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 "vgrid_distribution.hpp"
#include "plasma.hpp"
#include "moments.hpp"

class Diag3DF0F : public Diagnostic{
    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;

    struct Drifts{
        View<double*****,CLayout,HostType> v_exb;
        View<double*****,CLayout,HostType> v_mag;
        View<double*****,CLayout,HostType> v_par;
        View<double***,CLayout,HostType> boltz_fac;
        View<double***,CLayout,HostType> dpot;
        
        Drifts(){}
        
        Drifts(const VGridDistribution<HostType>& 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, int default_period);

    KOKKOS_INLINE_FUNCTION void get_diag_vals(double mass, double charge, double angular_conv_fac, double vp, double en, double* v_mag, double* v_exb, double* v_par,
                   double dpot0, double dpot1, double v_exb_n0_r, double v_exb_turb_r, double v_mag_tot_r, double v_pardrift_tot_r, bool include_vpar,
                   double* v);


    Drifts get_drift_velocities(const DomainDecomposition<DeviceType>& pol_decomp, const VelocityGrid& vgrid, ElectricField<DeviceType>& electric_field, Plasma& plasma,
                            const VGridDistribution<HostType>& f0_f, const View<double**,CLayout,HostType>& 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, const View<double****,CLayout,HostType>& ftot_n0, const VGridDistribution<HostType>& f0_f, const Drifts& d, const Drifts& dn0,
                            const View<double**,CLayout,HostType>& T0, const View<double**,CLayout,HostType>& n0,
                            const View<double**,CLayout,HostType>& n,
                            const View<double***,CLayout,HostType>& moments_adia_n0,
                            const View<double***,CLayout,HostType>& moments_nonadia_n0,
                            const View<double***,CLayout,HostType>& moments_adia_turb,
                            const View<double***,CLayout,HostType>& moments_nonadia_turb,
                            const 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, const VelocityGrid& vgrid, ElectricField<DeviceType>& electric_field,
                  const VGridDistribution<HostType>& f0_f, const Moments& moments);
};

#endif