Home

Context Navigation

source: palm/trunk/SOURCE/diagnostic_output_quantities_mod.f90 @ 4768

Last change on this file since 4768 was 4768, checked in by suehring, 4 years ago
Enable 3D data output also with 64-bit precision
Property svn:keywords set to `Id`
File size: 89.4 KB

Rev	Line
[3994]	1	!> @file diagnostic_output_quantities_mod.f90
[4583]	2	!--------------------------------------------------------------------------------------------------!
[3994]	3	! This file is part of the PALM model system.
	4	!
[4583]	5	! PALM is free software: you can redistribute it and/or modify it under the terms of the GNU General
	6	! Public License as published by the Free Software Foundation, either version 3 of the License, or
	7	! (at your option) any later version.
[3994]	8	!
[4583]	9	! PALM is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the
	10	! implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
	11	! Public License for more details.
[3994]	12	!
[4583]	13	! You should have received a copy of the GNU General Public License along with PALM. If not, see
	14	! <http://www.gnu.org/licenses/>.
[3994]	15	!
[4360]	16	! Copyright 1997-2020 Leibniz Universitaet Hannover
[4583]	17	!--------------------------------------------------------------------------------------------------!
[3994]	18	!
	19	! Current revisions:
	20	! ------------------
[4671]	21	!
	22	!
[3994]	23	! Former revisions:
	24	! -----------------
	25	! $Id: diagnostic_output_quantities_mod.f90 4768 2020-11-02 19:11:23Z suehring $
[4768]	26	! Enable 3D data output also with 64-bit precision
	27	!
	28	! 4757 2020-10-26 10:23:38Z schwenkel
[4757]	29	! Implement relative humidity as diagnostic output quantity
	30	!
	31	! 4671 2020-09-09 20:27:58Z pavelkrc
[4671]	32	! Implementation of downward facing USM and LSM surfaces
	33	!
	34	! 4583 2020-06-29 12:36:47Z raasch
[4583]	35	! file re-formatted to follow the PALM coding standard
	36	!
	37	! 4560 2020-06-11 12:19:47Z suehring
[4560]	38	! - Bugfix in calculation of vertical momentum and scalar fluxes
	39	! - remove averaged output variables from PUBLIC list
[4583]	40	!
[4560]	41	! 4535 2020-05-15 12:07:23Z raasch
[4535]	42	! bugfix for restart data format query
[4583]	43	!
[4535]	44	! 4518 2020-05-04 15:44:28Z suehring
[4583]	45	! * Define arrays over ghost points in order to allow for standard mpi-io treatment. By this
	46	! modularization of restart-data input is possible with the module interface.
[4518]	47	! * Bugfix: add missing restart input of wtheta_av, wq_av, wu_av, and wv_av.
[4583]	48	!
[4518]	49	! 4517 2020-05-03 14:29:30Z raasch
[4457]	50	! use statement for exchange horiz added,
	51	! bugfix for call of exchange horiz 2d
[4583]	52	!
[4457]	53	! 4431 2020-02-27 23:23:01Z gronemeier
[4431]	54	! added wspeed and wdir output; bugfix: set fill_value in case of masked output
	55	!
	56	! 4360 2020-01-07 11:25:50Z suehring
[4583]	57	! added output of wu, wv, wtheta and wq to enable covariance calculation according to temporal EC
	58	! method
[4431]	59	!
[4351]	60	! 4346 2019-12-18 11:55:56Z motisi
[4583]	61	! Introduction of wall_flags_total_0, which currently sets bits based on static topography
	62	! information used in wall_flags_static_0
[4431]	63	!
[4346]	64	! 4331 2019-12-10 18:25:02Z suehring
[4331]	65	! - Modularize 2-m potential temperature output
	66	! - New output for 10-m wind speed
[4431]	67	!
[4331]	68	! 4329 2019-12-10 15:46:36Z motisi
[4329]	69	! Renamed wall_flags_0 to wall_flags_static_0
[4431]	70	!
[4329]	71	! 4182 2019-08-22 15:20:23Z scharf
[4182]	72	! Corrected "Former revisions" section
[4431]	73	!
[4182]	74	! 4167 2019-08-16 11:01:48Z suehring
[4583]	75	! Changed behaviour of masked output over surface to follow terrain and ignore buildings
	76	! (J.Resler, T.Gronemeier)
[4431]	77	!
[4167]	78	! 4157 2019-08-14 09:19:12Z suehring
[4583]	79	! Initialization restructured, in order to work also when data output during spin-up is enabled.
[4431]	80	!
[4157]	81	! 4132 2019-08-02 12:34:17Z suehring
[4132]	82	! Bugfix in masked data output
[4431]	83	!
[4132]	84	! 4069 2019-07-01 14:05:51Z Giersch
[4583]	85	! Masked output running index mid has been introduced as a local variable to avoid runtime error
	86	! (Loop variable has been modified) in time_integration
[4431]	87	!
[4069]	88	! 4039 2019-06-18 10:32:41Z suehring
[4583]	89	! - Add output of uu, vv, ww to enable variance calculation according temporal EC method
[4039]	90	! - Allocate arrays only when they are required
	91	! - Formatting adjustment
	92	! - Rename subroutines
	93	! - Further modularization
[4431]	94	!
[4039]	95	! 3998 2019-05-23 13:38:11Z suehring
[4431]	96	! Bugfix in gathering all output strings
	97	!
[3998]	98	! 3995 2019-05-22 18:59:54Z suehring
[4583]	99	! Avoid compiler warnings about unused variable and fix string operation which is not allowed with
	100	! PGI compiler
[4431]	101	!
[3995]	102	! 3994 2019-05-22 18:08:09Z suehring
[3994]	103	! Initial revision
	104	!
[4182]	105	! Authors:
	106	! --------
[3994]	107	! @author Farah Kanani-Suehring
[4182]	108	!
[4431]	109	!
[3994]	110	! Description:
	111	! ------------
	112	!> ...
[4583]	113	!--------------------------------------------------------------------------------------------------!
[3994]	114	MODULE diagnostic_output_quantities_mod
[4431]	115
[4583]	116	USE arrays_3d, &
	117	ONLY: ddzu, &
[4757]	118	exner, &
	119	hyp, &
[4583]	120	pt, &
	121	q, &
	122	u, &
	123	v, &
	124	w, &
	125	zu, &
[4331]	126	zw
	127
[4583]	128	USE basic_constants_and_equations_mod, &
[4757]	129	ONLY: kappa, &
	130	pi, &
	131	magnus, &
	132	rd_d_rv
[4331]	133
[4583]	134	USE control_parameters, &
	135	ONLY: current_timestep_number, &
	136	data_output, &
[4757]	137	data_output_pr, &
	138	humidity, &
[4583]	139	message_string, &
	140	restart_data_format_output, &
[4331]	141	varnamelength
[4431]	142	!
[4583]	143	! USE cpulog, &
[3994]	144	! ONLY: cpu_log, log_point
[4039]	145
[4583]	146	USE exchange_horiz_mod, &
[4457]	147	ONLY: exchange_horiz_2d
	148
[4583]	149	USE grid_variables, &
[4039]	150	ONLY: ddx, ddy
[4331]	151
[4583]	152	USE indices, &
	153	ONLY: nbgp, &
	154	nxl, &
	155	nxlg, &
	156	nxr, &
	157	nxrg, &
	158	nyn, &
	159	nyng, &
	160	nys, &
	161	nysg, &
	162	nzb, &
	163	nzt, &
[4346]	164	wall_flags_total_0
[4331]	165
[3994]	166	USE kinds
	167
[4757]	168	USE profil_parameter, &
	169	ONLY: dopr_index
	170
[4518]	171	USE restart_data_mpi_io_mod, &
	172	ONLY: rd_mpi_io_check_array, &
	173	rrd_mpi_io, &
	174	wrd_mpi_io
	175
[4757]	176	USE statistics, &
	177	ONLY: hom, &
	178	pr_palm, &
	179	rmask, &
	180	statistic_regions, &
	181	sums_l
	182
[4583]	183	USE surface_mod, &
	184	ONLY: surf_def_h, &
	185	surf_lsm_h, &
	186	surf_type, &
[4331]	187	surf_usm_h
[3994]	188
	189	IMPLICIT NONE
	190
	191	CHARACTER(LEN=varnamelength), DIMENSION(500) :: do_all = ' '
	192
	193	INTEGER(iwp) :: timestep_number_at_prev_calc = 0 !< ...at previous diagnostic output calculation
[4431]	194
[4039]	195	LOGICAL :: initialized_diagnostic_output_quantities = .FALSE. !< flag indicating whether output is initialized
	196	LOGICAL :: prepared_diagnostic_output_quantities = .FALSE. !< flag indicating whether output is p
[3994]	197
[4331]	198	REAL(wp), DIMENSION(:,:), ALLOCATABLE :: pt_2m !< 2-m air potential temperature
	199	REAL(wp), DIMENSION(:,:), ALLOCATABLE :: pt_2m_av !< averaged 2-m air potential temperature
	200	REAL(wp), DIMENSION(:,:), ALLOCATABLE :: uv_10m !< horizontal wind speed at 10m
	201	REAL(wp), DIMENSION(:,:), ALLOCATABLE :: uv_10m_av !< averaged horizontal wind speed at 10m
	202
[4757]	203	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: rh !< relative humidity
	204	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: rh_av !< avg. relative humidity
[3994]	205	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: ti !< rotation(u,v,w) aka turbulence intensity
[4431]	206	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: ti_av !< avg. rotation(u,v,w) aka turbulence intensity
	207	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: u_center !< u at center of grid box
	208	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: u_center_av !< mean of u_center
	209	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: uu !< uu
	210	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: uu_av !< mean of uu
	211	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wspeed !< horizontal wind speed
	212	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wspeed_av !< mean of horizotal wind speed
	213	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: v_center !< v at center of grid box
	214	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: v_center_av !< mean of v_center
	215	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: vv !< vv
	216	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: vv_av !< mean of vv
	217	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wdir !< wind direction
	218	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wdir_av !< mean wind direction
	219	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: ww !< ww
	220	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: ww_av !< mean of ww
	221	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wu !< wu
	222	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wu_av !< mean of wu
	223	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wv !< wv
	224	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wv_av !< mean of wv
	225	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wtheta !< wtheta
	226	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wtheta_av !< mean of wtheta
	227	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wq !< wq
	228	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE, TARGET :: wq_av !< mean of wq
[3994]	229
	230
	231	SAVE
	232
	233	PRIVATE
	234
	235	!
	236	!-- Public variables
[4583]	237	PUBLIC do_all, &
	238	initialized_diagnostic_output_quantities, &
	239	prepared_diagnostic_output_quantities, &
[4560]	240	timestep_number_at_prev_calc
[4431]	241	!
	242	!-- Public routines
[4583]	243	PUBLIC doq_3d_data_averaging, &
	244	doq_calculate, &
	245	doq_check_data_output, &
[4757]	246	doq_check_data_output_pr, &
[4583]	247	doq_define_netcdf_grid, &
	248	doq_init, &
	249	doq_output_2d, &
	250	doq_output_3d, &
	251	doq_output_mask, &
[4757]	252	doq_statistics, &
[4583]	253	doq_rrd_local, &
[4039]	254	doq_wrd_local
[3994]	255
	256
[4039]	257	INTERFACE doq_3d_data_averaging
	258	MODULE PROCEDURE doq_3d_data_averaging
[4431]	259	END INTERFACE doq_3d_data_averaging
[3994]	260
[4039]	261	INTERFACE doq_calculate
	262	MODULE PROCEDURE doq_calculate
	263	END INTERFACE doq_calculate
[3994]	264
[4039]	265	INTERFACE doq_check_data_output
	266	MODULE PROCEDURE doq_check_data_output
	267	END INTERFACE doq_check_data_output
[4431]	268
[4757]	269	INTERFACE doq_check_data_output_pr
	270	MODULE PROCEDURE doq_check_data_output_pr
	271	END INTERFACE doq_check_data_output_pr
	272
[4039]	273	INTERFACE doq_define_netcdf_grid
	274	MODULE PROCEDURE doq_define_netcdf_grid
	275	END INTERFACE doq_define_netcdf_grid
[4431]	276
[4039]	277	INTERFACE doq_output_2d
	278	MODULE PROCEDURE doq_output_2d
	279	END INTERFACE doq_output_2d
[4431]	280
[4039]	281	INTERFACE doq_output_3d
	282	MODULE PROCEDURE doq_output_3d
	283	END INTERFACE doq_output_3d
[4431]	284
[4039]	285	INTERFACE doq_output_mask
	286	MODULE PROCEDURE doq_output_mask
	287	END INTERFACE doq_output_mask
[4431]	288
[4039]	289	INTERFACE doq_init
	290	MODULE PROCEDURE doq_init
	291	END INTERFACE doq_init
[3994]	292
[4757]	293	INTERFACE doq_statistics
	294	MODULE PROCEDURE doq_statistics
	295	END INTERFACE doq_statistics
	296
[4039]	297	INTERFACE doq_prepare
	298	MODULE PROCEDURE doq_prepare
	299	END INTERFACE doq_prepare
[4431]	300
[4518]	301	INTERFACE doq_rrd_local
	302	MODULE PROCEDURE doq_rrd_local_ftn
	303	MODULE PROCEDURE doq_rrd_local_mpi
	304	END INTERFACE doq_rrd_local
[4431]	305
[4039]	306	INTERFACE doq_wrd_local
	307	MODULE PROCEDURE doq_wrd_local
	308	END INTERFACE doq_wrd_local
[3994]	309
	310	CONTAINS
[4431]	311
[4583]	312	!--------------------------------------------------------------------------------------------------!
[4039]	313	! Description:
	314	! ------------
[4583]	315	!> Sum up and time-average diagnostic output quantities as well as allocate the array necessary for
	316	!> storing the average.
	317	!--------------------------------------------------------------------------------------------------!
[4039]	318	SUBROUTINE doq_3d_data_averaging( mode, variable )
[3994]	319
[4583]	320	USE control_parameters, &
[4039]	321	ONLY: average_count_3d
	322
[4431]	323	CHARACTER (LEN=*) :: mode !<
	324	CHARACTER (LEN=*) :: variable !<
[4039]	325
	326	INTEGER(iwp) :: i !<
	327	INTEGER(iwp) :: j !<
	328	INTEGER(iwp) :: k !<
	329
[4583]	330
[4039]	331	IF ( mode == 'allocate' ) THEN
	332
	333	SELECT CASE ( TRIM( variable ) )
	334
[4757]	335	CASE ( 'rh' )
	336	IF ( .NOT. ALLOCATED( rh_av ) ) THEN
	337	ALLOCATE( rh_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	338	ENDIF
	339	rh_av = 0.0_wp
	340
[4039]	341	CASE ( 'ti' )
	342	IF ( .NOT. ALLOCATED( ti_av ) ) THEN
[4518]	343	ALLOCATE( ti_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	344	ENDIF
	345	ti_av = 0.0_wp
[4431]	346
[4039]	347	CASE ( 'uu' )
	348	IF ( .NOT. ALLOCATED( uu_av ) ) THEN
[4518]	349	ALLOCATE( uu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	350	ENDIF
	351	uu_av = 0.0_wp
[4431]	352
[4039]	353	CASE ( 'vv' )
	354	IF ( .NOT. ALLOCATED( vv_av ) ) THEN
[4518]	355	ALLOCATE( vv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	356	ENDIF
	357	vv_av = 0.0_wp
[4431]	358
[4039]	359	CASE ( 'ww' )
	360	IF ( .NOT. ALLOCATED( ww_av ) ) THEN
[4518]	361	ALLOCATE( ww_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	362	ENDIF
	363	ww_av = 0.0_wp
[4431]	364
[4351]	365	CASE ( 'wu' )
	366	IF ( .NOT. ALLOCATED( wu_av ) ) THEN
[4518]	367	ALLOCATE( wu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	368	ENDIF
	369	wu_av = 0.0_wp
[4431]	370
[4351]	371	CASE ( 'wv' )
	372	IF ( .NOT. ALLOCATED( wv_av ) ) THEN
[4518]	373	ALLOCATE( wv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	374	ENDIF
	375	wv_av = 0.0_wp
[4431]	376
[4351]	377	CASE ( 'wtheta' )
	378	IF ( .NOT. ALLOCATED( wtheta_av ) ) THEN
[4518]	379	ALLOCATE( wtheta_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	380	ENDIF
	381	wtheta_av = 0.0_wp
[4431]	382
[4351]	383	CASE ( 'wq' )
	384	IF ( .NOT. ALLOCATED( wq_av ) ) THEN
[4518]	385	ALLOCATE( wq_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	386	ENDIF
	387	wq_av = 0.0_wp
[4431]	388
[4331]	389	CASE ( 'theta_2m*' )
	390	IF ( .NOT. ALLOCATED( pt_2m_av ) ) THEN
	391	ALLOCATE( pt_2m_av(nysg:nyng,nxlg:nxrg) )
	392	ENDIF
	393	pt_2m_av = 0.0_wp
[4431]	394
[4331]	395	CASE ( 'wspeed_10m*' )
	396	IF ( .NOT. ALLOCATED( uv_10m_av ) ) THEN
	397	ALLOCATE( uv_10m_av(nysg:nyng,nxlg:nxrg) )
	398	ENDIF
	399	uv_10m_av = 0.0_wp
[4039]	400
[4431]	401	CASE ( 'wspeed' )
	402	IF ( .NOT. ALLOCATED( wspeed_av ) ) THEN
[4518]	403	ALLOCATE( wspeed_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	404	ENDIF
	405	wspeed_av = 0.0_wp
	406
	407	CASE ( 'wdir' )
	408	IF ( .NOT. ALLOCATED( u_center_av ) ) THEN
[4518]	409	ALLOCATE( u_center_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	410	ENDIF
	411	IF ( .NOT. ALLOCATED( v_center_av ) ) THEN
[4518]	412	ALLOCATE( v_center_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	413	ENDIF
	414	u_center_av = 0.0_wp
	415	v_center_av = 0.0_wp
	416
[4039]	417	CASE DEFAULT
	418	CONTINUE
	419
	420	END SELECT
	421
	422	ELSEIF ( mode == 'sum' ) THEN
	423
	424	SELECT CASE ( TRIM( variable ) )
[4431]	425
[4757]	426	CASE ( 'rh' )
	427	IF ( ALLOCATED( rh_av ) ) THEN
	428	DO i = nxl, nxr
	429	DO j = nys, nyn
	430	DO k = nzb, nzt+1
	431	rh_av(k,j,i) = rh_av(k,j,i) + rh(k,j,i)
	432	ENDDO
	433	ENDDO
	434	ENDDO
	435	ENDIF
	436
[4039]	437	CASE ( 'ti' )
[4583]	438	IF ( ALLOCATED( ti_av ) ) THEN
[4039]	439	DO i = nxl, nxr
	440	DO j = nys, nyn
	441	DO k = nzb, nzt+1
	442	ti_av(k,j,i) = ti_av(k,j,i) + ti(k,j,i)
	443	ENDDO
	444	ENDDO
	445	ENDDO
	446	ENDIF
[4431]	447
[4039]	448	CASE ( 'uu' )
[4583]	449	IF ( ALLOCATED( uu_av ) ) THEN
[4039]	450	DO i = nxl, nxr
	451	DO j = nys, nyn
	452	DO k = nzb, nzt+1
	453	uu_av(k,j,i) = uu_av(k,j,i) + uu(k,j,i)
	454	ENDDO
	455	ENDDO
	456	ENDDO
	457	ENDIF
[4431]	458
[4039]	459	CASE ( 'vv' )
[4583]	460	IF ( ALLOCATED( vv_av ) ) THEN
[4039]	461	DO i = nxl, nxr
	462	DO j = nys, nyn
	463	DO k = nzb, nzt+1
	464	vv_av(k,j,i) = vv_av(k,j,i) + vv(k,j,i)
	465	ENDDO
	466	ENDDO
	467	ENDDO
	468	ENDIF
[4431]	469
[4039]	470	CASE ( 'ww' )
[4583]	471	IF ( ALLOCATED( ww_av ) ) THEN
[4039]	472	DO i = nxl, nxr
	473	DO j = nys, nyn
	474	DO k = nzb, nzt+1
	475	ww_av(k,j,i) = ww_av(k,j,i) + ww(k,j,i)
	476	ENDDO
	477	ENDDO
	478	ENDDO
	479	ENDIF
[4431]	480
[4351]	481	CASE ( 'wu' )
[4583]	482	IF ( ALLOCATED( wu_av ) ) THEN
[4351]	483	DO i = nxl, nxr
	484	DO j = nys, nyn
	485	DO k = nzb, nzt+1
	486	wu_av(k,j,i) = wu_av(k,j,i) + wu(k,j,i)
	487	ENDDO
	488	ENDDO
	489	ENDDO
	490	ENDIF
[4431]	491
[4351]	492	CASE ( 'wv' )
[4583]	493	IF ( ALLOCATED( wv_av ) ) THEN
[4351]	494	DO i = nxl, nxr
	495	DO j = nys, nyn
	496	DO k = nzb, nzt+1
	497	wv_av(k,j,i) = wv_av(k,j,i) + wv(k,j,i)
	498	ENDDO
	499	ENDDO
	500	ENDDO
	501	ENDIF
[4431]	502
[4351]	503	CASE ( 'wtheta' )
[4583]	504	IF ( ALLOCATED( wtheta_av ) ) THEN
[4351]	505	DO i = nxl, nxr
	506	DO j = nys, nyn
	507	DO k = nzb, nzt+1
	508	wtheta_av(k,j,i) = wtheta_av(k,j,i) + wtheta(k,j,i)
	509	ENDDO
	510	ENDDO
	511	ENDDO
	512	ENDIF
[4431]	513
[4351]	514	CASE ( 'wq' )
[4583]	515	IF ( ALLOCATED( wq_av ) ) THEN
[4351]	516	DO i = nxl, nxr
	517	DO j = nys, nyn
	518	DO k = nzb, nzt+1
	519	wq_av(k,j,i) = wq_av(k,j,i) + wq(k,j,i)
	520	ENDDO
	521	ENDDO
	522	ENDDO
	523	ENDIF
[4431]	524
[4331]	525	CASE ( 'theta_2m*' )
[4583]	526	IF ( ALLOCATED( pt_2m_av ) ) THEN
[4331]	527	DO i = nxl, nxr
	528	DO j = nys, nyn
	529	pt_2m_av(j,i) = pt_2m_av(j,i) + pt_2m(j,i)
	530	ENDDO
	531	ENDDO
	532	ENDIF
[4039]	533
[4331]	534	CASE ( 'wspeed_10m*' )
[4583]	535	IF ( ALLOCATED( uv_10m_av ) ) THEN
[4331]	536	DO i = nxl, nxr
	537	DO j = nys, nyn
	538	uv_10m_av(j,i) = uv_10m_av(j,i) + uv_10m(j,i)
	539	ENDDO
	540	ENDDO
	541	ENDIF
	542
[4431]	543	CASE ( 'wspeed' )
[4583]	544	IF ( ALLOCATED( wspeed_av ) ) THEN
[4431]	545	DO i = nxl, nxr
	546	DO j = nys, nyn
	547	DO k = nzb, nzt+1
	548	wspeed_av(k,j,i) = wspeed_av(k,j,i) + wspeed(k,j,i)
	549	ENDDO
	550	ENDDO
	551	ENDDO
	552	ENDIF
	553
	554	CASE ( 'wdir' )
[4583]	555	IF ( ALLOCATED( u_center_av ) .AND. ALLOCATED( v_center_av ) ) THEN
[4431]	556	DO i = nxl, nxr
	557	DO j = nys, nyn
	558	DO k = nzb, nzt+1
	559	u_center_av(k,j,i) = u_center_av(k,j,i) + u_center(k,j,i)
	560	v_center_av(k,j,i) = v_center_av(k,j,i) + v_center(k,j,i)
	561	ENDDO
	562	ENDDO
	563	ENDDO
	564	ENDIF
	565
[4039]	566	CASE DEFAULT
	567	CONTINUE
	568
	569	END SELECT
	570
	571	ELSEIF ( mode == 'average' ) THEN
	572
	573	SELECT CASE ( TRIM( variable ) )
	574
[4757]	575	CASE ( 'rh' )
	576	IF ( ALLOCATED( rh_av ) ) THEN
	577	DO i = nxl, nxr
	578	DO j = nys, nyn
	579	DO k = nzb, nzt+1
	580	rh_av(k,j,i) = rh_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	581	ENDDO
	582	ENDDO
	583	ENDDO
	584	ENDIF
	585
[4039]	586	CASE ( 'ti' )
[4583]	587	IF ( ALLOCATED( ti_av ) ) THEN
[4039]	588	DO i = nxl, nxr
	589	DO j = nys, nyn
	590	DO k = nzb, nzt+1
	591	ti_av(k,j,i) = ti_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	592	ENDDO
	593	ENDDO
	594	ENDDO
	595	ENDIF
[4431]	596
[4039]	597	CASE ( 'uu' )
[4583]	598	IF ( ALLOCATED( uu_av ) ) THEN
[4039]	599	DO i = nxl, nxr
	600	DO j = nys, nyn
	601	DO k = nzb, nzt+1
	602	uu_av(k,j,i) = uu_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	603	ENDDO
	604	ENDDO
	605	ENDDO
	606	ENDIF
[4431]	607
[4039]	608	CASE ( 'vv' )
[4583]	609	IF ( ALLOCATED( vv_av ) ) THEN
[4039]	610	DO i = nxl, nxr
	611	DO j = nys, nyn
	612	DO k = nzb, nzt+1
	613	vv_av(k,j,i) = vv_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	614	ENDDO
	615	ENDDO
	616	ENDDO
	617	ENDIF
[4431]	618
[4039]	619	CASE ( 'ww' )
[4583]	620	IF ( ALLOCATED( ww_av ) ) THEN
[4039]	621	DO i = nxl, nxr
	622	DO j = nys, nyn
	623	DO k = nzb, nzt+1
	624	ww_av(k,j,i) = ww_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	625	ENDDO
	626	ENDDO
	627	ENDDO
	628	ENDIF
	629
[4351]	630	CASE ( 'wu' )
[4583]	631	IF ( ALLOCATED( wu_av ) ) THEN
[4351]	632	DO i = nxl, nxr
	633	DO j = nys, nyn
	634	DO k = nzb, nzt+1
	635	wu_av(k,j,i) = wu_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	636	ENDDO
	637	ENDDO
	638	ENDDO
	639	ENDIF
[4431]	640
[4351]	641	CASE ( 'wv' )
[4583]	642	IF ( ALLOCATED( wv_av ) ) THEN
[4351]	643	DO i = nxl, nxr
	644	DO j = nys, nyn
	645	DO k = nzb, nzt+1
	646	wv_av(k,j,i) = wv_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	647	ENDDO
	648	ENDDO
	649	ENDDO
	650	ENDIF
[4431]	651
[4351]	652	CASE ( 'wtheta' )
[4583]	653	IF ( ALLOCATED( wtheta_av ) ) THEN
[4351]	654	DO i = nxl, nxr
	655	DO j = nys, nyn
	656	DO k = nzb, nzt+1
	657	wtheta_av(k,j,i) = wtheta_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	658	ENDDO
	659	ENDDO
	660	ENDDO
	661	ENDIF
[4431]	662
[4351]	663	CASE ( 'wq' )
[4583]	664	IF ( ALLOCATED( wq_av ) ) THEN
[4351]	665	DO i = nxl, nxr
	666	DO j = nys, nyn
	667	DO k = nzb, nzt+1
	668	wq_av(k,j,i) = wq_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	669	ENDDO
	670	ENDDO
	671	ENDDO
	672	ENDIF
	673
[4331]	674	CASE ( 'theta_2m*' )
[4583]	675	IF ( ALLOCATED( pt_2m_av ) ) THEN
[4331]	676	DO i = nxlg, nxrg
	677	DO j = nysg, nyng
	678	pt_2m_av(j,i) = pt_2m_av(j,i) / REAL( average_count_3d, KIND=wp )
	679	ENDDO
	680	ENDDO
[4457]	681	CALL exchange_horiz_2d( pt_2m_av )
[4331]	682	ENDIF
	683
	684	CASE ( 'wspeed_10m*' )
[4583]	685	IF ( ALLOCATED( uv_10m_av ) ) THEN
[4331]	686	DO i = nxlg, nxrg
	687	DO j = nysg, nyng
	688	uv_10m_av(j,i) = uv_10m_av(j,i) / REAL( average_count_3d, KIND=wp )
	689	ENDDO
	690	ENDDO
[4457]	691	CALL exchange_horiz_2d( uv_10m_av )
[4331]	692	ENDIF
	693
[4431]	694	CASE ( 'wspeed' )
[4583]	695	IF ( ALLOCATED( wspeed_av ) ) THEN
[4431]	696	DO i = nxl, nxr
	697	DO j = nys, nyn
	698	DO k = nzb, nzt+1
	699	wspeed_av(k,j,i) = wspeed_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	700	ENDDO
	701	ENDDO
	702	ENDDO
	703	ENDIF
	704
	705	CASE ( 'wdir' )
[4583]	706	IF ( ALLOCATED( u_center_av ) .AND. ALLOCATED( v_center_av ) ) THEN
[4431]	707
	708	IF ( .NOT. ALLOCATED( wdir_av ) ) THEN
[4518]	709	ALLOCATE( wdir_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	710	ENDIF
	711	wdir_av = 0.0_wp
	712
	713	DO i = nxl, nxr
	714	DO j = nys, nyn
	715	DO k = nzb, nzt+1
	716	u_center_av(k,j,i) = u_center_av(k,j,i) / REAL( average_count_3d, KIND=wp )
	717	v_center_av(k,j,i) = v_center_av(k,j,i) / REAL( average_count_3d, KIND=wp )
[4583]	718	wdir_av(k,j,i) = ATAN2( u_center_av(k,j,i), v_center_av(k,j,i) ) &
	719	/ pi * 180.0_wp + 180.0_wp
[4431]	720	ENDDO
	721	ENDDO
	722	ENDDO
	723	ENDIF
	724
[4039]	725	END SELECT
	726
	727	ENDIF
	728
	729
[4431]	730	END SUBROUTINE doq_3d_data_averaging
	731
[4583]	732	!--------------------------------------------------------------------------------------------------!
[3994]	733	! Description:
	734	! ------------
[4039]	735	!> Check data output for diagnostic output
[4583]	736	!--------------------------------------------------------------------------------------------------!
[4331]	737	SUBROUTINE doq_check_data_output( var, unit, i, ilen, k )
[3994]	738
[4039]	739	IMPLICIT NONE
[3994]	740
[4431]	741	CHARACTER (LEN=*) :: unit !<
[4039]	742	CHARACTER (LEN=*) :: var !<
[3994]	743
[4331]	744	INTEGER(iwp), OPTIONAL, INTENT(IN) :: i !< Current element of data_output
	745	INTEGER(iwp), OPTIONAL, INTENT(IN) :: ilen !< Length of current entry in data_output
	746	INTEGER(iwp), OPTIONAL, INTENT(IN) :: k !< Output is xy mode? 0 = no, 1 = yes
	747
[4583]	748
[4039]	749	SELECT CASE ( TRIM( var ) )
	750
[4757]	751	CASE ( 'rh' )
	752	unit = '%'
	753
[4039]	754	CASE ( 'ti' )
	755	unit = '1/s'
[4431]	756
[4039]	757	CASE ( 'uu' )
	758	unit = 'm2/s2'
[4431]	759
[4039]	760	CASE ( 'vv' )
	761	unit = 'm2/s2'
[4431]	762
[4039]	763	CASE ( 'ww' )
	764	unit = 'm2/s2'
[4431]	765
[4351]	766	CASE ( 'wu' )
	767	unit = 'm2/s2'
[4431]	768
[4351]	769	CASE ( 'wv' )
	770	unit = 'm2/s2'
[4431]	771
[4351]	772	CASE ( 'wtheta' )
	773	unit = 'Km/s'
[4431]	774
[4351]	775	CASE ( 'wq' )
	776	unit = 'm/s'
[4431]	777
	778	CASE ( 'wspeed' )
	779	unit = 'm/s'
	780
	781	CASE ( 'wdir' )
	782	unit = 'degree'
[4331]	783	!
	784	!-- Treat horizotal cross-section output quanatities
	785	CASE ( 'theta_2m', 'wspeed_10m' )
	786	!
	787	!-- Check if output quantity is _xy only.
	788	IF ( k == 0 .OR. data_output(i)(ilen-2:ilen) /= '_xy' ) THEN
[4583]	789	message_string = 'illegal value for data_output: "' // &
	790	TRIM( var ) // '" & only 2d-horizontal ' // &
[4331]	791	'cross sections are allowed for this value'
	792	CALL message( 'diagnostic_output', 'PA0111', 1, 2, 0, 6, 0 )
	793	ENDIF
	794
	795	IF ( TRIM( var ) == 'theta_2m*' ) unit = 'K'
	796	IF ( TRIM( var ) == 'wspeed_10m*' ) unit = 'm/s'
	797
[4039]	798	CASE DEFAULT
	799	unit = 'illegal'
	800
	801	END SELECT
	802
	803
	804	END SUBROUTINE doq_check_data_output
[4431]	805
[4757]	806
[4583]	807	!--------------------------------------------------------------------------------------------------!
[4757]	808	! Description:
	809	! ------------
	810	!> Set the unit of user defined profile output quantities. For those variables not recognized by the
	811	!> user, the parameter unit is set to "illegal", which tells the calling routine that the
	812	!> output variable is not defined and leads to a program abort.
	813	!--------------------------------------------------------------------------------------------------!
	814	SUBROUTINE doq_check_data_output_pr( variable, var_count, unit, dopr_unit )
	815
	816	CHARACTER (LEN=*) :: unit !<
	817	CHARACTER (LEN=*) :: variable !<
	818	CHARACTER (LEN=*) :: dopr_unit !< local value of dopr_unit
	819
	820	INTEGER(iwp) :: pr_index !<
	821	INTEGER(iwp) :: var_count !<
	822
	823	SELECT CASE ( TRIM( variable ) )
	824
	825	CASE ( 'rh' )
	826	IF ( .NOT. humidity ) THEN
	827	message_string = 'data_output_pr = ' // TRIM( data_output_pr(var_count) ) // &
	828	' requires humidity'
	829	CALL message( 'check_parameters', 'PA0358', 1, 2, 0, 6, 0 )
	830	ENDIF
	831	pr_index = 130
	832	dopr_index(var_count) = pr_index
	833	dopr_unit = '%'
	834	unit = dopr_unit
	835	hom(:,2,pr_index,:) = SPREAD( zu, 2, statistic_regions+1 )
	836
	837	CASE DEFAULT
	838	unit = 'illegal'
	839
	840	END SELECT
	841
	842
	843	END SUBROUTINE doq_check_data_output_pr
	844
	845
	846	!--------------------------------------------------------------------------------------------------!
[4039]	847	!
	848	! Description:
	849	! ------------
	850	!> Subroutine defining appropriate grid for netcdf variables.
[4583]	851	!--------------------------------------------------------------------------------------------------!
[4039]	852	SUBROUTINE doq_define_netcdf_grid( variable, found, grid_x, grid_y, grid_z )
[4431]	853
[3994]	854	IMPLICIT NONE
[4039]	855
	856	CHARACTER (LEN=*), INTENT(IN) :: variable !<
[4431]	857	CHARACTER (LEN=*), INTENT(OUT) :: grid_x !<
	858	CHARACTER (LEN=*), INTENT(OUT) :: grid_y !<
	859	CHARACTER (LEN=*), INTENT(OUT) :: grid_z !<
[4039]	860
[4583]	861	LOGICAL, INTENT(OUT) :: found !<
	862
	863
[4039]	864	found = .TRUE.
[4431]	865
[4039]	866	SELECT CASE ( TRIM( variable ) )
[3995]	867	!
[4039]	868	!-- s grid
[4757]	869	CASE ( 'rh', 'rh_xy', 'rh_xz', 'rh_yz', &
	870	'ti', 'ti_xy', 'ti_xz', 'ti_yz', &
[4583]	871	'wspeed', 'wspeed_xy', 'wspeed_xz', 'wspeed_yz', &
[4560]	872	'wdir', 'wdir_xy', 'wdir_xz', 'wdir_yz' )
[3994]	873
[4039]	874	grid_x = 'x'
	875	grid_y = 'y'
[4331]	876	grid_z = 'zu'
[4039]	877	!
[4331]	878	!-- s grid surface variables
	879	CASE ( 'theta_2m_xy', 'wspeed_10m' )
	880
	881	grid_x = 'x'
	882	grid_y = 'y'
	883	grid_z = 'zu'
	884	!
[4039]	885	!-- u grid
	886	CASE ( 'uu', 'uu_xy', 'uu_xz', 'uu_yz' )
[3994]	887
[4039]	888	grid_x = 'xu'
	889	grid_y = 'y'
	890	grid_z = 'zu'
	891	!
	892	!-- v grid
	893	CASE ( 'vv', 'vv_xy', 'vv_xz', 'vv_yz' )
	894
	895	grid_x = 'x'
	896	grid_y = 'yv'
	897	grid_z = 'zu'
	898	!
	899	!-- w grid
[4583]	900	CASE ( 'ww', 'ww_xy', 'ww_xz', 'ww_yz', &
	901	'wu', 'wu_xy', 'wu_xz', 'wu_yz', &
	902	'wv', 'wv_xy', 'wv_xz', 'wv_yz', &
	903	'wtheta', 'wtheta_xy', 'wtheta_xz', 'wtheta_yz', &
	904	'wq', 'wq_xy', 'wq_xz', 'wq_yz' )
[4039]	905
	906	grid_x = 'x'
	907	grid_y = 'y'
	908	grid_z = 'zw'
	909
	910	CASE DEFAULT
	911	found = .FALSE.
	912	grid_x = 'none'
	913	grid_y = 'none'
	914	grid_z = 'none'
	915
	916	END SELECT
	917
	918
	919	END SUBROUTINE doq_define_netcdf_grid
[4431]	920
[4583]	921	!--------------------------------------------------------------------------------------------------!
[4039]	922	!
	923	! Description:
	924	! ------------
	925	!> Subroutine defining 2D output variables
[4583]	926	!--------------------------------------------------------------------------------------------------!
	927	SUBROUTINE doq_output_2d( av, variable, found, grid, mode, local_pf, two_d, nzb_do, nzt_do, &
	928	fill_value )
[4039]	929
	930
	931	IMPLICIT NONE
	932
[4431]	933	CHARACTER (LEN=*) :: grid !<
	934	CHARACTER (LEN=*) :: mode !<
	935	CHARACTER (LEN=*) :: variable !<
[4039]	936
	937	INTEGER(iwp) :: av !< value indicating averaged or non-averaged output
	938	INTEGER(iwp) :: flag_nr !< number of the topography flag (0: scalar, 1: u, 2: v, 3: w)
	939	INTEGER(iwp) :: i !< grid index x-direction
	940	INTEGER(iwp) :: j !< grid index y-direction
	941	INTEGER(iwp) :: k !< grid index z-direction
[4431]	942	INTEGER(iwp) :: nzb_do !<
	943	INTEGER(iwp) :: nzt_do !<
[4039]	944
	945	LOGICAL :: found !< true if variable is in list
	946	LOGICAL :: resorted !< true if array is resorted
	947	LOGICAL :: two_d !< flag parameter that indicates 2D variables (horizontal cross sections)
	948
	949	REAL(wp) :: fill_value !< value for the _FillValue attribute
[4431]	950
	951	REAL(wp), DIMENSION(nxl:nxr,nys:nyn,nzb_do:nzt_do) :: local_pf !<
[4039]	952	REAL(wp), DIMENSION(:,:,:), POINTER :: to_be_resorted !< points to array which needs to be resorted for output
[4431]	953
[4583]	954
[4039]	955	flag_nr = 0
	956	found = .TRUE.
	957	resorted = .FALSE.
	958	two_d = .FALSE.
	959
	960	SELECT CASE ( TRIM( variable ) )
	961
[4757]	962	CASE ( 'rh_xy', 'rh_xz', 'rh_yz' )
	963	IF ( av == 0 ) THEN
	964	to_be_resorted => rh
	965	ELSE
	966	IF ( .NOT. ALLOCATED( rh_av ) ) THEN
	967	ALLOCATE( rh_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	968	rh_av = REAL( fill_value, KIND = wp )
	969	ENDIF
	970	to_be_resorted => rh_av
	971	ENDIF
	972	flag_nr = 0
	973
	974	IF ( mode == 'xy' ) grid = 'zu'
	975
[4039]	976	CASE ( 'ti_xy', 'ti_xz', 'ti_yz' )
	977	IF ( av == 0 ) THEN
	978	to_be_resorted => ti
	979	ELSE
[4583]	980	IF ( .NOT. ALLOCATED( ti_av ) ) THEN
[4518]	981	ALLOCATE( ti_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	982	ti_av = REAL( fill_value, KIND = wp )
	983	ENDIF
	984	to_be_resorted => ti_av
	985	ENDIF
	986	flag_nr = 0
[4431]	987
[4039]	988	IF ( mode == 'xy' ) grid = 'zu'
[4431]	989
[4039]	990	CASE ( 'uu_xy', 'uu_xz', 'uu_yz' )
	991	IF ( av == 0 ) THEN
	992	to_be_resorted => uu
	993	ELSE
[4583]	994	IF ( .NOT. ALLOCATED( uu_av ) ) THEN
[4518]	995	ALLOCATE( uu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	996	uu_av = REAL( fill_value, KIND = wp )
	997	ENDIF
	998	to_be_resorted => uu_av
	999	ENDIF
	1000	flag_nr = 1
[4431]	1001
[4039]	1002	IF ( mode == 'xy' ) grid = 'zu'
[4431]	1003
[4039]	1004	CASE ( 'vv_xy', 'vv_xz', 'vv_yz' )
	1005	IF ( av == 0 ) THEN
	1006	to_be_resorted => vv
	1007	ELSE
[4583]	1008	IF ( .NOT. ALLOCATED( vv_av ) ) THEN
[4518]	1009	ALLOCATE( vv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1010	vv_av = REAL( fill_value, KIND = wp )
	1011	ENDIF
	1012	to_be_resorted => vv_av
	1013	ENDIF
	1014	flag_nr = 2
[4431]	1015
[4039]	1016	IF ( mode == 'xy' ) grid = 'zu'
[4431]	1017
[4039]	1018	CASE ( 'ww_xy', 'ww_xz', 'ww_yz' )
	1019	IF ( av == 0 ) THEN
	1020	to_be_resorted => ww
	1021	ELSE
[4583]	1022	IF ( .NOT. ALLOCATED( ww_av ) ) THEN
[4518]	1023	ALLOCATE( ww_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1024	ww_av = REAL( fill_value, KIND = wp )
	1025	ENDIF
	1026	to_be_resorted => ww_av
	1027	ENDIF
	1028	flag_nr = 3
[4431]	1029
[4039]	1030	IF ( mode == 'xy' ) grid = 'zw'
[4431]	1031
[4351]	1032	CASE ( 'wu_xy', 'wu_xz', 'wu_yz' )
	1033	IF ( av == 0 ) THEN
	1034	to_be_resorted => wu
	1035	ELSE
[4583]	1036	IF ( .NOT. ALLOCATED( wu_av ) ) THEN
[4518]	1037	ALLOCATE( wu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1038	wu_av = REAL( fill_value, KIND = wp )
	1039	ENDIF
	1040	to_be_resorted => wu_av
	1041	ENDIF
	1042	flag_nr = 0
[4431]	1043
[4351]	1044	IF ( mode == 'xy' ) grid = 'zw'
[4431]	1045
[4351]	1046	CASE ( 'wv_xy', 'wv_xz', 'wv_yz' )
	1047	IF ( av == 0 ) THEN
	1048	to_be_resorted => wv
	1049	ELSE
[4583]	1050	IF ( .NOT. ALLOCATED( wv_av ) ) THEN
[4518]	1051	ALLOCATE( wv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1052	wv_av = REAL( fill_value, KIND = wp )
	1053	ENDIF
	1054	to_be_resorted => wv_av
	1055	ENDIF
	1056	flag_nr = 0
[4431]	1057
[4351]	1058	IF ( mode == 'xy' ) grid = 'zw'
[4431]	1059
[4351]	1060	CASE ( 'wtheta_xy', 'wtheta_xz', 'wtheta_yz' )
	1061	IF ( av == 0 ) THEN
	1062	to_be_resorted => wtheta
	1063	ELSE
[4583]	1064	IF ( .NOT. ALLOCATED( wtheta_av ) ) THEN
[4518]	1065	ALLOCATE( wtheta_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1066	wtheta_av = REAL( fill_value, KIND = wp )
	1067	ENDIF
	1068	to_be_resorted => wtheta_av
	1069	ENDIF
	1070	flag_nr = 0
[4431]	1071
[4351]	1072	IF ( mode == 'xy' ) grid = 'zw'
[4431]	1073
[4351]	1074	CASE ( 'wq_xy', 'wq_xz', 'wq_yz' )
	1075	IF ( av == 0 ) THEN
	1076	to_be_resorted => wq
	1077	ELSE
[4583]	1078	IF ( .NOT. ALLOCATED( wq_av ) ) THEN
[4518]	1079	ALLOCATE( wq_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1080	wq_av = REAL( fill_value, KIND = wp )
	1081	ENDIF
	1082	to_be_resorted => wq_av
	1083	ENDIF
	1084	flag_nr = 0
[4431]	1085
[4351]	1086	IF ( mode == 'xy' ) grid = 'zw'
[4431]	1087
[4331]	1088	CASE ( 'theta_2m*_xy' ) ! 2d-array
	1089	IF ( av == 0 ) THEN
	1090	DO i = nxl, nxr
	1091	DO j = nys, nyn
	1092	local_pf(i,j,nzb+1) = pt_2m(j,i)
	1093	ENDDO
	1094	ENDDO
	1095	ELSE
[4583]	1096	IF ( .NOT. ALLOCATED( pt_2m_av ) ) THEN
[4331]	1097	ALLOCATE( pt_2m_av(nysg:nyng,nxlg:nxrg) )
	1098	pt_2m_av = REAL( fill_value, KIND = wp )
	1099	ENDIF
	1100	DO i = nxl, nxr
	1101	DO j = nys, nyn
	1102	local_pf(i,j,nzb+1) = pt_2m_av(j,i)
	1103	ENDDO
	1104	ENDDO
	1105	ENDIF
	1106	resorted = .TRUE.
	1107	two_d = .TRUE.
	1108	grid = 'zu1'
[4431]	1109
[4331]	1110	CASE ( 'wspeed_10m*_xy' ) ! 2d-array
	1111	IF ( av == 0 ) THEN
	1112	DO i = nxl, nxr
	1113	DO j = nys, nyn
	1114	local_pf(i,j,nzb+1) = uv_10m(j,i)
	1115	ENDDO
	1116	ENDDO
	1117	ELSE
[4583]	1118	IF ( .NOT. ALLOCATED( uv_10m_av ) ) THEN
[4331]	1119	ALLOCATE( uv_10m_av(nysg:nyng,nxlg:nxrg) )
	1120	uv_10m_av = REAL( fill_value, KIND = wp )
	1121	ENDIF
	1122	DO i = nxl, nxr
	1123	DO j = nys, nyn
	1124	local_pf(i,j,nzb+1) = uv_10m_av(j,i)
	1125	ENDDO
	1126	ENDDO
	1127	ENDIF
	1128	resorted = .TRUE.
	1129	two_d = .TRUE.
	1130	grid = 'zu1'
[4039]	1131
[4431]	1132	CASE ( 'wspeed_xy', 'wspeed_xz', 'wspeed_yz' )
	1133	IF ( av == 0 ) THEN
	1134	to_be_resorted => wspeed
	1135	ELSE
[4583]	1136	IF ( .NOT. ALLOCATED( wspeed_av ) ) THEN
[4518]	1137	ALLOCATE( wspeed_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	1138	wspeed_av = REAL( fill_value, KIND = wp )
	1139	ENDIF
	1140	to_be_resorted => wspeed_av
	1141	ENDIF
	1142	flag_nr = 0
	1143
	1144	IF ( mode == 'xy' ) grid = 'zu'
	1145
	1146	CASE ( 'wdir_xy', 'wdir_xz', 'wdir_yz' )
	1147	IF ( av == 0 ) THEN
	1148	to_be_resorted => wdir
	1149	ELSE
[4583]	1150	IF ( .NOT. ALLOCATED( wdir_av ) ) THEN
[4518]	1151	ALLOCATE( wdir_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	1152	wdir_av = REAL( fill_value, KIND = wp )
	1153	ENDIF
	1154	to_be_resorted => wdir_av
	1155	ENDIF
	1156	flag_nr = 0
	1157
	1158	IF ( mode == 'xy' ) grid = 'zu'
	1159
[4039]	1160	CASE DEFAULT
	1161	found = .FALSE.
	1162	grid = 'none'
	1163
	1164	END SELECT
[4431]	1165
	1166	IF ( found .AND. .NOT. resorted ) THEN
[4039]	1167	DO i = nxl, nxr
	1168	DO j = nys, nyn
	1169	DO k = nzb_do, nzt_do
[4583]	1170	local_pf(i,j,k) = MERGE( to_be_resorted(k,j,i), &
	1171	REAL( fill_value, KIND = wp ), &
	1172	BTEST( wall_flags_total_0(k,j,i), flag_nr ) )
[4039]	1173	ENDDO
	1174	ENDDO
	1175	ENDDO
[3994]	1176	ENDIF
[4431]	1177
[4039]	1178	END SUBROUTINE doq_output_2d
[4431]	1179
	1180
[4583]	1181	!--------------------------------------------------------------------------------------------------!
[4039]	1182	!
	1183	! Description:
	1184	! ------------
	1185	!> Subroutine defining 3D output variables
[4583]	1186	!--------------------------------------------------------------------------------------------------!
	1187	SUBROUTINE doq_output_3d( av, variable, found, local_pf, fill_value, nzb_do, nzt_do )
[4431]	1188
[4039]	1189	IMPLICIT NONE
[3994]	1190
[4431]	1191	CHARACTER (LEN=*) :: variable !<
[3994]	1192
[4039]	1193	INTEGER(iwp) :: av !< index indicating averaged or instantaneous output
	1194	INTEGER(iwp) :: flag_nr !< number of the topography flag (0: scalar, 1: u, 2: v, 3: w)
	1195	INTEGER(iwp) :: i !< index variable along x-direction
	1196	INTEGER(iwp) :: j !< index variable along y-direction
	1197	INTEGER(iwp) :: k !< index variable along z-direction
	1198	INTEGER(iwp) :: nzb_do !< lower limit of the data output (usually 0)
	1199	INTEGER(iwp) :: nzt_do !< vertical upper limit of the data output (usually nz_do3d)
[3994]	1200
[4039]	1201	LOGICAL :: found !< true if variable is in list
	1202	LOGICAL :: resorted !< true if array is resorted
[3994]	1203
[4039]	1204	REAL(wp) :: fill_value !< value for the _FillValue attribute
	1205
[4768]	1206	REAL(wp), DIMENSION(nxl:nxr,nys:nyn,nzb_do:nzt_do) :: local_pf !<
[4039]	1207	REAL(wp), DIMENSION(:,:,:), POINTER :: to_be_resorted !< points to array which needs to be resorted for output
	1208
[4583]	1209
[4039]	1210	flag_nr = 0
	1211	found = .TRUE.
	1212	resorted = .FALSE.
[4431]	1213
[4039]	1214	SELECT CASE ( TRIM( variable ) )
	1215
[4757]	1216	CASE ( 'rh' )
	1217	IF ( av == 0 ) THEN
	1218	to_be_resorted => rh
	1219	ELSE
	1220	IF ( .NOT. ALLOCATED( rh ) ) THEN
	1221	ALLOCATE( rh(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	1222	rh = REAL( fill_value, KIND = wp )
	1223	ENDIF
	1224	to_be_resorted => rh
	1225	ENDIF
	1226	flag_nr = 0
	1227
[4039]	1228	CASE ( 'ti' )
	1229	IF ( av == 0 ) THEN
	1230	to_be_resorted => ti
	1231	ELSE
[4583]	1232	IF ( .NOT. ALLOCATED( ti_av ) ) THEN
[4518]	1233	ALLOCATE( ti_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1234	ti_av = REAL( fill_value, KIND = wp )
	1235	ENDIF
	1236	to_be_resorted => ti_av
	1237	ENDIF
	1238	flag_nr = 0
[4431]	1239
[4039]	1240	CASE ( 'uu' )
	1241	IF ( av == 0 ) THEN
	1242	to_be_resorted => uu
	1243	ELSE
[4583]	1244	IF ( .NOT. ALLOCATED( uu_av ) ) THEN
[4518]	1245	ALLOCATE( uu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1246	uu_av = REAL( fill_value, KIND = wp )
	1247	ENDIF
	1248	to_be_resorted => uu_av
	1249	ENDIF
	1250	flag_nr = 1
[4431]	1251
[4039]	1252	CASE ( 'vv' )
	1253	IF ( av == 0 ) THEN
	1254	to_be_resorted => vv
	1255	ELSE
[4583]	1256	IF ( .NOT. ALLOCATED( vv_av ) ) THEN
[4518]	1257	ALLOCATE( vv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1258	vv_av = REAL( fill_value, KIND = wp )
	1259	ENDIF
	1260	to_be_resorted => vv_av
	1261	ENDIF
	1262	flag_nr = 2
[4431]	1263
[4039]	1264	CASE ( 'ww' )
	1265	IF ( av == 0 ) THEN
	1266	to_be_resorted => ww
	1267	ELSE
[4583]	1268	IF ( .NOT. ALLOCATED( ww_av ) ) THEN
[4518]	1269	ALLOCATE( ww_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1270	ww_av = REAL( fill_value, KIND = wp )
	1271	ENDIF
	1272	to_be_resorted => ww_av
	1273	ENDIF
	1274	flag_nr = 3
	1275
[4351]	1276	CASE ( 'wu' )
	1277	IF ( av == 0 ) THEN
	1278	to_be_resorted => wu
	1279	ELSE
[4583]	1280	IF ( .NOT. ALLOCATED( wu_av ) ) THEN
[4518]	1281	ALLOCATE( wu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1282	wu_av = REAL( fill_value, KIND = wp )
	1283	ENDIF
	1284	to_be_resorted => wu_av
	1285	ENDIF
	1286	flag_nr = 0
	1287
	1288	CASE ( 'wv' )
	1289	IF ( av == 0 ) THEN
	1290	to_be_resorted => wv
	1291	ELSE
[4583]	1292	IF ( .NOT. ALLOCATED( wv_av ) ) THEN
[4518]	1293	ALLOCATE( wv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1294	wv_av = REAL( fill_value, KIND = wp )
	1295	ENDIF
	1296	to_be_resorted => wv_av
	1297	ENDIF
	1298	flag_nr = 0
[4431]	1299
[4351]	1300	CASE ( 'wtheta' )
	1301	IF ( av == 0 ) THEN
	1302	to_be_resorted => wtheta
	1303	ELSE
[4583]	1304	IF ( .NOT. ALLOCATED( wtheta_av ) ) THEN
[4518]	1305	ALLOCATE( wtheta_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1306	wtheta_av = REAL( fill_value, KIND = wp )
	1307	ENDIF
	1308	to_be_resorted => wtheta_av
	1309	ENDIF
	1310	flag_nr = 0
[4431]	1311
[4351]	1312	CASE ( 'wq' )
	1313	IF ( av == 0 ) THEN
	1314	to_be_resorted => wq
	1315	ELSE
[4583]	1316	IF ( .NOT. ALLOCATED( wq_av ) ) THEN
[4518]	1317	ALLOCATE( wq_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1318	wq_av = REAL( fill_value, KIND = wp )
	1319	ENDIF
	1320	to_be_resorted => wq_av
	1321	ENDIF
	1322	flag_nr = 0
	1323
[4431]	1324	CASE ( 'wspeed' )
	1325	IF ( av == 0 ) THEN
	1326	to_be_resorted => wspeed
	1327	ELSE
[4583]	1328	IF ( .NOT. ALLOCATED( wspeed_av ) ) THEN
[4518]	1329	ALLOCATE( wspeed_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	1330	wspeed_av = REAL( fill_value, KIND = wp )
	1331	ENDIF
	1332	to_be_resorted => wspeed_av
	1333	ENDIF
	1334	flag_nr = 0
	1335
	1336	CASE ( 'wdir' )
	1337	IF ( av == 0 ) THEN
	1338	to_be_resorted => wdir
	1339	ELSE
[4583]	1340	IF ( .NOT. ALLOCATED( wdir_av ) ) THEN
[4518]	1341	ALLOCATE( wdir_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	1342	wdir_av = REAL( fill_value, KIND = wp )
	1343	ENDIF
	1344	to_be_resorted => wdir_av
	1345	ENDIF
	1346	flag_nr = 0
	1347
[4039]	1348	CASE DEFAULT
	1349	found = .FALSE.
	1350
	1351	END SELECT
[4431]	1352
	1353	IF ( found .AND. .NOT. resorted ) THEN
[4039]	1354	DO i = nxl, nxr
	1355	DO j = nys, nyn
	1356	DO k = nzb_do, nzt_do
[4583]	1357	local_pf(i,j,k) = MERGE( to_be_resorted(k,j,i), &
	1358	REAL( fill_value, KIND = wp ), &
	1359	BTEST( wall_flags_total_0(k,j,i), flag_nr ) )
[4039]	1360	ENDDO
	1361	ENDDO
	1362	ENDDO
	1363	ENDIF
	1364
	1365	END SUBROUTINE doq_output_3d
[4431]	1366
[4583]	1367
	1368	!--------------------------------------------------------------------------------------------------!
	1369	!
[3994]	1370	! Description:
	1371	! ------------
[4583]	1372	!> Resorts the user-defined output quantity with indices (k,j,i) to a temporary array with indices
	1373	!> (i,j,k) for masked data output.
	1374	!--------------------------------------------------------------------------------------------------!
[4069]	1375	SUBROUTINE doq_output_mask( av, variable, found, local_pf, mid )
[4431]	1376
[4039]	1377	USE control_parameters
[4431]	1378
[4039]	1379	USE indices
[4167]	1380
[4039]	1381	IMPLICIT NONE
[3994]	1382
[4583]	1383	REAL(wp), PARAMETER :: fill_value = -9999.0_wp !< value for the _FillValue attribute
	1384
[4167]	1385	CHARACTER (LEN=*) :: variable !<
	1386	CHARACTER (LEN=5) :: grid !< flag to distinquish between staggered grids
[3994]	1387
[4167]	1388	INTEGER(iwp) :: av !< index indicating averaged or instantaneous output
	1389	INTEGER(iwp) :: flag_nr !< number of the topography flag (0: scalar, 1: u, 2: v, 3: w)
	1390	INTEGER(iwp) :: i !< index variable along x-direction
	1391	INTEGER(iwp) :: j !< index variable along y-direction
	1392	INTEGER(iwp) :: k !< index variable along z-direction
	1393	INTEGER(iwp) :: im !< loop index for masked variables
	1394	INTEGER(iwp) :: jm !< loop index for masked variables
	1395	INTEGER(iwp) :: kk !< masked output index variable along z-direction
	1396	INTEGER(iwp) :: mid !< masked output running index
	1397	INTEGER(iwp) :: ktt !< k index of highest horizontal surface
[3994]	1398
[4167]	1399	LOGICAL :: found !< true if variable is in list
	1400	LOGICAL :: resorted !< true if array is resorted
[3994]	1401
[4583]	1402	REAL(wp), DIMENSION(mask_size_l(mid,1),mask_size_l(mid,2),mask_size_l(mid,3)) :: local_pf !<
[3994]	1403
[4583]	1404	REAL(wp), DIMENSION(:,:,:), POINTER :: to_be_resorted !< points to array which needs to be resorted for output
[4167]	1405
[4583]	1406
[4039]	1407	flag_nr = 0
	1408	found = .TRUE.
	1409	resorted = .FALSE.
	1410	grid = 's'
	1411
	1412	SELECT CASE ( TRIM( variable ) )
	1413
[4757]	1414	CASE ( 'rh' )
	1415	IF ( av == 0 ) THEN
	1416	to_be_resorted => rh
	1417	ELSE
	1418	to_be_resorted => rh_av
	1419	ENDIF
	1420	grid = 's'
	1421	flag_nr = 0
	1422
[4039]	1423	CASE ( 'ti' )
	1424	IF ( av == 0 ) THEN
	1425	to_be_resorted => ti
	1426	ELSE
	1427	to_be_resorted => ti_av
	1428	ENDIF
	1429	grid = 's'
	1430	flag_nr = 0
[4431]	1431
[4039]	1432	CASE ( 'uu' )
	1433	IF ( av == 0 ) THEN
	1434	to_be_resorted => uu
	1435	ELSE
	1436	to_be_resorted => uu_av
	1437	ENDIF
	1438	grid = 'u'
	1439	flag_nr = 1
[4431]	1440
[4039]	1441	CASE ( 'vv' )
	1442	IF ( av == 0 ) THEN
	1443	to_be_resorted => vv
	1444	ELSE
	1445	to_be_resorted => vv_av
	1446	ENDIF
	1447	grid = 'v'
	1448	flag_nr = 2
[4431]	1449
[4039]	1450	CASE ( 'ww' )
	1451	IF ( av == 0 ) THEN
	1452	to_be_resorted => ww
	1453	ELSE
	1454	to_be_resorted => ww_av
	1455	ENDIF
	1456	grid = 'w'
	1457	flag_nr = 3
[4431]	1458
[4351]	1459	CASE ( 'wu' )
	1460	IF ( av == 0 ) THEN
	1461	to_be_resorted => wu
	1462	ELSE
	1463	to_be_resorted => wu_av
	1464	ENDIF
	1465	grid = 's'
	1466	flag_nr = 0
[4431]	1467
[4351]	1468	CASE ( 'wv' )
	1469	IF ( av == 0 ) THEN
	1470	to_be_resorted => wv
	1471	ELSE
	1472	to_be_resorted => wv_av
	1473	ENDIF
	1474	grid = 's'
	1475	flag_nr = 0
[4431]	1476
[4351]	1477	CASE ( 'wtheta' )
	1478	IF ( av == 0 ) THEN
	1479	to_be_resorted => wtheta
	1480	ELSE
	1481	to_be_resorted => wtheta_av
	1482	ENDIF
	1483	grid = 's'
	1484	flag_nr = 0
[4431]	1485
[4351]	1486	CASE ( 'wq' )
	1487	IF ( av == 0 ) THEN
	1488	to_be_resorted => wq
	1489	ELSE
	1490	to_be_resorted => wq_av
	1491	ENDIF
	1492	grid = 's'
	1493	flag_nr = 0
[4039]	1494
[4431]	1495	CASE ( 'wspeed' )
	1496	IF ( av == 0 ) THEN
	1497	to_be_resorted => wspeed
	1498	ELSE
	1499	to_be_resorted => wspeed_av
	1500	ENDIF
	1501	grid = 's'
	1502	flag_nr = 0
	1503
	1504	CASE ( 'wdir' )
	1505	IF ( av == 0 ) THEN
	1506	to_be_resorted => wdir
	1507	ELSE
	1508	to_be_resorted => wdir_av
	1509	ENDIF
	1510	grid = 's'
	1511	flag_nr = 0
	1512
[4039]	1513	CASE DEFAULT
	1514	found = .FALSE.
	1515
	1516	END SELECT
[4431]	1517
[4039]	1518	IF ( found .AND. .NOT. resorted ) THEN
	1519	IF ( .NOT. mask_surface(mid) ) THEN
	1520	!
	1521	!-- Default masked output
	1522	DO i = 1, mask_size_l(mid,1)
	1523	DO j = 1, mask_size_l(mid,2)
	1524	DO k = 1, mask_size_l(mid,3)
[4583]	1525	local_pf(i,j,k) = MERGE( to_be_resorted(mask_k(mid,k), &
	1526	mask_j(mid,j), &
	1527	mask_i(mid,i)), &
	1528	REAL( fill_value, KIND = wp ), &
	1529	BTEST( wall_flags_total_0(mask_k(mid,k), &
	1530	mask_j(mid,j), &
	1531	mask_i(mid,i)), &
[4431]	1532	flag_nr ) )
[4039]	1533	ENDDO
	1534	ENDDO
	1535	ENDDO
	1536
	1537	ELSE
	1538	!
	1539	!-- Terrain-following masked output
	1540	DO i = 1, mask_size_l(mid,1)
	1541	DO j = 1, mask_size_l(mid,2)
	1542	!
[4167]	1543	!-- Get k index of the highest terraing surface
	1544	im = mask_i(mid,i)
	1545	jm = mask_j(mid,j)
[4583]	1546	ktt = MINLOC( MERGE( 1, 0, BTEST( wall_flags_total_0(:,jm,im), 5 ) ), DIM=1 ) - 1
[4167]	1547	DO k = 1, mask_size_l(mid,3)
	1548	kk = MIN( ktt+mask_k(mid,k), nzt+1 )
[4039]	1549	!
[4167]	1550	!-- Set value if not in building
[4346]	1551	IF ( BTEST( wall_flags_total_0(kk,jm,im), 6 ) ) THEN
[4167]	1552	local_pf(i,j,k) = fill_value
	1553	ELSE
	1554	local_pf(i,j,k) = to_be_resorted(kk,jm,im)
	1555	ENDIF
[4039]	1556	ENDDO
	1557	ENDDO
	1558	ENDDO
	1559
	1560	ENDIF
	1561	ENDIF
[4431]	1562
[4039]	1563	END SUBROUTINE doq_output_mask
	1564
[4583]	1565	!--------------------------------------------------------------------------------------------------!
[4039]	1566	! Description:
	1567	! ------------
	1568	!> Allocate required arrays
[4583]	1569	!--------------------------------------------------------------------------------------------------!
[4039]	1570	SUBROUTINE doq_init
	1571
[3994]	1572	IMPLICIT NONE
[4431]	1573
[4039]	1574	INTEGER(iwp) :: ivar !< loop index over all 2d/3d/mask output quantities
[4157]	1575
[4583]	1576
[4039]	1577	!
	1578	!-- Next line is to avoid compiler warnings about unused variables
	1579	IF ( timestep_number_at_prev_calc == 0 ) CONTINUE
[4157]	1580	!
	1581	!-- Preparatory steps and initialization of output arrays
	1582	IF ( .NOT. prepared_diagnostic_output_quantities ) CALL doq_prepare
[3994]	1583
[4039]	1584	initialized_diagnostic_output_quantities = .FALSE.
[4431]	1585
[4039]	1586	ivar = 1
	1587
[4431]	1588	DO WHILE ( ivar <= SIZE( do_all ) )
	1589
[4039]	1590	SELECT CASE ( TRIM( do_all(ivar) ) )
	1591	!
[4757]	1592	!-- Allocate array for 'relative humidity'
	1593	CASE ( 'rh' )
	1594	IF ( .NOT. ALLOCATED( rh ) ) THEN
	1595	ALLOCATE( rh(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	1596	rh = 0.0_wp
	1597	ENDIF
	1598	!
[4039]	1599	!-- Allocate array for 'turbulence intensity'
	1600	CASE ( 'ti' )
	1601	IF ( .NOT. ALLOCATED( ti ) ) THEN
[4518]	1602	ALLOCATE( ti(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1603	ti = 0.0_wp
	1604	ENDIF
	1605	!
	1606	!-- Allocate array for uu
	1607	CASE ( 'uu' )
	1608	IF ( .NOT. ALLOCATED( uu ) ) THEN
[4518]	1609	ALLOCATE( uu(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1610	uu = 0.0_wp
	1611	ENDIF
	1612	!
	1613	!-- Allocate array for vv
	1614	CASE ( 'vv' )
	1615	IF ( .NOT. ALLOCATED( vv ) ) THEN
[4518]	1616	ALLOCATE( vv(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1617	vv = 0.0_wp
	1618	ENDIF
	1619	!
	1620	!-- Allocate array for ww
	1621	CASE ( 'ww' )
	1622	IF ( .NOT. ALLOCATED( ww ) ) THEN
[4518]	1623	ALLOCATE( ww(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4039]	1624	ww = 0.0_wp
	1625	ENDIF
[4331]	1626	!
[4351]	1627	!-- Allocate array for wu
	1628	CASE ( 'wu' )
	1629	IF ( .NOT. ALLOCATED( wu ) ) THEN
[4518]	1630	ALLOCATE( wu(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1631	wu = 0.0_wp
	1632	ENDIF
	1633	!
	1634	!-- Allocate array for wv
	1635	CASE ( 'wv' )
	1636	IF ( .NOT. ALLOCATED( wv ) ) THEN
[4518]	1637	ALLOCATE( wv(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1638	wv = 0.0_wp
	1639	ENDIF
	1640	!
	1641	!-- Allocate array for wtheta
	1642	CASE ( 'wtheta' )
	1643	IF ( .NOT. ALLOCATED( wtheta ) ) THEN
[4518]	1644	ALLOCATE( wtheta(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1645	wtheta = 0.0_wp
	1646	ENDIF
	1647	!
	1648	!-- Allocate array for wq
	1649	CASE ( 'wq' )
	1650	IF ( .NOT. ALLOCATED( wq ) ) THEN
[4518]	1651	ALLOCATE( wq(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4351]	1652	wq = 0.0_wp
	1653	ENDIF
	1654	!
[4331]	1655	!-- Allocate array for 2-m potential temperature
	1656	CASE ( 'theta_2m*' )
	1657	IF ( .NOT. ALLOCATED( pt_2m ) ) THEN
[4518]	1658	ALLOCATE( pt_2m(nysg:nyng,nxlg:nxrg) )
[4331]	1659	pt_2m = 0.0_wp
	1660	ENDIF
	1661	!
	1662	!-- Allocate array for 10-m wind speed
	1663	CASE ( 'wspeed_10m*' )
	1664	IF ( .NOT. ALLOCATED( uv_10m ) ) THEN
[4518]	1665	ALLOCATE( uv_10m(nysg:nyng,nxlg:nxrg) )
[4331]	1666	uv_10m = 0.0_wp
	1667	ENDIF
[4431]	1668	!
	1669	!-- Allocate array for wspeed
	1670	CASE ( 'wspeed' )
	1671	IF ( .NOT. ALLOCATED( wspeed ) ) THEN
[4518]	1672	ALLOCATE( wspeed(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	1673	wspeed = 0.0_wp
	1674	ENDIF
[4039]	1675
[4431]	1676	!
	1677	!-- Allocate array for wdir
	1678	CASE ( 'wdir' )
	1679	IF ( .NOT. ALLOCATED( u_center ) ) THEN
[4518]	1680	ALLOCATE( u_center(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	1681	u_center = 0.0_wp
	1682	ENDIF
	1683	IF ( .NOT. ALLOCATED( v_center ) ) THEN
[4518]	1684	ALLOCATE( v_center(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	1685	v_center = 0.0_wp
	1686	ENDIF
	1687	IF ( .NOT. ALLOCATED( wdir ) ) THEN
[4518]	1688	ALLOCATE( wdir(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
[4431]	1689	wdir = 0.0_wp
	1690	ENDIF
	1691
[4039]	1692	END SELECT
	1693
	1694	ivar = ivar + 1
	1695	ENDDO
	1696
	1697	initialized_diagnostic_output_quantities = .TRUE.
[4431]	1698
[4039]	1699	END SUBROUTINE doq_init
	1700
	1701
	1702	!--------------------------------------------------------------------------------------------------!
	1703	! Description:
	1704	! ------------
[4757]	1705	!> Calculation of module-specific statistics, i.e. horizontally averaged profiles and time series.
	1706	!> This is called for every statistic region sr, but at least for the region "total domain" (sr=0).
	1707	!--------------------------------------------------------------------------------------------------!
	1708	SUBROUTINE doq_statistics( mode, sr, tn )
	1709
	1710
	1711	CHARACTER (LEN=*) :: mode !<
	1712
	1713	! INTEGER(iwp) :: i !<
	1714	! INTEGER(iwp) :: j !<
	1715	! INTEGER(iwp) :: k !<
	1716	INTEGER(iwp) :: sr !<
	1717	INTEGER(iwp) :: tn !<
	1718	!
	1719	!-- Next line is to avoid compiler warning about unused variables. Please remove.
	1720	IF ( sr == 0 .OR. tn == 0 ) CONTINUE
	1721
	1722	IF ( mode == 'profiles' ) THEN
	1723
	1724	ELSEIF ( mode == 'time_series' ) THEN
	1725
	1726	ENDIF
	1727
	1728	END SUBROUTINE doq_statistics
	1729
	1730
	1731	!--------------------------------------------------------------------------------------------------!
	1732	! Description:
	1733	! ------------
[4039]	1734	!> Calculate diagnostic quantities
	1735	!--------------------------------------------------------------------------------------------------!
	1736	SUBROUTINE doq_calculate
	1737
	1738	IMPLICIT NONE
	1739
[4331]	1740	INTEGER(iwp) :: i !< grid index x-dimension
[4431]	1741	INTEGER(iwp) :: j !< grid index y-dimension
[4331]	1742	INTEGER(iwp) :: k !< grid index z-dimension
[4039]	1743	INTEGER(iwp) :: ivar !< loop index over all 2d/3d/mask output quantities
[4431]	1744
[4757]	1745	REAL(wp) :: temp !< temperature
	1746	REAL(wp) :: e_s !< saturation vapor pressure
	1747	REAL(wp) :: q_s !< saturation mixing ratio
	1748
[4331]	1749	TYPE(surf_type), POINTER :: surf !< surf-type array, used to generalize subroutines
[3994]	1750
	1751
	1752	! CALL cpu_log( log_point(41), 'calculate_quantities', 'start' )
[4157]	1753
[3994]	1754	!
[4583]	1755	!-- Save timestep number to check in time_integration if doq_calculate has been called already,
	1756	!-- since the CALL occurs at two locations, but the calculations need to be done only once per
	1757	!-- timestep.
[3994]	1758	timestep_number_at_prev_calc = current_timestep_number
	1759
[4039]	1760	ivar = 1
[3994]	1761
[4431]	1762	DO WHILE ( ivar <= SIZE( do_all ) )
[3994]	1763
[4039]	1764	SELECT CASE ( TRIM( do_all(ivar) ) )
[3994]	1765	!
[4757]	1766	!-- rh (relative humidity)
	1767	CASE ( 'rh' )
	1768	DO i = nxl, nxr
	1769	DO j = nys, nyn
	1770	DO k = nzb+1, nzt
	1771	IF ( humidity ) THEN
	1772	temp = exner(k) * pt(k,j,i)
	1773	e_s = magnus( temp )
	1774	q_s = rd_d_rv * e_s / ( hyp(k) - e_s )
	1775	rh(k,j,i) = q(k,j,i) / q_s * 100.0_wp &
	1776	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 1 ) )
	1777	ENDIF
	1778	ENDDO
	1779	ENDDO
	1780	ENDDO
	1781	!
[3994]	1782	!-- Calculate 'turbulence intensity' from rot[(u,v,w)] at scalar grid point
	1783	CASE ( 'ti' )
	1784	DO i = nxl, nxr
	1785	DO j = nys, nyn
	1786	DO k = nzb+1, nzt
[4583]	1787	ti(k,j,i) = 0.25_wp * SQRT( &
	1788	( ( w(k,j+1,i) + w(k-1,j+1,i) &
	1789	- w(k,j-1,i) - w(k-1,j-1,i) ) * ddy &
	1790	- ( v(k+1,j,i) + v(k+1,j+1,i) &
	1791	- v(k-1,j,i) - v(k-1,j+1,i) ) * ddzu(k) )**2 &
	1792	+ ( ( u(k+1,j,i) + u(k+1,j,i+1) &
	1793	- u(k-1,j,i) - u(k-1,j,i+1) ) * ddzu(k) &
	1794	- ( w(k,j,i+1) + w(k-1,j,i+1) &
	1795	- w(k,j,i-1) - w(k-1,j,i-1) ) * ddx )**2 &
	1796	+ ( ( v(k,j,i+1) + v(k,j+1,i+1) &
	1797	- v(k,j,i-1) - v(k,j+1,i-1) ) * ddx &
	1798	- ( u(k,j+1,i) + u(k,j+1,i+1) &
	1799	- u(k,j-1,i) - u(k,j-1,i+1) ) * ddy )**2 ) &
	1800	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 0 ) )
[3994]	1801	ENDDO
	1802	ENDDO
[4431]	1803	ENDDO
[4039]	1804	!
	1805	!-- uu
	1806	CASE ( 'uu' )
	1807	DO i = nxl, nxr
	1808	DO j = nys, nyn
	1809	DO k = nzb+1, nzt
[4583]	1810	uu(k,j,i) = u(k,j,i) * u(k,j,i) &
	1811	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 1 ) )
[4039]	1812	ENDDO
	1813	ENDDO
[3994]	1814	ENDDO
[4039]	1815	!
	1816	!-- vv
	1817	CASE ( 'vv' )
	1818	DO i = nxl, nxr
	1819	DO j = nys, nyn
	1820	DO k = nzb+1, nzt
[4583]	1821	vv(k,j,i) = v(k,j,i) * v(k,j,i) &
	1822	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 2 ) )
[4039]	1823	ENDDO
	1824	ENDDO
	1825	ENDDO
	1826	!
	1827	!-- ww
	1828	CASE ( 'ww' )
	1829	DO i = nxl, nxr
	1830	DO j = nys, nyn
	1831	DO k = nzb+1, nzt-1
[4583]	1832	ww(k,j,i) = w(k,j,i) * w(k,j,i) &
	1833	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 3 ) )
[4039]	1834	ENDDO
	1835	ENDDO
	1836	ENDDO
[4331]	1837	!
[4351]	1838	!-- wu
	1839	CASE ( 'wu' )
	1840	DO i = nxl, nxr
	1841	DO j = nys, nyn
	1842	DO k = nzb+1, nzt-1
[4583]	1843	wu(k,j,i) = w(k,j,i) &
	1844	* 0.25_wp * ( u(k,j,i) + u(k,j,i+1) + u(k+1,j,i) + u(k+1,j,i+1) )&
	1845	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 0 ) )
[4351]	1846	ENDDO
	1847	ENDDO
	1848	ENDDO
	1849	!
	1850	!-- wv
	1851	CASE ( 'wv' )
	1852	DO i = nxl, nxr
	1853	DO j = nys, nyn
	1854	DO k = nzb+1, nzt-1
[4583]	1855	wv(k,j,i) = w(k,j,i) &
	1856	* 0.25_wp * ( v(k,j,i) + v(k,j+1,i) + v(k+1,j,i) + v(k+1,j+1,i) )&
	1857	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 0 ) )
[4351]	1858	ENDDO
	1859	ENDDO
	1860	ENDDO
	1861	!
	1862	!-- wtheta
	1863	CASE ( 'wtheta' )
	1864	DO i = nxl, nxr
	1865	DO j = nys, nyn
	1866	DO k = nzb+1, nzt-1
[4583]	1867	wtheta(k,j,i) = w(k,j,i) &
	1868	* 0.5_wp * ( pt(k,j,i) + pt(k+1,j,i) ) &
	1869	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 3 ))
[4351]	1870	ENDDO
	1871	ENDDO
	1872	ENDDO
	1873	!
	1874	!-- wq
	1875	CASE ( 'wq' )
	1876	DO i = nxl, nxr
	1877	DO j = nys, nyn
	1878	DO k = nzb+1, nzt-1
[4583]	1879	wq(k,j,i) = w(k,j,i) * 0.5_wp * ( q(k,j,i) + q(k+1,j,i) ) &
	1880	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 3 ) )
[4351]	1881	ENDDO
	1882	ENDDO
	1883	ENDDO
	1884	!
[4331]	1885	!-- 2-m potential temperature
	1886	CASE ( 'theta_2m*' )
	1887	!
[4583]	1888	!-- 2-m potential temperature is caluclated from surface arrays. In case the 2m level is
	1889	!-- below the first grid point, MOST is applied, else, linear interpolation between two
	1890	!-- vertical grid levels is applied. To access all surfaces, iterate over all horizontally-
[4331]	1891	!-- upward facing surface types.
	1892	surf => surf_def_h(0)
	1893	CALL calc_pt_2m
[4671]	1894	surf => surf_lsm_h(0)
[4331]	1895	CALL calc_pt_2m
[4671]	1896	surf => surf_usm_h(0)
[4331]	1897	CALL calc_pt_2m
	1898	!
	1899	!-- 10-m wind speed
	1900	CASE ( 'wspeed_10m*' )
	1901	!
[4583]	1902	!-- 10-m wind speed is caluclated from surface arrays. In case the 10m level is below the
	1903	!-- first grid point, MOST is applied, else, linear interpolation between two vertical grid
	1904	!-- levels is applied. To access all surfaces, iterate over all horizontally-upward facing
	1905	!-- surface types.
[4331]	1906	surf => surf_def_h(0)
	1907	CALL calc_wind_10m
[4671]	1908	surf => surf_lsm_h(0)
[4331]	1909	CALL calc_wind_10m
[4671]	1910	surf => surf_usm_h(0)
[4331]	1911	CALL calc_wind_10m
[4431]	1912	!
[4583]	1913	!-- Horizontal wind speed
[4431]	1914	CASE ( 'wspeed' )
	1915	DO i = nxl, nxr
	1916	DO j = nys, nyn
	1917	DO k = nzb, nzt+1
[4583]	1918	wspeed(k,j,i) = SQRT( ( 0.5_wp * ( u(k,j,i) + u(k,j,i+1) ) )**2 &
	1919	+ ( 0.5_wp * ( v(k,j,i) + v(k,j+1,i) ) )**2 ) &
	1920	* MERGE( 1.0_wp, 0.0_wp, BTEST( wall_flags_total_0(k,j,i), 0 ))
[4431]	1921	ENDDO
	1922	ENDDO
	1923	ENDDO
[3994]	1924
[4431]	1925	!
[4583]	1926	!-- Horizontal wind direction
[4431]	1927	CASE ( 'wdir' )
	1928	DO i = nxl, nxr
	1929	DO j = nys, nyn
	1930	DO k = nzb, nzt+1
	1931	u_center(k,j,i) = 0.5_wp * ( u(k,j,i) + u(k,j,i+1) )
	1932	v_center(k,j,i) = 0.5_wp * ( v(k,j,i) + v(k,j+1,i) )
	1933
[4583]	1934	wdir(k,j,i) = ATAN2( u_center(k,j,i), v_center(k,j,i) ) &
	1935	/ pi * 180.0_wp + 180.0_wp
[4431]	1936	ENDDO
	1937	ENDDO
	1938	ENDDO
	1939
[4039]	1940	END SELECT
[3994]	1941
[4039]	1942	ivar = ivar + 1
[3994]	1943	ENDDO
	1944
	1945	! CALL cpu_log( log_point(41), 'calculate_quantities', 'stop' )
	1946
[4331]	1947	!
[4583]	1948	!-- The following block contains subroutines to calculate diagnostic surface quantities.
[4331]	1949	CONTAINS
[4583]	1950	!--------------------------------------------------------------------------------------------------!
[4331]	1951	! Description:
	1952	! ------------
	1953	!> Calculation of 2-m potential temperature.
[4583]	1954	!--------------------------------------------------------------------------------------------------!
[4331]	1955	SUBROUTINE calc_pt_2m
	1956
[4583]	1957	USE surface_layer_fluxes_mod, &
[4331]	1958	ONLY: psi_h
	1959
	1960	IMPLICIT NONE
	1961
	1962	INTEGER(iwp) :: kk !< running index along the z-dimension
	1963	INTEGER(iwp) :: m !< running index for surface elements
[4431]	1964
[4331]	1965	DO m = 1, surf%ns
	1966
	1967	i = surf%i(m)
	1968	j = surf%j(m)
	1969	k = surf%k(m)
	1970	!
[4583]	1971	!-- If 2-m level is below the first grid level, MOST is used for calculation of
	1972	!-- 2-m temperature.
[4331]	1973	IF ( surf%z_mo(m) > 2.0_wp ) THEN
[4583]	1974	pt_2m(j,i) = surf%pt_surface(m) + surf%ts(m) / kappa &
	1975	* ( LOG( 2.0_wp / surf%z0h(m) ) &
	1976	- psi_h( 2.0_wp / surf%ol(m) ) &
	1977	+ psi_h( surf%z0h(m) / surf%ol(m) ) )
[4331]	1978	!
[4583]	1979	!-- If 2-m level is above the first grid level, 2-m temperature is linearly interpolated
	1980	!-- between the two nearest vertical grid levels. Note, since 2-m temperature is only
	1981	!-- computed for horizontal upward-facing surfaces, only a vertical interpolation is
	1982	!-- necessary.
[4331]	1983	ELSE
	1984	!
[4431]	1985	!-- zw(k-1) defines the height of the surface.
[4331]	1986	kk = k
	1987	DO WHILE ( zu(kk) - zw(k-1) < 2.0_wp .AND. kk <= nzt )
[4431]	1988	kk = kk + 1
[4331]	1989	ENDDO
	1990	!
[4431]	1991	!-- kk defines the index of the first grid level >= 2m.
[4583]	1992	pt_2m(j,i) = pt(kk-1,j,i) + &
	1993	( zw(k-1) + 2.0_wp - zu(kk-1) ) * &
	1994	( pt(kk,j,i) - pt(kk-1,j,i) ) / &
[4331]	1995	( zu(kk) - zu(kk-1) )
	1996	ENDIF
	1997
	1998	ENDDO
	1999
	2000	END SUBROUTINE calc_pt_2m
[4431]	2001
[4583]	2002	!--------------------------------------------------------------------------------------------------!
[4331]	2003	! Description:
	2004	! ------------
	2005	!> Calculation of 10-m wind speed.
[4583]	2006	!--------------------------------------------------------------------------------------------------!
[4331]	2007	SUBROUTINE calc_wind_10m
	2008
[4583]	2009	USE surface_layer_fluxes_mod, &
[4331]	2010	ONLY: psi_m
	2011
	2012	IMPLICIT NONE
	2013
	2014	INTEGER(iwp) :: kk !< running index along the z-dimension
	2015	INTEGER(iwp) :: m !< running index for surface elements
	2016
	2017	REAL(wp) :: uv_l !< wind speed at lower grid point
	2018	REAL(wp) :: uv_u !< wind speed at upper grid point
[4431]	2019
[4583]	2020
[4331]	2021	DO m = 1, surf%ns
	2022
	2023	i = surf%i(m)
	2024	j = surf%j(m)
	2025	k = surf%k(m)
	2026	!
[4583]	2027	!-- If 10-m level is below the first grid level, MOST is used for calculation of 10-m
	2028	!-- temperature.
[4331]	2029	IF ( surf%z_mo(m) > 10.0_wp ) THEN
[4583]	2030	uv_10m(j,i) = surf%us(m) / kappa &
	2031	* ( LOG( 10.0_wp / surf%z0(m) ) &
	2032	- psi_m( 10.0_wp / surf%ol(m) ) &
	2033	+ psi_m( surf%z0(m) / surf%ol(m) ) )
[4331]	2034	!
[4583]	2035	!-- If 10-m level is above the first grid level, 10-m wind speed is linearly interpolated
	2036	!-- between the two nearest vertical grid levels. Note, since 10-m temperature is only
	2037	!-- computed for horizontal upward-facing surfaces, only a vertical interpolation is
	2038	!-- necessary.
[4331]	2039	ELSE
	2040	!
[4431]	2041	!-- zw(k-1) defines the height of the surface.
[4331]	2042	kk = k
	2043	DO WHILE ( zu(kk) - zw(k-1) < 10.0_wp .AND. kk <= nzt )
[4431]	2044	kk = kk + 1
[4331]	2045	ENDDO
	2046	!
	2047	!-- kk defines the index of the first grid level >= 10m.
[4583]	2048	uv_l = SQRT( ( 0.5_wp * ( u(kk-1,j,i) + u(kk-1,j,i+1) ) )**2 &
[4331]	2049	+ ( 0.5_wp * ( v(kk-1,j,i) + v(kk-1,j+1,i) ) )**2 )
	2050
[4583]	2051	uv_u = SQRT( ( 0.5_wp * ( u(kk,j,i) + u(kk,j,i+1) ) )**2 &
[4331]	2052	+ ( 0.5_wp * ( v(kk,j,i) + v(kk,j+1,i) ) )**2 )
	2053
[4583]	2054	uv_10m(j,i) = uv_l + ( zw(k-1) + 10.0_wp - zu(kk-1) ) * &
	2055	( uv_u - uv_l ) / &
[4331]	2056	( zu(kk) - zu(kk-1) )
	2057
	2058	ENDIF
	2059
	2060	ENDDO
	2061
	2062	END SUBROUTINE calc_wind_10m
	2063
[4039]	2064	END SUBROUTINE doq_calculate
[3994]	2065
	2066
[4583]	2067	!--------------------------------------------------------------------------------------------------!
[3994]	2068	! Description:
	2069	! ------------
[4583]	2070	!> Preparation of the diagnostic output, counting of the module-specific output quantities and
	2071	!> gathering of the output names.
	2072	!--------------------------------------------------------------------------------------------------!
[4039]	2073	SUBROUTINE doq_prepare
[3994]	2074
[4583]	2075	USE control_parameters, &
[4069]	2076	ONLY: do2d, do3d, domask, masks
[3994]	2077
	2078	IMPLICIT NONE
	2079
[4583]	2080	CHARACTER (LEN=varnamelength), DIMENSION(0:1,500) :: do2d_var = ' ' !< label array for 2d output quantities
[3994]	2081
	2082	INTEGER(iwp) :: av !< index defining type of output, av=0 instantaneous, av=1 averaged
	2083	INTEGER(iwp) :: ivar !< loop index
	2084	INTEGER(iwp) :: ivar_all !< loop index
	2085	INTEGER(iwp) :: l !< index for cutting string
[4431]	2086	INTEGER(iwp) :: mid !< masked output running index
[3994]	2087
[4583]	2088
[3994]	2089	prepared_diagnostic_output_quantities = .FALSE.
	2090
	2091	ivar = 1
	2092	ivar_all = 1
	2093
	2094	DO av = 0, 1
	2095	!
	2096	!-- Remove _xy, _xz, or _yz from string
	2097	l = MAX( 3, LEN_TRIM( do2d(av,ivar) ) )
[3998]	2098	do2d_var(av,ivar)(1:l-3) = do2d(av,ivar)(1:l-3)
[3994]	2099	!
[4039]	2100	!-- Gather 2d output quantity names.
[4583]	2101	!-- Check for double occurrence of output quantity, e.g. by _xy, _yz, _xz.
[3994]	2102	DO WHILE ( do2d_var(av,ivar)(1:1) /= ' ' )
	2103	IF ( .NOT. ANY( do_all == do2d_var(av,ivar) ) ) THEN
	2104	do_all(ivar_all) = do2d_var(av,ivar)
	2105	ENDIF
	2106	ivar = ivar + 1
	2107	ivar_all = ivar_all + 1
	2108	l = MAX( 3, LEN_TRIM( do2d(av,ivar) ) )
[3998]	2109	do2d_var(av,ivar)(1:l-3) = do2d(av,ivar)(1:l-3)
[3994]	2110	ENDDO
	2111
	2112	ivar = 1
	2113	!
[4039]	2114	!-- Gather 3d output quantity names
[3994]	2115	DO WHILE ( do3d(av,ivar)(1:1) /= ' ' )
	2116	do_all(ivar_all) = do3d(av,ivar)
	2117	ivar = ivar + 1
	2118	ivar_all = ivar_all + 1
	2119	ENDDO
	2120
	2121	ivar = 1
	2122	!
[4583]	2123	!-- Gather masked output quantity names. Also check for double output e.g. by different masks.
[3994]	2124	DO mid = 1, masks
	2125	DO WHILE ( domask(mid,av,ivar)(1:1) /= ' ' )
[4039]	2126	IF ( .NOT. ANY( do_all == domask(mid,av,ivar) ) ) THEN
[4132]	2127	do_all(ivar_all) = domask(mid,av,ivar)
[4039]	2128	ENDIF
[3994]	2129
	2130	ivar = ivar + 1
	2131	ivar_all = ivar_all + 1
	2132	ENDDO
	2133	ivar = 1
	2134	ENDDO
	2135
	2136	ENDDO
	2137
	2138	prepared_diagnostic_output_quantities = .TRUE.
	2139
[4039]	2140	END SUBROUTINE doq_prepare
[4431]	2141
[4583]	2142	!--------------------------------------------------------------------------------------------------!
[4039]	2143	! Description:
	2144	! ------------
	2145	!> Subroutine reads local (subdomain) restart data
[4583]	2146	!--------------------------------------------------------------------------------------------------!
	2147	SUBROUTINE doq_rrd_local_ftn( k, nxlf, nxlc, nxl_on_file, nxrf, nxrc, nxr_on_file, nynf, nync, &
	2148	nyn_on_file, nysf, nysc, nys_on_file, tmp_2d, tmp_3d, found )
[4431]	2149
[4518]	2150	USE control_parameters
	2151
	2152	USE indices
	2153
	2154	USE kinds
	2155
	2156	USE pegrid
	2157
	2158
	2159	IMPLICIT NONE
	2160
	2161	INTEGER(iwp) :: k !<
	2162	INTEGER(iwp) :: nxlc !<
	2163	INTEGER(iwp) :: nxlf !<
	2164	INTEGER(iwp) :: nxl_on_file !<
	2165	INTEGER(iwp) :: nxrc !<
	2166	INTEGER(iwp) :: nxrf !<
	2167	INTEGER(iwp) :: nxr_on_file !<
	2168	INTEGER(iwp) :: nync !<
	2169	INTEGER(iwp) :: nynf !<
	2170	INTEGER(iwp) :: nyn_on_file !<
	2171	INTEGER(iwp) :: nysc !<
	2172	INTEGER(iwp) :: nysf !<
	2173	INTEGER(iwp) :: nys_on_file !<
	2174
	2175	LOGICAL, INTENT(OUT) :: found
	2176
	2177	REAL(wp), DIMENSION(nys_on_file-nbgp:nyn_on_file+nbgp,nxl_on_file-nbgp:nxr_on_file+nbgp) :: tmp_2d !<
	2178
	2179	REAL(wp), DIMENSION(nzb:nzt+1,nys_on_file-nbgp:nyn_on_file+nbgp,nxl_on_file-nbgp:nxr_on_file+nbgp) :: tmp_3d !<
	2180
	2181
	2182	found = .TRUE.
	2183
	2184	SELECT CASE ( restart_string(1:length) )
	2185
	2186	CASE ( 'pt_2m_av' )
	2187	IF ( .NOT. ALLOCATED( pt_2m_av ) ) THEN
	2188	ALLOCATE( pt_2m_av(nysg:nyng,nxlg:nxrg) )
	2189	ENDIF
	2190	IF ( k == 1 ) READ ( 13 ) tmp_2d
	2191	pt_2m_av(nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2192	tmp_2d(nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
[4583]	2193
[4757]	2194	CASE ( 'rh_av' )
	2195	IF ( .NOT. ALLOCATED( rh_av ) ) THEN
	2196	ALLOCATE( rh_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2197	ENDIF
	2198	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2199	rh_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2200	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2201
[4518]	2202	CASE ( 'ti_av' )
	2203	IF ( .NOT. ALLOCATED( ti_av ) ) THEN
	2204	ALLOCATE( ti_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2205	ENDIF
	2206	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2207	ti_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2208	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2209
	2210	CASE ( 'u_center_av' )
	2211	IF ( .NOT. ALLOCATED( u_center_av ) ) THEN
	2212	ALLOCATE( u_center_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2213	ENDIF
	2214	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2215	u_center_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2216	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2217
	2218	CASE ( 'uu_av' )
	2219	IF ( .NOT. ALLOCATED( uu_av ) ) THEN
	2220	ALLOCATE( uu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2221	ENDIF
	2222	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2223	uu_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2224	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2225
	2226	CASE ( 'uv_10m_av' )
	2227	IF ( .NOT. ALLOCATED( uv_10m_av ) ) THEN
	2228	ALLOCATE( uv_10m_av(nysg:nyng,nxlg:nxrg) )
	2229	ENDIF
	2230	IF ( k == 1 ) READ ( 13 ) tmp_2d
	2231	uv_10m_av(nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2232	tmp_2d(nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2233
	2234	CASE ( 'v_center_av' )
	2235	IF ( .NOT. ALLOCATED( v_center_av ) ) THEN
	2236	ALLOCATE( v_center_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2237	ENDIF
	2238	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2239	v_center_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2240	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2241
	2242	CASE ( 'vv_av' )
	2243	IF ( .NOT. ALLOCATED( vv_av ) ) THEN
	2244	ALLOCATE( vv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2245	ENDIF
	2246	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2247	vv_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2248	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2249
	2250	CASE ( 'wtheta_av' )
	2251	IF ( .NOT. ALLOCATED( wtheta_av ) ) THEN
	2252	ALLOCATE( wtheta_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2253	ENDIF
	2254	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2255	wtheta_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2256	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2257
	2258	CASE ( 'wq_av' )
	2259	IF ( .NOT. ALLOCATED( wq_av ) ) THEN
	2260	ALLOCATE( wq_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2261	ENDIF
	2262	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2263	wq_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2264	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2265
	2266	CASE ( 'wspeed_av' )
	2267	IF ( .NOT. ALLOCATED( wspeed_av ) ) THEN
	2268	ALLOCATE( wspeed_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2269	ENDIF
	2270	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2271	wspeed_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2272	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2273
	2274	CASE ( 'wu_av' )
	2275	IF ( .NOT. ALLOCATED( wu_av ) ) THEN
	2276	ALLOCATE( wu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2277	ENDIF
	2278	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2279	wu_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2280	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2281
	2282	CASE ( 'wv_av' )
	2283	IF ( .NOT. ALLOCATED( wv_av ) ) THEN
	2284	ALLOCATE( wv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2285	ENDIF
	2286	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2287	wv_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2288	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2289
	2290	CASE ( 'ww_av' )
	2291	IF ( .NOT. ALLOCATED( ww_av ) ) THEN
	2292	ALLOCATE( ww_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2293	ENDIF
	2294	IF ( k == 1 ) READ ( 13 ) tmp_3d
	2295	ww_av(:,nysc-nbgp:nync+nbgp,nxlc-nbgp:nxrc+nbgp) = &
	2296	tmp_3d(:,nysf-nbgp:nynf+nbgp,nxlf-nbgp:nxrf+nbgp)
	2297
	2298
	2299	CASE DEFAULT
	2300
	2301	found = .FALSE.
	2302
	2303	END SELECT
	2304
	2305	END SUBROUTINE doq_rrd_local_ftn
	2306
[4583]	2307	!--------------------------------------------------------------------------------------------------!
[4039]	2308	! Description:
	2309	! ------------
[4518]	2310	!> Read module-specific local restart data arrays (MPI-IO).
[4583]	2311	!--------------------------------------------------------------------------------------------------!
[4518]	2312	SUBROUTINE doq_rrd_local_mpi
	2313
	2314	LOGICAL :: array_found !< control flad indicating whether the array is found in the file or not
	2315
	2316
	2317	CALL rd_mpi_io_check_array( 'pt_2m_av' , found = array_found )
	2318	IF ( array_found ) THEN
	2319	IF ( .NOT. ALLOCATED( pt_2m_av ) ) ALLOCATE( pt_2m_av(nysg:nyng,nxlg:nxrg) )
	2320	CALL rrd_mpi_io( 'pt_2m_av', pt_2m_av )
	2321	ENDIF
	2322
	2323	CALL rd_mpi_io_check_array( 'ti_av' , found = array_found )
	2324	IF ( array_found ) THEN
	2325	IF ( .NOT. ALLOCATED( ti_av ) ) ALLOCATE( ti_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2326	CALL rrd_mpi_io( 'ti_av', ti_av )
	2327	ENDIF
	2328
[4757]	2329	CALL rd_mpi_io_check_array( 'rh_av' , found = array_found )
	2330	IF ( array_found ) THEN
	2331	IF ( .NOT. ALLOCATED( rh_av ) ) ALLOCATE( ti_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2332	CALL rrd_mpi_io( 'rh_av', rh_av )
	2333	ENDIF
	2334
[4518]	2335	CALL rd_mpi_io_check_array( 'u_center_av' , found = array_found )
	2336	IF ( array_found ) THEN
	2337	IF ( .NOT. ALLOCATED( u_center_av ) ) ALLOCATE( u_center_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2338	CALL rrd_mpi_io( 'u_center_av', u_center_av )
	2339	ENDIF
	2340
	2341	CALL rd_mpi_io_check_array( 'uu_av' , found = array_found )
	2342	IF ( array_found ) THEN
	2343	IF ( .NOT. ALLOCATED( uu_av ) ) ALLOCATE( uu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2344	CALL rrd_mpi_io( 'uu_av', uu_av )
	2345	ENDIF
	2346
	2347	CALL rd_mpi_io_check_array( 'uv_10m_av' , found = array_found )
	2348	IF ( array_found ) THEN
	2349	IF ( .NOT. ALLOCATED( uv_10m_av ) ) ALLOCATE( uv_10m_av(nysg:nyng,nxlg:nxrg) )
	2350	CALL rrd_mpi_io( 'uv_10m_av', uv_10m_av )
	2351	ENDIF
	2352
	2353	CALL rd_mpi_io_check_array( 'v_center_av' , found = array_found )
	2354	IF ( array_found ) THEN
	2355	IF ( .NOT. ALLOCATED( v_center_av ) ) ALLOCATE( v_center_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2356	CALL rrd_mpi_io( 'v_center_av', v_center_av )
	2357	ENDIF
	2358
	2359	CALL rd_mpi_io_check_array( 'vv_av' , found = array_found )
	2360	IF ( array_found ) THEN
	2361	IF ( .NOT. ALLOCATED( vv_av ) ) ALLOCATE( vv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2362	CALL rrd_mpi_io( 'vv_av', vv_av )
	2363	ENDIF
	2364
	2365	CALL rd_mpi_io_check_array( 'ww_av' , found = array_found )
	2366	IF ( array_found ) THEN
	2367	IF ( .NOT. ALLOCATED( ww_av ) ) ALLOCATE( ww_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2368	CALL rrd_mpi_io( 'ww_av', ww_av )
	2369	ENDIF
	2370
	2371	CALL rd_mpi_io_check_array( 'wu_av' , found = array_found )
	2372	IF ( array_found ) THEN
	2373	IF ( .NOT. ALLOCATED( wu_av ) ) ALLOCATE( wu_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2374	CALL rrd_mpi_io( 'wu_av', wu_av )
	2375	ENDIF
	2376
	2377	CALL rd_mpi_io_check_array( 'wv_av' , found = array_found )
	2378	IF ( array_found ) THEN
	2379	IF ( .NOT. ALLOCATED( wv_av ) ) ALLOCATE( wv_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2380	CALL rrd_mpi_io( 'wv_av', wv_av )
	2381	ENDIF
	2382
	2383	CALL rd_mpi_io_check_array( 'wtheta_av' , found = array_found )
	2384	IF ( array_found ) THEN
	2385	IF ( .NOT. ALLOCATED( wtheta_av ) ) ALLOCATE( wtheta_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2386	CALL rrd_mpi_io( 'wtheta_av', wtheta_av )
	2387	ENDIF
	2388
	2389	CALL rd_mpi_io_check_array( 'wq_av' , found = array_found )
	2390	IF ( array_found ) THEN
	2391	IF ( .NOT. ALLOCATED( wq_av ) ) ALLOCATE( wq_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2392	CALL rrd_mpi_io( 'wq_av', wq_av )
	2393	ENDIF
	2394
	2395	CALL rd_mpi_io_check_array( 'wspeed_av' , found = array_found )
	2396	IF ( array_found ) THEN
	2397	IF ( .NOT. ALLOCATED( wspeed_av ) ) ALLOCATE( wspeed_av(nzb:nzt+1,nysg:nyng,nxlg:nxrg) )
	2398	CALL rrd_mpi_io( 'wspeed_av', wspeed_av )
	2399	ENDIF
	2400
	2401	END SUBROUTINE doq_rrd_local_mpi
	2402
[4583]	2403	!--------------------------------------------------------------------------------------------------!
[4518]	2404	! Description:
	2405	! ------------
[4039]	2406	!> Subroutine writes local (subdomain) restart data
[4583]	2407	!--------------------------------------------------------------------------------------------------!
[4039]	2408	SUBROUTINE doq_wrd_local
[3994]	2409
[4039]	2410	IMPLICIT NONE
	2411
[4583]	2412
[4517]	2413	IF ( TRIM( restart_data_format_output ) == 'fortran_binary' ) THEN
[4331]	2414
[4517]	2415	IF ( ALLOCATED( pt_2m_av ) ) THEN
	2416	CALL wrd_write_string( 'pt_2m_av' )
	2417	WRITE ( 14 ) pt_2m_av
	2418	ENDIF
[4331]	2419
[4757]	2420	IF ( ALLOCATED( rh_av ) ) THEN
	2421	CALL wrd_write_string( 'rh_av' )
	2422	WRITE ( 14 ) rh_av
	2423	ENDIF
	2424
[4517]	2425	IF ( ALLOCATED( ti_av ) ) THEN
	2426	CALL wrd_write_string( 'ti_av' )
	2427	WRITE ( 14 ) ti_av
	2428	ENDIF
[4331]	2429
[4518]	2430	IF ( ALLOCATED( u_center_av ) ) THEN
	2431	CALL wrd_write_string( 'u_center_av' )
	2432	WRITE ( 14 ) u_center_av
	2433	ENDIF
	2434
[4517]	2435	IF ( ALLOCATED( uu_av ) ) THEN
	2436	CALL wrd_write_string( 'uu_av' )
	2437	WRITE ( 14 ) uu_av
	2438	ENDIF
[4331]	2439
[4517]	2440	IF ( ALLOCATED( uv_10m_av ) ) THEN
	2441	CALL wrd_write_string( 'uv_10m_av' )
	2442	WRITE ( 14 ) uv_10m_av
	2443	ENDIF
[4331]	2444
[4518]	2445	IF ( ALLOCATED( v_center_av ) ) THEN
	2446	CALL wrd_write_string( 'v_center_av' )
	2447	WRITE ( 14 ) v_center_av
	2448	ENDIF
	2449
[4517]	2450	IF ( ALLOCATED( vv_av ) ) THEN
	2451	CALL wrd_write_string( 'vv_av' )
	2452	WRITE ( 14 ) vv_av
	2453	ENDIF
[4039]	2454
[4517]	2455	IF ( ALLOCATED( ww_av ) ) THEN
	2456	CALL wrd_write_string( 'ww_av' )
	2457	WRITE ( 14 ) ww_av
	2458	ENDIF
[4039]	2459
[4517]	2460	IF ( ALLOCATED( wu_av ) ) THEN
	2461	CALL wrd_write_string( 'wu_av' )
	2462	WRITE ( 14 ) wu_av
	2463	ENDIF
[4431]	2464
[4517]	2465	IF ( ALLOCATED( wv_av ) ) THEN
	2466	CALL wrd_write_string( 'wv_av' )
	2467	WRITE ( 14 ) wv_av
	2468	ENDIF
[4431]	2469
[4517]	2470	IF ( ALLOCATED( wtheta_av ) ) THEN
	2471	CALL wrd_write_string( 'wtheta_av' )
	2472	WRITE ( 14 ) wtheta_av
	2473	ENDIF
[4351]	2474
[4517]	2475	IF ( ALLOCATED( wq_av ) ) THEN
	2476	CALL wrd_write_string( 'wq_av' )
	2477	WRITE ( 14 ) wq_av
	2478	ENDIF
[4431]	2479
[4517]	2480	IF ( ALLOCATED( wspeed_av ) ) THEN
	2481	CALL wrd_write_string( 'wspeed_av' )
	2482	WRITE ( 14 ) wspeed_av
	2483	ENDIF
[4431]	2484
[4535]	2485	ELSEIF ( restart_data_format_output(1:3) == 'mpi' ) THEN
[4517]	2486
[4518]	2487	IF ( ALLOCATED( pt_2m_av ) ) CALL wrd_mpi_io( 'pt_2m_av', pt_2m_av )
[4757]	2488	IF ( ALLOCATED( rh_av ) ) CALL wrd_mpi_io( 'rh_av', rh_av )
[4518]	2489	IF ( ALLOCATED( ti_av ) ) CALL wrd_mpi_io( 'ti_av', ti_av )
	2490	IF ( ALLOCATED( u_center_av ) ) CALL wrd_mpi_io( 'u_center_av', u_center_av )
	2491	IF ( ALLOCATED( uu_av ) ) CALL wrd_mpi_io( 'uu_av', uu_av )
	2492	IF ( ALLOCATED( uv_10m_av ) ) CALL wrd_mpi_io( 'uv_10m_av', uv_10m_av )
	2493	IF ( ALLOCATED( vv_av ) ) CALL wrd_mpi_io( 'vv_av', vv_av )
	2494	IF ( ALLOCATED( v_center_av ) ) CALL wrd_mpi_io( 'v_center_av', v_center_av )
	2495	IF ( ALLOCATED( wtheta_av ) ) CALL wrd_mpi_io( 'wtheta_av', wtheta_av )
	2496	IF ( ALLOCATED( wq_av ) ) CALL wrd_mpi_io( 'wq_av', wq_av )
	2497	IF ( ALLOCATED( wspeed_av ) ) CALL wrd_mpi_io( 'wspeed_av', wspeed_av )
	2498	IF ( ALLOCATED( wu_av ) ) CALL wrd_mpi_io( 'wu_av', wu_av )
	2499	IF ( ALLOCATED( wv_av ) ) CALL wrd_mpi_io( 'wv_av', wv_av )
	2500	IF ( ALLOCATED( ww_av ) ) CALL wrd_mpi_io( 'ww_av', ww_av )
[4517]	2501
[4431]	2502	ENDIF
	2503
[4039]	2504	END SUBROUTINE doq_wrd_local
	2505
[4431]	2506
[3994]	2507	END MODULE diagnostic_output_quantities_mod

Note: See TracBrowser for help on using the repository browser.

Download in other formats:

| Impressum | ©Leibniz Universität Hannover |