(* 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 _ =
  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;
  if !level = 2 then begin
    let id =
      CicNotationParser.extend ~precedence:50 ~associativity:Gramext.LeftA
        (P.Layout (P.Box (P.H,
          [
            P.Magic
              (P.List1
                (P.Layout (P.Box (P.H,
                  [ P.Literal (`Symbol "|");
                    P.Variable (P.TermVar "ugo");
                    P.Magic (P.Opt (P.Layout (P.Box (P.H,
                      [ P.Literal (`Symbol ",");
                        P.Variable (P.TermVar "pino")]))));
                    P.Literal (`Symbol "|");
                  ])),
                  Some (`Symbol ";")));
(*             P.Literal (`Symbol "+");
            P.Magic (P.Opt (P.Layout (P.Box (P.H,
              [
                P.Variable (P.TermVar "ugo");
                P.Literal (`Symbol "+");
                P.Variable (P.TermVar "pino")
              ])))); *)
(*           P.Variable (P.TermVar "a");
            P.Literal (`Symbol "+");
            P.Variable (P.TermVar "b"); *)
          ])))
        (fun env _ ->
          prerr_endline "reducing rule" ;
          prerr_endline (sprintf "env = [ %s ]" (CicNotationPp.pp_env env));
          P.Sort `Prop)
    in
    CicNotationParser.print_l2_pattern ()
  end;
  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
            | 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 ->
                      prerr_endline "ENV";
                      prerr_endline (CicNotationPp.pp_env env);
                      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);
                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