local_fields.hpp

Go to the documentation of this file.

#ifndef LOCAL_FIELDS_HPP
#define LOCAL_FIELDS_HPP

#include "globals.hpp"
#include "pack.hpp"
#include "grid_field.hpp"

template<VarType VT> struct getLocalFldType;
template<> struct getLocalFldType<VarType::Scalar>{ using var_type = Simd<double>; };
template<> struct getLocalFldType<VarType::Vector>{ using var_type = SimdVector; };
// The result vector is still 3D even if the input is 2D. We could potentially change this if it is less confusing/saves memory
template<> struct getLocalFldType<VarType::Vector2D>{ using var_type = SimdVector; };

template<class LGFType> struct getLocalLabeledFld{
    using labeled_type = Labeled<typename getLocalFldType<LGFType::var_type::field_type::VT>::var_type, LGFType::name>;
};

template<class... Rest> struct LocalPackType;

template<class... Rest> struct LocalPackType<Pack<Rest...>>{
    using pack_type = Pack<typename getLocalLabeledFld<Rest>::labeled_type...>;
};

/* Struct used at compile time to hold the template parameters needed to determine the structure of the fields */
template<class GFPackType> struct getLocalPackType{
//    using pack_type = typename LocalPackType<typename GFPackType::pack_type>::pack_type;
// Temporary:
    using pack_type = typename LocalPackType<GFPackType>::pack_type;
};

// Fields at a set of points, interpolated from solved grid quantities
template<typename gen_pack_type>
struct LocalFields {
    using pack_type = typename getLocalPackType<gen_pack_type>::pack_type;

    pack_type pack;

    template<Label FN>
    KOKKOS_INLINE_FUNCTION const auto& get() const{
        return pack.template get<FN>();
    }

    template<Label FN>
    KOKKOS_INLINE_FUNCTION auto& get(){
        return pack.template get<FN>();
    }
};

#endif