aboutsummaryrefslogtreecommitdiffstats
path: root/src/asm_pass3.py
blob: 5365e3b480b2cc3194aa7d0947ee31f87c444633 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
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,
  )