/****************************************************************************
 * libwiigui
 *
 * Tantric 2009
 *
 * gui_sound_plugin_wav.cpp
 *
 * by ardi 2009
 *
 * Decoder for WAVE PCM
 *
 * GUI class definitions
 ***************************************************************************/

#include <asndlib.h>
#include <map>
#include <vector>
#include <unistd.h>

#include "gui_sound_decoder.h"

typedef struct
{
    u32 cueID;
    u32 len;
    u32 loops;
} plst_t;
typedef struct
{
    const u8    *start;
    const u8    *end;
    u32         loops;
} playlist_t;

class GuiSoundDecoderWAV : public GuiSoundDecoder
{
    protected:
        GuiSoundDecoderWAV( const u8 * snd, u32 len, bool snd_is_allocated )
        {
            sound           = snd;
            is_running      = false;
            is_allocated    = snd_is_allocated;

            const u8 *in_ptr = sound;

            if ( be32inc( in_ptr ) != 0x52494646 /*'RIFF' (WAV)*/ ) throw( "Not a WAV" );

            u32 riffsize = le32inc( in_ptr );
            if ( riffsize > ( len - 8 ) ) throw( "Wrong size" );

            if ( be32inc( in_ptr ) != 0x57415645 /*'WAVE'*/ ) throw( "No  WAVE-Tag" );

            if ( be32inc( in_ptr ) != 0x666D7420 /*'fmt '*/ ) throw( "No fmt-Tag" );

            u32 fmtLen = le32inc( in_ptr );

            if ( le16inc( in_ptr ) != 1 ) throw( "Not PCM data" );

            channelCount    = le16inc( in_ptr );
            if ( channelCount < 1 || channelCount > 2 ) throw( "only mono or stereo" );

            sampleRate      = le32inc( in_ptr );

            in_ptr += 6;                    // skip <bytes/second> and <block align>

            bytePerSample   = ( le16inc( in_ptr ) + 7 ) / 8;
            if ( bytePerSample < 1 || bytePerSample > 2 ) throw( "only 1-16 bit/Sample" );

            in_ptr += fmtLen - 16;

            if ( be32inc( in_ptr ) != 0x64617461 /*'data'*/ ) throw( "No data-Tag" );


            soundDataStart = in_ptr + 4;
            soundDataEnd = soundDataStart + le32( in_ptr );

            in_ptr = soundDataEnd;

            std::map<u32, u32> cue;
            std::vector<plst_t>plst;

            if ( ( ( u32 )in_ptr ) & 0x0001UL ) in_ptr++;
            while ( ( in_ptr + 4 ) < ( sound + riffsize ) )
            {
                u32 tag = be32inc( in_ptr );
                switch ( tag )
                {
                    case 0x63756520 /*'cue '*/:
                        in_ptr += 4;                    // skip size
                        for ( u32 count = le32inc( in_ptr ); count > 0; count-- )
                        {
                            u32 ID = be32inc( in_ptr );
                            in_ptr += 4;                // skip dwPosition
                            if ( be32inc( in_ptr ) == 0x64617461 /*'data'*/ )
                            {
                                in_ptr += 8;            // skip chunkStart - dwBlockStart
                                cue[ID] = le32inc( in_ptr );
                            }
                            else
                                in_ptr += 12;           // skip chunkStart - SammpleOffset
                        }
                        break;
                    case 0x706C7374 /*' plst'*/:
                        in_ptr += 4;                    // skip size
                        for ( u32 count = le32inc( in_ptr ); count > 0; count-- )
                            plst.push_back( ( plst_t ) {le32inc( in_ptr ), le32inc( in_ptr ), le32inc( in_ptr )} );
                        break;
                    default:
                        in_ptr -= 2;
                        break;
                }
            }
            for ( std::vector<plst_t>::iterator i = plst.begin(); i != plst.end(); ++i )
            {
                const u8 *start = soundDataStart + cue[i->cueID];
                const u8 *end   = soundDataStart + ( i->len * bytePerSample * channelCount );
                u32 loops       = i->loops;
                playlist.push_back( ( playlist_t ) {start, end, loops} );
            }
            if ( playlist.size() == 0 )
            {
                playlist.push_back( ( playlist_t ) {soundDataStart, soundDataEnd, 1} );
            }
            Rewind();
        }
    public:
        ~GuiSoundDecoderWAV()
        {
            while ( is_running ) usleep( 50 );
            if ( is_allocated ) delete [] sound;
        }
        static GuiSoundDecoder *Create( const u8 * snd, u32 len, bool snd_is_allocated )
        {
            if ( snd && len > 4 && snd[0] == 'R' && snd[1] == 'I' && snd[2] == 'F' && snd[3] == 'F'
                    && snd[8] == 'W' && snd[9] == 'A' && snd[10] == 'V' && snd[11] == 'E' )
                return new GuiSoundDecoderWAV( snd, len, snd_is_allocated );
            return NULL;
        }
        s32 GetFormat()
        {
            if ( bytePerSample == 2 )
                return channelCount == 2 ? VOICE_STEREO_16BIT : VOICE_MONO_16BIT;
            else
                return channelCount == 2 ? VOICE_STEREO_8BIT : VOICE_MONO_8BIT;
        }
        s32 GetSampleRate()
        {
            return sampleRate;
        }
        /*  Read reads data from stream to buffer
            return: >0 = readed bytes;
                    0 = EOF;
                    <0 = Error;
        */
        int Read( u8 * buffer, int buffer_size )
        {
            is_running = true;
            u8 *write_pos = buffer;
            u8 *write_end = buffer + buffer_size;

            for ( ;; )
            {
                while ( currentPos < currentEnd )
                {
                    if ( write_pos >= write_end )
                    {
                        is_running = false;
                        return write_pos - buffer;
                    }
                    if ( bytePerSample == 2 )
                    {
                        *( ( s16* )write_pos ) = le16inc( currentPos );
                        write_pos += 2;
                        if ( channelCount == 2 ) // stereo
                        {
                            *( ( s16* )write_pos ) = le16inc( currentPos );
                            write_pos += 2;
                        }
                    }
                    else
                    {
                        *write_pos++ = *currentPos++;
                        if ( channelCount == 2 ) // stereo
                            *write_pos++ = *currentPos++;
                    }
                }
                if ( currentLoops > 1 )
                {
                    currentLoops--;
                    currentPos = currentStart;
                    continue;
                }
                if ( currentPlaylist != playlist.end() )
                    currentPlaylist++;
                if ( currentPlaylist != playlist.end() )
                {
                    currentStart    = currentPos = currentPlaylist->start;
                    currentEnd      = currentPlaylist->end;
                    currentLoops    = currentPlaylist->loops;
                    continue;
                }
                else
                {
                    is_running = false;
                    return write_pos - buffer;
                }
            }
            is_running = false;
            return 0;
        }
        int Rewind()
        {
            currentPlaylist = playlist.begin();
            currentStart    = currentPos = currentPlaylist->start;
            currentEnd      = currentPlaylist->end;
            currentLoops    = currentPlaylist->loops;
            return 0;
        }
    private:
        const u8        *sound;
        bool            is_allocated;
        bool            is_running;

        u16             channelCount;
        u32             sampleRate;
        u16             bytePerSample;
        const u8        *soundDataStart;
        const u8        *soundDataEnd;
        std::vector<playlist_t> playlist;
        std::vector<playlist_t>::iterator currentPlaylist;
        const u8        *currentStart;
        const u8        *currentEnd;
        u32             currentLoops;
        const u8        *currentPos;

};
REGISTER_GUI_SOUND_DECODER( GuiSoundDecoderWAV );