dimgigov
636c49e9df
- Add isOwn field to Symbol for tracking own T declarations - Reinitialization after move: assigning to a moved variable removes it from movedVars - Move tracking in let/var initialization: var b: own Box = a moves a - Move tracking in assignment: x = y moves y if y is own T - Move tracking in return: return x moves x if x is own T - Expand borrow_test.nim to 10 tests (all passing)
46 lines
1.0 KiB
Nim
46 lines
1.0 KiB
Nim
import std/tables
|
|
import types, ast, source_location
|
|
|
|
type
|
|
SymbolKind* = enum
|
|
skVar
|
|
skFunc
|
|
skType
|
|
skConst
|
|
skModule
|
|
|
|
Symbol* = ref object
|
|
kind*: SymbolKind
|
|
name*: string
|
|
typ*: Type
|
|
decl*: Decl ## optional back-reference to AST decl
|
|
isMutable*: bool
|
|
isPublic*: bool
|
|
isOwn*: bool ## true if declared as own T
|
|
|
|
Scope* = ref object
|
|
parent*: Scope
|
|
table*: Table[string, Symbol] ## O(1) lookup via hash table
|
|
|
|
proc newScope*(parent: Scope = nil): Scope =
|
|
result = Scope(parent: parent)
|
|
|
|
proc define*(scope: Scope, sym: Symbol): bool =
|
|
## Returns false if name already exists in this scope
|
|
if scope.table.hasKey(sym.name):
|
|
return false
|
|
scope.table[sym.name] = sym
|
|
return true
|
|
|
|
proc lookup*(scope: Scope, name: string): Symbol =
|
|
var cur = scope
|
|
while cur != nil:
|
|
if cur.table.hasKey(name):
|
|
return cur.table[name]
|
|
cur = cur.parent
|
|
return nil
|
|
|
|
proc lookupLocal*(scope: Scope, name: string): Symbol =
|
|
if scope.table.hasKey(name):
|
|
return scope.table[name]
|
|
return nil |