blob: b61bfb5461b8ae2afec6b03ba1b5b034205dcb4b (
plain)
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
(.module:
[library
[lux #*
[abstract
[equivalence (#+ Equivalence)]
[order (#+ Order)]]
[data
[collection
["." list ("#\." fold)]
[dictionary
["/" ordered]]]]
[type
abstract]]])
(abstract: #export (Set a)
(/.Dictionary a a)
(def: #export new
(All [a] (-> (Order a) (Set a)))
(|>> /.new :abstraction))
(def: #export (member? set elem)
(All [a] (-> (Set a) a Bit))
(/.key? (:representation set) elem))
(template [<type> <name> <alias>]
[(def: #export <name>
(All [a] (-> (Set a) <type>))
(|>> :representation <alias>))]
[(Maybe a) min /.min]
[(Maybe a) max /.max]
[Nat size /.size]
[Bit empty? /.empty?]
)
(def: #export (add elem set)
(All [a] (-> a (Set a) (Set a)))
(|> set :representation (/.put elem elem) :abstraction))
(def: #export (remove elem set)
(All [a] (-> a (Set a) (Set a)))
(|> set :representation (/.remove elem) :abstraction))
(def: #export to_list
(All [a] (-> (Set a) (List a)))
(|>> :representation /.keys))
(def: #export (from_list &order list)
(All [a] (-> (Order a) (List a) (Set a)))
(list\fold add (..new &order) list))
(def: #export (union left right)
(All [a] (-> (Set a) (Set a) (Set a)))
(list\fold ..add right (..to_list left)))
(def: #export (intersection left right)
(All [a] (-> (Set a) (Set a) (Set a)))
(|> (..to_list right)
(list.filter (..member? left))
(..from_list (get@ #/.&order (:representation right)))))
(def: #export (difference param subject)
(All [a] (-> (Set a) (Set a) (Set a)))
(|> (..to_list subject)
(list.filter (|>> (..member? param) not))
(..from_list (get@ #/.&order (:representation subject)))))
(implementation: #export equivalence
(All [a] (Equivalence (Set a)))
(def: (= reference sample)
(\ (list.equivalence (\ (:representation reference) &equivalence))
= (..to_list reference) (..to_list sample))))
)
(def: #export (sub? super sub)
(All [a] (-> (Set a) (Set a) Bit))
(|> sub
..to_list
(list.every? (..member? super))))
(def: #export (super? sub super)
(All [a] (-> (Set a) (Set a) Bit))
(sub? super sub))
|