/[MITgcm]/MITgcm_contrib/dgoldberg/streamice/streamice_velmask_upd.F
ViewVC logotype

Diff of /MITgcm_contrib/dgoldberg/streamice/streamice_velmask_upd.F

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

revision 1.1 by heimbach, Thu Mar 29 15:59:21 2012 UTC revision 1.5 by dgoldberg, Fri May 10 17:00:50 2013 UTC
# Line 22  C     === Global variables === Line 22  C     === Global variables ===
22  #include "EEPARAMS.h"  #include "EEPARAMS.h"
23  #include "PARAMS.h"  #include "PARAMS.h"
24  #include "STREAMICE.h"  #include "STREAMICE.h"
25    #ifdef ALLOW_USE_MPI
26    # include "EESUPPORT.h"
27    #endif
28  ! #include "STREAMICE_ADV.h"  ! #include "STREAMICE_ADV.h"
29    
30        INTEGER myThid        INTEGER myThid
# Line 30  C     === Global variables === Line 33  C     === Global variables ===
33    
34        INTEGER i, j, bi, bj, ki, kj        INTEGER i, j, bi, bj, ki, kj
35        INTEGER maskFlag        INTEGER maskFlag
36          CHARACTER*(MAX_LEN_MBUF) msgBuf
37    #ifdef ALLOW_USE_MPI
38          integer mpiRC, mpiMyWid
39    #endif
40    #ifdef ALLOW_PETSC
41          _RS DoFCount
42          integer n_dofs_proc_loc (0:nPx*nPy-1)
43          integer n_dofs_cum_sum (0:nPx*nPy-1)
44    #endif
45    
46          _EXCH_XY_RL( H_streamice, myThid )
47          _EXCH_XY_RL( area_shelf_streamice, myThid )
48          _EXCH_XY_RL( STREAMICE_hmask, myThid )
49    
50        DO bj=myByLo(myThid),myByHi(myThid)        DO bj=myByLo(myThid),myByHi(myThid)
51         DO bi=myBxLo(myThid),myBxHi(myThid)         DO bi=myBxLo(myThid),myBxHi(myThid)
# Line 62  C     === Global variables === Line 78  C     === Global variables ===
78               IF (maskFlag.EQ.3) THEN               IF (maskFlag.EQ.3) THEN
79                DO kj=0,1                DO kj=0,1
80                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 3.0                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 3.0
81                 STREAMICE_vmask(i+ki,j+kj,bi,bj) = 0.0                 STREAMICE_vmask(i+ki,j+kj,bi,bj) = 3.0
82                ENDDO                ENDDO
83                STREAMICE_ufacemask(i+ki,j,bi,bj) = 3.0                STREAMICE_ufacemask(i+ki,j,bi,bj) = 3.0
84               ELSE IF (maskFlag.EQ.2) THEN               ELSE IF (maskFlag.EQ.2) THEN
85                DO kj=0,1                !DO kj=0,1
86                 STREAMICE_ufacemask(i+ki,j+kj,bi,bj) = 2.0                 STREAMICE_ufacemask(i+ki,j,bi,bj) = 2.0
87                ENDDO                !ENDDO
88               ELSE IF (maskFlag.EQ.4) THEN               ELSE IF (maskFlag.EQ.4) THEN
89                DO kj=0,1                DO kj=0,1
90                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 0.0                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 0.0
# Line 93  C     === Global variables === Line 109  C     === Global variables ===
109              IF (maskFlag.EQ.3) THEN              IF (maskFlag.EQ.3) THEN
110                DO ki=0,1                DO ki=0,1
111                 STREAMICE_vmask(i+ki,j+kj,bi,bj) = 3.0                 STREAMICE_vmask(i+ki,j+kj,bi,bj) = 3.0
112                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 0.0                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 3.0
113                ENDDO                ENDDO
114                STREAMICE_vfacemask(i+ki,j,bi,bj) = 3.0                STREAMICE_vfacemask(i,j+kj,bi,bj) = 3.0
115             ELSE IF (maskFlag.EQ.2) THEN             ELSE IF (maskFlag.EQ.2) THEN
116                DO ki=0,1                !DO ki=0,1
117                 STREAMICE_vfacemask(i+ki,j+kj,bi,bj) = 2.0                 STREAMICE_vfacemask(i,j+kj,bi,bj) = 2.0
118                ENDDO                !ENDDO
119             ELSE IF (maskFlag.EQ.4) THEN             ELSE IF (maskFlag.EQ.4) THEN
120                DO ki=0,1                DO ki=0,1
121                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 0.0                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 0.0
122                 STREAMICE_vmask(i+ki,j+kj,bi,bj) = 0.0                             STREAMICE_vmask(i+ki,j+kj,bi,bj) = 0.0            
123                ENDDO                ENDDO
124                STREAMICE_vfacemask(i+ki,j,bi,bj) = 4.0                STREAMICE_vfacemask(i,j+kj,bi,bj) = 4.0
125             ELSE IF (maskFlag.EQ.0) THEN             ELSE IF (maskFlag.EQ.0) THEN
126                DO ki=0,1                DO ki=0,1
127                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 0.0                 STREAMICE_umask(i+ki,j+kj,bi,bj) = 0.0
# Line 164  C     === Global variables === Line 180  C     === Global variables ===
180    
181  !       CALL WRITE_FULLARRAY_RL ("umask",STREAMICE_umask,  !       CALL WRITE_FULLARRAY_RL ("umask",STREAMICE_umask,
182  !      c   1,0,0,1,0,myThid)  !      c   1,0,0,1,0,myThid)
183  !       CALL WRITE_FLD_XY_RL ("umask2","",STREAMICE_umask,0,myThid)  !      CALL WRITE_FLD_XY_RL ("umask","",STREAMICE_umask,0,myThid)
184    !      CALL WRITE_FLD_XY_RL ("vmask","",STREAMICE_vmask,0,myThid)
185    !      CALL WRITE_FLD_XY_RL ("ufacemask","",STREAMICE_ufacemask,0,myThid)
186    !      CALL WRITE_FLD_XY_RL ("vfacemask","",STREAMICE_vfacemask,0,myThid)
187    
188    #ifdef ALLOW_PETSC
189    
190          DO bj = myByLo(myThid), myByHi(myThid)
191           DO bi = myBxLo(myThid), myBxHi(myThid)
192            DO j=1,sNy
193             DO i=1,sNx
194               streamice_petsc_dofs_u (i,j,bi,bj) = -2.0
195               streamice_petsc_dofs_v (i,j,bi,bj) = -2.0
196             ENDDO
197            ENDDO
198           ENDDO
199          ENDDO
200    
201          DoFCount = -1.0
202          DO bj = myByLo(myThid), myByHi(myThid)
203           DO bi = myBxLo(myThid), myBxHi(myThid)
204            DO j=1,sNy
205             DO i=1,sNx
206    
207    
208    C   DOFS ARE NUMBERED AS FOLLOWS ON PROCESSOR DOMAIN:
209    C    grid is stepped through in order bj, bi, j, i
210    C    1) if umask(i,j,bi,bj)==1, the counter is updated by 1;
211    C        streamice_petsc_dofs_u is assigned the counter;
212    C        o/w  streamice_petsc_dofs_u is assigned -1
213    C    2) if vmask(i,j,bi,bj)==1, the counter is updated by 1;
214    C        streamice_petsc_dofs_v is assigned the counter;
215    C        o/w  streamice_petsc_dofs_v is assigned -1
216    C    NOTE THESE NUMBERING ARRAYS ARE USED TO CONSTRUCT PETSC VECTORS AND MATRIX
217    
218              if (STREAMICE_umask (i,j,bi,bj).eq.1) THEN
219               DoFCount = DoFCount + 1.0
220               streamice_petsc_dofs_u (i,j,bi,bj) = DoFCount
221              else
222               streamice_petsc_dofs_u (i,j,bi,bj) = -1.0
223              endif
224    
225              if (STREAMICE_vmask (i,j,bi,bj).eq.1) THEN
226               DoFCount = DoFCount + 1.0
227               streamice_petsc_dofs_v (i,j,bi,bj) = DoFCount
228              else
229               streamice_petsc_dofs_v (i,j,bi,bj) = -1.0
230              endif
231    
232             ENDDO
233            ENDDO
234           ENDDO
235          ENDDO
236    
237    #ifdef ALLOW_USE_MPI
238    
239          DO i=0,nPx*nPy-1
240           n_dofs_proc_loc (i) = 0
241          ENDDO
242    
243          CALL MPI_COMM_RANK( MPI_COMM_WORLD, mpiMyWId, mpiRC )
244    
245          n_dofs_proc_loc (mpiMyWId) = INT(DoFCount)+1
246    
247          CALL MPI_Allreduce(n_dofs_proc_loc,n_dofs_process,nPx*nPy,
248         &       MPI_INTEGER, MPI_SUM,MPI_COMM_MODEL,mpiRC)
249    
250          n_dofs_cum_sum(0) = 0
251    
252          DO i=1,nPx*nPy-1
253           n_dofs_cum_sum(i) = n_dofs_cum_sum(i-1)+
254         &                     n_dofs_process(i-1)
255          ENDDO
256    
257    #else /* ALLOW_USE_MPI */
258    
259          n_dofs_process (0) = INT(DoFCount)+1
260          n_dofs_cum_sum (0) = INT(DoFCount)+1
261    
262    #endif /* ALLOW_USE_MPI */
263    
264          DO bj = myByLo(myThid), myByHi(myThid)
265           DO bi = myBxLo(myThid), myBxHi(myThid)
266            DO j=1,sNy
267             DO i=1,sNx
268              IF (streamice_petsc_dofs_u(i,j,bi,bj).ge.0 ) THEN
269               streamice_petsc_dofs_u(i,j,bi,bj) =
270         &      streamice_petsc_dofs_u(i,j,bi,bj) +
271         &      n_dofs_cum_sum(mpimywid)
272              ENDIF
273              IF (streamice_petsc_dofs_v(i,j,bi,bj).ge.0 ) THEN
274               streamice_petsc_dofs_v(i,j,bi,bj) =
275         &      streamice_petsc_dofs_v(i,j,bi,bj) +
276         &      n_dofs_cum_sum(mpimywid)
277              ENDIF
278             ENDDO
279            ENDDO
280           ENDDO
281          ENDDO
282    
283          _EXCH_XY_RS(streamice_petsc_dofs_u,myThid)
284          _EXCH_XY_RS(streamice_petsc_dofs_v,myThid)
285    
286    
287    #endif /* ALLOW_PETSC */
288    
289    
290  #endif  #endif
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.5
  ViewVC Help
Powered by ViewVC 1.1.22