]> matita.cs.unibo.it Git - helm.git/blob - components/grafite_parser/grafiteParser.ml
applyS now receives the same parameters that auto receives.
[helm.git] / components / grafite_parser / grafiteParser.ml
1 (* Copyright (C) 2005, HELM Team.
2  * 
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.
6  * 
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.
11  * 
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.
16  *
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,
20  * MA  02111-1307, USA.
21  * 
22  * For details, see the HELM World-Wide-Web page,
23  * http://helm.cs.unibo.it/
24  *)
25
26 (* $Id$ *)
27
28 open Printf
29
30 module Ast = CicNotationPt
31
32 type 'a localized_option =
33    LSome of 'a
34  | LNone of GrafiteAst.loc
35
36 type ast_statement =
37   (CicNotationPt.term, CicNotationPt.term,
38    CicNotationPt.term GrafiteAst.reduction, CicNotationPt.obj, string)
39     GrafiteAst.statement
40
41 type statement =
42   include_paths:string list ->
43   LexiconEngine.status ->
44     LexiconEngine.status * ast_statement localized_option
45
46 let grammar = CicNotationParser.level2_ast_grammar
47
48 let term = CicNotationParser.term
49 let statement = Grammar.Entry.create grammar "statement"
50
51 let add_raw_attribute ~text t = Ast.AttributedTerm (`Raw text, t)
52
53 let default_precedence = 50
54 let default_associativity = Gramext.NonA
55
56 type by_continuation =
57    BYC_done
58  | BYC_weproved of CicNotationPt.term * string option * CicNotationPt.term option
59  | BYC_letsuchthat of string * CicNotationPt.term * string * CicNotationPt.term
60  | BYC_wehaveand of string * CicNotationPt.term * string * CicNotationPt.term
61
62 EXTEND
63   GLOBAL: term statement;
64   arg: [
65    [ LPAREN; names = LIST1 IDENT SEP SYMBOL ",";
66       SYMBOL ":"; ty = term; RPAREN -> names,ty
67    | name = IDENT -> [name],Ast.Implicit
68    ]
69   ];
70   constructor: [ [ name = IDENT; SYMBOL ":"; typ = term -> (name, typ) ] ];
71   tactic_term: [ [ t = term LEVEL "90N" -> t ] ];
72   ident_list0: [ [ LPAREN; idents = LIST0 IDENT; RPAREN -> idents ] ];
73   tactic_term_list1: [
74     [ tactic_terms = LIST1 tactic_term SEP SYMBOL "," -> tactic_terms ]
75   ];
76   reduction_kind: [
77     [ IDENT "normalize" -> `Normalize
78     | IDENT "reduce" -> `Reduce
79     | IDENT "simplify" -> `Simpl
80     | IDENT "unfold"; t = OPT tactic_term -> `Unfold t
81     | IDENT "whd" -> `Whd ]
82   ];
83   sequent_pattern_spec: [
84    [ hyp_paths =
85       LIST0
86        [ id = IDENT ;
87          path = OPT [SYMBOL ":" ; path = tactic_term -> path ] ->
88          (id,match path with Some p -> p | None -> Ast.UserInput) ];
89      goal_path = OPT [ SYMBOL <:unicode<vdash>>; term = tactic_term -> term ] ->
90       let goal_path =
91        match goal_path, hyp_paths with
92           None, [] -> Some Ast.UserInput
93         | None, _::_ -> None
94         | Some goal_path, _ -> Some goal_path
95       in
96        hyp_paths,goal_path
97    ]
98   ];
99   pattern_spec: [
100     [ res = OPT [
101        "in";
102        wanted_and_sps =
103         [ "match" ; wanted = tactic_term ;
104           sps = OPT [ "in"; sps = sequent_pattern_spec -> sps ] ->
105            Some wanted,sps
106         | sps = sequent_pattern_spec ->
107            None,Some sps
108         ] ->
109          let wanted,hyp_paths,goal_path =
110           match wanted_and_sps with
111              wanted,None -> wanted, [], Some Ast.UserInput
112            | wanted,Some (hyp_paths,goal_path) -> wanted,hyp_paths,goal_path
113          in
114           wanted, hyp_paths, goal_path ] ->
115       match res with
116          None -> None,[],Some Ast.UserInput
117        | Some ps -> ps]
118   ];
119   direction: [
120     [ SYMBOL ">" -> `LeftToRight
121     | SYMBOL "<" -> `RightToLeft ]
122   ];
123   int: [ [ num = NUMBER -> int_of_string num ] ];
124   intros_spec: [
125     [ num = OPT [ num = int -> num ]; idents = OPT ident_list0 ->
126         let idents = match idents with None -> [] | Some idents -> idents in
127         num, idents
128     ]
129   ];
130   using: [ [ using = OPT [ IDENT "using"; t = tactic_term -> t ] -> using ] ];
131   tactic: [
132     [ IDENT "absurd"; t = tactic_term ->
133         GrafiteAst.Absurd (loc, t)
134     | IDENT "apply"; t = tactic_term ->
135         GrafiteAst.Apply (loc, t)
136     | IDENT "applyS"; t = tactic_term ; params = 
137         LIST0 [ i = IDENT -> i,"" | i = IDENT ; SYMBOL "="; v = [ v = int ->
138           string_of_int v | v = IDENT -> v ] -> i,v ] ->
139         GrafiteAst.ApplyS (loc, t, params)
140     | IDENT "assumption" ->
141         GrafiteAst.Assumption loc
142     | IDENT "auto"; params = 
143         LIST0 [ i = IDENT -> i,"" | i = IDENT ; SYMBOL "="; v = [ v = int ->
144           string_of_int v | v = IDENT -> v ] -> i,v ] ->
145         GrafiteAst.Auto (loc,params)
146     | IDENT "clear"; ids = LIST1 IDENT ->
147         GrafiteAst.Clear (loc, ids)
148     | IDENT "clearbody"; id = IDENT ->
149         GrafiteAst.ClearBody (loc,id)
150     | IDENT "change"; what = pattern_spec; "with"; t = tactic_term ->
151         GrafiteAst.Change (loc, what, t)
152     | IDENT "constructor"; n = int ->
153         GrafiteAst.Constructor (loc, n)
154     | IDENT "contradiction" ->
155         GrafiteAst.Contradiction loc
156     | IDENT "cut"; t = tactic_term; ident = OPT [ "as"; id = IDENT -> id] ->
157         GrafiteAst.Cut (loc, ident, t)
158     | IDENT "decompose"; types = OPT ident_list0; what = OPT IDENT;
159         idents = OPT [ "as"; idents = LIST1 IDENT -> idents ] ->
160         let types = match types with None -> [] | Some types -> types in
161         let idents = match idents with None -> [] | Some idents -> idents in
162         let to_spec id = GrafiteAst.Ident id in
163         GrafiteAst.Decompose (loc, List.rev_map to_spec types, what, idents)
164     | IDENT "demodulate" -> GrafiteAst.Demodulate loc
165     | IDENT "destruct"; t = tactic_term ->
166         GrafiteAst.Destruct (loc, t)
167     | IDENT "elim"; what = tactic_term; using = using;
168       (num, idents) = intros_spec ->
169         GrafiteAst.Elim (loc, what, using, num, idents)
170     | IDENT "elimType"; what = tactic_term; using = using;
171       (num, idents) = intros_spec ->
172         GrafiteAst.ElimType (loc, what, using, num, idents)
173     | IDENT "exact"; t = tactic_term ->
174         GrafiteAst.Exact (loc, t)
175     | IDENT "exists" ->
176         GrafiteAst.Exists loc
177     | IDENT "fail" -> GrafiteAst.Fail loc
178     | IDENT "fold"; kind = reduction_kind; t = tactic_term; p = pattern_spec ->
179         let (pt,_,_) = p in
180           if pt <> None then
181             raise (HExtlib.Localized (loc, CicNotationParser.Parse_error
182               ("the pattern cannot specify the term to replace, only its"
183               ^ " paths in the hypotheses and in the conclusion")))
184         else
185          GrafiteAst.Fold (loc, kind, t, p)
186     | IDENT "fourier" ->
187         GrafiteAst.Fourier loc
188     | IDENT "fwd"; hyp = IDENT; idents = OPT [ "as"; idents = LIST1 IDENT -> idents ] ->
189         let idents = match idents with None -> [] | Some idents -> idents in
190         GrafiteAst.FwdSimpl (loc, hyp, idents)
191     | IDENT "generalize"; p=pattern_spec; id = OPT ["as" ; id = IDENT -> id] ->
192        GrafiteAst.Generalize (loc,p,id)
193     | IDENT "goal"; n = int ->
194         GrafiteAst.Goal (loc, n)
195     | IDENT "id" -> GrafiteAst.IdTac loc
196     | IDENT "intro"; ident = OPT IDENT ->
197         let idents = match ident with None -> [] | Some id -> [id] in
198         GrafiteAst.Intros (loc, Some 1, idents)
199     | IDENT "intros"; (num, idents) = intros_spec ->
200         GrafiteAst.Intros (loc, num, idents)
201     | IDENT "inversion"; t = tactic_term ->
202         GrafiteAst.Inversion (loc, t)
203     | IDENT "lapply"; 
204       linear = OPT [ IDENT "linear" ];
205       depth = OPT [ IDENT "depth"; SYMBOL "="; i = int -> i ];
206       what = tactic_term; 
207       to_what = OPT [ "to" ; t = tactic_term_list1 -> t ];
208       ident = OPT [ "as" ; ident = IDENT -> ident ] ->
209         let linear = match linear with None -> false | Some _ -> true in 
210         let to_what = match to_what with None -> [] | Some to_what -> to_what in
211         GrafiteAst.LApply (loc, linear, depth, to_what, what, ident)
212     | IDENT "left" -> GrafiteAst.Left loc
213     | IDENT "letin"; where = IDENT ; SYMBOL <:unicode<def>> ; t = tactic_term ->
214         GrafiteAst.LetIn (loc, t, where)
215     | kind = reduction_kind; p = pattern_spec ->
216         GrafiteAst.Reduce (loc, kind, p)
217     | IDENT "reflexivity" ->
218         GrafiteAst.Reflexivity loc
219     | IDENT "replace"; p = pattern_spec; "with"; t = tactic_term ->
220         GrafiteAst.Replace (loc, p, t)
221     | IDENT "rewrite" ; d = direction; t = tactic_term ; p = pattern_spec ->
222        let (pt,_,_) = p in
223         if pt <> None then
224          raise
225           (HExtlib.Localized (loc,
226            (CicNotationParser.Parse_error
227             "the pattern cannot specify the term to rewrite, only its paths in the hypotheses and in the conclusion")))
228         else
229          GrafiteAst.Rewrite (loc, d, t, p)
230     | IDENT "right" ->
231         GrafiteAst.Right loc
232     | IDENT "ring" ->
233         GrafiteAst.Ring loc
234     | IDENT "split" ->
235         GrafiteAst.Split loc
236     | IDENT "subst" ->
237         GrafiteAst.Subst loc    
238     | IDENT "symmetry" ->
239         GrafiteAst.Symmetry loc
240     | IDENT "transitivity"; t = tactic_term ->
241         GrafiteAst.Transitivity (loc, t)
242      (* Produzioni Aggiunte *)
243     | IDENT "assume" ; id = IDENT ; SYMBOL ":" ; t = tactic_term ->
244         GrafiteAst.Assume (loc, id, t)
245     | IDENT "suppose" ; t = tactic_term ; LPAREN ; id = IDENT ; RPAREN ; t1 = OPT [IDENT "that" ; IDENT "is" ; IDENT "equivalent" ; "to" ; t' = tactic_term -> t']->
246         GrafiteAst.Suppose (loc, t, id, t1)
247     | IDENT "by" ; t = [t' = tactic_term -> LSome t' | SYMBOL "_" -> LNone loc];
248       cont=by_continuation ->
249        let t' = match t with LNone _ -> None | LSome t -> Some t in
250        (match cont with
251            BYC_done -> GrafiteAst.Bydone (loc, t')
252          | BYC_weproved (ty,id,t1) ->
253             GrafiteAst.By_term_we_proved(loc, t', ty, id, t1)
254          | BYC_letsuchthat (id1,t1,id2,t2) ->
255             (match t with
256                 LNone floc ->
257                  raise (HExtlib.Localized
258                   (floc,CicNotationParser.Parse_error
259                     "tactic_term expected here"))
260               | LSome t -> GrafiteAst.ExistsElim (loc, t, id1, t1, id2, t2))
261          | BYC_wehaveand (id1,t1,id2,t2) ->
262             (match t with
263                 LNone floc ->
264                  raise (HExtlib.Localized
265                   (floc,CicNotationParser.Parse_error
266                     "tactic_term expected here"))
267               | LSome t -> GrafiteAst.AndElim (loc, t, id1, t1, id2, t2)))
268     | IDENT "we" ; IDENT "need" ; "to" ; IDENT "prove" ; t = tactic_term ; id = OPT [ LPAREN ; id = IDENT ; RPAREN -> id ] ; t1 = OPT [IDENT "or" ; IDENT "equivalently"; t' = tactic_term -> t']->
269         GrafiteAst.We_need_to_prove (loc, t, id, t1)
270     | IDENT "we" ; IDENT "proceed" ; IDENT "by" ; IDENT "induction" ; "on" ; t=tactic_term ; "to" ; IDENT "prove" ; t1=tactic_term ->  
271         GrafiteAst.We_proceed_by_induction_on (loc, t, t1)
272     | IDENT "by" ; IDENT "induction" ; IDENT "hypothesis" ; IDENT "we" ; IDENT "know" ; t=tactic_term ; LPAREN ; id = IDENT ; RPAREN ->
273         GrafiteAst.Byinduction(loc, t, id)
274     | IDENT "the" ; IDENT "thesis" ; IDENT "becomes" ; t=tactic_term ->
275         GrafiteAst.Thesisbecomes(loc, t)
276     | IDENT "case" ; id = IDENT ; params=LIST0[LPAREN ; i=IDENT ;
277         SYMBOL":" ; t=tactic_term ; RPAREN -> i,t] ->
278          GrafiteAst.Case(loc,id,params)
279     | IDENT "obtain" ; termine=tactic_term ; SYMBOL "=" ; t1=tactic_term ; IDENT "by" ; t2=[ t=tactic_term -> Some t | SYMBOL "_" -> None  ] ; cont=rewriting_step_continuation ->
280      GrafiteAst.RewritingStep(loc, Some termine, t1, t2, cont)
281     | SYMBOL "=" ; t1=tactic_term ; IDENT "by" ; t2=[ t=tactic_term -> Some t | SYMBOL "_" -> None  ] ; cont=rewriting_step_continuation  ->
282      GrafiteAst.RewritingStep(loc, None, t1, t2, cont)
283   ]
284 ];
285   by_continuation: [
286     [ IDENT "we" ; IDENT "proved" ; ty = tactic_term ; LPAREN ; id = IDENT ; RPAREN ; t1 = OPT [IDENT "that" ; IDENT "is" ; IDENT "equivalent" ; "to" ; t2 = tactic_term -> t2] -> BYC_weproved (ty,Some id,t1)
287     | IDENT "we" ; IDENT "proved" ; ty = tactic_term ; t1 = OPT [IDENT "that" ; IDENT "is" ; IDENT "equivalent" ; "to" ; t2 = tactic_term -> t2] ; 
288             IDENT "done" -> BYC_weproved (ty,None,t1)
289     | IDENT "done" -> BYC_done
290     | "let" ; id1 = IDENT ; SYMBOL ":" ; t1 = tactic_term ;
291       IDENT "such" ; IDENT "that" ; t2=tactic_term ; LPAREN ; id2 = IDENT ; RPAREN -> BYC_letsuchthat (id1,t1,id2,t2)
292     | IDENT "we" ; IDENT "have" ; t1=tactic_term ; LPAREN ; id1=IDENT ; RPAREN ;"and" ; t2=tactic_term ; LPAREN ; id2=IDENT ; RPAREN ->
293               BYC_wehaveand (id1,t1,id2,t2)
294     ]
295 ];
296   rewriting_step_continuation : [
297     [ IDENT "done" -> None
298     | IDENT "we" ; IDENT "proved" ; id=IDENT -> Some (Cic.Name id)
299     | -> Some Cic.Anonymous
300     ]
301 ];
302   atomic_tactical:
303     [ "sequence" LEFTA
304       [ t1 = SELF; SYMBOL ";"; t2 = SELF ->
305           let ts =
306             match t1 with
307             | GrafiteAst.Seq (_, l) -> l @ [ t2 ]
308             | _ -> [ t1; t2 ]
309           in
310           GrafiteAst.Seq (loc, ts)
311       ]
312     | "then" NONA
313       [ tac = SELF; SYMBOL ";";
314         SYMBOL "["; tacs = LIST0 SELF SEP SYMBOL "|"; SYMBOL "]"->
315           (GrafiteAst.Then (loc, tac, tacs))
316       ]
317     | "loops" RIGHTA
318       [ IDENT "do"; count = int; tac = SELF; IDENT "end" ->
319           GrafiteAst.Do (loc, count, tac)
320       | IDENT "repeat"; tac = SELF; IDENT "end" -> GrafiteAst.Repeat (loc, tac)
321       ]
322     | "simple" NONA
323       [ IDENT "first";
324         SYMBOL "["; tacs = LIST0 SELF SEP SYMBOL "|"; SYMBOL "]"->
325           GrafiteAst.First (loc, tacs)
326       | IDENT "try"; tac = SELF -> GrafiteAst.Try (loc, tac)
327       | IDENT "solve";
328         SYMBOL "["; tacs = LIST0 SELF SEP SYMBOL "|"; SYMBOL "]"->
329           GrafiteAst.Solve (loc, tacs)
330       | IDENT "progress"; tac = SELF -> GrafiteAst.Progress (loc, tac)
331       | LPAREN; tac = SELF; RPAREN -> tac
332       | tac = tactic -> GrafiteAst.Tactic (loc, tac)
333       ]
334     ];
335   punctuation_tactical:
336     [
337       [ SYMBOL "[" -> GrafiteAst.Branch loc
338       | SYMBOL "|" -> GrafiteAst.Shift loc
339       | i = LIST1 int SEP SYMBOL ","; SYMBOL ":" -> GrafiteAst.Pos (loc, i)
340       | SYMBOL "*"; SYMBOL ":" -> GrafiteAst.Wildcard loc
341       | SYMBOL "]" -> GrafiteAst.Merge loc
342       | SYMBOL ";" -> GrafiteAst.Semicolon loc
343       | SYMBOL "." -> GrafiteAst.Dot loc
344       ]
345     ];
346   tactical:
347     [ "simple" NONA
348       [ IDENT "focus"; goals = LIST1 int -> GrafiteAst.Focus (loc, goals)
349       | IDENT "unfocus" -> GrafiteAst.Unfocus loc
350       | IDENT "skip" -> GrafiteAst.Skip loc
351       | tac = atomic_tactical LEVEL "loops" -> tac
352       ]
353     ];
354   theorem_flavour: [
355     [ [ IDENT "definition"  ] -> `Definition
356     | [ IDENT "fact"        ] -> `Fact
357     | [ IDENT "lemma"       ] -> `Lemma
358     | [ IDENT "remark"      ] -> `Remark
359     | [ IDENT "theorem"     ] -> `Theorem
360     ]
361   ];
362   inductive_spec: [ [
363     fst_name = IDENT; params = LIST0 [ arg=arg -> arg ];
364     SYMBOL ":"; fst_typ = term; SYMBOL <:unicode<def>>; OPT SYMBOL "|";
365     fst_constructors = LIST0 constructor SEP SYMBOL "|";
366     tl = OPT [ "with";
367       types = LIST1 [
368         name = IDENT; SYMBOL ":"; typ = term; SYMBOL <:unicode<def>>;
369        OPT SYMBOL "|"; constructors = LIST0 constructor SEP SYMBOL "|" ->
370           (name, true, typ, constructors) ] SEP "with" -> types
371     ] ->
372       let params =
373         List.fold_right
374           (fun (names, typ) acc ->
375             (List.map (fun name -> (name, typ)) names) @ acc)
376           params []
377       in
378       let fst_ind_type = (fst_name, true, fst_typ, fst_constructors) in
379       let tl_ind_types = match tl with None -> [] | Some types -> types in
380       let ind_types = fst_ind_type :: tl_ind_types in
381       (params, ind_types)
382   ] ];
383   
384   record_spec: [ [
385     name = IDENT; params = LIST0 [ arg = arg -> arg ] ;
386      SYMBOL ":"; typ = term; SYMBOL <:unicode<def>>; SYMBOL "{" ; 
387      fields = LIST0 [ 
388        name = IDENT ; 
389        coercion = [ 
390            SYMBOL ":" -> false,0 
391          | SYMBOL ":"; SYMBOL ">" -> true,0
392          | SYMBOL ":"; arity = int ; SYMBOL ">" -> true,arity
393        ]; 
394        ty = term -> 
395          let b,n = coercion in 
396          (name,ty,b,n) 
397      ] SEP SYMBOL ";"; SYMBOL "}" -> 
398       let params =
399         List.fold_right
400           (fun (names, typ) acc ->
401             (List.map (fun name -> (name, typ)) names) @ acc)
402           params []
403       in
404       (params,name,typ,fields)
405   ] ];
406   
407   macro: [
408     [ [ IDENT "check"   ]; t = term ->
409         GrafiteAst.Check (loc, t)
410     | [ IDENT "inline"]; suri = QSTRING ->
411          GrafiteAst.Inline (loc,suri)
412     | [ IDENT "hint" ] -> GrafiteAst.Hint loc
413     | [ IDENT "whelp"; "match" ] ; t = term -> 
414         GrafiteAst.WMatch (loc,t)
415     | [ IDENT "whelp"; IDENT "instance" ] ; t = term -> 
416         GrafiteAst.WInstance (loc,t)
417     | [ IDENT "whelp"; IDENT "locate" ] ; id = QSTRING -> 
418         GrafiteAst.WLocate (loc,id)
419     | [ IDENT "whelp"; IDENT "elim" ] ; t = term ->
420         GrafiteAst.WElim (loc, t)
421     | [ IDENT "whelp"; IDENT "hint" ] ; t = term -> 
422         GrafiteAst.WHint (loc,t)
423     ]
424   ];
425   alias_spec: [
426     [ IDENT "id"; id = QSTRING; SYMBOL "="; uri = QSTRING ->
427       let alpha = "[a-zA-Z]" in
428       let num = "[0-9]+" in
429       let ident_cont = "\\("^alpha^"\\|"^num^"\\|_\\|\\\\\\)" in
430       let ident = "\\("^alpha^ident_cont^"*\\|_"^ident_cont^"+\\)" in
431       let rex = Str.regexp ("^"^ident^"$") in
432       if Str.string_match rex id 0 then
433         if (try ignore (UriManager.uri_of_string uri); true
434             with UriManager.IllFormedUri _ -> false)
435         then
436           LexiconAst.Ident_alias (id, uri)
437         else 
438           raise
439            (HExtlib.Localized (loc, CicNotationParser.Parse_error (sprintf "Not a valid uri: %s" uri)))
440       else
441         raise (HExtlib.Localized (loc, CicNotationParser.Parse_error (
442           sprintf "Not a valid identifier: %s" id)))
443     | IDENT "symbol"; symbol = QSTRING;
444       instance = OPT [ LPAREN; IDENT "instance"; n = int; RPAREN -> n ];
445       SYMBOL "="; dsc = QSTRING ->
446         let instance =
447           match instance with Some i -> i | None -> 0
448         in
449         LexiconAst.Symbol_alias (symbol, instance, dsc)
450     | IDENT "num";
451       instance = OPT [ LPAREN; IDENT "instance"; n = int; RPAREN -> n ];
452       SYMBOL "="; dsc = QSTRING ->
453         let instance =
454           match instance with Some i -> i | None -> 0
455         in
456         LexiconAst.Number_alias (instance, dsc)
457     ]
458   ];
459   argument: [
460     [ l = LIST0 [ SYMBOL <:unicode<eta>> (* η *); SYMBOL "." -> () ];
461       id = IDENT ->
462         Ast.IdentArg (List.length l, id)
463     ]
464   ];
465   associativity: [
466     [ IDENT "left";  IDENT "associative" -> Gramext.LeftA
467     | IDENT "right"; IDENT "associative" -> Gramext.RightA
468     | IDENT "non"; IDENT "associative" -> Gramext.NonA
469     ]
470   ];
471   precedence: [
472     [ "with"; IDENT "precedence"; n = NUMBER -> int_of_string n ]
473   ];
474   notation: [
475     [ dir = OPT direction; s = QSTRING;
476       assoc = OPT associativity; prec = OPT precedence;
477       IDENT "for";
478       p2 = 
479         [ blob = UNPARSED_AST ->
480             add_raw_attribute ~text:(sprintf "@{%s}" blob)
481               (CicNotationParser.parse_level2_ast
482                 (Ulexing.from_utf8_string blob))
483         | blob = UNPARSED_META ->
484             add_raw_attribute ~text:(sprintf "${%s}" blob)
485               (CicNotationParser.parse_level2_meta
486                 (Ulexing.from_utf8_string blob))
487         ] ->
488           let assoc =
489             match assoc with
490             | None -> default_associativity
491             | Some assoc -> assoc
492           in
493           let prec =
494             match prec with
495             | None -> default_precedence
496             | Some prec -> prec
497           in
498           let p1 =
499             add_raw_attribute ~text:s
500               (CicNotationParser.parse_level1_pattern
501                 (Ulexing.from_utf8_string s))
502           in
503           (dir, p1, assoc, prec, p2)
504     ]
505   ];
506   level3_term: [
507     [ u = URI -> Ast.UriPattern (UriManager.uri_of_string u)
508     | id = IDENT -> Ast.VarPattern id
509     | SYMBOL "_" -> Ast.ImplicitPattern
510     | LPAREN; terms = LIST1 SELF; RPAREN ->
511         (match terms with
512         | [] -> assert false
513         | [term] -> term
514         | terms -> Ast.ApplPattern terms)
515     ]
516   ];
517   interpretation: [
518     [ s = CSYMBOL; args = LIST0 argument; SYMBOL "="; t = level3_term ->
519         (s, args, t)
520     ]
521   ];
522   
523   include_command: [ [
524       IDENT "include" ; path = QSTRING -> 
525         loc,path,LexiconAst.WithPreferences
526     | IDENT "include'" ; path = QSTRING -> 
527         loc,path,LexiconAst.WithoutPreferences
528    ]];
529
530   grafite_command: [ [
531       IDENT "set"; n = QSTRING; v = QSTRING ->
532         GrafiteAst.Set (loc, n, v)
533     | IDENT "drop" -> GrafiteAst.Drop loc
534     | IDENT "print"; s = IDENT -> GrafiteAst.Print (loc,s)
535     | IDENT "qed" -> GrafiteAst.Qed loc
536     | IDENT "variant" ; name = IDENT; SYMBOL ":"; 
537       typ = term; SYMBOL <:unicode<def>> ; newname = IDENT ->
538         GrafiteAst.Obj (loc, 
539           Ast.Theorem 
540             (`Variant,name,typ,Some (Ast.Ident (newname, None))))
541     | flavour = theorem_flavour; name = IDENT; SYMBOL ":"; typ = term;
542       body = OPT [ SYMBOL <:unicode<def>> (* ≝ *); body = term -> body ] ->
543         GrafiteAst.Obj (loc, Ast.Theorem (flavour, name, typ, body))
544     | flavour = theorem_flavour; name = IDENT; SYMBOL <:unicode<def>> (* ≝ *);
545       body = term ->
546         GrafiteAst.Obj (loc,
547           Ast.Theorem (flavour, name, Ast.Implicit, Some body))
548     | IDENT "axiom"; name = IDENT; SYMBOL ":"; typ = term ->
549         GrafiteAst.Obj (loc, Ast.Theorem (`Axiom, name, typ, None))
550     | "let"; ind_kind = [ "corec" -> `CoInductive | "rec"-> `Inductive ];
551         defs = CicNotationParser.let_defs -> 
552           let name,ty = 
553             match defs with
554             | ((Ast.Ident (name, None), Some ty),_,_) :: _ -> name,ty
555             | ((Ast.Ident (name, None), None),_,_) :: _ ->
556                 name, Ast.Implicit
557             | _ -> assert false 
558           in
559           let body = Ast.Ident (name,None) in
560           GrafiteAst.Obj (loc, Ast.Theorem(`Definition, name, ty,
561             Some (Ast.LetRec (ind_kind, defs, body))))
562     | IDENT "inductive"; spec = inductive_spec ->
563         let (params, ind_types) = spec in
564         GrafiteAst.Obj (loc, Ast.Inductive (params, ind_types))
565     | IDENT "coinductive"; spec = inductive_spec ->
566         let (params, ind_types) = spec in
567         let ind_types = (* set inductive flags to false (coinductive) *)
568           List.map (fun (name, _, term, ctors) -> (name, false, term, ctors))
569             ind_types
570         in
571         GrafiteAst.Obj (loc, Ast.Inductive (params, ind_types))
572     | IDENT "coercion" ; suri = URI ; arity = OPT int ->
573         let arity = match arity with None -> 0 | Some x -> x in
574         GrafiteAst.Coercion (loc, UriManager.uri_of_string suri, true, arity)
575     | IDENT "record" ; (params,name,ty,fields) = record_spec ->
576         GrafiteAst.Obj (loc, Ast.Record (params,name,ty,fields))
577     | IDENT "default" ; what = QSTRING ; uris = LIST1 URI ->
578        let uris = List.map UriManager.uri_of_string uris in
579         GrafiteAst.Default (loc,what,uris)
580     | IDENT "relation" ; aeq = tactic_term ; "on" ; a = tactic_term ;
581       refl = OPT [ IDENT "reflexivity" ; IDENT "proved" ; IDENT "by" ;
582                    refl = tactic_term -> refl ] ;
583       sym = OPT [ IDENT "symmetry" ; IDENT "proved" ; IDENT "by" ;
584                    sym = tactic_term -> sym ] ;
585       trans = OPT [ IDENT "transitivity" ; IDENT "proved" ; IDENT "by" ;
586                    trans = tactic_term -> trans ] ;
587       "as" ; id = IDENT ->
588        GrafiteAst.Relation (loc,id,a,aeq,refl,sym,trans)
589   ]];
590   lexicon_command: [ [
591       IDENT "alias" ; spec = alias_spec ->
592         LexiconAst.Alias (loc, spec)
593     | IDENT "notation"; (dir, l1, assoc, prec, l2) = notation ->
594         LexiconAst.Notation (loc, dir, l1, assoc, prec, l2)
595     | IDENT "interpretation"; id = QSTRING;
596       (symbol, args, l3) = interpretation ->
597         LexiconAst.Interpretation (loc, id, (symbol, args), l3)
598   ]];
599   executable: [
600     [ cmd = grafite_command; SYMBOL "." -> GrafiteAst.Command (loc, cmd)
601     | tac = tactical; punct = punctuation_tactical ->
602         GrafiteAst.Tactical (loc, tac, Some punct)
603     | punct = punctuation_tactical -> GrafiteAst.Tactical (loc, punct, None)
604     | mac = macro; SYMBOL "." -> GrafiteAst.Macro (loc, mac)
605     ]
606   ];
607   comment: [
608     [ BEGINCOMMENT ; ex = executable ; ENDCOMMENT -> 
609        GrafiteAst.Code (loc, ex)
610     | str = NOTE -> 
611        GrafiteAst.Note (loc, str)
612     ]
613   ];
614   statement: [
615     [ ex = executable ->
616        fun ~include_paths status -> status,LSome(GrafiteAst.Executable (loc,ex))
617     | com = comment ->
618        fun ~include_paths status -> status,LSome (GrafiteAst.Comment (loc, com))
619     | (iloc,fname,mode) = include_command ; SYMBOL "."  ->
620        fun ~include_paths status ->
621         let buri, fullpath = 
622           DependenciesParser.baseuri_of_script ~include_paths fname 
623         in
624         let status =
625          LexiconEngine.eval_command status 
626            (LexiconAst.Include (iloc,buri,mode,fullpath))
627         in
628          status,
629           LSome
630           (GrafiteAst.Executable
631            (loc,GrafiteAst.Command
632             (loc,GrafiteAst.Include (iloc,buri))))
633     | scom = lexicon_command ; SYMBOL "." ->
634        fun ~include_paths status ->
635         let status = LexiconEngine.eval_command status scom in
636          status,LNone loc
637     | EOI -> raise End_of_file
638     ]
639   ];
640 END
641
642 let exc_located_wrapper f =
643   try
644     f ()
645   with
646   | Stdpp.Exc_located (_, End_of_file) -> raise End_of_file
647   | Stdpp.Exc_located (floc, Stream.Error msg) ->
648       raise (HExtlib.Localized (floc,CicNotationParser.Parse_error msg))
649   | Stdpp.Exc_located (floc, exn) ->
650       raise
651        (HExtlib.Localized (floc,CicNotationParser.Parse_error (Printexc.to_string exn)))
652
653 let parse_statement lexbuf =
654   exc_located_wrapper
655     (fun () -> (Grammar.Entry.parse statement (Obj.magic lexbuf)))