10 #ifndef IMPACTX_DISTRIBUTION_KVDIST
11 #define IMPACTX_DISTRIBUTION_KVDIST
38 amrex::ParticleReal sigx,
39 amrex::ParticleReal sigy,
40 amrex::ParticleReal sigt,
41 amrex::ParticleReal sigpx,
42 amrex::ParticleReal sigpy,
43 amrex::ParticleReal sigpt,
44 amrex::ParticleReal muxpx=0.0,
45 amrex::ParticleReal muypy=0.0,
46 amrex::ParticleReal mutpt=0.0
60 void initialize ([[maybe_unused]] amrex::ParticleReal bunch_charge, [[maybe_unused]]
RefPart const & ref)
85 amrex::ParticleReal & AMREX_RESTRICT x,
86 amrex::ParticleReal & AMREX_RESTRICT y,
87 amrex::ParticleReal & AMREX_RESTRICT t,
88 amrex::ParticleReal & AMREX_RESTRICT px,
89 amrex::ParticleReal & AMREX_RESTRICT py,
90 amrex::ParticleReal & AMREX_RESTRICT pt,
94 using namespace amrex::literals;
97 amrex::ParticleReal v,phi,
r,
beta,p,u1,u2,ln1;
98 amrex::ParticleReal root,a1,a2;
117 t = 2.0_prt*(
t-0.5_prt);
121 pt = ln1*cos(2_prt*
pi*u2);
124 amrex::ParticleReal
const c =
sqrt(3.0_prt);
#define AMREX_FORCE_INLINE
#define AMREX_GPU_HOST_DEVICE
static constexpr amrex::Real pi
constexpr std::enable_if_t< std::is_floating_point< T >::value, T > pi()
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE GpuComplex< T > log(const GpuComplex< T > &a_z) noexcept
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE GpuComplex< T > pow(const GpuComplex< T > &a_z, const T &a_y) noexcept
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE GpuComplex< T > sqrt(const GpuComplex< T > &a_z) noexcept
@ t
fixed t as the independent variable
Definition: ReferenceParticle.H:30
amrex::ParticleReal m_sigmaPy
Definition: KVdist.H:152
KVdist(amrex::ParticleReal sigx, amrex::ParticleReal sigy, amrex::ParticleReal sigt, amrex::ParticleReal sigpx, amrex::ParticleReal sigpy, amrex::ParticleReal sigpt, amrex::ParticleReal muxpx=0.0, amrex::ParticleReal muypy=0.0, amrex::ParticleReal mutpt=0.0)
Definition: KVdist.H:37
amrex::ParticleReal m_sigmaT
Definition: KVdist.H:151
amrex::ParticleReal m_mutpt
Definition: KVdist.H:153
void finalize()
Definition: KVdist.H:69
void initialize([[maybe_unused]] amrex::ParticleReal bunch_charge, [[maybe_unused]] RefPart const &ref)
Definition: KVdist.H:60
amrex::ParticleReal m_sigmaPx
related RMS sizes (length)
Definition: KVdist.H:152
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void operator()(amrex::ParticleReal &AMREX_RESTRICT x, amrex::ParticleReal &AMREX_RESTRICT y, amrex::ParticleReal &AMREX_RESTRICT t, amrex::ParticleReal &AMREX_RESTRICT px, amrex::ParticleReal &AMREX_RESTRICT py, amrex::ParticleReal &AMREX_RESTRICT pt, amrex::RandomEngine const &engine) const
Definition: KVdist.H:84
amrex::ParticleReal m_sigmaPt
Definition: KVdist.H:152
amrex::ParticleReal m_muxpx
RMS momentum.
Definition: KVdist.H:153
amrex::ParticleReal m_muypy
Definition: KVdist.H:153
amrex::ParticleReal m_sigmaX
Definition: KVdist.H:151
amrex::ParticleReal m_sigmaY
Definition: KVdist.H:151