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

/* .AST - from Nintendo games [Super Mario Galaxy (Wii), Pac-Man Vs (GC)] */
VGMSTREAM * init_vgmstream_ast(STREAMFILE *streamFile) {
    VGMSTREAM * vgmstream = NULL;
    off_t start_offset;
    int loop_flag, channel_count, codec;


    /* checks */
    if (!check_extensions(streamFile, "ast"))
        goto fail;
    if (read_32bitBE(0x00,streamFile) != 0x5354524D) /* "STRM" */
        goto fail;
    if (read_16bitBE(0x0a,streamFile) != 0x10) /* ? */
        goto fail;

    if (read_32bitBE(0x04,streamFile)+0x40 != get_streamfile_size(streamFile))
        goto fail;
    codec         = read_16bitBE(0x08,streamFile);
    channel_count = read_16bitBE(0x0c,streamFile);
    loop_flag     = read_16bitBE(0x0e,streamFile);
    //max_block   = read_32bitBE(0x20,streamFile);

    start_offset = 0x40;
    if (read_32bitBE(start_offset,streamFile) != 0x424C434B)  /* "BLCK" */
        goto fail;


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

    vgmstream->meta_type = meta_AST;
    vgmstream->sample_rate = read_32bitBE(0x10,streamFile);
    vgmstream->num_samples = read_32bitBE(0x14,streamFile);
    vgmstream->loop_start_sample = read_32bitBE(0x18,streamFile);
    vgmstream->loop_end_sample = read_32bitBE(0x1c,streamFile);

    vgmstream->layout_type = layout_blocked_ast;
    switch (codec) {
        case 0x00: /* , Pikmin 2 (GC) */
            vgmstream->coding_type = coding_NGC_AFC;
            break;
        case 0x01: /* Mario Kart: Double Dash!! (GC) */
            vgmstream->coding_type = coding_PCM16BE;
            break;
        default:
            goto fail;
    }

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

fail:
    close_vgmstream(vgmstream);
    return NULL;
}