Supprimer des fonctions 'keep_elements' et 'keep_attributs'
[tatoo.git] / src / query_tree.ml
index a1b47a6..1f58ba6 100644 (file)
@@ -1,12 +1,18 @@
 open Table
 
-
+let compteur = ref 0
+let table_query_tree = Hashtbl.create 97 
+let table_qtree = QTreeHash.create 97
 
 let all_nodes tree = let root = Naive_tree.root tree in
                     eval_axis tree [root] (Descendant true)
 
-let element_by_tag tree tagset = let dom = all_nodes tree in
-                             List.filter (fun c -> QNameSet.mem (Naive_tree.tag tree c) tagset ) dom
+let element_by_tag tree tagset kind = let dom = all_nodes tree in
+                             List.filter (fun c ->
+                               Tree.NodeKind.is_a (Naive_tree.kind tree c) kind &&
+                               QNameSet.mem (Naive_tree.tag tree c) tagset ) dom
+
+let mk_node q = {desc = q; id = -1; hash = -1}
 
 let rec compile_single_path p =
   let open Xpath.Ast in
@@ -15,24 +21,24 @@ let rec compile_single_path p =
       
 and compile_step_list p = 
     match p with
-      | [] -> Start
-      | (a,(test,_),el) :: r ->
+      | [] -> mk_node Start
+      | (a,(test,kind),el) :: r ->
        let qtree = compile_step_list r in
-       let res = Binop ( Inter,Axis (a,qtree), Tag (test) ) in
+       let res = mk_node ( Binop ( Inter,mk_node( Axis (a,qtree)),mk_node (Tag (test,kind) )) ) in
        List.fold_left (fun acc e ->
-         Binop(Inter, acc, compile_expr e)) res el  (*avant j'ai utilise une function compile_expr_list ,c'est pas genial*)
+         mk_node (Binop(Inter, acc, compile_expr e))) res el  
 
   and compile_expr  (e : Xpath.Ast.expr )  = match e with
     | Fun_call (f, [ e0 ]) when (QName.to_string f) = "not" ->
       let qtree = compile_expr e0 in
-      Binop (Diff , Dom, qtree)
+      mk_node (Binop (Diff , mk_node (Dom), qtree))
 
     | Binop (e1,op,e2) -> let qtree1 = compile_expr e1 in
                          let qtree2 = compile_expr e2 in 
                          begin
                            match op with 
-                             | Or -> Binop (Union , qtree1,qtree2)
-                             | And -> Binop (Inter ,qtree1,qtree2)
+                             | Or -> mk_node (Binop (Union , qtree1,qtree2))
+                             | And -> mk_node (Binop (Inter ,qtree1,qtree2))
                              | _ -> failwith "Unknown operator"
                          end
     | Path p -> compile_path_rev p
@@ -42,28 +48,28 @@ and compile_step_list p =
     match p with
       | [] -> assert false
       | [p] -> compile_single_path_rev p  
-      | p::r -> List.fold_left (fun acc p -> Binop (Union , acc, compile_single_path_rev p) ) (compile_single_path_rev p) r
+      | p::r -> List.fold_left (fun acc p -> mk_node (Binop (Union , acc, compile_single_path_rev p)) ) (compile_single_path_rev p) r
        
   and compile_single_path_rev p = 
     match p with
-      | Absolute p | Relative p -> compile_step_list_rev (List.rev p)
+      | Absolute p | Relative p -> compile_step_list_rev 
 
   and compile_step_list_rev p = match p with
-    | [] -> Dom         (*assert false*) (*on fait rien , mais comment signifer ???*)
-    | (a,(test,_),el) :: r -> 
+    | [] -> mk_node Dom        
+    | (a,(test,kind),el) :: r -> 
       let qtree = compile_step_list_rev r in
-      let res = Binop (Inter , qtree, Tag(test)) in
+      let res = mk_node (Binop (Inter , qtree,mk_node (Tag(test,kind)))) in
       let qtree2 = List.fold_left (fun acc e ->
-       Binop(Inter, acc, compile_expr e)) res el in
+       mk_node (Binop(Inter, acc, compile_expr e))) res el in
       let a_rev = axis_rev a in
-      Axis (a_rev , qtree2)
+      mk_node (Axis (a_rev , qtree2))
                                                
       
     and axis_rev a =
       let open Xpath.Ast in
          match a with
              Self -> Self
-           | Attribute -> assert false
+           | Attribute -> Parent
            | Child -> Parent
            | Descendant b -> 
              if not b then (Ancestor false)
@@ -81,7 +87,7 @@ and compile_step_list p =
 let compile_xpath p = match p with
   | [] -> assert false
   | [p] -> compile_single_path p
-  | p::r -> List.fold_left (fun acc p -> Binop (Union , acc, compile_single_path p) ) (compile_single_path p) r
+  | p::r -> List.fold_left (fun acc p -> mk_node (Binop (Union , acc, compile_single_path p) )) (compile_single_path p) r
 
 let comp_node t n1 n2 = (Naive_tree.preorder t n1) < (Naive_tree.preorder t n2)
 
@@ -111,38 +117,131 @@ let rec diff_list t l1 l2 =
       else diff_list t ll1 ll2
 
 
-let do_debug = ref true
+let do_debug = ref false
 
 let debug tree q l =
   if !do_debug then begin
     Format.fprintf Format.std_formatter "Evaluation de: ";
     print_query_tree Format.std_formatter q;
-    Format.fprintf Format.std_formatter "\nResultat: %i"
+    Format.fprintf Format.std_formatter "\nResultat: %i\n"
     (List.length l);
     Format.pp_print_flush Format.std_formatter ();
     print_node_list tree l;
+    (*List.iter
+      (fun n -> Format.fprintf Format.std_formatter "%i, " (Naive_tree.preorder tree n)) l;*)
     Format.fprintf Format.std_formatter "\n----------------\n";
     Format.pp_print_flush Format.std_formatter ();
   end
 
 
+
+
+let rec compare_query_tree q1 q2 =
+     q1.id==q2.id|| match q1.desc,q2.desc with
+     | Binop(op1,qt1,qt2),Binop(op2,qt3,qt4)->op1==op2&& ((compare_query_tree qt1 qt3 && compare_query_tree qt2 qt4) 
+                                                         ||  (compare_query_tree qt1 qt4 && compare_query_tree qt2 qt3))
+     | Axis(a1,qt1),Axis(a2,qt2) -> compare_axis a1 a2 && compare_query_tree qt1 qt2
+     | Tag(t1,k1),Tag(t2,k2) ->t1==t2&& k1==k2
+     | Dom,Dom | Start,Start ->true
+     | _,_ ->false
+
+and compare_axis a1 a2 =
+  match a1,a2 with
+      Self ,Self | Attribute, Attribute | Child , Child | Parent , Parent
+    | FollowingSibling , FollowingSibling         
+    | PrecedingSibling , PrecedingSibling
+    | Preceding , Preceding | Following , Following -> true
+    | Descendant b1, Descendant b2 -> b1==b2 
+    | Ancestor b1, Ancestor b2 -> b1==b2
+    | _,_ -> false
+
+
 let rec eval_query_tree tree start q =
   let resultat = 
-    match q with
-      | Start ->  start
-      | Dom -> all_nodes tree
-      | Tag t -> element_by_tag tree t
-      | Axis (a,q1) -> let ls = eval_query_tree tree start q1 in
-                      eval_axis tree ls a
-      | Binop (op,q1,q2)-> begin
-       let ls1 = eval_query_tree tree start q1 in
-       let ls2 = eval_query_tree tree start q2 in
-       match op with     
-         | Union -> union_list tree ls1 ls2        
-         | Inter -> inter_list tree ls1 ls2
-         | Diff -> diff_list tree ls1 ls2
-      end
+    begin
+      try
+       Hashtbl.find table_query_tree q
+      with Not_found ->        
+       let res =
+       match q.desc with
+         | Start -> start
+         | Dom ->  all_nodes tree          
+         | Tag (t,k) -> element_by_tag tree t k                     
+         | Axis (a,q1) -> let ls = eval_query_tree tree start q1 in
+                          eval_axis tree ls a                     
+         | Binop (op,q1,q2)-> begin
+           let ls1 = eval_query_tree tree start q1 in
+           let ls2 = eval_query_tree tree start q2 in
+           match op with         
+             | Union -> union_list tree ls1 ls2            
+             | Inter -> inter_list tree ls1 ls2
+             | Diff -> diff_list tree ls1 ls2
+         end
+       in
+       let res = List.sort (Table.compare_node tree) res in
+       Hashtbl.add table_query_tree q res;
+       compteur := !compteur + (List.length res);
+       res
+    end
+  in
+  debug tree q resultat;
+  resultat
+
+let mini_id = ref 0
+let mini_table = QTreeHash.create 17 
+
+let rec minimize_qtree q =
+   if q.id != -1 then q
+   else
+    try
+      QTreeHash.find mini_table q
+    with Not_found ->
+      let mdesc =
+       match q.desc with
+           (Start | Dom | Tag _) as d -> d
+         | Binop(op,q1,q2) -> let mq1 = minimize_qtree q1 in
+                              let mq2 = minimize_qtree q2 in
+                              Binop(op,mq1,mq2)
+         | Axis(a,q1) -> let mq1 = minimize_qtree q1 in
+                         Axis(a,mq1)         
+      in
+      q.desc <- mdesc;
+      q.hash <- QTree.hash q;
+      q.id <- !mini_id;
+      incr mini_id;
+      QTreeHash.add mini_table q q;
+      q
+
+
+         
+let rec eval_qtree tree start q =
+  let resultat = 
+    begin
+      try
+       QTreeHash.find table_qtree q
+      with Not_found ->        
+       let res =
+       match q.desc with
+         | Start -> start
+         | Dom ->  all_nodes tree          
+         | Tag (t,k) -> element_by_tag tree t k                     
+         | Axis (a,q1) -> let ls = eval_qtree tree start q1 in
+                          eval_axis tree ls a                     
+         | Binop (op,q1,q2)-> begin
+           let ls1 = eval_qtree tree start q1 in
+           let ls2 = eval_qtree tree start q2 in
+           match op with         
+             | Union -> union_list tree ls1 ls2            
+             | Inter -> inter_list tree ls1 ls2
+             | Diff -> diff_list tree ls1 ls2
+         end
+       in
+       let res = Tas.sort_of_list tree res in
+(*     let res = List.sort (Table.compare_node tree) res in*)
+       QTreeHash.add table_qtree q res;
+       compteur := !compteur + (List.length res);
+       res
+    end
   in
   debug tree q resultat;
   resultat
-