Source file instances.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
open! Base
open Base_quickcheck
open Base_test_coverage_helpers
open Overrides
include Instances_intf.Definitions

module Make_elt
    (Types : Functor.Types with type 'elt elt = 'elt)
    (Cmp : sig
       val comparator : (int, Int.comparator_witness Types.cmp) Comparator.t
     end) : Functor.Elt with module Types := Types = struct
  type t = int [@@deriving compare, equal, quickcheck, sexp_of]
  type comparator_witness = Int.comparator_witness Types.cmp

  let comparator = Cmp.comparator

  include (Int : Comparable.Infix with type t := t)

  let of_int = Fn.id
  let to_int = Fn.id
end

module Poly = struct
  module Types = Types.Poly
  module Elt = Make_elt (Types) (Comparator.Poly)

  type t = Set.M(Elt).t [@@deriving compare, equal, sexp_of]

  let create x = x
  let access x = x
end

module Toplevel = struct
  module Types = Types.Toplevel
  module Elt = Make_elt (Types) (Int)

  type t = Set.M(Elt).t [@@deriving compare, equal, sexp_of]

  let create f = f ((module Int) : _ Comparator.Module.t)
  let access x = x
end

module Tree = struct
  module Types = Types.Tree
  module Elt = Make_elt (Types) (Int)

  type t = (int, Int.comparator_witness) Types.t

  let compare = Comparable.lift Toplevel.compare ~f:(Set.of_tree (module Elt))
  let equal = Comparable.lift Toplevel.equal ~f:(Set.of_tree (module Elt))
  let sexp_of_t t = Toplevel.sexp_of_t (Set.of_tree (module Elt) t)
  let create f = f ~comparator:Int.comparator
  let access f = f ~comparator:Int.comparator
end

module Using_comparator = struct
  module Types = Types.Using_comparator
  module Elt = Make_elt (Types) (Int)

  type t = Set.M(Elt).t [@@deriving compare, equal, sexp_of]

  let create f = f ~comparator:Int.comparator
  let access x = x
end