SnippyHolloW-BroodwarBotQ-f01ab56/src/Micro/UnitManager/UnitSet.h

Go to the documentation of this file.

#pragma once
#include <BWAPI.h>
#include <list>

#include "Objective.h"
class Objective;

typedef enum
{
        DEFEND_BASE_GO = 0,
        ATTACK_BASE_GO = 1,
        EXPLORE_GO     = 2,
        DROP_GO        = 3,
        PATROL_GO      = 4,
        BLOCK_SLOPE_GO = 5,
        ND_GO          = 0
} groupObjective_t;

typedef enum
{
        OBJ_IN_PROGRESS_GS = 1,
        MOVE_TO_OBJ_GS     = 2,
        WAIT_4_UNITS_GS    = 3,
        FLEE_OBJ_GS        = 4,
        ND_GS              = 0
} groupStatus_t;

class UnitSet
{
protected : 
        std::list<BWAPI::Unit*> lUnits_;          
        groupObjective_t objective_;              
        groupStatus_t status_;                    
        BWAPI::Position curPos_;                  
        BWAPI::Position objPos_;                  
        /*************** Constructors and destructors ***************/
public : 
        UnitSet();

        UnitSet(groupObjective_t obj,BWAPI::Position pos);

        UnitSet(BWAPI::Position regroupPos);

        ~UnitSet();

        /************************* Accessors ************************/

protected : 
        void updateAvgPos();

public : 
        inline BWAPI::Position curPos() const { return this->curPos_; }

        BWAPI::Position pos() const;

        BWAPI::Position objPos() const;

        inline bool empty() const { return lUnits_.empty(); }

        inline int nbUnits() const { return lUnits_.size(); }

        /************************** Actions *************************/

public : 
        void moveTo(BWAPI::Position pos);

        void attackTo(const BWAPI::Position pos);

        void patrolTo(BWAPI::Position pos);

        void stop();

        /******************** Objective evaluation ******************/

        bool canFulFillObj(Objective* obj) const;

        /********************** Units management ********************/

        bool requires(BWAPI::Unit* unit) const;

        void addUnit(BWAPI::Unit* unit);
};