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.
traatf_qco.F90 in NEMO/branches/2020/dev_r13898_Tiling_Cleanup_MPI3/src/OCE/TRA – NEMO

source: NEMO/branches/2020/dev_r13898_Tiling_Cleanup_MPI3/src/OCE/TRA/traatf_qco.F90 @ 13923

Last change on this file since 13923 was 13923, checked in by mocavero, 4 years ago

Final merge - conflicts solved

  • Property svn:keywords set to Id
File size: 19.2 KB
Line 
1MODULE traatfqco
2   !!======================================================================
3   !!                       ***  MODULE  traatfqco  ***
4   !! Ocean active tracers:  Asselin time filtering for temperature and salinity
5   !!======================================================================
6   !! History :  OPA  !  1991-11  (G. Madec)  Original code
7   !!            7.0  !  1993-03  (M. Guyon)  symetrical conditions
8   !!            8.0  !  1996-02  (G. Madec & M. Imbard)  opa release 8.0
9   !!             -   !  1996-04  (A. Weaver)  Euler forward step
10   !!            8.2  !  1999-02  (G. Madec, N. Grima)  semi-implicit pressure grad.
11   !!  NEMO      1.0  !  2002-08  (G. Madec)  F90: Free form and module
12   !!             -   !  2002-11  (C. Talandier, A-M Treguier) Open boundaries
13   !!             -   !  2005-04  (C. Deltel) Add Asselin trend in the ML budget
14   !!            2.0  !  2006-02  (L. Debreu, C. Mazauric) Agrif implementation
15   !!            3.0  !  2008-06  (G. Madec)  time stepping always done in trazdf
16   !!            3.1  !  2009-02  (G. Madec, R. Benshila)  re-introduce the vvl option
17   !!            3.3  !  2010-04  (M. Leclair, G. Madec)  semi-implicit hpg with asselin filter + modified LF-RA
18   !!             -   !  2010-05  (C. Ethe, G. Madec)  merge TRC-TRA
19   !!            4.1  !  2019-08  (A. Coward, D. Storkey) rename tranxt.F90 -> traatfLF.F90. Now only does time filtering.
20   !!----------------------------------------------------------------------
21
22   !!----------------------------------------------------------------------
23   !!   tra_atf       : time filtering on tracers
24   !!   tra_atf_fix   : time filtering on tracers : fixed    volume case
25   !!   tra_atf_vvl   : time filtering on tracers : variable volume case
26   !!----------------------------------------------------------------------
27   USE oce             ! ocean dynamics and tracers variables
28   USE dom_oce         ! ocean space and time domain variables
29   USE sbc_oce         ! surface boundary condition: ocean
30   USE sbcrnf          ! river runoffs
31   USE isf_oce         ! ice shelf melting
32   USE zdf_oce         ! ocean vertical mixing
33   USE domvvl          ! variable volume
34   USE trd_oce         ! trends: ocean variables
35   USE trdtra          ! trends manager: tracers
36   USE traqsr          ! penetrative solar radiation (needed for nksr)
37   USE phycst          ! physical constant
38   USE ldftra          ! lateral physics : tracers
39   USE ldfslp          ! lateral physics : slopes
40   USE bdy_oce  , ONLY : ln_bdy
41   USE bdytra          ! open boundary condition (bdy_tra routine)
42   !
43   USE in_out_manager  ! I/O manager
44   USE lbclnk          ! ocean lateral boundary conditions (or mpp link)
45   USE prtctl          ! Print control
46   USE timing          ! Timing
47#if defined key_agrif
48   USE agrif_oce_interp
49#endif
50
51   IMPLICIT NONE
52   PRIVATE
53
54   PUBLIC   tra_atf_qco       ! routine called by step.F90
55   PUBLIC   tra_atf_fix_lf    ! to be used in trcnxt !!st WARNING discrepancy here interpol is used by PISCES
56   PUBLIC   tra_atf_qco_lf    ! to be used in trcnxt !!st WARNING discrepancy here interpol is used by PISCES
57
58   !! * Substitutions
59#  include "do_loop_substitute.h90"
60#  include "domzgr_substitute.h90"
61   !!----------------------------------------------------------------------
62   !! NEMO/OCE 4.0 , NEMO Consortium (2018)
63   !! $Id$
64   !! Software governed by the CeCILL license (see ./LICENSE)
65   !!----------------------------------------------------------------------
66CONTAINS
67
68   SUBROUTINE tra_atf_qco( kt, Kbb, Kmm, Kaa, pts )
69      !!----------------------------------------------------------------------
70      !!                   ***  ROUTINE traatfLF  ***
71      !!
72      !! ** Purpose :   Apply the boundary condition on the after temperature
73      !!             and salinity fields and add the Asselin time filter on now fields.
74      !!
75      !! ** Method  :   At this stage of the computation, ta and sa are the
76      !!             after temperature and salinity as the time stepping has
77      !!             been performed in trazdf_imp or trazdf_exp module.
78      !!
79      !!              - Apply lateral boundary conditions on (ta,sa)
80      !!             at the local domain   boundaries through lbc_lnk call,
81      !!             at the one-way open boundaries (ln_bdy=T),
82      !!             at the AGRIF zoom   boundaries (lk_agrif=T)
83      !!
84      !!              - Update lateral boundary conditions on AGRIF children
85      !!             domains (lk_agrif=T)
86      !!
87      !! ** Action  : - ts(Kmm) time filtered
88      !!----------------------------------------------------------------------
89      INTEGER                                  , INTENT(in   ) :: kt             ! ocean time-step index
90      INTEGER                                  , INTENT(in   ) :: Kbb, Kmm, Kaa  ! time level indices
91      REAL(wp), DIMENSION(jpi,jpj,jpk,jpts,jpt), INTENT(inout) :: pts            ! active tracers
92      !!
93      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices
94      REAL(wp) ::   zfact            ! local scalars
95      REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) ::   ztrdt, ztrds
96      !!----------------------------------------------------------------------
97      !
98      IF( ln_timing )   CALL timing_start( 'tra_atf_qco')
99      !
100      IF( kt == nit000 ) THEN
101         IF(lwp) WRITE(numout,*)
102         IF(lwp) WRITE(numout,*) 'tra_atf_qco : apply Asselin time filter to "now" fields'
103         IF(lwp) WRITE(numout,*) '~~~~~~~'
104      ENDIF
105!!st  Update after tracer on domain lateral boundaries as been removed outside
106
107      ! trends computation initialisation
108      IF( l_trdtra )   THEN
109         ALLOCATE( ztrdt(jpi,jpj,jpk) , ztrds(jpi,jpj,jpk) )
110         ztrdt(:,:,jpk) = 0._wp
111         ztrds(:,:,jpk) = 0._wp
112         IF( ln_traldf_iso ) THEN              ! diagnose the "pure" Kz diffusive trend
113            CALL trd_tra( kt, Kmm, Kaa, 'TRA', jp_tem, jptra_zdfp, ztrdt )
114            CALL trd_tra( kt, Kmm, Kaa, 'TRA', jp_sal, jptra_zdfp, ztrds )
115         ENDIF
116         ! total trend for the non-time-filtered variables.
117         zfact = 1.0 / rn_Dt
118         ! G Nurser 23 Mar 2017. Recalculate trend as Delta(e3t*T)/e3tn; e3tn cancel from pts(Kmm) terms
119         DO jk = 1, jpkm1
120            ztrdt(:,:,jk) = ( pts(:,:,jk,jp_tem,Kaa) * (1._wp + r3t(:,:,Kaa) * tmask(:,:,jk))/(1._wp + r3t(:,:,Kmm) * tmask(:,:,jk))  &
121               &            - pts(:,:,jk,jp_tem,Kmm) ) * zfact
122            ztrds(:,:,jk) = ( pts(:,:,jk,jp_sal,Kaa) * (1._wp + r3t(:,:,Kaa) * tmask(:,:,jk))/(1._wp + r3t(:,:,Kmm) * tmask(:,:,jk))  &
123               &            - pts(:,:,jk,jp_sal,Kmm) ) * zfact
124         END DO
125         CALL trd_tra( kt, Kmm, Kaa, 'TRA', jp_tem, jptra_tot, ztrdt )
126         CALL trd_tra( kt, Kmm, Kaa, 'TRA', jp_sal, jptra_tot, ztrds )
127         IF( ln_linssh ) THEN       ! linear sea surface height only
128            ! Store now fields before applying the Asselin filter
129            ! in order to calculate Asselin filter trend later.
130            ztrdt(:,:,:) = pts(:,:,:,jp_tem,Kmm)
131            ztrds(:,:,:) = pts(:,:,:,jp_sal,Kmm)
132         ENDIF
133      ENDIF
134
135      IF( l_1st_euler ) THEN       ! Euler time-stepping
136         !
137         IF (l_trdtra .AND. .NOT. ln_linssh ) THEN   ! Zero Asselin filter contribution must be explicitly written out since for vvl
138            !                                        ! Asselin filter is output by tra_atf_vvl that is not called on this time step
139            ztrdt(:,:,:) = 0._wp
140            ztrds(:,:,:) = 0._wp
141            CALL trd_tra( kt, Kmm, Kaa, 'TRA', jp_tem, jptra_atf, ztrdt )
142            CALL trd_tra( kt, Kmm, Kaa, 'TRA', jp_sal, jptra_atf, ztrds )
143         END IF
144         !
145      ELSE                                            ! Leap-Frog + Asselin filter time stepping
146         !
147         IF ( ln_linssh ) THEN   ;   CALL tra_atf_fix_lf( kt, Kbb, Kmm, Kaa, nit000,        'TRA', pts, jpts )  ! linear free surface
148         ELSE                    ;   CALL tra_atf_qco_lf( kt, Kbb, Kmm, Kaa, nit000, rn_Dt, 'TRA', pts, sbc_tsc, sbc_tsc_b, jpts )  ! non-linear free surface
149         ENDIF
150         !
151#if defined key_mpi3
152         CALL lbc_lnk_nc_multi( 'traatfqco', pts(:,:,:,jp_tem,Kmm) , 'T', 1., pts(:,:,:,jp_sal,Kmm) , 'T', 1. )
153#else
154         CALL lbc_lnk_multi( 'traatfqco', pts(:,:,:,jp_tem,Kmm) , 'T', 1., pts(:,:,:,jp_sal,Kmm) , 'T', 1. )
155#endif
156
157      ENDIF
158      !
159      IF( l_trdtra .AND. ln_linssh ) THEN      ! trend of the Asselin filter (tb filtered - tb)/dt
160         DO jk = 1, jpkm1
161            ztrdt(:,:,jk) = ( pts(:,:,jk,jp_tem,Kmm) - ztrdt(:,:,jk) ) * r1_Dt
162            ztrds(:,:,jk) = ( pts(:,:,jk,jp_sal,Kmm) - ztrds(:,:,jk) ) * r1_Dt
163         END DO
164         CALL trd_tra( kt, Kmm, Kaa, 'TRA', jp_tem, jptra_atf, ztrdt )
165         CALL trd_tra( kt, Kmm, Kaa, 'TRA', jp_sal, jptra_atf, ztrds )
166      END IF
167      IF( l_trdtra )   DEALLOCATE( ztrdt , ztrds )
168      !
169      !                        ! control print
170      IF(sn_cfctl%l_prtctl)   CALL prt_ctl( tab3d_1=pts(:,:,:,jp_tem,Kmm), clinfo1=' nxt  - Tn: ', mask1=tmask,   &
171         &                                  tab3d_2=pts(:,:,:,jp_sal,Kmm), clinfo2=       ' Sn: ', mask2=tmask )
172      !
173      IF( ln_timing )   CALL timing_stop('tra_atf_qco')
174      !
175   END SUBROUTINE tra_atf_qco
176
177
178   SUBROUTINE tra_atf_fix_lf( kt, Kbb, Kmm, Kaa, kit000, cdtype, pt, kjpt )
179      !!----------------------------------------------------------------------
180      !!                   ***  ROUTINE tra_atf_fix  ***
181      !!
182      !! ** Purpose :   fixed volume: apply the Asselin time filter to the "now" field
183      !!
184      !! ** Method  : - Apply a Asselin time filter on now fields.
185      !!
186      !! ** Action  : - pt(Kmm) ready for the next time step
187      !!----------------------------------------------------------------------
188      INTEGER                                  , INTENT(in   ) ::  kt            ! ocean time-step index
189      INTEGER                                  , INTENT(in   ) ::  Kbb, Kmm, Kaa ! time level indices
190      INTEGER                                  , INTENT(in   ) ::  kit000        ! first time step index
191      CHARACTER(len=3)                         , INTENT(in   ) ::  cdtype        ! =TRA or TRC (tracer indicator)
192      INTEGER                                  , INTENT(in   ) ::  kjpt          ! number of tracers
193      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::  pt            ! tracer fields
194      !
195      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices
196      REAL(wp) ::   ztn, ztd         ! local scalars
197      !!----------------------------------------------------------------------
198      !
199      IF( kt == kit000 )  THEN
200         IF(lwp) WRITE(numout,*)
201         IF(lwp) WRITE(numout,*) 'tra_atf_fix_lf : time filtering', cdtype
202         IF(lwp) WRITE(numout,*) '~~~~~~~~~~~'
203      ENDIF
204      !
205      DO jn = 1, kjpt
206         !
207         DO_3D( nn_hls, nn_hls, nn_hls, nn_hls, 1, jpkm1 )
208            ztn = pt(ji,jj,jk,jn,Kmm)
209            ztd = pt(ji,jj,jk,jn,Kaa) - 2._wp * ztn + pt(ji,jj,jk,jn,Kbb)  ! time laplacian on tracers
210            !
211            pt(ji,jj,jk,jn,Kmm) = ztn + rn_atfp * ztd                      ! pt <-- filtered pt
212         END_3D
213         !
214      END DO
215      !
216   END SUBROUTINE tra_atf_fix_lf
217
218
219   SUBROUTINE tra_atf_qco_lf( kt, Kbb, Kmm, Kaa, kit000, p2dt, cdtype, pt, psbc_tc, psbc_tc_b, kjpt )
220      !!----------------------------------------------------------------------
221      !!                   ***  ROUTINE tra_atf_vvl  ***
222      !!
223      !! ** Purpose :   Time varying volume: apply the Asselin time filter
224      !!
225      !! ** Method  : - Apply a thickness weighted Asselin time filter on now fields.
226      !!             pt(Kmm)  = ( e3t_m*pt(Kmm) + rn_atfp*[ e3t_b*pt(Kbb) - 2 e3t_m*pt(Kmm) + e3t_a*pt(Kaa) ] )
227      !!                       /( e3t_m         + rn_atfp*[ e3t_b         - 2 e3t_m         + e3t_a    ] )
228      !!
229      !! ** Action  : - pt(Kmm) ready for the next time step
230      !!----------------------------------------------------------------------
231      INTEGER                                  , INTENT(in   ) ::  kt        ! ocean time-step index
232      INTEGER                                  , INTENT(in   ) ::  Kbb, Kmm, Kaa ! time level indices
233      INTEGER                                  , INTENT(in   ) ::  kit000    ! first time step index
234      REAL(wp)                                 , INTENT(in   ) ::  p2dt      ! time-step
235      CHARACTER(len=3)                         , INTENT(in   ) ::  cdtype    ! =TRA or TRC (tracer indicator)
236      INTEGER                                  , INTENT(in   ) ::  kjpt      ! number of tracers
237      REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt,jpt), INTENT(inout) ::  pt        ! tracer fields
238      REAL(wp), DIMENSION(jpi,jpj    ,kjpt)    , INTENT(in   ) ::  psbc_tc   ! surface tracer content
239      REAL(wp), DIMENSION(jpi,jpj    ,kjpt)    , INTENT(in   ) ::  psbc_tc_b ! before surface tracer content
240      !
241      LOGICAL  ::   ll_traqsr, ll_rnf, ll_isf   ! local logical
242      INTEGER  ::   ji, jj, jk, jn              ! dummy loop indices
243      REAL(wp) ::   zfact, zfact1, ztc_a , ztc_n , ztc_b , ztc_f , ztc_d    ! local scalar
244      REAL(wp) ::   zfact2, ze3t_b, ze3t_n, ze3t_a, ze3t_f                  !   -      -
245      REAL(wp), ALLOCATABLE, DIMENSION(:,:,:,:) ::   ztrd_atf
246      !!----------------------------------------------------------------------
247      !
248      IF( kt == kit000 )  THEN
249         IF(lwp) WRITE(numout,*)
250         IF(lwp) WRITE(numout,*) 'tra_atf_qco : time filtering', cdtype
251         IF(lwp) WRITE(numout,*) '~~~~~~~~~~~'
252      ENDIF
253      !
254      IF( cdtype == 'TRA' )  THEN
255         ll_traqsr  = ln_traqsr        ! active  tracers case  and  solar penetration
256         ll_rnf     = ln_rnf           ! active  tracers case  and  river runoffs
257         ll_isf     = ln_isf           ! active  tracers case  and  ice shelf melting
258      ELSE                          ! passive tracers case
259         ll_traqsr  = .FALSE.          ! NO solar penetration
260         ll_rnf     = .FALSE.          ! NO river runoffs ????          !!gm BUG ?
261         ll_isf     = .FALSE.          ! NO ice shelf melting/freezing  !!gm BUG ??
262      ENDIF
263      !
264      IF( ( l_trdtra .AND. cdtype == 'TRA' ) .OR. ( l_trdtrc .AND. cdtype == 'TRC' ) )   THEN
265         ALLOCATE( ztrd_atf(jpi,jpj,jpk,kjpt) )
266         ztrd_atf(:,:,:,:) = 0.0_wp
267      ENDIF
268      zfact = 1._wp / p2dt
269      zfact1 = rn_atfp * p2dt
270      zfact2 = zfact1 * r1_rho0
271      DO jn = 1, kjpt
272         DO_3D( nn_hls, nn_hls, nn_hls, nn_hls, 1, jpkm1 )
273            ze3t_b = e3t(ji,jj,jk,Kbb)
274            ze3t_n = e3t(ji,jj,jk,Kmm)
275            ze3t_a = e3t(ji,jj,jk,Kaa)
276            !                                         ! tracer content at Before, now and after
277            ztc_b  = pt(ji,jj,jk,jn,Kbb) * ze3t_b
278            ztc_n  = pt(ji,jj,jk,jn,Kmm) * ze3t_n
279            ztc_a  = pt(ji,jj,jk,jn,Kaa) * ze3t_a
280            !
281            ztc_d  = ztc_a  - 2. * ztc_n  + ztc_b
282            !
283            ztc_f  = ztc_n  + rn_atfp * ztc_d
284            !
285            ! Asselin correction on scale factors is done via ssh in r3t_f
286            ze3t_f = e3t_0(ji,jj,jk) * ( 1._wp + r3t_f(ji,jj) * tmask(ji,jj,jk) )
287
288            !
289            IF( jk == mikt(ji,jj) ) THEN           ! first level
290               ztc_f  = ztc_f  - zfact1 * ( psbc_tc(ji,jj,jn) - psbc_tc_b(ji,jj,jn) )
291            ENDIF
292            !
293            ! solar penetration (temperature only)
294            IF( ll_traqsr .AND. jn == jp_tem .AND. jk <= nksr )                            &
295               &     ztc_f  = ztc_f  - zfact1 * ( qsr_hc(ji,jj,jk) - qsr_hc_b(ji,jj,jk) )
296               !
297            !
298            IF( ll_rnf .AND. jk <= nk_rnf(ji,jj) )                                          &
299               &     ztc_f  = ztc_f  - zfact1 * ( rnf_tsc(ji,jj,jn) - rnf_tsc_b(ji,jj,jn) ) &
300               &                              * e3t(ji,jj,jk,Kmm) / h_rnf(ji,jj)
301
302            !
303            ! ice shelf
304            IF( ll_isf ) THEN
305               !
306               ! melt in the cavity
307               IF ( ln_isfcav_mlt ) THEN
308                  ! level fully include in the Losch_2008 ice shelf boundary layer
309                  IF ( jk >= misfkt_cav(ji,jj) .AND. jk < misfkb_cav(ji,jj) ) THEN
310                     ztc_f  = ztc_f  - zfact1 * ( risf_cav_tsc(ji,jj,jn) - risf_cav_tsc_b(ji,jj,jn) ) &
311                        &                     * e3t(ji,jj,jk,Kmm) / rhisf_tbl_cav(ji,jj)
312                  END IF
313                  ! level partially include in Losch_2008 ice shelf boundary layer
314                  IF ( jk == misfkb_cav(ji,jj) ) THEN
315                     ztc_f  = ztc_f  - zfact1 * ( risf_cav_tsc(ji,jj,jn) - risf_cav_tsc_b(ji,jj,jn) )  &
316                            &                 * e3t(ji,jj,jk,Kmm) / rhisf_tbl_cav(ji,jj)               &
317                            &                 * rfrac_tbl_cav(ji,jj)
318                  END IF
319               END IF
320               !
321               ! parametrised melt (cavity closed)
322               IF ( ln_isfpar_mlt ) THEN
323                  ! level fully include in the Losch_2008 ice shelf boundary layer
324                  IF ( jk >= misfkt_par(ji,jj) .AND. jk < misfkb_par(ji,jj) ) THEN
325                     ztc_f  = ztc_f  - zfact1 * ( risf_par_tsc(ji,jj,jn) - risf_par_tsc_b(ji,jj,jn) )  &
326                            &                 * e3t(ji,jj,jk,Kmm) / rhisf_tbl_par(ji,jj)
327                  END IF
328                  ! level partially include in Losch_2008 ice shelf boundary layer
329                  IF ( jk == misfkb_par(ji,jj) ) THEN
330                     ztc_f  = ztc_f  - zfact1 * ( risf_par_tsc(ji,jj,jn) - risf_par_tsc_b(ji,jj,jn) )  &
331                            &                 * e3t(ji,jj,jk,Kmm) / rhisf_tbl_par(ji,jj)               &
332                            &                 * rfrac_tbl_par(ji,jj)
333                  END IF
334               END IF
335               !
336               ! ice sheet coupling correction
337               IF ( ln_isfcpl ) THEN
338                  !
339                  ! at kt = nit000,  risfcpl_vol_n = 0 and risfcpl_vol_b = risfcpl_vol so contribution nul
340                  IF ( ln_rstart .AND. kt == nit000+1 ) THEN
341                     ztc_f  = ztc_f  + zfact1 * risfcpl_tsc(ji,jj,jk,jn) * r1_e1e2t(ji,jj)
342                     ! Shouldn't volume increment be spread according thanks to zscale  ?
343                  END IF
344                  !
345               END IF
346               !
347            END IF
348            !
349            ze3t_f = 1.e0 / ze3t_f
350            pt(ji,jj,jk,jn,Kmm) = ztc_f * ze3t_f    ! time filtered "now" field
351            !
352            IF( ( l_trdtra .and. cdtype == 'TRA' ) .OR. ( l_trdtrc .and. cdtype == 'TRC' ) ) THEN
353               ztrd_atf(ji,jj,jk,jn) = (ztc_f - ztc_n) * zfact/ze3t_n
354            ENDIF
355            !
356         END_3D
357         !
358      END DO
359      !
360      IF( ( l_trdtra .AND. cdtype == 'TRA' ) .OR. ( l_trdtrc .AND. cdtype == 'TRC' ) )   THEN
361         IF( l_trdtra .AND. cdtype == 'TRA' ) THEN
362            CALL trd_tra( kt, Kmm, Kaa, cdtype, jp_tem, jptra_atf, ztrd_atf(:,:,:,jp_tem) )
363            CALL trd_tra( kt, Kmm, Kaa, cdtype, jp_sal, jptra_atf, ztrd_atf(:,:,:,jp_sal) )
364         ENDIF
365         IF( l_trdtrc .AND. cdtype == 'TRC' ) THEN
366            DO jn = 1, kjpt
367               CALL trd_tra( kt, Kmm, Kaa, cdtype, jn, jptra_atf, ztrd_atf(:,:,:,jn) )
368            END DO
369         ENDIF
370         DEALLOCATE( ztrd_atf )
371      ENDIF
372      !
373   END SUBROUTINE tra_atf_qco_lf
374
375   !!======================================================================
376END MODULE traatfqco
Note: See TracBrowser for help on using the repository browser.