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

Diff of /MITgcm_contrib/dgoldberg/streamice/streamice_vel_solve.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.3 by dgoldberg, Thu Jul 19 18:46:56 2012 UTC
# Line 21  C     === Global variables === Line 21  C     === Global variables ===
21  #include "PARAMS.h"  #include "PARAMS.h"
22  #include "STREAMICE.h"  #include "STREAMICE.h"
23  #include "STREAMICE_CG.h"  #include "STREAMICE_CG.h"
24    #ifdef ALLOW_AUTODIFF_TAMC
25    # include "tamc.h"
26    #endif
27    
28  C     !INPUT/OUTPUT ARGUMENTS  C     !INPUT/OUTPUT ARGUMENTS
29        INTEGER myThid        INTEGER myThid
# Line 39  C     LOCAL VARIABLES Line 42  C     LOCAL VARIABLES
42    
43        INTEGER conv_flag, i, j, k, l, iter, cg_iters, bi, bj        INTEGER conv_flag, i, j, k, l, iter, cg_iters, bi, bj
44        INTEGER iter_numconv        INTEGER iter_numconv
45          INTEGER ikey_nl
46        _RL err_max, err_tempu, err_tempv, err_init, area        _RL err_max, err_tempu, err_tempv, err_init, area
47        _RL max_vel, tempu, tempv, err_lastchange, cgtol        _RL max_vel, tempu, tempv, err_lastchange, cgtol
48        CHARACTER*(MAX_LEN_MBUF) msgBuf        CHARACTER*(MAX_LEN_MBUF) msgBuf
# Line 67  C     LOCAL VARIABLES Line 71  C     LOCAL VARIABLES
71         ENDDO         ENDDO
72        ENDDO        ENDDO
73    
74    #ifdef ALLOW_AUTODIFF_TAMC
75    !$TAF STORE U_streamice = comlev1, key=ikey_dynamics
76    !$TAF STORE V_streamice = comlev1, key=ikey_dynamics
77    #endif
78    
79        CALL STREAMICE_VISC_BETA ( myThid )        CALL STREAMICE_VISC_BETA ( myThid )
80    
81        _EXCH_XY_RL ( tau_beta_eff_streamice , myThid )        _EXCH_XY_RL ( tau_beta_eff_streamice , myThid )
# Line 85  C     LOCAL VARIABLES Line 94  C     LOCAL VARIABLES
94         ENDDO         ENDDO
95        ENDDO        ENDDO
96    
97    
98        CALL STREAMICE_CG_BOUND_VALS( myThid,            CALL STREAMICE_CG_BOUND_VALS( myThid,    
99       O    ubd_SI,       O    ubd_SI,
100       O    vbd_SI)       O    vbd_SI)
101    
102    #ifdef ALLOW_AUTODIFF_TAMC
103    !$TAF STORE U_streamice = comlev1, key=ikey_dynamics
104    !$TAF STORE V_streamice = comlev1, key=ikey_dynamics
105    #endif
106    
107        CALL STREAMICE_CG_ACTION( myThid,        CALL STREAMICE_CG_ACTION( myThid,
108       O    Au_SI,       O    Au_SI,
109       O    Av_SI,       O    Av_SI,
# Line 97  C     LOCAL VARIABLES Line 112  C     LOCAL VARIABLES
112       I    0, sNx+1, 0, sNy+1 )       I    0, sNx+1, 0, sNy+1 )
113    
114        err_init = 0. _d 0        err_init = 0. _d 0
       err_tempu = 0. _d 0  
       err_tempv = 0. _d 0  
115    
116        DO bj = myByLo(myThid), myByHi(myThid)        DO bj = myByLo(myThid), myByHi(myThid)
117         DO bi = myBxLo(myThid), myBxHi(myThid)         DO bi = myBxLo(myThid), myBxHi(myThid)
118    #ifdef ALLOW_AUTODIFF_TAMC
119    !$TAF STORE err_init = comlev1, key=ikey_dynamics
120    #endif
121          DO j=1,sNy          DO j=1,sNy
122           DO i=1,sNx           DO i=1,sNx
123              err_tempu = 0. _d 0
124              err_tempv = 0. _d 0
125            IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN            IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN
126             err_tempu =             err_tempu =
127       &      ABS (Au_SI(i,j,bi,bj)+ubd_SI(i,j,bi,bj) -       &      ABS (Au_SI(i,j,bi,bj)+ubd_SI(i,j,bi,bj) -
# Line 119  C     LOCAL VARIABLES Line 137  C     LOCAL VARIABLES
137          ENDDO          ENDDO
138         ENDDO         ENDDO
139        ENDDO        ENDDO
140    #ifdef ALLOW_AUTODIFF_TAMC
141    !$TAF STORE err_init = comlev1, key=ikey_dynamics
142    #endif
143    
144        CALL GLOBAL_MAX_R8 (err_init, myThid)        CALL GLOBAL_MAX_R8 (err_init, myThid)
145    
# Line 126  C     LOCAL VARIABLES Line 147  C     LOCAL VARIABLES
147        err_max = err_init        err_max = err_init
148        err_lastchange = err_init        err_lastchange = err_init
149    
150    C START NL ITER. LOOP
151    C-------------------------------------------------------------------
152    
153        DO iter=1,streamice_max_nl_iter        DO iter=1,streamice_max_nl_iter
154    
155  C     To avoid using "exit", loop goes through all iterations  C     To avoid using "exit", loop goes through all iterations
156  C       but after convergence loop does nothing  C       but after convergence loop does nothing
157    
158    #ifdef ALLOW_AUTODIFF_TAMC
159             ikey_nl = (ikey_dynamics-1)*streamice_max_nl + iter
160    #endif
161    #ifdef ALLOW_AUTODIFF_TAMC
162    !$TAF STORE cgtol = comlev1_stream_nl, key=ikey_nl
163    !$TAF STORE err_max = comlev1_stream_nl, key=ikey_nl
164    !$TAF STORE err_tempu = comlev1_stream_nl, key=ikey_nl
165    !$TAF STORE err_tempv = comlev1_stream_nl, key=ikey_nl
166    !$TAF STORE err_lastchange = comlev1_stream_nl, key=ikey_nl
167    !$TAF STORE ru_old_si = comlev1_stream_nl, key=ikey_nl
168    !$TAF STORE rv_old_si = comlev1_stream_nl, key=ikey_nl
169    !$TAF STORE streamice_cg_a1 = comlev1_stream_nl, key=ikey_nl
170    !$TAF STORE streamice_cg_a2 = comlev1_stream_nl, key=ikey_nl
171    !$TAF STORE streamice_cg_a3 = comlev1_stream_nl, key=ikey_nl
172    !$TAF STORE streamice_cg_a4 = comlev1_stream_nl, key=ikey_nl
173    !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
174    !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
175    !$TAF STORE tau_beta_eff_streamice = comlev1_stream_nl, key=ikey_nl
176    !$TAF STORE visc_streamice = comlev1_stream_nl, key=ikey_nl
177    !$TAF STORE zu_old_si = comlev1_stream_nl, key=ikey_nl
178    !$TAF STORE zv_old_si = comlev1_stream_nl, key=ikey_nl
179    #endif
180    
181         IF (err_max .GT. streamice_nonlin_tol * err_init) THEN         IF (err_max .GT. streamice_nonlin_tol * err_init) THEN
182    
183         iter_numconv = iter_numconv + 1         iter_numconv = iter_numconv + 1
184    
185         CALL STREAMICE_CG_SOLVE(  #ifdef ALLOW_AUTODIFF_TAMC
186       &  U_streamice,  !$TAF STORE cgtol = comlev1_stream_nl, key=ikey_nl
187       &  V_streamice,  !$TAF STORE err_max = comlev1_stream_nl, key=ikey_nl
188       &  taudx_SI,  !$TAF STORE err_tempu = comlev1_stream_nl, key=ikey_nl
189       &  taudy_SI,  !$TAF STORE err_tempv = comlev1_stream_nl, key=ikey_nl
190       &  myThid , cgtol , cg_iters)  !$TAF STORE err_lastchange = comlev1_stream_nl, key=ikey_nl
191    !$TAF STORE ru_old_si = comlev1_stream_nl, key=ikey_nl
192    !$TAF STORE rv_old_si = comlev1_stream_nl, key=ikey_nl
193    !$TAF STORE streamice_cg_a1 = comlev1_stream_nl, key=ikey_nl
194    !$TAF STORE streamice_cg_a2 = comlev1_stream_nl, key=ikey_nl
195    !$TAF STORE streamice_cg_a3 = comlev1_stream_nl, key=ikey_nl
196    !$TAF STORE streamice_cg_a4 = comlev1_stream_nl, key=ikey_nl
197    !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
198    !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
199    !$TAF STORE tau_beta_eff_streamice = comlev1_stream_nl, key=ikey_nl
200    !$TAF STORE visc_streamice = comlev1_stream_nl, key=ikey_nl
201    !$TAF STORE zu_old_si = comlev1_stream_nl, key=ikey_nl
202    !$TAF STORE zv_old_si = comlev1_stream_nl, key=ikey_nl
203    #endif
204    
205           DO bj = myByLo(myThid), myByHi(myThid)
206            DO bi = myBxLo(myThid), myBxHi(myThid)
207             DO j=1,sNy
208              DO i=1,sNx
209               U_streamice (i,j,bi,bj) = 0. _d 0
210               V_streamice (i,j,bi,bj) = 0. _d 0
211              ENDDO
212             ENDDO
213            ENDDO
214           ENDDO
215    
216           CALL STREAMICE_CG_WRAPPER(
217         &       U_streamice,
218         &       V_streamice,
219         &       taudx_SI,
220         &       taudy_SI,
221         &       cgtol,
222         &       cg_iters,
223         &       myThid )
224    
225                
226    
# Line 151  C       but after convergence loop does Line 231  C       but after convergence loop does
231          CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,          CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,
232       &                    SQUEEZE_RIGHT , 1)       &                    SQUEEZE_RIGHT , 1)
233    
234    #ifdef ALLOW_AUTODIFF_TAMC
235    !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
236    !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
237    #endif
238    
239         CALL STREAMICE_VISC_BETA ( myThid )         CALL STREAMICE_VISC_BETA ( myThid )
240    
241    #ifdef ALLOW_AUTODIFF_TAMC
242    !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
243    !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
244    #endif
245    
246         _EXCH_XY_RL ( tau_beta_eff_streamice , myThid )         _EXCH_XY_RL ( tau_beta_eff_streamice , myThid )
247         _EXCH_XY_RL ( visc_streamice , myThid )         _EXCH_XY_RL ( visc_streamice , myThid )
248    
# Line 173  C       but after convergence loop does Line 263  C       but after convergence loop does
263       O    ubd_SI,       O    ubd_SI,
264       O    vbd_SI)       O    vbd_SI)
265    
266    #ifdef ALLOW_AUTODIFF_TAMC
267    !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
268    !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
269    #endif
270    
271         CALL STREAMICE_CG_ACTION( myThid,         CALL STREAMICE_CG_ACTION( myThid,
272       O    Au_SI,       O    Au_SI,
273       O    Av_SI,       O    Av_SI,
# Line 180  C       but after convergence loop does Line 275  C       but after convergence loop does
275       I    V_streamice,       I    V_streamice,
276       I    0, sNx+1, 0, sNy+1 )       I    0, sNx+1, 0, sNy+1 )
277    
278    #ifdef ALLOW_AUTODIFF_TAMC
279    !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
280    !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
281    #endif
282    
283         err_max = 0. _d 0         err_max = 0. _d 0
        err_tempu = 0. _d 0  
        err_tempv = 0. _d 0  
284    
285    #ifdef ALLOW_AUTODIFF_TAMC
286    !$TAF STORE err_max = comlev1_stream_nl, key=ikey_nl
287    #endif
288         DO bj = myByLo(myThid), myByHi(myThid)         DO bj = myByLo(myThid), myByHi(myThid)
289          DO bi = myBxLo(myThid), myBxHi(myThid)          DO bi = myBxLo(myThid), myBxHi(myThid)
290    #ifdef ALLOW_AUTODIFF_TAMC
291    !$TAF STORE err_max = comlev1_stream_nl, key=ikey_nl
292    #endif
293           DO j=1,sNy           DO j=1,sNy
294            DO i=1,sNx            DO i=1,sNx
295               err_tempu = 0. _d 0
296               err_tempv = 0. _d 0
297             IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN             IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN
298              err_tempu =              err_tempu =
299       &       ABS (Au_SI(i,j,bi,bj)+ubd_SI(i,j,bi,bj) -       &       ABS (Au_SI(i,j,bi,bj)+ubd_SI(i,j,bi,bj) -
# Line 224  C       but after convergence loop does Line 330  C       but after convergence loop does
330         ENDIF ! (err_max .GT. streamice_nonlin_tol * err_init)         ENDIF ! (err_max .GT. streamice_nonlin_tol * err_init)
331        ENDDO        ENDDO
332    
333    C END NL ITER. LOOP
334    C-------------------------------------------------------------------
335    
336        if (iter_numconv .lt. streamice_max_nl_iter) then        if (iter_numconv .lt. streamice_max_nl_iter) then
337         PRINT *, "VELOCITY SOLVE CONVERGED, ", iter_numconv,         PRINT *, "VELOCITY SOLVE CONVERGED, ", iter_numconv,
338       & " iterations"       & " iterations"
# Line 232  C       but after convergence loop does Line 341  C       but after convergence loop does
341       & iter, " iterations"       & iter, " iterations"
342        endif        endif
343    
344    !        DO bj = myByLo(myThid), myByHi(myThid)
345    !         DO bi = myBxLo(myThid), myBxHi(myThid)
346    !          DO j=1,sNy
347    !           DO i=1,sNx
348    !            U_streamice (i,j,bi,bj) = 0. _d 0
349    !            V_streamice (i,j,bi,bj) = 0. _d 0
350    !           ENDDO
351    !          ENDDO
352    !         ENDDO
353    !        ENDDO
354    !
355    !        CALL STREAMICE_CG_WRAPPER(
356    !      &       U_streamice,
357    !      &       V_streamice,
358    !      &       taudx_SI,
359    !      &       taudy_SI,
360    !      &       cgtol,
361    !      &       cg_iters,
362    !      &       myThid )
363    
364    
365        _EXCH_XY_RL (U_streamice, myThid)        _EXCH_XY_RL (U_streamice, myThid)
366        _EXCH_XY_RL (V_streamice, myThid)        _EXCH_XY_RL (V_streamice, myThid)
367    
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.3
  ViewVC Help
Powered by ViewVC 1.1.22