blob: c38f6afefdc3cafbc7ecce45e47fd12a650a9dcf (
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
86
87
88
|
(.module:
[lux (#- type)
["." meta]
[abstract
[predicate (#+ Predicate)]]
[macro
[syntax (#+ syntax:)]]
[type (#+ :by_example)
abstract]])
(abstract: #export (Refined t r)
{#value t
#predicate (Predicate t)}
{#.doc "A refined type 'r' of base type 't' using a predicate."}
(type: #export (Refiner t r)
(-> t (Maybe (Refined t r))))
(def: #export (refinement predicate)
(All [t]
(Ex [r]
(-> (Predicate t) (Refiner t r))))
(function (_ un_refined)
(if (predicate un_refined)
(#.Some (:abstraction {#value un_refined
#predicate predicate}))
#.None)))
(template [<name> <output> <slot>]
[(def: #export (<name> refined)
(All [t r] (-> (Refined t r) <output>))
(|> refined :representation (get@ <slot>)))]
[un_refine t #value]
[predicate (Predicate t) #predicate]
)
(def: #export (lift transform)
(All [t r]
(-> (-> t t)
(-> (Refined t r) (Maybe (Refined t r)))))
(function (_ refined)
(let [[value predicate] (:representation refined)
value' (transform value)]
(if (predicate value')
(#.Some (:abstraction {#value value'
#predicate predicate}))
#.None))))
)
(def: #export (filter refiner values)
(All [t r] (-> (Refiner t r) (List t) (List (Refined t r))))
(case values
#.Nil
#.Nil
(#.Cons head tail)
(case (refiner head)
(#.Some refined)
(#.Cons refined (filter refiner tail))
#.None
(filter refiner tail))))
(def: #export (partition refiner values)
(All [t r] (-> (Refiner t r) (List t) [(List (Refined t r)) (List t)]))
(case values
#.Nil
[#.Nil #.Nil]
(#.Cons head tail)
(let [[yes no] (partition refiner tail)]
(case (refiner head)
(#.Some refined)
[(#.Cons refined yes)
no]
#.None
[yes
(#.Cons head no)]))))
(syntax: #export (type refiner)
(meta.with_gensyms [g!t g!r]
(wrap (list (` ((~! :by_example) [(~ g!t) (~ g!r)]
{(..Refiner (~ g!t) (~ g!r))
(~ refiner)}
(..Refined (~ g!t) (~ g!r))))))))
|