/[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.7 by dgoldberg, Tue Sep 4 21:11:44 2012 UTC revision 1.13 by dgoldberg, Sat Apr 6 17:43:41 2013 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_PETSC
25    #include "finclude/petsc.h"
26    #endif
27    
28  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
29  # include "tamc.h"  # include "tamc.h"
30  #endif  #endif
# Line 46  C     LOCAL VARIABLES Line 50  C     LOCAL VARIABLES
50        _RL err_max, err_tempu, err_tempv, err_init, area, err_max_fp        _RL err_max, err_tempu, err_tempv, err_init, area, err_max_fp
51        _RL max_vel, tempu, tempv, err_lastchange, cgtol        _RL max_vel, tempu, tempv, err_lastchange, cgtol
52        CHARACTER*(MAX_LEN_MBUF) msgBuf        CHARACTER*(MAX_LEN_MBUF) msgBuf
53    #ifdef ALLOW_PETSC
54          PetscErrorCode ierr
55    #endif
56  !       _RL taudx (1-OLx:sNx+OLx,1-OLy:sNy+Oly,nSx,nSy)  !       _RL taudx (1-OLx:sNx+OLx,1-OLy:sNy+Oly,nSx,nSy)
57  !       _RL taudy (1-OLx:sNx+OLx,1-OLy:sNy+Oly,nSx,nSy)  !       _RL taudy (1-OLx:sNx+OLx,1-OLy:sNy+Oly,nSx,nSy)
58    
59        CALL STREAMICE_DRIVING_STRESS (myThid)        IF (STREAMICE_ppm_driving_stress) THEN
60           CALL STREAMICE_DRIVING_STRESS_PPM (myThid)
61          ELSE
62           CALL STREAMICE_DRIVING_STRESS (myThid)
63          ENDIF
64    
65        cgtol = streamice_cg_tol        cgtol = streamice_cg_tol
66    
# Line 74  C     LOCAL VARIABLES Line 85  C     LOCAL VARIABLES
85  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
86  !$TAF STORE U_streamice = comlev1, key=ikey_dynamics  !$TAF STORE U_streamice = comlev1, key=ikey_dynamics
87  !$TAF STORE V_streamice = comlev1, key=ikey_dynamics  !$TAF STORE V_streamice = comlev1, key=ikey_dynamics
88    #ifdef STREAMICE_HYBRID_STRESS
89    !$TAF STORE streamice_taubx = comlev1, key=ikey_dynamics
90    !$TAF STORE streamice_tauby = comlev1, key=ikey_dynamics
91    !$TAF STORE visc_streamice_full = comlev1, key=ikey_dynamics
92    #endif
93  #endif  #endif
94    
95        CALL STREAMICE_VISC_BETA ( myThid )  #ifdef STREAMICE_HYBRID_STRESS
96           CALL STREAMICE_VISC_BETA_HYBRID ( myThid )
97    #else
98           CALL STREAMICE_VISC_BETA ( myThid )
99    #endif
100    
101    #ifdef STREAMICE_HYBRID_STRESS
102    !$TAF STORE visc_streamice_full = comlev1, key=ikey_dynamics
103    #endif
104    
105        _EXCH_XY_RL ( tau_beta_eff_streamice , myThid )        _EXCH_XY_RL ( tau_beta_eff_streamice , myThid )
106        _EXCH_XY_RL ( visc_streamice , myThid )        _EXCH_XY_RL ( visc_streamice , myThid )
# Line 98  C     LOCAL VARIABLES Line 122  C     LOCAL VARIABLES
122        CALL STREAMICE_CG_BOUND_VALS( myThid,            CALL STREAMICE_CG_BOUND_VALS( myThid,    
123       O    ubd_SI,       O    ubd_SI,
124       O    vbd_SI)       O    vbd_SI)
125          
126    
127    !      CALL WRITE_FLD_XY_RL("u_bound_cont","",ubd_SI,0,myThid)
128    !      CALL WRITE_FLD_XY_RL("v_bound_cont","",vbd_SI,0,myThid)
129    !      CALL WRITE_FLD_XY_RL("taudx_u","",taudx_SI,0,myThid)
130    !      CALL WRITE_FLD_XY_RL("taudx_v","",taudy_SI,0,myThid)
131    
132  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
133  !$TAF STORE U_streamice = comlev1, key=ikey_dynamics  !$TAF STORE U_streamice = comlev1, key=ikey_dynamics
# Line 111  C     LOCAL VARIABLES Line 141  C     LOCAL VARIABLES
141       I    V_streamice,       I    V_streamice,
142       I    0, sNx+1, 0, sNy+1 )       I    0, sNx+1, 0, sNy+1 )
143    
144    
145    
146        err_init = 0. _d 0        err_init = 0. _d 0
147    
148        DO bj = myByLo(myThid), myByHi(myThid)        DO bj = myByLo(myThid), myByHi(myThid)
# Line 124  C     LOCAL VARIABLES Line 156  C     LOCAL VARIABLES
156            err_tempv = 0. _d 0            err_tempv = 0. _d 0
157            IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN            IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN
158             err_tempu =             err_tempu =
159       &      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) -
160       &           taudx_SI(i,j,bi,bj))       &           taudx_SI(i,j,bi,bj))
161    !            print *, "err_temp_u", err_tempu
162            ENDIF            ENDIF
163            IF (STREAMICE_vmask(i,j,bi,bj).eq.1) THEN            IF (STREAMICE_vmask(i,j,bi,bj).eq.1) THEN
164             err_tempv = MAX( err_tempu,             err_tempv = MAX( err_tempu,
165       &      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) -
166       &           taudy_SI(i,j,bi,bj)))       &           taudy_SI(i,j,bi,bj)))
167            ENDIF            ENDIF
168            IF (err_tempv .ge. err_init) err_init = err_tempv            IF (err_tempv .ge. err_init) THEN
169                err_init = err_tempv
170              ENDIF
171           ENDDO           ENDDO
172          ENDDO          ENDDO
173         ENDDO         ENDDO
# Line 156  C     LOCAL VARIABLES Line 191  C     LOCAL VARIABLES
191    
192  C START NL ITER. LOOP  C START NL ITER. LOOP
193  C-------------------------------------------------------------------  C-------------------------------------------------------------------
194    #ifdef ALLOW_PETSC
195          call petscInitialize(PETSC_NULL_CHARACTER,ierr)
196    #endif
197    
198        DO iter=1,streamice_max_nl_iter        DO iter=1,streamice_max_nl_iter
199    
# Line 235  C       but after convergence loop does Line 273  C       but after convergence loop does
273       &       cg_iters,       &       cg_iters,
274       &       myThid )       &       myThid )
275    
276          #ifdef STREAMICE_HYBRID_STRESS
277    #ifdef ALLOW_AUTODIFF_TAMC
278    !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
279    !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
280    #endif
281    #endif
282    
283    #ifdef STREAMICE_HYBRID_STRESS
284            CALL STREAMICE_TAUB (myThid)
285    #endif
286    
287         WRITE(msgBuf,'(A,I5,A,I4,A)') 'streamice linear solve number',         WRITE(msgBuf,'(A,I5,A,I4,A)') 'streamice linear solve number',
288       &                       iter, " ",       &                       iter, " ",
# Line 247  C       but after convergence loop does Line 294  C       but after convergence loop does
294  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
295  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
296  !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl  !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
297    #ifdef STREAMICE_HYBRID_STRESS
298    !$TAF STORE streamice_taubx = comlev1_stream_nl, key=ikey_nl
299    !$TAF STORE streamice_tauby = comlev1_stream_nl, key=ikey_nl
300    !$TAF STORE visc_streamice_full = comlev1_stream_nl, key=ikey_nl
301    #endif
302  #endif  #endif
303    
304    #ifdef STREAMICE_HYBRID_STRESS
305           CALL STREAMICE_VISC_BETA_HYBRID ( myThid )
306    #else
307         CALL STREAMICE_VISC_BETA ( myThid )         CALL STREAMICE_VISC_BETA ( myThid )
308    #endif
309    
310    
311  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
# Line 289  C       but after convergence loop does Line 345  C       but after convergence loop does
345       I    V_streamice,       I    V_streamice,
346       I    0, sNx+1, 0, sNy+1 )       I    0, sNx+1, 0, sNy+1 )
347    
348          if (iter .eq. streamice_max_nl_iter) then
349          CALL WRITE_FLD_XY_RL("u_bound_cont_A","",ubd_SI,0,myThid)
350          CALL WRITE_FLD_XY_RL("v_bound_cont_A","",vbd_SI,0,myThid)
351          CALL WRITE_FLD_XY_RL("u_bound_cont_B","",Au_SI,0,myThid)
352          CALL WRITE_FLD_XY_RL("v_bound_cont_B","",Av_SI,0,myThid)
353          endif
354    
355  #ifdef ALLOW_AUTODIFF_TAMC  #ifdef ALLOW_AUTODIFF_TAMC
356  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl  !$TAF STORE U_streamice = comlev1_stream_nl, key=ikey_nl
357  !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl  !$TAF STORE V_streamice = comlev1_stream_nl, key=ikey_nl
# Line 311  C       but after convergence loop does Line 374  C       but after convergence loop does
374             err_tempv = 0. _d 0             err_tempv = 0. _d 0
375             IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN             IF (STREAMICE_umask(i,j,bi,bj).eq.1) THEN
376              err_tempu =              err_tempu =
377       &       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) -
378       &            taudx_SI(i,j,bi,bj))       &            taudx_SI(i,j,bi,bj))
379             ENDIF             ENDIF
380             IF (STREAMICE_vmask(i,j,bi,bj).eq.1) THEN             IF (STREAMICE_vmask(i,j,bi,bj).eq.1) THEN
381              err_tempv = MAX( err_tempu,              err_tempv = MAX( err_tempu,
382       &       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) -
383       &            taudy_SI(i,j,bi,bj)))       &            taudy_SI(i,j,bi,bj)))
384             ENDIF             ENDIF
385             IF (err_tempv .ge. err_max) err_max = err_tempv  !           if (err_tempu.ge.1.e2.or.err_tempv.ge.1.e2) THEN
386    !            print *, "FOUND MAX ", i,j,err_tempu,err_tempv,
387    !     &      ubd_SI(i,j,bi,bj),vbd_SI(i,j,bi,bj)
388    !           endif
389               IF (err_tempv .ge. err_max) THEN
390                err_max = err_tempv
391               ENDIF
392            ENDDO            ENDDO
393           ENDDO           ENDDO
394          ENDDO          ENDDO
# Line 348  C       but after convergence loop does Line 417  C       but after convergence loop does
417         ENDDO         ENDDO
418    
419         CALL GLOBAL_MAX_R8 (err_max_fp, myThid)         CALL GLOBAL_MAX_R8 (err_max_fp, myThid)
420         print *, "err_max_fp", err_max_fp         WRITE(msgBuf,'(A,E15.7)') '||x_i-x_{i-1}||_inf',
421         &                       err_max_fp
422           CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,
423         &                    SQUEEZE_RIGHT , 1)
424    
425         DO bj = myByLo(myThid), myByHi(myThid)         DO bj = myByLo(myThid), myByHi(myThid)
426          DO bi = myBxLo(myThid), myBxHi(myThid)          DO bi = myBxLo(myThid), myBxHi(myThid)
# Line 361  C       but after convergence loop does Line 433  C       but after convergence loop does
433          ENDDO          ENDDO
434         ENDDO         ENDDO
435    
436         WRITE(msgBuf,'(A,F11.7)') 'err/err_init',         WRITE(msgBuf,'(A,E15.7)') 'err/err_init',
437       &                       err_max/err_init       &                       err_max/err_init
438         CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,         CALL PRINT_MESSAGE( msgBuf, standardMessageUnit,
439       &                    SQUEEZE_RIGHT , 1)       &                    SQUEEZE_RIGHT , 1)
# Line 380  C       but after convergence loop does Line 452  C       but after convergence loop does
452         ENDIF ! (err_max .GT. streamice_nonlin_tol * err_init)         ENDIF ! (err_max .GT. streamice_nonlin_tol * err_init)
453        ENDDO        ENDDO
454    
455    #ifdef ALLOW_PETSC
456          call PetscFinalize(ierr)
457    #endif
458    
459  C END NL ITER. LOOP  C END NL ITER. LOOP
460  C-------------------------------------------------------------------  C-------------------------------------------------------------------
461    
# Line 397  C--------------------------------------- Line 473  C---------------------------------------
473    
474        _EXCH_XY_RL (U_streamice, myThid)        _EXCH_XY_RL (U_streamice, myThid)
475        _EXCH_XY_RL (V_streamice, myThid)        _EXCH_XY_RL (V_streamice, myThid)
476    !      CALL WRITE_FLD_XY_RL("taubx","",streamice_taubx,0,myThid)
477    !      CALL WRITE_FLD_XY_RL("tauby","",streamice_tauby,0,myThid)
478    
479  #endif  #endif
480        RETURN        RETURN
Legend:
Removed from v.1.7  
changed lines
  Added in v.1.13
  ViewVC Help
Powered by ViewVC 1.1.22