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.
domqco.F90 in NEMO/branches/2020/dev_r14116_HPC-04_mcastril_Mixed_Precision_implementation_final/src/OCE/DOM – NEMO

source: NEMO/branches/2020/dev_r14116_HPC-04_mcastril_Mixed_Precision_implementation_final/src/OCE/DOM/domqco.F90 @ 14644

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

Merge trunk -r14642:HEAD

File size: 14.5 KB
Line 
1MODULE domqco
2   !!======================================================================
3   !!                       ***  MODULE domqco   ***
4   !! Ocean :
5   !!======================================================================
6   !! History :  2.0  !  2006-06  (B. Levier, L. Marie)  original code
7   !!            3.1  !  2009-02  (G. Madec, M. Leclair, R. Benshila)  pure z* coordinate
8   !!            3.3  !  2011-10  (M. Leclair) totally rewrote domvvl: vvl option includes z_star and z_tilde coordinates
9   !!            3.6  !  2014-11  (P. Mathiot) add ice shelf capability
10   !!            4.1  !  2019-08  (A. Coward, D. Storkey) add time level indices for prognostic variables
11   !!             -   !  2020-02  (S. Techene, G. Madec) quasi-eulerian coordinate (z* or s*)
12   !!----------------------------------------------------------------------
13
14   !!----------------------------------------------------------------------
15   !!   dom_qco_init  : define initial vertical scale factors, depths and column thickness
16   !!   dom_qco_zgr   : Set ssh/h_0 ratio at t
17   !!   dom_qco_r3c   : Compute ssh/h_0 ratio at t-, u-, v-, and optionally f-points
18   !!       qco_ctl   : Check the vvl options
19   !!----------------------------------------------------------------------
20   USE oce            ! ocean dynamics and tracers
21   USE phycst         ! physical constant
22   USE dom_oce        ! ocean space and time domain
23   USE dynadv  , ONLY : ln_dynadv_vec
24   USE isf_oce        ! iceshelf cavities
25   USE sbc_oce        ! ocean surface boundary condition
26   USE wet_dry        ! wetting and drying
27   USE usrdef_istate  ! user defined initial state (wad only)
28   USE restart        ! ocean restart
29   !
30   USE in_out_manager ! I/O manager
31   USE iom            ! I/O manager library
32   USE lib_mpp        ! distributed memory computing library
33   USE lbclnk         ! ocean lateral boundary conditions (or mpp link)
34   USE timing         ! Timing
35
36   IMPLICIT NONE
37   PRIVATE
38
39   PUBLIC  dom_qco_init       ! called by domain.F90
40   PUBLIC  dom_qco_zgr        ! called by isfcpl.F90
41   PUBLIC  dom_qco_r3c        ! called by steplf.F90
42
43   !                                                      !!* Namelist nam_vvl
44   LOGICAL , PUBLIC :: ln_vvl_zstar           = .FALSE.    ! zstar  vertical coordinate
45   LOGICAL , PUBLIC :: ln_vvl_ztilde          = .FALSE.    ! ztilde vertical coordinate
46   LOGICAL , PUBLIC :: ln_vvl_layer           = .FALSE.    ! level  vertical coordinate
47   LOGICAL , PUBLIC :: ln_vvl_ztilde_as_zstar = .FALSE.    ! ztilde vertical coordinate
48   LOGICAL , PUBLIC :: ln_vvl_zstar_at_eqtor  = .FALSE.    ! ztilde vertical coordinate
49   LOGICAL , PUBLIC :: ln_vvl_kepe            = .FALSE.    ! kinetic/potential energy transfer
50   !                                                       ! conservation: not used yet
51   REAL(wp)         :: rn_ahe3                             ! thickness diffusion coefficient
52   REAL(wp)         :: rn_rst_e3t                          ! ztilde to zstar restoration timescale [days]
53   REAL(wp)         :: rn_lf_cutoff                        ! cutoff frequency for low-pass filter  [days]
54   REAL(wp)         :: rn_zdef_max                         ! maximum fractional e3t deformation
55   LOGICAL , PUBLIC :: ln_vvl_dbg = .FALSE.                ! debug control prints
56
57   !! * Substitutions
58#  include "do_loop_substitute.h90"
59#  include "single_precision_substitute.h90"
60   !!----------------------------------------------------------------------
61   !! NEMO/OCE 4.0 , NEMO Consortium (2018)
62   !! $Id: domvvl.F90 12377 2020-02-12 14:39:06Z acc $
63   !! Software governed by the CeCILL license (see ./LICENSE)
64   !!----------------------------------------------------------------------
65CONTAINS
66
67   SUBROUTINE dom_qco_init( Kbb, Kmm, Kaa )
68      !!----------------------------------------------------------------------
69      !!                ***  ROUTINE dom_qco_init  ***
70      !!
71      !! ** Purpose :  Initialization of all ssh. to h._0 ratio
72      !!
73      !! ** Method  :  - use restart file and/or initialize
74      !!               - compute ssh. to h._0 ratio
75      !!
76      !! ** Action  : - r3(t/u/v)_b
77      !!              - r3(t/u/v/f)_n
78      !!
79      !!----------------------------------------------------------------------
80      INTEGER, INTENT(in) ::   Kbb, Kmm, Kaa   ! time level indices
81      !!----------------------------------------------------------------------
82      !
83      IF(lwp) WRITE(numout,*)
84      IF(lwp) WRITE(numout,*) 'dom_qco_init : Variable volume activated'
85      IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~'
86      !
87      CALL qco_ctl                            ! choose vertical coordinate (z_star, z_tilde or layer)
88      !
89      CALL dom_qco_zgr( Kbb, Kmm )            ! interpolation scale factor, depth and water column
90      !
91#if defined key_agrif
92      ! We need to define r3[tuv](Kaa) for AGRIF initialisation (should not be a
93      ! problem for the restartability...)
94      r3t(:,:,Kaa) = r3t(:,:,Kmm)
95      r3u(:,:,Kaa) = r3u(:,:,Kmm)
96      r3v(:,:,Kaa) = r3v(:,:,Kmm)
97#endif
98      !
99   END SUBROUTINE dom_qco_init
100
101
102   SUBROUTINE dom_qco_zgr( Kbb, Kmm )
103      !!----------------------------------------------------------------------
104      !!                ***  ROUTINE dom_qco_init  ***
105      !!
106      !! ** Purpose :  Initialization of all r3. = ssh./h._0 ratios
107      !!
108      !! ** Method  :  Call domqco using Kbb and Kmm
109      !!               NB: dom_qco_zgr is called by dom_qco_init it uses ssh from ssh_init
110      !!
111      !! ** Action  : - r3(t/u/v)(Kbb)
112      !!              - r3(t/u/v/f)(Kmm)
113      !!----------------------------------------------------------------------
114      INTEGER, INTENT(in) ::   Kbb, Kmm   ! time level indices
115      !!----------------------------------------------------------------------
116      !
117      !                    !== Set of all other vertical scale factors  ==!  (now and before)
118      !                                ! Horizontal interpolation of e3t
119#if defined key_RK3
120      CALL dom_qco_r3c( CASTWP(ssh(:,:,Kbb)), r3t(:,:,Kbb), r3u(:,:,Kbb), r3v(:,:,Kbb), r3f(:,:) )
121      CALL dom_qco_r3c( CASTWP(ssh(:,:,Kmm)), r3t(:,:,Kmm), r3u(:,:,Kmm), r3v(:,:,Kmm)           )
122#else
123      CALL dom_qco_r3c( CASTWP(ssh(:,:,Kbb)), r3t(:,:,Kbb), r3u(:,:,Kbb), r3v(:,:,Kbb)           )
124      CALL dom_qco_r3c( CASTWP(ssh(:,:,Kmm)), r3t(:,:,Kmm), r3u(:,:,Kmm), r3v(:,:,Kmm), r3f(:,:) )
125#endif
126      !
127   END SUBROUTINE dom_qco_zgr
128
129
130   SUBROUTINE dom_qco_r3c( pssh, pr3t, pr3u, pr3v, pr3f )
131      !!---------------------------------------------------------------------
132      !!                   ***  ROUTINE r3c  ***
133      !!
134      !! ** Purpose :   compute the filtered ratio ssh/h_0 at t-,u-,v-,f-points
135      !!
136      !! ** Method  : - compute the ssh at u- and v-points (f-point optional)
137      !!                   Vector Form : surface weighted averaging
138      !!                   Flux   Form : simple           averaging
139      !!              - compute the ratio ssh/h_0 at t-,u-,v-pts, (f-pt optional)
140      !!----------------------------------------------------------------------
141      REAL(wp), DIMENSION(:,:)          , INTENT(in   )  ::   pssh               ! sea surface height   [m]
142      REAL(wp), DIMENSION(:,:)          , INTENT(  out)  ::   pr3t, pr3u, pr3v   ! ssh/h0 ratio at t-, u-, v-,points  [-]
143      REAL(wp), DIMENSION(:,:), OPTIONAL, INTENT(  out)  ::   pr3f               ! ssh/h0 ratio at f-point   [-]
144      !
145      INTEGER ::   ji, jj   ! dummy loop indices
146      !!----------------------------------------------------------------------
147      !
148      !
149      pr3t(:,:) = pssh(:,:) * r1_ht_0(:,:)   !==  ratio at t-point  ==!
150      !
151      !
152      !                                      !==  ratio at u-,v-point  ==!
153      !
154!!st      IF( ln_dynadv_vec ) THEN                     !- Vector Form   (thickness weighted averaging)
155#if ! defined key_qcoTest_FluxForm
156      !                                ! no 'key_qcoTest_FluxForm' : surface weighted ssh average
157         DO_2D( 0, 0, 0, 0 )
158            pr3u(ji,jj) = 0.5_wp * (  e1e2t(ji  ,jj) * pssh(ji  ,jj)  &
159               &                    + e1e2t(ji+1,jj) * pssh(ji+1,jj)  ) * r1_hu_0(ji,jj) * r1_e1e2u(ji,jj)
160            pr3v(ji,jj) = 0.5_wp * (  e1e2t(ji,jj  ) * pssh(ji,jj  )  &
161               &                    + e1e2t(ji,jj+1) * pssh(ji,jj+1)  ) * r1_hv_0(ji,jj) * r1_e1e2v(ji,jj)
162         END_2D
163!!st      ELSE                                         !- Flux Form   (simple averaging)
164#else
165         DO_2D( 0, 0, 0, 0 )
166            pr3u(ji,jj) = 0.5_wp * (  pssh(ji,jj) + pssh(ji+1,jj  )  ) * r1_hu_0(ji,jj)
167            pr3v(ji,jj) = 0.5_wp * (  pssh(ji,jj) + pssh(ji  ,jj+1)  ) * r1_hv_0(ji,jj)
168         END_2D
169!!st      ENDIF
170#endif         
171      !
172      IF( .NOT.PRESENT( pr3f ) ) THEN              !- lbc on ratio at u-, v-points only
173         CALL lbc_lnk( 'dom_qco_r3c', pr3u, 'U', 1._wp, pr3v, 'V', 1._wp )
174         !
175         !
176      ELSE                                   !==  ratio at f-point  ==!
177         !
178!!st         IF( ln_dynadv_vec )   THEN                !- Vector Form   (thickness weighted averaging)
179#if ! defined key_qcoTest_FluxForm
180         !                                ! no 'key_qcoTest_FluxForm' : surface weighted ssh average
181
182            DO_2D( 0, 0, 0, 0 )                               ! start from 1 since lbc_lnk('F') doesn't update the 1st row/line
183               pr3f(ji,jj) = 0.25_wp * (  e1e2t(ji  ,jj  ) * pssh(ji  ,jj  )  &
184                  &                     + e1e2t(ji+1,jj  ) * pssh(ji+1,jj  )  &
185                  &                     + e1e2t(ji  ,jj+1) * pssh(ji  ,jj+1)  &
186                  &                     + e1e2t(ji+1,jj+1) * pssh(ji+1,jj+1)  ) * r1_hf_0(ji,jj) * r1_e1e2f(ji,jj)
187            END_2D
188!!st         ELSE                                      !- Flux Form   (simple averaging)
189#else
190            DO_2D( 0, 0, 0, 0 )                               ! start from 1 since lbc_lnk('F') doesn't update the 1st row/line
191               pr3f(ji,jj) = 0.25_wp * (  pssh(ji,jj  ) + pssh(ji+1,jj  )  &
192                  &                     + pssh(ji,jj+1) + pssh(ji+1,jj+1)  ) * r1_hf_0(ji,jj)
193            END_2D
194!!st         ENDIF
195#endif
196         !                                                 ! lbc on ratio at u-,v-,f-points
197         CALL lbc_lnk( 'dom_qco_r3c', pr3u, 'U', 1._wp, pr3v, 'V', 1._wp, pr3f, 'F', 1._wp )
198         !
199      ENDIF
200      !
201   END SUBROUTINE dom_qco_r3c
202
203
204   SUBROUTINE qco_ctl
205      !!---------------------------------------------------------------------
206      !!                  ***  ROUTINE qco_ctl  ***
207      !!
208      !! ** Purpose :   Control the consistency between namelist options
209      !!                for vertical coordinate
210      !!----------------------------------------------------------------------
211      INTEGER ::   ioptio, ios
212      !!
213      NAMELIST/nam_vvl/ ln_vvl_zstar, ln_vvl_ztilde, ln_vvl_layer, ln_vvl_ztilde_as_zstar, &
214         &              ln_vvl_zstar_at_eqtor      , rn_ahe3     , rn_rst_e3t            , &
215         &              rn_lf_cutoff               , rn_zdef_max , ln_vvl_dbg                ! not yet implemented: ln_vvl_kepe
216      !!----------------------------------------------------------------------
217      !
218      READ  ( numnam_ref, nam_vvl, IOSTAT = ios, ERR = 901)
219901   IF( ios /= 0 )   CALL ctl_nam ( ios , 'nam_vvl in reference namelist' )
220      READ  ( numnam_cfg, nam_vvl, IOSTAT = ios, ERR = 902 )
221902   IF( ios >  0 ) CALL ctl_nam ( ios , 'nam_vvl in configuration namelist' )
222      IF(lwm) WRITE ( numond, nam_vvl )
223      !
224      IF(lwp) THEN                    ! Namelist print
225         WRITE(numout,*)
226         WRITE(numout,*) 'qco_ctl : choice/control of the variable vertical coordinate'
227         WRITE(numout,*) '~~~~~~~~'
228         WRITE(numout,*) '   Namelist nam_vvl : chose a vertical coordinate'
229         WRITE(numout,*) '      zstar                      ln_vvl_zstar   = ', ln_vvl_zstar
230         WRITE(numout,*) '      ztilde                     ln_vvl_ztilde  = ', ln_vvl_ztilde
231         WRITE(numout,*) '      layer                      ln_vvl_layer   = ', ln_vvl_layer
232         WRITE(numout,*) '      ztilde as zstar   ln_vvl_ztilde_as_zstar  = ', ln_vvl_ztilde_as_zstar
233         WRITE(numout,*) '      ztilde near the equator    ln_vvl_zstar_at_eqtor  = ', ln_vvl_zstar_at_eqtor
234         WRITE(numout,*) '      !'
235         WRITE(numout,*) '      thickness diffusion coefficient                      rn_ahe3      = ', rn_ahe3
236         WRITE(numout,*) '      maximum e3t deformation fractional change            rn_zdef_max  = ', rn_zdef_max
237         IF( ln_vvl_ztilde_as_zstar ) THEN
238            WRITE(numout,*) '      ztilde running in zstar emulation mode (ln_vvl_ztilde_as_zstar=T) '
239            WRITE(numout,*) '         ignoring namelist timescale parameters and using:'
240            WRITE(numout,*) '            hard-wired : z-tilde to zstar restoration timescale (days)'
241            WRITE(numout,*) '                         rn_rst_e3t     = 0.e0'
242            WRITE(numout,*) '            hard-wired : z-tilde cutoff frequency of low-pass filter (days)'
243            WRITE(numout,*) '                         rn_lf_cutoff   = 1.0/rn_Dt'
244         ELSE
245            WRITE(numout,*) '      z-tilde to zstar restoration timescale (days)        rn_rst_e3t   = ', rn_rst_e3t
246            WRITE(numout,*) '      z-tilde cutoff frequency of low-pass filter (days)   rn_lf_cutoff = ', rn_lf_cutoff
247         ENDIF
248         WRITE(numout,*) '         debug prints flag                                 ln_vvl_dbg   = ', ln_vvl_dbg
249      ENDIF
250      !
251      ioptio = 0                      ! Parameter control
252      IF( ln_vvl_ztilde_as_zstar )   ln_vvl_ztilde = .true.
253      IF( ln_vvl_zstar           )   ioptio = ioptio + 1
254      IF( ln_vvl_ztilde          )   ioptio = ioptio + 1
255      IF( ln_vvl_layer           )   ioptio = ioptio + 1
256      !
257      IF( ioptio /= 1 )   CALL ctl_stop( 'Choose ONE vertical coordinate in namelist nam_vvl' )
258      !
259      IF(lwp) THEN                   ! Print the choice
260         WRITE(numout,*)
261         IF( ln_vvl_zstar           ) WRITE(numout,*) '      ==>>>   zstar vertical coordinate is used'
262         IF( ln_vvl_ztilde          ) WRITE(numout,*) '      ==>>>   ztilde vertical coordinate is used'
263         IF( ln_vvl_layer           ) WRITE(numout,*) '      ==>>>   layer vertical coordinate is used'
264         IF( ln_vvl_ztilde_as_zstar ) WRITE(numout,*) '      ==>>>   to emulate a zstar coordinate'
265      ENDIF
266      !
267#if defined key_agrif
268      IF( (.NOT.Agrif_Root()).AND.(.NOT.ln_vvl_zstar) )   CALL ctl_stop( 'AGRIF is implemented with zstar coordinate only' )
269#endif
270      !
271   END SUBROUTINE qco_ctl
272
273   !!======================================================================
274END MODULE domqco
Note: See TracBrowser for help on using the repository browser.