BTHAI/SCProjects/BTHAIModule/Source/ExplorationManager.h

Go to the documentation of this file.

#ifndef __EXPLORATIONMANAGER_H__
#define __EXPLORATIONMANAGER_H__

#include "UnitAgent.h"
#include "SpottedObject.h"
#include "Squad.h"

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

struct ForceData {
        int airAttackStr;
        int airDefendStr;
        int groundAttackStr;
        int groundDefendStr;

        int noRefineries;
        int noCommandCenters;
        int noFactories;
        int noAirports;
        int noBarracks;
        int noDefenseStructures;
        int noDetectorStructures;
        int noDetectorUnits;

        void reset() {
                airAttackStr = 0;
                airDefendStr = 0;
                groundAttackStr = 0;
                groundDefendStr = 0;

                noRefineries = 0;
                noCommandCenters = 0;
                noFactories = 0;
                noAirports = 0;
                noBarracks = 0;
                noDefenseStructures = 0;
                noDetectorStructures = 0;
                noDetectorUnits = 0;
        }

        void checkType(UnitType type) {
                if (type.isRefinery()) {
                        noRefineries++;
                }
                if (type.isResourceDepot()) {
                        noCommandCenters++;
                }
                if (type.isBuilding() && type.isDetector()) {
                        noDetectorStructures++;
                }
                if (!type.isBuilding() && type.isDetector()) {
                        noDetectorUnits++;
                }
                if (type.isBuilding() && type.canAttack()) {
                        noDefenseStructures++;
                }
                if (type.getID() == UnitTypes::Terran_Bunker.getID()) {
                        noDefenseStructures++;
                }
                if (type.getID() == UnitTypes::Terran_Starport.getID() || type.getID() == UnitTypes::Protoss_Stargate.getID()) {
                        noAirports++;   
                }
                if (type.getID() == UnitTypes::Terran_Barracks.getID() || type.getID() == UnitTypes::Protoss_Gateway.getID()) {
                        noBarracks++;
                }
                if (type.getID() == UnitTypes::Terran_Factory.getID() || type.getID() == UnitTypes::Protoss_Gateway.getID() || type.getID() == UnitTypes::Protoss_Robotics_Facility.getID()) {
                        noFactories++;
                }
        }
};

struct ExploreData {
        TilePosition center;
        int lastVisitFrame;
        
        ExploreData(Position tCenter) {
                center = TilePosition(tCenter);
                lastVisitFrame = 0;
        }

        int getX() {
                return center.x();
        }

        int getY() {
                return center.y();
        }

        bool matches(Region* region) {
                if (region == NULL) {
                        return false;
                }
                TilePosition tCenter = TilePosition(region->getCenter());
                return matches(tCenter);
        }

        bool matches(TilePosition tCenter) {
                double dist = tCenter.getDistance(center);
                if (dist <= 2) {
                        return true;
                }
                return false;
        }
};

class ExplorationManager {

private:
        vector<SpottedObject*> spottedBuildings;
        vector<SpottedObject*> spottedUnits;
        
        vector<ExploreData> exploreData;
        int getLastVisitFrame(Region* region);
        
        ForceData ownForce;
        ForceData enemyForce;

        ExplorationManager();
        static ExplorationManager* instance;
        static bool instanceFlag;
        
        void calcEnemyForceData();
        void calcOwnForceData();

        void cleanup();

        bool active;

        int lastCallFrame;

        int siteSetFrame;
        TilePosition expansionSite;

public:
        ~ExplorationManager();

        static ExplorationManager* getInstance();

        void setInactive();

        bool isActive();

        void computeActions();

        TilePosition getNextToExplore(Squad* squad);

        TilePosition searchExpansionSite();

        TilePosition getExpansionSite();

        void setExpansionSite(TilePosition pos);

        void printInfo();

        void addSpottedUnit(Unit* unit);

        void unitDestroyed(Unit* unit);

        vector<SpottedObject*> getSpottedBuildings();

        TilePosition getClosestSpottedBuilding(TilePosition start);

        int spottedBuildingsWithinRange(TilePosition pos, int range);

        bool buildingsSpotted();

        void showIntellData();

        static bool canReach(TilePosition a, TilePosition b);

        static bool canReach(BaseAgent* agent, TilePosition b);

        void setExplored(TilePosition goal);

        TilePosition scanForVulnerableBase();

        bool isDetectorCovering(TilePosition pos);

        bool isDetectorCovering(Position pos);

        static bool enemyIsProtoss();

        static bool enemyIsZerg();

        static bool enemyIsTerran();

        static bool enemyIsUnknown();
};

#endif