1 |
C $Header: /u/gcmpack/MITgcm/pkg/ptracers/ptracers_integrate.F,v 1.42 2011/01/18 22:58:20 heimbach Exp $ |
2 |
C $Name: $ |
3 |
|
4 |
#include "PTRACERS_OPTIONS.h" |
5 |
|
6 |
CBOP |
7 |
C !ROUTINE: PTRACERS_INTEGRATE |
8 |
|
9 |
C !INTERFACE: ========================================================== |
10 |
SUBROUTINE PTRACERS_INTEGRATE( |
11 |
I bi,bj,k, |
12 |
I xA, yA, maskUp, uFld, vFld, wFld, |
13 |
I uTrans, vTrans, rTrans, rTransKp1, |
14 |
I KappaRtr, |
15 |
U rFlx, |
16 |
I myTime,myIter,myThid ) |
17 |
|
18 |
C !DESCRIPTION: |
19 |
C Calculates tendency for passive tracers and integrates forward |
20 |
C in time. |
21 |
|
22 |
C !USES: =============================================================== |
23 |
IMPLICIT NONE |
24 |
#include "SIZE.h" |
25 |
#include "EEPARAMS.h" |
26 |
#include "PARAMS.h" |
27 |
#include "DYNVARS.h" |
28 |
#ifdef ALLOW_LONGSTEP |
29 |
#include "LONGSTEP_PARAMS.h" |
30 |
#endif |
31 |
#include "PTRACERS_SIZE.h" |
32 |
#include "PTRACERS_PARAMS.h" |
33 |
#include "PTRACERS_RESTART.h" |
34 |
#include "PTRACERS_FIELDS.h" |
35 |
#include "GAD.h" |
36 |
#ifdef ALLOW_AUTODIFF_TAMC |
37 |
# include "tamc.h" |
38 |
# include "tamc_keys.h" |
39 |
#endif |
40 |
|
41 |
C !INPUT PARAMETERS: =================================================== |
42 |
C bi,bj :: tile indices |
43 |
C k :: vertical level number |
44 |
C xA :: face area at U points in level k |
45 |
C yA :: face area at V points in level k |
46 |
C maskUp :: mask for vertical transport |
47 |
C uFld,vFld,wFld :: Local copy of velocity field (3 components) |
48 |
C uTrans :: zonal transport in level k |
49 |
C vTrans :: meridional transport in level k |
50 |
C rTrans :: vertical volume transport at interface k |
51 |
C rTransKp1 :: vertical volume transport at interface k+1 |
52 |
C KappaRtr :: vertical diffusion of passive tracers, interf k |
53 |
C rFlx :: vertical flux |
54 |
C myTime :: model time |
55 |
C myIter :: time-step number |
56 |
C myThid :: thread number |
57 |
INTEGER bi,bj,k |
58 |
_RS xA (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
59 |
_RS yA (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
60 |
_RS maskUp (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
61 |
_RL uFld (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
62 |
_RL vFld (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
63 |
_RL wFld (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
64 |
_RL uTrans (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
65 |
_RL vTrans (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
66 |
_RL rTrans (1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
67 |
_RL rTransKp1(1-OLx:sNx+OLx,1-OLy:sNy+OLy) |
68 |
_RL KappaRtr(1-Olx:sNx+Olx,1-Oly:sNy+Oly,PTRACERS_num) |
69 |
_RL rFlx (1-OLx:sNx+OLx,1-OLy:sNy+OLy,2,PTRACERS_num) |
70 |
_RL myTime |
71 |
INTEGER myIter |
72 |
INTEGER myThid |
73 |
|
74 |
C !OUTPUT PARAMETERS: ================================================== |
75 |
C none |
76 |
|
77 |
#ifdef ALLOW_PTRACERS |
78 |
|
79 |
C !LOCAL VARIABLES: ==================================================== |
80 |
C iTracer :: tracer index |
81 |
C iMin,iMax,jMin,jMax :: loop ranges |
82 |
C kUp,kDown :: toggle indices for even/odd level fluxes |
83 |
C km1 :: =min(1,k-1) |
84 |
C GAD_TR :: passive tracer id (GAD_TR1+iTracer-1) |
85 |
INTEGER iTracer |
86 |
INTEGER iMin,iMax,jMin,jMax |
87 |
INTEGER kUp,kDown,km1 |
88 |
INTEGER GAD_TR |
89 |
LOGICAL calcAdvection |
90 |
INTEGER iterNb |
91 |
CEOP |
92 |
|
93 |
C Loop ranges for daughter routines |
94 |
iMin = 1-OLx+2 |
95 |
iMax = sNx+OLx-1 |
96 |
jMin = 1-OLy+2 |
97 |
jMax = sNy+OLy-1 |
98 |
|
99 |
km1 = MAX(1,k-1) |
100 |
kUp = 1+MOD(k+1,2) |
101 |
kDown= 1+MOD(k,2) |
102 |
|
103 |
C Loop over tracers |
104 |
DO iTracer=1,PTRACERS_numInUse |
105 |
|
106 |
#ifdef ALLOW_AUTODIFF_TAMC |
107 |
act0 = iTracer - 1 |
108 |
max0 = PTRACERS_num |
109 |
act1 = bi - myBxLo(myThid) |
110 |
max1 = myBxHi(myThid) - myBxLo(myThid) + 1 |
111 |
act2 = bj - myByLo(myThid) |
112 |
max2 = myByHi(myThid) - myByLo(myThid) + 1 |
113 |
act3 = myThid - 1 |
114 |
max3 = nTx*nTy |
115 |
act4 = ikey_dynamics - 1 |
116 |
iptrkey = (act0 + 1) |
117 |
& + act1*max0 |
118 |
& + act2*max0*max1 |
119 |
& + act3*max0*max1*max2 |
120 |
& + act4*max0*max1*max2*max3 |
121 |
kkey = (iptrkey-1)*Nr + k |
122 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
123 |
|
124 |
#ifdef ALLOW_AUTODIFF_TAMC |
125 |
rFlx(1,1,kDown,iTracer) = rFlx(1,1,kDown,iTracer) |
126 |
c |
127 |
CADJ STORE pTracer(:,:,k,bi,bj,iTracer) |
128 |
CADJ & = comlev1_bibj_k_ptracers, key=kkey, byte=isbyte |
129 |
CADJ STORE gpTrNm1(:,:,k,bi,bj,iTracer) |
130 |
CADJ & = comlev1_bibj_k_ptracers, key=kkey, byte=isbyte |
131 |
# ifdef NONLIN_FRSURF |
132 |
CADJ STORE gpTr(:,:,k,bi,bj,iTracer) |
133 |
CADJ & = comlev1_bibj_k_ptracers, key=kkey, byte=isbyte |
134 |
CADJ STORE rFlx(:,:,:,iTracer) |
135 |
CADJ & = comlev1_bibj_k_ptracers, key=kkey, byte=isbyte |
136 |
# endif |
137 |
#endif /* ALLOW_AUTODIFF_TAMC */ |
138 |
|
139 |
C Calculate active tracer tendencies (gPtr) due to internal processes |
140 |
C (advection, [explicit] diffusion, parameterizations,...) |
141 |
calcAdvection = .NOT.PTRACERS_MultiDimAdv(iTracer) |
142 |
GAD_TR = GAD_TR1 + iTracer - 1 |
143 |
CALL GAD_CALC_RHS( |
144 |
I bi,bj,iMin,iMax,jMin,jMax,k,km1,kUp,kDown, |
145 |
I xA, yA, maskUp, uFld, vFld, wFld, |
146 |
I uTrans, vTrans, rTrans, rTransKp1, |
147 |
I PTRACERS_diffKh(iTracer), |
148 |
I PTRACERS_diffK4(iTracer), |
149 |
I KappaRtr(1-Olx,1-Oly,iTracer), |
150 |
I gpTrNm1(1-Olx,1-Oly,1,1,1,iTracer), |
151 |
I pTracer(1-Olx,1-Oly,1,1,1,iTracer), |
152 |
I PTRACERS_dTLev, GAD_TR, |
153 |
I PTRACERS_advScheme(iTracer), |
154 |
I PTRACERS_advScheme(iTracer), |
155 |
I calcAdvection, PTRACERS_ImplVertAdv(iTracer), |
156 |
I .FALSE., |
157 |
I PTRACERS_useGMRedi(iTracer), |
158 |
I PTRACERS_useKPP(iTracer), |
159 |
U rFlx(1-Olx,1-Oly,1,iTracer), |
160 |
U gPtr(1-Olx,1-Oly,1,1,1,iTracer), |
161 |
I myTime, myIter, myThid ) |
162 |
|
163 |
C External forcing term(s) |
164 |
IF ( tracForcingOutAB.NE.1 ) |
165 |
& CALL PTRACERS_FORCING( |
166 |
I bi,bj,iMin,iMax,jMin,jMax,k,iTracer, |
167 |
U gPtr(1-Olx,1-Oly,1,1,1,iTracer), |
168 |
I surfaceForcingPTr(1-Olx,1-Oly,1,1,iTracer), |
169 |
I myIter,myTime,myThid) |
170 |
|
171 |
C If using Adams-Bashforth II, then extrapolate tendencies |
172 |
C gPtr is now the tracer tendency for explicit advection/diffusion |
173 |
IF ( PTRACERS_AdamsBashGtr(iTracer) ) THEN |
174 |
#ifdef ALLOW_MATRIX |
175 |
C If matrix is being computed, block call to S/R ADAMS_BASHFORTH2 to |
176 |
C prevent gPtr from being replaced by the average of gPtr and gpTrNm1. |
177 |
IF (.NOT.useMATRIX) THEN |
178 |
#endif |
179 |
C compute iter at beginning of ptracer time step |
180 |
#ifdef ALLOW_LONGSTEP |
181 |
iterNb = myIter - LS_nIter + 1 |
182 |
IF (LS_whenToSample.GE.2) iterNb = myIter - LS_nIter |
183 |
#else |
184 |
iterNb = myIter |
185 |
IF (staggerTimeStep) iterNb = myIter - 1 |
186 |
#endif |
187 |
CALL ADAMS_BASHFORTH2( |
188 |
I bi,bj,K, |
189 |
U gPtr(1-Olx,1-Oly,1,1,1,iTracer), |
190 |
U gpTrNm1(1-Olx,1-Oly,1,1,1,iTracer), |
191 |
I PTRACERS_startAB(iTracer), iterNb, myThid ) |
192 |
#ifdef ALLOW_MATRIX |
193 |
ENDIF |
194 |
#endif |
195 |
ENDIF |
196 |
|
197 |
C External forcing term(s) |
198 |
IF ( tracForcingOutAB.EQ.1 ) |
199 |
& CALL PTRACERS_FORCING( |
200 |
I bi,bj,iMin,iMax,jMin,jMax,k,iTracer, |
201 |
U gPtr(1-Olx,1-Oly,1,1,1,iTracer), |
202 |
I surfaceForcingPTr(1-Olx,1-Oly,1,1,iTracer), |
203 |
I myIter,myTime,myThid) |
204 |
|
205 |
#ifdef NONLIN_FRSURF |
206 |
C Account for change in level thickness |
207 |
IF (nonlinFreeSurf.GT.0) THEN |
208 |
CALL FREESURF_RESCALE_G( |
209 |
I bi,bj,K, |
210 |
U gPtr(1-Olx,1-Oly,1,1,1,iTracer), |
211 |
I myThid ) |
212 |
IF ( PTRACERS_AdamsBashGtr(iTracer) ) |
213 |
& CALL FREESURF_RESCALE_G( |
214 |
I bi,bj,K, |
215 |
U gpTrNm1(1-Olx,1-Oly,1,1,1,iTracer), |
216 |
I myThid ) |
217 |
ENDIF |
218 |
#endif /* NONLIN_FRSURF */ |
219 |
|
220 |
C Integrate forward in time, storing in gPtr: G=T+dt*G |
221 |
CALL TIMESTEP_TRACER( |
222 |
I bi, bj, k, PTRACERS_dTLev(k), |
223 |
I pTracer(1-Olx,1-Oly,1,1,1,iTracer), |
224 |
U gPtr(1-Olx,1-Oly,1,1,1,iTracer), |
225 |
I myIter, myThid ) |
226 |
|
227 |
C end of tracer loop |
228 |
ENDDO |
229 |
|
230 |
#endif /* ALLOW_PTRACERS */ |
231 |
|
232 |
RETURN |
233 |
END |