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

Last change on this file since 4759 was 4559, checked in by raasch, 5 years ago

files re-formatted to follow the PALM coding standard

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