SnippyHolloW-BroodwarBotQ-f01ab56/src/Micro/UnitsGroup.h

Go to the documentation of this file.

#pragma once
#include <BWAPI.h>
#include <BWTA.h>
#include <windows.h>
#include <math.h>
#include "Micro/Units/BayesianUnit.h"
#include "Micro/Formations.h"
#include "Utils/Vec.h"
#include <set>
#include <boost/bimap/bimap.hpp>
#include "Intelligence/EUnitsFilter.h"

#define _UNITS_DEBUG 1

struct Dmg
{
    Dmg(int d, BWAPI::Unit* u, int h) 
        : dmg(d)
        , unit(u)
        , hp(h)
    { }
    Dmg(const Dmg& d)
        : dmg(d.dmg)
        , unit(d.unit)
        , hp(d.hp)
    { }
    int dmg;
    int hp;
    BWAPI::Unit* unit;
    bool operator<(const Dmg& d) const { return ((this->hp - this->dmg) < (d.hp - d.dmg) || (this->hp - this->dmg == d.hp - d.dmg) && this->unit < d.unit); }
    Dmg operator-=(int i) { this->dmg -= i; return *this; }
    Dmg operator+=(int i) { this->dmg += i; return *this; }
};

typedef boost::bimaps::bimap<BWAPI::Unit*, Dmg> UnitDmgBimap;
typedef UnitDmgBimap::relation UnitDmg;

class Formation;

struct i_dist 
{
        unsigned int ind;
        double dist;
        i_dist(unsigned int i, double d): ind(i), dist(d) {}
        //bool operator<(i_dist& ext) { return (ext.dist < dist); }
};

struct BasicUnitsGroup // should derive from vector<pBayesianUnit>
{
        std::vector<pBayesianUnit> units;
        virtual void update();
        bool emptyUnits();
    int size() const;
};

class UnitsGroup : public BasicUnitsGroup
{
private:
        int _totalHP;
    int _totalMinPrice;
    int _totalGasPrice;
    int _totalSupply;
        int _backFrame;
        int _offensiveFrame;
        double _maxRange;
    EUnitsFilter* _eUnitsFilter;
    std::set<BWAPI::Unit*> _mergersHT;
    bool _hasDetection;
    inline void updateNearbyEnemyUnitsFromFilter(BWAPI::Position p, double radius = 400.0);
    double evaluateForces();
        inline std::vector<BWAPI::Position> findRangePositions();
        inline void activeUnit(pBayesianUnit bu);
    bool removeUnit(BWAPI::Unit* u);
    bool removeArrivingUnit(BWAPI::Unit* u);
#ifndef __RELEASE_OPTIM__
    void displayTargets();  // debug purpose
#endif
public:
        double force;
        bool enemiesDefinedByGoal;
    std::list<pBayesianUnit> arrivingUnits;
    std::vector<BWAPI::Position> ppath;
    UnitDmgBimap unitDamages;
    pBayesianUnit leadingUnit;
    BWAPI::Unit* defaultTargetEnemy;
        BWAPI::Position groupTargetPosition;
        double distToTarget;
    BWTA::Chokepoint* nearestChoke;
    double distToNearestChoke;
    Vec centerSpeed;
        std::map<BWAPI::UnitSizeType, int> sizes;
        bool isFighting;
        bool suicide;
        bool readyToAttack;
        int nonFlyers;
        unit_mode groupMode;
    BWAPI::Position center;
    int groupAltitude;
    double stdDevRadius, maxRadius;
    BWAPI::Position enemiesCenter;
        bool enemyStatic;
    int enemiesAltitude;
    std::map<BWAPI::Unit*, BWAPI::Position> enemies;
        
        UnitsGroup();
        virtual ~UnitsGroup();
        void update();
#ifndef __RELEASE_OPTIM__
        void display();
#endif

        void move(BWAPI::Position& p);
        virtual void formation(pFormation f);
        void switchMode(unit_mode um);
        void idle();
        void dispatchCompleteUnit(pBayesianUnit bu);
    void giveUpControl(BWAPI::Unit* u);
    virtual void onUnitDestroy(BWAPI::Unit* u);
    virtual void onUnitShow(BWAPI::Unit* u);
    virtual void onUnitHide(BWAPI::Unit* u);
        inline void updateArrivingUnits();
        inline void chooseLeadingUnit();
        inline void updateCenter();
        inline void updateOurStats();
        inline void updateEnemiesCenter();
    virtual BWAPI::Position getCenter() const;
    inline double getDistance(BWAPI::Unit* u) const;
        double getDistance(BWAPI::Position p) const;
    int getTotalHP() const;
    std::vector<pBayesianUnit>* getUnits(); // a baaad getter, keep an eye
        const BayesianUnit& operator[](ptrdiff_t i);
    inline void templarMergingStuff();
    void signalMerge(BWAPI::Unit* u);
#ifndef __RELEASE_OPTIM__
    void selectedUnits(std::set<pBayesianUnit>& u);
#endif
};