ref: ee4feb8374694d684dad8905c2be2bf0171de6f9
dir: /tables/compile_resources.py/
import sys import text_compression import util from PIL import Image import yaml import tables import compile_music import array, hashlib, struct print_int_array = util.print_int_array PATH = '' def flatten(xss): return [x for xs in xss for x in xs] def invert_dict(xs): return {s:i for i,s in xs.items()} assets = {} def add_asset_uint8(name, data): assert name not in assets assets[name] = ('uint8', bytes(array.array('B', data))) def add_asset_uint16(name, data): assert name not in assets assets[name] = ('uint16', bytes(array.array('H', data))) def add_asset_int8(name, data): assert name not in assets assets[name] = ('int8', bytes(array.array('b', data))) def add_asset_int16(name, data): assert name not in assets assets[name] = ('int16', bytes(array.array('h', data))) def print_map32_to_map16(): tab = {} for line in open(PATH + 'map32_to_map16.txt'): line = line.strip() x, xs = line.split(':', 1) tab[int(x)] = [int(t) for t in xs.split(',')] def pack(*a): res=[0] * 6 res[0] = a[0] & 0xff res[1] = a[1] & 0xff res[2] = a[2] & 0xff res[3] = a[3] & 0xff res[4] = (a[0] >> 8) << 4 | (a[1] >> 8) res[5] = (a[2] >> 8) << 4 | (a[3] >> 8) return res res = [[],[],[],[]] for i in range(0, len(tab), 4): for j in range(4): res[j].extend(pack(tab[i][j], tab[i+1][j], tab[i+2][j], tab[i+3][j])) add_asset_uint8('kMap32ToMap16_0', res[0]) add_asset_uint8('kMap32ToMap16_1', res[1]) add_asset_uint8('kMap32ToMap16_2', res[2]) add_asset_uint8('kMap32ToMap16_3', res[3]) def print_dialogue(): new_r = [] offs = [] for line in open(PATH + 'dialogue.txt'): line = line.strip('\n') a, b = line.split(': ', 1) index = int(a) offs.append(len(new_r)) r = text_compression.compress_string(b) new_r.extend(r) add_asset_uint16('kDialogueOffs', offs) add_asset_uint8('kDialogueText', new_r) ROM = util.LoadedRom(sys.argv[1] if len(sys.argv) >= 2 else None) kCompSpritePtrs = [ 0x10f000,0x10f600,0x10fc00,0x118200,0x118800,0x118e00,0x119400,0x119a00, 0x11a000,0x11a600,0x11ac00,0x11b200,0x14fffc,0x1585d4,0x158ab6,0x158fbe, 0x1593f8,0x1599a6,0x159f32,0x15a3d7,0x15a8f1,0x15aec6,0x15b418,0x15b947, 0x15bed0,0x15c449,0x15c975,0x15ce7c,0x15d394,0x15d8ac,0x15ddc0,0x15e34c, 0x15e8e8,0x15ee31,0x15f3a6,0x15f92d,0x15feba,0x1682ff,0x1688e0,0x168e41, 0x1692df,0x169883,0x169cd0,0x16a26e,0x16a275,0x16a787,0x16aa06,0x16ae9d, 0x16b3ff,0x16b87e,0x16be6b,0x16c13d,0x16c619,0x16cbbb,0x16d0f1,0x16d641, 0x16d95a,0x16dd99,0x16e278,0x16e760,0x16ed25,0x16f20f,0x16f6b7,0x16fa5f, 0x16fd29,0x1781cd,0x17868d,0x178b62,0x178fd5,0x179527,0x17994b,0x179ea7, 0x17a30e,0x17a805,0x17acf8,0x17b2a2,0x17b7f9,0x17bc93,0x17c237,0x17c78e, 0x17cd55,0x17d2bc,0x17d82f,0x17dcec,0x17e1cc,0x17e36b,0x17e842,0x17eb38, 0x17ed58,0x17f06c,0x17f4fd,0x17fa39,0x17ff86,0x18845c,0x1889a1,0x188d64, 0x18919d,0x189610,0x189857,0x189b24,0x189dd2,0x18a03f,0x18a4ed,0x18a7ba, 0x18aedf,0x18af0d,0x18b520,0x18b953, ] kCompBgPtrs = [ 0x11b800,0x11bce2,0x11c15f,0x11c675,0x11cb84,0x11cf4c,0x11d2ce,0x11d726, 0x11d9cf,0x11dec4,0x11e393,0x11e893,0x11ed7d,0x11f283,0x11f746,0x11fc21, 0x11fff2,0x128498,0x128a0e,0x128f30,0x129326,0x129804,0x129d5b,0x12a272, 0x12a6fe,0x12aa77,0x12ad83,0x12b167,0x12b51d,0x12b840,0x12bd54,0x12c1c9, 0x12c73d,0x12cc86,0x12d198,0x12d6b1,0x12db6a,0x12e0ea,0x12e6bd,0x12eb51, 0x12f135,0x12f6c5,0x12fc71,0x138129,0x138693,0x138bad,0x139117,0x139609, 0x139b21,0x13a074,0x13a619,0x13ab2b,0x13b00c,0x13b4f5,0x13b9eb,0x13bebf, 0x13c3ce,0x13c817,0x13cb68,0x13cfb5,0x13d460,0x13d8c2,0x13dd7a,0x13e266, 0x13e7af,0x13ece5,0x13f245,0x13f6f0,0x13fc30,0x1480e9,0x14863b,0x148a7c, 0x148f2a,0x149346,0x1497ed,0x149cc2,0x14a173,0x14a61d,0x14ab5d,0x14b083, 0x14b4bd,0x14b94e,0x14be0e,0x14c291,0x14c7ba,0x14cce4,0x14d1db,0x14d6bd, 0x14db77,0x14ded1,0x14e2ac,0x14e754,0x14ebae,0x14ef4e,0x14f309,0x14f6f4, 0x14fa55,0x14ff8c,0x14ff93,0x14ff9a,0x14ffa1,0x14ffa8,0x14ffaf,0x14ffb6, 0x14ffbd,0x14ffc4,0x14ffcb,0x14ffd2,0x14ffd9,0x14ffe0,0x14ffe7,0x14ffee, 0x14fff5,0x18b520,0x18b953, ] def compress_store(r): rr = [] j, jend = 0, len(r) while j < jend: n = min(jend - j, 1024) rr.append(0xe0 | (n - 1) >> 8) rr.append((n - 1) & 0xff) rr.extend(r[j:j+n]) j += n rr.append(0xff) return rr def pack_u32_arrays(arr): all_offs, offs = [], len(arr) * 4 for i in range(len(arr)): all_offs.append(offs) offs += len(arr[i]) return b''.join([struct.pack('I', i) for i in all_offs] + arr) def print_images(): lengths = b'' all = [] for i in range(12): all.append(bytes(ROM.get_bytes(kCompSpritePtrs[i], 0x600))) for i in range(12, 108): decomp, comp_len = util.decomp(kCompSpritePtrs[i], ROM.get_byte, False, True) all.append(bytes(ROM.get_bytes(kCompSpritePtrs[i], comp_len))) add_asset_uint8('kSprGfx', pack_u32_arrays(all)) all = [] for i in range(len(kCompBgPtrs)): decomp, comp_len = util.decomp(kCompBgPtrs[i], ROM.get_byte, False, True) all.append(bytes(ROM.get_bytes(kCompBgPtrs[i], comp_len))) add_asset_uint8('kBgGfx', pack_u32_arrays(all)) def print_misc(): add_asset_uint8('kOverworldMapGfx', ROM.get_bytes(0x18c000, 0x4000)) add_asset_uint8('kLightOverworldTilemap', ROM.get_bytes(0xac727, 4096)) add_asset_uint8('kDarkOverworldTilemap', ROM.get_bytes(0xaD727, 1024)) add_asset_uint16('kPredefinedTileData', ROM.get_words(0x9B52, 6438)) add_asset_uint16('kFontData', ROM.get_words(0xe8000, 2048)) add_asset_uint16('kMap16ToMap8', ROM.get_words(0x8f8000, 3752 * 4)) add_asset_uint8('kGeneratedWishPondItem', ROM.get_bytes(0x888450, 256)) add_asset_uint8('kGeneratedBombosArr', ROM.get_bytes(0x8890FC, 256)) add_asset_uint8('kGeneratedEndSequence15', ROM.get_bytes(0x8ead25, 256)) add_asset_uint8('kEnding_Credits_Text', ROM.get_bytes(0x8EB178, 1989)) add_asset_uint16('kEnding_Credits_Offs', ROM.get_words(0x8EB93d, 394)) add_asset_uint16('kEnding_MapData', ROM.get_words(0x8EB038, 160)) add_asset_uint16('kEnding0_Offs', ROM.get_words(0x8EC2E1, 17)) add_asset_uint8('kEnding0_Data', ROM.get_bytes(0x8EBF4C, 917)) add_asset_uint16('kPalette_DungBgMain', ROM.get_words(0x9BD734, 1800)) add_asset_uint16('kPalette_MainSpr', ROM.get_words(0x9BD218, 120)) add_asset_uint16('kPalette_ArmorAndGloves', ROM.get_words(0x9BD308, 75)) add_asset_uint16('kPalette_Sword', ROM.get_words(0x9BD630, 12)) add_asset_uint16('kPalette_Shield', ROM.get_words(0x9BD648, 12)) add_asset_uint16('kPalette_SpriteAux3', ROM.get_words(0x9BD39E, 84)) add_asset_uint16('kPalette_MiscSprite_Indoors', ROM.get_words(0x9BD446, 77)) add_asset_uint16('kPalette_SpriteAux1', ROM.get_words(0x9BD4E0, 168)) add_asset_uint16('kPalette_OverworldBgMain', ROM.get_words(0x9BE6C8, 210)) add_asset_uint16('kPalette_OverworldBgAux12', ROM.get_words(0x9BE86C, 420)) add_asset_uint16('kPalette_OverworldBgAux3', ROM.get_words(0x9BE604, 98)) add_asset_uint16('kPalette_PalaceMapBg', ROM.get_words(0x9BE544, 96)) add_asset_uint16('kPalette_PalaceMapSpr', ROM.get_words(0x9BD70A, 21)) add_asset_uint16('kHudPalData', ROM.get_words(0x9BD660, 64)) add_asset_uint16('kOverworldMapPaletteData', ROM.get_words(0x8ADB27, 256)) g_overworld_yaml_cache = {} def load_overworld_yaml(room): if room not in g_overworld_yaml_cache: g_overworld_yaml_cache[room] = yaml.safe_load(open(PATH+'overworld/overworld-%d.yaml' % room, 'r')) return g_overworld_yaml_cache[room] def print_overworld(): r = [] for i in range(160): addr = ROM.get_24(0x82F94D + i * 3) decomp, comp_len = util.decomp(addr, ROM.get_byte, True, True) r.append(bytes(ROM.get_bytes(addr, comp_len))) add_asset_uint8('kOverworld_Hibytes_Comp', pack_u32_arrays(r)) r = [] for i in range(160): addr = ROM.get_24(0x82FB2D + i * 3) decomp, comp_len = util.decomp(addr, ROM.get_byte, True, True) r.append(bytes(ROM.get_bytes(addr, comp_len))) add_asset_uint8('kOverworld_Lobytes_Comp', pack_u32_arrays(r)) def is_area_head(i): return i >= 128 or ROM.get_byte(0x82A5EC + (i & 63)) == (i & 63) class OutArrays: def __init__(self): self.arrs = [] def add(self, type, name, size, initializer = None, items_per_line = 16): t = [initializer] * size self.arrs.append((type, name, t, items_per_line)) setattr(self, name, t) def write(self): for type, name, arr, items_per_line in self.arrs: for i, j in enumerate(arr): assert isinstance(j, int), (name, i, j, arr) if type == 'uint8': add_asset_uint8(name, arr) elif type == 'uint16': add_asset_uint16(name, arr) elif type == 'int8': add_asset_int8(name, arr) elif type == 'int16': add_asset_int16(name, arr) else: assert 0, type def print_overworld_tables(): A = OutArrays() A.add('uint8', 'kOverworldMapIsSmall', 192, initializer = 0, items_per_line = 8) A.add('uint8', 'kOverworldAuxTileThemeIndexes', 128, items_per_line = 8) A.add('uint8', 'kOverworldBgPalettes', 136, items_per_line = 8) A.add('uint16', 'kOverworld_SignText', 128, items_per_line = 8) A.add('uint8', 'kOwMusicSets', 256, items_per_line = 8) A.add('uint8', 'kOwMusicSets2', 96, items_per_line = 8) def get_music_byte(h, tag): return tables.kMusicNamesRev[h['music'][tag]] | tables.kAmbientSoundNameRev[h['ambient'][tag]] << 4 def get_loadoffs(c, d): x, y = c[0] >> 4, c[1] >> 4 x += d[0] y += d[1] return (y&0x3f) << 7 | (x&0x3f) << 1 def awrite(arr, area, key, value): arr[key] = value if area < 128 and not A.kOverworldMapIsSmall[area]: arr[key + 1], arr[key + 8], arr[key + 9] = value, value, value loaded_areas = [(i, load_overworld_yaml(i)) for i in range(160) if is_area_head(i)] for i,y in loaded_areas: h = y['Header'] A.kOverworldMapIsSmall[i] = {'small':1, 'big':0}[h['size']] if i < len(A.kOverworldAuxTileThemeIndexes): awrite(A.kOverworldAuxTileThemeIndexes, i, i, h['gfx']) if i < len(A.kOverworldBgPalettes): awrite(A.kOverworldBgPalettes, i, i, h['palette']) if i < len(A.kOverworld_SignText): awrite(A.kOverworld_SignText, i, i, h['sign_text']) if i < 64: awrite(A.kOwMusicSets, i, i, get_music_byte(h, 'beginning')) awrite(A.kOwMusicSets, i, i + 64, get_music_byte(h, 'zelda')) awrite(A.kOwMusicSets, i, i + 128, get_music_byte(h, 'sword')) awrite(A.kOwMusicSets, i, i + 192, get_music_byte(h, 'agahnim')) elif i < 64 + 96: awrite(A.kOwMusicSets2, i, i - 64, get_music_byte(h, 'agahnim')) for a in ['kBirdTravel_ScreenIndex', 'kBirdTravel_Map16LoadSrcOff', 'kBirdTravel_ScrollX', 'kBirdTravel_ScrollY', 'kBirdTravel_LinkXCoord', 'kBirdTravel_LinkYCoord', 'kBirdTravel_CameraXScroll', 'kBirdTravel_CameraYScroll']: A.add('uint16', a, 17) A.add('int8', 'kBirdTravel_Unk1', 17) A.add('int8', 'kBirdTravel_Unk3', 17) A.add('uint16', 'kWhirlpoolAreas', 8) next_whirlpool_id = 0 for i, y in loaded_areas: for t in y['Travel']: if 'bird_travel_id' in t: j = t['bird_travel_id'] else: A.kWhirlpoolAreas[next_whirlpool_id] = t['whirlpool_src_area'] j = next_whirlpool_id + 9 next_whirlpool_id += 1 base_x, base_y = (i & 7) << 9, (i & 56) << 6 A.kBirdTravel_ScreenIndex[j] = i A.kBirdTravel_Map16LoadSrcOff[j] = get_loadoffs(t['scroll_xy'], t['load_xy']) A.kBirdTravel_ScrollX[j] = t['scroll_xy'][0] + base_x A.kBirdTravel_ScrollY[j] = t['scroll_xy'][1] + base_y A.kBirdTravel_LinkXCoord[j] = t['xy'][0] + base_x A.kBirdTravel_LinkYCoord[j] = t['xy'][1] + base_y A.kBirdTravel_CameraXScroll[j] = t['camera_xy'][0] + base_x A.kBirdTravel_CameraYScroll[j] = t['camera_xy'][1] + base_y A.kBirdTravel_Unk1[j] = t['unk'][0] A.kBirdTravel_Unk3[j] = t['unk'][1] A.add('uint16', 'kOverworld_Entrance_Area', 129) A.add('uint16', 'kOverworld_Entrance_Pos', 129) A.add('uint8', 'kOverworld_Entrance_Id', 129) for i, y in loaded_areas: for e in y['Entrances']: j = e['index'] assert A.kOverworld_Entrance_Id[j] == None A.kOverworld_Entrance_Area[j] = i A.kOverworld_Entrance_Id[j] = e['entrance_id'] A.kOverworld_Entrance_Pos[j] = e['x'] << 1 | e['y'] << 7 A.add('uint16', 'kFallHole_Area', 19) A.add('uint16', 'kFallHole_Pos', 19) A.add('uint8', 'kFallHole_Entrances', 19) holes = [] for i, y in loaded_areas: if 'Holes' not in y: continue for e in y['Holes']: x, y, j = e['x'], e['y'], e['entrance_id'] holes.append((j, x << 1 | ((y - 8) & 0x3f) << 7, i)) for i, (entrance, pos, area) in enumerate(sorted(holes)): A.kFallHole_Area[i] = area A.kFallHole_Pos[i] = pos A.kFallHole_Entrances[i] = entrance A.add('uint8', 'kExitData_ScreenIndex', 79) for a in ['kExitDataRooms', 'kExitData_Map16LoadSrcOff', 'kExitData_ScrollX', 'kExitData_ScrollY', 'kExitData_XCoord', 'kExitData_YCoord', 'kExitData_CameraXScroll', 'kExitData_CameraYScroll']: A.add('uint16', a, 79) A.add('uint16', 'kExitData_NormalDoor', 79, initializer = 0) A.add('uint16', 'kExitData_FancyDoor', 79, initializer = 0) A.add('int8', 'kExitData_Unk1', 79) A.add('int8', 'kExitData_Unk3', 79) A.add('uint16', 'kSpExit_Top', 16, initializer = 0) A.add('uint16', 'kSpExit_Bottom', 16, initializer = 0) A.add('uint16', 'kSpExit_Left', 16, initializer = 0) A.add('uint16', 'kSpExit_Right', 16, initializer = 0) A.add('int16', 'kSpExit_Tab4', 16, initializer = 0) A.add('int16', 'kSpExit_Tab5', 16, initializer = 0) A.add('int16', 'kSpExit_Tab6', 16, initializer = 0) A.add('int16', 'kSpExit_Tab7', 16, initializer = 0) A.add('uint16', 'kSpExit_LeftEdgeOfMap', 16, initializer = 0) A.add('uint8', 'kSpExit_Dir', 16, initializer = 0) A.add('uint8', 'kSpExit_SprGfx', 16, initializer = 0) A.add('uint8', 'kSpExit_AuxGfx', 16, initializer = 0) A.add('uint8', 'kSpExit_PalBg', 16, initializer = 0) A.add('uint8', 'kSpExit_PalSpr', 16, initializer = 0) for i, y in loaded_areas: for e in y['Exits']: j = e['index'] base_x, base_y = (i & 7) << 9, (i & 56) << 6 assert A.kExitData_ScreenIndex[j] == None A.kExitData_ScreenIndex[j] = i room = e['room'] A.kExitDataRooms[j] = e['room'] A.kExitData_Map16LoadSrcOff[j] = get_loadoffs(e['scroll_xy'], e['load_xy']) A.kExitData_ScrollX[j] = e['scroll_xy'][0] + base_x A.kExitData_ScrollY[j] = e['scroll_xy'][1] + base_y A.kExitData_XCoord[j] = e['xy'][0] + base_x A.kExitData_YCoord[j] = e['xy'][1] + base_y A.kExitData_CameraXScroll[j] = e['camera_xy'][0] + base_x A.kExitData_CameraYScroll[j] = e['camera_xy'][1] + base_y A.kExitData_Unk1[j] = e['unk'][0] A.kExitData_Unk3[j] = e['unk'][1] door = e.get('door') if door != None: if door[0] in ('bombable', 'wooden'): A.kExitData_NormalDoor[j] = door[1] << 1 | door[2] << 7 | (0x8000 if door[0] == 'bombable' else 0) elif door[0] in ('palace', 'sanctuary'): A.kExitData_FancyDoor[j] = door[1] << 1 | door[2] << 7 | (0x8000 if door[0] == 'palace' else 0) else: assert e[0] == 'none' se = e.get('special_exit') if se: j = room - 0x180 A.kSpExit_Dir[j] = se['dir'] * 2 A.kSpExit_SprGfx[j] = se['spr_gfx'] A.kSpExit_AuxGfx[j] = se['aux_gfx'] A.kSpExit_PalBg[j] = se['pal_bg'] A.kSpExit_PalSpr[j] = se['pal_spr'] A.kSpExit_Top[j] = se['top'] A.kSpExit_Bottom[j] = se['bottom'] A.kSpExit_Left[j] = se['left'] A.kSpExit_Right[j] = se['right'] A.kSpExit_LeftEdgeOfMap[j] = se['left_edge_of_map'] A.kSpExit_Tab4[j] = se['unk4'] A.kSpExit_Tab5[j] = se['unk5'] A.kSpExit_Tab6[j] = se['unk6'] A.kSpExit_Tab7[j] = se['unk7'] A.add('uint16', 'kOverworldSecrets_Offs', 128, initializer = None) A.add('uint8', 'kOverworldSecrets', 0) for i, y in loaded_areas: if len(y['Items']): assert i < 128 j = len(A.kOverworldSecrets) awrite(A.kOverworldSecrets_Offs, i, i, j) for e in y['Items']: pos = e[0] << 1 | e[1] << 7 item = tables.kSecretNamesRev[e[2]] A.kOverworldSecrets.extend([pos & 0xff, pos >> 8, item]) A.kOverworldSecrets.extend([0xff, 0xff]) for i in range(128): if A.kOverworldSecrets_Offs[i] == None: A.kOverworldSecrets_Offs[i] = len(A.kOverworldSecrets) - 2 A.add('uint16', 'kOverworldSpriteOffs', 144 * 3, initializer = 0) A.add('uint8', 'kOverworldSprites', 0) A.add('uint8', 'kOverworldSpriteGfx', 256) A.add('uint8', 'kOverworldSpritePalettes', 256) A.kOverworldSprites.append(0xff) def do_sprite_range(start, end, stagename, sprite_stage_idxs, infostage): for i, y in loaded_areas: if i < start or i >= end: continue info = y[stagename]['info'] if i < 128: awrite(A.kOverworldSpriteGfx, i, (i & 63) + infostage * 64, info['gfx']) awrite(A.kOverworldSpritePalettes, i, (i & 63) + infostage * 64, info['palette']) if len(y[stagename]['sprites']): for stage in sprite_stage_idxs: A.kOverworldSpriteOffs[stage * 144 + i] = len(A.kOverworldSprites) for e in y[stagename]['sprites']: A.kOverworldSprites.extend([e[1], e[0], tables.kSpriteNamesRev[e[2]]]) A.kOverworldSprites.append(0xff) do_sprite_range(0, 64, 'Sprites.Beginning', [0], 0) do_sprite_range(0, 64, 'Sprites.FirstPart', [1], 1) do_sprite_range(0, 64, 'Sprites.SecondPart', [2], 2) do_sprite_range(64, 144, 'Sprites', [1, 2], 3) A.write() add_asset_uint8('kMap8DataToTileAttr', ROM.get_bytes(0x8E9459, 512)) add_asset_uint8('kSomeTileAttr', ROM.get_bytes(0x9bf110, 3824)) def print_dungeon_map(): r, r2 = [], [] for i in range(14): kSizes = [75, 125, 50, 75, 175, 75, 50, 75, 50, 200, 150, 75, 100, 200] addr = 0xa0000 + ROM.get_word(0x8AF605 + i * 2) b = ROM.get_bytes(addr, kSizes[i]) nonzero_bytes = len(b) - b.count(0xf) r.append(bytes(b)) addr = 0xa0000 + ROM.get_word(0x8AFBE4 + i * 2) b = ROM.get_bytes(addr, nonzero_bytes) r2.append(bytes(b)) add_asset_uint8('kDungMap_FloorLayout', pack_u32_arrays(r)) add_asset_uint8('kDungMap_Tiles', pack_u32_arrays(r2)) g_dungeon_yaml_cache = {} def load_dungeon_yaml(room): if room not in g_dungeon_yaml_cache: g_dungeon_yaml_cache[room] = yaml.safe_load(open(PATH+'dungeon/dungeon-%d.yaml' % room, 'r')) return g_dungeon_yaml_cache[room] def print_dungeon_sprites(): offsets=[0 for i in range(320)] data = [0, 0xff] for i in range(320): y = load_dungeon_yaml(i) sortmode = y['Header']['sort_sprites'] if len(y['Sprites']) == 0 and sortmode == 0: continue offsets[i] = len(data) data.append(sortmode) for s in y['Sprites']: xx, yy, f, name = s[:4] f = {'upper' : 0, 'lower' : 1}[f] assert xx >= 0 and xx <= 0x1f assert yy >= 0 and yy <= 0x1f # parse out subcode ss = 0 if len(name) > 2 and name[2] == '.': j = name.index('-') ss = int(name[3:j]) name = name[0:2] + name[j:] name = tables.kSpriteName2Idx[name] if name >= 0x100: data.extend((f << 7 | 0 << 5 | yy, xx | 7 << 5, name & 0xff)) else: data.extend((f << 7 | (ss >> 3) << 5 | yy, xx | (ss & 7) << 5, name)) if len(s) == 5: kDropTypesToCode = {'drop_key' : [0xfe, 0, 0xe4], 'drop_big_key' : [0xfd, 0, 0xe4]} data.extend(kDropTypesToCode[s[-1]]) data.append(0xff) add_asset_uint8('kDungeonSprites', data) add_asset_uint16('kDungeonSpriteOffs', offsets) def print_dungeon_secrets(): result = [None] * 640 for i in range(320): y = load_dungeon_yaml(i) def fixone(s): x, y, data = s[0], s[1], tables.kSecretNamesRev[s[2]] pos = (x + y * 64) * 2 return [pos & 0xff, pos >> 8, data] if len(y['Secrets']): result[i*2+0] = len(result) & 0xff result[i*2+1] = len(result) >> 8 for a in y['Secrets']: result.extend(fixone(a)) result.extend([0xff, 0xff]) for i in range(320): if result[i*2+0] == None: l = (len(result) - 2) result[i*2+0] = l & 0xff result[i*2+1] = l >> 8 add_asset_uint8('kDungeonSecrets', result) def append_scan_bytes(big, little): for n in range(len(little), -1, -1): if n == 0 or big[-n:] == little[:n]: offset = len(big) - n big.extend(little[n:]) return offset def print_dungeon_rooms(): def print_layer(objs, doors): door_offset = None for o in objs: if o['n'] in tables.kType0Names_rev: index = tables.kType0Names_rev[o['n']] # 0 - 0xf7 w = int(o['s'][0]) h = int(o['s'][2]) assert w >= 0 and w <= 3 and h >= 0 and h <= 3 p0 = o['x'] * 4 + w p1 = o['y'] * 4 + h p2 = index elif o['n'] in tables.kType1Names_rev: index = tables.kType1Names_rev[o['n']] p0 = o['x'] * 4 + (index >> 0 & 3) p1 = o['y'] * 4 + (index >> 2 & 3) p2 = (index >> 4) + 0xf8 elif o['n'] in tables.kType2Names_rev: index = tables.kType2Names_rev[o['n']] x, y = o['x'], o['y'] #111111xx xxxxyyyy yyiiiiii p0 = 0xfc + (x >> 4 & 3) p1 = (x << 4 & 0xf0) | (y >> 2 & 0x0f) p2 = index | (y << 6 & 0xc0) else: raise Exception('item %s not found' % o['n']) data.extend((p0, p1, p2)) if doors != None: data.extend([0xf0, 0xff]) door_offset = len(data) for d in doors: data.extend((d['dir'] | d['pos'] << 4, d['type'])) data.extend([0xff, 0xff]) return door_offset def get_room_header(y): h = y['Header'] p7 = h['hole0_dest'][1] | h['stair0_dest'][1] << 2 | h['stair1_dest'][1] << 4 | h['stair2_dest'][1] << 6 p8 = h['stair3_dest'][1] return [tables.kBg2_rev[h['bg2']] << 5 | tables.kCollisionNames_rev[h['collision']] << 2 | h['lights_out'], h['palette'], h['blockset'], h['enemyblk'], tables.kEffectNames_rev[h['effect']], tables.kTagNames_rev[h['tag0']], tables.kTagNames_rev[h['tag1']], p7, p8, h['hole0_dest'][0], h['stair0_dest'][0], h['stair1_dest'][0], h['stair2_dest'][0], h['stair3_dest'][0]] data = [] offsets = [0]*320 door_offsets = [0]*320 room_headers = [] header_offsets = [0] * 320 chests = [] entrances = [None] * 133 starting_points = [None] * 7 sign_texts = [0] * 320 pits_hurt_player = [] for i in range(320): y = load_dungeon_yaml(i) h = y['Header'] if h['pits_hurt_player']: pits_hurt_player.append(i) offsets[i]=len(data) data.append(h['floor1'] + h['floor2'] * 16) data.append(h['layout'] * 4 + h['start_quadrant']) print_layer(y['Layer1'], y.get('Layer1.doors')) print_layer(y['Layer2'], y.get('Layer2.doors')) door_offsets[i] = print_layer(y['Layer3'], y.get('Layer3.doors') or []) header_offsets[i] = append_scan_bytes(room_headers, get_room_header(y)) sign_texts[i] = h['tele_msg'] for a in y['Chests']: if isinstance(a, int): chests.extend([i & 0xff, i >> 8, a]) else: assert a.endswith('!') chests.extend([i & 0xff, (i >> 8) | 0x80, int(a[:-1])]) for e in y['Entrances']: a = e['entrance_index'] e['room'] = i assert entrances[a] == None entrances[a] = e if 'StartingPoints' in y: for e in y['StartingPoints']: a = e['starting_point_index'] e['room'] = i assert starting_points[a] == None starting_points[a] = e for i in range(133): if entrances[i] == None: raise Exception('Entrance %d not defined' % i) for i in range(7): if starting_points[i] == None: raise Exception('Starting point %d not defined' % i) def print_entrance_info(entrances, prefix): def get_rc(a): rep, room, quads, xy = a.get('repair_scroll_bounds'), a['room'], a['quadrants'], a['player_xy'] if rep == None: rep = (0, 0, 0, 0, 0, 0, 0, 0) base_x = (room & 0xf) * 2 base_y = (room >> 4) * 2 ym = (xy[1] & 0x100) >> 8 xm = (xy[0] & 0x100) >> 8 qqq = xm if room >= 242 and quads[0] == 'single_x' else 0 l = [base_y + ym, base_y, base_y + ym, base_y + 1, base_x + xm, base_x + qqq, base_x + xm, base_x + qqq + 1] return [a+b for a,b in zip(l, rep)] def get_palace(a): i = tables.kPalaceNames.index(a) return -1 if i == 0 else (i - 1) * 2 def get_quadrant1(a): return ['single_x', 'double_x'].index(a[0]) * 0x20 + ['single_y', 'double_y'].index(a[1]) * 0x2 def get_quadrant2(a): kScrollNames = { 'upper_left' : 0, 'lower_left' : 2, 'upper_right' : 16, 'lower_right' : 18 } return kScrollNames[a[2]] def get_exit_door(a): if a[0] == 'none': return 0 if a[0] == 'none_0xffff': return 65535 return ['wooden', 'bombable'].index(a[0]) << 15 | a[1] << 1 | a[2] << 7 add_asset_uint16(prefix+'rooms', [a['room'] for a in entrances]) add_asset_uint8(prefix+'relativeCoords', flatten([get_rc(a) for a in entrances])) def get_base_x(a): return ((a['room'] & 0x00f) << 9) def get_base_y(a): return ((a['room'] & 0x1f0) << 5) add_asset_uint16(prefix+'scrollX', [a['scroll_xy'][0] + get_base_x(a) for a in entrances]) add_asset_uint16(prefix+'scrollY', [a['scroll_xy'][1] + get_base_y(a) for a in entrances]) add_asset_uint16(prefix+'playerX', [a['player_xy'][0] + get_base_x(a) for a in entrances]) add_asset_uint16(prefix+'playerY', [a['player_xy'][1] + get_base_y(a) for a in entrances]) add_asset_uint16(prefix+'cameraX', [a['camera_xy'][0] for a in entrances]) add_asset_uint16(prefix+'cameraY', [a['camera_xy'][1] for a in entrances]) add_asset_uint8(prefix+'blockset', [a['blockset'] for a in entrances]) add_asset_int8(prefix+'floor', [a['floor'] for a in entrances]) add_asset_int8(prefix+'palace', [get_palace(a['palace']) for a in entrances]) add_asset_uint8(prefix+'doorwayOrientation', [a['doorway_orientation'] for a in entrances]) add_asset_uint8(prefix+'startingBg', [a['plane'] + a['ladder_level'] * 16 for a in entrances]) add_asset_uint8(prefix+'quadrant1', [get_quadrant1(a['quadrants']) for a in entrances]) add_asset_uint8(prefix+'quadrant2', [get_quadrant2(a['quadrants']) for a in entrances]) add_asset_uint16(prefix+'doorSettings', [get_exit_door(a['house_exit_door']) for a in entrances]) if prefix == 'kStartingPoint_': add_asset_uint8(prefix+'entrance', [a['associated_entrance_index'] for a in entrances]) m = invert_dict(tables.kMusicNames) add_asset_uint8(prefix+'musicTrack', [m[a['music']] for a in entrances]) add_asset_uint8('kDungeonRoom', data) add_asset_uint16('kDungeonRoomOffs', offsets) add_asset_uint16('kDungeonRoomDoorOffs', door_offsets) add_asset_uint8('kDungeonRoomHeaders', room_headers) add_asset_uint16('kDungeonRoomHeadersOffs', header_offsets) add_asset_uint8('kDungeonRoomChests', chests) add_asset_uint16('kDungeonRoomTeleMsg', sign_texts) add_asset_uint16('kDungeonPitsHurtPlayer', pits_hurt_player) print_entrance_info(entrances, 'kEntranceData_') print_entrance_info(starting_points, 'kStartingPoint_') data = [] offsets = [0] * 8 default_yaml = yaml.safe_load(open(PATH+'dungeon/default_rooms.yaml', 'r')) for i in range(len(offsets)): offsets[i] = len(data) print_layer(default_yaml['Default%d' % i], None) add_asset_uint8('kDungeonRoomDefault', data) add_asset_uint16('kDungeonRoomDefaultOffs', offsets) data = [] offsets = [0] * 19 overlay_yaml = yaml.safe_load(open(PATH+'dungeon/overlay_rooms.yaml', 'r')) for i in range(len(offsets)): offsets[i] = len(data) print_layer(overlay_yaml['Overlay%d' % i], None) add_asset_uint8('kDungeonRoomOverlay', data) add_asset_uint16('kDungeonRoomOverlayOffs', offsets) print_dungeon_secrets() add_asset_uint16('kDungAttrsForTile_Offs', ROM.get_words(0x8e9000, 21)) add_asset_uint8('kDungAttrsForTile', ROM.get_bytes(0x8e902a, 1024)) add_asset_uint16('kMovableBlockDataInit', ROM.get_words(0x84f1de, 198)) add_asset_uint16('kTorchDataInit', ROM.get_words(0x84F36A, 144)) add_asset_uint16('kTorchDataJunk', ROM.get_words(0x84F48a, 48)) def print_enemy_damage_data(): decomp, comp_len = util.decomp(0x83e800, ROM.get_byte, True, True) add_asset_uint8('kEnemyDamageData', decomp) def print_tilemaps(): kSrcs = [0xcdd6d, 0xce7bf, 0xce2a8, 0xce63c, 0xce456, 0xeda9c] def decode_one(p): p_org = p while not (ROM.get_byte(p) & 0x80): is_memset = ROM.get_byte(p+2) & 0x40 len = ((ROM.get_byte(p+2)*256+ROM.get_byte(p+3))&0x3fff) + 1 p += 4 p += 2 if is_memset else len return p - p_org + 1 for i,s in enumerate(kSrcs): l = decode_one(s) add_asset_uint8('kBgTilemap_%d' % i, ROM.get_bytes(s, l)) def print_link_graphics(): image = Image.open(PATH+'linksprite.png') data = image.tobytes() def encode_4bit_sprite(data, offset, pitch): b = [0] * 32 for y in range(8): for x in range(8): v = data[offset + y * pitch + x] b[y*2+0] |= (v & 1) << (7-x) b[y*2+1] |= (v >> 1 & 1) << (7-x) b[y*2+16] |= (v >> 2 & 1) << (7-x) b[y*2+17] |= (v >> 3 & 1) << (7-x) return bytes(b) b = b'' for y in range(56): for x in range(16): b += encode_4bit_sprite(data, y * 128 * 8 + x * 8, 128) add_asset_uint8('kLinkGraphics', b) def print_sound_banks(): for song in ['intro', 'indoor', 'ending']: name, data = compile_music.print_song(song) add_asset_uint8(name, data) def print_all(): print_sound_banks() print_dungeon_rooms() print_enemy_damage_data() print_link_graphics() print_dungeon_sprites() print_map32_to_map16() print_dialogue() print_images() print_misc() print_dungeon_map() print_tilemaps() print_overworld() print_overworld_tables() print_all() def write_assets_to_file(print_header = False): key_sig = b'' all_data = [] if print_header: print('''#pragma once #include "types.h" enum { kNumberOfAssets = %d }; extern const uint8 *g_asset_ptrs[kNumberOfAssets]; extern uint32 g_asset_sizes[kNumberOfAssets];''' % len(assets)) for i, (k, (tp, data)) in enumerate(assets.items()): if print_header: print('#define %s ((%s*)g_asset_ptrs[%d])' % (k, tp, i)) print('#define %s_SIZE (g_asset_sizes[%d])' % (k, i)) key_sig += k.encode('utf8') + b'\0' all_data.append(data) assets_sig = b'Zelda3_v0 \n\0' + hashlib.sha256(key_sig).digest() if print_header: print('#define kAssets_Sig %s' % ", ".join((str(a) for a in assets_sig))) hdr = assets_sig + b'\x00' * 32 + struct.pack('II', len(all_data), len(key_sig)) encoded_sizes = array.array('I', [len(i) for i in all_data]) file_data = hdr + encoded_sizes + key_sig for v in all_data: while len(file_data) & 3: file_data += b'\0' file_data += v open('zelda3_assets.dat', 'wb').write(file_data) write_assets_to_file(False)