HasEnemies.h

Go to the documentation of this file.

#pragma once

#include "conditions/ICondition.h"
#include "common/String.h"
#include "aggro/AggroMgr.h"

namespace ai {

class HasEnemies: public ICondition {
protected:
    int _enemyCount;
public:
    CONDITION_FACTORY(HasEnemies)

    explicit HasEnemies(const std::string& parameters) :
            ICondition("HasEnemies", parameters) {
        if (_parameters.empty()) {
            _enemyCount = -1;
        } else {
            _enemyCount = std::stoi(_parameters);
        }
    }

    bool evaluate(const AIPtr& entity) override {
        const AggroMgr& mgr = entity->getAggroMgr();
        if (_enemyCount == -1) {
            // TODO: check why boolean operator isn't working here
            const bool hasEnemy = mgr.getHighestEntry() != nullptr;
            return hasEnemy;
        }
        const int size = static_cast<int>(mgr.getEntries().size());
        return size >= _enemyCount;
    }
};

}