sc2_unit_filters.h

#pragma once

#include "sc2_typeenums.h"
#include "sc2_unit.h"

#include <vector>

namespace sc2 {
struct IsUnit {
    explicit IsUnit(UNIT_TYPEID type_);

    bool operator()(const Unit& unit_) const;

 private:
    UNIT_TYPEID m_type;
};

struct IsUnits {
    explicit IsUnits(const std::vector<UNIT_TYPEID>& types_);

    bool operator()(const Unit& unit_) const ;

 private:
    std::vector<UNIT_TYPEID> m_types;
};

struct IsTownHall {
    bool operator()(const Unit& unit_) const;

    bool operator()(UNIT_TYPEID type_) const;
};

struct IsMineralPatch {
    bool operator()(const Unit& unit_) const;

    bool operator()(UNIT_TYPEID type_) const;
};

struct IsVisibleMineralPatch {
    bool operator()(const Unit& unit_) const;
};

struct IsGeyser {
    bool operator()(const Unit& unit_) const;

    bool operator()(UNIT_TYPEID type_) const;
};

struct IsVisibleGeyser {
    bool operator()(const Unit& unit_) const;
};

}  // namespace sc2