properties.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_IMPL_PROPERTIES_HPP
#define KOKKOS_EXECUTION_IMPL_PROPERTIES_HPP
 
#include "kokkos-execution/stdexec.hpp"
 
#include "Kokkos_Macros.hpp"
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#else
#    include "Kokkos_Core.hpp"
#endif

 
namespace Kokkos::Execution::Impl {

template <typename... Properties>
struct properties { };
 
namespace detail {
 
template <template <typename...> typename T, typename... Args>
consteval void deduce_from_base(const T<Args...>&);
 
template <template <typename...> typename T, typename... Args>
constexpr T<Args...>& as_base(T<Args...>& base) noexcept {
    return base;
}
 
template <template <typename...> typename T, typename... Args>
constexpr const T<Args...>& as_base(const T<Args...>& base) noexcept {
    return base;
}
 
template <typename Property, typename Data>
constexpr Property& as_base(Data& data) noexcept {
    return data;
}
 
template <typename Property, typename Data>
constexpr const Property& as_base(const Data& data) noexcept {
    return data;
}
 
} // namespace detail

template <typename Data, typename Property>
consteval bool has_property() {
    return std::derived_from<Data, Property>;
}

template <typename Data, template <typename...> typename Property>
consteval bool has_property() {
    return requires(const Data& data) { detail::deduce_from_base<Property>(data); };
}
 
template <typename Property, typename Data>
requires(has_property<std::remove_cvref_t<Data>, Property>())
constexpr decltype(auto) get_property(Data&& data) noexcept { // NOLINT(cppcoreguidelines-missing-std-forward)
    return stdexec::__forward_like<Data>(detail::as_base<Property>(data).m_value);
}
 
template <template <typename...> typename Property, typename Data>
requires(has_property<std::remove_cvref_t<Data>, Property>())
constexpr decltype(auto) get_property(Data&& data) noexcept { // NOLINT(cppcoreguidelines-missing-std-forward)
    return stdexec::__forward_like<Data>(detail::as_base<Property>(data).m_value);
}
 
} // namespace Kokkos::Execution::Impl
 
#endif // KOKKOS_EXECUTION_IMPL_PROPERTIES_HPP