PrioritySelector.h

Go to the documentation of this file.

#pragma once

#include "tree/Selector.h"
#include "AI.h"

namespace ai {

class PrioritySelector: public Selector {
public:
    SELECTOR_CLASS(PrioritySelector)

    TreeNodeStatus execute(const AIPtr& entity, int64_t deltaMillis) override {
        if (Selector::execute(entity, deltaMillis) == CANNOTEXECUTE)
            return CANNOTEXECUTE;

        int index = getSelectorState(entity);
        if (index == AI_NOTHING_SELECTED) {
            index = 0;
        }
        const std::size_t size = _children.size();
        TreeNodeStatus overallResult = FINISHED;
        std::size_t i = index;
        for (std::size_t j = 0; j < i; ++j) {
            _children[j]->resetState(entity);
        }
        for (; i < size; ++i) {
            const TreeNodePtr& child = _children[i];
            const TreeNodeStatus result = child->execute(entity, deltaMillis);
            if (result == RUNNING) {
                setSelectorState(entity, static_cast<int>(i));
            } else if (result == CANNOTEXECUTE || result == FAILED) {
                child->resetState(entity);
                setSelectorState(entity, AI_NOTHING_SELECTED);
                continue;
            } else {
                setSelectorState(entity, AI_NOTHING_SELECTED);
            }
            child->resetState(entity);
            overallResult = result;
            break;
        }
        for (++i; i < size; ++i) {
            _children[i]->resetState(entity);
        }
        return state(entity, overallResult);
    }
};

}