Impose lateral boundary conditions on z velocity.
35 const auto& dom_lo = lbound(domain);
36 const auto& dom_hi = ubound(domain);
41 Vector<BCRec> bcrs_w(1);
44 bool l_use_terrain_fitted_coords = ( (terrain_type == TerrainType::StaticFittedMesh) ||
45 (terrain_type == TerrainType::MovingFittedMesh) );
54 Gpu::DeviceVector<BCRec> bcrs_w_d(1);
55 Gpu::copyAsync(Gpu::hostToDevice, bcrs_w.begin(), bcrs_w.end(), bcrs_w_d.begin());
56 const BCRec* bc_ptr_w = bcrs_w_d.data();
58 GpuArray<GpuArray<Real, AMREX_SPACEDIM*2>,1> l_bc_extdir_vals_d;
60 for (
int ori = 0; ori < 2*AMREX_SPACEDIM; ori++) {
64 GeometryData
const& geomdata =
m_geom.data();
65 bool is_periodic_in_x = geomdata.isPeriodic(0);
66 bool is_periodic_in_y = geomdata.isPeriodic(1);
71 if (!is_periodic_in_x)
74 Box bx_xlo(bx); bx_xlo.setBig (0,dom_lo.x-1);
75 Box bx_xhi(bx); bx_xhi.setSmall(0,dom_hi.x+1);
77 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
78 int iflip = dom_lo.x - 1 - i;
82 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][0];
83 if (l_use_terrain_fitted_coords) {
84 dest_arr(i,j,k) = WFromOmega(i,j,k,dest_arr(i,j,k),
86 mf_u,mf_v,z_phys_nd,dxInv);
89 dest_arr(i,j,k) = dest_arr(dom_lo.x,j,k);
91 dest_arr(i,j,k) = dest_arr(dom_lo.x,j,k);
93 dest_arr(i,j,k) = dest_arr(iflip,j,k);
95 dest_arr(i,j,k) = -dest_arr(iflip,j,k);
98 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
99 int iflip = 2*dom_hi.x + 1 - i;
103 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][3];
104 if (l_use_terrain_fitted_coords) {
105 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),
107 mf_u,mf_v,z_phys_nd,
dxInv);
110 dest_arr(i,j,k) = dest_arr(dom_hi.x,j,k);
112 dest_arr(i,j,k) = dest_arr(dom_hi.x,j,k);
114 dest_arr(i,j,k) = dest_arr(iflip,j,k);
116 dest_arr(i,j,k) = -dest_arr(iflip,j,k);
123 if (!is_periodic_in_y)
126 Box bx_ylo(bx); bx_ylo.setBig (1,dom_lo.y-1);
127 Box bx_yhi(bx); bx_yhi.setSmall(1,dom_hi.y+1);
129 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
130 int jflip = dom_lo.y - 1 - j;
134 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][1];
135 if (l_use_terrain_fitted_coords) {
136 dest_arr(i,j,k) = WFromOmega(i,j,k,dest_arr(i,j,k),
138 mf_u,mf_v,z_phys_nd,dxInv);
141 dest_arr(i,j,k) = dest_arr(i,dom_lo.y,k);
143 dest_arr(i,j,k) = dest_arr(i,dom_lo.y,k);
145 dest_arr(i,j,k) = dest_arr(i,jflip,k);
147 dest_arr(i,j,k) = -dest_arr(i,jflip,k);
150 [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) {
151 int jflip = 2*dom_hi.y + 1 - j;
155 dest_arr(i,j,k) = (zvel_bc_ptr) ? zvel_bc_ptr[k] : l_bc_extdir_vals_d[0][4];
156 if (l_use_terrain_fitted_coords) {
157 dest_arr(i,j,k) =
WFromOmega(i,j,k,dest_arr(i,j,k),
159 mf_u,mf_v,z_phys_nd,
dxInv);
162 dest_arr(i,j,k) = dest_arr(i,dom_hi.y,k);
164 dest_arr(i,j,k) = dest_arr(i,dom_hi.y,k);
166 dest_arr(i,j,k) = dest_arr(i,jflip,k);
168 dest_arr(i,j,k) = -dest_arr(i,jflip,k);
172 Gpu::streamSynchronize();
constexpr amrex::Real zero
Definition: ERF_Constants.H:8
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);})
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:464
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:256
@ reflect_odd
Definition: ERF_IndexDefines.H:245
@ foextrap
Definition: ERF_IndexDefines.H:248
@ ext_dir
Definition: ERF_IndexDefines.H:249
@ ext_dir_upwind
Definition: ERF_IndexDefines.H:257
@ reflect_even
Definition: ERF_IndexDefines.H:247