xref: /petsc/src/snes/interface/snes.c (revision 38f2d2fdb3b6f522a3102c6eb796cebecf3224c0)
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 {
45fa9f3622SBarry Smith   SNESKSPEW           *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) {
84fa9f3622SBarry Smith       kctx = (SNESKSPEW *)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:
169fa9f3622SBarry Smith +  -snes_ksp_ew - 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;
191fa9f3622SBarry Smith   SNESKSPEW               *kctx = (SNESKSPEW *)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 
230fa9f3622SBarry Smith     ierr = PetscOptionsTruth("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,PETSC_NULL);CHKERRQ(ierr);
231186905e3SBarry Smith 
232fa9f3622SBarry Smith     ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,0);CHKERRQ(ierr);
233fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,0);CHKERRQ(ierr);
234fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,0);CHKERRQ(ierr);
235fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,0);CHKERRQ(ierr);
236fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,0);CHKERRQ(ierr);
237fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,0);CHKERRQ(ierr);
238fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,0);CHKERRQ(ierr);
239186905e3SBarry Smith 
240b0a32e0cSBarry Smith     ierr = PetscOptionsName("-snes_no_convergence_test","Don't test for convergence","None",&flg);CHKERRQ(ierr);
241e7788613SBarry Smith     if (flg) {snes->ops->converged = 0;}
242a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",&flg);CHKERRQ(ierr);
243a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);}
244eabae89aSBarry Smith 
245a6570f20SBarry Smith     ierr = PetscOptionsString("-snes_monitor","Monitor norm of function","SNESMonitorSet","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
246e8105e01SRichard Katz     if (flg) {
24723d894e5SBarry Smith       ierr = PetscViewerASCIIMonitorCreate(snes->comm,monfilename,0,&monviewer);CHKERRQ(ierr);
24823d894e5SBarry Smith       ierr = SNESMonitorSet(snes,SNESMonitorDefault,monviewer,(PetscErrorCode (*)(void*))PetscViewerASCIIMonitorDestroy);CHKERRQ(ierr);
249e8105e01SRichard Katz     }
250eabae89aSBarry Smith 
251a6570f20SBarry Smith     ierr = PetscOptionsString("-snes_ratiomonitor","Monitor ratios of norms of function","SNESMonitorSetRatio","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
252eabae89aSBarry Smith     if (flg) {
25323d894e5SBarry Smith       ierr = PetscViewerASCIIMonitorCreate(snes->comm,monfilename,0,&monviewer);CHKERRQ(ierr);
254f1bef1bcSMatthew Knepley       ierr = SNESMonitorSetRatio(snes,monviewer);CHKERRQ(ierr);
255e8105e01SRichard Katz     }
256eabae89aSBarry Smith 
257a6570f20SBarry Smith     ierr = PetscOptionsString("-snes_monitor_short","Monitor norm of function (fewer digits)","SNESMonitorSet","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
258eabae89aSBarry Smith     if (flg) {
25923d894e5SBarry Smith       ierr = PetscViewerASCIIMonitorCreate(snes->comm,monfilename,0,&monviewer);CHKERRQ(ierr);
26023d894e5SBarry Smith       ierr = SNESMonitorSet(snes,SNESMonitorDefaultShort,monviewer,(PetscErrorCode (*)(void*))PetscViewerASCIIMonitorDestroy);CHKERRQ(ierr);
261eabae89aSBarry Smith     }
262eabae89aSBarry Smith 
263a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_solution","Plot solution at each iteration","SNESMonitorSolution",&flg);CHKERRQ(ierr);
264a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorSolution,0,0);CHKERRQ(ierr);}
265a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_solution_update","Plot correction at each iteration","SNESMonitorSolutionUpdate",&flg);CHKERRQ(ierr);
266a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorSolutionUpdate,0,0);CHKERRQ(ierr);}
267a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_residual","Plot residual at each iteration","SNESMonitorResidual",&flg);CHKERRQ(ierr);
268a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorResidual,0,0);CHKERRQ(ierr);}
269a6570f20SBarry Smith     ierr = PetscOptionsName("-snes_monitor_draw","Plot function norm at each iteration","SNESMonitorLG",&flg);CHKERRQ(ierr);
270a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorLG,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);}
271e24b481bSBarry Smith 
272b0a32e0cSBarry Smith     ierr = PetscOptionsName("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESDefaultComputeJacobian",&flg);CHKERRQ(ierr);
2734b27c08aSLois Curfman McInnes     if (flg) {
274186905e3SBarry Smith       ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESDefaultComputeJacobian,snes->funP);CHKERRQ(ierr);
275ae15b995SBarry Smith       ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr);
2769b94acceSBarry Smith     }
277639f9d9dSBarry Smith 
27876b2cf59SMatthew Knepley     for(i = 0; i < numberofsetfromoptions; i++) {
27976b2cf59SMatthew Knepley       ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr);
28076b2cf59SMatthew Knepley     }
28176b2cf59SMatthew Knepley 
282e7788613SBarry Smith     if (snes->ops->setfromoptions) {
283e7788613SBarry Smith       ierr = (*snes->ops->setfromoptions)(snes);CHKERRQ(ierr);
284639f9d9dSBarry Smith     }
285b0a32e0cSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
2864bbc92c1SBarry Smith 
28794b7f48cSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
28894b7f48cSBarry Smith   ierr = KSPSetFromOptions(ksp);CHKERRQ(ierr);
28993993e2dSLois Curfman McInnes 
2903a40ed3dSBarry Smith   PetscFunctionReturn(0);
2919b94acceSBarry Smith }
2929b94acceSBarry Smith 
293a847f771SSatish Balay 
2944a2ae208SSatish Balay #undef __FUNCT__
2954a2ae208SSatish Balay #define __FUNCT__ "SNESSetApplicationContext"
2969b94acceSBarry Smith /*@
2979b94acceSBarry Smith    SNESSetApplicationContext - Sets the optional user-defined context for
2989b94acceSBarry Smith    the nonlinear solvers.
2999b94acceSBarry Smith 
300fee21e36SBarry Smith    Collective on SNES
301fee21e36SBarry Smith 
302c7afd0dbSLois Curfman McInnes    Input Parameters:
303c7afd0dbSLois Curfman McInnes +  snes - the SNES context
304c7afd0dbSLois Curfman McInnes -  usrP - optional user context
305c7afd0dbSLois Curfman McInnes 
30636851e7fSLois Curfman McInnes    Level: intermediate
30736851e7fSLois Curfman McInnes 
3089b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context
3099b94acceSBarry Smith 
3109b94acceSBarry Smith .seealso: SNESGetApplicationContext()
3119b94acceSBarry Smith @*/
31263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetApplicationContext(SNES snes,void *usrP)
3139b94acceSBarry Smith {
3143a40ed3dSBarry Smith   PetscFunctionBegin;
3154482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
3169b94acceSBarry Smith   snes->user		= usrP;
3173a40ed3dSBarry Smith   PetscFunctionReturn(0);
3189b94acceSBarry Smith }
31974679c65SBarry Smith 
3204a2ae208SSatish Balay #undef __FUNCT__
3214a2ae208SSatish Balay #define __FUNCT__ "SNESGetApplicationContext"
3229b94acceSBarry Smith /*@C
3239b94acceSBarry Smith    SNESGetApplicationContext - Gets the user-defined context for the
3249b94acceSBarry Smith    nonlinear solvers.
3259b94acceSBarry Smith 
326c7afd0dbSLois Curfman McInnes    Not Collective
327c7afd0dbSLois Curfman McInnes 
3289b94acceSBarry Smith    Input Parameter:
3299b94acceSBarry Smith .  snes - SNES context
3309b94acceSBarry Smith 
3319b94acceSBarry Smith    Output Parameter:
3329b94acceSBarry Smith .  usrP - user context
3339b94acceSBarry Smith 
33436851e7fSLois Curfman McInnes    Level: intermediate
33536851e7fSLois Curfman McInnes 
3369b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context
3379b94acceSBarry Smith 
3389b94acceSBarry Smith .seealso: SNESSetApplicationContext()
3399b94acceSBarry Smith @*/
34063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetApplicationContext(SNES snes,void **usrP)
3419b94acceSBarry Smith {
3423a40ed3dSBarry Smith   PetscFunctionBegin;
3434482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
3449b94acceSBarry Smith   *usrP = snes->user;
3453a40ed3dSBarry Smith   PetscFunctionReturn(0);
3469b94acceSBarry Smith }
34774679c65SBarry Smith 
3484a2ae208SSatish Balay #undef __FUNCT__
3494a2ae208SSatish Balay #define __FUNCT__ "SNESGetIterationNumber"
3509b94acceSBarry Smith /*@
351c8228a4eSBarry Smith    SNESGetIterationNumber - Gets the number of nonlinear iterations completed
352c8228a4eSBarry Smith    at this time.
3539b94acceSBarry Smith 
354c7afd0dbSLois Curfman McInnes    Not Collective
355c7afd0dbSLois Curfman McInnes 
3569b94acceSBarry Smith    Input Parameter:
3579b94acceSBarry Smith .  snes - SNES context
3589b94acceSBarry Smith 
3599b94acceSBarry Smith    Output Parameter:
3609b94acceSBarry Smith .  iter - iteration number
3619b94acceSBarry Smith 
362c8228a4eSBarry Smith    Notes:
363c8228a4eSBarry Smith    For example, during the computation of iteration 2 this would return 1.
364c8228a4eSBarry Smith 
365c8228a4eSBarry Smith    This is useful for using lagged Jacobians (where one does not recompute the
36608405cd6SLois Curfman McInnes    Jacobian at each SNES iteration). For example, the code
36708405cd6SLois Curfman McInnes .vb
36808405cd6SLois Curfman McInnes       ierr = SNESGetIterationNumber(snes,&it);
36908405cd6SLois Curfman McInnes       if (!(it % 2)) {
37008405cd6SLois Curfman McInnes         [compute Jacobian here]
37108405cd6SLois Curfman McInnes       }
37208405cd6SLois Curfman McInnes .ve
373c8228a4eSBarry Smith    can be used in your ComputeJacobian() function to cause the Jacobian to be
37408405cd6SLois Curfman McInnes    recomputed every second SNES iteration.
375c8228a4eSBarry Smith 
37636851e7fSLois Curfman McInnes    Level: intermediate
37736851e7fSLois Curfman McInnes 
3782b668275SBarry Smith .keywords: SNES, nonlinear, get, iteration, number,
3792b668275SBarry Smith 
380b850b91aSLisandro Dalcin .seealso:   SNESGetFunctionNorm(), SNESGetLinearSolveIterations()
3819b94acceSBarry Smith @*/
38263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetIterationNumber(SNES snes,PetscInt* iter)
3839b94acceSBarry Smith {
3843a40ed3dSBarry Smith   PetscFunctionBegin;
3854482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
3864482741eSBarry Smith   PetscValidIntPointer(iter,2);
3879b94acceSBarry Smith   *iter = snes->iter;
3883a40ed3dSBarry Smith   PetscFunctionReturn(0);
3899b94acceSBarry Smith }
39074679c65SBarry Smith 
3914a2ae208SSatish Balay #undef __FUNCT__
3924a2ae208SSatish Balay #define __FUNCT__ "SNESGetFunctionNorm"
3939b94acceSBarry Smith /*@
3949b94acceSBarry Smith    SNESGetFunctionNorm - Gets the norm of the current function that was set
3959b94acceSBarry Smith    with SNESSSetFunction().
3969b94acceSBarry Smith 
397c7afd0dbSLois Curfman McInnes    Collective on SNES
398c7afd0dbSLois Curfman McInnes 
3999b94acceSBarry Smith    Input Parameter:
4009b94acceSBarry Smith .  snes - SNES context
4019b94acceSBarry Smith 
4029b94acceSBarry Smith    Output Parameter:
4039b94acceSBarry Smith .  fnorm - 2-norm of function
4049b94acceSBarry Smith 
40536851e7fSLois Curfman McInnes    Level: intermediate
40636851e7fSLois Curfman McInnes 
4079b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm
408a86d99e1SLois Curfman McInnes 
409b850b91aSLisandro Dalcin .seealso: SNESGetFunction(), SNESGetIterationNumber(), SNESGetLinearSolveIterations()
4109b94acceSBarry Smith @*/
41171f87433Sdalcinl PetscErrorCode PETSCSNES_DLLEXPORT SNESGetFunctionNorm(SNES snes,PetscReal *fnorm)
4129b94acceSBarry Smith {
4133a40ed3dSBarry Smith   PetscFunctionBegin;
4144482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
4154482741eSBarry Smith   PetscValidScalarPointer(fnorm,2);
4169b94acceSBarry Smith   *fnorm = snes->norm;
4173a40ed3dSBarry Smith   PetscFunctionReturn(0);
4189b94acceSBarry Smith }
41974679c65SBarry Smith 
4204a2ae208SSatish Balay #undef __FUNCT__
421b850b91aSLisandro Dalcin #define __FUNCT__ "SNESGetNonlinearStepFailures"
4229b94acceSBarry Smith /*@
423b850b91aSLisandro Dalcin    SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps
4249b94acceSBarry Smith    attempted by the nonlinear solver.
4259b94acceSBarry Smith 
426c7afd0dbSLois Curfman McInnes    Not Collective
427c7afd0dbSLois Curfman McInnes 
4289b94acceSBarry Smith    Input Parameter:
4299b94acceSBarry Smith .  snes - SNES context
4309b94acceSBarry Smith 
4319b94acceSBarry Smith    Output Parameter:
4329b94acceSBarry Smith .  nfails - number of unsuccessful steps attempted
4339b94acceSBarry Smith 
434c96a6f78SLois Curfman McInnes    Notes:
435c96a6f78SLois Curfman McInnes    This counter is reset to zero for each successive call to SNESSolve().
436c96a6f78SLois Curfman McInnes 
43736851e7fSLois Curfman McInnes    Level: intermediate
43836851e7fSLois Curfman McInnes 
4399b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps
4409b94acceSBarry Smith @*/
441b850b91aSLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESGetNonlinearStepFailures(SNES snes,PetscInt* nfails)
4429b94acceSBarry Smith {
4433a40ed3dSBarry Smith   PetscFunctionBegin;
4444482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
4454482741eSBarry Smith   PetscValidIntPointer(nfails,2);
44650ffb88aSMatthew Knepley   *nfails = snes->numFailures;
44750ffb88aSMatthew Knepley   PetscFunctionReturn(0);
44850ffb88aSMatthew Knepley }
44950ffb88aSMatthew Knepley 
45050ffb88aSMatthew Knepley #undef __FUNCT__
451b850b91aSLisandro Dalcin #define __FUNCT__ "SNESSetMaxNonlinearStepFailures"
45250ffb88aSMatthew Knepley /*@
453b850b91aSLisandro Dalcin    SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps
45450ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
45550ffb88aSMatthew Knepley 
45650ffb88aSMatthew Knepley    Not Collective
45750ffb88aSMatthew Knepley 
45850ffb88aSMatthew Knepley    Input Parameters:
45950ffb88aSMatthew Knepley +  snes     - SNES context
46050ffb88aSMatthew Knepley -  maxFails - maximum of unsuccessful steps
46150ffb88aSMatthew Knepley 
46250ffb88aSMatthew Knepley    Level: intermediate
46350ffb88aSMatthew Knepley 
46450ffb88aSMatthew Knepley .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
46550ffb88aSMatthew Knepley @*/
466b850b91aSLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails)
46750ffb88aSMatthew Knepley {
46850ffb88aSMatthew Knepley   PetscFunctionBegin;
4694482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
47050ffb88aSMatthew Knepley   snes->maxFailures = maxFails;
47150ffb88aSMatthew Knepley   PetscFunctionReturn(0);
47250ffb88aSMatthew Knepley }
47350ffb88aSMatthew Knepley 
47450ffb88aSMatthew Knepley #undef __FUNCT__
475b850b91aSLisandro Dalcin #define __FUNCT__ "SNESGetMaxNonlinearStepFailures"
47650ffb88aSMatthew Knepley /*@
477b850b91aSLisandro Dalcin    SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps
47850ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
47950ffb88aSMatthew Knepley 
48050ffb88aSMatthew Knepley    Not Collective
48150ffb88aSMatthew Knepley 
48250ffb88aSMatthew Knepley    Input Parameter:
48350ffb88aSMatthew Knepley .  snes     - SNES context
48450ffb88aSMatthew Knepley 
48550ffb88aSMatthew Knepley    Output Parameter:
48650ffb88aSMatthew Knepley .  maxFails - maximum of unsuccessful steps
48750ffb88aSMatthew Knepley 
48850ffb88aSMatthew Knepley    Level: intermediate
48950ffb88aSMatthew Knepley 
49050ffb88aSMatthew Knepley .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
49150ffb88aSMatthew Knepley @*/
492b850b91aSLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails)
49350ffb88aSMatthew Knepley {
49450ffb88aSMatthew Knepley   PetscFunctionBegin;
4954482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
4964482741eSBarry Smith   PetscValidIntPointer(maxFails,2);
49750ffb88aSMatthew Knepley   *maxFails = snes->maxFailures;
4983a40ed3dSBarry Smith   PetscFunctionReturn(0);
4999b94acceSBarry Smith }
500a847f771SSatish Balay 
5014a2ae208SSatish Balay #undef __FUNCT__
5023d4c4710SBarry Smith #define __FUNCT__ "SNESGetLinearSolveFailures"
5033d4c4710SBarry Smith /*@
5043d4c4710SBarry Smith    SNESGetLinearSolveFailures - Gets the number of failed (non-converged)
5053d4c4710SBarry Smith    linear solvers.
5063d4c4710SBarry Smith 
5073d4c4710SBarry Smith    Not Collective
5083d4c4710SBarry Smith 
5093d4c4710SBarry Smith    Input Parameter:
5103d4c4710SBarry Smith .  snes - SNES context
5113d4c4710SBarry Smith 
5123d4c4710SBarry Smith    Output Parameter:
5133d4c4710SBarry Smith .  nfails - number of failed solves
5143d4c4710SBarry Smith 
5153d4c4710SBarry Smith    Notes:
5163d4c4710SBarry Smith    This counter is reset to zero for each successive call to SNESSolve().
5173d4c4710SBarry Smith 
5183d4c4710SBarry Smith    Level: intermediate
5193d4c4710SBarry Smith 
5203d4c4710SBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps
5213d4c4710SBarry Smith @*/
5223d4c4710SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESGetLinearSolveFailures(SNES snes,PetscInt* nfails)
5233d4c4710SBarry Smith {
5243d4c4710SBarry Smith   PetscFunctionBegin;
5253d4c4710SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
5263d4c4710SBarry Smith   PetscValidIntPointer(nfails,2);
5273d4c4710SBarry Smith   *nfails = snes->numLinearSolveFailures;
5283d4c4710SBarry Smith   PetscFunctionReturn(0);
5293d4c4710SBarry Smith }
5303d4c4710SBarry Smith 
5313d4c4710SBarry Smith #undef __FUNCT__
5323d4c4710SBarry Smith #define __FUNCT__ "SNESSetMaxLinearSolveFailures"
5333d4c4710SBarry Smith /*@
5343d4c4710SBarry Smith    SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts
5353d4c4710SBarry Smith    allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE
5363d4c4710SBarry Smith 
5373d4c4710SBarry Smith    Collective on SNES
5383d4c4710SBarry Smith 
5393d4c4710SBarry Smith    Input Parameters:
5403d4c4710SBarry Smith +  snes     - SNES context
5413d4c4710SBarry Smith -  maxFails - maximum allowed linear solve failures
5423d4c4710SBarry Smith 
5433d4c4710SBarry Smith    Level: intermediate
5443d4c4710SBarry Smith 
5453d4c4710SBarry Smith    Notes: By default this is 1; that is SNES returns on the first failed linear solve
5463d4c4710SBarry Smith 
5473d4c4710SBarry Smith .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
5483d4c4710SBarry Smith 
5493d4c4710SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures()
5503d4c4710SBarry Smith @*/
5513d4c4710SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails)
5523d4c4710SBarry Smith {
5533d4c4710SBarry Smith   PetscFunctionBegin;
5543d4c4710SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
5553d4c4710SBarry Smith   snes->maxLinearSolveFailures = maxFails;
5563d4c4710SBarry Smith   PetscFunctionReturn(0);
5573d4c4710SBarry Smith }
5583d4c4710SBarry Smith 
5593d4c4710SBarry Smith #undef __FUNCT__
5603d4c4710SBarry Smith #define __FUNCT__ "SNESGetMaxLinearSolveFailures"
5613d4c4710SBarry Smith /*@
5623d4c4710SBarry Smith    SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that
5633d4c4710SBarry Smith      are allowed before SNES terminates
5643d4c4710SBarry Smith 
5653d4c4710SBarry Smith    Not Collective
5663d4c4710SBarry Smith 
5673d4c4710SBarry Smith    Input Parameter:
5683d4c4710SBarry Smith .  snes     - SNES context
5693d4c4710SBarry Smith 
5703d4c4710SBarry Smith    Output Parameter:
5713d4c4710SBarry Smith .  maxFails - maximum of unsuccessful solves allowed
5723d4c4710SBarry Smith 
5733d4c4710SBarry Smith    Level: intermediate
5743d4c4710SBarry Smith 
5753d4c4710SBarry Smith    Notes: By default this is 1; that is SNES returns on the first failed linear solve
5763d4c4710SBarry Smith 
5773d4c4710SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
5783d4c4710SBarry Smith 
5793d4c4710SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures()
5803d4c4710SBarry Smith @*/
5813d4c4710SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails)
5823d4c4710SBarry Smith {
5833d4c4710SBarry Smith   PetscFunctionBegin;
5843d4c4710SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
5853d4c4710SBarry Smith   PetscValidIntPointer(maxFails,2);
5863d4c4710SBarry Smith   *maxFails = snes->maxLinearSolveFailures;
5873d4c4710SBarry Smith   PetscFunctionReturn(0);
5883d4c4710SBarry Smith }
5893d4c4710SBarry Smith 
5903d4c4710SBarry Smith #undef __FUNCT__
591b850b91aSLisandro Dalcin #define __FUNCT__ "SNESGetLinearSolveIterations"
592c96a6f78SLois Curfman McInnes /*@
593b850b91aSLisandro Dalcin    SNESGetLinearSolveIterations - Gets the total number of linear iterations
594c96a6f78SLois Curfman McInnes    used by the nonlinear solver.
595c96a6f78SLois Curfman McInnes 
596c7afd0dbSLois Curfman McInnes    Not Collective
597c7afd0dbSLois Curfman McInnes 
598c96a6f78SLois Curfman McInnes    Input Parameter:
599c96a6f78SLois Curfman McInnes .  snes - SNES context
600c96a6f78SLois Curfman McInnes 
601c96a6f78SLois Curfman McInnes    Output Parameter:
602c96a6f78SLois Curfman McInnes .  lits - number of linear iterations
603c96a6f78SLois Curfman McInnes 
604c96a6f78SLois Curfman McInnes    Notes:
605c96a6f78SLois Curfman McInnes    This counter is reset to zero for each successive call to SNESSolve().
606c96a6f78SLois Curfman McInnes 
60736851e7fSLois Curfman McInnes    Level: intermediate
60836851e7fSLois Curfman McInnes 
609c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations
6102b668275SBarry Smith 
6112b668275SBarry Smith .seealso:  SNESGetIterationNumber(), SNESGetFunctionNorm()
612c96a6f78SLois Curfman McInnes @*/
613b850b91aSLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESGetLinearSolveIterations(SNES snes,PetscInt* lits)
614c96a6f78SLois Curfman McInnes {
6153a40ed3dSBarry Smith   PetscFunctionBegin;
6164482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
6174482741eSBarry Smith   PetscValidIntPointer(lits,2);
618c96a6f78SLois Curfman McInnes   *lits = snes->linear_its;
6193a40ed3dSBarry Smith   PetscFunctionReturn(0);
620c96a6f78SLois Curfman McInnes }
621c96a6f78SLois Curfman McInnes 
6224a2ae208SSatish Balay #undef __FUNCT__
62394b7f48cSBarry Smith #define __FUNCT__ "SNESGetKSP"
62452baeb72SSatish Balay /*@
62594b7f48cSBarry Smith    SNESGetKSP - Returns the KSP context for a SNES solver.
6269b94acceSBarry Smith 
62794b7f48cSBarry Smith    Not Collective, but if SNES object is parallel, then KSP object is parallel
628c7afd0dbSLois Curfman McInnes 
6299b94acceSBarry Smith    Input Parameter:
6309b94acceSBarry Smith .  snes - the SNES context
6319b94acceSBarry Smith 
6329b94acceSBarry Smith    Output Parameter:
63394b7f48cSBarry Smith .  ksp - the KSP context
6349b94acceSBarry Smith 
6359b94acceSBarry Smith    Notes:
63694b7f48cSBarry Smith    The user can then directly manipulate the KSP context to set various
6379b94acceSBarry Smith    options, etc.  Likewise, the user can then extract and manipulate the
6382999313aSBarry Smith    PC contexts as well.
6399b94acceSBarry Smith 
64036851e7fSLois Curfman McInnes    Level: beginner
64136851e7fSLois Curfman McInnes 
64294b7f48cSBarry Smith .keywords: SNES, nonlinear, get, KSP, context
6439b94acceSBarry Smith 
6442999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
6459b94acceSBarry Smith @*/
64663dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetKSP(SNES snes,KSP *ksp)
6479b94acceSBarry Smith {
6483a40ed3dSBarry Smith   PetscFunctionBegin;
6494482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
6504482741eSBarry Smith   PetscValidPointer(ksp,2);
65194b7f48cSBarry Smith   *ksp = snes->ksp;
6523a40ed3dSBarry Smith   PetscFunctionReturn(0);
6539b94acceSBarry Smith }
65482bf6240SBarry Smith 
6554a2ae208SSatish Balay #undef __FUNCT__
6562999313aSBarry Smith #define __FUNCT__ "SNESSetKSP"
6572999313aSBarry Smith /*@
6582999313aSBarry Smith    SNESSetKSP - Sets a KSP context for the SNES object to use
6592999313aSBarry Smith 
6602999313aSBarry Smith    Not Collective, but the SNES and KSP objects must live on the same MPI_Comm
6612999313aSBarry Smith 
6622999313aSBarry Smith    Input Parameters:
6632999313aSBarry Smith +  snes - the SNES context
6642999313aSBarry Smith -  ksp - the KSP context
6652999313aSBarry Smith 
6662999313aSBarry Smith    Notes:
6672999313aSBarry Smith    The SNES object already has its KSP object, you can obtain with SNESGetKSP()
6682999313aSBarry Smith    so this routine is rarely needed.
6692999313aSBarry Smith 
6702999313aSBarry Smith    The KSP object that is already in the SNES object has its reference count
6712999313aSBarry Smith    decreased by one.
6722999313aSBarry Smith 
6732999313aSBarry Smith    Level: developer
6742999313aSBarry Smith 
6752999313aSBarry Smith .keywords: SNES, nonlinear, get, KSP, context
6762999313aSBarry Smith 
6772999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
6782999313aSBarry Smith @*/
6792999313aSBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetKSP(SNES snes,KSP ksp)
6802999313aSBarry Smith {
6812999313aSBarry Smith   PetscErrorCode ierr;
6822999313aSBarry Smith 
6832999313aSBarry Smith   PetscFunctionBegin;
6842999313aSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
6852999313aSBarry Smith   PetscValidHeaderSpecific(ksp,KSP_COOKIE,2);
6862999313aSBarry Smith   PetscCheckSameComm(snes,1,ksp,2);
6877dcf0eaaSdalcinl   ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr);
688906ed7ccSBarry Smith   if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);}
6892999313aSBarry Smith   snes->ksp = ksp;
6902999313aSBarry Smith   PetscFunctionReturn(0);
6912999313aSBarry Smith }
6922999313aSBarry Smith 
6932999313aSBarry Smith #undef __FUNCT__
6944a2ae208SSatish Balay #define __FUNCT__ "SNESPublish_Petsc"
6956849ba73SBarry Smith static PetscErrorCode SNESPublish_Petsc(PetscObject obj)
696e24b481bSBarry Smith {
697e24b481bSBarry Smith   PetscFunctionBegin;
698e24b481bSBarry Smith   PetscFunctionReturn(0);
699e24b481bSBarry Smith }
700e24b481bSBarry Smith 
7019b94acceSBarry Smith /* -----------------------------------------------------------*/
7024a2ae208SSatish Balay #undef __FUNCT__
7034a2ae208SSatish Balay #define __FUNCT__ "SNESCreate"
70452baeb72SSatish Balay /*@
7059b94acceSBarry Smith    SNESCreate - Creates a nonlinear solver context.
7069b94acceSBarry Smith 
707c7afd0dbSLois Curfman McInnes    Collective on MPI_Comm
708c7afd0dbSLois Curfman McInnes 
709c7afd0dbSLois Curfman McInnes    Input Parameters:
710906ed7ccSBarry Smith .  comm - MPI communicator
7119b94acceSBarry Smith 
7129b94acceSBarry Smith    Output Parameter:
7139b94acceSBarry Smith .  outsnes - the new SNES context
7149b94acceSBarry Smith 
715c7afd0dbSLois Curfman McInnes    Options Database Keys:
716c7afd0dbSLois Curfman McInnes +   -snes_mf - Activates default matrix-free Jacobian-vector products,
717c7afd0dbSLois Curfman McInnes                and no preconditioning matrix
718c7afd0dbSLois Curfman McInnes .   -snes_mf_operator - Activates default matrix-free Jacobian-vector
719c7afd0dbSLois Curfman McInnes                products, and a user-provided preconditioning matrix
720c7afd0dbSLois Curfman McInnes                as set by SNESSetJacobian()
721c7afd0dbSLois Curfman McInnes -   -snes_fd - Uses (slow!) finite differences to compute Jacobian
722c1f60f51SBarry Smith 
72336851e7fSLois Curfman McInnes    Level: beginner
72436851e7fSLois Curfman McInnes 
7259b94acceSBarry Smith .keywords: SNES, nonlinear, create, context
7269b94acceSBarry Smith 
7274b27c08aSLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy(), SNES
7289b94acceSBarry Smith @*/
72963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESCreate(MPI_Comm comm,SNES *outsnes)
7309b94acceSBarry Smith {
731dfbe8321SBarry Smith   PetscErrorCode      ierr;
7329b94acceSBarry Smith   SNES                snes;
733fa9f3622SBarry Smith   SNESKSPEW           *kctx;
73437fcc0dbSBarry Smith 
7353a40ed3dSBarry Smith   PetscFunctionBegin;
736ed1caa07SMatthew Knepley   PetscValidPointer(outsnes,2);
7378ba1e511SMatthew Knepley   *outsnes = PETSC_NULL;
7388ba1e511SMatthew Knepley #ifndef PETSC_USE_DYNAMIC_LIBRARIES
7398ba1e511SMatthew Knepley   ierr = SNESInitializePackage(PETSC_NULL);CHKERRQ(ierr);
7408ba1e511SMatthew Knepley #endif
7418ba1e511SMatthew Knepley 
742e7788613SBarry Smith   ierr = PetscHeaderCreate(snes,_p_SNES,struct _SNESOps,SNES_COOKIE,0,"SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr);
743e24b481bSBarry Smith   snes->bops->publish     = SNESPublish_Petsc;
7449b94acceSBarry Smith   snes->max_its           = 50;
7459750a799SBarry Smith   snes->max_funcs	  = 10000;
7469b94acceSBarry Smith   snes->norm		  = 0.0;
747b4874afaSBarry Smith   snes->rtol		  = 1.e-8;
748b4874afaSBarry Smith   snes->ttol              = 0.0;
74970441072SBarry Smith   snes->abstol		  = 1.e-50;
7509b94acceSBarry Smith   snes->xtol		  = 1.e-8;
7514b27c08aSLois Curfman McInnes   snes->deltatol	  = 1.e-12;
7529b94acceSBarry Smith   snes->nfuncs            = 0;
75350ffb88aSMatthew Knepley   snes->numFailures       = 0;
75450ffb88aSMatthew Knepley   snes->maxFailures       = 1;
7557a00f4a9SLois Curfman McInnes   snes->linear_its        = 0;
756639f9d9dSBarry Smith   snes->numbermonitors    = 0;
7579b94acceSBarry Smith   snes->data              = 0;
7584dc4c822SBarry Smith   snes->setupcalled       = PETSC_FALSE;
759186905e3SBarry Smith   snes->ksp_ewconv        = PETSC_FALSE;
7606f24a144SLois Curfman McInnes   snes->vwork             = 0;
7616f24a144SLois Curfman McInnes   snes->nwork             = 0;
762758f92a0SBarry Smith   snes->conv_hist_len     = 0;
763758f92a0SBarry Smith   snes->conv_hist_max     = 0;
764758f92a0SBarry Smith   snes->conv_hist         = PETSC_NULL;
765758f92a0SBarry Smith   snes->conv_hist_its     = PETSC_NULL;
766758f92a0SBarry Smith   snes->conv_hist_reset   = PETSC_TRUE;
767184914b5SBarry Smith   snes->reason            = SNES_CONVERGED_ITERATING;
7689b94acceSBarry Smith 
7693d4c4710SBarry Smith   snes->numLinearSolveFailures = 0;
7703d4c4710SBarry Smith   snes->maxLinearSolveFailures = 1;
7713d4c4710SBarry Smith 
7729b94acceSBarry Smith   /* Create context to compute Eisenstat-Walker relative tolerance for KSP */
773*38f2d2fdSLisandro Dalcin   ierr = PetscNewLog(snes,SNESKSPEW,&kctx);CHKERRQ(ierr);
7749b94acceSBarry Smith   snes->kspconvctx  = (void*)kctx;
7759b94acceSBarry Smith   kctx->version     = 2;
7769b94acceSBarry Smith   kctx->rtol_0      = .3; /* Eisenstat and Walker suggest rtol_0=.5, but
7779b94acceSBarry Smith                              this was too large for some test cases */
7789b94acceSBarry Smith   kctx->rtol_last   = 0;
7799b94acceSBarry Smith   kctx->rtol_max    = .9;
7809b94acceSBarry Smith   kctx->gamma       = 1.0;
78171f87433Sdalcinl   kctx->alpha       = .5*(1.0 + sqrt(5.0));
78271f87433Sdalcinl   kctx->alpha2      = kctx->alpha;
7839b94acceSBarry Smith   kctx->threshold   = .1;
7849b94acceSBarry Smith   kctx->lresid_last = 0;
7859b94acceSBarry Smith   kctx->norm_last   = 0;
7869b94acceSBarry Smith 
78794b7f48cSBarry Smith   ierr = KSPCreate(comm,&snes->ksp);CHKERRQ(ierr);
78852e6d16bSBarry Smith   ierr = PetscLogObjectParent(snes,snes->ksp);CHKERRQ(ierr);
7895334005bSBarry Smith 
7909b94acceSBarry Smith   *outsnes = snes;
79100036973SBarry Smith   ierr = PetscPublishAll(snes);CHKERRQ(ierr);
7923a40ed3dSBarry Smith   PetscFunctionReturn(0);
7939b94acceSBarry Smith }
7949b94acceSBarry Smith 
7954a2ae208SSatish Balay #undef __FUNCT__
7964a2ae208SSatish Balay #define __FUNCT__ "SNESSetFunction"
7979b94acceSBarry Smith /*@C
7989b94acceSBarry Smith    SNESSetFunction - Sets the function evaluation routine and function
7999b94acceSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
8009b94acceSBarry Smith    equations.
8019b94acceSBarry Smith 
802fee21e36SBarry Smith    Collective on SNES
803fee21e36SBarry Smith 
804c7afd0dbSLois Curfman McInnes    Input Parameters:
805c7afd0dbSLois Curfman McInnes +  snes - the SNES context
806c7afd0dbSLois Curfman McInnes .  r - vector to store function value
807de044059SHong Zhang .  func - function evaluation routine
808c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
809c7afd0dbSLois Curfman McInnes          function evaluation routine (may be PETSC_NULL)
8109b94acceSBarry Smith 
811c7afd0dbSLois Curfman McInnes    Calling sequence of func:
8128d76a1e5SLois Curfman McInnes $    func (SNES snes,Vec x,Vec f,void *ctx);
813c7afd0dbSLois Curfman McInnes 
814313e4042SLois Curfman McInnes .  f - function vector
815c7afd0dbSLois Curfman McInnes -  ctx - optional user-defined function context
8169b94acceSBarry Smith 
8179b94acceSBarry Smith    Notes:
8189b94acceSBarry Smith    The Newton-like methods typically solve linear systems of the form
8199b94acceSBarry Smith $      f'(x) x = -f(x),
820c7afd0dbSLois Curfman McInnes    where f'(x) denotes the Jacobian matrix and f(x) is the function.
8219b94acceSBarry Smith 
82236851e7fSLois Curfman McInnes    Level: beginner
82336851e7fSLois Curfman McInnes 
8249b94acceSBarry Smith .keywords: SNES, nonlinear, set, function
8259b94acceSBarry Smith 
826a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian()
8279b94acceSBarry Smith @*/
82863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*func)(SNES,Vec,Vec,void*),void *ctx)
8299b94acceSBarry Smith {
8303a40ed3dSBarry Smith   PetscFunctionBegin;
8314482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
8324482741eSBarry Smith   PetscValidHeaderSpecific(r,VEC_COOKIE,2);
833c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,r,2);
834184914b5SBarry Smith 
835e7788613SBarry Smith   snes->ops->computefunction = func;
8369b94acceSBarry Smith   snes->vec_func             = snes->vec_func_always = r;
8379b94acceSBarry Smith   snes->funP                 = ctx;
8383a40ed3dSBarry Smith   PetscFunctionReturn(0);
8399b94acceSBarry Smith }
8409b94acceSBarry Smith 
8413ab0aad5SBarry Smith /* --------------------------------------------------------------- */
8423ab0aad5SBarry Smith #undef __FUNCT__
8433ab0aad5SBarry Smith #define __FUNCT__ "SNESSetRhs"
8443ab0aad5SBarry Smith /*@C
8453ab0aad5SBarry Smith    SNESSetRhs - Sets the vector for solving F(x) = rhs. If rhs is not set
8463ab0aad5SBarry Smith    it assumes a zero right hand side.
8473ab0aad5SBarry Smith 
8483ab0aad5SBarry Smith    Collective on SNES
8493ab0aad5SBarry Smith 
8503ab0aad5SBarry Smith    Input Parameters:
8513ab0aad5SBarry Smith +  snes - the SNES context
8523ab0aad5SBarry Smith -  rhs - the right hand side vector or PETSC_NULL for a zero right hand side
8533ab0aad5SBarry Smith 
8543ab0aad5SBarry Smith    Level: intermediate
8553ab0aad5SBarry Smith 
8563ab0aad5SBarry Smith .keywords: SNES, nonlinear, set, function, right hand side
8573ab0aad5SBarry Smith 
8581096aae1SMatthew Knepley .seealso: SNESGetRhs(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
8593ab0aad5SBarry Smith @*/
86063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetRhs(SNES snes,Vec rhs)
8613ab0aad5SBarry Smith {
862dfbe8321SBarry Smith   PetscErrorCode ierr;
8633ab0aad5SBarry Smith 
8643ab0aad5SBarry Smith   PetscFunctionBegin;
8653ab0aad5SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
8663ab0aad5SBarry Smith   if (rhs) {
8673ab0aad5SBarry Smith     PetscValidHeaderSpecific(rhs,VEC_COOKIE,2);
8683ab0aad5SBarry Smith     PetscCheckSameComm(snes,1,rhs,2);
8693ab0aad5SBarry Smith     ierr = PetscObjectReference((PetscObject)rhs);CHKERRQ(ierr);
8703ab0aad5SBarry Smith   }
8713ab0aad5SBarry Smith   if (snes->afine) {
8723ab0aad5SBarry Smith     ierr = VecDestroy(snes->afine);CHKERRQ(ierr);
8733ab0aad5SBarry Smith   }
8743ab0aad5SBarry Smith   snes->afine = rhs;
8753ab0aad5SBarry Smith   PetscFunctionReturn(0);
8763ab0aad5SBarry Smith }
8773ab0aad5SBarry Smith 
8784a2ae208SSatish Balay #undef __FUNCT__
8791096aae1SMatthew Knepley #define __FUNCT__ "SNESGetRhs"
8801096aae1SMatthew Knepley /*@C
8811096aae1SMatthew Knepley    SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set
8821096aae1SMatthew Knepley    it assumes a zero right hand side.
8831096aae1SMatthew Knepley 
8841096aae1SMatthew Knepley    Collective on SNES
8851096aae1SMatthew Knepley 
8861096aae1SMatthew Knepley    Input Parameter:
8871096aae1SMatthew Knepley .  snes - the SNES context
8881096aae1SMatthew Knepley 
8891096aae1SMatthew Knepley    Output Parameter:
8901096aae1SMatthew Knepley .  rhs - the right hand side vector or PETSC_NULL for a zero right hand side
8911096aae1SMatthew Knepley 
8921096aae1SMatthew Knepley    Level: intermediate
8931096aae1SMatthew Knepley 
8941096aae1SMatthew Knepley .keywords: SNES, nonlinear, get, function, right hand side
8951096aae1SMatthew Knepley 
8961096aae1SMatthew Knepley .seealso: SNESSetRhs(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
8971096aae1SMatthew Knepley @*/
8981096aae1SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESGetRhs(SNES snes,Vec *rhs)
8991096aae1SMatthew Knepley {
9001096aae1SMatthew Knepley   PetscFunctionBegin;
9011096aae1SMatthew Knepley   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
9021096aae1SMatthew Knepley   PetscValidPointer(rhs,2);
9031096aae1SMatthew Knepley   *rhs = snes->afine;
9041096aae1SMatthew Knepley   PetscFunctionReturn(0);
9051096aae1SMatthew Knepley }
9061096aae1SMatthew Knepley 
9071096aae1SMatthew Knepley #undef __FUNCT__
9084a2ae208SSatish Balay #define __FUNCT__ "SNESComputeFunction"
9099b94acceSBarry Smith /*@
91036851e7fSLois Curfman McInnes    SNESComputeFunction - Calls the function that has been set with
9119b94acceSBarry Smith                          SNESSetFunction().
9129b94acceSBarry Smith 
913c7afd0dbSLois Curfman McInnes    Collective on SNES
914c7afd0dbSLois Curfman McInnes 
9159b94acceSBarry Smith    Input Parameters:
916c7afd0dbSLois Curfman McInnes +  snes - the SNES context
917c7afd0dbSLois Curfman McInnes -  x - input vector
9189b94acceSBarry Smith 
9199b94acceSBarry Smith    Output Parameter:
9203638b69dSLois Curfman McInnes .  y - function vector, as set by SNESSetFunction()
9219b94acceSBarry Smith 
9221bffabb2SLois Curfman McInnes    Notes:
92336851e7fSLois Curfman McInnes    SNESComputeFunction() is typically used within nonlinear solvers
92436851e7fSLois Curfman McInnes    implementations, so most users would not generally call this routine
92536851e7fSLois Curfman McInnes    themselves.
92636851e7fSLois Curfman McInnes 
92736851e7fSLois Curfman McInnes    Level: developer
92836851e7fSLois Curfman McInnes 
9299b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function
9309b94acceSBarry Smith 
931a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction()
9329b94acceSBarry Smith @*/
93363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESComputeFunction(SNES snes,Vec x,Vec y)
9349b94acceSBarry Smith {
935dfbe8321SBarry Smith   PetscErrorCode ierr;
9369b94acceSBarry Smith 
9373a40ed3dSBarry Smith   PetscFunctionBegin;
9384482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
9394482741eSBarry Smith   PetscValidHeaderSpecific(x,VEC_COOKIE,2);
9404482741eSBarry Smith   PetscValidHeaderSpecific(y,VEC_COOKIE,3);
941c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,x,2);
942c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,3);
943184914b5SBarry Smith 
944d5ba7fb7SMatthew Knepley   ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
945e7788613SBarry Smith   if (snes->ops->computefunction) {
946d64ed03dSBarry Smith     PetscStackPush("SNES user function");
947e9a2bbcdSBarry Smith     CHKMEMQ;
948e7788613SBarry Smith     ierr = (*snes->ops->computefunction)(snes,x,y,snes->funP);
949e9a2bbcdSBarry Smith     CHKMEMQ;
950d64ed03dSBarry Smith     PetscStackPop;
951d5e45103SBarry Smith     if (PetscExceptionValue(ierr)) {
95219717074SBarry Smith       PetscErrorCode pierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(pierr);
95319717074SBarry Smith     }
954d5e45103SBarry Smith     CHKERRQ(ierr);
9551096aae1SMatthew Knepley   } else if (snes->afine) {
9561096aae1SMatthew Knepley     ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr);
9571096aae1SMatthew Knepley   } else {
9581096aae1SMatthew Knepley     SETERRQ(PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() before SNESComputeFunction(), likely called from SNESSolve().");
9591096aae1SMatthew Knepley   }
9603ab0aad5SBarry Smith   if (snes->afine) {
961016dedfbSBarry Smith     ierr = VecAXPY(y,-1.0,snes->afine);CHKERRQ(ierr);
9623ab0aad5SBarry Smith   }
963ae3c334cSLois Curfman McInnes   snes->nfuncs++;
964d5ba7fb7SMatthew Knepley   ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
9653a40ed3dSBarry Smith   PetscFunctionReturn(0);
9669b94acceSBarry Smith }
9679b94acceSBarry Smith 
9684a2ae208SSatish Balay #undef __FUNCT__
9694a2ae208SSatish Balay #define __FUNCT__ "SNESComputeJacobian"
97062fef451SLois Curfman McInnes /*@
97162fef451SLois Curfman McInnes    SNESComputeJacobian - Computes the Jacobian matrix that has been
97262fef451SLois Curfman McInnes    set with SNESSetJacobian().
97362fef451SLois Curfman McInnes 
974c7afd0dbSLois Curfman McInnes    Collective on SNES and Mat
975c7afd0dbSLois Curfman McInnes 
97662fef451SLois Curfman McInnes    Input Parameters:
977c7afd0dbSLois Curfman McInnes +  snes - the SNES context
978c7afd0dbSLois Curfman McInnes -  x - input vector
97962fef451SLois Curfman McInnes 
98062fef451SLois Curfman McInnes    Output Parameters:
981c7afd0dbSLois Curfman McInnes +  A - Jacobian matrix
98262fef451SLois Curfman McInnes .  B - optional preconditioning matrix
9832b668275SBarry Smith -  flag - flag indicating matrix structure (one of, SAME_NONZERO_PATTERN,DIFFERENT_NONZERO_PATTERN,SAME_PRECONDITIONER)
984fee21e36SBarry Smith 
98562fef451SLois Curfman McInnes    Notes:
98662fef451SLois Curfman McInnes    Most users should not need to explicitly call this routine, as it
98762fef451SLois Curfman McInnes    is used internally within the nonlinear solvers.
98862fef451SLois Curfman McInnes 
98994b7f48cSBarry Smith    See KSPSetOperators() for important information about setting the
990dc5a77f8SLois Curfman McInnes    flag parameter.
99162fef451SLois Curfman McInnes 
99236851e7fSLois Curfman McInnes    Level: developer
99336851e7fSLois Curfman McInnes 
99462fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix
99562fef451SLois Curfman McInnes 
9962b668275SBarry Smith .seealso:  SNESSetJacobian(), KSPSetOperators(), MatStructure
99762fef451SLois Curfman McInnes @*/
99863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg)
9999b94acceSBarry Smith {
1000dfbe8321SBarry Smith   PetscErrorCode ierr;
10013a40ed3dSBarry Smith 
10023a40ed3dSBarry Smith   PetscFunctionBegin;
10034482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
10044482741eSBarry Smith   PetscValidHeaderSpecific(X,VEC_COOKIE,2);
10054482741eSBarry Smith   PetscValidPointer(flg,5);
1006c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,X,2);
1007e7788613SBarry Smith   if (!snes->ops->computejacobian) PetscFunctionReturn(0);
1008d5ba7fb7SMatthew Knepley   ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,*A,*B);CHKERRQ(ierr);
1009c4fc05e7SBarry Smith   *flg = DIFFERENT_NONZERO_PATTERN;
1010d64ed03dSBarry Smith   PetscStackPush("SNES user Jacobian function");
1011dc67937bSBarry Smith   CHKMEMQ;
1012e7788613SBarry Smith   ierr = (*snes->ops->computejacobian)(snes,X,A,B,flg,snes->jacP);CHKERRQ(ierr);
1013dc67937bSBarry Smith   CHKMEMQ;
1014d64ed03dSBarry Smith   PetscStackPop;
1015d5ba7fb7SMatthew Knepley   ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,*A,*B);CHKERRQ(ierr);
10166d84be18SBarry Smith   /* make sure user returned a correct Jacobian and preconditioner */
10176ce558aeSBarry Smith   /* PetscValidHeaderSpecific(*A,MAT_COOKIE,3);
10186ce558aeSBarry Smith     PetscValidHeaderSpecific(*B,MAT_COOKIE,4);   */
10193a40ed3dSBarry Smith   PetscFunctionReturn(0);
10209b94acceSBarry Smith }
10219b94acceSBarry Smith 
10224a2ae208SSatish Balay #undef __FUNCT__
10234a2ae208SSatish Balay #define __FUNCT__ "SNESSetJacobian"
10249b94acceSBarry Smith /*@C
10259b94acceSBarry Smith    SNESSetJacobian - Sets the function to compute Jacobian as well as the
1026044dda88SLois Curfman McInnes    location to store the matrix.
10279b94acceSBarry Smith 
1028c7afd0dbSLois Curfman McInnes    Collective on SNES and Mat
1029c7afd0dbSLois Curfman McInnes 
10309b94acceSBarry Smith    Input Parameters:
1031c7afd0dbSLois Curfman McInnes +  snes - the SNES context
10329b94acceSBarry Smith .  A - Jacobian matrix
10339b94acceSBarry Smith .  B - preconditioner matrix (usually same as the Jacobian)
10349b94acceSBarry Smith .  func - Jacobian evaluation routine
1035c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
10362cd2dfdcSLois Curfman McInnes          Jacobian evaluation routine (may be PETSC_NULL)
10379b94acceSBarry Smith 
10389b94acceSBarry Smith    Calling sequence of func:
10398d76a1e5SLois Curfman McInnes $     func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx);
10409b94acceSBarry Smith 
1041c7afd0dbSLois Curfman McInnes +  x - input vector
10429b94acceSBarry Smith .  A - Jacobian matrix
10439b94acceSBarry Smith .  B - preconditioner matrix, usually the same as A
1044ac21db08SLois Curfman McInnes .  flag - flag indicating information about the preconditioner matrix
10452b668275SBarry Smith    structure (same as flag in KSPSetOperators()), one of SAME_NONZERO_PATTERN,DIFFERENT_NONZERO_PATTERN,SAME_PRECONDITIONER
1046c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined Jacobian context
10479b94acceSBarry Smith 
10489b94acceSBarry Smith    Notes:
104994b7f48cSBarry Smith    See KSPSetOperators() for important information about setting the flag
10502cd2dfdcSLois Curfman McInnes    output parameter in the routine func().  Be sure to read this information!
1051ac21db08SLois Curfman McInnes 
1052ac21db08SLois Curfman McInnes    The routine func() takes Mat * as the matrix arguments rather than Mat.
10539b94acceSBarry Smith    This allows the Jacobian evaluation routine to replace A and/or B with a
10549b94acceSBarry Smith    completely new new matrix structure (not just different matrix elements)
10559b94acceSBarry Smith    when appropriate, for instance, if the nonzero structure is changing
10569b94acceSBarry Smith    throughout the global iterations.
10579b94acceSBarry Smith 
105836851e7fSLois Curfman McInnes    Level: beginner
105936851e7fSLois Curfman McInnes 
10609b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix
10619b94acceSBarry Smith 
10623ec795f1SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESDefaultComputeJacobianColor(), MatStructure
10639b94acceSBarry Smith @*/
106463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetJacobian(SNES snes,Mat A,Mat B,PetscErrorCode (*func)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void *ctx)
10659b94acceSBarry Smith {
1066dfbe8321SBarry Smith   PetscErrorCode ierr;
10673a7fca6bSBarry Smith 
10683a40ed3dSBarry Smith   PetscFunctionBegin;
10694482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
10704482741eSBarry Smith   if (A) PetscValidHeaderSpecific(A,MAT_COOKIE,2);
10714482741eSBarry Smith   if (B) PetscValidHeaderSpecific(B,MAT_COOKIE,3);
1072c9780b6fSBarry Smith   if (A) PetscCheckSameComm(snes,1,A,2);
1073c9780b6fSBarry Smith   if (B) PetscCheckSameComm(snes,1,B,2);
1074e7788613SBarry Smith   if (func) snes->ops->computejacobian = func;
10753a7fca6bSBarry Smith   if (ctx)  snes->jacP                 = ctx;
10763a7fca6bSBarry Smith   if (A) {
10777dcf0eaaSdalcinl     ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
10783a7fca6bSBarry Smith     if (snes->jacobian) {ierr = MatDestroy(snes->jacobian);CHKERRQ(ierr);}
10799b94acceSBarry Smith     snes->jacobian = A;
10803a7fca6bSBarry Smith   }
10813a7fca6bSBarry Smith   if (B) {
10827dcf0eaaSdalcinl     ierr = PetscObjectReference((PetscObject)B);CHKERRQ(ierr);
10833a7fca6bSBarry Smith     if (snes->jacobian_pre) {ierr = MatDestroy(snes->jacobian_pre);CHKERRQ(ierr);}
10849b94acceSBarry Smith     snes->jacobian_pre = B;
10853a7fca6bSBarry Smith   }
10863a40ed3dSBarry Smith   PetscFunctionReturn(0);
10879b94acceSBarry Smith }
108862fef451SLois Curfman McInnes 
10894a2ae208SSatish Balay #undef __FUNCT__
10904a2ae208SSatish Balay #define __FUNCT__ "SNESGetJacobian"
1091c2aafc4cSSatish Balay /*@C
1092b4fd4287SBarry Smith    SNESGetJacobian - Returns the Jacobian matrix and optionally the user
1093b4fd4287SBarry Smith    provided context for evaluating the Jacobian.
1094b4fd4287SBarry Smith 
1095c7afd0dbSLois Curfman McInnes    Not Collective, but Mat object will be parallel if SNES object is
1096c7afd0dbSLois Curfman McInnes 
1097b4fd4287SBarry Smith    Input Parameter:
1098b4fd4287SBarry Smith .  snes - the nonlinear solver context
1099b4fd4287SBarry Smith 
1100b4fd4287SBarry Smith    Output Parameters:
1101c7afd0dbSLois Curfman McInnes +  A - location to stash Jacobian matrix (or PETSC_NULL)
1102b4fd4287SBarry Smith .  B - location to stash preconditioner matrix (or PETSC_NULL)
110370e92668SMatthew Knepley .  func - location to put Jacobian function (or PETSC_NULL)
110470e92668SMatthew Knepley -  ctx - location to stash Jacobian ctx (or PETSC_NULL)
1105fee21e36SBarry Smith 
110636851e7fSLois Curfman McInnes    Level: advanced
110736851e7fSLois Curfman McInnes 
1108b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian()
1109b4fd4287SBarry Smith @*/
111070e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESGetJacobian(SNES snes,Mat *A,Mat *B,PetscErrorCode (**func)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void **ctx)
1111b4fd4287SBarry Smith {
11123a40ed3dSBarry Smith   PetscFunctionBegin;
11134482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1114b4fd4287SBarry Smith   if (A)    *A    = snes->jacobian;
1115b4fd4287SBarry Smith   if (B)    *B    = snes->jacobian_pre;
1116e7788613SBarry Smith   if (func) *func = snes->ops->computejacobian;
111770e92668SMatthew Knepley   if (ctx)  *ctx  = snes->jacP;
11183a40ed3dSBarry Smith   PetscFunctionReturn(0);
1119b4fd4287SBarry Smith }
1120b4fd4287SBarry Smith 
11219b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */
112263dd3a1aSKris Buschelman EXTERN PetscErrorCode PETSCSNES_DLLEXPORT SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*);
11239b94acceSBarry Smith 
11244a2ae208SSatish Balay #undef __FUNCT__
11254a2ae208SSatish Balay #define __FUNCT__ "SNESSetUp"
11269b94acceSBarry Smith /*@
11279b94acceSBarry Smith    SNESSetUp - Sets up the internal data structures for the later use
1128272ac6f2SLois Curfman McInnes    of a nonlinear solver.
11299b94acceSBarry Smith 
1130fee21e36SBarry Smith    Collective on SNES
1131fee21e36SBarry Smith 
1132c7afd0dbSLois Curfman McInnes    Input Parameters:
113370e92668SMatthew Knepley .  snes - the SNES context
1134c7afd0dbSLois Curfman McInnes 
1135272ac6f2SLois Curfman McInnes    Notes:
1136272ac6f2SLois Curfman McInnes    For basic use of the SNES solvers the user need not explicitly call
1137272ac6f2SLois Curfman McInnes    SNESSetUp(), since these actions will automatically occur during
1138272ac6f2SLois Curfman McInnes    the call to SNESSolve().  However, if one wishes to control this
1139272ac6f2SLois Curfman McInnes    phase separately, SNESSetUp() should be called after SNESCreate()
1140272ac6f2SLois Curfman McInnes    and optional routines of the form SNESSetXXX(), but before SNESSolve().
1141272ac6f2SLois Curfman McInnes 
114236851e7fSLois Curfman McInnes    Level: advanced
114336851e7fSLois Curfman McInnes 
11449b94acceSBarry Smith .keywords: SNES, nonlinear, setup
11459b94acceSBarry Smith 
11469b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy()
11479b94acceSBarry Smith @*/
114870e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESSetUp(SNES snes)
11499b94acceSBarry Smith {
1150dfbe8321SBarry Smith   PetscErrorCode ierr;
115171f87433Sdalcinl   PetscTruth     flg;
11523a40ed3dSBarry Smith 
11533a40ed3dSBarry Smith   PetscFunctionBegin;
11544482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
11554dc4c822SBarry Smith   if (snes->setupcalled) PetscFunctionReturn(0);
11569b94acceSBarry Smith 
1157b0a32e0cSBarry Smith   ierr = PetscOptionsHasName(snes->prefix,"-snes_mf_operator",&flg);CHKERRQ(ierr);
1158c1f60f51SBarry Smith   /*
1159c1f60f51SBarry Smith       This version replaces the user provided Jacobian matrix with a
1160dfa02198SLois Curfman McInnes       matrix-free version but still employs the user-provided preconditioner matrix
1161c1f60f51SBarry Smith   */
1162c1f60f51SBarry Smith   if (flg) {
1163c1f60f51SBarry Smith     Mat J;
1164fef1beadSBarry Smith     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
11653ec795f1SBarry Smith     ierr = MatMFFDSetFromOptions(J);CHKERRQ(ierr);
1166ae15b995SBarry Smith     ierr = PetscInfo(snes,"Setting default matrix-free operator routines\n");CHKERRQ(ierr);
11673a7fca6bSBarry Smith     ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);
11683a7fca6bSBarry Smith     ierr = MatDestroy(J);CHKERRQ(ierr);
1169c1f60f51SBarry Smith   }
117045fc7adcSBarry Smith 
117103c60df9SBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_SINGLE) && !defined(PETSC_USE_MAT_SINGLE) && !defined(PETSC_USE_LONG_DOUBLE) && !defined(PETSC_USE_INT)
117245fc7adcSBarry Smith   ierr = PetscOptionsHasName(snes->prefix,"-snes_mf_operator2",&flg);CHKERRQ(ierr);
117345fc7adcSBarry Smith   if (flg) {
117445fc7adcSBarry Smith     Mat J;
117545fc7adcSBarry Smith     ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_sol,&J);CHKERRQ(ierr);
117675396ef9SLisandro Dalcin     ierr = PetscInfo(snes,"Setting default matrix-free operator routines (version 2)\n");CHKERRQ(ierr);
117745fc7adcSBarry Smith     ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);
117845fc7adcSBarry Smith     ierr = MatDestroy(J);CHKERRQ(ierr);
117945fc7adcSBarry Smith   }
118032a4b47aSMatthew Knepley #endif
118145fc7adcSBarry Smith 
1182b0a32e0cSBarry Smith   ierr = PetscOptionsHasName(snes->prefix,"-snes_mf",&flg);CHKERRQ(ierr);
1183c1f60f51SBarry Smith   /*
1184dfa02198SLois Curfman McInnes       This version replaces both the user-provided Jacobian and the user-
1185c1f60f51SBarry Smith       provided preconditioner matrix with the default matrix free version.
1186c1f60f51SBarry Smith    */
118731615d04SLois Curfman McInnes   if (flg) {
1188272ac6f2SLois Curfman McInnes     Mat  J;
1189b5d62d44SBarry Smith     KSP ksp;
119094b7f48cSBarry Smith     PC   pc;
119175396ef9SLisandro Dalcin     /* create and set matrix-free operator */
1192fef1beadSBarry Smith     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
11933ec795f1SBarry Smith     ierr = MatMFFDSetFromOptions(J);CHKERRQ(ierr);
119475396ef9SLisandro Dalcin     ierr = PetscInfo(snes,"Setting default matrix-free operator routines\n");CHKERRQ(ierr);
11953ec795f1SBarry Smith     ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,snes->funP);CHKERRQ(ierr);
11963a7fca6bSBarry Smith     ierr = MatDestroy(J);CHKERRQ(ierr);
1197f3ef73ceSBarry Smith     /* force no preconditioner */
119894b7f48cSBarry Smith     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
1199b5d62d44SBarry Smith     ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
1200a9815358SBarry Smith     ierr = PetscTypeCompare((PetscObject)pc,PCSHELL,&flg);CHKERRQ(ierr);
1201a9815358SBarry Smith     if (!flg) {
120275396ef9SLisandro Dalcin       ierr = PetscInfo(snes,"Setting default matrix-free preconditioner routines;\nThat is no preconditioner is being used\n");CHKERRQ(ierr);
1203f3ef73ceSBarry Smith       ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr);
1204272ac6f2SLois Curfman McInnes     }
1205a9815358SBarry Smith   }
1206f3ef73ceSBarry Smith 
12071096aae1SMatthew Knepley   if (!snes->vec_func && !snes->afine) {
12081096aae1SMatthew Knepley     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetFunction() first");
12091096aae1SMatthew Knepley   }
1210e7788613SBarry Smith   if (!snes->ops->computefunction && !snes->afine) {
12111096aae1SMatthew Knepley     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetFunction() first");
12121096aae1SMatthew Knepley   }
121375396ef9SLisandro Dalcin   if (!snes->jacobian) {
121475396ef9SLisandro Dalcin     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetJacobian() first \n or use -snes_mf option");
121575396ef9SLisandro Dalcin   }
1216a8c6a408SBarry Smith   if (snes->vec_func == snes->vec_sol) {
121729bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector");
1218a8c6a408SBarry Smith   }
1219a703fe33SLois Curfman McInnes 
1220410397dcSLisandro Dalcin   if (!snes->type_name) {
1221410397dcSLisandro Dalcin     ierr = SNESSetType(snes,SNESLS);CHKERRQ(ierr);
1222410397dcSLisandro Dalcin   }
1223410397dcSLisandro Dalcin   if (snes->ops->setup) {
1224410397dcSLisandro Dalcin     ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr);
1225410397dcSLisandro Dalcin   }
12267aaed0d8SBarry Smith   snes->setupcalled = PETSC_TRUE;
12273a40ed3dSBarry Smith   PetscFunctionReturn(0);
12289b94acceSBarry Smith }
12299b94acceSBarry Smith 
12304a2ae208SSatish Balay #undef __FUNCT__
12314a2ae208SSatish Balay #define __FUNCT__ "SNESDestroy"
123252baeb72SSatish Balay /*@
12339b94acceSBarry Smith    SNESDestroy - Destroys the nonlinear solver context that was created
12349b94acceSBarry Smith    with SNESCreate().
12359b94acceSBarry Smith 
1236c7afd0dbSLois Curfman McInnes    Collective on SNES
1237c7afd0dbSLois Curfman McInnes 
12389b94acceSBarry Smith    Input Parameter:
12399b94acceSBarry Smith .  snes - the SNES context
12409b94acceSBarry Smith 
124136851e7fSLois Curfman McInnes    Level: beginner
124236851e7fSLois Curfman McInnes 
12439b94acceSBarry Smith .keywords: SNES, nonlinear, destroy
12449b94acceSBarry Smith 
124563a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve()
12469b94acceSBarry Smith @*/
124763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESDestroy(SNES snes)
12489b94acceSBarry Smith {
12496849ba73SBarry Smith   PetscErrorCode ierr;
12503a40ed3dSBarry Smith 
12513a40ed3dSBarry Smith   PetscFunctionBegin;
12524482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
12533a40ed3dSBarry Smith   if (--snes->refct > 0) PetscFunctionReturn(0);
1254d4bb536fSBarry Smith 
1255be0abb6dSBarry Smith   /* if memory was published with AMS then destroy it */
12560f5bd95cSBarry Smith   ierr = PetscObjectDepublish(snes);CHKERRQ(ierr);
1257be0abb6dSBarry Smith 
1258e7788613SBarry Smith   if (snes->ops->destroy) {ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr);}
125905b42c5fSBarry Smith   ierr = PetscFree(snes->kspconvctx);CHKERRQ(ierr);
12603a7fca6bSBarry Smith   if (snes->jacobian) {ierr = MatDestroy(snes->jacobian);CHKERRQ(ierr);}
12613a7fca6bSBarry Smith   if (snes->jacobian_pre) {ierr = MatDestroy(snes->jacobian_pre);CHKERRQ(ierr);}
12623ab0aad5SBarry Smith   if (snes->afine) {ierr = VecDestroy(snes->afine);CHKERRQ(ierr);}
126394b7f48cSBarry Smith   ierr = KSPDestroy(snes->ksp);CHKERRQ(ierr);
1264522c5e43SBarry Smith   if (snes->vwork) {ierr = VecDestroyVecs(snes->vwork,snes->nvwork);CHKERRQ(ierr);}
1265a6570f20SBarry Smith   ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);
1266a79aaaedSSatish Balay   ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr);
12673a40ed3dSBarry Smith   PetscFunctionReturn(0);
12689b94acceSBarry Smith }
12699b94acceSBarry Smith 
12709b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */
12719b94acceSBarry Smith 
12724a2ae208SSatish Balay #undef __FUNCT__
12734a2ae208SSatish Balay #define __FUNCT__ "SNESSetTolerances"
12749b94acceSBarry Smith /*@
1275d7a720efSLois Curfman McInnes    SNESSetTolerances - Sets various parameters used in convergence tests.
12769b94acceSBarry Smith 
1277c7afd0dbSLois Curfman McInnes    Collective on SNES
1278c7afd0dbSLois Curfman McInnes 
12799b94acceSBarry Smith    Input Parameters:
1280c7afd0dbSLois Curfman McInnes +  snes - the SNES context
128170441072SBarry Smith .  abstol - absolute convergence tolerance
128233174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
128333174efeSLois Curfman McInnes .  stol -  convergence tolerance in terms of the norm
128433174efeSLois Curfman McInnes            of the change in the solution between steps
128533174efeSLois Curfman McInnes .  maxit - maximum number of iterations
1286c7afd0dbSLois Curfman McInnes -  maxf - maximum number of function evaluations
1287fee21e36SBarry Smith 
128833174efeSLois Curfman McInnes    Options Database Keys:
128970441072SBarry Smith +    -snes_atol <abstol> - Sets abstol
1290c7afd0dbSLois Curfman McInnes .    -snes_rtol <rtol> - Sets rtol
1291c7afd0dbSLois Curfman McInnes .    -snes_stol <stol> - Sets stol
1292c7afd0dbSLois Curfman McInnes .    -snes_max_it <maxit> - Sets maxit
1293c7afd0dbSLois Curfman McInnes -    -snes_max_funcs <maxf> - Sets maxf
12949b94acceSBarry Smith 
1295d7a720efSLois Curfman McInnes    Notes:
12969b94acceSBarry Smith    The default maximum number of iterations is 50.
12979b94acceSBarry Smith    The default maximum number of function evaluations is 1000.
12989b94acceSBarry Smith 
129936851e7fSLois Curfman McInnes    Level: intermediate
130036851e7fSLois Curfman McInnes 
130133174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances
13029b94acceSBarry Smith 
13032492ecdbSBarry Smith .seealso: SNESSetTrustRegionTolerance()
13049b94acceSBarry Smith @*/
130563dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf)
13069b94acceSBarry Smith {
13073a40ed3dSBarry Smith   PetscFunctionBegin;
13084482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
130970441072SBarry Smith   if (abstol != PETSC_DEFAULT)  snes->abstol      = abstol;
1310d7a720efSLois Curfman McInnes   if (rtol != PETSC_DEFAULT)  snes->rtol      = rtol;
1311d7a720efSLois Curfman McInnes   if (stol != PETSC_DEFAULT)  snes->xtol      = stol;
1312d7a720efSLois Curfman McInnes   if (maxit != PETSC_DEFAULT) snes->max_its   = maxit;
1313d7a720efSLois Curfman McInnes   if (maxf != PETSC_DEFAULT)  snes->max_funcs = maxf;
13143a40ed3dSBarry Smith   PetscFunctionReturn(0);
13159b94acceSBarry Smith }
13169b94acceSBarry Smith 
13174a2ae208SSatish Balay #undef __FUNCT__
13184a2ae208SSatish Balay #define __FUNCT__ "SNESGetTolerances"
13199b94acceSBarry Smith /*@
132033174efeSLois Curfman McInnes    SNESGetTolerances - Gets various parameters used in convergence tests.
132133174efeSLois Curfman McInnes 
1322c7afd0dbSLois Curfman McInnes    Not Collective
1323c7afd0dbSLois Curfman McInnes 
132433174efeSLois Curfman McInnes    Input Parameters:
1325c7afd0dbSLois Curfman McInnes +  snes - the SNES context
132670441072SBarry Smith .  abstol - absolute convergence tolerance
132733174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
132833174efeSLois Curfman McInnes .  stol -  convergence tolerance in terms of the norm
132933174efeSLois Curfman McInnes            of the change in the solution between steps
133033174efeSLois Curfman McInnes .  maxit - maximum number of iterations
1331c7afd0dbSLois Curfman McInnes -  maxf - maximum number of function evaluations
1332fee21e36SBarry Smith 
133333174efeSLois Curfman McInnes    Notes:
133433174efeSLois Curfman McInnes    The user can specify PETSC_NULL for any parameter that is not needed.
133533174efeSLois Curfman McInnes 
133636851e7fSLois Curfman McInnes    Level: intermediate
133736851e7fSLois Curfman McInnes 
133833174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances
133933174efeSLois Curfman McInnes 
134033174efeSLois Curfman McInnes .seealso: SNESSetTolerances()
134133174efeSLois Curfman McInnes @*/
134263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetTolerances(SNES snes,PetscReal *abstol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf)
134333174efeSLois Curfman McInnes {
13443a40ed3dSBarry Smith   PetscFunctionBegin;
13454482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
134670441072SBarry Smith   if (abstol)  *abstol  = snes->abstol;
134733174efeSLois Curfman McInnes   if (rtol)  *rtol  = snes->rtol;
134833174efeSLois Curfman McInnes   if (stol)  *stol  = snes->xtol;
134933174efeSLois Curfman McInnes   if (maxit) *maxit = snes->max_its;
135033174efeSLois Curfman McInnes   if (maxf)  *maxf  = snes->max_funcs;
13513a40ed3dSBarry Smith   PetscFunctionReturn(0);
135233174efeSLois Curfman McInnes }
135333174efeSLois Curfman McInnes 
13544a2ae208SSatish Balay #undef __FUNCT__
13554a2ae208SSatish Balay #define __FUNCT__ "SNESSetTrustRegionTolerance"
135633174efeSLois Curfman McInnes /*@
13579b94acceSBarry Smith    SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance.
13589b94acceSBarry Smith 
1359fee21e36SBarry Smith    Collective on SNES
1360fee21e36SBarry Smith 
1361c7afd0dbSLois Curfman McInnes    Input Parameters:
1362c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1363c7afd0dbSLois Curfman McInnes -  tol - tolerance
1364c7afd0dbSLois Curfman McInnes 
13659b94acceSBarry Smith    Options Database Key:
1366c7afd0dbSLois Curfman McInnes .  -snes_trtol <tol> - Sets tol
13679b94acceSBarry Smith 
136836851e7fSLois Curfman McInnes    Level: intermediate
136936851e7fSLois Curfman McInnes 
13709b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance
13719b94acceSBarry Smith 
13722492ecdbSBarry Smith .seealso: SNESSetTolerances()
13739b94acceSBarry Smith @*/
137463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetTrustRegionTolerance(SNES snes,PetscReal tol)
13759b94acceSBarry Smith {
13763a40ed3dSBarry Smith   PetscFunctionBegin;
13774482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
13789b94acceSBarry Smith   snes->deltatol = tol;
13793a40ed3dSBarry Smith   PetscFunctionReturn(0);
13809b94acceSBarry Smith }
13819b94acceSBarry Smith 
1382df9fa365SBarry Smith /*
1383df9fa365SBarry Smith    Duplicate the lg monitors for SNES from KSP; for some reason with
1384df9fa365SBarry Smith    dynamic libraries things don't work under Sun4 if we just use
1385df9fa365SBarry Smith    macros instead of functions
1386df9fa365SBarry Smith */
13874a2ae208SSatish Balay #undef __FUNCT__
1388a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorLG"
1389a6570f20SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorLG(SNES snes,PetscInt it,PetscReal norm,void *ctx)
1390ce1608b8SBarry Smith {
1391dfbe8321SBarry Smith   PetscErrorCode ierr;
1392ce1608b8SBarry Smith 
1393ce1608b8SBarry Smith   PetscFunctionBegin;
13944482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1395a6570f20SBarry Smith   ierr = KSPMonitorLG((KSP)snes,it,norm,ctx);CHKERRQ(ierr);
1396ce1608b8SBarry Smith   PetscFunctionReturn(0);
1397ce1608b8SBarry Smith }
1398ce1608b8SBarry Smith 
13994a2ae208SSatish Balay #undef __FUNCT__
1400a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorLGCreate"
1401a6570f20SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorLGCreate(const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *draw)
1402df9fa365SBarry Smith {
1403dfbe8321SBarry Smith   PetscErrorCode ierr;
1404df9fa365SBarry Smith 
1405df9fa365SBarry Smith   PetscFunctionBegin;
1406a6570f20SBarry Smith   ierr = KSPMonitorLGCreate(host,label,x,y,m,n,draw);CHKERRQ(ierr);
1407df9fa365SBarry Smith   PetscFunctionReturn(0);
1408df9fa365SBarry Smith }
1409df9fa365SBarry Smith 
14104a2ae208SSatish Balay #undef __FUNCT__
1411a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorLGDestroy"
1412a6570f20SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorLGDestroy(PetscDrawLG draw)
1413df9fa365SBarry Smith {
1414dfbe8321SBarry Smith   PetscErrorCode ierr;
1415df9fa365SBarry Smith 
1416df9fa365SBarry Smith   PetscFunctionBegin;
1417a6570f20SBarry Smith   ierr = KSPMonitorLGDestroy(draw);CHKERRQ(ierr);
1418df9fa365SBarry Smith   PetscFunctionReturn(0);
1419df9fa365SBarry Smith }
1420df9fa365SBarry Smith 
14219b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */
14229b94acceSBarry Smith 
14234a2ae208SSatish Balay #undef __FUNCT__
1424a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorSet"
14259b94acceSBarry Smith /*@C
1426a6570f20SBarry Smith    SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every
14279b94acceSBarry Smith    iteration of the nonlinear solver to display the iteration's
14289b94acceSBarry Smith    progress.
14299b94acceSBarry Smith 
1430fee21e36SBarry Smith    Collective on SNES
1431fee21e36SBarry Smith 
1432c7afd0dbSLois Curfman McInnes    Input Parameters:
1433c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1434c7afd0dbSLois Curfman McInnes .  func - monitoring routine
1435b8a78c4aSBarry Smith .  mctx - [optional] user-defined context for private data for the
1436e8105e01SRichard Katz           monitor routine (use PETSC_NULL if no context is desired)
1437b3006f0bSLois Curfman McInnes -  monitordestroy - [optional] routine that frees monitor context
1438b3006f0bSLois Curfman McInnes           (may be PETSC_NULL)
14399b94acceSBarry Smith 
1440c7afd0dbSLois Curfman McInnes    Calling sequence of func:
1441a7cc72afSBarry Smith $     int func(SNES snes,PetscInt its, PetscReal norm,void *mctx)
1442c7afd0dbSLois Curfman McInnes 
1443c7afd0dbSLois Curfman McInnes +    snes - the SNES context
1444c7afd0dbSLois Curfman McInnes .    its - iteration number
1445c7afd0dbSLois Curfman McInnes .    norm - 2-norm function value (may be estimated)
144640a0c1c6SLois Curfman McInnes -    mctx - [optional] monitoring context
14479b94acceSBarry Smith 
14489665c990SLois Curfman McInnes    Options Database Keys:
1449a6570f20SBarry Smith +    -snes_monitor        - sets SNESMonitorDefault()
1450a6570f20SBarry Smith .    -snes_monitor_draw    - sets line graph monitor,
1451a6570f20SBarry Smith                             uses SNESMonitorLGCreate()
1452a6570f20SBarry Smith _    -snes_monitor_cancel - cancels all monitors that have
1453c7afd0dbSLois Curfman McInnes                             been hardwired into a code by
1454a6570f20SBarry Smith                             calls to SNESMonitorSet(), but
1455c7afd0dbSLois Curfman McInnes                             does not cancel those set via
1456c7afd0dbSLois Curfman McInnes                             the options database.
14579665c990SLois Curfman McInnes 
1458639f9d9dSBarry Smith    Notes:
14596bc08f3fSLois Curfman McInnes    Several different monitoring routines may be set by calling
1460a6570f20SBarry Smith    SNESMonitorSet() multiple times; all will be called in the
14616bc08f3fSLois Curfman McInnes    order in which they were set.
1462639f9d9dSBarry Smith 
146336851e7fSLois Curfman McInnes    Level: intermediate
146436851e7fSLois Curfman McInnes 
14659b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor
14669b94acceSBarry Smith 
1467a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorCancel()
14689b94acceSBarry Smith @*/
1469b90d0a6eSBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorSet(SNES snes,PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void*))
14709b94acceSBarry Smith {
1471b90d0a6eSBarry Smith   PetscInt i;
1472b90d0a6eSBarry Smith 
14733a40ed3dSBarry Smith   PetscFunctionBegin;
14744482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1475639f9d9dSBarry Smith   if (snes->numbermonitors >= MAXSNESMONITORS) {
147629bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set");
1477639f9d9dSBarry Smith   }
1478b90d0a6eSBarry Smith   for (i=0; i<snes->numbermonitors;i++) {
1479b90d0a6eSBarry Smith     if (monitor == snes->monitor[i] && monitordestroy == snes->monitordestroy[i] && mctx == snes->monitorcontext[i]) PetscFunctionReturn(0);
1480b90d0a6eSBarry Smith 
1481b90d0a6eSBarry Smith     /* check if both default monitors that share common ASCII viewer */
1482b90d0a6eSBarry Smith     if (monitor == snes->monitor[i] && monitor == SNESMonitorDefault) {
1483b90d0a6eSBarry Smith       if (mctx && snes->monitorcontext[i]) {
1484b90d0a6eSBarry Smith         PetscErrorCode          ierr;
1485b90d0a6eSBarry Smith         PetscViewerASCIIMonitor viewer1 = (PetscViewerASCIIMonitor) mctx;
1486b90d0a6eSBarry Smith         PetscViewerASCIIMonitor viewer2 = (PetscViewerASCIIMonitor) snes->monitorcontext[i];
1487b90d0a6eSBarry Smith         if (viewer1->viewer == viewer2->viewer) {
1488b90d0a6eSBarry Smith           ierr = (*monitordestroy)(mctx);CHKERRQ(ierr);
1489b90d0a6eSBarry Smith           PetscFunctionReturn(0);
1490b90d0a6eSBarry Smith         }
1491b90d0a6eSBarry Smith       }
1492b90d0a6eSBarry Smith     }
1493b90d0a6eSBarry Smith   }
1494b90d0a6eSBarry Smith   snes->monitor[snes->numbermonitors]           = monitor;
1495b8a78c4aSBarry Smith   snes->monitordestroy[snes->numbermonitors]    = monitordestroy;
1496639f9d9dSBarry Smith   snes->monitorcontext[snes->numbermonitors++]  = (void*)mctx;
14973a40ed3dSBarry Smith   PetscFunctionReturn(0);
14989b94acceSBarry Smith }
14999b94acceSBarry Smith 
15004a2ae208SSatish Balay #undef __FUNCT__
1501a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorCancel"
15025cd90555SBarry Smith /*@C
1503a6570f20SBarry Smith    SNESMonitorCancel - Clears all the monitor functions for a SNES object.
15045cd90555SBarry Smith 
1505c7afd0dbSLois Curfman McInnes    Collective on SNES
1506c7afd0dbSLois Curfman McInnes 
15075cd90555SBarry Smith    Input Parameters:
15085cd90555SBarry Smith .  snes - the SNES context
15095cd90555SBarry Smith 
15101a480d89SAdministrator    Options Database Key:
1511a6570f20SBarry Smith .  -snes_monitor_cancel - cancels all monitors that have been hardwired
1512a6570f20SBarry Smith     into a code by calls to SNESMonitorSet(), but does not cancel those
1513c7afd0dbSLois Curfman McInnes     set via the options database
15145cd90555SBarry Smith 
15155cd90555SBarry Smith    Notes:
15165cd90555SBarry Smith    There is no way to clear one specific monitor from a SNES object.
15175cd90555SBarry Smith 
151836851e7fSLois Curfman McInnes    Level: intermediate
151936851e7fSLois Curfman McInnes 
15205cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor
15215cd90555SBarry Smith 
1522a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorSet()
15235cd90555SBarry Smith @*/
1524a6570f20SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESMonitorCancel(SNES snes)
15255cd90555SBarry Smith {
1526d952e501SBarry Smith   PetscErrorCode ierr;
1527d952e501SBarry Smith   PetscInt       i;
1528d952e501SBarry Smith 
15295cd90555SBarry Smith   PetscFunctionBegin;
15304482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1531d952e501SBarry Smith   for (i=0; i<snes->numbermonitors; i++) {
1532d952e501SBarry Smith     if (snes->monitordestroy[i]) {
1533d952e501SBarry Smith       ierr = (*snes->monitordestroy[i])(snes->monitorcontext[i]);CHKERRQ(ierr);
1534d952e501SBarry Smith     }
1535d952e501SBarry Smith   }
15365cd90555SBarry Smith   snes->numbermonitors = 0;
15375cd90555SBarry Smith   PetscFunctionReturn(0);
15385cd90555SBarry Smith }
15395cd90555SBarry Smith 
15404a2ae208SSatish Balay #undef __FUNCT__
15414a2ae208SSatish Balay #define __FUNCT__ "SNESSetConvergenceTest"
15429b94acceSBarry Smith /*@C
15439b94acceSBarry Smith    SNESSetConvergenceTest - Sets the function that is to be used
15449b94acceSBarry Smith    to test for convergence of the nonlinear iterative solution.
15459b94acceSBarry Smith 
1546fee21e36SBarry Smith    Collective on SNES
1547fee21e36SBarry Smith 
1548c7afd0dbSLois Curfman McInnes    Input Parameters:
1549c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1550c7afd0dbSLois Curfman McInnes .  func - routine to test for convergence
1551c7afd0dbSLois Curfman McInnes -  cctx - [optional] context for private data for the convergence routine
1552c7afd0dbSLois Curfman McInnes           (may be PETSC_NULL)
15539b94acceSBarry Smith 
1554c7afd0dbSLois Curfman McInnes    Calling sequence of func:
155506ee9f85SBarry Smith $     PetscErrorCode func (SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx)
1556c7afd0dbSLois Curfman McInnes 
1557c7afd0dbSLois Curfman McInnes +    snes - the SNES context
155806ee9f85SBarry Smith .    it - current iteration (0 is the first and is before any Newton step)
1559c7afd0dbSLois Curfman McInnes .    cctx - [optional] convergence context
1560184914b5SBarry Smith .    reason - reason for convergence/divergence
1561c7afd0dbSLois Curfman McInnes .    xnorm - 2-norm of current iterate
15624b27c08aSLois Curfman McInnes .    gnorm - 2-norm of current step
15634b27c08aSLois Curfman McInnes -    f - 2-norm of function
15649b94acceSBarry Smith 
156536851e7fSLois Curfman McInnes    Level: advanced
156636851e7fSLois Curfman McInnes 
15679b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test
15689b94acceSBarry Smith 
15693f149594SLisandro Dalcin .seealso: SNESDefaultConverged(), SNESSkipConverged(), SNESConverged_LS(), SNESConverged_TR()
15709b94acceSBarry Smith @*/
157106ee9f85SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetConvergenceTest(SNES snes,PetscErrorCode (*func)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx)
15729b94acceSBarry Smith {
15733a40ed3dSBarry Smith   PetscFunctionBegin;
15744482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1575e7788613SBarry Smith   (snes)->ops->converged = func;
15769b94acceSBarry Smith   (snes)->cnvP           = cctx;
15773a40ed3dSBarry Smith   PetscFunctionReturn(0);
15789b94acceSBarry Smith }
15799b94acceSBarry Smith 
15804a2ae208SSatish Balay #undef __FUNCT__
15814a2ae208SSatish Balay #define __FUNCT__ "SNESGetConvergedReason"
158252baeb72SSatish Balay /*@
1583184914b5SBarry Smith    SNESGetConvergedReason - Gets the reason the SNES iteration was stopped.
1584184914b5SBarry Smith 
1585184914b5SBarry Smith    Not Collective
1586184914b5SBarry Smith 
1587184914b5SBarry Smith    Input Parameter:
1588184914b5SBarry Smith .  snes - the SNES context
1589184914b5SBarry Smith 
1590184914b5SBarry Smith    Output Parameter:
1591e090d566SSatish Balay .  reason - negative value indicates diverged, positive value converged, see petscsnes.h or the
1592184914b5SBarry Smith             manual pages for the individual convergence tests for complete lists
1593184914b5SBarry Smith 
1594184914b5SBarry Smith    Level: intermediate
1595184914b5SBarry Smith 
1596184914b5SBarry Smith    Notes: Can only be called after the call the SNESSolve() is complete.
1597184914b5SBarry Smith 
1598184914b5SBarry Smith .keywords: SNES, nonlinear, set, convergence, test
1599184914b5SBarry Smith 
16004b27c08aSLois Curfman McInnes .seealso: SNESSetConvergenceTest(), SNESConverged_LS(), SNESConverged_TR(), SNESConvergedReason
1601184914b5SBarry Smith @*/
160263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason)
1603184914b5SBarry Smith {
1604184914b5SBarry Smith   PetscFunctionBegin;
16054482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
16064482741eSBarry Smith   PetscValidPointer(reason,2);
1607184914b5SBarry Smith   *reason = snes->reason;
1608184914b5SBarry Smith   PetscFunctionReturn(0);
1609184914b5SBarry Smith }
1610184914b5SBarry Smith 
16114a2ae208SSatish Balay #undef __FUNCT__
16124a2ae208SSatish Balay #define __FUNCT__ "SNESSetConvergenceHistory"
1613c9005455SLois Curfman McInnes /*@
1614c9005455SLois Curfman McInnes    SNESSetConvergenceHistory - Sets the array used to hold the convergence history.
1615c9005455SLois Curfman McInnes 
1616fee21e36SBarry Smith    Collective on SNES
1617fee21e36SBarry Smith 
1618c7afd0dbSLois Curfman McInnes    Input Parameters:
1619c7afd0dbSLois Curfman McInnes +  snes - iterative context obtained from SNESCreate()
1620c7afd0dbSLois Curfman McInnes .  a   - array to hold history
1621cd5578b5SBarry Smith .  its - integer array holds the number of linear iterations for each solve.
1622758f92a0SBarry Smith .  na  - size of a and its
162364731454SLois Curfman McInnes -  reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero,
1624758f92a0SBarry Smith            else it continues storing new values for new nonlinear solves after the old ones
1625c7afd0dbSLois Curfman McInnes 
1626c9005455SLois Curfman McInnes    Notes:
16274b27c08aSLois Curfman McInnes    If set, this array will contain the function norms computed
1628c9005455SLois Curfman McInnes    at each step.
1629c9005455SLois Curfman McInnes 
1630c9005455SLois Curfman McInnes    This routine is useful, e.g., when running a code for purposes
1631c9005455SLois Curfman McInnes    of accurate performance monitoring, when no I/O should be done
1632c9005455SLois Curfman McInnes    during the section of code that is being timed.
1633c9005455SLois Curfman McInnes 
163436851e7fSLois Curfman McInnes    Level: intermediate
163536851e7fSLois Curfman McInnes 
1636c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history
1637758f92a0SBarry Smith 
163808405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory()
1639758f92a0SBarry Smith 
1640c9005455SLois Curfman McInnes @*/
1641a562a398SLisandro Dalcin PetscErrorCode PETSCSNES_DLLEXPORT SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscTruth reset)
1642c9005455SLois Curfman McInnes {
16433a40ed3dSBarry Smith   PetscFunctionBegin;
16444482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
16454482741eSBarry Smith   if (na)  PetscValidScalarPointer(a,2);
1646a562a398SLisandro Dalcin   if (its) PetscValidIntPointer(its,3);
1647c9005455SLois Curfman McInnes   snes->conv_hist       = a;
1648758f92a0SBarry Smith   snes->conv_hist_its   = its;
1649758f92a0SBarry Smith   snes->conv_hist_max   = na;
1650758f92a0SBarry Smith   snes->conv_hist_reset = reset;
1651758f92a0SBarry Smith   PetscFunctionReturn(0);
1652758f92a0SBarry Smith }
1653758f92a0SBarry Smith 
16544a2ae208SSatish Balay #undef __FUNCT__
16554a2ae208SSatish Balay #define __FUNCT__ "SNESGetConvergenceHistory"
16560c4c9dddSBarry Smith /*@C
1657758f92a0SBarry Smith    SNESGetConvergenceHistory - Gets the array used to hold the convergence history.
1658758f92a0SBarry Smith 
1659758f92a0SBarry Smith    Collective on SNES
1660758f92a0SBarry Smith 
1661758f92a0SBarry Smith    Input Parameter:
1662758f92a0SBarry Smith .  snes - iterative context obtained from SNESCreate()
1663758f92a0SBarry Smith 
1664758f92a0SBarry Smith    Output Parameters:
1665758f92a0SBarry Smith .  a   - array to hold history
1666758f92a0SBarry Smith .  its - integer array holds the number of linear iterations (or
1667758f92a0SBarry Smith          negative if not converged) for each solve.
1668758f92a0SBarry Smith -  na  - size of a and its
1669758f92a0SBarry Smith 
1670758f92a0SBarry Smith    Notes:
1671758f92a0SBarry Smith     The calling sequence for this routine in Fortran is
1672758f92a0SBarry Smith $   call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr)
1673758f92a0SBarry Smith 
1674758f92a0SBarry Smith    This routine is useful, e.g., when running a code for purposes
1675758f92a0SBarry Smith    of accurate performance monitoring, when no I/O should be done
1676758f92a0SBarry Smith    during the section of code that is being timed.
1677758f92a0SBarry Smith 
1678758f92a0SBarry Smith    Level: intermediate
1679758f92a0SBarry Smith 
1680758f92a0SBarry Smith .keywords: SNES, get, convergence, history
1681758f92a0SBarry Smith 
1682758f92a0SBarry Smith .seealso: SNESSetConvergencHistory()
1683758f92a0SBarry Smith 
1684758f92a0SBarry Smith @*/
168563dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na)
1686758f92a0SBarry Smith {
1687758f92a0SBarry Smith   PetscFunctionBegin;
16884482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1689758f92a0SBarry Smith   if (a)   *a   = snes->conv_hist;
1690758f92a0SBarry Smith   if (its) *its = snes->conv_hist_its;
1691758f92a0SBarry Smith   if (na)  *na  = snes->conv_hist_len;
16923a40ed3dSBarry Smith   PetscFunctionReturn(0);
1693c9005455SLois Curfman McInnes }
1694c9005455SLois Curfman McInnes 
1695e74ef692SMatthew Knepley #undef __FUNCT__
1696e74ef692SMatthew Knepley #define __FUNCT__ "SNESSetUpdate"
1697ac226902SBarry Smith /*@C
169876b2cf59SMatthew Knepley   SNESSetUpdate - Sets the general-purpose update function called
16997e4bb74cSBarry Smith   at the beginning o every iteration of the nonlinear solve. Specifically
17007e4bb74cSBarry Smith   it is called just before the Jacobian is "evaluated".
170176b2cf59SMatthew Knepley 
170276b2cf59SMatthew Knepley   Collective on SNES
170376b2cf59SMatthew Knepley 
170476b2cf59SMatthew Knepley   Input Parameters:
170576b2cf59SMatthew Knepley . snes - The nonlinear solver context
170676b2cf59SMatthew Knepley . func - The function
170776b2cf59SMatthew Knepley 
170876b2cf59SMatthew Knepley   Calling sequence of func:
1709b5d30489SBarry Smith . func (SNES snes, PetscInt step);
171076b2cf59SMatthew Knepley 
171176b2cf59SMatthew Knepley . step - The current step of the iteration
171276b2cf59SMatthew Knepley 
171376b2cf59SMatthew Knepley   Level: intermediate
171476b2cf59SMatthew Knepley 
171576b2cf59SMatthew Knepley .keywords: SNES, update
1716b5d30489SBarry Smith 
171776b2cf59SMatthew Knepley .seealso SNESDefaultUpdate(), SNESSetRhsBC(), SNESSetSolutionBC()
171876b2cf59SMatthew Knepley @*/
171963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt))
172076b2cf59SMatthew Knepley {
172176b2cf59SMatthew Knepley   PetscFunctionBegin;
17224482741eSBarry Smith   PetscValidHeaderSpecific(snes, SNES_COOKIE,1);
1723e7788613SBarry Smith   snes->ops->update = func;
172476b2cf59SMatthew Knepley   PetscFunctionReturn(0);
172576b2cf59SMatthew Knepley }
172676b2cf59SMatthew Knepley 
1727e74ef692SMatthew Knepley #undef __FUNCT__
1728e74ef692SMatthew Knepley #define __FUNCT__ "SNESDefaultUpdate"
172976b2cf59SMatthew Knepley /*@
173076b2cf59SMatthew Knepley   SNESDefaultUpdate - The default update function which does nothing.
173176b2cf59SMatthew Knepley 
173276b2cf59SMatthew Knepley   Not collective
173376b2cf59SMatthew Knepley 
173476b2cf59SMatthew Knepley   Input Parameters:
173576b2cf59SMatthew Knepley . snes - The nonlinear solver context
173676b2cf59SMatthew Knepley . step - The current step of the iteration
173776b2cf59SMatthew Knepley 
1738205452f4SMatthew Knepley   Level: intermediate
1739205452f4SMatthew Knepley 
174076b2cf59SMatthew Knepley .keywords: SNES, update
1741a6570f20SBarry Smith .seealso SNESSetUpdate(), SNESDefaultRhsBC(), SNESDefaultShortolutionBC()
174276b2cf59SMatthew Knepley @*/
174363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESDefaultUpdate(SNES snes, PetscInt step)
174476b2cf59SMatthew Knepley {
174576b2cf59SMatthew Knepley   PetscFunctionBegin;
174676b2cf59SMatthew Knepley   PetscFunctionReturn(0);
174776b2cf59SMatthew Knepley }
174876b2cf59SMatthew Knepley 
17494a2ae208SSatish Balay #undef __FUNCT__
17504a2ae208SSatish Balay #define __FUNCT__ "SNESScaleStep_Private"
17519b94acceSBarry Smith /*
17529b94acceSBarry Smith    SNESScaleStep_Private - Scales a step so that its length is less than the
17539b94acceSBarry Smith    positive parameter delta.
17549b94acceSBarry Smith 
17559b94acceSBarry Smith     Input Parameters:
1756c7afd0dbSLois Curfman McInnes +   snes - the SNES context
17579b94acceSBarry Smith .   y - approximate solution of linear system
17589b94acceSBarry Smith .   fnorm - 2-norm of current function
1759c7afd0dbSLois Curfman McInnes -   delta - trust region size
17609b94acceSBarry Smith 
17619b94acceSBarry Smith     Output Parameters:
1762c7afd0dbSLois Curfman McInnes +   gpnorm - predicted function norm at the new point, assuming local
17639b94acceSBarry Smith     linearization.  The value is zero if the step lies within the trust
17649b94acceSBarry Smith     region, and exceeds zero otherwise.
1765c7afd0dbSLois Curfman McInnes -   ynorm - 2-norm of the step
17669b94acceSBarry Smith 
17679b94acceSBarry Smith     Note:
17684b27c08aSLois Curfman McInnes     For non-trust region methods such as SNESLS, the parameter delta
17699b94acceSBarry Smith     is set to be the maximum allowable step size.
17709b94acceSBarry Smith 
17719b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step
17729b94acceSBarry Smith */
1773dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm)
17749b94acceSBarry Smith {
1775064f8208SBarry Smith   PetscReal      nrm;
1776ea709b57SSatish Balay   PetscScalar    cnorm;
1777dfbe8321SBarry Smith   PetscErrorCode ierr;
17783a40ed3dSBarry Smith 
17793a40ed3dSBarry Smith   PetscFunctionBegin;
17804482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
17814482741eSBarry Smith   PetscValidHeaderSpecific(y,VEC_COOKIE,2);
1782c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,2);
1783184914b5SBarry Smith 
1784064f8208SBarry Smith   ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr);
1785064f8208SBarry Smith   if (nrm > *delta) {
1786064f8208SBarry Smith      nrm = *delta/nrm;
1787064f8208SBarry Smith      *gpnorm = (1.0 - nrm)*(*fnorm);
1788064f8208SBarry Smith      cnorm = nrm;
17892dcb1b2aSMatthew Knepley      ierr = VecScale(y,cnorm);CHKERRQ(ierr);
17909b94acceSBarry Smith      *ynorm = *delta;
17919b94acceSBarry Smith   } else {
17929b94acceSBarry Smith      *gpnorm = 0.0;
1793064f8208SBarry Smith      *ynorm = nrm;
17949b94acceSBarry Smith   }
17953a40ed3dSBarry Smith   PetscFunctionReturn(0);
17969b94acceSBarry Smith }
17979b94acceSBarry Smith 
17984a2ae208SSatish Balay #undef __FUNCT__
17994a2ae208SSatish Balay #define __FUNCT__ "SNESSolve"
18006ce558aeSBarry Smith /*@C
1801f69a0ea3SMatthew Knepley    SNESSolve - Solves a nonlinear system F(x) = b.
1802f69a0ea3SMatthew Knepley    Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX().
18039b94acceSBarry Smith 
1804c7afd0dbSLois Curfman McInnes    Collective on SNES
1805c7afd0dbSLois Curfman McInnes 
1806b2002411SLois Curfman McInnes    Input Parameters:
1807c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1808f69a0ea3SMatthew Knepley .  b - the constant part of the equation, or PETSC_NULL to use zero.
180970e92668SMatthew Knepley -  x - the solution vector, or PETSC_NULL if it was set with SNESSetSolution()
18109b94acceSBarry Smith 
1811b2002411SLois Curfman McInnes    Notes:
18128ddd3da0SLois Curfman McInnes    The user should initialize the vector,x, with the initial guess
18138ddd3da0SLois Curfman McInnes    for the nonlinear solve prior to calling SNESSolve.  In particular,
18148ddd3da0SLois Curfman McInnes    to employ an initial guess of zero, the user should explicitly set
18158ddd3da0SLois Curfman McInnes    this vector to zero by calling VecSet().
18168ddd3da0SLois Curfman McInnes 
181736851e7fSLois Curfman McInnes    Level: beginner
181836851e7fSLois Curfman McInnes 
18199b94acceSBarry Smith .keywords: SNES, nonlinear, solve
18209b94acceSBarry Smith 
182170e92668SMatthew Knepley .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetRhs(), SNESSetSolution()
18229b94acceSBarry Smith @*/
1823f69a0ea3SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESSolve(SNES snes,Vec b,Vec x)
18249b94acceSBarry Smith {
1825dfbe8321SBarry Smith   PetscErrorCode ierr;
1826f1af5d2fSBarry Smith   PetscTruth     flg;
1827eabae89aSBarry Smith   char           filename[PETSC_MAX_PATH_LEN];
1828eabae89aSBarry Smith   PetscViewer    viewer;
1829052efed2SBarry Smith 
18303a40ed3dSBarry Smith   PetscFunctionBegin;
18314482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
1832e7788613SBarry Smith   if (!snes->ops->solve) SETERRQ(PETSC_ERR_ORDER,"SNESSetType() or SNESSetFromOptions() must be called before SNESSolve()");
183374637425SBarry Smith 
1834f69a0ea3SMatthew Knepley   if (b) {
1835f69a0ea3SMatthew Knepley     ierr = SNESSetRhs(snes, b); CHKERRQ(ierr);
18361096aae1SMatthew Knepley     if (!snes->vec_func) {
18371096aae1SMatthew Knepley       Vec r;
18381096aae1SMatthew Knepley 
18391096aae1SMatthew Knepley       ierr = VecDuplicate(b, &r); CHKERRQ(ierr);
18401096aae1SMatthew Knepley       ierr = SNESSetFunction(snes, r, PETSC_NULL, PETSC_NULL); CHKERRQ(ierr);
18411096aae1SMatthew Knepley     }
1842f69a0ea3SMatthew Knepley   }
184370e92668SMatthew Knepley   if (x) {
1844f69a0ea3SMatthew Knepley     PetscValidHeaderSpecific(x,VEC_COOKIE,3);
1845f69a0ea3SMatthew Knepley     PetscCheckSameComm(snes,1,x,3);
184670e92668SMatthew Knepley   } else {
184770e92668SMatthew Knepley     ierr = SNESGetSolution(snes, &x); CHKERRQ(ierr);
184870e92668SMatthew Knepley     if (!x) {
184970e92668SMatthew Knepley       ierr = VecDuplicate(snes->vec_func_always, &x); CHKERRQ(ierr);
185070e92668SMatthew Knepley     }
185170e92668SMatthew Knepley   }
185270e92668SMatthew Knepley   snes->vec_sol = snes->vec_sol_always = x;
18533f149594SLisandro Dalcin 
185470e92668SMatthew Knepley   ierr = SNESSetUp(snes);CHKERRQ(ierr);
18553f149594SLisandro Dalcin 
1856abc0a331SBarry Smith   if (snes->conv_hist_reset) snes->conv_hist_len = 0;
185750ffb88aSMatthew Knepley   snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;
1858d5e45103SBarry Smith 
18593f149594SLisandro Dalcin   ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
18603f149594SLisandro Dalcin 
1861e7788613SBarry Smith   ierr = PetscExceptionTry1((*(snes)->ops->solve)(snes),PETSC_ERR_ARG_DOMAIN);
1862d5e45103SBarry Smith   if (PetscExceptionValue(ierr)) {
186338f152feSBarry Smith     /* this means that a caller above me has also tryed this exception so I don't handle it here, pass it up */
186419717074SBarry Smith     PetscErrorCode pierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(pierr);
1865c671dbe7SSatish Balay   } else if (PetscExceptionCaught(ierr,PETSC_ERR_ARG_DOMAIN)) {
1866d5e45103SBarry Smith     /* translate exception into SNES not converged reason */
1867d5e45103SBarry Smith     snes->reason = SNES_DIVERGED_FUNCTION_DOMAIN;
186838f152feSBarry Smith     ierr = 0;
186938f152feSBarry Smith   }
187038f152feSBarry Smith   CHKERRQ(ierr);
18713f149594SLisandro Dalcin   if (!snes->reason) {
18723f149594SLisandro Dalcin     SETERRQ(PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason");
18733f149594SLisandro Dalcin   }
18743f149594SLisandro Dalcin   if (!snes->ops->converged && snes->reason == SNES_DIVERGED_MAX_IT) {
18753f149594SLisandro Dalcin     snes->reason = SNES_CONVERGED_ITS;
18763f149594SLisandro Dalcin   }
1877d5ba7fb7SMatthew Knepley   ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
18783f149594SLisandro Dalcin 
1879eabae89aSBarry Smith   ierr = PetscOptionsGetString(snes->prefix,"-snes_view",filename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
1880eabae89aSBarry Smith   if (flg && !PetscPreLoadingOn) {
1881eabae89aSBarry Smith     ierr = PetscViewerASCIIOpen(snes->comm,filename,&viewer);CHKERRQ(ierr);
1882eabae89aSBarry Smith     ierr = SNESView(snes,viewer);CHKERRQ(ierr);
1883eabae89aSBarry Smith     ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr);
1884eabae89aSBarry Smith   }
1885eabae89aSBarry Smith 
1886da9b6338SBarry Smith   ierr = PetscOptionsHasName(snes->prefix,"-snes_test_local_min",&flg);CHKERRQ(ierr);
1887da9b6338SBarry Smith   if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); }
18885968eb51SBarry Smith   if (snes->printreason) {
18895968eb51SBarry Smith     if (snes->reason > 0) {
18909dcbbd2bSBarry Smith       ierr = PetscPrintf(snes->comm,"Nonlinear solve converged due to %s\n",SNESConvergedReasons[snes->reason]);CHKERRQ(ierr);
18915968eb51SBarry Smith     } else {
18929dcbbd2bSBarry Smith       ierr = PetscPrintf(snes->comm,"Nonlinear solve did not converge due to %s\n",SNESConvergedReasons[snes->reason]);CHKERRQ(ierr);
18935968eb51SBarry Smith     }
18945968eb51SBarry Smith   }
18955968eb51SBarry Smith 
18963a40ed3dSBarry Smith   PetscFunctionReturn(0);
18979b94acceSBarry Smith }
18989b94acceSBarry Smith 
18999b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */
19009b94acceSBarry Smith 
19014a2ae208SSatish Balay #undef __FUNCT__
19024a2ae208SSatish Balay #define __FUNCT__ "SNESSetType"
190382bf6240SBarry Smith /*@C
19044b0e389bSBarry Smith    SNESSetType - Sets the method for the nonlinear solver.
19059b94acceSBarry Smith 
1906fee21e36SBarry Smith    Collective on SNES
1907fee21e36SBarry Smith 
1908c7afd0dbSLois Curfman McInnes    Input Parameters:
1909c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1910454a90a3SBarry Smith -  type - a known method
1911c7afd0dbSLois Curfman McInnes 
1912c7afd0dbSLois Curfman McInnes    Options Database Key:
1913454a90a3SBarry Smith .  -snes_type <type> - Sets the method; use -help for a list
1914c7afd0dbSLois Curfman McInnes    of available methods (for instance, ls or tr)
1915ae12b187SLois Curfman McInnes 
19169b94acceSBarry Smith    Notes:
1917e090d566SSatish Balay    See "petsc/include/petscsnes.h" for available methods (for instance)
19184b27c08aSLois Curfman McInnes +    SNESLS - Newton's method with line search
1919c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
19204b27c08aSLois Curfman McInnes .    SNESTR - Newton's method with trust region
1921c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
19229b94acceSBarry Smith 
1923ae12b187SLois Curfman McInnes   Normally, it is best to use the SNESSetFromOptions() command and then
1924ae12b187SLois Curfman McInnes   set the SNES solver type from the options database rather than by using
1925ae12b187SLois Curfman McInnes   this routine.  Using the options database provides the user with
1926ae12b187SLois Curfman McInnes   maximum flexibility in evaluating the many nonlinear solvers.
1927ae12b187SLois Curfman McInnes   The SNESSetType() routine is provided for those situations where it
1928ae12b187SLois Curfman McInnes   is necessary to set the nonlinear solver independently of the command
1929ae12b187SLois Curfman McInnes   line or options database.  This might be the case, for example, when
1930ae12b187SLois Curfman McInnes   the choice of solver changes during the execution of the program,
1931ae12b187SLois Curfman McInnes   and the user's application is taking responsibility for choosing the
1932b0a32e0cSBarry Smith   appropriate method.
193336851e7fSLois Curfman McInnes 
193436851e7fSLois Curfman McInnes   Level: intermediate
1935a703fe33SLois Curfman McInnes 
1936454a90a3SBarry Smith .keywords: SNES, set, type
1937435da068SBarry Smith 
1938435da068SBarry Smith .seealso: SNESType, SNESCreate()
1939435da068SBarry Smith 
19409b94acceSBarry Smith @*/
1941e060cb09SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetType(SNES snes,SNESType type)
19429b94acceSBarry Smith {
1943dfbe8321SBarry Smith   PetscErrorCode ierr,(*r)(SNES);
19446831982aSBarry Smith   PetscTruth     match;
19453a40ed3dSBarry Smith 
19463a40ed3dSBarry Smith   PetscFunctionBegin;
19474482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
19484482741eSBarry Smith   PetscValidCharPointer(type,2);
194982bf6240SBarry Smith 
19506831982aSBarry Smith   ierr = PetscTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr);
19510f5bd95cSBarry Smith   if (match) PetscFunctionReturn(0);
195292ff6ae8SBarry Smith 
19531d280d73SBarry Smith   ierr =  PetscFListFind(SNESList,snes->comm,type,(void (**)(void)) &r);CHKERRQ(ierr);
1954958c9bccSBarry Smith   if (!r) SETERRQ1(PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type);
195575396ef9SLisandro Dalcin   /* Destroy the previous private SNES context */
195675396ef9SLisandro Dalcin   if (snes->ops->destroy) { ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr); }
195775396ef9SLisandro Dalcin   /* Reinitialize function pointers in SNESOps structure */
195875396ef9SLisandro Dalcin   snes->ops->converged      = 0;
195975396ef9SLisandro Dalcin   snes->ops->setup          = 0;
196075396ef9SLisandro Dalcin   snes->ops->solve          = 0;
196175396ef9SLisandro Dalcin   snes->ops->view           = 0;
196275396ef9SLisandro Dalcin   snes->ops->setfromoptions = 0;
196375396ef9SLisandro Dalcin   snes->ops->destroy        = 0;
196475396ef9SLisandro Dalcin   /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */
196575396ef9SLisandro Dalcin   snes->setupcalled = PETSC_FALSE;
19663a40ed3dSBarry Smith   ierr = (*r)(snes);CHKERRQ(ierr);
1967454a90a3SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr);
19683a40ed3dSBarry Smith   PetscFunctionReturn(0);
19699b94acceSBarry Smith }
19709b94acceSBarry Smith 
1971a847f771SSatish Balay 
19729b94acceSBarry Smith /* --------------------------------------------------------------------- */
19734a2ae208SSatish Balay #undef __FUNCT__
19744a2ae208SSatish Balay #define __FUNCT__ "SNESRegisterDestroy"
197552baeb72SSatish Balay /*@
19769b94acceSBarry Smith    SNESRegisterDestroy - Frees the list of nonlinear solvers that were
1977f1af5d2fSBarry Smith    registered by SNESRegisterDynamic().
19789b94acceSBarry Smith 
1979fee21e36SBarry Smith    Not Collective
1980fee21e36SBarry Smith 
198136851e7fSLois Curfman McInnes    Level: advanced
198236851e7fSLois Curfman McInnes 
19839b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy
19849b94acceSBarry Smith 
19859b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll()
19869b94acceSBarry Smith @*/
198763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESRegisterDestroy(void)
19889b94acceSBarry Smith {
1989dfbe8321SBarry Smith   PetscErrorCode ierr;
199082bf6240SBarry Smith 
19913a40ed3dSBarry Smith   PetscFunctionBegin;
19921441b1d3SBarry Smith   ierr = PetscFListDestroy(&SNESList);CHKERRQ(ierr);
19934c49b128SBarry Smith   SNESRegisterAllCalled = PETSC_FALSE;
19943a40ed3dSBarry Smith   PetscFunctionReturn(0);
19959b94acceSBarry Smith }
19969b94acceSBarry Smith 
19974a2ae208SSatish Balay #undef __FUNCT__
19984a2ae208SSatish Balay #define __FUNCT__ "SNESGetType"
19999b94acceSBarry Smith /*@C
20009a28b0a6SLois Curfman McInnes    SNESGetType - Gets the SNES method type and name (as a string).
20019b94acceSBarry Smith 
2002c7afd0dbSLois Curfman McInnes    Not Collective
2003c7afd0dbSLois Curfman McInnes 
20049b94acceSBarry Smith    Input Parameter:
20054b0e389bSBarry Smith .  snes - nonlinear solver context
20069b94acceSBarry Smith 
20079b94acceSBarry Smith    Output Parameter:
20083a7fca6bSBarry Smith .  type - SNES method (a character string)
20099b94acceSBarry Smith 
201036851e7fSLois Curfman McInnes    Level: intermediate
201136851e7fSLois Curfman McInnes 
2012454a90a3SBarry Smith .keywords: SNES, nonlinear, get, type, name
20139b94acceSBarry Smith @*/
201463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetType(SNES snes,SNESType *type)
20159b94acceSBarry Smith {
20163a40ed3dSBarry Smith   PetscFunctionBegin;
20174482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
20184482741eSBarry Smith   PetscValidPointer(type,2);
2019454a90a3SBarry Smith   *type = snes->type_name;
20203a40ed3dSBarry Smith   PetscFunctionReturn(0);
20219b94acceSBarry Smith }
20229b94acceSBarry Smith 
20234a2ae208SSatish Balay #undef __FUNCT__
20244a2ae208SSatish Balay #define __FUNCT__ "SNESGetSolution"
202552baeb72SSatish Balay /*@
20269b94acceSBarry Smith    SNESGetSolution - Returns the vector where the approximate solution is
20279b94acceSBarry Smith    stored.
20289b94acceSBarry Smith 
2029c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
2030c7afd0dbSLois Curfman McInnes 
20319b94acceSBarry Smith    Input Parameter:
20329b94acceSBarry Smith .  snes - the SNES context
20339b94acceSBarry Smith 
20349b94acceSBarry Smith    Output Parameter:
20359b94acceSBarry Smith .  x - the solution
20369b94acceSBarry Smith 
203770e92668SMatthew Knepley    Level: intermediate
203836851e7fSLois Curfman McInnes 
20399b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution
20409b94acceSBarry Smith 
204170e92668SMatthew Knepley .seealso: SNESSetSolution(), SNESGetFunction(), SNESGetSolutionUpdate()
20429b94acceSBarry Smith @*/
204363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetSolution(SNES snes,Vec *x)
20449b94acceSBarry Smith {
20453a40ed3dSBarry Smith   PetscFunctionBegin;
20464482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
20474482741eSBarry Smith   PetscValidPointer(x,2);
20489b94acceSBarry Smith   *x = snes->vec_sol_always;
20493a40ed3dSBarry Smith   PetscFunctionReturn(0);
20509b94acceSBarry Smith }
20519b94acceSBarry Smith 
20524a2ae208SSatish Balay #undef __FUNCT__
205370e92668SMatthew Knepley #define __FUNCT__ "SNESSetSolution"
20547e4bb74cSBarry Smith /*@
205570e92668SMatthew Knepley    SNESSetSolution - Sets the vector where the approximate solution is stored.
205670e92668SMatthew Knepley 
205770e92668SMatthew Knepley    Not Collective, but Vec is parallel if SNES is parallel
205870e92668SMatthew Knepley 
205970e92668SMatthew Knepley    Input Parameters:
206070e92668SMatthew Knepley +  snes - the SNES context
206170e92668SMatthew Knepley -  x - the solution
206270e92668SMatthew Knepley 
206370e92668SMatthew Knepley    Output Parameter:
206470e92668SMatthew Knepley 
206570e92668SMatthew Knepley    Level: intermediate
206670e92668SMatthew Knepley 
206742219521SBarry Smith    Notes: this is not normally used, rather one simply calls SNESSolve() with
206842219521SBarry Smith           the appropriate solution vector.
206942219521SBarry Smith 
207070e92668SMatthew Knepley .keywords: SNES, nonlinear, set, solution
207170e92668SMatthew Knepley 
207270e92668SMatthew Knepley .seealso: SNESGetSolution(), SNESGetFunction(), SNESGetSolutionUpdate()
207370e92668SMatthew Knepley @*/
207470e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESSetSolution(SNES snes,Vec x)
207570e92668SMatthew Knepley {
207670e92668SMatthew Knepley   PetscFunctionBegin;
207770e92668SMatthew Knepley   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
207870e92668SMatthew Knepley   PetscValidHeaderSpecific(x,VEC_COOKIE,2);
207970e92668SMatthew Knepley   PetscCheckSameComm(snes,1,x,2);
208070e92668SMatthew Knepley   snes->vec_sol_always = x;
208170e92668SMatthew Knepley   PetscFunctionReturn(0);
208270e92668SMatthew Knepley }
208370e92668SMatthew Knepley 
208470e92668SMatthew Knepley #undef __FUNCT__
20854a2ae208SSatish Balay #define __FUNCT__ "SNESGetSolutionUpdate"
208652baeb72SSatish Balay /*@
20879b94acceSBarry Smith    SNESGetSolutionUpdate - Returns the vector where the solution update is
20889b94acceSBarry Smith    stored.
20899b94acceSBarry Smith 
2090c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
2091c7afd0dbSLois Curfman McInnes 
20929b94acceSBarry Smith    Input Parameter:
20939b94acceSBarry Smith .  snes - the SNES context
20949b94acceSBarry Smith 
20959b94acceSBarry Smith    Output Parameter:
20969b94acceSBarry Smith .  x - the solution update
20979b94acceSBarry Smith 
209836851e7fSLois Curfman McInnes    Level: advanced
209936851e7fSLois Curfman McInnes 
21009b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update
21019b94acceSBarry Smith 
21029b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction
21039b94acceSBarry Smith @*/
210463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetSolutionUpdate(SNES snes,Vec *x)
21059b94acceSBarry Smith {
21063a40ed3dSBarry Smith   PetscFunctionBegin;
21074482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
21084482741eSBarry Smith   PetscValidPointer(x,2);
21099b94acceSBarry Smith   *x = snes->vec_sol_update_always;
21103a40ed3dSBarry Smith   PetscFunctionReturn(0);
21119b94acceSBarry Smith }
21129b94acceSBarry Smith 
21134a2ae208SSatish Balay #undef __FUNCT__
21144a2ae208SSatish Balay #define __FUNCT__ "SNESGetFunction"
21159b94acceSBarry Smith /*@C
21163638b69dSLois Curfman McInnes    SNESGetFunction - Returns the vector where the function is stored.
21179b94acceSBarry Smith 
2118c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
2119c7afd0dbSLois Curfman McInnes 
21209b94acceSBarry Smith    Input Parameter:
21219b94acceSBarry Smith .  snes - the SNES context
21229b94acceSBarry Smith 
21239b94acceSBarry Smith    Output Parameter:
21247bf4e008SBarry Smith +  r - the function (or PETSC_NULL)
212570e92668SMatthew Knepley .  func - the function (or PETSC_NULL)
212670e92668SMatthew Knepley -  ctx - the function context (or PETSC_NULL)
21279b94acceSBarry Smith 
212836851e7fSLois Curfman McInnes    Level: advanced
212936851e7fSLois Curfman McInnes 
2130a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function
21319b94acceSBarry Smith 
21324b27c08aSLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution()
21339b94acceSBarry Smith @*/
213470e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**func)(SNES,Vec,Vec,void*),void **ctx)
21359b94acceSBarry Smith {
21363a40ed3dSBarry Smith   PetscFunctionBegin;
21374482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
21387bf4e008SBarry Smith   if (r)    *r    = snes->vec_func_always;
2139e7788613SBarry Smith   if (func) *func = snes->ops->computefunction;
214070e92668SMatthew Knepley   if (ctx)  *ctx  = snes->funP;
21413a40ed3dSBarry Smith   PetscFunctionReturn(0);
21429b94acceSBarry Smith }
21439b94acceSBarry Smith 
21444a2ae208SSatish Balay #undef __FUNCT__
21454a2ae208SSatish Balay #define __FUNCT__ "SNESSetOptionsPrefix"
21463c7409f5SSatish Balay /*@C
21473c7409f5SSatish Balay    SNESSetOptionsPrefix - Sets the prefix used for searching for all
2148d850072dSLois Curfman McInnes    SNES options in the database.
21493c7409f5SSatish Balay 
2150fee21e36SBarry Smith    Collective on SNES
2151fee21e36SBarry Smith 
2152c7afd0dbSLois Curfman McInnes    Input Parameter:
2153c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2154c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
2155c7afd0dbSLois Curfman McInnes 
2156d850072dSLois Curfman McInnes    Notes:
2157a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2158c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
2159d850072dSLois Curfman McInnes 
216036851e7fSLois Curfman McInnes    Level: advanced
216136851e7fSLois Curfman McInnes 
21623c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database
2163a86d99e1SLois Curfman McInnes 
2164a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions()
21653c7409f5SSatish Balay @*/
216663dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetOptionsPrefix(SNES snes,const char prefix[])
21673c7409f5SSatish Balay {
2168dfbe8321SBarry Smith   PetscErrorCode ierr;
21693c7409f5SSatish Balay 
21703a40ed3dSBarry Smith   PetscFunctionBegin;
21714482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
2172639f9d9dSBarry Smith   ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
217394b7f48cSBarry Smith   ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
21743a40ed3dSBarry Smith   PetscFunctionReturn(0);
21753c7409f5SSatish Balay }
21763c7409f5SSatish Balay 
21774a2ae208SSatish Balay #undef __FUNCT__
21784a2ae208SSatish Balay #define __FUNCT__ "SNESAppendOptionsPrefix"
21793c7409f5SSatish Balay /*@C
2180f525115eSLois Curfman McInnes    SNESAppendOptionsPrefix - Appends to the prefix used for searching for all
2181d850072dSLois Curfman McInnes    SNES options in the database.
21823c7409f5SSatish Balay 
2183fee21e36SBarry Smith    Collective on SNES
2184fee21e36SBarry Smith 
2185c7afd0dbSLois Curfman McInnes    Input Parameters:
2186c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2187c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
2188c7afd0dbSLois Curfman McInnes 
2189d850072dSLois Curfman McInnes    Notes:
2190a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2191c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
2192d850072dSLois Curfman McInnes 
219336851e7fSLois Curfman McInnes    Level: advanced
219436851e7fSLois Curfman McInnes 
21953c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database
2196a86d99e1SLois Curfman McInnes 
2197a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix()
21983c7409f5SSatish Balay @*/
219963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESAppendOptionsPrefix(SNES snes,const char prefix[])
22003c7409f5SSatish Balay {
2201dfbe8321SBarry Smith   PetscErrorCode ierr;
22023c7409f5SSatish Balay 
22033a40ed3dSBarry Smith   PetscFunctionBegin;
22044482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
2205639f9d9dSBarry Smith   ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
220694b7f48cSBarry Smith   ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
22073a40ed3dSBarry Smith   PetscFunctionReturn(0);
22083c7409f5SSatish Balay }
22093c7409f5SSatish Balay 
22104a2ae208SSatish Balay #undef __FUNCT__
22114a2ae208SSatish Balay #define __FUNCT__ "SNESGetOptionsPrefix"
22129ab63eb5SSatish Balay /*@C
22133c7409f5SSatish Balay    SNESGetOptionsPrefix - Sets the prefix used for searching for all
22143c7409f5SSatish Balay    SNES options in the database.
22153c7409f5SSatish Balay 
2216c7afd0dbSLois Curfman McInnes    Not Collective
2217c7afd0dbSLois Curfman McInnes 
22183c7409f5SSatish Balay    Input Parameter:
22193c7409f5SSatish Balay .  snes - the SNES context
22203c7409f5SSatish Balay 
22213c7409f5SSatish Balay    Output Parameter:
22223c7409f5SSatish Balay .  prefix - pointer to the prefix string used
22233c7409f5SSatish Balay 
22249ab63eb5SSatish Balay    Notes: On the fortran side, the user should pass in a string 'prifix' of
22259ab63eb5SSatish Balay    sufficient length to hold the prefix.
22269ab63eb5SSatish Balay 
222736851e7fSLois Curfman McInnes    Level: advanced
222836851e7fSLois Curfman McInnes 
22293c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database
2230a86d99e1SLois Curfman McInnes 
2231a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix()
22323c7409f5SSatish Balay @*/
2233e060cb09SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESGetOptionsPrefix(SNES snes,const char *prefix[])
22343c7409f5SSatish Balay {
2235dfbe8321SBarry Smith   PetscErrorCode ierr;
22363c7409f5SSatish Balay 
22373a40ed3dSBarry Smith   PetscFunctionBegin;
22384482741eSBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
2239639f9d9dSBarry Smith   ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
22403a40ed3dSBarry Smith   PetscFunctionReturn(0);
22413c7409f5SSatish Balay }
22423c7409f5SSatish Balay 
2243b2002411SLois Curfman McInnes 
22444a2ae208SSatish Balay #undef __FUNCT__
22454a2ae208SSatish Balay #define __FUNCT__ "SNESRegister"
22463cea93caSBarry Smith /*@C
22473cea93caSBarry Smith   SNESRegister - See SNESRegisterDynamic()
22483cea93caSBarry Smith 
22497f6c08e0SMatthew Knepley   Level: advanced
22503cea93caSBarry Smith @*/
225163dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESRegister(const char sname[],const char path[],const char name[],PetscErrorCode (*function)(SNES))
2252b2002411SLois Curfman McInnes {
2253e2d1d2b7SBarry Smith   char           fullname[PETSC_MAX_PATH_LEN];
2254dfbe8321SBarry Smith   PetscErrorCode ierr;
2255b2002411SLois Curfman McInnes 
2256b2002411SLois Curfman McInnes   PetscFunctionBegin;
2257b0a32e0cSBarry Smith   ierr = PetscFListConcat(path,name,fullname);CHKERRQ(ierr);
2258c134de8dSSatish Balay   ierr = PetscFListAdd(&SNESList,sname,fullname,(void (*)(void))function);CHKERRQ(ierr);
2259b2002411SLois Curfman McInnes   PetscFunctionReturn(0);
2260b2002411SLois Curfman McInnes }
2261da9b6338SBarry Smith 
2262da9b6338SBarry Smith #undef __FUNCT__
2263da9b6338SBarry Smith #define __FUNCT__ "SNESTestLocalMin"
226463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESTestLocalMin(SNES snes)
2265da9b6338SBarry Smith {
2266dfbe8321SBarry Smith   PetscErrorCode ierr;
226777431f27SBarry Smith   PetscInt       N,i,j;
2268da9b6338SBarry Smith   Vec            u,uh,fh;
2269da9b6338SBarry Smith   PetscScalar    value;
2270da9b6338SBarry Smith   PetscReal      norm;
2271da9b6338SBarry Smith 
2272da9b6338SBarry Smith   PetscFunctionBegin;
2273da9b6338SBarry Smith   ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr);
2274da9b6338SBarry Smith   ierr = VecDuplicate(u,&uh);CHKERRQ(ierr);
2275da9b6338SBarry Smith   ierr = VecDuplicate(u,&fh);CHKERRQ(ierr);
2276da9b6338SBarry Smith 
2277da9b6338SBarry Smith   /* currently only works for sequential */
2278da9b6338SBarry Smith   ierr = PetscPrintf(PETSC_COMM_WORLD,"Testing FormFunction() for local min\n");
2279da9b6338SBarry Smith   ierr = VecGetSize(u,&N);CHKERRQ(ierr);
2280da9b6338SBarry Smith   for (i=0; i<N; i++) {
2281da9b6338SBarry Smith     ierr = VecCopy(u,uh);CHKERRQ(ierr);
228277431f27SBarry Smith     ierr  = PetscPrintf(PETSC_COMM_WORLD,"i = %D\n",i);CHKERRQ(ierr);
2283da9b6338SBarry Smith     for (j=-10; j<11; j++) {
2284ccae9161SBarry Smith       value = PetscSign(j)*exp(PetscAbs(j)-10.0);
2285da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
22863ab0aad5SBarry Smith       ierr  = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr);
2287da9b6338SBarry Smith       ierr  = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr);
228877431f27SBarry Smith       ierr  = PetscPrintf(PETSC_COMM_WORLD,"       j norm %D %18.16e\n",j,norm);CHKERRQ(ierr);
2289da9b6338SBarry Smith       value = -value;
2290da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
2291da9b6338SBarry Smith     }
2292da9b6338SBarry Smith   }
2293da9b6338SBarry Smith   ierr = VecDestroy(uh);CHKERRQ(ierr);
2294da9b6338SBarry Smith   ierr = VecDestroy(fh);CHKERRQ(ierr);
2295da9b6338SBarry Smith   PetscFunctionReturn(0);
2296da9b6338SBarry Smith }
229771f87433Sdalcinl 
229871f87433Sdalcinl #undef __FUNCT__
2299fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPSetUseEW"
230071f87433Sdalcinl /*@
2301fa9f3622SBarry Smith    SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for
230271f87433Sdalcinl    computing relative tolerance for linear solvers within an inexact
230371f87433Sdalcinl    Newton method.
230471f87433Sdalcinl 
230571f87433Sdalcinl    Collective on SNES
230671f87433Sdalcinl 
230771f87433Sdalcinl    Input Parameters:
230871f87433Sdalcinl +  snes - SNES context
230971f87433Sdalcinl -  flag - PETSC_TRUE or PETSC_FALSE
231071f87433Sdalcinl 
231171f87433Sdalcinl    Notes:
231271f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
231371f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
231471f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
231571f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
231671f87433Sdalcinl    solver.
231771f87433Sdalcinl 
231871f87433Sdalcinl    Level: advanced
231971f87433Sdalcinl 
232071f87433Sdalcinl    Reference:
232171f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
232271f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
232371f87433Sdalcinl 
232471f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
232571f87433Sdalcinl 
2326fa9f3622SBarry Smith .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
232771f87433Sdalcinl @*/
2328fa9f3622SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESKSPSetUseEW(SNES snes,PetscTruth flag)
232971f87433Sdalcinl {
233071f87433Sdalcinl   PetscFunctionBegin;
233171f87433Sdalcinl   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
233271f87433Sdalcinl   snes->ksp_ewconv = flag;
233371f87433Sdalcinl   PetscFunctionReturn(0);
233471f87433Sdalcinl }
233571f87433Sdalcinl 
233671f87433Sdalcinl #undef __FUNCT__
2337fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPGetUseEW"
233871f87433Sdalcinl /*@
2339fa9f3622SBarry Smith    SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method
234071f87433Sdalcinl    for computing relative tolerance for linear solvers within an
234171f87433Sdalcinl    inexact Newton method.
234271f87433Sdalcinl 
234371f87433Sdalcinl    Not Collective
234471f87433Sdalcinl 
234571f87433Sdalcinl    Input Parameter:
234671f87433Sdalcinl .  snes - SNES context
234771f87433Sdalcinl 
234871f87433Sdalcinl    Output Parameter:
234971f87433Sdalcinl .  flag - PETSC_TRUE or PETSC_FALSE
235071f87433Sdalcinl 
235171f87433Sdalcinl    Notes:
235271f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
235371f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
235471f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
235571f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
235671f87433Sdalcinl    solver.
235771f87433Sdalcinl 
235871f87433Sdalcinl    Level: advanced
235971f87433Sdalcinl 
236071f87433Sdalcinl    Reference:
236171f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
236271f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
236371f87433Sdalcinl 
236471f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
236571f87433Sdalcinl 
2366fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
236771f87433Sdalcinl @*/
2368fa9f3622SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESKSPGetUseEW(SNES snes, PetscTruth *flag)
236971f87433Sdalcinl {
237071f87433Sdalcinl   PetscFunctionBegin;
237171f87433Sdalcinl   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
237271f87433Sdalcinl   PetscValidPointer(flag,2);
237371f87433Sdalcinl   *flag = snes->ksp_ewconv;
237471f87433Sdalcinl   PetscFunctionReturn(0);
237571f87433Sdalcinl }
237671f87433Sdalcinl 
237771f87433Sdalcinl #undef __FUNCT__
2378fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPSetParametersEW"
237971f87433Sdalcinl /*@
2380fa9f3622SBarry Smith    SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker
238171f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
238271f87433Sdalcinl    Newton method.
238371f87433Sdalcinl 
238471f87433Sdalcinl    Collective on SNES
238571f87433Sdalcinl 
238671f87433Sdalcinl    Input Parameters:
238771f87433Sdalcinl +    snes - SNES context
238871f87433Sdalcinl .    version - version 1, 2 (default is 2) or 3
238971f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
239071f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
239171f87433Sdalcinl .    gamma - multiplicative factor for version 2 rtol computation
239271f87433Sdalcinl              (0 <= gamma2 <= 1)
239371f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
239471f87433Sdalcinl .    alpha2 - power for safeguard
239571f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
239671f87433Sdalcinl 
239771f87433Sdalcinl    Note:
239871f87433Sdalcinl    Version 3 was contributed by Luis Chacon, June 2006.
239971f87433Sdalcinl 
240071f87433Sdalcinl    Use PETSC_DEFAULT to retain the default for any of the parameters.
240171f87433Sdalcinl 
240271f87433Sdalcinl    Level: advanced
240371f87433Sdalcinl 
240471f87433Sdalcinl    Reference:
240571f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
240671f87433Sdalcinl    inexact Newton method", Utah State University Math. Stat. Dept. Res.
240771f87433Sdalcinl    Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput.
240871f87433Sdalcinl 
240971f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, set, parameters
241071f87433Sdalcinl 
2411fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW()
241271f87433Sdalcinl @*/
2413fa9f3622SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,
241471f87433Sdalcinl 							    PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold)
241571f87433Sdalcinl {
2416fa9f3622SBarry Smith   SNESKSPEW *kctx;
241771f87433Sdalcinl   PetscFunctionBegin;
241871f87433Sdalcinl   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
2419fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
242071f87433Sdalcinl   if (!kctx) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
242171f87433Sdalcinl 
242271f87433Sdalcinl   if (version != PETSC_DEFAULT)   kctx->version   = version;
242371f87433Sdalcinl   if (rtol_0 != PETSC_DEFAULT)    kctx->rtol_0    = rtol_0;
242471f87433Sdalcinl   if (rtol_max != PETSC_DEFAULT)  kctx->rtol_max  = rtol_max;
242571f87433Sdalcinl   if (gamma != PETSC_DEFAULT)     kctx->gamma     = gamma;
242671f87433Sdalcinl   if (alpha != PETSC_DEFAULT)     kctx->alpha     = alpha;
242771f87433Sdalcinl   if (alpha2 != PETSC_DEFAULT)    kctx->alpha2    = alpha2;
242871f87433Sdalcinl   if (threshold != PETSC_DEFAULT) kctx->threshold = threshold;
242971f87433Sdalcinl 
243071f87433Sdalcinl   if (kctx->version < 1 || kctx->version > 3) {
243171f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 and 3 are supported: %D",kctx->version);
243271f87433Sdalcinl   }
243371f87433Sdalcinl   if (kctx->rtol_0 < 0.0 || kctx->rtol_0 >= 1.0) {
243471f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_0 < 1.0: %G",kctx->rtol_0);
243571f87433Sdalcinl   }
243671f87433Sdalcinl   if (kctx->rtol_max < 0.0 || kctx->rtol_max >= 1.0) {
243771f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_max (%G) < 1.0\n",kctx->rtol_max);
243871f87433Sdalcinl   }
243971f87433Sdalcinl   if (kctx->gamma < 0.0 || kctx->gamma > 1.0) {
244071f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= gamma (%G) <= 1.0\n",kctx->gamma);
244171f87433Sdalcinl   }
244271f87433Sdalcinl   if (kctx->alpha <= 1.0 || kctx->alpha > 2.0) {
244371f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"1.0 < alpha (%G) <= 2.0\n",kctx->alpha);
244471f87433Sdalcinl   }
244571f87433Sdalcinl   if (kctx->threshold <= 0.0 || kctx->threshold >= 1.0) {
244671f87433Sdalcinl     SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"0.0 < threshold (%G) < 1.0\n",kctx->threshold);
244771f87433Sdalcinl   }
244871f87433Sdalcinl   PetscFunctionReturn(0);
244971f87433Sdalcinl }
245071f87433Sdalcinl 
245171f87433Sdalcinl #undef __FUNCT__
2452fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPGetParametersEW"
245371f87433Sdalcinl /*@
2454fa9f3622SBarry Smith    SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker
245571f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
245671f87433Sdalcinl    Newton method.
245771f87433Sdalcinl 
245871f87433Sdalcinl    Not Collective
245971f87433Sdalcinl 
246071f87433Sdalcinl    Input Parameters:
246171f87433Sdalcinl      snes - SNES context
246271f87433Sdalcinl 
246371f87433Sdalcinl    Output Parameters:
246471f87433Sdalcinl +    version - version 1, 2 (default is 2) or 3
246571f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
246671f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
246771f87433Sdalcinl .    gamma - multiplicative factor for version 2 rtol computation
246871f87433Sdalcinl              (0 <= gamma2 <= 1)
246971f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
247071f87433Sdalcinl .    alpha2 - power for safeguard
247171f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
247271f87433Sdalcinl 
247371f87433Sdalcinl    Level: advanced
247471f87433Sdalcinl 
247571f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, get, parameters
247671f87433Sdalcinl 
2477fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW()
247871f87433Sdalcinl @*/
2479fa9f3622SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,
248071f87433Sdalcinl 							    PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold)
248171f87433Sdalcinl {
2482fa9f3622SBarry Smith   SNESKSPEW *kctx;
248371f87433Sdalcinl   PetscFunctionBegin;
248471f87433Sdalcinl   PetscValidHeaderSpecific(snes,SNES_COOKIE,1);
2485fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
248671f87433Sdalcinl   if (!kctx) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
248771f87433Sdalcinl   if(version)   *version   = kctx->version;
248871f87433Sdalcinl   if(rtol_0)    *rtol_0    = kctx->rtol_0;
248971f87433Sdalcinl   if(rtol_max)  *rtol_max  = kctx->rtol_max;
249071f87433Sdalcinl   if(gamma)     *gamma     = kctx->gamma;
249171f87433Sdalcinl   if(alpha)     *alpha     = kctx->alpha;
249271f87433Sdalcinl   if(alpha2)    *alpha2    = kctx->alpha2;
249371f87433Sdalcinl   if(threshold) *threshold = kctx->threshold;
249471f87433Sdalcinl   PetscFunctionReturn(0);
249571f87433Sdalcinl }
249671f87433Sdalcinl 
249771f87433Sdalcinl #undef __FUNCT__
2498fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPEW_PreSolve"
2499fa9f3622SBarry Smith static PetscErrorCode SNESKSPEW_PreSolve(SNES snes, KSP ksp, Vec b, Vec x)
250071f87433Sdalcinl {
250171f87433Sdalcinl   PetscErrorCode ierr;
2502fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
250371f87433Sdalcinl   PetscReal      rtol=PETSC_DEFAULT,stol;
250471f87433Sdalcinl 
250571f87433Sdalcinl   PetscFunctionBegin;
250671f87433Sdalcinl   if (!kctx) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context exists");
250771f87433Sdalcinl   if (!snes->iter) { /* first time in, so use the original user rtol */
250871f87433Sdalcinl     rtol = kctx->rtol_0;
250971f87433Sdalcinl   } else {
251071f87433Sdalcinl     if (kctx->version == 1) {
251171f87433Sdalcinl       rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last;
251271f87433Sdalcinl       if (rtol < 0.0) rtol = -rtol;
251371f87433Sdalcinl       stol = pow(kctx->rtol_last,kctx->alpha2);
251471f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
251571f87433Sdalcinl     } else if (kctx->version == 2) {
251671f87433Sdalcinl       rtol = kctx->gamma * pow(snes->norm/kctx->norm_last,kctx->alpha);
251771f87433Sdalcinl       stol = kctx->gamma * pow(kctx->rtol_last,kctx->alpha);
251871f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
251971f87433Sdalcinl     } else if (kctx->version == 3) {/* contributed by Luis Chacon, June 2006. */
252071f87433Sdalcinl       rtol = kctx->gamma * pow(snes->norm/kctx->norm_last,kctx->alpha);
252171f87433Sdalcinl       /* safeguard: avoid sharp decrease of rtol */
252271f87433Sdalcinl       stol = kctx->gamma*pow(kctx->rtol_last,kctx->alpha);
252371f87433Sdalcinl       stol = PetscMax(rtol,stol);
252471f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
252571f87433Sdalcinl       /* safeguard: avoid oversolving */
252671f87433Sdalcinl       stol = kctx->gamma*(snes->ttol)/snes->norm;
252771f87433Sdalcinl       stol = PetscMax(rtol,stol);
252871f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
252971f87433Sdalcinl     } else SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version);
253071f87433Sdalcinl   }
253171f87433Sdalcinl   /* safeguard: avoid rtol greater than one */
253271f87433Sdalcinl   rtol = PetscMin(rtol,kctx->rtol_max);
253371f87433Sdalcinl   ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr);
253471f87433Sdalcinl   ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%G\n",snes->iter,kctx->version,rtol);CHKERRQ(ierr);
253571f87433Sdalcinl   PetscFunctionReturn(0);
253671f87433Sdalcinl }
253771f87433Sdalcinl 
253871f87433Sdalcinl #undef __FUNCT__
2539fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPEW_PostSolve"
2540fa9f3622SBarry Smith static PetscErrorCode SNESKSPEW_PostSolve(SNES snes, KSP ksp, Vec b, Vec x)
254171f87433Sdalcinl {
254271f87433Sdalcinl   PetscErrorCode ierr;
2543fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
254471f87433Sdalcinl   PCSide         pcside;
254571f87433Sdalcinl   Vec            lres;
254671f87433Sdalcinl 
254771f87433Sdalcinl   PetscFunctionBegin;
254871f87433Sdalcinl   if (!kctx) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context exists");
254971f87433Sdalcinl   ierr = KSPGetTolerances(ksp,&kctx->rtol_last,0,0,0);CHKERRQ(ierr);
255071f87433Sdalcinl   ierr = SNESGetFunctionNorm(snes,&kctx->norm_last);CHKERRQ(ierr);
255171f87433Sdalcinl   if (kctx->version == 1) {
255271f87433Sdalcinl     ierr = KSPGetPreconditionerSide(ksp,&pcside);CHKERRQ(ierr);
255371f87433Sdalcinl     if (pcside == PC_RIGHT) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */
255471f87433Sdalcinl       /* KSP residual is true linear residual */
255571f87433Sdalcinl       ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr);
255671f87433Sdalcinl     } else {
255771f87433Sdalcinl       /* KSP residual is preconditioned residual */
255871f87433Sdalcinl       /* compute true linear residual norm */
255971f87433Sdalcinl       ierr = VecDuplicate(b,&lres);CHKERRQ(ierr);
256071f87433Sdalcinl       ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr);
256171f87433Sdalcinl       ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr);
256271f87433Sdalcinl       ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr);
256371f87433Sdalcinl       ierr = VecDestroy(lres);CHKERRQ(ierr);
256471f87433Sdalcinl     }
256571f87433Sdalcinl   }
256671f87433Sdalcinl   PetscFunctionReturn(0);
256771f87433Sdalcinl }
256871f87433Sdalcinl 
256971f87433Sdalcinl #undef __FUNCT__
257071f87433Sdalcinl #define __FUNCT__ "SNES_KSPSolve"
257171f87433Sdalcinl PetscErrorCode SNES_KSPSolve(SNES snes, KSP ksp, Vec b, Vec x)
257271f87433Sdalcinl {
257371f87433Sdalcinl   PetscErrorCode ierr;
257471f87433Sdalcinl 
257571f87433Sdalcinl   PetscFunctionBegin;
2576fa9f3622SBarry Smith   if (snes->ksp_ewconv) { ierr = SNESKSPEW_PreSolve(snes,ksp,b,x);CHKERRQ(ierr);  }
257771f87433Sdalcinl   ierr = KSPSolve(ksp,b,x);CHKERRQ(ierr);
2578fa9f3622SBarry Smith   if (snes->ksp_ewconv) { ierr = SNESKSPEW_PostSolve(snes,ksp,b,x);CHKERRQ(ierr); }
257971f87433Sdalcinl   PetscFunctionReturn(0);
258071f87433Sdalcinl }
2581