#include "meta.h" #include "../coding/coding.h" typedef enum { PSX, PCM16, ATRAC9, HEVAG } bnk_codec; /* BNK - Sony's Scream Tool bank format [Puyo Puyo Tetris (PS4), NekoBuro: Cats Block (Vita)] */ VGMSTREAM * init_vgmstream_bnk_sony(STREAMFILE *streamFile) { #if 1 VGMSTREAM * vgmstream = NULL; off_t start_offset, stream_offset, name_offset = 0; size_t stream_size, interleave = 0; off_t sblk_offset, data_offset; size_t data_size; int channel_count = 0, loop_flag, sample_rate, parts, version; int loop_start = 0, loop_end = 0; uint32_t atrac9_info = 0; int total_subsongs, target_subsong = streamFile->stream_index; int32_t (*read_32bit)(off_t,STREAMFILE*) = NULL; int16_t (*read_16bit)(off_t,STREAMFILE*) = NULL; bnk_codec codec; /* checks */ if (!check_extensions(streamFile, "bnk")) goto fail; if (read_32bitBE(0x00,streamFile) == 0x00000003) { /* PS3 */ read_32bit = read_32bitBE; read_16bit = read_16bitBE; } else if (read_32bitBE(0x00,streamFile) == 0x03000000) { /* Vita/PS4 */ read_32bit = read_32bitLE; read_16bit = read_16bitLE; } else { goto fail; } parts = read_32bit(0x04,streamFile); if (parts < 2 || parts > 3) goto fail; sblk_offset = read_32bit(0x08,streamFile); /* 0x0c: sklb size */ data_offset = read_32bit(0x10,streamFile); data_size = read_32bit(0x14,streamFile); /* 0x18: ZLSD small footer, rare [Yakuza 6's Puyo Puyo (PS4)] */ /* 0x1c: ZLSD size */ /* SE banks, also used for music. Most table fields seems reserved/defaults and * don't change much between subsongs or files, so they aren't described in detail */ /* SBlk part: parse header */ if (read_32bit(sblk_offset+0x00,streamFile) != 0x6B6C4253) /* "klBS" (SBlk = sample block?) */ goto fail; version = read_32bit(sblk_offset+0x04,streamFile); /* 0x08: possibly when version=0x0d, 0x03=Vita, 0x06=PS4 */ //;VGM_LOG("BNK: sblk_offset=%lx, data_offset=%lx, version %x\n", sblk_offset, data_offset, version); { int i; off_t table1_offset, table2_offset, table3_offset, table4_offset; size_t section_entries, material_entries, stream_entries; size_t table1_entry_size; off_t table1_suboffset, table2_suboffset, table3_suboffset; off_t table2_entry_offset = 0, table3_entry_offset = 0; int table4_entry_id = -1; off_t table4_entries_offset, table4_names_offset; switch(version) { case 0x03: /* L@ove Once - Mermaid's Tears (PS3) */ case 0x04: /* Test banks */ case 0x09: /* Puyo Puyo Tetris (PS4) */ section_entries = (uint16_t)read_16bit(sblk_offset+0x16,streamFile); /* entry size: ~0x0c */ material_entries = (uint16_t)read_16bit(sblk_offset+0x18,streamFile); /* entry size: ~0x08 */ stream_entries = (uint16_t)read_16bit(sblk_offset+0x1a,streamFile); /* entry size: ~0x60 */ table1_offset = sblk_offset + read_32bit(sblk_offset+0x1c,streamFile); table2_offset = sblk_offset + read_32bit(sblk_offset+0x20,streamFile); table3_offset = sblk_offset + read_32bit(sblk_offset+0x34,streamFile); table4_offset = sblk_offset + read_32bit(sblk_offset+0x38,streamFile); table1_entry_size = 0x0c; table1_suboffset = 0x08; table2_suboffset = 0x00; table3_suboffset = 0x10; break; case 0x0d: /* Polara (Vita), Crypt of the Necrodancer (Vita) */ case 0x0e: /* Yakuza 6's Puyo Puyo (PS4) */ table1_offset = sblk_offset + read_32bit(sblk_offset+0x18,streamFile); table2_offset = sblk_offset + read_32bit(sblk_offset+0x1c,streamFile); table3_offset = sblk_offset + read_32bit(sblk_offset+0x2c,streamFile); table4_offset = sblk_offset + read_32bit(sblk_offset+0x30,streamFile); section_entries = (uint16_t)read_16bit(sblk_offset+0x38,streamFile); /* entry size: ~0x24 */ material_entries = (uint16_t)read_16bit(sblk_offset+0x3a,streamFile); /* entry size: ~0x08 */ stream_entries = (uint16_t)read_16bit(sblk_offset+0x3c,streamFile); /* entry size: ~0x90 + variable (sometimes) */ table1_entry_size = 0x24; table1_suboffset = 0x0c; table2_suboffset = 0x00; table3_suboffset = 0x44; break; default: VGM_LOG("BNK: unknown version %x\n", version); goto fail; } //;VGM_LOG("BNK: table offsets=%lx, %lx, %lx, %lx\n", table1_offset,table2_offset,table3_offset,table4_offset); //;VGM_LOG("BNK: table entries=%i, %i, %i\n", section_entries,material_entries,stream_entries); /* table defs: * - table1: sections, point to some materials (may be less than streams/materials) * - table2: materials, point to all sounds or others subtypes (may be more than sounds) * - table3: sounds, point to streams (multiple sounds can repeat stream) * - table4: names define section names (not all sounds may have a name) * * approximate table parsing * - check materials and skip non-sounds to get table3 offsets (since table3 entry size isn't always constant) * - get stream offsets * - find if one section points to the selected material, and get section name = stream name */ /* parse materials */ total_subsongs = 0; if (target_subsong == 0) target_subsong = 1; for (i = 0; i < material_entries; i++) { uint32_t table2_value, table2_subinfo, table2_subtype; table2_value = (uint32_t)read_32bit(table2_offset+(i*0x08)+table2_suboffset+0x00,streamFile); table2_subinfo = (table2_value >> 0) & 0xFFFF; table2_subtype = (table2_value >> 16) & 0xFFFF; if (table2_subtype != 0x100) continue; /* not sounds */ total_subsongs++; if (total_subsongs == target_subsong) { table2_entry_offset = (i*0x08); table3_entry_offset = table2_subinfo; /* continue to count all subsongs*/ } } //;VGM_LOG("BNK: subsongs %i, table2_entry=%lx, table3_entry=%lx\n", total_subsongs,table2_entry_offset,table3_entry_offset); if (target_subsong < 0 || target_subsong > total_subsongs || total_subsongs < 1) goto fail; /* this means some subsongs repeat streams, that can happen in some sfx banks, whatevs */ if (total_subsongs != stream_entries) { //;VGM_LOG("BNK: subsongs %i vs table3 %i don't match\n", total_subsongs, stream_entries); /* find_dupes...? */ } /* parse sounds */ stream_offset = read_32bit(table3_offset+table3_entry_offset+table3_suboffset+0x00,streamFile); stream_size = read_32bit(table3_offset+table3_entry_offset+table3_suboffset+0x04,streamFile); /* parse names */ switch(version) { //case 0x03: /* different format? */ //case 0x04: /* different format? */ case 0x09: case 0x0d: case 0x0e: /* find if this sound has an assigned name in table1 */ for (i = 0; i < section_entries; i++) { off_t entry_offset = (uint16_t)read_16bit(table1_offset+(i*table1_entry_size)+table1_suboffset+0x00,streamFile); /* rarely (ex. Polara sfx) one name applies to multiple materials, * from current entry_offset to next entry_offset (section offsets should be in order) */ if (entry_offset <= table2_entry_offset ) { table4_entry_id = i; //break; } } /* table4: */ /* 0x00: bank name (optional) */ /* 0x08: header size */ /* 0x0c: table4 size */ /* variable: entries */ /* variable: names (null terminated) */ table4_entries_offset = table4_offset + read_32bit(table4_offset+0x08, streamFile); table4_names_offset = table4_entries_offset + (0x10*section_entries); //;VGM_LOG("BNK: t4_entries=%lx, t4_names=%lx\n", table4_entries_offset, table4_names_offset); /* get assigned name from table4 names */ for (i = 0; i < section_entries; i++) { int entry_id = read_32bit(table4_entries_offset+(i*0x10)+0x0c, streamFile); if (entry_id == table4_entry_id) { name_offset = table4_names_offset + read_32bit(table4_entries_offset+(i*0x10)+0x00, streamFile); break; } } break; default: break; } //;VGM_LOG("BNK: stream_offset=%lx, stream_size=%x, name_offset=%lx\n", stream_offset, stream_size, name_offset); } /* data part: parse extradata before the codec, if needed */ { int type, loop_length; size_t extradata_size = 0, postdata_size = 0; start_offset = data_offset + stream_offset; switch(version) { case 0x03: case 0x04: sample_rate = 48000; /* seems ok */ channel_count = 1; /* hack for PS3 files that use dual subsongs as stereo */ if (total_subsongs == 2 && stream_size * 2 == data_size) { channel_count = 2; stream_size = stream_size*channel_count; total_subsongs = 1; } interleave = stream_size / channel_count; //postdata_size = 0x10; /* last frame may be garbage */ loop_flag = ps_find_loop_offsets(streamFile, start_offset, stream_size, channel_count, interleave, &loop_start, &loop_end); loop_flag = (loop_start > 28); /* ignore full loops since they just fadeout + repeat */ codec = PSX; break; case 0x09: type = read_16bit(start_offset+0x00,streamFile); extradata_size = 0x08 + read_32bit(start_offset+0x04,streamFile); /* 0x14 for AT9 */ switch(type) { case 0x02: /* ATRAC9 mono */ case 0x05: /* ATRAC9 stereo */ if (read_32bit(start_offset+0x08,streamFile) + 0x08 != extradata_size) /* repeat? */ goto fail; sample_rate = 48000; /* seems ok */ channel_count = (type == 0x02) ? 1 : 2; atrac9_info = (uint32_t)read_32bitBE(start_offset+0x0c,streamFile); /* 0x10: null? */ loop_length = read_32bit(start_offset+0x14,streamFile); loop_start = read_32bit(start_offset+0x18,streamFile); loop_end = loop_start + loop_length; /* loop_start is -1 if not set */ codec = ATRAC9; break; default: VGM_LOG("BNK: unknown type %x\n", type); goto fail; } break; case 0x0d: case 0x0e: type = read_16bit(start_offset+0x00,streamFile); if (read_32bit(start_offset+0x04,streamFile) != 0x01) /* type? */ goto fail; extradata_size = 0x10 + read_32bit(start_offset+0x08,streamFile); /* 0x80 for AT9, 0x10 for PCM/PS-ADPCM */ /* 0x0c: null? */ switch(type) { case 0x02: /* ATRAC9 mono */ case 0x05: /* ATRAC9 stereo */ if (read_32bit(start_offset+0x10,streamFile) + 0x10 != extradata_size) /* repeat? */ goto fail; sample_rate = 48000; /* seems ok */ channel_count = (type == 0x02) ? 1 : 2; atrac9_info = (uint32_t)read_32bitBE(start_offset+0x14,streamFile); /* 0x18: null? */ /* 0x1c: channels? */ /* 0x20: null? */ loop_length = read_32bit(start_offset+0x24,streamFile); loop_start = read_32bit(start_offset+0x28,streamFile); loop_end = loop_start + loop_length; /* loop_start is -1 if not set */ codec = ATRAC9; break; case 0x01: /* PCM16LE mono? (NekoBuro/Polara sfx) */ case 0x04: /* PCM16LE stereo? (NekoBuro/Polara sfx) */ sample_rate = 48000; /* seems ok */ /* 0x10: null? */ channel_count = read_32bit(start_offset+0x14,streamFile); interleave = 0x02; loop_start = read_32bit(start_offset+0x18,streamFile); loop_length = read_32bit(start_offset+0x1c,streamFile); loop_end = loop_start + loop_length; /* loop_start is -1 if not set */ codec = PCM16; break; case 0x00: /* PS-ADPCM (test banks) */ sample_rate = 48000; /* seems ok */ /* 0x10: null? */ channel_count = read_32bit(start_offset+0x14,streamFile); interleave = 0x02; loop_start = read_32bit(start_offset+0x18,streamFile); loop_length = read_32bit(start_offset+0x1c,streamFile); loop_end = loop_start + loop_length; /* loop_start is -1 if not set */ codec = HEVAG; break; default: VGM_LOG("BNK: unknown type %x\n", type); goto fail; } break; default: goto fail; } start_offset += extradata_size; stream_size -= extradata_size; stream_size -= postdata_size; //;VGM_LOG("BNK: offset=%lx, size=%x\n", start_offset, stream_size); } loop_flag = (loop_start >= 0) && (loop_end > 0); /* build the VGMSTREAM */ vgmstream = allocate_vgmstream(channel_count,loop_flag); if (!vgmstream) goto fail; vgmstream->sample_rate = sample_rate; vgmstream->num_streams = total_subsongs; vgmstream->stream_size = stream_size; vgmstream->meta_type = meta_BNK_SONY; switch(codec) { #ifdef VGM_USE_ATRAC9 case ATRAC9: { atrac9_config cfg = {0}; cfg.channels = vgmstream->channels; cfg.config_data = atrac9_info; //cfg.encoder_delay = 0x00; //todo vgmstream->codec_data = init_atrac9(&cfg); if (!vgmstream->codec_data) goto fail; vgmstream->coding_type = coding_ATRAC9; vgmstream->layout_type = layout_none; vgmstream->num_samples = atrac9_bytes_to_samples(stream_size, vgmstream->codec_data); vgmstream->loop_start_sample = loop_start; vgmstream->loop_end_sample = loop_end; break; } #endif case PCM16: vgmstream->coding_type = coding_PCM16LE; vgmstream->layout_type = layout_interleave; vgmstream->interleave_block_size = interleave; vgmstream->num_samples = pcm_bytes_to_samples(stream_size, vgmstream->channels, 16); vgmstream->loop_start_sample = loop_start; vgmstream->loop_end_sample = loop_end; break; case PSX: vgmstream->sample_rate = 48000; vgmstream->coding_type = coding_PSX; vgmstream->layout_type = layout_interleave; vgmstream->interleave_block_size = interleave; vgmstream->num_samples = ps_bytes_to_samples(stream_size,channel_count); vgmstream->loop_start_sample = loop_start; vgmstream->loop_end_sample = loop_end; break; case HEVAG: vgmstream->sample_rate = 48000; vgmstream->coding_type = coding_HEVAG; vgmstream->layout_type = layout_interleave; vgmstream->interleave_block_size = interleave; vgmstream->num_samples = ps_bytes_to_samples(stream_size,channel_count); vgmstream->loop_start_sample = loop_start; vgmstream->loop_end_sample = loop_end; break; } if (name_offset) read_string(vgmstream->stream_name,STREAM_NAME_SIZE, name_offset,streamFile); if (!vgmstream_open_stream(vgmstream,streamFile,start_offset)) goto fail; return vgmstream; fail: close_vgmstream(vgmstream); #endif return NULL; }