types.ml 131 KB
Newer Older
1
open Ident
2
open Encodings
3

4 5
let (@@) f a = f a

6
let count = ref 0
Pietro Abate's avatar
Pietro Abate committed
7

8 9 10 11
let () =
  Stats.register Stats.Summary
    (fun ppf -> Format.fprintf ppf "Allocated type nodes:%i@\n" !count)

Pietro Abate's avatar
Pietro Abate committed
12
(*
13 14 15 16 17 18 19 20
To be sure not to use generic comparison ...
*)
let (=) : int -> int -> bool = (==)
let (<) : int -> int -> bool = (<)
let (<=) : int -> int -> bool = (<=)
let (<>) : int -> int -> bool = (<>)
let compare = 1

Pietro Abate's avatar
Pietro Abate committed
21
type const =
22
  | Integer of Intervals.V.t
23
  | Atom of Atoms.V.t
24
  | Char of Chars.V.t
25 26 27 28
  | Pair of const * const
  | Xml of const * const
  | Record of const label_map
  | String of U.uindex * U.uindex * U.t * const
29

30 31
type service_params =
  | TProd of service_params * service_params
Pietro Abate's avatar
Pietro Abate committed
32
  | TOption of service_params
33 34 35 36 37 38 39
  | TList of string * service_params
  | TSet of service_params
  | TSum of service_params * service_params
  | TString of string
  | TInt of string
  | TInt32 of string
  | TInt64 of string
Pietro Abate's avatar
Pietro Abate committed
40
  | TFloat of string
41 42 43
  | TBool of string
  | TFile of string
      (* | TUserType of string * (string -> 'a) * ('a -> string) *)
Pietro Abate's avatar
Pietro Abate committed
44
  | TCoord of string
45
  | TCoordv of service_params * string
Pietro Abate's avatar
Pietro Abate committed
46
  | TESuffix of string
47 48 49
  | TESuffixs of string
      (*  | TESuffixu of (string * (string -> 'a) * ('a -> string)) *)
  | TSuffix of (bool * service_params)
Pietro Abate's avatar
Pietro Abate committed
50
  | TUnit
51 52 53
  | TAny
  | TConst of string;;

54 55 56
module Const = struct
  type t = const

57 58
  let check _ = ()
  let dump ppf _ = Format.fprintf ppf "<Types.Const.t>"
59 60

  let rec compare c1 c2 = match (c1,c2) with
61
    | Integer x, Integer y -> Intervals.V.compare x y
62 63
    | Integer _, _ -> -1
    | _, Integer _ -> 1
64
    | Atom x, Atom y -> Atoms.V.compare x y
65 66
    | Atom _, _ -> -1
    | _, Atom _ -> 1
67

68
    | Char x, Char y -> Chars.V.compare x y
69 70 71
    | Char _, _ -> -1
    | _, Char _ -> 1
    | Pair (x1,x2), Pair (y1,y2) ->
72 73
	let c = compare x1 y1 in
	if c <> 0 then c else compare x2 y2
74 75 76
    | Pair (_,_), _ -> -1
    | _, Pair (_,_) -> 1
    | Xml (x1,x2), Xml (y1,y2) ->
77 78
	let c = compare x1 y1 in
	if c <> 0 then c else compare x2 y2
79 80 81
    | Xml (_,_), _ -> -1
    | _, Xml (_,_) -> 1
    | Record x, Record y ->
82
	LabelMap.compare compare x y
83 84 85
    | Record _, _ -> -1
    | _, Record _ -> 1
    | String (i1,j1,s1,r1), String (i2,j2,s2,r2) ->
Pietro Abate's avatar
Pietro Abate committed
86
	let c = Pervasives.compare i1 i2 in if c <> 0 then c
87 88 89
	else let c = Pervasives.compare j1 j2 in if c <> 0 then c
	else let c = U.compare s1 s2 in if c <> 0 then c (* Should compare
							    only the substring *)
90 91 92 93 94 95 96 97 98 99
	else compare r1 r2

  let rec hash = function
    | Integer x -> 1 + 17 * (Intervals.V.hash x)
    | Atom x -> 2 + 17 * (Atoms.V.hash x)
    | Char x -> 3 + 17 * (Chars.V.hash x)
    | Pair (x,y) -> 4 + 17 * (hash x) + 257 * (hash y)
    | Xml (x,y) -> 5 + 17 * (hash x) + 257 * (hash y)
    | Record x -> 6 + 17 * (LabelMap.hash hash x)
    | String (i,j,s,r) -> 7 + 17 * (U.hash s) + 257 * hash r
100
      (* Note: improve hash for String *)
101

102 103
  let equal c1 c2 = compare c1 c2 = 0
end
104

105 106 107
module Abstracts
  =
struct
108
  module T = Custom.String
109

110
  module V = struct type t = T.t * Obj.t end
111 112

  include SortedList.FiniteCofinite(T)
113 114 115 116 117
  let trivially_disjoint = disjoint
  let compute ~empty ~full ~cup ~cap ~diff ~atom b = assert false
  let get _ = assert false
  let iter _ = assert false

118
  let full = any
119 120

  let print = function
121
    | Finite l -> List.map (fun x ppf -> Format.fprintf ppf "!%s" x) l
122
    | Cofinite l ->
123 124 125 126
	[ fun ppf ->
	  Format.fprintf ppf "@[Abstract";
	  List.iter (fun x -> Format.fprintf ppf " \\@ !%s" x) l;
	  Format.fprintf ppf "@]" ]
127

128 129 130 131
  let contains_sample s t = match s,t with
    | None, Cofinite _ -> true
    | None, Finite _ -> false
    | Some s, t -> contains s t
Pietro Abate's avatar
Pietro Abate committed
132

133 134
end

135 136
type pair_kind = [ `Normal | `XML ]

137 138 139 140
module BoolAtoms = BoolVar.Make(Atoms)
module BoolIntervals = BoolVar.Make(Intervals)
module BoolChars = BoolVar.Make(Chars)
module BoolAbstracts = BoolVar.Make(Abstracts)
141

Pietro Abate's avatar
Pietro Abate committed
142
module rec Descr :
143
sig
144
  type s = {
145 146 147
    atoms : BoolAtoms.t;
    ints  : BoolIntervals.t;
    chars : BoolChars.t;
148 149 150 151
    times : BoolPair.t;
    xml   : BoolPair.t;
    arrow : BoolPair.t;
    record: BoolRec.t;
152
    abstract: BoolAbstracts.t;
153
    absent: bool;
154
  }
155
  include Custom.T with type t = s
156
  val empty: t
157
  val any : t
Julien Lopez's avatar
Julien Lopez committed
158
  val is_empty : t -> bool
159 160
end =
struct
161
  type s = {
162 163 164
    atoms : BoolAtoms.t;
    ints  : BoolIntervals.t;
    chars : BoolChars.t;
165 166 167 168
    times : BoolPair.t;
    xml   : BoolPair.t;
    arrow : BoolPair.t;
    record: BoolRec.t;
169
    abstract: BoolAbstracts.t;
170
    absent: bool;
171
  }
172
  type t = s
173

174
  let dump ppf d =
175 176 177 178 179 180 181 182 183 184
    Format.fprintf ppf "@[<v 1>types:@\n\
@<1>    atoms: %a@\n\
@<1>     ints: %a@\n\
@<1>    chars: %a@\n\
@<1>    times: %a@\n\
@<1>    arrow: %a@\n\
@<1>   record: %a@\n\
@<1>      xml: %a@\n\
@<1> abstract: %a@\n\
@<1>   absent: %b@]@\n"
185
      BoolAtoms.dump d.atoms
186 187
      BoolIntervals.dump d.ints
      BoolChars.dump d.chars
188
      BoolPair.dump d.times
189
      BoolPair.dump d.arrow
190 191
      BoolRec.dump d.record
      BoolPair.dump d.xml
192
      BoolAbstracts.dump d.abstract
193
      d.absent
194

Pietro Abate's avatar
Pietro Abate committed
195 196 197 198
  let empty = {
    times = BoolPair.empty;
    xml   = BoolPair.empty;
    arrow = BoolPair.empty;
199
    record= BoolRec.empty;
200 201 202
    ints  = BoolIntervals.empty;
    atoms = BoolAtoms.empty;
    chars = BoolChars.empty;
203
    abstract = BoolAbstracts.empty;
204
    absent = false;
205 206
  }

207 208 209 210
  let any =  {
    times = BoolPair.full;
    xml   = BoolPair.full;
    arrow = BoolPair.full;
211
    record = BoolRec.full;
212 213 214
    ints  = BoolIntervals.full;
    atoms = BoolAtoms.full;
    chars = BoolChars.full;
215
    abstract = BoolAbstracts.full;
216
    absent = false;
217 218
  }

219 220 221 222 223 224 225 226
  let check a =
    BoolChars.check a.chars;
    BoolIntervals.check a.ints;
    BoolAtoms.check a.atoms;
    BoolPair.check a.times;
    BoolPair.check a.xml;
    BoolPair.check a.arrow;
    BoolRec.check a.record;
227
    BoolAbstracts.check a.abstract;
228 229
    ()

230
  let equal a b =
231
    (a == b) || (
232 233 234
      (BoolAtoms.equal a.atoms b.atoms) &&
      (BoolChars.equal a.chars b.chars) &&
      (BoolIntervals.equal a.ints  b.ints) &&
235 236 237 238
      (BoolPair.equal a.times b.times) &&
      (BoolPair.equal a.xml b.xml) &&
      (BoolPair.equal a.arrow b.arrow) &&
      (BoolRec.equal a.record b.record) &&
239
      (BoolAbstracts.equal a.abstract b.abstract) &&
240 241
      (a.absent == b.absent)
    )
242

Julien Lopez's avatar
Julien Lopez committed
243 244 245 246 247 248 249 250
  let is_empty a =
    (BoolAtoms.is_empty a.atoms) &&
      (BoolChars.is_empty a.chars) &&
      (BoolIntervals.is_empty a.ints) &&
      (BoolPair.is_empty a.times) &&
      (BoolPair.is_empty a.xml) &&
      (BoolPair.is_empty a.arrow) &&
      (BoolRec.is_empty a.record) &&
251
      (BoolAbstracts.is_empty a.abstract)
Julien Lopez's avatar
Julien Lopez committed
252

253
  let compare a b =
Pietro Abate's avatar
Pietro Abate committed
254
    if a == b then 0
255 256 257
    else let c = BoolAtoms.compare a.atoms b.atoms in if c <> 0 then c
    else let c = BoolChars.compare a.chars b.chars in if c <> 0 then c
    else let c = BoolIntervals.compare a.ints b.ints in if c <> 0 then c
258 259 260 261
    else let c = BoolPair.compare a.times b.times in if c <> 0 then c
    else let c = BoolPair.compare a.xml b.xml in if c <> 0 then c
    else let c = BoolPair.compare a.arrow b.arrow in if c <> 0 then c
    else let c = BoolRec.compare a.record b.record in if c <> 0 then c
262
    else let c = BoolAbstracts.compare a.abstract b.abstract in if c <> 0 then c
263 264 265
    else if a.absent && not b.absent then -1
    else if b.absent && not a.absent then 1
    else 0
Pietro Abate's avatar
Pietro Abate committed
266

267
  let hash a =
268 269 270
    let accu = BoolChars.hash a.chars in
    let accu = 17 * accu + BoolIntervals.hash a.ints in
    let accu = 17 * accu + BoolAtoms.hash a.atoms in
271 272 273 274
    let accu = 17 * accu + BoolPair.hash a.times in
    let accu = 17 * accu + BoolPair.hash a.xml in
    let accu = 17 * accu + BoolPair.hash a.arrow in
    let accu = 17 * accu + BoolRec.hash a.record in
275
    let accu = 17 * accu + BoolAbstracts.hash a.abstract in
276 277
    let accu = if a.absent then accu+5 else accu in
    accu
278

279 280 281
end
and Node :
sig
282
  type t = { id : int; cu: Compunit.t; mutable descr : Descr.t }
283 284 285 286 287
  val dump: Format.formatter -> t -> unit
  val check: t -> unit
  val equal: t -> t -> bool
  val hash: t -> int
  val compare:t -> t -> int
288
  val mk: int -> Descr.t -> t
289
end =
290

291
struct
292
  type t = { id : int; cu: Compunit.t; mutable descr : Descr.t }
293
  let check n = ()
294
  let dump ppf n = Format.fprintf ppf "X%i" n.id
295
  let hash x = x.id + Compunit.hash x.cu
Pietro Abate's avatar
Pietro Abate committed
296
  let compare x y =
297 298
    let c = x.id - y.id in if c = 0 then Compunit.compare x.cu y.cu else c
  let equal x y = x==y || (x.id == y.id && (Compunit.equal x.cu y.cu))
299

300
  let mk id d = { id = id; cu = Compunit.current (); descr = d }
301 302 303



304 305
end

306 307
and Pair : Bool.S with type elem = (Node.t * Node.t) =
  Bool.Make(Custom.Pair(Node)(Node))
308
and BoolPair : BoolVar.S with module Atom = Pair = BoolVar.Make(Pair)
309

310 311 312 313 314
(* bool = true means that the record is open that is, that
 * the labels that are not in the domain of the map are
 * equal to "any" *)
and Rec : Bool.S with type elem = bool * Node.t Ident.label_map =
  Bool.Make(Custom.Pair(Custom.Bool)(LabelSet.MakeMap(Node)))
315
and BoolRec : BoolVar.S with module Atom = Rec = BoolVar.Make(Rec)
316

317 318
module DescrHash = Hashtbl.Make(Descr)
module DescrMap = Map.Make(Descr)
319 320
module DescrSet = Set.Make(Descr)
module DescrSList = SortedList.Make(Descr)
321

322 323 324
type descr = Descr.t
type node = Node.t
include Descr
325

326 327
let dummy_print = (fun _ _ -> assert false)
let forward_print = ref dummy_print
328

Pietro Abate's avatar
Pietro Abate committed
329 330
let make () =
  incr count;
331 332
  Node.mk !count empty

Pietro Abate's avatar
Pietro Abate committed
333
let define n d =
334 335
  n.Node.descr <- d

Pietro Abate's avatar
Pietro Abate committed
336 337
let cons d =
  incr count;
338 339
  Node.mk !count d

340 341 342 343
let descr n = n.Node.descr
let internalize n = n
let id n = n.Node.id

344
let non_constructed =
Pietro Abate's avatar
Pietro Abate committed
345
  { any with
346
      times = empty.times; xml = empty.xml; record = empty.record }
Pietro Abate's avatar
Pietro Abate committed
347 348

let non_constructed_or_absent =
349
  { non_constructed with absent = true }
350

Pietro Abate's avatar
Pietro Abate committed
351
(* Descr.t type constructors *)
352 353 354
let times x y = { empty with times = BoolPair.atom (`Atm (Pair.atom (x,y))) }
let xml x y = { empty with xml = BoolPair.atom (`Atm (Pair.atom (x,y))) }
let arrow x y = { empty with arrow = BoolPair.atom (`Atm (Pair.atom (x,y))) }
355

Pietro Abate's avatar
Pietro Abate committed
356
let record label t =
357
  { empty with record = BoolRec.atom (`Atm (Rec.atom (true,LabelMap.singleton label t))) }
358

359
let record_fields x =
360
  { empty with record = BoolRec.atom (`Atm (Rec.atom x)) }
361

362
let atom a = { empty with atoms = BoolAtoms.atom (`Atm a) }
363 364

(* Atm = Any ^ a *)
365 366 367 368 369 370 371 372 373 374
let var a =
  {
  times = BoolPair.var a;
  xml   = BoolPair.var a;
  arrow = BoolPair.var a;
  record= BoolRec.var a;
  ints  = BoolIntervals.var a;
  atoms = BoolAtoms.var a;
  chars = BoolChars.var a;
  abstract = BoolAbstracts.var a;
375
  absent = false;
376 377
}

378 379
let char c = { empty with chars = BoolChars.atom (`Atm c) }
let interval i = { empty with ints = BoolIntervals.atom (`Atm i) }
380
let abstract a = { empty with abstract = BoolAbstracts.atom (`Atm a) }
381

Pietro Abate's avatar
Pietro Abate committed
382
let cup x y =
383
  if x == y then x else
384
    {
385 386 387 388 389 390 391
      times = BoolPair.cup x.times y.times;
      xml   = BoolPair.cup x.xml y.xml;
      arrow = BoolPair.cup x.arrow y.arrow;
      record= BoolRec.cup x.record y.record;
      ints  = BoolIntervals.cup x.ints  y.ints;
      atoms = BoolAtoms.cup x.atoms y.atoms;
      chars = BoolChars.cup x.chars y.chars;
392
      abstract = BoolAbstracts.cup x.abstract y.abstract;
393 394
      absent = x.absent || y.absent;
    }
395

Pietro Abate's avatar
Pietro Abate committed
396
let cap x y =
397
  if x == y then x else
398
    {
399 400 401
      ints  = BoolIntervals.cap x.ints y.ints;
      times = BoolPair.cap x.times y.times;
      xml   = BoolPair.cap x.xml y.xml;
402
      record = BoolRec.cap x.record y.record;
403 404 405
      arrow = BoolPair.cap x.arrow y.arrow;
      atoms = BoolAtoms.cap x.atoms y.atoms;
      chars = BoolChars.cap x.chars y.chars;
406
      abstract = BoolAbstracts.cap x.abstract y.abstract;
407
      absent= x.absent && y.absent;
408
    }
409

Pietro Abate's avatar
Pietro Abate committed
410
let diff x y =
411
  if x == y then empty else
412
    {
413 414 415 416 417 418 419
      times = BoolPair.diff x.times y.times;
      xml   = BoolPair.diff x.xml y.xml;
      arrow = BoolPair.diff x.arrow y.arrow;
      record= BoolRec.diff x.record y.record;
      ints  = BoolIntervals.diff x.ints y.ints;
      atoms = BoolAtoms.diff x.atoms y.atoms;
      chars = BoolChars.diff x.chars y.chars;
420
      abstract = BoolAbstracts.diff x.abstract y.abstract;
421
      absent= x.absent && not y.absent;
422
    }
Pietro Abate's avatar
Pietro Abate committed
423

424 425
(* TODO: optimize disjoint check for boolean combinations *)
let trivially_disjoint a b =
426 427 428
  (BoolChars.trivially_disjoint a.chars b.chars) &&
  (BoolIntervals.trivially_disjoint a.ints b.ints) &&
  (BoolAtoms.trivially_disjoint a.atoms b.atoms) &&
429 430 431
  (BoolPair.trivially_disjoint a.times b.times) &&
  (BoolPair.trivially_disjoint a.xml b.xml) &&
  (BoolPair.trivially_disjoint a.arrow b.arrow) &&
432
  (BoolRec.trivially_disjoint a.record b.record) &&
433
  (BoolAbstracts.trivially_disjoint a.abstract b.abstract) &&
434
  (not (a.absent && b.absent))
435

436
let rec constant = function
437 438 439
  | Integer i -> interval (Intervals.atom i)
  | Atom a -> atom (Atoms.atom a)
  | Char c -> char (Chars.atom c)
440
  | Pair (x,y) -> times (const_node x) (const_node y)
441
  | Xml (x,y) -> xml (const_node x) (const_node y)
442
  | Record x -> record_fields (false ,LabelMap.map const_node x)
443 444
  | String (i,j,s,c) ->
      if U.equal_index i j then constant c
Pietro Abate's avatar
Pietro Abate committed
445
      else
446 447 448
	let (ch,i') = U.next s i in
	constant (Pair (Char (Chars.V.mk_int ch), String (i',j,s,c)))
and const_node c = cons (constant c)
449

450 451
let neg x = diff any x

452
let any_node = cons any
453
let empty_node = cons empty
454

455
module LabelS = Set.Make(Label)
456

Pietro Abate's avatar
Pietro Abate committed
457
let any_or_absent = { any with absent = true }
458
let only_absent = { empty with absent = true }
459

460
let get_record r =
Pietro Abate's avatar
Pietro Abate committed
461 462
  let labs accu (_,r) =
    List.fold_left
463
      (fun accu (l,_) -> LabelS.add l accu) accu (LabelMap.get r) in
464
  let extend descrs labs (o,r) =
465 466 467 468 469
    let rec aux i labs r =
      match labs with
	| [] -> ()
	| l1::labs ->
	    match r with
Pietro Abate's avatar
Pietro Abate committed
470
	      | (l2,x)::r when l1 == l2 ->
471 472 473
		  descrs.(i) <- cap descrs.(i) (descr x);
		  aux (i+1) labs r
	      | r ->
Pietro Abate's avatar
Pietro Abate committed
474
		  if not o then
475
		    descrs.(i) <- cap descrs.(i) only_absent; (* TODO:OPT *)
476 477
		  aux (i+1) labs r
    in
478
    aux 0 labs (LabelMap.get r);
479 480 481
    o
  in
  let line (p,n) =
Pietro Abate's avatar
Pietro Abate committed
482
    let labels =
483 484
      List.fold_left labs (List.fold_left labs LabelS.empty p) n in
    let labels = LabelS.elements labels in
485
    let nlab = List.length labels in
486
    let mk () = Array.create nlab any_or_absent in
487 488

    let pos = mk () in
Pietro Abate's avatar
Pietro Abate committed
489 490
    let opos = List.fold_left
		 (fun accu x ->
491 492 493 494 495 496 497 498 499 500 501
		    (extend pos labels x) && accu)
		 true p in
    let p = (opos, pos) in

    let n = List.map (fun x ->
			let neg = mk () in
			let o = extend neg labels x in
			(o,neg)
		     ) n in
    (labels,p,n)
  in
502
  List.map line (Rec.get r)
503

504 505 506 507 508
(* Subtyping algorithm *)

let diff_t d t = diff d (descr t)
let cap_t d t = cap d (descr t)
let cup_t d t = cup d (descr t)
509
let cap_product any_left any_right l =
Pietro Abate's avatar
Pietro Abate committed
510
  List.fold_left
511
    (fun (d1,d2) (t1,t2) -> (cap_t d1 t1, cap_t d2 t2))
512
    (any_left,any_right)
513
    l
514
let any_pair = { empty with times = any.times }
515

516 517 518
let rec exists max f =
  (max > 0) && (f (max - 1) || exists (max - 1) f)

519
exception NotEmpty
520

521 522 523 524 525 526 527 528 529
module Witness = struct

  module NodeSet = Set.Make(Node)

  type witness =
    | WInt of Intervals.V.t
    | WAtom of Atoms.sample
    | WChar of Chars.V.t
    | WAbsent
530
    | WAbstract of Abstracts.elem option
531 532 533 534 535 536 537

    | WPair of witness * witness * witness_slot
    | WXml of witness * witness * witness_slot
    | WRecord of witness label_map * bool * witness_slot
	(* Invariant: WAbsent cannot actually appear *)

    | WFun of (witness * witness option) list * witness_slot
Pietro Abate's avatar
Pietro Abate committed
538
  and witness_slot =
539 540 541 542 543 544 545 546 547 548 549 550 551 552 553
      { mutable wnodes_in: NodeSet.t;
	mutable wnodes_out: NodeSet.t;
	mutable wuid: int }

  module WHash = Hashtbl.Make(
    struct
      type t = witness
      let hash_small = function
	| WInt i -> 17 * Intervals.V.hash i
	| WChar c -> 1 + 17 * Chars.V.hash c
	| WAtom None -> 2
	| WAtom (Some (ns,None)) -> 3 + 17 * Ns.Uri.hash ns
	| WAtom (Some (_,Some t)) -> 4 + 17 * Ns.Label.hash t
	| WAbsent -> 5
	| WAbstract None -> 6
554
	| WAbstract (Some t) -> 7 + 17 * Abstracts.T.hash t
Pietro Abate's avatar
Pietro Abate committed
555
	| WPair (_,_,s)
556 557 558 559 560 561
	| WXml (_,_,s)
	| WRecord (_,_,s)
	| WFun (_,s) -> 8 + 17 * s.wuid
      let hash = function
	| WPair (p1,p2,_) -> 257 * hash_small p1 + 65537 * hash_small p2
	| WXml (p1,p2,_) -> 1 + 257 * hash_small p1 + 65537 * hash_small p2
Pietro Abate's avatar
Pietro Abate committed
562
	| WRecord (r,o,_) ->
563 564 565
	    (if o then 2 else 3) + 257 * LabelMap.hash hash_small r
	| WFun (f,_) ->
	    4 + 257 *
Pietro Abate's avatar
Pietro Abate committed
566 567
	      (Hashtbl.hash
		 (List.map
568
		    (function (x,None) -> 17 * hash_small x
Pietro Abate's avatar
Pietro Abate committed
569
		       | (x,Some y) ->
570 571 572 573 574 575 576 577 578 579 580 581 582 583 584
			   1 + 17 * hash_small x + 257 * hash_small y)
		    f)
	      )
	| _ -> assert false

      let equal_small w1 w2 = match w1,w2 with
	| WInt i1, WInt i2 -> Intervals.V.equal i1 i2
	| WChar c1, WChar c2 -> Chars.V.equal c1 c2
	| WAtom None, WAtom None -> true
	| WAtom (Some (ns1,None)), WAtom (Some (ns2,None)) ->
	    Ns.Uri.equal ns1 ns2
	| WAtom (Some (_,Some t1)), WAtom (Some (_,Some t2)) ->
	    Ns.Label.equal t1 t2
	| WAbsent, WAbsent -> true
	| WAbstract None, WAbstract None -> false
585
	| WAbstract (Some t1), WAbstract (Some t2) -> Abstracts.T.equal t1 t2
586 587 588
	| _ -> w1 == w2

      let equal w1 w2 = match w1,w2 with
Pietro Abate's avatar
Pietro Abate committed
589 590
	| WPair (p1,q1,_), WPair (p2,q2,_)
	| WXml (p1,q1,_), WXml (p2,q2,_) ->
591 592 593 594 595 596 597 598
	    equal_small p1 p2 && equal_small q1 q2
	| WRecord (r1,o1,_), WRecord (r2,o2,_) ->
	    o1 == o2 && (LabelMap.equal equal_small r1 r2)
	| WFun (f1,_), WFun (f2,_) ->
	    List.length f1 = List.length f2 &&
		List.for_all2
		(fun (x1,y1) (x2,y2) ->
		   equal_small x1 x2 && (match y1,y2 with
Pietro Abate's avatar
Pietro Abate committed
599
					   | Some y1, Some y2 ->
600 601 602 603 604 605 606 607 608
					       equal_small y1 y2
					   | None, None -> true
					   | _ -> false)
		) f1 f2
	| _ -> false
    end)

  let wmemo = WHash.create 1024
  let wuid = ref 0
Pietro Abate's avatar
Pietro Abate committed
609
  let wslot () = { wuid = !wuid; wnodes_in = NodeSet.empty;
610 611
		   wnodes_out = NodeSet.empty }

612 613 614 615
  let () =
    Stats.register Stats.Summary
      (fun ppf -> Format.fprintf ppf "Allocated witnesses:%i@\n" !wuid)

616
  let rec pp ppf = function
617 618 619 620 621 622 623 624 625 626
    | WInt i ->
	Format.fprintf ppf "%a" Intervals.V.print i
    | WChar c ->
	Format.fprintf ppf "%a" Chars.V.print c
    | WAtom None ->
	Format.fprintf ppf "`#:#"
    | WAtom (Some (ns,None)) ->
	Format.fprintf ppf "`%a" Ns.InternalPrinter.print_any_ns ns
    | WAtom (Some (_,Some t)) ->
	Format.fprintf ppf "`%a" Ns.Label.print_attr t
Pietro Abate's avatar
Pietro Abate committed
627
    | WPair (w1,w2,_) ->
628
	Format.fprintf ppf "(%a,%a)" pp w1 pp w2
Pietro Abate's avatar
Pietro Abate committed
629
    | WXml (w1,w2,_) ->
630
	Format.fprintf ppf "XML(%a,%a)" pp w1 pp w2
631 632 633
    | WRecord (ws,o,_) ->
	Format.fprintf ppf "{";
	LabelMap.iteri
Pietro Abate's avatar
Pietro Abate committed
634
	  (fun l w -> Format.fprintf ppf " %a=%a"
635
	     Label.print_attr l pp w)
636 637 638 639 640 641
	  ws;
	if o then Format.fprintf ppf " ..";
	Format.fprintf ppf " }"
    | WFun (f,_) ->
	Format.fprintf ppf "FUN{";
	List.iter (fun (x,y) ->
642
		     Format.fprintf ppf " %a->" pp x;
643 644
		     match y with
		       | None -> Format.fprintf ppf "#"
645
		       | Some y -> pp ppf y) f;
646 647 648 649 650 651 652
	Format.fprintf ppf " }"
    | WAbstract None ->
	Format.fprintf ppf "Abstract(..)"
    | WAbstract (Some s) ->
	Format.fprintf ppf "Abstract(%s)" s
    | WAbsent ->
	Format.fprintf ppf "Absent"
Pietro Abate's avatar
Pietro Abate committed
653

654 655
  let printf = pp Format.std_formatter

656 657
  let wmk w =  (* incr wuid; w *)  (* hash-consing disabled *)
    try WHash.find wmemo w
Pietro Abate's avatar
Pietro Abate committed
658 659
    with Not_found ->
      incr wuid;
660
      WHash.add wmemo w w;
Pietro Abate's avatar
Pietro Abate committed
661
(*      Format.fprintf Format.std_formatter "W:%a@."
662
	pp w; *)
663 664 665 666 667 668 669
      w

  let wpair p1 p2 = wmk (WPair (p1,p2, wslot()))
  let wxml p1 p2 = wmk (WXml (p1,p2, wslot()))
  let wrecord r o = wmk (WRecord (r,o, wslot()))
  let wfun f = wmk (WFun (f, wslot()))

670
  let bool_pair f =
Pietro Abate's avatar
Pietro Abate committed
671 672 673
    Pair.compute
      ~empty:false ~full:true
      ~cup:(||) ~cap:(&&) ~diff:(fun x y -> x && not y)
674 675 676
      ~atom:f

  let bool_rec f =
Pietro Abate's avatar
Pietro Abate committed
677 678 679
    Rec.compute
      ~empty:false ~full:true
      ~cup:(||) ~cap:(&&) ~diff:(fun x y -> x && not y)
680 681
      ~atom:f

682
  let rec node_has n = function
683 684 685 686 687 688 689 690 691
    | WXml (_,_,s) | WPair (_,_,s) | WFun (_,s) | WRecord (_,_,s) as w ->
	if NodeSet.mem n s.wnodes_in then true
	else if NodeSet.mem n s.wnodes_out then false
	else (let r = type_has (descr n) w in
	      if r then s.wnodes_in <- NodeSet.add n s.wnodes_in
	      else s.wnodes_out <- NodeSet.add n s.wnodes_out;
	      r)
    | w -> type_has (descr n) w

692 693
  (* type_has checks if a witness is contained in the union of
   * the leafs of a bdd, ignoring all variables. *)
694
  and type_has t = function
695 696 697
    | WInt i -> Intervals.contains i (BoolIntervals.leafconj t.ints)
    | WChar c -> Chars.contains c (BoolChars.leafconj t.chars)
    | WAtom a -> Atoms.contains_sample a (BoolAtoms.leafconj t.atoms)
Pietro Abate's avatar
Pietro Abate committed
698 699 700
    | WPair (w1,w2,_) ->
	bool_pair
	  (fun (n1,n2) -> node_has n1 w1 && node_has n2 w2)
701
	  (BoolPair.leafconj t.times)
702
    | WXml (w1,w2,_) ->
Pietro Abate's avatar
Pietro Abate committed
703
	bool_pair
704
	  (fun (n1,n2) -> node_has n1 w1 && node_has n2 w2)
705
	  (BoolPair.leafconj t.xml)
706
    | WFun (f,_) ->
Pietro Abate's avatar
Pietro Abate committed
707
	bool_pair
708 709 710 711 712 713
	  (fun (n1,n2) ->
	     List.for_all
	       (fun (x,y) ->
		  not (node_has n1 x) ||
		    (match y with None -> false
		       | Some y -> node_has n2 y))
Pietro Abate's avatar
Pietro Abate committed
714
	       f)
715
	  (BoolPair.leafconj t.arrow)
716
    | WRecord (f,o,_) ->
Pietro Abate's avatar
Pietro Abate committed
717
	bool_rec
718 719 720
	  (fun (o',f') ->
	     ((not o) || o') && (
	       let checked = ref 0 in
Pietro Abate's avatar
Pietro Abate committed
721 722
	       try
		 LabelMap.iteri
723
		   (fun l n ->
Pietro Abate's avatar
Pietro Abate committed
724
		      let w =
725 726 727
			try let w = LabelMap.assoc l f in incr checked; w
			with Not_found -> WAbsent in
		      if not (node_has n w) then raise Exit
Pietro Abate's avatar
Pietro Abate committed
728
		   ) f';
729 730 731 732 733
		 o' || (LabelMap.length f == !checked)
		   (* All the remaining fields cannot be WAbsent
		      because of an invariant. Otherwise, we must
		      check that all are WAbsent here. *)
	       with Exit -> false))
734
	  (BoolRec.leafconj t.record)
735
    | WAbsent -> t.absent
736
    | WAbstract a -> Abstracts.contains_sample a (BoolAbstracts.leafconj t.abstract)
737 738
end

Pietro Abate's avatar
Pietro Abate committed
739
type slot = { mutable status : status;
740 741
	       mutable notify : notify;
	       mutable active : bool }
742 743
and status = Empty | NEmpty of Witness.witness | Maybe
and notify = Nothing | Do of slot * (Witness.witness -> unit) * notify
744 745

let slot_empty = { status = Empty; active = false; notify = Nothing }
746 747
let slot_nempty w = { status = NEmpty w;
		     active = false; notify = Nothing }
748

749
let rec notify w = function
750
  | Nothing -> ()
Pietro Abate's avatar
Pietro Abate committed
751
  | Do (n,f,rem) ->
752 753
      if n.status == Maybe then (try f w with NotEmpty -> ());
      notify w rem
754 755 756 757 758

let rec iter_s s f = function
  | [] -> ()
  | arg::rem -> f arg s; iter_s s f rem

759 760 761
let set s w =
  s.status <- NEmpty w;
  notify w s.notify;
Pietro Abate's avatar
Pietro Abate committed
762
  s.notify <- Nothing;
763 764
  raise NotEmpty

765
let rec big_conj f l n w =
766
  match l with
767 768
    | [] -> set n w
    | [arg] -> f w arg n
769
    | arg::rem ->
Pietro Abate's avatar
Pietro Abate committed
770 771
	let s =
	  { status = Maybe; active = false;
772
	    notify = Do (n,(big_conj f rem n), Nothing) } in
Pietro Abate's avatar
Pietro Abate committed
773
	try
774
	  f w arg s;
775
	  if s.active then n.active <- true
776
	with NotEmpty when n.status == Empty || n.status == Maybe -> ()
777

778
let memo = DescrHash.create 8191
Pietro Abate's avatar
Pietro Abate committed
779
let marks = ref []
780

Pietro Abate's avatar
Pietro Abate committed
781
let count_subtype = Stats.Counter.create "Subtyping internal loop"
782

783 784
let complex = ref 0

785
let rec slot d =
786
  incr complex;
Pietro Abate's avatar
Pietro Abate committed
787
  Stats.Counter.incr count_subtype;
788
  (* XXX here I call leafconj a zilliontime. REWRITE !!! *)
789
  if d.absent then slot_nempty Witness.WAbsent
Pietro Abate's avatar
Pietro Abate committed
790
  else if not (Intervals.is_empty (BoolIntervals.leafconj d.ints))
791
  then slot_nempty (Witness.WInt (Intervals.sample (BoolIntervals.leafconj d.ints)))
Pietro Abate's avatar
Pietro Abate committed
792
  else if not (Atoms.is_empty (BoolAtoms.leafconj d.atoms))
793
  then slot_nempty (Witness.WAtom (Atoms.sample (BoolAtoms.leafconj d.atoms)))
Pietro Abate's avatar
Pietro Abate committed
794
  else if not (Chars.is_empty (BoolChars.leafconj d.chars))
795
  then slot_nempty (Witness.WChar (Chars.sample (BoolChars.leafconj d.chars)))
796 797
  else if not (Abstracts.is_empty (BoolAbstracts.leafconj d.abstract))
  then slot_nempty (Witness.WAbstract (Abstracts.sample (BoolAbstracts.leafconj d.abstract)))
798 799
  else try
         DescrHash.find memo d
800 801 802 803
  with Not_found ->
    let s = { status = Maybe; active = false; notify = Nothing } in
    DescrHash.add memo d s;
    (try
804
       iter_s s check_times (Pair.get (BoolPair.leafconj d.times));
Pietro Abate's avatar
Pietro Abate committed
805
       iter_s s check_xml (Pair.get (BoolPair.leafconj d.xml));
806 807
       iter_s s check_arrow (Pair.get (BoolPair.leafconj d.arrow));
       iter_s s check_record (get_record (BoolRec.leafconj d.record));
808
       if s.active then marks := s :: !marks else s.status <- Empty;
809
     with NotEmpty -> ());
810 811
    s

812 813 814 815 816
and guard n t f = match (slot t) with
  | { status = Empty } -> ()
  | { status = Maybe } as s -> n.active <- true; s.notify <- Do (n,f,s.notify)
  | { status = NEmpty v } -> f v

817
and check_times (left,right) s =
818 819 820 821
  let rec aux w1 w2 accu1 accu2 seen = function
    (* Find a product in right which contains (w1,w2) *)
    | [] -> (* no such product: the current witness is in the difference. *)
	set s (Witness.wpair w1 w2)
Pietro Abate's avatar
Pietro Abate committed
822
    | (n1,n2) :: rest
823 824
	when (Witness.node_has n1 w1) && (Witness.node_has n2 w2) ->
	let right = seen @ rest in
Pietro Abate's avatar
Pietro Abate committed
825
	let accu2' = diff accu2 (descr n2) in
826 827 828
	guard s accu2' (fun w2 -> aux w1 w2 accu1 accu2' [] right);
	let accu1' = diff accu1 (descr n1) in
	guard s accu1' (fun w1 -> aux w1 w2 accu1' accu2 [] right)
829
    | k :: rest -> aux w1 w2 accu1 accu2 (k::seen) rest
830
  in
831 832
  let (t1,t2) = cap_product any any left in
  guard s t1 (fun w1 -> guard s t2 (fun w2 -> aux w1 w2 t1 t2 [] right))
833 834

and check_xml (left,right) s =
835 836 837 838
  let rec aux w1 w2 accu1 accu2 seen = function
    (* Find a product in right which contains (w1,w2) *)
    | [] -> (* no such product: the current witness is in the difference. *)
	set s (Witness.wxml w1 w2)
Pietro Abate's avatar
Pietro Abate committed
839
    | (n1,n2) :: rest
840 841
	when (Witness.node_has n1 w1) && (Witness.node_has n2 w2) ->
	let right = seen @ rest in
Pietro Abate's avatar
Pietro Abate committed
842
	let accu2' = diff accu2 (descr n2) in
843 844 845
	guard s accu2' (fun w2 -> aux w1 w2 accu1 accu2' [] right);
	let accu1' = diff accu1 (descr n1) in
	guard s accu1' (fun w1 -> aux w1 w2 accu1' accu2 [] right)
846
    | k :: rest -> aux w1 w2 accu1 accu2 (k::seen) rest
847
  in
848 849
  let (t1,t2) = cap_product any any_pair left in
  guard s t1 (fun w1 -> guard s t2 (fun w2 -> aux w1 w2 t1 t2 [] right))
850

851
and check_arrow (left,right) s =
852 853
  let single_right f (s1,s2) s =
    let rec aux w1 w2 accu1 accu2 left = match left with
854
      | (t1,t2)::left ->
Pietro Abate's avatar
Pietro Abate committed
855
          let accu1' = diff_t accu1 t1 in
856
          guard s accu1' (fun w1 -> aux w1 w2 accu1' accu2 left);
857

Pietro Abate's avatar
Pietro Abate committed
858
          let accu2' = cap_t  accu2 t2 in
859
          guard s accu2' (fun w2 -> aux w1 (Some w2) accu1 accu2' left)
Pietro Abate's avatar
Pietro Abate committed
860
      | [] ->
861 862
	  let f = match f with Witness.WFun (f,_) -> f | _ -> assert false in
	  set s (Witness.wfun ((w1,w2)::f))
863 864
    in
    let accu1 = descr s1 in
865
    guard s accu1 (fun w1 -> aux w1 None accu1 (neg (descr s2)) left)
866
  in
867
  big_conj single_right right s (Witness.wfun [])
868

869
and check_record (labels,(oleft,left),rights) s =
870 871
  let rec aux ws accus seen = function
    | [] ->
872 873
	let rec aux w i = function
	  | [] -> assert (i == Array.length ws); w
Pietro Abate's avatar
Pietro Abate committed
874 875
	  | l::labs ->
	      let w = match ws.(i) with
876
              | Witness.WAbsent -> w
Pietro Abate's avatar
Pietro Abate committed
877
              | wl -> LabelMap.add l wl w
878 879 880
            in
	      aux w (succ i) labs
      in
881 882
	set s (Witness.wrecord (aux LabelMap.empty 0 labels) oleft)
    | (false,_) :: rest when oleft -> aux ws accus seen rest
Pietro Abate's avatar
Pietro Abate committed
883
    | (_,f) :: rest
884 885 886 887 888 889
	when not (exists (Array.length left)
		    (fun i -> not (Witness.type_has f.(i) ws.(i)))) ->
	(* TODO: a version f get_record which keeps nodes in neg records. *)
	let right = seen @ rest in
	for i = 0 to Array.length left - 1 do
	  let di = diff accus.(i) f.(i) in
Pietro Abate's avatar
Pietro Abate committed
890
	  guard s di (fun wi ->
891 892 893 894 895
			let accus' = Array.copy accus in accus'.(i) <- di;
			let ws' = Array.copy ws in ws'.(i) <- wi;
			aux ws' accus' [] right);
	done
    | k :: rest -> aux ws accus (k::seen) rest
896
  in
897
  let rec start wl i =
898
    if (i < 0) then aux (Array.of_list wl) left [] rights
899
    else guard s left.(i) (fun w -> start (w::wl) (i - 1))
900
  in
901
  start [] (Array.length left - 1)
902

903
let timer_subtype = Stats.Timer.create "Types.is_empty"
904

905
let is_empty d =
906
  Stats.Timer.start timer_subtype;
907
  let s = slot d in
Pietro Abate's avatar
Pietro Abate committed
908 909 910
  List.iter
    (fun s' ->
       if s'.status == Maybe then s'.status <- Empty; s'.notify <- Nothing)
911 912
    !marks;
  marks := [];
913
  Stats.Timer.stop timer_subtype
914
    (s.status == Empty)
915

916 917 918 919
let getwit t = match (slot t).status with NEmpty w -> w | _ -> assert false
  (* Assumes that is_empty has been called on t before. *)

let witness t = if is_empty t then raise Not_found else getwit t
920

921
let non_empty d = not (is_empty d)
922

923 924 925
let disjoint d1 d2 = is_empty (cap d1 d2)

let subtype d1 d2 = is_empty (diff d1 d2)
926

927 928
let equiv d1 d2 = (subtype d1 d2) && (subtype d2 d1)

929

930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952
(* perform some semantic simplifications around type constructors *)
let atom a =
  let atm =
  if Atoms.(is_empty (diff full a)) then
    BoolAtoms.full
  else BoolAtoms.atom (`Atm a)
  in
  { empty with atoms = atm }

let times x y =
  if subtype any x.Node.descr
    && subtype any y.Node.descr
  then
    { empty with times = BoolPair.full }
  else times x y

let xml x y =
  if subtype any x.Node.descr
    && subtype any y.Node.descr
  then
    { empty with xml = BoolPair.full }
  else xml x y

953 954 955 956 957 958 959 960
module Cache = struct

  type 'a cache =
    | Empty
    | Type of t * 'a
    | Split of Witness.witness * 'a cache * 'a cache

  let rec find f t = function
Pietro Abate's avatar
Pietro Abate committed
961
    | Empty ->
962 963
	let r = f t in Type (t,r), r
    | Split (w,yes,no) ->
Pietro Abate's avatar
Pietro Abate committed
964
	if Witness.type_has t w
965 966 967
	then let yes,r = find f t yes in Split (w,yes,no), r
	else let no,r = find f t no in Split (w,yes,no), r
    | Type (s,rs) as c ->
Pietro Abate's avatar
Pietro Abate committed
968 969 970
	let f1 ()=
	  let w = witness (diff t s) in
	  let rt = f t in
971 972
	  Split (w, Type (t,rt), c), rt
	and f2 () =
Pietro Abate's avatar
Pietro Abate committed
973
	  let w = witness (diff s t) in
974 975 976 977 978 979 980 981
	  let rt = f t in
	  Split (w, c, Type (t,rt)), rt in

	if Random.int 2 = 0 then
	  try f1 () with Not_found -> try f2 () with Not_found -> c, rs
	else
	  try f2 () with Not_found -> try f1 () with Not_found -> c, rs

982 983 984 985 986
  let rec lookup t = function
    | Empty -> None
    | Split (w,yes,no) -> lookup t (if Witness.type_has t w then yes else no)
    | Type (s,rs) -> if equiv s t then Some rs else None

987 988 989 990 991 992 993
  let emp = Empty


  let rec dump_cache f ppf = function
    | Empty -> Format.fprintf ppf "Empty"
    | Type (_,s) -> Format.fprintf ppf "*%a" f s
    | Split (w,c1,c2) -> Format.fprintf ppf "?(%a,%a)"
994
	(*Witness.pp w *)(dump_cache f) c1 (dump_cache f) c2
995 996 997 998

  let memo f =
    let c = ref emp in
    fun t ->
999 1000 1001
       let c',r = find f t !c in
       c := c';
       r
1002 1003 1004

end

1005 1006 1007 1008
module Product =
struct
  type t = (descr * descr) list

Pietro Abate's avatar
Pietro Abate committed
1009
  let other ?(kind=`Normal) d =
1010
    match kind with
1011 1012
      | `Normal -> { d with times = empty.times }
      | `XML -> { d with xml = empty.xml }
1013 1014 1015 1016 1017

  let is_product ?kind d = is_empty (other ?kind d)

  let need_second = function _::_::_ -> true | _ -> false

1018 1019 1020
  let normal_aux = function
    | ([] | [ _ ]) as d -> d
    | d ->
1021 1022 1023 1024 1025 1026
      let res = ref [] in

      let add (t1,t2) =
        let rec loop t1 t2 = function
        | [] -> res := (ref (t1,t2)) :: !res
        | ({contents = (d1,d2)} as r)::l ->
Pietro Abate's avatar
Pietro Abate committed
1027
            (*OPT*)
1028
  (*	    if equal_descr d1 t1 then r := (d1,cup d2 t2) else*)
Pietro Abate's avatar
Pietro Abate committed
1029

1030 1031 1032 1033
              let i = cap t1 d1 in
              if is_empty i then loop t1 t2 l
              else (
              r := (i, cup t2 d2);
Pietro Abate's avatar
Pietro Abate committed
1034
              let k = diff d1 t1 in
1035
              if non_empty k then res := (ref (k,d2)) :: !res;
Pietro Abate's avatar
Pietro Abate committed
1036 1037

              let j = diff t1 d1 in
1038 1039 1040 1041
              if non_empty j then loop j t2 l
              )
        in
        loop t1 t2 !res
1042
      in