heat_diagnostics.hpp

Go to the documentation of this file.

#ifndef HEAT_DIAGNOSTICS_HPP
#define HEAT_DIAGNOSTICS_HPP

#include "grid.hpp"

namespace HeatDiag{
    enum HEAT_DIAG{
        NVAR = 5
    };
}

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

    // Constructor for Mode 1
    HeatDiagnostics(int heat_mode_in, int heat_nsection_in, int heat_npsi_in, int heat_nr_in, int heat_nz_in,
        double *heat_rmin_in, double *heat_rmax_in, double *heat_dr_in, double *heat_zmin_in, double *heat_zmax_in, double *heat_dz_in,
        double *heat_pmin_in, double *heat_dp_in, int nspecies, int nthreads);

    // Constructor for Mode 2
    HeatDiagnostics(int heat_mode_in, int heat_nseg_in, int heat_nphi_in, double heat_delta_phi_in, double *heat_r2_in, double *heat_z2_in, int nspecies, int nthreads);

    // Null constructor
    HeatDiagnostics(){}

    // Used in heat_port (in pushe)
    int heat_mode;              

    // Used in Mode 1
    int heat_nsection;          
    int heat_npsi;              
    int heat_nr;                
    int heat_nz;                
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_rmin; 
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_rmax; 
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_dr;   
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_zmin; 
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_zmax; 
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_dz;   
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_pmin; 
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_dp;   

    Kokkos::View<double******,Kokkos::LayoutRight,Device> heat_pv;           
    Kokkos::View<double*****,Kokkos::LayoutRight,Device> heat_pv_psi;        

    // Used in Mode 2
    int heat_nseg;              
    int heat_nphi;              
    double heat_delta_phi;      
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_r2;            
    Kokkos::View<double*,Kokkos::LayoutRight,Device> heat_z2;            
    Kokkos::View<double*****,Kokkos::LayoutRight,Device> heat_pv_psi2;       

    // Sheath heat diagnostic
    KOKKOS_INLINE_FUNCTION void heat_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