disambiguation even more abstracted

[helm.git] / helm / software / components / grafite_parser / grafiteDisambiguate.ml

diff --git a/helm/software/components/grafite_parser/grafiteDisambiguate.ml b/helm/software/components/grafite_parser/grafiteDisambiguate.ml
index ad7f70ef1a331e3430228e74f4e88a71b57e689c..4d7655f6dd65234e0128bbd2884c0813f9c85a54 100644 (file)
--- a/helm/software/components/grafite_parser/grafiteDisambiguate.ml
+++ b/helm/software/components/grafite_parser/grafiteDisambiguate.ml
@@ -45,15 +45,66 @@ let singleton msg = function
       in
       HLog.debug debug; assert false
 
+
+let lookup_in_library interactive_user_uri_choice input_or_locate_uri item =
+  let dbd = LibraryDb.instance () in
+  let choices_of_id id =
+    let uris = Whelp.locate ~dbd id in
+    let uris =
+     match uris with
+      | [] ->
+         (match 
+           (input_or_locate_uri 
+             ~title:("URI matching \"" ^ id ^ "\" unknown.") 
+             ?id:(Some id) ()) 
+         with
+         | None -> []
+         | Some uri -> [uri])
+      | [uri] -> [uri]
+      | _ ->
+          interactive_user_uri_choice ~selection_mode:`MULTIPLE
+           ?ok:(Some "Try selected.") 
+           ?enable_button_for_non_vars:(Some true)
+           ~title:"Ambiguous input."
+           ~msg: ("Ambiguous input \"" ^ id ^
+              "\". Please, choose one or more interpretations:")
+           ~id
+           uris
+    in
+    List.map
+      (fun uri ->
+        (UriManager.string_of_uri uri,
+         let term =
+           try
+             CicUtil.term_of_uri uri
+           with exn ->
+             assert false
+          in
+         fun _ _ _ -> term))
+      uris
+  in
+  match item with
+  | DisambiguateTypes.Id id -> choices_of_id id
+  | DisambiguateTypes.Symbol (symb, _) ->
+   (try
+     List.map DisambiguateChoices.mk_choice
+      (TermAcicContent.lookup_interpretations symb)
+    with
+     TermAcicContent.Interpretation_not_found -> [])
+  | DisambiguateTypes.Num instance ->
+    DisambiguateChoices.lookup_num_choices ()
+;;
+
   (** @param term not meaningful when context is given *)
 let disambiguate_term goal text prefix_len lexicon_status_ref context metasenv
 term =
   let lexicon_status = !lexicon_status_ref in
   let (diff, metasenv, subst, cic, _) =
     singleton "first"
-      (GrafiteDisambiguator.disambiguate_term ~dbd:(LibraryDb.instance ())
+      (GrafiteDisambiguator.disambiguate_term
         ~aliases:lexicon_status.LexiconEngine.aliases
         ?goal ~universe:(Some lexicon_status.LexiconEngine.multi_aliases)
+        ~lookup_in_library
         ~context ~metasenv ~subst:[] (text,prefix_len,term))
   in
   let lexicon_status = LexiconEngine.set_proof_aliases lexicon_status diff in
@@ -71,7 +122,7 @@ let disambiguate_lazy_term goal text prefix_len lexicon_status_ref term =
     let lexicon_status = !lexicon_status_ref in
     let (diff, metasenv, _, cic, ugraph) =
       singleton "second"
-        (GrafiteDisambiguator.disambiguate_term ~dbd:(LibraryDb.instance ())
+        (GrafiteDisambiguator.disambiguate_term ~lookup_in_library 
           ~initial_ugraph:ugraph ~aliases:lexicon_status.LexiconEngine.aliases
           ~universe:(Some lexicon_status.LexiconEngine.multi_aliases)
           ~context ~metasenv ~subst:[] ?goal
@@ -458,7 +509,7 @@ let disambiguate_obj lexicon_status ?baseuri metasenv (text,prefix_len,obj) =
     | CicNotationPt.Theorem _ -> None in
   let (diff, metasenv, _, cic, _) =
     singleton "third"
-      (GrafiteDisambiguator.disambiguate_obj ~dbd:(LibraryDb.instance ())
+      (GrafiteDisambiguator.disambiguate_obj ~lookup_in_library
         ~aliases:lexicon_status.LexiconEngine.aliases
         ~universe:(Some lexicon_status.LexiconEngine.multi_aliases) ~uri 
         (text,prefix_len,obj)) in