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 (sprintf "Found keywords: %s"
  80                 (String.concat " " (CicNotationUtil.keywords_of_term l1)));
  81               let time1 = Unix.gettimeofday () in
  82                 ignore (CicNotationParser.extend l1 ?precedence ?associativity
  83                    (fun env loc -> CicNotationFwd.instantiate_level2 env l2));
  84                 let time2 = Unix.gettimeofday () in
  85                   prerr_endline "Extending pretty printer ..."; flush stdout;
  86                   let time3 = Unix.gettimeofday () in
  87                   ignore (CicNotationRew.add_pretty_printer
  88                     ?precedence ?associativity l2 l1);
  89                     let time4 = Unix.gettimeofday () in
  90                       printf ("done (extending parser took %f, " ^^
  91                         "extending pretty printer took %f)\n")
  92                         (time2 -. time1) (time4 -. time3);
  93                       flush stdout
  94           | G.Executable (_, G.Command (_, G.Interpretation (_, l2, l3))) ->
  95               prerr_endline "Adding interpretation ..."; flush stdout;
  96               let time1 = Unix.gettimeofday () in
  97                 ignore (CicNotationRew.add_interpretation l2 l3);
  98                 let time2 = Unix.gettimeofday () in
  99                   printf "done (patterns compilation took %f seconds)\n"
 100                     (time2 -. time1);
 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 time1 = Unix.gettimeofday () in
 116               let t, id_to_uri =
 117                 CicNotationRew.ast_of_acic id_to_sort annterm
 118               in
 119               let time2 = Unix.gettimeofday () in
 120                 prerr_endline (sprintf "ast creation took %f seconds\n"
 121                   (time2 -. time1));
 122                 prerr_endline "AST";
 123                 prerr_endline (CicNotationPp.pp_term t);
 124                 flush stdout;
 125                 let time3 = Unix.gettimeofday () in
 126                 let t' = CicNotationRew.pp_ast t in
 127                 let time4 = Unix.gettimeofday () in
 128                   prerr_endline (sprintf "pretty printing took %f seconds\n"
 129                     (time4 -. time3));
 130                   prerr_endline (CicNotationPp.pp_term t');
 131                   dump_xml t' id_to_uri "out.xml"
 132           | _ ->
 133               prerr_endline "Unsupported statement"
 134         with
 135         | End_of_file -> raise End_of_file
 136         | CicNotationParser.Parse_error (floc, msg) ->
 137             let (x, y) = P.loc_of_floc floc in
 138 (*             let before = String.sub line 0 x in
 139             let error = String.sub line x (y - x) in
 140             let after = String.sub line y (String.length line - y) in
 141             eprintf "%s\e[01;31m%s\e[00m%s\n" before error after;
 142             prerr_endline (sprintf "at character %d-%d: %s" x y msg) *)
 143             prerr_endline (sprintf "Parse error at character %d-%d: %s"
 144               (!char_count + x) (!char_count + y) msg);
 145             errquit ignore_exn
 146         | exn ->
 147             prerr_endline
 148               (sprintf "Uncaught exception: %s" (Printexc.to_string exn));
 149             errquit ignore_exn
 150         done
 151     with End_of_file ->
 152       ()
 153
 154 let _ =
 155   let arg_spec = [ ] in
 156   let usage = "" in
 157   Arg.parse arg_spec (fun _ -> raise (Arg.Bad usage)) usage;
 158   print_endline "Loading builtin notation ...";
 159   let ic = open_in (Helm_registry.get "notation.core_file") in
 160   process_stream ~ignore_exn:true (Stream.of_channel ic);
 161   close_in ic;
 162   print_endline "done.";
 163   flush stdout;
 164   process_stream ~ignore_exn:false (Stream.of_channel stdin)
 165