1 #ifndef AMREX_MultiFabUtil_H_
2 #define AMREX_MultiFabUtil_H_
3 #include <AMReX_Config.H>
19 const MultiFab& nd,
int scomp,
20 int ncomp,
int ngrow = 0);
29 const Vector<const MultiFab*>& edge,
33 const Vector<const MultiFab*>& fc,
36 template <
typename CMF,
typename FMF,
37 std::enable_if_t<IsFabArray_v<CMF> && IsFabArray_v<FMF>,
int> = 0>
39 const Array<const FMF*,AMREX_SPACEDIM>& fc,
43 const Vector<const MultiFab*>& fc,
44 const Geometry& geom);
47 const Array<const MultiFab*,AMREX_SPACEDIM>& fc,
48 const Geometry& geom);
54 bool use_harmonic_averaging =
false);
60 bool use_harmonic_averaging =
false);
63 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int> = 0>
65 const Vector<MF*>&
crse,
69 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int> = 0>
71 const Vector<MF*>&
crse,
75 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int> = 0>
77 const Array<MF*,AMREX_SPACEDIM>&
crse,
81 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int> = 0>
83 const Array<MF*,AMREX_SPACEDIM>&
crse,
92 template <
typename FAB>
94 const IntVect& ratio,
int ngcrse=0);
97 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int> = 0>
99 const Array<MF*,AMREX_SPACEDIM>&
crse,
100 const IntVect& ratio,
const Geometry& crse_geom);
102 template <
typename FAB>
104 const IntVect& ratio,
const Geometry& crse_geom);
108 const Vector<MultiFab*>&
crse,
112 const Array<MultiFab*,AMREX_SPACEDIM>&
crse,
119 const IntVect& ratio,
int ngcrse=0);
122 template <
typename FAB>
124 FabArray<FAB>& S_crse,
127 bool mfiter_is_definitely_safe=
false);
135 void average_down (
const MultiFab& S_fine, MultiFab& S_crse,
136 const Geometry& fgeom,
const Geometry& cgeom,
137 int scomp,
int ncomp,
const IntVect& ratio);
138 void average_down (
const MultiFab& S_fine, MultiFab& S_crse,
139 const Geometry& fgeom,
const Geometry& cgeom,
140 int scomp,
int ncomp,
int rr);
145 template<
typename FAB>
146 void average_down (
const FabArray<FAB>& S_fine, FabArray<FAB>& S_crse,
147 int scomp,
int ncomp,
const IntVect& ratio);
148 template<
typename FAB>
149 void average_down (
const FabArray<FAB>& S_fine, FabArray<FAB>& S_crse,
150 int scomp,
int ncomp,
int rr);
155 int scomp,
int ncomp,
157 const Geometry& cgeom,
const Geometry& fgeom);
164 void writeFabs (
const MultiFab& mf,
const std::string& name);
165 void writeFabs (
const MultiFab& mf,
int comp,
int ncomp,
const std::string& name);
171 const Geometry& geom,
int start_comp,
int ncomp,
172 bool interpolate=
false);
181 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int> FOO = 0>
190 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int> FOO = 0>
196 template <
typename FAB>
197 iMultiFab
makeFineMask (
const FabArray<FAB>& cmf,
const BoxArray& fba,
const IntVect& ratio,
198 int crse_value = 0,
int fine_value = 1);
199 iMultiFab
makeFineMask (
const BoxArray& cba,
const DistributionMapping& cdm,
200 const BoxArray& fba,
const IntVect& ratio,
201 int crse_value = 0,
int fine_value = 1);
202 template <
typename FAB>
203 iMultiFab
makeFineMask (
const FabArray<FAB>& cmf,
const BoxArray& fba,
const IntVect& ratio,
204 Periodicity
const& period,
int crse_value,
int fine_value);
205 iMultiFab
makeFineMask (
const BoxArray& cba,
const DistributionMapping& cdm,
206 const IntVect& cnghost,
const BoxArray& fba,
const IntVect& ratio,
207 Periodicity
const& period,
int crse_value,
int fine_value);
208 template <
typename FAB>
209 iMultiFab
makeFineMask (
const FabArray<FAB>& cmf,
const FabArray<FAB>& fmf,
211 Periodicity
const& period,
int crse_value,
int fine_value);
212 template <
typename FAB>
213 iMultiFab
makeFineMask (
const FabArray<FAB>& cmf,
const FabArray<FAB>& fmf,
215 Periodicity
const& period,
int crse_value,
int fine_value,
216 LayoutData<int>& has_cf);
218 MultiFab
makeFineMask (
const BoxArray& cba,
const DistributionMapping& cdm,
219 const BoxArray& fba,
const IntVect& ratio,
220 Real crse_value, Real fine_value);
223 void computeDivergence (MultiFab& divu,
const Array<MultiFab const*,AMREX_SPACEDIM>& umac,
224 const Geometry& geom);
227 void computeGradient (MultiFab& grad,
const Array<MultiFab const*,AMREX_SPACEDIM>& umac,
228 const Geometry& geom);
237 Periodicity
const& period);
240 template <
typename T,
typename U>
243 T mf_out(mf_in.boxArray(), mf_in.DistributionMap(), mf_in.nComp(), mf_in.nGrowVect());
246 #pragma omp parallel if (Gpu::notInLaunchRegion())
250 const Long n = mfi.fabbox().numPts() * mf_in.nComp();
251 auto *
pdst = mf_out[mfi].dataPtr();
252 auto const* psrc = mf_in [mfi].dataPtr();
255 pdst[i] =
static_cast<typename U::value_type
>(psrc[i]);
321 template <
typename Op,
typename T,
typename FAB,
typename F,
322 std::enable_if_t<IsBaseFab<FAB>::value
323 #ifndef AMREX_USE_CUDA
324 && IsCallableR<T,F,int,int,int,int>::value
328 ReduceToPlane (
int direction,
Box const& domain, FabArray<FAB>
const& mf, F
const&
f);
345 Gpu::HostVector<Real>
sumToLine (MultiFab
const& mf,
int icomp,
int ncomp,
346 Box const& domain,
int direction,
bool local =
false);
356 Vector<Geometry>
const& geom,
357 Vector<IntVect>
const& ratio,
387 void FillRandom (MultiFab& mf,
int scomp,
int ncomp);
400 void FillRandomNormal (MultiFab& mf,
int scomp,
int ncomp, Real mean, Real stddev);
413 [[nodiscard]] Vector<MultiFab>
convexify (Vector<MultiFab const*>
const& mf,
414 Vector<IntVect>
const& refinement_ratio);
419 template <
typename FAB>
422 int crse_value,
int fine_value)
428 template <
typename FAB>
431 Periodicity const& period,
int crse_value,
int fine_value)
434 fba, ratio, period, crse_value, fine_value);
437 template <
typename FAB>
441 Periodicity const& period,
int crse_value,
int fine_value)
444 mask.setVal(crse_value);
447 1, 0,
MFInfo().SetAlloc(
false));
449 mask.setVal(fine_value, cpc, 0, 1);
454 template <
typename FAB>
458 Periodicity const& period,
int crse_value,
int fine_value,
462 mask.setVal(crse_value);
465 1, 0,
MFInfo().SetAlloc(
false));
467 mask.setVal(fine_value, cpc, 0, 1);
469 has_cf =
mask.RecvLayoutMask(cpc);
476 template <
typename FAB>
478 const IntVect& ratio,
int ngcrse,
bool mfiter_is_definitely_safe)
485 using value_type =
typename FAB::value_type;
490 #pragma omp parallel if (Gpu::notInLaunchRegion())
494 const Box& bx = mfi.growntilebox(ngcrse);
509 crse.ParallelCopy(ctmp,0,0,ncomp,ngcrse,ngcrse);
518 template<
typename FAB>
524 template<
typename FAB>
526 int scomp,
int ncomp,
const IntVect& ratio)
533 using value_type =
typename FAB::value_type;
546 auto const& crsema = S_crse.
arrays();
548 if (is_cell_centered) {
552 amrex_avgdown(i,j,k,n,crsema[box_no],finema[box_no],scomp,scomp,ratio);
568 #pragma omp parallel if (Gpu::notInLaunchRegion())
573 const Box& bx = mfi.tilebox();
577 if (is_cell_centered) {
597 auto const& crsema = crse_S_fine.
arrays();
599 if (is_cell_centered) {
603 amrex_avgdown(i,j,k,n,crsema[box_no],finema[box_no],0,scomp,ratio);
619 #pragma omp parallel if (Gpu::notInLaunchRegion())
624 const Box& bx = mfi.tilebox();
632 if (is_cell_centered) {
661 template <
typename F>
666 int numcomp,
IntVect nghost,
bool local)
675 auto const& xma =
x.const_arrays();
681 auto const& xfab = xma[box_no];
682 auto const& yfab = yma[box_no];
683 for (
int n = 0; n < numcomp; ++n) {
684 t +=
f(xfab(i,j,k,xcomp+n) , yfab(i,j,k,ycomp+n));
692 #pragma omp parallel if (!system::regtest_reduction) reduction(+:sm)
696 Box const& bx = mfi.growntilebox(nghost);
701 sm +=
f(xfab(i,j,k,xcomp+n) , yfab(i,j,k,ycomp+n));
721 template <
typename MMF,
typename Pred,
typename F>
728 int numcomp,
IntVect nghost,
bool local)
740 auto const& xma =
x.const_arrays();
742 auto const& mma =
mask.const_arrays();
747 if (pf(mma[box_no](i,j,k))) {
748 auto const& xfab = xma[box_no];
749 auto const& yfab = yma[box_no];
750 for (
int n = 0; n < numcomp; ++n) {
751 t +=
f(xfab(i,j,k,xcomp+n) , yfab(i,j,k,ycomp+n));
760 #pragma omp parallel if (!system::regtest_reduction) reduction(+:sm)
764 Box const& bx = mfi.growntilebox(nghost);
767 auto const& mfab =
mask.const_array(mfi);
770 if (pf(mfab(i,j,k))) {
771 sm +=
f(xfab(i,j,k,xcomp+n) , yfab(i,j,k,ycomp+n));
784 template <
typename CMF,
typename FMF,
785 std::enable_if_t<IsFabArray_v<CMF> && IsFabArray_v<FMF>,
int> FOO>
795 auto const& ccma = cc.arrays();
797 auto const& fyma = fc[1]->const_arrays();,
798 auto const& fzma = fc[2]->const_arrays(););
802 #if (AMREX_SPACEDIM == 1)
810 #
if (AMREX_SPACEDIM == 1)
822 #pragma omp parallel if (Gpu::notInLaunchRegion())
826 const Box bx = mfi.growntilebox(ngrow);
827 auto const& ccarr = cc.array(mfi);
828 AMREX_D_TERM(
auto const& fxarr = fc[0]->const_array(mfi);,
829 auto const& fyarr = fc[1]->const_array(mfi);,
830 auto const& fzarr = fc[2]->const_array(mfi););
832 #if (AMREX_SPACEDIM == 1)
849 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int>>
852 const IntVect& ratio,
int ngcrse)
862 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int>>
869 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int>>
872 int ratio,
int ngcrse)
877 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int>>
880 const IntVect& ratio,
int ngcrse)
882 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
887 template <
typename FAB>
889 const IntVect& ratio,
int ngcrse)
895 const auto type =
fine.ixType();
897 for (dir = 0; dir < AMREX_SPACEDIM; ++dir) {
898 if (type.nodeCentered(dir)) {
break; }
902 if (dir >= AMREX_SPACEDIM || !tmptype.cellCentered()) {
903 amrex::Abort(
"average_down_faces: not face index type");
910 auto const& crsema =
crse.arrays();
911 auto const& finema =
fine.const_arrays();
924 #pragma omp parallel if (Gpu::notInLaunchRegion())
928 const Box& bx = mfi.growntilebox(ngcrse);
929 auto const& crsearr =
crse.array(mfi);
930 auto const& finearr =
fine.const_array(mfi);
943 crse.ParallelCopy(ctmp,0,0,ncomp,ngcrse,ngcrse);
947 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int>>
952 for (
int idim = 0; idim < AMREX_SPACEDIM; ++idim) {
957 template <
typename FAB>
967 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int> FOO>
970 using T =
typename MF::value_type;
971 const int ncomp = mf.nComp();
975 auto loc = cell.
dim3();
978 Box const& box = mfi.validbox();
981 auto const& fab = mf.const_array(mfi);
984 for (
int n = 0; n < ncomp; ++n) {
985 dp[n] = fab(loc.x,loc.y,loc.z,n);
998 template <typename MF, std::enable_if_t<IsFabArray<MF>::value,
int> FOO>
1001 BoxArray const& ba = mf.boxArray();
1003 const auto nboxes =
static_cast<int>(ba.
size());
1008 for (
int i = 0; i < nboxes; ++i) {
1009 Box const&
b = ba[i];
1011 lo[dir] =
b.smallEnd(dir);
1012 if (
b.contains(lo)) {
1014 hi[dir] =
b.bigEnd(dir);
1015 Box b1d(lo,hi,
b.ixType());
1017 procmap.push_back(dm[i]);
1018 index_map.push_back(i);
1027 MF rmf(rba, rdm, mf.nComp(),
IntVect(0),
1028 MFInfo().SetArena(mf.arena()));
1029 #ifdef AMREX_USE_OMP
1030 #pragma omp parallel if (Gpu::notInLaunchRegion())
1033 Box const&
b = mfi.validbox();
1034 auto const& dfab = rmf.array(mfi);
1035 auto const& sfab = mf.const_array(index_map[mfi.index()]);
1039 dfab(i,j,k,n) = sfab(i,j,k,n);
1046 template <
typename Op,
typename T,
typename FAB,
typename F,
1047 std::enable_if_t<IsBaseFab<FAB>::value
1048 #ifndef AMREX_USE_CUDA
1049 && IsCallableR<T,F,int,int,int,int>::value
1055 Box domain2d = domain;
1062 r.template setVal<RunOn::Device>(initval);
1063 auto const& ar =
r.array();
1065 for (
MFIter mfi(mf,
MFItInfo().UseDefaultStream().DisableDeviceSync());
1068 Box bx = mfi.validbox() & domain;
1070 int box_no = mfi.LocalIndex();
1071 #if defined(AMREX_USE_GPU)
1076 constexpr
int nthreads = 128;
1077 auto nblocks =
static_cast<int>(b2d.
numPts());
1078 #ifdef AMREX_USE_SYCL
1080 amrex::launch<nthreads>(nblocks, shared_mem_bytes,
Gpu::gpuStream(),
1083 int bid = h.blockIdx();
1084 int tid = h.threadIdx();
1089 int bid = blockIdx.x;
1090 int tid = threadIdx.x;
1095 if (direction == 0) {
1096 int k = bid / len.y;
1097 int j = bid - k*len.y;
1100 for (
int i = blo.x + tid; i < blo.x+len.x; i += nthreads) {
1101 Op().local_update(tmp,
f(box_no,i,j,k));
1104 }
else if (direction == 1) {
1105 int k = bid / len.x;
1106 int i = bid - k*len.x;
1109 for (
int j = blo.y + tid; j < blo.y+len.y; j += nthreads) {
1110 Op().local_update(tmp,
f(box_no,i,j,k));
1114 int j = bid / len.x;
1115 int i = bid - j*len.x;
1118 for (
int k = blo.z + tid; k < blo.z+len.z; k += nthreads) {
1119 Op().local_update(tmp,
f(box_no,i,j,k));
1123 #ifdef AMREX_USE_SYCL
1124 Op().template parallel_update<T>(*p, tmp, h);
1126 Op().template parallel_update<T,nthreads>(*p, tmp);
1131 if (direction == 0) {
1134 Op().local_update(ar(0,j,k),
f(box_no,i,j,k));
1136 }
else if (direction == 1) {
1139 Op().local_update(ar(i,0,k),
f(box_no,i,j,k));
1144 Op().local_update(ar(i,j,0),
f(box_no,i,j,k));
#define BL_PROFILE(a)
Definition: AMReX_BLProfiler.H:551
#define BL_ASSERT(EX)
Definition: AMReX_BLassert.H:39
#define AMREX_ASSERT(EX)
Definition: AMReX_BLassert.H:38
#define AMREX_LAUNCH_HOST_DEVICE_LAMBDA(...)
Definition: AMReX_GpuLaunch.nolint.H:16
#define AMREX_HOST_DEVICE_PARALLEL_FOR_1D(...)
Definition: AMReX_GpuLaunch.nolint.H:53
#define AMREX_HOST_DEVICE_PARALLEL_FOR_3D(...)
Definition: AMReX_GpuLaunch.nolint.H:54
#define AMREX_HOST_DEVICE_PARALLEL_FOR_4D(...)
Definition: AMReX_GpuLaunch.nolint.H:55
#define AMREX_GPU_DEVICE
Definition: AMReX_GpuQualifiers.H:18
Array4< int const > offset
Definition: AMReX_HypreMLABecLap.cpp:1089
Real * pdst
Definition: AMReX_HypreMLABecLap.cpp:1090
Array4< Real > fine
Definition: AMReX_InterpFaceRegister.cpp:90
Array4< int const > mask
Definition: AMReX_InterpFaceRegister.cpp:93
Array4< Real const > crse
Definition: AMReX_InterpFaceRegister.cpp:92
#define AMREX_LOOP_3D(bx, i, j, k, block)
Definition: AMReX_Loop.nolint.H:4
#define AMREX_LOOP_4D(bx, ncomp, i, j, k, n, block)
Definition: AMReX_Loop.nolint.H:16
#define AMREX_D_TERM(a, b, c)
Definition: AMReX_SPACE.H:129
#define AMREX_D_DECL(a, b, c)
Definition: AMReX_SPACE.H:104
A FortranArrayBox(FAB)-like object.
Definition: AMReX_BaseFab.H:183
A collection of Boxes stored in an Array.
Definition: AMReX_BoxArray.H:530
IndexType ixType() const noexcept
Return index type of this BoxArray.
Definition: AMReX_BoxArray.H:817
BoxArray & coarsen(int refinement_ratio)
Coarsen each Box in the BoxArray to the specified ratio.
Definition: AMReX_BoxArray.cpp:662
Long size() const noexcept
Return the number of boxes in the BoxArray.
Definition: AMReX_BoxArray.H:577
A class for managing a List of Boxes that share a common IndexType. This class implements operations ...
Definition: AMReX_BoxList.H:52
bool isEmpty() const noexcept
Is this BoxList empty?
Definition: AMReX_BoxList.H:135
void push_back(const Box &bn)
Append a Box to this BoxList.
Definition: AMReX_BoxList.H:93
AMREX_GPU_HOST_DEVICE bool ok() const noexcept
Checks if it is a proper BoxND (including a valid type).
Definition: AMReX_Box.H:200
AMREX_GPU_HOST_DEVICE Long numPts() const noexcept
Returns the number of points contained in the BoxND.
Definition: AMReX_Box.H:346
AMREX_GPU_HOST_DEVICE bool contains(const IntVectND< dim > &p) const noexcept
Returns true if argument is contained within BoxND.
Definition: AMReX_Box.H:204
AMREX_GPU_HOST_DEVICE BoxND & setRange(int dir, int sm_index, int n_cells=1) noexcept
Set the entire range in a given direction, starting at sm_index with length n_cells....
Definition: AMReX_Box.H:1046
Definition: AMReX_FabFactory.H:76
Calculates the distribution of FABs to MPI processes.
Definition: AMReX_DistributionMapping.H:41
IntVect nGrowVect() const noexcept
Definition: AMReX_FabArrayBase.H:79
const BoxArray & boxArray() const noexcept
Return a constant reference to the BoxArray that defines the valid region associated with this FabArr...
Definition: AMReX_FabArrayBase.H:94
bool isFusingCandidate() const noexcept
Is this a good candidate for kernel fusing?
bool is_cell_centered() const noexcept
This tests on whether the FabArray is cell-centered.
bool is_nodal() const noexcept
This tests on whether the FabArray is fully nodal.
const DistributionMapping & DistributionMap() const noexcept
Return constant reference to associated DistributionMapping.
Definition: AMReX_FabArrayBase.H:130
int nComp() const noexcept
Return number of variables (aka components) associated with each point.
Definition: AMReX_FabArrayBase.H:82
An Array of FortranArrayBox(FAB)-like Objects.
Definition: AMReX_FabArray.H:343
Array4< typename FabArray< FAB >::value_type const > const_array(const MFIter &mfi) const noexcept
Definition: AMReX_FabArray.H:1592
void ParallelCopy(const FabArray< FAB > &src, const Periodicity &period=Periodicity::NonPeriodic(), CpOp op=FabArrayBase::COPY)
Definition: AMReX_FabArray.H:777
Array4< typename FabArray< FAB >::value_type const > array(const MFIter &mfi) const noexcept
Definition: AMReX_FabArray.H:1560
MultiArray4< typename FabArray< FAB >::value_type > arrays() noexcept
Definition: AMReX_FabArray.H:1656
MultiArray4< typename FabArray< FAB >::value_type const > const_arrays() const noexcept
Definition: AMReX_FabArray.H:1674
Rectangular problem domain geometry.
Definition: AMReX_Geometry.H:73
Periodicity periodicity() const noexcept
Definition: AMReX_Geometry.H:355
Definition: AMReX_Tuple.H:93
static constexpr AMREX_EXPORT int warp_size
Definition: AMReX_GpuDevice.H:173
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE bool allGE(const IntVectND< dim > &rhs) const noexcept
Returns true if this is greater than or equal to argument for all components. NOTE: This is NOT a str...
Definition: AMReX_IntVect.H:443
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE Dim3 dim3() const noexcept
Definition: AMReX_IntVect.H:163
AMREX_GPU_HOST_DEVICE static constexpr AMREX_FORCE_INLINE IntVectND< dim > TheUnitVector() noexcept
This static member function returns a reference to a constant IntVectND object, all of whose dim argu...
Definition: AMReX_IntVect.H:682
AMREX_GPU_HOST_DEVICE static constexpr AMREX_FORCE_INLINE IntVectND< dim > TheZeroVector() noexcept
This static member function returns a reference to a constant IntVectND object, all of whose dim argu...
Definition: AMReX_IntVect.H:672
Definition: AMReX_MFIter.H:57
bool isValid() const noexcept
Is the iterator valid i.e. is it associated with a FAB?
Definition: AMReX_MFIter.H:141
A collection (stored as an array) of FArrayBox objects.
Definition: AMReX_MultiFab.H:38
Definition: AMReX_PODVector.H:246
T * data() noexcept
Definition: AMReX_PODVector.H:593
iterator begin() noexcept
Definition: AMReX_PODVector.H:601
iterator end() noexcept
Definition: AMReX_PODVector.H:605
This provides length of period for periodic domains. 0 means it is not periodic in that direction....
Definition: AMReX_Periodicity.H:17
static const Periodicity & NonPeriodic() noexcept
Definition: AMReX_Periodicity.cpp:52
This class is a thin wrapper around std::vector. Unlike vector, Vector::operator[] provides bound che...
Definition: AMReX_Vector.H:27
Definition: AMReX_iMultiFab.H:32
@ FAB
Definition: AMReX_AmrvisConstants.H:86
void copy(HostToDevice, InIter begin, InIter end, OutIter result) noexcept
A host-to-device copy routine. Note this is just a wrapper around memcpy, so it assumes contiguous st...
Definition: AMReX_GpuContainers.H:121
static constexpr DeviceToHost deviceToHost
Definition: AMReX_GpuContainers.H:99
void streamSynchronize() noexcept
Definition: AMReX_GpuDevice.H:237
bool inLaunchRegion() noexcept
Definition: AMReX_GpuControl.H:86
bool inNoSyncRegion() noexcept
Definition: AMReX_GpuControl.H:146
gpuStream_t gpuStream() noexcept
Definition: AMReX_GpuDevice.H:218
void Sum(T &v, MPI_Comm comm)
Definition: AMReX_ParallelReduce.H:204
MPI_Comm CommunicatorSub() noexcept
sub-communicator for current frame
Definition: AMReX_ParallelContext.H:70
static int f(amrex::Real t, N_Vector y_data, N_Vector y_rhs, void *user_data)
Definition: AMReX_SundialsIntegrator.H:44
Definition: AMReX_Amr.cpp:49
std::enable_if_t< std::is_integral_v< T > > ParallelFor(TypeList< CTOs... > ctos, std::array< int, sizeof...(CTOs)> const &runtime_options, T N, F &&f)
Definition: AMReX_CTOParallelForImpl.H:200
void FillRandomNormal(MultiFab &mf, int scomp, int ncomp, Real mean, Real stddev)
Fill MultiFab with random numbers from normal distribution.
Definition: AMReX_MultiFabUtil.cpp:1207
int nComp(FabArrayBase const &fa)
void FillRandom(MultiFab &mf, int scomp, int ncomp)
Fill MultiFab with random numbers from uniform distribution.
Definition: AMReX_MultiFabUtil.cpp:1194
void average_down(const MultiFab &S_fine, MultiFab &S_crse, const Geometry &fgeom, const Geometry &cgeom, int scomp, int ncomp, int rr)
Definition: AMReX_MultiFabUtil.cpp:315
BoxND< AMREX_SPACEDIM > Box
Definition: AMReX_BaseFwd.H:27
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void amrex_avg_fc_to_cc(int i, int, int, Array4< CT > const &cc, Array4< FT const > const &fx, int cccomp, GeometryData const &gd) noexcept
Definition: AMReX_MultiFabUtil_1D_C.H:33
Vector< typename MF::value_type > get_cell_data(MF const &mf, IntVect const &cell)
Get data in a cell of MultiFab/FabArray.
Definition: AMReX_MultiFabUtil.H:968
BaseFab< T > ReduceToPlane(int direction, Box const &domain, FabArray< FAB > const &mf, F const &f)
Reduce FabArray/MultiFab data to a plane.
Definition: AMReX_MultiFabUtil.H:1053
Vector< MultiFab > convexify(Vector< MultiFab const * > const &mf, Vector< IntVect > const &refinement_ratio)
Convexify AMR data.
Definition: AMReX_MultiFabUtil.cpp:1220
std::unique_ptr< MultiFab > get_slice_data(int dir, Real coord, const MultiFab &cc, const Geometry &geom, int start_comp, int ncomp, bool interpolate)
Definition: AMReX_MultiFabUtil.cpp:553
void average_down_faces(const Vector< const MF * > &fine, const Vector< MF * > &crse, const IntVect &ratio, int ngcrse=0)
Average fine face-based FabArray onto crse face-based FabArray.
Definition: AMReX_MultiFabUtil.H:850
MultiFab periodicShift(MultiFab const &mf, IntVect const &offset, Periodicity const &period)
Periodic shift MultiFab.
Definition: AMReX_MultiFabUtil.cpp:787
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void amrex_avgdown_faces(Box const &bx, Array4< T > const &crse, Array4< T const > const &fine, int ccomp, int fcomp, int ncomp, IntVect const &ratio, int) noexcept
Definition: AMReX_MultiFabUtil_1D_C.H:130
Real volumeWeightedSum(Vector< MultiFab const * > const &mf, int icomp, Vector< Geometry > const &geom, Vector< IntVect > const &ratio, bool local)
Volume weighted sum for a vector of MultiFabs.
Definition: AMReX_MultiFabUtil.cpp:953
bool isMFIterSafe(const FabArrayBase &x, const FabArrayBase &y)
Definition: AMReX_MFIter.H:219
MultiFab ToMultiFab(const iMultiFab &imf)
Convert iMultiFab to MultiFab.
Definition: AMReX_MultiFabUtil.cpp:543
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE Dim3 lbound(Array4< T > const &a) noexcept
Definition: AMReX_Array4.H:308
ReduceData< Ts... >::Type ParReduce(TypeList< Ops... > operation_list, TypeList< Ts... > type_list, FabArray< FAB > const &fa, IntVect const &nghost, F &&f)
Parallel reduce for MultiFab/FabArray.
Definition: AMReX_ParReduce.H:47
void writeFabs(const MultiFab &mf, const std::string &name)
Write each fab individually.
Definition: AMReX_MultiFabUtil.cpp:530
Real NormHelper(const MultiFab &x, int xcomp, const MultiFab &y, int ycomp, F const &f, int numcomp, IntVect nghost, bool local)
Returns part of a norm based on two MultiFabs.
Definition: AMReX_MultiFabUtil.H:663
MF get_line_data(MF const &mf, int dir, IntVect const &cell)
Get data in a line of MultiFab/FabArray.
Definition: AMReX_MultiFabUtil.H:999
void average_down_edges(const Vector< const MultiFab * > &fine, const Vector< MultiFab * > &crse, const IntVect &ratio, int ngcrse)
Average fine edge-based MultiFab onto crse edge-based MultiFab.
Definition: AMReX_MultiFabUtil.cpp:448
IntVectND< AMREX_SPACEDIM > IntVect
Definition: AMReX_BaseFwd.H:30
Gpu::HostVector< Real > sumToLine(MultiFab const &mf, int icomp, int ncomp, Box const &domain, int direction, bool local)
Sum MultiFab data to line.
Definition: AMReX_MultiFabUtil.cpp:811
void average_face_to_cellcenter(MultiFab &cc, int dcomp, const Vector< const MultiFab * > &fc, int ngrow)
Average face-based MultiFab onto cell-centered MultiFab.
Definition: AMReX_MultiFabUtil.cpp:142
AMREX_GPU_HOST_DEVICE void cast(BaseFab< Tto > &tofab, BaseFab< Tfrom > const &fromfab, Box const &bx, SrcComp scomp, DestComp dcomp, NumComps ncomp) noexcept
Definition: AMReX_BaseFabUtility.H:13
void print_state(const MultiFab &mf, const IntVect &cell, const int n, const IntVect &ng)
Output state data for a single zone.
Definition: AMReX_MultiFabUtil.cpp:525
void computeDivergence(MultiFab &divu, const Array< MultiFab const *, AMREX_SPACEDIM > &umac, const Geometry &geom)
Computes divergence of face-data stored in the umac MultiFab.
Definition: AMReX_MultiFabUtil.cpp:686
bool TilingIfNotGPU() noexcept
Definition: AMReX_MFIter.H:12
void sum_fine_to_coarse(const MultiFab &S_fine, MultiFab &S_crse, int scomp, int ncomp, const IntVect &ratio, const Geometry &cgeom, const Geometry &)
Definition: AMReX_MultiFabUtil.cpp:394
void average_cellcenter_to_face(const Vector< MultiFab * > &fc, const MultiFab &cc, const Geometry &geom, int ncomp, bool use_harmonic_averaging)
Average cell-centered MultiFab onto face-based MultiFab with geometric weighting.
Definition: AMReX_MultiFabUtil.cpp:219
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE Dim3 length(Array4< T > const &a) noexcept
Definition: AMReX_Array4.H:322
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void amrex_avgdown(Box const &bx, Array4< T > const &crse, Array4< T const > const &fine, int ccomp, int fcomp, int ncomp, IntVect const &ratio) noexcept
Definition: AMReX_MultiFabUtil_1D_C.H:195
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE BoxND< dim > coarsen(const BoxND< dim > &b, int ref_ratio) noexcept
Coarsen BoxND by given (positive) refinement ratio. NOTE: if type(dir) = CELL centered: lo <- lo/rati...
Definition: AMReX_Box.H:1304
void Abort(const std::string &msg)
Print out message to cerr and exit via abort().
Definition: AMReX.cpp:221
void average_node_to_cellcenter(MultiFab &cc, int dcomp, const MultiFab &nd, int scomp, int ncomp, int ngrow)
Average nodal-based MultiFab onto cell-centered MultiFab.
Definition: AMReX_MultiFabUtil.cpp:63
void FourthOrderInterpFromFineToCoarse(MultiFab &cmf, int scomp, int ncomp, MultiFab const &fmf, IntVect const &ratio)
Fourth-order interpolation from fine to coarse level.
Definition: AMReX_MultiFabUtil.cpp:1107
FabArray< BaseFab< Long > > ToLongMultiFab(const iMultiFab &imf)
Convert iMultiFab to Long.
Definition: AMReX_MultiFabUtil.cpp:548
void average_edge_to_cellcenter(MultiFab &cc, int dcomp, const Vector< const MultiFab * > &edge, int ngrow)
Average edge-based MultiFab onto cell-centered MultiFab.
Definition: AMReX_MultiFabUtil.cpp:98
void computeGradient(MultiFab &grad, const Array< MultiFab const *, AMREX_SPACEDIM > &umac, const Geometry &geom)
Computes gradient of face-data stored in the umac MultiFab.
Definition: AMReX_MultiFabUtil.cpp:738
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void amrex_avgdown_nodes(Box const &bx, Array4< T > const &crse, Array4< T const > const &fine, int ccomp, int fcomp, int ncomp, IntVect const &ratio) noexcept
Definition: AMReX_MultiFabUtil_1D_C.H:253
iMultiFab makeFineMask(const BoxArray &cba, const DistributionMapping &cdm, const BoxArray &fba, const IntVect &ratio, int crse_value, int fine_value)
Definition: AMReX_MultiFabUtil.cpp:602
std::array< T, N > Array
Definition: AMReX_Array.H:23
void average_down_nodal(const FabArray< FAB > &S_fine, FabArray< FAB > &S_crse, const IntVect &ratio, int ngcrse=0, bool mfiter_is_definitely_safe=false)
Average fine node-based MultiFab onto crse node-centered MultiFab.
Definition: AMReX_MultiFabUtil.H:477
integer, parameter dp
Definition: AMReX_SDCquadrature.F90:8
Definition: AMReX_Array4.H:61
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE int nComp() const noexcept
Definition: AMReX_Array4.H:248
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE std::size_t size() const noexcept
Definition: AMReX_Array4.H:243
parallel copy or add
Definition: AMReX_FabArrayBase.H:536
Definition: AMReX_Geometry.H:25
int coord
Definition: AMReX_Geometry.H:59
Definition: AMReX_GpuTypes.H:86
FabArray memory allocation information.
Definition: AMReX_FabArray.H:65
Definition: AMReX_MFIter.H:20
Struct for holding types.
Definition: AMReX_TypeList.H:12