26 |
C == Global variables === |
C == Global variables === |
27 |
#include "SIZE.h" |
#include "SIZE.h" |
28 |
#include "EEPARAMS.h" |
#include "EEPARAMS.h" |
|
#include "CG2D.h" |
|
29 |
#include "PARAMS.h" |
#include "PARAMS.h" |
30 |
#include "DYNVARS.h" |
#include "DYNVARS.h" |
31 |
#include "GRID.h" |
#include "GRID.h" |
39 |
# include "KPP.h" |
# include "KPP.h" |
40 |
#endif |
#endif |
41 |
|
|
42 |
#ifdef INCLUDE_DIAGNOSTICS_INTERFACE_CODE |
#ifdef ALLOW_TIMEAVE |
43 |
#include "AVER.h" |
#include "TIMEAVE_STATV.h" |
44 |
#endif |
#endif |
45 |
|
|
46 |
C == Routine arguments == |
C == Routine arguments == |
67 |
C rhoK, rhoKM1 - Density at current level, and level above |
C rhoK, rhoKM1 - Density at current level, and level above |
68 |
C phiHyd - Hydrostatic part of the potential phiHydi. |
C phiHyd - Hydrostatic part of the potential phiHydi. |
69 |
C In z coords phiHydiHyd is the hydrostatic |
C In z coords phiHydiHyd is the hydrostatic |
70 |
C pressure anomaly |
C Potential (=pressure/rho0) anomaly |
71 |
C In p coords phiHydiHyd is the geopotential |
C In p coords phiHydiHyd is the geopotential |
72 |
C surface height |
C surface height anomaly. |
|
C anomaly. |
|
73 |
C phiSurfX, - gradient of Surface potentiel (Pressure/rho, ocean) |
C phiSurfX, - gradient of Surface potentiel (Pressure/rho, ocean) |
74 |
C phiSurfY or geopotentiel (atmos) in X and Y direction |
C phiSurfY or geopotentiel (atmos) in X and Y direction |
75 |
C KappaRT, - Total diffusion in vertical for T and S. |
C KappaRT, - Total diffusion in vertical for T and S. |
119 |
c EXTERNAL DIFFERENT_MULTIPLE |
c EXTERNAL DIFFERENT_MULTIPLE |
120 |
Cjmc(end) |
Cjmc(end) |
121 |
|
|
|
#ifdef ALLOW_AUTODIFF_TAMC |
|
|
INTEGER isbyte |
|
|
PARAMETER( isbyte = 4 ) |
|
|
|
|
|
INTEGER act1, act2, act3, act4 |
|
|
INTEGER max1, max2, max3 |
|
|
INTEGER iikey, kkey |
|
|
INTEGER maximpl |
|
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
|
|
|
|
122 |
C--- The algorithm... |
C--- The algorithm... |
123 |
C |
C |
124 |
C "Correction Step" |
C "Correction Step" |
262 |
jMax = sNy+OLy |
jMax = sNy+OLy |
263 |
|
|
264 |
|
|
265 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
266 |
|
CADJ STORE theta(:,:,:,bi,bj) = comlev1_bibj, key = ikey, byte = isbyte |
267 |
|
CADJ STORE salt (:,:,:,bi,bj) = comlev1_bibj, key = ikey, byte = isbyte |
268 |
|
CADJ STORE uvel(:,:,:,bi,bj) = comlev1_bibj, key = ikey, byte = isbyte |
269 |
|
CADJ STORE vvel (:,:,:,bi,bj) = comlev1_bibj, key = ikey, byte = isbyte |
270 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
271 |
|
|
272 |
C-- Start of diagnostic loop |
C-- Start of diagnostic loop |
273 |
DO k=Nr,1,-1 |
DO k=Nr,1,-1 |
274 |
|
|
275 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
276 |
C? Patrick, is this formula correct now that we change the loop range? |
C? Patrick, is this formula correct now that we change the loop range? |
277 |
C? Do we still need this? |
C? Do we still need this? |
278 |
kkey = (ikey-1)*(Nr-2+1) + (k-2) + 1 |
cph kkey formula corrected. |
279 |
|
cph Needed for rhok, rhokm1, in the case useGMREDI. |
280 |
|
kkey = (ikey-1)*Nr + k |
281 |
|
CADJ STORE rhokm1(:,:) = comlev1_bibj_k , key = kkey, byte = isbyte |
282 |
|
CADJ STORE rhok (:,:) = comlev1_bibj_k , key = kkey, byte = isbyte |
283 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
284 |
|
|
285 |
C-- Integrate continuity vertically for vertical velocity |
C-- Integrate continuity vertically for vertical velocity |
372 |
IF (useKPP) THEN |
IF (useKPP) THEN |
373 |
CALL KPP_CALC( |
CALL KPP_CALC( |
374 |
I bi, bj, myTime, myThid ) |
I bi, bj, myTime, myThid ) |
375 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
376 |
|
ELSE |
377 |
|
DO j=1-OLy,sNy+OLy |
378 |
|
DO i=1-OLx,sNx+OLx |
379 |
|
KPPhbl (i,j,bi,bj) = 1.0 |
380 |
|
KPPfrac(i,j,bi,bj) = 0.0 |
381 |
|
DO k = 1,Nr |
382 |
|
KPPghat (i,j,k,bi,bj) = 0.0 |
383 |
|
KPPviscAz (i,j,k,bi,bj) = viscAz |
384 |
|
KPPdiffKzT(i,j,k,bi,bj) = diffKzT |
385 |
|
KPPdiffKzS(i,j,k,bi,bj) = diffKzS |
386 |
|
ENDDO |
387 |
|
ENDDO |
388 |
|
ENDDO |
389 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
390 |
ENDIF |
ENDIF |
391 |
|
|
392 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
393 |
|
CADJ STORE KPPghat (:,:,:,bi,bj) |
394 |
|
CADJ & , KPPviscAz (:,:,:,bi,bj) |
395 |
|
CADJ & , KPPdiffKzT(:,:,:,bi,bj) |
396 |
|
CADJ & , KPPdiffKzS(:,:,:,bi,bj) |
397 |
|
CADJ & , KPPfrac (:,: ,bi,bj) |
398 |
|
CADJ & = comlev1_bibj, key=ikey, byte=isbyte |
399 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
400 |
|
|
401 |
#endif /* ALLOW_KPP */ |
#endif /* ALLOW_KPP */ |
402 |
|
|
403 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
437 |
jMax = sNy+OLy-1 |
jMax = sNy+OLy-1 |
438 |
|
|
439 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
440 |
CPatrick Is this formula correct? |
C? Patrick Is this formula correct? |
441 |
kkey = (ikey-1)*(Nr-1+1) + (k-1) + 1 |
cph Yes, but I rewrote it. |
442 |
CADJ STORE rTrans(:,:) = comlev1_bibj_k, key = kkey, byte = isbyte |
cph Also, the KappaR? need the index k! |
443 |
CADJ STORE KappaRT(:,:,:) = comlev1_bibj_k, key = kkey, byte = isbyte |
kkey = (ikey-1)*Nr + k |
444 |
CADJ STORE KappaRS(:,:,:) = comlev1_bibj_k, key = kkey, byte = isbyte |
CADJ STORE KappaRT(:,:,k) = comlev1_bibj_k, key = kkey, byte = isbyte |
445 |
|
CADJ STORE KappaRS(:,:,k) = comlev1_bibj_k, key = kkey, byte = isbyte |
446 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
447 |
|
|
448 |
C-- Get temporary terms used by tendency routines |
C-- Get temporary terms used by tendency routines |
510 |
|
|
511 |
|
|
512 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
513 |
CPatrick? What about this one? |
C? Patrick? What about this one? |
514 |
maximpl = 6 |
cph Keys iikey and idkey don't seem to be needed |
515 |
iikey = (ikey-1)*maximpl |
cph since storing occurs on different tape for each |
516 |
|
cph impldiff call anyways. |
517 |
|
cph Thus, common block comlev1_impl isn't needed either. |
518 |
|
cph Storing below needed in the case useGMREDI. |
519 |
|
iikey = (ikey-1)*maximpl |
520 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
521 |
|
|
522 |
C-- Implicit diffusion |
C-- Implicit diffusion |
525 |
IF (tempStepping) THEN |
IF (tempStepping) THEN |
526 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
527 |
idkey = iikey + 1 |
idkey = iikey + 1 |
528 |
|
CADJ STORE gTNm1(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
529 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
530 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
531 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
537 |
IF (saltStepping) THEN |
IF (saltStepping) THEN |
538 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
539 |
idkey = iikey + 2 |
idkey = iikey + 2 |
540 |
|
CADJ STORE gSNm1(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
541 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
542 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
543 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
564 |
jMin = 1-OLy+2 |
jMin = 1-OLy+2 |
565 |
jMax = sNy+OLy-1 |
jMax = sNy+OLy-1 |
566 |
|
|
567 |
C-- Explicit part of the Surface Pressure Gradient (add in TIMESTEP) |
C-- Explicit part of the Surface Potentiel Gradient (add in TIMESTEP) |
568 |
C (note: this loop will be replaced by CALL CALC_GRAD_ETA) |
C (note: this loop will be replaced by CALL CALC_GRAD_ETA) |
569 |
IF (implicSurfPress.NE.1.) THEN |
IF (implicSurfPress.NE.1.) THEN |
570 |
DO j=jMin,jMax |
CALL CALC_GRAD_PHI_SURF( |
571 |
DO i=iMin,iMax |
I bi,bj,iMin,iMax,jMin,jMax, |
572 |
phiSurfX(i,j) = _recip_dxC(i,j,bi,bj)*gBaro |
I etaN, |
573 |
& *(cg2d_x(i,j,bi,bj)-cg2d_x(i-1,j,bi,bj)) |
O phiSurfX,phiSurfY, |
574 |
phiSurfY(i,j) = _recip_dyC(i,j,bi,bj)*gBaro |
I myThid ) |
|
& *(cg2d_x(i,j,bi,bj)-cg2d_x(i,j-1,bi,bj)) |
|
|
ENDDO |
|
|
ENDDO |
|
575 |
ENDIF |
ENDIF |
576 |
|
|
577 |
C-- Start of dynamics loop |
C-- Start of dynamics loop |
645 |
IF (implicitViscosity.AND.momStepping) THEN |
IF (implicitViscosity.AND.momStepping) THEN |
646 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
647 |
idkey = iikey + 3 |
idkey = iikey + 3 |
648 |
|
CADJ STORE gUNm1(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
649 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
650 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
651 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
654 |
I myThid ) |
I myThid ) |
655 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
656 |
idkey = iikey + 4 |
idkey = iikey + 4 |
657 |
|
CADJ STORE gVNm1(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
658 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
659 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
660 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
674 |
#ifdef INCLUDE_CD_CODE |
#ifdef INCLUDE_CD_CODE |
675 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
676 |
idkey = iikey + 5 |
idkey = iikey + 5 |
677 |
|
CADJ STORE vVelD(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
678 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
679 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
680 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
683 |
I myThid ) |
I myThid ) |
684 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
685 |
idkey = iikey + 6 |
idkey = iikey + 6 |
686 |
|
CADJ STORE uVelD(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
687 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
688 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
689 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
702 |
c ENDIF |
c ENDIF |
703 |
Cjmc(end) |
Cjmc(end) |
704 |
|
|
705 |
#ifdef INCLUDE_DIAGNOSTICS_INTERFACE_CODE |
#ifdef ALLOW_TIMEAVE |
706 |
IF (taveFreq.GT.0.) THEN |
IF (taveFreq.GT.0.) THEN |
707 |
DO K=1,Nr |
CALL TIMEAVE_CUMULATE(phiHydtave, phiHyd, Nr, |
708 |
CALL TIMEAVER_1FLD_XYZ(phiHyd, phiHydtave, |
I deltaTclock, bi, bj, myThid) |
|
I deltaTclock, bi, bj, K, myThid) |
|
709 |
IF (ivdc_kappa.NE.0.) THEN |
IF (ivdc_kappa.NE.0.) THEN |
710 |
CALL TIMEAVER_1FLD_XYZ(ConvectCount, ConvectCountTave, |
CALL TIMEAVE_CUMULATE(ConvectCountTave, ConvectCount, Nr, |
711 |
I deltaTclock, bi, bj, K, myThid) |
I deltaTclock, bi, bj, myThid) |
712 |
ENDIF |
ENDIF |
|
ENDDO |
|
713 |
ENDIF |
ENDIF |
714 |
#endif /* INCLUDE_DIAGNOSTICS_INTERFACE_CODE */ |
#endif /* ALLOW_TIMEAVE */ |
715 |
|
|
716 |
ENDDO |
ENDDO |
717 |
ENDDO |
ENDDO |