helm/software/helena/src/basic_rg/brgELPI.ml

   1 (*
   2     ||M||  This file is part of HELM, an Hypertextual, Electronic
   3     ||A||  Library of Mathematics, developed at the Computer Science
   4     ||T||  Department, University of Bologna, Italy.
   5     ||I||
   6     ||T||  HELM is free software; you can redistribute it and/or
   7     ||A||  modify it under the terms of the GNU General Public License
   8     \   /  version 2 or (at your option) any later version.
   9      \ /   This software is distributed as is, NO WARRANTY.
  10       V_______________________________________________________________ *)
  11
  12 module KF = Filename
  13 module KP = Printf
  14
  15 module U  = NUri
  16 module C  = Cps
  17 module G  = Options
  18 module N  = Layer
  19 module E  = Entity
  20 module R  = Alpha
  21 module B  = Brg
  22
  23 (* Internal functions *******************************************************)
  24
  25 let ok = ref true
  26
  27 let level = ref 0
  28
  29 let base = "elpi"
  30
  31 let ext = ".elpi"
  32
  33 let reserved = ["pi"; "sigma"; "nil"; "delay"; "in"; "with"; "resume"; "context"]
  34
  35 let alpha n =
  36    if List.mem n reserved then !G.alpha ^ n else n
  37
  38 let out_preamble och =
  39    let ich = open_in !G.preamble in
  40    let rec aux () = KP.fprintf och "%s\n" (input_line ich); aux () in
  41    try aux () with End_of_file -> close_in ich
  42
  43 let out_top_comment och msg =
  44    KP.fprintf och "%% %s\n\n" msg
  45
  46 let out_comment och msg =
  47    KP.fprintf och "%% %s\n" msg
  48
  49 let out_clause och msg =
  50    KP.fprintf och "%s\n\n" msg
  51
  52 let out_uri och u =
  53    let str = U.string_of_uri u in
  54    let rec aux i =
  55      let c = str.[i] in
  56      if c = '.' then () else begin
  57         output_char och (if c = '/' then '_' else c);
  58         aux (succ i)
  59      end
  60    in
  61    let rec strip i n =
  62       if n <= 0 then succ i else
  63       strip (String.index_from str (succ i) '/') (pred n)
  64    in
  65    aux (strip 0 3)
  66
  67 let out_name och a =
  68    let f n = function
  69       | true  -> KP.fprintf och "%s" (alpha n)
  70       | false -> KP.fprintf och "_"
  71    in
  72    let err () = f "" false in
  73    E.name err f a
  74
  75 let rec out_term st e och = function
  76    | B.Sort (_, h)                   ->
  77       let sort = if h = 0 then "k+0" else if h = 1 then "k+1" else assert false in
  78       KP.fprintf och "(sort %s)" sort
  79    | B.LRef (_, i)                   ->
  80       let _, _, a, b = B.get e i in
  81       KP.fprintf och "%a" out_name a
  82    | B.GRef (_, s)                   ->
  83       KP.fprintf och "%a" out_uri s
  84    | B.Cast (_, u, t)                ->
  85       KP.fprintf och "(cast %a %a)" (out_term st e) u (out_term st e) t
  86    | B.Appl (_, v, t)                ->
  87       KP.fprintf och "(appr %a %a)" (out_term st e) v (out_term st e) t
  88    | B.Bind (a, B.Abst (x, n, w), t) ->
  89       let a = R.alpha B.mem e a in
  90       let ee = B.push e B.empty a (B.abst x n w) in
  91       let c = if x then "prod" else "abst" in
  92       let l = match N.to_string st n with
  93          | "1" -> "l+1"
  94          | "2" -> "l+2"
  95          | _   -> ok := false; "?"
  96       in
  97       KP.fprintf och "(%s %s %a %a\\ %a)"
  98          c l (out_term st e) w out_name a (out_term st ee) t
  99    | B.Bind (a, B.Abbr v, t)         ->
 100       let a = R.alpha B.mem e a in
 101       let ee = B.push e B.empty a (B.abbr v) in
 102       KP.fprintf och "(abbr %a %a\\ %a)"
 103           (out_term st e) v out_name a (out_term st ee) t
 104    | B.Bind (a, B.Void, t)   ->
 105       let a = R.alpha B.mem e a in
 106       let ee = B.push e B.empty a (B.Void) in
 107       KP.fprintf och "(void %a\\ %a)"
 108           out_name a (out_term st ee) t
 109
 110 let output_entity och st (_, na, s, b) =
 111 (*   if na.E.n_apix <= 4500 then begin *)
 112    out_comment och (KP.sprintf "constant %u" na.E.n_apix);
 113    match b with
 114       | E.Abbr t ->
 115          KP.fprintf och "(gdef c+%u %a\n %a\\\n" na.E.n_apix (out_term st B.empty) t out_uri s;
 116          incr level; !ok
 117       | E.Abst u ->
 118          KP.fprintf och "(gdec c+%u %a\n%a\\\n" na.E.n_apix (out_term st B.empty) u out_uri s;
 119          incr level; !ok
 120       | E.Void   -> C.err ()
 121 (*   end else !ok *)
 122
 123 let close_out och () =
 124    let rec aux () =
 125       if !level <= 0 then KP.fprintf och "%s" "\n\n).\n"
 126       else begin KP.fprintf och "%s" ")"; decr level; aux () end
 127    in
 128    KP.fprintf och "%s" "gtop";
 129    aux (); close_out och
 130
 131 (* Interface functions ******************************************************)
 132
 133 let open_out fname =
 134    let dir = KF.concat !G.manager_dir base in
 135    let path = KF.concat dir fname in
 136    let och = open_out (path ^ ext) in
 137    out_preamble och;
 138    out_top_comment och (KP.sprintf "This file was generated by %s: do not edit" G.version_string);
 139    out_clause och "main :- grundlagen.";
 140    out_clause och "grundlagen :- gv+ (";
 141    output_entity och, close_out och