Skynet/Skynet/TrainTask.h

Go to the documentation of this file.

#pragma once

#include "Interface.h"

#include "Task.h"

class TrainTask : public Task
{
public:
        TrainTask(TaskType priority, BWAPI::UnitType type);

        virtual int getEndTime() const;
        virtual int getEndTime(Unit unit) const;

        virtual int getPriority(Unit unit) const;

        virtual Position getStartLocation(Unit unit) const;
        virtual Position getEndLocation(Unit unit) const;

        virtual bool preUpdate();
        virtual bool update();

        virtual void updateRequirements();

        virtual bool waitingForUnit(Unit unit) const;
        virtual void giveUnit(Unit unit);
        virtual void returnUnit(Unit unit);
        virtual bool morph(Unit unit, BWAPI::UnitType previousType);
        virtual UnitGroup getFinishedUnits();

        virtual std::string getTaskName() const { return "Training"; }
        virtual std::string getOutputName() const { return mType.getName(); }

        virtual void setRequiredSatisfyTime(RequirementGroup requirements, int time, int delay) { mRequiredSatisfyTime = time; }

        virtual int dispatchTime() const { return mRequiredSatisfyTime; }
        virtual int startTime() const { return mRequiredSatisfyTime; }

        virtual bool hasDispatched() const { return mProductionBuilding; }
        virtual bool inProgress() const { return mProducedUnit; }

        BWAPI::UnitType getTrainType() { return mType; }

private:
        BWAPI::UnitType mType;

        Unit mProductionBuilding;
        Unit mProducedUnit;

        int mRequiredSatisfyTime;

        bool mReservedResources;

        int mOrderTime;

        void reserveResources();
        void freeResources();
};