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.
trdtra.F90 in branches/2012/dev_r3309_LOCEAN12_Ediag/NEMOGCM/NEMO/OPA_SRC/TRD – NEMO

source: branches/2012/dev_r3309_LOCEAN12_Ediag/NEMOGCM/NEMO/OPA_SRC/TRD/trdtra.F90 @ 3327

Last change on this file since 3327 was 3327, checked in by gm, 12 years ago

Ediag branche: #927 correct a few problems and clean trdtra
  • Property svn:keywords set to Id
File size: 19.3 KB
Line 
1MODULE trdtra
2   !!======================================================================
3   !!                       ***  MODULE  trdtra  ***
4   !! Ocean diagnostics:  ocean tracers trends pre-processing
5   !!=====================================================================
6   !! History :  3.3  !  2010-06  (C. Ethe) creation for the TRA/TRC merge
7   !!            3.5  !  2012-02  (G. Madec) update the comments
8   !!----------------------------------------------------------------------
9
10   !!----------------------------------------------------------------------
11   !!   trd_tra       : pre-process the tracer trends
12   !!   trd_tra_adv   : transform a div(U.T) trend into a U.grad(T) trend
13   !!   trd_tra_mng   : tracer trend manager: dispatch to the diagnostic modules
14   !!   trd_tra_iom   : output 3D tracer trends using IOM
15   !!----------------------------------------------------------------------
16   USE oce            ! ocean dynamics and tracers variables
17   USE dom_oce        ! ocean domain
18   USE sbc_oce        ! surface boundary condition: ocean
19   USE zdf_oce        ! ocean vertical physics
20   USE trd_oce        ! trends: ocean variables
21   USE trdmod_trc     ! ocean passive mixed layer tracers trends
22   USE trdglo         ! trends: global domain averaged
23   USE trdpen         ! trends: Potential ENergy
24   USE trdmld         ! ocean active mixed layer tracers trends
25   USE ldftra_oce     ! ocean active tracers lateral physics
26   USE zdfddm         ! vertical physics: double diffusion
27   USE phycst         ! physical constants
28   USE in_out_manager ! I/O manager
29   USE iom            ! I/O manager library
30   USE lib_mpp        ! MPP library
31   USE wrk_nemo       ! Memory allocation
32
33   IMPLICIT NONE
34   PRIVATE
35
36   PUBLIC   trd_tra   ! called by all tra_... modules
37
38   REAL(wp) ::   r2dt   ! time-step, = 2 rdttra except at nit000 (=rdttra) if neuler=0
39
40   REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:) ::   trdtx, trdty, trdt   ! use to store the temperature trends
41
42   !! * Substitutions
43#  include "domzgr_substitute.h90"
44#  include "zdfddm_substitute.h90"
45#  include "vectopt_loop_substitute.h90"
46   !!----------------------------------------------------------------------
47   !! NEMO/OPA 3.3 , NEMO Consortium (2010)
48   !! $Id$
49   !! Software governed by the CeCILL licence     (NEMOGCM/NEMO_CeCILL.txt)
50   !!----------------------------------------------------------------------
51CONTAINS
52
53   INTEGER FUNCTION trd_tra_alloc()
54      !!---------------------------------------------------------------------
55      !!                  ***  FUNCTION trd_tra_alloc  ***
56      !!---------------------------------------------------------------------
57      ALLOCATE( trdtx(jpi,jpj,jpk) , trdty(jpi,jpj,jpk) , trdt(jpi,jpj,jpk) , STAT= trd_tra_alloc )
58      !
59      IF( lk_mpp             )   CALL mpp_sum ( trd_tra_alloc )
60      IF( trd_tra_alloc /= 0 )   CALL ctl_warn('trd_tra_alloc: failed to allocate arrays')
61   END FUNCTION trd_tra_alloc
62
63
64   SUBROUTINE trd_tra( kt, ctype, ktra, ktrd, ptrd, pun, ptra )
65      !!---------------------------------------------------------------------
66      !!                  ***  ROUTINE trd_tra  ***
67      !!
68      !! ** Purpose : pre-process tracer trends
69      !!
70      !! ** Method  : - mask the trend
71      !!              - advection (ptra present) converte the incoming flux (U.T)
72      !!              into trend (U.T => -U.grat(T)=div(U.T)-T.div(U)) through a
73      !!              call to trd_tra_adv
74      !!              - 'TRA' case : regroup T & S trends
75      !!              - send the trends to trd_tra_mng (trdmod_trc) for further processing
76      !!----------------------------------------------------------------------
77      INTEGER                         , INTENT(in)           ::   kt      ! time step
78      CHARACTER(len=3)                , INTENT(in)           ::   ctype   ! tracers trends type 'TRA'/'TRC'
79      INTEGER                         , INTENT(in)           ::   ktra    ! tracer index
80      INTEGER                         , INTENT(in)           ::   ktrd    ! tracer trend index
81      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in)           ::   ptrd    ! tracer trend  or flux
82      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in), OPTIONAL ::   pun     ! now velocity
83      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in), OPTIONAL ::   ptra    ! now tracer variable
84      !
85      INTEGER  ::   jk   ! loop indices
86      REAL(wp), POINTER, DIMENSION(:,:,:)  ::   zwt, zws, ztrdt, ztrds   ! 3D workspace
87      !!----------------------------------------------------------------------
88      !
89      CALL wrk_alloc( jpi, jpj, jpk, ztrds )
90      !     
91      IF( .NOT. ALLOCATED( trdtx ) ) THEN      ! allocate trdtra arrays
92         IF( trd_tra_alloc() /= 0 )   CALL ctl_stop( 'STOP', 'trd_tra : unable to allocate arrays' )
93      ENDIF
94
95      IF( ctype == 'TRA' .AND. ktra == jp_tem ) THEN   !==  Temperature trend  ==!
96         !
97         SELECT CASE( ktrd )
98         !                            ! advection: transform the advective flux into a trend
99         CASE( jptra_xad )   ;   CALL trd_tra_adv( ptrd, pun, ptra, 'X', trdtx ) 
100         CASE( jptra_yad )   ;   CALL trd_tra_adv( ptrd, pun, ptra, 'Y', trdty ) 
101         CASE( jptra_zad )   ;   CALL trd_tra_adv( ptrd, pun, ptra, 'Z', trdt  ) 
102         CASE( jptra_bbc,    &        ! qsr, bbc: on temperature only, send to trd_tra_mng
103            &  jptra_qsr )   ;   ztrds(:,:,:) = 0._wp
104                                 CALL trd_tra_mng( trdt, ztrds, ktrd, kt )
105         CASE DEFAULT                 ! other trends: masked trends
106            trdt(:,:,:) = ptrd(:,:,:) * tmask(:,:,:)              ! mask & store
107         END SELECT
108         !
109      ENDIF
110
111      IF( ctype == 'TRA' .AND. ktra == jp_sal ) THEN      !==  Salinity trends  ==!
112         !
113         SELECT CASE( ktrd )
114         !                            ! advection: transform the advective flux into a trend
115         !                            !            and send T & S trends to trd_tra_mng
116         CASE( jptra_xad  )   ;   CALL trd_tra_adv( ptrd , pun  , ptra, 'X'  , ztrds ) 
117                                  CALL trd_tra_mng( trdtx, ztrds, ktrd, kt   )
118         CASE( jptra_yad  )   ;   CALL trd_tra_adv( ptrd , pun  , ptra, 'Y'  , ztrds ) 
119                              ;   CALL trd_tra_mng( trdty, ztrds, ktrd, kt   )
120         CASE( jptra_zad  )   ;   CALL trd_tra_adv( ptrd , pun  , ptra, 'Z'  , ztrds ) 
121                                  CALL trd_tra_mng( trdt , ztrds, ktrd, kt   )
122         CASE( jptra_zdfp )           ! diagnose the "PURE" Kz trend (here: just before the swap)
123            !                         ! iso-neutral diffusion case otherwise jptra_zdf is "PURE"
124            CALL wrk_alloc( jpi, jpj, jpk, zwt, zws, ztrdt )
125            !
126            zwt(:,:, 1 ) = 0._wp   ;   zws(:,:, 1 ) = 0._wp            ! vertical diffusive fluxes
127            zwt(:,:,jpk) = 0._wp   ;   zws(:,:,jpk) = 0._wp
128            DO jk = 2, jpk
129               zwt(:,:,jk) =   avt(:,:,jk) * ( tsa(:,:,jk-1,jp_tem) - tsa(:,:,jk,jp_tem) ) / fse3w(:,:,jk) * tmask(:,:,jk)
130               zws(:,:,jk) = fsavs(:,:,jk) * ( tsa(:,:,jk-1,jp_sal) - tsa(:,:,jk,jp_sal) ) / fse3w(:,:,jk) * tmask(:,:,jk)
131            END DO
132            !
133            ztrdt(:,:,jpk) = 0._wp   ;   ztrds(:,:,jpk) = 0._wp
134            DO jk = 1, jpkm1
135               ztrdt(:,:,jk) = ( zwt(:,:,jk) - zwt(:,:,jk+1) ) / fse3t(:,:,jk)
136               ztrds(:,:,jk) = ( zws(:,:,jk) - zws(:,:,jk+1) ) / fse3t(:,:,jk) 
137            END DO
138            CALL trd_tra_mng( ztrdt, ztrds, jptra_zdfp, kt ) 
139            !
140            CALL wrk_dealloc( jpi, jpj, jpk, zwt, zws, ztrdt )
141            !
142         CASE DEFAULT                 ! other trends: mask and send T & S trends to trd_tra_mng
143            ztrds(:,:,:) = ptrd(:,:,:) * tmask(:,:,:)
144            CALL trd_tra_mng( trdt, ztrds, ktrd, kt ) 
145         END SELECT
146      ENDIF
147
148      IF( ctype == 'TRC' ) THEN                           !==  passive tracer trend  ==!
149         !
150         SELECT CASE( ktrd )
151         !                            ! advection: transform the advective flux into a masked trend
152         CASE( jptra_xad )   ;   CALL trd_tra_adv( ptrd , pun , ptra, 'X', ztrds ) 
153         CASE( jptra_yad )   ;   CALL trd_tra_adv( ptrd , pun , ptra, 'Y', ztrds ) 
154         CASE( jptra_zad )   ;   CALL trd_tra_adv( ptrd , pun , ptra, 'Z', ztrds ) 
155         CASE DEFAULT                 ! other trends: just masked
156                                 ztrds(:,:,:) = ptrd(:,:,:) * tmask(:,:,:)
157         END SELECT
158         !                            ! send trend to trd_mod_trc
159         CALL trd_mod_trc( ztrds, ktra, ktrd, kt ) 
160         !
161      ENDIF
162      !
163      CALL wrk_dealloc( jpi, jpj, jpk, ztrds )
164      !
165   END SUBROUTINE trd_tra
166
167
168   SUBROUTINE trd_tra_adv( pf, pun, ptn, cdir, ptrd )
169      !!---------------------------------------------------------------------
170      !!                  ***  ROUTINE trd_tra_adv  ***
171      !!
172      !! ** Purpose :   transformed a advective flux into a masked advective trends
173      !!
174      !! ** Method  :   use the following transformation: -div(U.T) = - U grad(T) + T.div(U)
175      !!       i-advective trends = -un. di-1[T] = -( di-1[fi] - tn di-1[un] )
176      !!       j-advective trends = -un. di-1[T] = -( dj-1[fi] - tn dj-1[un] )
177      !!       k-advective trends = -un. di+1[T] = -( dk+1[fi] - tn dk+1[un] )
178      !!                where fi is the incoming advective flux.
179      !!----------------------------------------------------------------------
180      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   pf      ! advective flux in one direction
181      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   pun     ! now velocity   in one direction
182      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   ptn     ! now or before tracer
183      CHARACTER(len=1)                , INTENT(in   ) ::   cdir    ! X/Y/Z direction
184      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(  out) ::   ptrd    ! advective trend in one direction
185      !
186      INTEGER  ::   ji, jj, jk   ! dummy loop indices
187      INTEGER  ::   ii, ij, ik   ! index shift as function of the direction
188      !!----------------------------------------------------------------------
189      !
190      SELECT CASE( cdir )      ! shift depending on the direction
191      CASE( 'X' )   ;   ii = 1   ;   ij = 0   ;   ik = 0      ! i-trend
192      CASE( 'Y' )   ;   ii = 0   ;   ij = 1   ;   ik = 0      ! j-trend
193      CASE( 'Z' )   ;   ii = 0   ;   ij = 0   ;   ik =-1      ! k-trend
194      END SELECT
195      !
196      !                        ! set to zero uncomputed values
197      ptrd(jpi,:,:) = 0._wp   ;   ptrd(1,:,:) = 0._wp
198      ptrd(:,jpj,:) = 0._wp   ;   ptrd(:,1,:) = 0._wp
199      ptrd(:,:,jpk) = 0._wp
200      !
201      DO jk = 1, jpkm1         ! advective trend
202         DO jj = 2, jpjm1
203            DO ji = fs_2, fs_jpim1   ! vector opt.
204               ptrd(ji,jj,jk) = - (     pf (ji,jj,jk) - pf (ji-ii,jj-ij,jk-ik)                        &
205                 &                  - ( pun(ji,jj,jk) - pun(ji-ii,jj-ij,jk-ik) ) * ptn(ji,jj,jk)  )   &
206                 &              / ( e1t(ji,jj) * e2t(ji,jj) * fse3t(ji,jj,jk) )  * tmask(ji,jj,jk)
207            END DO
208         END DO
209      END DO
210      !
211   END SUBROUTINE trd_tra_adv
212
213
214   SUBROUTINE trd_tra_mng( ptrdx, ptrdy, ktrd, kt )
215      !!---------------------------------------------------------------------
216      !!                  ***  ROUTINE trd_tra_mng  ***
217      !!
218      !! ** Purpose :   Dispatch all trends computation, e.g. 3D output, integral
219      !!                constraints, barotropic vorticity, kinetic enrgy,
220      !!                potential energy, and/or mixed layer budget.
221      !!----------------------------------------------------------------------
222      REAL(wp), DIMENSION(:,:,:), INTENT(inout) ::   ptrdx   ! Temperature or U trend
223      REAL(wp), DIMENSION(:,:,:), INTENT(inout) ::   ptrdy   ! Salinity    or V trend
224      INTEGER                   , INTENT(in   ) ::   ktrd    ! tracer trend index
225      INTEGER                   , INTENT(in   ) ::   kt      ! time step
226      !!----------------------------------------------------------------------
227
228      IF( neuler == 0 .AND. kt == nit000    ) THEN   ;   r2dt =      rdt      ! = rdtra (restart with Euler time stepping)
229      ELSEIF(               kt <= nit000 + 1) THEN   ;   r2dt = 2. * rdt      ! = 2 rdttra (leapfrog)
230      ENDIF
231
232      !                   ! 3D output of tracers trends using IOM interface
233      IF( ln_tra_trd )   CALL trd_tra_iom ( ptrdx, ptrdy, ktrd, kt )
234
235      !                   ! Integral Constraints Properties for tracers trends                                       !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
236      IF( ln_glo_trd )   CALL trd_glo( ptrdx, ptrdy, ktrd, 'TRA', kt )
237
238      !                   ! Potential ENergy trends
239      IF( ln_glo_trd )   CALL trd_pen( ptrdx, ptrdy, ktrd, kt, r2dt )
240
241      !                   ! Mixed layer trends for active tracers
242      IF( ln_tra_mld )   THEN   
243         !-----------------------------------------------------------------------------------------------
244         ! W.A.R.N.I.N.G :
245         ! jptra_ldf : called by traldf.F90
246         !                 at this stage we store:
247         !                  - the lateral geopotential diffusion (here, lateral = horizontal)
248         !                  - and the iso-neutral diffusion if activated
249         ! jptra_zdf : called by trazdf.F90
250         !                 * in case of iso-neutral diffusion we store the vertical diffusion component in the
251         !                   lateral trend including the K_z contrib, which will be removed later (see trd_mld)
252         !-----------------------------------------------------------------------------------------------
253
254         SELECT CASE ( ktrd )
255         CASE ( jptra_xad )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_xad, '3D' )   ! zonal    advection
256         CASE ( jptra_yad )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_yad, '3D' )   ! merid.   advection
257         CASE ( jptra_zad )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_zad, '3D' )   ! vertical advection
258         CASE ( jptra_ldf )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_ldf, '3D' )   ! lateral  diffusion
259         CASE ( jptra_bbl )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_bbl, '3D' )   ! bottom boundary layer
260         CASE ( jptra_zdf )
261            IF( ln_traldf_iso ) THEN ; CALL trd_mld_zint( ptrdx, ptrdy, jpmld_ldf, '3D' )   ! lateral  diffusion (K_z)
262            ELSE                   ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_zdf, '3D' )   ! vertical diffusion (K_z)
263            ENDIF
264         CASE ( jptra_dmp )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_dmp, '3D' )   ! internal 3D restoring (tradmp)
265         CASE ( jptra_qsr )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_for, '3D' )   ! air-sea : penetrative sol radiat
266         CASE ( jptra_nsr )        ;   ptrdx(:,:,2:jpk) = 0._wp   ;   ptrdy(:,:,2:jpk) = 0._wp
267                                   ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_for, '2D' )   ! air-sea : non penetr sol radiation
268         CASE ( jptra_bbc )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_bbc, '3D' )   ! bottom bound cond (geoth flux)
269         CASE ( jptra_npc )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_npc, '3D' )   ! non penetr convect adjustment
270         CASE ( jptra_atf )        ;   CALL trd_mld_zint( ptrdx, ptrdy, jpmld_atf, '3D' )   ! asselin time filter (last trend)
271                                   !
272                                       CALL trd_mld( kt )                                   ! trends: Mixed-layer (output)
273         END SELECT
274         !
275      ENDIF
276      !
277   END SUBROUTINE trd_tra_mng
278
279
280   SUBROUTINE trd_tra_iom( ptrdx, ptrdy, ktrd, kt )
281      !!---------------------------------------------------------------------
282      !!                  ***  ROUTINE trd_tra_iom  ***
283      !!
284      !! ** Purpose :   output 3D tracer trends using IOM
285      !!----------------------------------------------------------------------
286      REAL(wp), DIMENSION(:,:,:), INTENT(inout) ::   ptrdx   ! Temperature or U trend
287      REAL(wp), DIMENSION(:,:,:), INTENT(inout) ::   ptrdy   ! Salinity    or V trend
288      INTEGER                   , INTENT(in   ) ::   ktrd    ! tracer trend index
289      INTEGER                   , INTENT(in   ) ::   kt      ! time step
290      !!
291      INTEGER ::   ji, jj, jk   ! dummy loop indices
292      INTEGER ::   ikbu, ikbv   ! local integers
293      REAL(wp), POINTER, DIMENSION(:,:)   ::   z2dx, z2dy   ! 2D workspace
294      !!----------------------------------------------------------------------
295      !
296      CALL wrk_alloc( jpi, jpj, z2dx, z2dy )
297      !
298!!gm Rq: mask the trends already masked in trd_tra, but lbc_lnk should probably be added
299      !
300      SELECT CASE( ktrd )
301      CASE( jptra_xad  )   ;   CALL iom_put( "ttrd_xad" , ptrdx )        ! x- horizontal advection
302                               CALL iom_put( "strd_xad" , ptrdy )
303      CASE( jptra_yad  )   ;   CALL iom_put( "ttrd_yad" , ptrdx )        ! y- horizontal advection
304                               CALL iom_put( "strd_yad" , ptrdy )
305      CASE( jptra_zad  )   ;   CALL iom_put( "ttrd_zad" , ptrdx )        ! z- vertical   advection
306                               CALL iom_put( "strd_zad" , ptrdy )
307                               IF( .NOT.lk_vvl ) THEN                   ! cst volume : adv flux through z=0 surface
308                                  z2dx(:,:) = wn(:,:,1) * tsn(:,:,1,jp_tem) / fse3t(:,:,1)
309                                  z2dy(:,:) = wn(:,:,1) * tsn(:,:,1,jp_sal) / fse3t(:,:,1)
310                                  CALL iom_put( "ttrd_sad", z2dx )
311                                  CALL iom_put( "strd_sad", z2dy )
312                               ENDIF
313      CASE( jptra_ldf  )   ;   CALL iom_put( "ttrd_ldf" , ptrdx )        ! lateral diffusion
314                               CALL iom_put( "strd_ldf" , ptrdy )
315      CASE( jptra_zdf  )   ;   CALL iom_put( "ttrd_zdf" , ptrdx )        ! vertical diffusion (including Kz contribution)
316                               CALL iom_put( "strd_zdf" , ptrdy )
317      CASE( jptra_zdfp )   ;   CALL iom_put( "ttrd_zdfp", ptrdx )        ! PURE vertical diffusion (no isoneutral contribution)
318                               CALL iom_put( "strd_zdfp", ptrdy )
319      CASE( jptra_dmp  )   ;   CALL iom_put( "ttrd_dmp" , ptrdx )        ! internal restoring (damping)
320                               CALL iom_put( "strd_dmp" , ptrdy )
321      CASE( jptra_bbl  )   ;   CALL iom_put( "ttrd_bbl" , ptrdx )        ! bottom boundary layer
322                               CALL iom_put( "strd_bbl" , ptrdy )
323      CASE( jptra_npc  )   ;   CALL iom_put( "ttrd_npc" , ptrdx )        ! static instability mixing
324                               CALL iom_put( "strd_npc" , ptrdy )
325      CASE( jptra_nsr  )   ;   CALL iom_put( "ttrd_qns" , ptrdx )        ! surface forcing + runoff (ln_rnf=T)
326                               CALL iom_put( "strd_cdt" , ptrdy )
327      CASE( jptra_qsr  )   ;   CALL iom_put( "ttrd_qsr" , ptrdx )        ! penetrative solar radiat. (only on temperature)
328      CASE( jptra_bbc  )   ;   CALL iom_put( "ttrd_bbc" , ptrdx )        ! geothermal heating   (only on temperature)
329      CASE( jptra_atf  )   ;   CALL iom_put( "ttrd_atf" , ptrdx )        ! asselin time Filter
330                               CALL iom_put( "strd_atf" , ptrdy )
331      END SELECT
332      !
333      CALL wrk_dealloc( jpi, jpj, z2dx, z2dy )
334      !
335   END SUBROUTINE trd_tra_iom
336
337   !!======================================================================
338END MODULE trdtra
Note: See TracBrowser for help on using the repository browser.