diag_3d.hpp

Go to the documentation of this file.

#ifndef DIAG_3D_HPP
#define DIAG_3D_HPP
 
#include "diagnostic.hpp"
#include "grid.hpp"
#include "plasma.hpp"
#include "charge.hpp"
#include "electric_field.hpp"
#include "sheath.hpp"
#include "perturbed_B_field.hpp"
 
class Diag3D : public Diagnostic{
 
    View<double**,CLayout,HostType> E_saved;
 
    View<double***,CLayout,HostType> diag_3d_add;
 
    View<double**,CLayout,HostType> marker_den;
    View<double**,CLayout,HostType> weight_mean;
    View<double**,CLayout,HostType> weight_variance;
 
    public:
 
    bool adios_stage;
    bool weight_stats;
    bool additional_moments;
 
    Diag3D() : Diagnostic() {}
 
    void init(NLReader::NamelistReader& nlr, const Grid<DeviceType>& grid, const Plasma& plasma, int default_period);
 
    void write(const Grid<DeviceType>& grid, const Simulation<DeviceType>& sml, const Sheath<DeviceType>& sheath, 
               const PerturbedBField<DeviceType>& perturbed_B_field, const DomainDecomposition<DeviceType>& pol_decomp,
               Plasma& plasma, const Charges& charges, const ElectricField<DeviceType>& electric_field,
               double time, int gstep);
 
    void additional(const DomainDecomposition<DeviceType>& pol_decomp, const Simulation<DeviceType>& sml, const Grid<DeviceType>& grid, const MagneticField<DeviceType>& magnetic_field, Plasma& plasma) const;
 
    void diag_marker_den(const DomainDecomposition<DeviceType>& pol_decomp, const Grid<DeviceType>& grid, const MagneticField<DeviceType>& magnetic_field, Plasma& plasma, bool is_XGCa) const;
 
    void calculate_E_out(const Grid<DeviceType>& grid, const ElectricField<DeviceType>& electric_field, const View<double**,CLayout,HostType>& E_grid,
                         View<double**,CLayout,HostType>& E_out, View<double**,CLayout,HostType>& dEdt_out, double dt);
 
    void store_E_field(const ElectricField<DeviceType>& electric_field);
 
    View<double**,CLayout,HostType> calculate_B_out_base(const Grid<DeviceType>& grid, const View<double**,CLayout,HostType>& B_grid);
 
    void add_B_out_ptb_3db(const Grid<DeviceType>& grid, const PerturbedBField<DeviceType>& perturbed_B_field, const View<double**,CLayout,HostType>& B_out, double phi);
 
    View<double**,CLayout,HostType> calculate_B_out(const Simulation<DeviceType>& sml, const Grid<DeviceType>& grid, const DomainDecomposition<DeviceType>& pol_decomp, const PerturbedBField<DeviceType>& perturbed_B_field, const View<double**,CLayout,HostType>& B_grid);
};
 
#endif