BTHAI/SCProjects/BTHAIModule/Source/UnitAgent.h

Go to the documentation of this file.

#ifndef __UNITAGENT_H__
#define __UNITAGENT_H__

#include <BWAPI.h>
#include "BaseAgent.h"
using namespace BWAPI;
using namespace std;

#define DISABLE_UNIT_AI 0

class UnitAgent : public BaseAgent {

protected:
        int currentState;
        static const int ATTACK = 1;
        static const int EXPLORE = 2;
        static const int DEFEND = 3;
        static const int BUNKERED = 4;
        
public:
        UnitAgent(Unit* mUnit);
        UnitAgent();

        int dropped;

        void computeActions();

        void computeKitingActions();

        string getTypeName();

        virtual void setGoal(TilePosition goal);

        virtual void clearGoal();

        int friendlyUnitsWithinRange();

        int friendlyUnitsWithinRange(int maxRange);

        int friendlyUnitsWithinRange(TilePosition tilePos, int maxRange);

        int enemyAttackingUnitsWithinRange();

        int enemyAttackingUnitsWithinRange(int maxRange, TilePosition center);

        int enemyAttackingUnitsWithinRange(UnitType type);

        int enemyUnitsWithinRange(int maxRange);

        int enemyGroundUnitsWithinRange(int maxRange);

        int enemySiegedTanksWithinRange(TilePosition center);

        int enemyGroundAttackingUnitsWithinRange(TilePosition center, int maxRange);

        int enemyAirUnitsWithinRange(int maxRange);

        bool useDefensiveMode();

        int enemyAirToGroundUnitsWithinRange(int maxRange);

        int enemyAirAttackingUnitsWithinRange(TilePosition center, int maxRange);

        int getGroundRange();

        static int getGroundRange(UnitType type);

        static bool canAttack(UnitType attacker, UnitType target);

        int getAirRange();

        static int getAirRange(UnitType type);

        void assignToExplore();

        void assignToAttack();

        void assignToDefend();

        bool isExploring();

        bool isAttacking();

        bool isDefending();

        bool isBunkered();
        
        Unit* getClosestOrganicEnemy(int maxRange);

        Unit* getClosestShieldedEnemy(int maxRange);

        Unit* getClosestEnemyTurret(int maxRange);

        Unit* getClosestEnemyAirDefense(int maxRange);

        bool chargeShields();

        virtual void printInfo();
};

#endif