X-Git-Url: http://git.nguyen.vg/gitweb/?a=blobdiff_plain;f=src%2Fquery_tree.ml;h=a81804792e063f7f381c816cef6e2e8b542238fa;hb=11fd750e5f19e421768ec72c511b631e7b009365;hp=a1b47a67beecfc95fb98988277dbb5e097150eed;hpb=f3a0235e4715d24d8e8b4053923d433e1d876851;p=tatoo.git diff --git a/src/query_tree.ml b/src/query_tree.ml index a1b47a6..a818047 100644 --- a/src/query_tree.ml +++ b/src/query_tree.ml @@ -1,12 +1,19 @@ 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 acc1 = get_descendant tree [root] in + root::acc1 + +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 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 mk_node q = {desc = q; id = -1; hash = -1} let rec compile_single_path p = let open Xpath.Ast in @@ -15,24 +22,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 +49,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 p 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,68 +88,102 @@ 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 - -let comp_node t n1 n2 = (Naive_tree.preorder t n1) < (Naive_tree.preorder t n2) - + | p::r -> List.fold_left (fun acc p -> mk_node (Binop (Union , acc, compile_single_path p) )) (compile_single_path p) r -let rec union_list t l1 l2 = - match l1,l2 with - | [],l2 -> l2 - | l1, [] -> l1 - | h1::ll1, h2::ll2 -> if (comp_node t h2 h1) then h2 :: (union_list t l1 ll2) - else if (comp_node t h1 h2) then h1::(union_list t ll1 l2) - else h1 ::(union_list t ll1 ll2) -let rec inter_list t l1 l2 = - match l1,l2 with - | [],l2 -> [] - | l1, [] -> [] - | h1::ll1, h2::ll2 -> if (comp_node t h1 h2) then inter_list t ll1 l2 - else if (comp_node t h2 h1) then inter_list t l1 ll2 - else h1 :: (inter_list t ll1 ll2) -let rec diff_list t l1 l2 = - match l1,l2 with - | [],l2 -> [] - | l1, [] -> l1 - | h1::ll1, h2::ll2 -> if (comp_node t h1 h2) then h1::(diff_list t ll1 l2) - else if (comp_node t h2 h1) then h2 :: (diff_list t l1 ll2) - 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 eval_query_tree tree start q = + + +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 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 = - 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 + 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 + QTreeHash.add table_qtree q res; + compteur := !compteur + (List.length res); + res + end in debug tree q resultat; resultat -