#include <SoftQuad.H>

Inheritance diagram for impactx::elements::SoftQuadrupole:

Public Types
using	PType = ImpactXParticleContainer::ParticleType

Public Member Functions
	SoftQuadrupole (amrex::ParticleReal ds, amrex::ParticleReal gscale, std::vector< amrex::ParticleReal > cos_coef, std::vector< amrex::ParticleReal > sin_coef, amrex::ParticleReal dx=0, amrex::ParticleReal dy=0, amrex::ParticleReal rotation_degree=0, amrex::ParticleReal aperture_x=0, amrex::ParticleReal aperture_y=0, int mapsteps=1, int nslice=1, std::optional< std::string > name=std::nullopt)

void	compute_constants (RefPart const &refpart)

template<typename T_Real = amrex::ParticleReal, typename T_IdCpu = uint64_t>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void	operator() (T_Real &AMREX_RESTRICT x, T_Real &AMREX_RESTRICT y, T_Real &AMREX_RESTRICT t, T_Real &AMREX_RESTRICT px, T_Real &AMREX_RESTRICT py, T_Real &AMREX_RESTRICT pt, T_IdCpu &AMREX_RESTRICT idcpu, RefPart const &AMREX_RESTRICT refpart) const

AMREX_GPU_HOST AMREX_FORCE_INLINE void	operator() (RefPart &AMREX_RESTRICT refpart) const

AMREX_GPU_HOST AMREX_FORCE_INLINE Map6x6	transport_map (RefPart const &AMREX_RESTRICT refpart) const

std::tuple< amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal > AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE	Quad_Bfield (amrex::ParticleReal const zeval) const

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void	map1 (amrex::ParticleReal const tau, RefPart &refpart, amrex::ParticleReal &zeval) const

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void	map2 (amrex::ParticleReal const tau, RefPart &refpart, amrex::ParticleReal &zeval) const

void	finalize ()

Public Member Functions inherited from impactx::elements::mixin::Named
AMREX_GPU_HOST void	set_name (std::string const &new_name)

AMREX_GPU_HOST	Named (std::optional< std::string > name)

AMREX_GPU_HOST_DEVICE	~Named ()

AMREX_GPU_HOST	Named (Named const &other)

AMREX_GPU_HOST Named &	operator= (Named const &other)

AMREX_GPU_HOST_DEVICE	Named (Named &&other) noexcept

AMREX_GPU_HOST_DEVICE Named &	operator= (Named &&other) noexcept

AMREX_FORCE_INLINE std::string	name () const

AMREX_FORCE_INLINE bool	has_name () const

Public Member Functions inherited from impactx::elements::mixin::BeamOptic< SoftQuadrupole >
void	operator() (ImpactXParticleContainer &pc, int step, int period)

void	operator() (ImpactXParticleContainer::iterator &pti, RefPart &AMREX_RESTRICT ref_part)

Public Member Functions inherited from impactx::elements::mixin::LinearTransport< SoftQuadrupole >
	LinearTransport ()=default

	LinearTransport (LinearTransport const &)=default

	LinearTransport (LinearTransport &&)=default

LinearTransport &	operator= (LinearTransport const &)=default

LinearTransport &	operator= (LinearTransport &&rhs)=default

	~LinearTransport ()=default

AMREX_GPU_HOST AMREX_FORCE_INLINE void	operator() (Map6x6 &AMREX_RESTRICT cm, RefPart const &AMREX_RESTRICT ref) const

Public Member Functions inherited from impactx::elements::mixin::Thick
	Thick (amrex::ParticleReal ds, int nslice)

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE int	nslice () const

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::ParticleReal	ds () const

Public Member Functions inherited from impactx::elements::mixin::Alignment
	Alignment (amrex::ParticleReal dx, amrex::ParticleReal dy, amrex::ParticleReal rotation_degree)

	Alignment ()=default

	Alignment (Alignment const &)=default

Alignment &	operator= (Alignment const &)=default

	Alignment (Alignment &&)=default

Alignment &	operator= (Alignment &&rhs)=default

	~Alignment ()=default

void	compute_constants (RefPart const &refpart)

template<typename T_Real>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void	shift_in (T_Real &AMREX_RESTRICT x, T_Real &AMREX_RESTRICT y, T_Real &AMREX_RESTRICT px, T_Real &AMREX_RESTRICT py) const

template<typename T_Real>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void	shift_out (T_Real &AMREX_RESTRICT x, T_Real &AMREX_RESTRICT y, T_Real &AMREX_RESTRICT px, T_Real &AMREX_RESTRICT py) const

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::ParticleReal	dx () const

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::ParticleReal	dy () const

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::ParticleReal	rotation () const

Public Member Functions inherited from impactx::elements::mixin::PipeAperture
	PipeAperture (amrex::ParticleReal aperture_x, amrex::ParticleReal aperture_y)

	PipeAperture ()=default

	PipeAperture (PipeAperture const &)=default

PipeAperture &	operator= (PipeAperture const &)=default

	PipeAperture (PipeAperture &&)=default

PipeAperture &	operator= (PipeAperture &&rhs)=default

	~PipeAperture ()=default

template<typename T_Real = amrex::ParticleReal, typename T_IdCpu = uint64_t>
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void	apply_aperture (T_Real &AMREX_RESTRICT x, T_Real &AMREX_RESTRICT y, T_IdCpu &AMREX_RESTRICT idcpu) const

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::ParticleReal	aperture_x () const

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::ParticleReal	aperture_y () const

Public Attributes
amrex::ParticleReal	m_gscale

int	m_mapsteps
	scaling factor for quad field gradient

int	m_id
	number of map integration steps per slice

int	m_ncoef = 0
	unique soft quad id used for data lookup map

amrex::ParticleReal *	m_cos_h_data = nullptr
	number of Fourier coefficients

amrex::ParticleReal *	m_sin_h_data = nullptr
	non-owning pointer to host cosine coefficients

amrex::ParticleReal *	m_cos_d_data = nullptr
	non-owning pointer to host sine coefficients

amrex::ParticleReal *	m_sin_d_data = nullptr
	non-owning pointer to device cosine coefficients

Public Attributes inherited from impactx::elements::mixin::Thick
amrex::ParticleReal	m_ds

int	m_nslice
	segment length in m

Public Attributes inherited from impactx::elements::mixin::Alignment
amrex::ParticleReal	m_dx = 0

amrex::ParticleReal	m_dy = 0
	horizontal translation error [m]

amrex::ParticleReal	m_rotation = 0
	vertical translation error [m]

Public Attributes inherited from impactx::elements::mixin::PipeAperture
amrex::ParticleReal	m_inv_aperture_x = 0

amrex::ParticleReal	m_inv_aperture_y = 0
	inverse of the horizontal aperture size [1/m]

Static Public Attributes
static constexpr auto	type = "SoftQuadrupole"

Static Public Attributes inherited from impactx::elements::mixin::Alignment
static constexpr amrex::ParticleReal	degree2rad = ablastr::constant::math::pi / 180.0

Static Public Attributes inherited from amrex::simd::Vectorized< amrex::simd::native_simd_size_particlereal >
static constexpr int	simd_width

Member Typedef Documentation

◆ PType

using impactx::elements::SoftQuadrupole::PType = ImpactXParticleContainer::ParticleType

Constructor & Destructor Documentation

◆ SoftQuadrupole()

impactx::elements::SoftQuadrupole::SoftQuadrupole	(	amrex::ParticleReal	ds,
		amrex::ParticleReal	gscale,
		std::vector< amrex::ParticleReal >	cos_coef,
		std::vector< amrex::ParticleReal >	sin_coef,
		amrex::ParticleReal	dx = 0,
		amrex::ParticleReal	dy = 0,
		amrex::ParticleReal	rotation_degree = 0,
		amrex::ParticleReal	aperture_x = 0,
		amrex::ParticleReal	aperture_y = 0,
		int	mapsteps = 1,
		int	nslice = 1,
		std::optional< std::string >	name = std::nullopt )

inline

A soft-edge quadrupole

Parameters

ds	Segment length in m
gscale	Scaling factor for on-axis field gradient Bz in 1/m^2
cos_coef	cosine coefficients in Fourier expansion of on-axis magnetic field Bz
sin_coef	sine coefficients in Fourier expansion of on-axis magnetic field Bz
dx	horizontal translation error in m
dy	vertical translation error in m
rotation_degree	rotation error in the transverse plane [degrees]
aperture_x	horizontal half-aperture in m
aperture_y	vertical half-aperture in m
mapsteps	number of integration steps per slice used for map and reference particle push in applied fields
nslice	number of slices used for the application of space charge
name	a user defined and not necessarily unique name of the element

Member Function Documentation

◆ compute_constants()

void impactx::elements::SoftQuadrupole::compute_constants ( RefPart const & refpart )

inline

Compute and cache the constants for the push.

In particular, used to pre-compute and cache variables that are independent of the individually tracked particle.

Parameters

refpart reference particle

◆ finalize()

void impactx::elements::SoftQuadrupole::finalize ( )

inline

Close and deallocate all data and handles.

◆ map1()

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void impactx::elements::SoftQuadrupole::map1	(	amrex::ParticleReal const	tau,
		RefPart &	refpart,
		amrex::ParticleReal &	zeval ) const

inline

This pushes the reference particle and the linear map matrix elements for a quadrupole through the symplectic map associated with H_1 in the Hamiltonian splitting H = H_1 + H_2.

Parameters

	tau	Map step size in m
[in,out]	refpart	reference particle
[in,out]	zeval	Longitudinal on-axis location in m

◆ map2()

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void impactx::elements::SoftQuadrupole::map2	(	amrex::ParticleReal const	tau,
		RefPart &	refpart,
		amrex::ParticleReal &	zeval ) const

inline

This pushes the reference particle and the linear map matrix elements for a quadrupole through the symplectic map associated with H_2 in the Hamiltonian splitting H = H_1 + H_2.

Parameters

	tau	Map step size in m
[in,out]	refpart	reference particle
[in,out]	zeval	Longitudinal on-axis location in m

◆ operator()() [1/2]

AMREX_GPU_HOST AMREX_FORCE_INLINE void impactx::elements::SoftQuadrupole::operator() ( RefPart &AMREX_RESTRICT refpart ) const

inline

This pushes the reference particle.

Parameters

[in,out] refpart reference particle

◆ operator()() [2/2]

template<typename T_Real = amrex::ParticleReal, typename T_IdCpu = uint64_t>

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void impactx::elements::SoftQuadrupole::operator()	(	T_Real &AMREX_RESTRICT	x,
		T_Real &AMREX_RESTRICT	y,
		T_Real &AMREX_RESTRICT	t,
		T_Real &AMREX_RESTRICT	px,
		T_Real &AMREX_RESTRICT	py,
		T_Real &AMREX_RESTRICT	pt,
		T_IdCpu &AMREX_RESTRICT	idcpu,
		RefPart const &AMREX_RESTRICT	refpart ) const

inline

This is a soft-edge quadrupole functor, so that a variable of this type can be used like a soft-edge quadrupole function.

The

See also: compute_constants method must be called before pushing particles through this operator.

Parameters

x	particle position in x
y	particle position in y
t	particle position in t
px	particle momentum in x
py	particle momentum in y
pt	particle momentum in t
idcpu	particle global index
refpart	reference particle

◆ Quad_Bfield()

std::tuple< amrex::ParticleReal, amrex::ParticleReal, amrex::ParticleReal > AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE impactx::elements::SoftQuadrupole::Quad_Bfield ( amrex::ParticleReal const zeval ) const

inline

This evaluates the on-axis magnetic field Bz at a fixed location z, together with certain required integrals and derivatives. The field returned is normalized to a peak value of 1.

Parameters

zeval Longitudinal on-axis location in m

◆ transport_map()

AMREX_GPU_HOST AMREX_FORCE_INLINE Map6x6 impactx::elements::SoftQuadrupole::transport_map ( RefPart const &AMREX_RESTRICT refpart ) const

inline

This function returns the linear transport map.

Parameters

[in] refpart reference particle

Returns: 6x6 transport matrix

Member Data Documentation

◆ m_cos_d_data

amrex::ParticleReal* impactx::elements::SoftQuadrupole::m_cos_d_data = nullptr

non-owning pointer to host sine coefficients

◆ m_cos_h_data

amrex::ParticleReal* impactx::elements::SoftQuadrupole::m_cos_h_data = nullptr

number of Fourier coefficients

◆ m_gscale

amrex::ParticleReal impactx::elements::SoftQuadrupole::m_gscale

◆ m_id

int impactx::elements::SoftQuadrupole::m_id

number of map integration steps per slice

◆ m_mapsteps

int impactx::elements::SoftQuadrupole::m_mapsteps

scaling factor for quad field gradient

◆ m_ncoef

int impactx::elements::SoftQuadrupole::m_ncoef = 0

unique soft quad id used for data lookup map

◆ m_sin_d_data

amrex::ParticleReal* impactx::elements::SoftQuadrupole::m_sin_d_data = nullptr

non-owning pointer to device cosine coefficients

◆ m_sin_h_data

amrex::ParticleReal* impactx::elements::SoftQuadrupole::m_sin_h_data = nullptr

non-owning pointer to host cosine coefficients

◆ type

auto impactx::elements::SoftQuadrupole::type = "SoftQuadrupole"

staticconstexpr

The documentation for this struct was generated from the following file:

/home/docs/checkouts/readthedocs.org/user_builds/impactx/checkouts/1182/src/elements/SoftQuad.H

Public Types

Public Member Functions

Public Attributes

Static Public Attributes

Member Typedef Documentation

◆ PType

Constructor & Destructor Documentation

◆ SoftQuadrupole()

Member Function Documentation

◆ compute_constants()

◆ finalize()

◆ map1()

◆ map2()

◆ operator()() [1/2]

◆ operator()() [2/2]

◆ Quad_Bfield()

◆ transport_map()

Member Data Documentation

◆ m_cos_d_data

◆ m_cos_h_data

◆ m_gscale

◆ m_id

◆ m_mapsteps

◆ m_ncoef

◆ m_sin_d_data

◆ m_sin_h_data

◆ type