local_equil_profiles.hpp

Go to the documentation of this file.

#ifndef LOCAL_EQUIL_PROFILES_HPP
#define LOCAL_EQUIL_PROFILES_HPP

#include "magnetic_field.hpp"
#include "species.hpp"

// f0 profile information is required for conventional delta-f method (in dw/dt)
template<MarkerType MT> struct LocalEquilProfiles;

template<>
struct LocalEquilProfiles<MarkerType::ReducedDeltaF>{
    Simd<double> f0tev;
    Simd<double> f0dden_n;
    Simd<double> f0dtev;
    Simd<double> f0flow;
    Simd<double> f0dflow;

    KOKKOS_INLINE_FUNCTION LocalEquilProfiles(const MagneticField<DeviceType>& magnetic_field, const Species<DeviceType>& species, const Simd<double>& psi, const SimdVector2D& x){
        bool ftype_non_zero=(species.eq_flow_type>0);

        for (int i_simd = 0; i_simd<SIMD_SIZE; i_simd++){
            f0tev[i_simd] = species.eq_temp.value(magnetic_field, psi[i_simd],x.r[i_simd],x.z[i_simd]);
            f0dtev[i_simd] = species.eq_temp.slope(magnetic_field, psi[i_simd],x.r[i_simd],x.z[i_simd]);
            f0dden_n[i_simd] = species.eq_den.slope(magnetic_field, psi[i_simd],x.r[i_simd],x.z[i_simd])/
                               species.eq_den.value(magnetic_field, psi[i_simd],x.r[i_simd],x.z[i_simd]);
            f0flow[i_simd] = species.eq_flow.value(magnetic_field, psi[i_simd],x.r[i_simd],x.z[i_simd]);
            f0dflow[i_simd] = species.eq_flow.slope(magnetic_field, psi[i_simd],x.r[i_simd],x.z[i_simd]);

            // Only flow_type=0 give consistent weight evolution
            // When flow_type>0, the poloidal derivative of flow is ignored and inconsitent.
            // Here, only unit conversion from omega to flow is considered.
            f0flow[i_simd]=(ftype_non_zero)? f0flow[i_simd]*x.r[i_simd] : f0flow[i_simd];
            f0dflow[i_simd]=(ftype_non_zero)? f0dflow[i_simd]*x.r[i_simd] : f0dflow[i_simd];
        }
    }
};

template<>
struct LocalEquilProfiles<MarkerType::TotalF>{
    KOKKOS_INLINE_FUNCTION LocalEquilProfiles(const MagneticField<DeviceType>& magnetic_field, const Species<DeviceType>& species, const Simd<double>& psi, const SimdVector2D& x){}
};

#endif