Added debugging messages

[SXSI/xpathcomp.git] / main.ml

diff --git a/main.ml b/main.ml
index ce37dfc..7418fa1 100644 (file)
--- a/main.ml
+++ b/main.ml
@@ -4,42 +4,37 @@
 (*  Copyright NICTA 2008                                                      *)
 (*  Distributed under the terms of the LGPL (see LICENCE)                     *)
 (******************************************************************************)
-INCLUDE "debug.ml"
 
 open Ata
+INCLUDE "utils.ml"
+let () = init_timer();;
 
-let l = ref [] ;;
-let time f x =
-  let t1 = Unix.gettimeofday () in
-  let r = f x in
-  let t2 = Unix.gettimeofday () in 
-  let t = (1000. *. (t2 -. t1)) in
-    l:= t::!l;
-    Printf.eprintf "  %fms\n%!" t ;
-    r
-;;
-let total_time () =  List.fold_left (+.) 0. !l;;
 let enabled_gc = Gc.get()
 let disabled_gc = { Gc.get() with
                      Gc.max_overhead = 1000000; 
                      Gc.space_overhead = 100 }
 
-let main v query output =
+
+
+
+
+let main v query_string output =
+  
     let _ = Tag.init (Tree.tag_pool v) in
       Printf.eprintf "Parsing query : ";    
       let query = try
        time
-         XPath.Parser.parse_string query
+         XPath.Parser.parse_string query_string
       with
          Ulexer.Loc.Exc_located ((x,y),e) -> Printf.eprintf "character %i-%i %s\n" x y (Printexc.to_string e);exit 1
       in
        XPath.Ast.print Format.err_formatter query;
        Format.fprintf Format.err_formatter "\n%!";
        Printf.eprintf "Compiling query : ";
-       let auto,ltags,contains = time XPath.Compile.compile  query in 
+       let auto,ltags,contains = time (XPath.Compile.compile ~querystring:query_string) query in 
        let _ = Ata.dump Format.err_formatter auto in
        let _ = Printf.eprintf "%!" in
-       let jump_to =
+       let jump_to = 
          match contains with
              None -> (max_int,`NOTHING)
            | Some s -> 
@@ -56,18 +51,20 @@ let main v query output =
                    time (Tree.init_naive_contains v) s
                  end;(r,`CONTAINS(s))
        in
+       let test_list = jump_to in
+       (*
        let test_list = 
          if (!Options.backward) then begin
            Printf.eprintf "Finding min occurences : ";
            time 
              ( List.fold_left (fun ((min_occ,kind)as acc)  (tag,_) ->
-                             let numtags = Tree.subtree_tags v tag in
+                             let numtags = Tree.subtree_tags v tag Tree.root in
                                if  ((numtags < min_occ) && numtags >= 2)
                                then (numtags,`TAG(tag))
                                else acc) jump_to) ltags
          end
          else (max_int,`NOTHING)
-       in
+       in*)
        let _ = if (snd test_list) != `NOTHING then
          let occ,s1,s2 = match test_list with
            | (x,`TAG (tag)) -> (x, "tag", (Tag.to_string tag))
@@ -80,6 +77,7 @@ let main v query output =
            (if !Options.count_only then "(counting only)" else if !Options.backward then "(bottomup)" else "");
          begin
            let _ = Gc.full_major();Gc.compact() in
+           let _ = Printf.eprintf "%!" in
              (*            let _ = Gc.set (disabled_gc) in  *)
              if !Options.backward && ((snd test_list) != `NOTHING )then 
                
@@ -96,7 +94,7 @@ let main v query output =
                  in ()
                else      
                  let result = time (top_down auto) v in          
-                 let rcount = IdSet.length result in
+                 let rcount = GResult.length result in
                    Printf.eprintf "Number of nodes in the result set : %i\n" rcount;
                    Printf.eprintf "\n%!";
                    begin
@@ -107,10 +105,10 @@ let main v query output =
                            time( fun () ->
                                    let oc = open_out f in
                                      output_string oc "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n";                          
-                                     IdSet.iter (fun t -> 
-                                                   Tree.print_xml_fast oc t;
+                                     GResult.iter (fun t -> 
+                                                   Tree.print_xml_fast oc v t;
                                                    output_char oc '\n';
-                                                   output_string oc "----------\n";
+
                                                 ) result) ();
                    end;
          end;
@@ -120,13 +118,13 @@ let main v query output =
 
 Options.parse_cmdline();;
 
-let v = 
+let v =
   if (Filename.check_suffix !Options.input_file ".srx")
   then 
     begin
       Printf.eprintf "Loading from file : ";
       time (Tree.load  ~sample:!Options.sample_factor )
-    (Filename.chop_suffix !Options.input_file ".srx");
+       !Options.input_file;
        end
   else 
     let v = 
@@ -143,10 +141,3 @@ let v =
 in
   main v !Options.query !Options.output_file;;
 
-IFDEF DEBUG
-THEN
-Printf.eprintf "\n=================================================\nDEBUGGING\n%!";
-
-Tree.DEBUGTREE.print_stats Format.err_formatter;;
-Gc.full_major()
-ENDIF