source: branches/2017/dev_r7881_HPC09_ZDF/NEMOGCM/NEMO/NST_SRC/agrif_opa_interp.F90 @ 7953

Last change on this file since 7953 was 7953, checked in by gm, 3 years ago

#1880 (HPC-09): add zdfphy (the ZDF manager) + remove all key_…

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