(***********************************************************************)
(*
- Time-stamp: <Last modified on 2013-03-13 21:29:17 CET by Kim Nguyen>
+ Time-stamp: <Last modified on 2013-04-22 18:48:48 CEST by Kim Nguyen>
*)
-let doc =
- let fd = open_in Sys.argv.(1) in
- let d = Tree.Naive.load_xml_file fd in
- close_in fd; d
+open Format
+let time f arg msg =
+ let t1 = Unix.gettimeofday () in
+ let r = f arg in
+ let t2 = Unix.gettimeofday () in
+ let time = (t2 -. t1) *. 1000. in
+ if !Options.stats then fprintf err_formatter "@[STATS: %s: %fms@]@." msg time;
+ r
-let query =
- let arg2 = Sys.argv.(2) in
- Xpath.Parser.parse (Ulexing.from_latin1_string arg2)
-let auto =
- Xpath.Compile.path query
+let main () =
+ let () = Options.parse () in
+ let doc =
+ let fd = open_in !Options.input_file in
+ let d = time Naive_tree.load_xml_file fd "parsing xml document" in
+ close_in fd; d
+ in
+ let query =
+ time
+ Xpath.Parser.parse
+ (Ulexing.from_latin1_string !Options.query)
+ "parsing XPath query"
+ in
+ let auto =
+ time Xpath.Compile.path query "compiling XPath query"
+ in
+ let output =
+ match !Options.output_file with
+ | None | Some "-" | Some "/dev/stdout" -> stdout
+ | Some f -> open_out f
+ in
+ if !Options.stats then begin
+ fprintf err_formatter "@[STATS: Query: %a @]@." Xpath.Ast.print_path query;
+ fprintf err_formatter "@[STATS: @[Automaton: @\n";
+ Ata.print err_formatter auto;
+ fprintf err_formatter "@]@]@.";
+ end;
+
+ let module Naive = Eval.Make(Naive_tree) in
+ let results =
+ time (Naive.eval auto doc) (Naive_tree.root doc) "evaluating query"
+ in
+ time (fun () ->
+ output_string output "<xml_result>\n";
+ if !Options.count then begin
+ output_string output (string_of_int (List.length results));
+ output_char output '\n';
+ end else
+ List.iter (fun n ->
+ Naive_tree.print_xml output doc n;
+ output_char output '\n'
+ ) results;
+ output_string output "</xml_result>\n";
+ flush output;
+ if output != stdout then close_out output
+
+) () "serializing results"
-open Format
let () =
- fprintf err_formatter "Query: %a\n%!" Xpath.Ast.print_path query;
- fprintf err_formatter "Automata: %a\n%!" Auto.Ata.print auto;
- fprintf err_formatter "Evaluating automaton:\n%!";
- let module Naive = Auto.Eval.Make(Tree.Naive) in
- let t1 = Unix.gettimeofday() in
- let results = Naive.eval auto doc (Tree.Naive.root doc) in
- let teval = (Unix.gettimeofday () -. t1) *. 1000. in
- let t1 = Unix.gettimeofday () in
- output_string stdout "<xml_result>\n";
- List.iter (fun n ->
- Tree.Naive.print_xml stdout doc n;
- output_char stdout '\n'
- ) results;
- output_string stdout "</xml_result>\n";
- let tprint = (Unix.gettimeofday () -. t1) *. 1000. in
- flush stdout;
- fprintf err_formatter "evaluation time: %fms\nserialization time: %fms\n%!" teval tprint
+ try
+ main ()
+ with
+ Arg.Bad msg -> eprintf "Error: %s\n%!" msg; Options.usage (); exit 1
+ | Sys_error msg -> eprintf "Error: %s\n%!" msg; exit 2
+ | Tree.Parse_error msg -> eprintf "Error: file %s, %s\n%!" !Options.input_file msg; exit 3
+ | Xpath.Ulexer.Error (s, e, msg) -> eprintf "Error: character %i-%i: %s\n%!" s e msg; exit 4
+ | e -> eprintf "FATAL ERROR: %s\n%!" (Printexc.to_string e); exit 128