Home

Context Navigation

turbulence_closure_mod.f90 @ 4110

Last change on this file since 4110 was 4110, checked in by suehring, 4 years ago
last changes documented
Property svn:keywords set to `Id`
File size: 206.2 KB

Line
1	!> @file turbulence_closure_mod.f90
2	!------------------------------------------------------------------------------!
3	! This file is part of the PALM model system.
4	!
5	! PALM is free software: you can redistribute it and/or modify it under the
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	!
10	! PALM is distributed in the hope that it will be useful, but WITHOUT ANY
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-2018 Leibniz Universitaet Hannover
18	!--------------------------------------------------------------------------------!
19	!
20	! Current revisions:
21	! -----------------
22	!
23	!
24	! Former revisions:
25	! -----------------
26	! $Id: turbulence_closure_mod.f90 4110 2019-07-22 17:05:21Z suehring $
27	! pass integer flag array as well as boundary flags to WS scalar advection
28	! routine
29	!
30	! 4109 2019-07-22 17:00:34Z suehring
31	! - Modularize setting of boundary conditions for TKE and dissipation
32	! - Neumann boundary condition for TKE at model top is set also in child domain
33	! - Revise setting of Neumann boundary conditions at non-cyclic lateral
34	! boundaries
35	! - Bugfix, set Neumann boundary condition for TKE at vertical wall instead of
36	! an implicit Dirichlet boundary condition which implied a sink of TKE
37	! at vertical walls
38	!
39	! 4048 2019-06-21 21:00:21Z knoop
40	! write out preprocessor directives; remove tailing whitespaces
41	!
42	! 3775 2019-03-04 12:40:20Z gronemeier
43	! removed unused variables
44	!
45	! 3724 2019-02-06 16:28:23Z kanani
46	! Correct double-used log_point_s units
47	!
48	! 3719 2019-02-06 13:10:18Z kanani
49	! Changed log_point to log_point_s, otherwise this overlaps with
50	! 'all progn.equations' cpu measurement.
51	!
52	! 3684 2019-01-20 20:20:58Z knoop
53	! Remove unused variable simulated_time
54	!
55	! 3636 2018-12-19 13:48:34Z raasch
56	! nopointer option removed
57	!
58	! 3634 2018-12-18 12:31:28Z knoop
59	! OpenACC port for SPEC
60	!
61	! 3550 2018-11-21 16:01:01Z gronemeier
62	! - calculate diss production same in vector and cache optimization
63	! - move boundary condition for e and diss to boundary_conds
64	!
65	! 3545 2018-11-21 11:19:41Z gronemeier
66	! - Set rans_mode according to value of turbulence_closure
67	! - removed debug output
68	!
69	! 3430 2018-10-25 13:36:23Z maronga
70	! Added support for buildings in the dynamic SGS model
71	!
72	! 3398 2018-10-22 19:30:24Z knoop
73	! Refactored production_e and production_e_ij (removed code redundancy)
74	!
75	! 3386 2018-10-19 16:28:22Z gronemeier
76	! Renamed tcm_prognostic to tcm_prognostic_equations
77	!
78	! 3385 2018-10-19 14:52:29Z knoop
79	! Restructured loops and ifs in production_e to ease vectorization on GPUs
80	!
81	! 3300 2018-10-02 14:16:54Z gronemeier
82	! - removed global array wall_flags_0_global, hence reduced accuracy of l_wall
83	! calculation
84	! - removed maxloc call as this produced different results for different
85	! compiler options
86	!
87	! 3294 2018-10-01 02:37:10Z raasch
88	! changes concerning modularization of ocean option
89	!
90	! 3274 2018-09-24 15:42:55Z knoop
91	! Modularization of all bulk cloud physics code components
92	!
93	! 3245 2018-09-13 14:08:16Z knoop
94	! unused variables removed, shortest_distance has wp now
95	!
96	! 3183 2018-07-27 14:25:55Z suehring
97	! Rename variables and remove unused variable from USE statement
98	!
99	! 3182 2018-07-27 13:36:03Z suehring
100	! Use MOST for km only in RANS mode
101	!
102	! 3130 2018-07-16 11:08:55Z gronemeier
103	! - move boundary condition of km and kh to tcm_diffusivities
104	! - calculate km at boundaries according to MOST
105	! - move phi_m to surface_layer_fluxes_mod
106	!
107	! 3129 2018-07-16 07:45:13Z gronemeier
108	! - move limitation of diss to boundary_conds
109	! - move boundary conditions for e and diss to boundary_conds
110	! - consider non-default surfaces in tcm_diffusivities
111	! - use z_mo within surface layer instead of calculating it
112	! - resort output after case select -> reduced code duplication
113	! - when using rans_tke_e and 1d-model, do not use e1d, km1d and diss1d
114	!
115	! 3121 2018-07-11 18:46:49Z gronemeier
116	! - created the function phi_m
117	! - implemented km = u* * kappa * zp / phi_m in production_e_init for all
118	! surfaces
119	!
120	! 3120 2018-07-11 18:30:57Z gronemeier
121	! - changed tcm_diffusivities to tcm_diffusivities_default
122	! - created subroutine tcm_diffusivities that calls tcm_diffusivities_default
123	! and tcm_diffusivities_dynamic
124	!
125	! 3086 2018-06-25 09:08:04Z gronemeier
126	! bugfix: set rans_const_sigma(1) = 1.3
127	!
128	! 3083 2018-06-19 14:03:12Z gronemeier
129	! - set limits of diss at the end of prognostic equations
130	! - call production_e to calculate production term of diss
131	! - limit change of diss to -90% to +100%
132	! - remove factor 0.5 from diffusion_diss_ij
133	! - rename c_m into c_0, and c_h into c_4
134	! - add rans_const_c and rans_const_sigma as namelist parameters
135	! - add calculation of mixing length for profile output in case of rans_tke_e
136	! - changed format of annotations to comply with doxygen standards
137	! - calculate and save dissipation rate during rans_tke_l mode
138	! - set bc at vertical walls for e, diss, km, kh
139	! - bugfix: set l_wall = 0.0 within buildings
140	! - set l_wall at bottom and top boundary (rans-mode)
141	! - bugfix in production term for dissipation rate
142	! - bugfix in diffusion of dissipation rate
143	! - disable check for 1D model if rans_tke_e is used
144	! - bugfixes for initialization (rans-mode):
145	! - correction of dissipation-rate formula
146	! - calculate km based on l_wall
147	! - initialize diss if 1D model is not used
148	!
149	! 3045 2018-05-28 07:55:41Z Giersch
150	! Error message revised
151	!
152	! 3014 2018-05-09 08:42:38Z maronga
153	! Bugfix: nzb_do and nzt_do were not used for 3d data output
154	!
155	! 3004 2018-04-27 12:33:25Z Giersch
156	! Further allocation checks implemented
157	!
158	! 2938 2018-03-27 15:52:42Z suehring
159	! Further todo's
160	!
161	! 2936 2018-03-27 14:49:27Z gronemeier
162	! - defined l_grid only within this module
163	! - Moved l_wall definition from modules.f90
164	! - Get level of highest topography, used to limit upward distance calculation
165	! - Consider cyclic boundary conditions for mixing length calculation
166	! - Moved copy of wall_flags into subarray to subroutine
167	! - Implemented l_wall calculation in case of RANS simulation
168	! - Moved init of l_black to tcm_init_mixing_length
169	! - Moved init_mixing_length from init_grid.f90 and
170	! renamed it to tcm_init_mixing_length
171	!
172	! 2764 2018-01-22 09:25:36Z gronemeier
173	! Bugfix: remove duplicate SAVE statements
174	!
175	! 2746 2018-01-15 12:06:04Z suehring
176	! Move flag plant canopy to modules
177	!
178	! 2718 2018-01-02 08:49:38Z maronga
179	! Corrected "Former revisions" section
180	!
181	! 2701 2017-12-15 15:40:50Z suehring
182	! Changes from last commit documented
183	!
184	! 2698 2017-12-14 18:46:24Z suehring
185	! Bugfix in get_topography_top_index
186	!
187	! 2696 2017-12-14 17:12:51Z kanani
188	! Initial revision
189	!
190	!
191	!
192	!
193	! Authors:
194	! --------
195	! @author Tobias Gronemeier
196	! @author Hauke Wurps
197	!
198	! Description:
199	! ------------
200	!> This module contains the available turbulence closures for PALM.
201	!>
202	!>
203	!> @todo test initialization for all possibilities
204	!> add OpenMP directives whereever possible
205	!> @todo Check for random disturbances
206	!> @note <Enter notes on the module>
207	!-----------------------------------------------------------------------------!
208	MODULE turbulence_closure_mod
209
210
211	USE arrays_3d, &
212	ONLY: diss, diss_1, diss_2, diss_3, diss_p, dzu, e, e_1, e_2, e_3, &
213	e_p, kh, km, mean_inflow_profiles, prho, pt, tdiss_m, &
214	te_m, tend, u, v, vpt, w
215
216	USE basic_constants_and_equations_mod, &
217	ONLY: g, kappa, lv_d_cp, lv_d_rd, rd_d_rv
218
219	USE control_parameters, &
220	ONLY: bc_dirichlet_l, &
221	bc_dirichlet_n, &
222	bc_dirichlet_r, &
223	bc_dirichlet_s, &
224	bc_radiation_l, &
225	bc_radiation_n, &
226	bc_radiation_r, &
227	bc_radiation_s, &
228	child_domain, &
229	constant_diffusion, dt_3d, e_init, humidity, &
230	initializing_actions, intermediate_timestep_count, &
231	intermediate_timestep_count_max, km_constant, &
232	les_dynamic, les_mw, ocean_mode, plant_canopy, prandtl_number, &
233	pt_reference, rans_mode, rans_tke_e, rans_tke_l, &
234	timestep_scheme, turbulence_closure, &
235	turbulent_inflow, use_upstream_for_tke, vpt_reference, &
236	ws_scheme_sca, current_timestep_number
237
238	USE advec_ws, &
239	ONLY: advec_s_ws
240
241	USE advec_s_bc_mod, &
242	ONLY: advec_s_bc
243
244	USE advec_s_pw_mod, &
245	ONLY: advec_s_pw
246
247	USE advec_s_up_mod, &
248	ONLY: advec_s_up
249
250	USE cpulog, &
251	ONLY: cpu_log, log_point_s
252
253	USE indices, &
254	ONLY: advc_flags_s, &
255	nbgp, nxl, nxlg, nxr, nxrg, nyn, nyng, nys, nysg, nzb, nzt, &
256	wall_flags_0
257
258	USE kinds
259
260	USE ocean_mod, &
261	ONLY: prho_reference
262
263	USE pegrid
264
265	USE plant_canopy_model_mod, &
266	ONLY: pcm_tendency
267
268	USE statistics, &
269	ONLY: hom, hom_sum, statistic_regions
270
271	USE surface_mod, &
272	ONLY: bc_h, &
273	bc_v, &
274	get_topography_top_index_ji, &
275	surf_def_h, &
276	surf_def_v, &
277	surf_lsm_h, &
278	surf_lsm_v, &
279	surf_usm_h, &
280	surf_usm_v
281
282	IMPLICIT NONE
283
284
285	REAL(wp) :: c_0 !< constant used for diffusion coefficient and dissipation (dependent on mode RANS/LES)
286	REAL(wp) :: c_1 !< model constant for RANS mode
287	REAL(wp) :: c_2 !< model constant for RANS mode
288	REAL(wp) :: c_3 !< model constant for RANS mode
289	REAL(wp) :: c_4 !< model constant for RANS mode
290	REAL(wp) :: l_max !< maximum length scale for Blackadar mixing length
291	REAL(wp) :: dsig_e = 1.0_wp !< factor to calculate Ke from Km (1/sigma_e)
292	REAL(wp) :: dsig_diss = 1.0_wp !< factor to calculate K_diss from Km (1/sigma_diss)
293
294	REAL(wp), DIMENSION(0:4) :: rans_const_c = & !< model constants for RANS mode (namelist param)
295	(/ 0.55_wp, 1.44_wp, 1.92_wp, 1.44_wp, 0.0_wp /) !> default values fit for standard-tke-e closure
296
297	REAL(wp), DIMENSION(2) :: rans_const_sigma = & !< model constants for RANS mode, sigma values (sigma_e, sigma_diss) (namelist param)
298	(/ 1.0_wp, 1.30_wp /)
299
300	REAL(wp), DIMENSION(:), ALLOCATABLE :: l_black !< mixing length according to Blackadar
301
302	REAL(wp), DIMENSION(:), ALLOCATABLE :: l_grid !< geometric mean of grid sizes dx, dy, dz
303
304	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE :: l_wall !< near-wall mixing length
305
306	!
307	!-- Public variables
308	PUBLIC c_0, rans_const_c, rans_const_sigma
309
310	SAVE
311
312	PRIVATE
313	!
314	!-- Public subroutines
315	PUBLIC &
316	tcm_boundary_conds, &
317	tcm_check_parameters, &
318	tcm_check_data_output, &
319	tcm_define_netcdf_grid, &
320	tcm_init_arrays, &
321	tcm_init, &
322	tcm_actions, &
323	tcm_prognostic_equations, &
324	tcm_swap_timelevel, &
325	tcm_3d_data_averaging, &
326	tcm_data_output_2d, &
327	tcm_data_output_3d, &
328	tcm_diffusivities
329
330	!
331	!-- PALM interfaces:
332	!-- Boundary conditions for subgrid TKE and dissipation
333	INTERFACE tcm_boundary_conds
334	MODULE PROCEDURE tcm_boundary_conds
335	END INTERFACE tcm_boundary_conds
336	!
337	!-- Input parameter checks to be done in check_parameters
338	INTERFACE tcm_check_parameters
339	MODULE PROCEDURE tcm_check_parameters
340	END INTERFACE tcm_check_parameters
341
342	!
343	!-- Data output checks for 2D/3D data to be done in check_parameters
344	INTERFACE tcm_check_data_output
345	MODULE PROCEDURE tcm_check_data_output
346	END INTERFACE tcm_check_data_output
347
348	!
349	!-- Definition of data output quantities
350	INTERFACE tcm_define_netcdf_grid
351	MODULE PROCEDURE tcm_define_netcdf_grid
352	END INTERFACE tcm_define_netcdf_grid
353
354	!
355	!-- Initialization of arrays
356	INTERFACE tcm_init_arrays
357	MODULE PROCEDURE tcm_init_arrays
358	END INTERFACE tcm_init_arrays
359
360	!
361	!-- Initialization actions
362	INTERFACE tcm_init
363	MODULE PROCEDURE tcm_init
364	END INTERFACE tcm_init
365
366	!
367	!-- Location specific actions
368	INTERFACE tcm_actions
369	MODULE PROCEDURE tcm_actions
370	MODULE PROCEDURE tcm_actions_ij
371	END INTERFACE tcm_actions
372
373	!
374	!-- Prognostic equations for TKE and TKE dissipation rate
375	INTERFACE tcm_prognostic_equations
376	MODULE PROCEDURE tcm_prognostic_equations
377	MODULE PROCEDURE tcm_prognostic_equations_ij
378	END INTERFACE tcm_prognostic_equations
379
380	!
381	!-- Swapping of time levels (required for prognostic variables)
382	INTERFACE tcm_swap_timelevel
383	MODULE PROCEDURE tcm_swap_timelevel
384	END INTERFACE tcm_swap_timelevel
385
386	!
387	!-- Averaging of 3D data for output
388	INTERFACE tcm_3d_data_averaging
389	MODULE PROCEDURE tcm_3d_data_averaging
390	END INTERFACE tcm_3d_data_averaging
391
392	!
393	!-- Data output of 2D quantities
394	INTERFACE tcm_data_output_2d
395	MODULE PROCEDURE tcm_data_output_2d
396	END INTERFACE tcm_data_output_2d
397
398	!
399	!-- Data output of 3D data
400	INTERFACE tcm_data_output_3d
401	MODULE PROCEDURE tcm_data_output_3d
402	END INTERFACE tcm_data_output_3d
403
404	!
405	!-- Call tcm_diffusivities_default and tcm_diffusivities_dynamic
406	INTERFACE tcm_diffusivities
407	MODULE PROCEDURE tcm_diffusivities
408	END INTERFACE tcm_diffusivities
409
410
411	CONTAINS
412
413	!------------------------------------------------------------------------------!
414	! Description:
415	! ------------
416	!> Check parameters routine for turbulence closure module.
417	!------------------------------------------------------------------------------!
418	SUBROUTINE tcm_boundary_conds
419
420	USE pmc_interface, &
421	ONLY : rans_mode_parent
422
423	IMPLICIT NONE
424
425	INTEGER(iwp) :: i !< grid index x direction
426	INTEGER(iwp) :: j !< grid index y direction
427	INTEGER(iwp) :: k !< grid index z direction
428	INTEGER(iwp) :: l !< running index boundary type, for up- and downward-facing walls
429	INTEGER(iwp) :: m !< running index surface elements
430	!
431	!-- Boundary conditions for TKE.
432	IF ( .NOT. constant_diffusion ) THEN
433	!
434	!-- In LES mode, Neumann conditions with de/x_i=0 are assumed at solid walls.
435	!-- Note, only TKE is prognostic in this case and dissipation is only
436	!-- a diagnostic quantity.
437	IF ( .NOT. rans_mode ) THEN
438	!
439	!-- Horizontal walls, upward- and downward-facing
440	DO l = 0, 1
441	!$OMP PARALLEL DO PRIVATE( i, j, k )
442	!$ACC PARALLEL LOOP PRIVATE(i, j, k) &
443	!$ACC PRESENT(bc_h, e_p)
444	DO m = 1, bc_h(l)%ns
445	i = bc_h(l)%i(m)
446	j = bc_h(l)%j(m)
447	k = bc_h(l)%k(m)
448	e_p(k+bc_h(l)%koff,j,i) = e_p(k,j,i)
449	ENDDO
450	ENDDO
451	!
452	!-- Vertical walls
453	DO l = 0, 3
454	!
455	!-- Note concerning missing ACC directive for this loop: Even though
456	!-- the data structure bc_v is present, it may not contain any
457	!-- allocated arrays in the flat but also in a topography case,
458	!-- leading to a runtime error. Therefore, omit ACC directives
459	!-- for this loop, in contrast to the bc_h loop.
460	!$OMP PARALLEL DO PRIVATE( i, j, k )
461	DO m = 1, bc_v(l)%ns
462	i = bc_v(l)%i(m)
463	j = bc_v(l)%j(m)
464	k = bc_v(l)%k(m)
465	e_p(k,j+bc_v(l)%joff,i+bc_v(l)%ioff) = e_p(k,j,i)
466	ENDDO
467	ENDDO
468	!
469	!-- In RANS mode, wall function is used as boundary condition for TKE
470	ELSE
471	!
472	!-- Use wall function within constant-flux layer
473	!-- Note, grid points listed in bc_h are not included in any calculations in RANS mode and
474	!-- are therefore not set here.
475	!
476	!-- Upward-facing surfaces
477	!-- Default surfaces
478	DO m = 1, surf_def_h(0)%ns
479	i = surf_def_h(0)%i(m)
480	j = surf_def_h(0)%j(m)
481	k = surf_def_h(0)%k(m)
482	e_p(k,j,i) = ( surf_def_h(0)%us(m) / c_0 )**2
483	ENDDO
484	!
485	!-- Natural surfaces
486	DO m = 1, surf_lsm_h%ns
487	i = surf_lsm_h%i(m)
488	j = surf_lsm_h%j(m)
489	k = surf_lsm_h%k(m)
490	e_p(k,j,i) = ( surf_lsm_h%us(m) / c_0 )**2
491	ENDDO
492	!
493	!-- Urban surfaces
494	DO m = 1, surf_usm_h%ns
495	i = surf_usm_h%i(m)
496	j = surf_usm_h%j(m)
497	k = surf_usm_h%k(m)
498	e_p(k,j,i) = ( surf_usm_h%us(m) / c_0 )**2
499	ENDDO
500	!
501	!-- Vertical surfaces
502	DO l = 0, 3
503	!
504	!-- Default surfaces
505	DO m = 1, surf_def_v(l)%ns
506	i = surf_def_v(l)%i(m)
507	j = surf_def_v(l)%j(m)
508	k = surf_def_v(l)%k(m)
509	e_p(k,j,i) = ( surf_def_v(l)%us(m) / c_0 )**2
510	ENDDO
511	!
512	!-- Natural surfaces
513	DO m = 1, surf_lsm_v(l)%ns
514	i = surf_lsm_v(l)%i(m)
515	j = surf_lsm_v(l)%j(m)
516	k = surf_lsm_v(l)%k(m)
517	e_p(k,j,i) = ( surf_lsm_v(l)%us(m) / c_0 )**2
518	ENDDO
519	!
520	!-- Urban surfaces
521	DO m = 1, surf_usm_v(l)%ns
522	i = surf_usm_v(l)%i(m)
523	j = surf_usm_v(l)%j(m)
524	k = surf_usm_v(l)%k(m)
525	e_p(k,j,i) = ( surf_usm_v(l)%us(m) / c_0 )**2
526	ENDDO
527	ENDDO
528	ENDIF
529	!
530	!-- Set Neumann boundary condition for TKE at model top. Do this also
531	!-- in case of a nested run.
532	!$ACC KERNELS PRESENT(e_p)
533	e_p(nzt+1,:,:) = e_p(nzt,:,:)
534	!$ACC END KERNELS
535	!
536	!-- Nesting case: if parent operates in RANS mode and child in LES mode,
537	!-- no TKE is transfered. This case, set Neumann conditions at lateral and
538	!-- top child boundaries.
539	!-- If not ( both either in RANS or in LES mode ), TKE boundary condition
540	!-- is treated in the nesting.
541	If ( child_domain ) THEN
542	IF ( rans_mode_parent .AND. .NOT. rans_mode ) THEN
543
544	e_p(nzt+1,:,:) = e_p(nzt,:,:)
545	IF ( bc_dirichlet_l ) e_p(:,:,nxl-1) = e_p(:,:,nxl)
546	IF ( bc_dirichlet_r ) e_p(:,:,nxr+1) = e_p(:,:,nxr)
547	IF ( bc_dirichlet_s ) e_p(:,nys-1,:) = e_p(:,nys,:)
548	IF ( bc_dirichlet_n ) e_p(:,nyn+1,:) = e_p(:,nyn,:)
549
550	ENDIF
551	ENDIF
552	!
553	!-- At in- and outflow boundaries also set Neumann boundary conditions
554	!-- for the SGS-TKE. An exception is made for the child domain if
555	!-- both parent and child operate in RANS mode. This case no
556	!-- lateral Neumann boundary conditions will be set but Dirichlet
557	!-- conditions will be set in the nesting.
558	IF ( .NOT. child_domain .AND. .NOT. rans_mode_parent .AND. &
559	.NOT. rans_mode ) THEN
560	IF ( bc_dirichlet_s .OR. bc_radiation_s ) THEN
561	e_p(:,nys-1,:) = e_p(:,nys,:)
562	IF ( rans_tke_e ) diss_p(:,nys-1,:) = diss_p(:,nys,:)
563	ENDIF
564	IF ( bc_dirichlet_n .OR. bc_radiation_n ) THEN
565	e_p(:,nyn+1,:) = e_p(:,nyn,:)
566	IF ( rans_tke_e ) diss_p(:,nyn+1,:) = diss_p(:,nyn,:)
567	ENDIF
568	IF ( bc_dirichlet_l .OR. bc_radiation_l ) THEN
569	e_p(:,:,nxl-1) = e_p(:,:,nxl)
570	IF ( rans_tke_e ) diss_p(:,nyn+1,:) = diss_p(:,nyn,:)
571	ENDIF
572	IF ( bc_dirichlet_r .OR. bc_radiation_r ) THEN
573	e_p(:,:,nxr+1) = e_p(:,:,nxr)
574	IF ( rans_tke_e ) diss_p(:,nyn+1,:) = diss_p(:,nyn,:)
575	ENDIF
576	ENDIF
577	ENDIF
578
579	!
580	!-- Boundary conditions for TKE dissipation rate in RANS mode.
581	IF ( rans_tke_e ) THEN
582	!
583	!-- Use wall function within constant-flux layer
584	!-- Upward-facing surfaces
585	!-- Default surfaces
586	DO m = 1, surf_def_h(0)%ns
587	i = surf_def_h(0)%i(m)
588	j = surf_def_h(0)%j(m)
589	k = surf_def_h(0)%k(m)
590	diss_p(k,j,i) = surf_def_h(0)%us(m)**3 &
591	/ ( kappa * surf_def_h(0)%z_mo(m) )
592	ENDDO
593	!
594	!-- Natural surfaces
595	DO m = 1, surf_lsm_h%ns
596	i = surf_lsm_h%i(m)
597	j = surf_lsm_h%j(m)
598	k = surf_lsm_h%k(m)
599	diss_p(k,j,i) = surf_lsm_h%us(m)**3 &
600	/ ( kappa * surf_lsm_h%z_mo(m) )
601	ENDDO
602	!
603	!-- Urban surfaces
604	DO m = 1, surf_usm_h%ns
605	i = surf_usm_h%i(m)
606	j = surf_usm_h%j(m)
607	k = surf_usm_h%k(m)
608	diss_p(k,j,i) = surf_usm_h%us(m)**3 &
609	/ ( kappa * surf_usm_h%z_mo(m) )
610	ENDDO
611	!
612	!-- Vertical surfaces
613	DO l = 0, 3
614	!
615	!-- Default surfaces
616	DO m = 1, surf_def_v(l)%ns
617	i = surf_def_v(l)%i(m)
618	j = surf_def_v(l)%j(m)
619	k = surf_def_v(l)%k(m)
620	diss_p(k,j,i) = surf_def_v(l)%us(m)**3 &
621	/ ( kappa * surf_def_v(l)%z_mo(m) )
622	ENDDO
623	!
624	!-- Natural surfaces
625	DO m = 1, surf_lsm_v(l)%ns
626	i = surf_lsm_v(l)%i(m)
627	j = surf_lsm_v(l)%j(m)
628	k = surf_lsm_v(l)%k(m)
629	diss_p(k,j,i) = surf_lsm_v(l)%us(m)**3 &
630	/ ( kappa * surf_lsm_v(l)%z_mo(m) )
631	ENDDO
632	!
633	!-- Urban surfaces
634	DO m = 1, surf_usm_v(l)%ns
635	i = surf_usm_v(l)%i(m)
636	j = surf_usm_v(l)%j(m)
637	k = surf_usm_v(l)%k(m)
638	diss_p(k,j,i) = surf_usm_v(l)%us(m)**3 &
639	/ ( kappa * surf_usm_v(l)%z_mo(m) )
640	ENDDO
641	ENDDO
642	!
643	!-- Limit change of diss to be between -90% and +100%. Also, set an absolute
644	!-- minimum value
645	DO i = nxl, nxr
646	DO j = nys, nyn
647	DO k = nzb, nzt+1
648	diss_p(k,j,i) = MAX( MIN( diss_p(k,j,i), &
649	2.0_wp * diss(k,j,i) ), &
650	0.1_wp * diss(k,j,i), &
651	0.0001_wp )
652	ENDDO
653	ENDDO
654	ENDDO
655
656
657	diss_p(nzt+1,:,:) = diss_p(nzt,:,:)
658
659	ENDIF
660	END SUBROUTINE tcm_boundary_conds
661
662	!------------------------------------------------------------------------------!
663	! Description:
664	! ------------
665	!> Check parameters routine for turbulence closure module.
666	!------------------------------------------------------------------------------!
667	SUBROUTINE tcm_check_parameters
668
669	USE control_parameters, &
670	ONLY: message_string, turbulent_inflow, turbulent_outflow
671
672	IMPLICIT NONE
673
674	!
675	!-- Define which turbulence closure is going to be used
676	SELECT CASE ( TRIM( turbulence_closure ) )
677
678	CASE ( 'dynamic' )
679	les_dynamic = .TRUE.
680
681	CASE ( 'Moeng_Wyngaard' )
682	les_mw = .TRUE.
683
684	CASE ( 'TKE-l' )
685	rans_tke_l = .TRUE.
686	rans_mode = .TRUE.
687
688	CASE ( 'TKE-e' )
689	rans_tke_e = .TRUE.
690	rans_mode = .TRUE.
691
692	CASE DEFAULT
693	message_string = 'Unknown turbulence closure: ' // &
694	TRIM( turbulence_closure )
695	CALL message( 'tcm_check_parameters', 'PA0500', 1, 2, 0, 6, 0 )
696
697	END SELECT
698	!
699	!-- Set variables for RANS mode or LES mode
700	IF ( rans_mode ) THEN
701	!
702	!-- Assign values to constants for RANS mode
703	dsig_e = 1.0_wp / rans_const_sigma(1)
704	dsig_diss = 1.0_wp / rans_const_sigma(2)
705
706	c_0 = rans_const_c(0)
707	c_1 = rans_const_c(1)
708	c_2 = rans_const_c(2)
709	c_3 = rans_const_c(3) !> @todo clarify how to switch between different models
710	c_4 = rans_const_c(4)
711
712	IF ( turbulent_inflow .OR. turbulent_outflow ) THEN
713	message_string = 'turbulent inflow/outflow is not yet '// &
714	'implemented for RANS mode'
715	CALL message( 'tcm_check_parameters', 'PA0501', 1, 2, 0, 6, 0 )
716	ENDIF
717
718	message_string = 'RANS mode is still in development! ' // &
719	'&Not all features of PALM are yet compatible '// &
720	'with RANS mode. &Use at own risk!'
721	CALL message( 'tcm_check_parameters', 'PA0502', 0, 1, 0, 6, 0 )
722
723	ELSE
724	!
725	!-- LES mode
726	c_0 = 0.1_wp !according to Lilly (1967) and Deardorff (1980)
727
728	dsig_e = 1.0_wp !assure to use K_m to calculate TKE instead
729	!of K_e which is used in RANS mode
730
731	ENDIF
732
733	END SUBROUTINE tcm_check_parameters
734
735	!------------------------------------------------------------------------------!
736	! Description:
737	! ------------
738	!> Check data output.
739	!------------------------------------------------------------------------------!
740	SUBROUTINE tcm_check_data_output( var, unit )
741
742	IMPLICIT NONE
743
744	CHARACTER (LEN=*) :: unit !< unit of output variable
745	CHARACTER (LEN=*) :: var !< name of output variable
746
747
748	SELECT CASE ( TRIM( var ) )
749
750	CASE ( 'diss' )
751	unit = 'm2/s3'
752
753	CASE ( 'kh', 'km' )
754	unit = 'm2/s'
755
756	CASE DEFAULT
757	unit = 'illegal'
758
759	END SELECT
760
761	END SUBROUTINE tcm_check_data_output
762
763
764	!------------------------------------------------------------------------------!
765	! Description:
766	! ------------
767	!> Define appropriate grid for netcdf variables.
768	!> It is called out from subroutine netcdf.
769	!------------------------------------------------------------------------------!
770	SUBROUTINE tcm_define_netcdf_grid( var, found, grid_x, grid_y, grid_z )
771
772	IMPLICIT NONE
773
774	CHARACTER (LEN=*), INTENT(OUT) :: grid_x !< x grid of output variable
775	CHARACTER (LEN=*), INTENT(OUT) :: grid_y !< y grid of output variable
776	CHARACTER (LEN=*), INTENT(OUT) :: grid_z !< z grid of output variable
777	CHARACTER (LEN=*), INTENT(IN) :: var !< name of output variable
778
779	LOGICAL, INTENT(OUT) :: found !< flag if output variable is found
780
781	found = .TRUE.
782
783	!
784	!-- Check for the grid
785	SELECT CASE ( TRIM( var ) )
786
787	CASE ( 'diss', 'diss_xy', 'diss_xz', 'diss_yz' )
788	grid_x = 'x'
789	grid_y = 'y'
790	grid_z = 'zu'
791
792	CASE ( 'kh', 'kh_xy', 'kh_xz', 'kh_yz' )
793	grid_x = 'x'
794	grid_y = 'y'
795	grid_z = 'zu'
796
797	CASE ( 'km', 'km_xy', 'km_xz', 'km_yz' )
798	grid_x = 'x'
799	grid_y = 'y'
800	grid_z = 'zu'
801
802	CASE DEFAULT
803	found = .FALSE.
804	grid_x = 'none'
805	grid_y = 'none'
806	grid_z = 'none'
807
808	END SELECT
809
810	END SUBROUTINE tcm_define_netcdf_grid
811
812
813	!------------------------------------------------------------------------------!
814	! Description:
815	! ------------
816	!> Average 3D data.
817	!------------------------------------------------------------------------------!
818	SUBROUTINE tcm_3d_data_averaging( mode, variable )
819
820
821	USE averaging, &
822	ONLY: diss_av, kh_av, km_av
823
824	USE control_parameters, &
825	ONLY: average_count_3d
826
827	IMPLICIT NONE
828
829	CHARACTER (LEN=*) :: mode !< flag defining mode 'allocate', 'sum' or 'average'
830	CHARACTER (LEN=*) :: variable !< name of variable
831
832	INTEGER(iwp) :: i !< loop index
833	INTEGER(iwp) :: j !< loop index
834	INTEGER(iwp) :: k !< loop index
835
836	IF ( mode == 'allocate' ) THEN
837
838	SELECT CASE ( TRIM( variable ) )
839
840	CASE ( 'diss' )
841	IF ( .NOT. ALLOCATED( diss_av ) ) THEN
842	ALLOCATE( diss_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
843	ENDIF
844	diss_av = 0.0_wp
845
846	CASE ( 'kh' )
847	IF ( .NOT. ALLOCATED( kh_av ) ) THEN
848	ALLOCATE( kh_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
849	ENDIF
850	kh_av = 0.0_wp
851
852	CASE ( 'km' )
853	IF ( .NOT. ALLOCATED( km_av ) ) THEN
854	ALLOCATE( km_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
855	ENDIF
856	km_av = 0.0_wp
857
858	CASE DEFAULT
859	CONTINUE
860
861	END SELECT
862
863	ELSEIF ( mode == 'sum' ) THEN
864
865	SELECT CASE ( TRIM( variable ) )
866
867	CASE ( 'diss' )
868	IF ( ALLOCATED( diss_av ) ) THEN
869	DO i = nxlg, nxrg
870	DO j = nysg, nyng
871	DO k = nzb, nzt+1
872	diss_av(k,j,i) = diss_av(k,j,i) + diss(k,j,i)
873	ENDDO
874	ENDDO
875	ENDDO
876	ENDIF
877
878	CASE ( 'kh' )
879	IF ( ALLOCATED( kh_av ) ) THEN
880	DO i = nxlg, nxrg
881	DO j = nysg, nyng
882	DO k = nzb, nzt+1
883	kh_av(k,j,i) = kh_av(k,j,i) + kh(k,j,i)
884	ENDDO
885	ENDDO
886	ENDDO
887	ENDIF
888
889	CASE ( 'km' )
890	IF ( ALLOCATED( km_av ) ) THEN
891	DO i = nxlg, nxrg
892	DO j = nysg, nyng
893	DO k = nzb, nzt+1
894	km_av(k,j,i) = km_av(k,j,i) + km(k,j,i)
895	ENDDO
896	ENDDO
897	ENDDO
898	ENDIF
899
900	CASE DEFAULT
901	CONTINUE
902
903	END SELECT
904
905	ELSEIF ( mode == 'average' ) THEN
906
907	SELECT CASE ( TRIM( variable ) )
908
909	CASE ( 'diss' )
910	IF ( ALLOCATED( diss_av ) ) THEN
911	DO i = nxlg, nxrg
912	DO j = nysg, nyng
913	DO k = nzb, nzt+1
914	diss_av(k,j,i) = diss_av(k,j,i) &
915	/ REAL( average_count_3d, KIND=wp )
916	ENDDO
917	ENDDO
918	ENDDO
919	ENDIF
920
921	CASE ( 'kh' )
922	IF ( ALLOCATED( kh_av ) ) THEN
923	DO i = nxlg, nxrg
924	DO j = nysg, nyng
925	DO k = nzb, nzt+1
926	kh_av(k,j,i) = kh_av(k,j,i) &
927	/ REAL( average_count_3d, KIND=wp )
928	ENDDO
929	ENDDO
930	ENDDO
931	ENDIF
932
933	CASE ( 'km' )
934	IF ( ALLOCATED( km_av ) ) THEN
935	DO i = nxlg, nxrg
936	DO j = nysg, nyng
937	DO k = nzb, nzt+1
938	km_av(k,j,i) = km_av(k,j,i) &
939	/ REAL( average_count_3d, KIND=wp )
940	ENDDO
941	ENDDO
942	ENDDO
943	ENDIF
944
945	END SELECT
946
947	ENDIF
948
949	END SUBROUTINE tcm_3d_data_averaging
950
951
952	!------------------------------------------------------------------------------!
953	! Description:
954	! ------------
955	!> Define 2D output variables.
956	!------------------------------------------------------------------------------!
957	SUBROUTINE tcm_data_output_2d( av, variable, found, grid, mode, local_pf, &
958	nzb_do, nzt_do )
959
960	USE averaging, &
961	ONLY: diss_av, kh_av, km_av
962
963	IMPLICIT NONE
964
965	CHARACTER (LEN=*) :: grid !< name of vertical grid
966	CHARACTER (LEN=*) :: mode !< either 'xy', 'xz' or 'yz'
967	CHARACTER (LEN=*) :: variable !< name of variable
968
969	INTEGER(iwp) :: av !< flag for (non-)average output
970	INTEGER(iwp) :: flag_nr !< number of masking flag
971	INTEGER(iwp) :: i !< loop index
972	INTEGER(iwp) :: j !< loop index
973	INTEGER(iwp) :: k !< loop index
974	INTEGER(iwp) :: nzb_do !< vertical output index (bottom)
975	INTEGER(iwp) :: nzt_do !< vertical output index (top)
976
977	LOGICAL :: found !< flag if output variable is found
978	LOGICAL :: resorted !< flag if output is already resorted
979
980	REAL(wp) :: fill_value = -9999.0_wp !< value for the _FillValue attribute
981
982	REAL(wp), DIMENSION(nxl:nxr,nys:nyn,nzb_do:nzt_do) :: local_pf !< local
983	!< array to which output data is resorted to
984
985	REAL(wp), DIMENSION(:,:,:), POINTER :: to_be_resorted !< points to selected output variable
986
987	found = .TRUE.
988	resorted = .FALSE.
989	!
990	!-- Set masking flag for topography for not resorted arrays
991	flag_nr = 0
992
993	SELECT CASE ( TRIM( variable ) )
994
995	CASE ( 'diss_xy', 'diss_xz', 'diss_yz' )
996	IF ( av == 0 ) THEN
997	to_be_resorted => diss
998	ELSE
999	IF ( .NOT. ALLOCATED( diss_av ) ) THEN
1000	ALLOCATE( diss_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1001	diss_av = REAL( fill_value, KIND = wp )
1002	ENDIF
1003	to_be_resorted => diss_av
1004	ENDIF
1005	IF ( mode == 'xy' ) grid = 'zu'
1006
1007	CASE ( 'kh_xy', 'kh_xz', 'kh_yz' )
1008	IF ( av == 0 ) THEN
1009	to_be_resorted => kh
1010	ELSE
1011	IF ( .NOT. ALLOCATED( kh_av ) ) THEN
1012	ALLOCATE( kh_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1013	kh_av = REAL( fill_value, KIND = wp )
1014	ENDIF
1015	to_be_resorted => kh_av
1016	ENDIF
1017	IF ( mode == 'xy' ) grid = 'zu'
1018
1019	CASE ( 'km_xy', 'km_xz', 'km_yz' )
1020	IF ( av == 0 ) THEN
1021	to_be_resorted => km
1022	ELSE
1023	IF ( .NOT. ALLOCATED( km_av ) ) THEN
1024	ALLOCATE( km_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1025	km_av = REAL( fill_value, KIND = wp )
1026	ENDIF
1027	to_be_resorted => km_av
1028	ENDIF
1029	IF ( mode == 'xy' ) grid = 'zu'
1030
1031	CASE DEFAULT
1032	found = .FALSE.
1033	grid = 'none'
1034
1035	END SELECT
1036
1037	IF ( found .AND. .NOT. resorted ) THEN
1038	DO i = nxl, nxr
1039	DO j = nys, nyn
1040	DO k = nzb_do, nzt_do
1041	local_pf(i,j,k) = MERGE( to_be_resorted(k,j,i), &
1042	REAL( fill_value, KIND = wp ), &
1043	BTEST( wall_flags_0(k,j,i), flag_nr ) )
1044	ENDDO
1045	ENDDO
1046	ENDDO
1047	ENDIF
1048
1049	END SUBROUTINE tcm_data_output_2d
1050
1051
1052	!------------------------------------------------------------------------------!
1053	! Description:
1054	! ------------
1055	!> Define 3D output variables.
1056	!------------------------------------------------------------------------------!
1057	SUBROUTINE tcm_data_output_3d( av, variable, found, local_pf, nzb_do, nzt_do )
1058
1059
1060	USE averaging, &
1061	ONLY: diss_av, kh_av, km_av
1062
1063	IMPLICIT NONE
1064
1065	CHARACTER (LEN=*) :: variable !< name of variable
1066
1067	INTEGER(iwp) :: av !< flag for (non-)average output
1068	INTEGER(iwp) :: flag_nr !< number of masking flag
1069	INTEGER(iwp) :: i !< loop index
1070	INTEGER(iwp) :: j !< loop index
1071	INTEGER(iwp) :: k !< loop index
1072	INTEGER(iwp) :: nzb_do !< lower limit of the data output (usually 0)
1073	INTEGER(iwp) :: nzt_do !< vertical upper limit of the data output (usually nz_do3d)
1074
1075	LOGICAL :: found !< flag if output variable is found
1076	LOGICAL :: resorted !< flag if output is already resorted
1077
1078	REAL(wp) :: fill_value = -9999.0_wp !< value for the _FillValue attribute
1079
1080	REAL(sp), DIMENSION(nxl:nxr,nys:nyn,nzb_do:nzt_do) :: local_pf !< local
1081	!< array to which output data is resorted to
1082
1083	REAL(wp), DIMENSION(:,:,:), POINTER :: to_be_resorted !< points to selected output variable
1084
1085	found = .TRUE.
1086	resorted = .FALSE.
1087	!
1088	!-- Set masking flag for topography for not resorted arrays
1089	flag_nr = 0
1090
1091	SELECT CASE ( TRIM( variable ) )
1092
1093	CASE ( 'diss' )
1094	IF ( av == 0 ) THEN
1095	to_be_resorted => diss
1096	ELSE
1097	IF ( .NOT. ALLOCATED( diss_av ) ) THEN
1098	ALLOCATE( diss_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1099	diss_av = REAL( fill_value, KIND = wp )
1100	ENDIF
1101	to_be_resorted => diss_av
1102	ENDIF
1103
1104	CASE ( 'kh' )
1105	IF ( av == 0 ) THEN
1106	to_be_resorted => kh
1107	ELSE
1108	IF ( .NOT. ALLOCATED( kh_av ) ) THEN
1109	ALLOCATE( kh_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1110	kh_av = REAL( fill_value, KIND = wp )
1111	ENDIF
1112	to_be_resorted => kh_av
1113	ENDIF
1114
1115	CASE ( 'km' )
1116	IF ( av == 0 ) THEN
1117	to_be_resorted => km
1118	ELSE
1119	IF ( .NOT. ALLOCATED( km_av ) ) THEN
1120	ALLOCATE( km_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1121	km_av = REAL( fill_value, KIND = wp )
1122	ENDIF
1123	to_be_resorted => km_av
1124	ENDIF
1125
1126	CASE DEFAULT
1127	found = .FALSE.
1128
1129	END SELECT
1130
1131
1132	IF ( found .AND. .NOT. resorted ) THEN
1133	DO i = nxl, nxr
1134	DO j = nys, nyn
1135	DO k = nzb_do, nzt_do
1136	local_pf(i,j,k) = MERGE( &
1137	to_be_resorted(k,j,i), &
1138	REAL( fill_value, KIND = wp ), &
1139	BTEST( wall_flags_0(k,j,i), flag_nr ) )
1140	ENDDO
1141	ENDDO
1142	ENDDO
1143	resorted = .TRUE.
1144	ENDIF
1145
1146	END SUBROUTINE tcm_data_output_3d
1147
1148
1149	!------------------------------------------------------------------------------!
1150	! Description:
1151	! ------------
1152	!> Allocate arrays and assign pointers.
1153	!------------------------------------------------------------------------------!
1154	SUBROUTINE tcm_init_arrays
1155
1156	USE bulk_cloud_model_mod, &
1157	ONLY: collision_turbulence
1158
1159	USE particle_attributes, &
1160	ONLY: use_sgs_for_particles, wang_kernel
1161
1162	USE pmc_interface, &
1163	ONLY: nested_run
1164
1165	IMPLICIT NONE
1166
1167	ALLOCATE( kh(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1168	ALLOCATE( km(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1169
1170	ALLOCATE( e_1(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1171	ALLOCATE( e_2(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1172	ALLOCATE( e_3(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1173
1174	!
1175	!-- Allocate arrays required for dissipation.
1176	!-- Please note, if it is a nested run, arrays need to be allocated even if
1177	!-- they do not necessarily need to be transferred, which is attributed to
1178	!-- the design of the model coupler which allocates memory for each variable.
1179	IF ( rans_mode .OR. use_sgs_for_particles .OR. wang_kernel .OR. &
1180	collision_turbulence .OR. nested_run ) THEN
1181
1182	ALLOCATE( diss_1(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1183	IF ( rans_tke_e .OR. nested_run ) THEN
1184	ALLOCATE( diss_2(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1185	ALLOCATE( diss_3(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1186	ENDIF
1187
1188	ENDIF
1189
1190	!
1191	!-- Initial assignment of pointers
1192	e => e_1; e_p => e_2; te_m => e_3
1193
1194	IF ( rans_mode .OR. use_sgs_for_particles .OR. &
1195	wang_kernel .OR. collision_turbulence .OR. nested_run ) THEN
1196	diss => diss_1
1197	IF ( rans_tke_e .OR. nested_run ) THEN
1198	diss_p => diss_2; tdiss_m => diss_3
1199	ENDIF
1200	ENDIF
1201
1202	END SUBROUTINE tcm_init_arrays
1203
1204
1205	!------------------------------------------------------------------------------!
1206	! Description:
1207	! ------------
1208	!> Initialization of turbulence closure module.
1209	!------------------------------------------------------------------------------!
1210	SUBROUTINE tcm_init
1211
1212	USE control_parameters, &
1213	ONLY: bc_dirichlet_l, complex_terrain, topography
1214
1215	USE model_1d_mod, &
1216	ONLY: e1d, kh1d, km1d
1217
1218	IMPLICIT NONE
1219
1220	INTEGER(iwp) :: i !< loop index
1221	INTEGER(iwp) :: j !< loop index
1222	INTEGER(iwp) :: k !< loop index
1223	INTEGER(iwp) :: nz_s_shift !< lower shift index for scalars
1224	INTEGER(iwp) :: nz_s_shift_l !< local lower shift index in case of turbulent inflow
1225
1226	!
1227	!-- Initialize mixing length
1228	CALL tcm_init_mixing_length
1229
1230	!
1231	!-- Actions for initial runs
1232	IF ( TRIM( initializing_actions ) /= 'read_restart_data' .AND. &
1233	TRIM( initializing_actions ) /= 'cyclic_fill' ) THEN
1234
1235	IF ( INDEX( initializing_actions, 'set_1d-model_profiles' ) /= 0 ) THEN
1236
1237	IF ( .NOT. rans_tke_e ) THEN
1238	!
1239	!-- Transfer initial profiles to the arrays of the 3D model
1240	DO i = nxlg, nxrg
1241	DO j = nysg, nyng
1242	e(:,j,i) = e1d
1243	kh(:,j,i) = kh1d
1244	km(:,j,i) = km1d
1245	ENDDO
1246	ENDDO
1247
1248	IF ( constant_diffusion ) THEN
1249	e = 0.0_wp
1250	ENDIF
1251
1252	ELSE
1253	!
1254	!-- In case of TKE-e closure in RANS mode, do not use e, diss, and km
1255	!-- profiles from 1D model. Instead, initialize with constant profiles
1256	IF ( constant_diffusion ) THEN
1257	km = km_constant
1258	kh = km / prandtl_number
1259	e = 0.0_wp
1260	ELSEIF ( e_init > 0.0_wp ) THEN
1261	DO i = nxlg, nxrg
1262	DO j = nysg, nyng
1263	DO k = nzb+1, nzt
1264	km(k,j,i) = c_0 * l_wall(k,j,i) * SQRT( e_init )
1265	ENDDO
1266	ENDDO
1267	ENDDO
1268	km(nzb,:,:) = km(nzb+1,:,:)
1269	km(nzt+1,:,:) = km(nzt,:,:)
1270	kh = km / prandtl_number
1271	e = e_init
1272	ELSE
1273	IF ( .NOT. ocean_mode ) THEN
1274	kh = 0.01_wp ! there must exist an initial diffusion, because
1275	km = 0.01_wp ! otherwise no TKE would be produced by the
1276	! production terms, as long as not yet
1277	! e = (u/cm)*2 at k=nzb+1
1278	ELSE
1279	kh = 0.00001_wp
1280	km = 0.00001_wp
1281	ENDIF
1282	e = 0.0_wp
1283	ENDIF
1284
1285	DO i = nxlg, nxrg
1286	DO j = nysg, nyng
1287	DO k = nzb+1, nzt
1288	diss(k,j,i) = c_0*4 e(k,j,i)**2 / km(k,j,i)
1289	ENDDO
1290	ENDDO
1291	ENDDO
1292	diss(nzb,:,:) = diss(nzb+1,:,:)
1293	diss(nzt+1,:,:) = diss(nzt,:,:)
1294
1295	ENDIF
1296
1297	ELSEIF ( INDEX(initializing_actions, 'set_constant_profiles') /= 0 .OR. &
1298	INDEX( initializing_actions, 'inifor' ) /= 0 ) THEN
1299
1300	IF ( constant_diffusion ) THEN
1301	km = km_constant
1302	kh = km / prandtl_number
1303	e = 0.0_wp
1304	ELSEIF ( e_init > 0.0_wp ) THEN
1305	DO i = nxlg, nxrg
1306	DO j = nysg, nyng
1307	DO k = nzb+1, nzt
1308	km(k,j,i) = c_0 * l_wall(k,j,i) * SQRT( e_init )
1309	ENDDO
1310	ENDDO
1311	ENDDO
1312	km(nzb,:,:) = km(nzb+1,:,:)
1313	km(nzt+1,:,:) = km(nzt,:,:)
1314	kh = km / prandtl_number
1315	e = e_init
1316	ELSE
1317	IF ( .NOT. ocean_mode ) THEN
1318	kh = 0.01_wp ! there must exist an initial diffusion, because
1319	km = 0.01_wp ! otherwise no TKE would be produced by the
1320	! production terms, as long as not yet
1321	! e = (u/cm)*2 at k=nzb+1
1322	ELSE
1323	kh = 0.00001_wp
1324	km = 0.00001_wp
1325	ENDIF
1326	e = 0.0_wp
1327	ENDIF
1328
1329	IF ( rans_tke_e ) THEN
1330	DO i = nxlg, nxrg
1331	DO j = nysg, nyng
1332	DO k = nzb+1, nzt
1333	diss(k,j,i) = c_0*4 e(k,j,i)**2 / km(k,j,i)
1334	ENDDO
1335	ENDDO
1336	ENDDO
1337	diss(nzb,:,:) = diss(nzb+1,:,:)
1338	diss(nzt+1,:,:) = diss(nzt,:,:)
1339	ENDIF
1340
1341	ENDIF
1342	!
1343	!-- Store initial profiles for output purposes etc.
1344	hom(:,1,23,:) = SPREAD( km(:,nys,nxl), 2, statistic_regions+1 )
1345	hom(:,1,24,:) = SPREAD( kh(:,nys,nxl), 2, statistic_regions+1 )
1346	!
1347	!-- Initialize old and new time levels.
1348	te_m = 0.0_wp
1349	e_p = e
1350	IF ( rans_tke_e ) THEN
1351	tdiss_m = 0.0_wp
1352	diss_p = diss
1353	ENDIF
1354
1355	ELSEIF ( TRIM( initializing_actions ) == 'read_restart_data' .OR. &
1356	TRIM( initializing_actions ) == 'cyclic_fill' ) &
1357	THEN
1358
1359	!
1360	!-- In case of complex terrain and cyclic fill method as initialization,
1361	!-- shift initial data in the vertical direction for each point in the
1362	!-- x-y-plane depending on local surface height
1363	IF ( complex_terrain .AND. &
1364	TRIM( initializing_actions ) == 'cyclic_fill' ) THEN
1365	DO i = nxlg, nxrg
1366	DO j = nysg, nyng
1367	nz_s_shift = get_topography_top_index_ji( j, i, 's' )
1368
1369	e(nz_s_shift:nzt+1,j,i) = e(0:nzt+1-nz_s_shift,j,i)
1370	km(nz_s_shift:nzt+1,j,i) = km(0:nzt+1-nz_s_shift,j,i)
1371	kh(nz_s_shift:nzt+1,j,i) = kh(0:nzt+1-nz_s_shift,j,i)
1372	ENDDO
1373	ENDDO
1374	IF ( rans_tke_e ) THEN
1375	DO i = nxlg, nxrg
1376	DO j = nysg, nyng
1377	nz_s_shift = get_topography_top_index_ji( j, i, 's' )
1378
1379	diss(nz_s_shift:nzt+1,j,i) = diss(0:nzt+1-nz_s_shift,j,i)
1380	ENDDO
1381	ENDDO
1382	ENDIF
1383	ENDIF
1384
1385	!
1386	!-- Initialization of the turbulence recycling method
1387	IF ( TRIM( initializing_actions ) == 'cyclic_fill' .AND. &
1388	turbulent_inflow ) THEN
1389	mean_inflow_profiles(:,5) = hom_sum(:,8,0) ! e
1390	!
1391	!-- In case of complex terrain, determine vertical displacement at inflow
1392	!-- boundary and adjust mean inflow profiles
1393	IF ( complex_terrain ) THEN
1394	IF ( nxlg <= 0 .AND. nxrg >= 0 .AND. &
1395	nysg <= 0 .AND. nyng >= 0 ) THEN
1396	nz_s_shift_l = get_topography_top_index_ji( 0, 0, 's' )
1397	ELSE
1398	nz_s_shift_l = 0
1399	ENDIF
1400	#if defined( __parallel )
1401	CALL MPI_ALLREDUCE(nz_s_shift_l, nz_s_shift, 1, MPI_INTEGER, &
1402	MPI_MAX, comm2d, ierr)
1403	#else
1404	nz_s_shift = nz_s_shift_l
1405	#endif
1406	mean_inflow_profiles(nz_s_shift:nzt+1,5) = &
1407	hom_sum(0:nzt+1-nz_s_shift,8,0) ! e
1408	ENDIF
1409	!
1410	!-- Use these mean profiles at the inflow (provided that Dirichlet
1411	!-- conditions are used)
1412	IF ( bc_dirichlet_l ) THEN
1413	DO j = nysg, nyng
1414	DO k = nzb, nzt+1
1415	e(k,j,nxlg:-1) = mean_inflow_profiles(k,5)
1416	ENDDO
1417	ENDDO
1418	ENDIF
1419	ENDIF
1420	!
1421	!-- Inside buildings set TKE back to zero
1422	IF ( TRIM( initializing_actions ) == 'cyclic_fill' .AND. &
1423	topography /= 'flat' ) THEN
1424	!
1425	!-- Inside buildings set TKE back to zero.
1426	!-- Other scalars (km, kh,...) are ignored at present,
1427	!-- maybe revise later.
1428	DO i = nxlg, nxrg
1429	DO j = nysg, nyng
1430	DO k = nzb, nzt
1431	e(k,j,i) = MERGE( e(k,j,i), 0.0_wp, &
1432	BTEST( wall_flags_0(k,j,i), 0 ) )
1433	ENDDO
1434	ENDDO
1435	ENDDO
1436
1437	IF ( rans_tke_e ) THEN
1438	DO i = nxlg, nxrg
1439	DO j = nysg, nyng
1440	DO k = nzb, nzt
1441	diss(k,j,i) = MERGE( diss(k,j,i), 0.0_wp, &
1442	BTEST( wall_flags_0(k,j,i), 0 ) )
1443	ENDDO
1444	ENDDO
1445	ENDDO
1446	ENDIF
1447	ENDIF
1448	!
1449	!-- Initialize new time levels (only done in order to set boundary values
1450	!-- including ghost points)
1451	e_p = e
1452	!
1453	!-- Allthough tendency arrays are set in prognostic_equations, they have
1454	!-- to be predefined here because there they are used (but multiplied with 0)
1455	!-- before they are set.
1456	te_m = 0.0_wp
1457
1458	IF ( rans_tke_e ) THEN
1459	diss_p = diss
1460	tdiss_m = 0.0_wp
1461	ENDIF
1462
1463	ENDIF
1464
1465	END SUBROUTINE tcm_init
1466
1467
1468	! Description:
1469	! -----------------------------------------------------------------------------!
1470	!> Pre-computation of grid-dependent and near-wall mixing length.
1471	!> @todo consider walls in horizontal direction at a distance further than a
1472	!> single grid point (RANS mode)
1473	!------------------------------------------------------------------------------!
1474	SUBROUTINE tcm_init_mixing_length
1475
1476	USE arrays_3d, &
1477	ONLY: dzw, ug, vg, zu, zw
1478
1479	USE control_parameters, &
1480	ONLY: f, message_string, wall_adjustment_factor
1481
1482	USE grid_variables, &
1483	ONLY: dx, dy
1484
1485	USE indices, &
1486	ONLY: nbgp, nx, nxl, nxlg, nxr, nxrg, ny, nyn, nyng, nys, nysg, nzb, &
1487	nzt, wall_flags_0
1488
1489	USE kinds
1490
1491
1492	IMPLICIT NONE
1493
1494	INTEGER(iwp) :: dist_dx !< found distance devided by dx
1495	INTEGER(iwp) :: i !< index variable along x
1496	INTEGER(iwp) :: ii !< index variable along x
1497	INTEGER(iwp) :: j !< index variable along y
1498	INTEGER(iwp) :: k !< index variable along z
1499	INTEGER(iwp) :: k_max_topo = 0 !< index of maximum topography height
1500	INTEGER(iwp) :: kk !< index variable along z
1501	INTEGER(iwp) :: rad_i !< search radius in grid points along x
1502	INTEGER(iwp) :: rad_i_l !< possible search radius to the left
1503	INTEGER(iwp) :: rad_i_r !< possible search radius to the right
1504	INTEGER(iwp) :: rad_j !< search radius in grid points along y
1505	INTEGER(iwp) :: rad_j_n !< possible search radius to north
1506	INTEGER(iwp) :: rad_j_s !< possible search radius to south
1507	INTEGER(iwp) :: rad_k !< search radius in grid points along z
1508	INTEGER(iwp) :: rad_k_b !< search radius in grid points along negative z
1509	INTEGER(iwp) :: rad_k_t !< search radius in grid points along positive z
1510
1511	INTEGER(KIND=1), DIMENSION(:,:), ALLOCATABLE :: vic_yz !< contains a quarter of a single yz-slice of vicinity
1512
1513	INTEGER(KIND=1), DIMENSION(:,:,:), ALLOCATABLE :: vicinity !< contains topography information of the vicinity of (i/j/k)
1514
1515	REAL(wp) :: radius !< search radius in meter
1516
1517	ALLOCATE( l_grid(1:nzt) )
1518	ALLOCATE( l_wall(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
1519	!
1520	!-- Initialize the mixing length in case of an LES-simulation
1521	IF ( .NOT. rans_mode ) THEN
1522	!
1523	!-- Compute the grid-dependent mixing length.
1524	DO k = 1, nzt
1525	l_grid(k) = ( dx * dy * dzw(k) )**0.33333333333333_wp
1526	ENDDO
1527	!
1528	!-- Initialize near-wall mixing length l_wall only in the vertical direction
1529	!-- for the moment, multiplication with wall_adjustment_factor further below
1530	l_wall(nzb,:,:) = l_grid(1)
1531	DO k = nzb+1, nzt
1532	l_wall(k,:,:) = l_grid(k)
1533	ENDDO
1534	l_wall(nzt+1,:,:) = l_grid(nzt)
1535
1536	DO k = 1, nzt
1537	IF ( l_grid(k) > 1.5_wp * dx * wall_adjustment_factor .OR. &
1538	l_grid(k) > 1.5_wp * dy * wall_adjustment_factor ) THEN
1539	WRITE( message_string, * ) 'grid anisotropy exceeds ', &
1540	'threshold given by only local', &
1541	' &horizontal reduction of near_wall ', &
1542	'mixing length l_wall', &
1543	' &starting from height level k = ', k, &
1544	'.'
1545	CALL message( 'init_grid', 'PA0202', 0, 1, 0, 6, 0 )
1546	EXIT
1547	ENDIF
1548	ENDDO
1549	!
1550	!-- In case of topography: limit near-wall mixing length l_wall further:
1551	!-- Go through all points of the subdomain one by one and look for the closest
1552	!-- surface.
1553	!-- Is this correct in the ocean case?
1554	DO i = nxl, nxr
1555	DO j = nys, nyn
1556	DO k = nzb+1, nzt
1557	!
1558	!-- Check if current gridpoint belongs to the atmosphere
1559	IF ( BTEST( wall_flags_0(k,j,i), 0 ) ) THEN
1560	!
1561	!-- Check for neighbouring grid-points.
1562	!-- Vertical distance, down
1563	IF ( .NOT. BTEST( wall_flags_0(k-1,j,i), 0 ) ) &
1564	l_wall(k,j,i) = MIN( l_grid(k), zu(k) - zw(k-1) )
1565	!
1566	!-- Vertical distance, up
1567	IF ( .NOT. BTEST( wall_flags_0(k+1,j,i), 0 ) ) &
1568	l_wall(k,j,i) = MIN( l_grid(k), zw(k) - zu(k) )
1569	!
1570	!-- y-distance
1571	IF ( .NOT. BTEST( wall_flags_0(k,j-1,i), 0 ) .OR. &
1572	.NOT. BTEST( wall_flags_0(k,j+1,i), 0 ) ) &
1573	l_wall(k,j,i) = MIN( l_wall(k,j,i), l_grid(k), 0.5_wp * dy )
1574	!
1575	!-- x-distance
1576	IF ( .NOT. BTEST( wall_flags_0(k,j,i-1), 0 ) .OR. &
1577	.NOT. BTEST( wall_flags_0(k,j,i+1), 0 ) ) &
1578	l_wall(k,j,i) = MIN( l_wall(k,j,i), l_grid(k), 0.5_wp * dx )
1579	!
1580	!-- yz-distance (vertical edges, down)
1581	IF ( .NOT. BTEST( wall_flags_0(k-1,j-1,i), 0 ) .OR. &
1582	.NOT. BTEST( wall_flags_0(k-1,j+1,i), 0 ) ) &
1583	l_wall(k,j,i) = MIN( l_wall(k,j,i), l_grid(k), &
1584	SQRT( 0.25_wp * dy**2 + &
1585	( zu(k) - zw(k-1) )**2 ) )
1586	!
1587	!-- yz-distance (vertical edges, up)
1588	IF ( .NOT. BTEST( wall_flags_0(k+1,j-1,i), 0 ) .OR. &
1589	.NOT. BTEST( wall_flags_0(k+1,j+1,i), 0 ) ) &
1590	l_wall(k,j,i) = MIN( l_wall(k,j,i), l_grid(k), &
1591	SQRT( 0.25_wp * dy**2 + &
1592	( zw(k) - zu(k) )**2 ) )
1593	!
1594	!-- xz-distance (vertical edges, down)
1595	IF ( .NOT. BTEST( wall_flags_0(k-1,j,i-1), 0 ) .OR. &
1596	.NOT. BTEST( wall_flags_0(k-1,j,i+1), 0 ) ) &
1597	l_wall(k,j,i) = MIN( l_wall(k,j,i), l_grid(k), &
1598	SQRT( 0.25_wp * dx**2 + &
1599	( zu(k) - zw(k-1) )**2 ) )
1600	!
1601	!-- xz-distance (vertical edges, up)
1602	IF ( .NOT. BTEST( wall_flags_0(k+1,j,i-1), 0 ) .OR. &
1603	.NOT. BTEST( wall_flags_0(k+1,j,i+1), 0 ) ) &
1604	l_wall(k,j,i) = MIN( l_wall(k,j,i), l_grid(k), &
1605	SQRT( 0.25_wp * dx**2 + &
1606	( zw(k) - zu(k) )**2 ) )
1607	!
1608	!-- xy-distance (horizontal edges)
1609	IF ( .NOT. BTEST( wall_flags_0(k,j-1,i-1), 0 ) .OR. &
1610	.NOT. BTEST( wall_flags_0(k,j+1,i-1), 0 ) .OR. &
1611	.NOT. BTEST( wall_flags_0(k,j-1,i+1), 0 ) .OR. &
1612	.NOT. BTEST( wall_flags_0(k,j+1,i+1), 0 ) ) &
1613	l_wall(k,j,i) = MIN( l_wall(k,j,i), l_grid(k), &
1614	SQRT( 0.25_wp * ( dx2 + dy2 ) ) )
1615	!
1616	!-- xyz distance (vertical and horizontal edges, down)
1617	IF ( .NOT. BTEST( wall_flags_0(k-1,j-1,i-1), 0 ) .OR. &
1618	.NOT. BTEST( wall_flags_0(k-1,j+1,i-1), 0 ) .OR. &
1619	.NOT. BTEST( wall_flags_0(k-1,j-1,i+1), 0 ) .OR. &
1620	.NOT. BTEST( wall_flags_0(k-1,j+1,i+1), 0 ) ) &
1621	l_wall(k,j,i) = MIN( l_wall(k,j,i), l_grid(k), &
1622	SQRT( 0.25_wp * ( dx2 + dy2 ) &
1623	+ ( zu(k) - zw(k-1) )**2 ) )
1624	!
1625	!-- xyz distance (vertical and horizontal edges, up)
1626	IF ( .NOT. BTEST( wall_flags_0(k+1,j-1,i-1), 0 ) .OR. &
1627	.NOT. BTEST( wall_flags_0(k+1,j+1,i-1), 0 ) .OR. &
1628	.NOT. BTEST( wall_flags_0(k+1,j-1,i+1), 0 ) .OR. &
1629	.NOT. BTEST( wall_flags_0(k+1,j+1,i+1), 0 ) ) &
1630	l_wall(k,j,i) = MIN( l_wall(k,j,i), l_grid(k), &
1631	SQRT( 0.25_wp * ( dx2 + dy2 ) &
1632	+ ( zw(k) - zu(k) )**2 ) )
1633
1634	ENDIF
1635	ENDDO
1636	ENDDO
1637	ENDDO
1638
1639	ELSE
1640	!
1641	!-- Initialize the mixing length in case of a RANS simulation
1642	ALLOCATE( l_black(nzb:nzt+1) )
1643
1644	!
1645	!-- Calculate mixing length according to Blackadar (1962)
1646	IF ( f /= 0.0_wp ) THEN
1647	l_max = 2.7E-4_wp * SQRT( ug(nzt+1)2 + vg(nzt+1)2 ) / &
1648	ABS( f ) + 1.0E-10_wp
1649	ELSE
1650	l_max = 30.0_wp
1651	ENDIF
1652
1653	DO k = nzb, nzt
1654	l_black(k) = kappa * zu(k) / ( 1.0_wp + kappa * zu(k) / l_max )
1655	ENDDO
1656
1657	l_black(nzt+1) = l_black(nzt)
1658
1659	!
1660	!-- Get height level of highest topography within local subdomain
1661	DO i = nxlg, nxrg
1662	DO j = nysg, nyng
1663	DO k = nzb+1, nzt-1
1664	IF ( .NOT. BTEST( wall_flags_0(k,j,i), 0 ) .AND. &
1665	k > k_max_topo ) &
1666	k_max_topo = k
1667	ENDDO
1668	ENDDO
1669	ENDDO
1670
1671	l_wall(nzb,:,:) = l_black(nzb)
1672	l_wall(nzt+1,:,:) = l_black(nzt+1)
1673	!
1674	!-- Limit mixing length to either nearest wall or Blackadar mixing length.
1675	!-- For that, analyze each grid point (i/j/k) ("analysed grid point") and
1676	!-- search within its vicinity for the shortest distance to a wall by cal-
1677	!-- culating the distance between the analysed grid point and the "viewed
1678	!-- grid point" if it contains a wall (belongs to topography).
1679	DO k = nzb+1, nzt
1680
1681	radius = l_black(k) ! radius within walls are searched
1682	!
1683	!-- Set l_wall to its default maximum value (l_back)
1684	l_wall(k,:,:) = radius
1685
1686	!
1687	!-- Compute search radius as number of grid points in all directions
1688	rad_i = CEILING( radius / dx )
1689	rad_j = CEILING( radius / dy )
1690
1691	DO kk = 0, nzt-k
1692	rad_k_t = kk
1693	!
1694	!-- Limit upward search radius to height of maximum topography
1695	IF ( zu(k+kk)-zu(k) >= radius .OR. k+kk >= k_max_topo ) EXIT
1696	ENDDO
1697
1698	DO kk = 0, k
1699	rad_k_b = kk
1700	IF ( zu(k)-zu(k-kk) >= radius ) EXIT
1701	ENDDO
1702
1703	!
1704	!-- Get maximum vertical radius; necessary for defining arrays
1705	rad_k = MAX( rad_k_b, rad_k_t )
1706	!
1707	!-- When analysed grid point lies above maximum topography, set search
1708	!-- radius to 0 if the distance between the analysed grid point and max
1709	!-- topography height is larger than the maximum search radius
1710	IF ( zu(k-rad_k_b) > zu(k_max_topo) ) rad_k_b = 0
1711	!
1712	!-- Search within vicinity only if the vertical search radius is >0
1713	IF ( rad_k_b /= 0 .OR. rad_k_t /= 0 ) THEN
1714
1715	!> @note shape of vicinity is larger in z direction
1716	!> Shape of vicinity is two grid points larger than actual search
1717	!> radius in vertical direction. The first and last grid point is
1718	!> always set to 1 to asure correct detection of topography. See
1719	!> function "shortest_distance" for details.
1720	!> 2018-03-16, gronemeier
1721	ALLOCATE( vicinity(-rad_k-1:rad_k+1,-rad_j:rad_j,-rad_i:rad_i) )
1722	ALLOCATE( vic_yz(0:rad_k+1,0:rad_j) )
1723
1724	vicinity = 1
1725
1726	DO i = nxl, nxr
1727	DO j = nys, nyn
1728	!
1729	!-- Start search only if (i/j/k) belongs to atmosphere
1730	IF ( BTEST( wall_flags_0(k,j,i), 0 ) ) THEN
1731	!
1732	!-- Reset topography within vicinity
1733	vicinity(-rad_k:rad_k,:,:) = 0
1734	!
1735	!-- Copy area surrounding analysed grid point into vicinity.
1736	!-- First, limit size of data copied to vicinity by the domain
1737	!-- border
1738	!> @note limit copied area to 1 grid point in hor. dir.
1739	!> Ignore walls in horizontal direction which are
1740	!> further away than a single grid point. This allows to
1741	!> only search within local subdomain without the need
1742	!> of global topography information.
1743	!> The error made by this assumption are acceptable at
1744	!> the moment.
1745	!> 2018-10-01, gronemeier
1746	rad_i_l = MIN( 1, rad_i, i )
1747	rad_i_r = MIN( 1, rad_i, nx-i )
1748
1749	rad_j_s = MIN( 1, rad_j, j )
1750	rad_j_n = MIN( 1, rad_j, ny-j )
1751
1752	CALL copy_into_vicinity( k, j, i, &
1753	-rad_k_b, rad_k_t, &
1754	-rad_j_s, rad_j_n, &
1755	-rad_i_l, rad_i_r )
1756	!> @note in case of cyclic boundaries, those parts of the
1757	!> topography which lies beyond the domain borders but
1758	!> still within the search radius still needs to be
1759	!> copied into 'vicinity'. As the effective search
1760	!> radius is limited to 1 at the moment, no further
1761	!> copying is needed. Old implementation (prior to
1762	!> 2018-10-01) had this covered but used a global array.
1763	!> 2018-10-01, gronemeier
1764
1765	!
1766	!-- Search for walls only if there is any within vicinity
1767	IF ( MAXVAL( vicinity(-rad_k:rad_k,:,:) ) /= 0 ) THEN
1768	!
1769	!-- Search within first half (positive x)
1770	dist_dx = rad_i
1771	DO ii = 0, dist_dx
1772	!
1773	!-- Search along vertical direction only if below
1774	!-- maximum topography
1775	IF ( rad_k_t > 0 ) THEN
1776	!
1777	!-- Search for walls within octant (+++)
1778	vic_yz = vicinity(0:rad_k+1,0:rad_j,ii)
1779	l_wall(k,j,i) = MIN( l_wall(k,j,i), &
1780	shortest_distance( vic_yz, .TRUE., ii ) )
1781	!
1782	!-- Search for walls within octant (+-+)
1783	!-- Switch order of array so that the analysed grid
1784	!-- point is always located at (0/0) (required by
1785	!-- shortest_distance").
1786	vic_yz = vicinity(0:rad_k+1,0:-rad_j:-1,ii)
1787	l_wall(k,j,i) = MIN( l_wall(k,j,i), &
1788	shortest_distance( vic_yz, .TRUE., ii ) )
1789
1790	ENDIF
1791	!
1792	!-- Search for walls within octant (+--)
1793	vic_yz = vicinity(0:-rad_k-1:-1,0:-rad_j:-1,ii)
1794	l_wall(k,j,i) = MIN( l_wall(k,j,i), &
1795	shortest_distance( vic_yz, .FALSE., ii ) )
1796	!
1797	!-- Search for walls within octant (++-)
1798	vic_yz = vicinity(0:-rad_k-1:-1,0:rad_j,ii)
1799	l_wall(k,j,i) = MIN( l_wall(k,j,i), &
1800	shortest_distance( vic_yz, .FALSE., ii ) )
1801	!
1802	!-- Reduce search along x by already found distance
1803	dist_dx = CEILING( l_wall(k,j,i) / dx )
1804
1805	ENDDO
1806	!
1807	!- Search within second half (negative x)
1808	DO ii = 0, -dist_dx, -1
1809	!
1810	!-- Search along vertical direction only if below
1811	!-- maximum topography
1812	IF ( rad_k_t > 0 ) THEN
1813	!
1814	!-- Search for walls within octant (-++)
1815	vic_yz = vicinity(0:rad_k+1,0:rad_j,ii)
1816	l_wall(k,j,i) = MIN( l_wall(k,j,i), &
1817	shortest_distance( vic_yz, .TRUE., -ii ) )
1818	!
1819	!-- Search for walls within octant (--+)
1820	!-- Switch order of array so that the analysed grid
1821	!-- point is always located at (0/0) (required by
1822	!-- shortest_distance").
1823	vic_yz = vicinity(0:rad_k+1,0:-rad_j:-1,ii)
1824	l_wall(k,j,i) = MIN( l_wall(k,j,i), &
1825	shortest_distance( vic_yz, .TRUE., -ii ) )
1826
1827	ENDIF
1828	!
1829	!-- Search for walls within octant (---)
1830	vic_yz = vicinity(0:-rad_k-1:-1,0:-rad_j:-1,ii)
1831	l_wall(k,j,i) = MIN( l_wall(k,j,i), &
1832	shortest_distance( vic_yz, .FALSE., -ii ) )
1833	!
1834	!-- Search for walls within octant (-+-)
1835	vic_yz = vicinity(0:-rad_k-1:-1,0:rad_j,ii)
1836	l_wall(k,j,i) = MIN( l_wall(k,j,i), &
1837	shortest_distance( vic_yz, .FALSE., -ii ) )
1838	!
1839	!-- Reduce search along x by already found distance
1840	dist_dx = CEILING( l_wall(k,j,i) / dx )
1841
1842	ENDDO
1843
1844	ENDIF !Check for any walls within vicinity
1845
1846	ELSE !Check if (i,j,k) belongs to atmosphere
1847
1848	l_wall(k,j,i) = l_black(k)
1849
1850	ENDIF
1851
1852	ENDDO !j loop
1853	ENDDO !i loop
1854
1855	DEALLOCATE( vicinity )
1856	DEALLOCATE( vic_yz )
1857
1858	ENDIF !check vertical size of vicinity
1859
1860	ENDDO !k loop
1861
1862	!$ACC ENTER DATA COPYIN(l_black(nzb:nzt+1))
1863
1864	ENDIF !LES or RANS mode
1865
1866	!
1867	!-- Set lateral boundary conditions for l_wall
1868	CALL exchange_horiz( l_wall, nbgp )
1869
1870	!$ACC ENTER DATA COPYIN(l_grid(nzb:nzt+1)) &
1871	!$ACC COPYIN(l_wall(nzb:nzt+1,nysg:nyng,nxlg:nxrg))
1872
1873	CONTAINS
1874	!------------------------------------------------------------------------------!
1875	! Description:
1876	! ------------
1877	!> Calculate the shortest distance between position (i/j/k)=(0/0/0) and
1878	!> (pos_i/jj/kk), where (jj/kk) is the position of the maximum of 'array'
1879	!> closest to the origin (0/0) of 'array'.
1880	!------------------------------------------------------------------------------!
1881	REAL(wp) FUNCTION shortest_distance( array, orientation, pos_i )
1882
1883	IMPLICIT NONE
1884
1885	LOGICAL, INTENT(IN) :: orientation !< flag if array represents an array oriented upwards (true) or downwards (false)
1886
1887	INTEGER(iwp), INTENT(IN) :: pos_i !< x position of the yz-plane 'array'
1888
1889	INTEGER(iwp) :: a !< loop index
1890	INTEGER(iwp) :: b !< loop index
1891	INTEGER(iwp) :: jj !< loop index
1892
1893	INTEGER(KIND=1) :: maximum !< maximum of array along z dimension
1894
1895	INTEGER(iwp), DIMENSION(0:rad_j) :: loc_k !< location of closest wall along vertical dimension
1896
1897	INTEGER(KIND=1), DIMENSION(0:rad_k+1,0:rad_j), INTENT(IN) :: array !< array containing a yz-plane at position pos_i
1898
1899	!
1900	!-- Get coordinate of first maximum along vertical dimension
1901	!-- at each y position of array (similar to function maxloc but more stable).
1902	DO a = 0, rad_j
1903	loc_k(a) = rad_k+1
1904	maximum = MAXVAL( array(:,a) )
1905	DO b = 0, rad_k+1
1906	IF ( array(b,a) == maximum ) THEN
1907	loc_k(a) = b
1908	EXIT
1909	ENDIF
1910	ENDDO
1911	ENDDO
1912	!
1913	!-- Set distance to the default maximum value (=search radius)
1914	shortest_distance = radius
1915	!
1916	!-- Calculate distance between position (0/0/0) and
1917	!-- position (pos_i/jj/loc(jj)) and only save the shortest distance.
1918	IF ( orientation ) THEN !if array is oriented upwards
1919	DO jj = 0, rad_j
1920	shortest_distance = &
1921	MIN( shortest_distance, &
1922	SQRT( MAX(REAL(pos_i, KIND=wp)dx-0.5_wpdx, 0.0_wp)**2 &
1923	+ MAX(REAL(jj, KIND=wp)dy-0.5_wpdy, 0.0_wp)**2 &
1924	+ MAX(zw(loc_k(jj)+k-1)-zu(k), 0.0_wp)**2 &
1925	) &
1926	)
1927	ENDDO
1928	ELSE !if array is oriented downwards
1929	!> @note MAX within zw required to circumvent error at domain border
1930	!> At the domain border, if non-cyclic boundary is present, the
1931	!> index for zw could be -1, which will be errorneous (zw(-1) does
1932	!> not exist). The MAX function limits the index to be at least 0.
1933	DO jj = 0, rad_j
1934	shortest_distance = &
1935	MIN( shortest_distance, &
1936	SQRT( MAX(REAL(pos_i, KIND=wp)dx-0.5_wpdx, 0.0_wp)**2 &
1937	+ MAX(REAL(jj, KIND=wp)dy-0.5_wpdy, 0.0_wp)**2 &
1938	+ MAX(zu(k)-zw(MAX(k-loc_k(jj),0_iwp)), 0.0_wp)**2 &
1939	) &
1940	)
1941	ENDDO
1942	ENDIF
1943
1944	END FUNCTION
1945
1946	!------------------------------------------------------------------------------!
1947	! Description:
1948	! ------------
1949	!> Copy a subarray of size (kb:kt,js:jn,il:ir) centered around grid point
1950	!> (kp,jp,ip) containing the first bit of wall_flags_0 into the array
1951	!> 'vicinity'. Only copy first bit as this indicates the presence of topography.
1952	!------------------------------------------------------------------------------!
1953	SUBROUTINE copy_into_vicinity( kp, jp, ip, kb, kt, js, jn, il, ir )
1954
1955	IMPLICIT NONE
1956
1957	INTEGER(iwp), INTENT(IN) :: il !< left loop boundary
1958	INTEGER(iwp), INTENT(IN) :: ip !< center position in x-direction
1959	INTEGER(iwp), INTENT(IN) :: ir !< right loop boundary
1960	INTEGER(iwp), INTENT(IN) :: jn !< northern loop boundary
1961	INTEGER(iwp), INTENT(IN) :: jp !< center position in y-direction
1962	INTEGER(iwp), INTENT(IN) :: js !< southern loop boundary
1963	INTEGER(iwp), INTENT(IN) :: kb !< bottom loop boundary
1964	INTEGER(iwp), INTENT(IN) :: kp !< center position in z-direction
1965	INTEGER(iwp), INTENT(IN) :: kt !< top loop boundary
1966
1967	INTEGER(iwp) :: i !< loop index
1968	INTEGER(iwp) :: j !< loop index
1969	INTEGER(iwp) :: k !< loop index
1970
1971	DO i = il, ir
1972	DO j = js, jn
1973	DO k = kb, kt
1974	vicinity(k,j,i) = MERGE( 0, 1, &
1975	BTEST( wall_flags_0(kp+k,jp+j,ip+i), 0 ) )
1976	ENDDO
1977	ENDDO
1978	ENDDO
1979
1980	END SUBROUTINE copy_into_vicinity
1981
1982	END SUBROUTINE tcm_init_mixing_length
1983
1984
1985	!------------------------------------------------------------------------------!
1986	! Description:
1987	! ------------
1988	!> Initialize virtual velocities used later in production_e.
1989	!------------------------------------------------------------------------------!
1990	SUBROUTINE production_e_init
1991
1992	USE arrays_3d, &
1993	ONLY: drho_air_zw, zu
1994
1995	USE control_parameters, &
1996	ONLY: constant_flux_layer
1997
1998	USE surface_layer_fluxes_mod, &
1999	ONLY: phi_m
2000
2001	IMPLICIT NONE
2002
2003	INTEGER(iwp) :: i !< grid index x-direction
2004	INTEGER(iwp) :: j !< grid index y-direction
2005	INTEGER(iwp) :: k !< grid index z-direction
2006	INTEGER(iwp) :: m !< running index surface elements
2007
2008	REAL(wp) :: km_sfc !< diffusion coefficient, used to compute virtual velocities
2009
2010	IF ( constant_flux_layer ) THEN
2011	!
2012	!-- Calculate a virtual velocity at the surface in a way that the
2013	!-- vertical velocity gradient at k = 1 (u(k+1)-u_0) matches the
2014	!-- Prandtl law (-w'u'/km). This gradient is used in the TKE shear
2015	!-- production term at k=1 (see production_e_ij).
2016	!-- The velocity gradient has to be limited in case of too small km
2017	!-- (otherwise the timestep may be significantly reduced by large
2018	!-- surface winds).
2019	!-- not available in case of non-cyclic boundary conditions.
2020	!-- Default surfaces, upward-facing
2021	!$OMP PARALLEL DO PRIVATE(i,j,k,m)
2022	!$ACC PARALLEL LOOP PRIVATE(i, j, k, m, km_sfc) &
2023	!$ACC PRESENT(surf_def_h(0), u, v, drho_air_zw, zu)
2024	DO m = 1, surf_def_h(0)%ns
2025
2026	i = surf_def_h(0)%i(m)
2027	j = surf_def_h(0)%j(m)
2028	k = surf_def_h(0)%k(m)
2029	!
2030	!-- Note, calculation of u_0 and v_0 is not fully accurate, as u/v
2031	!-- and km are not on the same grid. Actually, a further
2032	!-- interpolation of km onto the u/v-grid is necessary. However, the
2033	!-- effect of this error is negligible.
2034	km_sfc = kappa * surf_def_h(0)%us(m) * surf_def_h(0)%z_mo(m) / &
2035	phi_m( surf_def_h(0)%z_mo(m) / surf_def_h(0)%ol(m) )
2036
2037	surf_def_h(0)%u_0(m) = u(k+1,j,i) + surf_def_h(0)%usws(m) * &
2038	drho_air_zw(k-1) * &
2039	( zu(k+1) - zu(k-1) ) / &
2040	( km_sfc + 1.0E-20_wp )
2041	surf_def_h(0)%v_0(m) = v(k+1,j,i) + surf_def_h(0)%vsws(m) * &
2042	drho_air_zw(k-1) * &
2043	( zu(k+1) - zu(k-1) ) / &
2044	( km_sfc + 1.0E-20_wp )
2045
2046	IF ( ABS( u(k+1,j,i) - surf_def_h(0)%u_0(m) ) > &
2047	ABS( u(k+1,j,i) - u(k-1,j,i) ) &
2048	) surf_def_h(0)%u_0(m) = u(k-1,j,i)
2049
2050	IF ( ABS( v(k+1,j,i) - surf_def_h(0)%v_0(m) ) > &
2051	ABS( v(k+1,j,i) - v(k-1,j,i) ) &
2052	) surf_def_h(0)%v_0(m) = v(k-1,j,i)
2053
2054	ENDDO
2055	!
2056	!-- Default surfaces, downward-facing surfaces
2057	!$OMP PARALLEL DO PRIVATE(i,j,k,m)
2058	!$ACC PARALLEL LOOP PRIVATE(i, j, k, m, km_sfc) &
2059	!$ACC PRESENT(surf_def_h(1), u, v, drho_air_zw, zu, km)
2060	DO m = 1, surf_def_h(1)%ns
2061
2062	i = surf_def_h(1)%i(m)
2063	j = surf_def_h(1)%j(m)
2064	k = surf_def_h(1)%k(m)
2065	!
2066	!-- Note, calculation of u_0 and v_0 is not fully accurate, as u/v
2067	!-- and km are not on the same grid. Actually, a further
2068	!-- interpolation of km onto the u/v-grid is necessary. However, the
2069	!-- effect of this error is negligible.
2070	surf_def_h(1)%u_0(m) = u(k-1,j,i) - surf_def_h(1)%usws(m) * &
2071	drho_air_zw(k-1) * &
2072	( zu(k+1) - zu(k-1) ) / &
2073	( km(k,j,i) + 1.0E-20_wp )
2074	surf_def_h(1)%v_0(m) = v(k-1,j,i) - surf_def_h(1)%vsws(m) * &
2075	drho_air_zw(k-1) * &
2076	( zu(k+1) - zu(k-1) ) / &
2077	( km(k,j,i) + 1.0E-20_wp )
2078
2079	IF ( ABS( surf_def_h(1)%u_0(m) - u(k-1,j,i) ) > &
2080	ABS( u(k+1,j,i) - u(k-1,j,i) ) &
2081	) surf_def_h(1)%u_0(m) = u(k+1,j,i)
2082
2083	IF ( ABS( surf_def_h(1)%v_0(m) - v(k-1,j,i) ) > &
2084	ABS( v(k+1,j,i) - v(k-1,j,i) ) &
2085	) surf_def_h(1)%v_0(m) = v(k+1,j,i)
2086
2087	ENDDO
2088	!
2089	!-- Natural surfaces, upward-facing
2090	!$OMP PARALLEL DO PRIVATE(i,j,k,m)
2091	!$ACC PARALLEL LOOP PRIVATE(i, j, k, m, km_sfc) &
2092	!$ACC PRESENT(surf_lsm_h, u, v, drho_air_zw, zu)
2093	DO m = 1, surf_lsm_h%ns
2094
2095	i = surf_lsm_h%i(m)
2096	j = surf_lsm_h%j(m)
2097	k = surf_lsm_h%k(m)
2098	!
2099	!-- Note, calculation of u_0 and v_0 is not fully accurate, as u/v
2100	!-- and km are not on the same grid. Actually, a further
2101	!-- interpolation of km onto the u/v-grid is necessary. However, the
2102	!-- effect of this error is negligible.
2103	km_sfc = kappa * surf_lsm_h%us(m) * surf_lsm_h%z_mo(m) / &
2104	phi_m( surf_lsm_h%z_mo(m) / surf_lsm_h%ol(m) )
2105
2106	surf_lsm_h%u_0(m) = u(k+1,j,i) + surf_lsm_h%usws(m) * &
2107	drho_air_zw(k-1) * &
2108	( zu(k+1) - zu(k-1) ) / &
2109	( km_sfc + 1.0E-20_wp )
2110	surf_lsm_h%v_0(m) = v(k+1,j,i) + surf_lsm_h%vsws(m) * &
2111	drho_air_zw(k-1) * &
2112	( zu(k+1) - zu(k-1) ) / &
2113	( km_sfc + 1.0E-20_wp )
2114
2115	IF ( ABS( u(k+1,j,i) - surf_lsm_h%u_0(m) ) > &
2116	ABS( u(k+1,j,i) - u(k-1,j,i) ) &
2117	) surf_lsm_h%u_0(m) = u(k-1,j,i)
2118
2119	IF ( ABS( v(k+1,j,i) - surf_lsm_h%v_0(m) ) > &
2120	ABS( v(k+1,j,i) - v(k-1,j,i) ) &
2121	) surf_lsm_h%v_0(m) = v(k-1,j,i)
2122
2123	ENDDO
2124	!
2125	!-- Urban surfaces, upward-facing
2126	!$OMP PARALLEL DO PRIVATE(i,j,k,m)
2127	!$ACC PARALLEL LOOP PRIVATE(i, j, k, m, km_sfc) &
2128	!$ACC PRESENT(surf_usm_h, u, v, drho_air_zw, zu)
2129	DO m = 1, surf_usm_h%ns
2130
2131	i = surf_usm_h%i(m)
2132	j = surf_usm_h%j(m)
2133	k = surf_usm_h%k(m)
2134	!
2135	!-- Note, calculation of u_0 and v_0 is not fully accurate, as u/v
2136	!-- and km are not on the same grid. Actually, a further
2137	!-- interpolation of km onto the u/v-grid is necessary. However, the
2138	!-- effect of this error is negligible.
2139	km_sfc = kappa * surf_usm_h%us(m) * surf_usm_h%z_mo(m) / &
2140	phi_m( surf_usm_h%z_mo(m) / surf_usm_h%ol(m) )
2141
2142	surf_usm_h%u_0(m) = u(k+1,j,i) + surf_usm_h%usws(m) * &
2143	drho_air_zw(k-1) * &
2144	( zu(k+1) - zu(k-1) ) / &
2145	( km_sfc + 1.0E-20_wp )
2146	surf_usm_h%v_0(m) = v(k+1,j,i) + surf_usm_h%vsws(m) * &
2147	drho_air_zw(k-1) * &
2148	( zu(k+1) - zu(k-1) ) / &
2149	( km_sfc + 1.0E-20_wp )
2150
2151	IF ( ABS( u(k+1,j,i) - surf_usm_h%u_0(m) ) > &
2152	ABS( u(k+1,j,i) - u(k-1,j,i) ) &
2153	) surf_usm_h%u_0(m) = u(k-1,j,i)
2154
2155	IF ( ABS( v(k+1,j,i) - surf_usm_h%v_0(m) ) > &
2156	ABS( v(k+1,j,i) - v(k-1,j,i) ) &
2157	) surf_usm_h%v_0(m) = v(k-1,j,i)
2158
2159	ENDDO
2160
2161	ENDIF
2162
2163	END SUBROUTINE production_e_init
2164
2165
2166	!--------------------------------------------------------------------------------------------------!
2167	! Description:
2168	! ------------
2169	!> Execute module-specific actions for all grid points
2170	!--------------------------------------------------------------------------------------------------!
2171	SUBROUTINE tcm_actions( location )
2172
2173
2174	CHARACTER (LEN=*) :: location !<
2175
2176	! INTEGER(iwp) :: i !<
2177	! INTEGER(iwp) :: j !<
2178	! INTEGER(iwp) :: k !<
2179
2180	!
2181	!-- Here the module-specific actions follow
2182	!-- No calls for single grid points are allowed at locations before and
2183	!-- after the timestep, since these calls are not within an i,j-loop
2184	SELECT CASE ( location )
2185
2186	CASE ( 'before_timestep' )
2187
2188
2189	CASE ( 'before_prognostic_equations' )
2190
2191	IF ( .NOT. constant_diffusion ) CALL production_e_init
2192
2193
2194	CASE ( 'after_integration' )
2195
2196
2197	CASE ( 'after_timestep' )
2198
2199
2200	CASE ( 'u-tendency' )
2201
2202
2203	CASE ( 'v-tendency' )
2204
2205
2206	CASE ( 'w-tendency' )
2207
2208
2209	CASE ( 'pt-tendency' )
2210
2211
2212	CASE ( 'sa-tendency' )
2213
2214
2215	CASE ( 'e-tendency' )
2216
2217
2218	CASE ( 'q-tendency' )
2219
2220
2221	CASE ( 's-tendency' )
2222
2223
2224	CASE DEFAULT
2225	CONTINUE
2226
2227	END SELECT
2228
2229	END SUBROUTINE tcm_actions
2230
2231
2232	!--------------------------------------------------------------------------------------------------!
2233	! Description:
2234	! ------------
2235	!> Execute module-specific actions for grid point i,j
2236	!--------------------------------------------------------------------------------------------------!
2237	SUBROUTINE tcm_actions_ij( i, j, location )
2238
2239
2240	CHARACTER (LEN=*) :: location
2241
2242	INTEGER(iwp) :: i
2243	INTEGER(iwp) :: j
2244
2245	!
2246	!-- Here the module-specific actions follow
2247	SELECT CASE ( location )
2248
2249	CASE ( 'u-tendency' )
2250
2251	!-- Next line is to avoid compiler warning about unused variables. Please remove.
2252	IF ( i + j < 0 ) CONTINUE
2253
2254	CASE ( 'v-tendency' )
2255
2256
2257	CASE ( 'w-tendency' )
2258
2259
2260	CASE ( 'pt-tendency' )
2261
2262
2263	CASE ( 'sa-tendency' )
2264
2265
2266	CASE ( 'e-tendency' )
2267
2268
2269	CASE ( 'q-tendency' )
2270
2271
2272	CASE ( 's-tendency' )
2273
2274
2275	CASE DEFAULT
2276	CONTINUE
2277
2278	END SELECT
2279
2280	END SUBROUTINE tcm_actions_ij
2281
2282
2283	!------------------------------------------------------------------------------!
2284	! Description:
2285	! ------------
2286	!> Prognostic equation for subgrid-scale TKE and TKE dissipation rate.
2287	!> Vector-optimized version
2288	!------------------------------------------------------------------------------!
2289	SUBROUTINE tcm_prognostic_equations
2290
2291	USE control_parameters, &
2292	ONLY: scalar_advec, tsc
2293
2294	IMPLICIT NONE
2295
2296	INTEGER(iwp) :: i !< loop index
2297	INTEGER(iwp) :: j !< loop index
2298	INTEGER(iwp) :: k !< loop index
2299
2300	REAL(wp) :: sbt !< wheighting factor for sub-time step
2301
2302	!
2303	!-- If required, compute prognostic equation for turbulent kinetic
2304	!-- energy (TKE)
2305	IF ( .NOT. constant_diffusion ) THEN
2306
2307	CALL cpu_log( log_point_s(67), 'tke-equation', 'start' )
2308
2309	sbt = tsc(2)
2310	IF ( .NOT. use_upstream_for_tke ) THEN
2311	IF ( scalar_advec == 'bc-scheme' ) THEN
2312
2313	IF ( timestep_scheme(1:5) /= 'runge' ) THEN
2314	!
2315	!-- Bott-Chlond scheme always uses Euler time step. Thus:
2316	sbt = 1.0_wp
2317	ENDIF
2318	tend = 0.0_wp
2319	CALL advec_s_bc( e, 'e' )
2320
2321	ENDIF
2322	ENDIF
2323
2324	!
2325	!-- TKE-tendency terms with no communication
2326	IF ( scalar_advec /= 'bc-scheme' .OR. use_upstream_for_tke ) THEN
2327	IF ( use_upstream_for_tke ) THEN
2328	tend = 0.0_wp
2329	CALL advec_s_up( e )
2330	ELSE
2331	!$ACC KERNELS PRESENT(tend)
2332	tend = 0.0_wp
2333	!$ACC END KERNELS
2334	IF ( timestep_scheme(1:5) == 'runge' ) THEN
2335	IF ( ws_scheme_sca ) THEN
2336	CALL advec_s_ws( advc_flags_s, e, 'e', &
2337	bc_dirichlet_l .OR. bc_radiation_l, &
2338	bc_dirichlet_n .OR. bc_radiation_n, &
2339	bc_dirichlet_r .OR. bc_radiation_r, &
2340	bc_dirichlet_s .OR. bc_radiation_s )
2341	ELSE
2342	CALL advec_s_pw( e )
2343	ENDIF
2344	ELSE
2345	CALL advec_s_up( e )
2346	ENDIF
2347	ENDIF
2348	ENDIF
2349
2350	CALL production_e( .FALSE. )
2351
2352	IF ( .NOT. humidity ) THEN
2353	IF ( ocean_mode ) THEN
2354	CALL diffusion_e( prho, prho_reference )
2355	ELSE
2356	CALL diffusion_e( pt, pt_reference )
2357	ENDIF
2358	ELSE
2359	CALL diffusion_e( vpt, pt_reference )
2360	ENDIF
2361
2362	!
2363	!-- Additional sink term for flows through plant canopies
2364	IF ( plant_canopy ) CALL pcm_tendency( 6 )
2365
2366	! CALL user_actions( 'e-tendency' ) ToDo: find general solution for circular dependency between modules
2367
2368	!
2369	!-- Prognostic equation for TKE.
2370	!-- Eliminate negative TKE values, which can occur due to numerical
2371	!-- reasons in the course of the integration. In such cases the old TKE
2372	!-- value is reduced by 90%.
2373	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
2374	!$ACC PRESENT(e, tend, te_m, wall_flags_0) &
2375	!$ACC PRESENT(tsc(3:3)) &
2376	!$ACC PRESENT(e_p)
2377	DO i = nxl, nxr
2378	DO j = nys, nyn
2379	DO k = nzb+1, nzt
2380	e_p(k,j,i) = e(k,j,i) + ( dt_3d * ( sbt * tend(k,j,i) + &
2381	tsc(3) * te_m(k,j,i) ) &
2382	) &
2383	* MERGE( 1.0_wp, 0.0_wp, &
2384	BTEST( wall_flags_0(k,j,i), 0 ) &
2385	)
2386	IF ( e_p(k,j,i) < 0.0_wp ) e_p(k,j,i) = 0.1_wp * e(k,j,i)
2387	ENDDO
2388	ENDDO
2389	ENDDO
2390
2391	!
2392	!-- Calculate tendencies for the next Runge-Kutta step
2393	IF ( timestep_scheme(1:5) == 'runge' ) THEN
2394	IF ( intermediate_timestep_count == 1 ) THEN
2395	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
2396	!$ACC PRESENT(tend, te_m)
2397	DO i = nxl, nxr
2398	DO j = nys, nyn
2399	DO k = nzb+1, nzt
2400	te_m(k,j,i) = tend(k,j,i)
2401	ENDDO
2402	ENDDO
2403	ENDDO
2404	ELSEIF ( intermediate_timestep_count < &
2405	intermediate_timestep_count_max ) THEN
2406	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
2407	!$ACC PRESENT(tend, te_m)
2408	DO i = nxl, nxr
2409	DO j = nys, nyn
2410	DO k = nzb+1, nzt
2411	te_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
2412	+ 5.3125_wp * te_m(k,j,i)
2413	ENDDO
2414	ENDDO
2415	ENDDO
2416	ENDIF
2417	ENDIF
2418
2419	CALL cpu_log( log_point_s(67), 'tke-equation', 'stop' )
2420
2421	ENDIF ! TKE equation
2422
2423	!
2424	!-- If required, compute prognostic equation for TKE dissipation rate
2425	IF ( rans_tke_e ) THEN
2426
2427	CALL cpu_log( log_point_s(64), 'diss-equation', 'start' )
2428
2429	sbt = tsc(2)
2430	IF ( .NOT. use_upstream_for_tke ) THEN
2431	IF ( scalar_advec == 'bc-scheme' ) THEN
2432
2433	IF ( timestep_scheme(1:5) /= 'runge' ) THEN
2434	!
2435	!-- Bott-Chlond scheme always uses Euler time step. Thus:
2436	sbt = 1.0_wp
2437	ENDIF
2438	tend = 0.0_wp
2439	CALL advec_s_bc( diss, 'diss' )
2440
2441	ENDIF
2442	ENDIF
2443
2444	!
2445	!-- dissipation-tendency terms with no communication
2446	IF ( scalar_advec /= 'bc-scheme' .OR. use_upstream_for_tke ) THEN
2447	IF ( use_upstream_for_tke ) THEN
2448	tend = 0.0_wp
2449	CALL advec_s_up( diss )
2450	ELSE
2451	tend = 0.0_wp
2452	IF ( timestep_scheme(1:5) == 'runge' ) THEN
2453	IF ( ws_scheme_sca ) THEN
2454	CALL advec_s_ws( advc_flags_s, diss, 'diss', &
2455	bc_dirichlet_l .OR. bc_radiation_l, &
2456	bc_dirichlet_n .OR. bc_radiation_n, &
2457	bc_dirichlet_r .OR. bc_radiation_r, &
2458	bc_dirichlet_s .OR. bc_radiation_s )
2459	ELSE
2460	CALL advec_s_pw( diss )
2461	ENDIF
2462	ELSE
2463	CALL advec_s_up( diss )
2464	ENDIF
2465	ENDIF
2466	ENDIF
2467	!
2468	!-- Production of TKE dissipation rate
2469	CALL production_e( .TRUE. )
2470	!
2471	!-- Diffusion term of TKE dissipation rate
2472	CALL diffusion_diss
2473	!
2474	!-- Additional sink term for flows through plant canopies
2475	! IF ( plant_canopy ) CALL pcm_tendency( ? ) !> @todo not yet implemented
2476
2477	! CALL user_actions( 'e-tendency' ) ToDo: find general solution for circular dependency between modules
2478
2479	!
2480	!-- Prognostic equation for TKE dissipation.
2481	!-- Eliminate negative dissipation values, which can occur due to numerical
2482	!-- reasons in the course of the integration. In such cases the old
2483	!-- dissipation value is reduced by 90%.
2484	DO i = nxl, nxr
2485	DO j = nys, nyn
2486	DO k = nzb+1, nzt
2487	diss_p(k,j,i) = diss(k,j,i) + ( dt_3d * ( sbt * tend(k,j,i) + &
2488	tsc(3) * tdiss_m(k,j,i) ) &
2489	) &
2490	* MERGE( 1.0_wp, 0.0_wp, &
2491	BTEST( wall_flags_0(k,j,i), 0 ) &
2492	)
2493	IF ( diss_p(k,j,i) < 0.0_wp ) &
2494	diss_p(k,j,i) = 0.1_wp * diss(k,j,i)
2495	ENDDO
2496	ENDDO
2497	ENDDO
2498
2499	!
2500	!-- Calculate tendencies for the next Runge-Kutta step
2501	IF ( timestep_scheme(1:5) == 'runge' ) THEN
2502	IF ( intermediate_timestep_count == 1 ) THEN
2503	DO i = nxl, nxr
2504	DO j = nys, nyn
2505	DO k = nzb+1, nzt
2506	tdiss_m(k,j,i) = tend(k,j,i)
2507	ENDDO
2508	ENDDO
2509	ENDDO
2510	ELSEIF ( intermediate_timestep_count < &
2511	intermediate_timestep_count_max ) THEN
2512	DO i = nxl, nxr
2513	DO j = nys, nyn
2514	DO k = nzb+1, nzt
2515	tdiss_m(k,j,i) = -9.5625_wp * tend(k,j,i) &
2516	+ 5.3125_wp * tdiss_m(k,j,i)
2517	ENDDO
2518	ENDDO
2519	ENDDO
2520	ENDIF
2521	ENDIF
2522
2523	CALL cpu_log( log_point_s(64), 'diss-equation', 'stop' )
2524
2525	ENDIF
2526
2527	END SUBROUTINE tcm_prognostic_equations
2528
2529
2530	!------------------------------------------------------------------------------!
2531	! Description:
2532	! ------------
2533	!> Prognostic equation for subgrid-scale TKE and TKE dissipation rate.
2534	!> Cache-optimized version
2535	!------------------------------------------------------------------------------!
2536	SUBROUTINE tcm_prognostic_equations_ij( i, j, i_omp, tn )
2537
2538	USE arrays_3d, &
2539	ONLY: diss_l_diss, diss_l_e, diss_s_diss, diss_s_e, flux_l_diss, &
2540	flux_l_e, flux_s_diss, flux_s_e
2541
2542	USE control_parameters, &
2543	ONLY: tsc
2544
2545	IMPLICIT NONE
2546
2547	INTEGER(iwp) :: i !< loop index x direction
2548	INTEGER(iwp) :: i_omp !< first loop index of i-loop in prognostic_equations
2549	INTEGER(iwp) :: j !< loop index y direction
2550	INTEGER(iwp) :: k !< loop index z direction
2551	INTEGER(iwp) :: tn !< task number of openmp task
2552
2553	!
2554	!-- If required, compute prognostic equation for turbulent kinetic
2555	!-- energy (TKE)
2556	IF ( .NOT. constant_diffusion ) THEN
2557
2558	!
2559	!-- Tendency-terms for TKE
2560	tend(:,j,i) = 0.0_wp
2561	IF ( timestep_scheme(1:5) == 'runge' &
2562	.AND. .NOT. use_upstream_for_tke ) THEN
2563	IF ( ws_scheme_sca ) THEN
2564	CALL advec_s_ws( advc_flags_s, &
2565	i, j, e, 'e', flux_s_e, diss_s_e, &
2566	flux_l_e, diss_l_e , i_omp, tn, &
2567	bc_dirichlet_l .OR. bc_radiation_l, &
2568	bc_dirichlet_n .OR. bc_radiation_n, &
2569	bc_dirichlet_r .OR. bc_radiation_r, &
2570	bc_dirichlet_s .OR. bc_radiation_s )
2571	ELSE
2572	CALL advec_s_pw( i, j, e )
2573	ENDIF
2574	ELSE
2575	CALL advec_s_up( i, j, e )
2576	ENDIF
2577
2578	CALL production_e_ij( i, j, .FALSE. )
2579
2580	IF ( .NOT. humidity ) THEN
2581	IF ( ocean_mode ) THEN
2582	CALL diffusion_e_ij( i, j, prho, prho_reference )
2583	ELSE
2584	CALL diffusion_e_ij( i, j, pt, pt_reference )
2585	ENDIF
2586	ELSE
2587	CALL diffusion_e_ij( i, j, vpt, pt_reference )
2588	ENDIF
2589
2590	!
2591	!-- Additional sink term for flows through plant canopies
2592	IF ( plant_canopy ) CALL pcm_tendency( i, j, 6 )
2593
2594	! CALL user_actions( i, j, 'e-tendency' ) ToDo: find general solution for circular dependency between modules
2595
2596	!
2597	!-- Prognostic equation for TKE.
2598	!-- Eliminate negative TKE values, which can occur due to numerical
2599	!-- reasons in the course of the integration. In such cases the old
2600	!-- TKE value is reduced by 90%.
2601	DO k = nzb+1, nzt
2602	e_p(k,j,i) = e(k,j,i) + ( dt_3d * ( tsc(2) * tend(k,j,i) + &
2603	tsc(3) * te_m(k,j,i) ) &
2604	) &
2605	* MERGE( 1.0_wp, 0.0_wp, &
2606	BTEST( wall_flags_0(k,j,i), 0 ) &
2607	)
2608	IF ( e_p(k,j,i) <= 0.0_wp ) e_p(k,j,i) = 0.1_wp * e(k,j,i)
2609	ENDDO
2610
2611	!
2612	!-- Calculate tendencies for the next Runge-Kutta step
2613	IF ( timestep_scheme(1:5) == 'runge' ) THEN
2614	IF ( intermediate_timestep_count == 1 ) THEN
2615	DO k = nzb+1, nzt
2616	te_m(k,j,i) = tend(k,j,i)
2617	ENDDO
2618	ELSEIF ( intermediate_timestep_count < &
2619	intermediate_timestep_count_max ) THEN
2620	DO k = nzb+1, nzt
2621	te_m(k,j,i) = -9.5625_wp * tend(k,j,i) + &
2622	5.3125_wp * te_m(k,j,i)
2623	ENDDO
2624	ENDIF
2625	ENDIF
2626
2627	ENDIF ! TKE equation
2628
2629	!
2630	!-- If required, compute prognostic equation for TKE dissipation rate
2631	IF ( rans_tke_e ) THEN
2632	!
2633	!-- Tendency-terms for dissipation
2634	tend(:,j,i) = 0.0_wp
2635	IF ( timestep_scheme(1:5) == 'runge' &
2636	.AND. .NOT. use_upstream_for_tke ) THEN
2637	IF ( ws_scheme_sca ) THEN
2638	CALL advec_s_ws( advc_flags_s, &
2639	i, j, diss, 'diss', flux_s_diss, diss_s_diss, &
2640	flux_l_diss, diss_l_diss, i_omp, tn, &
2641	bc_dirichlet_l .OR. bc_radiation_l, &
2642	bc_dirichlet_n .OR. bc_radiation_n, &
2643	bc_dirichlet_r .OR. bc_radiation_r, &
2644	bc_dirichlet_s .OR. bc_radiation_s )
2645	ELSE
2646	CALL advec_s_pw( i, j, diss )
2647	ENDIF
2648	ELSE
2649	CALL advec_s_up( i, j, diss )
2650	ENDIF
2651	!
2652	!-- Production of TKE dissipation rate
2653	CALL production_e_ij( i, j, .TRUE. )
2654	!
2655	!-- Diffusion term of TKE dissipation rate
2656	CALL diffusion_diss_ij( i, j )
2657	!
2658	!-- Additional sink term for flows through plant canopies
2659	! IF ( plant_canopy ) CALL pcm_tendency( i, j, ? ) !> @todo not yet implemented
2660
2661	! CALL user_actions( i, j, 'diss-tendency' ) ToDo: find general solution for circular dependency between modules
2662
2663	!
2664	!-- Prognostic equation for TKE dissipation
2665	!-- Eliminate negative dissipation values, which can occur due to
2666	!-- numerical reasons in the course of the integration. In such cases
2667	!-- the old dissipation value is reduced by 90%.
2668	DO k = nzb+1, nzt
2669	diss_p(k,j,i) = diss(k,j,i) + ( dt_3d * ( tsc(2) * tend(k,j,i) + &
2670	tsc(3) * tdiss_m(k,j,i) ) &
2671	) &
2672	* MERGE( 1.0_wp, 0.0_wp, &
2673	BTEST( wall_flags_0(k,j,i), 0 )&
2674	)
2675	ENDDO
2676
2677	!
2678	!-- Calculate tendencies for the next Runge-Kutta step
2679	IF ( timestep_scheme(1:5) == 'runge' ) THEN
2680	IF ( intermediate_timestep_count == 1 ) THEN
2681	DO k = nzb+1, nzt
2682	tdiss_m(k,j,i) = tend(k,j,i)
2683	ENDDO
2684	ELSEIF ( intermediate_timestep_count < &
2685	intermediate_timestep_count_max ) THEN
2686	DO k = nzb+1, nzt
2687	tdiss_m(k,j,i) = -9.5625_wp * tend(k,j,i) + &
2688	5.3125_wp * tdiss_m(k,j,i)
2689	ENDDO
2690	ENDIF
2691	ENDIF
2692
2693	ENDIF ! dissipation equation
2694
2695	END SUBROUTINE tcm_prognostic_equations_ij
2696
2697
2698	!------------------------------------------------------------------------------!
2699	! Description:
2700	! ------------
2701	!> Production terms (shear + buoyancy) of the TKE.
2702	!> Vector-optimized version
2703	!> @warning The case with constant_flux_layer = F and use_surface_fluxes = T is
2704	!> not considered well!
2705	!------------------------------------------------------------------------------!
2706	SUBROUTINE production_e( diss_production )
2707
2708	USE arrays_3d, &
2709	ONLY: ddzw, dd2zu, drho_air_zw, q, ql, d_exner, exner
2710
2711	USE control_parameters, &
2712	ONLY: cloud_droplets, constant_flux_layer, neutral, &
2713	rho_reference, use_single_reference_value, use_surface_fluxes, &
2714	use_top_fluxes
2715
2716	USE grid_variables, &
2717	ONLY: ddx, dx, ddy, dy
2718
2719	USE bulk_cloud_model_mod, &
2720	ONLY: bulk_cloud_model
2721
2722	IMPLICIT NONE
2723
2724	LOGICAL :: diss_production
2725
2726	INTEGER(iwp) :: i !< running index x-direction
2727	INTEGER(iwp) :: j !< running index y-direction
2728	INTEGER(iwp) :: k !< running index z-direction
2729	INTEGER(iwp) :: l !< running index for different surface type orientation
2730	INTEGER(iwp) :: m !< running index surface elements
2731	INTEGER(iwp) :: surf_e !< end index of surface elements at given i-j position
2732	INTEGER(iwp) :: surf_s !< start index of surface elements at given i-j position
2733	INTEGER(iwp) :: flag_nr !< number of masking flag
2734
2735	REAL(wp) :: def !< ( du_i/dx_j + du_j/dx_i ) * du_i/dx_j
2736	REAL(wp) :: flag !< flag to mask topography
2737	REAL(wp) :: k1 !< temporary factor
2738	REAL(wp) :: k2 !< temporary factor
2739	REAL(wp) :: km_neutral !< diffusion coefficient assuming neutral conditions - used to compute shear production at surfaces
2740	REAL(wp) :: theta !< virtual potential temperature
2741	REAL(wp) :: temp !< theta * Exner-function
2742	REAL(wp) :: sign_dir !< sign of wall-tke flux, depending on wall orientation
2743	REAL(wp) :: usvs !< momentum flux u"v"
2744	REAL(wp) :: vsus !< momentum flux v"u"
2745	REAL(wp) :: wsus !< momentum flux w"u"
2746	REAL(wp) :: wsvs !< momentum flux w"v"
2747
2748	REAL(wp), DIMENSION(nzb+1:nzt) :: dudx !< Gradient of u-component in x-direction
2749	REAL(wp), DIMENSION(nzb+1:nzt) :: dudy !< Gradient of u-component in y-direction
2750	REAL(wp), DIMENSION(nzb+1:nzt) :: dudz !< Gradient of u-component in z-direction
2751	REAL(wp), DIMENSION(nzb+1:nzt) :: dvdx !< Gradient of v-component in x-direction
2752	REAL(wp), DIMENSION(nzb+1:nzt) :: dvdy !< Gradient of v-component in y-direction
2753	REAL(wp), DIMENSION(nzb+1:nzt) :: dvdz !< Gradient of v-component in z-direction
2754	REAL(wp), DIMENSION(nzb+1:nzt) :: dwdx !< Gradient of w-component in x-direction
2755	REAL(wp), DIMENSION(nzb+1:nzt) :: dwdy !< Gradient of w-component in y-direction
2756	REAL(wp), DIMENSION(nzb+1:nzt) :: dwdz !< Gradient of w-component in z-direction
2757	REAL(wp), DIMENSION(nzb+1:nzt) :: tmp_flux !< temporary flux-array in z-direction
2758
2759
2760
2761	!
2762	!-- Calculate TKE production by shear. Calculate gradients at all grid
2763	!-- points first, gradients at surface-bounded grid points will be
2764	!-- overwritten further below.
2765	!$ACC PARALLEL LOOP COLLAPSE(2) PRIVATE(i, j, l) &
2766	!$ACC PRIVATE(surf_s, surf_e) &
2767	!$ACC PRIVATE(dudx(:), dudy(:), dudz(:), dvdx(:), dvdy(:), dvdz(:), dwdx(:), dwdy(:), dwdz(:)) &
2768	!$ACC PRESENT(e, u, v, w, diss, dd2zu, ddzw, km, wall_flags_0) &
2769	!$ACC PRESENT(tend) &
2770	!$ACC PRESENT(surf_def_h(0:1), surf_def_v(0:3)) &
2771	!$ACC PRESENT(surf_lsm_h, surf_lsm_v(0:3)) &
2772	!$ACC PRESENT(surf_usm_h, surf_usm_v(0:3))
2773	DO i = nxl, nxr
2774	DO j = nys, nyn
2775	!$ACC LOOP PRIVATE(k)
2776	DO k = nzb+1, nzt
2777
2778	dudx(k) = ( u(k,j,i+1) - u(k,j,i) ) * ddx
2779	dudy(k) = 0.25_wp * ( u(k,j+1,i) + u(k,j+1,i+1) - &
2780	u(k,j-1,i) - u(k,j-1,i+1) ) * ddy
2781	dudz(k) = 0.5_wp * ( u(k+1,j,i) + u(k+1,j,i+1) - &
2782	u(k-1,j,i) - u(k-1,j,i+1) ) * dd2zu(k)
2783
2784	dvdx(k) = 0.25_wp * ( v(k,j,i+1) + v(k,j+1,i+1) - &
2785	v(k,j,i-1) - v(k,j+1,i-1) ) * ddx
2786	dvdy(k) = ( v(k,j+1,i) - v(k,j,i) ) * ddy
2787	dvdz(k) = 0.5_wp * ( v(k+1,j,i) + v(k+1,j+1,i) - &
2788	v(k-1,j,i) - v(k-1,j+1,i) ) * dd2zu(k)
2789
2790	dwdx(k) = 0.25_wp * ( w(k,j,i+1) + w(k-1,j,i+1) - &
2791	w(k,j,i-1) - w(k-1,j,i-1) ) * ddx
2792	dwdy(k) = 0.25_wp * ( w(k,j+1,i) + w(k-1,j+1,i) - &
2793	w(k,j-1,i) - w(k-1,j-1,i) ) * ddy
2794	dwdz(k) = ( w(k,j,i) - w(k-1,j,i) ) * ddzw(k)
2795
2796	ENDDO
2797
2798
2799	flag_nr = 29
2800
2801
2802	IF ( constant_flux_layer ) THEN
2803	!
2804
2805	flag_nr = 0
2806
2807	!-- Position beneath wall
2808	!-- (2) - Will allways be executed.
2809	!-- 'bottom and wall: use u_0,v_0 and wall functions'
2810	!
2811	!-- Compute gradients at north- and south-facing surfaces.
2812	!-- First, for default surfaces, then for urban surfaces.
2813	!-- Note, so far no natural vertical surfaces implemented
2814	DO l = 0, 1
2815	surf_s = surf_def_v(l)%start_index(j,i)
2816	surf_e = surf_def_v(l)%end_index(j,i)
2817	!$ACC LOOP PRIVATE(m, k, usvs, wsvs, km_neutral, sign_dir)
2818	DO m = surf_s, surf_e
2819	k = surf_def_v(l)%k(m)
2820	usvs = surf_def_v(l)%mom_flux_tke(0,m)
2821	wsvs = surf_def_v(l)%mom_flux_tke(1,m)
2822
2823	km_neutral = kappa * ( usvs2 + wsvs2 )**0.25_wp &
2824	* 0.5_wp * dy
2825	!
2826	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
2827	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
2828	BTEST( wall_flags_0(k,j-1,i), flag_nr ) )
2829	dudy(k) = sign_dir * usvs / ( km_neutral + 1E-10_wp )
2830	dwdy(k) = sign_dir * wsvs / ( km_neutral + 1E-10_wp )
2831	ENDDO
2832	!
2833	!-- Natural surfaces
2834	surf_s = surf_lsm_v(l)%start_index(j,i)
2835	surf_e = surf_lsm_v(l)%end_index(j,i)
2836	!$ACC LOOP PRIVATE(m, k, usvs, wsvs, km_neutral, sign_dir)
2837	DO m = surf_s, surf_e
2838	k = surf_lsm_v(l)%k(m)
2839	usvs = surf_lsm_v(l)%mom_flux_tke(0,m)
2840	wsvs = surf_lsm_v(l)%mom_flux_tke(1,m)
2841
2842	km_neutral = kappa * ( usvs2 + wsvs2 )**0.25_wp &
2843	* 0.5_wp * dy
2844	!
2845	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
2846	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
2847	BTEST( wall_flags_0(k,j-1,i), flag_nr ) )
2848	dudy(k) = sign_dir * usvs / ( km_neutral + 1E-10_wp )
2849	dwdy(k) = sign_dir * wsvs / ( km_neutral + 1E-10_wp )
2850	ENDDO
2851	!
2852	!-- Urban surfaces
2853	surf_s = surf_usm_v(l)%start_index(j,i)
2854	surf_e = surf_usm_v(l)%end_index(j,i)
2855	!$ACC LOOP PRIVATE(m, k, usvs, wsvs, km_neutral, sign_dir)
2856	DO m = surf_s, surf_e
2857	k = surf_usm_v(l)%k(m)
2858	usvs = surf_usm_v(l)%mom_flux_tke(0,m)
2859	wsvs = surf_usm_v(l)%mom_flux_tke(1,m)
2860
2861	km_neutral = kappa * ( usvs2 + wsvs2 )**0.25_wp &
2862	* 0.5_wp * dy
2863	!
2864	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
2865	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
2866	BTEST( wall_flags_0(k,j-1,i), flag_nr ) )
2867	dudy(k) = sign_dir * usvs / ( km_neutral + 1E-10_wp )
2868	dwdy(k) = sign_dir * wsvs / ( km_neutral + 1E-10_wp )
2869	ENDDO
2870	ENDDO
2871	!
2872	!-- Compute gradients at east- and west-facing walls
2873	DO l = 2, 3
2874	surf_s = surf_def_v(l)%start_index(j,i)
2875	surf_e = surf_def_v(l)%end_index(j,i)
2876	!$ACC LOOP PRIVATE(m, k, vsus, wsus, km_neutral, sign_dir)
2877	DO m = surf_s, surf_e
2878	k = surf_def_v(l)%k(m)
2879	vsus = surf_def_v(l)%mom_flux_tke(0,m)
2880	wsus = surf_def_v(l)%mom_flux_tke(1,m)
2881
2882	km_neutral = kappa * ( vsus2 + wsus2 )**0.25_wp &
2883	* 0.5_wp * dx
2884	!
2885	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
2886	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
2887	BTEST( wall_flags_0(k,j,i-1), flag_nr ) )
2888	dvdx(k) = sign_dir * vsus / ( km_neutral + 1E-10_wp )
2889	dwdx(k) = sign_dir * wsus / ( km_neutral + 1E-10_wp )
2890	ENDDO
2891	!
2892	!-- Natural surfaces
2893	surf_s = surf_lsm_v(l)%start_index(j,i)
2894	surf_e = surf_lsm_v(l)%end_index(j,i)
2895	!$ACC LOOP PRIVATE(m, k, vsus, wsus, km_neutral, sign_dir)
2896	DO m = surf_s, surf_e
2897	k = surf_lsm_v(l)%k(m)
2898	vsus = surf_lsm_v(l)%mom_flux_tke(0,m)
2899	wsus = surf_lsm_v(l)%mom_flux_tke(1,m)
2900
2901	km_neutral = kappa * ( vsus2 + wsus2 )**0.25_wp &
2902	* 0.5_wp * dx
2903	!
2904	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
2905	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
2906	BTEST( wall_flags_0(k,j,i-1), flag_nr ) )
2907	dvdx(k) = sign_dir * vsus / ( km_neutral + 1E-10_wp )
2908	dwdx(k) = sign_dir * wsus / ( km_neutral + 1E-10_wp )
2909	ENDDO
2910	!
2911	!-- Urban surfaces
2912	surf_s = surf_usm_v(l)%start_index(j,i)
2913	surf_e = surf_usm_v(l)%end_index(j,i)
2914	!$ACC LOOP PRIVATE(m, k, vsus, wsus, km_neutral, sign_dir)
2915	DO m = surf_s, surf_e
2916	k = surf_usm_v(l)%k(m)
2917	vsus = surf_usm_v(l)%mom_flux_tke(0,m)
2918	wsus = surf_usm_v(l)%mom_flux_tke(1,m)
2919
2920	km_neutral = kappa * ( vsus2 + wsus2 )**0.25_wp &
2921	* 0.5_wp * dx
2922	!
2923	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
2924	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
2925	BTEST( wall_flags_0(k,j,i-1), flag_nr ) )
2926	dvdx(k) = sign_dir * vsus / ( km_neutral + 1E-10_wp )
2927	dwdx(k) = sign_dir * wsus / ( km_neutral + 1E-10_wp )
2928	ENDDO
2929	ENDDO
2930	!
2931	!-- Compute gradients at upward-facing surfaces
2932	surf_s = surf_def_h(0)%start_index(j,i)
2933	surf_e = surf_def_h(0)%end_index(j,i)
2934	!$ACC LOOP PRIVATE(m, k)
2935	DO m = surf_s, surf_e
2936	k = surf_def_h(0)%k(m)
2937	!
2938	!-- Please note, actually, an interpolation of u_0 and v_0
2939	!-- onto the grid center would be required. However, this
2940	!-- would require several data transfers between 2D-grid and
2941	!-- wall type. The effect of this missing interpolation is
2942	!-- negligible. (See also production_e_init).
2943	dudz(k) = ( u(k+1,j,i) - surf_def_h(0)%u_0(m) ) * dd2zu(k)
2944	dvdz(k) = ( v(k+1,j,i) - surf_def_h(0)%v_0(m) ) * dd2zu(k)
2945
2946	ENDDO
2947	!
2948	!-- Natural surfaces
2949	surf_s = surf_lsm_h%start_index(j,i)
2950	surf_e = surf_lsm_h%end_index(j,i)
2951	!$ACC LOOP PRIVATE(m, k)
2952	DO m = surf_s, surf_e
2953	k = surf_lsm_h%k(m)
2954
2955	dudz(k) = ( u(k+1,j,i) - surf_lsm_h%u_0(m) ) * dd2zu(k)
2956	dvdz(k) = ( v(k+1,j,i) - surf_lsm_h%v_0(m) ) * dd2zu(k)
2957
2958	ENDDO
2959	!
2960	!-- Urban surfaces
2961	surf_s = surf_usm_h%start_index(j,i)
2962	surf_e = surf_usm_h%end_index(j,i)
2963	!$ACC LOOP PRIVATE(m, k)
2964	DO m = surf_s, surf_e
2965	k = surf_usm_h%k(m)
2966
2967	dudz(k) = ( u(k+1,j,i) - surf_usm_h%u_0(m) ) * dd2zu(k)
2968	dvdz(k) = ( v(k+1,j,i) - surf_usm_h%v_0(m) ) * dd2zu(k)
2969
2970	ENDDO
2971	!
2972	!-- Compute gradients at downward-facing walls, only for
2973	!-- non-natural default surfaces
2974	surf_s = surf_def_h(1)%start_index(j,i)
2975	surf_e = surf_def_h(1)%end_index(j,i)
2976	!$ACC LOOP PRIVATE(m, k)
2977	DO m = surf_s, surf_e
2978	k = surf_def_h(1)%k(m)
2979
2980	dudz(k) = ( surf_def_h(1)%u_0(m) - u(k-1,j,i) ) * dd2zu(k)
2981	dvdz(k) = ( surf_def_h(1)%v_0(m) - v(k-1,j,i) ) * dd2zu(k)
2982
2983	ENDDO
2984
2985
2986	ENDIF
2987
2988
2989	!$ACC LOOP PRIVATE(k, def, flag)
2990	DO k = nzb+1, nzt
2991
2992	def = 2.0_wp * ( dudx(k)2 + dvdy(k)2 + dwdz(k)**2 ) + &
2993	dudy(k)2 + dvdx(k)2 + dwdx(k)**2 + &
2994	dwdy(k)2 + dudz(k)2 + dvdz(k)**2 + &
2995	2.0_wp * ( dvdx(k)dudy(k) + dwdx(k)dudz(k) + &
2996	dwdy(k)*dvdz(k) )
2997
2998	IF ( def < 0.0_wp ) def = 0.0_wp
2999
3000	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),flag_nr) )
3001
3002	IF ( .NOT. diss_production ) THEN
3003
3004	!-- Compute tendency for TKE-production from shear
3005	tend(k,j,i) = tend(k,j,i) + km(k,j,i) * def * flag
3006
3007	ELSE
3008
3009	!-- RANS mode: Compute tendency for dissipation-rate-production from shear
3010	tend(k,j,i) = tend(k,j,i) + km(k,j,i) * def * flag * &
3011	diss(k,j,i)/( e(k,j,i) + 1.0E-20_wp ) * c_1
3012
3013	ENDIF
3014
3015	ENDDO
3016
3017
3018	ENDDO
3019	ENDDO
3020
3021	!
3022	!-- If required, calculate TKE production by buoyancy
3023	IF ( .NOT. neutral ) THEN
3024
3025	IF ( .NOT. humidity ) THEN
3026
3027	IF ( ocean_mode ) THEN
3028	!
3029	!-- So far in the ocean no special treatment of density flux
3030	!-- in the bottom and top surface layer
3031	DO i = nxl, nxr
3032	DO j = nys, nyn
3033
3034	DO k = nzb+1, nzt
3035	tmp_flux(k) = kh(k,j,i) * ( prho(k+1,j,i) - prho(k-1,j,i) ) * dd2zu(k)
3036	ENDDO
3037	!
3038	!-- Treatment of near-surface grid points, at up- and down-
3039	!-- ward facing surfaces
3040	IF ( use_surface_fluxes ) THEN
3041	DO l = 0, 1
3042	surf_s = surf_def_h(l)%start_index(j,i)
3043	surf_e = surf_def_h(l)%end_index(j,i)
3044	DO m = surf_s, surf_e
3045	k = surf_def_h(l)%k(m)
3046	tmp_flux(k) = drho_air_zw(k-1) * surf_def_h(l)%shf(m)
3047	ENDDO
3048	ENDDO
3049	ENDIF
3050
3051	IF ( use_top_fluxes ) THEN
3052	surf_s = surf_def_h(2)%start_index(j,i)
3053	surf_e = surf_def_h(2)%end_index(j,i)
3054	DO m = surf_s, surf_e
3055	k = surf_def_h(2)%k(m)
3056	tmp_flux(k) = drho_air_zw(k) * surf_def_h(2)%shf(m)
3057	ENDDO
3058	ENDIF
3059
3060	IF ( .NOT. diss_production ) THEN
3061
3062	!-- Compute tendency for TKE-production from shear
3063	DO k = nzb+1, nzt
3064	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3065	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3066	MERGE( rho_reference, prho(k,j,i), &
3067	use_single_reference_value ) )
3068	ENDDO
3069
3070	ELSE
3071
3072	!-- RANS mode: Compute tendency for dissipation-rate-production from shear
3073	DO k = nzb+1, nzt
3074	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3075	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3076	MERGE( rho_reference, prho(k,j,i), &
3077	use_single_reference_value ) ) * &
3078	diss(k,j,i)/( e(k,j,i) + 1.0E-20_wp ) * &
3079	c_3
3080	ENDDO
3081
3082	ENDIF
3083
3084	ENDDO
3085	ENDDO
3086
3087	ELSE ! or IF ( .NOT. ocean_mode ) THEN
3088
3089	!$ACC PARALLEL LOOP COLLAPSE(2) PRIVATE(i, j) &
3090	!$ACC PRIVATE(surf_s, surf_e) &
3091	!$ACC PRIVATE(tmp_flux(nzb+1:nzt)) &
3092	!$ACC PRESENT(e, diss, kh, pt, dd2zu, drho_air_zw, wall_flags_0) &
3093	!$ACC PRESENT(tend) &
3094	!$ACC PRESENT(surf_def_h(0:2)) &
3095	!$ACC PRESENT(surf_lsm_h) &
3096	!$ACC PRESENT(surf_usm_h)
3097	DO i = nxl, nxr
3098	DO j = nys, nyn
3099
3100	!$ACC LOOP PRIVATE(k)
3101	DO k = nzb+1, nzt
3102	tmp_flux(k) = -1.0_wp * kh(k,j,i) * ( pt(k+1,j,i) - pt(k-1,j,i) ) * dd2zu(k)
3103	ENDDO
3104
3105	IF ( use_surface_fluxes ) THEN
3106	!
3107	!-- Default surfaces, up- and downward-facing
3108	DO l = 0, 1
3109	surf_s = surf_def_h(l)%start_index(j,i)
3110	surf_e = surf_def_h(l)%end_index(j,i)
3111	!$ACC LOOP PRIVATE(m, k)
3112	DO m = surf_s, surf_e
3113	k = surf_def_h(l)%k(m)
3114	tmp_flux(k) = drho_air_zw(k-1) * surf_def_h(l)%shf(m)
3115	ENDDO
3116	ENDDO
3117	!
3118	!-- Natural surfaces
3119	surf_s = surf_lsm_h%start_index(j,i)
3120	surf_e = surf_lsm_h%end_index(j,i)
3121	!$ACC LOOP PRIVATE(m, k)
3122	DO m = surf_s, surf_e
3123	k = surf_lsm_h%k(m)
3124	tmp_flux(k) = drho_air_zw(k-1) * surf_lsm_h%shf(m)
3125	ENDDO
3126	!
3127	!-- Urban surfaces
3128	surf_s = surf_usm_h%start_index(j,i)
3129	surf_e = surf_usm_h%end_index(j,i)
3130	!$ACC LOOP PRIVATE(m, k)
3131	DO m = surf_s, surf_e
3132	k = surf_usm_h%k(m)
3133	tmp_flux(k) = drho_air_zw(k-1) * surf_usm_h%shf(m)
3134	ENDDO
3135	ENDIF
3136
3137	IF ( use_top_fluxes ) THEN
3138	surf_s = surf_def_h(2)%start_index(j,i)
3139	surf_e = surf_def_h(2)%end_index(j,i)
3140	!$ACC LOOP PRIVATE(m, k)
3141	DO m = surf_s, surf_e
3142	k = surf_def_h(2)%k(m)
3143	tmp_flux(k) = drho_air_zw(k) * surf_def_h(2)%shf(m)
3144	ENDDO
3145	ENDIF
3146
3147	IF ( .NOT. diss_production ) THEN
3148
3149	!-- Compute tendency for TKE-production from shear
3150	!$ACC LOOP PRIVATE(k, flag)
3151	DO k = nzb+1, nzt
3152	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3153	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3154	MERGE( pt_reference, pt(k,j,i), &
3155	use_single_reference_value ) )
3156	ENDDO
3157
3158	ELSE
3159
3160	!-- RANS mode: Compute tendency for dissipation-rate-production from shear
3161	DO k = nzb+1, nzt
3162	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3163	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3164	MERGE( pt_reference, pt(k,j,i), &
3165	use_single_reference_value ) ) * &
3166	diss(k,j,i)/( e(k,j,i) + 1.0E-20_wp ) * &
3167	c_3
3168	ENDDO
3169
3170	ENDIF
3171
3172	ENDDO
3173	ENDDO
3174
3175	ENDIF ! from IF ( .NOT. ocean_mode )
3176
3177	ELSE ! or IF ( humidity ) THEN
3178
3179	DO i = nxl, nxr
3180	DO j = nys, nyn
3181
3182	DO k = nzb+1, nzt
3183
3184	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3185	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3186	k2 = 0.61_wp * pt(k,j,i)
3187	tmp_flux(k) = -1.0_wp * kh(k,j,i) * &
3188	( k1 * ( pt(k+1,j,i) - pt(k-1,j,i) ) + &
3189	k2 * ( q(k+1,j,i) - q(k-1,j,i) ) &
3190	) * dd2zu(k)
3191	ELSE IF ( bulk_cloud_model ) THEN
3192	IF ( ql(k,j,i) == 0.0_wp ) THEN
3193	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3194	k2 = 0.61_wp * pt(k,j,i)
3195	ELSE
3196	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3197	temp = theta * exner(k)
3198	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3199	( q(k,j,i) - ql(k,j,i) ) * &
3200	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3201	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3202	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3203	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3204	ENDIF
3205	tmp_flux(k) = -1.0_wp * kh(k,j,i) * &
3206	( k1 * ( pt(k+1,j,i) - pt(k-1,j,i) ) + &
3207	k2 * ( q(k+1,j,i) - q(k-1,j,i) ) &
3208	) * dd2zu(k)
3209	ELSE IF ( cloud_droplets ) THEN
3210	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3211	k2 = 0.61_wp * pt(k,j,i)
3212	tmp_flux(k) = -1.0_wp * kh(k,j,i) * &
3213	( k1 * ( pt(k+1,j,i) - pt(k-1,j,i) ) + &
3214	k2 * ( q(k+1,j,i) - q(k-1,j,i) ) - &
3215	pt(k,j,i) * ( ql(k+1,j,i) - &
3216	ql(k-1,j,i) ) ) * dd2zu(k)
3217	ENDIF
3218
3219	ENDDO
3220
3221	IF ( use_surface_fluxes ) THEN
3222	!
3223	!-- Treat horizontal default surfaces
3224	DO l = 0, 1
3225	surf_s = surf_def_h(l)%start_index(j,i)
3226	surf_e = surf_def_h(l)%end_index(j,i)
3227	DO m = surf_s, surf_e
3228	k = surf_def_h(l)%k(m)
3229
3230	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3231	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3232	k2 = 0.61_wp * pt(k,j,i)
3233	ELSE IF ( bulk_cloud_model ) THEN
3234	IF ( ql(k,j,i) == 0.0_wp ) THEN
3235	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3236	k2 = 0.61_wp * pt(k,j,i)
3237	ELSE
3238	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3239	temp = theta * exner(k)
3240	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3241	( q(k,j,i) - ql(k,j,i) ) * &
3242	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3243	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3244	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3245	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3246	ENDIF
3247	ELSE IF ( cloud_droplets ) THEN
3248	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3249	k2 = 0.61_wp * pt(k,j,i)
3250	ENDIF
3251
3252	tmp_flux(k) = ( k1 * surf_def_h(l)%shf(m) + &
3253	k2 * surf_def_h(l)%qsws(m) &
3254	) * drho_air_zw(k-1)
3255	ENDDO
3256	ENDDO
3257	!
3258	!-- Treat horizontal natural surfaces
3259	surf_s = surf_lsm_h%start_index(j,i)
3260	surf_e = surf_lsm_h%end_index(j,i)
3261	DO m = surf_s, surf_e
3262	k = surf_lsm_h%k(m)
3263
3264	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3265	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3266	k2 = 0.61_wp * pt(k,j,i)
3267	ELSE IF ( bulk_cloud_model ) THEN
3268	IF ( ql(k,j,i) == 0.0_wp ) THEN
3269	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3270	k2 = 0.61_wp * pt(k,j,i)
3271	ELSE
3272	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3273	temp = theta * exner(k)
3274	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3275	( q(k,j,i) - ql(k,j,i) ) * &
3276	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3277	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3278	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3279	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3280	ENDIF
3281	ELSE IF ( cloud_droplets ) THEN
3282	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3283	k2 = 0.61_wp * pt(k,j,i)
3284	ENDIF
3285
3286	tmp_flux(k) = ( k1 * surf_lsm_h%shf(m) + &
3287	k2 * surf_lsm_h%qsws(m) &
3288	) * drho_air_zw(k-1)
3289	ENDDO
3290	!
3291	!-- Treat horizontal urban surfaces
3292	surf_s = surf_usm_h%start_index(j,i)
3293	surf_e = surf_usm_h%end_index(j,i)
3294	DO m = surf_s, surf_e
3295	k = surf_usm_h%k(m)
3296
3297	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3298	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3299	k2 = 0.61_wp * pt(k,j,i)
3300	ELSE IF ( bulk_cloud_model ) THEN
3301	IF ( ql(k,j,i) == 0.0_wp ) THEN
3302	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3303	k2 = 0.61_wp * pt(k,j,i)
3304	ELSE
3305	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3306	temp = theta * exner(k)
3307	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3308	( q(k,j,i) - ql(k,j,i) ) * &
3309	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3310	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3311	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3312	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3313	ENDIF
3314	ELSE IF ( cloud_droplets ) THEN
3315	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3316	k2 = 0.61_wp * pt(k,j,i)
3317	ENDIF
3318
3319	tmp_flux(k) = ( k1 * surf_usm_h%shf(m) + &
3320	k2 * surf_usm_h%qsws(m) &
3321	) * drho_air_zw(k-1)
3322	ENDDO
3323
3324	ENDIF ! from IF ( use_surface_fluxes ) THEN
3325
3326	IF ( use_top_fluxes ) THEN
3327
3328	surf_s = surf_def_h(2)%start_index(j,i)
3329	surf_e = surf_def_h(2)%end_index(j,i)
3330	DO m = surf_s, surf_e
3331	k = surf_def_h(2)%k(m)
3332
3333	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3334	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3335	k2 = 0.61_wp * pt(k,j,i)
3336	ELSE IF ( bulk_cloud_model ) THEN
3337	IF ( ql(k,j,i) == 0.0_wp ) THEN
3338	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3339	k2 = 0.61_wp * pt(k,j,i)
3340	ELSE
3341	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3342	temp = theta * exner(k)
3343	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3344	( q(k,j,i) - ql(k,j,i) ) * &
3345	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3346	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3347	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3348	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3349	ENDIF
3350	ELSE IF ( cloud_droplets ) THEN
3351	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3352	k2 = 0.61_wp * pt(k,j,i)
3353	ENDIF
3354
3355	tmp_flux(k) = ( k1 * surf_def_h(2)%shf(m) + &
3356	k2 * surf_def_h(2)%qsws(m) &
3357	) * drho_air_zw(k)
3358
3359	ENDDO
3360
3361	ENDIF ! from IF ( use_top_fluxes ) THEN
3362
3363	IF ( .NOT. diss_production ) THEN
3364
3365	!-- Compute tendency for TKE-production from shear
3366	DO k = nzb+1, nzt
3367	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3368	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3369	MERGE( vpt_reference, vpt(k,j,i), &
3370	use_single_reference_value ) )
3371	ENDDO
3372
3373	ELSE
3374
3375	!-- RANS mode: Compute tendency for dissipation-rate-production from shear
3376	DO k = nzb+1, nzt
3377	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3378	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3379	MERGE( vpt_reference, vpt(k,j,i), &
3380	use_single_reference_value ) ) * &
3381	diss(k,j,i)/( e(k,j,i) + 1.0E-20_wp ) * &
3382	c_3
3383	ENDDO
3384
3385	ENDIF
3386
3387	ENDDO
3388	ENDDO
3389
3390	ENDIF
3391
3392	ENDIF
3393
3394	END SUBROUTINE production_e
3395
3396
3397	!------------------------------------------------------------------------------!
3398	! Description:
3399	! ------------
3400	!> Production terms (shear + buoyancy) of the TKE.
3401	!> Cache-optimized version
3402	!> @warning The case with constant_flux_layer = F and use_surface_fluxes = T is
3403	!> not considered well!
3404	!------------------------------------------------------------------------------!
3405	SUBROUTINE production_e_ij( i, j, diss_production )
3406
3407	USE arrays_3d, &
3408	ONLY: ddzw, dd2zu, drho_air_zw, q, ql, d_exner, exner
3409
3410	USE control_parameters, &
3411	ONLY: cloud_droplets, constant_flux_layer, neutral, &
3412	rho_reference, use_single_reference_value, use_surface_fluxes, &
3413	use_top_fluxes
3414
3415	USE grid_variables, &
3416	ONLY: ddx, dx, ddy, dy
3417
3418	USE bulk_cloud_model_mod, &
3419	ONLY: bulk_cloud_model
3420
3421	IMPLICIT NONE
3422
3423	LOGICAL :: diss_production
3424
3425	INTEGER(iwp) :: i !< running index x-direction
3426	INTEGER(iwp) :: j !< running index y-direction
3427	INTEGER(iwp) :: k !< running index z-direction
3428	INTEGER(iwp) :: l !< running index for different surface type orientation
3429	INTEGER(iwp) :: m !< running index surface elements
3430	INTEGER(iwp) :: surf_e !< end index of surface elements at given i-j position
3431	INTEGER(iwp) :: surf_s !< start index of surface elements at given i-j position
3432	INTEGER(iwp) :: flag_nr !< number of masking flag
3433
3434	REAL(wp) :: def !< ( du_i/dx_j + du_j/dx_i ) * du_i/dx_j
3435	REAL(wp) :: flag !< flag to mask topography
3436	REAL(wp) :: k1 !< temporary factor
3437	REAL(wp) :: k2 !< temporary factor
3438	REAL(wp) :: km_neutral !< diffusion coefficient assuming neutral conditions - used to compute shear production at surfaces
3439	REAL(wp) :: theta !< virtual potential temperature
3440	REAL(wp) :: temp !< theta * Exner-function
3441	REAL(wp) :: sign_dir !< sign of wall-tke flux, depending on wall orientation
3442	REAL(wp) :: usvs !< momentum flux u"v"
3443	REAL(wp) :: vsus !< momentum flux v"u"
3444	REAL(wp) :: wsus !< momentum flux w"u"
3445	REAL(wp) :: wsvs !< momentum flux w"v"
3446
3447	REAL(wp), DIMENSION(nzb+1:nzt) :: dudx !< Gradient of u-component in x-direction
3448	REAL(wp), DIMENSION(nzb+1:nzt) :: dudy !< Gradient of u-component in y-direction
3449	REAL(wp), DIMENSION(nzb+1:nzt) :: dudz !< Gradient of u-component in z-direction
3450	REAL(wp), DIMENSION(nzb+1:nzt) :: dvdx !< Gradient of v-component in x-direction
3451	REAL(wp), DIMENSION(nzb+1:nzt) :: dvdy !< Gradient of v-component in y-direction
3452	REAL(wp), DIMENSION(nzb+1:nzt) :: dvdz !< Gradient of v-component in z-direction
3453	REAL(wp), DIMENSION(nzb+1:nzt) :: dwdx !< Gradient of w-component in x-direction
3454	REAL(wp), DIMENSION(nzb+1:nzt) :: dwdy !< Gradient of w-component in y-direction
3455	REAL(wp), DIMENSION(nzb+1:nzt) :: dwdz !< Gradient of w-component in z-direction
3456	REAL(wp), DIMENSION(nzb+1:nzt) :: tmp_flux !< temporary flux-array in z-direction
3457
3458
3459
3460	!
3461	!-- Calculate TKE production by shear. Calculate gradients at all grid
3462	!-- points first, gradients at surface-bounded grid points will be
3463	!-- overwritten further below.
3464	DO k = nzb+1, nzt
3465
3466	dudx(k) = ( u(k,j,i+1) - u(k,j,i) ) * ddx
3467	dudy(k) = 0.25_wp * ( u(k,j+1,i) + u(k,j+1,i+1) - &
3468	u(k,j-1,i) - u(k,j-1,i+1) ) * ddy
3469	dudz(k) = 0.5_wp * ( u(k+1,j,i) + u(k+1,j,i+1) - &
3470	u(k-1,j,i) - u(k-1,j,i+1) ) * dd2zu(k)
3471
3472	dvdx(k) = 0.25_wp * ( v(k,j,i+1) + v(k,j+1,i+1) - &
3473	v(k,j,i-1) - v(k,j+1,i-1) ) * ddx
3474	dvdy(k) = ( v(k,j+1,i) - v(k,j,i) ) * ddy
3475	dvdz(k) = 0.5_wp * ( v(k+1,j,i) + v(k+1,j+1,i) - &
3476	v(k-1,j,i) - v(k-1,j+1,i) ) * dd2zu(k)
3477
3478	dwdx(k) = 0.25_wp * ( w(k,j,i+1) + w(k-1,j,i+1) - &
3479	w(k,j,i-1) - w(k-1,j,i-1) ) * ddx
3480	dwdy(k) = 0.25_wp * ( w(k,j+1,i) + w(k-1,j+1,i) - &
3481	w(k,j-1,i) - w(k-1,j-1,i) ) * ddy
3482	dwdz(k) = ( w(k,j,i) - w(k-1,j,i) ) * ddzw(k)
3483
3484	ENDDO
3485
3486	flag_nr = 29
3487
3488	IF ( constant_flux_layer ) THEN
3489
3490	flag_nr = 0
3491
3492	!-- Position beneath wall
3493	!-- (2) - Will allways be executed.
3494	!-- 'bottom and wall: use u_0,v_0 and wall functions'
3495	!
3496	!-- Compute gradients at north- and south-facing surfaces.
3497	!-- First, for default surfaces, then for urban surfaces.
3498	!-- Note, so far no natural vertical surfaces implemented
3499	DO l = 0, 1
3500	surf_s = surf_def_v(l)%start_index(j,i)
3501	surf_e = surf_def_v(l)%end_index(j,i)
3502	DO m = surf_s, surf_e
3503	k = surf_def_v(l)%k(m)
3504	usvs = surf_def_v(l)%mom_flux_tke(0,m)
3505	wsvs = surf_def_v(l)%mom_flux_tke(1,m)
3506
3507	km_neutral = kappa * ( usvs2 + wsvs2 )**0.25_wp &
3508	* 0.5_wp * dy
3509	!
3510	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
3511	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
3512	BTEST( wall_flags_0(k,j-1,i), flag_nr ) )
3513	dudy(k) = sign_dir * usvs / ( km_neutral + 1E-10_wp )
3514	dwdy(k) = sign_dir * wsvs / ( km_neutral + 1E-10_wp )
3515	ENDDO
3516	!
3517	!-- Natural surfaces
3518	surf_s = surf_lsm_v(l)%start_index(j,i)
3519	surf_e = surf_lsm_v(l)%end_index(j,i)
3520	DO m = surf_s, surf_e
3521	k = surf_lsm_v(l)%k(m)
3522	usvs = surf_lsm_v(l)%mom_flux_tke(0,m)
3523	wsvs = surf_lsm_v(l)%mom_flux_tke(1,m)
3524
3525	km_neutral = kappa * ( usvs2 + wsvs2 )**0.25_wp &
3526	* 0.5_wp * dy
3527	!
3528	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
3529	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
3530	BTEST( wall_flags_0(k,j-1,i), flag_nr ) )
3531	dudy(k) = sign_dir * usvs / ( km_neutral + 1E-10_wp )
3532	dwdy(k) = sign_dir * wsvs / ( km_neutral + 1E-10_wp )
3533	ENDDO
3534	!
3535	!-- Urban surfaces
3536	surf_s = surf_usm_v(l)%start_index(j,i)
3537	surf_e = surf_usm_v(l)%end_index(j,i)
3538	DO m = surf_s, surf_e
3539	k = surf_usm_v(l)%k(m)
3540	usvs = surf_usm_v(l)%mom_flux_tke(0,m)
3541	wsvs = surf_usm_v(l)%mom_flux_tke(1,m)
3542
3543	km_neutral = kappa * ( usvs2 + wsvs2 )**0.25_wp &
3544	* 0.5_wp * dy
3545	!
3546	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
3547	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
3548	BTEST( wall_flags_0(k,j-1,i), flag_nr ) )
3549	dudy(k) = sign_dir * usvs / ( km_neutral + 1E-10_wp )
3550	dwdy(k) = sign_dir * wsvs / ( km_neutral + 1E-10_wp )
3551	ENDDO
3552	ENDDO
3553	!
3554	!-- Compute gradients at east- and west-facing walls
3555	DO l = 2, 3
3556	surf_s = surf_def_v(l)%start_index(j,i)
3557	surf_e = surf_def_v(l)%end_index(j,i)
3558	DO m = surf_s, surf_e
3559	k = surf_def_v(l)%k(m)
3560	vsus = surf_def_v(l)%mom_flux_tke(0,m)
3561	wsus = surf_def_v(l)%mom_flux_tke(1,m)
3562
3563	km_neutral = kappa * ( vsus2 + wsus2 )**0.25_wp &
3564	* 0.5_wp * dx
3565	!
3566	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
3567	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
3568	BTEST( wall_flags_0(k,j,i-1), flag_nr ) )
3569	dvdx(k) = sign_dir * vsus / ( km_neutral + 1E-10_wp )
3570	dwdx(k) = sign_dir * wsus / ( km_neutral + 1E-10_wp )
3571	ENDDO
3572	!
3573	!-- Natural surfaces
3574	surf_s = surf_lsm_v(l)%start_index(j,i)
3575	surf_e = surf_lsm_v(l)%end_index(j,i)
3576	DO m = surf_s, surf_e
3577	k = surf_lsm_v(l)%k(m)
3578	vsus = surf_lsm_v(l)%mom_flux_tke(0,m)
3579	wsus = surf_lsm_v(l)%mom_flux_tke(1,m)
3580
3581	km_neutral = kappa * ( vsus2 + wsus2 )**0.25_wp &
3582	* 0.5_wp * dx
3583	!
3584	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
3585	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
3586	BTEST( wall_flags_0(k,j,i-1), flag_nr ) )
3587	dvdx(k) = sign_dir * vsus / ( km_neutral + 1E-10_wp )
3588	dwdx(k) = sign_dir * wsus / ( km_neutral + 1E-10_wp )
3589	ENDDO
3590	!
3591	!-- Urban surfaces
3592	surf_s = surf_usm_v(l)%start_index(j,i)
3593	surf_e = surf_usm_v(l)%end_index(j,i)
3594	DO m = surf_s, surf_e
3595	k = surf_usm_v(l)%k(m)
3596	vsus = surf_usm_v(l)%mom_flux_tke(0,m)
3597	wsus = surf_usm_v(l)%mom_flux_tke(1,m)
3598
3599	km_neutral = kappa * ( vsus2 + wsus2 )**0.25_wp &
3600	* 0.5_wp * dx
3601	!
3602	!-- -1.0 for right-facing wall, 1.0 for left-facing wall
3603	sign_dir = MERGE( 1.0_wp, -1.0_wp, &
3604	BTEST( wall_flags_0(k,j,i-1), flag_nr ) )
3605	dvdx(k) = sign_dir * vsus / ( km_neutral + 1E-10_wp )
3606	dwdx(k) = sign_dir * wsus / ( km_neutral + 1E-10_wp )
3607	ENDDO
3608	ENDDO
3609	!
3610	!-- Compute gradients at upward-facing surfaces
3611	surf_s = surf_def_h(0)%start_index(j,i)
3612	surf_e = surf_def_h(0)%end_index(j,i)
3613	DO m = surf_s, surf_e
3614	k = surf_def_h(0)%k(m)
3615	!
3616	!-- Please note, actually, an interpolation of u_0 and v_0
3617	!-- onto the grid center would be required. However, this
3618	!-- would require several data transfers between 2D-grid and
3619	!-- wall type. The effect of this missing interpolation is
3620	!-- negligible. (See also production_e_init).
3621	dudz(k) = ( u(k+1,j,i) - surf_def_h(0)%u_0(m) ) * dd2zu(k)
3622	dvdz(k) = ( v(k+1,j,i) - surf_def_h(0)%v_0(m) ) * dd2zu(k)
3623
3624	ENDDO
3625	!
3626	!-- Natural surfaces
3627	surf_s = surf_lsm_h%start_index(j,i)
3628	surf_e = surf_lsm_h%end_index(j,i)
3629	DO m = surf_s, surf_e
3630	k = surf_lsm_h%k(m)
3631
3632	dudz(k) = ( u(k+1,j,i) - surf_lsm_h%u_0(m) ) * dd2zu(k)
3633	dvdz(k) = ( v(k+1,j,i) - surf_lsm_h%v_0(m) ) * dd2zu(k)
3634
3635	ENDDO
3636	!
3637	!-- Urban surfaces
3638	surf_s = surf_usm_h%start_index(j,i)
3639	surf_e = surf_usm_h%end_index(j,i)
3640	DO m = surf_s, surf_e
3641	k = surf_usm_h%k(m)
3642
3643	dudz(k) = ( u(k+1,j,i) - surf_usm_h%u_0(m) ) * dd2zu(k)
3644	dvdz(k) = ( v(k+1,j,i) - surf_usm_h%v_0(m) ) * dd2zu(k)
3645
3646	ENDDO
3647	!
3648	!-- Compute gradients at downward-facing walls, only for
3649	!-- non-natural default surfaces
3650	surf_s = surf_def_h(1)%start_index(j,i)
3651	surf_e = surf_def_h(1)%end_index(j,i)
3652	DO m = surf_s, surf_e
3653	k = surf_def_h(1)%k(m)
3654
3655	dudz(k) = ( surf_def_h(1)%u_0(m) - u(k-1,j,i) ) * dd2zu(k)
3656	dvdz(k) = ( surf_def_h(1)%v_0(m) - v(k-1,j,i) ) * dd2zu(k)
3657
3658	ENDDO
3659
3660	ENDIF
3661
3662	DO k = nzb+1, nzt
3663
3664	def = 2.0_wp * ( dudx(k)2 + dvdy(k)2 + dwdz(k)**2 ) + &
3665	dudy(k)2 + dvdx(k)2 + dwdx(k)**2 + &
3666	dwdy(k)2 + dudz(k)2 + dvdz(k)**2 + &
3667	2.0_wp * ( dvdx(k)dudy(k) + dwdx(k)dudz(k) + &
3668	dwdy(k)*dvdz(k) )
3669
3670	IF ( def < 0.0_wp ) def = 0.0_wp
3671
3672	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),flag_nr) )
3673
3674	IF ( .NOT. diss_production ) THEN
3675
3676	!-- Compute tendency for TKE-production from shear
3677	tend(k,j,i) = tend(k,j,i) + km(k,j,i) * def * flag
3678
3679	ELSE
3680
3681	!-- RANS mode: Compute tendency for dissipation-rate-production from shear
3682	tend(k,j,i) = tend(k,j,i) + km(k,j,i) * def * flag * &
3683	diss(k,j,i)/( e(k,j,i) + 1.0E-20_wp ) * c_1
3684
3685	ENDIF
3686
3687	ENDDO
3688
3689	!
3690	!-- If required, calculate TKE production by buoyancy
3691	IF ( .NOT. neutral ) THEN
3692
3693	IF ( .NOT. humidity ) THEN
3694
3695	IF ( ocean_mode ) THEN
3696	!
3697	!-- So far in the ocean no special treatment of density flux
3698	!-- in the bottom and top surface layer
3699	DO k = nzb+1, nzt
3700	tmp_flux(k) = kh(k,j,i) * ( prho(k+1,j,i) - prho(k-1,j,i) ) * dd2zu(k)
3701	ENDDO
3702	!
3703	!-- Treatment of near-surface grid points, at up- and down-
3704	!-- ward facing surfaces
3705	IF ( use_surface_fluxes ) THEN
3706	DO l = 0, 1
3707	surf_s = surf_def_h(l)%start_index(j,i)
3708	surf_e = surf_def_h(l)%end_index(j,i)
3709	DO m = surf_s, surf_e
3710	k = surf_def_h(l)%k(m)
3711	tmp_flux(k) = drho_air_zw(k-1) * surf_def_h(l)%shf(m)
3712	ENDDO
3713	ENDDO
3714	ENDIF
3715
3716	IF ( use_top_fluxes ) THEN
3717	surf_s = surf_def_h(2)%start_index(j,i)
3718	surf_e = surf_def_h(2)%end_index(j,i)
3719	DO m = surf_s, surf_e
3720	k = surf_def_h(2)%k(m)
3721	tmp_flux(k) = drho_air_zw(k) * surf_def_h(2)%shf(m)
3722	ENDDO
3723	ENDIF
3724
3725	IF ( .NOT. diss_production ) THEN
3726
3727	!-- Compute tendency for TKE-production from shear
3728	DO k = nzb+1, nzt
3729	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3730	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3731	MERGE( rho_reference, prho(k,j,i), &
3732	use_single_reference_value ) )
3733	ENDDO
3734
3735	ELSE
3736
3737	!-- RANS mode: Compute tendency for dissipation-rate-production from shear
3738	DO k = nzb+1, nzt
3739	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3740	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3741	MERGE( rho_reference, prho(k,j,i), &
3742	use_single_reference_value ) ) * &
3743	diss(k,j,i)/( e(k,j,i) + 1.0E-20_wp ) * &
3744	c_3
3745	ENDDO
3746
3747	ENDIF
3748
3749
3750	ELSE ! or IF ( .NOT. ocean_mode ) THEN
3751
3752	DO k = nzb+1, nzt
3753	tmp_flux(k) = -1.0_wp * kh(k,j,i) * ( pt(k+1,j,i) - pt(k-1,j,i) ) * dd2zu(k)
3754	ENDDO
3755
3756	IF ( use_surface_fluxes ) THEN
3757	!
3758	!-- Default surfaces, up- and downward-facing
3759	DO l = 0, 1
3760	surf_s = surf_def_h(l)%start_index(j,i)
3761	surf_e = surf_def_h(l)%end_index(j,i)
3762	DO m = surf_s, surf_e
3763	k = surf_def_h(l)%k(m)
3764	tmp_flux(k) = drho_air_zw(k-1) * surf_def_h(l)%shf(m)
3765	ENDDO
3766	ENDDO
3767	!
3768	!-- Natural surfaces
3769	surf_s = surf_lsm_h%start_index(j,i)
3770	surf_e = surf_lsm_h%end_index(j,i)
3771	DO m = surf_s, surf_e
3772	k = surf_lsm_h%k(m)
3773	tmp_flux(k) = drho_air_zw(k-1) * surf_lsm_h%shf(m)
3774	ENDDO
3775	!
3776	!-- Urban surfaces
3777	surf_s = surf_usm_h%start_index(j,i)
3778	surf_e = surf_usm_h%end_index(j,i)
3779	DO m = surf_s, surf_e
3780	k = surf_usm_h%k(m)
3781	tmp_flux(k) = drho_air_zw(k-1) * surf_usm_h%shf(m)
3782	ENDDO
3783	ENDIF
3784
3785	IF ( use_top_fluxes ) THEN
3786	surf_s = surf_def_h(2)%start_index(j,i)
3787	surf_e = surf_def_h(2)%end_index(j,i)
3788	DO m = surf_s, surf_e
3789	k = surf_def_h(2)%k(m)
3790	tmp_flux(k) = drho_air_zw(k) * surf_def_h(2)%shf(m)
3791	ENDDO
3792	ENDIF
3793
3794	IF ( .NOT. diss_production ) THEN
3795
3796	!-- Compute tendency for TKE-production from shear
3797	DO k = nzb+1, nzt
3798	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3799	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3800	MERGE( pt_reference, pt(k,j,i), &
3801	use_single_reference_value ) )
3802	ENDDO
3803
3804	ELSE
3805
3806	!-- RANS mode: Compute tendency for dissipation-rate-production from shear
3807	DO k = nzb+1, nzt
3808	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
3809	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
3810	MERGE( pt_reference, pt(k,j,i), &
3811	use_single_reference_value ) ) * &
3812	diss(k,j,i)/( e(k,j,i) + 1.0E-20_wp ) * &
3813	c_3
3814	ENDDO
3815
3816	ENDIF
3817
3818	ENDIF ! from IF ( .NOT. ocean_mode )
3819
3820	ELSE ! or IF ( humidity ) THEN
3821
3822	DO k = nzb+1, nzt
3823
3824	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3825	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3826	k2 = 0.61_wp * pt(k,j,i)
3827	tmp_flux(k) = -1.0_wp * kh(k,j,i) * &
3828	( k1 * ( pt(k+1,j,i) - pt(k-1,j,i) ) + &
3829	k2 * ( q(k+1,j,i) - q(k-1,j,i) ) &
3830	) * dd2zu(k)
3831	ELSE IF ( bulk_cloud_model ) THEN
3832	IF ( ql(k,j,i) == 0.0_wp ) THEN
3833	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3834	k2 = 0.61_wp * pt(k,j,i)
3835	ELSE
3836	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3837	temp = theta * exner(k)
3838	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3839	( q(k,j,i) - ql(k,j,i) ) * &
3840	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3841	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3842	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3843	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3844	ENDIF
3845	tmp_flux(k) = -1.0_wp * kh(k,j,i) * &
3846	( k1 * ( pt(k+1,j,i) - pt(k-1,j,i) ) + &
3847	k2 * ( q(k+1,j,i) - q(k-1,j,i) ) &
3848	) * dd2zu(k)
3849	ELSE IF ( cloud_droplets ) THEN
3850	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3851	k2 = 0.61_wp * pt(k,j,i)
3852	tmp_flux(k) = -1.0_wp * kh(k,j,i) * &
3853	( k1 * ( pt(k+1,j,i) - pt(k-1,j,i) ) + &
3854	k2 * ( q(k+1,j,i) - q(k-1,j,i) ) - &
3855	pt(k,j,i) * ( ql(k+1,j,i) - &
3856	ql(k-1,j,i) ) ) * dd2zu(k)
3857	ENDIF
3858
3859	ENDDO
3860
3861	IF ( use_surface_fluxes ) THEN
3862	!
3863	!-- Treat horizontal default surfaces
3864	DO l = 0, 1
3865	surf_s = surf_def_h(l)%start_index(j,i)
3866	surf_e = surf_def_h(l)%end_index(j,i)
3867	DO m = surf_s, surf_e
3868	k = surf_def_h(l)%k(m)
3869
3870	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3871	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3872	k2 = 0.61_wp * pt(k,j,i)
3873	ELSE IF ( bulk_cloud_model ) THEN
3874	IF ( ql(k,j,i) == 0.0_wp ) THEN
3875	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3876	k2 = 0.61_wp * pt(k,j,i)
3877	ELSE
3878	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3879	temp = theta * exner(k)
3880	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3881	( q(k,j,i) - ql(k,j,i) ) * &
3882	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3883	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3884	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3885	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3886	ENDIF
3887	ELSE IF ( cloud_droplets ) THEN
3888	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3889	k2 = 0.61_wp * pt(k,j,i)
3890	ENDIF
3891
3892	tmp_flux(k) = ( k1 * surf_def_h(l)%shf(m) + &
3893	k2 * surf_def_h(l)%qsws(m) &
3894	) * drho_air_zw(k-1)
3895	ENDDO
3896	ENDDO
3897	!
3898	!-- Treat horizontal natural surfaces
3899	surf_s = surf_lsm_h%start_index(j,i)
3900	surf_e = surf_lsm_h%end_index(j,i)
3901	DO m = surf_s, surf_e
3902	k = surf_lsm_h%k(m)
3903
3904	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3905	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3906	k2 = 0.61_wp * pt(k,j,i)
3907	ELSE IF ( bulk_cloud_model ) THEN
3908	IF ( ql(k,j,i) == 0.0_wp ) THEN
3909	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3910	k2 = 0.61_wp * pt(k,j,i)
3911	ELSE
3912	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3913	temp = theta * exner(k)
3914	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3915	( q(k,j,i) - ql(k,j,i) ) * &
3916	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3917	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3918	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3919	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3920	ENDIF
3921	ELSE IF ( cloud_droplets ) THEN
3922	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3923	k2 = 0.61_wp * pt(k,j,i)
3924	ENDIF
3925
3926	tmp_flux(k) = ( k1 * surf_lsm_h%shf(m) + &
3927	k2 * surf_lsm_h%qsws(m) &
3928	) * drho_air_zw(k-1)
3929	ENDDO
3930	!
3931	!-- Treat horizontal urban surfaces
3932	surf_s = surf_usm_h%start_index(j,i)
3933	surf_e = surf_usm_h%end_index(j,i)
3934	DO m = surf_s, surf_e
3935	k = surf_usm_h%k(m)
3936
3937	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3938	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3939	k2 = 0.61_wp * pt(k,j,i)
3940	ELSE IF ( bulk_cloud_model ) THEN
3941	IF ( ql(k,j,i) == 0.0_wp ) THEN
3942	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3943	k2 = 0.61_wp * pt(k,j,i)
3944	ELSE
3945	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3946	temp = theta * exner(k)
3947	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3948	( q(k,j,i) - ql(k,j,i) ) * &
3949	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3950	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3951	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3952	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3953	ENDIF
3954	ELSE IF ( cloud_droplets ) THEN
3955	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3956	k2 = 0.61_wp * pt(k,j,i)
3957	ENDIF
3958
3959	tmp_flux(k) = ( k1 * surf_usm_h%shf(m) + &
3960	k2 * surf_usm_h%qsws(m) &
3961	) * drho_air_zw(k-1)
3962	ENDDO
3963
3964	ENDIF ! from IF ( use_surface_fluxes ) THEN
3965
3966	IF ( use_top_fluxes ) THEN
3967
3968	surf_s = surf_def_h(2)%start_index(j,i)
3969	surf_e = surf_def_h(2)%end_index(j,i)
3970	DO m = surf_s, surf_e
3971	k = surf_def_h(2)%k(m)
3972
3973	IF ( .NOT. bulk_cloud_model .AND. .NOT. cloud_droplets ) THEN
3974	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3975	k2 = 0.61_wp * pt(k,j,i)
3976	ELSE IF ( bulk_cloud_model ) THEN
3977	IF ( ql(k,j,i) == 0.0_wp ) THEN
3978	k1 = 1.0_wp + 0.61_wp * q(k,j,i)
3979	k2 = 0.61_wp * pt(k,j,i)
3980	ELSE
3981	theta = pt(k,j,i) + d_exner(k) * lv_d_cp * ql(k,j,i)
3982	temp = theta * exner(k)
3983	k1 = ( 1.0_wp - q(k,j,i) + 1.61_wp * &
3984	( q(k,j,i) - ql(k,j,i) ) * &
3985	( 1.0_wp + rd_d_rv * lv_d_rd / temp ) ) / &
3986	( 1.0_wp + rd_d_rv * lv_d_rd * lv_d_cp * &
3987	( q(k,j,i) - ql(k,j,i) ) / ( temp * temp ) )
3988	k2 = theta * ( lv_d_cp / temp * k1 - 1.0_wp )
3989	ENDIF
3990	ELSE IF ( cloud_droplets ) THEN
3991	k1 = 1.0_wp + 0.61_wp * q(k,j,i) - ql(k,j,i)
3992	k2 = 0.61_wp * pt(k,j,i)
3993	ENDIF
3994
3995	tmp_flux(k) = ( k1 * surf_def_h(2)%shf(m) + &
3996	k2 * surf_def_h(2)%qsws(m) &
3997	) * drho_air_zw(k)
3998
3999	ENDDO
4000
4001	ENDIF ! from IF ( use_top_fluxes ) THEN
4002
4003	IF ( .NOT. diss_production ) THEN
4004
4005	!-- Compute tendency for TKE-production from shear
4006	DO k = nzb+1, nzt
4007	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
4008	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
4009	MERGE( vpt_reference, vpt(k,j,i), &
4010	use_single_reference_value ) )
4011	ENDDO
4012
4013	ELSE
4014
4015	!-- RANS mode: Compute tendency for dissipation-rate-production from shear
4016	DO k = nzb+1, nzt
4017	flag = MERGE( 1.0_wp, 0.0_wp, BTEST(wall_flags_0(k,j,i),0) )
4018	tend(k,j,i) = tend(k,j,i) + flag * tmp_flux(k) * ( g / &
4019	MERGE( vpt_reference, vpt(k,j,i), &
4020	use_single_reference_value ) ) * &
4021	diss(k,j,i)/( e(k,j,i) + 1.0E-20_wp ) * &
4022	c_3
4023	ENDDO
4024
4025	ENDIF
4026
4027	ENDIF
4028
4029	ENDIF
4030
4031	END SUBROUTINE production_e_ij
4032
4033
4034	!------------------------------------------------------------------------------!
4035	! Description:
4036	! ------------
4037	!> Diffusion and dissipation terms for the TKE.
4038	!> Vector-optimized version
4039	!------------------------------------------------------------------------------!
4040	SUBROUTINE diffusion_e( var, var_reference )
4041
4042	#if defined( _OPENACC )
4043	USE arrays_3d, &
4044	ONLY: ddzu, dd2zu, ddzw, drho_air, rho_air_zw
4045
4046	USE control_parameters, &
4047	ONLY: atmos_ocean_sign, use_single_reference_value, &
4048	wall_adjustment, wall_adjustment_factor
4049	#else
4050	USE arrays_3d, &
4051	ONLY: ddzu, ddzw, drho_air, rho_air_zw
4052	#endif
4053	USE grid_variables, &
4054	ONLY: ddx2, ddy2
4055
4056	USE bulk_cloud_model_mod, &
4057	ONLY: collision_turbulence
4058
4059	USE particle_attributes, &
4060	ONLY: use_sgs_for_particles, wang_kernel
4061
4062	IMPLICIT NONE
4063
4064	INTEGER(iwp) :: i !< running index x direction
4065	INTEGER(iwp) :: j !< running index y direction
4066	INTEGER(iwp) :: k !< running index z direction
4067	INTEGER(iwp) :: m !< running index surface elements
4068
4069	REAL(wp) :: flag !< flag to mask topography
4070	REAL(wp) :: l !< mixing length
4071	REAL(wp) :: ll !< adjusted l
4072	REAL(wp) :: var_reference !< reference temperature
4073	#if defined( _OPENACC )
4074	!
4075	!-- From mixing_length_les:
4076	REAL(wp) :: l_stable !< mixing length according to stratification
4077	!
4078	!-- From mixing_length_rans:
4079	REAL(wp) :: duv2_dz2 !< squared vertical gradient of wind vector
4080	REAL(wp) :: l_diss !< mixing length for dissipation
4081	REAL(wp) :: rif !< Richardson flux number
4082	!
4083	!-- From both:
4084	REAL(wp) :: dvar_dz !< vertical gradient of var
4085	#endif
4086
4087	REAL(wp) :: dissipation !< TKE dissipation
4088
4089	REAL(wp), DIMENSION(:,:,:), POINTER :: var !< temperature
4090
4091
4092	!
4093	!-- Calculate the tendency terms
4094	!$ACC PARALLEL LOOP COLLAPSE(3) PRIVATE(i, j, k) &
4095	!$ACC PRIVATE(flag, l, ll, dissipation) &
4096	!$ACC PRIVATE(l_stable, duv2_dz2, l_diss, rif, dvar_dz) &
4097	!$ACC PRESENT(e, u, v, km, var, wall_flags_0) &
4098	!$ACC PRESENT(ddzu, dd2zu, ddzw, rho_air_zw, drho_air) &
4099	!$ACC PRESENT(l_black, l_grid, l_wall) &
4100