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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
/* Copyright 2019 Andrew Myers, David Grote, Maxence Thevenet
* Weiqun Zhang
*
* This file is part of WarpX.
*
* License: BSD-3-Clause-LBNL
*/
#ifndef WARPX_RigidInjectedParticleContainer_H_
#define WARPX_RigidInjectedParticleContainer_H_
#include "PhysicalParticleContainer.H"
#include <AMReX_Vector.H>
/**
* When injecting a particle beam (typically for a plasma wakefield
* acceleration simulation), say propagating in the z direction, it can
* be necessary to make particles propagate in a straight line up to a given
* location z=z0. This is of particular importance when running in a boosted
* frame, where the beam may evolve due to its space charge fields before
* entering the plasma, causing the actual injected beam, and hence the whole
* simulation result, to depend on the Lorentz factor of the boost.
*
* This feature is implemented in RigidInjectedParticleContainer: At each
* iteration, for each particle, if z<z0 the particle moves in a straight line,
* and if z>z0 the particle evolves as a regular PhysicalParticleContainer.
*
* Note: This option is also useful to build self-consistent space charge
* fields for the particle beam.
*
* RigidInjectedParticleContainer derives from PhysicalParticleContainer.
*/
class RigidInjectedParticleContainer
: public PhysicalParticleContainer
{
public:
RigidInjectedParticleContainer (amrex::AmrCore* amr_core,
int ispecies,
const std::string& name);
virtual ~RigidInjectedParticleContainer () {}
virtual void InitData() override;
virtual void RemapParticles();
virtual void BoostandRemapParticles();
virtual void Evolve (int lev,
const amrex::MultiFab& Ex,
const amrex::MultiFab& Ey,
const amrex::MultiFab& Ez,
const amrex::MultiFab& Bx,
const amrex::MultiFab& By,
const amrex::MultiFab& Bz,
const amrex::MultiFab& Ex_avg,
const amrex::MultiFab& Ey_avg,
const amrex::MultiFab& Ez_avg,
const amrex::MultiFab& Bx_avg,
const amrex::MultiFab& By_avg,
const amrex::MultiFab& Bz_avg,
amrex::MultiFab& jx,
amrex::MultiFab& jy,
amrex::MultiFab& jz,
amrex::MultiFab* cjx,
amrex::MultiFab* cjy,
amrex::MultiFab* cjz,
amrex::MultiFab* rho,
amrex::MultiFab* crho,
const amrex::MultiFab* cEx,
const amrex::MultiFab* cEy,
const amrex::MultiFab* cEz,
const amrex::MultiFab* cBx,
const amrex::MultiFab* cBy,
const amrex::MultiFab* cBz,
amrex::Real t,
amrex::Real dt,
DtType a_dt_type=DtType::Full) override;
virtual void PushPX (WarpXParIter& pti,
amrex::FArrayBox const * exfab,
amrex::FArrayBox const * eyfab,
amrex::FArrayBox const * ezfab,
amrex::FArrayBox const * bxfab,
amrex::FArrayBox const * byfab,
amrex::FArrayBox const * bzfab,
const int ngE, const int /*e_is_nodal*/,
const long offset,
const long np_to_push,
int lev, int gather_lev,
amrex::Real dt, ScaleFields scaleFields,
DtType a_dt_type=DtType::Full) override;
virtual void PushP (int lev, amrex::Real dt,
const amrex::MultiFab& Ex,
const amrex::MultiFab& Ey,
const amrex::MultiFab& Ez,
const amrex::MultiFab& Bx,
const amrex::MultiFab& By,
const amrex::MultiFab& Bz) override;
virtual void ReadHeader (std::istream& is) override;
virtual void WriteHeader (std::ostream& os) const override;
private:
// User input quantities
amrex::Real zinject_plane = 0.;
bool projected = true; // When true, particle transverse positions are directly projected (without adjusment)
bool focused = false; // When true, particle transverse positions are adjusted to account for distance between zinject and z=0
bool rigid_advance = true; // When true, particles are advance with vzbar before injection
amrex::Real vzbeam_ave_boosted;
amrex::Vector<int> done_injecting;
amrex::Vector<amrex::Real> zinject_plane_levels;
// Temporary quantites
amrex::Real zinject_plane_lev;
amrex::Real zinject_plane_lev_previous;
bool done_injecting_lev;
};
#endif
|
