ground/GroundPY
2
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed some scope issues and added label and line reference support to the AST

Browse Source
This commit is contained in:
SpookyDervish
2025-09-01 07:01:30 +10:00
parent eee9325ab8
commit 62e95a24ed
2 changed files with 36 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
ground_ast.py
View File

@@ -43,6 +43,15 @@ class FunctionNode:
parent: FunctionNode | RootNode
return_type: Optional[str] = None
name: Optional[str] = None
@dataclass
class LabelDecNode:
name: str
@dataclass
class LabelRefNode:
name: str
@dataclass
class LineRefNode:
line: int
def generate_ast(tokens: list[Token], code: str) -> RootNode:
root_node = RootNode([])
@@ -55,7 +64,7 @@ def generate_ast(tokens: list[Token], code: str) -> RootNode:
# todo: this is the absolute WORST way i could do this, but i could not care less lmao
# its not even performant......
for token in tokens:
print(token)
#print(token.type)
if token.type == TokenType.INSTRUCTION:
if current_node:
scope.statements.append(current_node)
@@ -68,9 +77,15 @@ def generate_ast(tokens: list[Token], code: str) -> RootNode:
current_node_type = "inst"
if current_node.instruction == "endfun":
scope = scope.parent # go up one scope
if scope == root_node:
traceback(code, "ScopeError", "You can't use endfun from global scope, did you mean to use the \"end\" instruction?")
current_node.parent = scope
scope = scope.parent # go up one scope
current_node.parent = scope
scope.statements.append(current_node)
current_node = scope
else:
current_node.parent = scope
else:
current_node = FunctionNode([], [], scope)
current_node_type = "func"
@@ -88,11 +103,22 @@ def generate_ast(tokens: list[Token], code: str) -> RootNode:
else:
traceback(code, "SyntaxError", "Expected instruction, not number.")
elif token.type == TokenType.LINE_REFERENCE:
if current_node_type == "inst":
current_node.arguments.append(LineRefNode(token.value))
else:
traceback(code, "SyntaxError", "Expected instruction, not line reference")
elif token.type == TokenType.LABEL_REFERENCE:
if current_node_type == "inst":
current_node.arguments.append(LabelRefNode(token.value))
else:
traceback(code, "SyntaxError", "Expected instruction, not label reference")
elif token.type == TokenType.VARIABLE_POINTER:
if current_node_type == "inst":
current_node.arguments.append(VarPointerNode(token.value))
elif last_token and last_token.type == TokenType.TYPE and current_node_type == "func":
print(current_node)
current_node.args[-1].name = token.value
else:
traceback(code, "SyntaxError", "Expected instruction, not variable pointer.")
@@ -126,6 +152,9 @@ def generate_ast(tokens: list[Token], code: str) -> RootNode:
else:
traceback(code, "SyntaxError", "Expected instruction or function return type, got function reference.")
elif token.type == TokenType.LABEL_DECLERATION:
scope.statements.append(LabelDecNode(token.value))
elif token.type == TokenType.EOF:
root_node.statements.append(current_node)
last_token = token