source: palm/trunk/SOURCE/synthetic_turbulence_generator_mod.f90 @ 2984

Last change on this file since 2984 was 2967, checked in by raasch, 7 years ago

bugfix: missing parallel cpp-directives added

  • Property svn:keywords set to Id
File size: 61.4 KB
RevLine 
[2259]1!> @synthetic_turbulence_generator_mod.f90
2!------------------------------------------------------------------------------!
[2696]3! This file is part of the PALM model system.
[2259]4!
[2696]5! PALM is free software: you can redistribute it and/or modify it under the
[2259]6! terms of the GNU General Public License as published by the Free Software
7! Foundation, either version 3 of the License, or (at your option) any later
8! version.
9!
[2696]10! PALM is distributed in the hope that it will be useful, but WITHOUT ANY
[2259]11! WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
12! A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
13!
14! You should have received a copy of the GNU General Public License along with
15! PALM. If not, see <http://www.gnu.org/licenses/>.
16!
17! Copyright 2017 Leibniz Universitaet Hannover
18!------------------------------------------------------------------------------!
19!
20! Current revisions:
21! -----------------
22!
23!
24! Former revisions:
25! -----------------
26! $Id: synthetic_turbulence_generator_mod.f90 2967 2018-04-13 11:22:08Z hellstea $
[2967]27! bugfix: missing parallel cpp-directives added
28!
29! 2946 2018-04-04 17:01:23Z suehring
[2946]30! Remove unused module load
31!
32! 2945 2018-04-04 16:27:14Z suehring
[2945]33! - Bugfix in parallelization of synthetic turbulence generator in case the
34!   z-dimension is not an integral divisor of the number of processors along 
35!   the x- and y-dimension
36! - Revision in control mimic in case of RAN-LES nesting
37!
38! 2938 2018-03-27 15:52:42Z suehring
[2938]39! Apply turbulence generator at all non-cyclic lateral boundaries in case of
40! realistic Inifor large-scale forcing or RANS-LES nesting
41!
42! 2936 2018-03-27 14:49:27Z suehring
[2894]43! variable named found has been introduced for checking if restart data was found,
44! reading of restart strings has been moved completely to read_restart_data_mod,
45! redundant skipping function has been removed, stg_read/write_restart_data
46! have been renamed to stg_r/wrd_global, stg_rrd_global is called in
47! read_restart_data_mod now, flag syn_turb_gen_prerun and marker *** end stg
48! *** have been removed (Giersch), strings and their respective lengths are
49! written out and read now in case of restart runs to get rid of prescribed
50! character lengths (Giersch), CASE DEFAULT was added if restart data is read
51!
52! 2841 2018-02-27 15:02:57Z suehring
[2841]53! Bugfix: wrong placement of include 'mpif.h' corrected
54!
55! 2836 2018-02-26 13:40:05Z Giersch
[2836]56! The variables synthetic_turbulence_generator and
57! use_synthetic_turbulence_generator have been abbreviated + syn_turb_gen_prerun
58! flag is used to define if module related parameters were outputted as restart
59! data
[2776]60!
61! 2716 2017-12-29 16:35:59Z kanani
[2716]62! Corrected "Former revisions" section
63!
64! 2696 2017-12-14 17:12:51Z kanani
65! Change in file header (GPL part)
66!
67! 2669 2017-12-06 16:03:27Z raasch
[2669]68! unit number for file containing turbulence generator data changed to 90
69! bugfix: preprocessor directives added for MPI specific code
70!
71! 2576 2017-10-24 13:49:46Z Giersch
[2894]72! Definition of a new function called stg_skip_global to skip module
[2576]73! parameters during reading restart data
74!
75! 2563 2017-10-19 15:36:10Z Giersch
[2563]76! stg_read_restart_data is called in stg_parin in the case of a restart run
77!
78! 2259 2017-06-08 09:09:11Z gronemeier
[2259]79! Initial revision
80!
81!
82!
83! Authors:
84! --------
85! @author Tobias Gronemeier, Atsushi Inagaki, Micha Gryschka, Christoph Knigge
86!
87!
88! Description:
89! ------------
90!> The module generates turbulence at the inflow boundary based on a method by
91!> Xie and Castro (2008) utilizing a Lund rotation (Lund, 1998) and a mass-flux
92!> correction by Kim et al. (2013).
93!> The turbulence is correlated based on length scales in y- and z-direction and
94!> a time scale for each velocity component. The profiles of length and time
95!> scales, mean u, v, w, e and pt, and all components of the Reynolds stress
96!> tensor are read from file STG_PROFILES.
97!>
98!> @todo test restart
99!>       enable cyclic_fill
100!>       implement turbulence generation for e and pt
[2945]101!> @todo Input of height-constant length scales via namelist
[2259]102!> @note <Enter notes on the module>
103!> @bug  Height information from input file is not used. Profiles from input
104!>       must match with current PALM grid.
105!>       Transformation of length scales to number of gridpoints does not
106!>       consider grid stretching.
107!>       In case of restart, velocity seeds differ from precursor run if a11,
108!>       a22, or a33 are zero.
109!------------------------------------------------------------------------------!
110 MODULE synthetic_turbulence_generator_mod
111
112
113    USE arrays_3d,                                                             &
114        ONLY:  mean_inflow_profiles, u, v, w
115
116    USE constants,                                                             &
117        ONLY:  pi
118
119    USE control_parameters,                                                    &
[2938]120        ONLY:  initializing_actions, message_string, syn_turb_gen
[2259]121
122    USE cpulog,                                                                &
123        ONLY:  cpu_log, log_point, log_point_s
124
125    USE indices,                                                               &
[2938]126        ONLY:  nbgp, nzb, nzt, nxl, nxlg, nxr, nxrg, nys, nyn, nyng, nysg
[2259]127
128    USE kinds
129
[2967]130#if defined( __parallel )  &&  !defined( __mpifh )
[2259]131    USE MPI
[2841]132#endif
[2259]133
134    USE pegrid,                                                                &
[2938]135        ONLY:  comm1dx, comm1dy, comm2d, ierr, myidx, myidy, pdims
[2259]136
137    USE transpose_indices,                                                     &
138        ONLY: nzb_x, nzt_x
139
140
141    IMPLICIT NONE
142
[2967]143#if defined( __parallel )  &&  defined( __mpifh )
[2841]144    INCLUDE "mpif.h"
145#endif
146
[2776]147    LOGICAL :: velocity_seed_initialized = .FALSE.  !< true after first call of stg_main
148    LOGICAL :: use_syn_turb_gen = .FALSE.           !< switch to use synthetic turbulence generator
[2259]149
[2938]150    INTEGER(iwp) ::  id_stg_left        !< left lateral boundary core id in case of turbulence generator 
151    INTEGER(iwp) ::  id_stg_north       !< north lateral boundary core id in case of turbulence generator 
152    INTEGER(iwp) ::  id_stg_right       !< right lateral boundary core id in case of turbulence generator 
153    INTEGER(iwp) ::  id_stg_south       !< south lateral boundary core id in case of turbulence generator 
154    INTEGER(iwp) ::  stg_type_xz        !< MPI type for full z range
155    INTEGER(iwp) ::  stg_type_xz_small  !< MPI type for small z range
156    INTEGER(iwp) ::  stg_type_yz        !< MPI type for full z range
157    INTEGER(iwp) ::  stg_type_yz_small  !< MPI type for small z range
158    INTEGER(iwp) ::  merg               !< maximum length scale (in gp)
159    INTEGER(iwp) ::  mergp              !< merg + nbgp
160    INTEGER(iwp) ::  nzb_x_stg          !<
161    INTEGER(iwp) ::  nzt_x_stg          !<
162    INTEGER(iwp) ::  nzb_y_stg          !<
163    INTEGER(iwp) ::  nzt_y_stg          !<
[2259]164
165    REAL(wp) :: mc_factor    !< mass flux correction factor
166
[2938]167    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: displs_xz      !< displacement for MPI_GATHERV
168    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: recv_count_xz  !< receive count for MPI_GATHERV
169    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: displs_yz      !< displacement for MPI_GATHERV
170    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: recv_count_yz  !< receive count for MPI_GATHERV
171    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: nux            !< length scale of u in x direction (in gp)
172    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: nuy            !< length scale of u in y direction (in gp)
173    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: nuz            !< length scale of u in z direction (in gp)
174    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: nvx            !< length scale of v in x direction (in gp)
175    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: nvy            !< length scale of v in y direction (in gp)
176    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: nvz            !< length scale of v in z direction (in gp)
177    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: nwx            !< length scale of w in x direction (in gp)
178    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: nwy            !< length scale of w in y direction (in gp)
179    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: nwz            !< length scale of w in z direction (in gp)
[2259]180
181    INTEGER(iwp), DIMENSION(:), ALLOCATABLE :: seed        !< seed of random number for rn-generator
182
183    REAL(wp), DIMENSION(:), ALLOCATABLE :: a11             !< coefficient for Lund rotation
184    REAL(wp), DIMENSION(:), ALLOCATABLE :: a21             !< coefficient for Lund rotation
185    REAL(wp), DIMENSION(:), ALLOCATABLE :: a22             !< coefficient for Lund rotation
186    REAL(wp), DIMENSION(:), ALLOCATABLE :: a31             !< coefficient for Lund rotation
187    REAL(wp), DIMENSION(:), ALLOCATABLE :: a32             !< coefficient for Lund rotation
188    REAL(wp), DIMENSION(:), ALLOCATABLE :: a33             !< coefficient for Lund rotation
189    REAL(wp), DIMENSION(:), ALLOCATABLE :: tu              !< Lagrangian time scale of u
190    REAL(wp), DIMENSION(:), ALLOCATABLE :: tv              !< Lagrangian time scale of v
191    REAL(wp), DIMENSION(:), ALLOCATABLE :: tw              !< Lagrangian time scale of w
192
[2938]193    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: bux           !< filter function for u in x direction
[2259]194    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: buy           !< filter function for u in y direction
195    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: buz           !< filter function for u in z direction
[2938]196    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: bvx           !< filter function for v in x direction
[2259]197    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: bvy           !< filter function for v in y direction
198    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: bvz           !< filter function for v in z direction
[2938]199    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: bwx           !< filter function for w in y direction
[2259]200    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: bwy           !< filter function for w in y direction
201    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: bwz           !< filter function for w in z direction
[2938]202    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fu_xz         !< velocity seed for u at xz plane
203    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fuo_xz        !< velocity seed for u at xz plane with new random number
204    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fu_yz         !< velocity seed for u at yz plane
205    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fuo_yz        !< velocity seed for u at yz plane with new random number
206    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fv_xz         !< velocity seed for v at xz plane
207    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fvo_xz        !< velocity seed for v at xz plane with new random number
208    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fv_yz         !< velocity seed for v at yz plane
209    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fvo_yz        !< velocity seed for v at yz plane with new random number
210    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fw_xz         !< velocity seed for w at xz plane
211    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fwo_xz        !< velocity seed for w at xz plane with new random number
212    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fw_yz         !< velocity seed for w at yz plane
213    REAL(wp), DIMENSION(:,:), ALLOCATABLE :: fwo_yz        !< velocity seed for w at yz plane with new random number
[2259]214
215
216!
217!-- PALM interfaces:
218!-- Input parameter checks to be done in check_parameters
219    INTERFACE stg_check_parameters
220       MODULE PROCEDURE stg_check_parameters
221    END INTERFACE stg_check_parameters
222
223!
224!-- Calculate filter functions
225    INTERFACE stg_filter_func
226       MODULE PROCEDURE stg_filter_func
227    END INTERFACE stg_filter_func
228
229!
[2938]230!-- Generate velocity seeds at south and north domain boundary
231    INTERFACE stg_generate_seed_xz
232       MODULE PROCEDURE stg_generate_seed_xz
233    END INTERFACE stg_generate_seed_xz
234!
235!-- Generate velocity seeds at left and/or right domain boundary
[2259]236    INTERFACE stg_generate_seed_yz
237       MODULE PROCEDURE stg_generate_seed_yz
238    END INTERFACE stg_generate_seed_yz
239
240!
241!-- Output of information to the header file
242    INTERFACE stg_header
243       MODULE PROCEDURE stg_header
244    END INTERFACE stg_header
245
246!
247!-- Initialization actions
248    INTERFACE stg_init
249       MODULE PROCEDURE stg_init
250    END INTERFACE stg_init
251
252!
253!-- Main procedure of synth. turb. gen.
254    INTERFACE stg_main
255       MODULE PROCEDURE stg_main
256    END INTERFACE stg_main
257
258!
259!-- Reading of NAMELIST parameters
260    INTERFACE stg_parin
261       MODULE PROCEDURE stg_parin
262    END INTERFACE stg_parin
263
264!
265!-- Reading of parameters for restart runs
[2894]266    INTERFACE stg_rrd_global
267       MODULE PROCEDURE stg_rrd_global
268    END INTERFACE stg_rrd_global
[2259]269
270!
271!-- Writing of binary output for restart runs
[2894]272    INTERFACE stg_wrd_global
273       MODULE PROCEDURE stg_wrd_global
274    END INTERFACE stg_wrd_global
[2259]275
276    SAVE
277
278    PRIVATE
279
280!
281!-- Public interfaces
282    PUBLIC  stg_check_parameters, stg_header, stg_init, stg_main, stg_parin,   &
[2894]283            stg_wrd_global, stg_rrd_global
[2259]284
285!
286!-- Public variables
[2938]287    PUBLIC  id_stg_left, id_stg_north, id_stg_right, id_stg_south,             &
288            use_syn_turb_gen
[2259]289
290
291 CONTAINS
292
293
294!------------------------------------------------------------------------------!
295! Description:
296! ------------
297!> Check parameters routine for synthetic turbulence generator
298!------------------------------------------------------------------------------!
299 SUBROUTINE stg_check_parameters
300
301
302    USE control_parameters,                                                    &
[2938]303        ONLY:  bc_lr, bc_ns, forcing, nest_domain, rans_mode, turbulent_inflow
[2259]304
[2938]305    USE pmc_interface,                                                         &
306        ONLY : rans_mode_parent
[2259]307
[2938]308
[2259]309    IMPLICIT NONE
310
[2938]311    IF ( .NOT. use_syn_turb_gen  .AND.  .NOT. rans_mode  .AND.  forcing )  THEN
312       message_string = 'Synthetic turbulence generator has to be applied ' // &
313                        'when forcing is used and model operates in LES mode.'
314       CALL message( 'stg_check_parameters', 'PA0000', 1, 2, 0, 6, 0 )
315    ENDIF
316
317    IF ( .NOT. use_syn_turb_gen  .AND.  nest_domain                            &
318         .AND. rans_mode_parent  .AND.  .NOT. rans_mode )  THEN
319       message_string = 'Synthetic turbulence generator has to be applied ' // &
320                        'when nesting is applied and parent operates in '  //  &
321                        'RANS-mode but current child in LES mode.'
322       CALL message( 'stg_check_parameters', 'PA0000', 1, 2, 0, 6, 0 )
323    ENDIF
324
[2776]325    IF ( use_syn_turb_gen )  THEN
[2259]326
[2938]327       IF ( .NOT. forcing  .AND.  .NOT. nest_domain )  THEN
328
329          IF ( INDEX( initializing_actions, 'set_constant_profiles' ) == 0     &
330        .AND.  INDEX( initializing_actions, 'read_restart_data' ) == 0 )  THEN
331             message_string = 'Using synthetic turbulence generator ' //       &
332                              'requires &initializing_actions = '         //   &
333                              '"set_constant_profiles" or "read_restart_data"'
334             CALL message( 'stg_check_parameters', 'PA0015', 1, 2, 0, 6, 0 )
335          ENDIF
336
337          IF ( bc_lr /= 'dirichlet/radiation' )  THEN
338             message_string = 'Using synthetic turbulence generator ' //       &
339                              'requires &bc_lr = "dirichlet/radiation"'
340             CALL message( 'stg_check_parameters', 'PA0035', 1, 2, 0, 6, 0 )
341          ENDIF
342          IF ( bc_ns /= 'cyclic' )  THEN
343             message_string = 'Using synthetic turbulence generator ' //       &
344                              'requires &bc_ns = "cyclic"'
345             CALL message( 'stg_check_parameters', 'PA0037', 1, 2, 0, 6, 0 )
346          ENDIF
347
[2259]348       ENDIF
349
350       IF ( turbulent_inflow )  THEN
351          message_string = 'Using synthetic turbulence generator ' //          &
[2938]352                           'in combination &with turbulent_inflow = .T. '//    &
353                              'is not allowed'
[2259]354          CALL message( 'stg_check_parameters', 'PA0039', 1, 2, 0, 6, 0 )
355       ENDIF
356
357    ENDIF
358
359 END SUBROUTINE stg_check_parameters
360
361
362!------------------------------------------------------------------------------!
363! Description:
364! ------------
365!> Header output for synthetic turbulence generator
366!------------------------------------------------------------------------------!
367 SUBROUTINE stg_header ( io )
368
369
370    IMPLICIT NONE
371
372    INTEGER(iwp), INTENT(IN) ::  io   !< Unit of the output file
373
374!
375!-- Write synthetic turbulence generator Header
376    WRITE( io, 1 )
[2776]377    IF ( use_syn_turb_gen )  THEN
[2259]378       WRITE( io, 2 )
379    ELSE
380       WRITE( io, 3 )
381    ENDIF
382
3831   FORMAT (//' Synthetic turbulence generator information:'/                  &
384              ' ------------------------------------------'/)
3852   FORMAT ('    synthetic turbulence generator switched on')
3863   FORMAT ('    synthetic turbulence generator switched off')
387
388 END SUBROUTINE stg_header
389
390
391!------------------------------------------------------------------------------!
392! Description:
393! ------------
394!> Initialization of the synthetic turbulence generator
395!------------------------------------------------------------------------------!
396 SUBROUTINE stg_init
397
398
399    USE arrays_3d,                                                             &
[2938]400        ONLY:  ddzw, u_init, v_init, zu
[2259]401
402    USE control_parameters,                                                    &
[2945]403        ONLY:  coupling_char, dz, e_init, forcing, nest_domain, rans_mode
[2259]404
405    USE grid_variables,                                                        &
[2938]406        ONLY:  ddx, ddy
[2259]407
[2938]408    USE indices,                                                               &
409        ONLY:  nz
[2259]410
[2945]411    USE pmc_interface,                                                         &
412        ONLY : rans_mode_parent
[2938]413
[2945]414
[2259]415    IMPLICIT NONE
416
[2938]417    LOGICAL ::  file_stg_exist = .FALSE. !< flag indication whether parameter file for Reynolds stress and length scales exist
418
[2669]419#if defined( __parallel )
[2259]420    INTEGER(KIND=MPI_ADDRESS_KIND) :: extent !< extent of new MPI type
421    INTEGER(KIND=MPI_ADDRESS_KIND) :: tob=0  !< dummy variable
[2669]422#endif
[2259]423
[2938]424    INTEGER(iwp) :: i                        !> grid index in x-direction
[2259]425    INTEGER(iwp) :: j                        !> loop index
426    INTEGER(iwp) :: k                        !< index
427    INTEGER(iwp) :: newtype                  !< dummy MPI type
428    INTEGER(iwp) :: realsize                 !< size of REAL variables
429    INTEGER(iwp) :: nseed                    !< dimension of random number seed
430    INTEGER(iwp) :: startseed = 1234567890   !< start seed for random number generator
431
432!
433!-- Dummy variables used for reading profiles
434    REAL(wp) :: d1      !< u profile
435    REAL(wp) :: d2      !< v profile
436    REAL(wp) :: d3      !< w profile
437    REAL(wp) :: d5      !< e profile
438    REAL(wp) :: d11     !< vertical interpolation for a11
439    REAL(wp) :: d21     !< vertical interpolation for a21
440    REAL(wp) :: d22     !< vertical interpolation for a22
[2938]441    REAL(wp) :: dum_exp !< dummy variable used for exponential vertical decrease of turbulent length scales
[2259]442    REAL(wp) :: luy     !< length scale for u in y direction
443    REAL(wp) :: luz     !< length scale for u in z direction
444    REAL(wp) :: lvy     !< length scale for v in y direction
445    REAL(wp) :: lvz     !< length scale for v in z direction
446    REAL(wp) :: lwy     !< length scale for w in y direction
447    REAL(wp) :: lwz     !< length scale for w in z direction
[2945]448    REAL(wp) :: nnz     !< increment used to determine processor decomposition of z-axis along x and y direction 
[2259]449    REAL(wp) :: zz      !< height
450
[2938]451    REAL(wp) :: length_scale_surface, r_ii_0, time_scale, length_scale_z
452
[2259]453    REAL(wp),DIMENSION(nzb:nzt+1) :: r11  !< Reynolds parameter
454    REAL(wp),DIMENSION(nzb:nzt+1) :: r21  !< Reynolds parameter
455    REAL(wp),DIMENSION(nzb:nzt+1) :: r22  !< Reynolds parameter
456    REAL(wp),DIMENSION(nzb:nzt+1) :: r31  !< Reynolds parameter
457    REAL(wp),DIMENSION(nzb:nzt+1) :: r32  !< Reynolds parameter
458    REAL(wp),DIMENSION(nzb:nzt+1) :: r33  !< Reynolds parameter
459
460
461#if defined( __parallel )
462    CALL MPI_BARRIER( comm2d, ierr )
463#endif
464
465    CALL  cpu_log( log_point(57), 'synthetic_turbulence_gen', 'start' )
466
467    IF ( .NOT. ALLOCATED( mean_inflow_profiles ) )                             &
468       ALLOCATE( mean_inflow_profiles(nzb:nzt+1,5) )
469
470    ALLOCATE ( a11(nzb:nzt+1), a21(nzb:nzt+1), a22(nzb:nzt+1),                 &
471               a31(nzb:nzt+1), a32(nzb:nzt+1), a33(nzb:nzt+1),                 &
[2938]472               nux(nzb:nzt+1), nuy(nzb:nzt+1), nuz(nzb:nzt+1),                 &
473               nvx(nzb:nzt+1), nvy(nzb:nzt+1), nvz(nzb:nzt+1),                 &
474               nwx(nzb:nzt+1), nwy(nzb:nzt+1), nwz(nzb:nzt+1),                 &
[2259]475               tu(nzb:nzt+1),  tv(nzb:nzt+1),  tw(nzb:nzt+1)   )
476
[2669]477#if defined( __parallel )
[2259]478!
[2938]479!-- Determine processor decomposition of z-axis along x- and y-direction
480    nnz = nz / pdims(1)
[2945]481    nzb_x_stg = 1 + myidx * INT( nnz )
482    nzt_x_stg = ( myidx + 1 ) * INT( nnz )
[2938]483
484    IF ( MOD( nz , pdims(1) ) /= 0  .AND.  myidx == id_stg_right )             & 
[2945]485       nzt_x_stg = nzt_x_stg + myidx * ( nnz - INT( nnz ) )
486!        nzt_x_stg = myidx * nnz + MOD( nz , pdims(1) )
[2938]487
[2945]488    IF ( forcing  .OR.  ( nest_domain .AND.  rans_mode_parent  .AND.           &
489                   .NOT.  rans_mode ) )  THEN
[2938]490       nnz = nz / pdims(2)
[2945]491       nzb_y_stg = 1 + myidy * INT( nnz )
492       nzt_y_stg = ( myidy + 1 ) * INT( nnz )
[2938]493
[2945]494       IF ( MOD( nz , pdims(2) ) /= 0  .AND.  myidy == id_stg_north )          &
495          nzt_y_stg = nzt_y_stg + myidy * ( nnz - INT( nnz ) ) 
496!           nzt_y_stg = myidy * nnz + MOD( nz , pdims(2) )
[2938]497    ENDIF
498
499!
[2259]500!-- Define MPI type used in stg_generate_seed_yz to gather vertical splitted
501!-- velocity seeds
502    CALL MPI_TYPE_SIZE( MPI_REAL, realsize, ierr )
503    extent = 1 * realsize
[2938]504!
505!-- Set-up MPI datatyp to involve all cores for turbulence generation at yz
506!-- layer
507!-- stg_type_yz: yz-slice with vertical bounds nzb:nzt+1
[2259]508    CALL MPI_TYPE_CREATE_SUBARRAY( 2, [nzt-nzb+2,nyng-nysg+1],                 &
509            [1,nyng-nysg+1], [0,0], MPI_ORDER_FORTRAN, MPI_REAL, newtype, ierr )
510    CALL MPI_TYPE_CREATE_RESIZED( newtype, tob, extent, stg_type_yz, ierr )
511    CALL MPI_TYPE_COMMIT( stg_type_yz, ierr )
512    CALL MPI_TYPE_FREE( newtype, ierr )
513
[2938]514    ! stg_type_yz_small: yz-slice with vertical bounds nzb_x_stg:nzt_x_stg+1
[2945]515    CALL MPI_TYPE_CREATE_SUBARRAY( 2, [nzt_x_stg-nzb_x_stg+2,nyng-nysg+1],     &
[2259]516            [1,nyng-nysg+1], [0,0], MPI_ORDER_FORTRAN, MPI_REAL, newtype, ierr )
517    CALL MPI_TYPE_CREATE_RESIZED( newtype, tob, extent, stg_type_yz_small, ierr )
518    CALL MPI_TYPE_COMMIT( stg_type_yz_small, ierr )
519    CALL MPI_TYPE_FREE( newtype, ierr )
520
521    ! receive count and displacement for MPI_GATHERV in stg_generate_seed_yz
[2938]522    ALLOCATE( recv_count_yz(pdims(1)), displs_yz(pdims(1)) )
[2259]523
[2938]524    recv_count_yz           = nzt_x_stg-nzb_x_stg + 1
525    recv_count_yz(pdims(1)) = recv_count_yz(pdims(1)) + 1
[2259]526
527    DO  j = 1, pdims(1)
[2938]528       displs_yz(j) = 0 + (nzt_x_stg-nzb_x_stg+1) * (j-1)
[2259]529    ENDDO
[2938]530!
531!-- Set-up MPI datatyp to involve all cores for turbulence generation at xz
532!-- layer
533!-- stg_type_xz: xz-slice with vertical bounds nzb:nzt+1
[2945]534    IF ( forcing  .OR.  ( nest_domain .AND.  rans_mode_parent  .AND.           &
535                   .NOT.  rans_mode ) )  THEN
536       CALL MPI_TYPE_CREATE_SUBARRAY( 2, [nzt-nzb+2,nxrg-nxlg+1],              &
[2938]537               [1,nxrg-nxlg+1], [0,0], MPI_ORDER_FORTRAN, MPI_REAL, newtype, ierr )
538       CALL MPI_TYPE_CREATE_RESIZED( newtype, tob, extent, stg_type_xz, ierr )
539       CALL MPI_TYPE_COMMIT( stg_type_xz, ierr )
540       CALL MPI_TYPE_FREE( newtype, ierr )
541
542       ! stg_type_yz_small: xz-slice with vertical bounds nzb_x_stg:nzt_x_stg+1
[2945]543       CALL MPI_TYPE_CREATE_SUBARRAY( 2, [nzt_y_stg-nzb_y_stg+2,nxrg-nxlg+1],  &
[2938]544               [1,nxrg-nxlg+1], [0,0], MPI_ORDER_FORTRAN, MPI_REAL, newtype, ierr )
545       CALL MPI_TYPE_CREATE_RESIZED( newtype, tob, extent, stg_type_xz_small, ierr )
546       CALL MPI_TYPE_COMMIT( stg_type_xz_small, ierr )
547       CALL MPI_TYPE_FREE( newtype, ierr )
548
549       ! receive count and displacement for MPI_GATHERV in stg_generate_seed_yz
550       ALLOCATE( recv_count_xz(pdims(2)), displs_xz(pdims(2)) )
551
552       recv_count_xz           = nzt_y_stg-nzb_y_stg + 1
553       recv_count_xz(pdims(2)) = recv_count_xz(pdims(2)) + 1
554
555       DO  j = 1, pdims(2)
556          displs_xz(j) = 0 + (nzt_y_stg-nzb_y_stg+1) * (j-1)
557       ENDDO
558
559    ENDIF
560
[2669]561#endif
[2259]562!
563!-- Define seed of random number
564    CALL RANDOM_SEED()
565    CALL RANDOM_SEED( size=nseed )
566    ALLOCATE( seed(1:nseed) )
567    DO  j = 1, nseed
568       seed(j) = startseed + j
569    ENDDO
570    CALL RANDOM_SEED(put=seed)
571
572!-- Read inflow profile
573!-- Height levels of profiles in input profile is as follows:
574!-- zu: luy, luz, tu, lvy, lvz, tv, r11, r21, r22, d1, d2, d5
575!-- zw: lwy, lwz, tw, r31, r32, r33, d3
576!-- WARNING: zz is not used at the moment
[2938]577    INQUIRE( FILE = 'STG_PROFILES' // TRIM( coupling_char ),                   &
578             EXIST = file_stg_exist  )
[2259]579
[2938]580    IF ( file_stg_exist )  THEN
[2259]581
[2938]582       OPEN( 90, FILE='STG_PROFILES'//TRIM( coupling_char ), STATUS='OLD',     &
583                      FORM='FORMATTED')
584!
585!--    Skip header
586       READ( 90, * )
[2259]587
[2938]588       DO  k = nzb, nzt+1
589          READ( 90, * ) zz, luy, luz, tu(k), lvy, lvz, tv(k), lwy, lwz, tw(k), &
590                        r11(k), r21(k), r22(k), r31(k), r32(k), r33(k),        &
591                        d1, d2, d3, d5
592
[2259]593!
[2938]594!--       Convert length scales from meter to number of grid points
595          nuy(k) = INT( luy * ddy )
596          nuz(k) = INT( luz / dz  )
597          nvy(k) = INT( lvy * ddy )
598          nvz(k) = INT( lvz / dz  )
599          nwy(k) = INT( lwy * ddy )
600          nwz(k) = INT( lwz / dz  )
601!
602!--       Workaround, assume isotropic turbulence
603          nwx(k) = nwy(k)
604          nvx(k) = nvy(k)
605          nux(k) = nuy(k)
606!
607!--       Save Mean inflow profiles
608          IF ( TRIM( initializing_actions ) /= 'read_restart_data' ) THEN
609             mean_inflow_profiles(k,1) = d1
610             mean_inflow_profiles(k,2) = d2
611            !  mean_inflow_profiles(k,4) = d4
612             mean_inflow_profiles(k,5) = d5
613          ENDIF
614       ENDDO
[2259]615
[2938]616       CLOSE( 90 )
617   
618    ELSE
[2259]619!
[2938]620!--    Set-up defaul length scales. Assume exponentially decreasing length
621!--    scales and isotropic turbulence.
[2945]622!--    Typical length (time) scales of 100 m (s) should be a good compromise
[2938]623!--    between all stratrifications. Near-surface variances are fixed to
624!--    0.1 m2/s2, vertical fluxes are one order of magnitude smaller.
625!--    Vertical fluxes 
626       length_scale_surface = 100.0_wp
627       r_ii_0               = 0.1_wp
628       time_scale           = 100.0_wp
629       DO  k = nzb+1, nzt+1
630          dum_exp        = MERGE( -zu(k) / ( 0.3* zu(nzt) ),                   &
631                                  0.0_wp,                                      &
632                                  zu(k) > 0.3 * zu(nzt)                        &
633                                )
634          length_scale_z = length_scale_surface * EXP( dum_exp )
[2259]635
[2938]636          nux(k) = MAX( INT( length_scale_z * ddx     ), 1 )
637          nuy(k) = MAX( INT( length_scale_z * ddy     ), 1 )
638          nuz(k) = MAX( INT( length_scale_z * ddzw(k) ), 1 )
639          nvx(k) = MAX( INT( length_scale_z * ddx     ), 1 )
640          nvy(k) = MAX( INT( length_scale_z * ddy     ), 1 )
641          nvz(k) = MAX( INT( length_scale_z * ddzw(k) ), 1 )
642          nwx(k) = MAX( INT( length_scale_z * ddx     ), 1 )
643          nwy(k) = MAX( INT( length_scale_z * ddy     ), 1 )
644          nwz(k) = MAX( INT( length_scale_z * ddzw(k) ), 1 )
[2259]645
[2938]646          r11(k) = r_ii_0 * EXP( dum_exp )
647          r22(k) = r_ii_0 * EXP( dum_exp )
648          r33(k) = r_ii_0 * EXP( dum_exp )
[2259]649
[2938]650          r21(k) = 0.1_wp * r_ii_0 * EXP( dum_exp )
651          r31(k) = 0.1_wp * r_ii_0 * EXP( dum_exp )
652          r32(k) = 0.1_wp * r_ii_0 * EXP( dum_exp )
653
654          tu(k)  = time_scale
655          tv(k)  = time_scale
656          tw(k)  = time_scale
657
658       ENDDO
659       nux(nzb) = nux(nzb+1)
660       nuy(nzb) = nuy(nzb+1)
661       nuz(nzb) = nuz(nzb+1)
662       nvx(nzb) = nvx(nzb+1)
663       nvy(nzb) = nvy(nzb+1)
664       nvz(nzb) = nvz(nzb+1)
665       nwx(nzb) = nwx(nzb+1)
666       nwy(nzb) = nwy(nzb+1)
667       nwz(nzb) = nwz(nzb+1)
668
669       r11(nzb) = r11(nzb+1)
670       r22(nzb) = r22(nzb+1)
671       r33(nzb) = r33(nzb+1)
672
673       r21(nzb) = r11(nzb+1)
674       r31(nzb) = r31(nzb+1)
675       r32(nzb) = r32(nzb+1)
676
677       tu(nzb)  = time_scale
678       tv(nzb)  = time_scale
679       tw(nzb)  = time_scale
680
681    ENDIF
682
[2259]683!
[2938]684!-- Assign initial profiles
685    IF ( .NOT. forcing  .AND.  .NOT.  nest_domain )  THEN
686       u_init = mean_inflow_profiles(:,1)
687       v_init = mean_inflow_profiles(:,2)
688      !pt_init = mean_inflow_profiles(:,4)
689       e_init = MAXVAL( mean_inflow_profiles(:,5) )
690    ENDIF
691!
[2259]692!-- Calculate coefficient matrix from Reynolds stress tensor (Lund rotation)
693    DO  k = nzb, nzt+1
694       IF ( r11(k) > 0.0_wp )  THEN
695          a11(k) = SQRT( r11(k) )
696          a21(k) = r21(k) / a11(k)
697       ELSE
698          a11(k) = 0.0_wp
699          a21(k) = 0.0_wp
700       ENDIF
701
702       a22(k) = r22(k) - a21(k)**2
703       IF ( a22(k) > 0.0_wp )  THEN
704          a22(k) = SQRT( a22(k) )
705          a32(k) = ( r32(k) - a21(k) * a31(k) ) / a22(k)
706       ELSE
707          a22(k) = 0.0_wp
708          a32(k) = 0.0_wp
709       ENDIF
710
711!
712!--    a31, a32, a33 must be calculated with interpolated a11, a21, a22 (d11,
713!--    d21, d22) because of different vertical grid
714       IF ( k .le. nzt )  THEN
715          d11 = 0.5_wp * ( r11(k) + r11(k+1) )
716          IF ( d11 > 0.0_wp )  THEN
717             d11 = SQRT( d11 )
718             d21 = ( 0.5_wp * ( r21(k) + r21(k+1) ) ) / d11
719             a31(k) = r31(k) / d11
720          ELSE
721             d21 = 0.0_wp
722             a31(k) = 0.0_wp
723          ENDIF
724
725          d22 = 0.5_wp * ( r22(k) + r22(k+1) ) - d21 ** 2
726          IF ( d22 > 0.0_wp )  THEN
727             a32(k) = ( r32(k) - d21 * a31(k) ) / SQRT( d22 )
728          ELSE
729             a32(k) = 0.0_wp
730          ENDIF
731
732          a33(k) = r33(k) - a31(k) ** 2 - a32(k) ** 2
733          IF ( a33(k) > 0.0_wp )  THEN
734             a33(k) = SQRT( a33(k) )
735          ELSE
736             a33(k) = 0.0_wp
737          ENDIF
738       ELSE
739          a31(k) = a31(k-1)
740          a32(k) = a32(k-1)
741          a33(k) = a33(k-1)
742       ENDIF
743
744    ENDDO
745!
746!-- Define the size of the filter functions and allocate them.
747    merg = 0
748
749    ! arrays must be large enough to cover the largest length scale
750    DO  k = nzb, nzt+1
[2938]751       j = MAX( ABS(nux(k)), ABS(nuy(k)), ABS(nuz(k)), &
752                ABS(nvx(k)), ABS(nvy(k)), ABS(nvz(k)), &
753                ABS(nwx(k)), ABS(nwy(k)), ABS(nwz(k))  )
[2259]754       IF ( j > merg )  merg = j
755    ENDDO
756
757    merg  = 2 * merg
758    mergp = merg + nbgp
759
[2938]760    ALLOCATE ( bux(-merg:merg,nzb:nzt+1),                                      &
761               buy(-merg:merg,nzb:nzt+1),                                      &
762               buz(-merg:merg,nzb:nzt+1),                                      &
763               bvx(-merg:merg,nzb:nzt+1),                                      &
764               bvy(-merg:merg,nzb:nzt+1),                                      &
765               bvz(-merg:merg,nzb:nzt+1),                                      &
766               bwx(-merg:merg,nzb:nzt+1),                                      &
767               bwy(-merg:merg,nzb:nzt+1),                                      &
768               bwz(-merg:merg,nzb:nzt+1)  )
[2259]769
770!
[2938]771!-- Allocate velocity seeds for turbulence at xz-layer
772    ALLOCATE ( fu_xz( nzb:nzt+1,nxlg:nxrg), fuo_xz(nzb:nzt+1,nxlg:nxrg),       &
773               fv_xz( nzb:nzt+1,nxlg:nxrg), fvo_xz(nzb:nzt+1,nxlg:nxrg),       &
774               fw_xz( nzb:nzt+1,nxlg:nxrg), fwo_xz(nzb:nzt+1,nxlg:nxrg)  )
[2259]775
[2938]776!
777!-- Allocate velocity seeds for turbulence at yz-layer
778    ALLOCATE ( fu_yz( nzb:nzt+1,nysg:nyng), fuo_yz(nzb:nzt+1,nysg:nyng),       &
779               fv_yz( nzb:nzt+1,nysg:nyng), fvo_yz(nzb:nzt+1,nysg:nyng),       &
780               fw_yz( nzb:nzt+1,nysg:nyng), fwo_yz(nzb:nzt+1,nysg:nyng)  )
[2259]781
[2938]782    fu_xz  = 0.0_wp
783    fuo_xz = 0.0_wp
784    fv_xz  = 0.0_wp
785    fvo_xz = 0.0_wp
786    fw_xz  = 0.0_wp
787    fwo_xz = 0.0_wp
788
789    fu_yz  = 0.0_wp
790    fuo_yz = 0.0_wp
791    fv_yz  = 0.0_wp
792    fvo_yz = 0.0_wp
793    fw_yz  = 0.0_wp
794    fwo_yz = 0.0_wp
795
[2259]796!
797!-- Create filter functions
[2938]798    CALL stg_filter_func( nux, bux ) !filter ux
[2259]799    CALL stg_filter_func( nuy, buy ) !filter uy
800    CALL stg_filter_func( nuz, buz ) !filter uz
[2938]801    CALL stg_filter_func( nvx, bvx ) !filter vx
[2259]802    CALL stg_filter_func( nvy, bvy ) !filter vy
803    CALL stg_filter_func( nvz, bvz ) !filter vz
[2938]804    CALL stg_filter_func( nwx, bwx ) !filter wx
[2259]805    CALL stg_filter_func( nwy, bwy ) !filter wy
806    CALL stg_filter_func( nwz, bwz ) !filter wz
807
808#if defined( __parallel )
809    CALL MPI_BARRIER( comm2d, ierr )
810#endif
811
812!
[2938]813!-- In case of restart, calculate velocity seeds fu, fv, fw from former
814!   time step.
815!   Bug: fu, fv, fw are different in those heights where a11, a22, a33
816!        are 0 compared to the prerun. This is mostly for k=nzt+1.
[2259]817    IF ( TRIM( initializing_actions ) == 'read_restart_data' )  THEN
[2938]818       IF ( myidx == id_stg_left  .OR.  myidx == id_stg_right )  THEN
819
820          IF ( myidx == id_stg_left  )  i = -1
821          IF ( myidx == id_stg_right )  i = nxr+1
822
[2259]823          DO  j = nysg, nyng
824             DO  k = nzb, nzt+1
825
826                IF  ( a11(k) .NE. 0._wp ) THEN
[2938]827                   fu_yz(k,j) = ( u(k,j,i) / mc_factor - u_init(k) ) / a11(k)
[2259]828                ELSE
[2938]829                   fu_yz(k,j) = 0._wp
[2259]830                ENDIF
831
832                IF  ( a22(k) .NE. 0._wp ) THEN
[2938]833                   fv_yz(k,j) = ( v(k,j,i) / mc_factor - a21(k) * fu_yz(k,j) - &
834                               v_init(k) ) / a22(k)
[2259]835                ELSE
[2938]836                   fv_yz(k,j) = 0._wp
[2259]837                ENDIF
838
839                IF  ( a33(k) .NE. 0._wp ) THEN
[2938]840                   fw_yz(k,j) = ( w(k,j,i) / mc_factor - a31(k) * fu_yz(k,j) - &
841                               a32(k) * fv_yz(k,j) ) / a33(k)
[2259]842                ELSE
[2938]843                   fw_yz = 0._wp
[2259]844                ENDIF
845
846             ENDDO
847          ENDDO
848       ENDIF
849    ENDIF
850
851    CALL  cpu_log( log_point(57), 'synthetic_turbulence_gen', 'stop' )
852
853 END SUBROUTINE stg_init
854
855
856!------------------------------------------------------------------------------!
857! Description:
858! ------------
859!> Calculate filter function bxx from length scale nxx following Eg.9 and 10
860!> (Xie and Castro, 2008)
861!------------------------------------------------------------------------------!
862 SUBROUTINE stg_filter_func( nxx, bxx )
863
864
865    IMPLICIT NONE
866
867    INTEGER(iwp) :: k         !< loop index
868    INTEGER(iwp) :: n_k       !< length scale nXX in height k
869    INTEGER(iwp) :: n_k2      !< n_k * 2
870    INTEGER(iwp) :: nf        !< index for length scales
871
872    REAL(wp) :: bdenom        !< denominator for filter functions bXX
873    REAL(wp) :: qsi = 1.0_wp  !< minimization factor
874
875    INTEGER(iwp), DIMENSION(:) :: nxx(nzb:nzt+1)           !< length scale (in gp)
876
877    REAL(wp), DIMENSION(:,:) :: bxx(-merg:merg,nzb:nzt+1)  !< filter function
878
879
880    bxx = 0.0_wp
881
882    DO  k = nzb, nzt+1
883       bdenom = 0.0_wp
884       n_k    = nxx(k)
885       IF ( n_k /= 0 )  THEN
886          n_k2 = n_k * 2
887
888!
889!--       ( Eq.10 )^2
890          DO  nf = -n_k2, n_k2
891             bdenom = bdenom + EXP( -qsi * pi * ABS(nf) / n_k )**2
892          ENDDO
893
894!
895!--       ( Eq.9 )
896          bdenom = SQRT( bdenom )
897          DO  nf = -n_k2, n_k2
898             bxx(nf,k) = EXP( -qsi * pi * ABS(nf) / n_k ) / bdenom
899          ENDDO
900       ENDIF
901    ENDDO
902
903END SUBROUTINE stg_filter_func
904
905
906!------------------------------------------------------------------------------!
907! Description:
908! ------------
909!> Parin for &stg_par for synthetic turbulence generator
910!------------------------------------------------------------------------------!
911 SUBROUTINE stg_parin
912
913
914    IMPLICIT NONE
915
916    CHARACTER (LEN=80) ::  line   !< dummy string that contains the current line of the parameter file
917
918
[2776]919    NAMELIST /stg_par/   use_syn_turb_gen
[2259]920
921    line = ' '
922
923!
924!-- Try to find stg package
925    REWIND ( 11 )
926    line = ' '
927    DO WHILE ( INDEX( line, '&stg_par' ) == 0 )
928       READ ( 11, '(A)', END=10 )  line
929    ENDDO
930    BACKSPACE ( 11 )
931
932!
933!-- Read namelist
934    READ ( 11, stg_par )
935
936!
937!-- Set flag that indicates that the synthetic turbulence generator is switched
938!-- on
[2776]939    syn_turb_gen = .TRUE.
[2259]940
[2563]941
[2259]942 10 CONTINUE
943
944 END SUBROUTINE stg_parin
945
946
947!------------------------------------------------------------------------------!
948! Description:
949! ------------
[2894]950!> This routine reads the respective restart data.
[2576]951!------------------------------------------------------------------------------!
[2894]952 SUBROUTINE stg_rrd_global( found )
[2576]953
954
[2894]955    USE control_parameters,                                                    &
956        ONLY: length, restart_string
[2576]957
958
[2894]959    IMPLICIT NONE
[2576]960
[2894]961    LOGICAL, INTENT(OUT)  ::  found 
[2259]962
963
[2894]964    found = .TRUE.
965 
[2259]966
[2894]967    SELECT CASE ( restart_string(1:length) )
[2259]968
[2894]969       CASE ( 'mc_factor' )
970          READ ( 13 )  mc_factor
971       CASE ( 'use_syn_turb_gen' )
972          READ ( 13 )  use_syn_turb_gen
[2259]973
[2894]974       CASE DEFAULT
[2259]975
[2894]976          found = .FALSE.   
[2259]977
[2894]978    END SELECT
[2259]979
980
[2894]981 END SUBROUTINE stg_rrd_global
[2259]982
983
984!------------------------------------------------------------------------------!
985! Description:
986! ------------
987!> This routine writes the respective restart data.
988!------------------------------------------------------------------------------!
[2894]989 SUBROUTINE stg_wrd_global
[2259]990
991
992    IMPLICIT NONE
993
[2894]994    CALL wrd_write_string( 'mc_factor' )
[2259]995    WRITE ( 14 )  mc_factor
996
[2894]997    CALL wrd_write_string( 'use_syn_turb_gen' )
998    WRITE ( 14 )  use_syn_turb_gen
[2259]999
1000
[2894]1001 END SUBROUTINE stg_wrd_global
[2259]1002
[2894]1003
[2259]1004!------------------------------------------------------------------------------!
1005! Description:
1006! ------------
1007!> Create turbulent inflow fields for u, v, w with prescribed length scales and
1008!> Reynolds stress tensor after a method of Xie and Castro (2008), modified
1009!> following suggestions of Kim et al. (2013), and using a Lund rotation
1010!> (Lund, 1998).
1011!------------------------------------------------------------------------------!
1012 SUBROUTINE stg_main
1013
1014
1015    USE arrays_3d,                                                             &
1016        ONLY:  dzw
1017
1018    USE control_parameters,                                                    &
[2938]1019        ONLY:  dt_3d, forcing, intermediate_timestep_count,  nest_domain,      &
[2945]1020               rans_mode, simulated_time, volume_flow_initial
[2259]1021
[2938]1022    USE grid_variables,                                                        &
1023        ONLY:  dx, dy
1024
[2259]1025    USE indices,                                                               &
[2938]1026        ONLY:  wall_flags_0
[2259]1027
1028    USE statistics,                                                            &
1029        ONLY:  weight_substep
1030
[2945]1031    USE pmc_interface,                                                         &
1032        ONLY : rans_mode_parent
[2259]1033
[2945]1034
[2259]1035    IMPLICIT NONE
1036
[2938]1037    INTEGER(iwp) :: i           !< grid index in x-direction
[2259]1038    INTEGER(iwp) :: j           !< loop index in y-direction
1039    INTEGER(iwp) :: k           !< loop index in z-direction
1040
1041    REAL(wp) :: dt_stg          !< wheighted subtimestep
1042    REAL(wp) :: mc_factor_l     !< local mass flux correction factor
[2938]1043    REAL(wp) :: volume_flow     !< mass flux through lateral boundary
1044    REAL(wp) :: volume_flow_l   !< local mass flux through lateral boundary
[2259]1045
[2938]1046    REAL(wp), DIMENSION(nzb:nzt+1,nxlg:nxrg,5) :: dist_xz !< imposed disturbances at north/south boundary
1047    REAL(wp), DIMENSION(nzb:nzt+1,nysg:nyng,5) :: dist_yz !< imposed disturbances at left/right boundary
[2259]1048
1049
1050    CALL  cpu_log( log_point(57), 'synthetic_turbulence_gen', 'start' )
1051
1052!
1053!-- Calculate time step which is needed for filter functions
1054    dt_stg = dt_3d * weight_substep(intermediate_timestep_count)
1055
1056!
1057!-- Initial value of fu, fv, fw
1058    IF ( simulated_time == 0.0_wp .AND. .NOT. velocity_seed_initialized )  THEN
[2938]1059       CALL stg_generate_seed_yz( nuy, nuz, buy, buz, fu_yz, id_stg_left )
1060       CALL stg_generate_seed_yz( nvy, nvz, bvy, bvz, fv_yz, id_stg_left )
1061       CALL stg_generate_seed_yz( nwy, nwz, bwy, bwz, fw_yz, id_stg_left )
1062
[2945]1063       IF ( forcing  .OR.  ( nest_domain .AND.  rans_mode_parent  .AND.        &
1064                      .NOT.  rans_mode ) )  THEN
[2938]1065!
1066!--       Generate turbulence at right boundary
1067          CALL stg_generate_seed_yz( nuy, nuz, buy, buz, fu_yz, id_stg_right )
1068          CALL stg_generate_seed_yz( nvy, nvz, bvy, bvz, fv_yz, id_stg_right )
1069          CALL stg_generate_seed_yz( nwy, nwz, bwy, bwz, fw_yz, id_stg_right )
1070!
1071!--       Generate turbulence at north boundary
1072          CALL stg_generate_seed_xz( nux, nuz, bux, buz, fu_xz, id_stg_north )
1073          CALL stg_generate_seed_xz( nvx, nvz, bvx, bvz, fv_xz, id_stg_north )
1074          CALL stg_generate_seed_xz( nwx, nwz, bwx, bwz, fw_xz, id_stg_north )
1075!
1076!--       Generate turbulence at south boundary
1077          CALL stg_generate_seed_xz( nux, nuz, bux, buz, fu_xz, id_stg_south )
1078          CALL stg_generate_seed_xz( nvx, nvz, bvx, bvz, fv_xz, id_stg_south )
1079          CALL stg_generate_seed_xz( nwx, nwz, bwx, bwz, fw_xz, id_stg_south )
1080       ENDIF
[2259]1081       velocity_seed_initialized = .TRUE.
1082    ENDIF
1083!
1084!-- New set of fu, fv, fw
[2938]1085    CALL stg_generate_seed_yz( nuy, nuz, buy, buz, fuo_yz, id_stg_left )
1086    CALL stg_generate_seed_yz( nvy, nvz, bvy, bvz, fvo_yz, id_stg_left )
1087    CALL stg_generate_seed_yz( nwy, nwz, bwy, bwz, fwo_yz, id_stg_left )
[2259]1088
[2945]1089    IF ( forcing  .OR.  ( nest_domain .AND.  rans_mode_parent  .AND.           &
1090                   .NOT.  rans_mode ) )  THEN
[2938]1091!
1092!--       Generate turbulence at right boundary
1093          CALL stg_generate_seed_yz( nuy, nuz, buy, buz, fuo_yz, id_stg_right )
1094          CALL stg_generate_seed_yz( nvy, nvz, bvy, bvz, fvo_yz, id_stg_right )
1095          CALL stg_generate_seed_yz( nwy, nwz, bwy, bwz, fwo_yz, id_stg_right )
1096!
1097!--       Generate turbulence at north boundary
1098          CALL stg_generate_seed_xz( nux, nuz, bux, buz, fuo_xz, id_stg_north )
1099          CALL stg_generate_seed_xz( nvx, nvz, bvx, bvz, fvo_xz, id_stg_north )
1100          CALL stg_generate_seed_xz( nwx, nwz, bwx, bwz, fwo_xz, id_stg_north )
1101!
1102!--       Generate turbulence at south boundary
1103          CALL stg_generate_seed_xz( nux, nuz, bux, buz, fuo_xz, id_stg_south )
1104          CALL stg_generate_seed_xz( nvx, nvz, bvx, bvz, fvo_xz, id_stg_south )
1105          CALL stg_generate_seed_xz( nwx, nwz, bwx, bwz, fwo_xz, id_stg_south )
1106    ENDIF
1107!
1108!-- Turbulence generation at left and or right boundary
1109    IF ( myidx == id_stg_left  .OR.  myidx == id_stg_right )  THEN
[2259]1110
1111       DO  j = nysg, nyng
1112          DO  k = nzb, nzt + 1
1113!
1114!--          Update fu, fv, fw following Eq. 14 of Xie and Castro (2008)
1115             IF ( tu(k) == 0.0_wp )  THEN
[2938]1116                fu_yz(k,j) = fuo_yz(k,j)
[2259]1117             ELSE
[2938]1118                fu_yz(k,j) = fu_yz(k,j) * EXP( -pi * dt_stg * 0.5_wp / tu(k) ) +     &
1119                         fuo_yz(k,j) * SQRT( 1.0_wp - EXP( -pi * dt_stg / tu(k) ) )
[2259]1120             ENDIF
1121
1122             IF ( tv(k) == 0.0_wp )  THEN
[2938]1123                fv_yz(k,j) = fvo_yz(k,j)
[2259]1124             ELSE
[2938]1125                fv_yz(k,j) = fv_yz(k,j) * EXP( -pi * dt_stg * 0.5_wp / tv(k) ) +     &
1126                         fvo_yz(k,j) * SQRT( 1.0_wp - EXP( -pi * dt_stg / tv(k) ) )
[2259]1127             ENDIF
1128
1129             IF ( tw(k) == 0.0_wp )  THEN
[2938]1130                fw_yz(k,j) = fwo_yz(k,j)
[2259]1131             ELSE
[2938]1132                fw_yz(k,j) = fw_yz(k,j) * EXP( -pi * dt_stg * 0.5_wp / tw(k) ) +     &
1133                         fwo_yz(k,j) * SQRT( 1.0_wp - EXP( -pi * dt_stg / tw(k) ) )
[2259]1134             ENDIF
1135!
1136!--          Lund rotation following Eq. 17 in Xie and Castro (2008).
1137!--          Additional factors are added to improve the variance of v and w
1138             IF( k == 0 )  THEN
[2938]1139                dist_yz(k,j,1) = 0.0_wp
1140                dist_yz(k,j,2) = 0.0_wp
1141                dist_yz(k,j,3) = 0.0_wp
1142!                 dist_yz(k,j,4) = 0.0_wp
1143!                 dist_yz(k,j,5) = 0.0_wp
[2259]1144             ELSE
[2938]1145                dist_yz(k,j,1) = a11(k) * fu_yz(k,j)
[2259]1146                !experimental test of 1.2
[2945]1147                dist_yz(k,j,2) = ( SQRT( a22(k) / MAXVAL(a22) )                &
[2259]1148                                         * 1.2_wp )                            &
[2945]1149                                       * (   a21(k) * fu_yz(k,j)               &
[2938]1150                                           + a22(k) * fv_yz(k,j) )
[2945]1151                dist_yz(k,j,3) = ( SQRT(a33(k) / MAXVAL(a33) )                 &
[2259]1152                                         * 1.3_wp )                            &
[2945]1153                                       * (   a31(k) * fu_yz(k,j)               &
1154                                           + a32(k) * fv_yz(k,j)               &
[2938]1155                                           + a33(k) * fw_yz(k,j) )
[2259]1156                ! Calculation for pt and e not yet implemented
[2938]1157!                 dist_yz(k,j,4) = 0.0_wp
1158!                 dist_yz(k,j,5) = 0.0_wp
[2259]1159             ENDIF
1160
1161          ENDDO
1162       ENDDO
1163
1164!
1165!--    Mass flux correction following Kim et al. (2013)
1166!--    This correction factor insures that the mass flux is preserved at the
1167!--    inflow boundary
[2938]1168       IF ( .NOT. forcing  .AND.  .NOT. nest_domain )  THEN
1169          mc_factor_l = 0.0_wp
1170          mc_factor   = 0.0_wp
1171          DO  j = nys, nyn
1172             DO  k = nzb+1, nzt
1173                mc_factor_l = mc_factor_l + dzw(k)  *                          &
1174                              ( mean_inflow_profiles(k,1) + dist_yz(k,j,1) )
1175             ENDDO
[2259]1176          ENDDO
1177
1178#if defined( __parallel )
[2938]1179          CALL MPI_ALLREDUCE( mc_factor_l, mc_factor,  &
1180                              1, MPI_REAL, MPI_SUM, comm1dy, ierr )
[2259]1181#else
[2938]1182          mc_factor = mc_factor_l
[2259]1183#endif
1184
[2938]1185          mc_factor = volume_flow_initial(1) / mc_factor
[2259]1186
1187!
[2938]1188!--       Add disturbance at the inflow
1189          DO  j = nysg, nyng
1190             DO  k = nzb, nzt+1
1191                 u(k,j,-nbgp+1:0) = ( mean_inflow_profiles(k,1) +              &
1192                                      dist_yz(k,j,1)             ) * mc_factor
1193                 v(k,j,-nbgp:-1)  = ( mean_inflow_profiles(k,2) +              &
1194                                      dist_yz(k,j,2)             ) * mc_factor
1195                 w(k,j,-nbgp:-1)  =   dist_yz(k,j,3)               * mc_factor
1196             ENDDO
[2259]1197          ENDDO
[2938]1198
1199       ELSE
1200!
1201!--       First, calculate volume flow at yz boundary
1202          IF ( myidx == id_stg_left  )  i = nxl
1203          IF ( myidx == id_stg_right )  i = nxr+1
1204
1205          volume_flow_l = 0.0_wp
1206          volume_flow   = 0.0_wp
1207          mc_factor_l   = 0.0_wp
1208          mc_factor     = 0.0_wp
1209          DO  j = nys, nyn
1210             DO  k = nzb+1, nzt
1211                volume_flow_l = volume_flow_l + u(k,j,i) * dzw(k) * dy         &
1212                                     * MERGE( 1.0_wp, 0.0_wp,                  &   
1213                                              BTEST( wall_flags_0(k,j,i), 1 ) )
1214
1215                mc_factor_l = mc_factor_l     + ( u(k,j,i) + dist_yz(k,j,1) )  &
1216                                                         * dzw(k) * dy         &
1217                                     * MERGE( 1.0_wp, 0.0_wp,                  &   
1218                                              BTEST( wall_flags_0(k,j,i), 1 ) )
1219             ENDDO
1220          ENDDO
1221#if defined( __parallel )
1222          CALL MPI_ALLREDUCE( volume_flow_l, volume_flow,                      &
1223                              1, MPI_REAL, MPI_SUM, comm1dy, ierr )
1224          CALL MPI_ALLREDUCE( mc_factor_l, mc_factor,                          &
1225                              1, MPI_REAL, MPI_SUM, comm1dy, ierr )
1226#else
1227          volume_flow = volume_flow_l
1228          mc_factor   = mc_factor_l
1229#endif
1230
1231          mc_factor = volume_flow / mc_factor
1232
1233!
1234!--       Add disturbances
1235          IF ( myidx == id_stg_left  )  THEN
1236
1237             DO  j = nysg, nyng
1238                DO  k = nzb, nzt+1
1239                   u(k,j,-nbgp+1:0) = ( u(k,j,-nbgp+1:0) + dist_yz(k,j,1) )    &
1240                                        * mc_factor
1241                   v(k,j,-nbgp:-1)  = ( v(k,j,-nbgp:-1)  + dist_yz(k,j,2)  )   &
1242                                        * mc_factor
1243                   w(k,j,-nbgp:-1)  = ( w(k,j,-nbgp:-1)  + dist_yz(k,j,3)  )   &
1244                                        * mc_factor
1245                ENDDO
1246             ENDDO
1247          ENDIF
1248          IF ( myidx == id_stg_right  )  THEN
1249
1250             DO  j = nysg, nyng
1251                DO  k = nzb, nzt+1
1252                   u(k,j,nxr+1:nxr+nbgp) = ( u(k,j,nxr+1:nxr+nbgp) +           &
1253                                             dist_yz(k,j,1) ) * mc_factor
1254                   v(k,j,nxr+1:nxr+nbgp) = ( v(k,j,nxr+1:nxr+nbgp) +           &
1255                                             dist_yz(k,j,2) ) * mc_factor
1256                   w(k,j,nxr+1:nxr+nbgp) = ( w(k,j,nxr+1:nxr+nbgp) +           &
1257                                             dist_yz(k,j,3) ) * mc_factor
1258                ENDDO
1259             ENDDO
1260          ENDIF
1261       ENDIF
1262
1263    ENDIF
1264!
1265!-- Turbulence generation at north and south boundary
1266    IF ( myidy == id_stg_north  .OR.  myidy == id_stg_south )  THEN
1267
1268       DO  i = nxlg, nxrg
1269          DO  k = nzb, nzt + 1
1270!
1271!--          Update fu, fv, fw following Eq. 14 of Xie and Castro (2008)
1272             IF ( tu(k) == 0.0_wp )  THEN
1273                fu_xz(k,i) = fuo_xz(k,i)
1274             ELSE
1275                fu_xz(k,i) = fu_xz(k,i) * EXP( -pi * dt_stg * 0.5_wp / tu(k) ) +     &
1276                         fuo_xz(k,i) * SQRT( 1.0_wp - EXP( -pi * dt_stg / tu(k) ) )
1277             ENDIF
1278
1279             IF ( tv(k) == 0.0_wp )  THEN
1280                fv_xz(k,i) = fvo_xz(k,i)
1281             ELSE
1282                fv_xz(k,i) = fv_xz(k,i) * EXP( -pi * dt_stg * 0.5_wp / tv(k) ) +     &
1283                         fvo_xz(k,i) * SQRT( 1.0_wp - EXP( -pi * dt_stg / tv(k) ) )
1284             ENDIF
1285
1286             IF ( tw(k) == 0.0_wp )  THEN
1287                fw_xz(k,i) = fwo_xz(k,i)
1288             ELSE
1289                fw_xz(k,i) = fw_xz(k,i) * EXP( -pi * dt_stg * 0.5_wp / tw(k) ) +     &
1290                         fwo_xz(k,i) * SQRT( 1.0_wp - EXP( -pi * dt_stg / tw(k) ) )
1291             ENDIF
1292!
1293!--          Lund rotation following Eq. 17 in Xie and Castro (2008).
1294!--          Additional factors are added to improve the variance of v and w
1295             IF( k == 0 )  THEN
1296                dist_xz(k,i,1) = 0.0_wp
1297                dist_xz(k,i,2) = 0.0_wp
1298                dist_xz(k,i,3) = 0.0_wp
1299
1300             ELSE
1301                dist_xz(k,i,1) = a11(k) * fu_xz(k,i)
1302                !experimental test of 1.2
1303                dist_xz(k,i,2) = ( SQRT( a22(k) / MAXVAL(a22) )                &
1304                                         * 1.2_wp )                            &
1305                                       * (   a21(k) * fu_xz(k,i)               &
1306                                           + a22(k) * fv_xz(k,i) )
1307                dist_xz(k,i,3) = ( SQRT(a33(k) / MAXVAL(a33) )                 &
1308                                         * 1.3_wp )                            &
1309                                       * (   a31(k) * fu_xz(k,i)               &
1310                                           + a32(k) * fv_xz(k,i)               &
1311                                           + a33(k) * fw_xz(k,i) )
1312             ENDIF
1313
1314          ENDDO
[2259]1315       ENDDO
[2938]1316!
1317!--    Mass flux correction following Kim et al. (2013)
1318!--    This correction factor insures that the mass flux is preserved at the
1319!--    inflow boundary.
1320!--    First, calculate volume flow at xz boundary
1321       IF ( myidy == id_stg_south  ) j = nys
1322       IF ( myidy == id_stg_north )  j = nyn+1
[2259]1323
[2938]1324       volume_flow_l = 0.0_wp
1325       volume_flow   = 0.0_wp
1326       mc_factor_l   = 0.0_wp
1327       mc_factor     = 0.0_wp
1328       DO  i = nxl, nxr
1329          DO  k = nzb+1, nzt
[2945]1330             volume_flow_l = volume_flow_l + v(k,j,i) * dzw(k) * dx            &
1331                                  * MERGE( 1.0_wp, 0.0_wp,                     &   
[2938]1332                                           BTEST( wall_flags_0(k,j,i), 2 ) )
1333
[2945]1334             mc_factor_l = mc_factor_l     + ( v(k,j,i) + dist_xz(k,i,2) )     &
1335                                                      * dzw(k) * dx            &
1336                                  * MERGE( 1.0_wp, 0.0_wp,                     &   
[2938]1337                                           BTEST( wall_flags_0(k,j,i), 2 ) )
1338          ENDDO
1339       ENDDO
1340#if defined( __parallel )
[2945]1341       CALL MPI_ALLREDUCE( volume_flow_l, volume_flow,                         &
[2938]1342                           1, MPI_REAL, MPI_SUM, comm1dx, ierr )
[2945]1343       CALL MPI_ALLREDUCE( mc_factor_l, mc_factor,                             &
[2938]1344                           1, MPI_REAL, MPI_SUM, comm1dx, ierr )
1345#else
1346       volume_flow = volume_flow_l
1347       mc_factor   = mc_factor_l
1348#endif
1349
1350       mc_factor = volume_flow / mc_factor
1351
1352!
1353!--    Add disturbances
1354       IF ( myidy == id_stg_south  )  THEN
1355
1356          DO  i = nxlg, nxrg
1357             DO  k = nzb, nzt+1
1358                u(k,-nbgp:-1,i) = ( u(k,-nbgp:-1,i) + dist_xz(k,i,1) )         &
1359                                     * mc_factor
1360                v(k,-nbgp:0,i)  = ( v(k,-nbgp:0,i)  + dist_xz(k,i,2)  )        &
1361                                     * mc_factor
1362                w(k,-nbgp:-1,i) = ( w(k,-nbgp:-1,i) + dist_xz(k,i,3)  )        &
1363                                     * mc_factor
1364             ENDDO
1365          ENDDO
1366       ENDIF
1367       IF ( myidy == id_stg_north  )  THEN
1368
1369          DO  i = nxlg, nxrg
1370             DO  k = nzb, nzt+1
1371                u(k,nyn+1:nyn+nbgp,i) = ( u(k,nyn+1:nyn+nbgp,i) +              &
1372                                          dist_xz(k,i,1) ) * mc_factor
1373                v(k,nyn+1:nyn+nbgp,i) = ( v(k,nyn+1:nyn+nbgp,i) +              &
1374                                          dist_xz(k,i,2) ) * mc_factor
1375                w(k,nyn+1:nyn+nbgp,i) = ( w(k,nyn+1:nyn+nbgp,i) +              &
1376                                          dist_xz(k,i,3) ) * mc_factor
1377             ENDDO
1378          ENDDO
1379       ENDIF
[2259]1380    ENDIF
1381
1382    CALL  cpu_log( log_point(57), 'synthetic_turbulence_gen', 'stop' )
1383
1384 END SUBROUTINE stg_main
1385
1386!------------------------------------------------------------------------------!
1387! Description:
1388! ------------
1389!> Generate a set of random number rand_it wich is equal on each PE
1390!> and calculate the velocity seed f_n.
1391!> f_n is splitted in vertical direction by the number of PEs in x-direction and
1392!> and each PE calculates a vertical subsection of f_n. At the the end, all
1393!> parts are collected to form the full array.
1394!------------------------------------------------------------------------------!
[2938]1395 SUBROUTINE stg_generate_seed_yz( n_y, n_z, b_y, b_z, f_n, id )
[2259]1396
1397
1398    USE indices,                                                               &
1399        ONLY: ny
1400
1401    IMPLICIT NONE
1402
[2938]1403    INTEGER(iwp) :: id          !< core ids at respective boundaries
[2259]1404    INTEGER(iwp) :: j           !< loop index in y-direction
1405    INTEGER(iwp) :: jj          !< loop index in y-direction
1406    INTEGER(iwp) :: k           !< loop index in z-direction
1407    INTEGER(iwp) :: kk          !< loop index in z-direction
1408    INTEGER(iwp) :: send_count  !< send count for MPI_GATHERV
1409
1410    INTEGER(iwp), DIMENSION(nzb:nzt+1) :: n_y    !< length scale in y-direction
1411    INTEGER(iwp), DIMENSION(nzb:nzt+1) :: n_z    !< length scale in z-direction
1412    INTEGER(iwp), DIMENSION(nzb:nzt+1) :: n_y2   !< n_y*2
1413    INTEGER(iwp), DIMENSION(nzb:nzt+1) :: n_z2   !< n_z*2
1414
1415    REAL(wp) :: nyz_inv         !< inverse of number of grid points in yz-slice
1416    REAL(wp) :: rand_av         !< average of random number
1417    REAL(wp) :: rand_sigma_inv  !< inverse of stdev of random number
1418
1419    REAL(wp), DIMENSION(-merg:merg,nzb:nzt+1)    :: b_y     !< filter func in y-dir
1420    REAL(wp), DIMENSION(-merg:merg,nzb:nzt+1)    :: b_z     !< filter func in z-dir
[2938]1421    REAL(wp), DIMENSION(nzb_x_stg:nzt_x_stg+1,nysg:nyng) :: f_n_l   !<  local velocity seed
[2259]1422    REAL(wp), DIMENSION(nzb:nzt+1,nysg:nyng)     :: f_n     !<  velocity seed
1423    REAL(wp), DIMENSION(:,:), ALLOCATABLE        :: rand_it !<  random number
1424
1425
1426!
1427!-- Generate random numbers using a seed generated in stg_init.
1428!-- The set of random numbers are modified to have an average of 0 and
1429!-- unit variance.
1430    ALLOCATE( rand_it(nzb-mergp:nzt+1+mergp,-mergp:ny+mergp) )
1431
1432    rand_av        = 0.0_wp
1433    rand_sigma_inv = 0.0_wp
1434    nyz_inv        = 1.0_wp / REAL( ( nzt+1 - nzb+1 ) * ( ny+1 ), KIND=wp )
1435
1436    DO  j = 0, ny
1437       DO  k = nzb, nzt+1
1438          CALL RANDOM_NUMBER( rand_it(k,j) )
1439          rand_av = rand_av + rand_it(k,j)
1440       ENDDO
1441    ENDDO
1442
1443    rand_av = rand_av * nyz_inv
1444
1445    DO  j = 0, ny
1446       DO  k = nzb, nzt+1
1447          rand_it(k,j)   = rand_it(k,j) - rand_av
1448          rand_sigma_inv = rand_sigma_inv + rand_it(k,j) ** 2
1449       ENDDO
1450    ENDDO
1451
1452    rand_sigma_inv = 1.0_wp / SQRT(rand_sigma_inv * nyz_inv)
1453
1454    DO  j = 0, ny
1455       DO  k = nzb, nzt+1
1456          rand_it(k,j) = rand_it(k,j) * rand_sigma_inv
1457       ENDDO
1458    ENDDO
1459
1460!
1461!-- Periodic fill of random number in space
1462    DO  j = 0, ny
1463       DO  k = 1, mergp
1464          rand_it(nzb  -k,j) = rand_it(nzt+2-k,j)    ! bottom margin
1465          rand_it(nzt+1+k,j) = rand_it(nzb+k-1,j)    ! top margin
1466       ENDDO
1467    ENDDO
1468    DO  j = 1, mergp
1469       DO  k = nzb-mergp, nzt+1+mergp
1470          rand_it(k,  -j) = rand_it(k,ny-j+1)        ! south margin
1471          rand_it(k,ny+j) = rand_it(k,   j-1)        ! north margin
1472       ENDDO
1473    ENDDO
1474
1475!
1476!-- Generate velocity seed following Eq.6 of Xie and Castro (2008)
1477    n_y2 = n_y * 2
1478    n_z2 = n_z * 2
1479    f_n_l  = 0.0_wp
1480
1481    DO  j = nysg, nyng
[2938]1482       DO  k = nzb_x_stg, nzt_x_stg+1
[2259]1483          DO  jj = -n_y2(k), n_y2(k)
1484             DO  kk = -n_z2(k), n_z2(k)
1485                f_n_l(k,j) = f_n_l(k,j)                                        &
1486                           + b_y(jj,k) * b_z(kk,k) * rand_it(k+kk,j+jj)
1487             ENDDO
1488          ENDDO
1489       ENDDO
1490    ENDDO
1491
1492    DEALLOCATE( rand_it )
1493!
1494!-- Gather velocity seeds of full subdomain
[2938]1495    send_count = nzt_x_stg - nzb_x_stg + 1
1496    IF ( nzt_x_stg == nzt )  send_count = send_count + 1
[2259]1497
1498#if defined( __parallel )
[2945]1499    CALL MPI_GATHERV( f_n_l(nzb_x_stg,nysg), send_count, stg_type_yz_small,    &
1500                      f_n(nzb+1,nysg), recv_count_yz, displs_yz, stg_type_yz,  &
[2938]1501                      id, comm1dx, ierr )
[2259]1502#else
[2938]1503    f_n(nzb+1:nzt+1,nysg:nyng) = f_n_l(nzb_x_stg:nzt_x_stg+1,nysg:nyng)
[2259]1504#endif
1505
1506
1507 END SUBROUTINE stg_generate_seed_yz
1508
[2938]1509
1510
1511
1512!------------------------------------------------------------------------------!
1513! Description:
1514! ------------
1515!> Generate a set of random number rand_it wich is equal on each PE
1516!> and calculate the velocity seed f_n.
1517!> f_n is splitted in vertical direction by the number of PEs in y-direction and
1518!> and each PE calculates a vertical subsection of f_n. At the the end, all
1519!> parts are collected to form the full array.
1520!------------------------------------------------------------------------------!
1521 SUBROUTINE stg_generate_seed_xz( n_x, n_z, b_x, b_z, f_n, id )
1522
1523
1524    USE indices,                                                               &
1525        ONLY: nx
1526
1527
1528    IMPLICIT NONE
1529
1530    INTEGER(iwp) :: id          !< core ids at respective boundaries
1531    INTEGER(iwp) :: i           !< loop index in x-direction
1532    INTEGER(iwp) :: ii          !< loop index in x-direction
1533    INTEGER(iwp) :: k           !< loop index in z-direction
1534    INTEGER(iwp) :: kk          !< loop index in z-direction
1535    INTEGER(iwp) :: send_count  !< send count for MPI_GATHERV
1536
1537    INTEGER(iwp), DIMENSION(nzb:nzt+1) :: n_x    !< length scale in x-direction
1538    INTEGER(iwp), DIMENSION(nzb:nzt+1) :: n_z    !< length scale in z-direction
1539    INTEGER(iwp), DIMENSION(nzb:nzt+1) :: n_x2   !< n_y*2
1540    INTEGER(iwp), DIMENSION(nzb:nzt+1) :: n_z2   !< n_z*2
1541
1542    REAL(wp) :: nxz_inv         !< inverse of number of grid points in xz-slice
1543    REAL(wp) :: rand_av         !< average of random number
1544    REAL(wp) :: rand_sigma_inv  !< inverse of stdev of random number
1545
1546    REAL(wp), DIMENSION(-merg:merg,nzb:nzt+1)    :: b_x     !< filter func in y-dir
1547    REAL(wp), DIMENSION(-merg:merg,nzb:nzt+1)    :: b_z     !< filter func in z-dir
1548    REAL(wp), DIMENSION(nzb_y_stg:nzt_y_stg+1,nxlg:nxrg) :: f_n_l   !<  local velocity seed
1549    REAL(wp), DIMENSION(nzb:nzt+1,nxlg:nxrg)     :: f_n     !<  velocity seed
1550    REAL(wp), DIMENSION(:,:), ALLOCATABLE        :: rand_it !<  random number
1551
1552
1553!
1554!-- Generate random numbers using a seed generated in stg_init.
1555!-- The set of random numbers are modified to have an average of 0 and
1556!-- unit variance.
1557    ALLOCATE( rand_it(nzb-mergp:nzt+1+mergp,-mergp:nx+mergp) )
1558
1559    rand_av        = 0.0_wp
1560    rand_sigma_inv = 0.0_wp
1561    nxz_inv        = 1.0_wp / REAL( ( nzt+1 - nzb+1 ) * ( nx+1 ), KIND=wp )
1562
1563    DO  i = 0, nx
1564       DO  k = nzb, nzt+1
1565          CALL RANDOM_NUMBER( rand_it(k,i) )
1566          rand_av = rand_av + rand_it(k,i)
1567       ENDDO
1568    ENDDO
1569
1570    rand_av = rand_av * nxz_inv
1571
1572    DO  i = 0, nx
1573       DO  k = nzb, nzt+1
1574          rand_it(k,i)   = rand_it(k,i) - rand_av
1575          rand_sigma_inv = rand_sigma_inv + rand_it(k,i) ** 2
1576       ENDDO
1577    ENDDO
1578
1579    rand_sigma_inv = 1.0_wp / SQRT(rand_sigma_inv * nxz_inv)
1580
1581    DO  i = 0, nx
1582       DO  k = nzb, nzt+1
1583          rand_it(k,i) = rand_it(k,i) * rand_sigma_inv
1584       ENDDO
1585    ENDDO
1586
1587!
1588!-- Periodic fill of random number in space
1589    DO  i = 0, nx
1590       DO  k = 1, mergp
1591          rand_it(nzb-k,i)   = rand_it(nzt+2-k,i)    ! bottom margin
1592          rand_it(nzt+1+k,i) = rand_it(nzb+k-1,i)    ! top margin
1593       ENDDO
1594    ENDDO
1595    DO  i = 1, mergp
1596       DO  k = nzb-mergp, nzt+1+mergp
1597          rand_it(k,-i)   = rand_it(k,nx-i+1)        ! left margin
1598          rand_it(k,nx+i) = rand_it(k,i-1)           ! right margin
1599       ENDDO
1600    ENDDO
1601
1602!
1603!-- Generate velocity seed following Eq.6 of Xie and Castro (2008)
1604    n_x2 = n_x * 2
1605    n_z2 = n_z * 2
1606    f_n_l  = 0.0_wp
1607
1608    DO  i = nxlg, nxrg
1609       DO  k = nzb_y_stg, nzt_y_stg+1
1610          DO  ii = -n_x2(k), n_x2(k)
1611             DO  kk = -n_z2(k), n_z2(k)
1612                f_n_l(k,i) = f_n_l(k,i)                                        &
1613                           + b_x(ii,k) * b_z(kk,k) * rand_it(k+kk,i+ii)
1614             ENDDO
1615          ENDDO
1616       ENDDO
1617    ENDDO
1618
1619    DEALLOCATE( rand_it )
1620
1621!
1622!-- Gather velocity seeds of full subdomain
1623    send_count = nzt_y_stg - nzb_y_stg + 1
1624    IF ( nzt_y_stg == nzt )  send_count = send_count + 1
1625
1626
1627#if defined( __parallel )
[2945]1628    CALL MPI_GATHERV( f_n_l(nzb_y_stg,nxlg), send_count, stg_type_xz_small,    &
[2938]1629                      f_n(nzb+1,nxlg), recv_count_xz, displs_xz, stg_type_xz,  &
1630                      id, comm1dy, ierr )
1631#else
1632    f_n(nzb+1:nzt+1,nxlg:nxrg) = f_n_l(nzb_y_stg:nzt_y_stg+1,nxlg:nxrg)
1633#endif
1634
1635
1636 END SUBROUTINE stg_generate_seed_xz
1637
[2259]1638 END MODULE synthetic_turbulence_generator_mod
Note: See TracBrowser for help on using the repository browser.