forked from darwinproject/darwin3
-
Notifications
You must be signed in to change notification settings - Fork 1
/
mom_vi_u_vertshear.F
136 lines (119 loc) · 4.29 KB
/
mom_vi_u_vertshear.F
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
#include "MOM_VECINV_OPTIONS.h"
SUBROUTINE MOM_VI_U_VERTSHEAR(
I bi,bj,K,
I uFld,wFld,
U uShearTerm,
I myThid)
IMPLICIT NONE
C *==========================================================*
C | S/R MOM_U_VERTSHEAR
C *==========================================================*
C *==========================================================*
C == Global variables ==
#include "SIZE.h"
#include "EEPARAMS.h"
#include "GRID.h"
#include "PARAMS.h"
C == Routine arguments ==
INTEGER bi,bj,K
_RL uFld(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
_RL wFld(1-OLx:sNx+OLx,1-OLy:sNy+OLy,Nr,nSx,nSy)
_RL uShearTerm(1-OLx:sNx+OLx,1-OLy:sNy+OLy)
INTEGER myThid
C == Local variables ==
INTEGER I,J,Kp1,Km1
_RL mask_Kp1,mask_Km1,wBarXm,wBarXp
_RL uZm,uZp
LOGICAL rAdvAreaWeight
c _RL umask_Kp1,umask_K,umask_Km1
c LOGICAL freeslipK,noslipK
c PARAMETER(freeslipK=.TRUE.)
c PARAMETER(noslipK=.NOT.freeslipK)
c LOGICAL freeslip1,noslip1
c PARAMETER(freeslip1=.TRUE.)
c PARAMETER(noslip1=.NOT.freeslip1)
c1 _RL wBarXZ,uZbarZ
rAdvAreaWeight =.TRUE.
C- Area-weighted average either in KE or in vert. advection:
IF ( selectKEscheme.EQ.1 .OR. selectKEscheme.EQ.3 )
& rAdvAreaWeight =.FALSE.
Kp1=min(K+1,Nr)
mask_Kp1=1.
IF (K.EQ.Nr) mask_Kp1=0.
Km1=max(K-1,1)
mask_Km1=1.
IF (K.EQ.1) mask_Km1=0.
DO J=1-OLy,sNy+OLy
DO I=2-OLx,sNx+OLx
c umask_K=_maskW(i,j,k,bi,bj)
C barZ( barX( W ) )
c wBarXm=0.5*(wFld(I,J,K,bi,bj)+wFld(I-1,J,K,bi,bj))
c wBarXp=0.5*(wFld(I,J,Kp1,bi,bj)+wFld(I-1,J,Kp1,bi,bj))
c & *mask_Kp1
IF ( rAdvAreaWeight ) THEN
C Transport at interface k : Area weighted average
wBarXm=0.5*(
& wFld(I,J,K,bi,bj)*rA(i,j,bi,bj)*maskC(I,J,Km1,bi,bj)
& +wFld(I-1,J,K,bi,bj)*rA(i-1,j,bi,bj)*maskC(I-1,J,Km1,bi,bj)
& )*mask_Km1*deepFac2F(K)*rhoFacF(K)
& *recip_rAw(i,j,bi,bj)
C Transport at interface k+1 (here wFld is already masked)
wBarXp=0.5*(
& wFld(I,J,Kp1,bi,bj)*rA(i,j,bi,bj)
& +wFld(I-1,J,Kp1,bi,bj)*rA(i-1,j,bi,bj)
& )*mask_Kp1*deepFac2F(Kp1)*rhoFacF(Kp1)
& *recip_rAw(i,j,bi,bj)
ELSE
C Transport at interface k : simple average
wBarXm=0.5*(
& wFld(I,J,K,bi,bj)*maskC(I,J,Km1,bi,bj)
& +wFld(I-1,J,K,bi,bj)*maskC(I-1,J,Km1,bi,bj)
& )*mask_Km1*deepFac2F(K)*rhoFacF(K)
C Transport at interface k+1 (here wFld is already masked)
wBarXp=0.5*(
& wFld(I,J,Kp1,bi,bj)
& +wFld(I-1,J,Kp1,bi,bj)
& )*mask_Kp1*deepFac2F(Kp1)*rhoFacF(Kp1)
ENDIF
C delta_Z( U ) @ interface k
c umask_Km1=mask_Km1*maskW(i,j,Km1,bi,bj)
uZm=(uFld(I,J,K,bi,bj)-mask_Km1*uFld(I,J,Km1,bi,bj))*rkSign
c2 & *recip_dRC(K)
c IF (freeslip1) uZm=uZm*umask_Km1
c IF (noslip1.AND.umask_Km1.EQ.0.) uZm=uZm*2.
C delta_Z( U ) @ interface k+1
c umask_Kp1=mask_Kp1*maskW(i,j,Kp1,bi,bj)
uZp=(mask_Kp1*uFld(I,J,Kp1,bi,bj)-uFld(I,J,K,bi,bj))*rkSign
c2 & *recip_dRC(Kp1)
c IF (freeslipK) uZp=uZp*umask_Kp1
c IF (noslipK.AND.umask_Kp1.EQ.0.) uZp=uZp*2.
c1 IF (upwindShear) THEN
c1 wBarXZ=0.5*( wBarXm + wBarXp )
c1 IF (wBarXZ.GT.0.) THEN
c1 uZbarZ=uZp
c1 ELSE
c1 uZbarZ=uZm
c1 ENDIF
c1 ELSE
c1 uZbarZ=0.5*(uZm+uZp)
c1 ENDIF
c1 uShearTerm(I,J)=-wBarXZ*uZbarZ*_maskW(I,J,K,bi,bj)
c2 uShearTerm(I,J)=-0.5*(wBarXp*uZp+wBarXm*uZm)
c2 & *_maskW(I,J,K,bi,bj)
IF (upwindShear) THEN
uShearTerm(I,J)=-0.5*
& ( (wBarXp*uZp+wBarXm*uZm)
& +(ABS(wBarXp)*uZp-ABS(wBarXm)*uZm)
& )*_recip_hFacW(i,j,k,bi,bj)
& * recip_drF(K)
& * recip_deepFac2C(K)*recip_rhoFacC(K)
ELSE
uShearTerm(I,J)=-0.5*(wBarXp*uZp+wBarXm*uZm)
& *_recip_hFacW(i,j,k,bi,bj)
& * recip_drF(K)
& * recip_deepFac2C(K)*recip_rhoFacC(K)
ENDIF
ENDDO
ENDDO
RETURN
END