diff --git a/VM-Compiler b/VM-Compiler index b3b623c..c122470 100644 --- a/VM-Compiler +++ b/VM-Compiler @@ -13,31 +13,31 @@ class VirtualMachineCompiler: "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\nM=0\n(LABLE{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\nM=0\n(LABLE{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\nM=0\n(LABLE{number})\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}D=M\n@SP\nM=M+1\nA=M-1\nM=D\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\nAM=M-1\nM=D\n@LCL\nD=M\n@SP\nAM=M-1\nM=D\n@ARG\nD=M\n@SP\nAM=M-1\nM=D\n@THIS\nD=M\n@SP\nAM=M-1\nM=D\n@THAT\nD=M\n@SP\nAM=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@{f_arg}\n({r_address})\n", - "return":"@LCL\nD=M\n@R5\nM=D\nD=D-A\n@R6\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@R5\nAM=M-1\nD=M\n@THAT\nM=D\n@R5\nAM=M-1\nD=M\n@THIS\nM=D\n@R5\nAM=M-1\nD=M\n@ARG\nM=D\n@R5\nAM=M-1\nD=M\n@LCL\nM=D\n@R6\nA=M\n0;JMP\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+M\n", - "local":"@LCL\nD=M\n@{arg2}\nA=D+M\n", - "static":"@{s_arg}\n", - "constant":"@{arg2}\n", - "this":"@THIS\nD=M\n@{arg2}\nA=D+M\n", - "that":"@THAT\nD=M\n@{arg2}\nA=D+M\n", - "pointer":"@R3\nD=M\n@{arg2}\nA=D+M\n", - "temp":"@R5\nD=M\n@{arg2}\nA=D+M\n" + "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): @@ -63,12 +63,39 @@ class VirtualMachineCompiler: self.files_content.append(file_array) file_content.close() def code_writer_module(self): - file_path=self.file_path + 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=file_path[:file_path.find(".")] - assembly_file=open(file_path,"w") + 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]: - continue + 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()