]> matita.cs.unibo.it Git - helm.git/blobdiff - helm/ocaml/cic_disambiguation/disambiguate.ml
This commit removes the slowest identity function ever implemented!
[helm.git] / helm / ocaml / cic_disambiguation / disambiguate.ml
index 026f9ebd1ee238ad6e7b7de248fa2f4f415bec30..b66e12d5e4c15909c2a7badc1d454666168086d2 100644 (file)
@@ -35,8 +35,10 @@ exception PathNotWellFormed
   (** raised when an environment is not enough informative to decide *)
 exception Try_again
 
+type aliases = bool * DisambiguateTypes.environment
+
 let debug = false
-let debug_print = if debug then prerr_endline else ignore
+let debug_print s = if debug then prerr_endline (Lazy.force s) else ()
 
 (*
   (** print benchmark information *)
@@ -47,16 +49,6 @@ let domain_size = ref 0
 let choices_avg = ref 0.
 *)
 
-let floc_of_loc (loc_begin, loc_end) =
-  let floc_begin =
-    { Lexing.pos_fname = ""; Lexing.pos_lnum = -1; Lexing.pos_bol = -1;
-      Lexing.pos_cnum = loc_begin }
-  in
-  let floc_end = { floc_begin with Lexing.pos_cnum = loc_end } in
-  (floc_begin, floc_end)
-
-let dummy_floc = floc_of_loc (-1, -1)
-
 let descr_of_domain_item = function
  | Id s -> s
  | Symbol (s, _) -> s
@@ -72,37 +64,37 @@ let refine_term metasenv context uri term ugraph =
   assert (uri=None);
   let metasenv, term = 
     CicMkImplicit.expand_implicits metasenv [] context term in
-    debug_print (sprintf "TEST_INTERPRETATION: %s" (CicPp.ppterm term));
+    debug_print (lazy (sprintf "TEST_INTERPRETATION: %s" (CicPp.ppterm term)));
     try
       let term', _, metasenv',ugraph1 = 
        CicRefine.type_of_aux' metasenv context term ugraph in
        (Ok (term', metasenv')),ugraph1
     with
       | CicRefine.Uncertain s ->
-          debug_print ("UNCERTAIN!!! [" ^ s ^ "] " ^ CicPp.ppterm term) ;
+          debug_print (lazy ("UNCERTAIN!!! [" ^ s ^ "] " ^ CicPp.ppterm term)) ;
           Uncertain,ugraph
       | CicRefine.RefineFailure msg ->
-          debug_print (sprintf "PRUNED!!!\nterm%s\nmessage:%s"
-            (CicPp.ppterm term) msg);
+          debug_print (lazy (sprintf "PRUNED!!!\nterm%s\nmessage:%s"
+            (CicPp.ppterm term) (CicRefine.explain_error msg)));
           Ko,ugraph
 
 let refine_obj metasenv context uri obj ugraph =
  assert (context = []);
  let metasenv, obj = CicMkImplicit.expand_implicits_in_obj metasenv [] obj in
-   debug_print (sprintf "TEST_INTERPRETATION: %s" (CicPp.ppobj obj));
+   debug_print (lazy (sprintf "TEST_INTERPRETATION: %s" (CicPp.ppobj obj))) ;
    try
      let obj', metasenv,ugraph = CicRefine.typecheck metasenv uri obj in
        (Ok (obj', metasenv)),ugraph
    with
      | CicRefine.Uncertain s ->
-         debug_print ("UNCERTAIN!!! [" ^ s ^ "] " ^ CicPp.ppobj obj) ;
+         debug_print (lazy ("UNCERTAIN!!! [" ^ s ^ "] " ^ CicPp.ppobj obj)) ;
          Uncertain,ugraph
      | CicRefine.RefineFailure msg ->
-         debug_print (sprintf "PRUNED!!!\nterm%s\nmessage:%s"
-           (CicPp.ppobj obj) msg);
+         debug_print (lazy (sprintf "PRUNED!!!\nterm%s\nmessage:%s"
+           (CicPp.ppobj obj) (CicRefine.explain_error msg))) ;
          Ko,ugraph
 
-let resolve (env: environment) (item: domain_item) ?(num = "") ?(args = []) () =
+let resolve (env: codomain_item Environment.t) (item: domain_item) ?(num = "") ?(args = []) () =
   try
     snd (Environment.find item env) env num args
   with Not_found -> 
@@ -110,7 +102,7 @@ let resolve (env: environment) (item: domain_item) ?(num = "") ?(args = []) () =
       (DisambiguateTypes.string_of_domain_item item))
 
   (* TODO move it to Cic *)
-let find_in_environment name (context: Cic.name list) =
+let find_in_context name (context: Cic.name list) =
   let rec aux acc = function
     | [] -> raise Not_found
     | Cic.Name hd :: tl when hd = name -> acc
@@ -142,7 +134,7 @@ let interpretate_term ~(context: Cic.name list) ~env ~uri ~is_path ast =
     | CicNotationPt.Case (term, indty_ident, outtype, branches) ->
         let cic_term = aux loc context term in
         let cic_outtype = aux_option loc context None outtype in
-        let do_branch ((head, args), term) =
+        let do_branch ((head, _, args), term) =
           let rec do_branch' context = function
             | [] -> aux loc context term
             | (name, typ) :: tl ->
@@ -159,7 +151,7 @@ let interpretate_term ~(context: Cic.name list) ~env ~uri ~is_path ast =
         in
         let (indtype_uri, indtype_no) =
           match indty_ident with
-          | Some indty_ident ->
+          | Some (indty_ident, _) ->
               (match resolve env (Id indty_ident) () with
               | Cic.MutInd (uri, tyno, _) -> (uri, tyno)
               | Cic.Implicit _ -> raise Try_again
@@ -167,7 +159,7 @@ let interpretate_term ~(context: Cic.name list) ~env ~uri ~is_path ast =
           | None ->
               let fst_constructor =
                 match branches with
-                | ((head, _), _) :: _ -> head
+                | ((head, _, _), _) :: _ -> head
                 | [] -> raise Invalid_choice
               in
               (match resolve env (Id fst_constructor) () with
@@ -241,7 +233,7 @@ let interpretate_term ~(context: Cic.name list) ~env ~uri ~is_path ast =
         let is_uri = function CicNotationPt.Uri _ -> true | _ -> false in
         (try
           if is_uri ast then raise Not_found;(* don't search the env for URIs *)
-          let index = find_in_environment name context in
+          let index = find_in_context name context in
           if subst <> None then
             CicNotationPt.fail loc "Explicit substitutions not allowed here";
           Cic.Rel index
@@ -300,13 +292,13 @@ let interpretate_term ~(context: Cic.name list) ~env ~uri ~is_path ast =
                 Cic.MutConstruct (uri, i, j, mk_subst uris)
             | Cic.Meta _ | Cic.Implicit _ as t ->
 (*
-                debug_print (sprintf
+                debug_print (lazy (sprintf
                   "Warning: %s must be instantiated with _[%s] but we do not enforce it"
                   (CicPp.ppterm t)
                   (String.concat "; "
                     (List.map
                       (fun (s, term) -> s ^ " := " ^ CicNotationPtPp.pp_term term)
-                      subst)));
+                      subst))));
 *)
                 t
             | _ ->
@@ -339,8 +331,8 @@ let interpretate_term ~(context: Cic.name list) ~env ~uri ~is_path ast =
   | CicNotationPt.AttributedTerm (`Loc loc, term) -> aux loc context term
   | term -> aux dummy_floc context term
 
-let interpretate_path ~context ~env path =
- interpretate_term ~context ~env ~uri:None ~is_path:true path
+let interpretate_path ~context path =
+ interpretate_term ~context ~env:Environment.empty ~uri:None ~is_path:true path
 
 let interpretate_obj ~context ~env ~uri ~is_path obj =
  assert (context = []);
@@ -493,9 +485,9 @@ let rec domain_rev_of_term ?(loc = dummy_floc) context = function
       let outtype_dom = domain_rev_of_term_option loc context outtype in
       let get_first_constructor = function
         | [] -> []
-        | ((head, _), _) :: _ -> [ Id head ]
+        | ((head, _, _), _) :: _ -> [ Id head ]
       in
-      let do_branch ((head, args), term) =
+      let do_branch ((head, _, args), term) =
         let (term_context, args_domain) =
           List.fold_left
             (fun (cont, dom) (name, typ) ->
@@ -513,7 +505,7 @@ let rec domain_rev_of_term ?(loc = dummy_floc) context = function
       branches_dom @ outtype_dom @ term_dom @
       (match indty_ident with
        | None -> get_first_constructor branches
-       | Some ident -> [ Id ident ])
+       | Some (ident, _) -> [ Id ident ])
   | CicNotationPt.Cast (term, ty) ->
       let term_dom = domain_rev_of_term ~loc context term in
       let ty_dom = domain_rev_of_term ~loc context ty in
@@ -544,7 +536,7 @@ let rec domain_rev_of_term ?(loc = dummy_floc) context = function
       where_dom @ defs_dom
   | CicNotationPt.Ident (name, subst) ->
       (try
-        let index = find_in_environment name context in
+        let index = find_in_context name context in
         if subst <> None then
           CicNotationPt.fail loc "Explicit substitutions not allowed here"
         else
@@ -577,8 +569,7 @@ and domain_rev_of_term_option loc context = function
   | None -> []
   | Some t -> domain_rev_of_term ~loc context t
 
-let domain_of_term ~context ast =
- rev_uniq (domain_rev_of_term context ast)
+let domain_of_term ~context ast = rev_uniq (domain_rev_of_term context ast)
 
 let domain_of_obj ~context ast =
  assert (context = []);
@@ -639,26 +630,32 @@ let domain_diff dom1 dom2 =
 module type Disambiguator =
 sig
   val disambiguate_term :
+    ?fresh_instances:bool ->
     dbd:Mysql.dbd ->
     context:Cic.context ->
     metasenv:Cic.metasenv ->
     ?initial_ugraph:CicUniv.universe_graph -> 
-    aliases:environment ->  (* previous interpretation status *)
+    aliases:DisambiguateTypes.environment ->(* previous interpretation status *)
+    universe:DisambiguateTypes.multiple_environment option ->
     CicNotationPt.term ->
     (environment *                   (* new interpretation status *)
      Cic.metasenv *                  (* new metasenv *)
      Cic.term*
-     CicUniv.universe_graph) list    (* disambiguated term *)
+     CicUniv.universe_graph) list *  (* disambiguated term *)
+    bool
 
   val disambiguate_obj :
+    ?fresh_instances:bool ->
     dbd:Mysql.dbd ->
-    aliases:environment ->           (* previous interpretation status *)
+    aliases:DisambiguateTypes.environment ->(* previous interpretation status *)
+    universe:DisambiguateTypes.multiple_environment option ->
     uri:UriManager.uri option ->     (* required only for inductive types *)
     GrafiteAst.obj ->
     (environment *                   (* new interpretation status *)
      Cic.metasenv *                  (* new metasenv *)
      Cic.obj *
-     CicUniv.universe_graph) list    (* disambiguated obj *)
+     CicUniv.universe_graph) list *  (* disambiguated obj *)
+    bool
 end
 
 module Make (C: Callbacks) =
@@ -686,54 +683,61 @@ module Make (C: Callbacks) =
              try
                CicUtil.term_of_uri uri
              with exn ->
-               debug_print (UriManager.string_of_uri uri);
-               debug_print (Printexc.to_string exn);
+               debug_print (lazy (UriManager.string_of_uri uri));
+               debug_print (lazy (Printexc.to_string exn));
                assert false
             in
            fun _ _ _ -> term))
         uris
 
+let refine_profiler = CicUtil.profile "disambiguate_thing.refine_thing"
+
     let disambiguate_thing ~dbd ~context ~metasenv
-      ?(initial_ugraph = CicUniv.empty_ugraph) ~aliases:current_env
+      ?(initial_ugraph = CicUniv.empty_ugraph) ~aliases ~universe
       ~uri ~pp_thing ~domain_of_thing ~interpretate_thing ~refine_thing thing
     =
-      debug_print "NEW DISAMBIGUATE INPUT";
+      debug_print (lazy "DISAMBIGUATE INPUT");
       let disambiguate_context =  (* cic context -> disambiguate context *)
         List.map
           (function None -> Cic.Anonymous | Some (name, _) -> name)
           context
       in
-      debug_print ("TERM IS: " ^ (pp_thing thing));
+      debug_print (lazy ("TERM IS: " ^ (pp_thing thing)));
       let thing_dom = domain_of_thing ~context:disambiguate_context thing in
-      debug_print (sprintf "DISAMBIGUATION DOMAIN: %s"
-        (string_of_domain thing_dom));
+      debug_print (lazy (sprintf "DISAMBIGUATION DOMAIN: %s"
+        (string_of_domain thing_dom)));
+      debug_print (lazy (sprintf "DISAMBIGUATION ENVIRONMENT: %s"
+        (DisambiguatePp.pp_environment aliases)));
+      debug_print (lazy (sprintf "DISAMBIGUATION UNIVERSE: %s"
+        (match universe with None -> "None" | Some _ -> "Some _")));
       let current_dom =
-        Environment.fold (fun item _ dom -> item :: dom) current_env []
+        Environment.fold (fun item _ dom -> item :: dom) aliases []
       in
       let todo_dom = domain_diff thing_dom current_dom in
       (* (2) lookup function for any item (Id/Symbol/Num) *)
       let lookup_choices =
         let id_choices = Hashtbl.create 1023 in
         fun item ->
-        let choices =
-          match item with
-          | Id id ->
-              (try
-                Hashtbl.find id_choices id
-              with Not_found ->
-                let choices = choices_of_id dbd id in
-                Hashtbl.add id_choices id choices;
-                choices)
-          | Symbol (symb, _) ->
-              List.map DisambiguateChoices.mk_choice
-                (CicNotationRew.lookup_interpretations symb)
-(*               DisambiguateChoices.lookup_symbol_choices symb *)
-          | Num instance -> DisambiguateChoices.lookup_num_choices ()
-        in
-        if choices = [] then raise (No_choices item);
-        choices
+          let choices =
+            let lookup_in_library () =
+              match item with
+              | Id id -> choices_of_id dbd id
+              | Symbol (symb, _) ->
+                  List.map DisambiguateChoices.mk_choice
+                    (CicNotationRew.lookup_interpretations symb)
+              | Num instance ->
+                  DisambiguateChoices.lookup_num_choices ()
+            in
+            match universe with
+            | None -> lookup_in_library ()
+            | Some e ->
+                (try
+                  Environment.find item e
+                with Not_found -> lookup_in_library ())
+          in
+          if choices = [] then raise (No_choices item);
+          choices
       in
-
 (*
       (* <benchmark> *)
       let _ =
@@ -743,8 +747,8 @@ module Make (C: Callbacks) =
               (fun dom_item ->
                 try
                   let len = List.length (lookup_choices dom_item) in
-                  debug_print (sprintf "BENCHMARK %s: %d"
-                    (string_of_domain_item dom_item) len);
+                  debug_print (lazy (sprintf "BENCHMARK %s: %d"
+                    (string_of_domain_item dom_item) len));
                   len
                 with No_choices _ -> 0)
               thing_dom
@@ -761,7 +765,7 @@ module Make (C: Callbacks) =
 
       (* (3) test an interpretation filling with meta uninterpreted identifiers
        *)
-      let test_env current_env todo_dom ugraph = 
+      let test_env aliases todo_dom ugraph = 
         let filled_env =
           List.fold_left
             (fun env item ->
@@ -770,38 +774,38 @@ module Make (C: Callbacks) =
                  (match item with
                     | Id _ | Num _ -> (fun _ _ _ -> Cic.Implicit (Some `Closed))
                     | Symbol _ -> (fun _ _ _ -> Cic.Implicit None))) env)
-            current_env todo_dom 
+            aliases todo_dom 
         in
         try
           let cic_thing =
             interpretate_thing ~context:disambiguate_context ~env:filled_env
              ~uri ~is_path:false thing
           in
+let foo () =
           let k,ugraph1 = refine_thing metasenv context uri cic_thing ugraph in
             (k , ugraph1 )
+in refine_profiler.CicUtil.profile foo ()
         with
         | Try_again -> Uncertain, ugraph
         | Invalid_choice -> Ko, ugraph
       in
       (* (4) build all possible interpretations *)
-      let rec aux current_env todo_dom base_univ =
+      let rec aux aliases todo_dom base_univ =
         match todo_dom with
         | [] ->
-            (match test_env current_env [] base_univ with
+            (match test_env aliases [] base_univ with
             | Ok (thing, metasenv),new_univ -> 
-               [ current_env, metasenv, thing, new_univ ]
+               [ aliases, metasenv, thing, new_univ ]
             | Ko,_ | Uncertain,_ -> [])
         | item :: remaining_dom ->
-            debug_print (sprintf "CHOOSED ITEM: %s"
-             (string_of_domain_item item));
+            debug_print (lazy (sprintf "CHOOSED ITEM: %s"
+             (string_of_domain_item item)));
             let choices = lookup_choices item in
             let rec filter univ = function 
               | [] -> []
               | codomain_item :: tl ->
-                  debug_print (sprintf "%s CHOSEN" (fst codomain_item)) ;
-                  let new_env =
-                    Environment.add item codomain_item current_env
-                  in
+                  debug_print (lazy (sprintf "%s CHOSEN" (fst codomain_item))) ;
+                  let new_env = Environment.add item codomain_item aliases in
                   (match test_env new_env remaining_dom univ with
                   | Ok (thing, metasenv),new_univ ->
                       (match remaining_dom with
@@ -820,13 +824,13 @@ module Make (C: Callbacks) =
       let base_univ = initial_ugraph in
       try
         let res =
-         match aux current_env todo_dom base_univ with
+         match aux aliases todo_dom base_univ with
          | [] -> raise NoWellTypedInterpretation
-         | [ e,me,t,u ] as l ->
-             debug_print "UNA SOLA SCELTA";
-             [ e,me,t,u]
+         | [ e,me,t,u ] ->
+             debug_print (lazy "SINGLE INTERPRETATION");
+             [ e,me,t,u ], false
          | l ->
-             debug_print (sprintf "PIU' SCELTE (%d)" (List.length l));
+             debug_print (lazy (sprintf "MANY INTERPRETATIONS (%d)" (List.length l)));
              let choices =
                List.map
                  (fun (env, _, _, _) ->
@@ -840,35 +844,44 @@ module Make (C: Callbacks) =
                  l
              in
              let choosed = C.interactive_interpretation_choice choices in
-             List.map (List.nth l) choosed
+             (List.map (List.nth l) choosed), true
         in
 (*
         (if benchmark then
           let res_size = List.length res in
-          debug_print (sprintf
+          debug_print (lazy (sprintf
             ("BENCHMARK: %d/%d refinements performed, domain size %d, interps %d, k %.2f\n" ^^
             "BENCHMARK:   estimated %.2f")
             !actual_refinements !max_refinements !domain_size res_size
             !choices_avg
-            (float_of_int (!domain_size - 1) *. !choices_avg *. (float_of_int res_size) +. !choices_avg)));
+            (float_of_int (!domain_size - 1) *. !choices_avg *. (float_of_int res_size) +. !choices_avg))));
 *)
-        res
+      res
      with
       CicEnvironment.CircularDependency s -> 
         failwith "Disambiguate: circular dependency"
 
-    let disambiguate_term ~dbd ~context ~metasenv
-      ?(initial_ugraph = CicUniv.empty_ugraph) ~aliases term
+    let disambiguate_term ?(fresh_instances=false) ~dbd ~context ~metasenv
+      ?(initial_ugraph = CicUniv.empty_ugraph) ~aliases ~universe term
     =
-     disambiguate_thing ~dbd ~context ~metasenv ~initial_ugraph ~aliases
-      ~uri:None ~pp_thing:CicNotationPp.pp_term ~domain_of_thing:domain_of_term
-      ~interpretate_thing:interpretate_term ~refine_thing:refine_term term
-
-    let disambiguate_obj ~dbd ~aliases ~uri obj =
-     disambiguate_thing ~dbd ~context:[] ~metasenv:[] ~aliases ~uri
-      ~pp_thing:GrafiteAstPp.pp_obj ~domain_of_thing:domain_of_obj
-      ~interpretate_thing:interpretate_obj ~refine_thing:refine_obj
-      obj
+      let term =
+        if fresh_instances then CicNotationUtil.freshen_term term else term
+      in
+      disambiguate_thing ~dbd ~context ~metasenv ~initial_ugraph ~aliases
+        ~universe ~uri:None ~pp_thing:CicNotationPp.pp_term
+        ~domain_of_thing:domain_of_term ~interpretate_thing:interpretate_term
+        ~refine_thing:refine_term term
+
+    let disambiguate_obj ?(fresh_instances=false) ~dbd ~aliases ~universe ~uri
+     obj
+    =
+      let obj =
+        if fresh_instances then CicNotationUtil.freshen_obj obj else obj
+      in
+      disambiguate_thing ~dbd ~context:[] ~metasenv:[] ~aliases ~universe ~uri
+        ~pp_thing:GrafiteAstPp.pp_obj ~domain_of_thing:domain_of_obj
+        ~interpretate_thing:interpretate_obj ~refine_thing:refine_obj
+        obj
   end
 
 module Trivial =
@@ -887,10 +900,12 @@ struct
   let disambiguate_string ~dbd ?(context = []) ?(metasenv = []) ?initial_ugraph
     ?(aliases = DisambiguateTypes.Environment.empty) term
   =
-    let ast = CicNotationParser.parse_level2_ast (Stream.of_string term) in
+    let ast =
+      CicNotationParser.parse_level2_ast (Ulexing.from_utf8_string term)
+    in
     try
-      Disambiguator.disambiguate_term ~dbd ~context ~metasenv ast
-        ?initial_ugraph ~aliases
+      fst (Disambiguator.disambiguate_term ~dbd ~context ~metasenv ast
+        ?initial_ugraph ~aliases ~universe:None)
     with Exit -> raise (Ambiguous_term term)
 end