blob: 974fb7d4718e4ebdaf32edb38b3bdf992e68b414 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//=============================================================================//
#ifndef PLAYERENUMERATOR_H
#define PLAYERENUMERATOR_H
#ifdef _WIN32
#pragma once
#endif
#include "utlvector.h"
#include "ehandle.h"
#include "ispatialpartition.h"
class C_BaseEntity;
class CPlayerEnumerator : public IPartitionEnumerator
{
DECLARE_CLASS_NOBASE( CPlayerEnumerator );
public:
//Forced constructor
CPlayerEnumerator( float radius, Vector vecOrigin )
{
m_flRadiusSquared = radius * radius;
m_vecOrigin = vecOrigin;
m_Objects.RemoveAll();
}
int GetObjectCount() { return m_Objects.Size(); }
C_BaseEntity *GetObject( int index )
{
if ( index < 0 || index >= GetObjectCount() )
return NULL;
return m_Objects[ index ];
}
//Actual work code
virtual IterationRetval_t EnumElement( IHandleEntity *pHandleEntity )
{
C_BaseEntity *pEnt = ClientEntityList().GetBaseEntityFromHandle( pHandleEntity->GetRefEHandle() );
if ( pEnt == NULL )
return ITERATION_CONTINUE;
if ( !pEnt->IsPlayer() )
return ITERATION_CONTINUE;
Vector deltaPos = pEnt->GetAbsOrigin() - m_vecOrigin;
if ( deltaPos.LengthSqr() > m_flRadiusSquared )
return ITERATION_CONTINUE;
CHandle< C_BaseEntity > h;
h = pEnt;
m_Objects.AddToTail( h );
return ITERATION_CONTINUE;
}
public:
//Data members
float m_flRadiusSquared;
Vector m_vecOrigin;
CUtlVector< CHandle< C_BaseEntity > > m_Objects;
};
#endif // PLAYERENUMERATOR_H
|