trunk/bwapi/TestAIModule/Source/TransportTest.h

Go to the documentation of this file.

#pragma once
#include "TestCase.h"
#include <BWAPI.h>
#include <vector>

class TransportTest : public TestCase
{
  public:
    TransportTest(BWAPI::UnitType unitType);
    virtual void start();
    virtual void update();
    virtual void stop();
  private:
    bool verifyLoadedUnits();
    int startFrame;
    int nextFrame;
    BWAPI::UnitType transportType;
    BWAPI::Unit* transport;
    BWAPI::Unit* currentUnit;
    std::set<BWAPI::Unit*> unloadedUnits;
    std::set<BWAPI::Unit*> loadedUnits;
    int waitFrameStart;
    enum LoadState
    {
      Start,
      WaitForFirstUnit,
      WaitForSecondUnit,
      WaitForThirdUnit,
      WaitForFourthUnit,
      WaitDelay1,
      WaitUnloadUnit,
      WaitReloadUnit,
      WaitUnloadAll,
      WaitReloadAll,
      WaitUnloadAllPosition,
      End
    } state;
};