/*
   client/soundext.cc
   This file is part of the Osirion project and is distributed under
   the terms of the GNU General Public License version 2
*/


#include "audio/audio.h"
#include "audio/buffers.h"
#include "audio/sources.h"
#include "auxiliary/functions.h"
#include "core/gameinterface.h"
#include "core/entity.h"
#include "client/soundext.h"
#include "client/client.h"
#include "render/camera.h"

#include <sstream>
#include <iomanip>

namespace client
{

// cvars are initialized in client.cc
core::Cvar *snd_doppler = 0;
core::Cvar *snd_engines = 0;
core::Cvar *snd_volume = 0;

float master_volume = 0;
void render_listener_sound()
{
	if (!(snd_engines && snd_engines->value()))
		return;

	math::Vector3f velocity(0, 0 , 0);
	if (core::localcontrol()) {
		velocity.assign(core::localcontrol()->axis().forward() * core::localcontrol()->speed());
	}

	if (master_volume != snd_volume->value()) {
		master_volume = snd_volume->value();
		math::clamp(master_volume, 0, 1);
		(*snd_volume) = master_volume;
	}
	
	audio::update_listener(render::Camera::eye(), render::Camera::axis(), velocity, master_volume);
}

void render_entity_sound(core::Entity *entity)
{
	if (!(snd_engines && snd_engines->value())) {
		if (ext_sound(entity))
			delete ext_sound(entity);
		return;
	}

	if (!ext_render(entity) || (ext_render(entity) && !ext_render(entity)->visible())) {
		if (ext_sound(entity))
			delete ext_sound(entity);
		return;
	} else {
		if (!ext_sound(entity)) {
			new SoundExt(entity);
		}

		ext_sound(entity)->frame(0.0f);
	}
}

SoundExt::SoundExt(core::Entity *entity) : core::Extension(core::Extension::Sound, entity)
{
	state_thusterloopbuffer = 0;

	state_impulseloopbuffer = 0;
	state_impulsestartbuffer = 0;
	state_impulsestopbuffer = 0;

	state_engineloopbuffer = 0;
	state_engineloopsource = 0;

	state_engineeventbuffer = 0;
	state_engineeventsource = 0;

	// load engine sound
	if (entity->type() == core::Entity::Controlable) {
		core::EntityControlable *entityco = static_cast<core::EntityControlable *>(entity);
		unsigned int enginesoundset = 0;
		unsigned int impulsesoundset = 0;

		if (entityco->model()) {
			enginesoundset = entityco->model()->enginesound();
			impulsesoundset = entityco->model()->impulsesound();
		}


		std::stringstream soundname;
		soundname << "engines/loop" << std::setfill('0') << std::setw(2) << enginesoundset;
		state_thusterloopbuffer = audio::Buffers::load(soundname.str());

		// load impulse sound
		// FIXME load impulse sound set
		state_impulseloopbuffer = audio::Buffers::load("engines/impulse_loop00");
		state_impulsestartbuffer = audio::Buffers::load("engines/impulse_start00");
		state_impulsestopbuffer = audio::Buffers::load("engines/impulse_stop00");

		state_engineloopsource = audio::Sources::get();
		state_engineeventsource = audio::Sources::get();
	}
}

SoundExt::~SoundExt()
{
	clear();
}

void SoundExt::clear()
{
	if (state_engineloopsource) {
		audio::Sources::remove(state_engineloopsource);
	}

	if (state_engineeventsource) {
		audio::Sources::remove(state_engineeventsource);
	}

	state_thusterloopbuffer = 0;
	state_impulseloopbuffer = 0;
	state_impulsestartbuffer = 0;
	state_impulsestopbuffer = 0;

	state_engineloopbuffer = 0;
	state_engineloopsource = 0;

	state_engineeventbuffer = 0;
	state_engineeventsource = 0;
}

void SoundExt::frame(float elapsed)
{
	if (entity()->type() == core::Entity::Controlable) {
		core::EntityControlable *entity = static_cast<core::EntityControlable *>(this->entity());

		float speed = entity->speed();
		float pitch = 1.0f;
		float gain = 0.0;
		float r = (entity->model() ? entity->model()->maxbbox().x() : entity->radius());

		if (entity->state() == core::Entity::Impulse) {
			pitch = 1.0f;
			gain = 1.0f;
		} else if (entity->thrust() > 0) {
			pitch = 0.2f + entity->thrust() * 0.8f;
			gain = 1.0f;
			if (entity->thrust() < 0.2f) {
				gain *= entity->thrust() * 5.0f;
			}
		}

		if (entity->state() == core::Entity::ImpulseInitiate) {

			if (state_engineeventbuffer != state_impulsestartbuffer) {
				audio::update_source(state_engineeventsource,
						     entity->location() - entity->axis().forward() * r ,
						     entity->axis().forward() * speed);

				state_engineeventbuffer = audio::play(state_engineeventsource, state_impulsestartbuffer);
			}
		} else if (entity->state() == core::Entity::Impulse) {

			state_engineeventbuffer = state_impulseloopbuffer;

			if (state_engineloopbuffer != state_impulseloopbuffer) {
				state_engineloopbuffer = audio::loop(state_engineloopsource, state_impulseloopbuffer, pitch, 0);
			}
			pitch = 1.0f;
		} else {

			if (state_engineeventbuffer == state_impulseloopbuffer) {
				audio::update_source(state_engineeventsource,
						     entity->location() - entity->axis().forward() * r ,
						     entity->axis().forward() * speed);
				state_engineeventbuffer = audio::play(state_engineeventsource, state_impulsestopbuffer);
			}
			state_engineeventbuffer = 0;

			if (state_engineloopbuffer != state_thusterloopbuffer) {
				state_engineloopbuffer = audio::loop(state_engineloopsource, state_thusterloopbuffer, pitch, 0);
			}
		}

		audio::update_source(state_engineloopsource,
				     entity->location() - entity->axis().forward() * r , entity->axis().forward() * speed, pitch, gain);

		audio::update_source(state_engineeventsource,
				     entity->location() - entity->axis().forward() * r , entity->axis().forward() * speed);
	}
}

} // namespace client