3 |
|
|
4 |
#include "PACKAGES_CONFIG.h" |
#include "PACKAGES_CONFIG.h" |
5 |
#include "CPP_OPTIONS.h" |
#include "CPP_OPTIONS.h" |
6 |
|
#ifdef ALLOW_MOM_COMMON |
7 |
|
# include "MOM_COMMON_OPTIONS.h" |
8 |
|
#endif |
9 |
#ifdef ALLOW_OBCS |
#ifdef ALLOW_OBCS |
10 |
# include "OBCS_OPTIONS.h" |
# include "OBCS_OPTIONS.h" |
11 |
#endif |
#endif |
83 |
#include "PARAMS.h" |
#include "PARAMS.h" |
84 |
#include "GRID.h" |
#include "GRID.h" |
85 |
#include "DYNVARS.h" |
#include "DYNVARS.h" |
86 |
|
#ifdef ALLOW_MOM_COMMON |
87 |
|
# include "MOM_VISC.h" |
88 |
|
#endif |
89 |
#ifdef ALLOW_CD_CODE |
#ifdef ALLOW_CD_CODE |
90 |
# include "CD_CODE_VARS.h" |
# include "CD_CODE_VARS.h" |
91 |
#endif |
#endif |
122 |
C | |
C | |
123 |
C |-- CALC_VISCOSITY |
C |-- CALC_VISCOSITY |
124 |
C | |
C | |
125 |
|
C |-- MOM_CALC_3D_STRAIN |
126 |
|
C | |
127 |
C |-- CALC_EDDY_STRESS |
C |-- CALC_EDDY_STRESS |
128 |
C | |
C | |
129 |
C |-- CALC_PHI_HYD |
C |-- CALC_PHI_HYD |
132 |
C | |
C | |
133 |
C |-- MOM_VECINV |
C |-- MOM_VECINV |
134 |
C | |
C | |
135 |
|
C |-- MOM_CALC_SMAG_3D |
136 |
|
C |-- MOM_UV_SMAG_3D |
137 |
|
C | |
138 |
C |-- TIMESTEP |
C |-- TIMESTEP |
139 |
C | |
C | |
140 |
C |-- MOM_U_IMPLICIT_R |
C |-- MOM_U_IMPLICIT_R |
201 |
_RL gvDissip(1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
_RL gvDissip(1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
202 |
_RL KappaRU (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr) |
_RL KappaRU (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr) |
203 |
_RL KappaRV (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr) |
_RL KappaRV (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr) |
204 |
|
#ifdef ALLOW_SMAG_3D |
205 |
|
C str11 :: strain component Vxx @ grid-cell center |
206 |
|
C str22 :: strain component Vyy @ grid-cell center |
207 |
|
C str33 :: strain component Vzz @ grid-cell center |
208 |
|
C str12 :: strain component Vxy @ grid-cell corner |
209 |
|
C str13 :: strain component Vxz @ above uVel |
210 |
|
C str23 :: strain component Vyz @ above vVel |
211 |
|
C viscAh3d_00 :: Smagorinsky viscosity @ grid-cell center |
212 |
|
C viscAh3d_12 :: Smagorinsky viscosity @ grid-cell corner |
213 |
|
C viscAh3d_13 :: Smagorinsky viscosity @ above uVel |
214 |
|
C viscAh3d_23 :: Smagorinsky viscosity @ above vVel |
215 |
|
C addDissU :: zonal momentum tendency from 3-D Smag. viscosity |
216 |
|
C addDissV :: merid momentum tendency from 3-D Smag. viscosity |
217 |
|
_RL str11(1-OLx:sNx+OLx,1-OLy:sNy+OLy, Nr ) |
218 |
|
_RL str22(1-OLx:sNx+OLx,1-OLy:sNy+OLy, Nr ) |
219 |
|
_RL str33(1-OLx:sNx+OLx,1-OLy:sNy+OLy, Nr ) |
220 |
|
_RL str12(1-OLx:sNx+OLx,1-OLy:sNy+OLy, Nr ) |
221 |
|
_RL str13(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr+1) |
222 |
|
_RL str23(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr+1) |
223 |
|
_RL viscAh3d_00(1-OLx:sNx+OLx,1-OLy:sNy+OLy, Nr ) |
224 |
|
_RL viscAh3d_12(1-OLx:sNx+OLx,1-OLy:sNy+OLy, Nr ) |
225 |
|
_RL viscAh3d_13(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr+1) |
226 |
|
_RL viscAh3d_23(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr+1) |
227 |
|
_RL addDissU(1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
228 |
|
_RL addDissV(1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
229 |
|
#elif ( defined ALLOW_NONHYDROSTATIC ) |
230 |
|
_RL str13(1), str23(1), str33(1) |
231 |
|
_RL viscAh3d_00(1), viscAh3d_13(1), viscAh3d_23(1) |
232 |
|
#endif |
233 |
|
|
|
INTEGER iMin, iMax |
|
|
INTEGER jMin, jMax |
|
234 |
INTEGER bi, bj |
INTEGER bi, bj |
235 |
INTEGER i, j |
INTEGER i, j |
236 |
INTEGER k, km1, kp1, kUp, kDown |
INTEGER k, km1, kp1, kUp, kDown |
237 |
|
INTEGER iMin, iMax |
238 |
|
INTEGER jMin, jMax |
239 |
|
PARAMETER( iMin = 0 , iMax = sNx+1 ) |
240 |
|
PARAMETER( jMin = 0 , jMax = sNy+1 ) |
241 |
|
|
242 |
#ifdef ALLOW_DIAGNOSTICS |
#ifdef ALLOW_DIAGNOSTICS |
243 |
LOGICAL dPhiHydDiagIsOn |
LOGICAL dPhiHydDiagIsOn |
346 |
C point numbers. This prevents spurious hardware signals due to |
C point numbers. This prevents spurious hardware signals due to |
347 |
C uninitialised but inert locations. |
C uninitialised but inert locations. |
348 |
|
|
349 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF |
350 |
DO k=1,Nr |
DO k=1,Nr |
351 |
DO j=1-OLy,sNy+OLy |
DO j=1-OLy,sNy+OLy |
352 |
DO i=1-OLx,sNx+OLx |
DO i=1-OLx,sNx+OLx |
|
cph( |
|
353 |
c-- need some re-initialisation here to break dependencies |
c-- need some re-initialisation here to break dependencies |
|
cph) |
|
354 |
gU(i,j,k,bi,bj) = 0. _d 0 |
gU(i,j,k,bi,bj) = 0. _d 0 |
355 |
gV(i,j,k,bi,bj) = 0. _d 0 |
gV(i,j,k,bi,bj) = 0. _d 0 |
356 |
ENDDO |
ENDDO |
357 |
ENDDO |
ENDDO |
358 |
ENDDO |
ENDDO |
359 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF */ |
360 |
DO j=1-OLy,sNy+OLy |
DO j=1-OLy,sNy+OLy |
361 |
DO i=1-OLx,sNx+OLx |
DO i=1-OLx,sNx+OLx |
362 |
fVerU (i,j,1) = 0. _d 0 |
fVerU (i,j,1) = 0. _d 0 |
373 |
phiSurfY(i,j) = 0. _d 0 |
phiSurfY(i,j) = 0. _d 0 |
374 |
guDissip(i,j) = 0. _d 0 |
guDissip(i,j) = 0. _d 0 |
375 |
gvDissip(i,j) = 0. _d 0 |
gvDissip(i,j) = 0. _d 0 |
376 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF |
377 |
phiHydLow(i,j,bi,bj) = 0. _d 0 |
phiHydLow(i,j,bi,bj) = 0. _d 0 |
378 |
# if (defined NONLIN_FRSURF) && (defined ALLOW_MOM_FLUXFORM) |
# if (defined NONLIN_FRSURF) && (defined ALLOW_MOM_FLUXFORM) |
379 |
# ifndef DISABLE_RSTAR_CODE |
# ifndef DISABLE_RSTAR_CODE |
|
# ifndef ALLOW_AUTODIFF_OPENAD |
|
380 |
dWtransC(i,j,bi,bj) = 0. _d 0 |
dWtransC(i,j,bi,bj) = 0. _d 0 |
381 |
dWtransU(i,j,bi,bj) = 0. _d 0 |
dWtransU(i,j,bi,bj) = 0. _d 0 |
382 |
dWtransV(i,j,bi,bj) = 0. _d 0 |
dWtransV(i,j,bi,bj) = 0. _d 0 |
|
# endif |
|
383 |
# endif |
# endif |
384 |
# endif |
# endif |
385 |
#endif |
#endif /* ALLOW_AUTODIFF */ |
386 |
ENDDO |
ENDDO |
387 |
ENDDO |
ENDDO |
388 |
|
|
389 |
C-- Start computation of dynamics |
C-- Start computation of dynamics |
|
iMin = 0 |
|
|
iMax = sNx+1 |
|
|
jMin = 0 |
|
|
jMax = sNy+1 |
|
390 |
|
|
391 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
392 |
CADJ STORE wVel (:,:,:,bi,bj) = |
CADJ STORE wVel (:,:,:,bi,bj) = |
412 |
#endif /* ALLOW_KPP */ |
#endif /* ALLOW_KPP */ |
413 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
414 |
|
|
415 |
#if (defined INCLUDE_CALC_DIFFUSIVITY_CALL) && !(defined ALLOW_AUTODIFF) |
#ifndef ALLOW_AUTODIFF |
416 |
IF ( .NOT.momViscosity ) THEN |
IF ( .NOT.momViscosity ) THEN |
417 |
#endif /* INCLUDE_CALC_DIFFUSIVITY_CALL and not ALLOW_AUTODIFF */ |
#endif |
418 |
DO k=1,Nr |
DO k=1,Nr |
419 |
DO j=1-OLy,sNy+OLy |
DO j=1-OLy,sNy+OLy |
420 |
DO i=1-OLx,sNx+OLx |
DO i=1-OLx,sNx+OLx |
423 |
ENDDO |
ENDDO |
424 |
ENDDO |
ENDDO |
425 |
ENDDO |
ENDDO |
426 |
|
#ifndef ALLOW_AUTODIFF |
427 |
|
ENDIF |
428 |
|
#endif |
429 |
#ifdef INCLUDE_CALC_DIFFUSIVITY_CALL |
#ifdef INCLUDE_CALC_DIFFUSIVITY_CALL |
430 |
C-- Calculate the total vertical viscosity |
C-- Calculate the total vertical viscosity |
|
#ifdef ALLOW_AUTODIFF |
|
431 |
IF ( momViscosity ) THEN |
IF ( momViscosity ) THEN |
|
#else |
|
|
ELSE |
|
|
#endif |
|
432 |
CALL CALC_VISCOSITY( |
CALL CALC_VISCOSITY( |
433 |
I bi,bj, iMin,iMax,jMin,jMax, |
I bi,bj, iMin,iMax,jMin,jMax, |
434 |
O KappaRU, KappaRV, |
O KappaRU, KappaRV, |
436 |
ENDIF |
ENDIF |
437 |
#endif /* INCLUDE_CALC_DIFFUSIVITY_CALL */ |
#endif /* INCLUDE_CALC_DIFFUSIVITY_CALL */ |
438 |
|
|
439 |
|
#ifdef ALLOW_SMAG_3D |
440 |
|
IF ( useSmag3D ) THEN |
441 |
|
CALL MOM_CALC_3D_STRAIN( |
442 |
|
O str11, str22, str33, str12, str13, str23, |
443 |
|
I bi, bj, myThid ) |
444 |
|
ENDIF |
445 |
|
#endif /* ALLOW_SMAG_3D */ |
446 |
|
|
447 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
448 |
CADJ STORE KappaRU(:,:,:) |
CADJ STORE KappaRU(:,:,:) |
449 |
CADJ & = comlev1_bibj, key=idynkey, byte=isbyte |
CADJ & = comlev1_bibj, key=idynkey, byte=isbyte |
480 |
|
|
481 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
482 |
kkey = (idynkey-1)*Nr + k |
kkey = (idynkey-1)*Nr + k |
|
c |
|
483 |
CADJ STORE totPhiHyd (:,:,k,bi,bj) |
CADJ STORE totPhiHyd (:,:,k,bi,bj) |
484 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
CADJ STORE phiHydLow (:,:,bi,bj) |
|
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
485 |
CADJ STORE theta (:,:,k,bi,bj) |
CADJ STORE theta (:,:,k,bi,bj) |
486 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
487 |
CADJ STORE salt (:,:,k,bi,bj) |
CADJ STORE salt (:,:,k,bi,bj) |
496 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
497 |
CADJ STORE phiHydF (:,:) |
CADJ STORE phiHydF (:,:) |
498 |
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 |
|
499 |
CADJ STORE gU(:,:,k,bi,bj) |
CADJ STORE gU(:,:,k,bi,bj) |
500 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
501 |
CADJ STORE gV(:,:,k,bi,bj) |
CADJ STORE gV(:,:,k,bi,bj) |
525 |
CADJ STORE vVelD(:,:,k,bi,bj) |
CADJ STORE vVelD(:,:,k,bi,bj) |
526 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
527 |
# endif |
# endif |
528 |
# 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 |
|
529 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
530 |
|
|
531 |
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 |
557 |
C-- Calculate accelerations in the momentum equations (gU, gV, ...) |
C-- Calculate accelerations in the momentum equations (gU, gV, ...) |
558 |
C and step forward storing the result in gU, gV, etc... |
C and step forward storing the result in gU, gV, etc... |
559 |
IF ( momStepping ) THEN |
IF ( momStepping ) THEN |
560 |
|
#ifdef ALLOW_AUTODIFF |
561 |
|
DO j=1-OLy,sNy+OLy |
562 |
|
DO i=1-OLx,sNx+OLx |
563 |
|
guDissip(i,j) = 0. _d 0 |
564 |
|
gvDissip(i,j) = 0. _d 0 |
565 |
|
ENDDO |
566 |
|
ENDDO |
567 |
|
#endif /* ALLOW_AUTODIFF */ |
568 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
569 |
# ifdef NONLIN_FRSURF |
# if (defined NONLIN_FRSURF) && (defined ALLOW_MOM_FLUXFORM) |
570 |
# if (defined ALLOW_MOM_FLUXFORM) && !(defined DISABLE_RSTAR_CODE) |
# ifndef DISABLE_RSTAR_CODE |
571 |
CADJ STORE dWtransC(:,:,bi,bj) |
CADJ STORE dWtransC(:,:,bi,bj) |
572 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
573 |
CADJ STORE dWtransU(:,:,bi,bj) |
CADJ STORE dWtransU(:,:,bi,bj) |
575 |
CADJ STORE dWtransV(:,:,bi,bj) |
CADJ STORE dWtransV(:,:,bi,bj) |
576 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
577 |
# endif |
# endif |
578 |
CADJ STORE fVerU(:,:,:) |
# endif /* NONLIN_FRSURF and ALLOW_MOM_FLUXFORM */ |
579 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
# if (defined NONLIN_FRSURF) || (defined ALLOW_DEPTH_CONTROL) |
580 |
CADJ STORE fVerV(:,:,:) |
CADJ STORE fVerU(:,:,:) = comlev1_bibj_k, key=kkey, byte=isbyte |
581 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ STORE fVerV(:,:,:) = comlev1_bibj_k, key=kkey, byte=isbyte |
582 |
# endif /* NONLIN_FRSURF */ |
# endif |
583 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
584 |
IF (.NOT. vectorInvariantMomentum) THEN |
IF (.NOT. vectorInvariantMomentum) THEN |
585 |
#ifdef ALLOW_MOM_FLUXFORM |
#ifdef ALLOW_MOM_FLUXFORM |
603 |
#endif |
#endif |
604 |
ENDIF |
ENDIF |
605 |
|
|
606 |
|
#ifdef ALLOW_SMAG_3D |
607 |
|
IF ( useSmag3D ) THEN |
608 |
|
CALL MOM_CALC_SMAG_3D( |
609 |
|
I str11, str22, str33, str12, str13, str23, |
610 |
|
O viscAh3d_00, viscAh3d_12, viscAh3d_13, viscAh3d_23, |
611 |
|
I smag3D_hLsC, smag3D_hLsW, smag3D_hLsS, smag3D_hLsZ, |
612 |
|
I k, bi, bj, myThid ) |
613 |
|
CALL MOM_UV_SMAG_3D( |
614 |
|
I str11, str22, str12, str13, str23, |
615 |
|
I viscAh3d_00, viscAh3d_12, viscAh3d_13, viscAh3d_23, |
616 |
|
O addDissU, addDissV, |
617 |
|
I iMin,iMax,jMin,jMax, k, bi, bj, myThid ) |
618 |
|
DO j= jMin,jMax |
619 |
|
DO i= iMin,iMax |
620 |
|
guDissip(i,j) = guDissip(i,j) + addDissU(i,j) |
621 |
|
gvDissip(i,j) = gvDissip(i,j) + addDissV(i,j) |
622 |
|
ENDDO |
623 |
|
ENDDO |
624 |
|
ENDIF |
625 |
|
#endif /* ALLOW_SMAG_3D */ |
626 |
|
|
627 |
CALL TIMESTEP( |
CALL TIMESTEP( |
628 |
I bi,bj,iMin,iMax,jMin,jMax,k, |
I bi,bj,iMin,iMax,jMin,jMax,k, |
629 |
I dPhiHydX,dPhiHydY, phiSurfX, phiSurfY, |
I dPhiHydX,dPhiHydY, phiSurfX, phiSurfY, |
648 |
IF ( implicitViscosity ) THEN |
IF ( implicitViscosity ) THEN |
649 |
#endif /* INCLUDE_IMPLVERTADV_CODE */ |
#endif /* INCLUDE_IMPLVERTADV_CODE */ |
650 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
|
CADJ STORE KappaRU(:,:,:) = comlev1_bibj , key=idynkey, byte=isbyte |
|
651 |
CADJ STORE gU(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
CADJ STORE gU(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
652 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
653 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
656 |
U gU, |
U gU, |
657 |
I myThid ) |
I myThid ) |
658 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
|
CADJ STORE KappaRV(:,:,:) = comlev1_bibj , key=idynkey, byte=isbyte |
|
659 |
CADJ STORE gV(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
CADJ STORE gV(:,:,:,bi,bj) = comlev1_bibj , key=idynkey, byte=isbyte |
660 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
661 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
710 |
CALL TIMER_START('CALC_GW [DYNAMICS]',myThid) |
CALL TIMER_START('CALC_GW [DYNAMICS]',myThid) |
711 |
CALL CALC_GW( |
CALL CALC_GW( |
712 |
I bi,bj, KappaRU, KappaRV, |
I bi,bj, KappaRU, KappaRV, |
713 |
|
I str13, str23, str33, |
714 |
|
I viscAh3d_00, viscAh3d_13, viscAh3d_23, |
715 |
I myTime, myIter, myThid ) |
I myTime, myIter, myThid ) |
716 |
ENDIF |
ENDIF |
717 |
IF ( nonHydrostatic.OR.implicitIntGravWave ) |
IF ( nonHydrostatic.OR.implicitIntGravWave ) |