Fix: rename enterpreter to interpreter
This commit is contained in:
3
src/interpreter/__init__.py
Normal file
3
src/interpreter/__init__.py
Normal file
@@ -0,0 +1,3 @@
|
||||
from .__interpreter import Interpreter, opcode
|
||||
|
||||
__all__ = ["Interpreter", "opcode"]
|
||||
73
src/interpreter/__interpreter.py
Normal file
73
src/interpreter/__interpreter.py
Normal file
@@ -0,0 +1,73 @@
|
||||
from collections.abc import Callable
|
||||
from inspect import getmembers
|
||||
from typing import TypeAlias
|
||||
from traceback import print_exception
|
||||
from pprint import pprint
|
||||
|
||||
from .util import hexdump
|
||||
|
||||
Operation: TypeAlias = str | int
|
||||
Subroutine: TypeAlias = Callable
|
||||
|
||||
OPCODE_HEADER = "__enterpreter__opcode"
|
||||
|
||||
|
||||
def opcode(opcode: Operation):
|
||||
def _helper(func: Callable):
|
||||
setattr(func, OPCODE_HEADER, opcode)
|
||||
return func
|
||||
|
||||
return _helper
|
||||
|
||||
|
||||
class Interpreter:
|
||||
def __init__(self, bits=8, memsize=None) -> None:
|
||||
memsize = memsize or 2**bits
|
||||
self.__bits = bits
|
||||
self.__memory_max_size = 2**self.__bits
|
||||
self.__memory_size = min(memsize, self.__memory_max_size)
|
||||
|
||||
self._memory = bytearray(self.__memory_size)
|
||||
self._memory_pointer: int = 0
|
||||
|
||||
self._program_counter: int = 0
|
||||
|
||||
self.__operations: dict[Operation, Subroutine] = dict()
|
||||
self.__init_opcodes()
|
||||
|
||||
def __init_opcodes(self):
|
||||
for func_name, func in getmembers(
|
||||
self, lambda obj: hasattr(obj, OPCODE_HEADER)
|
||||
):
|
||||
self._register_opcode(getattr(func, OPCODE_HEADER), func)
|
||||
|
||||
def _register_opcode(self, opcode: Operation, callback: Subroutine):
|
||||
self.__operations[opcode] = callback
|
||||
|
||||
def is_running(self) -> bool:
|
||||
return self._memory[self._program_counter] != 0
|
||||
|
||||
def run(self) -> None:
|
||||
try:
|
||||
# Keep running until you reach a null byte
|
||||
while self.is_running():
|
||||
opcode = self._memory[self._program_counter]
|
||||
self.__operations[opcode]()
|
||||
self._program_counter += 1
|
||||
except Exception as e:
|
||||
self.dump()
|
||||
print_exception(e)
|
||||
|
||||
def dump(self):
|
||||
hexdump(
|
||||
data=self._memory,
|
||||
program_counter=self._program_counter,
|
||||
memory_pointer=self._memory_pointer,
|
||||
)
|
||||
pprint(self.__operations)
|
||||
|
||||
def load_program(self, program: bytearray) -> None:
|
||||
self._memory[: len(program)] = program[:]
|
||||
|
||||
# Set new pointer immediately after code
|
||||
self._memory_pointer = len(program) + 1
|
||||
0
src/interpreter/__main__.py
Normal file
0
src/interpreter/__main__.py
Normal file
3
src/interpreter/util/__init__.py
Normal file
3
src/interpreter/util/__init__.py
Normal file
@@ -0,0 +1,3 @@
|
||||
from .hexdump import hexdump
|
||||
|
||||
__all__ = ["hexdump"]
|
||||
45
src/interpreter/util/hexdump.py
Normal file
45
src/interpreter/util/hexdump.py
Normal file
@@ -0,0 +1,45 @@
|
||||
from colorama import Fore, Style, init
|
||||
|
||||
# Initialize colorama (especially important on Windows)
|
||||
init()
|
||||
|
||||
|
||||
def hexdump(
|
||||
data: bytearray,
|
||||
width: int = 16,
|
||||
program_counter: int = None,
|
||||
memory_pointer: int = None,
|
||||
):
|
||||
"""
|
||||
Print a hex+ASCII memory dump of the given bytearray.
|
||||
Highlights:
|
||||
- Program counter (PC) in bright red
|
||||
- Memory pointer (MP) in bright blue
|
||||
"""
|
||||
for i in range(0, len(data), width):
|
||||
chunk = data[i : i + width]
|
||||
hex_parts = []
|
||||
ascii_parts = []
|
||||
|
||||
for j, byte in enumerate(chunk):
|
||||
addr = i + j
|
||||
|
||||
# Apply color based on PC or MP
|
||||
if addr == program_counter:
|
||||
style = Style.BRIGHT + Fore.RED
|
||||
elif addr == memory_pointer:
|
||||
style = Style.BRIGHT + Fore.BLUE
|
||||
else:
|
||||
style = ""
|
||||
reset = Style.RESET_ALL if style else ""
|
||||
|
||||
# Hex and ASCII views
|
||||
hex_parts.append(f"{style}{byte:02x}{reset}")
|
||||
ascii_parts.append(
|
||||
f"{style}{chr(byte) if 32 <= byte <= 126 else '.'}{reset}"
|
||||
)
|
||||
|
||||
# Format line: address, hex bytes, ASCII chars
|
||||
hex_field = " ".join(hex_parts).ljust(width * 3 - 1)
|
||||
ascii_field = "".join(ascii_parts)
|
||||
print(f"{i:08x}: {hex_field} {ascii_field}")
|
||||
Reference in New Issue
Block a user