blob: f7cd8a06e88c8eab3dac9310e747e5749c034149 (
plain)
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
|
from atk16_asm import assemble
from atk16_emu import Machine
def pad_bytearray(ba: bytearray, to_length: int = 64 * 1024) -> bytearray:
"""
Pads a bytearray with zeros until it reaches a specified length.
:param to_length: Target length in bytes.
:param ba: The bytearray to be padded.
:return: The padded bytearray.
"""
padding_size = to_length - len(ba)
if padding_size > 0:
ba.extend(b'\x00' * padding_size)
return ba
def make_rom(words: list[int]) -> bytearray:
"""
Constructs a ROM image from a list of 16-bit words.
:param words: The list of 16-bit words.
:return: The ROM image as a bytearray.
"""
bytes = []
for word in words:
bytes.append((word >> 8) & 0xFF)
bytes.append(word & 0xFF)
return pad_bytearray(bytearray(bytes))
def assemble_and_run_until_halted(filename: str) -> Machine:
with open(filename, "r") as f:
source = f.read()
obj = assemble(source, filename)
rom_image = pad_bytearray(obj.program)
machine = Machine()
machine.load_rom_image(rom_image)
machine.reset()
machine.run_until_halted()
return machine
|
