]> matita.cs.unibo.it Git - helm.git/blobdiff - helm/software/components/ng_refiner/nCicUnification.ml
First attempt to implement unification hints.
[helm.git] / helm / software / components / ng_refiner / nCicUnification.ml
index c324b0ec6d90b0360eb75ebebf6d7af4c9ed19fe..abbafc4583dc09b9d1553414cbc0de47bde2324a 100644 (file)
@@ -99,14 +99,14 @@ let fix_sorts swap metasenv subst context meta t =
     aux () t
 ;;
 
-let rec beta_expand num test_eq_only swap metasenv subst context t arg =
+let rec beta_expand hdb num test_eq_only swap metasenv subst context t arg =
   let rec aux (n,context,test_eq_only as k) (metasenv, subst as acc) t' =
    try
     let metasenv, subst =
      if swap then
-      unify test_eq_only metasenv subst context t' (NCicSubstitution.lift n arg)
+      unify hdb test_eq_only metasenv subst context t' (NCicSubstitution.lift n arg)
      else
-      unify test_eq_only metasenv subst context (NCicSubstitution.lift n arg) t'
+      unify hdb test_eq_only metasenv subst context (NCicSubstitution.lift n arg) t'
     in
      (metasenv, subst), NCic.Rel (1 + n)
    with Uncertain _ | UnificationFailure _ ->
@@ -141,7 +141,7 @@ let rec beta_expand num test_eq_only swap metasenv subst context t arg =
   with NCicTypeChecker.TypeCheckerFailure _ -> 
     metasenv, subst, NCic.Lambda ("_", argty, NCicSubstitution.lift 1 arg)
 
-and beta_expand_many test_equality_only swap metasenv subst context t args =
+and beta_expand_many hdb test_equality_only swap metasenv subst context t args =
 (* (*D*)  inside 'B'; try let rc = *)
   pp (lazy (String.concat ", "
      (List.map (NCicPp.ppterm ~metasenv ~subst ~context)
@@ -150,7 +150,7 @@ and beta_expand_many test_equality_only swap metasenv subst context t args =
     List.fold_right
       (fun arg (num,subst,metasenv,t) ->
          let metasenv, subst, t =
-           beta_expand num test_equality_only swap metasenv subst context t arg
+           beta_expand hdb num test_equality_only swap metasenv subst context t arg
          in
            num+1,subst,metasenv,t)
       args (1,subst,metasenv,t) 
@@ -160,13 +160,13 @@ and beta_expand_many test_equality_only swap metasenv subst context t args =
     metasenv, subst, hd
 (* (*D*)  in outside (); rc with exn -> outside (); raise exn *)
 
-and instantiate test_eq_only metasenv subst context n lc t swap =
+and instantiate hdb test_eq_only metasenv subst context n lc t swap =
  (*D*)  inside 'I'; try let rc =  
          pp (lazy(string_of_int n ^ " :=?= "^
            NCicPp.ppterm ~metasenv ~subst ~context t));
   let unify test_eq_only m s c t1 t2 = 
-    if swap then unify test_eq_only m s c t2 t1 
-    else unify test_eq_only m s c t1 t2
+    if swap then unify hdb test_eq_only m s c t2 t1 
+    else unify hdb test_eq_only m s c t1 t2
   in
   let name, ctx, ty = NCicUtils.lookup_meta n metasenv in
   let metasenv, subst, t = 
@@ -245,200 +245,223 @@ and instantiate test_eq_only metasenv subst context n lc t swap =
     metasenv, subst
  (*D*)  in outside(); rc with exn -> outside (); raise exn 
 
-and unify test_eq_only metasenv subst context t1 t2 =
- (*D*) inside 'U'; try let rc = 
-   let fo_unif test_eq_only metasenv subst t1 t2 =
+and unify hdb test_eq_only metasenv subst context t1 t2 =
+ (*D*) inside 'U'; try let rc =
+   let rec fo_unif test_eq_only metasenv subst t1 t2 =
+     let try_hints metasenv subst context t1 t2 (* exc*) =
+       let candidates = NCicUnifHint.look_for_hint hdb metasenv subst context t1 t2
+       in
+       let rec cand_iter = function
+         | [] -> None (* raise exc *)
+         | (metasenv,c1,c2)::tl -> 
+             try 
+               prerr_endline (NCicPp.ppterm ~metasenv ~subst ~context t1);
+               prerr_endline (NCicPp.ppterm ~metasenv ~subst ~context c1);
+               prerr_endline (NCicPp.ppterm ~metasenv ~subst ~context t2);
+               prerr_endline (NCicPp.ppterm ~metasenv ~subst ~context c2);
+               let metasenv,subst = fo_unif test_eq_only metasenv subst t1 c1 in
+               let metasenv,subst = fo_unif test_eq_only metasenv subst c2 t2 in
+               Some (fo_unif test_eq_only metasenv subst t1 t2)
+             with
+               UnificationFailure _ | Uncertain _ -> cand_iter tl
+       in
+         cand_iter candidates
+     in
+
     (*D*) inside 'F'; try let rc =  
      pp (lazy("  " ^ NCicPp.ppterm ~metasenv ~subst ~context t1 ^ " === " ^ 
          NCicPp.ppterm ~metasenv ~subst ~context t2));
      if t1 === t2 then
        metasenv, subst
      else
-       match (t1,t2) with
-       | (C.Sort (C.Type a), C.Sort (C.Type b)) when not test_eq_only -> 
-           if NCicEnvironment.universe_leq a b then metasenv, subst
-           else raise (fail_exc metasenv subst context t1 t2)
-       | (C.Sort (C.Type a), C.Sort (C.Type b)) -> 
-           if NCicEnvironment.universe_eq a b then metasenv, subst
-           else raise (fail_exc metasenv subst context t1 t2)
-       | (C.Sort C.Prop,C.Sort (C.Type _)) -> 
-           if (not test_eq_only) then metasenv, subst
-           else raise (fail_exc metasenv subst context t1 t2)
-
-       | (C.Lambda (name1,s1,t1), C.Lambda(_,s2,t2)) 
-       | (C.Prod (name1,s1,t1), C.Prod(_,s2,t2)) ->
-           let metasenv, subst = unify true metasenv subst context s1 s2 in
-           unify test_eq_only metasenv subst ((name1, C.Decl s1)::context) t1 t2
-       | (C.LetIn (name1,ty1,s1,t1), C.LetIn(_,ty2,s2,t2)) ->
-           let metasenv,subst=unify test_eq_only metasenv subst context ty1 ty2 in
-           let metasenv,subst=unify test_eq_only metasenv subst context s1 s2 in
-           let context = (name1, C.Def (s1,ty1))::context in
-           unify test_eq_only metasenv subst context t1 t2
-
-       | (C.Meta (n1,(s1,l1 as lc1)),C.Meta (n2,(s2,l2 as lc2))) when n1 = n2 ->
-          (try 
-           let l1 = NCicUtils.expand_local_context l1 in
-           let l2 = NCicUtils.expand_local_context l2 in
-           let metasenv, subst, to_restrict, _ =
-            List.fold_right2 
-             (fun t1 t2 (metasenv, subst, to_restrict, i) -> 
-                try 
-                  let metasenv, subst = 
-                   unify test_eq_only metasenv subst context 
-                    (NCicSubstitution.lift s1 t1) (NCicSubstitution.lift s2 t2)
-                  in
-                  metasenv, subst, to_restrict, i-1  
-                with UnificationFailure _ | Uncertain _ ->
-                  metasenv, subst, i::to_restrict, i-1)
-             l1 l2 (metasenv, subst, [], List.length l1)
-           in
-           if to_restrict <> [] then
-             let metasenv, subst, _ = 
-               NCicMetaSubst.restrict metasenv subst n1 to_restrict
-             in
-               metasenv, subst
-           else metasenv, subst
-          with 
-           | Invalid_argument _ -> assert false
-           | NCicMetaSubst.MetaSubstFailure msg ->
-              try 
-                let _,_,term,_ = NCicUtils.lookup_subst n1 subst in
-                let term1 = NCicSubstitution.subst_meta lc1 term in
-                let term2 = NCicSubstitution.subst_meta lc2 term in
-                  unify test_eq_only metasenv subst context term1 term2
-              with NCicUtils.Subst_not_found _-> raise (UnificationFailure msg))
-
-       | C.Meta (n,lc), t -> 
-           (try 
-             let _,_,term,_ = NCicUtils.lookup_subst n subst in
-             let term = NCicSubstitution.subst_meta lc term in
-               unify test_eq_only metasenv subst context term t
-           with NCicUtils.Subst_not_found _-> 
-             instantiate test_eq_only metasenv subst context n lc t false)
-
-       | t, C.Meta (n,lc) -> 
-           (try 
-             let _,_,term,_ = NCicUtils.lookup_subst n subst in
-             let term = NCicSubstitution.subst_meta lc term in
-               unify test_eq_only metasenv subst context t term
-           with NCicUtils.Subst_not_found _-> 
-             instantiate test_eq_only metasenv subst context n lc t true)
-
-       | NCic.Appl (NCic.Meta (i,l)::args), _ when List.mem_assoc i subst ->
-            let _,_,term,_ = NCicUtils.lookup_subst i subst in
-            let term = NCicSubstitution.subst_meta l term in
-              unify test_eq_only metasenv subst context (mk_appl term args) t2
-
-       | _, NCic.Appl (NCic.Meta (i,l)::args) when List.mem_assoc i subst ->
-            let _,_,term,_ = NCicUtils.lookup_subst i subst in
-            let term = NCicSubstitution.subst_meta l term in
-              unify test_eq_only metasenv subst context t1 (mk_appl term args)
-
-       |  NCic.Appl (NCic.Meta (i,_)::_ as l1),
-          NCic.Appl (NCic.Meta (j,_)::_ as l2) when i=j ->
-            (try
-              List.fold_left2 
-                (fun (metasenv, subst) t1 t2 ->
-                  unify test_eq_only metasenv subst context t1 t2)
-                (metasenv,subst) l1 l2
-            with Invalid_argument _ -> 
-              raise (fail_exc metasenv subst context t1 t2))
-
-       | NCic.Appl (NCic.Meta (i,l)::args), _ when not (flexible args) ->
-           (* we verify that none of the args is a Meta, 
-              since beta expanding w.r.t a metavariable makes no sense  *)
-              let metasenv, subst, beta_expanded =
-                beta_expand_many 
-                  test_eq_only false 
-                  metasenv subst context t2 args
-              in
-                unify test_eq_only metasenv subst context 
-                  (C.Meta (i,l)) beta_expanded 
-
-       | _, NCic.Appl (NCic.Meta (i,l)::args) when not(flexible args) ->
-              let metasenv, subst, beta_expanded =
-                beta_expand_many 
-                  test_eq_only true 
-                  metasenv subst context t1 args
-              in
-                unify test_eq_only metasenv subst context 
-                  beta_expanded (C.Meta (i,l))
-
-       (* processing this case here we avoid a useless small delta step *)
-       | (C.Appl ((C.Const r1) as _hd1::tl1), C.Appl (C.Const r2::tl2)) 
-         when Ref.eq r1 r2 ->
-           let relevance = NCicEnvironment.get_relevance r1 in
-           let relevance = match r1 with
-             | Ref.Ref (_,Ref.Con (_,_,lno)) ->
-                 let _,relevance = HExtlib.split_nth lno relevance in
-                   HExtlib.mk_list false lno @ relevance
-             | _ -> relevance
-           in
-           let metasenv, subst, _ = 
-             try
-               List.fold_left2 
-                 (fun (metasenv, subst, relevance) t1 t2 ->
-                    let b, relevance = 
-                      match relevance with b::tl -> b,tl | _ -> true, [] in
+       match (try_hints metasenv subst context t1 t2) with
+       | Some x -> x
+       | None ->
+         match (t1,t2) with
+         | (C.Sort (C.Type a), C.Sort (C.Type b)) when not test_eq_only -> 
+             if NCicEnvironment.universe_leq a b then metasenv, subst
+             else raise (fail_exc metasenv subst context t1 t2)
+         | (C.Sort (C.Type a), C.Sort (C.Type b)) -> 
+             if NCicEnvironment.universe_eq a b then metasenv, subst
+             else raise (fail_exc metasenv subst context t1 t2)
+         | (C.Sort C.Prop,C.Sort (C.Type _)) -> 
+             if (not test_eq_only) then metasenv, subst
+             else raise (fail_exc metasenv subst context t1 t2)
+
+         | (C.Lambda (name1,s1,t1), C.Lambda(_,s2,t2)) 
+         | (C.Prod (name1,s1,t1), C.Prod(_,s2,t2)) ->
+             let metasenv, subst = unify hdb true metasenv subst context s1 s2 in
+             unify hdb test_eq_only metasenv subst ((name1, C.Decl s1)::context) t1 t2
+         | (C.LetIn (name1,ty1,s1,t1), C.LetIn(_,ty2,s2,t2)) ->
+             let metasenv,subst=unify hdb test_eq_only metasenv subst context ty1 ty2 in
+             let metasenv,subst=unify hdb test_eq_only metasenv subst context s1 s2 in
+             let context = (name1, C.Def (s1,ty1))::context in
+             unify hdb test_eq_only metasenv subst context t1 t2
+
+         | (C.Meta (n1,(s1,l1 as lc1)),C.Meta (n2,(s2,l2 as lc2))) when n1 = n2 ->
+            (try 
+             let l1 = NCicUtils.expand_local_context l1 in
+             let l2 = NCicUtils.expand_local_context l2 in
+             let metasenv, subst, to_restrict, _ =
+              List.fold_right2 
+               (fun t1 t2 (metasenv, subst, to_restrict, i) -> 
+                  try 
                     let metasenv, subst = 
-                      try unify test_eq_only metasenv subst context t1 t2
-                      with UnificationFailure _ | Uncertain _ when not b ->
-                        metasenv, subst
+                     unify hdb test_eq_only metasenv subst context 
+                      (NCicSubstitution.lift s1 t1) (NCicSubstitution.lift s2 t2)
                     in
-                      metasenv, subst, relevance)
-                 (metasenv, subst, relevance) tl1 tl2
-             with Invalid_argument _ -> 
-               raise (uncert_exc metasenv subst context t1 t2)
-           in 
-             metasenv, subst
-
-       | (C.Match (Ref.Ref (_,Ref.Ind (_,tyno,_)) as ref1,outtype1,term1,pl1),
-          C.Match (ref2,outtype2,term2,pl2)) ->
-          let _,_,itl,_,_ = NCicEnvironment.get_checked_indtys ref1 in
-          let _,_,ty,_ = List.nth itl tyno in
-          let rec remove_prods ~subst context ty = 
-             let ty = NCicReduction.whd ~subst context ty in
-             match ty with
-             | C.Sort _ -> ty
-            | C.Prod (name,so,ta) -> 
-                   remove_prods ~subst ((name,(C.Decl so))::context) ta
-             | _ -> assert false
-           in
-           let is_prop = 
-             match remove_prods ~subst [] ty with
-             | C.Sort C.Prop -> true
-             | _ -> false 
-           in
-          let rec remove_prods ~subst context ty = 
-             let ty = NCicReduction.whd ~subst context ty in
-             match ty with
-             | C.Sort _ -> ty
-            | C.Prod (name,so,ta) -> 
-                   remove_prods ~subst ((name,(C.Decl so))::context) ta
-             | _ -> assert false
-           in
-           if not (Ref.eq ref1 ref2) then 
-             raise (uncert_exc metasenv subst context t1 t2) 
-           else
-             let metasenv, subst = 
-               unify test_eq_only metasenv subst context outtype1 outtype2 in
-             let metasenv, subst = 
-               try unify test_eq_only metasenv subst context term1 term2 
-               with UnificationFailure _ | Uncertain _ when is_prop -> 
+                    metasenv, subst, to_restrict, i-1  
+                  with UnificationFailure _ | Uncertain _ ->
+                    metasenv, subst, i::to_restrict, i-1)
+               l1 l2 (metasenv, subst, [], List.length l1)
+             in
+             if to_restrict <> [] then
+               let metasenv, subst, _ = 
+                 NCicMetaSubst.restrict metasenv subst n1 to_restrict
+               in
                  metasenv, subst
+             else metasenv, subst
+            with 
+             | Invalid_argument _ -> assert false
+             | NCicMetaSubst.MetaSubstFailure msg ->
+                try 
+                  let _,_,term,_ = NCicUtils.lookup_subst n1 subst in
+                  let term1 = NCicSubstitution.subst_meta lc1 term in
+                  let term2 = NCicSubstitution.subst_meta lc2 term in
+                    unify hdb test_eq_only metasenv subst context term1 term2
+                with NCicUtils.Subst_not_found _-> raise (UnificationFailure msg))
+
+         | C.Meta (n,lc), t -> 
+             (try 
+               let _,_,term,_ = NCicUtils.lookup_subst n subst in
+               let term = NCicSubstitution.subst_meta lc term in
+                 unify hdb test_eq_only metasenv subst context term t
+             with NCicUtils.Subst_not_found _-> 
+               instantiate hdb test_eq_only metasenv subst context n lc t false)
+
+         | t, C.Meta (n,lc) -> 
+             (try 
+               let _,_,term,_ = NCicUtils.lookup_subst n subst in
+               let term = NCicSubstitution.subst_meta lc term in
+                 unify hdb test_eq_only metasenv subst context t term
+             with NCicUtils.Subst_not_found _-> 
+               instantiate hdb test_eq_only metasenv subst context n lc t true)
+
+         | NCic.Appl (NCic.Meta (i,l)::args), _ when List.mem_assoc i subst ->
+              let _,_,term,_ = NCicUtils.lookup_subst i subst in
+              let term = NCicSubstitution.subst_meta l term in
+                unify hdb test_eq_only metasenv subst context (mk_appl term args) t2
+
+         | _, NCic.Appl (NCic.Meta (i,l)::args) when List.mem_assoc i subst ->
+              let _,_,term,_ = NCicUtils.lookup_subst i subst in
+              let term = NCicSubstitution.subst_meta l term in
+                unify hdb test_eq_only metasenv subst context t1 (mk_appl term args)
+
+         |  NCic.Appl (NCic.Meta (i,_)::_ as l1),
+            NCic.Appl (NCic.Meta (j,_)::_ as l2) when i=j ->
+              (try
+                List.fold_left2 
+                  (fun (metasenv, subst) t1 t2 ->
+                    unify hdb test_eq_only metasenv subst context t1 t2)
+                  (metasenv,subst) l1 l2
+              with Invalid_argument _ -> 
+                raise (fail_exc metasenv subst context t1 t2))
+
+         | NCic.Appl (NCic.Meta (i,l)::args), _ when not (flexible args) ->
+             (* we verify that none of the args is a Meta, 
+                since beta expanding w.r.t a metavariable makes no sense  *)
+                let metasenv, subst, beta_expanded =
+                  beta_expand_many hdb 
+                    test_eq_only false 
+                    metasenv subst context t2 args
+                in
+                  unify hdb test_eq_only metasenv subst context 
+                    (C.Meta (i,l)) beta_expanded 
+
+         | _, NCic.Appl (NCic.Meta (i,l)::args) when not(flexible args) ->
+                let metasenv, subst, beta_expanded =
+                  beta_expand_many hdb 
+                    test_eq_only true 
+                    metasenv subst context t1 args
+                in
+                  unify hdb test_eq_only metasenv subst context 
+                    beta_expanded (C.Meta (i,l))
+
+         (* processing this case here we avoid a useless small delta step *)
+         | (C.Appl ((C.Const r1) as _hd1::tl1), C.Appl (C.Const r2::tl2)) 
+           when Ref.eq r1 r2 ->
+             let relevance = NCicEnvironment.get_relevance r1 in
+             let relevance = match r1 with
+               | Ref.Ref (_,Ref.Con (_,_,lno)) ->
+                   let _,relevance = HExtlib.split_nth lno relevance in
+                     HExtlib.mk_list false lno @ relevance
+               | _ -> relevance
+             in
+             let metasenv, subst, _ = 
+               try
+                 List.fold_left2 
+                   (fun (metasenv, subst, relevance) t1 t2 ->
+                      let b, relevance = 
+                        match relevance with b::tl -> b,tl | _ -> true, [] in
+                      let metasenv, subst = 
+                        try unify hdb test_eq_only metasenv subst context t1 t2
+                        with UnificationFailure _ | Uncertain _ when not b ->
+                          metasenv, subst
+                      in
+                        metasenv, subst, relevance)
+                   (metasenv, subst, relevance) tl1 tl2
+               with Invalid_argument _ -> 
+                 raise (uncert_exc metasenv subst context t1 t2)
+             in 
+               metasenv, subst
+
+         | (C.Match (Ref.Ref (_,Ref.Ind (_,tyno,_)) as ref1,outtype1,term1,pl1),
+            C.Match (ref2,outtype2,term2,pl2)) ->
+             let _,_,itl,_,_ = NCicEnvironment.get_checked_indtys ref1 in
+             let _,_,ty,_ = List.nth itl tyno in
+             let rec remove_prods ~subst context ty = 
+               let ty = NCicReduction.whd ~subst context ty in
+               match ty with
+               | C.Sort _ -> ty
+               | C.Prod (name,so,ta) -> 
+                     remove_prods ~subst ((name,(C.Decl so))::context) ta
+               | _ -> assert false
+             in
+             let is_prop = 
+               match remove_prods ~subst [] ty with
+               | C.Sort C.Prop -> true
+               | _ -> false 
+             in
+             let rec remove_prods ~subst context ty = 
+               let ty = NCicReduction.whd ~subst context ty in
+               match ty with
+               | C.Sort _ -> ty
+               | C.Prod (name,so,ta) -> 
+                     remove_prods ~subst ((name,(C.Decl so))::context) ta
+               | _ -> assert false
              in
-             (try
-              List.fold_left2 
-               (fun (metasenv,subst) -> 
-                  unify test_eq_only metasenv subst context)
-               (metasenv, subst) pl1 pl2
-             with Invalid_argument _ -> 
-               raise (uncert_exc metasenv subst context t1 t2))
-       | (C.Implicit _, _) | (_, C.Implicit _) -> assert false
-       | _ when NCicUntrusted.metas_of_term subst context t1 = [] && 
-                NCicUntrusted.metas_of_term subst context t2 = [] -> 
-                  raise (fail_exc metasenv subst context t1 t2)
-       | _ -> raise (uncert_exc metasenv subst context t1 t2)
+             if not (Ref.eq ref1 ref2) then 
+               raise (uncert_exc metasenv subst context t1 t2) 
+             else
+               let metasenv, subst = 
+                 unify hdb test_eq_only metasenv subst context outtype1 outtype2 in
+               let metasenv, subst = 
+                 try unify hdb test_eq_only metasenv subst context term1 term2 
+                 with UnificationFailure _ | Uncertain _ when is_prop -> 
+                   metasenv, subst
+               in
+               (try
+                List.fold_left2 
+                 (fun (metasenv,subst) -> 
+                    unify hdb test_eq_only metasenv subst context)
+                 (metasenv, subst) pl1 pl2
+               with Invalid_argument _ -> 
+                 raise (uncert_exc metasenv subst context t1 t2))
+         | (C.Implicit _, _) | (_, C.Implicit _) -> assert false
+         | _ when NCicUntrusted.metas_of_term subst context t1 = [] && 
+                  NCicUntrusted.metas_of_term subst context t2 = [] -> 
+                    raise (fail_exc metasenv subst context t1 t2)
+         | _ -> raise (uncert_exc metasenv subst context t1 t2)
      (*D*)  in outside(); rc with exn -> outside (); raise exn 
     in
     let height_of = function
@@ -520,8 +543,8 @@ and unify test_eq_only metasenv subst context t1 t2 =
  (*D*)  in outside(); rc with exn -> outside (); raise exn 
 ;;
 
-let unify = 
+let unify hdb 
   indent := "";      
-  unify false;;
+  unify hdb false;;