helm/software/components/ng_refiner/nCicCoercion.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 = NCic.term * int * int =
  15   struct
  16         type t = NCic.term * int * int
  17         let compare = Pervasives.compare
  18   end
  19
  20 module CoercionSet = Set.Make(COT)
  21
  22 module DB =
  23   Discrimination_tree.Make(NDiscriminationTree.NCicIndexable)(CoercionSet)
  24
  25 type db = DB.t * DB.t
  26
  27 let empty_db = DB.empty,DB.empty
  28
  29 class status =
  30  object
  31   val db = empty_db
  32   method coerc_db = db
  33   method set_coerc_db v = {< db = v >}
  34   method set_coercion_status (o : status) = {< db = o#coerc_db >}
  35  end
  36
  37 let index_coercion status c src tgt arity arg =
  38   let db_src,db_tgt = status#coerc_db in
  39   let data = (c,arity,arg) in
  40 (*
  41   prerr_endline ("INDEX:" ^
  42     NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] src ^ " ===> " ^
  43     NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] tgt ^ "  :=  " ^
  44     NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] c);
  45 *)
  46   let db_src = DB.index db_src src data in
  47   let db_tgt = DB.index db_tgt tgt data in
  48   status#set_coerc_db (db_src, db_tgt)
  49 ;;
  50
  51 let index_old_db odb (status : #status) =
  52   List.fold_left
  53     (fun status (_,tgt,clist) ->
  54        List.fold_left
  55          (fun status (uri,_,arg) ->
  56            try
  57             let c=fst (OCic2NCic.convert_term uri (CicUtil.term_of_uri uri)) in
  58             let arity = match tgt with | CoercDb.Fun i -> i | _ -> 0 in
  59             let src, tgt =
  60               let cty = NCicTypeChecker.typeof ~subst:[] ~metasenv:[] [] c in
  61               let scty, metasenv,_ =
  62                 NCicMetaSubst.saturate ~delta:max_int [] [] [] cty (arity+1)
  63               in
  64               match scty with
  65               | NCic.Prod (_, src, tgt) ->
  66                  let tgt =
  67                    NCicSubstitution.subst (NCic.Meta (-1,(0,NCic.Irl 0))) tgt
  68                  in
  69 (*
  70             prerr_endline (Printf.sprintf "indicizzo %s (%d) : %s ===> %s"
  71               (NCicPp.ppterm ~metasenv ~subst:[] ~context:[] scty) (arity+1)
  72               (NCicPp.ppterm ~metasenv ~subst:[] ~context:[] src)
  73               (NCicPp.ppterm ~metasenv ~subst:[] ~context:[] tgt));
  74 *)
  75                 src, tgt
  76               | t ->
  77                   prerr_endline (
  78                     NCicPp.ppterm ~metasenv ~subst:[] ~context:[] t);
  79                   assert false
  80             in
  81             index_coercion status c src tgt arity arg
  82            with
  83            | NCicEnvironment.BadDependency _
  84            | NCicTypeChecker.TypeCheckerFailure _ -> status)
  85          status clist)
  86     status (CoercDb.to_list odb)
  87 ;;
  88
  89 let look_for_coercion status metasenv subst context infty expty =
  90  let db_src,db_tgt = status#coerc_db in
  91   match infty, expty with
  92   | (NCic.Meta _ | NCic.Appl (NCic.Meta _::_)),
  93     (NCic.Meta _ | NCic.Appl (NCic.Meta _::_)) -> []
  94   | _ ->
  95 (*
  96     prerr_endline ("LOOK FOR COERCIONS: " ^
  97       NCicPp.ppterm ~metasenv ~subst ~context infty ^ "  |===> " ^
  98       NCicPp.ppterm ~metasenv ~subst ~context expty);
  99 *)
 100     let set_src = DB.retrieve_unifiables db_src infty in
 101     let set_tgt = DB.retrieve_unifiables db_tgt expty in
 102     let candidates = CoercionSet.inter set_src set_tgt in
 103 (*
 104     prerr_endline ("CANDIDATES: " ^
 105       String.concat "," (List.map (fun (t,_,_) ->
 106         NCicPp.ppterm ~metasenv ~subst ~context t)
 107       (CoercionSet.elements candidates)));
 108 *)
 109     List.map
 110       (fun (t,arity,arg) ->
 111           let ty = NCicTypeChecker.typeof ~metasenv:[] ~subst:[] [] t in
 112           let ty, metasenv, args =
 113            NCicMetaSubst.saturate ~delta:max_int metasenv subst context ty arity
 114           in
 115   (*       prerr_endline (
 116             NCicPp.ppterm ~metasenv ~subst:[] ~context:[] ty ^ " --- " ^
 117             NCicPp.ppterm ~metasenv ~subst ~context
 118             (NCicUntrusted.mk_appl t args) ^ " --- " ^
 119               string_of_int (List.length args) ^ " == " ^ string_of_int arg); *)
 120           metasenv, NCicUntrusted.mk_appl t args, ty, List.nth args arg)
 121       (CoercionSet.elements candidates)
 122 ;;