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
|
#include "CellCenterFunctor.H"
#include "Utils/Average.H"
using namespace amrex;
CellCenterFunctor::CellCenterFunctor(amrex::MultiFab const * mf_src, int lev,
bool convertRZmodes2cartesian, int ncomp)
: ComputeDiagFunctor(ncomp), m_mf_src(mf_src), m_lev(lev),
m_convertRZmodes2cartesian(convertRZmodes2cartesian)
{}
void
CellCenterFunctor::operator()(amrex::MultiFab& mf_dst, int dcomp) const
{
#ifdef WARPX_DIM_RZ
if (m_convertRZmodes2cartesian) {
// In cylindrical geometry, sum real part of all modes of m_mf_src in
// temporary multifab mf_dst_stag, and cell-center it to mf_dst.
AMREX_ALWAYS_ASSERT_WITH_MESSAGE(
nComp()==1,
"The RZ averaging over modes must write into 1 single component");
auto& warpx = WarpX::GetInstance();
MultiFab mf_dst_stag(m_mf_src->boxArray(), warpx.DistributionMap(m_lev), 1, m_mf_src->nGrowVect());
// Mode 0
MultiFab::Copy(mf_dst_stag, *m_mf_src, 0, 0, 1, m_mf_src->nGrowVect());
for (int ic=1 ; ic < m_mf_src->nComp() ; ic += 2) {
// All modes > 0
MultiFab::Add(mf_dst_stag, *m_mf_src, ic, 0, 1, m_mf_src->nGrowVect());
}
Average::CoarsenAndInterpolate( mf_dst, mf_dst_stag, dcomp, 0, nComp(), 0 );
} else {
Average::CoarsenAndInterpolate( mf_dst, *m_mf_src, dcomp, 0, nComp(), 0 );
}
#else
// In cartesian geometry, cell-center m_mf_src to mf_dst.
Average::CoarsenAndInterpolate( mf_dst, *m_mf_src, dcomp, 0, nComp(), 0 );
#endif
}
|
