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 branches/DEV_r1837_MLF/NEMO/OPA_SRC – NEMO

source: branches/DEV_r1837_MLF/NEMO/OPA_SRC/step.F90 @ 1975

Last change on this file since 1975 was 1975, checked in by mlelod, 14 years ago

ticket: #663 MLF: first part

  • Property svn:eol-style set to native
  • Property svn:keywords set to Id
File size: 20.5 KB
Line 
1MODULE step
2   !!======================================================================
3   !!                       ***  MODULE step  ***
4   !! Time-stepping    : manager of the ocean, tracer and ice time stepping
5   !!======================================================================
6   !! History :  OPA  !  1991-03  (G. Madec)  Original code
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
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
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
21   !!            3.2  !  2009-02  (G. Madec, R. Benshila)  reintroduicing z*-coordinate
22   !!             -   !  2009-06  (S. Masson, G. Madec)  TKE restart compatible with key_cpl
23   !!----------------------------------------------------------------------
24
25   !!----------------------------------------------------------------------
26   !!   stp            : OPA system time-stepping
27   !!----------------------------------------------------------------------
28   USE oce             ! ocean dynamics and tracers variables
29   USE dom_oce         ! ocean space and time domain variables
30   USE zdf_oce         ! ocean vertical physics variables
31   USE ldftra_oce      ! ocean tracer   - trends
32   USE ldfdyn_oce      ! ocean dynamics - trends
33   USE in_out_manager  ! I/O manager
34   USE iom             !
35   USE lbclnk
36
37   USE daymod          ! calendar                         (day     routine)
38
39   USE dtatem          ! ocean temperature data           (dta_tem routine)
40   USE dtasal          ! ocean salinity    data           (dta_sal routine)
41   USE sbcmod          ! surface boundary condition       (sbc     routine)
42   USE sbcrnf          ! surface boundary condition: runoff variables
43   USE sbccpl          ! surface boundary condition: coupled formulation (call send at end of step)
44   USE cpl_oasis3, ONLY : lk_cpl
45
46#if defined key_top
47   USE trcstp          ! passive tracer time-stepping      (trc_stp routine)
48#endif
49
50   USE traqsr          ! solar radiation penetration      (tra_qsr routine)
51   USE trasbc          ! surface boundary condition       (tra_sbc routine)
52   USE trabbc          ! bottom boundary condition        (tra_bbc routine)
53   USE trabbl          ! bottom boundary layer            (tra_bbl routine)
54   USE tradmp          ! internal damping                 (tra_dmp routine)
55   USE traadv          ! advection scheme control     (tra_adv_ctl routine)
56   USE traldf          ! lateral mixing                   (tra_ldf routine)
57   USE cla             ! cross land advection             (tra_cla routine)
58   !   zdfkpp          ! KPP non-local tracer fluxes      (tra_kpp routine)
59   USE trazdf          ! vertical mixing                  (tra_zdf routine)
60   USE tranxt          ! time-stepping                    (tra_nxt routine)
61   USE tranpc          ! non-penetrative convection       (tra_npc routine)
62
63   USE eosbn2          ! equation of state                (eos_bn2 routine)
64
65   USE dynadv          ! advection                        (dyn_adv routine)
66   USE dynbfr          ! Bottom friction terms            (dyn_bfr routine)
67   USE dynvor          ! vorticity term                   (dyn_vor routine)
68   USE dynhpg          ! hydrostatic pressure grad.       (dyn_hpg routine)
69   USE dynldf          ! lateral momentum diffusion       (dyn_ldf routine)
70   USE dynzdf          ! vertical diffusion               (dyn_zdf routine)
71   USE dynspg_oce      ! surface pressure gradient        (dyn_spg routine)
72   USE dynspg          ! surface pressure gradient        (dyn_spg routine)
73   USE dynnxt          ! time-stepping                    (dyn_nxt routine)
74
75   USE obc_par         ! open boundary condition variables
76   USE obcdta          ! open boundary condition data     (obc_dta routine)
77   USE obcrst          ! open boundary cond. restart      (obc_rst routine)
78   USE obcrad          ! open boundary cond. radiation    (obc_rad routine)
79
80   USE bdy_par         ! unstructured open boundary data variables
81   USE bdydta          ! unstructured open boundary data  (bdy_dta routine)
82
83   USE sshwzv          ! vertical velocity and ssh        (ssh_wzv routine)
84
85   USE ldfslp          ! iso-neutral slopes               (ldf_slp routine)
86   USE ldfeiv          ! eddy induced velocity coef.      (ldf_eiv routine)
87
88   USE zdftmx          ! tide-induced vertical mixing     (zdf_tmx routine)
89   USE zdfbfr          ! bottom friction                  (zdf_bfr routine)
90   USE zdftke_old      ! old TKE vertical mixing      (zdf_tke_old routine)
91   USE zdftke          ! TKE vertical mixing              (zdf_tke routine)
92   USE zdfkpp          ! KPP vertical mixing              (zdf_kpp routine)
93   USE zdfddm          ! double diffusion mixing          (zdf_ddm routine)
94   USE zdfevd          ! enhanced vertical diffusion      (zdf_evd routine)
95   USE zdfric          ! Richardson vertical mixing       (zdf_ric routine)
96   USE zdfmxl          ! Mixed-layer depth                (zdf_mxl routine)
97
98   USE zpshde          ! partial step: hor. derivative     (zps_hde routine)
99
100   USE diawri          ! Standard run outputs             (dia_wri routine)
101   USE trdicp          ! Ocean momentum/tracers trends    (trd_wri routine)
102   USE trdmld          ! mixed-layer trends               (trd_mld routine)
103   USE trdmld_rst      ! restart for mixed-layer trends
104   USE trdmod_oce      ! ocean momentum/tracers trends
105   USE trdvor          ! vorticity budget                 (trd_vor routine)
106   USE diagap          ! hor. mean model-data gap         (dia_gap routine)
107   USE diahdy          ! dynamic height                   (dia_hdy routine)
108   USE diaptr          ! poleward transports              (dia_ptr routine)
109   USE diaar5          ! AR5 diagnosics                   (dia_ar5 routine)
110   USE diahth          ! thermocline depth                (dia_hth routine)
111   USE diafwb          ! freshwater budget                (dia_fwb routine)
112   USE flo_oce         ! floats variables
113   USE floats          ! floats computation               (flo_stp routine)
114
115   USE stpctl          ! time stepping control            (stp_ctl routine)
116   USE restart         ! ocean restart                    (rst_wri routine)
117   USE prtctl          ! Print control                    (prt_ctl routine)
118   ! - ML -
119   USE diatrb          ! global tracer conservation       (dia_trb  routine)
120
121#if defined key_agrif
122   USE agrif_opa_sponge ! Momemtum and tracers sponges
123#endif
124
125   IMPLICIT NONE
126   PRIVATE
127
128   PUBLIC   stp        ! called by opa.F90
129
130   !! * Substitutions
131#  include "domzgr_substitute.h90"
132#  include "zdfddm_substitute.h90"
133   !!----------------------------------------------------------------------
134   !! NEMO/OPA 3.2 , LOCEAN-IPSL (2009)
135   !! $Id$
136   !! Software governed by the CeCILL licence (modipsl/doc/NEMO_CeCILL.txt)
137   !!----------------------------------------------------------------------
138
139CONTAINS
140
141#if defined key_agrif
142   SUBROUTINE stp( )
143      INTEGER             ::   kstp   ! ocean time-step index
144#else
145   SUBROUTINE stp( kstp )
146      INTEGER, INTENT(in) ::   kstp   ! ocean time-step index
147#endif
148      !!----------------------------------------------------------------------
149      !!                     ***  ROUTINE stp  ***
150      !!                     
151      !! ** Purpose : - Time stepping of OPA (momentum and active tracer eqs.)
152      !!              - Time stepping of LIM (dynamic and thermodynamic eqs.)
153      !!              - Tme stepping  of TRC (passive tracer eqs.)
154      !!
155      !! ** Method  : -1- Update forcings and data 
156      !!              -2- Update ocean physics
157      !!              -3- Compute the t and s trends
158      !!              -4- Update t and s
159      !!              -5- Compute the momentum trends
160      !!              -6- Update the horizontal velocity
161      !!              -7- Compute the diagnostics variables (rd,N2, div,cur,w)
162      !!              -8- Outputs and diagnostics
163      !!----------------------------------------------------------------------
164      INTEGER ::   jk       ! dummy loop indice
165      INTEGER ::   indic    ! error indicator if < 0
166      !! ---------------------------------------------------------------------
167
168#if defined key_agrif
169      kstp = nit000 + Agrif_Nb_Step()
170!      IF( Agrif_Root() .and. lwp) Write(*,*) '---'
171!      IF(lwp) Write(*,*) 'Grid Number',Agrif_Fixed(),' time step ',kstp
172# if defined key_iomput
173      IF( Agrif_Nbstepint() == 0) CALL iom_swap
174# endif   
175#endif   
176      indic = 1                                       ! reset to no error condition
177
178      IF( kstp /= nit000 )   CALL day( kstp )         ! Calendar (day was already called at nit000 in day_init)
179
180      CALL iom_setkt( kstp )                          ! say to iom that we are at time step kstp
181     
182      CALL rst_opn( kstp )                            ! Open the restart file
183
184      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
185      ! Update data, open boundaries, surface boundary condition (including sea-ice)
186      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
187      IF( lk_dtatem  )   CALL dta_tem( kstp )         ! update 3D temperature data
188      IF( lk_dtasal  )   CALL dta_sal( kstp )         ! update 3D salinity data
189                         CALL sbc    ( kstp )         ! Sea Boundary Condition (including sea-ice)
190      IF( lk_obc     )   CALL obc_dta( kstp )         ! update dynamic and tracer data at open boundaries
191      IF( lk_obc     )   CALL obc_rad( kstp )         ! compute phase velocities at open boundaries
192      IF( lk_bdy     )   CALL bdy_dta( kstp )         ! update dynamic and tracer data at unstructured open boundary
193
194      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
195      !  Ocean dynamics : ssh, wn, hdiv, rot                                 !
196      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
197                         CALL ssh_wzv( kstp )         ! after ssh & vertical velocity
198
199      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
200      ! Ocean physics update                (ua, va, ta, sa used as workspace)
201      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
202                         CALL bn2( tb, sb, rn2b )     ! before Brunt-Vaisala frequency
203                         CALL bn2( tn, sn, rn2  )     ! now    Brunt-Vaisala frequency
204      !
205      !  VERTICAL PHYSICS   
206                         CALL zdf_bfr( kstp )         ! bottom friction
207      !                                               ! Vertical eddy viscosity and diffusivity coefficients
208      IF( lk_zdfric     )   CALL zdf_ric    ( kstp )  ! Richardson number dependent Kz
209      IF( lk_zdftke_old )   CALL zdf_tke_old( kstp )  ! TKE closure scheme for Kz (old scheme)
210      IF( lk_zdftke     )   CALL zdf_tke    ( kstp )  ! TKE closure scheme for Kz
211      IF( lk_zdfkpp     )   CALL zdf_kpp    ( kstp )  ! KPP closure scheme for Kz
212      IF( lk_zdfcst     )   THEN                      ! Constant Kz (reset avt, avm[uv] to the background value)
213         avt (:,:,:) = rn_avt0 * tmask(:,:,:)
214         avmu(:,:,:) = rn_avm0 * umask(:,:,:)
215         avmv(:,:,:) = rn_avm0 * vmask(:,:,:)
216      ENDIF
217      IF( ln_rnf_mouth ) THEN                         ! increase diffusivity at rivers mouths
218         DO jk = 2, nkrnf   ;   avt(:,:,jk) = avt(:,:,jk) + 2.e0 * rn_avt_rnf * rnfmsk(:,:) * tmask(:,:,jk)   ;   END DO
219      ENDIF
220      IF( ln_zdfevd  )   CALL zdf_evd( kstp )         ! enhanced vertical eddy diffusivity
221
222      IF( lk_zdftmx  )   CALL zdf_tmx( kstp )         ! tidal vertical mixing
223
224      IF( lk_zdfddm .AND. .NOT. lk_zdfkpp )   &
225         &               CALL zdf_ddm( kstp )         ! double diffusive mixing
226                         CALL zdf_mxl( kstp )         ! mixed layer depth
227
228                                                      ! write tke information in the restart file
229      IF( lrst_oce .AND. lk_zdftke )   CALL tke_rst( kstp, 'WRITE' )
230      !
231      !  LATERAL  PHYSICS
232      !
233      IF( lk_ldfslp ) THEN                            ! slope of lateral mixing
234                         CALL eos( tb, sb, rhd )                ! before in situ density
235         IF( ln_zps )    CALL zps_hde( kstp, tb, sb, rhd,  &    ! Partial steps: before horizontal gradient
236            &                              gtu, gsu, gru,  &    ! of t, s, rd at the last ocean level
237            &                              gtv, gsv, grv )
238                         CALL ldf_slp( kstp, rhd, rn2b )        ! before slope of the lateral mixing
239      ENDIF
240#if defined key_traldf_c2d
241      IF( lk_traldf_eiv )   CALL ldf_eiv( kstp )      ! eddy induced velocity coefficient
242#  endif
243
244      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
245      ! diagnostics and outputs             (ua, va, ta, sa used as workspace)
246      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
247      IF( lk_floats  )   CALL flo_stp( kstp )         ! drifting Floats
248      IF( lk_diahth  )   CALL dia_hth( kstp )         ! Thermocline depth (20 degres isotherm depth)
249      IF( lk_diagap  )   CALL dia_gap( kstp )         ! basin averaged diagnostics
250      IF( lk_diahdy  )   CALL dia_hdy( kstp )         ! dynamical heigh diagnostics
251      IF( lk_diafwb  )   CALL dia_fwb( kstp )         ! Fresh water budget diagnostics
252      IF( ln_diaptr  )   CALL dia_ptr( kstp )         ! Poleward TRansports diagnostics
253      IF( lk_diaar5  )   CALL dia_ar5( kstp )         ! ar5 diag
254                         CALL dia_wri( kstp )         ! ocean model: outputs
255
256#if defined key_top
257      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
258      ! Passive Tracer Model
259      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
260                         CALL trc_stp( kstp )         ! time-stepping
261#endif
262
263      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
264      ! Active tracers                              (ua, va used as workspace)
265      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
266                             ta(:,:,:) = 0.e0               ! set tracer trends to zero
267                             sa(:,:,:) = 0.e0
268
269                             CALL tra_sbc    ( kstp )       ! surface boundary condition
270      IF( ln_traqsr      )   CALL tra_qsr    ( kstp )       ! penetrative solar radiation qsr
271      IF( lk_trabbc      )   CALL tra_bbc    ( kstp )       ! bottom heat flux
272      IF( lk_trabbl_dif  )   CALL tra_bbl_dif( kstp )       ! diffusive bottom boundary layer scheme
273      IF( lk_trabbl_adv  )   CALL tra_bbl_adv( kstp )       ! advective (and/or diffusive) bottom boundary layer scheme
274      IF( lk_tradmp      )   CALL tra_dmp    ( kstp )       ! internal damping trends
275                             CALL tra_adv    ( kstp )       ! horizontal & vertical advection
276      IF( n_cla == 1     )   CALL tra_cla    ( kstp )       ! Cross Land Advection (Update Hor. advection)
277      IF( lk_zdfkpp )        CALL tra_kpp    ( kstp )       ! KPP non-local tracer fluxes
278                             CALL tra_ldf    ( kstp )       ! lateral mixing
279#if defined key_agrif
280      IF(.NOT. Agrif_Root()) CALL Agrif_Sponge_tra          ! tracers sponge
281#endif
282                             CALL tra_zdf    ( kstp )       ! vertical mixing and after tracer fields
283
284      IF( ln_dynhpg_imp  ) THEN                             ! semi-implicit hpg (time stepping then eos)
285         IF( ln_zdfnpc   )   CALL tra_npc    ( kstp )            ! update after fields by non-penetrative convection
286                             CALL tra_nxt    ( kstp )            ! tracer fields at next time step
287                             CALL eos( ta, sa, rhd, rhop )       ! Time-filtered in situ density for hpg computation
288         IF( ln_zps      )   CALL zps_hde( kstp, ta, sa, rhd,   &   ! Partial steps: time filtered hor. derivative
289            &                                   gtu, gsu, gru,  &   ! of t, s, rd at the bottom ocean level
290            &                                   gtv, gsv, grv )
291         
292      ELSE                                                  ! centered hpg  (eos then time stepping)
293                             CALL eos( tn, sn, rhd, rhop )       ! now in situ density for hpg computation
294         IF( ln_zps      )   CALL zps_hde( kstp, tn, sn, rhd,   &   ! Partial steps: now horizontal derivative
295            &                                   gtu, gsu, gru,  &   ! of t, s, rd at the bottom ocean level
296            &                                   gtv, gsv, grv )
297         IF( ln_zdfnpc   )   CALL tra_npc    ( kstp )       ! update after fields by non-penetrative convection
298                             CALL tra_nxt    ( kstp )       ! tracer fields at next time step
299      ENDIF 
300
301      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
302      ! Dynamics                                    (ta, sa used as workspace)
303      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
304                               ua(:,:,:) = 0.e0             ! set dynamics trends to zero
305                               va(:,:,:) = 0.e0
306
307                               CALL dyn_adv( kstp )         ! advection (vector or flux form)
308                               CALL dyn_vor( kstp )         ! vorticity term including Coriolis
309                               CALL dyn_ldf( kstp )         ! lateral mixing
310#if defined key_agrif
311      IF(.NOT. Agrif_Root())   CALL Agrif_Sponge_dyn        ! momemtum sponge
312#endif
313                               CALL dyn_hpg( kstp )         ! horizontal gradient of Hydrostatic pressure
314                               CALL dyn_bfr( kstp )           ! bottom friction   
315                               CALL dyn_zdf( kstp )         ! vertical diffusion
316                               indic=0
317                               CALL dyn_spg( kstp, indic )  ! surface pressure gradient
318                               CALL dyn_nxt( kstp )         ! lateral velocity at next time step
319
320                               CALL ssh_nxt( kstp )         ! sea surface height at next time step
321                               CALL dia_trb( kstp )         ! - ML - global conservation diagnostics
322
323      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
324      ! Control and restarts
325      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
326                               CALL stp_ctl( kstp, indic )
327      IF( indic < 0        )   THEN
328                               CALL ctl_stop( 'step: indic < 0' )
329                               CALL dia_wri_state( 'output.abort', kstp )
330      ENDIF
331      IF( kstp == nit000   )   CALL iom_close( numror )     ! close input  ocean restart file
332      IF( lrst_oce         )   CALL rst_write    ( kstp )   ! write output ocean restart file
333      IF( lk_obc           )   CALL obc_rst_write( kstp )   ! write open boundary restart file
334
335      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
336      ! Trends                              (ua, va, ta, sa used as workspace)
337      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
338      IF( nstop == 0 ) THEN                         
339         IF( lk_trddyn     )   CALL trd_dwr( kstp )         ! trends: dynamics
340         IF( lk_trdtra     )   CALL trd_twr( kstp )         ! trends: active tracers
341         IF( lk_trdmld     )   CALL trd_mld( kstp )         ! trends: Mixed-layer
342         IF( lk_trdvor     )   CALL trd_vor( kstp )         ! trends: vorticity budget
343      ENDIF
344
345      !>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
346      ! Coupled mode
347      !<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
348      IF( lk_cpl           )   CALL sbc_cpl_snd( kstp )     ! coupled mode : field exchanges
349      !
350      !
351   END SUBROUTINE stp
352
353   !!======================================================================
354END MODULE step
Note: See TracBrowser for help on using the repository browser.