Home

Context Navigation

source: palm/trunk/SOURCE/data_output_mask.f90 @ 3537

Last change on this file since 3537 was 3467, checked in by suehring, 6 years ago
Branch salsa @3446 re-integrated into trunk
Property svn:keywords set to `Id`
File size: 31.8 KB

Rev	Line
[3298]	1	!> @file data_output_mask.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 1997-2018 Leibniz Universitaet Hannover
	18	!------------------------------------------------------------------------------!
	19	!
	20	! Current revisions:
	21	! -----------------
[2032]	22	!
	23	!
[3298]	24	! Former revisions:
	25	! -----------------
	26	! $Id: data_output_mask.f90 3467 2018-10-30 19:05:21Z eckhard $
[3467]	27	! Implementation of a new aerosol module salsa.
	28	!
	29	! 3435 2018-10-26 18:25:44Z gronemeier
[3435]	30	! Add terrain-following output
	31	!
	32	! 3421 2018-10-24 18:39:32Z gronemeier
[3421]	33	! Renamed output variables
	34	!
	35	! 3419 2018-10-24 17:27:31Z gronemeier
[3298]	36	! Minor formatting (kanani)
	37	! Call for chem_data_output_mask added (basit)
	38	!
	39	! 3274 2018-09-24 15:42:55Z knoop
	40	! Modularization of all bulk cloud physics code components
	41	!
[3274]	42	! 3241 2018-09-12 15:02:00Z raasch
[3298]	43	! unused variable removed
	44	!
	45	! 3083 2018-06-19 14:03:12Z gronemeier
	46	!
	47	!
	48	! 3045 2018-05-28 07:55:41Z Giersch
	49	! Error messages revised
	50	!
	51	! 3030 2018-05-23 14:37:00Z raasch
	52	! variable if renamed ivar
	53	!
	54	! 2718 2018-01-02 08:49:38Z maronga
	55	! Corrected "Former revisions" section
	56	!
	57	! 2696 2017-12-14 17:12:51Z kanani
	58	! Change in file header (GPL part)
	59	!
	60	! 2292 2017-06-20 09:51:42Z schwenkel
	61	! Implementation of new microphysic scheme: cloud_scheme = 'morrison'
	62	! includes two more prognostic equations for cloud drop concentration (nc)
	63	! and cloud water content (qc).
	64	!
	65	! 2101 2017-01-05 16:42:31Z suehring
	66	!
	67	! 2031 2016-10-21 15:11:58Z knoop
	68	! renamed variable rho to rho_ocean and rho_av to rho_ocean_av
	69	!
	70	! 2000 2016-08-20 18:09:15Z knoop
	71	! Forced header and separation lines into 80 columns
	72	!
	73	! 1980 2016-07-29 15:51:57Z suehring
	74	! Bugfix, in order to steer user-defined output, setting flag found explicitly
	75	! to .F.
	76	!
	77	! 1976 2016-07-27 13:28:04Z maronga
	78	! Output of radiation quantities is now done directly in the respective module
	79	!
	80	! 1960 2016-07-12 16:34:24Z suehring
	81	! Separate humidity and passive scalar
	82	!
	83	! 2016-03-06 18:36:17Z raasch
	84	! name change of netcdf routines and module + related changes,
	85	! switch back of netcdf data format moved from time integration routine to here
	86	!
	87	! 1691 2015-10-26 16:17:44Z maronga
	88	! Added output of radiative heating rates for RRTMG
	89	!
	90	! 1682 2015-10-07 23:56:08Z knoop
	91	! Code annotations made doxygen readable
	92	!
	93	! 1585 2015-04-30 07:05:52Z maronga
	94	! Added support for RRTMG
	95	!
	96	! 1438 2014-07-22 14:14:06Z heinze
	97	! +nr, qc, qr
	98	!
	99	! 1359 2014-04-11 17:15:14Z hoffmann
	100	! New particle structure integrated.
	101	!
	102	! 1353 2014-04-08 15:21:23Z heinze
	103	! REAL constants provided with KIND-attribute
	104	!
	105	! 1327 2014-03-21 11:00:16Z raasch
	106	!
	107	!
	108	! 1320 2014-03-20 08:40:49Z raasch
	109	! ONLY-attribute added to USE-statements,
	110	! kind-parameters added to all INTEGER and REAL declaration statements,
	111	! kinds are defined in new module kinds,
	112	! revision history before 2012 removed,
	113	! comment fields (!:) to be used for variable explanations added to
	114	! all variable declaration statements
	115	!
	116	! 1318 2014-03-17 13:35:16Z raasch
	117	! barrier argument removed from cpu_log,
	118	! module interfaces removed
	119	!
	120	! 1092 2013-02-02 11:24:22Z raasch
	121	! unused variables removed
	122	!
	123	! 1036 2012-10-22 13:43:42Z raasch
	124	! code put under GPL (PALM 3.9)
	125	!
	126	! 1031 2012-10-19 14:35:30Z raasch
	127	! netCDF4 without parallel file support implemented
	128	!
	129	! 1007 2012-09-19 14:30:36Z franke
	130	! Bugfix: calculation of pr must depend on the particle weighting factor,
	131	! missing calculation of ql_vp added
	132	!
	133	! 410 2009-12-04 17:05:40Z letzel
	134	! Initial version
	135	!
	136	! Description:
	137	! ------------
	138	!> Masked data output in netCDF format for current mask (current value of mid).
	139	!------------------------------------------------------------------------------!
	140	SUBROUTINE data_output_mask( av )
	141
	142
	143
	144	#if defined( __netcdf )
	145	USE arrays_3d, &
	146	ONLY: e, nc, nr, p, pt, q, qc, ql, ql_c, ql_v, qr, rho_ocean, s, sa, &
	147	tend, u, v, vpt, w, d_exner
	148
	149	USE averaging, &
	150	ONLY: e_av, lpt_av, nc_av, nr_av, p_av, pc_av, pr_av, pt_av, q_av, &
	151	qc_av, ql_av, ql_c_av, ql_v_av, ql_vp_av, qv_av, qr_av, &
	152	rho_ocean_av, s_av, sa_av, u_av, v_av, vpt_av, w_av
	153
	154	USE basic_constants_and_equations_mod, &
	155	ONLY: lv_d_cp
	156
	157	USE chemistry_model_mod, &
	158	ONLY: chem_data_output_mask
	159
	160	USE control_parameters, &
	161	ONLY: air_chemistry, domask, domask_no, domask_time_count, mask_i, &
	162	mask_j, mask_k, mask_size, mask_size_l, mask_start_l, &
[3435]	163	mask_surface, &
[3298]	164	max_masks, message_string, mid, nz_do3d, simulated_time
	165	USE cpulog, &
	166	ONLY: cpu_log, log_point
	167
	168	USE indices, &
[3435]	169	ONLY: nbgp, nxl, nxr, nyn, nys, nzb, nzt
[3298]	170
	171	USE kinds
	172
	173	USE bulk_cloud_model_mod, &
	174	ONLY: bulk_cloud_model
	175
	176	USE NETCDF
	177
	178	USE netcdf_interface, &
	179	ONLY: id_set_mask, id_var_domask, id_var_time_mask, nc_stat, &
	180	netcdf_data_format, netcdf_handle_error
	181
	182	USE particle_attributes, &
	183	ONLY: grid_particles, number_of_particles, particles, &
	184	particle_advection_start, prt_count
	185
	186	USE pegrid
	187
	188	USE radiation_model_mod, &
	189	ONLY: radiation, radiation_data_output_mask
	190
[3467]	191	USE salsa_mod, &
	192	ONLY: salsa, salsa_data_output_mask
	193
[3435]	194	USE surface_mod, &
[3467]	195	ONLY : surf_def_h, surf_lsm_h, surf_usm_h, get_topography_top_index_ji
[3435]	196
[3298]	197	IMPLICIT NONE
	198
[3435]	199	CHARACTER(LEN=5) :: grid !< flag to distinquish between staggered grids
	200
	201	INTEGER(iwp) :: av !<
	202	INTEGER(iwp) :: ngp !<
	203	INTEGER(iwp) :: i !<
	204	INTEGER(iwp) :: ivar !<
	205	INTEGER(iwp) :: j !<
	206	INTEGER(iwp) :: k !<
	207	INTEGER(iwp) :: kk !<
	208	INTEGER(iwp) :: n !<
[3298]	209	INTEGER(iwp) :: netcdf_data_format_save !<
[3435]	210	INTEGER(iwp) :: sender !<
	211	INTEGER(iwp) :: topo_top_ind !< k index of highest horizontal surface
	212	INTEGER(iwp) :: ind(6) !<
	213
	214	LOGICAL :: found !<
	215	LOGICAL :: resorted !<
	216
	217	REAL(wp) :: mean_r !<
	218	REAL(wp) :: s_r2 !<
	219	REAL(wp) :: s_r3 !<
[3298]	220
	221	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE :: local_pf !<
	222	#if defined( __parallel )
	223	REAL(wp), DIMENSION(:,:,:), ALLOCATABLE :: total_pf !<
	224	#endif
	225	REAL(wp), DIMENSION(:,:,:), POINTER :: to_be_resorted !<
	226
	227	!
	228	!-- Return, if nothing to output
	229	IF ( domask_no(mid,av) == 0 ) RETURN
	230
	231	CALL cpu_log (log_point(49),'data_output_mask','start')
	232
	233	!
	234	!-- Parallel netcdf output is not tested so far for masked data, hence
	235	!-- netcdf_data_format is switched back to non-paralell output.
	236	netcdf_data_format_save = netcdf_data_format
	237	IF ( netcdf_data_format == 5 ) netcdf_data_format = 3
	238	IF ( netcdf_data_format == 6 ) netcdf_data_format = 4
	239
	240	!
	241	!-- Open output file.
	242	IF ( myid == 0 .OR. netcdf_data_format > 4 ) THEN
	243	CALL check_open( 200+mid+av*max_masks )
	244	ENDIF
	245
	246	!
	247	!-- Allocate total and local output arrays.
	248	#if defined( __parallel )
	249	IF ( myid == 0 ) THEN
	250	ALLOCATE( total_pf(mask_size(mid,1),mask_size(mid,2),mask_size(mid,3)) )
	251	ENDIF
	252	#endif
	253	ALLOCATE( local_pf(mask_size_l(mid,1),mask_size_l(mid,2), &
	254	mask_size_l(mid,3)) )
	255
	256	!
	257	!-- Update the netCDF time axis.
	258	domask_time_count(mid,av) = domask_time_count(mid,av) + 1
	259	IF ( myid == 0 .OR. netcdf_data_format > 4 ) THEN
	260	nc_stat = NF90_PUT_VAR( id_set_mask(mid,av), id_var_time_mask(mid,av), &
	261	(/ simulated_time /), &
	262	start = (/ domask_time_count(mid,av) /), &
	263	count = (/ 1 /) )
	264	CALL netcdf_handle_error( 'data_output_mask', 460 )
	265	ENDIF
	266
	267	!
	268	!-- Loop over all variables to be written.
	269	ivar = 1
	270
	271	DO WHILE ( domask(mid,av,ivar)(1:1) /= ' ' )
	272	!
	273	!-- Reallocate local_pf on PE 0 since its shape changes during MPI exchange
	274	IF ( netcdf_data_format < 5 .AND. myid == 0 .AND. ivar > 1 ) THEN
	275	DEALLOCATE( local_pf )
	276	ALLOCATE( local_pf(mask_size_l(mid,1),mask_size_l(mid,2), &
	277	mask_size_l(mid,3)) )
	278	ENDIF
	279	!
[3435]	280	!-- Set default grid for terrain-following output
	281	grid = 's'
	282	!
[3298]	283	!-- Set flag to steer output of radiation, land-surface, or user-defined
	284	!-- quantities
	285	found = .FALSE.
	286	!
	287	!-- Store the variable chosen.
	288	resorted = .FALSE.
	289	SELECT CASE ( TRIM( domask(mid,av,ivar) ) )
	290
	291	CASE ( 'e' )
	292	IF ( av == 0 ) THEN
	293	to_be_resorted => e
	294	ELSE
	295	to_be_resorted => e_av
	296	ENDIF
	297
[3421]	298	CASE ( 'thetal' )
[3298]	299	IF ( av == 0 ) THEN
	300	to_be_resorted => pt
	301	ELSE
	302	to_be_resorted => lpt_av
	303	ENDIF
	304
	305	CASE ( 'nc' )
	306	IF ( av == 0 ) THEN
	307	to_be_resorted => nc
	308	ELSE
	309	to_be_resorted => nc_av
	310	ENDIF
	311
	312	CASE ( 'nr' )
	313	IF ( av == 0 ) THEN
	314	to_be_resorted => nr
	315	ELSE
	316	to_be_resorted => nr_av
	317	ENDIF
	318
	319	CASE ( 'p' )
	320	IF ( av == 0 ) THEN
	321	to_be_resorted => p
	322	ELSE
	323	to_be_resorted => p_av
	324	ENDIF
	325
	326	CASE ( 'pc' ) ! particle concentration (requires ghostpoint exchange)
	327	IF ( av == 0 ) THEN
	328	tend = prt_count
	329	CALL exchange_horiz( tend, nbgp )
[3435]	330	IF ( .NOT. mask_surface(mid) ) THEN
	331	DO i = 1, mask_size_l(mid,1)
	332	DO j = 1, mask_size_l(mid,2)
	333	DO k = 1, mask_size_l(mid,3)
	334	local_pf(i,j,k) = tend(mask_k(mid,k), &
	335	mask_j(mid,j),mask_i(mid,i))
	336	ENDDO
[3298]	337	ENDDO
	338	ENDDO
[3435]	339	ELSE
	340	!
	341	!-- Terrain-following masked output
	342	DO i = 1, mask_size_l(mid,1)
	343	DO j = 1, mask_size_l(mid,2)
	344	!
	345	!-- Get k index of highest horizontal surface
	346	topo_top_ind = &
	347	get_topography_top_index_ji( mask_j(mid,j), &
	348	mask_i(mid,i), &
	349	grid )
	350	DO k = 1, mask_size_l(mid,3)
	351	kk = MIN( topo_top_ind+mask_k(mid,k), nzt+1 )
	352	local_pf(i,j,k) = &
	353	tend(kk,mask_j(mid,j),mask_i(mid,i))
	354	ENDDO
	355	ENDDO
	356	ENDDO
	357	ENDIF
[3298]	358	resorted = .TRUE.
	359	ELSE
	360	CALL exchange_horiz( pc_av, nbgp )
	361	to_be_resorted => pc_av
	362	ENDIF
	363
	364	CASE ( 'pr' ) ! mean particle radius (effective radius)
	365	IF ( av == 0 ) THEN
	366	IF ( simulated_time >= particle_advection_start ) THEN
	367	DO i = nxl, nxr
	368	DO j = nys, nyn
	369	DO k = nzb, nz_do3d
	370	number_of_particles = prt_count(k,j,i)
	371	IF (number_of_particles <= 0) CYCLE
	372	particles => grid_particles(k,j,i)%particles(1:number_of_particles)
	373	s_r2 = 0.0_wp
	374	s_r3 = 0.0_wp
	375	DO n = 1, number_of_particles
	376	IF ( particles(n)%particle_mask ) THEN
	377	s_r2 = s_r2 + grid_particles(k,j,i)%particles(n)%radius*2 &
	378	grid_particles(k,j,i)%particles(n)%weight_factor
	379	s_r3 = s_r3 + grid_particles(k,j,i)%particles(n)%radius*3 &
	380	grid_particles(k,j,i)%particles(n)%weight_factor
	381	ENDIF
	382	ENDDO
	383	IF ( s_r2 > 0.0_wp ) THEN
	384	mean_r = s_r3 / s_r2
	385	ELSE
	386	mean_r = 0.0_wp
	387	ENDIF
	388	tend(k,j,i) = mean_r
	389	ENDDO
	390	ENDDO
	391	ENDDO
	392	CALL exchange_horiz( tend, nbgp )
	393	ELSE
	394	tend = 0.0_wp
	395	ENDIF
[3435]	396	IF ( .NOT. mask_surface(mid) ) THEN
	397	DO i = 1, mask_size_l(mid,1)
	398	DO j = 1, mask_size_l(mid,2)
	399	DO k = 1, mask_size_l(mid,3)
	400	local_pf(i,j,k) = tend(mask_k(mid,k), &
	401	mask_j(mid,j),mask_i(mid,i))
	402	ENDDO
[3298]	403	ENDDO
	404	ENDDO
[3435]	405	ELSE
	406	!
	407	!-- Terrain-following masked output
	408	DO i = 1, mask_size_l(mid,1)
	409	DO j = 1, mask_size_l(mid,2)
	410	!
	411	!-- Get k index of highest horizontal surface
	412	topo_top_ind = &
	413	get_topography_top_index_ji( mask_j(mid,j), &
	414	mask_i(mid,i), &
	415	grid )
	416	DO k = 1, mask_size_l(mid,3)
	417	kk = MIN( topo_top_ind+mask_k(mid,k), nzt+1 )
	418	local_pf(i,j,k) = &
	419	tend(kk,mask_j(mid,j),mask_i(mid,i))
	420	ENDDO
	421	ENDDO
	422	ENDDO
	423	ENDIF
[3298]	424	resorted = .TRUE.
	425	ELSE
	426	CALL exchange_horiz( pr_av, nbgp )
	427	to_be_resorted => pr_av
	428	ENDIF
	429
[3421]	430	CASE ( 'theta' )
[3298]	431	IF ( av == 0 ) THEN
	432	IF ( .NOT. bulk_cloud_model ) THEN
	433	to_be_resorted => pt
	434	ELSE
[3435]	435	IF ( .NOT. mask_surface(mid) ) THEN
	436	DO i = 1, mask_size_l(mid,1)
	437	DO j = 1, mask_size_l(mid,2)
	438	DO k = 1, mask_size_l(mid,3)
	439	local_pf(i,j,k) = &
	440	pt(mask_k(mid,k),mask_j(mid,j),mask_i(mid,i)) &
	441	+ lv_d_cp * d_exner(mask_k(mid,k)) * &
	442	ql(mask_k(mid,k),mask_j(mid,j),mask_i(mid,i))
	443	ENDDO
[3298]	444	ENDDO
	445	ENDDO
[3435]	446	ELSE
	447	!
	448	!-- Terrain-following masked output
	449	DO i = 1, mask_size_l(mid,1)
	450	DO j = 1, mask_size_l(mid,2)
	451	!
	452	!-- Get k index of highest horizontal surface
	453	topo_top_ind = &
	454	get_topography_top_index_ji( mask_j(mid,j), &
	455	mask_i(mid,i), &
	456	grid )
	457	DO k = 1, mask_size_l(mid,3)
	458	kk = MIN( topo_top_ind+mask_k(mid,k), nzt+1 )
	459	local_pf(i,j,k) = &
	460	pt(kk,mask_j(mid,j),mask_i(mid,i) ) &
	461	+ lv_d_cp * d_exner(kk) * &
	462	ql(kk,mask_j(mid,j),mask_i(mid,i))
	463	ENDDO
	464	ENDDO
	465	ENDDO
	466	ENDIF
[3298]	467	resorted = .TRUE.
	468	ENDIF
	469	ELSE
	470	to_be_resorted => pt_av
	471	ENDIF
	472
	473	CASE ( 'q' )
	474	IF ( av == 0 ) THEN
	475	to_be_resorted => q
	476	ELSE
	477	to_be_resorted => q_av
	478	ENDIF
	479
	480	CASE ( 'qc' )
	481	IF ( av == 0 ) THEN
	482	to_be_resorted => qc
	483	ELSE
	484	to_be_resorted => qc_av
	485	ENDIF
	486
	487	CASE ( 'ql' )
	488	IF ( av == 0 ) THEN
	489	to_be_resorted => ql
	490	ELSE
	491	to_be_resorted => ql_av
	492	ENDIF
	493
	494	CASE ( 'ql_c' )
	495	IF ( av == 0 ) THEN
	496	to_be_resorted => ql_c
	497	ELSE
	498	to_be_resorted => ql_c_av
	499	ENDIF
	500
	501	CASE ( 'ql_v' )
	502	IF ( av == 0 ) THEN
	503	to_be_resorted => ql_v
	504	ELSE
	505	to_be_resorted => ql_v_av
	506	ENDIF
	507
	508	CASE ( 'ql_vp' )
	509	IF ( av == 0 ) THEN
	510	IF ( simulated_time >= particle_advection_start ) THEN
	511	DO i = nxl, nxr
	512	DO j = nys, nyn
	513	DO k = nzb, nz_do3d
	514	number_of_particles = prt_count(k,j,i)
	515	IF (number_of_particles <= 0) CYCLE
	516	particles => grid_particles(k,j,i)%particles(1:number_of_particles)
	517	DO n = 1, number_of_particles
	518	IF ( particles(n)%particle_mask ) THEN
	519	tend(k,j,i) = tend(k,j,i) + &
	520	particles(n)%weight_factor / &
	521	prt_count(k,j,i)
	522	ENDIF
	523	ENDDO
	524	ENDDO
	525	ENDDO
	526	ENDDO
	527	CALL exchange_horiz( tend, nbgp )
	528	ELSE
	529	tend = 0.0_wp
	530	ENDIF
[3435]	531	IF ( .NOT. mask_surface(mid) ) THEN
	532	DO i = 1, mask_size_l(mid,1)
	533	DO j = 1, mask_size_l(mid,2)
	534	DO k = 1, mask_size_l(mid,3)
	535	local_pf(i,j,k) = tend(mask_k(mid,k), &
	536	mask_j(mid,j),mask_i(mid,i))
	537	ENDDO
[3298]	538	ENDDO
	539	ENDDO
[3435]	540	ELSE
	541	!
	542	!-- Terrain-following masked output
	543	DO i = 1, mask_size_l(mid,1)
	544	DO j = 1, mask_size_l(mid,2)
	545	!
	546	!-- Get k index of highest horizontal surface
	547	topo_top_ind = &
	548	get_topography_top_index_ji( mask_j(mid,j), &
	549	mask_i(mid,i), &
	550	grid )
	551	DO k = 1, mask_size_l(mid,3)
	552	kk = MIN( topo_top_ind+mask_k(mid,k), nzt+1 )
	553	local_pf(i,j,k) = &
	554	tend(kk,mask_j(mid,j),mask_i(mid,i))
	555	ENDDO
	556	ENDDO
	557	ENDDO
	558	ENDIF
[3298]	559	resorted = .TRUE.
	560	ELSE
	561	CALL exchange_horiz( ql_vp_av, nbgp )
	562	to_be_resorted => ql_vp_av
	563	ENDIF
	564
	565	CASE ( 'qv' )
	566	IF ( av == 0 ) THEN
[3435]	567	IF ( .NOT. mask_surface(mid) ) THEN
	568	DO i = 1, mask_size_l(mid,1)
	569	DO j = 1, mask_size_l(mid,2)
	570	DO k = 1, mask_size_l(mid,3)
	571	local_pf(i,j,k) = &
	572	q(mask_k(mid,k),mask_j(mid,j),mask_i(mid,i)) - &
	573	ql(mask_k(mid,k),mask_j(mid,j),mask_i(mid,i))
	574	ENDDO
[3298]	575	ENDDO
	576	ENDDO
[3435]	577	ELSE
	578	!
	579	!-- Terrain-following masked output
	580	DO i = 1, mask_size_l(mid,1)
	581	DO j = 1, mask_size_l(mid,2)
	582	!
	583	!-- Get k index of highest horizontal surface
	584	topo_top_ind = &
	585	get_topography_top_index_ji( mask_j(mid,j), &
	586	mask_i(mid,i), &
	587	grid )
	588	DO k = 1, mask_size_l(mid,3)
	589	kk = MIN( topo_top_ind+mask_k(mid,k), nzt+1 )
	590	local_pf(i,j,k) = &
	591	q(kk,mask_j(mid,j),mask_i(mid,i)) - &
	592	ql(kk,mask_j(mid,j),mask_i(mid,i))
	593	ENDDO
	594	ENDDO
	595	ENDDO
	596	ENDIF
[3298]	597	resorted = .TRUE.
	598	ELSE
	599	to_be_resorted => qv_av
	600	ENDIF
	601
	602	CASE ( 'qr' )
	603	IF ( av == 0 ) THEN
	604	to_be_resorted => qr
	605	ELSE
	606	to_be_resorted => qr_av
	607	ENDIF
	608
[3421]	609	CASE ( 'rho_sea_water' )
[3298]	610	IF ( av == 0 ) THEN
	611	to_be_resorted => rho_ocean
	612	ELSE
	613	to_be_resorted => rho_ocean_av
	614	ENDIF
	615
	616	CASE ( 's' )
	617	IF ( av == 0 ) THEN
	618	to_be_resorted => s
	619	ELSE
	620	to_be_resorted => s_av
	621	ENDIF
	622
	623	CASE ( 'sa' )
	624	IF ( av == 0 ) THEN
	625	to_be_resorted => sa
	626	ELSE
	627	to_be_resorted => sa_av
	628	ENDIF
	629
	630	CASE ( 'u' )
	631	IF ( av == 0 ) THEN
	632	to_be_resorted => u
	633	ELSE
	634	to_be_resorted => u_av
	635	ENDIF
	636
	637	CASE ( 'v' )
	638	IF ( av == 0 ) THEN
	639	to_be_resorted => v
	640	ELSE
	641	to_be_resorted => v_av
	642	ENDIF
	643
[3421]	644	CASE ( 'thetav' )
[3298]	645	IF ( av == 0 ) THEN
	646	to_be_resorted => vpt
	647	ELSE
	648	to_be_resorted => vpt_av
	649	ENDIF
	650
	651	CASE ( 'w' )
[3435]	652	grid = 'w'
[3298]	653	IF ( av == 0 ) THEN
	654	to_be_resorted => w
	655	ELSE
	656	to_be_resorted => w_av
	657	ENDIF
	658
	659	CASE DEFAULT
[3421]	660
[3298]	661	!
	662	!-- Radiation quantity
	663	IF ( radiation ) THEN
	664	CALL radiation_data_output_mask(av, domask(mid,av,ivar), found,&
	665	local_pf )
	666	ENDIF
	667
	668	IF ( air_chemistry ) THEN
	669	CALL chem_data_output_mask(av, domask(mid,av,ivar), found, &
	670	local_pf )
	671	ENDIF
	672	!
[3467]	673	!-- SALSA quantities
	674	IF ( salsa ) THEN
	675	CALL salsa_data_output_mask( av, domask(mid,av,ivar), found, &
	676	local_pf )
	677	ENDIF
	678	!
[3298]	679	!-- User defined quantity
	680	IF ( .NOT. found ) THEN
	681	CALL user_data_output_mask(av, domask(mid,av,ivar), found, &
	682	local_pf )
	683	ENDIF
	684
	685	resorted = .TRUE.
	686
	687	IF ( .NOT. found ) THEN
	688	WRITE ( message_string, * ) 'no masked output available for: ',&
	689	TRIM( domask(mid,av,ivar) )
	690	CALL message( 'data_output_mask', 'PA0327', 0, 0, 0, 6, 0 )
	691	ENDIF
	692
	693	END SELECT
	694
	695	!
	696	!-- Resort the array to be output, if not done above
	697	IF ( .NOT. resorted ) THEN
[3435]	698	IF ( .NOT. mask_surface(mid) ) THEN
	699	!
	700	!-- Default masked output
	701	DO i = 1, mask_size_l(mid,1)
	702	DO j = 1, mask_size_l(mid,2)
	703	DO k = 1, mask_size_l(mid,3)
	704	local_pf(i,j,k) = to_be_resorted(mask_k(mid,k), &
	705	mask_j(mid,j),mask_i(mid,i))
	706	ENDDO
[3298]	707	ENDDO
	708	ENDDO
[3435]	709
	710	ELSE
	711	!
	712	!-- Terrain-following masked output
	713	DO i = 1, mask_size_l(mid,1)
	714	DO j = 1, mask_size_l(mid,2)
	715	!
	716	!-- Get k index of highest horizontal surface
	717	topo_top_ind = get_topography_top_index_ji( mask_j(mid,j), &
	718	mask_i(mid,i), &
	719	grid )
	720	!
	721	!-- Save output array
	722	DO k = 1, mask_size_l(mid,3)
	723	local_pf(i,j,k) = to_be_resorted( &
	724	MIN( topo_top_ind+mask_k(mid,k), &
	725	nzt+1 ), &
	726	mask_j(mid,j), &
	727	mask_i(mid,i) )
	728	ENDDO
	729	ENDDO
	730	ENDDO
	731
	732	ENDIF
[3298]	733	ENDIF
	734
	735	!
	736	!-- I/O block. I/O methods are implemented
	737	!-- (1) for parallel execution
	738	!-- a. with netCDF 4 parallel I/O-enabled library
	739	!-- b. with netCDF 3 library
	740	!-- (2) for serial execution.
	741	!-- The choice of method depends on the correct setting of preprocessor
	742	!-- directives __parallel and __netcdf4_parallel as well as on the parameter
	743	!-- netcdf_data_format.
	744	#if defined( __parallel )
	745	#if defined( __netcdf4_parallel )
	746	IF ( netcdf_data_format > 4 ) THEN
	747	!
	748	!-- (1) a. Parallel I/O using netCDF 4 (not yet tested)
	749	nc_stat = NF90_PUT_VAR( id_set_mask(mid,av), &
	750	id_var_domask(mid,av,ivar), local_pf, &
	751	start = (/ mask_start_l(mid,1), mask_start_l(mid,2), &
	752	mask_start_l(mid,3), domask_time_count(mid,av) /), &
	753	count = (/ mask_size_l(mid,1), mask_size_l(mid,2), &
	754	mask_size_l(mid,3), 1 /) )
	755	CALL netcdf_handle_error( 'data_output_mask', 461 )
	756	ELSE
	757	#endif
	758	!
	759	!-- (1) b. Conventional I/O only through PE0
	760	!-- PE0 receives partial arrays from all processors of the respective mask
	761	!-- and outputs them. Here a barrier has to be set, because otherwise
	762	!-- "-MPI- FATAL: Remote protocol queue full" may occur.
	763	CALL MPI_BARRIER( comm2d, ierr )
	764
	765	ngp = mask_size_l(mid,1) * mask_size_l(mid,2) * mask_size_l(mid,3)
	766	IF ( myid == 0 ) THEN
	767	!
	768	!-- Local array can be relocated directly.
	769	total_pf( &
	770	mask_start_l(mid,1):mask_start_l(mid,1)+mask_size_l(mid,1)-1, &
	771	mask_start_l(mid,2):mask_start_l(mid,2)+mask_size_l(mid,2)-1, &
	772	mask_start_l(mid,3):mask_start_l(mid,3)+mask_size_l(mid,3)-1 ) &
	773	= local_pf
	774	!
	775	!-- Receive data from all other PEs.
	776	DO n = 1, numprocs-1
	777	!
	778	!-- Receive index limits first, then array.
	779	!-- Index limits are received in arbitrary order from the PEs.
	780	CALL MPI_RECV( ind(1), 6, MPI_INTEGER, MPI_ANY_SOURCE, 0, &
	781	comm2d, status, ierr )
	782	!
	783	!-- Not all PEs have data for the mask
	784	IF ( ind(1) /= -9999 ) THEN
	785	ngp = ( ind(2)-ind(1)+1 ) * (ind(4)-ind(3)+1 ) * &
	786	( ind(6)-ind(5)+1 )
	787	sender = status(MPI_SOURCE)
	788	DEALLOCATE( local_pf )
	789	ALLOCATE(local_pf(ind(1):ind(2),ind(3):ind(4),ind(5):ind(6)))
	790	CALL MPI_RECV( local_pf(ind(1),ind(3),ind(5)), ngp, &
	791	MPI_REAL, sender, 1, comm2d, status, ierr )
	792	total_pf(ind(1):ind(2),ind(3):ind(4),ind(5):ind(6)) &
	793	= local_pf
	794	ENDIF
	795	ENDDO
	796
	797	nc_stat = NF90_PUT_VAR( id_set_mask(mid,av), &
	798	id_var_domask(mid,av,ivar), total_pf, &
	799	start = (/ 1, 1, 1, domask_time_count(mid,av) /), &
	800	count = (/ mask_size(mid,1), mask_size(mid,2), &
	801	mask_size(mid,3), 1 /) )
	802	CALL netcdf_handle_error( 'data_output_mask', 462 )
	803
	804	ELSE
	805	!
	806	!-- If at least part of the mask resides on the PE, send the index
	807	!-- limits for the target array, otherwise send -9999 to PE0.
	808	IF ( mask_size_l(mid,1) > 0 .AND. mask_size_l(mid,2) > 0 .AND. &
	809	mask_size_l(mid,3) > 0 ) &
	810	THEN
	811	ind(1) = mask_start_l(mid,1)
	812	ind(2) = mask_start_l(mid,1) + mask_size_l(mid,1) - 1
	813	ind(3) = mask_start_l(mid,2)
	814	ind(4) = mask_start_l(mid,2) + mask_size_l(mid,2) - 1
	815	ind(5) = mask_start_l(mid,3)
	816	ind(6) = mask_start_l(mid,3) + mask_size_l(mid,3) - 1
	817	ELSE
	818	ind(1) = -9999; ind(2) = -9999
	819	ind(3) = -9999; ind(4) = -9999
	820	ind(5) = -9999; ind(6) = -9999
	821	ENDIF
	822	CALL MPI_SEND( ind(1), 6, MPI_INTEGER, 0, 0, comm2d, ierr )
	823	!
	824	!-- If applicable, send data to PE0.
	825	IF ( ind(1) /= -9999 ) THEN
	826	CALL MPI_SEND( local_pf(1,1,1), ngp, MPI_REAL, 0, 1, comm2d, &
	827	ierr )
	828	ENDIF
	829	ENDIF
	830	!
	831	!-- A barrier has to be set, because otherwise some PEs may proceed too
	832	!-- fast so that PE0 may receive wrong data on tag 0.
	833	CALL MPI_BARRIER( comm2d, ierr )
	834	#if defined( __netcdf4_parallel )
	835	ENDIF
	836	#endif
	837	#else
	838	!
	839	!-- (2) For serial execution of PALM, the single processor (PE0) holds all
	840	!-- data and writes them directly to file.
	841	nc_stat = NF90_PUT_VAR( id_set_mask(mid,av), &
	842	id_var_domask(mid,av,ivar), local_pf, &
	843	start = (/ 1, 1, 1, domask_time_count(mid,av) /), &
	844	count = (/ mask_size_l(mid,1), mask_size_l(mid,2),&
	845	mask_size_l(mid,3), 1 /) )
	846	CALL netcdf_handle_error( 'data_output_mask', 463 )
	847	#endif
	848
	849	ivar = ivar + 1
	850
	851	ENDDO
	852
	853	!
	854	!-- Deallocate temporary arrays.
	855	DEALLOCATE( local_pf )
	856	#if defined( __parallel )
	857	IF ( myid == 0 ) THEN
	858	DEALLOCATE( total_pf )
	859	ENDIF
	860	#endif
	861
	862	!
	863	!-- Switch back to original format given by user (see beginning of this routine)
	864	netcdf_data_format = netcdf_data_format_save
	865
	866	CALL cpu_log( log_point(49), 'data_output_mask', 'stop' )
	867	#endif
	868
[3435]	869
[3298]	870	END SUBROUTINE data_output_mask

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

Download in other formats:

| Impressum | ©Leibniz Universität Hannover |