25 const auto& dom_lo = lbound(domain);
26 const auto& dom_hi = ubound(domain);
32 Vector<BCRec> bcrs_w(1);
42 Gpu::DeviceVector<BCRec> bcrs_w_d(ncomp);
43 Gpu::copyAsync(Gpu::hostToDevice, bcrs_w.begin(), bcrs_w.end(), bcrs_w_d.begin());
44 const BCRec* bc_ptr_w = bcrs_w_d.data();
46 GpuArray<GpuArray<Real, AMREX_SPACEDIM*2>,AMREX_SPACEDIM+
NVAR_max> l_bc_extdir_vals_d;
50 for (
int i = 0; i < ncomp; i++)
51 for (
int ori = 0; ori < 2*AMREX_SPACEDIM; ori++)
54 GeometryData
const& geomdata =
m_geom.data();
55 bool is_periodic_in_x = geomdata.isPeriodic(0);
56 bool is_periodic_in_y = geomdata.isPeriodic(1);
61 if (!is_periodic_in_x)
63 Box bx_xlo(bx); bx_xlo.setBig (0,dom_lo.x-1);
64 Box bx_xhi(bx); bx_xhi.setSmall(0,dom_hi.x+1);
66 bx_xlo, ncomp, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) {
67 int iflip = dom_lo.x - 1 - i;
69 dest_arr(i,j,k) = l_bc_extdir_vals_d[n][0];
71 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),xvel_arr,yvel_arr,z_phys_nd,dxInv);
74 dest_arr(i,j,k) = dest_arr(dom_lo.x,j,k);
76 dest_arr(i,j,k) = dest_arr(dom_lo.x,j,k);
78 dest_arr(i,j,k) = dest_arr(iflip,j,k);
80 dest_arr(i,j,k) = -dest_arr(iflip,j,k);
83 bx_xhi, ncomp, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) {
84 int iflip = 2*dom_hi.x + 1 - i;
86 dest_arr(i,j,k) = l_bc_extdir_vals_d[n][3];
88 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),xvel_arr,yvel_arr,z_phys_nd,dxInv);
91 dest_arr(i,j,k) = dest_arr(dom_hi.x,j,k);
93 dest_arr(i,j,k) = dest_arr(dom_hi.x,j,k);
95 dest_arr(i,j,k) = dest_arr(iflip,j,k);
97 dest_arr(i,j,k) = -dest_arr(iflip,j,k);
104 if (!is_periodic_in_y)
106 Box bx_ylo(bx); bx_ylo.setBig (1,dom_lo.y-1);
107 Box bx_yhi(bx); bx_yhi.setSmall(1,dom_hi.y+1);
108 ParallelFor(bx_ylo, ncomp, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) {
109 int jflip = dom_lo.y - 1 - j;
111 dest_arr(i,j,k) = l_bc_extdir_vals_d[n][1];
113 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),xvel_arr,yvel_arr,z_phys_nd,dxInv);
116 dest_arr(i,j,k) = dest_arr(i,dom_lo.y,k);
118 dest_arr(i,j,k) = dest_arr(i,dom_lo.y,k);
120 dest_arr(i,j,k) = dest_arr(i,jflip,k);
122 dest_arr(i,j,k) = -dest_arr(i,jflip,k);
125 bx_yhi, ncomp, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) {
126 int jflip = 2*dom_hi.y + 1 - j;
128 dest_arr(i,j,k) = l_bc_extdir_vals_d[n][4];
129 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),xvel_arr,yvel_arr,z_phys_nd,dxInv);
131 dest_arr(i,j,k) = dest_arr(i,dom_hi.y,k);
133 dest_arr(i,j,k) = dest_arr(i,dom_hi.y,k);
135 dest_arr(i,j,k) = dest_arr(i,jflip,k);
137 dest_arr(i,j,k) = -dest_arr(i,jflip,k);
142 Gpu::streamSynchronize();
#define NVAR_max
Definition: IndexDefines.H:27
AMREX_GPU_DEVICE AMREX_FORCE_INLINE amrex::Real WFromOmega(int i, int j, int k, amrex::Real omega, amrex::Real u, amrex::Real v, const amrex::Array4< const amrex::Real > &z_nd, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv)
Definition: TerrainMetrics.H:399
void impose_lateral_zvel_bcs(const amrex::Array4< amrex::Real > &dest_arr, const amrex::Array4< amrex::Real const > &xvel_arr, const amrex::Array4< amrex::Real const > &yvel_arr, const amrex::Box &bx, const amrex::Box &domain, const amrex::Array4< amrex::Real const > &z_nd, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > dxInv, int bccomp_w)
Definition: BoundaryConditions_zvel.cpp:16
@ open
Definition: IndexDefines.H:157
@ reflect_odd
Definition: IndexDefines.H:148
@ foextrap
Definition: IndexDefines.H:151
@ ext_dir
Definition: IndexDefines.H:152
@ reflect_even
Definition: IndexDefines.H:150