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

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

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


Revision 1.42 - (hide annotations) (download)
Tue Jul 18 03:23:50 2006 UTC (17 years, 11 months ago) by heimbach
Branch: MAIN
CVS Tags: checkpoint58o_post, checkpoint58p_post, checkpoint58q_post, checkpoint58r_post, checkpoint58n_post
Changes since 1.41: +17 -2 lines
rtrans recomputation warning treated.

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

  ViewVC Help
Powered by ViewVC 1.1.22