blob: 9bf2c798188cafcb93bf4eb3ed95bf4d77c6404b (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
#include "Particles/Gather/GetExternalFields.H"
#include "Particles/MultiParticleContainer.H"
#include "Particles/WarpXParticleContainer.H"
#include "Utils/TextMsg.H"
#include "WarpX.H"
#include <AMReX_Vector.H>
#include <string>
using namespace amrex::literals;
GetExternalEBField::GetExternalEBField (const WarpXParIter& a_pti, int a_offset) noexcept
{
auto& warpx = WarpX::GetInstance();
auto& mypc = warpx.GetPartContainer();
m_gamma_boost = WarpX::gamma_boost;
m_uz_boost = std::sqrt(WarpX::gamma_boost*WarpX::gamma_boost - 1._prt)*PhysConst::c;
m_Etype = Unknown;
m_Btype = Unknown;
if (mypc.m_E_ext_particle_s == "none") m_Etype = None;
if (mypc.m_B_ext_particle_s == "none") m_Btype = None;
if (mypc.m_E_ext_particle_s == "constant")
{
m_Etype = Constant;
m_Efield_value[0] = mypc.m_E_external_particle[0];
m_Efield_value[1] = mypc.m_E_external_particle[1];
m_Efield_value[2] = mypc.m_E_external_particle[2];
}
if (mypc.m_B_ext_particle_s == "constant")
{
m_Btype = Constant;
m_Bfield_value[0] = mypc.m_B_external_particle[0];
m_Bfield_value[1] = mypc.m_B_external_particle[1];
m_Bfield_value[2] = mypc.m_B_external_particle[2];
}
if (mypc.m_E_ext_particle_s == "parse_e_ext_particle_function" ||
mypc.m_B_ext_particle_s == "parse_b_ext_particle_function" ||
mypc.m_E_ext_particle_s == "repeated_plasma_lens" ||
mypc.m_B_ext_particle_s == "repeated_plasma_lens")
{
m_time = warpx.gett_new(a_pti.GetLevel());
m_get_position = GetParticlePosition(a_pti, a_offset);
}
if (mypc.m_E_ext_particle_s == "parse_e_ext_particle_function")
{
m_Etype = ExternalFieldInitType::Parser;
constexpr auto num_arguments = 4; //x,y,z,t
m_Exfield_partparser = mypc.m_Ex_particle_parser->compile<num_arguments>();
m_Eyfield_partparser = mypc.m_Ey_particle_parser->compile<num_arguments>();
m_Ezfield_partparser = mypc.m_Ez_particle_parser->compile<num_arguments>();
}
if (mypc.m_B_ext_particle_s == "parse_b_ext_particle_function")
{
m_Btype = ExternalFieldInitType::Parser;
constexpr auto num_arguments = 4; //x,y,z,t
m_Bxfield_partparser = mypc.m_Bx_particle_parser->compile<num_arguments>();
m_Byfield_partparser = mypc.m_By_particle_parser->compile<num_arguments>();
m_Bzfield_partparser = mypc.m_Bz_particle_parser->compile<num_arguments>();
}
if (mypc.m_E_ext_particle_s == "repeated_plasma_lens" ||
mypc.m_B_ext_particle_s == "repeated_plasma_lens")
{
if (mypc.m_E_ext_particle_s == "repeated_plasma_lens") m_Etype = RepeatedPlasmaLens;
if (mypc.m_B_ext_particle_s == "repeated_plasma_lens") m_Btype = RepeatedPlasmaLens;
m_dt = warpx.getdt(a_pti.GetLevel());
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_E = mypc.d_repeated_plasma_lens_strengths_E.data();
m_repeated_plasma_lens_strengths_B = mypc.d_repeated_plasma_lens_strengths_B.data();
}
WARPX_ALWAYS_ASSERT_WITH_MESSAGE(m_Etype != Unknown, "Unknown E_ext_particle_init_style");
WARPX_ALWAYS_ASSERT_WITH_MESSAGE(m_Btype != Unknown, "Unknown B_ext_particle_init_style");
}
|
