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;

    public:

    bool adios_stage;
    bool weight_stats;
    bool additional_moments;

    Diag3D() : Diagnostic() {}

    void init(NLReader::NamelistReader& nlr, const Grid<DeviceType>& grid, 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 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