diag_3d_f0_f.hpp

Go to the documentation of this file.

#ifndef DIAG_3D_F0_F_HPP
#define DIAG_3D_F0_F_HPP
 
#include "diagnostic.hpp"
#include "grid.hpp"
#include "domain_decomposition.hpp"
#include "electric_field.hpp"
#include "perturbed_B_field.hpp"
#include "vgrid_distribution.hpp"
#include "plasma.hpp"
#include "moments.hpp"
 
class Diag3DF0F : public Diagnostic{
 
    public: // Cuda workaround for: A type that is a private or protected class member cannot be used in the type of a variable captured by an extended __device__ or __host__ __device__ lambda
 
    enum VarNames{
        ToroidalVelocity=0,
        PoloidalVelocity,
        ParallelVelocity,
        RadialFluxExB,
        RadialFluxMag,
        RadialFlux3DB,
        TorAngMom,
        RadMomFluxExB,
        RadMomFluxMag,
        RadMomFlux3DB,
        RadEnFluxExB,
        RadEnFluxMag,
        RadEnFlux3DB,
        RadPotEnFluxExB,
        RadPotEnFluxExBTurb,
        RadPotEnFluxMag,
        RadPotEnFlux3DB,
        ParMomFlux,
        ParEnFlux,
        ParPotEnFlux,
        NVARS
    };
 
    std::vector<std::string> var_names;
 
    // Precalculated values
    View<double*, CLayout, DeviceType> v_curv_rad_fsa;
    View<double*, CLayout, DeviceType> v_grad_B_rad_fsa;
 
    struct Drifts{
        View<double****,CLayout,DeviceType> v_exb;
        View<double****,CLayout,DeviceType> v_mag;
        View<double****,CLayout,DeviceType> v_par;
        View<double**,CLayout,DeviceType> boltz_fac;
        View<double**,CLayout,DeviceType> dpot;
        
        Drifts(){}
        
        Drifts(const VGridDistribution<DeviceType>& f0_f)
          : v_exb(NoInit("v_exb"), f0_f.n_vr(), f0_f.n_nodes(), f0_f.n_vz(),3),
            v_mag(NoInit("v_mag"), v_exb.layout()),
            v_par(NoInit("v_par"), v_exb.layout()),
            boltz_fac(NoInit("boltz_fac"), f0_f.n_vr(), f0_f.n_nodes()),
            dpot(NoInit("dpot"), boltz_fac.layout())
        {}
    };
 
    public:
 
    bool adios_stage;
 
    Diag3DF0F() : Diagnostic() {}
 
    void init(NLReader::NamelistReader& nlr, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, int default_period);
 
    Drifts get_drift_velocities(const MagneticField<DeviceType>& magnetic_field,
                                const Grid<DeviceType>& grid,
                                const DomainDecomposition<DeviceType>& pol_decomp,
                                const VelocityGrid& vgrid,
                                Species<DeviceType>& species,
                                const PerturbedBField<DeviceType> &perturbed_B_field,
                                const Simulation<DeviceType>& sml,
                                const VGridDistribution<DeviceType>& f0_f,
                                const View<double*,CLayout,DeviceType>& T0,
                                const View<double*,CLayout,DeviceType> pot0_d,
                                const View<double*,CLayout,DeviceType>& dpot_d,
                                const GridField<DeviceType,VarType::Scalar,PIT_GLOBAL,TorType::OnePlane,KinType::GyroKin>& pot,
                                const GridFieldPackPtr& gfpack_ptr,
                                bool use_turb);
 
    Drifts copy_drift_velocities(const Drifts& d_in);
 
    void calculate_diagnostics(const Simulation<DeviceType>& sml, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, const DomainDecomposition<DeviceType>& pol_decomp, Plasma& plasma, const VelocityGrid& vgrid, ElectricField<DeviceType>& electric_field, PerturbedBField<DeviceType> &perturbed_B_field, const VGridDistribution<DeviceType>& f0_f,
                            const View<double**,CLayout,DeviceType>& T0, const View<double**,CLayout,DeviceType>& n0,
                            const View<double**,CLayout,DeviceType>& n,
                            View<double***,CLayout,HostType>& moments_adia_n0,
                            View<double***,CLayout,HostType>& moments_nonadia_n0,
                            View<double***,CLayout,HostType>& moments_adia_turb,
                            View<double***,CLayout,HostType>& moments_nonadia_turb,
                            View<double****,CLayout,HostType>& moments_nonadia_turb_en);
 
    void write(const Simulation<DeviceType>& sml, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, const DomainDecomposition<DeviceType>& pol_decomp, Plasma& plasma, PerturbedBField<DeviceType> &perturbed_B_field, const VelocityGrid& vgrid, ElectricField<DeviceType>& electric_field,
                  const VGridDistribution<DeviceType>& f0_f, const Moments& moments);
};
 
#endif