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

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

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


Revision 1.6 - (hide annotations) (download)
Fri Oct 5 10:50:47 2007 UTC (16 years, 7 months ago) by mlosch
Branch: MAIN
CVS Tags: checkpoint59q, checkpoint59p, checkpoint59r, checkpoint59m, checkpoint59l, checkpoint59o, checkpoint59n, checkpoint59i, checkpoint59k, checkpoint59j
Changes since 1.5: +16 -1 lines
thanks to Jens-Olaf Beismann, these advection routines will now vectorize

1 mlosch 1.6 C $Header: /u/gcmpack/MITgcm/pkg/generic_advdiff/gad_os7mp_adv_r.F,v 1.5 2007/06/13 18:45:27 jmc Exp $
2 mlosch 1.2 C $Name: $
3 adcroft 1.1
4     #include "GAD_OPTIONS.h"
5    
6     SUBROUTINE GAD_OS7MP_ADV_R(
7     I bi,bj,k,deltaTloc,
8     I wTrans, wFld,
9     I Q,
10     O wT,
11     I myThid )
12     C /==========================================================\
13     C | SUBROUTINE GAD_OS7MP_ADV_R |
14     C | o Compute Vertical advective Flux of tracer Q using |
15     C | 7th Order DST Sceheme with monotone preserving limiter |
16     C |==========================================================|
17     IMPLICIT NONE
18    
19     C == GLobal variables ==
20     #include "SIZE.h"
21     #include "GRID.h"
22     #include "GAD.h"
23    
24     C == Routine arguments ==
25     INTEGER bi,bj,k
26     _RL deltaTloc
27     _RL wTrans(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
28     _RL wFld (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
29     _RL Q (1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr)
30     _RL wT (1-OLx:sNx+OLx,1-OLy:sNy+OLy)
31     INTEGER myThid
32    
33     C == Local variables ==
34     INTEGER i,j,kp3,kp2,kp1,km1,km2,km3,km4
35     _RL cfl,Psi
36 adcroft 1.4 _RL wLoc,Fac,DelIp,DelI,Phi,Eps,rp1h,rp1h_cfl
37 adcroft 1.1 _RL Qippp,Qipp,Qip,Qi,Qim,Qimm,Qimmm
38     _RL MskIpp,MskIp,MskI,MskIm,MskImm,MskImmm
39     _RL d2,d2p1,d2m1,A,B,C,D
40     _RL dp1h,dm1h,qMD,qUL,qLC,PhiMD,PhiLC,PhiMin,PhiMax
41 adcroft 1.4 _RL DelM,DelP,DelMM,DelPP,DelMMM,DelPPP
42     _RL Del2MM,Del2M,Del2,Del2P,Del2PP
43     _RL Del3MM,Del3M,Del3P,Del3PP
44     _RL Del4M,Del4,Del4P
45     _RL Del5M,Del5P
46     _RL Del6
47 adcroft 1.1
48     Eps = 1. _d -20
49    
50     km4=MAX(1,k-4)
51     km3=MAX(1,k-3)
52     km2=MAX(1,k-2)
53     km1=MAX(1,k-1)
54     kp1=MIN(Nr,k+1)
55     kp2=MIN(Nr,k+2)
56     kp3=MIN(Nr,k+3)
57    
58     DO j=1-Oly,sNy+Oly
59     DO i=1-Olx,sNx+Olx
60    
61     wLoc = wFld(i,j)
62     cfl = abs(wLoc*deltaTloc*recip_drC(k))
63    
64 adcroft 1.3 IF (wTrans(i,j).lt.0.) THEN
65 adcroft 1.1 Qippp = Q(i,j,kp2)
66     Qipp = Q(i,j,kp1)
67     Qip = Q(i,j,k)
68     Qi = Q(i,j,km1)
69     Qim = Q(i,j,km2)
70     Qimm = Q(i,j,km3)
71     Qimmm = Q(i,j,km4)
72    
73     MskIpp = maskC(i,j,kp2,bi,bj) * float(kp2-kp1)
74     MskIp = maskC(i,j,kp1,bi,bj) * float(kp1-k)
75     MskI = maskC(i,j,k,bi,bj) * float(k-km1)
76     MskIm = maskC(i,j,km1,bi,bj) * float(km1-km2)
77     MskImm = maskC(i,j,km2,bi,bj) * float(km2-km3)
78     MskImmm = maskC(i,j,km3,bi,bj) * float(km3-km4)
79 adcroft 1.3 ELSEIF (wTrans(i,j).gt.0.) THEN
80 adcroft 1.1 Qippp = Q(i,j,km3)
81     Qipp = Q(i,j,km2)
82     Qip = Q(i,j,km1)
83     Qi = Q(i,j,k)
84     Qim = Q(i,j,kp1)
85     Qimm = Q(i,j,kp2)
86     Qimmm = Q(i,j,kp3)
87    
88     MskIpp = maskC(i,j,km2,bi,bj) * float(km2-km3)
89     MskIp = maskC(i,j,km1,bi,bj) * float(km1-km2)
90     MskI = maskC(i,j,k,bi,bj) * float(k-km1)
91     MskIm = maskC(i,j,kp1,bi,bj) * float(kp1-k)
92     MskImm = maskC(i,j,kp2,bi,bj) * float(kp2-kp1)
93     MskImmm = maskC(i,j,kp3,bi,bj) * float(kp3-kp2)
94 mlosch 1.6 ELSE
95     Qippp = 0. _d 0
96     Qipp = 0. _d 0
97     Qip = 0. _d 0
98     Qi = 0. _d 0
99     Qim = 0. _d 0
100     Qimm = 0. _d 0
101     Qimmm = 0. _d 0
102    
103     MskIpp = 0. _d 0
104     MskIp = 0. _d 0
105     MskI = 0. _d 0
106     MskIm = 0. _d 0
107     MskImm = 0. _d 0
108     MskImmm = 0. _d 0
109 adcroft 1.1 ENDIF
110    
111 adcroft 1.3 IF (wTrans(i,j).ne.0.) THEN
112 adcroft 1.1 C 2nd order correction [i i-1]
113     Fac = 1.
114 adcroft 1.4 DelP = (Qip-Qi)*MskI
115     Phi = Fac * DelP
116 adcroft 1.1 C 3rd order correction [i i-1 i-2]
117     Fac = Fac * ( cfl + 1. )/3.
118 adcroft 1.4 DelM = (Qi-Qim)*MskIm
119     Del2 = DelP - DelM
120     Phi = Phi - Fac * Del2
121 adcroft 1.1 C 4th order correction [i+1 i i-1 i-2]
122     Fac = Fac * ( cfl - 2. )/4.
123 adcroft 1.4 DelPP = (Qipp-Qip)*MskIp*MskI
124     Del2P = DelPP - DelP
125     Del3P = Del2P - Del2
126     Phi = Phi + Fac * Del3p
127 adcroft 1.1 C 5th order correction [i+1 i i-1 i-2 i-3]
128     Fac = Fac * ( cfl - 3. )/5.
129 adcroft 1.4 DelMM = (Qim-Qimm)*MskImm*MskIm
130     Del2M = DelM - DelMM
131     Del3M = Del2 - Del2M
132     Del4 = Del3P - Del3M
133     Phi = Phi + Fac * Del4
134 adcroft 1.1 C 6th order correction [i+2 i+1 i i-1 i-2 i-3]
135     Fac = Fac * ( cfl + 2. )/6.
136 adcroft 1.4 DelPPP = (Qippp-Qipp)*MskIpp*MskIp*MskI
137     Del2PP = DelPP - DelP
138     Del3PP = Del2PP - Del2P
139     Del4P = Del3PP - Del3P
140     Del5P = Del4P - Del4
141     Phi = Phi + Fac * Del5P
142 adcroft 1.1 C 7th order correction [i+2 i+1 i i-1 i-2 i-3 i-4]
143     Fac = Fac * ( cfl + 2. )/7.
144 adcroft 1.4 DelMMM = (Qimm-Qimmm)*MskImmm*MskImm*MskIm
145     Del2MM = DelMM - DelMMM
146     Del3MM = Del2M - Del2MM
147     Del4M = Del3M - Del3MM
148     Del5M = Del4 - Del4M
149     Del6 = Del5P - Del5M
150     Phi = Phi - Fac * Del6
151 adcroft 1.1
152     DelIp = ( Qip - Qi ) * MskI
153 mlosch 1.2 Phi = sign(1. _d 0,Phi)*sign(1. _d 0,DelIp)
154     & *abs(Phi+Eps)/abs(DelIp+Eps)
155 adcroft 1.1
156     DelI = ( Qi - Qim ) * MskIm
157 mlosch 1.2 rp1h =sign(1. _d 0,DelI)*sign(1. _d 0,DelIp)
158     & *abs(DelI+Eps)/abs(DelIp+Eps)
159 adcroft 1.3 rp1h_cfl = rp1h/(cfl+Eps)
160 adcroft 1.1
161     C TVD limiter
162 adcroft 1.3 ! Phi = max(0. _d 0, min( 2./(1-cfl), Phi, 2.*rp1h_cfl ) )
163 adcroft 1.1
164     C MP limiter
165 adcroft 1.4 d2 = Del2 !( ( Qip + Qim ) - 2.*Qi ) * MskI * MskIm
166     d2p1 = Del2P !( ( Qipp + Qi ) - 2.*Qip ) * MskIp * MskI
167     d2m1 = Del2M !( ( Qi + Qimm ) - 2.*Qim ) * MskIm * MskImm
168 adcroft 1.1 A = 4.*d2 - d2p1
169     B = 4.*d2p1 - d2
170     C = d2
171 jmc 1.5 D = d2p1
172 mlosch 1.2 dp1h = max(min(A,B,C,D),0. _d 0)+min(max(A,B,C,D),0. _d 0)
173 adcroft 1.1 A = 4.*d2m1 - d2
174     B = 4.*d2 - d2m1
175     C = d2m1
176 jmc 1.5 D = d2
177 mlosch 1.2 dm1h = max(min(A,B,C,D),0. _d 0)+min(max(A,B,C,D),0. _d 0)
178 adcroft 1.4 !qMD = 0.5*( ( Qi + Qip ) - dp1h )
179     qMD = 0.5*( ( 2.*Qi + DelIp ) - dp1h )
180     qUL = Qi + (1.-cfl)/(cfl+Eps)*DelI
181     qLC = Qi + 0.5*( 1.+dm1h/(DelI+Eps) )*(qUL-Qi)
182     PhiMD = 2./(1.-cfl)*(qMD-Qi+Eps)/(DelIp+Eps)
183 adcroft 1.3 PhiLC = 2.*rp1h_cfl*(qLC-Qi+Eps)/(qUL-Qi+Eps)
184 mlosch 1.2 PhiMin = max(min(0. _d 0,PhiMD),
185 adcroft 1.3 & min(0. _d 0,2.*rp1h_cfl,PhiLC))
186 mlosch 1.2 PhiMax = min(max(2. _d 0/(1.-cfl),PhiMD),
187 adcroft 1.3 & max(0. _d 0,2.*rp1h_cfl,PhiLC))
188 adcroft 1.1 Phi = max(PhiMin,min(Phi,PhiMax))
189    
190     Psi = Phi * 0.5 * (1. - cfl)
191     wT(i,j) = wTrans(i,j)*( Qi + Psi*DelIp )
192     ELSE
193     wT(i,j) = 0.
194     ENDIF
195    
196     ENDDO
197     ENDDO
198    
199     RETURN
200     END

  ViewVC Help
Powered by ViewVC 1.1.22