(* Copyright (C) 2005, HELM Team.
 * 
 * This file is part of HELM, an Hypertextual, Electronic
 * Library of Mathematics, developed at the Computer Science
 * Department, University of Bologna, Italy.
 * 
 * HELM is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * HELM is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HELM; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
 * MA  02111-1307, USA.
 * 
 * For details, see the HELM World-Wide-Web page,
 * http://helm.cs.unibo.it/
 *)

open Printf

let _ =
  Helm_registry.load_from "test_parser.conf.xml"
(*   Http_getter.init () *)

let xml_stream_of_markup =
  let rec print_box (t: CicNotationPres.boxml_markup) =
    Box.box2xml print_mpres t
  and print_mpres (t: CicNotationPres.mathml_markup) =
    Mpresentation.print_mpres print_box t
  in
  print_mpres

let dump_xml t id_to_uri fname =
  print_endline (sprintf "dumping MathML to %s ..." fname);
  flush stdout;
  let oc = open_out fname in
  Xml.pp_to_outchan (CicNotationPres.render_to_boxml id_to_uri t) oc;
  close_out oc

let _ =
  let module P = CicNotationPt in
  let level = ref ~-1 in
  let arg_spec = [ "-level", Arg.Set_int level, "set the notation level" ] in
  let usage = "test_parser -level { 1 | 2 | 3 }" in
  Arg.parse arg_spec (fun _ -> raise (Arg.Bad usage)) usage;
  let ic = stdin in
  try
    printf "Parsing notation level %d\n" !level; flush stdout;
    while true do
      let line = input_line ic in
      let istream = Stream.of_string line in
      try
        (match !level with
        | -1 ->
            (match CicNotationParser.parse_phrase istream with
            | P.Print t -> 
                prerr_endline "====";
                print_endline (CicNotationPp.pp_term t); flush stdout;
                let t' = CicNotationRew.pp_ast t in
                print_endline (CicNotationPp.pp_term t'); flush stdout;
                let tbl = Hashtbl.create 0 in
                dump_xml t' tbl "out.xml"
            | P.Notation (l1, associativity, precedence, l2) ->
                print_endline "Extending parser ..."; flush stdout;
                let time1 = Unix.gettimeofday () in
                ignore
                  (CicNotationParser.extend l1 ?precedence ?associativity
                    (fun env loc -> CicNotationFwd.instantiate_level2 env l2));
                let time2 = Unix.gettimeofday () in
                print_endline "Extending pretty printer ..."; flush stdout;
                let time3 = Unix.gettimeofday () in
                ignore
                  (CicNotationRew.add_pretty_printer ?precedence ?associativity
                    l2 l1);
                let time4 = Unix.gettimeofday () in
                printf "done (extending parser took %f, extending pretty printer took %f)\n"
                  (time2 -. time1) (time4 -. time3);
                flush stdout
            | P.Interpretation (l2, l3) ->
                print_endline "Adding interpretation ..."; flush stdout;
                let time1 = Unix.gettimeofday () in
                ignore (CicNotationRew.add_interpretation l2 l3);
                let time2 = Unix.gettimeofday () in
                printf "done (patterns compilation took %f seconds)\n"
                  (time2 -. time1);
                flush stdout
            | P.Render uri ->
                let obj, _ = CicEnvironment.get_obj CicUniv.empty_ugraph uri in
                let annobj, _, _, id_to_sort, _, _, _ =
                  Cic2acic.acic_object_of_cic_object obj
                in
                let annterm =
                  match annobj with
                  | Cic.AConstant (_, _, _, _, ty, _, _)
                  | Cic.AVariable (_, _, _, ty, _, _) -> ty
                  | _ -> assert false
                in
                let time1 = Unix.gettimeofday () in
                let t, id_to_uri =
                  CicNotationRew.ast_of_acic id_to_sort annterm
                in
                let time2 = Unix.gettimeofday () in
                printf "ast creation took %f seconds\n" (time2 -. time1);
                prerr_endline "====";
                print_endline (CicNotationPp.pp_term t); flush stdout;
                flush stdout;
                let time3 = Unix.gettimeofday () in
                let t' = CicNotationRew.pp_ast t in
                let time4 = Unix.gettimeofday () in
                printf "pretty printing took %f seconds\n" (time4 -. time3);
                dump_xml t' id_to_uri "out.xml";
                print_endline (CicNotationPp.pp_term t'); flush stdout
                )
(*                 CicNotationParser.print_l2_pattern ()) *)
        | 1 -> ignore (CicNotationParser.parse_syntax_pattern istream)
        | 2 ->
            let ast = CicNotationParser.parse_ast_pattern istream in
            if ast = P.Sort `Prop then
              prerr_endline "eureka"
            else
              prerr_endline ":-("
        | 3 -> ignore (CicNotationParser.parse_interpretation istream)
        | _ -> Arg.usage arg_spec usage; exit 1);
      with CicNotationParser.Parse_error (floc, msg) ->
        let (x, y) = P.loc_of_floc floc in
        let before = String.sub line 0 x in
        let error = String.sub line x (y - x) in
        let after = String.sub line y (String.length line - y) in
        eprintf "%s%s%s\n" before error after;
        prerr_endline (sprintf "at character %d-%d: %s" x y msg)
    done
  with End_of_file ->
    close_in ic