]> matita.cs.unibo.it Git - helm.git/blob - helm/software/components/ng_refiner/nCicUnifHint.ml
Initial implementation of statuses using objects in place of nested records.
[helm.git] / helm / software / components / ng_refiner / nCicUnifHint.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 (* $Id: nCicRefiner.mli 9227 2008-11-21 16:00:06Z tassi $ *)
13
14 module COT : Set.OrderedType with type t = int * NCic.term = 
15   struct
16         type t = int * NCic.term
17         let compare = Pervasives.compare
18   end
19
20 module HintSet = Set.Make(COT)
21
22 module DB = 
23   Discrimination_tree.Make(NDiscriminationTree.NCicIndexable)(HintSet)
24
25 type db = DB.t
26
27 class status =
28  object
29   val db = DB.empty
30   method uhint_db = db
31   method set_uhint_db v = {< db = v >}
32  end
33
34 let dummy = NCic.Const (NReference.reference_of_string "cic:/dummy_conv.dec");;
35 let pair t1 t2 = (NCic.Appl [dummy;t1;t2]) ;;
36
37 let index_hint hdb context t1 t2 precedence =
38   assert (
39     (match t1 with
40     | NCic.Meta _ | NCic.Appl (NCic.Meta _ :: _) -> false | _ -> true)
41     &&
42     (match t2 with
43     | NCic.Meta _ | NCic.Appl (NCic.Meta _ :: _) -> false | _ -> true)
44   );
45   let pair' = pair t1 t2 in
46   let data = 
47     List.fold_left 
48      (fun t (n,e) -> 
49         match e with
50         | NCic.Decl ty -> NCic.Prod (n,ty,t)
51         | NCic.Def (b,ty) -> NCic.LetIn (n,ty,b,t))
52       pair' context in
53   let src = 
54     List.fold_left 
55      (fun t (_,e) -> 
56         match e with
57         | NCic.Decl _ -> 
58             NCicSubstitution.subst (NCic.Meta (-1,(0,NCic.Irl 0))) t
59         | NCic.Def _ -> 
60             NCicSubstitution.subst (NCic.Meta (-1,(0,NCic.Irl 0))) t)
61      pair' context in
62 (*
63   prerr_endline ("INDEXING: " ^ 
64     NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] src ^ " |==> " ^
65     NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] data);
66 *)
67   hdb#set_uhint_db (DB.index hdb#uhint_db src (precedence, data))
68 ;;
69
70 let add_user_provided_hint db t precedence =
71   let c, a, b = 
72     let rec aux ctx = function
73       | NCic.Appl l ->
74           (match List.rev l with
75           | b::a::_ -> ctx, a, b
76           | _ -> assert false)
77       | NCic.Prod (n,s,t) -> aux ((n, NCic.Decl s) :: ctx) t
78       | NCic.LetIn (n,ty,t,b) -> aux  ((n, NCic.Def (t,ty)) :: ctx) b
79       | _ -> assert false
80     in
81       aux [] t
82   in
83    index_hint db c a b precedence
84 ;;
85
86 let db () = 
87   let combine f l =
88    List.flatten
89      (let rec aux = function 
90       | u1 :: tl -> List.map (f u1) tl :: aux tl
91       | [] -> []
92      in aux l)
93   in
94   let mk_hint (u1,_,_) (u2,_,_) = 
95     let l = OCic2NCic.convert_obj u1 
96       (fst (CicEnvironment.get_obj CicUniv.oblivion_ugraph u1)) in
97     let r = OCic2NCic.convert_obj u2 
98       (fst (CicEnvironment.get_obj CicUniv.oblivion_ugraph u2)) in
99     match List.hd l,List.hd r with
100     | (_,h1,_,_,NCic.Constant (_,_,Some l,_,_)), 
101       (_,h2,_,_,NCic.Constant (_,_,Some r,_,_)) ->
102         let rec aux ctx t1 t2 =
103           match t1, t2 with
104           | NCic.Lambda (n1,s1,b1), NCic.Lambda(_,s2,b2) ->
105               if NCicReduction.are_convertible ~subst:[] ~metasenv:[] ctx s1 s2
106               then aux ((n1, NCic.Decl s1) :: ctx) b1 b2
107               else []
108           | b1,b2 -> 
109               if NCicReduction.are_convertible ~subst:[] ~metasenv:[] ctx b1 b2 
110               then begin
111 (*
112                 prerr_endline ("hint: " ^ NCicPp.ppterm ~metasenv:[] ~subst:[]
113                   ~context:ctx b1 ^ " === " ^ NCicPp.ppterm ~metasenv:[]
114                   ~subst:[] ~context:ctx b2);
115 *)
116               let rec mk_rels =  
117                  function 0 -> [] | n -> NCic.Rel n :: mk_rels (n-1) 
118               in 
119               let n1 = 
120                 NCic.Appl (NCic.Const(OCic2NCic.reference_of_ouri 
121                  u1 (NReference.Def h1)) :: mk_rels (List.length ctx))
122               in
123               let n2 = 
124                 NCic.Appl (NCic.Const(OCic2NCic.reference_of_ouri 
125                  u2 (NReference.Def h2)) :: mk_rels (List.length ctx))
126               in
127                 [ctx,b1,b2; ctx,b1,n2; ctx,n1,b2; ctx,n1,n2]
128               end else []
129         in
130           aux [] l r
131     | _ -> []
132   in
133   let hints = 
134     List.fold_left 
135       (fun acc (_,_,l) -> 
136           acc @ 
137           if List.length l > 1 then 
138            combine mk_hint l
139           else [])
140       [] (CoercDb.to_list (CoercDb.dump ()))
141   in
142   prerr_endline "MISTERO";
143   assert false (* ERA
144   List.fold_left 
145     (fun db -> function 
146      | (ctx,b1,b2) -> index_hint db ctx b1 b2 0)
147     !user_db (List.flatten hints)
148 *)
149 ;;
150
151 let saturate ?(delta=0) metasenv subst context ty goal_arity =
152  assert (goal_arity >= 0);
153   let rec aux metasenv = function
154    | NCic.Prod (name,s,t) as ty ->
155        let metasenv1, _, arg,_ = 
156           NCicMetaSubst.mk_meta ~name:name metasenv context (`WithType s) in
157        let t, metasenv1, args, pno = 
158            aux metasenv1 (NCicSubstitution.subst arg t) 
159        in
160        if pno + 1 = goal_arity then
161          ty, metasenv, [], goal_arity+1
162        else
163          t, metasenv1, arg::args, pno+1
164    | ty ->
165         match NCicReduction.whd ~subst context ty ~delta with
166         | NCic.Prod _ as ty -> aux metasenv ty
167         | _ -> ty, metasenv, [], 0 (* differs from the other impl in this line*)
168   in
169   let res, newmetasenv, arguments, _ = aux metasenv ty in
170   res, newmetasenv, arguments
171 ;;
172
173 let look_for_hint hdb metasenv subst context t1 t2 =
174 (*
175   prerr_endline ("KEY1: "^NCicPp.ppterm ~metasenv ~subst ~context (pair t1 t2));
176   prerr_endline ("KEY2: "^NCicPp.ppterm ~metasenv ~subst ~context (pair t2 t1));
177   DB.iter hdb
178    (fun p ds ->
179       prerr_endline ("ENTRY: " ^
180       NDiscriminationTree.NCicIndexable.string_of_path p ^ " |--> " ^
181       String.concat "|" (List.map (NCicPp.ppterm ~metasenv:[] ~subst:[]
182       ~context:[]) (HintSet.elements ds))));
183 *)
184   let candidates1 = DB.retrieve_unifiables hdb#uhint_db (pair t1 t2) in
185   let candidates2 = DB.retrieve_unifiables hdb#uhint_db (pair t2 t1) in
186   let candidates1 = 
187     List.map (fun (prec,ty) -> 
188        prec,true,saturate ~delta:max_int metasenv subst context ty 0) 
189     (HintSet.elements candidates1) 
190   in
191   let candidates2 = 
192     List.map (fun (prec,ty) -> 
193        prec,false,saturate ~delta:max_int metasenv subst context ty 0) 
194     (HintSet.elements candidates2) 
195   in
196   let rc = 
197     List.map
198       (fun (p,b,(t,m,_)) ->
199          let rec aux () (m,l as acc) = function
200            | NCic.Meta _ as t -> acc, t
201            | NCic.LetIn (name,ty,bo,t) ->
202                let m,_,i,_=
203                  NCicMetaSubst.mk_meta ~name m context (`WithType ty)in
204                let t = NCicSubstitution.subst i t in
205                aux () (m, (i,bo)::l) t
206            | t -> NCicUntrusted.map_term_fold_a (fun _ () -> ()) () aux acc t
207          in
208          let (m,l), t = aux () (m,[]) t in
209          p,b,(t,m,l))
210    (candidates1 @ candidates2)
211   in
212   let rc = 
213   List.map
214    (function 
215     | (prec,true,(NCic.Appl [_; t1; t2],metasenv,l))-> prec,metasenv,(t1,t2),l
216     | (prec,false,(NCic.Appl [_; t1; t2],metasenv,l))-> prec,metasenv,(t2,t1),l
217     | _ -> assert false)
218     rc
219   in
220   let rc = 
221     List.sort (fun (x,_,_,_) (y,_,_,_) -> Pervasives.compare x y) rc
222   in 
223   let rc = List.map (fun (_,x,y,z) -> x,y,z) rc in
224 (*
225     prerr_endline "Hints:";
226     List.iter 
227       (fun (metasenv, (t1, t2), premises) ->
228           prerr_endline 
229           ("\t" ^ String.concat ";  "
230                 (List.map (fun (a,b) -> 
231                    NCicPp.ppterm ~margin:max_int ~metasenv ~subst ~context a ^
232                    " =?= "^
233                    NCicPp.ppterm ~margin:max_int ~metasenv ~subst ~context b)
234                 premises) ^     
235               "  ==> "^
236               NCicPp.ppterm ~margin:max_int ~metasenv ~subst ~context t1 ^
237               " = "^NCicPp.ppterm ~margin:max_int ~metasenv ~subst ~context t2))
238       rc;
239 *)
240   rc
241 ;;