diff options
Diffstat (limited to 'projects')
| -rw-r--r-- | projects/hack-vm/__main__.py | 3 | ||||
| -rw-r--r-- | projects/hack-vm/branching.py | 24 |
2 files changed, 27 insertions, 0 deletions
diff --git a/projects/hack-vm/__main__.py b/projects/hack-vm/__main__.py index a1dce1f..5373fe9 100644 --- a/projects/hack-vm/__main__.py +++ b/projects/hack-vm/__main__.py @@ -2,6 +2,7 @@ from argparse import ArgumentParser from .memory import translate_memory from .arith_logic import translate_arith_logic from .compare import translate_compare +from .branching import translate_branching from .utils import * @@ -38,6 +39,8 @@ def vm_translate(input_path): asm += translate_arith_logic(command) case [("lt" | "eq" | "gt") as command]: asm += translate_compare(command) + case [("label" | "goto" | "if-goto") as action, label]: + asm += translate_branching(action, label) case _: exit_on_error( f"Syntax error: invalid command: {cmd}", EXIT_CODE_SYNTAX_ERROR diff --git a/projects/hack-vm/branching.py b/projects/hack-vm/branching.py new file mode 100644 index 0000000..b73ed94 --- /dev/null +++ b/projects/hack-vm/branching.py @@ -0,0 +1,24 @@ +from .utils import * + +LABEL_ASM = "({label})\n" + +GOTO_ASM = """@{label} +0;JMP +""" + +IF_GOTO_ASM = """@SP +AM=M-1 +D=M +@{label} +D;JNE +""" + +BRANCHING_ASM = { + "label": LABEL_ASM, + "goto": GOTO_ASM, + "if-goto": IF_GOTO_ASM, +} + + +def translate_branching(action, label): + return BRANCHING_ASM[action].format(label=label) |