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
|
#ifndef CDEFENSEMATRIX_H
#define CDEFENSEMATRIX_H
#include <map>
#include "headers/Defines.h"
#include "headers/HEngine.h"
class AIClasses;
class UnitType;
class CUnit;
class CDefenseMatrix {
public:
CDefenseMatrix(AIClasses *ai);
~CDefenseMatrix() {}
/* Determine all clusters currently ingame */
void update();
/* Get the spot that needs defense the most */
float3 getDefenseBuildSite(UnitType *tower);
/* Get clusters >= 2 */
int getClusters();
/* Get the nth best defended pos */
float3 getBestDefendedPos(int n);
private:
/* A group with spacing s between each building such that s < n */
struct Cluster {
Cluster() {
value = 0.0f;
center = ZEROVECTOR;
defenses = 0;
}
float value;
float3 center;
int defenses;
std::multimap<float, CUnit*> members;
};
int X, Z;
const float *hm;
/* Total clustervalue */
float totalValue;
/* The clusters, sorted on importance */
std::multimap<float, Cluster*> clusters;
/* The building to cluster table */
std::map<int, Cluster*> buildingToCluster;
/* The static buildings */
std::map<int, CUnit*> buildings;
AIClasses *ai;
float getValue(CUnit *building);
void draw();
};
#endif
|
