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.
agrif_opa_interp.F90 in branches/2017/dev_r8183_ICEMODEL/NEMOGCM/NEMO/NST_SRC – NEMO

source: branches/2017/dev_r8183_ICEMODEL/NEMOGCM/NEMO/NST_SRC/agrif_opa_interp.F90 @ 8226

Last change on this file since 8226 was 8226, checked in by clem, 7 years ago

merge with dev_r8127_AGRIF_LIM3_GHOST@r8189 and dev_r8126_ROBUST08_no_ghost@r8196
  • Property svn:keywords set to Id
File size: 50.0 KB
Line 
1MODULE agrif_opa_interp
2   !!======================================================================
3   !!                   ***  MODULE  agrif_opa_interp  ***
4   !! AGRIF: interpolation package
5   !!======================================================================
6   !! History :  2.0  !  2002-06  (XXX)  Original cade
7   !!             -   !  2005-11  (XXX)
8   !!            3.2  !  2009-04  (R. Benshila)
9   !!            3.6  !  2014-09  (R. Benshila)
10   !!----------------------------------------------------------------------
11#if defined key_agrif
12   !!----------------------------------------------------------------------
13   !!   'key_agrif'                                              AGRIF zoom
14   !!----------------------------------------------------------------------
15   !!   Agrif_tra     :
16   !!   Agrif_dyn     :
17   !!   interpu       :
18   !!   interpv       :
19   !!----------------------------------------------------------------------
20   USE par_oce
21   USE oce
22   USE dom_oce     
23   USE zdf_oce
24   USE agrif_oce
25   USE phycst
26   !
27   USE in_out_manager
28   USE agrif_opa_sponge
29   USE lib_mpp
30   USE wrk_nemo
31 
32   IMPLICIT NONE
33   PRIVATE
34
35   PUBLIC   Agrif_tra, Agrif_dyn, Agrif_ssh, Agrif_dyn_ts, Agrif_ssh_ts, Agrif_dta_ts
36   PUBLIC   interpun, interpvn
37   PUBLIC   interptsn, interpsshn
38   PUBLIC   interpunb, interpvnb, interpub2b, interpvb2b
39   PUBLIC   interpe3t, interpumsk, interpvmsk
40# if defined key_zdftke
41   PUBLIC   Agrif_tke, interpavm
42# endif
43
44   INTEGER ::   bdy_tinterp = 0
45
46#  include "vectopt_loop_substitute.h90"
47   !!----------------------------------------------------------------------
48   !! NEMO/NST 3.7 , NEMO Consortium (2015)
49   !! $Id$
50   !! Software governed by the CeCILL licence (NEMOGCM/NEMO_CeCILL.txt)
51   !!----------------------------------------------------------------------
52CONTAINS
53
54   SUBROUTINE Agrif_tra
55      !!----------------------------------------------------------------------
56      !!                  ***  ROUTINE Agrif_tra  ***
57      !!----------------------------------------------------------------------
58      !
59      IF( Agrif_Root() )   RETURN
60      !
61      Agrif_SpecialValue    = 0._wp
62      Agrif_UseSpecialValue = .TRUE.
63      !
64      CALL Agrif_Bc_variable( tsn_id, procname=interptsn )
65      !
66      Agrif_UseSpecialValue = .FALSE.
67      !
68   END SUBROUTINE Agrif_tra
69
70
71   SUBROUTINE Agrif_dyn( kt )
72      !!----------------------------------------------------------------------
73      !!                  ***  ROUTINE Agrif_DYN  ***
74      !!---------------------------------------------------------------------- 
75      INTEGER, INTENT(in) ::   kt
76      !
77      INTEGER ::   ji, jj, jk       ! dummy loop indices
78      INTEGER ::   j1, j2, i1, i2
79      REAL(wp), POINTER, DIMENSION(:,:) ::   zub, zvb
80      !!---------------------------------------------------------------------- 
81      !
82      IF( Agrif_Root() )   RETURN
83      !
84      CALL wrk_alloc( jpi,jpj,   zub, zvb )
85      !
86      Agrif_SpecialValue    = 0._wp
87      Agrif_UseSpecialValue = ln_spc_dyn
88      !
89      CALL Agrif_Bc_variable( un_interp_id, procname=interpun )
90      CALL Agrif_Bc_variable( vn_interp_id, procname=interpvn )
91      !
92      Agrif_UseSpecialValue = .FALSE.
93      !
94      ! prevent smoothing in ghost cells
95      i1 =  1   ;   i2 = jpi
96      j1 =  1   ;   j2 = jpj
97      IF( nbondj == -1 .OR. nbondj == 2 )   j1 = 3
98      IF( nbondj == +1 .OR. nbondj == 2 )   j2 = nlcj-2
99      IF( nbondi == -1 .OR. nbondi == 2 )   i1 = 3
100      IF( nbondi == +1 .OR. nbondi == 2 )   i2 = nlci-2
101
102      ! --- West --- !
103      IF( nbondi == -1 .OR. nbondi == 2 ) THEN
104         !
105         IF( .NOT.ln_dynspg_ts ) THEN  ! Store transport
106            ua_b(2:1+nbghostcells,:) = 0._wp
107            DO jk = 1, jpkm1
108               DO jj = 1, jpj
109                  ua_b(2:1+nbghostcells,jj) = ua_b(2:1+nbghostcells,jj) + e3u_a(2:1+nbghostcells,jj,jk) * ua(2:1+nbghostcells,jj,jk)
110               END DO
111            END DO
112            DO jj = 1, jpj
113               ua_b(2:1+nbghostcells,jj) = ua_b(2:1+nbghostcells,jj) * r1_hu_a(2:1+nbghostcells,jj)
114            END DO
115         ENDIF
116         !
117         ! Smoothing if only 1 ghostcell
118         ! -----------------------------
119         IF( nbghostcells == 1 ) THEN
120            DO jk=1,jpkm1                 ! Smooth
121               DO jj=j1,j2
122                  ua(2,jj,jk) = 0.25_wp*(ua(1,jj,jk)+2._wp*ua(2,jj,jk)+ua(3,jj,jk))
123                  ua(2,jj,jk) = ua(2,jj,jk) * umask(2,jj,jk)
124               END DO
125            END DO
126            !
127            zub(2,:) = 0._wp              ! Correct transport
128            DO jk = 1, jpkm1
129               DO jj = 1, jpj
130                  zub(2,jj) = zub(2,jj) + e3u_a(2,jj,jk) * ua(2,jj,jk)
131               END DO
132            END DO
133            DO jj=1,jpj
134               zub(2,jj) = zub(2,jj) * r1_hu_a(2,jj)
135            END DO
136           
137            DO jk=1,jpkm1
138               DO jj=1,jpj
139                  ua(2,jj,jk) = (ua(2,jj,jk)+ua_b(2,jj)-zub(2,jj))*umask(2,jj,jk)
140               END DO
141            END DO
142           
143            IF( ln_dynspg_ts ) THEN       ! Set tangential velocities to time splitting estimate
144               zvb(2,:) = 0._wp
145               DO jk = 1, jpkm1
146                  DO jj = 1, jpj
147                     zvb(2,jj) = zvb(2,jj) + e3v_a(2,jj,jk) * va(2,jj,jk)
148                  END DO
149               END DO
150               DO jj = 1, jpj
151                  zvb(2,jj) = zvb(2,jj) * r1_hv_a(2,jj)
152               END DO
153               DO jk = 1, jpkm1
154                  DO jj = 1, jpj
155                     va(2,jj,jk) = (va(2,jj,jk)+va_b(2,jj)-zvb(2,jj)) * vmask(2,jj,jk)
156                  END DO
157               END DO
158            ENDIF
159            !
160         ENDIF
161         !
162      ENDIF
163
164      ! --- East --- !
165      IF( nbondi == 1 .OR. nbondi == 2 ) THEN
166
167         IF( .NOT.ln_dynspg_ts ) THEN  ! Store transport
168            ua_b(nlci-nbghostcells-1:nlci-2,:) = 0._wp
169            DO jk=1,jpkm1
170               DO jj=1,jpj
171                  ua_b(nlci-nbghostcells-1:nlci-2,jj) = ua_b(nlci-nbghostcells-1:nlci-2,jj) + e3u_a(nlci-nbghostcells-1:nlci-2,jj,jk)  &
172                     &                                                                         * ua(nlci-nbghostcells-1:nlci-2,jj,jk)
173               END DO
174            END DO
175            DO jj=1,jpj
176               ua_b(nlci-nbghostcells-1:nlci-2,jj) = ua_b(nlci-nbghostcells-1:nlci-2,jj) * r1_hu_a(nlci-nbghostcells-1:nlci-2,jj) 
177            END DO
178         ENDIF
179         !
180         ! Smoothing if only 1 ghostcell
181         ! -----------------------------
182         IF( nbghostcells == 1 ) THEN
183            DO jk = 1, jpkm1              ! Smooth
184               DO jj = j1, j2
185                  ua(nlci-2,jj,jk) = 0.25_wp * umask(nlci-2,jj,jk)      &
186                     &             * ( ua(nlci-3,jj,jk) + 2._wp*ua(nlci-2,jj,jk) + ua(nlci-1,jj,jk) )
187               END DO
188            END DO
189           
190            zub(nlci-2,:) = 0._wp        ! Correct transport
191            DO jk = 1, jpkm1
192               DO jj = 1, jpj
193                  zub(nlci-2,jj) = zub(nlci-2,jj) + e3u_a(nlci-2,jj,jk) * ua(nlci-2,jj,jk)
194               END DO
195            END DO
196            DO jj = 1, jpj
197               zub(nlci-2,jj) = zub(nlci-2,jj) * r1_hu_a(nlci-2,jj)
198            END DO
199           
200            DO jk = 1, jpkm1
201               DO jj = 1, jpj
202                  ua(nlci-2,jj,jk) = ( ua(nlci-2,jj,jk) + ua_b(nlci-2,jj) - zub(nlci-2,jj) ) * umask(nlci-2,jj,jk)
203               END DO
204            END DO
205            !
206            IF( ln_dynspg_ts ) THEN       ! Set tangential velocities to time splitting estimate
207               zvb(nlci-1,:) = 0._wp
208               DO jk = 1, jpkm1
209                  DO jj = 1, jpj
210                     zvb(nlci-1,jj) = zvb(nlci-1,jj) + e3v_a(nlci-1,jj,jk) * va(nlci-1,jj,jk)
211                  END DO
212               END DO
213               DO jj=1,jpj
214                  zvb(nlci-1,jj) = zvb(nlci-1,jj) * r1_hv_a(nlci-1,jj)
215               END DO
216               DO jk = 1, jpkm1
217                  DO jj = 1, jpj
218                     va(nlci-1,jj,jk) = ( va(nlci-1,jj,jk) + va_b(nlci-1,jj) - zvb(nlci-1,jj) ) * vmask(nlci-1,jj,jk)
219                  END DO
220               END DO
221            ENDIF
222            !
223         ENDIF
224         !
225      ENDIF
226
227      ! --- South --- !
228      IF( nbondj == -1 .OR. nbondj == 2 ) THEN
229
230         IF( .NOT.ln_dynspg_ts ) THEN  ! Store transport
231            va_b(:,2:nbghostcells+1) = 0._wp
232            DO jk = 1, jpkm1
233               DO ji = 1, jpi
234                  va_b(ji,2:nbghostcells+1) = va_b(ji,2:nbghostcells+1) + e3v_a(ji,2:nbghostcells+1,jk) * va(ji,2:nbghostcells+1,jk)
235               END DO
236            END DO
237            DO ji=1,jpi
238               va_b(ji,2:nbghostcells+1) = va_b(ji,2:nbghostcells+1) * r1_hv_a(ji,2:nbghostcells+1)
239            END DO
240         ENDIF
241         !
242         ! Smoothing if only 1 ghostcell
243         ! -----------------------------
244         IF( nbghostcells == 1 ) THEN
245            DO jk = 1, jpkm1              ! Smooth
246               DO ji = i1, i2
247                  va(ji,2,jk) = 0.25_wp * vmask(ji,2,jk)    &
248                     &        * ( va(ji,1,jk) + 2._wp*va(ji,2,jk) + va(ji,3,jk) )
249               END DO
250            END DO
251            !
252            zvb(:,2) = 0._wp              ! Correct transport
253            DO jk=1,jpkm1
254               DO ji=1,jpi
255                  zvb(ji,2) = zvb(ji,2) + e3v_a(ji,2,jk) * va(ji,2,jk) * vmask(ji,2,jk)
256               END DO
257            END DO
258            DO ji = 1, jpi
259               zvb(ji,2) = zvb(ji,2) * r1_hv_a(ji,2)
260            END DO
261            DO jk = 1, jpkm1
262               DO ji = 1, jpi
263                  va(ji,2,jk) = ( va(ji,2,jk) + va_b(ji,2) - zvb(ji,2) ) * vmask(ji,2,jk)
264               END DO
265            END DO
266           
267            IF( ln_dynspg_ts ) THEN       ! Set tangential velocities to time splitting estimate
268               zub(:,2) = 0._wp
269               DO jk = 1, jpkm1
270                  DO ji = 1, jpi
271                     zub(ji,2) = zub(ji,2) + e3u_a(ji,2,jk) * ua(ji,2,jk) * umask(ji,2,jk)
272                  END DO
273               END DO
274               DO ji = 1, jpi
275                  zub(ji,2) = zub(ji,2) * r1_hu_a(ji,2)
276               END DO
277               
278               DO jk = 1, jpkm1
279                  DO ji = 1, jpi
280                     ua(ji,2,jk) = ( ua(ji,2,jk) + ua_b(ji,2) - zub(ji,2) ) * umask(ji,2,jk)
281                  END DO
282               END DO
283            ENDIF
284            !
285         ENDIF
286         !
287      ENDIF
288
289      ! --- North --- !
290      IF( nbondj == 1 .OR. nbondj == 2 ) THEN
291         !
292         IF( .NOT.ln_dynspg_ts ) THEN  ! Store transport
293            va_b(:,nlcj-nbghostcells-1:nlcj-2) = 0._wp
294            DO jk = 1, jpkm1
295               DO ji = 1, jpi
296                  va_b(ji,nlcj-nbghostcells-1:nlcj-2) = va_b(ji,nlcj-nbghostcells-1:nlcj-2) + e3v_a(ji,nlcj-nbghostcells-1:nlcj-2,jk)  &
297                     &                                                                         * va(ji,nlcj-nbghostcells-1:nlcj-2,jk)
298               END DO
299            END DO
300            DO ji = 1, jpi
301               va_b(ji,nlcj-nbghostcells-1:nlcj-2) = va_b(ji,nlcj-nbghostcells-1:nlcj-2) * r1_hv_a(ji,nlcj-nbghostcells-1:nlcj-2)
302            END DO
303         ENDIF
304         !
305         ! Smoothing if only 1 ghostcell
306         ! -----------------------------
307         IF( nbghostcells == 1 ) THEN
308            DO jk = 1, jpkm1              ! Smooth
309               DO ji = i1, i2
310                  va(ji,nlcj-2,jk) = 0.25_wp * vmask(ji,nlcj-2,jk)   &
311                     &             * ( va(ji,nlcj-3,jk) + 2._wp * va(ji,nlcj-2,jk) + va(ji,nlcj-1,jk) )
312               END DO
313            END DO
314            !
315            zvb(:,nlcj-2) = 0._wp         ! Correct transport
316            DO jk = 1, jpkm1
317               DO ji = 1, jpi
318                  zvb(ji,nlcj-2) = zvb(ji,nlcj-2) + e3v_a(ji,nlcj-2,jk) * va(ji,nlcj-2,jk) * vmask(ji,nlcj-2,jk)
319               END DO
320            END DO
321            DO ji = 1, jpi
322               zvb(ji,nlcj-2) = zvb(ji,nlcj-2) * r1_hv_a(ji,nlcj-2)
323            END DO
324            DO jk = 1, jpkm1
325               DO ji = 1, jpi
326                  va(ji,nlcj-2,jk) = ( va(ji,nlcj-2,jk) + va_b(ji,nlcj-2) - zvb(ji,nlcj-2) ) * vmask(ji,nlcj-2,jk)
327               END DO
328            END DO
329            !
330            IF( ln_dynspg_ts ) THEN       ! Set tangential velocities to time splitting estimate
331               zub(:,nlcj-1) = 0._wp
332               DO jk = 1, jpkm1
333                  DO ji = 1, jpi
334                     zub(ji,nlcj-1) = zub(ji,nlcj-1) + e3u_a(ji,nlcj-1,jk) * ua(ji,nlcj-1,jk) * umask(ji,nlcj-1,jk)
335                  END DO
336               END DO
337               DO ji = 1, jpi
338                  zub(ji,nlcj-1) = zub(ji,nlcj-1) * r1_hu_a(ji,nlcj-1)
339               END DO
340               !
341               DO jk = 1, jpkm1
342                  DO ji = 1, jpi
343                     ua(ji,nlcj-1,jk) = ( ua(ji,nlcj-1,jk) + ua_b(ji,nlcj-1) - zub(ji,nlcj-1) ) * umask(ji,nlcj-1,jk)
344                  END DO
345               END DO
346            ENDIF
347            !
348         ENDIF
349         !
350      ENDIF
351      !
352      CALL wrk_dealloc( jpi,jpj,   zub, zvb )
353      !
354   END SUBROUTINE Agrif_dyn
355
356
357   SUBROUTINE Agrif_dyn_ts( jn )
358      !!----------------------------------------------------------------------
359      !!                  ***  ROUTINE Agrif_dyn_ts  ***
360      !!---------------------------------------------------------------------- 
361      !!
362      INTEGER, INTENT(in) ::   jn
363      !!
364      INTEGER :: ji, jj
365      !!---------------------------------------------------------------------- 
366      !
367      IF( Agrif_Root() )   RETURN
368      !! clem ghost
369      IF((nbondi == -1).OR.(nbondi == 2)) THEN
370         DO jj=1,jpj
371            IF( vmask(2,jj,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
372               va_e(2:nbghostcells+1,jj) = vbdy_w(jj) * hvr_e(2:nbghostcells+1,jj)
373               ! Specified fluxes:
374               ua_e(2:nbghostcells+1,jj) = ubdy_w(jj) * hur_e(2:nbghostcells+1,jj)
375               ! Characteristics method (only if ghostcells=1):
376               !alt            ua_e(2,jj) = 0.5_wp * ( ubdy_w(jj) * hur_e(2,jj) + ua_e(3,jj) &
377               !alt                       &           - sqrt(grav * hur_e(2,jj)) * (sshn_e(3,jj) - hbdy_w(jj)) )
378            ENDIF
379         END DO
380      ENDIF
381      !
382      IF((nbondi == 1).OR.(nbondi == 2)) THEN
383         DO jj=1,jpj
384            IF( vmask(nlci-1,jj,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
385               va_e(nlci-nbghostcells:nlci-1,jj)   = vbdy_e(jj) * hvr_e(nlci-nbghostcells:nlci-1,jj)
386               ! Specified fluxes:
387               ua_e(nlci-nbghostcells-1:nlci-2,jj) = ubdy_e(jj) * hur_e(nlci-nbghostcells-1:nlci-2,jj)
388               ! Characteristics method (only if ghostcells=1):
389               !alt            ua_e(nlci-2,jj) = 0.5_wp * ( ubdy_e(jj) * hur_e(nlci-2,jj) + ua_e(nlci-3,jj) &
390               !alt                            &           + sqrt(grav * hur_e(nlci-2,jj)) * (sshn_e(nlci-2,jj) - hbdy_e(jj)) )
391            ENDIF
392         END DO
393      ENDIF
394      !
395      IF((nbondj == -1).OR.(nbondj == 2)) THEN
396         DO ji=1,jpi
397            IF( umask(ji,2,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
398               ua_e(ji,2:nbghostcells+1) = ubdy_s(ji) * hur_e(ji,2:nbghostcells+1)
399               ! Specified fluxes:
400               va_e(ji,2:nbghostcells+1) = vbdy_s(ji) * hvr_e(ji,2:nbghostcells+1)
401               ! Characteristics method (only if ghostcells=1):
402               !alt            va_e(ji,2) = 0.5_wp * ( vbdy_s(ji) * hvr_e(ji,2) + va_e(ji,3) &
403               !alt                       &           - sqrt(grav * hvr_e(ji,2)) * (sshn_e(ji,3) - hbdy_s(ji)) )
404            ENDIF
405         END DO
406      ENDIF
407      !
408      IF((nbondj == 1).OR.(nbondj == 2)) THEN
409         DO ji=1,jpi
410            IF( umask(ji,nlcj-1,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
411               ua_e(ji,nlcj-nbghostcells:nlcj-1)   = ubdy_n(ji) * hur_e(ji,nlcj-nbghostcells:nlcj-1)
412               ! Specified fluxes:
413               va_e(ji,nlcj-nbghostcells-1:nlcj-2) = vbdy_n(ji) * hvr_e(ji,nlcj-nbghostcells-1:nlcj-2)
414               ! Characteristics method (only if ghostcells=1):
415               !alt            va_e(ji,nlcj-2) = 0.5_wp * ( vbdy_n(ji) * hvr_e(ji,nlcj-2)  + va_e(ji,nlcj-3) &
416               !alt                            &           + sqrt(grav * hvr_e(ji,nlcj-2)) * (sshn_e(ji,nlcj-2) - hbdy_n(ji)) )
417            ENDIF
418         END DO
419      ENDIF
420      !
421   END SUBROUTINE Agrif_dyn_ts
422
423
424   SUBROUTINE Agrif_dta_ts( kt )
425      !!----------------------------------------------------------------------
426      !!                  ***  ROUTINE Agrif_dta_ts  ***
427      !!---------------------------------------------------------------------- 
428      !!
429      INTEGER, INTENT(in) ::   kt
430      !!
431      INTEGER :: ji, jj
432      LOGICAL :: ll_int_cons
433      REAL(wp) :: zrhot, zt
434      !!---------------------------------------------------------------------- 
435      !
436      IF( Agrif_Root() )   RETURN
437      !
438      ll_int_cons = ln_bt_fw ! Assume conservative temporal integration in the forward case only
439      !
440      zrhot = Agrif_rhot()
441      !
442      ! "Central" time index for interpolation:
443      IF( ln_bt_fw ) THEN
444         zt = REAL( Agrif_NbStepint()+0.5_wp, wp ) / zrhot
445      ELSE
446         zt = REAL( Agrif_NbStepint()       , wp ) / zrhot
447      ENDIF
448      !
449      ! Linear interpolation of sea level
450      Agrif_SpecialValue    = 0._wp
451      Agrif_UseSpecialValue = .TRUE.
452      CALL Agrif_Bc_variable( sshn_id, calledweight=zt, procname=interpsshn )
453      Agrif_UseSpecialValue = .FALSE.
454      !
455      ! Interpolate barotropic fluxes
456      Agrif_SpecialValue=0.
457      Agrif_UseSpecialValue = ln_spc_dyn
458      !
459      IF( ll_int_cons ) THEN  ! Conservative interpolation
460         ! order matters here !!!!!!
461         CALL Agrif_Bc_variable( ub2b_interp_id, calledweight=1._wp, procname=interpub2b ) ! Time integrated
462         CALL Agrif_Bc_variable( vb2b_interp_id, calledweight=1._wp, procname=interpvb2b )
463         bdy_tinterp = 1
464         CALL Agrif_Bc_variable( unb_id        , calledweight=1._wp, procname=interpunb  ) ! After
465         CALL Agrif_Bc_variable( vnb_id        , calledweight=1._wp, procname=interpvnb  )
466         bdy_tinterp = 2
467         CALL Agrif_Bc_variable( unb_id        , calledweight=0._wp, procname=interpunb  ) ! Before
468         CALL Agrif_Bc_variable( vnb_id        , calledweight=0._wp, procname=interpvnb  )         
469      ELSE ! Linear interpolation
470         bdy_tinterp = 0
471         ubdy_w(:) = 0._wp   ;   vbdy_w(:) = 0._wp 
472         ubdy_e(:) = 0._wp   ;   vbdy_e(:) = 0._wp 
473         ubdy_n(:) = 0._wp   ;   vbdy_n(:) = 0._wp 
474         ubdy_s(:) = 0._wp   ;   vbdy_s(:) = 0._wp
475         CALL Agrif_Bc_variable( unb_id, calledweight=zt, procname=interpunb )
476         CALL Agrif_Bc_variable( vnb_id, calledweight=zt, procname=interpvnb )
477      ENDIF
478      Agrif_UseSpecialValue = .FALSE.
479      !
480   END SUBROUTINE Agrif_dta_ts
481
482
483   SUBROUTINE Agrif_ssh( kt )
484      !!----------------------------------------------------------------------
485      !!                  ***  ROUTINE Agrif_DYN  ***
486      !!---------------------------------------------------------------------- 
487      INTEGER, INTENT(in) ::   kt
488      !
489      INTEGER  :: ji, jj, indx
490      !!---------------------------------------------------------------------- 
491      !
492      IF( Agrif_Root() )   RETURN
493      !! clem ghost
494      ! --- West --- !
495      IF((nbondi == -1).OR.(nbondi == 2)) THEN
496         indx = 1+nbghostcells
497         DO jj = 1, jpj
498            DO ji = 2, indx
499               IF( tmask(2,jj,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
500                  ssha(ji,jj)=ssha(indx+1,jj)
501                  sshn(ji,jj)=sshn(indx+1,jj)
502               ENDIF
503            ENDDO
504         ENDDO
505      ENDIF
506      !
507      ! --- East --- !
508      IF((nbondi == 1).OR.(nbondi == 2)) THEN
509         indx = nlci-nbghostcells
510         DO jj = 1, jpj
511            DO ji = indx, nlci-1
512               IF( tmask(nlci-1,jj,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
513                  ssha(ji,jj)=ssha(indx-1,jj)
514                  sshn(ji,jj)=sshn(indx-1,jj)
515               ENDIF
516            ENDDO
517         ENDDO
518      ENDIF
519      !
520      ! --- South --- !
521      IF((nbondj == -1).OR.(nbondj == 2)) THEN
522         indx = 1+nbghostcells
523         DO jj = 2, indx
524            DO ji = 1, jpi
525               IF( tmask(ji,2,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
526                  ssha(ji,jj)=ssha(ji,indx+1)
527                  sshn(ji,jj)=sshn(ji,indx+1)
528               ENDIF
529            ENDDO
530         ENDDO
531      ENDIF
532      !
533      ! --- North --- !
534      IF((nbondj == 1).OR.(nbondj == 2)) THEN
535         indx = nlcj-nbghostcells
536         DO jj = indx, nlcj-1
537            DO ji = 1, jpi
538               IF( tmask(ji,nlcj-1,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
539                  ssha(ji,jj)=ssha(ji,indx-1)
540                  sshn(ji,jj)=sshn(ji,indx-1)
541               ENDIF
542            ENDDO
543         ENDDO
544      ENDIF
545      !
546   END SUBROUTINE Agrif_ssh
547
548
549   SUBROUTINE Agrif_ssh_ts( jn )
550      !!----------------------------------------------------------------------
551      !!                  ***  ROUTINE Agrif_ssh_ts  ***
552      !!---------------------------------------------------------------------- 
553      INTEGER, INTENT(in) ::   jn
554      !!
555      INTEGER :: ji, jj
556      !!---------------------------------------------------------------------- 
557      !! clem ghost (starting at i,j=1 is important I think otherwise you introduce a grad(ssh)/=0 at point 2)
558      IF((nbondi == -1).OR.(nbondi == 2)) THEN
559         DO jj = 1, jpj
560            IF( tmask(2,jj,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
561               ssha_e(2:nbghostcells+1,jj) = hbdy_w(jj)
562            ENDIF
563         END DO
564      ENDIF
565      !
566      IF((nbondi == 1).OR.(nbondi == 2)) THEN
567         DO jj = 1, jpj
568            IF( tmask(nlci-1,jj,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
569               ssha_e(nlci-nbghostcells:nlci-1,jj) = hbdy_e(jj)
570            ENDIF
571         END DO
572      ENDIF
573      !
574      IF((nbondj == -1).OR.(nbondj == 2)) THEN
575         DO ji = 1, jpi
576            IF( tmask(ji,2,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
577               ssha_e(ji,2:nbghostcells+1) = hbdy_s(ji)
578            ENDIF
579         END DO
580      ENDIF
581      !
582      IF((nbondj == 1).OR.(nbondj == 2)) THEN
583         DO ji = 1, jpi
584            IF( tmask(ji,nlcj-1,1) == 1._wp ) THEN !clem: avoid interpolation if coast at the boundary
585               ssha_e(ji,nlcj-nbghostcells:nlcj-1) = hbdy_n(ji)
586            ENDIF
587         END DO
588      ENDIF
589      !
590   END SUBROUTINE Agrif_ssh_ts
591
592# if defined key_zdftke
593
594   SUBROUTINE Agrif_tke
595      !!----------------------------------------------------------------------
596      !!                  ***  ROUTINE Agrif_tke  ***
597      !!---------------------------------------------------------------------- 
598      REAL(wp) ::   zalpha
599      !!---------------------------------------------------------------------- 
600      !
601      zalpha = REAL( Agrif_NbStepint() + Agrif_IRhot() - 1, wp ) / REAL( Agrif_IRhot(), wp )
602      IF( zalpha > 1. )   zalpha = 1.
603      !
604      Agrif_SpecialValue    = 0.e0
605      Agrif_UseSpecialValue = .TRUE.
606      !
607      CALL Agrif_Bc_variable(avm_id ,calledweight=zalpha, procname=interpavm)       
608      !
609      Agrif_UseSpecialValue = .FALSE.
610      !
611   END SUBROUTINE Agrif_tke
612   
613# endif
614
615   SUBROUTINE interptsn( ptab, i1, i2, j1, j2, k1, k2, n1, n2, before, nb, ndir )
616      !!----------------------------------------------------------------------
617      !!   *** ROUTINE interptsn ***
618      !!----------------------------------------------------------------------
619      REAL(wp), DIMENSION(i1:i2,j1:j2,k1:k2,n1:n2), INTENT(inout) ::   ptab
620      INTEGER                                     , INTENT(in   ) ::   i1, i2, j1, j2, k1, k2, n1, n2
621      LOGICAL                                     , INTENT(in   ) ::   before
622      INTEGER                                     , INTENT(in   ) ::   nb , ndir
623      !
624      INTEGER  ::   ji, jj, jk, jn   ! dummy loop indices
625      INTEGER  ::   imin, imax, jmin, jmax
626      REAL(wp) ::   zrhox, z1, z2, z3, z4, z5, z6, z7
627      LOGICAL  ::   western_side, eastern_side,northern_side,southern_side
628      !!----------------------------------------------------------------------
629      !
630      IF (before) THEN         
631         ptab(i1:i2,j1:j2,k1:k2,n1:n2) = tsn(i1:i2,j1:j2,k1:k2,n1:n2)
632      ELSE
633         !
634         western_side  = (nb == 1).AND.(ndir == 1)  ;  eastern_side  = (nb == 1).AND.(ndir == 2)
635         southern_side = (nb == 2).AND.(ndir == 1)  ;  northern_side = (nb == 2).AND.(ndir == 2)
636         !
637         IF( nbghostcells > 1 ) THEN  ! no smoothing
638            tsa(i1:i2,j1:j2,k1:k2,n1:n2) = ptab(i1:i2,j1:j2,k1:k2,n1:n2)
639         ELSE                         ! smoothing
640            !
641            zrhox = Agrif_Rhox()
642            z1 = ( zrhox - 1. ) * 0.5
643            z3 = ( zrhox - 1. ) / ( zrhox + 1. )
644            z6 = 2. * ( zrhox - 1. ) / ( zrhox + 1. )
645            z7 =    - ( zrhox - 1. ) / ( zrhox + 3. )
646            !
647            z2 = 1. - z1
648            z4 = 1. - z3
649            z5 = 1. - z6 - z7
650            !
651            imin = i1 ; imax = i2
652            jmin = j1 ; jmax = j2
653            !
654            ! Remove CORNERS
655            IF((nbondj == -1).OR.(nbondj == 2)) jmin = 3
656            IF((nbondj == +1).OR.(nbondj == 2)) jmax = nlcj-2
657            IF((nbondi == -1).OR.(nbondi == 2)) imin = 3
658            IF((nbondi == +1).OR.(nbondi == 2)) imax = nlci-2       
659            !
660            IF( eastern_side ) THEN
661               DO jn = 1, jpts
662                  tsa(nlci,j1:j2,k1:k2,jn) = z1 * ptab(nlci,j1:j2,k1:k2,jn) + z2 * ptab(nlci-1,j1:j2,k1:k2,jn)
663                  DO jk = 1, jpkm1
664                     DO jj = jmin,jmax
665                        IF( umask(nlci-2,jj,jk) == 0._wp ) THEN
666                           tsa(nlci-1,jj,jk,jn) = tsa(nlci,jj,jk,jn) * tmask(nlci-1,jj,jk)
667                        ELSE
668                           tsa(nlci-1,jj,jk,jn)=(z4*tsa(nlci,jj,jk,jn)+z3*tsa(nlci-2,jj,jk,jn))*tmask(nlci-1,jj,jk)
669                           IF( un(nlci-2,jj,jk) > 0._wp ) THEN
670                              tsa(nlci-1,jj,jk,jn)=( z6*tsa(nlci-2,jj,jk,jn)+z5*tsa(nlci,jj,jk,jn) & 
671                                                   + z7*tsa(nlci-3,jj,jk,jn) ) * tmask(nlci-1,jj,jk)
672                           ENDIF
673                        ENDIF
674                     END DO
675                  END DO
676                  tsa(nlci,j1:j2,k1:k2,jn) = 0._wp
677               END DO
678            ENDIF
679            !
680            IF( northern_side ) THEN           
681               DO jn = 1, jpts
682                  tsa(i1:i2,nlcj,k1:k2,jn) = z1 * ptab(i1:i2,nlcj,k1:k2,jn) + z2 * ptab(i1:i2,nlcj-1,k1:k2,jn)
683                  DO jk = 1, jpkm1
684                     DO ji = imin,imax
685                        IF( vmask(ji,nlcj-2,jk) == 0._wp ) THEN
686                           tsa(ji,nlcj-1,jk,jn) = tsa(ji,nlcj,jk,jn) * tmask(ji,nlcj-1,jk)
687                        ELSE
688                           tsa(ji,nlcj-1,jk,jn)=(z4*tsa(ji,nlcj,jk,jn)+z3*tsa(ji,nlcj-2,jk,jn))*tmask(ji,nlcj-1,jk)       
689                           IF (vn(ji,nlcj-2,jk) > 0._wp ) THEN
690                              tsa(ji,nlcj-1,jk,jn)=( z6*tsa(ji,nlcj-2,jk,jn)+z5*tsa(ji,nlcj,jk,jn)  &
691                                                   + z7*tsa(ji,nlcj-3,jk,jn) ) * tmask(ji,nlcj-1,jk)
692                           ENDIF
693                        ENDIF
694                     END DO
695                  END DO
696                  tsa(i1:i2,nlcj,k1:k2,jn) = 0._wp
697               END DO
698            ENDIF
699            !
700            IF( western_side ) THEN           
701               DO jn = 1, jpts
702                  tsa(1,j1:j2,k1:k2,jn) = z1 * ptab(1,j1:j2,k1:k2,jn) + z2 * ptab(2,j1:j2,k1:k2,jn)
703                  DO jk = 1, jpkm1
704                     DO jj = jmin,jmax
705                        IF( umask(2,jj,jk) == 0._wp ) THEN
706                           tsa(2,jj,jk,jn) = tsa(1,jj,jk,jn) * tmask(2,jj,jk)
707                        ELSE
708                           tsa(2,jj,jk,jn)=(z4*tsa(1,jj,jk,jn)+z3*tsa(3,jj,jk,jn))*tmask(2,jj,jk)       
709                           IF( un(2,jj,jk) < 0._wp ) THEN
710                              tsa(2,jj,jk,jn)=(z6*tsa(3,jj,jk,jn)+z5*tsa(1,jj,jk,jn)+z7*tsa(4,jj,jk,jn))*tmask(2,jj,jk)
711                           ENDIF
712                        ENDIF
713                     END DO
714                  END DO
715                  tsa(1,j1:j2,k1:k2,jn) = 0._wp
716               END DO
717            ENDIF
718            !
719            IF( southern_side ) THEN           
720               DO jn = 1, jpts
721                  tsa(i1:i2,1,k1:k2,jn) = z1 * ptab(i1:i2,1,k1:k2,jn) + z2 * ptab(i1:i2,2,k1:k2,jn)
722                  DO jk = 1, jpk     
723                     DO ji=imin,imax
724                        IF( vmask(ji,2,jk) == 0._wp ) THEN
725                           tsa(ji,2,jk,jn)=tsa(ji,1,jk,jn) * tmask(ji,2,jk)
726                        ELSE
727                           tsa(ji,2,jk,jn)=(z4*tsa(ji,1,jk,jn)+z3*tsa(ji,3,jk,jn))*tmask(ji,2,jk)
728                           IF( vn(ji,2,jk) < 0._wp ) THEN
729                              tsa(ji,2,jk,jn)=(z6*tsa(ji,3,jk,jn)+z5*tsa(ji,1,jk,jn)+z7*tsa(ji,4,jk,jn))*tmask(ji,2,jk)
730                           ENDIF
731                        ENDIF
732                     END DO
733                  END DO
734                  tsa(i1:i2,1,k1:k2,jn) = 0._wp
735               END DO
736            ENDIF
737            !
738            ! Treatment of corners
739            IF ((eastern_side).AND.((nbondj == -1).OR.(nbondj == 2)))   tsa(nlci-1,2,:,:) = ptab(nlci-1,2,:,:)            ! East south
740            IF ((eastern_side).AND.((nbondj ==  1).OR.(nbondj == 2)))   tsa(nlci-1,nlcj-1,:,:) = ptab(nlci-1,nlcj-1,:,:)  ! East north
741            IF ((western_side).AND.((nbondj == -1).OR.(nbondj == 2)))   tsa(2,2,:,:) = ptab(2,2,:,:)                      ! West south
742            IF ((western_side).AND.((nbondj ==  1).OR.(nbondj == 2)))   tsa(2,nlcj-1,:,:) = ptab(2,nlcj-1,:,:)            ! West north
743            !
744         ENDIF
745      ENDIF
746      !
747   END SUBROUTINE interptsn
748
749
750   SUBROUTINE interpsshn( ptab, i1, i2, j1, j2, before, nb, ndir )
751      !!----------------------------------------------------------------------
752      !!                  ***  ROUTINE interpsshn  ***
753      !!---------------------------------------------------------------------- 
754      INTEGER                         , INTENT(in   ) ::   i1, i2, j1, j2
755      REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) ::   ptab
756      LOGICAL                         , INTENT(in   ) ::   before
757      INTEGER                         , INTENT(in   ) ::   nb , ndir
758      !
759      LOGICAL :: western_side, eastern_side,northern_side,southern_side
760      !!---------------------------------------------------------------------- 
761      !
762      IF( before) THEN
763         ptab(i1:i2,j1:j2) = sshn(i1:i2,j1:j2)
764      ELSE
765         western_side  = (nb == 1).AND.(ndir == 1)
766         eastern_side  = (nb == 1).AND.(ndir == 2)
767         southern_side = (nb == 2).AND.(ndir == 1)
768         northern_side = (nb == 2).AND.(ndir == 2)
769         !! clem ghost
770         IF(western_side)  hbdy_w(j1:j2) = ptab(i2,j1:j2) * tmask(i2,j1:j2,1)
771         IF(eastern_side)  hbdy_e(j1:j2) = ptab(i1,j1:j2) * tmask(i1,j1:j2,1) !clem previously i1
772         IF(southern_side) hbdy_s(i1:i2) = ptab(i1:i2,j2) * tmask(i1:i2,j2,1) !clem previously j1
773         IF(northern_side) hbdy_n(i1:i2) = ptab(i1:i2,j1) * tmask(i1:i2,j1,1)
774      ENDIF
775      !
776   END SUBROUTINE interpsshn
777
778
779   SUBROUTINE interpun( ptab, i1, i2, j1, j2, k1, k2, before )
780      !!----------------------------------------------------------------------
781      !!   *** ROUTINE interpun ***
782      !!----------------------------------------------------------------------
783      INTEGER                               , INTENT(in   ) ::   i1, i2, j1, j2, k1, k2
784      REAL(wp), DIMENSION(i1:i2,j1:j2,k1:k2), INTENT(inout) ::   ptab
785      LOGICAL                               , INTENT(in   ) ::   before
786      !
787      INTEGER  ::   ji, jj, jk
788      REAL(wp) ::   zrhoy 
789      !!----------------------------------------------------------------------
790      !
791      IF( before ) THEN
792         DO jk = k1, jpk
793            ptab(i1:i2,j1:j2,jk) = e2u(i1:i2,j1:j2) * e3u_n(i1:i2,j1:j2,jk) * un(i1:i2,j1:j2,jk)
794         END DO
795      ELSE
796         zrhoy = Agrif_Rhoy()
797         DO jk = 1, jpkm1
798            DO jj=j1,j2
799               ua(i1:i2,jj,jk) = ptab(i1:i2,jj,jk) / ( zrhoy * e2u(i1:i2,jj) * e3u_n(i1:i2,jj,jk) )
800            END DO
801         END DO
802      ENDIF
803      !
804   END SUBROUTINE interpun
805
806
807   SUBROUTINE interpvn( ptab, i1, i2, j1, j2, k1, k2, before )
808      !!----------------------------------------------------------------------
809      !!   *** ROUTINE interpvn ***
810      !!----------------------------------------------------------------------
811      INTEGER                               , INTENT(in   ) ::   i1, i2, j1, j2, k1, k2
812      REAL(wp), DIMENSION(i1:i2,j1:j2,k1:k2), INTENT(inout) ::   ptab
813      LOGICAL                               , INTENT(in   ) ::   before
814      !
815      INTEGER  ::   ji, jj, jk
816      REAL(wp) ::   zrhox 
817      !!----------------------------------------------------------------------
818      !     
819      IF( before ) THEN       !interpv entre 1 et k2 et interpv2d en jpkp1
820         DO jk = k1, jpk
821            ptab(i1:i2,j1:j2,jk) = e1v(i1:i2,j1:j2) * e3v_n(i1:i2,j1:j2,jk) * vn(i1:i2,j1:j2,jk)
822         END DO
823      ELSE         
824         zrhox= Agrif_Rhox()
825         DO jk = 1, jpkm1
826            va(i1:i2,j1:j2,jk) = ptab(i1:i2,j1:j2,jk) / ( zrhox * e1v(i1:i2,j1:j2) * e3v_n(i1:i2,j1:j2,jk) )
827         END DO
828      ENDIF
829      !       
830   END SUBROUTINE interpvn
831   
832
833   SUBROUTINE interpunb( ptab, i1, i2, j1, j2, before, nb, ndir )
834      !!----------------------------------------------------------------------
835      !!                  ***  ROUTINE interpunb  ***
836      !!---------------------------------------------------------------------- 
837      INTEGER                         , INTENT(in   ) ::   i1, i2, j1, j2
838      REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) ::   ptab
839      LOGICAL                         , INTENT(in   ) ::   before
840      INTEGER                         , INTENT(in   ) ::   nb , ndir
841      !
842      INTEGER  ::   ji, jj
843      REAL(wp) ::   zrhoy, zrhot, zt0, zt1, ztcoeff
844      LOGICAL  ::   western_side, eastern_side,northern_side,southern_side
845      !!---------------------------------------------------------------------- 
846      !
847      IF( before ) THEN
848         ptab(i1:i2,j1:j2) = e2u(i1:i2,j1:j2) * hu_n(i1:i2,j1:j2) * un_b(i1:i2,j1:j2)
849      ELSE
850         western_side  = (nb == 1).AND.(ndir == 1)
851         eastern_side  = (nb == 1).AND.(ndir == 2)
852         southern_side = (nb == 2).AND.(ndir == 1)
853         northern_side = (nb == 2).AND.(ndir == 2)
854         zrhoy = Agrif_Rhoy()
855         zrhot = Agrif_rhot()
856         ! Time indexes bounds for integration
857         zt0 = REAL(Agrif_NbStepint()  , wp) / zrhot
858         zt1 = REAL(Agrif_NbStepint()+1, wp) / zrhot     
859         ! Polynomial interpolation coefficients:
860         IF( bdy_tinterp == 1 ) THEN
861            ztcoeff = zrhot * (  zt1**2._wp * (       zt1 - 1._wp)        &
862               &               - zt0**2._wp * (       zt0 - 1._wp)        )
863         ELSEIF( bdy_tinterp == 2 ) THEN
864            ztcoeff = zrhot * (  zt1        * (       zt1 - 1._wp)**2._wp &
865               &               - zt0        * (       zt0 - 1._wp)**2._wp )
866         ELSE
867            ztcoeff = 1
868         ENDIF
869         !! clem ghost   
870         IF(western_side)   ubdy_w(j1:j2) = ubdy_w(j1:j2) + ztcoeff * ptab(i2,j1:j2) 
871         IF(eastern_side)   ubdy_e(j1:j2) = ubdy_e(j1:j2) + ztcoeff * ptab(i1,j1:j2) !clem previously i1 
872         IF(southern_side)  ubdy_s(i1:i2) = ubdy_s(i1:i2) + ztcoeff * ptab(i1:i2,j2) !clem previously j1
873         IF(northern_side)  ubdy_n(i1:i2) = ubdy_n(i1:i2) + ztcoeff * ptab(i1:i2,j1) 
874         !           
875         IF( bdy_tinterp == 0 .OR. bdy_tinterp == 2) THEN
876            IF(western_side)   ubdy_w(j1:j2) = ubdy_w(j1:j2) / (zrhoy*e2u(i2,j1:j2)) * umask(i2,j1:j2,1)
877            IF(eastern_side)   ubdy_e(j1:j2) = ubdy_e(j1:j2) / (zrhoy*e2u(i1,j1:j2)) * umask(i1,j1:j2,1)
878            IF(southern_side)  ubdy_s(i1:i2) = ubdy_s(i1:i2) / (zrhoy*e2u(i1:i2,j2)) * umask(i1:i2,j2,1)
879            IF(northern_side)  ubdy_n(i1:i2) = ubdy_n(i1:i2) / (zrhoy*e2u(i1:i2,j1)) * umask(i1:i2,j1,1)
880         ENDIF
881      ENDIF
882      !
883   END SUBROUTINE interpunb
884
885
886   SUBROUTINE interpvnb( ptab, i1, i2, j1, j2, before, nb, ndir )
887      !!----------------------------------------------------------------------
888      !!                  ***  ROUTINE interpvnb  ***
889      !!---------------------------------------------------------------------- 
890      INTEGER                         , INTENT(in   ) ::   i1, i2, j1, j2
891      REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) ::   ptab
892      LOGICAL                         , INTENT(in   ) ::   before
893      INTEGER                         , INTENT(in   ) ::   nb , ndir
894      !
895      INTEGER  ::   ji,jj
896      REAL(wp) ::   zrhox, zrhot, zt0, zt1, ztcoeff   
897      LOGICAL  ::   western_side, eastern_side,northern_side,southern_side
898      !!---------------------------------------------------------------------- 
899      !
900      IF( before ) THEN
901         ptab(i1:i2,j1:j2) = e1v(i1:i2,j1:j2) * hv_n(i1:i2,j1:j2) * vn_b(i1:i2,j1:j2)
902      ELSE
903         western_side  = (nb == 1).AND.(ndir == 1)
904         eastern_side  = (nb == 1).AND.(ndir == 2)
905         southern_side = (nb == 2).AND.(ndir == 1)
906         northern_side = (nb == 2).AND.(ndir == 2)
907         zrhox = Agrif_Rhox()
908         zrhot = Agrif_rhot()
909         ! Time indexes bounds for integration
910         zt0 = REAL(Agrif_NbStepint()  , wp) / zrhot
911         zt1 = REAL(Agrif_NbStepint()+1, wp) / zrhot     
912         IF( bdy_tinterp == 1 ) THEN
913            ztcoeff = zrhot * (  zt1**2._wp * (       zt1 - 1._wp)        &
914               &               - zt0**2._wp * (       zt0 - 1._wp)        )
915         ELSEIF( bdy_tinterp == 2 ) THEN
916            ztcoeff = zrhot * (  zt1        * (       zt1 - 1._wp)**2._wp &
917               &               - zt0        * (       zt0 - 1._wp)**2._wp ) 
918         ELSE
919            ztcoeff = 1
920         ENDIF
921         !! clem ghost
922         IF(western_side)   vbdy_w(j1:j2) = vbdy_w(j1:j2) + ztcoeff * ptab(i2,j1:j2) 
923         IF(eastern_side)   vbdy_e(j1:j2) = vbdy_e(j1:j2) + ztcoeff * ptab(i1,j1:j2) !clem previously i1 
924         IF(southern_side)  vbdy_s(i1:i2) = vbdy_s(i1:i2) + ztcoeff * ptab(i1:i2,j2) !clem previously j1
925         IF(northern_side)  vbdy_n(i1:i2) = vbdy_n(i1:i2) + ztcoeff * ptab(i1:i2,j1) 
926         !           
927         IF( bdy_tinterp == 0 .OR. bdy_tinterp == 2) THEN
928            IF(western_side)   vbdy_w(j1:j2) = vbdy_w(j1:j2) / (zrhox*e1v(i2,j1:j2)) * vmask(i2,j1:j2,1)
929            IF(eastern_side)   vbdy_e(j1:j2) = vbdy_e(j1:j2) / (zrhox*e1v(i1,j1:j2)) * vmask(i1,j1:j2,1)
930            IF(southern_side)  vbdy_s(i1:i2) = vbdy_s(i1:i2) / (zrhox*e1v(i1:i2,j2)) * vmask(i1:i2,j2,1)
931            IF(northern_side)  vbdy_n(i1:i2) = vbdy_n(i1:i2) / (zrhox*e1v(i1:i2,j1)) * vmask(i1:i2,j1,1)
932         ENDIF
933      ENDIF
934      !
935   END SUBROUTINE interpvnb
936
937
938   SUBROUTINE interpub2b( ptab, i1, i2, j1, j2, before, nb, ndir )
939      !!----------------------------------------------------------------------
940      !!                  ***  ROUTINE interpub2b  ***
941      !!---------------------------------------------------------------------- 
942      INTEGER                         , INTENT(in   ) ::   i1, i2, j1, j2
943      REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) ::   ptab
944      LOGICAL                         , INTENT(in   ) ::   before
945      INTEGER                         , INTENT(in   ) ::   nb , ndir
946      !
947      INTEGER  ::   ji,jj
948      REAL(wp) ::   zrhot, zt0, zt1,zat
949      LOGICAL  ::   western_side, eastern_side,northern_side,southern_side
950      !!---------------------------------------------------------------------- 
951      IF( before ) THEN
952         ptab(i1:i2,j1:j2) = e2u(i1:i2,j1:j2) * ub2_b(i1:i2,j1:j2)
953      ELSE
954         western_side  = (nb == 1).AND.(ndir == 1)
955         eastern_side  = (nb == 1).AND.(ndir == 2)
956         southern_side = (nb == 2).AND.(ndir == 1)
957         northern_side = (nb == 2).AND.(ndir == 2)
958         zrhot = Agrif_rhot()
959         ! Time indexes bounds for integration
960         zt0 = REAL(Agrif_NbStepint()  , wp) / zrhot
961         zt1 = REAL(Agrif_NbStepint()+1, wp) / zrhot
962         ! Polynomial interpolation coefficients:
963         zat = zrhot * (  zt1**2._wp * (-2._wp*zt1 + 3._wp)    &
964            &           - zt0**2._wp * (-2._wp*zt0 + 3._wp)    ) 
965         !! clem ghost
966         IF(western_side ) ubdy_w(j1:j2) = zat * ptab(i2,j1:j2) 
967         IF(eastern_side ) ubdy_e(j1:j2) = zat * ptab(i1,j1:j2) !clem previously i1 
968         IF(southern_side) ubdy_s(i1:i2) = zat * ptab(i1:i2,j2) !clem previously j1
969         IF(northern_side) ubdy_n(i1:i2) = zat * ptab(i1:i2,j1) 
970      ENDIF
971      !
972   END SUBROUTINE interpub2b
973   
974
975   SUBROUTINE interpvb2b( ptab, i1, i2, j1, j2, before, nb, ndir )
976      !!----------------------------------------------------------------------
977      !!                  ***  ROUTINE interpvb2b  ***
978      !!---------------------------------------------------------------------- 
979      INTEGER                         , INTENT(in   ) ::   i1, i2, j1, j2
980      REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) ::   ptab
981      LOGICAL                         , INTENT(in   ) ::   before
982      INTEGER                         , INTENT(in   ) ::   nb , ndir
983      !
984      INTEGER ::   ji,jj
985      REAL(wp) ::   zrhot, zt0, zt1,zat
986      LOGICAL ::   western_side, eastern_side,northern_side,southern_side
987      !!---------------------------------------------------------------------- 
988      !
989      IF( before ) THEN
990         ptab(i1:i2,j1:j2) = e1v(i1:i2,j1:j2) * vb2_b(i1:i2,j1:j2)
991      ELSE     
992         western_side  = (nb == 1).AND.(ndir == 1)
993         eastern_side  = (nb == 1).AND.(ndir == 2)
994         southern_side = (nb == 2).AND.(ndir == 1)
995         northern_side = (nb == 2).AND.(ndir == 2)
996         zrhot = Agrif_rhot()
997         ! Time indexes bounds for integration
998         zt0 = REAL(Agrif_NbStepint()  , wp) / zrhot
999         zt1 = REAL(Agrif_NbStepint()+1, wp) / zrhot
1000         ! Polynomial interpolation coefficients:
1001         zat = zrhot * (  zt1**2._wp * (-2._wp*zt1 + 3._wp)    &
1002            &           - zt0**2._wp * (-2._wp*zt0 + 3._wp)    ) 
1003         !
1004         IF(western_side )   vbdy_w(j1:j2) = zat * ptab(i2,j1:j2) 
1005         IF(eastern_side )   vbdy_e(j1:j2) = zat * ptab(i1,j1:j2) !clem previously i1 
1006         IF(southern_side)   vbdy_s(i1:i2) = zat * ptab(i1:i2,j2) !clem previously j1
1007         IF(northern_side)   vbdy_n(i1:i2) = zat * ptab(i1:i2,j1) 
1008      ENDIF
1009      !     
1010   END SUBROUTINE interpvb2b
1011
1012
1013   SUBROUTINE interpe3t( ptab, i1, i2, j1, j2, k1, k2, before, nb, ndir )
1014      !!----------------------------------------------------------------------
1015      !!                  ***  ROUTINE interpe3t  ***
1016      !!---------------------------------------------------------------------- 
1017      INTEGER                              , INTENT(in   ) :: i1, i2, j1, j2, k1, k2
1018      REAL(wp),DIMENSION(i1:i2,j1:j2,k1:k2), INTENT(inout) :: ptab
1019      LOGICAL                              , INTENT(in   ) :: before
1020      INTEGER                              , INTENT(in   ) :: nb , ndir
1021      !
1022      INTEGER :: ji, jj, jk
1023      LOGICAL :: western_side, eastern_side, northern_side, southern_side
1024      !!---------------------------------------------------------------------- 
1025      !   
1026      IF( before ) THEN
1027         ptab(i1:i2,j1:j2,k1:k2) = tmask(i1:i2,j1:j2,k1:k2) * e3t_0(i1:i2,j1:j2,k1:k2)
1028      ELSE
1029         western_side  = (nb == 1).AND.(ndir == 1)
1030         eastern_side  = (nb == 1).AND.(ndir == 2)
1031         southern_side = (nb == 2).AND.(ndir == 1)
1032         northern_side = (nb == 2).AND.(ndir == 2)
1033         !
1034         DO jk = k1, k2
1035            DO jj = j1, j2
1036               DO ji = i1, i2
1037                  !
1038                  IF( ABS( ptab(ji,jj,jk) - tmask(ji,jj,jk) * e3t_0(ji,jj,jk) ) > 1.D-2) THEN
1039                     IF (western_side) THEN
1040                        WRITE(numout,*) 'ERROR bathymetry merge at the western border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk
1041                     ELSEIF (eastern_side) THEN
1042                        WRITE(numout,*) 'ERROR bathymetry merge at the eastern border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk
1043                     ELSEIF (southern_side) THEN
1044                        WRITE(numout,*) 'ERROR bathymetry merge at the southern border ji,jj,jk', ji+nimpp-1,jj+njmpp-1,jk
1045                     ELSEIF (northern_side) THEN
1046                        WRITE(numout,*) 'ERROR bathymetry merge at the northen border ji,jj,jk', ji+nimpp-1,jj+njmpp-1,jk
1047                     ENDIF
1048                     WRITE(numout,*) '      ptab(ji,jj,jk), e3t(ji,jj,jk) ', ptab(ji,jj,jk), e3t_0(ji,jj,jk)
1049                     kindic_agr = kindic_agr + 1
1050                  ENDIF
1051               END DO
1052            END DO
1053         END DO
1054         !
1055      ENDIF
1056      !
1057   END SUBROUTINE interpe3t
1058
1059
1060   SUBROUTINE interpumsk( ptab, i1, i2, j1, j2, k1, k2, before, nb, ndir )
1061      !!----------------------------------------------------------------------
1062      !!                  ***  ROUTINE interpumsk  ***
1063      !!---------------------------------------------------------------------- 
1064      INTEGER                              , INTENT(in   ) ::   i1, i2, j1, j2, k1, k2
1065      REAL(wp),DIMENSION(i1:i2,j1:j2,k1:k2), INTENT(inout) ::   ptab
1066      LOGICAL                              , INTENT(in   ) ::   before
1067      INTEGER                              , INTENT(in   ) ::   nb , ndir
1068      !
1069      INTEGER ::   ji, jj, jk
1070      LOGICAL ::   western_side, eastern_side   
1071      !!---------------------------------------------------------------------- 
1072      !   
1073      IF( before ) THEN
1074         ptab(i1:i2,j1:j2,k1:k2) = umask(i1:i2,j1:j2,k1:k2)
1075      ELSE
1076         western_side = (nb == 1).AND.(ndir == 1)
1077         eastern_side = (nb == 1).AND.(ndir == 2)
1078         DO jk = k1, k2
1079            DO jj = j1, j2
1080               DO ji = i1, i2
1081                   ! Velocity mask at boundary edge points:
1082                  IF (ABS(ptab(ji,jj,jk) - umask(ji,jj,jk)) > 1.D-2) THEN
1083                     IF (western_side) THEN
1084                        WRITE(numout,*) 'ERROR with umask at the western border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk
1085                        WRITE(numout,*) '      masks: parent, child ', ptab(ji,jj,jk), umask(ji,jj,jk)
1086                        kindic_agr = kindic_agr + 1
1087                     ELSEIF (eastern_side) THEN
1088                        WRITE(numout,*) 'ERROR with umask at the eastern border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk
1089                        WRITE(numout,*) '      masks: parent, child ', ptab(ji,jj,jk), umask(ji,jj,jk)
1090                        kindic_agr = kindic_agr + 1
1091                     ENDIF
1092                  ENDIF
1093               END DO
1094            END DO
1095         END DO
1096         !
1097      ENDIF
1098      !
1099   END SUBROUTINE interpumsk
1100
1101
1102   SUBROUTINE interpvmsk( ptab, i1, i2, j1, j2, k1, k2, before, nb, ndir )
1103      !!----------------------------------------------------------------------
1104      !!                  ***  ROUTINE interpvmsk  ***
1105      !!---------------------------------------------------------------------- 
1106      INTEGER                              , INTENT(in   ) ::   i1,i2,j1,j2,k1,k2
1107      REAL(wp),DIMENSION(i1:i2,j1:j2,k1:k2), INTENT(inout) ::   ptab
1108      LOGICAL                              , INTENT(in   ) ::   before
1109      INTEGER                              , INTENT(in   ) :: nb , ndir
1110      !
1111      INTEGER ::   ji, jj, jk
1112      LOGICAL ::   northern_side, southern_side     
1113      !!---------------------------------------------------------------------- 
1114      !   
1115      IF( before ) THEN
1116         ptab(i1:i2,j1:j2,k1:k2) = vmask(i1:i2,j1:j2,k1:k2)
1117      ELSE
1118         southern_side = (nb == 2).AND.(ndir == 1)
1119         northern_side = (nb == 2).AND.(ndir == 2)
1120         DO jk = k1, k2
1121            DO jj = j1, j2
1122               DO ji = i1, i2
1123                   ! Velocity mask at boundary edge points:
1124                  IF (ABS(ptab(ji,jj,jk) - vmask(ji,jj,jk)) > 1.D-2) THEN
1125                     IF (southern_side) THEN
1126                        WRITE(numout,*) 'ERROR with vmask at the southern border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk
1127                        WRITE(numout,*) '      masks: parent, child ', ptab(ji,jj,jk), vmask(ji,jj,jk)
1128                        kindic_agr = kindic_agr + 1
1129                     ELSEIF (northern_side) THEN
1130                        WRITE(numout,*) 'ERROR with vmask at the northern border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk
1131                        WRITE(numout,*) '      masks: parent, child ', ptab(ji,jj,jk), vmask(ji,jj,jk)
1132                        kindic_agr = kindic_agr + 1
1133                     ENDIF
1134                  ENDIF
1135               END DO
1136            END DO
1137         END DO
1138         !
1139      ENDIF
1140      !
1141   END SUBROUTINE interpvmsk
1142
1143# if defined key_zdftke
1144
1145   SUBROUTINE interpavm( ptab, i1, i2, j1, j2, k1, k2, before )
1146      !!----------------------------------------------------------------------
1147      !!                  ***  ROUTINE interavm  ***
1148      !!---------------------------------------------------------------------- 
1149      INTEGER                              , INTENT(in   ) ::   i1, i2, j1, j2, k1, k2
1150      REAL(wp),DIMENSION(i1:i2,j1:j2,k1:k2), INTENT(inout) ::   ptab
1151      LOGICAL                              , INTENT(in   ) ::   before
1152      !!---------------------------------------------------------------------- 
1153      !     
1154      IF( before ) THEN
1155         ptab (i1:i2,j1:j2,k1:k2) = avm_k(i1:i2,j1:j2,k1:k2)
1156      ELSE
1157         avm_k(i1:i2,j1:j2,k1:k2) = ptab (i1:i2,j1:j2,k1:k2)
1158      ENDIF
1159      !
1160   END SUBROUTINE interpavm
1161
1162# endif /* key_zdftke */
1163
1164#else
1165   !!----------------------------------------------------------------------
1166   !!   Empty module                                          no AGRIF zoom
1167   !!----------------------------------------------------------------------
1168CONTAINS
1169   SUBROUTINE Agrif_OPA_Interp_empty
1170      WRITE(*,*)  'agrif_opa_interp : You should not have seen this print! error?'
1171   END SUBROUTINE Agrif_OPA_Interp_empty
1172#endif
1173
1174   !!======================================================================
1175END MODULE agrif_opa_interp
Note: See TracBrowser for help on using the repository browser.