helm/software/lambda-delta/basic_ag/bagReduction.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 C = Cps
  14 module L = Log
  15 module B = Bag
  16 module O = BagOutput
  17 module E = BagEnvironment
  18 module S = BagSubstitution
  19
  20 exception TypeError of B.message
  21
  22 type machine = {
  23    i: int;
  24    c: B.context;
  25    s: B.term list
  26 }
  27
  28 type whd_result =
  29    | Sort_ of int
  30    | LRef_ of int * B.term option
  31    | GRef_ of B.obj
  32    | Bind_ of int * B.id * B.term * B.term
  33
  34 type ho_whd_result =
  35    | Sort of int
  36    | Abst of B.term
  37
  38 (* Internal functions *******************************************************)
  39
  40 let term_of_whdr = function
  41    | Sort_ h             -> B.Sort h
  42    | LRef_ (i, _)        -> B.LRef i
  43    | GRef_ (_, uri, _)   -> B.GRef uri
  44    | Bind_ (l, id, w, t) -> B.bind_abst l id w t
  45
  46 let level = 5
  47
  48 let error i =
  49    let s = Printf.sprintf "local reference not found %u" i in
  50    raise (TypeError (L.items1 s))
  51
  52 let log1 s c t =
  53    let sc, st = s ^ " in the context", "the term" in
  54    L.log O.specs level (L.ct_items1 sc c st t)
  55
  56 let log2 s cu u ct t =
  57    let s1, s2, s3 = s ^ " in the context", "the term", "and in the context" in
  58    L.log O.specs level (L.ct_items2 s1 cu s2 u s3 ct s2 t)
  59
  60 let empty_machine = {i = 0; c = B.empty_context; s = []}
  61
  62 let inc m = {m with i = succ m.i}
  63
  64 let unwind_to_term f m t =
  65    let map f t (l, id, b) = f (B.Bind (l, id, b, t)) in
  66    let f mc = C.list_fold_left f map t mc in
  67    B.contents f m.c
  68
  69 let unwind_stack f m =
  70    let map f v = unwind_to_term f m v in
  71    C.list_map f map m.s
  72
  73 let get f c m i =
  74    let f = function
  75       | Some (_, b) -> f b
  76       | None        -> error i
  77    in
  78    let f c = B.get f c i in
  79    B.append f c m.c
  80
  81 let push msg f c m l id w =
  82    assert (m.s = []);
  83    let f w = B.push msg f c l id (B.Abst w) in
  84    unwind_to_term f m w
  85
  86 (* to share *)
  87 let rec whd f c m x =
  88 (*   L.warn "entering R.whd"; *)
  89    match x with
  90    | B.Sort h                    -> f m (Sort_ h)
  91    | B.GRef uri                  ->
  92       let f obj = f m (GRef_ obj) in
  93       E.get_obj f uri
  94    | B.LRef i                    ->
  95       let f = function
  96          | B.Void   -> f m (LRef_ (i, None))
  97          | B.Abst t -> f m (LRef_ (i, Some t))
  98          | B.Abbr t -> whd f c m t
  99       in
 100       get f c m i
 101    | B.Cast (_, t)               -> whd f c m t
 102    | B.Appl (v, t)               -> whd f c {m with s = v :: m.s} t
 103    | B.Bind (l, id, B.Abst w, t) ->
 104       begin match m.s with
 105          | []      -> f m (Bind_ (l, id, w, t))
 106          | v :: tl ->
 107             let nl = B.new_location () in
 108             let f mc = S.subst (whd f c {m with c = mc; s = tl}) nl l t in
 109             B.push "!" f m.c nl id (B.Abbr (B.Cast (w, v)))
 110       end
 111    | B.Bind (l, id, b, t)         ->
 112       let nl = B.new_location () in
 113       let f mc = S.subst (whd f c {m with c = mc}) nl l t in
 114       B.push "!" f m.c nl id b
 115
 116 (* Interface functions ******************************************************)
 117
 118 let rec ho_whd f c m x =
 119 (*   L.warn "entering R.ho_whd"; *)
 120    let aux m = function
 121       | Sort_ h                -> f (Sort h)
 122       | Bind_ (_, _, w, _)     ->
 123          let f w = f (Abst w) in unwind_to_term f m w
 124       | LRef_ (_, Some w)      -> ho_whd f c m w
 125       | GRef_ (_, _, B.Abst w) -> ho_whd f c m w
 126       | GRef_ (_, _, B.Abbr v) -> ho_whd f c m v
 127       | LRef_ (_, None)        -> assert false
 128       | GRef_ (_, _, B.Void)   -> assert false
 129    in
 130    whd aux c m x
 131
 132 let ho_whd f c t =
 133    let f r = L.unbox level; f r in
 134    L.box level; log1 "Now scanning" c t;
 135    ho_whd f c empty_machine t
 136
 137 let rec are_convertible f ~si a c m1 t1 m2 t2 =
 138 (*   L.warn "entering R.are_convertible"; *)
 139    let rec aux m1 r1 m2 r2 =
 140 (*   L.warn "entering R.are_convertible_aux"; *)
 141    let u, t = term_of_whdr r1, term_of_whdr r2 in
 142    log2 "Now really converting" c u c t;
 143    match r1, r2 with
 144       | Sort_ h1, Sort_ h2                                 ->
 145          if h1 = h2 then f a else f false
 146       | LRef_ (i1, _), LRef_ (i2, _)                       ->
 147          if i1 = i2 then are_convertible_stacks f ~si a c m1 m2 else f false
 148       | GRef_ (a1, _, B.Abst _), GRef_ (a2, _, B.Abst _)   ->
 149          if a1 = a2 then are_convertible_stacks f ~si a c m1 m2 else f false
 150       | GRef_ (a1, _, B.Abbr v1), GRef_ (a2, _, B.Abbr v2) ->
 151          if a1 = a2 then
 152             let f a =
 153                if a then f a else are_convertible f ~si true c m1 v1 m2 v2
 154             in
 155             are_convertible_stacks f ~si a c m1 m2
 156          else
 157          if a1 < a2 then whd (aux m1 r1) c m2 v2 else
 158          whd (aux_rev m2 r2) c m1 v1
 159       | _, GRef_ (_, _, B.Abbr v2)                         ->
 160          whd (aux m1 r1) c m2 v2
 161       | GRef_ (_, _, B.Abbr v1), _                         ->
 162          whd (aux_rev m2 r2) c m1 v1
 163       | Bind_ (l1, id1, w1, t1), Bind_ (l2, id2, w2, t2)   ->
 164            let l = B.new_location () in
 165            let h c =
 166               let m1, m2 = inc m1, inc m2 in
 167               let f t1 = S.subst (are_convertible f ~si a c m1 t1 m2) l l2 t2 in
 168               S.subst f l l1 t1
 169          in
 170          let f r = if r then push "!" h c m1 l id1 w1 else f false in
 171          are_convertible f ~si a c m1 w1 m2 w2
 172 (* we detect the AUT-QE reduction rule for type/prop inclusion *)
 173       | Sort_ _, Bind_ (l2, id2, w2, t2) when si           ->
 174          let m1, m2 = inc m1, inc m2 in
 175          let f c = are_convertible f ~si a c m1 (term_of_whdr r1) m2 t2 in
 176          push "nsi" f c m2 l2 id2 w2
 177       | _                                                  -> f false
 178    and aux_rev m2 r2 m1 r1 = aux m1 r1 m2 r2 in
 179    let g m1 r1 = whd (aux m1 r1) c m2 t2 in
 180    if a = false then f false else whd g c m1 t1
 181
 182 and are_convertible_stacks f ~si a c m1 m2 =
 183 (*   L.warn "entering R.are_convertible_stacks"; *)
 184    let mm1, mm2 = {m1 with s = []}, {m2 with s = []} in
 185    let map f a v1 v2 = are_convertible f ~si a c mm1 v1 mm2 v2 in
 186    if List.length m1.s <> List.length m2.s then
 187       begin
 188 (*         L.warn (Printf.sprintf "Different lengths: %u %u"
 189             (List.length m1.s) (List.length m2.s)
 190          ); *)
 191          f false
 192       end
 193    else
 194       C.list_fold_left2 f map a m1.s m2.s
 195
 196 let are_convertible f ?(si=false) c u t =
 197    let f b = L.unbox level; f b in
 198    L.box level; log2 "Now converting" c u c t;
 199    are_convertible f ~si true c empty_machine u empty_machine t