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
142
143
144
145
146
147
148
149
150
|
from typing import Callable
from assembler_eval import *
from dataclasses import dataclass
@dataclass
class Options:
stack_pointer = "rf"
csr_scratch = "rh"
@dataclass
class Meta:
address: int
def make_alr(meta: Meta, labels: Labels, alu_op: str, left: str, right: str, target: str) -> int:
"""ALR 0000 TTTL LLRR RSSS"""
target_e = eval_expr(labels, target, bits=3)
left_e = eval_expr(labels, left, bits=3)
right_e = eval_expr(labels, right, bits=3)
alu_op_e = eval_expr(labels, alu_op, bits=3)
word = (0b0000 << 12) + \
(target_e << 9) + \
(left_e << 6) + \
(right_e << 3) + \
alu_op_e
return word
def make_ali(meta: Meta, labels: Labels, alu_op: str, left: str, imm: str, target: str) -> int:
"""ALI 0001 TTTL LLII ISSS"""
target_e = eval_expr(labels, target, bits=3)
left_e = eval_expr(labels, left, bits=3)
imm_e = eval_expr(labels, imm, bits=3)
alu_op_e = eval_expr(labels, alu_op, bits=3)
word = (0b0001 << 12) + \
(target_e << 9) + \
(left_e << 6) + \
(imm_e << 3) + \
alu_op_e
return word
def make_ldr(meta: Meta, labels: Labels, addr_reg: str, to_reg: str) -> int:
"""LDR 0010 TTTR RRXX XXXX"""
to_reg_e = eval_expr(labels, to_reg, bits=3)
addr_reg_e = eval_expr(labels, addr_reg, bits=3)
word = (0b0010 << 12) + \
(to_reg_e << 9) + \
(addr_reg_e << 6)
return word
def make_str(meta: Meta, labels: Labels, from_reg: str, addr_reg: str) -> int:
"""STR 0011 XXXL LLRR RXXX"""
from_reg_e = eval_expr(labels, from_reg, bits=3)
addr_reg_e = eval_expr(labels, addr_reg, bits=3)
word = (0b0011 << 12) + \
(addr_reg_e << 6) + \
(from_reg_e << 3)
return word
def make_ldi(meta: Meta, labels: Labels, imm: str, to_reg: str) -> int:
"""LDI 0100 TTTI IIII IIII"""
to_reg_e = eval_expr(labels, to_reg, bits=3)
imm_e = eval_expr(labels, imm, bits=9)
word = (0b0100 << 12) + \
(to_reg_e << 9) + \
imm_e
return word
def make_jpr(meta: Meta, labels: Labels, addr_reg: str) -> int:
"""JPR 0101 XXXR RRXX XXXX"""
addr_reg_e = eval_expr(labels, addr_reg, bits=3)
word = (0b0101 << 12) + \
(addr_reg_e << 6)
return word
def make_jpi(meta: Meta, labels: Labels, imm: str) -> int:
"""JPI 0110 XXXI IIII IIII"""
imm_e = eval_expr(labels, imm, bits=9)
imm_e = imm_e - meta.address - 1
imm_e = imm_e & (0b111111111)
word = (0b0110 << 12) + \
imm_e
return word
def make_brr(meta: Meta, labels: Labels, flag_s: str, addr_reg: str) -> int:
"""brr 0111 XFFR RRXX XXXX"""
flag_s_e = eval_expr(labels, flag_s, bits=2)
addr_reg_e = eval_expr(labels, addr_reg, bits=3)
word = (0b0111 << 12) + \
(flag_s_e << 9) + \
(addr_reg_e << 6)
return word
def make_bri(meta: Meta, labels: Labels, flag_s: str, imm: str) -> int:
"""BRI 1000 XFFI IIII IIII"""
flag_s_e = eval_expr(labels, flag_s, bits=2)
imm_e = eval_expr(labels, imm, bits=9)
imm_e = imm_e - meta.address - 1
imm_e = imm_e & (0b111111111)
word = (0b1000 << 12) + \
(flag_s_e << 9) + \
imm_e
return word
def make_lpc(meta: Meta, labels: Labels, target: str) -> int:
"""LPC 1001 TTTX XXXX XXXX"""
target_e = eval_expr(labels, target, bits=3)
word = (0b1001 << 12) + \
(target_e << 9)
return word
def make_hlt(meta: Meta, labels: Labels) -> int:
"""HLT 1111 XXXX XXXX XXXX"""
word = (0b1111 << 12)
return word
OpWordDict = dict[str, Callable[..., int]]
operations: OpWordDict = {
"alr": make_alr,
"ali": make_ali,
"ldr": make_ldr,
"str": make_str,
"ldi": make_ldi,
"jpr": make_jpr,
"jpi": make_jpi,
"brr": make_brr,
"bri": make_bri,
"lpc": make_lpc,
"hlt": make_hlt,
}
ExpandResult = list[list[str]]
ExpandFn = Callable[..., ExpandResult]
OpExpansionDict = dict[str, ExpandFn]
def expand_id(*parts: str) -> ExpandResult:
return [list(parts)]
default_expansions: OpExpansionDict = {
"alr": lambda *args: expand_id("alr", *args),
"ali": lambda *args: expand_id("ali", *args),
"ldr": lambda *args: expand_id("ldr", *args),
"str": lambda *args: expand_id("str", *args),
"ldi": lambda *args: expand_id("ldi", *args),
"jpr": lambda *args: expand_id("jpr", *args),
"jpi": lambda *args: expand_id("jpi", *args),
"brr": lambda *args: expand_id("brr", *args),
"bri": lambda *args: expand_id("bri", *args),
"lpc": lambda *args: expand_id("lpc", *args),
"hlt": lambda *args: expand_id("hlt", *args),
}
|
