dvander@592: /**
ds@653:  * vim: set ts=4 :
dvander@592:  * ===============================================================
dvander@592:  * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
dvander@592:  * ===============================================================
dvander@592:  *
dvander@592:  * This file is not open source and may not be copied without explicit
dvander@592:  * written permission of AlliedModders LLC.  This file may not be redistributed 
dvander@592:  * in whole or significant part.
dvander@592:  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
dvander@592:  *
dvander@592:  * Version: $Id$
dvander@592:  */
dvander@592: 
dvander@592: #include "HalfLife2.h"
ds@696: #include "sourcemod.h"
dvander@592: #include "sourcemm_api.h"
ds@873: #include "UserMessages.h"
dvander@592: 
dvander@592: CHalfLife2 g_HL2;
dvander@685: bool g_IsOriginalEngine = false;
dvander@592: 
faluco@682: namespace SourceHook
faluco@682: {
faluco@682: 	template<>
faluco@682: 	int HashFunction<datamap_t *>(datamap_t * const &k)
faluco@682: 	{
faluco@682: 		return reinterpret_cast<int>(k);
faluco@682: 	}
faluco@682: 
faluco@682: 	template<>
faluco@682: 	int Compare<datamap_t *>(datamap_t * const &k1, datamap_t * const &k2)
faluco@682: 	{
faluco@682: 		return (k1-k2);
faluco@682: 	}
faluco@682: }
faluco@682: 
dvander@592: CHalfLife2::CHalfLife2()
dvander@592: {
dvander@592: 	m_pClasses = sm_trie_create();
dvander@592: }
dvander@592: 
dvander@592: CHalfLife2::~CHalfLife2()
dvander@592: {
dvander@592: 	sm_trie_destroy(m_pClasses);
dvander@592: 
dvander@592: 	List<DataTableInfo *>::iterator iter;
dvander@592: 	DataTableInfo *pInfo;
dvander@592: 	for (iter=m_Tables.begin(); iter!=m_Tables.end(); iter++)
dvander@592: 	{
dvander@592: 		pInfo = (*iter);
dvander@592: 		sm_trie_destroy(pInfo->lookup);
dvander@592: 		delete pInfo;
dvander@592: 	}
dvander@592: 
dvander@592: 	m_Tables.clear();
faluco@682: 
faluco@682: 	THash<datamap_t *, DataMapTrie>::iterator h_iter;
faluco@682: 	for (h_iter=m_Maps.begin(); h_iter!=m_Maps.end(); h_iter++)
faluco@682: 	{
faluco@682: 		if (h_iter->val.trie)
faluco@682: 		{
faluco@682: 			sm_trie_destroy(h_iter->val.trie);
faluco@682: 			h_iter->val.trie = NULL;
faluco@682: 		}
faluco@682: 	}
faluco@682: 
faluco@682: 	m_Maps.clear();
dvander@592: }
dvander@592: 
dvander@677: CSharedEdictChangeInfo *g_pSharedChangeInfo = NULL;
dvander@677: 
dvander@592: void CHalfLife2::OnSourceModStartup(bool late)
dvander@592: {
ds@696: 	/* The Ship currently is the only known game to use an older version of the engine */
ds@729: 	if (strcasecmp(g_SourceMod.GetGameFolderName(), "ship") == 0)
dvander@677: 	{
ds@696: 		/* :TODO: Better engine versioning - perhaps something added to SourceMM? */
ds@696: 		g_IsOriginalEngine = true;
ds@696: 	} else if (!g_pSharedChangeInfo) {
dvander@677: 		g_pSharedChangeInfo = engine->GetSharedEdictChangeInfo();
dvander@677: 	}
dvander@592: }
dvander@592: 
ds@873: void CHalfLife2::OnSourceModAllInitialized()
ds@873: {
ds@873: 	m_MsgTextMsg = g_UserMsgs.GetMessageIndex("TextMsg");
dvander@965: 	g_ShareSys.AddInterface(NULL, this);
ds@873: }
ds@873: 
dvander@678: IChangeInfoAccessor *CBaseEdict::GetChangeAccessor()
dvander@678: {
dvander@678: 	return engine->GetChangeAccessor( (const edict_t *)this );
dvander@678: }
dvander@678: 
dvander@592: SendProp *UTIL_FindInSendTable(SendTable *pTable, const char *name)
dvander@592: {
dvander@592: 	const char *pname;
dvander@592: 	int props = pTable->GetNumProps();
dvander@592: 	SendProp *prop;
dvander@592: 
dvander@592: 	for (int i=0; i<props; i++)
dvander@592: 	{
dvander@592: 		prop = pTable->GetProp(i);
dvander@592: 		pname = prop->GetName();
dvander@592: 		if (pname && strcmp(name, pname) == 0)
dvander@592: 		{
dvander@592: 			return prop;
dvander@592: 		}
dvander@592: 		if (prop->GetDataTable())
dvander@592: 		{
dvander@592: 			if ((prop=UTIL_FindInSendTable(prop->GetDataTable(), name)) != NULL)
dvander@592: 			{
dvander@592: 				return prop;
dvander@592: 			}
dvander@592: 		}
dvander@592: 	}
dvander@592: 
dvander@592: 	return NULL;
dvander@592: }
dvander@592: 
faluco@682: typedescription_t *UTIL_FindInDataMap(datamap_t *pMap, const char *name)
faluco@682: {
faluco@682: 	while (pMap)
faluco@682: 	{
faluco@682: 		for (int i=0; i<pMap->dataNumFields; i++)
faluco@682: 		{
dvander@709: 			if (pMap->dataDesc[i].fieldName == NULL)
dvander@709: 			{
dvander@709: 				continue;
dvander@709: 			}
faluco@682: 			if (strcmp(name, pMap->dataDesc[i].fieldName) == 0)
faluco@682: 			{
faluco@682: 				return &(pMap->dataDesc[i]);
faluco@682: 			}
faluco@682: 			if (pMap->dataDesc[i].td)
faluco@682: 			{
faluco@682: 				typedescription_t *_td;
faluco@682: 				if ((_td=UTIL_FindInDataMap(pMap->dataDesc[i].td, name)) != NULL)
faluco@682: 				{
faluco@682: 					return _td;
faluco@682: 				}
faluco@682: 			}
faluco@682: 		}
faluco@682: 		pMap = pMap->baseMap;
faluco@682: 	}
faluco@682: 
faluco@682: 	return NULL; 
faluco@682: }
faluco@682: 
dvander@592: ServerClass *CHalfLife2::FindServerClass(const char *classname)
dvander@592: {
dvander@592: 	DataTableInfo *pInfo = _FindServerClass(classname);
dvander@592: 	if (!pInfo)
dvander@592: 	{
dvander@592: 		return NULL;
dvander@592: 	}
dvander@592: 
dvander@592: 	return pInfo->sc;
dvander@592: }
dvander@592: 
dvander@592: DataTableInfo *CHalfLife2::_FindServerClass(const char *classname)
dvander@592: {
dvander@592: 	DataTableInfo *pInfo = NULL;
dvander@592: 
dvander@592: 	if (!sm_trie_retrieve(m_pClasses, classname, (void **)&pInfo))
dvander@592: 	{
dvander@592: 		ServerClass *sc = gamedll->GetAllServerClasses();
dvander@592: 		while (sc)
dvander@592: 		{
dvander@592: 			if (strcmp(classname, sc->GetName()) == 0)
dvander@592: 			{
dvander@592: 				pInfo = new DataTableInfo;
dvander@592: 				pInfo->lookup = sm_trie_create();
dvander@592: 				pInfo->sc = sc;
dvander@592: 				sm_trie_insert(m_pClasses, classname, pInfo);
faluco@682: 				m_Tables.push_back(pInfo);
dvander@592: 				break;
dvander@592: 			}
dvander@592: 			sc = sc->m_pNext;
dvander@592: 		}
dvander@592: 		if (!pInfo)
dvander@592: 		{
dvander@592: 			return NULL;
dvander@592: 		}
dvander@592: 	}
dvander@592: 
dvander@592: 	return pInfo;
dvander@592: }
dvander@592: 
dvander@592: SendProp *CHalfLife2::FindInSendTable(const char *classname, const char *offset)
dvander@592: {
dvander@592: 	DataTableInfo *pInfo = _FindServerClass(classname);
dvander@592: 
dvander@592: 	if (!pInfo)
dvander@592: 	{
dvander@592: 		return NULL;
dvander@592: 	}
dvander@592: 
faluco@682: 	SendProp *pProp = NULL;
dvander@592: 	if (!sm_trie_retrieve(pInfo->lookup, offset, (void **)&pProp))
dvander@592: 	{
dvander@592: 		if ((pProp = UTIL_FindInSendTable(pInfo->sc->m_pTable, offset)) != NULL)
dvander@592: 		{
dvander@592: 			sm_trie_insert(pInfo->lookup, offset, pProp);
dvander@592: 		}
dvander@592: 	}
dvander@592: 
dvander@592: 	return pProp;
dvander@592: }
faluco@682: 
faluco@682: typedescription_t *CHalfLife2::FindInDataMap(datamap_t *pMap, const char *offset)
faluco@682: {
faluco@682: 	typedescription_t *td = NULL;
faluco@682: 	DataMapTrie &val = m_Maps[pMap];
faluco@682: 
faluco@682: 	if (!val.trie)
faluco@682: 	{
faluco@682: 		val.trie = sm_trie_create();
faluco@682: 	}
faluco@682: 	if (!sm_trie_retrieve(val.trie, offset, (void **)&td))
faluco@682: 	{
faluco@682: 		if ((td = UTIL_FindInDataMap(pMap, offset)) != NULL)
faluco@682: 		{
faluco@682: 			sm_trie_insert(val.trie, offset, td);
faluco@682: 		}
faluco@682: 	}
faluco@682: 
faluco@682: 	return td;
faluco@682: }
dvander@685: 
dvander@685: void CHalfLife2::SetEdictStateChanged(edict_t *pEdict, unsigned short offset)
dvander@685: {
dvander@685: 	if (!g_IsOriginalEngine)
dvander@685: 	{
dvander@685: 		if (offset)
dvander@685: 		{
dvander@685: 			pEdict->StateChanged(offset);
dvander@685: 		} else {
dvander@685: 			pEdict->StateChanged();
dvander@685: 		}
dvander@685: 	} else {
dvander@685: 		pEdict->m_fStateFlags |= FL_EDICT_CHANGED;
dvander@685: 	}
dvander@685: }
ds@873: 
ds@873: void CHalfLife2::TextMsg(int client, int dest, const char *msg)
ds@873: {
ds@873: 	bf_write *pBitBuf = NULL;
ds@873: 	cell_t players[] = {client};
ds@873: 
ds@873: 	pBitBuf = g_UserMsgs.StartMessage(m_MsgTextMsg, players, 1, USERMSG_RELIABLE);
ds@873: 	pBitBuf->WriteByte(dest);
ds@873: 	pBitBuf->WriteString(msg);
ds@873: 	g_UserMsgs.EndMessage();
ds@873: }