MODULE agrif_top_interp !!====================================================================== !! *** MODULE agrif_top_interp *** !! AGRIF: interpolation package for TOP !!====================================================================== !! History : 2.0 ! ??? !!---------------------------------------------------------------------- #if defined key_agrif && defined key_top !!---------------------------------------------------------------------- !! 'key_agrif' AGRIF zoom !! 'key_top' on-line tracers !!---------------------------------------------------------------------- USE par_oce USE oce USE dom_oce USE agrif_oce USE agrif_top_sponge USE par_trc USE trc ! USE lib_mpp ! MPP library IMPLICIT NONE PRIVATE PUBLIC Agrif_trc, interptrn !!---------------------------------------------------------------------- !! NEMO/NST 4.0 , NEMO Consortium (2017) !! $Id$ !! Software governed by the CeCILL licence (NEMOGCM/NEMO_CeCILL.txt) !!---------------------------------------------------------------------- CONTAINS SUBROUTINE Agrif_trc !!---------------------------------------------------------------------- !! *** ROUTINE Agrif_trc *** !!---------------------------------------------------------------------- ! IF( Agrif_Root() ) RETURN ! Agrif_SpecialValue = 0._wp Agrif_UseSpecialValue = .TRUE. ! CALL Agrif_Bc_variable( trn_id, procname=interptrn ) Agrif_UseSpecialValue = .FALSE. ! END SUBROUTINE Agrif_trc SUBROUTINE interptrn( ptab, i1, i2, j1, j2, k1, k2, n1, n2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interptrn *** !!---------------------------------------------------------------------- REAL(wp), DIMENSION(i1:i2,j1:j2,k1:k2,n1:n2), INTENT(inout) :: ptab INTEGER , INTENT(in ) :: i1, i2, j1, j2, k1, k2, n1, n2 LOGICAL , INTENT(in ) :: before INTEGER , INTENT(in ) :: nb , ndir !! INTEGER :: ji, jj, jk, jn ! dummy loop indices INTEGER :: imin, imax, jmin, jmax LOGICAL :: ll_west, ll_east, ll_north, ll_south REAL(wp) :: zrhox, z1, z2, z3, z4, z5, z6, z7 !!---------------------------------------------------------------------- ! IF( before ) THEN ptab(i1:i2,j1:j2,k1:k2,n1:n2) = trn(i1:i2,j1:j2,k1:k2,n1:n2) ELSE ! IF( nbghostcells > 1 ) THEN ! no smoothing tra(i1:i2,j1:j2,k1:k2,n1:n2) = ptab(i1:i2,j1:j2,k1:k2,n1:n2) ELSE ! smoothing ! ll_west = (nb == 1).AND.(ndir == 1) ; ll_east = (nb == 1).AND.(ndir == 2) ll_south = (nb == 2).AND.(ndir == 1) ; ll_north = (nb == 2).AND.(ndir == 2) ! zrhox = Agrif_Rhox() z1 = ( zrhox - 1. ) * 0.5 z3 = ( zrhox - 1. ) / ( zrhox + 1. ) z6 = 2. * ( zrhox - 1. ) / ( zrhox + 1. ) z7 = - ( zrhox - 1. ) / ( zrhox + 3. ) ! z2 = 1. - z1 z4 = 1. - z3 z5 = 1. - z6 - z7 ! imin = i1 ; imax = i2 jmin = j1 ; jmax = j2 ! ! Remove CORNERS IF((nbondj == -1).OR.(nbondj == 2)) jmin = 3 IF((nbondj == +1).OR.(nbondj == 2)) jmax = nlcj-2 IF((nbondi == -1).OR.(nbondi == 2)) imin = 3 IF((nbondi == +1).OR.(nbondi == 2)) imax = nlci-2 ! IF( ll_east ) THEN !== eastern side ==! DO jn = 1, jptra tra(nlci,j1:j2,k1:k2,jn) = z1 * ptab(nlci,j1:j2,k1:k2,jn) + z2 * ptab(nlci-1,j1:j2,k1:k2,jn) DO jk = 1, jpkm1 DO jj = jmin,jmax IF( umask(nlci-2,jj,jk) == 0._wp ) THEN tra(nlci-1,jj,jk,jn) = tra(nlci,jj,jk,jn) * tmask(nlci-1,jj,jk) ELSE tra(nlci-1,jj,jk,jn) = ( z4*tra(nlci,jj,jk,jn)+z3*tra(nlci-2,jj,jk,jn) ) * tmask(nlci-1,jj,jk) IF( un(nlci-2,jj,jk) > 0._wp ) THEN tra(nlci-1,jj,jk,jn) = ( z6*tra(nlci-2,jj,jk,jn)+z5*tra(nlci,jj,jk,jn) & & +z7*tra(nlci-3,jj,jk,jn) ) * tmask(nlci-1,jj,jk) ENDIF ENDIF END DO END DO ENDDO ENDIF ! IF( ll_north ) THEN !== northern side ==! DO jn = 1, jptra tra(i1:i2,nlcj,k1:k2,jn) = z1 * ptab(i1:i2,nlcj,k1:k2,jn) + z2 * ptab(i1:i2,nlcj-1,k1:k2,jn) DO jk = 1, jpkm1 DO ji = imin, imax IF( vmask(ji,nlcj-2,jk) == 0._wp ) THEN tra(ji,nlcj-1,jk,jn) = tra(ji,nlcj,jk,jn) * tmask(ji,nlcj-1,jk) ELSE tra(ji,nlcj-1,jk,jn) = ( z4*tra(ji,nlcj,jk,jn)+z3*tra(ji,nlcj-2,jk,jn) ) * tmask(ji,nlcj-1,jk) IF (vn(ji,nlcj-2,jk) > 0._wp ) THEN tra(ji,nlcj-1,jk,jn) = ( z6*tra(ji,nlcj-2,jk,jn)+z5*tra(ji,nlcj,jk,jn) & & +z7*tra(ji,nlcj-3,jk,jn) ) * tmask(ji,nlcj-1,jk) ENDIF ENDIF END DO END DO END DO ENDIF ! IF( ll_west ) THEN !== western side ==! DO jn = 1, jptra tra(1,j1:j2,k1:k2,jn) = z1 * ptab(1,j1:j2,k1:k2,jn) + z2 * ptab(2,j1:j2,k1:k2,jn) DO jk = 1, jpkm1 DO jj = jmin,jmax IF( umask(2,jj,jk) == 0._wp ) THEN tra(2,jj,jk,jn) = tra(1,jj,jk,jn) * tmask(2,jj,jk) ELSE tra(2,jj,jk,jn) = ( z4*tra(1,jj,jk,jn)+z3*tra(3,jj,jk,jn) ) * tmask(2,jj,jk) IF( un(2,jj,jk) < 0._wp ) THEN tra(2,jj,jk,jn) = ( z6*tra(3,jj,jk,jn)+z5*tra(1,jj,jk,jn)+z7*tra(4,jj,jk,jn) ) * tmask(2,jj,jk) ENDIF ENDIF END DO END DO END DO ENDIF ! IF( ll_south ) THEN !== southern side ==! DO jn = 1, jptra tra(i1:i2,1,k1:k2,jn) = z1 * ptab(i1:i2,1,k1:k2,jn) + z2 * ptab(i1:i2,2,k1:k2,jn) DO jk = 1, jpk DO ji = imin, imax IF( vmask(ji,2,jk) == 0._wp ) THEN tra(ji,2,jk,jn) = tra(ji,1,jk,jn) * tmask(ji,2,jk) ELSE tra(ji,2,jk,jn) = ( z4*tra(ji,1,jk,jn)+z3*tra(ji,3,jk,jn) ) * tmask(ji,2,jk) IF( vn(ji,2,jk) < 0._wp ) THEN tra(ji,2,jk,jn) = ( z6*tra(ji,3,jk,jn)+z5*tra(ji,1,jk,jn)+z7*tra(ji,4,jk,jn) ) * tmask(ji,2,jk) ENDIF ENDIF END DO END DO END DO ENDIF ! ! Treatment of corners IF( ll_east .AND.((nbondj == -1).OR.(nbondj == 2)) ) tra(nlci-1, 2 ,:,:) = ptab(nlci-1, 2 ,:,:) ! East south IF( ll_east .AND.((nbondj == 1).OR.(nbondj == 2)) ) tra(nlci-1,nlcj-1,:,:) = ptab(nlci-1,nlcj-1,:,:) ! East north IF( ll_west .AND.((nbondj == -1).OR.(nbondj == 2)) ) tra( 2 , 2 ,:,:) = ptab( 2 , 2 ,:,:) ! West south IF( ll_west .AND.((nbondj == 1).OR.(nbondj == 2)) ) tra( 2 ,nlcj-1,:,:) = ptab( 2 ,nlcj-1,:,:) ! West north ! ENDIF ENDIF ! END SUBROUTINE interptrn #else !!---------------------------------------------------------------------- !! Empty module no TOP AGRIF !!---------------------------------------------------------------------- CONTAINS SUBROUTINE Agrif_TOP_Interp_empty !!--------------------------------------------- !! *** ROUTINE agrif_Top_Interp_empty *** !!--------------------------------------------- WRITE(*,*) 'agrif_top_interp : You should not have seen this print! error?' END SUBROUTINE Agrif_TOP_Interp_empty #endif !!====================================================================== END MODULE agrif_top_interp