pushe_diagnostics.hpp

Go to the documentation of this file.

#ifndef PUSHE_DIAGNOSTICS_HPP
#define PUSHE_DIAGNOSTICS_HPP

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

namespace Diag{
    enum VF_DIAG{
        N_VF_DIAG = 9
    };

    enum D1D_PV1{
#ifdef XGC1
        D1D_NPV1 = 16
#else
        D1D_NPV1 = 18
#endif
    };
}

// Diagnostics class
template<class Device>
class Diagnostics {
    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
    Diagnostics(NLReader::NamelistReader& nlr, bool allocate_df_diagnostic, int np, int nspecies, int nthreads);

    // Null constructor
    Diagnostics(){}

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

    // 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 sp_type, const Simd<double> *vd) const;

};

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

#include "pushe_diagnostics.tpp"
#endif