helm/ocaml/cic_notation/test_parser.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 open Printf
  27
  28 let _ = Helm_registry.load_from "test_parser.conf.xml"
  29
  30 let xml_stream_of_markup =
  31   let rec print_box (t: CicNotationPres.boxml_markup) =
  32     Box.box2xml print_mpres t
  33   and print_mpres (t: CicNotationPres.mathml_markup) =
  34     Mpresentation.print_mpres print_box t
  35   in
  36   print_mpres
  37
  38 let dump_xml t id_to_uri fname =
  39   prerr_endline (sprintf "dumping MathML to %s ..." fname);
  40   flush stdout;
  41   let oc = open_out fname in
  42   Xml.pp_to_outchan (CicNotationPres.render_to_boxml id_to_uri t) oc;
  43   close_out oc
  44
  45 let extract_loc =
  46   function
  47     | GrafiteAst.Executable (loc, _)
  48     | GrafiteAst.Comment (loc, _) -> loc
  49
  50 let errquit ignore_exn =
  51   if not ignore_exn then begin
  52     prerr_endline "Error, exiting!";
  53     exit 2
  54   end
  55
  56 let pp_associativity = function
  57   | Gramext.LeftA -> "left"
  58   | Gramext.RightA -> "right"
  59   | Gramext.NonA -> "non"
  60
  61 let pp_precedence = string_of_int
  62
  63 let process_stream ?(ignore_exn = false) istream =
  64   let char_count = ref 0 in
  65   let module P = CicNotationPt in
  66   let module G = GrafiteAst in
  67     try
  68       while Stream.peek istream <> None do
  69         try
  70           let statement = GrafiteParser.parse_statement istream in
  71           let floc = extract_loc statement in
  72           let (_, y) = P.loc_of_floc floc in
  73           char_count := y + !char_count;
  74           match statement with
  75           | G.Executable (_, G.Macro (_, G.Check (_, t))) ->
  76               prerr_endline (sprintf "ast: %s" (CicNotationPp.pp_term t));
  77               let t' = CicNotationRew.pp_ast t in
  78               prerr_endline (sprintf "rendered ast: %s"
  79                 (CicNotationPp.pp_term t'));
  80               let tbl = Hashtbl.create 0 in
  81               dump_xml t' tbl "out.xml"
  82           | G.Executable (_, G.Command (_,
  83             G.Notation (_, l1, associativity, precedence, l2))) ->
  84               prerr_endline "notation";
  85               prerr_endline (sprintf "l1: %s" (CicNotationPp.pp_term l1));
  86               prerr_endline (sprintf "l2: %s" (CicNotationPp.pp_term l2));
  87               prerr_endline (sprintf "prec: %s" (pp_precedence precedence));
  88               prerr_endline (sprintf "assoc: %s" (pp_associativity associativity));
  89               let keywords = CicNotationUtil.keywords_of_term l1 in
  90               if keywords <> [] then
  91                 prerr_endline (sprintf "keywords: %s"
  92                   (String.concat " " keywords));
  93               ignore (CicNotationParser.extend l1 ?precedence ?associativity
  94                 (fun env loc -> CicNotationFwd.instantiate_level2 env l2));
  95               ignore (CicNotationRew.add_pretty_printer
  96                 ?precedence ?associativity l2 l1)
  97           | G.Executable (_, G.Command (_, G.Interpretation (_, id, l2, l3))) ->
  98               prerr_endline "interpretation";
  99               prerr_endline (sprintf "dsc: %s" id);
 100               ignore (CicNotationRew.add_interpretation id l2 l3);
 101               flush stdout
 102           | G.Executable (_, G.Command (_, G.Dump _)) ->
 103               CicNotationParser.print_l2_pattern (); print_newline ()
 104           | G.Executable (_, G.Command (_, G.Render (_, uri))) ->
 105               let obj, _ = CicEnvironment.get_obj CicUniv.empty_ugraph uri in
 106               let annobj, _, _, id_to_sort, _, _, _ =
 107                 Cic2acic.acic_object_of_cic_object obj
 108               in
 109               let annterm =
 110                 match annobj with
 111                   | Cic.AConstant (_, _, _, _, ty, _, _)
 112                   | Cic.AVariable (_, _, _, ty, _, _) -> ty
 113                   | _ -> assert false
 114               in
 115               let t, id_to_uri =
 116                 CicNotationRew.ast_of_acic id_to_sort annterm
 117               in
 118               prerr_endline "AST";
 119               prerr_endline (CicNotationPp.pp_term t);
 120               let t' = CicNotationRew.pp_ast t in
 121               prerr_endline (CicNotationPp.pp_term t');
 122               dump_xml t' id_to_uri "out.xml"
 123           | _ -> prerr_endline "Unsupported statement"
 124         with
 125         | End_of_file -> raise End_of_file
 126         | CicNotationParser.Parse_error (floc, msg) ->
 127             let (x, y) = P.loc_of_floc floc in
 128 (*             let before = String.sub line 0 x in
 129             let error = String.sub line x (y - x) in
 130             let after = String.sub line y (String.length line - y) in
 131             eprintf "%s\e[01;31m%s\e[00m%s\n" before error after;
 132             prerr_endline (sprintf "at character %d-%d: %s" x y msg) *)
 133             prerr_endline (sprintf "Parse error at character %d-%d: %s"
 134               (!char_count + x) (!char_count + y) msg);
 135             errquit ignore_exn
 136         | exn ->
 137             prerr_endline
 138               (sprintf "Uncaught exception: %s" (Printexc.to_string exn));
 139             errquit ignore_exn
 140         done
 141     with End_of_file ->
 142       ()
 143
 144 let _ =
 145   let arg_spec = [ ] in
 146   let usage = "" in
 147   Arg.parse arg_spec (fun _ -> raise (Arg.Bad usage)) usage;
 148   print_endline "Loading builtin notation ...";
 149   CicNotation.load_notation (Helm_registry.get "notation.core_file");
 150   print_endline "done.";
 151   flush stdout;
 152   process_stream ~ignore_exn:false (Stream.of_channel stdin)
 153