10 val var : (int*State.t) -> t
11 val close : ((int*State.t), t) Hashtbl.t -> t -> t
12 val is_open : t -> bool
13 val singleton : elt -> t
14 val cons : elt -> t -> t
15 val snoc : t -> elt -> t
16 val concat : t -> t -> t
17 val concat3 : t -> t -> t -> t
18 val concat4 : t -> t -> t -> t -> t
19 val conscat : elt -> t -> t -> t
20 val conscat3 : elt -> t -> t -> t -> t
21 val conscat4 : elt -> t -> t -> t -> t -> t
22 val subtree_tags : Tree.t -> elt -> Tag.t -> t
23 val subtree_elements : Tree.t -> elt -> t
24 val iter : ( elt -> unit) -> t -> unit
25 val fold : ( elt -> 'a -> 'a) -> t -> 'a -> 'a
27 val serialize : string -> Tree.t -> t -> unit
31 module Count : S with type t = int =
43 let concat x y = x + y
44 let concat3 x y z = x + y + z
45 let concat4 x y z t = x + y + z + t
46 let conscat _ x y = 1 + x + y
47 let conscat3 _ x y z = 1 + x + y + z
48 let conscat4 _ x y z t = 1 + x + y + z + t
49 let subtree_tags tree node tag = Tree.subtree_tags tree node tag
50 let subtree_elements tree node = Tree.subtree_elements tree node
51 let iter _ _ = failwith "iter not implemented"
52 let fold _ _ _ = failwith "fold not implemented"
53 let map _ _ = failwith "map not implemented"
55 let serialize _ _ _ = ()
60 | Cons of Tree.node * clist
61 | Concat of clist * clist
62 | ConsCat of Tree.node * clist * clist
63 | SubtreeTags of Tree.t * Tree.node * Tag.t
64 | SubtreeElts of Tree.t * Tree.node
67 type 'a mat = { mutable clist : clist;
68 mutable length : int }
70 module Mat : S with type t = Tree.node mat =
72 type t = Tree.node mat
75 let empty = { clist = Nil; length = 0 }
76 let singleton e = { clist = Cons(e, Nil) ; length = 1 }
77 let cons e l = { clist = Cons(e, l.clist); length = l.length + 1 }
79 let ll1 = l1.length in
80 if ll1 == 0 then l2 else
81 let ll2 = l2.length in if ll2 == 0 then l1 else
82 { clist = Concat(l1.clist, l2.clist); length = ll1 + ll2 }
84 let snoc l e = concat l (singleton e)
85 let concat3 l1 l2 l3 = concat l1 (concat l2 l3)
86 let concat4 l1 l2 l3 l4 = concat (concat l1 l2) (concat l3 l4)
92 let ll1 = l1.length in
93 if ll1 == 0 then cons e l2 else
94 let ll2 = l2.length in if ll2 == 0 then cons e l1 else
95 { clist = ConsCat(e, l1.clist, l2.clist); length = 1 + ll1 + ll2 }
97 (* let conscat e l1 l2 = cons e (concat l1 l2) *)
99 let conscat3 e l1 l2 l3 = conscat e l1 (concat l2 l3)
100 let conscat4 e l1 l2 l3 l4 = conscat e l1 (concat l2 (concat l3 l4))
102 let subtree_tags tree node tag =
103 let len = Tree.subtree_tags tree node tag in
104 if len == 0 then empty
106 { clist = SubtreeTags(tree, node, tag);
109 let subtree_elements tree node =
110 let len = Tree.subtree_elements tree node in
111 if len == 0 then empty
113 { clist = SubtreeElts(tree, node);
116 let fst_tagged tree t tag =
117 if Tree.tag tree t == tag then t
118 else Tree.tagged_descendant tree t tag
120 let fst_element tree t =
121 let tag = Tree.tag tree t in
122 let t = if Ptset.Int.mem tag
123 (Ptset.Int.remove Tag.document_node (Tree.element_tags tree))
125 else Tree.first_element tree t
126 in Tree.first_element tree t
128 let element_fold f tree t acc =
129 let rec loop node acc =
130 if node == Tree.nil then acc
132 let acc = f node acc in
133 let acc' = loop (Tree.first_element tree node) acc in
134 loop (Tree.next_element tree node) acc'
136 loop (fst_element tree t) acc
138 let element_iter f tree t =
140 if node != Tree.nil then begin
142 loop (Tree.first_element tree node);
143 loop (Tree.next_element tree node)
146 let t' = fst_element tree t in loop t'
148 let tag_fold f tree t tag acc =
149 let rec loop close node acc =
150 if node > Tree.nil && node < close then acc
152 let acc = f node acc in
153 loop close (Tree.tagged_next tree node tag) acc
155 let t' = fst_tagged tree t tag in
156 loop (Tree.closing tree t) t' acc
158 let tag_iter f tree t tag =
159 let rec loop close node =
160 if node > Tree.nil && node < close then begin
162 loop close (Tree.tagged_next tree node tag);
165 let t' = fst_tagged tree t tag in
166 loop (Tree.closing tree t) t'
172 | Cons(e, ll) -> loop ll (f e acc)
173 | Concat(l1, l2) -> loop l2 (loop l1 acc)
174 | ConsCat(e, l1, l2) -> loop l2 (loop l1 (f e acc))
175 | SubtreeTags(tree, t, tag) -> tag_fold f tree t tag acc
176 | SubtreeElts(tree, t) -> element_fold f tree t acc
184 | Cons(e, l) -> f e; loop l
185 | Concat(l1, l2) -> loop l1; loop l2
186 | ConsCat(e, l1, l2) -> f e; loop l1; loop l2
187 | SubtreeTags(tree, t, tag) -> tag_iter f tree t tag
188 | SubtreeElts(tree, t) ->
189 element_iter f tree t
193 let length l = l.length
195 let serialize name v l =
197 if name = "-" then Unix.stdout, ignore
199 Unix.openfile name [ Unix.O_WRONLY; Unix.O_TRUNC; Unix.O_CREAT ] 0o666,
202 ignore (Unix.write fd "<xml_result>\n" 0 13);
203 iter (fun node -> Tree.print_xml v node fd) l;
205 ignore (Unix.write fd "</xml_result>\n" 0 14);
209 let rec debug_clist =
211 Nil -> Printf.eprintf "Nil"
213 Printf.eprintf "Cons(%i," (Obj.magic e);
216 | Concat(clist1, clist2) ->
217 Printf.eprintf "Concat(";
222 | ConsCat(_, clist1, clist2) ->
223 Printf.eprintf "Concat(";
229 | SubtreeTags(tree, node, tag) ->
230 Printf.eprintf "SubtreeTags(tree, %i, %s)"
233 | SubtreeElts(tree, node) ->
234 Printf.eprintf "SubtreeElts(tree, %i)"
237 let debug l = debug_clist l.clist
241 module Partial(N : S) : S =
245 type t = { env : ((int*State.t), t) Hashtbl.t;
250 | Var of (int * State.t)
253 | Concat of list * list
256 let dummy = Hashtbl.create 0
257 let empty = { env = dummy;
260 let is_open t = t.opened
265 Lambda { t with env = h; opened = false } }
267 let singleton i = { empty with elem = Cons(i, Nil) }
268 let cons e t = { t with elem = Cons(e, t.elem) }
270 { t1 with elem = Concat (t1.elem, t2.elem) }
272 let snoc t e = concat t (singleton e)
273 let concat3 t1 t2 t3 = concat t1 (concat t2 t3)
274 let concat4 t1 t2 t3 t4 = concat (concat t1 t2) (concat t3 t4)
275 let conscat e t1 t2 = cons e (concat t1 t2)
276 let conscat3 e t1 t2 t3 = cons e (concat3 t1 t2 t3)
277 let conscat4 e t1 t2 t3 t4 = cons e (concat4 t1 t2 t3 t4)
278 let subtree_tags _ = failwith "not implemented"
279 let subtree_elements _ = failwith "not_implemented"
283 loop_list t.env t.elem
284 and loop_list h = function
286 | Var i -> loop (Hashtbl.find h i)
287 | Cons (e, l) -> f e; loop_list h l
288 | Concat (l1, l2) -> loop_list h l1; loop_list h l2
295 loop_list t.env acc t.elem
296 and loop_list h acc = function
298 | Var i -> loop (try Hashtbl.find h i with Not_found -> let a,b = i in Printf.eprintf "%i,%i not found\n%!" a b; empty) acc
299 | Cons (e, l) -> loop_list h (f e acc) l
300 | Concat (l1, l2) -> loop_list h (loop_list h acc l1) l2
301 | Lambda t -> loop t acc
307 Hashtbl.iter (fun (i,j) t ->
308 Format.eprintf "%i, %a ->" i State.print j;
310 Format.eprintf "----------------\n%!";
313 and dump_list = function
315 | Var (i,j) -> Format.eprintf "Var(%i, %a) " i State.print j;
316 | Cons (e, l) -> Format.eprintf "%i " (Node.to_int e); dump_list l
317 | Concat (l1, l2) -> dump_list l1 ; dump_list l2
321 let length t = fold (fun _ acc -> 1 + acc) t 0
325 { empty with elem = Var i; opened = true }
327 let serialize _ = failwith "not implemented"