kwz-restoration.hpp

#pragma once

typedef uint8_t  u8;
typedef int16_t  s16;
typedef uint16_t u16;
typedef uint32_t u32;

std::vector<u8> file_buffer;

template<typename X, typename Y, typename Z>
X clampValue(X value, Y min, Z max) {
    if (value < min) value = (X)min;
    if (value > max) value = (X)max;
    return value;
}

template<typename T>
T getInt(int pos) {
    return *reinterpret_cast<T*>(file_buffer.data() + pos);
}

int track_sizes[4] = {};
int track_offsets[4] = {};
int step_indexes[4] = {};

// ADPCM Tables
const int ADPCM_INDEX_TABLE_2[4] = { -1, 2, -1, 2 };

const int ADPCM_INDEX_TABLE_4[16] = { -1, -1, -1, -1, 2, 4, 6, 8,
                                          -1, -1, -1, -1, 2, 4, 6, 8 };

const s16 ADPCM_STEP_TABLE[89] = {     7,     8,     9,    10,    11,    12,
                                      13,    14,    16,    17,    19,    21,
                                      23,    25,    28,    31,    34,    37,
                                      41,    45,    50,    55,    60,    66,
                                      73,    80,    88,    97,   107,   118,
                                     130,   143,   157,   173,   190,   209,
                                     230,   253,   279,   307,   337,   371,
                                     408,   449,   494,   544,   598,   658,
                                     724,   796,   876,   963,  1060,  1166,
                                    1282,  1411,  1552,  1707,  1878,  2066,
                                    2272,  2499,  2749,  3024,  3327,  3660,
                                    4026,  4428,  4871,  5358,  5894,  6484,
                                    7132,  7845,  8630,  9493, 10442, 11487,
                                   12635, 13899, 15289, 16818, 18500, 20350,
                                   22385, 24623, 27086, 29794, 32767 };

typedef struct wav_header {
    // RIFF chunk
    u8 riff_header[4] = { 'R', 'I', 'F', 'F' };
    u32 chunk_size = 0; // Data size + 36
    u8 wave_header[4] = { 'W', 'A', 'V', 'E' };
    // fmt sub chunk
    u8 fmt_header[4] = { 'f', 'm', 't', ' ' };
    u32 subchunk_1_size = 16;
    u16 audio_format = 1; // Audio format, 1=PCM
    u16 num_channels = 1; // Number of channels, 1=Mono
    u32 sample_rate = 16364; // Sampling Frequency in Hz
    u32 byte_rate = 32728; // sample rate * number of channels * bits per sample / 8
    u16 block_align = 2; // 2=16-bit mono
    u16 bits_per_sample = 16; // Bits per sample
    // data sub chunk
    u8 data_header[4] = { 'd', 'a', 't', 'a' };
    u32 subchunk_2_size = 0; // Data size in bytes (*2 for 16 bit)
} wav_hdr;