(*
    ||M||  This file is part of HELM, an Hypertextual, Electronic        
    ||A||  Library of Mathematics, developed at the Computer Science     
    ||T||  Department, University of Bologna, Italy.                     
    ||I||                                                                
    ||T||  HELM is free software; you can redistribute it and/or         
    ||A||  modify it under the terms of the GNU General Public License   
    \   /  version 2 or (at your option) any later version.              
     \ /   This software is distributed as is, NO WARRANTY.              
      V_______________________________________________________________ *)

module KF = Filename
module KP = Printf

module U  = NUri
module C  = Cps
module G  = Options
module N  = Layer
module E  = Entity
module R  = Alpha
module B  = Brg

IFDEF MANAGER THEN

(* Internal functions *******************************************************)

let version = KP.sprintf "This file was generated by %s: do not edit" (G.version_string true)

let ok = ref true

let uris = ref []

let chunk = ref 0

let sub_och = ref stdout

let top_age = 7000

let size = 250

let base = "lp"

let ext_lp = ".elpi"
let ext_tj = ".mod"
let ext_tj_sig = ".sig"

let reserved = ["pi"; "sigma"; "nil"; "delay"; "in"; "with"; "resume"; "context"]

let alpha n =
   if List.mem n reserved then !G.alpha ^ n else n

let out_preamble och =
   let ich = open_in !G.preamble in
   let rec aux () = KP.fprintf och "%s\n" (input_line ich); aux () in
   try aux () with End_of_file -> close_in ich

let out_top_comment och msg =
   KP.fprintf och "%% %s\n\n" msg

let out_comment och msg =
   KP.fprintf och "%% %s\n" msg 

let out_clause och msg =
   KP.fprintf och "%s\n\n" msg 

let out_uri och u =
   let str = U.string_of_uri u in
   let rec aux i =
     let c = str.[i] in
     if c = '.' then () else begin 
        output_char och (if c = '/' then '_' else c);
        aux (succ i)
     end
   in
   let rec strip i n = 
      if n <= 0 then succ i else
      strip (String.index_from str (succ i) '/') (pred n)
   in
   aux (strip 0 3)

let out_name och a =
   let f n = function 
      | true  -> KP.fprintf och "%s" (alpha n)
      | false -> KP.fprintf och "_"
   in
   let err () = f "" false in
   E.name err f a

let rec out_term st e och = function
   | B.Sort k                        ->
      let sort = if k = 0 then "k+set" else if k = 1 then "k+prop" else assert false in
      KP.fprintf och "(sort %s)" sort
   | B.LRef (_, i)                   ->
      let _, _, _, y, b = B.get e i in
      KP.fprintf och "%a" out_name y
   | B.GRef (_, s)                   ->
      KP.fprintf och "%a" out_uri s
   | B.Cast (u, t)                   ->
      KP.fprintf och "(cast %a %a)" (out_term st e) u (out_term st e) t 
   | B.Appl (a, v, t)                ->
      let c = if a.E.a_rest then "appr" else "appx" in
      KP.fprintf och "(%s %a %a)" c (out_term st e) v (out_term st e) t
   | B.Bind (a, B.Abst (r, n, w), t) ->
      let a = R.alpha B.mem e a in
      let ee = B.push e B.empty E.empty_node a (B.abst r n w) in
      let c = if r then "prod" else "abst" in
      let l = match N.to_string st n with
         | "1" -> "l+1"
         | "2" -> "l+2"
         | _   -> ok := false; "?"
      in
      KP.fprintf och "(%s %s %a %a\\ %a)"
         c l (out_term st e) w out_name a (out_term st ee) t
   | B.Bind (a, B.Abbr v, t)         ->
      let a = R.alpha B.mem e a in
      let ee = B.push e B.empty E.empty_node a (B.abbr v) in
      KP.fprintf och "(abbr %a %a\\ %a)"
          (out_term st e) v out_name a (out_term st ee) t
   | B.Bind (_, B.Void, _)           -> C.err ()

(* elpi variant 1 ***********************************************************)

let output_entity_lp1 och st (_, na, u, b) =
   if na.E.n_apix <= !G.last then begin
   match b with
      | E.Abbr (_, v) ->
         KP.fprintf och "(gdef+1 c+%u %a\n %a\\\n" na.E.n_apix (out_term st B.empty) v out_uri u;
         uris := (true, u) :: !uris; !ok
      | E.Abst (_, w) ->
         KP.fprintf och "(gdec+1 c+%u %a\n%a\\\n" na.E.n_apix (out_term st B.empty) w out_uri u;
         uris := (false, u) :: !uris; !ok
      | E.Void        -> C.err ()
   end else !ok

let close_out_lp1 och () =
   let aux_sep _ = KP.fprintf och "%s" ")" in
   KP.fprintf och "%s" "gtop";   
   List.iter aux_sep !uris;
   out_clause och "\n\n.";
   close_out och

(* elpi variant 2 ***********************************************************)

let output_entity_lp2 och st (_, na, u, b) =
   if na.E.n_apix <= !G.last then begin
   match b with
      | E.Abbr (_, v) ->
         KP.fprintf och "g+line %a c+%u\n       %a\n.\n\n"
            out_uri u na.E.n_apix (out_term st B.empty) v;
         uris := (true, u) :: !uris; !ok
      | E.Abst (_, w) ->
         KP.fprintf och "g+line %a c+%u\n       %a\n.\n\n"
            out_uri u na.E.n_apix (out_term st B.empty) w;
         uris := (false, u) :: !uris; !ok
      | E.Void        -> C.err ()
   end else !ok

let close_out_lp2 och () =
   let aux_name (b, s) =
      let gde = if b then "gdef+2" else "gdec+2" in 
      KP.fprintf och "(%s %a\n" gde out_uri s
   in
   let aux_sep _ = KP.fprintf och "%s" ")" in
   if !G.first > 0 then begin
      let s = KP.sprintf "tv+c C T :- $lt C c+%u, !." !G.first in
      out_clause och s;
      out_clause och "tv+c C T :- tv+ T."
   end;
   out_clause och "main :- grundlagen.";
   out_clause och "grundlagen :- gv+";
   List.iter aux_name (List.rev !uris);
   KP.fprintf och "%s" "gtop";
   List.iter aux_sep !uris;
   out_clause och "\n\n.";
   close_out och

(* teyjus variant 2 *************************************************)

let append_out name = 
   open_out_gen [Open_append; Open_creat; Open_text] 0o666 name

let sub_close () = if !sub_och <> stdout then close_out !sub_och

let mk_name chunk =
   let dir = KF.concat !G.manager_dir base in
   let name = KP.sprintf "grundlagen%02u" chunk in
   dir, name

let output_entity_tj2 och st (_, na, u, b) =
   if na.E.n_apix <= !G.last then begin
   if pred na.E.n_apix mod size = 0 then begin
      sub_close ();
      incr chunk;
      let dir, name = mk_name !chunk in
      let soch = open_out (KF.concat dir name ^ ext_tj_sig) in
      out_preamble soch;
      out_top_comment soch version;
      out_clause soch (KP.sprintf "sig %s." name);
      out_clause soch "accum_sig grundlagen.";
      out_clause soch (KP.sprintf "type line+%02u t -> int -> t -> o." !chunk);
      close_out soch;
      let soch = open_out (KF.concat dir name ^ ext_tj) in
      out_preamble soch;
      out_top_comment soch version;
      out_clause soch (KP.sprintf "module %s." name);
      sub_och := soch
   end;
   out_comment !sub_och (KP.sprintf "constant %u" na.E.n_apix); 
   match b with
      | E.Abbr (_, v) ->
         KP.fprintf !sub_och "line+%02u %a %u\n        %a\n.\n\n"
            !chunk out_uri u (top_age - na.E.n_apix) (out_term st B.empty) v;
         uris := (true, u) :: !uris; !ok
      | E.Abst (_, w) ->
         KP.fprintf !sub_och "line+%02u %a %u\n        %a\n.\n\n"
            !chunk out_uri u (top_age - na.E.n_apix) (out_term st B.empty) w;
         uris := (false, u) :: !uris; !ok
      | E.Void        -> C.err ()
   end else !ok

let close_out_tj2 och () =
   let out_name och (b, u) =
      let gde = if b then "gdef+2" else "gdec+2" in 
      KP.fprintf och "(%s %a\n" gde out_uri u
   in
   let rec out_pars och p =
      if p > 0 then begin KP.fprintf och "%s" ")"; out_pars och (pred p) end
   in
   let rec out_list chunk pars first items uris = match first, items, uris with
      | true, _, _         ->
         let dir, name = mk_name chunk in
         let soch = append_out (KF.concat dir name ^ ext_tj) in
         out_clause soch (KP.sprintf "g+line R C T :- line+%02u R C T, !." chunk);
         KP.fprintf soch "g+list %u\n" chunk;
         sub_och := soch;
         out_list (succ chunk) pars false items uris
      | false, _, []       ->
         KP.fprintf !sub_och "gtop%a.\n\n" out_pars pars;
         sub_close ()
      | false, 0, _        ->
         KP.fprintf !sub_och "(genv %u)%a.\n\n" chunk out_pars pars;
         sub_close ();
         out_list chunk 0 true size uris
      | false, _, hd :: tl ->
         out_name !sub_och hd; out_list chunk (succ pars) false (pred items) tl
   in
   let rec out_accumulate c =
      if !chunk < c then KP.fprintf och "\n" else begin
         let _, name = mk_name c in
         KP.fprintf och "accumulate %s.\n" name;
         out_accumulate (succ c)
      end
   in
   sub_close ();
   out_list 1 0 true size (List.rev !uris);
   KP.fprintf och "accumulate helena.\n";
   out_accumulate 1;
   out_clause och "main :- grundlagen.";
   out_clause och "grundlagen :- gv+ (genv 1).";
   close_out och

(* teyjus variant 3 *************************************************)

let output_entity_tj3 och st (_, na, u, b) =
   if na.E.n_apix <= !G.last then begin
      out_comment och (KP.sprintf "constant %u" na.E.n_apix);
      let age = top_age - na.E.n_apix in
      match b with
         | E.Abbr (_, v) ->
            KP.fprintf och "g+line %a %u\n       %a\n.\n\n"
               out_uri u age (out_term st B.empty) v;
            KP.fprintf och "tv+ %a.\n\n" out_uri u;
            KP.fprintf och "r+exp %a M C E M V :- g+line %a C V.\n\n"
               out_uri u out_uri u;
            uris := (true, u) :: !uris; !ok
         | E.Abst (_, w) ->
            KP.fprintf och "g+line %a %u\n       %a\n.\n\n"
               out_uri u age (out_term st B.empty) w;
            KP.fprintf och "tv+ %a.\n\n" out_uri u;
            KP.fprintf och "r+exp %a M1 C E M2 W :- m+pred M1 M2, g+line %a C W.\n\n"
               out_uri u out_uri u;
            uris := (false, u) :: !uris; !ok
         | E.Void        -> C.err ()
   end else !ok

let close_out_tj3 och () =
   let out_name (_, u) =
      KP.fprintf och "gv+3 %a,\n" out_uri u
   in
   let rec out_list och chunk first items uris = match first, items, uris with
      | true, _, _         ->
         KP.fprintf och "chunk %u :-\n" chunk;
         out_list och (succ chunk) false items uris
      | false, _, []       -> KP.fprintf och "!.\n\n"; chunk
      | false, 0, _        ->
         KP.fprintf och "!.\n\n";
         out_list och chunk true size uris
      | false, _, hd :: tl ->
         out_name hd; out_list och chunk false (pred items) tl
   in
   let rec out_chunks och chunks c =
      if chunks < c then out_clause och "!." else begin
         KP.fprintf och "chunk %u,\n" c; out_chunks och chunks (succ c)
      end
   in
   let chunks = out_list och 1 true size (List.rev !uris) in
   out_clause och "main :- grundlagen.";
   KP.fprintf och "grundlagen :-\n";
   out_chunks och (pred chunks) 1;
   close_out och

(* Interface functions ******************************************************)

let open_out_lp1 fname =
   let dir = KF.concat !G.manager_dir base in 
   let path = KF.concat dir fname in
   let och = open_out (path ^ "1" ^ ext_lp) in
   out_preamble och;
   out_top_comment och version;
   out_clause och "accumulate helena.";
   out_clause och "main :- grundlagen.";
   out_clause och "grundlagen :- gv+";
   output_entity_lp1 och, close_out_lp1 och

let open_out_lp2 fname =
   let dir = KF.concat !G.manager_dir base in 
   let path = KF.concat dir fname in
   let och = open_out (path ^ "2" ^ ext_lp) in
   out_preamble och;
   out_top_comment och version;
   out_clause och "accumulate helena.";
   output_entity_lp2 och, close_out_lp2 och

let open_out_tj2 fname =
   let dir = KF.concat !G.manager_dir base in
   let path = KF.concat dir fname ^ "2" in
   let och = open_out (path ^ ext_tj) in
   out_preamble och;
   out_top_comment och version;
   out_clause och "module grundlagen.";
   output_entity_tj2 och, close_out_tj2 och

let open_out_tj3 fname =
   let dir = KF.concat !G.manager_dir base in 
   let path = KF.concat dir fname in
   let och = open_out (path ^ "3" ^ ext_tj) in
   out_preamble och;
   out_top_comment och version;
   out_clause och "module grundlagen.";
   out_clause och "accumulate helena.";
   output_entity_tj3 och, close_out_tj3 och

END