Home

Context Navigation

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

Last change on this file since 4471 was 4457, checked in by raasch, 5 years ago
ghost point exchange modularized, bugfix for wrong 2d-exchange
Property svn:keywords set to `Id`
File size: 32.7 KB

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

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

Download in other formats:

| Impressum | ©Leibniz Universität Hannover |