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
140
141
|
#!/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: dict[str, str] = {
"stack_pointer": "6", # RG
"csr_scratch": "5", # RF
}
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
continue
case "@opt":
opt_name = args[0]
opt_value = constants.get(args[1], args[1])
options[opt_name] = opt_value
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
# def make_spu(reg: str) -> list[int]:
# stack_pointer = options["stack_pointer"]
# reg = eval_symbol(reg)
# words_inc: list[int] = make_inc(stack_pointer, stack_pointer)
# words_str: list[int] = make_str(reg, stack_pointer)
# return words_inc + words_str
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())
if keyword in operations:
words = operations[keyword](address, *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}")
if len(result) < 2 * address + 1:
result.extend((2 * address + 1 - len(result)) * nop)
for (label, label_addr) in labels.items():
if address == label_addr:
print(f"{label}:")
for word in words:
print(f"{address:>08x} 0x{word:>04x} {line}")
result[2 * address + 0] = ((word >> 8) & 0xff)
result[2 * address + 1] = ((word >> 0) & 0xff)
address += 1
with open(outfile_path, "wb") as f:
f.write(result)
print(f"Wrote {len(result)} bytes to {outfile_path}")
|
