]> matita.cs.unibo.it Git - helm.git/blobdiff - helm/ocaml/cic_disambiguation/disambiguate.ml
merged cic_notation with disambiguation: good luck!
[helm.git] / helm / ocaml / cic_disambiguation / disambiguate.ml
index 6848338c3bf06b5c09659bacf069df8deb1385ed..026f9ebd1ee238ad6e7b7de248fa2f4f415bec30 100644 (file)
@@ -47,6 +47,16 @@ 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
@@ -100,7 +110,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 =
+let find_in_environment name (context: Cic.name list) =
   let rec aux acc = function
     | [] -> raise Not_found
     | Cic.Name hd :: tl when hd = name -> acc
@@ -108,39 +118,42 @@ let find_in_environment name context =
   in
   aux 1 context
 
-let interpretate_term ~context ~env ~uri ~is_path ast =
+let interpretate_term ~(context: Cic.name list) ~env ~uri ~is_path ast =
   assert (uri = None);
-  let rec aux loc context = function
-    | CicAst.AttributedTerm (`Loc loc, term) ->
+  let rec aux loc (context: Cic.name list) = function
+    | CicNotationPt.AttributedTerm (`Loc loc, term) ->
         aux loc context term
-    | CicAst.AttributedTerm (_, term) -> aux loc context term
-    | CicAst.Appl (CicAst.Symbol (symb, i) :: args) ->
+    | CicNotationPt.AttributedTerm (_, term) -> aux loc context term
+    | CicNotationPt.Appl (CicNotationPt.Symbol (symb, i) :: args) ->
         let cic_args = List.map (aux loc context) args in
         resolve env (Symbol (symb, i)) ~args:cic_args ()
-    | CicAst.Appl terms -> Cic.Appl (List.map (aux loc context) terms)
-    | CicAst.Binder (binder_kind, (var, typ), body) ->
+    | CicNotationPt.Appl terms -> Cic.Appl (List.map (aux loc context) terms)
+    | CicNotationPt.Binder (binder_kind, (var, typ), body) ->
         let cic_type = aux_option loc context (Some `Type) typ in
-        let cic_body = aux loc (var :: context) body in
+        let cic_name = CicNotationUtil.cic_name_of_name var in
+        let cic_body = aux loc (cic_name :: context) body in
         (match binder_kind with
-        | `Lambda -> Cic.Lambda (var, cic_type, cic_body)
-        | `Pi | `Forall -> Cic.Prod (var, cic_type, cic_body)
+        | `Lambda -> Cic.Lambda (cic_name, cic_type, cic_body)
+        | `Pi
+        | `Forall -> Cic.Prod (cic_name, cic_type, cic_body)
         | `Exists ->
             resolve env (Symbol ("exists", 0))
-              ~args:[ cic_type; Cic.Lambda (var, cic_type, cic_body) ] ())
-    | CicAst.Case (term, indty_ident, outtype, branches) ->
+              ~args:[ cic_type; Cic.Lambda (cic_name, cic_type, cic_body) ] ())
+    | 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 rec do_branch' context = function
             | [] -> aux loc context term
             | (name, typ) :: tl ->
-                let cic_body = do_branch' (name :: context) tl in
+                let cic_name = CicNotationUtil.cic_name_of_name name in
+                let cic_body = do_branch' (cic_name :: context) tl in
                 let typ =
                   match typ with
                   | None -> Cic.Implicit (Some `Type)
                   | Some typ -> aux loc context typ
                 in
-                Cic.Lambda (name, typ, cic_body)
+                Cic.Lambda (cic_name, typ, cic_body)
           in
           do_branch' context args
         in
@@ -150,33 +163,40 @@ let interpretate_term ~context ~env ~uri ~is_path ast =
               (match resolve env (Id indty_ident) () with
               | Cic.MutInd (uri, tyno, _) -> (uri, tyno)
               | Cic.Implicit _ -> raise Try_again
-              | _ -> raise DisambiguateChoices.Invalid_choice)
+              | _ -> raise Invalid_choice)
           | None ->
               let fst_constructor =
                 match branches with
                 | ((head, _), _) :: _ -> head
-                | [] -> raise DisambiguateChoices.Invalid_choice
+                | [] -> raise Invalid_choice
               in
               (match resolve env (Id fst_constructor) () with
               | Cic.MutConstruct (indtype_uri, indtype_no, _, _) ->
                   (indtype_uri, indtype_no)
               | Cic.Implicit _ -> raise Try_again
-              | _ -> raise DisambiguateChoices.Invalid_choice)
+              | _ -> raise Invalid_choice)
         in
         Cic.MutCase (indtype_uri, indtype_no, cic_outtype, cic_term,
           (List.map do_branch branches))
-    | CicAst.LetIn ((name, typ), def, body) ->
+    | CicNotationPt.Cast (t1, t2) ->
+        let cic_t1 = aux loc context t1 in
+        let cic_t2 = aux loc context t2 in
+        Cic.Cast (cic_t1, cic_t2)
+    | CicNotationPt.LetIn ((name, typ), def, body) ->
         let cic_def = aux loc context def in
+        let cic_name = CicNotationUtil.cic_name_of_name name in
         let cic_def =
           match typ with
           | None -> cic_def
           | Some t -> Cic.Cast (cic_def, aux loc context t)
         in
-        let cic_body = aux loc (name :: context) body in
-        Cic.LetIn (name, cic_def, cic_body)
-    | CicAst.LetRec (kind, defs, body) ->
+        let cic_body = aux loc (cic_name :: context) body in
+        Cic.LetIn (cic_name, cic_def, cic_body)
+    | CicNotationPt.LetRec (kind, defs, body) ->
         let context' =
-          List.fold_left (fun acc ((name, _), _, _) -> name :: acc)
+          List.fold_left
+            (fun acc ((name, _), _, _) ->
+              CicNotationUtil.cic_name_of_name name :: acc)
             context defs
         in
         let cic_body = aux loc context' body in
@@ -186,9 +206,9 @@ let interpretate_term ~context ~env ~uri ~is_path ast =
               let cic_body = aux loc context' body in
               let cic_type = aux_option loc context (Some `Type) typ in
               let name =
-                match name with
+                match CicNotationUtil.cic_name_of_name name with
                 | Cic.Anonymous ->
-                    CicTextualParser2.fail loc
+                    CicNotationPt.fail loc
                       "Recursive functions cannot be anonymous"
                 | Cic.Name name -> name
               in
@@ -214,17 +234,16 @@ let interpretate_term ~context ~env ~uri ~is_path ast =
                 Cic.LetIn (Cic.Name var, Cic.CoFix (!counter, funs), cic))
         in
         List.fold_right (build_term inductiveFuns) inductiveFuns cic_body
-    | CicAst.Ident _
-    | CicAst.Uri _ when is_path -> raise PathNotWellFormed
-    | CicAst.Ident (name, subst)
-    | CicAst.Uri (name, subst) as ast ->
-        let is_uri = function CicAst.Uri _ -> true | _ -> false in
+    | CicNotationPt.Ident _
+    | CicNotationPt.Uri _ when is_path -> raise PathNotWellFormed
+    | CicNotationPt.Ident (name, subst)
+    | CicNotationPt.Uri (name, subst) as 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
           if subst <> None then
-            CicTextualParser2.fail loc
-              "Explicit substitutions not allowed here";
+            CicNotationPt.fail loc "Explicit substitutions not allowed here";
           Cic.Rel index
         with Not_found ->
           let cic =
@@ -232,7 +251,7 @@ let interpretate_term ~context ~env ~uri ~is_path ast =
               try
                 CicUtil.term_of_uri (UriManager.uri_of_string name)
               with UriManager.IllFormedUri _ ->
-                CicTextualParser2.fail loc "Ill formed URI"
+                CicNotationPt.fail loc "Ill formed URI"
             else
               resolve env (Id name) ()
           in
@@ -247,7 +266,7 @@ let interpretate_term ~context ~env ~uri ~is_path ast =
                     (try
                       List.assoc s ids_to_uris, aux loc context term
                      with Not_found ->
-                       raise DisambiguateChoices.Invalid_choice))
+                       raise Invalid_choice))
                   subst
             | None -> List.map (fun uri -> uri, Cic.Implicit None) uris)
           in
@@ -286,38 +305,39 @@ let interpretate_term ~context ~env ~uri ~is_path ast =
                   (CicPp.ppterm t)
                   (String.concat "; "
                     (List.map
-                      (fun (s, term) -> s ^ " := " ^ CicAstPp.pp_term term)
+                      (fun (s, term) -> s ^ " := " ^ CicNotationPtPp.pp_term term)
                       subst)));
 *)
                 t
             | _ ->
-              raise DisambiguateChoices.Invalid_choice
+              raise Invalid_choice
            with 
              CicEnvironment.CircularDependency _ -> 
-               raise DisambiguateChoices.Invalid_choice))
-    | CicAst.Implicit -> Cic.Implicit None
-    | CicAst.UserInput -> Cic.Implicit (Some `Hole)
-    | CicAst.Num (num, i) -> resolve env (Num i) ~num ()
-    | CicAst.Meta (index, subst) ->
+               raise Invalid_choice))
+    | CicNotationPt.Implicit -> Cic.Implicit None
+    | CicNotationPt.UserInput -> Cic.Implicit (Some `Hole)
+    | CicNotationPt.Num (num, i) -> resolve env (Num i) ~num ()
+    | CicNotationPt.Meta (index, subst) ->
         let cic_subst =
           List.map
             (function None -> None | Some term -> Some (aux loc context term))
             subst
         in
         Cic.Meta (index, cic_subst)
-    | CicAst.Sort `Prop -> Cic.Sort Cic.Prop
-    | CicAst.Sort `Set -> Cic.Sort Cic.Set
-    | CicAst.Sort `Type -> Cic.Sort (Cic.Type (CicUniv.fresh())) (* TASSI *)
-    | CicAst.Sort `CProp -> Cic.Sort Cic.CProp
-    | CicAst.Symbol (symbol, instance) ->
+    | CicNotationPt.Sort `Prop -> Cic.Sort Cic.Prop
+    | CicNotationPt.Sort `Set -> Cic.Sort Cic.Set
+    | CicNotationPt.Sort `Type -> Cic.Sort (Cic.Type (CicUniv.fresh())) (* TASSI *)
+    | CicNotationPt.Sort `CProp -> Cic.Sort Cic.CProp
+    | CicNotationPt.Symbol (symbol, instance) ->
         resolve env (Symbol (symbol, instance)) ()
-  and aux_option loc context annotation = function
+    | _ -> assert false (* god bless Bologna *)
+  and aux_option loc (context: Cic.name list) annotation = function
     | None -> Cic.Implicit annotation
     | Some term -> aux loc context term
   in
   match ast with
-  | CicAst.AttributedTerm (`Loc loc, term) -> aux loc context term
-  | term -> aux CicAst.dummy_floc context term
+  | 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
@@ -326,13 +346,13 @@ let interpretate_obj ~context ~env ~uri ~is_path obj =
  assert (context = []);
  assert (is_path = false);
  match obj with
-  | TacticAst.Inductive (params,tyl) ->
+  | GrafiteAst.Inductive (params,tyl) ->
      let uri = match uri with Some uri -> uri | None -> assert false in
      let context,params =
       let context,res =
        List.fold_left
         (fun (context,res) (name,t) ->
-          (Cic.Name name)::context,
+          Cic.Name name :: context,
           (name, interpretate_term context env None false t)::res
         ) ([],[]) params
       in
@@ -374,13 +394,13 @@ let interpretate_obj ~context ~env ~uri ~is_path obj =
        ) tyl
      in
       Cic.InductiveDefinition (tyl,[],List.length params,[])
-  | TacticAst.Record (params,name,ty,fields) ->
+  | GrafiteAst.Record (params,name,ty,fields) ->
      let uri = match uri with Some uri -> uri | None -> assert false in
      let context,params =
       let context,res =
        List.fold_left
         (fun (context,res) (name,t) ->
-          (Cic.Name name)::context,
+          (Cic.Name name :: context),
           (name, interpretate_term context env None false t)::res
         ) ([],[]) params
       in
@@ -393,7 +413,7 @@ let interpretate_obj ~context ~env ~uri ~is_path obj =
       snd (
        List.fold_left
         (fun (context,res) (name,ty) ->
-          let context' = (Cic.Name name)::context in
+          let context' = Cic.Name name :: context in
            context',(name,interpretate_term context env None false ty)::res
         ) (context,[]) fields) in
      let concl =
@@ -412,7 +432,7 @@ let interpretate_obj ~context ~env ~uri ~is_path obj =
      let field_names = List.map fst fields in
       Cic.InductiveDefinition
        (tyl,[],List.length params,[`Class (`Record field_names)])
-  | TacticAst.Theorem (flavour, name, ty, bo) ->
+  | GrafiteAst.Theorem (flavour, name, ty, bo) ->
      let attrs = [`Flavour flavour] in
      let ty' = interpretate_term [] env None false ty in
      (match bo with
@@ -448,23 +468,27 @@ let rev_uniq =
 (* "aux" keeps domain in reverse order and doesn't care about duplicates.
  * Domain item more in deep in the list will be processed first.
  *)
-let rec domain_rev_of_term ?(loc = CicAst.dummy_floc) context = function
-  | CicAst.AttributedTerm (`Loc loc, term) ->
+let rec domain_rev_of_term ?(loc = dummy_floc) context = function
+  | CicNotationPt.AttributedTerm (`Loc loc, term) ->
      domain_rev_of_term ~loc context term
-  | CicAst.AttributedTerm (_, term) -> domain_rev_of_term ~loc context term
-  | CicAst.Appl terms ->
+  | CicNotationPt.AttributedTerm (_, term) ->
+      domain_rev_of_term ~loc context term
+  | CicNotationPt.Appl terms ->
       List.fold_left
        (fun dom term -> domain_rev_of_term ~loc context term @ dom) [] terms
-  | CicAst.Binder (kind, (var, typ), body) ->
+  | CicNotationPt.Binder (kind, (var, typ), body) ->
       let kind_dom =
         match kind with
         | `Exists -> [ Symbol ("exists", 0) ]
         | _ -> []
       in
       let type_dom = domain_rev_of_term_option loc context typ in
-      let body_dom = domain_rev_of_term ~loc (var :: context) body in
+      let body_dom =
+        domain_rev_of_term ~loc
+          (CicNotationUtil.cic_name_of_name var :: context) body
+      in
       body_dom @ type_dom @ kind_dom
-  | CicAst.Case (term, indty_ident, outtype, branches) ->
+  | CicNotationPt.Case (term, indty_ident, outtype, branches) ->
       let term_dom = domain_rev_of_term ~loc context term in
       let outtype_dom = domain_rev_of_term_option loc context outtype in
       let get_first_constructor = function
@@ -475,7 +499,7 @@ let rec domain_rev_of_term ?(loc = CicAst.dummy_floc) context = function
         let (term_context, args_domain) =
           List.fold_left
             (fun (cont, dom) (name, typ) ->
-              (name :: cont,
+              (CicNotationUtil.cic_name_of_name name :: cont,
                (match typ with
                | None -> dom
                | Some typ -> domain_rev_of_term ~loc cont typ @ dom)))
@@ -490,14 +514,23 @@ let rec domain_rev_of_term ?(loc = CicAst.dummy_floc) context = function
       (match indty_ident with
        | None -> get_first_constructor branches
        | Some ident -> [ Id ident ])
-  | CicAst.LetIn ((var, typ), body, where) ->
+  | 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
+      ty_dom @ term_dom
+  | CicNotationPt.LetIn ((var, typ), body, where) ->
       let body_dom = domain_rev_of_term ~loc context body in
       let type_dom = domain_rev_of_term_option loc context typ in
-      let where_dom = domain_rev_of_term ~loc (var :: context) where in
+      let where_dom =
+        domain_rev_of_term ~loc
+          (CicNotationUtil.cic_name_of_name var :: context) where
+      in
       where_dom @ type_dom @ body_dom
-  | CicAst.LetRec (kind, defs, where) ->
+  | CicNotationPt.LetRec (kind, defs, where) ->
       let context' =
-        List.fold_left (fun acc ((var, typ), _, _) -> var :: acc)
+        List.fold_left
+          (fun acc ((var, typ), _, _) ->
+            CicNotationUtil.cic_name_of_name var :: acc)
           context defs
       in
       let where_dom = domain_rev_of_term ~loc context' where in
@@ -509,12 +542,11 @@ let rec domain_rev_of_term ?(loc = CicAst.dummy_floc) context = function
           [] defs
       in
       where_dom @ defs_dom
-  | CicAst.Ident (name, subst) ->
+  | CicNotationPt.Ident (name, subst) ->
       (try
         let index = find_in_environment name context in
         if subst <> None then
-          CicTextualParser2.fail loc
-            "Explicit substitutions not allowed here"
+          CicNotationPt.fail loc "Explicit substitutions not allowed here"
         else
           []
       with Not_found ->
@@ -526,16 +558,20 @@ let rec domain_rev_of_term ?(loc = CicAst.dummy_floc) context = function
                 let dom' = domain_rev_of_term ~loc context term in
                 dom' @ dom)
               [Id name] subst))
-  | CicAst.Uri _ -> []
-  | CicAst.Implicit -> []
-  | CicAst.Num (num, i) -> [ Num i ]
-  | CicAst.Meta (index, local_context) ->
+  | CicNotationPt.Uri _ -> []
+  | CicNotationPt.Implicit -> []
+  | CicNotationPt.Num (num, i) -> [ Num i ]
+  | CicNotationPt.Meta (index, local_context) ->
       List.fold_left
        (fun dom term -> domain_rev_of_term_option loc context term @ dom) []
         local_context
-  | CicAst.Sort _ -> []
-  | CicAst.Symbol (symbol, instance) -> [ Symbol (symbol, instance) ]
-  | CicAst.UserInput -> assert false
+  | CicNotationPt.Sort _ -> []
+  | CicNotationPt.Symbol (symbol, instance) -> [ Symbol (symbol, instance) ]
+  | CicNotationPt.UserInput
+  | CicNotationPt.Literal _
+  | CicNotationPt.Layout _
+  | CicNotationPt.Magic _
+  | CicNotationPt.Variable _ -> assert false
 
 and domain_rev_of_term_option loc context = function
   | None -> []
@@ -548,12 +584,12 @@ let domain_of_obj ~context ast =
  assert (context = []);
  let domain_rev =
   match ast with
-   | TacticAst.Theorem (_,_,ty,bo) ->
+   | GrafiteAst.Theorem (_,_,ty,bo) ->
       (match bo with
           None -> []
         | Some bo -> domain_rev_of_term [] bo) @
       domain_of_term [] ty
-   | TacticAst.Inductive (params,tyl) ->
+   | GrafiteAst.Inductive (params,tyl) ->
       let dom =
        List.flatten (
         List.rev_map
@@ -573,7 +609,7 @@ let domain_of_obj ~context ast =
           not (  List.exists (fun (name',_) -> name = Id name') params
               || List.exists (fun (name',_,_,_) -> name = Id name') tyl)
         ) dom
-   | TacticAst.Record (params,_,ty,fields) ->
+   | GrafiteAst.Record (params,_,ty,fields) ->
       let dom =
        List.flatten
         (List.rev_map (fun (_,ty) -> domain_rev_of_term [] ty) fields) in
@@ -608,7 +644,7 @@ sig
     metasenv:Cic.metasenv ->
     ?initial_ugraph:CicUniv.universe_graph -> 
     aliases:environment ->  (* previous interpretation status *)
-    CicAst.term ->
+    CicNotationPt.term ->
     (environment *                   (* new interpretation status *)
      Cic.metasenv *                  (* new metasenv *)
      Cic.term*
@@ -618,7 +654,7 @@ sig
     dbd:Mysql.dbd ->
     aliases:environment ->           (* previous interpretation status *)
     uri:UriManager.uri option ->     (* required only for inductive types *)
-    TacticAst.obj ->
+    GrafiteAst.obj ->
     (environment *                   (* new interpretation status *)
      Cic.metasenv *                  (* new metasenv *)
      Cic.obj *
@@ -688,7 +724,10 @@ module Make (C: Callbacks) =
                 let choices = choices_of_id dbd id in
                 Hashtbl.add id_choices id choices;
                 choices)
-          | Symbol (symb, _) -> DisambiguateChoices.lookup_symbol_choices symb
+          | 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);
@@ -741,8 +780,8 @@ module Make (C: Callbacks) =
           let k,ugraph1 = refine_thing metasenv context uri cic_thing ugraph in
             (k , ugraph1 )
         with
-        | Try_again -> Uncertain,ugraph
-        | DisambiguateChoices.Invalid_choice -> Ko,ugraph
+        | Try_again -> Uncertain, ugraph
+        | Invalid_choice -> Ko, ugraph
       in
       (* (4) build all possible interpretations *)
       let rec aux current_env todo_dom base_univ =
@@ -822,12 +861,12 @@ module Make (C: Callbacks) =
       ?(initial_ugraph = CicUniv.empty_ugraph) ~aliases term
     =
      disambiguate_thing ~dbd ~context ~metasenv ~initial_ugraph ~aliases
-      ~uri:None ~pp_thing:CicAstPp.pp_term ~domain_of_thing:domain_of_term
+      ~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:TacticAstPp.pp_obj ~domain_of_thing:domain_of_obj
+      ~pp_thing:GrafiteAstPp.pp_obj ~domain_of_thing:domain_of_obj
       ~interpretate_thing:interpretate_obj ~refine_thing:refine_obj
       obj
   end
@@ -845,9 +884,10 @@ struct
     let input_or_locate_uri ~(title:string) ?id = raise Exit
   end
   module Disambiguator = Make (Callbacks)
-  let disambiguate_string ~dbd ?(context=[]) ?(metasenv=[]) ?initial_ugraph
-        ?(aliases=DisambiguateTypes.Environment.empty) term =
-    let ast = CicTextualParser2.parse_term (Stream.of_string term) in
+  let disambiguate_string ~dbd ?(context = []) ?(metasenv = []) ?initial_ugraph
+    ?(aliases = DisambiguateTypes.Environment.empty) term
+  =
+    let ast = CicNotationParser.parse_level2_ast (Stream.of_string term) in
     try
       Disambiguator.disambiguate_term ~dbd ~context ~metasenv ast
         ?initial_ugraph ~aliases