DerekSchaefer.NET I do stuff, you read about it!

Starcraft AI: Misbehaving Workers

Since we changed the official tournament map to The Fastest Possible Map V.2, the built in Worker Manager has not been gathering resources correctly (or at all, really). The problem arises from the map breaking Starcraft layout and positioning rules in order to be as fast as possible; minerals and geysers overlap, bases are placed impossibly close to resources, etc. I have created a fix that actually ends up making the worker AI more efficient, and gives you more control over how your workers behave.

There are only a few steps. First, copy this code:

// When an assimilator is constructed, assign workers to it
for (std::set<std::pair<Unit*, int> >::iterator i = workersPerAssimilator.begin(); i != workersPerAssimilator.end(); i++)
{
    // If the assimilator is still being constructed, or WORKERS_PER_ASSIMILATOR workers are
    // already asigned to it, ignore it for now
    if (!((*i).first->isCompleted()) || (*i).second >= WORKERS_PER_ASSIMILATOR) continue;

    // Assign a maximum of WORKERS_PER_ASSIMILATOR workers to each available assimilator
    int probeCount = 0;
    UnitGroup tempGasWorkers = SelectAll(Broodwar->self(), UnitTypes::Protoss_Probe)(FilterFlag::isIdle) + SelectAll(Broodwar->self(), UnitTypes::Protoss_Probe)(FilterFlag::isGatheringMinerals);
    tempGasWorkers = tempGasWorkers - gasWorkers;
    for (UnitGroup::iterator w = tempGasWorkers.begin(); w != tempGasWorkers.end() && probeCount < WORKERS_PER_ASSIMILATOR; w++, probeCount++)
    {
        (*w)->stop();
        (*w)->rightClick((*i).first);
        gasWorkers.insert((*w));
        (*i).second = (*i).second + 1;
    }
}

// Make sure all idle workers are always gathering minerals
UnitGroup workers = SelectAll(Broodwar->self(), UnitTypes::Protoss_Probe)(FilterFlag::isIdle);
workers = workers - gasWorkers;
for each (Unit* w in workers)
{
    w->rightClick(closestMinerals);
}

...and then place it in the onFrame event handler immediately following this chunk of code:

if (Broodwar->isReplay()) return;
if (!this->analyzed) return;
this->buildManager->update();
this->buildOrderManager->update();
this->baseManager->update();
this->workerManager->update();
this->techManager->update();
this->upgradeManager->update();
this->supplyManager->update();
this->scoutManager->update();
this->defenseManager->update();
this->arbitrator.update();

this->enhancedUI->update();

If you will not be playing as Protoss, be sure to replace all occurrences of UnitTypes::Protoss_Probe with your race's worker type.

I also suggest that you only have this code execute every second or half a second, for performance reasons. It really does not need to be executed every frame. You can do this by placing the above code that you just copied within the following condition:

// Only do these every second, for performance reasons
if (Broodwar->getFrameCount() % 24 == 0)
{
    // Put all that code here
}

Next, place the following global variables at the very top of your BasicAIModule.cpp file:

// Starting base location
BWTA::BaseLocation* home;
// All workers assigned to gathering gas
UnitGroup gasWorkers;
// Set of all assimilators and the number workers assigned to them
std::set<std::pair<Unit*, int> > workersPerAssimilator;
// Closest minerals to start location
Unit* closestMinerals;

int WORKERS_PER_ASSIMILATOR = 3;

Set this to whatever you feel is appropriate.

Finally, place this code in your onUnitMorph:

// Once an assimilator has been constructed, add it to the set of assimilators
if (unit->getPlayer() == Broodwar->self() && unit->getType() == UnitTypes::Protoss_Assimilator)
{
    workersPerAssimilator.insert(std::pair<Unit*, int>(unit, 0));
}

Your workers will now behave efficiently and as expected!

Note: If you're still having problems, please see the full template here.