1 (* Copyright (C) 2004, HELM Team.
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.
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.
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.
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,
22 * For details, see the HELM World-Wide-Web page,
23 * http://helm.cs.unibo.it/
28 open DisambiguateTypes
31 exception No_choices of domain_item
32 exception NoWellTypedInterpretation
34 (** raised when an environment is not enough informative to decide *)
38 let debug_print = if debug then prerr_endline else ignore
40 let descr_of_domain_item = function
43 | Num i -> string_of_int i
46 | Ok of Cic.term * Cic.metasenv
50 let refine metasenv context term =
51 let metasenv, term = CicMkImplicit.expand_implicits metasenv context term in
52 debug_print (sprintf "TEST_INTERPRETATION: %s" (CicPp.ppterm term));
54 let term', _, metasenv' = CicRefine.type_of_aux' metasenv context term in
57 | CicRefine.Uncertain _ ->
58 debug_print ("%%% UNCERTAIN!!! " ^ CicPp.ppterm term) ;
61 debug_print ("%%% PRUNED!!! " ^ CicPp.ppterm term) ;
64 let resolve (env: environment) (item: domain_item) ?(num = "") ?(args = []) () =
65 snd (Environment.find item env) env num args
67 (* TODO move it to Cic *)
68 let find_in_environment name context =
69 let rec aux acc = function
70 | [] -> raise Not_found
71 | Cic.Name hd :: tl when hd = name -> acc
72 | _ :: tl -> aux (acc + 1) tl
76 let interpretate ~context ~env ast =
77 let rec aux loc context = function
78 | CicTextualParser2Ast.AttributedTerm (`Loc loc, term) ->
80 | CicTextualParser2Ast.AttributedTerm (_, term) -> aux loc context term
81 | CicTextualParser2Ast.Appl terms ->
82 Cic.Appl (List.map (aux loc context) terms)
83 | CicTextualParser2Ast.Appl_symbol (symb, i, args) ->
84 let cic_args = List.map (aux loc context) args in
85 resolve env (Symbol (symb, i)) ~args:cic_args ()
86 | CicTextualParser2Ast.Binder (binder_kind, (var, typ), body) ->
87 let cic_type = aux_option loc context typ in
88 let cic_body = aux loc (var :: context) body in
89 (match binder_kind with
90 | `Lambda -> Cic.Lambda (var, cic_type, cic_body)
91 | `Pi | `Forall -> Cic.Prod (var, cic_type, cic_body)
93 resolve env (Symbol ("exists", 0))
94 ~args:[ cic_type; Cic.Lambda (var, cic_type, cic_body) ] ())
95 | CicTextualParser2Ast.Case (term, indty_ident, outtype, branches) ->
96 let cic_term = aux loc context term in
97 let cic_outtype = aux_option loc context outtype in
98 let do_branch ((head, args), term) =
99 let rec do_branch' context = function
100 | [] -> aux loc context term
101 | (name, typ) :: tl ->
102 let cic_body = do_branch' (name :: context) tl in
105 | None -> Cic.Implicit
106 | Some typ -> aux loc context typ
108 Cic.Lambda (name, typ, cic_body)
110 do_branch' context args
112 let (indtype_uri, indtype_no) =
113 match resolve env (Id indty_ident) () with
114 | Cic.MutInd (uri, tyno, _) -> uri, tyno
115 | Cic.Implicit -> raise Try_again
116 | _ -> raise DisambiguateChoices.Invalid_choice
118 Cic.MutCase (indtype_uri, indtype_no, cic_outtype, cic_term,
119 (List.map do_branch branches))
120 | CicTextualParser2Ast.LetIn ((name, typ), def, body) ->
121 let cic_def = aux loc context def in
125 | Some t -> Cic.Cast (cic_def, aux loc context t)
127 let cic_body = aux loc (name :: context) body in
128 Cic.LetIn (name, cic_def, cic_body)
129 | CicTextualParser2Ast.LetRec (kind, defs, body) ->
131 List.fold_left (fun acc ((name, _), _, _) -> name :: acc)
134 let cic_body = aux loc context' body in
137 (fun ((name, typ), body, decr_idx) ->
138 let cic_body = aux loc context' body in
139 let cic_type = aux_option loc context typ in
143 CicTextualParser2.fail loc
144 "Recursive functions cannot be anonymous"
145 | Cic.Name name -> name
147 (name, decr_idx, cic_type, cic_body))
150 let counter = ref ~-1 in
151 let build_term funs =
152 (* this is the body of the fold_right function below. Rationale: Fix
153 * and CoFix cases differs only in an additional index in the
154 * inductiveFun list, see Cic.term *)
157 (fun (var, _, _, _) cic ->
159 Cic.LetIn (Cic.Name var, Cic.Fix (!counter, funs), cic))
162 List.map (fun (name, _, typ, body) -> (name, typ, body)) funs
164 (fun (var, _, _, _) cic ->
166 Cic.LetIn (Cic.Name var, Cic.CoFix (!counter, funs), cic))
168 List.fold_right (build_term inductiveFuns) inductiveFuns cic_body
169 | CicTextualParser2Ast.Ident (name, subst) ->
170 (* TODO hanlde explicit substitutions *)
172 let index = find_in_environment name context in
174 CicTextualParser2.fail loc
175 "Explicit substitutions not allowed here";
177 with Not_found -> resolve env (Id name) ())
178 | CicTextualParser2Ast.Num (num, i) -> resolve env (Num i) ~num ()
179 | CicTextualParser2Ast.Meta (index, subst) ->
182 (function None -> None | Some term -> Some (aux loc context term))
185 Cic.Meta (index, cic_subst)
186 | CicTextualParser2Ast.Sort `Prop -> Cic.Sort Cic.Prop
187 | CicTextualParser2Ast.Sort `Set -> Cic.Sort Cic.Set
188 | CicTextualParser2Ast.Sort `Type -> Cic.Sort Cic.Type
189 | CicTextualParser2Ast.Sort `CProp -> Cic.Sort Cic.CProp
190 and aux_option loc context = function
191 | None -> Cic.Implicit
192 | Some term -> aux loc context term
195 | CicTextualParser2Ast.AttributedTerm (`Loc loc, term) -> aux loc context term
198 let domain_of_term ~context ast =
199 let rec aux loc context = function
200 | CicTextualParser2Ast.AttributedTerm (`Loc loc, term) ->
202 | CicTextualParser2Ast.AttributedTerm (_, term) -> aux loc context term
203 | CicTextualParser2Ast.Appl terms ->
204 List.fold_left (fun dom term -> Domain.union dom (aux loc context term))
206 | CicTextualParser2Ast.Appl_symbol (symb, i, args) ->
207 List.fold_left (fun dom term -> Domain.union dom (aux loc context term))
208 (Domain.singleton (Symbol (symb, i))) args
209 | CicTextualParser2Ast.Binder (_, (var, typ), body) ->
210 let type_dom = aux_option loc context typ in
211 let body_dom = aux loc (var :: context) body in
212 Domain.union type_dom body_dom
213 | CicTextualParser2Ast.Case (term, indty_ident, outtype, branches) ->
214 let term_dom = aux loc context term in
215 let outtype_dom = aux_option loc context outtype in
216 let do_branch ((head, args), term) =
217 let (term_context, args_domain) =
219 (fun (cont, dom) (name, typ) ->
223 | Some typ -> Domain.union dom (aux loc cont typ))))
224 (context, Domain.empty) args
226 Domain.union (aux loc term_context term) args_domain
229 List.fold_left (fun dom branch -> Domain.union dom (do_branch branch))
230 Domain.empty branches
232 Domain.add (Id indty_ident)
233 (Domain.union outtype_dom (Domain.union term_dom branches_dom))
234 | CicTextualParser2Ast.LetIn ((var, typ), body, where) ->
235 let body_dom = aux loc context body in
238 | None -> Domain.empty
239 | Some typ -> aux loc context typ
241 let where_dom = aux loc (var :: context) where in
242 Domain.union (Domain.union body_dom where_dom) type_dom
243 | CicTextualParser2Ast.LetRec (kind, defs, where) ->
245 List.fold_left (fun acc ((var, typ), _, _) -> var :: acc)
248 let where_dom = aux loc context' where in
251 (fun dom ((_, typ), body, _) ->
252 Domain.union (aux loc context' body) (aux_option loc context typ))
255 Domain.union where_dom defs_dom
256 | CicTextualParser2Ast.Ident (name, subst) ->
257 (* TODO hanlde explicit substitutions *)
259 let index = find_in_environment name context in
261 CicTextualParser2.fail loc
262 "Explicit substitutions not allowed here";
264 with Not_found -> Domain.singleton (Id name))
265 | CicTextualParser2Ast.Num (num, i) -> Domain.singleton (Num i)
266 | CicTextualParser2Ast.Meta (index, local_context) ->
268 (fun dom term -> Domain.union dom (aux_option loc context term))
269 Domain.empty local_context
270 | CicTextualParser2Ast.Sort _ -> Domain.empty
271 and aux_option loc context = function
272 | None -> Domain.empty
273 | Some t -> aux loc context t
276 | CicTextualParser2Ast.AttributedTerm (`Loc loc, term) -> aux loc context term
279 module Make (C: Callbacks) =
281 let choices_of_id mqi_handle id =
282 let query = MQueryGenerator.locate id in
283 let result = MQueryInterpreter.execute mqi_handle query in
287 MQueryMisc.wrong_xpointer_format_from_wrong_xpointer_format' uri
289 C.output_html (`Msg (`T "Locate query:"));
290 MQueryUtil.text_of_query
291 (fun s -> C.output_html ~append_NL:false (`Msg (`T s)))
293 C.output_html (`Msg (`T "Result:"));
294 MQueryUtil.text_of_result
295 (fun s -> C.output_html (`Msg (`T s))) "" result;
299 [UriManager.string_of_uri (C.input_or_locate_uri
300 ~title:("URI matching \"" ^ id ^ "\" unknown."))]
303 C.interactive_user_uri_choice ~selection_mode:`MULTIPLE
304 ~ok:"Try selected." ~enable_button_for_non_vars:true
305 ~title:"Ambiguous input." ~id
306 ~msg: ("Ambiguous input \"" ^ id ^
307 "\". Please, choose one or more interpretations:")
315 HelmLibraryObjects.term_of_uri (UriManager.uri_of_string uri)
316 with _ -> assert false
321 let disambiguate_term mqi_handle context metasenv term ~aliases:current_env
323 let current_dom = (* TODO temporary, remove ASAP *)
324 Environment.fold (fun item _ dom -> Domain.add item dom)
325 current_env Domain.empty
327 debug_print "NEW DISAMBIGUATE INPUT";
328 let disambiguate_context = (* cic context -> disambiguate context *)
330 (function None -> Cic.Anonymous | Some (name, _) -> name)
333 let term_dom = domain_of_term ~context:disambiguate_context term in
334 debug_print (sprintf "DISAMBIGUATION DOMAIN: %s"
335 (string_of_domain term_dom));
336 let todo_dom = Domain.diff term_dom current_dom in
337 (* (2) lookup function for any item (Id/Symbol/Num) *)
339 let id_choices = Hashtbl.create 1023 in
345 Hashtbl.find id_choices id
347 let choices = choices_of_id mqi_handle id in
348 Hashtbl.add id_choices id choices;
350 | Symbol (symb, _) -> DisambiguateChoices.lookup_symbol_choices symb
351 | Num instance -> DisambiguateChoices.lookup_num_choices ()
353 if choices = [] then raise (No_choices item);
356 (* (3) test an interpretation filling with meta uninterpreted identifiers
358 let test_env current_env todo_dom =
362 Environment.add item ("Implicit", fun _ _ _ -> Cic.Implicit) env)
367 interpretate ~context:disambiguate_context ~env:filled_env term
369 refine metasenv context cic_term
371 | Try_again -> Uncertain
372 | DisambiguateChoices.Invalid_choice -> Ko
374 (* (4) build all possible interpretations *)
375 let rec aux current_env todo_dom =
376 if Domain.is_empty todo_dom then
377 match test_env current_env Domain.empty with
378 | Ok (term, metasenv) -> [ current_env, term, metasenv ]
379 | Ko | Uncertain -> []
381 let item = Domain.choose todo_dom in
382 let remaining_dom = Domain.remove item todo_dom in
383 debug_print (sprintf "CHOOSED ITEM: %s" (string_of_domain_item item));
384 let choices = lookup_choices item in
385 let rec filter = function
387 | codomain_item :: tl ->
388 debug_print (sprintf "%s CHOSEN" (fst codomain_item)) ;
389 let new_env = Environment.add item codomain_item current_env in
390 (match test_env new_env remaining_dom with
391 | Ok (term, metasenv) ->
392 (if Domain.is_empty remaining_dom then
393 [ new_env, term, metasenv ]
395 aux new_env remaining_dom)
398 (if Domain.is_empty remaining_dom then
401 aux new_env remaining_dom)
407 let (choosed_env, choosed_term, choosed_metasenv) =
408 match aux current_env todo_dom with
409 | [] -> raise NoWellTypedInterpretation
411 debug_print "UNA SOLA SCELTA";
414 debug_print (sprintf "PIU' SCELTE (%d)" (List.length l));
421 fst (Environment.find domain_item env)
423 (descr_of_domain_item domain_item, description))
424 (Domain.elements term_dom))
427 let choosed = C.interactive_interpretation_choice choices in
430 (choosed_env, choosed_metasenv, choosed_term)