source: NEMO/branches/UKMO/NEMO_4.0_mirror_SI3_GPU/src/OCE/DYN/dynldf_lap_blp.F90 @ 13153

Last change on this file since 13153 was 13153, checked in by andmirek, 15 months ago

Ticket #2482: fixes after testing with ifort

File size: 8.1 KB
Line 
1MODULE dynldf_lap_blp
2   !!======================================================================
3   !!                   ***  MODULE  dynldf_lap_blp  ***
4   !! Ocean dynamics:  lateral viscosity trend (laplacian and bilaplacian)
5   !!======================================================================
6   !! History : 3.7  ! 2014-01  (G. Madec, S. Masson)  Original code, re-entrant laplacian
7   !!----------------------------------------------------------------------
8
9   !!----------------------------------------------------------------------
10   !!   dyn_ldf_lap   : update the momentum trend with the lateral viscosity using an iso-level   laplacian operator
11   !!   dyn_ldf_blp   : update the momentum trend with the lateral viscosity using an iso-level bilaplacian operator
12   !!----------------------------------------------------------------------
13   USE oce            ! ocean dynamics and tracers
14   USE dom_oce        ! ocean space and time domain
15   USE ldfdyn         ! lateral diffusion: eddy viscosity coef.
16   USE ldfslp         ! iso-neutral slopes
17   USE zdf_oce        ! ocean vertical physics
18   !
19   USE in_out_manager ! I/O manager
20   USE lbclnk         ! ocean lateral boundary conditions (or mpp link)
21
22   IMPLICIT NONE
23   PRIVATE
24
25   PUBLIC dyn_ldf_lap  ! called by dynldf.F90
26   PUBLIC dyn_ldf_blp  ! called by dynldf.F90
27
28   REAL(wp), ALLOCATABLE, DIMENSION(:, :, :) :: zulap, zvlap
29   REAL(wp), ALLOCATABLE, DIMENSION(:, :) :: zcur, zdiv
30
31   !! * Substitutions
32#  include "vectopt_loop_substitute.h90"
33   !!----------------------------------------------------------------------
34   !! NEMO/OCE 4.0 , NEMO Consortium (2018)
35   !! $Id$
36   !! Software governed by the CeCILL license (see ./LICENSE)
37   !!----------------------------------------------------------------------
38CONTAINS
39
40   SUBROUTINE dyn_ldf_lap( kt, pub, pvb, pua, pva, kpass )
41      !!----------------------------------------------------------------------
42      !!                     ***  ROUTINE dyn_ldf_lap  ***
43      !!                       
44      !! ** Purpose :   Compute the before horizontal momentum diffusive
45      !!      trend and add it to the general trend of momentum equation.
46      !!
47      !! ** Method  :   The Laplacian operator apply on horizontal velocity is
48      !!      writen as :   grad_h( ahmt div_h(U )) - curl_h( ahmf curl_z(U) )
49      !!
50      !! ** Action : - pua, pva increased by the harmonic operator applied on pub, pvb.
51      !!----------------------------------------------------------------------
52      INTEGER                         , INTENT(in   ) ::   kt         ! ocean time-step index
53      INTEGER                         , INTENT(in   ) ::   kpass      ! =1/2 first or second passage
54      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   pub, pvb   ! before velocity  [m/s]
55      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   pua, pva   ! velocity trend   [m/s2]
56      !
57      INTEGER  ::   ji, jj, jk   ! dummy loop indices
58      REAL(wp) ::   zsign        ! local scalars
59      REAL(wp) ::   zua, zva     ! local scalars
60!     REAL(wp), DIMENSION(jpi,jpj) ::   zcur, zdiv
61      !!----------------------------------------------------------------------
62      !
63      IF( kt == nit000 ) THEN
64         IF(lwp) THEN
65            WRITE(numout,*)
66            WRITE(numout,*) 'dyn_ldf : iso-level harmonic (laplacian) operator, pass=', kpass
67            WRITE(numout,*) '~~~~~~~ '
68         ENDIF
69         IF(kpass == 1) ALLOCATE(zcur(jpi,jpj), zdiv(jpi,jpj))
70      ENDIF
71      !
72      IF( kpass == 1 ) THEN   ;   zsign =  1._wp      ! bilaplacian operator require a minus sign
73      ELSE                    ;   zsign = -1._wp      !  (eddy viscosity coef. >0)
74      ENDIF
75      !
76      !                                                ! ===============
77      DO jk = 1, jpkm1                                 ! Horizontal slab
78         !                                             ! ===============
79         DO jj = 2, jpj
80            DO ji = fs_2, jpi   ! vector opt.
81               !                                      ! ahm * e3 * curl  (computed from 1 to jpim1/jpjm1)
82!!gm open question here : e3f  at before or now ?    probably now...
83!!gm note that ahmf has already been multiplied by fmask
84               zcur(ji-1,jj-1) = ahmf(ji-1,jj-1,jk) * e3f_n(ji-1,jj-1,jk) * r1_e1e2f(ji-1,jj-1)       &
85                  &     * (  e2v(ji  ,jj-1) * pvb(ji  ,jj-1,jk) - e2v(ji-1,jj-1) * pvb(ji-1,jj-1,jk)  &
86                  &        - e1u(ji-1,jj  ) * pub(ji-1,jj  ,jk) + e1u(ji-1,jj-1) * pub(ji-1,jj-1,jk)  )
87               !                                      ! ahm * div        (computed from 2 to jpi/jpj)
88!!gm note that ahmt has already been multiplied by tmask
89               zdiv(ji,jj)     = ahmt(ji,jj,jk) * r1_e1e2t(ji,jj) / e3t_b(ji,jj,jk)                                         &
90                  &     * (  e2u(ji,jj)*e3u_b(ji,jj,jk) * pub(ji,jj,jk) - e2u(ji-1,jj)*e3u_b(ji-1,jj,jk) * pub(ji-1,jj,jk)  &
91                  &        + e1v(ji,jj)*e3v_b(ji,jj,jk) * pvb(ji,jj,jk) - e1v(ji,jj-1)*e3v_b(ji,jj-1,jk) * pvb(ji,jj-1,jk)  )
92            END DO 
93         END DO 
94         !
95         DO jj = 2, jpjm1                             ! - curl( curl) + grad( div )
96            DO ji = fs_2, fs_jpim1   ! vector opt.
97               pua(ji,jj,jk) = pua(ji,jj,jk) + zsign * (                                                 &
98                  &              - ( zcur(ji  ,jj) - zcur(ji,jj-1) ) * r1_e2u(ji,jj) / e3u_n(ji,jj,jk)   &
99                  &              + ( zdiv(ji+1,jj) - zdiv(ji,jj  ) ) * r1_e1u(ji,jj)                     )
100                  !
101               pva(ji,jj,jk) = pva(ji,jj,jk) + zsign * (                                                 &
102                  &                ( zcur(ji,jj  ) - zcur(ji-1,jj) ) * r1_e1v(ji,jj) / e3v_n(ji,jj,jk)   &
103                  &              + ( zdiv(ji,jj+1) - zdiv(ji  ,jj) ) * r1_e2v(ji,jj)                     )
104            END DO
105         END DO
106         !                                             ! ===============
107      END DO                                           !   End of slab
108      !                                                ! ===============
109      !
110   END SUBROUTINE dyn_ldf_lap
111
112
113   SUBROUTINE dyn_ldf_blp( kt, pub, pvb, pua, pva )
114      !!----------------------------------------------------------------------
115      !!                 ***  ROUTINE dyn_ldf_blp  ***
116      !!                   
117      !! ** Purpose :   Compute the before lateral momentum viscous trend
118      !!              and add it to the general trend of momentum equation.
119      !!
120      !! ** Method  :   The lateral viscous trends is provided by a bilaplacian
121      !!      operator applied to before field (forward in time).
122      !!      It is computed by two successive calls to dyn_ldf_lap routine
123      !!
124      !! ** Action :   pta   updated with the before rotated bilaplacian diffusion
125      !!----------------------------------------------------------------------
126      INTEGER                         , INTENT(in   ) ::   kt         ! ocean time-step index
127      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(in   ) ::   pub, pvb   ! before velocity fields
128      REAL(wp), DIMENSION(jpi,jpj,jpk), INTENT(inout) ::   pua, pva   ! momentum trend
129      !
130!     REAL(wp), DIMENSION(jpi,jpj,jpk) ::   zulap, zvlap   ! laplacian at u- and v-point
131      !!----------------------------------------------------------------------
132      !
133      IF( kt == nit000 )  THEN
134         IF(lwp) WRITE(numout,*)
135         IF(lwp) WRITE(numout,*) 'dyn_ldf_blp : bilaplacian operator momentum '
136         IF(lwp) WRITE(numout,*) '~~~~~~~~~~~~'
137         ALLOCATE(zulap(jpi,jpj,jpk), zvlap(jpi,jpj,jpk))
138      ENDIF
139      !
140      zulap(:,:,:) = 0._wp
141      zvlap(:,:,:) = 0._wp
142      !
143      CALL dyn_ldf_lap( kt, pub, pvb, zulap, zvlap, 1 )   ! rotated laplacian applied to ptb (output in zlap)
144      !
145      CALL lbc_lnk_multi( 'dynldf_lap_blp', zulap, 'U', -1., zvlap, 'V', -1. )             ! Lateral boundary conditions
146      !
147      CALL dyn_ldf_lap( kt, zulap, zvlap, pua, pva, 2 )   ! rotated laplacian applied to zlap (output in pta)
148      !
149   END SUBROUTINE dyn_ldf_blp
150
151   !!======================================================================
152END MODULE dynldf_lap_blp
Note: See TracBrowser for help on using the repository browser.