33 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
34 |
# include "tamc.h" |
# include "tamc.h" |
35 |
# include "tamc_keys.h" |
# include "tamc_keys.h" |
36 |
|
# include "FFIELDS.h" |
37 |
|
# ifdef ALLOW_KPP |
38 |
|
# include "KPP.h" |
39 |
|
# endif |
40 |
|
# ifdef ALLOW_GMREDI |
41 |
|
# include "GMREDI.h" |
42 |
|
# endif |
43 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
44 |
|
|
|
#ifdef ALLOW_KPP |
|
|
# include "KPP.h" |
|
|
#endif |
|
|
|
|
45 |
#ifdef ALLOW_TIMEAVE |
#ifdef ALLOW_TIMEAVE |
46 |
#include "TIMEAVE_STATV.h" |
#include "TIMEAVE_STATV.h" |
47 |
#endif |
#endif |
122 |
c EXTERNAL DIFFERENT_MULTIPLE |
c EXTERNAL DIFFERENT_MULTIPLE |
123 |
Cjmc(end) |
Cjmc(end) |
124 |
|
|
|
#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 */ |
|
|
|
|
125 |
C--- The algorithm... |
C--- The algorithm... |
126 |
C |
C |
127 |
C "Correction Step" |
C "Correction Step" |
265 |
jMax = sNy+OLy |
jMax = sNy+OLy |
266 |
|
|
267 |
|
|
268 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
269 |
|
CADJ STORE theta(:,:,:,bi,bj) = comlev1_bibj, key = ikey, byte = isbyte |
270 |
|
CADJ STORE salt (:,:,:,bi,bj) = comlev1_bibj, key = ikey, byte = isbyte |
271 |
|
CADJ STORE uvel (:,:,:,bi,bj) = comlev1_bibj, key = ikey, byte = isbyte |
272 |
|
CADJ STORE vvel (:,:,:,bi,bj) = comlev1_bibj, key = ikey, byte = isbyte |
273 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
274 |
|
|
275 |
C-- Start of diagnostic loop |
C-- Start of diagnostic loop |
276 |
DO k=Nr,1,-1 |
DO k=Nr,1,-1 |
277 |
|
|
278 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
279 |
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? |
280 |
C? Do we still need this? |
C? Do we still need this? |
281 |
kkey = (ikey-1)*(Nr-2+1) + (k-2) + 1 |
cph kkey formula corrected. |
282 |
|
cph Needed for rhok, rhokm1, in the case useGMREDI. |
283 |
|
kkey = (ikey-1)*Nr + k |
284 |
|
CADJ STORE rhokm1(:,:) = comlev1_bibj_k , key=kkey, byte=isbyte |
285 |
|
CADJ STORE rhok (:,:) = comlev1_bibj_k , key=kkey, byte=isbyte |
286 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
287 |
|
|
288 |
C-- Integrate continuity vertically for vertical velocity |
C-- Integrate continuity vertically for vertical velocity |
304 |
C slope terms (e.g. GM/Redi tensor or IVDC diffusivity) |
C slope terms (e.g. GM/Redi tensor or IVDC diffusivity) |
305 |
c IF ( k.GT.1 .AND. (useGMRedi.OR.ivdc_kappa.NE.0.) ) THEN |
c IF ( k.GT.1 .AND. (useGMRedi.OR.ivdc_kappa.NE.0.) ) THEN |
306 |
IF ( useGMRedi .OR. (k.GT.1 .AND. ivdc_kappa.NE.0.) ) THEN |
IF ( useGMRedi .OR. (k.GT.1 .AND. ivdc_kappa.NE.0.) ) THEN |
307 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
308 |
|
CADJ STORE theta(:,:,k,bi,bj) = comlev1_bibj_k, key=kkey, byte=isbyte |
309 |
|
CADJ STORE salt (:,:,k,bi,bj) = comlev1_bibj_k, key=kkey, byte=isbyte |
310 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
311 |
CALL FIND_RHO( |
CALL FIND_RHO( |
312 |
I bi, bj, iMin, iMax, jMin, jMax, k, k, eosType, |
I bi, bj, iMin, iMax, jMin, jMax, k, k, eosType, |
313 |
I theta, salt, |
I theta, salt, |
314 |
O rhoK, |
O rhoK, |
315 |
I myThid ) |
I myThid ) |
316 |
IF (k.GT.1) CALL FIND_RHO( |
IF (k.GT.1) THEN |
317 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
318 |
|
CADJ STORE theta(:,:,k-1,bi,bj) = comlev1_bibj_k, key=kkey, byte=isbyte |
319 |
|
CADJ STORE salt (:,:,k-1,bi,bj) = comlev1_bibj_k, key=kkey, byte=isbyte |
320 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
321 |
|
CALL FIND_RHO( |
322 |
I bi, bj, iMin, iMax, jMin, jMax, k-1, k, eosType, |
I bi, bj, iMin, iMax, jMin, jMax, k-1, k, eosType, |
323 |
I theta, salt, |
I theta, salt, |
324 |
O rhoKm1, |
O rhoKm1, |
325 |
I myThid ) |
I myThid ) |
326 |
|
ENDIF |
327 |
CALL GRAD_SIGMA( |
CALL GRAD_SIGMA( |
328 |
I bi, bj, iMin, iMax, jMin, jMax, k, |
I bi, bj, iMin, iMax, jMin, jMax, k, |
329 |
I rhoK, rhoKm1, rhoK, |
I rhoK, rhoKm1, rhoK, |
344 |
C-- end of diagnostic k loop (Nr:1) |
C-- end of diagnostic k loop (Nr:1) |
345 |
ENDDO |
ENDDO |
346 |
|
|
347 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
348 |
|
cph avoids recomputation of integrate_for_w |
349 |
|
CADJ STORE wvel (:,:,:,bi,bj) = comlev1_bibj, key = ikey, byte = isbyte |
350 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
351 |
|
|
352 |
#ifdef ALLOW_OBCS |
#ifdef ALLOW_OBCS |
353 |
C-- Calculate future values on open boundaries |
C-- Calculate future values on open boundaries |
354 |
IF (useOBCS) THEN |
IF (useOBCS) THEN |
363 |
CALL EXTERNAL_FORCING_SURF( |
CALL EXTERNAL_FORCING_SURF( |
364 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
365 |
I myThid ) |
I myThid ) |
366 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
367 |
|
cph needed for KPP |
368 |
|
CADJ STORE surfacetendencyU(:,:,bi,bj) |
369 |
|
CADJ & = comlev1_bibj, key=ikey, byte=isbyte |
370 |
|
CADJ STORE surfacetendencyV(:,:,bi,bj) |
371 |
|
CADJ & = comlev1_bibj, key=ikey, byte=isbyte |
372 |
|
CADJ STORE surfacetendencyS(:,:,bi,bj) |
373 |
|
CADJ & = comlev1_bibj, key=ikey, byte=isbyte |
374 |
|
CADJ STORE surfacetendencyT(:,:,bi,bj) |
375 |
|
CADJ & = comlev1_bibj, key=ikey, byte=isbyte |
376 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
377 |
|
|
378 |
#ifdef ALLOW_GMREDI |
#ifdef ALLOW_GMREDI |
379 |
|
|
380 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
381 |
|
CADJ STORE sigmaX(:,:,:) = comlev1, key=ikey, byte=isbyte |
382 |
|
CADJ STORE sigmaY(:,:,:) = comlev1, key=ikey, byte=isbyte |
383 |
|
CADJ STORE sigmaR(:,:,:) = comlev1, key=ikey, byte=isbyte |
384 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
385 |
C-- Calculate iso-neutral slopes for the GM/Redi parameterisation |
C-- Calculate iso-neutral slopes for the GM/Redi parameterisation |
386 |
IF (useGMRedi) THEN |
IF (useGMRedi) THEN |
387 |
DO k=1,Nr |
DO k=1,Nr |
400 |
ENDDO |
ENDDO |
401 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
402 |
ENDIF |
ENDIF |
403 |
|
|
404 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
405 |
|
CADJ STORE Kwx(:,:,:,bi,bj) = comlev1_bibj, key=ikey, byte=isbyte |
406 |
|
CADJ STORE Kwy(:,:,:,bi,bj) = comlev1_bibj, key=ikey, byte=isbyte |
407 |
|
CADJ STORE Kwz(:,:,:,bi,bj) = comlev1_bibj, key=ikey, byte=isbyte |
408 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
409 |
|
|
410 |
#endif /* ALLOW_GMREDI */ |
#endif /* ALLOW_GMREDI */ |
411 |
|
|
412 |
#ifdef ALLOW_KPP |
#ifdef ALLOW_KPP |
414 |
IF (useKPP) THEN |
IF (useKPP) THEN |
415 |
CALL KPP_CALC( |
CALL KPP_CALC( |
416 |
I bi, bj, myTime, myThid ) |
I bi, bj, myTime, myThid ) |
417 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
418 |
|
ELSE |
419 |
|
CALL KPP_CALC_DUMMY( |
420 |
|
I bi, bj, myTime, myThid ) |
421 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
422 |
ENDIF |
ENDIF |
423 |
|
|
424 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
425 |
|
CADJ STORE KPPghat (:,:,:,bi,bj) |
426 |
|
CADJ & , KPPviscAz (:,:,:,bi,bj) |
427 |
|
CADJ & , KPPdiffKzT(:,:,:,bi,bj) |
428 |
|
CADJ & , KPPdiffKzS(:,:,:,bi,bj) |
429 |
|
CADJ & , KPPfrac (:,: ,bi,bj) |
430 |
|
CADJ & = comlev1_bibj, key=ikey, byte=isbyte |
431 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
432 |
|
|
433 |
#endif /* ALLOW_KPP */ |
#endif /* ALLOW_KPP */ |
434 |
|
|
435 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
454 |
|
|
455 |
C-- Start of thermodynamics loop |
C-- Start of thermodynamics loop |
456 |
DO k=Nr,1,-1 |
DO k=Nr,1,-1 |
457 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
458 |
|
C? Patrick Is this formula correct? |
459 |
|
cph Yes, but I rewrote it. |
460 |
|
cph Also, the KappaR? need the index and subscript k! |
461 |
|
kkey = (ikey-1)*Nr + k |
462 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
463 |
|
|
464 |
C-- km1 Points to level above k (=k-1) |
C-- km1 Points to level above k (=k-1) |
465 |
C-- kup Cycles through 1,2 to point to layer above |
C-- kup Cycles through 1,2 to point to layer above |
474 |
jMin = 1-OLy+2 |
jMin = 1-OLy+2 |
475 |
jMax = sNy+OLy-1 |
jMax = sNy+OLy-1 |
476 |
|
|
|
#ifdef ALLOW_AUTODIFF_TAMC |
|
|
CPatrick Is this formula correct? |
|
|
kkey = (ikey-1)*(Nr-1+1) + (k-1) + 1 |
|
|
CADJ STORE rTrans(:,:) = comlev1_bibj_k, key = kkey, byte = isbyte |
|
|
CADJ STORE KappaRT(:,:,:) = comlev1_bibj_k, key = kkey, byte = isbyte |
|
|
CADJ STORE KappaRS(:,:,:) = comlev1_bibj_k, key = kkey, byte = isbyte |
|
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
|
|
|
|
477 |
C-- Get temporary terms used by tendency routines |
C-- Get temporary terms used by tendency routines |
478 |
CALL CALC_COMMON_FACTORS ( |
CALL CALC_COMMON_FACTORS ( |
479 |
I bi,bj,iMin,iMax,jMin,jMax,k,km1,kup,kDown, |
I bi,bj,iMin,iMax,jMin,jMax,k,km1,kup,kDown, |
480 |
O xA,yA,uTrans,vTrans,rTrans,maskC,maskUp, |
O xA,yA,uTrans,vTrans,rTrans,maskC,maskUp, |
481 |
I myThid) |
I myThid) |
482 |
|
|
483 |
|
#ifdef ALLOW_AUTODIFF_TAMC |
484 |
|
CADJ STORE KappaRT(:,:,k) = comlev1_bibj_k, key=kkey, byte=isbyte |
485 |
|
CADJ STORE KappaRS(:,:,k) = comlev1_bibj_k, key=kkey, byte=isbyte |
486 |
|
#endif /* ALLOW_AUTODIFF_TAMC */ |
487 |
|
|
488 |
#ifdef INCLUDE_CALC_DIFFUSIVITY_CALL |
#ifdef INCLUDE_CALC_DIFFUSIVITY_CALL |
489 |
C-- Calculate the total vertical diffusivity |
C-- Calculate the total vertical diffusivity |
490 |
CALL CALC_DIFFUSIVITY( |
CALL CALC_DIFFUSIVITY( |
544 |
|
|
545 |
|
|
546 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
547 |
CPatrick? What about this one? |
C? Patrick? What about this one? |
548 |
maximpl = 6 |
cph Keys iikey and idkey don't seem to be needed |
549 |
iikey = (ikey-1)*maximpl |
cph since storing occurs on different tape for each |
550 |
|
cph impldiff call anyways. |
551 |
|
cph Thus, common block comlev1_impl isn't needed either. |
552 |
|
cph Storing below needed in the case useGMREDI. |
553 |
|
iikey = (ikey-1)*maximpl |
554 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
555 |
|
|
556 |
C-- Implicit diffusion |
C-- Implicit diffusion |
559 |
IF (tempStepping) THEN |
IF (tempStepping) THEN |
560 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
561 |
idkey = iikey + 1 |
idkey = iikey + 1 |
562 |
|
CADJ STORE gTNm1(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
563 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
564 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
565 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
571 |
IF (saltStepping) THEN |
IF (saltStepping) THEN |
572 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
573 |
idkey = iikey + 2 |
idkey = iikey + 2 |
574 |
|
CADJ STORE gSNm1(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
575 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
576 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
577 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
679 |
IF (implicitViscosity.AND.momStepping) THEN |
IF (implicitViscosity.AND.momStepping) THEN |
680 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
681 |
idkey = iikey + 3 |
idkey = iikey + 3 |
682 |
|
CADJ STORE gUNm1(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
683 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
684 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
685 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
688 |
I myThid ) |
I myThid ) |
689 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
690 |
idkey = iikey + 4 |
idkey = iikey + 4 |
691 |
|
CADJ STORE gVNm1(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
692 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
693 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
694 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
708 |
#ifdef INCLUDE_CD_CODE |
#ifdef INCLUDE_CD_CODE |
709 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
710 |
idkey = iikey + 5 |
idkey = iikey + 5 |
711 |
|
CADJ STORE vVelD(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
712 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
713 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
714 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |
717 |
I myThid ) |
I myThid ) |
718 |
#ifdef ALLOW_AUTODIFF_TAMC |
#ifdef ALLOW_AUTODIFF_TAMC |
719 |
idkey = iikey + 6 |
idkey = iikey + 6 |
720 |
|
CADJ STORE uVelD(:,:,:,bi,bj) = comlev1_bibj , key=ikey, byte=isbyte |
721 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
#endif /* ALLOW_AUTODIFF_TAMC */ |
722 |
CALL IMPLDIFF( |
CALL IMPLDIFF( |
723 |
I bi, bj, iMin, iMax, jMin, jMax, |
I bi, bj, iMin, iMax, jMin, jMax, |