-
Notifications
You must be signed in to change notification settings - Fork 0
/
resolver.jl
187 lines (148 loc) · 4.3 KB
/
resolver.jl
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
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
include("parser.jl")
function resolve(expr::Block, scopes, locals)
begin_scope(scopes)
resolve(expr.stmts, scopes, locals)
end_scope(scopes)
end
function resolve(expr::Vector{Stmt}, scopes, locals)
for stmt in expr
resolve(stmt, scopes, locals)
end
end
function resolve(expr::Stmt, scopes, locals)
resolve(expr.expr, scopes, locals)
end
function begin_scope(scopes)
push!(scopes, Dict())
end
function end_scope(scopes)
pop!(scopes)
end
function resolve(expr::Var, scopes, locals)
declare(expr.name.lexeme, scopes)
if !isnothing(expr.val)
resolve(expr.val, scopes, locals)
end
define(expr.name.lexeme, scopes)
end
function declare(name, scopes)
if !isempty(scopes)
if haskey(scopes[end], name)
Base.error("Variable already declared with the same name at this local scope")
end
scopes[end][name] = false
end
end
function define(name, scopes)
if !isempty(scopes)
scopes[end][name] = true
end
end
function resolve(expr::Variable, scopes, locals)
if !isempty(scopes) && get(scopes[end], expr.name, nothing) == false
Base.error("Cannot use a variable in its own definition")
end
resolve_local(expr, expr.name, scopes, locals)
end
function resolve(expr::This, scopes, locals)
resolve_local(expr, expr.keyword.lexeme, scopes, locals)
end
function resolve_local(expr, name, scopes, locals)
for i in reverse(eachindex(scopes))
if haskey(scopes[i], name)
locals[expr] = size(scopes)[1] - i
return
end
end
end
function resolve(expr::Assign, scopes, locals)
resolve(expr.val, scopes, locals)
resolve_local(expr, expr.name, scopes, locals)
end
function resolve(expr::Function, scopes, locals)
declare(expr.name.lexeme, scopes)
define(expr.name.lexeme, scopes)
resolve_function(expr, scopes, locals)
end
function resolve_function(fct::Function, scopes, locals)
begin_scope(scopes)
for param in fct.params
declare(param, scopes)
define(param, scopes)
end
resolve(fct.body, scopes, locals)
end_scope(scopes)
end
function resolve(expr::IfStmt, scopes, locals)
resolve(expr.condition, scopes, locals)
resolve(expr.thenBr, scopes, locals)
if !isnothing(expr.elseBr)
resolve(expr.elseBr, scopes, locals)
end
end
function resolve(expr::Print, scopes, locals)
resolve(expr.expr, scopes, locals)
end
function resolve(expr::Return, scopes, locals)
if !isnothing(expr.val)
resolve(expr.val, scopes, locals)
end
end
function resolve(expr::While, scopes, locals)
resolve(expr.condition, scopes, locals)
resolve(expr.body, scopes, locals)
end
function resolve(expr::Binary, scopes, locals)
resolve(expr.left, scopes, locals)
resolve(expr.right, scopes, locals)
end
function resolve(expr::Call, scopes, locals)
resolve(expr.callee, scopes, locals)
for arg in expr.args
resolve(arg, scopes, locals)
end
end
function resolve(expr::Grouping, scopes, locals)
resolve(expr.expression, scopes, locals)
end
function resolve(expr::Literal, scopes, locals) end
function resolve(expr::Logical, scopes, locals)
resolve(expr.left, scopes, locals)
resolve(expr.right, scopes, locals)
end
function resolve(expr::Unary, scopes, locals)
resolve(expr.right, scopes, locals)
end
function resolve(expr::Class, scopes, locals)
declare(expr.name, scopes)
define(expr.name, scopes)
if !isnothing(expr.superclass)
if expr.superclass.name == expr.name.lexeme
Base.error("Cannot inherit from own class")
end
resolve(expr.superclass, scopes, locals)
end
if !isnothing(expr.superclass)
begin_scope(scopes)
scopes[end]["super"] = true
end
begin_scope(scopes)
scopes[end]["this"] = true
for method in expr.methods
resolve_function(method, scopes, locals)
end
if !isnothing(expr.superclass)
end_scope(scopes)
end
end_scope(scopes)
end
function resolve(expr::Super, scopes, locals)
resolve_local(expr, expr.keyword.lexeme, scopes, locals)
end
function resolve(expr::Get, scopes, locals)
resolve(expr.obj, scopes, locals)
end
function resolve(expr::Set, scopes, locals)
resolve(expr.obj, scopes, locals)
resolve(expr.val, scopes, locals)
end