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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
#!/usr/bin/env python3
# Assemble ATK16 assembly to bytecode
import sys
from typing import Callable
import importlib
from assembler_ops import *
from assembler_eval import *
if len(sys.argv) != 3:
print("usage: assembler.py <infile> <outfile> # read from file")
print(" assembler.py - <outfile> # read from stdin")
sys.exit(1)
infile_path = sys.argv[1]
outfile_path = sys.argv[2]
src = ""
if (infile_path == "-"):
for line in sys.stdin:
src += line
else:
with open(infile_path, "r") as f:
src = f.read()
src_lines = src.splitlines()
### Utils
def parse(line: str) -> list[str]:
depth = 0
result: list[str] = []
acc: str = ""
for c in line:
if c.isspace() and depth == 0:
result.append(acc)
acc = ""
elif c == "(":
depth += 1
acc += "("
elif c == ")":
depth -= 1
acc += ")"
else:
acc += c
result.append(acc)
return list(filter(lambda x: len(x) > 0, result))
# 1st pass, gather labels, set options, eval operations
options = Options()
address: int = 0
for (lineNo, line) in enumerate(src_lines):
line = line.split(";")[0].strip()
if line == "": continue
keyword, *args = line.lower().split()
match keyword:
case "@address":
address = eval(args[0])
continue
case "@label":
labels[args[0]] = address
# todo bugi: 1st pass ei ota monirivisiä käskyjä huomioon
continue
case "@opt":
opt_name, opt_value = args
match opt_name:
case "stack_pointer": options.stack_pointer = opt_value
case "csr_scratch": options.csr_scratch = opt_value
case _: raise Exception("Unknown @opt: " + opt_name)
case "@use":
module_name, ops = args[0].split(":")
ops_split = ops.split(",")
module = importlib.import_module(module_name)
mod_operations: dict[str, Callable[..., list[int]]] = module.operations
for op in mod_operations:
callable = mod_operations[op]
if ops == "*" or op in ops_split:
operations[op] = callable
case _:
pass
address += 1
# 2nd pass
address: int = 0
nop = bytearray([0b1000_0000, 0])
result = bytearray()
# initially one nop
result.extend(nop)
for (lineNo, line) in enumerate(src_lines):
line = line.split(";")[0].strip()
if line == "": continue
keyword, *args = parse(line.lower())
meta = Meta(
address=address,
options=options
)
if keyword in operations:
words = operations[keyword](meta, *args)
else:
match keyword:
# Directives
case "@address":
address = eval_expr(args[0])
continue
case "@label" | "@opt" | "@use":
continue
# Default case: evaluate as is (e.g. data word)
case _:
try:
words = [eval_expr(keyword)]
except:
raise Exception(f"Invalid assembly at {infile_path}:{lineNo + 1}\n\n{line}")
for (label, label_addr) in labels.items():
if address == label_addr:
print(f"{label}:")
line_printed = False
for word in words:
if len(result) < 2 * address + 1:
result.extend((2 * address + 1 - len(result)) * nop)
print(f"{address:>08x} 0x{word:>04x} {line if not line_printed else '...'}")
result[2 * address + 0] = ((word >> 8) & 0xff)
result[2 * address + 1] = ((word >> 0) & 0xff)
address += 1
line_printed = True
with open(outfile_path, "wb") as f:
f.write(result)
print(f"Wrote {len(result)} bytes to {outfile_path}")
|
