/[MITgcm]/MITgcm/pkg/generic_advdiff/gad_advection.F
ViewVC logotype

Contents of /MITgcm/pkg/generic_advdiff/gad_advection.F

Parent Directory Parent Directory | Revision Log Revision Log | View Revision Graph Revision Graph


Revision 1.71 - (show annotations) (download)
Fri Mar 9 20:13:44 2012 UTC (12 years, 3 months ago) by jmc
Branch: MAIN
CVS Tags: checkpoint64, checkpoint63p, checkpoint63q, checkpoint63r, checkpoint63s, checkpoint63l, checkpoint63m, checkpoint63n, checkpoint63o, checkpoint63k, checkpoint64a, checkpoint64c, checkpoint64b, checkpoint64d
Changes since 1.70: +87 -69 lines
- allow to switch to upwind 1rst order advection scheme for the advective
  flux computaion at the open-boundary.

1 C $Header: /u/gcmpack/MITgcm/pkg/generic_advdiff/gad_advection.F,v 1.70 2011/12/12 15:33:53 mlosch Exp $
2 C $Name: $
3
4 #include "GAD_OPTIONS.h"
5
6 C---+----1----+----2----+----3----+----4----+----5----+----6----+----7-|--+----|
7 CBOP
8 C !ROUTINE: GAD_ADVECTION
9
10 C !INTERFACE: ==========================================================
11 SUBROUTINE GAD_ADVECTION(
12 I implicitAdvection, advectionScheme, vertAdvecScheme,
13 I trIdentity, deltaTLev,
14 I uVel, vVel, wVel, tracer,
15 O gTracer,
16 I bi,bj, myTime,myIter,myThid)
17
18 C !DESCRIPTION:
19 C Calculates the tendency of a tracer due to advection.
20 C It uses the multi-dimensional method given in \ref{sect:multiDimAdvection}
21 C and can only be used for the non-linear advection schemes such as the
22 C direct-space-time method and flux-limiters.
23 C
24 C The algorithm is as follows:
25 C \begin{itemize}
26 C \item{$\theta^{(n+1/3)} = \theta^{(n)}
27 C - \Delta t \partial_x (u\theta^{(n)}) + \theta^{(n)} \partial_x u$}
28 C \item{$\theta^{(n+2/3)} = \theta^{(n+1/3)}
29 C - \Delta t \partial_y (v\theta^{(n+1/3)}) + \theta^{(n)} \partial_y v$}
30 C \item{$\theta^{(n+3/3)} = \theta^{(n+2/3)}
31 C - \Delta t \partial_r (w\theta^{(n+2/3)}) + \theta^{(n)} \partial_r w$}
32 C \item{$G_\theta = ( \theta^{(n+3/3)} - \theta^{(n)} )/\Delta t$}
33 C \end{itemize}
34 C
35 C The tendency (output) is over-written by this routine.
36
37 C !USES: ===============================================================
38 IMPLICIT NONE
39 #include "SIZE.h"
40 #include "EEPARAMS.h"
41 #include "PARAMS.h"
42 #include "GRID.h"
43 #include "GAD.h"
44 #ifdef ALLOW_AUTODIFF_TAMC
45 # include "tamc.h"
46 # include "tamc_keys.h"
47 # ifdef ALLOW_PTRACERS
48 # include "PTRACERS_SIZE.h"
49 # endif
50 #endif
51 #ifdef ALLOW_EXCH2
52 #include "W2_EXCH2_SIZE.h"
53 #include "W2_EXCH2_TOPOLOGY.h"
54 #endif /* ALLOW_EXCH2 */
55
56 C !INPUT PARAMETERS: ===================================================
57 C implicitAdvection :: implicit vertical advection (later on)
58 C advectionScheme :: advection scheme to use (Horizontal plane)
59 C vertAdvecScheme :: advection scheme to use (vertical direction)
60 C trIdentity :: tracer identifier
61 C uVel :: velocity, zonal component
62 C vVel :: velocity, meridional component
63 C wVel :: velocity, vertical component
64 C tracer :: tracer field
65 C bi,bj :: tile indices
66 C myTime :: current time
67 C myIter :: iteration number
68 C myThid :: thread number
69 LOGICAL implicitAdvection
70 INTEGER advectionScheme, vertAdvecScheme
71 INTEGER trIdentity
72 _RL deltaTLev(Nr)
73 _RL uVel (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
74 _RL vVel (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
75 _RL wVel (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
76 _RL tracer(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
77 INTEGER bi,bj
78 _RL myTime
79 INTEGER myIter
80 INTEGER myThid
81
82 C !OUTPUT PARAMETERS: ==================================================
83 C gTracer :: tendency array
84 _RL gTracer(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
85
86 C !LOCAL VARIABLES: ====================================================
87 C maskUp :: 2-D array for mask at W points
88 C maskLocW :: 2-D array for mask at West points
89 C maskLocS :: 2-D array for mask at South points
90 C [iMin,iMax]Upd :: loop range to update tracer field
91 C [jMin,jMax]Upd :: loop range to update tracer field
92 C i,j,k :: loop indices
93 C kUp :: index into 2 1/2D array, toggles between 1 and 2
94 C kDown :: index into 2 1/2D array, toggles between 2 and 1
95 C kp1 :: =k+1 for k<Nr, =Nr for k=Nr
96 C xA,yA :: areas of X and Y face of tracer cells
97 C uFld,vFld :: 2-D local copy of horizontal velocity, U,V components
98 C wFld :: 2-D local copy of vertical velocity
99 C uTrans,vTrans :: 2-D arrays of volume transports at U,V points
100 C rTrans :: 2-D arrays of volume transports at W points
101 C rTransKp1 :: vertical volume transport at interface k+1
102 C af :: 2-D array for horizontal advective flux
103 C afx :: 2-D array for horizontal advective flux, x direction
104 C afy :: 2-D array for horizontal advective flux, y direction
105 C fVerT :: 2 1/2D arrays for vertical advective flux
106 C localTij :: 2-D array, temporary local copy of tracer fld
107 C localTijk :: 3-D array, temporary local copy of tracer fld
108 C kp1Msk :: flag (0,1) for over-riding mask for W levels
109 C calc_fluxes_X :: logical to indicate to calculate fluxes in X dir
110 C calc_fluxes_Y :: logical to indicate to calculate fluxes in Y dir
111 C interiorOnly :: only update the interior of myTile, but not the edges
112 C overlapOnly :: only update the edges of myTile, but not the interior
113 C npass :: number of passes in multi-dimensional method
114 C ipass :: number of the current pass being made
115 C myTile :: variables used to determine which cube face
116 C nCFace :: owns a tile for cube grid runs using
117 C :: multi-dim advection.
118 C [N,S,E,W]_edge :: true if N,S,E,W edge of myTile is an Edge of the cube
119 c _RS maskUp (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
120 _RS maskLocW(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
121 _RS maskLocS(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
122 INTEGER iMinUpd,iMaxUpd,jMinUpd,jMaxUpd
123 INTEGER i,j,k,kUp,kDown
124 _RS xA (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
125 _RS yA (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
126 _RL uFld (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
127 _RL vFld (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
128 _RL wFld (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
129 _RL uTrans (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
130 _RL vTrans (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
131 _RL rTrans (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
132 _RL rTransKp1(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
133 _RL af (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
134 _RL afx (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
135 _RL afy (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
136 _RL fVerT (1-OLx:sNx+OLx,1-OLy:sNy+OLy,2)
137 _RL localTij(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
138 _RL localTijk(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr)
139 _RL kp1Msk
140 LOGICAL calc_fluxes_X, calc_fluxes_Y, withSigns
141 LOGICAL interiorOnly, overlapOnly
142 INTEGER npass, ipass
143 INTEGER nCFace
144 LOGICAL N_edge, S_edge, E_edge, W_edge
145 #ifdef ALLOW_AUTODIFF_TAMC
146 C msgBuf :: Informational/error message buffer
147 CHARACTER*(MAX_LEN_MBUF) msgBuf
148 #endif
149 #ifdef ALLOW_EXCH2
150 INTEGER myTile
151 #endif
152 #ifdef ALLOW_DIAGNOSTICS
153 CHARACTER*8 diagName
154 CHARACTER*4 diagSufx
155 LOGICAL doDiagAdvX, doDiagAdvY, doDiagAdvR
156 C- Functions:
157 CHARACTER*4 GAD_DIAG_SUFX
158 EXTERNAL GAD_DIAG_SUFX
159 LOGICAL DIAGNOSTICS_IS_ON
160 EXTERNAL DIAGNOSTICS_IS_ON
161 #endif
162 CEOP
163
164 #ifdef ALLOW_AUTODIFF_TAMC
165 act0 = trIdentity
166 max0 = maxpass
167 act1 = bi - myBxLo(myThid)
168 max1 = myBxHi(myThid) - myBxLo(myThid) + 1
169 act2 = bj - myByLo(myThid)
170 max2 = myByHi(myThid) - myByLo(myThid) + 1
171 act3 = myThid - 1
172 max3 = nTx*nTy
173 act4 = ikey_dynamics - 1
174 igadkey = act0
175 & + act1*max0
176 & + act2*max0*max1
177 & + act3*max0*max1*max2
178 & + act4*max0*max1*max2*max3
179 IF (trIdentity.GT.maxpass) THEN
180 WRITE(msgBuf,'(A,2I3)')
181 & 'GAD_ADVECTION: maxpass < trIdentity ',
182 & maxpass, trIdentity
183 CALL PRINT_ERROR( msgBuf, myThid )
184 STOP 'ABNORMAL END: S/R GAD_ADVECTION'
185 ENDIF
186 #endif /* ALLOW_AUTODIFF_TAMC */
187
188 #ifdef ALLOW_DIAGNOSTICS
189 C-- Set diagnostics flags and suffix for the current tracer
190 doDiagAdvX = .FALSE.
191 doDiagAdvY = .FALSE.
192 doDiagAdvR = .FALSE.
193 IF ( useDiagnostics ) THEN
194 diagSufx = GAD_DIAG_SUFX( trIdentity, myThid )
195 diagName = 'ADVx'//diagSufx
196 doDiagAdvX = DIAGNOSTICS_IS_ON( diagName, myThid )
197 diagName = 'ADVy'//diagSufx
198 doDiagAdvY = DIAGNOSTICS_IS_ON( diagName, myThid )
199 diagName = 'ADVr'//diagSufx
200 doDiagAdvR = DIAGNOSTICS_IS_ON( diagName, myThid )
201 ENDIF
202 #endif
203
204 C-- Set up work arrays with valid (i.e. not NaN) values
205 C These inital values do not alter the numerical results. They
206 C just ensure that all memory references are to valid floating
207 C point numbers. This prevents spurious hardware signals due to
208 C uninitialised but inert locations.
209 DO j=1-OLy,sNy+OLy
210 DO i=1-OLx,sNx+OLx
211 xA(i,j) = 0. _d 0
212 yA(i,j) = 0. _d 0
213 uTrans(i,j) = 0. _d 0
214 vTrans(i,j) = 0. _d 0
215 rTrans(i,j) = 0. _d 0
216 fVerT(i,j,1) = 0. _d 0
217 fVerT(i,j,2) = 0. _d 0
218 rTransKp1(i,j)= 0. _d 0
219 #ifdef ALLOW_AUTODIFF_TAMC
220 localTij(i,j) = 0. _d 0
221 wfld(i,j) = 0. _d 0
222 #endif
223 ENDDO
224 ENDDO
225
226 C-- Set tile-specific parameters for horizontal fluxes
227 IF (useCubedSphereExchange) THEN
228 npass = 3
229 #ifdef ALLOW_AUTODIFF_TAMC
230 IF ( npass.GT.maxcube ) STOP 'maxcube needs to be = 3'
231 #endif
232 #ifdef ALLOW_EXCH2
233 myTile = W2_myTileList(bi,bj)
234 nCFace = exch2_myFace(myTile)
235 N_edge = exch2_isNedge(myTile).EQ.1
236 S_edge = exch2_isSedge(myTile).EQ.1
237 E_edge = exch2_isEedge(myTile).EQ.1
238 W_edge = exch2_isWedge(myTile).EQ.1
239 #else
240 nCFace = bi
241 N_edge = .TRUE.
242 S_edge = .TRUE.
243 E_edge = .TRUE.
244 W_edge = .TRUE.
245 #endif
246 ELSE
247 npass = 2
248 nCFace = 0
249 N_edge = .FALSE.
250 S_edge = .FALSE.
251 E_edge = .FALSE.
252 W_edge = .FALSE.
253 ENDIF
254
255 C-- Start of k loop for horizontal fluxes
256 DO k=1,Nr
257 #ifdef ALLOW_AUTODIFF_TAMC
258 kkey = (igadkey-1)*Nr + k
259 CADJ STORE tracer(:,:,k,bi,bj) =
260 CADJ & comlev1_bibj_k_gad, key=kkey, kind=isbyte
261 #endif /* ALLOW_AUTODIFF_TAMC */
262
263 C-- Get temporary terms used by tendency routines
264 CALL CALC_COMMON_FACTORS (
265 I uVel, vVel,
266 O uFld, vFld, uTrans, vTrans, xA, yA,
267 I k,bi,bj, myThid )
268
269 #ifdef ALLOW_GMREDI
270 C-- Residual transp = Bolus transp + Eulerian transp
271 IF (useGMRedi)
272 & CALL GMREDI_CALC_UVFLOW(
273 U uFld, vFld, uTrans, vTrans,
274 I k, bi, bj, myThid )
275 #endif /* ALLOW_GMREDI */
276
277 C-- Make local copy of tracer array and mask West & South
278 DO j=1-OLy,sNy+OLy
279 DO i=1-OLx,sNx+OLx
280 localTij(i,j) = tracer(i,j,k,bi,bj)
281 #ifdef ALLOW_OBCS
282 maskLocW(i,j) = _maskW(i,j,k,bi,bj)*maskInW(i,j,bi,bj)
283 maskLocS(i,j) = _maskS(i,j,k,bi,bj)*maskInS(i,j,bi,bj)
284 #else /* ALLOW_OBCS */
285 maskLocW(i,j) = _maskW(i,j,k,bi,bj)
286 maskLocS(i,j) = _maskS(i,j,k,bi,bj)
287 #endif /* ALLOW_OBCS */
288 ENDDO
289 ENDDO
290
291 cph-exch2#ifndef ALLOW_AUTODIFF_TAMC
292 IF (useCubedSphereExchange) THEN
293 withSigns = .FALSE.
294 CALL FILL_CS_CORNER_UV_RS(
295 & withSigns, maskLocW,maskLocS, bi,bj, myThid )
296 ENDIF
297 cph-exch2#endif
298
299 C-- Multiple passes for different directions on different tiles
300 C-- For cube need one pass for each of red, green and blue axes.
301 DO ipass=1,npass
302 #ifdef ALLOW_AUTODIFF_TAMC
303 passkey = ipass
304 & + (k-1) *maxpass
305 & + (igadkey-1)*maxpass*Nr
306 IF (npass .GT. maxpass) THEN
307 STOP 'GAD_ADVECTION: npass > maxcube. check tamc.h'
308 ENDIF
309 #endif /* ALLOW_AUTODIFF_TAMC */
310
311 interiorOnly = .FALSE.
312 overlapOnly = .FALSE.
313 IF (useCubedSphereExchange) THEN
314 C- CubedSphere : pass 3 times, with partial update of local tracer field
315 IF (ipass.EQ.1) THEN
316 overlapOnly = MOD(nCFace,3).EQ.0
317 interiorOnly = MOD(nCFace,3).NE.0
318 calc_fluxes_X = nCFace.EQ.6 .OR. nCFace.EQ.1 .OR. nCFace.EQ.2
319 calc_fluxes_Y = nCFace.EQ.3 .OR. nCFace.EQ.4 .OR. nCFace.EQ.5
320 ELSEIF (ipass.EQ.2) THEN
321 overlapOnly = MOD(nCFace,3).EQ.2
322 interiorOnly = MOD(nCFace,3).EQ.1
323 calc_fluxes_X = nCFace.EQ.2 .OR. nCFace.EQ.3 .OR. nCFace.EQ.4
324 calc_fluxes_Y = nCFace.EQ.5 .OR. nCFace.EQ.6 .OR. nCFace.EQ.1
325 ELSE
326 interiorOnly = .TRUE.
327 calc_fluxes_X = nCFace.EQ.5 .OR. nCFace.EQ.6
328 calc_fluxes_Y = nCFace.EQ.2 .OR. nCFace.EQ.3
329 ENDIF
330 ELSE
331 C- not CubedSphere
332 calc_fluxes_X = MOD(ipass,2).EQ.1
333 calc_fluxes_Y = .NOT.calc_fluxes_X
334 ENDIF
335
336 C---+----1----+----2----+----3----+----4----+----5----+----6----+----7-|--+----|
337 C-- X direction
338 C- Advective flux in X
339 DO j=1-OLy,sNy+OLy
340 DO i=1-OLx,sNx+OLx
341 af(i,j) = 0.
342 ENDDO
343 ENDDO
344 C
345 #ifdef ALLOW_AUTODIFF_TAMC
346 # ifndef DISABLE_MULTIDIM_ADVECTION
347 CADJ STORE localTij(:,:) =
348 CADJ & comlev1_bibj_k_gad_pass, key=passkey, kind=isbyte
349 CADJ STORE af(:,:) =
350 CADJ & comlev1_bibj_k_gad_pass, key=passkey, kind=isbyte
351 # endif
352 #endif /* ALLOW_AUTODIFF_TAMC */
353 C
354 IF (calc_fluxes_X) THEN
355
356 C- Do not compute fluxes if
357 C a) needed in overlap only
358 C and b) the overlap of myTile are not cube-face Edges
359 IF ( .NOT.overlapOnly .OR. N_edge .OR. S_edge ) THEN
360
361 C- Internal exchange for calculations in X
362 IF ( overlapOnly ) THEN
363 CALL FILL_CS_CORNER_TR_RL( 1, .FALSE.,
364 & localTij, bi,bj, myThid )
365 ENDIF
366
367 #ifdef ALLOW_AUTODIFF_TAMC
368 # ifndef DISABLE_MULTIDIM_ADVECTION
369 CADJ STORE localTij(:,:) =
370 CADJ & comlev1_bibj_k_gad_pass, key=passkey, kind=isbyte
371 # endif
372 #endif /* ALLOW_AUTODIFF_TAMC */
373
374 IF ( advectionScheme.EQ.ENUM_UPWIND_1RST
375 & .OR. advectionScheme.EQ.ENUM_DST2 ) THEN
376 CALL GAD_DST2U1_ADV_X( bi,bj,k, advectionScheme, .TRUE.,
377 I deltaTLev(k),uTrans,uFld,localTij,
378 O af, myThid )
379 ELSEIF (advectionScheme.EQ.ENUM_FLUX_LIMIT) THEN
380 CALL GAD_FLUXLIMIT_ADV_X( bi,bj,k, .TRUE., deltaTLev(k),
381 I uTrans, uFld, maskLocW, localTij,
382 O af, myThid )
383 ELSEIF (advectionScheme.EQ.ENUM_DST3 ) THEN
384 CALL GAD_DST3_ADV_X( bi,bj,k, .TRUE., deltaTLev(k),
385 I uTrans, uFld, maskLocW, localTij,
386 O af, myThid )
387 ELSEIF (advectionScheme.EQ.ENUM_DST3_FLUX_LIMIT ) THEN
388 CALL GAD_DST3FL_ADV_X( bi,bj,k, .TRUE., deltaTLev(k),
389 I uTrans, uFld, maskLocW, localTij,
390 O af, myThid )
391 #ifndef ALLOW_AUTODIFF_TAMC
392 ELSEIF (advectionScheme.EQ.ENUM_OS7MP ) THEN
393 CALL GAD_OS7MP_ADV_X( bi,bj,k, .TRUE., deltaTLev(k),
394 I uTrans, uFld, maskLocW, localTij,
395 O af, myThid )
396 #endif
397 ELSE
398 STOP 'GAD_ADVECTION: adv. scheme incompatibale with multi-dim'
399 ENDIF
400
401 #ifdef ALLOW_OBCS
402 IF ( useOBCS ) THEN
403 C- replace advective flux with 1st order upwind scheme estimate
404 CALL OBCS_U1_ADV_TRACER( .TRUE., trIdentity, bi, bj, k,
405 I maskLocW, uTrans, localTij,
406 U af, myThid )
407 ENDIF
408 #endif /* ALLOW_OBCS */
409
410 C- Internal exchange for next calculations in Y
411 IF ( overlapOnly .AND. ipass.EQ.1 ) THEN
412 CALL FILL_CS_CORNER_TR_RL( 2, .FALSE.,
413 & localTij, bi,bj, myThid )
414 ENDIF
415
416 C- Advective flux in X : done
417 ENDIF
418
419 C- Update the local tracer field where needed:
420 C use "maksInC" to prevent updating tracer field in OB regions
421
422 C update in overlap-Only
423 IF ( overlapOnly ) THEN
424 iMinUpd = 1-OLx+1
425 iMaxUpd = sNx+OLx-1
426 C- notes: these 2 lines below have no real effect (because recip_hFac=0
427 C in corner region) but safer to keep them.
428 IF ( W_edge ) iMinUpd = 1
429 IF ( E_edge ) iMaxUpd = sNx
430
431 IF ( S_edge ) THEN
432 DO j=1-OLy,0
433 DO i=iMinUpd,iMaxUpd
434 localTij(i,j) = localTij(i,j)
435 & -deltaTLev(k)*recip_rhoFacC(k)
436 & *_recip_hFacC(i,j,k,bi,bj)*recip_drF(k)
437 & *recip_rA(i,j,bi,bj)*recip_deepFac2C(k)
438 & *( af(i+1,j)-af(i,j)
439 & -tracer(i,j,k,bi,bj)*(uTrans(i+1,j)-uTrans(i,j))
440 & )*maskInC(i,j,bi,bj)
441 ENDDO
442 ENDDO
443 ENDIF
444 IF ( N_edge ) THEN
445 DO j=sNy+1,sNy+OLy
446 DO i=iMinUpd,iMaxUpd
447 localTij(i,j) = localTij(i,j)
448 & -deltaTLev(k)*recip_rhoFacC(k)
449 & *_recip_hFacC(i,j,k,bi,bj)*recip_drF(k)
450 & *recip_rA(i,j,bi,bj)*recip_deepFac2C(k)
451 & *( af(i+1,j)-af(i,j)
452 & -tracer(i,j,k,bi,bj)*(uTrans(i+1,j)-uTrans(i,j))
453 & )*maskInC(i,j,bi,bj)
454 ENDDO
455 ENDDO
456 ENDIF
457
458 ELSE
459 C do not only update the overlap
460 jMinUpd = 1-OLy
461 jMaxUpd = sNy+OLy
462 IF ( interiorOnly .AND. S_edge ) jMinUpd = 1
463 IF ( interiorOnly .AND. N_edge ) jMaxUpd = sNy
464 DO j=jMinUpd,jMaxUpd
465 DO i=1-OLx+1,sNx+OLx-1
466 localTij(i,j) = localTij(i,j)
467 & -deltaTLev(k)*recip_rhoFacC(k)
468 & *_recip_hFacC(i,j,k,bi,bj)*recip_drF(k)
469 & *recip_rA(i,j,bi,bj)*recip_deepFac2C(k)
470 & *( af(i+1,j)-af(i,j)
471 & -tracer(i,j,k,bi,bj)*(uTrans(i+1,j)-uTrans(i,j))
472 & )*maskInC(i,j,bi,bj)
473 ENDDO
474 ENDDO
475 C- keep advective flux (for diagnostics)
476 DO j=1-OLy,sNy+OLy
477 DO i=1-OLx,sNx+OLx
478 afx(i,j) = af(i,j)
479 ENDDO
480 ENDDO
481
482 C- end if/else update overlap-Only
483 ENDIF
484
485 C-- End of X direction
486 ENDIF
487
488 C---+----1----+----2----+----3----+----4----+----5----+----6----+----7-|--+----|
489 C-- Y direction
490 cph-test
491 C- Advective flux in Y
492 DO j=1-OLy,sNy+OLy
493 DO i=1-OLx,sNx+OLx
494 af(i,j) = 0.
495 ENDDO
496 ENDDO
497 C
498 #ifdef ALLOW_AUTODIFF_TAMC
499 # ifndef DISABLE_MULTIDIM_ADVECTION
500 CADJ STORE localTij(:,:) =
501 CADJ & comlev1_bibj_k_gad_pass, key=passkey, kind=isbyte
502 CADJ STORE af(:,:) =
503 CADJ & comlev1_bibj_k_gad_pass, key=passkey, kind=isbyte
504 # endif
505 #endif /* ALLOW_AUTODIFF_TAMC */
506 C
507 IF (calc_fluxes_Y) THEN
508
509 C- Do not compute fluxes if
510 C a) needed in overlap only
511 C and b) the overlap of myTile are not cube-face edges
512 IF ( .NOT.overlapOnly .OR. E_edge .OR. W_edge ) THEN
513
514 C- Internal exchange for calculations in Y
515 IF ( overlapOnly ) THEN
516 CALL FILL_CS_CORNER_TR_RL( 2, .FALSE.,
517 & localTij, bi,bj, myThid )
518 ENDIF
519
520 C- Advective flux in Y
521 DO j=1-OLy,sNy+OLy
522 DO i=1-OLx,sNx+OLx
523 af(i,j) = 0.
524 ENDDO
525 ENDDO
526
527 #ifdef ALLOW_AUTODIFF_TAMC
528 #ifndef DISABLE_MULTIDIM_ADVECTION
529 CADJ STORE localTij(:,:) =
530 CADJ & comlev1_bibj_k_gad_pass, key=passkey, kind=isbyte
531 #endif
532 #endif /* ALLOW_AUTODIFF_TAMC */
533
534 IF ( advectionScheme.EQ.ENUM_UPWIND_1RST
535 & .OR. advectionScheme.EQ.ENUM_DST2 ) THEN
536 CALL GAD_DST2U1_ADV_Y( bi,bj,k, advectionScheme, .TRUE.,
537 I deltaTLev(k),vTrans,vFld,localTij,
538 O af, myThid )
539 ELSEIF (advectionScheme.EQ.ENUM_FLUX_LIMIT) THEN
540 CALL GAD_FLUXLIMIT_ADV_Y( bi,bj,k, .TRUE., deltaTLev(k),
541 I vTrans, vFld, maskLocS, localTij,
542 O af, myThid )
543 ELSEIF (advectionScheme.EQ.ENUM_DST3 ) THEN
544 CALL GAD_DST3_ADV_Y( bi,bj,k, .TRUE., deltaTLev(k),
545 I vTrans, vFld, maskLocS, localTij,
546 O af, myThid )
547 ELSEIF (advectionScheme.EQ.ENUM_DST3_FLUX_LIMIT ) THEN
548 CALL GAD_DST3FL_ADV_Y( bi,bj,k, .TRUE., deltaTLev(k),
549 I vTrans, vFld, maskLocS, localTij,
550 O af, myThid )
551 #ifndef ALLOW_AUTODIFF_TAMC
552 ELSEIF (advectionScheme.EQ.ENUM_OS7MP ) THEN
553 CALL GAD_OS7MP_ADV_Y( bi,bj,k, .TRUE., deltaTLev(k),
554 I vTrans, vFld, maskLocS, localTij,
555 O af, myThid )
556 #endif
557 ELSE
558 STOP 'GAD_ADVECTION: adv. scheme incompatibale with mutli-dim'
559 ENDIF
560
561 #ifdef ALLOW_OBCS
562 IF ( useOBCS ) THEN
563 C- replace advective flux with 1st order upwind scheme estimate
564 CALL OBCS_U1_ADV_TRACER( .FALSE., trIdentity, bi, bj, k,
565 I maskLocS, vTrans, localTij,
566 U af, myThid )
567 ENDIF
568 #endif /* ALLOW_OBCS */
569
570 C- Internal exchange for next calculations in X
571 IF ( overlapOnly .AND. ipass.EQ.1 ) THEN
572 CALL FILL_CS_CORNER_TR_RL( 1, .FALSE.,
573 & localTij, bi,bj, myThid )
574 ENDIF
575
576 C- Advective flux in Y : done
577 ENDIF
578
579 C- Update the local tracer field where needed:
580 C use "maksInC" to prevent updating tracer field in OB regions
581
582 C update in overlap-Only
583 IF ( overlapOnly ) THEN
584 jMinUpd = 1-OLy+1
585 jMaxUpd = sNy+OLy-1
586 C- notes: these 2 lines below have no real effect (because recip_hFac=0
587 C in corner region) but safer to keep them.
588 IF ( S_edge ) jMinUpd = 1
589 IF ( N_edge ) jMaxUpd = sNy
590
591 IF ( W_edge ) THEN
592 DO j=jMinUpd,jMaxUpd
593 DO i=1-OLx,0
594 localTij(i,j) = localTij(i,j)
595 & -deltaTLev(k)*recip_rhoFacC(k)
596 & *_recip_hFacC(i,j,k,bi,bj)*recip_drF(k)
597 & *recip_rA(i,j,bi,bj)*recip_deepFac2C(k)
598 & *( af(i,j+1)-af(i,j)
599 & -tracer(i,j,k,bi,bj)*(vTrans(i,j+1)-vTrans(i,j))
600 & )*maskInC(i,j,bi,bj)
601 ENDDO
602 ENDDO
603 ENDIF
604 IF ( E_edge ) THEN
605 DO j=jMinUpd,jMaxUpd
606 DO i=sNx+1,sNx+OLx
607 localTij(i,j) = localTij(i,j)
608 & -deltaTLev(k)*recip_rhoFacC(k)
609 & *_recip_hFacC(i,j,k,bi,bj)*recip_drF(k)
610 & *recip_rA(i,j,bi,bj)*recip_deepFac2C(k)
611 & *( af(i,j+1)-af(i,j)
612 & -tracer(i,j,k,bi,bj)*(vTrans(i,j+1)-vTrans(i,j))
613 & )*maskInC(i,j,bi,bj)
614 ENDDO
615 ENDDO
616 ENDIF
617
618 ELSE
619 C do not only update the overlap
620 iMinUpd = 1-OLx
621 iMaxUpd = sNx+OLx
622 IF ( interiorOnly .AND. W_edge ) iMinUpd = 1
623 IF ( interiorOnly .AND. E_edge ) iMaxUpd = sNx
624 DO j=1-OLy+1,sNy+OLy-1
625 DO i=iMinUpd,iMaxUpd
626 localTij(i,j) = localTij(i,j)
627 & -deltaTLev(k)*recip_rhoFacC(k)
628 & *_recip_hFacC(i,j,k,bi,bj)*recip_drF(k)
629 & *recip_rA(i,j,bi,bj)*recip_deepFac2C(k)
630 & *( af(i,j+1)-af(i,j)
631 & -tracer(i,j,k,bi,bj)*(vTrans(i,j+1)-vTrans(i,j))
632 & )*maskInC(i,j,bi,bj)
633 ENDDO
634 ENDDO
635 C- keep advective flux (for diagnostics)
636 DO j=1-OLy,sNy+OLy
637 DO i=1-OLx,sNx+OLx
638 afy(i,j) = af(i,j)
639 ENDDO
640 ENDDO
641
642 C end if/else update overlap-Only
643 ENDIF
644
645 C-- End of Y direction
646 ENDIF
647
648 C-- End of ipass loop
649 ENDDO
650
651 IF ( implicitAdvection ) THEN
652 C- explicit advection is done ; store tendency in gTracer:
653 DO j=1-OLy,sNy+OLy
654 DO i=1-OLx,sNx+OLx
655 gTracer(i,j,k,bi,bj)=
656 & (localTij(i,j)-tracer(i,j,k,bi,bj))/deltaTLev(k)
657 ENDDO
658 ENDDO
659 ELSE
660 C- horizontal advection done; store intermediate result in 3D array:
661 DO j=1-OLy,sNy+OLy
662 DO i=1-OLx,sNx+OLx
663 localTijk(i,j,k) = localTij(i,j)
664 ENDDO
665 ENDDO
666 ENDIF
667
668 #ifdef ALLOW_DIAGNOSTICS
669 IF ( doDiagAdvX ) THEN
670 diagName = 'ADVx'//diagSufx
671 CALL DIAGNOSTICS_FILL( afx, diagName, k,1, 2,bi,bj, myThid )
672 ENDIF
673 IF ( doDiagAdvY ) THEN
674 diagName = 'ADVy'//diagSufx
675 CALL DIAGNOSTICS_FILL( afy, diagName, k,1, 2,bi,bj, myThid )
676 ENDIF
677 #endif
678
679 #ifdef ALLOW_DEBUG
680 IF ( debugLevel .GE. debLevC
681 & .AND. trIdentity.EQ.GAD_TEMPERATURE
682 & .AND. k.LE.3 .AND. myIter.EQ.1+nIter0
683 & .AND. nPx.EQ.1 .AND. nPy.EQ.1
684 & .AND. useCubedSphereExchange ) THEN
685 CALL DEBUG_CS_CORNER_UV( ' afx,afy from GAD_ADVECTION',
686 & afx,afy, k, standardMessageUnit,bi,bj,myThid )
687 ENDIF
688 #endif /* ALLOW_DEBUG */
689
690 C-- End of K loop for horizontal fluxes
691 ENDDO
692
693 C---+----1----+----2----+----3----+----4----+----5----+----6----+----7-|--+----|
694
695 IF ( .NOT.implicitAdvection ) THEN
696 C-- Start of k loop for vertical flux
697 DO k=Nr,1,-1
698 #ifdef ALLOW_AUTODIFF_TAMC
699 kkey = (igadkey-1)*Nr + (Nr-k+1)
700 #endif /* ALLOW_AUTODIFF_TAMC */
701 C-- kUp Cycles through 1,2 to point to w-layer above
702 C-- kDown Cycles through 2,1 to point to w-layer below
703 kUp = 1+MOD(k+1,2)
704 kDown= 1+MOD(k,2)
705 c kp1=min(Nr,k+1)
706 kp1Msk=1.
707 if (k.EQ.Nr) kp1Msk=0.
708
709 #ifdef ALLOW_AUTODIFF_TAMC
710 CADJ STORE rtrans(:,:) =
711 CADJ & comlev1_bibj_k_gad, key=kkey, kind=isbyte
712 cphCADJ STORE wfld(:,:) =
713 cphCADJ & comlev1_bibj_k_gad, key=kkey, kind=isbyte
714 #endif
715
716 C-- Compute Vertical transport
717 #ifdef ALLOW_AIM
718 C- a hack to prevent Water-Vapor vert.transport into the stratospheric level Nr
719 IF ( k.EQ.1 .OR.
720 & (useAIM .AND. trIdentity.EQ.GAD_SALINITY .AND. k.EQ.Nr)
721 & ) THEN
722 #else
723 IF ( k.EQ.1 ) THEN
724 #endif
725
726 #ifdef ALLOW_AUTODIFF_TAMC
727 cphmultiCADJ STORE wfld(:,:) =
728 cphmultiCADJ & comlev1_bibj_k_gad, key=kkey, kind=isbyte
729 #endif /* ALLOW_AUTODIFF_TAMC */
730
731 C- Surface interface :
732 DO j=1-OLy,sNy+OLy
733 DO i=1-OLx,sNx+OLx
734 rTransKp1(i,j) = kp1Msk*rTrans(i,j)
735 wFld(i,j) = 0.
736 rTrans(i,j) = 0.
737 fVerT(i,j,kUp) = 0.
738 ENDDO
739 ENDDO
740
741 ELSE
742
743 #ifdef ALLOW_AUTODIFF_TAMC
744 cphmultiCADJ STORE wfld(:,:) =
745 cphmultiCADJ & comlev1_bibj_k_gad, key=kkey, kind=isbyte
746 #endif /* ALLOW_AUTODIFF_TAMC */
747
748 C- Interior interface :
749 DO j=1-OLy,sNy+OLy
750 DO i=1-OLx,sNx+OLx
751 rTransKp1(i,j) = kp1Msk*rTrans(i,j)
752 wFld(i,j) = wVel(i,j,k,bi,bj)
753 rTrans(i,j) = wVel(i,j,k,bi,bj)*rA(i,j,bi,bj)
754 & *deepFac2F(k)*rhoFacF(k)
755 & *maskC(i,j,k-1,bi,bj)
756 fVerT(i,j,kUp) = 0.
757 ENDDO
758 ENDDO
759
760 #ifdef ALLOW_GMREDI
761 C-- Residual transp = Bolus transp + Eulerian transp
762 IF (useGMRedi)
763 & CALL GMREDI_CALC_WFLOW(
764 U wFld, rTrans,
765 I k, bi, bj, myThid )
766 #endif /* ALLOW_GMREDI */
767
768 #ifdef ALLOW_AUTODIFF_TAMC
769 cphmultiCADJ STORE localTijk(:,:,k)
770 cphmultiCADJ & = comlev1_bibj_k_gad, key=kkey, kind=isbyte
771 cphmultiCADJ STORE rTrans(:,:)
772 cphmultiCADJ & = comlev1_bibj_k_gad, key=kkey, kind=isbyte
773 #endif /* ALLOW_AUTODIFF_TAMC */
774
775 C- Compute vertical advective flux in the interior:
776 IF ( vertAdvecScheme.EQ.ENUM_UPWIND_1RST
777 & .OR. vertAdvecScheme.EQ.ENUM_DST2 ) THEN
778 CALL GAD_DST2U1_ADV_R( bi,bj,k, advectionScheme,
779 I deltaTLev(k),rTrans,wFld,localTijk,
780 O fVerT(1-OLx,1-OLy,kUp), myThid )
781 ELSEIF( vertAdvecScheme.EQ.ENUM_FLUX_LIMIT) THEN
782 CALL GAD_FLUXLIMIT_ADV_R( bi,bj,k, deltaTLev(k),
783 I rTrans, wFld, localTijk,
784 O fVerT(1-OLx,1-OLy,kUp), myThid )
785 ELSEIF( vertAdvecScheme.EQ.ENUM_DST3 ) THEN
786 CALL GAD_DST3_ADV_R( bi,bj,k, deltaTLev(k),
787 I rTrans, wFld, localTijk,
788 O fVerT(1-OLx,1-OLy,kUp), myThid )
789 ELSEIF( vertAdvecScheme.EQ.ENUM_DST3_FLUX_LIMIT ) THEN
790 CALL GAD_DST3FL_ADV_R( bi,bj,k, deltaTLev(k),
791 I rTrans, wFld, localTijk,
792 O fVerT(1-OLx,1-OLy,kUp), myThid )
793 #ifndef ALLOW_AUTODIFF_TAMC
794 ELSEIF (vertAdvecScheme.EQ.ENUM_OS7MP ) THEN
795 CALL GAD_OS7MP_ADV_R( bi,bj,k, deltaTLev(k),
796 I rTrans, wFld, localTijk,
797 O fVerT(1-OLx,1-OLy,kUp), myThid )
798 #endif
799 ELSE
800 STOP 'GAD_ADVECTION: adv. scheme incompatibale with mutli-dim'
801 ENDIF
802
803 C- end Surface/Interior if bloc
804 ENDIF
805
806 #ifdef ALLOW_AUTODIFF_TAMC
807 cphmultiCADJ STORE rTrans(:,:)
808 cphmultiCADJ & = comlev1_bibj_k_gad, key=kkey, kind=isbyte
809 cphmultiCADJ STORE rTranskp1(:,:)
810 cphmultiCADJ & = comlev1_bibj_k_gad, key=kkey, kind=isbyte
811 cph --- following storing of fVerT is critical for correct
812 cph --- gradient with multiDimAdvection
813 cph --- Without it, kDown component is not properly recomputed
814 cph --- This is a TAF bug (and no warning available)
815 CADJ STORE fVerT(:,:,:)
816 CADJ & = comlev1_bibj_k_gad, key=kkey, kind=isbyte
817 #endif /* ALLOW_AUTODIFF_TAMC */
818
819 C-- Divergence of vertical fluxes
820 DO j=1-OLy,sNy+OLy
821 DO i=1-OLx,sNx+OLx
822 localTij(i,j) = localTijk(i,j,k)
823 & -deltaTLev(k)*recip_rhoFacC(k)
824 & *_recip_hFacC(i,j,k,bi,bj)*recip_drF(k)
825 & *recip_rA(i,j,bi,bj)*recip_deepFac2C(k)
826 & *( fVerT(i,j,kDown)-fVerT(i,j,kUp)
827 & -tracer(i,j,k,bi,bj)*(rTransKp1(i,j)-rTrans(i,j))
828 & )*rkSign*maskInC(i,j,bi,bj)
829 gTracer(i,j,k,bi,bj)=
830 & (localTij(i,j)-tracer(i,j,k,bi,bj))/deltaTLev(k)
831 ENDDO
832 ENDDO
833
834 #ifdef ALLOW_DIAGNOSTICS
835 IF ( doDiagAdvR ) THEN
836 diagName = 'ADVr'//diagSufx
837 CALL DIAGNOSTICS_FILL( fVerT(1-OLx,1-OLy,kUp),
838 & diagName, k,1, 2,bi,bj, myThid )
839 ENDIF
840 #endif
841
842 C-- End of K loop for vertical flux
843 ENDDO
844 C-- end of if not.implicitAdvection block
845 ENDIF
846
847 RETURN
848 END

  ViewVC Help
Powered by ViewVC 1.1.22