/**

  * vim: set ts=4 :

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

  * SourceMod

  * 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 "sm_globals.h"

 #include <IGameConfigs.h>

 #include "UserMessages.h"

 #include "TimerSys.h"

 #include "PlayerManager.h"

 #include "logic_bridge.h"

 #include "sourcemod.h"

 #if SOURCE_ENGINE == SE_DOTA

 #include <game/shared/protobuf/usermessages.pb.h>

 #elif SOURCE_ENGINE == SE_CSGO

 #include <game/shared/csgo/protobuf/cstrike15_usermessages.pb.h>

 #endif

 #define MAX_HUD_CHANNELS		6

 int g_HudMsgNum = -1;

 struct hud_syncobj_t

 {

 	int player_channels[256 + 1];

 };

 struct player_chaninfo_t

 {

 	double chan_times[MAX_HUD_CHANNELS];

 	hud_syncobj_t *chan_syncobjs[MAX_HUD_CHANNELS];

 };

 struct hud_text_parms

 {

 	float       x;

 	float       y;

 	int         effect;

 	byte        r1, g1, b1, a1;

 	byte        r2, g2, b2, a2;

 	float       fadeinTime;

 	float       fadeoutTime;

 	float       holdTime;

 	float       fxTime;

 	int         channel;

 };

 class HudMsgHelpers : 

 	public SMGlobalClass,

 	public IHandleTypeDispatch,

 	public IClientListener

 {

 public:

 	bool IsSupported()

 	{

 		return (g_HudMsgNum != -1);

 	}

 	virtual void OnSourceModAllInitialized_Post()

 	{

 		const char *key;

 		key = g_pGameConf->GetKeyValue("HudTextMsg");

 		if (key != NULL)

 		{

 			g_HudMsgNum = g_UserMsgs.GetMessageIndex(key);

 		}

 		if (!IsSupported())

 		{

 			m_hHudSyncObj = 0;

 			m_PlayerHuds = NULL;

 			return;

 		}

 		m_PlayerHuds = new player_chaninfo_t[256+1];

 		m_hHudSyncObj = handlesys->CreateType("HudSyncObj", this, 0, NULL, NULL, g_pCoreIdent, NULL);

 		g_Players.AddClientListener(this);

 	}

 	virtual void OnSourceModShutdown()

 	{

 		if (!IsSupported())

 		{

 			return;

 		}

 		delete [] m_PlayerHuds;

 		handlesys->RemoveType(m_hHudSyncObj, g_pCoreIdent);

 		g_Players.RemoveClientListener(this);

 	}

 	virtual void OnHandleDestroy(HandleType_t type, void *object)

 	{

 		hud_syncobj_t *obj = (hud_syncobj_t *)object;

 		delete obj;

 	}

 	virtual bool GetHandleApproxSize(HandleType_t type, void *object, unsigned int *pSize)

 	{

 		*pSize = sizeof(unsigned int) * g_Players.GetMaxClients();

 		return true;

 	}

 	virtual void OnClientConnected(int client)

 	{

 		player_chaninfo_t *player;

 		player = &m_PlayerHuds[client];

 		memset(player->chan_syncobjs, 0, sizeof(player->chan_syncobjs));

 		memset(player->chan_times, 0, sizeof(player->chan_times));

 	}

 	Handle_t CreateHudSyncObj(IdentityToken_t *pIdent)

 	{

 		Handle_t hndl;

 		HandleError err;

 		hud_syncobj_t *obj;

 		HandleSecurity sec;

 		obj = new hud_syncobj_t;

 		memset(obj->player_channels, 0, sizeof(obj->player_channels));

 		sec = HandleSecurity(pIdent, g_pCoreIdent);

 		if ((hndl = handlesys->CreateHandleEx(m_hHudSyncObj, obj, &sec, NULL, &err))

 			== BAD_HANDLE)

 		{

 			delete obj;

 		}

 		return hndl;

 	}

 	HandleError ReadHudSyncObj(Handle_t hndl, 

 		IdentityToken_t *pOwner, 

 		hud_syncobj_t **pObj)

 	{

 		HandleSecurity sec(pOwner, g_pCoreIdent);

 		return handlesys->ReadHandle(hndl, m_hHudSyncObj, &sec, (void **)pObj);

 	}

 	unsigned int AutoSelectChannel(unsigned int client)

 	{

 		int last_channel;

 		player_chaninfo_t *player;

 		player = &m_PlayerHuds[client];

 		last_channel = 0;

 		for (unsigned int i = 1; i < MAX_HUD_CHANNELS; i++)

 		{

 			if (player->chan_times[i] < player->chan_times[last_channel])

 			{

 				last_channel = i;

 			}

 		}

 		ManualSelectChannel(client, last_channel);

 		return last_channel;

 	}

 	int TryReuseLastChannel(unsigned int client, hud_syncobj_t *obj)

 	{

 		int last_channel;

 		player_chaninfo_t *player;

 		player = &m_PlayerHuds[client];

 		/* First, see if we can re-use the previous channel. */

 		last_channel = obj->player_channels[client];

 		if (player->chan_syncobjs[last_channel] == obj)

 		{

 			player->chan_times[last_channel] = *g_pUniversalTime;

 			return last_channel;

 		}

 		return -1;

 	}

 	int AutoSelectChannel(unsigned int client, hud_syncobj_t *obj)

 	{

 		int last_channel;

 		player_chaninfo_t *player;

 		player = &m_PlayerHuds[client];

 		/* First, see if we can re-use the previous channel. */

 		last_channel = obj->player_channels[client];

 		if (player->chan_syncobjs[last_channel] == obj)

 		{

 			player->chan_times[last_channel] = *g_pUniversalTime;

 			return last_channel;

 		}

 		last_channel = 0;

 		for (unsigned int i = 1; i < MAX_HUD_CHANNELS; i++)

 		{

 			if (player->chan_times[i] < player->chan_times[last_channel])

 			{

 				last_channel = i;

 			}

 		}

 		obj->player_channels[client] = last_channel;

 		player->chan_syncobjs[last_channel] = obj;

 		player->chan_times[last_channel] = *g_pUniversalTime;

 		return last_channel;

 	}

 	int ManualSelectChannel(unsigned int client, int channel)

 	{

 		player_chaninfo_t *player;

 		player = &m_PlayerHuds[client];

 		player->chan_times[channel] = *g_pUniversalTime;

 		player->chan_syncobjs[channel] = NULL;

 		return channel;

 	}

 private:

 	HandleType_t m_hHudSyncObj;

 	player_chaninfo_t *m_PlayerHuds;

 } s_HudMsgHelpers;

 hud_text_parms g_hud_params;

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

 {

 	return s_HudMsgHelpers.CreateHudSyncObj(pContext->GetIdentity());

 }

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

 {

 	g_hud_params.x = sp_ctof(params[1]);

 	g_hud_params.y = sp_ctof(params[2]);

 	g_hud_params.holdTime = sp_ctof(params[3]);

 	g_hud_params.r1 = static_cast<byte>(params[4]);

 	g_hud_params.g1 = static_cast<byte>(params[5]);

 	g_hud_params.b1 = static_cast<byte>(params[6]);

 	g_hud_params.a1 = static_cast<byte>(params[7]);

 	g_hud_params.effect = params[8];

 	g_hud_params.fxTime = sp_ctof(params[9]);

 	g_hud_params.fadeinTime = sp_ctof(params[10]);

 	g_hud_params.fadeoutTime = sp_ctof(params[11]);

 	g_hud_params.r2 = 255;

 	g_hud_params.g2 = 255;

 	g_hud_params.b2 = 250;

 	g_hud_params.a2 = 0;

 	return 1;

 }

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

 {

 	cell_t *color1, *color2;

 	pContext->LocalToPhysAddr(params[4], &color1);

 	pContext->LocalToPhysAddr(params[5], &color2);

 	g_hud_params.x = sp_ctof(params[1]);

 	g_hud_params.y = sp_ctof(params[2]);

 	g_hud_params.holdTime = sp_ctof(params[3]);

 	g_hud_params.r1 = static_cast<byte>(color1[0]);

 	g_hud_params.g1 = static_cast<byte>(color1[1]);

 	g_hud_params.b1 = static_cast<byte>(color1[2]);

 	g_hud_params.a1 = static_cast<byte>(color1[3]);

 	g_hud_params.effect = params[6];

 	g_hud_params.fxTime = sp_ctof(params[7]);

 	g_hud_params.fadeinTime = sp_ctof(params[8]);

 	g_hud_params.fadeoutTime = sp_ctof(params[9]);

 	g_hud_params.r2 = static_cast<byte>(color2[0]);

 	g_hud_params.g2 = static_cast<byte>(color2[1]);

 	g_hud_params.b2 = static_cast<byte>(color2[2]);

 	g_hud_params.a2 = static_cast<byte>(color2[3]);

 	return 1;

 }

 void UTIL_SendHudText(int client, const hud_text_parms &textparms, const char *pMessage)

 {

 	cell_t players[1];

 	players[0] = client;

 #if SOURCE_ENGINE == SE_DOTA

 	CUserMsg_HudMsg *msg = (CUserMsg_HudMsg *)g_UserMsgs.StartProtobufMessage(g_HudMsgNum, players, 1, 0);

 	msg->set_channel(textparms.channel & 0xFF);

 	msg->set_x(textparms.x);

 	msg->set_y(textparms.y);

 	Color c1(textparms.r1, textparms.g1, textparms.b1, textparms.a1);

 	msg->set_color1(c1.GetRawColor());

 	Color c2(textparms.r2, textparms.g2, textparms.b2, textparms.a2);

 	msg->set_color2(c2.GetRawColor());

 	msg->set_effect(textparms.effect);

 	msg->set_fade_in_time(textparms.fadeinTime);

 	msg->set_fade_out_time(textparms.fadeoutTime);

 	msg->set_hold_time(textparms.holdTime);

 	msg->set_fx_time(textparms.fxTime);

 	msg->set_message(pMessage);

 #elif SOURCE_ENGINE == SE_CSGO

 	CCSUsrMsg_HudMsg *msg = (CCSUsrMsg_HudMsg *)g_UserMsgs.StartProtobufMessage(g_HudMsgNum, players, 1, 0);

 	msg->set_channel(textparms.channel & 0xFF);

 	CMsgVector2D *pos = msg->mutable_pos();

 	pos->set_x(textparms.x);

 	pos->set_y(textparms.y);

 	CMsgRGBA *color1 = msg->mutable_clr1();

 	color1->set_r(textparms.r1);

 	color1->set_g(textparms.g1);

 	color1->set_b(textparms.b1);

 	color1->set_a(textparms.a1);

 	CMsgRGBA *color2 = msg->mutable_clr2();

 	color2->set_r(textparms.r2);

 	color2->set_g(textparms.g2);

 	color2->set_b(textparms.b2);

 	color2->set_a(textparms.a2);

 	msg->set_effect(textparms.effect);

 	msg->set_fade_in_time(textparms.fadeinTime);

 	msg->set_fade_out_time(textparms.fadeoutTime);

 	msg->set_hold_time(textparms.holdTime);

 	msg->set_fx_time(textparms.fxTime);

 	msg->set_text(pMessage);

 #else

 	bf_write *bf = g_UserMsgs.StartBitBufMessage(g_HudMsgNum, players, 1, 0);

 	bf->WriteByte(textparms.channel & 0xFF );

 	bf->WriteFloat(textparms.x);

 	bf->WriteFloat(textparms.y);

 	bf->WriteByte(textparms.r1);

 	bf->WriteByte(textparms.g1);

 	bf->WriteByte(textparms.b1);

 	bf->WriteByte(textparms.a1);

 	bf->WriteByte(textparms.r2);

 	bf->WriteByte(textparms.g2);

 	bf->WriteByte(textparms.b2);

 	bf->WriteByte(textparms.a2);

 	bf->WriteByte(textparms.effect);

 	bf->WriteFloat(textparms.fadeinTime);

 	bf->WriteFloat(textparms.fadeoutTime);

 	bf->WriteFloat(textparms.holdTime);

 	bf->WriteFloat(textparms.fxTime);

 	bf->WriteString(pMessage);

 #endif

 	g_UserMsgs.EndMessage();

 }

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

 {

 	int client;

 	Handle_t err;

 	CPlayer *pPlayer;

 	hud_syncobj_t *obj;

 	char message_buffer[255-36];

 	if (!s_HudMsgHelpers.IsSupported())

 	{

 		return -1;

 	}

 	if ((err = s_HudMsgHelpers.ReadHudSyncObj(params[2], pContext->GetIdentity(), &obj)) != HandleError_None)

 	{

 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", params[2], err);

 	}

 	client = params[1];

 	if ((pPlayer = g_Players.GetPlayerByIndex(client)) == NULL)

 	{

 		return pContext->ThrowNativeError("Invalid client index %d", client);

 	}

 	else if (!pPlayer->IsInGame())

 	{

 		return pContext->ThrowNativeError("Client %d is not in-game", client);

 	}

 	g_SourceMod.FormatString(message_buffer, sizeof(message_buffer), pContext, params, 3);

 	if (pContext->GetLastNativeError() != SP_ERROR_NONE)

 	{

 		return 0;

 	}

 	g_hud_params.channel = s_HudMsgHelpers.AutoSelectChannel(client, obj);

 	UTIL_SendHudText(client, g_hud_params, message_buffer);

 	return 1;

 }

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

 {

 	int client;

 	int channel;

 	Handle_t err;

 	CPlayer *pPlayer;

 	hud_syncobj_t *obj;

 	if (!s_HudMsgHelpers.IsSupported())

 	{

 		return -1;

 	}

 	if ((err = s_HudMsgHelpers.ReadHudSyncObj(params[2], pContext->GetIdentity(), &obj)) != HandleError_None)

 	{

 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", params[2], err);

 	}

 	client = params[1];

 	if ((pPlayer = g_Players.GetPlayerByIndex(client)) == NULL)

 	{

 		return pContext->ThrowNativeError("Invalid client index %d", client);

 	}

 	else if (!pPlayer->IsInGame())

 	{

 		return pContext->ThrowNativeError("Client %d is not in-game", client);

 	}

 	if ((channel = s_HudMsgHelpers.TryReuseLastChannel(client, obj)) == -1)

 	{

 		return -1;

 	}

 	g_hud_params.channel = channel;

 	UTIL_SendHudText(client, g_hud_params, "");

 	return g_hud_params.channel;

 }

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

 {

 	int client;

 	CPlayer *pPlayer;

 	char message_buffer[255-36];

 	if (!s_HudMsgHelpers.IsSupported())

 	{

 		return -1;

 	}

 	client = params[1];

 	if ((pPlayer = g_Players.GetPlayerByIndex(client)) == NULL)

 	{

 		return pContext->ThrowNativeError("Invalid client index %d", client);

 	}

 	else if (!pPlayer->IsInGame())

 	{

 		return pContext->ThrowNativeError("Client %d is not in-game", client);

 	}

 	g_SourceMod.FormatString(message_buffer, sizeof(message_buffer), pContext, params, 3);

 	if (pContext->GetLastNativeError() != SP_ERROR_NONE)

 	{

 		return 0;

 	}

 	if (params[2] == -1)

 	{

 		g_hud_params.channel = s_HudMsgHelpers.AutoSelectChannel(client);

 	}

 	else

 	{

 		g_hud_params.channel = params[2] % MAX_HUD_CHANNELS;

 		s_HudMsgHelpers.ManualSelectChannel(client, g_hud_params.channel);

 	}

 	UTIL_SendHudText(client, g_hud_params, message_buffer);

 	return g_hud_params.channel;

 }

 REGISTER_NATIVES(hudNatives)

 {

 	{"ClearSyncHud",				ClearSyncHud},

 	{"CreateHudSynchronizer",		CreateHudSynchronizer},

 	{"SetHudTextParams",			SetHudTextParams},

 	{"SetHudTextParamsEx",			SetHudTextParamsEx},

 	{"ShowHudText",					ShowHudText},

 	{"ShowSyncHudText",				ShowSyncHudText},

 	{NULL,							NULL},

 };