28 lines
738 B
Python
28 lines
738 B
Python
from ground_ast import RootNode
|
|
|
|
|
|
class Generator:
|
|
def __init__(self, ast: RootNode, code: str, output_path: str):
|
|
self.ast = ast
|
|
self.lines: list[str] = []
|
|
self.code = code
|
|
self.output_path = output_path
|
|
self.variables = {}
|
|
|
|
def init(self):
|
|
pass
|
|
|
|
def generate_node(self, node):
|
|
self.lines.append(f"; {node}\n\t")
|
|
node_type = str(type(node))[19:-2]
|
|
if not hasattr(self, f"generate_{node_type}"):
|
|
raise NotImplementedError(f"Generator has no generate method for {node_type}.")
|
|
getattr(self, f"generate_{node_type}")(node)
|
|
|
|
def generate(self):
|
|
for statement in self.ast.statements:
|
|
self.generate_node(statement)
|
|
|
|
def write(self):
|
|
with open(self.output_path + ".asm", "w") as f:
|
|
f.writelines(self.lines) |