Impose lateral boundary conditions on z velocity.
35 const auto& dom_lo = lbound(domain);
36 const auto& dom_hi = ubound(domain);
49 const auto& xvel_lo = lbound(xvel_arr);
50 const auto& xvel_hi = ubound(xvel_arr);
51 const auto& yvel_lo = lbound(yvel_arr);
52 const auto& yvel_hi = ubound(yvel_arr);
57 Vector<BCRec> bcrs_w(1);
60 bool l_use_terrain_fitted_coords = ( (terrain_type == TerrainType::StaticFittedMesh) ||
61 (terrain_type == TerrainType::MovingFittedMesh) );
70 Gpu::DeviceVector<BCRec> bcrs_w_d(1);
71 Gpu::copyAsync(Gpu::hostToDevice, bcrs_w.begin(), bcrs_w.end(), bcrs_w_d.begin());
72 const BCRec* bc_ptr_w = bcrs_w_d.data();
74 GpuArray<GpuArray<Real, AMREX_SPACEDIM*2>,1> l_bc_extdir_vals_d;
76 for (
int ori = 0; ori < 2*AMREX_SPACEDIM; ori++) {
80 GeometryData
const& geomdata =
m_geom.data();
81 bool is_periodic_in_x = geomdata.isPeriodic(0);
82 bool is_periodic_in_y = geomdata.isPeriodic(1);
87 if (!is_periodic_in_x)
90 Box bx_xlo(bx); bx_xlo.setBig (0,dom_lo.x-1);
91 Box bx_xhi(bx); bx_xhi.setSmall(0,dom_hi.x+1);
93 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
94 int iflip = dom_lo.x - 1 - i;
98 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][0];
99 if (l_use_terrain_fitted_coords) {
100 dest_arr(i,j,k) = WFromOmega(i,j,k,dest_arr(i,j,k),
102 mf_u,mf_v,z_phys_nd,dxInv);
105 dest_arr(i,j,k) = dest_arr(dom_lo.x,j,k);
107 dest_arr(i,j,k) = dest_arr(dom_lo.x,j,k);
109 dest_arr(i,j,k) = dest_arr(iflip,j,k);
111 dest_arr(i,j,k) = -dest_arr(iflip,j,k);
114 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
115 int iflip = 2*dom_hi.x + 1 - i;
119 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][3];
120 if (l_use_terrain_fitted_coords) {
121 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),
123 mf_u,mf_v,z_phys_nd,
dxInv);
126 dest_arr(i,j,k) = dest_arr(dom_hi.x,j,k);
128 dest_arr(i,j,k) = dest_arr(dom_hi.x,j,k);
130 dest_arr(i,j,k) = dest_arr(iflip,j,k);
132 dest_arr(i,j,k) = -dest_arr(iflip,j,k);
139 if (!is_periodic_in_y)
142 Box bx_ylo(bx); bx_ylo.setBig (1,dom_lo.y-1);
143 Box bx_yhi(bx); bx_yhi.setSmall(1,dom_hi.y+1);
145 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
146 int jflip = dom_lo.y - 1 - j;
150 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][1];
151 if (l_use_terrain_fitted_coords) {
152 dest_arr(i,j,k) = WFromOmega(i,j,k,dest_arr(i,j,k),
154 mf_u,mf_v,z_phys_nd,dxInv);
157 dest_arr(i,j,k) = dest_arr(i,dom_lo.y,k);
159 dest_arr(i,j,k) = dest_arr(i,dom_lo.y,k);
161 dest_arr(i,j,k) = dest_arr(i,jflip,k);
163 dest_arr(i,j,k) = -dest_arr(i,jflip,k);
166 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
167 int jflip = 2*dom_hi.y + 1 - j;
171 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][4];
172 if (l_use_terrain_fitted_coords) {
173 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),
175 mf_u,mf_v,z_phys_nd,
dxInv);
178 dest_arr(i,j,k) = dest_arr(i,dom_hi.y,k);
180 dest_arr(i,j,k) = dest_arr(i,dom_hi.y,k);
182 dest_arr(i,j,k) = dest_arr(i,jflip,k);
184 dest_arr(i,j,k) = -dest_arr(i,jflip,k);
188 Gpu::streamSynchronize();
amrex::GpuArray< Real, AMREX_SPACEDIM > dxInv
Definition: ERF_InitCustomPertVels_ParticleTests.H:17
ParallelFor(fab_box, [=] AMREX_GPU_DEVICE(int i, int j, int k) { qrcuten_arr(i, j, k)=Real(0);qscuten_arr(i, j, k)=Real(0);qicuten_arr(i, j, k)=Real(0);})
constexpr amrex::Real zero
Definition: ERF_NumericalConstants.H:36
amrex::Real Real
Definition: ERF_ShocInterface.H:19
AMREX_GPU_DEVICE AMREX_FORCE_INLINE amrex::Real WFromOmega(int &i, int &j, int &k, amrex::Real omega, const amrex::Array4< const amrex::Real > &u_arr, const amrex::Array4< const amrex::Real > &v_arr, const amrex::Array4< const amrex::Real > &mf_u, const amrex::Array4< const amrex::Real > &mf_v, const amrex::Array4< const amrex::Real > &z_nd, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv)
Definition: ERF_TerrainMetrics.H:863
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 > &mf_u, const amrex::Array4< amrex::Real const > &mf_v, const amrex::Array4< amrex::Real const > &z_nd, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > dxInv, TerrainType terrain_type, int bccomp, const double time)
Definition: ERF_BoundaryConditionsZvel.cpp:23
@ open
Definition: ERF_IndexDefines.H:307
@ reflect_odd
Definition: ERF_IndexDefines.H:296
@ foextrap
Definition: ERF_IndexDefines.H:299
@ ext_dir
Definition: ERF_IndexDefines.H:300
@ ext_dir_upwind
Definition: ERF_IndexDefines.H:308
@ reflect_even
Definition: ERF_IndexDefines.H:298