/****************************************************************************
 * libwiigui
 *
 * Tantric 2009
 *
 * gui_sound.cpp
 *
 * decoder modification by ardi 2009
 *
 * GUI class definitions
 ***************************************************************************/

#include "gui.h"
#include <unistd.h>
#include "gecko.h"

#include "gui_sound_decoder.h"


#define BUFFER_SIZE 8192


/***************************************************************
 *
 * D E C O D E R – L I S T
 *
 *
 ***************************************************************/

GuiSoundDecoder::DecoderListEntry *GuiSoundDecoder::DecoderList = NULL;
GuiSoundDecoder::DecoderListEntry &GuiSoundDecoder::RegisterDecoder(DecoderListEntry &Decoder, GuiSoundDecoderCreate fnc)
{
	if(Decoder.fnc != fnc)
	{
		Decoder.fnc = fnc;
		Decoder.next = DecoderList;
		DecoderList = &Decoder;
	}
	return Decoder;
}
GuiSoundDecoder *GuiSoundDecoder::GetDecoder(const u8 * snd, u32 len, bool snd_is_allocated)
{
	for(DecoderListEntry *de = DecoderList; de; de=de->next)
	{
		GuiSoundDecoder *d = NULL;
		try{ d = de->fnc(snd, len, snd_is_allocated); }
		catch(const char *error){ 
		gprintf("%s", error); }
		catch(...){}
		if(d) return d;
	}
	return NULL;
}

 
/***************************************************************
 *
 * D E C O D E R – T H R E A D
 *
 *
 ***************************************************************/

static GuiSound *GuiSoundPlayer[16] = {NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL};

static lwp_t GuiSoundDecoderThreadHandle	= LWP_THREAD_NULL;
static bool GuiSoundDecoderThreadRunning	= false;
static bool GuiSoundDecoderDataRquested	= false;

void *GuiSoundDecoderThread(void *args)
{
	GuiSoundDecoderThreadRunning = true;
	do
	{
		if(GuiSoundDecoderDataRquested)
		{
			GuiSoundDecoderDataRquested = false;
			GuiSound **players = GuiSoundPlayer;
			for( int i = 0; i < 16; ++i , ++players)
			{
				GuiSound *player = *players;
				if(player)
					player->DecoderCallback();
			}
		}
		if(!GuiSoundDecoderDataRquested)
			usleep(50);
	} while(GuiSoundDecoderThreadRunning);
	return 0;
}

/***************************************************************
 *
 * A S N D – C A L L B A C K
 *
 *
 ***************************************************************/

void GuiSoundPlayerCallback(s32 Voice)
{
	if(Voice >= 0 && Voice < 16 && GuiSoundPlayer[Voice])
	{
		GuiSoundPlayer[Voice]->PlayerCallback();
		GuiSoundDecoderDataRquested = true;
	}
}

/***************************************************************
 *
 *   R A W - D E C O D E R
 *   Decoder for Raw-PCM-Datas (16bit Stereo 48kHz)
 *
 ***************************************************************/
class GuiSoundDecoderRAW : public GuiSoundDecoder
{
protected:
	GuiSoundDecoderRAW(const u8 * snd, u32 len, bool snd_is_allocated)
	{
		pcm_start		= snd;
		is_allocated	= snd_is_allocated;
		pcm_end			= pcm_start+len;
		pos				= pcm_start;
		is_running		= false;

    }
public:
	~GuiSoundDecoderRAW()
	{
		while(is_running) usleep(50);
		if(is_allocated) delete [] pcm_start;
	}
	static GuiSoundDecoder *Create(const u8 * snd, u32 len, bool snd_is_allocated)
	{
		try { return new GuiSoundDecoderRAW(snd, len, snd_is_allocated); }
		catch(...) {}
		return NULL;
	}
	s32 GetFormat()
	{
		return VOICE_STEREO_16BIT;
	}
	s32 GetSampleRate()
	{
		return 48000;
	}
	/*  Read reads data from stream to buffer
		return:	>0 = readed bytes;
	            0 = EOF;
				<0 = Error;
	*/
	int Read(u8 * buffer, int buffer_size)
	{
		if(pos >= pcm_end)
			return 0; // EOF

		is_running = true;
		if(pos + buffer_size > pcm_end)
			buffer_size = pcm_end-pos;
		memcpy(buffer, pos, buffer_size);
		pos += buffer_size;
		is_running = false;
		return buffer_size;
	}
	int Rewind()
	{
		pos = pcm_start;
		return 0;
	}
private:
	const u8		*pcm_start;
	const u8		*pcm_end;
	bool			 is_allocated;
	const u8		*pos;
	bool			is_running;

};

/***************************************************************
 *
 *   G u i S o u n d
 *
 *
 ***************************************************************/
#define GuiSoundBufferReady	0x01
#define GuiSoundBufferEOF	0x02
#define GuiSoundFinish		0x04
static int GuiSoundCount = 0;
/**
 * Constructor for the GuiSound class.
 */
GuiSound::GuiSound(const u8 *s, int l, int v/*=100*/, bool r/*=true*/, bool a/*=false*/)
{
	if(GuiSoundCount++ == 0 || GuiSoundDecoderThreadHandle == LWP_THREAD_NULL)
	{
		LWP_CreateThread(&GuiSoundDecoderThreadHandle,GuiSoundDecoderThread,NULL,NULL,32*1024,80);
	}
	voice = -1;
	play_buffer[0]	= (u8*)memalign(32, BUFFER_SIZE*3);	// tripple-buffer first is played
	play_buffer[1]	= play_buffer[0] + BUFFER_SIZE;		//						second is waiting
	play_buffer[2]	= play_buffer[1] + BUFFER_SIZE;		//						third is decoding
	buffer_nr		= 0;			// current playbuffer
	buffer_pos		= 0;			// current idx to write in buffer
	buffer_ready	= false;
	buffer_eof		= false;
	loop			= false;		// play looped
	volume			= v;			// volume
	decoder			= NULL;
	if(play_buffer[0])			// playbuffer ok
		Load(s, l, r, a);
}
bool GuiSound::Load(const u8 *s, int l, bool r/*=false*/, bool a/*=false*/)
{
	Stop();
	if(!play_buffer[0]) return false;
	GuiSoundDecoder *newDecoder = GuiSoundDecoder::GetDecoder(s, l, a);
	if(!newDecoder && r) newDecoder = GuiSoundDecoderRAW::Create(s, l, a);
	if(newDecoder)
	{
		delete decoder;
		decoder = newDecoder;
		return true;
	}
	else if(a)
		delete [] s;
	return false;
}
GuiSound::GuiSound(const char *p, int v/*=100*/)
{
	if(GuiSoundCount++ == 0 || GuiSoundDecoderThreadHandle == LWP_THREAD_NULL)
	{
		LWP_CreateThread(&GuiSoundDecoderThreadHandle,GuiSoundDecoderThread,NULL,NULL,32*1024,80);
	}
	voice = -1;
	play_buffer[0]	= (u8*)memalign(32, BUFFER_SIZE*3);	// tripple-buffer first is played
	play_buffer[1]	= play_buffer[0] + BUFFER_SIZE;		//						second is waiting
	play_buffer[2]	= play_buffer[1] + BUFFER_SIZE;		//						third is decoding
	buffer_nr		= 0;					// current playbuffer
	buffer_pos		= 0;					// current idx to write in buffer
	buffer_ready	= false;
	buffer_eof		= false;
	loop			= false;				// play looped
	volume			= v;					// volume
	decoder			= NULL;
	if(play_buffer[0])			// playbuffer ok
		Load(p);
}
bool GuiSound::Load(const char *p)
{
	Stop();							// stop playing
	if(!play_buffer[0]) return false;
	
	bool ret = false;
	voice = -2;						// -2 marks loading from file
	u32 filesize = 0;
	u8 *buffer = NULL;
	size_t result;

	FILE * pFile = fopen (p, "rb");
	if(pFile)
	{
		// get file size:
		fseek (pFile , 0 , SEEK_END);
		filesize = ftell (pFile);
		fseek (pFile , 0 , SEEK_SET);

		// allocate memory to contain the whole file:
		buffer = new(std::nothrow) u8[filesize];
		if (buffer)
		{
			// copy the file into the buffer:
			result = fread (buffer, 1, filesize, pFile);
			if (result == filesize)
				ret= Load(buffer, filesize, false, true);
			else
				delete [] buffer;
		}
		fclose (pFile);
	}
	return ret;
}

/**
 * Destructor for the GuiSound class.
 */
GuiSound::~GuiSound()
{
	if(!loop)	while(voice >= 0) usleep(50);
	Stop();
	if(--GuiSoundCount == 0 && GuiSoundDecoderThreadHandle != LWP_THREAD_NULL)
	{
		GuiSoundDecoderThreadRunning = false;
		LWP_JoinThread(GuiSoundDecoderThreadHandle,NULL);
		GuiSoundDecoderThreadHandle = LWP_THREAD_NULL;
	}
	delete decoder;
	free(play_buffer[0]);
}

void GuiSound::Play()
{
	Stop();									// stop playing if it played
	if(!play_buffer[0]) return;
	if(!decoder) return;					// no decoder or no play_buffer -> no playing
	// initialize the buffer 
	buffer_nr		= 0; 					// allways starts with buffer 0					
	buffer_pos		= 0;					// reset position
	buffer_ready	= false;
	buffer_eof		= false;
	decoder->Rewind();						// play from begin
	DecoderCallback();						// fill first buffer;
	if(!buffer_ready || buffer_eof)			// if first buffer not ready -> no play
		return; 
	voice = ASND_GetFirstUnusedVoice();
	if(voice >= 0)
	{
		s32 vol			= (255*volume)/100;
		s32 format		= decoder->GetFormat();
		s32 samplerate	= decoder->GetSampleRate();
		s32 first_pos	= buffer_pos;
		// switch to next buffer
		buffer_nr		= 1;
		buffer_pos		= 0;
		buffer_ready	= false;
		buffer_eof		= false;
		DecoderCallback();						// fill second buffer;
		GuiSoundPlayer[voice] = this;			// activate Callbacks for this voice
		// Play the voice
		ASND_SetVoice(voice, format, samplerate, 0, play_buffer[0], first_pos, vol, vol, GuiSoundPlayerCallback);
	}
}
/*
int GuiSound::PlayOggFile(char * path)
{
	if(Load(path))
		Play();
	return 1;
}
*/
void GuiSound::Stop()
{
	if(voice < 0) return ;
	GuiSoundPlayer[voice] = NULL; // disable Callbacks
	SND_StopVoice(voice);
	voice = -1;
}

void GuiSound::Pause()
{
	if(voice < 0) return ;
	ASND_PauseVoice(voice, 1);
}

void GuiSound::Resume()
{
	if(voice < 0) return ;
	ASND_PauseVoice(voice, 0);
}

bool GuiSound::IsPlaying()
{
	return voice >= 0;
}

void GuiSound::SetVolume(int vol)
{
	volume = vol;
	if(voice < 0) return ;
	int newvol = 255*(volume/100.0);
	ASND_ChangeVolumeVoice(voice, newvol, newvol);
}

void GuiSound::SetLoop(bool l)
{
	loop = l;
}

void GuiSound::DecoderCallback()
{
	if(buffer_ready || buffer_eof) // if buffer ready or EOF -> nothing
		return;
	bool error=false;
	while(buffer_pos < BUFFER_SIZE)
	{
		int ret = decoder->Read(&play_buffer[buffer_nr][buffer_pos], BUFFER_SIZE-buffer_pos);
		if(ret > 0)
			buffer_pos += ret;					// ok -> fill the buffer more
		else if(ret == 0)						// EOF from decoder
		{
			if(loop)
				decoder->Rewind();				// if loop -> rewind and fill the buffer more
			else if(buffer_pos)	
				break;							// has data in buffer -> play the buffer
			else
				buffer_eof = true;				// no data in buffer -> return EOF
				return;
		}
		else if(ret < 0)						// an ERROR
		{
			if(buffer_pos)
				break;							// has data in buffer -> play the buffer
			else if(loop)
			{
				if(!error)						// if no prev error
				{
					decoder->Rewind();			// if loop -> rewind
					error = true;				// set error-state
					continue;					// and fill the buffer more
				}
				buffer_eof = true;				// has prev error -> error in first block -> return EOF
				return;
			}
			else
			{
				buffer_eof = true;				// no loop -> return EOF
				return;
			}
		}
		error = false;							// clear error-state
	}
	buffer_ready = true;
}
void GuiSound::PlayerCallback()
{
	if(buffer_eof)								// if EOF 
	{
		if(ASND_TestPointer(voice, play_buffer[(buffer_nr+2)%3])==0)	// test prev. Buffer
			Stop();
	}
	else if(buffer_ready)				// if buffer ready
	{
		if(ASND_AddVoice(voice, play_buffer[buffer_nr], buffer_pos)==SND_OK)	// add buffer
		{
			// next buffer 
			buffer_nr	= (buffer_nr+1)%3;
			buffer_pos	= 0;
			buffer_ready= false;
			buffer_eof	= false;
		}
	}
}