Method start

bwapi/TestAIModule/Source/TransportTest.cpp:19–69

  running = false;
}
void TransportTest::start()
{
  if (fail) return;
  running = true;

  int transportCount = Broodwar->self()->completedUnitCount(transportType);
  BWAssertF(transportCount>=1,{fail=true;return;});
  for (Unit u : Broodwar->self()->getUnits())
    if (u->getType()==transportType)
      transport = u;

  if (transportType.getRace()==Races::Terran)
  {
    for (Unit u : Broodwar->self()->getUnits())
      if (u->getType()==UnitTypes::Terran_Marine ||
          u->getType()==UnitTypes::Terran_Firebat ||
          u->getType()==UnitTypes::Terran_Ghost ||
          u->getType()==UnitTypes::Terran_Medic)
        unloadedUnits.insert(u);
  }
  else if (transportType.getRace()==Races::Protoss)
  {
    for (Unit u : Broodwar->self()->getUnits())
      if (u->getType()==UnitTypes::Protoss_Probe ||
          u->getType()==UnitTypes::Protoss_Zealot ||
          u->getType()==UnitTypes::Protoss_Dragoon)
        unloadedUnits.insert(u);
  }
  else if (transportType.getRace()==Races::Zerg)
  {
    for (Unit u : Broodwar->self()->getUnits())
      if (u->getType()==UnitTypes::Zerg_Drone ||
          u->getType()==UnitTypes::Zerg_Hydralisk ||
          u->getType()==UnitTypes::Zerg_Lurker)
        unloadedUnits.insert(u);
  }
  BWAssertF(transport!=NULL,{fail=true;return;});
  BWAssertF(transport->exists(),{fail=true;return;});
  BWAssertF(transport->getLoadedUnits().size()==0,{fail=true;return;});
  BWAssertF(unloadedUnits.size()==4,{fail=true;return;});
  for (Unit u : unloadedUnits)
  {
    BWAssertF(u!=NULL,{fail=true;return;});
    BWAssertF(u->exists()==true,{fail=true;return;});
    BWAssertF(u->isLoaded()==false,{fail=true;return;});
    BWAssertF(u->getTransport()==NULL,{fail=true;return;});
  }
  startFrame = Broodwar->getFrameCount();
  nextFrame = Broodwar->getFrameCount();
  state = Start;
}
bool TransportTest::verifyLoadedUnits()
{
  BWAssertF(transport!=NULL,{fail=true;return false;});