ERF
Energy Research and Forecasting: An Atmospheric Modeling Code
ERF_Diffusion.H
Go to the documentation of this file.
1 #ifndef ERF_DIFFUSION_H_
2 #define ERF_DIFFUSION_H_
3 
4 #include "AMReX.H"
5 #include "AMReX_MultiFab.H"
6 #include "AMReX_BCRec.H"
7 
8 #include "ERF_DataStruct.H"
9 #include "ERF_IndexDefines.H"
10 #include "ERF_SurfaceLayer.H"
11 #include "ERF_EB.H"
12 
13 void DiffusionSrcForMom (const amrex::Box& bxx, const amrex::Box& bxy, const amrex::Box& bxz,
14  const amrex::Array4< amrex::Real>& rho_u_rhs,
15  const amrex::Array4< amrex::Real>& rho_v_rhs,
16  const amrex::Array4< amrex::Real>& rho_w_rhs,
17  const amrex::Array4<const amrex::Real>& tau11 ,
18  const amrex::Array4<const amrex::Real>& tau22 ,
19  const amrex::Array4<const amrex::Real>& tau33 ,
20  const amrex::Array4<const amrex::Real>& tau12 , const amrex::Array4<const amrex::Real>& tau21,
21  const amrex::Array4<const amrex::Real>& tau13 , const amrex::Array4<const amrex::Real>& tau31,
22  const amrex::Array4<const amrex::Real>& tau23 , const amrex::Array4<const amrex::Real>& tau32,
23  const amrex::Array4<const amrex::Real>& detJ_arr ,
24  const amrex::Gpu::DeviceVector<amrex::Real>& stretched_dz_d,
25  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& dxInv,
26  const amrex::Array4<const amrex::Real>& mf_mx ,
27  const amrex::Array4<const amrex::Real>& mf_ux ,
28  const amrex::Array4<const amrex::Real>& mf_vx ,
29  const amrex::Array4<const amrex::Real>& mf_my ,
30  const amrex::Array4<const amrex::Real>& mf_uy ,
31  const amrex::Array4<const amrex::Real>& mf_vy ,
32  const bool use_stretched_dz, const bool use_variable_dz);
33 
34 void DiffusionSrcForMom_EB (const amrex::MFIter& mfi,
35  [[maybe_unused]] const amrex::Box& domain,
36  const amrex::Box& bxx, const amrex::Box& bxy, const amrex::Box& bxz,
37  const amrex::Array4< amrex::Real>& rho_u_rhs,
38  const amrex::Array4< amrex::Real>& rho_v_rhs,
39  const amrex::Array4< amrex::Real>& rho_w_rhs,
40  const amrex::Array4<const amrex::Real>& u_arr ,
41  const amrex::Array4<const amrex::Real>& v_arr ,
42  const amrex::Array4<const amrex::Real>& w_arr ,
43  const amrex::Array4<const amrex::Real>& tau11 ,
44  const amrex::Array4<const amrex::Real>& tau22 ,
45  const amrex::Array4<const amrex::Real>& tau33 ,
46  const amrex::Array4<const amrex::Real>& tau12 ,
47  const amrex::Array4<const amrex::Real>& tau13 ,
48  const amrex::Array4<const amrex::Real>& tau23 ,
49  const amrex::Real* dx_arr,
50  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& dxInv,
51  const amrex::Array4<const amrex::Real>& mf_mx ,
52  const amrex::Array4<const amrex::Real>& mf_ux ,
53  const amrex::Array4<const amrex::Real>& mf_vx ,
54  const amrex::Array4<const amrex::Real>& mf_my ,
55  const amrex::Array4<const amrex::Real>& mf_uy ,
56  const amrex::Array4<const amrex::Real>& mf_vy ,
57  const SolverChoice& solverChoice,
58  const eb_& ebfact,
59  [[maybe_unused]] const amrex::BCRec* bc_ptr);
60 
61 void DiffusionSrcForState_N (const amrex::Box& bx, const amrex::Box& domain,
62  int start_comp, int num_comp,
63  const amrex::Array4<const amrex::Real>& u,
64  const amrex::Array4<const amrex::Real>& v,
65  const amrex::Array4<const amrex::Real>& cell_data,
66  const amrex::Array4<const amrex::Real>& cell_prim,
67  const amrex::Array4<amrex::Real>& cell_rhs,
68  const amrex::Array4<amrex::Real>& xflux,
69  const amrex::Array4<amrex::Real>& yflux,
70  const amrex::Array4<amrex::Real>& zflux,
71  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& cellSizeInv,
72  const amrex::Array4<const amrex::Real>& SmnSmn_a,
73  const amrex::Array4<const amrex::Real>& mf_mx,
74  const amrex::Array4<const amrex::Real>& mf_ux,
75  const amrex::Array4<const amrex::Real>& mf_vx,
76  const amrex::Array4<const amrex::Real>& mf_my,
77  const amrex::Array4<const amrex::Real>& mf_uy,
78  const amrex::Array4<const amrex::Real>& mf_vy,
79  amrex::Array4< amrex::Real>& hfx_z,
80  amrex::Array4< amrex::Real>& qfx1_z,
81  amrex::Array4< amrex::Real>& qfx2_z,
82  amrex::Array4< amrex::Real>& diss,
83  const amrex::Array4<const amrex::Real>& mu_turb,
84  const SolverChoice& solverChoice,
85  const int level,
86  const amrex::Array4<const amrex::Real>& tm_arr,
87  const amrex::GpuArray<amrex::Real,AMREX_SPACEDIM> grav_gpu,
88  const amrex::BCRec* bc_ptr,
89  const bool use_SurfLayer,
90  const amrex::Real implicit_fac);
91 
92 void DiffusionSrcForState_S (const amrex::Box& bx, const amrex::Box& domain,
93  int start_comp, int num_comp,
94  const amrex::Array4<const amrex::Real>& u,
95  const amrex::Array4<const amrex::Real>& v,
96  const amrex::Array4<const amrex::Real>& cell_data,
97  const amrex::Array4<const amrex::Real>& cell_prim,
98  const amrex::Array4<amrex::Real>& cell_rhs,
99  const amrex::Array4<amrex::Real>& xflux,
100  const amrex::Array4<amrex::Real>& yflux,
101  const amrex::Array4<amrex::Real>& zflux,
102  const amrex::Gpu::DeviceVector<amrex::Real>& stretched_dz_d,
103  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& dxInv,
104  const amrex::Array4<const amrex::Real>& SmnSmn_a,
105  const amrex::Array4<const amrex::Real>& mf_mx,
106  const amrex::Array4<const amrex::Real>& mf_ux,
107  const amrex::Array4<const amrex::Real>& mf_vx,
108  const amrex::Array4<const amrex::Real>& mf_my,
109  const amrex::Array4<const amrex::Real>& mf_uy,
110  const amrex::Array4<const amrex::Real>& mf_vy,
111  amrex::Array4< amrex::Real>& hfx_z,
112  amrex::Array4< amrex::Real>& qfx1_z,
113  amrex::Array4< amrex::Real>& qfx2_z,
114  amrex::Array4< amrex::Real>& diss,
115  const amrex::Array4<const amrex::Real>& mu_turb,
116  const SolverChoice& solverChoice,
117  const int level,
118  const amrex::Array4<const amrex::Real>& tm_arr,
119  const amrex::GpuArray<amrex::Real,AMREX_SPACEDIM> grav_gpu,
120  const amrex::BCRec* bc_ptr,
121  const bool use_SurfLayer,
122  const amrex::Real implicit_fac);
123 
124 void DiffusionSrcForState_T (const amrex::Box& bx, const amrex::Box& domain,
125  int start_comp, int num_comp,
126  const bool& rotate,
127  const amrex::Array4<const amrex::Real>& u,
128  const amrex::Array4<const amrex::Real>& v,
129  const amrex::Array4<const amrex::Real>& cell_data,
130  const amrex::Array4<const amrex::Real>& cell_prim,
131  const amrex::Array4<amrex::Real>& cell_rhs,
132  const amrex::Array4<amrex::Real>& xflux,
133  const amrex::Array4<amrex::Real>& yflux,
134  const amrex::Array4<amrex::Real>& zflux,
135  const amrex::Array4<const amrex::Real>& z_nd,
136  const amrex::Array4<const amrex::Real>& z_cc,
137  const amrex::Array4<const amrex::Real>& ax,
138  const amrex::Array4<const amrex::Real>& ay,
139  const amrex::Array4<const amrex::Real>& az,
140  const amrex::Array4<const amrex::Real>& detJ,
141  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& dxInv,
142  const amrex::Array4<const amrex::Real>& SmnSmn_a,
143  const amrex::Array4<const amrex::Real>& mf_mx,
144  const amrex::Array4<const amrex::Real>& mf_ux,
145  const amrex::Array4<const amrex::Real>& mf_vx,
146  const amrex::Array4<const amrex::Real>& mf_my,
147  const amrex::Array4<const amrex::Real>& mf_uy,
148  const amrex::Array4<const amrex::Real>& mf_vy,
149  amrex::Array4< amrex::Real>& hfx_x,
150  amrex::Array4< amrex::Real>& hfx_y,
151  amrex::Array4< amrex::Real>& hfx_z,
152  amrex::Array4< amrex::Real>& qfx1_x,
153  amrex::Array4< amrex::Real>& qfx1_y,
154  amrex::Array4< amrex::Real>& qfx1_z,
155  amrex::Array4< amrex::Real>& qfx2_z,
156  amrex::Array4< amrex::Real>& diss,
157  const amrex::Array4<const amrex::Real>& mu_turb,
158  const SolverChoice& solverChoice,
159  const int level,
160  const amrex::Array4<const amrex::Real>& tm_arr,
161  const amrex::GpuArray<amrex::Real,AMREX_SPACEDIM> grav_gpu,
162  const amrex::BCRec* bc_ptr,
163  const bool use_SurfLayer,
164  const amrex::Real implicit_fac);
165 
166 void ComputeStressConsVisc_N (amrex::Box bxcc,
167  amrex::Box tbxxy,
168  amrex::Box tbxxz,
169  amrex::Box tbxyz,
170  amrex::Real mu_eff,
171  const amrex::Array4<const amrex::Real>& cell_data,
172  amrex::Array4<amrex::Real>& tau11,
173  amrex::Array4<amrex::Real>& tau22,
174  amrex::Array4<amrex::Real>& tau33,
175  amrex::Array4<amrex::Real>& tau12,
176  amrex::Array4<amrex::Real>& tau13,
177  amrex::Array4<amrex::Real>& tau23,
178  const amrex::Array4<const amrex::Real>& er_arr);
179 
180 void ComputeStressConsVisc_S (amrex::Box bxcc,
181  amrex::Box tbxxy,
182  amrex::Box tbxxz,
183  amrex::Box tbxyz,
184  amrex::Real mu_eff,
185  const amrex::Array4<const amrex::Real>& cell_data,
186  amrex::Array4<amrex::Real>& tau11,
187  amrex::Array4<amrex::Real>& tau22,
188  amrex::Array4<amrex::Real>& tau33,
189  amrex::Array4<amrex::Real>& tau12, amrex::Array4<amrex::Real>& tau21,
190  amrex::Array4<amrex::Real>& tau13, amrex::Array4<amrex::Real>& tau31,
191  amrex::Array4<amrex::Real>& tau23, amrex::Array4<amrex::Real>& tau32,
192  const amrex::Array4<const amrex::Real>& er_arr,
193  const amrex::Array4<const amrex::Real>& mf_mx,
194  const amrex::Array4<const amrex::Real>& mf_ux,
195  const amrex::Array4<const amrex::Real>& mf_vx,
196  const amrex::Array4<const amrex::Real>& mf_my,
197  const amrex::Array4<const amrex::Real>& mf_uy,
198  const amrex::Array4<const amrex::Real>& mf_vy);
199 
200 void ComputeStressConsVisc_T (amrex::Box bxcc,
201  amrex::Box tbxxy,
202  amrex::Box tbxxz,
203  amrex::Box tbxyz,
204  amrex::Real mu_eff,
205  const amrex::Array4<const amrex::Real>& cell_data,
206  amrex::Array4<amrex::Real>& tau11,
207  amrex::Array4<amrex::Real>& tau22,
208  amrex::Array4<amrex::Real>& tau33,
209  amrex::Array4<amrex::Real>& tau12, amrex::Array4<amrex::Real>& tau21,
210  amrex::Array4<amrex::Real>& tau13, amrex::Array4<amrex::Real>& tau31,
211  amrex::Array4<amrex::Real>& tau23, amrex::Array4<amrex::Real>& tau32,
212  const amrex::Array4<const amrex::Real>& er_arr,
213  const amrex::Array4<const amrex::Real>& z_nd,
214  const amrex::Array4<const amrex::Real>& detJ,
215  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& dxInv,
216  const amrex::Array4<const amrex::Real>& mf_mx,
217  const amrex::Array4<const amrex::Real>& mf_ux,
218  const amrex::Array4<const amrex::Real>& mf_vx,
219  const amrex::Array4<const amrex::Real>& mf_my,
220  const amrex::Array4<const amrex::Real>& mf_uy,
221  const amrex::Array4<const amrex::Real>& mf_vy);
222 
223 
224 
225 void ComputeStressVarVisc_N (amrex::Box bxcc,
226  amrex::Box tbxxy,
227  amrex::Box tbxxz,
228  amrex::Box tbxyz,
229  amrex::Real mu_eff,
230  const amrex::Array4<const amrex::Real>& mu_turb,
231  const amrex::Array4<const amrex::Real>& cell_data,
232  amrex::Array4<amrex::Real>& tau11,
233  amrex::Array4<amrex::Real>& tau22,
234  amrex::Array4<amrex::Real>& tau33,
235  amrex::Array4<amrex::Real>& tau12,
236  amrex::Array4<amrex::Real>& tau13,
237  amrex::Array4<amrex::Real>& tau23,
238  const amrex::Array4<const amrex::Real>& er_arr);
239 
240 void ComputeStressVarVisc_S (amrex::Box bxcc,
241  amrex::Box tbxxy,
242  amrex::Box tbxxz,
243  amrex::Box tbxyz,
244  amrex::Real mu_eff,
245  const amrex::Array4<const amrex::Real>& mu_turb,
246  const amrex::Array4<const amrex::Real>& cell_data,
247  amrex::Array4<amrex::Real>& tau11,
248  amrex::Array4<amrex::Real>& tau22,
249  amrex::Array4<amrex::Real>& tau33,
250  amrex::Array4<amrex::Real>& tau12, amrex::Array4<amrex::Real>& tau21,
251  amrex::Array4<amrex::Real>& tau13, amrex::Array4<amrex::Real>& tau31,
252  amrex::Array4<amrex::Real>& tau23, amrex::Array4<amrex::Real>& tau32,
253  const amrex::Array4<const amrex::Real>& er_arr,
254  const amrex::Array4<const amrex::Real>& mf_mx,
255  const amrex::Array4<const amrex::Real>& mf_ux,
256  const amrex::Array4<const amrex::Real>& mf_vx,
257  const amrex::Array4<const amrex::Real>& mf_my,
258  const amrex::Array4<const amrex::Real>& mf_uy,
259  const amrex::Array4<const amrex::Real>& mf_vy);
260 
261 void ComputeStressVarVisc_T (amrex::Box bxcc,
262  amrex::Box tbxxy,
263  amrex::Box tbxxz,
264  amrex::Box tbxyz,
265  amrex::Real mu_eff,
266  const amrex::Array4<const amrex::Real>& mu_turb,
267  const amrex::Array4<const amrex::Real>& cell_data,
268  amrex::Array4<amrex::Real>& tau11,
269  amrex::Array4<amrex::Real>& tau22,
270  amrex::Array4<amrex::Real>& tau33,
271  amrex::Array4<amrex::Real>& tau12, amrex::Array4<amrex::Real>& tau21,
272  amrex::Array4<amrex::Real>& tau13, amrex::Array4<amrex::Real>& tau31,
273  amrex::Array4<amrex::Real>& tau23, amrex::Array4<amrex::Real>& tau32,
274  const amrex::Array4<const amrex::Real>& er_arr,
275  const amrex::Array4<const amrex::Real>& z_nd,
276  const amrex::Array4<const amrex::Real>& detJ,
277  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& dxInv,
278  const amrex::Array4<const amrex::Real>& mf_mx,
279  const amrex::Array4<const amrex::Real>& mf_ux,
280  const amrex::Array4<const amrex::Real>& mf_vx,
281  const amrex::Array4<const amrex::Real>& mf_my,
282  const amrex::Array4<const amrex::Real>& mf_uy,
283  const amrex::Array4<const amrex::Real>& mf_vy);
284 
285 
286 
287 void ComputeStrain_N (amrex::Box bx, amrex::Box bxcc, amrex::Box tbxxy,
288  amrex::Box tbxxz, amrex::Box tbxyz, amrex::Box domain,
289  const amrex::Array4<const amrex::Real>& u,
290  const amrex::Array4<const amrex::Real>& v,
291  const amrex::Array4<const amrex::Real>& w,
292  amrex::Array4<amrex::Real>& tau11,
293  amrex::Array4<amrex::Real>& tau22,
294  amrex::Array4<amrex::Real>& tau33,
295  amrex::Array4<amrex::Real>& tau12,
296  amrex::Array4<amrex::Real>& tau13, amrex::Array4<amrex::Real>& tau31,
297  amrex::Array4<amrex::Real>& tau23, amrex::Array4<amrex::Real>& tau32,
298  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& dxInv,
299  const amrex::Array4<const amrex::Real>& mf_mx,
300  const amrex::Array4<const amrex::Real>& mf_ux,
301  const amrex::Array4<const amrex::Real>& mf_vx,
302  const amrex::Array4<const amrex::Real>& mf_my,
303  const amrex::Array4<const amrex::Real>& mf_uy,
304  const amrex::Array4<const amrex::Real>& mf_vy,
305  const amrex::BCRec* bc_ptr,
306  amrex::Array4<amrex::Real>& SmnSmn_a,
307  const amrex::Real implicit_fac);
308 
309 void ComputeStrain_S (amrex::Box bx, amrex::Box bxcc, amrex::Box tbxxy,
310  amrex::Box tbxxz, amrex::Box tbxyz, amrex::Box domain,
311  const amrex::Array4<const amrex::Real>& u,
312  const amrex::Array4<const amrex::Real>& v,
313  const amrex::Array4<const amrex::Real>& w,
314  amrex::Array4<amrex::Real>& tau11,
315  amrex::Array4<amrex::Real>& tau22,
316  amrex::Array4<amrex::Real>& tau33,
317  amrex::Array4<amrex::Real>& tau12, amrex::Array4<amrex::Real>& tau21,
318  amrex::Array4<amrex::Real>& tau13, amrex::Array4<amrex::Real>& tau31,
319  amrex::Array4<amrex::Real>& tau23, amrex::Array4<amrex::Real>& tau32,
320  const amrex::Gpu::DeviceVector<amrex::Real>& stretched_dz_d,
321  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& dxInv,
322  const amrex::Array4<const amrex::Real>& mf_mx,
323  const amrex::Array4<const amrex::Real>& mf_ux,
324  const amrex::Array4<const amrex::Real>& mf_vx,
325  const amrex::Array4<const amrex::Real>& mf_my,
326  const amrex::Array4<const amrex::Real>& mf_uy,
327  const amrex::Array4<const amrex::Real>& mf_vy,
328  const amrex::BCRec* bc_ptr,
329  amrex::Array4<amrex::Real>& SmnSmn_a,
330  const amrex::Real implicit_fac);
331 
332 void ComputeStrain_T (amrex::Box bx, amrex::Box bxcc, amrex::Box tbxxy,
333  amrex::Box tbxxz, amrex::Box tbxyz, amrex::Box domain,
334  const amrex::Array4<const amrex::Real>& u,
335  const amrex::Array4<const amrex::Real>& v,
336  const amrex::Array4<const amrex::Real>& w,
337  amrex::Array4<amrex::Real>& tau11,
338  amrex::Array4<amrex::Real>& tau22,
339  amrex::Array4<amrex::Real>& tau33,
340  amrex::Array4<amrex::Real>& tau12, amrex::Array4<amrex::Real>& tau21,
341  amrex::Array4<amrex::Real>& tau13, amrex::Array4<amrex::Real>& tau31,
342  amrex::Array4<amrex::Real>& tau23, amrex::Array4<amrex::Real>& tau32,
343  const amrex::Array4<const amrex::Real>& z_nd,
344  const amrex::Array4<const amrex::Real>& detJ,
345  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& dxInv,
346  const amrex::Array4<const amrex::Real>& mf_mx,
347  const amrex::Array4<const amrex::Real>& mf_ux,
348  const amrex::Array4<const amrex::Real>& mf_vx,
349  const amrex::Array4<const amrex::Real>& mf_my,
350  const amrex::Array4<const amrex::Real>& mf_uy,
351  const amrex::Array4<const amrex::Real>& mf_vy,
352  const amrex::BCRec* bc_ptr,
353  amrex::Array4<amrex::Real>& SmnSmn_a,
354  const amrex::Real implicit_fac);
355 
356 void ImplicitDiffForState_N (const amrex::Box& bx, const amrex::Box& domain, const int level,
357  const amrex::Real dt,
358  /*int start_comp, int num_comp,*/
359  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM*2>& bc_neumann_vals,
360  const amrex::Array4< amrex::Real>& cell_data,
361  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& cellSizeInv,
362  const amrex::Array4< amrex::Real>& hfx_z,
363  const amrex::Array4<const amrex::Real>& mu_turb,
364  const SolverChoice& solverChoice,
365  const amrex::BCRec* bc_ptr,
366  const bool use_SurfLayer,
367  const amrex::Real implicit_fac);
368 
369 void ImplicitDiffForState_S (const amrex::Box& bx, const amrex::Box& domain, const int level,
370  const amrex::Real dt,
371  /*int start_comp, int num_comp,*/
372  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM*2>& bc_neumann_vals,
373  const amrex::Array4< amrex::Real>& cell_data,
374  const amrex::Gpu::DeviceVector<amrex::Real>& stretched_dz_d,
375  const amrex::Array4< amrex::Real>& hfx_z,
376  const amrex::Array4<const amrex::Real>& mu_turb,
377  const SolverChoice& solverChoice,
378  const amrex::BCRec* bc_ptr,
379  const bool use_SurfLayer,
380  const amrex::Real implicit_fac);
381 
382 void ImplicitDiffForState_T (const amrex::Box& bx, const amrex::Box& domain, const int level,
383  const amrex::Real dt,
384  /*int start_comp, int num_comp,*/
385  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM*2>& bc_neumann_vals,
386  const amrex::Array4< amrex::Real>& cell_data,
387  const amrex::Array4<const amrex::Real>& z_nd,
388  const amrex::Array4<const amrex::Real>& detJ,
389  const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM>& cellSizeInv,
390  const amrex::Array4< amrex::Real>& hfx_z,
391  const amrex::Array4<const amrex::Real>& mu_turb,
392  const SolverChoice& solverChoice,
393  const amrex::BCRec* bc_ptr,
394  const bool use_SurfLayer,
395  const amrex::Real implicit_fac);
396 #endif
@ tau12
Definition: ERF_DataStruct.H:30
@ tau23
Definition: ERF_DataStruct.H:30
@ tau33
Definition: ERF_DataStruct.H:30
@ tau22
Definition: ERF_DataStruct.H:30
@ tau11
Definition: ERF_DataStruct.H:30
@ tau32
Definition: ERF_DataStruct.H:30
@ tau31
Definition: ERF_DataStruct.H:30
@ tau21
Definition: ERF_DataStruct.H:30
@ tau13
Definition: ERF_DataStruct.H:30
void ComputeStressConsVisc_N(amrex::Box bxcc, amrex::Box tbxxy, amrex::Box tbxxz, amrex::Box tbxyz, amrex::Real mu_eff, const amrex::Array4< const amrex::Real > &cell_data, amrex::Array4< amrex::Real > &tau11, amrex::Array4< amrex::Real > &tau22, amrex::Array4< amrex::Real > &tau33, amrex::Array4< amrex::Real > &tau12, amrex::Array4< amrex::Real > &tau13, amrex::Array4< amrex::Real > &tau23, const amrex::Array4< const amrex::Real > &er_arr)
void DiffusionSrcForMom(const amrex::Box &bxx, const amrex::Box &bxy, const amrex::Box &bxz, const amrex::Array4< amrex::Real > &rho_u_rhs, const amrex::Array4< amrex::Real > &rho_v_rhs, const amrex::Array4< amrex::Real > &rho_w_rhs, const amrex::Array4< const amrex::Real > &tau11, const amrex::Array4< const amrex::Real > &tau22, const amrex::Array4< const amrex::Real > &tau33, const amrex::Array4< const amrex::Real > &tau12, const amrex::Array4< const amrex::Real > &tau21, const amrex::Array4< const amrex::Real > &tau13, const amrex::Array4< const amrex::Real > &tau31, const amrex::Array4< const amrex::Real > &tau23, const amrex::Array4< const amrex::Real > &tau32, const amrex::Array4< const amrex::Real > &detJ_arr, const amrex::Gpu::DeviceVector< amrex::Real > &stretched_dz_d, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy, const bool use_stretched_dz, const bool use_variable_dz)
void DiffusionSrcForMom_EB(const amrex::MFIter &mfi, [[maybe_unused]] const amrex::Box &domain, const amrex::Box &bxx, const amrex::Box &bxy, const amrex::Box &bxz, const amrex::Array4< amrex::Real > &rho_u_rhs, const amrex::Array4< amrex::Real > &rho_v_rhs, const amrex::Array4< amrex::Real > &rho_w_rhs, const amrex::Array4< const amrex::Real > &u_arr, const amrex::Array4< const amrex::Real > &v_arr, const amrex::Array4< const amrex::Real > &w_arr, const amrex::Array4< const amrex::Real > &tau11, const amrex::Array4< const amrex::Real > &tau22, const amrex::Array4< const amrex::Real > &tau33, const amrex::Array4< const amrex::Real > &tau12, const amrex::Array4< const amrex::Real > &tau13, const amrex::Array4< const amrex::Real > &tau23, const amrex::Real *dx_arr, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy, const SolverChoice &solverChoice, const eb_ &ebfact, [[maybe_unused]] const amrex::BCRec *bc_ptr)
void ComputeStrain_S(amrex::Box bx, amrex::Box bxcc, amrex::Box tbxxy, amrex::Box tbxxz, amrex::Box tbxyz, amrex::Box domain, const amrex::Array4< const amrex::Real > &u, const amrex::Array4< const amrex::Real > &v, const amrex::Array4< const amrex::Real > &w, amrex::Array4< amrex::Real > &tau11, amrex::Array4< amrex::Real > &tau22, amrex::Array4< amrex::Real > &tau33, amrex::Array4< amrex::Real > &tau12, amrex::Array4< amrex::Real > &tau21, amrex::Array4< amrex::Real > &tau13, amrex::Array4< amrex::Real > &tau31, amrex::Array4< amrex::Real > &tau23, amrex::Array4< amrex::Real > &tau32, const amrex::Gpu::DeviceVector< amrex::Real > &stretched_dz_d, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy, const amrex::BCRec *bc_ptr, amrex::Array4< amrex::Real > &SmnSmn_a, const amrex::Real implicit_fac)
void DiffusionSrcForState_S(const amrex::Box &bx, const amrex::Box &domain, int start_comp, int num_comp, const amrex::Array4< const amrex::Real > &u, const amrex::Array4< const amrex::Real > &v, const amrex::Array4< const amrex::Real > &cell_data, const amrex::Array4< const amrex::Real > &cell_prim, const amrex::Array4< amrex::Real > &cell_rhs, const amrex::Array4< amrex::Real > &xflux, const amrex::Array4< amrex::Real > &yflux, const amrex::Array4< amrex::Real > &zflux, const amrex::Gpu::DeviceVector< amrex::Real > &stretched_dz_d, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv, const amrex::Array4< const amrex::Real > &SmnSmn_a, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy, amrex::Array4< amrex::Real > &hfx_z, amrex::Array4< amrex::Real > &qfx1_z, amrex::Array4< amrex::Real > &qfx2_z, amrex::Array4< amrex::Real > &diss, const amrex::Array4< const amrex::Real > &mu_turb, const SolverChoice &solverChoice, const int level, const amrex::Array4< const amrex::Real > &tm_arr, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > grav_gpu, const amrex::BCRec *bc_ptr, const bool use_SurfLayer, const amrex::Real implicit_fac)
void DiffusionSrcForState_T(const amrex::Box &bx, const amrex::Box &domain, int start_comp, int num_comp, const bool &rotate, const amrex::Array4< const amrex::Real > &u, const amrex::Array4< const amrex::Real > &v, const amrex::Array4< const amrex::Real > &cell_data, const amrex::Array4< const amrex::Real > &cell_prim, const amrex::Array4< amrex::Real > &cell_rhs, const amrex::Array4< amrex::Real > &xflux, const amrex::Array4< amrex::Real > &yflux, const amrex::Array4< amrex::Real > &zflux, const amrex::Array4< const amrex::Real > &z_nd, const amrex::Array4< const amrex::Real > &z_cc, const amrex::Array4< const amrex::Real > &ax, const amrex::Array4< const amrex::Real > &ay, const amrex::Array4< const amrex::Real > &az, const amrex::Array4< const amrex::Real > &detJ, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv, const amrex::Array4< const amrex::Real > &SmnSmn_a, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy, amrex::Array4< amrex::Real > &hfx_x, amrex::Array4< amrex::Real > &hfx_y, amrex::Array4< amrex::Real > &hfx_z, amrex::Array4< amrex::Real > &qfx1_x, amrex::Array4< amrex::Real > &qfx1_y, amrex::Array4< amrex::Real > &qfx1_z, amrex::Array4< amrex::Real > &qfx2_z, amrex::Array4< amrex::Real > &diss, const amrex::Array4< const amrex::Real > &mu_turb, const SolverChoice &solverChoice, const int level, const amrex::Array4< const amrex::Real > &tm_arr, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > grav_gpu, const amrex::BCRec *bc_ptr, const bool use_SurfLayer, const amrex::Real implicit_fac)
void ImplicitDiffForState_S(const amrex::Box &bx, const amrex::Box &domain, const int level, const amrex::Real dt, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM *2 > &bc_neumann_vals, const amrex::Array4< amrex::Real > &cell_data, const amrex::Gpu::DeviceVector< amrex::Real > &stretched_dz_d, const amrex::Array4< amrex::Real > &hfx_z, const amrex::Array4< const amrex::Real > &mu_turb, const SolverChoice &solverChoice, const amrex::BCRec *bc_ptr, const bool use_SurfLayer, const amrex::Real implicit_fac)
void ImplicitDiffForState_N(const amrex::Box &bx, const amrex::Box &domain, const int level, const amrex::Real dt, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM *2 > &bc_neumann_vals, const amrex::Array4< amrex::Real > &cell_data, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &cellSizeInv, const amrex::Array4< amrex::Real > &hfx_z, const amrex::Array4< const amrex::Real > &mu_turb, const SolverChoice &solverChoice, const amrex::BCRec *bc_ptr, const bool use_SurfLayer, const amrex::Real implicit_fac)
void DiffusionSrcForState_N(const amrex::Box &bx, const amrex::Box &domain, int start_comp, int num_comp, const amrex::Array4< const amrex::Real > &u, const amrex::Array4< const amrex::Real > &v, const amrex::Array4< const amrex::Real > &cell_data, const amrex::Array4< const amrex::Real > &cell_prim, const amrex::Array4< amrex::Real > &cell_rhs, const amrex::Array4< amrex::Real > &xflux, const amrex::Array4< amrex::Real > &yflux, const amrex::Array4< amrex::Real > &zflux, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &cellSizeInv, const amrex::Array4< const amrex::Real > &SmnSmn_a, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy, amrex::Array4< amrex::Real > &hfx_z, amrex::Array4< amrex::Real > &qfx1_z, amrex::Array4< amrex::Real > &qfx2_z, amrex::Array4< amrex::Real > &diss, const amrex::Array4< const amrex::Real > &mu_turb, const SolverChoice &solverChoice, const int level, const amrex::Array4< const amrex::Real > &tm_arr, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > grav_gpu, const amrex::BCRec *bc_ptr, const bool use_SurfLayer, const amrex::Real implicit_fac)
void ComputeStressVarVisc_S(amrex::Box bxcc, amrex::Box tbxxy, amrex::Box tbxxz, amrex::Box tbxyz, amrex::Real mu_eff, const amrex::Array4< const amrex::Real > &mu_turb, const amrex::Array4< const amrex::Real > &cell_data, amrex::Array4< amrex::Real > &tau11, amrex::Array4< amrex::Real > &tau22, amrex::Array4< amrex::Real > &tau33, amrex::Array4< amrex::Real > &tau12, amrex::Array4< amrex::Real > &tau21, amrex::Array4< amrex::Real > &tau13, amrex::Array4< amrex::Real > &tau31, amrex::Array4< amrex::Real > &tau23, amrex::Array4< amrex::Real > &tau32, const amrex::Array4< const amrex::Real > &er_arr, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy)
void ComputeStrain_N(amrex::Box bx, amrex::Box bxcc, amrex::Box tbxxy, amrex::Box tbxxz, amrex::Box tbxyz, amrex::Box domain, const amrex::Array4< const amrex::Real > &u, const amrex::Array4< const amrex::Real > &v, const amrex::Array4< const amrex::Real > &w, amrex::Array4< amrex::Real > &tau11, amrex::Array4< amrex::Real > &tau22, amrex::Array4< amrex::Real > &tau33, amrex::Array4< amrex::Real > &tau12, amrex::Array4< amrex::Real > &tau13, amrex::Array4< amrex::Real > &tau31, amrex::Array4< amrex::Real > &tau23, amrex::Array4< amrex::Real > &tau32, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy, const amrex::BCRec *bc_ptr, amrex::Array4< amrex::Real > &SmnSmn_a, const amrex::Real implicit_fac)
void ImplicitDiffForState_T(const amrex::Box &bx, const amrex::Box &domain, const int level, const amrex::Real dt, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM *2 > &bc_neumann_vals, const amrex::Array4< amrex::Real > &cell_data, const amrex::Array4< const amrex::Real > &z_nd, const amrex::Array4< const amrex::Real > &detJ, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &cellSizeInv, const amrex::Array4< amrex::Real > &hfx_z, const amrex::Array4< const amrex::Real > &mu_turb, const SolverChoice &solverChoice, const amrex::BCRec *bc_ptr, const bool use_SurfLayer, const amrex::Real implicit_fac)
void ComputeStrain_T(amrex::Box bx, amrex::Box bxcc, amrex::Box tbxxy, amrex::Box tbxxz, amrex::Box tbxyz, amrex::Box domain, const amrex::Array4< const amrex::Real > &u, const amrex::Array4< const amrex::Real > &v, const amrex::Array4< const amrex::Real > &w, amrex::Array4< amrex::Real > &tau11, amrex::Array4< amrex::Real > &tau22, amrex::Array4< amrex::Real > &tau33, amrex::Array4< amrex::Real > &tau12, amrex::Array4< amrex::Real > &tau21, amrex::Array4< amrex::Real > &tau13, amrex::Array4< amrex::Real > &tau31, amrex::Array4< amrex::Real > &tau23, amrex::Array4< amrex::Real > &tau32, const amrex::Array4< const amrex::Real > &z_nd, const amrex::Array4< const amrex::Real > &detJ, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy, const amrex::BCRec *bc_ptr, amrex::Array4< amrex::Real > &SmnSmn_a, const amrex::Real implicit_fac)
void ComputeStressVarVisc_T(amrex::Box bxcc, amrex::Box tbxxy, amrex::Box tbxxz, amrex::Box tbxyz, amrex::Real mu_eff, const amrex::Array4< const amrex::Real > &mu_turb, const amrex::Array4< const amrex::Real > &cell_data, amrex::Array4< amrex::Real > &tau11, amrex::Array4< amrex::Real > &tau22, amrex::Array4< amrex::Real > &tau33, amrex::Array4< amrex::Real > &tau12, amrex::Array4< amrex::Real > &tau21, amrex::Array4< amrex::Real > &tau13, amrex::Array4< amrex::Real > &tau31, amrex::Array4< amrex::Real > &tau23, amrex::Array4< amrex::Real > &tau32, const amrex::Array4< const amrex::Real > &er_arr, const amrex::Array4< const amrex::Real > &z_nd, const amrex::Array4< const amrex::Real > &detJ, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy)
void ComputeStressConsVisc_T(amrex::Box bxcc, amrex::Box tbxxy, amrex::Box tbxxz, amrex::Box tbxyz, amrex::Real mu_eff, const amrex::Array4< const amrex::Real > &cell_data, amrex::Array4< amrex::Real > &tau11, amrex::Array4< amrex::Real > &tau22, amrex::Array4< amrex::Real > &tau33, amrex::Array4< amrex::Real > &tau12, amrex::Array4< amrex::Real > &tau21, amrex::Array4< amrex::Real > &tau13, amrex::Array4< amrex::Real > &tau31, amrex::Array4< amrex::Real > &tau23, amrex::Array4< amrex::Real > &tau32, const amrex::Array4< const amrex::Real > &er_arr, const amrex::Array4< const amrex::Real > &z_nd, const amrex::Array4< const amrex::Real > &detJ, const amrex::GpuArray< amrex::Real, AMREX_SPACEDIM > &dxInv, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy)
void ComputeStressVarVisc_N(amrex::Box bxcc, amrex::Box tbxxy, amrex::Box tbxxz, amrex::Box tbxyz, amrex::Real mu_eff, const amrex::Array4< const amrex::Real > &mu_turb, const amrex::Array4< const amrex::Real > &cell_data, amrex::Array4< amrex::Real > &tau11, amrex::Array4< amrex::Real > &tau22, amrex::Array4< amrex::Real > &tau33, amrex::Array4< amrex::Real > &tau12, amrex::Array4< amrex::Real > &tau13, amrex::Array4< amrex::Real > &tau23, const amrex::Array4< const amrex::Real > &er_arr)
void ComputeStressConsVisc_S(amrex::Box bxcc, amrex::Box tbxxy, amrex::Box tbxxz, amrex::Box tbxyz, amrex::Real mu_eff, const amrex::Array4< const amrex::Real > &cell_data, amrex::Array4< amrex::Real > &tau11, amrex::Array4< amrex::Real > &tau22, amrex::Array4< amrex::Real > &tau33, amrex::Array4< amrex::Real > &tau12, amrex::Array4< amrex::Real > &tau21, amrex::Array4< amrex::Real > &tau13, amrex::Array4< amrex::Real > &tau31, amrex::Array4< amrex::Real > &tau23, amrex::Array4< amrex::Real > &tau32, const amrex::Array4< const amrex::Real > &er_arr, const amrex::Array4< const amrex::Real > &mf_mx, const amrex::Array4< const amrex::Real > &mf_ux, const amrex::Array4< const amrex::Real > &mf_vx, const amrex::Array4< const amrex::Real > &mf_my, const amrex::Array4< const amrex::Real > &mf_uy, const amrex::Array4< const amrex::Real > &mf_vy)
amrex::Real Real
Definition: ERF_ShocInterface.H:19
Definition: ERF_EB.H:13
Definition: ERF_DataStruct.H:123