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.
step.F90 in NEMO/branches/2021/dev_r14116_HPC-10_mcastril_Mixed_Precision_implementation/src/OCE – NEMO

source: NEMO/branches/2021/dev_r14116_HPC-10_mcastril_Mixed_Precision_implementation/src/OCE/step.F90 @ 15606

Last change on this file since 15606 was 15540, checked in by sparonuz, 3 years ago

Mixed precision version, tested up to 30 years on ORCA2.

  • Property svn:keywords set to Id
File size: 25.9 KB
RevLine 
[3]1MODULE step
2   !!======================================================================
3   !!                       ***  MODULE step  ***
[6140]4   !! Time-stepping   : manager of the ocean, tracer and ice time stepping
[3]5   !!======================================================================
[1438]6   !! History :  OPA  !  1991-03  (G. Madec)  Original code
[1492]7   !!             -   !  1991-11  (G. Madec)
8   !!             -   !  1992-06  (M. Imbard)  add a first output record
9   !!             -   !  1996-04  (G. Madec)  introduction of dynspg
10   !!             -   !  1996-04  (M.A. Foujols)  introduction of passive tracer
[1438]11   !!            8.0  !  1997-06  (G. Madec)  new architecture of call
12   !!            8.2  !  1997-06  (G. Madec, M. Imbard, G. Roullet)  free surface
13   !!             -   !  1999-02  (G. Madec, N. Grima)  hpg implicit
14   !!             -   !  2000-07  (J-M Molines, M. Imbard)  Open Bondary Conditions
15   !!   NEMO     1.0  !  2002-06  (G. Madec)  free form, suppress macro-tasking
16   !!             -   !  2004-08  (C. Talandier) New trends organization
[1492]17   !!             -   !  2005-01  (C. Ethe) Add the KPP closure scheme
18   !!             -   !  2005-11  (G. Madec)  Reorganisation of tra and dyn calls
19   !!             -   !  2006-01  (L. Debreu, C. Mazauric)  Agrif implementation
20   !!             -   !  2006-07  (S. Masson)  restart using iom
[1438]21   !!            3.2  !  2009-02  (G. Madec, R. Benshila)  reintroduicing z*-coordinate
[1533]22   !!             -   !  2009-06  (S. Masson, G. Madec)  TKE restart compatible with key_cpl
[2528]23   !!            3.3  !  2010-05  (K. Mogensen, A. Weaver, M. Martin, D. Lea) Assimilation interface
24   !!             -   !  2010-10  (C. Ethe, G. Madec) reorganisation of initialisation phase + merge TRC-TRA
[3294]25   !!            3.4  !  2011-04  (G. Madec, C. Ethe) Merge of dtatem and dtasal
[5836]26   !!            3.6  !  2012-07  (J. Simeon, G. Madec. C. Ethe)  Online coarsening of outputs
27   !!            3.6  !  2014-04  (F. Roquet, G. Madec) New equations of state
[7646]28   !!            3.6  !  2014-10  (E. Clementi, P. Oddo) Add Qiao vertical mixing in case of waves
[14072]29   !!            3.7  !  2014-10  (G. Madec)  LDF simplication
[5836]30   !!             -   !  2014-12  (G. Madec) remove KPP scheme
[9019]31   !!             -   !  2015-11  (J. Chanut) free surface simplification (remove filtered free surface)
32   !!            4.0  !  2017-05  (G. Madec)  introduction of the vertical physics manager (zdfphy)
[12377]33   !!            4.1  !  2019-08  (A. Coward, D. Storkey) rewrite in preparation for new timestepping scheme
[3]34   !!----------------------------------------------------------------------
[14644]35
[14200]36#if defined key_qco   ||   defined key_linssh
[888]37   !!----------------------------------------------------------------------
[14200]38   !!   'key_qco'      EMPTY MODULE      Quasi-Eulerian vertical coordinate
39   !!                                OR
40   !!   'key_linssh    EMPTY MODULE       Fixed in time vertical coordinate
[13237]41   !!----------------------------------------------------------------------
42#else
43   !!----------------------------------------------------------------------
[14644]44   !!   stp             : OCE system time-stepping
[3]45   !!----------------------------------------------------------------------
[3764]46   USE step_oce         ! time stepping definition modules
[389]47
[3]48   IMPLICIT NONE
49   PRIVATE
50
[5836]51   PUBLIC   stp   ! called by nemogcm.F90
[3]52
[14644]53   !                                          !**  time level indices  **!
54   INTEGER, PUBLIC ::   Nbb, Nnn, Naa, Nrhs   !: used by nemo_init
[12377]55
[13982]56   !! * Substitutions
57#  include "do_loop_substitute.h90"
[14219]58#  include "single_precision_substitute.h90"
[12377]59   !!----------------------------------------------------------------------
[9598]60   !! NEMO/OCE 4.0 , NEMO Consortium (2018)
[888]61   !! $Id$
[10068]62   !! Software governed by the CeCILL license (see ./LICENSE)
[3]63   !!----------------------------------------------------------------------
64CONTAINS
65
[508]66#if defined key_agrif
[5656]67   RECURSIVE SUBROUTINE stp( )
[1438]68      INTEGER             ::   kstp   ! ocean time-step index
[508]69#else
[467]70   SUBROUTINE stp( kstp )
[1438]71      INTEGER, INTENT(in) ::   kstp   ! ocean time-step index
[467]72#endif
73      !!----------------------------------------------------------------------
[3]74      !!                     ***  ROUTINE stp  ***
[3764]75      !!
[14644]76      !! ** Purpose : - Time stepping of OCE  (momentum and active tracer eqs.)
[9570]77      !!              - Time stepping of SI3 (dynamic and thermodynamic eqs.)
[9019]78      !!              - Time stepping of TRC  (passive tracer eqs.)
[3764]79      !!
80      !! ** Method  : -1- Update forcings and data
81      !!              -2- Update ocean physics
82      !!              -3- Compute the t and s trends
83      !!              -4- Update t and s
[3]84      !!              -5- Compute the momentum trends
85      !!              -6- Update the horizontal velocity
[5836]86      !!              -7- Compute the diagnostics variables (rd,N2, hdiv,w)
[3]87      !!              -8- Outputs and diagnostics
88      !!----------------------------------------------------------------------
[13982]89      INTEGER ::   ji, jj, jk, jtile   ! dummy loop indice
[3]90      !! ---------------------------------------------------------------------
[392]91#if defined key_agrif
[12933]92      IF( nstop > 0 ) RETURN   ! avoid to go further if an error was detected during previous time step (child grid)
[389]93      kstp = nit000 + Agrif_Nb_Step()
[12377]94      Kbb_a = Nbb; Kmm_a = Nnn; Krhs_a = Nrhs   ! agrif_oce module copies of time level indices
[5836]95      IF( lk_agrif_debug ) THEN
96         IF( Agrif_Root() .and. lwp)   WRITE(*,*) '---'
97         IF(lwp)   WRITE(*,*) 'Grid Number', Agrif_Fixed(),' time step ', kstp, 'int tstep', Agrif_NbStepint()
[5656]98      ENDIF
[5836]99      IF( kstp == nit000 + 1 )   lk_agrif_fstep = .FALSE.
[14644]100# if defined key_xios
[5407]101      IF( Agrif_Nbstepint() == 0 )   CALL iom_swap( cxios_context )
[3764]102# endif
103#endif
[9124]104      !
105      IF( ln_timing )   CALL timing_start('stp')
106      !
[5836]107      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[12489]108      ! model timestep
109      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
110      !
[14644]111      IF( l_1st_euler ) THEN     ! start or restart with Euler 1st time-step
112         rDt   = rn_Dt   
[12489]113         r1_Dt = 1._wp / rDt
114      ENDIF
115      !
116      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[14072]117      ! update I/O and calendar
[5836]118      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[14644]119      !
[5836]120      IF( kstp == nit000 ) THEN                       ! initialize IOM context (must be done after nemo_init for AGRIF+XIOS+OASIS)
[12933]121                             CALL iom_init( cxios_context, ld_closedef=.FALSE. )   ! for model grid (including possible AGRIF zoom)
[12377]122         IF( lk_diamlr   )   CALL dia_mlr_iom_init    ! with additional setup for multiple-linear-regression analysis
123                             CALL iom_init_closedef
[5836]124         IF( ln_crs      )   CALL iom_init( TRIM(cxios_context)//"_crs" )  ! for coarse grid
[4152]125      ENDIF
[14644]126      IF( kstp == nitrst .AND. lwxios ) THEN
127                             CALL iom_swap(                     cw_ocerst_cxt )
128                             CALL iom_init_closedef(            cw_ocerst_cxt )
129                             CALL iom_setkt( kstp - nit000 + 1, cw_ocerst_cxt )
[13970]130#if defined key_top
[14644]131                             CALL iom_swap(                     cw_toprst_cxt )
132                             CALL iom_init_closedef(            cw_toprst_cxt )
133                             CALL iom_setkt( kstp - nit000 + 1, cw_toprst_cxt )
[13970]134#endif
135      ENDIF
[14644]136      IF( kstp + nn_fsbc - 1 == nitrst .AND. lwxios ) THEN
[13970]137#if defined key_si3
[14644]138                             CALL iom_swap(                     cw_icerst_cxt )
139                             CALL iom_init_closedef(            cw_icerst_cxt )
140                             CALL iom_setkt( kstp - nit000 + 1, cw_icerst_cxt )
141#endif
142         IF( ln_abl      ) THEN
143                             CALL iom_swap(                     cw_ablrst_cxt )
144                             CALL iom_init_closedef(            cw_ablrst_cxt )
145                             CALL iom_setkt( kstp - nit000 + 1, cw_ablrst_cxt )
146         ENDIF
[13970]147      ENDIF
[1725]148      IF( kstp /= nit000 )   CALL day( kstp )         ! Calendar (day was already called at nit000 in day_init)
[5836]149                             CALL iom_setkt( kstp - nit000 + 1,      cxios_context          )   ! tell IOM we are at time step kstp
150      IF( ln_crs         )   CALL iom_setkt( kstp - nit000 + 1, TRIM(cxios_context)//"_crs" )   ! tell IOM we are at time step kstp
[3]151
152      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[12377]153      ! Update external forcing (tides, open boundaries, ice shelf interaction and surface boundary condition (including sea-ice)
[3]154      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[12377]155      IF( ln_tide    )   CALL tide_update( kstp )                     ! update tide potential
[14072]156      IF( ln_apr_dyn )   CALL sbc_apr ( kstp )                        ! atmospheric pressure (NB: call before bdy_dta which needs ssh_ib)
[12377]157      IF( ln_bdy     )   CALL bdy_dta ( kstp, Nnn )                   ! update dynamic & tracer data at open boundaries
158      IF( ln_isf     )   CALL isf_stp ( kstp, Nnn )
159                         CALL sbc     ( kstp, Nbb, Nnn )              ! Sea Boundary Condition (including sea-ice)
[5836]160
[3]161      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[5329]162      ! Update stochastic parameters and random T/S fluctuations
163      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[14644]164      IF( ln_sto_eos )   CALL sto_par( kstp )                         ! Stochastic parameters
165      IF( ln_sto_eos )   CALL sto_pts( ts(:,:,:,:,Nnn)  )             ! Random T/S fluctuations
[5329]166
167      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[9490]168      ! Ocean physics update
[3]169      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[4990]170      !  THERMODYNAMICS
[15540]171                         CALL eos_rab( ts(:,:,:,:,Nbb), rab_b, Nnn )       ! before local thermal/haline expension ratio at T-points
172                         CALL eos_rab( ts(:,:,:,:,Nnn), rab_n, Nnn )       ! now    local thermal/haline expension ratio at T-points
[15066]173                         CALL bn2    ( ts(:,:,:,:,Nbb), rab_b, rn2b, Nnn ) ! before Brunt-Vaisala frequency
174                         CALL bn2    ( ts(:,:,:,:,Nnn), rab_n, rn2, Nnn  ) ! now    Brunt-Vaisala frequency
[5836]175
[3764]176      !  VERTICAL PHYSICS
[14986]177      ! lbc_lnk needed for zdf_sh2 when using nn_hls = 2, moved here to allow tiling in zdf_phy
178      IF( nn_hls == 2 .AND. l_zdfsh2 ) CALL lbc_lnk( 'stp', avm_k, 'W', 1.0_wp )
179
180      IF( ln_tile ) CALL dom_tile_start         ! [tiling] ZDF tiling loop
181      DO jtile = 1, nijtile
182         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = jtile )
183
[12377]184                         CALL zdf_phy( kstp, Nbb, Nnn, Nrhs )   ! vertical physics update (top/bot drag, avt, avs, avm + MLD)
[14986]185      END DO
186      IF( ln_tile ) CALL dom_tile_stop
[5836]187
[3764]188      !  LATERAL  PHYSICS
[1492]189      !
[5836]190      IF( l_ldfslp ) THEN                             ! slope of lateral mixing
[15540]191                         CALL eos( ts(:,:,:,:,Nbb), rhd, gdept_0(:,:,:) )               ! before in situ density
[5836]192
[14986]193      IF( ln_zps .AND. .NOT. ln_isfcav)                                    &
[12377]194            &            CALL zps_hde    ( kstp, Nnn, jpts, ts(:,:,:,:,Nbb), gtsu, gtsv,  &  ! Partial steps: before horizontal gradient
195            &                                          rhd, gru , grv    )       ! of t, s, rd at the last ocean level
[5836]196
[14986]197      IF( ln_zps .AND.       ln_isfcav)                                                &
[12377]198            &            CALL zps_hde_isf( kstp, Nnn, jpts, ts(:,:,:,:,Nbb), gtsu, gtsv, gtui, gtvi,  &  ! Partial steps for top cell (ISF)
199            &                                          rhd, gru , grv , grui, grvi   )       ! of t, s, rd at the first ocean level
[14072]200         IF( ln_traldf_triad ) THEN
[12377]201                         CALL ldf_slp_triad( kstp, Nbb, Nnn )             ! before slope for triad operator
[14072]202         ELSE
[12377]203                         CALL ldf_slp     ( kstp, rhd, rn2b, Nbb, Nnn )   ! before slope for standard operator
[2528]204         ENDIF
[1481]205      ENDIF
[12377]206      !                                                                        ! eddy diffusivity coeff.
207      IF( l_ldftra_time .OR. l_ldfeiv_time )   CALL ldf_tra( kstp, Nbb, Nnn )  !       and/or eiv coeff.
[14072]208      IF( l_ldfdyn_time                    )   CALL ldf_dyn( kstp, Nbb )       ! eddy viscosity coeff.
[12377]209
[4369]210      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[5930]211      !  Ocean dynamics : hdiv, ssh, e3, u, v, w
[4369]212      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[5930]213
[14644]214                         CALL ssh_nxt       ( kstp, Nbb, Nnn, ssh, Naa )   ! after ssh (includes call to div_hor)
215      IF( .NOT.ln_linssh )   &
216                       & CALL dom_vvl_sf_nxt( kstp, Nbb, Nnn,      Naa )   ! after vertical scale factors
217                         CALL wzv           ( kstp, Nbb, Nnn, Naa, ww  )   ! now cross-level velocity
218      IF( ln_zad_Aimp )  CALL wAimp         ( kstp,      Nnn           )   ! Adaptive-implicit vertical advection partitioning
219                         CALL eos    ( ts(:,:,:,:,Nnn), rhd, rhop, gdept(:,:,:,Nnn) )  ! now in situ density for hpg computation
[14072]220
221
[12377]222                         uu(:,:,:,Nrhs) = 0._wp            ! set dynamics trends to zero
223                         vv(:,:,:,Nrhs) = 0._wp
[4369]224
[14986]225      IF( ln_tile ) CALL dom_tile_start         ! [tiling] DYN tiling loop (1)
226      DO jtile = 1, nijtile
227         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = jtile )
228
229         IF(  lk_asminc .AND. ln_asmiau .AND. ln_dyninc )   &
230                  &         CALL dyn_asm_inc   ( kstp, Nbb, Nnn, uu, vv, Nrhs )  ! apply dynamics assimilation increment
231         IF( ln_bdy     )   CALL bdy_dyn3d_dmp ( kstp, Nbb,      uu, vv, Nrhs )  ! bdy damping trends
[4369]232#if defined key_agrif
[14986]233      END DO
234      IF( ln_tile ) CALL dom_tile_stop
235
[14072]236      IF(.NOT. Agrif_Root())  &
[5930]237               &         CALL Agrif_Sponge_dyn        ! momentum sponge
[14986]238
239      IF( ln_tile ) CALL dom_tile_start         ! [tiling] DYN tiling loop (1, continued)
240      DO jtile = 1, nijtile
241         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = jtile )
[4369]242#endif
[14986]243                            CALL dyn_adv( kstp, Nbb, Nnn      , uu, vv, Nrhs )  ! advection (VF or FF)   ==> RHS
244                            CALL dyn_vor( kstp,      Nnn      , uu, vv, Nrhs )  ! vorticity              ==> RHS
245                            CALL dyn_ldf( kstp, Nbb, Nnn      , uu, vv, Nrhs )  ! lateral mixing
246         IF( ln_zdfosm  )   CALL dyn_osm( kstp,      Nnn      , uu, vv, Nrhs )  ! OSMOSIS non-local velocity fluxes ==> RHS
247                            CALL dyn_hpg( kstp,      Nnn      , uu, vv, Nrhs )  ! horizontal gradient of Hydrostatic pressure
248      END DO
249      IF( ln_tile ) CALL dom_tile_stop
[4369]250
[14986]251                            CALL dyn_spg( kstp, Nbb, Nnn, Nrhs, uu, vv, ssh, uu_b, vv_b, Naa )  ! surface pressure gradient
252
[12377]253                                                      ! With split-explicit free surface, since now transports have been updated and ssh(:,:,Nrhs) as well
[5930]254      IF( ln_dynspg_ts ) THEN                         ! vertical scale factors and vertical velocity need to be updated
[14986]255         IF( ln_tile ) CALL dom_tile_start      ! [tiling] DYN tiling loop (2- div_hor only)
256         DO jtile = 1, nijtile
257            IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = jtile )
258
259                             CALL div_hor       ( kstp, Nbb, Nnn )               ! Horizontal divergence  (2nd call in time-split case)
260         END DO
261         IF( ln_tile ) CALL dom_tile_stop
262
263         IF(.NOT. ln_linssh) CALL dom_vvl_sf_nxt( kstp, Nbb, Nnn, Naa, kcall=2 )  ! after vertical scale factors (update depth average component)
[11287]264      ENDIF
[14986]265
266      IF( ln_tile ) CALL dom_tile_start         ! [tiling] DYN tiling loop (3- dyn_zdf only)
267      DO jtile = 1, nijtile
268         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = jtile )
269
270                               CALL dyn_zdf    ( kstp, Nbb, Nnn, Nrhs, uu, vv, Naa  )  ! vertical diffusion
271      END DO
272      IF( ln_tile ) CALL dom_tile_stop
273
[12377]274      IF( ln_dynspg_ts ) THEN                                                       ! vertical scale factors and vertical velocity need to be updated
[14644]275                            CALL wzv        ( kstp, Nbb, Nnn, Naa, ww )             ! Nnn cross-level velocity
[12377]276         IF( ln_zad_Aimp )  CALL wAimp      ( kstp,      Nnn )                      ! Adaptive-implicit vertical advection partitioning
[4369]277      ENDIF
[5930]278
[14072]279
[1482]280      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[6140]281      ! cool skin
[14072]282      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[12377]283      IF ( ln_diurnal )  CALL diurnal_layers( kstp )
[14072]284
[6140]285      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[9490]286      ! diagnostics and outputs
[1482]287      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[12377]288      IF( ln_floats  )   CALL flo_stp   ( kstp, Nbb, Nnn )      ! drifting Floats
289      IF( ln_diacfl  )   CALL dia_cfl   ( kstp,      Nnn )      ! Courant number diagnostics
290                         CALL dia_hth   ( kstp,      Nnn )      ! Thermocline depth (20 degres isotherm depth)
291      IF( ln_diadct  )   CALL dia_dct   ( kstp,      Nnn )      ! Transports
292                         CALL dia_ar5   ( kstp,      Nnn )      ! ar5 diag
293                         CALL dia_ptr   ( kstp,      Nnn )      ! Poleward adv/ldf TRansports diagnostics
294                         CALL dia_wri   ( kstp,      Nnn )      ! ocean model: outputs
295      IF( ln_crs     )   CALL crs_fld   ( kstp,      Nnn )      ! ocean model: online field coarsening & output
296      IF( lk_diadetide ) CALL dia_detide( kstp )                ! Weights computation for daily detiding of model diagnostics
297      IF( lk_diamlr  )   CALL dia_mlr                           ! Update time used in multiple-linear-regression analysis
[14072]298
[923]299#if defined key_top
[3]300      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
301      ! Passive Tracer Model
302      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[14644]303                         CALL trc_stp    ( kstp, Nbb, Nnn, Nrhs, Naa )           ! time-stepping
[3]304#endif
305
306      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[14072]307      ! Active tracers
[3]308      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[14986]309                         ts(:,:,:,:,Nrhs) = 0._wp         ! set tracer trends to zero
310
311      IF( ln_tile ) CALL dom_tile_start         ! [tiling] TRA tiling loop (1)
[13982]312      DO jtile = 1, nijtile
[14986]313         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = jtile )
[3]314
[13982]315         IF(  lk_asminc .AND. ln_asmiau .AND. &
316            & ln_trainc )   CALL tra_asm_inc( kstp, Nbb, Nnn, ts, Nrhs )  ! apply tracer assimilation increment
317                            CALL tra_sbc    ( kstp,      Nnn, ts, Nrhs )  ! surface boundary condition
318         IF( ln_traqsr  )   CALL tra_qsr    ( kstp,      Nnn, ts, Nrhs )  ! penetrative solar radiation qsr
319         IF( ln_isf     )   CALL tra_isf    ( kstp,      Nnn, ts, Nrhs )  ! ice shelf heat flux
320         IF( ln_trabbc  )   CALL tra_bbc    ( kstp,      Nnn, ts, Nrhs )  ! bottom heat flux
321         IF( ln_trabbl  )   CALL tra_bbl    ( kstp, Nbb, Nnn, ts, Nrhs )  ! advective (and/or diffusive) bottom boundary layer scheme
322         IF( ln_tradmp  )   CALL tra_dmp    ( kstp, Nbb, Nnn, ts, Nrhs )  ! internal damping trends
323         IF( ln_bdy     )   CALL bdy_tra_dmp( kstp, Nbb,      ts, Nrhs )  ! bdy damping trends
324      END DO
[14986]325      IF( ln_tile ) CALL dom_tile_stop
[13982]326
[392]327#if defined key_agrif
[14644]328      IF(.NOT. Agrif_Root() )   THEN
329                            CALL Agrif_Sponge_tra        ! tracers sponge
330      ENDIF
[389]331#endif
[1239]332
[13982]333      ! TEMP: [tiling] Separate loop over tile domains (due to tra_adv workarounds for tiling)
[14986]334      IF( ln_tile ) CALL dom_tile_start         ! [tiling] TRA tiling loop (2)
[13982]335      DO jtile = 1, nijtile
[14986]336         IF( ln_tile ) CALL dom_tile( ntsi, ntsj, ntei, ntej, ktile = jtile )
[1481]337
[13982]338                            CALL tra_adv    ( kstp, Nbb, Nnn, ts, Nrhs )  ! hor. + vert. advection ==> RHS
[14072]339         IF( ln_zdfmfc  )   CALL tra_mfc    ( kstp, Nbb,      ts, Nrhs )  ! Mass Flux Convection
[14644]340         IF( ln_zdfosm  ) THEN
341                            CALL tra_osm    ( kstp,      Nnn, ts, Nrhs )  ! OSMOSIS non-local tracer fluxes ==> RHS
342            IF( lrst_oce )  CALL osm_rst    ( kstp,      Nnn, 'WRITE'  )  ! write OSMOSIS outputs + ww (so must do here) to restarts
343         ENDIF
[13982]344                            CALL tra_ldf    ( kstp, Nbb, Nnn, ts, Nrhs )  ! lateral mixing
345
346                            CALL tra_zdf    ( kstp, Nbb, Nnn, Nrhs, ts, Naa  )  ! vertical mixing and after tracer fields
347         IF( ln_zdfnpc  )   CALL tra_npc    ( kstp,      Nnn, Nrhs, ts, Naa  )  ! update after fields by non-penetrative convection
348      END DO
[14986]349      IF( ln_tile ) CALL dom_tile_stop
[13982]350
[3]351      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[12377]352      ! Set boundary conditions, time filter and swap time levels
[3]353      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[14072]354!!jc1: For agrif, it would be much better to finalize tracers/momentum here (e.g. bdy conditions) and move the swap
355!!    (and time filtering) after Agrif update. Then restart would be done after and would contain updated fields.
356!!    If so:
[5930]357!!    (i) no need to call agrif update at initialization time
[14072]358!!    (ii) no need to update "before" fields
[5930]359!!
[14072]360!!    Apart from creating new tra_swp/dyn_swp routines, this however:
361!!    (i) makes boundary conditions at initialization time computed from updated fields which is not the case between
362!!    two restarts => restartability issue. One can circumvent this, maybe, by assuming "interface separation",
363!!    e.g. a shift of the feedback interface inside child domain.
[5930]364!!    (ii) requires that all restart outputs of updated variables by agrif (e.g. passive tracers/tke/barotropic arrays) are done at the same
365!!    place.
[14072]366!!
[12377]367!!jc2: dynnxt must be the latest call. e3t(:,:,:,Nbb) are indeed updated in that routine
368                         CALL tra_atf       ( kstp, Nbb, Nnn, Naa, ts )                      ! time filtering of "now" tracer arrays
369                         CALL dyn_atf       ( kstp, Nbb, Nnn, Naa, uu, vv, e3t, e3u, e3v  )  ! time filtering of "now" velocities and scale factors
370                         CALL ssh_atf       ( kstp, Nbb, Nnn, Naa, ssh )                     ! time filtering of "now" sea surface height
[5656]371      !
[12377]372      ! Swap time levels
373      Nrhs = Nbb
374      Nbb = Nnn
375      Nnn = Naa
376      Naa = Nrhs
377      !
378      IF(.NOT.ln_linssh) CALL dom_vvl_sf_update( kstp, Nbb, Nnn, Naa )  ! recompute vertical scale factors
379      !
380      IF( ln_diahsb  )   CALL dia_hsb       ( kstp, Nbb, Nnn )  ! - ML - global conservation diagnostics
[5836]381
382!!gm : This does not only concern the dynamics ==>>> add a new title
383!!gm2: why ouput restart before AGRIF update?
[5930]384!!
385!!jc: That would be better, but see comment above
386!!
[12377]387      IF( lrst_oce   )   CALL rst_write    ( kstp, Nbb, Nnn )   ! write output ocean restart file
[9210]388      IF( ln_sto_eos )   CALL sto_rst_write( kstp )   ! write restart file for stochastic parameters
[593]389
[5656]390#if defined key_agrif
391      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[12650]392      ! AGRIF recursive integration
[14072]393      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[12377]394                         Kbb_a = Nbb; Kmm_a = Nnn; Krhs_a = Nrhs      ! agrif_oce module copies of time level indices
395                         CALL Agrif_Integrate_ChildGrids( stp )       ! allows to finish all the Child Grids before updating
[12933]396
[9485]397#endif
[3]398      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[5656]399      ! Control
[3]400      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[12933]401                         CALL stp_ctl      ( kstp, Nnn )
402
[12650]403#if defined key_agrif
404      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
405      ! AGRIF update
[14072]406      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[14644]407      IF( Agrif_NbStepint() == 0 .AND. nstop == 0 )   &
408         &               CALL Agrif_update_all( )                  ! Update all components
[12933]409
[12650]410#endif
[14644]411      IF( ln_diaobs .AND. nstop == 0 )   &
412         &               CALL dia_obs( kstp, Nnn )  ! obs-minus-model (assimilation) diags (after dynamics update)
[12650]413
414      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
415      ! File manipulation at the end of the first time step
[14072]416      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[9210]417      IF( kstp == nit000 ) THEN                          ! 1st time step only
418                                        CALL iom_close( numror )   ! close input  ocean restart file
[14644]419         IF( lrxios )                   CALL iom_context_finalize( cr_ocerst_cxt )
[9210]420         IF(lwm)                        CALL FLUSH    ( numond )   ! flush output namelist oce
421         IF(lwm .AND. numoni /= -1 )    CALL FLUSH    ( numoni )   ! flush output namelist ice (if exist)
[1482]422      ENDIF
[3]423
[508]424      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[75]425      ! Coupled mode
426      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
[12933]427      IF( lk_oasis .AND. nstop == 0 )   CALL sbc_cpl_snd( kstp, Nbb, Nnn )     ! coupled mode : field exchanges
[508]428      !
[14644]429#if defined key_xios
[12650]430      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
431      ! Finalize contextes if end of simulation or error detected
[14072]432      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
433      IF( kstp == nitend .OR. nstop > 0 ) THEN
[5407]434                      CALL iom_context_finalize(      cxios_context          ) ! needed for XIOS+AGRIF
[14072]435         IF( ln_crs ) CALL iom_context_finalize( trim(cxios_context)//"_crs" ) !
[4152]436      ENDIF
[4153]437#endif
[3769]438      !
[12489]439      IF( l_1st_euler ) THEN         ! recover Leap-frog timestep
[14644]440         rDt   = 2._wp * rn_Dt
[12489]441         r1_Dt = 1._wp / rDt
[14072]442         l_1st_euler = .FALSE.
[12489]443      ENDIF
444      !
[9210]445      IF( ln_timing )   CALL timing_stop('stp')
[1481]446      !
[3]447   END SUBROUTINE stp
[12377]448   !
[13237]449#endif
[9019]450   !!======================================================================
[3]451END MODULE step
Note: See TracBrowser for help on using the repository browser.