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_r7881_ENHANCE09_RK3/NEMOGCM/NEMO/NST_SRC – NEMO

source: branches/2017/dev_r7881_ENHANCE09_RK3/NEMOGCM/NEMO/NST_SRC/agrif_opa_interp.F90 @ 8863

Last change on this file since 8863 was 8863, checked in by flavoni, 6 years ago

(ENHANCE-09): fix AGRIF reproducibility problem

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