4 let index_empty_texts = ref false
5 let sample_factor = ref 64
6 let disable_text_collection = ref false
7 let tc_threshold = ref 60000
10 let input_file = ref ""
11 let output_file = ref None
12 let save_file = ref ""
13 let count_only = ref false
15 let bottom_up = ref false
16 let no_jump = ref false
17 let verbose = ref false
18 let text_index_type = ref 0
19 let do_perf = ref false
21 (* Only valid if compiled with -DTRACE *)
22 let trace_file = ref "trace.dot"
25 let set_index_type = function
26 | "default" -> text_index_type := 0
27 | "swcsa" -> text_index_type := 1
28 | "rlcsa" -> text_index_type := 2
29 | s -> raise (Arg.Bad(s))
31 let usage_msg = Printf.sprintf "%s [options] <input.{xml|srx}> 'query' [output]" Sys.argv.(0)
35 fun s -> match !pos with
36 | 0 -> input_file:= s;incr pos
37 | 1 -> query := s; incr pos
38 | 2 -> output_file := Some s; incr pos
39 | _ -> raise (Arg.Bad(s))
45 match String.explode t ':' with
47 let l = try int_of_string lvl with _ -> raise (Arg.Bad (lvl)) in
48 if Tracer.is_tracer tr then Tracer.activate tr l
49 else raise (Arg.Bad (t))
50 | _ -> raise (Arg.Bad (t))
51 ) (String.explode s ',')
53 let pretty_tracers () =
54 ignore(flush_str_formatter());
57 (fun f s -> fprintf f "%s" s)
59 (Tracer.available ());
60 flush_str_formatter ()
63 [ "-c", Arg.Set(count_only),
64 " counting only (don't materialize the result set)";
66 "-f", Arg.Set_int(sample_factor),
67 "<n> sample factor [default=64]";
69 "-i", Arg.Set(index_empty_texts),
70 " index empty texts [default=false]";
72 "-d", Arg.Set(disable_text_collection),
73 " disable text collection[default=false]";
75 "-s", Arg.Set_string(save_file),
76 "<save_file> save the intermediate representation into file.srx";
78 "-b", Arg.Set(bottom_up), " real bottom up run";
80 "-nj", Arg.Set(no_jump), " disable jumping";
82 "-p", Arg.Set(do_perf), " dump perf counters (Linux only)";
84 "-index-type", Arg.Symbol ([ "default"; "swcsa"; "rlcsa" ],
86 " choose text index type";
88 "-v", Arg.Set(verbose), " verbose mode"; ] @
91 "-trace-file", Arg.Set_string(trace_file),
92 "<trace_file> save the full trace in dot format in <trace_file>";
93 "-trace", Arg.String (set_tracer),
94 "<tracer1:l1,...,tracern:ln> enable tracing with the specified level. Valid tracers are: "
101 let parse_cmdline() =
102 let _ = Arg.parse spec anon_fun usage_msg
104 if (!pos > 3 || !pos < 2)
105 then begin Arg.usage spec usage_msg; exit 1 end