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

cleaning up some stuff and optimising

Browse Source
This commit is contained in:
SpookyDervish
2025-09-02 16:57:56 +10:00
parent ef48978bb5
commit a32598f7f0
4 changed files with 47 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
generators/x86_64.py
View File

@@ -24,17 +24,34 @@ class X86_64Generator(Generator):
self.stack_size -= 1
def get_variable(self, var_name: str, reg: str):
self.push(
f"QWORD [rsp + {(self.stack_size - self.variables.get(var_name)['stack_loc'] - 1) * 8}]"
)
try:
self.push(
f"QWORD [rsp + {(self.stack_size - self.variables.get(var_name, None)['stack_loc'] - 1) * 8}]"
)
except TypeError: # variable doesnt exist
traceback(self.code, "NameError", f"\"{var_name}\" is not defined.")
self.pop(reg)
def get_var_pos(self, var_name: str):
return (self.stack_size - self.variables.get(var_name)['stack_loc'] - 1) * 8
def create_variable(self, var_name: str, starting_value):
new_var = {"stack_loc": (self.stack_size), "type": type(starting_value)}
if type(starting_value) == NumberNode:
self.lines.append(f"mov rax, {starting_value.value}\n\t")
self.push("rax")
elif type(starting_value) == VarRefNode:
self.get_variable(starting_value.var_name, "rax")
#self.lines.append(f"mov QWORD [rsp + {var_pos}], rax\n\t")
self.push("rax")
self.variables[var_name] = new_var
def change_variable(self, var_name: str, new_value):
var_pos = (self.stack_size - self.variables.get(var_name)['stack_loc'] - 1) * 8
var_pos = self.get_var_pos(var_name)
if type(new_value) == NumberNode: # we're changing a variable to a number
self.lines.append(f"mov rax, {new_value.value}\n\t")
self.lines.append(f"mov QWORD [rsp + {var_pos}], rax\n\t")
#self.lines.append(f"mov rax, {new_value.value}\n\t")
self.lines.append(f"mov QWORD [rsp + {var_pos}], {new_value.value}\n\t")
elif type(new_value) == VarRefNode: # we're changing a variable to the value of another variable
self.get_variable(new_value.var_name, "rax")
@@ -70,10 +87,7 @@ class X86_64Generator(Generator):
variable_exists = self.variables.get(node.arguments[0].var_name, None) != None
if not variable_exists: # create a new variable
self.variables[node.arguments[0].var_name] = {"stack_loc": (self.stack_size), "type": type(node.arguments[1])}
if type(node.arguments[1]) == NumberNode:
self.lines.append(f"mov rax, {node.arguments[1].value}\n\t")
self.push("rax")
self.create_variable(node.arguments[0].var_name, node.arguments[1])
else: # modify the existing one
self.change_variable(node.arguments[0].var_name, node.arguments[1])