pack.hpp

Go to the documentation of this file.

#ifndef PACK_HPP
#define PACK_HPP
#include "space_settings.hpp"
#include "label.hpp"

template <class T, class... Rest> class Pack : Pack<Rest...> {
    using Base = Pack<Rest...>; 
    T v_; 

    public:

    using var_type = typename T::var_type; 

    template<typename Func>
    KOKKOS_INLINE_FUNCTION void for_each(Func func) {
        func(v_.v);
        Base::for_each(func); // Forward to base for recursion
    }

    template<typename Func>
    KOKKOS_INLINE_FUNCTION void for_each_labeled(Func func) const {
        func(v_);
        Base::for_each_labeled(func); // Forward to base for recursion
    }

    template<Label FN>
    KOKKOS_INLINE_FUNCTION const auto& get() const{
        if constexpr(T::name==FN){
            return v_.v;
        }else{
            return Base::template get<FN>();
        }
    }

    template<Label FN>
    KOKKOS_INLINE_FUNCTION auto& get() {
        if constexpr(T::name==FN){
            return v_.v;
        }else{
            return Base::template get<FN>();
        }
    }
};

template <class T> class Pack<T> {
    T v_; 

    public:

    using var_type = typename T::var_type; 

    template<typename Func>
    KOKKOS_INLINE_FUNCTION void for_each(Func func){
        func(v_.v);
    }

    template<typename Func>
    KOKKOS_INLINE_FUNCTION void for_each_labeled(Func func) const{
        func(v_);
    }

    template<Label FN>
    KOKKOS_INLINE_FUNCTION const auto& get() const{
        if constexpr(T::name==FN){
            return v_.v;
        }else{
            static_assert(T::name==FN, "No such field name found in this pack");
        }
    }

    template<Label FN>
    KOKKOS_INLINE_FUNCTION auto& get(){
        if constexpr(T::name==FN){
            return v_.v;
        }else{
            static_assert(T::name==FN, "No such field name found in this pack");
        }
    }
};

template<typename Pack1, typename Pack2>
struct PackConcatenator;

template<typename... Types1, typename... Types2>
struct PackConcatenator<Pack<Types1...>, Pack<Types2...>> {
    using type = Pack<Types1..., Types2...>; 
};

template<typename Pack1, typename Pack2>
using ConcatenatePacks = typename PackConcatenator<Pack1, Pack2>::type;


#endif