xref: /petsc/src/snes/interface/snes.c (revision b850b91a487c21fc39a8177f06cac97e82c9be37)
163dd3a1aSKris Buschelman #define PETSCSNES_DLL
29b94acceSBarry Smith 
3b9147fbbSdalcinl #include "include/private/snesimpl.h"      /*I "petscsnes.h"  I*/
49b94acceSBarry Smith 
54c49b128SBarry Smith PetscTruth SNESRegisterAllCalled = PETSC_FALSE;
68ba1e511SMatthew Knepley PetscFList SNESList              = PETSC_NULL;
78ba1e511SMatthew Knepley 
88ba1e511SMatthew Knepley /* Logging support */
963dd3a1aSKris Buschelman PetscCookie PETSCSNES_DLLEXPORT SNES_COOKIE = 0;
106849ba73SBarry Smith PetscEvent  SNES_Solve = 0, SNES_LineSearch = 0, SNES_FunctionEval = 0, SNES_JacobianEval = 0;
11a09944afSBarry Smith 
12a09944afSBarry Smith #undef __FUNCT__
134a2ae208SSatish Balay #define __FUNCT__ "SNESView"
147e2c5f70SBarry Smith /*@C
159b94acceSBarry Smith    SNESView - Prints the SNES data structure.
169b94acceSBarry Smith 
174c49b128SBarry Smith    Collective on SNES
18fee21e36SBarry Smith 
19c7afd0dbSLois Curfman McInnes    Input Parameters:
20c7afd0dbSLois Curfman McInnes +  SNES - the SNES context
21c7afd0dbSLois Curfman McInnes -  viewer - visualization context
22c7afd0dbSLois Curfman McInnes 
239b94acceSBarry Smith    Options Database Key:
24c8a8ba5cSLois Curfman McInnes .  -snes_view - Calls SNESView() at end of SNESSolve()
259b94acceSBarry Smith 
269b94acceSBarry Smith    Notes:
279b94acceSBarry Smith    The available visualization contexts include
28b0a32e0cSBarry Smith +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
29b0a32e0cSBarry Smith -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
30c8a8ba5cSLois Curfman McInnes          output where only the first processor opens
31c8a8ba5cSLois Curfman McInnes          the file.  All other processors send their
32c8a8ba5cSLois Curfman McInnes          data to the first processor to print.
339b94acceSBarry Smith 
343e081fefSLois Curfman McInnes    The user can open an alternative visualization context with
35b0a32e0cSBarry Smith    PetscViewerASCIIOpen() - output to a specified file.
369b94acceSBarry Smith 
3736851e7fSLois Curfman McInnes    Level: beginner
3836851e7fSLois Curfman McInnes 
399b94acceSBarry Smith .keywords: SNES, view
409b94acceSBarry Smith 
41b0a32e0cSBarry Smith .seealso: PetscViewerASCIIOpen()
429b94acceSBarry Smith @*/
4363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESView(SNES snes,PetscViewer viewer)
449b94acceSBarry Smith {
459b94acceSBarry Smith   SNES_KSP_EW_ConvCtx *kctx;
46dfbe8321SBarry Smith   PetscErrorCode      ierr;
4794b7f48cSBarry Smith   KSP                 ksp;
48e060cb09SBarry Smith   SNESType            type;
4932077d6dSBarry Smith   PetscTruth          iascii,isstring;
509b94acceSBarry Smith 
513a40ed3dSBarry Smith   PetscFunctionBegin;
524482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
533050cee2SBarry Smith   if (!viewer) {
543050cee2SBarry Smith     ierr = PetscViewerASCIIGetStdout(snes->comm,&viewer);CHKERRQ(ierr);
553050cee2SBarry Smith   }
564482741eSBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_COOKIE,2);
57c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,viewer,2);
5874679c65SBarry Smith 
5932077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
60b0a32e0cSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_STRING,&isstring);CHKERRQ(ierr);
6132077d6dSBarry Smith   if (iascii) {
623a7fca6bSBarry Smith     if (snes->prefix) {
633a7fca6bSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"SNES Object:(%s)\n",snes->prefix);CHKERRQ(ierr);
643a7fca6bSBarry Smith     } else {
65b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"SNES Object:\n");CHKERRQ(ierr);
663a7fca6bSBarry Smith     }
67454a90a3SBarry Smith     ierr = SNESGetType(snes,&type);CHKERRQ(ierr);
68454a90a3SBarry Smith     if (type) {
69b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  type: %s\n",type);CHKERRQ(ierr);
70184914b5SBarry Smith     } else {
71b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  type: not set yet\n");CHKERRQ(ierr);
72184914b5SBarry Smith     }
73e7788613SBarry Smith     if (snes->ops->view) {
74b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
75e7788613SBarry Smith       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
76b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
770ef38995SBarry Smith     }
7877431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr);
79a83599f4SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  tolerances: relative=%G, absolute=%G, solution=%G\n",
8070441072SBarry Smith                  snes->rtol,snes->abstol,snes->xtol);CHKERRQ(ierr);
8177431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr);
8277431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr);
839b94acceSBarry Smith     if (snes->ksp_ewconv) {
849b94acceSBarry Smith       kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx;
859b94acceSBarry Smith       if (kctx) {
8677431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"  Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr);
87a83599f4SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"    rtol_0=%G, rtol_max=%G, threshold=%G\n",kctx->rtol_0,kctx->rtol_max,kctx->threshold);CHKERRQ(ierr);
88a83599f4SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"    gamma=%G, alpha=%G, alpha2=%G\n",kctx->gamma,kctx->alpha,kctx->alpha2);CHKERRQ(ierr);
899b94acceSBarry Smith       }
909b94acceSBarry Smith     }
910f5bd95cSBarry Smith   } else if (isstring) {
92454a90a3SBarry Smith     ierr = SNESGetType(snes,&type);CHKERRQ(ierr);
93b0a32e0cSBarry Smith     ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr);
9419bcc07fSBarry Smith   }
9594b7f48cSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
96b0a32e0cSBarry Smith   ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
9794b7f48cSBarry Smith   ierr = KSPView(ksp,viewer);CHKERRQ(ierr);
98b0a32e0cSBarry Smith   ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
993a40ed3dSBarry Smith   PetscFunctionReturn(0);
1009b94acceSBarry Smith }
1019b94acceSBarry Smith 
10276b2cf59SMatthew Knepley /*
10376b2cf59SMatthew Knepley   We retain a list of functions that also take SNES command
10476b2cf59SMatthew Knepley   line options. These are called at the end SNESSetFromOptions()
10576b2cf59SMatthew Knepley */
10676b2cf59SMatthew Knepley #define MAXSETFROMOPTIONS 5
107a7cc72afSBarry Smith static PetscInt numberofsetfromoptions;
1086849ba73SBarry Smith static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES);
10976b2cf59SMatthew Knepley 
110e74ef692SMatthew Knepley #undef __FUNCT__
111e74ef692SMatthew Knepley #define __FUNCT__ "SNESAddOptionsChecker"
112ac226902SBarry Smith /*@C
11376b2cf59SMatthew Knepley   SNESAddOptionsChecker - Adds an additional function to check for SNES options.
11476b2cf59SMatthew Knepley 
11576b2cf59SMatthew Knepley   Not Collective
11676b2cf59SMatthew Knepley 
11776b2cf59SMatthew Knepley   Input Parameter:
11876b2cf59SMatthew Knepley . snescheck - function that checks for options
11976b2cf59SMatthew Knepley 
12076b2cf59SMatthew Knepley   Level: developer
12176b2cf59SMatthew Knepley 
12276b2cf59SMatthew Knepley .seealso: SNESSetFromOptions()
12376b2cf59SMatthew Knepley @*/
12463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES))
12576b2cf59SMatthew Knepley {
12676b2cf59SMatthew Knepley   PetscFunctionBegin;
12776b2cf59SMatthew Knepley   if (numberofsetfromoptions >= MAXSETFROMOPTIONS) {
12877431f27SBarry Smith     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS);
12976b2cf59SMatthew Knepley   }
13076b2cf59SMatthew Knepley   othersetfromoptions[numberofsetfromoptions++] = snescheck;
13176b2cf59SMatthew Knepley   PetscFunctionReturn(0);
13276b2cf59SMatthew Knepley }
13376b2cf59SMatthew Knepley 
1344a2ae208SSatish Balay #undef __FUNCT__
1354a2ae208SSatish Balay #define __FUNCT__ "SNESSetFromOptions"
1369b94acceSBarry Smith /*@
13794b7f48cSBarry Smith    SNESSetFromOptions - Sets various SNES and KSP parameters from user options.
1389b94acceSBarry Smith 
139c7afd0dbSLois Curfman McInnes    Collective on SNES
140c7afd0dbSLois Curfman McInnes 
1419b94acceSBarry Smith    Input Parameter:
1429b94acceSBarry Smith .  snes - the SNES context
1439b94acceSBarry Smith 
14436851e7fSLois Curfman McInnes    Options Database Keys:
1456831982aSBarry Smith +  -snes_type <type> - ls, tr, umls, umtr, test
14682738288SBarry Smith .  -snes_stol - convergence tolerance in terms of the norm
14782738288SBarry Smith                 of the change in the solution between steps
14870441072SBarry Smith .  -snes_atol <abstol> - absolute tolerance of residual norm
149b39c3a46SLois Curfman McInnes .  -snes_rtol <rtol> - relative decrease in tolerance norm from initial
150b39c3a46SLois Curfman McInnes .  -snes_max_it <max_it> - maximum number of iterations
151b39c3a46SLois Curfman McInnes .  -snes_max_funcs <max_funcs> - maximum number of function evaluations
15250ffb88aSMatthew Knepley .  -snes_max_fail <max_fail> - maximum number of failures
153b39c3a46SLois Curfman McInnes .  -snes_trtol <trtol> - trust region tolerance
1542492ecdbSBarry Smith .  -snes_no_convergence_test - skip convergence test in nonlinear
15582738288SBarry Smith                                solver; hence iterations will continue until max_it
1561fbbfb26SLois Curfman McInnes                                or some other criterion is reached. Saves expense
15782738288SBarry Smith                                of convergence test
158e8105e01SRichard Katz .  -snes_monitor <optional filename> - prints residual norm at each iteration. if no
159e8105e01SRichard Katz                                        filename given prints to stdout
160a6570f20SBarry Smith .  -snes_monitor_solution - plots solution at each iteration
161a6570f20SBarry Smith .  -snes_monitor_residual - plots residual (not its norm) at each iteration
162a6570f20SBarry Smith .  -snes_monitor_solution_update - plots update to solution at each iteration
163a6570f20SBarry Smith .  -snes_monitor_draw - plots residual norm at each iteration
164e24b481bSBarry Smith .  -snes_fd - use finite differences to compute Jacobian; very slow, only for testing
1655968eb51SBarry Smith .  -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration
166fee2055bSBarry Smith -  -snes_converged_reason - print the reason for convergence/divergence after each solve
16782738288SBarry Smith 
16882738288SBarry Smith     Options Database for Eisenstat-Walker method:
1694b27c08aSLois Curfman McInnes +  -snes_ksp_ew_conv - use Eisenstat-Walker method for determining linear system convergence
1704b27c08aSLois Curfman McInnes .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
17136851e7fSLois Curfman McInnes .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
17236851e7fSLois Curfman McInnes .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
17336851e7fSLois Curfman McInnes .  -snes_ksp_ew_gamma <gamma> - Sets gamma
17436851e7fSLois Curfman McInnes .  -snes_ksp_ew_alpha <alpha> - Sets alpha
17536851e7fSLois Curfman McInnes .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
17636851e7fSLois Curfman McInnes -  -snes_ksp_ew_threshold <threshold> - Sets threshold
17782738288SBarry Smith 
17811ca99fdSLois Curfman McInnes    Notes:
17911ca99fdSLois Curfman McInnes    To see all options, run your program with the -help option or consult
18011ca99fdSLois Curfman McInnes    the users manual.
18183e2fdc7SBarry Smith 
18236851e7fSLois Curfman McInnes    Level: beginner
18336851e7fSLois Curfman McInnes 
1849b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database
1859b94acceSBarry Smith 
18669ed319cSSatish Balay .seealso: SNESSetOptionsPrefix()
1879b94acceSBarry Smith @*/
18863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetFromOptions(SNES snes)
1899b94acceSBarry Smith {
19094b7f48cSBarry Smith   KSP                     ksp;
191186905e3SBarry Smith   SNES_KSP_EW_ConvCtx     *kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx;
192f1af5d2fSBarry Smith   PetscTruth              flg;
193dfbe8321SBarry Smith   PetscErrorCode          ierr;
19477431f27SBarry Smith   PetscInt                i;
1952fc52814SBarry Smith   const char              *deft;
196e8105e01SRichard Katz   char                    type[256], monfilename[PETSC_MAX_PATH_LEN];
19723d894e5SBarry Smith   PetscViewerASCIIMonitor monviewer;
1989b94acceSBarry Smith 
1993a40ed3dSBarry Smith   PetscFunctionBegin;
2004482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
201ca161407SBarry Smith 
202b0a32e0cSBarry Smith   ierr = PetscOptionsBegin(snes->comm,snes->prefix,"Nonlinear solver (SNES) options","SNES");CHKERRQ(ierr);
203186905e3SBarry Smith     if (snes->type_name) {
204186905e3SBarry Smith       deft = snes->type_name;
205186905e3SBarry Smith     } else {
2064b27c08aSLois Curfman McInnes       deft = SNESLS;
207d64ed03dSBarry Smith     }
2084bbc92c1SBarry Smith 
209186905e3SBarry Smith     if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL);CHKERRQ(ierr);}
210b0a32e0cSBarry Smith     ierr = PetscOptionsList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr);
211d64ed03dSBarry Smith     if (flg) {
212186905e3SBarry Smith       ierr = SNESSetType(snes,type);CHKERRQ(ierr);
213186905e3SBarry Smith     } else if (!snes->type_name) {
214186905e3SBarry Smith       ierr = SNESSetType(snes,deft);CHKERRQ(ierr);
215d64ed03dSBarry Smith     }
216909c8a9fSBarry Smith     ierr = PetscOptionsName("-snes_view","Print detailed information on solver used","SNESView",0);CHKERRQ(ierr);
21793c39befSBarry Smith 
21887828ca2SBarry Smith     ierr = PetscOptionsReal("-snes_stol","Stop if step length less then","SNESSetTolerances",snes->xtol,&snes->xtol,0);CHKERRQ(ierr);
21970441072SBarry Smith     ierr = PetscOptionsReal("-snes_atol","Stop if function norm less then","SNESSetTolerances",snes->abstol,&snes->abstol,0);CHKERRQ(ierr);
220186905e3SBarry Smith 
22187828ca2SBarry Smith     ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less then","SNESSetTolerances",snes->rtol,&snes->rtol,0);CHKERRQ(ierr);
222b0a32e0cSBarry Smith     ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,PETSC_NULL);CHKERRQ(ierr);
223b0a32e0cSBarry Smith     ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,PETSC_NULL);CHKERRQ(ierr);
22450ffb88aSMatthew Knepley     ierr = PetscOptionsInt("-snes_max_fail","Maximum failures","SNESSetTolerances",snes->maxFailures,&snes->maxFailures,PETSC_NULL);CHKERRQ(ierr);
2255968eb51SBarry Smith     ierr = PetscOptionsName("-snes_converged_reason","Print reason for converged or diverged","SNESSolve",&flg);CHKERRQ(ierr);
2265968eb51SBarry Smith     if (flg) {
2275968eb51SBarry Smith       snes->printreason = PETSC_TRUE;
2285968eb51SBarry Smith     }
229186905e3SBarry Smith 
23071f87433Sdalcinl     ierr = PetscOptionsTruth("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNES_KSP_SetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,PETSC_NULL);CHKERRQ(ierr);
23171f87433Sdalcinl     ierr = PetscOptionsTruth("-snes_ksp_ew_conv","Use Eisentat-Walker linear system convergence test","SNES_KSP_SetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,PETSC_NULL);CHKERRQ(ierr);
232186905e3SBarry Smith 
233cf502942SBarry Smith     ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNES_KSP_SetParametersEW",kctx->version,&kctx->version,0);CHKERRQ(ierr);
23487828ca2SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNES_KSP_SetParametersEW",kctx->rtol_0,&kctx->rtol_0,0);CHKERRQ(ierr);
23587828ca2SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNES_KSP_SetParametersEW",kctx->rtol_max,&kctx->rtol_max,0);CHKERRQ(ierr);
23687828ca2SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNES_KSP_SetParametersEW",kctx->gamma,&kctx->gamma,0);CHKERRQ(ierr);
23787828ca2SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNES_KSP_SetParametersEW",kctx->alpha,&kctx->alpha,0);CHKERRQ(ierr);
23887828ca2SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNES_KSP_SetParametersEW",kctx->alpha2,&kctx->alpha2,0);CHKERRQ(ierr);
23987828ca2SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNES_KSP_SetParametersEW",kctx->threshold,&kctx->threshold,0);CHKERRQ(ierr);
240186905e3SBarry Smith 
241b0a32e0cSBarry Smith     ierr = PetscOptionsName("-snes_no_convergence_test","Don't test for convergence","None",&flg);CHKERRQ(ierr);
242e7788613SBarry Smith     if (flg) {snes->ops->converged = 0;}
243a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",&flg);CHKERRQ(ierr);
244a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);}
245eabae89aSBarry Smith 
246a6570f20SBarry Smith     ierr = PetscOptionsString("-snes_monitor","Monitor norm of function","SNESMonitorSet","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
247e8105e01SRichard Katz     if (flg) {
24823d894e5SBarry Smith       ierr = PetscViewerASCIIMonitorCreate(snes->comm,monfilename,0,&monviewer);CHKERRQ(ierr);
24923d894e5SBarry Smith       ierr = SNESMonitorSet(snes,SNESMonitorDefault,monviewer,(PetscErrorCode (*)(void*))PetscViewerASCIIMonitorDestroy);CHKERRQ(ierr);
250e8105e01SRichard Katz     }
251eabae89aSBarry Smith 
252a6570f20SBarry Smith     ierr = PetscOptionsString("-snes_ratiomonitor","Monitor ratios of norms of function","SNESMonitorSetRatio","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
253eabae89aSBarry Smith     if (flg) {
25423d894e5SBarry Smith       ierr = PetscViewerASCIIMonitorCreate(snes->comm,monfilename,0,&monviewer);CHKERRQ(ierr);
255f1bef1bcSMatthew Knepley       ierr = SNESMonitorSetRatio(snes,monviewer);CHKERRQ(ierr);
256e8105e01SRichard Katz     }
257eabae89aSBarry Smith 
258a6570f20SBarry Smith     ierr = PetscOptionsString("-snes_monitor_short","Monitor norm of function (fewer digits)","SNESMonitorSet","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
259eabae89aSBarry Smith     if (flg) {
26023d894e5SBarry Smith       ierr = PetscViewerASCIIMonitorCreate(snes->comm,monfilename,0,&monviewer);CHKERRQ(ierr);
26123d894e5SBarry Smith       ierr = SNESMonitorSet(snes,SNESMonitorDefaultShort,monviewer,(PetscErrorCode (*)(void*))PetscViewerASCIIMonitorDestroy);CHKERRQ(ierr);
262eabae89aSBarry Smith     }
263eabae89aSBarry Smith 
264a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_solution","Plot solution at each iteration","SNESMonitorSolution",&flg);CHKERRQ(ierr);
265a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorSolution,0,0);CHKERRQ(ierr);}
266a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_solution_update","Plot correction at each iteration","SNESMonitorSolutionUpdate",&flg);CHKERRQ(ierr);
267a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorSolutionUpdate,0,0);CHKERRQ(ierr);}
268a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_residual","Plot residual at each iteration","SNESMonitorResidual",&flg);CHKERRQ(ierr);
269a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorResidual,0,0);CHKERRQ(ierr);}
270a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_draw","Plot function norm at each iteration","SNESMonitorLG",&flg);CHKERRQ(ierr);
271a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorLG,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);}
272e24b481bSBarry Smith 
273b0a32e0cSBarry Smith     ierr = PetscOptionsName("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESDefaultComputeJacobian",&flg);CHKERRQ(ierr);
2744b27c08aSLois Curfman McInnes     if (flg) {
275186905e3SBarry Smith       ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESDefaultComputeJacobian,snes->funP);CHKERRQ(ierr);
276ae15b995SBarry Smith       ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr);
2779b94acceSBarry Smith     }
278639f9d9dSBarry Smith 
27976b2cf59SMatthew Knepley     for(i = 0; i < numberofsetfromoptions; i++) {
28076b2cf59SMatthew Knepley       ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr);
28176b2cf59SMatthew Knepley     }
28276b2cf59SMatthew Knepley 
283e7788613SBarry Smith     if (snes->ops->setfromoptions) {
284e7788613SBarry Smith       ierr = (*snes->ops->setfromoptions)(snes);CHKERRQ(ierr);
285639f9d9dSBarry Smith     }
286b0a32e0cSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
2874bbc92c1SBarry Smith 
28894b7f48cSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
28994b7f48cSBarry Smith   ierr = KSPSetFromOptions(ksp);CHKERRQ(ierr);
29093993e2dSLois Curfman McInnes 
2913a40ed3dSBarry Smith   PetscFunctionReturn(0);
2929b94acceSBarry Smith }
2939b94acceSBarry Smith 
294a847f771SSatish Balay 
2954a2ae208SSatish Balay #undef __FUNCT__
2964a2ae208SSatish Balay #define __FUNCT__ "SNESSetApplicationContext"
2979b94acceSBarry Smith /*@
2989b94acceSBarry Smith    SNESSetApplicationContext - Sets the optional user-defined context for
2999b94acceSBarry Smith    the nonlinear solvers.
3009b94acceSBarry Smith 
301fee21e36SBarry Smith    Collective on SNES
302fee21e36SBarry Smith 
303c7afd0dbSLois Curfman McInnes    Input Parameters:
304c7afd0dbSLois Curfman McInnes +  snes - the SNES context
305c7afd0dbSLois Curfman McInnes -  usrP - optional user context
306c7afd0dbSLois Curfman McInnes 
30736851e7fSLois Curfman McInnes    Level: intermediate
30836851e7fSLois Curfman McInnes 
3099b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context
3109b94acceSBarry Smith 
3119b94acceSBarry Smith .seealso: SNESGetApplicationContext()
3129b94acceSBarry Smith @*/
31363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetApplicationContext(SNES snes,void *usrP)
3149b94acceSBarry Smith {
3153a40ed3dSBarry Smith   PetscFunctionBegin;
3164482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
3179b94acceSBarry Smith   snes->user		= usrP;
3183a40ed3dSBarry Smith   PetscFunctionReturn(0);
3199b94acceSBarry Smith }
32074679c65SBarry Smith 
3214a2ae208SSatish Balay #undef __FUNCT__
3224a2ae208SSatish Balay #define __FUNCT__ "SNESGetApplicationContext"
3239b94acceSBarry Smith /*@C
3249b94acceSBarry Smith    SNESGetApplicationContext - Gets the user-defined context for the
3259b94acceSBarry Smith    nonlinear solvers.
3269b94acceSBarry Smith 
327c7afd0dbSLois Curfman McInnes    Not Collective
328c7afd0dbSLois Curfman McInnes 
3299b94acceSBarry Smith    Input Parameter:
3309b94acceSBarry Smith .  snes - SNES context
3319b94acceSBarry Smith 
3329b94acceSBarry Smith    Output Parameter:
3339b94acceSBarry Smith .  usrP - user context
3349b94acceSBarry Smith 
33536851e7fSLois Curfman McInnes    Level: intermediate
33636851e7fSLois Curfman McInnes 
3379b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context
3389b94acceSBarry Smith 
3399b94acceSBarry Smith .seealso: SNESSetApplicationContext()
3409b94acceSBarry Smith @*/
34163dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetApplicationContext(SNES snes,void **usrP)
3429b94acceSBarry Smith {
3433a40ed3dSBarry Smith   PetscFunctionBegin;
3444482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
3459b94acceSBarry Smith   *usrP = snes->user;
3463a40ed3dSBarry Smith   PetscFunctionReturn(0);
3479b94acceSBarry Smith }
34874679c65SBarry Smith 
3494a2ae208SSatish Balay #undef __FUNCT__
3504a2ae208SSatish Balay #define __FUNCT__ "SNESGetIterationNumber"
3519b94acceSBarry Smith /*@
352c8228a4eSBarry Smith    SNESGetIterationNumber - Gets the number of nonlinear iterations completed
353c8228a4eSBarry Smith    at this time.
3549b94acceSBarry Smith 
355c7afd0dbSLois Curfman McInnes    Not Collective
356c7afd0dbSLois Curfman McInnes 
3579b94acceSBarry Smith    Input Parameter:
3589b94acceSBarry Smith .  snes - SNES context
3599b94acceSBarry Smith 
3609b94acceSBarry Smith    Output Parameter:
3619b94acceSBarry Smith .  iter - iteration number
3629b94acceSBarry Smith 
363c8228a4eSBarry Smith    Notes:
364c8228a4eSBarry Smith    For example, during the computation of iteration 2 this would return 1.
365c8228a4eSBarry Smith 
366c8228a4eSBarry Smith    This is useful for using lagged Jacobians (where one does not recompute the
36708405cd6SLois Curfman McInnes    Jacobian at each SNES iteration). For example, the code
36808405cd6SLois Curfman McInnes .vb
36908405cd6SLois Curfman McInnes       ierr = SNESGetIterationNumber(snes,&it);
37008405cd6SLois Curfman McInnes       if (!(it % 2)) {
37108405cd6SLois Curfman McInnes         [compute Jacobian here]
37208405cd6SLois Curfman McInnes       }
37308405cd6SLois Curfman McInnes .ve
374c8228a4eSBarry Smith    can be used in your ComputeJacobian() function to cause the Jacobian to be
37508405cd6SLois Curfman McInnes    recomputed every second SNES iteration.
376c8228a4eSBarry Smith 
37736851e7fSLois Curfman McInnes    Level: intermediate
37836851e7fSLois Curfman McInnes 
3792b668275SBarry Smith .keywords: SNES, nonlinear, get, iteration, number,
3802b668275SBarry Smith 
381*b850b91aSLisandro Dalcin .seealso:   SNESGetFunctionNorm(), SNESGetLinearSolveIterations()
3829b94acceSBarry Smith @*/
38363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetIterationNumber(SNES snes,PetscInt* iter)
3849b94acceSBarry Smith {
3853a40ed3dSBarry Smith   PetscFunctionBegin;
3864482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
3874482741eSBarry Smith   PetscValidIntPointer(iter,2);
3889b94acceSBarry Smith   *iter = snes->iter;
3893a40ed3dSBarry Smith   PetscFunctionReturn(0);
3909b94acceSBarry Smith }
39174679c65SBarry Smith 
3924a2ae208SSatish Balay #undef __FUNCT__
3934a2ae208SSatish Balay #define __FUNCT__ "SNESGetFunctionNorm"
3949b94acceSBarry Smith /*@
3959b94acceSBarry Smith    SNESGetFunctionNorm - Gets the norm of the current function that was set
3969b94acceSBarry Smith    with SNESSSetFunction().
3979b94acceSBarry Smith 
398c7afd0dbSLois Curfman McInnes    Collective on SNES
399c7afd0dbSLois Curfman McInnes 
4009b94acceSBarry Smith    Input Parameter:
4019b94acceSBarry Smith .  snes - SNES context
4029b94acceSBarry Smith 
4039b94acceSBarry Smith    Output Parameter:
4049b94acceSBarry Smith .  fnorm - 2-norm of function
4059b94acceSBarry Smith 
40636851e7fSLois Curfman McInnes    Level: intermediate
40736851e7fSLois Curfman McInnes 
4089b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm
409a86d99e1SLois Curfman McInnes 
410*b850b91aSLisandro Dalcin .seealso: SNESGetFunction(), SNESGetIterationNumber(), SNESGetLinearSolveIterations()
4119b94acceSBarry Smith @*/
41271f87433Sdalcinl PetscErrorCode PETSCSNES_DLLEXPORT SNESGetFunctionNorm(SNES snes,PetscReal *fnorm)
4139b94acceSBarry Smith {
4143a40ed3dSBarry Smith   PetscFunctionBegin;
4154482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
4164482741eSBarry Smith   PetscValidScalarPointer(fnorm,2);
4179b94acceSBarry Smith   *fnorm = snes->norm;
4183a40ed3dSBarry Smith   PetscFunctionReturn(0);
4199b94acceSBarry Smith }
42074679c65SBarry Smith 
4214a2ae208SSatish Balay #undef __FUNCT__
422*b850b91aSLisandro Dalcin #define __FUNCT__ "SNESGetNonlinearStepFailures"
4239b94acceSBarry Smith /*@
424*b850b91aSLisandro Dalcin    SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps
4259b94acceSBarry Smith    attempted by the nonlinear solver.
4269b94acceSBarry Smith 
427c7afd0dbSLois Curfman McInnes    Not Collective
428c7afd0dbSLois Curfman McInnes 
4299b94acceSBarry Smith    Input Parameter:
4309b94acceSBarry Smith .  snes - SNES context
4319b94acceSBarry Smith 
4329b94acceSBarry Smith    Output Parameter:
4339b94acceSBarry Smith .  nfails - number of unsuccessful steps attempted
4349b94acceSBarry Smith 
435c96a6f78SLois Curfman McInnes    Notes:
436c96a6f78SLois Curfman McInnes    This counter is reset to zero for each successive call to SNESSolve().
437c96a6f78SLois Curfman McInnes 
43836851e7fSLois Curfman McInnes    Level: intermediate
43936851e7fSLois Curfman McInnes 
4409b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps
4419b94acceSBarry Smith @*/
442*b850b91aSLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESGetNonlinearStepFailures(SNES snes,PetscInt* nfails)
4439b94acceSBarry Smith {
4443a40ed3dSBarry Smith   PetscFunctionBegin;
4454482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
4464482741eSBarry Smith   PetscValidIntPointer(nfails,2);
44750ffb88aSMatthew Knepley   *nfails = snes->numFailures;
44850ffb88aSMatthew Knepley   PetscFunctionReturn(0);
44950ffb88aSMatthew Knepley }
45050ffb88aSMatthew Knepley 
45150ffb88aSMatthew Knepley #undef __FUNCT__
452*b850b91aSLisandro Dalcin #define __FUNCT__ "SNESSetMaxNonlinearStepFailures"
45350ffb88aSMatthew Knepley /*@
454*b850b91aSLisandro Dalcin    SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps
45550ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
45650ffb88aSMatthew Knepley 
45750ffb88aSMatthew Knepley    Not Collective
45850ffb88aSMatthew Knepley 
45950ffb88aSMatthew Knepley    Input Parameters:
46050ffb88aSMatthew Knepley +  snes     - SNES context
46150ffb88aSMatthew Knepley -  maxFails - maximum of unsuccessful steps
46250ffb88aSMatthew Knepley 
46350ffb88aSMatthew Knepley    Level: intermediate
46450ffb88aSMatthew Knepley 
46550ffb88aSMatthew Knepley .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
46650ffb88aSMatthew Knepley @*/
467*b850b91aSLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails)
46850ffb88aSMatthew Knepley {
46950ffb88aSMatthew Knepley   PetscFunctionBegin;
4704482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
47150ffb88aSMatthew Knepley   snes->maxFailures = maxFails;
47250ffb88aSMatthew Knepley   PetscFunctionReturn(0);
47350ffb88aSMatthew Knepley }
47450ffb88aSMatthew Knepley 
47550ffb88aSMatthew Knepley #undef __FUNCT__
476*b850b91aSLisandro Dalcin #define __FUNCT__ "SNESGetMaxNonlinearStepFailures"
47750ffb88aSMatthew Knepley /*@
478*b850b91aSLisandro Dalcin    SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps
47950ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
48050ffb88aSMatthew Knepley 
48150ffb88aSMatthew Knepley    Not Collective
48250ffb88aSMatthew Knepley 
48350ffb88aSMatthew Knepley    Input Parameter:
48450ffb88aSMatthew Knepley .  snes     - SNES context
48550ffb88aSMatthew Knepley 
48650ffb88aSMatthew Knepley    Output Parameter:
48750ffb88aSMatthew Knepley .  maxFails - maximum of unsuccessful steps
48850ffb88aSMatthew Knepley 
48950ffb88aSMatthew Knepley    Level: intermediate
49050ffb88aSMatthew Knepley 
49150ffb88aSMatthew Knepley .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
49250ffb88aSMatthew Knepley @*/
493*b850b91aSLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails)
49450ffb88aSMatthew Knepley {
49550ffb88aSMatthew Knepley   PetscFunctionBegin;
4964482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
4974482741eSBarry Smith   PetscValidIntPointer(maxFails,2);
49850ffb88aSMatthew Knepley   *maxFails = snes->maxFailures;
4993a40ed3dSBarry Smith   PetscFunctionReturn(0);
5009b94acceSBarry Smith }
501a847f771SSatish Balay 
5024a2ae208SSatish Balay #undef __FUNCT__
5033d4c4710SBarry Smith #define __FUNCT__ "SNESGetLinearSolveFailures"
5043d4c4710SBarry Smith /*@
5053d4c4710SBarry Smith    SNESGetLinearSolveFailures - Gets the number of failed (non-converged)
5063d4c4710SBarry Smith    linear solvers.
5073d4c4710SBarry Smith 
5083d4c4710SBarry Smith    Not Collective
5093d4c4710SBarry Smith 
5103d4c4710SBarry Smith    Input Parameter:
5113d4c4710SBarry Smith .  snes - SNES context
5123d4c4710SBarry Smith 
5133d4c4710SBarry Smith    Output Parameter:
5143d4c4710SBarry Smith .  nfails - number of failed solves
5153d4c4710SBarry Smith 
5163d4c4710SBarry Smith    Notes:
5173d4c4710SBarry Smith    This counter is reset to zero for each successive call to SNESSolve().
5183d4c4710SBarry Smith 
5193d4c4710SBarry Smith    Level: intermediate
5203d4c4710SBarry Smith 
5213d4c4710SBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps
5223d4c4710SBarry Smith @*/
5233d4c4710SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESGetLinearSolveFailures(SNES snes,PetscInt* nfails)
5243d4c4710SBarry Smith {
5253d4c4710SBarry Smith   PetscFunctionBegin;
5263d4c4710SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
5273d4c4710SBarry Smith   PetscValidIntPointer(nfails,2);
5283d4c4710SBarry Smith   *nfails = snes->numLinearSolveFailures;
5293d4c4710SBarry Smith   PetscFunctionReturn(0);
5303d4c4710SBarry Smith }
5313d4c4710SBarry Smith 
5323d4c4710SBarry Smith #undef __FUNCT__
5333d4c4710SBarry Smith #define __FUNCT__ "SNESSetMaxLinearSolveFailures"
5343d4c4710SBarry Smith /*@
5353d4c4710SBarry Smith    SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts
5363d4c4710SBarry Smith    allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE
5373d4c4710SBarry Smith 
5383d4c4710SBarry Smith    Collective on SNES
5393d4c4710SBarry Smith 
5403d4c4710SBarry Smith    Input Parameters:
5413d4c4710SBarry Smith +  snes     - SNES context
5423d4c4710SBarry Smith -  maxFails - maximum allowed linear solve failures
5433d4c4710SBarry Smith 
5443d4c4710SBarry Smith    Level: intermediate
5453d4c4710SBarry Smith 
5463d4c4710SBarry Smith    Notes: By default this is 1; that is SNES returns on the first failed linear solve
5473d4c4710SBarry Smith 
5483d4c4710SBarry Smith .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
5493d4c4710SBarry Smith 
5503d4c4710SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures()
5513d4c4710SBarry Smith @*/
5523d4c4710SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails)
5533d4c4710SBarry Smith {
5543d4c4710SBarry Smith   PetscFunctionBegin;
5553d4c4710SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
5563d4c4710SBarry Smith   snes->maxLinearSolveFailures = maxFails;
5573d4c4710SBarry Smith   PetscFunctionReturn(0);
5583d4c4710SBarry Smith }
5593d4c4710SBarry Smith 
5603d4c4710SBarry Smith #undef __FUNCT__
5613d4c4710SBarry Smith #define __FUNCT__ "SNESGetMaxLinearSolveFailures"
5623d4c4710SBarry Smith /*@
5633d4c4710SBarry Smith    SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that
5643d4c4710SBarry Smith      are allowed before SNES terminates
5653d4c4710SBarry Smith 
5663d4c4710SBarry Smith    Not Collective
5673d4c4710SBarry Smith 
5683d4c4710SBarry Smith    Input Parameter:
5693d4c4710SBarry Smith .  snes     - SNES context
5703d4c4710SBarry Smith 
5713d4c4710SBarry Smith    Output Parameter:
5723d4c4710SBarry Smith .  maxFails - maximum of unsuccessful solves allowed
5733d4c4710SBarry Smith 
5743d4c4710SBarry Smith    Level: intermediate
5753d4c4710SBarry Smith 
5763d4c4710SBarry Smith    Notes: By default this is 1; that is SNES returns on the first failed linear solve
5773d4c4710SBarry Smith 
5783d4c4710SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
5793d4c4710SBarry Smith 
5803d4c4710SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures()
5813d4c4710SBarry Smith @*/
5823d4c4710SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails)
5833d4c4710SBarry Smith {
5843d4c4710SBarry Smith   PetscFunctionBegin;
5853d4c4710SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
5863d4c4710SBarry Smith   PetscValidIntPointer(maxFails,2);
5873d4c4710SBarry Smith   *maxFails = snes->maxLinearSolveFailures;
5883d4c4710SBarry Smith   PetscFunctionReturn(0);
5893d4c4710SBarry Smith }
5903d4c4710SBarry Smith 
5913d4c4710SBarry Smith #undef __FUNCT__
592*b850b91aSLisandro Dalcin #define __FUNCT__ "SNESGetLinearSolveIterations"
593c96a6f78SLois Curfman McInnes /*@
594*b850b91aSLisandro Dalcin    SNESGetLinearSolveIterations - Gets the total number of linear iterations
595c96a6f78SLois Curfman McInnes    used by the nonlinear solver.
596c96a6f78SLois Curfman McInnes 
597c7afd0dbSLois Curfman McInnes    Not Collective
598c7afd0dbSLois Curfman McInnes 
599c96a6f78SLois Curfman McInnes    Input Parameter:
600c96a6f78SLois Curfman McInnes .  snes - SNES context
601c96a6f78SLois Curfman McInnes 
602c96a6f78SLois Curfman McInnes    Output Parameter:
603c96a6f78SLois Curfman McInnes .  lits - number of linear iterations
604c96a6f78SLois Curfman McInnes 
605c96a6f78SLois Curfman McInnes    Notes:
606c96a6f78SLois Curfman McInnes    This counter is reset to zero for each successive call to SNESSolve().
607c96a6f78SLois Curfman McInnes 
60836851e7fSLois Curfman McInnes    Level: intermediate
60936851e7fSLois Curfman McInnes 
610c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations
6112b668275SBarry Smith 
6122b668275SBarry Smith .seealso:  SNESGetIterationNumber(), SNESGetFunctionNorm()
613c96a6f78SLois Curfman McInnes @*/
614*b850b91aSLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESGetLinearSolveIterations(SNES snes,PetscInt* lits)
615c96a6f78SLois Curfman McInnes {
6163a40ed3dSBarry Smith   PetscFunctionBegin;
6174482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
6184482741eSBarry Smith   PetscValidIntPointer(lits,2);
619c96a6f78SLois Curfman McInnes   *lits = snes->linear_its;
6203a40ed3dSBarry Smith   PetscFunctionReturn(0);
621c96a6f78SLois Curfman McInnes }
622c96a6f78SLois Curfman McInnes 
6234a2ae208SSatish Balay #undef __FUNCT__
62494b7f48cSBarry Smith #define __FUNCT__ "SNESGetKSP"
62552baeb72SSatish Balay /*@
62694b7f48cSBarry Smith    SNESGetKSP - Returns the KSP context for a SNES solver.
6279b94acceSBarry Smith 
62894b7f48cSBarry Smith    Not Collective, but if SNES object is parallel, then KSP object is parallel
629c7afd0dbSLois Curfman McInnes 
6309b94acceSBarry Smith    Input Parameter:
6319b94acceSBarry Smith .  snes - the SNES context
6329b94acceSBarry Smith 
6339b94acceSBarry Smith    Output Parameter:
63494b7f48cSBarry Smith .  ksp - the KSP context
6359b94acceSBarry Smith 
6369b94acceSBarry Smith    Notes:
63794b7f48cSBarry Smith    The user can then directly manipulate the KSP context to set various
6389b94acceSBarry Smith    options, etc.  Likewise, the user can then extract and manipulate the
6392999313aSBarry Smith    PC contexts as well.
6409b94acceSBarry Smith 
64136851e7fSLois Curfman McInnes    Level: beginner
64236851e7fSLois Curfman McInnes 
64394b7f48cSBarry Smith .keywords: SNES, nonlinear, get, KSP, context
6449b94acceSBarry Smith 
6452999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
6469b94acceSBarry Smith @*/
64763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetKSP(SNES snes,KSP *ksp)
6489b94acceSBarry Smith {
6493a40ed3dSBarry Smith   PetscFunctionBegin;
6504482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
6514482741eSBarry Smith   PetscValidPointer(ksp,2);
65294b7f48cSBarry Smith   *ksp = snes->ksp;
6533a40ed3dSBarry Smith   PetscFunctionReturn(0);
6549b94acceSBarry Smith }
65582bf6240SBarry Smith 
6564a2ae208SSatish Balay #undef __FUNCT__
6572999313aSBarry Smith #define __FUNCT__ "SNESSetKSP"
6582999313aSBarry Smith /*@
6592999313aSBarry Smith    SNESSetKSP - Sets a KSP context for the SNES object to use
6602999313aSBarry Smith 
6612999313aSBarry Smith    Not Collective, but the SNES and KSP objects must live on the same MPI_Comm
6622999313aSBarry Smith 
6632999313aSBarry Smith    Input Parameters:
6642999313aSBarry Smith +  snes - the SNES context
6652999313aSBarry Smith -  ksp - the KSP context
6662999313aSBarry Smith 
6672999313aSBarry Smith    Notes:
6682999313aSBarry Smith    The SNES object already has its KSP object, you can obtain with SNESGetKSP()
6692999313aSBarry Smith    so this routine is rarely needed.
6702999313aSBarry Smith 
6712999313aSBarry Smith    The KSP object that is already in the SNES object has its reference count
6722999313aSBarry Smith    decreased by one.
6732999313aSBarry Smith 
6742999313aSBarry Smith    Level: developer
6752999313aSBarry Smith 
6762999313aSBarry Smith .keywords: SNES, nonlinear, get, KSP, context
6772999313aSBarry Smith 
6782999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
6792999313aSBarry Smith @*/
6802999313aSBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetKSP(SNES snes,KSP ksp)
6812999313aSBarry Smith {
6822999313aSBarry Smith   PetscErrorCode ierr;
6832999313aSBarry Smith 
6842999313aSBarry Smith   PetscFunctionBegin;
6852999313aSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
6862999313aSBarry Smith   PetscValidHeaderSpecific(ksp,KSP_COOKIE,2);
6872999313aSBarry Smith   PetscCheckSameComm(snes,1,ksp,2);
6887dcf0eaaSdalcinl   ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr);
689906ed7ccSBarry Smith   if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);}
6902999313aSBarry Smith   snes->ksp = ksp;
6912999313aSBarry Smith   PetscFunctionReturn(0);
6922999313aSBarry Smith }
6932999313aSBarry Smith 
6942999313aSBarry Smith #undef __FUNCT__
6954a2ae208SSatish Balay #define __FUNCT__ "SNESPublish_Petsc"
6966849ba73SBarry Smith static PetscErrorCode SNESPublish_Petsc(PetscObject obj)
697e24b481bSBarry Smith {
698e24b481bSBarry Smith   PetscFunctionBegin;
699e24b481bSBarry Smith   PetscFunctionReturn(0);
700e24b481bSBarry Smith }
701e24b481bSBarry Smith 
7029b94acceSBarry Smith /* -----------------------------------------------------------*/
7034a2ae208SSatish Balay #undef __FUNCT__
7044a2ae208SSatish Balay #define __FUNCT__ "SNESCreate"
70552baeb72SSatish Balay /*@
7069b94acceSBarry Smith    SNESCreate - Creates a nonlinear solver context.
7079b94acceSBarry Smith 
708c7afd0dbSLois Curfman McInnes    Collective on MPI_Comm
709c7afd0dbSLois Curfman McInnes 
710c7afd0dbSLois Curfman McInnes    Input Parameters:
711906ed7ccSBarry Smith .  comm - MPI communicator
7129b94acceSBarry Smith 
7139b94acceSBarry Smith    Output Parameter:
7149b94acceSBarry Smith .  outsnes - the new SNES context
7159b94acceSBarry Smith 
716c7afd0dbSLois Curfman McInnes    Options Database Keys:
717c7afd0dbSLois Curfman McInnes +   -snes_mf - Activates default matrix-free Jacobian-vector products,
718c7afd0dbSLois Curfman McInnes                and no preconditioning matrix
719c7afd0dbSLois Curfman McInnes .   -snes_mf_operator - Activates default matrix-free Jacobian-vector
720c7afd0dbSLois Curfman McInnes                products, and a user-provided preconditioning matrix
721c7afd0dbSLois Curfman McInnes                as set by SNESSetJacobian()
722c7afd0dbSLois Curfman McInnes -   -snes_fd - Uses (slow!) finite differences to compute Jacobian
723c1f60f51SBarry Smith 
72436851e7fSLois Curfman McInnes    Level: beginner
72536851e7fSLois Curfman McInnes 
7269b94acceSBarry Smith .keywords: SNES, nonlinear, create, context
7279b94acceSBarry Smith 
7284b27c08aSLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy(), SNES
7299b94acceSBarry Smith @*/
73063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESCreate(MPI_Comm comm,SNES *outsnes)
7319b94acceSBarry Smith {
732dfbe8321SBarry Smith   PetscErrorCode      ierr;
7339b94acceSBarry Smith   SNES                snes;
7349b94acceSBarry Smith   SNES_KSP_EW_ConvCtx *kctx;
73537fcc0dbSBarry Smith 
7363a40ed3dSBarry Smith   PetscFunctionBegin;
737ed1caa07SMatthew Knepley   PetscValidPointer(outsnes,2);
7388ba1e511SMatthew Knepley   *outsnes = PETSC_NULL;
7398ba1e511SMatthew Knepley #ifndef PETSC_USE_DYNAMIC_LIBRARIES
7408ba1e511SMatthew Knepley   ierr = SNESInitializePackage(PETSC_NULL);CHKERRQ(ierr);
7418ba1e511SMatthew Knepley #endif
7428ba1e511SMatthew Knepley 
743e7788613SBarry Smith   ierr = PetscHeaderCreate(snes,_p_SNES,struct _SNESOps,SNES_COOKIE,0,"SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr);
744e24b481bSBarry Smith   snes->bops->publish     = SNESPublish_Petsc;
7459b94acceSBarry Smith   snes->max_its           = 50;
7469750a799SBarry Smith   snes->max_funcs	  = 10000;
7479b94acceSBarry Smith   snes->norm		  = 0.0;
748b4874afaSBarry Smith   snes->rtol		  = 1.e-8;
749b4874afaSBarry Smith   snes->ttol              = 0.0;
75070441072SBarry Smith   snes->abstol		  = 1.e-50;
7519b94acceSBarry Smith   snes->xtol		  = 1.e-8;
7524b27c08aSLois Curfman McInnes   snes->deltatol	  = 1.e-12;
7539b94acceSBarry Smith   snes->nfuncs            = 0;
75450ffb88aSMatthew Knepley   snes->numFailures       = 0;
75550ffb88aSMatthew Knepley   snes->maxFailures       = 1;
7567a00f4a9SLois Curfman McInnes   snes->linear_its        = 0;
757639f9d9dSBarry Smith   snes->numbermonitors    = 0;
7589b94acceSBarry Smith   snes->data              = 0;
7593f149594SLisandro Dalcin   snes->ops->converged    = 0;
760e7788613SBarry Smith   snes->ops->view         = 0;
7614dc4c822SBarry Smith   snes->setupcalled       = PETSC_FALSE;
762186905e3SBarry Smith   snes->ksp_ewconv        = PETSC_FALSE;
7636f24a144SLois Curfman McInnes   snes->vwork             = 0;
7646f24a144SLois Curfman McInnes   snes->nwork             = 0;
765758f92a0SBarry Smith   snes->conv_hist_len     = 0;
766758f92a0SBarry Smith   snes->conv_hist_max     = 0;
767758f92a0SBarry Smith   snes->conv_hist         = PETSC_NULL;
768758f92a0SBarry Smith   snes->conv_hist_its     = PETSC_NULL;
769758f92a0SBarry Smith   snes->conv_hist_reset   = PETSC_TRUE;
770184914b5SBarry Smith   snes->reason            = SNES_CONVERGED_ITERATING;
7719b94acceSBarry Smith 
7723d4c4710SBarry Smith   snes->numLinearSolveFailures = 0;
7733d4c4710SBarry Smith   snes->maxLinearSolveFailures = 1;
7743d4c4710SBarry Smith 
7759b94acceSBarry Smith   /* Create context to compute Eisenstat-Walker relative tolerance for KSP */
776b0a32e0cSBarry Smith   ierr = PetscNew(SNES_KSP_EW_ConvCtx,&kctx);CHKERRQ(ierr);
77752e6d16bSBarry Smith   ierr = PetscLogObjectMemory(snes,sizeof(SNES_KSP_EW_ConvCtx));CHKERRQ(ierr);
7789b94acceSBarry Smith   snes->kspconvctx  = (void*)kctx;
7799b94acceSBarry Smith   kctx->version     = 2;
7809b94acceSBarry Smith   kctx->rtol_0      = .3; /* Eisenstat and Walker suggest rtol_0=.5, but
7819b94acceSBarry Smith                              this was too large for some test cases */
7829b94acceSBarry Smith   kctx->rtol_last   = 0;
7839b94acceSBarry Smith   kctx->rtol_max    = .9;
7849b94acceSBarry Smith   kctx->gamma       = 1.0;
78571f87433Sdalcinl   kctx->alpha       = .5*(1.0 + sqrt(5.0));
78671f87433Sdalcinl   kctx->alpha2      = kctx->alpha;
7879b94acceSBarry Smith   kctx->threshold   = .1;
7889b94acceSBarry Smith   kctx->lresid_last = 0;
7899b94acceSBarry Smith   kctx->norm_last   = 0;
7909b94acceSBarry Smith 
79194b7f48cSBarry Smith   ierr = KSPCreate(comm,&snes->ksp);CHKERRQ(ierr);
79252e6d16bSBarry Smith   ierr = PetscLogObjectParent(snes,snes->ksp);CHKERRQ(ierr);
7935334005bSBarry Smith 
7949b94acceSBarry Smith   *outsnes = snes;
79500036973SBarry Smith   ierr = PetscPublishAll(snes);CHKERRQ(ierr);
7963a40ed3dSBarry Smith   PetscFunctionReturn(0);
7979b94acceSBarry Smith }
7989b94acceSBarry Smith 
7994a2ae208SSatish Balay #undef __FUNCT__
8004a2ae208SSatish Balay #define __FUNCT__ "SNESSetFunction"
8019b94acceSBarry Smith /*@C
8029b94acceSBarry Smith    SNESSetFunction - Sets the function evaluation routine and function
8039b94acceSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
8049b94acceSBarry Smith    equations.
8059b94acceSBarry Smith 
806fee21e36SBarry Smith    Collective on SNES
807fee21e36SBarry Smith 
808c7afd0dbSLois Curfman McInnes    Input Parameters:
809c7afd0dbSLois Curfman McInnes +  snes - the SNES context
810c7afd0dbSLois Curfman McInnes .  r - vector to store function value
811de044059SHong Zhang .  func - function evaluation routine
812c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
813c7afd0dbSLois Curfman McInnes          function evaluation routine (may be PETSC_NULL)
8149b94acceSBarry Smith 
815c7afd0dbSLois Curfman McInnes    Calling sequence of func:
8168d76a1e5SLois Curfman McInnes $    func (SNES snes,Vec x,Vec f,void *ctx);
817c7afd0dbSLois Curfman McInnes 
818313e4042SLois Curfman McInnes .  f - function vector
819c7afd0dbSLois Curfman McInnes -  ctx - optional user-defined function context
8209b94acceSBarry Smith 
8219b94acceSBarry Smith    Notes:
8229b94acceSBarry Smith    The Newton-like methods typically solve linear systems of the form
8239b94acceSBarry Smith $      f'(x) x = -f(x),
824c7afd0dbSLois Curfman McInnes    where f'(x) denotes the Jacobian matrix and f(x) is the function.
8259b94acceSBarry Smith 
82636851e7fSLois Curfman McInnes    Level: beginner
82736851e7fSLois Curfman McInnes 
8289b94acceSBarry Smith .keywords: SNES, nonlinear, set, function
8299b94acceSBarry Smith 
830a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian()
8319b94acceSBarry Smith @*/
83263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*func)(SNES,Vec,Vec,void*),void *ctx)
8339b94acceSBarry Smith {
8343a40ed3dSBarry Smith   PetscFunctionBegin;
8354482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
8364482741eSBarry Smith   PetscValidHeaderSpecific(r,VEC_COOKIE,2);
837c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,r,2);
838184914b5SBarry Smith 
839e7788613SBarry Smith   snes->ops->computefunction = func;
8409b94acceSBarry Smith   snes->vec_func             = snes->vec_func_always = r;
8419b94acceSBarry Smith   snes->funP                 = ctx;
8423a40ed3dSBarry Smith   PetscFunctionReturn(0);
8439b94acceSBarry Smith }
8449b94acceSBarry Smith 
8453ab0aad5SBarry Smith /* --------------------------------------------------------------- */
8463ab0aad5SBarry Smith #undef __FUNCT__
8473ab0aad5SBarry Smith #define __FUNCT__ "SNESSetRhs"
8483ab0aad5SBarry Smith /*@C
8493ab0aad5SBarry Smith    SNESSetRhs - Sets the vector for solving F(x) = rhs. If rhs is not set
8503ab0aad5SBarry Smith    it assumes a zero right hand side.
8513ab0aad5SBarry Smith 
8523ab0aad5SBarry Smith    Collective on SNES
8533ab0aad5SBarry Smith 
8543ab0aad5SBarry Smith    Input Parameters:
8553ab0aad5SBarry Smith +  snes - the SNES context
8563ab0aad5SBarry Smith -  rhs - the right hand side vector or PETSC_NULL for a zero right hand side
8573ab0aad5SBarry Smith 
8583ab0aad5SBarry Smith    Level: intermediate
8593ab0aad5SBarry Smith 
8603ab0aad5SBarry Smith .keywords: SNES, nonlinear, set, function, right hand side
8613ab0aad5SBarry Smith 
8621096aae1SMatthew Knepley .seealso: SNESGetRhs(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
8633ab0aad5SBarry Smith @*/
86463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetRhs(SNES snes,Vec rhs)
8653ab0aad5SBarry Smith {
866dfbe8321SBarry Smith   PetscErrorCode ierr;
8673ab0aad5SBarry Smith 
8683ab0aad5SBarry Smith   PetscFunctionBegin;
8693ab0aad5SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
8703ab0aad5SBarry Smith   if (rhs) {
8713ab0aad5SBarry Smith     PetscValidHeaderSpecific(rhs,VEC_COOKIE,2);
8723ab0aad5SBarry Smith     PetscCheckSameComm(snes,1,rhs,2);
8733ab0aad5SBarry Smith     ierr = PetscObjectReference((PetscObject)rhs);CHKERRQ(ierr);
8743ab0aad5SBarry Smith   }
8753ab0aad5SBarry Smith   if (snes->afine) {
8763ab0aad5SBarry Smith     ierr = VecDestroy(snes->afine);CHKERRQ(ierr);
8773ab0aad5SBarry Smith   }
8783ab0aad5SBarry Smith   snes->afine = rhs;
8793ab0aad5SBarry Smith   PetscFunctionReturn(0);
8803ab0aad5SBarry Smith }
8813ab0aad5SBarry Smith 
8824a2ae208SSatish Balay #undef __FUNCT__
8831096aae1SMatthew Knepley #define __FUNCT__ "SNESGetRhs"
8841096aae1SMatthew Knepley /*@C
8851096aae1SMatthew Knepley    SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set
8861096aae1SMatthew Knepley    it assumes a zero right hand side.
8871096aae1SMatthew Knepley 
8881096aae1SMatthew Knepley    Collective on SNES
8891096aae1SMatthew Knepley 
8901096aae1SMatthew Knepley    Input Parameter:
8911096aae1SMatthew Knepley .  snes - the SNES context
8921096aae1SMatthew Knepley 
8931096aae1SMatthew Knepley    Output Parameter:
8941096aae1SMatthew Knepley .  rhs - the right hand side vector or PETSC_NULL for a zero right hand side
8951096aae1SMatthew Knepley 
8961096aae1SMatthew Knepley    Level: intermediate
8971096aae1SMatthew Knepley 
8981096aae1SMatthew Knepley .keywords: SNES, nonlinear, get, function, right hand side
8991096aae1SMatthew Knepley 
9001096aae1SMatthew Knepley .seealso: SNESSetRhs(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
9011096aae1SMatthew Knepley @*/
9021096aae1SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESGetRhs(SNES snes,Vec *rhs)
9031096aae1SMatthew Knepley {
9041096aae1SMatthew Knepley   PetscFunctionBegin;
9051096aae1SMatthew Knepley   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
9061096aae1SMatthew Knepley   PetscValidPointer(rhs,2);
9071096aae1SMatthew Knepley   *rhs = snes->afine;
9081096aae1SMatthew Knepley   PetscFunctionReturn(0);
9091096aae1SMatthew Knepley }
9101096aae1SMatthew Knepley 
9111096aae1SMatthew Knepley #undef __FUNCT__
9124a2ae208SSatish Balay #define __FUNCT__ "SNESComputeFunction"
9139b94acceSBarry Smith /*@
91436851e7fSLois Curfman McInnes    SNESComputeFunction - Calls the function that has been set with
9159b94acceSBarry Smith                          SNESSetFunction().
9169b94acceSBarry Smith 
917c7afd0dbSLois Curfman McInnes    Collective on SNES
918c7afd0dbSLois Curfman McInnes 
9199b94acceSBarry Smith    Input Parameters:
920c7afd0dbSLois Curfman McInnes +  snes - the SNES context
921c7afd0dbSLois Curfman McInnes -  x - input vector
9229b94acceSBarry Smith 
9239b94acceSBarry Smith    Output Parameter:
9243638b69dSLois Curfman McInnes .  y - function vector, as set by SNESSetFunction()
9259b94acceSBarry Smith 
9261bffabb2SLois Curfman McInnes    Notes:
92736851e7fSLois Curfman McInnes    SNESComputeFunction() is typically used within nonlinear solvers
92836851e7fSLois Curfman McInnes    implementations, so most users would not generally call this routine
92936851e7fSLois Curfman McInnes    themselves.
93036851e7fSLois Curfman McInnes 
93136851e7fSLois Curfman McInnes    Level: developer
93236851e7fSLois Curfman McInnes 
9339b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function
9349b94acceSBarry Smith 
935a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction()
9369b94acceSBarry Smith @*/
93763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESComputeFunction(SNES snes,Vec x,Vec y)
9389b94acceSBarry Smith {
939dfbe8321SBarry Smith   PetscErrorCode ierr;
9409b94acceSBarry Smith 
9413a40ed3dSBarry Smith   PetscFunctionBegin;
9424482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
9434482741eSBarry Smith   PetscValidHeaderSpecific(x,VEC_COOKIE,2);
9444482741eSBarry Smith   PetscValidHeaderSpecific(y,VEC_COOKIE,3);
945c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,x,2);
946c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,3);
947184914b5SBarry Smith 
948d5ba7fb7SMatthew Knepley   ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
949e7788613SBarry Smith   if (snes->ops->computefunction) {
950d64ed03dSBarry Smith     PetscStackPush("SNES user function");
951e9a2bbcdSBarry Smith     CHKMEMQ;
952e7788613SBarry Smith     ierr = (*snes->ops->computefunction)(snes,x,y,snes->funP);
953e9a2bbcdSBarry Smith     CHKMEMQ;
954d64ed03dSBarry Smith     PetscStackPop;
955d5e45103SBarry Smith     if (PetscExceptionValue(ierr)) {
95619717074SBarry Smith       PetscErrorCode pierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(pierr);
95719717074SBarry Smith     }
958d5e45103SBarry Smith     CHKERRQ(ierr);
9591096aae1SMatthew Knepley   } else if (snes->afine) {
9601096aae1SMatthew Knepley     ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr);
9611096aae1SMatthew Knepley   } else {
9621096aae1SMatthew Knepley     SETERRQ(PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() before SNESComputeFunction(), likely called from SNESSolve().");
9631096aae1SMatthew Knepley   }
9643ab0aad5SBarry Smith   if (snes->afine) {
965016dedfbSBarry Smith     ierr = VecAXPY(y,-1.0,snes->afine);CHKERRQ(ierr);
9663ab0aad5SBarry Smith   }
967ae3c334cSLois Curfman McInnes   snes->nfuncs++;
968d5ba7fb7SMatthew Knepley   ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
9693a40ed3dSBarry Smith   PetscFunctionReturn(0);
9709b94acceSBarry Smith }
9719b94acceSBarry Smith 
9724a2ae208SSatish Balay #undef __FUNCT__
9734a2ae208SSatish Balay #define __FUNCT__ "SNESComputeJacobian"
97462fef451SLois Curfman McInnes /*@
97562fef451SLois Curfman McInnes    SNESComputeJacobian - Computes the Jacobian matrix that has been
97662fef451SLois Curfman McInnes    set with SNESSetJacobian().
97762fef451SLois Curfman McInnes 
978c7afd0dbSLois Curfman McInnes    Collective on SNES and Mat
979c7afd0dbSLois Curfman McInnes 
98062fef451SLois Curfman McInnes    Input Parameters:
981c7afd0dbSLois Curfman McInnes +  snes - the SNES context
982c7afd0dbSLois Curfman McInnes -  x - input vector
98362fef451SLois Curfman McInnes 
98462fef451SLois Curfman McInnes    Output Parameters:
985c7afd0dbSLois Curfman McInnes +  A - Jacobian matrix
98662fef451SLois Curfman McInnes .  B - optional preconditioning matrix
9872b668275SBarry Smith -  flag - flag indicating matrix structure (one of, SAME_NONZERO_PATTERN,DIFFERENT_NONZERO_PATTERN,SAME_PRECONDITIONER)
988fee21e36SBarry Smith 
98962fef451SLois Curfman McInnes    Notes:
99062fef451SLois Curfman McInnes    Most users should not need to explicitly call this routine, as it
99162fef451SLois Curfman McInnes    is used internally within the nonlinear solvers.
99262fef451SLois Curfman McInnes 
99394b7f48cSBarry Smith    See KSPSetOperators() for important information about setting the
994dc5a77f8SLois Curfman McInnes    flag parameter.
99562fef451SLois Curfman McInnes 
99636851e7fSLois Curfman McInnes    Level: developer
99736851e7fSLois Curfman McInnes 
99862fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix
99962fef451SLois Curfman McInnes 
10002b668275SBarry Smith .seealso:  SNESSetJacobian(), KSPSetOperators(), MatStructure
100162fef451SLois Curfman McInnes @*/
100263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg)
10039b94acceSBarry Smith {
1004dfbe8321SBarry Smith   PetscErrorCode ierr;
10053a40ed3dSBarry Smith 
10063a40ed3dSBarry Smith   PetscFunctionBegin;
10074482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
10084482741eSBarry Smith   PetscValidHeaderSpecific(X,VEC_COOKIE,2);
10094482741eSBarry Smith   PetscValidPointer(flg,5);
1010c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,X,2);
1011e7788613SBarry Smith   if (!snes->ops->computejacobian) PetscFunctionReturn(0);
1012d5ba7fb7SMatthew Knepley   ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,*A,*B);CHKERRQ(ierr);
1013c4fc05e7SBarry Smith   *flg = DIFFERENT_NONZERO_PATTERN;
1014d64ed03dSBarry Smith   PetscStackPush("SNES user Jacobian function");
1015dc67937bSBarry Smith   CHKMEMQ;
1016e7788613SBarry Smith   ierr = (*snes->ops->computejacobian)(snes,X,A,B,flg,snes->jacP);CHKERRQ(ierr);
1017dc67937bSBarry Smith   CHKMEMQ;
1018d64ed03dSBarry Smith   PetscStackPop;
1019d5ba7fb7SMatthew Knepley   ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,*A,*B);CHKERRQ(ierr);
10206d84be18SBarry Smith   /* make sure user returned a correct Jacobian and preconditioner */
10216ce558aeSBarry Smith   /* PetscValidHeaderSpecific(*A,MAT_COOKIE,3);
10226ce558aeSBarry Smith     PetscValidHeaderSpecific(*B,MAT_COOKIE,4);   */
10233a40ed3dSBarry Smith   PetscFunctionReturn(0);
10249b94acceSBarry Smith }
10259b94acceSBarry Smith 
10264a2ae208SSatish Balay #undef __FUNCT__
10274a2ae208SSatish Balay #define __FUNCT__ "SNESSetJacobian"
10289b94acceSBarry Smith /*@C
10299b94acceSBarry Smith    SNESSetJacobian - Sets the function to compute Jacobian as well as the
1030044dda88SLois Curfman McInnes    location to store the matrix.
10319b94acceSBarry Smith 
1032c7afd0dbSLois Curfman McInnes    Collective on SNES and Mat
1033c7afd0dbSLois Curfman McInnes 
10349b94acceSBarry Smith    Input Parameters:
1035c7afd0dbSLois Curfman McInnes +  snes - the SNES context
10369b94acceSBarry Smith .  A - Jacobian matrix
10379b94acceSBarry Smith .  B - preconditioner matrix (usually same as the Jacobian)
10389b94acceSBarry Smith .  func - Jacobian evaluation routine
1039c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
10402cd2dfdcSLois Curfman McInnes          Jacobian evaluation routine (may be PETSC_NULL)
10419b94acceSBarry Smith 
10429b94acceSBarry Smith    Calling sequence of func:
10438d76a1e5SLois Curfman McInnes $     func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx);
10449b94acceSBarry Smith 
1045c7afd0dbSLois Curfman McInnes +  x - input vector
10469b94acceSBarry Smith .  A - Jacobian matrix
10479b94acceSBarry Smith .  B - preconditioner matrix, usually the same as A
1048ac21db08SLois Curfman McInnes .  flag - flag indicating information about the preconditioner matrix
10492b668275SBarry Smith    structure (same as flag in KSPSetOperators()), one of SAME_NONZERO_PATTERN,DIFFERENT_NONZERO_PATTERN,SAME_PRECONDITIONER
1050c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined Jacobian context
10519b94acceSBarry Smith 
10529b94acceSBarry Smith    Notes:
105394b7f48cSBarry Smith    See KSPSetOperators() for important information about setting the flag
10542cd2dfdcSLois Curfman McInnes    output parameter in the routine func().  Be sure to read this information!
1055ac21db08SLois Curfman McInnes 
1056ac21db08SLois Curfman McInnes    The routine func() takes Mat * as the matrix arguments rather than Mat.
10579b94acceSBarry Smith    This allows the Jacobian evaluation routine to replace A and/or B with a
10589b94acceSBarry Smith    completely new new matrix structure (not just different matrix elements)
10599b94acceSBarry Smith    when appropriate, for instance, if the nonzero structure is changing
10609b94acceSBarry Smith    throughout the global iterations.
10619b94acceSBarry Smith 
106236851e7fSLois Curfman McInnes    Level: beginner
106336851e7fSLois Curfman McInnes 
10649b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix
10659b94acceSBarry Smith 
10662b668275SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatSNESMFComputeJacobian(), SNESDefaultComputeJacobianColor(), MatStructure
10679b94acceSBarry Smith @*/
106863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetJacobian(SNES snes,Mat A,Mat B,PetscErrorCode (*func)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void *ctx)
10699b94acceSBarry Smith {
1070dfbe8321SBarry Smith   PetscErrorCode ierr;
10713a7fca6bSBarry Smith 
10723a40ed3dSBarry Smith   PetscFunctionBegin;
10734482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
10744482741eSBarry Smith   if (A) PetscValidHeaderSpecific(A,MAT_COOKIE,2);
10754482741eSBarry Smith   if (B) PetscValidHeaderSpecific(B,MAT_COOKIE,3);
1076c9780b6fSBarry Smith   if (A) PetscCheckSameComm(snes,1,A,2);
1077c9780b6fSBarry Smith   if (B) PetscCheckSameComm(snes,1,B,2);
1078e7788613SBarry Smith   if (func) snes->ops->computejacobian = func;
10793a7fca6bSBarry Smith   if (ctx)  snes->jacP                 = ctx;
10803a7fca6bSBarry Smith   if (A) {
10817dcf0eaaSdalcinl     ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
10823a7fca6bSBarry Smith     if (snes->jacobian) {ierr = MatDestroy(snes->jacobian);CHKERRQ(ierr);}
10839b94acceSBarry Smith     snes->jacobian = A;
10843a7fca6bSBarry Smith   }
10853a7fca6bSBarry Smith   if (B) {
10867dcf0eaaSdalcinl     ierr = PetscObjectReference((PetscObject)B);CHKERRQ(ierr);
10873a7fca6bSBarry Smith     if (snes->jacobian_pre) {ierr = MatDestroy(snes->jacobian_pre);CHKERRQ(ierr);}
10889b94acceSBarry Smith     snes->jacobian_pre = B;
10893a7fca6bSBarry Smith   }
10903a40ed3dSBarry Smith   PetscFunctionReturn(0);
10919b94acceSBarry Smith }
109262fef451SLois Curfman McInnes 
10934a2ae208SSatish Balay #undef __FUNCT__
10944a2ae208SSatish Balay #define __FUNCT__ "SNESGetJacobian"
1095c2aafc4cSSatish Balay /*@C
1096b4fd4287SBarry Smith    SNESGetJacobian - Returns the Jacobian matrix and optionally the user
1097b4fd4287SBarry Smith    provided context for evaluating the Jacobian.
1098b4fd4287SBarry Smith 
1099c7afd0dbSLois Curfman McInnes    Not Collective, but Mat object will be parallel if SNES object is
1100c7afd0dbSLois Curfman McInnes 
1101b4fd4287SBarry Smith    Input Parameter:
1102b4fd4287SBarry Smith .  snes - the nonlinear solver context
1103b4fd4287SBarry Smith 
1104b4fd4287SBarry Smith    Output Parameters:
1105c7afd0dbSLois Curfman McInnes +  A - location to stash Jacobian matrix (or PETSC_NULL)
1106b4fd4287SBarry Smith .  B - location to stash preconditioner matrix (or PETSC_NULL)
110770e92668SMatthew Knepley .  func - location to put Jacobian function (or PETSC_NULL)
110870e92668SMatthew Knepley -  ctx - location to stash Jacobian ctx (or PETSC_NULL)
1109fee21e36SBarry Smith 
111036851e7fSLois Curfman McInnes    Level: advanced
111136851e7fSLois Curfman McInnes 
1112b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian()
1113b4fd4287SBarry Smith @*/
111470e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESGetJacobian(SNES snes,Mat *A,Mat *B,PetscErrorCode (**func)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void **ctx)
1115b4fd4287SBarry Smith {
11163a40ed3dSBarry Smith   PetscFunctionBegin;
11174482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1118b4fd4287SBarry Smith   if (A)    *A    = snes->jacobian;
1119b4fd4287SBarry Smith   if (B)    *B    = snes->jacobian_pre;
1120e7788613SBarry Smith   if (func) *func = snes->ops->computejacobian;
112170e92668SMatthew Knepley   if (ctx)  *ctx  = snes->jacP;
11223a40ed3dSBarry Smith   PetscFunctionReturn(0);
1123b4fd4287SBarry Smith }
1124b4fd4287SBarry Smith 
11259b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */
112663dd3a1aSKris Buschelman EXTERN PetscErrorCode PETSCSNES_DLLEXPORT SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*);
11279b94acceSBarry Smith 
11284a2ae208SSatish Balay #undef __FUNCT__
11294a2ae208SSatish Balay #define __FUNCT__ "SNESSetUp"
11309b94acceSBarry Smith /*@
11319b94acceSBarry Smith    SNESSetUp - Sets up the internal data structures for the later use
1132272ac6f2SLois Curfman McInnes    of a nonlinear solver.
11339b94acceSBarry Smith 
1134fee21e36SBarry Smith    Collective on SNES
1135fee21e36SBarry Smith 
1136c7afd0dbSLois Curfman McInnes    Input Parameters:
113770e92668SMatthew Knepley .  snes - the SNES context
1138c7afd0dbSLois Curfman McInnes 
1139272ac6f2SLois Curfman McInnes    Notes:
1140272ac6f2SLois Curfman McInnes    For basic use of the SNES solvers the user need not explicitly call
1141272ac6f2SLois Curfman McInnes    SNESSetUp(), since these actions will automatically occur during
1142272ac6f2SLois Curfman McInnes    the call to SNESSolve().  However, if one wishes to control this
1143272ac6f2SLois Curfman McInnes    phase separately, SNESSetUp() should be called after SNESCreate()
1144272ac6f2SLois Curfman McInnes    and optional routines of the form SNESSetXXX(), but before SNESSolve().
1145272ac6f2SLois Curfman McInnes 
114636851e7fSLois Curfman McInnes    Level: advanced
114736851e7fSLois Curfman McInnes 
11489b94acceSBarry Smith .keywords: SNES, nonlinear, setup
11499b94acceSBarry Smith 
11509b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy()
11519b94acceSBarry Smith @*/
115270e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESSetUp(SNES snes)
11539b94acceSBarry Smith {
1154dfbe8321SBarry Smith   PetscErrorCode ierr;
115571f87433Sdalcinl   PetscTruth     flg;
11563a40ed3dSBarry Smith 
11573a40ed3dSBarry Smith   PetscFunctionBegin;
11584482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
11594dc4c822SBarry Smith   if (snes->setupcalled) PetscFunctionReturn(0);
11609b94acceSBarry Smith 
1161b0a32e0cSBarry Smith   ierr = PetscOptionsHasName(snes->prefix,"-snes_mf_operator",&flg);CHKERRQ(ierr);
1162c1f60f51SBarry Smith   /*
1163c1f60f51SBarry Smith       This version replaces the user provided Jacobian matrix with a
1164dfa02198SLois Curfman McInnes       matrix-free version but still employs the user-provided preconditioner matrix
1165c1f60f51SBarry Smith   */
1166c1f60f51SBarry Smith   if (flg) {
1167c1f60f51SBarry Smith     Mat J;
11685a655dc6SBarry Smith     ierr = MatCreateSNESMF(snes,snes->vec_sol,&J);CHKERRQ(ierr);
11695a655dc6SBarry Smith     ierr = MatSNESMFSetFromOptions(J);CHKERRQ(ierr);
1170ae15b995SBarry Smith     ierr = PetscInfo(snes,"Setting default matrix-free operator routines\n");CHKERRQ(ierr);
11713a7fca6bSBarry Smith     ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);
11723a7fca6bSBarry Smith     ierr = MatDestroy(J);CHKERRQ(ierr);
1173c1f60f51SBarry Smith   }
117445fc7adcSBarry Smith 
117503c60df9SBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_SINGLE) && !defined(PETSC_USE_MAT_SINGLE) && !defined(PETSC_USE_LONG_DOUBLE) && !defined(PETSC_USE_INT)
117645fc7adcSBarry Smith   ierr = PetscOptionsHasName(snes->prefix,"-snes_mf_operator2",&flg);CHKERRQ(ierr);
117745fc7adcSBarry Smith   if (flg) {
117845fc7adcSBarry Smith     Mat J;
117945fc7adcSBarry Smith     ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_sol,&J);CHKERRQ(ierr);
118045fc7adcSBarry Smith     ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);
118145fc7adcSBarry Smith     ierr = MatDestroy(J);CHKERRQ(ierr);
118245fc7adcSBarry Smith   }
118332a4b47aSMatthew Knepley #endif
118445fc7adcSBarry Smith 
1185b0a32e0cSBarry Smith   ierr = PetscOptionsHasName(snes->prefix,"-snes_mf",&flg);CHKERRQ(ierr);
1186c1f60f51SBarry Smith   /*
1187dfa02198SLois Curfman McInnes       This version replaces both the user-provided Jacobian and the user-
1188c1f60f51SBarry Smith       provided preconditioner matrix with the default matrix free version.
1189c1f60f51SBarry Smith    */
119031615d04SLois Curfman McInnes   if (flg) {
1191272ac6f2SLois Curfman McInnes     Mat  J;
1192b5d62d44SBarry Smith     KSP ksp;
119394b7f48cSBarry Smith     PC   pc;
1194f3ef73ceSBarry Smith 
11955a655dc6SBarry Smith     ierr = MatCreateSNESMF(snes,snes->vec_sol,&J);CHKERRQ(ierr);
11963a7fca6bSBarry Smith     ierr = MatSNESMFSetFromOptions(J);CHKERRQ(ierr);
1197ae15b995SBarry Smith     ierr = PetscInfo(snes,"Setting default matrix-free operator and preconditioner routines;\nThat is no preconditioner is being used.\n");CHKERRQ(ierr);
11983a7fca6bSBarry Smith     ierr = SNESSetJacobian(snes,J,J,MatSNESMFComputeJacobian,snes->funP);CHKERRQ(ierr);
11993a7fca6bSBarry Smith     ierr = MatDestroy(J);CHKERRQ(ierr);
12003a7fca6bSBarry Smith 
1201f3ef73ceSBarry Smith     /* force no preconditioner */
120294b7f48cSBarry Smith     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
1203b5d62d44SBarry Smith     ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
1204a9815358SBarry Smith     ierr = PetscTypeCompare((PetscObject)pc,PCSHELL,&flg);CHKERRQ(ierr);
1205a9815358SBarry Smith     if (!flg) {
1206f3ef73ceSBarry Smith       ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr);
1207272ac6f2SLois Curfman McInnes     }
1208a9815358SBarry Smith   }
1209f3ef73ceSBarry Smith 
12101096aae1SMatthew Knepley   if (!snes->vec_func && !snes->afine) {
12111096aae1SMatthew Knepley     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetFunction() first");
12121096aae1SMatthew Knepley   }
1213e7788613SBarry Smith   if (!snes->ops->computefunction && !snes->afine) {
12141096aae1SMatthew Knepley     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetFunction() first");
12151096aae1SMatthew Knepley   }
121629bbc08cSBarry Smith   if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetJacobian() first \n or use -snes_mf option");
1217a8c6a408SBarry Smith   if (snes->vec_func == snes->vec_sol) {
121829bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector");
1219a8c6a408SBarry Smith   }
1220a703fe33SLois Curfman McInnes 
1221410397dcSLisandro Dalcin   if (!snes->type_name) {
1222410397dcSLisandro Dalcin     ierr = SNESSetType(snes,SNESLS);CHKERRQ(ierr);
1223410397dcSLisandro Dalcin   }
122471f87433Sdalcinl 
1225410397dcSLisandro Dalcin   if (snes->ops->setup) {
1226410397dcSLisandro Dalcin     ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr);
1227410397dcSLisandro Dalcin   }
12287aaed0d8SBarry Smith   snes->setupcalled = PETSC_TRUE;
12293a40ed3dSBarry Smith   PetscFunctionReturn(0);
12309b94acceSBarry Smith }
12319b94acceSBarry Smith 
12324a2ae208SSatish Balay #undef __FUNCT__
12334a2ae208SSatish Balay #define __FUNCT__ "SNESDestroy"
123452baeb72SSatish Balay /*@
12359b94acceSBarry Smith    SNESDestroy - Destroys the nonlinear solver context that was created
12369b94acceSBarry Smith    with SNESCreate().
12379b94acceSBarry Smith 
1238c7afd0dbSLois Curfman McInnes    Collective on SNES
1239c7afd0dbSLois Curfman McInnes 
12409b94acceSBarry Smith    Input Parameter:
12419b94acceSBarry Smith .  snes - the SNES context
12429b94acceSBarry Smith 
124336851e7fSLois Curfman McInnes    Level: beginner
124436851e7fSLois Curfman McInnes 
12459b94acceSBarry Smith .keywords: SNES, nonlinear, destroy
12469b94acceSBarry Smith 
124763a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve()
12489b94acceSBarry Smith @*/
124963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESDestroy(SNES snes)
12509b94acceSBarry Smith {
12516849ba73SBarry Smith   PetscErrorCode ierr;
12523a40ed3dSBarry Smith 
12533a40ed3dSBarry Smith   PetscFunctionBegin;
12544482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
12553a40ed3dSBarry Smith   if (--snes->refct > 0) PetscFunctionReturn(0);
1256d4bb536fSBarry Smith 
1257be0abb6dSBarry Smith   /* if memory was published with AMS then destroy it */
12580f5bd95cSBarry Smith   ierr = PetscObjectDepublish(snes);CHKERRQ(ierr);
1259be0abb6dSBarry Smith 
1260e7788613SBarry Smith   if (snes->ops->destroy) {ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr);}
126105b42c5fSBarry Smith   ierr = PetscFree(snes->kspconvctx);CHKERRQ(ierr);
12623a7fca6bSBarry Smith   if (snes->jacobian) {ierr = MatDestroy(snes->jacobian);CHKERRQ(ierr);}
12633a7fca6bSBarry Smith   if (snes->jacobian_pre) {ierr = MatDestroy(snes->jacobian_pre);CHKERRQ(ierr);}
12643ab0aad5SBarry Smith   if (snes->afine) {ierr = VecDestroy(snes->afine);CHKERRQ(ierr);}
126594b7f48cSBarry Smith   ierr = KSPDestroy(snes->ksp);CHKERRQ(ierr);
1266522c5e43SBarry Smith   if (snes->vwork) {ierr = VecDestroyVecs(snes->vwork,snes->nvwork);CHKERRQ(ierr);}
1267a6570f20SBarry Smith   ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);
1268a79aaaedSSatish Balay   ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr);
12693a40ed3dSBarry Smith   PetscFunctionReturn(0);
12709b94acceSBarry Smith }
12719b94acceSBarry Smith 
12729b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */
12739b94acceSBarry Smith 
12744a2ae208SSatish Balay #undef __FUNCT__
12754a2ae208SSatish Balay #define __FUNCT__ "SNESSetTolerances"
12769b94acceSBarry Smith /*@
1277d7a720efSLois Curfman McInnes    SNESSetTolerances - Sets various parameters used in convergence tests.
12789b94acceSBarry Smith 
1279c7afd0dbSLois Curfman McInnes    Collective on SNES
1280c7afd0dbSLois Curfman McInnes 
12819b94acceSBarry Smith    Input Parameters:
1282c7afd0dbSLois Curfman McInnes +  snes - the SNES context
128370441072SBarry Smith .  abstol - absolute convergence tolerance
128433174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
128533174efeSLois Curfman McInnes .  stol -  convergence tolerance in terms of the norm
128633174efeSLois Curfman McInnes            of the change in the solution between steps
128733174efeSLois Curfman McInnes .  maxit - maximum number of iterations
1288c7afd0dbSLois Curfman McInnes -  maxf - maximum number of function evaluations
1289fee21e36SBarry Smith 
129033174efeSLois Curfman McInnes    Options Database Keys:
129170441072SBarry Smith +    -snes_atol <abstol> - Sets abstol
1292c7afd0dbSLois Curfman McInnes .    -snes_rtol <rtol> - Sets rtol
1293c7afd0dbSLois Curfman McInnes .    -snes_stol <stol> - Sets stol
1294c7afd0dbSLois Curfman McInnes .    -snes_max_it <maxit> - Sets maxit
1295c7afd0dbSLois Curfman McInnes -    -snes_max_funcs <maxf> - Sets maxf
12969b94acceSBarry Smith 
1297d7a720efSLois Curfman McInnes    Notes:
12989b94acceSBarry Smith    The default maximum number of iterations is 50.
12999b94acceSBarry Smith    The default maximum number of function evaluations is 1000.
13009b94acceSBarry Smith 
130136851e7fSLois Curfman McInnes    Level: intermediate
130236851e7fSLois Curfman McInnes 
130333174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances
13049b94acceSBarry Smith 
13052492ecdbSBarry Smith .seealso: SNESSetTrustRegionTolerance()
13069b94acceSBarry Smith @*/
130763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf)
13089b94acceSBarry Smith {
13093a40ed3dSBarry Smith   PetscFunctionBegin;
13104482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
131170441072SBarry Smith   if (abstol != PETSC_DEFAULT)  snes->abstol      = abstol;
1312d7a720efSLois Curfman McInnes   if (rtol != PETSC_DEFAULT)  snes->rtol      = rtol;
1313d7a720efSLois Curfman McInnes   if (stol != PETSC_DEFAULT)  snes->xtol      = stol;
1314d7a720efSLois Curfman McInnes   if (maxit != PETSC_DEFAULT) snes->max_its   = maxit;
1315d7a720efSLois Curfman McInnes   if (maxf != PETSC_DEFAULT)  snes->max_funcs = maxf;
13163a40ed3dSBarry Smith   PetscFunctionReturn(0);
13179b94acceSBarry Smith }
13189b94acceSBarry Smith 
13194a2ae208SSatish Balay #undef __FUNCT__
13204a2ae208SSatish Balay #define __FUNCT__ "SNESGetTolerances"
13219b94acceSBarry Smith /*@
132233174efeSLois Curfman McInnes    SNESGetTolerances - Gets various parameters used in convergence tests.
132333174efeSLois Curfman McInnes 
1324c7afd0dbSLois Curfman McInnes    Not Collective
1325c7afd0dbSLois Curfman McInnes 
132633174efeSLois Curfman McInnes    Input Parameters:
1327c7afd0dbSLois Curfman McInnes +  snes - the SNES context
132870441072SBarry Smith .  abstol - absolute convergence tolerance
132933174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
133033174efeSLois Curfman McInnes .  stol -  convergence tolerance in terms of the norm
133133174efeSLois Curfman McInnes            of the change in the solution between steps
133233174efeSLois Curfman McInnes .  maxit - maximum number of iterations
1333c7afd0dbSLois Curfman McInnes -  maxf - maximum number of function evaluations
1334fee21e36SBarry Smith 
133533174efeSLois Curfman McInnes    Notes:
133633174efeSLois Curfman McInnes    The user can specify PETSC_NULL for any parameter that is not needed.
133733174efeSLois Curfman McInnes 
133836851e7fSLois Curfman McInnes    Level: intermediate
133936851e7fSLois Curfman McInnes 
134033174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances
134133174efeSLois Curfman McInnes 
134233174efeSLois Curfman McInnes .seealso: SNESSetTolerances()
134333174efeSLois Curfman McInnes @*/
134463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetTolerances(SNES snes,PetscReal *abstol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf)
134533174efeSLois Curfman McInnes {
13463a40ed3dSBarry Smith   PetscFunctionBegin;
13474482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
134870441072SBarry Smith   if (abstol)  *abstol  = snes->abstol;
134933174efeSLois Curfman McInnes   if (rtol)  *rtol  = snes->rtol;
135033174efeSLois Curfman McInnes   if (stol)  *stol  = snes->xtol;
135133174efeSLois Curfman McInnes   if (maxit) *maxit = snes->max_its;
135233174efeSLois Curfman McInnes   if (maxf)  *maxf  = snes->max_funcs;
13533a40ed3dSBarry Smith   PetscFunctionReturn(0);
135433174efeSLois Curfman McInnes }
135533174efeSLois Curfman McInnes 
13564a2ae208SSatish Balay #undef __FUNCT__
13574a2ae208SSatish Balay #define __FUNCT__ "SNESSetTrustRegionTolerance"
135833174efeSLois Curfman McInnes /*@
13599b94acceSBarry Smith    SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance.
13609b94acceSBarry Smith 
1361fee21e36SBarry Smith    Collective on SNES
1362fee21e36SBarry Smith 
1363c7afd0dbSLois Curfman McInnes    Input Parameters:
1364c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1365c7afd0dbSLois Curfman McInnes -  tol - tolerance
1366c7afd0dbSLois Curfman McInnes 
13679b94acceSBarry Smith    Options Database Key:
1368c7afd0dbSLois Curfman McInnes .  -snes_trtol <tol> - Sets tol
13699b94acceSBarry Smith 
137036851e7fSLois Curfman McInnes    Level: intermediate
137136851e7fSLois Curfman McInnes 
13729b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance
13739b94acceSBarry Smith 
13742492ecdbSBarry Smith .seealso: SNESSetTolerances()
13759b94acceSBarry Smith @*/
137663dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetTrustRegionTolerance(SNES snes,PetscReal tol)
13779b94acceSBarry Smith {
13783a40ed3dSBarry Smith   PetscFunctionBegin;
13794482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
13809b94acceSBarry Smith   snes->deltatol = tol;
13813a40ed3dSBarry Smith   PetscFunctionReturn(0);
13829b94acceSBarry Smith }
13839b94acceSBarry Smith 
1384df9fa365SBarry Smith /*
1385df9fa365SBarry Smith    Duplicate the lg monitors for SNES from KSP; for some reason with
1386df9fa365SBarry Smith    dynamic libraries things don't work under Sun4 if we just use
1387df9fa365SBarry Smith    macros instead of functions
1388df9fa365SBarry Smith */
13894a2ae208SSatish Balay #undef __FUNCT__
1390a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorLG"
1391a6570f20SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorLG(SNES snes,PetscInt it,PetscReal norm,void *ctx)
1392ce1608b8SBarry Smith {
1393dfbe8321SBarry Smith   PetscErrorCode ierr;
1394ce1608b8SBarry Smith 
1395ce1608b8SBarry Smith   PetscFunctionBegin;
13964482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1397a6570f20SBarry Smith   ierr = KSPMonitorLG((KSP)snes,it,norm,ctx);CHKERRQ(ierr);
1398ce1608b8SBarry Smith   PetscFunctionReturn(0);
1399ce1608b8SBarry Smith }
1400ce1608b8SBarry Smith 
14014a2ae208SSatish Balay #undef __FUNCT__
1402a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorLGCreate"
1403a6570f20SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorLGCreate(const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *draw)
1404df9fa365SBarry Smith {
1405dfbe8321SBarry Smith   PetscErrorCode ierr;
1406df9fa365SBarry Smith 
1407df9fa365SBarry Smith   PetscFunctionBegin;
1408a6570f20SBarry Smith   ierr = KSPMonitorLGCreate(host,label,x,y,m,n,draw);CHKERRQ(ierr);
1409df9fa365SBarry Smith   PetscFunctionReturn(0);
1410df9fa365SBarry Smith }
1411df9fa365SBarry Smith 
14124a2ae208SSatish Balay #undef __FUNCT__
1413a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorLGDestroy"
1414a6570f20SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorLGDestroy(PetscDrawLG draw)
1415df9fa365SBarry Smith {
1416dfbe8321SBarry Smith   PetscErrorCode ierr;
1417df9fa365SBarry Smith 
1418df9fa365SBarry Smith   PetscFunctionBegin;
1419a6570f20SBarry Smith   ierr = KSPMonitorLGDestroy(draw);CHKERRQ(ierr);
1420df9fa365SBarry Smith   PetscFunctionReturn(0);
1421df9fa365SBarry Smith }
1422df9fa365SBarry Smith 
14239b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */
14249b94acceSBarry Smith 
14254a2ae208SSatish Balay #undef __FUNCT__
1426a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorSet"
14279b94acceSBarry Smith /*@C
1428a6570f20SBarry Smith    SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every
14299b94acceSBarry Smith    iteration of the nonlinear solver to display the iteration's
14309b94acceSBarry Smith    progress.
14319b94acceSBarry Smith 
1432fee21e36SBarry Smith    Collective on SNES
1433fee21e36SBarry Smith 
1434c7afd0dbSLois Curfman McInnes    Input Parameters:
1435c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1436c7afd0dbSLois Curfman McInnes .  func - monitoring routine
1437b8a78c4aSBarry Smith .  mctx - [optional] user-defined context for private data for the
1438e8105e01SRichard Katz           monitor routine (use PETSC_NULL if no context is desired)
1439b3006f0bSLois Curfman McInnes -  monitordestroy - [optional] routine that frees monitor context
1440b3006f0bSLois Curfman McInnes           (may be PETSC_NULL)
14419b94acceSBarry Smith 
1442c7afd0dbSLois Curfman McInnes    Calling sequence of func:
1443a7cc72afSBarry Smith $     int func(SNES snes,PetscInt its, PetscReal norm,void *mctx)
1444c7afd0dbSLois Curfman McInnes 
1445c7afd0dbSLois Curfman McInnes +    snes - the SNES context
1446c7afd0dbSLois Curfman McInnes .    its - iteration number
1447c7afd0dbSLois Curfman McInnes .    norm - 2-norm function value (may be estimated)
144840a0c1c6SLois Curfman McInnes -    mctx - [optional] monitoring context
14499b94acceSBarry Smith 
14509665c990SLois Curfman McInnes    Options Database Keys:
1451a6570f20SBarry Smith +    -snes_monitor        - sets SNESMonitorDefault()
1452a6570f20SBarry Smith .    -snes_monitor_draw    - sets line graph monitor,
1453a6570f20SBarry Smith                             uses SNESMonitorLGCreate()
1454a6570f20SBarry Smith _    -snes_monitor_cancel - cancels all monitors that have
1455c7afd0dbSLois Curfman McInnes                             been hardwired into a code by
1456a6570f20SBarry Smith                             calls to SNESMonitorSet(), but
1457c7afd0dbSLois Curfman McInnes                             does not cancel those set via
1458c7afd0dbSLois Curfman McInnes                             the options database.
14599665c990SLois Curfman McInnes 
1460639f9d9dSBarry Smith    Notes:
14616bc08f3fSLois Curfman McInnes    Several different monitoring routines may be set by calling
1462a6570f20SBarry Smith    SNESMonitorSet() multiple times; all will be called in the
14636bc08f3fSLois Curfman McInnes    order in which they were set.
1464639f9d9dSBarry Smith 
146536851e7fSLois Curfman McInnes    Level: intermediate
146636851e7fSLois Curfman McInnes 
14679b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor
14689b94acceSBarry Smith 
1469a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorCancel()
14709b94acceSBarry Smith @*/
1471b90d0a6eSBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorSet(SNES snes,PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void*))
14729b94acceSBarry Smith {
1473b90d0a6eSBarry Smith   PetscInt i;
1474b90d0a6eSBarry Smith 
14753a40ed3dSBarry Smith   PetscFunctionBegin;
14764482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1477639f9d9dSBarry Smith   if (snes->numbermonitors >= MAXSNESMONITORS) {
147829bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set");
1479639f9d9dSBarry Smith   }
1480b90d0a6eSBarry Smith   for (i=0; i<snes->numbermonitors;i++) {
1481b90d0a6eSBarry Smith     if (monitor == snes->monitor[i] && monitordestroy == snes->monitordestroy[i] && mctx == snes->monitorcontext[i]) PetscFunctionReturn(0);
1482b90d0a6eSBarry Smith 
1483b90d0a6eSBarry Smith     /* check if both default monitors that share common ASCII viewer */
1484b90d0a6eSBarry Smith     if (monitor == snes->monitor[i] && monitor == SNESMonitorDefault) {
1485b90d0a6eSBarry Smith       if (mctx && snes->monitorcontext[i]) {
1486b90d0a6eSBarry Smith         PetscErrorCode          ierr;
1487b90d0a6eSBarry Smith         PetscViewerASCIIMonitor viewer1 = (PetscViewerASCIIMonitor) mctx;
1488b90d0a6eSBarry Smith         PetscViewerASCIIMonitor viewer2 = (PetscViewerASCIIMonitor) snes->monitorcontext[i];
1489b90d0a6eSBarry Smith         if (viewer1->viewer == viewer2->viewer) {
1490b90d0a6eSBarry Smith           ierr = (*monitordestroy)(mctx);CHKERRQ(ierr);
1491b90d0a6eSBarry Smith           PetscFunctionReturn(0);
1492b90d0a6eSBarry Smith         }
1493b90d0a6eSBarry Smith       }
1494b90d0a6eSBarry Smith     }
1495b90d0a6eSBarry Smith   }
1496b90d0a6eSBarry Smith   snes->monitor[snes->numbermonitors]           = monitor;
1497b8a78c4aSBarry Smith   snes->monitordestroy[snes->numbermonitors]    = monitordestroy;
1498639f9d9dSBarry Smith   snes->monitorcontext[snes->numbermonitors++]  = (void*)mctx;
14993a40ed3dSBarry Smith   PetscFunctionReturn(0);
15009b94acceSBarry Smith }
15019b94acceSBarry Smith 
15024a2ae208SSatish Balay #undef __FUNCT__
1503a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorCancel"
15045cd90555SBarry Smith /*@C
1505a6570f20SBarry Smith    SNESMonitorCancel - Clears all the monitor functions for a SNES object.
15065cd90555SBarry Smith 
1507c7afd0dbSLois Curfman McInnes    Collective on SNES
1508c7afd0dbSLois Curfman McInnes 
15095cd90555SBarry Smith    Input Parameters:
15105cd90555SBarry Smith .  snes - the SNES context
15115cd90555SBarry Smith 
15121a480d89SAdministrator    Options Database Key:
1513a6570f20SBarry Smith .  -snes_monitor_cancel - cancels all monitors that have been hardwired
1514a6570f20SBarry Smith     into a code by calls to SNESMonitorSet(), but does not cancel those
1515c7afd0dbSLois Curfman McInnes     set via the options database
15165cd90555SBarry Smith 
15175cd90555SBarry Smith    Notes:
15185cd90555SBarry Smith    There is no way to clear one specific monitor from a SNES object.
15195cd90555SBarry Smith 
152036851e7fSLois Curfman McInnes    Level: intermediate
152136851e7fSLois Curfman McInnes 
15225cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor
15235cd90555SBarry Smith 
1524a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorSet()
15255cd90555SBarry Smith @*/
1526a6570f20SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorCancel(SNES snes)
15275cd90555SBarry Smith {
1528d952e501SBarry Smith   PetscErrorCode ierr;
1529d952e501SBarry Smith   PetscInt       i;
1530d952e501SBarry Smith 
15315cd90555SBarry Smith   PetscFunctionBegin;
15324482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1533d952e501SBarry Smith   for (i=0; i<snes->numbermonitors; i++) {
1534d952e501SBarry Smith     if (snes->monitordestroy[i]) {
1535d952e501SBarry Smith       ierr = (*snes->monitordestroy[i])(snes->monitorcontext[i]);CHKERRQ(ierr);
1536d952e501SBarry Smith     }
1537d952e501SBarry Smith   }
15385cd90555SBarry Smith   snes->numbermonitors = 0;
15395cd90555SBarry Smith   PetscFunctionReturn(0);
15405cd90555SBarry Smith }
15415cd90555SBarry Smith 
15424a2ae208SSatish Balay #undef __FUNCT__
15434a2ae208SSatish Balay #define __FUNCT__ "SNESSetConvergenceTest"
15449b94acceSBarry Smith /*@C
15459b94acceSBarry Smith    SNESSetConvergenceTest - Sets the function that is to be used
15469b94acceSBarry Smith    to test for convergence of the nonlinear iterative solution.
15479b94acceSBarry Smith 
1548fee21e36SBarry Smith    Collective on SNES
1549fee21e36SBarry Smith 
1550c7afd0dbSLois Curfman McInnes    Input Parameters:
1551c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1552c7afd0dbSLois Curfman McInnes .  func - routine to test for convergence
1553c7afd0dbSLois Curfman McInnes -  cctx - [optional] context for private data for the convergence routine
1554c7afd0dbSLois Curfman McInnes           (may be PETSC_NULL)
15559b94acceSBarry Smith 
1556c7afd0dbSLois Curfman McInnes    Calling sequence of func:
155706ee9f85SBarry Smith $     PetscErrorCode func (SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx)
1558c7afd0dbSLois Curfman McInnes 
1559c7afd0dbSLois Curfman McInnes +    snes - the SNES context
156006ee9f85SBarry Smith .    it - current iteration (0 is the first and is before any Newton step)
1561c7afd0dbSLois Curfman McInnes .    cctx - [optional] convergence context
1562184914b5SBarry Smith .    reason - reason for convergence/divergence
1563c7afd0dbSLois Curfman McInnes .    xnorm - 2-norm of current iterate
15644b27c08aSLois Curfman McInnes .    gnorm - 2-norm of current step
15654b27c08aSLois Curfman McInnes -    f - 2-norm of function
15669b94acceSBarry Smith 
156736851e7fSLois Curfman McInnes    Level: advanced
156836851e7fSLois Curfman McInnes 
15699b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test
15709b94acceSBarry Smith 
15713f149594SLisandro Dalcin .seealso: SNESDefaultConverged(), SNESSkipConverged(), SNESConverged_LS(), SNESConverged_TR()
15729b94acceSBarry Smith @*/
157306ee9f85SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetConvergenceTest(SNES snes,PetscErrorCode (*func)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx)
15749b94acceSBarry Smith {
15753a40ed3dSBarry Smith   PetscFunctionBegin;
15764482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1577e7788613SBarry Smith   (snes)->ops->converged = func;
15789b94acceSBarry Smith   (snes)->cnvP           = cctx;
15793a40ed3dSBarry Smith   PetscFunctionReturn(0);
15809b94acceSBarry Smith }
15819b94acceSBarry Smith 
15824a2ae208SSatish Balay #undef __FUNCT__
15834a2ae208SSatish Balay #define __FUNCT__ "SNESGetConvergedReason"
158452baeb72SSatish Balay /*@
1585184914b5SBarry Smith    SNESGetConvergedReason - Gets the reason the SNES iteration was stopped.
1586184914b5SBarry Smith 
1587184914b5SBarry Smith    Not Collective
1588184914b5SBarry Smith 
1589184914b5SBarry Smith    Input Parameter:
1590184914b5SBarry Smith .  snes - the SNES context
1591184914b5SBarry Smith 
1592184914b5SBarry Smith    Output Parameter:
1593e090d566SSatish Balay .  reason - negative value indicates diverged, positive value converged, see petscsnes.h or the
1594184914b5SBarry Smith             manual pages for the individual convergence tests for complete lists
1595184914b5SBarry Smith 
1596184914b5SBarry Smith    Level: intermediate
1597184914b5SBarry Smith 
1598184914b5SBarry Smith    Notes: Can only be called after the call the SNESSolve() is complete.
1599184914b5SBarry Smith 
1600184914b5SBarry Smith .keywords: SNES, nonlinear, set, convergence, test
1601184914b5SBarry Smith 
16024b27c08aSLois Curfman McInnes .seealso: SNESSetConvergenceTest(), SNESConverged_LS(), SNESConverged_TR(), SNESConvergedReason
1603184914b5SBarry Smith @*/
160463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason)
1605184914b5SBarry Smith {
1606184914b5SBarry Smith   PetscFunctionBegin;
16074482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
16084482741eSBarry Smith   PetscValidPointer(reason,2);
1609184914b5SBarry Smith   *reason = snes->reason;
1610184914b5SBarry Smith   PetscFunctionReturn(0);
1611184914b5SBarry Smith }
1612184914b5SBarry Smith 
16134a2ae208SSatish Balay #undef __FUNCT__
16144a2ae208SSatish Balay #define __FUNCT__ "SNESSetConvergenceHistory"
1615c9005455SLois Curfman McInnes /*@
1616c9005455SLois Curfman McInnes    SNESSetConvergenceHistory - Sets the array used to hold the convergence history.
1617c9005455SLois Curfman McInnes 
1618fee21e36SBarry Smith    Collective on SNES
1619fee21e36SBarry Smith 
1620c7afd0dbSLois Curfman McInnes    Input Parameters:
1621c7afd0dbSLois Curfman McInnes +  snes - iterative context obtained from SNESCreate()
1622c7afd0dbSLois Curfman McInnes .  a   - array to hold history
1623cd5578b5SBarry Smith .  its - integer array holds the number of linear iterations for each solve.
1624758f92a0SBarry Smith .  na  - size of a and its
162564731454SLois Curfman McInnes -  reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero,
1626758f92a0SBarry Smith            else it continues storing new values for new nonlinear solves after the old ones
1627c7afd0dbSLois Curfman McInnes 
1628c9005455SLois Curfman McInnes    Notes:
16294b27c08aSLois Curfman McInnes    If set, this array will contain the function norms computed
1630c9005455SLois Curfman McInnes    at each step.
1631c9005455SLois Curfman McInnes 
1632c9005455SLois Curfman McInnes    This routine is useful, e.g., when running a code for purposes
1633c9005455SLois Curfman McInnes    of accurate performance monitoring, when no I/O should be done
1634c9005455SLois Curfman McInnes    during the section of code that is being timed.
1635c9005455SLois Curfman McInnes 
163636851e7fSLois Curfman McInnes    Level: intermediate
163736851e7fSLois Curfman McInnes 
1638c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history
1639758f92a0SBarry Smith 
164008405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory()
1641758f92a0SBarry Smith 
1642c9005455SLois Curfman McInnes @*/
1643a562a398SLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscTruth reset)
1644c9005455SLois Curfman McInnes {
16453a40ed3dSBarry Smith   PetscFunctionBegin;
16464482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
16474482741eSBarry Smith   if (na)  PetscValidScalarPointer(a,2);
1648a562a398SLisandro Dalcin   if (its) PetscValidIntPointer(its,3);
1649c9005455SLois Curfman McInnes   snes->conv_hist       = a;
1650758f92a0SBarry Smith   snes->conv_hist_its   = its;
1651758f92a0SBarry Smith   snes->conv_hist_max   = na;
1652758f92a0SBarry Smith   snes->conv_hist_reset = reset;
1653758f92a0SBarry Smith   PetscFunctionReturn(0);
1654758f92a0SBarry Smith }
1655758f92a0SBarry Smith 
16564a2ae208SSatish Balay #undef __FUNCT__
16574a2ae208SSatish Balay #define __FUNCT__ "SNESGetConvergenceHistory"
16580c4c9dddSBarry Smith /*@C
1659758f92a0SBarry Smith    SNESGetConvergenceHistory - Gets the array used to hold the convergence history.
1660758f92a0SBarry Smith 
1661758f92a0SBarry Smith    Collective on SNES
1662758f92a0SBarry Smith 
1663758f92a0SBarry Smith    Input Parameter:
1664758f92a0SBarry Smith .  snes - iterative context obtained from SNESCreate()
1665758f92a0SBarry Smith 
1666758f92a0SBarry Smith    Output Parameters:
1667758f92a0SBarry Smith .  a   - array to hold history
1668758f92a0SBarry Smith .  its - integer array holds the number of linear iterations (or
1669758f92a0SBarry Smith          negative if not converged) for each solve.
1670758f92a0SBarry Smith -  na  - size of a and its
1671758f92a0SBarry Smith 
1672758f92a0SBarry Smith    Notes:
1673758f92a0SBarry Smith     The calling sequence for this routine in Fortran is
1674758f92a0SBarry Smith $   call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr)
1675758f92a0SBarry Smith 
1676758f92a0SBarry Smith    This routine is useful, e.g., when running a code for purposes
1677758f92a0SBarry Smith    of accurate performance monitoring, when no I/O should be done
1678758f92a0SBarry Smith    during the section of code that is being timed.
1679758f92a0SBarry Smith 
1680758f92a0SBarry Smith    Level: intermediate
1681758f92a0SBarry Smith 
1682758f92a0SBarry Smith .keywords: SNES, get, convergence, history
1683758f92a0SBarry Smith 
1684758f92a0SBarry Smith .seealso: SNESSetConvergencHistory()
1685758f92a0SBarry Smith 
1686758f92a0SBarry Smith @*/
168763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na)
1688758f92a0SBarry Smith {
1689758f92a0SBarry Smith   PetscFunctionBegin;
16904482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1691758f92a0SBarry Smith   if (a)   *a   = snes->conv_hist;
1692758f92a0SBarry Smith   if (its) *its = snes->conv_hist_its;
1693758f92a0SBarry Smith   if (na)  *na  = snes->conv_hist_len;
16943a40ed3dSBarry Smith   PetscFunctionReturn(0);
1695c9005455SLois Curfman McInnes }
1696c9005455SLois Curfman McInnes 
1697e74ef692SMatthew Knepley #undef __FUNCT__
1698e74ef692SMatthew Knepley #define __FUNCT__ "SNESSetUpdate"
1699ac226902SBarry Smith /*@C
170076b2cf59SMatthew Knepley   SNESSetUpdate - Sets the general-purpose update function called
17017e4bb74cSBarry Smith   at the beginning o every iteration of the nonlinear solve. Specifically
17027e4bb74cSBarry Smith   it is called just before the Jacobian is "evaluated".
170376b2cf59SMatthew Knepley 
170476b2cf59SMatthew Knepley   Collective on SNES
170576b2cf59SMatthew Knepley 
170676b2cf59SMatthew Knepley   Input Parameters:
170776b2cf59SMatthew Knepley . snes - The nonlinear solver context
170876b2cf59SMatthew Knepley . func - The function
170976b2cf59SMatthew Knepley 
171076b2cf59SMatthew Knepley   Calling sequence of func:
1711b5d30489SBarry Smith . func (SNES snes, PetscInt step);
171276b2cf59SMatthew Knepley 
171376b2cf59SMatthew Knepley . step - The current step of the iteration
171476b2cf59SMatthew Knepley 
171576b2cf59SMatthew Knepley   Level: intermediate
171676b2cf59SMatthew Knepley 
171776b2cf59SMatthew Knepley .keywords: SNES, update
1718b5d30489SBarry Smith 
171976b2cf59SMatthew Knepley .seealso SNESDefaultUpdate(), SNESSetRhsBC(), SNESSetSolutionBC()
172076b2cf59SMatthew Knepley @*/
172163dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt))
172276b2cf59SMatthew Knepley {
172376b2cf59SMatthew Knepley   PetscFunctionBegin;
17244482741eSBarry Smith   PetscValidHeaderSpecific(snes, SNES_COOKIE,1);
1725e7788613SBarry Smith   snes->ops->update = func;
172676b2cf59SMatthew Knepley   PetscFunctionReturn(0);
172776b2cf59SMatthew Knepley }
172876b2cf59SMatthew Knepley 
1729e74ef692SMatthew Knepley #undef __FUNCT__
1730e74ef692SMatthew Knepley #define __FUNCT__ "SNESDefaultUpdate"
173176b2cf59SMatthew Knepley /*@
173276b2cf59SMatthew Knepley   SNESDefaultUpdate - The default update function which does nothing.
173376b2cf59SMatthew Knepley 
173476b2cf59SMatthew Knepley   Not collective
173576b2cf59SMatthew Knepley 
173676b2cf59SMatthew Knepley   Input Parameters:
173776b2cf59SMatthew Knepley . snes - The nonlinear solver context
173876b2cf59SMatthew Knepley . step - The current step of the iteration
173976b2cf59SMatthew Knepley 
1740205452f4SMatthew Knepley   Level: intermediate
1741205452f4SMatthew Knepley 
174276b2cf59SMatthew Knepley .keywords: SNES, update
1743a6570f20SBarry Smith .seealso SNESSetUpdate(), SNESDefaultRhsBC(), SNESDefaultShortolutionBC()
174476b2cf59SMatthew Knepley @*/
174563dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESDefaultUpdate(SNES snes, PetscInt step)
174676b2cf59SMatthew Knepley {
174776b2cf59SMatthew Knepley   PetscFunctionBegin;
174876b2cf59SMatthew Knepley   PetscFunctionReturn(0);
174976b2cf59SMatthew Knepley }
175076b2cf59SMatthew Knepley 
17514a2ae208SSatish Balay #undef __FUNCT__
17524a2ae208SSatish Balay #define __FUNCT__ "SNESScaleStep_Private"
17539b94acceSBarry Smith /*
17549b94acceSBarry Smith    SNESScaleStep_Private - Scales a step so that its length is less than the
17559b94acceSBarry Smith    positive parameter delta.
17569b94acceSBarry Smith 
17579b94acceSBarry Smith     Input Parameters:
1758c7afd0dbSLois Curfman McInnes +   snes - the SNES context
17599b94acceSBarry Smith .   y - approximate solution of linear system
17609b94acceSBarry Smith .   fnorm - 2-norm of current function
1761c7afd0dbSLois Curfman McInnes -   delta - trust region size
17629b94acceSBarry Smith 
17639b94acceSBarry Smith     Output Parameters:
1764c7afd0dbSLois Curfman McInnes +   gpnorm - predicted function norm at the new point, assuming local
17659b94acceSBarry Smith     linearization.  The value is zero if the step lies within the trust
17669b94acceSBarry Smith     region, and exceeds zero otherwise.
1767c7afd0dbSLois Curfman McInnes -   ynorm - 2-norm of the step
17689b94acceSBarry Smith 
17699b94acceSBarry Smith     Note:
17704b27c08aSLois Curfman McInnes     For non-trust region methods such as SNESLS, the parameter delta
17719b94acceSBarry Smith     is set to be the maximum allowable step size.
17729b94acceSBarry Smith 
17739b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step
17749b94acceSBarry Smith */
1775dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm)
17769b94acceSBarry Smith {
1777064f8208SBarry Smith   PetscReal      nrm;
1778ea709b57SSatish Balay   PetscScalar    cnorm;
1779dfbe8321SBarry Smith   PetscErrorCode ierr;
17803a40ed3dSBarry Smith 
17813a40ed3dSBarry Smith   PetscFunctionBegin;
17824482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
17834482741eSBarry Smith   PetscValidHeaderSpecific(y,VEC_COOKIE,2);
1784c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,2);
1785184914b5SBarry Smith 
1786064f8208SBarry Smith   ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr);
1787064f8208SBarry Smith   if (nrm > *delta) {
1788064f8208SBarry Smith      nrm = *delta/nrm;
1789064f8208SBarry Smith      *gpnorm = (1.0 - nrm)*(*fnorm);
1790064f8208SBarry Smith      cnorm = nrm;
17912dcb1b2aSMatthew Knepley      ierr = VecScale(y,cnorm);CHKERRQ(ierr);
17929b94acceSBarry Smith      *ynorm = *delta;
17939b94acceSBarry Smith   } else {
17949b94acceSBarry Smith      *gpnorm = 0.0;
1795064f8208SBarry Smith      *ynorm = nrm;
17969b94acceSBarry Smith   }
17973a40ed3dSBarry Smith   PetscFunctionReturn(0);
17989b94acceSBarry Smith }
17999b94acceSBarry Smith 
18004a2ae208SSatish Balay #undef __FUNCT__
18014a2ae208SSatish Balay #define __FUNCT__ "SNESSolve"
18026ce558aeSBarry Smith /*@C
1803f69a0ea3SMatthew Knepley    SNESSolve - Solves a nonlinear system F(x) = b.
1804f69a0ea3SMatthew Knepley    Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX().
18059b94acceSBarry Smith 
1806c7afd0dbSLois Curfman McInnes    Collective on SNES
1807c7afd0dbSLois Curfman McInnes 
1808b2002411SLois Curfman McInnes    Input Parameters:
1809c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1810f69a0ea3SMatthew Knepley .  b - the constant part of the equation, or PETSC_NULL to use zero.
181170e92668SMatthew Knepley -  x - the solution vector, or PETSC_NULL if it was set with SNESSetSolution()
18129b94acceSBarry Smith 
1813b2002411SLois Curfman McInnes    Notes:
18148ddd3da0SLois Curfman McInnes    The user should initialize the vector,x, with the initial guess
18158ddd3da0SLois Curfman McInnes    for the nonlinear solve prior to calling SNESSolve.  In particular,
18168ddd3da0SLois Curfman McInnes    to employ an initial guess of zero, the user should explicitly set
18178ddd3da0SLois Curfman McInnes    this vector to zero by calling VecSet().
18188ddd3da0SLois Curfman McInnes 
181936851e7fSLois Curfman McInnes    Level: beginner
182036851e7fSLois Curfman McInnes 
18219b94acceSBarry Smith .keywords: SNES, nonlinear, solve
18229b94acceSBarry Smith 
182370e92668SMatthew Knepley .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetRhs(), SNESSetSolution()
18249b94acceSBarry Smith @*/
1825f69a0ea3SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESSolve(SNES snes,Vec b,Vec x)
18269b94acceSBarry Smith {
1827dfbe8321SBarry Smith   PetscErrorCode ierr;
1828f1af5d2fSBarry Smith   PetscTruth     flg;
1829eabae89aSBarry Smith   char           filename[PETSC_MAX_PATH_LEN];
1830eabae89aSBarry Smith   PetscViewer    viewer;
1831052efed2SBarry Smith 
18323a40ed3dSBarry Smith   PetscFunctionBegin;
18334482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1834e7788613SBarry Smith   if (!snes->ops->solve) SETERRQ(PETSC_ERR_ORDER,"SNESSetType() or SNESSetFromOptions() must be called before SNESSolve()");
183574637425SBarry Smith 
1836f69a0ea3SMatthew Knepley   if (b) {
1837f69a0ea3SMatthew Knepley     ierr = SNESSetRhs(snes, b); CHKERRQ(ierr);
18381096aae1SMatthew Knepley     if (!snes->vec_func) {
18391096aae1SMatthew Knepley       Vec r;
18401096aae1SMatthew Knepley 
18411096aae1SMatthew Knepley       ierr = VecDuplicate(b, &r); CHKERRQ(ierr);
18421096aae1SMatthew Knepley       ierr = SNESSetFunction(snes, r, PETSC_NULL, PETSC_NULL); CHKERRQ(ierr);
18431096aae1SMatthew Knepley     }
1844f69a0ea3SMatthew Knepley   }
184570e92668SMatthew Knepley   if (x) {
1846f69a0ea3SMatthew Knepley     PetscValidHeaderSpecific(x,VEC_COOKIE,3);
1847f69a0ea3SMatthew Knepley     PetscCheckSameComm(snes,1,x,3);
184870e92668SMatthew Knepley   } else {
184970e92668SMatthew Knepley     ierr = SNESGetSolution(snes, &x); CHKERRQ(ierr);
185070e92668SMatthew Knepley     if (!x) {
185170e92668SMatthew Knepley       ierr = VecDuplicate(snes->vec_func_always, &x); CHKERRQ(ierr);
185270e92668SMatthew Knepley     }
185370e92668SMatthew Knepley   }
185470e92668SMatthew Knepley   snes->vec_sol = snes->vec_sol_always = x;
18553f149594SLisandro Dalcin 
185670e92668SMatthew Knepley   ierr = SNESSetUp(snes);CHKERRQ(ierr);
18573f149594SLisandro Dalcin 
1858abc0a331SBarry Smith   if (snes->conv_hist_reset) snes->conv_hist_len = 0;
185950ffb88aSMatthew Knepley   snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;
1860d5e45103SBarry Smith 
18613f149594SLisandro Dalcin   ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
18623f149594SLisandro Dalcin 
1863e7788613SBarry Smith   ierr = PetscExceptionTry1((*(snes)->ops->solve)(snes),PETSC_ERR_ARG_DOMAIN);
1864d5e45103SBarry Smith   if (PetscExceptionValue(ierr)) {
186538f152feSBarry Smith     /* this means that a caller above me has also tryed this exception so I don't handle it here, pass it up */
186619717074SBarry Smith     PetscErrorCode pierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(pierr);
1867c671dbe7SSatish Balay   } else if (PetscExceptionCaught(ierr,PETSC_ERR_ARG_DOMAIN)) {
1868d5e45103SBarry Smith     /* translate exception into SNES not converged reason */
1869d5e45103SBarry Smith     snes->reason = SNES_DIVERGED_FUNCTION_DOMAIN;
187038f152feSBarry Smith     ierr = 0;
187138f152feSBarry Smith   }
187238f152feSBarry Smith   CHKERRQ(ierr);
18733f149594SLisandro Dalcin   if (!snes->reason) {
18743f149594SLisandro Dalcin     SETERRQ(PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason");
18753f149594SLisandro Dalcin   }
18763f149594SLisandro Dalcin   if (!snes->ops->converged && snes->reason == SNES_DIVERGED_MAX_IT) {
18773f149594SLisandro Dalcin     snes->reason = SNES_CONVERGED_ITS;
18783f149594SLisandro Dalcin   }
1879d5ba7fb7SMatthew Knepley   ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
18803f149594SLisandro Dalcin 
1881eabae89aSBarry Smith   ierr = PetscOptionsGetString(snes->prefix,"-snes_view",filename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
1882eabae89aSBarry Smith   if (flg && !PetscPreLoadingOn) {
1883eabae89aSBarry Smith     ierr = PetscViewerASCIIOpen(snes->comm,filename,&viewer);CHKERRQ(ierr);
1884eabae89aSBarry Smith     ierr = SNESView(snes,viewer);CHKERRQ(ierr);
1885eabae89aSBarry Smith     ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr);
1886eabae89aSBarry Smith   }
1887eabae89aSBarry Smith 
1888da9b6338SBarry Smith   ierr = PetscOptionsHasName(snes->prefix,"-snes_test_local_min",&flg);CHKERRQ(ierr);
1889da9b6338SBarry Smith   if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); }
18905968eb51SBarry Smith   if (snes->printreason) {
18915968eb51SBarry Smith     if (snes->reason > 0) {
18929dcbbd2bSBarry Smith       ierr = PetscPrintf(snes->comm,"Nonlinear solve converged due to %s\n",SNESConvergedReasons[snes->reason]);CHKERRQ(ierr);
18935968eb51SBarry Smith     } else {
18949dcbbd2bSBarry Smith       ierr = PetscPrintf(snes->comm,"Nonlinear solve did not converge due to %s\n",SNESConvergedReasons[snes->reason]);CHKERRQ(ierr);
18955968eb51SBarry Smith     }
18965968eb51SBarry Smith   }
18975968eb51SBarry Smith 
18983a40ed3dSBarry Smith   PetscFunctionReturn(0);
18999b94acceSBarry Smith }
19009b94acceSBarry Smith 
19019b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */
19029b94acceSBarry Smith 
19034a2ae208SSatish Balay #undef __FUNCT__
19044a2ae208SSatish Balay #define __FUNCT__ "SNESSetType"
190582bf6240SBarry Smith /*@C
19064b0e389bSBarry Smith    SNESSetType - Sets the method for the nonlinear solver.
19079b94acceSBarry Smith 
1908fee21e36SBarry Smith    Collective on SNES
1909fee21e36SBarry Smith 
1910c7afd0dbSLois Curfman McInnes    Input Parameters:
1911c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1912454a90a3SBarry Smith -  type - a known method
1913c7afd0dbSLois Curfman McInnes 
1914c7afd0dbSLois Curfman McInnes    Options Database Key:
1915454a90a3SBarry Smith .  -snes_type <type> - Sets the method; use -help for a list
1916c7afd0dbSLois Curfman McInnes    of available methods (for instance, ls or tr)
1917ae12b187SLois Curfman McInnes 
19189b94acceSBarry Smith    Notes:
1919e090d566SSatish Balay    See "petsc/include/petscsnes.h" for available methods (for instance)
19204b27c08aSLois Curfman McInnes +    SNESLS - Newton's method with line search
1921c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
19224b27c08aSLois Curfman McInnes .    SNESTR - Newton's method with trust region
1923c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
19249b94acceSBarry Smith 
1925ae12b187SLois Curfman McInnes   Normally, it is best to use the SNESSetFromOptions() command and then
1926ae12b187SLois Curfman McInnes   set the SNES solver type from the options database rather than by using
1927ae12b187SLois Curfman McInnes   this routine.  Using the options database provides the user with
1928ae12b187SLois Curfman McInnes   maximum flexibility in evaluating the many nonlinear solvers.
1929ae12b187SLois Curfman McInnes   The SNESSetType() routine is provided for those situations where it
1930ae12b187SLois Curfman McInnes   is necessary to set the nonlinear solver independently of the command
1931ae12b187SLois Curfman McInnes   line or options database.  This might be the case, for example, when
1932ae12b187SLois Curfman McInnes   the choice of solver changes during the execution of the program,
1933ae12b187SLois Curfman McInnes   and the user's application is taking responsibility for choosing the
1934b0a32e0cSBarry Smith   appropriate method.
193536851e7fSLois Curfman McInnes 
193636851e7fSLois Curfman McInnes   Level: intermediate
1937a703fe33SLois Curfman McInnes 
1938454a90a3SBarry Smith .keywords: SNES, set, type
1939435da068SBarry Smith 
1940435da068SBarry Smith .seealso: SNESType, SNESCreate()
1941435da068SBarry Smith 
19429b94acceSBarry Smith @*/
1943e060cb09SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetType(SNES snes,SNESType type)
19449b94acceSBarry Smith {
1945dfbe8321SBarry Smith   PetscErrorCode ierr,(*r)(SNES);
19466831982aSBarry Smith   PetscTruth     match;
19473a40ed3dSBarry Smith 
19483a40ed3dSBarry Smith   PetscFunctionBegin;
19494482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
19504482741eSBarry Smith   PetscValidCharPointer(type,2);
195182bf6240SBarry Smith 
19526831982aSBarry Smith   ierr = PetscTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr);
19530f5bd95cSBarry Smith   if (match) PetscFunctionReturn(0);
195482bf6240SBarry Smith 
195582bf6240SBarry Smith   if (snes->setupcalled) {
19564dc4c822SBarry Smith     snes->setupcalled = PETSC_FALSE;
1957e7788613SBarry Smith     ierr              = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr);
195882bf6240SBarry Smith     snes->data        = 0;
195982bf6240SBarry Smith   }
19607d1a2b2bSBarry Smith 
19619b94acceSBarry Smith   /* Get the function pointers for the iterative method requested */
196282bf6240SBarry Smith   if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL);CHKERRQ(ierr);}
1963b9617806SBarry Smith   ierr =  PetscFListFind(snes->comm,SNESList,type,(void (**)(void)) &r);CHKERRQ(ierr);
1964958c9bccSBarry Smith   if (!r) SETERRQ1(PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type);
196505b42c5fSBarry Smith   ierr = PetscFree(snes->data);CHKERRQ(ierr);
196682bf6240SBarry Smith   snes->data = 0;
19673a40ed3dSBarry Smith   ierr = (*r)(snes);CHKERRQ(ierr);
1968454a90a3SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr);
19693a40ed3dSBarry Smith   PetscFunctionReturn(0);
19709b94acceSBarry Smith }
19719b94acceSBarry Smith 
1972a847f771SSatish Balay 
19739b94acceSBarry Smith /* --------------------------------------------------------------------- */
19744a2ae208SSatish Balay #undef __FUNCT__
19754a2ae208SSatish Balay #define __FUNCT__ "SNESRegisterDestroy"
197652baeb72SSatish Balay /*@
19779b94acceSBarry Smith    SNESRegisterDestroy - Frees the list of nonlinear solvers that were
1978f1af5d2fSBarry Smith    registered by SNESRegisterDynamic().
19799b94acceSBarry Smith 
1980fee21e36SBarry Smith    Not Collective
1981fee21e36SBarry Smith 
198236851e7fSLois Curfman McInnes    Level: advanced
198336851e7fSLois Curfman McInnes 
19849b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy
19859b94acceSBarry Smith 
19869b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll()
19879b94acceSBarry Smith @*/
198863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESRegisterDestroy(void)
19899b94acceSBarry Smith {
1990dfbe8321SBarry Smith   PetscErrorCode ierr;
199182bf6240SBarry Smith 
19923a40ed3dSBarry Smith   PetscFunctionBegin;
199382bf6240SBarry Smith   if (SNESList) {
1994b0a32e0cSBarry Smith     ierr = PetscFListDestroy(&SNESList);CHKERRQ(ierr);
199582bf6240SBarry Smith     SNESList = 0;
19969b94acceSBarry Smith   }
19974c49b128SBarry Smith   SNESRegisterAllCalled = PETSC_FALSE;
19983a40ed3dSBarry Smith   PetscFunctionReturn(0);
19999b94acceSBarry Smith }
20009b94acceSBarry Smith 
20014a2ae208SSatish Balay #undef __FUNCT__
20024a2ae208SSatish Balay #define __FUNCT__ "SNESGetType"
20039b94acceSBarry Smith /*@C
20049a28b0a6SLois Curfman McInnes    SNESGetType - Gets the SNES method type and name (as a string).
20059b94acceSBarry Smith 
2006c7afd0dbSLois Curfman McInnes    Not Collective
2007c7afd0dbSLois Curfman McInnes 
20089b94acceSBarry Smith    Input Parameter:
20094b0e389bSBarry Smith .  snes - nonlinear solver context
20109b94acceSBarry Smith 
20119b94acceSBarry Smith    Output Parameter:
20123a7fca6bSBarry Smith .  type - SNES method (a character string)
20139b94acceSBarry Smith 
201436851e7fSLois Curfman McInnes    Level: intermediate
201536851e7fSLois Curfman McInnes 
2016454a90a3SBarry Smith .keywords: SNES, nonlinear, get, type, name
20179b94acceSBarry Smith @*/
201863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetType(SNES snes,SNESType *type)
20199b94acceSBarry Smith {
20203a40ed3dSBarry Smith   PetscFunctionBegin;
20214482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
20224482741eSBarry Smith   PetscValidPointer(type,2);
2023454a90a3SBarry Smith   *type = snes->type_name;
20243a40ed3dSBarry Smith   PetscFunctionReturn(0);
20259b94acceSBarry Smith }
20269b94acceSBarry Smith 
20274a2ae208SSatish Balay #undef __FUNCT__
20284a2ae208SSatish Balay #define __FUNCT__ "SNESGetSolution"
202952baeb72SSatish Balay /*@
20309b94acceSBarry Smith    SNESGetSolution - Returns the vector where the approximate solution is
20319b94acceSBarry Smith    stored.
20329b94acceSBarry Smith 
2033c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
2034c7afd0dbSLois Curfman McInnes 
20359b94acceSBarry Smith    Input Parameter:
20369b94acceSBarry Smith .  snes - the SNES context
20379b94acceSBarry Smith 
20389b94acceSBarry Smith    Output Parameter:
20399b94acceSBarry Smith .  x - the solution
20409b94acceSBarry Smith 
204170e92668SMatthew Knepley    Level: intermediate
204236851e7fSLois Curfman McInnes 
20439b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution
20449b94acceSBarry Smith 
204570e92668SMatthew Knepley .seealso: SNESSetSolution(), SNESGetFunction(), SNESGetSolutionUpdate()
20469b94acceSBarry Smith @*/
204763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetSolution(SNES snes,Vec *x)
20489b94acceSBarry Smith {
20493a40ed3dSBarry Smith   PetscFunctionBegin;
20504482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
20514482741eSBarry Smith   PetscValidPointer(x,2);
20529b94acceSBarry Smith   *x = snes->vec_sol_always;
20533a40ed3dSBarry Smith   PetscFunctionReturn(0);
20549b94acceSBarry Smith }
20559b94acceSBarry Smith 
20564a2ae208SSatish Balay #undef __FUNCT__
205770e92668SMatthew Knepley #define __FUNCT__ "SNESSetSolution"
20587e4bb74cSBarry Smith /*@
205970e92668SMatthew Knepley    SNESSetSolution - Sets the vector where the approximate solution is stored.
206070e92668SMatthew Knepley 
206170e92668SMatthew Knepley    Not Collective, but Vec is parallel if SNES is parallel
206270e92668SMatthew Knepley 
206370e92668SMatthew Knepley    Input Parameters:
206470e92668SMatthew Knepley +  snes - the SNES context
206570e92668SMatthew Knepley -  x - the solution
206670e92668SMatthew Knepley 
206770e92668SMatthew Knepley    Output Parameter:
206870e92668SMatthew Knepley 
206970e92668SMatthew Knepley    Level: intermediate
207070e92668SMatthew Knepley 
207142219521SBarry Smith    Notes: this is not normally used, rather one simply calls SNESSolve() with
207242219521SBarry Smith           the appropriate solution vector.
207342219521SBarry Smith 
207470e92668SMatthew Knepley .keywords: SNES, nonlinear, set, solution
207570e92668SMatthew Knepley 
207670e92668SMatthew Knepley .seealso: SNESGetSolution(), SNESGetFunction(), SNESGetSolutionUpdate()
207770e92668SMatthew Knepley @*/
207870e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESSetSolution(SNES snes,Vec x)
207970e92668SMatthew Knepley {
208070e92668SMatthew Knepley   PetscFunctionBegin;
208170e92668SMatthew Knepley   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
208270e92668SMatthew Knepley   PetscValidHeaderSpecific(x,VEC_COOKIE,2);
208370e92668SMatthew Knepley   PetscCheckSameComm(snes,1,x,2);
208470e92668SMatthew Knepley   snes->vec_sol_always = x;
208570e92668SMatthew Knepley   PetscFunctionReturn(0);
208670e92668SMatthew Knepley }
208770e92668SMatthew Knepley 
208870e92668SMatthew Knepley #undef __FUNCT__
20894a2ae208SSatish Balay #define __FUNCT__ "SNESGetSolutionUpdate"
209052baeb72SSatish Balay /*@
20919b94acceSBarry Smith    SNESGetSolutionUpdate - Returns the vector where the solution update is
20929b94acceSBarry Smith    stored.
20939b94acceSBarry Smith 
2094c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
2095c7afd0dbSLois Curfman McInnes 
20969b94acceSBarry Smith    Input Parameter:
20979b94acceSBarry Smith .  snes - the SNES context
20989b94acceSBarry Smith 
20999b94acceSBarry Smith    Output Parameter:
21009b94acceSBarry Smith .  x - the solution update
21019b94acceSBarry Smith 
210236851e7fSLois Curfman McInnes    Level: advanced
210336851e7fSLois Curfman McInnes 
21049b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update
21059b94acceSBarry Smith 
21069b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction
21079b94acceSBarry Smith @*/
210863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetSolutionUpdate(SNES snes,Vec *x)
21099b94acceSBarry Smith {
21103a40ed3dSBarry Smith   PetscFunctionBegin;
21114482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
21124482741eSBarry Smith   PetscValidPointer(x,2);
21139b94acceSBarry Smith   *x = snes->vec_sol_update_always;
21143a40ed3dSBarry Smith   PetscFunctionReturn(0);
21159b94acceSBarry Smith }
21169b94acceSBarry Smith 
21174a2ae208SSatish Balay #undef __FUNCT__
21184a2ae208SSatish Balay #define __FUNCT__ "SNESGetFunction"
21199b94acceSBarry Smith /*@C
21203638b69dSLois Curfman McInnes    SNESGetFunction - Returns the vector where the function is stored.
21219b94acceSBarry Smith 
2122c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
2123c7afd0dbSLois Curfman McInnes 
21249b94acceSBarry Smith    Input Parameter:
21259b94acceSBarry Smith .  snes - the SNES context
21269b94acceSBarry Smith 
21279b94acceSBarry Smith    Output Parameter:
21287bf4e008SBarry Smith +  r - the function (or PETSC_NULL)
212970e92668SMatthew Knepley .  func - the function (or PETSC_NULL)
213070e92668SMatthew Knepley -  ctx - the function context (or PETSC_NULL)
21319b94acceSBarry Smith 
213236851e7fSLois Curfman McInnes    Level: advanced
213336851e7fSLois Curfman McInnes 
2134a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function
21359b94acceSBarry Smith 
21364b27c08aSLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution()
21379b94acceSBarry Smith @*/
213870e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**func)(SNES,Vec,Vec,void*),void **ctx)
21399b94acceSBarry Smith {
21403a40ed3dSBarry Smith   PetscFunctionBegin;
21414482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
21427bf4e008SBarry Smith   if (r)    *r    = snes->vec_func_always;
2143e7788613SBarry Smith   if (func) *func = snes->ops->computefunction;
214470e92668SMatthew Knepley   if (ctx)  *ctx  = snes->funP;
21453a40ed3dSBarry Smith   PetscFunctionReturn(0);
21469b94acceSBarry Smith }
21479b94acceSBarry Smith 
21484a2ae208SSatish Balay #undef __FUNCT__
21494a2ae208SSatish Balay #define __FUNCT__ "SNESSetOptionsPrefix"
21503c7409f5SSatish Balay /*@C
21513c7409f5SSatish Balay    SNESSetOptionsPrefix - Sets the prefix used for searching for all
2152d850072dSLois Curfman McInnes    SNES options in the database.
21533c7409f5SSatish Balay 
2154fee21e36SBarry Smith    Collective on SNES
2155fee21e36SBarry Smith 
2156c7afd0dbSLois Curfman McInnes    Input Parameter:
2157c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2158c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
2159c7afd0dbSLois Curfman McInnes 
2160d850072dSLois Curfman McInnes    Notes:
2161a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2162c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
2163d850072dSLois Curfman McInnes 
216436851e7fSLois Curfman McInnes    Level: advanced
216536851e7fSLois Curfman McInnes 
21663c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database
2167a86d99e1SLois Curfman McInnes 
2168a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions()
21693c7409f5SSatish Balay @*/
217063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetOptionsPrefix(SNES snes,const char prefix[])
21713c7409f5SSatish Balay {
2172dfbe8321SBarry Smith   PetscErrorCode ierr;
21733c7409f5SSatish Balay 
21743a40ed3dSBarry Smith   PetscFunctionBegin;
21754482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
2176639f9d9dSBarry Smith   ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
217794b7f48cSBarry Smith   ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
21783a40ed3dSBarry Smith   PetscFunctionReturn(0);
21793c7409f5SSatish Balay }
21803c7409f5SSatish Balay 
21814a2ae208SSatish Balay #undef __FUNCT__
21824a2ae208SSatish Balay #define __FUNCT__ "SNESAppendOptionsPrefix"
21833c7409f5SSatish Balay /*@C
2184f525115eSLois Curfman McInnes    SNESAppendOptionsPrefix - Appends to the prefix used for searching for all
2185d850072dSLois Curfman McInnes    SNES options in the database.
21863c7409f5SSatish Balay 
2187fee21e36SBarry Smith    Collective on SNES
2188fee21e36SBarry Smith 
2189c7afd0dbSLois Curfman McInnes    Input Parameters:
2190c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2191c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
2192c7afd0dbSLois Curfman McInnes 
2193d850072dSLois Curfman McInnes    Notes:
2194a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2195c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
2196d850072dSLois Curfman McInnes 
219736851e7fSLois Curfman McInnes    Level: advanced
219836851e7fSLois Curfman McInnes 
21993c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database
2200a86d99e1SLois Curfman McInnes 
2201a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix()
22023c7409f5SSatish Balay @*/
220363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESAppendOptionsPrefix(SNES snes,const char prefix[])
22043c7409f5SSatish Balay {
2205dfbe8321SBarry Smith   PetscErrorCode ierr;
22063c7409f5SSatish Balay 
22073a40ed3dSBarry Smith   PetscFunctionBegin;
22084482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
2209639f9d9dSBarry Smith   ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
221094b7f48cSBarry Smith   ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
22113a40ed3dSBarry Smith   PetscFunctionReturn(0);
22123c7409f5SSatish Balay }
22133c7409f5SSatish Balay 
22144a2ae208SSatish Balay #undef __FUNCT__
22154a2ae208SSatish Balay #define __FUNCT__ "SNESGetOptionsPrefix"
22169ab63eb5SSatish Balay /*@C
22173c7409f5SSatish Balay    SNESGetOptionsPrefix - Sets the prefix used for searching for all
22183c7409f5SSatish Balay    SNES options in the database.
22193c7409f5SSatish Balay 
2220c7afd0dbSLois Curfman McInnes    Not Collective
2221c7afd0dbSLois Curfman McInnes 
22223c7409f5SSatish Balay    Input Parameter:
22233c7409f5SSatish Balay .  snes - the SNES context
22243c7409f5SSatish Balay 
22253c7409f5SSatish Balay    Output Parameter:
22263c7409f5SSatish Balay .  prefix - pointer to the prefix string used
22273c7409f5SSatish Balay 
22289ab63eb5SSatish Balay    Notes: On the fortran side, the user should pass in a string 'prifix' of
22299ab63eb5SSatish Balay    sufficient length to hold the prefix.
22309ab63eb5SSatish Balay 
223136851e7fSLois Curfman McInnes    Level: advanced
223236851e7fSLois Curfman McInnes 
22333c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database
2234a86d99e1SLois Curfman McInnes 
2235a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix()
22363c7409f5SSatish Balay @*/
2237e060cb09SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESGetOptionsPrefix(SNES snes,const char *prefix[])
22383c7409f5SSatish Balay {
2239dfbe8321SBarry Smith   PetscErrorCode ierr;
22403c7409f5SSatish Balay 
22413a40ed3dSBarry Smith   PetscFunctionBegin;
22424482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
2243639f9d9dSBarry Smith   ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
22443a40ed3dSBarry Smith   PetscFunctionReturn(0);
22453c7409f5SSatish Balay }
22463c7409f5SSatish Balay 
2247b2002411SLois Curfman McInnes 
22484a2ae208SSatish Balay #undef __FUNCT__
22494a2ae208SSatish Balay #define __FUNCT__ "SNESRegister"
22503cea93caSBarry Smith /*@C
22513cea93caSBarry Smith   SNESRegister - See SNESRegisterDynamic()
22523cea93caSBarry Smith 
22537f6c08e0SMatthew Knepley   Level: advanced
22543cea93caSBarry Smith @*/
225563dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESRegister(const char sname[],const char path[],const char name[],PetscErrorCode (*function)(SNES))
2256b2002411SLois Curfman McInnes {
2257e2d1d2b7SBarry Smith   char           fullname[PETSC_MAX_PATH_LEN];
2258dfbe8321SBarry Smith   PetscErrorCode ierr;
2259b2002411SLois Curfman McInnes 
2260b2002411SLois Curfman McInnes   PetscFunctionBegin;
2261b0a32e0cSBarry Smith   ierr = PetscFListConcat(path,name,fullname);CHKERRQ(ierr);
2262c134de8dSSatish Balay   ierr = PetscFListAdd(&SNESList,sname,fullname,(void (*)(void))function);CHKERRQ(ierr);
2263b2002411SLois Curfman McInnes   PetscFunctionReturn(0);
2264b2002411SLois Curfman McInnes }
2265da9b6338SBarry Smith 
2266da9b6338SBarry Smith #undef __FUNCT__
2267da9b6338SBarry Smith #define __FUNCT__ "SNESTestLocalMin"
226863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESTestLocalMin(SNES snes)
2269da9b6338SBarry Smith {
2270dfbe8321SBarry Smith   PetscErrorCode ierr;
227177431f27SBarry Smith   PetscInt       N,i,j;
2272da9b6338SBarry Smith   Vec            u,uh,fh;
2273da9b6338SBarry Smith   PetscScalar    value;
2274da9b6338SBarry Smith   PetscReal      norm;
2275da9b6338SBarry Smith 
2276da9b6338SBarry Smith   PetscFunctionBegin;
2277da9b6338SBarry Smith   ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr);
2278da9b6338SBarry Smith   ierr = VecDuplicate(u,&uh);CHKERRQ(ierr);
2279da9b6338SBarry Smith   ierr = VecDuplicate(u,&fh);CHKERRQ(ierr);
2280da9b6338SBarry Smith 
2281da9b6338SBarry Smith   /* currently only works for sequential */
2282da9b6338SBarry Smith   ierr = PetscPrintf(PETSC_COMM_WORLD,"Testing FormFunction() for local min\n");
2283da9b6338SBarry Smith   ierr = VecGetSize(u,&N);CHKERRQ(ierr);
2284da9b6338SBarry Smith   for (i=0; i<N; i++) {
2285da9b6338SBarry Smith     ierr = VecCopy(u,uh);CHKERRQ(ierr);
228677431f27SBarry Smith     ierr  = PetscPrintf(PETSC_COMM_WORLD,"i = %D\n",i);CHKERRQ(ierr);
2287da9b6338SBarry Smith     for (j=-10; j<11; j++) {
2288ccae9161SBarry Smith       value = PetscSign(j)*exp(PetscAbs(j)-10.0);
2289da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
22903ab0aad5SBarry Smith       ierr  = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr);
2291da9b6338SBarry Smith       ierr  = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr);
229277431f27SBarry Smith       ierr  = PetscPrintf(PETSC_COMM_WORLD,"       j norm %D %18.16e\n",j,norm);CHKERRQ(ierr);
2293da9b6338SBarry Smith       value = -value;
2294da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
2295da9b6338SBarry Smith     }
2296da9b6338SBarry Smith   }
2297da9b6338SBarry Smith   ierr = VecDestroy(uh);CHKERRQ(ierr);
2298da9b6338SBarry Smith   ierr = VecDestroy(fh);CHKERRQ(ierr);
2299da9b6338SBarry Smith   PetscFunctionReturn(0);
2300da9b6338SBarry Smith }
230171f87433Sdalcinl 
230271f87433Sdalcinl #undef __FUNCT__
230371f87433Sdalcinl #define __FUNCT__ "SNES_KSP_SetUseEW"
230471f87433Sdalcinl /*@
230571f87433Sdalcinl    SNES_KSP_SetUseEW - Sets SNES use Eisenstat-Walker method for
230671f87433Sdalcinl    computing relative tolerance for linear solvers within an inexact
230771f87433Sdalcinl    Newton method.
230871f87433Sdalcinl 
230971f87433Sdalcinl    Collective on SNES
231071f87433Sdalcinl 
231171f87433Sdalcinl    Input Parameters:
231271f87433Sdalcinl +  snes - SNES context
231371f87433Sdalcinl -  flag - PETSC_TRUE or PETSC_FALSE
231471f87433Sdalcinl 
231571f87433Sdalcinl    Notes:
231671f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
231771f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
231871f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
231971f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
232071f87433Sdalcinl    solver.
232171f87433Sdalcinl 
232271f87433Sdalcinl    Level: advanced
232371f87433Sdalcinl 
232471f87433Sdalcinl    Reference:
232571f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
232671f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
232771f87433Sdalcinl 
232871f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
232971f87433Sdalcinl 
233071f87433Sdalcinl .seealso: SNES_KSP_GetUseEW(), SNES_KSP_GetParametersEW(), SNES_KSP_SetParametersEW()
233171f87433Sdalcinl @*/
233271f87433Sdalcinl PetscErrorCode PETSCSNES_DLLEXPORT SNES_KSP_SetUseEW(SNES snes,PetscTruth flag)
233371f87433Sdalcinl {
233471f87433Sdalcinl   PetscFunctionBegin;
233571f87433Sdalcinl   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
233671f87433Sdalcinl   snes->ksp_ewconv = flag;
233771f87433Sdalcinl   PetscFunctionReturn(0);
233871f87433Sdalcinl }
233971f87433Sdalcinl 
234071f87433Sdalcinl #undef __FUNCT__
234171f87433Sdalcinl #define __FUNCT__ "SNES_KSP_GetUseEW"
234271f87433Sdalcinl /*@
234371f87433Sdalcinl    SNES_KSP_GetUseEW - Gets if SNES is using Eisenstat-Walker method
234471f87433Sdalcinl    for computing relative tolerance for linear solvers within an
234571f87433Sdalcinl    inexact Newton method.
234671f87433Sdalcinl 
234771f87433Sdalcinl    Not Collective
234871f87433Sdalcinl 
234971f87433Sdalcinl    Input Parameter:
235071f87433Sdalcinl .  snes - SNES context
235171f87433Sdalcinl 
235271f87433Sdalcinl    Output Parameter:
235371f87433Sdalcinl .  flag - PETSC_TRUE or PETSC_FALSE
235471f87433Sdalcinl 
235571f87433Sdalcinl    Notes:
235671f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
235771f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
235871f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
235971f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
236071f87433Sdalcinl    solver.
236171f87433Sdalcinl 
236271f87433Sdalcinl    Level: advanced
236371f87433Sdalcinl 
236471f87433Sdalcinl    Reference:
236571f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
236671f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
236771f87433Sdalcinl 
236871f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
236971f87433Sdalcinl 
237071f87433Sdalcinl .seealso: SNES_KSP_SetUseEW(), SNES_KSP_GetParametersEW(), SNES_KSP_SetParametersEW()
237171f87433Sdalcinl @*/
237271f87433Sdalcinl PetscErrorCode PETSCSNES_DLLEXPORT SNES_KSP_GetUseEW(SNES snes, PetscTruth *flag)
237371f87433Sdalcinl {
237471f87433Sdalcinl   PetscFunctionBegin;
237571f87433Sdalcinl   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
237671f87433Sdalcinl   PetscValidPointer(flag,2);
237771f87433Sdalcinl   *flag = snes->ksp_ewconv;
237871f87433Sdalcinl   PetscFunctionReturn(0);
237971f87433Sdalcinl }
238071f87433Sdalcinl 
238171f87433Sdalcinl #undef __FUNCT__
238271f87433Sdalcinl #define __FUNCT__ "SNES_KSP_SetParametersEW"
238371f87433Sdalcinl /*@
238471f87433Sdalcinl    SNES_KSP_SetParametersEW - Sets parameters for Eisenstat-Walker
238571f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
238671f87433Sdalcinl    Newton method.
238771f87433Sdalcinl 
238871f87433Sdalcinl    Collective on SNES
238971f87433Sdalcinl 
239071f87433Sdalcinl    Input Parameters:
239171f87433Sdalcinl +    snes - SNES context
239271f87433Sdalcinl .    version - version 1, 2 (default is 2) or 3
239371f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
239471f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
239571f87433Sdalcinl .    gamma - multiplicative factor for version 2 rtol computation
239671f87433Sdalcinl              (0 <= gamma2 <= 1)
239771f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
239871f87433Sdalcinl .    alpha2 - power for safeguard
239971f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
240071f87433Sdalcinl 
240171f87433Sdalcinl    Note:
240271f87433Sdalcinl    Version 3 was contributed by Luis Chacon, June 2006.
240371f87433Sdalcinl 
240471f87433Sdalcinl    Use PETSC_DEFAULT to retain the default for any of the parameters.
240571f87433Sdalcinl 
240671f87433Sdalcinl    Level: advanced
240771f87433Sdalcinl 
240871f87433Sdalcinl    Reference:
240971f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
241071f87433Sdalcinl    inexact Newton method", Utah State University Math. Stat. Dept. Res.
241171f87433Sdalcinl    Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput.
241271f87433Sdalcinl 
241371f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, set, parameters
241471f87433Sdalcinl 
241571f87433Sdalcinl .seealso: SNES_KSP_SetUseEW(), SNES_KSP_GetUseEW(), SNES_KSP_GetParametersEW()
241671f87433Sdalcinl @*/
241771f87433Sdalcinl PetscErrorCode PETSCSNES_DLLEXPORT SNES_KSP_SetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,
241871f87433Sdalcinl 							    PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold)
241971f87433Sdalcinl {
242071f87433Sdalcinl   SNES_KSP_EW_ConvCtx *kctx;
242171f87433Sdalcinl   PetscFunctionBegin;
242271f87433Sdalcinl   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
242371f87433Sdalcinl   kctx = (SNES_KSP_EW_ConvCtx*)snes->kspconvctx;
242471f87433Sdalcinl   if (!kctx) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
242571f87433Sdalcinl 
242671f87433Sdalcinl   if (version != PETSC_DEFAULT)   kctx->version   = version;
242771f87433Sdalcinl   if (rtol_0 != PETSC_DEFAULT)    kctx->rtol_0    = rtol_0;
242871f87433Sdalcinl   if (rtol_max != PETSC_DEFAULT)  kctx->rtol_max  = rtol_max;
242971f87433Sdalcinl   if (gamma != PETSC_DEFAULT)     kctx->gamma     = gamma;
243071f87433Sdalcinl   if (alpha != PETSC_DEFAULT)     kctx->alpha     = alpha;
243171f87433Sdalcinl   if (alpha2 != PETSC_DEFAULT)    kctx->alpha2    = alpha2;
243271f87433Sdalcinl   if (threshold != PETSC_DEFAULT) kctx->threshold = threshold;
243371f87433Sdalcinl 
243471f87433Sdalcinl   if (kctx->version < 1 || kctx->version > 3) {
243571f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 and 3 are supported: %D",kctx->version);
243671f87433Sdalcinl   }
243771f87433Sdalcinl   if (kctx->rtol_0 < 0.0 || kctx->rtol_0 >= 1.0) {
243871f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_0 < 1.0: %G",kctx->rtol_0);
243971f87433Sdalcinl   }
244071f87433Sdalcinl   if (kctx->rtol_max < 0.0 || kctx->rtol_max >= 1.0) {
244171f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_max (%G) < 1.0\n",kctx->rtol_max);
244271f87433Sdalcinl   }
244371f87433Sdalcinl   if (kctx->gamma < 0.0 || kctx->gamma > 1.0) {
244471f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= gamma (%G) <= 1.0\n",kctx->gamma);
244571f87433Sdalcinl   }
244671f87433Sdalcinl   if (kctx->alpha <= 1.0 || kctx->alpha > 2.0) {
244771f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"1.0 < alpha (%G) <= 2.0\n",kctx->alpha);
244871f87433Sdalcinl   }
244971f87433Sdalcinl   if (kctx->threshold <= 0.0 || kctx->threshold >= 1.0) {
245071f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"0.0 < threshold (%G) < 1.0\n",kctx->threshold);
245171f87433Sdalcinl   }
245271f87433Sdalcinl   PetscFunctionReturn(0);
245371f87433Sdalcinl }
245471f87433Sdalcinl 
245571f87433Sdalcinl #undef __FUNCT__
245671f87433Sdalcinl #define __FUNCT__ "SNES_KSP_GetParametersEW"
245771f87433Sdalcinl /*@
245871f87433Sdalcinl    SNES_KSP_GetParametersEW - Gets parameters for Eisenstat-Walker
245971f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
246071f87433Sdalcinl    Newton method.
246171f87433Sdalcinl 
246271f87433Sdalcinl    Not Collective
246371f87433Sdalcinl 
246471f87433Sdalcinl    Input Parameters:
246571f87433Sdalcinl      snes - SNES context
246671f87433Sdalcinl 
246771f87433Sdalcinl    Output Parameters:
246871f87433Sdalcinl +    version - version 1, 2 (default is 2) or 3
246971f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
247071f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
247171f87433Sdalcinl .    gamma - multiplicative factor for version 2 rtol computation
247271f87433Sdalcinl              (0 <= gamma2 <= 1)
247371f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
247471f87433Sdalcinl .    alpha2 - power for safeguard
247571f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
247671f87433Sdalcinl 
247771f87433Sdalcinl    Level: advanced
247871f87433Sdalcinl 
247971f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, get, parameters
248071f87433Sdalcinl 
248171f87433Sdalcinl .seealso: SNES_KSP_SetUseEW(), SNES_KSP_GetUseEW(), SNES_KSP_SetParametersEW()
248271f87433Sdalcinl @*/
248371f87433Sdalcinl PetscErrorCode PETSCSNES_DLLEXPORT SNES_KSP_GetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,
248471f87433Sdalcinl 							    PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold)
248571f87433Sdalcinl {
248671f87433Sdalcinl   SNES_KSP_EW_ConvCtx *kctx;
248771f87433Sdalcinl   PetscFunctionBegin;
248871f87433Sdalcinl   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
248971f87433Sdalcinl   kctx = (SNES_KSP_EW_ConvCtx*)snes->kspconvctx;
249071f87433Sdalcinl   if (!kctx) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
249171f87433Sdalcinl   if(version)   *version   = kctx->version;
249271f87433Sdalcinl   if(rtol_0)    *rtol_0    = kctx->rtol_0;
249371f87433Sdalcinl   if(rtol_max)  *rtol_max  = kctx->rtol_max;
249471f87433Sdalcinl   if(gamma)     *gamma     = kctx->gamma;
249571f87433Sdalcinl   if(alpha)     *alpha     = kctx->alpha;
249671f87433Sdalcinl   if(alpha2)    *alpha2    = kctx->alpha2;
249771f87433Sdalcinl   if(threshold) *threshold = kctx->threshold;
249871f87433Sdalcinl   PetscFunctionReturn(0);
249971f87433Sdalcinl }
250071f87433Sdalcinl 
250171f87433Sdalcinl #undef __FUNCT__
250271f87433Sdalcinl #define __FUNCT__ "SNES_KSP_EW_PreSolve"
250371f87433Sdalcinl static PetscErrorCode SNES_KSP_EW_PreSolve(SNES snes, KSP ksp, Vec b, Vec x)
250471f87433Sdalcinl {
250571f87433Sdalcinl   PetscErrorCode       ierr;
250671f87433Sdalcinl   SNES_KSP_EW_ConvCtx *kctx = (SNES_KSP_EW_ConvCtx*)snes->kspconvctx;
250771f87433Sdalcinl   PetscReal            rtol=PETSC_DEFAULT,stol;
250871f87433Sdalcinl 
250971f87433Sdalcinl   PetscFunctionBegin;
251071f87433Sdalcinl   if (!kctx) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context exists");
251171f87433Sdalcinl   if (!snes->iter) { /* first time in, so use the original user rtol */
251271f87433Sdalcinl     rtol = kctx->rtol_0;
251371f87433Sdalcinl   } else {
251471f87433Sdalcinl     if (kctx->version == 1) {
251571f87433Sdalcinl       rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last;
251671f87433Sdalcinl       if (rtol < 0.0) rtol = -rtol;
251771f87433Sdalcinl       stol = pow(kctx->rtol_last,kctx->alpha2);
251871f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
251971f87433Sdalcinl     } else if (kctx->version == 2) {
252071f87433Sdalcinl       rtol = kctx->gamma * pow(snes->norm/kctx->norm_last,kctx->alpha);
252171f87433Sdalcinl       stol = kctx->gamma * pow(kctx->rtol_last,kctx->alpha);
252271f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
252371f87433Sdalcinl     } else if (kctx->version == 3) {/* contributed by Luis Chacon, June 2006. */
252471f87433Sdalcinl       rtol = kctx->gamma * pow(snes->norm/kctx->norm_last,kctx->alpha);
252571f87433Sdalcinl       /* safeguard: avoid sharp decrease of rtol */
252671f87433Sdalcinl       stol = kctx->gamma*pow(kctx->rtol_last,kctx->alpha);
252771f87433Sdalcinl       stol = PetscMax(rtol,stol);
252871f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
252971f87433Sdalcinl       /* safeguard: avoid oversolving */
253071f87433Sdalcinl       stol = kctx->gamma*(snes->ttol)/snes->norm;
253171f87433Sdalcinl       stol = PetscMax(rtol,stol);
253271f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
253371f87433Sdalcinl     } else SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version);
253471f87433Sdalcinl   }
253571f87433Sdalcinl   /* safeguard: avoid rtol greater than one */
253671f87433Sdalcinl   rtol = PetscMin(rtol,kctx->rtol_max);
253771f87433Sdalcinl   ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr);
253871f87433Sdalcinl   ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%G\n",snes->iter,kctx->version,rtol);CHKERRQ(ierr);
253971f87433Sdalcinl   PetscFunctionReturn(0);
254071f87433Sdalcinl }
254171f87433Sdalcinl 
254271f87433Sdalcinl #undef __FUNCT__
254371f87433Sdalcinl #define __FUNCT__ "SNES_KSP_EW_PostSolve"
254471f87433Sdalcinl static PetscErrorCode SNES_KSP_EW_PostSolve(SNES snes, KSP ksp, Vec b, Vec x)
254571f87433Sdalcinl {
254671f87433Sdalcinl   PetscErrorCode      ierr;
254771f87433Sdalcinl   SNES_KSP_EW_ConvCtx *kctx = (SNES_KSP_EW_ConvCtx*)snes->kspconvctx;
254871f87433Sdalcinl   PCSide              pcside;
254971f87433Sdalcinl   Vec                 lres;
255071f87433Sdalcinl 
255171f87433Sdalcinl   PetscFunctionBegin;
255271f87433Sdalcinl   if (!kctx) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context exists");
255371f87433Sdalcinl   ierr = KSPGetTolerances(ksp,&kctx->rtol_last,0,0,0);CHKERRQ(ierr);
255471f87433Sdalcinl   ierr = SNESGetFunctionNorm(snes,&kctx->norm_last);CHKERRQ(ierr);
255571f87433Sdalcinl   if (kctx->version == 1) {
255671f87433Sdalcinl     ierr = KSPGetPreconditionerSide(ksp,&pcside);CHKERRQ(ierr);
255771f87433Sdalcinl     if (pcside == PC_RIGHT) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */
255871f87433Sdalcinl       /* KSP residual is true linear residual */
255971f87433Sdalcinl       ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr);
256071f87433Sdalcinl     } else {
256171f87433Sdalcinl       /* KSP residual is preconditioned residual */
256271f87433Sdalcinl       /* compute true linear residual norm */
256371f87433Sdalcinl       ierr = VecDuplicate(b,&lres);CHKERRQ(ierr);
256471f87433Sdalcinl       ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr);
256571f87433Sdalcinl       ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr);
256671f87433Sdalcinl       ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr);
256771f87433Sdalcinl       ierr = VecDestroy(lres);CHKERRQ(ierr);
256871f87433Sdalcinl     }
256971f87433Sdalcinl   }
257071f87433Sdalcinl   PetscFunctionReturn(0);
257171f87433Sdalcinl }
257271f87433Sdalcinl 
257371f87433Sdalcinl #undef __FUNCT__
257471f87433Sdalcinl #define __FUNCT__ "SNES_KSPSolve"
257571f87433Sdalcinl PetscErrorCode SNES_KSPSolve(SNES snes, KSP ksp, Vec b, Vec x)
257671f87433Sdalcinl {
257771f87433Sdalcinl   PetscErrorCode ierr;
257871f87433Sdalcinl 
257971f87433Sdalcinl   PetscFunctionBegin;
258071f87433Sdalcinl   if (snes->ksp_ewconv) { ierr = SNES_KSP_EW_PreSolve(snes,ksp,b,x);CHKERRQ(ierr);  }
258171f87433Sdalcinl   ierr = KSPSolve(ksp,b,x);CHKERRQ(ierr);
258271f87433Sdalcinl   if (snes->ksp_ewconv) { ierr = SNES_KSP_EW_PostSolve(snes,ksp,b,x);CHKERRQ(ierr); }
258371f87433Sdalcinl   PetscFunctionReturn(0);
258471f87433Sdalcinl }
2585