X-Git-Url: http://git.nguyen.vg/gitweb/?a=blobdiff_plain;f=src%2Foptions.ml;h=cec31bc6b91f364af7a8a32f06738094d31f5549;hb=fcdb6b70f80c4777e319917e9318fc8d54e3a28f;hp=d72763ec8ba29a2277cb223530acf3716a4db5a1;hpb=a84e4fba7daf4f600958487d88f0b733fb19042e;p=SXSI%2Fxpathcomp.git

diff --git a/src/options.ml b/src/options.ml
index d72763e..cec31bc 100644
--- a/src/options.ml
+++ b/src/options.ml
@@ -1,3 +1,6 @@
+open Utils
+open Format
+
 let index_empty_texts = ref false
 let sample_factor = ref 64
 let disable_text_collection = ref false
@@ -14,10 +17,8 @@ let no_jump = ref false
 let verbose = ref false
 let text_index_type = ref 0
 let do_perf = ref false
-
-(* Only valid if compiled with -DTRACE *)
-let trace_file = ref "trace.dot"
-
+let twopass = ref false
+let repeat = ref 1
 
 let set_index_type = function
   | "default" -> text_index_type := 0
@@ -27,7 +28,6 @@ let set_index_type = function
 
 let usage_msg = Printf.sprintf "%s [options] <input.{xml|srx}> 'query' [output]" Sys.argv.(0)
 
-
 let pos = ref 0
 let anon_fun =
   fun s -> match !pos with
@@ -36,10 +36,34 @@ let anon_fun =
     | 2 -> output_file := Some s; incr pos
     | _ -> raise (Arg.Bad(s))
 
+let set_logger s =
+  List.iter (fun t ->
+    if t = "" then ()
+    else
+      match String.explode t ':' with
+	[ tr; lvl ] ->
+	  let l = try int_of_string lvl with _ -> raise (Arg.Bad (lvl)) in
+	  if Logger.is_logger tr then Logger.activate tr l
+	  else raise (Arg.Bad (t))
+      | _ -> raise (Arg.Bad (t))
+  ) (String.explode s ',')
+
+let pretty_loggers () =
+  ignore(flush_str_formatter());
+  Pretty.print_list
+    ~sep:", "
+    (fun f s -> fprintf f "%s" s)
+    str_formatter
+    (Logger.available ());
+  flush_str_formatter ()
+
 let spec = Arg.align
   [ "-c", Arg.Set(count_only),
     " counting only (don't materialize the result set)";
 
+    "-two", Arg.Set(twopass),
+    " Use twopass algorithm";
+
     "-f", Arg.Set_int(sample_factor),
     "<n> sample factor [default=64]";
 
@@ -59,14 +83,20 @@ let spec = Arg.align
     "-p",  Arg.Set(do_perf), " dump perf counters (Linux only)";
 
     "-index-type", Arg.Symbol ([ "default"; "swcsa"; "rlcsa" ],
-			       set_index_type),
+                               set_index_type),
     " choose text index type";
 
+    "-r", Arg.Set_int(repeat),
+    " repeat query execution n time (benchmarking only, default 1)";
+
+
+
     "-v", Arg.Set(verbose), " verbose mode"; ] @
-IFDEF TRACE
+IFNDEF NTRACE
 THEN [
-    "-trace-file", Arg.Set_string(trace_file),
-    "<trace_file> save the full trace in dot format in <trace_file>"
+    "-log", Arg.String (set_logger),
+    "<logger1:l1,...,loggern:ln> enable logging with the specified level. Valid loggers are: "
+      ^ (pretty_loggers ())
      ]
 ELSE []
 END