/[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.2 by heimbach, Wed May 2 02:36:01 2012 UTC revision 1.12 by dgoldberg, Thu Mar 7 15:23:19 2013 UTC
# Line 43  C     LOCAL VARIABLES Line 43  C     LOCAL VARIABLES
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        INTEGER ikey_nl
46        _RL err_max, err_tempu, err_tempv, err_init, area        _RL err_max, err_tempu, err_tempv, err_init, area, err_max_fp
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
49  !       _RL taudx (1-OLx:sNx+OLx,1-OLy:sNy+Oly,nSx,nSy)  !       _RL taudx (1-OLx:sNx+OLx,1-OLy:sNy+Oly,nSx,nSy)
50  !       _RL taudy (1-OLx:sNx+OLx,1-OLy:sNy+Oly,nSx,nSy)  !       _RL taudy (1-OLx:sNx+OLx,1-OLy:sNy+Oly,nSx,nSy)
51    
52        CALL STREAMICE_DRIVING_STRESS (myThid)        IF (STREAMICE_ppm_driving_stress) THEN
53           CALL STREAMICE_DRIVING_STRESS_PPM (myThid)
54          ELSE
55           CALL STREAMICE_DRIVING_STRESS (myThid)
56          ENDIF
57    
58        cgtol = streamice_cg_tol        cgtol = streamice_cg_tol
59    
# Line 74  C     LOCAL VARIABLES Line 78  C     LOCAL VARIABLES
78  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
79  !$TAF STORE U_streamice = comlev1, key=ikey_dynamics  !$TAF STORE U_streamice = comlev1, key=ikey_dynamics
80  !$TAF STORE V_streamice = comlev1, key=ikey_dynamics  !$TAF STORE V_streamice = comlev1, key=ikey_dynamics
81    #ifdef STREAMICE_HYBRID_STRESS
82    !$TAF STORE streamice_taubx = comlev1, key=ikey_dynamics
83    !$TAF STORE streamice_tauby = comlev1, key=ikey_dynamics
84    !$TAF STORE visc_streamice_full = comlev1, key=ikey_dynamics
85    #endif
86    #endif
87    
88    #ifdef STREAMICE_HYBRID_STRESS
89           CALL STREAMICE_VISC_BETA_HYBRID ( myThid )
90    #else
91           CALL STREAMICE_VISC_BETA ( myThid )
92  #endif  #endif
93    
94        CALL STREAMICE_VISC_BETA ( myThid )  #ifdef STREAMICE_HYBRID_STRESS
95    !$TAF STORE visc_streamice_full = comlev1, key=ikey_dynamics
96    #endif
97    
98        _EXCH_XY_RL ( tau_beta_eff_streamice , myThid )        _EXCH_XY_RL ( tau_beta_eff_streamice , myThid )
99        _EXCH_XY_RL ( visc_streamice , myThid )        _EXCH_XY_RL ( visc_streamice , myThid )
# Line 94  C     LOCAL VARIABLES Line 111  C     LOCAL VARIABLES
111         ENDDO         ENDDO
112        ENDDO        ENDDO
113    
114    
115        CALL STREAMICE_CG_BOUND_VALS( myThid,            CALL STREAMICE_CG_BOUND_VALS( myThid,    
116       O    ubd_SI,       O    ubd_SI,
117       O    vbd_SI)       O    vbd_SI)
118          
119    
120    !      CALL WRITE_FLD_XY_RL("u_bound_cont","",ubd_SI,0,myThid)
121    !      CALL WRITE_FLD_XY_RL("v_bound_cont","",vbd_SI,0,myThid)
122    !      CALL WRITE_FLD_XY_RL("taudx_u","",taudx_SI,0,myThid)
123    !      CALL WRITE_FLD_XY_RL("taudx_v","",taudy_SI,0,myThid)
124    
125  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
126  !$TAF STORE U_streamice = comlev1, key=ikey_dynamics  !$TAF STORE U_streamice = comlev1, key=ikey_dynamics
# Line 110  C     LOCAL VARIABLES Line 134  C     LOCAL VARIABLES
134       I    V_streamice,       I    V_streamice,
135       I    0, sNx+1, 0, sNy+1 )       I    0, sNx+1, 0, sNy+1 )
136    
137    
138    
139        err_init = 0. _d 0        err_init = 0. _d 0
140    
141        DO bj = myByLo(myThid), myByHi(myThid)        DO bj = myByLo(myThid), myByHi(myThid)
# Line 123  C     LOCAL VARIABLES Line 149  C     LOCAL VARIABLES
149            err_tempv = 0. _d 0            err_tempv = 0. _d 0
150            IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN            IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN
151             err_tempu =             err_tempu =
152       &      ABS (Au_SI(i,j,bi,bj)+ubd_SI(i,j,bi,bj) -       &      ABS (Au_SI(i,j,bi,bj)+0*ubd_SI(i,j,bi,bj) -
153       &           taudx_SI(i,j,bi,bj))       &           taudx_SI(i,j,bi,bj))
154    !            print *, "err_temp_u", err_tempu
155            ENDIF            ENDIF
156            IF (STREAMICE_vmask(i,j,bi,bj).eq.1) THEN            IF (STREAMICE_vmask(i,j,bi,bj).eq.1) THEN
157             err_tempv = MAX( err_tempu,             err_tempv = MAX( err_tempu,
158       &      ABS (Av_SI(i,j,bi,bj)+vbd_SI(i,j,bi,bj) -       &      ABS (Av_SI(i,j,bi,bj)+0*vbd_SI(i,j,bi,bj) -
159       &           taudy_SI(i,j,bi,bj)))       &           taudy_SI(i,j,bi,bj)))
160            ENDIF            ENDIF
161            IF (err_tempv .ge. err_init) err_init = err_tempv            IF (err_tempv .ge. err_init) THEN
162                err_init = err_tempv
163              ENDIF
164           ENDDO           ENDDO
165          ENDDO          ENDDO
166         ENDDO         ENDDO
# Line 142  C     LOCAL VARIABLES Line 171  C     LOCAL VARIABLES
171    
172        CALL GLOBAL_MAX_R8 (err_init, myThid)        CALL GLOBAL_MAX_R8 (err_init, myThid)
173    
174          WRITE(msgBuf,'(A,E15.7)') 'initial nonlinear resid (error): ',
175         &                       err_init
176          CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,
177         &                    SQUEEZE_RIGHT , 1)
178    
179    
180        iter_numconv = 0        iter_numconv = 0
181        err_max = err_init        err_max = err_init
182          err_max_fp = streamice_nonlin_tol_fp * 10.
183        err_lastchange = err_init        err_lastchange = err_init
184    
185  C START NL ITER. LOOP  C START NL ITER. LOOP
# Line 160  C       but after convergence loop does Line 196  C       but after convergence loop does
196  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
197  !$TAF STORE cgtol = comlev1_stream_nl, key=ikey_nl  !$TAF STORE cgtol = comlev1_stream_nl, key=ikey_nl
198  !$TAF STORE err_max = comlev1_stream_nl, key=ikey_nl  !$TAF STORE err_max = comlev1_stream_nl, key=ikey_nl
199    !$TAF STORE err_max_fp = comlev1_stream_nl, key=ikey_nl
200  !$TAF STORE err_tempu = comlev1_stream_nl, key=ikey_nl  !$TAF STORE err_tempu = comlev1_stream_nl, key=ikey_nl
201  !$TAF STORE err_tempv = comlev1_stream_nl, key=ikey_nl  !$TAF STORE err_tempv = comlev1_stream_nl, key=ikey_nl
202  !$TAF STORE err_lastchange = comlev1_stream_nl, key=ikey_nl  !$TAF STORE err_lastchange = comlev1_stream_nl, key=ikey_nl
203  !$TAF STORE ru_old_si = comlev1_stream_nl, key=ikey_nl  !$TAF STORE ru_old_si = comlev1_stream_nl, key=ikey_nl
204  !$TAF STORE rv_old_si = comlev1_stream_nl, key=ikey_nl  !$TAF STORE rv_old_si = comlev1_stream_nl, key=ikey_nl
205    !$TAF STORE u_old_si = comlev1_stream_nl, key=ikey_nl
206    !$TAF STORE v_old_si = comlev1_stream_nl, key=ikey_nl
207  !$TAF STORE streamice_cg_a1 = comlev1_stream_nl, key=ikey_nl  !$TAF STORE streamice_cg_a1 = comlev1_stream_nl, key=ikey_nl
208  !$TAF STORE streamice_cg_a2 = comlev1_stream_nl, key=ikey_nl  !$TAF STORE streamice_cg_a2 = comlev1_stream_nl, key=ikey_nl
209  !$TAF STORE streamice_cg_a3 = comlev1_stream_nl, key=ikey_nl  !$TAF STORE streamice_cg_a3 = comlev1_stream_nl, key=ikey_nl
# Line 177  C       but after convergence loop does Line 216  C       but after convergence loop does
216  !$TAF STORE zv_old_si = comlev1_stream_nl, key=ikey_nl  !$TAF STORE zv_old_si = comlev1_stream_nl, key=ikey_nl
217  #endif  #endif
218    
219         IF (err_max .GT. streamice_nonlin_tol * err_init) THEN         IF ((err_max .GT. streamice_nonlin_tol * err_init) .and.
220         &     (err_max_fp .GT. streamice_nonlin_tol_fp)) THEN
221    
222         iter_numconv = iter_numconv + 1         iter_numconv = iter_numconv + 1
223    
# Line 201  C       but after convergence loop does Line 241  C       but after convergence loop does
241  !$TAF STORE zv_old_si = comlev1_stream_nl, key=ikey_nl  !$TAF STORE zv_old_si = comlev1_stream_nl, key=ikey_nl
242  #endif  #endif
243    
244         CALL STREAMICE_CG_SOLVE(  #ifdef ALLOW_AUTODIFF_TAMC
245       &  U_streamice,  !       DO bj = myByLo(myThid), myByHi(myThid)
246       &  V_streamice,  !        DO bi = myBxLo(myThid), myBxHi(myThid)
247       &  taudx_SI,  !         DO j=1-OLy,sNy+OLy
248       &  taudy_SI,  !          DO i=1-OLx,sNx+OLx
249       &  cgtol,  !           U_streamice (i,j,bi,bj) = 0. _d 0
250       &  cg_iters,  !           V_streamice (i,j,bi,bj) = 0. _d 0
251       &  myThid )  !          ENDDO
252    !         ENDDO
253    !        ENDDO
254    !       ENDDO
255    #endif
256    
257           CALL STREAMICE_CG_WRAPPER(
258         &       U_streamice,
259         &       V_streamice,
260         &       taudx_SI,
261         &       taudy_SI,
262         &       cgtol,
263         &       cg_iters,
264         &       myThid )
265    
266          #ifdef STREAMICE_HYBRID_STRESS
267    #ifdef ALLOW_AUTODIFF_TAMC
268    !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
269    !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
270    #endif
271    #endif
272    
273    #ifdef STREAMICE_HYBRID_STRESS
274            CALL STREAMICE_TAUB (myThid)
275    #endif
276    
277         WRITE(msgBuf,'(A,I5,A,I4,A)') 'streamice linear solve number',         WRITE(msgBuf,'(A,I5,A,I4,A)') 'streamice linear solve number',
278       &                       iter, " ",       &                       iter, " ",
# Line 222  C       but after convergence loop does Line 284  C       but after convergence loop does
284  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
285  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
286  !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl  !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
287    #ifdef STREAMICE_HYBRID_STRESS
288    !$TAF STORE streamice_taubx = comlev1_stream_nl, key=ikey_nl
289    !$TAF STORE streamice_tauby = comlev1_stream_nl, key=ikey_nl
290    !$TAF STORE visc_streamice_full = comlev1_stream_nl, key=ikey_nl
291    #endif
292  #endif  #endif
293    
294    #ifdef STREAMICE_HYBRID_STRESS
295           CALL STREAMICE_VISC_BETA_HYBRID ( myThid )
296    #else
297         CALL STREAMICE_VISC_BETA ( myThid )         CALL STREAMICE_VISC_BETA ( myThid )
298    #endif
299    
300    
301  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
302  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
# Line 263  C       but after convergence loop does Line 335  C       but after convergence loop does
335       I    V_streamice,       I    V_streamice,
336       I    0, sNx+1, 0, sNy+1 )       I    0, sNx+1, 0, sNy+1 )
337    
338          if (iter .eq. streamice_max_nl_iter) then
339          CALL WRITE_FLD_XY_RL("u_bound_cont_A","",ubd_SI,0,myThid)
340          CALL WRITE_FLD_XY_RL("v_bound_cont_A","",vbd_SI,0,myThid)
341          CALL WRITE_FLD_XY_RL("u_bound_cont_B","",Au_SI,0,myThid)
342          CALL WRITE_FLD_XY_RL("v_bound_cont_B","",Av_SI,0,myThid)
343          endif
344    
345  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
346  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
347  !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl  !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
348  #endif  #endif
349    
350         err_max = 0. _d 0         err_max = 0. _d 0
351           err_max_fp = 0. _d 0
352    
353  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
354  !$TAF STORE err_max = comlev1_stream_nl, key=ikey_nl  !$TAF STORE err_max = comlev1_stream_nl, key=ikey_nl
# Line 284  C       but after convergence loop does Line 364  C       but after convergence loop does
364             err_tempv = 0. _d 0             err_tempv = 0. _d 0
365             IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN             IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN
366              err_tempu =              err_tempu =
367       &       ABS (Au_SI(i,j,bi,bj)+ubd_SI(i,j,bi,bj) -       &       ABS (Au_SI(i,j,bi,bj)+0*ubd_SI(i,j,bi,bj) -
368       &            taudx_SI(i,j,bi,bj))       &            taudx_SI(i,j,bi,bj))
369             ENDIF             ENDIF
370             IF (STREAMICE_vmask(i,j,bi,bj).eq.1) THEN             IF (STREAMICE_vmask(i,j,bi,bj).eq.1) THEN
371              err_tempv = MAX( err_tempu,              err_tempv = MAX( err_tempu,
372       &       ABS (Av_SI(i,j,bi,bj)+vbd_SI(i,j,bi,bj) -       &       ABS (Av_SI(i,j,bi,bj)+0*vbd_SI(i,j,bi,bj) -
373       &            taudy_SI(i,j,bi,bj)))       &            taudy_SI(i,j,bi,bj)))
374             ENDIF             ENDIF
375             IF (err_tempv .ge. err_max) err_max = err_tempv  !           if (err_tempu.ge.1.e2.or.err_tempv.ge.1.e2) THEN
376    !            print *, "FOUND MAX ", i,j,err_tempu,err_tempv,
377    !     &      ubd_SI(i,j,bi,bj),vbd_SI(i,j,bi,bj)
378    !           endif
379               IF (err_tempv .ge. err_max) THEN
380                err_max = err_tempv
381               ENDIF
382            ENDDO            ENDDO
383           ENDDO           ENDDO
384          ENDDO          ENDDO
# Line 300  C       but after convergence loop does Line 386  C       but after convergence loop does
386    
387         CALL GLOBAL_MAX_R8 (err_max, myThid)         CALL GLOBAL_MAX_R8 (err_max, myThid)
388    
389         WRITE(msgBuf,'(A,F11.7)') 'err/err_init',         DO bj = myByLo(myThid), myByHi(myThid)
390            DO bi = myBxLo(myThid), myBxHi(myThid)
391             DO j=1,sNy
392              DO i=1,sNx
393               err_tempu = 0. _d 0
394               err_tempv = 0. _d 0
395               IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN
396                err_tempu =
397         &       ABS (U_streamice(i,j,bi,bj)-u_old_SI(i,j,bi,bj))
398               ENDIF
399               IF (STREAMICE_vmask(i,j,bi,bj).eq.1) THEN
400                err_tempv = MAX( err_tempu,
401         &       ABS (V_streamice(i,j,bi,bj)-v_old_SI(i,j,bi,bj)))
402               ENDIF
403               IF (err_tempv .ge. err_max_fp) err_max_fp = err_tempv
404              ENDDO
405             ENDDO
406            ENDDO
407           ENDDO
408    
409           CALL GLOBAL_MAX_R8 (err_max_fp, myThid)
410           WRITE(msgBuf,'(A,E15.7)') '||x_i-x_{i-1}||_inf',
411         &                       err_max_fp
412           CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,
413         &                    SQUEEZE_RIGHT , 1)
414    
415           DO bj = myByLo(myThid), myByHi(myThid)
416            DO bi = myBxLo(myThid), myBxHi(myThid)
417             DO j=1-OLy,sNy+OLy
418              DO i=1-OLx,sNx+OLx
419               u_old_SI (i,j,bi,bj) = U_streamice (i,j,bi,bj)
420               v_old_SI (i,j,bi,bj) = V_streamice (i,j,bi,bj)
421              ENDDO
422             ENDDO
423            ENDDO
424           ENDDO
425    
426           WRITE(msgBuf,'(A,E15.7)') 'err/err_init',
427       &                       err_max/err_init       &                       err_max/err_init
428         CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,         CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,
429       &                    SQUEEZE_RIGHT , 1)       &                    SQUEEZE_RIGHT , 1)
# Line 309  C       but after convergence loop does Line 432  C       but after convergence loop does
432       &   STREAMICE_lower_cg_tol) THEN       &   STREAMICE_lower_cg_tol) THEN
433           cgtol = cgtol * 5.e-2           cgtol = cgtol * 5.e-2
434           err_lastchange = err_max           err_lastchange = err_max
435           WRITE(msgBuf,'(A,F11.7)') 'new cg tol: ',           WRITE(msgBuf,'(A,E15.7)') 'new cg tol: ',
436       &                       cgtol       &                       cgtol
437           CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,           CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,
438       &                    SQUEEZE_RIGHT , 1)       &                    SQUEEZE_RIGHT , 1)
439         ENDIF         ENDIF
440    
441    
442         ENDIF ! (err_max .GT. streamice_nonlin_tol * err_init)         ENDIF ! (err_max .GT. streamice_nonlin_tol * err_init)
443        ENDDO        ENDDO
444    
# Line 322  C END NL ITER. LOOP Line 446  C END NL ITER. LOOP
446  C-------------------------------------------------------------------  C-------------------------------------------------------------------
447    
448        if (iter_numconv .lt. streamice_max_nl_iter) then        if (iter_numconv .lt. streamice_max_nl_iter) then
449         PRINT *, "VELOCITY SOLVE CONVERGED, ", iter_numconv,         WRITE(msgBuf,'(A,I5,A)') 'VELOCITY SOLVE CONVERGED, ',
450       & " iterations"       &         iter_numconv, ' iterations'
451           CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,
452         &                     SQUEEZE_RIGHT , 1)
453        else        else
454         PRINT *, "VELOCITY SOLVE DID NOT CONVERGE IN ",         WRITE(msgBuf,'(A,I5,A)') 'VELOCITY SOLVE NOT CONVERGED IN ',
455       & iter, " iterations"       &         iter_numconv, ' iterations'
456           CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,
457         &                     SQUEEZE_RIGHT , 1)
458        endif        endif
459    
460        _EXCH_XY_RL (U_streamice, myThid)        _EXCH_XY_RL (U_streamice, myThid)
461        _EXCH_XY_RL (V_streamice, myThid)        _EXCH_XY_RL (V_streamice, myThid)
462    !      CALL WRITE_FLD_XY_RL("taubx","",streamice_taubx,0,myThid)
463    !      CALL WRITE_FLD_XY_RL("tauby","",streamice_tauby,0,myThid)
464    
465  #endif  #endif
466        RETURN        RETURN
Legend:
Removed from v.1.2  
changed lines
  Added in v.1.12
  ViewVC Help
Powered by ViewVC 1.1.22