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 process_stream ?(ignore_exn = false) istream =
  57   let char_count = ref 0 in
  58   let module P = CicNotationPt in
  59   let module G = GrafiteAst in
  60     try
  61       while Stream.peek istream <> None do
  62         try
  63           let statement = GrafiteParser.parse_statement istream in
  64           let floc = extract_loc statement in
  65           let (_, y) = P.loc_of_floc floc in
  66           char_count := y + !char_count;
  67           match statement with
  68           | G.Executable (_, G.Macro (_, G.Check (_, t))) ->
  69               prerr_endline "====";
  70               prerr_endline (CicNotationPp.pp_term t); flush stdout;
  71               let t' = CicNotationRew.pp_ast t in
  72                 prerr_endline (CicNotationPp.pp_term t'); flush stdout;
  73                 let tbl = Hashtbl.create 0 in
  74                   dump_xml t' tbl "out.xml"
  75           | G.Executable (_, G.Command (_,
  76             G.Notation (_, l1, associativity, precedence, l2))) ->
  77               prerr_endline "Extending parser ..."; flush stdout;
  78               prerr_endline (CicNotationPp.pp_term l1) ;
  79               prerr_endline (CicNotationPp.pp_term l2) ;
  80               prerr_endline (sprintf "Found keywords: %s"
  81                 (String.concat " " (CicNotationUtil.keywords_of_term l1)));
  82               let time1 = Unix.gettimeofday () in
  83                 ignore (CicNotationParser.extend l1 ?precedence ?associativity
  84                    (fun env loc -> CicNotationFwd.instantiate_level2 env l2));
  85                 let time2 = Unix.gettimeofday () in
  86                   prerr_endline "Extending pretty printer ..."; flush stdout;
  87                   let time3 = Unix.gettimeofday () in
  88                   ignore (CicNotationRew.add_pretty_printer
  89                     ?precedence ?associativity l2 l1);
  90                     let time4 = Unix.gettimeofday () in
  91                       printf ("done (extending parser took %f, " ^^
  92                         "extending pretty printer took %f)\n")
  93                         (time2 -. time1) (time4 -. time3);
  94                       flush stdout
  95           | G.Executable (_, G.Command (_, G.Interpretation (_, id, l2, l3))) ->
  96               prerr_endline "Adding interpretation ..."; flush stdout;
  97               let time1 = Unix.gettimeofday () in
  98                 ignore (CicNotationRew.add_interpretation id l2 l3);
  99                 let time2 = Unix.gettimeofday () in
 100                   printf "done (patterns compilation took %f seconds)\n"
 101                     (time2 -. time1);
 102                   flush stdout
 103           | G.Executable (_, G.Command (_, G.Dump _)) ->
 104               CicNotationParser.print_l2_pattern (); print_newline ()
 105           | G.Executable (_, G.Command (_, G.Render (_, uri))) ->
 106               let obj, _ = CicEnvironment.get_obj CicUniv.empty_ugraph uri in
 107               let annobj, _, _, id_to_sort, _, _, _ =
 108                 Cic2acic.acic_object_of_cic_object obj
 109               in
 110               let annterm =
 111                 match annobj with
 112                   | Cic.AConstant (_, _, _, _, ty, _, _)
 113                   | Cic.AVariable (_, _, _, ty, _, _) -> ty
 114                   | _ -> assert false
 115               in
 116               let time1 = Unix.gettimeofday () in
 117               let t, id_to_uri =
 118                 CicNotationRew.ast_of_acic id_to_sort annterm
 119               in
 120               let time2 = Unix.gettimeofday () in
 121                 prerr_endline (sprintf "ast creation took %f seconds\n"
 122                   (time2 -. time1));
 123                 prerr_endline "AST";
 124                 prerr_endline (CicNotationPp.pp_term t);
 125                 flush stdout;
 126                 let time3 = Unix.gettimeofday () in
 127                 let t' = CicNotationRew.pp_ast t in
 128                 let time4 = Unix.gettimeofday () in
 129                   prerr_endline (sprintf "pretty printing took %f seconds\n"
 130                     (time4 -. time3));
 131                   prerr_endline (CicNotationPp.pp_term t');
 132                   dump_xml t' id_to_uri "out.xml"
 133           | _ ->
 134               prerr_endline "Unsupported statement"
 135         with
 136         | End_of_file -> raise End_of_file
 137         | CicNotationParser.Parse_error (floc, msg) ->
 138             let (x, y) = P.loc_of_floc floc in
 139 (*             let before = String.sub line 0 x in
 140             let error = String.sub line x (y - x) in
 141             let after = String.sub line y (String.length line - y) in
 142             eprintf "%s\e[01;31m%s\e[00m%s\n" before error after;
 143             prerr_endline (sprintf "at character %d-%d: %s" x y msg) *)
 144             prerr_endline (sprintf "Parse error at character %d-%d: %s"
 145               (!char_count + x) (!char_count + y) msg);
 146             errquit ignore_exn
 147         | exn ->
 148             prerr_endline
 149               (sprintf "Uncaught exception: %s" (Printexc.to_string exn));
 150             errquit ignore_exn
 151         done
 152     with End_of_file ->
 153       ()
 154
 155 let _ =
 156   let arg_spec = [ ] in
 157   let usage = "" in
 158   Arg.parse arg_spec (fun _ -> raise (Arg.Bad usage)) usage;
 159   print_endline "Loading builtin notation ...";
 160   let ic = open_in (Helm_registry.get "notation.core_file") in
 161   process_stream ~ignore_exn:true (Stream.of_channel ic);
 162   close_in ic;
 163   print_endline "done.";
 164   flush stdout;
 165   process_stream ~ignore_exn:false (Stream.of_channel stdin)
 166