BaczekKPAI.h

// GroupAI.h: interface for the CGroupAI class.
// Dont modify this file
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_GroupAI_H__10718E36_5CDF_4CD4_8D90_F41311DD2694__INCLUDED_)
#define AFX_GroupAI_H__10718E36_5CDF_4CD4_8D90_F41311DD2694__INCLUDED_

#pragma once

#include <map>
#include <set>
#include <vector>


#include "ExternalAI/IGlobalAI.h"
#include "ExternalAI/IAICallback.h"
#include "ExternalAI/IAICheats.h"


#include "GUI/StatusFrame.h"
#include "InfluenceMap.h"
#include "PythonScripting.h"
#include "TopLevelAI.h"


using namespace std;

class Log;
class Unit;

class BaczekKPAI : public IGlobalAI  
{
public:
	static const char AI_NAME[];
	static const char AI_VERSION[];

	BaczekKPAI();
	virtual ~BaczekKPAI();

	void InitAI(IGlobalAICallback* callback, int team);

	void UnitCreated(int unit, int builder);					//called when a new unit is created on ai team
	void UnitFinished(int unit);								//called when an unit has finished building
	void UnitDestroyed(int unit,int attacker);								//called when a unit is destroyed

	void EnemyEnterLOS(int enemy);
	void EnemyLeaveLOS(int enemy);

	void EnemyEnterRadar(int enemy);				
	void EnemyLeaveRadar(int enemy);				

	void EnemyDamaged(int damaged,int attacker,float damage,float3 dir);	//called when an enemy inside los or radar is damaged
	void EnemyDestroyed(int enemy,int attacker);							//will be called if an enemy inside los or radar dies (note that leave los etc will not be called then)

	void UnitIdle(int unit);										//called when a unit go idle and is not assigned to any group

	void GotChatMsg(const char* msg,int player);					//called when someone writes a chat msg

	void UnitDamaged(int damaged,int attacker,float damage,float3 dir);					//called when one of your units are damaged

	void UnitMoveFailed(int unit);
	int HandleEvent (int msg,const void *data);

	//called every frame
	void Update();

	void DumpStatus();
	void FindGeovents();

	IGlobalAICallback* callback;
	IAICallback* cb;
	IAICheats* cheatcb;

	set<int> myUnits;
	set<int> losEnemies;

	// oldEnemies is used to find units that changed
	vector<int> oldEnemies;
	vector<int> allEnemies;
	vector<int> friends;

	// units
	Unit* unitTable[MAX_UNITS];

	virtual void Load(IGlobalAICallback* callback,std::ifstream *ifs){};
	virtual void Save(std::ifstream *ifs){};

	const char *datadir;
	const char *statusName;

	struct MapInfo {
		int w, h;
		int squareSize;
	};
	MapInfo map;

	vector<float3> geovents;
	set<int> enemyBases;

	InfluenceMap *influence;
	PythonScripting *python;

	TopLevelAI* toplevel;

	bool debugLines;
	bool debugMsgs;

#ifdef USE_STATUS_WINDOW
	MyFrame *frame;
#endif

	std::vector<const UnitDef*> unitDefById;
	void InitializeUnitDefs();
	const UnitDef* GetUnitDefById(int id) { return unitDefById[id]; }


	Unit* GetUnit(int id) { return unitTable[id]; }
	
	void GetAllUnitsInRadius(std::vector<int>& vec, float3 pos, float radius);

	float EstimateSqDistancePF(int unitID, const float3& start, const float3& end);
	float EstimateSqDistancePF(const UnitDef* unitdef, const float3& start, const float3& end);
	// slower versions for precise distance
	float EstimateDistancePF(int unitID, const float3& start, const float3& end);
	float EstimateDistancePF(const UnitDef* unitdef, const float3& start, const float3& end);

	// heightmap
	float GetGroundHeight(float x, float y);

	// easier spatial queries
	void GetEnemiesInRadius(float3 pos, float radius, std::vector<int>& output)
	{
		int enemies[MAX_UNITS];
		int numenemies;
		numenemies = cheatcb->GetEnemyUnits(enemies, pos, radius);
		output.clear();
		output.reserve(numenemies);
		for (int i=0; i<numenemies; ++i)
			output.push_back(enemies[i]);
	}

	std::string GetRoleUnitName(const char* role)
	{
		const char* side = cb->GetTeamSide(cb->GetMyTeam());
		std::string configval = (std::string(side)+"_"+role);
		return python->GetStringValue(configval.c_str(), std::string());
	}

	void SendTextMsg(const char *msg, int zone)
	{
		if (debugMsgs)
			cb->SendTextMsg(msg, zone);
	}

	int CreateLineFigure(float3 pos1, float3 pos2, float width, int arrow, int lifeTime, int figureGroupId)
	{
		if (debugLines)
			return cb->CreateLineFigure(pos1, pos2, width, arrow, lifeTime, figureGroupId);
		else return 0;
	}
};

#endif // !defined(AFX_GroupAI_H__10718E36_5CDF_4CD4_8D90_F41311DD2694__INCLUDED_)