Changeset 10123

Timestamp:

2018-09-13T14:35:00+02:00 (6 years ago)

Author:

ldouriez

Message:

1 loop optimization in zdftke.F90: BULL_OPTIM_ZDFTKE_LANGMUIR_LOOP. Disabled by default

File:

• NEMO/branches/2018/dev_r9759_HPC09_ESIWACE/src/OCE/ZDF/zdftke.F90 (modified) (3 diffs)

NEMO/branches/2018/dev_r9759_HPC09_ESIWACE/src/OCE/ZDF/zdftke.F90

@@ -202 +202 @@
       REAL(wp) ::   zzd_up, zzd_lw             !   -         -
       INTEGER , DIMENSION(jpi,jpj)     ::   imlc
+#if defined(BULL_OPTIM_ZDFTKE_LANGMUIR_LOOP)
+      REAL(wp), DIMENSION(jpi,jpj)     ::   zhlc, fr_i_tmp
+#else      
       REAL(wp), DIMENSION(jpi,jpj)     ::   zhlc
+#endif      
       REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zpelc, zdiag, zd_up, zd_lw
       !!--------------------------------------------------------------------
@@ -290 +294 @@
          END DO
          zcof = 0.016 / SQRT( zrhoa * zcdrag )
+#if defined(BULL_OPTIM_ZDFTKE_LANGMUIR_LOOP)
+         DO jj = 2, jpjm1
+            DO ji = fs_2, fs_jpim1   ! vector opt.
+               fr_i_tmp(ji,jj) = (1._wp - 4.*fr_i(ji,jj))
+               IF (fr_i_tmp(ji,jj) < 0 ) THEN
+                  fr_i_tmp(ji,jj) = 0
+               ENDIF
+            END DO
+         END DO         
+#endif         
          DO jk = 2, jpkm1         !* TKE Langmuir circulation source term added to en
             DO jj = 2, jpjm1
                DO ji = fs_2, fs_jpim1   ! vector opt.
+#if defined(BULL_OPTIM_ZDFTKE_LANGMUIR_LOOP)
+                  IF ( fr_i_tmp(ji,jj) /= 0 ) THEN               
+                     ! vertical velocity due to LC   
+                     IF ( (pdepw(ji,jj,jk) - zhlc(ji,jj)) < 0 ) THEN
+                        zus  = zcof * SQRT( taum(ji,jj) )           ! Stokes drift
+                        zwlc = rn_lc * zus * SIN( rpi * pdepw(ji,jj,jk) / zhlc(ji,jj) )
+                        ! TKE Langmuir circulation source term
+                        en(ji,jj,jk) = en(ji,jj,jk) + rdt * fr_i_tmp(ji,jj) * ( zwlc * zwlc * zwlc )   &
+                             &                              / zhlc(ji,jj) * wmask(ji,jj,jk) * tmask(ji,jj,1)
+                     ENDIF
+                  ENDIF
+#else                  
                   zus  = zcof * SQRT( taum(ji,jj) )           ! Stokes drift
                   !                                           ! vertical velocity due to LC
@@ -299 +325 @@
                   !                                           ! TKE Langmuir circulation source term
                   en(ji,jj,jk) = en(ji,jj,jk) + rdt * MAX(0.,1._wp - 4.*fr_i(ji,jj) ) * ( zwlc * zwlc * zwlc )   &
-                     &                              / zhlc(ji,jj) * wmask(ji,jj,jk) * tmask(ji,jj,1)
+                       &                              / zhlc(ji,jj) * wmask(ji,jj,jk) * tmask(ji,jj,1)
+#endif                  
                END DO
             END DO