Changeset 11934


Ignore:
Timestamp:
2019-11-20T10:47:16+01:00 (2 months ago)
Author:
davestorkey
Message:

UKMO/NEMO_4.0.1_momentum_trends : code changes (equivalent to rev 11917 of NEMO 4.0 branch).

Location:
NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends
Files:
8 edited

Legend:

Unmodified
Added
Removed
  • NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/cfgs/SHARED/field_def_nemo-oce.xml

    r11536 r11934  
    806806     <!-- variables available with ln_dyn_trd --> 
    807807     <field id="utrd_hpg"       long_name="i-trend: hydrostatic pressure gradient"          unit="m/s^2"                        /> 
    808      <field id="utrd_spg"       long_name="i-trend: surface     pressure gradient"          unit="m/s^2"                        /> 
     808     <field id="utrd_hpg_corr"  long_name="i-trend: HPG: correction"       unit="m/s^2"    grid_ref="grid_U_2D" /> 
     809     <field id="utrd_spg2d"     long_name="i-trend: surface pressure gradient: true trend"       unit="m/s^2"    grid_ref="grid_U_2D" /> 
    809810     <field id="utrd_spgexp"    long_name="i-trend: surface pressure gradient (explicit)"   unit="m/s^2"                        /> 
    810      <field id="utrd_spgflt"    long_name="i-trend: surface pressure gradient (filtered)"   unit="m/s^2"                        /> 
    811811     <field id="utrd_keg"       long_name="i-trend: KE gradient         or hor. adv."       unit="m/s^2"                        /> 
    812812     <field id="utrd_rvo"       long_name="i-trend: relative  vorticity or metric term"     unit="m/s^2"                        /> 
    813      <field id="utrd_pvo"       long_name="i-trend: planetary vorticity"                    unit="m/s^2"                        /> 
     813     <field id="utrd_pvo"       long_name="i-trend: planetary vorticity: 3D component"      unit="m/s^2"                        /> 
     814     <field id="utrd_pvo2d"     long_name="i-trend: planetary vorticity: 2D component"      unit="m/s^2"    grid_ref="grid_U_2D" /> 
     815     <field id="utrd_pvo_corr"     long_name="i-trend: planetary vorticity: correction"      unit="m/s^2"    grid_ref="grid_U_2D" /> 
    814816     <field id="utrd_zad"       long_name="i-trend: vertical  advection"                    unit="m/s^2"                        /> 
    815817     <field id="utrd_udx"       long_name="i-trend: U.dx[U]"                                unit="m/s^2"                        /> 
     
    819821     <field id="utrd_bfr"       long_name="i-trend: bottom friction (explicit)"             unit="m/s^2"                        />    
    820822     <field id="utrd_bfri"      long_name="i-trend: bottom friction (implicit)"             unit="m/s^2"                        />    
     823     <field id="utrd_bfr2d"     long_name="i-trend: bottom friction: 2D component"      unit="m/s^2"    grid_ref="grid_U_2D" /> 
    821824     <field id="utrd_tot"       long_name="i-trend: total momentum trend before atf"        unit="m/s^2"                        />    
    822825     <field id="utrd_atf"       long_name="i-trend: asselin time filter trend"              unit="m/s^2"                        />    
     826     <!-- thickness weighted versions --> 
     827     <field id="utrd_hpg_e3u"       unit="m2/s^2" > utrd_hpg * e3u </field> 
     828     <field id="utrd_spg_e3u"       unit="m2/s^2" > utrd_spg * e3u </field> 
     829     <field id="utrd_spgexp_e3u"    unit="m2/s^2" > utrd_spgexp * e3u </field> 
     830     <field id="utrd_keg_e3u"       unit="m2/s^2" > utrd_keg * e3u </field> 
     831     <field id="utrd_rvo_e3u"       unit="m2/s^2" > utrd_rvo * e3u </field> 
     832     <field id="utrd_pvo_e3u"       unit="m2/s^2" > utrd_pvo * e3u </field> 
     833     <field id="utrd_zad_e3u"       unit="m2/s^2" > utrd_zad * e3u </field> 
     834     <field id="utrd_ldf_e3u"       unit="m2/s^2" > utrd_ldf * e3u </field> 
     835     <field id="utrd_zdf_e3u"       unit="m2/s^2" > utrd_zdf * e3u </field> 
     836     <field id="utrd_tot_e3u"       unit="m2/s^2" > utrd_tot * e3u </field> 
     837     <field id="utrd_atf_e3u"       unit="m2/s^2" > utrd_atf * e3u </field> 
     838     <field id="utrd_bta_e3u"       unit="m2/s^2" > utrd_bta * e3u </field>   
     839     <field id="utrd_bfr_e3u"       unit="m2/s^2" > utrd_bfr * e3u </field>   
     840     <field id="utrd_bfri_e3u"      unit="m2/s^2" > utrd_bfri * e3u </field>   
    823841   </field_group> 
    824842 
     
    826844     <!-- variables available with ln_dyn_trd --> 
    827845     <field id="vtrd_hpg"       long_name="j-trend: hydrostatic pressure gradient"          unit="m/s^2"                        /> 
     846     <field id="vtrd_hpg_corr"  long_name="j-trend: HPG: correction"       unit="m/s^2"    grid_ref="grid_V_2D" /> 
    828847     <field id="vtrd_spg"       long_name="j-trend: surface     pressure gradient"          unit="m/s^2"                        /> 
     848     <field id="vtrd_spgbt"     long_name="j-trend: surface pressure gradient: barotropic part"  unit="m/s^2"    grid_ref="grid_V_2D" /> 
     849     <field id="vtrd_spg2d"     long_name="j-trend: surface pressure gradient: true trend"       unit="m/s^2"    grid_ref="grid_V_2D" /> 
    829850     <field id="vtrd_spgexp"    long_name="j-trend: surface pressure gradient (explicit)"   unit="m/s^2"                        /> 
    830      <field id="vtrd_spgflt"    long_name="j-trend: surface pressure gradient (filtered)"   unit="m/s^2"                        /> 
    831851     <field id="vtrd_keg"       long_name="j-trend: KE gradient         or hor. adv."       unit="m/s^2"                        /> 
    832852     <field id="vtrd_rvo"       long_name="j-trend: relative  vorticity or metric term"     unit="m/s^2"                        /> 
    833      <field id="vtrd_pvo"       long_name="j-trend: planetary vorticity"                    unit="m/s^2"                        /> 
     853     <field id="vtrd_pvo"       long_name="j-trend: planetary vorticity: 3D component"      unit="m/s^2"                        /> 
     854     <field id="vtrd_pvo2d"     long_name="j-trend: planetary vorticity: 2D component"      unit="m/s^2"    grid_ref="grid_V_2D" /> 
     855     <field id="vtrd_pvo_corr"     long_name="j-trend: planetary vorticity: correction"      unit="m/s^2"    grid_ref="grid_V_2D" /> 
    834856     <field id="vtrd_zad"       long_name="j-trend: vertical  advection"                    unit="m/s^2"                        /> 
    835      <field id="vtrd_vdy"       long_name="i-trend: V.dx[V]"                                unit="m/s^2"                        /> 
     857     <field id="vtrd_vdy"       long_name="j-trend: V.dx[V]"                                unit="m/s^2"                        /> 
    836858     <field id="vtrd_ldf"       long_name="j-trend: lateral   diffusion"                    unit="m/s^2"                        /> 
    837859     <field id="vtrd_zdf"       long_name="j-trend: vertical  diffusion"                    unit="m/s^2"                        /> 
     
    839861     <field id="vtrd_bfr"       long_name="j-trend: bottom friction (explicit)"             unit="m/s^2"                        />    
    840862     <field id="vtrd_bfri"      long_name="j-trend: bottom friction (implicit)"             unit="m/s^2"                        />    
     863     <field id="vtrd_bfr2d"     long_name="j-trend: bottom friction: 2D component"      unit="m/s^2"    grid_ref="grid_V_2D" /> 
    841864     <field id="vtrd_tot"       long_name="j-trend: total momentum trend before atf"        unit="m/s^2"                        />    
    842865     <field id="vtrd_atf"       long_name="j-trend: asselin time filter trend"              unit="m/s^2"                        />    
     866     <!-- thickness weighted versions --> 
     867     <field id="vtrd_hpg_e3v"       unit="m2/s^2" > vtrd_hpg * e3v </field> 
     868     <field id="vtrd_spg_e3v"       unit="m2/s^2" > vtrd_spg * e3v </field> 
     869     <field id="vtrd_spgexp_e3v"    unit="m2/s^2" > vtrd_spgexp * e3v </field> 
     870     <field id="vtrd_keg_e3v"       unit="m2/s^2" > vtrd_keg * e3v </field> 
     871     <field id="vtrd_rvo_e3v"       unit="m2/s^2" > vtrd_rvo * e3v </field> 
     872     <field id="vtrd_pvo_e3v"       unit="m2/s^2" > vtrd_pvo * e3v </field> 
     873     <field id="vtrd_zad_e3v"       unit="m2/s^2" > vtrd_zad * e3v </field> 
     874     <field id="vtrd_ldf_e3v"       unit="m2/s^2" > vtrd_ldf * e3v </field> 
     875     <field id="vtrd_zdf_e3v"       unit="m2/s^2" > vtrd_zdf * e3v </field> 
     876     <field id="vtrd_tot_e3v"       unit="m2/s^2" > vtrd_tot * e3v </field> 
     877     <field id="vtrd_atf_e3v"       unit="m2/s^2" > vtrd_atf * e3v </field> 
     878     <field id="vtrd_bta_e3v"       unit="m2/s^2" > vtrd_bta * e3v </field>   
     879     <field id="vtrd_bfr_e3v"       unit="m2/s^2" > vtrd_bfr * e3v </field>   
     880     <field id="vtrd_bfri_e3v"      unit="m2/s^2" > vtrd_bfri * e3v </field>   
    843881   </field_group> 
    844882 
  • NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/DYN/dynhpg.F90

    r11715 r11934  
    115115         ztrdu(:,:,:) = ua(:,:,:) - ztrdu(:,:,:) 
    116116         ztrdv(:,:,:) = va(:,:,:) - ztrdv(:,:,:) 
    117          CALL trd_dyn( ztrdu, ztrdv, jpdyn_hpg, kt ) 
     117         CALL trd_dyn( ztrdu, ztrdv, jpdyn_hpg_save, kt ) 
    118118         DEALLOCATE( ztrdu , ztrdv ) 
    119119      ENDIF 
  • NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/DYN/dynspg.F90

    r11715 r11934  
    8181      IF( ln_timing )   CALL timing_start('dyn_spg') 
    8282      ! 
    83       IF( l_trddyn )   THEN                      ! temporary save of ta and sa trends 
     83      IF( l_trddyn .AND. nspg == np_EXP )   THEN                      ! temporary save of ta and sa trends 
    8484         ALLOCATE( ztrdu(jpi,jpj,jpk) , ztrdv(jpi,jpj,jpk) )  
    8585         ztrdu(:,:,:) = ua(:,:,:) 
     
    165165      END SELECT 
    166166      !                     
    167       IF( l_trddyn )   THEN                  ! save the surface pressure gradient trends for further diagnostics 
     167      IF( l_trddyn .AND. nspg == np_EXP )   THEN     ! save the surface pressure gradient trends for further diagnostics 
    168168         ztrdu(:,:,:) = ua(:,:,:) - ztrdu(:,:,:) 
    169169         ztrdv(:,:,:) = va(:,:,:) - ztrdv(:,:,:) 
    170170         CALL trd_dyn( ztrdu, ztrdv, jpdyn_spg, kt ) 
     171         ! In this case we also need to finalise the write-out of the PVO 3D diagnostic with zero correction 
     172         ztrdu(:,:,1) = 0._wp 
     173         ztrdv(:,:,1) = 0._wp 
     174         CALL trd_dyn( ztrdu(:,:,1), ztrdv(:,:,1), jpdyn_pvo, kt ) 
    171175         DEALLOCATE( ztrdu , ztrdv )  
    172176      ENDIF 
  • NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/DYN/dynspg_ts.F90

    r11715 r11934  
    6363   USE restart         ! only for lrst_oce 
    6464   USE diatmb          ! Top,middle,bottom output 
     65   USE trd_oce        ! trends: ocean variables 
     66   USE trddyn         ! trend manager: dynamics 
    6567 
    6668   USE iom   ! to remove 
     
    170172      REAL(wp), ALLOCATABLE, DIMENSION(:,:) :: ztwdmask, zuwdmask, zvwdmask ! ROMS wetting and drying masks at t,u,v points 
    171173      REAL(wp), ALLOCATABLE, DIMENSION(:,:) :: zuwdav2, zvwdav2    ! averages over the sub-steps of zuwdmask and zvwdmask 
     174      REAL(wp), ALLOCATABLE, DIMENSION(:,:) :: zspgtrdu, zspgtrdv, zpvotrdu, zpvotrdv  ! SPG and PVO trends (if l_trddyn) 
     175      REAL(wp), ALLOCATABLE, DIMENSION(:,:) :: ztautrdu, ztautrdv, zbfrtrdu, zbfrtrdv  ! TAU and BFR trends (if l_trddyn) 
    172176      !!---------------------------------------------------------------------- 
    173177      ! 
     
    175179      !                                         !* Allocate temporary arrays 
    176180      IF( ln_wd_dl ) ALLOCATE( ztwdmask(jpi,jpj), zuwdmask(jpi,jpj), zvwdmask(jpi,jpj), zuwdav2(jpi,jpj), zvwdav2(jpi,jpj)) 
     181      ! 
     182      IF( l_trddyn ) THEN 
     183          ALLOCATE( zspgtrdu(jpi,jpj), zspgtrdv(jpi,jpj), zpvotrdu(jpi,jpj), zpvotrdv(jpi,jpj), & 
     184         &          ztautrdu(jpi,jpj), ztautrdv(jpi,jpj), zbfrtrdu(jpi,jpj), zbfrtrdv(jpi,jpj) ) 
     185          zspgtrdu(:,:) = 0._wp 
     186          zspgtrdv(:,:) = 0._wp 
     187          zpvotrdu(:,:) = 0._wp 
     188          zpvotrdv(:,:) = 0._wp 
     189          ztautrdu(:,:) = 0._wp 
     190          ztautrdv(:,:) = 0._wp 
     191          zbfrtrdu(:,:) = 0._wp 
     192          zbfrtrdv(:,:) = 0._wp 
     193      ENDIF 
     194      ! 
     195      zu_trd(:,:) = 0._wp 
     196      zv_trd(:,:) = 0._wp 
     197      zu_spg(:,:) = 0._wp 
     198      zv_spg(:,:) = 0._wp 
    177199      ! 
    178200      zmdi=1.e+20                               !  missing data indicator for masking 
     
    253275         &                               zu_trd, zv_trd   )   ! ==>> out 
    254276      ! 
     277      IF( l_trddyn ) THEN 
     278         ! send correction to baroclinic planetary vorticity trend to trd_dyn 
     279         CALL trd_dyn( zu_trd, zv_trd, jpdyn_pvo_corr, kt ) 
     280      ENDIF 
     281      ! 
    255282      IF( .NOT.ln_linssh ) THEN                 !* surface pressure gradient   (variable volume only) 
     283         ! 
     284         IF( l_trddyn ) THEN 
     285            zspgtrdu(:,:) = zu_trd(:,:) 
     286            zspgtrdv(:,:) = zv_trd(:,:) 
     287         ENDIF 
    256288         ! 
    257289         IF( ln_wd_il ) THEN                       ! W/D : limiter applied to spgspg 
     
    274306         ENDIF 
    275307         ! 
     308         IF( l_trddyn ) THEN 
     309            zspgtrdu(:,:) = zu_trd(:,:) - zspgtrdu(:,:)  
     310            zspgtrdv(:,:) = zv_trd(:,:) - zspgtrdv(:,:)  
     311            ! send correction to HPG trend to trd_dyn 
     312            CALL trd_dyn( zspgtrdu, zspgtrdv, jpdyn_hpg_corr, kt ) 
     313            ! reset temporary arrays for use later 
     314            zspgtrdu(:,:) = 0._wp 
     315            zspgtrdv(:,:) = 0._wp 
     316         ENDIF 
     317         ! 
    276318      ENDIF 
    277319      ! 
     
    283325      END DO  
    284326      ! 
     327      IF( l_trddyn ) THEN 
     328         zbfrtrdu(:,:) = zu_frc(:,:) 
     329         zbfrtrdv(:,:) = zv_frc(:,:) 
     330      ENDIF 
    285331      !                                   !=  Add bottom stress contribution from baroclinic velocities  =! 
    286332      !                                   !  -----------------------------------------------------------  ! 
    287333      CALL dyn_drg_init( zu_frc, zv_frc,  zCdU_u, zCdU_v )      ! also provide the barotropic drag coefficients 
     334      ! 
     335      IF( l_trddyn ) THEN 
     336         ! bottom friction trend diagnostic: bottom friction due to baroclinic currents 
     337         zbfrtrdu(:,:) = zu_frc(:,:) - zbfrtrdu(:,:) 
     338         zbfrtrdv(:,:) = zv_frc(:,:) - zbfrtrdv(:,:)  
     339      ENDIF 
    288340      ! 
    289341      !                                   !=  Add atmospheric pressure forcing  =! 
     
    312364      !                                   !=  Add atmospheric pressure forcing  =! 
    313365      !                                   !  ----------------------------------  ! 
     366      IF( l_trddyn ) THEN 
     367         ztautrdu(:,:) = zu_frc(:,:) 
     368         ztautrdv(:,:) = zv_frc(:,:) 
     369      ENDIF 
     370      ! 
    314371      IF( ln_bt_fw ) THEN                        ! Add wind forcing 
    315372         DO jj = 2, jpjm1 
     
    329386      ENDIF   
    330387      ! 
     388      IF( l_trddyn ) THEN 
     389         ! wind stress trend diagnostic 
     390         ztautrdu(:,:) = zu_frc(:,:) - ztautrdu(:,:) 
     391         ztautrdv(:,:) = zv_frc(:,:) - ztautrdv(:,:)  
     392      ENDIF 
    331393      !              !----------------! 
    332394      !              !==  sssh_frc  ==!   Right-Hand-Side of the barotropic ssh equation   (over the FULL domain) 
     
    591653         ENDIF 
    592654         ! 
     655         IF( l_trddyn ) THEN 
     656            za2 = wgtbtp2(jn) 
     657            zspgtrdu(:,:) = zspgtrdu(:,:) + za2 * zu_spg(:,:) * ssumask(:,:) 
     658            zspgtrdv(:,:) = zspgtrdv(:,:) + za2 * zv_spg(:,:) * ssvmask(:,:) 
     659         ENDIF 
     660         ! 
    593661         ! Add Coriolis trend: 
    594662         ! zwz array below or triads normally depend on sea level with ln_linssh=F and should be updated 
     
    596664         ! Recall that zhU and zhV hold fluxes at jn+0.5 (extrapolated not backward interpolated) 
    597665         CALL dyn_cor_2d( zhup2_e, zhvp2_e, ua_e, va_e, zhU, zhV,    zu_trd, zv_trd   ) 
     666         ! 
     667         IF( l_trddyn ) THEN 
     668            za2 = wgtbtp2(jn) 
     669            zpvotrdu(:,:) = zpvotrdu(:,:) + za2 * zu_trd(:,:) * ssumask(:,:) 
     670            zpvotrdv(:,:) = zpvotrdv(:,:) + za2 * zv_trd(:,:) * ssvmask(:,:) 
     671         ENDIF 
    598672         ! 
    599673         ! Add tidal astronomical forcing if defined 
     
    616690               END DO 
    617691            END DO 
     692            IF( l_trddyn ) THEN 
     693               za2 = wgtbtp2(jn) 
     694               zbfrtrdu(:,:) = zbfrtrdu(:,:) + za2 * zCdU_u(:,:) * un_e(:,:) * hur_e(:,:) 
     695               zbfrtrdv(:,:) = zbfrtrdv(:,:) + za2 * zCdU_v(:,:) * vn_e(:,:) * hvr_e(:,:) 
     696            ENDIF 
    618697         ENDIF 
    619698         ! 
     
    835914      IF( ln_wd_il )   DEALLOCATE( zcpx, zcpy ) 
    836915      IF( ln_wd_dl )   DEALLOCATE( ztwdmask, zuwdmask, zvwdmask, zuwdav2, zvwdav2 ) 
     916      ! 
     917      IF( l_trddyn ) THEN 
     918         CALL trd_dyn( zspgtrdu, zspgtrdv, jpdyn_spg, kt ) 
     919         CALL trd_dyn( zpvotrdu, zpvotrdv, jpdyn_pvo, kt ) 
     920         CALL trd_dyn( ztautrdu, ztautrdv, jpdyn_tau, kt ) 
     921         CALL trd_dyn( zbfrtrdu, zbfrtrdv, jpdyn_bfr, kt ) 
     922         DEALLOCATE( zspgtrdu, zspgtrdv, zpvotrdu, zpvotrdv, ztautrdu, ztautrdv, zbfrtrdu, zbfrtrdv ) 
     923      ENDIF 
    837924      ! 
    838925      IF( ln_diatmb ) THEN 
  • NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/DYN/dynvor.F90

    r11715 r11934  
    133133         ztrdu(:,:,:) = ua(:,:,:) - ztrdu(:,:,:) 
    134134         ztrdv(:,:,:) = va(:,:,:) - ztrdv(:,:,:) 
    135          CALL trd_dyn( ztrdu, ztrdv, jpdyn_pvo, kt ) 
     135         CALL trd_dyn( ztrdu, ztrdv, jpdyn_pvo_save, kt ) 
    136136         ! 
    137137         IF( n_dynadv /= np_LIN_dyn ) THEN   !* relative vorticity or metric trend (only in non-linear case) 
  • NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/DYN/dynzdf.F90

    r11715 r11934  
    2525   ! 
    2626   USE in_out_manager ! I/O manager 
     27   USE iom             ! IOM library 
    2728   USE lib_mpp        ! MPP library 
    2829   USE prtctl         ! Print control 
     
    6970      INTEGER  ::   ji, jj, jk         ! dummy loop indices 
    7071      INTEGER  ::   iku, ikv           ! local integers 
     72      INTEGER  ::   ikbu, ikbv         ! local integers 
    7173      REAL(wp) ::   zzwi, ze3ua, zdt   ! local scalars 
    7274      REAL(wp) ::   zzws, ze3va        !   -      - 
     
    99101      ! 
    100102      ! 
    101       IF( l_trddyn )   THEN         !* temporary save of ta and sa trends 
    102          ALLOCATE( ztrdu(jpi,jpj,jpk), ztrdv(jpi,jpj,jpk) )  
    103          ztrdu(:,:,:) = ua(:,:,:) 
    104          ztrdv(:,:,:) = va(:,:,:) 
    105       ENDIF 
    106       ! 
    107103      !              !==  RHS: Leap-Frog time stepping on all trends but the vertical mixing  ==!   (put in ua,va) 
    108104      ! 
     
    131127            va(:,:,jk) = ( va(:,:,jk) - va_b(:,:) ) * vmask(:,:,jk) 
    132128         END DO 
     129         IF( l_trddyn )   THEN         !* temporary save of ta and sa trends 
     130            ALLOCATE( ztrdu(jpi,jpj,jpk), ztrdv(jpi,jpj,jpk) )  
     131            ztrdu(:,:,:) = ua(:,:,:) 
     132            ztrdv(:,:,:) = va(:,:,:) 
     133         ENDIF 
    133134         DO jj = 2, jpjm1        ! Add bottom/top stress due to barotropic component only 
    134135            DO ji = fs_2, fs_jpim1   ! vector opt. 
     
    141142            END DO 
    142143         END DO 
     144         IF( l_trddyn )   THEN                      ! save explicit part of bottom friction trends 
     145            ztrdu(:,:,:) = ( ua(:,:,:) - ztrdu(:,:,:) ) / r2dt  
     146            ztrdv(:,:,:) = ( va(:,:,:) - ztrdv(:,:,:) ) / r2dt  
     147            CALL trd_dyn( ztrdu, ztrdv, jpdyn_bfr, kt ) 
     148         ENDIF 
    143149         IF( ln_isfcav ) THEN    ! Ocean cavities (ISF) 
    144150            DO jj = 2, jpjm1         
     
    153159            END DO 
    154160         END IF 
     161      ENDIF 
     162      ! 
     163      IF( l_trddyn )   THEN         !* temporary save of ta and sa trends 
     164         IF( .NOT. ALLOCATED(ztrdu) ) ALLOCATE( ztrdu(jpi,jpj,jpk), ztrdv(jpi,jpj,jpk) )  
     165         ztrdu(:,:,:) = ua(:,:,:) 
     166         ztrdv(:,:,:) = va(:,:,:) 
    155167      ENDIF 
    156168      ! 
     
    486498      ! 
    487499      IF( l_trddyn )   THEN                      ! save the vertical diffusive trends for further diagnostics 
    488          ztrdu(:,:,:) = ( ua(:,:,:) - ub(:,:,:) ) / r2dt - ztrdu(:,:,:) 
    489          ztrdv(:,:,:) = ( va(:,:,:) - vb(:,:,:) ) / r2dt - ztrdv(:,:,:) 
     500 
     501         ztrdu(:,:,:) = ( ua(:,:,:) - ztrdu(:,:,:) ) / r2dt  
     502         ztrdv(:,:,:) = ( va(:,:,:) - ztrdv(:,:,:) ) / r2dt  
    490503         CALL trd_dyn( ztrdu, ztrdv, jpdyn_zdf, kt ) 
     504         ! 
     505         IF( ln_drgimp ) THEN 
     506            ztrdu(:,:,:) = 0._wp    ;   ztrdv(:,:,:) = 0._wp 
     507            DO jk = 1, jpkm1 
     508               DO jj = 2, jpjm1 
     509                  DO ji = 2, jpim1 
     510                     ikbu = mbku(ji,jj)          ! deepest ocean u- & v-levels 
     511                     ikbv = mbkv(ji,jj) 
     512                     ztrdu(ji,jj,jk) = 0.5 * ( rCdU_bot(ji+1,jj) + rCdU_bot(ji,jj) )  &  
     513     &                               * un(ji,jj,ikbu) / e3u_n(ji,jj,ikbu) 
     514                     ztrdv(ji,jj,jk) = 0.5 * ( rCdU_bot(ji,jj+1) + rCdU_bot(ji,jj) )  & 
     515     &                               * vn(ji,jj,ikbv) / e3v_n(ji,jj,ikbv) 
     516                  END DO 
     517               END DO 
     518            END DO 
     519            CALL trd_dyn( ztrdu, ztrdv, jpdyn_bfri, kt ) 
     520         ENDIF 
     521         ! 
    491522         DEALLOCATE( ztrdu, ztrdv )  
    492523      ENDIF 
  • NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/TRD/trd_oce.F90

    r11715 r11934  
    5959   ! 
    6060   !                                                  !!!* Momentum trends indices 
    61    INTEGER, PUBLIC, PARAMETER ::   jptot_dyn  = 13     !: Total trend nb: change it when adding/removing one indice below 
     61   INTEGER, PUBLIC, PARAMETER ::   jptot_dyn  = 17     !: Total trend nb: change it when adding/removing one indice below 
    6262   !                               ===============     !   
    63    INTEGER, PUBLIC, PARAMETER ::   jpdyn_hpg  =  1     !: hydrostatic pressure gradient  
    64    INTEGER, PUBLIC, PARAMETER ::   jpdyn_spg  =  2     !: surface     pressure gradient 
    65    INTEGER, PUBLIC, PARAMETER ::   jpdyn_keg  =  3     !: kinetic energy gradient  or horizontal advection 
    66    INTEGER, PUBLIC, PARAMETER ::   jpdyn_rvo  =  4     !: relative  vorticity      or metric term 
    67    INTEGER, PUBLIC, PARAMETER ::   jpdyn_pvo  =  5     !: planetary vorticity 
    68    INTEGER, PUBLIC, PARAMETER ::   jpdyn_zad  =  6     !: vertical advection 
    69    INTEGER, PUBLIC, PARAMETER ::   jpdyn_ldf  =  7     !: horizontal diffusion    
    70    INTEGER, PUBLIC, PARAMETER ::   jpdyn_zdf  =  8     !: vertical   diffusion 
    71    INTEGER, PUBLIC, PARAMETER ::   jpdyn_bfr  =  9     !: bottom  stress  
    72    INTEGER, PUBLIC, PARAMETER ::   jpdyn_atf  = 10     !: Asselin time filter 
    73    INTEGER, PUBLIC, PARAMETER ::   jpdyn_tau  = 11     !: surface stress 
    74    INTEGER, PUBLIC, PARAMETER ::   jpdyn_bfri = 12     !: implicit bottom friction (ln_drgimp=.TRUE.) 
    75    INTEGER, PUBLIC, PARAMETER ::   jpdyn_ken  = 13     !: use for calculation of KE 
     63   INTEGER, PUBLIC, PARAMETER ::   jpdyn_hpg   =  1     !: hydrostatic pressure gradient  
     64   INTEGER, PUBLIC, PARAMETER ::   jpdyn_hpg_save =  2  !: hydrostatic pressure gradient (saved value) 
     65   INTEGER, PUBLIC, PARAMETER ::   jpdyn_hpg_corr =  3  !: hydrostatic pressure gradient (initial correction) 
     66   INTEGER, PUBLIC, PARAMETER ::   jpdyn_spg   =  4     !: surface     pressure gradient 
     67   INTEGER, PUBLIC, PARAMETER ::   jpdyn_keg   =  5     !: kinetic energy gradient  or horizontal advection 
     68   INTEGER, PUBLIC, PARAMETER ::   jpdyn_rvo   =  6     !: relative  vorticity      or metric term 
     69   INTEGER, PUBLIC, PARAMETER ::   jpdyn_pvo   =  7     !: planetary vorticity 
     70   INTEGER, PUBLIC, PARAMETER ::   jpdyn_pvo_save =  8  !: planetary vorticity (saved value) 
     71   INTEGER, PUBLIC, PARAMETER ::   jpdyn_pvo_corr =  9  !: planetary vorticity (initial correction) 
     72   INTEGER, PUBLIC, PARAMETER ::   jpdyn_zad   = 10     !: vertical advection 
     73   INTEGER, PUBLIC, PARAMETER ::   jpdyn_ldf   = 11     !: horizontal diffusion    
     74   INTEGER, PUBLIC, PARAMETER ::   jpdyn_zdf   = 12     !: vertical   diffusion 
     75   INTEGER, PUBLIC, PARAMETER ::   jpdyn_bfr   = 13     !: bottom  stress  
     76   INTEGER, PUBLIC, PARAMETER ::   jpdyn_atf   = 14     !: Asselin time filter 
     77   INTEGER, PUBLIC, PARAMETER ::   jpdyn_tau   = 15     !: surface stress 
     78   INTEGER, PUBLIC, PARAMETER ::   jpdyn_bfri  = 16     !: implicit bottom friction (ln_drgimp=.TRUE.) 
     79   INTEGER, PUBLIC, PARAMETER ::   jpdyn_ken   = 17     !: use for calculation of KE 
    7680   ! 
    7781   !!---------------------------------------------------------------------- 
  • NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/TRD/trddyn.F90

    r11715 r11934  
    1818   USE sbc_oce        ! surface boundary condition: ocean 
    1919   USE zdf_oce        ! ocean vertical physics: variables 
    20 !!gm   USE zdfdrg         ! ocean vertical physics: bottom friction 
    2120   USE trd_oce        ! trends: ocean variables 
    2221   USE trdken         ! trends: Kinetic ENergy  
     
    3534   PUBLIC trd_dyn        ! called by all dynXXX modules 
    3635 
     36   INTERFACE trd_dyn 
     37      module procedure trd_dyn_3d, trd_dyn_2d 
     38   END INTERFACE 
     39 
     40   REAL(wp), ALLOCATABLE, DIMENSION(:,:,:), SAVE :: zutrd_hpg, zvtrd_hpg 
     41   REAL(wp), ALLOCATABLE, DIMENSION(:,:,:), SAVE :: zutrd_pvo, zvtrd_pvo 
     42   REAL(wp), ALLOCATABLE, DIMENSION(:,:,:), SAVE :: zutrd_bfr, zvtrd_bfr 
     43   REAL(wp), ALLOCATABLE, DIMENSION(:,:)  , SAVE :: zutrd_tau, zvtrd_tau 
     44 
    3745   !! * Substitutions 
    3846#  include "vectopt_loop_substitute.h90" 
     
    4452CONTAINS 
    4553 
    46    SUBROUTINE trd_dyn( putrd, pvtrd, ktrd, kt ) 
     54   SUBROUTINE trd_dyn_3d( putrd, pvtrd, ktrd, kt ) 
    4755      !!--------------------------------------------------------------------- 
    48       !!                  ***  ROUTINE trd_mod  *** 
     56      !!                  ***  ROUTINE trd_dyn_3d  *** 
    4957      !!  
    5058      !! ** Purpose :   Dispatch momentum trend computation, e.g. 3D output,  
     
    5563      INTEGER                   , INTENT(in   ) ::   ktrd           ! trend index 
    5664      INTEGER                   , INTENT(in   ) ::   kt             ! time step 
     65      REAL(wp), ALLOCATABLE, DIMENSION(:,:)     ::   zue, zve       ! temporary 2D arrays 
     66      INTEGER                                   ::   jk 
    5767      !!---------------------------------------------------------------------- 
    5868      ! 
     
    6373!!gm NB : here a lbc_lnk should probably be added 
    6474 
     75      SELECT CASE( ktrd ) 
     76      CASE( jpdyn_hpg_save )  
     77         ! 
     78         ! save 3D HPG trends to possibly have barotropic part corrected later before writing out 
     79         ALLOCATE( zutrd_hpg(jpi,jpj,jpk), zvtrd_hpg(jpi,jpj,jpk) ) 
     80         zutrd_hpg(:,:,:) = putrd(:,:,:) 
     81         zvtrd_hpg(:,:,:) = pvtrd(:,:,:) 
     82 
     83      CASE( jpdyn_pvo_save )  
     84         ! 
     85         ! save 3D coriolis trends to possibly have barotropic part corrected later before writing out 
     86         ALLOCATE( zutrd_pvo(jpi,jpj,jpk), zvtrd_pvo(jpi,jpj,jpk) ) 
     87         zutrd_pvo(:,:,:) = putrd(:,:,:) 
     88         zvtrd_pvo(:,:,:) = pvtrd(:,:,:) 
     89 
     90      CASE( jpdyn_spg )  
     91         ! For explicit scheme SPG trends come here as 3D fields 
     92         ! Add SPG trend to 3D HPG trend and also output as 2D diagnostic in own right. 
     93         CALL trd_dyn_iom_2d( putrd(:,:,1), pvtrd(:,:,1), jpdyn_spg, kt )  
     94         putrd(:,:,:) = putrd(:,:,:) + zutrd_hpg(:,:,:)  
     95         pvtrd(:,:,:) = pvtrd(:,:,:) + zvtrd_hpg(:,:,:)  
     96         DEALLOCATE( zutrd_hpg, zvtrd_hpg ) 
     97 
     98      CASE( jpdyn_bfr ) 
     99         ! 
     100         ! Add 3D part of BFR trend minus its depth-mean part to depth-mean trend already saved. 
     101         ALLOCATE( zue(jpi,jpj), zve(jpi,jpj) ) 
     102         zue(:,:) = e3u_a(:,:,1) * putrd(:,:,1) * umask(:,:,1) 
     103         zve(:,:) = e3v_a(:,:,1) * pvtrd(:,:,1) * vmask(:,:,1) 
     104         DO jk = 2, jpkm1 
     105            zue(:,:) = zue(:,:) + e3u_a(:,:,jk) * putrd(:,:,jk) * umask(:,:,jk) 
     106            zve(:,:) = zve(:,:) + e3v_a(:,:,jk) * pvtrd(:,:,jk) * vmask(:,:,jk) 
     107         END DO 
     108         DO jk = 1, jpkm1 
     109            putrd(:,:,jk) = zutrd_bfr(:,:,jk) + putrd(:,:,jk) - zue(:,:) * r1_hu_a(:,:) 
     110            pvtrd(:,:,jk) = zvtrd_bfr(:,:,jk) + pvtrd(:,:,jk) - zve(:,:) * r1_hv_a(:,:) 
     111         END DO 
     112         ! Update locally saved BFR trends to add to ZDF trend. 
     113         zutrd_bfr(:,:,:) = putrd(:,:,:)  
     114         zvtrd_bfr(:,:,:) = pvtrd(:,:,:) 
     115 
     116      CASE( jpdyn_zdf )  
     117         ! ZDF trend: Remove barotropic component and add wind stress and bottom friction 
     118         !            trends from dynspg_ts. Also adding on the bottom stress for the  
     119         !            baroclinic solution in the case of explicit bottom friction.  
     120         ALLOCATE( zue(jpi,jpj), zve(jpi,jpj) ) 
     121         zue(:,:) = e3u_a(:,:,1) * putrd(:,:,1) * umask(:,:,1) 
     122         zve(:,:) = e3v_a(:,:,1) * pvtrd(:,:,1) * vmask(:,:,1) 
     123         DO jk = 2, jpkm1 
     124            zue(:,:) = zue(:,:) + e3u_a(:,:,jk) * putrd(:,:,jk) * umask(:,:,jk) 
     125            zve(:,:) = zve(:,:) + e3v_a(:,:,jk) * pvtrd(:,:,jk) * vmask(:,:,jk) 
     126         END DO 
     127         DO jk = 1, jpkm1 
     128            putrd(:,:,jk) = zutrd_tau(:,:) + zutrd_bfr(:,:,jk) + putrd(:,:,jk) - zue(:,:) * r1_hu_a(:,:) 
     129            pvtrd(:,:,jk) = zutrd_tau(:,:) + zvtrd_bfr(:,:,jk) + pvtrd(:,:,jk) - zve(:,:) * r1_hv_a(:,:) 
     130         END DO 
     131         DEALLOCATE( zue, zve, zutrd_tau, zvtrd_tau, zutrd_bfr, zvtrd_bfr ) 
     132 
     133      END SELECT 
     134 
     135      IF ( ktrd /= jpdyn_hpg_save .AND. ktrd /= jpdyn_pvo_save ) THEN 
     136         ! 
     137         !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     138         !   3D output of momentum and/or tracers trends using IOM interface 
     139         !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     140         IF( ln_dyn_trd )   CALL trd_dyn_iom_3d( putrd, pvtrd, ktrd, kt ) 
     141 
     142         !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     143         !  Integral Constraints Properties for momentum and/or tracers trends 
     144         !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     145         IF( ln_glo_trd )   CALL trd_glo( putrd, pvtrd, ktrd, 'DYN', kt ) 
     146 
     147         !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     148         !  Kinetic Energy trends 
     149         !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> 
     150         IF( ln_KE_trd  )   CALL trd_ken( putrd, pvtrd, ktrd, kt ) 
     151 
     152         !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     153         !  Vorticity trends 
     154         !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> 
     155         IF( ln_vor_trd )   CALL trd_vor( putrd, pvtrd, ktrd, kt ) 
     156 
     157         !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> 
     158         !  Mixed layer trends for active tracers 
     159         !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     160         !!gm      IF( ln_dyn_mxl )   CALL trd_mxl_dyn    
     161         ! 
     162      ENDIF 
     163      ! 
     164   END SUBROUTINE trd_dyn_3d 
     165 
     166 
     167   SUBROUTINE trd_dyn_2d( putrd, pvtrd, ktrd, kt ) 
     168      !!--------------------------------------------------------------------- 
     169      !!                  ***  ROUTINE trd_mod  *** 
     170      !!  
     171      !! ** Purpose :   Dispatch momentum trend computation, e.g. 2D output,  
     172      !!              integral constraints, barotropic vorticity, kinetic enrgy,  
     173      !!              and/or mixed layer budget. 
     174      !!---------------------------------------------------------------------- 
     175      REAL(wp), DIMENSION(:,:), INTENT(inout) ::   putrd, pvtrd   ! U and V trends  
     176      INTEGER                 , INTENT(in   ) ::   ktrd           ! trend index 
     177      INTEGER                 , INTENT(in   ) ::   kt             ! time step 
     178      INTEGER                                 ::   jk 
     179      !!---------------------------------------------------------------------- 
     180      ! 
     181      putrd(:,:) = putrd(:,:) * umask(:,:,1)                       ! mask the trends 
     182      pvtrd(:,:) = pvtrd(:,:) * vmask(:,:,1) 
     183      ! 
     184 
     185!!gm NB : here a lbc_lnk should probably be added 
     186 
     187      SELECT CASE(ktrd) 
     188 
     189      CASE ( jpdyn_hpg_corr ) 
     190         ! 
     191         ! Remove "first-guess" SPG trend from 3D HPG trend.  
     192         DO jk = 1, jpkm1 
     193            zutrd_hpg(:,:,jk) = zutrd_hpg(:,:,jk) - putrd(:,:) 
     194            zvtrd_hpg(:,:,jk) = zvtrd_hpg(:,:,jk) - pvtrd(:,:) 
     195         ENDDO 
     196 
     197      CASE( jpdyn_pvo_corr ) 
     198         ! 
     199         ! Remove "first-guess" barotropic coriolis trend from 3D PVO trend.  
     200         DO jk = 1, jpkm1 
     201            zutrd_pvo(:,:,jk) = zutrd_pvo(:,:,jk) - putrd(:,:) 
     202            zvtrd_pvo(:,:,jk) = zvtrd_pvo(:,:,jk) - pvtrd(:,:) 
     203         ENDDO 
     204 
     205      CASE( jpdyn_spg ) 
     206          ! 
     207          ! For split-explicit scheme SPG trends come here as 2D fields 
     208          ! Add SPG trend to 3D HPG trend and also output as 2D diagnostic in own right. 
     209          DO jk = 1, jpkm1 
     210             zutrd_hpg(:,:,jk) = zutrd_hpg(:,:,jk) + putrd(:,:) 
     211             zvtrd_hpg(:,:,jk) = zvtrd_hpg(:,:,jk) + pvtrd(:,:) 
     212          ENDDO 
     213          CALL trd_dyn_3d( zutrd_hpg, zvtrd_hpg, jpdyn_hpg, kt ) 
     214          DEALLOCATE( zutrd_hpg, zvtrd_hpg ) 
     215 
     216      CASE( jpdyn_pvo ) 
     217          ! 
     218          ! Add 2D PVO trend to 3D PVO trend and also output as diagnostic in own right. 
     219          DO jk = 1, jpkm1 
     220             zutrd_pvo(:,:,jk) = zutrd_pvo(:,:,jk) + putrd(:,:) 
     221             zvtrd_pvo(:,:,jk) = zvtrd_pvo(:,:,jk) + pvtrd(:,:) 
     222          ENDDO 
     223          CALL trd_dyn_3d( zutrd_pvo, zvtrd_pvo, jpdyn_pvo, kt ) 
     224          DEALLOCATE( zutrd_pvo, zvtrd_pvo ) 
     225 
     226      CASE( jpdyn_tau ) 
     227          ! 
     228          ! Save 2D wind forcing trend locally (to be added to ZDF trend) 
     229          ! and output as a trend in its own right. 
     230          ALLOCATE( zutrd_tau(jpi,jpj), zvtrd_tau(jpi,jpj) ) 
     231          zutrd_tau(:,:) = putrd(:,:) 
     232          zvtrd_tau(:,:) = pvtrd(:,:) 
     233 
     234      CASE( jpdyn_bfr ) 
     235          ! 
     236          ! Create 3D BFR trend from 2D field and also output 2D field as diagnostic in own right. 
     237          ALLOCATE( zutrd_bfr(jpi,jpj,jpk), zvtrd_bfr(jpi,jpj,jpk) ) 
     238          zutrd_bfr(:,:,:) = 0.0 
     239          zvtrd_bfr(:,:,:) = 0.0 
     240          DO jk = 1, jpkm1 
     241             zutrd_bfr(:,:,jk) = putrd(:,:) * umask(:,:,jk) 
     242             zvtrd_bfr(:,:,jk) = pvtrd(:,:) * vmask(:,:,jk) 
     243          ENDDO 
     244 
     245      END SELECT 
     246 
    65247      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
    66       !   3D output of momentum and/or tracers trends using IOM interface 
     248      !   2D output of momentum and/or tracers trends using IOM interface 
    67249      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
    68       IF( ln_dyn_trd )   CALL trd_dyn_iom( putrd, pvtrd, ktrd, kt ) 
     250      IF( ln_dyn_trd )   CALL trd_dyn_iom_2d( putrd, pvtrd, ktrd, kt ) 
    69251          
    70       !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
    71       !  Integral Constraints Properties for momentum and/or tracers trends 
    72       !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
    73       IF( ln_glo_trd )   CALL trd_glo( putrd, pvtrd, ktrd, 'DYN', kt ) 
    74  
    75       !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
    76       !  Kinetic Energy trends 
    77       !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> 
    78       IF( ln_KE_trd  )   CALL trd_ken( putrd, pvtrd, ktrd, kt ) 
    79  
    80       !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
    81       !  Vorticity trends 
    82       !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> 
    83       IF( ln_vor_trd )   CALL trd_vor( putrd, pvtrd, ktrd, kt ) 
     252 
     253!!$   CALLS TO THESE ROUTINES FOR 2D DIAGOSTICS NOT CODED YET 
     254!!$      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     255!!$      !  Integral Constraints Properties for momentum and/or tracers trends 
     256!!$      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     257!!$      IF( ln_glo_trd )   CALL trd_glo( putrd, pvtrd, ktrd, 'DYN', kt ) 
     258!!$ 
     259!!$      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     260!!$      !  Kinetic Energy trends 
     261!!$      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> 
     262!!$      IF( ln_KE_trd  )   CALL trd_ken( putrd, pvtrd, ktrd, kt ) 
     263!!$ 
     264!!$      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< 
     265!!$      !  Vorticity trends 
     266!!$      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> 
     267!!$      IF( ln_vor_trd )   CALL trd_vor( putrd, pvtrd, ktrd, kt ) 
    84268 
    85269      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> 
     
    88272!!gm      IF( ln_dyn_mxl )   CALL trd_mxl_dyn    
    89273      ! 
    90    END SUBROUTINE trd_dyn 
    91  
    92  
    93    SUBROUTINE trd_dyn_iom( putrd, pvtrd, ktrd, kt ) 
     274   END SUBROUTINE trd_dyn_2d 
     275 
     276 
     277   SUBROUTINE trd_dyn_iom_3d( putrd, pvtrd, ktrd, kt ) 
    94278      !!--------------------------------------------------------------------- 
    95279      !!                  ***  ROUTINE trd_dyn_iom  *** 
     
    110294      CASE( jpdyn_hpg )   ;   CALL iom_put( "utrd_hpg", putrd )    ! hydrostatic pressure gradient 
    111295                              CALL iom_put( "vtrd_hpg", pvtrd ) 
    112       CASE( jpdyn_spg )   ;   CALL iom_put( "utrd_spg", putrd )    ! surface pressure gradient 
    113                               CALL iom_put( "vtrd_spg", pvtrd ) 
    114296      CASE( jpdyn_pvo )   ;   CALL iom_put( "utrd_pvo", putrd )    ! planetary vorticity 
    115297                              CALL iom_put( "vtrd_pvo", pvtrd ) 
     
    147329                              CALL iom_put( "vtrd_tau", z2dy ) 
    148330                              DEALLOCATE( z2dx , z2dy ) 
    149 !!gm  to be changed : computation should be done in dynzdf.F90 
    150 !!gm                + missing the top friction  
    151 !                              !                                    ! bottom stress tends (implicit case) 
    152 !                              IF( ln_drgimp ) THEN 
    153 !                                 ALLOCATE( z3dx(jpi,jpj,jpk) , z3dy(jpi,jpj,jpk) ) 
    154 !                             z3dx(:,:,:) = 0._wp   ;   z3dy(:,:,:) = 0._wp  ! after velocity known (now filed at this stage) 
    155 !                            DO jk = 1, jpkm1 
    156 !                                    DO jj = 2, jpjm1 
    157 !                                       DO ji = 2, jpim1 
    158 !                                      ikbu = mbku(ji,jj)          ! deepest ocean u- & v-levels 
    159 !                                          ikbv = mbkv(ji,jj) 
    160 !                                          z3dx(ji,jj,jk) = 0.5 * ( rCdU_bot(ji+1,jj) + rCdU_bot(ji,jj) ) &  
    161 !                                               &         * un(ji,jj,ikbu) / e3u_n(ji,jj,ikbu) 
    162 !                                          z3dy(ji,jj,jk) = 0.5 * ( rCdU_bot(ji,jj+1) + rCdU_bot(ji,jj) ) & 
    163 !                                               &         * vn(ji,jj,ikbv) / e3v_n(ji,jj,ikbv) 
    164 !                                    END DO 
    165 !                                 END DO 
    166 !                              END DO 
    167 !                              CALL lbc_lnk_multi( 'trddyn', z3dx, 'U', -1., z3dy, 'V', -1. ) 
    168 !                              CALL iom_put( "utrd_bfr", z3dx ) 
    169 !                              CALL iom_put( "vtrd_bfr", z3dy ) 
    170 !                                 DEALLOCATE( z3dx , z3dy ) 
    171 !                              ENDIF 
    172 !!gm end 
    173       CASE( jpdyn_bfr )       ! called if ln_drgimp=F 
    174                               CALL iom_put( "utrd_bfr", putrd )    ! bottom friction (explicit case) 
     331      CASE( jpdyn_bfr )   ;   CALL iom_put( "utrd_bfr", putrd )    ! bottom friction (explicit case) 
    175332                              CALL iom_put( "vtrd_bfr", pvtrd ) 
     333      CASE( jpdyn_bfri)   ;   CALL iom_put( "utrd_bfri", putrd )    ! bottom friction (implicit case) 
     334                              CALL iom_put( "vtrd_bfri", pvtrd ) 
    176335      CASE( jpdyn_atf )   ;   CALL iom_put( "utrd_atf", putrd )        ! asselin filter trends  
    177336                              CALL iom_put( "vtrd_atf", pvtrd ) 
    178337      END SELECT 
    179338      ! 
    180    END SUBROUTINE trd_dyn_iom 
     339   END SUBROUTINE trd_dyn_iom_3d 
     340 
     341 
     342   SUBROUTINE trd_dyn_iom_2d( putrd, pvtrd, ktrd, kt ) 
     343      !!--------------------------------------------------------------------- 
     344      !!                  ***  ROUTINE trd_dyn_iom  *** 
     345      !!  
     346      !! ** Purpose :   output 2D trends using IOM 
     347      !!---------------------------------------------------------------------- 
     348      REAL(wp), DIMENSION(:,:), INTENT(inout) ::   putrd, pvtrd   ! U and V trends 
     349      INTEGER                 , INTENT(in   ) ::   ktrd           ! trend index 
     350      INTEGER                 , INTENT(in   ) ::   kt             ! time step 
     351      ! 
     352      INTEGER ::   ji, jj, jk   ! dummy loop indices 
     353      INTEGER ::   ikbu, ikbv   ! local integers 
     354      REAL(wp), ALLOCATABLE, DIMENSION(:,:)   ::   z2dx, z2dy   ! 2D workspace  
     355      REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) ::   z3dx, z3dy   ! 3D workspace  
     356      !!---------------------------------------------------------------------- 
     357      ! 
     358      SELECT CASE( ktrd ) 
     359      CASE( jpdyn_spg )      ;   CALL iom_put( "utrd_spg2d", putrd )      ! surface pressure gradient 
     360                                 CALL iom_put( "vtrd_spg2d", pvtrd ) 
     361      CASE( jpdyn_pvo )      ;   CALL iom_put( "utrd_pvo2d", putrd )      ! planetary vorticity (barotropic part) 
     362                                 CALL iom_put( "vtrd_pvo2d", pvtrd ) 
     363      CASE( jpdyn_hpg_corr ) ;   CALL iom_put( "utrd_hpg_corr", putrd )   ! horizontal pressure gradient correction 
     364                                 CALL iom_put( "vtrd_hpg_corr", pvtrd ) 
     365      CASE( jpdyn_pvo_corr ) ;   CALL iom_put( "utrd_pvo_corr", putrd )   ! planetary vorticity correction 
     366                                 CALL iom_put( "vtrd_pvo_corr", pvtrd ) 
     367      CASE( jpdyn_bfr )      ;   CALL iom_put( "utrd_bfr2d", putrd )      ! bottom friction due to barotropic currents 
     368                                 CALL iom_put( "vtrd_bfr2d", pvtrd ) 
     369      END SELECT 
     370      ! 
     371   END SUBROUTINE trd_dyn_iom_2d 
    181372 
    182373   !!====================================================================== 
Note: See TracChangeset for help on using the changeset viewer.