V_______________________________________________________________ *)
module U = NUri
+module C = Cps
+module L = Log
+module P = Output
module B = Brg
+module O = BrgOutput
module E = BrgEnvironment
-exception LRefNotFound of string Lazy.t
+exception TypeError of B.message
-type environment = int * (B.bind * B.term) list
+type machine = {
+ c: B.context;
+ s: (B.context * B.term) list;
+ i: int
+}
-type stack = B.term list
+(* Internal functions *******************************************************)
-type context = {
- g: environment;
- l: environment;
- s: stack
-}
+let level = 5
-type whd_result =
- | Sort_ of int
- | LRef_ of int
- | GRef_ of int * U.uri * B.bind
- | Abst_ of B.term
+let log1 s c t =
+ let sc, st = s ^ " in the context", "the term" in
+ L.log O.specs level (L.ct_items1 sc c st t)
-(* Internal functions *******************************************************)
+let log2 s cu u ct t =
+ let s1, s2, s3 = s ^ " in the context", "the term", "and in the context" in
+ L.log O.specs level (L.ct_items2 s1 cu s2 u s3 ct s2 t)
-let push_e f b t (l, e) =
- f (succ l, (b, t) :: e)
+let error0 i =
+ let s = Printf.sprintf "local reference not found %u" i in
+ raise (TypeError (L.items1 s))
-let rec find_e f c i =
- let (gl, ge), (ll, le) = c.g, c.l in
- if i >= gl + ll then raise (LRefNotFound (lazy (string_of_int i)));
- let b, t =
- if i < gl then List.nth ge (gl - (succ i))
- else List.nth le (gl + ll - (succ i))
+let error1 st c t =
+ let sc = "In the context" in
+ raise (TypeError (L.ct_items1 sc c st t))
+
+let error3 c t1 t2 t3 =
+ let sc, st1, st2, st3 =
+ "In the context", "the term", "is of type", "but must be of type"
in
- f t b
-
-let rec whd f c t = match t with
- | B.Sort h -> f c t (Sort_ h)
- | B.GRef uri ->
- let f (i, _, b, t) = f c t (GRef_ (i, uri, b)) in
- E.get_obj f uri
- | B.LRef i ->
- let f t = function
- | B.Abst -> f c t (LRef_ i)
- | B.Abbr -> whd f c t
+ raise (TypeError (L.ct_items3 sc c st1 t1 st2 t2 st3 t3))
+
+let err () = assert false
+
+let get f m i =
+ B.get error0 f m.c i
+
+(* to share *)
+let rec step f ?(delta=false) ?(rt=false) m x =
+(* L.warn "entering R.step"; *)
+ match x with
+ | B.Sort _ -> f m None x
+ | B.GRef (a, uri) ->
+ let f = function
+ | _, _, B.Abbr v when delta ->
+ P.add ~gdelta:1 ();
+ step f ~delta ~rt m v
+ | _, _, B.Abst w when rt ->
+ P.add ~grt:1 ();
+ step f ~delta ~rt m w
+ | e, _, b ->
+ f m (Some (e, b)) x
in
- find_e f c i
- | B.Appl (v, t) -> whd f {c with s = v :: c.s} t
- | B.Bind (_, B.Abbr, v, t) ->
- let f l = whd f {c with l = l} t in
- push_e f B.Abbr v c.l
- | B.Bind (_, B.Abst, w, t) ->
- begin match c.s with
- | [] -> f c t (Abst_ w)
- | v :: tl ->
- let f tl l = whd f {c with l = l; s = tl} t in
- push_e (f tl) B.Abbr v c.l
+ E.get_obj f uri
+ | B.LRef (a, i) ->
+ let f c a = function
+ | B.Abbr v ->
+ P.add ~ldelta:1 ();
+ step f ~delta ~rt {m with c = c} v
+ | B.Abst w when rt ->
+ P.add ~lrt:1 ();
+ step f ~delta ~rt {m with c = c} w
+ | B.Void ->
+ f {m with c = c} None x
+ | b ->
+ let f e = f {m with c = c} (Some (e, b)) x in
+ B.apix err f a
+ in
+ get f m i
+ | B.Cast (_, _, t) ->
+ P.add ~tau:1 ();
+ step f ~delta ~rt m t
+ | B.Appl (_, v, t) ->
+ step f ~delta ~rt {m with s = (m.c, v) :: m.s} t
+ | B.Bind (a, B.Abst w, t) ->
+ begin match m.s with
+ | [] -> f m None x
+ | (c, v) :: s ->
+ P.add ~beta:1 ~upsilon:(List.length s) ();
+ let f c = step f ~delta ~rt {m with c = c; s = s} t in
+ B.push f m.c ~c a (B.Abbr v) (* (B.Cast ([], w, v)) *)
end
- | B.Cast (_, t) -> whd f c t
+ | B.Bind (a, b, t) ->
+ P.add ~upsilon:(List.length m.s) ();
+ let f c = step f ~delta ~rt {m with c = c} t in
+ B.push f m.c ~c:m.c a b
+
+let domain f m t =
+ let f r = L.unbox level; f r in
+ let f m _ = function
+ | B.Bind (_, B.Abst w, _) -> f m w
+ | _ -> error1 "not a function" m.c t
+ in
+ L.box level; log1 "Now scanning" m.c t;
+ step f ~delta:true ~rt:true m t
+
+let push f m a b =
+ assert (m.s = []);
+ let a, i = match b with
+ | B.Abst _ -> B.Apix m.i :: a, succ m.i
+ | _ -> a, m.i
+ in
+ let f c = f {m with c = c; i = i} in
+ B.push f m.c ~c:m.c a b
+
+let rec ac_nfs f ~si r m1 a1 u m2 a2 t =
+ log2 "Now converting nfs" m1.c u m2.c t;
+ match a1, u, a2, t with
+ | _, B.Sort (_, h1),
+ _, B.Sort (_, h2) ->
+ if h1 = h2 then f r else f false
+ | Some (e1, B.Abst _), _, Some (e2, B.Abst _), _ ->
+ if e1 = e2 then ac_stacks f ~si r m1 m2 else f false
+ | Some (e1, B.Abbr v1), _, Some (e2, B.Abbr v2), _ ->
+ if e1 = e2 then
+ let f r =
+ if r then f r
+ else begin
+ P.add ~gdelta:2 ();
+ ac f ~si true m1 v1 m2 v2
+ end
+ in
+ ac_stacks f ~si r m1 m2
+ else if e1 < e2 then begin
+ P.add ~gdelta:1 ();
+ step (ac_nfs f ~si r m1 a1 u) m2 v2
+ end else begin
+ P.add ~gdelta:1 ();
+ step (ac_nfs_rev f ~si r m2 a2 t) m1 v1
+ end
+ | _, _, Some (_, B.Abbr v2), _ ->
+ P.add ~gdelta:1 ();
+ step (ac_nfs f ~si r m1 a1 u) m2 v2
+ | Some (_, B.Abbr v1), _, _, _ ->
+ P.add ~gdelta:1 ();
+ step (ac_nfs_rev f ~si r m2 a2 t) m1 v1
+ | _, B.Bind (a1, (B.Abst w1 as b1), t1),
+ _, B.Bind (a2, (B.Abst w2 as b2), t2) ->
+ let g m1 m2 = ac f ~si r m1 t1 m2 t2 in
+ let g m1 = push (g m1) m2 a2 b2 in
+ let f r = if r then push g m1 a1 b1 else f false in
+ ac f ~si r m1 w1 m2 w2
+ | _, B.Sort _, _, B.Bind (a, b, t) when si ->
+ P.add ~si:1 ();
+ let f m1 m2 = ac f ~si r m1 u m2 t in
+ let f m1 = push (f m1) m2 a b in
+ push f m1 a b
+ | _ -> f false
+
+and ac_nfs_rev f ~si r m2 a2 t m1 a1 u = ac_nfs f ~si r m1 a1 u m2 a2 t
+
+and ac f ~si r m1 t1 m2 t2 =
+(* L.warn "entering R.are_convertible"; *)
+ let g m1 a1 t1 = step (ac_nfs f ~si r m1 a1 t1) m2 t2 in
+ if r = false then f false else step g m1 t1
+
+and ac_stacks f ~si r m1 m2 =
+(* L.warn "entering R.are_convertible_stacks"; *)
+ if List.length m1.s <> List.length m2.s then f false else
+ let map f r (c1, v1) (c2, v2) =
+ let m1, m2 = {m1 with c = c1; s = []}, {m2 with c = c2; s = []} in
+ ac f ~si r m1 v1 m2 v2
+ in
+ C.list_fold_left2 f map r m1.s m2.s
(* Interface functions ******************************************************)
-let push f c b t =
- assert (fst c.l = 0 && c.s = []);
- let f g = f {c with g = g} in
- push_e f b t c.g
+let empty_machine = {
+ c = B.empty_context; s = []; i = 0
+}
+
+let get f m i =
+ assert (m.s = []);
+ let f c = f in
+ get f m i
+
+let assert_conversion f ?(si=false) ?(rt=false) mw u w v =
+ let f b = L.unbox level; f b in
+ let f mu u =
+ let f = function
+ | true -> f ()
+ | false -> error3 mw.c v w u
+ in
+ L.box level; log2 "Now converting" mu.c u mw.c w;
+ ac f ~si true mu u mw w
+ in
+ if rt then domain f mw u else f mw u
+
+let message1 st m t =
+ L.ct_items1 "In the context" m.c st t