10 #ifndef IMPACTX_PARTICLE_CONTAINER_H
11 #define IMPACTX_PARTICLE_CONTAINER_H
28 #include <unordered_map>
64 static constexpr
auto names_s = {
"position_x",
"position_y",
"position_t",
"momentum_x",
"momentum_y",
"momentum_t",
"qm",
"weighting" };
66 static constexpr
auto names_t = {
"position_x",
"position_y",
"position_z",
"momentum_x",
"momentum_y",
"momentum_z",
"qm",
"weighting" };
82 static constexpr std::initializer_list<char const *>
names_s = {};
84 static constexpr std::initializer_list<char const *>
names_t = {};
150 void AddRealComp (std::string
const & name,
bool communicate=
true);
157 void AddIntComp (std::string
const & name,
bool communicate=
true);
183 amrex::ParticleReal qm,
184 amrex::ParticleReal bchchg
251 amrex::ParticleReal, amrex::ParticleReal,
252 amrex::ParticleReal, amrex::ParticleReal,
253 amrex::ParticleReal, amrex::ParticleReal>
261 amrex::ParticleReal, amrex::ParticleReal,
262 amrex::ParticleReal, amrex::ParticleReal,
263 amrex::ParticleReal, amrex::ParticleReal>
276 DepositCharge (std::unordered_map<int, amrex::MultiFab> & rho,
280 std::vector<std::string>
284 std::vector<std::string>
Definition: ImpactXParticleContainer.H:131
virtual ~ImpactXParticleContainer()=default
Destruct a particle container.
int GetParticleShape() const
Definition: ImpactXParticleContainer.H:226
void SetLostParticleContainer(ImpactXParticleContainer *lost_pc)
Definition: ImpactXParticleContainer.cpp:84
void DepositCharge(std::unordered_map< int, amrex::MultiFab > &rho, amrex::Vector< amrex::IntVect > const &ref_ratio)
Definition: ChargeDeposition.cpp:27
void AddNParticles(amrex::Gpu::DeviceVector< amrex::ParticleReal > const &x, amrex::Gpu::DeviceVector< amrex::ParticleReal > const &y, amrex::Gpu::DeviceVector< amrex::ParticleReal > const &t, amrex::Gpu::DeviceVector< amrex::ParticleReal > const &px, amrex::Gpu::DeviceVector< amrex::ParticleReal > const &py, amrex::Gpu::DeviceVector< amrex::ParticleReal > const &pt, amrex::ParticleReal qm, amrex::ParticleReal bchchg)
Definition: ImpactXParticleContainer.cpp:126
std::vector< std::string > intSoA_names() const
Definition: ImpactXParticleContainer.cpp:273
std::tuple< amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal > MinAndMaxPositions()
Definition: ImpactXParticleContainer.cpp:248
void SetRefParticle(RefPart const &refpart)
Definition: ImpactXParticleContainer.cpp:221
void AddIntComp(std::string const &name, bool communicate=true)
Definition: ImpactXParticleContainer.cpp:77
std::vector< std::string > m_int_soa_names
Int component names.
Definition: ImpactXParticleContainer.H:316
void AddRealComp(std::string const &name, bool communicate=true)
Definition: ImpactXParticleContainer.cpp:70
std::vector< std::string > RealSoA_names() const
Definition: ImpactXParticleContainer.cpp:267
CoordSystem GetCoordSystem() const
Definition: ImpactXParticleContainer.cpp:279
void SetRefParticleEdge()
Definition: ImpactXParticleContainer.cpp:239
ImpactXParticleContainer * m_particles_lost
a non-owning reference to lost particles, i.e., due to apertures
Definition: ImpactXParticleContainer.H:307
ImpactXParticleContainer(initialization::AmrCoreData *amr_core)
Construct a new particle container.
Definition: ImpactXParticleContainer.cpp:57
void SetParticleShape()
Definition: ImpactXParticleContainer.cpp:115
void SetCoordSystem(CoordSystem coord_system)
Definition: ImpactXParticleContainer.cpp:285
ImpactXParticleContainer * GetLostParticleContainer()
Definition: ImpactXParticleContainer.cpp:90
std::tuple< amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal > MeanAndStdPositions()
Definition: ImpactXParticleContainer.cpp:258
RefPart m_refpart
the reference particle for the beam in the particle container
Definition: ImpactXParticleContainer.H:301
std::vector< std::string > m_real_soa_names
ParticleReal component names.
Definition: ImpactXParticleContainer.H:313
std::optional< int > m_particle_shape
the particle shape
Definition: ImpactXParticleContainer.H:304
CoordSystem m_coordsystem
the current coordinate system of particles in this container
Definition: ImpactXParticleContainer.H:310
RefPart & GetRefParticle()
Definition: ImpactXParticleContainer.cpp:227
Definition: ImpactXParticleContainer.H:114
ContainerType const * pc() const
Definition: ImpactXParticleContainer.H:122
ParConstIterSoA(ContainerType &pc, int level)
Definition: ImpactXParticleContainer.cpp:49
Definition: ImpactXParticleContainer.H:96
ContainerType * pc()
Definition: ImpactXParticleContainer.H:104
ParIterSoA(ContainerType &pc, int level)
Definition: ImpactXParticleContainer.cpp:41
Definition: AmrCoreData.H:42
Definition: ImpactX.cpp:33
CoordSystem
Definition: ImpactXParticleContainer.H:34
@ s
fixed s as the independent variable
@ t
fixed t as the independent variable
Definition: ImpactXParticleContainer.H:75
static constexpr std::initializer_list< char const * > names_s
named labels for fixed s
Definition: ImpactXParticleContainer.H:82
static constexpr std::initializer_list< char const * > names_t
named labels for fixed t
Definition: ImpactXParticleContainer.H:84
@ nattribs
the number of attributes above (always last)
Definition: ImpactXParticleContainer.H:78
Definition: ImpactXParticleContainer.H:43
@ pt
energy deviation, scaled by speed of light * the magnitude of the reference momentum [unitless] (at f...
Definition: ImpactXParticleContainer.H:51
@ y
position in y [m] (at fixed s or t)
Definition: ImpactXParticleContainer.H:47
@ t
time-of-flight ct [m] (at fixed s)
Definition: ImpactXParticleContainer.H:48
@ w
particle weight, number of real particles represented by this macroparticle [unitless]
Definition: ImpactXParticleContainer.H:53
@ px
momentum in x, scaled by the magnitude of the reference momentum [unitless] (at fixed s or t)
Definition: ImpactXParticleContainer.H:49
@ nattribs
the number of attributes above (always last)
Definition: ImpactXParticleContainer.H:54
@ py
momentum in y, scaled by the magnitude of the reference momentum [unitless] (at fixed s or t)
Definition: ImpactXParticleContainer.H:50
@ x
position in x [m] (at fixed s or t)
Definition: ImpactXParticleContainer.H:46
@ qm
charge to mass ratio, in q_e/m_e [q_e/eV]
Definition: ImpactXParticleContainer.H:52
static constexpr auto names_s
named labels for fixed s
Definition: ImpactXParticleContainer.H:64
@ z
position in z [m] (at fixed t)
Definition: ImpactXParticleContainer.H:59
@ pz
momentum in z, scaled by the magnitude of the reference momentum [unitless] (at fixed t)
Definition: ImpactXParticleContainer.H:60
static constexpr auto names_t
named labels for fixed t
Definition: ImpactXParticleContainer.H:66
Definition: ReferenceParticle.H:30