ExampleAIModule/Source/InformationManager.h

Go to the documentation of this file.

#pragma once
#include <BWAPI.h>
#include <BWTA.h>
#include <windows.h>
#include "Common.h"

#define SCANNER_SWEEP_FREQUENCY 30

DWORD WINAPI AnalyzeMapThread();

typedef std::map<BWAPI::UnitType, BWAPI::UnitType> BuildingToUnitMap;
typedef std::map<BWAPI::UnitType, std::vector<BWAPI::TechType> > BuildingToTechMap;
typedef std::map<BWAPI::UnitType, std::vector<BWAPI::UpgradeType> > BuildingToUpgradeMap;
typedef std::map<BWAPI::UnitType, std::vector<BWAPI::UnitType> > BuildingToAddonMap;
typedef std::map<BWAPI::Unit*, BWAPI::UnitType> UnitToTypeMap;
typedef std::map<BWAPI::Unit*, int> UnitToTimeMap;

class InformationManager
{
public:
        InformationManager();
        ~InformationManager();
        void analyzeMap();

        // Economical information
        void reserveMinerals(int minerals);
        void removeReservedMinerals(int minerals);
        void reserveGas(int gas);
        void removeReservedGas(int gas);
        int minerals();
        int gas();
        bool haveResources(BWAPI::UnitType type);
        bool haveResources(BWAPI::TechType type);
        bool haveResources(BWAPI::UpgradeType type);
        void frameSpend(BWAPI::UnitType type);
        void frameSpend(BWAPI::TechType type);
        void frameSpend(BWAPI::UpgradeType type);
        int _mineralsReserved;
        int _gasReserved;
        int _frameMineralSpend;
        int _frameGasSpend;

        // Build request
        void buildRequest(BWAPI::UnitType type);
        void buildRequest(BWAPI::UnitType type, bool checkUnic);
        void criticalBuildRequest(BWAPI::UnitType type);
        void criticalBuildRequest(BWAPI::UnitType type, bool first);
        void addonRequest(BWAPI::UnitType addonType);
        void researchRequest(BWAPI::TechType techType);
        void upgradeRequest(BWAPI::UpgradeType upgradeType);
        void upgradeRequest(BWAPI::UpgradeType upgradeType, int level);
        // Base
        enum Expand
        {
                Natural,
                Gas
        };
        void naturalExpandRequest(bool critical = false);
        void gasExpandRequest(bool critical = false);
        void onCommandCenterShow(BWAPI::Unit* unit);
        void onEnemyResourceDepotShow(BWAPI::Unit* unit);
        void onCommandCenterDestroy(BWAPI::Unit* unit);
        void onEnemyResourceDepotDestroy(BWAPI::Unit* unit);
        BWAPI::TilePosition getNaturalExpandPosition(BWAPI::TilePosition ignorePosition = BWAPI::TilePositions::None);
        BWAPI::TilePosition getGasExpandPosition(BWAPI::TilePosition ignorePosition = BWAPI::TilePositions::None);

        // enemy DPS map:
        double get_enemy_air_dps(BWAPI::TilePosition, int cycle);
        double get_enemy_ground_dps(BWAPI::TilePosition, int cycle);
        double get_enemy_air_dps(int x, int y, int cycle);
        double get_enemy_ground_dps(int x, int y, int cycle);
        void recompute_enemy_dps(int cycle);
        void drawAirDPSMap();
        void drawGroundDPSMap();

        // Military information
        UnitSet _seenEnemies;
        UnitToTypeMap _enemiesType;
        unsigned int _minSquadSize;
        bool _needAntiAirUnits;
        bool _retreatDisabled;
        UnitToTimeMap _tankSiege;
                // Self army
                double _marines, _medics, _firebats, _ghosts;
                double _vultures, _tank, _goliath;
                double _wraiths;
                // Enemy Terran army
                double _enemyMarines, _enemyMedics, _enemyFirebats, _enemyGhosts;
                double _enemyVultures, _enemyTank, _enemyGoliath;
                double _enemyWraiths, _enemyDropship, _enemyScienceVessel, _enemyBattlecruiser, _enemyValkyrie;
                double _enemyTurrets;
                // Enemy Protoss army
                double _enemyZealot, _enemyDragoon, _enemyHTemplar, _enemyDTemplar, _enemyArchon, _enemyDArchon;
                double _enemyReaver, _enemyObserver, _enemyShuttle;
                double _enemyScout, _enemyCarrier, _enemyArbiter, _enemyCorsair;
                double _enemyPhotonCanon;
                // Enemy Zerg army
                double _enemyZergling, _enemyHydralisk, _enemyOverlord;
                double _enemyLurker, _enemyMutalisk;
                double _enemyUltralisk, _enemyGuardian, _enemyDevourer;
                double _enemySunkenColony, _enemySporeColony;
                // stats
                double _ourAirDPS, _ourAntiAirHP, _ourGroundDPS, _ourAirHP, _ourGroundHP;
                double _armySize;
                double _enemyAirDPS, _enemyAntiAirHP, _enemyGroundDPS, _enemyAirHP, _enemyGroundHP;
                double _enemyArmySize;

        // Production information
        BuildingToUnitMap _trainOrder;
        std::vector<BWAPI::UnitType> _buildRequest;
        std::vector<BWAPI::UnitType> _criticalBuildRequest;
        BuildingToAddonMap _addonOrder;
        BuildingToTechMap _researchOrder;
        BuildingToUpgradeMap _upgradeOrder;
        UnitToPercent _percentList;
        void checkRequirements(BWAPI::UnitType type);
        bool _autoVehicleUpgrade;
        bool _autoShipUpgrade;
        bool _autoInfanteryUpgrade;

        // Events information
        void seenCloakedEnemy(BWAPI::TilePosition pos);
        TilePositionSet _cloakedEnemyPositions;
        bool _turretDefenses;
        //bool _enemyWorkerScout;
        bool _scienceVesselDetector;

        // Base and Buildings
        Expand _expandType;
        std::map<BWTA::BaseLocation*, BWAPI::TilePosition> _ourBases;
        std::map<BWTA::BaseLocation*, BWAPI::Unit*> _basesUnderConstruction;
        std::set<BWTA::BaseLocation*> _emptyBases;
        std::set<BWTA::BaseLocation*> _enemyBases;
        std::set<BWAPI::TilePosition> _ignoreBases;
        BWAPI::TilePosition _proxyPosition;
        bool _useProxyPosition;
        void onMissileTurretShow(BWAPI::Unit* unit);
        void onMissileTurretDestroy(BWAPI::Unit* unit);
        std::map<BWAPI::Unit*, BWTA::Region*> _missileTurrets;
        bool _autoBuildSuplies;
        bool _priorCommandCenters;

        // Map information //TODO revise this information
        BWAPI::Position _enemyStartPosition;
        bool mapAnalyzed;
        BWTA::Region* home;
        std::set<BWTA::BaseLocation*> startLocationCouldContainEnemy;
        bool _scoutedAnEnemyBase;
        BWTA::Chokepoint* _choke;

        // Debug variables
        BWAPI::Position _topLeft;
        BWAPI::Position _bottomRight;
        BWAPI::Position _center;
        int _radius;

        // Enemy DPS map: (stores, for each cell of the map, the aggregated dps that the seen units of the enemy can do to our units in that cell)
        int _last_cycle_dps_map_was_recomputed;
        double *_ground_enemy_dps_map;
        double *_air_enemy_dps_map;


        // TODO move to scoutManager
        UnitToTimeMap _comsatStation;
        void scanBases();
        bool useScanner(BWAPI::Unit* comsat, BWAPI::Position pos);
        std::set<BWTA::BaseLocation*>::iterator _lastBaseScan;
        std::set<BWAPI::TilePosition>::iterator _lastIgnoreBaseScan;
};