MFC
Exascale flow solver
Loading...
Searching...
No Matches
m_start_up.fpp.f90
Go to the documentation of this file.
1# 1 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
2# 1 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp" 1
3# 1 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp" 1
4# 1 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp" 1
5# 2 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
6# 3 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
7# 4 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
8# 5 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
9# 6 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
10
11# 8 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
12# 9 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
13# 10 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
14
15# 17 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
16
17# 46 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
18
19# 58 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
20
21# 68 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
22
23# 98 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
24
25# 110 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
26
27# 120 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
28! New line at end of file is required for FYPP
29# 2 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp" 2
30# 1 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp" 1
31# 1 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp" 1
32# 2 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
33# 3 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
34# 4 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
35# 5 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
36# 6 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
37
38# 8 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
39# 9 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
40# 10 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
41
42# 17 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
43
44# 46 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
45
46# 58 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
47
48# 68 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
49
50# 98 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
51
52# 110 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
53
54# 120 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
55! New line at end of file is required for FYPP
56# 2 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp" 2
57
58# 4 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
59# 5 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
60# 6 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
61# 7 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
62# 8 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
63
64# 20 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
65
66# 43 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
67
68# 48 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
69
70# 53 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
71
72# 58 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
73
74# 63 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
75
76# 68 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
77
78# 76 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
79
80# 81 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
81
82# 86 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
83
84# 91 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
85
86# 96 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
87
88# 101 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
89
90# 106 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
91
92# 111 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
93
94# 116 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
95
96# 121 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
97
98# 151 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
99
100# 192 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
101
102# 206 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
103
104# 231 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
105
106# 242 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
107
108# 244 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
109# 255 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
110
111# 284 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
112
113# 294 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
114
115# 304 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
116
117# 313 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
118
119# 330 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
120
121# 340 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
122
123# 347 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
124
125# 353 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
126
127# 359 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
128
129# 365 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
130
131# 371 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
132
133# 377 "/home/runner/work/MFC/MFC/src/common/include/omp_macros.fpp"
134! New line at end of file is required for FYPP
135# 3 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp" 2
136# 1 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp" 1
137# 1 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp" 1
138# 2 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
139# 3 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
140# 4 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
141# 5 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
142# 6 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
143
144# 8 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
145# 9 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
146# 10 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
147
148# 17 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
149
150# 46 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
151
152# 58 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
153
154# 68 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
155
156# 98 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
157
158# 110 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
159
160# 120 "/home/runner/work/MFC/MFC/src/common/include/shared_parallel_macros.fpp"
161! New line at end of file is required for FYPP
162# 2 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp" 2
163
164# 7 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
165
166# 17 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
167
168# 22 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
169
170# 27 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
171
172# 32 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
173
174# 37 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
175
176# 42 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
177
178# 47 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
179
180# 52 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
181
182# 57 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
183
184# 62 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
185
186# 73 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
187
188# 78 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
189
190# 83 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
191
192# 88 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
193
194# 103 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
195
196# 131 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
197
198# 160 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
199
200# 175 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
201
202# 193 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
203
204# 215 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
205
206# 244 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
207
208# 259 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
209
210# 269 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
211
212# 278 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
213
214# 294 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
215
216# 304 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
217
218# 311 "/home/runner/work/MFC/MFC/src/common/include/acc_macros.fpp"
219! New line at end of file is required for FYPP
220# 4 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp" 2
221
222! GPU parallel region (scalar reductions, maxval/minval)
223# 23 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
224
225! GPU parallel loop over threads (most common GPU macro)
226# 43 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
227
228! Required closing for GPU_PARALLEL_LOOP
229# 55 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
230
231! Mark routine for device compilation
232# 112 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
233
234! Declare device-resident data
235# 130 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
236
237! Inner loop within a GPU parallel region
238# 145 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
239
240! Scoped GPU data region
241# 164 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
242
243! Host code with device pointers (for MPI with GPU buffers)
244# 193 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
245
246! Allocate device memory (unscoped)
247# 207 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
248
249! Free device memory
250# 219 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
251
252! Atomic operation on device
253# 231 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
254
255! End atomic capture block
256# 242 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
257
258! Copy data between host and device
259# 254 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
260
261! Synchronization barrier
262# 266 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
263
264! Import GPU library module (openacc or omp_lib)
265# 275 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
266
267! Emit code only for AMD compiler
268# 282 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
269
270! Emit code for non-Cray compilers
271# 289 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
272
273! Emit code only for Cray compiler
274# 296 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
275
276! Emit code for non-NVIDIA compilers
277# 303 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
278
279# 305 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
280# 306 "/home/runner/work/MFC/MFC/src/common/include/parallel_macros.fpp"
281! New line at end of file is required for FYPP
282# 2 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp" 2
283
284# 14 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
285
286! Caution: This macro requires the use of a binding script to set CUDA_VISIBLE_DEVICES, such that we have one GPU device per MPI
287! rank. That's because for both cudaMemAdvise (preferred location) and cudaMemPrefetchAsync we use location = device_id = 0. For an
288! example see misc/nvidia_uvm/bind.sh. NVIDIA unified memory page placement hint
289# 57 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
290
291! Allocate and create GPU device memory
292# 77 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
293
294! Free GPU device memory and deallocate
295# 85 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
296
297! Cray-specific GPU pointer setup for vector fields
298# 109 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
299
300! Cray-specific GPU pointer setup for scalar fields
301# 125 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
302
303! Cray-specific GPU pointer setup for acoustic source spatials
304# 150 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
305
306# 156 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
307
308# 163 "/home/runner/work/MFC/MFC/src/common/include/macros.fpp"
309! New line at end of file is required for FYPP
310# 2 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp" 2
311
312!>
313!! @file
314!! @brief Contains module m_start_up
315
316!> @brief Reads and validates user inputs, allocates variables, and configures MPI decomposition and I/O for post-processing
317
319
320 use, intrinsic :: iso_c_binding
321
324 use m_mpi_proxy
325 use m_mpi_common
328 use m_data_input
329 use m_data_output
331 use m_helper
334 use m_checker
335 use m_thermochem, only: num_species, species_names
338 use m_chemistry
339
340#ifdef MFC_MPI
341 use mpi
342#endif
343
344 implicit none
345
346 include 'fftw3.f03'
347
349 complex(c_double_complex), allocatable :: data_in(:), data_out(:)
350 complex(c_double_complex), allocatable :: data_cmplx(:,:,:), data_cmplx_y(:,:,:), data_cmplx_z(:,:,:)
351 real(wp), allocatable, dimension(:,:,:) :: en_real
352 real(wp), allocatable, dimension(:) :: en
354 integer :: nx, ny, nz, nxloc, nyloc, nyloc2, nzloc, nf
355 integer :: ierr
357 integer, dimension(3) :: cart3d_coords
358 integer, dimension(2) :: cart2d12_coords, cart2d13_coords
360
361contains
362
363 !> Reads the configuration file post_process.inp, in order to populate parameters in module m_global_parameters.f90 with the
364 !! user provided inputs
365 impure subroutine s_read_input_file
366
367 character(LEN=name_len) :: file_loc
368 logical :: file_check
369 integer :: iostatus
370 character(len=1000) :: line
371
372# 1 "/home/runner/work/MFC/MFC/build/include/post_process/generated_namelist.fpp" 1
373! AUTO-GENERATED - do not edit directly. Regenerate: cmake reconfigure
374!
375namelist /user_inputs/ bx0, ca, e_wrt, g, r0ref, re_inv, web, adv_n, alpha_rho_e_wrt, alpha_rho_wrt, alpha_wrt, alt_soundspeed, &
376 & avg_state, bc_x, bc_y, bc_z, bub_pp, bubbles_euler, bubbles_lagrange, c_wrt, case_dir, cf_wrt, cfl_adap_dt, cfl_const_dt, &
377 & cfl_target, chem_wrt_t, chem_wrt_y, cons_vars_wrt, cont_damage, cyl_coord, down_sample, fd_order, fft_wrt, &
378 & file_per_process, fluid_pp, flux_lim, flux_wrt, format, gamma_wrt, heat_ratio_wrt, hyper_cleaning, hyperelasticity, &
379 & hypoelasticity, ib, ib_state_wrt, igr, igr_order, lag_betac_wrt, lag_betat_wrt, lag_db_wrt, lag_dphidt_wrt, lag_header, &
380 & lag_id_wrt, lag_mg_wrt, lag_mv_wrt, lag_pos_prev_wrt, lag_pos_wrt, lag_pres_wrt, lag_r0_wrt, lag_rad_wrt, lag_rmax_wrt, &
381 & lag_rmin_wrt, lag_rvel_wrt, lag_txt_wrt, lag_vel_wrt, liutex_wrt, m, mhd, mixture_err, model_eqns, mom_wrt, mpp_lim, &
382 & muscl_order, n, n_start, nb, num_bc_patches, num_fluids, num_ibs, omega_wrt, output_partial_domain, p, parallel_io, &
383 & pi_inf_wrt, poly_sigma, polydisperse, polytropic, precision, pref, pres_inf_wrt, pres_wrt, prim_vars_wrt, qbmm, qm_wrt, &
384 & recon_type, relativity, relax, relax_model, rho_wrt, rhoref, schlieren_alpha, schlieren_wrt, sigr, sigma, sim_data, &
385 & surface_tension, t_save, t_step_save, t_step_start, t_step_stop, t_stop, thermal, vel_wrt, weno_order, x_output, y_output, &
386 & z_output
387# 64 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp" 2
388
389 file_loc = 'post_process.inp'
390 inquire (file=trim(file_loc), exist=file_check)
391
392 if (file_check) then
393 open (1, file=trim(file_loc), form='formatted', status='old', action='read')
394 read (1, nml=user_inputs, iostat=iostatus)
395
396 if (iostatus /= 0) then
397 backspace(1)
398 read (1, fmt='(A)') line
399 print *, 'Invalid line in namelist: ' // trim(line)
400 call s_mpi_abort('Invalid line in post_process.inp. It is ' // 'likely due to a datatype mismatch. Exiting.')
401 end if
402
403 close (1)
404
405 call s_update_cell_bounds(cells_bounds, m, n, p)
406
407 if (down_sample) then
408 m = int((m + 1)/3) - 1
409 n = int((n + 1)/3) - 1
410 p = int((p + 1)/3) - 1
411 end if
412
413 m_glb = m
414 n_glb = n
415 p_glb = p
416
417 nglobal = int(m_glb + 1, kind=8)*int(n_glb + 1, kind=8)*int(p_glb + 1, kind=8)
418
419 if (cfl_adap_dt .or. cfl_const_dt) cfl_dt = .true.
420
421 if (any((/bc_x%beg, bc_x%end, bc_y%beg, bc_y%end, bc_z%beg, bc_z%end/) == -17) .or. num_bc_patches > 0) then
422 bc_io = .true.
423 end if
424 else
425 call s_mpi_abort('File post_process.inp is missing. Exiting.')
426 end if
427
428 end subroutine s_read_input_file
429
430 !> Checking that the user inputs make sense, i.e. that the individual choices are compatible with the code's options and that
431 !! the combination of these choices results into a valid configuration for the post-process
432 impure subroutine s_check_input_file
433
434 character(LEN=len_trim(case_dir)) :: file_loc
435 logical :: dir_check
436
437 case_dir = adjustl(case_dir)
438
439 file_loc = trim(case_dir) // '/.'
440
441 call my_inquire(file_loc, dir_check)
442
443 if (dir_check .neqv. .true.) then
444 call s_mpi_abort('Unsupported choice for the value of ' // 'case_dir. Exiting.')
445 end if
446
448 call s_check_inputs()
449
450 end subroutine s_check_input_file
451
452 !> Load grid and conservative data for a time step, fill ghost-cell buffers, and convert to primitive variables.
453 impure subroutine s_perform_time_step(t_step)
454
455 integer, intent(inout) :: t_step
456 integer :: eta_hh, eta_mm, eta_ss
457 real(wp) :: eta_sec
458
459 if (proc_rank == 0) then
460 if (cfl_dt) then
461 eta_sec = wall_time_avg*real(n_save - 1 - t_step, wp)
462 eta_hh = int(eta_sec)/3600
463 eta_mm = mod(int(eta_sec), 3600)/60
464 eta_ss = mod(int(eta_sec), 60)
465 print '(" [", I3, "%] Saving ", I8, " of ", I0, " Time Avg = ", ES16.6, " Time/step = ", ES12.6, " ETA (HH:MM:SS) = ", I0, ":", I2.2, ":", I2.2)', &
466 & int(ceiling(100._wp*(real(t_step - n_start)/(n_save)))), t_step, n_save, wall_time_avg, wall_time, eta_hh, &
467 & eta_mm, eta_ss
468 else
469 eta_sec = wall_time_avg*real((t_step_stop - t_step)/t_step_save, wp)
470 eta_hh = int(eta_sec)/3600
471 eta_mm = mod(int(eta_sec), 3600)/60
472 eta_ss = mod(int(eta_sec), 60)
473 print '(" [", I3, "%] Saving ", I8, " of ", I0, " @ t_step = ", I8, " Time Avg = ", ES16.6, " Time/step = ", ES12.6, " ETA (HH:MM:SS) = ", I0, ":", I2.2, ":", I2.2)', &
474 & int(ceiling(100._wp*(real(t_step - t_step_start)/(t_step_stop - t_step_start + 1)))), &
475 & (t_step - t_step_start)/t_step_save + 1, (t_step_stop - t_step_start)/t_step_save + 1, t_step, &
476 & wall_time_avg, wall_time, eta_hh, eta_mm, eta_ss
477 end if
478 end if
479
480 call s_read_data_files(t_step)
481
482 if (chemistry) call s_compute_q_t_sf(q_t_sf, q_cons_vf, idwbuff)
483
484 if (buff_size > 0) then
485 call s_populate_grid_variables_buffers()
487 end if
488
490
491 end subroutine s_perform_time_step
492
493 !> Derive requested flow quantities from primitive variables and write them to the formatted database files.
494 impure subroutine s_save_data(t_step, varname, pres, c, H)
495
496 integer, intent(inout) :: t_step
497 character(LEN=name_len), intent(inout) :: varname
498 real(wp), intent(inout) :: pres, c, h
499
500 real(wp), dimension(-offset_x%beg:m + offset_x%end,-offset_y%beg:n + offset_y%end, & & -offset_z%beg:p + offset_z%end) :: liutex_mag
501 real(wp), dimension(-offset_x%beg:m + offset_x%end,-offset_y%beg:n + offset_y%end,-offset_z%beg:p + offset_z%end, & & 3) :: liutex_axis
502 integer :: i, j, k, l, kx, ky, kz, kf, j_glb, k_glb, l_glb
503 character(50) :: filename
504 logical :: file_exists
505 integer :: x_beg, x_end, y_beg, y_end, z_beg, z_end
506
507 if (output_partial_domain) then
509 x_beg = -offset_x%beg + x_output_idx%beg
510 x_end = offset_x%end + x_output_idx%end
511 y_beg = -offset_y%beg + y_output_idx%beg
512 y_end = offset_y%end + y_output_idx%end
513 z_beg = -offset_z%beg + z_output_idx%beg
514 z_end = offset_z%end + z_output_idx%end
515 else
516 x_beg = -offset_x%beg
517 x_end = offset_x%end + m
518 y_beg = -offset_y%beg
519 y_end = offset_y%end + n
520 z_beg = -offset_z%beg
521 z_end = offset_z%end + p
522 end if
523
525
526 if (sim_data .and. proc_rank == 0) then
529 end if
530
531 if (sim_data) then
534 end if
535
537
538 if (omega_wrt(2) .or. omega_wrt(3) .or. qm_wrt .or. liutex_wrt .or. schlieren_wrt) then
540 end if
541
542 if (omega_wrt(1) .or. omega_wrt(3) .or. qm_wrt .or. liutex_wrt .or. (n > 0 .and. schlieren_wrt)) then
544 end if
545
546 if (omega_wrt(1) .or. omega_wrt(2) .or. qm_wrt .or. liutex_wrt .or. (p > 0 .and. schlieren_wrt)) then
548 end if
549
550 if ((model_eqns == model_eqns_5eq) .or. (model_eqns == model_eqns_6eq) .or. (model_eqns == model_eqns_4eq)) then
551 do i = 1, num_fluids
552 if (alpha_rho_wrt(i) .or. (cons_vars_wrt .or. prim_vars_wrt)) then
553 out%q_sf(:,:,:) = q_cons_vf(i)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
554 if (model_eqns /= model_eqns_4eq) then
555 write (varname, '(A,I0)') 'alpha_rho', i
556 else
557 write (varname, '(A,I0)') 'rho', i
558 end if
560
561 varname(:) = ' '
562 end if
563 end do
564 end if
565
566 if ((rho_wrt .or. (model_eqns == model_eqns_gamma_law .and. (cons_vars_wrt .or. prim_vars_wrt))) .and. (.not. relativity)) &
567 & then
568 out%q_sf(:,:,:) = rho_sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
569 write (varname, '(A)') 'rho'
571
572 varname(:) = ' '
573 end if
574
575 if (relativity .and. (rho_wrt .or. prim_vars_wrt)) then
576 out%q_sf(:,:,:) = q_prim_vf(1)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
577 write (varname, '(A)') 'rho'
579
580 varname(:) = ' '
581 end if
582
583 if (relativity .and. (rho_wrt .or. cons_vars_wrt)) then
584 ! For relativistic flow, conservative and primitive densities are different Hard-coded single-component for now
585 out%q_sf(:,:,:) = q_cons_vf(1)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
586 write (varname, '(A)') 'D'
588
589 varname(:) = ' '
590 end if
591
592 do i = 1, eqn_idx%E - eqn_idx%mom%beg
593 if (mom_wrt(i) .or. cons_vars_wrt) then
594 out%q_sf(:,:,:) = q_cons_vf(i + eqn_idx%cont%end)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
595 write (varname, '(A,I0)') 'mom', i
597
598 varname(:) = ' '
599 end if
600 end do
601
602 do i = 1, eqn_idx%E - eqn_idx%mom%beg
603 if (vel_wrt(i) .or. prim_vars_wrt) then
604 out%q_sf(:,:,:) = q_prim_vf(i + eqn_idx%cont%end)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
605 write (varname, '(A,I0)') 'vel', i
607
608 varname(:) = ' '
609 end if
610 end do
611
612 if (chemistry) then
613 do i = 1, num_species
614 if (chem_wrt_y(i) .or. prim_vars_wrt) then
615 out%q_sf(:,:,:) = q_prim_vf(eqn_idx%species%beg + i - 1)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
616 write (varname, '(A,A)') 'Y_', trim(species_names(i))
618
619 varname(:) = ' '
620 end if
621 end do
622
623 if (chem_wrt_t) then
624 out%q_sf(:,:,:) = q_t_sf%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
625 write (varname, '(A)') 'T'
627
628 varname(:) = ' '
629 end if
630 end if
631
632 do i = 1, eqn_idx%E - eqn_idx%mom%beg
633 if (flux_wrt(i)) then
634 call s_derive_flux_limiter(i, q_prim_vf, out%q_sf)
635
636 write (varname, '(A,I0)') 'flux', i
638
639 varname(:) = ' '
640 end if
641 end do
642
643 if (e_wrt .or. cons_vars_wrt) then
644 out%q_sf(:,:,:) = q_cons_vf(eqn_idx%E)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
645 write (varname, '(A)') 'E'
647
648 varname(:) = ' '
649 end if
650
651 if (model_eqns == model_eqns_6eq) then
652 do i = 1, num_fluids
653 if (alpha_rho_e_wrt(i) .or. cons_vars_wrt) then
654 out%q_sf = q_cons_vf(i + eqn_idx%int_en%beg - 1)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
655 write (varname, '(A,I0)') 'alpha_rho_e', i
657
658 varname(:) = ' '
659 end if
660 end do
661 end if
662
663 if (fft_wrt) then
664 do l = 0, p
665 do k = 0, n
666 do j = 0, m
667 data_cmplx(j + 1, k + 1, l + 1) = cmplx(q_cons_vf(eqn_idx%mom%beg)%sf(j, k, l)/q_cons_vf(1)%sf(j, k, l), &
668 & 0._wp)
669 end do
670 end do
671 end do
672
673 call s_mpi_fft_fwd()
674
675 en_real = 0.5_wp*abs(data_cmplx_z)**2._wp/(1._wp*nx*ny*nz)**2._wp
676
677 do l = 0, p
678 do k = 0, n
679 do j = 0, m
680 data_cmplx(j + 1, k + 1, l + 1) = cmplx(q_cons_vf(eqn_idx%mom%beg + 1)%sf(j, k, l)/q_cons_vf(1)%sf(j, k, &
681 & l), 0._wp)
682 end do
683 end do
684 end do
685
686 call s_mpi_fft_fwd()
687
688 en_real = en_real + 0.5_wp*abs(data_cmplx_z)**2._wp/(1._wp*nx*ny*nz)**2._wp
689
690 do l = 0, p
691 do k = 0, n
692 do j = 0, m
693 data_cmplx(j + 1, k + 1, l + 1) = cmplx(q_cons_vf(eqn_idx%mom%beg + 2)%sf(j, k, l)/q_cons_vf(1)%sf(j, k, &
694 & l), 0._wp)
695 end do
696 end do
697 end do
698
699 call s_mpi_fft_fwd()
700
701 en_real = en_real + 0.5_wp*abs(data_cmplx_z)**2._wp/(1._wp*nx*ny*nz)**2._wp
702
703 do kf = 1, nf
704 en(kf) = 0._wp
705 end do
706
707 do l = 1, nz
708 do k = 1, nyloc2
709 do j = 1, nxloc
710 j_glb = j + cart3d_coords(2)*nxloc
711 k_glb = k + cart3d_coords(3)*nyloc2
712 l_glb = l
713
714 if (j_glb >= (m_glb + 1)/2) then
715 kx = (j_glb - 1) - (m_glb + 1)
716 else
717 kx = j_glb - 1
718 end if
719
720 if (k_glb >= (n_glb + 1)/2) then
721 ky = (k_glb - 1) - (n_glb + 1)
722 else
723 ky = k_glb - 1
724 end if
725
726 if (l_glb >= (p_glb + 1)/2) then
727 kz = (l_glb - 1) - (p_glb + 1)
728 else
729 kz = l_glb - 1
730 end if
731
732 kf = nint(sqrt(kx**2._wp + ky**2._wp + kz**2._wp)) + 1
733
734 en(kf) = en(kf) + en_real(j, k, l)
735 end do
736 end do
737 end do
738
739#ifdef MFC_MPI
740 call mpi_allreduce(mpi_in_place, en, nf, mpi_p, mpi_sum, mpi_comm_world, ierr)
741#endif
742
743 if (proc_rank == 0) then
744 call s_create_directory('En_FFT_DATA')
745 write (filename, '(a,i0,a)') 'En_FFT_DATA/En_tot', t_step, '.dat'
746 inquire (file=filename, exist=file_exists)
747 if (file_exists) then
748 call s_delete_file(trim(filename))
749 end if
750 end if
751
752 do kf = 1, nf
753 if (proc_rank == 0) then
754 write (filename, '(a,i0,a)') 'En_FFT_DATA/En_tot', t_step, '.dat'
755 inquire (file=filename, exist=file_exists)
756 if (file_exists) then
757 open (1, file=filename, position='append', status='old')
758 write (1, *) en(kf), t_step
759 close (1)
760 else
761 open (1, file=filename, status='new')
762 write (1, *) en(kf), t_step
763 close (1)
764 end if
765 end if
766 end do
767 end if
768
769 if (mhd .and. prim_vars_wrt) then
770 do i = eqn_idx%B%beg, eqn_idx%B%end
771 out%q_sf(:,:,:) = q_prim_vf(i)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
772
773 ! 1D: output By, Bz
774 if (n == 0) then
775 if (i == eqn_idx%B%beg) then
776 write (varname, '(A)') 'By'
777 else
778 write (varname, '(A)') 'Bz'
779 end if
780 ! 2D/3D: output Bx, By, Bz
781 else
782 if (i == eqn_idx%B%beg) then
783 write (varname, '(A)') 'Bx'
784 else if (i == eqn_idx%B%beg + 1) then
785 write (varname, '(A)') 'By'
786 else
787 write (varname, '(A)') 'Bz'
788 end if
789 end if
790
792 varname(:) = ' '
793 end do
794 end if
795
796 if (elasticity) then
797 do i = 1, eqn_idx%stress%end - eqn_idx%stress%beg + 1
798 if (prim_vars_wrt) then
799 out%q_sf(:,:,:) = q_prim_vf(i - 1 + eqn_idx%stress%beg)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
800 write (varname, '(A,I0)') 'tau', i
802 end if
803 varname(:) = ' '
804 end do
805 end if
806
807 if (hyperelasticity) then
808 do i = 1, eqn_idx%xi%end - eqn_idx%xi%beg + 1
809 if (prim_vars_wrt) then
810 out%q_sf(:,:,:) = q_prim_vf(i - 1 + eqn_idx%xi%beg)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
811 write (varname, '(A,I0)') 'xi', i
813 end if
814 varname(:) = ' '
815 end do
816 end if
817
818 if (cont_damage) then
819 out%q_sf(:,:,:) = q_cons_vf(eqn_idx%damage)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
820 write (varname, '(A)') 'damage_state'
822
823 varname(:) = ' '
824 end if
825
826 if (hyper_cleaning) then
827 out%q_sf = q_cons_vf(eqn_idx%psi)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
828 write (varname, '(A)') 'psi'
830
831 varname(:) = ' '
832 end if
833
834 if (pres_wrt .or. prim_vars_wrt) then
835 out%q_sf(:,:,:) = q_prim_vf(eqn_idx%E)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
836 write (varname, '(A)') 'pres'
838
839 varname(:) = ' '
840 end if
841
842 if (((model_eqns == model_eqns_5eq) .and. (bubbles_euler .neqv. .true.)) .or. (model_eqns == model_eqns_6eq)) then
843 do i = 1, num_fluids - 1
844 if (alpha_wrt(i) .or. (cons_vars_wrt .or. prim_vars_wrt)) then
845 out%q_sf(:,:,:) = q_cons_vf(i + eqn_idx%E)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
846 write (varname, '(A,I0)') 'alpha', i
848
849 varname(:) = ' '
850 end if
851 end do
852
853 if (alpha_wrt(num_fluids) .or. (cons_vars_wrt .or. prim_vars_wrt)) then
854 if (igr) then
855 do k = z_beg, z_end
856 do j = y_beg, y_end
857 do i = x_beg, x_end
858 out%q_sf(i, j, k) = 1._wp
859 do l = 1, num_fluids - 1
860 out%q_sf(i, j, k) = out%q_sf(i, j, k) - q_cons_vf(eqn_idx%E + l)%sf(i, j, k)
861 end do
862 end do
863 end do
864 end do
865 else
866 out%q_sf(:,:,:) = q_cons_vf(eqn_idx%adv%end)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
867 end if
868 write (varname, '(A,I0)') 'alpha', num_fluids
870
871 varname(:) = ' '
872 end if
873 end if
874
875 if (gamma_wrt .or. (model_eqns == model_eqns_gamma_law .and. (cons_vars_wrt .or. prim_vars_wrt))) then
876 out%q_sf(:,:,:) = gamma_sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
877 write (varname, '(A)') 'gamma'
879
880 varname(:) = ' '
881 end if
882
883 if (heat_ratio_wrt) then
885
886 write (varname, '(A)') 'heat_ratio'
888
889 varname(:) = ' '
890 end if
891
892 if (pi_inf_wrt .or. (model_eqns == model_eqns_gamma_law .and. (cons_vars_wrt .or. prim_vars_wrt))) then
893 out%q_sf(:,:,:) = pi_inf_sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
894 write (varname, '(A)') 'pi_inf'
896
897 varname(:) = ' '
898 end if
899
900 if (pres_inf_wrt) then
902
903 write (varname, '(A)') 'pres_inf'
905
906 varname(:) = ' '
907 end if
908
909 if (c_wrt) then
910 do k = -offset_z%beg, p + offset_z%end
911 do j = -offset_y%beg, n + offset_y%end
912 do i = -offset_x%beg, m + offset_x%end
913 do l = 1, eqn_idx%adv%end - eqn_idx%E
914 adv(l) = q_prim_vf(eqn_idx%E + l)%sf(i, j, k)
915 end do
916
917 pres = q_prim_vf(eqn_idx%E)%sf(i, j, k)
918
919 h = ((gamma_sf(i, j, k) + 1._wp)*pres + pi_inf_sf(i, j, k) + qv_sf(i, j, k))/rho_sf(i, j, k)
920
921 call s_compute_speed_of_sound(pres, rho_sf(i, j, k), gamma_sf(i, j, k), pi_inf_sf(i, j, k), h, adv, &
922 & 0._wp, 0._wp, c, qv_sf(i, j, k))
923
924 out%q_sf(i, j, k) = c
925 end do
926 end do
927 end do
928
929 write (varname, '(A)') 'c'
931
932 varname(:) = ' '
933 end if
934
935 do i = 1, 3
936 if (omega_wrt(i)) then
938
939 write (varname, '(A,I0)') 'omega', i
941
942 varname(:) = ' '
943 end if
944 end do
945
946 if (ib) then
947 out%q_sf(:,:,:) = real(ib_markers%sf(-offset_x%beg:m + offset_x%end,-offset_y%beg:n + offset_y%end, &
948 & -offset_z%beg:p + offset_z%end))
949 varname = 'ib_markers'
951 end if
952
953 if (p > 0 .and. qm_wrt) then
954 call s_derive_qm(q_prim_vf, out%q_sf)
955
956 write (varname, '(A)') 'qm'
958
959 varname(:) = ' '
960 end if
961
962 if (liutex_wrt) then
963 call s_derive_liutex(q_prim_vf, liutex_mag, liutex_axis)
964
965 out%q_sf = liutex_mag
966
967 write (varname, '(A)') 'liutex_mag'
969
970 varname(:) = ' '
971
972 do i = 1, 3
973 out%q_sf = liutex_axis(:,:,:,i)
974
975 write (varname, '(A,I0)') 'liutex_axis', i
977
978 varname(:) = ' '
979 end do
980 end if
981
982 if (schlieren_wrt) then
984
985 write (varname, '(A)') 'schlieren'
987
988 varname(:) = ' '
989 end if
990
991 if (cf_wrt) then
992 out%q_sf(:,:,:) = q_cons_vf(eqn_idx%c)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
993 write (varname, '(A,I0)') 'color_function'
995 varname(:) = ' '
996 end if
997
998 if (bubbles_euler) then
999 do i = eqn_idx%adv%beg, eqn_idx%adv%end
1000 out%q_sf(:,:,:) = q_cons_vf(i)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
1001 write (varname, '(A,I0)') 'alpha', i - eqn_idx%E
1003 varname(:) = ' '
1004 end do
1005 end if
1006
1007 if (bubbles_euler) then
1008 ! nR
1009 do i = 1, nb
1010 out%q_sf(:,:,:) = q_cons_vf(qbmm_idx%rs(i))%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
1011 write (varname, '(A,I3.3)') 'nR', i
1013 varname(:) = ' '
1014 end do
1015
1016 ! nRdot
1017 do i = 1, nb
1018 out%q_sf(:,:,:) = q_cons_vf(qbmm_idx%vs(i))%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
1019 write (varname, '(A,I3.3)') 'nV', i
1021 varname(:) = ' '
1022 end do
1023 if ((polytropic .neqv. .true.) .and. (.not. qbmm)) then
1024 ! nP
1025 do i = 1, nb
1026 out%q_sf(:,:,:) = q_cons_vf(qbmm_idx%ps(i))%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
1027 write (varname, '(A,I3.3)') 'nP', i
1029 varname(:) = ' '
1030 end do
1031
1032 ! nM
1033 do i = 1, nb
1034 out%q_sf(:,:,:) = q_cons_vf(qbmm_idx%ms(i))%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
1035 write (varname, '(A,I3.3)') 'nM', i
1037 varname(:) = ' '
1038 end do
1039 end if
1040
1041 ! number density
1042 if (adv_n) then
1043 out%q_sf(:,:,:) = q_cons_vf(eqn_idx%n)%sf(x_beg:x_end,y_beg:y_end,z_beg:z_end)
1044 write (varname, '(A)') 'n'
1046 varname(:) = ' '
1047 end if
1048 end if
1049
1050 if (bubbles_lagrange) then
1051 ! Void fraction field
1052 out%q_sf(:,:,:) = 1._wp - q_cons_vf(beta_idx)%sf(-offset_x%beg:m + offset_x%end,-offset_y%beg:n + offset_y%end, &
1053 & -offset_z%beg:p + offset_z%end)
1054 write (varname, '(A)') 'voidFraction'
1056 varname(:) = ' '
1057
1058 if (lag_txt_wrt) call s_write_lag_bubbles_results_to_text(t_step) ! text output
1059 if (lag_db_wrt) call s_write_lag_bubbles_to_formatted_database_file(t_step) ! silo file output
1060 end if
1061
1062 if (ib_state_wrt) call s_write_ib_bodies_to_formatted_database_file(t_step)
1063
1064 if (sim_data .and. proc_rank == 0) then
1067 end if
1068
1070
1071 end subroutine s_save_data
1072
1073 !> Transpose 3-D complex data from x-pencil to y-pencil layout via MPI_Alltoall.
1074 subroutine s_mpi_transpose_x2y
1075
1076 complex(c_double_complex), allocatable :: sendbuf(:), recvbuf(:)
1077 integer :: dest_rank, src_rank
1078 integer :: i, j, k, l
1079
1080#ifdef MFC_MPI
1081 allocate (sendbuf(nx*nyloc*nzloc))
1082 allocate (recvbuf(nx*nyloc*nzloc))
1083
1084 do dest_rank = 0, num_procs_y - 1
1085 do l = 1, nzloc
1086 do k = 1, nyloc
1087 do j = 1, nxloc
1088 sendbuf(j + (k - 1)*nxloc + (l - 1)*nxloc*nyloc + dest_rank*nxloc*nyloc*nzloc) = data_cmplx(j &
1089 & + dest_rank*nxloc, k, l)
1090 end do
1091 end do
1092 end do
1093 end do
1094
1095 call mpi_alltoall(sendbuf, nxloc*nyloc*nzloc, mpi_c_double_complex, recvbuf, nxloc*nyloc*nzloc, mpi_c_double_complex, &
1097
1098 do src_rank = 0, num_procs_y - 1
1099 do l = 1, nzloc
1100 do k = 1, nyloc
1101 do j = 1, nxloc
1102 data_cmplx_y(j, k + src_rank*nyloc, &
1103 & l) = recvbuf(j + (k - 1)*nxloc + (l - 1)*nxloc*nyloc + src_rank*nxloc*nyloc*nzloc)
1104 end do
1105 end do
1106 end do
1107 end do
1108
1109 deallocate (sendbuf)
1110 deallocate (recvbuf)
1111#endif
1112
1113 end subroutine s_mpi_transpose_x2y
1114
1115 !> Transpose 3-D complex data from y-pencil to z-pencil layout via MPI_Alltoall.
1116 subroutine s_mpi_transpose_y2z
1117
1118 complex(c_double_complex), allocatable :: sendbuf(:), recvbuf(:)
1119 integer :: dest_rank, src_rank
1120 integer :: j, k, l
1121
1122#ifdef MFC_MPI
1123 allocate (sendbuf(ny*nxloc*nzloc))
1124 allocate (recvbuf(ny*nxloc*nzloc))
1125
1126 do dest_rank = 0, num_procs_z - 1
1127 do l = 1, nzloc
1128 do j = 1, nxloc
1129 do k = 1, nyloc2
1130 sendbuf(k + (j - 1)*nyloc2 + (l - 1)*(nyloc2*nxloc) + dest_rank*nyloc2*nxloc*nzloc) = data_cmplx_y(j, &
1131 & k + dest_rank*nyloc2, l)
1132 end do
1133 end do
1134 end do
1135 end do
1136
1137 call mpi_alltoall(sendbuf, nyloc2*nxloc*nzloc, mpi_c_double_complex, recvbuf, nyloc2*nxloc*nzloc, mpi_c_double_complex, &
1139
1140 do src_rank = 0, num_procs_z - 1
1141 do l = 1, nzloc
1142 do j = 1, nxloc
1143 do k = 1, nyloc2
1144 data_cmplx_z(j, k, &
1145 & l + src_rank*nzloc) = recvbuf(k + (j - 1)*nyloc2 + (l - 1)*(nyloc2*nxloc) &
1146 & + src_rank*nyloc2*nxloc*nzloc)
1147 end do
1148 end do
1149 end do
1150 end do
1151
1152 deallocate (sendbuf)
1153 deallocate (recvbuf)
1154#endif
1155
1156 end subroutine s_mpi_transpose_y2z
1157
1158 !> Initialize all post-process sub-modules, set up I/O pointers, and prepare FFTW plans and MPI communicators.
1159 impure subroutine s_initialize_modules
1160
1161 integer :: size_n(1), inembed(1), onembed(1)
1162
1164 if (bubbles_euler .or. bubbles_lagrange) then
1166 end if
1167 if (num_procs > 1) then
1170 end if
1176
1177 if (parallel_io .neqv. .true.) then
1179 else
1181 end if
1182
1183#ifdef MFC_MPI
1184 if (fft_wrt) then
1185 num_procs_x = (m_glb + 1)/(m + 1)
1186 num_procs_y = (n_glb + 1)/(n + 1)
1187 num_procs_z = (p_glb + 1)/(p + 1)
1188
1189 nx = m_glb + 1
1190 ny = n_glb + 1
1191 nz = p_glb + 1
1192
1193 nxloc = (m_glb + 1)/num_procs_y
1194 nyloc = n + 1
1195 nyloc2 = (n_glb + 1)/num_procs_z
1196 nzloc = p + 1
1197
1198 nf = max(nx, ny, nz)
1199
1200#ifdef MFC_DEBUG
1201# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1202 block
1203# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1204 use iso_fortran_env, only: output_unit
1205# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1206
1207# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1208 print *, 'm_start_up.fpp:878: ', '@:ALLOCATE(data_in(Nx*Nyloc*Nzloc))'
1209# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1210
1211# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1212 call flush (output_unit)
1213# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1214 end block
1215# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1216#endif
1217# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1218 allocate (data_in(nx*nyloc*nzloc))
1219# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1220
1221# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1222
1223# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1224#if defined(MFC_OpenACC)
1225# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1226!$acc enter data create(data_in)
1227# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1228#elif defined(MFC_OpenMP)
1229# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1230!$omp target enter data map(always,alloc:data_in)
1231# 878 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1232#endif
1233#ifdef MFC_DEBUG
1234# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1235 block
1236# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1237 use iso_fortran_env, only: output_unit
1238# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1239
1240# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1241 print *, 'm_start_up.fpp:879: ', '@:ALLOCATE(data_out(Nx*Nyloc*Nzloc))'
1242# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1243
1244# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1245 call flush (output_unit)
1246# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1247 end block
1248# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1249#endif
1250# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1251 allocate (data_out(nx*nyloc*nzloc))
1252# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1253
1254# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1255
1256# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1257#if defined(MFC_OpenACC)
1258# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1259!$acc enter data create(data_out)
1260# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1261#elif defined(MFC_OpenMP)
1262# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1263!$omp target enter data map(always,alloc:data_out)
1264# 879 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1265#endif
1266
1267#ifdef MFC_DEBUG
1268# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1269 block
1270# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1271 use iso_fortran_env, only: output_unit
1272# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1273
1274# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1275 print *, 'm_start_up.fpp:881: ', '@:ALLOCATE(data_cmplx(Nx, Nyloc, Nzloc))'
1276# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1277
1278# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1279 call flush (output_unit)
1280# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1281 end block
1282# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1283#endif
1284# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1285 allocate (data_cmplx(nx, nyloc, nzloc))
1286# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1287
1288# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1289
1290# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1291#if defined(MFC_OpenACC)
1292# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1293!$acc enter data create(data_cmplx)
1294# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1295#elif defined(MFC_OpenMP)
1296# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1297!$omp target enter data map(always,alloc:data_cmplx)
1298# 881 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1299#endif
1300#ifdef MFC_DEBUG
1301# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1302 block
1303# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1304 use iso_fortran_env, only: output_unit
1305# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1306
1307# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1308 print *, 'm_start_up.fpp:882: ', '@:ALLOCATE(data_cmplx_y(Nxloc, Ny, Nzloc))'
1309# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1310
1311# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1312 call flush (output_unit)
1313# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1314 end block
1315# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1316#endif
1317# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1318 allocate (data_cmplx_y(nxloc, ny, nzloc))
1319# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1320
1321# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1322
1323# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1324#if defined(MFC_OpenACC)
1325# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1326!$acc enter data create(data_cmplx_y)
1327# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1328#elif defined(MFC_OpenMP)
1329# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1330!$omp target enter data map(always,alloc:data_cmplx_y)
1331# 882 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1332#endif
1333#ifdef MFC_DEBUG
1334# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1335 block
1336# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1337 use iso_fortran_env, only: output_unit
1338# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1339
1340# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1341 print *, 'm_start_up.fpp:883: ', '@:ALLOCATE(data_cmplx_z(Nxloc, Nyloc2, Nz))'
1342# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1343
1344# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1345 call flush (output_unit)
1346# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1347 end block
1348# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1349#endif
1350# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1351 allocate (data_cmplx_z(nxloc, nyloc2, nz))
1352# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1353
1354# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1355
1356# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1357#if defined(MFC_OpenACC)
1358# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1359!$acc enter data create(data_cmplx_z)
1360# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1361#elif defined(MFC_OpenMP)
1362# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1363!$omp target enter data map(always,alloc:data_cmplx_z)
1364# 883 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1365#endif
1366
1367#ifdef MFC_DEBUG
1368# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1369 block
1370# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1371 use iso_fortran_env, only: output_unit
1372# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1373
1374# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1375 print *, 'm_start_up.fpp:885: ', '@:ALLOCATE(En_real(Nxloc, Nyloc2, Nz))'
1376# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1377
1378# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1379 call flush (output_unit)
1380# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1381 end block
1382# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1383#endif
1384# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1385 allocate (en_real(nxloc, nyloc2, nz))
1386# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1387
1388# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1389
1390# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1391#if defined(MFC_OpenACC)
1392# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1393!$acc enter data create(En_real)
1394# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1395#elif defined(MFC_OpenMP)
1396# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1397!$omp target enter data map(always,alloc:En_real)
1398# 885 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1399#endif
1400#ifdef MFC_DEBUG
1401# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1402 block
1403# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1404 use iso_fortran_env, only: output_unit
1405# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1406
1407# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1408 print *, 'm_start_up.fpp:886: ', '@:ALLOCATE(En(Nf))'
1409# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1410
1411# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1412 call flush (output_unit)
1413# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1414 end block
1415# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1416#endif
1417# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1418 allocate (en(nf))
1419# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1420
1421# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1422
1423# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1424#if defined(MFC_OpenACC)
1425# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1426!$acc enter data create(En)
1427# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1428#elif defined(MFC_OpenMP)
1429# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1430!$omp target enter data map(always,alloc:En)
1431# 886 "/home/runner/work/MFC/MFC/src/post_process/m_start_up.fpp"
1432#endif
1433
1434 size_n(1) = nx
1435 inembed(1) = nx
1436 onembed(1) = nx
1437
1438 fwd_plan_x = fftw_plan_many_dft(1, size_n, nyloc*nzloc, data_in, inembed, 1, nx, data_out, onembed, 1, nx, &
1439 & fftw_forward, fftw_measure)
1440
1441 size_n(1) = ny
1442 inembed(1) = ny
1443 onembed(1) = ny
1444
1445 fwd_plan_y = fftw_plan_many_dft(1, size_n, nxloc*nzloc, data_out, inembed, 1, ny, data_in, onembed, 1, ny, &
1446 & fftw_forward, fftw_measure)
1447
1448 size_n(1) = nz
1449 inembed(1) = nz
1450 onembed(1) = nz
1451
1452 fwd_plan_z = fftw_plan_many_dft(1, size_n, nxloc*nyloc2, data_in, inembed, 1, nz, data_out, onembed, 1, nz, &
1453 & fftw_forward, fftw_measure)
1454
1455 call mpi_cart_create(mpi_comm_world, 3, (/num_procs_x, num_procs_y, num_procs_z/), (/.true., .true., .true./), &
1456 & .false., mpi_comm_cart, ierr)
1457 call mpi_cart_coords(mpi_comm_cart, proc_rank, 3, cart3d_coords, ierr)
1458
1459 call mpi_cart_sub(mpi_comm_cart, (/.true., .true., .false./), mpi_comm_cart12, ierr)
1460 call mpi_comm_rank(mpi_comm_cart12, proc_rank12, ierr)
1461 call mpi_cart_coords(mpi_comm_cart12, proc_rank12, 2, cart2d12_coords, ierr)
1462
1463 call mpi_cart_sub(mpi_comm_cart, (/.true., .false., .true./), mpi_comm_cart13, ierr)
1464 call mpi_comm_rank(mpi_comm_cart13, proc_rank13, ierr)
1465 call mpi_cart_coords(mpi_comm_cart13, proc_rank13, 2, cart2d13_coords, ierr)
1466 end if
1467#endif
1468
1469 end subroutine s_initialize_modules
1470
1471 !> Perform a distributed forward 3-D FFT using pencil decomposition with FFTW and MPI transposes.
1472 subroutine s_mpi_fft_fwd
1473
1474 integer :: j, k, l
1475
1476#ifdef MFC_MPI
1477 do l = 1, nzloc
1478 do k = 1, nyloc
1479 do j = 1, nx
1480 data_in(j + (k - 1)*nx + (l - 1)*nx*nyloc) = data_cmplx(j, k, l)
1481 end do
1482 end do
1483 end do
1484
1485 call fftw_execute_dft(fwd_plan_x, data_in, data_out)
1486
1487 do l = 1, nzloc
1488 do k = 1, nyloc
1489 do j = 1, nx
1490 data_cmplx(j, k, l) = data_out(j + (k - 1)*nx + (l - 1)*nx*nyloc)
1491 end do
1492 end do
1493 end do
1494
1495 call s_mpi_transpose_x2y !!Change Pencil from data_cmplx to data_cmpx_y
1496
1497 do l = 1, nzloc
1498 do k = 1, nxloc
1499 do j = 1, ny
1500 data_out(j + (k - 1)*ny + (l - 1)*ny*nxloc) = data_cmplx_y(k, j, l)
1501 end do
1502 end do
1503 end do
1504
1505 call fftw_execute_dft(fwd_plan_y, data_out, data_in)
1506
1507 do l = 1, nzloc
1508 do k = 1, nxloc
1509 do j = 1, ny
1510 data_cmplx_y(k, j, l) = data_in(j + (k - 1)*ny + (l - 1)*ny*nxloc)
1511 end do
1512 end do
1513 end do
1514
1515 call s_mpi_transpose_y2z !!Change Pencil from data_cmplx_y to data_cmpx_z
1516
1517 do l = 1, nyloc2
1518 do k = 1, nxloc
1519 do j = 1, nz
1520 data_in(j + (k - 1)*nz + (l - 1)*nz*nxloc) = data_cmplx_z(k, l, j)
1521 end do
1522 end do
1523 end do
1524
1525 call fftw_execute_dft(fwd_plan_z, data_in, data_out)
1526
1527 do l = 1, nyloc2
1528 do k = 1, nxloc
1529 do j = 1, nz
1530 data_cmplx_z(k, l, j) = data_out(j + (k - 1)*nz + (l - 1)*nz*nxloc)
1531 end do
1532 end do
1533 end do
1534#endif
1535
1536 end subroutine s_mpi_fft_fwd
1537
1538 !> Set up the MPI environment, read and broadcast user inputs, and decompose the computational domain.
1539 impure subroutine s_initialize_mpi_domain
1540
1541 num_dims = 1 + min(1, n) + min(1, p)
1542
1543 call s_mpi_initialize()
1544
1545 if (proc_rank == 0) then
1546 call s_assign_default_values_to_user_inputs()
1547 call s_read_input_file()
1548 call s_check_input_file()
1549
1550 print '(" Post-processing a ", I0, "x", I0, "x", I0, " case on ", I0, " rank(s)")', m, n, p, num_procs
1551 end if
1552
1553 call s_mpi_bcast_user_inputs()
1554 call s_initialize_parallel_io()
1555 call s_mpi_decompose_computational_domain()
1556 call s_check_inputs_fft()
1557
1558 end subroutine s_initialize_mpi_domain
1559
1560 !> Destroy FFTW plans, free MPI communicators, and finalize all post-process sub-modules.
1561 impure subroutine s_finalize_modules
1562
1563 s_read_data_files => null()
1564
1565 if (fft_wrt) then
1566 if (c_associated(fwd_plan_x)) call fftw_destroy_plan(fwd_plan_x)
1567 if (c_associated(fwd_plan_y)) call fftw_destroy_plan(fwd_plan_y)
1568 if (c_associated(fwd_plan_z)) call fftw_destroy_plan(fwd_plan_z)
1569 if (allocated(data_in)) deallocate (data_in)
1570 if (allocated(data_out)) deallocate (data_out)
1571 if (allocated(data_cmplx)) deallocate (data_cmplx)
1572 if (allocated(data_cmplx_y)) deallocate (data_cmplx_y)
1573 if (allocated(data_cmplx_z)) deallocate (data_cmplx_z)
1574 if (allocated(en_real)) deallocate (en_real)
1575 if (allocated(en)) deallocate (en)
1576 call fftw_cleanup()
1577 end if
1578
1579#ifdef MFC_MPI
1580 if (fft_wrt) then
1581 if (mpi_comm_cart12 /= mpi_comm_null) call mpi_comm_free(mpi_comm_cart12, ierr)
1582 if (mpi_comm_cart13 /= mpi_comm_null) call mpi_comm_free(mpi_comm_cart13, ierr)
1583 if (mpi_comm_cart /= mpi_comm_null) call mpi_comm_free(mpi_comm_cart, ierr)
1584 end if
1585#endif
1586
1587 call s_finalize_data_output_module()
1588 call s_finalize_derived_variables_module()
1589 call s_finalize_data_input_module()
1590 call s_finalize_variables_conversion_module()
1591 if (num_procs > 1) then
1592 call s_finalize_mpi_proxy_module()
1593 call s_finalize_mpi_common_module()
1594 end if
1595 call s_finalize_global_parameters_module()
1596
1597 call s_mpi_finalize()
1598
1599 end subroutine s_finalize_modules
1600
1601end module m_start_up
1602
integer, intent(in) k
integer, intent(in) j
integer, intent(in) l
Noncharacteristic and processor boundary condition application for ghost cells and buffer regions.
impure subroutine, public s_initialize_boundary_common_module()
Allocate and set up boundary condition buffer arrays for all coordinate directions.
impure subroutine, public s_populate_variables_buffers(bc_type, q_prim_vf, pb_in, mv_in, q_t_sf)
Populate the buffers of the primitive variables based on the selected boundary conditions.
Shared input validation checks for grid dimensions and AMD GPU compiler limits.
impure subroutine, public s_check_inputs_common
Checks compatibility of parameters in the input file. Used by all three stages.
Validates post-process input parameters and output format consistency.
impure subroutine, public s_check_inputs
Checks compatibility of parameters in the input file. Used by the post_process stage.
Multi-species chemistry interface for thermodynamic properties, reaction rates, and transport coeffic...
subroutine s_compute_q_t_sf(q_t_sf, q_cons_vf, bounds)
Initialize the temperature field from conservative variables by inverting the energy equation.
Platform-specific file and directory operations: create, delete, inquire, getcwd, and basename.
impure subroutine s_delete_file(filepath)
Delete a file at the given path using a platform-specific system command.
impure subroutine my_inquire(fileloc, dircheck)
Inquires on the existence of a directory.
impure subroutine s_create_directory(dir_name)
Create a directory and all its parents if it does not exist.
Compile-time constant parameters: default values, tolerances, and physical constants.
integer, parameter model_eqns_4eq
integer, parameter model_eqns_5eq
integer, parameter model_eqns_6eq
integer, parameter model_eqns_gamma_law
Reads raw simulation grid and conservative-variable data for a given time-step and fills buffer regio...
impure subroutine, public s_read_parallel_data_files(t_step)
Parallel-read the raw data files present in the corresponding time-step directory and to populate the...
type(scalar_field), public q_t_sf
Temperature field.
type(scalar_field), dimension(:), allocatable, public q_cons_vf
Conservative variables.
impure subroutine, public s_initialize_data_input_module
Computation of parameters, allocation procedures, and/or any other tasks needed to properly setup the...
type(scalar_field), dimension(:), allocatable, public q_prim_vf
Primitive variables.
impure subroutine, public s_read_serial_data_files(t_step)
Read the raw data files present in the corresponding time-step directory and to populate the associat...
procedure(s_read_abstract_data_files), pointer, public s_read_data_files
type(integer_field), dimension(:,:), allocatable, public bc_type
Boundary condition identifiers.
type(integer_field), public ib_markers
Writes post-processed grid and flow-variable data to Silo-HDF5 or binary database files.
impure subroutine, public s_write_grid_to_formatted_database_file(t_step)
Write the computational grid (cell-boundary coordinates) to the formatted database slave and master f...
impure subroutine, public s_write_variable_to_formatted_database_file(varname, t_step)
Write a single flow variable field to the formatted database slave and master files for a given time ...
impure subroutine, public s_open_energy_data_file()
Open the energy data file for appending volume-integrated energy budget quantities.
impure subroutine, public s_open_intf_data_file()
Open the interface data file for appending extracted interface coordinates.
impure subroutine, public s_write_energy_data_file(q_prim_vf, q_cons_vf)
Compute volume-integrated kinetic, potential, and internal energies and write the energy budget to th...
impure subroutine, public s_write_lag_bubbles_to_formatted_database_file(t_step)
Read Lagrangian bubble restart data and write bubble positions and scalar fields to the Silo database...
impure subroutine, public s_write_intf_data_file(q_prim_vf)
Extract the volume-fraction interface contour from primitive fields and write the coordinates to the ...
impure subroutine, public s_initialize_data_output_module()
Allocate storage arrays, configure output directories, and count flow variables for formatted databas...
impure subroutine, public s_close_energy_data_file()
Close the energy data file.
impure subroutine, public s_close_formatted_database_file()
Close the formatted database slave file and, for the root process, the master file.
impure subroutine, public s_open_formatted_database_file(t_step)
Open (or create) the Silo-HDF5 or Binary formatted database slave and master files for a given time s...
impure subroutine, public s_close_intf_data_file()
Close the interface data file.
impure subroutine, public s_write_lag_bubbles_results_to_text(t_step)
Write the post-processed results in the folder 'lag_bubbles_data'.
impure subroutine, public s_define_output_region
Compute the cell-index bounds for the user-specified partial output domain in each coordinate directi...
impure subroutine, public s_write_ib_bodies_to_formatted_database_file(t_step)
Read IB state and write a Silo point mesh with per-body scalar fields.
type(output_context), public out
Output workspace: flow variable buffers, VisIt extents/offsets, directory paths, file handles,...
Shared derived types for field data, patch geometry, bubble dynamics, and MPI I/O structures.
Computes derived flow quantities (sound speed, vorticity, Schlieren, etc.) from conservative and prim...
type(fd_context), public fd
Finite-difference state: density gradient magnitude and centered FD coefficients in x-,...
impure subroutine, public s_derive_liutex(q_prim_vf, liutex_mag, liutex_axis)
Compute the Liutex vector and its magnitude based on Xu et al. (2019).
subroutine, public s_derive_specific_heat_ratio(q_sf)
Derive the specific heat ratio from the specific heat ratio function gamma_sf. The latter is stored i...
subroutine, public s_derive_liquid_stiffness(q_sf)
Compute the liquid stiffness from the specific heat ratio function gamma_sf and the liquid stiffness ...
impure subroutine, public s_initialize_derived_variables_module
Computation of parameters, allocation procedures, and/or any other tasks needed to properly setup the...
subroutine, public s_derive_vorticity_component(i, q_prim_vf, q_sf)
Compute the specified component of the vorticity from the primitive variables. From those inputs,...
impure subroutine, public s_derive_numerical_schlieren_function(q_cons_vf, q_sf)
Compute the values of the numerical Schlieren function, which are subsequently stored in the derived ...
subroutine, public s_derive_flux_limiter(i, q_prim_vf, q_sf)
Derive the flux limiter at cell boundary i+1/2. This is an approximation because the velocity used to...
subroutine, public s_derive_qm(q_prim_vf, q_sf)
Compute the Q_M criterion from the primitive variables. The Q_M function, which are subsequently stor...
Finite difference operators for computing divergence of velocity fields.
subroutine s_compute_finite_difference_coefficients(q, s_cc, fd_coeff_s, local_buff_size, fd_number_in, fd_order_in, offset_s)
Compute the centered finite-difference coefficients for first-order spatial derivatives in the s-coor...
Global parameters for the post-process: domain geometry, equation of state, and output database setti...
integer beta_idx
Index of lagrange bubbles beta.
type(int_bounds_info) offset_y
type(qbmm_idx_info) qbmm_idx
QBMM moment index mappings.
real(wp), dimension(:), allocatable y_cc
integer proc_rank
Rank of the local processor.
real(wp), dimension(:), allocatable adv
Advection variables.
type(int_bounds_info) z_output_idx
Indices of domain to output for post-processing.
integer fd_number
Finite-difference half-stencil size: MAX(1, fd_order/2).
type(int_bounds_info), dimension(1:3) idwbuff
integer buff_size
Number of ghost cells for boundary condition storage.
type(bounds_info) z_output
Portion of domain to output for post-processing.
type(int_bounds_info) x_output_idx
impure subroutine s_initialize_global_parameters_module
Computation of parameters, allocation procedures, and/or any other tasks needed to properly setup the...
real(wp), dimension(:), allocatable x_cc
type(int_bounds_info) offset_x
real(wp), dimension(:), allocatable z_cc
integer num_procs
Number of processors.
type(int_bounds_info) y_output_idx
type(int_bounds_info) offset_z
type(cell_num_bounds) cells_bounds
real(wp) wall_time_avg
Wall time measurements.
integer(kind=8) nglobal
Total number of cells in global domain.
Utility routines for bubble model setup, coordinate transforms, array sampling, and special functions...
impure subroutine, public s_initialize_bubbles_model()
Initialize bubble model arrays for Euler or Lagrangian bubbles with polytropic or non-polytropic gas.
MPI communication layer: domain decomposition, halo exchange, reductions, and parallel I/O setup.
impure subroutine s_mpi_abort(prnt, code)
The subroutine terminates the MPI execution environment.
impure subroutine s_initialize_mpi_common_module
Initialize the module.
MPI gather and scatter operations for distributing post-process grid and flow-variable data.
impure subroutine s_initialize_mpi_proxy_module
Computation of parameters, allocation procedures, and/or any other tasks needed to properly setup the...
Reads and validates user inputs, allocates variables, and configures MPI decomposition and I/O for po...
impure subroutine s_save_data(t_step, varname, pres, c, h)
Derive requested flow quantities from primitive variables and write them to the formatted database fi...
impure subroutine s_check_input_file
Checking that the user inputs make sense, i.e. that the individual choices are compatible with the co...
real(wp), dimension(:), allocatable en
complex(c_double_complex), dimension(:,:,:), allocatable data_cmplx_y
subroutine s_mpi_fft_fwd
Perform a distributed forward 3-D FFT using pencil decomposition with FFTW and MPI transposes.
type(c_ptr) fwd_plan_y
integer mpi_comm_cart13
integer num_procs_x
impure subroutine s_initialize_mpi_domain
Set up the MPI environment, read and broadcast user inputs, and decompose the computational domain.
complex(c_double_complex), dimension(:,:,:), allocatable data_cmplx_z
impure subroutine s_read_input_file
Reads the configuration file post_process.inp, in order to populate parameters in module m_global_par...
complex(c_double_complex), dimension(:), allocatable data_out
integer, dimension(2) cart2d13_coords
type(c_ptr) fwd_plan_z
integer num_procs_z
real(wp), dimension(:,:,:), allocatable en_real
complex(c_double_complex), dimension(:), allocatable data_in
integer, dimension(3) cart3d_coords
impure subroutine s_perform_time_step(t_step)
Load grid and conservative data for a time step, fill ghost-cell buffers, and convert to primitive va...
integer num_procs_y
integer mpi_comm_cart
complex(c_double_complex), dimension(:,:,:), allocatable data_cmplx
integer proc_rank12
subroutine s_mpi_transpose_x2y
Transpose 3-D complex data from x-pencil to y-pencil layout via MPI_Alltoall.
impure subroutine s_finalize_modules
Destroy FFTW plans, free MPI communicators, and finalize all post-process sub-modules.
subroutine s_mpi_transpose_y2z
Transpose 3-D complex data from y-pencil to z-pencil layout via MPI_Alltoall.
impure subroutine s_initialize_modules
Initialize all post-process sub-modules, set up I/O pointers, and prepare FFTW plans and MPI communic...
integer mpi_comm_cart12
type(c_ptr) fwd_plan_x
integer proc_rank13
integer, dimension(2) cart2d12_coords
Conservative-to-primitive variable conversion, mixture property evaluation, and pressure computation.
subroutine s_compute_speed_of_sound(pres, rho, gamma, pi_inf, h, adv, vel_sum, c_c, c, qv)
Compute the speed of sound from thermodynamic state variables, supporting multiple equation-of-state ...
subroutine, public s_convert_conservative_to_primitive_variables(qk_cons_vf, q_t_sf, qk_prim_vf, ibounds)
Convert conserved variables (rho*alpha, rho*u, E, alpha) to primitives (rho, u, p,...
impure subroutine, public s_initialize_variables_conversion_module
Initialize the variables conversion module.
real(wp), dimension(:,:,:), allocatable, public qv_sf
Scalar liquid energy reference function.
real(wp), dimension(:,:,:), allocatable, public pi_inf_sf
Scalar liquid stiffness function.
real(wp), dimension(:,:,:), allocatable, public gamma_sf
Scalar sp. heat ratio function.
real(wp), dimension(:,:,:), allocatable, public rho_sf
Scalar density function.