helm/software/helena/src/automath/autCrg.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 U  = NUri
  13 module UH = U.UriHash
  14 module C  = Cps
  15 module G  = Options
  16 module N  = Level
  17 module E  = Entity
  18 module S  = Status
  19 module A  = Aut
  20 module D  = Crg
  21
  22 (* qualified identifier: uri, name, qualifiers *)
  23 type qid = D.uri * D.id * D.id list
  24
  25 type context_node = qid option (* context node: None = root *)
  26
  27 type status = {
  28    path: D.id list;          (* current section path *)
  29    node: context_node;       (* current context node *)
  30    nodes: context_node list; (* context node list *)
  31    line: int;                (* line number *)
  32    mk_uri: G.uri_generator;  (* uri generator *)
  33 }
  34
  35 let henv_size, hcnt_size = 7000, 4300 (* hash tables initial sizes *)
  36
  37 let henv = UH.create henv_size (* optimized global environment *)
  38
  39 let hcnt = UH.create hcnt_size (* optimized context *)
  40
  41 (* Internal functions *******************************************************)
  42
  43 let empty_cnt = D.empty_lenv
  44
  45 let alpha id =
  46    if id.[0] >= '0' && id.[0] <= '9' then !G.alpha ^ id else id
  47
  48 let add_abst cnt id aw w =
  49    let id = if !G.alpha <> "" then alpha id else id in
  50    let aw = {aw with E.n_name = Some (id, true); E.n_degr = succ aw.E.n_degr} in
  51    D.EBind (cnt, aw, D.Abst (N.two, w))
  52
  53 let mk_lref f a i = f a (D.TLRef (a, i))
  54
  55 let id_of_name (id, _, _) =
  56    if !G.alpha <> "" then alpha id else id
  57
  58 let mk_qid f lst id path =
  59    let str = String.concat "/" path in
  60    let str = Filename.concat str id in
  61    let str = lst.mk_uri str in
  62    f (U.uri_of_string str, id, path)
  63
  64 let uri_of_qid (uri, _, _) = uri
  65
  66 let complete_qid f lst (id, is_local, qs) =
  67    let f path = C.list_rev_append (mk_qid f lst id) path ~tail:qs in
  68    let rec skip f = function
  69       | phd :: ptl, qshd :: _ when phd = qshd -> f ptl
  70       | _ :: ptl, _ :: _                      -> skip f (ptl, qs)
  71       | _                                     -> f []
  72    in
  73    if is_local then f lst.path else skip f (lst.path, qs)
  74
  75 let relax_qid f lst (_, id, path) =
  76    let f = function
  77       | _ :: tl -> C.list_rev (mk_qid f lst id) tl
  78       | []      -> assert false
  79    in
  80    C.list_rev f path
  81
  82 let relax_opt_qid f lst = function
  83    | None     -> f None
  84    | Some qid -> let f qid = f (Some qid) in relax_qid f lst qid
  85
  86 let resolve_gref err f lst qid =
  87    try let a, cnt = UH.find henv (uri_of_qid qid) in f qid a cnt
  88    with Not_found -> err qid
  89
  90 let resolve_gref_relaxed f lst qid =
  91 (* this is not tail recursive *)
  92    let rec err qid = relax_qid (resolve_gref err f lst) lst qid in
  93    resolve_gref err f lst qid
  94
  95 let get_cnt err f lst = function
  96    | None             -> f empty_cnt
  97    | Some qid as node ->
  98       try let cnt = UH.find hcnt (uri_of_qid qid) in f cnt
  99       with Not_found -> err node
 100
 101 let get_cnt_relaxed f lst =
 102 (* this is not tail recursive *)
 103    let rec err node = relax_opt_qid (get_cnt err f lst) lst node in
 104    get_cnt err f lst lst.node
 105
 106 let push_abst f a w lenv =
 107    let bw = D.Abst (N.infinite, w) in
 108    D.push_bind f a bw lenv
 109
 110 let add_proj e t = match e with
 111    | D.ESort                 -> t
 112    | D.EBind (D.ESort, a, b) -> D.TBind (a, b, t)
 113    | _                       -> D.TProj (E.empty_node, e, t)
 114
 115 (* this is not tail recursive in the GRef branch *)
 116 let rec xlate_term f st lst y lenv = function
 117    | A.Sort s            ->
 118       let h = if s then 0 else 1 in
 119       let a = E.node_attrs ~sort:h () in
 120       f a (D.TSort (a, h))
 121    | A.Appl (v, t)       ->
 122       let f vv at tt = f at (D.TAppl (at, vv, tt)) in
 123       let f _ vv = xlate_term (f vv) st lst y lenv t in
 124       xlate_term f st lst false lenv v
 125    | A.Abst (name, w, t) ->
 126       let name = if !G.alpha <> "" then alpha name else name in
 127       let name = Some (name, true) in
 128       let f aw ww =
 129          let f at tt =
 130             let l = if !G.cc then match y, at.E.n_degr with
 131                | true, _ -> N.one
 132                | _   , 0 -> N.one
 133                | _   , 1 -> N.unknown st.S.lenv
 134                | _   , 2 -> N.two
 135                | _       -> assert false
 136                else N.infinite
 137             in
 138             let b = D.Abst (l, ww) in
 139             let at = {at with E.n_name = name} in
 140             f at (D.TBind (at, b, tt))
 141          in
 142          let f lenv = xlate_term f st lst y lenv t in
 143          push_abst f {aw with E.n_name = name; E.n_degr = succ aw.E.n_degr} ww lenv
 144       in
 145       xlate_term f st lst true lenv w
 146    | A.GRef (name, args) ->
 147       let map1 args (id, _) = A.GRef ((id, true, []), []) :: args in
 148       let map2 f arg args =
 149          let f _ arg = f (D.EAppl (args, E.empty_node, arg)) in
 150          xlate_term f st lst false lenv arg
 151       in
 152       let g qid a cnt =
 153          let gref = D.TGRef (a, uri_of_qid qid) in
 154          if cnt = D.ESort then f a gref else
 155          let f = function
 156             | D.EAppl (D.ESort, _, v) -> f a (D.TAppl (a, v, gref))
 157             | args                    -> f a (D.TProj (a, args, gref))
 158          in
 159          let f args = C.list_fold_right f map2 args D.ESort in
 160          D.sub_list_strict (D.fold_names f map1 args) cnt args
 161       in
 162       let g qid = resolve_gref_relaxed g lst qid in
 163       let err () = complete_qid g lst name in
 164       D.resolve_lref err (mk_lref f) (id_of_name name) lenv
 165
 166 let xlate_entity err f st lst = function
 167    | A.Section (Some (_, name))     ->
 168       err {lst with path = name :: lst.path; nodes = lst.node :: lst.nodes}
 169    | A.Section None            ->
 170       begin match lst.path, lst.nodes with
 171          | _ :: ptl, nhd :: ntl ->
 172             err {lst with path = ptl; node = nhd; nodes = ntl}
 173          | _                    -> assert false
 174       end
 175    | A.Context None            ->
 176       err {lst with node = None}
 177    | A.Context (Some name)     ->
 178       let f name = err {lst with node = Some name} in
 179       complete_qid f lst name
 180    | A.Block (name, w)         ->
 181       let f qid =
 182          let f cnt =
 183             let f aw ww =
 184                UH.add hcnt (uri_of_qid qid) (add_abst cnt name aw ww);
 185                err {lst with node = Some qid}
 186             in
 187             xlate_term f st lst true cnt w
 188          in
 189          get_cnt_relaxed f lst
 190       in
 191       complete_qid f lst (name, true, [])
 192    | A.Decl (name, w)          ->
 193       let f lenv =
 194          let f qid =
 195             let f aw ww =
 196                let aw = {aw with E.n_apix = lst.line; E.n_degr = succ aw.E.n_degr} in
 197                UH.add henv (uri_of_qid qid) (aw, lenv);
 198                let t = add_proj lenv ww in
 199 (*
 200             print_newline (); CrgOutput.pp_term print_string t;
 201 *)
 202                let b = E.Abst t in
 203                let entity = E.empty_root, aw, uri_of_qid qid, b in
 204                f {lst with line = succ lst.line} entity
 205             in
 206             xlate_term f st lst true lenv w
 207          in
 208          complete_qid f lst (name, true, [])
 209       in
 210       get_cnt_relaxed (D.sta f) lst
 211    | A.Def (name, w, trans, v) ->
 212       let f lenv =
 213          let f qid =
 214             let f _ ww =
 215                let f av vv =
 216                   let na = {av with E.n_apix = lst.line} in
 217                   UH.add henv (uri_of_qid qid) (na, lenv);
 218                   let t = add_proj lenv (D.TCast (na, ww, vv)) in
 219 (*
 220             print_newline (); CrgOutput.pp_term print_string t;
 221 *)
 222                   let b = E.Abbr t in
 223                   let ra = if trans then E.empty_root else E.root_attrs ~meta:[E.Private] () in
 224                   let entity = ra, na, uri_of_qid qid, b in
 225                   f {lst with line = succ lst.line} entity
 226                in
 227                xlate_term f st lst false lenv v
 228             in
 229             xlate_term f st lst true lenv w
 230          in
 231          complete_qid f lst (name, true, [])
 232       in
 233       get_cnt_relaxed f lst
 234
 235 (* Interface functions ******************************************************)
 236
 237 let initial_status () =
 238    UH.clear henv; UH.clear hcnt; {
 239    path = []; node = None; nodes = []; line = 1; mk_uri = G.get_mk_uri ();
 240 }
 241
 242 let refresh_status lst = {lst with
 243    mk_uri = G.get_mk_uri ()
 244 }
 245
 246 let crg_of_aut = xlate_entity