BTHAI/SCProjects/BTHAIModule/Source/Squad.h

Go to the documentation of this file.

#ifndef __SQUAD_H__
#define __SQUAD_H__

#include "BaseAgent.h"
#include "Squad.h"
#include "UnitSetup.h"

using namespace BWAPI;
using namespace std;

class Squad {

protected:
        vector<BaseAgent*> agents;
        vector<UnitSetup> setup;

        TilePosition goal;
        
        void setMemberGoals();
        void setMemberState(BaseAgent* agent);
        Unit* findTarget();
        
        int id;
        bool active;
        int type;
        int priority;
        int activePriority;
        int moveType;
        bool required;
        string name;
        int goalSetFrame;

        void checkAttack();
        bool isVisible(TilePosition pos);
        TilePosition getNextStartLocation();
        vector<TilePosition> hasVisited;

public:
        Squad();
        
        Squad(int mId, int mType, int mMoveType, string mName, int mPriority);

        int getID();

        void addRequirement(UnitType type, int no);

        string getName();

        bool isRequired();

        void setRequired(bool mRequired);

        int getPriority();

        void setPriority(int mPriority);

        void setActivePriority(int mPriority);

        void addSetup(UnitType type, int no);

        vector<UnitSetup> getSetup();

        virtual bool isActive();

        void forceActive();

        bool isFull();

        int size();

        int getHealthPct();

        virtual void computeActions();

        bool isGathered();

        bool isAttacking();

        bool isUnderAttack();

        bool needUnit(UnitType type);

        bool addMember(BaseAgent* agent);

        vector<BaseAgent*> getMembers();

        void removeMember(BaseAgent* agent);

        BaseAgent* removeMember(UnitType type);

        virtual void setGoal(TilePosition mGoal);

        virtual void clearGoal();

        virtual TilePosition getGoal();

        virtual bool hasGoal();

        bool isThisGoal(TilePosition mGoal);

        bool isCloseTo(TilePosition mGoal);

        TilePosition getCenter();

        BaseAgent* getCenterAgent();

        bool isOffensive();

        bool isDefensive();

        bool isExplorer();

        bool isBunkerDefend();

        bool isShuttle();

        bool isKite();

        bool isRush();

        bool isSupport();

        bool isGround();

        bool isAir();

        int getSize();

        int getTotalUnits();

        int getStrength();

        void disband(TilePosition retreatPoint);

        virtual void printInfo();

        void printFullInfo();

        bool canMerge(Squad* squad);

        bool hasUnits(UnitType type, int no);

        bool contains(UnitType type);

        TilePosition getClosestStartLocation(TilePosition pos);

        static const int OFFENSIVE = 0;
        static const int DEFENSIVE = 1;
        static const int EXPLORER = 2;
        static const int SUPPORT = 3;
        static const int BUNKER = 4;
        static const int SHUTTLE = 5;
        static const int KITE = 6;
        static const int RUSH = 7;
        static const int CHOKEHARASS = 8;
        static const int GROUND = 0;
        static const int AIR = 1;

        
};

#endif