helm/software/lambda-delta/dual_rg/drgAut.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 H = U.UriHash
  14 module C = Cps
  15 module Y = Entity
  16 module A = Aut
  17 module D = Drg
  18
  19 (* qualified identifier: uri, name, qualifiers *)
  20 type qid = D.uri * D.id * D.id list
  21
  22 type context = Y.attrs * D.term list
  23
  24 type environment = context H.t
  25
  26 type context_node = qid option (* context node: None = root *)
  27
  28 type status = {
  29    henv: environment;        (* optimized global environment *)
  30    path: D.id list;          (* current section path *)
  31    hcnt: environment;        (* optimized context *)
  32    node: context_node;       (* current context node *)
  33    nodes: context_node list; (* context node list *)
  34    line: int;                (* line number *)
  35    mk_uri:Y.uri_generator    (* uri generator *)
  36 }
  37
  38 type resolver = Local of int
  39               | Global of context
  40
  41 let henv_size, hcnt_size = 7000, 4300 (* hash tables initial sizes *)
  42
  43 (* Internal functions *******************************************************)
  44
  45 let initial_status mk_uri = {
  46    path = []; node = None; nodes = []; line = 1; mk_uri = mk_uri;
  47    henv = H.create henv_size; hcnt = H.create hcnt_size
  48 }
  49
  50 let empty_cnt = [], []
  51
  52 let add_abst (a, ws) id w =
  53    Y.Name (id, true) :: a, w :: ws
  54
  55 let lenv_of_cnt (a, ws) =
  56    D.push_bind C.start D.empty_lenv a (D.Abst ws)
  57
  58 let mk_lref f i j k = f (D.TLRef ([Y.Apix k], i, j))
  59
  60 let id_of_name (id, _, _) = id
  61
  62 let mk_qid f st id path =
  63    let str = String.concat "/" path in
  64    let str = Filename.concat str id in
  65    let f str = f (U.uri_of_string str, id, path) in
  66    f (st.mk_uri str)
  67
  68 let uri_of_qid (uri, _, _) = uri
  69
  70 let complete_qid f st (id, is_local, qs) =
  71    let f path = C.list_rev_append (mk_qid f st id) path ~tail:qs in
  72    let rec skip f = function
  73       | phd :: ptl, qshd :: _ when phd = qshd -> f ptl
  74       | _ :: ptl, _ :: _                      -> skip f (ptl, qs)
  75       | _                                     -> f []
  76    in
  77    if is_local then f st.path else skip f (st.path, qs)
  78
  79 let relax_qid f st (_, id, path) =
  80    let f = function
  81       | _ :: tl -> C.list_rev (mk_qid f st id) tl
  82       | []      -> assert false
  83    in
  84    C.list_rev f path
  85
  86 let relax_opt_qid f st = function
  87    | None     -> f None
  88    | Some qid -> let f qid = f (Some qid) in relax_qid f st qid
  89
  90 let resolve_gref err f st qid =
  91    try let cnt = H.find st.henv (uri_of_qid qid) in f qid cnt
  92    with Not_found -> err qid
  93
  94 let resolve_gref_relaxed f st qid =
  95 (* this is not tail recursive *)
  96    let rec err qid = relax_qid (resolve_gref err f st) st qid in
  97    resolve_gref err f st qid
  98
  99 let get_cnt err f st = function
 100    | None              -> f empty_cnt
 101    | Some qid as node ->
 102       try let cnt = H.find st.hcnt (uri_of_qid qid) in f cnt
 103       with Not_found -> err node
 104
 105 let get_cnt_relaxed f st =
 106 (* this is not tail recursive *)
 107    let rec err node = relax_opt_qid (get_cnt err f st) st node in
 108    get_cnt err f st st.node
 109
 110 (* this is not tail recursive in the GRef branch *)
 111 let rec xlate_term f st lenv = function
 112    | A.Sort s            ->
 113       let f h = f (D.TSort ([], h)) in
 114       if s then f 0 else f 1
 115    | A.Appl (v, t)       ->
 116       let f vv tt = f (D.TAppl ([], [vv], tt)) in
 117       let f vv = xlate_term (f vv) st lenv t in
 118       xlate_term f st lenv v
 119    | A.Abst (name, w, t) ->
 120       let f ww =
 121          let a, b = [Y.Name (name, true)], (D.Abst [ww]) in
 122          let f tt = f (D.TBind (a, b, tt)) in
 123          let f lenv = xlate_term f st lenv t in
 124          D.push_bind f lenv a b
 125       in
 126       xlate_term f st lenv w
 127    | A.GRef (name, args) ->
 128       let g qid (a, _) =
 129          let gref = D.TGRef ([], uri_of_qid qid) in
 130          let map1 f = xlate_term f st lenv in
 131          let map2 f = function
 132             | Y.Name (id, _) -> D.resolve_lref Cps.err (mk_lref f) id lenv
 133             | _              -> assert false
 134          in
 135          let f tail =
 136             let f = function
 137                | []   -> f gref
 138                | args -> f (D.TAppl ([], args, gref))
 139             in
 140             let f a = C.list_rev_map_append f map2 a ~tail in
 141             C.list_sub_strict f a args
 142          in
 143          C.list_map f map1 args
 144       in
 145       let g qid = resolve_gref_relaxed g st qid in
 146       let err () = complete_qid g st name in
 147       D.resolve_lref err (mk_lref f) (id_of_name name) lenv
 148
 149 let xlate_entity err f st = function
 150    | A.Section (Some (_, name))     ->
 151       err {st with path = name :: st.path; nodes = st.node :: st.nodes}
 152    | A.Section None            ->
 153       begin match st.path, st.nodes with
 154          | _ :: ptl, nhd :: ntl ->
 155             err {st with path = ptl; node = nhd; nodes = ntl}
 156          | _                    -> assert false
 157       end
 158    | A.Context None            ->
 159       err {st with node = None}
 160    | A.Context (Some name)     ->
 161       let f name = err {st with node = Some name} in
 162       complete_qid f st name
 163    | A.Block (name, w)         ->
 164       let f qid =
 165          let f cnt =
 166             let lenv = lenv_of_cnt cnt in
 167             let f ww =
 168                H.add st.hcnt (uri_of_qid qid) (add_abst cnt name ww);
 169                err {st with node = Some qid}
 170             in
 171             xlate_term f st lenv w
 172          in
 173          get_cnt_relaxed f st
 174       in
 175       complete_qid f st (name, true, [])
 176    | A.Decl (name, w)          ->
 177       let f cnt =
 178          let a, ws = cnt in
 179          let lenv = lenv_of_cnt cnt in
 180          let f qid =
 181             let f ww =
 182                H.add st.henv (uri_of_qid qid) cnt;
 183                let b = Y.Abst (D.TBind (a, D.Abst ws, ww)) in
 184                let entity = [Y.Mark st.line], uri_of_qid qid, b in
 185                f {st with line = succ st.line} entity
 186             in
 187             xlate_term f st lenv w
 188          in
 189          complete_qid f st (name, true, [])
 190       in
 191       get_cnt_relaxed f st
 192    | A.Def (name, w, trans, v) ->
 193       let f cnt =
 194          let a, ws = cnt in
 195          let lenv = lenv_of_cnt cnt in
 196          let f qid =
 197             let f ww vv =
 198                H.add st.henv (uri_of_qid qid) cnt;
 199                let b = Y.Abbr (D.TBind (a, D.Abst ws, D.TCast ([], ww, vv))) in
 200                let a =
 201                   if trans then [Y.Mark st.line] else [Y.Mark st.line; Y.Priv]
 202                in
 203                let entity = a, uri_of_qid qid, b in
 204                f {st with line = succ st.line} entity
 205             in
 206             let f ww = xlate_term (f ww) st lenv v in
 207             xlate_term f st lenv w
 208          in
 209          complete_qid f st (name, true, [])
 210       in
 211       get_cnt_relaxed f st
 212
 213 (* Interface functions ******************************************************)
 214
 215 let initial_status mk_uri =
 216    initial_status mk_uri
 217
 218 let drg_of_aut = xlate_entity