neutrals.hpp

Go to the documentation of this file.

#ifndef NEUTRALS_HPP
#define NEUTRALS_HPP

#include "view_arithmetic.hpp"
#include "xgc_io.hpp"
#include "grid.hpp"
#include "plasma.hpp"
#include "vgrid_distribution.hpp"
#include "col_grid.hpp"
#include "diag_f0_df.hpp"

// Neutrals class
template<class Device>
class Neutrals {
public:

    // Neutrals variables
    bool use_ion_loss;         
    View<double*,CLayout,Device> weight_sum_lost;   
    View<double**,CLayout,Device> weight_wall_lost;  

    View<double*,CLayout,HostType> weight_sum_lost_h;   
    View<double*,CLayout,HostType> weight_wall_lost_h;  
    View<double*,CLayout,HostType> node_dens_h;
    View<double*,CLayout,HostType> node_temp_h;

    // Constructor
    Neutrals( int nthreads, int nbins, int nnode);

    // "Null constructor"
    Neutrals(){}

    void write_checkpoint_files(const XGC_IO_Stream& stream) const;

    void read_checkpoint_files(const XGC_IO_Stream& stream);

    void copy_to_host() const;

    void clear_device_views() const;

    void apply(const Grid<DeviceType>& grid, const DomainDecomposition<DeviceType>& pol_decomp, const VelocityGrid& vgrid, Plasma& plasma, const VGridDistribution<DeviceType>& f0_f, const VGridDistribution<HostType>&  f0_f_h, VGridDistribution<HostType>& f0_df0g_cumul, const int axisym_parallelization_int, int gstep, DiagF0DF& diag_f0_df) const;
};
#endif