3 |
|
|
4 |
#include "PACKAGES_CONFIG.h" |
#include "PACKAGES_CONFIG.h" |
5 |
#include "CPP_OPTIONS.h" |
#include "CPP_OPTIONS.h" |
6 |
|
#ifdef ALLOW_AUTODIFF |
7 |
|
# include "AUTODIFF_OPTIONS.h" |
8 |
|
#endif |
9 |
#ifdef ALLOW_MOM_COMMON |
#ifdef ALLOW_MOM_COMMON |
10 |
# include "MOM_COMMON_OPTIONS.h" |
# include "MOM_COMMON_OPTIONS.h" |
11 |
#endif |
#endif |
92 |
#ifdef ALLOW_CD_CODE |
#ifdef ALLOW_CD_CODE |
93 |
# include "CD_CODE_VARS.h" |
# include "CD_CODE_VARS.h" |
94 |
#endif |
#endif |
95 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF |
96 |
# include "tamc.h" |
# include "tamc.h" |
97 |
# include "tamc_keys.h" |
# include "tamc_keys.h" |
98 |
# include "FFIELDS.h" |
# include "FFIELDS.h" |
114 |
# ifdef ALLOW_MOM_FLUXFORM |
# ifdef ALLOW_MOM_FLUXFORM |
115 |
# include "MOM_FLUXFORM.h" |
# include "MOM_FLUXFORM.h" |
116 |
# endif |
# endif |
117 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF */ |
118 |
|
|
119 |
C !CALLING SEQUENCE: |
C !CALLING SEQUENCE: |
120 |
C DYNAMICS() |
C DYNAMICS() |
349 |
C point numbers. This prevents spurious hardware signals due to |
C point numbers. This prevents spurious hardware signals due to |
350 |
C uninitialised but inert locations. |
C uninitialised but inert locations. |
351 |
|
|
352 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF |
353 |
DO k=1,Nr |
DO k=1,Nr |
354 |
DO j=1-OLy,sNy+OLy |
DO j=1-OLy,sNy+OLy |
355 |
DO i=1-OLx,sNx+OLx |
DO i=1-OLx,sNx+OLx |
|
cph( |
|
356 |
c-- need some re-initialisation here to break dependencies |
c-- need some re-initialisation here to break dependencies |
|
cph) |
|
357 |
gU(i,j,k,bi,bj) = 0. _d 0 |
gU(i,j,k,bi,bj) = 0. _d 0 |
358 |
gV(i,j,k,bi,bj) = 0. _d 0 |
gV(i,j,k,bi,bj) = 0. _d 0 |
359 |
ENDDO |
ENDDO |
360 |
ENDDO |
ENDDO |
361 |
ENDDO |
ENDDO |
362 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF */ |
363 |
DO j=1-OLy,sNy+OLy |
DO j=1-OLy,sNy+OLy |
364 |
DO i=1-OLx,sNx+OLx |
DO i=1-OLx,sNx+OLx |
365 |
fVerU (i,j,1) = 0. _d 0 |
fVerU (i,j,1) = 0. _d 0 |
376 |
phiSurfY(i,j) = 0. _d 0 |
phiSurfY(i,j) = 0. _d 0 |
377 |
guDissip(i,j) = 0. _d 0 |
guDissip(i,j) = 0. _d 0 |
378 |
gvDissip(i,j) = 0. _d 0 |
gvDissip(i,j) = 0. _d 0 |
379 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF |
380 |
phiHydLow(i,j,bi,bj) = 0. _d 0 |
phiHydLow(i,j,bi,bj) = 0. _d 0 |
381 |
# if (defined NONLIN_FRSURF) && (defined ALLOW_MOM_FLUXFORM) |
# if (defined NONLIN_FRSURF) && (defined ALLOW_MOM_FLUXFORM) |
382 |
# ifndef DISABLE_RSTAR_CODE |
# ifndef DISABLE_RSTAR_CODE |
|
# ifndef ALLOW_AUTODIFF_OPENAD |
|
383 |
dWtransC(i,j,bi,bj) = 0. _d 0 |
dWtransC(i,j,bi,bj) = 0. _d 0 |
384 |
dWtransU(i,j,bi,bj) = 0. _d 0 |
dWtransU(i,j,bi,bj) = 0. _d 0 |
385 |
dWtransV(i,j,bi,bj) = 0. _d 0 |
dWtransV(i,j,bi,bj) = 0. _d 0 |
|
# endif |
|
386 |
# endif |
# endif |
387 |
# endif |
# endif |
388 |
#endif |
#endif /* ALLOW_AUTODIFF */ |
389 |
ENDDO |
ENDDO |
390 |
ENDDO |
ENDDO |
391 |
|
|
415 |
#endif /* ALLOW_KPP */ |
#endif /* ALLOW_KPP */ |
416 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
417 |
|
|
418 |
#if (defined INCLUDE_CALC_DIFFUSIVITY_CALL) && !(defined ALLOW_AUTODIFF) |
#ifndef ALLOW_AUTODIFF |
419 |
IF ( .NOT.momViscosity ) THEN |
IF ( .NOT.momViscosity ) THEN |
420 |
#endif /* INCLUDE_CALC_DIFFUSIVITY_CALL and not ALLOW_AUTODIFF */ |
#endif |
421 |
DO k=1,Nr |
DO k=1,Nr |
422 |
DO j=1-OLy,sNy+OLy |
DO j=1-OLy,sNy+OLy |
423 |
DO i=1-OLx,sNx+OLx |
DO i=1-OLx,sNx+OLx |
426 |
ENDDO |
ENDDO |
427 |
ENDDO |
ENDDO |
428 |
ENDDO |
ENDDO |
429 |
|
#ifndef ALLOW_AUTODIFF |
430 |
|
ENDIF |
431 |
|
#endif |
432 |
#ifdef INCLUDE_CALC_DIFFUSIVITY_CALL |
#ifdef INCLUDE_CALC_DIFFUSIVITY_CALL |
433 |
C-- Calculate the total vertical viscosity |
C-- Calculate the total vertical viscosity |
|
#ifdef ALLOW_AUTODIFF |
|
434 |
IF ( momViscosity ) THEN |
IF ( momViscosity ) THEN |
|
#else |
|
|
ELSE |
|
|
#endif |
|
435 |
CALL CALC_VISCOSITY( |
CALL CALC_VISCOSITY( |
436 |
I bi,bj, iMin,iMax,jMin,jMax, |
I bi,bj, iMin,iMax,jMin,jMax, |
437 |
O KappaRU, KappaRV, |
O KappaRU, KappaRV, |
483 |
|
|
484 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
485 |
kkey = (idynkey-1)*Nr + k |
kkey = (idynkey-1)*Nr + k |
|
c |
|
486 |
CADJ STORE totPhiHyd (:,:,k,bi,bj) |
CADJ STORE totPhiHyd (:,:,k,bi,bj) |
487 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
488 |
CADJ STORE phiHydLow (:,:,bi,bj) |
CADJ STORE phiHydLow (:,:,bi,bj) |
501 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
502 |
CADJ STORE phiHydF (:,:) |
CADJ STORE phiHydF (:,:) |
503 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
CADJ STORE guDissip (:,:) |
|
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
|
CADJ STORE gvDissip (:,:) |
|
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
|
CADJ STORE fVerU (:,:,:) |
|
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
|
CADJ STORE fVerV (:,:,:) |
|
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
504 |
CADJ STORE gU(:,:,k,bi,bj) |
CADJ STORE gU(:,:,k,bi,bj) |
505 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
506 |
CADJ STORE gV(:,:,k,bi,bj) |
CADJ STORE gV(:,:,k,bi,bj) |
530 |
CADJ STORE vVelD(:,:,k,bi,bj) |
CADJ STORE vVelD(:,:,k,bi,bj) |
531 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
532 |
# endif |
# endif |
533 |
# endif |
# endif /* NONLIN_FRSURF */ |
|
# ifdef ALLOW_DEPTH_CONTROL |
|
|
CADJ STORE fVerU (:,:,:) |
|
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
|
CADJ STORE fVerV (:,:,:) |
|
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
|
# endif |
|
534 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
535 |
|
|
536 |
C-- Integrate hydrostatic balance for phiHyd with BC of phiHyd(z=0)=0 |
C-- Integrate hydrostatic balance for phiHyd with BC of phiHyd(z=0)=0 |
537 |
IF ( implicitIntGravWave ) THEN |
CALL CALC_PHI_HYD( |
|
CALL CALC_PHI_HYD( |
|
|
I bi,bj,iMin,iMax,jMin,jMax,k, |
|
|
I gT, gS, |
|
|
U phiHydF, |
|
|
O phiHydC, dPhiHydX, dPhiHydY, |
|
|
I myTime, myIter, myThid ) |
|
|
ELSE |
|
|
CALL CALC_PHI_HYD( |
|
538 |
I bi,bj,iMin,iMax,jMin,jMax,k, |
I bi,bj,iMin,iMax,jMin,jMax,k, |
539 |
I theta, salt, |
I theta, salt, |
540 |
U phiHydF, |
U phiHydF, |
541 |
O phiHydC, dPhiHydX, dPhiHydY, |
O phiHydC, dPhiHydX, dPhiHydY, |
542 |
I myTime, myIter, myThid ) |
I myTime, myIter, myThid ) |
|
ENDIF |
|
543 |
#ifdef ALLOW_DIAGNOSTICS |
#ifdef ALLOW_DIAGNOSTICS |
544 |
IF ( dPhiHydDiagIsOn ) THEN |
IF ( dPhiHydDiagIsOn ) THEN |
545 |
tmpFac = -1. _d 0 |
tmpFac = -1. _d 0 |
553 |
C-- Calculate accelerations in the momentum equations (gU, gV, ...) |
C-- Calculate accelerations in the momentum equations (gU, gV, ...) |
554 |
C and step forward storing the result in gU, gV, etc... |
C and step forward storing the result in gU, gV, etc... |
555 |
IF ( momStepping ) THEN |
IF ( momStepping ) THEN |
556 |
|
#ifdef ALLOW_AUTODIFF |
557 |
|
DO j=1-OLy,sNy+OLy |
558 |
|
DO i=1-OLx,sNx+OLx |
559 |
|
guDissip(i,j) = 0. _d 0 |
560 |
|
gvDissip(i,j) = 0. _d 0 |
561 |
|
ENDDO |
562 |
|
ENDDO |
563 |
|
#endif /* ALLOW_AUTODIFF */ |
564 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
565 |
# ifdef NONLIN_FRSURF |
# if (defined NONLIN_FRSURF) && (defined ALLOW_MOM_FLUXFORM) |
566 |
# if (defined ALLOW_MOM_FLUXFORM) && !(defined DISABLE_RSTAR_CODE) |
# ifndef DISABLE_RSTAR_CODE |
567 |
CADJ STORE dWtransC(:,:,bi,bj) |
CADJ STORE dWtransC(:,:,bi,bj) |
568 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
569 |
CADJ STORE dWtransU(:,:,bi,bj) |
CADJ STORE dWtransU(:,:,bi,bj) |
571 |
CADJ STORE dWtransV(:,:,bi,bj) |
CADJ STORE dWtransV(:,:,bi,bj) |
572 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
573 |
# endif |
# endif |
574 |
CADJ STORE fVerU(:,:,:) |
# endif /* NONLIN_FRSURF and ALLOW_MOM_FLUXFORM */ |
575 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
# if (defined NONLIN_FRSURF) || (defined ALLOW_DEPTH_CONTROL) |
576 |
CADJ STORE fVerV(:,:,:) |
CADJ STORE fVerU(:,:,:) = comlev1_bibj_k, key=kkey, byte=isbyte |
577 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ STORE fVerV(:,:,:) = comlev1_bibj_k, key=kkey, byte=isbyte |
578 |
# endif /* NONLIN_FRSURF */ |
# endif |
579 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
580 |
IF (.NOT. vectorInvariantMomentum) THEN |
IF (.NOT. vectorInvariantMomentum) THEN |
581 |
#ifdef ALLOW_MOM_FLUXFORM |
#ifdef ALLOW_MOM_FLUXFORM |
644 |
IF ( implicitViscosity ) THEN |
IF ( implicitViscosity ) THEN |
645 |
#endif /* INCLUDE_IMPLVERTADV_CODE */ |
#endif /* INCLUDE_IMPLVERTADV_CODE */ |
646 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
|
CADJ STORE KappaRU(:,:,:) = comlev1_bibj , key=idynkey, byte=isbyte |
|
647 |
CADJ STORE gU(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
CADJ STORE gU(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
648 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
649 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
650 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
651 |
I -1, KappaRU, recip_hFacW(1-OLx,1-OLy,1,bi,bj), |
I -1, KappaRU, recip_hFacW(1-OLx,1-OLy,1,bi,bj), |
652 |
U gU, |
U gU(1-OLx,1-OLy,1,bi,bj), |
653 |
I myThid ) |
I myThid ) |
654 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
|
CADJ STORE KappaRV(:,:,:) = comlev1_bibj , key=idynkey, byte=isbyte |
|
655 |
CADJ STORE gV(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
CADJ STORE gV(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
656 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
657 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
658 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
659 |
I -2, KappaRV, recip_hFacS(1-OLx,1-OLy,1,bi,bj), |
I -2, KappaRV, recip_hFacS(1-OLx,1-OLy,1,bi,bj), |
660 |
U gV, |
U gV(1-OLx,1-OLy,1,bi,bj), |
661 |
I myThid ) |
I myThid ) |
662 |
ENDIF |
ENDIF |
663 |
|
|
681 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
682 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
683 |
I 0, KappaRU, recip_hFacW(1-OLx,1-OLy,1,bi,bj), |
I 0, KappaRU, recip_hFacW(1-OLx,1-OLy,1,bi,bj), |
684 |
U vVelD, |
U vVelD(1-OLx,1-OLy,1,bi,bj), |
685 |
I myThid ) |
I myThid ) |
686 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
687 |
CADJ STORE uVelD(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
CADJ STORE uVelD(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
689 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
690 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
691 |
I 0, KappaRV, recip_hFacS(1-OLx,1-OLy,1,bi,bj), |
I 0, KappaRV, recip_hFacS(1-OLx,1-OLy,1,bi,bj), |
692 |
U uVelD, |
U uVelD(1-OLx,1-OLy,1,bi,bj), |
693 |
I myThid ) |
I myThid ) |
694 |
ENDIF |
ENDIF |
695 |
#endif /* ALLOW_CD_CODE */ |
#endif /* ALLOW_CD_CODE */ |