78 |
#include "SIZE.h" |
#include "SIZE.h" |
79 |
#include "EEPARAMS.h" |
#include "EEPARAMS.h" |
80 |
#include "PARAMS.h" |
#include "PARAMS.h" |
81 |
|
#include "GRID.h" |
82 |
#include "DYNVARS.h" |
#include "DYNVARS.h" |
83 |
#ifdef ALLOW_CD_CODE |
#ifdef ALLOW_CD_CODE |
84 |
#include "CD_CODE_VARS.h" |
# include "CD_CODE_VARS.h" |
85 |
#endif |
#endif |
|
#include "GRID.h" |
|
86 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
87 |
# include "tamc.h" |
# include "tamc.h" |
88 |
# include "tamc_keys.h" |
# include "tamc_keys.h" |
96 |
# include "PTRACERS_FIELDS.h" |
# include "PTRACERS_FIELDS.h" |
97 |
# endif |
# endif |
98 |
# ifdef ALLOW_OBCS |
# ifdef ALLOW_OBCS |
99 |
|
# include "OBCS_PARAMS.h" |
100 |
# include "OBCS_FIELDS.h" |
# include "OBCS_FIELDS.h" |
101 |
# ifdef ALLOW_PTRACERS |
# ifdef ALLOW_PTRACERS |
102 |
# include "OBCS_PTRACERS.h" |
# include "OBCS_PTRACERS.h" |
116 |
C | |
C | |
117 |
C |-- CALC_VISCOSITY |
C |-- CALC_VISCOSITY |
118 |
C | |
C | |
119 |
|
C |-- CALC_EDDY_STRESS |
120 |
|
C | |
121 |
C |-- CALC_PHI_HYD |
C |-- CALC_PHI_HYD |
122 |
C | |
C | |
123 |
C |-- MOM_FLUXFORM |
C |-- MOM_FLUXFORM |
169 |
C phiSurfY or geopotential (atmos) in X and Y direction |
C phiSurfY or geopotential (atmos) in X and Y direction |
170 |
C guDissip :: dissipation tendency (all explicit terms), u component |
C guDissip :: dissipation tendency (all explicit terms), u component |
171 |
C gvDissip :: dissipation tendency (all explicit terms), v component |
C gvDissip :: dissipation tendency (all explicit terms), v component |
172 |
C KappaRU :: vertical viscosity |
C KappaRU :: vertical viscosity for velocity U-component |
173 |
C KappaRV :: vertical viscosity |
C KappaRV :: vertical viscosity for velocity V-component |
174 |
C iMin, iMax - Ranges and sub-block indices on which calculations |
C iMin, iMax :: Ranges and sub-block indices on which calculations |
175 |
C jMin, jMax are applied. |
C jMin, jMax are applied. |
176 |
C bi, bj |
C bi, bj :: tile indices |
177 |
C k, kup, - Index for layer above and below. kup and kDown |
C k :: current level index |
178 |
C kDown, km1 are switched with layer to be the appropriate |
C km1, kp1 :: index of level above (k-1) and below (k+1) |
179 |
C index into fVerTerm. |
C kUp, kDown :: Index for interface above and below. kUp and kDown are |
180 |
|
C are switched with k to be the appropriate index into fVerU,V |
181 |
_RL fVerU (1-OLx:sNx+OLx,1-OLy:sNy+OLy,2) |
_RL fVerU (1-OLx:sNx+OLx,1-OLy:sNy+OLy,2) |
182 |
_RL fVerV (1-OLx:sNx+OLx,1-OLy:sNy+OLy,2) |
_RL fVerV (1-OLx:sNx+OLx,1-OLy:sNy+OLy,2) |
183 |
_RL phiHydF (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
_RL phiHydF (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
195 |
INTEGER jMin, jMax |
INTEGER jMin, jMax |
196 |
INTEGER bi, bj |
INTEGER bi, bj |
197 |
INTEGER i, j |
INTEGER i, j |
198 |
INTEGER k, km1, kp1, kup, kDown |
INTEGER k, km1, kp1, kUp, kDown |
199 |
|
|
200 |
#ifdef ALLOW_DIAGNOSTICS |
#ifdef ALLOW_DIAGNOSTICS |
201 |
LOGICAL dPhiHydDiagIsOn |
LOGICAL dPhiHydDiagIsOn |
202 |
_RL tmpFac |
_RL tmpFac |
203 |
#endif /* ALLOW_DIAGNOSTICS */ |
#endif /* ALLOW_DIAGNOSTICS */ |
204 |
|
|
|
|
|
205 |
C--- The algorithm... |
C--- The algorithm... |
206 |
C |
C |
207 |
C "Correction Step" |
C "Correction Step" |
258 |
& .OR. DIAGNOSTICS_IS_ON( 'Vm_dPHdy', myThid ) |
& .OR. DIAGNOSTICS_IS_ON( 'Vm_dPHdy', myThid ) |
259 |
#endif |
#endif |
260 |
|
|
261 |
C-- Call to routine for calculation of |
C-- Call to routine for calculation of Eliassen-Palm-flux-forced |
262 |
C Eliassen-Palm-flux-forced U-tendency, |
C U-tendency, if desired: |
|
C if desired: |
|
263 |
#ifdef INCLUDE_EP_FORCING_CODE |
#ifdef INCLUDE_EP_FORCING_CODE |
264 |
CALL CALC_EP_FORCING(myThid) |
CALL CALC_EP_FORCING(myThid) |
265 |
#endif |
#endif |
308 |
DO k=1,Nr |
DO k=1,Nr |
309 |
DO j=1-OLy,sNy+OLy |
DO j=1-OLy,sNy+OLy |
310 |
DO i=1-OLx,sNx+OLx |
DO i=1-OLx,sNx+OLx |
|
KappaRU(i,j,k) = 0. _d 0 |
|
|
KappaRV(i,j,k) = 0. _d 0 |
|
311 |
cph( |
cph( |
312 |
c-- need some re-initialisation here to break dependencies |
c-- need some re-initialisation here to break dependencies |
313 |
cph) |
cph) |
337 |
phiHydLow(i,j,bi,bj) = 0. _d 0 |
phiHydLow(i,j,bi,bj) = 0. _d 0 |
338 |
# if (defined NONLIN_FRSURF) && (defined ALLOW_MOM_FLUXFORM) |
# if (defined NONLIN_FRSURF) && (defined ALLOW_MOM_FLUXFORM) |
339 |
# ifndef DISABLE_RSTAR_CODE |
# ifndef DISABLE_RSTAR_CODE |
340 |
|
# ifndef ALLOW_AUTODIFF_OPENAD |
341 |
dWtransC(i,j,bi,bj) = 0. _d 0 |
dWtransC(i,j,bi,bj) = 0. _d 0 |
342 |
dWtransU(i,j,bi,bj) = 0. _d 0 |
dWtransU(i,j,bi,bj) = 0. _d 0 |
343 |
dWtransV(i,j,bi,bj) = 0. _d 0 |
dWtransV(i,j,bi,bj) = 0. _d 0 |
344 |
|
# endif |
345 |
# endif |
# endif |
346 |
# endif |
# endif |
347 |
#endif |
#endif |
378 |
#endif /* ALLOW_KPP */ |
#endif /* ALLOW_KPP */ |
379 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
380 |
|
|
381 |
#ifdef INCLUDE_CALC_DIFFUSIVITY_CALL |
#if (defined INCLUDE_CALC_DIFFUSIVITY_CALL) && !(defined ALLOW_AUTODIFF) |
382 |
|
IF ( .NOT.momViscosity ) THEN |
383 |
|
#endif /* INCLUDE_CALC_DIFFUSIVITY_CALL and not ALLOW_AUTODIFF */ |
384 |
|
DO k=1,Nr |
385 |
|
DO j=1-OLy,sNy+OLy |
386 |
|
DO i=1-OLx,sNx+OLx |
387 |
|
KappaRU(i,j,k) = 0. _d 0 |
388 |
|
KappaRV(i,j,k) = 0. _d 0 |
389 |
|
ENDDO |
390 |
|
ENDDO |
391 |
|
ENDDO |
392 |
|
#ifdef INCLUDE_CALC_DIFFUSIVITY_CALL |
393 |
C-- Calculate the total vertical viscosity |
C-- Calculate the total vertical viscosity |
394 |
CALL CALC_VISCOSITY( |
#ifdef ALLOW_AUTODIFF |
395 |
|
IF ( momViscosity ) THEN |
396 |
|
#else |
397 |
|
ELSE |
398 |
|
#endif |
399 |
|
CALL CALC_VISCOSITY( |
400 |
I bi,bj, iMin,iMax,jMin,jMax, |
I bi,bj, iMin,iMax,jMin,jMax, |
401 |
O KappaRU, KappaRV, |
O KappaRU, KappaRV, |
402 |
I myThid ) |
I myThid ) |
403 |
#else |
ENDIF |
404 |
DO k=1,Nr |
#endif /* INCLUDE_CALC_DIFFUSIVITY_CALL */ |
|
DO j=1-OLy,sNy+OLy |
|
|
DO i=1-OLx,sNx+OLx |
|
|
KappaRU(i,j,k) = 0. _d 0 |
|
|
KappaRV(i,j,k) = 0. _d 0 |
|
|
ENDDO |
|
|
ENDDO |
|
|
ENDDO |
|
|
#endif |
|
405 |
|
|
406 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
407 |
CADJ STORE KappaRU(:,:,:) |
CADJ STORE KappaRU(:,:,:) |
413 |
#ifdef ALLOW_OBCS |
#ifdef ALLOW_OBCS |
414 |
C-- For Stevens boundary conditions velocities need to be extrapolated |
C-- For Stevens boundary conditions velocities need to be extrapolated |
415 |
C (copied) to a narrow strip outside the domain |
C (copied) to a narrow strip outside the domain |
416 |
IF ( useOBCS ) THEN |
IF ( useOBCS ) THEN |
417 |
CALL OBCS_COPY_UV_N( |
CALL OBCS_COPY_UV_N( |
418 |
U uVel(1-OLx,1-OLy,1,bi,bj), |
U uVel(1-OLx,1-OLy,1,bi,bj), |
419 |
U vVel(1-OLx,1-OLy,1,bi,bj), |
U vVel(1-OLx,1-OLy,1,bi,bj), |
420 |
I Nr, bi, bj, myThid ) |
I Nr, bi, bj, myThid ) |
421 |
ENDIF |
ENDIF |
422 |
#endif /* ALLOW_OBCS */ |
#endif /* ALLOW_OBCS */ |
423 |
|
|
424 |
|
#ifdef ALLOW_EDDYPSI |
425 |
|
CALL CALC_EDDY_STRESS(bi,bj,myThid) |
426 |
|
#endif |
427 |
|
|
428 |
C-- Start of dynamics loop |
C-- Start of dynamics loop |
429 |
DO k=1,Nr |
DO k=1,Nr |
430 |
|
|
504 |
# endif |
# endif |
505 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
506 |
|
|
507 |
C-- Integrate hydrostatic balance for phiHyd with BC of |
C-- Integrate hydrostatic balance for phiHyd with BC of phiHyd(z=0)=0 |
|
C phiHyd(z=0)=0 |
|
508 |
IF ( implicitIntGravWave ) THEN |
IF ( implicitIntGravWave ) THEN |
509 |
CALL CALC_PHI_HYD( |
CALL CALC_PHI_HYD( |
510 |
I bi,bj,iMin,iMax,jMin,jMax,k, |
I bi,bj,iMin,iMax,jMin,jMax,k, |
534 |
C and step forward storing the result in gU, gV, etc... |
C and step forward storing the result in gU, gV, etc... |
535 |
IF ( momStepping ) THEN |
IF ( momStepping ) THEN |
536 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
537 |
# if (defined NONLIN_FRSURF) && (defined ALLOW_MOM_FLUXFORM) |
# ifdef NONLIN_FRSURF |
538 |
# ifndef DISABLE_RSTAR_CODE |
# if (defined ALLOW_MOM_FLUXFORM) && !(defined DISABLE_RSTAR_CODE) |
539 |
CADJ STORE dWtransC(:,:,bi,bj) |
CADJ STORE dWtransC(:,:,bi,bj) |
540 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
541 |
CADJ STORE dWtransU(:,:,bi,bj) |
CADJ STORE dWtransU(:,:,bi,bj) |
543 |
CADJ STORE dWtransV(:,:,bi,bj) |
CADJ STORE dWtransV(:,:,bi,bj) |
544 |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
545 |
# endif |
# endif |
546 |
# endif |
CADJ STORE fVerU(:,:,:) |
547 |
#endif |
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
548 |
|
CADJ STORE fVerV(:,:,:) |
549 |
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
550 |
|
# endif /* NONLIN_FRSURF */ |
551 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
552 |
IF (.NOT. vectorInvariantMomentum) THEN |
IF (.NOT. vectorInvariantMomentum) THEN |
553 |
#ifdef ALLOW_MOM_FLUXFORM |
#ifdef ALLOW_MOM_FLUXFORM |
|
C |
|
554 |
CALL MOM_FLUXFORM( |
CALL MOM_FLUXFORM( |
555 |
I bi,bj,iMin,iMax,jMin,jMax,k,kup,kDown, |
I bi,bj,k,iMin,iMax,jMin,jMax, |
556 |
I KappaRU, KappaRV, |
I KappaRU, KappaRV, |
557 |
U fVerU, fVerV, |
U fVerU(1-OLx,1-OLy,kUp), fVerV(1-OLx,1-OLy,kUp), |
558 |
|
O fVerU(1-OLx,1-OLy,kDown), fVerV(1-OLx,1-OLy,kDown), |
559 |
O guDissip, gvDissip, |
O guDissip, gvDissip, |
560 |
I myTime, myIter, myThid) |
I myTime, myIter, myThid) |
561 |
#endif |
#endif |
562 |
ELSE |
ELSE |
563 |
#ifdef ALLOW_MOM_VECINV |
#ifdef ALLOW_MOM_VECINV |
|
C |
|
|
# ifdef ALLOW_AUTODIFF_TAMC |
|
|
# ifdef NONLIN_FRSURF |
|
|
CADJ STORE fVerU(:,:,:) |
|
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
|
CADJ STORE fVerV(:,:,:) |
|
|
CADJ & = comlev1_bibj_k, key=kkey, byte=isbyte |
|
|
# endif |
|
|
# endif /* ALLOW_AUTODIFF_TAMC */ |
|
|
C |
|
564 |
CALL MOM_VECINV( |
CALL MOM_VECINV( |
565 |
I bi,bj,iMin,iMax,jMin,jMax,k,kup,kDown, |
I bi,bj,k,iMin,iMax,jMin,jMax, |
566 |
I KappaRU, KappaRV, |
I KappaRU, KappaRV, |
567 |
U fVerU, fVerV, |
I fVerU(1-OLx,1-OLy,kUp), fVerV(1-OLx,1-OLy,kUp), |
568 |
|
O fVerU(1-OLx,1-OLy,kDown), fVerV(1-OLx,1-OLy,kDown), |
569 |
O guDissip, gvDissip, |
O guDissip, gvDissip, |
570 |
I myTime, myIter, myThid) |
I myTime, myIter, myThid) |
571 |
#endif |
#endif |
572 |
ENDIF |
ENDIF |
573 |
C |
|
574 |
CALL TIMESTEP( |
CALL TIMESTEP( |
575 |
I bi,bj,iMin,iMax,jMin,jMax,k, |
I bi,bj,iMin,iMax,jMin,jMax,k, |
576 |
I dPhiHydX,dPhiHydY, phiSurfX, phiSurfY, |
I dPhiHydX,dPhiHydY, phiSurfX, phiSurfY, |