BWAPI: Aiur/include/UpgradeManager.h Source File

Go to the documentation of this file.
00001 #pragma once
00002 #include <Arbitrator.h>
00003 #include <BWAPI.h>
00004 #include <BuildingPlacer.h>
00005 class UpgradeManager : public Arbitrator::Controller<BWAPI::Unit*,double>
00006 {
00007 public:
00008         class Upgrade
00009         {
00010         public:
00011                 BWAPI::UpgradeType type;
00012                 int level;
00013         };
00014         UpgradeManager(Arbitrator::Arbitrator<BWAPI::Unit*,double>* arbitrator);
00015         void setBuildingPlacer(BuildingPlacer* placer);
00016         virtual void onOffer(std::set<BWAPI::Unit*> units);
00017         virtual void onRevoke(BWAPI::Unit* unit, double bid);
00018         virtual void update();
00019         virtual std::string getName() const;
00020         void onRemoveUnit(BWAPI::Unit* unit);
00021         bool upgrade(BWAPI::UpgradeType type, int level = -1);
00022         int getPlannedLevel(BWAPI::UpgradeType type) const;
00023         int getStartedLevel(BWAPI::UpgradeType type) const;
00024         int getCompletedLevel(BWAPI::UpgradeType type) const;
00025 
00026 private:
00027         Arbitrator::Arbitrator<BWAPI::Unit*,double>* arbitrator;
00028         BuildingPlacer* placer;
00029         std::map<BWAPI::UnitType,std::list< Upgrade > > upgradeQueues;
00030         std::map<BWAPI::Unit*,Upgrade> upgradingUnits;
00031         std::map<BWAPI::UpgradeType, int> plannedLevel;
00032         std::map<BWAPI::UpgradeType, int> startedLevel;
00033         int lastFrameCheck;
00034 };