heat_diagnostics.hpp

Go to the documentation of this file.

#ifndef HEAT_DIAGNOSTICS_HPP
#define HEAT_DIAGNOSTICS_HPP

#include "grid.hpp"
#include "plasma.hpp"
#include "diagnostic.hpp"

namespace HeatDiag{
    enum HEAT_DIAG{
        NUMBER=0,
        PARA_ENERGY,
        PERP_ENERGY,
        POTENTIAL,
        DISTANCE,
        NVAR
    };
}

// Heat Diagnostics class
template<class Device>
class HeatDiagnostics : public Diagnostic {
    public:

    // Null constructor
    HeatDiagnostics(){}

    void init(NLReader::NamelistReader& nlr, const MagneticField<Device>& magnetic_field, const Grid<Device>& grid, int nspecies);

    HeatDiagnostics(NLReader::NamelistReader& nlr, const MagneticField<Device>& magnetic_field, const Grid<Device>& grid, int nspecies);

    // Used in port (in pushe)
    int mode;              

    // Used in Mode 1
    int nsection;          
    int npsi;              
    int nr;                
    int nz;                
    View<double*,CLayout,Device> rmin; 
    View<double*,CLayout,Device> rmax; 
    View<double*,CLayout,Device> dr;   
    View<double*,CLayout,Device> zmin; 
    View<double*,CLayout,Device> zmax; 
    View<double*,CLayout,Device> dz;   
    View<double*,CLayout,Device> pmin; 
    View<double*,CLayout,Device> dp;   

    View<double******,CLayout,Device> pv;           
    View<double*****,CLayout,Device> pv_psi;        

    // Used in Mode 2
    int nseg;              
    int nphi;              
    double delta_phi;      
    View<RZPair*,CLayout,Device> x;
    View<double*,CLayout,HostType> psi2; // Could be removed, calculated in first write
    View<double*,CLayout,HostType> ds2; // Could be removed, calculated in first write
    View<double*,CLayout,HostType> strike_angle2; // Could be removed, calculated in first write
    View<double*****,CLayout,Device> pv_psi2;       

    int get_nspecies() const;

    void write(Plasma& plasma, double time, int gstep);

    // Sheath heat diagnostic
    KOKKOS_INLINE_FUNCTION void port(const Grid<Device> &grid, const MagneticField<Device> &magnetic_field, const Simd<bool>& just_left_the_grid, const Simd<double>& w, const Simd<double>& pot, const Simd<double>& epara, const Simd<double>& eperp, const SimdConstants &ct, const SimdPhase &old_ph, const SimdVector2D &xff_old, const SimdPhase &new_ph, int stype) const;

    // Routine used in heat diagnostics - should be elsewhere?
    KOKKOS_INLINE_FUNCTION int wall_cross2(double r1, double z1, double r2, double z2) const;

};

#include "heat_diagnostics.tpp"
#endif