radiation.hpp

Go to the documentation of this file.

#ifndef RADIATION_HPP
#define RADIATION_HPP

#include "grid.hpp"
#include "magnetic_field.hpp"
#include "domain_decomposition.hpp"
#include "velocity_grid.hpp"
#include "plasma.hpp"
#include "vgrid_distribution.hpp"
#include "moments.hpp"
#include "source.hpp"

class Radiation : public Source{

    double psi_min; 
    double psi_max; 

    View<double*, CLayout, HostType> rad_Te_ev_h;
    View<double*, CLayout, HostType> rad_n0_ne_h;
    View<double**, CLayout, HostType> rad_Lz_h;
    View<double**, CLayout, HostType> rad_avgZ_h;
    View<double**, CLayout, HostType> rad_avgZ2_h;

    View<double*, CLayout, DeviceType> rad_Te_ev;
    View<double*, CLayout, DeviceType> rad_n0_ne;
    View<double**, CLayout, DeviceType> rad_Lz;
    View<double**, CLayout, DeviceType> rad_avgZ;
    View<double**, CLayout, DeviceType> rad_avgZ2;

    double impurity_fraction;
    bool use_zeff_profile;
    bool use_fix_charge;
    double fixedZ;

    Eq::Profile<DeviceType> eq_zeff;

    public:

    Radiation(){}

    Radiation(NLReader::NamelistReader& nlr, const Grid<DeviceType>& grid, const MagneticField<DeviceType>& magnetic_field);

    KOKKOS_INLINE_FUNCTION void Rad_Emissivity2(double tval_ev_in,double n0_ne_in, double& Lz,double& avgZ,double& avgZ2) const;

    KOKKOS_INLINE_FUNCTION double get_impurity_fraction(const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, int inode, double avgZ, double avgZ2) const;

    KOKKOS_INLINE_FUNCTION double power_loss(double n, double imp_frac, double Lz, double dt) const;

    KOKKOS_INLINE_FUNCTION void f_shift_from_f_heat_torque(const VelocityGrid& vgrid, double alpha, const VGridDistribution<DeviceType>& f0_f, const VGridDistribution<DeviceType>& f0_df0g, int inode_local) const;

    KOKKOS_INLINE_FUNCTION bool out_of_bounds(const Grid<DeviceType>& grid, int inode) const;

    void apply(const Grid<DeviceType>& grid, const MagneticField<DeviceType>& magnetic_field, const DomainDecomposition<DeviceType>& pol_decomp, const VelocityGrid& vgrid, const Moments& moments, const View<double*, CLayout, HostType>& neu_node_dens, const VGridDistribution<HostType>& f0_f, const VGridDistribution<HostType>& f0_df0g, double dt);
};

#endif