BasicAIModule/source/Managers/SupplyManager.cpp

Go to the documentation of this file.

/*
 *  SupplyManager.cpp
 *  
 *  Used to ensure SupplyDepot count keeps supply below capacity
 */
#include "SupplyManager.h"
#include "BuildManager.h"
#include "Strategizer.h"

#include <BWAPI.h>

using namespace BWAPI;


SupplyManager::SupplyManager()
    : depotCount(0)
        , plannedDepotCount(0)
{ }

void SupplyManager::update()
{
        /* Build supply if running low */
        int currentSupply = Broodwar->self()->supplyUsed();
        if (plannedSupply() - currentSupply < 12)
        {
                Broodwar->sendText("SM: Nearing capacity (%d/%d), Supply Depot ordered",
                        currentSupply, plannedSupply());
                plannedDepotCount++;
                Strategizer::instance().buildManager.build(UnitTypes::Terran_Supply_Depot, true);
        }

        /* Detect Supply Depot completion */
        int completedDepots = Broodwar->self()->completedUnitCount(UnitTypes::Terran_Supply_Depot);
        if (completedDepots > depotCount)
        {
                depotCount++;
                plannedDepotCount--;
        }

        /* Base class updates Agents */
        Manager::update();
}

int SupplyManager::plannedSupply() const
{
    // Note: supplyProvided() returns double the onscreen supply count
        const int plannedSupply = plannedDepotCount * UnitTypes::Terran_Supply_Depot.supplyProvided();
        const int totalSupply   = Broodwar->self()->supplyTotal();
        return plannedSupply + totalSupply;
}

void SupplyManager::draw()
{
        Broodwar->drawTextScreen(2, 10, "\x1E SM : %d planned", plannedSupply()/2);
        Manager::draw();
}