BTHAI/SCProjects/BTHAIModule/Source/WorkerAgent.h

Go to the documentation of this file.

#ifndef __WORKERAGENT_H__
#define __WORKERAGENT_H__

#include "BaseAgent.h"
using namespace BWAPI;
using namespace std;

class WorkerAgent : public BaseAgent {

private:
        int currentState;

        UnitType toBuild;
        TilePosition buildSpot;
        TilePosition startSpot;
        bool buildSpotExplored();
        bool areaFree();
        bool isBuilt();
        int startBuildFrame;

        void handleKitingWorker();
        Unit* getEnemyUnit();
        Unit* getEnemyBuilding();
        Unit* getEnemyWorker();

        int lastFrame;

public:
        static const int GATHER_MINERALS = 0;
        static const int GATHER_GAS = 1;
        static const int FIND_BUILDSPOT = 2;
        static const int MOVE_TO_SPOT = 3;
        static const int CONSTRUCT = 4;
        static const int REPAIRING = 5;
        static const int ATTACKING = 6;

        WorkerAgent(Unit* mUnit);

        void computeActions();

        string getTypeName();

        void setState(int state);

        int getState();

        bool canBuild(UnitType type);

        bool assignToRepair(Unit* building);

        bool assignToBuild(UnitType type);

        bool assignToFinishBuild(Unit* building);

        string getStateAsText();

        void destroyed();

        void reset();

        bool isConstructing(UnitType type);
};

#endif