...

[helm.git] / components / cic_disambiguation / disambiguateTypes.ml

diff --git a/components/cic_disambiguation/disambiguateTypes.ml b/components/cic_disambiguation/disambiguateTypes.ml
index 4a2e43a205dc88b25c6be961955246ca19dd5050..1eade4ca09850d1b6d6636d5b3bb8e203142e5ee 100644 (file)
--- a/components/cic_disambiguation/disambiguateTypes.ml
+++ b/components/cic_disambiguation/disambiguateTypes.ml
@@ -40,7 +40,7 @@ type domain_item =
   | Symbol of string * int     (* literal, instance num *)
   | Num of int                 (* instance num *)
 
-exception Invalid_choice of string Lazy.t
+exception Invalid_choice of Stdpp.location option * string Lazy.t
 
 module OrderedDomain =
   struct
@@ -55,6 +55,16 @@ struct
 
   include Environment'
 
+  let find k env =
+   match k with
+      Symbol (sym,n) ->
+       (try find k env
+        with Not_found -> find (Symbol (sym,0)) env)
+    | Num n ->
+       (try find k env
+        with Not_found -> find (Num 0) env)
+    | _ -> find k env
+
   let cons k v env =
     try
       let current = find k env in
@@ -105,9 +115,10 @@ module type Callbacks =
       title:string -> msg:string -> id:string -> UriManager.uri list ->
       UriManager.uri list
     val interactive_interpretation_choice:
-      (string * string) list list -> int list
+      string -> int ->
+      (Stdpp.location list * string * string) list list -> int list
     val input_or_locate_uri:
-      title:string -> ?id:string -> unit -> UriManager.uri
+      title:string -> ?id:string -> unit -> UriManager.uri option
   end
 
 let string_of_domain_item = function