/[MITgcm]/MITgcm/model/src/ini_cg2d.F
ViewVC logotype

Diff of /MITgcm/model/src/ini_cg2d.F

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

revision 1.14 by cnh, Mon Jun 15 05:13:56 1998 UTC revision 1.41 by jmc, Mon Jan 13 01:31:22 2003 UTC
# Line 1  Line 1 
1  C $Header$  C $Header$
2    C $Name$
3    
4  #include "CPP_OPTIONS.h"  #include "CPP_OPTIONS.h"
5    
6  CStartOfInterface  CBOP
7    C     !ROUTINE: INI_CG2D
8    C     !INTERFACE:
9        SUBROUTINE INI_CG2D( myThid )        SUBROUTINE INI_CG2D( myThid )
 C     /==========================================================\  
 C     | SUBROUTINE INI_CG2D                                      |  
 C     | o Initialise 2d conjugate gradient solver operators.     |  
 C     |==========================================================|  
 C     | These arrays are purely a function of the basin geom.    |  
 C     | We set then here once and them use then repeatedly.      |  
 C     \==========================================================/  
10    
11    C     !DESCRIPTION: \bv
12    C     *==========================================================*
13    C     | SUBROUTINE INI_CG2D                                      
14    C     | o Initialise 2d conjugate gradient solver operators.      
15    C     *==========================================================*
16    C     | These arrays are purely a function of the basin geom.    
17    C     | We set then here once and them use then repeatedly.      
18    C     *==========================================================*
19    C     \ev
20    
21    C     !USES:
22          IMPLICIT NONE
23  C     === Global variables ===  C     === Global variables ===
24  #include "SIZE.h"  #include "SIZE.h"
25  #include "EEPARAMS.h"  #include "EEPARAMS.h"
26  #include "PARAMS.h"  #include "PARAMS.h"
27  #include "GRID.h"  #include "GRID.h"
28  #include "DYNVARS.h"  c#include "DYNVARS.h"
29    #include "SURFACE.h"
30  #include "CG2D.h"  #include "CG2D.h"
31    #ifdef ALLOW_OBCS
32    #include "OBCS.h"
33    #endif
34    
35    C     !INPUT/OUTPUT PARAMETERS:
36  C     === Routine arguments ===  C     === Routine arguments ===
37  C     myThid - Thread no. that called this routine.  C     myThid - Thread no. that called this routine.
38        INTEGER myThid        INTEGER myThid
 CEndOfInterface  
39    
40    C     !LOCAL VARIABLES:
41  C     === Local variables ===  C     === Local variables ===
42  C     xG, yG - Global coordinate location.  C     xG, yG - Global coordinate location.
43  C     zG  C     zG
# Line 32  C     iG, jG - Global coordinate index Line 45  C     iG, jG - Global coordinate index
45  C     bi,bj  - Loop counters  C     bi,bj  - Loop counters
46  C     faceArea - Temporary used to hold cell face areas.  C     faceArea - Temporary used to hold cell face areas.
47  C     I,J,K  C     I,J,K
48  C     myNorm - Work variable used in clculating normalisation factor  C     myNorm - Work variable used in calculating normalisation factor
49    C     sumArea - Work variable used to compute the total Domain Area
50        CHARACTER*(MAX_LEN_MBUF) msgBuf        CHARACTER*(MAX_LEN_MBUF) msgBuf
51        INTEGER bi, bj        INTEGER bi, bj
52        INTEGER I,  J, K        INTEGER I,  J, K
53        _RL     faceArea        _RL     faceArea
54        _RL     myNorm        _RS     myNorm
55          _RL     sumArea
56        _RL     aC, aCw, aCs        _RL     aC, aCw, aCs
57    CEOP
58    
59    C--   Initialize arrays in common blocs (CG2D.h) ; not really necessary
60    C     but safer when EXCH do not fill all the overlap regions.
61          DO bj=myByLo(myThid),myByHi(myThid)
62           DO bi=myBxLo(myThid),myBxHi(myThid)
63            DO J=1-OLy,sNy+OLy
64             DO I=1-OLx,sNx+OLx
65              aW2d(I,J,bi,bj) = 0. _d 0
66              aS2d(I,J,bi,bj) = 0. _d 0
67              pW(I,J,bi,bj) = 0. _d 0
68              pS(I,J,bi,bj) = 0. _d 0
69              pC(I,J,bi,bj) = 0. _d 0
70              cg2d_q(I,J,bi,bj) = 0. _d 0
71             ENDDO
72            ENDDO
73            DO J=1-1,sNy+1
74             DO I=1-1,sNx+1
75              cg2d_r(I,J,bi,bj) = 0. _d 0
76              cg2d_s(I,J,bi,bj) = 0. _d 0
77             ENDDO
78            ENDDO
79           ENDDO
80          ENDDO
81    
82  C--   Initialise laplace operator  C--   Initialise laplace operator
83  C     aW2d: integral in Z Ax/dX  C     aW2d: integral in Z Ax/dX
# Line 52  C     aS2d: integral in Z Ay/dY Line 91  C     aS2d: integral in Z Ay/dY
91            aS2d(I,J,bi,bj) = 0. _d 0            aS2d(I,J,bi,bj) = 0. _d 0
92           ENDDO           ENDDO
93          ENDDO          ENDDO
94          DO K=1,Nz          DO K=1,Nr
95           DO J=1,sNy           DO J=1,sNy
96            DO I=1,sNx            DO I=1,sNx
97             faceArea = _dyG(I,J,bi,bj)*dzF(K)*_hFacW(I,J,K,bi,bj)             faceArea = _dyG(I,J,bi,bj)*drF(K)
98         &               *_hFacW(I,J,K,bi,bj)
99             aW2d(I,J,bi,bj) = aW2d(I,J,bi,bj) +             aW2d(I,J,bi,bj) = aW2d(I,J,bi,bj) +
100       &      gBaro*faceArea*_rdxC(I,J,bi,bj)       &      implicSurfPress*implicDiv2DFlow
101             faceArea = _dxG(I,J,bi,bj)*dzF(K)*_hFacS(I,J,K,bi,bj)       &           *faceArea*recip_dxC(I,J,bi,bj)
102               faceArea = _dxG(I,J,bi,bj)*drF(K)
103         &               *_hFacS(I,J,K,bi,bj)
104             aS2d(I,J,bi,bj) = aS2d(I,J,bi,bj) +             aS2d(I,J,bi,bj) = aS2d(I,J,bi,bj) +
105       &      gBaro*faceArea*_rdyC(I,J,bi,bj)       &      implicSurfPress*implicDiv2DFlow
106         &           *faceArea*recip_dyC(I,J,bi,bj)
107            ENDDO            ENDDO
108           ENDDO           ENDDO
109          ENDDO          ENDDO
110    #ifdef ALLOW_OBCS
111            IF (useOBCS) THEN
112             DO I=1,sNx
113              IF (OB_Jn(I,bi,bj).NE.0) aS2d(I,OB_Jn(I,bi,bj),bi,bj)=0.
114              IF (OB_Jn(I,bi,bj).NE.0) aS2d(I,OB_Jn(I,bi,bj)+1,bi,bj)=0.
115              IF (OB_Js(I,bi,bj).NE.0) aS2d(I,OB_Js(I,bi,bj)+1,bi,bj)=0.
116              IF (OB_Js(I,bi,bj).NE.0) aS2d(I,OB_Js(I,bi,bj),bi,bj)=0.
117             ENDDO
118             DO J=1,sNy
119              IF (OB_Ie(J,bi,bj).NE.0) aW2d(OB_Ie(J,bi,bj),J,bi,bj)=0.
120              IF (OB_Ie(J,bi,bj).NE.0) aW2d(OB_Ie(J,bi,bj)+1,J,bi,bj)=0.
121              IF (OB_Iw(J,bi,bj).NE.0) aW2d(OB_Iw(J,bi,bj)+1,J,bi,bj)=0.
122              IF (OB_Iw(J,bi,bj).NE.0) aW2d(OB_Iw(J,bi,bj),J,bi,bj)=0.
123             ENDDO
124            ENDIF
125    #endif
126          DO J=1,sNy          DO J=1,sNy
127           DO I=1,sNx           DO I=1,sNx
128            myNorm = MAX(ABS(aW2d(I,J,bi,bj)),myNorm)            myNorm = MAX(ABS(aW2d(I,J,bi,bj)),myNorm)
# Line 72  C     aS2d: integral in Z Ay/dY Line 131  C     aS2d: integral in Z Ay/dY
131          ENDDO          ENDDO
132         ENDDO         ENDDO
133        ENDDO        ENDDO
134        cg2dNbuf(1,myThid) = myNorm        _GLOBAL_MAX_R4( myNorm, myThid )
135        _GLOBAL_MAX_R8( cg2dNbuf, myNorm, myThid )        IF ( myNorm .NE. 0. _d 0 ) THEN
136        IF ( cg2dNbuf(1,1) .NE. 0. _d 0 ) THEN         myNorm = 1. _d 0/myNorm
        myNorm = 1. _d 0/cg2dNbuf(1,1)  
137        ELSE        ELSE
138         myNorm = 1. _d 0         myNorm = 1. _d 0
139        ENDIF        ENDIF
140         cg2dNorm = myNorm         cg2dNorm = myNorm
141        _BEGIN_MASTER( myThid )        _BEGIN_MASTER( myThid )
142  CcnhDebugStarts  CcnhDebugStarts
143         WRITE(msgBuf,'(A,F)') '// CG2D normalisation factor = ', cg2dNorm         WRITE(msgBuf,'(A,E40.25)') '// CG2D normalisation factor = ',
144         &               cg2dNorm
145         CALL PRINT_MESSAGE( msgBuf,standardMessageUnit,SQUEEZE_RIGHT,1)         CALL PRINT_MESSAGE( msgBuf,standardMessageUnit,SQUEEZE_RIGHT,1)
146         WRITE(msgBuf,*) '                               '         WRITE(msgBuf,*) '                               '
147         CALL PRINT_MESSAGE( msgBuf,standardMessageUnit,SQUEEZE_RIGHT,1)         CALL PRINT_MESSAGE( msgBuf,standardMessageUnit,SQUEEZE_RIGHT,1)
# Line 104  CcnhDebugStarts Line 163  CcnhDebugStarts
163  C     CALL PLOT_FIELD_XYRS( aW2d, 'AW2D INI_CG2D.1' , 1, myThid )  C     CALL PLOT_FIELD_XYRS( aW2d, 'AW2D INI_CG2D.1' , 1, myThid )
164  C     CALL PLOT_FIELD_XYRS( aS2d, 'AS2D INI_CG2D.1' , 1, myThid )  C     CALL PLOT_FIELD_XYRS( aS2d, 'AS2D INI_CG2D.1' , 1, myThid )
165  CcnhDebugEnds  CcnhDebugEnds
166        _EXCH_XY_R4(aW2d, myThid)  c     _EXCH_XY_R4(aW2d, myThid)
167        _EXCH_XY_R4(aS2d, myThid)  c     _EXCH_XY_R4(aS2d, myThid)
168          CALL EXCH_UV_XY_RS(aW2d,aS2d,.FALSE.,myThid)
169  CcnhDebugStarts  CcnhDebugStarts
170  C     CALL PLOT_FIELD_XYRS( aW2d, 'AW2D INI_CG2D.2' , 1, myThid )  C     CALL PLOT_FIELD_XYRS( aW2d, 'AW2D INI_CG2D.2' , 1, myThid )
171  C     CALL PLOT_FIELD_XYRS( aS2d, 'AS2D INI_CG2D.2' , 1, myThid )  C     CALL PLOT_FIELD_XYRS( aS2d, 'AS2D INI_CG2D.2' , 1, myThid )
172  CcnhDebugEnds  CcnhDebugEnds
173    
174    C-- Define the solver tolerance in the appropriate Unit :
175          cg2dNormaliseRHS = cg2dTargetResWunit.LE.0
176          IF (cg2dNormaliseRHS) THEN
177    C-  when using a normalisation of RHS, tolerance has no unit => no conversion
178            cg2dTolerance = cg2dTargetResidual
179          ELSE
180    C-  compute the total Area of the domain :
181            sumArea = 0.
182            DO bj=myByLo(myThid),myByHi(myThid)
183             DO bi=myBxLo(myThid),myBxHi(myThid)
184              DO j=1,sNy
185               DO i=1,sNx
186                 IF (Ro_surf(i,j,bi,bj).GT.R_low(i,j,bi,bj)) THEN
187                   sumArea = sumArea + rA(i,j,bi,bj)
188                 ENDIF
189               ENDDO
190              ENDDO
191             ENDDO
192            ENDDO
193    c       WRITE(*,*) ' mythid, sumArea = ', mythid, sumArea
194            _GLOBAL_SUM_R8( sumArea, myThid )
195    C-  convert Target-Residual (in W unit) to cg2d-solver residual unit [m^2/s^2]
196            cg2dTolerance = cg2dNorm * cg2dTargetResWunit
197         &                           * sumArea / deltaTmom
198            _BEGIN_MASTER( myThid )
199            WRITE(standardMessageUnit,'(2A,1P2E22.14)') ' ini_cg2d: ',
200         &          'sumArea,cg2dTolerance =', sumArea,cg2dTolerance
201            _END_MASTER( myThid )
202          ENDIF
203        
204  C--   Initialise preconditioner  C--   Initialise preconditioner
205  C     Note. 20th May 1998  C     Note. 20th May 1998
206  C           I made a weird discovery! In the model paper we argue  C           I made a weird discovery! In the model paper we argue
# Line 132  C           defaults to 0.51 but can be Line 222  C           defaults to 0.51 but can be
222            pC(I,J,bi,bj) = 1. _d 0            pC(I,J,bi,bj) = 1. _d 0
223            aC = -(            aC = -(
224       &     aW2d(I,J,bi,bj) + aW2d(I+1,J  ,bi,bj)       &     aW2d(I,J,bi,bj) + aW2d(I+1,J  ,bi,bj)
225       &    +aS2d(I,J,bi,bj) + aS2D(I  ,J+1,bi,bj)       &    +aS2d(I,J,bi,bj) + aS2d(I  ,J+1,bi,bj)
226       &    +freeSurfFac*myNorm*       &    +freeSurfFac*myNorm*recip_Bo(I,J,bi,bj)*
227       &     zA(I,J,bi,bj)/deltaTMom/deltaTMom       &     rA(I,J,bi,bj)/deltaTMom/deltaTfreesurf
228       &    )       &    )
229            aCs = -(            aCs = -(
230       &     aW2d(I,J-1,bi,bj) + aW2d(I+1,J-1,bi,bj)       &     aW2d(I,J-1,bi,bj) + aW2d(I+1,J-1,bi,bj)
231       &    +aS2d(I,J-1,bi,bj) + aS2d(I  ,J  ,bi,bj)       &    +aS2d(I,J-1,bi,bj) + aS2d(I  ,J  ,bi,bj)
232       &    +freeSurfFac*myNorm*       &    +freeSurfFac*myNorm*recip_Bo(I,J-1,bi,bj)*
233       &     zA(I,J-1,bi,bj)/deltaTMom/deltaTMom       &     rA(I,J-1,bi,bj)/deltaTMom/deltaTfreesurf
234       &    )       &    )
235            aCw = -(            aCw = -(
236       &     aW2d(I-1,J,bi,bj) + aW2d(I  ,J  ,bi,bj)       &     aW2d(I-1,J,bi,bj) + aW2d(I  ,J  ,bi,bj)
237       &    +aS2d(I-1,J,bi,bj) + aS2d(I-1,J+1,bi,bj)       &    +aS2d(I-1,J,bi,bj) + aS2d(I-1,J+1,bi,bj)
238       &    +freeSurfFac*myNorm*       &    +freeSurfFac*myNorm*recip_Bo(I-1,J,bi,bj)*
239       &     zA(I-1,J,bi,bj)/deltaTMom/deltaTMom       &     rA(I-1,J,bi,bj)/deltaTMom/deltaTfreesurf
240       &    )       &    )
241            IF ( aC .EQ. 0. ) THEN            IF ( aC .EQ. 0. ) THEN
242              pC(I,J,bi,bj) = 0. _d 0              pC(I,J,bi,bj) = 1. _d 0
243            ELSE            ELSE
244             pC(I,J,bi,bj) =  1. _d 0 / aC             pC(I,J,bi,bj) =  1. _d 0 / aC
245            ENDIF            ENDIF
# Line 174  C         pS(I,J,bi,bj) = 0. Line 264  C         pS(I,J,bi,bj) = 0.
264        ENDDO        ENDDO
265  C--   Update overlap regions  C--   Update overlap regions
266        _EXCH_XY_R4(pC, myThid)        _EXCH_XY_R4(pC, myThid)
267        _EXCH_XY_R4(pW, myThid)  c     _EXCH_XY_R4(pW, myThid)
268        _EXCH_XY_R4(pS, myThid)  c     _EXCH_XY_R4(pS, myThid)
269          CALL EXCH_UV_XY_RS(pW,pS,.FALSE.,myThid)
270  C--   Set default values for initial guess and RHS  CcnhDebugStarts
271        IF ( startTime .EQ. 0 ) THEN  C     CALL PLOT_FIELD_XYRS( pC, 'pC   INI_CG2D.2' , 1, myThid )
272         DO bj=myByLo(myThid),myByHi(myThid)  C     CALL PLOT_FIELD_XYRS( pW, 'pW   INI_CG2D.2' , 1, myThid )
273          DO bi=myBxLo(myThid),myBxHi(myThid)  C     CALL PLOT_FIELD_XYRS( pS, 'pS   INI_CG2D.2' , 1, myThid )
274           DO J=1,sNy  CcnhDebugEnds
           DO I=1,sNx  
            cg2d_x(I,J,bi,bj) = 0. _d 0  
            cg2d_b(I,J,bi,bj) = 0. _d 0  
 #ifdef ALLOW_CD  
            cg2d_xNM1(I,J,bi,bj) = 0. _d 0  
 #endif  
           ENDDO  
          ENDDO  
         ENDDO  
        ENDDO  
 C--    Update overlap regions  
        _EXCH_XY_R8(cg2d_x, myThid)  
        _EXCH_XY_R8(cg2d_b, myThid)  
 #ifdef ALLOW_CD  
        _EXCH_XY_R8(cg2d_xNM1, myThid)  
 #endif  
       ENDIF  
275    
276        RETURN        RETURN
277        END        END
Legend:
Removed from v.1.14  
changed lines
  Added in v.1.41
  ViewVC Help
Powered by ViewVC 1.1.22