BTHAI/SCProjects/BTHAIModule/Source/Commander.h

Go to the documentation of this file.

#ifndef __COMMANDER_H__
#define __COMMANDER_H__

#include "Squad.h"
#include "BaseAgent.h"

using namespace BWAPI;
using namespace BWTA;
using namespace std;

struct SortSquadList {
        bool operator()(Squad*& sq1, Squad*& sq2) {
                if (sq1->getPriority() != sq2->getPriority()) {
                        return sq1->getPriority() < sq2->getPriority();
                }
                else {
                        if (sq1->isRequired() && !sq2->isRequired()) return true;
                        else return false;
                }
        }
};

class Commander {

private:
        bool chokePointFortified(TilePosition center);
        bool chokePointGuarded(TilePosition center);
        void sortSquadList();
        bool isOccupied(Region* region);
        bool isEdgeChokepoint(Chokepoint* choke);
        double getDistToBase(Chokepoint* choke);
        TilePosition findDefensePos(Chokepoint* choke);
        double getChokepointPrio(TilePosition center);

        void checkNoSquadUnits();
        void assignUnit(BaseAgent* agent);

        int lastCallFrame;

protected:
        vector<Squad*> squads;
        int currentID;
        static Commander* instance;

        int currentState;
        static const int DEFEND = 0;
        static const int ATTACK = 1;

        Commander();

        TilePosition findOffensiveSquadPosition(TilePosition closeEnemy);

public:
        ~Commander();

        static Commander* getInstance();

        void computeActions();

        bool shallEngage();

        void updateGoals();

        void unitCreated(BaseAgent* agent);

        void unitDestroyed(BaseAgent* agent);

        /* Checks if the specified unittype needs to be built. */
        bool needUnit(UnitType type);

        Squad* getSquad(int id);

        vector<Squad*> getSquads();

        TilePosition getClosestEnemyBuilding(TilePosition start);
        
        int noOffensiveSquadsWithin(TilePosition center, int maxRange);

        bool checkWorkersAttack(BaseAgent* base);

        void assistBuilding(BaseAgent* building);

        void assistWorker(BaseAgent* worker);

        int enemyUnitsWithinRange(TilePosition start, int range);

        void handleCloakedEnemy(TilePosition pos, Squad* squad);

        void checkRemovableObstacles();

        void forceAttack();

        void printInfo();

        TilePosition findUnfortifiedChokePoint();
        
        TilePosition findChokePoint();

        bool isPowered(TilePosition pos);

        bool isBuildable(TilePosition pos);

        bool checkUnfinishedBuildings();

        bool checkRepairUnits();

        bool isImportantUnit(Unit* unit);

        void repair(BaseAgent* agent);

        void finishBuild(BaseAgent* agent);

        int ownDeadScore;

        int enemyDeadScore;

        void addBunkerSquad();
};

#endif