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
struct LocalFields {
#ifdef XGC1
    // Temporarily here to make PR simpler:
# ifdef EXPLICIT_EM
    using gfpack_type = Pack<Labeled<EfieldType, Label::E>,
                             Labeled<EfieldType, Label::dAh>,
                             Labeled<EfieldType, Label::dAs>,
                             Labeled<PotType, Label::Ah>,
                             Labeled<PotType, Label::As>,
                             Labeled<PotType, Label::Epar_em>>;
# else
#  ifdef DELTAF_CONV
    using gfpack_type = Pack<Labeled<EfieldType, Label::E>,
                             Labeled<E00Type, Label::E00>,
                             Labeled<PotType, Label::ddpotdt>>;
#  else
    using gfpack_type = Pack<Labeled<EfieldType, Label::E>>;
#  endif
# endif
#else
# ifdef SONIC_GK
    using gfpack_type = Pack<Labeled<EfieldType, Label::E>,
                             Labeled<E00GyroType, Label::dEr_B2>,
                             Labeled<E00GyroType, Label::dEz_B2>,
                             Labeled<E00GyroType, Label::du2_E_rho>>;
# else
    using gfpack_type = Pack<Labeled<EfieldType, Label::E>>;
# endif
#endif
    //

    using pack_type = typename getLocalPackType<gfpack_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