/**

 * vim: set ts=4 :

 * =============================================================================

 * SourceMod SDKTools Extension

 * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.

 * =============================================================================

 *

 * This program is free software; you can redistribute it and/or modify it under

 * the terms of the GNU General Public License, version 3.0, as published by the

 * Free Software Foundation.

 * 

 * This program 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

 * this program.  If not, see <http://www.gnu.org/licenses/>.

 *

 * As a special exception, AlliedModders LLC gives you permission to link the

 * code of this program (as well as its derivative works) to "Half-Life 2," the

 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

 * by the Valve Corporation.  You must obey the GNU General Public License in

 * all respects for all other code used.  Additionally, AlliedModders LLC grants

 * this exception to all derivative works.  AlliedModders LLC defines further

 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

 * or <http://www.sourcemod.net/license.php>.

 *

 * Version: $Id$

 */

#include "vsound.h"

#include <IForwardSys.h>

SH_DECL_HOOK8_void(IVEngineServer, EmitAmbientSound, SH_NOATTRIB, 0, int, const Vector &, const char *, float, soundlevel_t, int, int, float);

#if SOURCE_ENGINE >= SE_PORTAL2

SH_DECL_HOOK17(IEngineSound, EmitSound, SH_NOATTRIB, 0, int, IRecipientFilter &, int, int, const char *, unsigned int, const char *, float, float, int, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int);

SH_DECL_HOOK17(IEngineSound, EmitSound, SH_NOATTRIB, 1, int, IRecipientFilter &, int, int, const char *, unsigned int, const char *, float, soundlevel_t, int, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int);

#elif SOURCE_ENGINE == SE_ORANGEBOXVALVE || SOURCE_ENGINE == SE_CSS

SH_DECL_HOOK15_void(IEngineSound, EmitSound, SH_NOATTRIB, 0, IRecipientFilter &, int, int, const char *, float, float, int, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int);

SH_DECL_HOOK15_void(IEngineSound, EmitSound, SH_NOATTRIB, 1, IRecipientFilter &, int, int, const char *, float, soundlevel_t, int, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int);

#else

SH_DECL_HOOK14_void(IEngineSound, EmitSound, SH_NOATTRIB, 0, IRecipientFilter &, int, int, const char *, float, float, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int);

SH_DECL_HOOK14_void(IEngineSound, EmitSound, SH_NOATTRIB, 1, IRecipientFilter &, int, int, const char *, float, soundlevel_t, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int);

#endif

bool g_InSoundHook = false;

/***************************

*                          *

* Sound Related Hook Class *

*                          *

****************************/

cell_t SoundReferenceToIndex(cell_t ref)

{

	if (ref == 0 || ref == -1 || ref == -2)

	{

		return ref;

	}

	return gamehelpers->ReferenceToIndex(ref);

}

size_t SoundHooks::_FillInPlayers(int *pl_array, IRecipientFilter *pFilter)

{

	size_t size = static_cast<size_t>(pFilter->GetRecipientCount());

	for (size_t i=0; i<size; i++)

	{

		int index;

#if SOURCE_ENGINE == SE_DOTA

		pFilter->GetRecipientIndex(&index, i);

#else

		index = pFilter->GetRecipientIndex(i);

#endif

		pl_array[i] = index;

	}

	return size;

}

void SoundHooks::_IncRefCounter(int type)

{

	if (type == NORMAL_SOUND_HOOK)

	{

		if (m_NormalCount++ == 0)

		{

			SH_ADD_HOOK(IEngineSound, EmitSound, engsound, SH_MEMBER(this, &SoundHooks::OnEmitSound), false);

			SH_ADD_HOOK(IEngineSound, EmitSound, engsound, SH_MEMBER(this, &SoundHooks::OnEmitSound2), false);

		}

	}

	else if (type == AMBIENT_SOUND_HOOK)

	{

		if (m_AmbientCount++ == 0)

		{

			SH_ADD_HOOK(IVEngineServer, EmitAmbientSound, engine, SH_MEMBER(this, &SoundHooks::OnEmitAmbientSound), false);

		}

	}

}

void SoundHooks::_DecRefCounter(int type)

{

	if (type == NORMAL_SOUND_HOOK)

	{

		if (--m_NormalCount == 0)

		{

			SH_REMOVE_HOOK(IEngineSound, EmitSound, engsound, SH_MEMBER(this, &SoundHooks::OnEmitSound), false);

			SH_REMOVE_HOOK(IEngineSound, EmitSound, engsound, SH_MEMBER(this, &SoundHooks::OnEmitSound2), false);

		}

	}

	else if (type == AMBIENT_SOUND_HOOK)

	{

		if (--m_AmbientCount == 0)

		{

			SH_REMOVE_HOOK(IVEngineServer, EmitAmbientSound, engine, SH_MEMBER(this, &SoundHooks::OnEmitAmbientSound), false);

		}

	}

}

void SoundHooks::Initialize()

{

	plsys->AddPluginsListener(this);

}

void SoundHooks::Shutdown()

{

	plsys->RemovePluginsListener(this);

	if (m_NormalCount)

	{

		SH_REMOVE_HOOK(IEngineSound, EmitSound, engsound, SH_MEMBER(this, &SoundHooks::OnEmitSound), false);

		SH_REMOVE_HOOK(IEngineSound, EmitSound, engsound, SH_MEMBER(this, &SoundHooks::OnEmitSound2), false);

	}

	if (m_AmbientCount)

	{

		SH_REMOVE_HOOK(IVEngineServer, EmitAmbientSound, engine, SH_MEMBER(this, &SoundHooks::OnEmitAmbientSound), false);

	}

}

void SoundHooks::OnPluginUnloaded(IPlugin *plugin)

{

	SoundHookIter iter;

	IPluginContext *pContext = plugin->GetBaseContext();

	if (m_AmbientCount)

	{

		for (iter=m_AmbientFuncs.begin(); iter!=m_AmbientFuncs.end(); )

		{

			if ((*iter)->GetParentContext() == pContext)

			{

				iter = m_AmbientFuncs.erase(iter);

				_DecRefCounter(AMBIENT_SOUND_HOOK);

			}

			else

			{

				iter++;

			}

		}

	}

	if (m_NormalCount)

	{

		for (iter=m_NormalFuncs.begin(); iter!=m_NormalFuncs.end(); )

		{

			if ((*iter)->GetParentContext() == pContext)

			{

				iter = m_NormalFuncs.erase(iter);

				_DecRefCounter(NORMAL_SOUND_HOOK);

			}

			else

			{

				iter++;

			}

		}

	}

}

void SoundHooks::AddHook(int type, IPluginFunction *pFunc)

{

	if (type == NORMAL_SOUND_HOOK)

	{

		m_NormalFuncs.push_back(pFunc);

		_IncRefCounter(NORMAL_SOUND_HOOK);

	}

	else if (type == AMBIENT_SOUND_HOOK)

	{

		m_AmbientFuncs.push_back(pFunc);

		_IncRefCounter(AMBIENT_SOUND_HOOK);

	}

}

bool SoundHooks::RemoveHook(int type, IPluginFunction *pFunc)

{

	SoundHookIter iter;

	if (type == NORMAL_SOUND_HOOK)

	{

		if ((iter=m_NormalFuncs.find(pFunc)) != m_NormalFuncs.end())

		{

			m_NormalFuncs.erase(iter);

			_DecRefCounter(NORMAL_SOUND_HOOK);

			return true;

		}

		else

		{

			return false;

		}

	}

	else if (type == AMBIENT_SOUND_HOOK)

	{

		if ((iter=m_AmbientFuncs.find(pFunc)) != m_AmbientFuncs.end())

		{

			m_AmbientFuncs.erase(iter);

			_DecRefCounter(AMBIENT_SOUND_HOOK);

			return true;

		}

		else

		{

			return false;

		}

	}

	return false;

}

void SoundHooks::OnEmitAmbientSound(int entindex, const Vector &pos, const char *samp, float vol, 

									soundlevel_t soundlevel, int fFlags, int pitch, float delay)

{

	SoundHookIter iter;

	IPluginFunction *pFunc;

	cell_t vec[3] = {sp_ftoc(pos.x), sp_ftoc(pos.y), sp_ftoc(pos.z)};

	cell_t res = static_cast<ResultType>(Pl_Continue);

	char buffer[PLATFORM_MAX_PATH];

	strcpy(buffer, samp);

	for (iter=m_AmbientFuncs.begin(); iter!=m_AmbientFuncs.end(); iter++)

	{

		pFunc = (*iter);

		pFunc->PushStringEx(buffer, sizeof(buffer), SM_PARAM_STRING_COPY, SM_PARAM_COPYBACK);

		pFunc->PushCellByRef(&entindex);

		pFunc->PushFloatByRef(&vol);

		pFunc->PushCellByRef(reinterpret_cast<cell_t *>(&soundlevel));

		pFunc->PushCellByRef(&pitch);

		pFunc->PushArray(vec, 3, SM_PARAM_COPYBACK);

		pFunc->PushCellByRef(&fFlags);

		pFunc->PushFloatByRef(&delay);

		g_InSoundHook = true;

		pFunc->Execute(&res);

		g_InSoundHook = false;

		switch (res)

		{

		case Pl_Handled:

		case Pl_Stop:

			{

				RETURN_META(MRES_SUPERCEDE);

			}

		case Pl_Changed:

			{

				Vector vec2;

				vec2.x = sp_ctof(vec[0]);

				vec2.y = sp_ctof(vec[1]);

				vec2.z = sp_ctof(vec[2]);

				RETURN_META_NEWPARAMS(MRES_IGNORED, &IVEngineServer::EmitAmbientSound, 

										(entindex, vec2, buffer, vol, soundlevel, fFlags, pitch, delay));

			}

		}

	}

}

#if SOURCE_ENGINE >= SE_PORTAL2

int SoundHooks::OnEmitSound(IRecipientFilter &filter, int iEntIndex, int iChannel, const char *pSoundEntry, unsigned int nSoundEntryHash, const char *pSample, 

							 float flVolume, soundlevel_t iSoundlevel, int nSeed, int iFlags, int iPitch, const Vector *pOrigin, 

							 const Vector *pDirection, CUtlVector<Vector> *pUtlVecOrigins, bool bUpdatePositions, 

							 float soundtime, int speakerentity)

#elif SOURCE_ENGINE == SE_ORANGEBOXVALVE || SOURCE_ENGINE == SE_CSS

void SoundHooks::OnEmitSound(IRecipientFilter &filter, int iEntIndex, int iChannel, const char *pSample, 

							 float flVolume, soundlevel_t iSoundlevel, int iFlags, int iPitch, int iSpecialDSP, const Vector *pOrigin, 

							 const Vector *pDirection, CUtlVector<Vector> *pUtlVecOrigins, bool bUpdatePositions, 

							 float soundtime, int speakerentity)

#else

void SoundHooks::OnEmitSound(IRecipientFilter &filter, int iEntIndex, int iChannel, const char *pSample, 

							 float flVolume, soundlevel_t iSoundlevel, int iFlags, int iPitch, const Vector *pOrigin, 

							 const Vector *pDirection, CUtlVector<Vector> *pUtlVecOrigins, bool bUpdatePositions, 

							 float soundtime, int speakerentity)

#endif

{

	SoundHookIter iter;

	IPluginFunction *pFunc;

	cell_t res = static_cast<ResultType>(Pl_Continue);

	char buffer[PLATFORM_MAX_PATH];

	strcpy(buffer, pSample);

	for (iter=m_NormalFuncs.begin(); iter!=m_NormalFuncs.end(); iter++)

	{

		int players[64], size;

		size = _FillInPlayers(players, &filter);

		pFunc = (*iter);

		pFunc->PushArray(players, 64, SM_PARAM_COPYBACK);

		pFunc->PushCellByRef(&size);

		pFunc->PushStringEx(buffer, sizeof(buffer), SM_PARAM_STRING_COPY, SM_PARAM_COPYBACK);

		pFunc->PushCellByRef(&iEntIndex);

		pFunc->PushCellByRef(&iChannel);

		pFunc->PushFloatByRef(&flVolume);

		pFunc->PushCellByRef(reinterpret_cast<cell_t *>(&iSoundlevel));

		pFunc->PushCellByRef(&iPitch);

		pFunc->PushCellByRef(&iFlags);

		g_InSoundHook = true;

		pFunc->Execute(&res);

		g_InSoundHook = false;

		switch (res)

		{

		case Pl_Handled:

		case Pl_Stop:

			{

#if SOURCE_ENGINE >= SE_PORTAL2

				RETURN_META_VALUE(MRES_SUPERCEDE, -1);

#else

				RETURN_META(MRES_SUPERCEDE);

#endif

			}

		case Pl_Changed:

			{

				CellRecipientFilter crf;

				crf.Initialize(players, size);

#if SOURCE_ENGINE >= SE_PORTAL2

				RETURN_META_VALUE_NEWPARAMS(

					MRES_IGNORED,

					-1,

					static_cast<int (IEngineSound::*)(IRecipientFilter &, int, int, const char*, unsigned int, const char*, float, soundlevel_t, 

					int, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int)>(&IEngineSound::EmitSound), 

					(crf, iEntIndex, iChannel, buffer, -1, buffer, flVolume, iSoundlevel, nSeed, iFlags, iPitch, pOrigin, 

					pDirection, pUtlVecOrigins, bUpdatePositions, soundtime, speakerentity)

					);

#elif SOURCE_ENGINE == SE_ORANGEBOXVALVE || SOURCE_ENGINE == SE_CSS

				RETURN_META_NEWPARAMS(

					MRES_IGNORED,

					static_cast<void (IEngineSound::*)(IRecipientFilter &, int, int, const char*, float, soundlevel_t, 

					int, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int)>(&IEngineSound::EmitSound), 

					(crf, iEntIndex, iChannel, buffer, flVolume, iSoundlevel, iFlags, iPitch, iSpecialDSP, pOrigin, 

					pDirection, pUtlVecOrigins, bUpdatePositions, soundtime, speakerentity)

					);

#else

				RETURN_META_NEWPARAMS(

					MRES_IGNORED,

					static_cast<void (IEngineSound::*)(IRecipientFilter &, int, int, const char*, float, soundlevel_t, 

					int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int)>(&IEngineSound::EmitSound), 

					(crf, iEntIndex, iChannel, buffer, flVolume, iSoundlevel, iFlags, iPitch, pOrigin, 

					pDirection, pUtlVecOrigins, bUpdatePositions, soundtime, speakerentity)

					);

#endif

			}

		}

	}

#if SOURCE_ENGINE >= SE_PORTAL2

	RETURN_META_VALUE(MRES_IGNORED, -1 );

#endif

}

#if SOURCE_ENGINE >= SE_PORTAL2

int SoundHooks::OnEmitSound2(IRecipientFilter &filter, int iEntIndex, int iChannel, const char *pSoundEntry, unsigned int nSoundEntryHash, const char *pSample, 

							 float flVolume, float flAttenuation, int nSeed, int iFlags, int iPitch, const Vector *pOrigin, 

							 const Vector *pDirection, CUtlVector<Vector> *pUtlVecOrigins, bool bUpdatePositions, 

							 float soundtime, int speakerentity)

#elif SOURCE_ENGINE == SE_ORANGEBOXVALVE || SOURCE_ENGINE == SE_CSS

void SoundHooks::OnEmitSound2(IRecipientFilter &filter, int iEntIndex, int iChannel, const char *pSample, 

							 float flVolume, float flAttenuation, int iFlags, int iPitch, int iSpecialDSP, const Vector *pOrigin, 

							 const Vector *pDirection, CUtlVector<Vector> *pUtlVecOrigins, bool bUpdatePositions, 

							 float soundtime, int speakerentity)

#else

void SoundHooks::OnEmitSound2(IRecipientFilter &filter, int iEntIndex, int iChannel, const char *pSample, 

							 float flVolume, float flAttenuation, int iFlags, int iPitch, const Vector *pOrigin, 

							 const Vector *pDirection, CUtlVector<Vector> *pUtlVecOrigins, bool bUpdatePositions, 

							 float soundtime, int speakerentity)

#endif

{

	SoundHookIter iter;

	IPluginFunction *pFunc;

	cell_t res = static_cast<ResultType>(Pl_Continue);

	cell_t sndlevel = static_cast<cell_t>(ATTN_TO_SNDLVL(flAttenuation));

	char buffer[PLATFORM_MAX_PATH];

	strcpy(buffer, pSample);

	for (iter=m_NormalFuncs.begin(); iter!=m_NormalFuncs.end(); iter++)

	{

		int players[64], size;

		size = _FillInPlayers(players, &filter);

		pFunc = (*iter);

		pFunc->PushArray(players, 64, SM_PARAM_COPYBACK);

		pFunc->PushCellByRef(&size);

		pFunc->PushStringEx(buffer, sizeof(buffer), SM_PARAM_STRING_COPY, SM_PARAM_COPYBACK);

		pFunc->PushCellByRef(&iEntIndex);

		pFunc->PushCellByRef(&iChannel);

		pFunc->PushFloatByRef(&flVolume);

		pFunc->PushCellByRef(&sndlevel);

		pFunc->PushCellByRef(&iPitch);

		pFunc->PushCellByRef(&iFlags);

		g_InSoundHook = true;

		pFunc->Execute(&res);

		g_InSoundHook = false;

		switch (res)

		{

		case Pl_Handled:

		case Pl_Stop:

			{

#if SOURCE_ENGINE >= SE_PORTAL2

				RETURN_META_VALUE(MRES_SUPERCEDE, -1);

#else

				RETURN_META(MRES_SUPERCEDE);

#endif

			}

		case Pl_Changed:

			{

				CellRecipientFilter crf;

				crf.Initialize(players, size);

#if SOURCE_ENGINE >= SE_PORTAL2

				RETURN_META_VALUE_NEWPARAMS(

					MRES_IGNORED,

					-1,

					static_cast<int (IEngineSound::*)(IRecipientFilter &, int, int, const char *, unsigned int, const char *, float, float, 

					int, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int)>(&IEngineSound::EmitSound), 

					(crf, iEntIndex, iChannel, buffer, -1, buffer, flVolume, SNDLVL_TO_ATTN(static_cast<soundlevel_t>(sndlevel)), 

					nSeed, iFlags, iPitch, pOrigin, pDirection, pUtlVecOrigins, bUpdatePositions, soundtime, speakerentity)

					);

#elif SOURCE_ENGINE == SE_ORANGEBOXVALVE || SOURCE_ENGINE == SE_CSS

RETURN_META_NEWPARAMS(

					MRES_IGNORED,

					static_cast<void (IEngineSound::*)(IRecipientFilter &, int, int, const char *, float, float, 

					int, int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int)>(&IEngineSound::EmitSound), 

					(crf, iEntIndex, iChannel, buffer, flVolume, SNDLVL_TO_ATTN(static_cast<soundlevel_t>(sndlevel)), 

					iFlags, iPitch, iSpecialDSP, pOrigin, pDirection, pUtlVecOrigins, bUpdatePositions, soundtime, speakerentity)

					);

#else

				RETURN_META_NEWPARAMS(

					MRES_IGNORED,

					static_cast<void (IEngineSound::*)(IRecipientFilter &, int, int, const char *, float, float, 

					int, int, const Vector *, const Vector *, CUtlVector<Vector> *, bool, float, int)>(&IEngineSound::EmitSound), 

					(crf, iEntIndex, iChannel, buffer, flVolume, SNDLVL_TO_ATTN(static_cast<soundlevel_t>(sndlevel)), 

					iFlags, iPitch, pOrigin, pDirection, pUtlVecOrigins, bUpdatePositions, soundtime, speakerentity)

					);

#endif

			}

		}

	}

#if SOURCE_ENGINE >= SE_PORTAL2

	RETURN_META_VALUE(MRES_IGNORED, -1 );

#endif

}

/************************

*                       *

* Sound Related Natives *

*                       *

*************************/

SoundHooks s_SoundHooks;

static cell_t PrefetchSound(IPluginContext *pContext, const cell_t *params)

{

	char *name;

	pContext->LocalToString(params[1], &name);

	engsound->PrefetchSound(name);

	return 1;

}

static cell_t GetSoundDuration(IPluginContext *pContext, const cell_t *params)

{

	char *name;

	pContext->LocalToString(params[1], &name);

	return sp_ftoc(engsound->GetSoundDuration(name));

}