102 lines
5.4 KiB
Plaintext
102 lines
5.4 KiB
Plaintext
from os import listdir
|
|
class VirtualMachineCompiler:
|
|
def __init__(self,file_path):
|
|
self.file_path=file_path
|
|
self.files=[]
|
|
self.files_content=[]
|
|
self.is_folder=True
|
|
if ".vm" in file_path:
|
|
self.is_folder=False
|
|
self.files.append(file_path)
|
|
self.assembly_commands={
|
|
None:"",
|
|
"add":"@SP\nA=M-1\nD=M\n@SP\nM=M-1\nA=M-1\nM=D+M\n",
|
|
"sub":"@SP\nA=M-1\nD=M\n@SP\nM=M-1\nA=M-1\nM=M-D\n",
|
|
"neg":"@SP\nA=M-1\nM=-M\n",
|
|
"eq":"@SP\nA=M-1\nD=M\n@SP\nM=M-1\nA=M-1\nD=M-D\nM=-1\n@LABEL{number}\nD;JEQ\n@SP\nA=M-1\nM=0\n(LABEL{number})\n",
|
|
"gt":"@SP\nA=M-1\nD=M\n@SP\nM=M-1\nA=M-1\nD=M-D\nM=-1\n@LABEL{number}\nD;JGT\n@SP\nA=M-1\nM=0\n(LABEL{number})\n",
|
|
"lt":"@SP\nA=M-1\nD=M\n@SP\nM=M-1\nA=M-1\nD=M-D\nM=-1\n@LABEL{number}\nD;JLT\n@SP\nA=M-1\nM=0\n(LABEL{number})\n",
|
|
"and":"@SP\nA=M-1\nD=M\n@SP\nM=M-1\nA=M-1\nM=D&M\n",
|
|
"or":"@SP\nA=M-1\nD=M\n@SP\nM=M-1\nA=M-1\nM=D|M\n",
|
|
"not":"@SP\nA=M-1\nM=!M\n",
|
|
"push":"{m_arg}@SP\nM=M+1\nA=M-1\nM=D\n",
|
|
"pop":"{m_arg}D=A\n@R13\nM=D\n@SP\nAM=M-1\nD=M\n@R13\nA=M\nM=D\n",
|
|
"label":"({l_arg})\n",
|
|
"goto":"@{l_arg}\n0;JMP\n",
|
|
"if-goto":"@SP\nAM=M-1\nD=M\n@{l_arg}\nD;JNE\n",
|
|
"function":"({f_arg})\n",
|
|
"call":"@{r_address}\nD=A\n@SP\nM=M+1\nA=M-1\nM=D\n@LCL\nD=M\n@SP\nM=M+1\nA=M-1\nM=D\n@ARG\nD=M\n@SP\nM=M+1\nA=M-1\nM=D\n@THIS\nD=M\n@SP\nM=M+1\nA=M-1\nM=D\n@THAT\nD=M\n@SP\nM=M+1\nA=M-1\nM=D\n@SP\nD=M\n@5\nD=D-A\n@{arg2}\nD=D-A\n@ARG\nM=D\n@SP\nD=M\n@LCL\nM=D\n@{c_arg}\n0;JMP\n({r_address})\n",
|
|
"return":"@LCL\nD=M\n@R14\nM=D\n@R5\nA=D-A\nD=M\n@R15\nM=D\n@ARG\nAD=M\n@R13\nM=D\n@SP\nAM=M-1\nD=M\n@R13\nA=M\nM=D\n@ARG\nD=M+1\n@SP\nM=D\n@R14\nAM=M-1\nD=M\n@THAT\nM=D\n@R14\nAM=M-1\nD=M\n@THIS\nM=D\n@R14\nAM=M-1\nD=M\n@ARG\nM=D\n@R14\nAM=M-1\nD=M\n@LCL\nM=D\n@R15\nA=M\n0;JMP\n"
|
|
}
|
|
self.arguments={
|
|
None:"",
|
|
"argument":"@ARG\nD=M\n@{arg2}\nA=D+A\nD=M\n",
|
|
"local":"@LCL\nD=M\n@{arg2}\nA=D+A\nD=M\n",
|
|
"static":"@{s_arg}\nD=M\n",
|
|
"constant":"@{arg2}\nD=A\n",
|
|
"this":"@THIS\nD=M\n@{arg2}\nA=D+A\nD=M\n",
|
|
"that":"@THAT\nD=M\n@{arg2}\nA=D+A\nD=M\n",
|
|
"pointer":"@R3\nD=A\n@{arg2}\nA=D+A\nD=M\n",
|
|
"temp":"@R5\nD=A\n@{arg2}\nA=D+A\nD=M\n"
|
|
}
|
|
self.label_number=0
|
|
def folder_search(self):
|
|
for item in listdir(self.file_path):
|
|
if ".vm" in item:
|
|
self.files.append(self.file_path+"\\"+item)
|
|
def parser_module(self):
|
|
if self.is_folder:
|
|
self.folder_search()
|
|
for file in self.files:
|
|
file_content=open(file)
|
|
file_array=[]
|
|
for line in file_content:
|
|
command=[None,None,None]
|
|
line=line[:line.find("//")]
|
|
line = line.strip()
|
|
if line =="":
|
|
continue
|
|
line_array=line.split()
|
|
for index in range(len(line_array)):
|
|
command[index]=line_array[index]
|
|
file_array.append(command)
|
|
self.files_content.append(file_array)
|
|
file_content.close()
|
|
def code_writer_module(self):
|
|
file_path=self.file_path+"\\"+self.file_path.split("\\")[-1]
|
|
bootstrap_code="@256\nD=A\n@SP\nM=D\n@BOOT\nD=A\n@SP\nM=M+1\nA=M-1\nM=D\n@LCL\nD=M\n@SP\nM=M+1\nA=M-1\nM=D\n@ARG\nD=M\n@SP\nM=M+1\nA=M-1\nM=D\n@THIS\nD=M\n@SP\nM=M+1\nA=M-1\nM=D\n@THAT\nD=M\n@SP\nM=M+1\nA=M-1\nM=D\n@SP\nD=M\n@5\nD=D-A\n@0\nD=D-A\n@ARG\nM=D\n@SP\nD=M\n@LCL\nM=D\n@Sys.init\n0;JMP\n(BOOT)\n"
|
|
if not self.is_folder:
|
|
file_path=self.file_path[:file_path.find(".")]
|
|
bootstrap_code=""
|
|
assembly_file=open(file_path+".asm","w")
|
|
assembly_file.write(bootstrap_code)
|
|
for file in range(len(self.files_content)):
|
|
current_function=""
|
|
for line in self.files_content[file]:
|
|
assembly_command=self.assembly_commands[line[0]]
|
|
if line[0]=="function":
|
|
current_function=line[1]
|
|
assembly_command=assembly_command.format(f_arg=line[1])+"@SP\nM=M+1\nA=M-1\nM=0\n"*int(line[2])
|
|
elif "m_arg" in assembly_command:
|
|
m_argument=self.arguments[line[1]]
|
|
if line[1] == "static":
|
|
current_file=self.files[file].split("\\")[-1]
|
|
m_argument=m_argument.format(s_arg=current_file[:current_file.find(".")]+"."+line[2])
|
|
else:
|
|
m_argument=m_argument.format(arg2=line[2])
|
|
assembly_command=assembly_command.format(m_arg=m_argument)
|
|
elif "l_arg" in assembly_command:
|
|
assembly_command=assembly_command.format(l_arg=current_function+"$"+line[1])
|
|
elif "c_arg" in assembly_command:
|
|
assembly_command=assembly_command.format(r_address="RETURN"+str(self.label_number),arg2=line[2],c_arg=line[1])
|
|
self.label_number+=1
|
|
else:
|
|
assembly_command=assembly_command.format(number=str(self.label_number))
|
|
self.label_number+=1
|
|
assembly_file.write(assembly_command)
|
|
assembly_file.close()
|
|
|
|
a=VirtualMachineCompiler(input("Enter path: "))
|
|
a.parser_module()
|
|
a.code_writer_module()
|