main.ml

   1 (******************************************************************************)
   2 (*  SXSI : XPath evaluator                                                    *)
   3 (*  Kim Nguyen (Kim.Nguyen@nicta.com.au)                                      *)
   4 (*  Copyright NICTA 2008                                                      *)
   5 (*  Distributed under the terms of the LGPL (see LICENCE)                     *)
   6 (******************************************************************************)
   7 INCLUDE "debug.ml"
   8
   9 open Automaton
  10 let a = ref None
  11
  12 let l = ref [] ;;
  13 let time f x =
  14   let t1 = Unix.gettimeofday () in
  15   let r = f x in
  16   let t2 = Unix.gettimeofday () in
  17   let t = (1000. *.(t2 -. t1)) in
  18     l:= t::!l;
  19     Printf.eprintf "  %fms\n%!" t ;
  20     r
  21 ;;
  22 let total_time () =  List.fold_left (+.) 0. !l;;
  23
  24 let test filename query output =
  25   Printf.eprintf "Parsing document : %!";
  26   let v = time Tree.Binary.parse_xml_uri filename in
  27 (*    Tree.dump v;
  28     Printf.eprintf "\n\n\n\n\n%!";
  29     Tree.Binary.print_xml_fast stderr v  *)
  30     Printf.eprintf "Full Traversal\n%!";
  31     time (Tree.full_traversal) v;
  32     Printf.eprintf "Traversal\n%!";
  33     time (Tree.traversal) v;
  34     Printf.eprintf "CPP Traversal\n%!";
  35     time (Tree.cpp_traversal) v
  36
  37
  38 let main filename query output =
  39   Printf.eprintf "Parsing document : %!";
  40   let v = time Tree.Binary.parse_xml_uri filename in
  41     MM(v,__LOCATION__);
  42     a := Some (v);
  43     a := None;
  44     Printf.eprintf "Parsing query : ";
  45     let query = try
  46       time
  47         XPath.Parser.parse_string query
  48     with
  49         Ulexer.Loc.Exc_located ((x,y),e) -> Printf.eprintf "character %i-%i %s\n" x y (Printexc.to_string e);exit 1
  50     in
  51       Printf.eprintf "Compiling query : ";
  52       let auto = time XPath.Compile.compile  query in
  53         XPath.Ast.print Format.err_formatter query;
  54         Format.eprintf "\n%!";
  55
  56         Printf.eprintf "TopDown (No BackTrack) : \n";
  57         time (fun v -> ignore (TopDown.accept auto v)) v;
  58         Printf.eprintf "Number of nodes in the result set : %i\n" (BST.cardinal auto.result);
  59         begin
  60           match output with
  61             | None -> ()
  62             | Some f ->
  63
  64                 Printf.eprintf "Serializing results : ";
  65                 time( fun () ->
  66                         let oc = open_out f in
  67                           output_string oc "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n";
  68                           BST.iter (fun t -> Tree.Binary.print_xml_fast oc t;
  69                                       output_string oc "\n------------------";
  70                                     output_char oc '\n') auto.result) ();
  71         end;
  72         Printf.eprintf "Total time : %fms\n Coherence : %i\n%!" (total_time())
  73 ;;
  74
  75
  76 let argc = Array.length Sys.argv;;
  77 if (argc < 3 || argc >4)
  78 then
  79   (prerr_endline ("usage : " ^ Sys.argv.(0) ^ " <document> \'query\'[ <output> ]");
  80    exit 1)
  81 ;;
  82
  83
  84 test Sys.argv.(1) Sys.argv.(2) (if argc == 4 then Some Sys.argv.(3) else None) ;;
  85