label.hpp

 
#ifndef UFO_MAP_LABELS_PREDICATE_LABEL_HPP
#define UFO_MAP_LABELS_PREDICATE_LABEL_HPP
 
// UFO
#include <ufo/container/tree/predicate/filter.hpp>
#include <ufo/container/tree/predicate/predicate_compare.hpp>
#include <ufo/core/label.hpp>
#include <ufo/map/labels/propagation_criteria.hpp>
 
namespace ufo::pred
{
template <bool Negated = false>
struct Label {
    Label(label_t label) : label(label) {}
 
    label_t label;
};
 
template <bool Negated>
Label<!Negated> operator!(Label<Negated>)
{
    return {};
}
 
template <bool Negated>
struct Filter<Label<Negated>> {
    using Pred = Label<Negated>;
 
    template <class Tree>
    static constexpr void init(Pred& p, Tree const& t)
    {
    }
 
    template <class Tree, class Node>
    [[nodiscard]] static constexpr bool returnable(Pred const& p, Tree const& t,
                                                   Node const& n)
    {
        // TODO: Do this better
        auto ls = t.labels(n.index);
        if constexpr (Negated) {
            return ls.find(p.label) == ls.end();
        } else {
            return ls.find(p.label) != ls.end();
        }
    }
 
    template <class Tree, class Node>
    [[nodiscard]] static constexpr bool traversable(Pred const& p, Tree const& t,
                                                    Node const& n)
    {
        if constexpr (Negated) {
            return true;
        } else {
            // TODO: Do this better
            auto ls = t.labels(n.index);
            switch (t.labelsPropagationCriteria()) {
                case LabelsPropagationCriteria::ALL: {
                    return ls.find(p.label) != ls.end();
                }
                case LabelsPropagationCriteria::SUMMARY: {
                    return !ls.empty() && *ls.begin() & p.label == p.label;
                }
                case LabelsPropagationCriteria::MIN: {
                    return !ls.empty() && *ls.begin() < p.label;
                }
                case LabelsPropagationCriteria::MAX: {
                    return !ls.empty() && *ls.begin() > p.label;
                    return true;
                }
                case LabelsPropagationCriteria::NONE: {
                    return true;
                }
            }
        }
        return true;
    }
};
 
}  // namespace ufo::pred
 
#endif  // UFO_MAP_LABELS_PREDICATE_LABEL_HPP