UAlbertaBot_src/Projects/UAlbertaBot/Source/SquadData.h

Go to the documentation of this file.

#pragma once

#include "Common.h"
#include "Squad.h"

class SquadData
{
        // All squads. Indexed by SquadOrder enum
        std::vector<Squad>      squads;

        void                            updateAllSquads();
        int                                     getNumType(UnitVector & units, BWAPI::UnitType type);

public:

        SquadData();
        ~SquadData() {}

        void                            clearSquadData();

        void                            addSquad(Squad & squad);
        void                            drawSquadInformation(int x, int y);

        void                            update();
        void                            setSquad(Squad & squad);
        void                            setRegroup();
        UnitVector &            getUnits();
        SquadOrder &            getSquadOrder();
};