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
|
constants: dict[str, str] = {
# Registers
"RA": "0",
"RB": "1",
"RC": "2",
"RD": "3",
"RE": "4",
"RF": "5",
"RG": "6",
"RH": "7",
# ALU instructions
"al_plus": "0",
"al_minus": "1",
"al_and": "2",
"al_or": "3",
"al_xor": "4",
"al_slr": "5",
"al_sar": "6",
"al_sll": "7",
# ALU flags
"carry": "0",
"overflow": "1",
"zero": "2",
"sign": "3",
}
Symbols = dict[str, int]
def check_size(bits: int, val: int) -> None:
if val >= 2 ** bits:
raise Exception(f"Value does not fit in {bits} bits: 0x{val:>04x}")
def eval_symbol(symbols: Symbols, c: str) -> str:
if c in symbols:
return str(symbols[c])
if c in constants:
return constants[c]
return c
def eval_expr(symbols: Symbols, expr: str, bits: int = 16) -> int:
expr = eval_symbol(symbols, expr)
ret = eval(expr, symbols.copy()) # eval as Python expr
check_size(bits, ret)
return ret
|
