#include "meta.h"
#include "../layout/layout.h"

/* THP - Nintendo movie format found in GC/Wii games */
VGMSTREAM* init_vgmstream_thp(STREAMFILE *streamFile) {
    VGMSTREAM *vgmstream = NULL;
    off_t start_offset, component_type_offset, component_data_offset;
    uint32_t version, max_audio_size;
    int num_components;
    int loop_flag, channel_count;
    int i;


    /* checks */
    /* .thp: actual extension
     * .dsp: fake extension?
     * (extensionless): Fragile (Wii) */
    if (!check_extensions(streamFile, "thp,dsp,"))
        goto fail;
    if (read_32bitBE(0x00,streamFile) != 0x54485000) /* "THP\0" */
        goto fail;

    version = read_32bitBE(0x04,streamFile); /* 16b+16b major/minor */
    /* 0x08: max buffer size */
    max_audio_size = read_32bitBE(0x0C,streamFile);
    /* 0x10: fps in float */
    /* 0x14: block count */
    /* 0x18: first block size */
    /* 0x1c: data size */

    if (version != 0x00010000 && version != 0x00011000) /* v1.0 (~2002) or v1.1 (rest) */
        goto fail;
    if (max_audio_size == 0) /* no sound */
        goto fail;

    component_type_offset = read_32bitBE(0x20,streamFile);
    /* 0x24: block offsets table offset (optional, for seeking) */
    start_offset = read_32bitBE(0x28,streamFile);
    /* 0x2c: last block offset */

    /* first component "type" x16 then component headers */
    num_components = read_32bitBE(component_type_offset,streamFile);
    component_type_offset += 0x04;
    component_data_offset = component_type_offset + 0x10;

    /* parse "component" (data that goes into blocks) */
    for (i = 0; i < num_components; i++) {
        int type = read_8bit(component_type_offset + i,streamFile);

        if (type == 0x00) { /* video */
            if (version == 0x00010000)
                component_data_offset += 0x08; /* width + height */
            else
                component_data_offset += 0x0c; /* width + height + format? */
        }
        else if (type == 0x01) { /* audio */
            /* parse below */
#if 0
            if (version == 0x00010000)
                component_data_offset += 0x0c; /* channels + sample rate + samples */
            else
                component_data_offset += 0x10; /* channels + sample rate + samples + format? */
#endif
            break;
        }
        else { /* 0xFF / no data (reserved as THP is meant to be extensible) */
            goto fail;
        }
    }

    /* official docs remark original's audio is adjusted to match GC's hardware rate
     * (48000 > 48043 / 32000 > 32028), not sure if that means ouput sample rate should
     * adjusted, but we can't detect Wii (non adjusted) .thp tho */

    loop_flag = 0;
    channel_count = read_32bitBE(component_data_offset + 0x00,streamFile);


    /* build the VGMSTREAM */
    vgmstream = allocate_vgmstream(channel_count,loop_flag);
    if (!vgmstream) goto fail;

    vgmstream->sample_rate = read_32bitBE(component_data_offset + 0x04,streamFile);
    vgmstream->num_samples = read_32bitBE(component_data_offset + 0x08,streamFile);

    vgmstream->meta_type = meta_THP;
    vgmstream->coding_type = coding_NGC_DSP;
    vgmstream->layout_type = layout_blocked_thp;
    /* coefs are in every block */

    vgmstream->full_block_size = read_32bitBE(0x18,streamFile); /* next block size */

    if (!vgmstream_open_stream(vgmstream,streamFile,start_offset))
        goto fail;
    return vgmstream;

fail:
    close_vgmstream(vgmstream);
    return NULL;
}