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.
zdftke.F90 in NEMO/trunk/src/OCE/ZDF – NEMO

source: NEMO/trunk/src/OCE/ZDF/zdftke.F90 @ 13461

Last change on this file since 13461 was 13461, checked in by smasson, 4 years ago

trunk: merge r4.0-HEAD 13263:13278 into the trunk, see #2523

  • Property svn:keywords set to Id
File size: 41.9 KB
RevLine 
[1531]1MODULE zdftke
[1239]2   !!======================================================================
[1531]3   !!                       ***  MODULE  zdftke  ***
[1239]4   !! Ocean physics:  vertical mixing coefficient computed from the tke
5   !!                 turbulent closure parameterization
6   !!=====================================================================
[1492]7   !! History :  OPA  !  1991-03  (b. blanke)  Original code
8   !!            7.0  !  1991-11  (G. Madec)   bug fix
9   !!            7.1  !  1992-10  (G. Madec)   new mixing length and eav
10   !!            7.2  !  1993-03  (M. Guyon)   symetrical conditions
11   !!            7.3  !  1994-08  (G. Madec, M. Imbard)  nn_pdl flag
12   !!            7.5  !  1996-01  (G. Madec)   s-coordinates
13   !!            8.0  !  1997-07  (G. Madec)   lbc
14   !!            8.1  !  1999-01  (E. Stretta) new option for the mixing length
15   !!  NEMO      1.0  !  2002-06  (G. Madec) add tke_init routine
16   !!             -   !  2004-10  (C. Ethe )  1D configuration
17   !!            2.0  !  2006-07  (S. Masson)  distributed restart using iom
18   !!            3.0  !  2008-05  (C. Ethe,  G.Madec) : update TKE physics:
19   !!                 !           - tke penetration (wind steering)
20   !!                 !           - suface condition for tke & mixing length
21   !!                 !           - Langmuir cells
22   !!             -   !  2008-05  (J.-M. Molines, G. Madec)  2D form of avtb
23   !!             -   !  2008-06  (G. Madec)  style + DOCTOR name for namelist parameters
24   !!             -   !  2008-12  (G. Reffray) stable discretization of the production term
25   !!            3.2  !  2009-06  (G. Madec, S. Masson) TKE restart compatible with key_cpl
26   !!                 !                                + cleaning of the parameters + bugs correction
[2528]27   !!            3.3  !  2010-10  (C. Ethe, G. Madec) reorganisation of initialisation phase
[5120]28   !!            3.6  !  2014-11  (P. Mathiot) add ice shelf capability
[9019]29   !!            4.0  !  2017-04  (G. Madec)  remove CPP ddm key & avm at t-point only
[13461]30   !!             -   !  2017-05  (G. Madec)  add top/bottom friction as boundary condition
[1239]31   !!----------------------------------------------------------------------
[9019]32
[1239]33   !!----------------------------------------------------------------------
[3625]34   !!   zdf_tke       : update momentum and tracer Kz from a tke scheme
35   !!   tke_tke       : tke time stepping: update tke at now time step (en)
36   !!   tke_avn       : compute mixing length scale and deduce avm and avt
37   !!   zdf_tke_init  : initialization, namelist read, and parameters control
38   !!   tke_rst       : read/write tke restart in ocean restart file
[1239]39   !!----------------------------------------------------------------------
[2528]40   USE oce            ! ocean: dynamics and active tracers variables
41   USE phycst         ! physical constants
42   USE dom_oce        ! domain: ocean
43   USE domvvl         ! domain: variable volume layer
[1492]44   USE sbc_oce        ! surface boundary condition: ocean
[9019]45   USE zdfdrg         ! vertical physics: top/bottom drag coef.
[2528]46   USE zdfmxl         ! vertical physics: mixed layer
[13007]47#if defined key_si3
48   USE ice, ONLY: hm_i, h_i
49#endif
50#if defined key_cice
51   USE sbc_ice, ONLY: h_i
52#endif
[9019]53   !
[1492]54   USE in_out_manager ! I/O manager
55   USE iom            ! I/O manager library
[2715]56   USE lib_mpp        ! MPP library
[9019]57   USE lbclnk         ! ocean lateral boundary conditions (or mpp link)
58   USE prtctl         ! Print control
[3625]59   USE lib_fortran    ! Fortran utilities (allows no signed zero when 'key_nosignedzero' defined) 
[1239]60
61   IMPLICIT NONE
62   PRIVATE
63
[2528]64   PUBLIC   zdf_tke        ! routine called in step module
65   PUBLIC   zdf_tke_init   ! routine called in opa module
66   PUBLIC   tke_rst        ! routine called in step module
[1239]67
[4147]68   !                      !!** Namelist  namzdf_tke  **
69   LOGICAL  ::   ln_mxl0   ! mixing length scale surface value as function of wind stress or not
70   INTEGER  ::   nn_mxl    ! type of mixing length (=0/1/2/3)
71   REAL(wp) ::   rn_mxl0   ! surface  min value of mixing length (kappa*z_o=0.4*0.1 m)  [m]
[13012]72   INTEGER  ::      nn_mxlice ! type of scaling under sea-ice
73   REAL(wp) ::      rn_mxlice ! max constant ice thickness value when scaling under sea-ice ( nn_mxlice=1)
[4147]74   INTEGER  ::   nn_pdl    ! Prandtl number or not (ratio avt/avm) (=0/1)
75   REAL(wp) ::   rn_ediff  ! coefficient for avt: avt=rn_ediff*mxl*sqrt(e)
76   REAL(wp) ::   rn_ediss  ! coefficient of the Kolmogoroff dissipation
77   REAL(wp) ::   rn_ebb    ! coefficient of the surface input of tke
78   REAL(wp) ::   rn_emin   ! minimum value of tke           [m2/s2]
79   REAL(wp) ::   rn_emin0  ! surface minimum value of tke   [m2/s2]
80   REAL(wp) ::   rn_bshear ! background shear (>0) currently a numerical threshold (do not change it)
81   INTEGER  ::   nn_etau   ! type of depth penetration of surface tke (=0/1/2/3)
[9019]82   INTEGER  ::      nn_htau   ! type of tke profile of penetration (=0/1)
83   REAL(wp) ::      rn_efr    ! fraction of TKE surface value which penetrates in the ocean
[9546]84   REAL(wp) ::      rn_eice   ! =0 ON below sea-ice, =4 OFF when ice fraction > 1/4   
[4147]85   LOGICAL  ::   ln_lc     ! Langmuir cells (LC) as a source term of TKE or not
[9019]86   REAL(wp) ::      rn_lc     ! coef to compute vertical velocity of Langmuir cells
[1239]87
[4147]88   REAL(wp) ::   ri_cri    ! critic Richardson number (deduced from rn_ediff and rn_ediss values)
89   REAL(wp) ::   rmxl_min  ! minimum mixing length value (deduced from rn_ediff and rn_emin values)  [m]
[2528]90   REAL(wp) ::   rhftau_add = 1.e-3_wp     ! add offset   applied to HF part of taum  (nn_etau=3)
91   REAL(wp) ::   rhftau_scl = 1.0_wp       ! scale factor applied to HF part of taum  (nn_etau=3)
[1239]92
[9019]93   REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:)   ::   htau    ! depth of tke penetration (nn_htau)
94   REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::   dissl   ! now mixing lenght of dissipation
95   REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::   apdlr   ! now mixing lenght of dissipation
[1492]96
[1239]97   !! * Substitutions
[12377]98#  include "do_loop_substitute.h90"
[13237]99#  include "domzgr_substitute.h90"
[1239]100   !!----------------------------------------------------------------------
[9598]101   !! NEMO/OCE 4.0 , NEMO Consortium (2018)
[2528]102   !! $Id$
[10068]103   !! Software governed by the CeCILL license (see ./LICENSE)
[1239]104   !!----------------------------------------------------------------------
105CONTAINS
106
[2715]107   INTEGER FUNCTION zdf_tke_alloc()
108      !!----------------------------------------------------------------------
109      !!                ***  FUNCTION zdf_tke_alloc  ***
110      !!----------------------------------------------------------------------
[9019]111      ALLOCATE( htau(jpi,jpj) , dissl(jpi,jpj,jpk) , apdlr(jpi,jpj,jpk) ,   STAT= zdf_tke_alloc )
112      !
[10425]113      CALL mpp_sum ( 'zdftke', zdf_tke_alloc )
114      IF( zdf_tke_alloc /= 0 )   CALL ctl_stop( 'STOP', 'zdf_tke_alloc: failed to allocate arrays' )
[2715]115      !
116   END FUNCTION zdf_tke_alloc
117
118
[12377]119   SUBROUTINE zdf_tke( kt, Kbb, Kmm, p_sh2, p_avm, p_avt )
[1239]120      !!----------------------------------------------------------------------
[1531]121      !!                   ***  ROUTINE zdf_tke  ***
[1239]122      !!
123      !! ** Purpose :   Compute the vertical eddy viscosity and diffusivity
[1492]124      !!              coefficients using a turbulent closure scheme (TKE).
[1239]125      !!
[1492]126      !! ** Method  :   The time evolution of the turbulent kinetic energy (tke)
127      !!              is computed from a prognostic equation :
128      !!         d(en)/dt = avm (d(u)/dz)**2             ! shear production
129      !!                  + d( avm d(en)/dz )/dz         ! diffusion of tke
130      !!                  + avt N^2                      ! stratif. destruc.
131      !!                  - rn_ediss / emxl en**(2/3)    ! Kolmogoroff dissipation
[1239]132      !!      with the boundary conditions:
[1695]133      !!         surface: en = max( rn_emin0, rn_ebb * taum )
[1239]134      !!         bottom : en = rn_emin
[1492]135      !!      The associated critical Richardson number is: ri_cri = 2/(2+rn_ediss/rn_ediff)
136      !!
137      !!        The now Turbulent kinetic energy is computed using the following
138      !!      time stepping: implicit for vertical diffusion term, linearized semi
139      !!      implicit for kolmogoroff dissipation term, and explicit forward for
140      !!      both buoyancy and shear production terms. Therefore a tridiagonal
141      !!      linear system is solved. Note that buoyancy and shear terms are
142      !!      discretized in a energy conserving form (Bruchard 2002).
143      !!
144      !!        The dissipative and mixing length scale are computed from en and
145      !!      the stratification (see tke_avn)
146      !!
147      !!        The now vertical eddy vicosity and diffusivity coefficients are
148      !!      given by:
149      !!              avm = max( avtb, rn_ediff * zmxlm * en^1/2 )
150      !!              avt = max( avmb, pdl * avm                 ) 
[1239]151      !!              eav = max( avmb, avm )
[1492]152      !!      where pdl, the inverse of the Prandtl number is 1 if nn_pdl=0 and
153      !!      given by an empirical funtion of the localRichardson number if nn_pdl=1
[1239]154      !!
155      !! ** Action  :   compute en (now turbulent kinetic energy)
[9019]156      !!                update avt, avm (before vertical eddy coef.)
[1239]157      !!
158      !! References : Gaspar et al., JGR, 1990,
159      !!              Blanke and Delecluse, JPO, 1991
160      !!              Mellor and Blumberg, JPO 2004
161      !!              Axell, JGR, 2002
[1492]162      !!              Bruchard OM 2002
[1239]163      !!----------------------------------------------------------------------
[9019]164      INTEGER                   , INTENT(in   ) ::   kt             ! ocean time step
[12377]165      INTEGER                   , INTENT(in   ) ::   Kbb, Kmm       ! ocean time level indices
[9019]166      REAL(wp), DIMENSION(:,:,:), INTENT(in   ) ::   p_sh2          ! shear production term
167      REAL(wp), DIMENSION(:,:,:), INTENT(inout) ::   p_avm, p_avt   !  momentum and tracer Kz (w-points)
[1492]168      !!----------------------------------------------------------------------
[1481]169      !
[12377]170      CALL tke_tke( Kbb, Kmm, p_sh2, p_avm, p_avt )   ! now tke (en)
[5656]171      !
[12377]172      CALL tke_avn( Kbb, Kmm,        p_avm, p_avt )   ! now avt, avm, dissl
[3632]173      !
[5656]174  END SUBROUTINE zdf_tke
[1239]175
[1492]176
[12377]177   SUBROUTINE tke_tke( Kbb, Kmm, p_sh2, p_avm, p_avt )
[1239]178      !!----------------------------------------------------------------------
[1492]179      !!                   ***  ROUTINE tke_tke  ***
180      !!
181      !! ** Purpose :   Compute the now Turbulente Kinetic Energy (TKE)
182      !!
183      !! ** Method  : - TKE surface boundary condition
[2528]184      !!              - source term due to Langmuir cells (Axell JGR 2002) (ln_lc=T)
[9019]185      !!              - source term due to shear (= Kz dz[Ub] * dz[Un] )
[1492]186      !!              - Now TKE : resolution of the TKE equation by inverting
187      !!                a tridiagonal linear system by a "methode de chasse"
188      !!              - increase TKE due to surface and internal wave breaking
[9019]189      !!             NB: when sea-ice is present, both LC parameterization
190      !!                 and TKE penetration are turned off when the ice fraction
191      !!                 is smaller than 0.25
[1492]192      !!
193      !! ** Action  : - en : now turbulent kinetic energy)
[1239]194      !! ---------------------------------------------------------------------
[9019]195      USE zdf_oce , ONLY : en   ! ocean vertical physics
196      !!
[12377]197      INTEGER                    , INTENT(in   ) ::   Kbb, Kmm       ! ocean time level indices
[9019]198      REAL(wp), DIMENSION(:,:,:) , INTENT(in   ) ::   p_sh2          ! shear production term
199      REAL(wp), DIMENSION(:,:,:) , INTENT(in   ) ::   p_avm, p_avt   ! vertical eddy viscosity & diffusivity (w-points)
200      !
201      INTEGER ::   ji, jj, jk              ! dummy loop arguments
202      REAL(wp) ::   zetop, zebot, zmsku, zmskv ! local scalars
203      REAL(wp) ::   zrhoa  = 1.22              ! Air density kg/m3
204      REAL(wp) ::   zcdrag = 1.5e-3            ! drag coefficient
205      REAL(wp) ::   zbbrau, zri                ! local scalars
206      REAL(wp) ::   zfact1, zfact2, zfact3     !   -         -
207      REAL(wp) ::   ztx2  , zty2  , zcof       !   -         -
208      REAL(wp) ::   ztau  , zdif               !   -         -
209      REAL(wp) ::   zus   , zwlc  , zind       !   -         -
210      REAL(wp) ::   zzd_up, zzd_lw             !   -         -
211      INTEGER , DIMENSION(jpi,jpj)     ::   imlc
[10425]212      REAL(wp), DIMENSION(jpi,jpj)     ::   zhlc, zfr_i
[9019]213      REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zpelc, zdiag, zd_up, zd_lw
[1239]214      !!--------------------------------------------------------------------
[1492]215      !
[12489]216      zbbrau = rn_ebb / rho0       ! Local constant initialisation
217      zfact1 = -.5_wp * rn_Dt 
218      zfact2 = 1.5_wp * rn_Dt * rn_ediss
[2528]219      zfact3 = 0.5_wp       * rn_ediss
[1492]220      !
221      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[9019]222      !                     !  Surface/top/bottom boundary condition on tke
[1492]223      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[12698]224      !
[13295]225      DO_2D( 0, 0, 0, 0 )
[12377]226         en(ji,jj,1) = MAX( rn_emin0, zbbrau * taum(ji,jj) ) * tmask(ji,jj,1)
227      END_2D
[9019]228      !
[1492]229      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
230      !                     !  Bottom boundary condition on tke
231      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[1719]232      !
[12489]233      !   en(bot)   = (ebb0/rho0)*0.5*sqrt(u_botfr^2+v_botfr^2) (min value rn_emin)
[9019]234      ! where ebb0 does not includes surface wave enhancement (i.e. ebb0=3.75)
235      ! Note that stress averaged is done using an wet-only calculation of u and v at t-point like in zdfsh2
[1492]236      !
[13461]237      IF( .NOT.ln_drg_OFF ) THEN    !== friction used as top/bottom boundary condition on TKE
[9019]238         !
[13461]239         DO_2D( 0, 0, 0, 0 )        ! bottom friction
[12377]240            zmsku = ( 2. - umask(ji-1,jj,mbkt(ji,jj)) * umask(ji,jj,mbkt(ji,jj)) )
241            zmskv = ( 2. - vmask(ji,jj-1,mbkt(ji,jj)) * vmask(ji,jj,mbkt(ji,jj)) )
[12489]242            !                       ! where 0.001875 = (rn_ebb0/rho0) * 0.5 = 3.75*0.5/1000. (CAUTION CdU<0)
[12377]243            zebot = - 0.001875_wp * rCdU_bot(ji,jj) * SQRT(  ( zmsku*( uu(ji,jj,mbkt(ji,jj),Kbb)+uu(ji-1,jj,mbkt(ji,jj),Kbb) ) )**2  &
244               &                                           + ( zmskv*( vv(ji,jj,mbkt(ji,jj),Kbb)+vv(ji,jj-1,mbkt(ji,jj),Kbb) ) )**2  )
245            en(ji,jj,mbkt(ji,jj)+1) = MAX( zebot, rn_emin ) * ssmask(ji,jj)
246         END_2D
[13461]247         IF( ln_isfcav ) THEN
248            DO_2D( 0, 0, 0, 0 )     ! top friction
[12377]249               zmsku = ( 2. - umask(ji-1,jj,mikt(ji,jj)) * umask(ji,jj,mikt(ji,jj)) )
250               zmskv = ( 2. - vmask(ji,jj-1,mikt(ji,jj)) * vmask(ji,jj,mikt(ji,jj)) )
[12489]251               !                             ! where 0.001875 = (rn_ebb0/rho0) * 0.5 = 3.75*0.5/1000.  (CAUTION CdU<0)
[12377]252               zetop = - 0.001875_wp * rCdU_top(ji,jj) * SQRT(  ( zmsku*( uu(ji,jj,mikt(ji,jj),Kbb)+uu(ji-1,jj,mikt(ji,jj),Kbb) ) )**2  &
253                  &                                           + ( zmskv*( vv(ji,jj,mikt(ji,jj),Kbb)+vv(ji,jj-1,mikt(ji,jj),Kbb) ) )**2  )
[12702]254               ! (1._wp - tmask(ji,jj,1)) * ssmask(ji,jj) = 1 where ice shelves are present
255               en(ji,jj,mikt(ji,jj)) = en(ji,jj,1)           * tmask(ji,jj,1) &
256                  &                  + MAX( zetop, rn_emin ) * (1._wp - tmask(ji,jj,1)) * ssmask(ji,jj)
[12377]257            END_2D
[9019]258         ENDIF
259         !
260      ENDIF
261      !
[1492]262      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[9019]263      IF( ln_lc ) THEN      !  Langmuir circulation source term added to tke   !   (Axell JGR 2002)
[1492]264         !                  !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[1239]265         !
[1492]266         !                        !* total energy produce by LC : cumulative sum over jk
[12377]267         zpelc(:,:,1) =  MAX( rn2b(:,:,1), 0._wp ) * gdepw(:,:,1,Kmm) * e3w(:,:,1,Kmm)
[1239]268         DO jk = 2, jpk
[13237]269            zpelc(:,:,jk)  = zpelc(:,:,jk-1) +   &
270               &        MAX( rn2b(:,:,jk), 0._wp ) * gdepw(:,:,jk,Kmm) * e3w(:,:,jk,Kmm)
[1239]271         END DO
[1492]272         !                        !* finite Langmuir Circulation depth
[1705]273         zcof = 0.5 * 0.016 * 0.016 / ( zrhoa * zcdrag )
[7753]274         imlc(:,:) = mbkt(:,:) + 1       ! Initialization to the number of w ocean point (=2 over land)
[13295]275         DO_3DS( 1, 1, 1, 1, jpkm1, 2, -1 )
[12377]276            zus  = zcof * taum(ji,jj)
277            IF( zpelc(ji,jj,jk) > zus )   imlc(ji,jj) = jk
278         END_3D
[1492]279         !                               ! finite LC depth
[13295]280         DO_2D( 1, 1, 1, 1 )
[12377]281            zhlc(ji,jj) = gdepw(ji,jj,imlc(ji,jj),Kmm)
282         END_2D
[1705]283         zcof = 0.016 / SQRT( zrhoa * zcdrag )
[13295]284         DO_2D( 0, 0, 0, 0 )
[12377]285            zus  = zcof * SQRT( taum(ji,jj) )           ! Stokes drift
286            zfr_i(ji,jj) = ( 1._wp - 4._wp * fr_i(ji,jj) ) * zus * zus * zus * tmask(ji,jj,1) ! zus > 0. ok
287            IF (zfr_i(ji,jj) < 0. ) zfr_i(ji,jj) = 0.
288         END_2D
[13295]289         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]290            IF ( zfr_i(ji,jj) /= 0. ) THEN               
291               ! vertical velocity due to LC   
292               IF ( gdepw(ji,jj,jk,Kmm) - zhlc(ji,jj) < 0 .AND. wmask(ji,jj,jk) /= 0. ) THEN
293                  !                                           ! vertical velocity due to LC
294                  zwlc = rn_lc * SIN( rpi * gdepw(ji,jj,jk,Kmm) / zhlc(ji,jj) )   ! warning: optimization: zus^3 is in zfr_i
295                  !                                           ! TKE Langmuir circulation source term
[12489]296                  en(ji,jj,jk) = en(ji,jj,jk) + rn_Dt * zfr_i(ji,jj) * ( zwlc * zwlc * zwlc ) / zhlc(ji,jj)
[12377]297               ENDIF
298            ENDIF
299         END_3D
[1239]300         !
301      ENDIF
[1492]302      !
303      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
304      !                     !  Now Turbulent kinetic energy (output in en)
305      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
306      !                     ! Resolution of a tridiagonal linear system by a "methode de chasse"
307      !                     ! computation from level 2 to jpkm1  (e(1) already computed and e(jpk)=0 ).
308      !                     ! zdiag : diagonal zd_up : upper diagonal zd_lw : lower diagonal
309      !
[9019]310      IF( nn_pdl == 1 ) THEN      !* Prandtl number = F( Ri )
[13295]311         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]312            !                             ! local Richardson number
[13226]313            IF (rn2b(ji,jj,jk) <= 0.0_wp) then
314                zri = 0.0_wp
315            ELSE
316                zri = rn2b(ji,jj,jk) * p_avm(ji,jj,jk) / ( p_sh2(ji,jj,jk) + rn_bshear )
317            ENDIF
[12377]318            !                             ! inverse of Prandtl number
319            apdlr(ji,jj,jk) = MAX(  0.1_wp,  ri_cri / MAX( ri_cri , zri )  )
320         END_3D
[5656]321      ENDIF
[5836]322      !         
[13295]323      DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]324         zcof   = zfact1 * tmask(ji,jj,jk)
325         !                                   ! A minimum of 2.e-5 m2/s is imposed on TKE vertical
326         !                                   ! eddy coefficient (ensure numerical stability)
327         zzd_up = zcof * MAX(  p_avm(ji,jj,jk+1) + p_avm(ji,jj,jk  ) , 2.e-5_wp  )   &  ! upper diagonal
[13237]328            &          /    (  e3t(ji,jj,jk  ,Kmm)   &
329            &                * e3w(ji,jj,jk  ,Kmm)  )
[12377]330         zzd_lw = zcof * MAX(  p_avm(ji,jj,jk  ) + p_avm(ji,jj,jk-1) , 2.e-5_wp  )   &  ! lower diagonal
[13237]331            &          /    (  e3t(ji,jj,jk-1,Kmm)   &
332            &                * e3w(ji,jj,jk  ,Kmm)  )
[12377]333         !
334         zd_up(ji,jj,jk) = zzd_up            ! Matrix (zdiag, zd_up, zd_lw)
335         zd_lw(ji,jj,jk) = zzd_lw
336         zdiag(ji,jj,jk) = 1._wp - zzd_lw - zzd_up + zfact2 * dissl(ji,jj,jk) * wmask(ji,jj,jk)
337         !
338         !                                   ! right hand side in en
[12489]339         en(ji,jj,jk) = en(ji,jj,jk) + rn_Dt * (  p_sh2(ji,jj,jk)                        &   ! shear
[12377]340            &                                 - p_avt(ji,jj,jk) * rn2(ji,jj,jk)          &   ! stratification
341            &                                 + zfact3 * dissl(ji,jj,jk) * en(ji,jj,jk)  &   ! dissipation
342            &                                ) * wmask(ji,jj,jk)
343      END_3D
[5120]344      !                          !* Matrix inversion from level 2 (tke prescribed at level 1)
[13295]345      DO_3D( 0, 0, 0, 0, 3, jpkm1 )
[12377]346         zdiag(ji,jj,jk) = zdiag(ji,jj,jk) - zd_lw(ji,jj,jk) * zd_up(ji,jj,jk-1) / zdiag(ji,jj,jk-1)
347      END_3D
[13295]348      DO_2D( 0, 0, 0, 0 )
[12377]349         zd_lw(ji,jj,2) = en(ji,jj,2) - zd_lw(ji,jj,2) * en(ji,jj,1)    ! Surface boudary conditions on tke
350      END_2D
[13295]351      DO_3D( 0, 0, 0, 0, 3, jpkm1 )
[12377]352         zd_lw(ji,jj,jk) = en(ji,jj,jk) - zd_lw(ji,jj,jk) / zdiag(ji,jj,jk-1) *zd_lw(ji,jj,jk-1)
353      END_3D
[13295]354      DO_2D( 0, 0, 0, 0 )
[12377]355         en(ji,jj,jpkm1) = zd_lw(ji,jj,jpkm1) / zdiag(ji,jj,jpkm1)
356      END_2D
[13295]357      DO_3DS( 0, 0, 0, 0, jpk-2, 2, -1 )
[12377]358         en(ji,jj,jk) = ( zd_lw(ji,jj,jk) - zd_up(ji,jj,jk) * en(ji,jj,jk+1) ) / zdiag(ji,jj,jk)
359      END_3D
[13295]360      DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]361         en(ji,jj,jk) = MAX( en(ji,jj,jk), rn_emin ) * wmask(ji,jj,jk)
362      END_3D
[9019]363      !
[1492]364      !                            !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
365      !                            !  TKE due to surface and internal wave breaking
366      !                            !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[6140]367!!gm BUG : in the exp  remove the depth of ssh !!!
[12377]368!!gm       i.e. use gde3w in argument (gdepw(:,:,:,Kmm))
[6140]369     
370     
[2528]371      IF( nn_etau == 1 ) THEN           !* penetration below the mixed layer (rn_efr fraction)
[13295]372         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]373            en(ji,jj,jk) = en(ji,jj,jk) + rn_efr * en(ji,jj,1) * EXP( -gdepw(ji,jj,jk,Kmm) / htau(ji,jj) )   &
374               &                                 * MAX(0.,1._wp - rn_eice *fr_i(ji,jj) )  * wmask(ji,jj,jk) * tmask(ji,jj,1)
375         END_3D
[2528]376      ELSEIF( nn_etau == 2 ) THEN       !* act only at the base of the mixed layer (jk=nmln)  (rn_efr fraction)
[13295]377         DO_2D( 0, 0, 0, 0 )
[12377]378            jk = nmln(ji,jj)
379            en(ji,jj,jk) = en(ji,jj,jk) + rn_efr * en(ji,jj,1) * EXP( -gdepw(ji,jj,jk,Kmm) / htau(ji,jj) )   &
380               &                                 * MAX(0.,1._wp - rn_eice *fr_i(ji,jj) )  * wmask(ji,jj,jk) * tmask(ji,jj,1)
381         END_2D
[2528]382      ELSEIF( nn_etau == 3 ) THEN       !* penetration belox the mixed layer (HF variability)
[13295]383         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]384            ztx2 = utau(ji-1,jj  ) + utau(ji,jj)
385            zty2 = vtau(ji  ,jj-1) + vtau(ji,jj)
386            ztau = 0.5_wp * SQRT( ztx2 * ztx2 + zty2 * zty2 ) * tmask(ji,jj,1)    ! module of the mean stress
387            zdif = taum(ji,jj) - ztau                            ! mean of modulus - modulus of the mean
388            zdif = rhftau_scl * MAX( 0._wp, zdif + rhftau_add )  ! apply some modifications...
389            en(ji,jj,jk) = en(ji,jj,jk) + zbbrau * zdif * EXP( -gdepw(ji,jj,jk,Kmm) / htau(ji,jj) )   &
390               &                        * MAX(0.,1._wp - rn_eice *fr_i(ji,jj) ) * wmask(ji,jj,jk) * tmask(ji,jj,1)
391         END_3D
[1239]392      ENDIF
[1492]393      !
[1239]394   END SUBROUTINE tke_tke
395
[1492]396
[12377]397   SUBROUTINE tke_avn( Kbb, Kmm, p_avm, p_avt )
[1239]398      !!----------------------------------------------------------------------
[1492]399      !!                   ***  ROUTINE tke_avn  ***
[1239]400      !!
[1492]401      !! ** Purpose :   Compute the vertical eddy viscosity and diffusivity
402      !!
403      !! ** Method  :   At this stage, en, the now TKE, is known (computed in
404      !!              the tke_tke routine). First, the now mixing lenth is
405      !!      computed from en and the strafification (N^2), then the mixings
406      !!      coefficients are computed.
407      !!              - Mixing length : a first evaluation of the mixing lengh
408      !!      scales is:
409      !!                      mxl = sqrt(2*en) / N 
410      !!      where N is the brunt-vaisala frequency, with a minimum value set
[2528]411      !!      to rmxl_min (rn_mxl0) in the interior (surface) ocean.
[1492]412      !!        The mixing and dissipative length scale are bound as follow :
413      !!         nn_mxl=0 : mxl bounded by the distance to surface and bottom.
414      !!                        zmxld = zmxlm = mxl
415      !!         nn_mxl=1 : mxl bounded by the e3w and zmxld = zmxlm = mxl
416      !!         nn_mxl=2 : mxl bounded such that the vertical derivative of mxl is
417      !!                    less than 1 (|d/dz(mxl)|<1) and zmxld = zmxlm = mxl
418      !!         nn_mxl=3 : mxl is bounded from the surface to the bottom usings
419      !!                    |d/dz(xml)|<1 to obtain lup, and from the bottom to
420      !!                    the surface to obtain ldown. the resulting length
421      !!                    scales are:
422      !!                        zmxld = sqrt( lup * ldown )
423      !!                        zmxlm = min ( lup , ldown )
424      !!              - Vertical eddy viscosity and diffusivity:
425      !!                      avm = max( avtb, rn_ediff * zmxlm * en^1/2 )
426      !!                      avt = max( avmb, pdlr * avm ) 
427      !!      with pdlr=1 if nn_pdl=0, pdlr=1/pdl=F(Ri) otherwise.
428      !!
[9019]429      !! ** Action  : - avt, avm : now vertical eddy diffusivity and viscosity (w-point)
[1239]430      !!----------------------------------------------------------------------
[9019]431      USE zdf_oce , ONLY : en, avtb, avmb, avtb_2d   ! ocean vertical physics
432      !!
[12377]433      INTEGER                   , INTENT(in   ) ::   Kbb, Kmm       ! ocean time level indices
[9019]434      REAL(wp), DIMENSION(:,:,:), INTENT(  out) ::   p_avm, p_avt   ! vertical eddy viscosity & diffusivity (w-points)
435      !
[2715]436      INTEGER  ::   ji, jj, jk   ! dummy loop indices
[9019]437      REAL(wp) ::   zrn2, zraug, zcoef, zav   ! local scalars
438      REAL(wp) ::   zdku,   zdkv, zsqen       !   -      -
[13007]439      REAL(wp) ::   zemxl, zemlm, zemlp, zmaxice       !   -      -
[9019]440      REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zmxlm, zmxld   ! 3D workspace
[1239]441      !!--------------------------------------------------------------------
[3294]442      !
[1492]443      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
444      !                     !  Mixing length
445      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
446      !
447      !                     !* Buoyancy length scale: l=sqrt(2*e/n**2)
448      !
[5120]449      ! initialisation of interior minimum value (avoid a 2d loop with mikt)
[7753]450      zmxlm(:,:,:)  = rmxl_min   
451      zmxld(:,:,:)  = rmxl_min
[5120]452      !
[13012]453     IF( ln_mxl0 ) THEN            ! surface mixing length = F(stress) : l=vkarmn*2.e5*taum/(rho0*g)
[13007]454         !
[13012]455         zraug = vkarmn * 2.e5_wp / ( rho0 * grav )
[13007]456#if ! defined key_si3 && ! defined key_cice
[13295]457         DO_2D( 0, 0, 0, 0 )
[13007]458            zmxlm(ji,jj,1) =  zraug * taum(ji,jj) * tmask(ji,jj,1)
[12377]459         END_2D
[13007]460#else
461         SELECT CASE( nn_mxlice )             ! Type of scaling under sea-ice
462         !
463         CASE( 0 )                      ! No scaling under sea-ice
[13295]464            DO_2D( 0, 0, 0, 0 )
[13007]465               zmxlm(ji,jj,1) = zraug * taum(ji,jj) * tmask(ji,jj,1)
466            END_2D
467            !
468         CASE( 1 )                           ! scaling with constant sea-ice thickness
[13295]469            DO_2D( 0, 0, 0, 0 )
[13007]470               zmxlm(ji,jj,1) =  ( ( 1. - fr_i(ji,jj) ) * zraug * taum(ji,jj) + fr_i(ji,jj) * rn_mxlice ) * tmask(ji,jj,1)
471            END_2D
472            !
473         CASE( 2 )                                 ! scaling with mean sea-ice thickness
[13295]474            DO_2D( 0, 0, 0, 0 )
[13007]475#if defined key_si3
476               zmxlm(ji,jj,1) = ( ( 1. - fr_i(ji,jj) ) * zraug * taum(ji,jj) + fr_i(ji,jj) * hm_i(ji,jj) * 2. ) * tmask(ji,jj,1)
477#elif defined key_cice
478               zmaxice = MAXVAL( h_i(ji,jj,:) )
479               zmxlm(ji,jj,1) = ( ( 1. - fr_i(ji,jj) ) * zraug * taum(ji,jj) + fr_i(ji,jj) * zmaxice ) * tmask(ji,jj,1)
480#endif
481            END_2D
482            !
483         CASE( 3 )                                 ! scaling with max sea-ice thickness
[13295]484            DO_2D( 0, 0, 0, 0 )
[13007]485               zmaxice = MAXVAL( h_i(ji,jj,:) )
486               zmxlm(ji,jj,1) = ( ( 1. - fr_i(ji,jj) ) * zraug * taum(ji,jj) + fr_i(ji,jj) * zmaxice ) * tmask(ji,jj,1)
487            END_2D
488            !
489         END SELECT
490#endif
491         !
[13295]492         DO_2D( 0, 0, 0, 0 )
[13007]493            zmxlm(ji,jj,1) = MAX( rn_mxl0, zmxlm(ji,jj,1) )
494         END_2D
495         !
496      ELSE
[7753]497         zmxlm(:,:,1) = rn_mxl0
[1239]498      ENDIF
[13007]499
[1239]500      !
[13295]501      DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]502         zrn2 = MAX( rn2(ji,jj,jk), rsmall )
503         zmxlm(ji,jj,jk) = MAX(  rmxl_min,  SQRT( 2._wp * en(ji,jj,jk) / zrn2 )  )
504      END_3D
[1492]505      !
506      !                     !* Physical limits for the mixing length
507      !
[7753]508      zmxld(:,:, 1 ) = zmxlm(:,:,1)   ! surface set to the minimum value
509      zmxld(:,:,jpk) = rmxl_min       ! last level  set to the minimum value
[1492]510      !
[1239]511      SELECT CASE ( nn_mxl )
512      !
[5836]513 !!gm Not sure of that coding for ISF....
[12377]514      ! where wmask = 0 set zmxlm == e3w(:,:,:,Kmm)
[1239]515      CASE ( 0 )           ! bounded by the distance to surface and bottom
[13295]516         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]517            zemxl = MIN( gdepw(ji,jj,jk,Kmm) - gdepw(ji,jj,mikt(ji,jj),Kmm), zmxlm(ji,jj,jk),   &
518            &            gdepw(ji,jj,mbkt(ji,jj)+1,Kmm) - gdepw(ji,jj,jk,Kmm) )
519            ! wmask prevent zmxlm = 0 if jk = mikt(ji,jj)
[13237]520            zmxlm(ji,jj,jk) = zemxl * wmask(ji,jj,jk)   &
521               &            + MIN( zmxlm(ji,jj,jk) , e3w(ji,jj,jk,Kmm) ) * (1 - wmask(ji,jj,jk))
522            zmxld(ji,jj,jk) = zemxl * wmask(ji,jj,jk)   &
523               &            + MIN( zmxlm(ji,jj,jk) , e3w(ji,jj,jk,Kmm) ) * (1 - wmask(ji,jj,jk))
[12377]524         END_3D
[1239]525         !
526      CASE ( 1 )           ! bounded by the vertical scale factor
[13295]527         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]528            zemxl = MIN( e3w(ji,jj,jk,Kmm), zmxlm(ji,jj,jk) )
529            zmxlm(ji,jj,jk) = zemxl
530            zmxld(ji,jj,jk) = zemxl
531         END_3D
[1239]532         !
533      CASE ( 2 )           ! |dk[xml]| bounded by e3t :
[13295]534         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[13237]535            zmxlm(ji,jj,jk) =   &
536               &    MIN( zmxlm(ji,jj,jk-1) + e3t(ji,jj,jk-1,Kmm), zmxlm(ji,jj,jk) )
[12377]537         END_3D
[13295]538         DO_3DS( 0, 0, 0, 0, jpkm1, 2, -1 )
[12377]539            zemxl = MIN( zmxlm(ji,jj,jk+1) + e3t(ji,jj,jk+1,Kmm), zmxlm(ji,jj,jk) )
540            zmxlm(ji,jj,jk) = zemxl
541            zmxld(ji,jj,jk) = zemxl
542         END_3D
[1239]543         !
544      CASE ( 3 )           ! lup and ldown, |dk[xml]| bounded by e3t :
[13295]545         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[13237]546            zmxld(ji,jj,jk) =    &
547               &    MIN( zmxld(ji,jj,jk-1) + e3t(ji,jj,jk-1,Kmm), zmxlm(ji,jj,jk) )
[12377]548         END_3D
[13295]549         DO_3DS( 0, 0, 0, 0, jpkm1, 2, -1 )
[13237]550            zmxlm(ji,jj,jk) =   &
551               &    MIN( zmxlm(ji,jj,jk+1) + e3t(ji,jj,jk+1,Kmm), zmxlm(ji,jj,jk) )
[12377]552         END_3D
[13295]553         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12377]554            zemlm = MIN ( zmxld(ji,jj,jk),  zmxlm(ji,jj,jk) )
555            zemlp = SQRT( zmxld(ji,jj,jk) * zmxlm(ji,jj,jk) )
556            zmxlm(ji,jj,jk) = zemlm
557            zmxld(ji,jj,jk) = zemlp
558         END_3D
[1239]559         !
560      END SELECT
[1492]561      !
562      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[9019]563      !                     !  Vertical eddy viscosity and diffusivity  (avm and avt)
[1492]564      !                     !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[13295]565      DO_3D( 0, 0, 0, 0, 1, jpkm1 )
[12377]566         zsqen = SQRT( en(ji,jj,jk) )
567         zav   = rn_ediff * zmxlm(ji,jj,jk) * zsqen
568         p_avm(ji,jj,jk) = MAX( zav,                  avmb(jk) ) * wmask(ji,jj,jk)
569         p_avt(ji,jj,jk) = MAX( zav, avtb_2d(ji,jj) * avtb(jk) ) * wmask(ji,jj,jk)
570         dissl(ji,jj,jk) = zsqen / zmxld(ji,jj,jk)
571      END_3D
[1492]572      !
573      !
574      IF( nn_pdl == 1 ) THEN      !* Prandtl number case: update avt
[13295]575         DO_3D( 0, 0, 0, 0, 2, jpkm1 )
[12698]576            p_avt(ji,jj,jk)   = MAX( apdlr(ji,jj,jk) * p_avt(ji,jj,jk), avtb_2d(ji,jj) * avtb(jk) ) * wmask(ji,jj,jk)
[12377]577         END_3D
[1239]578      ENDIF
[9019]579      !
[12377]580      IF(sn_cfctl%l_prtctl) THEN
[9440]581         CALL prt_ctl( tab3d_1=en   , clinfo1=' tke  - e: ', tab3d_2=p_avt, clinfo2=' t: ', kdim=jpk)
582         CALL prt_ctl( tab3d_1=p_avm, clinfo1=' tke  - m: ', kdim=jpk )
[1239]583      ENDIF
584      !
[1492]585   END SUBROUTINE tke_avn
[1239]586
[1492]587
[12377]588   SUBROUTINE zdf_tke_init( Kmm )
[1239]589      !!----------------------------------------------------------------------
[2528]590      !!                  ***  ROUTINE zdf_tke_init  ***
[1239]591      !!                     
592      !! ** Purpose :   Initialization of the vertical eddy diffivity and
[1492]593      !!              viscosity when using a tke turbulent closure scheme
[1239]594      !!
[1601]595      !! ** Method  :   Read the namzdf_tke namelist and check the parameters
[1492]596      !!              called at the first timestep (nit000)
[1239]597      !!
[1601]598      !! ** input   :   Namlist namzdf_tke
[1239]599      !!
600      !! ** Action  :   Increase by 1 the nstop flag is setting problem encounter
601      !!----------------------------------------------------------------------
[9019]602      USE zdf_oce , ONLY : ln_zdfiwm   ! Internal Wave Mixing flag
603      !!
[12377]604      INTEGER, INTENT(in) ::   Kmm          ! time level index
605      INTEGER             ::   ji, jj, jk   ! dummy loop indices
606      INTEGER             ::   ios
[1239]607      !!
[13007]608      NAMELIST/namzdf_tke/ rn_ediff, rn_ediss , rn_ebb   , rn_emin  ,  &
609         &                 rn_emin0, rn_bshear, nn_mxl   , ln_mxl0  ,  &
610         &                 rn_mxl0 , nn_mxlice, rn_mxlice,             &
[13461]611         &                 nn_pdl  , ln_lc    , rn_lc    ,             &
[13007]612         &                 nn_etau , nn_htau  , rn_efr   , rn_eice 
[1239]613      !!----------------------------------------------------------------------
[2715]614      !
[4147]615      READ  ( numnam_ref, namzdf_tke, IOSTAT = ios, ERR = 901)
[11536]616901   IF( ios /= 0 ) CALL ctl_nam ( ios , 'namzdf_tke in reference namelist' )
[4147]617
618      READ  ( numnam_cfg, namzdf_tke, IOSTAT = ios, ERR = 902 )
[11536]619902   IF( ios >  0 ) CALL ctl_nam ( ios , 'namzdf_tke in configuration namelist' )
[4624]620      IF(lwm) WRITE ( numond, namzdf_tke )
[2715]621      !
[2528]622      ri_cri   = 2._wp    / ( 2._wp + rn_ediss / rn_ediff )   ! resulting critical Richardson number
[2715]623      !
[1492]624      IF(lwp) THEN                    !* Control print
[1239]625         WRITE(numout,*)
[2528]626         WRITE(numout,*) 'zdf_tke_init : tke turbulent closure scheme - initialisation'
627         WRITE(numout,*) '~~~~~~~~~~~~'
[1601]628         WRITE(numout,*) '   Namelist namzdf_tke : set tke mixing parameters'
[1705]629         WRITE(numout,*) '      coef. to compute avt                        rn_ediff  = ', rn_ediff
630         WRITE(numout,*) '      Kolmogoroff dissipation coef.               rn_ediss  = ', rn_ediss
631         WRITE(numout,*) '      tke surface input coef.                     rn_ebb    = ', rn_ebb
632         WRITE(numout,*) '      minimum value of tke                        rn_emin   = ', rn_emin
633         WRITE(numout,*) '      surface minimum value of tke                rn_emin0  = ', rn_emin0
[9019]634         WRITE(numout,*) '      prandl number flag                          nn_pdl    = ', nn_pdl
[1705]635         WRITE(numout,*) '      background shear (>0)                       rn_bshear = ', rn_bshear
636         WRITE(numout,*) '      mixing length type                          nn_mxl    = ', nn_mxl
[9019]637         WRITE(numout,*) '         surface mixing length = F(stress) or not    ln_mxl0   = ', ln_mxl0
[13007]638         IF( ln_mxl0 ) THEN
639            WRITE(numout,*) '      type of scaling under sea-ice               nn_mxlice = ', nn_mxlice
640            IF( nn_mxlice == 1 ) &
641            WRITE(numout,*) '      ice thickness when scaling under sea-ice    rn_mxlice = ', rn_mxlice
642         ENDIF         
[9019]643         WRITE(numout,*) '         surface  mixing length minimum value        rn_mxl0   = ', rn_mxl0
644         WRITE(numout,*) '      Langmuir cells parametrization              ln_lc     = ', ln_lc
645         WRITE(numout,*) '         coef to compute vertical velocity of LC     rn_lc  = ', rn_lc
[1705]646         WRITE(numout,*) '      test param. to add tke induced by wind      nn_etau   = ', nn_etau
[9019]647         WRITE(numout,*) '          type of tke penetration profile            nn_htau   = ', nn_htau
648         WRITE(numout,*) '          fraction of TKE that penetrates            rn_efr    = ', rn_efr
[9546]649         WRITE(numout,*) '          below sea-ice:  =0 ON                      rn_eice   = ', rn_eice
650         WRITE(numout,*) '          =4 OFF when ice fraction > 1/4   '
[13461]651         IF( .NOT.ln_drg_OFF ) THEN
[9169]652            WRITE(numout,*)
[9019]653            WRITE(numout,*) '   Namelist namdrg_top/_bot:   used values:'
654            WRITE(numout,*) '      top    ocean cavity roughness (m)          rn_z0(_top)= ', r_z0_top
655            WRITE(numout,*) '      Bottom seafloor     roughness (m)          rn_z0(_bot)= ', r_z0_bot
656         ENDIF
657         WRITE(numout,*)
[9190]658         WRITE(numout,*) '   ==>>>   critical Richardson nb with your parameters  ri_cri = ', ri_cri
[9019]659         WRITE(numout,*)
[1239]660      ENDIF
[2715]661      !
[9019]662      IF( ln_zdfiwm ) THEN          ! Internal wave-driven mixing
663         rn_emin  = 1.e-10_wp             ! specific values of rn_emin & rmxl_min are used
664         rmxl_min = 1.e-03_wp             ! associated avt minimum = molecular salt diffusivity (10^-9 m2/s)
[9190]665         IF(lwp) WRITE(numout,*) '   ==>>>   Internal wave-driven mixing case:   force   rn_emin = 1.e-10 and rmxl_min = 1.e-3'
[9019]666      ELSE                          ! standard case : associated avt minimum = molecular viscosity (10^-6 m2/s)
667         rmxl_min = 1.e-6_wp / ( rn_ediff * SQRT( rn_emin ) )    ! resulting minimum length to recover molecular viscosity
[9190]668         IF(lwp) WRITE(numout,*) '   ==>>>   minimum mixing length with your parameters rmxl_min = ', rmxl_min
[9019]669      ENDIF
670      !
[2715]671      !                              ! allocate tke arrays
672      IF( zdf_tke_alloc() /= 0 )   CALL ctl_stop( 'STOP', 'zdf_tke_init : unable to allocate arrays' )
673      !
[1492]674      !                               !* Check of some namelist values
[4990]675      IF( nn_mxl  < 0   .OR.  nn_mxl  > 3 )   CALL ctl_stop( 'bad flag: nn_mxl is  0, 1 or 2 ' )
676      IF( nn_pdl  < 0   .OR.  nn_pdl  > 1 )   CALL ctl_stop( 'bad flag: nn_pdl is  0 or 1    ' )
677      IF( nn_htau < 0   .OR.  nn_htau > 1 )   CALL ctl_stop( 'bad flag: nn_htau is 0, 1 or 2 ' )
[5407]678      IF( nn_etau == 3 .AND. .NOT. ln_cpl )   CALL ctl_stop( 'nn_etau == 3 : HF taum only known in coupled mode' )
[9019]679      !
[2528]680      IF( ln_mxl0 ) THEN
[9169]681         IF(lwp) WRITE(numout,*)
[9190]682         IF(lwp) WRITE(numout,*) '   ==>>>   use a surface mixing length = F(stress) :   set rn_mxl0 = rmxl_min'
[2528]683         rn_mxl0 = rmxl_min
684      ENDIF
685     
[12377]686      IF( nn_etau == 2  )   CALL zdf_mxl( nit000, Kmm )      ! Initialization of nmln
[1239]687
[1492]688      !                               !* depth of penetration of surface tke
689      IF( nn_etau /= 0 ) THEN     
[1601]690         SELECT CASE( nn_htau )             ! Choice of the depth of penetration
[2528]691         CASE( 0 )                                 ! constant depth penetration (here 10 meters)
[7753]692            htau(:,:) = 10._wp
[2528]693         CASE( 1 )                                 ! F(latitude) : 0.5m to 30m poleward of 40 degrees
[7753]694            htau(:,:) = MAX(  0.5_wp, MIN( 30._wp, 45._wp* ABS( SIN( rpi/180._wp * gphit(:,:) ) ) )   )           
[1492]695         END SELECT
696      ENDIF
[9019]697      !                                !* read or initialize all required files
698      CALL tke_rst( nit000, 'READ' )      ! (en, avt_k, avm_k, dissl)
[1239]699      !
[9367]700      IF( lwxios ) THEN
701         CALL iom_set_rstw_var_active('en')
702         CALL iom_set_rstw_var_active('avt_k')
703         CALL iom_set_rstw_var_active('avm_k')
704         CALL iom_set_rstw_var_active('dissl')
705      ENDIF
[2528]706   END SUBROUTINE zdf_tke_init
[1239]707
708
[1531]709   SUBROUTINE tke_rst( kt, cdrw )
[9019]710      !!---------------------------------------------------------------------
711      !!                   ***  ROUTINE tke_rst  ***
712      !!                     
713      !! ** Purpose :   Read or write TKE file (en) in restart file
714      !!
715      !! ** Method  :   use of IOM library
716      !!                if the restart does not contain TKE, en is either
717      !!                set to rn_emin or recomputed
718      !!----------------------------------------------------------------------
719      USE zdf_oce , ONLY : en, avt_k, avm_k   ! ocean vertical physics
720      !!
721      INTEGER         , INTENT(in) ::   kt     ! ocean time-step
722      CHARACTER(len=*), INTENT(in) ::   cdrw   ! "READ"/"WRITE" flag
723      !
724      INTEGER ::   jit, jk              ! dummy loop indices
725      INTEGER ::   id1, id2, id3, id4   ! local integers
726      !!----------------------------------------------------------------------
727      !
728      IF( TRIM(cdrw) == 'READ' ) THEN        ! Read/initialise
729         !                                   ! ---------------
730         IF( ln_rstart ) THEN                   !* Read the restart file
731            id1 = iom_varid( numror, 'en'   , ldstop = .FALSE. )
732            id2 = iom_varid( numror, 'avt_k', ldstop = .FALSE. )
733            id3 = iom_varid( numror, 'avm_k', ldstop = .FALSE. )
734            id4 = iom_varid( numror, 'dissl', ldstop = .FALSE. )
735            !
736            IF( MIN( id1, id2, id3, id4 ) > 0 ) THEN      ! fields exist
[13286]737               CALL iom_get( numror, jpdom_auto, 'en'   , en   , ldxios = lrxios )
738               CALL iom_get( numror, jpdom_auto, 'avt_k', avt_k, ldxios = lrxios )
739               CALL iom_get( numror, jpdom_auto, 'avm_k', avm_k, ldxios = lrxios )
740               CALL iom_get( numror, jpdom_auto, 'dissl', dissl, ldxios = lrxios )
[9019]741            ELSE                                          ! start TKE from rest
[9169]742               IF(lwp) WRITE(numout,*)
[9190]743               IF(lwp) WRITE(numout,*) '   ==>>>   previous run without TKE scheme, set en to background values'
[9019]744               en   (:,:,:) = rn_emin * wmask(:,:,:)
745               dissl(:,:,:) = 1.e-12_wp
746               ! avt_k, avm_k already set to the background value in zdf_phy_init
747            ENDIF
748         ELSE                                   !* Start from rest
[9169]749            IF(lwp) WRITE(numout,*)
[9190]750            IF(lwp) WRITE(numout,*) '   ==>>>   start from rest: set en to the background value'
[9019]751            en   (:,:,:) = rn_emin * wmask(:,:,:)
752            dissl(:,:,:) = 1.e-12_wp
753            ! avt_k, avm_k already set to the background value in zdf_phy_init
754         ENDIF
755         !
756      ELSEIF( TRIM(cdrw) == 'WRITE' ) THEN   ! Create restart file
757         !                                   ! -------------------
[9169]758         IF(lwp) WRITE(numout,*) '---- tke_rst ----'
[9367]759         IF( lwxios ) CALL iom_swap(      cwxios_context          ) 
760         CALL iom_rstput( kt, nitrst, numrow, 'en'   , en   , ldxios = lwxios )
761         CALL iom_rstput( kt, nitrst, numrow, 'avt_k', avt_k, ldxios = lwxios )
762         CALL iom_rstput( kt, nitrst, numrow, 'avm_k', avm_k, ldxios = lwxios )
763         CALL iom_rstput( kt, nitrst, numrow, 'dissl', dissl, ldxios = lwxios )
764         IF( lwxios ) CALL iom_swap(      cxios_context          )
[9019]765         !
766      ENDIF
767      !
[1531]768   END SUBROUTINE tke_rst
[1239]769
770   !!======================================================================
[1531]771END MODULE zdftke
Note: See TracBrowser for help on using the repository browser.