def test_find():
rom_bytes = bytearray(b"\x00\x01\x02\x03\x04\x05\x06\x00")
rom = Rom(rom_bytes)
assert rom.find(b"\x00") == 0
assert rom.find(b"\x00", 1) == 7
assert rom.find(b"\x07") == -1
def __init__(self, layout_address: int, rom: Rom):
super(WorldMap, self).__init__(WORLD_MAP_OBJECT_SET, layout_address)
self._rom = rom
self._minimal_enterable_tiles = _get_normal_enterable_tiles(self._rom)
self._special_enterable_tiles = _get_special_enterable_tiles(self._rom)
self._completable_tiles = _get_completable_tiles(self._rom)
memory_addresses = list_world_map_addresses(rom)
try:
self.number = memory_addresses.index(layout_address) + 1
except ValueError:
raise ValueError(f"World map was not found at given memory address {hex(layout_address)}.")
self.height = WORLD_MAP_HEIGHT
layout_end_index = rom.find(b"\xFF", layout_address)
self.layout_bytes = rom.read(layout_address, layout_end_index - layout_address)
if len(self.layout_bytes) % WORLD_MAP_SCREEN_SIZE != 0:
raise ValueError(
f"Invalid length of layout bytes for world map ({self.layout_bytes}). "
f"Should be divisible by {WORLD_MAP_SCREEN_SIZE}."
)
self.screen_count = len(self.layout_bytes) // WORLD_MAP_SCREEN_SIZE
self.width = int(self.screen_count * WORLD_MAP_SCREEN_WIDTH)
self._parse_structure_data_block(rom)
def _get_completable_tiles(rom: Rom) -> bytearray:
completable_tile_amount = rom.find(COMPLETABLE_LIST_END_MARKER, COMPLETABLE_TILES_LIST) - COMPLETABLE_TILES_LIST - 1
return rom.read(COMPLETABLE_TILES_LIST, completable_tile_amount)