if __name__ == "__main__":
if len(sys.argv) != 2:
print(f"Usage: {sys.argv[0]} <{sys.argv[1]}>")
files = []
if sys.argv[1].endswith('.vm'):
files.append(sys.argv[1])
elif os.path.isdir(sys.argv[1]):
files.extend(glob.glob("*.vm"))
cw = CodeWriter(sys.argv[1].strip(".vm") + ".asm")
print(f"Creating {sys.argv[1].strip('.vm') + '.asm'} ...")
for f in files:
p = Parser(f)
cw.file_name = f
while p.advance():
if p.command_type == CommandType.C_ARITHMETIC:
cw.write_arithmetic(p.arg1())
elif p.command_type == CommandType.C_PUSH:
cw.write_push_pop(CommandType.C_PUSH, segment_map[p.arg1()],
p.arg2())
elif p.command_type == CommandType.C_POP:
cw.write_push_pop(CommandType.C_POP, segment_map[p.arg1()],
p.arg2())
print("Success.")
