neutrals.hpp

Go to the documentation of this file.

#ifndef NEUTRALS_HPP
#define NEUTRALS_HPP
 
#include "view_arithmetic.hpp"
#include "xgc_io.hpp"
#include "NamelistReader.hpp"
#include "grid.hpp"
#include "plasma.hpp"
#include "vgrid_distribution.hpp"
#include "diag_f0_df.hpp"
#include "step_trigger.hpp"
 
// Neutrals class
template<class Device>
class Neutrals {
public:
 
    // Neutrals variables
    bool use_ion_loss;         
    int nebins;
    int background_period;
    double ebin_min;
    double ebin_max;
    double ebin_delta;
    bool ebin_log;
 
    // Step trigger for the neutral background update
    StepTrigger trigger_neutral_update;
                               //
    View<double*,CLayout,Device> weight_sum_lost;   
    View<double**,CLayout,Device> weight_wall_lost;  
    View<double***,CLayout,Device> weight_lost_ebin;  
 
    View<double*,CLayout,HostType> weight_sum_lost_h;   
    View<double*,CLayout,HostType> weight_wall_lost_h;  
    View<double**,CLayout,HostType> weight_lost_ebin_h;  
    View<double*,CLayout,HostType> node_dens_h;
    View<double*,CLayout,HostType> node_temp_h;
 
    // Constructor
    Neutrals( int nthreads, int nbins, int nnode, NLReader::NamelistReader& nlr);
 
    // "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 clear_host_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<DeviceType>& f0_df0g_cumul, const int axisym_parallelization_int, int gstep, DiagF0DF& diag_f0_df, double dt) const;
};
#endif