GroundPY/generators/generator.py

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)