sheath.hpp

Go to the documentation of this file.

#ifndef SHEATH_HPP
#define SHEATH_HPP
#include "push_controls.hpp"
#include "species.hpp"
#include "neutrals.hpp"
#include "heat_diagnostics.hpp"

struct SheathVec {
    double r;
    double z;
    double phi;
};

// Sheath particle class to store particle info if using sheath mode 2
template<class Device>
class SheathParticles {
    public:

    // Constructor
    SheathParticles(int n_threads, int sheath_nphi_in, int grid_nwall_in, double *en_max_initial, int n_ptl_in);

    // Null constructor
    SheathParticles(){}

    // Sheath variables
    int n_ptl;     

    // Sheath mode 2 array
    Kokkos::View<double***,Kokkos::LayoutRight,Device> en_max; 
    // Arrays of particle properties to track
    Kokkos::View<int*,Kokkos::LayoutRight,Device> widx;           
    Kokkos::View<int*,Kokkos::LayoutRight,Device> iphi;           
    Kokkos::View<double*,Kokkos::LayoutRight,Device> en_para;     
    Kokkos::View<double*,Kokkos::LayoutRight,Device> en_perp;     
    Kokkos::View<SheathVec*,Kokkos::LayoutRight,Device> ph_back;  
    Kokkos::View<SheathVec*,Kokkos::LayoutRight,Device> ph_out;   
};


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

    // Constructor
    Sheath(const Grid<Device>& grid, const Simulation<Device>& sml);

    // Null constructor
    Sheath(){}

    // Sheath variables
    Kokkos::View<double**,Kokkos::LayoutRight,Device> pot;   
    Kokkos::View<double***,Kokkos::LayoutRight,Device> lost;  
    Kokkos::View<double***,Kokkos::LayoutRight,Device> ilost;  

    KOKKOS_INLINE_FUNCTION void deposition_diagnostics_and_w1(const Species<Device> &species, const Grid<Device> &grid, const MagneticField<Device> &magnetic_field, const Neutrals<Device> &neutrals, const HeatDiagnostics<Device> &heat_diag, const SimdParticles &old_part,SimdParticles &part, const SimdVector2D& xff, const Simd<int>& iphi, const Simd<double>& en_para, const Simd<double>& en_perp, SimdPhase& new_phase, const Simd<int>& widx, const Simd<bool>& just_left_the_grid, bool update_lost, bool update_ilost, bool update_neutrals, bool update_heat_diag, Simd<double>& dw) const;

    KOKKOS_INLINE_FUNCTION void save_sheath_ptl_info(const PushControls &push_controls, const Species<Device> &species, const Grid<Device> &grid, const SheathParticles<Device> &sheath_particles, int i_item, const SimdParticles &old_part,SimdParticles &part, const Simd<int>& iphi, const Simd<double>& en_para, const Simd<double>& en_perp, const SimdPhase& new_phase, const Simd<int>& widx, const Simd<bool>& just_left_the_grid) const;

    KOKKOS_INLINE_FUNCTION void sheath_calculation(const PushControls &push_controls, const Species<Device> &species, const Grid<Device> &grid, const MagneticField<Device> &magnetic_field, const Neutrals<Device> &neutrals, const HeatDiagnostics<Device> &heat_diag, const SheathParticles<Device> &sheath_particles, int i_item, const SimdParticles &old_part,const Simd<bool>& not_in_triangle,SimdParticles &part,Simd<int>& itrout,SimdGridVec &pout, Simd<double>& dw) const;

};

#include "sheath.tpp"
#endif