void WorkerManager::onUnitMorph(BWAPI::Unit unit)
{
UAB_ASSERT(unit != nullptr, "Unit was null");
// if something morphs into a worker, add it
if (unit->getType().isWorker() && unit->getPlayer() == BWAPI::Broodwar->self() && unit->getHitPoints() >= 0)
{
workerData.addWorker(unit);
}
// if something morphs into a building, it was a worker?
if (unit->getType().isBuilding() && unit->getPlayer() == BWAPI::Broodwar->self() && unit->getPlayer()->getRace() == BWAPI::Races::Zerg)
{
//BWAPI::Broodwar->printf("A Drone started building");
workerData.workerDestroyed(unit);
}
}
void WorkerManager::onUnitShow(BWAPI::Unit unit)
{
UAB_ASSERT(unit != nullptr, "Unit was null");
// add the depot if it exists
if (unit->getType().isResourceDepot() && unit->getPlayer() == BWAPI::Broodwar->self())
{
workerData.addDepot(unit);
}
// if something morphs into a worker, add it
if (unit->getType().isWorker() && unit->getPlayer() == BWAPI::Broodwar->self() && unit->getHitPoints() >= 0)
{
//BWAPI::Broodwar->printf("A worker was shown %d", unit->getID());
workerData.addWorker(unit);
}
}
bool UnitUtil::IsValidUnit(BWAPI::Unit unit)
{
if (!unit)
{
return false;
}
if (unit->isCompleted()
&& unit->getHitPoints() > 0
&& unit->exists()
&& unit->getType() != BWAPI::UnitTypes::Unknown
&& unit->getPosition().x != BWAPI::Positions::Unknown.x
&& unit->getPosition().y != BWAPI::Positions::Unknown.y)
{
return true;
}
else
{
return false;
}
}
