MODULE agrif_opa_interp !!====================================================================== !! *** MODULE agrif_opa_interp *** !! AGRIF: interpolation package !!====================================================================== !! History : 2.0 ! 2002-06 (XXX) Original cade !! - ! 2005-11 (XXX) !! 3.2 ! 2009-04 (R. Benshila) !! 3.6 ! 2014-09 (R. Benshila) !!---------------------------------------------------------------------- #if defined key_agrif !!---------------------------------------------------------------------- !! 'key_agrif' AGRIF zoom !!---------------------------------------------------------------------- !! Agrif_tra : !! Agrif_dyn : !! interpu : !! interpv : !!---------------------------------------------------------------------- USE par_oce USE oce USE dom_oce USE zdf_oce USE agrif_oce USE phycst ! USE in_out_manager USE agrif_opa_sponge USE lib_mpp USE wrk_nemo IMPLICIT NONE PRIVATE PUBLIC Agrif_tra, Agrif_dyn, Agrif_ssh, Agrif_dyn_ts, Agrif_ssh_ts, Agrif_dta_ts PUBLIC interpun, interpvn PUBLIC interptsn, interpsshn PUBLIC interpunb, interpvnb, interpub2b, interpvb2b PUBLIC interpe3t, interpumsk, interpvmsk # if defined key_zdftke PUBLIC Agrif_tke, interpavm # endif INTEGER :: bdy_tinterp = 0 # include "vectopt_loop_substitute.h90" !!---------------------------------------------------------------------- !! NEMO/NST 3.7 , NEMO Consortium (2015) !! $Id$ !! Software governed by the CeCILL licence (NEMOGCM/NEMO_CeCILL.txt) !!---------------------------------------------------------------------- CONTAINS SUBROUTINE Agrif_tra !!---------------------------------------------------------------------- !! *** ROUTINE Agrif_tra *** !!---------------------------------------------------------------------- ! IF( Agrif_Root() ) RETURN ! Agrif_SpecialValue = 0._wp Agrif_UseSpecialValue = .TRUE. ! CALL Agrif_Bc_variable( tsn_id, procname=interptsn ) ! Agrif_UseSpecialValue = .FALSE. ! END SUBROUTINE Agrif_tra SUBROUTINE Agrif_dyn( kt ) !!---------------------------------------------------------------------- !! *** ROUTINE Agrif_DYN *** !!---------------------------------------------------------------------- INTEGER, INTENT(in) :: kt ! INTEGER :: ji, jj, jk ! dummy loop indices INTEGER :: j1, j2, i1, i2 REAL(wp), POINTER, DIMENSION(:,:) :: zub, zvb !!---------------------------------------------------------------------- ! IF( Agrif_Root() ) RETURN ! CALL wrk_alloc( jpi,jpj, zub, zvb ) ! Agrif_SpecialValue = 0._wp Agrif_UseSpecialValue = ln_spc_dyn ! CALL Agrif_Bc_variable( un_interp_id, procname=interpun ) CALL Agrif_Bc_variable( vn_interp_id, procname=interpvn ) ! Agrif_UseSpecialValue = .FALSE. ! ! prevent smoothing in ghost cells i1 = 1 ; i2 = jpi j1 = 1 ; j2 = jpj IF( nbondj == -1 .OR. nbondj == 2 ) j1 = 3 IF( nbondj == +1 .OR. nbondj == 2 ) j2 = nlcj-2 IF( nbondi == -1 .OR. nbondi == 2 ) i1 = 3 IF( nbondi == +1 .OR. nbondi == 2 ) i2 = nlci-2 IF( nbondi == -1 .OR. nbondi == 2 ) THEN ! ! Smoothing ! --------- IF( .NOT.ln_dynspg_ts ) THEN ! Store transport ua_b(2,:) = 0._wp DO jk = 1, jpkm1 DO jj = 1, jpj ua_b(2,jj) = ua_b(2,jj) + e3u_a(2,jj,jk) * ua(2,jj,jk) END DO END DO DO jj = 1, jpj ua_b(2,jj) = ua_b(2,jj) * r1_hu_a(2,jj) END DO ENDIF ! DO jk=1,jpkm1 ! Smooth DO jj=j1,j2 ua(2,jj,jk) = 0.25_wp*(ua(1,jj,jk)+2._wp*ua(2,jj,jk)+ua(3,jj,jk)) ua(2,jj,jk) = ua(2,jj,jk) * umask(2,jj,jk) END DO END DO ! zub(2,:) = 0._wp ! Correct transport DO jk = 1, jpkm1 DO jj = 1, jpj zub(2,jj) = zub(2,jj) + e3u_a(2,jj,jk) * ua(2,jj,jk) END DO END DO DO jj=1,jpj zub(2,jj) = zub(2,jj) * r1_hu_a(2,jj) END DO DO jk=1,jpkm1 DO jj=1,jpj ua(2,jj,jk) = (ua(2,jj,jk)+ua_b(2,jj)-zub(2,jj))*umask(2,jj,jk) END DO END DO ! Set tangential velocities to time splitting estimate !----------------------------------------------------- IF( ln_dynspg_ts ) THEN zvb(2,:) = 0._wp DO jk = 1, jpkm1 DO jj = 1, jpj zvb(2,jj) = zvb(2,jj) + e3v_a(2,jj,jk) * va(2,jj,jk) END DO END DO DO jj = 1, jpj zvb(2,jj) = zvb(2,jj) * r1_hv_a(2,jj) END DO DO jk = 1, jpkm1 DO jj = 1, jpj va(2,jj,jk) = (va(2,jj,jk)+va_b(2,jj)-zvb(2,jj)) * vmask(2,jj,jk) END DO END DO ENDIF ! ! Mask domain edges: !------------------- DO jk = 1, jpkm1 DO jj = 1, jpj ua(1,jj,jk) = 0._wp va(1,jj,jk) = 0._wp END DO END DO ! ENDIF IF( nbondi == 1 .OR. nbondi == 2 ) THEN ! Smoothing ! --------- IF( .NOT.ln_dynspg_ts ) THEN ! Store transport ua_b(nlci-2,:) = 0._wp DO jk=1,jpkm1 DO jj=1,jpj ua_b(nlci-2,jj) = ua_b(nlci-2,jj) + e3u_a(nlci-2,jj,jk) * ua(nlci-2,jj,jk) END DO END DO DO jj=1,jpj ua_b(nlci-2,jj) = ua_b(nlci-2,jj) * r1_hu_a(nlci-2,jj) END DO ENDIF DO jk = 1, jpkm1 ! Smooth DO jj = j1, j2 ua(nlci-2,jj,jk) = 0.25_wp * umask(nlci-2,jj,jk) & & * ( ua(nlci-3,jj,jk) + 2._wp*ua(nlci-2,jj,jk) + ua(nlci-1,jj,jk) ) END DO END DO zub(nlci-2,:) = 0._wp ! Correct transport DO jk = 1, jpkm1 DO jj = 1, jpj zub(nlci-2,jj) = zub(nlci-2,jj) + e3u_a(nlci-2,jj,jk) * ua(nlci-2,jj,jk) END DO END DO DO jj = 1, jpj zub(nlci-2,jj) = zub(nlci-2,jj) * r1_hu_a(nlci-2,jj) END DO DO jk = 1, jpkm1 DO jj = 1, jpj ua(nlci-2,jj,jk) = ( ua(nlci-2,jj,jk) + ua_b(nlci-2,jj) - zub(nlci-2,jj) ) * umask(nlci-2,jj,jk) END DO END DO ! ! Set tangential velocities to time splitting estimate !----------------------------------------------------- IF( ln_dynspg_ts ) THEN zvb(nlci-1,:) = 0._wp DO jk = 1, jpkm1 DO jj = 1, jpj zvb(nlci-1,jj) = zvb(nlci-1,jj) + e3v_a(nlci-1,jj,jk) * va(nlci-1,jj,jk) END DO END DO DO jj=1,jpj zvb(nlci-1,jj) = zvb(nlci-1,jj) * r1_hv_a(nlci-1,jj) END DO DO jk = 1, jpkm1 DO jj = 1, jpj va(nlci-1,jj,jk) = ( va(nlci-1,jj,jk) + va_b(nlci-1,jj) - zvb(nlci-1,jj) ) * vmask(nlci-1,jj,jk) END DO END DO ENDIF ! ! Mask domain edges: !------------------- DO jk = 1, jpkm1 DO jj = 1, jpj ua(nlci-1,jj,jk) = 0._wp va(nlci ,jj,jk) = 0._wp END DO END DO ! ENDIF IF( nbondj == -1 .OR. nbondj == 2 ) THEN ! Smoothing ! --------- IF( .NOT.ln_dynspg_ts ) THEN ! Store transport va_b(:,2) = 0._wp DO jk = 1, jpkm1 DO ji = 1, jpi va_b(ji,2) = va_b(ji,2) + e3v_a(ji,2,jk) * va(ji,2,jk) END DO END DO DO ji=1,jpi va_b(ji,2) = va_b(ji,2) * r1_hv_a(ji,2) END DO ENDIF ! DO jk = 1, jpkm1 ! Smooth DO ji = i1, i2 va(ji,2,jk) = 0.25_wp * vmask(ji,2,jk) & & * ( va(ji,1,jk) + 2._wp*va(ji,2,jk) + va(ji,3,jk) ) END DO END DO ! zvb(:,2) = 0._wp ! Correct transport DO jk=1,jpkm1 DO ji=1,jpi zvb(ji,2) = zvb(ji,2) + e3v_a(ji,2,jk) * va(ji,2,jk) * vmask(ji,2,jk) END DO END DO DO ji = 1, jpi zvb(ji,2) = zvb(ji,2) * r1_hv_a(ji,2) END DO DO jk = 1, jpkm1 DO ji = 1, jpi va(ji,2,jk) = ( va(ji,2,jk) + va_b(ji,2) - zvb(ji,2) ) * vmask(ji,2,jk) END DO END DO ! Set tangential velocities to time splitting estimate !----------------------------------------------------- IF( ln_dynspg_ts ) THEN zub(:,2) = 0._wp DO jk = 1, jpkm1 DO ji = 1, jpi zub(ji,2) = zub(ji,2) + e3u_a(ji,2,jk) * ua(ji,2,jk) * umask(ji,2,jk) END DO END DO DO ji = 1, jpi zub(ji,2) = zub(ji,2) * r1_hu_a(ji,2) END DO DO jk = 1, jpkm1 DO ji = 1, jpi ua(ji,2,jk) = ( ua(ji,2,jk) + ua_b(ji,2) - zub(ji,2) ) * umask(ji,2,jk) END DO END DO ENDIF ! Mask domain edges: !------------------- DO jk = 1, jpkm1 DO ji = 1, jpi ua(ji,1,jk) = 0._wp va(ji,1,jk) = 0._wp END DO END DO ENDIF IF( nbondj == 1 .OR. nbondj == 2 ) THEN ! ! Smoothing ! --------- IF( .NOT.ln_dynspg_ts ) THEN ! Store transport va_b(:,nlcj-2) = 0._wp DO jk = 1, jpkm1 DO ji = 1, jpi va_b(ji,nlcj-2) = va_b(ji,nlcj-2) + e3v_a(ji,nlcj-2,jk) * va(ji,nlcj-2,jk) END DO END DO DO ji = 1, jpi va_b(ji,nlcj-2) = va_b(ji,nlcj-2) * r1_hv_a(ji,nlcj-2) END DO ENDIF ! DO jk = 1, jpkm1 ! Smooth DO ji = i1, i2 va(ji,nlcj-2,jk) = 0.25_wp * vmask(ji,nlcj-2,jk) & & * ( va(ji,nlcj-3,jk) + 2._wp * va(ji,nlcj-2,jk) + va(ji,nlcj-1,jk) ) END DO END DO ! zvb(:,nlcj-2) = 0._wp ! Correct transport DO jk = 1, jpkm1 DO ji = 1, jpi zvb(ji,nlcj-2) = zvb(ji,nlcj-2) + e3v_a(ji,nlcj-2,jk) * va(ji,nlcj-2,jk) * vmask(ji,nlcj-2,jk) END DO END DO DO ji = 1, jpi zvb(ji,nlcj-2) = zvb(ji,nlcj-2) * r1_hv_a(ji,nlcj-2) END DO DO jk = 1, jpkm1 DO ji = 1, jpi va(ji,nlcj-2,jk) = ( va(ji,nlcj-2,jk) + va_b(ji,nlcj-2) - zvb(ji,nlcj-2) ) * vmask(ji,nlcj-2,jk) END DO END DO ! ! Set tangential velocities to time splitting estimate !----------------------------------------------------- IF( ln_dynspg_ts ) THEN zub(:,nlcj-1) = 0._wp DO jk = 1, jpkm1 DO ji = 1, jpi zub(ji,nlcj-1) = zub(ji,nlcj-1) + e3u_a(ji,nlcj-1,jk) * ua(ji,nlcj-1,jk) * umask(ji,nlcj-1,jk) END DO END DO DO ji = 1, jpi zub(ji,nlcj-1) = zub(ji,nlcj-1) * r1_hu_a(ji,nlcj-1) END DO ! DO jk = 1, jpkm1 DO ji = 1, jpi ua(ji,nlcj-1,jk) = ( ua(ji,nlcj-1,jk) + ua_b(ji,nlcj-1) - zub(ji,nlcj-1) ) * umask(ji,nlcj-1,jk) END DO END DO ENDIF ! ! Mask domain edges: !------------------- DO jk = 1, jpkm1 DO ji = 1, jpi ua(ji,nlcj ,jk) = 0._wp va(ji,nlcj-1,jk) = 0._wp END DO END DO ! ENDIF ! CALL wrk_dealloc( jpi,jpj, zub, zvb ) ! END SUBROUTINE Agrif_dyn SUBROUTINE Agrif_dyn_ts( jn ) !!---------------------------------------------------------------------- !! *** ROUTINE Agrif_dyn_ts *** !!---------------------------------------------------------------------- !! INTEGER, INTENT(in) :: jn !! INTEGER :: ji, jj !!---------------------------------------------------------------------- ! IF( Agrif_Root() ) RETURN ! IF((nbondi == -1).OR.(nbondi == 2)) THEN DO jj=1,jpj va_e(2,jj) = vbdy_w(jj) * hvr_e(2,jj) ! Specified fluxes: ua_e(2,jj) = ubdy_w(jj) * hur_e(2,jj) ! Characteristics method: !alt ua_e(2,jj) = 0.5_wp * ( ubdy_w(jj) * hur_e(2,jj) + ua_e(3,jj) & !alt & - sqrt(grav * hur_e(2,jj)) * (sshn_e(3,jj) - hbdy_w(jj)) ) END DO ENDIF ! IF((nbondi == 1).OR.(nbondi == 2)) THEN DO jj=1,jpj va_e(nlci-1,jj) = vbdy_e(jj) * hvr_e(nlci-1,jj) ! Specified fluxes: ua_e(nlci-2,jj) = ubdy_e(jj) * hur_e(nlci-2,jj) ! Characteristics method: !alt ua_e(nlci-2,jj) = 0.5_wp * ( ubdy_e(jj) * hur_e(nlci-2,jj) + ua_e(nlci-3,jj) & !alt & + sqrt(grav * hur_e(nlci-2,jj)) * (sshn_e(nlci-2,jj) - hbdy_e(jj)) ) END DO ENDIF ! IF((nbondj == -1).OR.(nbondj == 2)) THEN DO ji=1,jpi ua_e(ji,2) = ubdy_s(ji) * hur_e(ji,2) ! Specified fluxes: va_e(ji,2) = vbdy_s(ji) * hvr_e(ji,2) ! Characteristics method: !alt va_e(ji,2) = 0.5_wp * ( vbdy_s(ji) * hvr_e(ji,2) + va_e(ji,3) & !alt & - sqrt(grav * hvr_e(ji,2)) * (sshn_e(ji,3) - hbdy_s(ji)) ) END DO ENDIF ! IF((nbondj == 1).OR.(nbondj == 2)) THEN DO ji=1,jpi ua_e(ji,nlcj-1) = ubdy_n(ji) * hur_e(ji,nlcj-1) ! Specified fluxes: va_e(ji,nlcj-2) = vbdy_n(ji) * hvr_e(ji,nlcj-2) ! Characteristics method: !alt va_e(ji,nlcj-2) = 0.5_wp * ( vbdy_n(ji) * hvr_e(ji,nlcj-2) + va_e(ji,nlcj-3) & !alt & + sqrt(grav * hvr_e(ji,nlcj-2)) * (sshn_e(ji,nlcj-2) - hbdy_n(ji)) ) END DO ENDIF ! END SUBROUTINE Agrif_dyn_ts SUBROUTINE Agrif_dta_ts( kt ) !!---------------------------------------------------------------------- !! *** ROUTINE Agrif_dta_ts *** !!---------------------------------------------------------------------- !! INTEGER, INTENT(in) :: kt !! INTEGER :: ji, jj LOGICAL :: ll_int_cons REAL(wp) :: zrhot, zt !!---------------------------------------------------------------------- ! IF( Agrif_Root() ) RETURN ! ll_int_cons = ln_bt_fw ! Assume conservative temporal integration in the forward case only ! zrhot = Agrif_rhot() ! ! "Central" time index for interpolation: IF( ln_bt_fw ) THEN zt = REAL( Agrif_NbStepint()+0.5_wp, wp ) / zrhot ELSE zt = REAL( Agrif_NbStepint() , wp ) / zrhot ENDIF ! ! Linear interpolation of sea level Agrif_SpecialValue = 0._wp Agrif_UseSpecialValue = .TRUE. CALL Agrif_Bc_variable( sshn_id, calledweight=zt, procname=interpsshn ) Agrif_UseSpecialValue = .FALSE. ! ! Interpolate barotropic fluxes Agrif_SpecialValue=0. Agrif_UseSpecialValue = ln_spc_dyn ! IF( ll_int_cons ) THEN ! Conservative interpolation ! orders matters here !!!!!! CALL Agrif_Bc_variable( ub2b_interp_id, calledweight=1._wp, procname=interpub2b ) ! Time integrated CALL Agrif_Bc_variable( vb2b_interp_id, calledweight=1._wp, procname=interpvb2b ) bdy_tinterp = 1 CALL Agrif_Bc_variable( unb_id , calledweight=1._wp, procname=interpunb ) ! After CALL Agrif_Bc_variable( vnb_id , calledweight=1._wp, procname=interpvnb ) bdy_tinterp = 2 CALL Agrif_Bc_variable( unb_id , calledweight=0._wp, procname=interpunb ) ! Before CALL Agrif_Bc_variable( vnb_id , calledweight=0._wp, procname=interpvnb ) ELSE ! Linear interpolation bdy_tinterp = 0 ubdy_w(:) = 0._wp ; vbdy_w(:) = 0._wp ubdy_e(:) = 0._wp ; vbdy_e(:) = 0._wp ubdy_n(:) = 0._wp ; vbdy_n(:) = 0._wp ubdy_s(:) = 0._wp ; vbdy_s(:) = 0._wp CALL Agrif_Bc_variable( unb_id, calledweight=zt, procname=interpunb ) CALL Agrif_Bc_variable( vnb_id, calledweight=zt, procname=interpvnb ) ENDIF Agrif_UseSpecialValue = .FALSE. ! END SUBROUTINE Agrif_dta_ts SUBROUTINE Agrif_ssh( kt ) !!---------------------------------------------------------------------- !! *** ROUTINE Agrif_DYN *** !!---------------------------------------------------------------------- INTEGER, INTENT(in) :: kt !! !!---------------------------------------------------------------------- ! IF( Agrif_Root() ) RETURN ! IF((nbondi == -1).OR.(nbondi == 2)) THEN ssha(2,:)=ssha(3,:) sshn(2,:)=sshn(3,:) ENDIF ! IF((nbondi == 1).OR.(nbondi == 2)) THEN ssha(nlci-1,:)=ssha(nlci-2,:) sshn(nlci-1,:)=sshn(nlci-2,:) ENDIF ! IF((nbondj == -1).OR.(nbondj == 2)) THEN ssha(:,2)=ssha(:,3) sshn(:,2)=sshn(:,3) ENDIF ! IF((nbondj == 1).OR.(nbondj == 2)) THEN ssha(:,nlcj-1)=ssha(:,nlcj-2) sshn(:,nlcj-1)=sshn(:,nlcj-2) ENDIF ! END SUBROUTINE Agrif_ssh SUBROUTINE Agrif_ssh_ts( jn ) !!---------------------------------------------------------------------- !! *** ROUTINE Agrif_ssh_ts *** !!---------------------------------------------------------------------- INTEGER, INTENT(in) :: jn !! INTEGER :: ji,jj !!---------------------------------------------------------------------- ! IF((nbondi == -1).OR.(nbondi == 2)) THEN DO jj = 1, jpj ssha_e(2,jj) = hbdy_w(jj) END DO ENDIF ! IF((nbondi == 1).OR.(nbondi == 2)) THEN DO jj = 1, jpj ssha_e(nlci-1,jj) = hbdy_e(jj) END DO ENDIF ! IF((nbondj == -1).OR.(nbondj == 2)) THEN DO ji = 1, jpi ssha_e(ji,2) = hbdy_s(ji) END DO ENDIF ! IF((nbondj == 1).OR.(nbondj == 2)) THEN DO ji = 1, jpi ssha_e(ji,nlcj-1) = hbdy_n(ji) END DO ENDIF ! END SUBROUTINE Agrif_ssh_ts # if defined key_zdftke SUBROUTINE Agrif_tke !!---------------------------------------------------------------------- !! *** ROUTINE Agrif_tke *** !!---------------------------------------------------------------------- REAL(wp) :: zalpha !!---------------------------------------------------------------------- ! zalpha = REAL( Agrif_NbStepint() + Agrif_IRhot() - 1, wp ) / REAL( Agrif_IRhot(), wp ) IF( zalpha > 1. ) zalpha = 1. ! Agrif_SpecialValue = 0.e0 Agrif_UseSpecialValue = .TRUE. ! CALL Agrif_Bc_variable(avm_id ,calledweight=zalpha, procname=interpavm) ! Agrif_UseSpecialValue = .FALSE. ! END SUBROUTINE Agrif_tke # endif SUBROUTINE interptsn( ptab, i1, i2, j1, j2, k1, k2, n1, n2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interptsn *** !!---------------------------------------------------------------------- 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, iref, jref ! dummy loop indices INTEGER :: imin, imax, jmin, jmax, N_in, N_out REAL(wp) :: zrhox , zalpha1, zalpha2, zalpha3 REAL(wp) :: zalpha4, zalpha5, zalpha6, zalpha7 LOGICAL :: western_side, eastern_side,northern_side,southern_side REAL(wp), DIMENSION(i1:i2,j1:j2,1:jpk,n1:n2) :: ptab_child REAL(wp), DIMENSION(k1:k2,n1:n2-1) :: tabin REAL(wp) :: h_in(k1:k2) REAL(wp) :: h_out(1:jpk) REAL(wp) :: h_diff, zrhoxy zrhoxy = Agrif_rhox()*Agrif_rhoy() IF (before) THEN DO jn = n1,n2-1 DO jk=k1,k2 DO jj=j1,j2 DO ji=i1,i2 ptab(ji,jj,jk,jn) = tsn(ji,jj,jk,jn) * e1e2t(ji,jj) * e3t_n(ji,jj,jk) END DO END DO END DO END DO DO jk=k1,k2 DO jj=j1,j2 DO ji=i1,i2 ptab(ji,jj,jk,n2) = tmask(ji,jj,jk) * e1e2t(ji,jj) * e3t_n(ji,jj,jk) END DO END DO END DO ELSE 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) #ifdef key_vertical ptab_child(:,:,:,:) = 0. DO jj=j1,j2 DO ji=i1,i2 iref = ji jref = jj if(western_side) iref=2 if(eastern_side) iref=nlci-1 if(southern_side) jref=2 if(northern_side) jref=nlcj-1 N_in = 0 DO jk=k1,k2 !k2 = jpk of parent grid IF (ptab(ji,jj,jk,n2) == 0) EXIT N_in = N_in + 1 tabin(jk,:) = ptab(ji,jj,jk,n1:n2-1)/(ptab(ji,jj,jk,n2)) h_in(N_in) = ptab(ji,jj,jk,n2)/(e1e2t(ji,jj)*zrhoxy) END DO N_out = 0 DO jk=1,jpk ! jpk of child grid IF (tmask(iref,jref,jk) == 0) EXIT N_out = N_out + 1 h_out(jk) = e3t_n(iref,jref,jk) ENDDO IF (N_in > 0) THEN h_diff = sum(h_out(1:N_out))-sum(h_in(1:N_in)) DO jn=1,jpts call reconstructandremap(tabin(1:N_in,jn),h_in,ptab_child(ji,jj,1:N_out,jn),h_out,N_in,N_out) ENDDO ENDIF ENDDO ENDDO #else DO jk=k1,k2 DO jj=j1,j2 DO ji=i1,i2 ptab_child(ji,jj,jk,n1:n2-1) = ptab(ji,jj,jk,n1:n2-1)/(zrhoxy*e1e2t(ji,jj)*e3t_n(ji,jj,jk) ENDDO ENDDO ENDDO #endif ! zrhox = Agrif_Rhox() ! zalpha1 = ( zrhox - 1. ) * 0.5 zalpha2 = 1. - zalpha1 ! zalpha3 = ( zrhox - 1. ) / ( zrhox + 1. ) zalpha4 = 1. - zalpha3 ! zalpha6 = 2. * ( zrhox - 1. ) / ( zrhox + 1. ) zalpha7 = - ( zrhox - 1. ) / ( zrhox + 3. ) zalpha5 = 1. - zalpha6 - zalpha7 ! 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, jpts tsa(nlci,j1:j2,1:jpk,jn) = zalpha1 * ptab_child(nlci,j1:j2,1:jpk,jn) + zalpha2 * ptab_child(nlci-1,j1:j2,1:jpk,jn) DO jk = 1, jpkm1 DO jj = jmin,jmax IF( umask(nlci-2,jj,jk) == 0._wp ) THEN tsa(nlci-1,jj,jk,jn) = tsa(nlci,jj,jk,jn) * tmask(nlci-1,jj,jk) ELSE tsa(nlci-1,jj,jk,jn)=(zalpha4*tsa(nlci,jj,jk,jn)+zalpha3*tsa(nlci-2,jj,jk,jn))*tmask(nlci-1,jj,jk) IF( un(nlci-2,jj,jk) > 0._wp ) THEN tsa(nlci-1,jj,jk,jn)=( zalpha6*tsa(nlci-2,jj,jk,jn)+zalpha5*tsa(nlci,jj,jk,jn) & + zalpha7*tsa(nlci-3,jj,jk,jn) ) * tmask(nlci-1,jj,jk) ENDIF ENDIF END DO END DO tsa(nlci,j1:j2,k1:k2,jn) = 0._wp END DO ENDIF ! IF( northern_side ) THEN DO jn = 1, jpts tsa(i1:i2,nlcj,1:jpk,jn) = zalpha1 * ptab_child(i1:i2,nlcj,1:jpk,jn) + zalpha2 * ptab_child(i1:i2,nlcj-1,1:jpk,jn) DO jk = 1, jpkm1 DO ji = imin,imax IF( vmask(ji,nlcj-2,jk) == 0._wp ) THEN tsa(ji,nlcj-1,jk,jn) = tsa(ji,nlcj,jk,jn) * tmask(ji,nlcj-1,jk) ELSE tsa(ji,nlcj-1,jk,jn)=(zalpha4*tsa(ji,nlcj,jk,jn)+zalpha3*tsa(ji,nlcj-2,jk,jn))*tmask(ji,nlcj-1,jk) IF (vn(ji,nlcj-2,jk) > 0._wp ) THEN tsa(ji,nlcj-1,jk,jn)=( zalpha6*tsa(ji,nlcj-2,jk,jn)+zalpha5*tsa(ji,nlcj,jk,jn) & + zalpha7*tsa(ji,nlcj-3,jk,jn) ) * tmask(ji,nlcj-1,jk) ENDIF ENDIF END DO END DO tsa(i1:i2,nlcj,k1:k2,jn) = 0._wp END DO ENDIF ! IF( western_side ) THEN DO jn = 1, jpts tsa(1,j1:j2,1:jpk,jn) = zalpha1 * ptab_child(1,j1:j2,1:jpk,jn) + zalpha2 * ptab_child(2,j1:j2,1:jpk,jn) DO jk = 1, jpkm1 DO jj = jmin,jmax IF( umask(2,jj,jk) == 0._wp ) THEN tsa(2,jj,jk,jn) = tsa(1,jj,jk,jn) * tmask(2,jj,jk) ELSE tsa(2,jj,jk,jn)=(zalpha4*tsa(1,jj,jk,jn)+zalpha3*tsa(3,jj,jk,jn))*tmask(2,jj,jk) IF( un(2,jj,jk) < 0._wp ) THEN 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) ENDIF ENDIF END DO END DO tsa(1,j1:j2,k1:k2,jn) = 0._wp END DO ENDIF ! IF( southern_side ) THEN DO jn = 1, jpts tsa(i1:i2,1,1:jpk,jn) = zalpha1 * ptab_child(i1:i2,1,1:jpk,jn) + zalpha2 * ptab_child(i1:i2,2,1:jpk,jn) DO jk = 1, jpk DO ji=imin,imax IF( vmask(ji,2,jk) == 0._wp ) THEN tsa(ji,2,jk,jn)=tsa(ji,1,jk,jn) * tmask(ji,2,jk) ELSE tsa(ji,2,jk,jn)=(zalpha4*tsa(ji,1,jk,jn)+zalpha3*tsa(ji,3,jk,jn))*tmask(ji,2,jk) IF( vn(ji,2,jk) < 0._wp ) THEN 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) ENDIF ENDIF END DO END DO tsa(i1:i2,1,k1:k2,jn) = 0._wp END DO ENDIF ! ! Treatment of corners ! ! East south IF ((eastern_side).AND.((nbondj == -1).OR.(nbondj == 2))) THEN tsa(nlci-1,2,:,:) = ptab_child(nlci-1,2,:,1:jpts) ENDIF ! East north IF ((eastern_side).AND.((nbondj == 1).OR.(nbondj == 2))) THEN tsa(nlci-1,nlcj-1,:,:) = ptab_child(nlci-1,nlcj-1,:,1:jpts) ENDIF ! West south IF ((western_side).AND.((nbondj == -1).OR.(nbondj == 2))) THEN tsa(2,2,:,:) = ptab_child(2,2,:,1:jpts) ENDIF ! West north IF ((western_side).AND.((nbondj == 1).OR.(nbondj == 2))) THEN tsa(2,nlcj-1,:,:) = ptab_child(2,nlcj-1,:,1:jpts) ENDIF ! ENDIF ! END SUBROUTINE interptsn SUBROUTINE interpsshn( ptab, i1, i2, j1, j2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpsshn *** !!---------------------------------------------------------------------- INTEGER , INTENT(in ) :: i1, i2, j1, j2 REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) :: ptab LOGICAL , INTENT(in ) :: before INTEGER , INTENT(in ) :: nb , ndir ! LOGICAL :: western_side, eastern_side,northern_side,southern_side !!---------------------------------------------------------------------- ! IF( before) THEN ptab(i1:i2,j1:j2) = sshn(i1:i2,j1:j2) ELSE 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) IF(western_side) hbdy_w(j1:j2) = ptab(i1,j1:j2) * tmask(i1,j1:j2,1) IF(eastern_side) hbdy_e(j1:j2) = ptab(i1,j1:j2) * tmask(i1,j1:j2,1) IF(southern_side) hbdy_s(i1:i2) = ptab(i1:i2,j1) * tmask(i1:i2,j1,1) IF(northern_side) hbdy_n(i1:i2) = ptab(i1:i2,j1) * tmask(i1:i2,j1,1) ENDIF ! END SUBROUTINE interpsshn SUBROUTINE interpun( ptab, i1, i2, j1, j2, k1, k2, m1, m2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpun *** !!--------------------------------------------- !! INTEGER, INTENT(in) :: i1,i2,j1,j2,k1,k2,m1,m2 REAL(wp), DIMENSION(i1:i2,j1:j2,k1:k2,m1:m2), INTENT(inout) :: ptab LOGICAL, INTENT(in) :: before INTEGER, INTENT(in) :: nb , ndir !! INTEGER :: ji,jj,jk REAL(wp) :: zrhoy ! VERTICAL REFINEMENT BEGIN REAL(wp), DIMENSION(i1:i2,j1:j2,1:jpk) :: ptab_child REAL(wp), DIMENSION(k1:k2) :: tabin REAL(wp) :: h_in(k1:k2) REAL(wp) :: h_out(1:jpk) INTEGER :: N_in, N_out REAL(wp) :: h_diff LOGICAL :: western_side, eastern_side INTEGER :: iref ! VERTICAL REFINEMENT END !!--------------------------------------------- ! zrhoy = Agrif_rhoy() IF (before) THEN !We can't use zero as the special value because we need to include zeros !when interpolating the scale factors IF(Agrif_UseSpecialValue) THEN ! Agrif_SpecialValue = -999._wp Agrif_SpecialValue = 0._wp ELSE Agrif_SpecialValue = 0._wp ENDIF DO jk=1,jpk DO jj=j1,j2 DO ji=i1,i2 ptab(ji,jj,jk,1) = (e2u(ji,jj) * e3u_n(ji,jj,jk) * un(ji,jj,jk)*umask(ji,jj,jk)) - & & ((umask(ji,jj,jk)-1) * Agrif_SpecialValue) ptab(ji,jj,jk,2) = (umask(ji,jj,jk) * e2u(ji,jj) * e3u_n(ji,jj,jk)) END DO END DO END DO ELSE ! VERTICAL REFINEMENT BEGIN western_side = (nb == 1).AND.(ndir == 1) eastern_side = (nb == 1).AND.(ndir == 2) Agrif_SpecialValue = 0._wp ! reset specialvalue to zero now interpolation completed ptab_child(:,:,:) = 0. #ifdef key_vertical ! VERTICAL REFINEMENT BEGIN DO jj=j1,j2 DO ji=i1,i2 iref = ji IF (western_side) iref = 2 IF (eastern_side) iref = nlci-2 N_in = 0 DO jk=k1,k2 IF (ptab(ji,jj,jk,2) == 0) EXIT N_in = N_in + 1 tabin(jk) = ptab(ji,jj,jk,1)/ptab(ji,jj,jk,2) h_in(N_in) = ptab(ji,jj,jk,2)/(e2u(ji,jj)*zrhoy) ENDDO IF (N_in == 0) THEN ptab_child(ji,jj,:) = 0. CYCLE ENDIF N_out = 0 DO jk=1,jpk if (umask(ji,jj,jk) == 0) EXIT N_out = N_out + 1 h_out(N_out) = e3u_n(ji,jj,jk) ENDDO IF (N_out == 0) THEN ptab_child(ji,jj,:) = 0. CYCLE ENDIF IF (N_in * N_out > 0) THEN h_diff = sum(h_out(1:N_out))-sum(h_in(1:N_in)) ! Should be able to remove the next IF/ELSEIF statement once scale factors are dealt with properly if (h_diff < -1.e4) then print *,'CHECK YOUR BATHY ...', h_diff, sum(h_out(1:N_out)), sum(h_in(1:N_in)) ! stop endif ENDIF call reconstructandremap(tabin(1:N_in),h_in(1:N_in),ptab_child(ji,jj,1:N_out),h_out(1:N_out),N_in,N_out) ENDDO ENDDO ! VERTICAL REFINEMENT END DO jk = 1, jpkm1 DO jj=j1,j2 ua(i1:i2,jj,jk) = ptab_child(i1:i2,jj,jk) END DO END DO #else DO jk = 1, jpkm1 DO jj=j1,j2 ua(i1:i2,jj,jk) = umask(i1:i2,jj,jk) * ptab(i1:i2,jj,jk,1) / ( zrhoy * e2u(i1:i2,jj) * e3u_n(i1:i2,jj,jk) ) END DO END DO #endif ENDIF ! END SUBROUTINE interpun SUBROUTINE interpvn( ptab, i1, i2, j1, j2, k1, k2, m1, m2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpvn *** !!---------------------------------------------------------------------- ! INTEGER, INTENT(in) :: i1,i2,j1,j2,k1,k2,m1,m2 REAL(wp), DIMENSION(i1:i2,j1:j2,k1:k2,m1:m2), INTENT(inout) :: ptab LOGICAL, INTENT(in) :: before INTEGER, INTENT(in) :: nb , ndir ! INTEGER :: ji,jj,jk REAL(wp) :: zrhox REAL(wp), DIMENSION(i1:i2,j1:j2,1:jpk) :: ptab_child REAL(wp), DIMENSION(k1:k2) :: tabin REAL(wp) :: h_in(k1:k2) REAL(wp) :: h_out(1:jpk) INTEGER :: N_in, N_out REAL(wp) :: h_diff LOGICAL :: northern_side,southern_side INTEGER :: jref !!--------------------------------------------- ! zrhox = Agrif_rhox() IF (before) THEN IF(Agrif_UseSpecialValue) THEN ! Agrif_SpecialValue = -999._wp Agrif_SpecialValue = 0._wp ELSE Agrif_SpecialValue = 0._wp ENDIF DO jk=k1,k2 DO jj=j1,j2 DO ji=i1,i2 ptab(ji,jj,jk,1) = e1v(ji,jj) * e3v_n(ji,jj,jk) * vn(ji,jj,jk) ptab(ji,jj,jk,2) = vmask(ji,jj,jk) * e1v(ji,jj) * e3v_n(ji,jj,jk) END DO END DO END DO ELSE Agrif_SpecialValue = 0._wp !Reset special value to zero now interpolation is done ptab_child(:,:,:) = 0. #ifdef key_vertical southern_side = (nb == 2).AND.(ndir == 1) northern_side = (nb == 2).AND.(ndir == 2) do jj=j1,j2 jref = jj IF (southern_side) jref = 2 IF (northern_side) jref = nlcj-2 do ji=i1,i2 N_in = 0 do jk=k1,k2 if (ptab(ji,jj,jk,2) == 0) EXIT N_in = N_in + 1 tabin(jk) = ptab(ji,jj,jk,1)/ptab(ji,jj,jk,2) h_in(N_in) = ptab(ji,jj,jk,2)/(e1v(ji,jj)*zrhox) enddo IF (N_in == 0) THEN ptab_child(ji,jj,:) = 0. CYCLE ENDIF N_out = 0 do jk=1,jpk if (vmask(ji,jref,jk) == 0) EXIT N_out = N_out + 1 h_out(N_out) = e3v_n(ji,jj,jk) enddo IF (N_out == 0) THEN ptab_child(ji,jj,:) = 0. CYCLE ENDIF call reconstructandremap(tabin(1:N_in),h_in(1:N_in),ptab_child(ji,jj,1:N_out),h_out(1:N_out),N_in,N_out) enddo enddo DO jk=1,jpkm1 DO jj=j1,j2 va(i1:i2,jj,jk) = ptab_child(i1:i2,jj,jk) END DO END DO #else DO jk=1,jpkm1 va(i1:i2,j1:j2,jk) = vmask(i1:i2,j1:j2,jk) * ptab(i1:i2,j1:j2,jk,1) / & & ( zrhox * e1v(i1:i2,j1:j2) * e3v_n(i1:i2,j1:j2,jk) ) END DO #endif ENDIF ! END SUBROUTINE interpvn SUBROUTINE interpunb( ptab, i1, i2, j1, j2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpunb *** !!---------------------------------------------------------------------- INTEGER , INTENT(in ) :: i1, i2, j1, j2 REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) :: ptab LOGICAL , INTENT(in ) :: before INTEGER , INTENT(in ) :: nb , ndir ! INTEGER :: ji, jj REAL(wp) :: zrhoy, zrhot, zt0, zt1, ztcoeff LOGICAL :: western_side, eastern_side,northern_side,southern_side !!---------------------------------------------------------------------- ! IF( before ) THEN ptab(i1:i2,j1:j2) = e2u(i1:i2,j1:j2) * hu_n(i1:i2,j1:j2) * un_b(i1:i2,j1:j2) ELSE 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) zrhoy = Agrif_Rhoy() zrhot = Agrif_rhot() ! Time indexes bounds for integration zt0 = REAL(Agrif_NbStepint() , wp) / zrhot zt1 = REAL(Agrif_NbStepint()+1, wp) / zrhot ! Polynomial interpolation coefficients: IF( bdy_tinterp == 1 ) THEN ztcoeff = zrhot * ( zt1**2._wp * ( zt1 - 1._wp) & & - zt0**2._wp * ( zt0 - 1._wp) ) ELSEIF( bdy_tinterp == 2 ) THEN ztcoeff = zrhot * ( zt1 * ( zt1 - 1._wp)**2._wp & & - zt0 * ( zt0 - 1._wp)**2._wp ) ELSE ztcoeff = 1 ENDIF ! IF(western_side) THEN ubdy_w(j1:j2) = ubdy_w(j1:j2) + ztcoeff * ptab(i1,j1:j2) ENDIF IF(eastern_side) THEN ubdy_e(j1:j2) = ubdy_e(j1:j2) + ztcoeff * ptab(i1,j1:j2) ENDIF IF(southern_side) THEN ubdy_s(i1:i2) = ubdy_s(i1:i2) + ztcoeff * ptab(i1:i2,j1) ENDIF IF(northern_side) THEN ubdy_n(i1:i2) = ubdy_n(i1:i2) + ztcoeff * ptab(i1:i2,j1) ENDIF ! IF( bdy_tinterp == 0 .OR. bdy_tinterp == 2) THEN IF(western_side) THEN ubdy_w(j1:j2) = ubdy_w(j1:j2) / (zrhoy*e2u(i1,j1:j2)) * umask(i1,j1:j2,1) ENDIF IF(eastern_side) THEN ubdy_e(j1:j2) = ubdy_e(j1:j2) / (zrhoy*e2u(i1,j1:j2)) * umask(i1,j1:j2,1) ENDIF IF(southern_side) THEN ubdy_s(i1:i2) = ubdy_s(i1:i2) / (zrhoy*e2u(i1:i2,j1)) * umask(i1:i2,j1,1) ENDIF IF(northern_side) THEN ubdy_n(i1:i2) = ubdy_n(i1:i2) / (zrhoy*e2u(i1:i2,j1)) * umask(i1:i2,j1,1) ENDIF ENDIF ENDIF ! END SUBROUTINE interpunb SUBROUTINE interpvnb( ptab, i1, i2, j1, j2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpvnb *** !!---------------------------------------------------------------------- INTEGER , INTENT(in ) :: i1, i2, j1, j2 REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) :: ptab LOGICAL , INTENT(in ) :: before INTEGER , INTENT(in ) :: nb , ndir ! INTEGER :: ji,jj REAL(wp) :: zrhox, zrhot, zt0, zt1, ztcoeff LOGICAL :: western_side, eastern_side,northern_side,southern_side !!---------------------------------------------------------------------- ! IF( before ) THEN ptab(i1:i2,j1:j2) = e1v(i1:i2,j1:j2) * hv_n(i1:i2,j1:j2) * vn_b(i1:i2,j1:j2) ELSE 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() zrhot = Agrif_rhot() ! Time indexes bounds for integration zt0 = REAL(Agrif_NbStepint() , wp) / zrhot zt1 = REAL(Agrif_NbStepint()+1, wp) / zrhot IF( bdy_tinterp == 1 ) THEN ztcoeff = zrhot * ( zt1**2._wp * ( zt1 - 1._wp) & & - zt0**2._wp * ( zt0 - 1._wp) ) ELSEIF( bdy_tinterp == 2 ) THEN ztcoeff = zrhot * ( zt1 * ( zt1 - 1._wp)**2._wp & & - zt0 * ( zt0 - 1._wp)**2._wp ) ELSE ztcoeff = 1 ENDIF ! IF(western_side) THEN vbdy_w(j1:j2) = vbdy_w(j1:j2) + ztcoeff * ptab(i1,j1:j2) ENDIF IF(eastern_side) THEN vbdy_e(j1:j2) = vbdy_e(j1:j2) + ztcoeff * ptab(i1,j1:j2) ENDIF IF(southern_side) THEN vbdy_s(i1:i2) = vbdy_s(i1:i2) + ztcoeff * ptab(i1:i2,j1) ENDIF IF(northern_side) THEN vbdy_n(i1:i2) = vbdy_n(i1:i2) + ztcoeff * ptab(i1:i2,j1) ENDIF ! IF( bdy_tinterp == 0 .OR. bdy_tinterp == 2) THEN IF(western_side) THEN vbdy_w(j1:j2) = vbdy_w(j1:j2) / (zrhox*e1v(i1,j1:j2)) & & * vmask(i1,j1:j2,1) ENDIF IF(eastern_side) THEN vbdy_e(j1:j2) = vbdy_e(j1:j2) / (zrhox*e1v(i1,j1:j2)) & & * vmask(i1,j1:j2,1) ENDIF IF(southern_side) THEN vbdy_s(i1:i2) = vbdy_s(i1:i2) / (zrhox*e1v(i1:i2,j1)) & & * vmask(i1:i2,j1,1) ENDIF IF(northern_side) THEN vbdy_n(i1:i2) = vbdy_n(i1:i2) / (zrhox*e1v(i1:i2,j1)) & & * vmask(i1:i2,j1,1) ENDIF ENDIF ENDIF ! END SUBROUTINE interpvnb SUBROUTINE interpub2b( ptab, i1, i2, j1, j2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpub2b *** !!---------------------------------------------------------------------- INTEGER , INTENT(in ) :: i1, i2, j1, j2 REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) :: ptab LOGICAL , INTENT(in ) :: before INTEGER , INTENT(in ) :: nb , ndir ! INTEGER :: ji,jj REAL(wp) :: zrhot, zt0, zt1,zat LOGICAL :: western_side, eastern_side,northern_side,southern_side !!---------------------------------------------------------------------- IF( before ) THEN ptab(i1:i2,j1:j2) = e2u(i1:i2,j1:j2) * ub2_b(i1:i2,j1:j2) ELSE 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) zrhot = Agrif_rhot() ! Time indexes bounds for integration zt0 = REAL(Agrif_NbStepint() , wp) / zrhot zt1 = REAL(Agrif_NbStepint()+1, wp) / zrhot ! Polynomial interpolation coefficients: zat = zrhot * ( zt1**2._wp * (-2._wp*zt1 + 3._wp) & & - zt0**2._wp * (-2._wp*zt0 + 3._wp) ) ! IF(western_side ) ubdy_w(j1:j2) = zat * ptab(i1,j1:j2) IF(eastern_side ) ubdy_e(j1:j2) = zat * ptab(i1,j1:j2) IF(southern_side) ubdy_s(i1:i2) = zat * ptab(i1:i2,j1) IF(northern_side) ubdy_n(i1:i2) = zat * ptab(i1:i2,j1) ENDIF ! END SUBROUTINE interpub2b SUBROUTINE interpvb2b( ptab, i1, i2, j1, j2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpvb2b *** !!---------------------------------------------------------------------- INTEGER , INTENT(in ) :: i1, i2, j1, j2 REAL(wp), DIMENSION(i1:i2,j1:j2), INTENT(inout) :: ptab LOGICAL , INTENT(in ) :: before INTEGER , INTENT(in ) :: nb , ndir ! INTEGER :: ji,jj REAL(wp) :: zrhot, zt0, zt1,zat LOGICAL :: western_side, eastern_side,northern_side,southern_side !!---------------------------------------------------------------------- ! IF( before ) THEN ptab(i1:i2,j1:j2) = e1v(i1:i2,j1:j2) * vb2_b(i1:i2,j1:j2) ELSE 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) zrhot = Agrif_rhot() ! Time indexes bounds for integration zt0 = REAL(Agrif_NbStepint() , wp) / zrhot zt1 = REAL(Agrif_NbStepint()+1, wp) / zrhot ! Polynomial interpolation coefficients: zat = zrhot * ( zt1**2._wp * (-2._wp*zt1 + 3._wp) & & - zt0**2._wp * (-2._wp*zt0 + 3._wp) ) ! IF(western_side ) vbdy_w(j1:j2) = zat * ptab(i1,j1:j2) IF(eastern_side ) vbdy_e(j1:j2) = zat * ptab(i1,j1:j2) IF(southern_side) vbdy_s(i1:i2) = zat * ptab(i1:i2,j1) IF(northern_side) vbdy_n(i1:i2) = zat * ptab(i1:i2,j1) ENDIF ! END SUBROUTINE interpvb2b SUBROUTINE interpe3t( ptab, i1, i2, j1, j2, k1, k2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpe3t *** !!---------------------------------------------------------------------- INTEGER , INTENT(in ) :: i1, i2, j1, j2, k1, k2 REAL(wp),DIMENSION(i1:i2,j1:j2,k1:k2), INTENT(inout) :: ptab LOGICAL , INTENT(in ) :: before INTEGER , INTENT(in ) :: nb , ndir ! INTEGER :: ji, jj, jk LOGICAL :: western_side, eastern_side, northern_side, southern_side REAL(wp) :: ztmpmsk !!---------------------------------------------------------------------- ! IF( before ) THEN ptab(i1:i2,j1:j2,k1:k2) = tmask(i1:i2,j1:j2,k1:k2) * e3t_0(i1:i2,j1:j2,k1:k2) ELSE 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) DO jk = k1, k2 DO jj = j1, j2 DO ji = i1, i2 ! Get velocity mask at boundary edge points: IF( western_side ) ztmpmsk = umask(ji ,jj ,1) IF( eastern_side ) ztmpmsk = umask(nlci-2,jj ,1) IF( northern_side) ztmpmsk = vmask(ji ,nlcj-2,1) IF( southern_side) ztmpmsk = vmask(ji ,2 ,1) ! IF( ABS( ptab(ji,jj,jk) - tmask(ji,jj,jk) * e3t_0(ji,jj,jk) )*ztmpmsk > 1.D-2) THEN IF (western_side) THEN WRITE(numout,*) 'ERROR bathymetry merge at the western border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk ELSEIF (eastern_side) THEN WRITE(numout,*) 'ERROR bathymetry merge at the eastern border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk ELSEIF (southern_side) THEN WRITE(numout,*) 'ERROR bathymetry merge at the southern border ji,jj,jk', ji+nimpp-1,jj+njmpp-1,jk ELSEIF (northern_side) THEN WRITE(numout,*) 'ERROR bathymetry merge at the northen border ji,jj,jk', ji+nimpp-1,jj+njmpp-1,jk ENDIF WRITE(numout,*) ' ptab(ji,jj,jk), e3t(ji,jj,jk) ', ptab(ji,jj,jk), e3t_0(ji,jj,jk) kindic_agr = kindic_agr + 1 ENDIF END DO END DO END DO ! ENDIF ! END SUBROUTINE interpe3t SUBROUTINE interpumsk( ptab, i1, i2, j1, j2, k1, k2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpumsk *** !!---------------------------------------------------------------------- INTEGER , INTENT(in ) :: i1, i2, j1, j2, k1, k2 REAL(wp),DIMENSION(i1:i2,j1:j2,k1:k2), INTENT(inout) :: ptab LOGICAL , INTENT(in ) :: before INTEGER , INTENT(in ) :: nb , ndir ! INTEGER :: ji, jj, jk LOGICAL :: western_side, eastern_side !!---------------------------------------------------------------------- ! IF( before ) THEN ptab(i1:i2,j1:j2,k1:k2) = umask(i1:i2,j1:j2,k1:k2) ELSE western_side = (nb == 1).AND.(ndir == 1) eastern_side = (nb == 1).AND.(ndir == 2) DO jk = k1, k2 DO jj = j1, j2 DO ji = i1, i2 ! Velocity mask at boundary edge points: IF (ABS(ptab(ji,jj,jk) - umask(ji,jj,jk)) > 1.D-2) THEN IF (western_side) THEN WRITE(numout,*) 'ERROR with umask at the western border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk WRITE(numout,*) ' masks: parent, child ', ptab(ji,jj,jk), umask(ji,jj,jk) kindic_agr = kindic_agr + 1 ELSEIF (eastern_side) THEN WRITE(numout,*) 'ERROR with umask at the eastern border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk WRITE(numout,*) ' masks: parent, child ', ptab(ji,jj,jk), umask(ji,jj,jk) kindic_agr = kindic_agr + 1 ENDIF ENDIF END DO END DO END DO ! ENDIF ! END SUBROUTINE interpumsk SUBROUTINE interpvmsk( ptab, i1, i2, j1, j2, k1, k2, before, nb, ndir ) !!---------------------------------------------------------------------- !! *** ROUTINE interpvmsk *** !!---------------------------------------------------------------------- INTEGER , INTENT(in ) :: i1,i2,j1,j2,k1,k2 REAL(wp),DIMENSION(i1:i2,j1:j2,k1:k2), INTENT(inout) :: ptab LOGICAL , INTENT(in ) :: before INTEGER , INTENT(in ) :: nb , ndir ! INTEGER :: ji, jj, jk LOGICAL :: northern_side, southern_side !!---------------------------------------------------------------------- ! IF( before ) THEN ptab(i1:i2,j1:j2,k1:k2) = vmask(i1:i2,j1:j2,k1:k2) ELSE southern_side = (nb == 2).AND.(ndir == 1) northern_side = (nb == 2).AND.(ndir == 2) DO jk = k1, k2 DO jj = j1, j2 DO ji = i1, i2 ! Velocity mask at boundary edge points: IF (ABS(ptab(ji,jj,jk) - vmask(ji,jj,jk)) > 1.D-2) THEN IF (southern_side) THEN WRITE(numout,*) 'ERROR with vmask at the southern border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk WRITE(numout,*) ' masks: parent, child ', ptab(ji,jj,jk), vmask(ji,jj,jk) kindic_agr = kindic_agr + 1 ELSEIF (northern_side) THEN WRITE(numout,*) 'ERROR with vmask at the northern border ji,jj,jk ', ji+nimpp-1,jj+njmpp-1,jk WRITE(numout,*) ' masks: parent, child ', ptab(ji,jj,jk), vmask(ji,jj,jk) kindic_agr = kindic_agr + 1 ENDIF ENDIF END DO END DO END DO ! ENDIF ! END SUBROUTINE interpvmsk # if defined key_zdftke SUBROUTINE interpavm( ptab, i1, i2, j1, j2, k1, k2, m1, m2, before ) !!---------------------------------------------------------------------- !! *** ROUTINE interavm *** !!---------------------------------------------------------------------- INTEGER , INTENT(in ) :: i1, i2, j1, j2, k1, k2, m1, m2 REAL(wp),DIMENSION(i1:i2,j1:j2,k1:k2,m1:m2), INTENT(inout) :: ptab LOGICAL , INTENT(in ) :: before REAL(wp), DIMENSION(k1:k2) :: tabin REAL(wp) :: h_in(k1:k2) REAL(wp) :: h_out(1:jpk) REAL(wp) :: zrhoxy INTEGER :: N_in, N_out, ji, jj, jk !!---------------------------------------------------------------------- ! zrhoxy = Agrif_rhox()*Agrif_rhoy() IF (before) THEN DO jk=k1,k2 DO jj=j1,j2 DO ji=i1,i2 ptab(ji,jj,jk,1) = avm_k(ji,jj,jk) END DO END DO END DO #ifdef key_vertical DO jk=k1,k2 DO jj=j1,j2 DO ji=i1,i2 ptab(ji,jj,jk,2) = wmask(ji,jj,jk) * e1e2t(ji,jj) * e3w_n(ji,jj,jk) END DO END DO END DO #else ptab(i1:i2,j1:j2,k1:k2,2) = 0._wp #endif ELSE #ifdef key_vertical avm_k(i1:i2,j1:j2,1:jpk) = 0. DO jj=j1,j2 DO ji=i1,i2 N_in = 0 DO jk=k1,k2 !k2 = jpk of parent grid IF (ptab(ji,jj,jk,2) == 0) EXIT N_in = N_in + 1 tabin(jk) = ptab(ji,jj,jk,1) h_in(N_in) = ptab(ji,jj,jk,2)/(e1e2t(ji,jj)*zrhoxy) END DO N_out = 0 DO jk=1,jpk ! jpk of child grid IF (wmask(ji,jj,jk) == 0) EXIT N_out = N_out + 1 h_out(jk) = e3t_n(ji,jj,jk) ENDDO IF (N_in > 0) THEN CALL reconstructandremap(tabin(1:N_in),h_in,avm_k(ji,jj,1:N_out),h_out,N_in,N_out) ENDIF ENDDO ENDDO #else avm_k(i1:i2,j1:j2,k1:k2) = ptab (i1:i2,j1:j2,k1:k2,1) #endif ENDIF ! END SUBROUTINE interpavm # endif /* key_zdftke */ #else !!---------------------------------------------------------------------- !! Empty module no AGRIF zoom !!---------------------------------------------------------------------- CONTAINS SUBROUTINE Agrif_OPA_Interp_empty WRITE(*,*) 'agrif_opa_interp : You should not have seen this print! error?' END SUBROUTINE Agrif_OPA_Interp_empty #endif !!====================================================================== END MODULE agrif_opa_interp