EventRegionMatcher.hpp

Go to the documentation of this file.

#ifndef KOKKOS_UTILS_CALLBACKS_EVENTREGIONMATCHER_HPP
#define KOKKOS_UTILS_CALLBACKS_EVENTREGIONMATCHER_HPP
 
#include "kokkos-utils/callbacks/Events.hpp"
#include "kokkos-utils/callbacks/Matcher.hpp"
 
namespace Kokkos::utils::callbacks
{

template <typename MatcherType, bool TrueOnBoundary = false> requires MatcherFor<MatcherType, PushRegionEvent>
struct EventRegionMatcher
{
    static constexpr uint32_t invalid_nested_level = Kokkos::Experimental::finite_max_v<uint32_t>;

    bool operator()(const PushRegionEvent& event)
    {
        if (this->matching)
        {
            ++this->nested_level;
            return ! TrueOnBoundary;
        }
        else if (matcher(event))
        {
            this->nested_level = 1;
            this->matching     = true;
            return TrueOnBoundary;
        }
 
        return TrueOnBoundary ? false : this->matching;
    }

    bool operator()(const PopRegionEvent&)
    {
        if (this->matching)
        {
            --this->nested_level;
 
            if (this->nested_level == 0)
            {
                this->matching = false;
                return TrueOnBoundary;
            }
        }
 
        return TrueOnBoundary ? false : this->matching;
    }
 
    template <Event EventType>
    bool operator()(const EventType&) const {
        return TrueOnBoundary ? false : this->matching;
    }
 
    MatcherType matcher {};
    bool matching = false;
    uint32_t nested_level = invalid_nested_level;
};
 
} // namespace Kokkos::utils::callbacks
 
#endif // KOKKOS_UTILS_CALLBACKS_EVENTREGIONMATCHER_HPP