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

static STREAMFILE* setup_lyn_streamfile(STREAMFILE *streamFile, off_t subfile_offset, size_t subfile_size);

/* LyN RIFF - from Ubisoft LyN engine games [Red Steel 2 (Wii), Adventures of Tintin (Multi), From Dust (Multi), Just Dance 3/4 (multi)] */
VGMSTREAM * init_vgmstream_ubi_lyn(STREAMFILE *streamFile) {
    VGMSTREAM * vgmstream = NULL;
    off_t start_offset, first_offset = 0xc;
    off_t fmt_offset, data_offset, fact_offset;
    size_t fmt_size, data_size, fact_size;
    int loop_flag, channel_count, sample_rate, codec;
    int num_samples;


    /* checks */
    /* .sns: Red Steel 2, .wav: Tintin, .son: From Dust */
    if (!check_extensions(streamFile,"sns,wav,lwav,son"))
        goto fail;

    /* a slightly eccentric RIFF with custom codecs */
    if (read_32bitBE(0x00,streamFile) != 0x52494646 ||  /* "RIFF" */
        read_32bitBE(0x08,streamFile) != 0x57415645)    /* "WAVE" */
        goto fail;
    if (read_32bitLE(0x04,streamFile)+0x04+0x04 != get_streamfile_size(streamFile))
        goto fail;

    if (!find_chunk(streamFile, 0x666d7420,first_offset,0, &fmt_offset,&fmt_size, 0, 0))   /* "fmt " */
        goto fail;
    if (!find_chunk(streamFile, 0x64617461,first_offset,0, &data_offset,&data_size, 0, 0)) /* "data" */
        goto fail;

    /* always found, even with PCM (LyN subchunk seems to contain the engine version, ex. 0x0d/10) */
    if (!find_chunk(streamFile, 0x66616374,first_offset,0, &fact_offset,&fact_size, 0, 0)) /* "fact" */
        goto fail;
    if (fact_size != 0x10 || read_32bitBE(fact_offset+0x04, streamFile) != 0x4C794E20) /* "LyN " */
        goto fail;
    num_samples = read_32bitLE(fact_offset+0x00, streamFile);
    /* sometimes there is a LySE chunk */


    /* parse format */
    {
        if (fmt_size < 0x12)
            goto fail;
        codec = (uint16_t)read_16bitLE(fmt_offset+0x00,streamFile);
        channel_count   = read_16bitLE(fmt_offset+0x02,streamFile);
        sample_rate     = read_32bitLE(fmt_offset+0x04,streamFile);
        /* 0x08: average bytes, 0x0c: block align, 0x0e: bps, etc */

        /* fake WAVEFORMATEX, used with > 2ch */
        if (codec == 0xFFFE) {
            if (fmt_size < 0x28)
                goto fail;
            /* fake GUID with first value doubling as codec */
            codec = read_32bitLE(fmt_offset+0x18,streamFile);
            if (read_32bitBE(fmt_offset+0x1c,streamFile) != 0x00001000 &&
                read_32bitBE(fmt_offset+0x20,streamFile) != 0x800000AA &&
                read_32bitBE(fmt_offset+0x24,streamFile) != 0x00389B71) {
                goto fail;
            }
        }
    }

    /* most songs simply repeat, loop if it looks long enough */
    loop_flag = (num_samples > 20*sample_rate); /* in seconds */
    start_offset = data_offset;


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

    vgmstream->sample_rate = sample_rate;
    vgmstream->meta_type = meta_UBI_LYN;
    vgmstream->num_samples = num_samples;
    vgmstream->loop_start_sample = 0;
    vgmstream->loop_end_sample = vgmstream->num_samples;

    switch(codec) {
        case 0x0001: /* PCM */
            vgmstream->coding_type = coding_PCM16LE; /* LE even in X360 */
            vgmstream->layout_type = layout_interleave;
            vgmstream->interleave_block_size = 0x02;
            break;

        case 0x5050: /* DSP (Wii) */
            vgmstream->coding_type = coding_NGC_DSP;
            vgmstream->layout_type = layout_interleave;
            vgmstream->interleave_block_size = 0x08;

            /* setup default Ubisoft coefs */
            {
                static const int16_t coef[16] = {
                        0x04ab,0xfced,0x0789,0xfedf,0x09a2,0xfae5,0x0c90,0xfac1,
                        0x084d,0xfaa4,0x0982,0xfdf7,0x0af6,0xfafa,0x0be6,0xfbf5
                };
                int i, ch;

                for (ch = 0; ch < channel_count; ch++) {
                    for (i = 0; i < 16; i++) {
                        vgmstream->ch[ch].adpcm_coef[i] = coef[i];
                    }
                }
            }

            break;

#ifdef VGM_USE_VORBIS
        case 0x3157: { /* Ogg (PC), interleaved 1ch */
            size_t interleave_size, stride_size;
            layered_layout_data* data = NULL;
            int i;

            if (read_32bitLE(start_offset+0x00,streamFile) != 1) /* id? */
               goto fail;

            interleave_size = read_32bitLE(start_offset+0x04,streamFile);
            stride_size = interleave_size * channel_count;
            /* interleave is adjusted so there is no smaller last block, it seems */

            vgmstream->coding_type = coding_OGG_VORBIS;
            vgmstream->layout_type = layout_layered;

            /* init layout */
            data = init_layout_layered(channel_count);
            if (!data) goto fail;
            vgmstream->layout_data = data;

            /* open each layer subfile */
            for (i = 0; i < channel_count; i++) {
                STREAMFILE* temp_streamFile = NULL;
                size_t total_size = read_32bitLE(start_offset+0x08 + 0x04*i,streamFile);
                off_t layer_offset = start_offset+0x08 + 0x04*channel_count + interleave_size*i;

                temp_streamFile = setup_lyn_ogg_streamfile(streamFile, layer_offset, interleave_size, stride_size, total_size);
                if (!temp_streamFile) goto fail;

                data->layers[i] = init_vgmstream_ogg_vorbis(temp_streamFile);
                close_streamfile(temp_streamFile);
                if (!data->layers[i]) goto fail;

                /* could validate between layers, meh */
            }

            /* setup layered VGMSTREAMs */
            if (!setup_layout_layered(data))
                goto fail;

            break;
        }
#endif

#ifdef VGM_USE_MPEG
        case 0x5051: { /* MPEG (PS3/PC), interleaved 1ch */
            mpeg_codec_data *mpeg_data = NULL;
            mpeg_custom_config cfg = {0};
            int i;

            cfg.interleave = read_32bitLE(start_offset+0x04,streamFile);
            cfg.chunk_size = read_32bitLE(start_offset+0x08,streamFile); /* frame size (not counting MPEG padding?) */
            /* 0x00: id? (2=Tintin, 3=Michael Jackson), 0x0c: frame per interleave, 0x10: samples per frame */

            /* skip seek tables and find actual start */
            start_offset += 0x14;
            data_size -= 0x14;
            for (i = 0; i < channel_count; i++) {
                int entries = read_32bitLE(start_offset,streamFile);

                start_offset += 0x04 + entries*0x08;
                data_size -= 0x04 + entries*0x08;
            }

            cfg.data_size = data_size;

            //todo data parsing looks correct but some files decode a bit wrong at the end (ex. Tintin: Music~Boss~Allan~Victory~02)
            mpeg_data = init_mpeg_custom(streamFile, start_offset, &vgmstream->coding_type, vgmstream->channels, MPEG_LYN, &cfg);
            if (!mpeg_data) goto fail;
            vgmstream->codec_data = mpeg_data;
            vgmstream->layout_type = layout_none;

            break;
        }
#endif

#ifdef VGM_USE_FFMPEG
        case 0x0166: { /* XMA (X360), standard */
            uint8_t buf[0x100];
            int bytes;
            off_t chunk_offset;
            size_t chunk_size, seek_size;

            if (read_32bitLE(start_offset+0x00,streamFile) != 3) /* id? */
               goto fail;

            /* skip standard XMA header + seek table */
            chunk_offset = start_offset + 0x04 + 0x04;
            chunk_size = read_32bitLE(start_offset + 0x04, streamFile);
            seek_size = read_32bitLE(chunk_offset+chunk_size, streamFile);
            start_offset += (0x04 + 0x04 + chunk_size + 0x04 + seek_size);
            data_size    -= (0x04 + 0x04 + chunk_size + 0x04 + seek_size);

            bytes = ffmpeg_make_riff_xma_from_fmt_chunk(buf,0x100, chunk_offset,chunk_size, data_size, streamFile, 1);
            vgmstream->codec_data = init_ffmpeg_header_offset(streamFile, buf,bytes, start_offset,data_size);
            if ( !vgmstream->codec_data ) goto fail;
            vgmstream->coding_type = coding_FFmpeg;
            vgmstream->layout_type = layout_none;

            break;
        }
#endif

        default:
            goto fail;
    }


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

fail:
    close_vgmstream(vgmstream);
    return NULL;
}


/* LyN RIFF in containers */
VGMSTREAM * init_vgmstream_ubi_lyn_container(STREAMFILE *streamFile) {
    VGMSTREAM *vgmstream = NULL;
    STREAMFILE *temp_streamFile = NULL;
    off_t subfile_offset;
    size_t subfile_size;

    /* LyN packs files in bigfiles, and once extracted the sound files have extra engine
     * data before the RIFF. Might as well support them in case the RIFF wasn't extracted. */

    /* checks */
    if (!check_extensions(streamFile,"sns,wav,lwav,son"))
        goto fail;

    /* find "RIFF" position */
    if (read_32bitBE(0x00,streamFile) == 0x4C795345 && /* "LySE" */
        read_32bitBE(0x14,streamFile) == 0x52494646) { /* "RIFF" */
        subfile_offset = 0x14; /* Adventures of Tintin */
    }
    else if (read_32bitLE(0x00,streamFile)+0x20 == get_streamfile_size(streamFile) &&
             read_32bitBE(0x20,streamFile) == 0x52494646) { /* "RIFF" */
        subfile_offset = 0x20; /* Red Steel 2, From Dust */
    }
    else {
        goto fail;
    }
    
    subfile_size = read_32bitLE(subfile_offset+0x04,streamFile) + 0x04+0x04;

    temp_streamFile = setup_lyn_streamfile(streamFile, subfile_offset,subfile_size);
    if (!temp_streamFile) goto fail;

    vgmstream = init_vgmstream_ubi_lyn(temp_streamFile);
    
    close_streamfile(temp_streamFile);
    return vgmstream;

fail:
    close_streamfile(temp_streamFile);
    close_vgmstream(vgmstream);
    return NULL;
}

static STREAMFILE* setup_lyn_streamfile(STREAMFILE *streamFile, off_t subfile_offset, size_t subfile_size) {
    STREAMFILE *temp_streamFile = NULL, *new_streamFile = NULL;

    /* setup subfile */
    new_streamFile = open_wrap_streamfile(streamFile);
    if (!new_streamFile) goto fail;
    temp_streamFile = new_streamFile;

    new_streamFile = open_clamp_streamfile(temp_streamFile, subfile_offset,subfile_size);
    if (!new_streamFile) goto fail;
    temp_streamFile = new_streamFile;

    return temp_streamFile;

fail:
    close_streamfile(temp_streamFile);
    return NULL;
}