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)