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,
        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_species(), 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_species(), 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, ElectricField<DeviceType>& electric_field, Plasma& plasma, PerturbedBField<DeviceType> &perturbed_B_field, const Simulation<DeviceType>& sml,
                            const VGridDistribution<DeviceType>& f0_f, const View<double**,CLayout,DeviceType>& T0, 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