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

Contents 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 - (show 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 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 C $Name: $
3
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 _RL wLoc,Fac,DelIp,DelI,Phi,Eps,rp1h,rp1h_cfl
37 _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 _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
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 IF (wTrans(i,j).lt.0.) THEN
65 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 ELSEIF (wTrans(i,j).gt.0.) THEN
80 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 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 ENDIF
110
111 IF (wTrans(i,j).ne.0.) THEN
112 C 2nd order correction [i i-1]
113 Fac = 1.
114 DelP = (Qip-Qi)*MskI
115 Phi = Fac * DelP
116 C 3rd order correction [i i-1 i-2]
117 Fac = Fac * ( cfl + 1. )/3.
118 DelM = (Qi-Qim)*MskIm
119 Del2 = DelP - DelM
120 Phi = Phi - Fac * Del2
121 C 4th order correction [i+1 i i-1 i-2]
122 Fac = Fac * ( cfl - 2. )/4.
123 DelPP = (Qipp-Qip)*MskIp*MskI
124 Del2P = DelPP - DelP
125 Del3P = Del2P - Del2
126 Phi = Phi + Fac * Del3p
127 C 5th order correction [i+1 i i-1 i-2 i-3]
128 Fac = Fac * ( cfl - 3. )/5.
129 DelMM = (Qim-Qimm)*MskImm*MskIm
130 Del2M = DelM - DelMM
131 Del3M = Del2 - Del2M
132 Del4 = Del3P - Del3M
133 Phi = Phi + Fac * Del4
134 C 6th order correction [i+2 i+1 i i-1 i-2 i-3]
135 Fac = Fac * ( cfl + 2. )/6.
136 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 C 7th order correction [i+2 i+1 i i-1 i-2 i-3 i-4]
143 Fac = Fac * ( cfl + 2. )/7.
144 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
152 DelIp = ( Qip - Qi ) * MskI
153 Phi = sign(1. _d 0,Phi)*sign(1. _d 0,DelIp)
154 & *abs(Phi+Eps)/abs(DelIp+Eps)
155
156 DelI = ( Qi - Qim ) * MskIm
157 rp1h =sign(1. _d 0,DelI)*sign(1. _d 0,DelIp)
158 & *abs(DelI+Eps)/abs(DelIp+Eps)
159 rp1h_cfl = rp1h/(cfl+Eps)
160
161 C TVD limiter
162 ! Phi = max(0. _d 0, min( 2./(1-cfl), Phi, 2.*rp1h_cfl ) )
163
164 C MP limiter
165 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 A = 4.*d2 - d2p1
169 B = 4.*d2p1 - d2
170 C = d2
171 D = d2p1
172 dp1h = max(min(A,B,C,D),0. _d 0)+min(max(A,B,C,D),0. _d 0)
173 A = 4.*d2m1 - d2
174 B = 4.*d2 - d2m1
175 C = d2m1
176 D = d2
177 dm1h = max(min(A,B,C,D),0. _d 0)+min(max(A,B,C,D),0. _d 0)
178 !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 PhiLC = 2.*rp1h_cfl*(qLC-Qi+Eps)/(qUL-Qi+Eps)
184 PhiMin = max(min(0. _d 0,PhiMD),
185 & min(0. _d 0,2.*rp1h_cfl,PhiLC))
186 PhiMax = min(max(2. _d 0/(1.-cfl),PhiMD),
187 & max(0. _d 0,2.*rp1h_cfl,PhiLC))
188 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