2020-06-29 13:45:27 +00:00
|
|
|
package compiler
|
|
|
|
|
|
|
|
type varScope struct {
|
|
|
|
localsCnt int
|
|
|
|
argCnt int
|
|
|
|
arguments map[string]int
|
2021-02-04 13:26:33 +00:00
|
|
|
locals []map[string]varInfo
|
|
|
|
}
|
|
|
|
|
|
|
|
type varInfo struct {
|
|
|
|
refType varType
|
|
|
|
index int
|
2020-06-29 13:45:27 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func newVarScope() varScope {
|
|
|
|
return varScope{
|
|
|
|
arguments: make(map[string]int),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *varScope) newScope() {
|
2021-02-04 13:26:33 +00:00
|
|
|
c.locals = append(c.locals, map[string]varInfo{})
|
2020-06-29 13:45:27 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (c *varScope) dropScope() {
|
|
|
|
c.locals = c.locals[:len(c.locals)-1]
|
|
|
|
}
|
|
|
|
|
2021-02-04 13:26:33 +00:00
|
|
|
func (c *varScope) addAlias(name string, vt varType, index int) {
|
|
|
|
c.locals[len(c.locals)-1][name] = varInfo{
|
|
|
|
refType: vt,
|
|
|
|
index: index,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-06-29 13:45:27 +00:00
|
|
|
func (c *varScope) getVarIndex(name string) (varType, int) {
|
|
|
|
for i := len(c.locals) - 1; i >= 0; i-- {
|
2021-02-04 13:26:33 +00:00
|
|
|
if vi, ok := c.locals[i][name]; ok {
|
|
|
|
return vi.refType, vi.index
|
2020-06-29 13:45:27 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if i, ok := c.arguments[name]; ok {
|
|
|
|
return varArgument, i
|
|
|
|
}
|
|
|
|
return 0, -1
|
|
|
|
}
|
|
|
|
|
|
|
|
// newVariable creates a new local variable or argument in the scope of the function.
|
|
|
|
func (c *varScope) newVariable(t varType, name string) int {
|
|
|
|
var n int
|
|
|
|
switch t {
|
|
|
|
case varLocal:
|
|
|
|
return c.newLocal(name)
|
|
|
|
case varArgument:
|
|
|
|
_, ok := c.arguments[name]
|
|
|
|
if ok {
|
|
|
|
panic("argument is already allocated")
|
|
|
|
}
|
|
|
|
n = len(c.arguments)
|
|
|
|
c.arguments[name] = n
|
|
|
|
default:
|
|
|
|
panic("invalid type")
|
|
|
|
}
|
|
|
|
return n
|
|
|
|
}
|
|
|
|
|
|
|
|
// newLocal creates a new local variable in the current scope.
|
|
|
|
func (c *varScope) newLocal(name string) int {
|
|
|
|
idx := len(c.locals) - 1
|
|
|
|
m := c.locals[idx]
|
2021-02-04 13:26:33 +00:00
|
|
|
m[name] = varInfo{
|
|
|
|
refType: varLocal,
|
|
|
|
index: c.localsCnt,
|
|
|
|
}
|
2020-06-29 13:45:27 +00:00
|
|
|
c.localsCnt++
|
|
|
|
c.locals[idx] = m
|
|
|
|
return c.localsCnt - 1
|
|
|
|
}
|