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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
#include "CIntel.h"
#include "CAI.h"
#include "CUnitTable.h"
#include "CUnit.h"
#include "CMilitary.h"
CIntel::CIntel(AIClasses *ai) {
this->ai = ai;
units = &ai->unitIDs[0]; // save about 4x32KB of memory
selector.push_back(ASSAULT);
selector.push_back(SCOUTER);
selector.push_back(SNIPER);
selector.push_back(ARTILLERY);
selector.push_back(ANTIAIR);
selector.push_back(AIR);
for (size_t i = 0; i < selector.size(); i++)
counts[selector[i]] = 1;
numUnits = 1;
}
float3 CIntel::getEnemyVector() {
return enemyvector;
}
void CIntel::init() {
numUnits = ai->cbc->GetEnemyUnits(units, MAX_UNITS);
assert(numUnits > 0);
enemyvector = float3(0.0f, 0.0f, 0.0f);
for (int i = 0; i < numUnits; i++) {
enemyvector += ai->cbc->GetUnitPos(units[i]);
}
enemyvector /= numUnits;
LOG_II("Number of enemies: " << numUnits)
}
void CIntel::update(int frame) {
mobileBuilders.clear();
factories.clear();
attackers.clear();
metalMakers.clear();
energyMakers.clear();
restUnarmedUnits.clear();
rest.clear();
resetCounters();
numUnits = ai->cbc->GetEnemyUnits(units, MAX_UNITS);
for (int i = 0; i < numUnits; i++) {
const UnitDef *ud = ai->cbc->GetUnitDef(units[i]);
UnitType *ut = UT(ud->id);
unsigned int c = ut->cats;
bool armed = !ud->weapons.empty();
/* Ignore units being built and cloaked units */
if ((ai->cbc->UnitBeingBuilt(units[i]) && (armed || !(c&STATIC)))
|| ai->cbc->IsUnitCloaked(units[i]))
continue;
if (c&ATTACKER && !(c&AIR)) {
attackers.push_back(units[i]);
}
else if (c&FACTORY) {
factories.push_back(units[i]);
}
else if (c&BUILDER && c&MOBILE && !(c&AIR)) {
mobileBuilders.push_back(units[i]);
}
else if (c&MEXTRACTOR || c&MMAKER) {
metalMakers.push_back(units[i]);
}
else if (c&EMAKER) {
energyMakers.push_back(units[i]);
}
else if (!armed)
restUnarmedUnits.push_back(units[i]);
else {
rest.push_back(units[i]);
}
if (c&ATTACKER && c&MOBILE)
updateCounts(c);
}
}
unitCategory CIntel::counter(unitCategory c) {
switch(c) {
case ASSAULT: return SNIPER;
case SCOUTER: return ASSAULT;
case SNIPER: return ARTILLERY;
case ARTILLERY: return ASSAULT;
case ANTIAIR: return ARTILLERY;
case AIR: return ANTIAIR;
default: return ARTILLERY;
}
}
void CIntel::updateCounts(unsigned c) {
for (size_t i = 0; i < selector.size(); i++) {
if (selector[i] & c) {
// TODO: counts[ai->cfgparser->counter(selector[i])]++
counts[counter(selector[i])]++;
totalCount++;
}
}
}
void CIntel::resetCounters() {
roulette.clear();
/* Put the counts in a normalized reversed map first and reset counters*/
for (size_t i = 0; i < selector.size(); i++) {
roulette.insert(std::pair<float,unitCategory>(counts[selector[i]]/float(totalCount), selector[i]));
counts[selector[i]] = 1;
}
counts[ANTIAIR] = 0;
counts[AIR] = 0;
counts[ASSAULT] = 3;
if(ai->military->idleScoutGroupsNum() >= MAX_IDLE_SCOUT_GROUPS)
counts[SCOUTER] = 0;
else
counts[SCOUTER] = 1; // default value
totalCount = selector.size();
}
bool CIntel::enemyInbound() {
return false;
}
|
