Method update

bwapi/TestAIModule/Source/CancelTrainTest.cpp:130–240

}
void CancelTrainTest::update()
{
  if (running == false) return;
  if (fail)
  {
    running = false;
    return;
  }
  int thisFrame = Broodwar->getFrameCount();
  BWAssert(thisFrame==nextFrame);
  BWAssertF(producer!=NULL,{fail=true;return;});
  nextFrame++;
  Broodwar->setScreenPosition(producer->getPosition() - Position(320,240));
  if (state==Start)
  {
    BWAssertF(producer->isTraining()==true,{fail=true;return;});
    BWAssertF(producer->isConstructing()==false,{fail=true;return;});
    BWAssertF(producer->isIdle()==false,{fail=true;return;});
    BWAssertF(producer->isLifted()==false,{fail=true;return;});
    BWAssertF(verifyTrainingQueue(),{log("Error training queue failed at %u frame %d",__LINE__,thisFrame-startFrame);fail=true;return;});
    BWAssertF(Broodwar->self()->minerals() == correctMineralCount,{fail=true;return;});
    BWAssertF(Broodwar->self()->gas() == correctGasCount,{fail=true;return;});
    BWAssertF(Broodwar->self()->supplyUsed() == correctSupplyUsedCount,{log("Error supply used %d != %d",Broodwar->self()->supplyUsed(),correctSupplyUsedCount);fail=true;return;});
    BWAssertF(Broodwar->self()->allUnitCount(unitType2) == originalAllUnit2Count,{fail=true;return;});
    BWAssertF(Broodwar->self()->allUnitCount(unitType3) == originalAllUnit3Count,{fail=true;return;});
    if (thisFrame>=startFrame+15)
    {
      BWAssertF(producer->cancelTrain(0),{Broodwar->printf("%s",Broodwar->getLastError().c_str());fail=true;return;});
      correctTrainingQueue.pop_front();
      BWAssertF(verifyTrainingQueue(),{log("Error training queue failed at %u frame %d",__LINE__,thisFrame-startFrame-150);fail=true;return;});
      correctMineralCount = correctMineralCount + unitType1.mineralPrice();
      correctGasCount = correctGasCount + unitType1.gasPrice();
      correctSupplyUsedCount = correctSupplyUsedCount - unitType1.supplyRequired() + unitType2.supplyRequired();
      BWAssertF(Broodwar->self()->minerals() == correctMineralCount,{Broodwar->printf("%d != %d",Broodwar->self()->minerals(),correctMineralCount);fail=true;return;});
      BWAssertF(Broodwar->self()->gas() == correctGasCount,{fail=true;return;});
      BWAssertF(Broodwar->self()->supplyUsed() == correctSupplyUsedCount,{fail=true;return;});
      state = CancelledFirstSlot;
    }
  }
  else if (state == CancelledFirstSlot)
  {
    BWAssertF(producer->isTraining()==true,{fail=true;return;});
    BWAssertF(producer->isConstructing()==false,{fail=true;return;});
    BWAssertF(producer->isIdle()==false,{fail=true;return;});
    BWAssertF(producer->isLifted()==false,{fail=true;return;});
    BWAssertF(verifyTrainingQueue(),{log("Error training queue failed at %u frame %d",__LINE__,thisFrame-startFrame-150);fail=true;return;});
    BWAssertF(Broodwar->self()->minerals() == correctMineralCount,{fail=true;return;});
    BWAssertF(Broodwar->self()->gas() == correctGasCount,{fail=true;return;});
    BWAssertF(Broodwar->self()->supplyUsed() == correctSupplyUsedCount,{log("Error supply used %d: %d != %d",thisFrame-startFrame-150,Broodwar->self()->supplyUsed(),correctSupplyUsedCount);});
    if (thisFrame>=startFrame+30)
    {
      BWAssertF(producer->cancelTrain(),{Broodwar->printf("%s",Broodwar->getLastError().c_str());fail=true;return;});
      correctTrainingQueue.pop_back();
      BWAssertF(verifyTrainingQueue(),{log("Error training queue failed at %u frame %d",__LINE__,thisFrame-startFrame-30);fail=true;return;});
      correctMineralCount = correctMineralCount + unitType3.mineralPrice();
      correctGasCount = correctGasCount + unitType3.gasPrice();
      BWAssertF(Broodwar->self()->minerals() == correctMineralCount,{fail=true;return;});
      BWAssertF(Broodwar->self()->gas() == correctGasCount,{fail=true;return;});