upload "kind" alien swarm

game/shared/swarm/asw_scanner_objects_enumerator.cpp

@@ -0,0 +1,108 @@
+#include "cbase.h"
+#include "asw_scanner_objects_enumerator.h"
+
+#ifdef CLIENT_DLL
+	#include "c_ai_basenpc.h"
+	#include "c_asw_alien.h"
+	#include "c_asw_buzzer.h"
+	#include "c_asw_door.h"
+	#include "c_asw_grub.h"
+	#include "c_asw_computer_area.h"
+	#include "c_asw_button_area.h"
+	#define CASW_Buzzer C_ASW_Buzzer
+	#define CASW_Alien C_ASW_Alien
+	#define CASW_Door C_ASW_Door
+	#define CASW_Grub C_ASW_Grub
+	#define CASW_Computer_Area C_ASW_Computer_Area
+	#define CASW_Button_Area C_ASW_Button_Area
+#else
+	#include "ai_basenpc.h"
+	#include "asw_alien.h"
+	#include "asw_buzzer.h"
+	#include "asw_door.h"	
+	#include "asw_grub.h"
+	#include "asw_computer_area.h"
+	#include "asw_button_area.h"
+#endif
+
+// memdbgon must be the last include file in a .cpp file!!!
+#include "tier0/memdbgon.h"
+
+namespace
+{
+	bool ValidScannerObject(CBaseEntity *pEnt)
+	{
+		// dron't show grubs, they're too small
+		CASW_Grub *pGrub = dynamic_cast<CASW_Grub*>(pEnt);
+		if (pGrub)
+			return false;
+		CASW_Alien *pAlien = dynamic_cast<CASW_Alien*>(pEnt);
+		if (pAlien)
+			return pAlien->GetHealth() > 0;
+		CASW_Door *pDoor = dynamic_cast<CASW_Door*>(pEnt);
+		if (pDoor && pDoor->IsMoving())
+			return true;
+		CASW_Buzzer *pBuzzer = dynamic_cast<CASW_Buzzer*>(pEnt);
+		if (pBuzzer)
+			return pBuzzer->GetHealth() > 0;
+		
+		CASW_Computer_Area *pComputer = dynamic_cast<CASW_Computer_Area*>(pEnt);
+		if (pComputer)
+			return true;
+		CASW_Button_Area *pButton = dynamic_cast<CASW_Button_Area*>(pEnt);
+		if (pButton)
+			return true;
+		return false;
+	}
+}
+
+// Enumator class for ragdolls being affected by explosive forces
+CASW_Scanner_Objects_Enumerator::CASW_Scanner_Objects_Enumerator( float radius, Vector &vecCenter ) :
+	m_vecScannerCenter(vecCenter)
+{
+	m_flRadius = radius;
+	m_flRadiusSquared = radius * radius;
+	m_Objects.RemoveAll();
+}
+
+int	CASW_Scanner_Objects_Enumerator::GetObjectCount()
+{
+	return m_Objects.Count();
+}
+
+CBaseEntity *CASW_Scanner_Objects_Enumerator::GetObject( int index )
+{
+	if ( index < 0 || index >= GetObjectCount() )
+		return NULL;
+
+	return m_Objects[ index ];
+}
+
+// Actual work code
+IterationRetval_t CASW_Scanner_Objects_Enumerator::EnumElement( IHandleEntity *pHandleEntity )
+{
+#ifndef CLIENT_DLL
+		CBaseEntity *pEnt = gEntList.GetBaseEntity( pHandleEntity->GetRefEHandle() );
+#else
+		C_BaseEntity *pEnt = ClientEntityList().GetBaseEntityFromHandle( pHandleEntity->GetRefEHandle() );
+#endif
+	
+	if ( pEnt == NULL )
+		return ITERATION_CONTINUE;
+
+	if (!ValidScannerObject(pEnt))
+		return ITERATION_CONTINUE;	
+
+	Vector deltaPos = pEnt->GetAbsOrigin() - m_vecScannerCenter;
+	if ( deltaPos.LengthSqr() > m_flRadiusSquared )
+		return ITERATION_CONTINUE;
+
+	if ( m_vecScannerCenter.z - pEnt->GetAbsOrigin().z > m_flRadius / 3 )
+		return ITERATION_CONTINUE;
+	
+	CHandle< CBaseEntity > h;
+	h = pEnt;
+	m_Objects.AddToTail( h );
+
+	return ITERATION_CONTINUE;
+}