ERF_WriteBndryPlanes.cpp File Reference

#include "AMReX_Gpu.H"
#include "AMReX_ParmParse.H"
#include "AMReX_PlotFileUtil.H"
#include "AMReX_MultiFabUtil.H"
#include "ERF_WriteBndryPlanes.H"
#include "ERF_IndexDefines.H"
#include "ERF_Derive.H"

Include dependency graph for ERF_WriteBndryPlanes.cpp:

Functions
void	br_shift (OrientationIter oit, const BndryRegister &b1, BndryRegister &b2)

Function Documentation

br_shift()

void br_shift	(	OrientationIter	oit,
		const BndryRegister &	b1,
		BndryRegister &	b2
	)

Copies the contents of one boundary register to another in either x or y dimensions.

Parameters

oit	Orientation iterator used for determining which dimension we are to copy
b1	Boundary register containing the source data
b2	Boundary register containing the destination data

{
    auto ori = oit();
    int ncomp = b1[ori].nComp();
    if (ori.coordDir() < 2) {
#ifdef AMREX_USE_OMP
#pragma omp parallel if (Gpu::notInLaunchRegion())
#endif
        for (FabSetIter bfsi(b1[ori]); bfsi.isValid(); ++bfsi) {
            int idx = bfsi.index();
            const Box& bx1 = b1[ori].boxArray()[idx];
            const Box& bx2 = b2[ori].boxArray()[idx];
 
            // Copy onto a boundary register based on a box starting at (0,0,0)
            Array4<Real>       dest_arr = b2[ori][idx].array();
            Array4<Real const>  src_arr = b1[ori][idx].const_array();
            int ioff = bx1.smallEnd(0) - bx2.smallEnd(0);
            int joff = bx1.smallEnd(1) - bx2.smallEnd(1);
            ParallelFor(bx2, ncomp,
            [=] AMREX_GPU_DEVICE(int i, int j, int k, int n) noexcept
            {
                dest_arr(i,j,k,n) = src_arr(i+ioff,j+joff,k,n);
            });
        }
    }
}

Referenced by WriteBndryPlanes::write_planes().

Here is the call graph for this function:

Here is the caller graph for this function: