Changeset 14086 for NEMO/trunk/src/NST/agrif_top_sponge.F90
- Timestamp:
- 2020-12-04T12:37:14+01:00 (4 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
NEMO/trunk/src/NST/agrif_top_sponge.F90
r12489 r14086 45 45 ! 46 46 #if defined SPONGE_TOP 47 !! Assume persistence 47 !! Assume persistence: 48 48 zcoef = REAL(Agrif_rhot()-1,wp)/REAL(Agrif_rhot()) 49 CALL Agrif_sponge 49 50 50 Agrif_SpecialValue = 0._wp 51 51 Agrif_UseSpecialValue = .TRUE. 52 l_vremap = ln_vert_remap 52 53 tabspongedone_trn = .FALSE. 54 ! 53 55 CALL Agrif_Bc_Variable( trn_sponge_id, calledweight=zcoef, procname=interptrn_sponge ) 56 ! 54 57 Agrif_UseSpecialValue = .FALSE. 58 l_vremap = .FALSE. 55 59 #endif 56 60 ! … … 58 62 59 63 60 SUBROUTINE interptrn_sponge( tabres, i1, i2, j1, j2, k1, k2, n1, n2, before )61 !!---------------------------------------------------------------------- 62 !! 64 SUBROUTINE interptrn_sponge( tabres, i1, i2, j1, j2, k1, k2, n1, n2, before) 65 !!---------------------------------------------------------------------- 66 !! *** ROUTINE interptrn_sponge *** 63 67 !!---------------------------------------------------------------------- 64 68 INTEGER , INTENT(in ) :: i1, i2, j1, j2, k1, k2, n1, n2 … … 67 71 ! 68 72 INTEGER :: ji, jj, jk, jn ! dummy loop indices 69 REAL(wp) :: zabe1, zabe2, ztrelax 70 REAL(wp), DIMENSION(i1:i2,j1:j2) :: ztu, ztv 71 REAL(wp), DIMENSION(i1:i2,j1:j2,k1:k2,1:jptra) :: trbdiff 73 INTEGER :: iku, ikv 74 REAL(wp) :: ztra, zabe1, zabe2, zbtr, zhtot 75 REAL(wp), DIMENSION(i1-1:i2,j1-1:j2,jpk) :: ztu, ztv 76 REAL(wp), DIMENSION(i1:i2,j1:j2,jpk,n1:n2) ::trbdiff 72 77 ! vertical interpolation: 73 REAL(wp), DIMENSION(i1:i2,j1:j2,jpk, 1:jptra) ::tabres_child74 REAL(wp), DIMENSION(k1:k2, 1:jptra) :: tabin75 REAL(wp), DIMENSION(k1:k2) :: h_in76 REAL(wp), DIMENSION(1:jpk) :: h_out 78 REAL(wp), DIMENSION(i1:i2,j1:j2,jpk,n1:n2) ::tabres_child 79 REAL(wp), DIMENSION(k1:k2,n1:n2-1) :: tabin, tabin_i 80 REAL(wp), DIMENSION(k1:k2) :: z_in, z_in_i, h_in_i 81 REAL(wp), DIMENSION(1:jpk) :: h_out, z_out 77 82 INTEGER :: N_in, N_out 78 REAL(wp) :: h_diff79 83 !!---------------------------------------------------------------------- 80 84 ! … … 90 94 END DO 91 95 92 # if defined key_vertical 93 DO jk=k1,k2 94 DO jj=j1,j2 95 DO ji=i1,i2 96 tabres(ji,jj,jk,jpts+1) = tmask(ji,jj,jk) * e3t(ji,jj,jk,Kbb_a) 97 END DO 98 END DO 99 END DO 100 # endif 101 ELSE 102 # if defined key_vertical 103 tabres_child(:,:,:,:) = 0. 104 DO jj=j1,j2 105 DO ji=i1,i2 106 N_in = 0 107 DO jk=k1,k2 !k2 = jpk of parent grid 108 IF (tabres(ji,jj,jk,n2) == 0) EXIT 109 N_in = N_in + 1 110 tabin(jk,:) = tabres(ji,jj,jk,n1:n2-1) 111 h_in(N_in) = tabres(ji,jj,jk,n2) 112 END DO 113 N_out = 0 114 DO jk=1,jpk ! jpk of child grid 115 IF (tmask(ji,jj,jk) == 0) EXIT 116 N_out = N_out + 1 117 h_out(jk) = e3t(ji,jj,jk,Kbb_a) !Child grid scale factors. Could multiply by e1e2t here instead of division above 118 ENDDO 119 IF (N_in > 0) THEN 120 CALL reconstructandremap(tabin(1:N_in,1:jptra),h_in,tabres_child(ji,jj,1:N_out,1:jptra),h_out,N_in,N_out,jptra) 96 IF ( l_vremap.OR.ln_zps ) THEN 97 98 ! Fill cell depths (i.e. gdept) to be interpolated 99 ! Warning: these are masked, hence extrapolated prior interpolation. 100 DO jj=j1,j2 101 DO ji=i1,i2 102 tabres(ji,jj,k1,jptra+1) = 0.5_wp * tmask(ji,jj,k1) * e3t(ji,jj,k1,Kbb_a) 103 DO jk=k1+1,k2 104 tabres(ji,jj,jk,jptra+1) = tmask(ji,jj,jk) * & 105 & ( tabres(ji,jj,jk-1,jptra+1) + 0.5_wp * (e3t(ji,jj,jk-1,Kbb_a)+e3t(ji,jj,jk,Kbb_a)) ) 106 END DO 107 END DO 108 END DO 109 110 ! Save ssh at last level: 111 IF ( .NOT.ln_linssh ) THEN 112 tabres(i1:i2,j1:j2,k2,jptra+1) = ssh(i1:i2,j1:j2,Kbb_a)*tmask(i1:i2,j1:j2,1) 113 END IF 114 115 END IF 116 117 ELSE 118 ! 119 IF ( l_vremap ) THEN 120 121 IF (ln_linssh) tabres(i1:i2,j1:j2,k2,n2) = 0._wp 122 123 DO jj=j1,j2 124 DO ji=i1,i2 125 126 tabres_child(ji,jj,:,:) = 0._wp 127 ! Build vertical grids: 128 N_in = mbkt_parent(ji,jj) 129 ! Input grid (account for partial cells if any): 130 DO jk=1,N_in 131 z_in(jk) = tabres(ji,jj,jk,n2) - tabres(ji,jj,k2,n2) 132 tabin(jk,1:jptra) = tabres(ji,jj,jk,1:jptra) 133 END DO 134 135 ! Intermediate grid: 136 DO jk = 1, N_in 137 h_in_i(jk) = e3t0_parent(ji,jj,jk) * & 138 & (1._wp + tabres(ji,jj,k2,n2)/(ht0_parent(ji,jj)*ssmask(ji,jj) + 1._wp - ssmask(ji,jj))) 139 END DO 140 z_in_i(1) = 0.5_wp * h_in_i(1) 141 DO jk=2,N_in 142 z_in_i(jk) = z_in_i(jk-1) + 0.5_wp * ( h_in_i(jk) + h_in_i(jk-1) ) 143 END DO 144 z_in_i(1:N_in) = z_in_i(1:N_in) - tabres(ji,jj,k2,n2) 145 146 ! Output (Child) grid: 147 N_out = mbkt(ji,jj) 148 DO jk=1,N_out 149 h_out(jk) = e3t(ji,jj,jk,Kbb_a) 150 END DO 151 z_out(1) = 0.5_wp * h_out(1) 152 DO jk=2,N_out 153 z_out(jk) = z_out(jk-1) + 0.5_wp * ( h_out(jk)+h_out(jk-1) ) 154 END DO 155 IF (.NOT.ln_linssh) z_out(1:N_out) = z_out(1:N_out) - ssh(ji,jj,Kbb_a) 156 157 ! Account for small differences in the free-surface 158 IF ( sum(h_out(1:N_out)) > sum(h_in_i(1:N_in) )) THEN 159 h_out(1) = h_out(1) - ( sum(h_out(1:N_out))-sum(h_in_i(1:N_in)) ) 160 ELSE 161 h_in_i(1)= h_in_i(1) - ( sum(h_in_i(1:N_in))-sum(h_out(1:N_out)) ) 162 END IF 163 IF (N_in*N_out > 0) THEN 164 CALL remap_linear(tabin(1:N_in,1:jptra),z_in(1:N_in),tabin_i(1:N_in,1:jptra),z_in_i(1:N_in),N_in,N_in,jptra) 165 CALL reconstructandremap(tabin_i(1:N_in,1:jptra),h_in_i(1:N_in),tabres_child(ji,jj,1:N_out,1:jptra),h_out(1:N_out),N_in,N_out,jptra) 166 ! CALL remap_linear(tabin(1:N_in,1:jptra),z_in(1:N_in),tabres_child(ji,jj,1:N_out,1:jptra),z_out(1:N_in),N_in,N_out,jptra) 167 ENDIF 168 END DO 169 END DO 170 171 DO jj=j1,j2 172 DO ji=i1,i2 173 DO jk=1,jpkm1 174 trbdiff(ji,jj,jk,1:jptra) = (tr(ji,jj,jk,1:jptra,Kbb_a) - tabres_child(ji,jj,jk,1:jptra)) * tmask(ji,jj,jk) 175 END DO 176 END DO 177 END DO 178 179 ELSE 180 181 IF ( Agrif_Parent(ln_zps) ) THEN ! Account for partial cells 182 183 DO jj=j1,j2 184 DO ji=i1,i2 185 ! 186 N_in = mbkt(ji,jj) 187 N_out = mbkt(ji,jj) 188 z_in(1) = tabres(ji,jj,1,n2) 189 tabin(1,1:jptra) = tabres(ji,jj,1,1:jptra) 190 DO jk=2, N_in 191 z_in(jk) = tabres(ji,jj,jk,n2) 192 tabin(jk,1:jptra) = tabres(ji,jj,jk,1:jptra) 193 END DO 194 IF (.NOT.ln_linssh) z_in(1:N_in) = z_in(1:N_in) - tabres(ji,jj,k2,n2) 195 196 z_out(1) = 0.5_wp * e3t(ji,jj,1,Kbb_a) 197 DO jk=2, N_out 198 z_out(jk) = z_out(jk-1) + 0.5_wp * (e3t(ji,jj,jk-1,Kbb_a) + e3t(ji,jj,jk,Kbb_a)) 199 END DO 200 IF (.NOT.ln_linssh) z_out(1:N_out) = z_out(1:N_out) - ssh(ji,jj,Kbb_a) 201 202 CALL remap_linear(tabin(1:N_in,1:jptra), z_in(1:N_in), tabres(ji,jj,1:N_out,1:jptra), & 203 & z_out(1:N_out), N_in, N_out, jptra) 204 END DO 205 END DO 206 ENDIF 207 208 DO jj=j1,j2 209 DO ji=i1,i2 210 DO jk=1,jpkm1 211 trbdiff(ji,jj,jk,1:jptra) = (tr(ji,jj,jk,1:jptra,Kbb_a) - tabres(ji,jj,jk,1:jptra))*tmask(ji,jj,jk) 212 END DO 213 END DO 214 END DO 215 216 END IF 217 218 DO jn = 1, jptra 219 DO jk = 1, jpkm1 220 ztu(i1-1:i2,j1-1:j2,jk) = 0._wp 221 DO jj = j1,j2 222 DO ji = i1,i2-1 223 zabe1 = rn_sponge_tra * r1_Dt * umask(ji,jj,jk) * e1e2u(ji,jj) * e3u(ji,jj,jk,Kmm_a) 224 ztu(ji,jj,jk) = zabe1 * fspu(ji,jj) * ( trbdiff(ji+1,jj ,jk,jn) - trbdiff(ji,jj,jk,jn) ) 225 END DO 226 END DO 227 ztv(i1-1:i2,j1-1:j2,jk) = 0._wp 228 DO ji = i1,i2 229 DO jj = j1,j2-1 230 zabe2 = rn_sponge_tra * r1_Dt * vmask(ji,jj,jk) * e1e2v(ji,jj) * e3v(ji,jj,jk,Kmm_a) 231 ztv(ji,jj,jk) = zabe2 * fspv(ji,jj) * ( trbdiff(ji ,jj+1,jk,jn) - trbdiff(ji,jj,jk,jn) ) 232 END DO 233 END DO 234 ! 235 IF( ln_zps ) THEN ! set gradient at partial step level 236 DO jj = j1,j2 237 DO ji = i1,i2 238 ! last level 239 iku = mbku(ji,jj) 240 ikv = mbkv(ji,jj) 241 IF( iku == jk ) ztu(ji,jj,jk) = 0._wp 242 IF( ikv == jk ) ztv(ji,jj,jk) = 0._wp 243 END DO 244 END DO 121 245 ENDIF 122 ENDDO 123 ENDDO 124 # endif 125 126 DO jj=j1,j2 127 DO ji=i1,i2 128 DO jk=1,jpkm1 129 # if defined key_vertical 130 trbdiff(ji,jj,jk,1:jptra) = tr(ji,jj,jk,1:jptra,Kbb_a) - tabres_child(ji,jj,jk,1:jptra) 131 # else 132 trbdiff(ji,jj,jk,1:jptra) = tr(ji,jj,jk,1:jptra,Kbb_a) - tabres(ji,jj,jk,1:jptra) 133 # endif 134 ENDDO 135 ENDDO 136 ENDDO 137 138 !* set relaxation time scale 139 IF( l_1st_euler .AND. lk_agrif_fstep ) THEN ; ztrelax = rn_trelax_tra / ( rn_Dt ) 140 ELSE ; ztrelax = rn_trelax_tra / (2._wp * rn_Dt ) 141 ENDIF 142 143 DO jn = 1, jptra 246 END DO 247 ! 248 ! JC: there is something wrong with the Laplacian in corners 144 249 DO jk = 1, jpkm1 145 DO jj = j1,j2-1 146 DO ji = i1,i2-1 147 zabe1 = rn_sponge_tra * fspu(ji,jj) * e2_e1u(ji,jj) * e3u(ji,jj,jk,Kmm_a) * umask(ji,jj,jk) 148 zabe2 = rn_sponge_tra * fspv(ji,jj) * e1_e2v(ji,jj) * e3v(ji,jj,jk,Kmm_a) * vmask(ji,jj,jk) 149 ztu(ji,jj) = zabe1 * ( trbdiff(ji+1,jj ,jk,jn) - trbdiff(ji,jj,jk,jn) ) 150 ztv(ji,jj) = zabe2 * ( trbdiff(ji ,jj+1,jk,jn) - trbdiff(ji,jj,jk,jn) ) 151 END DO 152 END DO 153 ! 154 DO jj = j1+1,j2-1 155 DO ji = i1+1,i2-1 156 IF( .NOT. tabspongedone_trn(ji,jj) ) THEN 157 tr(ji,jj,jk,jn,Krhs_a) = tr(ji,jj,jk,jn,Krhs_a) + ( ztu(ji,jj) - ztu(ji-1,jj ) & 158 & + ztv(ji,jj) - ztv(ji ,jj-1) ) & 159 & * r1_e1e2t(ji,jj) / e3t(ji,jj,jk,Kmm_a) & 160 & - ztrelax * fspt(ji,jj) * trbdiff(ji,jj,jk,jn) 250 DO jj = j1,j2 251 DO ji = i1,i2 252 IF (.NOT. tabspongedone_trn(ji,jj)) THEN 253 zbtr = r1_e1e2t(ji,jj) / e3t(ji,jj,jk,Kmm_a) 254 ! horizontal diffusive trends 255 ztra = zbtr * ( ztu(ji,jj,jk) - ztu(ji-1,jj,jk) & 256 & + ztv(ji,jj,jk) - ztv(ji,jj-1,jk) ) & 257 & - rn_trelax_tra * r1_Dt * fspt(ji,jj) * trbdiff(ji,jj,jk,jn) 258 ! add it to the general tracer trends 259 tr(ji,jj,jk,jn,Krhs_a) = tr(ji,jj,jk,jn,Krhs_a) + ztra 161 260 ENDIF 162 261 END DO 163 262 END DO 263 164 264 END DO 165 265 ! 166 266 END DO 167 267 ! 168 tabspongedone_trn(i1+1:i2-1,j1+1:j2-1) = .TRUE. 268 tabspongedone_trn(i1:i2,j1:j2) = .TRUE. 269 ! 169 270 ENDIF 170 ! 271 ! 171 272 END SUBROUTINE interptrn_sponge 172 273
Note: See TracChangeset
for help on using the changeset viewer.