Split the Options module in two to remove a circular dependency in
[SXSI/xpathcomp.git] / src / main.ml
index 1981fc2..af00682 100644 (file)
@@ -13,127 +13,113 @@ let () = init_timer();;
 
 let default_gc = Gc.get()
 let tuned_gc = { default_gc with
-                  Gc.minor_heap_size = 32*1024*1024;
-                  Gc.major_heap_increment = 8*1024*1024;
-                  Gc.max_overhead = 1000000;
-                  Gc.space_overhead = 100;
-              }
-
+  Gc.minor_heap_size = 32*1024*1024;
+  Gc.major_heap_increment = 8*1024*1024;
+  Gc.max_overhead = 1000000;
+  Gc.space_overhead = 100;
+}
 
 let mk_runtime run auto doc arg count print outfile =
   fun () ->
-    let r = time ~count:1 ~msg:"Execution time" (run auto doc) arg in
-    Printf.eprintf "Number of results: %i\n%!" (count r);
+    if !Config.do_perf then start_perf ();
+    let r = time ~count:!Config.repeat ~msg:"Execution time" (run auto doc) arg in
+    if !Config.do_perf then stop_perf ();
+    Logger.verbose Format.err_formatter "Number of results: %i@\n" (count r);
     match outfile with
        None -> ()
       | Some file ->
-       time ~count:1 ~msg:"Serialization time" (print file doc) r
+       time ~count:1 ~msg:"Serialization time" (print file !Config.no_wrap_results doc) r
 ;;
 
-
 let main v query_string output =
-  Tag.init (Tree.tag_pool v);
+  Tag.init (Tree.tag_operations v);
+  if !Config.docstats then Tree.stats v;
   let query =
     time ~msg:"Parsing query" XPath.parse query_string
   in
-  if !Options.verbose then begin
-    Printf.eprintf "Parsed query:\n%!";
-    XPath.Ast.print Format.err_formatter query;
-    Format.fprintf Format.err_formatter "\n%!"
-  end;
+  Logger.verbose Format.err_formatter "Parsed query:%a@\n"
+    XPath.Ast.print query;
   let auto, bu_info =
-    time ~msg:"Compiling query" (Compile.compile) query
+    time ~msg:"Compiling query" Compile.compile query
   in
-  if !Options.verbose then Ata.print Format.err_formatter auto;
+  if !Config.verbose then Ata.print Format.err_formatter auto;
   Gc.full_major();
   Gc.compact();
   Gc.set (tuned_gc);
   let runtime =
-  match !Options.bottom_up, bu_info with
+    match !Config.bottom_up, bu_info with
 
     | true, Some [ (query, pattern) ] ->
-      (* let nodes =
-       time
-         ~count:1 ~msg:"Computing full text query"
-         (Tree.full_text_query query v) pattern
-      in
-      let nodes = Array.to_list nodes in *)
-      if !Options.count_only then
-       let module R = ResJIT.Count in
-       let module M = Runtime.Make(R) in
-       mk_runtime M.bottom_up_run auto v (query, pattern) R.NS.length R.NS.serialize None
+      if !Config.count_only then
+        let module R = ResJIT.Count in
+        let module M = Runtime.Make(R) in
+        mk_runtime M.bottom_up_run auto v (query, pattern) R.NS.length R.NS.serialize !Config.output_file
       else
-       let module R = ResJIT.Mat in
-       let module M = Runtime.Make(R) in
-       mk_runtime M.bottom_up_run auto v (query, pattern) R.NS.length R.NS.serialize !Options.output_file
+        let module R = ResJIT.Mat in
+        let module M = Runtime.Make(R) in
+        mk_runtime M.bottom_up_run auto v (query, pattern) R.NS.length R.NS.serialize !Config.output_file
 
     | _ ->
-       (* run the query top_down *)
-
-      if !Options.bottom_up then
-       Printf.eprintf "Cannot run the query in bottom-up mode, using top-down evaluator\n%!";
-
-      if !Options.count_only then
-       let module R = ResJIT.Count in
-       let module M = Runtime.Make(R) in
-       (* mk_runtime run auto doc arg count print outfile  *)
-       mk_runtime M.top_down_run auto v Tree.root R.NS.length R.NS.serialize None
+      (* run the query top_down *)
+
+      if !Config.bottom_up then
+        Logger.verbose Format.err_formatter "Cannot run the query in bottom-up mode, using top-down evaluator@\n@?";
+      if !Config.count_only then
+        let module R = ResJIT.Count in
+        let module M = Runtime.Make(R) in
+        if !Config.twopass then
+          mk_runtime M.twopass_top_down_run auto v Tree.root R.NS.length R.NS.serialize None
+        else
+          mk_runtime M.top_down_run auto v Tree.root R.NS.length R.NS.serialize !Config.output_file
       else
-       let module R = ResJIT.Mat in
-       let module M = Runtime.Make(R) in
-       mk_runtime M.top_down_run auto v Tree.root R.NS.length R.NS.serialize !Options.output_file
+        let module R = ResJIT.Mat in
+        let module M = Runtime.Make(R) in
+        mk_runtime M.top_down_run auto v Tree.root R.NS.length R.NS.serialize !Config.output_file
   in
   runtime ()
 ;;
 
 let () = Options.parse_cmdline()
 ;;
-
-let document =
-  if Filename.check_suffix !Options.input_file ".srx"
-  then
-    time
-      ~msg:"Loading file"
-      (Tree.load
-        ~sample:!Options.sample_factor
-        ~load_text:true)
-      !Options.input_file
-  else
-    let v =
-      time
-       ~msg:"Parsing document"
-       (Tree.parse_xml_uri)
-       !Options.input_file
-    in
-    let () =
-      if !Options.save_file <> ""
+let _ =
+  try
+    Printexc.record_backtrace true;
+    let document =
+      if Filename.check_suffix !Config.input_file ".srx"
       then
        time
-         ~msg:"Writing file to disk"
-         (Tree.save v)
-         !Options.save_file;
+         ~msg:"Loading file"
+         (Tree.load
+            ~sample:!Config.sample_factor
+            ~load_text:(not !Config.disable_text_collection))
+         !Config.input_file
+      else
+       let v =
+         time
+           ~msg:"Parsing document"
+           (Tree.parse_xml_uri)
+           !Config.input_file
+       in
+        let () =
+          if !Config.save_file <> ""
+          then
+           time
+             ~msg:"Writing file to disk"
+             (Tree.save v)
+             !Config.save_file;
+       in
+       v
     in
-      v
-in
-  try
-    (*Printexc.record_backtrace true; *)
-    main document !Options.query !Options.output_file;
-    if !Options.verbose then Printf.eprintf "Maximum resident set size: %s\n" (read_procmem());
+    main document !Config.query !Config.output_file;
+    Logger.verbose Format.err_formatter "Maximum resident set size: %s @\n" (read_procmem());
     Gc.full_major();
     Profile.summary Format.err_formatter
   with
-    | Ulexer.Loc.Exc_located ((x,y),e) ->
-       Printf.eprintf "character %i-%i %s\n" x y (Printexc.to_string e);
-       exit 1
-
-    | e ->
-       output_string stderr "\n";
-       flush stderr;
-       Printexc.print_backtrace stderr;
-       Printf.eprintf "FATAL ERROR: %s\n%!" (Printexc.to_string e);
-       output_string stderr "\n";
-       flush stderr;
-(*     Ptset.Int.stats(); *)
-       exit 2
-
-
+  | Ulexer.Loc.Exc_located ((x,y),e) ->
+    Logger.print Format.err_formatter "character %i-%i %s@\n" x y (Printexc.to_string e);
+    exit 1
+
+  | e ->
+    Logger.print Format.err_formatter "BACKTRACE: %s@\n@?" (Printexc.get_backtrace());
+    Logger.print Format.err_formatter "FATAL ERROR: %s@\n@?" (Printexc.to_string e);
+    exit 2