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"

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;

    public:

    Diag1D() : Diagnostic() {}

    void write_volume();

    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);

    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, 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