type value = [ `NodeSet of Automaton.BST.t
| `Int of int | `String of string
| `Bool of bool | `True | `False ]
+
type expr = [ value | `Call of (string*(expr list))
- | `Auto of Automaton.t ]
+ | `Auto of Automaton.t | `Contains of expr list ]
let count = function [`NodeSet(s) ] -> `Int(Automaton.BST.cardinal s)
| _ -> failwith "count"
-
+ let contains_old = function [`NodeSet(s) ; `String(str) ] ->
+ `Bool(Automaton.BST.exists (fun tree -> Tree.Binary.contains_old tree str
+ ) s)
+ | _ -> failwith "contains_old"
+
let equal = function [ `Int i; `Int j ] -> `Bool (i == j)
|_ -> failwith "equal"
("count",count);
("equal",equal);
+ ("contains_old",contains_old);
]
let text t = Tree.Binary.string (Tree.Binary.left t)
+
+
let rec eval_expr tree (e:expr) : value = match e with
| `Call (f,args) -> (List.assoc f globals) (List.map (eval_expr tree) args)
- | `Auto(a) -> `NodeSet(ignore (Automaton.BottomUp.accept a tree);a.Automaton.result)
+ | `Auto(a) -> `NodeSet(ignore (Automaton.BottomUp.accept a tree);
+ a.Automaton.result)
+ | `Contains(args) ->
+ begin
+ match args with
+ [ `Auto(a); `String(s) ] ->
+ let docs = try
+ Hashtbl.find a.Automaton.contains s
+ with
+ | Not_found ->
+ let r = Tree.Binary.contains tree s
+ in
+ (* Tree.Binary.DocIdSet.iter (fun id ->
+ Printf.eprintf "%s matches %s\n%!" (Tree.Binary.get_string tree id) s) r; *)
+
+ Hashtbl.add a.Automaton.contains s r;r
+ in
+ let _ = Automaton.BottomUp.accept ~strings:(Some docs) a tree
+ in `NodeSet(a.Automaton.result)
+ | _ -> failwith "contains invalid"
+ end
| #value as x -> x
let truth_value =
- function `NodeSet s -> Automaton.BST.is_empty s
+ function `NodeSet s -> not (Automaton.BST.is_empty s)
|`Bool(b) -> b
| _ -> failwith "truth_value"
let rec map_dir (d,acc) = function
| [] -> acc
| s::r -> map_dir ((dir s),(s,d)::acc) r
- in let l = match p with
- | Absolute p | Relative p -> map_dir (Final,[]) p
- | AbsoluteDoS p ->
- let l = (map_dir (Final,[]) p)
- in ((DescendantOrSelf,TagSet.Xml.node,Expr True),dir (fst(List.hd l)))::l
- in ((Child,TagSet.Xml.node,Expr True),dir (fst(List.hd l)))::l
-
+ in
+ let l =
+ match p with
+ | Absolute p
+ | Relative p -> map_dir (Final,[]) p
+ | AbsoluteDoS p ->
+ let l = (map_dir (Final,[]) p)
+ in ((DescendantOrSelf,TagSet.Xml.node,Expr True),dir (fst(List.hd l)))::l
+ in ((Child,TagSet.Xml.node,Expr True),dir (fst(List.hd l)))::l
+
let rec compile_step q dir trs final initial ignore (axis,test,pred) =
let q' = State.mk() in
(* q' is not returned and thus not added to the set of final states.
It's ok since we should never be in a final state on a node
<@> *)
+
| Attribute,ts -> let q'' = State.mk() in
(mk_tag_t Left q (TagSet.Xml.attribute) q' ignore)::
(mk_tag_t Left q' (ts) q'' ignore)::( q==> any @@ (ignore,q))::trs, q''::q'::final,initial
| Path p -> `Auto(compile p)
| Int i -> `Int i
| String s -> `String s
+ | Function ("contains",elist) ->`Contains(List.map compile_expr elist)
| Function (f,elist) -> `Call(f,List.map compile_expr elist)
and cup a b = match a,b with