80 |
INTEGER exchWidthX |
INTEGER exchWidthX |
81 |
INTEGER exchWidthY |
INTEGER exchWidthY |
82 |
INTEGER myNz |
INTEGER myNz |
83 |
_RL topLevFac |
_RL topLevTerm |
84 |
|
|
85 |
C-- Initialise inverter |
C-- Initialise inverter |
86 |
eta_qrNM1 = 1. D0 |
eta_qrNM1 = 1. D0 |
119 |
_EXCH_XYZ_R8( cg3d_b, myThid ) |
_EXCH_XYZ_R8( cg3d_b, myThid ) |
120 |
_EXCH_XYZ_R8( cg3d_x, myThid ) |
_EXCH_XYZ_R8( cg3d_x, myThid ) |
121 |
|
|
122 |
#ifdef NONO |
C-- Initial residual calculation (with free-Surface term) |
|
CcnhDebugStarts |
|
|
C-- Initial residual calculation |
|
|
err = 0. _d 0 |
|
|
sumRHS = 0. _d 0 |
|
|
DO bj=myByLo(myThid),myByHi(myThid) |
|
|
DO bi=myBxLo(myThid),myBxHi(myThid) |
|
|
DO J=1,sNy |
|
|
DO I=1,sNx |
|
|
alpha = 0. _d 0 |
|
|
DO K=1,Nr |
|
|
KM1 = K-1 |
|
|
IF ( KM1 .EQ. 0 ) KM1 = 1 |
|
|
KP1 = K+1 |
|
|
IF ( KP1 .EQ. Nr+1 ) KP1 = 1 |
|
|
cg3d_r(I,J,K,bi,bj) = cg3d_b(I,J,K,bi,bj) -( 0. |
|
|
& +aW3d(I ,J ,K ,bi,bj)*cg3d_x(I-1,J ,K ,bi,bj) |
|
|
& +aW3d(I+1,J ,K ,bi,bj)*cg3d_x(I+1,J ,K ,bi,bj) |
|
|
& +aS3d(I ,J ,K ,bi,bj)*cg3d_x(I ,J-1,K ,bi,bj) |
|
|
& +aS3d(I ,J+1,K ,bi,bj)*cg3d_x(I ,J+1,K ,bi,bj) |
|
|
& +aV3d(I ,J ,K ,bi,bj)*cg3d_x(I ,J ,KM1,bi,bj) |
|
|
& +aV3d(I ,J ,KP1,bi,bj)*cg3d_x(I ,J ,KP1,bi,bj) |
|
|
& -aW3d(I ,J ,K ,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
|
|
& -aW3d(I+1,J ,K ,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
|
|
& -aS3d(I ,J ,K ,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
|
|
& -aS3d(I ,J+1,K ,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
|
|
& -aV3d(I ,J ,K ,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
|
|
& -aV3d(I ,J ,KP1,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
|
|
& ) |
|
|
alpha = alpha |
|
|
& +cg3d_r(I,J,K,bi,bj) |
|
|
sumRHS = sumRHS |
|
|
& +cg3d_b(I,J,K,bi,bj) |
|
|
ENDDO |
|
|
err = err + alpha*alpha |
|
|
ENDDO |
|
|
ENDDO |
|
|
ENDDO |
|
|
ENDDO |
|
|
WRITE(6,*) 'DEBUG mythid, err = ', mythid, SQRT(err) |
|
|
_GLOBAL_SUM_R8( err , myThid ) |
|
|
_GLOBAL_SUM_R8( sumRHS , myThid ) |
|
|
_BEGIN_MASTER( myThid ) |
|
|
write(0,*) 'DEBUG cg3d: Sum(rhs) = ',sumRHS |
|
|
_END_MASTER( ) |
|
|
actualIts = 0 |
|
|
actualResidual = SQRT(err) |
|
|
C _BARRIER |
|
|
_BEGIN_MASTER( myThid ) |
|
|
WRITE(0,'(A,I6,1PE30.14)') 'DEBUG CG3D iters, err = ', |
|
|
& actualIts, actualResidual |
|
|
_END_MASTER( ) |
|
|
CcnhDebugEnds |
|
|
#endif |
|
|
|
|
|
C-- Initial residual calculation |
|
123 |
err = 0. _d 0 |
err = 0. _d 0 |
124 |
sumRHS = 0. _d 0 |
sumRHS = 0. _d 0 |
125 |
DO bj=myByLo(myThid),myByHi(myThid) |
DO bj=myByLo(myThid),myByHi(myThid) |
129 |
IF ( K .EQ. 1 ) KM1 = 1 |
IF ( K .EQ. 1 ) KM1 = 1 |
130 |
KP1 = K+1 |
KP1 = K+1 |
131 |
IF ( K .EQ. Nr ) KP1 = 1 |
IF ( K .EQ. Nr ) KP1 = 1 |
132 |
topLevFac = 0. |
topLevTerm = 0. |
133 |
IF ( K .EQ. 1) topLevFac = 1. |
IF ( K .EQ. 1) topLevTerm = freeSurfFac*cg3dNorm* |
134 |
|
& (horiVertRatio/gravity)/deltaTMom/deltaTMom |
135 |
DO J=1,sNy |
DO J=1,sNy |
136 |
DO I=1,sNx |
DO I=1,sNx |
137 |
cg3d_s(I,J,K,bi,bj) = 0. |
cg3d_s(I,J,K,bi,bj) = 0. |
148 |
& -aS3d(I ,J+1,K ,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
& -aS3d(I ,J+1,K ,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
149 |
& -aV3d(I ,J ,K ,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
& -aV3d(I ,J ,K ,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
150 |
& -aV3d(I ,J ,KP1,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
& -aV3d(I ,J ,KP1,bi,bj)*cg3d_x(I ,J ,K ,bi,bj) |
151 |
& -freeSurfFac*_rA(i,j,bi,bj)* horiVertRatio* |
& -topLevTerm*_rA(I,J,bi,bj)*cg3d_x(I,J,K,bi,bj) |
|
& cg3d_x(I ,J ,K,bi,bj)/deltaTMom/deltaTMom*cg3dNorm |
|
|
& *topLevFac |
|
152 |
& ) |
& ) |
153 |
err = err |
err = err |
154 |
& +cg3d_r(I,J,K,bi,bj)*cg3d_r(I,J,K,bi,bj) |
& +cg3d_r(I,J,K,bi,bj)*cg3d_r(I,J,K,bi,bj) |
311 |
C== Evaluate laplace operator on conjugate gradient vector |
C== Evaluate laplace operator on conjugate gradient vector |
312 |
C== q = A.s |
C== q = A.s |
313 |
alpha = 0. _d 0 |
alpha = 0. _d 0 |
314 |
|
topLevTerm = freeSurfFac*cg3dNorm* |
315 |
|
& (horiVertRatio/gravity)/deltaTMom/deltaTMom |
316 |
DO bj=myByLo(myThid),myByHi(myThid) |
DO bj=myByLo(myThid),myByHi(myThid) |
317 |
DO bi=myBxLo(myThid),myBxHi(myThid) |
DO bi=myBxLo(myThid),myBxHi(myThid) |
318 |
IF ( Nr .GT. 1 ) THEN |
IF ( Nr .GT. 1 ) THEN |
330 |
& -aS3d(I ,J ,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
& -aS3d(I ,J ,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
331 |
& -aS3d(I ,J+1,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
& -aS3d(I ,J+1,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
332 |
& -aV3d(I ,J ,K+1,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
& -aV3d(I ,J ,K+1,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
333 |
& -freeSurfFac*_rA(i,j,bi,bj)* horiVertRatio* |
& -topLevTerm*_rA(I,J,bi,bj)*cg3d_s(I,J,K,bi,bj) |
|
& cg3d_s(I ,J ,K,bi,bj)/deltaTMom/deltaTMom*cg3dNorm |
|
334 |
alpha = alpha+cg3d_s(I,J,K,bi,bj)*cg3d_q(I,J,K,bi,bj) |
alpha = alpha+cg3d_s(I,J,K,bi,bj)*cg3d_q(I,J,K,bi,bj) |
335 |
ENDDO |
ENDDO |
336 |
ENDDO |
ENDDO |
348 |
& -aW3d(I+1,J ,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
& -aW3d(I+1,J ,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
349 |
& -aS3d(I ,J ,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
& -aS3d(I ,J ,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
350 |
& -aS3d(I ,J+1,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
& -aS3d(I ,J+1,K ,bi,bj)*cg3d_s(I ,J ,K ,bi,bj) |
351 |
& -freeSurfFac*_rA(i,j,bi,bj)* horiVertRatio* |
& -topLevTerm*_rA(I,J,bi,bj)*cg3d_s(I,J,K,bi,bj) |
|
& cg3d_s(I ,J ,K,bi,bj)/deltaTMom/deltaTMom*cg3dNorm |
|
352 |
alpha = alpha+cg3d_s(I,J,K,bi,bj)*cg3d_q(I,J,K,bi,bj) |
alpha = alpha+cg3d_s(I,J,K,bi,bj)*cg3d_q(I,J,K,bi,bj) |
353 |
ENDDO |
ENDDO |
354 |
ENDDO |
ENDDO |
464 |
& actualIts, actualResidual |
& actualIts, actualResidual |
465 |
_END_MASTER( ) |
_END_MASTER( ) |
466 |
|
|
|
CcnhDebugStarts |
|
|
C CALL PLOT_FIELD_XYRL( cg2d_x, 'CALC_MOM_RHS CG2D_X' , 1, myThid ) |
|
|
C err = 0. _d 0 |
|
|
C DO bj=myByLo(myThid),myByHi(myThid) |
|
|
C DO bi=myBxLo(myThid),myBxHi(myThid) |
|
|
C DO J=1,sNy |
|
|
C DO I=1,sNx |
|
|
C cg2d_r(I,J,bi,bj) = cg2d_b(I,J,bi,bj) - |
|
|
C & (aW2d(I ,J ,bi,bj)*cg2d_x(I-1,J ,bi,bj) |
|
|
C & +aW2d(I+1,J ,bi,bj)*cg2d_x(I+1,J ,bi,bj) |
|
|
C & +aS2d(I ,J ,bi,bj)*cg2d_x(I ,J-1,bi,bj) |
|
|
C & +aS2d(I ,J+1,bi,bj)*cg2d_x(I ,J+1,bi,bj) |
|
|
C & -aW2d(I ,J ,bi,bj)*cg2d_x(I ,J ,bi,bj) |
|
|
C & -aW2d(I+1,J ,bi,bj)*cg2d_x(I ,J ,bi,bj) |
|
|
C & -aS2d(I ,J ,bi,bj)*cg2d_x(I ,J ,bi,bj) |
|
|
C & -aS2d(I ,J+1,bi,bj)*cg2d_x(I ,J ,bi,bj)) |
|
|
C err = err + |
|
|
C & cg2d_r(I,J,bi,bj)*cg2d_r(I,J,bi,bj) |
|
|
C ENDDO |
|
|
C ENDDO |
|
|
C ENDDO |
|
|
C ENDDO |
|
|
C _GLOBAL_SUM_R8( err , myThid ) |
|
|
C write(0,*) 'cg2d: Ax - b = ',SQRT(err) |
|
|
CcnhDebugEnds |
|
|
|
|
467 |
#endif /* ALLOW_NONHYDROSTATIC */ |
#endif /* ALLOW_NONHYDROSTATIC */ |
468 |
|
|
469 |
RETURN |
RETURN |