diag_1d.hpp

Go to the documentation of this file.

#ifndef DIAG_1D_HPP
#define DIAG_1D_HPP

#include "diagnostic.hpp"
#include "grid.hpp"
#include "plasma.hpp"
#include "charge.hpp"
#include "perturbed_B_field.hpp"

class Diag1D : public Diagnostic{

    enum VarNames{
        GC_Density=0,
        GC_ToroidalFlow,
        GC_PoloidalFlow,
        ParallelFlow,
        TorAngMom,
        RadialFlux,
        ParallelMeanEnergy,
        PerpTemp,
        RadialEnFlux,
        RadialMomFluxExB,
        RadialMomFlux,
        RadialFluxExB,
        RadialEnFluxExB,
        PoloidalExBFlow,
        VParxB,
        GradPsiSqr,
        RadialFlux3DB,
        RadialEnFlux3DB,
        MarkerDensity,
        NVARS
    };

    std::vector<std::string> var_names;

    bool deltaf_diag_on;
    bool f0_diag_on;

    View<double***,CLayout,HostType> d1d_f_pv1_h;
    View<double***,CLayout,HostType> d1d_df_pv1_h;

    View<double*,CLayout,HostType> diag_1d_vol_h;

    // Precalculated values
    View<double*, CLayout, DeviceType> v_curv_rad_fsa;
    View<double*, CLayout, DeviceType> v_grad_B_rad_fsa;

    public:

    Diag1D() : Diagnostic() {}

    void write_volume(const VolumesAndAreas& volumes_and_areas);

    void init(NLReader::NamelistReader& nlr, const DomainDecomposition<DeviceType>& pol_decomp, const Grid<DeviceType>& grid, const MagneticField<DeviceType>& magnetic_field, const Plasma& plasma, bool allocate_df_diagnostic);

    View<double***,CLayout,HostType> diag_1d_f0(const Grid<DeviceType>& grid, const MagneticField<DeviceType>& magnetic_field, const DomainDecomposition<DeviceType>& pol_decomp, const VelocityGrid& vgrid, Plasma& plasma, PerturbedBField<DeviceType> &perturbed_B_field, const ElectricField<DeviceType>& electric_field, const Simulation<DeviceType>& sml);

    void write(const Grid<DeviceType>& grid, const MagneticField<DeviceType>& magnetic_field, const DomainDecomposition<DeviceType>& pol_decomp, const VelocityGrid& vgrid, Plasma& plasma, const Charges& charges, const ElectricField<DeviceType>& electric_field, PerturbedBField<DeviceType> &perturbed_B_field, const Simulation<DeviceType>& sml, double time, int gstep);

    void reduce_and_copy_to_host(const View<double****,CLayout,DeviceType>& d1d_f_pv1, const View<double****,CLayout,DeviceType>& d1d_df_pv1) const;

    void clear();
};

#endif