open Table

let compteur = ref 0
let table_qtree = QTreeHash.create 97

let all_nodes tree = let root = Naive_tree.root tree in
		     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 mk_node q = {desc = q; id = -1; hash = -1}

let rec compile_single_path p =
  let open Xpath.Ast in
      match p with
	| Absolute p | Relative p -> compile_step_list (List.rev p)
      
and compile_step_list p = 
    match p with
      | [] -> mk_node Start
      | (a,(test,kind),el) :: r ->
	let qtree = compile_step_list r in
	let res = mk_node ( Binop ( Inter,mk_node( Axis (a,qtree)),mk_node (Tag (test,kind) )) ) in
	List.fold_left (fun acc e ->
	  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
      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 -> mk_node (Binop (Union , qtree1,qtree2))
			      | And -> mk_node (Binop (Inter ,qtree1,qtree2))
			      | _ -> failwith "Unknown operator"
			  end
    | Path p -> compile_path_rev p
    | _ -> failwith "Unknown expression"
      
  and compile_path_rev p = 
    match p with
      | [] -> assert false
      | [p] -> compile_single_path_rev p  
      | 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 p 

  and compile_step_list_rev p = match p with
    | [] -> mk_node Dom        
    | (a,(test,kind),el) :: r -> 
      let qtree = compile_step_list_rev r in
      let res = mk_node (Binop (Inter , qtree,mk_node (Tag(test,kind)))) in
      let qtree2 = List.fold_left (fun acc e ->
	mk_node (Binop(Inter, acc, compile_expr e))) res el in
      let a_rev = axis_rev a in
      mk_node (Axis (a_rev , qtree2))
						
      
    and axis_rev a =
      let open Xpath.Ast in
	  match a with
	      Self -> Self
	    | Attribute -> Parent
	    | Child -> Parent
	    | Descendant b -> 
	      if not b then (Ancestor false)
	      else (Ancestor true)    (* true = descendant-or-self, false = descendant *)
	    | FollowingSibling -> PrecedingSibling
	    | Parent -> Child
	    | Ancestor b -> 
	      if not b then (Descendant false) 
	      else (Descendant true)  (* true = ancestor-or-self, false = ancestor *)
	    | PrecedingSibling -> FollowingSibling
	    | Preceding -> Following
	    | Following -> Preceding
	    
	    
let compile_xpath p = match p with
  | [] -> assert false
  | [p] -> compile_single_path p
  | p::r -> List.fold_left (fun acc p -> mk_node (Binop (Union , acc, compile_single_path p) )) (compile_single_path p) r




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\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 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
	QTreeHash.add table_qtree q res;
	compteur := !compteur + (List.length res);
	res
    end
  in
  debug tree q resultat;
  resultat