.
This commit is contained in:
QkoSad
Executable
+246
@@ -0,0 +1,246 @@
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
def clear_file(clearfile):
|
||||
temp_string_cf = '' # clears the empty lines, comments and spaces and writes them in a string
|
||||
for vmline in clearfile:
|
||||
flag = 0
|
||||
if vmline == '\n':
|
||||
continue
|
||||
for index in range(len(vmline)):
|
||||
if vmline[index:index + 2] == '//':
|
||||
if flag == 1:
|
||||
temp_string_cf += '\n'
|
||||
break
|
||||
if flag == 1 and vmline[index:index + 2] == ' ':
|
||||
temp_string_cf += '\n'
|
||||
break
|
||||
temp_string_cf += vmline[index]
|
||||
flag = 1
|
||||
return temp_string_cf
|
||||
|
||||
|
||||
def push(pushline, funcname):
|
||||
if 'this' in pushline:
|
||||
number = int(pushline[10:])
|
||||
pushline = '@' + str(number) + '\nD=A\n@THIS\nA=M+D\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n//push THIS\n'
|
||||
elif 'that' in pushline:
|
||||
number = int(pushline[10:])
|
||||
pushline = '@' + str(number) + '\nD=A\n@THAT\nA=M+D\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n//push THAT\n'
|
||||
elif 'temp' in pushline:
|
||||
number = int(pushline[10:])
|
||||
number += 5
|
||||
pushline = '@' + str(number) + '\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n//push TMP\n'
|
||||
elif 'local' in pushline:
|
||||
number = int(pushline[11:])
|
||||
pushline = '@' + str(number) + '\nD=A\n@LCL\nA=M+D\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n//push LOCAL\n'
|
||||
elif 'static' in pushline:
|
||||
number = int(pushline[12:])
|
||||
pushline = '@' + funcname + '.' + str(number) + '\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n//push STATIC\n'
|
||||
elif 'pointer' in pushline:
|
||||
number = int(pushline[13:])
|
||||
number += 3
|
||||
pushline = '@' + str(number) + '\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n//push POINTER\n'
|
||||
elif 'argument' in pushline:
|
||||
number = int(pushline[14:])
|
||||
pushline = '@' + str(number) + '\nD=A\n@ARG\nA=M+D\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n//push ARGUMENT\n'
|
||||
elif 'constant' in pushline:
|
||||
number = pushline[14:]
|
||||
pushline = '@' + str(number) + '\nD=A\n@SP\nA=M\nM=D\n@SP\nM=M+1\n//push CONSTANT\n'
|
||||
return pushline
|
||||
|
||||
|
||||
def pop(popline, funcname):
|
||||
if 'this' in popline:
|
||||
number = int(popline[9:])
|
||||
popline = '@' + str(number) + '\nD=A\n@THIS\nM=M+D\n@SP\nM=M-1\nA=M\nD=M\n@THIS\nA=M\nM=D\n' + '@' + str(
|
||||
number) + '\nD=A\n@THIS\nM=M-D\n//pop THIS\n'
|
||||
elif 'that' in popline:
|
||||
number = int(popline[9:])
|
||||
popline = '@' + str(number) + '\nD=A\n@THAT\nM=M+D\n@SP\nM=M-1\nA=M\nD=M\n@THAT\nA=M\nM=D\n' + '@' + str(
|
||||
number) + '\nD=A\n@THAT\nM=M-D\n//pop THAT\n'
|
||||
elif 'temp' in popline:
|
||||
number = int(popline[9:])
|
||||
number = number + 5
|
||||
popline = '@SP\nM=M-1\nA=M\nD=M\n@' + str(number) + '\nM=D\n//pop TMP\n'
|
||||
elif 'local' in popline:
|
||||
number = int(popline[10:])
|
||||
popline = '@' + str(number) + '\nD=A\n@LCL\nM=M+D\n@SP\nM=M-1\nA=M\nD=M\n@LCL\nA=M\nM=D\n' + '@' + str(
|
||||
number) + '\nD=A\n@LCL\nM=M-D\n//pop LOCAL\n'
|
||||
elif 'static' in popline:
|
||||
number = int(popline[11:])
|
||||
popline = '@SP\nM=M-1\nA=M\nD=M\n@' + funcname + '.' + str(number) + '\nM=D\n//pop STATIC\n'
|
||||
elif 'pointer' in popline:
|
||||
number = int(popline[12:])
|
||||
number = number + 3
|
||||
popline = '@SP\nM=M-1\nA=M\nD=M\n@' + str(number) + '\nM=D\n//pop POINTER\n'
|
||||
elif 'argument' in popline:
|
||||
number = int(popline[13:])
|
||||
popline = '@' + str(number) + '\nD=A\n@ARG\nM=M+D\n@SP\nM=M-1\nA=M\nD=M\n@ARG\nA=M\nM=D\n' + '@' + str(
|
||||
number) + '\nD=A\n@ARG\nM=M-D\n//pop ARG\n'
|
||||
return popline
|
||||
|
||||
|
||||
def aritmetic(armline):
|
||||
global trueCounter
|
||||
global falseCounter
|
||||
if armline == 'add':
|
||||
armline = "@SP\nM=M-1\nA=M\nD=M\nA=A-1\nM=M+D\n//ADD\n"
|
||||
elif armline == 'sub':
|
||||
armline = "@SP\nM=M-1\nA=M\nD=M\nA=A-1\nM=M-D\n//SUB\n"
|
||||
elif armline == 'neg':
|
||||
armline = "@SP\nA=M\nA=A-1\nM=-M\n//NEG\n"
|
||||
elif armline == 'eq':
|
||||
armline = '@SP\nM=M-1\nA=M\nD=M\nA=A-1\nD=M-D\n@TRUE' + str(
|
||||
trueCounter) + '\nD;JEQ\n@SP\nA=M\nA=A-1\nM=0\n@FALSE' + str(
|
||||
falseCounter) + '\n0;JMP\n(TRUE' + str(trueCounter) + ')\n@SP\nA=M\nA=A-1\nM=-1\n(FALSE' + str(
|
||||
falseCounter) + ')\n//EQ\n'
|
||||
trueCounter += 1
|
||||
falseCounter += 1
|
||||
elif armline == 'gt':
|
||||
armline = '@SP\nM=M-1\nA=M\nD=M\nA=A-1\nD=M-D\n@TRUE' + str(
|
||||
trueCounter) + '\nD;JGT\n@SP\nA=M\nA=A-1\nM=0\n@FALSE' + str(
|
||||
falseCounter) + '\n0;JMP\n(TRUE' + str(trueCounter) + ')\n@SP\nA=M\nA=A-1\nM=-1\n(FALSE' + str(
|
||||
falseCounter) + ')\n//GT\n'
|
||||
trueCounter += 1
|
||||
falseCounter += 1
|
||||
elif armline == 'lt':
|
||||
armline = '@SP\nM=M-1\nA=M\nD=M\nA=A-1\nD=M-D\n@TRUE' + str(
|
||||
trueCounter) + '\nD;JLT\n@SP\nA=M\nA=A-1\nM=0\n@FALSE' + str(
|
||||
falseCounter) + '\n0;JMP\n(TRUE' + str(trueCounter) + ')\n@SP\nA=M\nA=A-1\nM=-1\n(FALSE' + str(
|
||||
falseCounter) + ')\n//LT\n'
|
||||
trueCounter += 1
|
||||
falseCounter += 1
|
||||
elif armline == 'and':
|
||||
armline = "@SP\nM=M-1\nA=M\nD=M\nA=A-1\nM=M&D\n//AND\n"
|
||||
elif armline == 'or':
|
||||
armline = "@SP\nM=M-1\nA=M\nD=M\nA=A-1\nM=M|D\n//OR\n"
|
||||
elif armline == 'not':
|
||||
armline = "@SP\nA=M\nA=A-1\nM=!M\n//NOT\n"
|
||||
return armline
|
||||
|
||||
|
||||
def label(labelline):
|
||||
labelline = '(' + func_label_name + '$' + labelline[6:] + ')\n//LABEL\n'
|
||||
return labelline
|
||||
|
||||
|
||||
def goto(gotoline):
|
||||
gotoline = '@' + func_label_name + '$' + gotoline[5:] + '\n0;JMP\n//GOTO\n'
|
||||
return gotoline
|
||||
|
||||
|
||||
def if_goto(ifline):
|
||||
ifline = '@SP\nM=M-1\nA=M\nD=M\n@' + func_label_name + '$' + ifline[8:] + '\nD;JNE\n//IFGOTO\n'
|
||||
return ifline
|
||||
|
||||
|
||||
def define_func(funcline):
|
||||
global func_label_name
|
||||
def_func_list = funcline.rsplit(' ')
|
||||
func_label_name = def_func_list[1]
|
||||
funcline = '(' + def_func_list[1] + ')\n'
|
||||
for ind in range(int(def_func_list[2])):
|
||||
funcline += '@0\nD=A\n@SP\nA=M\nM=D\n@SP\nM=M+1\n//DEFINEFUNC\n'
|
||||
return funcline
|
||||
|
||||
|
||||
def call_func(callline):
|
||||
global f
|
||||
call_func_list = callline.rsplit(' ')
|
||||
callline = '@' + call_func_list[1] + str(f) + 'RA\nD=A\n@SP\nA=M\nM=D\n@SP\nM=M+1\n'
|
||||
callline += '@LCL\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n'
|
||||
callline += '@ARG\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n'
|
||||
callline += '@THIS\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n'
|
||||
callline += '@THAT\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n'
|
||||
callline += '@SP\nD=M\n@' + call_func_list[2] + '\nD=D-A\n@5\nD=D-A\n@ARG\nM=D\n'
|
||||
callline += '@SP\nD=M\n@LCL\nM=D\n'
|
||||
callline += '@' + call_func_list[1] + '\n0;JMP\n'
|
||||
callline += '(' + call_func_list[1] + str(f) + 'RA)\n//CALLFUNC\n'
|
||||
f += 1
|
||||
return callline
|
||||
|
||||
|
||||
def return_func(returnline):
|
||||
returnline = '@LCL\nD=M\n@R13\nM=D\n'
|
||||
returnline += '@5\nD=A\n@R13\nD=M-D\nA=D\nD=M\n@R14\nM=D\n'
|
||||
returnline += '@SP\nA=M-1\nD=M\n@ARG\nA=M\nM=D\n'
|
||||
returnline += '@ARG\nD=M+1\n@SP\nM=D\n'
|
||||
returnline += '@R13\nM=M-1\nA=M\nD=M\n@THAT\nM=D\n'
|
||||
returnline += '@R13\nM=M-1\nA=M\nD=M\n@THIS\nM=D\n'
|
||||
returnline += '@R13\nM=M-1\nA=M\nD=M\n@ARG\nM=D\n'
|
||||
returnline += '@R13\nM=M-1\nA=M\nD=M\n@LCL\nM=D\n'
|
||||
returnline += '@R14\nA=M\n0;JMP\n//RETURN\n'
|
||||
return returnline
|
||||
|
||||
|
||||
def iterate(name_iterate, yourpath_iterate):
|
||||
line = ''
|
||||
new_string = ''
|
||||
with open(Path(yourpath_iterate, name_iterate + '.vm'), "r+") as vmfile:
|
||||
string_iterate = clear_file(vmfile)
|
||||
for j in string_iterate: # zapisva liniqta v string koito posle se obrabotva
|
||||
if j != '\n':
|
||||
line += j
|
||||
else:
|
||||
if line[:4] == 'push':
|
||||
new_string += push(line, name_iterate)
|
||||
line = ''
|
||||
elif line[:3] == 'pop':
|
||||
new_string += pop(line, name_iterate)
|
||||
line = ''
|
||||
elif line[:2] in aritmeticList2 or line[:3] in aritmeticList3:
|
||||
new_string += aritmetic(line)
|
||||
line = ''
|
||||
elif line[:5] == 'label':
|
||||
new_string += label(line)
|
||||
line = ''
|
||||
elif line[:4] == 'goto':
|
||||
new_string += goto(line)
|
||||
line = ''
|
||||
elif line[:7] == 'if-goto':
|
||||
new_string += if_goto(line)
|
||||
line = ''
|
||||
elif line[:8] == 'function':
|
||||
new_string += define_func(line)
|
||||
line = ''
|
||||
elif line[:4] == 'call':
|
||||
new_string += call_func(line)
|
||||
line = ''
|
||||
elif line[:6] == 'return':
|
||||
new_string += return_func(line)
|
||||
line = ''
|
||||
return new_string
|
||||
|
||||
|
||||
trueCounter = falseCounter = f = 0
|
||||
staticCounter = 16
|
||||
func_label_name = ''
|
||||
aritmeticList3 = ['add', 'sub', 'neg', 'and', 'not']
|
||||
aritmeticList2 = ['or', 'gt', 'lt', 'eq']
|
||||
yourpath = os.getcwd()
|
||||
|
||||
bootstrap = '''@256\nD=A\n@SP\nM=D\n@MainLoopXYZRA\nD=A\n@SP\nA=M\nM=D\n@SP\nM=M+1\n@LCL\nD=M\n@SP\nA=M\nM=D\n@SP
|
||||
M=M+1\n@ARG\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n@THIS\nD=M\n@SP\nA=M\nM=D\n@SP\nM=M+1\n@THAT\nD=M\n@SP\nA=M\nM=D\n@SP
|
||||
M=M+1\n@SP\nD=M\n@0\nD=D-A\n@5\nD=D-A\n@ARG\nM=D\n@SP\nD=M\n@LCL\nM=D\n@Sys.init\n0;JMP\n(MainLoopXYZRARA)
|
||||
//CALLFUNC\n'''
|
||||
|
||||
for root, dirs, files in os.walk(yourpath, topdown=False):
|
||||
nameslist = []
|
||||
sysstring, functstring = '', ''
|
||||
for name in files:
|
||||
tempString = ''
|
||||
if name[-3:] == '.vm':
|
||||
nameslist += [name[:-3]]
|
||||
for i in nameslist:
|
||||
if i == 'Sys':
|
||||
sysstring = bootstrap
|
||||
sysstring += iterate(i, root)
|
||||
|
||||
else:
|
||||
functstring += iterate(i, root)
|
||||
if nameslist:
|
||||
with open(Path(root, f'{os.path.split(root)[1]}.asm'), 'w+') as newfile:
|
||||
newfile.write(sysstring + functstring)
|
||||
|
||||
Reference in New Issue
Block a user