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

/* .XWC - Starbreeze games [Chronicles of Riddick: Assault on Dark Athena, Syndicate] */
VGMSTREAM * init_vgmstream_xwc(STREAMFILE *streamFile) {
    VGMSTREAM * vgmstream = NULL;
	off_t start_offset, extra_offset;
	size_t data_size;
    int loop_flag, channel_count, codec, num_samples;


    /* checks */
    /* .xwc: extension of the bigfile, individual files don't have one */
    if ( !check_extensions(streamFile,"xwc"))
        goto fail;


    /* version */
    if (read_32bitBE(0x00,streamFile) == 0x00030000 &&
        read_32bitBE(0x04,streamFile) == 0x00900000) { /* The Darkness */
        data_size = read_32bitLE(0x08, streamFile) + 0x1c; /* not including subheader */
        channel_count = read_32bitLE(0x0c, streamFile);
        /* 0x10: num_samples */
        /* 0x14: 0x8000? */
        /* 0x18: null */
        codec = read_32bitBE(0x1c, streamFile);
        num_samples = read_32bitLE(0x20, streamFile);
        /* 0x24: config data >> 2? (0x00(1): channels; 0x01(2): ?, 0x03(2): sample_rate) */
        extra_offset = 0x28;
    }
    else if (read_32bitBE(0x00,streamFile) == 0x00040000 &&
             read_32bitBE(0x04,streamFile) == 0x00900000) { /* Riddick, Syndicate */
        data_size = read_32bitLE(0x08, streamFile) + 0x24; /* not including subheader */
        channel_count = read_32bitLE(0x0c, streamFile);
        /* 0x10: num_samples */
        /* 0x14: 0x8000? */
        codec = read_32bitBE(0x24, streamFile);
        num_samples = read_32bitLE(0x28, streamFile);
        /* 0x2c: config data >> 2? (0x00(1): channels; 0x01(2): ?, 0x03(2): sample_rate) */
        /* 0x30+: codec dependant */
        extra_offset = 0x30;
    }
    else {
        goto fail;
    }

    loop_flag = 0;


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

    vgmstream->meta_type = meta_XWC;
    vgmstream->num_samples = num_samples;

    switch(codec) {
#ifdef VGM_USE_MPEG
        case 0x4D504547: { /* "MPEG" (PS3) */
            mpeg_custom_config cfg = {0};

            start_offset = 0x800;
            vgmstream->num_samples = read_32bitLE(extra_offset+0x00, streamFile); /* with encoder delay */ //todo improve
            cfg.data_size = read_32bitLE(extra_offset+0x04, streamFile); /* without padding */

            vgmstream->codec_data = init_mpeg_custom(streamFile, start_offset, &vgmstream->coding_type, vgmstream->channels, MPEG_STANDARD, &cfg);
            if (!vgmstream->codec_data) goto fail;
            vgmstream->layout_type = layout_none;

            vgmstream->sample_rate = mpeg_get_sample_rate(vgmstream->codec_data);
            break;
        }
#endif
#ifdef VGM_USE_FFMPEG
        case 0x584D4100: { /* "XMA\0" (X360) */
            uint8_t buf[0x100];
            int32_t bytes, seek_size, block_size, block_count, sample_rate, chunk_size;

            seek_size  = read_32bitLE(extra_offset + 0x00, streamFile);
            chunk_size = read_32bitLE(extra_offset + 0x04 + seek_size, streamFile);

            start_offset = extra_offset+ 0x04 + seek_size + chunk_size + 0x08;
            start_offset += (start_offset % 0x800) ? 0x800 - (start_offset % 0x800) : 0; /* padded */
            data_size = data_size - start_offset;

            if (chunk_size == 0x34) { /* new XMA2 */
                sample_rate = read_32bitLE(extra_offset+0x04+seek_size+0x08, streamFile);
                block_size  = read_32bitLE(extra_offset+0x04+seek_size+0x20, streamFile);
                block_count = data_size / block_size;
                /* others: standard RIFF XMA2 fmt? */
            }
            else if (chunk_size == 0x2c) { /* old XMA2 */
                sample_rate = read_32bitBE(extra_offset+0x04+seek_size+0x10, streamFile);
                block_size  = read_32bitBE(extra_offset+0x04+seek_size+0x1c, streamFile);
                block_count = read_32bitBE(extra_offset+0x04+seek_size+0x28, streamFile);
                /* others: scrambled RIFF fmt BE values */
            }
            else {
                goto fail;
            }

            bytes = ffmpeg_make_riff_xma2(buf,0x100, vgmstream->num_samples, data_size, vgmstream->channels, sample_rate, block_count, block_size);
            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;

            vgmstream->sample_rate = sample_rate;

            xma_fix_raw_samples(vgmstream, streamFile, start_offset,data_size, 0, 0,0); /* samples are ok, fix delay */
            break;
        }
#endif
#ifdef VGM_USE_VORBIS
        case 0x564F5242: { /* "VORB" (PC) */
            start_offset = 0x30;
            data_size = data_size - start_offset;

            vgmstream->codec_data = init_ogg_vorbis(streamFile, start_offset, data_size, NULL);
            if ( !vgmstream->codec_data ) goto fail;
            vgmstream->coding_type = coding_OGG_VORBIS;
            vgmstream->layout_type = layout_none;

            vgmstream->sample_rate = read_32bitLE(start_offset + 0x28, streamFile);
            break;
        }
#endif
        default:
            goto fail;
    }


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

fail:
    close_vgmstream(vgmstream);
    return NULL;
}