25 const auto& dom_lo = lbound(domain);
26 const auto& dom_hi = ubound(domain);
31 Vector<BCRec> bcrs_w(1);
34 bool l_use_terrain_fitted_coords = ( (terrain_type == TerrainType::StaticFittedMesh) ||
35 (terrain_type == TerrainType::MovingFittedMesh) );
44 Gpu::DeviceVector<BCRec> bcrs_w_d(1);
45 Gpu::copyAsync(Gpu::hostToDevice, bcrs_w.begin(), bcrs_w.end(), bcrs_w_d.begin());
46 const BCRec* bc_ptr_w = bcrs_w_d.data();
48 GpuArray<GpuArray<Real, AMREX_SPACEDIM*2>,1> l_bc_extdir_vals_d;
50 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)
64 Box bx_xlo(bx); bx_xlo.setBig (0,dom_lo.x-1);
65 Box bx_xhi(bx); bx_xhi.setSmall(0,dom_hi.x+1);
66 ParallelFor(bx_xlo, bx_xhi,
67 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
68 int iflip = dom_lo.x - 1 - i;
72 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][0];
73 if (l_use_terrain_fitted_coords) {
74 dest_arr(i,j,k) = WFromOmega(i,j,k,dest_arr(i,j,k),xvel_arr,yvel_arr,z_phys_nd,dxInv);
77 dest_arr(i,j,k) = dest_arr(dom_lo.x,j,k);
79 dest_arr(i,j,k) = dest_arr(dom_lo.x,j,k);
81 dest_arr(i,j,k) = dest_arr(iflip,j,k);
83 dest_arr(i,j,k) = -dest_arr(iflip,j,k);
86 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
87 int iflip = 2*dom_hi.x + 1 - i;
91 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][3];
92 if (l_use_terrain_fitted_coords) {
93 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),xvel_arr,yvel_arr,z_phys_nd,dxInv);
96 dest_arr(i,j,k) = dest_arr(dom_hi.x,j,k);
98 dest_arr(i,j,k) = dest_arr(dom_hi.x,j,k);
100 dest_arr(i,j,k) = dest_arr(iflip,j,k);
102 dest_arr(i,j,k) = -dest_arr(iflip,j,k);
109 if (!is_periodic_in_y)
112 Box bx_ylo(bx); bx_ylo.setBig (1,dom_lo.y-1);
113 Box bx_yhi(bx); bx_yhi.setSmall(1,dom_hi.y+1);
114 ParallelFor(bx_ylo, bx_yhi,
115 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
116 int jflip = dom_lo.y - 1 - j;
120 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][1];
121 if (l_use_terrain_fitted_coords) {
122 dest_arr(i,j,k) = WFromOmega(i,j,k,dest_arr(i,j,k),xvel_arr,yvel_arr,z_phys_nd,dxInv);
125 dest_arr(i,j,k) = dest_arr(i,dom_lo.y,k);
127 dest_arr(i,j,k) = dest_arr(i,dom_lo.y,k);
129 dest_arr(i,j,k) = dest_arr(i,jflip,k);
131 dest_arr(i,j,k) = -dest_arr(i,jflip,k);
134 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
135 int jflip = 2*dom_hi.y + 1 - j;
139 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][4];
140 if (l_use_terrain_fitted_coords) {
141 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),xvel_arr,yvel_arr,z_phys_nd,dxInv);
144 dest_arr(i,j,k) = dest_arr(i,dom_hi.y,k);
146 dest_arr(i,j,k) = dest_arr(i,dom_hi.y,k);
148 dest_arr(i,j,k) = dest_arr(i,jflip,k);
150 dest_arr(i,j,k) = -dest_arr(i,jflip,k);
154 Gpu::streamSynchronize();
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: ERF_TerrainMetrics.H:414
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, TerrainType terrain_type, int bccomp)
Definition: ERF_BoundaryConditionsZvel.cpp:16
@ open
Definition: ERF_IndexDefines.H:197
@ reflect_odd
Definition: ERF_IndexDefines.H:187
@ foextrap
Definition: ERF_IndexDefines.H:190
@ ext_dir
Definition: ERF_IndexDefines.H:191
@ ext_dir_upwind
Definition: ERF_IndexDefines.H:199
@ reflect_even
Definition: ERF_IndexDefines.H:189