10 #ifndef IMPACTX_DISTRIBUTION_KVDIST
11 #define IMPACTX_DISTRIBUTION_KVDIST
38 amrex::ParticleReal lambdax,
39 amrex::ParticleReal lambday,
40 amrex::ParticleReal lambdat,
41 amrex::ParticleReal lambdapx,
42 amrex::ParticleReal lambdapy,
43 amrex::ParticleReal lambdapt,
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_v< T >, 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_lambdaPt
Definition: KVdist.H:152
amrex::ParticleReal m_lambdaY
Definition: KVdist.H:151
amrex::ParticleReal m_lambdaT
Definition: KVdist.H:151
amrex::ParticleReal m_mutpt
Definition: KVdist.H:153
amrex::ParticleReal m_lambdaX
Definition: KVdist.H:151
void finalize()
Definition: KVdist.H:69
void initialize([[maybe_unused]] amrex::ParticleReal bunch_charge, [[maybe_unused]] RefPart const &ref)
Definition: KVdist.H:60
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_muxpx
related momentum axis intercepts of the phase space ellipse
Definition: KVdist.H:153
amrex::ParticleReal m_lambdaPx
related position axis intercepts (length) of the phase space ellipse
Definition: KVdist.H:152
amrex::ParticleReal m_muypy
Definition: KVdist.H:153
amrex::ParticleReal m_lambdaPy
Definition: KVdist.H:152
KVdist(amrex::ParticleReal lambdax, amrex::ParticleReal lambday, amrex::ParticleReal lambdat, amrex::ParticleReal lambdapx, amrex::ParticleReal lambdapy, amrex::ParticleReal lambdapt, amrex::ParticleReal muxpx=0.0, amrex::ParticleReal muypy=0.0, amrex::ParticleReal mutpt=0.0)
Definition: KVdist.H:37