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
72
73
74
75
76
77
78
79
80
81
|
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */
#ifndef UNITHANDLER_H
#define UNITHANDLER_H
#include <vector>
#include "UnitDef.h"
#include "UnitSet.h"
#include "Sim/Misc/SimObjectIDPool.h"
#include "System/creg/STL_Map.h"
#include "System/creg/STL_List.h"
class CUnit;
class CBuilderCAI;
class CUnitHandler
{
CR_DECLARE_STRUCT(CUnitHandler)
public:
CUnitHandler();
~CUnitHandler();
void Update();
void DeleteUnit(CUnit* unit);
void DeleteUnitNow(CUnit* unit);
bool AddUnit(CUnit* unit);
void PostLoad();
bool CanAddUnit(int id) const {
// do we want to be assigned a random ID and are any left in pool?
if (id < 0)
return (!idPool.IsEmpty());
// is this ID not already in use?
if (id < MaxUnits())
return (units[id] == NULL);
// AddUnit will not make new room for us
return false;
}
unsigned int MaxUnits() const { return maxUnits; }
float MaxUnitRadius() const { return maxUnitRadius; }
/// Returns true if a unit of type unitID can be built, false otherwise
bool CanBuildUnit(const UnitDef* unitdef, int team) const;
void AddBuilderCAI(CBuilderCAI*);
void RemoveBuilderCAI(CBuilderCAI*);
// note: negative ID's are implicitly converted
CUnit* GetUnitUnsafe(unsigned int unitID) const { return units[unitID]; }
CUnit* GetUnit(unsigned int unitID) const { return (unitID < MaxUnits()? units[unitID]: NULL); }
std::vector<CUnit*> units; ///< used to get units from IDs (0 if not created)
std::vector< std::vector<CUnitSet> > unitsByDefs; ///< units sorted by team and unitDef
std::list<CUnit*> activeUnits; ///< used to get all active units
std::map<unsigned int, CBuilderCAI*> builderCAIs;
private:
void InsertActiveUnit(CUnit* unit);
private:
SimObjectIDPool idPool;
std::vector<CUnit*> unitsToBeRemoved; ///< units that will be removed at start of next update
std::list<CUnit*>::iterator activeSlowUpdateUnit; ///< first unit of batch that will be SlowUpdate'd this frame
///< global unit-limit (derived from the per-team limit)
///< units.size() is equal to this and constant at runtime
unsigned int maxUnits;
///< largest radius of any unit added so far (some
///< spatial query filters in GameHelper use this)
float maxUnitRadius;
};
extern CUnitHandler* unitHandler;
#endif /* UNITHANDLER_H */
|
