Neko 0.9.1
A portable framework for high-order spectral element flow simulations
Loading...
Searching...
No Matches
fluid_stats.f90
Go to the documentation of this file.
1! Copyright (c) 2022-2024, The Neko Authors
2! All rights reserved.
3!
4! Redistribution and use in source and binary forms, with or without
5! modification, are permitted provided that the following conditions
6! are met:
7!
8! * Redistributions of source code must retain the above copyright
9! notice, this list of conditions and the following disclaimer.
10!
11! * Redistributions in binary form must reproduce the above
12! copyright notice, this list of conditions and the following
13! disclaimer in the documentation and/or other materials provided
14! with the distribution.
15!
16! * Neither the name of the authors nor the names of its
17! contributors may be used to endorse or promote products derived
18! from this software without specific prior written permission.
19!
20! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21! "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22! LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23! FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24! COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
25! INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
26! BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
27! LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
28! CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29! LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
30! ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
31! POSSIBILITY OF SUCH DAMAGE.
32!
36module fluid_stats
37 use mean_field, only : mean_field_t
38 use device_math, only : device_col3, device_col2, device_cfill, &
39 device_invcol2, device_addcol3
40 use num_types, only : rp
41 use math, only : invers2, col2, addcol3, col3, copy, subcol3
42 use operators, only : opgrad
43 use coefs, only : coef_t
44 use field, only : field_t
45 use field_list, only : field_list_t
46 use stats_quant, only : stats_quant_t
47 use device
48 use neko_config, only : neko_bcknd_device
49 use utils, only : neko_warning
50 implicit none
51 private
52
53 type, public, extends(stats_quant_t) :: fluid_stats_t
55 type(field_t) :: stats_u
56 type(field_t) :: stats_v
57 type(field_t) :: stats_w
58 type(field_t) :: stats_p
59 type(field_t) :: stats_work
60
62 type(field_t), pointer :: u
63 type(field_t), pointer :: v
64 type(field_t), pointer :: w
65 type(field_t), pointer :: p
66
67 type(mean_field_t) :: u_mean
68 type(mean_field_t) :: v_mean
69 type(mean_field_t) :: w_mean
70 type(mean_field_t) :: p_mean
72 type(mean_field_t) :: uu
73 type(mean_field_t) :: vv
74 type(mean_field_t) :: ww
75 type(mean_field_t) :: uv
76 type(mean_field_t) :: uw
77 type(mean_field_t) :: vw
79 type(mean_field_t) :: uuu
80 type(mean_field_t) :: vvv
81 type(mean_field_t) :: www
82 type(mean_field_t) :: uuv
83 type(mean_field_t) :: uuw
84 type(mean_field_t) :: uvv
85 type(mean_field_t) :: uvw
86 type(mean_field_t) :: vvw
87 type(mean_field_t) :: uww
88 type(mean_field_t) :: vww
90 type(mean_field_t) :: uuuu
91 type(mean_field_t) :: vvvv
92 type(mean_field_t) :: wwww
94 type(mean_field_t) :: pp
95 type(mean_field_t) :: ppp
96 type(mean_field_t) :: pppp
98 type(mean_field_t) :: pu
99 type(mean_field_t) :: pv
100 type(mean_field_t) :: pw
101
103 type(mean_field_t) :: pdudx
104 type(mean_field_t) :: pdudy
105 type(mean_field_t) :: pdudz
106 type(mean_field_t) :: pdvdx
107 type(mean_field_t) :: pdvdy
108 type(mean_field_t) :: pdvdz
109 type(mean_field_t) :: pdwdx
110 type(mean_field_t) :: pdwdy
111 type(mean_field_t) :: pdwdz
112
114 type(mean_field_t) :: e11
115 type(mean_field_t) :: e22
116 type(mean_field_t) :: e33
117 type(mean_field_t) :: e12
118 type(mean_field_t) :: e13
119 type(mean_field_t) :: e23
121 type(field_t) :: dudx
122 type(field_t) :: dudy
123 type(field_t) :: dudz
124 type(field_t) :: dvdx
125 type(field_t) :: dvdy
126 type(field_t) :: dvdz
127 type(field_t) :: dwdx
128 type(field_t) :: dwdy
129 type(field_t) :: dwdz
130
132 type(coef_t), pointer :: coef
134 integer :: n_stats = 44
137 character(5) :: stat_set
140 type(field_list_t) :: stat_fields
141 contains
143 procedure, pass(this) :: init => fluid_stats_init
145 procedure, pass(this) :: free => fluid_stats_free
147 procedure, pass(this) :: update => fluid_stats_update
149 procedure, pass(this) :: reset => fluid_stats_reset
150 ! Convert computed weak gradients to strong.
151 procedure, pass(this) :: make_strong_grad => fluid_stats_make_strong_grad
154 procedure, pass(this) :: post_process => fluid_stats_post_process
155 end type fluid_stats_t
156
157contains
158
167 subroutine fluid_stats_init(this, coef, u, v, w, p, set)
168 class(fluid_stats_t), intent(inout), target:: this
169 type(coef_t), target, optional :: coef
170 type(field_t), target, intent(inout) :: u, v, w, p
171 character(*), intent(in), optional :: set
172
173 call this%free()
174 this%coef => coef
175
176 this%u => u
177 this%v => v
178 this%w => w
179 this%p => p
180
181 if (present(set)) then
182 this%stat_set = trim(set)
183 if (this%stat_set .eq. 'basic') then
184 this%n_stats = 11
185 end if
186 else
187 this%stat_set = 'full'
188 this%n_stats = 44
189 end if
190
191 call this%stats_work%init(this%u%dof, 'stats')
192 call this%stats_u%init(this%u%dof, 'u temp')
193 call this%stats_v%init(this%u%dof, 'v temp')
194 call this%stats_w%init(this%u%dof, 'w temp')
195 call this%stats_p%init(this%u%dof, 'p temp')
196 call this%u_mean%init(this%u)
197 call this%v_mean%init(this%v)
198 call this%w_mean%init(this%w)
199 call this%p_mean%init(this%p)
200 call this%uu%init(this%stats_u, 'uu')
201 call this%vv%init(this%stats_v, 'vv')
202 call this%ww%init(this%stats_w, 'ww')
203 call this%uv%init(this%stats_work, 'uv')
204 call this%uw%init(this%stats_work, 'uw')
205 call this%vw%init(this%stats_work, 'vw')
206 call this%pp%init(this%stats_p, 'pp')
207
208 if (this%n_stats .eq. 44) then
209 call this%dudx%init(this%u%dof, 'dudx')
210 call this%dudy%init(this%u%dof, 'dudy')
211 call this%dudz%init(this%u%dof, 'dudz')
212 call this%dvdx%init(this%u%dof, 'dvdx')
213 call this%dvdy%init(this%u%dof, 'dvdy')
214 call this%dvdz%init(this%u%dof, 'dvdz')
215 call this%dwdx%init(this%u%dof, 'dwdx')
216 call this%dwdy%init(this%u%dof, 'dwdy')
217 call this%dwdz%init(this%u%dof, 'dwdz')
218
219 call this%uuu%init(this%stats_work, 'uuu')
220 call this%vvv%init(this%stats_work, 'vvv')
221 call this%www%init(this%stats_work, 'www')
222 call this%uuv%init(this%stats_work, 'uuv')
223 call this%uuw%init(this%stats_work, 'uuw')
224 call this%uvv%init(this%stats_work, 'uvv')
225 call this%uvw%init(this%stats_work, 'uvw')
226 call this%vvw%init(this%stats_work, 'vvw')
227 call this%uww%init(this%stats_work, 'uww')
228 call this%vww%init(this%stats_work, 'vww')
229 call this%uuuu%init(this%stats_work, 'uuuu')
230 call this%vvvv%init(this%stats_work, 'vvvv')
231 call this%wwww%init(this%stats_work, 'wwww')
233 call this%ppp%init(this%stats_work, 'ppp')
234 call this%pppp%init(this%stats_work, 'pppp')
236 call this%pu%init(this%stats_work, 'pu')
237 call this%pv%init(this%stats_work, 'pv')
238 call this%pw%init(this%stats_work, 'pw')
239
240 call this%pdudx%init(this%stats_work, 'pdudx')
241 call this%pdudy%init(this%stats_work, 'pdudy')
242 call this%pdudz%init(this%stats_work, 'pdudz')
243 call this%pdvdx%init(this%stats_work, 'pdvdx')
244 call this%pdvdy%init(this%stats_work, 'pdvdy')
245 call this%pdvdz%init(this%stats_work, 'pdvdz')
246 call this%pdwdx%init(this%stats_work, 'pdwdx')
247 call this%pdwdy%init(this%stats_work, 'pdwdy')
248 call this%pdwdz%init(this%stats_work, 'pdwdz')
249
250 call this%e11%init(this%stats_work, 'e11')
251 call this%e22%init(this%stats_work, 'e22')
252 call this%e33%init(this%stats_work, 'e33')
253 call this%e12%init(this%stats_work, 'e12')
254 call this%e13%init(this%stats_work, 'e13')
255 call this%e23%init(this%stats_work, 'e23')
256 end if
257
258 allocate(this%stat_fields%items(this%n_stats))
259
260 call this%stat_fields%assign_to_field(1, this%p_mean%mf)
261 call this%stat_fields%assign_to_field(2, this%u_mean%mf)
262 call this%stat_fields%assign_to_field(3, this%v_mean%mf)
263 call this%stat_fields%assign_to_field(4, this%w_mean%mf)
264 call this%stat_fields%assign_to_field(5, this%pp%mf)
265 call this%stat_fields%assign_to_field(6, this%uu%mf)
266 call this%stat_fields%assign_to_field(7, this%vv%mf)
267 call this%stat_fields%assign_to_field(8, this%ww%mf)
268 call this%stat_fields%assign_to_field(9, this%uv%mf)
269 call this%stat_fields%assign_to_field(10, this%uw%mf)
270 call this%stat_fields%assign_to_field(11, this%vw%mf)
271
272 if (this%n_stats .eq. 44) then
273 call this%stat_fields%assign_to_field(12, this%uuu%mf)
274 call this%stat_fields%assign_to_field(13, this%vvv%mf)
275 call this%stat_fields%assign_to_field(14, this%www%mf)
276 call this%stat_fields%assign_to_field(15, this%uuv%mf)
277 call this%stat_fields%assign_to_field(16, this%uuw%mf)
278 call this%stat_fields%assign_to_field(17, this%uvv%mf)
279 call this%stat_fields%assign_to_field(18, this%uvw%mf)
280 call this%stat_fields%assign_to_field(19, this%vvw%mf)
281 call this%stat_fields%assign_to_field(20, this%uww%mf)
282 call this%stat_fields%assign_to_field(21, this%vww%mf)
283 call this%stat_fields%assign_to_field(22, this%uuuu%mf)
284 call this%stat_fields%assign_to_field(23, this%vvvv%mf)
285 call this%stat_fields%assign_to_field(24, this%wwww%mf)
286 call this%stat_fields%assign_to_field(25, this%ppp%mf)
287 call this%stat_fields%assign_to_field(26, this%pppp%mf)
288 call this%stat_fields%assign_to_field(27, this%pu%mf)
289 call this%stat_fields%assign_to_field(28, this%pv%mf)
290 call this%stat_fields%assign_to_field(29, this%pw%mf)
291
292 call this%stat_fields%assign_to_field(30, this%pdudx%mf)
293 call this%stat_fields%assign_to_field(31, this%pdudy%mf)
294 call this%stat_fields%assign_to_field(32, this%pdudz%mf)
295 call this%stat_fields%assign_to_field(33, this%pdvdx%mf)
296 call this%stat_fields%assign_to_field(34, this%pdvdy%mf)
297 call this%stat_fields%assign_to_field(35, this%pdvdz%mf)
298 call this%stat_fields%assign_to_field(36, this%pdwdx%mf)
299 call this%stat_fields%assign_to_field(37, this%pdwdy%mf)
300 call this%stat_fields%assign_to_field(38, this%pdwdz%mf)
301 call this%stat_fields%assign_to_field(39, this%e11%mf)
302 call this%stat_fields%assign_to_field(40, this%e22%mf)
303 call this%stat_fields%assign_to_field(41, this%e33%mf)
304 call this%stat_fields%assign_to_field(42, this%e12%mf)
305 call this%stat_fields%assign_to_field(43, this%e13%mf)
306 call this%stat_fields%assign_to_field(44, this%e23%mf)
307 end if
308
309 end subroutine fluid_stats_init
310
313 subroutine fluid_stats_update(this, k)
314 class(fluid_stats_t), intent(inout) :: this
315 real(kind=rp), intent(in) :: k
316 integer :: n
317
318 associate(stats_work => this%stats_work, stats_u => this%stats_u, &
319 stats_v => this%stats_v, stats_w => this%stats_w, &
320 stats_p => this%stats_p)
321 n = stats_work%dof%size()
322
324 if (neko_bcknd_device .eq. 1) then
325
326 call this%u_mean%update(k)
327 call this%v_mean%update(k)
328 call this%w_mean%update(k)
329 call this%p_mean%update(k)
330
331 call device_col3(stats_u%x_d, this%u%x_d, this%u%x_d, n)
332 call device_col3(stats_v%x_d, this%v%x_d, this%v%x_d, n)
333 call device_col3(stats_w%x_d, this%w%x_d, this%w%x_d, n)
334 call device_col3(stats_p%x_d, this%p%x_d, this%p%x_d, n)
335
336 call this%uu%update(k)
337 call this%vv%update(k)
338 call this%ww%update(k)
339 call this%pp%update(k)
340
341 call device_col3(stats_work%x_d, this%u%x_d, this%v%x_d, n)
342 call this%uv%update(k)
343 call device_col3(stats_work%x_d, this%u%x_d, this%w%x_d, n)
344 call this%uw%update(k)
345 call device_col3(stats_work%x_d, this%v%x_d, this%w%x_d, n)
346 call this%vw%update(k)
347 if (this%n_stats .eq. 11) return
348 call device_col2(stats_work%x_d, this%u%x_d, n)
349 call this%uvw%update(k)
350 call device_col3(stats_work%x_d, this%stats_u%x_d, this%u%x_d, n)
351 call this%uuu%update(k)
352 call device_col3(stats_work%x_d, this%stats_v%x_d, this%v%x_d, n)
353 call this%vvv%update(k)
354 call device_col3(stats_work%x_d, this%stats_w%x_d, this%w%x_d, n)
355 call this%www%update(k)
356 call device_col3(stats_work%x_d, this%stats_u%x_d, this%v%x_d, n)
357 call this%uuv%update(k)
358 call device_col3(stats_work%x_d, this%stats_u%x_d, this%w%x_d, n)
359 call this%uuw%update(k)
360 call device_col3(stats_work%x_d, this%stats_v%x_d, this%u%x_d, n)
361 call this%uvv%update(k)
362 call device_col3(stats_work%x_d, this%stats_v%x_d, this%w%x_d, n)
363 call this%vvw%update(k)
364 call device_col3(stats_work%x_d, this%stats_w%x_d, this%u%x_d, n)
365 call this%uww%update(k)
366 call device_col3(stats_work%x_d, this%stats_w%x_d, this%v%x_d, n)
367 call this%vww%update(k)
368
369 call device_col3(stats_work%x_d, this%stats_u%x_d, this%stats_u%x_d, n)
370 call this%uuuu%update(k)
371 call device_col3(stats_work%x_d, this%stats_v%x_d, this%stats_v%x_d, n)
372 call this%vvvv%update(k)
373 call device_col3(stats_work%x_d, this%stats_w%x_d, this%stats_w%x_d, n)
374 call this%wwww%update(k)
375
376 call device_col3(stats_work%x_d, this%stats_p%x_d, this%p%x_d, n)
377 call this%ppp%update(k)
378 call device_col3(stats_work%x_d, this%stats_p%x_d, this%stats_p%x_d, n)
379 call this%pppp%update(k)
380
381 call device_col3(stats_work%x_d, this%p%x_d, this%u%x_d, n)
382 call this%pu%update(k)
383 call device_col3(stats_work%x_d, this%p%x_d, this%v%x_d, n)
384 call this%pv%update(k)
385 call device_col3(stats_work%x_d, this%p%x_d, this%w%x_d, n)
386 call this%pw%update(k)
387
388 else
389
390 call this%u_mean%update(k)
391 call this%v_mean%update(k)
392 call this%w_mean%update(k)
393 call this%p_mean%update(k)
394 call col3(stats_u%x, this%u%x, this%u%x, n)
395 call col3(stats_v%x, this%v%x, this%v%x, n)
396 call col3(stats_w%x, this%w%x, this%w%x, n)
397 call col3(stats_p%x, this%p%x, this%p%x, n)
398
399 call this%uu%update(k)
400 call this%vv%update(k)
401 call this%ww%update(k)
402 call this%pp%update(k)
403
404 call col3(stats_work%x, this%u%x, this%v%x, n)
405 call this%uv%update(k)
406 call col3(stats_work%x, this%u%x, this%w%x, n)
407 call this%uw%update(k)
408 call col3(stats_work%x, this%v%x, this%w%x, n)
409 call this%vw%update(k)
410
411 if (this%n_stats .eq. 11) return
412
413 call col2(stats_work%x, this%u%x, n)
414 call this%uvw%update(k)
415 call col3(stats_work%x, this%stats_u%x, this%u%x, n)
416 call this%uuu%update(k)
417 call col3(stats_work%x, this%stats_v%x, this%v%x, n)
418 call this%vvv%update(k)
419 call col3(stats_work%x, this%stats_w%x, this%w%x, n)
420 call this%www%update(k)
421 call col3(stats_work%x, this%stats_u%x, this%v%x, n)
422 call this%uuv%update(k)
423 call col3(stats_work%x, this%stats_u%x, this%w%x, n)
424 call this%uuw%update(k)
425 call col3(stats_work%x, this%stats_v%x, this%u%x, n)
426 call this%uvv%update(k)
427 call col3(stats_work%x, this%stats_v%x, this%w%x, n)
428 call this%vvw%update(k)
429 call col3(stats_work%x, this%stats_w%x, this%u%x, n)
430 call this%uww%update(k)
431 call col3(stats_work%x, this%stats_w%x, this%v%x, n)
432 call this%vww%update(k)
433
434 call col3(stats_work%x, this%stats_u%x, this%stats_u%x, n)
435 call this%uuuu%update(k)
436 call col3(stats_work%x, this%stats_v%x, this%stats_v%x, n)
437 call this%vvvv%update(k)
438 call col3(stats_work%x, this%stats_w%x, this%stats_w%x, n)
439 call this%wwww%update(k)
440
441 call col3(stats_work%x, this%stats_p%x, this%p%x, n)
442 call this%ppp%update(k)
443 call col3(stats_work%x, this%stats_p%x, this%stats_p%x, n)
444 call this%pppp%update(k)
445
446 call col3(stats_work%x, this%p%x, this%u%x,n)
447 call this%pu%update(k)
448 call col3(stats_work%x, this%p%x, this%v%x,n)
449 call this%pv%update(k)
450 call col3(stats_work%x, this%p%x, this%w%x,n)
451 call this%pw%update(k)
452
453
454 end if
455 call opgrad(this%dudx%x, this%dudy%x, this%dudz%x, this%u%x, this%coef)
456 call opgrad(this%dvdx%x, this%dvdy%x, this%dvdz%x, this%v%x, this%coef)
457 call opgrad(this%dwdx%x, this%dwdy%x, this%dwdz%x, this%w%x, this%coef)
458
459 if (neko_bcknd_device .eq. 1) then
460 call device_col3(stats_work%x_d, this%dudx%x_d, this%p%x_d, n)
461 call this%pdudx%update(k)
462 call device_col3(stats_work%x_d, this%dudy%x_d, this%p%x_d, n)
463 call this%pdudy%update(k)
464 call device_col3(stats_work%x_d, this%dudz%x_d, this%p%x_d, n)
465 call this%pdudz%update(k)
466
467 call device_col3(stats_work%x_d, this%dvdx%x_d, this%p%x_d, n)
468 call this%pdvdx%update(k)
469 call device_col3(stats_work%x_d, this%dvdy%x_d, this%p%x_d, n)
470 call this%pdvdy%update(k)
471 call device_col3(stats_work%x_d, this%dvdz%x_d, this%p%x_d, n)
472 call this%pdvdz%update(k)
473
474 call device_col3(stats_work%x_d, this%dwdx%x_d, this%p%x_d, n)
475 call this%pdwdx%update(k)
476 call device_col3(stats_work%x_d, this%dwdy%x_d, this%p%x_d, n)
477 call this%pdwdy%update(k)
478 call device_col3(stats_work%x_d, this%dwdz%x_d, this%p%x_d, n)
479 call this%pdwdz%update(k)
480
481 call device_col3(this%stats_work%x_d, this%dudx%x_d, this%dudx%x_d, n)
482 call device_addcol3(this%stats_work%x_d, this%dudy%x_d, &
483 this%dudy%x_d, n)
484 call device_addcol3(this%stats_work%x_d, this%dudz%x_d, &
485 this%dudz%x_d, n)
486 call this%e11%update(k)
487 call device_col3(this%stats_work%x_d, this%dvdx%x_d, this%dvdx%x_d, n)
488 call device_addcol3(this%stats_work%x_d, this%dvdy%x_d, &
489 this%dvdy%x_d, n)
490 call device_addcol3(this%stats_work%x_d, this%dvdz%x_d, &
491 this%dvdz%x_d, n)
492 call this%e22%update(k)
493 call device_col3(this%stats_work%x_d, this%dwdx%x_d, this%dwdx%x_d, n)
494 call device_addcol3(this%stats_work%x_d, this%dwdy%x_d, &
495 this%dwdy%x_d, n)
496 call device_addcol3(this%stats_work%x_d, this%dwdz%x_d, &
497 this%dwdz%x_d, n)
498 call this%e33%update(k)
499 call device_col3(this%stats_work%x_d, this%dudx%x_d, &
500 this%dvdx%x_d, n)
501 call device_addcol3(this%stats_work%x_d, this%dudy%x_d, &
502 this%dvdy%x_d, n)
503 call device_addcol3(this%stats_work%x_d, this%dudz%x_d, &
504 this%dvdz%x_d, n)
505 call this%e12%update(k)
506 call device_col3(this%stats_work%x_d, this%dudx%x_d, this%dwdx%x_d, n)
507 call device_addcol3(this%stats_work%x_d, this%dudy%x_d, &
508 this%dwdy%x_d, n)
509 call device_addcol3(this%stats_work%x_d, this%dudz%x_d, &
510 this%dwdz%x_d, n)
511 call this%e13%update(k)
512 call device_col3(this%stats_work%x_d, this%dvdx%x_d, this%dwdx%x_d, n)
513 call device_addcol3(this%stats_work%x_d, this%dvdy%x_d, &
514 this%dwdy%x_d, n)
515 call device_addcol3(this%stats_work%x_d, this%dvdz%x_d, &
516 this%dwdz%x_d, n)
517 call this%e23%update(k)
518 else
519 call col3(stats_work%x, this%dudx%x, this%p%x, n)
520 call this%pdudx%update(k)
521 call col3(stats_work%x, this%dudy%x, this%p%x, n)
522 call this%pdudy%update(k)
523 call col3(stats_work%x, this%dudz%x, this%p%x, n)
524 call this%pdudz%update(k)
525
526 call col3(stats_work%x, this%dvdx%x, this%p%x, n)
527 call this%pdvdx%update(k)
528 call col3(stats_work%x, this%dvdy%x, this%p%x, n)
529 call this%pdvdy%update(k)
530 call col3(stats_work%x, this%dvdz%x, this%p%x, n)
531 call this%pdvdz%update(k)
532
533 call col3(stats_work%x, this%dwdx%x, this%p%x, n)
534 call this%pdwdx%update(k)
535 call col3(stats_work%x, this%dwdy%x, this%p%x, n)
536 call this%pdwdy%update(k)
537 call col3(stats_work%x, this%dwdz%x, this%p%x, n)
538 call this%pdwdz%update(k)
539
540 call col3(this%stats_work%x, this%dudx%x, this%dudx%x, n)
541 call addcol3(this%stats_work%x, this%dudy%x, this%dudy%x, n)
542 call addcol3(this%stats_work%x, this%dudz%x, this%dudz%x, n)
543 call this%e11%update(k)
544 call col3(this%stats_work%x, this%dvdx%x, this%dvdx%x, n)
545 call addcol3(this%stats_work%x, this%dvdy%x, this%dvdy%x, n)
546 call addcol3(this%stats_work%x, this%dvdz%x, this%dvdz%x, n)
547 call this%e22%update(k)
548 call col3(this%stats_work%x, this%dwdx%x, this%dwdx%x, n)
549 call addcol3(this%stats_work%x, this%dwdy%x, this%dwdy%x, n)
550 call addcol3(this%stats_work%x, this%dwdz%x, this%dwdz%x, n)
551 call this%e33%update(k)
552 call col3(this%stats_work%x, this%dudx%x, this%dvdx%x, n)
553 call addcol3(this%stats_work%x, this%dudy%x, this%dvdy%x, n)
554 call addcol3(this%stats_work%x, this%dudz%x, this%dvdz%x, n)
555 call this%e12%update(k)
556 call col3(this%stats_work%x,this%dudx%x, this%dwdx%x,n)
557 call addcol3(this%stats_work%x,this%dudy%x, this%dwdy%x,n)
558 call addcol3(this%stats_work%x,this%dudz%x, this%dwdz%x,n)
559 call this%e13%update(k)
560 call col3(this%stats_work%x, this%dvdx%x, this%dwdx%x, n)
561 call addcol3(this%stats_work%x, this%dvdy%x, this%dwdy%x, n)
562 call addcol3(this%stats_work%x, this%dvdz%x, this%dwdz%x, n)
563 call this%e23%update(k)
564
565 end if
566 end associate
567
568 end subroutine fluid_stats_update
569
570
572 subroutine fluid_stats_free(this)
573 class(fluid_stats_t), intent(inout) :: this
574
575 call this%stats_work%free()
576 call this%stats_u%free()
577 call this%stats_v%free()
578 call this%stats_w%free()
579
580 call this%u_mean%free()
581 call this%v_mean%free()
582 call this%w_mean%free()
583 call this%p_mean%free()
584
585 call this%uu%free()
586 call this%vv%free()
587 call this%ww%free()
588 call this%uv%free()
589 call this%uw%free()
590 call this%vw%free()
591 call this%pp%free()
592
593 call this%dUdx%free()
594 call this%dUdy%free()
595 call this%dUdz%free()
596 call this%dVdx%free()
597 call this%dVdy%free()
598 call this%dVdz%free()
599 call this%dWdx%free()
600 call this%dWdy%free()
601 call this%dWdz%free()
602
603 end subroutine fluid_stats_free
604
606 subroutine fluid_stats_reset(this)
607 class(fluid_stats_t), intent(inout), target:: this
608
609 call this%p_mean%reset()
610 call this%u_mean%reset()
611 call this%v_mean%reset()
612 call this%w_mean%reset()
613
614 call this%uu%reset()
615 call this%vv%reset()
616 call this%ww%reset()
617 call this%uv%reset()
618 call this%uw%reset()
619 call this%vw%reset()
620 call this%pp%reset()
621 if (this%n_stats .eq. 44) then
622 call this%uuu%reset()
623 call this%vvv%reset()
624 call this%www%reset()
625 call this%uuv%reset()
626 call this%uuw%reset()
627 call this%uvv%reset()
628 call this%uvw%reset()
629 call this%vvw%reset()
630 call this%uww%reset()
631 call this%vww%reset()
632 call this%uuuu%reset()
633 call this%vvvv%reset()
634 call this%wwww%reset()
635 call this%ppp%reset()
636 call this%pppp%reset()
637 call this%pu%reset()
638 call this%pv%reset()
639 call this%pw%reset()
640
641 call this%pdudx%reset()
642 call this%pdudy%reset()
643 call this%pdudz%reset()
644 call this%pdvdx%reset()
645 call this%pdvdy%reset()
646 call this%pdvdz%reset()
647 call this%pdwdx%reset()
648 call this%pdwdy%reset()
649 call this%pdwdz%reset()
650
651 call this%e11%reset()
652 call this%e22%reset()
653 call this%e33%reset()
654 call this%e12%reset()
655 call this%e13%reset()
656 call this%e23%reset()
657 end if
658
659 end subroutine fluid_stats_reset
660
661 ! Convert computed weak gradients to strong.
662 subroutine fluid_stats_make_strong_grad(this)
663 class(fluid_stats_t) :: this
664 integer :: n
665
666 if (this%n_stats .eq. 11) return
667
668 n = size(this%coef%B)
669
670 if (neko_bcknd_device .eq. 1) then
671 call device_cfill(this%stats_work%x_d, 1.0_rp, n)
672 call device_invcol2(this%stats_work%x_d, this%coef%B_d, n)
673 call device_col2(this%pdudx%mf%x_d, this%stats_work%x_d, n)
674 call device_col2(this%pdudy%mf%x_d, this%stats_work%x_d, n)
675 call device_col2(this%pdudz%mf%x_d, this%stats_work%x_d, n)
676 call device_col2(this%pdvdx%mf%x_d, this%stats_work%x_d, n)
677 call device_col2(this%pdvdy%mf%x_d, this%stats_work%x_d, n)
678 call device_col2(this%pdvdz%mf%x_d, this%stats_work%x_d, n)
679 call device_col2(this%pdwdx%mf%x_d, this%stats_work%x_d, n)
680 call device_col2(this%pdwdy%mf%x_d, this%stats_work%x_d, n)
681 call device_col2(this%pdwdz%mf%x_d, this%stats_work%x_d, n)
682
683 call device_col2(this%stats_work%x_d, this%stats_work%x_d, n)
684 call device_col2(this%e11%mf%x_d, this%stats_work%x_d, n)
685 call device_col2(this%e22%mf%x_d, this%stats_work%x_d, n)
686 call device_col2(this%e33%mf%x_d, this%stats_work%x_d, n)
687 call device_col2(this%e12%mf%x_d, this%stats_work%x_d, n)
688 call device_col2(this%e13%mf%x_d, this%stats_work%x_d, n)
689 call device_col2(this%e23%mf%x_d, this%stats_work%x_d, n)
690
691
692 else
693 call invers2(this%stats_work%x, this%coef%B, n)
694 call col2(this%pdudx%mf%x, this%stats_work%x, n)
695 call col2(this%pdudy%mf%x, this%stats_work%x, n)
696 call col2(this%pdudz%mf%x, this%stats_work%x, n)
697 call col2(this%pdvdx%mf%x, this%stats_work%x, n)
698 call col2(this%pdvdy%mf%x, this%stats_work%x, n)
699 call col2(this%pdvdz%mf%x, this%stats_work%x, n)
700 call col2(this%pdwdx%mf%x, this%stats_work%x, n)
701 call col2(this%pdwdy%mf%x, this%stats_work%x, n)
702 call col2(this%pdwdz%mf%x, this%stats_work%x, n)
703
704 call col2(this%stats_work%x, this%stats_work%x, n)
705 call col2(this%e11%mf%x, this%stats_work%x, n)
706 call col2(this%e22%mf%x, this%stats_work%x, n)
707 call col2(this%e33%mf%x, this%stats_work%x, n)
708 call col2(this%e12%mf%x, this%stats_work%x, n)
709 call col2(this%e13%mf%x, this%stats_work%x, n)
710 call col2(this%e23%mf%x, this%stats_work%x, n)
711
712 end if
713
714 end subroutine fluid_stats_make_strong_grad
715
718 subroutine fluid_stats_post_process(this, mean, reynolds, pressure_flatness,&
719 pressure_skewness, skewness_tensor, mean_vel_grad, dissipation_tensor)
720 class(fluid_stats_t) :: this
721 type(field_list_t), intent(inout), optional :: mean
722 type(field_list_t), intent(inout), optional :: reynolds
723 type(field_list_t), intent(inout), optional :: pressure_skewness
724 type(field_list_t), intent(inout), optional :: pressure_flatness
725 type(field_list_t), intent(inout), optional :: skewness_tensor
726 type(field_list_t), intent(inout), optional :: mean_vel_grad
727 type(field_list_t), intent(inout), optional :: dissipation_tensor
728 integer :: n
729
730 if (present(mean)) then
731 n = mean%item_size(1)
732 call copy(mean%items(1)%ptr%x, this%u_mean%mf%x, n)
733 call copy(mean%items(2)%ptr%x, this%v_mean%mf%x, n)
734 call copy(mean%items(3)%ptr%x, this%w_mean%mf%x, n)
735 call copy(mean%items(4)%ptr%x, this%p_mean%mf%x, n)
736 end if
737
738 if (present(reynolds)) then
739 n = reynolds%item_size(1)
740 call copy(reynolds%items(1)%ptr%x, this%pp%mf%x, n)
741 call subcol3(reynolds%items(1)%ptr%x, this%p_mean%mf%x, &
742 this%p_mean%mf%x, n)
743
744 call copy(reynolds%items(2)%ptr%x, this%uu%mf%x, n)
745 call subcol3(reynolds%items(2)%ptr%x, this%u_mean%mf%x, &
746 this%u_mean%mf%x, n)
747
748 call copy(reynolds%items(3)%ptr%x, this%vv%mf%x, n)
749 call subcol3(reynolds%items(3)%ptr%x, this%v_mean%mf%x, &
750 this%v_mean%mf%x,n)
751
752 call copy(reynolds%items(4)%ptr%x, this%ww%mf%x, n)
753 call subcol3(reynolds%items(4)%ptr%x, this%w_mean%mf%x, &
754 this%w_mean%mf%x,n)
755
756 call copy(reynolds%items(5)%ptr%x, this%uv%mf%x, n)
757 call subcol3(reynolds%items(5)%ptr%x, this%u_mean%mf%x, &
758 this%v_mean%mf%x, n)
759
760 call copy(reynolds%items(6)%ptr%x, this%uw%mf%x, n)
761 call subcol3(reynolds%items(6)%ptr%x, this%u_mean%mf%x, &
762 this%w_mean%mf%x, n)
763
764 call copy(reynolds%items(7)%ptr%x, this%vw%mf%x, n)
765 call subcol3(reynolds%items(7)%ptr%x, this%v_mean%mf%x, &
766 this%w_mean%mf%x, n)
767 end if
768 if (present(pressure_skewness)) then
769
770 call neko_warning('Presssure skewness stat not implemented'// &
771 ' in fluid_stats, process stats in python instead')
772
773 end if
774
775 if (present(pressure_flatness)) then
776 call neko_warning('Presssure flatness stat not implemented'// &
777 ' in fluid_stats, process stats in python instead')
778
779 end if
780
781 if (present(skewness_tensor)) then
782 call neko_warning('Skewness tensor stat not implemented'// &
783 ' in fluid_stats, process stats in python instead')
784 end if
785
786 if (present(mean_vel_grad)) then
787 !Compute gradient of mean flow
788 n = mean_vel_grad%item_size(1)
789 if (neko_bcknd_device .eq. 1) then
790 call device_memcpy(this%u_mean%mf%x, this%u_mean%mf%x_d, n, &
791 host_to_device, sync = .false.)
792 call device_memcpy(this%v_mean%mf%x, this%v_mean%mf%x_d, n, &
793 host_to_device, sync = .false.)
794 call device_memcpy(this%w_mean%mf%x, this%w_mean%mf%x_d, n, &
795 host_to_device, sync = .false.)
796 call opgrad(this%dudx%x, this%dudy%x, this%dudz%x, &
797 this%u_mean%mf%x, this%coef)
798 call opgrad(this%dvdx%x, this%dvdy%x, this%dvdz%x, &
799 this%v_mean%mf%x, this%coef)
800 call opgrad(this%dwdx%x, this%dwdy%x, this%dwdz%x, &
801 this%w_mean%mf%x, this%coef)
802 call device_memcpy(this%dudx%x, this%dudx%x_d, n, &
803 device_to_host, sync = .false.)
804 call device_memcpy(this%dvdx%x, this%dvdx%x_d, n, &
805 device_to_host, sync = .false.)
806 call device_memcpy(this%dwdx%x, this%dwdx%x_d, n, &
807 device_to_host, sync = .false.)
808 call device_memcpy(this%dudy%x, this%dudy%x_d, n, &
809 device_to_host, sync = .false.)
810 call device_memcpy(this%dvdy%x, this%dvdy%x_d, n, &
811 device_to_host, sync = .false.)
812 call device_memcpy(this%dwdy%x, this%dwdy%x_d, n, &
813 device_to_host, sync = .false.)
814 call device_memcpy(this%dudz%x, this%dudz%x_d, n, &
815 device_to_host, sync = .false.)
816 call device_memcpy(this%dvdz%x, this%dvdz%x_d, n, &
817 device_to_host, sync = .false.)
818 call device_memcpy(this%dwdz%x, this%dwdz%x_d, n, &
819 device_to_host, sync = .true.)
820 else
821 call opgrad(this%dudx%x, this%dudy%x, this%dudz%x, &
822 this%u_mean%mf%x, this%coef)
823 call opgrad(this%dvdx%x, this%dvdy%x, this%dvdz%x, &
824 this%v_mean%mf%x, this%coef)
825 call opgrad(this%dwdx%x, this%dwdy%x, this%dwdz%x, &
826 this%w_mean%mf%x, this%coef)
827 end if
828 call invers2(this%stats_work%x, this%coef%B,n)
829 call col3(mean_vel_grad%items(1)%ptr%x, this%dudx%x, &
830 this%stats_work%x, n)
831 call col3(mean_vel_grad%items(2)%ptr%x, this%dudy%x, &
832 this%stats_work%x, n)
833 call col3(mean_vel_grad%items(3)%ptr%x, this%dudz%x, &
834 this%stats_work%x, n)
835 call col3(mean_vel_grad%items(4)%ptr%x, this%dvdx%x, &
836 this%stats_work%x, n)
837 call col3(mean_vel_grad%items(5)%ptr%x, this%dvdy%x, &
838 this%stats_work%x, n)
839 call col3(mean_vel_grad%items(6)%ptr%x, this%dvdz%x, &
840 this%stats_work%x, n)
841 call col3(mean_vel_grad%items(7)%ptr%x, this%dwdx%x, &
842 this%stats_work%x, n)
843 call col3(mean_vel_grad%items(8)%ptr%x, this%dwdy%x, &
844 this%stats_work%x, n)
845 call col3(mean_vel_grad%items(9)%ptr%x, this%dwdz%x, &
846 this%stats_work%x, n)
847
848 end if
849
850 if (present(dissipation_tensor)) then
851 call neko_warning('Dissipation tensor stat not implemented'// &
852 ' in fluid_stats, process stats in python instead')
853 end if
854
855 end subroutine fluid_stats_post_process
856
857end module fluid_stats
coefs
Coefficients.
Definition coef.f90:34
device_math
Definition device_math.F90:33
device_math::device_addcol3
subroutine, public device_addcol3(a_d, b_d, c_d, n)
Returns .
Definition device_math.F90:460
device_math::device_col2
subroutine, public device_col2(a_d, b_d, n)
Vector multiplication .
Definition device_math.F90:385
device_math::device_col3
subroutine, public device_col3(a_d, b_d, c_d, n)
Vector multiplication with 3 vectors .
Definition device_math.F90:400
device_math::device_cfill
subroutine, public device_cfill(a_d, c, n)
Set all elements to a constant c .
Definition device_math.F90:231
device_math::device_invcol2
subroutine, public device_invcol2(a_d, b_d, n)
Vector division .
Definition device_math.F90:370
device
Device abstraction, common interface for various accelerators.
Definition device.F90:34
field_list
Definition field_list.f90:1
field
Defines a field.
Definition field.f90:34
fluid_stats
Computes various statistics for the fluid fields. We use the Reynolds decomposition for a field u = ...
Definition fluid_stats.f90:36
fluid_stats::fluid_stats_reset
subroutine fluid_stats_reset(this)
Resets all the computed means values and sampling times to zero.
Definition fluid_stats.f90:607
fluid_stats::fluid_stats_make_strong_grad
subroutine fluid_stats_make_strong_grad(this)
Definition fluid_stats.f90:663
fluid_stats::fluid_stats_update
subroutine fluid_stats_update(this, k)
Updates all fields with a new sample.
Definition fluid_stats.f90:314
fluid_stats::fluid_stats_free
subroutine fluid_stats_free(this)
Destructor.
Definition fluid_stats.f90:573
fluid_stats::fluid_stats_post_process
subroutine fluid_stats_post_process(this, mean, reynolds, pressure_flatness, pressure_skewness, skewness_tensor, mean_vel_grad, dissipation_tensor)
Compute certain physical statistical quantities based on existing mean fields.
Definition fluid_stats.f90:720
fluid_stats::fluid_stats_init
subroutine fluid_stats_init(this, coef, u, v, w, p, set)
Constructor. Initialize the fields associated with fluid_stats.
Definition fluid_stats.f90:168
math
Definition math.f90:60
math::invers2
subroutine, public invers2(a, b, n)
Compute inverted vector .
Definition math.f90:500
math::subcol3
subroutine, public subcol3(a, b, c, n)
Returns .
Definition math.f90:756
math::addcol3
subroutine, public addcol3(a, b, c, n)
Returns .
Definition math.f90:801
math::col2
subroutine, public col2(a, b, n)
Vector multiplication .
Definition math.f90:729
math::copy
subroutine, public copy(a, b, n)
Copy a vector .
Definition math.f90:239
math::col3
subroutine, public col3(a, b, c, n)
Vector multiplication with 3 vectors .
Definition math.f90:742
mean_field
Implements mean_field_t.
Definition mean_field.f90:35
neko_config
Build configurations.
Definition neko_config.f90:34
neko_config::neko_bcknd_device
integer, parameter neko_bcknd_device
Definition neko_config.f90:44
num_types
Definition num_types.f90:1
num_types::rp
integer, parameter, public rp
Global precision used in computations.
Definition num_types.f90:12
operators
Operators.
Definition operators.f90:34
operators::opgrad
subroutine, public opgrad(ux, uy, uz, u, coef, es, ee)
Compute the weak gradient of a scalar field, i.e. the gradient multiplied by the mass matrix.
Definition operators.f90:171
stats_quant
Defines a statistical quantity.
Definition stats_quant.f90:34
utils
Utilities.
Definition utils.f90:35
utils::neko_warning
subroutine, public neko_warning(warning_msg)
Reports a warning to standard output.
Definition utils.f90:245
coefs::coef_t
Coefficients defined on a given (mesh, ) tuple. Arrays use indices (i,j,k,e): element e,...
Definition coef.f90:55
field::field_t
Definition field.f90:46
field_list::field_list_t
field_list_t, To be able to group fields together
Definition field_list.f90:13
fluid_stats::fluid_stats_t
Definition fluid_stats.f90:53
mean_field::mean_field_t
Computes the temporal mean of a field.
Definition mean_field.f90:45
stats_quant::stats_quant_t
Abstract type defining a statistical quantity.
Definition stats_quant.f90:40