Limit.h

Go to the documentation of this file.

#pragma once

#include "tree/TreeNode.h"

namespace ai {

class Limit: public TreeNode {
private:
    int _amount;
public:
    NODE_FACTORY(Limit)

    Limit(const std::string& name, const std::string& parameters, const ConditionPtr& condition) :
        TreeNode(name, parameters, condition) {
        _type = "Limit";
        if (!parameters.empty()) {
            _amount = ::atoi(parameters.c_str());
        } else {
            _amount = 1;
        }
    }

    TreeNodeStatus execute(const AIPtr& entity, int64_t deltaMillis) override {
        ai_assert(_children.size() == 1, "Limit must have exactly one node");

        if (TreeNode::execute(entity, deltaMillis) == CANNOTEXECUTE) {
            return CANNOTEXECUTE;
        }

        const int alreadyExecuted = getLimitState(entity);
        if (alreadyExecuted >= _amount) {
            return state(entity, FINISHED);
        }

        const TreeNodePtr& treeNode = *_children.begin();
        const TreeNodeStatus status = treeNode->execute(entity, deltaMillis);
        setLimitState(entity, alreadyExecuted + 1);
        if (status == RUNNING) {
            return state(entity, RUNNING);
        }
        return state(entity, FAILED);
    }
};

}