blob: fb60d7a5f6e0821363386cc9804b8530588df7b2 (
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
/* Copyright 2021 Axel Huebl
*
* This file is part of WarpX.
*
* License: BSD-3-Clause-LBNL
*/
#ifndef PARTICLEIO_H_
#define PARTICLEIO_H_
#include "Particles/WarpXParticleContainer.H"
#include <AMReX_AmrParticles.H>
#include <AMReX_Gpu.H>
#include <AMReX_REAL.H>
/** Convert particle momentum to/from SI
*
* Particle momentum is defined as gamma*velocity, which is neither
* SI mass*gamma*velocity nor normalized gamma*velocity/c.
* This converts momentum to SI units (or vice-versa) to write SI data
* to file.
* Photons are a special case, since particle momentum is defined as
* (photon_energy/(m_e * c) ) * u, where u is the photon direction (a
* unit vector).
*
* @tparam T_ParticleContainer a WarpX particle container or AmrParticleContainer
* @param convert_dir convert to or from SI
* @param pc the particle container to manipulate
* @param mass the particle rest mass to use for conversion
*/
template< typename T_ParticleContainer >
void
particlesConvertUnits (ConvertDirection convert_direction, T_ParticleContainer* pc, amrex::ParticleReal const mass )
{
using namespace amrex;
// Compute conversion factor
auto factor = 1_rt;
if (convert_direction == ConvertDirection::WarpX_to_SI){
factor = mass;
} else if (convert_direction == ConvertDirection::SI_to_WarpX){
factor = 1._rt/mass;
}
const int nLevels = pc->finestLevel();
for (int lev=0; lev<=nLevels; lev++){
#ifdef AMREX_USE_OMP
#pragma omp parallel if (Gpu::notInLaunchRegion())
#endif
for (WarpXParIter pti(*pc, lev); pti.isValid(); ++pti)
{
// - momenta are stored as a struct of array, in `attribs`
auto& attribs = pti.GetAttribs();
ParticleReal* AMREX_RESTRICT ux = attribs[PIdx::ux].dataPtr();
ParticleReal* AMREX_RESTRICT uy = attribs[PIdx::uy].dataPtr();
ParticleReal* AMREX_RESTRICT uz = attribs[PIdx::uz].dataPtr();
// Loop over the particles and convert momentum
const long np = pti.numParticles();
ParallelFor( np,
[=] AMREX_GPU_DEVICE (long i) {
ux[i] *= factor;
uy[i] *= factor;
uz[i] *= factor;
}
);
}
}
}
#endif /* PARTICLEIO_H_ */
|
