MODULE zpshde_crs !!====================================================================== !! *** MODULE zpshde *** !! z-coordinate + partial step : Horizontal Derivative at ocean bottom level !!====================================================================== !! History : OPA ! 2002-04 (A. Bozec) Original code !! NEMO 1.0 ! 2002-08 (G. Madec E. Durand) Optimization and Free form !! - ! 2004-03 (C. Ethe) adapted for passive tracers !! 3.3 ! 2010-05 (C. Ethe, G. Madec) merge TRC-TRA !!====================================================================== !!---------------------------------------------------------------------- !! zps_hde : Horizontal DErivative of T, S and rd at the last !! ocean level (Z-coord. with Partial Steps) !!---------------------------------------------------------------------- USE oce ! ocean: dynamics and tracers variables USE dom_oce ! domain: ocean variables USE phycst ! physical constants USE eosbn2_crs ! ocean equation of state USE in_out_manager ! I/O manager USE crslbclnk ! lateral boundary conditions (or mpp link) USE lib_mpp ! MPP library USE wrk_nemo ! Memory allocation USE timing ! Timing USE crs IMPLICIT NONE PRIVATE PUBLIC zps_hde_crs ! routine called by step.F90 !! * Substitutions # include "domzgr_substitute.h90" # include "vectopt_loop_substitute.h90" !!---------------------------------------------------------------------- !! NEMO/OPA 3.3 , NEMO Consortium (2010) !! $Id: zpshde.F90 3294 2012-01-28 16:44:18Z rblod $ !! Software governed by the CeCILL licence (NEMOGCM/NEMO_CeCILL.txt) !!---------------------------------------------------------------------- CONTAINS SUBROUTINE zps_hde_crs( kt, kjpt, pta, pgtu, pgtv, & prd, pgru, pgrv ) !!---------------------------------------------------------------------- !! *** ROUTINE zps_hde *** !! !! ** Purpose : Compute the horizontal derivative of T, S and rho !! at u- and v-points with a linear interpolation for z-coordinate !! with partial steps. !! !! ** Method : In z-coord with partial steps, scale factors on last !! levels are different for each grid point, so that T, S and rd !! points are not at the same depth as in z-coord. To have horizontal !! gradients again, we interpolate T and S at the good depth : !! Linear interpolation of T, S !! Computation of di(tb) and dj(tb) by vertical interpolation: !! di(t) = t~ - t(i,j,k) or t(i+1,j,k) - t~ !! dj(t) = t~ - t(i,j,k) or t(i,j+1,k) - t~ !! This formulation computes the two cases: !! CASE 1 CASE 2 !! k-1 ___ ___________ k-1 ___ ___________ !! Ti T~ T~ Ti+1 !! _____ _____ !! k | |Ti+1 k Ti | | !! | |____ ____| | !! ___ | | | ___ | | | !! !! case 1-> e3w(i+1) >= e3w(i) ( and e3w(j+1) >= e3w(j) ) then !! t~ = t(i+1,j ,k) + (e3w(i+1) - e3w(i)) * dk(Ti+1)/e3w(i+1) !! ( t~ = t(i ,j+1,k) + (e3w(j+1) - e3w(j)) * dk(Tj+1)/e3w(j+1) ) !! or !! case 2-> e3w(i+1) <= e3w(i) ( and e3w(j+1) <= e3w(j) ) then !! t~ = t(i,j,k) + (e3w(i) - e3w(i+1)) * dk(Ti)/e3w(i ) !! ( t~ = t(i,j,k) + (e3w(j) - e3w(j+1)) * dk(Tj)/e3w(j ) ) !! Idem for di(s) and dj(s) !! !! For rho, we call eos_insitu_2d which will compute rd~(t~,s~) at !! the good depth zh from interpolated T and S for the different !! formulation of the equation of state (eos). !! Gradient formulation for rho : !! di(rho) = rd~ - rd(i,j,k) or rd(i+1,j,k) - rd~ !! !! ** Action : - pgtu, pgtv: horizontal gradient of tracer at u- & v-points !! - pgru, pgrv: horizontal gradient of rho (if present) at u- & v-points !!---------------------------------------------------------------------- ! INTEGER , INTENT(in ) :: kt ! ocean time-step index INTEGER , INTENT(in ) :: kjpt ! number of tracers REAL(wp), DIMENSION(jpi,jpj,jpk,kjpt), INTENT(in ) :: pta ! 4D tracers fields REAL(wp), DIMENSION(jpi,jpj, kjpt), INTENT( out) :: pgtu, pgtv ! hor. grad. of ptra at u- & v-pts REAL(wp), DIMENSION(jpi,jpj,jpk ), INTENT(in ), OPTIONAL :: prd ! 3D density anomaly fields REAL(wp), DIMENSION(jpi,jpj ), INTENT( out), OPTIONAL :: pgru, pgrv ! hor. grad. of prd at u- & v-pts ! INTEGER :: ji, jj, jn ! Dummy loop indices INTEGER :: iku, ikv, ikum1, ikvm1 ! partial step level (ocean bottom level) at u- and v-points REAL(wp) :: ze3wu, ze3wv, zmaxu, zmaxv ! temporary scalars !cc REAL(wp), POINTER, DIMENSION(:,: ) :: zri, zrj, zhi, zhj !cc REAL(wp), POINTER, DIMENSION(:,:,:) :: zti, zte ! interpolated value of tracer REAL(wp), ALLOCATABLE, DIMENSION(:,: ) :: zri, zrj, zhi, zhj REAL(wp), ALLOCATABLE, DIMENSION(:,:,:) :: zti, zte ! interpolated value of tracer !!---------------------------------------------------------------------- ! IF( nn_timing == 1 ) CALL timing_start( 'zps_hde_crs') ! !! CALL wrk_alloc( jpi, jpj, zri, zrj, zhi, zhj ) !! CALL wrk_alloc( jpi, jpj, kjpt, zti, zte ) ALLOCATE( zri(jpi_crs,jpj_crs) , zrj(jpi_crs,jpj_crs), zte(jpi_crs ,jpj_crs ,kjpt), & & zhi(jpi_crs,jpj_crs) , zhj(jpi_crs,jpj_crs), zti(jpi_crs ,jpj_crs ,kjpt)) ! DO jn = 1, kjpt !== Interpolation of tracers at the last ocean level ==! ! # if defined key_vectopt_loop jj = 1 DO ji = 1, jpij-jpi ! vector opt. (forced unrolled) # else DO jj = 1, jpjm1 DO ji = 1, jpim1 # endif iku = mbku_crs(ji,jj) ; ikum1 = MAX( iku - 1 , 1 ) ! last and before last ocean level at u- & v-points ikv = mbkv_crs(ji,jj) ; ikvm1 = MAX( ikv - 1 , 1 ) ! if level first is a p-step, ik.m1=1 ! ze3wu = e3w_crs(ji+1,jj ,iku) - e3w_crs(ji,jj,iku) ! ze3wv = e3w_crs(ji ,jj+1,ikv) - e3w_crs(ji,jj,ikv) ze3wu = e3w_max_crs(ji+1,jj ,iku) - e3w_max_crs(ji,jj,iku) ze3wv = e3w_max_crs(ji ,jj+1,ikv) - e3w_max_crs(ji,jj,ikv) ! ! i- direction IF( ze3wu >= 0._wp ) THEN ! case 1 zmaxu = ze3wu / e3w_max_crs(ji+1,jj,iku) ! zmaxu = ze3wu / e3w_crs(ji+1,jj,iku) ! interpolated values of tracers zti(ji,jj,jn) = pta(ji+1,jj,iku,jn) + zmaxu * ( pta(ji+1,jj,ikum1,jn) - pta(ji+1,jj,iku,jn) ) ! gradient of tracers pgtu(ji,jj,jn) = umask_crs(ji,jj,1) * ( zti(ji,jj,jn) - pta(ji,jj,iku,jn) ) ELSE ! case 2 zmaxu = -ze3wu / e3w_max_crs(ji,jj,iku) ! zmaxu = -ze3wu / e3w_crs(ji,jj,iku) ! interpolated values of tracers zti(ji,jj,jn) = pta(ji,jj,iku,jn) + zmaxu * ( pta(ji,jj,ikum1,jn) - pta(ji,jj,iku,jn) ) ! gradient of tracers pgtu(ji,jj,jn) = umask_crs(ji,jj,1) * ( pta(ji+1,jj,iku,jn) - zti(ji,jj,jn) ) ENDIF ! ! j- direction IF( ze3wv >= 0._wp ) THEN ! case 1 zmaxv = ze3wv / e3w_max_crs(ji,jj+1,ikv) ! zmaxv = ze3wv / e3w_crs(ji,jj+1,ikv) ! interpolated values of tracers zte(ji,jj,jn) = pta(ji,jj+1,ikv,jn) + zmaxv * ( pta(ji,jj+1,ikvm1,jn) - pta(ji,jj+1,ikv,jn) ) ! gradient of tracers pgtv(ji,jj,jn) = vmask_crs(ji,jj,1) * ( zte(ji,jj,jn) - pta(ji,jj,ikv,jn) ) ELSE ! case 2 zmaxv = -ze3wv / e3w_max_crs(ji,jj,ikv) ! zmaxv = -ze3wv / e3w_crs(ji,jj,ikv) ! interpolated values of tracers zte(ji,jj,jn) = pta(ji,jj,ikv,jn) + zmaxv * ( pta(ji,jj,ikvm1,jn) - pta(ji,jj,ikv,jn) ) ! gradient of tracers pgtv(ji,jj,jn) = vmask_crs(ji,jj,1) * ( pta(ji,jj+1,ikv,jn) - zte(ji,jj,jn) ) ENDIF # if ! defined key_vectopt_loop END DO # endif END DO CALL crs_lbc_lnk( pgtu(:,:,jn), 'U', -1. ) ; CALL crs_lbc_lnk( pgtv(:,:,jn), 'V', -1. ) ! Lateral boundary cond. ! END DO !WRITE(numout,*) ' test24 ', e3w_max_crs ! horizontal derivative of density anomalies (rd) IF( PRESENT( prd ) ) THEN ! depth of the partial step level # if defined key_vectopt_loop jj = 1 DO ji = 1, jpij-jpi ! vector opt. (forced unrolled) # else DO jj = 1, jpjm1 DO ji = 1, jpim1 # endif iku = mbku_crs(ji,jj) ikv = mbkv_crs(ji,jj) !cc ze3wu = e3w_max_crs(ji+1,jj ,iku) - e3w_max_crs(ji,jj,iku) !gradiant horizontal pas de max ze3wu = e3w_crs(ji+1,jj ,iku) - e3w_crs(ji,jj,iku) !cc ze3wv = e3w_max_crs(ji ,jj+1,ikv) - e3w_max_crs(ji,jj,ikv) ze3wv = e3w_crs(ji ,jj+1,ikv) - e3w_crs(ji,jj,ikv) IF( ze3wu >= 0._wp ) THEN ; zhi(ji,jj) = gdept_crs(ji ,jj,iku) ! i-direction: case 1 ELSE ; zhi(ji,jj) = gdept_crs(ji+1,jj,iku) ! - - case 2 ENDIF IF( ze3wv >= 0._wp ) THEN ; zhj(ji,jj) = gdept_crs(ji,jj ,ikv) ! j-direction: case 1 ELSE ; zhj(ji,jj) = gdept_crs(ji,jj+1,ikv) ! - - case 2 ENDIF # if ! defined key_vectopt_loop END DO # endif END DO CALL eos_crs( zti, zhi, zri ) CALL eos_crs( zte, zhj, zrj ) ! Gradient of density at the last level # if defined key_vectopt_loop jj = 1 DO ji = 1, jpij-jpi ! vector opt. (forced unrolled) # else DO jj = 1, jpjm1 DO ji = 1, jpim1 # endif iku = mbku_crs(ji,jj) ikv = mbkv_crs(ji,jj) ! ze3wu = e3w_max_crs(ji+1,jj ,iku) - e3w_max_crs(ji,jj,iku) gradient horizontal ze3wu = e3w_crs(ji+1,jj ,iku) - e3w_crs(ji,jj,iku) ! ze3wv = e3w_max_crs(ji ,jj+1,ikv) - e3w_max_crs(ji,jj,ikv) gradient horizontal ze3wv = e3w_crs(ji ,jj+1,ikv) - e3w_crs(ji,jj,ikv) IF( ze3wu >= 0._wp ) THEN ; pgru(ji,jj) = umask_crs(ji,jj,1) * ( zri(ji ,jj) - prd(ji,jj,iku) ) ! i: 1 ELSE ; pgru(ji,jj) = umask_crs(ji,jj,1) * ( prd(ji+1,jj,iku) - zri(ji,jj) ) ! i: 2 ENDIF IF( ze3wv >= 0._wp ) THEN ; pgrv(ji,jj) = vmask_crs(ji,jj,1) * ( zrj(ji,jj ) - prd(ji,jj,ikv) ) ! j: 1 ELSE ; pgrv(ji,jj) = vmask_crs(ji,jj,1) * ( prd(ji,jj+1,ikv) - zrj(ji,jj) ) ! j: 2 ENDIF # if ! defined key_vectopt_loop END DO # endif END DO CALL crs_lbc_lnk( pgru , 'U', -1. ) ; CALL crs_lbc_lnk( pgrv , 'V', -1. ) ! Lateral boundary conditions ! END IF ! !!ccCALL wrk_dealloc( jpi, jpj, zri, zrj, zhi, zhj ) !!ccCALL wrk_dealloc( jpi, jpj, kjpt, zti, zte ) DEALLOCATE( zri , zrj, zte, zhi, zhj, zti) ! IF( nn_timing == 1 ) CALL timing_stop( 'zps_hde_crs') ! END SUBROUTINE zps_hde_crs !!====================================================================== END MODULE zpshde_crs