#include "meta.h"
#include "../coding/coding.h"

/* SPSD - Naomi (arcade) streams [Guilty Gear X (Naomi), Crazy Taxi (Naomi), Virtua Tennis 2 (Naomi)] */
VGMSTREAM * init_vgmstream_naomi_spsd(STREAMFILE *streamFile) {
    VGMSTREAM * vgmstream = NULL;
    off_t start_offset;
    size_t data_size;
    int loop_flag, channel_count, codec, flags, index;


    /* checks */
    /* .spsd: header id */
    if (!check_extensions(streamFile, "spsd"))
        goto fail;
    if (read_32bitBE(0x00,streamFile) != 0x53505344) /* "SPSD" */
        goto fail;

    if (read_32bitBE(0x04,streamFile) != 0x01010004 && /* standard version */
        read_32bitBE(0x04,streamFile) != 0x00010004)   /* uncommon version [Crazy Taxi (Naomi)] */
        goto fail;


    codec = read_8bit(0x08,streamFile);
    flags = read_8bit(0x09,streamFile);
    index = read_16bitLE(0x0a,streamFile);
    data_size = read_32bitLE(0x0c,streamFile);
    //if (data_size + start_offset != get_streamfile_size(streamFile))
    //    goto fail; /* some rips out there have incorrect padding */

    //todo with 0x80 seems 0x2c is a loop_start_sample but must be adjusted to +1 block? (uncommon flag though)
    loop_flag = (flags & 0x80);
    channel_count = ((flags & 0x01) || (flags & 0x02)) ? 2 : 1; /* 0x02 is rare but looks normal (Virtua Tennis 2) */
    start_offset = 0x40;


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

    vgmstream->sample_rate = (uint16_t)read_16bitLE(0x2A,streamFile);

    vgmstream->meta_type = meta_NAOMI_SPSD;
    switch (codec) {
        case 0x00: /* [Virtua Tennis 2 (Naomi), Club Kart - European Session (Naomi)] */
            vgmstream->coding_type = coding_PCM16LE;
            vgmstream->num_samples = pcm_bytes_to_samples(data_size,channel_count,16);
            vgmstream->loop_start_sample = read_32bitLE(0x2c,streamFile) + pcm_bytes_to_samples(0x2000*channel_count,channel_count,16);
            vgmstream->loop_end_sample = vgmstream->num_samples;
            break;

        case 0x01: /* [Virtua Tennis 2 (Naomi)] */
            vgmstream->coding_type = coding_PCM8;
            vgmstream->num_samples = pcm_bytes_to_samples(data_size,channel_count,8);
            vgmstream->loop_start_sample = read_32bitLE(0x2c,streamFile) + pcm_bytes_to_samples(0x2000*channel_count,channel_count,8);
            vgmstream->loop_end_sample = vgmstream->num_samples;

            break;

        case 0x03: /* standard */
            vgmstream->coding_type = coding_AICA_int;
            vgmstream->num_samples = aica_bytes_to_samples(data_size,channel_count);
            vgmstream->loop_start_sample = /*read_32bitLE(0x2c,streamFile) +*/ aica_bytes_to_samples(0x2000*channel_count,channel_count);
            vgmstream->loop_end_sample = vgmstream->num_samples;
            break;

        default:
            goto fail;
    }

    /* interleave index, maybe */
    switch(index) {
        case 0x0000:
            if (channel_count != 1) goto fail;
            vgmstream->layout_type = layout_none;
            break;

        case 0x000d:
            vgmstream->layout_type = layout_interleave;
            vgmstream->interleave_block_size = 0x2000;
            if (vgmstream->interleave_block_size)
                vgmstream->interleave_last_block_size = (data_size % (vgmstream->interleave_block_size*vgmstream->channels)) / vgmstream->channels;
            break;

        case 0x00ff:
            vgmstream->layout_type = layout_interleave;
            vgmstream->interleave_block_size = data_size / channel_count;
            break;

    default:
        goto fail;
    }

    /* todo seems to decode slightly incorrectly in after certain data (loop section start?)
     *  may depend on values in 0x20 or 0x2c [ex. Marvel vs Capcom 2 (Naomi)]
     *  at 0x30(4*ch) is some config per channel but doesn't seem to affect ADPCM (found with PCM too) */
    {
        int i;
        for (i = 0; i < channel_count; i++) {
            vgmstream->ch[i].adpcm_step_index = 0x7f;
        }
    }


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

fail:
    close_vgmstream(vgmstream);
    return NULL;
}