X-Git-Url: http://git.nguyen.vg/gitweb/?a=blobdiff_plain;f=src%2Foptions.ml;h=fbbd73e44b68ddcb7b7fbd1a7d5e153f0e8e3279;hb=b281387c0d6115086ad3c3f3d0510be734fc29a7;hp=70d702d665347cf73fb77bf1da759af20f6cef6e;hpb=43906e89a76c67491e2a567990980df787036088;p=SXSI%2Fxpathcomp.git diff --git a/src/options.ml b/src/options.ml index 70d702d..fbbd73e 100644 --- a/src/options.ml +++ b/src/options.ml @@ -1,26 +1,5 @@ -open Utils open Format - -let index_empty_texts = ref false -let sample_factor = ref 64 -let disable_text_collection = ref false -let tc_threshold = ref 60000 - -let query = ref "" -let input_file = ref "" -let output_file = ref None -let save_file = ref "" -let count_only = ref false -let time = ref false -let bottom_up = ref false -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" - +open Config let set_index_type = function | "default" -> text_index_type := 0 @@ -38,36 +17,39 @@ let anon_fun = | 2 -> output_file := Some s; incr pos | _ -> raise (Arg.Bad(s)) -let set_tracer s = +let set_logger s = List.iter (fun t -> if t = "" then () else - match String.explode t ':' with + match Utils.String.explode t ':' with [ tr; lvl ] -> let l = try int_of_string lvl with _ -> raise (Arg.Bad (lvl)) in - if Tracer.is_tracer tr then Tracer.activate tr l + if Logger.is_logger tr then Logger.activate tr l else raise (Arg.Bad (t)) | _ -> raise (Arg.Bad (t)) - ) (String.explode s ',') + ) (Utils.String.explode s ',') -let pretty_tracers () = +let pretty_loggers () = ignore(flush_str_formatter()); Pretty.print_list ~sep:", " (fun f s -> fprintf f "%s" s) str_formatter - (Tracer.available ()); + (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), " sample factor [default=64]"; - "-i", Arg.Set(index_empty_texts), - " index empty texts [default=false]"; + "-ne", Arg.Clear(index_empty_texts), + " don't index empty texts [default=index]"; "-d", Arg.Set(disable_text_collection), " disable text collection[default=false]"; @@ -79,30 +61,49 @@ let spec = Arg.align "-nj", Arg.Set(no_jump), " disable jumping"; + "-nc", Arg.Set(no_cache), " disable caching"; + + "-nw", Arg.Set(no_wrap_results), " do not wrap results in "; + "-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)"; + + "-doc-stats", Arg.Set(docstats), + " Compute document statistics (performs full traversal)"; + "-v", Arg.Set(verbose), " verbose mode"; ] @ IFNDEF NTRACE THEN [ - "-trace-file", Arg.Set_string(trace_file), - " save the full trace in dot format in "; - "-trace", Arg.String (set_tracer), - " enable tracing with the specified level. Valid tracers are: " - ^ (pretty_tracers ()) + "-log", Arg.String (set_logger), + " enable logging with the specified level. Valid loggers are: " + ^ (pretty_loggers ()) ] ELSE [] END +let real_spec = + spec @ [ + "-", Arg.Unit (fun () -> anon_fun "-"), ""; + ] + let parse_cmdline() = - let _ = Arg.parse spec anon_fun usage_msg + let _ = + try + Arg.parse_argv Sys.argv real_spec anon_fun usage_msg; + if (!pos > 3 || !pos < 2) then raise (Arg.Bad "Invalid number of arguments") + with + Arg.Bad s | Arg.Help s -> + eprintf "%s\n" (Arg.usage_string spec usage_msg); + exit 1 in - if (!pos > 3 || !pos < 2) - then begin Arg.usage spec usage_msg; exit 1 end + Logger.set_verbose !verbose