/**

  * vim: set ts=4 :

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

  * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.

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

  *

  * This file is not open source and may not be copied without explicit

  * written permission of AlliedModders LLC.  This file may not be redistributed 

  * in whole or significant part.

  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php

  *

  * Version: $Id$

  */

 #include "HalfLife2.h"

 #include "sourcemod.h"

 #include "sourcemm_api.h"

 #include "UserMessages.h"

 CHalfLife2 g_HL2;

 bool g_IsOriginalEngine = false;

 namespace SourceHook

 {

 	template<>

 	int HashFunction<datamap_t *>(datamap_t * const &k)

 	{

 		return reinterpret_cast<int>(k);

 	}

 	template<>

 	int Compare<datamap_t *>(datamap_t * const &k1, datamap_t * const &k2)

 	{

 		return (k1-k2);

 	}

 }

 CHalfLife2::CHalfLife2()

 {

 	m_pClasses = sm_trie_create();

 }

 CHalfLife2::~CHalfLife2()

 {

 	sm_trie_destroy(m_pClasses);

 	List<DataTableInfo *>::iterator iter;

 	DataTableInfo *pInfo;

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

 	{

 		pInfo = (*iter);

 		sm_trie_destroy(pInfo->lookup);

 		delete pInfo;

 	}

 	m_Tables.clear();

 	THash<datamap_t *, DataMapTrie>::iterator h_iter;

 	for (h_iter=m_Maps.begin(); h_iter!=m_Maps.end(); h_iter++)

 	{

 		if (h_iter->val.trie)

 		{

 			sm_trie_destroy(h_iter->val.trie);

 			h_iter->val.trie = NULL;

 		}

 	}

 	m_Maps.clear();

 }

 CSharedEdictChangeInfo *g_pSharedChangeInfo = NULL;

 void CHalfLife2::OnSourceModStartup(bool late)

 {

 	/* The Ship currently is the only known game to use an older version of the engine */

 	if (strcasecmp(g_SourceMod.GetGameFolderName(), "ship") == 0)

 	{

 		/* :TODO: Better engine versioning - perhaps something added to SourceMM? */

 		g_IsOriginalEngine = true;

 	} else if (!g_pSharedChangeInfo) {

 		g_pSharedChangeInfo = engine->GetSharedEdictChangeInfo();

 	}

 }

 void CHalfLife2::OnSourceModAllInitialized()

 {

 	m_MsgTextMsg = g_UserMsgs.GetMessageIndex("TextMsg");

 	m_HinTextMsg = g_UserMsgs.GetMessageIndex("HintText");

 	m_VGUIMenu = g_UserMsgs.GetMessageIndex("VGUIMenu");

 	g_ShareSys.AddInterface(NULL, this);

 }

 IChangeInfoAccessor *CBaseEdict::GetChangeAccessor()

 {

 	return engine->GetChangeAccessor( (const edict_t *)this );

 }

 SendProp *UTIL_FindInSendTable(SendTable *pTable, const char *name)

 {

 	const char *pname;

 	int props = pTable->GetNumProps();

 	SendProp *prop;

 	for (int i=0; i<props; i++)

 	{

 		prop = pTable->GetProp(i);

 		pname = prop->GetName();

 		if (pname && strcmp(name, pname) == 0)

 		{

 			return prop;

 		}

 		if (prop->GetDataTable())

 		{

 			if ((prop=UTIL_FindInSendTable(prop->GetDataTable(), name)) != NULL)

 			{

 				return prop;

 			}

 		}

 	}

 	return NULL;

 }

 typedescription_t *UTIL_FindInDataMap(datamap_t *pMap, const char *name)

 {

 	while (pMap)

 	{

 		for (int i=0; i<pMap->dataNumFields; i++)

 		{

 			if (pMap->dataDesc[i].fieldName == NULL)

 			{

 				continue;

 			}

 			if (strcmp(name, pMap->dataDesc[i].fieldName) == 0)

 			{

 				return &(pMap->dataDesc[i]);

 			}

 			if (pMap->dataDesc[i].td)

 			{

 				typedescription_t *_td;

 				if ((_td=UTIL_FindInDataMap(pMap->dataDesc[i].td, name)) != NULL)

 				{

 					return _td;

 				}

 			}

 		}

 		pMap = pMap->baseMap;

 	}

 	return NULL; 

 }

 ServerClass *CHalfLife2::FindServerClass(const char *classname)

 {

 	DataTableInfo *pInfo = _FindServerClass(classname);

 	if (!pInfo)

 	{

 		return NULL;

 	}

 	return pInfo->sc;

 }

 DataTableInfo *CHalfLife2::_FindServerClass(const char *classname)

 {

 	DataTableInfo *pInfo = NULL;

 	if (!sm_trie_retrieve(m_pClasses, classname, (void **)&pInfo))

 	{

 		ServerClass *sc = gamedll->GetAllServerClasses();

 		while (sc)

 		{

 			if (strcmp(classname, sc->GetName()) == 0)

 			{

 				pInfo = new DataTableInfo;

 				pInfo->lookup = sm_trie_create();

 				pInfo->sc = sc;

 				sm_trie_insert(m_pClasses, classname, pInfo);

 				m_Tables.push_back(pInfo);

 				break;

 			}

 			sc = sc->m_pNext;

 		}

 		if (!pInfo)

 		{

 			return NULL;

 		}

 	}

 	return pInfo;

 }

 SendProp *CHalfLife2::FindInSendTable(const char *classname, const char *offset)

 {

 	DataTableInfo *pInfo = _FindServerClass(classname);

 	if (!pInfo)

 	{

 		return NULL;

 	}

 	SendProp *pProp = NULL;

 	if (!sm_trie_retrieve(pInfo->lookup, offset, (void **)&pProp))

 	{

 		if ((pProp = UTIL_FindInSendTable(pInfo->sc->m_pTable, offset)) != NULL)

 		{

 			sm_trie_insert(pInfo->lookup, offset, pProp);

 		}

 	}

 	return pProp;

 }

 typedescription_t *CHalfLife2::FindInDataMap(datamap_t *pMap, const char *offset)

 {

 	typedescription_t *td = NULL;

 	DataMapTrie &val = m_Maps[pMap];

 	if (!val.trie)

 	{

 		val.trie = sm_trie_create();

 	}

 	if (!sm_trie_retrieve(val.trie, offset, (void **)&td))

 	{

 		if ((td = UTIL_FindInDataMap(pMap, offset)) != NULL)

 		{

 			sm_trie_insert(val.trie, offset, td);

 		}

 	}

 	return td;

 }

 void CHalfLife2::SetEdictStateChanged(edict_t *pEdict, unsigned short offset)

 {

 	if (!g_IsOriginalEngine)

 	{

 		if (offset)

 		{

 			pEdict->StateChanged(offset);

 		} else {

 			pEdict->StateChanged();

 		}

 	} else {

 		pEdict->m_fStateFlags |= FL_EDICT_CHANGED;

 	}

 }

 bool CHalfLife2::TextMsg(int client, int dest, const char *msg)

 {

 	bf_write *pBitBuf = NULL;

 	cell_t players[] = {client};

 	if ((pBitBuf = g_UserMsgs.StartMessage(m_MsgTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)

 	{

 		return false;

 	}

 	pBitBuf->WriteByte(dest);

 	pBitBuf->WriteString(msg);

 	g_UserMsgs.EndMessage();

 	return true;

 }

 bool CHalfLife2::HinTextMsg(int client, const char *msg)

 {

 	bf_write *pBitBuf = NULL;

 	cell_t players[] = {client};

 	if ((pBitBuf = g_UserMsgs.StartMessage(m_HinTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)

 	{

 		return false;

 	}

 	pBitBuf->WriteByte(1);

 	pBitBuf->WriteString(msg);

 	g_UserMsgs.EndMessage();

 	return true;

 }

 bool CHalfLife2::ShowVGUIMenu(int client, const char *name, KeyValues *data, bool show)

 {

 	bf_write *pBitBuf = NULL;

 	KeyValues *SubKey = NULL;

 	int count = 0;

 	cell_t players[] = {client};

 	if ((pBitBuf = g_UserMsgs.StartMessage(m_VGUIMenu, players, 1, USERMSG_RELIABLE)) == NULL)

 	{

 		return false;

 	}

 	if (data)

 	{

 		SubKey = data->GetFirstSubKey();

 		while (SubKey)

 		{

 			count++;

 			SubKey = SubKey->GetNextKey();

 		}

 		SubKey = data->GetFirstSubKey();

 	}

 	pBitBuf->WriteString(name);

 	pBitBuf->WriteByte((show) ? 1 : 0);

 	pBitBuf->WriteByte(count);

 	while (SubKey)

 	{

 		pBitBuf->WriteString(SubKey->GetName());

 		pBitBuf->WriteString(SubKey->GetString());

 		SubKey = SubKey->GetNextKey();

 	}

 	g_UserMsgs.EndMessage();

 	return true;

 }