push_diagnostic.hpp

Go to the documentation of this file.

#ifndef PUSH_DIAGNOSTIC_HPP
#define PUSH_DIAGNOSTIC_HPP

#include "push_controls.hpp"
#include "electric_field.hpp"
#include "perturbed_B_field.hpp"

enum class PushDiagToggle{On, Off};

namespace Diag{
    enum D1D_PV1{
        D1D_NPV1 = 19
    };
}

template<PushDiagToggle PDT>
struct VFDiag{};

template<>
struct VFDiag<PushDiagToggle::On>{
    enum VF_DIAG{
        PSI=0,
        BMAG,
        BPHI,
        V_ExB_GRADPSI,
        V_d_GRADPSI,
        GRADPSI2,
        V_d_B_POL,
        V_ExB_B_POL,
        TDB_GRADPSI,
        N_VF_DIAG
    };

    Simd<double> vd[N_VF_DIAG];

    KOKKOS_INLINE_FUNCTION const Simd<double>& operator [](int i) const {return vd[i];}
    KOKKOS_INLINE_FUNCTION Simd<double>& operator [](int i) {return vd[i];}

    KOKKOS_INLINE_FUNCTION void set(int i_simd, double fac, const Simd<double>& psi, double B, const SimdVector& bfield, const SimdVector2D& gradpsi, const SimdVector& tdb, const double (&yp_exb)[3], const SimdPhase& yprime){
        vd[PSI][i_simd]  = psi[i_simd];
        vd[BMAG][i_simd] = B;
        vd[BPHI][i_simd] = bfield.phi[i_simd];
        vd[V_ExB_GRADPSI][i_simd] = yp_exb[PIR]*gradpsi.r[i_simd] + yp_exb[PIZ]*gradpsi.z[i_simd];   // V_ExB dot grad psi
        vd[V_d_GRADPSI][i_simd] = yprime.r[i_simd]*gradpsi.r[i_simd] + yprime.z[i_simd]*gradpsi.z[i_simd];   // V_d dot grad psi
        vd[GRADPSI2][i_simd]    = gradpsi.r[i_simd]*gradpsi.r[i_simd] + gradpsi.z[i_simd]*gradpsi.z[i_simd];  // |grad psi|^2
        double Bp = bfield.poloidal_magnitude(i_simd);
        vd[V_d_B_POL][i_simd]   = (yprime.r[i_simd]*bfield.r[i_simd] + yprime.z[i_simd]*bfield.z[i_simd])/Bp;
        vd[V_ExB_B_POL][i_simd] = (yp_exb[PIR]*bfield.r[i_simd] + yp_exb[PIZ]*bfield.z[i_simd])/Bp;
        vd[TDB_GRADPSI][i_simd] = fac * (tdb.r[i_simd]*gradpsi.r[i_simd] + tdb.z[i_simd]*gradpsi.z[i_simd]); // v_para*(deltaB.grad(psi))
    }
};

typedef VFDiag<PushDiagToggle::On> VF;

// PushDiagnostic class
template<class Device>
class PushDiagnostic {

    public:

    // Used in diag_1d_port1 (in pushe)
    int d1d_period;       

    // Arrays
    Kokkos::View<double****,Kokkos::LayoutRight,Device> d1d_f_pv1;  
    Kokkos::View<double****,Kokkos::LayoutRight,Device> d1d_df_pv1; 

    // Constructor
    PushDiagnostic(NLReader::NamelistReader& nlr, bool allocate_df_diagnostic, int np, int nspecies, int nthreads);

    // Null constructor
    PushDiagnostic(){}

    // Check whether to record diagnostics
    inline bool determine_diag_on(int istep,int ipc);

    KOKKOS_INLINE_FUNCTION void scatter_single_diag(bool is_deltaf, int ithread, int isp, int ip, double w, double dw, double wp, int i_v, double v) const;

    // General 1D push diagnostics
    KOKKOS_INLINE_FUNCTION void d1d_port1(const Grid<Device> &grid, const Species<Device> &species, const MagneticField<Device> &magnetic_field, const SimdParticles &part,const SimdPhase &derivs,int isp, const VFDiag<PushDiagToggle::On>& vf) const;

};

template<class Device>
KOKKOS_INLINE_FUNCTION void push_diagnostics_c(const Grid<Device> &grid, 
    const MagneticField<Device> &magnetic_field, const ElectricField<Device> &electric_field, 
    const PushDiagnostic<Device> &diag, const PushControls &push_controls, const Species<Device> &species, 
    const PerturbedBField<Device> &perturbed_B_field, int);

#include "push_diagnostic.tpp"
#endif