blob: f6b3746e7c5243023ecbb3561bac185ad9452ce2 (
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
[functor (#+ Functor)]
[equivalence (#+ Equivalence)]
[fold (#+ Fold)]
[monad (#+ do)]]
[control
["<>" parser
["<c>" code (#+ Parser)]]]
[data
[collection
["." list ("#\." monad fold)]]]
[macro
[syntax (#+ syntax:)]
["." code]]]])
(type: #export (Tree a)
{#value a
#children (List (Tree a))})
(def: #export (flatten tree)
(All [a] (-> (Tree a) (List a)))
(#.Cons (get@ #value tree)
(list\join (list\map flatten (get@ #children tree)))))
(def: #export (leaf value)
(All [a] (-> a (Tree a)))
{#value value
#children (list)})
(def: #export (branch value children)
(All [a] (-> a (List (Tree a)) (Tree a)))
{#value value
#children children})
(type: #rec Tree-Code
[Code (List Tree-Code)])
(def: tree^
(Parser Tree-Code)
(|> (|>> <>.some
<c>.record
(<>.and <c>.any))
<>.rec
<>.some
<c>.record
(<>.default (list))
(<>.and <c>.any)))
(syntax: #export (tree {root tree^})
{#.doc (doc "Tree literals."
(: (Tree Nat)
(tree 10
{20 {}
30 {}
40 {}})))}
(wrap (list (` (~ (loop [[value children] root]
(` {#value (~ value)
#children (list (~+ (list\map recur children)))})))))))
(implementation: #export (equivalence super)
(All [a] (-> (Equivalence a) (Equivalence (Tree a))))
(def: (= tx ty)
(and (\ super = (get@ #value tx) (get@ #value ty))
(\ (list.equivalence (equivalence super)) = (get@ #children tx) (get@ #children ty)))))
(implementation: #export functor
(Functor Tree)
(def: (map f fa)
{#value (f (get@ #value fa))
#children (list\map (map f)
(get@ #children fa))}))
(implementation: #export fold
(Fold Tree)
(def: (fold f init tree)
(list\fold (function (_ tree' init') (fold f init' tree'))
(f (get@ #value tree)
init)
(get@ #children tree))))
|