fix to BOSSManager

fixed bug where BOSS would not trigger if zerg only had non-hatchery depots
earthiverse · Nov 20, 2015 · 0e845bb · 0e845bb
1 parent 9446caf
commit 0e845bb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/UAlbertaBot/Source/BOSSManager.cpp b/UAlbertaBot/Source/BOSSManager.cpp
@@ -31,8 +31,10 @@ void BOSSManager::reset()
 // start a new search for a new goal
 void BOSSManager::startNewSearch(const std::vector<MetaPair> & goalUnits)
 {
-    size_t numWorkers = UnitUtil::GetAllUnitCount(BWAPI::Broodwar->self()->getRace().getWorker());
-    size_t numDepots = UnitUtil::GetAllUnitCount(BWAPI::Broodwar->self()->getRace().getCenter());
+    size_t numWorkers   = UnitUtil::GetAllUnitCount(BWAPI::Broodwar->self()->getRace().getWorker());
+    size_t numDepots    = UnitUtil::GetAllUnitCount(BWAPI::Broodwar->self()->getRace().getCenter())
+                        + UnitUtil::GetAllUnitCount(BWAPI::UnitTypes::Zerg_Lair)
+                        + UnitUtil::GetAllUnitCount(BWAPI::UnitTypes::Zerg_Hive);
 
     if (numWorkers == 0)
     {