Home

Context Navigation

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

Last change on this file since 3461 was 3435, checked in by gronemeier, 6 years ago
new: terrain-following masked output; bugfixes: increase vertical dimension of gamma_w_green_sat by 1, add checks for masked output for chemistry_model and radiation_model, reordered calls to xxx_define_netcdf_grid in masked output part
Property svn:keywords set to `Id`
File size: 31.4 KB

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

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

Download in other formats:

| Impressum | ©Leibniz Universität Hannover |