Changeset 11934 for NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/DYN/dynspg_ts.F90

Timestamp:

2019-11-20T10:47:16+01:00 (4 years ago)

Author:

davestorkey

Message:

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

File:

• NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/DYN/dynspg_ts.F90 (modified) (12 diffs)

NEMO/branches/UKMO/NEMO_4.0.1_momentum_trends/src/OCE/DYN/dynspg_ts.F90

@@ -63 +63 @@
    USE restart         ! only for lrst_oce
    USE diatmb          ! Top,middle,bottom output
+   USE trd_oce        ! trends: ocean variables
+   USE trddyn         ! trend manager: dynamics
 
    USE iom   ! to remove
@@ -170 +172 @@
       REAL(wp), ALLOCATABLE, DIMENSION(:,:) :: ztwdmask, zuwdmask, zvwdmask ! ROMS wetting and drying masks at t,u,v points
       REAL(wp), ALLOCATABLE, DIMENSION(:,:) :: zuwdav2, zvwdav2    ! averages over the sub-steps of zuwdmask and zvwdmask
+      REAL(wp), ALLOCATABLE, DIMENSION(:,:) :: zspgtrdu, zspgtrdv, zpvotrdu, zpvotrdv  ! SPG and PVO trends (if l_trddyn)
+      REAL(wp), ALLOCATABLE, DIMENSION(:,:) :: ztautrdu, ztautrdv, zbfrtrdu, zbfrtrdv  ! TAU and BFR trends (if l_trddyn)
       !!----------------------------------------------------------------------
       !
@@ -175 +179 @@
       !                                         !* Allocate temporary arrays
       IF( ln_wd_dl ) ALLOCATE( ztwdmask(jpi,jpj), zuwdmask(jpi,jpj), zvwdmask(jpi,jpj), zuwdav2(jpi,jpj), zvwdav2(jpi,jpj))
+      !
+      IF( l_trddyn ) THEN
+          ALLOCATE( zspgtrdu(jpi,jpj), zspgtrdv(jpi,jpj), zpvotrdu(jpi,jpj), zpvotrdv(jpi,jpj), &
+         &          ztautrdu(jpi,jpj), ztautrdv(jpi,jpj), zbfrtrdu(jpi,jpj), zbfrtrdv(jpi,jpj) )
+          zspgtrdu(:,:) = 0._wp
+          zspgtrdv(:,:) = 0._wp
+          zpvotrdu(:,:) = 0._wp
+          zpvotrdv(:,:) = 0._wp
+          ztautrdu(:,:) = 0._wp
+          ztautrdv(:,:) = 0._wp
+          zbfrtrdu(:,:) = 0._wp
+          zbfrtrdv(:,:) = 0._wp
+      ENDIF
+      !
+      zu_trd(:,:) = 0._wp
+      zv_trd(:,:) = 0._wp
+      zu_spg(:,:) = 0._wp
+      zv_spg(:,:) = 0._wp
       !
       zmdi=1.e+20                               !  missing data indicator for masking
@@ -253 +275 @@
          &                               zu_trd, zv_trd   )   ! ==>> out
       !
+      IF( l_trddyn ) THEN
+         ! send correction to baroclinic planetary vorticity trend to trd_dyn
+         CALL trd_dyn( zu_trd, zv_trd, jpdyn_pvo_corr, kt )
+      ENDIF
+      !
       IF( .NOT.ln_linssh ) THEN                 !* surface pressure gradient   (variable volume only)
+         !
+         IF( l_trddyn ) THEN
+            zspgtrdu(:,:) = zu_trd(:,:)
+            zspgtrdv(:,:) = zv_trd(:,:)
+         ENDIF
          !
          IF( ln_wd_il ) THEN                       ! W/D : limiter applied to spgspg
@@ -274 +306 @@
          ENDIF
          !
+         IF( l_trddyn ) THEN
+            zspgtrdu(:,:) = zu_trd(:,:) - zspgtrdu(:,:) 
+            zspgtrdv(:,:) = zv_trd(:,:) - zspgtrdv(:,:) 
+            ! send correction to HPG trend to trd_dyn
+            CALL trd_dyn( zspgtrdu, zspgtrdv, jpdyn_hpg_corr, kt )
+            ! reset temporary arrays for use later
+            zspgtrdu(:,:) = 0._wp
+            zspgtrdv(:,:) = 0._wp
+         ENDIF
+         !
       ENDIF
       !
@@ -283 +325 @@
       END DO 
       !
+      IF( l_trddyn ) THEN
+         zbfrtrdu(:,:) = zu_frc(:,:)
+         zbfrtrdv(:,:) = zv_frc(:,:)
+      ENDIF
       !                                   !=  Add bottom stress contribution from baroclinic velocities  =!
       !                                   !  -----------------------------------------------------------  !
       CALL dyn_drg_init( zu_frc, zv_frc,  zCdU_u, zCdU_v )      ! also provide the barotropic drag coefficients
+      !
+      IF( l_trddyn ) THEN
+         ! bottom friction trend diagnostic: bottom friction due to baroclinic currents
+         zbfrtrdu(:,:) = zu_frc(:,:) - zbfrtrdu(:,:)
+         zbfrtrdv(:,:) = zv_frc(:,:) - zbfrtrdv(:,:) 
+      ENDIF
       !
       !                                   !=  Add atmospheric pressure forcing  =!
@@ -312 +364 @@
       !                                   !=  Add atmospheric pressure forcing  =!
       !                                   !  ----------------------------------  !
+      IF( l_trddyn ) THEN
+         ztautrdu(:,:) = zu_frc(:,:)
+         ztautrdv(:,:) = zv_frc(:,:)
+      ENDIF
+      !
       IF( ln_bt_fw ) THEN                        ! Add wind forcing
          DO jj = 2, jpjm1
@@ -329 +386 @@
       ENDIF  
       !
+      IF( l_trddyn ) THEN
+         ! wind stress trend diagnostic
+         ztautrdu(:,:) = zu_frc(:,:) - ztautrdu(:,:)
+         ztautrdv(:,:) = zv_frc(:,:) - ztautrdv(:,:) 
+      ENDIF
       !              !----------------!
       !              !==  sssh_frc  ==!   Right-Hand-Side of the barotropic ssh equation   (over the FULL domain)
@@ -591 +653 @@
          ENDIF
          !
+         IF( l_trddyn ) THEN
+            za2 = wgtbtp2(jn)
+            zspgtrdu(:,:) = zspgtrdu(:,:) + za2 * zu_spg(:,:) * ssumask(:,:)
+            zspgtrdv(:,:) = zspgtrdv(:,:) + za2 * zv_spg(:,:) * ssvmask(:,:)
+         ENDIF
+         !
          ! Add Coriolis trend:
          ! zwz array below or triads normally depend on sea level with ln_linssh=F and should be updated
@@ -596 +664 @@
          ! Recall that zhU and zhV hold fluxes at jn+0.5 (extrapolated not backward interpolated)
          CALL dyn_cor_2d( zhup2_e, zhvp2_e, ua_e, va_e, zhU, zhV,    zu_trd, zv_trd   )
+         !
+         IF( l_trddyn ) THEN
+            za2 = wgtbtp2(jn)
+            zpvotrdu(:,:) = zpvotrdu(:,:) + za2 * zu_trd(:,:) * ssumask(:,:)
+            zpvotrdv(:,:) = zpvotrdv(:,:) + za2 * zv_trd(:,:) * ssvmask(:,:)
+         ENDIF
          !
          ! Add tidal astronomical forcing if defined
@@ -616 +690 @@
                END DO
             END DO
+            IF( l_trddyn ) THEN
+               za2 = wgtbtp2(jn)
+               zbfrtrdu(:,:) = zbfrtrdu(:,:) + za2 * zCdU_u(:,:) * un_e(:,:) * hur_e(:,:)
+               zbfrtrdv(:,:) = zbfrtrdv(:,:) + za2 * zCdU_v(:,:) * vn_e(:,:) * hvr_e(:,:)
+            ENDIF
          ENDIF
          !
@@ -835 +914 @@
       IF( ln_wd_il )   DEALLOCATE( zcpx, zcpy )
       IF( ln_wd_dl )   DEALLOCATE( ztwdmask, zuwdmask, zvwdmask, zuwdav2, zvwdav2 )
+      !
+      IF( l_trddyn ) THEN
+         CALL trd_dyn( zspgtrdu, zspgtrdv, jpdyn_spg, kt )
+         CALL trd_dyn( zpvotrdu, zpvotrdv, jpdyn_pvo, kt )
+         CALL trd_dyn( ztautrdu, ztautrdv, jpdyn_tau, kt )
+         CALL trd_dyn( zbfrtrdu, zbfrtrdv, jpdyn_bfr, kt )
+         DEALLOCATE( zspgtrdu, zspgtrdv, zpvotrdu, zpvotrdv, ztautrdu, ztautrdv, zbfrtrdu, zbfrtrdv )
+      ENDIF
       !
       IF( ln_diatmb ) THEN