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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
import Const from require 'base'
import Scope from require 'scope'
unpack or= table.unpack
hash = (tbl) ->
mt = getmetatable tbl
setmetatable tbl, nil
str = tostring tbl
setmetatable tbl, mt
'#' .. str\sub 10
class ASTNode
-- first pass (outin):
-- * expand macros
-- * define scoped values
-- * evaluate references
expand: (scope) =>
-- second pass (inout):
-- * setup OPs (spawn/patch)
link: =>
class Atom
type: 'Atom'
new: (@raw, @atom_type) =>
unescape = (str) ->
str = str\gsub '\\"', '"'
str = str\gsub "\\'", "'"
str = str\gsub "\\\\", "\\"
str
expand: (scope) =>
@value = switch @atom_type
when 'num'
Const 'num', tonumber @raw
when 'strq', 'strd'
Const 'str', unescape @raw
when 'sym'
assert (scope\get @raw), "undefined reference to symbol '#{@raw}'"
else
error "unknown atom type: '#{@atom_type}'"
link: =>
_walk: => coroutine.yield @type, @
stringify: =>
switch @atom_type
when 'sym', 'num'
@raw
when 'strq'
"'#{@raw}'"
when 'strd'
"\"#{@raw}\""
else
error "unknown atom type: '#{@atom_type}'"
@make_num: (match) -> Atom match, 'num'
@make_sym: (match) -> Atom match, 'sym'
@make_strd: (match) -> Atom match, 'strd'
@make_strq: (match) -> Atom match, 'strq'
__tostring: =>
"<Atom#{hash @} #{@stringify!}>"
class Xpr
type: 'Xpr'
-- either:
-- * style, tag, parts
-- * style, parts
new: (@style, tag, parts) =>
if not parts
parts = tag
tag = nil
@tag = tag
@white = {}
@white[0] = parts[1]
for i = 2,#parts,2
@[i/2] = parts[i]
@white[i/2] = parts[i+1]
expand: (scope) =>
head = @[1]
head\expand scope
@scope = Scope @, scope
if head.value.type == 'macro'
macro = head.value\getc!
macro scope, @
else
for child in *@[2,]
child\expand scope
link: =>
head = @head!
head: => @[1].value
tail: => unpack [p.value for p in *@[2,]]
_walk: (dir, yield_self=true) =>
coroutine.yield @type, @ if yield_self and dir == 'outin'
for frag in *@
frag\_walk dir
coroutine.yield @type, @ if yield_self and dir == 'inout'
walk: (dir, yield_self=true) =>
assert dir == 'inout' or dir == 'outin', "dir has to be either inout or outin"
coroutine.wrap -> @_walk dir, yield_self
stringify: =>
buf = ''
buf ..= @white[0]
for i, frag in ipairs @
buf ..= frag\stringify!
buf ..= @white[i]
switch @style
when 'naked'
buf
when '('
tag = if @tag then "[#{@tag}]" else ''
'(' .. tag .. buf .. ')'
else
error "unknown sexpr style: '#{@style}'"
make_sexpr: (...) -> Xpr '(', ...
make_nexpr: (...) -> Xpr 'naked', ...
__tostring: =>
if @style == 'naked'
return 'ROOT'
buf = "("
buf ..= "[#{@tag}]" if @tag
buf ..= "#{@[1].raw}"
buf ..= " ..." if #@ > 1
buf ..= ")"
buf
{
:Atom
:Xpr
}
|
