5 #include <AMReX_MultiFab.H>
6 #include <AMReX_BCRec.H>
15 void ChopGrids2D (amrex::BoxArray& ba,
const amrex::Box& domain,
int target_size);
27 void make_J (
const amrex::Geometry& geom,
28 amrex::MultiFab& z_phys_nd,
29 amrex::MultiFab& detJ_cc);
32 amrex::MultiFab& z_phys_nd,
41 amrex::MultiFab& z_phys_nd,
42 amrex::MultiFab& z_phys_cc);
48 amrex::MultiFab& yvel_out,
49 amrex::MultiFab& zvel_out,
50 const amrex::MultiFab& cons_in,
51 const amrex::MultiFab& xmom_in,
52 const amrex::MultiFab& ymom_in,
53 const amrex::MultiFab& zmom_in,
54 const amrex::Box& domain,
55 const amrex::Vector<amrex::BCRec>& domain_bcs_type_h);
61 const amrex::IntVect & xvel_ngrow,
62 const amrex::MultiFab& yvel_in,
63 const amrex::IntVect & yvel_ngrow,
64 const amrex::MultiFab& zvel_in,
65 const amrex::IntVect & zvel_ngrow,
66 const amrex::MultiFab& cons_in,
67 amrex::MultiFab& xmom_out,
68 amrex::MultiFab& ymom_out,
69 amrex::MultiFab& zmom_out,
70 const amrex::Box& domain,
71 const amrex::Vector<amrex::BCRec>& domain_bcs_type_h);
78 amrex::MultiFab&
xmom, amrex::MultiFab&
ymom, amrex::MultiFab&
zmom,
79 const amrex::Box& domain,
const amrex::Vector<amrex::BCRec>& domain_bcs_type_h);
86 const amrex::Box& domain,
92 const int& set_width=0,
93 const amrex::IntVect& ng_vect=amrex::IntVect(0,0,0),
94 const bool get_int_ng=
false);
101 const amrex::Box& domain,
102 const int& set_width,
107 const amrex::IntVect& ng_vect=amrex::IntVect(0,0,0));
113 const amrex::Real& start_bdy_time,
114 const amrex::Real& time,
115 const amrex::Real& delta_t,
118 const amrex::Geometry& geom,
119 amrex::Vector<amrex::MultiFab>& S_rhs,
120 amrex::Vector<amrex::MultiFab>& S_old_data,
121 amrex::Vector<amrex::MultiFab>& S_cur_data,
122 amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_xlo,
123 amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_xhi,
124 amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_ylo,
125 amrex::Vector<amrex::Vector<amrex::FArrayBox>>& bdy_data_yhi);
132 const amrex::Real& delta_t,
134 const int& set_width,
135 const amrex::Geometry& geom,
140 amrex::Vector<amrex::BCRec>& domain_bcs_type,
141 amrex::Vector<amrex::MultiFab>& S_rhs_f,
142 amrex::Vector<amrex::MultiFab>& S_data_f);
149 amrex::Real& t_avg_cnt,
150 amrex::MultiFab* vel_t_avg,
151 amrex::MultiFab&
xvel,
152 amrex::MultiFab&
yvel,
153 amrex::MultiFab&
zvel);
173 const int& set_width_x,
174 const int& set_width_y,
175 const amrex::Dim3& dom_lo,
176 const amrex::Dim3& dom_hi,
177 const amrex::Box& bx_xlo,
178 const amrex::Box& bx_xhi,
179 const amrex::Box& bx_ylo,
180 const amrex::Box& bx_yhi,
181 const amrex::Array4<const amrex::Real>& arr_xlo,
182 const amrex::Array4<const amrex::Real>& arr_xhi,
183 const amrex::Array4<const amrex::Real>& arr_ylo,
184 const amrex::Array4<const amrex::Real>& arr_yhi,
185 const amrex::Array4<const amrex::Real>& data_arr,
186 const amrex::Array4<amrex::Real>& rhs_arr)
188 int Spec_z_x = set_width_x;
189 int Spec_z_y = set_width_y;
190 amrex::ParallelFor(bx_xlo, num_var, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) noexcept
193 int j_lo = std::min(j-dom_lo.y,width-1);
194 int j_hi = std::min(dom_hi.y-j,width-1);
195 int jj = std::min(j_lo,j_hi);
196 int n_ind_y = jj + 1;
197 int n_ind_x = i - dom_lo.x + 1;
198 if ((n_ind_x <= Spec_z_x) || (n_ind_y <= Spec_z_y)) {
199 amrex::Real tend = ( arr_xlo(i,j,k) - data_arr(i,j,k,n+icomp) ) / dt;
200 rhs_arr(i,j,k,n+icomp) = tend;
203 bx_xhi, num_var, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) noexcept
206 int j_lo = std::min(j-dom_lo.y,width-1);
207 int j_hi = std::min(dom_hi.y-j,width-1);
208 int jj = std::min(j_lo,j_hi);
209 int n_ind_y = jj + 1;
210 int n_ind_x = dom_hi.x - i + 1;
211 if ((n_ind_x <= Spec_z_x) || (n_ind_y <= Spec_z_y)) {
212 amrex::Real tend = ( arr_xhi(i,j,k) - data_arr(i,j,k,n+icomp) ) / dt;
213 rhs_arr(i,j,k,n+icomp) = tend;
217 amrex::ParallelFor(bx_ylo, num_var, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) noexcept
220 int n_ind = j - dom_lo.y + 1;
221 if (n_ind <= Spec_z_y) {
222 amrex::Real tend = ( arr_ylo(i,j,k) - data_arr(i,j,k,n+icomp) ) / dt;
223 rhs_arr(i,j,k,n+icomp) = tend;
226 bx_yhi, num_var, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) noexcept
229 int n_ind = dom_hi.y - j + 1;
230 if (n_ind <= Spec_z_y) {
231 amrex::Real tend = ( arr_yhi(i,j,k) - data_arr(i,j,k,n+icomp) ) / dt;
232 rhs_arr(i,j,k,n+icomp) = tend;
266 const int& set_width,
267 const amrex::Dim3& dom_lo,
268 const amrex::Dim3& dom_hi,
269 const amrex::Real& F1,
270 const amrex::Real& F2,
271 const amrex::Box& bx_xlo,
272 const amrex::Box& bx_xhi,
273 const amrex::Box& bx_ylo,
274 const amrex::Box& bx_yhi,
275 const amrex::Array4<const amrex::Real>& arr_xlo,
276 const amrex::Array4<const amrex::Real>& arr_xhi,
277 const amrex::Array4<const amrex::Real>& arr_ylo,
278 const amrex::Array4<const amrex::Real>& arr_yhi,
279 const amrex::Array4<const amrex::Real>& data_arr,
280 const amrex::Array4<amrex::Real>& rhs_arr)
283 int Spec_z = set_width;
284 int Relax_z = width - Spec_z + 1;
285 amrex::Real num = amrex::Real(Spec_z + Relax_z);
286 amrex::Real denom = amrex::Real(Relax_z - 1);
287 amrex::Real SpecExp = -std::log(0.1) / amrex::Real(width - Spec_z);
288 amrex::ParallelFor(bx_xlo, num_var, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) noexcept
291 int j_lo = std::min(j-dom_lo.y,width-1);
292 int j_hi = std::min(dom_hi.y-j,width-1);
293 int jj = std::min(j_lo,j_hi);
294 int n_ind = std::min(i-dom_lo.x,jj) + 1;
296 amrex::Real Factor = (num - amrex::Real(n_ind))/denom
297 * std::exp(-SpecExp * amrex::Real(n_ind - Spec_z));
298 amrex::Real d = data_arr(i ,j ,k ,n+icomp);
299 amrex::Real d_ip1 = data_arr(i+1,j ,k ,n+icomp);
300 amrex::Real d_im1 = data_arr(i-1,j ,k ,n+icomp);
301 amrex::Real d_jp1 = data_arr(i ,j+1,k ,n+icomp);
302 amrex::Real d_jm1 = data_arr(i ,j-1,k ,n+icomp);
303 amrex::Real delta = arr_xlo(i ,j ,k,n) - d;
304 amrex::Real delta_xp = arr_xlo(i+1,j ,k,n) - d_ip1;
305 amrex::Real delta_xm = arr_xlo(i-1,j ,k,n) - d_im1;
306 amrex::Real delta_yp = arr_xlo(i ,j+1,k,n) - d_jp1;
307 amrex::Real delta_ym = arr_xlo(i ,j-1,k,n) - d_jm1;
308 amrex::Real Laplacian = delta_xp + delta_xm + delta_yp + delta_ym - 4.0*delta;
309 amrex::Real Temp = (F1*delta - F2*Laplacian) * Factor;
310 rhs_arr(i,j,k,n+icomp) += Temp;
312 bx_xhi, num_var, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) noexcept
315 int j_lo = std::min(j-dom_lo.y,width-1);
316 int j_hi = std::min(dom_hi.y-j,width-1);
317 int jj = std::min(j_lo,j_hi);
318 int n_ind = std::min(dom_hi.x-i,jj) + 1;
320 amrex::Real Factor = (num - amrex::Real(n_ind))/denom
321 * std::exp(-SpecExp * amrex::Real(n_ind - Spec_z));
322 amrex::Real d = data_arr(i ,j ,k ,n+icomp);
323 amrex::Real d_ip1 = data_arr(i+1,j ,k ,n+icomp);
324 amrex::Real d_im1 = data_arr(i-1,j ,k ,n+icomp);
325 amrex::Real d_jp1 = data_arr(i ,j+1,k ,n+icomp);
326 amrex::Real d_jm1 = data_arr(i ,j-1,k ,n+icomp);
327 amrex::Real delta = arr_xhi(i ,j ,k,n) - d;
328 amrex::Real delta_xp = arr_xhi(i+1,j ,k,n) - d_ip1;
329 amrex::Real delta_xm = arr_xhi(i-1,j ,k,n) - d_im1;
330 amrex::Real delta_yp = arr_xhi(i ,j+1,k,n) - d_jp1;
331 amrex::Real delta_ym = arr_xhi(i ,j-1,k,n) - d_jm1;
332 amrex::Real Laplacian = delta_xp + delta_xm + delta_yp + delta_ym - 4.0*delta;
333 amrex::Real Temp = (F1*delta - F2*Laplacian) * Factor;
334 rhs_arr(i,j,k,n+icomp) += Temp;
337 amrex::ParallelFor(bx_ylo, num_var, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) noexcept
340 int n_ind = j - dom_lo.y + 1;
342 amrex::Real Factor = (num - amrex::Real(n_ind))/denom
343 * std::exp(-SpecExp * amrex::Real(n_ind - Spec_z));
344 amrex::Real d = data_arr(i ,j ,k ,n+icomp);
345 amrex::Real d_ip1 = data_arr(i+1,j ,k ,n+icomp);
346 amrex::Real d_im1 = data_arr(i-1,j ,k ,n+icomp);
347 amrex::Real d_jp1 = data_arr(i ,j+1,k ,n+icomp);
348 amrex::Real d_jm1 = data_arr(i ,j-1,k ,n+icomp);
349 amrex::Real delta = arr_ylo(i ,j ,k,n) - d;
350 amrex::Real delta_xp = arr_ylo(i+1,j ,k,n) - d_ip1;
351 amrex::Real delta_xm = arr_ylo(i-1,j ,k,n) - d_im1;
352 amrex::Real delta_yp = arr_ylo(i ,j+1,k,n) - d_jp1;
353 amrex::Real delta_ym = arr_ylo(i ,j-1,k,n) - d_jm1;
354 amrex::Real Laplacian = delta_xp + delta_xm + delta_yp + delta_ym - 4.0*delta;
355 amrex::Real Temp = (F1*delta - F2*Laplacian) * Factor;
356 rhs_arr(i,j,k,n+icomp) += Temp;
358 bx_yhi, num_var, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k,
int n) noexcept
361 int n_ind = dom_hi.y - j + 1;
363 amrex::Real Factor = (num - amrex::Real(n_ind))/denom
364 * std::exp(-SpecExp * amrex::Real(n_ind - Spec_z));
365 amrex::Real d = data_arr(i ,j ,k ,n+icomp);
366 amrex::Real d_ip1 = data_arr(i+1,j ,k ,n+icomp);
367 amrex::Real d_im1 = data_arr(i-1,j ,k ,n+icomp);
368 amrex::Real d_jp1 = data_arr(i ,j+1,k ,n+icomp);
369 amrex::Real d_jm1 = data_arr(i ,j-1,k ,n+icomp);
370 amrex::Real delta = arr_yhi(i ,j ,k,n) - d;
371 amrex::Real delta_xp = arr_yhi(i+1,j ,k,n) - d_ip1;
372 amrex::Real delta_xm = arr_yhi(i-1,j ,k,n) - d_im1;
373 amrex::Real delta_yp = arr_yhi(i ,j+1,k,n) - d_jp1;
374 amrex::Real delta_ym = arr_yhi(i ,j-1,k,n) - d_jm1;
375 amrex::Real Laplacian = delta_xp + delta_xm + delta_yp + delta_ym - 4.0*delta;
376 amrex::Real Temp = (F1*delta - F2*Laplacian) * Factor;
377 rhs_arr(i,j,k,n+icomp) += Temp;
388 const amrex::iMultiFab& imask,
389 const int nghost = 0)
391 for (amrex::MFIter mfi(dst,amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi)
393 const amrex::Box& bx = mfi.growntilebox(nghost);
396 auto dstFab = dst.array(mfi);
397 const auto maskFab = imask.const_array(mfi);
398 ParallelFor(bx, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) noexcept
400 dstFab(i,j,k) *= maskFab(i,j,k);
410 const amrex::iMultiFab& imask,
411 const int nghost = 0)
413 for (amrex::MFIter mfi(dst,amrex::TilingIfNotGPU()); mfi.isValid(); ++mfi)
415 const amrex::Box& bx = mfi.growntilebox(nghost);
418 auto dstFab = dst.array(mfi);
419 const auto maskFab = imask.const_array(mfi);
420 ParallelFor(bx, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k) noexcept
422 dstFab(i,j,k) *= (1-maskFab(i,j,k));
430 amrex::Vector<amrex::IntVect>& xfaces,
431 amrex::Vector<amrex::IntVect>& yfaces,
432 amrex::Vector<amrex::IntVect>& zfaces,
433 const amrex::Geometry& geomdata,
434 std::unique_ptr<amrex::MultiFab>& z_phys_cc);
AMREX_GPU_HOST AMREX_FORCE_INLINE void realbdy_set_rhs_in_spec_region(const amrex::Real &dt, const int &icomp, const int &num_var, const int &width, const int &set_width_x, const int &set_width_y, const amrex::Dim3 &dom_lo, const amrex::Dim3 &dom_hi, const amrex::Box &bx_xlo, const amrex::Box &bx_xhi, const amrex::Box &bx_ylo, const amrex::Box &bx_yhi, const amrex::Array4< const amrex::Real > &arr_xlo, const amrex::Array4< const amrex::Real > &arr_xhi, const amrex::Array4< const amrex::Real > &arr_ylo, const amrex::Array4< const amrex::Real > &arr_yhi, const amrex::Array4< const amrex::Real > &data_arr, const amrex::Array4< amrex::Real > &rhs_arr)
Definition: ERF_Utils.H:169
void MomentumToVelocity(amrex::MultiFab &xvel_out, amrex::MultiFab &yvel_out, amrex::MultiFab &zvel_out, const amrex::MultiFab &cons_in, const amrex::MultiFab &xmom_in, const amrex::MultiFab &ymom_in, const amrex::MultiFab &zmom_in, const amrex::Box &domain, const amrex::Vector< amrex::BCRec > &domain_bcs_type_h)
void VelocityToMomentum(const amrex::MultiFab &xvel_in, const amrex::IntVect &xvel_ngrow, const amrex::MultiFab &yvel_in, const amrex::IntVect &yvel_ngrow, const amrex::MultiFab &zvel_in, const amrex::IntVect &zvel_ngrow, const amrex::MultiFab &cons_in, amrex::MultiFab &xmom_out, amrex::MultiFab &ymom_out, amrex::MultiFab &zmom_out, const amrex::Box &domain, const amrex::Vector< amrex::BCRec > &domain_bcs_type_h)
void fine_compute_interior_ghost_rhs(const amrex::Real &time, const amrex::Real &delta_t, const int &width, const int &set_width, const amrex::Geometry &geom, ERFFillPatcher *FPr_c, ERFFillPatcher *FPr_u, ERFFillPatcher *FPr_v, ERFFillPatcher *FPr_w, amrex::Vector< amrex::BCRec > &domain_bcs_type, amrex::Vector< amrex::MultiFab > &S_rhs_f, amrex::Vector< amrex::MultiFab > &S_data_f)
void thinbody_wall_dist(std::unique_ptr< amrex::MultiFab > &wdist, amrex::Vector< amrex::IntVect > &xfaces, amrex::Vector< amrex::IntVect > &yfaces, amrex::Vector< amrex::IntVect > &zfaces, const amrex::Geometry &geomdata, std::unique_ptr< amrex::MultiFab > &z_phys_cc)
void realbdy_interior_bxs_xy(const amrex::Box &bx, const amrex::Box &domain, const int &width, amrex::Box &bx_xlo, amrex::Box &bx_xhi, amrex::Box &bx_ylo, amrex::Box &bx_yhi, const int &set_width=0, const amrex::IntVect &ng_vect=amrex::IntVect(0, 0, 0), const bool get_int_ng=false)
amrex::BoxArray ERFPostProcessBaseGrids(const amrex::Box &domain, bool decompose_in_z)
void make_zcc(const amrex::Geometry &geom, amrex::MultiFab &z_phys_nd, amrex::MultiFab &z_phys_cc)
void realbdy_compute_interior_ghost_rhs(const amrex::Real &bdy_time_interval, const amrex::Real &start_bdy_time, const amrex::Real &time, const amrex::Real &delta_t, int width, int set_width, const amrex::Geometry &geom, amrex::Vector< amrex::MultiFab > &S_rhs, amrex::Vector< amrex::MultiFab > &S_old_data, amrex::Vector< amrex::MultiFab > &S_cur_data, amrex::Vector< amrex::Vector< amrex::FArrayBox >> &bdy_data_xlo, amrex::Vector< amrex::Vector< amrex::FArrayBox >> &bdy_data_xhi, amrex::Vector< amrex::Vector< amrex::FArrayBox >> &bdy_data_ylo, amrex::Vector< amrex::Vector< amrex::FArrayBox >> &bdy_data_yhi)
void realbdy_bc_bxs_xy(const amrex::Box &bx, const amrex::Box &domain, const int &set_width, amrex::Box &bx_xlo, amrex::Box &bx_xhi, amrex::Box &bx_ylo, amrex::Box &bx_yhi, const amrex::IntVect &ng_vect=amrex::IntVect(0, 0, 0))
void Time_Avg_Vel_atCC(const amrex::Real &dt, amrex::Real &t_avg_cnt, amrex::MultiFab *vel_t_avg, amrex::MultiFab &xvel, amrex::MultiFab &yvel, amrex::MultiFab &zvel)
AMREX_GPU_HOST AMREX_FORCE_INLINE void ApplyInvertedMask(amrex::MultiFab &dst, const amrex::iMultiFab &imask, const int nghost=0)
Definition: ERF_Utils.H:409
void make_J(const amrex::Geometry &geom, amrex::MultiFab &z_phys_nd, amrex::MultiFab &detJ_cc)
AMREX_GPU_HOST AMREX_FORCE_INLINE void ApplyMask(amrex::MultiFab &dst, const amrex::iMultiFab &imask, const int nghost=0)
Definition: ERF_Utils.H:387
void ConvertForProjection(const amrex::MultiFab &den_div, const amrex::MultiFab &den_mlt, amrex::MultiFab &xmom, amrex::MultiFab &ymom, amrex::MultiFab &zmom, const amrex::Box &domain, const amrex::Vector< amrex::BCRec > &domain_bcs_type_h)
void ChopGrids2D(amrex::BoxArray &ba, const amrex::Box &domain, int target_size)
AMREX_GPU_HOST AMREX_FORCE_INLINE void realbdy_compute_laplacian_relaxation(const int &icomp, const int &num_var, const int &width, const int &set_width, const amrex::Dim3 &dom_lo, const amrex::Dim3 &dom_hi, const amrex::Real &F1, const amrex::Real &F2, const amrex::Box &bx_xlo, const amrex::Box &bx_xhi, const amrex::Box &bx_ylo, const amrex::Box &bx_yhi, const amrex::Array4< const amrex::Real > &arr_xlo, const amrex::Array4< const amrex::Real > &arr_xhi, const amrex::Array4< const amrex::Real > &arr_ylo, const amrex::Array4< const amrex::Real > &arr_yhi, const amrex::Array4< const amrex::Real > &data_arr, const amrex::Array4< amrex::Real > &rhs_arr)
Definition: ERF_Utils.H:263
void make_areas(const amrex::Geometry &geom, amrex::MultiFab &z_phys_nd, amrex::MultiFab &ax, amrex::MultiFab &ay, amrex::MultiFab &az)
Definition: ERF_FillPatcher.H:9
@ ymom
Definition: ERF_IndexDefines.H:152
@ zmom
Definition: ERF_IndexDefines.H:153
@ xmom
Definition: ERF_IndexDefines.H:151
@ xvel
Definition: ERF_IndexDefines.H:141
@ zvel
Definition: ERF_IndexDefines.H:143
@ yvel
Definition: ERF_IndexDefines.H:142