Changeset 15048 for NEMO/trunk/src/ICE

Timestamp:

2021-06-23T18:02:14+02:00 (3 years ago)

Author:

clem

Message:

reduce drastically the number of global communications when using diagnostic outputs. New functions are created: glob_sum_vec and glob_sum_full_vec, which gave a vector as an output. This vector is composed of different summed arrays (such as temperature, salinity etc). Global diagnostics are identical as before. See examples in icedia and diahsb.

File:

• NEMO/trunk/src/ICE/icedia.F90 (modified) (2 diffs)

NEMO/trunk/src/ICE/icedia.F90

@@ -65 +65 @@
       INTEGER, INTENT(in) ::   kt   ! ocean time step
       !!
-      REAL(wp)   ::   zbg_ivol, zbg_item, zbg_area, zbg_isal
-      REAL(wp)   ::   zbg_svol, zbg_stem
-      REAL(wp)   ::   zbg_ipvol, zbg_ilvol
-      REAL(wp)   ::   z_frc_voltop, z_frc_temtop, z_frc_sal
-      REAL(wp)   ::   z_frc_volbot, z_frc_tembot
-      REAL(wp)   ::   zdiff_vol, zdiff_sal, zdiff_tem
+      REAL(wp), DIMENSION(jpi,jpj,16) ::   ztmp
+      REAL(wp), DIMENSION(16)         ::   zbg          
       !!---------------------------------------------------------------------------
       IF( ln_timing )   CALL timing_start('ice_dia')
@@ -84 +80 @@
       ENDIF
 
-      ! ----------------------- !
-      ! 1 -  Contents           !
-      ! ----------------------- !
-      IF(  iom_use('ibgvol_tot' ) .OR. iom_use('sbgvol_tot' ) .OR. iom_use('ibgarea_tot') .OR. &
-         & iom_use('ibgsalt_tot') .OR. iom_use('ibgheat_tot') .OR. iom_use('sbgheat_tot') .OR. &
-         & iom_use('ipbgvol_tot' ) .OR. iom_use('ilbgvol_tot' ) ) THEN
-
-         zbg_ivol = glob_sum( 'icedia', vt_i(:,:) * e1e2t(:,:) ) * 1.e-9  ! ice volume (km3)
-         zbg_svol = glob_sum( 'icedia', vt_s(:,:) * e1e2t(:,:) ) * 1.e-9  ! snow volume (km3)
-         zbg_area = glob_sum( 'icedia', at_i(:,:) * e1e2t(:,:) ) * 1.e-6  ! area (km2)
-         zbg_isal = glob_sum( 'icedia', st_i(:,:) * e1e2t(:,:) ) * 1.e-9  ! salt content (pss*km3)
-         zbg_item = glob_sum( 'icedia', et_i(:,:) * e1e2t(:,:) ) * 1.e-20 ! heat content (1.e20 J)
-         zbg_stem = glob_sum( 'icedia', et_s(:,:) * e1e2t(:,:) ) * 1.e-20 ! heat content (1.e20 J)
-         ! ponds
-         zbg_ipvol = glob_sum( 'icedia', vt_ip(:,:) * e1e2t(:,:) ) * 1.e-9  ! ice pond volume (km3)
-         zbg_ilvol = glob_sum( 'icedia', vt_il(:,:) * e1e2t(:,:) ) * 1.e-9  ! ice pond lid volume (km3)
-
-         CALL iom_put( 'ibgvol_tot'  , zbg_ivol )
-         CALL iom_put( 'sbgvol_tot'  , zbg_svol )
-         CALL iom_put( 'ibgarea_tot' , zbg_area )
-         CALL iom_put( 'ibgsalt_tot' , zbg_isal )
-         CALL iom_put( 'ibgheat_tot' , zbg_item )
-         CALL iom_put( 'sbgheat_tot' , zbg_stem )
-         ! ponds
-         CALL iom_put( 'ipbgvol_tot' , zbg_ipvol )
-         CALL iom_put( 'ilbgvol_tot' , zbg_ilvol )
-
-      ENDIF
-
+      ztmp(:,:,:) = 0._wp ! should be better coded
+      
       ! ---------------------------!
-      ! 2 - Trends due to forcing  !
+      ! 1 - Trends due to forcing  !
       ! ---------------------------!
       ! they must be kept outside an IF(iom_use) because of the call to dia_rst below
-      z_frc_volbot = r1_rho0 * glob_sum( 'icedia', -( wfx_ice(:,:) + wfx_snw(:,:) + wfx_err_sub(:,:) ) * e1e2t(:,:) ) * 1.e-9   ! freshwater flux ice/snow-ocean
-      z_frc_voltop = r1_rho0 * glob_sum( 'icedia', -( wfx_sub(:,:) + wfx_spr(:,:) )                    * e1e2t(:,:) ) * 1.e-9   ! freshwater flux ice/snow-atm
-      z_frc_sal    = r1_rho0 * glob_sum( 'icedia', -      sfx(:,:)                                     * e1e2t(:,:) ) * 1.e-9   ! salt fluxes ice/snow-ocean
-      z_frc_tembot =           glob_sum( 'icedia',  qt_oce_ai(:,:)                                     * e1e2t(:,:) ) * 1.e-20  ! heat on top of ocean (and below ice)
-      z_frc_temtop =           glob_sum( 'icedia',  qt_atm_oi(:,:)                                     * e1e2t(:,:) ) * 1.e-20  ! heat on top of ice-coean
-      !
-      frc_voltop  = frc_voltop  + z_frc_voltop  * rDt_ice ! km3
-      frc_volbot  = frc_volbot  + z_frc_volbot  * rDt_ice ! km3
-      frc_sal     = frc_sal     + z_frc_sal     * rDt_ice ! km3*pss
-      frc_temtop  = frc_temtop  + z_frc_temtop  * rDt_ice ! 1.e20 J
-      frc_tembot  = frc_tembot  + z_frc_tembot  * rDt_ice ! 1.e20 J
-
-      CALL iom_put( 'ibgfrcvoltop' , frc_voltop )   ! vol  forcing ice/snw-atm          (km3 equivalent ocean water)
-      CALL iom_put( 'ibgfrcvolbot' , frc_volbot )   ! vol  forcing ice/snw-ocean        (km3 equivalent ocean water)
-      CALL iom_put( 'ibgfrcsal'    , frc_sal    )   ! sal - forcing                     (psu*km3 equivalent ocean water)
-      CALL iom_put( 'ibgfrctemtop' , frc_temtop )   ! heat on top of ice/snw/ocean      (1.e20 J)
-      CALL iom_put( 'ibgfrctembot' , frc_tembot )   ! heat on top of ocean(below ice)   (1.e20 J)
-
-      IF(  iom_use('ibgfrchfxtop') .OR. iom_use('ibgfrchfxbot') ) THEN
-         CALL iom_put( 'ibgfrchfxtop' , frc_temtop * z1_e1e2 * 1.e-20 * kt*rn_Dt ) ! heat on top of ice/snw/ocean      (W/m2)
-         CALL iom_put( 'ibgfrchfxbot' , frc_tembot * z1_e1e2 * 1.e-20 * kt*rn_Dt ) ! heat on top of ocean(below ice)   (W/m2)
-      ENDIF
+      ztmp(:,:,1) = - ( wfx_ice(:,:) + wfx_snw(:,:) + wfx_err_sub(:,:) ) * e1e2t(:,:) ! freshwater flux ice/snow-ocean
+      ztmp(:,:,2) = - ( wfx_sub(:,:) + wfx_spr(:,:) )                    * e1e2t(:,:) ! freshwater flux ice/snow-atm
+      ztmp(:,:,3) = -   sfx    (:,:)                                     * e1e2t(:,:) ! salt fluxes ice/snow-ocean
+      ztmp(:,:,4) =   qt_atm_oi(:,:)                                     * e1e2t(:,:) ! heat on top of ice-ocean
+      ztmp(:,:,5) =   qt_oce_ai(:,:)                                     * e1e2t(:,:) ! heat on top of ocean (and below ice)
+      
+      ! ----------------------- !
+      ! 2 -  Contents           !
+      ! ----------------------- !
+      IF( iom_use('ibgvol_tot' ) )   ztmp(:,:,6 ) = vt_i (:,:) * e1e2t(:,:) ! ice volume
+      IF( iom_use('sbgvol_tot' ) )   ztmp(:,:,7 ) = vt_s (:,:) * e1e2t(:,:) ! snow volume
+      IF( iom_use('ibgarea_tot') )   ztmp(:,:,8 ) = at_i (:,:) * e1e2t(:,:) ! area
+      IF( iom_use('ibgsalt_tot') )   ztmp(:,:,9 ) = st_i (:,:) * e1e2t(:,:) ! salt content
+      IF( iom_use('ibgheat_tot') )   ztmp(:,:,10) = et_i (:,:) * e1e2t(:,:) ! heat content
+      IF( iom_use('sbgheat_tot') )   ztmp(:,:,11) = et_s (:,:) * e1e2t(:,:) ! heat content
+      IF( iom_use('ipbgvol_tot') )   ztmp(:,:,12) = vt_ip(:,:) * e1e2t(:,:) ! ice pond volume
+      IF( iom_use('ilbgvol_tot') )   ztmp(:,:,13) = vt_il(:,:) * e1e2t(:,:) ! ice pond lid volume
 
       ! ---------------------------------- !
       ! 3 -  Content variations and drifts !
       ! ---------------------------------- !
-      IF(  iom_use('ibgvolume') .OR. iom_use('ibgsaltco') .OR. iom_use('ibgheatco') .OR. iom_use('ibgheatfx') ) THEN
-
-         zdiff_vol = r1_rho0 * glob_sum( 'icedia', ( rhoi*vt_i(:,:) + rhos*vt_s(:,:) - vol_loc_ini(:,:) ) * e1e2t(:,:) ) * 1.e-9   ! freshwater trend (km3)
-         zdiff_sal = r1_rho0 * glob_sum( 'icedia', ( rhoi*st_i(:,:)                  - sal_loc_ini(:,:) ) * e1e2t(:,:) ) * 1.e-9   ! salt content trend (km3*pss)
-         zdiff_tem =           glob_sum( 'icedia', ( et_i(:,:) + et_s(:,:)           - tem_loc_ini(:,:) ) * e1e2t(:,:) ) * 1.e-20  ! heat content trend (1.e20 J)
-         !                               + SUM( qevap_ice * a_i_b, dim=3 )       !! clem: I think this term should not be there (but needs a check)
-
-         zdiff_vol = zdiff_vol - ( frc_voltop + frc_volbot )
-         zdiff_sal = zdiff_sal - frc_sal
-         zdiff_tem = zdiff_tem - ( frc_tembot - frc_temtop )
-
-         CALL iom_put( 'ibgvolume' , zdiff_vol )   ! ice/snow volume  drift            (km3 equivalent ocean water)
-         CALL iom_put( 'ibgsaltco' , zdiff_sal )   ! ice salt content drift            (psu*km3 equivalent ocean water)
-         CALL iom_put( 'ibgheatco' , zdiff_tem )   ! ice/snow heat content drift       (1.e20 J)
-         !
-      ENDIF
-
+      IF( iom_use('ibgvolume') ) ztmp(:,:,14) = ( rhoi*vt_i(:,:) + rhos*vt_s(:,:) - vol_loc_ini(:,:) ) * e1e2t(:,:) ! freshwater trend
+      IF( iom_use('ibgsaltco') ) ztmp(:,:,15) = ( rhoi*st_i(:,:)                  - sal_loc_ini(:,:) ) * e1e2t(:,:) ! salt content trend
+      IF( iom_use('ibgheatco') .OR. iom_use('ibgheatfx') ) &
+         &                       ztmp(:,:,16) = ( et_i(:,:) + et_s(:,:)           - tem_loc_ini(:,:) ) * e1e2t(:,:) ! heat content trend
+      
+      ! global sum
+      zbg(1:16) = glob_sum_vec( 'icedia', ztmp(:,:,1:16) )
+
+      ! change units for trends
+      zbg(1) = zbg(1) * r1_rho0 * 1.e-9  * rDt_ice ! freshwater flux ice/snow-ocean (km3)
+      zbg(2) = zbg(2) * r1_rho0 * 1.e-9  * rDt_ice ! freshwater flux ice/snow-atm (km3)
+      zbg(3) = zbg(3) * r1_rho0 * 1.e-9  * rDt_ice ! salt fluxes ice/snow-ocean (km3*pss)
+      zbg(4) = zbg(4)           * 1.e-20 * rDt_ice ! heat on top of ice-ocean (1.e20 J)
+      zbg(5) = zbg(5)           * 1.e-20 * rDt_ice ! heat on top of ocean (and below ice) (1.e20 J)
+      ! cumulative sum
+      frc_voltop  = frc_voltop  + zbg(1)
+      frc_volbot  = frc_volbot  + zbg(2)
+      frc_sal     = frc_sal     + zbg(3)
+      frc_temtop  = frc_temtop  + zbg(4)
+      frc_tembot  = frc_tembot  + zbg(5)
+
+      ! change units for contents
+      zbg(6)  = zbg(6)  * 1.e-9  ! ice volume (km3)
+      zbg(7)  = zbg(7)  * 1.e-9  ! snw volume (km3)
+      zbg(8)  = zbg(8)  * 1.e-6  ! ice area (km2)
+      zbg(9)  = zbg(9)  * 1.e-9  ! salt content (km3*pss)
+      zbg(10) = zbg(10) * 1.e-20 ! ice heat content (1.e20 J)
+      zbg(11) = zbg(11) * 1.e-20 ! snw heat content (1.e20 J)
+      zbg(12) = zbg(12) * 1.e-9  ! pnd volume (km3)
+      zbg(13) = zbg(13) * 1.e-9  ! pnd lid volume (km3)
+
+      ! change units for trends
+      zbg(14) = zbg(14) * r1_rho0 * 1.e-9  ! freshwater trend (km3)
+      zbg(15) = zbg(15) * r1_rho0 * 1.e-9  ! salt content trend (km3*pss)
+      zbg(16) = zbg(16)           * 1.e-20 ! heat content trend (1.e20 J)
+      ! difference
+      zbg(14) = zbg(14) - ( frc_voltop + frc_volbot )
+      zbg(15) = zbg(15) -   frc_sal
+      zbg(16) = zbg(16) - ( frc_tembot - frc_temtop )
+
+      ! outputs
+      CALL iom_put( 'ibgfrcvoltop' , frc_voltop )   ! vol  forcing ice/snw-atm          (km3 equivalent ocean water)
+      CALL iom_put( 'ibgfrcvolbot' , frc_volbot )   ! vol  forcing ice/snw-ocean        (km3 equivalent ocean water)
+      CALL iom_put( 'ibgfrcsal'    , frc_sal    )   ! sal  forcing                      (psu*km3 equivalent ocean water)
+      CALL iom_put( 'ibgfrctemtop' , frc_temtop )   ! heat on top of ice/snw/ocean      (1.e20 J)
+      CALL iom_put( 'ibgfrctembot' , frc_tembot )   ! heat on top of ocean(below ice)   (1.e20 J)
+      CALL iom_put( 'ibgfrchfxtop' , frc_temtop * z1_e1e2 * 1.e-20 * kt*rn_Dt ) ! heat on top of ice/snw/ocean      (W/m2)
+      CALL iom_put( 'ibgfrchfxbot' , frc_tembot * z1_e1e2 * 1.e-20 * kt*rn_Dt ) ! heat on top of ocean(below ice)   (W/m2)
+
+      CALL iom_put( 'ibgvol_tot'  , zbg(6)  )
+      CALL iom_put( 'sbgvol_tot'  , zbg(7)  )
+      CALL iom_put( 'ibgarea_tot' , zbg(8)  )
+      CALL iom_put( 'ibgsalt_tot' , zbg(9)  )
+      CALL iom_put( 'ibgheat_tot' , zbg(10) )
+      CALL iom_put( 'sbgheat_tot' , zbg(11) )
+      CALL iom_put( 'ipbgvol_tot' , zbg(12) )
+      CALL iom_put( 'ilbgvol_tot' , zbg(13) )
+     
+      CALL iom_put( 'ibgvolume' , zbg(14) )   ! ice/snow volume  drift            (km3 equivalent ocean water)
+      CALL iom_put( 'ibgsaltco' , zbg(15) )   ! ice salt content drift            (psu*km3 equivalent ocean water)
+      CALL iom_put( 'ibgheatco' , zbg(16) )   ! ice/snow heat content drift       (1.e20 J)
+      !
+      ! restarts
       IF( lrst_ice )   CALL ice_dia_rst( 'WRITE', kt_ice )
       !