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" |
# 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 :: tile indices |
C bi, bj :: tile indices |
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, |
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, |