Grinch-AP/worlds/pokemon_emerald/data.py

"""
Pulls data from JSON files in worlds/pokemon_emerald/data/ into classes.
This also includes marrying automatically extracted data with manually
defined data (like location labels or usable pokemon species), some cleanup
and sorting, and Warp methods.
"""
from dataclasses import dataclass
from enum import IntEnum, Enum
import orjson
from typing import Dict, List, NamedTuple, Optional, Set, FrozenSet, Tuple, Any, Union
import pkgutil
import pkg_resources

from BaseClasses import ItemClassification


BASE_OFFSET = 3860000
POKEDEX_OFFSET = 10000

IGNORABLE_MAPS = {
    "MAP_ALTERING_CAVE",
    "MAP_CAVE_OF_ORIGIN_UNUSED_RUBY_SAPPHIRE_MAP1",
    "MAP_CAVE_OF_ORIGIN_UNUSED_RUBY_SAPPHIRE_MAP2",
    "MAP_CAVE_OF_ORIGIN_UNUSED_RUBY_SAPPHIRE_MAP3",
}
"""These maps exist but don't show up in the rando or are unused, and so should be discarded"""

OUT_OF_LOGIC_MAPS = {
    "MAP_DESERT_UNDERPASS",
    "MAP_SAFARI_ZONE_NORTHEAST",
    "MAP_SAFARI_ZONE_SOUTHEAST",
    "MAP_METEOR_FALLS_STEVENS_CAVE",
    "MAP_MIRAGE_TOWER_1F",
    "MAP_MIRAGE_TOWER_2F",
    "MAP_MIRAGE_TOWER_3F",
    "MAP_MIRAGE_TOWER_4F",
}
"""
These maps have encounters and are locked behind beating the champion or are missable.
Those encounter slots should be ignored for logical access to a species.
"""

NUM_REAL_SPECIES = 386


class Warp:
    """
    Represents warp events in the game like doorways or warp pads
    """
    is_one_way: bool
    source_map: str
    source_ids: List[int]
    dest_map: str
    dest_ids: List[int]
    parent_region: Optional[str]

    def __init__(self, encoded_string: Optional[str] = None, parent_region: Optional[str] = None) -> None:
        if encoded_string is not None:
            decoded_warp = Warp.decode(encoded_string)
            self.is_one_way = decoded_warp.is_one_way
            self.source_map = decoded_warp.source_map
            self.source_ids = decoded_warp.source_ids
            self.dest_map = decoded_warp.dest_map
            self.dest_ids = decoded_warp.dest_ids
        self.parent_region = parent_region

    def encode(self) -> str:
        """
        Returns a string encoding of this warp
        """
        source_ids_string = ""
        for source_id in self.source_ids:
            source_ids_string += str(source_id) + ","
        source_ids_string = source_ids_string[:-1]  # Remove last ","

        dest_ids_string = ""
        for dest_id in self.dest_ids:
            dest_ids_string += str(dest_id) + ","
        dest_ids_string = dest_ids_string[:-1]  # Remove last ","

        return f"{self.source_map}:{source_ids_string}/{self.dest_map}:{dest_ids_string}{'!' if self.is_one_way else ''}"

    def connects_to(self, other: "Warp") -> bool:
        """
        Returns true if this warp sends the player to `other`
        """
        return self.dest_map == other.source_map and set(self.dest_ids) <= set(other.source_ids)

    @staticmethod
    def decode(encoded_string: str) -> "Warp":
        """
        Create a Warp object from an encoded string
        """
        warp = Warp()
        warp.is_one_way = encoded_string.endswith("!")
        if warp.is_one_way:
            encoded_string = encoded_string[:-1]

        warp_source, warp_dest = encoded_string.split("/")
        warp_source_map, warp_source_indices = warp_source.split(":")
        warp_dest_map, warp_dest_indices = warp_dest.split(":")

        warp.source_map = warp_source_map
        warp.dest_map = warp_dest_map

        warp.source_ids = [int(index) for index in warp_source_indices.split(",")]
        warp.dest_ids = [int(index) for index in warp_dest_indices.split(",")]

        return warp


class ItemData(NamedTuple):
    label: str
    item_id: int
    modern_id: Optional[int]
    classification: ItemClassification
    tags: FrozenSet[str]


class LocationCategory(IntEnum):
    BADGE = 0
    HM = 1
    KEY = 2
    ROD = 3
    BIKE = 4
    TICKET = 5
    OVERWORLD_ITEM = 6
    HIDDEN_ITEM = 7
    GIFT = 8
    BERRY_TREE = 9
    TRAINER = 10
    POKEDEX = 11


class LocationData(NamedTuple):
    name: str
    label: str
    parent_region: str
    default_item: int
    address: Union[int, List[int]]
    flag: int
    category: LocationCategory
    tags: FrozenSet[str]


class EncounterTableData(NamedTuple):
    slots: List[int]
    address: int


# class EncounterType(StrEnum):  # StrEnum introduced in python 3.11
class EncounterType(Enum):
    LAND = "LAND"
    WATER = "WATER"
    FISHING = "FISHING"
    ROCK_SMASH = "ROCK_SMASH"


@dataclass
class MapData:
    name: str
    label: str
    header_address: int
    encounters: Dict[EncounterType, EncounterTableData]


class EventData(NamedTuple):
    name: str
    parent_region: str


class RegionData:
    name: str
    parent_map: MapData
    has_grass: bool
    has_water: bool
    has_fishing: bool
    exits: List[str]
    warps: List[str]
    locations: List[str]
    events: List[EventData]

    def __init__(self, name: str, parent_map: MapData, has_grass: bool, has_water: bool, has_fishing: bool):
        self.name = name
        self.parent_map = parent_map
        self.has_grass = has_grass
        self.has_water = has_water
        self.has_fishing = has_fishing
        self.exits = []
        self.warps = []
        self.locations = []
        self.events = []


class BaseStats(NamedTuple):
    hp: int
    attack: int
    defense: int
    speed: int
    special_attack: int
    special_defense: int


class LearnsetMove(NamedTuple):
    level: int
    move_id: int


class EvolutionMethodEnum(IntEnum):
    LEVEL = 0
    LEVEL_ATK_LT_DEF = 1
    LEVEL_ATK_EQ_DEF = 2
    LEVEL_ATK_GT_DEF = 3
    LEVEL_SILCOON = 4
    LEVEL_CASCOON = 5
    LEVEL_NINJASK = 6
    LEVEL_SHEDINJA = 7
    ITEM = 8
    FRIENDSHIP = 9
    FRIENDSHIP_DAY = 10
    FRIENDSHIP_NIGHT = 11


class EvolutionData(NamedTuple):
    method: EvolutionMethodEnum
    param: int  # Level/item id/friendship/etc.; depends on method
    species_id: int


class MiscPokemonData(NamedTuple):
    species_id: int
    address: int


@dataclass
class SpeciesData:
    name: str
    label: str
    species_id: int
    national_dex_number: int
    base_stats: BaseStats
    types: Tuple[int, int]
    abilities: Tuple[int, int]
    evolutions: List[EvolutionData]
    pre_evolution: Optional[int]
    catch_rate: int
    friendship: int
    learnset: List[LearnsetMove]
    tm_hm_compatibility: int
    learnset_address: int
    address: int


class AbilityData(NamedTuple):
    ability_id: int
    label: str


class TrainerPokemonDataTypeEnum(IntEnum):
    NO_ITEM_DEFAULT_MOVES = 0
    ITEM_DEFAULT_MOVES = 1
    NO_ITEM_CUSTOM_MOVES = 2
    ITEM_CUSTOM_MOVES = 3


def _str_to_pokemon_data_type(string: str) -> TrainerPokemonDataTypeEnum:
    if string == "NO_ITEM_DEFAULT_MOVES":
        return TrainerPokemonDataTypeEnum.NO_ITEM_DEFAULT_MOVES
    if string == "ITEM_DEFAULT_MOVES":
        return TrainerPokemonDataTypeEnum.ITEM_DEFAULT_MOVES
    if string == "NO_ITEM_CUSTOM_MOVES":
        return TrainerPokemonDataTypeEnum.NO_ITEM_CUSTOM_MOVES
    if string == "ITEM_CUSTOM_MOVES":
        return TrainerPokemonDataTypeEnum.ITEM_CUSTOM_MOVES


class TrainerPokemonData(NamedTuple):
    species_id: int
    level: int
    moves: Optional[Tuple[int, int, int, int]]


class TrainerPartyData(NamedTuple):
    pokemon: List[TrainerPokemonData]
    pokemon_data_type: TrainerPokemonDataTypeEnum
    address: int


@dataclass
class TrainerData:
    trainer_id: int
    party: TrainerPartyData
    address: int
    script_address: int
    battle_type: int


class PokemonEmeraldData:
    starters: Tuple[int, int, int]
    constants: Dict[str, int]
    ram_addresses: Dict[str, int]
    rom_addresses: Dict[str, int]
    regions: Dict[str, RegionData]
    locations: Dict[str, LocationData]
    items: Dict[int, ItemData]
    species: Dict[int, SpeciesData]
    legendary_encounters: List[MiscPokemonData]
    misc_pokemon: List[MiscPokemonData]
    tmhm_moves: List[int]
    abilities: List[AbilityData]
    move_labels: Dict[str, int]
    maps: Dict[str, MapData]
    warps: Dict[str, Warp]
    warp_map: Dict[str, Optional[str]]
    trainers: List[TrainerData]

    def __init__(self) -> None:
        self.starters = (277, 280, 283)
        self.constants = {}
        self.ram_addresses = {}
        self.rom_addresses = {}
        self.regions = {}
        self.locations = {}
        self.items = {}
        self.species = {}
        self.legendary_encounters = []
        self.misc_pokemon = []
        self.tmhm_moves = []
        self.abilities = []
        self.move_labels = {}
        self.maps = {}
        self.warps = {}
        self.warp_map = {}
        self.trainers = []


def load_json_data(data_name: str) -> Union[List[Any], Dict[str, Any]]:
    return orjson.loads(pkgutil.get_data(__name__, "data/" + data_name).decode("utf-8-sig"))


def _init() -> None:
    import re

    extracted_data: Dict[str, Any] = load_json_data("extracted_data.json")
    data.constants = extracted_data["constants"]
    data.ram_addresses = extracted_data["misc_ram_addresses"]
    data.rom_addresses = extracted_data["misc_rom_addresses"]

    location_attributes_json = load_json_data("locations.json")

    # Create map data
    for map_name, map_json in extracted_data["maps"].items():
        assert isinstance(map_name, str)
        if map_name in IGNORABLE_MAPS:
            continue

        encounter_tables: Dict[EncounterType, EncounterTableData] = {}
        if "land_encounters" in map_json:
            encounter_tables[EncounterType.LAND] = EncounterTableData(
                map_json["land_encounters"]["slots"],
                map_json["land_encounters"]["address"]
            )
        if "water_encounters" in map_json:
            encounter_tables[EncounterType.WATER] = EncounterTableData(
                map_json["water_encounters"]["slots"],
                map_json["water_encounters"]["address"]
            )
        if "fishing_encounters" in map_json:
            encounter_tables[EncounterType.FISHING] = EncounterTableData(
                map_json["fishing_encounters"]["slots"],
                map_json["fishing_encounters"]["address"]
            )
        if "rock_smash_encounters" in map_json:
            encounter_tables[EncounterType.ROCK_SMASH] = EncounterTableData(
                map_json["rock_smash_encounters"]["slots"],
                map_json["rock_smash_encounters"]["address"]
            )

        # Derive a user-facing label
        label = []
        for word in map_name[4:].split("_"):
            # 1F, B1F, 2R, etc.
            re_match = re.match(r"^B?\d+[FRP]$", word)
            if re_match:
                label.append(word)
                continue

            # Route 103, Hall 1, House 5, etc.
            re_match = re.match(r"^([A-Z]+)(\d+)$", word)
            if re_match:
                label.append(re_match.group(1).capitalize())
                label.append(re_match.group(2).lstrip("0"))
                continue

            if word == "OF":
                label.append("of")
                continue

            if word == "SS":
                label.append("S.S.")
                continue

            label.append(word.capitalize())

        data.maps[map_name] = MapData(
            map_name,
            " ".join(label),
            map_json["header_address"],
            encounter_tables
        )

    # Load/merge region json files
    region_json_list = []
    for file in pkg_resources.resource_listdir(__name__, "data/regions"):
        if not pkg_resources.resource_isdir(__name__, "data/regions/" + file):
            region_json_list.append(load_json_data("regions/" + file))

    regions_json = {}
    for region_subset in region_json_list:
        for region_name, region_json in region_subset.items():
            if region_name in regions_json:
                raise AssertionError("Region [{region_name}] was defined multiple times")
            regions_json[region_name] = region_json

    # Create region data
    claimed_locations: Set[str] = set()
    claimed_warps: Set[str] = set()

    data.regions = {}
    for region_name, region_json in regions_json.items():
        new_region = RegionData(
            region_name,
            data.maps[region_json["parent_map"]],
            region_json["has_grass"],
            region_json["has_water"],
            region_json["has_fishing"]
        )

        # Locations
        for location_name in region_json["locations"]:
            if location_name in claimed_locations:
                raise AssertionError(f"Location [{location_name}] was claimed by multiple regions")

            location_json = extracted_data["locations"][location_name]
            if location_name.startswith("TRAINER_BRENDAN_") or location_name.startswith("TRAINER_MAY_"):
                import re
                locale = re.match("TRAINER_BRENDAN_([A-Z0-9_]+)_MUDKIP_REWARD", location_name).group(1)
                alternate_rival_jsons = [extracted_data["locations"][alternate] for alternate in [
                    f"TRAINER_BRENDAN_{locale}_TORCHIC_REWARD",
                    f"TRAINER_BRENDAN_{locale}_TREECKO_REWARD",
                    f"TRAINER_MAY_{locale}_MUDKIP_REWARD",
                    f"TRAINER_MAY_{locale}_TORCHIC_REWARD",
                    f"TRAINER_MAY_{locale}_TREECKO_REWARD",
                ]]
                new_location = LocationData(
                    location_name,
                    location_attributes_json[location_name]["label"],
                    region_name,
                    location_json["default_item"],
                    [location_json["address"]] + [j["address"] for j in alternate_rival_jsons],
                    location_json["flag"],
                    LocationCategory[location_attributes_json[location_name]["category"]],
                    frozenset(location_attributes_json[location_name]["tags"])
                )
            else:
                new_location = LocationData(
                    location_name,
                    location_attributes_json[location_name]["label"],
                    region_name,
                    location_json["default_item"],
                    location_json["address"],
                    location_json["flag"],
                    LocationCategory[location_attributes_json[location_name]["category"]],
                    frozenset(location_attributes_json[location_name]["tags"])
                )
            new_region.locations.append(location_name)
            data.locations[location_name] = new_location
            claimed_locations.add(location_name)

        new_region.locations.sort()

        # Events
        for event in region_json["events"]:
            new_region.events.append(EventData(event, region_name))

        # Exits
        for region_exit in region_json["exits"]:
            new_region.exits.append(region_exit)

        # Warps
        for encoded_warp in region_json["warps"]:
            if encoded_warp in claimed_warps:
                raise AssertionError(f"Warp [{encoded_warp}] was claimed by multiple regions")
            new_region.warps.append(encoded_warp)
            data.warps[encoded_warp] = Warp(encoded_warp, region_name)
            claimed_warps.add(encoded_warp)

        new_region.warps.sort()

        data.regions[region_name] = new_region

    # Create item data
    items_json = load_json_data("items.json")

    data.items = {}
    for item_constant_name, attributes in items_json.items():
        item_classification = None
        if attributes["classification"] == "PROGRESSION":
            item_classification = ItemClassification.progression
        elif attributes["classification"] == "USEFUL":
            item_classification = ItemClassification.useful
        elif attributes["classification"] == "FILLER":
            item_classification = ItemClassification.filler
        elif attributes["classification"] == "TRAP":
            item_classification = ItemClassification.trap
        else:
            raise ValueError(f"Unknown classification {attributes['classification']} for item {item_constant_name}")

        data.items[data.constants[item_constant_name]] = ItemData(
            attributes["label"],
            data.constants[item_constant_name],
            attributes["modern_id"],
            item_classification,
            frozenset(attributes["tags"])
        )

    # Create species data

    # Excludes extras like copies of Unown and special species values like SPECIES_EGG.
    all_species: List[Tuple[str, str, int]] = [
        ("SPECIES_BULBASAUR", "Bulbasaur", 1),
        ("SPECIES_IVYSAUR", "Ivysaur", 2),
        ("SPECIES_VENUSAUR", "Venusaur", 3),
        ("SPECIES_CHARMANDER", "Charmander", 4),
        ("SPECIES_CHARMELEON", "Charmeleon", 5),
        ("SPECIES_CHARIZARD", "Charizard", 6),
        ("SPECIES_SQUIRTLE", "Squirtle", 7),
        ("SPECIES_WARTORTLE", "Wartortle", 8),
        ("SPECIES_BLASTOISE", "Blastoise", 9),
        ("SPECIES_CATERPIE", "Caterpie", 10),
        ("SPECIES_METAPOD", "Metapod", 11),
        ("SPECIES_BUTTERFREE", "Butterfree", 12),
        ("SPECIES_WEEDLE", "Weedle", 13),
        ("SPECIES_KAKUNA", "Kakuna", 14),
        ("SPECIES_BEEDRILL", "Beedrill", 15),
        ("SPECIES_PIDGEY", "Pidgey", 16),
        ("SPECIES_PIDGEOTTO", "Pidgeotto", 17),
        ("SPECIES_PIDGEOT", "Pidgeot", 18),
        ("SPECIES_RATTATA", "Rattata", 19),
        ("SPECIES_RATICATE", "Raticate", 20),
        ("SPECIES_SPEAROW", "Spearow", 21),
        ("SPECIES_FEAROW", "Fearow", 22),
        ("SPECIES_EKANS", "Ekans", 23),
        ("SPECIES_ARBOK", "Arbok", 24),
        ("SPECIES_PIKACHU", "Pikachu", 25),
        ("SPECIES_RAICHU", "Raichu", 26),
        ("SPECIES_SANDSHREW", "Sandshrew", 27),
        ("SPECIES_SANDSLASH", "Sandslash", 28),
        ("SPECIES_NIDORAN_F", "Nidoran Female", 29),
        ("SPECIES_NIDORINA", "Nidorina", 30),
        ("SPECIES_NIDOQUEEN", "Nidoqueen", 31),
        ("SPECIES_NIDORAN_M", "Nidoran Male", 32),
        ("SPECIES_NIDORINO", "Nidorino", 33),
        ("SPECIES_NIDOKING", "Nidoking", 34),
        ("SPECIES_CLEFAIRY", "Clefairy", 35),
        ("SPECIES_CLEFABLE", "Clefable", 36),
        ("SPECIES_VULPIX", "Vulpix", 37),
        ("SPECIES_NINETALES", "Ninetales", 38),
        ("SPECIES_JIGGLYPUFF", "Jigglypuff", 39),
        ("SPECIES_WIGGLYTUFF", "Wigglytuff", 40),
        ("SPECIES_ZUBAT", "Zubat", 41),
        ("SPECIES_GOLBAT", "Golbat", 42),
        ("SPECIES_ODDISH", "Oddish", 43),
        ("SPECIES_GLOOM", "Gloom", 44),
        ("SPECIES_VILEPLUME", "Vileplume", 45),
        ("SPECIES_PARAS", "Paras", 46),
        ("SPECIES_PARASECT", "Parasect", 47),
        ("SPECIES_VENONAT", "Venonat", 48),
        ("SPECIES_VENOMOTH", "Venomoth", 49),
        ("SPECIES_DIGLETT", "Diglett", 50),
        ("SPECIES_DUGTRIO", "Dugtrio", 51),
        ("SPECIES_MEOWTH", "Meowth", 52),
        ("SPECIES_PERSIAN", "Persian", 53),
        ("SPECIES_PSYDUCK", "Psyduck", 54),
        ("SPECIES_GOLDUCK", "Golduck", 55),
        ("SPECIES_MANKEY", "Mankey", 56),
        ("SPECIES_PRIMEAPE", "Primeape", 57),
        ("SPECIES_GROWLITHE", "Growlithe", 58),
        ("SPECIES_ARCANINE", "Arcanine", 59),
        ("SPECIES_POLIWAG", "Poliwag", 60),
        ("SPECIES_POLIWHIRL", "Poliwhirl", 61),
        ("SPECIES_POLIWRATH", "Poliwrath", 62),
        ("SPECIES_ABRA", "Abra", 63),
        ("SPECIES_KADABRA", "Kadabra", 64),
        ("SPECIES_ALAKAZAM", "Alakazam", 65),
        ("SPECIES_MACHOP", "Machop", 66),
        ("SPECIES_MACHOKE", "Machoke", 67),
        ("SPECIES_MACHAMP", "Machamp", 68),
        ("SPECIES_BELLSPROUT", "Bellsprout", 69),
        ("SPECIES_WEEPINBELL", "Weepinbell", 70),
        ("SPECIES_VICTREEBEL", "Victreebel", 71),
        ("SPECIES_TENTACOOL", "Tentacool", 72),
        ("SPECIES_TENTACRUEL", "Tentacruel", 73),
        ("SPECIES_GEODUDE", "Geodude", 74),
        ("SPECIES_GRAVELER", "Graveler", 75),
        ("SPECIES_GOLEM", "Golem", 76),
        ("SPECIES_PONYTA", "Ponyta", 77),
        ("SPECIES_RAPIDASH", "Rapidash", 78),
        ("SPECIES_SLOWPOKE", "Slowpoke", 79),
        ("SPECIES_SLOWBRO", "Slowbro", 80),
        ("SPECIES_MAGNEMITE", "Magnemite", 81),
        ("SPECIES_MAGNETON", "Magneton", 82),
        ("SPECIES_FARFETCHD", "Farfetch'd", 83),
        ("SPECIES_DODUO", "Doduo", 84),
        ("SPECIES_DODRIO", "Dodrio", 85),
        ("SPECIES_SEEL", "Seel", 86),
        ("SPECIES_DEWGONG", "Dewgong", 87),
        ("SPECIES_GRIMER", "Grimer", 88),
        ("SPECIES_MUK", "Muk", 89),
        ("SPECIES_SHELLDER", "Shellder", 90),
        ("SPECIES_CLOYSTER", "Cloyster", 91),
        ("SPECIES_GASTLY", "Gastly", 92),
        ("SPECIES_HAUNTER", "Haunter", 93),
        ("SPECIES_GENGAR", "Gengar", 94),
        ("SPECIES_ONIX", "Onix", 95),
        ("SPECIES_DROWZEE", "Drowzee", 96),
        ("SPECIES_HYPNO", "Hypno", 97),
        ("SPECIES_KRABBY", "Krabby", 98),
        ("SPECIES_KINGLER", "Kingler", 99),
        ("SPECIES_VOLTORB", "Voltorb", 100),
        ("SPECIES_ELECTRODE", "Electrode", 101),
        ("SPECIES_EXEGGCUTE", "Exeggcute", 102),
        ("SPECIES_EXEGGUTOR", "Exeggutor", 103),
        ("SPECIES_CUBONE", "Cubone", 104),
        ("SPECIES_MAROWAK", "Marowak", 105),
        ("SPECIES_HITMONLEE", "Hitmonlee", 106),
        ("SPECIES_HITMONCHAN", "Hitmonchan", 107),
        ("SPECIES_LICKITUNG", "Lickitung", 108),
        ("SPECIES_KOFFING", "Koffing", 109),
        ("SPECIES_WEEZING", "Weezing", 110),
        ("SPECIES_RHYHORN", "Rhyhorn", 111),
        ("SPECIES_RHYDON", "Rhydon", 112),
        ("SPECIES_CHANSEY", "Chansey", 113),
        ("SPECIES_TANGELA", "Tangela", 114),
        ("SPECIES_KANGASKHAN", "Kangaskhan", 115),
        ("SPECIES_HORSEA", "Horsea", 116),
        ("SPECIES_SEADRA", "Seadra", 117),
        ("SPECIES_GOLDEEN", "Goldeen", 118),
        ("SPECIES_SEAKING", "Seaking", 119),
        ("SPECIES_STARYU", "Staryu", 120),
        ("SPECIES_STARMIE", "Starmie", 121),
        ("SPECIES_MR_MIME", "Mr. Mime", 122),
        ("SPECIES_SCYTHER", "Scyther", 123),
        ("SPECIES_JYNX", "Jynx", 124),
        ("SPECIES_ELECTABUZZ", "Electabuzz", 125),
        ("SPECIES_MAGMAR", "Magmar", 126),
        ("SPECIES_PINSIR", "Pinsir", 127),
        ("SPECIES_TAUROS", "Tauros", 128),
        ("SPECIES_MAGIKARP", "Magikarp", 129),
        ("SPECIES_GYARADOS", "Gyarados", 130),
        ("SPECIES_LAPRAS", "Lapras", 131),
        ("SPECIES_DITTO", "Ditto", 132),
        ("SPECIES_EEVEE", "Eevee", 133),
        ("SPECIES_VAPOREON", "Vaporeon", 134),
        ("SPECIES_JOLTEON", "Jolteon", 135),
        ("SPECIES_FLAREON", "Flareon", 136),
        ("SPECIES_PORYGON", "Porygon", 137),
        ("SPECIES_OMANYTE", "Omanyte", 138),
        ("SPECIES_OMASTAR", "Omastar", 139),
        ("SPECIES_KABUTO", "Kabuto", 140),
        ("SPECIES_KABUTOPS", "Kabutops", 141),
        ("SPECIES_AERODACTYL", "Aerodactyl", 142),
        ("SPECIES_SNORLAX", "Snorlax", 143),
        ("SPECIES_ARTICUNO", "Articuno", 144),
        ("SPECIES_ZAPDOS", "Zapdos", 145),
        ("SPECIES_MOLTRES", "Moltres", 146),
        ("SPECIES_DRATINI", "Dratini", 147),
        ("SPECIES_DRAGONAIR", "Dragonair", 148),
        ("SPECIES_DRAGONITE", "Dragonite", 149),
        ("SPECIES_MEWTWO", "Mewtwo", 150),
        ("SPECIES_MEW", "Mew", 151),
        ("SPECIES_CHIKORITA", "Chikorita", 152),
        ("SPECIES_BAYLEEF", "Bayleef", 153),
        ("SPECIES_MEGANIUM", "Meganium", 154),
        ("SPECIES_CYNDAQUIL", "Cyndaquil", 155),
        ("SPECIES_QUILAVA", "Quilava", 156),
        ("SPECIES_TYPHLOSION", "Typhlosion", 157),
        ("SPECIES_TOTODILE", "Totodile", 158),
        ("SPECIES_CROCONAW", "Croconaw", 159),
        ("SPECIES_FERALIGATR", "Feraligatr", 160),
        ("SPECIES_SENTRET", "Sentret", 161),
        ("SPECIES_FURRET", "Furret", 162),
        ("SPECIES_HOOTHOOT", "Hoothoot", 163),
        ("SPECIES_NOCTOWL", "Noctowl", 164),
        ("SPECIES_LEDYBA", "Ledyba", 165),
        ("SPECIES_LEDIAN", "Ledian", 166),
        ("SPECIES_SPINARAK", "Spinarak", 167),
        ("SPECIES_ARIADOS", "Ariados", 168),
        ("SPECIES_CROBAT", "Crobat", 169),
        ("SPECIES_CHINCHOU", "Chinchou", 170),
        ("SPECIES_LANTURN", "Lanturn", 171),
        ("SPECIES_PICHU", "Pichu", 172),
        ("SPECIES_CLEFFA", "Cleffa", 173),
        ("SPECIES_IGGLYBUFF", "Igglybuff", 174),
        ("SPECIES_TOGEPI", "Togepi", 175),
        ("SPECIES_TOGETIC", "Togetic", 176),
        ("SPECIES_NATU", "Natu", 177),
        ("SPECIES_XATU", "Xatu", 178),
        ("SPECIES_MAREEP", "Mareep", 179),
        ("SPECIES_FLAAFFY", "Flaaffy", 180),
        ("SPECIES_AMPHAROS", "Ampharos", 181),
        ("SPECIES_BELLOSSOM", "Bellossom", 182),
        ("SPECIES_MARILL", "Marill", 183),
        ("SPECIES_AZUMARILL", "Azumarill", 184),
        ("SPECIES_SUDOWOODO", "Sudowoodo", 185),
        ("SPECIES_POLITOED", "Politoed", 186),
        ("SPECIES_HOPPIP", "Hoppip", 187),
        ("SPECIES_SKIPLOOM", "Skiploom", 188),
        ("SPECIES_JUMPLUFF", "Jumpluff", 189),
        ("SPECIES_AIPOM", "Aipom", 190),
        ("SPECIES_SUNKERN", "Sunkern", 191),
        ("SPECIES_SUNFLORA", "Sunflora", 192),
        ("SPECIES_YANMA", "Yanma", 193),
        ("SPECIES_WOOPER", "Wooper", 194),
        ("SPECIES_QUAGSIRE", "Quagsire", 195),
        ("SPECIES_ESPEON", "Espeon", 196),
        ("SPECIES_UMBREON", "Umbreon", 197),
        ("SPECIES_MURKROW", "Murkrow", 198),
        ("SPECIES_SLOWKING", "Slowking", 199),
        ("SPECIES_MISDREAVUS", "Misdreavus", 200),
        ("SPECIES_UNOWN", "Unown", 201),
        ("SPECIES_WOBBUFFET", "Wobbuffet", 202),
        ("SPECIES_GIRAFARIG", "Girafarig", 203),
        ("SPECIES_PINECO", "Pineco", 204),
        ("SPECIES_FORRETRESS", "Forretress", 205),
        ("SPECIES_DUNSPARCE", "Dunsparce", 206),
        ("SPECIES_GLIGAR", "Gligar", 207),
        ("SPECIES_STEELIX", "Steelix", 208),
        ("SPECIES_SNUBBULL", "Snubbull", 209),
        ("SPECIES_GRANBULL", "Granbull", 210),
        ("SPECIES_QWILFISH", "Qwilfish", 211),
        ("SPECIES_SCIZOR", "Scizor", 212),
        ("SPECIES_SHUCKLE", "Shuckle", 213),
        ("SPECIES_HERACROSS", "Heracross", 214),
        ("SPECIES_SNEASEL", "Sneasel", 215),
        ("SPECIES_TEDDIURSA", "Teddiursa", 216),
        ("SPECIES_URSARING", "Ursaring", 217),
        ("SPECIES_SLUGMA", "Slugma", 218),
        ("SPECIES_MAGCARGO", "Magcargo", 219),
        ("SPECIES_SWINUB", "Swinub", 220),
        ("SPECIES_PILOSWINE", "Piloswine", 221),
        ("SPECIES_CORSOLA", "Corsola", 222),
        ("SPECIES_REMORAID", "Remoraid", 223),
        ("SPECIES_OCTILLERY", "Octillery", 224),
        ("SPECIES_DELIBIRD", "Delibird", 225),
        ("SPECIES_MANTINE", "Mantine", 226),
        ("SPECIES_SKARMORY", "Skarmory", 227),
        ("SPECIES_HOUNDOUR", "Houndour", 228),
        ("SPECIES_HOUNDOOM", "Houndoom", 229),
        ("SPECIES_KINGDRA", "Kingdra", 230),
        ("SPECIES_PHANPY", "Phanpy", 231),
        ("SPECIES_DONPHAN", "Donphan", 232),
        ("SPECIES_PORYGON2", "Porygon2", 233),
        ("SPECIES_STANTLER", "Stantler", 234),
        ("SPECIES_SMEARGLE", "Smeargle", 235),
        ("SPECIES_TYROGUE", "Tyrogue", 236),
        ("SPECIES_HITMONTOP", "Hitmontop", 237),
        ("SPECIES_SMOOCHUM", "Smoochum", 238),
        ("SPECIES_ELEKID", "Elekid", 239),
        ("SPECIES_MAGBY", "Magby", 240),
        ("SPECIES_MILTANK", "Miltank", 241),
        ("SPECIES_BLISSEY", "Blissey", 242),
        ("SPECIES_RAIKOU", "Raikou", 243),
        ("SPECIES_ENTEI", "Entei", 244),
        ("SPECIES_SUICUNE", "Suicune", 245),
        ("SPECIES_LARVITAR", "Larvitar", 246),
        ("SPECIES_PUPITAR", "Pupitar", 247),
        ("SPECIES_TYRANITAR", "Tyranitar", 248),
        ("SPECIES_LUGIA", "Lugia", 249),
        ("SPECIES_HO_OH", "Ho-Oh", 250),
        ("SPECIES_CELEBI", "Celebi", 251),
        ("SPECIES_TREECKO", "Treecko", 252),
        ("SPECIES_GROVYLE", "Grovyle", 253),
        ("SPECIES_SCEPTILE", "Sceptile", 254),
        ("SPECIES_TORCHIC", "Torchic", 255),
        ("SPECIES_COMBUSKEN", "Combusken", 256),
        ("SPECIES_BLAZIKEN", "Blaziken", 257),
        ("SPECIES_MUDKIP", "Mudkip", 258),
        ("SPECIES_MARSHTOMP", "Marshtomp", 259),
        ("SPECIES_SWAMPERT", "Swampert", 260),
        ("SPECIES_POOCHYENA", "Poochyena", 261),
        ("SPECIES_MIGHTYENA", "Mightyena", 262),
        ("SPECIES_ZIGZAGOON", "Zigzagoon", 263),
        ("SPECIES_LINOONE", "Linoone", 264),
        ("SPECIES_WURMPLE", "Wurmple", 265),
        ("SPECIES_SILCOON", "Silcoon", 266),
        ("SPECIES_BEAUTIFLY", "Beautifly", 267),
        ("SPECIES_CASCOON", "Cascoon", 268),
        ("SPECIES_DUSTOX", "Dustox", 269),
        ("SPECIES_LOTAD", "Lotad", 270),
        ("SPECIES_LOMBRE", "Lombre", 271),
        ("SPECIES_LUDICOLO", "Ludicolo", 272),
        ("SPECIES_SEEDOT", "Seedot", 273),
        ("SPECIES_NUZLEAF", "Nuzleaf", 274),
        ("SPECIES_SHIFTRY", "Shiftry", 275),
        ("SPECIES_NINCADA", "Nincada", 290),
        ("SPECIES_NINJASK", "Ninjask", 291),
        ("SPECIES_SHEDINJA", "Shedinja", 292),
        ("SPECIES_TAILLOW", "Taillow", 276),
        ("SPECIES_SWELLOW", "Swellow", 277),
        ("SPECIES_SHROOMISH", "Shroomish", 285),
        ("SPECIES_BRELOOM", "Breloom", 286),
        ("SPECIES_SPINDA", "Spinda", 327),
        ("SPECIES_WINGULL", "Wingull", 278),
        ("SPECIES_PELIPPER", "Pelipper", 279),
        ("SPECIES_SURSKIT", "Surskit", 283),
        ("SPECIES_MASQUERAIN", "Masquerain", 284),
        ("SPECIES_WAILMER", "Wailmer", 320),
        ("SPECIES_WAILORD", "Wailord", 321),
        ("SPECIES_SKITTY", "Skitty", 300),
        ("SPECIES_DELCATTY", "Delcatty", 301),
        ("SPECIES_KECLEON", "Kecleon", 352),
        ("SPECIES_BALTOY", "Baltoy", 343),
        ("SPECIES_CLAYDOL", "Claydol", 344),
        ("SPECIES_NOSEPASS", "Nosepass", 299),
        ("SPECIES_TORKOAL", "Torkoal", 324),
        ("SPECIES_SABLEYE", "Sableye", 302),
        ("SPECIES_BARBOACH", "Barboach", 339),
        ("SPECIES_WHISCASH", "Whiscash", 340),
        ("SPECIES_LUVDISC", "Luvdisc", 370),
        ("SPECIES_CORPHISH", "Corphish", 341),
        ("SPECIES_CRAWDAUNT", "Crawdaunt", 342),
        ("SPECIES_FEEBAS", "Feebas", 349),
        ("SPECIES_MILOTIC", "Milotic", 350),
        ("SPECIES_CARVANHA", "Carvanha", 318),
        ("SPECIES_SHARPEDO", "Sharpedo", 319),
        ("SPECIES_TRAPINCH", "Trapinch", 328),
        ("SPECIES_VIBRAVA", "Vibrava", 329),
        ("SPECIES_FLYGON", "Flygon", 330),
        ("SPECIES_MAKUHITA", "Makuhita", 296),
        ("SPECIES_HARIYAMA", "Hariyama", 297),
        ("SPECIES_ELECTRIKE", "Electrike", 309),
        ("SPECIES_MANECTRIC", "Manectric", 310),
        ("SPECIES_NUMEL", "Numel", 322),
        ("SPECIES_CAMERUPT", "Camerupt", 323),
        ("SPECIES_SPHEAL", "Spheal", 363),
        ("SPECIES_SEALEO", "Sealeo", 364),
        ("SPECIES_WALREIN", "Walrein", 365),
        ("SPECIES_CACNEA", "Cacnea", 331),
        ("SPECIES_CACTURNE", "Cacturne", 332),
        ("SPECIES_SNORUNT", "Snorunt", 361),
        ("SPECIES_GLALIE", "Glalie", 362),
        ("SPECIES_LUNATONE", "Lunatone", 337),
        ("SPECIES_SOLROCK", "Solrock", 338),
        ("SPECIES_AZURILL", "Azurill", 298),
        ("SPECIES_SPOINK", "Spoink", 325),
        ("SPECIES_GRUMPIG", "Grumpig", 326),
        ("SPECIES_PLUSLE", "Plusle", 311),
        ("SPECIES_MINUN", "Minun", 312),
        ("SPECIES_MAWILE", "Mawile", 303),
        ("SPECIES_MEDITITE", "Meditite", 307),
        ("SPECIES_MEDICHAM", "Medicham", 308),
        ("SPECIES_SWABLU", "Swablu", 333),
        ("SPECIES_ALTARIA", "Altaria", 334),
        ("SPECIES_WYNAUT", "Wynaut", 360),
        ("SPECIES_DUSKULL", "Duskull", 355),
        ("SPECIES_DUSCLOPS", "Dusclops", 356),
        ("SPECIES_ROSELIA", "Roselia", 315),
        ("SPECIES_SLAKOTH", "Slakoth", 287),
        ("SPECIES_VIGOROTH", "Vigoroth", 288),
        ("SPECIES_SLAKING", "Slaking", 289),
        ("SPECIES_GULPIN", "Gulpin", 316),
        ("SPECIES_SWALOT", "Swalot", 317),
        ("SPECIES_TROPIUS", "Tropius", 357),
        ("SPECIES_WHISMUR", "Whismur", 293),
        ("SPECIES_LOUDRED", "Loudred", 294),
        ("SPECIES_EXPLOUD", "Exploud", 295),
        ("SPECIES_CLAMPERL", "Clamperl", 366),
        ("SPECIES_HUNTAIL", "Huntail", 367),
        ("SPECIES_GOREBYSS", "Gorebyss", 368),
        ("SPECIES_ABSOL", "Absol", 359),
        ("SPECIES_SHUPPET", "Shuppet", 353),
        ("SPECIES_BANETTE", "Banette", 354),
        ("SPECIES_SEVIPER", "Seviper", 336),
        ("SPECIES_ZANGOOSE", "Zangoose", 335),
        ("SPECIES_RELICANTH", "Relicanth", 369),
        ("SPECIES_ARON", "Aron", 304),
        ("SPECIES_LAIRON", "Lairon", 305),
        ("SPECIES_AGGRON", "Aggron", 306),
        ("SPECIES_CASTFORM", "Castform", 351),
        ("SPECIES_VOLBEAT", "Volbeat", 313),
        ("SPECIES_ILLUMISE", "Illumise", 314),
        ("SPECIES_LILEEP", "Lileep", 345),
        ("SPECIES_CRADILY", "Cradily", 346),
        ("SPECIES_ANORITH", "Anorith", 347),
        ("SPECIES_ARMALDO", "Armaldo", 348),
        ("SPECIES_RALTS", "Ralts", 280),
        ("SPECIES_KIRLIA", "Kirlia", 281),
        ("SPECIES_GARDEVOIR", "Gardevoir", 282),
        ("SPECIES_BAGON", "Bagon", 371),
        ("SPECIES_SHELGON", "Shelgon", 372),
        ("SPECIES_SALAMENCE", "Salamence", 373),
        ("SPECIES_BELDUM", "Beldum", 374),
        ("SPECIES_METANG", "Metang", 375),
        ("SPECIES_METAGROSS", "Metagross", 376),
        ("SPECIES_REGIROCK", "Regirock", 377),
        ("SPECIES_REGICE", "Regice", 378),
        ("SPECIES_REGISTEEL", "Registeel", 379),
        ("SPECIES_KYOGRE", "Kyogre", 382),
        ("SPECIES_GROUDON", "Groudon", 383),
        ("SPECIES_RAYQUAZA", "Rayquaza", 384),
        ("SPECIES_LATIAS", "Latias", 380),
        ("SPECIES_LATIOS", "Latios", 381),
        ("SPECIES_JIRACHI", "Jirachi", 385),
        ("SPECIES_DEOXYS", "Deoxys", 386),
        ("SPECIES_CHIMECHO", "Chimecho", 358),
    ]

    max_species_id = 0
    for species_name, species_label, species_dex_number in all_species:
        species_id = data.constants[species_name]
        max_species_id = max(species_id, max_species_id)
        species_data = extracted_data["species"][species_id]

        learnset = [LearnsetMove(item["level"], item["move_id"]) for item in species_data["learnset"]["moves"]]

        data.species[species_id] = SpeciesData(
            species_name,
            species_label,
            species_id,
            species_dex_number,
            BaseStats(
                species_data["base_stats"][0],
                species_data["base_stats"][1],
                species_data["base_stats"][2],
                species_data["base_stats"][3],
                species_data["base_stats"][4],
                species_data["base_stats"][5]
            ),
            (species_data["types"][0], species_data["types"][1]),
            (species_data["abilities"][0], species_data["abilities"][1]),
            [EvolutionData(
                EvolutionMethodEnum[evolution_json["method"]],
                evolution_json["param"],
                evolution_json["species"],
            ) for evolution_json in species_data["evolutions"]],
            None,
            species_data["catch_rate"],
            species_data["friendship"],
            learnset,
            int(species_data["tmhm_learnset"], 16),
            species_data["learnset"]["address"],
            species_data["address"]
        )

    for species in data.species.values():
        for evolution in species.evolutions:
            data.species[evolution.species_id].pre_evolution = species.species_id

    # Replace default item for dex entry locations based on evo stage of species
    evo_stage_to_ball_map: Dict[int, int] = {
        0: data.constants["ITEM_POKE_BALL"],
        1: data.constants["ITEM_GREAT_BALL"],
        2: data.constants["ITEM_ULTRA_BALL"],
    }

    for species in data.species.values():
        default_item: Optional[int] = None
        pre_evolution = species.pre_evolution

        if pre_evolution is not None:
            evo_data = next(evo for evo in data.species[pre_evolution].evolutions if evo.species_id == species.species_id)
            if evo_data.method == EvolutionMethodEnum.ITEM:
                default_item = evo_data.param

        evo_stage = 0
        if default_item is None:
            while pre_evolution is not None:
                evo_stage += 1
                pre_evolution = data.species[pre_evolution].pre_evolution
            default_item = evo_stage_to_ball_map[evo_stage]

        dex_location_name = f"POKEDEX_REWARD_{str(species.national_dex_number).zfill(3)}"
        data.locations[dex_location_name] = LocationData(
            data.locations[dex_location_name].name,
            data.locations[dex_location_name].label,
            data.locations[dex_location_name].parent_region,
            default_item,
            data.locations[dex_location_name].address,
            data.locations[dex_location_name].flag,
            data.locations[dex_location_name].category,
            data.locations[dex_location_name].tags
        )

    # Create legendary encounter data
    for legendary_encounter_json in extracted_data["legendary_encounters"]:
        data.legendary_encounters.append(MiscPokemonData(
            legendary_encounter_json["species"],
            legendary_encounter_json["address"]
        ))

    for misc_pokemon_json in extracted_data["misc_pokemon"]:
        data.misc_pokemon.append(MiscPokemonData(
            misc_pokemon_json["species"],
            misc_pokemon_json["address"]
        ))

    # TM moves
    data.tmhm_moves = extracted_data["tmhm_moves"]

    # Create ability data
    data.abilities = [AbilityData(data.constants[ability_data[0]], ability_data[1]) for ability_data in [
        ("ABILITY_STENCH", "Stench"),
        ("ABILITY_DRIZZLE", "Drizzle"),
        ("ABILITY_SPEED_BOOST", "Speed Boost"),
        ("ABILITY_BATTLE_ARMOR", "Battle Armor"),
        ("ABILITY_STURDY", "Sturdy"),
        ("ABILITY_DAMP", "Damp"),
        ("ABILITY_LIMBER", "Limber"),
        ("ABILITY_SAND_VEIL", "Sand Veil"),
        ("ABILITY_STATIC", "Static"),
        ("ABILITY_VOLT_ABSORB", "Volt Absorb"),
        ("ABILITY_WATER_ABSORB", "Water Absorb"),
        ("ABILITY_OBLIVIOUS", "Oblivious"),
        ("ABILITY_CLOUD_NINE", "Cloud Nine"),
        ("ABILITY_COMPOUND_EYES", "Compoundeyes"),
        ("ABILITY_INSOMNIA", "Insomnia"),
        ("ABILITY_COLOR_CHANGE", "Color Change"),
        ("ABILITY_IMMUNITY", "Immunity"),
        ("ABILITY_FLASH_FIRE", "Flash Fire"),
        ("ABILITY_SHIELD_DUST", "Shield Dust"),
        ("ABILITY_OWN_TEMPO", "Own Tempo"),
        ("ABILITY_SUCTION_CUPS", "Suction Cups"),
        ("ABILITY_INTIMIDATE", "Intimidate"),
        ("ABILITY_SHADOW_TAG", "Shadow Tag"),
        ("ABILITY_ROUGH_SKIN", "Rough Skin"),
        ("ABILITY_WONDER_GUARD", "Wonder Guard"),
        ("ABILITY_LEVITATE", "Levitate"),
        ("ABILITY_EFFECT_SPORE", "Effect Spore"),
        ("ABILITY_SYNCHRONIZE", "Synchronize"),
        ("ABILITY_CLEAR_BODY", "Clear Body"),
        ("ABILITY_NATURAL_CURE", "Natural Cure"),
        ("ABILITY_LIGHTNING_ROD", "Lightningrod"),
        ("ABILITY_SERENE_GRACE", "Serene Grace"),
        ("ABILITY_SWIFT_SWIM", "Swift Swim"),
        ("ABILITY_CHLOROPHYLL", "Chlorophyll"),
        ("ABILITY_ILLUMINATE", "Illuminate"),
        ("ABILITY_TRACE", "Trace"),
        ("ABILITY_HUGE_POWER", "Huge Power"),
        ("ABILITY_POISON_POINT", "Poison Point"),
        ("ABILITY_INNER_FOCUS", "Inner Focus"),
        ("ABILITY_MAGMA_ARMOR", "Magma Armor"),
        ("ABILITY_WATER_VEIL", "Water Veil"),
        ("ABILITY_MAGNET_PULL", "Magnet Pull"),
        ("ABILITY_SOUNDPROOF", "Soundproof"),
        ("ABILITY_RAIN_DISH", "Rain Dish"),
        ("ABILITY_SAND_STREAM", "Sand Stream"),
        ("ABILITY_PRESSURE", "Pressure"),
        ("ABILITY_THICK_FAT", "Thick Fat"),
        ("ABILITY_EARLY_BIRD", "Early Bird"),
        ("ABILITY_FLAME_BODY", "Flame Body"),
        ("ABILITY_RUN_AWAY", "Run Away"),
        ("ABILITY_KEEN_EYE", "Keen Eye"),
        ("ABILITY_HYPER_CUTTER", "Hyper Cutter"),
        ("ABILITY_PICKUP", "Pickup"),
        ("ABILITY_TRUANT", "Truant"),
        ("ABILITY_HUSTLE", "Hustle"),
        ("ABILITY_CUTE_CHARM", "Cute Charm"),
        ("ABILITY_PLUS", "Plus"),
        ("ABILITY_MINUS", "Minus"),
        ("ABILITY_FORECAST", "Forecast"),
        ("ABILITY_STICKY_HOLD", "Sticky Hold"),
        ("ABILITY_SHED_SKIN", "Shed Skin"),
        ("ABILITY_GUTS", "Guts"),
        ("ABILITY_MARVEL_SCALE", "Marvel Scale"),
        ("ABILITY_LIQUID_OOZE", "Liquid Ooze"),
        ("ABILITY_OVERGROW", "Overgrow"),
        ("ABILITY_BLAZE", "Blaze"),
        ("ABILITY_TORRENT", "Torrent"),
        ("ABILITY_SWARM", "Swarm"),
        ("ABILITY_ROCK_HEAD", "Rock Head"),
        ("ABILITY_DROUGHT", "Drought"),
        ("ABILITY_ARENA_TRAP", "Arena Trap"),
        ("ABILITY_VITAL_SPIRIT", "Vital Spirit"),
        ("ABILITY_WHITE_SMOKE", "White Smoke"),
        ("ABILITY_PURE_POWER", "Pure Power"),
        ("ABILITY_SHELL_ARMOR", "Shell Armor"),
        ("ABILITY_CACOPHONY", "Cacophony"),
        ("ABILITY_AIR_LOCK", "Air Lock")
    ]]

    # Move labels
    data.move_labels = {r: data.constants[l] for l, r in [
        ("MOVE_POUND", "Pound"),
        ("MOVE_KARATE_CHOP", "Karate Chop"),
        ("MOVE_DOUBLE_SLAP", "Doubleslap"),
        ("MOVE_COMET_PUNCH", "Comet Punch"),
        ("MOVE_MEGA_PUNCH", "Mega Punch"),
        ("MOVE_PAY_DAY", "Pay Day"),
        ("MOVE_FIRE_PUNCH", "Fire Punch"),
        ("MOVE_ICE_PUNCH", "Ice Punch"),
        ("MOVE_THUNDER_PUNCH", "Thunderpunch"),
        ("MOVE_SCRATCH", "Scratch"),
        ("MOVE_VICE_GRIP", "Vicegrip"),
        ("MOVE_GUILLOTINE", "Guillotine"),
        ("MOVE_RAZOR_WIND", "Razor Wind"),
        ("MOVE_SWORDS_DANCE", "Swords Dance"),
        ("MOVE_CUT", "Cut"),
        ("MOVE_GUST", "Gust"),
        ("MOVE_WING_ATTACK", "Wing Attack"),
        ("MOVE_WHIRLWIND", "Whirlwind"),
        ("MOVE_FLY", "Fly"),
        ("MOVE_BIND", "Bind"),
        ("MOVE_SLAM", "Slam"),
        ("MOVE_VINE_WHIP", "Vine Whip"),
        ("MOVE_STOMP", "Stomp"),
        ("MOVE_DOUBLE_KICK", "Double Kick"),
        ("MOVE_MEGA_KICK", "Mega Kick"),
        ("MOVE_JUMP_KICK", "Jump Kick"),
        ("MOVE_ROLLING_KICK", "Rolling Kick"),
        ("MOVE_SAND_ATTACK", "Sand-Attack"),
        ("MOVE_HEADBUTT", "Headbutt"),
        ("MOVE_HORN_ATTACK", "Horn Attack"),
        ("MOVE_FURY_ATTACK", "Fury Attack"),
        ("MOVE_HORN_DRILL", "Horn Drill"),
        ("MOVE_TACKLE", "Tackle"),
        ("MOVE_BODY_SLAM", "Body Slam"),
        ("MOVE_WRAP", "Wrap"),
        ("MOVE_TAKE_DOWN", "Take Down"),
        ("MOVE_THRASH", "Thrash"),
        ("MOVE_DOUBLE_EDGE", "Double-Edge"),
        ("MOVE_TAIL_WHIP", "Tail Whip"),
        ("MOVE_POISON_STING", "Poison Sting"),
        ("MOVE_TWINEEDLE", "Twineedle"),
        ("MOVE_PIN_MISSILE", "Pin Missile"),
        ("MOVE_LEER", "Leer"),
        ("MOVE_BITE", "Bite"),
        ("MOVE_GROWL", "Growl"),
        ("MOVE_ROAR", "Roar"),
        ("MOVE_SING", "Sing"),
        ("MOVE_SUPERSONIC", "Supersonic"),
        ("MOVE_SONIC_BOOM", "Sonicboom"),
        ("MOVE_DISABLE", "Disable"),
        ("MOVE_ACID", "Acid"),
        ("MOVE_EMBER", "Ember"),
        ("MOVE_FLAMETHROWER", "Flamethrower"),
        ("MOVE_MIST", "Mist"),
        ("MOVE_WATER_GUN", "Water Gun"),
        ("MOVE_HYDRO_PUMP", "Hydro Pump"),
        ("MOVE_SURF", "Surf"),
        ("MOVE_ICE_BEAM", "Ice Beam"),
        ("MOVE_BLIZZARD", "Blizzard"),
        ("MOVE_PSYBEAM", "Psybeam"),
        ("MOVE_BUBBLE_BEAM", "Bubblebeam"),
        ("MOVE_AURORA_BEAM", "Aurora Beam"),
        ("MOVE_HYPER_BEAM", "Hyper Beam"),
        ("MOVE_PECK", "Peck"),
        ("MOVE_DRILL_PECK", "Drill Peck"),
        ("MOVE_SUBMISSION", "Submission"),
        ("MOVE_LOW_KICK", "Low Kick"),
        ("MOVE_COUNTER", "Counter"),
        ("MOVE_SEISMIC_TOSS", "Seismic Toss"),
        ("MOVE_STRENGTH", "Strength"),
        ("MOVE_ABSORB", "Absorb"),
        ("MOVE_MEGA_DRAIN", "Mega Drain"),
        ("MOVE_LEECH_SEED", "Leech Seed"),
        ("MOVE_GROWTH", "Growth"),
        ("MOVE_RAZOR_LEAF", "Razor Leaf"),
        ("MOVE_SOLAR_BEAM", "Solarbeam"),
        ("MOVE_POISON_POWDER", "Poisonpowder"),
        ("MOVE_STUN_SPORE", "Stun Spore"),
        ("MOVE_SLEEP_POWDER", "Sleep Powder"),
        ("MOVE_PETAL_DANCE", "Petal Dance"),
        ("MOVE_STRING_SHOT", "String Shot"),
        ("MOVE_DRAGON_RAGE", "Dragon Rage"),
        ("MOVE_FIRE_SPIN", "Fire Spin"),
        ("MOVE_THUNDER_SHOCK", "Thundershock"),
        ("MOVE_THUNDERBOLT", "Thunderbolt"),
        ("MOVE_THUNDER_WAVE", "Thunder Wave"),
        ("MOVE_THUNDER", "Thunder"),
        ("MOVE_ROCK_THROW", "Rock Throw"),
        ("MOVE_EARTHQUAKE", "Earthquake"),
        ("MOVE_FISSURE", "Fissure"),
        ("MOVE_DIG", "Dig"),
        ("MOVE_TOXIC", "Toxic"),
        ("MOVE_CONFUSION", "Confusion"),
        ("MOVE_PSYCHIC", "Psychic"),
        ("MOVE_HYPNOSIS", "Hypnosis"),
        ("MOVE_MEDITATE", "Meditate"),
        ("MOVE_AGILITY", "Agility"),
        ("MOVE_QUICK_ATTACK", "Quick Attack"),
        ("MOVE_RAGE", "Rage"),
        ("MOVE_TELEPORT", "Teleport"),
        ("MOVE_NIGHT_SHADE", "Night Shade"),
        ("MOVE_MIMIC", "Mimic"),
        ("MOVE_SCREECH", "Screech"),
        ("MOVE_DOUBLE_TEAM", "Double Team"),
        ("MOVE_RECOVER", "Recover"),
        ("MOVE_HARDEN", "Harden"),
        ("MOVE_MINIMIZE", "Minimize"),
        ("MOVE_SMOKESCREEN", "Smokescreen"),
        ("MOVE_CONFUSE_RAY", "Confuse Ray"),
        ("MOVE_WITHDRAW", "Withdraw"),
        ("MOVE_DEFENSE_CURL", "Defense Curl"),
        ("MOVE_BARRIER", "Barrier"),
        ("MOVE_LIGHT_SCREEN", "Light Screen"),
        ("MOVE_HAZE", "Haze"),
        ("MOVE_REFLECT", "Reflect"),
        ("MOVE_FOCUS_ENERGY", "Focus Energy"),
        ("MOVE_BIDE", "Bide"),
        ("MOVE_METRONOME", "Metronome"),
        ("MOVE_MIRROR_MOVE", "Mirror Move"),
        ("MOVE_SELF_DESTRUCT", "Selfdestruct"),
        ("MOVE_EGG_BOMB", "Egg Bomb"),
        ("MOVE_LICK", "Lick"),
        ("MOVE_SMOG", "Smog"),
        ("MOVE_SLUDGE", "Sludge"),
        ("MOVE_BONE_CLUB", "Bone Club"),
        ("MOVE_FIRE_BLAST", "Fire Blast"),
        ("MOVE_WATERFALL", "Waterfall"),
        ("MOVE_CLAMP", "Clamp"),
        ("MOVE_SWIFT", "Swift"),
        ("MOVE_SKULL_BASH", "Skull Bash"),
        ("MOVE_SPIKE_CANNON", "Spike Cannon"),
        ("MOVE_CONSTRICT", "Constrict"),
        ("MOVE_AMNESIA", "Amnesia"),
        ("MOVE_KINESIS", "Kinesis"),
        ("MOVE_SOFT_BOILED", "Softboiled"),
        ("MOVE_HI_JUMP_KICK", "Hi Jump Kick"),
        ("MOVE_GLARE", "Glare"),
        ("MOVE_DREAM_EATER", "Dream Eater"),
        ("MOVE_POISON_GAS", "Poison Gas"),
        ("MOVE_BARRAGE", "Barrage"),
        ("MOVE_LEECH_LIFE", "Leech Life"),
        ("MOVE_LOVELY_KISS", "Lovely Kiss"),
        ("MOVE_SKY_ATTACK", "Sky Attack"),
        ("MOVE_TRANSFORM", "Transform"),
        ("MOVE_BUBBLE", "Bubble"),
        ("MOVE_DIZZY_PUNCH", "Dizzy Punch"),
        ("MOVE_SPORE", "Spore"),
        ("MOVE_FLASH", "Flash"),
        ("MOVE_PSYWAVE", "Psywave"),
        ("MOVE_SPLASH", "Splash"),
        ("MOVE_ACID_ARMOR", "Acid Armor"),
        ("MOVE_CRABHAMMER", "Crabhammer"),
        ("MOVE_EXPLOSION", "Explosion"),
        ("MOVE_FURY_SWIPES", "Fury Swipes"),
        ("MOVE_BONEMERANG", "Bonemerang"),
        ("MOVE_REST", "Rest"),
        ("MOVE_ROCK_SLIDE", "Rock Slide"),
        ("MOVE_HYPER_FANG", "Hyper Fang"),
        ("MOVE_SHARPEN", "Sharpen"),
        ("MOVE_CONVERSION", "Conversion"),
        ("MOVE_TRI_ATTACK", "Tri Attack"),
        ("MOVE_SUPER_FANG", "Super Fang"),
        ("MOVE_SLASH", "Slash"),
        ("MOVE_SUBSTITUTE", "Substitute"),
        ("MOVE_SKETCH", "Sketch"),
        ("MOVE_TRIPLE_KICK", "Triple Kick"),
        ("MOVE_THIEF", "Thief"),
        ("MOVE_SPIDER_WEB", "Spider Web"),
        ("MOVE_MIND_READER", "Mind Reader"),
        ("MOVE_NIGHTMARE", "Nightmare"),
        ("MOVE_FLAME_WHEEL", "Flame Wheel"),
        ("MOVE_SNORE", "Snore"),
        ("MOVE_CURSE", "Curse"),
        ("MOVE_FLAIL", "Flail"),
        ("MOVE_CONVERSION_2", "Conversion 2"),
        ("MOVE_AEROBLAST", "Aeroblast"),
        ("MOVE_COTTON_SPORE", "Cotton Spore"),
        ("MOVE_REVERSAL", "Reversal"),
        ("MOVE_SPITE", "Spite"),
        ("MOVE_POWDER_SNOW", "Powder Snow"),
        ("MOVE_PROTECT", "Protect"),
        ("MOVE_MACH_PUNCH", "Mach Punch"),
        ("MOVE_SCARY_FACE", "Scary Face"),
        ("MOVE_FAINT_ATTACK", "Faint Attack"),
        ("MOVE_SWEET_KISS", "Sweet Kiss"),
        ("MOVE_BELLY_DRUM", "Belly Drum"),
        ("MOVE_SLUDGE_BOMB", "Sludge Bomb"),
        ("MOVE_MUD_SLAP", "Mud-Slap"),
        ("MOVE_OCTAZOOKA", "Octazooka"),
        ("MOVE_SPIKES", "Spikes"),
        ("MOVE_ZAP_CANNON", "Zap Cannon"),
        ("MOVE_FORESIGHT", "Foresight"),
        ("MOVE_DESTINY_BOND", "Destiny Bond"),
        ("MOVE_PERISH_SONG", "Perish Song"),
        ("MOVE_ICY_WIND", "Icy Wind"),
        ("MOVE_DETECT", "Detect"),
        ("MOVE_BONE_RUSH", "Bone Rush"),
        ("MOVE_LOCK_ON", "Lock-On"),
        ("MOVE_OUTRAGE", "Outrage"),
        ("MOVE_SANDSTORM", "Sandstorm"),
        ("MOVE_GIGA_DRAIN", "Giga Drain"),
        ("MOVE_ENDURE", "Endure"),
        ("MOVE_CHARM", "Charm"),
        ("MOVE_ROLLOUT", "Rollout"),
        ("MOVE_FALSE_SWIPE", "False Swipe"),
        ("MOVE_SWAGGER", "Swagger"),
        ("MOVE_MILK_DRINK", "Milk Drink"),
        ("MOVE_SPARK", "Spark"),
        ("MOVE_FURY_CUTTER", "Fury Cutter"),
        ("MOVE_STEEL_WING", "Steel Wing"),
        ("MOVE_MEAN_LOOK", "Mean Look"),
        ("MOVE_ATTRACT", "Attract"),
        ("MOVE_SLEEP_TALK", "Sleep Talk"),
        ("MOVE_HEAL_BELL", "Heal Bell"),
        ("MOVE_RETURN", "Return"),
        ("MOVE_PRESENT", "Present"),
        ("MOVE_FRUSTRATION", "Frustration"),
        ("MOVE_SAFEGUARD", "Safeguard"),
        ("MOVE_PAIN_SPLIT", "Pain Split"),
        ("MOVE_SACRED_FIRE", "Sacred Fire"),
        ("MOVE_MAGNITUDE", "Magnitude"),
        ("MOVE_DYNAMIC_PUNCH", "Dynamicpunch"),
        ("MOVE_MEGAHORN", "Megahorn"),
        ("MOVE_DRAGON_BREATH", "Dragonbreath"),
        ("MOVE_BATON_PASS", "Baton Pass"),
        ("MOVE_ENCORE", "Encore"),
        ("MOVE_PURSUIT", "Pursuit"),
        ("MOVE_RAPID_SPIN", "Rapid Spin"),
        ("MOVE_SWEET_SCENT", "Sweet Scent"),
        ("MOVE_IRON_TAIL", "Iron Tail"),
        ("MOVE_METAL_CLAW", "Metal Claw"),
        ("MOVE_VITAL_THROW", "Vital Throw"),
        ("MOVE_MORNING_SUN", "Morning Sun"),
        ("MOVE_SYNTHESIS", "Synthesis"),
        ("MOVE_MOONLIGHT", "Moonlight"),
        ("MOVE_HIDDEN_POWER", "Hidden Power"),
        ("MOVE_CROSS_CHOP", "Cross Chop"),
        ("MOVE_TWISTER", "Twister"),
        ("MOVE_RAIN_DANCE", "Rain Dance"),
        ("MOVE_SUNNY_DAY", "Sunny Day"),
        ("MOVE_CRUNCH", "Crunch"),
        ("MOVE_MIRROR_COAT", "Mirror Coat"),
        ("MOVE_PSYCH_UP", "Psych Up"),
        ("MOVE_EXTREME_SPEED", "Extremespeed"),
        ("MOVE_ANCIENT_POWER", "Ancientpower"),
        ("MOVE_SHADOW_BALL", "Shadow Ball"),
        ("MOVE_FUTURE_SIGHT", "Future Sight"),
        ("MOVE_ROCK_SMASH", "Rock Smash"),
        ("MOVE_WHIRLPOOL", "Whirlpool"),
        ("MOVE_BEAT_UP", "Beat Up"),
        ("MOVE_FAKE_OUT", "Fake Out"),
        ("MOVE_UPROAR", "Uproar"),
        ("MOVE_STOCKPILE", "Stockpile"),
        ("MOVE_SPIT_UP", "Spit Up"),
        ("MOVE_SWALLOW", "Swallow"),
        ("MOVE_HEAT_WAVE", "Heat Wave"),
        ("MOVE_HAIL", "Hail"),
        ("MOVE_TORMENT", "Torment"),
        ("MOVE_FLATTER", "Flatter"),
        ("MOVE_WILL_O_WISP", "Will-O-Wisp"),
        ("MOVE_MEMENTO", "Memento"),
        ("MOVE_FACADE", "Facade"),
        ("MOVE_FOCUS_PUNCH", "Focus Punch"),
        ("MOVE_SMELLING_SALT", "Smellingsalt"),
        ("MOVE_FOLLOW_ME", "Follow Me"),
        ("MOVE_NATURE_POWER", "Nature Power"),
        ("MOVE_CHARGE", "Charge"),
        ("MOVE_TAUNT", "Taunt"),
        ("MOVE_HELPING_HAND", "Helping Hand"),
        ("MOVE_TRICK", "Trick"),
        ("MOVE_ROLE_PLAY", "Role Play"),
        ("MOVE_WISH", "Wish"),
        ("MOVE_ASSIST", "Assist"),
        ("MOVE_INGRAIN", "Ingrain"),
        ("MOVE_SUPERPOWER", "Superpower"),
        ("MOVE_MAGIC_COAT", "Magic Coat"),
        ("MOVE_RECYCLE", "Recycle"),
        ("MOVE_REVENGE", "Revenge"),
        ("MOVE_BRICK_BREAK", "Brick Break"),
        ("MOVE_YAWN", "Yawn"),
        ("MOVE_KNOCK_OFF", "Knock Off"),
        ("MOVE_ENDEAVOR", "Endeavor"),
        ("MOVE_ERUPTION", "Eruption"),
        ("MOVE_SKILL_SWAP", "Skill Swap"),
        ("MOVE_IMPRISON", "Imprison"),
        ("MOVE_REFRESH", "Refresh"),
        ("MOVE_GRUDGE", "Grudge"),
        ("MOVE_SNATCH", "Snatch"),
        ("MOVE_SECRET_POWER", "Secret Power"),
        ("MOVE_DIVE", "Dive"),
        ("MOVE_ARM_THRUST", "Arm Thrust"),
        ("MOVE_CAMOUFLAGE", "Camouflage"),
        ("MOVE_TAIL_GLOW", "Tail Glow"),
        ("MOVE_LUSTER_PURGE", "Luster Purge"),
        ("MOVE_MIST_BALL", "Mist Ball"),
        ("MOVE_FEATHER_DANCE", "Featherdance"),
        ("MOVE_TEETER_DANCE", "Teeter Dance"),
        ("MOVE_BLAZE_KICK", "Blaze Kick"),
        ("MOVE_MUD_SPORT", "Mud Sport"),
        ("MOVE_ICE_BALL", "Ice Ball"),
        ("MOVE_NEEDLE_ARM", "Needle Arm"),
        ("MOVE_SLACK_OFF", "Slack Off"),
        ("MOVE_HYPER_VOICE", "Hyper Voice"),
        ("MOVE_POISON_FANG", "Poison Fang"),
        ("MOVE_CRUSH_CLAW", "Crush Claw"),
        ("MOVE_BLAST_BURN", "Blast Burn"),
        ("MOVE_HYDRO_CANNON", "Hydro Cannon"),
        ("MOVE_METEOR_MASH", "Meteor Mash"),
        ("MOVE_ASTONISH", "Astonish"),
        ("MOVE_WEATHER_BALL", "Weather Ball"),
        ("MOVE_AROMATHERAPY", "Aromatherapy"),
        ("MOVE_FAKE_TEARS", "Fake Tears"),
        ("MOVE_AIR_CUTTER", "Air Cutter"),
        ("MOVE_OVERHEAT", "Overheat"),
        ("MOVE_ODOR_SLEUTH", "Odor Sleuth"),
        ("MOVE_ROCK_TOMB", "Rock Tomb"),
        ("MOVE_SILVER_WIND", "Silver Wind"),
        ("MOVE_METAL_SOUND", "Metal Sound"),
        ("MOVE_GRASS_WHISTLE", "Grasswhistle"),
        ("MOVE_TICKLE", "Tickle"),
        ("MOVE_COSMIC_POWER", "Cosmic Power"),
        ("MOVE_WATER_SPOUT", "Water Spout"),
        ("MOVE_SIGNAL_BEAM", "Signal Beam"),
        ("MOVE_SHADOW_PUNCH", "Shadow Punch"),
        ("MOVE_EXTRASENSORY", "Extrasensory"),
        ("MOVE_SKY_UPPERCUT", "Sky Uppercut"),
        ("MOVE_SAND_TOMB", "Sand Tomb"),
        ("MOVE_SHEER_COLD", "Sheer Cold"),
        ("MOVE_MUDDY_WATER", "Muddy Water"),
        ("MOVE_BULLET_SEED", "Bullet Seed"),
        ("MOVE_AERIAL_ACE", "Aerial Ace"),
        ("MOVE_ICICLE_SPEAR", "Icicle Spear"),
        ("MOVE_IRON_DEFENSE", "Iron Defense"),
        ("MOVE_BLOCK", "Block"),
        ("MOVE_HOWL", "Howl"),
        ("MOVE_DRAGON_CLAW", "Dragon Claw"),
        ("MOVE_FRENZY_PLANT", "Frenzy Plant"),
        ("MOVE_BULK_UP", "Bulk Up"),
        ("MOVE_BOUNCE", "Bounce"),
        ("MOVE_MUD_SHOT", "Mud Shot"),
        ("MOVE_POISON_TAIL", "Poison Tail"),
        ("MOVE_COVET", "Covet"),
        ("MOVE_VOLT_TACKLE", "Volt Tackle"),
        ("MOVE_MAGICAL_LEAF", "Magical Leaf"),
        ("MOVE_WATER_SPORT", "Water Sport"),
        ("MOVE_CALM_MIND", "Calm Mind"),
        ("MOVE_LEAF_BLADE", "Leaf Blade"),
        ("MOVE_DRAGON_DANCE", "Dragon Dance"),
        ("MOVE_ROCK_BLAST", "Rock Blast"),
        ("MOVE_SHOCK_WAVE", "Shock Wave"),
        ("MOVE_WATER_PULSE", "Water Pulse"),
        ("MOVE_DOOM_DESIRE", "Doom Desire"),
        ("MOVE_PSYCHO_BOOST", "Psycho Boost"),
    ]}

    # Create warp map
    for warp, destination in extracted_data["warps"].items():
        data.warp_map[warp] = None if destination == "" else destination

    # Create trainer data
    for i, trainer_json in enumerate(extracted_data["trainers"]):
        party_json = trainer_json["party"]
        pokemon_data_type = _str_to_pokemon_data_type(trainer_json["data_type"])
        data.trainers.append(TrainerData(
            i,
            TrainerPartyData(
                [TrainerPokemonData(
                    p["species"],
                    p["level"],
                    (p["moves"][0], p["moves"][1], p["moves"][2], p["moves"][3]) if "moves" in p else None
                ) for p in party_json],
                pokemon_data_type,
                trainer_json["party_address"]
            ),
            trainer_json["address"],
            trainer_json["script_address"],
            trainer_json["battle_type"]
        ))


data = PokemonEmeraldData()
_init()

LEGENDARY_POKEMON = frozenset([data.constants[species] for species in [
    "SPECIES_ARTICUNO",
    "SPECIES_ZAPDOS",
    "SPECIES_MOLTRES",
    "SPECIES_MEWTWO",
    "SPECIES_MEW",
    "SPECIES_RAIKOU",
    "SPECIES_ENTEI",
    "SPECIES_SUICUNE",
    "SPECIES_LUGIA",
    "SPECIES_HO_OH",
    "SPECIES_CELEBI",
    "SPECIES_REGIROCK",
    "SPECIES_REGICE",
    "SPECIES_REGISTEEL",
    "SPECIES_LATIAS",
    "SPECIES_LATIOS",
    "SPECIES_KYOGRE",
    "SPECIES_GROUDON",
    "SPECIES_RAYQUAZA",
    "SPECIES_JIRACHI",
    "SPECIES_DEOXYS",
]])
"""Species IDs of legendary pokemon"""

UNEVOLVED_POKEMON = frozenset({
    species.species_id
    for species in data.species.values()
    if len(species.evolutions) > 0
})
"""Species IDs of pokemon which have further evolution stages in the vanilla game"""

NATIONAL_ID_TO_SPECIES_ID = {species.national_dex_number: i for i, species in data.species.items()}