BTHAI/SCProjects/BTHAIModule/Source/BaseAgent.h

Go to the documentation of this file.

#ifndef __BASEAGENT_H__
#define __BASEAGENT_H__

#include <windows.h>
#include <BWAPI.h>
#include <BWTA.h>
#include "BWTAExtern.h"
using namespace BWAPI;
using namespace std;

class BaseAgent {

private:

protected:
        Unit* unit;
        TilePosition goal;
        int unitID;
        int squadID;
        bool alive;
        int lastActionFrame;
        string info;
        bool bBlock;

public:
        BaseAgent();
        BaseAgent(Unit* mUnit);
        ~BaseAgent();

        virtual void computeActions() {
        }

        int noUnitsInWeaponRange();

        bool canAttack(Unit* target);

        bool canAttack(UnitType type);

        virtual void setGoal(TilePosition goal);

        virtual void clearGoal();

        TilePosition getGoal();

        void setActionFrame();

        int getLastActionFrame();

        virtual string getTypeName() {
                return "BaseAgent";
        }

        virtual bool assignToBuild(UnitType type) {
                return false;
        }

        virtual bool assignToRepair(Unit* building) {
                return false;
        }

        virtual bool assignToFinishBuild(Unit* building) {
                return false;
        }

        virtual void assignToExplore() {
        }

        virtual void assignToAttack() {
        }

        virtual void assignToDefend() {
        }

        virtual bool isExploring() {
                return false;
        }

        virtual bool isAttacking() {
                return false;
        }

        virtual bool isDefending() {
                return false;
        }

        virtual void printInfo();

        void setInfo(string mInfo);

        virtual bool canBuild(UnitType type);

        int getUnitID();

        UnitType getUnitType();

        Unit* getUnit();

        virtual void destroyed();

        bool isAlive();

        bool matches(Unit* mUnit);

        bool isOfType(UnitType type);

        static bool isOfType(Unit* mUnit, UnitType type);

        static bool isOfType(UnitType mType, UnitType toCheckType);

        bool isDetectorWithinRange(TilePosition pos, int range);

        bool isBuilding();

        bool isWorker();

        bool isFreeWorker();

        bool isUnit();

        bool isUnderAttack();

        bool isDamaged();

        static bool doScannerSweep(TilePosition pos);

        static bool doEnsnare(TilePosition pos);

        void setSquadID(int id);

        int getSquadID();

};

#endif