/***************************************************************************
 * Copyright (C) 2010
 * by Dimok
 *
 * This software is provided 'as-is', without any express or implied
 * warranty. In no event will the authors be held liable for any
 * damages arising from the use of this software.
 *
 * Permission is granted to anyone to use this software for any
 * purpose, including commercial applications, and to alter it and
 * redistribute it freely, subject to the following restrictions:
 *
 * 1. The origin of this software must not be misrepresented; you
 * must not claim that you wrote the original software. If you use
 * this software in a product, an acknowledgment in the product
 * documentation would be appreciated but is not required.
 *
 * 2. Altered source versions must be plainly marked as such, and
 * must not be misrepresented as being the original software.
 *
 * 3. This notice may not be removed or altered from any source
 * distribution.
 *
 * for WiiXplorer 2010
 ***************************************************************************/
#include <string.h>
#include <gui/sounds/WavDecoder.hpp>
#include "fs/CFile.hpp"
#include "utils/utils.h"

WavDecoder::WavDecoder(const char *filepath)
        : SoundDecoder(filepath) {
    SoundType = SOUND_WAV;
    SampleRate = 48000;
    Format = CHANNELS_STEREO | FORMAT_PCM_16_BIT;

    if (!file_fd) {
        return;
    }

    OpenFile();
}

WavDecoder::WavDecoder(const uint8_t *snd, int32_t len)
        : SoundDecoder(snd, len) {
    SoundType = SOUND_WAV;
    SampleRate = 48000;
    Format = CHANNELS_STEREO | FORMAT_PCM_16_BIT;

    if (!file_fd) {
        return;
    }

    OpenFile();
}

WavDecoder::~WavDecoder() {
}


void WavDecoder::OpenFile() {
    SWaveHdr Header;
    SWaveFmtChunk FmtChunk;
    memset(&Header, 0, sizeof(SWaveHdr));
    memset(&FmtChunk, 0, sizeof(SWaveFmtChunk));

    file_fd->read((uint8_t *) &Header, sizeof(SWaveHdr));
    file_fd->read((uint8_t *) &FmtChunk, sizeof(SWaveFmtChunk));

    if (Header.magicRIFF != 0x52494646) { // 'RIFF'
        CloseFile();
        return;
    } else if (Header.magicWAVE != 0x57415645) { // 'WAVE'
        CloseFile();
        return;
    } else if (FmtChunk.magicFMT != 0x666d7420) { // 'fmt '
        CloseFile();
        return;
    }

    DataOffset = sizeof(SWaveHdr) + le32(FmtChunk.size) + 8;
    file_fd->seek(DataOffset, SEEK_SET);
    SWaveChunk DataChunk;
    file_fd->read((uint8_t *) &DataChunk, sizeof(SWaveChunk));

    while (DataChunk.magicDATA != 0x64617461) { // 'data'
        DataOffset += 8 + le32(DataChunk.size);
        file_fd->seek(DataOffset, SEEK_SET);
        int32_t ret = file_fd->read((uint8_t *) &DataChunk, sizeof(SWaveChunk));
        if (ret <= 0) {
            CloseFile();
            return;
        }
    }

    DataOffset += 8;
    DataSize = le32(DataChunk.size);
    Is16Bit = (le16(FmtChunk.bps) == 16);
    SampleRate = le32(FmtChunk.freq);

    if (le16(FmtChunk.channels) == 1 && le16(FmtChunk.bps) == 8 && le16(FmtChunk.alignment) <= 1) {
        Format = CHANNELS_MONO | FORMAT_PCM_8_BIT;
    } else if (le16(FmtChunk.channels) == 1 && le16(FmtChunk.bps) == 16 && le16(FmtChunk.alignment) <= 2) {
        Format = CHANNELS_MONO | FORMAT_PCM_16_BIT;
    } else if (le16(FmtChunk.channels) == 2 && le16(FmtChunk.bps) == 8 && le16(FmtChunk.alignment) <= 2) {
        Format = CHANNELS_STEREO | FORMAT_PCM_8_BIT;
    } else if (le16(FmtChunk.channels) == 2 && le16(FmtChunk.bps) == 16 && le16(FmtChunk.alignment) <= 4) {
        Format = CHANNELS_STEREO | FORMAT_PCM_16_BIT;
    }
}

void WavDecoder::CloseFile() {
    if (file_fd) {
        delete file_fd;
    }

    file_fd = NULL;
}

int32_t WavDecoder::Read(uint8_t *buffer, int32_t buffer_size, int32_t pos) {
    if (!file_fd) {
        return -1;
    }

    if (CurPos >= (int32_t) DataSize) {
        return 0;
    }

    file_fd->seek(DataOffset + CurPos, SEEK_SET);

    if (buffer_size > (int32_t) DataSize - CurPos) {
        buffer_size = DataSize - CurPos;
    }

    int32_t read = file_fd->read(buffer, buffer_size);
    if (read > 0) {
        if (Is16Bit) {
            read &= ~0x0001;

            for (uint32_t i = 0; i < (uint32_t) (read / sizeof(uint16_t)); ++i) {
                ((uint16_t *) buffer)[i] = le16(((uint16_t *) buffer)[i]);
            }
        }
        CurPos += read;
    }

    return read;
}