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
|
#include <limits>
#include <ParticleContainer.H>
#include <ParticleIterator.H>
#include <WarpX_f.H>
#include <WarpX.H>
MultiParticleContainer::MultiParticleContainer (AmrCore* amr_core)
{
ReadParameters();
int n = WarpX::use_laser ? nspecies+1 : nspecies;
allcontainers.resize(n);
for (int i = 0; i < nspecies; ++i) {
allcontainers[i].reset(new PhysicalParticleContainer(amr_core, i));
}
if (WarpX::use_laser) {
allcontainers[n-1].reset(new LaserContainer(amr_core,n-1));
}
}
void
MultiParticleContainer::ReadParameters ()
{
static bool initialized = false;
if (!initialized)
{
ParmParse pp("particles");
pp.query("nspecies", nspecies);
BL_ASSERT(nspecies >= 1);
initialized = true;
}
}
void
MultiParticleContainer::AllocData ()
{
for (auto& pc : allcontainers) {
pc->AllocData();
}
}
void
MultiParticleContainer::InitData ()
{
for (auto& pc : allcontainers) {
pc->InitData();
}
}
void
MultiParticleContainer::Evolve (int lev,
const MultiFab& Ex, const MultiFab& Ey, const MultiFab& Ez,
const MultiFab& Bx, const MultiFab& By, const MultiFab& Bz,
MultiFab& jx, MultiFab& jy, MultiFab& jz, Real dt)
{
jx.setVal(0.0);
jy.setVal(0.0);
jz.setVal(0.0);
for (auto& pc : allcontainers) {
pc->Evolve(lev, Ex, Ey, Ez, Bx, By, Bz, jx, jy, jz, dt);
}
const Geometry& gm = allcontainers[0]->GDB().Geom(lev);
jx.SumBoundary(gm.periodicity());
jy.SumBoundary(gm.periodicity());
jz.SumBoundary(gm.periodicity());
}
std::unique_ptr<MultiFab>
MultiParticleContainer::GetChargeDensity (int lev, bool local)
{
std::unique_ptr<MultiFab> rho = allcontainers[0]->GetChargeDensity(lev, true);
for (unsigned i = 1, n = allcontainers.size(); i < n; ++i) {
std::unique_ptr<MultiFab> rhoi = allcontainers[i]->GetChargeDensity(lev, true);
MultiFab::Add(*rho, *rhoi, 0, 0, 1, rho->nGrow());
}
if (!local) {
const Geometry& gm = allcontainers[0]->GDB().Geom(lev);
rho->SumBoundary(gm.periodicity());
}
return rho;
}
void
MultiParticleContainer::Redistribute (bool where_called)
{
for (auto& pc : allcontainers) {
pc->Redistribute(where_called,true);
}
}
|
