 211456242e
			
		
	
	211456242e
	
	
	
		
			
			* gift version 3 * update settings usage * that really has just been broken this entire time * remove unnecessary print * Update client.py * fix random flavor handling * fix incorrect sender/receiver --------- Co-authored-by: Exempt-Medic <60412657+Exempt-Medic@users.noreply.github.com>
		
			
				
	
	
		
			601 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			601 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| import typing
 | |
| from pkgutil import get_data
 | |
| 
 | |
| import Utils
 | |
| from typing import Optional, TYPE_CHECKING, Tuple, Dict, List
 | |
| import hashlib
 | |
| import os
 | |
| import struct
 | |
| 
 | |
| from worlds.Files import APProcedurePatch, APTokenMixin, APTokenTypes, APPatchExtension
 | |
| from .aesthetics import get_palette_bytes, kirby_target_palettes, get_kirby_palette, gooey_target_palettes, \
 | |
|     get_gooey_palette
 | |
| from .compression import hal_decompress
 | |
| import bsdiff4
 | |
| 
 | |
| if TYPE_CHECKING:
 | |
|     from . import KDL3World
 | |
| 
 | |
| KDL3UHASH = "201e7658f6194458a3869dde36bf8ec2"
 | |
| KDL3JHASH = "b2f2d004ea640c3db66df958fce122b2"
 | |
| 
 | |
| level_pointers = {
 | |
|     0x770000: 0x0084,
 | |
|     0x770001: 0x009C,
 | |
|     0x770002: 0x00B8,
 | |
|     0x770003: 0x00D8,
 | |
|     0x770004: 0x0104,
 | |
|     0x770005: 0x0124,
 | |
|     0x770006: 0x014C,
 | |
|     0x770007: 0x0170,
 | |
|     0x770008: 0x0190,
 | |
|     0x770009: 0x01B0,
 | |
|     0x77000A: 0x01E8,
 | |
|     0x77000B: 0x0218,
 | |
|     0x77000C: 0x024C,
 | |
|     0x77000D: 0x0270,
 | |
|     0x77000E: 0x02A0,
 | |
|     0x77000F: 0x02C4,
 | |
|     0x770010: 0x02EC,
 | |
|     0x770011: 0x0314,
 | |
|     0x770012: 0x03CC,
 | |
|     0x770013: 0x0404,
 | |
|     0x770014: 0x042C,
 | |
|     0x770015: 0x044C,
 | |
|     0x770016: 0x0478,
 | |
|     0x770017: 0x049C,
 | |
|     0x770018: 0x04E4,
 | |
|     0x770019: 0x0504,
 | |
|     0x77001A: 0x0530,
 | |
|     0x77001B: 0x0554,
 | |
|     0x77001C: 0x05A8,
 | |
|     0x77001D: 0x0640,
 | |
|     0x770200: 0x0148,
 | |
|     0x770201: 0x0248,
 | |
|     0x770202: 0x03C8,
 | |
|     0x770203: 0x04E0,
 | |
|     0x770204: 0x06A4,
 | |
|     0x770205: 0x06A8,
 | |
| }
 | |
| 
 | |
| bb_bosses = {
 | |
|     0x770200: 0xED85F1,
 | |
|     0x770201: 0xF01360,
 | |
|     0x770202: 0xEDA3DF,
 | |
|     0x770203: 0xEDC2B9,
 | |
|     0x770204: 0xED7C3F,
 | |
|     0x770205: 0xEC29D2,
 | |
| }
 | |
| 
 | |
| level_sprites = {
 | |
|     0x19B2C6: 1827,
 | |
|     0x1A195C: 1584,
 | |
|     0x19F6F3: 1679,
 | |
|     0x19DC8B: 1717,
 | |
|     0x197900: 1872
 | |
| }
 | |
| 
 | |
| stage_tiles = {
 | |
|     0: [
 | |
|         0, 1, 2,
 | |
|         16, 17, 18,
 | |
|         32, 33, 34,
 | |
|         48, 49, 50
 | |
|     ],
 | |
|     1: [
 | |
|         3, 4, 5,
 | |
|         19, 20, 21,
 | |
|         35, 36, 37,
 | |
|         51, 52, 53
 | |
|     ],
 | |
|     2: [
 | |
|         6, 7, 8,
 | |
|         22, 23, 24,
 | |
|         38, 39, 40,
 | |
|         54, 55, 56
 | |
|     ],
 | |
|     3: [
 | |
|         9, 10, 11,
 | |
|         25, 26, 27,
 | |
|         41, 42, 43,
 | |
|         57, 58, 59,
 | |
|     ],
 | |
|     4: [
 | |
|         12, 13, 64,
 | |
|         28, 29, 65,
 | |
|         44, 45, 66,
 | |
|         60, 61, 67
 | |
|     ],
 | |
|     5: [
 | |
|         14, 15, 68,
 | |
|         30, 31, 69,
 | |
|         46, 47, 70,
 | |
|         62, 63, 71
 | |
|     ]
 | |
| }
 | |
| 
 | |
| heart_star_address = 0x2D0000
 | |
| heart_star_size = 456
 | |
| consumable_address = 0x2F91DD
 | |
| consumable_size = 698
 | |
| 
 | |
| stage_palettes = [0x60964, 0x60B64, 0x60D64, 0x60F64, 0x61164]
 | |
| 
 | |
| music_choices = [
 | |
|     2,  # Boss 1
 | |
|     3,  # Boss 2 (Unused)
 | |
|     4,  # Boss 3 (Miniboss)
 | |
|     7,  # Dedede
 | |
|     9,  # Event 2 (used once)
 | |
|     10,  # Field 1
 | |
|     11,  # Field 2
 | |
|     12,  # Field 3
 | |
|     13,  # Field 4
 | |
|     14,  # Field 5
 | |
|     15,  # Field 6
 | |
|     16,  # Field 7
 | |
|     17,  # Field 8
 | |
|     18,  # Field 9
 | |
|     19,  # Field 10
 | |
|     20,  # Field 11
 | |
|     21,  # Field 12 (Gourmet Race)
 | |
|     23,  # Dark Matter in the Hyper Zone
 | |
|     24,  # Zero
 | |
|     25,  # Level 1
 | |
|     26,  # Level 2
 | |
|     27,  # Level 4
 | |
|     28,  # Level 3
 | |
|     29,  # Heart Star Failed
 | |
|     30,  # Level 5
 | |
|     31,  # Minigame
 | |
|     38,  # Animal Friend 1
 | |
|     39,  # Animal Friend 2
 | |
|     40,  # Animal Friend 3
 | |
| ]
 | |
| # extra room pointers we don't want to track other than for music
 | |
| room_music = {
 | |
|     3079990: 23,  # Zero
 | |
|     2983409: 2,  # BB Whispy
 | |
|     3150688: 2,  # BB Acro
 | |
|     2991071: 2,  # BB PonCon
 | |
|     2998969: 2,  # BB Ado
 | |
|     2980927: 7,  # BB Dedede
 | |
|     2894290: 23  # BB Zero
 | |
| }
 | |
| 
 | |
| enemy_remap = {
 | |
|     "Waddle Dee": 0,
 | |
|     "Bronto Burt": 2,
 | |
|     "Rocky": 3,
 | |
|     "Bobo": 5,
 | |
|     "Chilly": 6,
 | |
|     "Poppy Bros Jr.": 7,
 | |
|     "Sparky": 8,
 | |
|     "Polof": 9,
 | |
|     "Broom Hatter": 11,
 | |
|     "Cappy": 12,
 | |
|     "Bouncy": 13,
 | |
|     "Nruff": 15,
 | |
|     "Glunk": 16,
 | |
|     "Togezo": 18,
 | |
|     "Kabu": 19,
 | |
|     "Mony": 20,
 | |
|     "Blipper": 21,
 | |
|     "Squishy": 22,
 | |
|     "Gabon": 24,
 | |
|     "Oro": 25,
 | |
|     "Galbo": 26,
 | |
|     "Sir Kibble": 27,
 | |
|     "Nidoo": 28,
 | |
|     "Kany": 29,
 | |
|     "Sasuke": 30,
 | |
|     "Yaban": 32,
 | |
|     "Boten": 33,
 | |
|     "Coconut": 34,
 | |
|     "Doka": 35,
 | |
|     "Icicle": 36,
 | |
|     "Pteran": 39,
 | |
|     "Loud": 40,
 | |
|     "Como": 41,
 | |
|     "Klinko": 42,
 | |
|     "Babut": 43,
 | |
|     "Wappa": 44,
 | |
|     "Mariel": 45,
 | |
|     "Tick": 48,
 | |
|     "Apolo": 49,
 | |
|     "Popon Ball": 50,
 | |
|     "KeKe": 51,
 | |
|     "Magoo": 53,
 | |
|     "Raft Waddle Dee": 57,
 | |
|     "Madoo": 58,
 | |
|     "Corori": 60,
 | |
|     "Kapar": 67,
 | |
|     "Batamon": 68,
 | |
|     "Peran": 72,
 | |
|     "Bobin": 73,
 | |
|     "Mopoo": 74,
 | |
|     "Gansan": 75,
 | |
|     "Bukiset (Burning)": 76,
 | |
|     "Bukiset (Stone)": 77,
 | |
|     "Bukiset (Ice)": 78,
 | |
|     "Bukiset (Needle)": 79,
 | |
|     "Bukiset (Clean)": 80,
 | |
|     "Bukiset (Parasol)": 81,
 | |
|     "Bukiset (Spark)": 82,
 | |
|     "Bukiset (Cutter)": 83,
 | |
|     "Waddle Dee Drawing": 84,
 | |
|     "Bronto Burt Drawing": 85,
 | |
|     "Bouncy Drawing": 86,
 | |
|     "Kabu (Dekabu)": 87,
 | |
|     "Wapod": 88,
 | |
|     "Propeller": 89,
 | |
|     "Dogon": 90,
 | |
|     "Joe": 91
 | |
| }
 | |
| 
 | |
| miniboss_remap = {
 | |
|     "Captain Stitch": 0,
 | |
|     "Yuki": 1,
 | |
|     "Blocky": 2,
 | |
|     "Jumper Shoot": 3,
 | |
|     "Boboo": 4,
 | |
|     "Haboki": 5
 | |
| }
 | |
| 
 | |
| ability_remap = {
 | |
|     "No Ability": 0,
 | |
|     "Burning Ability": 1,
 | |
|     "Stone Ability": 2,
 | |
|     "Ice Ability": 3,
 | |
|     "Needle Ability": 4,
 | |
|     "Clean Ability": 5,
 | |
|     "Parasol Ability": 6,
 | |
|     "Spark Ability": 7,
 | |
|     "Cutter Ability": 8,
 | |
| }
 | |
| 
 | |
| 
 | |
| class RomData:
 | |
|     def __init__(self, file: bytes, name: typing.Optional[str] = None):
 | |
|         self.file = bytearray(file)
 | |
|         self.name = name
 | |
| 
 | |
|     def read_byte(self, offset: int) -> int:
 | |
|         return self.file[offset]
 | |
| 
 | |
|     def read_bytes(self, offset: int, length: int) -> bytearray:
 | |
|         return self.file[offset:offset + length]
 | |
| 
 | |
|     def write_byte(self, offset: int, value: int) -> None:
 | |
|         self.file[offset] = value
 | |
| 
 | |
|     def write_bytes(self, offset: int, values: typing.Sequence[int]) -> None:
 | |
|         self.file[offset:offset + len(values)] = values
 | |
| 
 | |
|     def get_bytes(self) -> bytes:
 | |
|         return bytes(self.file)
 | |
| 
 | |
| 
 | |
| def handle_level_sprites(stages: List[Tuple[int, ...]], sprites: List[bytearray], palettes: List[List[bytearray]]) \
 | |
|         -> Tuple[List[bytearray], List[bytearray]]:
 | |
|     palette_by_level = list()
 | |
|     for palette in palettes:
 | |
|         palette_by_level.extend(palette[10:16])
 | |
|     out_palettes = list()
 | |
|     for i in range(5):
 | |
|         for j in range(6):
 | |
|             palettes[i][10 + j] = palette_by_level[stages[i][j]]
 | |
|         out_palettes.append(bytearray([x for palette in palettes[i] for x in palette]))
 | |
|     tiles_by_level = list()
 | |
|     for spritesheet in sprites:
 | |
|         decompressed = hal_decompress(spritesheet)
 | |
|         tiles = [decompressed[i:i + 32] for i in range(0, 2304, 32)]
 | |
|         tiles_by_level.extend([[tiles[x] for x in stage_tiles[stage]] for stage in stage_tiles])
 | |
|     out_sprites = list()
 | |
|     for world in range(5):
 | |
|         levels = [stages[world][x] for x in range(6)]
 | |
|         world_tiles: typing.List[bytes] = [bytes() for _ in range(72)]
 | |
|         for i in range(6):
 | |
|             for x in range(12):
 | |
|                 world_tiles[stage_tiles[i][x]] = tiles_by_level[levels[i]][x]
 | |
|         out_sprites.append(bytearray())
 | |
|         for tile in world_tiles:
 | |
|             out_sprites[world].extend(tile)
 | |
|         # insert our fake compression
 | |
|         out_sprites[world][0:0] = [0xe3, 0xff]
 | |
|         out_sprites[world][1026:1026] = [0xe3, 0xff]
 | |
|         out_sprites[world][2052:2052] = [0xe0, 0xff]
 | |
|         out_sprites[world].append(0xff)
 | |
|     return out_sprites, out_palettes
 | |
| 
 | |
| 
 | |
| def write_heart_star_sprites(rom: RomData) -> None:
 | |
|     compressed = rom.read_bytes(heart_star_address, heart_star_size)
 | |
|     decompressed = hal_decompress(compressed)
 | |
|     patch = get_data(__name__, "data/APHeartStar.bsdiff4")
 | |
|     patched = bytearray(bsdiff4.patch(decompressed, patch))
 | |
|     rom.write_bytes(0x1AF7DF, patched)
 | |
|     patched[0:0] = [0xE3, 0xFF]
 | |
|     patched.append(0xFF)
 | |
|     rom.write_bytes(0x1CD000, patched)
 | |
|     rom.write_bytes(0x3F0EBF, [0x00, 0xD0, 0x39])
 | |
| 
 | |
| 
 | |
| def write_consumable_sprites(rom: RomData, consumables: bool, stars: bool) -> None:
 | |
|     compressed = rom.read_bytes(consumable_address, consumable_size)
 | |
|     decompressed = hal_decompress(compressed)
 | |
|     patched = bytearray(decompressed)
 | |
|     if consumables:
 | |
|         patch = get_data(__name__, "data/APConsumable.bsdiff4")
 | |
|         patched = bytearray(bsdiff4.patch(bytes(patched), patch))
 | |
|     if stars:
 | |
|         patch = get_data(__name__, "data/APStars.bsdiff4")
 | |
|         patched = bytearray(bsdiff4.patch(bytes(patched), patch))
 | |
|     patched[0:0] = [0xE3, 0xFF]
 | |
|     patched.append(0xFF)
 | |
|     rom.write_bytes(0x1CD500, patched)
 | |
|     rom.write_bytes(0x3F0DAE, [0x00, 0xD5, 0x39])
 | |
| 
 | |
| 
 | |
| class KDL3PatchExtensions(APPatchExtension):
 | |
|     game = "Kirby's Dream Land 3"
 | |
| 
 | |
|     @staticmethod
 | |
|     def apply_post_patch(_: APProcedurePatch, rom: bytes) -> bytes:
 | |
|         rom_data = RomData(rom)
 | |
|         write_heart_star_sprites(rom_data)
 | |
|         if rom_data.read_bytes(0x3D014, 1)[0] > 0:
 | |
|             stages = [struct.unpack("HHHHHHH", rom_data.read_bytes(0x3D020 + x * 14, 14)) for x in range(5)]
 | |
|             palettes = [rom_data.read_bytes(full_pal, 512) for full_pal in stage_palettes]
 | |
|             read_palettes = [[palette[i:i + 32] for i in range(0, 512, 32)] for palette in palettes]
 | |
|             sprites = [rom_data.read_bytes(offset, level_sprites[offset]) for offset in level_sprites]
 | |
|             sprites, palettes = handle_level_sprites(stages, sprites, read_palettes)
 | |
|             for addr, palette in zip(stage_palettes, palettes):
 | |
|                 rom_data.write_bytes(addr, palette)
 | |
|             for addr, level_sprite in zip([0x1CA000, 0x1CA920, 0x1CB230, 0x1CBB40, 0x1CC450], sprites):
 | |
|                 rom_data.write_bytes(addr, level_sprite)
 | |
|             rom_data.write_bytes(0x460A, [0x00, 0xA0, 0x39, 0x20, 0xA9, 0x39, 0x30, 0xB2, 0x39, 0x40, 0xBB, 0x39,
 | |
|                                           0x50, 0xC4, 0x39])
 | |
|         write_consumable_sprites(rom_data, rom_data.read_byte(0x3D018) > 0, rom_data.read_byte(0x3D01A) > 0)
 | |
|         return rom_data.get_bytes()
 | |
| 
 | |
| 
 | |
| class KDL3ProcedurePatch(APProcedurePatch, APTokenMixin):
 | |
|     hash = [KDL3UHASH, KDL3JHASH]
 | |
|     game = "Kirby's Dream Land 3"
 | |
|     patch_file_ending = ".apkdl3"
 | |
|     procedure = [
 | |
|         ("apply_bsdiff4", ["kdl3_basepatch.bsdiff4"]),
 | |
|         ("apply_tokens", ["token_patch.bin"]),
 | |
|         ("apply_post_patch", []),
 | |
|         ("calc_snes_crc", [])
 | |
|     ]
 | |
|     name: bytes  # used to pass to __init__
 | |
| 
 | |
|     @classmethod
 | |
|     def get_source_data(cls) -> bytes:
 | |
|         return get_base_rom_bytes()
 | |
| 
 | |
| 
 | |
| def patch_rom(world: "KDL3World", patch: KDL3ProcedurePatch) -> None:
 | |
|     patch.write_file("kdl3_basepatch.bsdiff4",
 | |
|                      get_data(__name__, "data/kdl3_basepatch.bsdiff4"))
 | |
| 
 | |
|     # Write open world patch
 | |
|     if world.options.open_world:
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x143C7, bytes([0xAD, 0xC1, 0x5A, 0xCD, 0xC1, 0x5A, ]))
 | |
|         # changes the stage flag function to compare $5AC1 to $5AC1,
 | |
|         # always running the "new stage" function
 | |
|         # This has further checks present for bosses already, so we just
 | |
|         # need to handle regular stages
 | |
|         # write check for boss to be unlocked
 | |
| 
 | |
|     if world.options.consumables:
 | |
|         # reroute maxim tomatoes to use the 1-UP function, then null out the function
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x3002F, bytes([0x37, 0x00]))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x30037, bytes([0xA9, 0x26, 0x00,  # LDA #$0026
 | |
|                                                               0x22, 0x27, 0xD9, 0x00,  # JSL $00D927
 | |
|                                                               0xA4, 0xD2,  # LDY $D2
 | |
|                                                               0x6B,  # RTL
 | |
|                                                               0xEA, 0xEA, 0xEA, 0xEA, 0xEA, 0xEA, 0xEA, 0xEA, 0xEA,
 | |
|                                                               0xEA,  # NOP #10
 | |
|                                                               ]))
 | |
| 
 | |
|     # stars handling is built into the rom, so no changes there
 | |
| 
 | |
|     rooms = world.rooms
 | |
|     if world.options.music_shuffle > 0:
 | |
|         if world.options.music_shuffle == 1:
 | |
|             shuffled_music = music_choices.copy()
 | |
|             world.random.shuffle(shuffled_music)
 | |
|             music_map = dict(zip(music_choices, shuffled_music))
 | |
|             # Avoid putting star twinkle in the pool
 | |
|             music_map[5] = world.random.choice(music_choices)
 | |
|             # Heart Star music doesn't work on regular stages
 | |
|             music_map[8] = world.random.choice(music_choices)
 | |
|             for room in rooms:
 | |
|                 room.music = music_map[room.music]
 | |
|             for room_ptr in room_music:
 | |
|                 patch.write_token(APTokenTypes.WRITE, room_ptr + 2, bytes([music_map[room_music[room_ptr]]]))
 | |
|             for i, old_music in zip(range(5), [25, 26, 28, 27, 30]):
 | |
|                 # level themes
 | |
|                 patch.write_token(APTokenTypes.WRITE, 0x133F2 + i, bytes([music_map[old_music]]))
 | |
|             # Zero
 | |
|             patch.write_token(APTokenTypes.WRITE, 0x9AE79, music_map[0x18].to_bytes(1, "little"))
 | |
|             # Heart Star success and fail
 | |
|             patch.write_token(APTokenTypes.WRITE, 0x4A388, music_map[0x08].to_bytes(1, "little"))
 | |
|             patch.write_token(APTokenTypes.WRITE, 0x4A38D, music_map[0x1D].to_bytes(1, "little"))
 | |
|         elif world.options.music_shuffle == 2:
 | |
|             for room in rooms:
 | |
|                 room.music = world.random.choice(music_choices)
 | |
|             for room_ptr in room_music:
 | |
|                 patch.write_token(APTokenTypes.WRITE, room_ptr + 2,
 | |
|                                   world.random.choice(music_choices).to_bytes(1, "little"))
 | |
|             for i in range(5):
 | |
|                 # level themes
 | |
|                 patch.write_token(APTokenTypes.WRITE, 0x133F2 + i,
 | |
|                                   world.random.choice(music_choices).to_bytes(1, "little"))
 | |
|             # Zero
 | |
|             patch.write_token(APTokenTypes.WRITE, 0x9AE79, world.random.choice(music_choices).to_bytes(1, "little"))
 | |
|             # Heart Star success and fail
 | |
|             patch.write_token(APTokenTypes.WRITE, 0x4A388, world.random.choice(music_choices).to_bytes(1, "little"))
 | |
|             patch.write_token(APTokenTypes.WRITE, 0x4A38D, world.random.choice(music_choices).to_bytes(1, "little"))
 | |
| 
 | |
|     for room in rooms:
 | |
|         room.patch(patch, bool(world.options.consumables.value), not bool(world.options.remote_items.value))
 | |
| 
 | |
|     if world.options.virtual_console in [1, 3]:
 | |
|         # Flash Reduction
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x9AE68, b"\x10")
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x9AE8E, bytes([0x08, 0x00, 0x22, 0x5D, 0xF7, 0x00, 0xA2, 0x08, ]))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x9AEA1, b"\x08")
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x9AEC9, b"\x01")
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x9AED2, bytes([0xA9, 0x1F]))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x9AEE1, b"\x08")
 | |
| 
 | |
|     if world.options.virtual_console in [2, 3]:
 | |
|         # Hyper Zone BB colors
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2C5E16, bytes([0xEE, 0x1B, 0x18, 0x5B, 0xD3, 0x4A, 0xF4, 0x3B, ]))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2C8217, bytes([0xFF, 0x1E, ]))
 | |
| 
 | |
|     # boss requirements
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D000,
 | |
|                       struct.pack("HHHHH", world.boss_requirements[0], world.boss_requirements[1],
 | |
|                                   world.boss_requirements[2], world.boss_requirements[3],
 | |
|                                   world.boss_requirements[4]))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D00A,
 | |
|                       struct.pack("H", world.required_heart_stars if world.options.goal_speed == 1 else 0xFFFF))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D00C, world.options.goal_speed.value.to_bytes(2, "little"))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D00E, world.options.open_world.value.to_bytes(2, "little"))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D010, ((world.options.remote_items.value << 1) +
 | |
|                                                     world.options.death_link.value).to_bytes(2, "little"))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D012, world.options.goal.value.to_bytes(2, "little"))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D014, world.options.stage_shuffle.value.to_bytes(2, "little"))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D016, world.options.ow_boss_requirement.value.to_bytes(2, "little"))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D018, world.options.consumables.value.to_bytes(2, "little"))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D01A, world.options.starsanity.value.to_bytes(2, "little"))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D01C, world.options.gifting.value.to_bytes(2, "little"))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3D01E, world.options.strict_bosses.value.to_bytes(2, "little"))
 | |
|     # don't write gifting for solo game, since there's no one to send anything to
 | |
| 
 | |
|     for level in world.player_levels:
 | |
|         for i in range(len(world.player_levels[level])):
 | |
|             patch.write_token(APTokenTypes.WRITE, 0x3F002E + ((level - 1) * 14) + (i * 2),
 | |
|                               struct.pack("H", level_pointers[world.player_levels[level][i]]))
 | |
|             patch.write_token(APTokenTypes.WRITE, 0x3D020 + (level - 1) * 14 + (i * 2),
 | |
|                               struct.pack("H", world.player_levels[level][i] & 0x00FFFF))
 | |
|             if (i == 0) or (i > 0 and i % 6 != 0):
 | |
|                 patch.write_token(APTokenTypes.WRITE, 0x3D080 + (level - 1) * 12 + (i * 2),
 | |
|                                   struct.pack("H", (world.player_levels[level][i] & 0x00FFFF) % 6))
 | |
| 
 | |
|     for i in range(6):
 | |
|         if world.boss_butch_bosses[i]:
 | |
|             patch.write_token(APTokenTypes.WRITE, 0x3F0000 + (level_pointers[0x770200 + i]),
 | |
|                               struct.pack("I", bb_bosses[0x770200 + i]))
 | |
| 
 | |
|     # copy ability shuffle
 | |
|     if world.options.copy_ability_randomization.value > 0:
 | |
|         for enemy in world.copy_abilities:
 | |
|             if enemy in miniboss_remap:
 | |
|                 patch.write_token(APTokenTypes.WRITE, 0xB417E + (miniboss_remap[enemy] << 1),
 | |
|                                   struct.pack("H", ability_remap[world.copy_abilities[enemy]]))
 | |
|             else:
 | |
|                 patch.write_token(APTokenTypes.WRITE, 0xB3CAC + (enemy_remap[enemy] << 1),
 | |
|                                   struct.pack("H", ability_remap[world.copy_abilities[enemy]]))
 | |
|         # following only needs done on non-door rando
 | |
|         # incredibly lucky this follows the same order (including 5E == star block)
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2F77EA,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Sparky"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2F7811,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Sparky"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2F9BC4,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Blocky"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2F9BEB,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Blocky"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2FAC06,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Jumper Shoot"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2FAC2D,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Jumper Shoot"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2F9E7B,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Yuki"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2F9EA2,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Yuki"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2FA951,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Sir Kibble"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2FA978,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Sir Kibble"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2FA132,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Haboki"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2FA159,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Haboki"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2FA3E8,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Boboo"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2FA40F,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Boboo"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2F90E2,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Captain Stitch"]] << 1)).to_bytes(1, "little"))
 | |
|         patch.write_token(APTokenTypes.WRITE, 0x2F9109,
 | |
|                           (0x5E + (ability_remap[world.copy_abilities["Captain Stitch"]] << 1)).to_bytes(1, "little"))
 | |
| 
 | |
|         if world.options.copy_ability_randomization == 2:
 | |
|             for enemy in enemy_remap:
 | |
|                 # we just won't include it for minibosses
 | |
|                 patch.write_token(APTokenTypes.WRITE, 0xB3E40 + (enemy_remap[enemy] << 1),
 | |
|                                   struct.pack("h", world.random.randint(-1, 2)))
 | |
| 
 | |
|     # write jumping goal
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x94F8, struct.pack("H", world.options.jumping_target))
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x944E, struct.pack("H", world.options.jumping_target))
 | |
| 
 | |
|     from Utils import __version__
 | |
|     patch_name = bytearray(
 | |
|         f'KDL3{__version__.replace(".", "")[0:3]}_{world.player}_{world.multiworld.seed:11}\0', 'utf8')[:21]
 | |
|     patch_name.extend([0] * (21 - len(patch_name)))
 | |
|     patch.name = bytes(patch_name)
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3C000, patch.name)
 | |
|     patch.write_token(APTokenTypes.WRITE, 0x3C020, world.options.game_language.value.to_bytes(1, "little"))
 | |
| 
 | |
|     patch.write_token(APTokenTypes.COPY, 0x7FC0, (21, 0x3C000))
 | |
|     patch.write_token(APTokenTypes.COPY, 0x7FD9, (1, 0x3C020))
 | |
| 
 | |
|     # handle palette
 | |
|     if world.options.kirby_flavor_preset.value != 0:
 | |
|         for addr in kirby_target_palettes:
 | |
|             target = kirby_target_palettes[addr]
 | |
|             palette = get_kirby_palette(world)
 | |
|             if palette is not None:
 | |
|                 patch.write_token(APTokenTypes.WRITE, addr, get_palette_bytes(palette, target[0], target[1], target[2]))
 | |
| 
 | |
|     if world.options.gooey_flavor_preset.value != 0:
 | |
|         for addr in gooey_target_palettes:
 | |
|             target = gooey_target_palettes[addr]
 | |
|             palette = get_gooey_palette(world)
 | |
|             if palette is not None:
 | |
|                 patch.write_token(APTokenTypes.WRITE, addr, get_palette_bytes(palette, target[0], target[1], target[2]))
 | |
| 
 | |
|     patch.write_file("token_patch.bin", patch.get_token_binary())
 | |
| 
 | |
| 
 | |
| def get_base_rom_bytes() -> bytes:
 | |
|     rom_file: str = get_base_rom_path()
 | |
|     base_rom_bytes: Optional[bytes] = getattr(get_base_rom_bytes, "base_rom_bytes", None)
 | |
|     if not base_rom_bytes:
 | |
|         base_rom_bytes = bytes(Utils.read_snes_rom(open(rom_file, "rb")))
 | |
| 
 | |
|         basemd5 = hashlib.md5()
 | |
|         basemd5.update(base_rom_bytes)
 | |
|         if basemd5.hexdigest() not in {KDL3UHASH, KDL3JHASH}:
 | |
|             raise Exception("Supplied Base Rom does not match known MD5 for US or JP release. "
 | |
|                             "Get the correct game and version, then dump it")
 | |
|         get_base_rom_bytes.base_rom_bytes = base_rom_bytes
 | |
|     return base_rom_bytes
 | |
| 
 | |
| 
 | |
| def get_base_rom_path(file_name: str = "") -> str:
 | |
|     from . import KDL3World
 | |
|     if not file_name:
 | |
|         file_name = KDL3World.settings.rom_file
 | |
|     if not os.path.exists(file_name):
 | |
|         file_name = Utils.user_path(file_name)
 | |
|     return file_name
 |