MODULE agrif_top_interp #if defined key_agrif && defined key_top USE par_oce USE oce USE dom_oce USE agrif_oce USE agrif_top_sponge USE par_trc USE trc USE lib_mpp USE wrk_nemo IMPLICIT NONE PRIVATE PUBLIC Agrif_trc, interptrn # include "vectopt_loop_substitute.h90" !!---------------------------------------------------------------------- !! NEMO/NST 3.6 , NEMO Consortium (2010) !! $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.e0 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 REAL(wp) :: zrhox, z1, z2, z3, z4, z5, z6, z7 LOGICAL :: western_side, eastern_side,northern_side,southern_side !!----------------------------------------------------------------------- ! 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 ! western_side = (nb == 1).AND.(ndir == 1) ; eastern_side = (nb == 1).AND.(ndir == 2) southern_side = (nb == 2).AND.(ndir == 1) ; northern_side = (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( eastern_side) THEN 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.e0 ) 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.e0 ) 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( northern_side ) THEN 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.e0 ) 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.e0 ) 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 ENDDO ENDIF ! IF( western_side) THEN 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.e0 ) 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.e0 ) 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( southern_side ) THEN 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.e0 ) 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.e0 ) 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 ENDDO ENDIF ! ! Treatment of corners IF ((eastern_side).AND.((nbondj == -1).OR.(nbondj == 2))) tra(nlci-1,2,:,:) = ptab(nlci-1,2,:,:) ! East south IF ((eastern_side).AND.((nbondj == 1).OR.(nbondj == 2))) tra(nlci-1,nlcj-1,:,:) = ptab(nlci-1,nlcj-1,:,:) ! East north IF ((western_side).AND.((nbondj == -1).OR.(nbondj == 2))) tra(2,2,:,:) = ptab(2,2,:,:) ! West south IF ((western_side).AND.((nbondj == 1).OR.(nbondj == 2))) tra(2,nlcj-1,:,:) = ptab(2,nlcj-1,:,:) ! West north ! ENDIF ENDIF ! END SUBROUTINE interptrn #else 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