7 #ifndef WARPX_ACCELERATORLATTICE_LATTICEELEMENTS_HARDEDGEDQUADRUPOLE_H_
8 #define WARPX_ACCELERATORLATTICE_LATTICEELEMENTS_HARDEDGEDQUADRUPOLE_H_
47 std::vector<amrex::ParticleReal>
h_dEdx;
48 std::vector<amrex::ParticleReal>
h_dBdx;
91 const amrex::ParticleReal x,
92 const amrex::ParticleReal y,
93 const amrex::ParticleReal z,
94 const amrex::ParticleReal zpvdt,
95 amrex::ParticleReal& Ex,
96 amrex::ParticleReal& Ey,
97 amrex::ParticleReal& Bx,
98 amrex::ParticleReal& By)
const
105 const amrex::ParticleReal dEdx = frac*
d_dEdx_arr[ielement];
106 const amrex::ParticleReal dBdx = frac*
d_dBdx_arr[ielement];
#define AMREX_FORCE_INLINE
#define AMREX_GPU_HOST_DEVICE
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE amrex::ParticleReal hard_edged_fraction(const amrex::ParticleReal z, const amrex::ParticleReal zpvdt, const amrex::ParticleReal zs, const amrex::ParticleReal ze)
Calculate the residence correction, the fraction of the time step the particle spends inside of the h...
Definition: HardEdged_K.H:25
Definition: HardEdgedQuadrupole.H:64
const amrex::ParticleReal *AMREX_RESTRICT d_dEdx_arr
Definition: HardEdgedQuadrupole.H:78
const amrex::ParticleReal *AMREX_RESTRICT d_dBdx_arr
Definition: HardEdgedQuadrupole.H:79
int nelements
Definition: HardEdgedQuadrupole.H:73
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void get_field(const int ielement, const amrex::ParticleReal x, const amrex::ParticleReal y, const amrex::ParticleReal z, const amrex::ParticleReal zpvdt, amrex::ParticleReal &Ex, amrex::ParticleReal &Ey, amrex::ParticleReal &Bx, amrex::ParticleReal &By) const
Fetch the field of the specified element at the given location.
Definition: HardEdgedQuadrupole.H:90
void InitHardEdgedQuadrupoleDevice(HardEdgedQuadrupole const &h_quad)
Initializes the data and pointer needed to reference the lattice element info.
Definition: HardEdgedQuadrupole.cpp:56
const amrex::ParticleReal *AMREX_RESTRICT d_zs_arr
Definition: HardEdgedQuadrupole.H:75
const amrex::ParticleReal *AMREX_RESTRICT d_ze_arr
Definition: HardEdgedQuadrupole.H:76
Definition: HardEdgedQuadrupole.H:26
amrex::Gpu::DeviceVector< amrex::ParticleReal > d_dEdx
Definition: HardEdgedQuadrupole.H:50
HardEdgedQuadrupoleDevice GetDeviceInstance() const
Returns the device level instance with the lattice information.
Definition: HardEdgedQuadrupole.cpp:48
void AddElement(amrex::ParmParse &pp_element, amrex::ParticleReal &z_location)
Read in an element and add it to the lists.
Definition: HardEdgedQuadrupole.cpp:21
void WriteToDevice()
Write the element information to the device.
Definition: HardEdgedQuadrupole.cpp:37
HardEdgedQuadrupole()
Definition: HardEdgedQuadrupole.cpp:15
amrex::Gpu::DeviceVector< amrex::ParticleReal > d_dBdx
Definition: HardEdgedQuadrupole.H:51
std::vector< amrex::ParticleReal > h_dBdx
Definition: HardEdgedQuadrupole.H:48
std::vector< amrex::ParticleReal > h_dEdx
Definition: HardEdgedQuadrupole.H:47
Definition: LatticeElementBase.H:18