]> matita.cs.unibo.it Git - helm.git/blob - helm/matita/matitaDisambiguator.ml
A bit of profiling functions added here and there.
[helm.git] / helm / matita / matitaDisambiguator.ml
1 (* Copyright (C) 2004-2005, HELM Team.
2  * 
3  * This file is part of HELM, an Hypertextual, Electronic
4  * Library of Mathematics, developed at the Computer Science
5  * Department, University of Bologna, Italy.
6  * 
7  * HELM is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License
9  * as published by the Free Software Foundation; either version 2
10  * of the License, or (at your option) any later version.
11  * 
12  * HELM is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with HELM; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
20  * MA  02111-1307, USA.
21  * 
22  * For details, see the HELM World-Wide-Web page,
23  * http://helm.cs.unibo.it/
24  *)
25
26 open Printf
27
28 open MatitaTypes
29
30 exception Ambiguous_input
31
32 type choose_uris_callback =
33   id:string -> UriManager.uri list -> UriManager.uri list
34
35 type choose_interp_callback = (string * string) list list -> int list
36
37 let mono_uris_callback ~id =
38  if Helm_registry.get_bool "matita.auto_disambiguation" then
39   function l -> l
40  else
41   raise Ambiguous_input
42
43 let mono_interp_callback _ = raise Ambiguous_input
44
45 let _choose_uris_callback = ref mono_uris_callback
46 let _choose_interp_callback = ref mono_interp_callback
47 let set_choose_uris_callback f = _choose_uris_callback := f
48 let set_choose_interp_callback f = _choose_interp_callback := f
49
50 module Callbacks =
51   struct
52     let interactive_user_uri_choice ~selection_mode ?ok
53           ?(enable_button_for_non_vars = true) ~title ~msg ~id uris =
54               !_choose_uris_callback ~id uris
55
56     let interactive_interpretation_choice interp =
57       !_choose_interp_callback interp
58
59     let input_or_locate_uri ~(title:string) ?id =
60       (* Zack: I try to avoid using this callback. I therefore assume that
61       * the presence of an identifier that can't be resolved via "locate"
62       * query is a syntax error *)
63       let msg = match id with Some id -> id | _ -> "_" in
64       raise (Unbound_identifier msg)
65   end
66   
67 module Disambiguator = Disambiguate.Make (Callbacks)
68
69 (* implement module's API *)
70
71 let disambiguate_thing ~aliases ~universe
72   ~(f:?fresh_instances:bool ->
73       aliases:DisambiguateTypes.environment ->
74       universe:DisambiguateTypes.multiple_environment option ->
75       'a -> 'b)
76   ~(set_aliases: DisambiguateTypes.environment -> 'b -> 'b)
77   (thing: 'a)
78 =
79   assert (universe <> None);
80   let library = false, DisambiguateTypes.Environment.empty, None in
81   let multi_aliases=false, DisambiguateTypes.Environment.empty, universe in
82   let mono_aliases = true, aliases, None in
83   let passes =  (* <fresh_instances?, aliases, coercions?> *)
84     [ (false, mono_aliases, false);
85       (false, multi_aliases, false);
86       (true, mono_aliases, false);
87       (true, multi_aliases, false);
88       (true, mono_aliases, true);
89       (true, multi_aliases, true);
90       (true, library, true);
91     ]
92   in
93   let try_pass (fresh_instances, (_, aliases, universe), use_coercions) =
94     CoercDb.use_coercions := use_coercions;
95     f ~fresh_instances ~aliases ~universe thing
96   in
97   let set_aliases (instances,(use_mono_aliases,_,_),_) (_, user_asked as res) =
98    if use_mono_aliases && not instances then
99     res
100    else if user_asked then
101     res (* one shot aliases *)
102    else
103     set_aliases aliases res
104   in
105   let rec aux =
106     function
107     | [ pass ] -> set_aliases pass (try_pass pass)
108     | hd :: tl ->
109         (try
110           set_aliases hd (try_pass hd)
111         with Disambiguate.NoWellTypedInterpretation -> aux tl)
112     | [] -> assert false
113   in
114   let saved_use_coercions = !CoercDb.use_coercions in
115   try
116     let res = aux passes in
117     CoercDb.use_coercions := saved_use_coercions;
118     res
119   with exn ->
120     CoercDb.use_coercions := saved_use_coercions;
121     raise exn
122
123 let disambiguate_thing =
124  let profiler = CicUtil.profile "disambiguate_thing" in
125   fun ~aliases ~universe
126       ~(f:?fresh_instances:bool ->
127           aliases:DisambiguateTypes.environment ->
128           universe:DisambiguateTypes.multiple_environment option ->
129           'a -> 'b)
130       ~(set_aliases: DisambiguateTypes.environment -> 'b -> 'b)
131       (thing: 'a)
132   -> profiler.CicUtil.profile
133       (disambiguate_thing ~aliases ~universe ~f ~set_aliases) thing
134
135 let disambiguate_thing ~aliases ~universe
136   ~(f:?fresh_instances:bool ->
137       aliases:DisambiguateTypes.environment ->
138       universe:DisambiguateTypes.multiple_environment option ->
139       'a -> 'b)
140   ~(set_aliases: DisambiguateTypes.environment -> 'b -> 'b)
141   (thing: 'a)
142 =
143   Obj.magic disambiguate_thing ~aliases ~universe ~f ~set_aliases thing
144
145 let set_aliases aliases (choices, user_asked) =
146   (List.map (fun (_, a, b, c) -> aliases, a, b, c) choices),
147   user_asked
148
149 let disambiguate_term ?fresh_instances ~dbd ~context ~metasenv ?initial_ugraph
150   ~aliases ~universe term
151  =
152   assert (fresh_instances = None);
153   let f =
154     Disambiguator.disambiguate_term ~dbd ~context ~metasenv ?initial_ugraph
155   in
156   disambiguate_thing ~aliases ~universe ~f ~set_aliases term
157
158 let disambiguate_obj ?fresh_instances ~dbd ~aliases ~universe ~uri obj =
159   assert (fresh_instances = None);
160   let f = Disambiguator.disambiguate_obj ~dbd ~uri in
161   disambiguate_thing ~aliases ~universe ~f ~set_aliases obj