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

Diff of /MITgcm/pkg/generic_advdiff/gad_calc_rhs.F

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

revision 1.12.6.1 by heimbach, Sun Mar 24 17:22:22 2002 UTC revision 1.36 by jmc, Wed Jun 22 00:27:47 2005 UTC
# Line 9  C !ROUTINE: GAD_CALC_RHS Line 9  C !ROUTINE: GAD_CALC_RHS
9  C !INTERFACE: ==========================================================  C !INTERFACE: ==========================================================
10        SUBROUTINE GAD_CALC_RHS(        SUBROUTINE GAD_CALC_RHS(
11       I           bi,bj,iMin,iMax,jMin,jMax,k,kM1,kUp,kDown,       I           bi,bj,iMin,iMax,jMin,jMax,k,kM1,kUp,kDown,
12       I           xA,yA,uTrans,vTrans,rTrans,maskUp,       I           xA,yA,uTrans,vTrans,rTrans,rTransKp1,maskUp,
13       I           diffKh, diffK4, KappaRT, Tracer,       I           uVel, vVel, wVel,
14       I           tracerIdentity, advectionScheme,       I           diffKh, diffK4, KappaR, Tracer,
15         I           tracerIdentity, advectionScheme, vertAdvecScheme,
16         I           calcAdvection, implicitAdvection,
17       U           fVerT, gTracer,       U           fVerT, gTracer,
18       I           myThid )       I           myTime, myIter, myThid )
19    
20  C !DESCRIPTION:  C !DESCRIPTION:
21  C Calculates the tendancy of a tracer due to advection and diffusion.  C Calculates the tendancy of a tracer due to advection and diffusion.
# Line 40  C !USES: =============================== Line 42  C !USES: ===============================
42  #include "EEPARAMS.h"  #include "EEPARAMS.h"
43  #include "PARAMS.h"  #include "PARAMS.h"
44  #include "GRID.h"  #include "GRID.h"
45  #include "DYNVARS.h"  #include "SURFACE.h"
46  #include "GAD.h"  #include "GAD.h"
47    
48  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
# Line 49  C !USES: =============================== Line 51  C !USES: ===============================
51  #endif /* ALLOW_AUTODIFF_TAMC */  #endif /* ALLOW_AUTODIFF_TAMC */
52    
53  C !INPUT PARAMETERS: ===================================================  C !INPUT PARAMETERS: ===================================================
54  C  bi,bj                :: tile indices  C bi,bj            :: tile indices
55  C  iMin,iMax,jMin,jMax  :: loop range for called routines  C iMin,iMax        :: loop range for called routines
56  C  kup                  :: index into 2 1/2D array, toggles between 1 and 2  C jMin,jMax        :: loop range for called routines
57  C  kdown                :: index into 2 1/2D array, toggles between 2 and 1  C kup              :: index into 2 1/2D array, toggles between 1|2
58  C  kp1                  :: =k+1 for k<Nr, =Nr for k=Nr  C kdown            :: index into 2 1/2D array, toggles between 2|1
59  C  xA,yA                :: areas of X and Y face of tracer cells  C kp1              :: =k+1 for k<Nr, =Nr for k=Nr
60  C  uTrans,vTrans,rTrans :: 2-D arrays of volume transports at U,V and W points  C xA,yA            :: areas of X and Y face of tracer cells
61  C  maskUp               :: 2-D array for mask at W points  C uTrans,vTrans    :: 2-D arrays of volume transports at U,V points
62  C  diffKh               :: horizontal diffusion coefficient  C rTrans           :: 2-D arrays of volume transports at W points
63  C  diffK4               :: bi-harmonic diffusion coefficient  C rTransKp1        :: 2-D array of volume trans at W pts, interf k+1
64  C  KappaRT              :: 3-D array for vertical diffusion coefficient  C maskUp           :: 2-D array for mask at W points
65  C  Tracer               :: tracer field  C uVel,vVel,wVel   :: 3 components of the velcity field (3-D array)
66  C  tracerIdentity       :: identifier for the tracer (required only for KPP)  C diffKh           :: horizontal diffusion coefficient
67  C  advectionScheme      :: advection scheme to use  C diffK4           :: bi-harmonic diffusion coefficient
68  C  myThid               :: thread number  C KappaR           :: 2-D array for vertical diffusion coefficient, interf k
69    C Tracer           :: tracer field
70    C tracerIdentity   :: tracer identifier (required for KPP,GM)
71    C advectionScheme  :: advection scheme to use (Horizontal plane)
72    C vertAdvecScheme  :: advection scheme to use (Vertical direction)
73    C calcAdvection    :: =False if Advec computed with multiDim scheme
74    C implicitAdvection:: =True if vertical Advec computed implicitly
75    C myTime           :: current time
76    C myIter           :: iteration number
77    C myThid           :: thread number
78        INTEGER bi,bj,iMin,iMax,jMin,jMax        INTEGER bi,bj,iMin,iMax,jMin,jMax
79        INTEGER k,kUp,kDown,kM1        INTEGER k,kUp,kDown,kM1
80        _RS xA    (1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RS xA    (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
# Line 71  C  myThid               :: thread number Line 82  C  myThid               :: thread number
82        _RL uTrans(1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RL uTrans(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
83        _RL vTrans(1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RL vTrans(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
84        _RL rTrans(1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RL rTrans(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
85          _RL rTransKp1(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
86        _RS maskUp(1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RS maskUp(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
87          _RL uVel  (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
88          _RL vVel  (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
89          _RL wVel  (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
90        _RL diffKh, diffK4        _RL diffKh, diffK4
91        _RL KappaRT(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr)        _RL KappaR(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
92        _RL Tracer(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)        _RL Tracer(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
93        INTEGER tracerIdentity        INTEGER tracerIdentity
94        INTEGER advectionScheme        INTEGER advectionScheme, vertAdvecScheme
95        INTEGER myThid        LOGICAL calcAdvection
96          LOGICAL implicitAdvection
97          _RL     myTime
98          INTEGER myIter, myThid
99    
100  C !OUTPUT PARAMETERS: ==================================================  C !OUTPUT PARAMETERS: ==================================================
101  C  gTracer              :: tendancy array  C gTracer          :: tendancy array
102  C  fVerT                :: 2 1/2D arrays for vertical advective flux  C fVerT            :: 2 1/2D arrays for vertical advective flux
103        _RL gTracer(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)        _RL gTracer(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
104        _RL fVerT (1-OLx:sNx+OLx,1-OLy:sNy+OLy,2)        _RL fVerT (1-OLx:sNx+OLx,1-OLy:sNy+OLy,2)
105    
106  C !LOCAL VARIABLES: ====================================================  C !LOCAL VARIABLES: ====================================================
107  C  i,j                  :: loop indices  C i,j              :: loop indices
108  C  df4                  :: used for storing del^2 T for bi-harmonic term  C df4              :: used for storing del^2 T for bi-harmonic term
109  C  fZon                 :: zonal flux  C fZon             :: zonal flux
110  C  fmer                 :: meridional flux  C fMer             :: meridional flux
111  C  af                   :: advective flux  C af               :: advective flux
112  C  df                   :: diffusive flux  C df               :: diffusive flux
113  C  localT               :: local copy of tracer field  C localT           :: local copy of tracer field
114    #ifdef ALLOW_DIAGNOSTICS
115          CHARACTER*8 diagName
116          CHARACTER*4 GAD_DIAG_SUFX, diagSufx
117          EXTERNAL    GAD_DIAG_SUFX
118    #endif
119        INTEGER i,j        INTEGER i,j
120        _RL df4   (1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RL df4   (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
121        _RL fZon  (1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RL fZon  (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
# Line 100  C  localT               :: local copy of Line 123  C  localT               :: local copy of
123        _RL af    (1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RL af    (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
124        _RL df    (1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RL df    (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
125        _RL localT(1-OLx:sNx+OLx,1-OLy:sNy+OLy)        _RL localT(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
126          _RL advFac, rAdvFac
127  CEOP  CEOP
128    
129  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
# Line 108  C--   the kDown is still required Line 132  C--   the kDown is still required
132        fVerT(1,1,kDown) = fVerT(1,1,kDown)        fVerT(1,1,kDown) = fVerT(1,1,kDown)
133  #endif  #endif
134    
135    #ifdef ALLOW_DIAGNOSTICS
136    C--   Set diagnostic suffix for the current tracer
137          IF ( useDiagnostics ) THEN
138            diagSufx = GAD_DIAG_SUFX( tracerIdentity, myThid )
139          ENDIF
140    #endif
141    
142          advFac  = 0. _d 0
143          IF (calcAdvection) advFac = 1. _d 0
144          rAdvFac = rkSign*advFac
145          IF (implicitAdvection) rAdvFac = 0. _d 0
146    
147        DO j=1-OLy,sNy+OLy        DO j=1-OLy,sNy+OLy
148         DO i=1-OLx,sNx+OLx         DO i=1-OLx,sNx+OLx
149          fZon(i,j)      = 0. _d 0          fZon(i,j)      = 0. _d 0
# Line 115  C--   the kDown is still required Line 151  C--   the kDown is still required
151          fVerT(i,j,kUp) = 0. _d 0          fVerT(i,j,kUp) = 0. _d 0
152          df(i,j)        = 0. _d 0          df(i,j)        = 0. _d 0
153          df4(i,j)       = 0. _d 0          df4(i,j)       = 0. _d 0
         localT(i,j)    = 0. _d 0  
154         ENDDO         ENDDO
155        ENDDO        ENDDO
156    
# Line 128  C--   Make local copy of tracer array Line 163  C--   Make local copy of tracer array
163    
164  C--   Unless we have already calculated the advection terms we initialize  C--   Unless we have already calculated the advection terms we initialize
165  C     the tendency to zero.  C     the tendency to zero.
166        IF (.NOT. multiDimAdvection .OR.  C     <== now done earlier at the beginning of thermodynamics.
167       &    advectionScheme.EQ.ENUM_CENTERED_2ND .OR.  c     IF (calcAdvection) THEN
168       &    advectionScheme.EQ.ENUM_UPWIND_3RD .OR.  c      DO j=1-Oly,sNy+Oly
169       &    advectionScheme.EQ.ENUM_CENTERED_4TH ) THEN  c       DO i=1-Olx,sNx+Olx
170         DO j=1-Oly,sNy+Oly  c        gTracer(i,j,k,bi,bj)=0. _d 0
171          DO i=1-Olx,sNx+Olx  c       ENDDO
172           gTracer(i,j,k,bi,bj)=0. _d 0  c      ENDDO
173          ENDDO  c     ENDIF
        ENDDO  
       ENDIF  
174    
175  C--   Pre-calculate del^2 T if bi-harmonic coefficient is non-zero  C--   Pre-calculate del^2 T if bi-harmonic coefficient is non-zero
176        IF (diffK4 .NE. 0.) THEN        IF (diffK4 .NE. 0.) THEN
# Line 154  C--   Initialize net flux in X direction Line 187  C--   Initialize net flux in X direction
187        ENDDO        ENDDO
188    
189  C-    Advective flux in X  C-    Advective flux in X
190        IF (.NOT. multiDimAdvection .OR.        IF (calcAdvection) THEN
191       &    advectionScheme.EQ.ENUM_CENTERED_2ND .OR.          IF (advectionScheme.EQ.ENUM_CENTERED_2ND) THEN
192       &    advectionScheme.EQ.ENUM_UPWIND_3RD .OR.            CALL GAD_C2_ADV_X(bi,bj,k,uTrans,localT,af,myThid)
193       &    advectionScheme.EQ.ENUM_CENTERED_4TH ) THEN          ELSEIF (advectionScheme.EQ.ENUM_FLUX_LIMIT) THEN
194        IF (advectionScheme.EQ.ENUM_CENTERED_2ND) THEN            CALL GAD_FLUXLIMIT_ADV_X( bi,bj,k, dTtracerLev(k),
195         CALL GAD_C2_ADV_X(bi,bj,k,uTrans,localT,af,myThid)       I            uTrans, uVel, maskW(1-Olx,1-Oly,k,bi,bj), localT,
196        ELSEIF (advectionScheme.EQ.ENUM_FLUX_LIMIT) THEN       O            af, myThid )
197         CALL GAD_FLUXLIMIT_ADV_X(          ELSEIF (advectionScheme.EQ.ENUM_UPWIND_3RD ) THEN
198       &      bi,bj,k,deltaTtracer,uTrans,uVel,localT,af,myThid)            CALL GAD_U3_ADV_X(bi,bj,k,uTrans,localT,af,myThid)
199        ELSEIF (advectionScheme.EQ.ENUM_UPWIND_3RD ) THEN          ELSEIF (advectionScheme.EQ.ENUM_CENTERED_4TH) THEN
200         CALL GAD_U3_ADV_X(bi,bj,k,uTrans,localT,af,myThid)            CALL GAD_C4_ADV_X(bi,bj,k,uTrans,localT,af,myThid)
201        ELSEIF (advectionScheme.EQ.ENUM_CENTERED_4TH) THEN          ELSEIF (advectionScheme.EQ.ENUM_DST3 ) THEN
202         CALL GAD_C4_ADV_X(bi,bj,k,uTrans,localT,af,myThid)            CALL GAD_DST3_ADV_X( bi,bj,k, dTtracerLev(k),
203        ELSEIF (advectionScheme.EQ.ENUM_DST3 ) THEN       I            uTrans, uVel, maskW(1-Olx,1-Oly,k,bi,bj), localT,
204         CALL GAD_DST3_ADV_X(       O            af, myThid )
205       &       bi,bj,k,deltaTtracer,uTrans,uVel,localT,af,myThid)          ELSEIF (advectionScheme.EQ.ENUM_DST3_FLUX_LIMIT ) THEN
206        ELSEIF (advectionScheme.EQ.ENUM_DST3_FLUX_LIMIT ) THEN           IF ( inAdMode ) THEN
207         CALL GAD_DST3FL_ADV_X(  cph This block is to trick the adjoint:
208       &       bi,bj,k,deltaTtracer,uTrans,uVel,localT,af,myThid)  cph IF inAdExact=.FALSE., we want to use DST3
209        ELSE  cph with limiters in forward, but without limiters in reverse.
210         STOP 'GAD_CALC_RHS: Bad advectionScheme (X)'            CALL GAD_DST3_ADV_X( bi,bj,k, dTtracerLev(k),
211        ENDIF       I           uTrans, uVel, maskW(1-Olx,1-Oly,k,bi,bj), localT,
212        DO j=1-Oly,sNy+Oly       O           af, myThid )
213         DO i=1-Olx,sNx+Olx           ELSE
214          fZon(i,j) = fZon(i,j) + af(i,j)            CALL GAD_DST3FL_ADV_X( bi,bj,k, dTtracerLev(k),
215         ENDDO       I           uTrans, uVel, maskW(1-Olx,1-Oly,k,bi,bj), localT,
216        ENDDO       O           af, myThid )
217             ENDIF
218            ELSE
219             STOP 'GAD_CALC_RHS: Bad advectionScheme (X)'
220            ENDIF
221            DO j=1-Oly,sNy+Oly
222             DO i=1-Olx,sNx+Olx
223              fZon(i,j) = fZon(i,j) + af(i,j)
224             ENDDO
225            ENDDO
226    #ifdef ALLOW_DIAGNOSTICS
227            IF ( useDiagnostics ) THEN
228              diagName = 'ADVx'//diagSufx
229              CALL DIAGNOSTICS_FILL(af,diagName, k,1, 2,bi,bj, myThid)
230            ENDIF
231    #endif
232        ENDIF        ENDIF
233    
234  C-    Diffusive flux in X  C-    Diffusive flux in X
# Line 194  C-    Diffusive flux in X Line 242  C-    Diffusive flux in X
242         ENDDO         ENDDO
243        ENDIF        ENDIF
244    
245    C-    Add bi-harmonic diffusive flux in X
246          IF (diffK4 .NE. 0.) THEN
247           CALL GAD_BIHARM_X(bi,bj,k,xA,df4,diffK4,df,myThid)
248          ENDIF
249    
250  #ifdef ALLOW_GMREDI  #ifdef ALLOW_GMREDI
251  C-    GM/Redi flux in X  C-    GM/Redi flux in X
252        IF (useGMRedi) THEN        IF (useGMRedi) THEN
253  C *note* should update GMREDI_XTRANSPORT to use localT and set df  *aja*  C *note* should update GMREDI_XTRANSPORT to use localT and set df  *aja*
254          CALL GMREDI_XTRANSPORT(          CALL GMREDI_XTRANSPORT(
255       I     iMin,iMax,jMin,jMax,bi,bj,K,       I     iMin,iMax,jMin,jMax,bi,bj,K,
256       I     xA,Tracer,       I     xA,Tracer,tracerIdentity,
257       U     df,       U     df,
258       I     myThid)       I     myThid)
259        ENDIF        ENDIF
# Line 211  C *note* should update GMREDI_XTRANSPORT Line 264  C *note* should update GMREDI_XTRANSPORT
264         ENDDO         ENDDO
265        ENDDO        ENDDO
266    
267  C-    Bi-harmonic duffusive flux in X  #ifdef ALLOW_DIAGNOSTICS
268        IF (diffK4 .NE. 0.) THEN  C-    Diagnostics of Tracer flux in X dir (mainly Diffusive term),
269         CALL GAD_BIHARM_X(bi,bj,k,xA,df4,diffK4,df,myThid)  C       excluding advective terms:
270         DO j=1-Oly,sNy+Oly        IF ( useDiagnostics .AND.
271          DO i=1-Olx,sNx+Olx       &    (diffKh.NE.0. .OR. diffK4 .NE.0. .OR. useGMRedi) ) THEN
272           fZon(i,j) = fZon(i,j) + df(i,j)            diagName = 'DIFx'//diagSufx
273          ENDDO            CALL DIAGNOSTICS_FILL(df,diagName, k,1, 2,bi,bj, myThid)
        ENDDO  
274        ENDIF        ENDIF
275    #endif
276    
277  C--   Initialize net flux in Y direction  C--   Initialize net flux in Y direction
278        DO j=1-Oly,sNy+Oly        DO j=1-Oly,sNy+Oly
# Line 229  C--   Initialize net flux in Y direction Line 282  C--   Initialize net flux in Y direction
282        ENDDO        ENDDO
283    
284  C-    Advective flux in Y  C-    Advective flux in Y
285        IF (.NOT. multiDimAdvection .OR.        IF (calcAdvection) THEN
286       &    advectionScheme.EQ.ENUM_CENTERED_2ND .OR.          IF (advectionScheme.EQ.ENUM_CENTERED_2ND) THEN
287       &    advectionScheme.EQ.ENUM_UPWIND_3RD .OR.            CALL GAD_C2_ADV_Y(bi,bj,k,vTrans,localT,af,myThid)
288       &    advectionScheme.EQ.ENUM_CENTERED_4TH ) THEN          ELSEIF (advectionScheme.EQ.ENUM_FLUX_LIMIT) THEN
289        IF (advectionScheme.EQ.ENUM_CENTERED_2ND) THEN            CALL GAD_FLUXLIMIT_ADV_Y( bi,bj,k, dTtracerLev(k),
290         CALL GAD_C2_ADV_Y(bi,bj,k,vTrans,localT,af,myThid)       I            vTrans, vVel, maskS(1-Olx,1-Oly,k,bi,bj), localT,
291        ELSEIF (advectionScheme.EQ.ENUM_FLUX_LIMIT) THEN       O            af, myThid )
292         CALL GAD_FLUXLIMIT_ADV_Y(          ELSEIF (advectionScheme.EQ.ENUM_UPWIND_3RD ) THEN
293       &       bi,bj,k,deltaTtracer,vTrans,vVel,localT,af,myThid)            CALL GAD_U3_ADV_Y(bi,bj,k,vTrans,localT,af,myThid)
294        ELSEIF (advectionScheme.EQ.ENUM_UPWIND_3RD ) THEN          ELSEIF (advectionScheme.EQ.ENUM_CENTERED_4TH) THEN
295         CALL GAD_U3_ADV_Y(bi,bj,k,vTrans,localT,af,myThid)            CALL GAD_C4_ADV_Y(bi,bj,k,vTrans,localT,af,myThid)
296        ELSEIF (advectionScheme.EQ.ENUM_CENTERED_4TH) THEN          ELSEIF (advectionScheme.EQ.ENUM_DST3 ) THEN
297         CALL GAD_C4_ADV_Y(bi,bj,k,vTrans,localT,af,myThid)            CALL GAD_DST3_ADV_Y( bi,bj,k, dTtracerLev(k),
298        ELSEIF (advectionScheme.EQ.ENUM_DST3 ) THEN       I            vTrans, vVel, maskS(1-Olx,1-Oly,k,bi,bj), localT,
299         CALL GAD_DST3_ADV_Y(       O            af, myThid )
300       &       bi,bj,k,deltaTtracer,vTrans,vVel,localT,af,myThid)          ELSEIF (advectionScheme.EQ.ENUM_DST3_FLUX_LIMIT ) THEN
301        ELSEIF (advectionScheme.EQ.ENUM_DST3_FLUX_LIMIT ) THEN           IF ( inAdMode ) THEN
302         CALL GAD_DST3FL_ADV_Y(  cph This block is to trick the adjoint:
303       &       bi,bj,k,deltaTtracer,vTrans,vVel,localT,af,myThid)  cph IF inAdExact=.FALSE., we want to use DST3
304        ELSE  cph with limiters in forward, but without limiters in reverse.
305         STOP 'GAD_CALC_RHS: Bad advectionScheme (Y)'            CALL GAD_DST3_ADV_Y( bi,bj,k, dTtracerLev(k),
306        ENDIF       I           vTrans, vVel, maskS(1-Olx,1-Oly,k,bi,bj), localT,
307        DO j=1-Oly,sNy+Oly       O           af, myThid )
308         DO i=1-Olx,sNx+Olx           ELSE
309          fMer(i,j) = fMer(i,j) + af(i,j)            CALL GAD_DST3FL_ADV_Y( bi,bj,k, dTtracerLev(k),
310         ENDDO       I           vTrans, vVel, maskS(1-Olx,1-Oly,k,bi,bj), localT,
311        ENDDO       O           af, myThid )
312             ENDIF
313            ELSE
314              STOP 'GAD_CALC_RHS: Bad advectionScheme (Y)'
315            ENDIF
316            DO j=1-Oly,sNy+Oly
317             DO i=1-Olx,sNx+Olx
318              fMer(i,j) = fMer(i,j) + af(i,j)
319             ENDDO
320            ENDDO
321    #ifdef ALLOW_DIAGNOSTICS
322            IF ( useDiagnostics ) THEN
323              diagName = 'ADVy'//diagSufx
324              CALL DIAGNOSTICS_FILL(af,diagName, k,1, 2,bi,bj, myThid)
325            ENDIF
326    #endif
327        ENDIF        ENDIF
328    
329  C-    Diffusive flux in Y  C-    Diffusive flux in Y
# Line 269  C-    Diffusive flux in Y Line 337  C-    Diffusive flux in Y
337         ENDDO         ENDDO
338        ENDIF        ENDIF
339    
340    C-    Add bi-harmonic flux in Y
341          IF (diffK4 .NE. 0.) THEN
342           CALL GAD_BIHARM_Y(bi,bj,k,yA,df4,diffK4,df,myThid)
343          ENDIF
344    
345  #ifdef ALLOW_GMREDI  #ifdef ALLOW_GMREDI
346  C-    GM/Redi flux in Y  C-    GM/Redi flux in Y
347        IF (useGMRedi) THEN        IF (useGMRedi) THEN
348  C *note* should update GMREDI_YTRANSPORT to use localT and set df  *aja*  C *note* should update GMREDI_YTRANSPORT to use localT and set df  *aja*
349         CALL GMREDI_YTRANSPORT(         CALL GMREDI_YTRANSPORT(
350       I     iMin,iMax,jMin,jMax,bi,bj,K,       I     iMin,iMax,jMin,jMax,bi,bj,K,
351       I     yA,Tracer,       I     yA,Tracer,tracerIdentity,
352       U     df,       U     df,
353       I     myThid)       I     myThid)
354        ENDIF        ENDIF
# Line 286  C *note* should update GMREDI_YTRANSPORT Line 359  C *note* should update GMREDI_YTRANSPORT
359         ENDDO         ENDDO
360        ENDDO        ENDDO
361    
362  C-    Bi-harmonic flux in Y  #ifdef ALLOW_DIAGNOSTICS
363        IF (diffK4 .NE. 0.) THEN  C-    Diagnostics of Tracer flux in Y dir (mainly Diffusive terms),
364         CALL GAD_BIHARM_Y(bi,bj,k,yA,df4,diffK4,df,myThid)  C       excluding advective terms:
365         DO j=1-Oly,sNy+Oly        IF ( useDiagnostics .AND.
366          DO i=1-Olx,sNx+Olx       &    (diffKh.NE.0. .OR. diffK4 .NE.0. .OR. useGMRedi) ) THEN
367           fMer(i,j) = fMer(i,j) + df(i,j)            diagName = 'DIFy'//diagSufx
368          ENDDO            CALL DIAGNOSTICS_FILL(df,diagName, k,1, 2,bi,bj, myThid)
        ENDDO  
369        ENDIF        ENDIF
370    #endif
371    
372    C--   Compute vertical flux fVerT(kUp) at interface k (between k-1 & k):
373  C-    Advective flux in R  C-    Advective flux in R
374        IF (.NOT. multiDimAdvection .OR.  #ifdef ALLOW_AIM
375       &    advectionScheme.EQ.ENUM_CENTERED_2ND .OR.  C- a hack to prevent Water-Vapor vert.transport into the stratospheric level Nr
376       &    advectionScheme.EQ.ENUM_UPWIND_3RD .OR.        IF (calcAdvection .AND. .NOT.implicitAdvection .AND. K.GE.2 .AND.
377       &    advectionScheme.EQ.ENUM_CENTERED_4TH ) THEN       &     (.NOT.useAIM .OR.tracerIdentity.NE.GAD_SALINITY .OR.K.LT.Nr)
378  C     Note: wVel needs to be masked       &   ) THEN
379        IF (K.GE.2) THEN  #else
380          IF (calcAdvection .AND. .NOT.implicitAdvection .AND. K.GE.2) THEN
381    #endif
382  C-    Compute vertical advective flux in the interior:  C-    Compute vertical advective flux in the interior:
383         IF (advectionScheme.EQ.ENUM_CENTERED_2ND) THEN          IF (vertAdvecScheme.EQ.ENUM_CENTERED_2ND) THEN
384          CALL GAD_C2_ADV_R(bi,bj,k,rTrans,tracer,af,myThid)            CALL GAD_C2_ADV_R(bi,bj,k,rTrans,tracer,af,myThid)
385         ELSEIF (advectionScheme.EQ.ENUM_FLUX_LIMIT) THEN          ELSEIF (vertAdvecScheme.EQ.ENUM_FLUX_LIMIT) THEN
386          CALL GAD_FLUXLIMIT_ADV_R(            CALL GAD_FLUXLIMIT_ADV_R(
387       &       bi,bj,k,deltaTtracer,rTrans,wVel,tracer,af,myThid)       &         bi,bj,k,dTtracerLev(k),rTrans,wVel,tracer,af,myThid)
388         ELSEIF (advectionScheme.EQ.ENUM_UPWIND_3RD ) THEN          ELSEIF (vertAdvecScheme.EQ.ENUM_UPWIND_3RD ) THEN
389          CALL GAD_U3_ADV_R(bi,bj,k,rTrans,tracer,af,myThid)            CALL GAD_U3_ADV_R(bi,bj,k,rTrans,tracer,af,myThid)
390         ELSEIF (advectionScheme.EQ.ENUM_CENTERED_4TH) THEN          ELSEIF (vertAdvecScheme.EQ.ENUM_CENTERED_4TH) THEN
391          CALL GAD_C4_ADV_R(bi,bj,k,rTrans,tracer,af,myThid)            CALL GAD_C4_ADV_R(bi,bj,k,rTrans,tracer,af,myThid)
392         ELSEIF (advectionScheme.EQ.ENUM_DST3 ) THEN          ELSEIF (vertAdvecScheme.EQ.ENUM_DST3 ) THEN
393          CALL GAD_DST3_ADV_R(            CALL GAD_DST3_ADV_R(
394       &       bi,bj,k,deltaTtracer,rTrans,wVel,tracer,af,myThid)       &         bi,bj,k,dTtracerLev(k),rTrans,wVel,tracer,af,myThid)
395         ELSEIF (advectionScheme.EQ.ENUM_DST3_FLUX_LIMIT ) THEN          ELSEIF (vertAdvecScheme.EQ.ENUM_DST3_FLUX_LIMIT ) THEN
396          CALL GAD_DST3FL_ADV_R(  cph This block is to trick the adjoint:
397       &       bi,bj,k,deltaTtracer,rTrans,wVel,tracer,af,myThid)  cph IF inAdExact=.FALSE., we want to use DST3
398         ELSE  cph with limiters in forward, but without limiters in reverse.
399          STOP 'GAD_CALC_RHS: Bad advectionScheme (R)'            IF ( inAdMode ) THEN
400         ENDIF             CALL GAD_DST3_ADV_R(
401  C-    Surface "correction" term at k>1 :       &        bi,bj,k,dTtracerLev(k),rTrans,wVel,tracer,af,myThid)
402         DO j=1-Oly,sNy+Oly            ELSE
403          DO i=1-Olx,sNx+Olx             CALL GAD_DST3FL_ADV_R(
404           af(i,j) = af(i,j)       &        bi,bj,k,dTtracerLev(k),rTrans,wVel,tracer,af,myThid)
405       &           + (maskC(i,j,k,bi,bj)-maskC(i,j,k-1,bi,bj))*            ENDIF
406       &             rTrans(i,j)*Tracer(i,j,k,bi,bj)          ELSE
407              STOP 'GAD_CALC_RHS: Bad vertAdvecScheme (R)'
408            ENDIF
409    C-     add the advective flux to fVerT
410            DO j=1-Oly,sNy+Oly
411             DO i=1-Olx,sNx+Olx
412              fVerT(i,j,kUp) = fVerT(i,j,kUp) + af(i,j)
413             ENDDO
414          ENDDO          ENDDO
415         ENDDO  #ifdef ALLOW_DIAGNOSTICS
416        ELSE          IF ( useDiagnostics ) THEN
417  C-    Surface "correction" term at k=1 :            diagName = 'ADVr'//diagSufx
418         DO j=1-Oly,sNy+Oly            CALL DIAGNOSTICS_FILL(af,diagName, k,1, 2,bi,bj, myThid)
419          DO i=1-Olx,sNx+Olx  C- note: needs to explicitly increment the counter since DIAGNOSTICS_FILL
420           af(i,j) = rTrans(i,j)*Tracer(i,j,k,bi,bj)  C        does it only if k=1 (never the case here)
421          ENDDO            IF ( k.EQ.2 ) CALL DIAGNOSTICS_COUNT(diagName,bi,bj,myThid)
422         ENDDO          ENDIF
423        ENDIF  #endif
 C-    add the advective flux to fVerT  
       DO j=1-Oly,sNy+Oly  
        DO i=1-Olx,sNx+Olx  
         fVerT(i,j,kUp) = fVerT(i,j,kUp) + af(i,j)  
        ENDDO  
       ENDDO  
424        ENDIF        ENDIF
425    
426  C-    Diffusive flux in R  C-    Diffusive flux in R
# Line 356  C           boundary condition. Line 433  C           boundary condition.
433          ENDDO          ENDDO
434         ENDDO         ENDDO
435        ELSE        ELSE
436         CALL GAD_DIFF_R(bi,bj,k,KappaRT,tracer,df,myThid)         CALL GAD_DIFF_R(bi,bj,k,KappaR,tracer,df,myThid)
437        ENDIF        ENDIF
 c     DO j=1-Oly,sNy+Oly  
 c      DO i=1-Olx,sNx+Olx  
 c       fVerT(i,j,kUp) = fVerT(i,j,kUp) + df(i,j)*maskUp(i,j)  
 c      ENDDO  
 c     ENDDO  
438    
439  #ifdef ALLOW_GMREDI  #ifdef ALLOW_GMREDI
440  C-    GM/Redi flux in R  C-    GM/Redi flux in R
# Line 370  C-    GM/Redi flux in R Line 442  C-    GM/Redi flux in R
442  C *note* should update GMREDI_RTRANSPORT to set df  *aja*  C *note* should update GMREDI_RTRANSPORT to set df  *aja*
443         CALL GMREDI_RTRANSPORT(         CALL GMREDI_RTRANSPORT(
444       I     iMin,iMax,jMin,jMax,bi,bj,K,       I     iMin,iMax,jMin,jMax,bi,bj,K,
445       I     Tracer,       I     Tracer,tracerIdentity,
446       U     df,       U     df,
447       I     myThid)       I     myThid)
 c      DO j=1-Oly,sNy+Oly  
 c       DO i=1-Olx,sNx+Olx  
 c        fVerT(i,j,kUp) = fVerT(i,j,kUp) + df(i,j)*maskUp(i,j)  
 c       ENDDO  
 c      ENDDO  
448        ENDIF        ENDIF
449  #endif  #endif
450    
# Line 387  c      ENDDO Line 454  c      ENDDO
454         ENDDO         ENDDO
455        ENDDO        ENDDO
456    
457    #ifdef ALLOW_DIAGNOSTICS
458    C-    Diagnostics of Tracer flux in R dir (mainly Diffusive terms),
459    C       Explicit terms only & excluding advective terms:
460          IF ( useDiagnostics .AND.
461         &    (.NOT.implicitDiffusion .OR. useGMRedi) ) THEN
462              diagName = 'DFrE'//diagSufx
463              CALL DIAGNOSTICS_FILL(df,diagName, k,1, 2,bi,bj, myThid)
464          ENDIF
465    #endif
466    
467  #ifdef ALLOW_KPP  #ifdef ALLOW_KPP
468  C-    Add non local KPP transport term (ghat) to diffusive T flux.  C-    Set non local KPP transport term (ghat):
469        IF (useKPP) THEN        IF ( useKPP .AND. k.GE.2 ) THEN
470         DO j=1-Oly,sNy+Oly         DO j=1-Oly,sNy+Oly
471          DO i=1-Olx,sNx+Olx          DO i=1-Olx,sNx+Olx
472           df(i,j) = 0. _d 0           df(i,j) = 0. _d 0
473          ENDDO          ENDDO
474         ENDDO         ENDDO
475         IF (tracerIdentity.EQ.GAD_TEMPERATURE) THEN         IF (tracerIdentity.EQ.GAD_TEMPERATURE) THEN
 C *note* should update KPP_TRANSPORT_T to set df  *aja*  
476          CALL KPP_TRANSPORT_T(          CALL KPP_TRANSPORT_T(
477       I     iMin,iMax,jMin,jMax,bi,bj,k,km1,       I     iMin,iMax,jMin,jMax,bi,bj,k,km1,
478       I     KappaRT,       O     df )
      U     df )  
479         ELSEIF (tracerIdentity.EQ.GAD_SALINITY) THEN         ELSEIF (tracerIdentity.EQ.GAD_SALINITY) THEN
480          CALL KPP_TRANSPORT_S(          CALL KPP_TRANSPORT_S(
481       I     iMin,iMax,jMin,jMax,bi,bj,k,km1,       I     iMin,iMax,jMin,jMax,bi,bj,k,km1,
482       I     KappaRT,       O     df )
483       U     df )  #ifdef ALLOW_PTRACERS
484           ELSEIF (tracerIdentity .GE. GAD_TR1) THEN
485            CALL KPP_TRANSPORT_PTR(
486         I     iMin,iMax,jMin,jMax,bi,bj,k,km1,
487         I     tracerIdentity-GAD_TR1+1,
488         O     df )
489    #endif
490         ELSE         ELSE
491            PRINT*,'invalid tracer indentity: ', tracerIdentity
492          STOP 'GAD_CALC_RHS: Ooops'          STOP 'GAD_CALC_RHS: Ooops'
493         ENDIF         ENDIF
494         DO j=1-Oly,sNy+Oly         DO j=1-Oly,sNy+Oly
# Line 421  C--   Divergence of fluxes Line 503  C--   Divergence of fluxes
503        DO j=1-Oly,sNy+Oly-1        DO j=1-Oly,sNy+Oly-1
504         DO i=1-Olx,sNx+Olx-1         DO i=1-Olx,sNx+Olx-1
505          gTracer(i,j,k,bi,bj)=gTracer(i,j,k,bi,bj)          gTracer(i,j,k,bi,bj)=gTracer(i,j,k,bi,bj)
506       &   -_recip_hFacC(i,j,k,bi,bj)*recip_drF(k)       &   -_recip_hFacC(i,j,k,bi,bj)*recip_drF(k)*recip_rA(i,j,bi,bj)
507       &    *recip_rA(i,j,bi,bj)       &   *( (fZon(i+1,j)-fZon(i,j))
508       &    *(       &     +(fMer(i,j+1)-fMer(i,j))
509       &    +( fZon(i+1,j)-fZon(i,j) )       &     +(fVerT(i,j,kDown)-fVerT(i,j,kUp))*rkSign
510       &    +( fMer(i,j+1)-fMer(i,j) )       &     -localT(i,j)*( (uTrans(i+1,j)-uTrans(i,j))
511       &    +( fVerT(i,j,kUp)-fVerT(i,j,kDown) )*rkFac       &                   +(vTrans(i,j+1)-vTrans(i,j))
512         &                   +(rTransKp1(i,j)-rTrans(i,j))*rAdvFac
513         &                  )*advFac
514       &    )       &    )
515         ENDDO         ENDDO
516        ENDDO        ENDDO
517    
518    #ifdef ALLOW_DEBUG
519          IF ( debugLevel .GE. debLevB
520         &   .AND. tracerIdentity.EQ.GAD_TEMPERATURE
521         &   .AND. k.EQ.2 .AND. myIter.EQ.1+nIter0
522         &   .AND. nPx.EQ.1 .AND. nPy.EQ.1
523         &   .AND. useCubedSphereExchange ) THEN
524            CALL DEBUG_CS_CORNER_UV( ' fZon,fMer from GAD_CALC_RHS',
525         &             fZon,fMer, k, standardMessageUnit,bi,bj,myThid )
526          ENDIF
527    #endif /* ALLOW_DEBUG */
528    
529        RETURN        RETURN
530        END        END
Legend:
Removed from v.1.12.6.1  
changed lines
  Added in v.1.36
  ViewVC Help
Powered by ViewVC 1.1.22