New URL for NEMO forge!   http://forge.nemo-ocean.eu

Since March 2022 along with NEMO 4.2 release, the code development moved to a self-hosted GitLab.
This present forge is now archived and remained online for history.
traldf_iso_tile.F90 in NEMO/branches/UKMO/BPC_miniapp/OpenMP – NEMO

source: NEMO/branches/UKMO/BPC_miniapp/OpenMP/traldf_iso_tile.F90 @ 10838

Last change on this file since 10838 was 10838, checked in by wayne_gaudin, 5 years ago

Ticket #2197 - extracted versions added

File size: 23.6 KB
Line 
1MODULE traldf_iso_tile
2   !!======================================================================
3   !!                   ***  MODULE  traldf_iso  ***
4   !! Ocean  tracers:  horizontal component of the lateral tracer mixing trend
5   !!======================================================================
6   !! History :  OPA  ! 1994-08  (G. Madec, M. Imbard)
7   !!            8.0  ! 1997-05  (G. Madec)  split into traldf and trazdf
8   !!            NEMO ! 2002-08  (G. Madec)  Free form, F90
9   !!            1.0  ! 2005-11  (G. Madec)  merge traldf and trazdf :-)
10   !!            3.3  ! 2010-09  (C. Ethe, G. Madec) Merge TRA-TRC
11   !!            3.7  ! 2014-01  (G. Madec, S. Masson)  restructuration/simplification of aht/aeiv specification
12   !!             -   ! 2014-02  (F. Lemarie, G. Madec)  triad operator (Griffies) + Method of Stabilizing Correction
13   !!----------------------------------------------------------------------
14
15   !!----------------------------------------------------------------------
16   !!   tra_ldf_iso_tile   : update the tracer trend with the horizontal component of a iso-neutral laplacian operator
17   !!                   and with the vertical part of the isopycnal or geopotential s-coord. operator
18   !!----------------------------------------------------------------------
19   USE par_kind
20   USE len_oce            ! ocean dynamics and active tracers
21   USE tiletype
22   !
23   USE in_out_manager ! I/O manager
24
25   IMPLICIT NONE
26
27   PUBLIC   tra_ldf_iso_tile   ! routine called by step.F90
28
29   !! * Substitutions
30#  include "vectopt_loop_substitute.h90"
31   !!----------------------------------------------------------------------
32   !! NEMO/OCE 4.0 , NEMO Consortium (2018)
33   !! $Id$
34   !! Software governed by the CeCILL licence     (./LICENSE)
35   !!----------------------------------------------------------------------
36CONTAINS
37
38  SUBROUTINE tra_ldf_iso_tile( tile, ln_traldf_msc, ln_traldf_blp, ln_traldf_lap, ln_zps, ln_isfcav, kjpt, kpass, p2dt, &
39      &                   e1t, e1u, e1v, e2t, e2u, e2v, e1e2t, e1_e2v, e2_e1u, r1_e1e2t, mbku, mbkv, miku, mikv, &
40      &                   umask, vmask, wmask, e3t_n, e3u_n, e3v_n, e3w_n, wslpi, wslpj, uslp, vslp, &
41      &                   pahu, pahv, pgu, pgv, pgui, pgvi, ptb , ptbb, pta, &
42      &                   l_hst, l_ptr, pakz, pah_wslp2, pftu, pftv )
43
44      !!----------------------------------------------------------------------
45      !!                  ***  ROUTINE tra_ldf_iso_tile  ***
46      !!
47      !! ** Purpose :   Compute the before horizontal tracer (t & s) diffusive
48      !!      trend for a laplacian tensor (ezxcept the dz[ dz[.] ] term) and
49      !!      add it to the general trend of tracer equation.
50      !!
51      !! ** Method  :   The horizontal component of the lateral diffusive trends
52      !!      is provided by a 2nd order operator rotated along neural or geopo-
53      !!      tential surfaces to which an eddy induced advection can be added
54      !!      It is computed using before fields (forward in time) and isopyc-
55      !!      nal or geopotential slopes computed in routine ldfslp.
56      !!
57      !!      1st part :  masked horizontal derivative of T  ( di[ t ] )
58      !!      ========    with partial cell update if ln_zps=T
59      !!                  with top     cell update if ln_isfcav
60      !!
61      !!      2nd part :  horizontal fluxes of the lateral mixing operator
62      !!      ========   
63      !!         zftu =  pahu e2u*e3u/e1u di[ tb ]
64      !!               - pahu e2u*uslp    dk[ mi(mk(tb)) ]
65      !!         zftv =  pahv e1v*e3v/e2v dj[ tb ]
66      !!               - pahv e2u*vslp    dk[ mj(mk(tb)) ]
67      !!      take the horizontal divergence of the fluxes:
68      !!         difft = 1/(e1e2t*e3t) {  di-1[ zftu ] +  dj-1[ zftv ]  }
69      !!      Add this trend to the general trend (ta,sa):
70      !!         ta = ta + difft
71      !!
72      !!      3rd part: vertical trends of the lateral mixing operator
73      !!      ========  (excluding the vertical flux proportional to dk[t] )
74      !!      vertical fluxes associated with the rotated lateral mixing:
75      !!         zftw = - {  mi(mk(pahu)) * e2t*wslpi di[ mi(mk(tb)) ]
76      !!                   + mj(mk(pahv)) * e1t*wslpj dj[ mj(mk(tb)) ]  }
77      !!      take the horizontal divergence of the fluxes:
78      !!         difft = 1/(e1e2t*e3t) dk[ zftw ]
79      !!      Add this trend to the general trend (ta,sa):
80      !!         pta = pta + difft
81      !!
82      !! ** Action :   Update pta arrays with the before rotated diffusion
83      !!----------------------------------------------------------------------
84      TYPE(tile_type)                      , INTENT(in   ) ::   tile
85      LOGICAL                              , INTENT(in   ) ::   ln_traldf_msc, ln_traldf_blp, ln_traldf_lap, ln_zps, ln_isfcav
86      INTEGER                              , INTENT(in   ) ::   kjpt       ! number of tracers passed in
87      INTEGER                              , INTENT(in   ) ::   kpass      ! =1/2 first or second passage
88      REAL(wp)                             , INTENT(in   ) ::   p2dt       ! 2 . dt
89      REAL(wp), DIMENSION(jpi,jpj)         , INTENT(in   ) ::   e1t, e1u, e1v, e2t, e2u, e2v, e1e2t, e1_e2v, e2_e1u, r1_e1e2t     
90      INTEGER, DIMENSION(jpi,jpj)          , INTENT(in   ) ::   mbku, mbkv, miku, mikv
91      REAL(wp), DIMENSION(jpi,jpj,jpk)     , INTENT(in   ) ::   umask, vmask, wmask
92      REAL(wp), DIMENSION(jpi,jpj,jpk)     , INTENT(in   ) ::   e3t_n, e3u_n, e3v_n, e3w_n, wslpi, wslpj, uslp, vslp
93
94      REAL(wp), DIMENSION(jpi,jpj,jpk)     , INTENT(in   ) ::   pahu, pahv ! eddy diffusivity at u- and v-points  [m2/s]
95      REAL(wp), DIMENSION(jpi,jpj    ,kjpt), INTENT(in   ) ::   pgu, pgv   ! tracer gradient at pstep levels
96      REAL(wp), DIMENSION(jpi,jpj,    kjpt), INTENT(in   ) ::   pgui, pgvi ! tracer gradient at top   levels
97      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in   ) ::   ptb        ! tracer (kpass=1) or laplacian of tracer (kpass=2)
98      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in   ) ::   ptbb       ! tracer (only used in kpass=2)
99      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(inout) ::   pta        ! tracer trend
100
101      LOGICAL                              , INTENT(in   ) ::   l_ptr      ! flag to compute poleward transport
102      LOGICAL                              , INTENT(in   ) ::   l_hst      ! flag to compute heat transport
103      REAL(wp), DIMENSION(jpi,jpj,jpk)     , INTENT(inout) ::   pakz, pah_wslp2 !       
104      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(inout) ::   pftu, pftv ! diagnostic outputs
105
106      !
107      INTEGER  ::  ji, jj, jk, jn   ! dummy loop indices
108      INTEGER  ::  ikt
109      INTEGER  ::  ierr             ! local integer
110      REAL(wp) ::  zmsku, zahu_w, zabe1, zcof1, zcoef3   ! local scalars
111      REAL(wp) ::  zmskv, zahv_w, zabe2, zcof2, zcoef4   !   -      -
112      REAL(wp) ::  zcoef0, ze3w_2, zsign, z2dt, z1_2dt   !   -      -
113
114      REAL(wp) ,ALLOCATABLE, DIMENSION(:,:)     ::   zdkt, zdk1t, z2d      ! Make these use global scratch at some point
115      REAL(wp) ,ALLOCATABLE, DIMENSION(:,:,:)   ::   zdit, zdjt, zftu, zftv, ztfw,  zakz, zah_wslp2 
116
117      ALLOCATE(zdkt (tile%jsih:tile%jeih, tile%jsjh:tile%jejh))
118      ALLOCATE(zdk1t(tile%jsih:tile%jeih, tile%jsjh:tile%jejh))
119      ALLOCATE(z2d  (tile%jsih:tile%jeih, tile%jsjh:tile%jejh))
120
121      ALLOCATE(zdit(     tile%jsih:tile%jeih, tile%jsjh:tile%jejh,tile%jskh:tile%jekh))
122      ALLOCATE(zdjt(     tile%jsih:tile%jeih, tile%jsjh:tile%jejh,tile%jskh:tile%jekh))
123      ALLOCATE(zftu(     tile%jsih:tile%jeih, tile%jsjh:tile%jejh,tile%jskh:tile%jekh))
124      ALLOCATE(zftv(     tile%jsih:tile%jeih, tile%jsjh:tile%jejh,tile%jskh:tile%jekh))
125      ALLOCATE(ztfw(     tile%jsih:tile%jeih, tile%jsjh:tile%jejh,tile%jskh:tile%jekh))
126      ALLOCATE(zakz(     tile%jsih:tile%jeih, tile%jsjh:tile%jejh,tile%jskh:tile%jekh))
127      ALLOCATE(zah_wslp2(tile%jsih:tile%jeih, tile%jsjh:tile%jejh,tile%jskh:tile%jekh))
128
129write(0,*)"tile%jsih",tile%jsih
130write(0,*)"tile%jeih",tile%jeih
131write(0,*)"tile%jsjh",tile%jsjh
132write(0,*)"tile%jejh",tile%jejh
133write(0,*)"tile%jskh",tile%jskh
134write(0,*)"tile%jekh",tile%jekh
135
136      !!----------------------------------------------------------------------
137      !
138      !   
139      !
140      !                                            ! set time step size (Euler/Leapfrog)
141      z1_2dt = 1._wp / p2dt   !  note this was dependent on kt and neuler before
142      !
143      IF( kpass == 1 ) THEN   ;   zsign =  1._wp      ! bilaplacian operator require a minus sign (eddy diffusivity >0)
144      ELSE                    ;   zsign = -1._wp
145      ENDIF
146         
147      !!----------------------------------------------------------------------
148      !!   0 - calculate  ah_wslp2 and akz
149      !!----------------------------------------------------------------------
150      !
151      IF( kpass == 1 ) THEN                  !==  first pass only  ==!
152         !
153         DO jk = tile % jsk_2, tile % jekp1 
154            DO jj = tile % jsj, tile % jej
155               DO ji = tile % jsi, tile % jei   ! vector opt.
156
157                  !
158                  zmsku = wmask(ji,jj,jk) / MAX(   umask(ji  ,jj,jk-1) + umask(ji-1,jj,jk)          &
159                     &                           + umask(ji-1,jj,jk-1) + umask(ji  ,jj,jk) , 1._wp  )
160                  zmskv = wmask(ji,jj,jk) / MAX(   vmask(ji,jj  ,jk-1) + vmask(ji,jj-1,jk)          &
161                     &                           + vmask(ji,jj-1,jk-1) + vmask(ji,jj  ,jk) , 1._wp  )
162                     !
163                  zahu_w = (   pahu(ji  ,jj,jk-1) + pahu(ji-1,jj,jk)    &
164                     &       + pahu(ji-1,jj,jk-1) + pahu(ji  ,jj,jk)  ) * zmsku
165                  zahv_w = (   pahv(ji,jj  ,jk-1) + pahv(ji,jj-1,jk)    &
166                     &       + pahv(ji,jj-1,jk-1) + pahv(ji,jj  ,jk)  ) * zmskv
167                     !
168                  zah_wslp2(ji,jj,jk) = zahu_w * wslpi(ji,jj,jk) * wslpi(ji,jj,jk)   &
169                     &               + zahv_w * wslpj(ji,jj,jk) * wslpj(ji,jj,jk)
170               END DO
171            END DO
172         END DO
173         !
174         IF( ln_traldf_msc ) THEN                ! stabilizing vertical diffusivity coefficient
175            DO jk = tile % jsk_2, tile % jekp1 
176               DO jj = tile % jsj, tile % jej
177                  DO ji = tile % jsi, tile % jei
178                     zakz(ji,jj,jk) = 0.25_wp * (                                                                     &
179                        &              ( pahu(ji  ,jj,jk) + pahu(ji  ,jj,jk-1) ) / ( e1u(ji  ,jj) * e1u(ji  ,jj) )   &
180                        &            + ( pahu(ji-1,jj,jk) + pahu(ji-1,jj,jk-1) ) / ( e1u(ji-1,jj) * e1u(ji-1,jj) )   &
181                        &            + ( pahv(ji,jj  ,jk) + pahv(ji,jj  ,jk-1) ) / ( e2v(ji,jj  ) * e2v(ji,jj  ) )   &
182                        &            + ( pahv(ji,jj-1,jk) + pahv(ji,jj-1,jk-1) ) / ( e2v(ji,jj-1) * e2v(ji,jj-1) )   )
183                  END DO
184               END DO
185            END DO
186            !
187            IF( ln_traldf_blp ) THEN                ! bilaplacian operator
188               DO jk = tile % jsk_2, tile % jekp1 
189                  DO jj = tile % jsj, tile % jej
190                     DO ji = tile % jsi, tile % jei
191                        zakz(ji,jj,jk) = 16._wp * zah_wslp2(ji,jj,jk)   &
192                           &          * (  zakz(ji,jj,jk) + zah_wslp2(ji,jj,jk) / ( e3w_n(ji,jj,jk) * e3w_n(ji,jj,jk) )  )
193                     END DO
194                  END DO
195               END DO
196            ELSEIF( ln_traldf_lap ) THEN              ! laplacian operator
197               DO jk = tile % jsk_2, tile % jekp1
198                  DO jj = tile % jsj, tile % jej
199                     DO ji = tile % jsi, tile % jei
200                        ze3w_2 = e3w_n(ji,jj,jk) * e3w_n(ji,jj,jk)
201                        zcoef0 = p2dt * (  zakz(ji,jj,jk) + zah_wslp2(ji,jj,jk) / ze3w_2  )
202                        zakz(ji,jj,jk) = MAX( zcoef0 - 0.5_wp , 0._wp ) * ze3w_2 * z1_2dt
203                     END DO
204                  END DO
205               END DO
206           ENDIF
207           !
208         ELSE                                    ! 33 flux set to zero with zakz=zah_wslp2 ==>> computed in full implicit
209           DO jk = tile % jsk_2, tile % jekp1
210              DO jj = tile % jsj, tile % jej
211                 DO ji = tile % jsi, tile % jei
212                    zakz(ji,jj,jk) = zah_wslp2(ji,jj,jk)
213                 END DO
214               END DO
215            END DO
216         ENDIF
217      ENDIF
218
219! Save zakz and zah_wslp2 (private arrays) in non-overlapping elements of akz and ah_wslp2 arrays (for re-use in trazdf)
220      DO jk = tile % jsk, tile % jek
221         DO jj = tile % jsj, tile % jej
222            DO ji = tile % jsi, tile % jei
223               pakz(ji,jj,jk) = zakz(ji,jj,jk) ; pah_wslp2(ji,jj,jk) = zah_wslp2(ji,jj,jk)
224            END DO
225         END DO
226      END DO
227
228      !
229      !                                                          ! ===========
230      DO jn = 1, kjpt                                            ! tracer loop
231         !                                                       ! ===========
232         !                                               
233         !!----------------------------------------------------------------------
234         !!   I - masked horizontal derivative
235         !!----------------------------------------------------------------------
236
237         ! Horizontal tracer gradient
238         DO jk = tile % jskm1, tile % jekp1
239            DO jj = tile % jsjm1, tile % jej
240               DO ji = tile % jsim1, tile % jei   ! vector opt.
241                  zdit(ji,jj,jk) = ( ptb(ji+1,jj  ,jk,jn) - ptb(ji,jj,jk,jn) ) * umask(ji,jj,jk)
242                  zdjt(ji,jj,jk) = ( ptb(ji  ,jj+1,jk,jn) - ptb(ji,jj,jk,jn) ) * vmask(ji,jj,jk)
243               END DO
244            END DO
245         END DO
246         IF( ln_zps ) THEN      ! botton and surface ocean correction of the horizontal gradient
247            DO jj = tile % jsjm1, tile % jej              ! bottom correction (partial bottom cell)
248               DO ji = tile % jsim1, tile % jei   ! vector opt.
249                  IF( tile % jskm1 <= mbku(ji,jj) .AND. mbku(ji,jj) <= tile % jekp1 ) zdit(ji,jj,mbku(ji,jj)) = pgu(ji,jj,jn)         
250                  IF( tile % jskm1 <= mbkv(ji,jj) .AND. mbkv(ji,jj) <= tile % jekp1 ) zdjt(ji,jj,mbkv(ji,jj)) = pgv(ji,jj,jn)
251               END DO
252            END DO
253            IF( ln_isfcav ) THEN      ! first wet level beneath a cavity
254               DO jj = tile % jsjm1, tile % jej
255                  DO ji = tile % jsim1, tile % jei   ! vector opt.
256                     IF( tile % jskm1 <= miku(ji,jj) .AND. miku(ji,jj) <= tile % jekp1 ) zdit(ji,jj,miku(ji,jj)) = pgui(ji,jj,jn)         
257                     IF( tile % jskm1 <= mikv(ji,jj) .AND. mikv(ji,jj) <= tile % jekp1 ) zdjt(ji,jj,mikv(ji,jj)) = pgvi(ji,jj,jn)
258                  END DO
259               END DO
260            ENDIF
261         ENDIF
262         !
263         !!----------------------------------------------------------------------
264         !!   II - horizontal trend  (full)
265         !!----------------------------------------------------------------------
266         !
267         DO jk = tile % jsk, tile % jek                                 ! Horizontal slab
268            !
269            !                             !== Vertical tracer gradient (loops expanded for low-level tiling)
270            DO jj = tile % jsjm1, tile % jejp1
271              DO ji = tile % jsim1, tile % jeip1
272                zdk1t(ji,jj) = ( ptb(ji,jj,jk,jn) - ptb(ji,jj,jk+1,jn) ) * wmask(ji,jj,jk+1)     ! level jk+1
273              END DO
274            END DO
275            !
276            IF( jk == 1 ) THEN   
277              DO jj = tile % jsjm1, tile % jejp1
278                DO ji = tile % jsim1, tile % jeip1
279                  zdkt(ji,jj) = zdk1t(ji,jj)    ! surface: zdkt(jk=1)=zdkt(jk=2)
280                END DO
281              END DO
282            ELSE                 
283              DO jj = tile % jsjm1, tile % jejp1
284                DO ji = tile % jsim1, tile % jeip1
285                  zdkt(ji,jj) = ( ptb(ji,jj,jk-1,jn) - ptb(ji,jj,jk,jn) ) * wmask(ji,jj,jk)
286                END DO
287              END DO
288            ENDIF
289            DO jj = tile % jsjm1, tile % jej            !==  Horizontal fluxes
290               DO ji = tile % jsim1, tile % jei   ! vector opt.
291                  zabe1 = pahu(ji,jj,jk) * e2_e1u(ji,jj) * e3u_n(ji,jj,jk)
292                  zabe2 = pahv(ji,jj,jk) * e1_e2v(ji,jj) * e3v_n(ji,jj,jk)
293                  !
294                  zmsku = 1._wp / MAX(  wmask(ji+1,jj,jk  ) + wmask(ji,jj,jk+1)   &
295                     &             + wmask(ji+1,jj,jk+1) + wmask(ji,jj,jk  ), 1._wp )
296                  !
297                  zmskv = 1._wp / MAX(  wmask(ji,jj+1,jk  ) + wmask(ji,jj,jk+1)   &
298                     &             + wmask(ji,jj+1,jk+1) + wmask(ji,jj,jk  ), 1._wp )
299                  !
300                  zcof1 = - pahu(ji,jj,jk) * e2u(ji,jj) * uslp(ji,jj,jk) * zmsku
301                  zcof2 = - pahv(ji,jj,jk) * e1v(ji,jj) * vslp(ji,jj,jk) * zmskv
302                  !
303                  zftu(ji,jj,jk) = (  zabe1 * zdit(ji,jj,jk)   &
304                     &               + zcof1 * (  zdkt (ji+1,jj) + zdk1t(ji,jj)      &
305                     &                          + zdk1t(ji+1,jj) + zdkt (ji,jj)  )  ) * umask(ji,jj,jk)
306                  zftv(ji,jj,jk) = (  zabe2 * zdjt(ji,jj,jk)   &
307                     &               + zcof2 * (  zdkt (ji,jj+1) + zdk1t(ji,jj)      &
308                     &                          + zdk1t(ji,jj+1) + zdkt (ji,jj)  )  ) * vmask(ji,jj,jk)                 
309               END DO
310            END DO
311            !
312            DO jj = tile % jsj , tile % jej          !== horizontal divergence and add to pta
313               DO ji = tile % jsi, tile % jei    ! vector opt.
314                  pta(ji,jj,jk,jn) = pta(ji,jj,jk,jn) + zsign * (  zftu(ji,jj,jk) - zftu(ji-1,jj,jk)      &
315                     &                                           + zftv(ji,jj,jk) - zftv(ji,jj-1,jk)  )   &
316                     &                                        * r1_e1e2t(ji,jj) / e3t_n(ji,jj,jk)
317               END DO
318            END DO
319         END DO                                        !   End of slab 
320
321         !!----------------------------------------------------------------------
322         !!   III - vertical trend (full)
323         !!----------------------------------------------------------------------
324         !
325         ! Vertical fluxes
326         ! ---------------
327         !                          ! Surface and bottom vertical fluxes set to zero
328         IF ( tile % jsk == 1  ) THEN
329            DO jj = tile % jsj, tile % jej
330               DO ji = tile % jsi, tile % jei   ! vector opt.     
331                 ztfw(ji,jj,1) = 0._wp   
332               END DO
333             END DO
334         END IF   
335         IF ( tile % jek == jpkm1 ) THEN
336            DO jj = tile % jsj, tile % jej
337               DO ji = tile % jsi, tile % jei   ! vector opt.     
338                 ztfw(ji,jj,jpk) = 0._wp   
339               END DO
340             END DO
341         END IF   
342
343         DO jk = tile % jsk_2, tile % jekp1           ! interior (2=<jk=<jpk-1)
344            DO jj = tile % jsj, tile % jej
345               DO ji = tile % jsi, tile % jei   ! vector opt.
346                  !
347                  zmsku = wmask(ji,jj,jk) / MAX(   umask(ji  ,jj,jk-1) + umask(ji-1,jj,jk)          &
348                     &                           + umask(ji-1,jj,jk-1) + umask(ji  ,jj,jk) , 1._wp  )
349                  zmskv = wmask(ji,jj,jk) / MAX(   vmask(ji,jj  ,jk-1) + vmask(ji,jj-1,jk)          &
350                     &                           + vmask(ji,jj-1,jk-1) + vmask(ji,jj  ,jk) , 1._wp  )
351                     !
352                  zahu_w = (   pahu(ji  ,jj,jk-1) + pahu(ji-1,jj,jk)    &
353                     &       + pahu(ji-1,jj,jk-1) + pahu(ji  ,jj,jk)  ) * zmsku
354                  zahv_w = (   pahv(ji,jj  ,jk-1) + pahv(ji,jj-1,jk)    &
355                     &       + pahv(ji,jj-1,jk-1) + pahv(ji,jj  ,jk)  ) * zmskv
356                     !
357                  zcoef3 = - zahu_w * e2t(ji,jj) * zmsku * wslpi (ji,jj,jk)   !wslpi & j are already w-masked
358                  zcoef4 = - zahv_w * e1t(ji,jj) * zmskv * wslpj (ji,jj,jk)
359                  !
360                  ztfw(ji,jj,jk) = zcoef3 * (   zdit(ji  ,jj  ,jk-1) + zdit(ji-1,jj  ,jk)      &
361                     &                        + zdit(ji-1,jj  ,jk-1) + zdit(ji  ,jj  ,jk)  )   &
362                     &           + zcoef4 * (   zdjt(ji  ,jj  ,jk-1) + zdjt(ji  ,jj-1,jk)      &
363                     &                        + zdjt(ji  ,jj-1,jk-1) + zdjt(ji  ,jj  ,jk)  )
364               END DO
365            END DO
366         END DO
367
368         !                                !==  add the vertical 33 flux  ==!
369         IF( ln_traldf_lap ) THEN               ! laplacian case: eddy coef = zah_wslp2 - zakz
370            DO jk = tile % jsk_2, tile % jekp1       
371               DO jj = tile % jsj, tile % jej
372                  DO ji = tile % jsi, tile % jei
373                     ztfw(ji,jj,jk) = ztfw(ji,jj,jk) + e1e2t(ji,jj) / e3w_n(ji,jj,jk) * wmask(ji,jj,jk)   &
374                        &                            * ( zah_wslp2(ji,jj,jk) - zakz(ji,jj,jk) )           &
375                        &                            * ( ptb(ji,jj,jk-1,jn) - ptb(ji,jj,jk,jn) )
376                  END DO
377               END DO
378            END DO
379            !
380         ELSE                                   ! bilaplacian
381            SELECT CASE( kpass )
382            CASE(  1  )                            ! 1st pass : eddy coef = zah_wslp2
383               DO jk = tile % jsk_2, tile % jekp1 
384                  DO jj = tile % jsj, tile % jej
385                     DO ji = tile % jsi, tile % jei
386                        ztfw(ji,jj,jk) = ztfw(ji,jj,jk)                       &
387                           &           + zah_wslp2(ji,jj,jk) * e1e2t(ji,jj)   &
388                           &           * ( ptb(ji,jj,jk-1,jn) - ptb(ji,jj,jk,jn) ) / e3w_n(ji,jj,jk) * wmask(ji,jj,jk)
389                     END DO
390                  END DO
391               END DO
392            CASE(  2  )                         ! 2nd pass : eddy flux = zah_wslp2 and akz applied on ptb  and ptbb gradients, resp.
393               DO jk = tile % jsk_2, tile % jekp1
394                  DO jj = tile % jsj, tile % jej
395                     DO ji = tile % jsi, tile % jei
396                        ztfw(ji,jj,jk) = ztfw(ji,jj,jk) + e1e2t(ji,jj) / e3w_n(ji,jj,jk) * wmask(ji,jj,jk)                       &
397                           &                            * (  zah_wslp2(ji,jj,jk) * ( ptb (ji,jj,jk-1,jn) - ptb (ji,jj,jk,jn) )   &
398                           &                               +      zakz(ji,jj,jk) * ( ptbb(ji,jj,jk-1,jn) - ptbb(ji,jj,jk,jn) )   )
399                     END DO
400                  END DO
401               END DO
402            END SELECT
403         ENDIF
404         !         
405         DO jk = tile % jsk, tile % jek                 !==  Divergence of vertical fluxes added to pta  ==!
406            DO jj = tile % jsj, tile % jej
407               DO ji = tile % jsi, tile % jei   ! vector opt.
408                  pta(ji,jj,jk,jn) = pta(ji,jj,jk,jn) + zsign * (  ztfw (ji,jj,jk) - ztfw(ji,jj,jk+1)  )   &
409                     &                                        * r1_e1e2t(ji,jj) / e3t_n(ji,jj,jk)
410               END DO
411            END DO
412         END DO
413         !
414         IF ( l_ptr .OR. l_hst ) THEN
415            DO jk = tile % jsk, tile % jek                 
416               DO jj = tile % jsj, tile % jej
417                  DO ji = tile % jsi, tile % jei   ! vector opt.
418                     pftu(ji,jj,jk,jn) = zftu(ji,jj,jk) ; pftv(ji,jj,jk,jn) = zftv(ji,jj,jk)
419                  END DO
420               END DO
421            END DO
422         END IF 
423 
424         !                                                        ! ===============
425      END DO                                                      ! end tracer loop
426      !
427   END SUBROUTINE tra_ldf_iso_tile
428
429   !!==============================================================================
430END MODULE traldf_iso_tile
Note: See TracBrowser for help on using the repository browser.