aboutsummaryrefslogtreecommitdiff
path: root/stdlib/source/lux/data/sum.lux
blob: 6a048153cc399594e8660d02aac468cd49b4acb2 (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
89
90
91
92
93
(.module:
  {#.doc "Functionality for working with variants (particularly 2-variants)."}
  [lux #*
   [abstract
    [equivalence (#+ Equivalence)]
    [hash (#+ Hash)]]])

(template [<name> <type> <right?>]
  [(def: #export (<name> value)
     (All [a b] (-> <type> (| a b)))
     (0 <right?> value))]

  [left  a #0]
  [right b #1])

(def: #export (either fl fr)
  (All [a b c]
    (-> (-> a c) (-> b c)
        (-> (| a b) c)))
  (function (_ input)
    (case input
      (0 #0 l) (fl l)
      (0 #1 r) (fr r))))

(def: #export (each fl fr)
  (All [l l' r r']
    (-> (-> l l') (-> r r')
        (-> (| l r) (| l' r'))))
  (function (_ input)
    (case input
      (0 #0 l) (0 #0 (fl l))
      (0 #1 r) (0 #1 (fr r)))))

(template [<name> <side> <right?>]
  [(def: #export (<name> es)
     (All [a b] (-> (List (| a b)) (List <side>)))
     (case es
       #.Nil
       #.Nil
       
       (#.Cons (0 <right?> x) es')
       (#.Cons [x (<name> es')])
       
       (#.Cons _ es')
       (<name> es')))]

  [lefts  a #0]
  [rights b #1]
  )

(def: #export (partition xs)
  (All [a b] (-> (List (| a b)) [(List a) (List b)]))
  (case xs
    #.Nil
    [#.Nil #.Nil]

    (#.Cons x xs')
    (let [[lefts rights] (partition xs')]
      (case x
        (0 #0 x')  [(#.Cons x' lefts) rights]
        (0 #1 x') [lefts (#.Cons x' rights)]))))

(structure: #export (equivalence l@= r@=)
  (All [l r]
    (-> (Equivalence l) (Equivalence r)
        (Equivalence (| l r))))
  (def: (= reference sample)
    (case [reference sample]
      [(#.Left reference) (#.Left sample)]
      (l@= reference sample)
      
      [(#.Right reference) (#.Right sample)]
      (r@= reference sample)

      _
      false)))

(structure: #export (hash (^open "l@.") (^open "r@."))
  (All [l r]
    (-> (Hash l) (Hash r)
        (Hash (| l r))))

  (def: &equivalence (..equivalence l@= r@=))

  (def: (hash value)
    (case value
      (#.Left value)
      (l@hash value)
      
      (#.Right value)
      (.nat ("lux i64 *"
             (.int 2)
             (.int (r@hash value)))))))