/* Copyright (C) 2012 Doubango Telecom <http://www.doubango.org>
 *	
 * This file is part of Open Source Doubango Framework.
 *
 * DOUBANGO is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *	
 * DOUBANGO is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *	
 * You should have received a copy of the GNU General Public License
 * along with DOUBANGO.
 */
 #include <tinydav.h>
 #include "audio_opensles_consumer.h"
 #include "audio_opensles.h"
 
 #include "tinydav/audio/tdav_consumer_audio.h"
 
 #include "tsk_string.h"
 #include "tsk_string.h"
 #include "tsk_memory.h"
 #include "tsk_debug.h"
 
 typedef struct audio_consumer_opensles_s
 {
 	TDAV_DECLARE_CONSUMER_AUDIO;
 	audio_opensles_instance_handle_t* audioInstHandle;
 	bool isSpeakerOn;
 	struct{
 		void* ptr;
 		bool isFull;
 		int size;
 		int index;
 	} buffer;
 }
 audio_consumer_opensles_t;
 
 int audio_consumer_opensles_get_data_10ms(const audio_consumer_opensles_t* _self, void* audioSamples, int nSamples, int nBytesPerSample, int nChannels, int samplesPerSec, uint32_t &nSamplesOut)
 {
 	nSamplesOut = 0;
 	if(!_self || !audioSamples || !nSamples){
 		AUDIO_OPENSLES_DEBUG_ERROR("Invalid parameter");
 		return -1;
 	}
 	if((nSamples != (samplesPerSec / 100))){
 		AUDIO_OPENSLES_DEBUG_ERROR("Not producing 10ms samples (nSamples=%d, samplesPerSec=%d)", nSamples, samplesPerSec);
 		return -2;
 	}
 	if((nBytesPerSample != (TMEDIA_CONSUMER(_self)->audio.bits_per_sample >> 3))){
 		AUDIO_OPENSLES_DEBUG_ERROR("%d not valid bytes/samples", nBytesPerSample);
 		return -3;
 	}
 	if((nChannels != TMEDIA_CONSUMER(_self)->audio.out.channels)){
 		AUDIO_OPENSLES_DEBUG_ERROR("Playout - %d not the expected number of channels but should be %d", nChannels, TMEDIA_CONSUMER(_self)->audio.out.channels);
 		return -4;
 	}
 
 	audio_consumer_opensles_t* self = const_cast<audio_consumer_opensles_t*>(_self);
 
 	if(self->buffer.index == self->buffer.size){
         //self->buffer.ptr= tsk_realloc(self->buffer.ptr, (self->buffer.size + 2));
 		if((tdav_consumer_audio_get(TDAV_CONSUMER_AUDIO(self), self->buffer.ptr, self->buffer.size)) != self->buffer.size){
 			nSamplesOut = 0;
 			self->buffer.index = self->buffer.size;
 			return 0;
 		}
         self->buffer.index = 0;
 		tdav_consumer_audio_tick(TDAV_CONSUMER_AUDIO(self));
 	}
 	
 	int nSamplesInBits = (nSamples * nBytesPerSample);
 	if(_self->buffer.index + nSamplesInBits <= _self->buffer.size){
 		memcpy(audioSamples, (((uint8_t*)self->buffer.ptr) + self->buffer.index), nSamplesInBits);
 	}
 	self->buffer.index += nSamplesInBits;
 	TSK_CLAMP(0, self->buffer.index, self->buffer.size);
 	nSamplesOut = nSamples;
 	
 	return 0;
 }
 
 bool audio_consumer_opensles_is_speakerOn(const audio_consumer_opensles_t* self)
 {
 	if(!self){
 		AUDIO_OPENSLES_DEBUG_ERROR("Invalid parameter");
 		return false;
 	}
 	return self->isSpeakerOn;
 }
 
 
 /* ============ Media Consumer Interface ================= */
 static int audio_consumer_opensles_set(tmedia_consumer_t* _self, const tmedia_param_t* param)
 {
 	audio_consumer_opensles_t* self = (audio_consumer_opensles_t*)_self;
 	int ret = tdav_consumer_audio_set(TDAV_CONSUMER_AUDIO(self), param);
     int32_t* valueInt;
 	if(ret == 0){
 		if(tsk_striequals(param->key, "volume")){
 			return audio_opensles_instance_set_speaker_volume(self->audioInstHandle, *((int32_t*)param->value));
 		}
 		else if(tsk_striequals(param->key, "speaker-on")){
 			self->isSpeakerOn = (TSK_TO_INT32((uint8_t*)param->value) != 0);
 			if(self->audioInstHandle){
 				return audio_opensles_instance_set_speakerOn(self->audioInstHandle, self->isSpeakerOn);
 			}
 			else return 0; // will be set when instance is initialized
 		}
 	}
 
 	return ret;
 }
 
 static int audio_consumer_opensles_prepare(tmedia_consumer_t* _self, const tmedia_codec_t* codec)
 {
 	audio_consumer_opensles_t* self = (audio_consumer_opensles_t*)_self;
 	if(!self){
 		AUDIO_OPENSLES_DEBUG_ERROR("Invalid parameter");
 		return -1;
 	}
 
 	// create audio instance
 	if(!(self->audioInstHandle = audio_opensles_instance_create(TMEDIA_CONSUMER(self)->session_id))){
 		AUDIO_OPENSLES_DEBUG_ERROR("Failed to create audio instance handle");
 		return -1;
 	}
 
 	// initialize input parameters from the codec information
 	TMEDIA_CONSUMER(self)->audio.ptime = TMEDIA_CODEC_PTIME_AUDIO_DECODING(codec);
 	TMEDIA_CONSUMER(self)->audio.in.channels = TMEDIA_CODEC_CHANNELS_AUDIO_DECODING(codec);
 	TMEDIA_CONSUMER(self)->audio.in.rate = TMEDIA_CODEC_RATE_DECODING(codec);
 
 	AUDIO_OPENSLES_DEBUG_INFO("audio_consumer_opensles_prepare(channels=%d, rate=%d, ptime=%d)", codec->plugin->audio.channels, codec->plugin->rate, codec->plugin->audio.ptime);
 	
 	// prepare playout device and update output parameters
 	int ret = audio_opensles_instance_prepare_consumer(self->audioInstHandle, &_self);
 
 	// now that the producer is prepared we can initialize internal buffer using device caps
 	if(ret == 0){
 		// allocate buffer
 		int xsize = ((TMEDIA_CONSUMER(self)->audio.ptime * TMEDIA_CONSUMER(self)->audio.out.rate) / 1000) * (TMEDIA_CONSUMER(self)->audio.bits_per_sample >> 3);
 		if(!(self->buffer.ptr = tsk_realloc(self->buffer.ptr, xsize))){
 			AUDIO_OPENSLES_DEBUG_ERROR("Failed to allocate buffer with size = %d", xsize);
 			self->buffer.size = 0;
 			return -1;
 		}
 		memset(self->buffer.ptr, 0, xsize);
 		self->buffer.size = xsize;
 		self->buffer.index = 0;
 		self->buffer.isFull = false;
 	}
 	return ret;
 }
 
 static int audio_consumer_opensles_start(tmedia_consumer_t* _self)
 {
 	audio_consumer_opensles_t* self = (audio_consumer_opensles_t*)_self;
 	if(!self){
 		AUDIO_OPENSLES_DEBUG_ERROR("Invalid parameter");
 		return -1;
 	}
 
 	return audio_opensles_instance_start_consumer(self->audioInstHandle);
 }
 
 static int audio_consumer_opensles_consume(tmedia_consumer_t* _self, const void* data, tsk_size_t data_size, const tsk_object_t* proto_hdr)
 {
 	audio_consumer_opensles_t* self = (audio_consumer_opensles_t*)_self;
 	if(!self || !data || !data_size){
 		AUDIO_OPENSLES_DEBUG_ERROR("1Invalid parameter");
 		return -1;
 	}	
 	/* buffer is already decoded */
 	return tdav_consumer_audio_put(TDAV_CONSUMER_AUDIO(self), data,
 								   data_size, proto_hdr);
 }
 
 static int audio_consumer_opensles_pause(tmedia_consumer_t* self)
 {
 	return 0;
 }
 
 static int audio_consumer_opensles_stop(tmedia_consumer_t* _self)
 {
 	audio_consumer_opensles_t* self = (audio_consumer_opensles_t*)_self;
 	if(!self){
 		AUDIO_OPENSLES_DEBUG_ERROR("Invalid parameter");
 		return -1;
 	}
 
 	return audio_opensles_instance_stop_consumer(self->audioInstHandle);
 }
 
 
 //
 //	SLES audio consumer object definition
 //
 /* constructor */
 static tsk_object_t* audio_consumer_opensles_ctor(tsk_object_t *_self, va_list * app)
 {
 	audio_consumer_opensles_t *self = (audio_consumer_opensles_t *)_self;
 	if(self){
 		/* init base */
 		tdav_consumer_audio_init(TDAV_CONSUMER_AUDIO(self));
 		/* init self */
 		
 	}
 	return self;
 }
 /* destructor */
 static tsk_object_t* audio_consumer_opensles_dtor(tsk_object_t *_self)
 { 
 	audio_consumer_opensles_t *self = (audio_consumer_opensles_t *)_self;
 	if(self){
 		/* stop */
 		audio_consumer_opensles_stop(TMEDIA_CONSUMER(self));
 		/* deinit self */
 		if(self->audioInstHandle){
 			audio_opensles_instance_destroy(&self->audioInstHandle);
 		}
 		TSK_FREE(self->buffer.ptr);
 		/* deinit base */
 		tdav_consumer_audio_deinit(TDAV_CONSUMER_AUDIO(self));
 	}
 
 	return self;
 }
 /* object definition */
 static const tsk_object_def_t audio_consumer_opensles_def_s = 
 {
 	sizeof(audio_consumer_opensles_t),
 	audio_consumer_opensles_ctor, 
 	audio_consumer_opensles_dtor,
 	tdav_consumer_audio_cmp, 
 };
 /* plugin definition*/
 static const tmedia_consumer_plugin_def_t audio_consumer_opensles_plugin_def_s = 
 {
 	&audio_consumer_opensles_def_s,
 	
 	tmedia_audio,
 	"SLES audio consumer",
 	
 	audio_consumer_opensles_set,
 	audio_consumer_opensles_prepare,
 	audio_consumer_opensles_start,
 	audio_consumer_opensles_consume,
 	audio_consumer_opensles_pause,
 	audio_consumer_opensles_stop
 };
 const tmedia_consumer_plugin_def_t *audio_consumer_opensles_plugin_def_t = &audio_consumer_opensles_plugin_def_s;