ImpactX
ImpactXParticleContainer.H
Go to the documentation of this file.
1 /* Copyright 2022-2023 The Regents of the University of California, through Lawrence
2  * Berkeley National Laboratory (subject to receipt of any required
3  * approvals from the U.S. Dept. of Energy). All rights reserved.
4  *
5  * This file is part of ImpactX.
6  *
7  * Authors: Axel Huebl
8  * License: BSD-3-Clause-LBNL
9  */
10 #ifndef IMPACTX_PARTICLE_CONTAINER_H
11 #define IMPACTX_PARTICLE_CONTAINER_H
12 
13 #include "ReferenceParticle.H"
15 
16 #include <AMReX_AmrCoreFwd.H>
17 #include <AMReX_BaseFwd.H>
18 #include <AMReX_MultiFab.H>
19 #include <AMReX_ParIter.H>
20 #include <AMReX_Particles.H>
21 #include <AMReX_ParticleTile.H>
22 
23 #include <AMReX_IntVect.H>
24 #include <AMReX_Vector.H>
25 
26 #include <optional>
27 #include <tuple>
28 #include <unordered_map>
29 
30 
31 namespace impactx
32 {
33  enum class CoordSystem
34  {
35  s,
36  t
37  };
38 
42  struct RealSoA
43  {
44  enum
45  {
46  x,
47  y,
48  t,
49  px,
50  py,
51  pt,
52  qm,
53  w,
54  nattribs
55  };
56 
57  // at fixed t, the third component represents the position z, the 6th component represents the momentum in z
58  enum {
59  z = t,
60  pz = pt
61  };
62 
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" };
67  static_assert(names_s.size() == nattribs);
68  static_assert(names_t.size() == nattribs);
69  };
70 
74  struct IntSoA
75  {
76  enum
77  {
78  nattribs
79  };
80 
82  static constexpr std::initializer_list<char const *> names_s = {};
84  static constexpr std::initializer_list<char const *> names_t = {};
85  static_assert(names_s.size() == nattribs);
86  static_assert(names_t.size() == nattribs);
87  };
88 
94  class ParIterSoA
95  : public amrex::ParIterSoA<RealSoA::nattribs, IntSoA::nattribs>
96  {
97  public:
99 
100  ParIterSoA (ContainerType& pc, int level);
101 
102  ParIterSoA (ContainerType& pc, int level, amrex::MFItInfo& info);
103 
104  ContainerType * pc () { return m_pc; }
105  };
106 
113  : public amrex::ParConstIterSoA<RealSoA::nattribs, IntSoA::nattribs>
114  {
115  public:
117 
118  ParConstIterSoA (ContainerType& pc, int level);
119 
120  ParConstIterSoA (ContainerType& pc, int level, amrex::MFItInfo& info);
121 
122  ContainerType const * pc () const { return m_pc; }
123  };
124 
130  : public amrex::ParticleContainerPureSoA<RealSoA::nattribs, IntSoA::nattribs>
131  {
132  public:
135 
138 
141 
143  virtual ~ImpactXParticleContainer() = default;
144 
150  void AddRealComp (std::string const & name, bool communicate=true);
151 
157  void AddIntComp (std::string const & name, bool communicate=true);
158 
175  void
176  AddNParticles (
183  amrex::ParticleReal qm,
184  amrex::ParticleReal bchchg
185  );
186 
191  void
193 
196 
201  void
202  SetRefParticle (RefPart const & refpart);
203 
208  RefPart &
209  GetRefParticle ();
210 
215  RefPart const &
216  GetRefParticle () const;
217 
221  void SetRefParticleEdge ();
222 
225  int
226  GetParticleShape () const { return m_particle_shape.value(); }
227 
234  void SetParticleShape ();
235 
244  void SetParticleShape (int order);
245 
250  std::tuple<
251  amrex::ParticleReal, amrex::ParticleReal,
252  amrex::ParticleReal, amrex::ParticleReal,
253  amrex::ParticleReal, amrex::ParticleReal>
255 
260  std::tuple<
261  amrex::ParticleReal, amrex::ParticleReal,
262  amrex::ParticleReal, amrex::ParticleReal,
263  amrex::ParticleReal, amrex::ParticleReal>
265 
275  void
276  DepositCharge (std::unordered_map<int, amrex::MultiFab> & rho,
277  amrex::Vector<amrex::IntVect> const & ref_ratio);
278 
280  std::vector<std::string>
281  RealSoA_names () const;
282 
284  std::vector<std::string>
285  intSoA_names () const;
286 
289  GetCoordSystem () const;
290 
295  void
296  SetCoordSystem (CoordSystem coord_system);
297 
298  private:
299 
302 
304  std::optional<int> m_particle_shape;
305 
308 
311 
313  std::vector<std::string> m_real_soa_names;
314 
316  std::vector<std::string> m_int_soa_names;
317 
318  }; // ImpactXParticleContainer
319 
320 } // namespace impactx
321 
322 #endif // IMPACTX_PARTICLE_CONTAINER_H
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