""" 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 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""" POSTGAME_MAPS = { "MAP_DESERT_UNDERPASS", "MAP_SAFARI_ZONE_NORTHEAST", "MAP_SAFARI_ZONE_SOUTHEAST", "MAP_METEOR_FALLS_STEVENS_CAVE", } """These maps have encounters and are locked behind beating the champion. 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 LocationData(NamedTuple): name: str label: str parent_region: str default_item: int address: Union[int, List[int]] flag: int tags: FrozenSet[str] class EncounterTableData(NamedTuple): slots: List[int] address: int @dataclass class MapData: name: str header_address: int land_encounters: Optional[EncounterTableData] water_encounters: Optional[EncounterTableData] fishing_encounters: Optional[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 def _str_to_evolution_method(string: str) -> EvolutionMethodEnum: if string == "LEVEL": return EvolutionMethodEnum.LEVEL if string == "LEVEL_ATK_LT_DEF": return EvolutionMethodEnum.LEVEL_ATK_LT_DEF if string == "LEVEL_ATK_EQ_DEF": return EvolutionMethodEnum.LEVEL_ATK_EQ_DEF if string == "LEVEL_ATK_GT_DEF": return EvolutionMethodEnum.LEVEL_ATK_GT_DEF if string == "LEVEL_SILCOON": return EvolutionMethodEnum.LEVEL_SILCOON if string == "LEVEL_CASCOON": return EvolutionMethodEnum.LEVEL_CASCOON if string == "LEVEL_NINJASK": return EvolutionMethodEnum.LEVEL_NINJASK if string == "LEVEL_SHEDINJA": return EvolutionMethodEnum.LEVEL_SHEDINJA if string == "FRIENDSHIP": return EvolutionMethodEnum.FRIENDSHIP if string == "FRIENDSHIP_DAY": return EvolutionMethodEnum.FRIENDSHIP_DAY if string == "FRIENDSHIP_NIGHT": return EvolutionMethodEnum.FRIENDSHIP_NIGHT class EvolutionData(NamedTuple): method: EvolutionMethodEnum param: int 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 @dataclass class TrainerPokemonData: species_id: int level: int moves: Optional[Tuple[int, int, int, int]] @dataclass class TrainerPartyData: pokemon: List[TrainerPokemonData] pokemon_data_type: TrainerPokemonDataTypeEnum address: int @dataclass class TrainerData: trainer_id: int party: TrainerPartyData address: int script_address: 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: 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(): if map_name in IGNORABLE_MAPS: continue land_encounters = None water_encounters = None fishing_encounters = None if "land_encounters" in map_json: land_encounters = EncounterTableData( map_json["land_encounters"]["slots"], map_json["land_encounters"]["address"] ) if "water_encounters" in map_json: water_encounters = EncounterTableData( map_json["water_encounters"]["slots"], map_json["water_encounters"]["address"] ) if "fishing_encounters" in map_json: fishing_encounters = EncounterTableData( map_json["fishing_encounters"]["slots"], map_json["fishing_encounters"]["address"] ) data.maps[map_name] = MapData( map_name, map_json["header_address"], land_encounters, water_encounters, fishing_encounters ) # 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"], 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"], 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( _str_to_evolution_method(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 = { 0: data.constants["ITEM_POKE_BALL"], 1: data.constants["ITEM_GREAT_BALL"], 2: data.constants["ITEM_ULTRA_BALL"], } for species in data.species.values(): evo_stage = 0 pre_evolution = species.pre_evolution while pre_evolution is not None: evo_stage += 1 pre_evolution = data.species[pre_evolution].pre_evolution 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, evo_stage_to_ball_map[evo_stage], data.locations[dex_location_name].address, data.locations[dex_location_name].flag, 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 if encoded_warp not in data.warp_map: data.warp_map[encoded_warp] = None # 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"] )) 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()}