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 let dummy = NCic.Const (NReference.reference_of_string "cic:/dummy_conv.dec");;
  28 let pair t1 t2 = (NCic.Appl [dummy;t1;t2]) ;;
  29
  30 let index_hint hdb context t1 t2 precedence =
  31   assert (
  32     (match t1 with
  33     | NCic.Meta _ | NCic.Appl (NCic.Meta _ :: _) -> false | _ -> true)
  34     &&
  35     (match t2 with
  36     | NCic.Meta _ | NCic.Appl (NCic.Meta _ :: _) -> false | _ -> true)
  37   );
  38   let pair' = pair t1 t2 in
  39   let data =
  40     List.fold_left
  41      (fun t (n,e) ->
  42         match e with
  43         | NCic.Decl ty -> NCic.Prod (n,ty,t)
  44         | NCic.Def (b,ty) -> NCic.LetIn (n,ty,b,t))
  45       pair' context in
  46   let src =
  47     List.fold_left
  48      (fun t (_,e) ->
  49         match e with
  50         | NCic.Decl _ ->
  51             NCicSubstitution.subst (NCic.Meta (-1,(0,NCic.Irl 0))) t
  52         | NCic.Def _ ->
  53             NCicSubstitution.subst (NCic.Meta (-1,(0,NCic.Irl 0))) t)
  54      pair' context in
  55 (*
  56   prerr_endline ("INDEXING: " ^
  57     NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] src ^ " |==> " ^
  58     NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] data);
  59 *)
  60   DB.index hdb src (precedence, data)
  61 ;;
  62
  63 let empty_db = DB.empty ;;
  64
  65 let user_db = ref empty_db ;;
  66
  67 let add_user_provided_hint t precedence =
  68   let u = UriManager.uri_of_string "cic:/foo/bar.con" in
  69   let c, a, b =
  70     let rec aux ctx = function
  71       | NCic.Appl l ->
  72           (match List.rev l with
  73           | b::a::_ -> ctx, a, b
  74           | _ -> assert false)
  75       | NCic.Prod (n,s,t) ->
  76           aux ((n, NCic.Decl s) :: ctx) t
  77       | NCic.LetIn (n,ty,t,b) ->
  78           aux  ((n, NCic.Def (t,ty)) :: ctx) b
  79       | _ -> assert false
  80     in
  81       aux [] (fst (OCic2NCic.convert_term u t))
  82   in
  83   user_db := index_hint !user_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 ())
 141   in
 142   List.fold_left
 143     (fun db -> function
 144      | (ctx,b1,b2) -> index_hint db ctx b1 b2 0)
 145     !user_db (List.flatten hints)
 146 ;;
 147
 148
 149 let look_for_hint hdb metasenv subst context t1 t2 =
 150 (*
 151   prerr_endline ("KEY1: "^NCicPp.ppterm ~metasenv ~subst ~context (pair t1 t2));
 152   prerr_endline ("KEY2: "^NCicPp.ppterm ~metasenv ~subst ~context (pair t2 t1));
 153   DB.iter hdb
 154    (fun p ds ->
 155       prerr_endline ("ENTRY: " ^
 156       NDiscriminationTree.NCicIndexable.string_of_path p ^ " |--> " ^
 157       String.concat "|" (List.map (NCicPp.ppterm ~metasenv:[] ~subst:[]
 158       ~context:[]) (HintSet.elements ds))));
 159 *)
 160   let candidates1 = DB.retrieve_unifiables hdb (pair t1 t2) in
 161   let candidates2 = DB.retrieve_unifiables hdb (pair t2 t1) in
 162   let candidates1 =
 163     List.map (fun (prec,ty) ->
 164        prec,true,NCicMetaSubst.saturate ~delta:max_int metasenv subst context ty 0)
 165     (HintSet.elements candidates1)
 166   in
 167   let candidates2 =
 168     List.map (fun (prec,ty) ->
 169        prec,false,NCicMetaSubst.saturate ~delta:max_int metasenv subst context ty 0)
 170     (HintSet.elements candidates2)
 171   in
 172   let rc =
 173   List.map
 174    (function
 175     | (prec,true,(NCic.Appl [_; t1; t2],metasenv,_)) -> prec,metasenv, t1, t2
 176     | (prec,false,(NCic.Appl [_; t1; t2],metasenv,_)) -> prec,metasenv, t2, t1
 177     | _ -> assert false)
 178    (candidates1 @ candidates2)
 179   in
 180   let rc =
 181     List.sort (fun (x,_,_,_) (y,_,_,_) -> Pervasives.compare x y) rc
 182   in
 183   let rc = List.map (fun (_,x,y,z) -> x,y,z) rc in
 184 (*
 185     List.iter
 186       (fun (metasenv, t1, t2) ->
 187           prerr_endline
 188             ("CAND: "^NCicPp.ppterm ~metasenv ~subst ~context t1 ^
 189             " == "^NCicPp.ppterm ~metasenv ~subst ~context t2))
 190       rc;
 191 *)
 192   rc
 193 ;;