98 & LBi, UBi, LBj, UBj, &
99 & IminS, ImaxS, JminS, JmaxS, &
100 & nrhs, nstp, nnew, &
104#ifdef WET_DRY_NOT_YET
105 & umask_wet, vmask_wet, &
107 & om_v, on_u, pm, pn, &
130 integer,
intent(in) :: ng, tile
131 integer,
intent(in) :: LBi, UBi, LBj, UBj
132 integer,
intent(in) :: IminS, ImaxS, JminS, JmaxS
133 integer,
intent(in) :: nrhs, nstp, nnew
137 real(r8),
intent(in) :: umask(LBi:,LBj:)
138 real(r8),
intent(in) :: vmask(LBi:,LBj:)
140# ifdef WET_DRY_NOT_YET
141 real(r8),
intent(in) :: umask_wet(LBi:,LBj:)
142 real(r8),
intent(in) :: vmask_wet(LBi:,LBj:)
145 real(r8),
intent(in) :: diff3d_r(LBi:,LBj:,:)
147 real(r8),
intent(in) :: diff2(LBi:,LBj:,:)
149 real(r8),
intent(in) :: om_v(LBi:,LBj:)
150 real(r8),
intent(in) :: on_u(LBi:,LBj:)
151 real(r8),
intent(in) :: pm(LBi:,LBj:)
152 real(r8),
intent(in) :: pn(LBi:,LBj:)
153 real(r8),
intent(in) :: Hz(LBi:,LBj:,:)
154 real(r8),
intent(in) :: z_r(LBi:,LBj:,:)
155 real(r8),
intent(in) :: pden(LBi:,LBj:,:)
156 real(r8),
intent(in) :: t(LBi:,LBj:,:,:,:)
158 real(r8),
intent(in) :: tclm(LBi:,LBj:,:,:)
160 real(r8),
intent(in) :: tl_Hz(LBi:,LBj:,:)
161 real(r8),
intent(in) :: tl_z_r(LBi:,LBj:,:)
162 real(r8),
intent(in) :: tl_pden(LBi:,LBj:,:)
163# ifdef DIAGNOSTICS_TS
167 real(r8),
intent(inout) :: tl_t(LBi:,LBj:,:,:,:)
170 real(r8),
intent(in) :: umask(LBi:UBi,LBj:UBj)
171 real(r8),
intent(in) :: vmask(LBi:UBi,LBj:UBj)
173# ifdef WET_DRY_NOT_YET
174 real(r8),
intent(in) :: umask_wet(LBi:UBi,LBj:UBj)
175 real(r8),
intent(in) :: vmask_wet(LBi:UBi,LBj:UBj)
178 real(r8),
intent(in) :: diff3d_r(LBi:UBi,LBj:UBj,N(ng))
180 real(r8),
intent(in) :: diff2(LBi:UBi,LBj:UBj,NT(ng))
182 real(r8),
intent(in) :: om_v(LBi:UBi,LBj:UBj)
183 real(r8),
intent(in) :: on_u(LBi:UBi,LBj:UBj)
184 real(r8),
intent(in) :: pm(LBi:UBi,LBj:UBj)
185 real(r8),
intent(in) :: pn(LBi:UBi,LBj:UBj)
186 real(r8),
intent(in) :: Hz(LBi:UBi,LBj:UBj,N(ng))
187 real(r8),
intent(in) :: z_r(LBi:UBi,LBj:UBj,N(ng))
188 real(r8),
intent(in) :: pden(LBi:UBi,LBj:UBj,N(ng))
189 real(r8),
intent(in) :: t(LBi:UBi,LBj:UBj,N(ng),3,NT(ng))
191 real(r8),
intent(in) :: tclm(LBi:UBi,LBj:UBj,N(ng),NT(ng))
193 real(r8),
intent(in) :: tl_Hz(LBi:UBi,LBj:UBj,N(ng))
194 real(r8),
intent(in) :: tl_z_r(LBi:UBi,LBj:UBj,N(ng))
195 real(r8),
intent(in) :: tl_pden(LBi:UBi,LBj:UBj,N(ng))
196# ifdef DIAGNOSTICS_TS
200 real(r8),
intent(inout) :: tl_t(LBi:UBi,LBj:UBj,N(ng),3,NT(ng))
205 integer :: i, itrc, j, k, k1, k2
207 real(r8),
parameter :: eps = 0.5_r8
208 real(r8),
parameter :: small = 1.0e-14_r8
209 real(r8),
parameter :: slope_max = 0.0001_r8
210 real(r8),
parameter :: strat_min = 0.1_r8
212 real(r8) :: cff, cff1, cff2, cff3, cff4
213 real(r8) :: tl_cff, tl_cff1, tl_cff2, tl_cff3, tl_cff4
215 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS) :: tl_FE
216 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS) :: tl_FX
218 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: FS
219 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: dRde
220 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: dRdx
221 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: dTde
222 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: dTdr
223 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: dTdx
225 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: tl_FS
226 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: tl_dRde
227 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: tl_dRdx
228 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: tl_dTde
229 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: tl_dTdr
230 real(r8),
dimension(IminS:ImaxS,JminS:JmaxS,2) :: tl_dTdx
232#include "set_bounds.h"
246 t_loop :
DO itrc=1,nt(ng)
248 k_loop :
DO k=0,n(ng)
254 cff=0.5_r8*(pm(i,j)+pm(i-1,j))
258#ifdef WET_DRY_NOT_YET
259 cff=cff*umask_wet(i,j)
261 drdx(i,j,k2)=cff*(pden(i ,j,k+1)- &
263 tl_drdx(i,j,k2)=cff*(tl_pden(i ,j,k+1)- &
264 & tl_pden(i-1,j,k+1))
265#if defined TS_MIX_STABILITY
266 dtdx(i,j,k2)=cff*(0.75_r8*(t(i ,j,k+1,nrhs,itrc)- &
267 & t(i-1,j,k+1,nrhs,itrc))+ &
268 & 0.25_r8*(t(i ,j,k+1,nstp,itrc)- &
269 & t(i-1,j,k+1,nstp,itrc)))
270 tl_dtdx(i,j,k2)=cff* &
271 & (0.75_r8*(tl_t(i ,j,k+1,nrhs,itrc)- &
272 & tl_t(i-1,j,k+1,nrhs,itrc))+ &
273 & 0.25_r8*(tl_t(i ,j,k+1,nstp,itrc)- &
274 & tl_t(i-1,j,k+1,nstp,itrc)))
275#elif defined TS_MIX_CLIMA
277 dtdx(i,j,k2)=cff*((t(i ,j,k+1,nrhs,itrc)- &
278 & tclm(i ,j,k+1,itrc))- &
279 & (t(i-1,j,k+1,nrhs,itrc)- &
280 & tclm(i-1,j,k+1,itrc)))
282 dtdx(i,j,k2)=cff*(t(i ,j,k+1,nrhs,itrc)- &
283 & t(i-1,j,k+1,nrhs,itrc))
285 tl_dtdx(i,j,k2)=cff*(tl_t(i ,j,k+1,nrhs,itrc)- &
286 & tl_t(i-1,j,k+1,nrhs,itrc))
288 dtdx(i,j,k2)=cff*(t(i ,j,k+1,nrhs,itrc)- &
289 & t(i-1,j,k+1,nrhs,itrc))
290 tl_dtdx(i,j,k2)=cff*(tl_t(i ,j,k+1,nrhs,itrc)- &
291 & tl_t(i-1,j,k+1,nrhs,itrc))
297 cff=0.5_r8*(pn(i,j)+pn(i,j-1))
301#ifdef WET_DRY_NOT_YET
302 cff=cff*vmask_wet(i,j)
304 drde(i,j,k2)=cff*(pden(i,j ,k+1)- &
306 tl_drde(i,j,k2)=cff*(tl_pden(i,j ,k+1)- &
307 & tl_pden(i,j-1,k+1))
308#if defined TS_MIX_STABILITY
309 dtde(i,j,k2)=cff*(0.75_r8*(t(i,j ,k+1,nrhs,itrc)- &
310 & t(i,j-1,k+1,nrhs,itrc))+ &
311 & 0.25_r8*(t(i,j ,k+1,nstp,itrc)- &
312 & t(i,j-1,k+1,nstp,itrc)))
313 tl_dtde(i,j,k2)=cff* &
314 & (0.75_r8*(tl_t(i,j ,k+1,nrhs,itrc)- &
315 & tl_t(i,j-1,k+1,nrhs,itrc))+ &
316 & 0.25_r8*(tl_t(i,j ,k+1,nstp,itrc)- &
317 & tl_t(i,j-1,k+1,nstp,itrc)))
318#elif defined TS_MIX_CLIMA
320 dtde(i,j,k2)=cff*((t(i,j ,k+1,nrhs,itrc)- &
321 & tclm(i,j ,k+1,itrc))- &
322 & (t(i,j-1,k+1,nrhs,itrc)- &
323 & tclm(i,j-1,k+1,itrc)))
325 dtde(i,j,k2)=cff*(t(i,j ,k+1,nrhs,itrc)- &
326 & t(i,j-1,k+1,nrhs,itrc))
328 tl_dtde(i,j,k2)=cff*(tl_t(i,j ,k+1,nrhs,itrc)- &
329 & tl_t(i,j-1,k+1,nrhs,itrc))
331 dtde(i,j,k2)=cff*(t(i,j ,k+1,nrhs,itrc)- &
332 & t(i,j-1,k+1,nrhs,itrc))
333 tl_dtde(i,j,k2)=cff*(tl_t(i,j ,k+1,nrhs,itrc)- &
334 & tl_t(i,j-1,k+1,nrhs,itrc))
339 IF ((k.eq.0).or.(k.eq.n(ng)))
THEN
343 tl_dtdr(i,j,k2)=0.0_r8
351#if defined TS_MIX_MAX_SLOPE
352 cff1=sqrt(drdx(i,j,k2)**2+drdx(i+1,j,k2)**2+ &
353 & drdx(i,j,k1)**2+drdx(i+1,j,k1)**2+ &
354 & drde(i,j,k2)**2+drde(i,j+1,k2)**2+ &
355 & drde(i,j,k1)**2+drde(i,j+1,k1)**2)
356 IF (cff1.ne.0.0_r8)
THEN
357 tl_cff1=(drdx(i ,j,k2)*tl_drdx(i ,j,k2)+ &
358 & drdx(i+1,j,k2)*tl_drdx(i+1,j,k2)+ &
359 & drdx(i ,j,k1)*tl_drdx(i ,j,k1)+ &
360 & drdx(i+1,j,k1)*tl_drdx(i+1,j,k1)+ &
361 & drde(i,j ,k2)*tl_drde(i,j ,k2)+ &
362 & drde(i,j+1,k2)*tl_drde(i,j+1,k2)+ &
363 & drde(i,j ,k1)*tl_drde(i,j ,k1)+ &
364 & drde(i,j+1,k1)*tl_drde(i,j+1,k1))/cff1
368 cff2=0.25_r8*slope_max* &
369 & (z_r(i,j,k+1)-z_r(i,j,k))*cff1
370 tl_cff2=0.25_r8*slope_max* &
371 & ((tl_z_r(i,j,k+1)-tl_z_r(i,j,k))*cff1+ &
372 & (z_r(i,j,k+1)-z_r(i,j,k))*tl_cff1)
373 cff3=max(pden(i,j,k)-pden(i,j,k+1),small)
374 tl_cff3=(0.5_r8+sign(0.5_r8,pden(i,j,k)-pden(i,j,k+1)- &
376 & (tl_pden(i,j,k)-tl_pden(i,j,k+1))
378 tl_cff4=(0.5_r8+sign(0.5_r8,cff2-cff3))*tl_cff2+ &
379 & (0.5_r8-sign(0.5_r8,cff2-cff3))*tl_cff3
381 tl_cff=cff*cff*tl_cff4
382#elif defined TS_MIX_MIN_STRAT
383 cff1=max(pden(i,j,k)-pden(i,j,k+1), &
384 & strat_min*(z_r(i,j,k+1)-z_r(i,j,k)))
385 tl_cff1=(0.5_r8+sign(0.5_r8, &
386 & pden(i,j,k)-pden(i,j,k+1)- &
387 & strat_min*(z_r(i,j,k+1)- &
389 & (tl_pden(i,j,k)-tl_pden(i,j,k+1))+ &
390 & (0.5_r8-sign(0.5_r8, &
391 & pden(i,j,k)-pden(i,j,k+1)- &
392 & strat_min*(z_r(i,j,k+1)- &
394 & (strat_min*(tl_z_r(i,j,k+1)-tl_z_r(i,j,k )))
396 tl_cff=cff*cff*tl_cff1
398 cff1=max(pden(i,j,k)-pden(i,j,k+1),eps)
399 tl_cff1=(0.5_r8+sign(0.5_r8, &
400 & pden(i,j,k)-pden(i,j,k+1)-eps))* &
401 & (tl_pden(i,j,k)-tl_pden(i,j,k+1))
403 tl_cff=cff*cff*tl_cff1
405#if defined TS_MIX_STABILITY
406 dtdr(i,j,k2)=cff*(0.75_r8*(t(i,j,k+1,nrhs,itrc)- &
407 & t(i,j,k ,nrhs,itrc))+ &
408 & 0.25_r8*(t(i,j,k+1,nstp,itrc)- &
409 & t(i,j,k ,nstp,itrc)))
410 tl_dtdr(i,j,k2)=tl_cff* &
411 & (0.75_r8*(t(i,j,k+1,nrhs,itrc)- &
412 & t(i,j,k ,nrhs,itrc))+ &
413 & 0.25_r8*(t(i,j,k+1,nstp,itrc)- &
414 & t(i,j,k ,nstp,itrc)))+ &
416 & (0.75_r8*(tl_t(i,j,k+1,nrhs,itrc)- &
417 & tl_t(i,j,k ,nrhs,itrc))+ &
418 & 0.25_r8*(tl_t(i,j,k+1,nstp,itrc)- &
419 & tl_t(i,j,k ,nstp,itrc)))
420#elif defined TS_MIX_CLIMA
422 dtdr(i,j,k2)=cff*((t(i,j,k+1,nrhs,itrc)- &
423 & tclm(i,j,k+1,itrc))- &
424 & (t(i,j,k ,nrhs,itrc)- &
425 & tclm(i,j,k ,itrc)))
426 tl_dtdr(i,j,k2)=tl_cff*((t(i,j,k+1,nrhs,itrc)- &
427 & tclm(i,j,k+1,itrc))- &
428 & (t(i,j,k ,nrhs,itrc)- &
429 & tclm(i,j,k ,itrc)))+ &
430 & cff*(tl_t(i,j,k+1,nrhs,itrc)- &
431 & tl_t(i,j,k ,nrhs,itrc))
433 dtdr(i,j,k2)=cff*(t(i,j,k+1,nrhs,itrc)- &
434 & t(i,j,k ,nrhs,itrc))
435 tl_dtdr(i,j,k2)=tl_cff*(t(i,j,k+1,nrhs,itrc)- &
436 & t(i,j,k ,nrhs,itrc))+ &
437 & cff*(tl_t(i,j,k+1,nrhs,itrc)- &
438 & tl_t(i,j,k ,nrhs,itrc))
441 dtdr(i,j,k2)=cff*(t(i,j,k+1,nrhs,itrc)- &
442 & t(i,j,k ,nrhs,itrc))
443 tl_dtdr(i,j,k2)=tl_cff*(t(i,j,k+1,nrhs,itrc)- &
444 & t(i,j,k ,nrhs,itrc))+ &
445 & cff*(tl_t(i,j,k+1,nrhs,itrc)- &
446 & tl_t(i,j,k ,nrhs,itrc))
448 fs(i,j,k2)=cff*(z_r(i,j,k+1)-z_r(i,j,k))
449 tl_fs(i,j,k2)=tl_cff*(z_r(i,j,k+1)-z_r(i,j,k))+ &
450 & cff*(tl_z_r(i,j,k+1)-tl_z_r(i,j,k))
462 cff=0.25_r8*(diff3d_r(i,j,k)+diff3d_r(i-1,j,k))* &
465 cff=0.25_r8*(diff2(i,j,itrc)+diff2(i-1,j,itrc))* &
479 & (((tl_hz(i,j,k)+tl_hz(i-1,j,k))* &
481 & 0.5_r8*(max(drdx(i,j,k1),0.0_r8)* &
484 & min(drdx(i,j,k1),0.0_r8)* &
486 & dtdr(i ,j,k1)))))+ &
487 & ((hz(i,j,k)+hz(i-1,j,k))* &
488 & (tl_dtdx(i,j,k1)- &
489 & 0.5_r8*(max(drdx(i,j,k1),0.0_r8)* &
490 & (tl_dtdr(i-1,j,k1)+ &
491 & tl_dtdr(i ,j,k2))+ &
492 & min(drdx(i,j,k1),0.0_r8)* &
493 & (tl_dtdr(i-1,j,k2)+ &
494 & tl_dtdr(i ,j,k1)))- &
496 & sign(0.5_r8, drdx(i,j,k1)))* &
498 & (dtdr(i-1,j,k1)+dtdr(i,j,k2))+ &
500 & sign(0.5_r8,-drdx(i,j,k1)))* &
502 & (dtdr(i-1,j,k2)+dtdr(i,j,k1))))))
508 cff=0.25_r8*(diff3d_r(i,j,k)+diff3d_r(i,j-1,k))* &
511 cff=0.25_r8*(diff2(i,j,itrc)+diff2(i,j-1,itrc))* &
525 & (((tl_hz(i,j,k)+tl_hz(i,j-1,k))* &
527 & 0.5_r8*(max(drde(i,j,k1),0.0_r8)* &
530 & min(drde(i,j,k1),0.0_r8)* &
532 & dtdr(i,j ,k1)))))+ &
533 & ((hz(i,j,k)+hz(i,j-1,k))* &
534 & (tl_dtde(i,j,k1)- &
535 & 0.5_r8*(max(drde(i,j,k1),0.0_r8)* &
536 & (tl_dtdr(i,j-1,k1)+ &
537 & tl_dtdr(i,j ,k2))+ &
538 & min(drde(i,j,k1),0.0_r8)* &
539 & (tl_dtdr(i,j-1,k2)+ &
540 & tl_dtdr(i,j ,k1)))- &
542 & sign(0.5_r8, drde(i,j,k1)))* &
544 & (dtdr(i,j-1,k1)+dtdr(i,j,k2))+ &
546 & sign(0.5_r8,-drde(i,j,k1)))* &
548 & (dtdr(i,j-1,k2)+dtdr(i,j,k1))))))
554 cff1=max(drdx(i ,j,k1),0.0_r8)
555 cff2=max(drdx(i+1,j,k2),0.0_r8)
556 cff3=min(drdx(i ,j,k2),0.0_r8)
557 cff4=min(drdx(i+1,j,k1),0.0_r8)
558 tl_cff1=(0.5_r8+sign(0.5_r8, drdx(i ,j,k1)))* &
560 tl_cff2=(0.5_r8+sign(0.5_r8, drdx(i+1,j,k2)))* &
562 tl_cff3=(0.5_r8+sign(0.5_r8,-drdx(i ,j,k2)))* &
564 tl_cff4=(0.5_r8+sign(0.5_r8,-drdx(i+1,j,k1)))* &
566 cff=cff1*(cff1*dtdr(i,j,k2)-dtdx(i ,j,k1))+ &
567 & cff2*(cff2*dtdr(i,j,k2)-dtdx(i+1,j,k2))+ &
568 & cff3*(cff3*dtdr(i,j,k2)-dtdx(i ,j,k2))+ &
569 & cff4*(cff4*dtdr(i,j,k2)-dtdx(i+1,j,k1))
570 tl_cff=tl_cff1*(cff1*dtdr(i ,j,k2)- &
572 & tl_cff2*(cff2*dtdr(i,j,k2)- &
574 & tl_cff3*(cff3*dtdr(i,j,k2)- &
576 & tl_cff4*(cff4*dtdr(i,j,k2)- &
578 & cff1*(tl_cff1*dtdr(i,j,k2)+ &
579 & cff1*tl_dtdr(i,j,k2)- &
580 & tl_dtdx(i ,j,k1))+ &
581 & cff2*(tl_cff2*dtdr(i,j,k2)+ &
582 & cff2*tl_dtdr(i,j,k2)- &
583 & tl_dtdx(i+1,j,k2))+ &
584 & cff3*(tl_cff3*dtdr(i,j,k2)+ &
585 & cff3*tl_dtdr(i,j,k2)- &
586 & tl_dtdx(i ,j,k2))+ &
587 & cff4*(tl_cff4*dtdr(i,j,k2)+ &
588 & cff4*tl_dtdr(i,j,k2)- &
590 cff1=max(drde(i,j ,k1),0.0_r8)
591 cff2=max(drde(i,j+1,k2),0.0_r8)
592 cff3=min(drde(i,j ,k2),0.0_r8)
593 cff4=min(drde(i,j+1,k1),0.0_r8)
594 tl_cff1=(0.5_r8+sign(0.5_r8, drde(i,j ,k1)))* &
596 tl_cff2=(0.5_r8+sign(0.5_r8, drde(i,j+1,k2)))* &
598 tl_cff3=(0.5_r8+sign(0.5_r8,-drde(i,j ,k2)))* &
600 tl_cff4=(0.5_r8+sign(0.5_r8,-drde(i,j+1,k1)))* &
603 & cff1*(cff1*dtdr(i,j,k2)-dtde(i,j ,k1))+ &
604 & cff2*(cff2*dtdr(i,j,k2)-dtde(i,j+1,k2))+ &
605 & cff3*(cff3*dtdr(i,j,k2)-dtde(i,j ,k2))+ &
606 & cff4*(cff4*dtdr(i,j,k2)-dtde(i,j+1,k1))
608 & tl_cff1*(cff1*dtdr(i,j,k2)- &
610 & tl_cff2*(cff2*dtdr(i,j,k2)- &
612 & tl_cff3*(cff3*dtdr(i,j,k2)- &
614 & tl_cff4*(cff4*dtdr(i,j,k2)- &
616 & cff1*(tl_cff1*dtdr(i,j,k2)+ &
617 & cff1*tl_dtdr(i,j,k2)- &
618 & tl_dtde(i,j ,k1))+ &
619 & cff2*(tl_cff2*dtdr(i,j,k2)+ &
620 & cff2*tl_dtdr(i,j,k2)- &
621 & tl_dtde(i,j+1,k2))+ &
622 & cff3*(tl_cff3*dtdr(i,j,k2)+ &
623 & cff3*tl_dtdr(i,j,k2)- &
624 & tl_dtde(i,j ,k2))+ &
625 & cff4*(tl_cff4*dtdr(i,j,k2)+ &
626 & cff4*tl_dtdr(i,j,k2)- &
631 tl_fs(i,j,k2)=0.5_r8*diff3d_r(i,j,k)* &
632 & (tl_cff*fs(i,j,k2)+ &
637 tl_fs(i,j,k2)=0.5_r8*diff2(i,j,itrc)* &
638 & (tl_cff*fs(i,j,k2)+ &
654 tl_cff=
dt(ng)*pm(i,j)*pn(i,j)* &
655 & (tl_fx(i+1,j)-tl_fx(i,j)+ &
656 & tl_fe(i,j+1)-tl_fe(i,j))+ &
657 &
dt(ng)*(tl_fs(i,j,k2)-tl_fs(i,j,k1))
660 tl_t(i,j,k,nnew,itrc)=tl_t(i,j,k,nnew,itrc)+tl_cff