compiler_cfg.py

class CFGCompiler:
    def __init__(self):
        self.label_id = 0

    def new_label(self):
        self.label_id += 1
        return f"L{self.label_id}"

    def compile(self, ast):
        program = []

        if isinstance(ast, If):
            end_label = self.new_label()

            # condition
            program.append("LOAD R0 word")
            program.append("IN_INDEX R1 R0")
            program.append("BLACKLIST R3 R0")
            program.append("NOT R3 R3")

            # decision gate
            program.append("AND R4 R1 R3")

            # branch
            program.append(f"JMPZ R4 {end_label}")

            # body
            program.append(f"EXEC {ast.body.name}")

            # end
            program.append(f"LABEL {end_label}")

        return program