]> matita.cs.unibo.it Git - helm.git/blob - helm/ocaml/cic/cicUniv.ml
Universes speedup:
[helm.git] / helm / ocaml / cic / cicUniv.ml
1 (* Copyright (C) 2000, HELM Team.
2  * 
3  * This file is part of HELM, an Hypertextual, Electronic
4  * Library of Mathematics, developed at the Computer Science
5  * Department, University of Bologna, Italy.
6  * 
7  * HELM is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License
9  * as published by the Free Software Foundation; either version 2
10  * of the License, or (at your option) any later version.
11  * 
12  * HELM is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with HELM; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
20  * MA  02111-1307, USA.
21  * 
22  * For details, see the HELM World-Wide-Web page,
23  * http://cs.unibo.it/helm/.
24  *)
25
26 (*****************************************************************************)
27 (*                                                                           *)
28 (*                               PROJECT HELM                                *)
29 (*                                                                           *)
30 (*                      Enrico Tassi <tassi@cs.unibo.it>                     *)
31 (*                                 23/04/2004                                *)
32 (*                                                                           *)
33 (* This module implements the aciclic graph of universes.                    *)
34 (*                                                                           *)
35 (*****************************************************************************)
36
37 (* $Id$ *)
38
39 (*****************************************************************************)
40 (** switch implementation                                                   **)
41 (*****************************************************************************)
42
43 let fast_implementation = ref true ;;
44
45 (*****************************************************************************)
46 (** open                                                                    **)
47 (*****************************************************************************)
48
49 open Printf
50
51 (*****************************************************************************)
52 (** Types and default values                                                **)
53 (*****************************************************************************)
54
55 type universe = int * UriManager.uri option 
56     
57 module UniverseType = struct
58   type t = universe
59   let compare = Pervasives.compare
60 end
61   
62 module SOF = Set.Make(UniverseType)
63   
64 type entry = {
65   eq_closure : SOF.t;
66   ge_closure : SOF.t;
67   gt_closure : SOF.t;
68   in_gegt_of   : SOF.t;
69   one_s_eq   : SOF.t;
70   one_s_ge   : SOF.t;
71   one_s_gt   : SOF.t;
72 }
73     
74 module MAL = Map.Make(UniverseType)
75   
76 type arc_type = GE | GT | EQ
77     
78 type bag = entry MAL.t 
79     
80 let empty_entry = {
81   eq_closure=SOF.empty;
82   ge_closure=SOF.empty;
83   gt_closure=SOF.empty;
84   in_gegt_of=SOF.empty;
85   one_s_eq=SOF.empty;
86   one_s_ge=SOF.empty;
87   one_s_gt=SOF.empty;
88 }
89 let empty_bag = MAL.empty
90
91 let are_set_eq s1 s2 = 
92   SOF.equal s1 s2
93
94 let are_entry_eq v1 v2 =
95   (are_set_eq v1.gt_closure v2.gt_closure ) &&
96   (are_set_eq v1.ge_closure v2.ge_closure ) &&
97   (are_set_eq v1.eq_closure v2.eq_closure ) &&
98   (*(are_set_eq v1.in_gegt_of v2.in_gegt_of ) &&*)
99   (are_set_eq v1.one_s_ge v2.one_s_ge ) &&
100   (are_set_eq v1.one_s_gt v2.one_s_gt ) &&
101   (are_set_eq v1.one_s_eq v2.one_s_eq )
102
103 let are_ugraph_eq = MAL.equal are_entry_eq
104
105 (*****************************************************************************)
106 (** Pretty printings                                                        **)
107 (*****************************************************************************)
108
109 let string_of_universe (i,u) = 
110   match u with
111       Some u ->
112         "(" ^ ((string_of_int i) ^ "," ^ (UriManager.string_of_uri u) ^ ")")
113     | None -> "(" ^ (string_of_int i) ^ ",None)"
114
115 let string_of_universe_set l = 
116   SOF.fold (fun x s -> s ^ (string_of_universe x) ^ " ") l ""
117
118 let string_of_node n =
119   "{"^
120   "eq_c: " ^ (string_of_universe_set n.eq_closure) ^ "; " ^ 
121   "ge_c: " ^ (string_of_universe_set n.ge_closure) ^ "; " ^ 
122   "gt_c: " ^ (string_of_universe_set n.gt_closure) ^ "; " ^ 
123   "i_gegt: " ^ (string_of_universe_set n.in_gegt_of) ^ "}\n"
124
125 let string_of_arc (a,u,v) = 
126   (string_of_universe u) ^ " " ^ a ^ " " ^ (string_of_universe v)
127   
128 let string_of_mal m =
129   let rc = ref "" in
130   MAL.iter (fun k v ->  
131     rc := !rc ^ sprintf "%s --> %s" (string_of_universe k) 
132               (string_of_node v)) m;
133   !rc
134
135 let string_of_bag b = 
136   string_of_mal b
137
138 (*****************************************************************************)
139 (** Helpers                                                                 **)
140 (*****************************************************************************)
141
142 (* find the repr *)
143 let repr u m =
144   try 
145     MAL.find u m
146   with
147     Not_found -> empty_entry
148     
149 (* FIXME: May be faster if we make it by hand *)
150 let merge_closures f nodes m =  
151   SOF.fold (fun x i -> SOF.union (f (repr x m)) i ) nodes SOF.empty
152
153 (*****************************************************************************)
154 (** Benchmarking                                                            **)
155 (*****************************************************************************)
156 let time_spent = ref 0.0;;
157 let partial = ref 0.0 ;;
158
159 let reset_spent_time () = time_spent := 0.0;;
160 let get_spent_time () = !time_spent ;;
161 let begin_spending () =
162   (*assert (!partial = 0.0);*)
163   partial := Unix.gettimeofday ()
164 ;;
165
166 let end_spending () =
167   assert (!partial > 0.0);
168   let interval = (Unix.gettimeofday ()) -. !partial in
169     partial := 0.0;
170     time_spent := !time_spent +. interval
171 ;;
172   
173 \f
174 (*****************************************************************************)
175 (** _fats implementation                                                    **)
176 (*****************************************************************************)
177
178 let rec closure_of_fast ru m =
179   let eq_c = closure_eq_fast ru m in
180   let ge_c = closure_ge_fast ru m in
181   let gt_c = closure_gt_fast ru m in
182     {
183       eq_closure = eq_c;
184       ge_closure = ge_c;
185       gt_closure = gt_c;
186       in_gegt_of = ru.in_gegt_of;
187       one_s_eq = ru.one_s_eq;
188       one_s_ge = ru.one_s_ge;
189       one_s_gt = ru.one_s_gt
190     }
191       
192 and closure_eq_fast ru m = 
193   let eq_c =
194     let j = ru.one_s_eq in
195     let _Uj = merge_closures (fun x -> x.eq_closure) j m in
196     let one_step_eq = ru.one_s_eq in
197       (SOF.union one_step_eq _Uj)
198   in
199     eq_c
200       
201 and closure_ge_fast ru m =
202   let ge_c = 
203     let j = SOF.union ru.one_s_ge (SOF.union ru.one_s_gt ru.one_s_eq) in
204     let _Uj = merge_closures (fun x -> x.ge_closure) j m in
205     let _Ux = j in
206       (SOF.union _Uj _Ux)
207   in
208     ge_c
209       
210 and closure_gt_fast ru m =
211   let gt_c =
212     let j = ru.one_s_gt in
213     let k = ru.one_s_ge in
214     let l = ru.one_s_eq in
215     let _Uj = merge_closures (fun x -> x.ge_closure) j m in
216     let _Uk = merge_closures (fun x -> x.gt_closure) k m in
217     let _Ul = merge_closures (fun x -> x.gt_closure) l m in
218     let one_step_gt = ru.one_s_gt in
219       (SOF.union (SOF.union (SOF.union _Ul one_step_gt) _Uk) _Uj)
220   in
221     gt_c
222       
223 and print_rec_status u ru =
224   print_endline ("Aggiusto " ^ (string_of_universe u) ^ 
225                  "e ottengo questa chiusura\n " ^ (string_of_node ru))
226
227 and adjust_fast u m =
228   let ru = repr u m in
229   let gt_c = closure_gt_fast ru m in
230   let ge_c = closure_ge_fast ru m in
231   let eq_c = closure_eq_fast ru m in
232   let changed_eq = not (are_set_eq eq_c ru.eq_closure) in
233   let changed_gegt = 
234     (not (are_set_eq gt_c ru.gt_closure)) || 
235     (not (are_set_eq ge_c ru.ge_closure))
236   in
237     if ((not changed_gegt) &&  (not changed_eq)) then
238       m
239     else
240       begin
241         let ru' = {
242           eq_closure = eq_c;
243           ge_closure = ge_c;
244           gt_closure = gt_c;
245           in_gegt_of = ru.in_gegt_of;
246           one_s_eq = ru.one_s_eq;
247           one_s_ge = ru.one_s_ge;
248           one_s_gt = ru.one_s_gt}
249         in
250         let m = MAL.add u ru' m in
251         let m =
252             SOF.fold (fun x m -> adjust_fast  x m) 
253               (SOF.union ru'.eq_closure ru'.in_gegt_of) m
254               (* TESI: 
255                    ru'.in_gegt_of m 
256               *)
257         in
258           m (*adjust_fast  u m*)
259       end
260         
261 and add_gt_arc_fast u v m =
262   let ru = repr u m in
263   let ru' = {ru with one_s_gt = SOF.add v ru.one_s_gt} in
264   let m' = MAL.add u ru' m in
265   let rv = repr v m' in
266   let rv' = {rv with in_gegt_of = SOF.add u rv.in_gegt_of} in
267   let m'' = MAL.add v rv' m' in
268     adjust_fast u m''
269       
270 and add_ge_arc_fast u v m =
271   let ru = repr u m in
272   let ru' = { ru with one_s_ge = SOF.add v ru.one_s_ge} in
273   let m' = MAL.add u ru' m in
274   let rv = repr v m' in
275   let rv' = {rv with in_gegt_of = SOF.add u rv.in_gegt_of} in
276   let m'' = MAL.add v rv' m' in
277   adjust_fast u m''
278
279 and add_eq_arc_fast u v m =
280   let ru = repr u m in
281   let rv = repr v m in 
282   let ru' = {ru  with one_s_eq = SOF.add v ru.one_s_eq} in
283   (*TESI: let ru' = {ru' with in_gegt_of = SOF.add v ru.in_gegt_of} in *)
284   let m' = MAL.add u ru' m in
285   let rv' = {rv  with one_s_eq = SOF.add u rv.one_s_eq} in
286   (*TESI: let rv' = {rv' with in_gegt_of = SOF.add u rv.in_gegt_of} in *)
287   let m'' = MAL.add v rv' m' in
288     adjust_fast v (*(adjust_fast u*) m'' (* ) *)
289 ;;
290
291 \f
292 (*****************************************************************************)
293 (** safe implementation                                                     **)
294 (*****************************************************************************)
295
296 let closure_of u m =
297   let ru = repr u m in
298   let eq_c =
299     let j = ru.one_s_eq in
300     let _Uj = merge_closures (fun x -> x.eq_closure) j m in
301     let one_step_eq = ru.one_s_eq in
302             (SOF.union one_step_eq _Uj)
303   in
304   let ge_c = 
305     let j = SOF.union ru.one_s_ge (SOF.union ru.one_s_gt ru.one_s_eq) in
306     let _Uj = merge_closures (fun x -> x.ge_closure) j m in
307     let _Ux = j in
308       (SOF.union _Uj _Ux)
309   in
310   let gt_c =
311     let j = ru.one_s_gt in
312     let k = ru.one_s_ge in
313     let l = ru.one_s_eq in
314     let _Uj = merge_closures (fun x -> x.ge_closure) j m in
315     let _Uk = merge_closures (fun x -> x.gt_closure) k m in
316     let _Ul = merge_closures (fun x -> x.gt_closure) l m in
317     let one_step_gt = ru.one_s_gt in
318       (SOF.union (SOF.union (SOF.union _Ul one_step_gt) _Uk) _Uj)
319   in
320     {
321       eq_closure = eq_c;
322       ge_closure = ge_c;
323       gt_closure = gt_c;
324       in_gegt_of = ru.in_gegt_of;
325       one_s_eq = ru.one_s_eq;
326       one_s_ge = ru.one_s_ge;
327       one_s_gt = ru.one_s_gt
328     }
329
330 let rec simple_adjust m =
331   let m' = 
332     MAL.mapi (fun x _ -> closure_of x m) m
333   in
334     if not (are_ugraph_eq m  m') then(
335       simple_adjust m')
336     else
337       m'
338
339 let add_eq_arc u v m =
340   let ru = repr u m in
341   let rv = repr v m in
342   let ru' = {ru with one_s_eq = SOF.add v ru.one_s_eq} in
343   let m' = MAL.add u ru' m in
344   let rv' = {rv with one_s_eq = SOF.add u rv.one_s_eq} in
345   let m'' = MAL.add v rv' m' in
346     simple_adjust m''
347
348 let add_ge_arc u v m =
349   let ru = repr u m in
350   let ru' = { ru with one_s_ge = SOF.add v ru.one_s_ge} in
351   let m' = MAL.add u ru' m in
352     simple_adjust m'
353
354 let add_gt_arc u v m =
355   let ru = repr u m in
356   let ru' = {ru with one_s_gt = SOF.add v ru.one_s_gt} in
357   let m' = MAL.add u ru' m in
358     simple_adjust m'
359
360 \f
361 (*****************************************************************************)
362 (** Outhern interface, that chooses between _fast and safe                  **)
363 (*****************************************************************************)
364
365 (*                                                                            
366     given the 2 nodes plus the current bag, adds the arc, recomputes the 
367     closures and returns the new map
368 *) 
369 let add_eq fast u v b =
370   if fast then
371     add_eq_arc_fast u v b
372   else
373     add_eq_arc u v b
374
375 (*                                                                            
376     given the 2 nodes plus the current bag, adds the arc, recomputes the 
377     closures and returns the new map
378 *) 
379 let add_ge fast u v b =
380   if fast then
381     add_ge_arc_fast u v b
382   else
383     add_ge_arc u v b
384 (*                                                                            
385     given the 2 nodes plus the current bag, adds the arc, recomputes the 
386     closures and returns the new map
387 *)                                                                            
388 let add_gt fast u v b =
389   if fast then
390     add_gt_arc_fast u v b
391   else
392     add_gt_arc u v b
393
394
395 (*****************************************************************************)
396 (** Other real code                                                         **)
397 (*****************************************************************************)
398
399 exception UniverseInconsistency of string 
400
401 let error arc node1 closure_type node2 closure =
402   let s = "\n  ===== Universe Inconsistency detected =====\n\n" ^
403    "   Unable to add\n" ^ 
404    "\t" ^ (string_of_arc arc) ^ "\n" ^
405    "   cause\n" ^ 
406    "\t" ^ (string_of_universe node1) ^ "\n" ^
407    "   is in the " ^ closure_type ^ " closure\n" ^
408    "\t{" ^ (string_of_universe_set closure) ^ "}\n" ^ 
409    "   of\n" ^ 
410    "\t" ^ (string_of_universe node2) ^ "\n\n" ^
411    "  ===== Universe Inconsistency detected =====\n" in
412   prerr_endline s;
413   raise (UniverseInconsistency s)
414
415
416 let fill_empty_nodes_with_uri (g, already_contained) l uri =
417   let fill_empty_universe u =
418     match u with
419         (i,None) -> (i,Some uri)
420       | (i,Some _) as u -> u
421   in
422   let fill_empty_set s =
423     SOF.fold (fun e s -> SOF.add (fill_empty_universe e) s) s SOF.empty 
424   in
425   let fill_empty_entry e = {
426     eq_closure = (fill_empty_set e.eq_closure) ;
427     ge_closure = (fill_empty_set e.ge_closure) ;
428     gt_closure = (fill_empty_set e.gt_closure) ;
429     in_gegt_of = (fill_empty_set e.in_gegt_of) ;
430     one_s_eq = (fill_empty_set e.one_s_eq) ;
431     one_s_ge = (fill_empty_set e.one_s_ge) ;
432     one_s_gt = (fill_empty_set e.one_s_gt) ;
433   } in  
434   let m = g in
435   let m' = MAL.fold (
436     fun k v m -> 
437       MAL.add (fill_empty_universe k) (fill_empty_entry v) m) m MAL.empty
438   in
439   let l' = List.map fill_empty_universe l in
440     (m', already_contained),l'
441
442
443 (*****************************************************************************)
444 (** World interface                                                         **)
445 (*****************************************************************************)
446
447 type universe_graph = bag * UriManager.UriSet.t 
448 (* the graph , the cache of already merged ugraphs *)
449
450 let empty_ugraph = empty_bag, UriManager.UriSet.empty
451
452 let current_index_anon = ref (-1)
453 let current_index_named = ref (-1)
454
455 let restart_numbering () = current_index_named := (-1) 
456
457 let fresh ?uri ?id () =
458   let i =
459     match uri,id with
460     | None,None -> 
461         current_index_anon := !current_index_anon + 1;
462         !current_index_anon
463     | None, Some _ -> assert false
464     | Some _, None -> 
465         current_index_named := !current_index_named + 1;
466         !current_index_named
467     | Some _, Some id -> id
468   in
469   (i,uri)
470
471 let name_universe u uri =
472   match u with
473   | (i, None) -> (i, Some uri)
474   | _ -> u
475   
476 let print_ugraph (g, _) = 
477   prerr_endline (string_of_bag g)
478
479 let add_eq ?(fast=(!fast_implementation)) u v b =
480   (* should we check to no add twice the same?? *)
481   let m = b in
482   let ru = repr u m in
483   if SOF.mem v ru.gt_closure then
484     error ("EQ",u,v) v "GT" u ru.gt_closure
485   else
486     begin
487     let rv = repr v m in
488     if SOF.mem u rv.gt_closure then
489       error ("EQ",u,v) u "GT" v rv.gt_closure
490     else
491       add_eq fast u v b
492     end
493
494 let add_ge ?(fast=(!fast_implementation)) u v b =
495   (* should we check to no add twice the same?? *)
496   let m = b in
497   let rv = repr v m in
498   if SOF.mem u rv.gt_closure then
499     error ("GE",u,v) u "GT" v rv.gt_closure
500   else
501     add_ge fast u v b
502   
503 let add_gt ?(fast=(!fast_implementation)) u v b =
504   (* should we check to no add twice the same?? *)
505   (* 
506      FIXME : check the thesis... no need to check GT and EQ closure since the 
507      GE is a superset of both 
508   *)
509   let m = b in
510   let rv = repr v m in
511
512   if u = v then
513     error ("GT",u,v) u "==" v SOF.empty
514   else
515   
516   (*if SOF.mem u rv.gt_closure then
517     error ("GT",u,v) u "GT" v rv.gt_closure
518   else
519     begin*)
520       if SOF.mem u rv.ge_closure then
521         error ("GT",u,v) u "GE" v rv.ge_closure
522       else
523 (*        begin
524           if SOF.mem u rv.eq_closure then
525             error ("GT",u,v) u "EQ" v rv.eq_closure
526           else*)
527             add_gt fast u v b
528 (*        end
529     end*)
530
531 (*****************************************************************************)
532 (** START: Decomment this for performance comparisons                       **)
533 (*****************************************************************************)
534
535 let add_eq ?(fast=(!fast_implementation))  u v (b,already_contained) =
536   (*prerr_endline "add_eq";*)
537   begin_spending ();
538   let rc = add_eq ~fast u v b in
539     end_spending();
540     rc,already_contained
541
542 let add_ge ?(fast=(!fast_implementation)) u v (b,already_contained) =
543 (*   prerr_endline "add_ge"; *)
544   begin_spending ();
545   let rc = add_ge ~fast u v b in
546  end_spending();
547     rc,already_contained
548     
549 let add_gt ?(fast=(!fast_implementation)) u v (b,already_contained) =
550 (*   prerr_endline "add_gt"; *)
551   begin_spending ();
552   let rc = add_gt ~fast u v b in
553     end_spending();
554     rc,already_contained
555
556 (*****************************************************************************)
557 (** END: Decomment this for performance comparisons                         **)
558 (*****************************************************************************)
559
560 let merge_ugraphs ~base_ugraph ~increment:(increment, uri_of_increment) =
561   (* this sucks *)
562   let merge_brutal (u,_) v =
563     if u = empty_bag then 
564       ((*prerr_endline "HIT2";*)v) 
565     else if fst v = empty_bag then 
566       ((*prerr_endline "HIT3";*) u, snd v)
567     else
568       ((*prerr_endline "MISS";*)
569       let m1 = u in 
570       let m2 = v in 
571         MAL.fold (
572           fun k v x -> 
573             (SOF.fold (
574                fun u x -> 
575                  let m = add_gt k u x in m) 
576                   (SOF.union v.one_s_gt v.gt_closure)
577                (SOF.fold (
578                   fun u x -> 
579                     let m = add_ge k u x in m) 
580                       (SOF.union v.one_s_ge v.ge_closure)
581                   (SOF.fold (
582                      fun u x ->
583                        let m = add_eq k u x in m) 
584                         (SOF.union v.one_s_eq v.eq_closure) x)))
585             ) m1 m2)
586   in
587   let _, already_contained = base_ugraph in
588   if UriManager.UriSet.mem uri_of_increment already_contained then
589     ((*prerr_endline "HIT1";*)
590     base_ugraph)
591   else
592     fst (merge_brutal increment base_ugraph), 
593     UriManager.UriSet.add uri_of_increment already_contained
594
595 (*****************************************************************************)
596 (** Xml sesialization and parsing                                           **)
597 (*****************************************************************************)
598
599 let xml_of_universe name u = 
600   match u with
601   | (i,Some u) -> 
602       Xml.xml_empty name [
603         None,"id",(string_of_int i) ;
604         None,"uri",(UriManager.string_of_uri u)]
605   | (_,None) -> 
606       raise (Failure "we can serialize only universes with uri")
607
608 let xml_of_set s =
609   let l = 
610     List.map (xml_of_universe "node") (SOF.elements s) 
611   in
612     List.fold_left (fun s x -> [< s ; x >] ) [<>] l
613       
614 let xml_of_entry_content e =
615   let stream_of_field f name =
616     let eq_c = xml_of_set f in
617     if eq_c != [<>] then
618       Xml.xml_nempty name [] eq_c
619     else
620       [<>]
621   in
622   [<
623     (stream_of_field e.eq_closure "eq_closure");
624     (stream_of_field e.gt_closure "gt_closure");
625     (stream_of_field e.ge_closure "ge_closure");
626     (stream_of_field e.in_gegt_of "in_gegt_of");
627     (stream_of_field e.one_s_eq "one_s_eq");
628     (stream_of_field e.one_s_gt "one_s_gt");
629     (stream_of_field e.one_s_ge "one_s_ge")
630   >]
631
632 let xml_of_entry u e =
633   let (i,u') = u in
634   let u'' = 
635     match u' with 
636         Some x -> x 
637       | None -> 
638           raise (Failure "we can serialize only universes (entry) with uri")
639   in
640   let ent = Xml.xml_nempty "entry" [
641     None,"id",(string_of_int i) ; 
642     None,"uri",(UriManager.string_of_uri u'')] in
643   let content = xml_of_entry_content e in
644   ent content
645
646 let write_xml_of_ugraph filename (m,_) l =
647     let tokens = 
648       [< 
649         Xml.xml_cdata "<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?>\n";
650         Xml.xml_nempty "ugraph" [] 
651           ([< (MAL.fold ( fun k v s -> [< s ; (xml_of_entry k v) >]) m [<>]) ; 
652            (List.fold_left 
653              (fun s u -> [< s ; xml_of_universe "owned_node" u >]) [<>] l) >])>]
654     in
655     Xml.pp ~gzip:true tokens (Some filename)
656
657 let univno = fst
658
659  
660 let rec clean_ugraph (m,already_contained) f =
661   let m' = 
662     MAL.fold (fun k v x -> if (f k) then MAL.add k v x else x ) m MAL.empty in
663   let m'' =  MAL.fold (fun k v x -> 
664     let v' = {
665       eq_closure = SOF.filter f v.eq_closure;
666       ge_closure = SOF.filter f v.ge_closure;
667       gt_closure = SOF.filter f v.gt_closure;
668       in_gegt_of = SOF.filter f v.in_gegt_of;
669       one_s_eq = SOF.filter f v.one_s_eq;
670       one_s_ge = SOF.filter f v.one_s_ge;
671       one_s_gt = SOF.filter f v.one_s_gt
672     } in 
673     MAL.add k v' x ) m' MAL.empty in
674   let e_l = 
675     MAL.fold (fun k v l -> if v = empty_entry && not(f k) then
676       begin
677       k::l end else l) m'' []
678   in
679     if e_l != [] then
680       clean_ugraph 
681         (m'', already_contained) (fun u -> (f u) && not (List.mem u e_l))
682     else
683       MAL.fold 
684         (fun k v x -> if v <> empty_entry then MAL.add k v x else x) 
685         m'' MAL.empty,
686       already_contained
687
688 let clean_ugraph g l =
689   clean_ugraph g (fun u -> List.mem u l)
690
691 let assigner_of = 
692   function
693     "ge_closure" -> (fun e u->{e with ge_closure=SOF.add u e.ge_closure})
694   | "gt_closure" -> (fun e u->{e with gt_closure=SOF.add u e.gt_closure})
695   | "eq_closure" -> (fun e u->{e with eq_closure=SOF.add u e.eq_closure})
696   | "in_gegt_of"   -> (fun e u->{e with in_gegt_of  =SOF.add u e.in_gegt_of})
697   | "one_s_ge"   -> (fun e u->{e with one_s_ge  =SOF.add u e.one_s_ge})
698   | "one_s_gt"   -> (fun e u->{e with one_s_gt  =SOF.add u e.one_s_gt})
699   | "one_s_eq"   -> (fun e u->{e with one_s_eq  =SOF.add u e.one_s_eq})
700   | s -> raise (Failure ("unsupported tag " ^ s))
701 ;;
702
703 let cb_factory m l = 
704   let module XPP = XmlPushParser in
705   let current_node = ref (0,None) in
706   let current_entry = ref empty_entry in
707   let current_assign = ref (assigner_of "in_gegt_of") in
708   { XPP.default_callbacks with
709     XPP.end_element = Some( fun name ->
710       match name with
711       | "entry" -> 
712           m := MAL.add !current_node !current_entry !m;
713           current_entry := empty_entry
714       | _ -> ()
715     );
716     XPP.start_element = Some( fun name attlist ->
717       match name with
718       | "ugraph" -> ()
719       | "entry" -> 
720           let id = List.assoc "id" attlist in      
721           let uri = List.assoc "uri" attlist in
722           current_node := (int_of_string id,Some (UriManager.uri_of_string uri))
723       | "node" -> 
724           let id = int_of_string (List.assoc "id" attlist) in
725           let uri = List.assoc "uri" attlist in        
726             current_entry := !current_assign !current_entry 
727               (id,Some (UriManager.uri_of_string uri))
728       | "owned_node" -> 
729           let id = int_of_string (List.assoc "id" attlist) in
730           let uri = List.assoc "uri" attlist in        
731           l := (id,Some (UriManager.uri_of_string uri)) :: !l
732       | s -> current_assign := assigner_of s
733     )
734   }
735 ;; 
736
737 let ugraph_and_univlist_of_xml filename =
738   let module XPP = XmlPushParser in
739   let result_map = ref MAL.empty in
740   let result_list = ref [] in
741   let cb = cb_factory result_map result_list in
742   let xml_parser = XPP.create_parser cb in
743   let xml_source = `Gzip_file filename in
744   (try XPP.parse xml_parser xml_source
745    with (XPP.Parse_error err) as exn -> raise exn);
746   (!result_map,UriManager.UriSet.empty), !result_list
747
748 \f
749 (*****************************************************************************)
750 (** the main, only for testing                                              **)
751 (*****************************************************************************)
752
753 (* 
754
755 type arc = Ge | Gt | Eq ;;
756
757 let randomize_actionlist n m =
758   let ge_percent = 0.7 in
759   let gt_percent = 0.15 in
760   let random_step () =
761     let node1 = Random.int m in
762     let node2 = Random.int m in
763     let op = 
764       let r = Random.float 1.0 in
765         if r < ge_percent then 
766           Ge 
767         else (if r < (ge_percent +. gt_percent) then 
768           Gt 
769         else 
770           Eq) 
771     in
772       op,node1,node2      
773   in
774   let rec aux n =
775     match n with 
776         0 -> []
777       | n -> (random_step ())::(aux (n-1))
778   in
779     aux n
780
781 let print_action_list l =
782   let string_of_step (op,node1,node2) =
783     (match op with
784          Ge -> "Ge"
785        | Gt -> "Gt"
786        | Eq -> "Eq") ^ 
787     "," ^ (string_of_int node1) ^ ","   ^ (string_of_int node2) 
788   in
789   let rec aux l =
790     match l with 
791         [] -> "]"
792       | a::tl ->
793           ";" ^ (string_of_step a) ^ (aux tl)
794   in
795   let body = aux l in
796   let l_body = (String.length body) - 1 in
797     prerr_endline ("[" ^ (String.sub body 1 l_body))
798   
799 let debug = false
800 let d_print_endline = if debug then print_endline else ignore 
801 let d_print_ugraph = if debug then print_ugraph else ignore
802
803 let _ = 
804   (if Array.length Sys.argv < 2 then
805     prerr_endline ("Usage " ^ Sys.argv.(0) ^ " max_edges max_nodes"));
806   Random.self_init ();
807   let max_edges = int_of_string Sys.argv.(1) in
808   let max_nodes = int_of_string Sys.argv.(2) in
809   let action_listR = randomize_actionlist max_edges max_nodes in
810
811   let action_list = [Ge,1,4;Ge,2,6;Ge,1,1;Eq,6,4;Gt,6,3] in
812   let action_list = action_listR in
813   
814   print_action_list action_list;
815   let prform_step ?(fast=false) (t,u,v) g =
816     let f,str = 
817       match t with
818           Ge -> add_ge,">="
819         | Gt -> add_gt,">"
820         | Eq -> add_eq,"="
821     in
822       d_print_endline (
823         "Aggiungo " ^ 
824         (string_of_int u) ^
825         " " ^ str ^ " " ^ 
826         (string_of_int v));
827       let g' = f ~fast (u,None) (v,None) g in
828         (*print_ugraph g' ;*)
829         g'
830   in
831   let fail = ref false in
832   let time1 = Unix.gettimeofday () in
833   let n_safe = ref 0 in
834   let g_safe =  
835     try 
836       d_print_endline "SAFE";
837       List.fold_left (
838         fun g e -> 
839           n_safe := !n_safe + 1;
840           prform_step e g
841       ) empty_ugraph action_list
842     with
843         UniverseInconsistency s -> fail:=true;empty_bag
844   in
845   let time2 = Unix.gettimeofday () in
846   d_print_ugraph g_safe;
847   let time3 = Unix.gettimeofday () in
848   let n_test = ref 0 in
849   let g_test = 
850     try
851       d_print_endline "FAST";
852       List.fold_left (
853         fun g e ->
854           n_test := !n_test + 1;
855           prform_step ~fast:true e g
856       ) empty_ugraph action_list
857     with
858         UniverseInconsistency s -> empty_bag
859   in
860   let time4 = Unix.gettimeofday () in
861   d_print_ugraph g_test;
862     if are_ugraph_eq g_safe g_test && !n_test = !n_safe then
863       begin
864         let num_eq = 
865           List.fold_left (
866             fun s (e,_,_) -> 
867               if e = Eq then s+1 else s 
868           ) 0 action_list 
869         in
870         let num_gt = 
871           List.fold_left (
872             fun s (e,_,_) ->
873               if e = Gt then s+1 else s
874           ) 0 action_list
875         in
876         let num_ge = max_edges - num_gt - num_eq in
877         let time_fast = (time4 -. time3) in
878         let time_safe = (time2 -. time1) in
879         let gap = ((time_safe -. time_fast) *. 100.0) /. time_safe in
880         let fail = if !fail then 1 else 0 in
881           print_endline 
882             (sprintf 
883                "OK %d safe %1.4f fast %1.4f %% %1.2f #eq %d #gt %d #ge %d %d" 
884                fail time_safe time_fast gap num_eq num_gt num_ge !n_safe);
885           exit 0
886       end
887     else
888       begin
889         print_endline "FAIL";
890         print_ugraph g_safe;
891         print_ugraph g_test;
892         exit 1
893       end
894 ;;
895
896  *)
897
898 let recons_univ u =
899   match u with
900   | i, None -> u
901   | i, Some uri ->
902       i, Some (UriManager.uri_of_string (UriManager.string_of_uri uri))
903
904 let recons_entry entry =
905   let recons_set set =
906     SOF.fold (fun univ set -> SOF.add (recons_univ univ) set) set SOF.empty
907   in
908   {
909     eq_closure = recons_set entry.eq_closure;
910     ge_closure = recons_set entry.ge_closure;
911     gt_closure = recons_set entry.gt_closure;
912     in_gegt_of = recons_set entry.in_gegt_of;
913     one_s_eq = recons_set entry.one_s_eq;
914     one_s_ge = recons_set entry.one_s_ge;
915     one_s_gt = recons_set entry.one_s_gt;
916   }
917
918 let recons_graph (graph,uriset) =
919   MAL.fold
920     (fun universe entry map ->
921       MAL.add (recons_univ universe) (recons_entry entry) map)
922     graph 
923     MAL.empty,
924   UriManager.UriSet.fold 
925     (fun u acc -> 
926       UriManager.UriSet.add 
927         (UriManager.uri_of_string (UriManager.string_of_uri u)) acc) 
928     uriset UriManager.UriSet.empty 
929
930 let assert_univ u =
931     match u with 
932     | (_,None) -> raise (UniverseInconsistency "This universe graph has a hole")
933     | _ -> ()
934     
935 let assert_univs_have_uri (graph,_) univlist =
936   let assert_set s =
937     SOF.iter (fun u -> assert_univ u) s
938   in
939   let assert_entry e =
940     assert_set e.eq_closure;
941     assert_set e.ge_closure;
942     assert_set e.gt_closure;
943     assert_set e.in_gegt_of;
944     assert_set e.one_s_eq;
945     assert_set e.one_s_ge;
946     assert_set e.one_s_gt;
947   in
948   MAL.iter (fun k v -> assert_univ k; assert_entry v)graph;
949   List.iter assert_univ univlist
950   
951 let eq u1 u2 = 
952   match u1,u2 with
953   | (id1, Some uri1),(id2, Some uri2) -> 
954       id1 = id2 && UriManager.eq uri1 uri2
955   | (id1, None),(id2, None) -> id1 = id2
956   | _ -> false
957   
958 let compare (id1, uri1) (id2, uri2) = 
959   let cmp = id1 - id2 in
960   if cmp = 0 then
961     match uri1,uri2 with
962     | None, None -> 0 
963     | Some _, None -> 1
964     | None, Some _ -> ~-1
965     | Some uri1, Some uri2 -> UriManager.compare uri1 uri2
966   else
967     cmp
968   
969 (* EOF *)