skamonkey@3233: /**
skamonkey@3233:  * vim: set ts=4 :
skamonkey@3233:  * =============================================================================
skamonkey@3233:  * SourceMod Team Fortress 2 Extension
skamonkey@3233:  * Copyright (C) 2004-2011 AlliedModders LLC.  All rights reserved.
skamonkey@3233:  * =============================================================================
skamonkey@3233:  *
skamonkey@3233:  * This program is free software; you can redistribute it and/or modify it under
skamonkey@3233:  * the terms of the GNU General Public License, version 3.0, as published by the
skamonkey@3233:  * Free Software Foundation.
skamonkey@3233:  * 
skamonkey@3233:  * This program is distributed in the hope that it will be useful, but WITHOUT
skamonkey@3233:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
skamonkey@3233:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
skamonkey@3233:  * details.
skamonkey@3233:  *
skamonkey@3233:  * You should have received a copy of the GNU General Public License along with
skamonkey@3233:  * this program.  If not, see <http://www.gnu.org/licenses/>.
skamonkey@3233:  *
skamonkey@3233:  * As a special exception, AlliedModders LLC gives you permission to link the
skamonkey@3233:  * code of this program (as well as its derivative works) to "Half-Life 2," the
skamonkey@3233:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
skamonkey@3233:  * by the Valve Corporation.  You must obey the GNU General Public License in
skamonkey@3233:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
skamonkey@3233:  * this exception to all derivative works.  AlliedModders LLC defines further
skamonkey@3233:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
skamonkey@3233:  * or <http://www.sourcemod.net/license.php>.
skamonkey@3233:  *
skamonkey@3233:  * Version: $Id$
skamonkey@3233:  */
skamonkey@3233: 
skamonkey@3233: #include "extension.h"
skamonkey@3233: #include "conditions.h"
skamonkey@3233: #include "util.h"
skamonkey@3233: 
skamonkey@3332: #include <iplayerinfo.h>
skamonkey@3332: 
flaminsarge@4000: typedef struct condflags_s {
flaminsarge@4000: 	uint64_t upper;
flaminsarge@4000: 	uint64_t lower;
flaminsarge@4000: } condflags_t;
flaminsarge@4000: condflags_t condflags_empty = { 0, 0 };
skamonkey@3332: 
kyle@4345: condflags_t g_PlayerConds[SM_MAXPLAYERS+1];
skamonkey@3233: 
skamonkey@3233: IForward *g_addCondForward = NULL;
skamonkey@3233: IForward *g_removeCondForward = NULL;
skamonkey@3233: 
skamonkey@3233: int playerCondOffset = -1;
skamonkey@3332: int playerCondExOffset = -1;
flaminsarge@4000: int playerCondEx2Offset = -1;
skamonkey@3233: int conditionBitsOffset = -1;
skamonkey@3233: 
skamonkey@3233: bool g_bIgnoreRemove;
skamonkey@3233: 
skamonkey@3737: #define MAX_CONDS (sizeof(uint64_t) * 8)
skamonkey@3737: 
skamonkey@3737: inline condflags_t GetPlayerConds(CBaseEntity *pPlayer)
skamonkey@3233: {
flaminsarge@4000: 	condflags_t result;
flaminsarge@4000: 	uint32_t playerCond    =  *(uint32_t *)((intptr_t)pPlayer + playerCondOffset);
flaminsarge@4000: 	uint32_t condBits      =  *(uint32_t *)((intptr_t)pPlayer + conditionBitsOffset);
flaminsarge@4000: 	uint32_t playerCondEx  =  *(uint32_t *)((intptr_t)pPlayer + playerCondExOffset);
flaminsarge@4000: 	uint32_t playerCondEx2 =  *(uint32_t *)((intptr_t)pPlayer + playerCondEx2Offset);
skamonkey@3233: 
skamonkey@3737: 	uint64_t playerCondExAdj = playerCondEx;
skamonkey@3737: 	playerCondExAdj <<= 32;
skamonkey@3737: 
flaminsarge@4000: 	result.lower = playerCond|condBits|playerCondExAdj;
flaminsarge@4000: 	result.upper = playerCondEx2;
flaminsarge@4000: 	return result;
skamonkey@3233: }
skamonkey@3233: 
skamonkey@3332: void Conditions_OnGameFrame(bool simulating)
skamonkey@3233: {
skamonkey@3332: 	if (!simulating)
skamonkey@3233: 		return;
skamonkey@3233: 
skamonkey@3737: 	condflags_t oldconds;
skamonkey@3737: 	condflags_t newconds;
skamonkey@3332: 	
skamonkey@3737: 	condflags_t addedconds;
skamonkey@3737: 	condflags_t removedconds;
skamonkey@3233: 
skamonkey@3332: 	int maxClients = gpGlobals->maxClients;
skamonkey@3737: 	for (int i = 1; i <= maxClients; i++)
skamonkey@3233: 	{
skamonkey@3332: 		IGamePlayer *pPlayer = playerhelpers->GetGamePlayer(i);
skamonkey@3332: 		if (!pPlayer || !pPlayer->IsInGame())
skamonkey@3332: 			continue;
skamonkey@3332: 
skamonkey@3332: 		IPlayerInfo *info = pPlayer->GetPlayerInfo();
skamonkey@3332: 		if (info->IsHLTV() || info->IsReplay())
skamonkey@3332: 			continue;
skamonkey@3233: 
skamonkey@3332: 		CBaseEntity *pEntity = gamehelpers->ReferenceToEntity(i);
skamonkey@3737: 		oldconds = g_PlayerConds[i];
skamonkey@3737: 		newconds = GetPlayerConds(pEntity);
skamonkey@3332: 
flaminsarge@4000: 		if (oldconds.lower == newconds.lower && oldconds.upper == newconds.upper)
skamonkey@3332: 			continue;
skamonkey@3332: 
flaminsarge@4000: 		addedconds.lower = newconds.lower &~ oldconds.lower;
flaminsarge@4000: 		removedconds.lower = oldconds.lower &~ newconds.lower;
flaminsarge@4000: 		addedconds.upper = newconds.upper &~ oldconds.upper;
flaminsarge@4000: 		removedconds.upper = oldconds.upper &~ newconds.upper;
skamonkey@3737: 
skamonkey@3737: 		uint64_t j;
flaminsarge@4000: 		uint64_t bit;
skamonkey@3737: 
asherkin@4002: 		uint64_t maxbit = MAX(addedconds.lower, addedconds.upper);
flaminsarge@4000: 		for (j = 0; j < MAX_CONDS && (bit = ((uint64_t)1 << j)) <= maxbit; j++)
skamonkey@3720: 		{
flaminsarge@4000: 			if ((addedconds.lower & bit) == bit)
skamonkey@3737: 			{
skamonkey@3737: 				g_addCondForward->PushCell(i);
skamonkey@3737: 				g_addCondForward->PushCell(j & 0xFFFFFFFF);
skamonkey@3737: 				g_addCondForward->Execute(NULL, NULL);
skamonkey@3737: 			}
flaminsarge@4000: 			if ((addedconds.upper & bit) == bit)
flaminsarge@4000: 			{
flaminsarge@4000: 				g_addCondForward->PushCell(i);
flaminsarge@4000: 				g_addCondForward->PushCell((j+64) & 0xFFFFFFFF);
flaminsarge@4000: 				g_addCondForward->Execute(NULL, NULL);
flaminsarge@4000: 			}
skamonkey@3720: 		}
skamonkey@3233: 
asherkin@4002: 		maxbit = MAX(removedconds.lower, removedconds.upper);
flaminsarge@4000: 		for (j = 0; j < MAX_CONDS && (bit = ((uint64_t)1 << j)) <= maxbit; j++)
skamonkey@3332: 		{
flaminsarge@4000: 			if ((removedconds.lower & bit) == bit)
skamonkey@3737: 			{
skamonkey@3737: 				g_removeCondForward->PushCell(i);
skamonkey@3737: 				g_removeCondForward->PushCell(j & 0xFFFFFFFF);
skamonkey@3737: 				g_removeCondForward->Execute(NULL, NULL);
skamonkey@3737: 			}
flaminsarge@4000: 			if ((removedconds.upper & bit) == bit)
flaminsarge@4000: 			{
flaminsarge@4000: 				g_removeCondForward->PushCell(i);
flaminsarge@4000: 				g_removeCondForward->PushCell((j+64) & 0xFFFFFFFF);
flaminsarge@4000: 				g_removeCondForward->Execute(NULL, NULL);
flaminsarge@4000: 			}
skamonkey@3332: 		}
skamonkey@3332: 
skamonkey@3737: 		g_PlayerConds[i] = newconds;
skamonkey@3233: 	}
skamonkey@3233: }
skamonkey@3233: 
skamonkey@3332: bool InitialiseConditionChecks()
skamonkey@3233: {
skamonkey@3233: 	sm_sendprop_info_t prop;
skamonkey@3233: 	if (!gamehelpers->FindSendPropInfo("CTFPlayer", "m_nPlayerCond", &prop))
skamonkey@3233: 	{
skamonkey@3233: 		g_pSM->LogError(myself, "Failed to find m_nPlayerCond prop offset");
skamonkey@3233: 		return false;
skamonkey@3233: 	}
skamonkey@3233: 	
skamonkey@3233: 	playerCondOffset = prop.actual_offset;
skamonkey@3233: 	
skamonkey@3233: 	if (!gamehelpers->FindSendPropInfo("CTFPlayer", "_condition_bits", &prop))
skamonkey@3233: 	{
skamonkey@3233: 		g_pSM->LogError(myself, "Failed to find _condition_bits prop offset");
skamonkey@3233: 		return false;
skamonkey@3233: 	}
skamonkey@3233: 
skamonkey@3233: 	conditionBitsOffset = prop.actual_offset;
skamonkey@3233: 
skamonkey@3332: 	if (!gamehelpers->FindSendPropInfo("CTFPlayer", "m_nPlayerCondEx", &prop))
skamonkey@3332: 	{
skamonkey@3332: 		g_pSM->LogError(myself, "Failed to find m_nPlayerCondEx prop offset");
skamonkey@3233: 		return false;
skamonkey@3332: 	}
skamonkey@3332: 	
skamonkey@3332: 	playerCondExOffset = prop.actual_offset;
skamonkey@3233: 
flaminsarge@4000: 	if (!gamehelpers->FindSendPropInfo("CTFPlayer", "m_nPlayerCondEx2", &prop))
flaminsarge@4000: 	{
flaminsarge@4000: 		g_pSM->LogError(myself, "Failed to find m_nPlayerCondEx2 prop offset");
flaminsarge@4000: 		return false;
flaminsarge@4000: 	}
flaminsarge@4000: 	
flaminsarge@4000: 	playerCondEx2Offset = prop.actual_offset;
flaminsarge@4000: 
flaminsarge@4000: 	if (playerCondOffset == -1 || playerCondExOffset == -1 || conditionBitsOffset == -1 || playerCondEx2Offset == -1)
skamonkey@3332: 		return false;
skamonkey@3332: 	
skamonkey@3332: 	int maxClients = gpGlobals->maxClients;
skamonkey@3332: 	for (int i = 1; i <= maxClients; i++)
skamonkey@3233: 	{
skamonkey@3332: 		IGamePlayer *pPlayer = playerhelpers->GetGamePlayer(i);
skamonkey@3332: 		if (!pPlayer || !pPlayer->IsInGame())
skamonkey@3332: 			continue;
skamonkey@3332: 
skamonkey@3737: 		g_PlayerConds[i] = GetPlayerConds(gamehelpers->ReferenceToEntity(i));
skamonkey@3233: 	}
skamonkey@3233: 
skamonkey@3332: 	g_pSM->AddGameFrameHook(Conditions_OnGameFrame);
skamonkey@3233: 
skamonkey@3233: 	return true;
skamonkey@3233: }
skamonkey@3233: 
skamonkey@3332: void Conditions_OnClientPutInServer(int client)
skamonkey@3233: {
flaminsarge@4000: 	g_PlayerConds[client] = condflags_empty;
skamonkey@3233: }
skamonkey@3233: 
skamonkey@3332: void RemoveConditionChecks()
skamonkey@3332: {
skamonkey@3332: 	g_pSM->RemoveGameFrameHook(Conditions_OnGameFrame);
skamonkey@3332: }
skamonkey@3332: