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
|
#ifndef KAIK_THREATMAP_HDR
#define KAIK_THREATMAP_HDR
#include <map>
#include <vector>
struct AIClasses;
class CThreatMap {
public:
CR_DECLARE(CThreatMap)
CThreatMap(AIClasses* ai);
~CThreatMap();
void PostLoad();
void Update();
void EnemyCreated(int);
void EnemyFinished(int);
void EnemyDamaged(int, int);
void EnemyDestroyed(int, int);
float GetAverageThreat() const { return (currAvgThreat + 1.0f); }
float ThreatAtThisPoint(const float3&) const;
float* GetThreatArray() { return &threatCellsRaw[0]; }
int GetThreatMapWidth() const { return width; }
int GetThreatMapHeight() const { return height; }
void ToggleVisOverlay();
private:
struct EnemyUnit {
bool operator < (const EnemyUnit& e) const {
return (id < e.id);
}
int id;
float3 pos;
float threat;
float range;
};
void AddEnemyUnit(const EnemyUnit&, const float s);
void DelEnemyUnit(const EnemyUnit&);
float GetEnemyUnitThreat(const EnemyUnit&) const;
float currAvgThreat;
float currMaxThreat;
float currSumThreat;
int area;
int width;
int height;
int threatMapTexID;
std::map<int, EnemyUnit> enemyUnits;
std::vector<float> threatCellsRaw;
std::vector<float> threatCellsVis;
AIClasses* ai;
};
#endif
|