compiler.py

INDENT = "    "
def indented(lst):
    return ["%s%s" % (INDENT, s) for s in lst]

def stack_swap(indices):
    max_index = int(max(indices))
    return ["s%s:" % indices] +\
           indented("popdata r%s" % (i+1) for i in range(max_index)) +\
           indented("pushdata r%s" % i for i in reversed(indices)) +\
           indented(["ret"])

array_template = """
%(array)s.len:
    pushdata [%(array)s_length]
    ret
%(array)s.push:
%(array)s.append:
    popdata r1
    mov r2, [%(array)s_length]
    mov %(width_word)s[%(array)s + %(width)s*r2], r1%(width_word)s
    inc qword [%(array)s_length]
    ret
%(array)s.pop:
    dec qword [%(array)s_length]
    mov r1, [%(array)s_length]
    mov%(width_trunc)s r1, %(width_word)s[%(array)s + %(width)s*r1]
    pushdata r1
    ret
%(array)s.get:
    popdata r1
    mov%(width_trunc)s r1, %(width_word)s[%(array)s + %(width)s*r1]
    pushdata r1
    ret
%(array)s.set_at:
    popdata r1
    popdata r2
    mov [%(array)s + %(width)s*r1], r2%(width_word)s
    ret
%(array)s.address:
    popdata r1
    lea r1, [%(array)s + %(width)s*r1]
    pushdata r1
    ret
"""
array_template = array_template.strip()
# print(array_template % {"array": "array", "width": 8, "width_word": "", "width_char": ""})
def array(name, width=8):
    width_word = {8: ("", ""), 1: ("byte ", "zx")}
    return (array_template % {"array": name, "width": width,
                             "width_word": width_word[width][0],
                             "width_trunc": width_word[width][1]}).split("\n")

def array_footer(name, width=8, size=10000):
    width_char = {8: "q", 1: "b"}
    return ["%s_length: resq 1" % name,
            "%s: res%s %s" % (name, width_char[width], size)]

new_names = {"+": "add", "-": "sub", "==": "equal", "[": "write-loop",
             "(": "comment", ">": "greater", "push:": "push"}
old_names = {v: k for k, v in new_names.items()}
def rename(s):
    return new_names.get(s, s).replace("-", "_")

def compile_(commands, stop=None):
    output = []
    while True:
        try:
            command = rename(next(commands))
        except StopIteration:
            return output
        if command == stop:
            return output
        elif command == "write_loop":
            body = compile_(commands, "]")
            assert(next(commands) == "bind:")
            output.append("%s:" % rename(next(commands)))
            output.extend(indented(body + ["ret"]))
        elif command == "comment":
            _ = compile_(commands, ")")
        elif command == "prints(":
            next_command = next(commands)
            inner = []
            while next_command != ")":
                inner.append(next_command)
                next_command = next(commands)
            output.append('prints %s' % " ".join(inner))
        elif command == "pushe:":
            output.append("pushdata %s" % rename(next(commands)))
        elif command == "push":
            next_command = next(commands)
            output.append("pushdata %s" % rename(next_command))
        elif command == "label:":
            output.append("%s:" % rename(next(commands)))
        elif command == "goto:":
            output.append("jmp %s" % rename(next(commands)))
        elif command == "goto_if:":
            output.extend(["popdata r1", "test r1, r1",
                           "jnz %s" % rename(next(commands))])
        elif command == "ret":
            output.append("ret")
        else:
            output.append("call %s" % command)

if __name__ == "__main__":
    import sys
    commands = open("forth.f" if len(sys.argv) < 2 else sys.argv[1]).read()
    commands = iter(commands.split())

    writable = []
    output = open("header.asm").read().split("\n")
    output += compile_(commands)
    output += [line for indices in ["11", "21", "2"]
               for line in stack_swap(indices)]

    def add_array(name, width=8, size=10000):
        output.extend(array(name, width))
        writable.extend(array_footer(name, width, size))

    for name in ["call_stack", "memory"]:
        add_array(name)
    add_array("strings", 1)
    add_array("input_buffer", 1)
    writable.append("input_buffer_counter: resq 1")
    writable.extend(array_footer("data_stack"))

    print("\n".join(output))
    print("section .bss")
    print("\n".join(writable))