aboutsummaryrefslogtreecommitdiff
path: root/Source/Particles/Gather
diff options
context:
space:
mode:
Diffstat (limited to 'Source/Particles/Gather')
-rw-r--r--Source/Particles/Gather/GetExternalFields.H10
-rw-r--r--Source/Particles/Gather/GetExternalFields.cpp19
2 files changed, 26 insertions, 3 deletions
diff --git a/Source/Particles/Gather/GetExternalFields.H b/Source/Particles/Gather/GetExternalFields.H
index 468b8d078..c09d13ab3 100644
--- a/Source/Particles/Gather/GetExternalFields.H
+++ b/Source/Particles/Gather/GetExternalFields.H
@@ -34,6 +34,7 @@ struct GetExternalField
const amrex::Real* AMREX_RESTRICT m_repeated_plasma_lens_lengths = nullptr;
const amrex::Real* AMREX_RESTRICT m_repeated_plasma_lens_strengths = nullptr;
int m_n_lenses;
+ int m_lens_is_electric;
amrex::Real m_dt;
const amrex::ParticleReal* AMREX_RESTRICT m_ux = nullptr;
const amrex::ParticleReal* AMREX_RESTRICT m_uy = nullptr;
@@ -94,8 +95,13 @@ struct GetExternalField
if (fl > fr) frac = (lens_end - zl)*dzi;
if (fr > fl) frac = (zr - lens_start)*dzi;
- field_x += x*frac*m_repeated_plasma_lens_strengths[i_lens];
- field_y += y*frac*m_repeated_plasma_lens_strengths[i_lens];
+ if (m_lens_is_electric) {
+ field_x += x*frac*m_repeated_plasma_lens_strengths[i_lens];
+ field_y += y*frac*m_repeated_plasma_lens_strengths[i_lens];
+ } else {
+ field_x += y*frac*m_repeated_plasma_lens_strengths[i_lens];
+ field_y -= x*frac*m_repeated_plasma_lens_strengths[i_lens];
+ }
}
else
diff --git a/Source/Particles/Gather/GetExternalFields.cpp b/Source/Particles/Gather/GetExternalFields.cpp
index 03c4dc2b8..d97a69985 100644
--- a/Source/Particles/Gather/GetExternalFields.cpp
+++ b/Source/Particles/Gather/GetExternalFields.cpp
@@ -32,6 +32,7 @@ GetExternalEField::GetExternalEField (const WarpXParIter& a_pti, int a_offset) n
else if (mypc.m_E_ext_particle_s=="repeated_plasma_lens")
{
m_type = RepeatedPlasmaLens;
+ m_lens_is_electric = 1;
m_dt = warpx.getdt(a_pti.GetLevel());
m_get_position = GetParticlePosition(a_pti, a_offset);
auto& attribs = a_pti.GetAttribs();
@@ -42,7 +43,7 @@ GetExternalEField::GetExternalEField (const WarpXParIter& a_pti, int a_offset) n
m_n_lenses = static_cast<int>(mypc.h_repeated_plasma_lens_starts.size());
m_repeated_plasma_lens_starts = mypc.d_repeated_plasma_lens_starts.data();
m_repeated_plasma_lens_lengths = mypc.d_repeated_plasma_lens_lengths.data();
- m_repeated_plasma_lens_strengths = mypc.d_repeated_plasma_lens_strengths.data();
+ m_repeated_plasma_lens_strengths = mypc.d_repeated_plasma_lens_strengths_E.data();
}
}
@@ -66,4 +67,20 @@ GetExternalBField::GetExternalBField (const WarpXParIter& a_pti, int a_offset) n
m_yfield_partparser = mypc.m_By_particle_parser->compile<4>();
m_zfield_partparser = mypc.m_Bz_particle_parser->compile<4>();
}
+ else if (mypc.m_B_ext_particle_s=="repeated_plasma_lens")
+ {
+ m_type = RepeatedPlasmaLens;
+ m_lens_is_electric = 0;
+ m_dt = warpx.getdt(a_pti.GetLevel());
+ m_get_position = GetParticlePosition(a_pti, a_offset);
+ auto& attribs = a_pti.GetAttribs();
+ m_ux = attribs[PIdx::ux].dataPtr() + a_offset;
+ m_uy = attribs[PIdx::uy].dataPtr() + a_offset;
+ m_uz = attribs[PIdx::uz].dataPtr() + a_offset;
+ m_repeated_plasma_lens_period = mypc.m_repeated_plasma_lens_period;
+ m_n_lenses = static_cast<int>(mypc.h_repeated_plasma_lens_starts.size());
+ m_repeated_plasma_lens_starts = mypc.d_repeated_plasma_lens_starts.data();
+ m_repeated_plasma_lens_lengths = mypc.d_repeated_plasma_lens_lengths.data();
+ m_repeated_plasma_lens_strengths = mypc.d_repeated_plasma_lens_strengths_B.data();
+ }
}