xref: /petsc/src/snes/interface/snes.c (revision dc424cfae46ffd055acbb99fbc61fc85fc92f9ad)
1 
2 #include <petsc/private/snesimpl.h>      /*I "petscsnes.h"  I*/
3 #include <petscdmshell.h>
4 #include <petscdraw.h>
5 #include <petscds.h>
6 #include <petscconvest.h>
7 
8 PetscBool         SNESRegisterAllCalled = PETSC_FALSE;
9 PetscFunctionList SNESList              = NULL;
10 
11 /* Logging support */
12 PetscClassId  SNES_CLASSID, DMSNES_CLASSID;
13 PetscLogEvent SNES_Solve, SNES_FunctionEval, SNES_JacobianEval, SNES_NGSEval, SNES_NGSFuncEval, SNES_NPCSolve, SNES_ObjectiveEval;
14 
15 /*@
16    SNESSetErrorIfNotConverged - Causes SNESSolve() to generate an error if the solver has not converged.
17 
18    Logically Collective on SNES
19 
20    Input Parameters:
21 +  snes - iterative context obtained from SNESCreate()
22 -  flg - PETSC_TRUE indicates you want the error generated
23 
24    Options database keys:
25 .  -snes_error_if_not_converged : this takes an optional truth value (0/1/no/yes/true/false)
26 
27    Level: intermediate
28 
29    Notes:
30     Normally PETSc continues if a linear solver fails to converge, you can call SNESGetConvergedReason() after a SNESSolve()
31     to determine if it has converged.
32 
33 .keywords: SNES, set, initial guess, nonzero
34 
35 .seealso: SNESGetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged()
36 @*/
37 PetscErrorCode  SNESSetErrorIfNotConverged(SNES snes,PetscBool flg)
38 {
39   PetscFunctionBegin;
40   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
41   PetscValidLogicalCollectiveBool(snes,flg,2);
42   snes->errorifnotconverged = flg;
43   PetscFunctionReturn(0);
44 }
45 
46 /*@
47    SNESGetErrorIfNotConverged - Will SNESSolve() generate an error if the solver does not converge?
48 
49    Not Collective
50 
51    Input Parameter:
52 .  snes - iterative context obtained from SNESCreate()
53 
54    Output Parameter:
55 .  flag - PETSC_TRUE if it will generate an error, else PETSC_FALSE
56 
57    Level: intermediate
58 
59 .keywords: SNES, set, initial guess, nonzero
60 
61 .seealso:  SNESSetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged()
62 @*/
63 PetscErrorCode  SNESGetErrorIfNotConverged(SNES snes,PetscBool  *flag)
64 {
65   PetscFunctionBegin;
66   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
67   PetscValidPointer(flag,2);
68   *flag = snes->errorifnotconverged;
69   PetscFunctionReturn(0);
70 }
71 
72 /*@
73     SNESSetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution?
74 
75    Logically Collective on SNES
76 
77     Input Parameters:
78 +   snes - the shell SNES
79 -   flg - is the residual computed?
80 
81    Level: advanced
82 
83 .seealso: SNESGetAlwaysComputesFinalResidual()
84 @*/
85 PetscErrorCode  SNESSetAlwaysComputesFinalResidual(SNES snes, PetscBool flg)
86 {
87   PetscFunctionBegin;
88   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
89   snes->alwayscomputesfinalresidual = flg;
90   PetscFunctionReturn(0);
91 }
92 
93 /*@
94     SNESGetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution?
95 
96    Logically Collective on SNES
97 
98     Input Parameter:
99 .   snes - the shell SNES
100 
101     Output Parameter:
102 .   flg - is the residual computed?
103 
104    Level: advanced
105 
106 .seealso: SNESSetAlwaysComputesFinalResidual()
107 @*/
108 PetscErrorCode  SNESGetAlwaysComputesFinalResidual(SNES snes, PetscBool *flg)
109 {
110   PetscFunctionBegin;
111   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
112   *flg = snes->alwayscomputesfinalresidual;
113   PetscFunctionReturn(0);
114 }
115 
116 /*@
117    SNESSetFunctionDomainError - tells SNES that the input vector to your SNESFunction is not
118      in the functions domain. For example, negative pressure.
119 
120    Logically Collective on SNES
121 
122    Input Parameters:
123 .  snes - the SNES context
124 
125    Level: advanced
126 
127 .keywords: SNES, view
128 
129 .seealso: SNESCreate(), SNESSetFunction(), SNESFunction
130 @*/
131 PetscErrorCode  SNESSetFunctionDomainError(SNES snes)
132 {
133   PetscFunctionBegin;
134   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
135   if (snes->errorifnotconverged) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates input vector is not in the function domain");
136   snes->domainerror = PETSC_TRUE;
137   PetscFunctionReturn(0);
138 }
139 
140 /*@
141    SNESGetFunctionDomainError - Gets the status of the domain error after a call to SNESComputeFunction;
142 
143    Logically Collective on SNES
144 
145    Input Parameters:
146 .  snes - the SNES context
147 
148    Output Parameters:
149 .  domainerror - Set to PETSC_TRUE if there's a domain error; PETSC_FALSE otherwise.
150 
151    Level: advanced
152 
153 .keywords: SNES, view
154 
155 .seealso: SNESSetFunctionDomainError(), SNESComputeFunction()
156 @*/
157 PetscErrorCode  SNESGetFunctionDomainError(SNES snes, PetscBool *domainerror)
158 {
159   PetscFunctionBegin;
160   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
161   PetscValidPointer(domainerror, 2);
162   *domainerror = snes->domainerror;
163   PetscFunctionReturn(0);
164 }
165 
166 /*@C
167   SNESLoad - Loads a SNES that has been stored in binary  with SNESView().
168 
169   Collective on PetscViewer
170 
171   Input Parameters:
172 + newdm - the newly loaded SNES, this needs to have been created with SNESCreate() or
173            some related function before a call to SNESLoad().
174 - viewer - binary file viewer, obtained from PetscViewerBinaryOpen()
175 
176    Level: intermediate
177 
178   Notes:
179    The type is determined by the data in the file, any type set into the SNES before this call is ignored.
180 
181   Notes for advanced users:
182   Most users should not need to know the details of the binary storage
183   format, since SNESLoad() and TSView() completely hide these details.
184   But for anyone who's interested, the standard binary matrix storage
185   format is
186 .vb
187      has not yet been determined
188 .ve
189 
190 .seealso: PetscViewerBinaryOpen(), SNESView(), MatLoad(), VecLoad()
191 @*/
192 PetscErrorCode  SNESLoad(SNES snes, PetscViewer viewer)
193 {
194   PetscErrorCode ierr;
195   PetscBool      isbinary;
196   PetscInt       classid;
197   char           type[256];
198   KSP            ksp;
199   DM             dm;
200   DMSNES         dmsnes;
201 
202   PetscFunctionBegin;
203   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
204   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
205   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
206   if (!isbinary) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen()");
207 
208   ierr = PetscViewerBinaryRead(viewer,&classid,1,NULL,PETSC_INT);CHKERRQ(ierr);
209   if (classid != SNES_FILE_CLASSID) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_WRONG,"Not SNES next in file");
210   ierr = PetscViewerBinaryRead(viewer,type,256,NULL,PETSC_CHAR);CHKERRQ(ierr);
211   ierr = SNESSetType(snes, type);CHKERRQ(ierr);
212   if (snes->ops->load) {
213     ierr = (*snes->ops->load)(snes,viewer);CHKERRQ(ierr);
214   }
215   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
216   ierr = DMGetDMSNES(dm,&dmsnes);CHKERRQ(ierr);
217   ierr = DMSNESLoad(dmsnes,viewer);CHKERRQ(ierr);
218   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
219   ierr = KSPLoad(ksp,viewer);CHKERRQ(ierr);
220   PetscFunctionReturn(0);
221 }
222 
223 #include <petscdraw.h>
224 #if defined(PETSC_HAVE_SAWS)
225 #include <petscviewersaws.h>
226 #endif
227 /*@C
228    SNESView - Prints the SNES data structure.
229 
230    Collective on SNES
231 
232    Input Parameters:
233 +  SNES - the SNES context
234 -  viewer - visualization context
235 
236    Options Database Key:
237 .  -snes_view - Calls SNESView() at end of SNESSolve()
238 
239    Notes:
240    The available visualization contexts include
241 +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
242 -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
243          output where only the first processor opens
244          the file.  All other processors send their
245          data to the first processor to print.
246 
247    The user can open an alternative visualization context with
248    PetscViewerASCIIOpen() - output to a specified file.
249 
250    Level: beginner
251 
252 .keywords: SNES, view
253 
254 .seealso: PetscViewerASCIIOpen()
255 @*/
256 PetscErrorCode  SNESView(SNES snes,PetscViewer viewer)
257 {
258   SNESKSPEW      *kctx;
259   PetscErrorCode ierr;
260   KSP            ksp;
261   SNESLineSearch linesearch;
262   PetscBool      iascii,isstring,isbinary,isdraw;
263   DMSNES         dmsnes;
264 #if defined(PETSC_HAVE_SAWS)
265   PetscBool      issaws;
266 #endif
267 
268   PetscFunctionBegin;
269   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
270   if (!viewer) {
271     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&viewer);CHKERRQ(ierr);
272   }
273   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
274   PetscCheckSameComm(snes,1,viewer,2);
275 
276   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
277   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr);
278   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
279   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
280 #if defined(PETSC_HAVE_SAWS)
281   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSAWS,&issaws);CHKERRQ(ierr);
282 #endif
283   if (iascii) {
284     SNESNormSchedule normschedule;
285 
286     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)snes,viewer);CHKERRQ(ierr);
287     if (!snes->setupcalled) {
288       ierr = PetscViewerASCIIPrintf(viewer,"  SNES has not been set up so information may be incomplete\n");CHKERRQ(ierr);
289     }
290     if (snes->ops->view) {
291       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
292       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
293       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
294     }
295     ierr = PetscViewerASCIIPrintf(viewer,"  maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr);
296     ierr = PetscViewerASCIIPrintf(viewer,"  tolerances: relative=%g, absolute=%g, solution=%g\n",(double)snes->rtol,(double)snes->abstol,(double)snes->stol);CHKERRQ(ierr);
297     if (snes->usesksp) {
298       ierr = PetscViewerASCIIPrintf(viewer,"  total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr);
299     }
300     ierr = PetscViewerASCIIPrintf(viewer,"  total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr);
301     ierr = SNESGetNormSchedule(snes, &normschedule);CHKERRQ(ierr);
302     if (normschedule > 0) {ierr = PetscViewerASCIIPrintf(viewer,"  norm schedule %s\n",SNESNormSchedules[normschedule]);CHKERRQ(ierr);}
303     if (snes->gridsequence) {
304       ierr = PetscViewerASCIIPrintf(viewer,"  total number of grid sequence refinements=%D\n",snes->gridsequence);CHKERRQ(ierr);
305     }
306     if (snes->ksp_ewconv) {
307       kctx = (SNESKSPEW*)snes->kspconvctx;
308       if (kctx) {
309         ierr = PetscViewerASCIIPrintf(viewer,"  Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr);
310         ierr = PetscViewerASCIIPrintf(viewer,"    rtol_0=%g, rtol_max=%g, threshold=%g\n",(double)kctx->rtol_0,(double)kctx->rtol_max,(double)kctx->threshold);CHKERRQ(ierr);
311         ierr = PetscViewerASCIIPrintf(viewer,"    gamma=%g, alpha=%g, alpha2=%g\n",(double)kctx->gamma,(double)kctx->alpha,(double)kctx->alpha2);CHKERRQ(ierr);
312       }
313     }
314     if (snes->lagpreconditioner == -1) {
315       ierr = PetscViewerASCIIPrintf(viewer,"  Preconditioned is never rebuilt\n");CHKERRQ(ierr);
316     } else if (snes->lagpreconditioner > 1) {
317       ierr = PetscViewerASCIIPrintf(viewer,"  Preconditioned is rebuilt every %D new Jacobians\n",snes->lagpreconditioner);CHKERRQ(ierr);
318     }
319     if (snes->lagjacobian == -1) {
320       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is never rebuilt\n");CHKERRQ(ierr);
321     } else if (snes->lagjacobian > 1) {
322       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is rebuilt every %D SNES iterations\n",snes->lagjacobian);CHKERRQ(ierr);
323     }
324   } else if (isstring) {
325     const char *type;
326     ierr = SNESGetType(snes,&type);CHKERRQ(ierr);
327     ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr);
328   } else if (isbinary) {
329     PetscInt    classid = SNES_FILE_CLASSID;
330     MPI_Comm    comm;
331     PetscMPIInt rank;
332     char        type[256];
333 
334     ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr);
335     ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
336     if (!rank) {
337       ierr = PetscViewerBinaryWrite(viewer,&classid,1,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr);
338       ierr = PetscStrncpy(type,((PetscObject)snes)->type_name,sizeof(type));CHKERRQ(ierr);
339       ierr = PetscViewerBinaryWrite(viewer,type,sizeof(type),PETSC_CHAR,PETSC_FALSE);CHKERRQ(ierr);
340     }
341     if (snes->ops->view) {
342       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
343     }
344   } else if (isdraw) {
345     PetscDraw draw;
346     char      str[36];
347     PetscReal x,y,bottom,h;
348 
349     ierr   = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
350     ierr   = PetscDrawGetCurrentPoint(draw,&x,&y);CHKERRQ(ierr);
351     ierr   = PetscStrcpy(str,"SNES: ");CHKERRQ(ierr);
352     ierr   = PetscStrcat(str,((PetscObject)snes)->type_name);CHKERRQ(ierr);
353     ierr   = PetscDrawStringBoxed(draw,x,y,PETSC_DRAW_BLUE,PETSC_DRAW_BLACK,str,NULL,&h);CHKERRQ(ierr);
354     bottom = y - h;
355     ierr   = PetscDrawPushCurrentPoint(draw,x,bottom);CHKERRQ(ierr);
356     if (snes->ops->view) {
357       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
358     }
359 #if defined(PETSC_HAVE_SAWS)
360   } else if (issaws) {
361     PetscMPIInt rank;
362     const char *name;
363 
364     ierr = PetscObjectGetName((PetscObject)snes,&name);CHKERRQ(ierr);
365     ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
366     if (!((PetscObject)snes)->amsmem && !rank) {
367       char       dir[1024];
368 
369       ierr = PetscObjectViewSAWs((PetscObject)snes,viewer);CHKERRQ(ierr);
370       ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/its",name);CHKERRQ(ierr);
371       PetscStackCallSAWs(SAWs_Register,(dir,&snes->iter,1,SAWs_READ,SAWs_INT));
372       if (!snes->conv_hist) {
373         ierr = SNESSetConvergenceHistory(snes,NULL,NULL,PETSC_DECIDE,PETSC_TRUE);CHKERRQ(ierr);
374       }
375       ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/conv_hist",name);CHKERRQ(ierr);
376       PetscStackCallSAWs(SAWs_Register,(dir,snes->conv_hist,10,SAWs_READ,SAWs_DOUBLE));
377     }
378 #endif
379   }
380   if (snes->linesearch) {
381     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
382     ierr = SNESGetLineSearch(snes, &linesearch);CHKERRQ(ierr);
383     ierr = SNESLineSearchView(linesearch, viewer);CHKERRQ(ierr);
384     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
385   }
386   if (snes->npc && snes->usesnpc) {
387     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
388     ierr = SNESView(snes->npc, viewer);CHKERRQ(ierr);
389     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
390   }
391   ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
392   ierr = DMGetDMSNES(snes->dm,&dmsnes);CHKERRQ(ierr);
393   ierr = DMSNESView(dmsnes, viewer);CHKERRQ(ierr);
394   ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
395   if (snes->usesksp) {
396     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
397     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
398     ierr = KSPView(ksp,viewer);CHKERRQ(ierr);
399     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
400   }
401   if (isdraw) {
402     PetscDraw draw;
403     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
404     ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr);
405   }
406   PetscFunctionReturn(0);
407 }
408 
409 /*
410   We retain a list of functions that also take SNES command
411   line options. These are called at the end SNESSetFromOptions()
412 */
413 #define MAXSETFROMOPTIONS 5
414 static PetscInt numberofsetfromoptions;
415 static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES);
416 
417 /*@C
418   SNESAddOptionsChecker - Adds an additional function to check for SNES options.
419 
420   Not Collective
421 
422   Input Parameter:
423 . snescheck - function that checks for options
424 
425   Level: developer
426 
427 .seealso: SNESSetFromOptions()
428 @*/
429 PetscErrorCode  SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES))
430 {
431   PetscFunctionBegin;
432   if (numberofsetfromoptions >= MAXSETFROMOPTIONS) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS);
433   othersetfromoptions[numberofsetfromoptions++] = snescheck;
434   PetscFunctionReturn(0);
435 }
436 
437 extern PetscErrorCode  SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*);
438 
439 static PetscErrorCode SNESSetUpMatrixFree_Private(SNES snes, PetscBool hasOperator, PetscInt version)
440 {
441   Mat            J;
442   KSP            ksp;
443   PC             pc;
444   PetscBool      match;
445   PetscErrorCode ierr;
446   MatNullSpace   nullsp;
447 
448   PetscFunctionBegin;
449   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
450 
451   if (!snes->vec_func && (snes->jacobian || snes->jacobian_pre)) {
452     Mat A = snes->jacobian, B = snes->jacobian_pre;
453     ierr = MatCreateVecs(A ? A : B, NULL,&snes->vec_func);CHKERRQ(ierr);
454   }
455 
456   if (version == 1) {
457     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
458     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
459     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
460   } else if (version == 2) {
461     if (!snes->vec_func) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"SNESSetFunction() must be called first");
462 #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_REAL_SINGLE) && !defined(PETSC_USE_REAL___FLOAT128) && !defined(PETSC_USE_REAL___FP16)
463     ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_func,&J);CHKERRQ(ierr);
464 #else
465     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP, "matrix-free operator rutines (version 2)");
466 #endif
467   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "matrix-free operator rutines, only version 1 and 2");
468 
469   /* attach any user provided null space that was on Amat to the newly created matrix free matrix */
470   if (snes->jacobian) {
471     ierr = MatGetNullSpace(snes->jacobian,&nullsp);CHKERRQ(ierr);
472     if (nullsp) {
473       ierr = MatSetNullSpace(J,nullsp);CHKERRQ(ierr);
474     }
475   }
476 
477   ierr = PetscInfo1(snes,"Setting default matrix-free operator routines (version %D)\n", version);CHKERRQ(ierr);
478   if (hasOperator) {
479 
480     /* This version replaces the user provided Jacobian matrix with a
481        matrix-free version but still employs the user-provided preconditioner matrix. */
482     ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);
483   } else {
484     /* This version replaces both the user-provided Jacobian and the user-
485      provided preconditioner Jacobian with the default matrix free version. */
486     if ((snes->npcside== PC_LEFT) && snes->npc) {
487       if (!snes->jacobian){ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);}
488     } else {
489       ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,0);CHKERRQ(ierr);
490     }
491     /* Force no preconditioner */
492     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
493     ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
494     ierr = PetscObjectTypeCompare((PetscObject)pc,PCSHELL,&match);CHKERRQ(ierr);
495     if (!match) {
496       ierr = PetscInfo(snes,"Setting default matrix-free preconditioner routines\nThat is no preconditioner is being used\n");CHKERRQ(ierr);
497       ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr);
498     }
499   }
500   ierr = MatDestroy(&J);CHKERRQ(ierr);
501   PetscFunctionReturn(0);
502 }
503 
504 static PetscErrorCode DMRestrictHook_SNESVecSol(DM dmfine,Mat Restrict,Vec Rscale,Mat Inject,DM dmcoarse,void *ctx)
505 {
506   SNES           snes = (SNES)ctx;
507   PetscErrorCode ierr;
508   Vec            Xfine,Xfine_named = NULL,Xcoarse;
509 
510   PetscFunctionBegin;
511   if (PetscLogPrintInfo) {
512     PetscInt finelevel,coarselevel,fineclevel,coarseclevel;
513     ierr = DMGetRefineLevel(dmfine,&finelevel);CHKERRQ(ierr);
514     ierr = DMGetCoarsenLevel(dmfine,&fineclevel);CHKERRQ(ierr);
515     ierr = DMGetRefineLevel(dmcoarse,&coarselevel);CHKERRQ(ierr);
516     ierr = DMGetCoarsenLevel(dmcoarse,&coarseclevel);CHKERRQ(ierr);
517     ierr = PetscInfo4(dmfine,"Restricting SNES solution vector from level %D-%D to level %D-%D\n",finelevel,fineclevel,coarselevel,coarseclevel);CHKERRQ(ierr);
518   }
519   if (dmfine == snes->dm) Xfine = snes->vec_sol;
520   else {
521     ierr  = DMGetNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);
522     Xfine = Xfine_named;
523   }
524   ierr = DMGetNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr);
525   if (Inject) {
526     ierr = MatRestrict(Inject,Xfine,Xcoarse);CHKERRQ(ierr);
527   } else {
528     ierr = MatRestrict(Restrict,Xfine,Xcoarse);CHKERRQ(ierr);
529     ierr = VecPointwiseMult(Xcoarse,Xcoarse,Rscale);CHKERRQ(ierr);
530   }
531   ierr = DMRestoreNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr);
532   if (Xfine_named) {ierr = DMRestoreNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);}
533   PetscFunctionReturn(0);
534 }
535 
536 static PetscErrorCode DMCoarsenHook_SNESVecSol(DM dm,DM dmc,void *ctx)
537 {
538   PetscErrorCode ierr;
539 
540   PetscFunctionBegin;
541   ierr = DMCoarsenHookAdd(dmc,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,ctx);CHKERRQ(ierr);
542   PetscFunctionReturn(0);
543 }
544 
545 /* This may be called to rediscretize the operator on levels of linear multigrid. The DM shuffle is so the user can
546  * safely call SNESGetDM() in their residual evaluation routine. */
547 static PetscErrorCode KSPComputeOperators_SNES(KSP ksp,Mat A,Mat B,void *ctx)
548 {
549   SNES           snes = (SNES)ctx;
550   PetscErrorCode ierr;
551   Mat            Asave = A,Bsave = B;
552   Vec            X,Xnamed = NULL;
553   DM             dmsave;
554   void           *ctxsave;
555   PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*);
556 
557   PetscFunctionBegin;
558   dmsave = snes->dm;
559   ierr   = KSPGetDM(ksp,&snes->dm);CHKERRQ(ierr);
560   if (dmsave == snes->dm) X = snes->vec_sol; /* We are on the finest level */
561   else {                                     /* We are on a coarser level, this vec was initialized using a DM restrict hook */
562     ierr = DMGetNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr);
563     X    = Xnamed;
564     ierr = SNESGetJacobian(snes,NULL,NULL,&jac,&ctxsave);CHKERRQ(ierr);
565     /* If the DM's don't match up, the MatFDColoring context needed for the jacobian won't match up either -- fixit. */
566     if (jac == SNESComputeJacobianDefaultColor) {
567       ierr = SNESSetJacobian(snes,NULL,NULL,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr);
568     }
569   }
570   /* put the previous context back */
571 
572   ierr = SNESComputeJacobian(snes,X,A,B);CHKERRQ(ierr);
573   if (snes->dm != dmsave && jac == SNESComputeJacobianDefaultColor) {
574     ierr = SNESSetJacobian(snes,NULL,NULL,jac,ctxsave);CHKERRQ(ierr);
575   }
576 
577   if (A != Asave || B != Bsave) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_SUP,"No support for changing matrices at this time");
578   if (Xnamed) {
579     ierr = DMRestoreNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr);
580   }
581   snes->dm = dmsave;
582   PetscFunctionReturn(0);
583 }
584 
585 /*@
586    SNESSetUpMatrices - ensures that matrices are available for SNES, to be called by SNESSetUp_XXX()
587 
588    Collective
589 
590    Input Arguments:
591 .  snes - snes to configure
592 
593    Level: developer
594 
595 .seealso: SNESSetUp()
596 @*/
597 PetscErrorCode SNESSetUpMatrices(SNES snes)
598 {
599   PetscErrorCode ierr;
600   DM             dm;
601   DMSNES         sdm;
602 
603   PetscFunctionBegin;
604   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
605   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
606   if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_PLIB,"DMSNES not properly configured");
607   else if (!snes->jacobian && snes->mf) {
608     Mat  J;
609     void *functx;
610     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
611     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
612     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
613     ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr);
614     ierr = SNESSetJacobian(snes,J,J,0,0);CHKERRQ(ierr);
615     ierr = MatDestroy(&J);CHKERRQ(ierr);
616   } else if (snes->mf_operator && !snes->jacobian_pre && !snes->jacobian) {
617     Mat J,B;
618     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
619     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
620     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
621     ierr = DMCreateMatrix(snes->dm,&B);CHKERRQ(ierr);
622     /* sdm->computejacobian was already set to reach here */
623     ierr = SNESSetJacobian(snes,J,B,NULL,NULL);CHKERRQ(ierr);
624     ierr = MatDestroy(&J);CHKERRQ(ierr);
625     ierr = MatDestroy(&B);CHKERRQ(ierr);
626   } else if (!snes->jacobian_pre) {
627     PetscDS   prob;
628     Mat       J, B;
629     PetscBool hasPrec = PETSC_FALSE;
630 
631     J    = snes->jacobian;
632     ierr = DMGetDS(dm, &prob);CHKERRQ(ierr);
633     if (prob) {ierr = PetscDSHasJacobianPreconditioner(prob, &hasPrec);CHKERRQ(ierr);}
634     if (J)            {ierr = PetscObjectReference((PetscObject) J);CHKERRQ(ierr);}
635     else if (hasPrec) {ierr = DMCreateMatrix(snes->dm, &J);CHKERRQ(ierr);}
636     ierr = DMCreateMatrix(snes->dm, &B);CHKERRQ(ierr);
637     ierr = SNESSetJacobian(snes, J ? J : B, B, NULL, NULL);CHKERRQ(ierr);
638     ierr = MatDestroy(&J);CHKERRQ(ierr);
639     ierr = MatDestroy(&B);CHKERRQ(ierr);
640   }
641   {
642     KSP ksp;
643     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
644     ierr = KSPSetComputeOperators(ksp,KSPComputeOperators_SNES,snes);CHKERRQ(ierr);
645     ierr = DMCoarsenHookAdd(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr);
646   }
647   PetscFunctionReturn(0);
648 }
649 
650 /*@C
651    SNESMonitorSetFromOptions - Sets a monitor function and viewer appropriate for the type indicated by the user
652 
653    Collective on SNES
654 
655    Input Parameters:
656 +  snes - SNES object you wish to monitor
657 .  name - the monitor type one is seeking
658 .  help - message indicating what monitoring is done
659 .  manual - manual page for the monitor
660 .  monitor - the monitor function
661 -  monitorsetup - a function that is called once ONLY if the user selected this monitor that may set additional features of the SNES or PetscViewer objects
662 
663    Level: developer
664 
665 .seealso: PetscOptionsGetViewer(), PetscOptionsGetReal(), PetscOptionsHasName(), PetscOptionsGetString(),
666           PetscOptionsGetIntArray(), PetscOptionsGetRealArray(), PetscOptionsBool()
667           PetscOptionsInt(), PetscOptionsString(), PetscOptionsReal(), PetscOptionsBool(),
668           PetscOptionsName(), PetscOptionsBegin(), PetscOptionsEnd(), PetscOptionsHead(),
669           PetscOptionsStringArray(),PetscOptionsRealArray(), PetscOptionsScalar(),
670           PetscOptionsBoolGroupBegin(), PetscOptionsBoolGroup(), PetscOptionsBoolGroupEnd(),
671           PetscOptionsFList(), PetscOptionsEList()
672 @*/
673 PetscErrorCode  SNESMonitorSetFromOptions(SNES snes,const char name[],const char help[], const char manual[],PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,PetscViewerAndFormat*),PetscErrorCode (*monitorsetup)(SNES,PetscViewerAndFormat*))
674 {
675   PetscErrorCode    ierr;
676   PetscViewer       viewer;
677   PetscViewerFormat format;
678   PetscBool         flg;
679 
680   PetscFunctionBegin;
681   ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,name,&viewer,&format,&flg);CHKERRQ(ierr);
682   if (flg) {
683     PetscViewerAndFormat *vf;
684     ierr = PetscViewerAndFormatCreate(viewer,format,&vf);CHKERRQ(ierr);
685     ierr = PetscObjectDereference((PetscObject)viewer);CHKERRQ(ierr);
686     if (monitorsetup) {
687       ierr = (*monitorsetup)(snes,vf);CHKERRQ(ierr);
688     }
689     ierr = SNESMonitorSet(snes,(PetscErrorCode (*)(SNES,PetscInt,PetscReal,void*))monitor,vf,(PetscErrorCode (*)(void**))PetscViewerAndFormatDestroy);CHKERRQ(ierr);
690   }
691   PetscFunctionReturn(0);
692 }
693 
694 /*@
695    SNESSetFromOptions - Sets various SNES and KSP parameters from user options.
696 
697    Collective on SNES
698 
699    Input Parameter:
700 .  snes - the SNES context
701 
702    Options Database Keys:
703 +  -snes_type <type> - newtonls, newtontr, ngmres, ncg, nrichardson, qn, vi, fas, SNESType for complete list
704 .  -snes_stol - convergence tolerance in terms of the norm
705                 of the change in the solution between steps
706 .  -snes_atol <abstol> - absolute tolerance of residual norm
707 .  -snes_rtol <rtol> - relative decrease in tolerance norm from initial
708 .  -snes_divergence_tolerance <divtol> - if the residual goes above divtol*rnorm0, exit with divergence
709 .  -snes_force_iteration <force> - force SNESSolve() to take at least one iteration
710 .  -snes_max_it <max_it> - maximum number of iterations
711 .  -snes_max_funcs <max_funcs> - maximum number of function evaluations
712 .  -snes_max_fail <max_fail> - maximum number of line search failures allowed before stopping, default is none
713 .  -snes_max_linear_solve_fail - number of linear solver failures before SNESSolve() stops
714 .  -snes_lag_preconditioner <lag> - how often preconditioner is rebuilt (use -1 to never rebuild)
715 .  -snes_lag_jacobian <lag> - how often Jacobian is rebuilt (use -1 to never rebuild)
716 .  -snes_trtol <trtol> - trust region tolerance
717 .  -snes_no_convergence_test - skip convergence test in nonlinear
718                                solver; hence iterations will continue until max_it
719                                or some other criterion is reached. Saves expense
720                                of convergence test
721 .  -snes_monitor [ascii][:filename][:viewer format] - prints residual norm at each iteration. if no filename given prints to stdout
722 .  -snes_monitor_solution [ascii binary draw][:filename][:viewer format] - plots solution at each iteration
723 .  -snes_monitor_residual [ascii binary draw][:filename][:viewer format] - plots residual (not its norm) at each iteration
724 .  -snes_monitor_solution_update [ascii binary draw][:filename][:viewer format] - plots update to solution at each iteration
725 .  -snes_monitor_lg_residualnorm - plots residual norm at each iteration
726 .  -snes_monitor_lg_range - plots residual norm at each iteration
727 .  -snes_fd - use finite differences to compute Jacobian; very slow, only for testing
728 .  -snes_fd_color - use finite differences with coloring to compute Jacobian
729 .  -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration
730 -  -snes_converged_reason - print the reason for convergence/divergence after each solve
731 
732     Options Database for Eisenstat-Walker method:
733 +  -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence
734 .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
735 .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
736 .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
737 .  -snes_ksp_ew_gamma <gamma> - Sets gamma
738 .  -snes_ksp_ew_alpha <alpha> - Sets alpha
739 .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
740 -  -snes_ksp_ew_threshold <threshold> - Sets threshold
741 
742    Notes:
743    To see all options, run your program with the -help option or consult
744    Users-Manual: ch_snes
745 
746    Level: beginner
747 
748 .keywords: SNES, nonlinear, set, options, database
749 
750 .seealso: SNESSetOptionsPrefix()
751 @*/
752 PetscErrorCode  SNESSetFromOptions(SNES snes)
753 {
754   PetscBool      flg,pcset,persist,set;
755   PetscInt       i,indx,lag,grids;
756   const char     *deft        = SNESNEWTONLS;
757   const char     *convtests[] = {"default","skip"};
758   SNESKSPEW      *kctx        = NULL;
759   char           type[256], monfilename[PETSC_MAX_PATH_LEN];
760   PetscErrorCode ierr;
761   PCSide         pcside;
762   const char     *optionsprefix;
763 
764   PetscFunctionBegin;
765   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
766   ierr = SNESRegisterAll();CHKERRQ(ierr);
767   ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr);
768   if (((PetscObject)snes)->type_name) deft = ((PetscObject)snes)->type_name;
769   ierr = PetscOptionsFList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr);
770   if (flg) {
771     ierr = SNESSetType(snes,type);CHKERRQ(ierr);
772   } else if (!((PetscObject)snes)->type_name) {
773     ierr = SNESSetType(snes,deft);CHKERRQ(ierr);
774   }
775   ierr = PetscOptionsReal("-snes_stol","Stop if step length less than","SNESSetTolerances",snes->stol,&snes->stol,NULL);CHKERRQ(ierr);
776   ierr = PetscOptionsReal("-snes_atol","Stop if function norm less than","SNESSetTolerances",snes->abstol,&snes->abstol,NULL);CHKERRQ(ierr);
777 
778   ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less than","SNESSetTolerances",snes->rtol,&snes->rtol,NULL);CHKERRQ(ierr);
779   ierr = PetscOptionsReal("-snes_divergence_tolerance","Stop if residual norm increases by this factor","SNESSetDivergenceTolerance",snes->divtol,&snes->divtol,NULL);CHKERRQ(ierr);
780   ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,NULL);CHKERRQ(ierr);
781   ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,NULL);CHKERRQ(ierr);
782   ierr = PetscOptionsInt("-snes_max_fail","Maximum nonlinear step failures","SNESSetMaxNonlinearStepFailures",snes->maxFailures,&snes->maxFailures,NULL);CHKERRQ(ierr);
783   ierr = PetscOptionsInt("-snes_max_linear_solve_fail","Maximum failures in linear solves allowed","SNESSetMaxLinearSolveFailures",snes->maxLinearSolveFailures,&snes->maxLinearSolveFailures,NULL);CHKERRQ(ierr);
784   ierr = PetscOptionsBool("-snes_error_if_not_converged","Generate error if solver does not converge","SNESSetErrorIfNotConverged",snes->errorifnotconverged,&snes->errorifnotconverged,NULL);CHKERRQ(ierr);
785   ierr = PetscOptionsBool("-snes_force_iteration","Force SNESSolve() to take at least one iteration","SNESForceIteration",snes->forceiteration,&snes->forceiteration,NULL);CHKERRQ(ierr);
786 
787   ierr = PetscOptionsInt("-snes_lag_preconditioner","How often to rebuild preconditioner","SNESSetLagPreconditioner",snes->lagpreconditioner,&lag,&flg);CHKERRQ(ierr);
788   if (flg) {
789     ierr = SNESSetLagPreconditioner(snes,lag);CHKERRQ(ierr);
790   }
791   ierr = PetscOptionsBool("-snes_lag_preconditioner_persists","Preconditioner lagging through multiple solves","SNESSetLagPreconditionerPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr);
792   if (flg) {
793     ierr = SNESSetLagPreconditionerPersists(snes,persist);CHKERRQ(ierr);
794   }
795   ierr = PetscOptionsInt("-snes_lag_jacobian","How often to rebuild Jacobian","SNESSetLagJacobian",snes->lagjacobian,&lag,&flg);CHKERRQ(ierr);
796   if (flg) {
797     ierr = SNESSetLagJacobian(snes,lag);CHKERRQ(ierr);
798   }
799   ierr = PetscOptionsBool("-snes_lag_jacobian_persists","Jacobian lagging through multiple solves","SNESSetLagJacobianPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr);
800   if (flg) {
801     ierr = SNESSetLagJacobianPersists(snes,persist);CHKERRQ(ierr);
802   }
803 
804   ierr = PetscOptionsInt("-snes_grid_sequence","Use grid sequencing to generate initial guess","SNESSetGridSequence",snes->gridsequence,&grids,&flg);CHKERRQ(ierr);
805   if (flg) {
806     ierr = SNESSetGridSequence(snes,grids);CHKERRQ(ierr);
807   }
808 
809   ierr = PetscOptionsEList("-snes_convergence_test","Convergence test","SNESSetConvergenceTest",convtests,2,"default",&indx,&flg);CHKERRQ(ierr);
810   if (flg) {
811     switch (indx) {
812     case 0: ierr = SNESSetConvergenceTest(snes,SNESConvergedDefault,NULL,NULL);CHKERRQ(ierr); break;
813     case 1: ierr = SNESSetConvergenceTest(snes,SNESConvergedSkip,NULL,NULL);CHKERRQ(ierr);    break;
814     }
815   }
816 
817   ierr = PetscOptionsEList("-snes_norm_schedule","SNES Norm schedule","SNESSetNormSchedule",SNESNormSchedules,5,"function",&indx,&flg);CHKERRQ(ierr);
818   if (flg) { ierr = SNESSetNormSchedule(snes,(SNESNormSchedule)indx);CHKERRQ(ierr); }
819 
820   ierr = PetscOptionsEList("-snes_function_type","SNES Norm schedule","SNESSetFunctionType",SNESFunctionTypes,2,"unpreconditioned",&indx,&flg);CHKERRQ(ierr);
821   if (flg) { ierr = SNESSetFunctionType(snes,(SNESFunctionType)indx);CHKERRQ(ierr); }
822 
823   kctx = (SNESKSPEW*)snes->kspconvctx;
824 
825   ierr = PetscOptionsBool("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,NULL);CHKERRQ(ierr);
826 
827   ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,NULL);CHKERRQ(ierr);
828   ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,NULL);CHKERRQ(ierr);
829   ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,NULL);CHKERRQ(ierr);
830   ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,NULL);CHKERRQ(ierr);
831   ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,NULL);CHKERRQ(ierr);
832   ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,NULL);CHKERRQ(ierr);
833   ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,NULL);CHKERRQ(ierr);
834 
835   flg  = PETSC_FALSE;
836   ierr = PetscOptionsBool("-snes_check_jacobian","Check each Jacobian with a differenced one","SNESUpdateCheckJacobian",flg,&flg,&set);CHKERRQ(ierr);
837   if (set && flg) {
838     ierr = SNESSetUpdate(snes,SNESUpdateCheckJacobian);CHKERRQ(ierr);
839   }
840 
841   flg  = PETSC_FALSE;
842   ierr = PetscOptionsBool("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",flg,&flg,&set);CHKERRQ(ierr);
843   if (set && flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);}
844 
845   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor","Monitor norm of function","SNESMonitorDefault",SNESMonitorDefault,NULL);CHKERRQ(ierr);
846   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_short","Monitor norm of function with fewer digits","SNESMonitorDefaultShort",SNESMonitorDefaultShort,NULL);CHKERRQ(ierr);
847   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_range","Monitor range of elements of function","SNESMonitorRange",SNESMonitorRange,NULL);CHKERRQ(ierr);
848 
849   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_ratio","Monitor ratios of the norm of function for consecutive steps","SNESMonitorRatio",SNESMonitorRatio,SNESMonitorRatioSetUp);CHKERRQ(ierr);
850   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_field","Monitor norm of function (split into fields)","SNESMonitorDefaultField",SNESMonitorDefaultField,NULL);CHKERRQ(ierr);
851   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution","View solution at each iteration","SNESMonitorSolution",SNESMonitorSolution,NULL);CHKERRQ(ierr);
852   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution_update","View correction at each iteration","SNESMonitorSolutionUpdate",SNESMonitorSolutionUpdate,NULL);CHKERRQ(ierr);
853   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_residual","View residual at each iteration","SNESMonitorResidual",SNESMonitorResidual,NULL);CHKERRQ(ierr);
854   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_jacupdate_spectrum","Print the change in the spectrum of the Jacobian","SNESMonitorJacUpdateSpectrum",SNESMonitorJacUpdateSpectrum,NULL);CHKERRQ(ierr);
855   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_fields","Monitor norm of function per field","SNESMonitorSet",SNESMonitorFields,NULL);CHKERRQ(ierr);
856 
857   ierr = PetscOptionsString("-snes_monitor_python","Use Python function","SNESMonitorSet",0,monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
858   if (flg) {ierr = PetscPythonMonitorSet((PetscObject)snes,monfilename);CHKERRQ(ierr);}
859 
860 
861   flg  = PETSC_FALSE;
862   ierr = PetscOptionsBool("-snes_monitor_lg_residualnorm","Plot function norm at each iteration","SNESMonitorLGResidualNorm",flg,&flg,NULL);CHKERRQ(ierr);
863   if (flg) {
864     PetscDrawLG ctx;
865 
866     ierr = SNESMonitorLGCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr);
867     ierr = SNESMonitorSet(snes,SNESMonitorLGResidualNorm,ctx,(PetscErrorCode (*)(void**))PetscDrawLGDestroy);CHKERRQ(ierr);
868   }
869   flg  = PETSC_FALSE;
870   ierr = PetscOptionsBool("-snes_monitor_lg_range","Plot function range at each iteration","SNESMonitorLGRange",flg,&flg,NULL);CHKERRQ(ierr);
871   if (flg) {
872     PetscViewer ctx;
873 
874     ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr);
875     ierr = SNESMonitorSet(snes,SNESMonitorLGRange,ctx,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
876   }
877 
878 
879 
880   flg  = PETSC_FALSE;
881   ierr = PetscOptionsBool("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESComputeJacobianDefault",flg,&flg,NULL);CHKERRQ(ierr);
882   if (flg) {
883     void    *functx;
884     DM      dm;
885     DMSNES  sdm;
886     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
887     ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
888     sdm->jacobianctx = NULL;
889     ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr);
890     ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefault,functx);CHKERRQ(ierr);
891     ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr);
892   }
893 
894   flg  = PETSC_FALSE;
895   ierr = PetscOptionsBool("-snes_fd_function","Use finite differences (slow) to compute function from user objective","SNESObjectiveComputeFunctionDefaultFD",flg,&flg,NULL);CHKERRQ(ierr);
896   if (flg) {
897     ierr = SNESSetFunction(snes,NULL,SNESObjectiveComputeFunctionDefaultFD,NULL);CHKERRQ(ierr);
898   }
899 
900   flg  = PETSC_FALSE;
901   ierr = PetscOptionsBool("-snes_fd_color","Use finite differences with coloring to compute Jacobian","SNESComputeJacobianDefaultColor",flg,&flg,NULL);CHKERRQ(ierr);
902   if (flg) {
903     DM             dm;
904     DMSNES         sdm;
905     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
906     ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
907     sdm->jacobianctx = NULL;
908     ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr);
909     ierr = PetscInfo(snes,"Setting default finite difference coloring Jacobian matrix\n");CHKERRQ(ierr);
910   }
911 
912   flg  = PETSC_FALSE;
913   ierr = PetscOptionsBool("-snes_mf_operator","Use a Matrix-Free Jacobian with user-provided preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf_operator,&flg);CHKERRQ(ierr);
914   if (flg && snes->mf_operator) {
915     snes->mf_operator = PETSC_TRUE;
916     snes->mf          = PETSC_TRUE;
917   }
918   flg  = PETSC_FALSE;
919   ierr = PetscOptionsBool("-snes_mf","Use a Matrix-Free Jacobian with no preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf,&flg);CHKERRQ(ierr);
920   if (!flg && snes->mf_operator) snes->mf = PETSC_TRUE;
921   ierr = PetscOptionsInt("-snes_mf_version","Matrix-Free routines version 1 or 2","None",snes->mf_version,&snes->mf_version,0);CHKERRQ(ierr);
922 
923   flg  = PETSC_FALSE;
924   ierr = SNESGetNPCSide(snes,&pcside);CHKERRQ(ierr);
925   ierr = PetscOptionsEnum("-snes_npc_side","SNES nonlinear preconditioner side","SNESSetNPCSide",PCSides,(PetscEnum)pcside,(PetscEnum*)&pcside,&flg);CHKERRQ(ierr);
926   if (flg) {ierr = SNESSetNPCSide(snes,pcside);CHKERRQ(ierr);}
927 
928 #if defined(PETSC_HAVE_SAWS)
929   /*
930     Publish convergence information using SAWs
931   */
932   flg  = PETSC_FALSE;
933   ierr = PetscOptionsBool("-snes_monitor_saws","Publish SNES progress using SAWs","SNESMonitorSet",flg,&flg,NULL);CHKERRQ(ierr);
934   if (flg) {
935     void *ctx;
936     ierr = SNESMonitorSAWsCreate(snes,&ctx);CHKERRQ(ierr);
937     ierr = SNESMonitorSet(snes,SNESMonitorSAWs,ctx,SNESMonitorSAWsDestroy);CHKERRQ(ierr);
938   }
939 #endif
940 #if defined(PETSC_HAVE_SAWS)
941   {
942   PetscBool set;
943   flg  = PETSC_FALSE;
944   ierr = PetscOptionsBool("-snes_saws_block","Block for SAWs at end of SNESSolve","PetscObjectSAWsBlock",((PetscObject)snes)->amspublishblock,&flg,&set);CHKERRQ(ierr);
945   if (set) {
946     ierr = PetscObjectSAWsSetBlock((PetscObject)snes,flg);CHKERRQ(ierr);
947   }
948   }
949 #endif
950 
951   for (i = 0; i < numberofsetfromoptions; i++) {
952     ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr);
953   }
954 
955   if (snes->ops->setfromoptions) {
956     ierr = (*snes->ops->setfromoptions)(PetscOptionsObject,snes);CHKERRQ(ierr);
957   }
958 
959   /* process any options handlers added with PetscObjectAddOptionsHandler() */
960   ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject)snes);CHKERRQ(ierr);
961   ierr = PetscOptionsEnd();CHKERRQ(ierr);
962 
963   if (!snes->linesearch) {
964     ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr);
965   }
966   ierr = SNESLineSearchSetFromOptions(snes->linesearch);CHKERRQ(ierr);
967 
968   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
969   ierr = KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre);CHKERRQ(ierr);
970   ierr = KSPSetFromOptions(snes->ksp);CHKERRQ(ierr);
971 
972   /* if someone has set the SNES NPC type, create it. */
973   ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr);
974   ierr = PetscOptionsHasName(((PetscObject)snes)->options,optionsprefix, "-npc_snes_type", &pcset);CHKERRQ(ierr);
975   if (pcset && (!snes->npc)) {
976     ierr = SNESGetNPC(snes, &snes->npc);CHKERRQ(ierr);
977   }
978   PetscFunctionReturn(0);
979 }
980 
981 /*@C
982    SNESSetComputeApplicationContext - Sets an optional function to compute a user-defined context for
983    the nonlinear solvers.
984 
985    Logically Collective on SNES
986 
987    Input Parameters:
988 +  snes - the SNES context
989 .  compute - function to compute the context
990 -  destroy - function to destroy the context
991 
992    Level: intermediate
993 
994    Notes:
995    This function is currently not available from Fortran.
996 
997 .keywords: SNES, nonlinear, set, application, context
998 
999 .seealso: SNESGetApplicationContext(), SNESSetComputeApplicationContext(), SNESGetApplicationContext()
1000 @*/
1001 PetscErrorCode  SNESSetComputeApplicationContext(SNES snes,PetscErrorCode (*compute)(SNES,void**),PetscErrorCode (*destroy)(void**))
1002 {
1003   PetscFunctionBegin;
1004   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1005   snes->ops->usercompute = compute;
1006   snes->ops->userdestroy = destroy;
1007   PetscFunctionReturn(0);
1008 }
1009 
1010 /*@
1011    SNESSetApplicationContext - Sets the optional user-defined context for
1012    the nonlinear solvers.
1013 
1014    Logically Collective on SNES
1015 
1016    Input Parameters:
1017 +  snes - the SNES context
1018 -  usrP - optional user context
1019 
1020    Level: intermediate
1021 
1022    Fortran Notes: To use this from Fortran you must write a Fortran interface definition for this
1023     function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument.
1024 
1025 .keywords: SNES, nonlinear, set, application, context
1026 
1027 .seealso: SNESGetApplicationContext()
1028 @*/
1029 PetscErrorCode  SNESSetApplicationContext(SNES snes,void *usrP)
1030 {
1031   PetscErrorCode ierr;
1032   KSP            ksp;
1033 
1034   PetscFunctionBegin;
1035   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1036   ierr       = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
1037   ierr       = KSPSetApplicationContext(ksp,usrP);CHKERRQ(ierr);
1038   snes->user = usrP;
1039   PetscFunctionReturn(0);
1040 }
1041 
1042 /*@
1043    SNESGetApplicationContext - Gets the user-defined context for the
1044    nonlinear solvers.
1045 
1046    Not Collective
1047 
1048    Input Parameter:
1049 .  snes - SNES context
1050 
1051    Output Parameter:
1052 .  usrP - user context
1053 
1054    Fortran Notes: To use this from Fortran you must write a Fortran interface definition for this
1055     function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument.
1056 
1057    Level: intermediate
1058 
1059 .keywords: SNES, nonlinear, get, application, context
1060 
1061 .seealso: SNESSetApplicationContext()
1062 @*/
1063 PetscErrorCode  SNESGetApplicationContext(SNES snes,void *usrP)
1064 {
1065   PetscFunctionBegin;
1066   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1067   *(void**)usrP = snes->user;
1068   PetscFunctionReturn(0);
1069 }
1070 
1071 /*@
1072    SNESSetUseMatrixFree - indicates that SNES should use matrix free finite difference matrix vector products internally to apply
1073                           the Jacobian.
1074 
1075    Collective on SNES
1076 
1077    Input Parameters:
1078 +  snes - SNES context
1079 .  mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored
1080 -  mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used
1081 
1082    Options Database:
1083 + -snes_mf - use matrix free for both the mat and pmat operator
1084 - -snes_mf_operator - use matrix free only for the mat operator
1085 
1086    Level: intermediate
1087 
1088 .keywords: SNES, nonlinear, get, iteration, number,
1089 
1090 .seealso:   SNESGetUseMatrixFree(), MatCreateSNESMF()
1091 @*/
1092 PetscErrorCode  SNESSetUseMatrixFree(SNES snes,PetscBool mf_operator,PetscBool mf)
1093 {
1094   PetscFunctionBegin;
1095   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1096   PetscValidLogicalCollectiveBool(snes,mf_operator,2);
1097   PetscValidLogicalCollectiveBool(snes,mf,3);
1098   if (mf && !mf_operator) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"If using mf must also use mf_operator");
1099   snes->mf          = mf;
1100   snes->mf_operator = mf_operator;
1101   PetscFunctionReturn(0);
1102 }
1103 
1104 /*@
1105    SNESGetUseMatrixFree - indicates if the SNES uses matrix free finite difference matrix vector products to apply
1106                           the Jacobian.
1107 
1108    Collective on SNES
1109 
1110    Input Parameter:
1111 .  snes - SNES context
1112 
1113    Output Parameters:
1114 +  mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored
1115 -  mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used
1116 
1117    Options Database:
1118 + -snes_mf - use matrix free for both the mat and pmat operator
1119 - -snes_mf_operator - use matrix free only for the mat operator
1120 
1121    Level: intermediate
1122 
1123 .keywords: SNES, nonlinear, get, iteration, number,
1124 
1125 .seealso:   SNESSetUseMatrixFree(), MatCreateSNESMF()
1126 @*/
1127 PetscErrorCode  SNESGetUseMatrixFree(SNES snes,PetscBool *mf_operator,PetscBool *mf)
1128 {
1129   PetscFunctionBegin;
1130   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1131   if (mf)          *mf          = snes->mf;
1132   if (mf_operator) *mf_operator = snes->mf_operator;
1133   PetscFunctionReturn(0);
1134 }
1135 
1136 /*@
1137    SNESGetIterationNumber - Gets the number of nonlinear iterations completed
1138    at this time.
1139 
1140    Not Collective
1141 
1142    Input Parameter:
1143 .  snes - SNES context
1144 
1145    Output Parameter:
1146 .  iter - iteration number
1147 
1148    Notes:
1149    For example, during the computation of iteration 2 this would return 1.
1150 
1151    This is useful for using lagged Jacobians (where one does not recompute the
1152    Jacobian at each SNES iteration). For example, the code
1153 .vb
1154       ierr = SNESGetIterationNumber(snes,&it);
1155       if (!(it % 2)) {
1156         [compute Jacobian here]
1157       }
1158 .ve
1159    can be used in your ComputeJacobian() function to cause the Jacobian to be
1160    recomputed every second SNES iteration.
1161 
1162    After the SNES solve is complete this will return the number of nonlinear iterations used.
1163 
1164    Level: intermediate
1165 
1166 .keywords: SNES, nonlinear, get, iteration, number,
1167 
1168 .seealso:   SNESGetLinearSolveIterations()
1169 @*/
1170 PetscErrorCode  SNESGetIterationNumber(SNES snes,PetscInt *iter)
1171 {
1172   PetscFunctionBegin;
1173   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1174   PetscValidIntPointer(iter,2);
1175   *iter = snes->iter;
1176   PetscFunctionReturn(0);
1177 }
1178 
1179 /*@
1180    SNESSetIterationNumber - Sets the current iteration number.
1181 
1182    Not Collective
1183 
1184    Input Parameter:
1185 .  snes - SNES context
1186 .  iter - iteration number
1187 
1188    Level: developer
1189 
1190 .keywords: SNES, nonlinear, set, iteration, number,
1191 
1192 .seealso:   SNESGetLinearSolveIterations()
1193 @*/
1194 PetscErrorCode  SNESSetIterationNumber(SNES snes,PetscInt iter)
1195 {
1196   PetscErrorCode ierr;
1197 
1198   PetscFunctionBegin;
1199   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1200   ierr       = PetscObjectSAWsTakeAccess((PetscObject)snes);CHKERRQ(ierr);
1201   snes->iter = iter;
1202   ierr       = PetscObjectSAWsGrantAccess((PetscObject)snes);CHKERRQ(ierr);
1203   PetscFunctionReturn(0);
1204 }
1205 
1206 /*@
1207    SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps
1208    attempted by the nonlinear solver.
1209 
1210    Not Collective
1211 
1212    Input Parameter:
1213 .  snes - SNES context
1214 
1215    Output Parameter:
1216 .  nfails - number of unsuccessful steps attempted
1217 
1218    Notes:
1219    This counter is reset to zero for each successive call to SNESSolve().
1220 
1221    Level: intermediate
1222 
1223 .keywords: SNES, nonlinear, get, number, unsuccessful, steps
1224 
1225 .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
1226           SNESSetMaxNonlinearStepFailures(), SNESGetMaxNonlinearStepFailures()
1227 @*/
1228 PetscErrorCode  SNESGetNonlinearStepFailures(SNES snes,PetscInt *nfails)
1229 {
1230   PetscFunctionBegin;
1231   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1232   PetscValidIntPointer(nfails,2);
1233   *nfails = snes->numFailures;
1234   PetscFunctionReturn(0);
1235 }
1236 
1237 /*@
1238    SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps
1239    attempted by the nonlinear solver before it gives up.
1240 
1241    Not Collective
1242 
1243    Input Parameters:
1244 +  snes     - SNES context
1245 -  maxFails - maximum of unsuccessful steps
1246 
1247    Level: intermediate
1248 
1249 .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
1250 
1251 .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
1252           SNESGetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures()
1253 @*/
1254 PetscErrorCode  SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails)
1255 {
1256   PetscFunctionBegin;
1257   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1258   snes->maxFailures = maxFails;
1259   PetscFunctionReturn(0);
1260 }
1261 
1262 /*@
1263    SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps
1264    attempted by the nonlinear solver before it gives up.
1265 
1266    Not Collective
1267 
1268    Input Parameter:
1269 .  snes     - SNES context
1270 
1271    Output Parameter:
1272 .  maxFails - maximum of unsuccessful steps
1273 
1274    Level: intermediate
1275 
1276 .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
1277 
1278 .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
1279           SNESSetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures()
1280 
1281 @*/
1282 PetscErrorCode  SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails)
1283 {
1284   PetscFunctionBegin;
1285   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1286   PetscValidIntPointer(maxFails,2);
1287   *maxFails = snes->maxFailures;
1288   PetscFunctionReturn(0);
1289 }
1290 
1291 /*@
1292    SNESGetNumberFunctionEvals - Gets the number of user provided function evaluations
1293      done by SNES.
1294 
1295    Not Collective
1296 
1297    Input Parameter:
1298 .  snes     - SNES context
1299 
1300    Output Parameter:
1301 .  nfuncs - number of evaluations
1302 
1303    Level: intermediate
1304 
1305    Notes: Reset every time SNESSolve is called unless SNESSetCountersReset() is used.
1306 
1307 .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
1308 
1309 .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), SNESSetCountersReset()
1310 @*/
1311 PetscErrorCode  SNESGetNumberFunctionEvals(SNES snes, PetscInt *nfuncs)
1312 {
1313   PetscFunctionBegin;
1314   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1315   PetscValidIntPointer(nfuncs,2);
1316   *nfuncs = snes->nfuncs;
1317   PetscFunctionReturn(0);
1318 }
1319 
1320 /*@
1321    SNESGetLinearSolveFailures - Gets the number of failed (non-converged)
1322    linear solvers.
1323 
1324    Not Collective
1325 
1326    Input Parameter:
1327 .  snes - SNES context
1328 
1329    Output Parameter:
1330 .  nfails - number of failed solves
1331 
1332    Level: intermediate
1333 
1334    Options Database Keys:
1335 . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated
1336 
1337    Notes:
1338    This counter is reset to zero for each successive call to SNESSolve().
1339 
1340 .keywords: SNES, nonlinear, get, number, unsuccessful, steps
1341 
1342 .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures()
1343 @*/
1344 PetscErrorCode  SNESGetLinearSolveFailures(SNES snes,PetscInt *nfails)
1345 {
1346   PetscFunctionBegin;
1347   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1348   PetscValidIntPointer(nfails,2);
1349   *nfails = snes->numLinearSolveFailures;
1350   PetscFunctionReturn(0);
1351 }
1352 
1353 /*@
1354    SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts
1355    allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE
1356 
1357    Logically Collective on SNES
1358 
1359    Input Parameters:
1360 +  snes     - SNES context
1361 -  maxFails - maximum allowed linear solve failures
1362 
1363    Level: intermediate
1364 
1365    Options Database Keys:
1366 . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated
1367 
1368    Notes: By default this is 0; that is SNES returns on the first failed linear solve
1369 
1370 .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
1371 
1372 .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations()
1373 @*/
1374 PetscErrorCode  SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails)
1375 {
1376   PetscFunctionBegin;
1377   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1378   PetscValidLogicalCollectiveInt(snes,maxFails,2);
1379   snes->maxLinearSolveFailures = maxFails;
1380   PetscFunctionReturn(0);
1381 }
1382 
1383 /*@
1384    SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that
1385      are allowed before SNES terminates
1386 
1387    Not Collective
1388 
1389    Input Parameter:
1390 .  snes     - SNES context
1391 
1392    Output Parameter:
1393 .  maxFails - maximum of unsuccessful solves allowed
1394 
1395    Level: intermediate
1396 
1397    Notes: By default this is 1; that is SNES returns on the first failed linear solve
1398 
1399 .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
1400 
1401 .seealso: SNESGetLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(),
1402 @*/
1403 PetscErrorCode  SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails)
1404 {
1405   PetscFunctionBegin;
1406   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1407   PetscValidIntPointer(maxFails,2);
1408   *maxFails = snes->maxLinearSolveFailures;
1409   PetscFunctionReturn(0);
1410 }
1411 
1412 /*@
1413    SNESGetLinearSolveIterations - Gets the total number of linear iterations
1414    used by the nonlinear solver.
1415 
1416    Not Collective
1417 
1418    Input Parameter:
1419 .  snes - SNES context
1420 
1421    Output Parameter:
1422 .  lits - number of linear iterations
1423 
1424    Notes:
1425    This counter is reset to zero for each successive call to SNESSolve() unless SNESSetCountersReset() is used.
1426 
1427    If the linear solver fails inside the SNESSolve() the iterations for that call to the linear solver are not included. If you wish to count them
1428    then call KSPGetIterationNumber() after the failed solve.
1429 
1430    Level: intermediate
1431 
1432 .keywords: SNES, nonlinear, get, number, linear, iterations
1433 
1434 .seealso:  SNESGetIterationNumber(), SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESSetCountersReset()
1435 @*/
1436 PetscErrorCode  SNESGetLinearSolveIterations(SNES snes,PetscInt *lits)
1437 {
1438   PetscFunctionBegin;
1439   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1440   PetscValidIntPointer(lits,2);
1441   *lits = snes->linear_its;
1442   PetscFunctionReturn(0);
1443 }
1444 
1445 /*@
1446    SNESSetCountersReset - Sets whether or not the counters for linear iterations and function evaluations
1447    are reset every time SNESSolve() is called.
1448 
1449    Logically Collective on SNES
1450 
1451    Input Parameter:
1452 +  snes - SNES context
1453 -  reset - whether to reset the counters or not
1454 
1455    Notes:
1456    This defaults to PETSC_TRUE
1457 
1458    Level: developer
1459 
1460 .keywords: SNES, nonlinear, set, reset, number, linear, iterations
1461 
1462 .seealso:  SNESGetNumberFunctionEvals(), SNESGetLinearSolveIterations(), SNESGetNPC()
1463 @*/
1464 PetscErrorCode  SNESSetCountersReset(SNES snes,PetscBool reset)
1465 {
1466   PetscFunctionBegin;
1467   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1468   PetscValidLogicalCollectiveBool(snes,reset,2);
1469   snes->counters_reset = reset;
1470   PetscFunctionReturn(0);
1471 }
1472 
1473 
1474 /*@
1475    SNESSetKSP - Sets a KSP context for the SNES object to use
1476 
1477    Not Collective, but the SNES and KSP objects must live on the same MPI_Comm
1478 
1479    Input Parameters:
1480 +  snes - the SNES context
1481 -  ksp - the KSP context
1482 
1483    Notes:
1484    The SNES object already has its KSP object, you can obtain with SNESGetKSP()
1485    so this routine is rarely needed.
1486 
1487    The KSP object that is already in the SNES object has its reference count
1488    decreased by one.
1489 
1490    Level: developer
1491 
1492 .keywords: SNES, nonlinear, get, KSP, context
1493 
1494 .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
1495 @*/
1496 PetscErrorCode  SNESSetKSP(SNES snes,KSP ksp)
1497 {
1498   PetscErrorCode ierr;
1499 
1500   PetscFunctionBegin;
1501   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1502   PetscValidHeaderSpecific(ksp,KSP_CLASSID,2);
1503   PetscCheckSameComm(snes,1,ksp,2);
1504   ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr);
1505   if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);}
1506   snes->ksp = ksp;
1507   PetscFunctionReturn(0);
1508 }
1509 
1510 /* -----------------------------------------------------------*/
1511 /*@
1512    SNESCreate - Creates a nonlinear solver context.
1513 
1514    Collective on MPI_Comm
1515 
1516    Input Parameters:
1517 .  comm - MPI communicator
1518 
1519    Output Parameter:
1520 .  outsnes - the new SNES context
1521 
1522    Options Database Keys:
1523 +   -snes_mf - Activates default matrix-free Jacobian-vector products,
1524                and no preconditioning matrix
1525 .   -snes_mf_operator - Activates default matrix-free Jacobian-vector
1526                products, and a user-provided preconditioning matrix
1527                as set by SNESSetJacobian()
1528 -   -snes_fd - Uses (slow!) finite differences to compute Jacobian
1529 
1530    Level: beginner
1531 
1532    Developer Notes: SNES always creates a KSP object even though many SNES methods do not use it. This is
1533                     unfortunate and should be fixed at some point. The flag snes->usesksp indicates if the
1534                     particular method does use KSP and regulates if the information about the KSP is printed
1535                     in SNESView(). TSSetFromOptions() does call SNESSetFromOptions() which can lead to users being confused
1536                     by help messages about meaningless SNES options.
1537 
1538                     SNES always creates the snes->kspconvctx even though it is used by only one type. This should
1539                     be fixed.
1540 
1541 .keywords: SNES, nonlinear, create, context
1542 
1543 .seealso: SNESSolve(), SNESDestroy(), SNES, SNESSetLagPreconditioner()
1544 
1545 @*/
1546 PetscErrorCode  SNESCreate(MPI_Comm comm,SNES *outsnes)
1547 {
1548   PetscErrorCode ierr;
1549   SNES           snes;
1550   SNESKSPEW      *kctx;
1551 
1552   PetscFunctionBegin;
1553   PetscValidPointer(outsnes,2);
1554   *outsnes = NULL;
1555   ierr = SNESInitializePackage();CHKERRQ(ierr);
1556 
1557   ierr = PetscHeaderCreate(snes,SNES_CLASSID,"SNES","Nonlinear solver","SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr);
1558 
1559   snes->ops->converged    = SNESConvergedDefault;
1560   snes->usesksp           = PETSC_TRUE;
1561   snes->tolerancesset     = PETSC_FALSE;
1562   snes->max_its           = 50;
1563   snes->max_funcs         = 10000;
1564   snes->norm              = 0.0;
1565   snes->normschedule      = SNES_NORM_ALWAYS;
1566   snes->functype          = SNES_FUNCTION_DEFAULT;
1567 #if defined(PETSC_USE_REAL_SINGLE)
1568   snes->rtol              = 1.e-5;
1569 #else
1570   snes->rtol              = 1.e-8;
1571 #endif
1572   snes->ttol              = 0.0;
1573 #if defined(PETSC_USE_REAL_SINGLE)
1574   snes->abstol            = 1.e-25;
1575 #else
1576   snes->abstol            = 1.e-50;
1577 #endif
1578 #if defined(PETSC_USE_REAL_SINGLE)
1579   snes->stol              = 1.e-5;
1580 #else
1581   snes->stol              = 1.e-8;
1582 #endif
1583 #if defined(PETSC_USE_REAL_SINGLE)
1584   snes->deltatol          = 1.e-6;
1585 #else
1586   snes->deltatol          = 1.e-12;
1587 #endif
1588   snes->divtol            = 1.e4;
1589   snes->rnorm0            = 0;
1590   snes->nfuncs            = 0;
1591   snes->numFailures       = 0;
1592   snes->maxFailures       = 1;
1593   snes->linear_its        = 0;
1594   snes->lagjacobian       = 1;
1595   snes->jac_iter          = 0;
1596   snes->lagjac_persist    = PETSC_FALSE;
1597   snes->lagpreconditioner = 1;
1598   snes->pre_iter          = 0;
1599   snes->lagpre_persist    = PETSC_FALSE;
1600   snes->numbermonitors    = 0;
1601   snes->data              = 0;
1602   snes->setupcalled       = PETSC_FALSE;
1603   snes->ksp_ewconv        = PETSC_FALSE;
1604   snes->nwork             = 0;
1605   snes->work              = 0;
1606   snes->nvwork            = 0;
1607   snes->vwork             = 0;
1608   snes->conv_hist_len     = 0;
1609   snes->conv_hist_max     = 0;
1610   snes->conv_hist         = NULL;
1611   snes->conv_hist_its     = NULL;
1612   snes->conv_hist_reset   = PETSC_TRUE;
1613   snes->counters_reset    = PETSC_TRUE;
1614   snes->vec_func_init_set = PETSC_FALSE;
1615   snes->reason            = SNES_CONVERGED_ITERATING;
1616   snes->npcside           = PC_RIGHT;
1617 
1618   snes->mf          = PETSC_FALSE;
1619   snes->mf_operator = PETSC_FALSE;
1620   snes->mf_version  = 1;
1621 
1622   snes->numLinearSolveFailures = 0;
1623   snes->maxLinearSolveFailures = 1;
1624 
1625   snes->vizerotolerance = 1.e-8;
1626 
1627   /* Set this to true if the implementation of SNESSolve_XXX does compute the residual at the final solution. */
1628   snes->alwayscomputesfinalresidual = PETSC_FALSE;
1629 
1630   /* Create context to compute Eisenstat-Walker relative tolerance for KSP */
1631   ierr = PetscNewLog(snes,&kctx);CHKERRQ(ierr);
1632 
1633   snes->kspconvctx  = (void*)kctx;
1634   kctx->version     = 2;
1635   kctx->rtol_0      = .3; /* Eisenstat and Walker suggest rtol_0=.5, but
1636                              this was too large for some test cases */
1637   kctx->rtol_last   = 0.0;
1638   kctx->rtol_max    = .9;
1639   kctx->gamma       = 1.0;
1640   kctx->alpha       = .5*(1.0 + PetscSqrtReal(5.0));
1641   kctx->alpha2      = kctx->alpha;
1642   kctx->threshold   = .1;
1643   kctx->lresid_last = 0.0;
1644   kctx->norm_last   = 0.0;
1645 
1646   *outsnes = snes;
1647   PetscFunctionReturn(0);
1648 }
1649 
1650 /*MC
1651     SNESFunction - Functional form used to convey the nonlinear function to be solved by SNES
1652 
1653      Synopsis:
1654      #include "petscsnes.h"
1655      PetscErrorCode SNESFunction(SNES snes,Vec x,Vec f,void *ctx);
1656 
1657      Input Parameters:
1658 +     snes - the SNES context
1659 .     x    - state at which to evaluate residual
1660 -     ctx     - optional user-defined function context, passed in with SNESSetFunction()
1661 
1662      Output Parameter:
1663 .     f  - vector to put residual (function value)
1664 
1665    Level: intermediate
1666 
1667 .seealso:   SNESSetFunction(), SNESGetFunction()
1668 M*/
1669 
1670 /*@C
1671    SNESSetFunction - Sets the function evaluation routine and function
1672    vector for use by the SNES routines in solving systems of nonlinear
1673    equations.
1674 
1675    Logically Collective on SNES
1676 
1677    Input Parameters:
1678 +  snes - the SNES context
1679 .  r - vector to store function value
1680 .  f - function evaluation routine; see SNESFunction for calling sequence details
1681 -  ctx - [optional] user-defined context for private data for the
1682          function evaluation routine (may be NULL)
1683 
1684    Notes:
1685    The Newton-like methods typically solve linear systems of the form
1686 $      f'(x) x = -f(x),
1687    where f'(x) denotes the Jacobian matrix and f(x) is the function.
1688 
1689    Level: beginner
1690 
1691 .keywords: SNES, nonlinear, set, function
1692 
1693 .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetPicard(), SNESFunction
1694 @*/
1695 PetscErrorCode  SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx)
1696 {
1697   PetscErrorCode ierr;
1698   DM             dm;
1699 
1700   PetscFunctionBegin;
1701   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1702   if (r) {
1703     PetscValidHeaderSpecific(r,VEC_CLASSID,2);
1704     PetscCheckSameComm(snes,1,r,2);
1705     ierr = PetscObjectReference((PetscObject)r);CHKERRQ(ierr);
1706     ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr);
1707 
1708     snes->vec_func = r;
1709   }
1710   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
1711   ierr = DMSNESSetFunction(dm,f,ctx);CHKERRQ(ierr);
1712   PetscFunctionReturn(0);
1713 }
1714 
1715 
1716 /*@C
1717    SNESSetInitialFunction - Sets the function vector to be used as the
1718    function norm at the initialization of the method.  In some
1719    instances, the user has precomputed the function before calling
1720    SNESSolve.  This function allows one to avoid a redundant call
1721    to SNESComputeFunction in that case.
1722 
1723    Logically Collective on SNES
1724 
1725    Input Parameters:
1726 +  snes - the SNES context
1727 -  f - vector to store function value
1728 
1729    Notes:
1730    This should not be modified during the solution procedure.
1731 
1732    This is used extensively in the SNESFAS hierarchy and in nonlinear preconditioning.
1733 
1734    Level: developer
1735 
1736 .keywords: SNES, nonlinear, set, function
1737 
1738 .seealso: SNESSetFunction(), SNESComputeFunction(), SNESSetInitialFunctionNorm()
1739 @*/
1740 PetscErrorCode  SNESSetInitialFunction(SNES snes, Vec f)
1741 {
1742   PetscErrorCode ierr;
1743   Vec            vec_func;
1744 
1745   PetscFunctionBegin;
1746   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1747   PetscValidHeaderSpecific(f,VEC_CLASSID,2);
1748   PetscCheckSameComm(snes,1,f,2);
1749   if (snes->npcside== PC_LEFT && snes->functype == SNES_FUNCTION_PRECONDITIONED) {
1750     snes->vec_func_init_set = PETSC_FALSE;
1751     PetscFunctionReturn(0);
1752   }
1753   ierr = SNESGetFunction(snes,&vec_func,NULL,NULL);CHKERRQ(ierr);
1754   ierr = VecCopy(f, vec_func);CHKERRQ(ierr);
1755 
1756   snes->vec_func_init_set = PETSC_TRUE;
1757   PetscFunctionReturn(0);
1758 }
1759 
1760 /*@
1761    SNESSetNormSchedule - Sets the SNESNormSchedule used in covergence and monitoring
1762    of the SNES method.
1763 
1764    Logically Collective on SNES
1765 
1766    Input Parameters:
1767 +  snes - the SNES context
1768 -  normschedule - the frequency of norm computation
1769 
1770    Options Database Key:
1771 .  -snes_norm_schedule <none, always, initialonly, finalonly, initalfinalonly>
1772 
1773    Notes:
1774    Only certain SNES methods support certain SNESNormSchedules.  Most require evaluation
1775    of the nonlinear function and the taking of its norm at every iteration to
1776    even ensure convergence at all.  However, methods such as custom Gauss-Seidel methods
1777    (SNESNGS) and the like do not require the norm of the function to be computed, and therfore
1778    may either be monitored for convergence or not.  As these are often used as nonlinear
1779    preconditioners, monitoring the norm of their error is not a useful enterprise within
1780    their solution.
1781 
1782    Level: developer
1783 
1784 .keywords: SNES, nonlinear, set, function, norm, type
1785 
1786 .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1787 @*/
1788 PetscErrorCode  SNESSetNormSchedule(SNES snes, SNESNormSchedule normschedule)
1789 {
1790   PetscFunctionBegin;
1791   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1792   snes->normschedule = normschedule;
1793   PetscFunctionReturn(0);
1794 }
1795 
1796 
1797 /*@
1798    SNESGetNormSchedule - Gets the SNESNormSchedule used in covergence and monitoring
1799    of the SNES method.
1800 
1801    Logically Collective on SNES
1802 
1803    Input Parameters:
1804 +  snes - the SNES context
1805 -  normschedule - the type of the norm used
1806 
1807    Level: advanced
1808 
1809 .keywords: SNES, nonlinear, set, function, norm, type
1810 
1811 .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1812 @*/
1813 PetscErrorCode  SNESGetNormSchedule(SNES snes, SNESNormSchedule *normschedule)
1814 {
1815   PetscFunctionBegin;
1816   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1817   *normschedule = snes->normschedule;
1818   PetscFunctionReturn(0);
1819 }
1820 
1821 
1822 /*@
1823   SNESSetFunctionNorm - Sets the last computed residual norm.
1824 
1825   Logically Collective on SNES
1826 
1827   Input Parameters:
1828 + snes - the SNES context
1829 
1830 - normschedule - the frequency of norm computation
1831 
1832   Level: developer
1833 
1834 .keywords: SNES, nonlinear, set, function, norm, type
1835 .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1836 @*/
1837 PetscErrorCode SNESSetFunctionNorm(SNES snes, PetscReal norm)
1838 {
1839   PetscFunctionBegin;
1840   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1841   snes->norm = norm;
1842   PetscFunctionReturn(0);
1843 }
1844 
1845 /*@
1846   SNESGetFunctionNorm - Gets the last computed norm of the residual
1847 
1848   Not Collective
1849 
1850   Input Parameter:
1851 . snes - the SNES context
1852 
1853   Output Parameter:
1854 . norm - the last computed residual norm
1855 
1856   Level: developer
1857 
1858 .keywords: SNES, nonlinear, set, function, norm, type
1859 .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1860 @*/
1861 PetscErrorCode SNESGetFunctionNorm(SNES snes, PetscReal *norm)
1862 {
1863   PetscFunctionBegin;
1864   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1865   PetscValidPointer(norm, 2);
1866   *norm = snes->norm;
1867   PetscFunctionReturn(0);
1868 }
1869 
1870 /*@C
1871    SNESSetFunctionType - Sets the SNESNormSchedule used in covergence and monitoring
1872    of the SNES method.
1873 
1874    Logically Collective on SNES
1875 
1876    Input Parameters:
1877 +  snes - the SNES context
1878 -  normschedule - the frequency of norm computation
1879 
1880    Notes:
1881    Only certain SNES methods support certain SNESNormSchedules.  Most require evaluation
1882    of the nonlinear function and the taking of its norm at every iteration to
1883    even ensure convergence at all.  However, methods such as custom Gauss-Seidel methods
1884    (SNESNGS) and the like do not require the norm of the function to be computed, and therfore
1885    may either be monitored for convergence or not.  As these are often used as nonlinear
1886    preconditioners, monitoring the norm of their error is not a useful enterprise within
1887    their solution.
1888 
1889    Level: developer
1890 
1891 .keywords: SNES, nonlinear, set, function, norm, type
1892 
1893 .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1894 @*/
1895 PetscErrorCode  SNESSetFunctionType(SNES snes, SNESFunctionType type)
1896 {
1897   PetscFunctionBegin;
1898   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1899   snes->functype = type;
1900   PetscFunctionReturn(0);
1901 }
1902 
1903 
1904 /*@C
1905    SNESGetFunctionType - Gets the SNESNormSchedule used in covergence and monitoring
1906    of the SNES method.
1907 
1908    Logically Collective on SNES
1909 
1910    Input Parameters:
1911 +  snes - the SNES context
1912 -  normschedule - the type of the norm used
1913 
1914    Level: advanced
1915 
1916 .keywords: SNES, nonlinear, set, function, norm, type
1917 
1918 .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1919 @*/
1920 PetscErrorCode  SNESGetFunctionType(SNES snes, SNESFunctionType *type)
1921 {
1922   PetscFunctionBegin;
1923   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1924   *type = snes->functype;
1925   PetscFunctionReturn(0);
1926 }
1927 
1928 /*MC
1929     SNESNGSFunction - function used to convey a Gauss-Seidel sweep on the nonlinear function
1930 
1931      Synopsis:
1932      #include <petscsnes.h>
1933 $    SNESNGSFunction(SNES snes,Vec x,Vec b,void *ctx);
1934 
1935 +  X   - solution vector
1936 .  B   - RHS vector
1937 -  ctx - optional user-defined Gauss-Seidel context
1938 
1939    Level: intermediate
1940 
1941 .seealso:   SNESSetNGS(), SNESGetNGS()
1942 M*/
1943 
1944 /*@C
1945    SNESSetNGS - Sets the user nonlinear Gauss-Seidel routine for
1946    use with composed nonlinear solvers.
1947 
1948    Input Parameters:
1949 +  snes   - the SNES context
1950 .  f - function evaluation routine to apply Gauss-Seidel see SNESNGSFunction
1951 -  ctx    - [optional] user-defined context for private data for the
1952             smoother evaluation routine (may be NULL)
1953 
1954    Notes:
1955    The NGS routines are used by the composed nonlinear solver to generate
1956     a problem appropriate update to the solution, particularly FAS.
1957 
1958    Level: intermediate
1959 
1960 .keywords: SNES, nonlinear, set, Gauss-Seidel
1961 
1962 .seealso: SNESGetFunction(), SNESComputeNGS()
1963 @*/
1964 PetscErrorCode SNESSetNGS(SNES snes,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx)
1965 {
1966   PetscErrorCode ierr;
1967   DM             dm;
1968 
1969   PetscFunctionBegin;
1970   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1971   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
1972   ierr = DMSNESSetNGS(dm,f,ctx);CHKERRQ(ierr);
1973   PetscFunctionReturn(0);
1974 }
1975 
1976 PETSC_EXTERN PetscErrorCode SNESPicardComputeFunction(SNES snes,Vec x,Vec f,void *ctx)
1977 {
1978   PetscErrorCode ierr;
1979   DM             dm;
1980   DMSNES         sdm;
1981 
1982   PetscFunctionBegin;
1983   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
1984   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
1985   /*  A(x)*x - b(x) */
1986   if (sdm->ops->computepfunction) {
1987     ierr = (*sdm->ops->computepfunction)(snes,x,f,sdm->pctx);CHKERRQ(ierr);
1988   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard function.");
1989 
1990   if (sdm->ops->computepjacobian) {
1991     ierr = (*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx);CHKERRQ(ierr);
1992   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard matrix.");
1993   ierr = VecScale(f,-1.0);CHKERRQ(ierr);
1994   ierr = MatMultAdd(snes->jacobian,x,f,f);CHKERRQ(ierr);
1995   PetscFunctionReturn(0);
1996 }
1997 
1998 PETSC_EXTERN PetscErrorCode SNESPicardComputeJacobian(SNES snes,Vec x1,Mat J,Mat B,void *ctx)
1999 {
2000   PetscFunctionBegin;
2001   /* the jacobian matrix should be pre-filled in SNESPicardComputeFunction */
2002   PetscFunctionReturn(0);
2003 }
2004 
2005 /*@C
2006    SNESSetPicard - Use SNES to solve the semilinear-system A(x) x = b(x) via a Picard type iteration (Picard linearization)
2007 
2008    Logically Collective on SNES
2009 
2010    Input Parameters:
2011 +  snes - the SNES context
2012 .  r - vector to store function value
2013 .  b - function evaluation routine
2014 .  Amat - matrix with which A(x) x - b(x) is to be computed
2015 .  Pmat - matrix from which preconditioner is computed (usually the same as Amat)
2016 .  J  - function to compute matrix value, see SNESJacobianFunction for details on its calling sequence
2017 -  ctx - [optional] user-defined context for private data for the
2018          function evaluation routine (may be NULL)
2019 
2020    Notes:
2021     We do not recomemend using this routine. It is far better to provide the nonlinear function F() and some approximation to the Jacobian and use
2022     an approximate Newton solver. This interface is provided to allow porting/testing a previous Picard based code in PETSc before converting it to approximate Newton.
2023 
2024     One can call SNESSetPicard() or SNESSetFunction() (and possibly SNESSetJacobian()) but cannot call both
2025 
2026 $     Solves the equation A(x) x = b(x) via the defect correction algorithm A(x^{n}) (x^{n+1} - x^{n}) = b(x^{n}) - A(x^{n})x^{n}
2027 $     Note that when an exact solver is used this corresponds to the "classic" Picard A(x^{n}) x^{n+1} = b(x^{n}) iteration.
2028 
2029      Run with -snes_mf_operator to solve the system with Newton's method using A(x^{n}) to construct the preconditioner.
2030 
2031    We implement the defect correction form of the Picard iteration because it converges much more generally when inexact linear solvers are used then
2032    the direct Picard iteration A(x^n) x^{n+1} = b(x^n)
2033 
2034    There is some controversity over the definition of a Picard iteration for nonlinear systems but almost everyone agrees that it involves a linear solve and some
2035    believe it is the iteration  A(x^{n}) x^{n+1} = b(x^{n}) hence we use the name Picard. If anyone has an authoritative  reference that defines the Picard iteration
2036    different please contact us at petsc-dev@mcs.anl.gov and we'll have an entirely new argument :-).
2037 
2038    Level: intermediate
2039 
2040 .keywords: SNES, nonlinear, set, function
2041 
2042 .seealso: SNESGetFunction(), SNESSetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESGetPicard(), SNESLineSearchPreCheckPicard(), SNESJacobianFunction
2043 @*/
2044 PetscErrorCode  SNESSetPicard(SNES snes,Vec r,PetscErrorCode (*b)(SNES,Vec,Vec,void*),Mat Amat, Mat Pmat, PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx)
2045 {
2046   PetscErrorCode ierr;
2047   DM             dm;
2048 
2049   PetscFunctionBegin;
2050   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2051   ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
2052   ierr = DMSNESSetPicard(dm,b,J,ctx);CHKERRQ(ierr);
2053   ierr = SNESSetFunction(snes,r,SNESPicardComputeFunction,ctx);CHKERRQ(ierr);
2054   ierr = SNESSetJacobian(snes,Amat,Pmat,SNESPicardComputeJacobian,ctx);CHKERRQ(ierr);
2055   PetscFunctionReturn(0);
2056 }
2057 
2058 /*@C
2059    SNESGetPicard - Returns the context for the Picard iteration
2060 
2061    Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet.
2062 
2063    Input Parameter:
2064 .  snes - the SNES context
2065 
2066    Output Parameter:
2067 +  r - the function (or NULL)
2068 .  f - the function (or NULL); see SNESFunction for calling sequence details
2069 .  Amat - the matrix used to defined the operation A(x) x - b(x) (or NULL)
2070 .  Pmat  - the matrix from which the preconditioner will be constructed (or NULL)
2071 .  J - the function for matrix evaluation (or NULL); see SNESJacobianFunction for calling sequence details
2072 -  ctx - the function context (or NULL)
2073 
2074    Level: advanced
2075 
2076 .keywords: SNES, nonlinear, get, function
2077 
2078 .seealso: SNESSetPicard(), SNESGetFunction(), SNESGetJacobian(), SNESGetDM(), SNESFunction, SNESJacobianFunction
2079 @*/
2080 PetscErrorCode  SNESGetPicard(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),Mat *Amat, Mat *Pmat, PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx)
2081 {
2082   PetscErrorCode ierr;
2083   DM             dm;
2084 
2085   PetscFunctionBegin;
2086   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2087   ierr = SNESGetFunction(snes,r,NULL,NULL);CHKERRQ(ierr);
2088   ierr = SNESGetJacobian(snes,Amat,Pmat,NULL,NULL);CHKERRQ(ierr);
2089   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2090   ierr = DMSNESGetPicard(dm,f,J,ctx);CHKERRQ(ierr);
2091   PetscFunctionReturn(0);
2092 }
2093 
2094 /*@C
2095    SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem
2096 
2097    Logically Collective on SNES
2098 
2099    Input Parameters:
2100 +  snes - the SNES context
2101 .  func - function evaluation routine
2102 -  ctx - [optional] user-defined context for private data for the
2103          function evaluation routine (may be NULL)
2104 
2105    Calling sequence of func:
2106 $    func (SNES snes,Vec x,void *ctx);
2107 
2108 .  f - function vector
2109 -  ctx - optional user-defined function context
2110 
2111    Level: intermediate
2112 
2113 .keywords: SNES, nonlinear, set, function
2114 
2115 .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian()
2116 @*/
2117 PetscErrorCode  SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx)
2118 {
2119   PetscFunctionBegin;
2120   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2121   if (func) snes->ops->computeinitialguess = func;
2122   if (ctx)  snes->initialguessP            = ctx;
2123   PetscFunctionReturn(0);
2124 }
2125 
2126 /* --------------------------------------------------------------- */
2127 /*@C
2128    SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set
2129    it assumes a zero right hand side.
2130 
2131    Logically Collective on SNES
2132 
2133    Input Parameter:
2134 .  snes - the SNES context
2135 
2136    Output Parameter:
2137 .  rhs - the right hand side vector or NULL if the right hand side vector is null
2138 
2139    Level: intermediate
2140 
2141 .keywords: SNES, nonlinear, get, function, right hand side
2142 
2143 .seealso: SNESGetSolution(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
2144 @*/
2145 PetscErrorCode  SNESGetRhs(SNES snes,Vec *rhs)
2146 {
2147   PetscFunctionBegin;
2148   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2149   PetscValidPointer(rhs,2);
2150   *rhs = snes->vec_rhs;
2151   PetscFunctionReturn(0);
2152 }
2153 
2154 /*@
2155    SNESComputeFunction - Calls the function that has been set with SNESSetFunction().
2156 
2157    Collective on SNES
2158 
2159    Input Parameters:
2160 +  snes - the SNES context
2161 -  x - input vector
2162 
2163    Output Parameter:
2164 .  y - function vector, as set by SNESSetFunction()
2165 
2166    Notes:
2167    SNESComputeFunction() is typically used within nonlinear solvers
2168    implementations, so most users would not generally call this routine
2169    themselves.
2170 
2171    Level: developer
2172 
2173 .keywords: SNES, nonlinear, compute, function
2174 
2175 .seealso: SNESSetFunction(), SNESGetFunction()
2176 @*/
2177 PetscErrorCode  SNESComputeFunction(SNES snes,Vec x,Vec y)
2178 {
2179   PetscErrorCode ierr;
2180   DM             dm;
2181   DMSNES         sdm;
2182 
2183   PetscFunctionBegin;
2184   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2185   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2186   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2187   PetscCheckSameComm(snes,1,x,2);
2188   PetscCheckSameComm(snes,1,y,3);
2189   ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr);
2190 
2191   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2192   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
2193   if (sdm->ops->computefunction) {
2194     if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) {
2195       ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
2196     }
2197     ierr = VecLockPush(x);CHKERRQ(ierr);
2198     PetscStackPush("SNES user function");
2199     ierr = (*sdm->ops->computefunction)(snes,x,y,sdm->functionctx);CHKERRQ(ierr);
2200     PetscStackPop;
2201     ierr = VecLockPop(x);CHKERRQ(ierr);
2202     if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) {
2203       ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
2204     }
2205   } else if (snes->vec_rhs) {
2206     ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr);
2207   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve().");
2208   if (snes->vec_rhs) {
2209     ierr = VecAXPY(y,-1.0,snes->vec_rhs);CHKERRQ(ierr);
2210   }
2211   snes->nfuncs++;
2212   /*
2213      domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will
2214      propagate the value to all processes
2215   */
2216   if (snes->domainerror) {
2217     ierr = VecSetInf(y);CHKERRQ(ierr);
2218   }
2219   PetscFunctionReturn(0);
2220 }
2221 
2222 /*@
2223    SNESComputeNGS - Calls the Gauss-Seidel function that has been set with  SNESSetNGS().
2224 
2225    Collective on SNES
2226 
2227    Input Parameters:
2228 +  snes - the SNES context
2229 .  x - input vector
2230 -  b - rhs vector
2231 
2232    Output Parameter:
2233 .  x - new solution vector
2234 
2235    Notes:
2236    SNESComputeNGS() is typically used within composed nonlinear solver
2237    implementations, so most users would not generally call this routine
2238    themselves.
2239 
2240    Level: developer
2241 
2242 .keywords: SNES, nonlinear, compute, function
2243 
2244 .seealso: SNESSetNGS(), SNESComputeFunction()
2245 @*/
2246 PetscErrorCode  SNESComputeNGS(SNES snes,Vec b,Vec x)
2247 {
2248   PetscErrorCode ierr;
2249   DM             dm;
2250   DMSNES         sdm;
2251 
2252   PetscFunctionBegin;
2253   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2254   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2255   if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,3);
2256   PetscCheckSameComm(snes,1,x,2);
2257   if (b) PetscCheckSameComm(snes,1,b,3);
2258   if (b) {ierr = VecValidValues(b,2,PETSC_TRUE);CHKERRQ(ierr);}
2259   ierr = PetscLogEventBegin(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr);
2260   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2261   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
2262   if (sdm->ops->computegs) {
2263     if (b) {ierr = VecLockPush(b);CHKERRQ(ierr);}
2264     PetscStackPush("SNES user NGS");
2265     ierr = (*sdm->ops->computegs)(snes,x,b,sdm->gsctx);CHKERRQ(ierr);
2266     PetscStackPop;
2267     if (b) {ierr = VecLockPop(b);CHKERRQ(ierr);}
2268   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetNGS() before SNESComputeNGS(), likely called from SNESSolve().");
2269   ierr = PetscLogEventEnd(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr);
2270   PetscFunctionReturn(0);
2271 }
2272 
2273 /*@
2274    SNESComputeJacobian - Computes the Jacobian matrix that has been set with SNESSetJacobian().
2275 
2276    Collective on SNES and Mat
2277 
2278    Input Parameters:
2279 +  snes - the SNES context
2280 -  x - input vector
2281 
2282    Output Parameters:
2283 +  A - Jacobian matrix
2284 -  B - optional preconditioning matrix
2285 
2286   Options Database Keys:
2287 +    -snes_lag_preconditioner <lag>
2288 .    -snes_lag_jacobian <lag>
2289 .    -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences
2290 .    -snes_compare_explicit_draw  - Compare the computed Jacobian to the finite difference Jacobian and draw the result
2291 .    -snes_compare_explicit_contour  - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result
2292 .    -snes_compare_operator  - Make the comparison options above use the operator instead of the preconditioning matrix
2293 .    -snes_compare_coloring - Compute the finite difference Jacobian using coloring and display norms of difference
2294 .    -snes_compare_coloring_display - Compute the finite differece Jacobian using coloring and display verbose differences
2295 .    -snes_compare_coloring_threshold - Display only those matrix entries that differ by more than a given threshold
2296 .    -snes_compare_coloring_threshold_atol - Absolute tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold
2297 .    -snes_compare_coloring_threshold_rtol - Relative tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold
2298 .    -snes_compare_coloring_draw - Compute the finite differece Jacobian using coloring and draw differences
2299 -    -snes_compare_coloring_draw_contour - Compute the finite differece Jacobian using coloring and show contours of matrices and differences
2300 
2301 
2302    Notes:
2303    Most users should not need to explicitly call this routine, as it
2304    is used internally within the nonlinear solvers.
2305 
2306    Level: developer
2307 
2308 .keywords: SNES, compute, Jacobian, matrix
2309 
2310 .seealso:  SNESSetJacobian(), KSPSetOperators(), MatStructure, SNESSetLagPreconditioner(), SNESSetLagJacobian()
2311 @*/
2312 PetscErrorCode  SNESComputeJacobian(SNES snes,Vec X,Mat A,Mat B)
2313 {
2314   PetscErrorCode ierr;
2315   PetscBool      flag;
2316   DM             dm;
2317   DMSNES         sdm;
2318   KSP            ksp;
2319 
2320   PetscFunctionBegin;
2321   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2322   PetscValidHeaderSpecific(X,VEC_CLASSID,2);
2323   PetscCheckSameComm(snes,1,X,2);
2324   ierr = VecValidValues(X,2,PETSC_TRUE);CHKERRQ(ierr);
2325   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2326   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
2327 
2328   if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Must call SNESSetJacobian(), DMSNESSetJacobian(), DMDASNESSetJacobianLocal(), etc");
2329 
2330   /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */
2331 
2332   if (snes->lagjacobian == -2) {
2333     snes->lagjacobian = -1;
2334 
2335     ierr = PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n");CHKERRQ(ierr);
2336   } else if (snes->lagjacobian == -1) {
2337     ierr = PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n");CHKERRQ(ierr);
2338     ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr);
2339     if (flag) {
2340       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2341       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2342     }
2343     PetscFunctionReturn(0);
2344   } else if (snes->lagjacobian > 1 && (snes->iter + snes->jac_iter) % snes->lagjacobian) {
2345     ierr = PetscInfo2(snes,"Reusing Jacobian/preconditioner because lag is %D and SNES iteration is %D\n",snes->lagjacobian,snes->iter);CHKERRQ(ierr);
2346     ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr);
2347     if (flag) {
2348       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2349       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2350     }
2351     PetscFunctionReturn(0);
2352   }
2353   if (snes->npc && snes->npcside== PC_LEFT) {
2354       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2355       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2356       PetscFunctionReturn(0);
2357   }
2358 
2359   ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr);
2360   ierr = VecLockPush(X);CHKERRQ(ierr);
2361   PetscStackPush("SNES user Jacobian function");
2362   ierr = (*sdm->ops->computejacobian)(snes,X,A,B,sdm->jacobianctx);CHKERRQ(ierr);
2363   PetscStackPop;
2364   ierr = VecLockPop(X);CHKERRQ(ierr);
2365   ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr);
2366 
2367   /* the next line ensures that snes->ksp exists */
2368   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
2369   if (snes->lagpreconditioner == -2) {
2370     ierr = PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n");CHKERRQ(ierr);
2371     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr);
2372     snes->lagpreconditioner = -1;
2373   } else if (snes->lagpreconditioner == -1) {
2374     ierr = PetscInfo(snes,"Reusing preconditioner because lag is -1\n");CHKERRQ(ierr);
2375     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr);
2376   } else if (snes->lagpreconditioner > 1 && (snes->iter + snes->pre_iter) % snes->lagpreconditioner) {
2377     ierr = PetscInfo2(snes,"Reusing preconditioner because lag is %D and SNES iteration is %D\n",snes->lagpreconditioner,snes->iter);CHKERRQ(ierr);
2378     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr);
2379   } else {
2380     ierr = PetscInfo(snes,"Rebuilding preconditioner\n");CHKERRQ(ierr);
2381     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr);
2382   }
2383 
2384   /* make sure user returned a correct Jacobian and preconditioner */
2385   /* PetscValidHeaderSpecific(A,MAT_CLASSID,3);
2386     PetscValidHeaderSpecific(B,MAT_CLASSID,4);   */
2387   {
2388     PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE;
2389     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit",NULL,NULL,&flag);CHKERRQ(ierr);
2390     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr);
2391     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr);
2392     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_operator",NULL,NULL,&flag_operator);CHKERRQ(ierr);
2393     if (flag || flag_draw || flag_contour) {
2394       Mat          Bexp_mine = NULL,Bexp,FDexp;
2395       PetscViewer  vdraw,vstdout;
2396       PetscBool    flg;
2397       if (flag_operator) {
2398         ierr = MatComputeExplicitOperator(A,&Bexp_mine);CHKERRQ(ierr);
2399         Bexp = Bexp_mine;
2400       } else {
2401         /* See if the preconditioning matrix can be viewed and added directly */
2402         ierr = PetscObjectTypeCompareAny((PetscObject)B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr);
2403         if (flg) Bexp = B;
2404         else {
2405           /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */
2406           ierr = MatComputeExplicitOperator(B,&Bexp_mine);CHKERRQ(ierr);
2407           Bexp = Bexp_mine;
2408         }
2409       }
2410       ierr = MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp);CHKERRQ(ierr);
2411       ierr = SNESComputeJacobianDefault(snes,X,FDexp,FDexp,NULL);CHKERRQ(ierr);
2412       ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr);
2413       if (flag_draw || flag_contour) {
2414         ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
2415         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
2416       } else vdraw = NULL;
2417       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator ? "Jacobian" : "preconditioning Jacobian");CHKERRQ(ierr);
2418       if (flag) {ierr = MatView(Bexp,vstdout);CHKERRQ(ierr);}
2419       if (vdraw) {ierr = MatView(Bexp,vdraw);CHKERRQ(ierr);}
2420       ierr = PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n");CHKERRQ(ierr);
2421       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
2422       if (vdraw) {ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);}
2423       ierr = MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
2424       ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n");CHKERRQ(ierr);
2425       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
2426       if (vdraw) {              /* Always use contour for the difference */
2427         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
2428         ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);
2429         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
2430       }
2431       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
2432       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
2433       ierr = MatDestroy(&Bexp_mine);CHKERRQ(ierr);
2434       ierr = MatDestroy(&FDexp);CHKERRQ(ierr);
2435     }
2436   }
2437   {
2438     PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE;
2439     PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON;
2440     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring",NULL,NULL,&flag);CHKERRQ(ierr);
2441     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_display",NULL,NULL,&flag_display);CHKERRQ(ierr);
2442     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr);
2443     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr);
2444     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",NULL,NULL,&flag_threshold);CHKERRQ(ierr);
2445     if (flag_threshold) {
2446       ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,NULL);CHKERRQ(ierr);
2447       ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,NULL);CHKERRQ(ierr);
2448     }
2449     if (flag || flag_display || flag_draw || flag_contour || flag_threshold) {
2450       Mat            Bfd;
2451       PetscViewer    vdraw,vstdout;
2452       MatColoring    coloring;
2453       ISColoring     iscoloring;
2454       MatFDColoring  matfdcoloring;
2455       PetscErrorCode (*func)(SNES,Vec,Vec,void*);
2456       void           *funcctx;
2457       PetscReal      norm1,norm2,normmax;
2458 
2459       ierr = MatDuplicate(B,MAT_DO_NOT_COPY_VALUES,&Bfd);CHKERRQ(ierr);
2460       ierr = MatColoringCreate(Bfd,&coloring);CHKERRQ(ierr);
2461       ierr = MatColoringSetType(coloring,MATCOLORINGSL);CHKERRQ(ierr);
2462       ierr = MatColoringSetFromOptions(coloring);CHKERRQ(ierr);
2463       ierr = MatColoringApply(coloring,&iscoloring);CHKERRQ(ierr);
2464       ierr = MatColoringDestroy(&coloring);CHKERRQ(ierr);
2465       ierr = MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring);CHKERRQ(ierr);
2466       ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr);
2467       ierr = MatFDColoringSetUp(Bfd,iscoloring,matfdcoloring);CHKERRQ(ierr);
2468       ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr);
2469 
2470       /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */
2471       ierr = SNESGetFunction(snes,NULL,&func,&funcctx);CHKERRQ(ierr);
2472       ierr = MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode (*)(void))func,funcctx);CHKERRQ(ierr);
2473       ierr = PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix);CHKERRQ(ierr);
2474       ierr = PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_");CHKERRQ(ierr);
2475       ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr);
2476       ierr = MatFDColoringApply(Bfd,matfdcoloring,X,snes);CHKERRQ(ierr);
2477       ierr = MatFDColoringDestroy(&matfdcoloring);CHKERRQ(ierr);
2478 
2479       ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr);
2480       if (flag_draw || flag_contour) {
2481         ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
2482         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
2483       } else vdraw = NULL;
2484       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n");CHKERRQ(ierr);
2485       if (flag_display) {ierr = MatView(B,vstdout);CHKERRQ(ierr);}
2486       if (vdraw) {ierr = MatView(B,vdraw);CHKERRQ(ierr);}
2487       ierr = PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n");CHKERRQ(ierr);
2488       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
2489       if (vdraw) {ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);}
2490       ierr = MatAYPX(Bfd,-1.0,B,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
2491       ierr = MatNorm(Bfd,NORM_1,&norm1);CHKERRQ(ierr);
2492       ierr = MatNorm(Bfd,NORM_FROBENIUS,&norm2);CHKERRQ(ierr);
2493       ierr = MatNorm(Bfd,NORM_MAX,&normmax);CHKERRQ(ierr);
2494       ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian, norm1=%g normFrob=%g normmax=%g\n",(double)norm1,(double)norm2,(double)normmax);CHKERRQ(ierr);
2495       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
2496       if (vdraw) {              /* Always use contour for the difference */
2497         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
2498         ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);
2499         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
2500       }
2501       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
2502 
2503       if (flag_threshold) {
2504         PetscInt bs,rstart,rend,i;
2505         ierr = MatGetBlockSize(B,&bs);CHKERRQ(ierr);
2506         ierr = MatGetOwnershipRange(B,&rstart,&rend);CHKERRQ(ierr);
2507         for (i=rstart; i<rend; i++) {
2508           const PetscScalar *ba,*ca;
2509           const PetscInt    *bj,*cj;
2510           PetscInt          bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1;
2511           PetscReal         maxentry = 0,maxdiff = 0,maxrdiff = 0;
2512           ierr = MatGetRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr);
2513           ierr = MatGetRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
2514           if (bn != cn) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold");
2515           for (j=0; j<bn; j++) {
2516             PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
2517             if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) {
2518               maxentrycol = bj[j];
2519               maxentry    = PetscRealPart(ba[j]);
2520             }
2521             if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) {
2522               maxdiffcol = bj[j];
2523               maxdiff    = PetscRealPart(ca[j]);
2524             }
2525             if (rdiff > maxrdiff) {
2526               maxrdiffcol = bj[j];
2527               maxrdiff    = rdiff;
2528             }
2529           }
2530           if (maxrdiff > 1) {
2531             ierr = PetscViewerASCIIPrintf(vstdout,"row %D (maxentry=%g at %D, maxdiff=%g at %D, maxrdiff=%g at %D):",i,(double)maxentry,maxentrycol,(double)maxdiff,maxdiffcol,(double)maxrdiff,maxrdiffcol);CHKERRQ(ierr);
2532             for (j=0; j<bn; j++) {
2533               PetscReal rdiff;
2534               rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
2535               if (rdiff > 1) {
2536                 ierr = PetscViewerASCIIPrintf(vstdout," (%D,%g:%g)",bj[j],(double)PetscRealPart(ba[j]),(double)PetscRealPart(ca[j]));CHKERRQ(ierr);
2537               }
2538             }
2539             ierr = PetscViewerASCIIPrintf(vstdout,"\n",i,maxentry,maxdiff,maxrdiff);CHKERRQ(ierr);
2540           }
2541           ierr = MatRestoreRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr);
2542           ierr = MatRestoreRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
2543         }
2544       }
2545       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
2546       ierr = MatDestroy(&Bfd);CHKERRQ(ierr);
2547     }
2548   }
2549   PetscFunctionReturn(0);
2550 }
2551 
2552 /*MC
2553     SNESJacobianFunction - Function used to convey the nonlinear Jacobian of the function to be solved by SNES
2554 
2555      Synopsis:
2556      #include "petscsnes.h"
2557      PetscErrorCode SNESJacobianFunction(SNES snes,Vec x,Mat Amat,Mat Pmat,void *ctx);
2558 
2559 +  x - input vector
2560 .  Amat - the matrix that defines the (approximate) Jacobian
2561 .  Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat.
2562 -  ctx - [optional] user-defined Jacobian context
2563 
2564    Level: intermediate
2565 
2566 .seealso:   SNESSetFunction(), SNESGetFunction(), SNESSetJacobian(), SNESGetJacobian()
2567 M*/
2568 
2569 /*@C
2570    SNESSetJacobian - Sets the function to compute Jacobian as well as the
2571    location to store the matrix.
2572 
2573    Logically Collective on SNES and Mat
2574 
2575    Input Parameters:
2576 +  snes - the SNES context
2577 .  Amat - the matrix that defines the (approximate) Jacobian
2578 .  Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat.
2579 .  J - Jacobian evaluation routine (if NULL then SNES retains any previously set value), see SNESJacobianFunction for details
2580 -  ctx - [optional] user-defined context for private data for the
2581          Jacobian evaluation routine (may be NULL) (if NULL then SNES retains any previously set value)
2582 
2583    Notes:
2584    If the Amat matrix and Pmat matrix are different you must call MatAssemblyBegin/End() on
2585    each matrix.
2586 
2587    If you know the operator Amat has a null space you can use MatSetNullSpace() and MatSetTransposeNullSpace() to supply the null
2588    space to Amat and the KSP solvers will automatically use that null space as needed during the solution process.
2589 
2590    If using SNESComputeJacobianDefaultColor() to assemble a Jacobian, the ctx argument
2591    must be a MatFDColoring.
2592 
2593    Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian.  One common
2594    example is to use the "Picard linearization" which only differentiates through the highest order parts of each term.
2595 
2596    Level: beginner
2597 
2598 .keywords: SNES, nonlinear, set, Jacobian, matrix
2599 
2600 .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESComputeJacobianDefaultColor(), MatStructure, J,
2601           SNESSetPicard(), SNESJacobianFunction
2602 @*/
2603 PetscErrorCode  SNESSetJacobian(SNES snes,Mat Amat,Mat Pmat,PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx)
2604 {
2605   PetscErrorCode ierr;
2606   DM             dm;
2607 
2608   PetscFunctionBegin;
2609   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2610   if (Amat) PetscValidHeaderSpecific(Amat,MAT_CLASSID,2);
2611   if (Pmat) PetscValidHeaderSpecific(Pmat,MAT_CLASSID,3);
2612   if (Amat) PetscCheckSameComm(snes,1,Amat,2);
2613   if (Pmat) PetscCheckSameComm(snes,1,Pmat,3);
2614   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2615   ierr = DMSNESSetJacobian(dm,J,ctx);CHKERRQ(ierr);
2616   if (Amat) {
2617     ierr = PetscObjectReference((PetscObject)Amat);CHKERRQ(ierr);
2618     ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
2619 
2620     snes->jacobian = Amat;
2621   }
2622   if (Pmat) {
2623     ierr = PetscObjectReference((PetscObject)Pmat);CHKERRQ(ierr);
2624     ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
2625 
2626     snes->jacobian_pre = Pmat;
2627   }
2628   PetscFunctionReturn(0);
2629 }
2630 
2631 /*@C
2632    SNESGetJacobian - Returns the Jacobian matrix and optionally the user
2633    provided context for evaluating the Jacobian.
2634 
2635    Not Collective, but Mat object will be parallel if SNES object is
2636 
2637    Input Parameter:
2638 .  snes - the nonlinear solver context
2639 
2640    Output Parameters:
2641 +  Amat - location to stash (approximate) Jacobian matrix (or NULL)
2642 .  Pmat - location to stash matrix used to compute the preconditioner (or NULL)
2643 .  J - location to put Jacobian function (or NULL), see SNESJacobianFunction for details on its calling sequence
2644 -  ctx - location to stash Jacobian ctx (or NULL)
2645 
2646    Level: advanced
2647 
2648 .seealso: SNESSetJacobian(), SNESComputeJacobian(), SNESJacobianFunction, SNESGetFunction()
2649 @*/
2650 PetscErrorCode SNESGetJacobian(SNES snes,Mat *Amat,Mat *Pmat,PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx)
2651 {
2652   PetscErrorCode ierr;
2653   DM             dm;
2654   DMSNES         sdm;
2655 
2656   PetscFunctionBegin;
2657   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2658   if (Amat) *Amat = snes->jacobian;
2659   if (Pmat) *Pmat = snes->jacobian_pre;
2660   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2661   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
2662   if (J) *J = sdm->ops->computejacobian;
2663   if (ctx) *ctx = sdm->jacobianctx;
2664   PetscFunctionReturn(0);
2665 }
2666 
2667 /*@
2668    SNESSetUp - Sets up the internal data structures for the later use
2669    of a nonlinear solver.
2670 
2671    Collective on SNES
2672 
2673    Input Parameters:
2674 .  snes - the SNES context
2675 
2676    Notes:
2677    For basic use of the SNES solvers the user need not explicitly call
2678    SNESSetUp(), since these actions will automatically occur during
2679    the call to SNESSolve().  However, if one wishes to control this
2680    phase separately, SNESSetUp() should be called after SNESCreate()
2681    and optional routines of the form SNESSetXXX(), but before SNESSolve().
2682 
2683    Level: advanced
2684 
2685 .keywords: SNES, nonlinear, setup
2686 
2687 .seealso: SNESCreate(), SNESSolve(), SNESDestroy()
2688 @*/
2689 PetscErrorCode  SNESSetUp(SNES snes)
2690 {
2691   PetscErrorCode ierr;
2692   DM             dm;
2693   DMSNES         sdm;
2694   SNESLineSearch linesearch, pclinesearch;
2695   void           *lsprectx,*lspostctx;
2696   PetscErrorCode (*precheck)(SNESLineSearch,Vec,Vec,PetscBool*,void*);
2697   PetscErrorCode (*postcheck)(SNESLineSearch,Vec,Vec,Vec,PetscBool*,PetscBool*,void*);
2698   PetscErrorCode (*func)(SNES,Vec,Vec,void*);
2699   Vec            f,fpc;
2700   void           *funcctx;
2701   PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*);
2702   void           *jacctx,*appctx;
2703   Mat            j,jpre;
2704 
2705   PetscFunctionBegin;
2706   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2707   if (snes->setupcalled) PetscFunctionReturn(0);
2708 
2709   if (!((PetscObject)snes)->type_name) {
2710     ierr = SNESSetType(snes,SNESNEWTONLS);CHKERRQ(ierr);
2711   }
2712 
2713   ierr = SNESGetFunction(snes,&snes->vec_func,NULL,NULL);CHKERRQ(ierr);
2714 
2715   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2716   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
2717   if (!sdm->ops->computefunction) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE,"Function never provided to SNES object");
2718   if (!sdm->ops->computejacobian) {
2719     ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr);
2720   }
2721   if (!snes->vec_func) {
2722     ierr = DMCreateGlobalVector(dm,&snes->vec_func);CHKERRQ(ierr);
2723   }
2724 
2725   if (!snes->ksp) {
2726     ierr = SNESGetKSP(snes, &snes->ksp);CHKERRQ(ierr);
2727   }
2728 
2729   if (!snes->linesearch) {
2730     ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr);
2731   }
2732   ierr = SNESLineSearchSetFunction(snes->linesearch,SNESComputeFunction);CHKERRQ(ierr);
2733 
2734   if (snes->npc && (snes->npcside== PC_LEFT)) {
2735     snes->mf          = PETSC_TRUE;
2736     snes->mf_operator = PETSC_FALSE;
2737   }
2738 
2739   if (snes->npc) {
2740     /* copy the DM over */
2741     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2742     ierr = SNESSetDM(snes->npc,dm);CHKERRQ(ierr);
2743 
2744     ierr = SNESGetFunction(snes,&f,&func,&funcctx);CHKERRQ(ierr);
2745     ierr = VecDuplicate(f,&fpc);CHKERRQ(ierr);
2746     ierr = SNESSetFunction(snes->npc,fpc,func,funcctx);CHKERRQ(ierr);
2747     ierr = SNESGetJacobian(snes,&j,&jpre,&jac,&jacctx);CHKERRQ(ierr);
2748     ierr = SNESSetJacobian(snes->npc,j,jpre,jac,jacctx);CHKERRQ(ierr);
2749     ierr = SNESGetApplicationContext(snes,&appctx);CHKERRQ(ierr);
2750     ierr = SNESSetApplicationContext(snes->npc,appctx);CHKERRQ(ierr);
2751     ierr = VecDestroy(&fpc);CHKERRQ(ierr);
2752 
2753     /* copy the function pointers over */
2754     ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr);
2755 
2756     /* default to 1 iteration */
2757     ierr = SNESSetTolerances(snes->npc,0.0,0.0,0.0,1,snes->npc->max_funcs);CHKERRQ(ierr);
2758     if (snes->npcside==PC_RIGHT) {
2759       ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_FINAL_ONLY);CHKERRQ(ierr);
2760     } else {
2761       ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_NONE);CHKERRQ(ierr);
2762     }
2763     ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr);
2764 
2765     /* copy the line search context over */
2766     ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr);
2767     ierr = SNESGetLineSearch(snes->npc,&pclinesearch);CHKERRQ(ierr);
2768     ierr = SNESLineSearchGetPreCheck(linesearch,&precheck,&lsprectx);CHKERRQ(ierr);
2769     ierr = SNESLineSearchGetPostCheck(linesearch,&postcheck,&lspostctx);CHKERRQ(ierr);
2770     ierr = SNESLineSearchSetPreCheck(pclinesearch,precheck,lsprectx);CHKERRQ(ierr);
2771     ierr = SNESLineSearchSetPostCheck(pclinesearch,postcheck,lspostctx);CHKERRQ(ierr);
2772     ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)linesearch, (PetscObject)pclinesearch);CHKERRQ(ierr);
2773   }
2774   if (snes->mf) {
2775     ierr = SNESSetUpMatrixFree_Private(snes, snes->mf_operator, snes->mf_version);CHKERRQ(ierr);
2776   }
2777   if (snes->ops->usercompute && !snes->user) {
2778     ierr = (*snes->ops->usercompute)(snes,(void**)&snes->user);CHKERRQ(ierr);
2779   }
2780 
2781   snes->jac_iter = 0;
2782   snes->pre_iter = 0;
2783 
2784   if (snes->ops->setup) {
2785     ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr);
2786   }
2787 
2788   if (snes->npc && (snes->npcside== PC_LEFT)) {
2789     if (snes->functype == SNES_FUNCTION_PRECONDITIONED) {
2790       ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr);
2791       ierr = SNESLineSearchSetFunction(linesearch,SNESComputeFunctionDefaultNPC);CHKERRQ(ierr);
2792     }
2793   }
2794 
2795   snes->setupcalled = PETSC_TRUE;
2796   PetscFunctionReturn(0);
2797 }
2798 
2799 /*@
2800    SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats
2801 
2802    Collective on SNES
2803 
2804    Input Parameter:
2805 .  snes - iterative context obtained from SNESCreate()
2806 
2807    Level: intermediate
2808 
2809    Notes: Also calls the application context destroy routine set with SNESSetComputeApplicationContext()
2810 
2811 .keywords: SNES, destroy
2812 
2813 .seealso: SNESCreate(), SNESSetUp(), SNESSolve()
2814 @*/
2815 PetscErrorCode  SNESReset(SNES snes)
2816 {
2817   PetscErrorCode ierr;
2818 
2819   PetscFunctionBegin;
2820   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2821   if (snes->ops->userdestroy && snes->user) {
2822     ierr       = (*snes->ops->userdestroy)((void**)&snes->user);CHKERRQ(ierr);
2823     snes->user = NULL;
2824   }
2825   if (snes->npc) {
2826     ierr = SNESReset(snes->npc);CHKERRQ(ierr);
2827   }
2828 
2829   if (snes->ops->reset) {
2830     ierr = (*snes->ops->reset)(snes);CHKERRQ(ierr);
2831   }
2832   if (snes->ksp) {
2833     ierr = KSPReset(snes->ksp);CHKERRQ(ierr);
2834   }
2835 
2836   if (snes->linesearch) {
2837     ierr = SNESLineSearchReset(snes->linesearch);CHKERRQ(ierr);
2838   }
2839 
2840   ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
2841   ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
2842   ierr = VecDestroy(&snes->vec_sol_update);CHKERRQ(ierr);
2843   ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr);
2844   ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
2845   ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
2846   ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr);
2847   ierr = VecDestroyVecs(snes->nvwork,&snes->vwork);CHKERRQ(ierr);
2848 
2849   snes->alwayscomputesfinalresidual = PETSC_FALSE;
2850 
2851   snes->nwork       = snes->nvwork = 0;
2852   snes->setupcalled = PETSC_FALSE;
2853   PetscFunctionReturn(0);
2854 }
2855 
2856 /*@
2857    SNESDestroy - Destroys the nonlinear solver context that was created
2858    with SNESCreate().
2859 
2860    Collective on SNES
2861 
2862    Input Parameter:
2863 .  snes - the SNES context
2864 
2865    Level: beginner
2866 
2867 .keywords: SNES, nonlinear, destroy
2868 
2869 .seealso: SNESCreate(), SNESSolve()
2870 @*/
2871 PetscErrorCode  SNESDestroy(SNES *snes)
2872 {
2873   PetscErrorCode ierr;
2874 
2875   PetscFunctionBegin;
2876   if (!*snes) PetscFunctionReturn(0);
2877   PetscValidHeaderSpecific((*snes),SNES_CLASSID,1);
2878   if (--((PetscObject)(*snes))->refct > 0) {*snes = 0; PetscFunctionReturn(0);}
2879 
2880   ierr = SNESReset((*snes));CHKERRQ(ierr);
2881   ierr = SNESDestroy(&(*snes)->npc);CHKERRQ(ierr);
2882 
2883   /* if memory was published with SAWs then destroy it */
2884   ierr = PetscObjectSAWsViewOff((PetscObject)*snes);CHKERRQ(ierr);
2885   if ((*snes)->ops->destroy) {ierr = (*((*snes))->ops->destroy)((*snes));CHKERRQ(ierr);}
2886 
2887   ierr = DMDestroy(&(*snes)->dm);CHKERRQ(ierr);
2888   ierr = KSPDestroy(&(*snes)->ksp);CHKERRQ(ierr);
2889   ierr = SNESLineSearchDestroy(&(*snes)->linesearch);CHKERRQ(ierr);
2890 
2891   ierr = PetscFree((*snes)->kspconvctx);CHKERRQ(ierr);
2892   if ((*snes)->ops->convergeddestroy) {
2893     ierr = (*(*snes)->ops->convergeddestroy)((*snes)->cnvP);CHKERRQ(ierr);
2894   }
2895   if ((*snes)->conv_malloc) {
2896     ierr = PetscFree((*snes)->conv_hist);CHKERRQ(ierr);
2897     ierr = PetscFree((*snes)->conv_hist_its);CHKERRQ(ierr);
2898   }
2899   ierr = SNESMonitorCancel((*snes));CHKERRQ(ierr);
2900   ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr);
2901   PetscFunctionReturn(0);
2902 }
2903 
2904 /* ----------- Routines to set solver parameters ---------- */
2905 
2906 /*@
2907    SNESSetLagPreconditioner - Determines when the preconditioner is rebuilt in the nonlinear solve.
2908 
2909    Logically Collective on SNES
2910 
2911    Input Parameters:
2912 +  snes - the SNES context
2913 -  lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
2914          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
2915 
2916    Options Database Keys:
2917 .    -snes_lag_preconditioner <lag>
2918 
2919    Notes:
2920    The default is 1
2921    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
2922    If  -1 is used before the very first nonlinear solve the preconditioner is still built because there is no previous preconditioner to use
2923 
2924    Level: intermediate
2925 
2926 .keywords: SNES, nonlinear, set, convergence, tolerances
2927 
2928 .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian()
2929 
2930 @*/
2931 PetscErrorCode  SNESSetLagPreconditioner(SNES snes,PetscInt lag)
2932 {
2933   PetscFunctionBegin;
2934   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2935   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
2936   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
2937   PetscValidLogicalCollectiveInt(snes,lag,2);
2938   snes->lagpreconditioner = lag;
2939   PetscFunctionReturn(0);
2940 }
2941 
2942 /*@
2943    SNESSetGridSequence - sets the number of steps of grid sequencing that SNES does
2944 
2945    Logically Collective on SNES
2946 
2947    Input Parameters:
2948 +  snes - the SNES context
2949 -  steps - the number of refinements to do, defaults to 0
2950 
2951    Options Database Keys:
2952 .    -snes_grid_sequence <steps>
2953 
2954    Level: intermediate
2955 
2956    Notes:
2957    Use SNESGetSolution() to extract the fine grid solution after grid sequencing.
2958 
2959 .keywords: SNES, nonlinear, set, convergence, tolerances
2960 
2961 .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetGridSequence()
2962 
2963 @*/
2964 PetscErrorCode  SNESSetGridSequence(SNES snes,PetscInt steps)
2965 {
2966   PetscFunctionBegin;
2967   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2968   PetscValidLogicalCollectiveInt(snes,steps,2);
2969   snes->gridsequence = steps;
2970   PetscFunctionReturn(0);
2971 }
2972 
2973 /*@
2974    SNESGetGridSequence - gets the number of steps of grid sequencing that SNES does
2975 
2976    Logically Collective on SNES
2977 
2978    Input Parameter:
2979 .  snes - the SNES context
2980 
2981    Output Parameter:
2982 .  steps - the number of refinements to do, defaults to 0
2983 
2984    Options Database Keys:
2985 .    -snes_grid_sequence <steps>
2986 
2987    Level: intermediate
2988 
2989    Notes:
2990    Use SNESGetSolution() to extract the fine grid solution after grid sequencing.
2991 
2992 .keywords: SNES, nonlinear, set, convergence, tolerances
2993 
2994 .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetGridSequence()
2995 
2996 @*/
2997 PetscErrorCode  SNESGetGridSequence(SNES snes,PetscInt *steps)
2998 {
2999   PetscFunctionBegin;
3000   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3001   *steps = snes->gridsequence;
3002   PetscFunctionReturn(0);
3003 }
3004 
3005 /*@
3006    SNESGetLagPreconditioner - Indicates how often the preconditioner is rebuilt
3007 
3008    Not Collective
3009 
3010    Input Parameter:
3011 .  snes - the SNES context
3012 
3013    Output Parameter:
3014 .   lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
3015          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
3016 
3017    Options Database Keys:
3018 .    -snes_lag_preconditioner <lag>
3019 
3020    Notes:
3021    The default is 1
3022    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3023 
3024    Level: intermediate
3025 
3026 .keywords: SNES, nonlinear, set, convergence, tolerances
3027 
3028 .seealso: SNESSetTrustRegionTolerance(), SNESSetLagPreconditioner()
3029 
3030 @*/
3031 PetscErrorCode  SNESGetLagPreconditioner(SNES snes,PetscInt *lag)
3032 {
3033   PetscFunctionBegin;
3034   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3035   *lag = snes->lagpreconditioner;
3036   PetscFunctionReturn(0);
3037 }
3038 
3039 /*@
3040    SNESSetLagJacobian - Determines when the Jacobian is rebuilt in the nonlinear solve. See SNESSetLagPreconditioner() for determining how
3041      often the preconditioner is rebuilt.
3042 
3043    Logically Collective on SNES
3044 
3045    Input Parameters:
3046 +  snes - the SNES context
3047 -  lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
3048          the Jacobian is built etc. -2 means rebuild at next chance but then never again
3049 
3050    Options Database Keys:
3051 .    -snes_lag_jacobian <lag>
3052 
3053    Notes:
3054    The default is 1
3055    The Jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3056    If  -1 is used before the very first nonlinear solve the CODE WILL FAIL! because no Jacobian is used, use -2 to indicate you want it recomputed
3057    at the next Newton step but never again (unless it is reset to another value)
3058 
3059    Level: intermediate
3060 
3061 .keywords: SNES, nonlinear, set, convergence, tolerances
3062 
3063 .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagPreconditioner(), SNESGetLagJacobian()
3064 
3065 @*/
3066 PetscErrorCode  SNESSetLagJacobian(SNES snes,PetscInt lag)
3067 {
3068   PetscFunctionBegin;
3069   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3070   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
3071   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
3072   PetscValidLogicalCollectiveInt(snes,lag,2);
3073   snes->lagjacobian = lag;
3074   PetscFunctionReturn(0);
3075 }
3076 
3077 /*@
3078    SNESGetLagJacobian - Indicates how often the Jacobian is rebuilt. See SNESGetLagPreconditioner() to determine when the preconditioner is rebuilt
3079 
3080    Not Collective
3081 
3082    Input Parameter:
3083 .  snes - the SNES context
3084 
3085    Output Parameter:
3086 .   lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
3087          the Jacobian is built etc.
3088 
3089    Options Database Keys:
3090 .    -snes_lag_jacobian <lag>
3091 
3092    Notes:
3093    The default is 1
3094    The jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3095 
3096    Level: intermediate
3097 
3098 .keywords: SNES, nonlinear, set, convergence, tolerances
3099 
3100 .seealso: SNESSetTrustRegionTolerance(), SNESSetLagJacobian(), SNESSetLagPreconditioner(), SNESGetLagPreconditioner()
3101 
3102 @*/
3103 PetscErrorCode  SNESGetLagJacobian(SNES snes,PetscInt *lag)
3104 {
3105   PetscFunctionBegin;
3106   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3107   *lag = snes->lagjacobian;
3108   PetscFunctionReturn(0);
3109 }
3110 
3111 /*@
3112    SNESSetLagJacobianPersists - Set whether or not the Jacobian lagging persists through multiple solves
3113 
3114    Logically collective on SNES
3115 
3116    Input Parameter:
3117 +  snes - the SNES context
3118 -   flg - jacobian lagging persists if true
3119 
3120    Options Database Keys:
3121 .    -snes_lag_jacobian_persists <flg>
3122 
3123    Notes: This is useful both for nonlinear preconditioning, where it's appropriate to have the Jacobian be stale by
3124    several solves, and for implicit time-stepping, where Jacobian lagging in the inner nonlinear solve over several
3125    timesteps may present huge efficiency gains.
3126 
3127    Level: developer
3128 
3129 .keywords: SNES, nonlinear, lag
3130 
3131 .seealso: SNESSetLagPreconditionerPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC()
3132 
3133 @*/
3134 PetscErrorCode  SNESSetLagJacobianPersists(SNES snes,PetscBool flg)
3135 {
3136   PetscFunctionBegin;
3137   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3138   PetscValidLogicalCollectiveBool(snes,flg,2);
3139   snes->lagjac_persist = flg;
3140   PetscFunctionReturn(0);
3141 }
3142 
3143 /*@
3144    SNESSetLagPreconditionerPersists - Set whether or not the preconditioner lagging persists through multiple solves
3145 
3146    Logically Collective on SNES
3147 
3148    Input Parameter:
3149 +  snes - the SNES context
3150 -   flg - preconditioner lagging persists if true
3151 
3152    Options Database Keys:
3153 .    -snes_lag_jacobian_persists <flg>
3154 
3155    Notes: This is useful both for nonlinear preconditioning, where it's appropriate to have the preconditioner be stale
3156    by several solves, and for implicit time-stepping, where preconditioner lagging in the inner nonlinear solve over
3157    several timesteps may present huge efficiency gains.
3158 
3159    Level: developer
3160 
3161 .keywords: SNES, nonlinear, lag
3162 
3163 .seealso: SNESSetLagJacobianPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC()
3164 
3165 @*/
3166 PetscErrorCode  SNESSetLagPreconditionerPersists(SNES snes,PetscBool flg)
3167 {
3168   PetscFunctionBegin;
3169   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3170   PetscValidLogicalCollectiveBool(snes,flg,2);
3171   snes->lagpre_persist = flg;
3172   PetscFunctionReturn(0);
3173 }
3174 
3175 /*@
3176    SNESSetForceIteration - force SNESSolve() to take at least one iteration regardless of the initial residual norm
3177 
3178    Logically Collective on SNES
3179 
3180    Input Parameters:
3181 +  snes - the SNES context
3182 -  force - PETSC_TRUE require at least one iteration
3183 
3184    Options Database Keys:
3185 .    -snes_force_iteration <force> - Sets forcing an iteration
3186 
3187    Notes:
3188    This is used sometimes with TS to prevent TS from detecting a false steady state solution
3189 
3190    Level: intermediate
3191 
3192 .keywords: SNES, nonlinear, set, convergence, tolerances
3193 
3194 .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance()
3195 @*/
3196 PetscErrorCode  SNESSetForceIteration(SNES snes,PetscBool force)
3197 {
3198   PetscFunctionBegin;
3199   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3200   snes->forceiteration = force;
3201   PetscFunctionReturn(0);
3202 }
3203 
3204 
3205 /*@
3206    SNESSetTolerances - Sets various parameters used in convergence tests.
3207 
3208    Logically Collective on SNES
3209 
3210    Input Parameters:
3211 +  snes - the SNES context
3212 .  abstol - absolute convergence tolerance
3213 .  rtol - relative convergence tolerance
3214 .  stol -  convergence tolerance in terms of the norm of the change in the solution between steps,  || delta x || < stol*|| x ||
3215 .  maxit - maximum number of iterations
3216 -  maxf - maximum number of function evaluations
3217 
3218    Options Database Keys:
3219 +    -snes_atol <abstol> - Sets abstol
3220 .    -snes_rtol <rtol> - Sets rtol
3221 .    -snes_stol <stol> - Sets stol
3222 .    -snes_max_it <maxit> - Sets maxit
3223 -    -snes_max_funcs <maxf> - Sets maxf
3224 
3225    Notes:
3226    The default maximum number of iterations is 50.
3227    The default maximum number of function evaluations is 1000.
3228 
3229    Level: intermediate
3230 
3231 .keywords: SNES, nonlinear, set, convergence, tolerances
3232 
3233 .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance(), SNESSetForceIteration()
3234 @*/
3235 PetscErrorCode  SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf)
3236 {
3237   PetscFunctionBegin;
3238   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3239   PetscValidLogicalCollectiveReal(snes,abstol,2);
3240   PetscValidLogicalCollectiveReal(snes,rtol,3);
3241   PetscValidLogicalCollectiveReal(snes,stol,4);
3242   PetscValidLogicalCollectiveInt(snes,maxit,5);
3243   PetscValidLogicalCollectiveInt(snes,maxf,6);
3244 
3245   if (abstol != PETSC_DEFAULT) {
3246     if (abstol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Absolute tolerance %g must be non-negative",(double)abstol);
3247     snes->abstol = abstol;
3248   }
3249   if (rtol != PETSC_DEFAULT) {
3250     if (rtol < 0.0 || 1.0 <= rtol) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Relative tolerance %g must be non-negative and less than 1.0",(double)rtol);
3251     snes->rtol = rtol;
3252   }
3253   if (stol != PETSC_DEFAULT) {
3254     if (stol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Step tolerance %g must be non-negative",(double)stol);
3255     snes->stol = stol;
3256   }
3257   if (maxit != PETSC_DEFAULT) {
3258     if (maxit < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of iterations %D must be non-negative",maxit);
3259     snes->max_its = maxit;
3260   }
3261   if (maxf != PETSC_DEFAULT) {
3262     if (maxf < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of function evaluations %D must be non-negative",maxf);
3263     snes->max_funcs = maxf;
3264   }
3265   snes->tolerancesset = PETSC_TRUE;
3266   PetscFunctionReturn(0);
3267 }
3268 
3269 /*@
3270    SNESSetDivergenceTolerance - Sets the divergence tolerance used for the SNES divergence test.
3271 
3272    Logically Collective on SNES
3273 
3274    Input Parameters:
3275 +  snes - the SNES context
3276 -  divtol - the divergence tolerance. Use -1 to deactivate the test.
3277 
3278    Options Database Keys:
3279 +    -snes_divergence_tolerance <divtol> - Sets divtol
3280 
3281    Notes:
3282    The default divergence tolerance is 1e4.
3283 
3284    Level: intermediate
3285 
3286 .keywords: SNES, nonlinear, set, divergence, tolerance
3287 
3288 .seealso: SNESSetTolerances(), SNESGetDivergenceTolerance
3289 @*/
3290 PetscErrorCode  SNESSetDivergenceTolerance(SNES snes,PetscReal divtol)
3291 {
3292   PetscFunctionBegin;
3293   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3294   PetscValidLogicalCollectiveReal(snes,divtol,2);
3295 
3296   if (divtol != PETSC_DEFAULT) {
3297     snes->divtol = divtol;
3298   }
3299   else {
3300     snes->divtol = 1.0e4;
3301   }
3302   PetscFunctionReturn(0);
3303 }
3304 
3305 /*@
3306    SNESGetTolerances - Gets various parameters used in convergence tests.
3307 
3308    Not Collective
3309 
3310    Input Parameters:
3311 +  snes - the SNES context
3312 .  atol - absolute convergence tolerance
3313 .  rtol - relative convergence tolerance
3314 .  stol -  convergence tolerance in terms of the norm
3315            of the change in the solution between steps
3316 .  maxit - maximum number of iterations
3317 -  maxf - maximum number of function evaluations
3318 
3319    Notes:
3320    The user can specify NULL for any parameter that is not needed.
3321 
3322    Level: intermediate
3323 
3324 .keywords: SNES, nonlinear, get, convergence, tolerances
3325 
3326 .seealso: SNESSetTolerances()
3327 @*/
3328 PetscErrorCode  SNESGetTolerances(SNES snes,PetscReal *atol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf)
3329 {
3330   PetscFunctionBegin;
3331   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3332   if (atol)  *atol  = snes->abstol;
3333   if (rtol)  *rtol  = snes->rtol;
3334   if (stol)  *stol  = snes->stol;
3335   if (maxit) *maxit = snes->max_its;
3336   if (maxf)  *maxf  = snes->max_funcs;
3337   PetscFunctionReturn(0);
3338 }
3339 
3340 /*@
3341    SNESGetDivergenceTolerance - Gets divergence tolerance used in divergence test.
3342 
3343    Not Collective
3344 
3345    Input Parameters:
3346 +  snes - the SNES context
3347 -  divtol - divergence tolerance
3348 
3349    Level: intermediate
3350 
3351 .keywords: SNES, nonlinear, get, divergence, tolerance
3352 
3353 .seealso: SNESSetDivergenceTolerance()
3354 @*/
3355 PetscErrorCode  SNESGetDivergenceTolerance(SNES snes,PetscReal *divtol)
3356 {
3357   PetscFunctionBegin;
3358   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3359   if (divtol) *divtol = snes->divtol;
3360   PetscFunctionReturn(0);
3361 }
3362 
3363 /*@
3364    SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance.
3365 
3366    Logically Collective on SNES
3367 
3368    Input Parameters:
3369 +  snes - the SNES context
3370 -  tol - tolerance
3371 
3372    Options Database Key:
3373 .  -snes_trtol <tol> - Sets tol
3374 
3375    Level: intermediate
3376 
3377 .keywords: SNES, nonlinear, set, trust region, tolerance
3378 
3379 .seealso: SNESSetTolerances()
3380 @*/
3381 PetscErrorCode  SNESSetTrustRegionTolerance(SNES snes,PetscReal tol)
3382 {
3383   PetscFunctionBegin;
3384   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3385   PetscValidLogicalCollectiveReal(snes,tol,2);
3386   snes->deltatol = tol;
3387   PetscFunctionReturn(0);
3388 }
3389 
3390 /*
3391    Duplicate the lg monitors for SNES from KSP; for some reason with
3392    dynamic libraries things don't work under Sun4 if we just use
3393    macros instead of functions
3394 */
3395 PetscErrorCode  SNESMonitorLGResidualNorm(SNES snes,PetscInt it,PetscReal norm,void *ctx)
3396 {
3397   PetscErrorCode ierr;
3398 
3399   PetscFunctionBegin;
3400   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3401   ierr = KSPMonitorLGResidualNorm((KSP)snes,it,norm,ctx);CHKERRQ(ierr);
3402   PetscFunctionReturn(0);
3403 }
3404 
3405 PetscErrorCode  SNESMonitorLGCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *lgctx)
3406 {
3407   PetscErrorCode ierr;
3408 
3409   PetscFunctionBegin;
3410   ierr = KSPMonitorLGResidualNormCreate(comm,host,label,x,y,m,n,lgctx);CHKERRQ(ierr);
3411   PetscFunctionReturn(0);
3412 }
3413 
3414 PETSC_INTERN PetscErrorCode  SNESMonitorRange_Private(SNES,PetscInt,PetscReal*);
3415 
3416 PetscErrorCode  SNESMonitorLGRange(SNES snes,PetscInt n,PetscReal rnorm,void *monctx)
3417 {
3418   PetscDrawLG      lg;
3419   PetscErrorCode   ierr;
3420   PetscReal        x,y,per;
3421   PetscViewer      v = (PetscViewer)monctx;
3422   static PetscReal prev; /* should be in the context */
3423   PetscDraw        draw;
3424 
3425   PetscFunctionBegin;
3426   PetscValidHeaderSpecific(v,PETSC_VIEWER_CLASSID,4);
3427   ierr = PetscViewerDrawGetDrawLG(v,0,&lg);CHKERRQ(ierr);
3428   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3429   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3430   ierr = PetscDrawSetTitle(draw,"Residual norm");CHKERRQ(ierr);
3431   x    = (PetscReal)n;
3432   if (rnorm > 0.0) y = PetscLog10Real(rnorm);
3433   else y = -15.0;
3434   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
3435   if (n < 20 || !(n % 5) || snes->reason) {
3436     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
3437     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3438   }
3439 
3440   ierr = PetscViewerDrawGetDrawLG(v,1,&lg);CHKERRQ(ierr);
3441   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3442   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3443   ierr = PetscDrawSetTitle(draw,"% elemts > .2*max elemt");CHKERRQ(ierr);
3444   ierr =  SNESMonitorRange_Private(snes,n,&per);CHKERRQ(ierr);
3445   x    = (PetscReal)n;
3446   y    = 100.0*per;
3447   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
3448   if (n < 20 || !(n % 5) || snes->reason) {
3449     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
3450     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3451   }
3452 
3453   ierr = PetscViewerDrawGetDrawLG(v,2,&lg);CHKERRQ(ierr);
3454   if (!n) {prev = rnorm;ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3455   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3456   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm");CHKERRQ(ierr);
3457   x    = (PetscReal)n;
3458   y    = (prev - rnorm)/prev;
3459   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
3460   if (n < 20 || !(n % 5) || snes->reason) {
3461     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
3462     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3463   }
3464 
3465   ierr = PetscViewerDrawGetDrawLG(v,3,&lg);CHKERRQ(ierr);
3466   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3467   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3468   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr);
3469   x    = (PetscReal)n;
3470   y    = (prev - rnorm)/(prev*per);
3471   if (n > 2) { /*skip initial crazy value */
3472     ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
3473   }
3474   if (n < 20 || !(n % 5) || snes->reason) {
3475     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
3476     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3477   }
3478   prev = rnorm;
3479   PetscFunctionReturn(0);
3480 }
3481 
3482 /*@
3483    SNESMonitor - runs the user provided monitor routines, if they exist
3484 
3485    Collective on SNES
3486 
3487    Input Parameters:
3488 +  snes - nonlinear solver context obtained from SNESCreate()
3489 .  iter - iteration number
3490 -  rnorm - relative norm of the residual
3491 
3492    Notes:
3493    This routine is called by the SNES implementations.
3494    It does not typically need to be called by the user.
3495 
3496    Level: developer
3497 
3498 .seealso: SNESMonitorSet()
3499 @*/
3500 PetscErrorCode  SNESMonitor(SNES snes,PetscInt iter,PetscReal rnorm)
3501 {
3502   PetscErrorCode ierr;
3503   PetscInt       i,n = snes->numbermonitors;
3504 
3505   PetscFunctionBegin;
3506   ierr = VecLockPush(snes->vec_sol);CHKERRQ(ierr);
3507   for (i=0; i<n; i++) {
3508     ierr = (*snes->monitor[i])(snes,iter,rnorm,snes->monitorcontext[i]);CHKERRQ(ierr);
3509   }
3510   ierr = VecLockPop(snes->vec_sol);CHKERRQ(ierr);
3511   PetscFunctionReturn(0);
3512 }
3513 
3514 /* ------------ Routines to set performance monitoring options ----------- */
3515 
3516 /*MC
3517     SNESMonitorFunction - functional form passed to SNESMonitorSet() to monitor convergence of nonlinear solver
3518 
3519      Synopsis:
3520      #include <petscsnes.h>
3521 $    PetscErrorCode SNESMonitorFunction(SNES snes,PetscInt its, PetscReal norm,void *mctx)
3522 
3523 +    snes - the SNES context
3524 .    its - iteration number
3525 .    norm - 2-norm function value (may be estimated)
3526 -    mctx - [optional] monitoring context
3527 
3528    Level: advanced
3529 
3530 .seealso:   SNESMonitorSet(), SNESMonitorGet()
3531 M*/
3532 
3533 /*@C
3534    SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every
3535    iteration of the nonlinear solver to display the iteration's
3536    progress.
3537 
3538    Logically Collective on SNES
3539 
3540    Input Parameters:
3541 +  snes - the SNES context
3542 .  f - the monitor function, see SNESMonitorFunction for the calling sequence
3543 .  mctx - [optional] user-defined context for private data for the
3544           monitor routine (use NULL if no context is desired)
3545 -  monitordestroy - [optional] routine that frees monitor context
3546           (may be NULL)
3547 
3548    Options Database Keys:
3549 +    -snes_monitor        - sets SNESMonitorDefault()
3550 .    -snes_monitor_lg_residualnorm    - sets line graph monitor,
3551                             uses SNESMonitorLGCreate()
3552 -    -snes_monitor_cancel - cancels all monitors that have
3553                             been hardwired into a code by
3554                             calls to SNESMonitorSet(), but
3555                             does not cancel those set via
3556                             the options database.
3557 
3558    Notes:
3559    Several different monitoring routines may be set by calling
3560    SNESMonitorSet() multiple times; all will be called in the
3561    order in which they were set.
3562 
3563    Fortran notes: Only a single monitor function can be set for each SNES object
3564 
3565    Level: intermediate
3566 
3567 .keywords: SNES, nonlinear, set, monitor
3568 
3569 .seealso: SNESMonitorDefault(), SNESMonitorCancel(), SNESMonitorFunction
3570 @*/
3571 PetscErrorCode  SNESMonitorSet(SNES snes,PetscErrorCode (*f)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void**))
3572 {
3573   PetscInt       i;
3574   PetscErrorCode ierr;
3575   PetscBool      identical;
3576 
3577   PetscFunctionBegin;
3578   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3579   for (i=0; i<snes->numbermonitors;i++) {
3580     ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,mctx,monitordestroy,(PetscErrorCode (*)(void))snes->monitor[i],snes->monitorcontext[i],snes->monitordestroy[i],&identical);CHKERRQ(ierr);
3581     if (identical) PetscFunctionReturn(0);
3582   }
3583   if (snes->numbermonitors >= MAXSNESMONITORS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set");
3584   snes->monitor[snes->numbermonitors]          = f;
3585   snes->monitordestroy[snes->numbermonitors]   = monitordestroy;
3586   snes->monitorcontext[snes->numbermonitors++] = (void*)mctx;
3587   PetscFunctionReturn(0);
3588 }
3589 
3590 /*@
3591    SNESMonitorCancel - Clears all the monitor functions for a SNES object.
3592 
3593    Logically Collective on SNES
3594 
3595    Input Parameters:
3596 .  snes - the SNES context
3597 
3598    Options Database Key:
3599 .  -snes_monitor_cancel - cancels all monitors that have been hardwired
3600     into a code by calls to SNESMonitorSet(), but does not cancel those
3601     set via the options database
3602 
3603    Notes:
3604    There is no way to clear one specific monitor from a SNES object.
3605 
3606    Level: intermediate
3607 
3608 .keywords: SNES, nonlinear, set, monitor
3609 
3610 .seealso: SNESMonitorDefault(), SNESMonitorSet()
3611 @*/
3612 PetscErrorCode  SNESMonitorCancel(SNES snes)
3613 {
3614   PetscErrorCode ierr;
3615   PetscInt       i;
3616 
3617   PetscFunctionBegin;
3618   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3619   for (i=0; i<snes->numbermonitors; i++) {
3620     if (snes->monitordestroy[i]) {
3621       ierr = (*snes->monitordestroy[i])(&snes->monitorcontext[i]);CHKERRQ(ierr);
3622     }
3623   }
3624   snes->numbermonitors = 0;
3625   PetscFunctionReturn(0);
3626 }
3627 
3628 /*MC
3629     SNESConvergenceTestFunction - functional form used for testing of convergence of nonlinear solver
3630 
3631      Synopsis:
3632      #include <petscsnes.h>
3633 $     PetscErrorCode SNESConvergenceTest(SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx)
3634 
3635 +    snes - the SNES context
3636 .    it - current iteration (0 is the first and is before any Newton step)
3637 .    cctx - [optional] convergence context
3638 .    reason - reason for convergence/divergence
3639 .    xnorm - 2-norm of current iterate
3640 .    gnorm - 2-norm of current step
3641 -    f - 2-norm of function
3642 
3643    Level: intermediate
3644 
3645 .seealso:   SNESSetConvergenceTest(), SNESGetConvergenceTest()
3646 M*/
3647 
3648 /*@C
3649    SNESSetConvergenceTest - Sets the function that is to be used
3650    to test for convergence of the nonlinear iterative solution.
3651 
3652    Logically Collective on SNES
3653 
3654    Input Parameters:
3655 +  snes - the SNES context
3656 .  SNESConvergenceTestFunction - routine to test for convergence
3657 .  cctx - [optional] context for private data for the convergence routine  (may be NULL)
3658 -  destroy - [optional] destructor for the context (may be NULL; NULL_FUNCTION in Fortran)
3659 
3660    Level: advanced
3661 
3662 .keywords: SNES, nonlinear, set, convergence, test
3663 
3664 .seealso: SNESConvergedDefault(), SNESConvergedSkip(), SNESConvergenceTestFunction
3665 @*/
3666 PetscErrorCode  SNESSetConvergenceTest(SNES snes,PetscErrorCode (*SNESConvergenceTestFunction)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx,PetscErrorCode (*destroy)(void*))
3667 {
3668   PetscErrorCode ierr;
3669 
3670   PetscFunctionBegin;
3671   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3672   if (!SNESConvergenceTestFunction) SNESConvergenceTestFunction = SNESConvergedSkip;
3673   if (snes->ops->convergeddestroy) {
3674     ierr = (*snes->ops->convergeddestroy)(snes->cnvP);CHKERRQ(ierr);
3675   }
3676   snes->ops->converged        = SNESConvergenceTestFunction;
3677   snes->ops->convergeddestroy = destroy;
3678   snes->cnvP                  = cctx;
3679   PetscFunctionReturn(0);
3680 }
3681 
3682 /*@
3683    SNESGetConvergedReason - Gets the reason the SNES iteration was stopped.
3684 
3685    Not Collective
3686 
3687    Input Parameter:
3688 .  snes - the SNES context
3689 
3690    Output Parameter:
3691 .  reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the
3692             manual pages for the individual convergence tests for complete lists
3693 
3694    Options Database:
3695 .   -snes_converged_reason - prints the reason to standard out
3696 
3697    Level: intermediate
3698 
3699    Notes: Should only be called after the call the SNESSolve() is complete, if it is called earlier it returns the value SNES__CONVERGED_ITERATING.
3700 
3701 .keywords: SNES, nonlinear, set, convergence, test
3702 
3703 .seealso: SNESSetConvergenceTest(), SNESSetConvergedReason(), SNESConvergedReason
3704 @*/
3705 PetscErrorCode SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason)
3706 {
3707   PetscFunctionBegin;
3708   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3709   PetscValidPointer(reason,2);
3710   *reason = snes->reason;
3711   PetscFunctionReturn(0);
3712 }
3713 
3714 /*@
3715    SNESSetConvergedReason - Sets the reason the SNES iteration was stopped.
3716 
3717    Not Collective
3718 
3719    Input Parameters:
3720 +  snes - the SNES context
3721 -  reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the
3722             manual pages for the individual convergence tests for complete lists
3723 
3724    Level: intermediate
3725 
3726 .keywords: SNES, nonlinear, set, convergence, test
3727 .seealso: SNESGetConvergedReason(), SNESSetConvergenceTest(), SNESConvergedReason
3728 @*/
3729 PetscErrorCode SNESSetConvergedReason(SNES snes,SNESConvergedReason reason)
3730 {
3731   PetscFunctionBegin;
3732   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3733   snes->reason = reason;
3734   PetscFunctionReturn(0);
3735 }
3736 
3737 /*@
3738    SNESSetConvergenceHistory - Sets the array used to hold the convergence history.
3739 
3740    Logically Collective on SNES
3741 
3742    Input Parameters:
3743 +  snes - iterative context obtained from SNESCreate()
3744 .  a   - array to hold history, this array will contain the function norms computed at each step
3745 .  its - integer array holds the number of linear iterations for each solve.
3746 .  na  - size of a and its
3747 -  reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero,
3748            else it continues storing new values for new nonlinear solves after the old ones
3749 
3750    Notes:
3751    If 'a' and 'its' are NULL then space is allocated for the history. If 'na' PETSC_DECIDE or PETSC_DEFAULT then a
3752    default array of length 10000 is allocated.
3753 
3754    This routine is useful, e.g., when running a code for purposes
3755    of accurate performance monitoring, when no I/O should be done
3756    during the section of code that is being timed.
3757 
3758    Level: intermediate
3759 
3760 .keywords: SNES, set, convergence, history
3761 
3762 .seealso: SNESGetConvergenceHistory()
3763 
3764 @*/
3765 PetscErrorCode  SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscBool reset)
3766 {
3767   PetscErrorCode ierr;
3768 
3769   PetscFunctionBegin;
3770   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3771   if (a) PetscValidScalarPointer(a,2);
3772   if (its) PetscValidIntPointer(its,3);
3773   if (!a) {
3774     if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
3775     ierr = PetscCalloc1(na,&a);CHKERRQ(ierr);
3776     ierr = PetscCalloc1(na,&its);CHKERRQ(ierr);
3777 
3778     snes->conv_malloc = PETSC_TRUE;
3779   }
3780   snes->conv_hist       = a;
3781   snes->conv_hist_its   = its;
3782   snes->conv_hist_max   = na;
3783   snes->conv_hist_len   = 0;
3784   snes->conv_hist_reset = reset;
3785   PetscFunctionReturn(0);
3786 }
3787 
3788 #if defined(PETSC_HAVE_MATLAB_ENGINE)
3789 #include <engine.h>   /* MATLAB include file */
3790 #include <mex.h>      /* MATLAB include file */
3791 
3792 PETSC_EXTERN mxArray *SNESGetConvergenceHistoryMatlab(SNES snes)
3793 {
3794   mxArray   *mat;
3795   PetscInt  i;
3796   PetscReal *ar;
3797 
3798   PetscFunctionBegin;
3799   mat = mxCreateDoubleMatrix(snes->conv_hist_len,1,mxREAL);
3800   ar  = (PetscReal*) mxGetData(mat);
3801   for (i=0; i<snes->conv_hist_len; i++) ar[i] = snes->conv_hist[i];
3802   PetscFunctionReturn(mat);
3803 }
3804 #endif
3805 
3806 /*@C
3807    SNESGetConvergenceHistory - Gets the array used to hold the convergence history.
3808 
3809    Not Collective
3810 
3811    Input Parameter:
3812 .  snes - iterative context obtained from SNESCreate()
3813 
3814    Output Parameters:
3815 .  a   - array to hold history
3816 .  its - integer array holds the number of linear iterations (or
3817          negative if not converged) for each solve.
3818 -  na  - size of a and its
3819 
3820    Notes:
3821     The calling sequence for this routine in Fortran is
3822 $   call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr)
3823 
3824    This routine is useful, e.g., when running a code for purposes
3825    of accurate performance monitoring, when no I/O should be done
3826    during the section of code that is being timed.
3827 
3828    Level: intermediate
3829 
3830 .keywords: SNES, get, convergence, history
3831 
3832 .seealso: SNESSetConvergencHistory()
3833 
3834 @*/
3835 PetscErrorCode  SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na)
3836 {
3837   PetscFunctionBegin;
3838   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3839   if (a)   *a   = snes->conv_hist;
3840   if (its) *its = snes->conv_hist_its;
3841   if (na)  *na  = snes->conv_hist_len;
3842   PetscFunctionReturn(0);
3843 }
3844 
3845 /*@C
3846   SNESSetUpdate - Sets the general-purpose update function called
3847   at the beginning of every iteration of the nonlinear solve. Specifically
3848   it is called just before the Jacobian is "evaluated".
3849 
3850   Logically Collective on SNES
3851 
3852   Input Parameters:
3853 . snes - The nonlinear solver context
3854 . func - The function
3855 
3856   Calling sequence of func:
3857 . func (SNES snes, PetscInt step);
3858 
3859 . step - The current step of the iteration
3860 
3861   Level: advanced
3862 
3863   Note: This is NOT what one uses to update the ghost points before a function evaluation, that should be done at the beginning of your FormFunction()
3864         This is not used by most users.
3865 
3866 .keywords: SNES, update
3867 
3868 .seealso SNESSetJacobian(), SNESSolve()
3869 @*/
3870 PetscErrorCode  SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt))
3871 {
3872   PetscFunctionBegin;
3873   PetscValidHeaderSpecific(snes, SNES_CLASSID,1);
3874   snes->ops->update = func;
3875   PetscFunctionReturn(0);
3876 }
3877 
3878 /*
3879    SNESScaleStep_Private - Scales a step so that its length is less than the
3880    positive parameter delta.
3881 
3882     Input Parameters:
3883 +   snes - the SNES context
3884 .   y - approximate solution of linear system
3885 .   fnorm - 2-norm of current function
3886 -   delta - trust region size
3887 
3888     Output Parameters:
3889 +   gpnorm - predicted function norm at the new point, assuming local
3890     linearization.  The value is zero if the step lies within the trust
3891     region, and exceeds zero otherwise.
3892 -   ynorm - 2-norm of the step
3893 
3894     Note:
3895     For non-trust region methods such as SNESNEWTONLS, the parameter delta
3896     is set to be the maximum allowable step size.
3897 
3898 .keywords: SNES, nonlinear, scale, step
3899 */
3900 PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm)
3901 {
3902   PetscReal      nrm;
3903   PetscScalar    cnorm;
3904   PetscErrorCode ierr;
3905 
3906   PetscFunctionBegin;
3907   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3908   PetscValidHeaderSpecific(y,VEC_CLASSID,2);
3909   PetscCheckSameComm(snes,1,y,2);
3910 
3911   ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr);
3912   if (nrm > *delta) {
3913     nrm     = *delta/nrm;
3914     *gpnorm = (1.0 - nrm)*(*fnorm);
3915     cnorm   = nrm;
3916     ierr    = VecScale(y,cnorm);CHKERRQ(ierr);
3917     *ynorm  = *delta;
3918   } else {
3919     *gpnorm = 0.0;
3920     *ynorm  = nrm;
3921   }
3922   PetscFunctionReturn(0);
3923 }
3924 
3925 /*@
3926    SNESReasonView - Displays the reason a SNES solve converged or diverged to a viewer
3927 
3928    Collective on SNES
3929 
3930    Parameter:
3931 +  snes - iterative context obtained from SNESCreate()
3932 -  viewer - the viewer to display the reason
3933 
3934 
3935    Options Database Keys:
3936 .  -snes_converged_reason - print reason for converged or diverged, also prints number of iterations
3937 
3938    Level: beginner
3939 
3940 .keywords: SNES, solve, linear system
3941 
3942 .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault()
3943 
3944 @*/
3945 PetscErrorCode  SNESReasonView(SNES snes,PetscViewer viewer)
3946 {
3947   PetscErrorCode ierr;
3948   PetscBool      isAscii;
3949 
3950   PetscFunctionBegin;
3951   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isAscii);CHKERRQ(ierr);
3952   if (isAscii) {
3953     ierr = PetscViewerASCIIAddTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr);
3954     if (snes->reason > 0) {
3955       if (((PetscObject) snes)->prefix) {
3956         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve converged due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
3957       } else {
3958         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve converged due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
3959       }
3960     } else {
3961       if (((PetscObject) snes)->prefix) {
3962         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve did not converge due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
3963       } else {
3964         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve did not converge due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
3965       }
3966     }
3967     ierr = PetscViewerASCIISubtractTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr);
3968   }
3969   PetscFunctionReturn(0);
3970 }
3971 
3972 /*@C
3973   SNESReasonViewFromOptions - Processes command line options to determine if/how a SNESReason is to be viewed.
3974 
3975   Collective on SNES
3976 
3977   Input Parameters:
3978 . snes   - the SNES object
3979 
3980   Level: intermediate
3981 
3982 @*/
3983 PetscErrorCode SNESReasonViewFromOptions(SNES snes)
3984 {
3985   PetscErrorCode    ierr;
3986   PetscViewer       viewer;
3987   PetscBool         flg;
3988   static PetscBool  incall = PETSC_FALSE;
3989   PetscViewerFormat format;
3990 
3991   PetscFunctionBegin;
3992   if (incall) PetscFunctionReturn(0);
3993   incall = PETSC_TRUE;
3994   ierr   = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_converged_reason",&viewer,&format,&flg);CHKERRQ(ierr);
3995   if (flg) {
3996     ierr = PetscViewerPushFormat(viewer,format);CHKERRQ(ierr);
3997     ierr = SNESReasonView(snes,viewer);CHKERRQ(ierr);
3998     ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
3999     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
4000   }
4001   incall = PETSC_FALSE;
4002   PetscFunctionReturn(0);
4003 }
4004 
4005 /*@C
4006    SNESSolve - Solves a nonlinear system F(x) = b.
4007    Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX().
4008 
4009    Collective on SNES
4010 
4011    Input Parameters:
4012 +  snes - the SNES context
4013 .  b - the constant part of the equation F(x) = b, or NULL to use zero.
4014 -  x - the solution vector.
4015 
4016    Notes:
4017    The user should initialize the vector,x, with the initial guess
4018    for the nonlinear solve prior to calling SNESSolve.  In particular,
4019    to employ an initial guess of zero, the user should explicitly set
4020    this vector to zero by calling VecSet().
4021 
4022    Level: beginner
4023 
4024 .keywords: SNES, nonlinear, solve
4025 
4026 .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetGridSequence(), SNESGetSolution()
4027 @*/
4028 PetscErrorCode  SNESSolve(SNES snes,Vec b,Vec x)
4029 {
4030   PetscErrorCode    ierr;
4031   PetscBool         flg;
4032   PetscInt          grid;
4033   Vec               xcreated = NULL;
4034   DM                dm;
4035 
4036   PetscFunctionBegin;
4037   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4038   if (x) PetscValidHeaderSpecific(x,VEC_CLASSID,3);
4039   if (x) PetscCheckSameComm(snes,1,x,3);
4040   if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2);
4041   if (b) PetscCheckSameComm(snes,1,b,2);
4042 
4043   {
4044     PetscViewer       viewer;
4045     PetscViewerFormat format;
4046     PetscBool         flg;
4047     static PetscBool  incall = PETSC_FALSE;
4048 
4049     if (!incall) {
4050       ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) snes), ((PetscObject) snes)->prefix, "-snes_convergence_estimate", &viewer, &format, &flg);CHKERRQ(ierr);
4051       if (flg) {
4052         PetscConvEst conv;
4053         PetscReal    alpha; /* Convergence rate of the solution error in the L_2 norm */
4054 
4055         incall = PETSC_TRUE;
4056         ierr = PetscConvEstCreate(PetscObjectComm((PetscObject) snes), &conv);CHKERRQ(ierr);
4057         ierr = PetscConvEstSetSolver(conv, snes);CHKERRQ(ierr);
4058         ierr = PetscConvEstSetFromOptions(conv);CHKERRQ(ierr);
4059         ierr = PetscConvEstSetUp(conv);CHKERRQ(ierr);
4060         ierr = PetscConvEstGetConvRate(conv, &alpha);CHKERRQ(ierr);
4061         ierr = PetscViewerPushFormat(viewer, format);CHKERRQ(ierr);
4062         ierr = PetscConvEstRateView(conv, alpha, viewer);CHKERRQ(ierr);
4063         ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
4064         ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
4065         ierr = PetscConvEstDestroy(&conv);CHKERRQ(ierr);
4066         incall = PETSC_FALSE;
4067       }
4068     }
4069   }
4070   if (!x) {
4071     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4072     ierr = DMCreateGlobalVector(dm,&xcreated);CHKERRQ(ierr);
4073     x    = xcreated;
4074   }
4075   ierr = SNESViewFromOptions(snes,NULL,"-snes_view_pre");CHKERRQ(ierr);
4076 
4077   for (grid=0; grid<snes->gridsequence; grid++) {ierr = PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);}
4078   for (grid=0; grid<snes->gridsequence+1; grid++) {
4079 
4080     /* set solution vector */
4081     if (!grid) {ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);}
4082     ierr          = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
4083     snes->vec_sol = x;
4084     ierr          = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4085 
4086     /* set affine vector if provided */
4087     if (b) { ierr = PetscObjectReference((PetscObject)b);CHKERRQ(ierr); }
4088     ierr          = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
4089     snes->vec_rhs = b;
4090 
4091     if (snes->vec_func == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector");
4092     if (snes->vec_rhs  == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be right hand side vector");
4093     if (!snes->vec_sol_update /* && snes->vec_sol */) {
4094       ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr);
4095       ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->vec_sol_update);CHKERRQ(ierr);
4096     }
4097     ierr = DMShellSetGlobalVector(dm,snes->vec_sol);CHKERRQ(ierr);
4098     ierr = SNESSetUp(snes);CHKERRQ(ierr);
4099 
4100     if (!grid) {
4101       if (snes->ops->computeinitialguess) {
4102         ierr = (*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP);CHKERRQ(ierr);
4103       }
4104     }
4105 
4106     if (snes->conv_hist_reset) snes->conv_hist_len = 0;
4107     if (snes->counters_reset) {snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;}
4108 
4109     ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
4110     ierr = (*snes->ops->solve)(snes);CHKERRQ(ierr);
4111     ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
4112     if (!snes->reason) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason");
4113     snes->domainerror = PETSC_FALSE; /* clear the flag if it has been set */
4114 
4115     if (snes->lagjac_persist) snes->jac_iter += snes->iter;
4116     if (snes->lagpre_persist) snes->pre_iter += snes->iter;
4117 
4118     ierr   = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_test_local_min",NULL,NULL,&flg);CHKERRQ(ierr);
4119     if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); }
4120     ierr = SNESReasonViewFromOptions(snes);CHKERRQ(ierr);
4121 
4122     if (snes->errorifnotconverged && snes->reason < 0) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged");
4123     if (snes->reason < 0) break;
4124     if (grid <  snes->gridsequence) {
4125       DM  fine;
4126       Vec xnew;
4127       Mat interp;
4128 
4129       ierr = DMRefine(snes->dm,PetscObjectComm((PetscObject)snes),&fine);CHKERRQ(ierr);
4130       if (!fine) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"DMRefine() did not perform any refinement, cannot continue grid sequencing");
4131       ierr = DMCreateInterpolation(snes->dm,fine,&interp,NULL);CHKERRQ(ierr);
4132       ierr = DMCreateGlobalVector(fine,&xnew);CHKERRQ(ierr);
4133       ierr = MatInterpolate(interp,x,xnew);CHKERRQ(ierr);
4134       ierr = DMInterpolate(snes->dm,interp,fine);CHKERRQ(ierr);
4135       ierr = MatDestroy(&interp);CHKERRQ(ierr);
4136       x    = xnew;
4137 
4138       ierr = SNESReset(snes);CHKERRQ(ierr);
4139       ierr = SNESSetDM(snes,fine);CHKERRQ(ierr);
4140       ierr = DMDestroy(&fine);CHKERRQ(ierr);
4141       ierr = PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);
4142     }
4143   }
4144   ierr = SNESViewFromOptions(snes,NULL,"-snes_view");CHKERRQ(ierr);
4145   ierr = VecViewFromOptions(snes->vec_sol,(PetscObject)snes,"-snes_view_solution");CHKERRQ(ierr);
4146 
4147   ierr = VecDestroy(&xcreated);CHKERRQ(ierr);
4148   ierr = PetscObjectSAWsBlock((PetscObject)snes);CHKERRQ(ierr);
4149   PetscFunctionReturn(0);
4150 }
4151 
4152 /* --------- Internal routines for SNES Package --------- */
4153 
4154 /*@C
4155    SNESSetType - Sets the method for the nonlinear solver.
4156 
4157    Collective on SNES
4158 
4159    Input Parameters:
4160 +  snes - the SNES context
4161 -  type - a known method
4162 
4163    Options Database Key:
4164 .  -snes_type <type> - Sets the method; use -help for a list
4165    of available methods (for instance, newtonls or newtontr)
4166 
4167    Notes:
4168    See "petsc/include/petscsnes.h" for available methods (for instance)
4169 +    SNESNEWTONLS - Newton's method with line search
4170      (systems of nonlinear equations)
4171 .    SNESNEWTONTR - Newton's method with trust region
4172      (systems of nonlinear equations)
4173 
4174   Normally, it is best to use the SNESSetFromOptions() command and then
4175   set the SNES solver type from the options database rather than by using
4176   this routine.  Using the options database provides the user with
4177   maximum flexibility in evaluating the many nonlinear solvers.
4178   The SNESSetType() routine is provided for those situations where it
4179   is necessary to set the nonlinear solver independently of the command
4180   line or options database.  This might be the case, for example, when
4181   the choice of solver changes during the execution of the program,
4182   and the user's application is taking responsibility for choosing the
4183   appropriate method.
4184 
4185     Developer Notes: SNESRegister() adds a constructor for a new SNESType to SNESList, SNESSetType() locates
4186     the constructor in that list and calls it to create the spexific object.
4187 
4188   Level: intermediate
4189 
4190 .keywords: SNES, set, type
4191 
4192 .seealso: SNESType, SNESCreate(), SNESDestroy(), SNESGetType(), SNESSetFromOptions()
4193 
4194 @*/
4195 PetscErrorCode  SNESSetType(SNES snes,SNESType type)
4196 {
4197   PetscErrorCode ierr,(*r)(SNES);
4198   PetscBool      match;
4199 
4200   PetscFunctionBegin;
4201   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4202   PetscValidCharPointer(type,2);
4203 
4204   ierr = PetscObjectTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr);
4205   if (match) PetscFunctionReturn(0);
4206 
4207   ierr =  PetscFunctionListFind(SNESList,type,&r);CHKERRQ(ierr);
4208   if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type);
4209   /* Destroy the previous private SNES context */
4210   if (snes->ops->destroy) {
4211     ierr               = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr);
4212     snes->ops->destroy = NULL;
4213   }
4214   /* Reinitialize function pointers in SNESOps structure */
4215   snes->ops->setup          = 0;
4216   snes->ops->solve          = 0;
4217   snes->ops->view           = 0;
4218   snes->ops->setfromoptions = 0;
4219   snes->ops->destroy        = 0;
4220   /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */
4221   snes->setupcalled = PETSC_FALSE;
4222 
4223   ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr);
4224   ierr = (*r)(snes);CHKERRQ(ierr);
4225   PetscFunctionReturn(0);
4226 }
4227 
4228 /*@C
4229    SNESGetType - Gets the SNES method type and name (as a string).
4230 
4231    Not Collective
4232 
4233    Input Parameter:
4234 .  snes - nonlinear solver context
4235 
4236    Output Parameter:
4237 .  type - SNES method (a character string)
4238 
4239    Level: intermediate
4240 
4241 .keywords: SNES, nonlinear, get, type, name
4242 @*/
4243 PetscErrorCode  SNESGetType(SNES snes,SNESType *type)
4244 {
4245   PetscFunctionBegin;
4246   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4247   PetscValidPointer(type,2);
4248   *type = ((PetscObject)snes)->type_name;
4249   PetscFunctionReturn(0);
4250 }
4251 
4252 /*@
4253   SNESSetSolution - Sets the solution vector for use by the SNES routines.
4254 
4255   Logically Collective on SNES and Vec
4256 
4257   Input Parameters:
4258 + snes - the SNES context obtained from SNESCreate()
4259 - u    - the solution vector
4260 
4261   Level: beginner
4262 
4263 .keywords: SNES, set, solution
4264 @*/
4265 PetscErrorCode SNESSetSolution(SNES snes, Vec u)
4266 {
4267   DM             dm;
4268   PetscErrorCode ierr;
4269 
4270   PetscFunctionBegin;
4271   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
4272   PetscValidHeaderSpecific(u, VEC_CLASSID, 2);
4273   ierr = PetscObjectReference((PetscObject) u);CHKERRQ(ierr);
4274   ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
4275 
4276   snes->vec_sol = u;
4277 
4278   ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
4279   ierr = DMShellSetGlobalVector(dm, u);CHKERRQ(ierr);
4280   PetscFunctionReturn(0);
4281 }
4282 
4283 /*@
4284    SNESGetSolution - Returns the vector where the approximate solution is
4285    stored. This is the fine grid solution when using SNESSetGridSequence().
4286 
4287    Not Collective, but Vec is parallel if SNES is parallel
4288 
4289    Input Parameter:
4290 .  snes - the SNES context
4291 
4292    Output Parameter:
4293 .  x - the solution
4294 
4295    Level: intermediate
4296 
4297 .keywords: SNES, nonlinear, get, solution
4298 
4299 .seealso:  SNESGetSolutionUpdate(), SNESGetFunction()
4300 @*/
4301 PetscErrorCode  SNESGetSolution(SNES snes,Vec *x)
4302 {
4303   PetscFunctionBegin;
4304   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4305   PetscValidPointer(x,2);
4306   *x = snes->vec_sol;
4307   PetscFunctionReturn(0);
4308 }
4309 
4310 /*@
4311    SNESGetSolutionUpdate - Returns the vector where the solution update is
4312    stored.
4313 
4314    Not Collective, but Vec is parallel if SNES is parallel
4315 
4316    Input Parameter:
4317 .  snes - the SNES context
4318 
4319    Output Parameter:
4320 .  x - the solution update
4321 
4322    Level: advanced
4323 
4324 .keywords: SNES, nonlinear, get, solution, update
4325 
4326 .seealso: SNESGetSolution(), SNESGetFunction()
4327 @*/
4328 PetscErrorCode  SNESGetSolutionUpdate(SNES snes,Vec *x)
4329 {
4330   PetscFunctionBegin;
4331   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4332   PetscValidPointer(x,2);
4333   *x = snes->vec_sol_update;
4334   PetscFunctionReturn(0);
4335 }
4336 
4337 /*@C
4338    SNESGetFunction - Returns the vector where the function is stored.
4339 
4340    Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet.
4341 
4342    Input Parameter:
4343 .  snes - the SNES context
4344 
4345    Output Parameter:
4346 +  r - the vector that is used to store residuals (or NULL if you don't want it)
4347 .  f - the function (or NULL if you don't want it); see SNESFunction for calling sequence details
4348 -  ctx - the function context (or NULL if you don't want it)
4349 
4350    Level: advanced
4351 
4352 .keywords: SNES, nonlinear, get, function
4353 
4354 .seealso: SNESSetFunction(), SNESGetSolution(), SNESFunction
4355 @*/
4356 PetscErrorCode  SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),void **ctx)
4357 {
4358   PetscErrorCode ierr;
4359   DM             dm;
4360 
4361   PetscFunctionBegin;
4362   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4363   if (r) {
4364     if (!snes->vec_func) {
4365       if (snes->vec_rhs) {
4366         ierr = VecDuplicate(snes->vec_rhs,&snes->vec_func);CHKERRQ(ierr);
4367       } else if (snes->vec_sol) {
4368         ierr = VecDuplicate(snes->vec_sol,&snes->vec_func);CHKERRQ(ierr);
4369       } else if (snes->dm) {
4370         ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr);
4371       }
4372     }
4373     *r = snes->vec_func;
4374   }
4375   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4376   ierr = DMSNESGetFunction(dm,f,ctx);CHKERRQ(ierr);
4377   PetscFunctionReturn(0);
4378 }
4379 
4380 /*@C
4381    SNESGetNGS - Returns the NGS function and context.
4382 
4383    Input Parameter:
4384 .  snes - the SNES context
4385 
4386    Output Parameter:
4387 +  f - the function (or NULL) see SNESNGSFunction for details
4388 -  ctx    - the function context (or NULL)
4389 
4390    Level: advanced
4391 
4392 .keywords: SNES, nonlinear, get, function
4393 
4394 .seealso: SNESSetNGS(), SNESGetFunction()
4395 @*/
4396 
4397 PetscErrorCode SNESGetNGS (SNES snes, PetscErrorCode (**f)(SNES, Vec, Vec, void*), void ** ctx)
4398 {
4399   PetscErrorCode ierr;
4400   DM             dm;
4401 
4402   PetscFunctionBegin;
4403   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4404   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4405   ierr = DMSNESGetNGS(dm,f,ctx);CHKERRQ(ierr);
4406   PetscFunctionReturn(0);
4407 }
4408 
4409 /*@C
4410    SNESSetOptionsPrefix - Sets the prefix used for searching for all
4411    SNES options in the database.
4412 
4413    Logically Collective on SNES
4414 
4415    Input Parameter:
4416 +  snes - the SNES context
4417 -  prefix - the prefix to prepend to all option names
4418 
4419    Notes:
4420    A hyphen (-) must NOT be given at the beginning of the prefix name.
4421    The first character of all runtime options is AUTOMATICALLY the hyphen.
4422 
4423    Level: advanced
4424 
4425 .keywords: SNES, set, options, prefix, database
4426 
4427 .seealso: SNESSetFromOptions()
4428 @*/
4429 PetscErrorCode  SNESSetOptionsPrefix(SNES snes,const char prefix[])
4430 {
4431   PetscErrorCode ierr;
4432 
4433   PetscFunctionBegin;
4434   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4435   ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
4436   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
4437   if (snes->linesearch) {
4438     ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr);
4439     ierr = PetscObjectSetOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr);
4440   }
4441   ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
4442   PetscFunctionReturn(0);
4443 }
4444 
4445 /*@C
4446    SNESAppendOptionsPrefix - Appends to the prefix used for searching for all
4447    SNES options in the database.
4448 
4449    Logically Collective on SNES
4450 
4451    Input Parameters:
4452 +  snes - the SNES context
4453 -  prefix - the prefix to prepend to all option names
4454 
4455    Notes:
4456    A hyphen (-) must NOT be given at the beginning of the prefix name.
4457    The first character of all runtime options is AUTOMATICALLY the hyphen.
4458 
4459    Level: advanced
4460 
4461 .keywords: SNES, append, options, prefix, database
4462 
4463 .seealso: SNESGetOptionsPrefix()
4464 @*/
4465 PetscErrorCode  SNESAppendOptionsPrefix(SNES snes,const char prefix[])
4466 {
4467   PetscErrorCode ierr;
4468 
4469   PetscFunctionBegin;
4470   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4471   ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
4472   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
4473   if (snes->linesearch) {
4474     ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr);
4475     ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr);
4476   }
4477   ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
4478   PetscFunctionReturn(0);
4479 }
4480 
4481 /*@C
4482    SNESGetOptionsPrefix - Sets the prefix used for searching for all
4483    SNES options in the database.
4484 
4485    Not Collective
4486 
4487    Input Parameter:
4488 .  snes - the SNES context
4489 
4490    Output Parameter:
4491 .  prefix - pointer to the prefix string used
4492 
4493    Notes: On the fortran side, the user should pass in a string 'prefix' of
4494    sufficient length to hold the prefix.
4495 
4496    Level: advanced
4497 
4498 .keywords: SNES, get, options, prefix, database
4499 
4500 .seealso: SNESAppendOptionsPrefix()
4501 @*/
4502 PetscErrorCode  SNESGetOptionsPrefix(SNES snes,const char *prefix[])
4503 {
4504   PetscErrorCode ierr;
4505 
4506   PetscFunctionBegin;
4507   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4508   ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
4509   PetscFunctionReturn(0);
4510 }
4511 
4512 
4513 /*@C
4514   SNESRegister - Adds a method to the nonlinear solver package.
4515 
4516    Not collective
4517 
4518    Input Parameters:
4519 +  name_solver - name of a new user-defined solver
4520 -  routine_create - routine to create method context
4521 
4522    Notes:
4523    SNESRegister() may be called multiple times to add several user-defined solvers.
4524 
4525    Sample usage:
4526 .vb
4527    SNESRegister("my_solver",MySolverCreate);
4528 .ve
4529 
4530    Then, your solver can be chosen with the procedural interface via
4531 $     SNESSetType(snes,"my_solver")
4532    or at runtime via the option
4533 $     -snes_type my_solver
4534 
4535    Level: advanced
4536 
4537     Note: If your function is not being put into a shared library then use SNESRegister() instead
4538 
4539 .keywords: SNES, nonlinear, register
4540 
4541 .seealso: SNESRegisterAll(), SNESRegisterDestroy()
4542 
4543   Level: advanced
4544 @*/
4545 PetscErrorCode  SNESRegister(const char sname[],PetscErrorCode (*function)(SNES))
4546 {
4547   PetscErrorCode ierr;
4548 
4549   PetscFunctionBegin;
4550   ierr = PetscFunctionListAdd(&SNESList,sname,function);CHKERRQ(ierr);
4551   PetscFunctionReturn(0);
4552 }
4553 
4554 PetscErrorCode  SNESTestLocalMin(SNES snes)
4555 {
4556   PetscErrorCode ierr;
4557   PetscInt       N,i,j;
4558   Vec            u,uh,fh;
4559   PetscScalar    value;
4560   PetscReal      norm;
4561 
4562   PetscFunctionBegin;
4563   ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr);
4564   ierr = VecDuplicate(u,&uh);CHKERRQ(ierr);
4565   ierr = VecDuplicate(u,&fh);CHKERRQ(ierr);
4566 
4567   /* currently only works for sequential */
4568   ierr = PetscPrintf(PETSC_COMM_WORLD,"Testing FormFunction() for local min\n");CHKERRQ(ierr);
4569   ierr = VecGetSize(u,&N);CHKERRQ(ierr);
4570   for (i=0; i<N; i++) {
4571     ierr = VecCopy(u,uh);CHKERRQ(ierr);
4572     ierr = PetscPrintf(PETSC_COMM_WORLD,"i = %D\n",i);CHKERRQ(ierr);
4573     for (j=-10; j<11; j++) {
4574       value = PetscSign(j)*PetscExpReal(PetscAbs(j)-10.0);
4575       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
4576       ierr  = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr);
4577       ierr  = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr);
4578       ierr  = PetscPrintf(PETSC_COMM_WORLD,"       j norm %D %18.16e\n",j,norm);CHKERRQ(ierr);
4579       value = -value;
4580       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
4581     }
4582   }
4583   ierr = VecDestroy(&uh);CHKERRQ(ierr);
4584   ierr = VecDestroy(&fh);CHKERRQ(ierr);
4585   PetscFunctionReturn(0);
4586 }
4587 
4588 /*@
4589    SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for
4590    computing relative tolerance for linear solvers within an inexact
4591    Newton method.
4592 
4593    Logically Collective on SNES
4594 
4595    Input Parameters:
4596 +  snes - SNES context
4597 -  flag - PETSC_TRUE or PETSC_FALSE
4598 
4599     Options Database:
4600 +  -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence
4601 .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
4602 .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
4603 .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
4604 .  -snes_ksp_ew_gamma <gamma> - Sets gamma
4605 .  -snes_ksp_ew_alpha <alpha> - Sets alpha
4606 .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
4607 -  -snes_ksp_ew_threshold <threshold> - Sets threshold
4608 
4609    Notes:
4610    Currently, the default is to use a constant relative tolerance for
4611    the inner linear solvers.  Alternatively, one can use the
4612    Eisenstat-Walker method, where the relative convergence tolerance
4613    is reset at each Newton iteration according progress of the nonlinear
4614    solver.
4615 
4616    Level: advanced
4617 
4618    Reference:
4619    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
4620    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
4621 
4622 .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
4623 
4624 .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
4625 @*/
4626 PetscErrorCode  SNESKSPSetUseEW(SNES snes,PetscBool flag)
4627 {
4628   PetscFunctionBegin;
4629   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4630   PetscValidLogicalCollectiveBool(snes,flag,2);
4631   snes->ksp_ewconv = flag;
4632   PetscFunctionReturn(0);
4633 }
4634 
4635 /*@
4636    SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method
4637    for computing relative tolerance for linear solvers within an
4638    inexact Newton method.
4639 
4640    Not Collective
4641 
4642    Input Parameter:
4643 .  snes - SNES context
4644 
4645    Output Parameter:
4646 .  flag - PETSC_TRUE or PETSC_FALSE
4647 
4648    Notes:
4649    Currently, the default is to use a constant relative tolerance for
4650    the inner linear solvers.  Alternatively, one can use the
4651    Eisenstat-Walker method, where the relative convergence tolerance
4652    is reset at each Newton iteration according progress of the nonlinear
4653    solver.
4654 
4655    Level: advanced
4656 
4657    Reference:
4658    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
4659    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
4660 
4661 .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
4662 
4663 .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
4664 @*/
4665 PetscErrorCode  SNESKSPGetUseEW(SNES snes, PetscBool  *flag)
4666 {
4667   PetscFunctionBegin;
4668   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4669   PetscValidPointer(flag,2);
4670   *flag = snes->ksp_ewconv;
4671   PetscFunctionReturn(0);
4672 }
4673 
4674 /*@
4675    SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker
4676    convergence criteria for the linear solvers within an inexact
4677    Newton method.
4678 
4679    Logically Collective on SNES
4680 
4681    Input Parameters:
4682 +    snes - SNES context
4683 .    version - version 1, 2 (default is 2) or 3
4684 .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
4685 .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
4686 .    gamma - multiplicative factor for version 2 rtol computation
4687              (0 <= gamma2 <= 1)
4688 .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
4689 .    alpha2 - power for safeguard
4690 -    threshold - threshold for imposing safeguard (0 < threshold < 1)
4691 
4692    Note:
4693    Version 3 was contributed by Luis Chacon, June 2006.
4694 
4695    Use PETSC_DEFAULT to retain the default for any of the parameters.
4696 
4697    Level: advanced
4698 
4699    Reference:
4700    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
4701    inexact Newton method", Utah State University Math. Stat. Dept. Res.
4702    Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput.
4703 
4704 .keywords: SNES, KSP, Eisenstat, Walker, set, parameters
4705 
4706 .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW()
4707 @*/
4708 PetscErrorCode  SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold)
4709 {
4710   SNESKSPEW *kctx;
4711 
4712   PetscFunctionBegin;
4713   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4714   kctx = (SNESKSPEW*)snes->kspconvctx;
4715   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
4716   PetscValidLogicalCollectiveInt(snes,version,2);
4717   PetscValidLogicalCollectiveReal(snes,rtol_0,3);
4718   PetscValidLogicalCollectiveReal(snes,rtol_max,4);
4719   PetscValidLogicalCollectiveReal(snes,gamma,5);
4720   PetscValidLogicalCollectiveReal(snes,alpha,6);
4721   PetscValidLogicalCollectiveReal(snes,alpha2,7);
4722   PetscValidLogicalCollectiveReal(snes,threshold,8);
4723 
4724   if (version != PETSC_DEFAULT)   kctx->version   = version;
4725   if (rtol_0 != PETSC_DEFAULT)    kctx->rtol_0    = rtol_0;
4726   if (rtol_max != PETSC_DEFAULT)  kctx->rtol_max  = rtol_max;
4727   if (gamma != PETSC_DEFAULT)     kctx->gamma     = gamma;
4728   if (alpha != PETSC_DEFAULT)     kctx->alpha     = alpha;
4729   if (alpha2 != PETSC_DEFAULT)    kctx->alpha2    = alpha2;
4730   if (threshold != PETSC_DEFAULT) kctx->threshold = threshold;
4731 
4732   if (kctx->version < 1 || kctx->version > 3) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 and 3 are supported: %D",kctx->version);
4733   if (kctx->rtol_0 < 0.0 || kctx->rtol_0 >= 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_0 < 1.0: %g",(double)kctx->rtol_0);
4734   if (kctx->rtol_max < 0.0 || kctx->rtol_max >= 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_max (%g) < 1.0\n",(double)kctx->rtol_max);
4735   if (kctx->gamma < 0.0 || kctx->gamma > 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= gamma (%g) <= 1.0\n",(double)kctx->gamma);
4736   if (kctx->alpha <= 1.0 || kctx->alpha > 2.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"1.0 < alpha (%g) <= 2.0\n",(double)kctx->alpha);
4737   if (kctx->threshold <= 0.0 || kctx->threshold >= 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 < threshold (%g) < 1.0\n",(double)kctx->threshold);
4738   PetscFunctionReturn(0);
4739 }
4740 
4741 /*@
4742    SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker
4743    convergence criteria for the linear solvers within an inexact
4744    Newton method.
4745 
4746    Not Collective
4747 
4748    Input Parameters:
4749      snes - SNES context
4750 
4751    Output Parameters:
4752 +    version - version 1, 2 (default is 2) or 3
4753 .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
4754 .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
4755 .    gamma - multiplicative factor for version 2 rtol computation (0 <= gamma2 <= 1)
4756 .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
4757 .    alpha2 - power for safeguard
4758 -    threshold - threshold for imposing safeguard (0 < threshold < 1)
4759 
4760    Level: advanced
4761 
4762 .keywords: SNES, KSP, Eisenstat, Walker, get, parameters
4763 
4764 .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW()
4765 @*/
4766 PetscErrorCode  SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold)
4767 {
4768   SNESKSPEW *kctx;
4769 
4770   PetscFunctionBegin;
4771   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4772   kctx = (SNESKSPEW*)snes->kspconvctx;
4773   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
4774   if (version)   *version   = kctx->version;
4775   if (rtol_0)    *rtol_0    = kctx->rtol_0;
4776   if (rtol_max)  *rtol_max  = kctx->rtol_max;
4777   if (gamma)     *gamma     = kctx->gamma;
4778   if (alpha)     *alpha     = kctx->alpha;
4779   if (alpha2)    *alpha2    = kctx->alpha2;
4780   if (threshold) *threshold = kctx->threshold;
4781   PetscFunctionReturn(0);
4782 }
4783 
4784  PetscErrorCode KSPPreSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes)
4785 {
4786   PetscErrorCode ierr;
4787   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
4788   PetscReal      rtol  = PETSC_DEFAULT,stol;
4789 
4790   PetscFunctionBegin;
4791   if (!snes->ksp_ewconv) PetscFunctionReturn(0);
4792   if (!snes->iter) {
4793     rtol = kctx->rtol_0; /* first time in, so use the original user rtol */
4794     ierr = VecNorm(snes->vec_func,NORM_2,&kctx->norm_first);CHKERRQ(ierr);
4795   }
4796   else {
4797     if (kctx->version == 1) {
4798       rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last;
4799       if (rtol < 0.0) rtol = -rtol;
4800       stol = PetscPowReal(kctx->rtol_last,kctx->alpha2);
4801       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
4802     } else if (kctx->version == 2) {
4803       rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha);
4804       stol = kctx->gamma * PetscPowReal(kctx->rtol_last,kctx->alpha);
4805       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
4806     } else if (kctx->version == 3) { /* contributed by Luis Chacon, June 2006. */
4807       rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha);
4808       /* safeguard: avoid sharp decrease of rtol */
4809       stol = kctx->gamma*PetscPowReal(kctx->rtol_last,kctx->alpha);
4810       stol = PetscMax(rtol,stol);
4811       rtol = PetscMin(kctx->rtol_0,stol);
4812       /* safeguard: avoid oversolving */
4813       stol = kctx->gamma*(kctx->norm_first*snes->rtol)/snes->norm;
4814       stol = PetscMax(rtol,stol);
4815       rtol = PetscMin(kctx->rtol_0,stol);
4816     } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version);
4817   }
4818   /* safeguard: avoid rtol greater than one */
4819   rtol = PetscMin(rtol,kctx->rtol_max);
4820   ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr);
4821   ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%g\n",snes->iter,kctx->version,(double)rtol);CHKERRQ(ierr);
4822   PetscFunctionReturn(0);
4823 }
4824 
4825 PetscErrorCode KSPPostSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes)
4826 {
4827   PetscErrorCode ierr;
4828   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
4829   PCSide         pcside;
4830   Vec            lres;
4831 
4832   PetscFunctionBegin;
4833   if (!snes->ksp_ewconv) PetscFunctionReturn(0);
4834   ierr = KSPGetTolerances(ksp,&kctx->rtol_last,0,0,0);CHKERRQ(ierr);
4835   kctx->norm_last = snes->norm;
4836   if (kctx->version == 1) {
4837     PC        pc;
4838     PetscBool isNone;
4839 
4840     ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr);
4841     ierr = PetscObjectTypeCompare((PetscObject) pc, PCNONE, &isNone);CHKERRQ(ierr);
4842     ierr = KSPGetPCSide(ksp,&pcside);CHKERRQ(ierr);
4843      if (pcside == PC_RIGHT || isNone) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */
4844       /* KSP residual is true linear residual */
4845       ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr);
4846     } else {
4847       /* KSP residual is preconditioned residual */
4848       /* compute true linear residual norm */
4849       ierr = VecDuplicate(b,&lres);CHKERRQ(ierr);
4850       ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr);
4851       ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr);
4852       ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr);
4853       ierr = VecDestroy(&lres);CHKERRQ(ierr);
4854     }
4855   }
4856   PetscFunctionReturn(0);
4857 }
4858 
4859 /*@
4860    SNESGetKSP - Returns the KSP context for a SNES solver.
4861 
4862    Not Collective, but if SNES object is parallel, then KSP object is parallel
4863 
4864    Input Parameter:
4865 .  snes - the SNES context
4866 
4867    Output Parameter:
4868 .  ksp - the KSP context
4869 
4870    Notes:
4871    The user can then directly manipulate the KSP context to set various
4872    options, etc.  Likewise, the user can then extract and manipulate the
4873    PC contexts as well.
4874 
4875    Level: beginner
4876 
4877 .keywords: SNES, nonlinear, get, KSP, context
4878 
4879 .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
4880 @*/
4881 PetscErrorCode  SNESGetKSP(SNES snes,KSP *ksp)
4882 {
4883   PetscErrorCode ierr;
4884 
4885   PetscFunctionBegin;
4886   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4887   PetscValidPointer(ksp,2);
4888 
4889   if (!snes->ksp) {
4890     PetscBool monitor = PETSC_FALSE;
4891 
4892     ierr = KSPCreate(PetscObjectComm((PetscObject)snes),&snes->ksp);CHKERRQ(ierr);
4893     ierr = PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1);CHKERRQ(ierr);
4894     ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->ksp);CHKERRQ(ierr);
4895 
4896     ierr = KSPSetPreSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_SNESEW,snes);CHKERRQ(ierr);
4897     ierr = KSPSetPostSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_SNESEW,snes);CHKERRQ(ierr);
4898 
4899     ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes",&monitor,NULL);CHKERRQ(ierr);
4900     if (monitor) {
4901       ierr = KSPMonitorSet(snes->ksp,KSPMonitorSNES,snes,NULL);CHKERRQ(ierr);
4902     }
4903     monitor = PETSC_FALSE;
4904     ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes_lg",&monitor,NULL);CHKERRQ(ierr);
4905     if (monitor) {
4906       PetscObject *objs;
4907       ierr = KSPMonitorSNESLGResidualNormCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,600,600,&objs);CHKERRQ(ierr);
4908       objs[0] = (PetscObject) snes;
4909       ierr = KSPMonitorSet(snes->ksp,(PetscErrorCode (*)(KSP,PetscInt,PetscReal,void*))KSPMonitorSNESLGResidualNorm,objs,(PetscErrorCode (*)(void**))KSPMonitorSNESLGResidualNormDestroy);CHKERRQ(ierr);
4910     }
4911   }
4912   *ksp = snes->ksp;
4913   PetscFunctionReturn(0);
4914 }
4915 
4916 
4917 #include <petsc/private/dmimpl.h>
4918 /*@
4919    SNESSetDM - Sets the DM that may be used by some nonlinear solvers or their underlying preconditioners
4920 
4921    Logically Collective on SNES
4922 
4923    Input Parameters:
4924 +  snes - the nonlinear solver context
4925 -  dm - the dm, cannot be NULL
4926 
4927    Level: intermediate
4928 
4929 .seealso: SNESGetDM(), SNESHasDM(), KSPSetDM(), KSPGetDM()
4930 @*/
4931 PetscErrorCode  SNESSetDM(SNES snes,DM dm)
4932 {
4933   PetscErrorCode ierr;
4934   KSP            ksp;
4935   DMSNES         sdm;
4936 
4937   PetscFunctionBegin;
4938   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4939   PetscValidHeaderSpecific(dm,DM_CLASSID,2);
4940   ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr);
4941   if (snes->dm) {               /* Move the DMSNES context over to the new DM unless the new DM already has one */
4942     if (snes->dm->dmsnes && !dm->dmsnes) {
4943       ierr = DMCopyDMSNES(snes->dm,dm);CHKERRQ(ierr);
4944       ierr = DMGetDMSNES(snes->dm,&sdm);CHKERRQ(ierr);
4945       if (sdm->originaldm == snes->dm) sdm->originaldm = dm; /* Grant write privileges to the replacement DM */
4946     }
4947     ierr = DMDestroy(&snes->dm);CHKERRQ(ierr);
4948   }
4949   snes->dm     = dm;
4950   snes->dmAuto = PETSC_FALSE;
4951 
4952   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
4953   ierr = KSPSetDM(ksp,dm);CHKERRQ(ierr);
4954   ierr = KSPSetDMActive(ksp,PETSC_FALSE);CHKERRQ(ierr);
4955   if (snes->npc) {
4956     ierr = SNESSetDM(snes->npc, snes->dm);CHKERRQ(ierr);
4957     ierr = SNESSetNPCSide(snes,snes->npcside);CHKERRQ(ierr);
4958   }
4959   PetscFunctionReturn(0);
4960 }
4961 
4962 /*@
4963    SNESGetDM - Gets the DM that may be used by some preconditioners
4964 
4965    Not Collective but DM obtained is parallel on SNES
4966 
4967    Input Parameter:
4968 . snes - the preconditioner context
4969 
4970    Output Parameter:
4971 .  dm - the dm
4972 
4973    Level: intermediate
4974 
4975 .seealso: SNESSetDM(), SNESHasDM(), KSPSetDM(), KSPGetDM()
4976 @*/
4977 PetscErrorCode  SNESGetDM(SNES snes,DM *dm)
4978 {
4979   PetscErrorCode ierr;
4980 
4981   PetscFunctionBegin;
4982   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4983   if (!snes->dm) {
4984     ierr         = DMShellCreate(PetscObjectComm((PetscObject)snes),&snes->dm);CHKERRQ(ierr);
4985     snes->dmAuto = PETSC_TRUE;
4986   }
4987   *dm = snes->dm;
4988   PetscFunctionReturn(0);
4989 }
4990 
4991 
4992 /*@
4993    SNESHasDM - Whether snes has dm
4994 
4995    Not collective but all processes must return the same value
4996 
4997    Input Parameter:
4998 . snes - the nonlinear solver object
4999 
5000    Output Parameter:
5001 .  hasdm - a flag indicates whether there is dm in snes
5002 
5003    Level: intermediate
5004 
5005 .seealso: SNESGetDM(), SNESSetDM(), KSPSetDM(), KSPGetDM()
5006 @*/
5007 PetscErrorCode  SNESHasDM(SNES snes,PetscBool *hasdm)
5008 {
5009   PetscFunctionBegin;
5010   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5011   PetscValidPointer(hasdm,2);
5012   if (snes->dm) *hasdm = PETSC_TRUE;
5013   else *hasdm = PETSC_FALSE;
5014   PetscFunctionReturn(0);
5015 }
5016 
5017 /*@
5018   SNESSetNPC - Sets the nonlinear preconditioner to be used.
5019 
5020   Collective on SNES
5021 
5022   Input Parameters:
5023 + snes - iterative context obtained from SNESCreate()
5024 - pc   - the preconditioner object
5025 
5026   Notes:
5027   Use SNESGetNPC() to retrieve the preconditioner context (for example,
5028   to configure it using the API).
5029 
5030   Level: developer
5031 
5032 .keywords: SNES, set, precondition
5033 .seealso: SNESGetNPC(), SNESHasNPC()
5034 @*/
5035 PetscErrorCode SNESSetNPC(SNES snes, SNES pc)
5036 {
5037   PetscErrorCode ierr;
5038 
5039   PetscFunctionBegin;
5040   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5041   PetscValidHeaderSpecific(pc, SNES_CLASSID, 2);
5042   PetscCheckSameComm(snes, 1, pc, 2);
5043   ierr     = PetscObjectReference((PetscObject) pc);CHKERRQ(ierr);
5044   ierr     = SNESDestroy(&snes->npc);CHKERRQ(ierr);
5045   snes->npc = pc;
5046   ierr     = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->npc);CHKERRQ(ierr);
5047   PetscFunctionReturn(0);
5048 }
5049 
5050 /*@
5051   SNESGetNPC - Creates a nonlinear preconditioning solver (SNES) to be used to precondition the nonlinear solver.
5052 
5053   Not Collective
5054 
5055   Input Parameter:
5056 . snes - iterative context obtained from SNESCreate()
5057 
5058   Output Parameter:
5059 . pc - preconditioner context
5060 
5061   Notes: If a SNES was previously set with SNESSetNPC() then that SNES is returned.
5062 
5063   Level: developer
5064 
5065 .keywords: SNES, get, preconditioner
5066 .seealso: SNESSetNPC(), SNESHasNPC()
5067 @*/
5068 PetscErrorCode SNESGetNPC(SNES snes, SNES *pc)
5069 {
5070   PetscErrorCode ierr;
5071   const char     *optionsprefix;
5072 
5073   PetscFunctionBegin;
5074   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5075   PetscValidPointer(pc, 2);
5076   if (!snes->npc) {
5077     ierr = SNESCreate(PetscObjectComm((PetscObject)snes),&snes->npc);CHKERRQ(ierr);
5078     ierr = PetscObjectIncrementTabLevel((PetscObject)snes->npc,(PetscObject)snes,1);CHKERRQ(ierr);
5079     ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr);
5080     ierr = SNESGetOptionsPrefix(snes,&optionsprefix);CHKERRQ(ierr);
5081     ierr = SNESSetOptionsPrefix(snes->npc,optionsprefix);CHKERRQ(ierr);
5082     ierr = SNESAppendOptionsPrefix(snes->npc,"npc_");CHKERRQ(ierr);
5083     ierr = SNESSetCountersReset(snes->npc,PETSC_FALSE);CHKERRQ(ierr);
5084   }
5085   *pc = snes->npc;
5086   PetscFunctionReturn(0);
5087 }
5088 
5089 /*@
5090   SNESHasNPC - Returns whether a nonlinear preconditioner exists
5091 
5092   Not Collective
5093 
5094   Input Parameter:
5095 . snes - iterative context obtained from SNESCreate()
5096 
5097   Output Parameter:
5098 . has_npc - whether the SNES has an NPC or not
5099 
5100   Level: developer
5101 
5102 .keywords: SNES, has, preconditioner
5103 .seealso: SNESSetNPC(), SNESGetNPC()
5104 @*/
5105 PetscErrorCode SNESHasNPC(SNES snes, PetscBool *has_npc)
5106 {
5107   PetscFunctionBegin;
5108   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5109   *has_npc = (PetscBool) (snes->npc ? PETSC_TRUE : PETSC_FALSE);
5110   PetscFunctionReturn(0);
5111 }
5112 
5113 /*@
5114     SNESSetNPCSide - Sets the preconditioning side.
5115 
5116     Logically Collective on SNES
5117 
5118     Input Parameter:
5119 .   snes - iterative context obtained from SNESCreate()
5120 
5121     Output Parameter:
5122 .   side - the preconditioning side, where side is one of
5123 .vb
5124       PC_LEFT - left preconditioning
5125       PC_RIGHT - right preconditioning (default for most nonlinear solvers)
5126 .ve
5127 
5128     Options Database Keys:
5129 .   -snes_pc_side <right,left>
5130 
5131     Notes: SNESNRICHARDSON and SNESNCG only support left preconditioning.
5132 
5133     Level: intermediate
5134 
5135 .keywords: SNES, set, right, left, side, preconditioner, flag
5136 
5137 .seealso: SNESGetNPCSide(), KSPSetPCSide()
5138 @*/
5139 PetscErrorCode  SNESSetNPCSide(SNES snes,PCSide side)
5140 {
5141   PetscFunctionBegin;
5142   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5143   PetscValidLogicalCollectiveEnum(snes,side,2);
5144   snes->npcside= side;
5145   PetscFunctionReturn(0);
5146 }
5147 
5148 /*@
5149     SNESGetNPCSide - Gets the preconditioning side.
5150 
5151     Not Collective
5152 
5153     Input Parameter:
5154 .   snes - iterative context obtained from SNESCreate()
5155 
5156     Output Parameter:
5157 .   side - the preconditioning side, where side is one of
5158 .vb
5159       PC_LEFT - left preconditioning
5160       PC_RIGHT - right preconditioning (default for most nonlinear solvers)
5161 .ve
5162 
5163     Level: intermediate
5164 
5165 .keywords: SNES, get, right, left, side, preconditioner, flag
5166 
5167 .seealso: SNESSetNPCSide(), KSPGetPCSide()
5168 @*/
5169 PetscErrorCode  SNESGetNPCSide(SNES snes,PCSide *side)
5170 {
5171   PetscFunctionBegin;
5172   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5173   PetscValidPointer(side,2);
5174   *side = snes->npcside;
5175   PetscFunctionReturn(0);
5176 }
5177 
5178 /*@
5179   SNESSetLineSearch - Sets the linesearch on the SNES instance.
5180 
5181   Collective on SNES
5182 
5183   Input Parameters:
5184 + snes - iterative context obtained from SNESCreate()
5185 - linesearch   - the linesearch object
5186 
5187   Notes:
5188   Use SNESGetLineSearch() to retrieve the preconditioner context (for example,
5189   to configure it using the API).
5190 
5191   Level: developer
5192 
5193 .keywords: SNES, set, linesearch
5194 .seealso: SNESGetLineSearch()
5195 @*/
5196 PetscErrorCode SNESSetLineSearch(SNES snes, SNESLineSearch linesearch)
5197 {
5198   PetscErrorCode ierr;
5199 
5200   PetscFunctionBegin;
5201   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5202   PetscValidHeaderSpecific(linesearch, SNESLINESEARCH_CLASSID, 2);
5203   PetscCheckSameComm(snes, 1, linesearch, 2);
5204   ierr = PetscObjectReference((PetscObject) linesearch);CHKERRQ(ierr);
5205   ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr);
5206 
5207   snes->linesearch = linesearch;
5208 
5209   ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr);
5210   PetscFunctionReturn(0);
5211 }
5212 
5213 /*@
5214   SNESGetLineSearch - Returns a pointer to the line search context set with SNESSetLineSearch()
5215   or creates a default line search instance associated with the SNES and returns it.
5216 
5217   Not Collective
5218 
5219   Input Parameter:
5220 . snes - iterative context obtained from SNESCreate()
5221 
5222   Output Parameter:
5223 . linesearch - linesearch context
5224 
5225   Level: beginner
5226 
5227 .keywords: SNES, get, linesearch
5228 .seealso: SNESSetLineSearch(), SNESLineSearchCreate()
5229 @*/
5230 PetscErrorCode SNESGetLineSearch(SNES snes, SNESLineSearch *linesearch)
5231 {
5232   PetscErrorCode ierr;
5233   const char     *optionsprefix;
5234 
5235   PetscFunctionBegin;
5236   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5237   PetscValidPointer(linesearch, 2);
5238   if (!snes->linesearch) {
5239     ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr);
5240     ierr = SNESLineSearchCreate(PetscObjectComm((PetscObject)snes), &snes->linesearch);CHKERRQ(ierr);
5241     ierr = SNESLineSearchSetSNES(snes->linesearch, snes);CHKERRQ(ierr);
5242     ierr = SNESLineSearchAppendOptionsPrefix(snes->linesearch, optionsprefix);CHKERRQ(ierr);
5243     ierr = PetscObjectIncrementTabLevel((PetscObject) snes->linesearch, (PetscObject) snes, 1);CHKERRQ(ierr);
5244     ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr);
5245   }
5246   *linesearch = snes->linesearch;
5247   PetscFunctionReturn(0);
5248 }
5249 
5250 #if defined(PETSC_HAVE_MATLAB_ENGINE)
5251 #include <mex.h>
5252 
5253 typedef struct {char *funcname; mxArray *ctx;} SNESMatlabContext;
5254 
5255 /*
5256    SNESComputeFunction_Matlab - Calls the function that has been set with SNESSetFunctionMatlab().
5257 
5258    Collective on SNES
5259 
5260    Input Parameters:
5261 +  snes - the SNES context
5262 -  x - input vector
5263 
5264    Output Parameter:
5265 .  y - function vector, as set by SNESSetFunction()
5266 
5267    Notes:
5268    SNESComputeFunction() is typically used within nonlinear solvers
5269    implementations, so most users would not generally call this routine
5270    themselves.
5271 
5272    Level: developer
5273 
5274 .keywords: SNES, nonlinear, compute, function
5275 
5276 .seealso: SNESSetFunction(), SNESGetFunction()
5277 */
5278 PetscErrorCode  SNESComputeFunction_Matlab(SNES snes,Vec x,Vec y, void *ctx)
5279 {
5280   PetscErrorCode    ierr;
5281   SNESMatlabContext *sctx = (SNESMatlabContext*)ctx;
5282   int               nlhs  = 1,nrhs = 5;
5283   mxArray           *plhs[1],*prhs[5];
5284   long long int     lx = 0,ly = 0,ls = 0;
5285 
5286   PetscFunctionBegin;
5287   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5288   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
5289   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
5290   PetscCheckSameComm(snes,1,x,2);
5291   PetscCheckSameComm(snes,1,y,3);
5292 
5293   /* call Matlab function in ctx with arguments x and y */
5294 
5295   ierr    = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr);
5296   ierr    = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr);
5297   ierr    = PetscMemcpy(&ly,&y,sizeof(x));CHKERRQ(ierr);
5298   prhs[0] = mxCreateDoubleScalar((double)ls);
5299   prhs[1] = mxCreateDoubleScalar((double)lx);
5300   prhs[2] = mxCreateDoubleScalar((double)ly);
5301   prhs[3] = mxCreateString(sctx->funcname);
5302   prhs[4] = sctx->ctx;
5303   ierr    = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeFunctionInternal");CHKERRQ(ierr);
5304   ierr    = mxGetScalar(plhs[0]);CHKERRQ(ierr);
5305   mxDestroyArray(prhs[0]);
5306   mxDestroyArray(prhs[1]);
5307   mxDestroyArray(prhs[2]);
5308   mxDestroyArray(prhs[3]);
5309   mxDestroyArray(plhs[0]);
5310   PetscFunctionReturn(0);
5311 }
5312 
5313 /*
5314    SNESSetFunctionMatlab - Sets the function evaluation routine and function
5315    vector for use by the SNES routines in solving systems of nonlinear
5316    equations from MATLAB. Here the function is a string containing the name of a MATLAB function
5317 
5318    Logically Collective on SNES
5319 
5320    Input Parameters:
5321 +  snes - the SNES context
5322 .  r - vector to store function value
5323 -  f - function evaluation routine
5324 
5325    Notes:
5326    The Newton-like methods typically solve linear systems of the form
5327 $      f'(x) x = -f(x),
5328    where f'(x) denotes the Jacobian matrix and f(x) is the function.
5329 
5330    Level: beginner
5331 
5332    Developer Note:  This bleeds the allocated memory SNESMatlabContext *sctx;
5333 
5334 .keywords: SNES, nonlinear, set, function
5335 
5336 .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
5337 */
5338 PetscErrorCode  SNESSetFunctionMatlab(SNES snes,Vec r,const char *f,mxArray *ctx)
5339 {
5340   PetscErrorCode    ierr;
5341   SNESMatlabContext *sctx;
5342 
5343   PetscFunctionBegin;
5344   /* currently sctx is memory bleed */
5345   ierr = PetscNew(&sctx);CHKERRQ(ierr);
5346   ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr);
5347   /*
5348      This should work, but it doesn't
5349   sctx->ctx = ctx;
5350   mexMakeArrayPersistent(sctx->ctx);
5351   */
5352   sctx->ctx = mxDuplicateArray(ctx);
5353   ierr      = SNESSetFunction(snes,r,SNESComputeFunction_Matlab,sctx);CHKERRQ(ierr);
5354   PetscFunctionReturn(0);
5355 }
5356 
5357 /*
5358    SNESComputeJacobian_Matlab - Calls the function that has been set with SNESSetJacobianMatlab().
5359 
5360    Collective on SNES
5361 
5362    Input Parameters:
5363 +  snes - the SNES context
5364 .  x - input vector
5365 .  A, B - the matrices
5366 -  ctx - user context
5367 
5368    Level: developer
5369 
5370 .keywords: SNES, nonlinear, compute, function
5371 
5372 .seealso: SNESSetFunction(), SNESGetFunction()
5373 @*/
5374 PetscErrorCode  SNESComputeJacobian_Matlab(SNES snes,Vec x,Mat A,Mat B,void *ctx)
5375 {
5376   PetscErrorCode    ierr;
5377   SNESMatlabContext *sctx = (SNESMatlabContext*)ctx;
5378   int               nlhs  = 2,nrhs = 6;
5379   mxArray           *plhs[2],*prhs[6];
5380   long long int     lx = 0,lA = 0,ls = 0, lB = 0;
5381 
5382   PetscFunctionBegin;
5383   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5384   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
5385 
5386   /* call Matlab function in ctx with arguments x and y */
5387 
5388   ierr    = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr);
5389   ierr    = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr);
5390   ierr    = PetscMemcpy(&lA,A,sizeof(x));CHKERRQ(ierr);
5391   ierr    = PetscMemcpy(&lB,B,sizeof(x));CHKERRQ(ierr);
5392   prhs[0] = mxCreateDoubleScalar((double)ls);
5393   prhs[1] = mxCreateDoubleScalar((double)lx);
5394   prhs[2] = mxCreateDoubleScalar((double)lA);
5395   prhs[3] = mxCreateDoubleScalar((double)lB);
5396   prhs[4] = mxCreateString(sctx->funcname);
5397   prhs[5] = sctx->ctx;
5398   ierr    = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeJacobianInternal");CHKERRQ(ierr);
5399   ierr    = mxGetScalar(plhs[0]);CHKERRQ(ierr);
5400   mxDestroyArray(prhs[0]);
5401   mxDestroyArray(prhs[1]);
5402   mxDestroyArray(prhs[2]);
5403   mxDestroyArray(prhs[3]);
5404   mxDestroyArray(prhs[4]);
5405   mxDestroyArray(plhs[0]);
5406   mxDestroyArray(plhs[1]);
5407   PetscFunctionReturn(0);
5408 }
5409 
5410 /*
5411    SNESSetJacobianMatlab - Sets the Jacobian function evaluation routine and two empty Jacobian matrices
5412    vector for use by the SNES routines in solving systems of nonlinear
5413    equations from MATLAB. Here the function is a string containing the name of a MATLAB function
5414 
5415    Logically Collective on SNES
5416 
5417    Input Parameters:
5418 +  snes - the SNES context
5419 .  A,B - Jacobian matrices
5420 .  J - function evaluation routine
5421 -  ctx - user context
5422 
5423    Level: developer
5424 
5425    Developer Note:  This bleeds the allocated memory SNESMatlabContext *sctx;
5426 
5427 .keywords: SNES, nonlinear, set, function
5428 
5429 .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction(), J
5430 */
5431 PetscErrorCode  SNESSetJacobianMatlab(SNES snes,Mat A,Mat B,const char *J,mxArray *ctx)
5432 {
5433   PetscErrorCode    ierr;
5434   SNESMatlabContext *sctx;
5435 
5436   PetscFunctionBegin;
5437   /* currently sctx is memory bleed */
5438   ierr = PetscNew(&sctx);CHKERRQ(ierr);
5439   ierr = PetscStrallocpy(J,&sctx->funcname);CHKERRQ(ierr);
5440   /*
5441      This should work, but it doesn't
5442   sctx->ctx = ctx;
5443   mexMakeArrayPersistent(sctx->ctx);
5444   */
5445   sctx->ctx = mxDuplicateArray(ctx);
5446   ierr      = SNESSetJacobian(snes,A,B,SNESComputeJacobian_Matlab,sctx);CHKERRQ(ierr);
5447   PetscFunctionReturn(0);
5448 }
5449 
5450 /*
5451    SNESMonitor_Matlab - Calls the function that has been set with SNESMonitorSetMatlab().
5452 
5453    Collective on SNES
5454 
5455 .seealso: SNESSetFunction(), SNESGetFunction()
5456 @*/
5457 PetscErrorCode  SNESMonitor_Matlab(SNES snes,PetscInt it, PetscReal fnorm, void *ctx)
5458 {
5459   PetscErrorCode    ierr;
5460   SNESMatlabContext *sctx = (SNESMatlabContext*)ctx;
5461   int               nlhs  = 1,nrhs = 6;
5462   mxArray           *plhs[1],*prhs[6];
5463   long long int     lx = 0,ls = 0;
5464   Vec               x  = snes->vec_sol;
5465 
5466   PetscFunctionBegin;
5467   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5468 
5469   ierr    = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr);
5470   ierr    = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr);
5471   prhs[0] = mxCreateDoubleScalar((double)ls);
5472   prhs[1] = mxCreateDoubleScalar((double)it);
5473   prhs[2] = mxCreateDoubleScalar((double)fnorm);
5474   prhs[3] = mxCreateDoubleScalar((double)lx);
5475   prhs[4] = mxCreateString(sctx->funcname);
5476   prhs[5] = sctx->ctx;
5477   ierr    = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESMonitorInternal");CHKERRQ(ierr);
5478   ierr    = mxGetScalar(plhs[0]);CHKERRQ(ierr);
5479   mxDestroyArray(prhs[0]);
5480   mxDestroyArray(prhs[1]);
5481   mxDestroyArray(prhs[2]);
5482   mxDestroyArray(prhs[3]);
5483   mxDestroyArray(prhs[4]);
5484   mxDestroyArray(plhs[0]);
5485   PetscFunctionReturn(0);
5486 }
5487 
5488 /*
5489    SNESMonitorSetMatlab - Sets the monitor function from MATLAB
5490 
5491    Level: developer
5492 
5493    Developer Note:  This bleeds the allocated memory SNESMatlabContext *sctx;
5494 
5495 .keywords: SNES, nonlinear, set, function
5496 
5497 .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
5498 */
5499 PetscErrorCode  SNESMonitorSetMatlab(SNES snes,const char *f,mxArray *ctx)
5500 {
5501   PetscErrorCode    ierr;
5502   SNESMatlabContext *sctx;
5503 
5504   PetscFunctionBegin;
5505   /* currently sctx is memory bleed */
5506   ierr = PetscNew(&sctx);CHKERRQ(ierr);
5507   ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr);
5508   /*
5509      This should work, but it doesn't
5510   sctx->ctx = ctx;
5511   mexMakeArrayPersistent(sctx->ctx);
5512   */
5513   sctx->ctx = mxDuplicateArray(ctx);
5514   ierr      = SNESMonitorSet(snes,SNESMonitor_Matlab,sctx,NULL);CHKERRQ(ierr);
5515   PetscFunctionReturn(0);
5516 }
5517 
5518 #endif
5519