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
RevLine 
[12732]1MODULE domqco
[12482]2   !!======================================================================
[12732]3   !!                       ***  MODULE domqco   ***
[12482]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
[14053]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*)
[12482]12   !!----------------------------------------------------------------------
13
14   !!----------------------------------------------------------------------
[14053]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
[12482]19   !!----------------------------------------------------------------------
[12680]20   USE oce            ! ocean dynamics and tracers
21   USE phycst         ! physical constant
22   USE dom_oce        ! ocean space and time domain
[12492]23   USE dynadv  , ONLY : ln_dynadv_vec
[12680]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
[12482]29   !
[12680]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
[12482]35
36   IMPLICIT NONE
37   PRIVATE
38
[12732]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
[12482]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"
[14219]59#  include "single_precision_substitute.h90"
[12482]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
[12732]67   SUBROUTINE dom_qco_init( Kbb, Kmm, Kaa )
[12482]68      !!----------------------------------------------------------------------
[12732]69      !!                ***  ROUTINE dom_qco_init  ***
[12482]70      !!
[12732]71      !! ** Purpose :  Initialization of all ssh. to h._0 ratio
[12482]72      !!
73      !! ** Method  :  - use restart file and/or initialize
[12732]74      !!               - compute ssh. to h._0 ratio
[12482]75      !!
[12732]76      !! ** Action  : - r3(t/u/v)_b
77      !!              - r3(t/u/v/f)_n
[12482]78      !!
79      !!----------------------------------------------------------------------
[14053]80      INTEGER, INTENT(in) ::   Kbb, Kmm, Kaa   ! time level indices
81      !!----------------------------------------------------------------------
[12482]82      !
83      IF(lwp) WRITE(numout,*)
[12732]84      IF(lwp) WRITE(numout,*) 'dom_qco_init : Variable volume activated'
[12482]85      IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~'
86      !
[14053]87      CALL qco_ctl                            ! choose vertical coordinate (z_star, z_tilde or layer)
[12482]88      !
[14053]89      CALL dom_qco_zgr( Kbb, Kmm )            ! interpolation scale factor, depth and water column
[12482]90      !
[14053]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
[12482]98      !
[12732]99   END SUBROUTINE dom_qco_init
[12492]100
101
[14053]102   SUBROUTINE dom_qco_zgr( Kbb, Kmm )
[12482]103      !!----------------------------------------------------------------------
[12732]104      !!                ***  ROUTINE dom_qco_init  ***
[12482]105      !!
[14053]106      !! ** Purpose :  Initialization of all r3. = ssh./h._0 ratios
[12482]107      !!
[14053]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
[12482]110      !!
[14053]111      !! ** Action  : - r3(t/u/v)(Kbb)
112      !!              - r3(t/u/v/f)(Kmm)
[12482]113      !!----------------------------------------------------------------------
[14053]114      INTEGER, INTENT(in) ::   Kbb, Kmm   ! time level indices
[12482]115      !!----------------------------------------------------------------------
116      !
117      !                    !== Set of all other vertical scale factors  ==!  (now and before)
118      !                                ! Horizontal interpolation of e3t
[14200]119#if defined key_RK3
[14219]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)           )
[14200]122#else
[14219]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(:,:) )
[14200]125#endif
[12492]126      !
[12732]127   END SUBROUTINE dom_qco_zgr
[12482]128
[12624]129
[12732]130   SUBROUTINE dom_qco_r3c( pssh, pr3t, pr3u, pr3v, pr3f )
[12492]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      !
[14053]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
[13295]157         DO_2D( 0, 0, 0, 0 )
[12492]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
[14053]163!!st      ELSE                                         !- Flux Form   (simple averaging)
164#else
[13295]165         DO_2D( 0, 0, 0, 0 )
[14053]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)
[12492]168         END_2D
[14053]169!!st      ENDIF
170#endif         
[12492]171      !
172      IF( .NOT.PRESENT( pr3f ) ) THEN              !- lbc on ratio at u-, v-points only
[14644]173         CALL lbc_lnk( 'dom_qco_r3c', pr3u, 'U', 1._wp, pr3v, 'V', 1._wp )
[12492]174         !
175         !
176      ELSE                                   !==  ratio at f-point  ==!
177         !
[14053]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
[12492]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
[14053]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)
[12492]193            END_2D
[14053]194!!st         ENDIF
195#endif
[12492]196         !                                                 ! lbc on ratio at u-,v-,f-points
[14644]197         CALL lbc_lnk( 'dom_qco_r3c', pr3u, 'U', 1._wp, pr3v, 'V', 1._wp, pr3f, 'F', 1._wp )
[12492]198         !
199      ENDIF
200      !
[12732]201   END SUBROUTINE dom_qco_r3c
[12492]202
203
[14053]204   SUBROUTINE qco_ctl
[12482]205      !!---------------------------------------------------------------------
[14053]206      !!                  ***  ROUTINE qco_ctl  ***
[12482]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,*)
[14053]226         WRITE(numout,*) 'qco_ctl : choice/control of the variable vertical coordinate'
227         WRITE(numout,*) '~~~~~~~~'
[12482]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)'
[12724]243            WRITE(numout,*) '                         rn_lf_cutoff   = 1.0/rn_Dt'
[12482]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      !
[14053]271   END SUBROUTINE qco_ctl
[12482]272
273   !!======================================================================
[12732]274END MODULE domqco
Note: See TracBrowser for help on using the repository browser.