168 Real h_xi, h_eta, h_zeta;
175 Real px_hi = (
x(i+1,j,k) -
x(i,j,k)) * dxinv;
178 Real pz_hi_md_hi =
Real(0.5) * (
x(i+1,j ,k+1) +
x(i ,j ,k+1)
179 -
x(i+1,j ,k ) -
x(i ,j ,k ) );
180 h_xi =
Real(0.25) * ( zp(i+1,j ,k+1) - zp(i-1,j ,k+1)
181 +zp(i+1,j+1,k+1) - zp(i-1,j+1,k+1) ) * dxinv;
182 h_zeta =
Real(0.25) * ( zp(i+1,j ,k+2) - zp(i+1,j ,k )
183 +zp(i+1,j+1,k+2) - zp(i+1,j+1,k ) );
184 pz_hi_md_hi *= h_xi / h_zeta;
187 Real pz_hi_md_lo =
Real(0.5) * (
x(i+1,j ,k ) +
x(i ,j ,k )
188 -
x(i+1,j ,k-1) -
x(i ,j ,k-1) );
189 h_xi =
Real(0.25) * ( zp(i+1,j ,k ) - zp(i-1,j ,k )
190 +zp(i+1,j+1,k ) - zp(i-1,j+1,k ) ) * dxinv;
191 h_zeta =
Real(0.25) * ( zp(i+1,j ,k+1) - zp(i+1,j ,k-1)
192 +zp(i+1,j+1,k+1) - zp(i+1,j+1,k-1) );
193 pz_hi_md_lo *= h_xi / h_zeta;
196 px_hi -=
Real(0.5) * ( pz_hi_md_hi + pz_hi_md_lo );
201 Real px_lo = (
x(i,j,k) -
x(i-1,j,k)) * dxinv;
204 Real pz_lo_md_hi =
Real(0.5) * (
x(i,j,k+1) +
x(i-1,j,k+1)
205 -
x(i,j,k ) -
x(i-1,j,k ) );
206 h_xi =
Real(0.25) * ( zp(i,j ,k+1) - zp(i-2,j ,k+1)
207 +zp(i,j+1,k+1) - zp(i-2,j+1,k+1) ) * dxinv;
208 h_zeta =
Real(0.25) * ( zp(i,j ,k+2) - zp(i ,j ,k )
209 +zp(i,j+1,k+2) - zp(i ,j+1,k ) );
210 pz_lo_md_hi *= h_xi / h_zeta;
213 Real pz_lo_md_lo =
Real(0.5) * (
x(i,j,k ) +
x(i-1,j,k )
214 -
x(i,j,k-1) -
x(i-1,j,k-1) );
215 h_xi =
Real(0.25) * ( zp(i,j ,k ) - zp(i-2,j ,k )
216 +zp(i,j+1,k ) - zp(i-2,j+1,k ) ) * dxinv;
217 h_zeta =
Real(0.25) * ( zp(i,j ,k+1) - zp(i ,j ,k-1)
218 +zp(i,j+1,k+1) - zp(i ,j+1,k-1) );
219 pz_lo_md_lo *= h_xi / h_zeta;
222 px_lo -=
Real(0.5) * ( pz_lo_md_hi + pz_lo_md_lo );
228 Real py_hi = (
x(i,j+1,k) -
x(i,j,k)) * dyinv;
231 Real pz_md_hi_hi =
Real(0.5) * (
x(i,j+1,k+1) +
x(i,j,k+1)
232 -
x(i,j+1,k ) -
x(i,j,k ) );
233 h_eta =
Real(0.25) * ( zp(i ,j+1,k+1) - zp(i ,j-1,k+1)
234 +zp(i+1,j+1,k+1) - zp(i+1,j-1,k+1) ) * dyinv;
235 h_zeta =
Real(0.25) * ( zp(i ,j+1,k+2) - zp(i ,j+1,k )
236 +zp(i+1,j+1,k+2) - zp(i+1,j+1,k ) );
237 pz_md_hi_hi *= h_eta/ h_zeta;
240 Real pz_md_hi_lo =
Real(0.5) * (
x(i,j+1,k ) +
x(i,j,k )
241 -
x(i,j+1,k-1) -
x(i,j,k-1) );
242 h_eta =
Real(0.25) * ( zp(i ,j+1,k ) - zp(i ,j-1,k)
243 +zp(i+1,j+1,k ) - zp(i+1,j-1,k) ) * dyinv;
244 h_zeta =
Real(0.25) * ( zp(i ,j+1,k+1) - zp(i ,j+1,k-1)
245 +zp(i+1,j+1,k+1) - zp(i+1,j+1,k-1) );
246 pz_md_hi_lo *= h_eta/ h_zeta;
249 py_hi -=
Real(0.5) * ( pz_md_hi_hi + pz_md_hi_lo );
255 Real py_lo = (
x(i,j,k) -
x(i,j-1,k)) * dyinv;
258 Real pz_md_lo_hi =
Real(0.5) * (
x(i ,j,k+1) +
x(i ,j-1,k+1)
259 -
x(i ,j,k ) -
x(i ,j-1,k ) );
260 h_eta =
Real(0.25) * ( zp(i ,j,k+1) - zp(i ,j-2,k+1)
261 +zp(i+1,j,k+1) - zp(i+1,j-2,k+1) ) * dyinv;
262 h_zeta =
Real(0.25) * ( zp(i ,j,k+2) - zp(i ,j ,k )
263 +zp(i+1,j,k+2) - zp(i+1,j ,k ) );
264 pz_md_lo_hi *= h_eta/ h_zeta;
267 Real pz_md_lo_lo =
Real(0.5) * (
x(i ,j,k ) +
x(i ,j-1,k )
268 -
x(i ,j,k-1) -
x(i ,j-1,k-1) );
269 h_eta =
Real(0.25) * ( zp(i ,j,k ) - zp(i ,j-2,k )
270 +zp(i+1,j,k ) - zp(i+1,j-2,k ) ) * dyinv;
271 h_zeta =
Real(0.25) * ( zp(i ,j,k+1) - zp(i ,j ,k-1)
272 +zp(i+1,j,k+1) - zp(i+1,j ,k-1) );
273 pz_md_lo_lo *= h_eta/ h_zeta;
276 py_lo -=
Real(0.5) * ( pz_md_lo_hi + pz_md_lo_lo );
282 Real pz_hi =
x(i,j,k+1) -
x(i,j,k );
283 Real hzeta_inv_on_zhi = 8.0 / ( (zp(i,j,k+2) + zp(i+1,j,k+2) + zp(i,j+1,k+2) + zp(i+1,j+1,k+2))
284 -(zp(i,j,k ) + zp(i+1,j,k ) + zp(i,j+1,k ) + zp(i+1,j+1,k )) );
285 pz_hi *= hzeta_inv_on_zhi;
288 Real px_hi_md_hi =
Real(0.5) * (
x(i+1,j,k+1) -
x(i ,j ,k+1)
289 +
x(i+1,j,k ) -
x(i ,j ,k )) * dxinv;
290 Real px_lo_md_hi =
Real(0.5) * (
x(i ,j,k+1) -
x(i-1,j ,k+1)
291 +
x(i ,j,k ) -
x(i-1,j ,k )) * dxinv;
292 Real py_md_hi_hi =
Real(0.5) * (
x(i,j+1,k+1) -
x(i ,j ,k+1)
293 +
x(i,j+1,k ) -
x(i ,j ,k )) * dyinv;
294 Real py_md_lo_hi =
Real(0.5) * (
x(i,j ,k+1) -
x(i ,j-1,k+1)
295 +
x(i,j ,k ) -
x(i ,j-1,k )) * dyinv;
298 Real h_xi_on_zhi = 0.5 * ( zp(i+1,j+1,k+1) + zp(i+1,j,k+1) - zp(i,j+1,k+1) - zp(i,j,k+1) ) * dxinv;
299 Real h_eta_on_zhi = 0.5 * ( zp(i+1,j+1,k+1) + zp(i,j+1,k+1) - zp(i+1,j,k+1) - zp(i,j,k+1) ) * dyinv;
303 pz_hi -= 0.5 * h_xi_on_zhi * ( (px_hi_md_hi + px_lo_md_hi) - (pz_hi_md_hi + pz_lo_md_hi) );
304 pz_hi -= 0.5 * h_eta_on_zhi * ( (py_md_hi_hi + py_md_lo_hi) - (pz_md_hi_hi + pz_md_lo_hi) );
310 Real pz_lo =
x(i,j,k ) -
x(i,j,k-1);
311 Real hzeta_inv_on_zlo = 8.0 / ( (zp(i,j,k+1) + zp(i+1,j,k+1) + zp(i,j+1,k+1) + zp(i+1,j+1,k+1))
312 -(zp(i,j,k-1) + zp(i+1,j,k-1) + zp(i,j+1,k-1) + zp(i+1,j+1,k-1)) );
313 pz_lo *= hzeta_inv_on_zlo;
316 Real px_hi_md_lo =
Real(0.5) * (
x(i+1,j ,k ) -
x(i ,j ,k )
317 +
x(i+1,j ,k-1) -
x(i ,j ,k-1)) * dxinv;
318 Real px_lo_md_lo =
Real(0.5) * (
x(i ,j ,k ) -
x(i-1,j ,k )
319 +
x(i ,j ,k-1) -
x(i-1,j ,k-1)) * dxinv;
320 Real py_md_hi_lo =
Real(0.5) * (
x(i ,j+1,k ) -
x(i ,j ,k )
321 +
x(i ,j+1,k-1) -
x(i ,j ,k-1)) * dyinv;
322 Real py_md_lo_lo =
Real(0.5) * (
x(i ,j ,k ) -
x(i ,j-1,k )
323 +
x(i ,j ,k-1) -
x(i ,j-1,k-1)) * dyinv;
326 Real h_xi_on_zlo = 0.5 * (zp(i+1,j+1,k) + zp(i+1,j,k) - zp(i,j+1,k) - zp(i,j,k)) * dxinv;
327 Real h_eta_on_zlo = 0.5 * (zp(i+1,j+1,k) + zp(i,j+1,k) - zp(i+1,j,k) - zp(i,j,k)) * dyinv;
331 pz_lo -= 0.5 * h_xi_on_zlo * ( (px_hi_md_lo + px_lo_md_lo) - (pz_hi_md_lo + pz_lo_md_lo) );
332 pz_lo -= 0.5 * h_eta_on_zlo * ( (py_md_hi_lo + py_md_lo_lo) - (pz_md_hi_lo + pz_md_lo_lo) );
355 if (k == 0) pz_lo = 0.0;
357 y(i,j,k) = (ax(i+1,j,k)*px_hi - ax(i,j,k)*px_lo) * dxinv
358 +(ay(i,j+1,k)*py_hi - ay(i,j,k)*py_lo) * dyinv
359 +(az(i,j,k+1)*pz_hi - az(i,j,k)*pz_lo) * dzinv;
360 y(i,j,k) /= dJ(i,j,k);
amrex::Real Real
Definition: ERF_ShocInterface.H:19
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE T terrpoisson_flux_x(int i, int j, int k, amrex::Array4< T const > const &sol, amrex::Array4< T const > const &zp, T dxinv) noexcept
Definition: ERF_TerrainPoisson_3D_K.H:8
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE T terrpoisson_flux_z(int i, int j, int k, amrex::Array4< T const > const &sol, amrex::Array4< T const > const &zp, T dxinv, T dyinv) noexcept
Definition: ERF_TerrainPoisson_3D_K.H:83
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE T terrpoisson_flux_y(int i, int j, int k, amrex::Array4< T const > const &sol, amrex::Array4< T const > const &zp, T dyinv) noexcept
Definition: ERF_TerrainPoisson_3D_K.H:46