From 39ed87570bdb2f86969d4be821c94b722dc71179 Mon Sep 17 00:00:00 2001
From: Joe Ludwig <joe@valvesoftware.com>
Date: Wed, 26 Jun 2013 15:22:04 -0700
Subject: First version of the SOurce SDK 2013

---
 mp/src/game/server/test_stressentities.cpp | 152 +++++++++++++++++++++++++++++
 1 file changed, 152 insertions(+)
 create mode 100644 mp/src/game/server/test_stressentities.cpp

(limited to 'mp/src/game/server/test_stressentities.cpp')

diff --git a/mp/src/game/server/test_stressentities.cpp b/mp/src/game/server/test_stressentities.cpp
new file mode 100644
index 00000000..9bcf2ed0
--- /dev/null
+++ b/mp/src/game/server/test_stressentities.cpp
@@ -0,0 +1,152 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//=============================================================================//
+
+#include "cbase.h"
+#include "test_stressentities.h"
+#include "vstdlib/random.h"
+#include "world.h"
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include "tier0/memdbgon.h"
+
+CStressEntityReg	*CStressEntityReg::s_pHead = NULL;
+
+
+// CStressEntityReg::s_pHead in array form for convenient access.
+CUtlVector<CStressEntityReg*> g_StressEntityRegs;
+
+CUtlVector<EHANDLE> g_StressEntities;
+
+
+CBaseEntity* MoveToRandomSpot( CBaseEntity *pEnt )
+{
+	if ( pEnt )
+	{
+		CBasePlayer *pLocalPlayer = UTIL_GetLocalPlayer();
+		if ( pLocalPlayer )
+		{			
+			Vector vForward;
+			pLocalPlayer->EyeVectors(&vForward );
+
+			UTIL_SetOrigin( pEnt, GetRandomSpot() );
+		}
+	}
+
+	return pEnt;
+}
+
+
+Vector GetRandomSpot()
+{
+	CWorld *pEnt = GetWorldEntity();
+	if ( pEnt )
+	{
+		Vector vMin, vMax;
+		pEnt->GetWorldBounds( vMin, vMax );
+		return Vector(
+			RandomFloat( vMin.x, vMax.x ),
+			RandomFloat( vMin.y, vMax.y ),
+			RandomFloat( vMin.z, vMax.z ) );
+	}
+	else
+	{
+		return Vector( 0, 0, 0 );
+	}
+}
+
+
+void Test_InitRandomEntitySpawner( const CCommand &args )
+{
+	// Put the list of registered functions into array form for convenience.
+	g_StressEntityRegs.Purge();
+	for ( CStressEntityReg *pCur=CStressEntityReg::GetListHead(); pCur; pCur=pCur->GetNext() )
+		g_StressEntityRegs.AddToTail( pCur );
+
+	// Create slots for all the entities..
+	int nSlots = 100;
+	if ( args.ArgC() >= 2 )
+		nSlots = atoi( args[ 1 ] );
+
+	g_StressEntities.Purge();
+	g_StressEntities.SetSize( nSlots );
+
+	Msg( "Test_InitRandomEntitySpawner: created %d slots.\n", nSlots );
+}
+
+
+void Test_SpawnRandomEntities( const CCommand &args )
+{
+	if ( args.ArgC() < 3 )
+	{
+		Error( "Test_SpawnRandomEntities <min # entities> <max # entities> missing arguments." );
+	}
+
+	if ( g_StressEntities.Count() == 0 )
+	{
+		Error( "Test_SpawnRandomEntities: not initialized (call Test_InitRandomEntitySpawner frst)." );
+	} 
+
+	int nMin = atoi( args[ 1 ] );
+	int nMax = atoi( args[ 2 ] );
+	int count = RandomInt( nMin, nMax );
+
+	for ( int i=0; i < count; i++ )
+	{
+		int iSlot = RandomInt( 0, g_StressEntities.Count() - 1 );
+
+		// Remove any old entity in this slot.
+		if ( g_StressEntities[iSlot].Get() )
+			UTIL_RemoveImmediate( g_StressEntities[iSlot] );
+
+		// Create a new one in this slot.
+		int iType = RandomInt( 0, g_StressEntityRegs.Count() - 1 );
+		g_StressEntities[iSlot] = g_StressEntityRegs[iType]->GetFn()();
+	}
+}
+
+
+void Test_RandomizeInPVS( const CCommand &args )
+{
+	if ( args.ArgC() < 2 )
+	{
+		Error( "Test_RandomizeInPVS <percentage chance to change>" );
+	}
+
+	int percent = atoi( args[ 1 ] );
+	for ( int i=0; i < g_StressEntities.Count(); i++ )
+	{
+		CBaseEntity *pEnt = g_StressEntities[i];
+
+		if ( pEnt )
+		{
+			if ( RandomInt( 0, 100 ) < percent )
+			{
+				if ( pEnt->IsEffectActive( EF_NODRAW ) )
+					pEnt->RemoveEffects( EF_NODRAW );
+				else
+					pEnt->AddEffects( EF_NODRAW );
+			}
+		}
+	}
+}
+
+
+void Test_RemoveAllRandomEntities()
+{
+	for ( int i=0; i < g_StressEntities.Count(); i++ )
+	{
+		if ( g_StressEntities[i].Get() )
+			UTIL_Remove( g_StressEntities[i] );
+	}
+}
+
+
+ConCommand cc_Test_InitRandomEntitySpawner( "Test_InitRandomEntitySpawner", Test_InitRandomEntitySpawner, 0, FCVAR_CHEAT );
+ConCommand cc_Test_SpawnRandomEntities( "Test_SpawnRandomEntities", Test_SpawnRandomEntities, 0, FCVAR_CHEAT );
+ConCommand cc_Test_RandomizeInPVS( "Test_RandomizeInPVS", Test_RandomizeInPVS, 0, FCVAR_CHEAT );
+ConCommand cc_Test_RemoveAllRandomEntities( "Test_RemoveAllRandomEntities", Test_RemoveAllRandomEntities, 0, FCVAR_CHEAT );
+
-- 
cgit v1.2.3