| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | import os | 
					
						
							|  |  |  | import hashlib | 
					
						
							|  |  |  | import Utils | 
					
						
							|  |  |  | import bsdiff4 | 
					
						
							|  |  |  | from copy import deepcopy | 
					
						
							|  |  |  | from Patch import APDeltaPatch | 
					
						
							|  |  |  | from .text import encode_text | 
					
						
							|  |  |  | from .rom_addresses import rom_addresses | 
					
						
							|  |  |  | from .locations import location_data | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  | from .items import item_table | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  | from .rock_tunnel import randomize_rock_tunnel | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | import worlds.pokemon_rb.poke_data as poke_data | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def choose_forced_type(chances, random): | 
					
						
							|  |  |  |     n = random.randint(1, 100) | 
					
						
							|  |  |  |     for chance in chances: | 
					
						
							|  |  |  |         if chance[0] >= n: | 
					
						
							|  |  |  |             return chance[1] | 
					
						
							|  |  |  |     return None | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def filter_moves(moves, type, random): | 
					
						
							|  |  |  |     ret = [] | 
					
						
							|  |  |  |     for move in moves: | 
					
						
							|  |  |  |         if poke_data.moves[move]["type"] == type or type is None: | 
					
						
							|  |  |  |             ret.append(move) | 
					
						
							|  |  |  |     random.shuffle(ret) | 
					
						
							|  |  |  |     return ret | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  | def get_move(local_move_data, moves, chances, random, starting_move=False): | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     type = choose_forced_type(chances, random) | 
					
						
							|  |  |  |     filtered_moves = filter_moves(moves, type, random) | 
					
						
							|  |  |  |     for move in filtered_moves: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |         if local_move_data[move]["accuracy"] > 80 and local_move_data[move]["power"] > 0 or not starting_move: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             moves.remove(move) | 
					
						
							|  |  |  |             return move | 
					
						
							|  |  |  |     else: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |         return get_move(local_move_data, moves, [], random, starting_move) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def get_encounter_slots(self): | 
					
						
							| 
									
										
										
										
											2022-12-08 04:51:01 -05:00
										 |  |  |     encounter_slots = deepcopy([location for location in location_data if location.type == "Wild Encounter"]) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |     for location in encounter_slots: | 
					
						
							|  |  |  |         if isinstance(location.original_item, list): | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |             location.original_item = location.original_item[not self.multiworld.game_version[self.player].value] | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     return encounter_slots | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def get_base_stat_total(mon): | 
					
						
							|  |  |  |     return (poke_data.pokemon_data[mon]["atk"] + poke_data.pokemon_data[mon]["def"] | 
					
						
							|  |  |  |             + poke_data.pokemon_data[mon]["hp"] + poke_data.pokemon_data[mon]["spd"] | 
					
						
							|  |  |  |             + poke_data.pokemon_data[mon]["spc"]) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  | def randomize_pokemon(self, mon, mons_list, randomize_type, random): | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     if randomize_type in [1, 3]: | 
					
						
							|  |  |  |         type_mons = [pokemon for pokemon in mons_list if any([poke_data.pokemon_data[mon][ | 
					
						
							|  |  |  |              "type1"] in [self.local_poke_data[pokemon]["type1"], self.local_poke_data[pokemon]["type2"]], | 
					
						
							|  |  |  |              poke_data.pokemon_data[mon]["type2"] in [self.local_poke_data[pokemon]["type1"], | 
					
						
							|  |  |  |                                                       self.local_poke_data[pokemon]["type2"]]])] | 
					
						
							|  |  |  |         if not type_mons: | 
					
						
							|  |  |  |             type_mons = mons_list.copy() | 
					
						
							|  |  |  |         if randomize_type == 3: | 
					
						
							|  |  |  |             stat_base = get_base_stat_total(mon) | 
					
						
							|  |  |  |             type_mons.sort(key=lambda mon: abs(get_base_stat_total(mon) - stat_base)) | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  |         mon = type_mons[round(random.triangular(0, len(type_mons) - 1, 0))] | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     if randomize_type == 2: | 
					
						
							|  |  |  |         stat_base = get_base_stat_total(mon) | 
					
						
							|  |  |  |         mons_list.sort(key=lambda mon: abs(get_base_stat_total(mon) - stat_base)) | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  |         mon = mons_list[round(random.triangular(0, 50, 0))] | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     elif randomize_type == 4: | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  |         mon = random.choice(mons_list) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     return mon | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  | def set_mon_palettes(self, random, data): | 
					
						
							|  |  |  |     if self.multiworld.randomize_pokemon_palettes[self.player] == "vanilla": | 
					
						
							|  |  |  |         return | 
					
						
							|  |  |  |     pallet_map = { | 
					
						
							|  |  |  |         "Poison": 0x0F, | 
					
						
							|  |  |  |         "Normal": 0x10, | 
					
						
							|  |  |  |         "Ice": 0x11, | 
					
						
							|  |  |  |         "Fire": 0x12, | 
					
						
							|  |  |  |         "Water": 0x13, | 
					
						
							|  |  |  |         "Ghost": 0x14, | 
					
						
							|  |  |  |         "Ground": 0x15, | 
					
						
							|  |  |  |         "Grass": 0x16, | 
					
						
							|  |  |  |         "Psychic": 0x17, | 
					
						
							|  |  |  |         "Electric": 0x18, | 
					
						
							|  |  |  |         "Rock": 0x19, | 
					
						
							|  |  |  |         "Dragon": 0x1F, | 
					
						
							|  |  |  |         "Flying": 0x20, | 
					
						
							|  |  |  |         "Fighting": 0x21, | 
					
						
							|  |  |  |         "Bug": 0x22 | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     palettes = [] | 
					
						
							|  |  |  |     for mon in poke_data.pokemon_data: | 
					
						
							|  |  |  |         if self.multiworld.randomize_pokemon_palettes[self.player] == "primary_type": | 
					
						
							|  |  |  |             pallet = pallet_map[self.local_poke_data[mon]["type1"]] | 
					
						
							|  |  |  |         elif (self.multiworld.randomize_pokemon_palettes[self.player] == "follow_evolutions" and mon in | 
					
						
							|  |  |  |               poke_data.evolves_from and poke_data.evolves_from[mon] != "Eevee"): | 
					
						
							|  |  |  |             pallet = palettes[-1] | 
					
						
							|  |  |  |         else:  # completely_random or follow_evolutions and it is not an evolved form (except eeveelutions) | 
					
						
							|  |  |  |             pallet = random.choice(list(pallet_map.values())) | 
					
						
							|  |  |  |         palettes.append(pallet) | 
					
						
							|  |  |  |     address = rom_addresses["Mon_Palettes"] | 
					
						
							|  |  |  |     for pallet in palettes: | 
					
						
							|  |  |  |         data[address] = pallet | 
					
						
							|  |  |  |         address += 1 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  | def process_trainer_data(self, data, random): | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     mons_list = [pokemon for pokemon in poke_data.pokemon_data.keys() if pokemon not in poke_data.legendary_pokemon | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                  or self.multiworld.trainer_legendaries[self.player].value] | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     address = rom_addresses["Trainer_Data"] | 
					
						
							|  |  |  |     while address < rom_addresses["Trainer_Data_End"]: | 
					
						
							|  |  |  |         if data[address] == 255: | 
					
						
							|  |  |  |             mode = 1 | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             mode = 0 | 
					
						
							|  |  |  |         while True: | 
					
						
							|  |  |  |             address += 1 | 
					
						
							|  |  |  |             if data[address] == 0: | 
					
						
							|  |  |  |                 address += 1 | 
					
						
							|  |  |  |                 break | 
					
						
							|  |  |  |             address += mode | 
					
						
							|  |  |  |             mon = None | 
					
						
							|  |  |  |             for i in range(1, 4): | 
					
						
							|  |  |  |                 for l in ["A", "B", "C", "D", "E", "F", "G", "H"]: | 
					
						
							|  |  |  |                     if rom_addresses[f"Rival_Starter{i}_{l}"] == address: | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  |                         mon = " ".join(self.multiworld.get_location(f"Pallet Town - Starter {i}", | 
					
						
							|  |  |  |                                                                     self.player).item.name.split()[1:]) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                         if l in ["D", "E", "F", "G", "H"] and mon in poke_data.evolves_to: | 
					
						
							|  |  |  |                             mon = poke_data.evolves_to[mon] | 
					
						
							|  |  |  |                         if l in ["F", "G", "H"] and mon in poke_data.evolves_to: | 
					
						
							|  |  |  |                             mon = poke_data.evolves_to[mon] | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |             if mon is None and self.multiworld.randomize_trainer_parties[self.player].value: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                 mon = poke_data.id_to_mon[data[address]] | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  |                 mon = randomize_pokemon(self, mon, mons_list, | 
					
						
							|  |  |  |                                         self.multiworld.randomize_trainer_parties[self.player].value, random) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             if mon is not None: | 
					
						
							|  |  |  |                 data[address] = poke_data.pokemon_data[mon]["id"] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def process_static_pokemon(self): | 
					
						
							| 
									
										
										
										
											2022-12-08 04:51:01 -05:00
										 |  |  |     starter_slots = deepcopy([location for location in location_data if location.type == "Starter Pokemon"]) | 
					
						
							|  |  |  |     legendary_slots = deepcopy([location for location in location_data if location.type == "Legendary Pokemon"]) | 
					
						
							|  |  |  |     static_slots = deepcopy([location for location in location_data if location.type in | 
					
						
							|  |  |  |                     ["Static Pokemon", "Missable Pokemon"]]) | 
					
						
							|  |  |  |     legendary_mons = deepcopy([slot.original_item for slot in legendary_slots]) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  |     tower_6F_mons = set() | 
					
						
							|  |  |  |     for i in range(1, 11): | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         tower_6F_mons.add(self.multiworld.get_location(f"Pokemon Tower 6F - Wild Pokemon - {i}", self.player).item.name) | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     mons_list = [pokemon for pokemon in poke_data.first_stage_pokemon if pokemon not in poke_data.legendary_pokemon | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                  or self.multiworld.randomize_legendary_pokemon[self.player].value == 3] | 
					
						
							|  |  |  |     if self.multiworld.randomize_legendary_pokemon[self.player].value == 0: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         for slot in legendary_slots: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |             location = self.multiworld.get_location(slot.name, self.player) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             location.place_locked_item(self.create_item("Missable " + slot.original_item)) | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     elif self.multiworld.randomize_legendary_pokemon[self.player].value == 1: | 
					
						
							|  |  |  |         self.multiworld.random.shuffle(legendary_mons) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         for slot in legendary_slots: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |             location = self.multiworld.get_location(slot.name, self.player) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             location.place_locked_item(self.create_item("Missable " + legendary_mons.pop())) | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     elif self.multiworld.randomize_legendary_pokemon[self.player].value == 2: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         static_slots = static_slots + legendary_slots | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         self.multiworld.random.shuffle(static_slots) | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  |         static_slots.sort(key=lambda s: 0 if s.name == "Pokemon Tower 6F - Restless Soul" else 1) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         while legendary_slots: | 
					
						
							|  |  |  |             swap_slot = legendary_slots.pop() | 
					
						
							|  |  |  |             slot = static_slots.pop() | 
					
						
							|  |  |  |             slot_type = slot.type.split()[0] | 
					
						
							|  |  |  |             if slot_type == "Legendary": | 
					
						
							|  |  |  |                 slot_type = "Missable" | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |             location = self.multiworld.get_location(slot.name, self.player) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             location.place_locked_item(self.create_item(slot_type + " " + swap_slot.original_item)) | 
					
						
							|  |  |  |             swap_slot.original_item = slot.original_item | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     elif self.multiworld.randomize_legendary_pokemon[self.player].value == 3: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         static_slots = static_slots + legendary_slots | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for slot in static_slots: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         location = self.multiworld.get_location(slot.name, self.player) | 
					
						
							|  |  |  |         randomize_type = self.multiworld.randomize_static_pokemon[self.player].value | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         slot_type = slot.type.split()[0] | 
					
						
							|  |  |  |         if slot_type == "Legendary": | 
					
						
							|  |  |  |             slot_type = "Missable" | 
					
						
							|  |  |  |         if not randomize_type: | 
					
						
							|  |  |  |             location.place_locked_item(self.create_item(slot_type + " " + slot.original_item)) | 
					
						
							|  |  |  |         else: | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  |             mon = self.create_item(slot_type + " " + | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  |                                        randomize_pokemon(self, slot.original_item, mons_list, randomize_type, | 
					
						
							|  |  |  |                                                          self.multiworld.random)) | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  |             while location.name == "Pokemon Tower 6F - Restless Soul" and mon in tower_6F_mons: | 
					
						
							|  |  |  |                 mon = self.create_item(slot_type + " " + randomize_pokemon(self, slot.original_item, mons_list, | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  |                                                                            randomize_type, self.multiworld.random)) | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  |             location.place_locked_item(mon) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     chosen_mons = set() | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     for slot in starter_slots: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         location = self.multiworld.get_location(slot.name, self.player) | 
					
						
							|  |  |  |         randomize_type = self.multiworld.randomize_starter_pokemon[self.player].value | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         slot_type = "Missable" | 
					
						
							|  |  |  |         if not randomize_type: | 
					
						
							|  |  |  |             location.place_locked_item(self.create_item(slot_type + " " + slot.original_item)) | 
					
						
							|  |  |  |         else: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |             mon = self.create_item(slot_type + " " + randomize_pokemon(self, slot.original_item, mons_list, | 
					
						
							|  |  |  |                                                                        randomize_type, self.multiworld.random)) | 
					
						
							|  |  |  |             while mon.name in chosen_mons: | 
					
						
							|  |  |  |                 mon = self.create_item(slot_type + " " + randomize_pokemon(self, slot.original_item, mons_list, | 
					
						
							|  |  |  |                                                                            randomize_type, self.multiworld.random)) | 
					
						
							|  |  |  |             chosen_mons.add(mon.name) | 
					
						
							|  |  |  |             location.place_locked_item(mon) | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  | def process_wild_pokemon(self): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     encounter_slots = get_encounter_slots(self) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     placed_mons = {pokemon: 0 for pokemon in poke_data.pokemon_data.keys()} | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     zone_mapping = {} | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.randomize_wild_pokemon[self.player].value: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         mons_list = [pokemon for pokemon in poke_data.pokemon_data.keys() if pokemon not in poke_data.legendary_pokemon | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                      or self.multiworld.randomize_legendary_pokemon[self.player].value == 3] | 
					
						
							|  |  |  |         self.multiworld.random.shuffle(encounter_slots) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         locations = [] | 
					
						
							|  |  |  |         for slot in encounter_slots: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |             location = self.multiworld.get_location(slot.name, self.player) | 
					
						
							|  |  |  |             zone = " - ".join(location.name.split(" - ")[:-1]) | 
					
						
							|  |  |  |             if zone not in zone_mapping: | 
					
						
							|  |  |  |                 zone_mapping[zone] = {} | 
					
						
							|  |  |  |             original_mon = slot.original_item | 
					
						
							|  |  |  |             if self.multiworld.area_1_to_1_mapping[self.player] and original_mon in zone_mapping[zone]: | 
					
						
							|  |  |  |                 mon = zone_mapping[zone][original_mon] | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 mon = randomize_pokemon(self, original_mon, mons_list, | 
					
						
							|  |  |  |                                         self.multiworld.randomize_wild_pokemon[self.player].value, self.multiworld.random) | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  |             # if static Pokemon are not randomized, we make sure nothing on Pokemon Tower 6F is a Marowak | 
					
						
							|  |  |  |             # if static Pokemon are randomized we deal with that during static encounter randomization | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |             while (self.multiworld.randomize_static_pokemon[self.player].value == 0 and mon == "Marowak" | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  |                    and "Pokemon Tower 6F" in slot.name): | 
					
						
							|  |  |  |                 # to account for the possibility that only one ground type Pokemon exists, match only stats for this fix | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |                 mon = randomize_pokemon(self, original_mon, mons_list, 2, self.multiworld.random) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             placed_mons[mon] += 1 | 
					
						
							|  |  |  |             location.item = self.create_item(mon) | 
					
						
							|  |  |  |             location.event = True | 
					
						
							|  |  |  |             location.locked = True | 
					
						
							|  |  |  |             location.item.location = location | 
					
						
							|  |  |  |             locations.append(location) | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |             zone_mapping[zone][original_mon] = mon | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |         mons_to_add = [] | 
					
						
							|  |  |  |         remaining_pokemon = [pokemon for pokemon in poke_data.pokemon_data.keys() if placed_mons[pokemon] == 0 and | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                              (pokemon not in poke_data.legendary_pokemon or self.multiworld.randomize_legendary_pokemon[self.player].value == 3)] | 
					
						
							|  |  |  |         if self.multiworld.catch_em_all[self.player].value == 1: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             mons_to_add = [pokemon for pokemon in poke_data.first_stage_pokemon if placed_mons[pokemon] == 0 and | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                             (pokemon not in poke_data.legendary_pokemon or self.multiworld.randomize_legendary_pokemon[self.player].value == 3)] | 
					
						
							|  |  |  |         elif self.multiworld.catch_em_all[self.player].value == 2: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             mons_to_add = remaining_pokemon.copy() | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         logic_needed_mons = max(self.multiworld.oaks_aide_rt_2[self.player].value, | 
					
						
							|  |  |  |                                 self.multiworld.oaks_aide_rt_11[self.player].value, | 
					
						
							|  |  |  |                                 self.multiworld.oaks_aide_rt_15[self.player].value) | 
					
						
							|  |  |  |         if self.multiworld.accessibility[self.player] == "minimal": | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             logic_needed_mons = 0 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         self.multiworld.random.shuffle(remaining_pokemon) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         while (len([pokemon for pokemon in placed_mons if placed_mons[pokemon] > 0]) | 
					
						
							|  |  |  |                + len(mons_to_add) < logic_needed_mons): | 
					
						
							|  |  |  |             mons_to_add.append(remaining_pokemon.pop()) | 
					
						
							|  |  |  |         for mon in mons_to_add: | 
					
						
							|  |  |  |             stat_base = get_base_stat_total(mon) | 
					
						
							|  |  |  |             candidate_locations = get_encounter_slots(self) | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |             if self.multiworld.randomize_wild_pokemon[self.player].current_key in ["match_base_stats", "match_types_and_base_stats"]: | 
					
						
							| 
									
										
										
										
											2023-03-26 15:22:35 -04:00
										 |  |  |                 candidate_locations.sort(key=lambda slot: abs(get_base_stat_total(slot.original_item) - stat_base)) | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |             if self.multiworld.randomize_wild_pokemon[self.player].current_key in ["match_types", "match_types_and_base_stats"]: | 
					
						
							|  |  |  |                 candidate_locations.sort(key=lambda slot: not any([poke_data.pokemon_data[slot.original_item]["type1"] in | 
					
						
							|  |  |  |                      [self.local_poke_data[mon]["type1"], self.local_poke_data[mon]["type2"]], | 
					
						
							|  |  |  |                      poke_data.pokemon_data[slot.original_item]["type2"] in | 
					
						
							|  |  |  |                      [self.local_poke_data[mon]["type1"], self.local_poke_data[mon]["type2"]]])) | 
					
						
							| 
									
										
										
										
											2023-03-26 15:22:35 -04:00
										 |  |  |             candidate_locations = [self.multiworld.get_location(location.name, self.player) for location in candidate_locations] | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             for location in candidate_locations: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |                 zone = " - ".join(location.name.split(" - ")[:-1]) | 
					
						
							|  |  |  |                 if self.multiworld.catch_em_all[self.player] == "all_pokemon" and self.multiworld.area_1_to_1_mapping[self.player]: | 
					
						
							|  |  |  |                     if not [self.multiworld.get_location(l.name, self.player) for l in get_encounter_slots(self) | 
					
						
							|  |  |  |                             if (not l.name.startswith(zone)) and | 
					
						
							|  |  |  |                                self.multiworld.get_location(l.name, self.player).item.name == location.item.name]: | 
					
						
							|  |  |  |                         continue | 
					
						
							|  |  |  |                 if self.multiworld.catch_em_all[self.player] == "first_stage" and self.multiworld.area_1_to_1_mapping[self.player]: | 
					
						
							|  |  |  |                     if not [self.multiworld.get_location(l.name, self.player) for l in get_encounter_slots(self) | 
					
						
							|  |  |  |                             if (not l.name.startswith(zone)) and | 
					
						
							|  |  |  |                                self.multiworld.get_location(l.name, self.player).item.name == location.item.name and l.name | 
					
						
							|  |  |  |                             not in poke_data.evolves_from]: | 
					
						
							|  |  |  |                         continue | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 if placed_mons[location.item.name] < 2 and (location.item.name in poke_data.first_stage_pokemon | 
					
						
							|  |  |  |                                                             or self.multiworld.catch_em_all[self.player]): | 
					
						
							|  |  |  |                     continue | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 if self.multiworld.area_1_to_1_mapping[self.player]: | 
					
						
							|  |  |  |                     place_locations = [place_location for place_location in candidate_locations if | 
					
						
							|  |  |  |                         place_location.name.startswith(zone) and | 
					
						
							|  |  |  |                         place_location.item.name == location.item.name] | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     place_locations = [location] | 
					
						
							|  |  |  |                 for place_location in place_locations: | 
					
						
							|  |  |  |                     placed_mons[place_location.item.name] -= 1 | 
					
						
							|  |  |  |                     place_location.item = self.create_item(mon) | 
					
						
							|  |  |  |                     place_location.item.location = place_location | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                     placed_mons[mon] += 1 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |                 break | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 raise Exception | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         for slot in encounter_slots: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |             location = self.multiworld.get_location(slot.name, self.player) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             location.item = self.create_item(slot.original_item) | 
					
						
							|  |  |  |             location.event = True | 
					
						
							|  |  |  |             location.locked = True | 
					
						
							|  |  |  |             location.item.location = location | 
					
						
							|  |  |  |             placed_mons[location.item.name] += 1 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  | def process_move_data(self): | 
					
						
							|  |  |  |     self.local_move_data = deepcopy(poke_data.moves) | 
					
						
							|  |  |  |     if self.multiworld.move_balancing[self.player]: | 
					
						
							|  |  |  |         self.local_move_data["Sing"]["accuracy"] = 30 | 
					
						
							|  |  |  |         self.local_move_data["Sleep Powder"]["accuracy"] = 40 | 
					
						
							|  |  |  |         self.local_move_data["Spore"]["accuracy"] = 50 | 
					
						
							|  |  |  |         self.local_move_data["Sonicboom"]["effect"] = 0 | 
					
						
							|  |  |  |         self.local_move_data["Sonicboom"]["power"] = 50 | 
					
						
							|  |  |  |         self.local_move_data["Dragon Rage"]["effect"] = 0 | 
					
						
							|  |  |  |         self.local_move_data["Dragon Rage"]["power"] = 80 | 
					
						
							|  |  |  |         self.local_move_data["Horn Drill"]["effect"] = 0 | 
					
						
							|  |  |  |         self.local_move_data["Horn Drill"]["power"] = 70 | 
					
						
							|  |  |  |         self.local_move_data["Horn Drill"]["accuracy"] = 90 | 
					
						
							|  |  |  |         self.local_move_data["Guillotine"]["effect"] = 0 | 
					
						
							|  |  |  |         self.local_move_data["Guillotine"]["power"] = 70 | 
					
						
							|  |  |  |         self.local_move_data["Guillotine"]["accuracy"] = 90 | 
					
						
							|  |  |  |         self.local_move_data["Fissure"]["effect"] = 0 | 
					
						
							|  |  |  |         self.local_move_data["Fissure"]["power"] = 70 | 
					
						
							|  |  |  |         self.local_move_data["Fissure"]["accuracy"] = 90 | 
					
						
							|  |  |  |         self.local_move_data["Blizzard"]["accuracy"] = 70 | 
					
						
							|  |  |  |     if self.multiworld.randomize_tm_moves[self.player]: | 
					
						
							|  |  |  |         self.local_tms = self.multiworld.random.sample([move for move in poke_data.moves.keys() if move not in | 
					
						
							|  |  |  |                                                         ["No Move"] + poke_data.hm_moves], 50) | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         self.local_tms = poke_data.tm_moves.copy() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | def process_pokemon_data(self): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     local_poke_data = deepcopy(poke_data.pokemon_data) | 
					
						
							|  |  |  |     learnsets = deepcopy(poke_data.learnsets) | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     tms_hms = self.local_tms + poke_data.hm_moves | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     compat_hms = set() | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |     for mon, mon_data in local_poke_data.items(): | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         if self.multiworld.randomize_pokemon_stats[self.player].value == 1: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             stats = [mon_data["hp"], mon_data["atk"], mon_data["def"], mon_data["spd"], mon_data["spc"]] | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |             self.multiworld.random.shuffle(stats) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             mon_data["hp"] = stats[0] | 
					
						
							|  |  |  |             mon_data["atk"] = stats[1] | 
					
						
							|  |  |  |             mon_data["def"] = stats[2] | 
					
						
							|  |  |  |             mon_data["spd"] = stats[3] | 
					
						
							|  |  |  |             mon_data["spc"] = stats[4] | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         elif self.multiworld.randomize_pokemon_stats[self.player].value == 2: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |             first_run = True | 
					
						
							|  |  |  |             while (mon_data["hp"] > 255 or mon_data["atk"] > 255 or mon_data["def"] > 255 or mon_data["spd"] > 255 | 
					
						
							|  |  |  |                    or mon_data["spc"] > 255 or first_run): | 
					
						
							|  |  |  |                 first_run = False | 
					
						
							|  |  |  |                 total_stats = mon_data["hp"] + mon_data["atk"] + mon_data["def"] + mon_data["spd"] + mon_data["spc"] - 60 | 
					
						
							|  |  |  |                 dist = [self.multiworld.random.randint(1, 101) / 100, self.multiworld.random.randint(1, 101) / 100, | 
					
						
							|  |  |  |                         self.multiworld.random.randint(1, 101) / 100, self.multiworld.random.randint(1, 101) / 100, | 
					
						
							|  |  |  |                         self.multiworld.random.randint(1, 101) / 100] | 
					
						
							|  |  |  |                 total_dist = sum(dist) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 mon_data["hp"] = int(round(dist[0] / total_dist * total_stats) + 20) | 
					
						
							|  |  |  |                 mon_data["atk"] = int(round(dist[1] / total_dist * total_stats) + 10) | 
					
						
							|  |  |  |                 mon_data["def"] = int(round(dist[2] / total_dist * total_stats) + 10) | 
					
						
							|  |  |  |                 mon_data["spd"] = int(round(dist[3] / total_dist * total_stats) + 10) | 
					
						
							|  |  |  |                 mon_data["spc"] = int(round(dist[4] / total_dist * total_stats) + 10) | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         if self.multiworld.randomize_pokemon_types[self.player].value: | 
					
						
							|  |  |  |             if self.multiworld.randomize_pokemon_types[self.player].value == 1 and mon in poke_data.evolves_from: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                 type1 = local_poke_data[poke_data.evolves_from[mon]]["type1"] | 
					
						
							|  |  |  |                 type2 = local_poke_data[poke_data.evolves_from[mon]]["type2"] | 
					
						
							|  |  |  |                 if type1 == type2: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                     if self.multiworld.secondary_type_chance[self.player].value == -1: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                         if mon_data["type1"] != mon_data["type2"]: | 
					
						
							|  |  |  |                             while type2 == type1: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                                 type2 = self.multiworld.random.choice(list(poke_data.type_names.values())) | 
					
						
							|  |  |  |                     elif self.multiworld.random.randint(1, 100) <= self.multiworld.secondary_type_chance[self.player].value: | 
					
						
							|  |  |  |                         type2 = self.multiworld.random.choice(list(poke_data.type_names.values())) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             else: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                 type1 = self.multiworld.random.choice(list(poke_data.type_names.values())) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                 type2 = type1 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                 if ((self.multiworld.secondary_type_chance[self.player].value == -1 and mon_data["type1"] | 
					
						
							|  |  |  |                      != mon_data["type2"]) or self.multiworld.random.randint(1, 100) | 
					
						
							|  |  |  |                         <= self.multiworld.secondary_type_chance[self.player].value): | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                     while type2 == type1: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                         type2 = self.multiworld.random.choice(list(poke_data.type_names.values())) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |             mon_data["type1"] = type1 | 
					
						
							|  |  |  |             mon_data["type2"] = type2 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         if self.multiworld.randomize_pokemon_movesets[self.player].value: | 
					
						
							|  |  |  |             if self.multiworld.randomize_pokemon_movesets[self.player].value == 1: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                 if mon_data["type1"] == "Normal" and mon_data["type2"] == "Normal": | 
					
						
							|  |  |  |                     chances = [[75, "Normal"]] | 
					
						
							|  |  |  |                 elif mon_data["type1"] == "Normal" or mon_data["type2"] == "Normal": | 
					
						
							|  |  |  |                     if mon_data["type1"] == "Normal": | 
					
						
							|  |  |  |                         second_type = mon_data["type2"] | 
					
						
							|  |  |  |                     else: | 
					
						
							|  |  |  |                         second_type = mon_data["type1"] | 
					
						
							|  |  |  |                     chances = [[30, "Normal"], [85, second_type]] | 
					
						
							|  |  |  |                 elif mon_data["type1"] == mon_data["type2"]: | 
					
						
							|  |  |  |                     chances = [[60, mon_data["type1"]], [80, "Normal"]] | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     chances = [[50, mon_data["type1"]], [80, mon_data["type2"]], [85, "Normal"]] | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 chances = [] | 
					
						
							| 
									
										
										
										
											2022-10-13 13:55:21 -04:00
										 |  |  |             moves = list(poke_data.moves.keys()) | 
					
						
							|  |  |  |             for move in ["No Move"] + poke_data.hm_moves: | 
					
						
							|  |  |  |                 moves.remove(move) | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |             if self.multiworld.confine_transform_to_ditto[self.player]: | 
					
						
							|  |  |  |                 moves.remove("Transform") | 
					
						
							|  |  |  |             if self.multiworld.start_with_four_moves[self.player]: | 
					
						
							|  |  |  |                 num_moves = 4 | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 num_moves = len([i for i in [mon_data["start move 1"], mon_data["start move 2"], | 
					
						
							|  |  |  |                                              mon_data["start move 3"], mon_data["start move 4"]] if i != "No Move"]) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |             if mon in learnsets: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |                 num_moves += len(learnsets[mon]) | 
					
						
							|  |  |  |             non_power_moves = [] | 
					
						
							|  |  |  |             learnsets[mon] = [] | 
					
						
							|  |  |  |             for i in range(num_moves): | 
					
						
							|  |  |  |                 if i == 0 and mon == "Ditto" and self.multiworld.confine_transform_to_ditto[self.player]: | 
					
						
							|  |  |  |                     move = "Transform" | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     move = get_move(self.local_move_data, moves, chances, self.multiworld.random) | 
					
						
							|  |  |  |                     while move == "Transform" and self.multiworld.confine_transform_to_ditto[self.player]: | 
					
						
							|  |  |  |                         move = get_move(self.local_move_data, moves, chances, self.multiworld.random) | 
					
						
							|  |  |  |                 if self.local_move_data[move]["power"] < 5: | 
					
						
							|  |  |  |                     non_power_moves.append(move) | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     learnsets[mon].append(move) | 
					
						
							|  |  |  |             learnsets[mon].sort(key=lambda move: self.local_move_data[move]["power"]) | 
					
						
							|  |  |  |             if learnsets[mon]: | 
					
						
							|  |  |  |                 for move in non_power_moves: | 
					
						
							|  |  |  |                     learnsets[mon].insert(self.multiworld.random.randint(1, len(learnsets[mon])), move) | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 learnsets[mon] = non_power_moves | 
					
						
							|  |  |  |             for i in range(1, 5): | 
					
						
							|  |  |  |                 if mon_data[f"start move {i}"] != "No Move" or self.multiworld.start_with_four_moves[self.player]: | 
					
						
							|  |  |  |                     mon_data[f"start move {i}"] = learnsets[mon].pop(0) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         if self.multiworld.randomize_pokemon_catch_rates[self.player].value: | 
					
						
							|  |  |  |             mon_data["catch rate"] = self.multiworld.random.randint(self.multiworld.minimum_catch_rate[self.player], 255) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         else: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |             mon_data["catch rate"] = max(self.multiworld.minimum_catch_rate[self.player], mon_data["catch rate"]) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |         def roll_tm_compat(roll_move): | 
					
						
							|  |  |  |             if self.local_move_data[roll_move]["type"] in [mon_data["type1"], mon_data["type2"]]: | 
					
						
							|  |  |  |                 if roll_move in poke_data.hm_moves: | 
					
						
							|  |  |  |                     if self.multiworld.hm_same_type_compatibility[self.player].value == -1: | 
					
						
							|  |  |  |                         return mon_data["tms"][int(flag / 8)] | 
					
						
							|  |  |  |                     r = self.multiworld.random.randint(1, 100) <= self.multiworld.hm_same_type_compatibility[self.player].value | 
					
						
							|  |  |  |                     if r and mon not in poke_data.legendary_pokemon: | 
					
						
							|  |  |  |                         compat_hms.add(roll_move) | 
					
						
							|  |  |  |                     return r | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                 else: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |                     if self.multiworld.tm_same_type_compatibility[self.player].value == -1: | 
					
						
							|  |  |  |                         return mon_data["tms"][int(flag / 8)] | 
					
						
							|  |  |  |                     return self.multiworld.random.randint(1, 100) <= self.multiworld.tm_same_type_compatibility[self.player].value | 
					
						
							|  |  |  |             elif self.local_move_data[roll_move]["type"] == "Normal" and "Normal" not in [mon_data["type1"], mon_data["type2"]]: | 
					
						
							|  |  |  |                 if roll_move in poke_data.hm_moves: | 
					
						
							|  |  |  |                     if self.multiworld.hm_normal_type_compatibility[self.player].value == -1: | 
					
						
							|  |  |  |                         return mon_data["tms"][int(flag / 8)] | 
					
						
							|  |  |  |                     r = self.multiworld.random.randint(1, 100) <= self.multiworld.hm_normal_type_compatibility[self.player].value | 
					
						
							|  |  |  |                     if r and mon not in poke_data.legendary_pokemon: | 
					
						
							|  |  |  |                         compat_hms.add(roll_move) | 
					
						
							|  |  |  |                     return r | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     if self.multiworld.tm_normal_type_compatibility[self.player].value == -1: | 
					
						
							|  |  |  |                         return mon_data["tms"][int(flag / 8)] | 
					
						
							|  |  |  |                     return self.multiworld.random.randint(1, 100) <= self.multiworld.tm_normal_type_compatibility[self.player].value | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 if roll_move in poke_data.hm_moves: | 
					
						
							|  |  |  |                     if self.multiworld.hm_other_type_compatibility[self.player].value == -1: | 
					
						
							|  |  |  |                         return mon_data["tms"][int(flag / 8)] | 
					
						
							|  |  |  |                     r = self.multiworld.random.randint(1, 100) <= self.multiworld.hm_other_type_compatibility[self.player].value | 
					
						
							|  |  |  |                     if r and mon not in poke_data.legendary_pokemon: | 
					
						
							|  |  |  |                         compat_hms.add(roll_move) | 
					
						
							|  |  |  |                     return r | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |                 else: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |                     if self.multiworld.tm_other_type_compatibility[self.player].value == -1: | 
					
						
							|  |  |  |                         return mon_data["tms"][int(flag / 8)] | 
					
						
							|  |  |  |                     return self.multiworld.random.randint(1, 100) <= self.multiworld.tm_other_type_compatibility[self.player].value | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for flag, tm_move in enumerate(tms_hms): | 
					
						
							|  |  |  |             if mon in poke_data.evolves_from.keys() and self.multiworld.inherit_tm_hm_compatibility[self.player]: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 if local_poke_data[poke_data.evolves_from[mon]]["tms"][int(flag / 8)] & 1 << (flag % 8): | 
					
						
							|  |  |  |                     # always inherit learnable tms/hms | 
					
						
							|  |  |  |                     bit = 1 | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     if self.local_move_data[tm_move]["type"] in [mon_data["type1"], mon_data["type2"]] and \ | 
					
						
							|  |  |  |                             self.local_move_data[tm_move]["type"] not in [ | 
					
						
							|  |  |  |                             local_poke_data[poke_data.evolves_from[mon]]["type1"], | 
					
						
							|  |  |  |                             local_poke_data[poke_data.evolves_from[mon]]["type2"]]: | 
					
						
							|  |  |  |                         # the tm/hm is for a move whose type matches current mon, but not pre-evolved form | 
					
						
							|  |  |  |                         # so this gets full chance roll | 
					
						
							|  |  |  |                         bit = roll_tm_compat(tm_move) | 
					
						
							|  |  |  |                     # otherwise 50% reduced chance to add compatibility over pre-evolved form | 
					
						
							|  |  |  |                     elif self.multiworld.random.randint(1, 100) > 50 and roll_tm_compat(tm_move): | 
					
						
							|  |  |  |                         bit = 1 | 
					
						
							|  |  |  |                     else: | 
					
						
							|  |  |  |                         bit = 0 | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 bit = roll_tm_compat(tm_move) | 
					
						
							|  |  |  |             if bit: | 
					
						
							|  |  |  |                 mon_data["tms"][int(flag / 8)] |= 1 << (flag % 8) | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 mon_data["tms"][int(flag / 8)] &= ~(1 << (flag % 8)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     hm_verify = ["Surf", "Strength"] | 
					
						
							|  |  |  |     if self.multiworld.accessibility[self.player] != "minimal" or ((not | 
					
						
							|  |  |  |                 self.multiworld.badgesanity[self.player]) and max(self.multiworld.elite_four_condition[self.player], | 
					
						
							|  |  |  |                                                                  self.multiworld.victory_road_condition[self.player]) > 7): | 
					
						
							|  |  |  |         hm_verify += ["Cut"] | 
					
						
							|  |  |  |     if self.multiworld.accessibility[self.player] != "minimal" and (self.multiworld.trainersanity[self.player] or | 
					
						
							|  |  |  |                                                                     self.multiworld.extra_key_items[self.player]): | 
					
						
							|  |  |  |         hm_verify += ["Flash"] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for hm_move in hm_verify: | 
					
						
							|  |  |  |         if hm_move not in compat_hms: | 
					
						
							|  |  |  |             mon = self.multiworld.random.choice([mon for mon in poke_data.pokemon_data if mon not in | 
					
						
							|  |  |  |                                                  poke_data.legendary_pokemon]) | 
					
						
							|  |  |  |             flag = tms_hms.index(hm_move) | 
					
						
							|  |  |  |             local_poke_data[mon]["tms"][int(flag / 8)] |= 1 << (flag % 8) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |     self.local_poke_data = local_poke_data | 
					
						
							|  |  |  |     self.learnsets = learnsets | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  | def write_quizzes(self, data, random): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def get_quiz(q, a): | 
					
						
							|  |  |  |         if q == 0: | 
					
						
							|  |  |  |             r = random.randint(0, 3) | 
					
						
							|  |  |  |             if r == 0: | 
					
						
							|  |  |  |                 mon = self.trade_mons["Trade_Dux"] | 
					
						
							|  |  |  |                 text = "A woman in<LINE>Vermilion City<CONT>" | 
					
						
							|  |  |  |             elif r == 1: | 
					
						
							|  |  |  |                 mon = self.trade_mons["Trade_Lola"] | 
					
						
							|  |  |  |                 text = "A man in<LINE>Cerulean City<CONT>" | 
					
						
							|  |  |  |             elif r == 2: | 
					
						
							|  |  |  |                 mon = self.trade_mons["Trade_Marcel"] | 
					
						
							|  |  |  |                 text = "Someone on Route 2<LINE>" | 
					
						
							|  |  |  |             elif r == 3: | 
					
						
							|  |  |  |                 mon = self.trade_mons["Trade_Spot"] | 
					
						
							|  |  |  |                 text = "Someone on Route 5<LINE>" | 
					
						
							|  |  |  |             if not a: | 
					
						
							|  |  |  |                 answers.append(0) | 
					
						
							|  |  |  |                 old_mon = mon | 
					
						
							|  |  |  |                 while old_mon == mon: | 
					
						
							|  |  |  |                     mon = random.choice(list(poke_data.pokemon_data.keys())) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             return encode_text(f"{text}was looking for<CONT>{mon}?<DONE>") | 
					
						
							|  |  |  |         elif q == 1: | 
					
						
							|  |  |  |             for location in self.multiworld.get_filled_locations(): | 
					
						
							|  |  |  |                 if location.item.name == "Secret Key" and location.item.player == self.player: | 
					
						
							|  |  |  |                     break | 
					
						
							| 
									
										
										
										
											2023-03-29 22:28:00 -04:00
										 |  |  |             player_name = self.multiworld.player_name[location.player] | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |             if not a: | 
					
						
							|  |  |  |                 if len(self.multiworld.player_name) > 1: | 
					
						
							|  |  |  |                     old_name = player_name | 
					
						
							|  |  |  |                     while old_name == player_name: | 
					
						
							|  |  |  |                         player_name = random.choice(list(self.multiworld.player_name.values())) | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     return encode_text("You're playing<LINE>in a multiworld<CONT>with other<CONT>players?<DONE>") | 
					
						
							| 
									
										
										
										
											2023-03-29 22:28:00 -04:00
										 |  |  |             if player_name == self.multiworld.player_name[self.player]: | 
					
						
							|  |  |  |                 player_name = "yourself" | 
					
						
							|  |  |  |             player_name = encode_text(player_name, force=True, safety=True) | 
					
						
							|  |  |  |             return encode_text(f"The Secret Key was<LINE>found by<CONT>") + player_name + encode_text("<DONE>") | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |         elif q == 2: | 
					
						
							|  |  |  |             if a: | 
					
						
							|  |  |  |                 return encode_text(f"#mon is<LINE>pronounced<CONT>Po-kay-mon?<DONE>") | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 if random.randint(0, 1): | 
					
						
							|  |  |  |                     return encode_text(f"#mon is<LINE>pronounced<CONT>Po-key-mon?<DONE>") | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     return encode_text(f"#mon is<LINE>pronounced<CONT>Po-kuh-mon?<DONE>") | 
					
						
							|  |  |  |         elif q == 3: | 
					
						
							|  |  |  |             starters = [" ".join(self.multiworld.get_location( | 
					
						
							|  |  |  |                 f"Pallet Town - Starter {i}", self.player).item.name.split(" ")[1:]) for i in range(1, 4)] | 
					
						
							|  |  |  |             mon = random.choice(starters) | 
					
						
							|  |  |  |             nots = random.choice(range(8, 16, 2)) | 
					
						
							|  |  |  |             if random.randint(0, 1): | 
					
						
							|  |  |  |                 while mon in starters: | 
					
						
							|  |  |  |                     mon = random.choice(list(poke_data.pokemon_data.keys())) | 
					
						
							|  |  |  |                     if a: | 
					
						
							|  |  |  |                         nots += 1 | 
					
						
							|  |  |  |             elif not a: | 
					
						
							|  |  |  |                 nots += 1 | 
					
						
							|  |  |  |             text = f"{mon} was<LINE>" | 
					
						
							|  |  |  |             while nots > 0: | 
					
						
							|  |  |  |                 i = random.randint(1, min(4, nots)) | 
					
						
							|  |  |  |                 text += ("not " * i) + "<CONT>" | 
					
						
							|  |  |  |                 nots -= i | 
					
						
							|  |  |  |             text += "a starter choice?<DONE>" | 
					
						
							|  |  |  |             return encode_text(text) | 
					
						
							|  |  |  |         elif q == 4: | 
					
						
							|  |  |  |             if a: | 
					
						
							|  |  |  |                 tm_text = self.local_tms[27] | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 if self.multiworld.randomize_tm_moves[self.player]: | 
					
						
							|  |  |  |                     wrong_tms = self.local_tms.copy() | 
					
						
							|  |  |  |                     wrong_tms.pop(27) | 
					
						
							|  |  |  |                     tm_text = random.choice(wrong_tms) | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     tm_text = "TOMBSTONER" | 
					
						
							|  |  |  |             return encode_text(f"TM28 contains<LINE>{tm_text.upper()}?<DONE>") | 
					
						
							|  |  |  |         elif q == 5: | 
					
						
							|  |  |  |             i = 8 | 
					
						
							|  |  |  |             while not a and i in [1, 8]: | 
					
						
							|  |  |  |                 i = random.randint(0, 99999999) | 
					
						
							|  |  |  |             return encode_text(f"There are {i}<LINE>certified #MON<CONT>LEAGUE BADGEs?<DONE>") | 
					
						
							|  |  |  |         elif q == 6: | 
					
						
							|  |  |  |             i = 2 | 
					
						
							|  |  |  |             while not a and i in [1, 2]: | 
					
						
							|  |  |  |                 i = random.randint(0, 99) | 
					
						
							|  |  |  |             return encode_text(f"POLIWAG evolves {i}<LINE>times?<DONE>") | 
					
						
							|  |  |  |         elif q == 7: | 
					
						
							|  |  |  |             entity = "Motor Carrier" | 
					
						
							|  |  |  |             if not a: | 
					
						
							|  |  |  |                 entity = random.choice(["Driver", "Shipper"]) | 
					
						
							|  |  |  |             return encode_text("Title 49 of the<LINE>U.S. Code of<CONT>Federal<CONT>Regulations part<CONT>397.67 states" | 
					
						
							|  |  |  |                                f"<CONT>that the<CONT>{entity}<CONT>is responsible<CONT>for planning<CONT>routes when" | 
					
						
							|  |  |  |                                "<CONT>hazardous<CONT>materials are<CONT>transported?<DONE>") | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     answers = [random.randint(0, 1), random.randint(0, 1), random.randint(0, 1), | 
					
						
							|  |  |  |                random.randint(0, 1), random.randint(0, 1), random.randint(0, 1)] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     questions = random.sample((range(0, 8)), 6) | 
					
						
							|  |  |  |     question_texts = [] | 
					
						
							|  |  |  |     for i, question in enumerate(questions): | 
					
						
							|  |  |  |         question_texts.append(get_quiz(question, answers[i])) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for i, quiz in enumerate(["A", "B", "C", "D", "E", "F"]): | 
					
						
							|  |  |  |         data[rom_addresses[f"Quiz_Answer_{quiz}"]] = int(not answers[i]) << 4 | (i + 1) | 
					
						
							|  |  |  |         write_bytes(data, question_texts[i], rom_addresses[f"Text_Quiz_{quiz}"]) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | def generate_output(self, output_directory: str): | 
					
						
							| 
									
										
										
										
											2023-02-02 01:14:23 +01:00
										 |  |  |     random = self.multiworld.per_slot_randoms[self.player] | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     game_version = self.multiworld.game_version[self.player].current_key | 
					
						
							| 
									
										
										
										
											2022-11-06 03:07:41 -05:00
										 |  |  |     data = bytes(get_base_rom_bytes(game_version)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     with open(os.path.join(os.path.dirname(__file__), f'basepatch_{game_version}.bsdiff4'), 'rb') as stream: | 
					
						
							|  |  |  |         base_patch = bytes(stream.read()) | 
					
						
							|  |  |  |     data = bytearray(bsdiff4.patch(data, base_patch)) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |     basemd5 = hashlib.md5() | 
					
						
							|  |  |  |     basemd5.update(data) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     for location in self.multiworld.get_locations(): | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         if location.player != self.player or location.rom_address is None: | 
					
						
							|  |  |  |             continue | 
					
						
							|  |  |  |         if location.item and location.item.player == self.player: | 
					
						
							|  |  |  |             if location.rom_address: | 
					
						
							|  |  |  |                 rom_address = location.rom_address | 
					
						
							|  |  |  |                 if not isinstance(rom_address, list): | 
					
						
							|  |  |  |                     rom_address = [rom_address] | 
					
						
							|  |  |  |                 for address in rom_address: | 
					
						
							|  |  |  |                     if location.item.name in poke_data.pokemon_data.keys(): | 
					
						
							|  |  |  |                         data[address] = poke_data.pokemon_data[location.item.name]["id"] | 
					
						
							|  |  |  |                     elif " ".join(location.item.name.split()[1:]) in poke_data.pokemon_data.keys(): | 
					
						
							|  |  |  |                         data[address] = poke_data.pokemon_data[" ".join(location.item.name.split()[1:])]["id"] | 
					
						
							|  |  |  |                     else: | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |                         item_id = self.item_name_to_id[location.item.name] - 172000000 | 
					
						
							|  |  |  |                         if item_id > 255: | 
					
						
							|  |  |  |                             item_id -= 256 | 
					
						
							|  |  |  |                         data[address] = item_id | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             data[location.rom_address] = 0x2C  # AP Item | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def set_trade_mon(address, loc): | 
					
						
							|  |  |  |         mon = self.multiworld.get_location(loc, self.player).item.name | 
					
						
							|  |  |  |         data[rom_addresses[address]] = poke_data.pokemon_data[mon]["id"] | 
					
						
							|  |  |  |         self.trade_mons[address] = mon | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if game_version == "red": | 
					
						
							|  |  |  |         set_trade_mon("Trade_Terry", "Safari Zone Center - Wild Pokemon - 5") | 
					
						
							|  |  |  |         set_trade_mon("Trade_Spot", "Safari Zone East - Wild Pokemon - 1") | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         set_trade_mon("Trade_Terry", "Safari Zone Center - Wild Pokemon - 7") | 
					
						
							|  |  |  |         set_trade_mon("Trade_Spot", "Safari Zone East - Wild Pokemon - 7") | 
					
						
							|  |  |  |     set_trade_mon("Trade_Marcel", "Route 24 - Wild Pokemon - 6") | 
					
						
							|  |  |  |     set_trade_mon("Trade_Sailor", "Pokemon Mansion 1F - Wild Pokemon - 3") | 
					
						
							|  |  |  |     set_trade_mon("Trade_Dux", "Route 3 - Wild Pokemon - 2") | 
					
						
							|  |  |  |     set_trade_mon("Trade_Marc", "Route 23 - Super Rod Pokemon - 1") | 
					
						
							|  |  |  |     set_trade_mon("Trade_Lola", "Route 10 - Super Rod Pokemon - 1") | 
					
						
							|  |  |  |     set_trade_mon("Trade_Doris", "Cerulean Cave 1F - Wild Pokemon - 9") | 
					
						
							|  |  |  |     set_trade_mon("Trade_Crinkles", "Route 12 - Wild Pokemon - 4") | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     data[rom_addresses['Fly_Location']] = self.fly_map_code | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.tea[self.player].value: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         data[rom_addresses["Option_Tea"]] = 1 | 
					
						
							|  |  |  |         data[rom_addresses["Guard_Drink_List"]] = 0x54 | 
					
						
							|  |  |  |         data[rom_addresses["Guard_Drink_List"] + 1] = 0 | 
					
						
							|  |  |  |         data[rom_addresses["Guard_Drink_List"] + 2] = 0 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  |     data[rom_addresses["Fossils_Needed_For_Second_Item"]] = ( | 
					
						
							|  |  |  |         self.multiworld.second_fossil_check_condition[self.player].value) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-22 10:14:33 -05:00
										 |  |  |     data[rom_addresses["Option_Lose_Money"]] = int(not self.multiworld.lose_money_on_blackout[self.player].value) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.extra_key_items[self.player].value: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         data[rom_addresses['Options']] |= 4 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     data[rom_addresses["Option_Blind_Trainers"]] = round(self.multiworld.blind_trainers[self.player].value * 2.55) | 
					
						
							|  |  |  |     data[rom_addresses['Option_Cerulean_Cave_Condition']] = self.multiworld.cerulean_cave_condition[self.player].value | 
					
						
							|  |  |  |     data[rom_addresses['Option_Encounter_Minimum_Steps']] = self.multiworld.minimum_steps_between_encounters[self.player].value | 
					
						
							|  |  |  |     data[rom_addresses['Option_Victory_Road_Badges']] = self.multiworld.victory_road_condition[self.player].value | 
					
						
							|  |  |  |     data[rom_addresses['Option_Pokemon_League_Badges']] = self.multiworld.elite_four_condition[self.player].value | 
					
						
							|  |  |  |     data[rom_addresses['Option_Viridian_Gym_Badges']] = self.multiworld.viridian_gym_condition[self.player].value | 
					
						
							|  |  |  |     data[rom_addresses['Option_EXP_Modifier']] = self.multiworld.exp_modifier[self.player].value | 
					
						
							|  |  |  |     if not self.multiworld.require_item_finder[self.player].value: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         data[rom_addresses['Option_Itemfinder']] = 0 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.extra_strength_boulders[self.player].value: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         for i in range(0, 3): | 
					
						
							|  |  |  |             data[rom_addresses['Option_Boulders'] + (i * 3)] = 0x15 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.extra_key_items[self.player].value: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         for i in range(0, 4): | 
					
						
							|  |  |  |             data[rom_addresses['Option_Rock_Tunnel_Extra_Items'] + (i * 3)] = 0x15 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.old_man[self.player].value == 2: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         data[rom_addresses['Option_Old_Man']] = 0x11 | 
					
						
							|  |  |  |         data[rom_addresses['Option_Old_Man_Lying']] = 0x15 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     if self.multiworld.require_pokedex[self.player]: | 
					
						
							|  |  |  |         data[rom_addresses["Require_Pokedex_A"]] = 1 | 
					
						
							|  |  |  |         data[rom_addresses["Require_Pokedex_B"]] = 1 | 
					
						
							| 
									
										
										
										
											2023-04-05 00:59:59 -04:00
										 |  |  |         data[rom_addresses["Require_Pokedex_C"]] = 1 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     if self.multiworld.dexsanity[self.player]: | 
					
						
							|  |  |  |         data[rom_addresses["Option_Dexsanity_A"]] = 1 | 
					
						
							|  |  |  |         data[rom_addresses["Option_Dexsanity_B"]] = 1 | 
					
						
							|  |  |  |     if self.multiworld.all_pokemon_seen[self.player]: | 
					
						
							|  |  |  |         data[rom_addresses["Option_Pokedex_Seen"]] = 1 | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  |     money = str(self.multiworld.starting_money[self.player].value).zfill(6) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     data[rom_addresses["Starting_Money_High"]] = int(money[:2], 16) | 
					
						
							|  |  |  |     data[rom_addresses["Starting_Money_Middle"]] = int(money[2:4], 16) | 
					
						
							|  |  |  |     data[rom_addresses["Starting_Money_Low"]] = int(money[4:], 16) | 
					
						
							| 
									
										
										
										
											2022-11-01 02:02:15 -04:00
										 |  |  |     data[rom_addresses["Text_Badges_Needed_Viridian_Gym"]] = encode_text( | 
					
						
							|  |  |  |         str(self.multiworld.viridian_gym_condition[self.player].value))[0] | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     data[rom_addresses["Text_Badges_Needed"]] = encode_text( | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         str(max(self.multiworld.victory_road_condition[self.player].value, | 
					
						
							|  |  |  |                 self.multiworld.elite_four_condition[self.player].value)))[0] | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  |     write_bytes(data, encode_text( | 
					
						
							|  |  |  |         " ".join(self.multiworld.get_location("Route 3 - Pokemon For Sale", self.player).item.name.upper().split()[1:])), | 
					
						
							|  |  |  |                 rom_addresses["Text_Magikarp_Salesman"]) | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     write_quizzes(self, data, random) | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.badges_needed_for_hm_moves[self.player].value == 0: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         for hm_move in poke_data.hm_moves: | 
					
						
							|  |  |  |             write_bytes(data, bytearray([0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00]), | 
					
						
							|  |  |  |                         rom_addresses["HM_" + hm_move + "_Badge_a"]) | 
					
						
							|  |  |  |     elif self.extra_badges: | 
					
						
							|  |  |  |         written_badges = {} | 
					
						
							|  |  |  |         for hm_move, badge in self.extra_badges.items(): | 
					
						
							|  |  |  |             data[rom_addresses["HM_" + hm_move + "_Badge_b"]] = {"Boulder Badge": 0x47, "Cascade Badge": 0x4F, | 
					
						
							|  |  |  |                                                                  "Thunder Badge": 0x57, "Rainbow Badge": 0x5F, | 
					
						
							|  |  |  |                                                                  "Soul Badge": 0x67, "Marsh Badge": 0x6F, | 
					
						
							|  |  |  |                                                                  "Volcano Badge": 0x77, "Earth Badge": 0x7F}[badge] | 
					
						
							|  |  |  |             move_text = hm_move | 
					
						
							|  |  |  |             if badge not in ["Marsh Badge", "Volcano Badge", "Earth Badge"]: | 
					
						
							|  |  |  |                 move_text = ", " + move_text | 
					
						
							|  |  |  |             rom_address = rom_addresses["Badge_Text_" + badge.replace(" ", "_")] | 
					
						
							|  |  |  |             if badge in written_badges: | 
					
						
							|  |  |  |                 rom_address += len(written_badges[badge]) | 
					
						
							|  |  |  |                 move_text = ", " + move_text | 
					
						
							|  |  |  |             write_bytes(data, encode_text(move_text.upper()), rom_address) | 
					
						
							|  |  |  |             written_badges[badge] = move_text | 
					
						
							|  |  |  |         for badge in ["Marsh Badge", "Volcano Badge", "Earth Badge"]: | 
					
						
							|  |  |  |             if badge not in written_badges: | 
					
						
							|  |  |  |                 write_bytes(data, encode_text("Nothing"), rom_addresses["Badge_Text_" + badge.replace(" ", "_")]) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  |     type_loc = rom_addresses["Type_Chart"] | 
					
						
							| 
									
										
										
										
											2022-12-11 14:51:28 -05:00
										 |  |  |     for matchup in self.type_chart: | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  |         if matchup[2] != 10:  # don't needlessly divide damage by 10 and multiply by 10 | 
					
						
							|  |  |  |             data[type_loc] = poke_data.type_ids[matchup[0]] | 
					
						
							|  |  |  |             data[type_loc + 1] = poke_data.type_ids[matchup[1]] | 
					
						
							|  |  |  |             data[type_loc + 2] = matchup[2] | 
					
						
							|  |  |  |             type_loc += 3 | 
					
						
							|  |  |  |     data[type_loc] = 0xFF | 
					
						
							|  |  |  |     data[type_loc + 1] = 0xFF | 
					
						
							|  |  |  |     data[type_loc + 2] = 0xFF | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.normalize_encounter_chances[self.player].value: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         chances = [25, 51, 77, 103, 129, 155, 180, 205, 230, 255] | 
					
						
							|  |  |  |         for i, chance in enumerate(chances): | 
					
						
							|  |  |  |             data[rom_addresses['Encounter_Chances'] + (i * 2)] = chance | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for mon, mon_data in self.local_poke_data.items(): | 
					
						
							|  |  |  |         if mon == "Mew": | 
					
						
							|  |  |  |             address = rom_addresses["Base_Stats_Mew"] | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             address = rom_addresses["Base_Stats"] + (28 * (mon_data["dex"] - 1)) | 
					
						
							|  |  |  |         data[address + 1] = self.local_poke_data[mon]["hp"] | 
					
						
							|  |  |  |         data[address + 2] = self.local_poke_data[mon]["atk"] | 
					
						
							|  |  |  |         data[address + 3] = self.local_poke_data[mon]["def"] | 
					
						
							|  |  |  |         data[address + 4] = self.local_poke_data[mon]["spd"] | 
					
						
							|  |  |  |         data[address + 5] = self.local_poke_data[mon]["spc"] | 
					
						
							|  |  |  |         data[address + 6] = poke_data.type_ids[self.local_poke_data[mon]["type1"]] | 
					
						
							|  |  |  |         data[address + 7] = poke_data.type_ids[self.local_poke_data[mon]["type2"]] | 
					
						
							|  |  |  |         data[address + 8] = self.local_poke_data[mon]["catch rate"] | 
					
						
							|  |  |  |         data[address + 15] = poke_data.moves[self.local_poke_data[mon]["start move 1"]]["id"] | 
					
						
							|  |  |  |         data[address + 16] = poke_data.moves[self.local_poke_data[mon]["start move 2"]]["id"] | 
					
						
							|  |  |  |         data[address + 17] = poke_data.moves[self.local_poke_data[mon]["start move 3"]]["id"] | 
					
						
							|  |  |  |         data[address + 18] = poke_data.moves[self.local_poke_data[mon]["start move 4"]]["id"] | 
					
						
							|  |  |  |         write_bytes(data, self.local_poke_data[mon]["tms"], address + 20) | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |         if mon in self.learnsets and self.learnsets[mon]: | 
					
						
							|  |  |  |                 address = rom_addresses["Learnset_" + mon.replace(" ", "")] | 
					
						
							|  |  |  |                 for i, move in enumerate(self.learnsets[mon]): | 
					
						
							|  |  |  |                     data[(address + 1) + i * 2] = poke_data.moves[move]["id"] | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  |     data[rom_addresses["Option_Aide_Rt2"]] = self.multiworld.oaks_aide_rt_2[self.player].value | 
					
						
							|  |  |  |     data[rom_addresses["Option_Aide_Rt11"]] = self.multiworld.oaks_aide_rt_11[self.player].value | 
					
						
							|  |  |  |     data[rom_addresses["Option_Aide_Rt15"]] = self.multiworld.oaks_aide_rt_15[self.player].value | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.safari_zone_normal_battles[self.player].value == 1: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         data[rom_addresses["Option_Safari_Zone_Battle_Type"]] = 255 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.reusable_tms[self.player].value: | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         data[rom_addresses["Option_Reusable_TMs"]] = 0xC9 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     for i in range(1, 10): | 
					
						
							|  |  |  |         data[rom_addresses[f"Option_Trainersanity{i}"]] = self.multiworld.trainersanity[self.player].value | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  | 
 | 
					
						
							|  |  |  |     data[rom_addresses["Option_Always_Half_STAB"]] = int(not self.multiworld.same_type_attack_bonus[self.player].value) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if self.multiworld.better_shops[self.player].value: | 
					
						
							|  |  |  |         inventory = ["Poke Ball", "Great Ball", "Ultra Ball"] | 
					
						
							|  |  |  |         if self.multiworld.better_shops[self.player].value == 2: | 
					
						
							|  |  |  |             inventory.append("Master Ball") | 
					
						
							| 
									
										
										
										
											2023-01-02 13:21:08 -05:00
										 |  |  |         inventory += ["Potion", "Super Potion", "Hyper Potion", "Max Potion", "Full Restore", "Revive", "Antidote", | 
					
						
							|  |  |  |                       "Awakening", "Burn Heal", "Ice Heal", "Paralyze Heal", "Full Heal", "Repel", "Super Repel", | 
					
						
							|  |  |  |                       "Max Repel", "Escape Rope"] | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  |         shop_data = bytearray([0xFE, len(inventory)]) | 
					
						
							|  |  |  |         shop_data += bytearray([item_table[item].id - 172000000 for item in inventory]) | 
					
						
							|  |  |  |         shop_data.append(0xFF) | 
					
						
							|  |  |  |         for shop in range(1, 10): | 
					
						
							|  |  |  |             write_bytes(data, shop_data, rom_addresses[f"Shop{shop}"]) | 
					
						
							|  |  |  |     price = str(self.multiworld.master_ball_price[self.player].value).zfill(6) | 
					
						
							|  |  |  |     price = bytearray([int(price[:2], 16), int(price[2:4], 16), int(price[4:], 16)]) | 
					
						
							|  |  |  |     write_bytes(data, price, rom_addresses["Price_Master_Ball"])  # Money values in Red and Blue are weird | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for item in reversed(self.multiworld.precollected_items[self.player]): | 
					
						
							|  |  |  |         if data[rom_addresses["Start_Inventory"] + item.code - 172000000] < 255: | 
					
						
							|  |  |  |             data[rom_addresses["Start_Inventory"] + item.code - 172000000] += 1 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     set_mon_palettes(self, random, data) | 
					
						
							| 
									
										
										
										
											2023-01-21 12:59:00 -05:00
										 |  |  |     process_trainer_data(self, data, random) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     for move_data in self.local_move_data.values(): | 
					
						
							|  |  |  |         if move_data["id"] == 0: | 
					
						
							|  |  |  |             continue | 
					
						
							|  |  |  |         address = rom_addresses["Move_Data"] + ((move_data["id"] - 1) * 6) | 
					
						
							|  |  |  |         write_bytes(data, bytearray([move_data["id"], move_data["effect"], move_data["power"], | 
					
						
							|  |  |  |                     poke_data.type_ids[move_data["type"]], round(move_data["accuracy"] * 2.55), move_data["pp"]]), address) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     TM_IDs = bytearray([poke_data.moves[move]["id"] for move in self.local_tms]) | 
					
						
							|  |  |  |     write_bytes(data, TM_IDs, rom_addresses["TM_Moves"]) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if self.multiworld.randomize_rock_tunnel[self.player]: | 
					
						
							|  |  |  |         seed = randomize_rock_tunnel(data, random) | 
					
						
							|  |  |  |         write_bytes(data, encode_text(f"SEED: <LINE>{seed}"), rom_addresses["Text_Rock_Tunnel_Sign"]) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     mons = [mon["id"] for mon in poke_data.pokemon_data.values()] | 
					
						
							|  |  |  |     random.shuffle(mons) | 
					
						
							|  |  |  |     data[rom_addresses['Title_Mon_First']] = mons.pop() | 
					
						
							|  |  |  |     for mon in range(0, 16): | 
					
						
							|  |  |  |         data[rom_addresses['Title_Mons'] + mon] = mons.pop() | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.game_version[self.player].value: | 
					
						
							|  |  |  |         mons.sort(key=lambda mon: 0 if mon == self.multiworld.get_location("Pallet Town - Starter 1", self.player).item.name | 
					
						
							|  |  |  |                   else 1 if mon == self.multiworld.get_location("Pallet Town - Starter 2", self.player).item.name else | 
					
						
							|  |  |  |                   2 if mon == self.multiworld.get_location("Pallet Town - Starter 3", self.player).item.name else 3) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     else: | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |         mons.sort(key=lambda mon: 0 if mon == self.multiworld.get_location("Pallet Town - Starter 2", self.player).item.name | 
					
						
							|  |  |  |                   else 1 if mon == self.multiworld.get_location("Pallet Town - Starter 1", self.player).item.name else | 
					
						
							|  |  |  |                   2 if mon == self.multiworld.get_location("Pallet Town - Starter 3", self.player).item.name else 3) | 
					
						
							| 
									
										
										
										
											2022-11-01 02:02:15 -04:00
										 |  |  |     write_bytes(data, encode_text(self.multiworld.seed_name[-20:], 20, True), rom_addresses['Title_Seed']) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     slot_name = self.multiworld.player_name[self.player] | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     slot_name.replace("@", " ") | 
					
						
							|  |  |  |     slot_name.replace("<", " ") | 
					
						
							|  |  |  |     slot_name.replace(">", " ") | 
					
						
							|  |  |  |     write_bytes(data, encode_text(slot_name, 16, True, True), rom_addresses['Title_Slot_Name']) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-07 18:38:34 -05:00
										 |  |  |     if self.trainer_name == "choose_in_game": | 
					
						
							|  |  |  |         data[rom_addresses["Skip_Player_Name"]] = 0 | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         write_bytes(data, self.trainer_name, rom_addresses['Player_Name']) | 
					
						
							|  |  |  |     if self.rival_name == "choose_in_game": | 
					
						
							|  |  |  |         data[rom_addresses["Skip_Rival_Name"]] = 0 | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         write_bytes(data, self.rival_name, rom_addresses['Rival_Name']) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-03-13 18:40:55 -04:00
										 |  |  |     data[0xFF00] = 2 # client compatibility version | 
					
						
							| 
									
										
										
										
											2022-11-01 02:02:15 -04:00
										 |  |  |     write_bytes(data, self.multiworld.seed_name.encode(), 0xFFDB) | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     write_bytes(data, self.multiworld.player_name[self.player].encode(), 0xFFF0) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |     outfilepname = f'_P{self.player}' | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     outfilepname += f"_{self.multiworld.get_file_safe_player_name(self.player).replace(' ', '_')}" \ | 
					
						
							|  |  |  |         if self.multiworld.player_name[self.player] != 'Player%d' % self.player else '' | 
					
						
							|  |  |  |     rompath = os.path.join(output_directory, f'AP_{self.multiworld.seed_name}{outfilepname}.gb') | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     with open(rompath, 'wb') as outfile: | 
					
						
							|  |  |  |         outfile.write(data) | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |     if self.multiworld.game_version[self.player].current_key == "red": | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |         patch = RedDeltaPatch(os.path.splitext(rompath)[0] + RedDeltaPatch.patch_file_ending, player=self.player, | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                               player_name=self.multiworld.player_name[self.player], patched_path=rompath) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     else: | 
					
						
							|  |  |  |         patch = BlueDeltaPatch(os.path.splitext(rompath)[0] + BlueDeltaPatch.patch_file_ending, player=self.player, | 
					
						
							| 
									
										
										
										
											2022-10-31 21:41:21 -05:00
										 |  |  |                                player_name=self.multiworld.player_name[self.player], patched_path=rompath) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  | 
 | 
					
						
							|  |  |  |     patch.write() | 
					
						
							|  |  |  |     os.unlink(rompath) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def write_bytes(data, byte_array, address): | 
					
						
							|  |  |  |     for byte in byte_array: | 
					
						
							|  |  |  |         data[address] = byte | 
					
						
							|  |  |  |         address += 1 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def get_base_rom_bytes(game_version: str, hash: str="") -> bytes: | 
					
						
							|  |  |  |     file_name = get_base_rom_path(game_version) | 
					
						
							|  |  |  |     with open(file_name, "rb") as file: | 
					
						
							|  |  |  |         base_rom_bytes = bytes(file.read()) | 
					
						
							|  |  |  |     if hash: | 
					
						
							|  |  |  |         basemd5 = hashlib.md5() | 
					
						
							|  |  |  |         basemd5.update(base_rom_bytes) | 
					
						
							|  |  |  |         if hash != basemd5.hexdigest(): | 
					
						
							| 
									
										
										
										
											2022-12-11 14:51:28 -05:00
										 |  |  |             raise Exception(f"Supplied Base Rom does not match known MD5 for Pokémon {game_version.title()} UE " | 
					
						
							|  |  |  |                             "release. Get the correct game and version, then dump it") | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     return base_rom_bytes | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def get_base_rom_path(game_version: str) -> str: | 
					
						
							|  |  |  |     options = Utils.get_options() | 
					
						
							|  |  |  |     file_name = options["pokemon_rb_options"][f"{game_version}_rom_file"] | 
					
						
							|  |  |  |     if not os.path.exists(file_name): | 
					
						
							| 
									
										
										
										
											2023-03-29 20:14:45 +02:00
										 |  |  |         file_name = Utils.user_path(file_name) | 
					
						
							| 
									
										
										
										
											2022-10-13 01:45:52 -04:00
										 |  |  |     return file_name | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class BlueDeltaPatch(APDeltaPatch): | 
					
						
							|  |  |  |     patch_file_ending = ".apblue" | 
					
						
							|  |  |  |     hash = "50927e843568814f7ed45ec4f944bd8b" | 
					
						
							|  |  |  |     game_version = "blue" | 
					
						
							|  |  |  |     game = "Pokemon Red and Blue" | 
					
						
							|  |  |  |     result_file_ending = ".gb" | 
					
						
							|  |  |  |     @classmethod | 
					
						
							|  |  |  |     def get_source_data(cls) -> bytes: | 
					
						
							|  |  |  |         return get_base_rom_bytes(cls.game_version, cls.hash) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class RedDeltaPatch(APDeltaPatch): | 
					
						
							|  |  |  |     patch_file_ending = ".apred" | 
					
						
							|  |  |  |     hash = "3d45c1ee9abd5738df46d2bdda8b57dc" | 
					
						
							|  |  |  |     game_version = "red" | 
					
						
							|  |  |  |     game = "Pokemon Red and Blue" | 
					
						
							|  |  |  |     result_file_ending = ".gb" | 
					
						
							|  |  |  |     @classmethod | 
					
						
							|  |  |  |     def get_source_data(cls) -> bytes: | 
					
						
							|  |  |  |         return get_base_rom_bytes(cls.game_version, cls.hash) |