blob: 1e5385d7645cb725b6cc462615d9f0a343d2f143 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
#ifndef TRANSFORMPARTICLE_H_
#define TRANSFORMPARTICLE_H_
#include "WarpXParticleContainer.H"
enum struct elementaryProcessType { Ionization };
/**
* \brief small modifications on source particle
* \param i index of particle
* \param particle particle struct
* \param runtime_iattribs integer attribs
*/
template < elementaryProcessType ProcessT >
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
void transformSourceParticle(
int i,
WarpXParticleContainer::ParticleType& particle,
amrex::GpuArray<int*,1> runtime_iattribs){}
/**
* \brief small modifications on product particle
* \param i index of particle
* \param particle particle struct
*/
template < elementaryProcessType ProcessT >
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
void transformProductParticle(
int i,
amrex::Vector<WarpXParticleContainer::ParticleType*> v_particle){}
// For ionization, increase ionization level of source particle
template <>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
void transformSourceParticle < elementaryProcessType::Ionization > (
int i,
WarpXParticleContainer::ParticleType& particle,
amrex::GpuArray<int*,1> runtime_iattribs)
{
// increment particle's ionization level
runtime_iattribs[0][i] += 1;
}
#endif // TRANSFORMPARTICLE_H_
|
