]> matita.cs.unibo.it Git - helm.git/blobdiff - helm/ocaml/cic_notation/cicNotationParser.ml
use uniform naming for referencing cicNotation* modules
[helm.git] / helm / ocaml / cic_notation / cicNotationParser.ml
index 7c52560bec68b7517a52ff5db2d9e55cc13822bf..13817e248e08b013c31c212cb92a5fdb8478f24d 100644 (file)
@@ -25,8 +25,8 @@
 
 open Printf
 
-open CicNotationEnv
-open CicNotationPt
+module Ast = CicNotationPt
+module Env = CicNotationEnv
 
 exception Parse_error of Token.flocation * string
 exception Level_not_found of int
@@ -70,33 +70,35 @@ let gram_of_literal =
 
 type binding =
   | NoBinding
-  | Binding of string * value_type
-  | Env of (string * value_type) list
+  | Binding of string * Env.value_type
+  | Env of (string * Env.value_type) list
 
 let make_action action bindings =
   let rec aux (vl : CicNotationEnv.t) =
     function
-      [] -> Gramext.action (fun (loc: location) -> action vl loc)
+      [] -> Gramext.action (fun (loc: Ast.location) -> action vl loc)
     | NoBinding :: tl -> Gramext.action (fun _ -> aux vl tl)
     (* LUCA: DEFCON 5 BEGIN *)
-    | Binding (name, TermType) :: tl ->
+    | Binding (name, Env.TermType) :: tl ->
         Gramext.action
-          (fun (v:term) -> aux ((name, (TermType, TermValue v))::vl) tl)
-    | Binding (name, StringType) :: tl ->
+          (fun (v:Ast.term) ->
+            aux ((name, (Env.TermType, Env.TermValue v))::vl) tl)
+    | Binding (name, Env.StringType) :: tl ->
         Gramext.action
           (fun (v:string) ->
-            aux ((name, (StringType, StringValue v)) :: vl) tl)
-    | Binding (name, NumType) :: tl ->
+            aux ((name, (Env.StringType, Env.StringValue v)) :: vl) tl)
+    | Binding (name, Env.NumType) :: tl ->
         Gramext.action
-          (fun (v:string) -> aux ((name, (NumType, NumValue v)) :: vl) tl)
-    | Binding (name, OptType t) :: tl ->
+          (fun (v:string) ->
+            aux ((name, (Env.NumType, Env.NumValue v)) :: vl) tl)
+    | Binding (name, Env.OptType t) :: tl ->
         Gramext.action
           (fun (v:'a option) ->
-            aux ((name, (OptType t, OptValue v)) :: vl) tl)
-    | Binding (name, ListType t) :: tl ->
+            aux ((name, (Env.OptType t, Env.OptValue v)) :: vl) tl)
+    | Binding (name, Env.ListType t) :: tl ->
         Gramext.action
           (fun (v:'a list) ->
-            aux ((name, (ListType t, ListValue v)) :: vl) tl)
+            aux ((name, (Env.ListType t, Env.ListValue v)) :: vl) tl)
     | Env _ :: tl ->
         Gramext.action (fun (v:CicNotationEnv.t) -> aux (v @ vl) tl)
     (* LUCA: DEFCON 5 END *)
@@ -116,11 +118,11 @@ let flatten_opt =
   (* given a level 1 pattern computes the new RHS of "term" grammar entry *)
 let extract_term_production pattern =
   let rec aux = function
-    | AttributedTerm (_, t) -> aux t
-    | Literal l -> aux_literal l
-    | Layout l -> aux_layout l
-    | Magic m -> aux_magic m
-    | Variable v -> aux_variable v
+    | Ast.AttributedTerm (_, t) -> aux t
+    | Ast.Literal l -> aux_literal l
+    | Ast.Layout l -> aux_layout l
+    | Ast.Magic m -> aux_magic m
+    | Ast.Variable v -> aux_variable v
     | t ->
         prerr_endline (CicNotationPp.pp_term t);
         assert false
@@ -132,35 +134,35 @@ let extract_term_production pattern =
         [NoBinding, gram_keyword s]
     | `Number s -> [NoBinding, gram_number s]
   and aux_layout = function
-    | Sub (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\sub"] @ aux p2
-    | Sup (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\sup"] @ aux p2
-    | Below (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\below"] @ aux p2
-    | Above (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\above"] @ aux p2
-    | Frac (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\frac"] @ aux p2
-    | Atop (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\atop"] @ aux p2
-    | Over (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\over"] @ aux p2
-    | Root (p1, p2) ->
+    | Ast.Sub (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\sub"] @ aux p2
+    | Ast.Sup (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\sup"] @ aux p2
+    | Ast.Below (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\below"] @ aux p2
+    | Ast.Above (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\above"] @ aux p2
+    | Ast.Frac (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\frac"] @ aux p2
+    | Ast.Atop (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\atop"] @ aux p2
+    | Ast.Over (p1, p2) -> aux p1 @ [NoBinding, gram_symbol "\\over"] @ aux p2
+    | Ast.Root (p1, p2) ->
         [NoBinding, gram_symbol "\\root"] @ aux p2
         @ [NoBinding, gram_symbol "\\of"] @ aux p1
-    | Sqrt p -> [NoBinding, gram_symbol "\\sqrt"] @ aux p
-    | Break -> []
-    | Box (_, pl) -> List.flatten (List.map aux pl)
-    | Group pl -> List.flatten (List.map aux pl)
+    | Ast.Sqrt p -> [NoBinding, gram_symbol "\\sqrt"] @ aux p
+    | Ast.Break -> []
+    | Ast.Box (_, pl) -> List.flatten (List.map aux pl)
+    | Ast.Group pl -> List.flatten (List.map aux pl)
   and aux_magic magic =
     match magic with
-    | Opt p ->
+    | Ast.Opt p ->
         let p_bindings, p_atoms, p_names, p_action = inner_pattern p in
-        let action (env_opt : CicNotationEnv.t option) (loc : location) =
+        let action (env_opt : CicNotationEnv.t option) (loc : Ast.location) =
           match env_opt with
-          | Some env -> List.map opt_binding_some env
-          | None -> List.map opt_binding_of_name p_names
+          | Some env -> List.map Env.opt_binding_some env
+          | None -> List.map Env.opt_binding_of_name p_names
         in
-        [ Env (List.map opt_declaration p_names),
+        [ Env (List.map Env.opt_declaration p_names),
           Gramext.srules
             [ [ Gramext.Sopt (Gramext.srules [ p_atoms, p_action ]) ],
               Gramext.action action ] ]
-    | List0 (p, _)
-    | List1 (p, _) ->
+    | Ast.List0 (p, _)
+    | Ast.List1 (p, _) ->
         let p_bindings, p_atoms, p_names, p_action = inner_pattern p in
 (*         let env0 = List.map list_binding_of_name p_names in
         let grow_env_entry env n v =
@@ -176,34 +178,34 @@ let extract_term_production pattern =
             (fun env (n, (_, v)) -> grow_env_entry env n v)
             env env_i
         in *)
-        let action (env_list : CicNotationEnv.t list) (loc : location) =
+        let action (env_list : CicNotationEnv.t list) (loc : Ast.location) =
           CicNotationEnv.coalesce_env p_names env_list
         in
         let gram_of_list s =
           match magic with
-          | List0 (_, None) -> Gramext.Slist0 s
-          | List1 (_, None) -> Gramext.Slist1 s
-          | List0 (_, Some l) -> Gramext.Slist0sep (s, gram_of_literal l)
-          | List1 (_, Some l) -> Gramext.Slist1sep (s, gram_of_literal l)
+          | Ast.List0 (_, None) -> Gramext.Slist0 s
+          | Ast.List1 (_, None) -> Gramext.Slist1 s
+          | Ast.List0 (_, Some l) -> Gramext.Slist0sep (s, gram_of_literal l)
+          | Ast.List1 (_, Some l) -> Gramext.Slist1sep (s, gram_of_literal l)
           | _ -> assert false
         in
-        [ Env (List.map list_declaration p_names),
+        [ Env (List.map Env.list_declaration p_names),
           Gramext.srules
             [ [ gram_of_list (Gramext.srules [ p_atoms, p_action ]) ],
               Gramext.action action ] ]
     | _ -> assert false
   and aux_variable =
     function
-    | NumVar s -> [Binding (s, NumType), gram_number ""]
-    | TermVar s -> [Binding (s, TermType), gram_term]
-    | IdentVar s -> [Binding (s, StringType), gram_ident ""]
-    | Ascription (p, s) -> assert false (* TODO *)
-    | FreshVar _ -> assert false
+    | Ast.NumVar s -> [Binding (s, Env.NumType), gram_number ""]
+    | Ast.TermVar s -> [Binding (s, Env.TermType), gram_term]
+    | Ast.IdentVar s -> [Binding (s, Env.StringType), gram_ident ""]
+    | Ast.Ascription (p, s) -> assert false (* TODO *)
+    | Ast.FreshVar _ -> assert false
   and inner_pattern p =
     let p_bindings, p_atoms = List.split (aux p) in
     let p_names = flatten_opt p_bindings in
     let action =
-      make_action (fun (env : CicNotationEnv.t) (loc : location) -> env)
+      make_action (fun (env : CicNotationEnv.t) (loc : Ast.location) -> env)
         p_bindings
     in
     p_bindings, p_atoms, p_names, action
@@ -240,7 +242,8 @@ let extend level1_pattern ~precedence ~associativity action =
           Some associativity,
           [ p_atoms, 
             (make_action
-              (fun (env: CicNotationEnv.t) (loc: location) -> (action env loc))
+              (fun (env: CicNotationEnv.t) (loc: Ast.location) ->
+                (action env loc))
               p_bindings) ]]]
   in
   let keywords = CicNotationUtil.keywords_of_term level1_pattern in
@@ -265,7 +268,7 @@ let parse_level2_meta_ref = ref (fun _ -> assert false)
 
 let fold_cluster binder terms ty body =
   List.fold_right
-    (fun term body -> Binder (binder, (term, ty), body))
+    (fun term body -> Ast.Binder (binder, (term, ty), body))
     terms body  (* terms are names: either Ident or FreshVar *)
 
 let fold_binder binder pt_names body =
@@ -273,7 +276,7 @@ let fold_binder binder pt_names body =
     (fun (names, ty) body -> fold_cluster binder names ty body)
     pt_names body
 
-let return_term loc term = AttributedTerm (`Loc loc, term)
+let return_term loc term = Ast.AttributedTerm (`Loc loc, term)
 
 let _ = (* create empty precedence level for "term" *)
   let mk_level_list first last =
@@ -310,56 +313,56 @@ EXTEND
 (*   row_sep:   [ [ "rowsep";   sep = literal -> sep ] ];
   field_sep: [ [ "fieldsep"; sep = literal -> sep ] ]; *)
   l1_magic_pattern: [
-    [ "list0"; p = l1_simple_pattern; sep = OPT sep -> List0 (p, sep)
-    | "list1"; p = l1_simple_pattern; sep = OPT sep -> List1 (p, sep)
-    | "opt";   p = l1_simple_pattern -> Opt p
+    [ "list0"; p = l1_simple_pattern; sep = OPT sep -> Ast.List0 (p, sep)
+    | "list1"; p = l1_simple_pattern; sep = OPT sep -> Ast.List1 (p, sep)
+    | "opt";   p = l1_simple_pattern -> Ast.Opt p
     ]
   ];
   l1_pattern_variable: [
-    [ "term"; id = IDENT -> TermVar id
-    | "number"; id = IDENT -> NumVar id
-    | "ident"; id = IDENT -> IdentVar id
+    [ "term"; id = IDENT -> Ast.TermVar id
+    | "number"; id = IDENT -> Ast.NumVar id
+    | "ident"; id = IDENT -> Ast.IdentVar id
     ]
   ];
   l1_simple_pattern:
     [ "layout" LEFTA
       [ p1 = SELF; SYMBOL "\\sub"; p2 = SELF ->
-          return_term loc (Layout (Sub (p1, p2)))
+          return_term loc (Ast.Layout (Ast.Sub (p1, p2)))
       | p1 = SELF; SYMBOL "\\sup"; p2 = SELF ->
-          return_term loc (Layout (Sup (p1, p2)))
+          return_term loc (Ast.Layout (Ast.Sup (p1, p2)))
       | p1 = SELF; SYMBOL "\\below"; p2 = SELF ->
-          return_term loc (Layout (Below (p1, p2)))
+          return_term loc (Ast.Layout (Ast.Below (p1, p2)))
       | p1 = SELF; SYMBOL "\\above"; p2 = SELF ->
-          return_term loc (Layout (Above (p1, p2)))
+          return_term loc (Ast.Layout (Ast.Above (p1, p2)))
       | p1 = SELF; SYMBOL "\\over"; p2 = SELF ->
-          return_term loc (Layout (Over (p1, p2)))
+          return_term loc (Ast.Layout (Ast.Over (p1, p2)))
       | p1 = SELF; SYMBOL "\\atop"; p2 = SELF ->
-          return_term loc (Layout (Atop (p1, p2)))
+          return_term loc (Ast.Layout (Ast.Atop (p1, p2)))
 (*       | "array"; p = SELF; csep = OPT field_sep; rsep = OPT row_sep ->
           return_term loc (Array (p, csep, rsep)) *)
       | SYMBOL "\\frac"; p1 = SELF; p2 = SELF ->
-          return_term loc (Layout (Frac (p1, p2)))
-      | SYMBOL "\\sqrt"; p = SELF -> return_term loc (Layout (Sqrt p))
+          return_term loc (Ast.Layout (Ast.Frac (p1, p2)))
+      | SYMBOL "\\sqrt"; p = SELF -> return_term loc (Ast.Layout (Ast.Sqrt p))
       | SYMBOL "\\root"; index = SELF; SYMBOL "\\of"; arg = SELF ->
-          return_term loc (Layout (Root (arg, index)))
+          return_term loc (Ast.Layout (Ast.Root (arg, index)))
       | "hbox"; LPAREN; p = l1_pattern; RPAREN ->
-          return_term loc (Layout (Box ((H, false, false), p)))
+          return_term loc (Ast.Layout (Ast.Box ((Ast.H, false, false), p)))
       | "vbox"; LPAREN; p = l1_pattern; RPAREN ->
-          return_term loc (Layout (Box ((V, false, false), p)))
+          return_term loc (Ast.Layout (Ast.Box ((Ast.V, false, false), p)))
       | "hvbox"; LPAREN; p = l1_pattern; RPAREN ->
-          return_term loc (Layout (Box ((HV, false, false), p)))
+          return_term loc (Ast.Layout (Ast.Box ((Ast.HV, false, false), p)))
       | "hovbox"; LPAREN; p = l1_pattern; RPAREN ->
-          return_term loc (Layout (Box ((HOV, false, false), p)))
-      | "break" -> return_term loc (Layout Break)
+          return_term loc (Ast.Layout (Ast.Box ((Ast.HOV, false, false), p)))
+      | "break" -> return_term loc (Ast.Layout Ast.Break)
 (*       | SYMBOL "\\SPACE" -> return_term loc (Layout Space) *)
       | LPAREN; p = l1_pattern; RPAREN ->
           return_term loc (CicNotationUtil.group p)
       ]
     | "simple" NONA
-      [ i = IDENT -> return_term loc (Variable (TermVar i))
-      | m = l1_magic_pattern -> return_term loc (Magic m)
-      | v = l1_pattern_variable -> return_term loc (Variable v)
-      | l = literal -> return_term loc (Literal l)
+      [ i = IDENT -> return_term loc (Ast.Variable (Ast.TermVar i))
+      | m = l1_magic_pattern -> return_term loc (Ast.Magic m)
+      | v = l1_pattern_variable -> return_term loc (Ast.Variable v)
+      | l = literal -> return_term loc (Ast.Literal l)
       ]
     ];
   END
@@ -369,29 +372,30 @@ EXTEND
 EXTEND
   GLOBAL: level2_meta;
   l2_variable: [
-    [ "term"; id = IDENT -> TermVar id
-    | "number"; id = IDENT -> NumVar id
-    | "ident"; id = IDENT -> IdentVar id
-    | "fresh"; id = IDENT -> FreshVar id
-    | "anonymous" -> TermVar "_"
-    | id = IDENT -> TermVar id
+    [ "term"; id = IDENT -> Ast.TermVar id
+    | "number"; id = IDENT -> Ast.NumVar id
+    | "ident"; id = IDENT -> Ast.IdentVar id
+    | "fresh"; id = IDENT -> Ast.FreshVar id
+    | "anonymous" -> Ast.TermVar "_"
+    | id = IDENT -> Ast.TermVar id
     ]
   ];
   l2_magic: [
     [ "fold"; kind = [ "left" -> `Left | "right" -> `Right ];
       base = level2_meta; "rec"; id = IDENT; recursive = level2_meta ->
-        Fold (kind, base, [id], recursive)
-    | "default"; some = level2_meta; none = level2_meta -> Default (some, none)
+        Ast.Fold (kind, base, [id], recursive)
+    | "default"; some = level2_meta; none = level2_meta ->
+        Ast.Default (some, none)
     | "if"; p_test = level2_meta;
       "then"; p_true = level2_meta;
       "else"; p_false = level2_meta ->
-        If (p_test, p_true, p_false)
-    | "fail" -> Fail
+        Ast.If (p_test, p_true, p_false)
+    | "fail" -> Ast.Fail
     ]
   ];
   level2_meta: [
-    [ magic = l2_magic -> Magic magic
-    | var = l2_variable -> Variable var
+    [ magic = l2_magic -> Ast.Magic magic
+    | var = l2_variable -> Ast.Variable var
     | blob = UNPARSED_AST -> !parse_level2_ast_ref (Stream.of_string blob)
     ]
   ];
@@ -448,23 +452,23 @@ EXTEND
   arg: [
     [ LPAREN; names = LIST1 IDENT SEP SYMBOL ",";
       SYMBOL ":"; ty = term; RPAREN ->
-        List.map (fun n -> Ident (n, None)) names, Some ty
-    | name = IDENT -> [Ident (name, None)], None
+        List.map (fun n -> Ast.Ident (n, None)) names, Some ty
+    | name = IDENT -> [Ast.Ident (name, None)], None
     | blob = UNPARSED_META ->
         let meta = !parse_level2_meta_ref (Stream.of_string blob) in
         match meta with
-        | Variable (FreshVar _) -> [meta], None
-        | Variable (TermVar "_") -> [Ident ("_", None)], None
+        | Ast.Variable (Ast.FreshVar _) -> [meta], None
+        | Ast.Variable (Ast.TermVar "_") -> [Ast.Ident ("_", None)], None
         | _ -> failwith "Invalid bound name."
    ]
   ];
   single_arg: [
-    [ name = IDENT -> Ident (name, None)
+    [ name = IDENT -> Ast.Ident (name, None)
     | blob = UNPARSED_META ->
         let meta = !parse_level2_meta_ref (Stream.of_string blob) in
         match meta with
-        | Variable (FreshVar _) -> meta
-        | Variable (TermVar "_") -> Ident ("_", None)
+        | Ast.Variable (Ast.FreshVar _) -> meta
+        | Ast.Variable (Ast.TermVar "_") -> Ast.Ident ("_", None)
         | _ -> failwith "Invalid index name."
     ]
   ];
@@ -493,7 +497,7 @@ EXTEND
           in
           let rec find_arg name n = function 
             | [] ->
-                fail loc (sprintf "Argument %s not found"
+                Ast.fail loc (sprintf "Argument %s not found"
                   (CicNotationPp.pp_term name))
             | (l,_) :: tl -> 
                 (match position_of name 0 l with
@@ -513,12 +517,12 @@ EXTEND
   binder_vars: [
     [ vars = [
           l = LIST1 single_arg SEP SYMBOL "," -> l
-        | SYMBOL "_" -> [Ident ("_", None)] ];
+        | SYMBOL "_" -> [Ast.Ident ("_", None)] ];
       typ = OPT [ SYMBOL ":"; t = term -> t ] -> (vars, typ)
     | LPAREN; 
         vars = [
             l =  LIST1 single_arg SEP SYMBOL "," -> l
-          | SYMBOL "_" -> [Ident ("_", None)] ];
+          | SYMBOL "_" -> [Ast.Ident ("_", None)] ];
       typ = OPT [ SYMBOL ":"; t = term -> t ]; 
       RPAREN -> (vars, typ)
     ]
@@ -526,10 +530,10 @@ EXTEND
   term: LEVEL "10N" [ (* let in *)
     [ "let"; var = possibly_typed_name; SYMBOL <:unicode<def>> (* ≝ *);
       p1 = term; "in"; p2 = term ->
-        return_term loc (LetIn (var, p1, p2))
+        return_term loc (Ast.LetIn (var, p1, p2))
     | "let"; k = induction_kind; defs = let_defs; "in";
       body = term ->
-        return_term loc (LetRec (k, defs, body))
+        return_term loc (Ast.LetRec (k, defs, body))
     ]
   ];
   term: LEVEL "20R"  (* binder *)
@@ -542,26 +546,28 @@ EXTEND
     [
       [ p1 = term; p2 = term ->
           let rec aux = function
-            | Appl (hd :: tl)
-            | AttributedTerm (_, Appl (hd :: tl)) ->
+            | Ast.Appl (hd :: tl)
+            | Ast.AttributedTerm (_, Ast.Appl (hd :: tl)) ->
                 aux hd @ tl
             | term -> [term]
           in
-          return_term loc (Appl (aux p1 @ [p2]))
+          return_term loc (Ast.Appl (aux p1 @ [p2]))
       ]
     ];
   term: LEVEL "90N"  (* simple *)
     [
-      [ id = IDENT -> return_term loc (Ident (id, None))
-      | id = IDENT; s = explicit_subst -> return_term loc (Ident (id, Some s))
-      | s = CSYMBOL -> return_term loc (Symbol (s, 0))
-      | u = URI -> return_term loc (Uri (u, None))
-      | n = NUMBER -> return_term loc (Num (n, 0))
-      | IMPLICIT -> return_term loc (Implicit)
-      | PLACEHOLDER -> return_term loc UserInput
-      | m = META -> return_term loc (Meta (int_of_string m, []))
-      | m = META; s = meta_substs -> return_term loc (Meta (int_of_string m, s))
-      | s = sort -> return_term loc (Sort s)
+      [ id = IDENT -> return_term loc (Ast.Ident (id, None))
+      | id = IDENT; s = explicit_subst ->
+          return_term loc (Ast.Ident (id, Some s))
+      | s = CSYMBOL -> return_term loc (Ast.Symbol (s, 0))
+      | u = URI -> return_term loc (Ast.Uri (u, None))
+      | n = NUMBER -> return_term loc (Ast.Num (n, 0))
+      | IMPLICIT -> return_term loc (Ast.Implicit)
+      | PLACEHOLDER -> return_term loc Ast.UserInput
+      | m = META -> return_term loc (Ast.Meta (int_of_string m, []))
+      | m = META; s = meta_substs ->
+          return_term loc (Ast.Meta (int_of_string m, s))
+      | s = sort -> return_term loc (Ast.Sort s)
       | outtyp = OPT [ SYMBOL "["; ty = term; SYMBOL "]" -> ty ];
         "match"; t = term;
         indty_ident = OPT [ "in"; id = IDENT -> id ];
@@ -572,9 +578,9 @@ EXTEND
             lhs, rhs
         ] SEP SYMBOL "|";
         SYMBOL "]" ->
-          return_term loc (Case (t, indty_ident, outtyp, patterns))
+          return_term loc (Ast.Case (t, indty_ident, outtyp, patterns))
       | LPAREN; p1 = term; SYMBOL ":"; p2 = term; RPAREN ->
-          return_term loc (Cast (p1, p2))
+          return_term loc (Ast.Cast (p1, p2))
       | LPAREN; p = term; RPAREN -> p
       | blob = UNPARSED_META -> !parse_level2_meta_ref (Stream.of_string blob)
       ]