creole/creole_asm.py

165 lines
4.1 KiB
Python
Raw Normal View History

2023-02-07 11:38:04 -05:00
#!/usr/bin/python3
from enum import Enum
2023-02-07 12:25:52 -05:00
class MalformedArgument(Exception):
pass
2023-02-07 11:38:04 -05:00
class ArgType(Enum):
2023-02-07 12:25:52 -05:00
IMM = 1
REG = 2
VAL = 3
LAB = 4
2023-02-07 11:38:04 -05:00
def gettype(s):
if s.isnumeric():
2023-02-07 12:25:52 -05:00
return (ArgType.IMM, int(s))
2023-02-07 11:48:22 -05:00
elif s[0] == 'r' and s[1:].isnumeric():
2023-02-07 12:25:52 -05:00
return (ArgType.REG, int(s[1:]))
2023-02-07 11:48:22 -05:00
elif s[0] == 'l' and s[1:].isnumeric():
2023-02-07 12:25:52 -05:00
return (ArgType.LAB, int(s[1:]))
2023-02-07 11:38:04 -05:00
else:
2023-02-07 12:25:52 -05:00
raise MalformedArgument(s)
2023-02-07 11:38:04 -05:00
def typecheck(self, s):
2023-02-07 11:48:22 -05:00
t = ArgType.gettype(s)
2023-02-07 12:25:52 -05:00
if self == ArgType.VAL:
return t[0] == ArgType.REG or t[0] == ArgType.IMM
2023-02-07 11:38:04 -05:00
else:
return t[0] == self
2023-02-07 12:25:52 -05:00
class OpcodeException(Exception):
pass
class TypecheckLenException(Exception):
pass
class TypecheckException(Exception):
pass
2023-02-07 11:38:04 -05:00
class Instruction:
2023-02-07 12:25:52 -05:00
def __init__(self, opcode, argtypes):
if opcode > 0x7F or opcode < 0:
raise OpcodeException(opcode)
2023-02-07 11:38:04 -05:00
self.opcode = opcode
self.argtypes = argtypes
def typecheck(self, sargs):
rargs = []
if len(sargs) != len(self.argtypes):
2023-02-07 12:25:52 -05:00
raise TypecheckLenException(sargs, self.argtypes)
2023-02-07 11:38:04 -05:00
for i in range(0, len(sargs)):
if not self.argtypes[i].typecheck(sargs[i]):
2023-02-07 12:25:52 -05:00
raise TypecheckException(self.argtypes[i],
sargs[i])
2023-02-07 11:38:04 -05:00
rargs.append(ArgType.gettype(sargs[i]))
return rargs
instructions = {
2023-02-07 12:25:52 -05:00
"nop" : Instruction(0, []),
"push" : Instruction(1, [ArgType.REG]),
"pop" : Instruction(2, [ArgType.REG]),
"add" : Instruction(3, [ArgType.VAL, ArgType.VAL, ArgType.VAL]),
"mul" : Instruction(4, [ArgType.VAL, ArgType.VAL, ArgType.VAL]),
"div" : Instruction(5, [ArgType.VAL, ArgType.VAL, ArgType.VAL]),
"jl" : Instruction(6, [ArgType.LAB, ArgType.VAL, ArgType.VAL]),
"clb" : Instruction(7, [ArgType.LAB]),
"sys" : Instruction(8, [ArgType.VAL])
2023-02-07 11:38:04 -05:00
}
encoding_types = {
# start mask A B
2: (0x7F, 0xC0, 7),
3: (0xFFF, 0xE0, 12),
4: (0x1FFFF, 0xF0, 16),
5: (0x3FFFFF, 0xF8, 21),
6: (0x7FFFFFF, 0xFC, 26),
7: (0xFFFFFFFF, 0xFE, 36),
# A : number of bits in start byte
# B : Total number of bits excluding high bits
}
2023-02-07 12:25:52 -05:00
class InvalidNumberException(Exception):
pass
class InvalidLengthException(Exception):
pass
2023-02-07 11:38:04 -05:00
def encode_pseudo_utf8(n, high_bits, to):
2023-02-07 12:25:52 -05:00
if n < 0:
raise InvalidNumberException(n)
if to is None or to < 0:
for k in sorted(encoding_types):
if n <= encoding_types[k][0]:
to = k
break
if to is None:
raise InvalidNumberException(n)
2023-02-07 11:38:04 -05:00
if to > 8 or to < 0:
2023-02-07 12:25:52 -05:00
raise InvalidLengthException(to)
2023-02-07 11:38:04 -05:00
elif to == 1:
if n < 0x80:
return bytes([n])
else:
2023-02-07 12:25:52 -05:00
raise InvalidNumberException(n,to)
2023-02-07 11:38:04 -05:00
(maxval, start_byte, n_tot) = encoding_types[to]
if n > maxval or high_bits > 15:
2023-02-07 12:25:52 -05:00
raise InvalidNumberException(n, high_bits)
2023-02-07 11:38:04 -05:00
n = n | (high_bits << n_tot)
all_bytes = []
for i in range(0, to - 1):
all_bytes.append(0x80 | (n & 0x3F))
n >>= 6
all_bytes.append(start_byte | n)
return bytes(reversed(all_bytes))
2023-02-07 12:25:52 -05:00
class RangeCheckException(Exception):
pass
2023-02-07 11:38:04 -05:00
class Line:
2023-02-07 12:25:52 -05:00
def __init__(self, opcode, args, labnum, regnum):
2023-02-07 11:38:04 -05:00
self.opcode = opcode
self.args = args
2023-02-07 12:25:52 -05:00
for a in args:
if a[0] == ArgType.REG:
if a[1] < 0 or a[1] >= regnum:
raise RangeCheckException(a[0],
a[1],
regnum)
elif a[0] == ArgType.LAB:
if a[1] < 0 or a[1] >= labnum:
raise RangeCheckException(a[0],
a[1],
regnum)
2023-02-07 11:48:22 -05:00
def __call__(self):
b = bytes([self.opcode])
2023-02-07 12:25:52 -05:00
for a in self.args:
if a[0] == ArgType.REG:
2023-02-07 11:48:22 -05:00
b = b + encode_pseudo_utf8(a[1],1,None)
else:
b = b + encode_pseudo_utf8(a[1],0,None)
return b + bytes([0])
2023-02-07 11:38:04 -05:00
2023-02-07 12:25:52 -05:00
class InstructionNotFoundException(Exception):
pass
2023-02-07 11:38:04 -05:00
class Program:
def asm_push_line(self, ins, args):
2023-02-07 12:25:52 -05:00
self.asm.append(Line(ins, args, self.labnum, self.regnum))
2023-02-07 11:38:04 -05:00
def parse_asm_line(self, line):
2023-02-07 11:48:22 -05:00
line = line.split()
2023-02-07 12:25:52 -05:00
line[0] = line[0].casefold()
2023-02-07 11:38:04 -05:00
if line[0] not in instructions:
2023-02-07 12:25:52 -05:00
raise InstructionNotFoundException(line[0])
2023-02-07 11:38:04 -05:00
else:
2023-02-07 11:48:22 -05:00
ins = instructions[line[0]]
args_w_type = ins.typecheck(line[1:])
self.asm_push_line(ins.opcode, args_w_type)
def __call__(self):
b = bytes()
for line in self.asm:
b = b + line()
return b
2023-02-07 12:25:52 -05:00
def __init__(self, labnum=16, regnum=16):
2023-02-07 11:48:22 -05:00
self.asm = []
2023-02-07 12:25:52 -05:00
self.labnum = labnum
self.regnum = regnum