Refactor into multipass compiler

1 files changed, 46 insertions, 0 deletions

diff --git a/src/asm_pass3.py b/src/asm_pass3.py
new file mode 100644
index 0000000..5365e3b
--- /dev/null
+++ b/src/asm_pass3.py
@@ -0,0 +1,46 @@
+from dataclasses import dataclass
+from assembler_ops import *
+from assembler_eval import *
+from asm_pass2 import *
+
+@dataclass
+class Result3Line:
+  line_num: int
+  address: int
+  parts: list[str]
+
+@dataclass
+class Result3:
+  lines: list[Result3Line]
+  options: Options
+  operations: OpExpansionDict
+  labels: dict[str, int]
+
+def pass_3(result2: Result2) -> Result3:
+  result_lines: list[Result3Line] = []
+  labels: dict[str, int] = {}
+  address = 0
+
+  for line in result2.lines:
+    keyword, *args = line.parts
+    match keyword:
+      case "@address":
+        address = eval(args[0])
+        continue
+      case "@label":
+        labels[args[0]] = address
+        continue
+      case _:
+        result_lines.append(Result3Line(
+          line_num=line.line_num,
+          parts=line.parts,
+          address=address,
+        ))
+        address += 1
+
+  return Result3(
+    operations=result2.operations,
+    options=result2.options,
+    lines=result_lines,
+    labels=labels,
+  )