Method start

bwapi/TestAIModule/Source/UpgradeTest.cpp:20–68

  BWAssertF(upgraderType!=UnitTypes::Unknown,{fail=true;return;});
}
void UpgradeTest::start()
{
  if (fail) return;
  running = true;

  int upgraderCount = Broodwar->self()->completedUnitCount(upgraderType);
  BWAssertF(upgraderCount>=1,{fail=true;LOGTYPE;return;});
  for (Unit u : Broodwar->self()->getUnits())
  {
    if (u->getType()==upgraderType)
    {
      upgrader = u;
      break;
    }
  }
  BWAssertF(upgrader!=NULL,{fail=true;LOGTYPE;return;});

  BWAssertF(upgrader->getOrder()==Orders::Nothing,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->isIdle()==true,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->isLifted()==false,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->isResearching()==false,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->isUpgrading()==false,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->getTech()==TechTypes::None,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->getUpgrade()==UpgradeTypes::None,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->getRemainingResearchTime()==0,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->getRemainingUpgradeTime()==0,{fail=true;LOGTYPE;return;});
  previousUpgradeLevel = Broodwar->self()->getUpgradeLevel(upgradeType);
  correctMineralCount  = Broodwar->self()->minerals() - upgradeType.mineralPrice(previousUpgradeLevel+1);
  correctGasCount      = Broodwar->self()->gas() - upgradeType.gasPrice(previousUpgradeLevel+1);

  upgrader->upgrade(upgradeType);

  BWAssertF(upgrader->getOrder()==Orders::Upgrade,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->isIdle()==false,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->isLifted()==false,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->isResearching()==false,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->isUpgrading()==true,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->getTech()==TechTypes::None,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->getUpgrade()==upgradeType,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->getRemainingResearchTime()==0,{fail=true;LOGTYPE;return;});
  BWAssertF(upgrader->getRemainingUpgradeTime()==upgradeType.upgradeTime(previousUpgradeLevel+1),{fail=true;LOGTYPE;return;});
  BWAssertF(Broodwar->self()->getUpgradeLevel(upgradeType)==previousUpgradeLevel,{fail=true;LOGTYPE;return;});
  BWAssertF(Broodwar->self()->minerals() == correctMineralCount,{fail=true;LOGTYPE;return;});
  BWAssertF(Broodwar->self()->gas() == correctGasCount,{fail=true;LOGTYPE;return;});

  startUpgradeFrame = Broodwar->getFrameCount();
  nextUpdateFrame = startUpgradeFrame;

}
void UpgradeTest::update()
{
  if (running == false) return;