xref: /petsc/src/snes/interface/snes.c (revision 6719d8e4221aa2df40223c855908a52ac3a76cfd)
19b94acceSBarry Smith 
2c6db04a5SJed Brown #include <private/snesimpl.h>      /*I "petscsnes.h"  I*/
39b94acceSBarry Smith 
4ace3abfcSBarry Smith PetscBool  SNESRegisterAllCalled = PETSC_FALSE;
58ba1e511SMatthew Knepley PetscFList SNESList              = PETSC_NULL;
68ba1e511SMatthew Knepley 
78ba1e511SMatthew Knepley /* Logging support */
87087cfbeSBarry Smith PetscClassId  SNES_CLASSID;
9166c7f25SBarry Smith PetscLogEvent  SNES_Solve, SNES_LineSearch, SNES_FunctionEval, SNES_JacobianEval;
10a09944afSBarry Smith 
11a09944afSBarry Smith #undef __FUNCT__
12cab2e9ccSBarry Smith #define __FUNCT__ "SNESDMComputeJacobian"
13cab2e9ccSBarry Smith /*
14cab2e9ccSBarry Smith     Translates from a SNES call to a DM call in computing a Jacobian
15cab2e9ccSBarry Smith */
16cab2e9ccSBarry Smith PetscErrorCode SNESDMComputeJacobian(SNES snes,Vec X,Mat *J,Mat *B,MatStructure *flag,void *ptr)
17cab2e9ccSBarry Smith {
18cab2e9ccSBarry Smith   PetscErrorCode ierr;
19cab2e9ccSBarry Smith   DM             dm;
20cab2e9ccSBarry Smith 
21cab2e9ccSBarry Smith   PetscFunctionBegin;
22cab2e9ccSBarry Smith   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
23cab2e9ccSBarry Smith   ierr = DMComputeJacobian(dm,X,*J,*B,flag);CHKERRQ(ierr);
24cab2e9ccSBarry Smith   PetscFunctionReturn(0);
25cab2e9ccSBarry Smith }
26cab2e9ccSBarry Smith 
27cab2e9ccSBarry Smith #undef __FUNCT__
28e113a28aSBarry Smith #define __FUNCT__ "SNESSetErrorIfNotConverged"
29e113a28aSBarry Smith /*@
30e113a28aSBarry Smith    SNESSetErrorIfNotConverged - Causes SNESSolve() to generate an error if the solver has not converged.
31e113a28aSBarry Smith 
323f9fe445SBarry Smith    Logically Collective on SNES
33e113a28aSBarry Smith 
34e113a28aSBarry Smith    Input Parameters:
35e113a28aSBarry Smith +  snes - iterative context obtained from SNESCreate()
36e113a28aSBarry Smith -  flg - PETSC_TRUE indicates you want the error generated
37e113a28aSBarry Smith 
38e113a28aSBarry Smith    Options database keys:
39e113a28aSBarry Smith .  -snes_error_if_not_converged : this takes an optional truth value (0/1/no/yes/true/false)
40e113a28aSBarry Smith 
41e113a28aSBarry Smith    Level: intermediate
42e113a28aSBarry Smith 
43e113a28aSBarry Smith    Notes:
44e113a28aSBarry Smith     Normally PETSc continues if a linear solver fails to converge, you can call SNESGetConvergedReason() after a SNESSolve()
45e113a28aSBarry Smith     to determine if it has converged.
46e113a28aSBarry Smith 
47e113a28aSBarry Smith .keywords: SNES, set, initial guess, nonzero
48e113a28aSBarry Smith 
49e113a28aSBarry Smith .seealso: SNESGetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged()
50e113a28aSBarry Smith @*/
517087cfbeSBarry Smith PetscErrorCode  SNESSetErrorIfNotConverged(SNES snes,PetscBool  flg)
52e113a28aSBarry Smith {
53e113a28aSBarry Smith   PetscFunctionBegin;
54e113a28aSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
55acfcf0e5SJed Brown   PetscValidLogicalCollectiveBool(snes,flg,2);
56e113a28aSBarry Smith   snes->errorifnotconverged = flg;
57e113a28aSBarry Smith   PetscFunctionReturn(0);
58e113a28aSBarry Smith }
59e113a28aSBarry Smith 
60e113a28aSBarry Smith #undef __FUNCT__
61e113a28aSBarry Smith #define __FUNCT__ "SNESGetErrorIfNotConverged"
62e113a28aSBarry Smith /*@
63e113a28aSBarry Smith    SNESGetErrorIfNotConverged - Will SNESSolve() generate an error if the solver does not converge?
64e113a28aSBarry Smith 
65e113a28aSBarry Smith    Not Collective
66e113a28aSBarry Smith 
67e113a28aSBarry Smith    Input Parameter:
68e113a28aSBarry Smith .  snes - iterative context obtained from SNESCreate()
69e113a28aSBarry Smith 
70e113a28aSBarry Smith    Output Parameter:
71e113a28aSBarry Smith .  flag - PETSC_TRUE if it will generate an error, else PETSC_FALSE
72e113a28aSBarry Smith 
73e113a28aSBarry Smith    Level: intermediate
74e113a28aSBarry Smith 
75e113a28aSBarry Smith .keywords: SNES, set, initial guess, nonzero
76e113a28aSBarry Smith 
77e113a28aSBarry Smith .seealso:  SNESSetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged()
78e113a28aSBarry Smith @*/
797087cfbeSBarry Smith PetscErrorCode  SNESGetErrorIfNotConverged(SNES snes,PetscBool  *flag)
80e113a28aSBarry Smith {
81e113a28aSBarry Smith   PetscFunctionBegin;
82e113a28aSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
83e113a28aSBarry Smith   PetscValidPointer(flag,2);
84e113a28aSBarry Smith   *flag = snes->errorifnotconverged;
85e113a28aSBarry Smith   PetscFunctionReturn(0);
86e113a28aSBarry Smith }
87e113a28aSBarry Smith 
88e113a28aSBarry Smith #undef __FUNCT__
894936397dSBarry Smith #define __FUNCT__ "SNESSetFunctionDomainError"
90e725d27bSBarry Smith /*@
914936397dSBarry Smith    SNESSetFunctionDomainError - tells SNES that the input vector to your FormFunction is not
924936397dSBarry Smith      in the functions domain. For example, negative pressure.
934936397dSBarry Smith 
943f9fe445SBarry Smith    Logically Collective on SNES
954936397dSBarry Smith 
964936397dSBarry Smith    Input Parameters:
974936397dSBarry Smith .  SNES - the SNES context
984936397dSBarry Smith 
9928529972SSatish Balay    Level: advanced
1004936397dSBarry Smith 
1014936397dSBarry Smith .keywords: SNES, view
1024936397dSBarry Smith 
1034936397dSBarry Smith .seealso: SNESCreate(), SNESSetFunction()
1044936397dSBarry Smith @*/
1057087cfbeSBarry Smith PetscErrorCode  SNESSetFunctionDomainError(SNES snes)
1064936397dSBarry Smith {
1074936397dSBarry Smith   PetscFunctionBegin;
1080700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1094936397dSBarry Smith   snes->domainerror = PETSC_TRUE;
1104936397dSBarry Smith   PetscFunctionReturn(0);
1114936397dSBarry Smith }
1124936397dSBarry Smith 
1134936397dSBarry Smith #undef __FUNCT__
1144a2ae208SSatish Balay #define __FUNCT__ "SNESView"
1157e2c5f70SBarry Smith /*@C
1169b94acceSBarry Smith    SNESView - Prints the SNES data structure.
1179b94acceSBarry Smith 
1184c49b128SBarry Smith    Collective on SNES
119fee21e36SBarry Smith 
120c7afd0dbSLois Curfman McInnes    Input Parameters:
121c7afd0dbSLois Curfman McInnes +  SNES - the SNES context
122c7afd0dbSLois Curfman McInnes -  viewer - visualization context
123c7afd0dbSLois Curfman McInnes 
1249b94acceSBarry Smith    Options Database Key:
125c8a8ba5cSLois Curfman McInnes .  -snes_view - Calls SNESView() at end of SNESSolve()
1269b94acceSBarry Smith 
1279b94acceSBarry Smith    Notes:
1289b94acceSBarry Smith    The available visualization contexts include
129b0a32e0cSBarry Smith +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
130b0a32e0cSBarry Smith -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
131c8a8ba5cSLois Curfman McInnes          output where only the first processor opens
132c8a8ba5cSLois Curfman McInnes          the file.  All other processors send their
133c8a8ba5cSLois Curfman McInnes          data to the first processor to print.
1349b94acceSBarry Smith 
1353e081fefSLois Curfman McInnes    The user can open an alternative visualization context with
136b0a32e0cSBarry Smith    PetscViewerASCIIOpen() - output to a specified file.
1379b94acceSBarry Smith 
13836851e7fSLois Curfman McInnes    Level: beginner
13936851e7fSLois Curfman McInnes 
1409b94acceSBarry Smith .keywords: SNES, view
1419b94acceSBarry Smith 
142b0a32e0cSBarry Smith .seealso: PetscViewerASCIIOpen()
1439b94acceSBarry Smith @*/
1447087cfbeSBarry Smith PetscErrorCode  SNESView(SNES snes,PetscViewer viewer)
1459b94acceSBarry Smith {
146fa9f3622SBarry Smith   SNESKSPEW           *kctx;
147dfbe8321SBarry Smith   PetscErrorCode      ierr;
14894b7f48cSBarry Smith   KSP                 ksp;
149ace3abfcSBarry Smith   PetscBool           iascii,isstring;
1509b94acceSBarry Smith 
1513a40ed3dSBarry Smith   PetscFunctionBegin;
1520700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1533050cee2SBarry Smith   if (!viewer) {
1547adad957SLisandro Dalcin     ierr = PetscViewerASCIIGetStdout(((PetscObject)snes)->comm,&viewer);CHKERRQ(ierr);
1553050cee2SBarry Smith   }
1560700a824SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
157c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,viewer,2);
15874679c65SBarry Smith 
1592692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
1602692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr);
16132077d6dSBarry Smith   if (iascii) {
162317d6ea6SBarry Smith     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)snes,viewer,"SNES Object");CHKERRQ(ierr);
163e7788613SBarry Smith     if (snes->ops->view) {
164b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
165e7788613SBarry Smith       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
166b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
1670ef38995SBarry Smith     }
16877431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr);
169a83599f4SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  tolerances: relative=%G, absolute=%G, solution=%G\n",
17070441072SBarry Smith                  snes->rtol,snes->abstol,snes->xtol);CHKERRQ(ierr);
17177431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr);
17277431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr);
1739b94acceSBarry Smith     if (snes->ksp_ewconv) {
174fa9f3622SBarry Smith       kctx = (SNESKSPEW *)snes->kspconvctx;
1759b94acceSBarry Smith       if (kctx) {
17677431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"  Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr);
177a83599f4SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"    rtol_0=%G, rtol_max=%G, threshold=%G\n",kctx->rtol_0,kctx->rtol_max,kctx->threshold);CHKERRQ(ierr);
178a83599f4SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"    gamma=%G, alpha=%G, alpha2=%G\n",kctx->gamma,kctx->alpha,kctx->alpha2);CHKERRQ(ierr);
1799b94acceSBarry Smith       }
1809b94acceSBarry Smith     }
181eb1f6c34SBarry Smith     if (snes->lagpreconditioner == -1) {
182eb1f6c34SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Preconditioned is never rebuilt\n");CHKERRQ(ierr);
183eb1f6c34SBarry Smith     } else if (snes->lagpreconditioner > 1) {
184eb1f6c34SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Preconditioned is rebuilt every %D new Jacobians\n",snes->lagpreconditioner);CHKERRQ(ierr);
185eb1f6c34SBarry Smith     }
186eb1f6c34SBarry Smith     if (snes->lagjacobian == -1) {
187eb1f6c34SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is never rebuilt\n");CHKERRQ(ierr);
188eb1f6c34SBarry Smith     } else if (snes->lagjacobian > 1) {
18942f4f86dSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is rebuilt every %D SNES iterations\n",snes->lagjacobian);CHKERRQ(ierr);
190eb1f6c34SBarry Smith     }
1910f5bd95cSBarry Smith   } else if (isstring) {
192317d6ea6SBarry Smith     const char *type;
193454a90a3SBarry Smith     ierr = SNESGetType(snes,&type);CHKERRQ(ierr);
194b0a32e0cSBarry Smith     ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr);
19519bcc07fSBarry Smith   }
19642f4f86dSBarry Smith   if (snes->pc && snes->usespc) {
1974a0c5b0cSMatthew G Knepley     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
1984a0c5b0cSMatthew G Knepley     ierr = SNESView(snes->pc, viewer);CHKERRQ(ierr);
1994a0c5b0cSMatthew G Knepley     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
2004a0c5b0cSMatthew G Knepley   }
2012c155ee1SBarry Smith   if (snes->usesksp) {
2022c155ee1SBarry Smith     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
203b0a32e0cSBarry Smith     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
20494b7f48cSBarry Smith     ierr = KSPView(ksp,viewer);CHKERRQ(ierr);
205b0a32e0cSBarry Smith     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
2062c155ee1SBarry Smith   }
2073a40ed3dSBarry Smith   PetscFunctionReturn(0);
2089b94acceSBarry Smith }
2099b94acceSBarry Smith 
21076b2cf59SMatthew Knepley /*
21176b2cf59SMatthew Knepley   We retain a list of functions that also take SNES command
21276b2cf59SMatthew Knepley   line options. These are called at the end SNESSetFromOptions()
21376b2cf59SMatthew Knepley */
21476b2cf59SMatthew Knepley #define MAXSETFROMOPTIONS 5
215a7cc72afSBarry Smith static PetscInt numberofsetfromoptions;
2166849ba73SBarry Smith static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES);
21776b2cf59SMatthew Knepley 
218e74ef692SMatthew Knepley #undef __FUNCT__
219e74ef692SMatthew Knepley #define __FUNCT__ "SNESAddOptionsChecker"
220ac226902SBarry Smith /*@C
22176b2cf59SMatthew Knepley   SNESAddOptionsChecker - Adds an additional function to check for SNES options.
22276b2cf59SMatthew Knepley 
22376b2cf59SMatthew Knepley   Not Collective
22476b2cf59SMatthew Knepley 
22576b2cf59SMatthew Knepley   Input Parameter:
22676b2cf59SMatthew Knepley . snescheck - function that checks for options
22776b2cf59SMatthew Knepley 
22876b2cf59SMatthew Knepley   Level: developer
22976b2cf59SMatthew Knepley 
23076b2cf59SMatthew Knepley .seealso: SNESSetFromOptions()
23176b2cf59SMatthew Knepley @*/
2327087cfbeSBarry Smith PetscErrorCode  SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES))
23376b2cf59SMatthew Knepley {
23476b2cf59SMatthew Knepley   PetscFunctionBegin;
23576b2cf59SMatthew Knepley   if (numberofsetfromoptions >= MAXSETFROMOPTIONS) {
236e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS);
23776b2cf59SMatthew Knepley   }
23876b2cf59SMatthew Knepley   othersetfromoptions[numberofsetfromoptions++] = snescheck;
23976b2cf59SMatthew Knepley   PetscFunctionReturn(0);
24076b2cf59SMatthew Knepley }
24176b2cf59SMatthew Knepley 
2427087cfbeSBarry Smith extern PetscErrorCode  SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*);
243aa3661deSLisandro Dalcin 
244aa3661deSLisandro Dalcin #undef __FUNCT__
245aa3661deSLisandro Dalcin #define __FUNCT__ "SNESSetUpMatrixFree_Private"
246ace3abfcSBarry Smith static PetscErrorCode SNESSetUpMatrixFree_Private(SNES snes, PetscBool  hasOperator, PetscInt version)
247aa3661deSLisandro Dalcin {
248aa3661deSLisandro Dalcin   Mat            J;
249aa3661deSLisandro Dalcin   KSP            ksp;
250aa3661deSLisandro Dalcin   PC             pc;
251ace3abfcSBarry Smith   PetscBool      match;
252aa3661deSLisandro Dalcin   PetscErrorCode ierr;
253aa3661deSLisandro Dalcin 
254aa3661deSLisandro Dalcin   PetscFunctionBegin;
2550700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
256aa3661deSLisandro Dalcin 
25798613b67SLisandro Dalcin   if(!snes->vec_func && (snes->jacobian || snes->jacobian_pre)) {
25898613b67SLisandro Dalcin     Mat A = snes->jacobian, B = snes->jacobian_pre;
25998613b67SLisandro Dalcin     ierr = MatGetVecs(A ? A : B, PETSC_NULL,&snes->vec_func);CHKERRQ(ierr);
26098613b67SLisandro Dalcin   }
26198613b67SLisandro Dalcin 
262aa3661deSLisandro Dalcin   if (version == 1) {
263aa3661deSLisandro Dalcin     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
26498613b67SLisandro Dalcin     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
2659c6ac3b3SBarry Smith     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
266aa3661deSLisandro Dalcin   } else if (version == 2) {
267e32f2f54SBarry Smith     if (!snes->vec_func) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"SNESSetFunction() must be called first");
26882a7e548SBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_REAL_SINGLE) && !defined(PETSC_USE_REAL___FLOAT128)
269aa3661deSLisandro Dalcin     ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_func,&J);CHKERRQ(ierr);
270aa3661deSLisandro Dalcin #else
271e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP, "matrix-free operator rutines (version 2)");
272aa3661deSLisandro Dalcin #endif
273a8248277SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "matrix-free operator rutines, only version 1 and 2");
274aa3661deSLisandro Dalcin 
275aa3661deSLisandro Dalcin   ierr = PetscInfo1(snes,"Setting default matrix-free operator routines (version %D)\n", version);CHKERRQ(ierr);
276d3462f78SMatthew Knepley   if (hasOperator) {
277aa3661deSLisandro Dalcin     /* This version replaces the user provided Jacobian matrix with a
278aa3661deSLisandro Dalcin        matrix-free version but still employs the user-provided preconditioner matrix. */
279aa3661deSLisandro Dalcin     ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);
280aa3661deSLisandro Dalcin   } else {
281aa3661deSLisandro Dalcin     /* This version replaces both the user-provided Jacobian and the user-
282aa3661deSLisandro Dalcin        provided preconditioner matrix with the default matrix free version. */
283aa3661deSLisandro Dalcin     ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,snes->funP);CHKERRQ(ierr);
284aa3661deSLisandro Dalcin     /* Force no preconditioner */
285aa3661deSLisandro Dalcin     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
286aa3661deSLisandro Dalcin     ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
287aa3661deSLisandro Dalcin     ierr = PetscTypeCompare((PetscObject)pc,PCSHELL,&match);CHKERRQ(ierr);
288aa3661deSLisandro Dalcin     if (!match) {
289aa3661deSLisandro Dalcin       ierr = PetscInfo(snes,"Setting default matrix-free preconditioner routines\nThat is no preconditioner is being used\n");CHKERRQ(ierr);
290aa3661deSLisandro Dalcin       ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr);
291aa3661deSLisandro Dalcin     }
292aa3661deSLisandro Dalcin   }
2936bf464f9SBarry Smith   ierr = MatDestroy(&J);CHKERRQ(ierr);
294aa3661deSLisandro Dalcin   PetscFunctionReturn(0);
295aa3661deSLisandro Dalcin }
296aa3661deSLisandro Dalcin 
2974a2ae208SSatish Balay #undef __FUNCT__
2984a2ae208SSatish Balay #define __FUNCT__ "SNESSetFromOptions"
2999b94acceSBarry Smith /*@
30094b7f48cSBarry Smith    SNESSetFromOptions - Sets various SNES and KSP parameters from user options.
3019b94acceSBarry Smith 
302c7afd0dbSLois Curfman McInnes    Collective on SNES
303c7afd0dbSLois Curfman McInnes 
3049b94acceSBarry Smith    Input Parameter:
3059b94acceSBarry Smith .  snes - the SNES context
3069b94acceSBarry Smith 
30736851e7fSLois Curfman McInnes    Options Database Keys:
308ea630c6eSPeter Brune +  -snes_type <type> - ls, tr, ngmres, ncg, richardson, qn, vi, fas
30982738288SBarry Smith .  -snes_stol - convergence tolerance in terms of the norm
31082738288SBarry Smith                 of the change in the solution between steps
31170441072SBarry Smith .  -snes_atol <abstol> - absolute tolerance of residual norm
312b39c3a46SLois Curfman McInnes .  -snes_rtol <rtol> - relative decrease in tolerance norm from initial
313b39c3a46SLois Curfman McInnes .  -snes_max_it <max_it> - maximum number of iterations
314b39c3a46SLois Curfman McInnes .  -snes_max_funcs <max_funcs> - maximum number of function evaluations
31550ffb88aSMatthew Knepley .  -snes_max_fail <max_fail> - maximum number of failures
316ddf469c8SBarry Smith .  -snes_max_linear_solve_fail - number of linear solver failures before SNESSolve() stops
317a8054027SBarry Smith .  -snes_lag_preconditioner <lag> - how often preconditioner is rebuilt (use -1 to never rebuild)
318e35cf81dSBarry Smith .  -snes_lag_jacobian <lag> - how often Jacobian is rebuilt (use -1 to never rebuild)
319b39c3a46SLois Curfman McInnes .  -snes_trtol <trtol> - trust region tolerance
3202492ecdbSBarry Smith .  -snes_no_convergence_test - skip convergence test in nonlinear
32182738288SBarry Smith                                solver; hence iterations will continue until max_it
3221fbbfb26SLois Curfman McInnes                                or some other criterion is reached. Saves expense
32382738288SBarry Smith                                of convergence test
324e8105e01SRichard Katz .  -snes_monitor <optional filename> - prints residual norm at each iteration. if no
325e8105e01SRichard Katz                                        filename given prints to stdout
326a6570f20SBarry Smith .  -snes_monitor_solution - plots solution at each iteration
327a6570f20SBarry Smith .  -snes_monitor_residual - plots residual (not its norm) at each iteration
328a6570f20SBarry Smith .  -snes_monitor_solution_update - plots update to solution at each iteration
329a6570f20SBarry Smith .  -snes_monitor_draw - plots residual norm at each iteration
330e24b481bSBarry Smith .  -snes_fd - use finite differences to compute Jacobian; very slow, only for testing
3315968eb51SBarry Smith .  -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration
332fee2055bSBarry Smith -  -snes_converged_reason - print the reason for convergence/divergence after each solve
33382738288SBarry Smith 
33482738288SBarry Smith     Options Database for Eisenstat-Walker method:
335fa9f3622SBarry Smith +  -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence
3364b27c08aSLois Curfman McInnes .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
33736851e7fSLois Curfman McInnes .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
33836851e7fSLois Curfman McInnes .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
33936851e7fSLois Curfman McInnes .  -snes_ksp_ew_gamma <gamma> - Sets gamma
34036851e7fSLois Curfman McInnes .  -snes_ksp_ew_alpha <alpha> - Sets alpha
34136851e7fSLois Curfman McInnes .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
34236851e7fSLois Curfman McInnes -  -snes_ksp_ew_threshold <threshold> - Sets threshold
34382738288SBarry Smith 
34411ca99fdSLois Curfman McInnes    Notes:
34511ca99fdSLois Curfman McInnes    To see all options, run your program with the -help option or consult
3460598bfebSBarry Smith    the <A href="../../docs/manual.pdf#nameddest=ch_snes">SNES chapter of the users manual</A>.
34783e2fdc7SBarry Smith 
34836851e7fSLois Curfman McInnes    Level: beginner
34936851e7fSLois Curfman McInnes 
3509b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database
3519b94acceSBarry Smith 
35269ed319cSSatish Balay .seealso: SNESSetOptionsPrefix()
3539b94acceSBarry Smith @*/
3547087cfbeSBarry Smith PetscErrorCode  SNESSetFromOptions(SNES snes)
3559b94acceSBarry Smith {
356ea630c6eSPeter Brune   PetscBool               flg,set,mf,mf_operator,pcset;
357efd51863SBarry Smith   PetscInt                i,indx,lag,mf_version,grids;
358aa3661deSLisandro Dalcin   MatStructure            matflag;
35985385478SLisandro Dalcin   const char              *deft = SNESLS;
36085385478SLisandro Dalcin   const char              *convtests[] = {"default","skip"};
36185385478SLisandro Dalcin   SNESKSPEW               *kctx = NULL;
362e8105e01SRichard Katz   char                    type[256], monfilename[PETSC_MAX_PATH_LEN];
36351e86f29SPeter Brune   const char              *optionsprefix;
364649052a6SBarry Smith   PetscViewer             monviewer;
36585385478SLisandro Dalcin   PetscErrorCode          ierr;
3669b94acceSBarry Smith 
3673a40ed3dSBarry Smith   PetscFunctionBegin;
3680700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
369ca161407SBarry Smith 
370186905e3SBarry Smith   if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL);CHKERRQ(ierr);}
3713194b578SJed Brown   ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr);
3727adad957SLisandro Dalcin     if (((PetscObject)snes)->type_name) { deft = ((PetscObject)snes)->type_name; }
373b0a32e0cSBarry Smith     ierr = PetscOptionsList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr);
374d64ed03dSBarry Smith     if (flg) {
375186905e3SBarry Smith       ierr = SNESSetType(snes,type);CHKERRQ(ierr);
3767adad957SLisandro Dalcin     } else if (!((PetscObject)snes)->type_name) {
377186905e3SBarry Smith       ierr = SNESSetType(snes,deft);CHKERRQ(ierr);
378d64ed03dSBarry Smith     }
37990d69ab7SBarry Smith     /* not used here, but called so will go into help messaage */
380909c8a9fSBarry Smith     ierr = PetscOptionsName("-snes_view","Print detailed information on solver used","SNESView",0);CHKERRQ(ierr);
38193c39befSBarry Smith 
38257034d6fSHong Zhang     ierr = PetscOptionsReal("-snes_stol","Stop if step length less than","SNESSetTolerances",snes->xtol,&snes->xtol,0);CHKERRQ(ierr);
38357034d6fSHong Zhang     ierr = PetscOptionsReal("-snes_atol","Stop if function norm less than","SNESSetTolerances",snes->abstol,&snes->abstol,0);CHKERRQ(ierr);
384186905e3SBarry Smith 
38557034d6fSHong Zhang     ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less than","SNESSetTolerances",snes->rtol,&snes->rtol,0);CHKERRQ(ierr);
386b0a32e0cSBarry Smith     ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,PETSC_NULL);CHKERRQ(ierr);
387b0a32e0cSBarry Smith     ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,PETSC_NULL);CHKERRQ(ierr);
38850ffb88aSMatthew Knepley     ierr = PetscOptionsInt("-snes_max_fail","Maximum failures","SNESSetTolerances",snes->maxFailures,&snes->maxFailures,PETSC_NULL);CHKERRQ(ierr);
389ddf469c8SBarry Smith     ierr = PetscOptionsInt("-snes_max_linear_solve_fail","Maximum failures in linear solves allowed","SNESSetMaxLinearSolveFailures",snes->maxLinearSolveFailures,&snes->maxLinearSolveFailures,PETSC_NULL);CHKERRQ(ierr);
390acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_error_if_not_converged","Generate error if solver does not converge","SNESSetErrorIfNotConverged",snes->errorifnotconverged,&snes->errorifnotconverged,PETSC_NULL);CHKERRQ(ierr);
39185385478SLisandro Dalcin 
392a8054027SBarry Smith     ierr = PetscOptionsInt("-snes_lag_preconditioner","How often to rebuild preconditioner","SNESSetLagPreconditioner",snes->lagpreconditioner,&lag,&flg);CHKERRQ(ierr);
393a8054027SBarry Smith     if (flg) {
394a8054027SBarry Smith       ierr = SNESSetLagPreconditioner(snes,lag);CHKERRQ(ierr);
395a8054027SBarry Smith     }
396e35cf81dSBarry Smith     ierr = PetscOptionsInt("-snes_lag_jacobian","How often to rebuild Jacobian","SNESSetLagJacobian",snes->lagjacobian,&lag,&flg);CHKERRQ(ierr);
397e35cf81dSBarry Smith     if (flg) {
398e35cf81dSBarry Smith       ierr = SNESSetLagJacobian(snes,lag);CHKERRQ(ierr);
399e35cf81dSBarry Smith     }
400efd51863SBarry Smith     ierr = PetscOptionsInt("-snes_grid_sequence","Use grid sequencing to generate initial guess","SNESSetGridSequence",snes->gridsequence,&grids,&flg);CHKERRQ(ierr);
401efd51863SBarry Smith     if (flg) {
402efd51863SBarry Smith       ierr = SNESSetGridSequence(snes,grids);CHKERRQ(ierr);
403efd51863SBarry Smith     }
404a8054027SBarry Smith 
40585385478SLisandro Dalcin     ierr = PetscOptionsEList("-snes_convergence_test","Convergence test","SNESSetConvergenceTest",convtests,2,"default",&indx,&flg);CHKERRQ(ierr);
40685385478SLisandro Dalcin     if (flg) {
40785385478SLisandro Dalcin       switch (indx) {
4087f7931b9SBarry Smith       case 0: ierr = SNESSetConvergenceTest(snes,SNESDefaultConverged,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr); break;
4097f7931b9SBarry Smith       case 1: ierr = SNESSetConvergenceTest(snes,SNESSkipConverged,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);    break;
41085385478SLisandro Dalcin       }
41185385478SLisandro Dalcin     }
41285385478SLisandro Dalcin 
413acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_converged_reason","Print reason for converged or diverged","SNESSolve",snes->printreason,&snes->printreason,PETSC_NULL);CHKERRQ(ierr);
414186905e3SBarry Smith 
41585385478SLisandro Dalcin     kctx = (SNESKSPEW *)snes->kspconvctx;
41685385478SLisandro Dalcin 
417acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,PETSC_NULL);CHKERRQ(ierr);
418186905e3SBarry Smith 
419fa9f3622SBarry Smith     ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,0);CHKERRQ(ierr);
420fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,0);CHKERRQ(ierr);
421fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,0);CHKERRQ(ierr);
422fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,0);CHKERRQ(ierr);
423fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,0);CHKERRQ(ierr);
424fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,0);CHKERRQ(ierr);
425fa9f3622SBarry Smith     ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,0);CHKERRQ(ierr);
426186905e3SBarry Smith 
42790d69ab7SBarry Smith     flg  = PETSC_FALSE;
428acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
429a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);}
430eabae89aSBarry Smith 
431a6570f20SBarry Smith     ierr = PetscOptionsString("-snes_monitor","Monitor norm of function","SNESMonitorSet","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
432e8105e01SRichard Katz     if (flg) {
433649052a6SBarry Smith       ierr = PetscViewerASCIIOpen(((PetscObject)snes)->comm,monfilename,&monviewer);CHKERRQ(ierr);
434649052a6SBarry Smith       ierr = SNESMonitorSet(snes,SNESMonitorDefault,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
435e8105e01SRichard Katz     }
436eabae89aSBarry Smith 
437b271bb04SBarry Smith     ierr = PetscOptionsString("-snes_monitor_range","Monitor range of elements of function","SNESMonitorSet","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
438b271bb04SBarry Smith     if (flg) {
439b271bb04SBarry Smith       ierr = SNESMonitorSet(snes,SNESMonitorRange,0,0);CHKERRQ(ierr);
440b271bb04SBarry Smith     }
441b271bb04SBarry Smith 
442a6570f20SBarry Smith     ierr = PetscOptionsString("-snes_ratiomonitor","Monitor ratios of norms of function","SNESMonitorSetRatio","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
443eabae89aSBarry Smith     if (flg) {
444649052a6SBarry Smith       ierr = PetscViewerASCIIOpen(((PetscObject)snes)->comm,monfilename,&monviewer);CHKERRQ(ierr);
445f1bef1bcSMatthew Knepley       ierr = SNESMonitorSetRatio(snes,monviewer);CHKERRQ(ierr);
446e8105e01SRichard Katz     }
447eabae89aSBarry Smith 
448a6570f20SBarry Smith     ierr = PetscOptionsString("-snes_monitor_short","Monitor norm of function (fewer digits)","SNESMonitorSet","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
449eabae89aSBarry Smith     if (flg) {
450649052a6SBarry Smith       ierr = PetscViewerASCIIOpen(((PetscObject)snes)->comm,monfilename,&monviewer);CHKERRQ(ierr);
451649052a6SBarry Smith       ierr = SNESMonitorSet(snes,SNESMonitorDefaultShort,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
452eabae89aSBarry Smith     }
453eabae89aSBarry Smith 
4545180491cSLisandro Dalcin     ierr = PetscOptionsString("-snes_monitor_python","Use Python function","SNESMonitorSet",0,monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
4555180491cSLisandro Dalcin     if (flg) {ierr = PetscPythonMonitorSet((PetscObject)snes,monfilename);CHKERRQ(ierr);}
4565180491cSLisandro Dalcin 
45790d69ab7SBarry Smith     flg  = PETSC_FALSE;
458acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_monitor_solution","Plot solution at each iteration","SNESMonitorSolution",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
459a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorSolution,0,0);CHKERRQ(ierr);}
46090d69ab7SBarry Smith     flg  = PETSC_FALSE;
461acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_monitor_solution_update","Plot correction at each iteration","SNESMonitorSolutionUpdate",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
462a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorSolutionUpdate,0,0);CHKERRQ(ierr);}
46390d69ab7SBarry Smith     flg  = PETSC_FALSE;
464acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_monitor_residual","Plot residual at each iteration","SNESMonitorResidual",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
465a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorResidual,0,0);CHKERRQ(ierr);}
46690d69ab7SBarry Smith     flg  = PETSC_FALSE;
467acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_monitor_draw","Plot function norm at each iteration","SNESMonitorLG",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
468a6570f20SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorLG,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);}
46990d69ab7SBarry Smith     flg  = PETSC_FALSE;
470acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_monitor_range_draw","Plot function range at each iteration","SNESMonitorLG",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
471b271bb04SBarry Smith     if (flg) {ierr = SNESMonitorSet(snes,SNESMonitorLGRange,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);}
472e24b481bSBarry Smith 
47390d69ab7SBarry Smith     flg  = PETSC_FALSE;
474acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESDefaultComputeJacobian",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
4754b27c08aSLois Curfman McInnes     if (flg) {
476186905e3SBarry Smith       ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESDefaultComputeJacobian,snes->funP);CHKERRQ(ierr);
477ae15b995SBarry Smith       ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr);
4789b94acceSBarry Smith     }
479639f9d9dSBarry Smith 
480aa3661deSLisandro Dalcin     mf = mf_operator = PETSC_FALSE;
481aa3661deSLisandro Dalcin     flg = PETSC_FALSE;
482acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_mf_operator","Use a Matrix-Free Jacobian with user-provided preconditioner matrix","MatCreateSNESMF",PETSC_FALSE,&mf_operator,&flg);CHKERRQ(ierr);
483a8248277SBarry Smith     if (flg && mf_operator) {
484a8248277SBarry Smith       snes->mf_operator = PETSC_TRUE;
485a8248277SBarry Smith       mf = PETSC_TRUE;
486a8248277SBarry Smith     }
487aa3661deSLisandro Dalcin     flg = PETSC_FALSE;
488acfcf0e5SJed Brown     ierr = PetscOptionsBool("-snes_mf","Use a Matrix-Free Jacobian with no preconditioner matrix","MatCreateSNESMF",PETSC_FALSE,&mf,&flg);CHKERRQ(ierr);
489aa3661deSLisandro Dalcin     if (!flg && mf_operator) mf = PETSC_TRUE;
490aa3661deSLisandro Dalcin     mf_version = 1;
491aa3661deSLisandro Dalcin     ierr = PetscOptionsInt("-snes_mf_version","Matrix-Free routines version 1 or 2","None",mf_version,&mf_version,0);CHKERRQ(ierr);
492aa3661deSLisandro Dalcin 
493ea630c6eSPeter Brune     /* line search options */
494ea630c6eSPeter Brune     ierr = PetscOptionsReal("-snes_ls_alpha","Constant function norm must decrease by","None",snes->ls_alpha,&snes->ls_alpha,0);CHKERRQ(ierr);
495ea630c6eSPeter Brune     ierr = PetscOptionsReal("-snes_ls_maxstep","Step must be less than","None",snes->maxstep,&snes->maxstep,0);CHKERRQ(ierr);
496ea630c6eSPeter Brune     ierr = PetscOptionsReal("-snes_ls_steptol","Minimum lambda allowed","None",snes->steptol,&snes->steptol,0);CHKERRQ(ierr);
497ea630c6eSPeter Brune     ierr = PetscOptionsReal("-snes_ls_damping","Damping parameter","SNES",snes->damping,&snes->damping,&flg);CHKERRQ(ierr);
498af60355fSPeter Brune     ierr = PetscOptionsInt("-snes_ls_it"      ,"Line search iterations","SNES",snes->ls_its,&snes->ls_its,&flg);CHKERRQ(ierr);
499ea630c6eSPeter Brune     ierr = PetscOptionsBool("-snes_ls_monitor","Print progress of line searches","SNESLineSearchSetMonitor",snes->ls_monitor ? PETSC_TRUE : PETSC_FALSE,&flg,&set);CHKERRQ(ierr);
500ea630c6eSPeter Brune     if (set) {ierr = SNESLineSearchSetMonitor(snes,flg);CHKERRQ(ierr);}
50115f5eeeaSPeter Brune     ierr = PetscOptionsEnum("-snes_ls","Line search used","SNESLineSearchSet",SNESLineSearchTypes,(PetscEnum)snes->ls_type,(PetscEnum*)&indx,&flg);CHKERRQ(ierr);
50215f5eeeaSPeter Brune     if (flg) {
503ea630c6eSPeter Brune       ierr = SNESLineSearchSetType(snes,(SNESLineSearchType)indx);CHKERRQ(ierr);
50415f5eeeaSPeter Brune     }
5058e3fc8c0SJed Brown     flg = snes->ops->precheckstep == SNESLineSearchPreCheckPicard ? PETSC_TRUE : PETSC_FALSE;
5068e3fc8c0SJed Brown     ierr = PetscOptionsBool("-snes_ls_precheck_picard","Use a correction that sometimes improves convergence of Picard iteration","SNESLineSearchPreCheckPicard",flg,&flg,&set);CHKERRQ(ierr);
5078e3fc8c0SJed Brown     if (set) {
5088e3fc8c0SJed Brown       if (flg) {
5098e3fc8c0SJed Brown         snes->precheck_picard_angle = 10.; /* correction only active if angle is less than 10 degrees */
5108e3fc8c0SJed Brown         ierr = PetscOptionsReal("-snes_ls_precheck_picard_angle","Maximum angle at which to activate the correction","none",snes->precheck_picard_angle,&snes->precheck_picard_angle,PETSC_NULL);CHKERRQ(ierr);
5118e3fc8c0SJed Brown         ierr = SNESLineSearchSetPreCheck(snes,SNESLineSearchPreCheckPicard,&snes->precheck_picard_angle);CHKERRQ(ierr);
5128e3fc8c0SJed Brown       } else {
5138e3fc8c0SJed Brown         ierr = SNESLineSearchSetPreCheck(snes,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);
5148e3fc8c0SJed Brown       }
5158e3fc8c0SJed Brown     }
5168e3fc8c0SJed Brown 
51776b2cf59SMatthew Knepley     for(i = 0; i < numberofsetfromoptions; i++) {
51876b2cf59SMatthew Knepley       ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr);
51976b2cf59SMatthew Knepley     }
52076b2cf59SMatthew Knepley 
521e7788613SBarry Smith     if (snes->ops->setfromoptions) {
522e7788613SBarry Smith       ierr = (*snes->ops->setfromoptions)(snes);CHKERRQ(ierr);
523639f9d9dSBarry Smith     }
5245d973c19SBarry Smith 
5255d973c19SBarry Smith     /* process any options handlers added with PetscObjectAddOptionsHandler() */
5265d973c19SBarry Smith     ierr = PetscObjectProcessOptionsHandlers((PetscObject)snes);CHKERRQ(ierr);
527b0a32e0cSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
5284bbc92c1SBarry Smith 
529aa3661deSLisandro Dalcin   if (mf) { ierr = SNESSetUpMatrixFree_Private(snes, mf_operator, mf_version);CHKERRQ(ierr); }
5301cee3971SBarry Smith 
5311cee3971SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
532aa3661deSLisandro Dalcin   ierr = KSPGetOperators(snes->ksp,PETSC_NULL,PETSC_NULL,&matflag);
533aa3661deSLisandro Dalcin   ierr = KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre,matflag);CHKERRQ(ierr);
53485385478SLisandro Dalcin   ierr = KSPSetFromOptions(snes->ksp);CHKERRQ(ierr);
53593993e2dSLois Curfman McInnes 
53651e86f29SPeter Brune   /* if someone has set the SNES PC type, create it. */
53751e86f29SPeter Brune   ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr);
53851e86f29SPeter Brune   ierr = PetscOptionsHasName(optionsprefix, "-npc_snes_type", &pcset);CHKERRQ(ierr);
53951e86f29SPeter Brune   if (pcset && (!snes->pc)) {
54051e86f29SPeter Brune     ierr = SNESGetPC(snes, &snes->pc);CHKERRQ(ierr);
54151e86f29SPeter Brune   }
54251e86f29SPeter Brune 
5434a0c5b0cSMatthew G Knepley   if (snes->pc) {
5444a0c5b0cSMatthew G Knepley     ierr = SNESSetOptionsPrefix(snes->pc, "npc_");CHKERRQ(ierr);
5454a0c5b0cSMatthew G Knepley     ierr = SNESSetDM(snes->pc, snes->dm);CHKERRQ(ierr);
5464a0c5b0cSMatthew G Knepley     /* Should we make a duplicate vector and matrix? Leave the DM to make it? */
5474a0c5b0cSMatthew G Knepley     ierr = SNESSetFunction(snes->pc, snes->vec_func, snes->ops->computefunction, snes->funP);CHKERRQ(ierr);
5484a0c5b0cSMatthew G Knepley     ierr = SNESSetJacobian(snes->pc, snes->jacobian, snes->jacobian_pre, snes->ops->computejacobian, snes->jacP);CHKERRQ(ierr);
5494a0c5b0cSMatthew G Knepley     ierr = SNESSetFromOptions(snes->pc);CHKERRQ(ierr);
5504a0c5b0cSMatthew G Knepley   }
5513a40ed3dSBarry Smith   PetscFunctionReturn(0);
5529b94acceSBarry Smith }
5539b94acceSBarry Smith 
554d25893d9SBarry Smith #undef __FUNCT__
555d25893d9SBarry Smith #define __FUNCT__ "SNESSetComputeApplicationContext"
556d25893d9SBarry Smith /*@
557d25893d9SBarry Smith    SNESSetComputeApplicationContext - Sets an optional function to compute a user-defined context for
558d25893d9SBarry Smith    the nonlinear solvers.
559d25893d9SBarry Smith 
560d25893d9SBarry Smith    Logically Collective on SNES
561d25893d9SBarry Smith 
562d25893d9SBarry Smith    Input Parameters:
563d25893d9SBarry Smith +  snes - the SNES context
564d25893d9SBarry Smith .  compute - function to compute the context
565d25893d9SBarry Smith -  destroy - function to destroy the context
566d25893d9SBarry Smith 
567d25893d9SBarry Smith    Level: intermediate
568d25893d9SBarry Smith 
569d25893d9SBarry Smith .keywords: SNES, nonlinear, set, application, context
570d25893d9SBarry Smith 
571d25893d9SBarry Smith .seealso: SNESGetApplicationContext(), SNESSetComputeApplicationContext(), SNESGetApplicationContext()
572d25893d9SBarry Smith @*/
573d25893d9SBarry Smith PetscErrorCode  SNESSetComputeApplicationContext(SNES snes,PetscErrorCode (*compute)(SNES,void**),PetscErrorCode (*destroy)(void**))
574d25893d9SBarry Smith {
575d25893d9SBarry Smith   PetscFunctionBegin;
576d25893d9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
577d25893d9SBarry Smith   snes->ops->usercompute = compute;
578d25893d9SBarry Smith   snes->ops->userdestroy = destroy;
579d25893d9SBarry Smith   PetscFunctionReturn(0);
580d25893d9SBarry Smith }
581a847f771SSatish Balay 
5824a2ae208SSatish Balay #undef __FUNCT__
5834a2ae208SSatish Balay #define __FUNCT__ "SNESSetApplicationContext"
584b07ff414SBarry Smith /*@
5859b94acceSBarry Smith    SNESSetApplicationContext - Sets the optional user-defined context for
5869b94acceSBarry Smith    the nonlinear solvers.
5879b94acceSBarry Smith 
5883f9fe445SBarry Smith    Logically Collective on SNES
589fee21e36SBarry Smith 
590c7afd0dbSLois Curfman McInnes    Input Parameters:
591c7afd0dbSLois Curfman McInnes +  snes - the SNES context
592c7afd0dbSLois Curfman McInnes -  usrP - optional user context
593c7afd0dbSLois Curfman McInnes 
59436851e7fSLois Curfman McInnes    Level: intermediate
59536851e7fSLois Curfman McInnes 
5969b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context
5979b94acceSBarry Smith 
598d25893d9SBarry Smith .seealso: SNESGetApplicationContext(), SNESSetApplicationContext()
5999b94acceSBarry Smith @*/
6007087cfbeSBarry Smith PetscErrorCode  SNESSetApplicationContext(SNES snes,void *usrP)
6019b94acceSBarry Smith {
6021b2093e4SBarry Smith   PetscErrorCode ierr;
603b07ff414SBarry Smith   KSP            ksp;
6041b2093e4SBarry Smith 
6053a40ed3dSBarry Smith   PetscFunctionBegin;
6060700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
607b07ff414SBarry Smith   ierr       = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
608b07ff414SBarry Smith   ierr       = KSPSetApplicationContext(ksp,usrP);CHKERRQ(ierr);
6099b94acceSBarry Smith   snes->user = usrP;
6103a40ed3dSBarry Smith   PetscFunctionReturn(0);
6119b94acceSBarry Smith }
61274679c65SBarry Smith 
6134a2ae208SSatish Balay #undef __FUNCT__
6144a2ae208SSatish Balay #define __FUNCT__ "SNESGetApplicationContext"
615b07ff414SBarry Smith /*@
6169b94acceSBarry Smith    SNESGetApplicationContext - Gets the user-defined context for the
6179b94acceSBarry Smith    nonlinear solvers.
6189b94acceSBarry Smith 
619c7afd0dbSLois Curfman McInnes    Not Collective
620c7afd0dbSLois Curfman McInnes 
6219b94acceSBarry Smith    Input Parameter:
6229b94acceSBarry Smith .  snes - SNES context
6239b94acceSBarry Smith 
6249b94acceSBarry Smith    Output Parameter:
6259b94acceSBarry Smith .  usrP - user context
6269b94acceSBarry Smith 
62736851e7fSLois Curfman McInnes    Level: intermediate
62836851e7fSLois Curfman McInnes 
6299b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context
6309b94acceSBarry Smith 
6319b94acceSBarry Smith .seealso: SNESSetApplicationContext()
6329b94acceSBarry Smith @*/
633e71120c6SJed Brown PetscErrorCode  SNESGetApplicationContext(SNES snes,void *usrP)
6349b94acceSBarry Smith {
6353a40ed3dSBarry Smith   PetscFunctionBegin;
6360700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
637e71120c6SJed Brown   *(void**)usrP = snes->user;
6383a40ed3dSBarry Smith   PetscFunctionReturn(0);
6399b94acceSBarry Smith }
64074679c65SBarry Smith 
6414a2ae208SSatish Balay #undef __FUNCT__
6424a2ae208SSatish Balay #define __FUNCT__ "SNESGetIterationNumber"
6439b94acceSBarry Smith /*@
644c8228a4eSBarry Smith    SNESGetIterationNumber - Gets the number of nonlinear iterations completed
645c8228a4eSBarry Smith    at this time.
6469b94acceSBarry Smith 
647c7afd0dbSLois Curfman McInnes    Not Collective
648c7afd0dbSLois Curfman McInnes 
6499b94acceSBarry Smith    Input Parameter:
6509b94acceSBarry Smith .  snes - SNES context
6519b94acceSBarry Smith 
6529b94acceSBarry Smith    Output Parameter:
6539b94acceSBarry Smith .  iter - iteration number
6549b94acceSBarry Smith 
655c8228a4eSBarry Smith    Notes:
656c8228a4eSBarry Smith    For example, during the computation of iteration 2 this would return 1.
657c8228a4eSBarry Smith 
658c8228a4eSBarry Smith    This is useful for using lagged Jacobians (where one does not recompute the
65908405cd6SLois Curfman McInnes    Jacobian at each SNES iteration). For example, the code
66008405cd6SLois Curfman McInnes .vb
66108405cd6SLois Curfman McInnes       ierr = SNESGetIterationNumber(snes,&it);
66208405cd6SLois Curfman McInnes       if (!(it % 2)) {
66308405cd6SLois Curfman McInnes         [compute Jacobian here]
66408405cd6SLois Curfman McInnes       }
66508405cd6SLois Curfman McInnes .ve
666c8228a4eSBarry Smith    can be used in your ComputeJacobian() function to cause the Jacobian to be
66708405cd6SLois Curfman McInnes    recomputed every second SNES iteration.
668c8228a4eSBarry Smith 
66936851e7fSLois Curfman McInnes    Level: intermediate
67036851e7fSLois Curfman McInnes 
6712b668275SBarry Smith .keywords: SNES, nonlinear, get, iteration, number,
6722b668275SBarry Smith 
673b850b91aSLisandro Dalcin .seealso:   SNESGetFunctionNorm(), SNESGetLinearSolveIterations()
6749b94acceSBarry Smith @*/
6757087cfbeSBarry Smith PetscErrorCode  SNESGetIterationNumber(SNES snes,PetscInt* iter)
6769b94acceSBarry Smith {
6773a40ed3dSBarry Smith   PetscFunctionBegin;
6780700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
6794482741eSBarry Smith   PetscValidIntPointer(iter,2);
6809b94acceSBarry Smith   *iter = snes->iter;
6813a40ed3dSBarry Smith   PetscFunctionReturn(0);
6829b94acceSBarry Smith }
68374679c65SBarry Smith 
6844a2ae208SSatish Balay #undef __FUNCT__
6854a2ae208SSatish Balay #define __FUNCT__ "SNESGetFunctionNorm"
6869b94acceSBarry Smith /*@
6879b94acceSBarry Smith    SNESGetFunctionNorm - Gets the norm of the current function that was set
6889b94acceSBarry Smith    with SNESSSetFunction().
6899b94acceSBarry Smith 
690c7afd0dbSLois Curfman McInnes    Collective on SNES
691c7afd0dbSLois Curfman McInnes 
6929b94acceSBarry Smith    Input Parameter:
6939b94acceSBarry Smith .  snes - SNES context
6949b94acceSBarry Smith 
6959b94acceSBarry Smith    Output Parameter:
6969b94acceSBarry Smith .  fnorm - 2-norm of function
6979b94acceSBarry Smith 
69836851e7fSLois Curfman McInnes    Level: intermediate
69936851e7fSLois Curfman McInnes 
7009b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm
701a86d99e1SLois Curfman McInnes 
702b850b91aSLisandro Dalcin .seealso: SNESGetFunction(), SNESGetIterationNumber(), SNESGetLinearSolveIterations()
7039b94acceSBarry Smith @*/
7047087cfbeSBarry Smith PetscErrorCode  SNESGetFunctionNorm(SNES snes,PetscReal *fnorm)
7059b94acceSBarry Smith {
7063a40ed3dSBarry Smith   PetscFunctionBegin;
7070700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
7084482741eSBarry Smith   PetscValidScalarPointer(fnorm,2);
7099b94acceSBarry Smith   *fnorm = snes->norm;
7103a40ed3dSBarry Smith   PetscFunctionReturn(0);
7119b94acceSBarry Smith }
71274679c65SBarry Smith 
7134a2ae208SSatish Balay #undef __FUNCT__
714b850b91aSLisandro Dalcin #define __FUNCT__ "SNESGetNonlinearStepFailures"
7159b94acceSBarry Smith /*@
716b850b91aSLisandro Dalcin    SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps
7179b94acceSBarry Smith    attempted by the nonlinear solver.
7189b94acceSBarry Smith 
719c7afd0dbSLois Curfman McInnes    Not Collective
720c7afd0dbSLois Curfman McInnes 
7219b94acceSBarry Smith    Input Parameter:
7229b94acceSBarry Smith .  snes - SNES context
7239b94acceSBarry Smith 
7249b94acceSBarry Smith    Output Parameter:
7259b94acceSBarry Smith .  nfails - number of unsuccessful steps attempted
7269b94acceSBarry Smith 
727c96a6f78SLois Curfman McInnes    Notes:
728c96a6f78SLois Curfman McInnes    This counter is reset to zero for each successive call to SNESSolve().
729c96a6f78SLois Curfman McInnes 
73036851e7fSLois Curfman McInnes    Level: intermediate
73136851e7fSLois Curfman McInnes 
7329b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps
73358ebbce7SBarry Smith 
734e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
73558ebbce7SBarry Smith           SNESSetMaxNonlinearStepFailures(), SNESGetMaxNonlinearStepFailures()
7369b94acceSBarry Smith @*/
7377087cfbeSBarry Smith PetscErrorCode  SNESGetNonlinearStepFailures(SNES snes,PetscInt* nfails)
7389b94acceSBarry Smith {
7393a40ed3dSBarry Smith   PetscFunctionBegin;
7400700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
7414482741eSBarry Smith   PetscValidIntPointer(nfails,2);
74250ffb88aSMatthew Knepley   *nfails = snes->numFailures;
74350ffb88aSMatthew Knepley   PetscFunctionReturn(0);
74450ffb88aSMatthew Knepley }
74550ffb88aSMatthew Knepley 
74650ffb88aSMatthew Knepley #undef __FUNCT__
747b850b91aSLisandro Dalcin #define __FUNCT__ "SNESSetMaxNonlinearStepFailures"
74850ffb88aSMatthew Knepley /*@
749b850b91aSLisandro Dalcin    SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps
75050ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
75150ffb88aSMatthew Knepley 
75250ffb88aSMatthew Knepley    Not Collective
75350ffb88aSMatthew Knepley 
75450ffb88aSMatthew Knepley    Input Parameters:
75550ffb88aSMatthew Knepley +  snes     - SNES context
75650ffb88aSMatthew Knepley -  maxFails - maximum of unsuccessful steps
75750ffb88aSMatthew Knepley 
75850ffb88aSMatthew Knepley    Level: intermediate
75950ffb88aSMatthew Knepley 
76050ffb88aSMatthew Knepley .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
76158ebbce7SBarry Smith 
762e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
76358ebbce7SBarry Smith           SNESGetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures()
76450ffb88aSMatthew Knepley @*/
7657087cfbeSBarry Smith PetscErrorCode  SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails)
76650ffb88aSMatthew Knepley {
76750ffb88aSMatthew Knepley   PetscFunctionBegin;
7680700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
76950ffb88aSMatthew Knepley   snes->maxFailures = maxFails;
77050ffb88aSMatthew Knepley   PetscFunctionReturn(0);
77150ffb88aSMatthew Knepley }
77250ffb88aSMatthew Knepley 
77350ffb88aSMatthew Knepley #undef __FUNCT__
774b850b91aSLisandro Dalcin #define __FUNCT__ "SNESGetMaxNonlinearStepFailures"
77550ffb88aSMatthew Knepley /*@
776b850b91aSLisandro Dalcin    SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps
77750ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
77850ffb88aSMatthew Knepley 
77950ffb88aSMatthew Knepley    Not Collective
78050ffb88aSMatthew Knepley 
78150ffb88aSMatthew Knepley    Input Parameter:
78250ffb88aSMatthew Knepley .  snes     - SNES context
78350ffb88aSMatthew Knepley 
78450ffb88aSMatthew Knepley    Output Parameter:
78550ffb88aSMatthew Knepley .  maxFails - maximum of unsuccessful steps
78650ffb88aSMatthew Knepley 
78750ffb88aSMatthew Knepley    Level: intermediate
78850ffb88aSMatthew Knepley 
78950ffb88aSMatthew Knepley .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
79058ebbce7SBarry Smith 
791e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
79258ebbce7SBarry Smith           SNESSetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures()
79358ebbce7SBarry Smith 
79450ffb88aSMatthew Knepley @*/
7957087cfbeSBarry Smith PetscErrorCode  SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails)
79650ffb88aSMatthew Knepley {
79750ffb88aSMatthew Knepley   PetscFunctionBegin;
7980700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
7994482741eSBarry Smith   PetscValidIntPointer(maxFails,2);
80050ffb88aSMatthew Knepley   *maxFails = snes->maxFailures;
8013a40ed3dSBarry Smith   PetscFunctionReturn(0);
8029b94acceSBarry Smith }
803a847f771SSatish Balay 
8044a2ae208SSatish Balay #undef __FUNCT__
8052541af92SBarry Smith #define __FUNCT__ "SNESGetNumberFunctionEvals"
8062541af92SBarry Smith /*@
8072541af92SBarry Smith    SNESGetNumberFunctionEvals - Gets the number of user provided function evaluations
8082541af92SBarry Smith      done by SNES.
8092541af92SBarry Smith 
8102541af92SBarry Smith    Not Collective
8112541af92SBarry Smith 
8122541af92SBarry Smith    Input Parameter:
8132541af92SBarry Smith .  snes     - SNES context
8142541af92SBarry Smith 
8152541af92SBarry Smith    Output Parameter:
8162541af92SBarry Smith .  nfuncs - number of evaluations
8172541af92SBarry Smith 
8182541af92SBarry Smith    Level: intermediate
8192541af92SBarry Smith 
8202541af92SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
82158ebbce7SBarry Smith 
822e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures()
8232541af92SBarry Smith @*/
8247087cfbeSBarry Smith PetscErrorCode  SNESGetNumberFunctionEvals(SNES snes, PetscInt *nfuncs)
8252541af92SBarry Smith {
8262541af92SBarry Smith   PetscFunctionBegin;
8270700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
8282541af92SBarry Smith   PetscValidIntPointer(nfuncs,2);
8292541af92SBarry Smith   *nfuncs = snes->nfuncs;
8302541af92SBarry Smith   PetscFunctionReturn(0);
8312541af92SBarry Smith }
8322541af92SBarry Smith 
8332541af92SBarry Smith #undef __FUNCT__
8343d4c4710SBarry Smith #define __FUNCT__ "SNESGetLinearSolveFailures"
8353d4c4710SBarry Smith /*@
8363d4c4710SBarry Smith    SNESGetLinearSolveFailures - Gets the number of failed (non-converged)
8373d4c4710SBarry Smith    linear solvers.
8383d4c4710SBarry Smith 
8393d4c4710SBarry Smith    Not Collective
8403d4c4710SBarry Smith 
8413d4c4710SBarry Smith    Input Parameter:
8423d4c4710SBarry Smith .  snes - SNES context
8433d4c4710SBarry Smith 
8443d4c4710SBarry Smith    Output Parameter:
8453d4c4710SBarry Smith .  nfails - number of failed solves
8463d4c4710SBarry Smith 
8473d4c4710SBarry Smith    Notes:
8483d4c4710SBarry Smith    This counter is reset to zero for each successive call to SNESSolve().
8493d4c4710SBarry Smith 
8503d4c4710SBarry Smith    Level: intermediate
8513d4c4710SBarry Smith 
8523d4c4710SBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps
85358ebbce7SBarry Smith 
854e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures()
8553d4c4710SBarry Smith @*/
8567087cfbeSBarry Smith PetscErrorCode  SNESGetLinearSolveFailures(SNES snes,PetscInt* nfails)
8573d4c4710SBarry Smith {
8583d4c4710SBarry Smith   PetscFunctionBegin;
8590700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
8603d4c4710SBarry Smith   PetscValidIntPointer(nfails,2);
8613d4c4710SBarry Smith   *nfails = snes->numLinearSolveFailures;
8623d4c4710SBarry Smith   PetscFunctionReturn(0);
8633d4c4710SBarry Smith }
8643d4c4710SBarry Smith 
8653d4c4710SBarry Smith #undef __FUNCT__
8663d4c4710SBarry Smith #define __FUNCT__ "SNESSetMaxLinearSolveFailures"
8673d4c4710SBarry Smith /*@
8683d4c4710SBarry Smith    SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts
8693d4c4710SBarry Smith    allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE
8703d4c4710SBarry Smith 
8713f9fe445SBarry Smith    Logically Collective on SNES
8723d4c4710SBarry Smith 
8733d4c4710SBarry Smith    Input Parameters:
8743d4c4710SBarry Smith +  snes     - SNES context
8753d4c4710SBarry Smith -  maxFails - maximum allowed linear solve failures
8763d4c4710SBarry Smith 
8773d4c4710SBarry Smith    Level: intermediate
8783d4c4710SBarry Smith 
879a6796414SBarry Smith    Notes: By default this is 0; that is SNES returns on the first failed linear solve
8803d4c4710SBarry Smith 
8813d4c4710SBarry Smith .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
8823d4c4710SBarry Smith 
88358ebbce7SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations()
8843d4c4710SBarry Smith @*/
8857087cfbeSBarry Smith PetscErrorCode  SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails)
8863d4c4710SBarry Smith {
8873d4c4710SBarry Smith   PetscFunctionBegin;
8880700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
889c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxFails,2);
8903d4c4710SBarry Smith   snes->maxLinearSolveFailures = maxFails;
8913d4c4710SBarry Smith   PetscFunctionReturn(0);
8923d4c4710SBarry Smith }
8933d4c4710SBarry Smith 
8943d4c4710SBarry Smith #undef __FUNCT__
8953d4c4710SBarry Smith #define __FUNCT__ "SNESGetMaxLinearSolveFailures"
8963d4c4710SBarry Smith /*@
8973d4c4710SBarry Smith    SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that
8983d4c4710SBarry Smith      are allowed before SNES terminates
8993d4c4710SBarry Smith 
9003d4c4710SBarry Smith    Not Collective
9013d4c4710SBarry Smith 
9023d4c4710SBarry Smith    Input Parameter:
9033d4c4710SBarry Smith .  snes     - SNES context
9043d4c4710SBarry Smith 
9053d4c4710SBarry Smith    Output Parameter:
9063d4c4710SBarry Smith .  maxFails - maximum of unsuccessful solves allowed
9073d4c4710SBarry Smith 
9083d4c4710SBarry Smith    Level: intermediate
9093d4c4710SBarry Smith 
9103d4c4710SBarry Smith    Notes: By default this is 1; that is SNES returns on the first failed linear solve
9113d4c4710SBarry Smith 
9123d4c4710SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
9133d4c4710SBarry Smith 
914e1c61ce8SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(),
9153d4c4710SBarry Smith @*/
9167087cfbeSBarry Smith PetscErrorCode  SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails)
9173d4c4710SBarry Smith {
9183d4c4710SBarry Smith   PetscFunctionBegin;
9190700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
9203d4c4710SBarry Smith   PetscValidIntPointer(maxFails,2);
9213d4c4710SBarry Smith   *maxFails = snes->maxLinearSolveFailures;
9223d4c4710SBarry Smith   PetscFunctionReturn(0);
9233d4c4710SBarry Smith }
9243d4c4710SBarry Smith 
9253d4c4710SBarry Smith #undef __FUNCT__
926b850b91aSLisandro Dalcin #define __FUNCT__ "SNESGetLinearSolveIterations"
927c96a6f78SLois Curfman McInnes /*@
928b850b91aSLisandro Dalcin    SNESGetLinearSolveIterations - Gets the total number of linear iterations
929c96a6f78SLois Curfman McInnes    used by the nonlinear solver.
930c96a6f78SLois Curfman McInnes 
931c7afd0dbSLois Curfman McInnes    Not Collective
932c7afd0dbSLois Curfman McInnes 
933c96a6f78SLois Curfman McInnes    Input Parameter:
934c96a6f78SLois Curfman McInnes .  snes - SNES context
935c96a6f78SLois Curfman McInnes 
936c96a6f78SLois Curfman McInnes    Output Parameter:
937c96a6f78SLois Curfman McInnes .  lits - number of linear iterations
938c96a6f78SLois Curfman McInnes 
939c96a6f78SLois Curfman McInnes    Notes:
940c96a6f78SLois Curfman McInnes    This counter is reset to zero for each successive call to SNESSolve().
941c96a6f78SLois Curfman McInnes 
94236851e7fSLois Curfman McInnes    Level: intermediate
94336851e7fSLois Curfman McInnes 
944c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations
9452b668275SBarry Smith 
9468c7482ecSBarry Smith .seealso:  SNESGetIterationNumber(), SNESGetFunctionNorm(), SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures()
947c96a6f78SLois Curfman McInnes @*/
9487087cfbeSBarry Smith PetscErrorCode  SNESGetLinearSolveIterations(SNES snes,PetscInt* lits)
949c96a6f78SLois Curfman McInnes {
9503a40ed3dSBarry Smith   PetscFunctionBegin;
9510700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
9524482741eSBarry Smith   PetscValidIntPointer(lits,2);
953c96a6f78SLois Curfman McInnes   *lits = snes->linear_its;
9543a40ed3dSBarry Smith   PetscFunctionReturn(0);
955c96a6f78SLois Curfman McInnes }
956c96a6f78SLois Curfman McInnes 
9574a2ae208SSatish Balay #undef __FUNCT__
95894b7f48cSBarry Smith #define __FUNCT__ "SNESGetKSP"
95952baeb72SSatish Balay /*@
96094b7f48cSBarry Smith    SNESGetKSP - Returns the KSP context for a SNES solver.
9619b94acceSBarry Smith 
96294b7f48cSBarry Smith    Not Collective, but if SNES object is parallel, then KSP object is parallel
963c7afd0dbSLois Curfman McInnes 
9649b94acceSBarry Smith    Input Parameter:
9659b94acceSBarry Smith .  snes - the SNES context
9669b94acceSBarry Smith 
9679b94acceSBarry Smith    Output Parameter:
96894b7f48cSBarry Smith .  ksp - the KSP context
9699b94acceSBarry Smith 
9709b94acceSBarry Smith    Notes:
97194b7f48cSBarry Smith    The user can then directly manipulate the KSP context to set various
9729b94acceSBarry Smith    options, etc.  Likewise, the user can then extract and manipulate the
9732999313aSBarry Smith    PC contexts as well.
9749b94acceSBarry Smith 
97536851e7fSLois Curfman McInnes    Level: beginner
97636851e7fSLois Curfman McInnes 
97794b7f48cSBarry Smith .keywords: SNES, nonlinear, get, KSP, context
9789b94acceSBarry Smith 
9792999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
9809b94acceSBarry Smith @*/
9817087cfbeSBarry Smith PetscErrorCode  SNESGetKSP(SNES snes,KSP *ksp)
9829b94acceSBarry Smith {
9831cee3971SBarry Smith   PetscErrorCode ierr;
9841cee3971SBarry Smith 
9853a40ed3dSBarry Smith   PetscFunctionBegin;
9860700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
9874482741eSBarry Smith   PetscValidPointer(ksp,2);
9881cee3971SBarry Smith 
9891cee3971SBarry Smith   if (!snes->ksp) {
9901cee3971SBarry Smith     ierr = KSPCreate(((PetscObject)snes)->comm,&snes->ksp);CHKERRQ(ierr);
9911cee3971SBarry Smith     ierr = PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1);CHKERRQ(ierr);
9921cee3971SBarry Smith     ierr = PetscLogObjectParent(snes,snes->ksp);CHKERRQ(ierr);
9931cee3971SBarry Smith   }
99494b7f48cSBarry Smith   *ksp = snes->ksp;
9953a40ed3dSBarry Smith   PetscFunctionReturn(0);
9969b94acceSBarry Smith }
99782bf6240SBarry Smith 
9984a2ae208SSatish Balay #undef __FUNCT__
9992999313aSBarry Smith #define __FUNCT__ "SNESSetKSP"
10002999313aSBarry Smith /*@
10012999313aSBarry Smith    SNESSetKSP - Sets a KSP context for the SNES object to use
10022999313aSBarry Smith 
10032999313aSBarry Smith    Not Collective, but the SNES and KSP objects must live on the same MPI_Comm
10042999313aSBarry Smith 
10052999313aSBarry Smith    Input Parameters:
10062999313aSBarry Smith +  snes - the SNES context
10072999313aSBarry Smith -  ksp - the KSP context
10082999313aSBarry Smith 
10092999313aSBarry Smith    Notes:
10102999313aSBarry Smith    The SNES object already has its KSP object, you can obtain with SNESGetKSP()
10112999313aSBarry Smith    so this routine is rarely needed.
10122999313aSBarry Smith 
10132999313aSBarry Smith    The KSP object that is already in the SNES object has its reference count
10142999313aSBarry Smith    decreased by one.
10152999313aSBarry Smith 
10162999313aSBarry Smith    Level: developer
10172999313aSBarry Smith 
10182999313aSBarry Smith .keywords: SNES, nonlinear, get, KSP, context
10192999313aSBarry Smith 
10202999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
10212999313aSBarry Smith @*/
10227087cfbeSBarry Smith PetscErrorCode  SNESSetKSP(SNES snes,KSP ksp)
10232999313aSBarry Smith {
10242999313aSBarry Smith   PetscErrorCode ierr;
10252999313aSBarry Smith 
10262999313aSBarry Smith   PetscFunctionBegin;
10270700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
10280700a824SBarry Smith   PetscValidHeaderSpecific(ksp,KSP_CLASSID,2);
10292999313aSBarry Smith   PetscCheckSameComm(snes,1,ksp,2);
10307dcf0eaaSdalcinl   ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr);
1031906ed7ccSBarry Smith   if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);}
10322999313aSBarry Smith   snes->ksp = ksp;
10332999313aSBarry Smith   PetscFunctionReturn(0);
10342999313aSBarry Smith }
10352999313aSBarry Smith 
10367adad957SLisandro Dalcin #if 0
10372999313aSBarry Smith #undef __FUNCT__
10384a2ae208SSatish Balay #define __FUNCT__ "SNESPublish_Petsc"
10396849ba73SBarry Smith static PetscErrorCode SNESPublish_Petsc(PetscObject obj)
1040e24b481bSBarry Smith {
1041e24b481bSBarry Smith   PetscFunctionBegin;
1042e24b481bSBarry Smith   PetscFunctionReturn(0);
1043e24b481bSBarry Smith }
10447adad957SLisandro Dalcin #endif
1045e24b481bSBarry Smith 
10469b94acceSBarry Smith /* -----------------------------------------------------------*/
10474a2ae208SSatish Balay #undef __FUNCT__
10484a2ae208SSatish Balay #define __FUNCT__ "SNESCreate"
104952baeb72SSatish Balay /*@
10509b94acceSBarry Smith    SNESCreate - Creates a nonlinear solver context.
10519b94acceSBarry Smith 
1052c7afd0dbSLois Curfman McInnes    Collective on MPI_Comm
1053c7afd0dbSLois Curfman McInnes 
1054c7afd0dbSLois Curfman McInnes    Input Parameters:
1055906ed7ccSBarry Smith .  comm - MPI communicator
10569b94acceSBarry Smith 
10579b94acceSBarry Smith    Output Parameter:
10589b94acceSBarry Smith .  outsnes - the new SNES context
10599b94acceSBarry Smith 
1060c7afd0dbSLois Curfman McInnes    Options Database Keys:
1061c7afd0dbSLois Curfman McInnes +   -snes_mf - Activates default matrix-free Jacobian-vector products,
1062c7afd0dbSLois Curfman McInnes                and no preconditioning matrix
1063c7afd0dbSLois Curfman McInnes .   -snes_mf_operator - Activates default matrix-free Jacobian-vector
1064c7afd0dbSLois Curfman McInnes                products, and a user-provided preconditioning matrix
1065c7afd0dbSLois Curfman McInnes                as set by SNESSetJacobian()
1066c7afd0dbSLois Curfman McInnes -   -snes_fd - Uses (slow!) finite differences to compute Jacobian
1067c1f60f51SBarry Smith 
106836851e7fSLois Curfman McInnes    Level: beginner
106936851e7fSLois Curfman McInnes 
10709b94acceSBarry Smith .keywords: SNES, nonlinear, create, context
10719b94acceSBarry Smith 
1072a8054027SBarry Smith .seealso: SNESSolve(), SNESDestroy(), SNES, SNESSetLagPreconditioner()
1073a8054027SBarry Smith 
10749b94acceSBarry Smith @*/
10757087cfbeSBarry Smith PetscErrorCode  SNESCreate(MPI_Comm comm,SNES *outsnes)
10769b94acceSBarry Smith {
1077dfbe8321SBarry Smith   PetscErrorCode      ierr;
10789b94acceSBarry Smith   SNES                snes;
1079fa9f3622SBarry Smith   SNESKSPEW           *kctx;
108037fcc0dbSBarry Smith 
10813a40ed3dSBarry Smith   PetscFunctionBegin;
1082ed1caa07SMatthew Knepley   PetscValidPointer(outsnes,2);
10838ba1e511SMatthew Knepley   *outsnes = PETSC_NULL;
10848ba1e511SMatthew Knepley #ifndef PETSC_USE_DYNAMIC_LIBRARIES
10858ba1e511SMatthew Knepley   ierr = SNESInitializePackage(PETSC_NULL);CHKERRQ(ierr);
10868ba1e511SMatthew Knepley #endif
10878ba1e511SMatthew Knepley 
10883194b578SJed Brown   ierr = PetscHeaderCreate(snes,_p_SNES,struct _SNESOps,SNES_CLASSID,0,"SNES","Nonlinear solver","SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr);
10897adad957SLisandro Dalcin 
109085385478SLisandro Dalcin   snes->ops->converged    = SNESDefaultConverged;
10912c155ee1SBarry Smith   snes->usesksp           = PETSC_TRUE;
10929b94acceSBarry Smith   snes->max_its           = 50;
10939750a799SBarry Smith   snes->max_funcs	  = 10000;
10949b94acceSBarry Smith   snes->norm		  = 0.0;
1095b4874afaSBarry Smith   snes->rtol		  = 1.e-8;
1096b4874afaSBarry Smith   snes->ttol              = 0.0;
109770441072SBarry Smith   snes->abstol		  = 1.e-50;
10989b94acceSBarry Smith   snes->xtol		  = 1.e-8;
10994b27c08aSLois Curfman McInnes   snes->deltatol	  = 1.e-12;
11009b94acceSBarry Smith   snes->nfuncs            = 0;
110150ffb88aSMatthew Knepley   snes->numFailures       = 0;
110250ffb88aSMatthew Knepley   snes->maxFailures       = 1;
11037a00f4a9SLois Curfman McInnes   snes->linear_its        = 0;
1104e35cf81dSBarry Smith   snes->lagjacobian       = 1;
1105a8054027SBarry Smith   snes->lagpreconditioner = 1;
1106639f9d9dSBarry Smith   snes->numbermonitors    = 0;
11079b94acceSBarry Smith   snes->data              = 0;
11084dc4c822SBarry Smith   snes->setupcalled       = PETSC_FALSE;
1109186905e3SBarry Smith   snes->ksp_ewconv        = PETSC_FALSE;
11106f24a144SLois Curfman McInnes   snes->nwork             = 0;
111158c9b817SLisandro Dalcin   snes->work              = 0;
111258c9b817SLisandro Dalcin   snes->nvwork            = 0;
111358c9b817SLisandro Dalcin   snes->vwork             = 0;
1114758f92a0SBarry Smith   snes->conv_hist_len     = 0;
1115758f92a0SBarry Smith   snes->conv_hist_max     = 0;
1116758f92a0SBarry Smith   snes->conv_hist         = PETSC_NULL;
1117758f92a0SBarry Smith   snes->conv_hist_its     = PETSC_NULL;
1118758f92a0SBarry Smith   snes->conv_hist_reset   = PETSC_TRUE;
1119184914b5SBarry Smith   snes->reason            = SNES_CONVERGED_ITERATING;
11209b94acceSBarry Smith 
1121ea630c6eSPeter Brune   /* initialize the line search options */
1122ea630c6eSPeter Brune   snes->ls_type           = SNES_LS_BASIC;
1123af60355fSPeter Brune   snes->ls_its            = 1;
1124ea630c6eSPeter Brune   snes->damping           = 1.0;
1125ea630c6eSPeter Brune   snes->maxstep           = 1e8;
1126ea630c6eSPeter Brune   snes->steptol           = 1e-12;
1127ea630c6eSPeter Brune   snes->ls_alpha          = 1e-4;
1128ea630c6eSPeter Brune   snes->ls_monitor        = PETSC_NULL;
1129ea630c6eSPeter Brune 
1130ea630c6eSPeter Brune   snes->ops->linesearch   = PETSC_NULL;
1131ea630c6eSPeter Brune   snes->precheck          = PETSC_NULL;
1132ea630c6eSPeter Brune   snes->ops->precheckstep = PETSC_NULL;
1133ea630c6eSPeter Brune   snes->postcheck         = PETSC_NULL;
1134ea630c6eSPeter Brune   snes->ops->postcheckstep= PETSC_NULL;
1135ea630c6eSPeter Brune 
11363d4c4710SBarry Smith   snes->numLinearSolveFailures = 0;
11373d4c4710SBarry Smith   snes->maxLinearSolveFailures = 1;
11383d4c4710SBarry Smith 
11399b94acceSBarry Smith   /* Create context to compute Eisenstat-Walker relative tolerance for KSP */
114038f2d2fdSLisandro Dalcin   ierr = PetscNewLog(snes,SNESKSPEW,&kctx);CHKERRQ(ierr);
11419b94acceSBarry Smith   snes->kspconvctx  = (void*)kctx;
11429b94acceSBarry Smith   kctx->version     = 2;
11439b94acceSBarry Smith   kctx->rtol_0      = .3; /* Eisenstat and Walker suggest rtol_0=.5, but
11449b94acceSBarry Smith                              this was too large for some test cases */
114575567043SBarry Smith   kctx->rtol_last   = 0.0;
11469b94acceSBarry Smith   kctx->rtol_max    = .9;
11479b94acceSBarry Smith   kctx->gamma       = 1.0;
114871f87433Sdalcinl   kctx->alpha       = .5*(1.0 + sqrt(5.0));
114971f87433Sdalcinl   kctx->alpha2      = kctx->alpha;
11509b94acceSBarry Smith   kctx->threshold   = .1;
115175567043SBarry Smith   kctx->lresid_last = 0.0;
115275567043SBarry Smith   kctx->norm_last   = 0.0;
11539b94acceSBarry Smith 
11549b94acceSBarry Smith   *outsnes = snes;
11553a40ed3dSBarry Smith   PetscFunctionReturn(0);
11569b94acceSBarry Smith }
11579b94acceSBarry Smith 
11584a2ae208SSatish Balay #undef __FUNCT__
11594a2ae208SSatish Balay #define __FUNCT__ "SNESSetFunction"
11609b94acceSBarry Smith /*@C
11619b94acceSBarry Smith    SNESSetFunction - Sets the function evaluation routine and function
11629b94acceSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
11639b94acceSBarry Smith    equations.
11649b94acceSBarry Smith 
11653f9fe445SBarry Smith    Logically Collective on SNES
1166fee21e36SBarry Smith 
1167c7afd0dbSLois Curfman McInnes    Input Parameters:
1168c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1169c7afd0dbSLois Curfman McInnes .  r - vector to store function value
1170de044059SHong Zhang .  func - function evaluation routine
1171c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
1172c7afd0dbSLois Curfman McInnes          function evaluation routine (may be PETSC_NULL)
11739b94acceSBarry Smith 
1174c7afd0dbSLois Curfman McInnes    Calling sequence of func:
11758d76a1e5SLois Curfman McInnes $    func (SNES snes,Vec x,Vec f,void *ctx);
1176c7afd0dbSLois Curfman McInnes 
1177313e4042SLois Curfman McInnes .  f - function vector
1178c7afd0dbSLois Curfman McInnes -  ctx - optional user-defined function context
11799b94acceSBarry Smith 
11809b94acceSBarry Smith    Notes:
11819b94acceSBarry Smith    The Newton-like methods typically solve linear systems of the form
11829b94acceSBarry Smith $      f'(x) x = -f(x),
1183c7afd0dbSLois Curfman McInnes    where f'(x) denotes the Jacobian matrix and f(x) is the function.
11849b94acceSBarry Smith 
118536851e7fSLois Curfman McInnes    Level: beginner
118636851e7fSLois Curfman McInnes 
11879b94acceSBarry Smith .keywords: SNES, nonlinear, set, function
11889b94acceSBarry Smith 
1189a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian()
11909b94acceSBarry Smith @*/
11917087cfbeSBarry Smith PetscErrorCode  SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*func)(SNES,Vec,Vec,void*),void *ctx)
11929b94acceSBarry Smith {
119385385478SLisandro Dalcin   PetscErrorCode ierr;
11943a40ed3dSBarry Smith   PetscFunctionBegin;
11950700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1196d2a683ecSLisandro Dalcin   if (r) {
1197d2a683ecSLisandro Dalcin     PetscValidHeaderSpecific(r,VEC_CLASSID,2);
1198d2a683ecSLisandro Dalcin     PetscCheckSameComm(snes,1,r,2);
119985385478SLisandro Dalcin     ierr = PetscObjectReference((PetscObject)r);CHKERRQ(ierr);
12006bf464f9SBarry Smith     ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr);
120185385478SLisandro Dalcin     snes->vec_func = r;
1202d2a683ecSLisandro Dalcin   } else if (!snes->vec_func && snes->dm) {
1203d2a683ecSLisandro Dalcin     ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr);
1204d2a683ecSLisandro Dalcin   }
1205d2a683ecSLisandro Dalcin   if (func) snes->ops->computefunction = func;
1206d2a683ecSLisandro Dalcin   if (ctx)  snes->funP                 = ctx;
12073a40ed3dSBarry Smith   PetscFunctionReturn(0);
12089b94acceSBarry Smith }
12099b94acceSBarry Smith 
1210d25893d9SBarry Smith #undef __FUNCT__
1211d25893d9SBarry Smith #define __FUNCT__ "SNESSetComputeInitialGuess"
1212d25893d9SBarry Smith /*@C
1213d25893d9SBarry Smith    SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem
1214d25893d9SBarry Smith 
1215d25893d9SBarry Smith    Logically Collective on SNES
1216d25893d9SBarry Smith 
1217d25893d9SBarry Smith    Input Parameters:
1218d25893d9SBarry Smith +  snes - the SNES context
1219d25893d9SBarry Smith .  func - function evaluation routine
1220d25893d9SBarry Smith -  ctx - [optional] user-defined context for private data for the
1221d25893d9SBarry Smith          function evaluation routine (may be PETSC_NULL)
1222d25893d9SBarry Smith 
1223d25893d9SBarry Smith    Calling sequence of func:
1224d25893d9SBarry Smith $    func (SNES snes,Vec x,void *ctx);
1225d25893d9SBarry Smith 
1226d25893d9SBarry Smith .  f - function vector
1227d25893d9SBarry Smith -  ctx - optional user-defined function context
1228d25893d9SBarry Smith 
1229d25893d9SBarry Smith    Level: intermediate
1230d25893d9SBarry Smith 
1231d25893d9SBarry Smith .keywords: SNES, nonlinear, set, function
1232d25893d9SBarry Smith 
1233d25893d9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian()
1234d25893d9SBarry Smith @*/
1235d25893d9SBarry Smith PetscErrorCode  SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx)
1236d25893d9SBarry Smith {
1237d25893d9SBarry Smith   PetscFunctionBegin;
1238d25893d9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1239d25893d9SBarry Smith   if (func) snes->ops->computeinitialguess = func;
1240d25893d9SBarry Smith   if (ctx)  snes->initialguessP            = ctx;
1241d25893d9SBarry Smith   PetscFunctionReturn(0);
1242d25893d9SBarry Smith }
1243d25893d9SBarry Smith 
12443ab0aad5SBarry Smith /* --------------------------------------------------------------- */
12453ab0aad5SBarry Smith #undef __FUNCT__
12461096aae1SMatthew Knepley #define __FUNCT__ "SNESGetRhs"
12471096aae1SMatthew Knepley /*@C
12481096aae1SMatthew Knepley    SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set
12491096aae1SMatthew Knepley    it assumes a zero right hand side.
12501096aae1SMatthew Knepley 
12513f9fe445SBarry Smith    Logically Collective on SNES
12521096aae1SMatthew Knepley 
12531096aae1SMatthew Knepley    Input Parameter:
12541096aae1SMatthew Knepley .  snes - the SNES context
12551096aae1SMatthew Knepley 
12561096aae1SMatthew Knepley    Output Parameter:
1257bc08b0f1SBarry Smith .  rhs - the right hand side vector or PETSC_NULL if the right hand side vector is null
12581096aae1SMatthew Knepley 
12591096aae1SMatthew Knepley    Level: intermediate
12601096aae1SMatthew Knepley 
12611096aae1SMatthew Knepley .keywords: SNES, nonlinear, get, function, right hand side
12621096aae1SMatthew Knepley 
126385385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
12641096aae1SMatthew Knepley @*/
12657087cfbeSBarry Smith PetscErrorCode  SNESGetRhs(SNES snes,Vec *rhs)
12661096aae1SMatthew Knepley {
12671096aae1SMatthew Knepley   PetscFunctionBegin;
12680700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
12691096aae1SMatthew Knepley   PetscValidPointer(rhs,2);
127085385478SLisandro Dalcin   *rhs = snes->vec_rhs;
12711096aae1SMatthew Knepley   PetscFunctionReturn(0);
12721096aae1SMatthew Knepley }
12731096aae1SMatthew Knepley 
12741096aae1SMatthew Knepley #undef __FUNCT__
12754a2ae208SSatish Balay #define __FUNCT__ "SNESComputeFunction"
12769b94acceSBarry Smith /*@
127736851e7fSLois Curfman McInnes    SNESComputeFunction - Calls the function that has been set with
12789b94acceSBarry Smith                          SNESSetFunction().
12799b94acceSBarry Smith 
1280c7afd0dbSLois Curfman McInnes    Collective on SNES
1281c7afd0dbSLois Curfman McInnes 
12829b94acceSBarry Smith    Input Parameters:
1283c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1284c7afd0dbSLois Curfman McInnes -  x - input vector
12859b94acceSBarry Smith 
12869b94acceSBarry Smith    Output Parameter:
12873638b69dSLois Curfman McInnes .  y - function vector, as set by SNESSetFunction()
12889b94acceSBarry Smith 
12891bffabb2SLois Curfman McInnes    Notes:
129036851e7fSLois Curfman McInnes    SNESComputeFunction() is typically used within nonlinear solvers
129136851e7fSLois Curfman McInnes    implementations, so most users would not generally call this routine
129236851e7fSLois Curfman McInnes    themselves.
129336851e7fSLois Curfman McInnes 
129436851e7fSLois Curfman McInnes    Level: developer
129536851e7fSLois Curfman McInnes 
12969b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function
12979b94acceSBarry Smith 
1298a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction()
12999b94acceSBarry Smith @*/
13007087cfbeSBarry Smith PetscErrorCode  SNESComputeFunction(SNES snes,Vec x,Vec y)
13019b94acceSBarry Smith {
1302dfbe8321SBarry Smith   PetscErrorCode ierr;
13039b94acceSBarry Smith 
13043a40ed3dSBarry Smith   PetscFunctionBegin;
13050700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
13060700a824SBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
13070700a824SBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
1308c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,x,2);
1309c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,3);
1310184914b5SBarry Smith 
1311d5ba7fb7SMatthew Knepley   ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
1312e7788613SBarry Smith   if (snes->ops->computefunction) {
1313d64ed03dSBarry Smith     PetscStackPush("SNES user function");
131439d508bbSBarry Smith     ierr = (*snes->ops->computefunction)(snes,x,y,snes->funP);CHKERRQ(ierr);
1315d64ed03dSBarry Smith     PetscStackPop;
131673250ac0SBarry Smith   } else if (snes->dm) {
1317644e2e5bSBarry Smith     ierr = DMComputeFunction(snes->dm,x,y);CHKERRQ(ierr);
1318c90fad12SPeter Brune   } else if (snes->vec_rhs) {
1319c90fad12SPeter Brune     ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr);
1320644e2e5bSBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve().");
132185385478SLisandro Dalcin   if (snes->vec_rhs) {
132285385478SLisandro Dalcin     ierr = VecAXPY(y,-1.0,snes->vec_rhs);CHKERRQ(ierr);
13233ab0aad5SBarry Smith   }
1324ae3c334cSLois Curfman McInnes   snes->nfuncs++;
1325d5ba7fb7SMatthew Knepley   ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
13263a40ed3dSBarry Smith   PetscFunctionReturn(0);
13279b94acceSBarry Smith }
13289b94acceSBarry Smith 
13294a2ae208SSatish Balay #undef __FUNCT__
13304a2ae208SSatish Balay #define __FUNCT__ "SNESComputeJacobian"
133162fef451SLois Curfman McInnes /*@
133262fef451SLois Curfman McInnes    SNESComputeJacobian - Computes the Jacobian matrix that has been
133362fef451SLois Curfman McInnes    set with SNESSetJacobian().
133462fef451SLois Curfman McInnes 
1335c7afd0dbSLois Curfman McInnes    Collective on SNES and Mat
1336c7afd0dbSLois Curfman McInnes 
133762fef451SLois Curfman McInnes    Input Parameters:
1338c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1339c7afd0dbSLois Curfman McInnes -  x - input vector
134062fef451SLois Curfman McInnes 
134162fef451SLois Curfman McInnes    Output Parameters:
1342c7afd0dbSLois Curfman McInnes +  A - Jacobian matrix
134362fef451SLois Curfman McInnes .  B - optional preconditioning matrix
13442b668275SBarry Smith -  flag - flag indicating matrix structure (one of, SAME_NONZERO_PATTERN,DIFFERENT_NONZERO_PATTERN,SAME_PRECONDITIONER)
1345fee21e36SBarry Smith 
1346e35cf81dSBarry Smith   Options Database Keys:
1347e35cf81dSBarry Smith +    -snes_lag_preconditioner <lag>
1348693365a8SJed Brown .    -snes_lag_jacobian <lag>
1349693365a8SJed Brown .    -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences
1350693365a8SJed Brown .    -snes_compare_explicit_draw  - Compare the computed Jacobian to the finite difference Jacobian and draw the result
1351693365a8SJed Brown .    -snes_compare_explicit_contour  - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result
13524c30e9fbSJed Brown .    -snes_compare_operator  - Make the comparison options above use the operator instead of the preconditioning matrix
13534c30e9fbSJed Brown .    -snes_compare_coloring - Compute the finite differece Jacobian using coloring and display differences
13544c30e9fbSJed Brown .    -snes_compare_coloring_draw - Compute the finite differece Jacobian using coloring and draw differences
13554c30e9fbSJed Brown -    -snes_compare_coloring_contour - Compute the finite differece Jacobian using coloring and show contours of matrices and differences
1356e35cf81dSBarry Smith 
135762fef451SLois Curfman McInnes    Notes:
135862fef451SLois Curfman McInnes    Most users should not need to explicitly call this routine, as it
135962fef451SLois Curfman McInnes    is used internally within the nonlinear solvers.
136062fef451SLois Curfman McInnes 
136194b7f48cSBarry Smith    See KSPSetOperators() for important information about setting the
1362dc5a77f8SLois Curfman McInnes    flag parameter.
136362fef451SLois Curfman McInnes 
136436851e7fSLois Curfman McInnes    Level: developer
136536851e7fSLois Curfman McInnes 
136662fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix
136762fef451SLois Curfman McInnes 
1368e35cf81dSBarry Smith .seealso:  SNESSetJacobian(), KSPSetOperators(), MatStructure, SNESSetLagPreconditioner(), SNESSetLagJacobian()
136962fef451SLois Curfman McInnes @*/
13707087cfbeSBarry Smith PetscErrorCode  SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg)
13719b94acceSBarry Smith {
1372dfbe8321SBarry Smith   PetscErrorCode ierr;
1373ace3abfcSBarry Smith   PetscBool      flag;
13743a40ed3dSBarry Smith 
13753a40ed3dSBarry Smith   PetscFunctionBegin;
13760700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
13770700a824SBarry Smith   PetscValidHeaderSpecific(X,VEC_CLASSID,2);
13784482741eSBarry Smith   PetscValidPointer(flg,5);
1379c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,X,2);
1380e7788613SBarry Smith   if (!snes->ops->computejacobian) PetscFunctionReturn(0);
1381ebd3b9afSBarry Smith 
1382ebd3b9afSBarry Smith   /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */
1383ebd3b9afSBarry Smith 
1384fe3ffe1eSBarry Smith   if (snes->lagjacobian == -2) {
1385fe3ffe1eSBarry Smith     snes->lagjacobian = -1;
1386fe3ffe1eSBarry Smith     ierr = PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n");CHKERRQ(ierr);
1387fe3ffe1eSBarry Smith   } else if (snes->lagjacobian == -1) {
1388e35cf81dSBarry Smith     *flg = SAME_PRECONDITIONER;
1389e35cf81dSBarry Smith     ierr = PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n");CHKERRQ(ierr);
1390ebd3b9afSBarry Smith     ierr = PetscTypeCompare((PetscObject)*A,MATMFFD,&flag);CHKERRQ(ierr);
1391ebd3b9afSBarry Smith     if (flag) {
1392ebd3b9afSBarry Smith       ierr = MatAssemblyBegin(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1393ebd3b9afSBarry Smith       ierr = MatAssemblyEnd(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1394ebd3b9afSBarry Smith     }
1395e35cf81dSBarry Smith     PetscFunctionReturn(0);
1396e35cf81dSBarry Smith   } else if (snes->lagjacobian > 1 && snes->iter % snes->lagjacobian) {
1397e35cf81dSBarry Smith     *flg = SAME_PRECONDITIONER;
1398e35cf81dSBarry Smith     ierr = PetscInfo2(snes,"Reusing Jacobian/preconditioner because lag is %D and SNES iteration is %D\n",snes->lagjacobian,snes->iter);CHKERRQ(ierr);
1399ebd3b9afSBarry Smith     ierr = PetscTypeCompare((PetscObject)*A,MATMFFD,&flag);CHKERRQ(ierr);
1400ebd3b9afSBarry Smith     if (flag) {
1401ebd3b9afSBarry Smith       ierr = MatAssemblyBegin(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1402ebd3b9afSBarry Smith       ierr = MatAssemblyEnd(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1403ebd3b9afSBarry Smith     }
1404e35cf81dSBarry Smith     PetscFunctionReturn(0);
1405e35cf81dSBarry Smith   }
1406e35cf81dSBarry Smith 
1407c4fc05e7SBarry Smith   *flg = DIFFERENT_NONZERO_PATTERN;
1408e35cf81dSBarry Smith   ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,*A,*B);CHKERRQ(ierr);
1409d64ed03dSBarry Smith   PetscStackPush("SNES user Jacobian function");
1410e7788613SBarry Smith   ierr = (*snes->ops->computejacobian)(snes,X,A,B,flg,snes->jacP);CHKERRQ(ierr);
1411d64ed03dSBarry Smith   PetscStackPop;
1412d5ba7fb7SMatthew Knepley   ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,*A,*B);CHKERRQ(ierr);
1413a8054027SBarry Smith 
14143b4f5425SBarry Smith   if (snes->lagpreconditioner == -2) {
14153b4f5425SBarry Smith     ierr = PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n");CHKERRQ(ierr);
14163b4f5425SBarry Smith     snes->lagpreconditioner = -1;
14173b4f5425SBarry Smith   } else if (snes->lagpreconditioner == -1) {
1418a8054027SBarry Smith     *flg = SAME_PRECONDITIONER;
1419a8054027SBarry Smith     ierr = PetscInfo(snes,"Reusing preconditioner because lag is -1\n");CHKERRQ(ierr);
1420a8054027SBarry Smith   } else if (snes->lagpreconditioner > 1 && snes->iter % snes->lagpreconditioner) {
1421a8054027SBarry Smith     *flg = SAME_PRECONDITIONER;
1422a8054027SBarry Smith     ierr = PetscInfo2(snes,"Reusing preconditioner because lag is %D and SNES iteration is %D\n",snes->lagpreconditioner,snes->iter);CHKERRQ(ierr);
1423a8054027SBarry Smith   }
1424a8054027SBarry Smith 
14256d84be18SBarry Smith   /* make sure user returned a correct Jacobian and preconditioner */
14260700a824SBarry Smith   /* PetscValidHeaderSpecific(*A,MAT_CLASSID,3);
14270700a824SBarry Smith     PetscValidHeaderSpecific(*B,MAT_CLASSID,4);   */
1428693365a8SJed Brown   {
1429693365a8SJed Brown     PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE;
1430693365a8SJed Brown     ierr  = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_compare_explicit",&flag,PETSC_NULL);CHKERRQ(ierr);
1431693365a8SJed Brown     ierr  = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",&flag_draw,PETSC_NULL);CHKERRQ(ierr);
1432693365a8SJed Brown     ierr  = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",&flag_contour,PETSC_NULL);CHKERRQ(ierr);
1433693365a8SJed Brown     ierr  = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_compare_operator",&flag_operator,PETSC_NULL);CHKERRQ(ierr);
1434693365a8SJed Brown     if (flag || flag_draw || flag_contour) {
1435693365a8SJed Brown       Mat Bexp_mine = PETSC_NULL,Bexp,FDexp;
1436693365a8SJed Brown       MatStructure mstruct;
1437693365a8SJed Brown       PetscViewer vdraw,vstdout;
14386b3a5b13SJed Brown       PetscBool flg;
1439693365a8SJed Brown       if (flag_operator) {
1440693365a8SJed Brown         ierr = MatComputeExplicitOperator(*A,&Bexp_mine);CHKERRQ(ierr);
1441693365a8SJed Brown         Bexp = Bexp_mine;
1442693365a8SJed Brown       } else {
1443693365a8SJed Brown         /* See if the preconditioning matrix can be viewed and added directly */
1444693365a8SJed Brown         ierr = PetscTypeCompareAny((PetscObject)*B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr);
1445693365a8SJed Brown         if (flg) Bexp = *B;
1446693365a8SJed Brown         else {
1447693365a8SJed Brown           /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */
1448693365a8SJed Brown           ierr = MatComputeExplicitOperator(*B,&Bexp_mine);CHKERRQ(ierr);
1449693365a8SJed Brown           Bexp = Bexp_mine;
1450693365a8SJed Brown         }
1451693365a8SJed Brown       }
1452693365a8SJed Brown       ierr = MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp);CHKERRQ(ierr);
1453693365a8SJed Brown       ierr = SNESDefaultComputeJacobian(snes,X,&FDexp,&FDexp,&mstruct,NULL);CHKERRQ(ierr);
1454693365a8SJed Brown       ierr = PetscViewerASCIIGetStdout(((PetscObject)snes)->comm,&vstdout);CHKERRQ(ierr);
1455693365a8SJed Brown       if (flag_draw || flag_contour) {
1456693365a8SJed Brown         ierr = PetscViewerDrawOpen(((PetscObject)snes)->comm,0,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
1457693365a8SJed Brown         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
1458693365a8SJed Brown       } else vdraw = PETSC_NULL;
1459693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator?"Jacobian":"preconditioning Jacobian");CHKERRQ(ierr);
1460693365a8SJed Brown       if (flag) {ierr = MatView(Bexp,vstdout);CHKERRQ(ierr);}
1461693365a8SJed Brown       if (vdraw) {ierr = MatView(Bexp,vdraw);CHKERRQ(ierr);}
1462693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n");CHKERRQ(ierr);
1463693365a8SJed Brown       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
1464693365a8SJed Brown       if (vdraw) {ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);}
1465693365a8SJed Brown       ierr = MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
1466693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n");CHKERRQ(ierr);
1467693365a8SJed Brown       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
1468693365a8SJed Brown       if (vdraw) {              /* Always use contour for the difference */
1469693365a8SJed Brown         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
1470693365a8SJed Brown         ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);
1471693365a8SJed Brown         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
1472693365a8SJed Brown       }
1473693365a8SJed Brown       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
1474693365a8SJed Brown       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
1475693365a8SJed Brown       ierr = MatDestroy(&Bexp_mine);CHKERRQ(ierr);
1476693365a8SJed Brown       ierr = MatDestroy(&FDexp);CHKERRQ(ierr);
1477693365a8SJed Brown     }
1478693365a8SJed Brown   }
14794c30e9fbSJed Brown   {
1480*6719d8e4SJed Brown     PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE;
1481*6719d8e4SJed Brown     PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON;
14824c30e9fbSJed Brown     ierr = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_compare_coloring",&flag,PETSC_NULL);CHKERRQ(ierr);
1483*6719d8e4SJed Brown     ierr = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_compare_coloring_display",&flag_display,PETSC_NULL);CHKERRQ(ierr);
14844c30e9fbSJed Brown     ierr = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",&flag_draw,PETSC_NULL);CHKERRQ(ierr);
14854c30e9fbSJed Brown     ierr = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",&flag_contour,PETSC_NULL);CHKERRQ(ierr);
1486*6719d8e4SJed Brown     ierr = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",&flag_threshold,PETSC_NULL);CHKERRQ(ierr);
1487*6719d8e4SJed Brown     ierr = PetscOptionsGetReal(((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,PETSC_NULL);CHKERRQ(ierr);
1488*6719d8e4SJed Brown     ierr = PetscOptionsGetReal(((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,PETSC_NULL);CHKERRQ(ierr);
1489*6719d8e4SJed Brown     if (flag || flag_display || flag_draw || flag_contour || flag_threshold) {
14904c30e9fbSJed Brown       Mat Bfd;
14914c30e9fbSJed Brown       MatStructure mstruct;
14924c30e9fbSJed Brown       PetscViewer vdraw,vstdout;
14934c30e9fbSJed Brown       ISColoring iscoloring;
14944c30e9fbSJed Brown       MatFDColoring matfdcoloring;
14954c30e9fbSJed Brown       PetscErrorCode (*func)(SNES,Vec,Vec,void*);
14964c30e9fbSJed Brown       void *funcctx;
1497*6719d8e4SJed Brown       PetscReal norm1,norm2,normmax;
14984c30e9fbSJed Brown 
14994c30e9fbSJed Brown       ierr = MatDuplicate(*B,MAT_DO_NOT_COPY_VALUES,&Bfd);CHKERRQ(ierr);
15004c30e9fbSJed Brown       ierr = MatGetColoring(Bfd,MATCOLORINGSL,&iscoloring);CHKERRQ(ierr);
15014c30e9fbSJed Brown       ierr = MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring);CHKERRQ(ierr);
15024c30e9fbSJed Brown       ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr);
15034c30e9fbSJed Brown 
15044c30e9fbSJed Brown       /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */
15054c30e9fbSJed Brown       ierr = SNESGetFunction(snes,PETSC_NULL,&func,&funcctx);CHKERRQ(ierr);
15064c30e9fbSJed Brown       ierr = MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode(*)(void))func,funcctx);CHKERRQ(ierr);
15074c30e9fbSJed Brown       ierr = PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix);CHKERRQ(ierr);
15084c30e9fbSJed Brown       ierr = PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_");CHKERRQ(ierr);
15094c30e9fbSJed Brown       ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr);
15104c30e9fbSJed Brown       ierr = MatFDColoringApply(Bfd,matfdcoloring,X,&mstruct,snes);CHKERRQ(ierr);
15114c30e9fbSJed Brown       ierr = MatFDColoringDestroy(&matfdcoloring);CHKERRQ(ierr);
15124c30e9fbSJed Brown 
15134c30e9fbSJed Brown       ierr = PetscViewerASCIIGetStdout(((PetscObject)snes)->comm,&vstdout);CHKERRQ(ierr);
15144c30e9fbSJed Brown       if (flag_draw || flag_contour) {
15154c30e9fbSJed Brown         ierr = PetscViewerDrawOpen(((PetscObject)snes)->comm,0,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
15164c30e9fbSJed Brown         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
15174c30e9fbSJed Brown       } else vdraw = PETSC_NULL;
15184c30e9fbSJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n");CHKERRQ(ierr);
1519*6719d8e4SJed Brown       if (flag_display) {ierr = MatView(*B,vstdout);CHKERRQ(ierr);}
15204c30e9fbSJed Brown       if (vdraw) {ierr = MatView(*B,vdraw);CHKERRQ(ierr);}
15214c30e9fbSJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n");CHKERRQ(ierr);
1522*6719d8e4SJed Brown       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
15234c30e9fbSJed Brown       if (vdraw) {ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);}
15244c30e9fbSJed Brown       ierr = MatAYPX(Bfd,-1.0,*B,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
15254c30e9fbSJed Brown       ierr = MatNorm(Bfd,NORM_1,&norm1);CHKERRQ(ierr);
1526*6719d8e4SJed Brown       ierr = MatNorm(Bfd,NORM_FROBENIUS,&norm2);CHKERRQ(ierr);
15274c30e9fbSJed Brown       ierr = MatNorm(Bfd,NORM_MAX,&normmax);CHKERRQ(ierr);
1528*6719d8e4SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian, norm1=%G normFrob=%G normmax=%G\n",norm1,norm2,normmax);CHKERRQ(ierr);
1529*6719d8e4SJed Brown       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
15304c30e9fbSJed Brown       if (vdraw) {              /* Always use contour for the difference */
15314c30e9fbSJed Brown         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
15324c30e9fbSJed Brown         ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);
15334c30e9fbSJed Brown         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
15344c30e9fbSJed Brown       }
15354c30e9fbSJed Brown       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
1536*6719d8e4SJed Brown 
1537*6719d8e4SJed Brown       if (flag_threshold) {
1538*6719d8e4SJed Brown         PetscInt bs,rstart,rend,i;
1539*6719d8e4SJed Brown         ierr = MatGetBlockSize(*B,&bs);CHKERRQ(ierr);
1540*6719d8e4SJed Brown         ierr = MatGetOwnershipRange(*B,&rstart,&rend);CHKERRQ(ierr);
1541*6719d8e4SJed Brown         for (i=rstart; i<rend; i++) {
1542*6719d8e4SJed Brown           const PetscScalar *ba,*ca;
1543*6719d8e4SJed Brown           const PetscInt *bj,*cj;
1544*6719d8e4SJed Brown           PetscInt bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1;
1545*6719d8e4SJed Brown           PetscReal maxentry = 0,maxdiff = 0,maxrdiff = 0;
1546*6719d8e4SJed Brown           ierr = MatGetRow(*B,i,&bn,&bj,&ba);CHKERRQ(ierr);
1547*6719d8e4SJed Brown           ierr = MatGetRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
1548*6719d8e4SJed Brown           if (bn != cn) SETERRQ(((PetscObject)*A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold");
1549*6719d8e4SJed Brown           for (j=0; j<bn; j++) {
1550*6719d8e4SJed Brown             PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
1551*6719d8e4SJed Brown             if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) {
1552*6719d8e4SJed Brown               maxentrycol = bj[j];
1553*6719d8e4SJed Brown               maxentry = PetscRealPart(ba[j]);
1554*6719d8e4SJed Brown             }
1555*6719d8e4SJed Brown             if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) {
1556*6719d8e4SJed Brown               maxdiffcol = bj[j];
1557*6719d8e4SJed Brown               maxdiff = PetscRealPart(ca[j]);
1558*6719d8e4SJed Brown             }
1559*6719d8e4SJed Brown             if (rdiff > maxrdiff) {
1560*6719d8e4SJed Brown               maxrdiffcol = bj[j];
1561*6719d8e4SJed Brown               maxrdiff = rdiff;
1562*6719d8e4SJed Brown             }
1563*6719d8e4SJed Brown           }
1564*6719d8e4SJed Brown           if (maxrdiff > 1) {
1565*6719d8e4SJed Brown             ierr = PetscViewerASCIIPrintf(vstdout,"row %D (maxentry=%G at %D, maxdiff=%G at %D, maxrdiff=%G at %D):",i,maxentry,maxentrycol,maxdiff,maxdiffcol,maxrdiff,maxrdiffcol);CHKERRQ(ierr);
1566*6719d8e4SJed Brown             for (j=0; j<bn; j++) {
1567*6719d8e4SJed Brown               PetscReal rdiff;
1568*6719d8e4SJed Brown               rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
1569*6719d8e4SJed Brown               if (rdiff > 1) {
1570*6719d8e4SJed Brown                 ierr = PetscViewerASCIIPrintf(vstdout," (%D,%G:%G)",bj[j],PetscRealPart(ba[j]),PetscRealPart(ca[j]));CHKERRQ(ierr);
1571*6719d8e4SJed Brown               }
1572*6719d8e4SJed Brown             }
1573*6719d8e4SJed Brown             ierr = PetscViewerASCIIPrintf(vstdout,"\n",i,maxentry,maxdiff,maxrdiff);CHKERRQ(ierr);
1574*6719d8e4SJed Brown           }
1575*6719d8e4SJed Brown           ierr = MatRestoreRow(*B,i,&bn,&bj,&ba);CHKERRQ(ierr);
1576*6719d8e4SJed Brown           ierr = MatRestoreRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
1577*6719d8e4SJed Brown         }
1578*6719d8e4SJed Brown       }
15794c30e9fbSJed Brown       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
15804c30e9fbSJed Brown       ierr = MatDestroy(&Bfd);CHKERRQ(ierr);
15814c30e9fbSJed Brown     }
15824c30e9fbSJed Brown   }
15833a40ed3dSBarry Smith   PetscFunctionReturn(0);
15849b94acceSBarry Smith }
15859b94acceSBarry Smith 
15864a2ae208SSatish Balay #undef __FUNCT__
15874a2ae208SSatish Balay #define __FUNCT__ "SNESSetJacobian"
15889b94acceSBarry Smith /*@C
15899b94acceSBarry Smith    SNESSetJacobian - Sets the function to compute Jacobian as well as the
1590044dda88SLois Curfman McInnes    location to store the matrix.
15919b94acceSBarry Smith 
15923f9fe445SBarry Smith    Logically Collective on SNES and Mat
1593c7afd0dbSLois Curfman McInnes 
15949b94acceSBarry Smith    Input Parameters:
1595c7afd0dbSLois Curfman McInnes +  snes - the SNES context
15969b94acceSBarry Smith .  A - Jacobian matrix
15979b94acceSBarry Smith .  B - preconditioner matrix (usually same as the Jacobian)
1598efd51863SBarry Smith .  func - Jacobian evaluation routine (if PETSC_NULL then SNES retains any previously set value)
1599c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
1600efd51863SBarry Smith          Jacobian evaluation routine (may be PETSC_NULL) (if PETSC_NULL then SNES retains any previously set value)
16019b94acceSBarry Smith 
16029b94acceSBarry Smith    Calling sequence of func:
16038d76a1e5SLois Curfman McInnes $     func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx);
16049b94acceSBarry Smith 
1605c7afd0dbSLois Curfman McInnes +  x - input vector
16069b94acceSBarry Smith .  A - Jacobian matrix
16079b94acceSBarry Smith .  B - preconditioner matrix, usually the same as A
1608ac21db08SLois Curfman McInnes .  flag - flag indicating information about the preconditioner matrix
16092b668275SBarry Smith    structure (same as flag in KSPSetOperators()), one of SAME_NONZERO_PATTERN,DIFFERENT_NONZERO_PATTERN,SAME_PRECONDITIONER
1610c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined Jacobian context
16119b94acceSBarry Smith 
16129b94acceSBarry Smith    Notes:
161394b7f48cSBarry Smith    See KSPSetOperators() for important information about setting the flag
16142cd2dfdcSLois Curfman McInnes    output parameter in the routine func().  Be sure to read this information!
1615ac21db08SLois Curfman McInnes 
1616ac21db08SLois Curfman McInnes    The routine func() takes Mat * as the matrix arguments rather than Mat.
16179b94acceSBarry Smith    This allows the Jacobian evaluation routine to replace A and/or B with a
16189b94acceSBarry Smith    completely new new matrix structure (not just different matrix elements)
16199b94acceSBarry Smith    when appropriate, for instance, if the nonzero structure is changing
16209b94acceSBarry Smith    throughout the global iterations.
16219b94acceSBarry Smith 
162216913363SBarry Smith    If the A matrix and B matrix are different you must call MatAssemblyBegin/End() on
162316913363SBarry Smith    each matrix.
162416913363SBarry Smith 
1625a8a26c1eSJed Brown    If using SNESDefaultComputeJacobianColor() to assemble a Jacobian, the ctx argument
1626a8a26c1eSJed Brown    must be a MatFDColoring.
1627a8a26c1eSJed Brown 
1628c3cc8fd1SJed Brown    Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian.  One common
1629c3cc8fd1SJed Brown    example is to use the "Picard linearization" which only differentiates through the highest order parts of each term.
1630c3cc8fd1SJed Brown 
163136851e7fSLois Curfman McInnes    Level: beginner
163236851e7fSLois Curfman McInnes 
16339b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix
16349b94acceSBarry Smith 
16353ec795f1SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESDefaultComputeJacobianColor(), MatStructure
16369b94acceSBarry Smith @*/
16377087cfbeSBarry Smith PetscErrorCode  SNESSetJacobian(SNES snes,Mat A,Mat B,PetscErrorCode (*func)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void *ctx)
16389b94acceSBarry Smith {
1639dfbe8321SBarry Smith   PetscErrorCode ierr;
16403a7fca6bSBarry Smith 
16413a40ed3dSBarry Smith   PetscFunctionBegin;
16420700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
16430700a824SBarry Smith   if (A) PetscValidHeaderSpecific(A,MAT_CLASSID,2);
16440700a824SBarry Smith   if (B) PetscValidHeaderSpecific(B,MAT_CLASSID,3);
1645c9780b6fSBarry Smith   if (A) PetscCheckSameComm(snes,1,A,2);
164606975374SJed Brown   if (B) PetscCheckSameComm(snes,1,B,3);
1647e7788613SBarry Smith   if (func) snes->ops->computejacobian = func;
16483a7fca6bSBarry Smith   if (ctx)  snes->jacP                 = ctx;
16493a7fca6bSBarry Smith   if (A) {
16507dcf0eaaSdalcinl     ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
16516bf464f9SBarry Smith     ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
16529b94acceSBarry Smith     snes->jacobian = A;
16533a7fca6bSBarry Smith   }
16543a7fca6bSBarry Smith   if (B) {
16557dcf0eaaSdalcinl     ierr = PetscObjectReference((PetscObject)B);CHKERRQ(ierr);
16566bf464f9SBarry Smith     ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
16579b94acceSBarry Smith     snes->jacobian_pre = B;
16583a7fca6bSBarry Smith   }
16593a40ed3dSBarry Smith   PetscFunctionReturn(0);
16609b94acceSBarry Smith }
166162fef451SLois Curfman McInnes 
16624a2ae208SSatish Balay #undef __FUNCT__
16634a2ae208SSatish Balay #define __FUNCT__ "SNESGetJacobian"
1664c2aafc4cSSatish Balay /*@C
1665b4fd4287SBarry Smith    SNESGetJacobian - Returns the Jacobian matrix and optionally the user
1666b4fd4287SBarry Smith    provided context for evaluating the Jacobian.
1667b4fd4287SBarry Smith 
1668c7afd0dbSLois Curfman McInnes    Not Collective, but Mat object will be parallel if SNES object is
1669c7afd0dbSLois Curfman McInnes 
1670b4fd4287SBarry Smith    Input Parameter:
1671b4fd4287SBarry Smith .  snes - the nonlinear solver context
1672b4fd4287SBarry Smith 
1673b4fd4287SBarry Smith    Output Parameters:
1674c7afd0dbSLois Curfman McInnes +  A - location to stash Jacobian matrix (or PETSC_NULL)
1675b4fd4287SBarry Smith .  B - location to stash preconditioner matrix (or PETSC_NULL)
167670e92668SMatthew Knepley .  func - location to put Jacobian function (or PETSC_NULL)
167770e92668SMatthew Knepley -  ctx - location to stash Jacobian ctx (or PETSC_NULL)
1678fee21e36SBarry Smith 
167936851e7fSLois Curfman McInnes    Level: advanced
168036851e7fSLois Curfman McInnes 
1681b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian()
1682b4fd4287SBarry Smith @*/
16837087cfbeSBarry Smith PetscErrorCode  SNESGetJacobian(SNES snes,Mat *A,Mat *B,PetscErrorCode (**func)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void **ctx)
1684b4fd4287SBarry Smith {
16853a40ed3dSBarry Smith   PetscFunctionBegin;
16860700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1687b4fd4287SBarry Smith   if (A)    *A    = snes->jacobian;
1688b4fd4287SBarry Smith   if (B)    *B    = snes->jacobian_pre;
1689e7788613SBarry Smith   if (func) *func = snes->ops->computejacobian;
169070e92668SMatthew Knepley   if (ctx)  *ctx  = snes->jacP;
16913a40ed3dSBarry Smith   PetscFunctionReturn(0);
1692b4fd4287SBarry Smith }
1693b4fd4287SBarry Smith 
16949b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */
16959b94acceSBarry Smith 
16964a2ae208SSatish Balay #undef __FUNCT__
16974a2ae208SSatish Balay #define __FUNCT__ "SNESSetUp"
16989b94acceSBarry Smith /*@
16999b94acceSBarry Smith    SNESSetUp - Sets up the internal data structures for the later use
1700272ac6f2SLois Curfman McInnes    of a nonlinear solver.
17019b94acceSBarry Smith 
1702fee21e36SBarry Smith    Collective on SNES
1703fee21e36SBarry Smith 
1704c7afd0dbSLois Curfman McInnes    Input Parameters:
170570e92668SMatthew Knepley .  snes - the SNES context
1706c7afd0dbSLois Curfman McInnes 
1707272ac6f2SLois Curfman McInnes    Notes:
1708272ac6f2SLois Curfman McInnes    For basic use of the SNES solvers the user need not explicitly call
1709272ac6f2SLois Curfman McInnes    SNESSetUp(), since these actions will automatically occur during
1710272ac6f2SLois Curfman McInnes    the call to SNESSolve().  However, if one wishes to control this
1711272ac6f2SLois Curfman McInnes    phase separately, SNESSetUp() should be called after SNESCreate()
1712272ac6f2SLois Curfman McInnes    and optional routines of the form SNESSetXXX(), but before SNESSolve().
1713272ac6f2SLois Curfman McInnes 
171436851e7fSLois Curfman McInnes    Level: advanced
171536851e7fSLois Curfman McInnes 
17169b94acceSBarry Smith .keywords: SNES, nonlinear, setup
17179b94acceSBarry Smith 
17189b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy()
17199b94acceSBarry Smith @*/
17207087cfbeSBarry Smith PetscErrorCode  SNESSetUp(SNES snes)
17219b94acceSBarry Smith {
1722dfbe8321SBarry Smith   PetscErrorCode ierr;
17233a40ed3dSBarry Smith 
17243a40ed3dSBarry Smith   PetscFunctionBegin;
17250700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
17264dc4c822SBarry Smith   if (snes->setupcalled) PetscFunctionReturn(0);
17279b94acceSBarry Smith 
17287adad957SLisandro Dalcin   if (!((PetscObject)snes)->type_name) {
172985385478SLisandro Dalcin     ierr = SNESSetType(snes,SNESLS);CHKERRQ(ierr);
173085385478SLisandro Dalcin   }
173185385478SLisandro Dalcin 
1732c9b9fda1SJed Brown   if (!snes->vec_func) {
1733c9b9fda1SJed Brown     if (snes->vec_rhs) {
1734c9b9fda1SJed Brown       ierr = VecDuplicate(snes->vec_rhs,&snes->vec_func);CHKERRQ(ierr);
1735c9b9fda1SJed Brown     } else if (snes->vec_sol) {
1736c9b9fda1SJed Brown       ierr = VecDuplicate(snes->vec_sol,&snes->vec_func);CHKERRQ(ierr);
1737c9b9fda1SJed Brown     } else if (snes->dm) {
1738efd51863SBarry Smith       ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr);
1739efd51863SBarry Smith     }
1740c9b9fda1SJed Brown   }
174117186662SBarry Smith   if (snes->vec_func == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector");
174258c9b817SLisandro Dalcin   if (snes->vec_rhs  == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be right hand side vector");
174358c9b817SLisandro Dalcin 
174458c9b817SLisandro Dalcin   if (!snes->vec_sol_update /* && snes->vec_sol */) {
174558c9b817SLisandro Dalcin     ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr);
174658c9b817SLisandro Dalcin     ierr = PetscLogObjectParent(snes,snes->vec_sol_update);CHKERRQ(ierr);
174758c9b817SLisandro Dalcin   }
174858c9b817SLisandro Dalcin 
1749ef8dffc7SBarry Smith   if (!snes->ops->computejacobian && snes->dm) {
1750ef8dffc7SBarry Smith     Mat J;
1751ef8dffc7SBarry Smith     ierr = DMGetMatrix(snes->dm,MATAIJ,&J);CHKERRQ(ierr);
1752cab2e9ccSBarry Smith     ierr = SNESSetJacobian(snes,J,J,SNESDMComputeJacobian,PETSC_NULL);CHKERRQ(ierr);
1753cab2e9ccSBarry Smith     ierr = MatDestroy(&J);CHKERRQ(ierr);
17542ef29e96SBarry Smith   } else if (!snes->jacobian && snes->ops->computejacobian == MatMFFDComputeJacobian) {
1755a8248277SBarry Smith     Mat J;
1756a8248277SBarry Smith     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
1757a8248277SBarry Smith     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
1758a8248277SBarry Smith     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
1759a8248277SBarry Smith     ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,snes->funP);CHKERRQ(ierr);
1760a8248277SBarry Smith     ierr = MatDestroy(&J);CHKERRQ(ierr);
1761a8248277SBarry Smith   } else if (snes->dm && snes->mf_operator && !snes->jacobian_pre && !snes->jacobian) {
1762a8248277SBarry Smith     Mat J,B;
1763a8248277SBarry Smith     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
1764a8248277SBarry Smith     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
1765a8248277SBarry Smith     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
1766a8248277SBarry Smith     ierr = DMGetMatrix(snes->dm,MATAIJ,&B);CHKERRQ(ierr);
1767a8248277SBarry Smith     ierr = SNESSetJacobian(snes,J,B,SNESDMComputeJacobian,snes->funP);CHKERRQ(ierr);
1768a8248277SBarry Smith     ierr = MatDestroy(&J);CHKERRQ(ierr);
1769a8248277SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
1770efd51863SBarry Smith   } else if (snes->dm && !snes->jacobian_pre){
1771efd51863SBarry Smith     Mat J;
1772efd51863SBarry Smith     ierr = DMGetMatrix(snes->dm,MATAIJ,&J);CHKERRQ(ierr);
17733cbb28f5SBarry Smith     ierr = SNESSetJacobian(snes,J,J,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);
1774efd51863SBarry Smith     ierr = MatDestroy(&J);CHKERRQ(ierr);
1775ef8dffc7SBarry Smith   }
1776cfaf3a74SBarry Smith   if (!snes->ops->computefunction && !snes->dm) SETERRQ(((PetscObject)snes)->comm,PETSC_ERR_ARG_WRONGSTATE,"Must provide a residual function with SNESSetFunction() or call SNESSetDM()");
1777c9b9fda1SJed Brown   if (!snes->vec_func) SETERRQ(((PetscObject)snes)->comm,PETSC_ERR_ARG_WRONGSTATE,"Must provide a vector when calling SNESSetFunction() or call SNESSetDM()");
1778efd51863SBarry Smith 
1779b710008aSBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes, &snes->ksp);CHKERRQ(ierr);}
1780b710008aSBarry Smith 
1781d25893d9SBarry Smith   if (snes->ops->usercompute && !snes->user) {
1782d25893d9SBarry Smith     ierr = (*snes->ops->usercompute)(snes,(void**)&snes->user);CHKERRQ(ierr);
1783d25893d9SBarry Smith   }
1784d25893d9SBarry Smith 
1785410397dcSLisandro Dalcin   if (snes->ops->setup) {
1786410397dcSLisandro Dalcin     ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr);
1787410397dcSLisandro Dalcin   }
178858c9b817SLisandro Dalcin 
17897aaed0d8SBarry Smith   snes->setupcalled = PETSC_TRUE;
17903a40ed3dSBarry Smith   PetscFunctionReturn(0);
17919b94acceSBarry Smith }
17929b94acceSBarry Smith 
17934a2ae208SSatish Balay #undef __FUNCT__
179437596af1SLisandro Dalcin #define __FUNCT__ "SNESReset"
179537596af1SLisandro Dalcin /*@
179637596af1SLisandro Dalcin    SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats
179737596af1SLisandro Dalcin 
179837596af1SLisandro Dalcin    Collective on SNES
179937596af1SLisandro Dalcin 
180037596af1SLisandro Dalcin    Input Parameter:
180137596af1SLisandro Dalcin .  snes - iterative context obtained from SNESCreate()
180237596af1SLisandro Dalcin 
1803d25893d9SBarry Smith    Level: intermediate
1804d25893d9SBarry Smith 
1805d25893d9SBarry Smith    Notes: Also calls the application context destroy routine set with SNESSetComputeApplicationContext()
180637596af1SLisandro Dalcin 
180737596af1SLisandro Dalcin .keywords: SNES, destroy
180837596af1SLisandro Dalcin 
180937596af1SLisandro Dalcin .seealso: SNESCreate(), SNESSetUp(), SNESSolve()
181037596af1SLisandro Dalcin @*/
181137596af1SLisandro Dalcin PetscErrorCode  SNESReset(SNES snes)
181237596af1SLisandro Dalcin {
181337596af1SLisandro Dalcin   PetscErrorCode ierr;
181437596af1SLisandro Dalcin 
181537596af1SLisandro Dalcin   PetscFunctionBegin;
181637596af1SLisandro Dalcin   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1817d25893d9SBarry Smith   if (snes->ops->userdestroy && snes->user) {
1818d25893d9SBarry Smith     ierr       = (*snes->ops->userdestroy)((void**)&snes->user);CHKERRQ(ierr);
1819d25893d9SBarry Smith     snes->user = PETSC_NULL;
1820d25893d9SBarry Smith   }
18218a23116dSBarry Smith   if (snes->pc) {
18228a23116dSBarry Smith     ierr = SNESReset(snes->pc);CHKERRQ(ierr);
18238a23116dSBarry Smith   }
18248a23116dSBarry Smith 
182537596af1SLisandro Dalcin   if (snes->ops->reset) {
182637596af1SLisandro Dalcin     ierr = (*snes->ops->reset)(snes);CHKERRQ(ierr);
182737596af1SLisandro Dalcin   }
182837596af1SLisandro Dalcin   if (snes->ksp) {ierr = KSPReset(snes->ksp);CHKERRQ(ierr);}
18296bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
18306bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
18316bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_sol_update);CHKERRQ(ierr);
18326bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr);
18336bf464f9SBarry Smith   ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
18346bf464f9SBarry Smith   ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
183537596af1SLisandro Dalcin   if (snes->work) {ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr);}
183637596af1SLisandro Dalcin   if (snes->vwork) {ierr = VecDestroyVecs(snes->nvwork,&snes->vwork);CHKERRQ(ierr);}
183737596af1SLisandro Dalcin   snes->nwork = snes->nvwork = 0;
183837596af1SLisandro Dalcin   snes->setupcalled = PETSC_FALSE;
183937596af1SLisandro Dalcin   PetscFunctionReturn(0);
184037596af1SLisandro Dalcin }
184137596af1SLisandro Dalcin 
184237596af1SLisandro Dalcin #undef __FUNCT__
18434a2ae208SSatish Balay #define __FUNCT__ "SNESDestroy"
184452baeb72SSatish Balay /*@
18459b94acceSBarry Smith    SNESDestroy - Destroys the nonlinear solver context that was created
18469b94acceSBarry Smith    with SNESCreate().
18479b94acceSBarry Smith 
1848c7afd0dbSLois Curfman McInnes    Collective on SNES
1849c7afd0dbSLois Curfman McInnes 
18509b94acceSBarry Smith    Input Parameter:
18519b94acceSBarry Smith .  snes - the SNES context
18529b94acceSBarry Smith 
185336851e7fSLois Curfman McInnes    Level: beginner
185436851e7fSLois Curfman McInnes 
18559b94acceSBarry Smith .keywords: SNES, nonlinear, destroy
18569b94acceSBarry Smith 
185763a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve()
18589b94acceSBarry Smith @*/
18596bf464f9SBarry Smith PetscErrorCode  SNESDestroy(SNES *snes)
18609b94acceSBarry Smith {
18616849ba73SBarry Smith   PetscErrorCode ierr;
18623a40ed3dSBarry Smith 
18633a40ed3dSBarry Smith   PetscFunctionBegin;
18646bf464f9SBarry Smith   if (!*snes) PetscFunctionReturn(0);
18656bf464f9SBarry Smith   PetscValidHeaderSpecific((*snes),SNES_CLASSID,1);
18666bf464f9SBarry Smith   if (--((PetscObject)(*snes))->refct > 0) {*snes = 0; PetscFunctionReturn(0);}
1867d4bb536fSBarry Smith 
18686bf464f9SBarry Smith   ierr = SNESReset((*snes));CHKERRQ(ierr);
18698a23116dSBarry Smith   ierr = SNESDestroy(&(*snes)->pc);CHKERRQ(ierr);
18706b8b9a38SLisandro Dalcin 
1871be0abb6dSBarry Smith   /* if memory was published with AMS then destroy it */
18726bf464f9SBarry Smith   ierr = PetscObjectDepublish((*snes));CHKERRQ(ierr);
18736bf464f9SBarry Smith   if ((*snes)->ops->destroy) {ierr = (*((*snes))->ops->destroy)((*snes));CHKERRQ(ierr);}
18746d4c513bSLisandro Dalcin 
18756bf464f9SBarry Smith   ierr = DMDestroy(&(*snes)->dm);CHKERRQ(ierr);
18766bf464f9SBarry Smith   ierr = KSPDestroy(&(*snes)->ksp);CHKERRQ(ierr);
18776b8b9a38SLisandro Dalcin 
18786bf464f9SBarry Smith   ierr = PetscFree((*snes)->kspconvctx);CHKERRQ(ierr);
18796bf464f9SBarry Smith   if ((*snes)->ops->convergeddestroy) {
18806bf464f9SBarry Smith     ierr = (*(*snes)->ops->convergeddestroy)((*snes)->cnvP);CHKERRQ(ierr);
18816b8b9a38SLisandro Dalcin   }
18826bf464f9SBarry Smith   if ((*snes)->conv_malloc) {
18836bf464f9SBarry Smith     ierr = PetscFree((*snes)->conv_hist);CHKERRQ(ierr);
18846bf464f9SBarry Smith     ierr = PetscFree((*snes)->conv_hist_its);CHKERRQ(ierr);
188558c9b817SLisandro Dalcin   }
1886ea630c6eSPeter Brune   ierr = PetscViewerDestroy(&(*snes)->ls_monitor);CHKERRQ(ierr);
18876bf464f9SBarry Smith   ierr = SNESMonitorCancel((*snes));CHKERRQ(ierr);
1888a79aaaedSSatish Balay   ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr);
18893a40ed3dSBarry Smith  PetscFunctionReturn(0);
18909b94acceSBarry Smith }
18919b94acceSBarry Smith 
18929b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */
18939b94acceSBarry Smith 
18944a2ae208SSatish Balay #undef __FUNCT__
1895a8054027SBarry Smith #define __FUNCT__ "SNESSetLagPreconditioner"
1896a8054027SBarry Smith /*@
1897a8054027SBarry Smith    SNESSetLagPreconditioner - Determines when the preconditioner is rebuilt in the nonlinear solve.
1898a8054027SBarry Smith 
18993f9fe445SBarry Smith    Logically Collective on SNES
1900a8054027SBarry Smith 
1901a8054027SBarry Smith    Input Parameters:
1902a8054027SBarry Smith +  snes - the SNES context
1903a8054027SBarry Smith -  lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
19043b4f5425SBarry Smith          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
1905a8054027SBarry Smith 
1906a8054027SBarry Smith    Options Database Keys:
1907a8054027SBarry Smith .    -snes_lag_preconditioner <lag>
1908a8054027SBarry Smith 
1909a8054027SBarry Smith    Notes:
1910a8054027SBarry Smith    The default is 1
1911a8054027SBarry Smith    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
1912a8054027SBarry Smith    If  -1 is used before the very first nonlinear solve the preconditioner is still built because there is no previous preconditioner to use
1913a8054027SBarry Smith 
1914a8054027SBarry Smith    Level: intermediate
1915a8054027SBarry Smith 
1916a8054027SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
1917a8054027SBarry Smith 
1918e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian()
1919a8054027SBarry Smith 
1920a8054027SBarry Smith @*/
19217087cfbeSBarry Smith PetscErrorCode  SNESSetLagPreconditioner(SNES snes,PetscInt lag)
1922a8054027SBarry Smith {
1923a8054027SBarry Smith   PetscFunctionBegin;
19240700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1925e32f2f54SBarry Smith   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
1926e32f2f54SBarry Smith   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
1927c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,lag,2);
1928a8054027SBarry Smith   snes->lagpreconditioner = lag;
1929a8054027SBarry Smith   PetscFunctionReturn(0);
1930a8054027SBarry Smith }
1931a8054027SBarry Smith 
1932a8054027SBarry Smith #undef __FUNCT__
1933efd51863SBarry Smith #define __FUNCT__ "SNESSetGridSequence"
1934efd51863SBarry Smith /*@
1935efd51863SBarry Smith    SNESSetGridSequence - sets the number of steps of grid sequencing that SNES does
1936efd51863SBarry Smith 
1937efd51863SBarry Smith    Logically Collective on SNES
1938efd51863SBarry Smith 
1939efd51863SBarry Smith    Input Parameters:
1940efd51863SBarry Smith +  snes - the SNES context
1941efd51863SBarry Smith -  steps - the number of refinements to do, defaults to 0
1942efd51863SBarry Smith 
1943efd51863SBarry Smith    Options Database Keys:
1944efd51863SBarry Smith .    -snes_grid_sequence <steps>
1945efd51863SBarry Smith 
1946efd51863SBarry Smith    Level: intermediate
1947efd51863SBarry Smith 
1948efd51863SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
1949efd51863SBarry Smith 
1950efd51863SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian()
1951efd51863SBarry Smith 
1952efd51863SBarry Smith @*/
1953efd51863SBarry Smith PetscErrorCode  SNESSetGridSequence(SNES snes,PetscInt steps)
1954efd51863SBarry Smith {
1955efd51863SBarry Smith   PetscFunctionBegin;
1956efd51863SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1957efd51863SBarry Smith   PetscValidLogicalCollectiveInt(snes,steps,2);
1958efd51863SBarry Smith   snes->gridsequence = steps;
1959efd51863SBarry Smith   PetscFunctionReturn(0);
1960efd51863SBarry Smith }
1961efd51863SBarry Smith 
1962efd51863SBarry Smith #undef __FUNCT__
1963a8054027SBarry Smith #define __FUNCT__ "SNESGetLagPreconditioner"
1964a8054027SBarry Smith /*@
1965a8054027SBarry Smith    SNESGetLagPreconditioner - Indicates how often the preconditioner is rebuilt
1966a8054027SBarry Smith 
19673f9fe445SBarry Smith    Not Collective
1968a8054027SBarry Smith 
1969a8054027SBarry Smith    Input Parameter:
1970a8054027SBarry Smith .  snes - the SNES context
1971a8054027SBarry Smith 
1972a8054027SBarry Smith    Output Parameter:
1973a8054027SBarry Smith .   lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
19743b4f5425SBarry Smith          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
1975a8054027SBarry Smith 
1976a8054027SBarry Smith    Options Database Keys:
1977a8054027SBarry Smith .    -snes_lag_preconditioner <lag>
1978a8054027SBarry Smith 
1979a8054027SBarry Smith    Notes:
1980a8054027SBarry Smith    The default is 1
1981a8054027SBarry Smith    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
1982a8054027SBarry Smith 
1983a8054027SBarry Smith    Level: intermediate
1984a8054027SBarry Smith 
1985a8054027SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
1986a8054027SBarry Smith 
1987a8054027SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagPreconditioner()
1988a8054027SBarry Smith 
1989a8054027SBarry Smith @*/
19907087cfbeSBarry Smith PetscErrorCode  SNESGetLagPreconditioner(SNES snes,PetscInt *lag)
1991a8054027SBarry Smith {
1992a8054027SBarry Smith   PetscFunctionBegin;
19930700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1994a8054027SBarry Smith   *lag = snes->lagpreconditioner;
1995a8054027SBarry Smith   PetscFunctionReturn(0);
1996a8054027SBarry Smith }
1997a8054027SBarry Smith 
1998a8054027SBarry Smith #undef __FUNCT__
1999e35cf81dSBarry Smith #define __FUNCT__ "SNESSetLagJacobian"
2000e35cf81dSBarry Smith /*@
2001e35cf81dSBarry Smith    SNESSetLagJacobian - Determines when the Jacobian is rebuilt in the nonlinear solve. See SNESSetLagPreconditioner() for determining how
2002e35cf81dSBarry Smith      often the preconditioner is rebuilt.
2003e35cf81dSBarry Smith 
20043f9fe445SBarry Smith    Logically Collective on SNES
2005e35cf81dSBarry Smith 
2006e35cf81dSBarry Smith    Input Parameters:
2007e35cf81dSBarry Smith +  snes - the SNES context
2008e35cf81dSBarry Smith -  lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
2009fe3ffe1eSBarry Smith          the Jacobian is built etc. -2 means rebuild at next chance but then never again
2010e35cf81dSBarry Smith 
2011e35cf81dSBarry Smith    Options Database Keys:
2012e35cf81dSBarry Smith .    -snes_lag_jacobian <lag>
2013e35cf81dSBarry Smith 
2014e35cf81dSBarry Smith    Notes:
2015e35cf81dSBarry Smith    The default is 1
2016e35cf81dSBarry Smith    The Jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
2017fe3ffe1eSBarry Smith    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
2018fe3ffe1eSBarry Smith    at the next Newton step but never again (unless it is reset to another value)
2019e35cf81dSBarry Smith 
2020e35cf81dSBarry Smith    Level: intermediate
2021e35cf81dSBarry Smith 
2022e35cf81dSBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
2023e35cf81dSBarry Smith 
2024e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagPreconditioner(), SNESGetLagJacobian()
2025e35cf81dSBarry Smith 
2026e35cf81dSBarry Smith @*/
20277087cfbeSBarry Smith PetscErrorCode  SNESSetLagJacobian(SNES snes,PetscInt lag)
2028e35cf81dSBarry Smith {
2029e35cf81dSBarry Smith   PetscFunctionBegin;
20300700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2031e32f2f54SBarry Smith   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
2032e32f2f54SBarry Smith   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
2033c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,lag,2);
2034e35cf81dSBarry Smith   snes->lagjacobian = lag;
2035e35cf81dSBarry Smith   PetscFunctionReturn(0);
2036e35cf81dSBarry Smith }
2037e35cf81dSBarry Smith 
2038e35cf81dSBarry Smith #undef __FUNCT__
2039e35cf81dSBarry Smith #define __FUNCT__ "SNESGetLagJacobian"
2040e35cf81dSBarry Smith /*@
2041e35cf81dSBarry Smith    SNESGetLagJacobian - Indicates how often the Jacobian is rebuilt. See SNESGetLagPreconditioner() to determine when the preconditioner is rebuilt
2042e35cf81dSBarry Smith 
20433f9fe445SBarry Smith    Not Collective
2044e35cf81dSBarry Smith 
2045e35cf81dSBarry Smith    Input Parameter:
2046e35cf81dSBarry Smith .  snes - the SNES context
2047e35cf81dSBarry Smith 
2048e35cf81dSBarry Smith    Output Parameter:
2049e35cf81dSBarry Smith .   lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
2050e35cf81dSBarry Smith          the Jacobian is built etc.
2051e35cf81dSBarry Smith 
2052e35cf81dSBarry Smith    Options Database Keys:
2053e35cf81dSBarry Smith .    -snes_lag_jacobian <lag>
2054e35cf81dSBarry Smith 
2055e35cf81dSBarry Smith    Notes:
2056e35cf81dSBarry Smith    The default is 1
2057e35cf81dSBarry Smith    The jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
2058e35cf81dSBarry Smith 
2059e35cf81dSBarry Smith    Level: intermediate
2060e35cf81dSBarry Smith 
2061e35cf81dSBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
2062e35cf81dSBarry Smith 
2063e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagJacobian(), SNESSetLagPreconditioner(), SNESGetLagPreconditioner()
2064e35cf81dSBarry Smith 
2065e35cf81dSBarry Smith @*/
20667087cfbeSBarry Smith PetscErrorCode  SNESGetLagJacobian(SNES snes,PetscInt *lag)
2067e35cf81dSBarry Smith {
2068e35cf81dSBarry Smith   PetscFunctionBegin;
20690700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2070e35cf81dSBarry Smith   *lag = snes->lagjacobian;
2071e35cf81dSBarry Smith   PetscFunctionReturn(0);
2072e35cf81dSBarry Smith }
2073e35cf81dSBarry Smith 
2074e35cf81dSBarry Smith #undef __FUNCT__
20754a2ae208SSatish Balay #define __FUNCT__ "SNESSetTolerances"
20769b94acceSBarry Smith /*@
2077d7a720efSLois Curfman McInnes    SNESSetTolerances - Sets various parameters used in convergence tests.
20789b94acceSBarry Smith 
20793f9fe445SBarry Smith    Logically Collective on SNES
2080c7afd0dbSLois Curfman McInnes 
20819b94acceSBarry Smith    Input Parameters:
2082c7afd0dbSLois Curfman McInnes +  snes - the SNES context
208370441072SBarry Smith .  abstol - absolute convergence tolerance
208433174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
208533174efeSLois Curfman McInnes .  stol -  convergence tolerance in terms of the norm
208633174efeSLois Curfman McInnes            of the change in the solution between steps
208733174efeSLois Curfman McInnes .  maxit - maximum number of iterations
2088c7afd0dbSLois Curfman McInnes -  maxf - maximum number of function evaluations
2089fee21e36SBarry Smith 
209033174efeSLois Curfman McInnes    Options Database Keys:
209170441072SBarry Smith +    -snes_atol <abstol> - Sets abstol
2092c7afd0dbSLois Curfman McInnes .    -snes_rtol <rtol> - Sets rtol
2093c7afd0dbSLois Curfman McInnes .    -snes_stol <stol> - Sets stol
2094c7afd0dbSLois Curfman McInnes .    -snes_max_it <maxit> - Sets maxit
2095c7afd0dbSLois Curfman McInnes -    -snes_max_funcs <maxf> - Sets maxf
20969b94acceSBarry Smith 
2097d7a720efSLois Curfman McInnes    Notes:
20989b94acceSBarry Smith    The default maximum number of iterations is 50.
20999b94acceSBarry Smith    The default maximum number of function evaluations is 1000.
21009b94acceSBarry Smith 
210136851e7fSLois Curfman McInnes    Level: intermediate
210236851e7fSLois Curfman McInnes 
210333174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances
21049b94acceSBarry Smith 
21052492ecdbSBarry Smith .seealso: SNESSetTrustRegionTolerance()
21069b94acceSBarry Smith @*/
21077087cfbeSBarry Smith PetscErrorCode  SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf)
21089b94acceSBarry Smith {
21093a40ed3dSBarry Smith   PetscFunctionBegin;
21100700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2111c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,abstol,2);
2112c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol,3);
2113c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,stol,4);
2114c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxit,5);
2115c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxf,6);
2116c5eb9154SBarry Smith 
2117ab54825eSJed Brown   if (abstol != PETSC_DEFAULT) {
2118ab54825eSJed Brown     if (abstol < 0.0) SETERRQ1(((PetscObject)snes)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Absolute tolerance %G must be non-negative",abstol);
2119ab54825eSJed Brown     snes->abstol = abstol;
2120ab54825eSJed Brown   }
2121ab54825eSJed Brown   if (rtol != PETSC_DEFAULT) {
2122ab54825eSJed Brown     if (rtol < 0.0 || 1.0 <= rtol) SETERRQ1(((PetscObject)snes)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Relative tolerance %G must be non-negative and less than 1.0",rtol);
2123ab54825eSJed Brown     snes->rtol = rtol;
2124ab54825eSJed Brown   }
2125ab54825eSJed Brown   if (stol != PETSC_DEFAULT) {
2126ab54825eSJed Brown     if (stol < 0.0) SETERRQ1(((PetscObject)snes)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Step tolerance %G must be non-negative",stol);
2127ab54825eSJed Brown     snes->xtol = stol;
2128ab54825eSJed Brown   }
2129ab54825eSJed Brown   if (maxit != PETSC_DEFAULT) {
2130ab54825eSJed Brown     if (maxit < 0) SETERRQ1(((PetscObject)snes)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of iterations %D must be non-negative",maxit);
2131ab54825eSJed Brown     snes->max_its = maxit;
2132ab54825eSJed Brown   }
2133ab54825eSJed Brown   if (maxf != PETSC_DEFAULT) {
2134ab54825eSJed Brown     if (maxf < 0) SETERRQ1(((PetscObject)snes)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of function evaluations %D must be non-negative",maxf);
2135ab54825eSJed Brown     snes->max_funcs = maxf;
2136ab54825eSJed Brown   }
21373a40ed3dSBarry Smith   PetscFunctionReturn(0);
21389b94acceSBarry Smith }
21399b94acceSBarry Smith 
21404a2ae208SSatish Balay #undef __FUNCT__
21414a2ae208SSatish Balay #define __FUNCT__ "SNESGetTolerances"
21429b94acceSBarry Smith /*@
214333174efeSLois Curfman McInnes    SNESGetTolerances - Gets various parameters used in convergence tests.
214433174efeSLois Curfman McInnes 
2145c7afd0dbSLois Curfman McInnes    Not Collective
2146c7afd0dbSLois Curfman McInnes 
214733174efeSLois Curfman McInnes    Input Parameters:
2148c7afd0dbSLois Curfman McInnes +  snes - the SNES context
214985385478SLisandro Dalcin .  atol - absolute convergence tolerance
215033174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
215133174efeSLois Curfman McInnes .  stol -  convergence tolerance in terms of the norm
215233174efeSLois Curfman McInnes            of the change in the solution between steps
215333174efeSLois Curfman McInnes .  maxit - maximum number of iterations
2154c7afd0dbSLois Curfman McInnes -  maxf - maximum number of function evaluations
2155fee21e36SBarry Smith 
215633174efeSLois Curfman McInnes    Notes:
215733174efeSLois Curfman McInnes    The user can specify PETSC_NULL for any parameter that is not needed.
215833174efeSLois Curfman McInnes 
215936851e7fSLois Curfman McInnes    Level: intermediate
216036851e7fSLois Curfman McInnes 
216133174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances
216233174efeSLois Curfman McInnes 
216333174efeSLois Curfman McInnes .seealso: SNESSetTolerances()
216433174efeSLois Curfman McInnes @*/
21657087cfbeSBarry Smith PetscErrorCode  SNESGetTolerances(SNES snes,PetscReal *atol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf)
216633174efeSLois Curfman McInnes {
21673a40ed3dSBarry Smith   PetscFunctionBegin;
21680700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
216985385478SLisandro Dalcin   if (atol)  *atol  = snes->abstol;
217033174efeSLois Curfman McInnes   if (rtol)  *rtol  = snes->rtol;
217133174efeSLois Curfman McInnes   if (stol)  *stol  = snes->xtol;
217233174efeSLois Curfman McInnes   if (maxit) *maxit = snes->max_its;
217333174efeSLois Curfman McInnes   if (maxf)  *maxf  = snes->max_funcs;
21743a40ed3dSBarry Smith   PetscFunctionReturn(0);
217533174efeSLois Curfman McInnes }
217633174efeSLois Curfman McInnes 
21774a2ae208SSatish Balay #undef __FUNCT__
21784a2ae208SSatish Balay #define __FUNCT__ "SNESSetTrustRegionTolerance"
217933174efeSLois Curfman McInnes /*@
21809b94acceSBarry Smith    SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance.
21819b94acceSBarry Smith 
21823f9fe445SBarry Smith    Logically Collective on SNES
2183fee21e36SBarry Smith 
2184c7afd0dbSLois Curfman McInnes    Input Parameters:
2185c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2186c7afd0dbSLois Curfman McInnes -  tol - tolerance
2187c7afd0dbSLois Curfman McInnes 
21889b94acceSBarry Smith    Options Database Key:
2189c7afd0dbSLois Curfman McInnes .  -snes_trtol <tol> - Sets tol
21909b94acceSBarry Smith 
219136851e7fSLois Curfman McInnes    Level: intermediate
219236851e7fSLois Curfman McInnes 
21939b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance
21949b94acceSBarry Smith 
21952492ecdbSBarry Smith .seealso: SNESSetTolerances()
21969b94acceSBarry Smith @*/
21977087cfbeSBarry Smith PetscErrorCode  SNESSetTrustRegionTolerance(SNES snes,PetscReal tol)
21989b94acceSBarry Smith {
21993a40ed3dSBarry Smith   PetscFunctionBegin;
22000700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2201c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,tol,2);
22029b94acceSBarry Smith   snes->deltatol = tol;
22033a40ed3dSBarry Smith   PetscFunctionReturn(0);
22049b94acceSBarry Smith }
22059b94acceSBarry Smith 
2206df9fa365SBarry Smith /*
2207df9fa365SBarry Smith    Duplicate the lg monitors for SNES from KSP; for some reason with
2208df9fa365SBarry Smith    dynamic libraries things don't work under Sun4 if we just use
2209df9fa365SBarry Smith    macros instead of functions
2210df9fa365SBarry Smith */
22114a2ae208SSatish Balay #undef __FUNCT__
2212a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorLG"
22137087cfbeSBarry Smith PetscErrorCode  SNESMonitorLG(SNES snes,PetscInt it,PetscReal norm,void *ctx)
2214ce1608b8SBarry Smith {
2215dfbe8321SBarry Smith   PetscErrorCode ierr;
2216ce1608b8SBarry Smith 
2217ce1608b8SBarry Smith   PetscFunctionBegin;
22180700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2219a6570f20SBarry Smith   ierr = KSPMonitorLG((KSP)snes,it,norm,ctx);CHKERRQ(ierr);
2220ce1608b8SBarry Smith   PetscFunctionReturn(0);
2221ce1608b8SBarry Smith }
2222ce1608b8SBarry Smith 
22234a2ae208SSatish Balay #undef __FUNCT__
2224a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorLGCreate"
22257087cfbeSBarry Smith PetscErrorCode  SNESMonitorLGCreate(const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *draw)
2226df9fa365SBarry Smith {
2227dfbe8321SBarry Smith   PetscErrorCode ierr;
2228df9fa365SBarry Smith 
2229df9fa365SBarry Smith   PetscFunctionBegin;
2230a6570f20SBarry Smith   ierr = KSPMonitorLGCreate(host,label,x,y,m,n,draw);CHKERRQ(ierr);
2231df9fa365SBarry Smith   PetscFunctionReturn(0);
2232df9fa365SBarry Smith }
2233df9fa365SBarry Smith 
22344a2ae208SSatish Balay #undef __FUNCT__
2235a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorLGDestroy"
22366bf464f9SBarry Smith PetscErrorCode  SNESMonitorLGDestroy(PetscDrawLG *draw)
2237df9fa365SBarry Smith {
2238dfbe8321SBarry Smith   PetscErrorCode ierr;
2239df9fa365SBarry Smith 
2240df9fa365SBarry Smith   PetscFunctionBegin;
2241a6570f20SBarry Smith   ierr = KSPMonitorLGDestroy(draw);CHKERRQ(ierr);
2242df9fa365SBarry Smith   PetscFunctionReturn(0);
2243df9fa365SBarry Smith }
2244df9fa365SBarry Smith 
22457087cfbeSBarry Smith extern PetscErrorCode  SNESMonitorRange_Private(SNES,PetscInt,PetscReal*);
2246b271bb04SBarry Smith #undef __FUNCT__
2247b271bb04SBarry Smith #define __FUNCT__ "SNESMonitorLGRange"
22487087cfbeSBarry Smith PetscErrorCode  SNESMonitorLGRange(SNES snes,PetscInt n,PetscReal rnorm,void *monctx)
2249b271bb04SBarry Smith {
2250b271bb04SBarry Smith   PetscDrawLG      lg;
2251b271bb04SBarry Smith   PetscErrorCode   ierr;
2252b271bb04SBarry Smith   PetscReal        x,y,per;
2253b271bb04SBarry Smith   PetscViewer      v = (PetscViewer)monctx;
2254b271bb04SBarry Smith   static PetscReal prev; /* should be in the context */
2255b271bb04SBarry Smith   PetscDraw        draw;
2256b271bb04SBarry Smith   PetscFunctionBegin;
2257b271bb04SBarry Smith   if (!monctx) {
2258b271bb04SBarry Smith     MPI_Comm    comm;
2259b271bb04SBarry Smith 
2260b271bb04SBarry Smith     ierr   = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr);
2261b271bb04SBarry Smith     v      = PETSC_VIEWER_DRAW_(comm);
2262b271bb04SBarry Smith   }
2263b271bb04SBarry Smith   ierr   = PetscViewerDrawGetDrawLG(v,0,&lg);CHKERRQ(ierr);
2264b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
2265b271bb04SBarry Smith   ierr   = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
2266b271bb04SBarry Smith   ierr   = PetscDrawSetTitle(draw,"Residual norm");CHKERRQ(ierr);
2267b271bb04SBarry Smith   x = (PetscReal) n;
2268b271bb04SBarry Smith   if (rnorm > 0.0) y = log10(rnorm); else y = -15.0;
2269b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
2270b271bb04SBarry Smith   if (n < 20 || !(n % 5)) {
2271b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
2272b271bb04SBarry Smith   }
2273b271bb04SBarry Smith 
2274b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,1,&lg);CHKERRQ(ierr);
2275b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
2276b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
2277b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"% elemts > .2*max elemt");CHKERRQ(ierr);
2278b271bb04SBarry Smith   ierr =  SNESMonitorRange_Private(snes,n,&per);CHKERRQ(ierr);
2279b271bb04SBarry Smith   x = (PetscReal) n;
2280b271bb04SBarry Smith   y = 100.0*per;
2281b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
2282b271bb04SBarry Smith   if (n < 20 || !(n % 5)) {
2283b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
2284b271bb04SBarry Smith   }
2285b271bb04SBarry Smith 
2286b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,2,&lg);CHKERRQ(ierr);
2287b271bb04SBarry Smith   if (!n) {prev = rnorm;ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
2288b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
2289b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm");CHKERRQ(ierr);
2290b271bb04SBarry Smith   x = (PetscReal) n;
2291b271bb04SBarry Smith   y = (prev - rnorm)/prev;
2292b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
2293b271bb04SBarry Smith   if (n < 20 || !(n % 5)) {
2294b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
2295b271bb04SBarry Smith   }
2296b271bb04SBarry Smith 
2297b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,3,&lg);CHKERRQ(ierr);
2298b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
2299b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
2300b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr);
2301b271bb04SBarry Smith   x = (PetscReal) n;
2302b271bb04SBarry Smith   y = (prev - rnorm)/(prev*per);
2303b271bb04SBarry Smith   if (n > 2) { /*skip initial crazy value */
2304b271bb04SBarry Smith     ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
2305b271bb04SBarry Smith   }
2306b271bb04SBarry Smith   if (n < 20 || !(n % 5)) {
2307b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
2308b271bb04SBarry Smith   }
2309b271bb04SBarry Smith   prev = rnorm;
2310b271bb04SBarry Smith   PetscFunctionReturn(0);
2311b271bb04SBarry Smith }
2312b271bb04SBarry Smith 
2313b271bb04SBarry Smith #undef __FUNCT__
2314b271bb04SBarry Smith #define __FUNCT__ "SNESMonitorLGRangeCreate"
23157087cfbeSBarry Smith PetscErrorCode  SNESMonitorLGRangeCreate(const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *draw)
2316b271bb04SBarry Smith {
2317b271bb04SBarry Smith   PetscErrorCode ierr;
2318b271bb04SBarry Smith 
2319b271bb04SBarry Smith   PetscFunctionBegin;
2320b271bb04SBarry Smith   ierr = KSPMonitorLGCreate(host,label,x,y,m,n,draw);CHKERRQ(ierr);
2321b271bb04SBarry Smith   PetscFunctionReturn(0);
2322b271bb04SBarry Smith }
2323b271bb04SBarry Smith 
2324b271bb04SBarry Smith #undef __FUNCT__
2325b271bb04SBarry Smith #define __FUNCT__ "SNESMonitorLGRangeDestroy"
23266bf464f9SBarry Smith PetscErrorCode  SNESMonitorLGRangeDestroy(PetscDrawLG *draw)
2327b271bb04SBarry Smith {
2328b271bb04SBarry Smith   PetscErrorCode ierr;
2329b271bb04SBarry Smith 
2330b271bb04SBarry Smith   PetscFunctionBegin;
2331b271bb04SBarry Smith   ierr = KSPMonitorLGDestroy(draw);CHKERRQ(ierr);
2332b271bb04SBarry Smith   PetscFunctionReturn(0);
2333b271bb04SBarry Smith }
2334b271bb04SBarry Smith 
23357a03ce2fSLisandro Dalcin #undef __FUNCT__
23367a03ce2fSLisandro Dalcin #define __FUNCT__ "SNESMonitor"
2337228d79bcSJed Brown /*@
2338228d79bcSJed Brown    SNESMonitor - runs the user provided monitor routines, if they exist
2339228d79bcSJed Brown 
2340228d79bcSJed Brown    Collective on SNES
2341228d79bcSJed Brown 
2342228d79bcSJed Brown    Input Parameters:
2343228d79bcSJed Brown +  snes - nonlinear solver context obtained from SNESCreate()
2344228d79bcSJed Brown .  iter - iteration number
2345228d79bcSJed Brown -  rnorm - relative norm of the residual
2346228d79bcSJed Brown 
2347228d79bcSJed Brown    Notes:
2348228d79bcSJed Brown    This routine is called by the SNES implementations.
2349228d79bcSJed Brown    It does not typically need to be called by the user.
2350228d79bcSJed Brown 
2351228d79bcSJed Brown    Level: developer
2352228d79bcSJed Brown 
2353228d79bcSJed Brown .seealso: SNESMonitorSet()
2354228d79bcSJed Brown @*/
23557a03ce2fSLisandro Dalcin PetscErrorCode  SNESMonitor(SNES snes,PetscInt iter,PetscReal rnorm)
23567a03ce2fSLisandro Dalcin {
23577a03ce2fSLisandro Dalcin   PetscErrorCode ierr;
23587a03ce2fSLisandro Dalcin   PetscInt       i,n = snes->numbermonitors;
23597a03ce2fSLisandro Dalcin 
23607a03ce2fSLisandro Dalcin   PetscFunctionBegin;
23617a03ce2fSLisandro Dalcin   for (i=0; i<n; i++) {
23627a03ce2fSLisandro Dalcin     ierr = (*snes->monitor[i])(snes,iter,rnorm,snes->monitorcontext[i]);CHKERRQ(ierr);
23637a03ce2fSLisandro Dalcin   }
23647a03ce2fSLisandro Dalcin   PetscFunctionReturn(0);
23657a03ce2fSLisandro Dalcin }
23667a03ce2fSLisandro Dalcin 
23679b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */
23689b94acceSBarry Smith 
23694a2ae208SSatish Balay #undef __FUNCT__
2370a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorSet"
23719b94acceSBarry Smith /*@C
2372a6570f20SBarry Smith    SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every
23739b94acceSBarry Smith    iteration of the nonlinear solver to display the iteration's
23749b94acceSBarry Smith    progress.
23759b94acceSBarry Smith 
23763f9fe445SBarry Smith    Logically Collective on SNES
2377fee21e36SBarry Smith 
2378c7afd0dbSLois Curfman McInnes    Input Parameters:
2379c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2380c7afd0dbSLois Curfman McInnes .  func - monitoring routine
2381b8a78c4aSBarry Smith .  mctx - [optional] user-defined context for private data for the
2382e8105e01SRichard Katz           monitor routine (use PETSC_NULL if no context is desired)
2383b3006f0bSLois Curfman McInnes -  monitordestroy - [optional] routine that frees monitor context
2384b3006f0bSLois Curfman McInnes           (may be PETSC_NULL)
23859b94acceSBarry Smith 
2386c7afd0dbSLois Curfman McInnes    Calling sequence of func:
2387a7cc72afSBarry Smith $     int func(SNES snes,PetscInt its, PetscReal norm,void *mctx)
2388c7afd0dbSLois Curfman McInnes 
2389c7afd0dbSLois Curfman McInnes +    snes - the SNES context
2390c7afd0dbSLois Curfman McInnes .    its - iteration number
2391c7afd0dbSLois Curfman McInnes .    norm - 2-norm function value (may be estimated)
239240a0c1c6SLois Curfman McInnes -    mctx - [optional] monitoring context
23939b94acceSBarry Smith 
23949665c990SLois Curfman McInnes    Options Database Keys:
2395a6570f20SBarry Smith +    -snes_monitor        - sets SNESMonitorDefault()
2396a6570f20SBarry Smith .    -snes_monitor_draw    - sets line graph monitor,
2397a6570f20SBarry Smith                             uses SNESMonitorLGCreate()
2398cca6129bSJed Brown -    -snes_monitor_cancel - cancels all monitors that have
2399c7afd0dbSLois Curfman McInnes                             been hardwired into a code by
2400a6570f20SBarry Smith                             calls to SNESMonitorSet(), but
2401c7afd0dbSLois Curfman McInnes                             does not cancel those set via
2402c7afd0dbSLois Curfman McInnes                             the options database.
24039665c990SLois Curfman McInnes 
2404639f9d9dSBarry Smith    Notes:
24056bc08f3fSLois Curfman McInnes    Several different monitoring routines may be set by calling
2406a6570f20SBarry Smith    SNESMonitorSet() multiple times; all will be called in the
24076bc08f3fSLois Curfman McInnes    order in which they were set.
2408639f9d9dSBarry Smith 
2409025f1a04SBarry Smith    Fortran notes: Only a single monitor function can be set for each SNES object
2410025f1a04SBarry Smith 
241136851e7fSLois Curfman McInnes    Level: intermediate
241236851e7fSLois Curfman McInnes 
24139b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor
24149b94acceSBarry Smith 
2415a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorCancel()
24169b94acceSBarry Smith @*/
2417c2efdce3SBarry Smith PetscErrorCode  SNESMonitorSet(SNES snes,PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void**))
24189b94acceSBarry Smith {
2419b90d0a6eSBarry Smith   PetscInt       i;
2420649052a6SBarry Smith   PetscErrorCode ierr;
2421b90d0a6eSBarry Smith 
24223a40ed3dSBarry Smith   PetscFunctionBegin;
24230700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
242417186662SBarry Smith   if (snes->numbermonitors >= MAXSNESMONITORS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set");
2425b90d0a6eSBarry Smith   for (i=0; i<snes->numbermonitors;i++) {
2426649052a6SBarry Smith     if (monitor == snes->monitor[i] && monitordestroy == snes->monitordestroy[i] && mctx == snes->monitorcontext[i]) {
2427649052a6SBarry Smith       if (monitordestroy) {
2428c2efdce3SBarry Smith         ierr = (*monitordestroy)(&mctx);CHKERRQ(ierr);
2429649052a6SBarry Smith       }
2430b90d0a6eSBarry Smith       PetscFunctionReturn(0);
2431b90d0a6eSBarry Smith     }
2432b90d0a6eSBarry Smith   }
2433b90d0a6eSBarry Smith   snes->monitor[snes->numbermonitors]           = monitor;
2434b8a78c4aSBarry Smith   snes->monitordestroy[snes->numbermonitors]    = monitordestroy;
2435639f9d9dSBarry Smith   snes->monitorcontext[snes->numbermonitors++]  = (void*)mctx;
24363a40ed3dSBarry Smith   PetscFunctionReturn(0);
24379b94acceSBarry Smith }
24389b94acceSBarry Smith 
24394a2ae208SSatish Balay #undef __FUNCT__
2440a6570f20SBarry Smith #define __FUNCT__ "SNESMonitorCancel"
24415cd90555SBarry Smith /*@C
2442a6570f20SBarry Smith    SNESMonitorCancel - Clears all the monitor functions for a SNES object.
24435cd90555SBarry Smith 
24443f9fe445SBarry Smith    Logically Collective on SNES
2445c7afd0dbSLois Curfman McInnes 
24465cd90555SBarry Smith    Input Parameters:
24475cd90555SBarry Smith .  snes - the SNES context
24485cd90555SBarry Smith 
24491a480d89SAdministrator    Options Database Key:
2450a6570f20SBarry Smith .  -snes_monitor_cancel - cancels all monitors that have been hardwired
2451a6570f20SBarry Smith     into a code by calls to SNESMonitorSet(), but does not cancel those
2452c7afd0dbSLois Curfman McInnes     set via the options database
24535cd90555SBarry Smith 
24545cd90555SBarry Smith    Notes:
24555cd90555SBarry Smith    There is no way to clear one specific monitor from a SNES object.
24565cd90555SBarry Smith 
245736851e7fSLois Curfman McInnes    Level: intermediate
245836851e7fSLois Curfman McInnes 
24595cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor
24605cd90555SBarry Smith 
2461a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorSet()
24625cd90555SBarry Smith @*/
24637087cfbeSBarry Smith PetscErrorCode  SNESMonitorCancel(SNES snes)
24645cd90555SBarry Smith {
2465d952e501SBarry Smith   PetscErrorCode ierr;
2466d952e501SBarry Smith   PetscInt       i;
2467d952e501SBarry Smith 
24685cd90555SBarry Smith   PetscFunctionBegin;
24690700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2470d952e501SBarry Smith   for (i=0; i<snes->numbermonitors; i++) {
2471d952e501SBarry Smith     if (snes->monitordestroy[i]) {
24723c4aec1bSBarry Smith       ierr = (*snes->monitordestroy[i])(&snes->monitorcontext[i]);CHKERRQ(ierr);
2473d952e501SBarry Smith     }
2474d952e501SBarry Smith   }
24755cd90555SBarry Smith   snes->numbermonitors = 0;
24765cd90555SBarry Smith   PetscFunctionReturn(0);
24775cd90555SBarry Smith }
24785cd90555SBarry Smith 
24794a2ae208SSatish Balay #undef __FUNCT__
24804a2ae208SSatish Balay #define __FUNCT__ "SNESSetConvergenceTest"
24819b94acceSBarry Smith /*@C
24829b94acceSBarry Smith    SNESSetConvergenceTest - Sets the function that is to be used
24839b94acceSBarry Smith    to test for convergence of the nonlinear iterative solution.
24849b94acceSBarry Smith 
24853f9fe445SBarry Smith    Logically Collective on SNES
2486fee21e36SBarry Smith 
2487c7afd0dbSLois Curfman McInnes    Input Parameters:
2488c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2489c7afd0dbSLois Curfman McInnes .  func - routine to test for convergence
24907f7931b9SBarry Smith .  cctx - [optional] context for private data for the convergence routine  (may be PETSC_NULL)
24917f7931b9SBarry Smith -  destroy - [optional] destructor for the context (may be PETSC_NULL; PETSC_NULL_FUNCTION in Fortran)
24929b94acceSBarry Smith 
2493c7afd0dbSLois Curfman McInnes    Calling sequence of func:
249406ee9f85SBarry Smith $     PetscErrorCode func (SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx)
2495c7afd0dbSLois Curfman McInnes 
2496c7afd0dbSLois Curfman McInnes +    snes - the SNES context
249706ee9f85SBarry Smith .    it - current iteration (0 is the first and is before any Newton step)
2498c7afd0dbSLois Curfman McInnes .    cctx - [optional] convergence context
2499184914b5SBarry Smith .    reason - reason for convergence/divergence
2500c7afd0dbSLois Curfman McInnes .    xnorm - 2-norm of current iterate
25014b27c08aSLois Curfman McInnes .    gnorm - 2-norm of current step
25024b27c08aSLois Curfman McInnes -    f - 2-norm of function
25039b94acceSBarry Smith 
250436851e7fSLois Curfman McInnes    Level: advanced
250536851e7fSLois Curfman McInnes 
25069b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test
25079b94acceSBarry Smith 
250885385478SLisandro Dalcin .seealso: SNESDefaultConverged(), SNESSkipConverged()
25099b94acceSBarry Smith @*/
25107087cfbeSBarry Smith PetscErrorCode  SNESSetConvergenceTest(SNES snes,PetscErrorCode (*func)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx,PetscErrorCode (*destroy)(void*))
25119b94acceSBarry Smith {
25127f7931b9SBarry Smith   PetscErrorCode ierr;
25137f7931b9SBarry Smith 
25143a40ed3dSBarry Smith   PetscFunctionBegin;
25150700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
251685385478SLisandro Dalcin   if (!func) func = SNESSkipConverged;
25177f7931b9SBarry Smith   if (snes->ops->convergeddestroy) {
25187f7931b9SBarry Smith     ierr = (*snes->ops->convergeddestroy)(snes->cnvP);CHKERRQ(ierr);
25197f7931b9SBarry Smith   }
252085385478SLisandro Dalcin   snes->ops->converged        = func;
25217f7931b9SBarry Smith   snes->ops->convergeddestroy = destroy;
252285385478SLisandro Dalcin   snes->cnvP                  = cctx;
25233a40ed3dSBarry Smith   PetscFunctionReturn(0);
25249b94acceSBarry Smith }
25259b94acceSBarry Smith 
25264a2ae208SSatish Balay #undef __FUNCT__
25274a2ae208SSatish Balay #define __FUNCT__ "SNESGetConvergedReason"
252852baeb72SSatish Balay /*@
2529184914b5SBarry Smith    SNESGetConvergedReason - Gets the reason the SNES iteration was stopped.
2530184914b5SBarry Smith 
2531184914b5SBarry Smith    Not Collective
2532184914b5SBarry Smith 
2533184914b5SBarry Smith    Input Parameter:
2534184914b5SBarry Smith .  snes - the SNES context
2535184914b5SBarry Smith 
2536184914b5SBarry Smith    Output Parameter:
25374d0a8057SBarry Smith .  reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the
2538184914b5SBarry Smith             manual pages for the individual convergence tests for complete lists
2539184914b5SBarry Smith 
2540184914b5SBarry Smith    Level: intermediate
2541184914b5SBarry Smith 
2542184914b5SBarry Smith    Notes: Can only be called after the call the SNESSolve() is complete.
2543184914b5SBarry Smith 
2544184914b5SBarry Smith .keywords: SNES, nonlinear, set, convergence, test
2545184914b5SBarry Smith 
254685385478SLisandro Dalcin .seealso: SNESSetConvergenceTest(), SNESConvergedReason
2547184914b5SBarry Smith @*/
25487087cfbeSBarry Smith PetscErrorCode  SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason)
2549184914b5SBarry Smith {
2550184914b5SBarry Smith   PetscFunctionBegin;
25510700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
25524482741eSBarry Smith   PetscValidPointer(reason,2);
2553184914b5SBarry Smith   *reason = snes->reason;
2554184914b5SBarry Smith   PetscFunctionReturn(0);
2555184914b5SBarry Smith }
2556184914b5SBarry Smith 
25574a2ae208SSatish Balay #undef __FUNCT__
25584a2ae208SSatish Balay #define __FUNCT__ "SNESSetConvergenceHistory"
2559c9005455SLois Curfman McInnes /*@
2560c9005455SLois Curfman McInnes    SNESSetConvergenceHistory - Sets the array used to hold the convergence history.
2561c9005455SLois Curfman McInnes 
25623f9fe445SBarry Smith    Logically Collective on SNES
2563fee21e36SBarry Smith 
2564c7afd0dbSLois Curfman McInnes    Input Parameters:
2565c7afd0dbSLois Curfman McInnes +  snes - iterative context obtained from SNESCreate()
25668c7482ecSBarry Smith .  a   - array to hold history, this array will contain the function norms computed at each step
2567cd5578b5SBarry Smith .  its - integer array holds the number of linear iterations for each solve.
2568758f92a0SBarry Smith .  na  - size of a and its
256964731454SLois Curfman McInnes -  reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero,
2570758f92a0SBarry Smith            else it continues storing new values for new nonlinear solves after the old ones
2571c7afd0dbSLois Curfman McInnes 
2572308dcc3eSBarry Smith    Notes:
2573308dcc3eSBarry Smith    If 'a' and 'its' are PETSC_NULL then space is allocated for the history. If 'na' PETSC_DECIDE or PETSC_DEFAULT then a
2574308dcc3eSBarry Smith    default array of length 10000 is allocated.
2575308dcc3eSBarry Smith 
2576c9005455SLois Curfman McInnes    This routine is useful, e.g., when running a code for purposes
2577c9005455SLois Curfman McInnes    of accurate performance monitoring, when no I/O should be done
2578c9005455SLois Curfman McInnes    during the section of code that is being timed.
2579c9005455SLois Curfman McInnes 
258036851e7fSLois Curfman McInnes    Level: intermediate
258136851e7fSLois Curfman McInnes 
2582c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history
2583758f92a0SBarry Smith 
258408405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory()
2585758f92a0SBarry Smith 
2586c9005455SLois Curfman McInnes @*/
25877087cfbeSBarry Smith PetscErrorCode  SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscBool  reset)
2588c9005455SLois Curfman McInnes {
2589308dcc3eSBarry Smith   PetscErrorCode ierr;
2590308dcc3eSBarry Smith 
25913a40ed3dSBarry Smith   PetscFunctionBegin;
25920700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
25934482741eSBarry Smith   if (na)  PetscValidScalarPointer(a,2);
2594a562a398SLisandro Dalcin   if (its) PetscValidIntPointer(its,3);
2595308dcc3eSBarry Smith   if (na == PETSC_DECIDE || na == PETSC_DEFAULT || !a) {
2596308dcc3eSBarry Smith     if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2597308dcc3eSBarry Smith     ierr = PetscMalloc(na*sizeof(PetscReal),&a);CHKERRQ(ierr);
2598308dcc3eSBarry Smith     ierr = PetscMalloc(na*sizeof(PetscInt),&its);CHKERRQ(ierr);
2599308dcc3eSBarry Smith     snes->conv_malloc   = PETSC_TRUE;
2600308dcc3eSBarry Smith   }
2601c9005455SLois Curfman McInnes   snes->conv_hist       = a;
2602758f92a0SBarry Smith   snes->conv_hist_its   = its;
2603758f92a0SBarry Smith   snes->conv_hist_max   = na;
2604a12bc48fSLisandro Dalcin   snes->conv_hist_len   = 0;
2605758f92a0SBarry Smith   snes->conv_hist_reset = reset;
2606758f92a0SBarry Smith   PetscFunctionReturn(0);
2607758f92a0SBarry Smith }
2608758f92a0SBarry Smith 
2609308dcc3eSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
2610c6db04a5SJed Brown #include <engine.h>   /* MATLAB include file */
2611c6db04a5SJed Brown #include <mex.h>      /* MATLAB include file */
2612308dcc3eSBarry Smith EXTERN_C_BEGIN
2613308dcc3eSBarry Smith #undef __FUNCT__
2614308dcc3eSBarry Smith #define __FUNCT__ "SNESGetConvergenceHistoryMatlab"
2615308dcc3eSBarry Smith mxArray *SNESGetConvergenceHistoryMatlab(SNES snes)
2616308dcc3eSBarry Smith {
2617308dcc3eSBarry Smith   mxArray        *mat;
2618308dcc3eSBarry Smith   PetscInt       i;
2619308dcc3eSBarry Smith   PetscReal      *ar;
2620308dcc3eSBarry Smith 
2621308dcc3eSBarry Smith   PetscFunctionBegin;
2622308dcc3eSBarry Smith   mat  = mxCreateDoubleMatrix(snes->conv_hist_len,1,mxREAL);
2623308dcc3eSBarry Smith   ar   = (PetscReal*) mxGetData(mat);
2624308dcc3eSBarry Smith   for (i=0; i<snes->conv_hist_len; i++) {
2625308dcc3eSBarry Smith     ar[i] = snes->conv_hist[i];
2626308dcc3eSBarry Smith   }
2627308dcc3eSBarry Smith   PetscFunctionReturn(mat);
2628308dcc3eSBarry Smith }
2629308dcc3eSBarry Smith EXTERN_C_END
2630308dcc3eSBarry Smith #endif
2631308dcc3eSBarry Smith 
2632308dcc3eSBarry Smith 
26334a2ae208SSatish Balay #undef __FUNCT__
26344a2ae208SSatish Balay #define __FUNCT__ "SNESGetConvergenceHistory"
26350c4c9dddSBarry Smith /*@C
2636758f92a0SBarry Smith    SNESGetConvergenceHistory - Gets the array used to hold the convergence history.
2637758f92a0SBarry Smith 
26383f9fe445SBarry Smith    Not Collective
2639758f92a0SBarry Smith 
2640758f92a0SBarry Smith    Input Parameter:
2641758f92a0SBarry Smith .  snes - iterative context obtained from SNESCreate()
2642758f92a0SBarry Smith 
2643758f92a0SBarry Smith    Output Parameters:
2644758f92a0SBarry Smith .  a   - array to hold history
2645758f92a0SBarry Smith .  its - integer array holds the number of linear iterations (or
2646758f92a0SBarry Smith          negative if not converged) for each solve.
2647758f92a0SBarry Smith -  na  - size of a and its
2648758f92a0SBarry Smith 
2649758f92a0SBarry Smith    Notes:
2650758f92a0SBarry Smith     The calling sequence for this routine in Fortran is
2651758f92a0SBarry Smith $   call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr)
2652758f92a0SBarry Smith 
2653758f92a0SBarry Smith    This routine is useful, e.g., when running a code for purposes
2654758f92a0SBarry Smith    of accurate performance monitoring, when no I/O should be done
2655758f92a0SBarry Smith    during the section of code that is being timed.
2656758f92a0SBarry Smith 
2657758f92a0SBarry Smith    Level: intermediate
2658758f92a0SBarry Smith 
2659758f92a0SBarry Smith .keywords: SNES, get, convergence, history
2660758f92a0SBarry Smith 
2661758f92a0SBarry Smith .seealso: SNESSetConvergencHistory()
2662758f92a0SBarry Smith 
2663758f92a0SBarry Smith @*/
26647087cfbeSBarry Smith PetscErrorCode  SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na)
2665758f92a0SBarry Smith {
2666758f92a0SBarry Smith   PetscFunctionBegin;
26670700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2668758f92a0SBarry Smith   if (a)   *a   = snes->conv_hist;
2669758f92a0SBarry Smith   if (its) *its = snes->conv_hist_its;
2670758f92a0SBarry Smith   if (na)  *na  = snes->conv_hist_len;
26713a40ed3dSBarry Smith   PetscFunctionReturn(0);
2672c9005455SLois Curfman McInnes }
2673c9005455SLois Curfman McInnes 
2674e74ef692SMatthew Knepley #undef __FUNCT__
2675e74ef692SMatthew Knepley #define __FUNCT__ "SNESSetUpdate"
2676ac226902SBarry Smith /*@C
267776b2cf59SMatthew Knepley   SNESSetUpdate - Sets the general-purpose update function called
2678eec8f646SJed Brown   at the beginning of every iteration of the nonlinear solve. Specifically
26797e4bb74cSBarry Smith   it is called just before the Jacobian is "evaluated".
268076b2cf59SMatthew Knepley 
26813f9fe445SBarry Smith   Logically Collective on SNES
268276b2cf59SMatthew Knepley 
268376b2cf59SMatthew Knepley   Input Parameters:
268476b2cf59SMatthew Knepley . snes - The nonlinear solver context
268576b2cf59SMatthew Knepley . func - The function
268676b2cf59SMatthew Knepley 
268776b2cf59SMatthew Knepley   Calling sequence of func:
2688b5d30489SBarry Smith . func (SNES snes, PetscInt step);
268976b2cf59SMatthew Knepley 
269076b2cf59SMatthew Knepley . step - The current step of the iteration
269176b2cf59SMatthew Knepley 
2692fe97e370SBarry Smith   Level: advanced
2693fe97e370SBarry Smith 
2694fe97e370SBarry Smith   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()
2695fe97e370SBarry Smith         This is not used by most users.
269676b2cf59SMatthew Knepley 
269776b2cf59SMatthew Knepley .keywords: SNES, update
2698b5d30489SBarry Smith 
269985385478SLisandro Dalcin .seealso SNESDefaultUpdate(), SNESSetJacobian(), SNESSolve()
270076b2cf59SMatthew Knepley @*/
27017087cfbeSBarry Smith PetscErrorCode  SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt))
270276b2cf59SMatthew Knepley {
270376b2cf59SMatthew Knepley   PetscFunctionBegin;
27040700a824SBarry Smith   PetscValidHeaderSpecific(snes, SNES_CLASSID,1);
2705e7788613SBarry Smith   snes->ops->update = func;
270676b2cf59SMatthew Knepley   PetscFunctionReturn(0);
270776b2cf59SMatthew Knepley }
270876b2cf59SMatthew Knepley 
2709e74ef692SMatthew Knepley #undef __FUNCT__
2710e74ef692SMatthew Knepley #define __FUNCT__ "SNESDefaultUpdate"
271176b2cf59SMatthew Knepley /*@
271276b2cf59SMatthew Knepley   SNESDefaultUpdate - The default update function which does nothing.
271376b2cf59SMatthew Knepley 
271476b2cf59SMatthew Knepley   Not collective
271576b2cf59SMatthew Knepley 
271676b2cf59SMatthew Knepley   Input Parameters:
271776b2cf59SMatthew Knepley . snes - The nonlinear solver context
271876b2cf59SMatthew Knepley . step - The current step of the iteration
271976b2cf59SMatthew Knepley 
2720205452f4SMatthew Knepley   Level: intermediate
2721205452f4SMatthew Knepley 
272276b2cf59SMatthew Knepley .keywords: SNES, update
2723a6570f20SBarry Smith .seealso SNESSetUpdate(), SNESDefaultRhsBC(), SNESDefaultShortolutionBC()
272476b2cf59SMatthew Knepley @*/
27257087cfbeSBarry Smith PetscErrorCode  SNESDefaultUpdate(SNES snes, PetscInt step)
272676b2cf59SMatthew Knepley {
272776b2cf59SMatthew Knepley   PetscFunctionBegin;
272876b2cf59SMatthew Knepley   PetscFunctionReturn(0);
272976b2cf59SMatthew Knepley }
273076b2cf59SMatthew Knepley 
27314a2ae208SSatish Balay #undef __FUNCT__
27324a2ae208SSatish Balay #define __FUNCT__ "SNESScaleStep_Private"
27339b94acceSBarry Smith /*
27349b94acceSBarry Smith    SNESScaleStep_Private - Scales a step so that its length is less than the
27359b94acceSBarry Smith    positive parameter delta.
27369b94acceSBarry Smith 
27379b94acceSBarry Smith     Input Parameters:
2738c7afd0dbSLois Curfman McInnes +   snes - the SNES context
27399b94acceSBarry Smith .   y - approximate solution of linear system
27409b94acceSBarry Smith .   fnorm - 2-norm of current function
2741c7afd0dbSLois Curfman McInnes -   delta - trust region size
27429b94acceSBarry Smith 
27439b94acceSBarry Smith     Output Parameters:
2744c7afd0dbSLois Curfman McInnes +   gpnorm - predicted function norm at the new point, assuming local
27459b94acceSBarry Smith     linearization.  The value is zero if the step lies within the trust
27469b94acceSBarry Smith     region, and exceeds zero otherwise.
2747c7afd0dbSLois Curfman McInnes -   ynorm - 2-norm of the step
27489b94acceSBarry Smith 
27499b94acceSBarry Smith     Note:
27504b27c08aSLois Curfman McInnes     For non-trust region methods such as SNESLS, the parameter delta
27519b94acceSBarry Smith     is set to be the maximum allowable step size.
27529b94acceSBarry Smith 
27539b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step
27549b94acceSBarry Smith */
2755dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm)
27569b94acceSBarry Smith {
2757064f8208SBarry Smith   PetscReal      nrm;
2758ea709b57SSatish Balay   PetscScalar    cnorm;
2759dfbe8321SBarry Smith   PetscErrorCode ierr;
27603a40ed3dSBarry Smith 
27613a40ed3dSBarry Smith   PetscFunctionBegin;
27620700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
27630700a824SBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,2);
2764c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,2);
2765184914b5SBarry Smith 
2766064f8208SBarry Smith   ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr);
2767064f8208SBarry Smith   if (nrm > *delta) {
2768064f8208SBarry Smith      nrm = *delta/nrm;
2769064f8208SBarry Smith      *gpnorm = (1.0 - nrm)*(*fnorm);
2770064f8208SBarry Smith      cnorm = nrm;
27712dcb1b2aSMatthew Knepley      ierr = VecScale(y,cnorm);CHKERRQ(ierr);
27729b94acceSBarry Smith      *ynorm = *delta;
27739b94acceSBarry Smith   } else {
27749b94acceSBarry Smith      *gpnorm = 0.0;
2775064f8208SBarry Smith      *ynorm = nrm;
27769b94acceSBarry Smith   }
27773a40ed3dSBarry Smith   PetscFunctionReturn(0);
27789b94acceSBarry Smith }
27799b94acceSBarry Smith 
27804a2ae208SSatish Balay #undef __FUNCT__
27814a2ae208SSatish Balay #define __FUNCT__ "SNESSolve"
27826ce558aeSBarry Smith /*@C
2783f69a0ea3SMatthew Knepley    SNESSolve - Solves a nonlinear system F(x) = b.
2784f69a0ea3SMatthew Knepley    Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX().
27859b94acceSBarry Smith 
2786c7afd0dbSLois Curfman McInnes    Collective on SNES
2787c7afd0dbSLois Curfman McInnes 
2788b2002411SLois Curfman McInnes    Input Parameters:
2789c7afd0dbSLois Curfman McInnes +  snes - the SNES context
27903cebf274SJed Brown .  b - the constant part of the equation F(x) = b, or PETSC_NULL to use zero.
279185385478SLisandro Dalcin -  x - the solution vector.
27929b94acceSBarry Smith 
2793b2002411SLois Curfman McInnes    Notes:
27948ddd3da0SLois Curfman McInnes    The user should initialize the vector,x, with the initial guess
27958ddd3da0SLois Curfman McInnes    for the nonlinear solve prior to calling SNESSolve.  In particular,
27968ddd3da0SLois Curfman McInnes    to employ an initial guess of zero, the user should explicitly set
27978ddd3da0SLois Curfman McInnes    this vector to zero by calling VecSet().
27988ddd3da0SLois Curfman McInnes 
279936851e7fSLois Curfman McInnes    Level: beginner
280036851e7fSLois Curfman McInnes 
28019b94acceSBarry Smith .keywords: SNES, nonlinear, solve
28029b94acceSBarry Smith 
280385385478SLisandro Dalcin .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian()
28049b94acceSBarry Smith @*/
28057087cfbeSBarry Smith PetscErrorCode  SNESSolve(SNES snes,Vec b,Vec x)
28069b94acceSBarry Smith {
2807dfbe8321SBarry Smith   PetscErrorCode ierr;
2808ace3abfcSBarry Smith   PetscBool      flg;
2809eabae89aSBarry Smith   char           filename[PETSC_MAX_PATH_LEN];
2810eabae89aSBarry Smith   PetscViewer    viewer;
2811efd51863SBarry Smith   PetscInt       grid;
2812052efed2SBarry Smith 
28133a40ed3dSBarry Smith   PetscFunctionBegin;
28140700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
28150700a824SBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
2816f69a0ea3SMatthew Knepley   PetscCheckSameComm(snes,1,x,3);
28170700a824SBarry Smith   if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2);
281885385478SLisandro Dalcin   if (b) PetscCheckSameComm(snes,1,b,2);
281985385478SLisandro Dalcin 
2820a8248277SBarry Smith   for (grid=0; grid<snes->gridsequence; grid++) {ierr = PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(((PetscObject)snes)->comm));CHKERRQ(ierr);}
2821efd51863SBarry Smith   for (grid=0; grid<snes->gridsequence+1; grid++) {
2822efd51863SBarry Smith 
282385385478SLisandro Dalcin     /* set solution vector */
2824efd51863SBarry Smith     if (!grid) {ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);}
28256bf464f9SBarry Smith     ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
282685385478SLisandro Dalcin     snes->vec_sol = x;
282785385478SLisandro Dalcin     /* set afine vector if provided */
282885385478SLisandro Dalcin     if (b) { ierr = PetscObjectReference((PetscObject)b);CHKERRQ(ierr); }
28296bf464f9SBarry Smith     ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
283085385478SLisandro Dalcin     snes->vec_rhs = b;
283185385478SLisandro Dalcin 
283270e92668SMatthew Knepley     ierr = SNESSetUp(snes);CHKERRQ(ierr);
28333f149594SLisandro Dalcin 
2834d25893d9SBarry Smith     if (!grid && snes->ops->computeinitialguess) {
2835d25893d9SBarry Smith       ierr = (*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP);CHKERRQ(ierr);
2836d25893d9SBarry Smith     }
2837d25893d9SBarry Smith 
2838abc0a331SBarry Smith     if (snes->conv_hist_reset) snes->conv_hist_len = 0;
283950ffb88aSMatthew Knepley     snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;
2840d5e45103SBarry Smith 
28413f149594SLisandro Dalcin     ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
28424936397dSBarry Smith     ierr = (*snes->ops->solve)(snes);CHKERRQ(ierr);
284385385478SLisandro Dalcin     ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
28444936397dSBarry Smith     if (snes->domainerror){
28454936397dSBarry Smith       snes->reason      = SNES_DIVERGED_FUNCTION_DOMAIN;
28464936397dSBarry Smith       snes->domainerror = PETSC_FALSE;
28474936397dSBarry Smith     }
284817186662SBarry Smith     if (!snes->reason) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason");
28493f149594SLisandro Dalcin 
28507adad957SLisandro Dalcin     ierr = PetscOptionsGetString(((PetscObject)snes)->prefix,"-snes_view",filename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
2851eabae89aSBarry Smith     if (flg && !PetscPreLoadingOn) {
28527adad957SLisandro Dalcin       ierr = PetscViewerASCIIOpen(((PetscObject)snes)->comm,filename,&viewer);CHKERRQ(ierr);
2853eabae89aSBarry Smith       ierr = SNESView(snes,viewer);CHKERRQ(ierr);
28546bf464f9SBarry Smith       ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
2855eabae89aSBarry Smith     }
2856eabae89aSBarry Smith 
285790d69ab7SBarry Smith     flg  = PETSC_FALSE;
2858acfcf0e5SJed Brown     ierr = PetscOptionsGetBool(((PetscObject)snes)->prefix,"-snes_test_local_min",&flg,PETSC_NULL);CHKERRQ(ierr);
2859da9b6338SBarry Smith     if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); }
28605968eb51SBarry Smith     if (snes->printreason) {
2861a8248277SBarry Smith       ierr = PetscViewerASCIIAddTab(PETSC_VIEWER_STDOUT_(((PetscObject)snes)->comm),((PetscObject)snes)->tablevel);CHKERRQ(ierr);
28625968eb51SBarry Smith       if (snes->reason > 0) {
2863a8248277SBarry Smith         ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_(((PetscObject)snes)->comm),"Nonlinear solve converged due to %s\n",SNESConvergedReasons[snes->reason]);CHKERRQ(ierr);
28645968eb51SBarry Smith       } else {
2865a8248277SBarry Smith         ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_(((PetscObject)snes)->comm),"Nonlinear solve did not converge due to %s\n",SNESConvergedReasons[snes->reason]);CHKERRQ(ierr);
28665968eb51SBarry Smith       }
2867a8248277SBarry Smith       ierr = PetscViewerASCIISubtractTab(PETSC_VIEWER_STDOUT_(((PetscObject)snes)->comm),((PetscObject)snes)->tablevel);CHKERRQ(ierr);
28685968eb51SBarry Smith     }
28695968eb51SBarry Smith 
28708501fc72SJed Brown     flg = PETSC_FALSE;
28718501fc72SJed Brown     ierr = PetscOptionsGetString(((PetscObject)snes)->prefix,"-snes_view_solution_vtk",filename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
28728501fc72SJed Brown     if (flg) {
28738501fc72SJed Brown       PetscViewer viewer;
28748501fc72SJed Brown       ierr = PetscViewerCreate(((PetscObject)snes)->comm,&viewer);CHKERRQ(ierr);
28758501fc72SJed Brown       ierr = PetscViewerSetType(viewer,PETSCVIEWERVTK);CHKERRQ(ierr);
28768501fc72SJed Brown       ierr = PetscViewerFileSetName(viewer,filename);CHKERRQ(ierr);
28778501fc72SJed Brown       ierr = VecView(snes->vec_sol,viewer);CHKERRQ(ierr);
28788501fc72SJed Brown       ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
28798501fc72SJed Brown     }
28808501fc72SJed Brown 
2881e113a28aSBarry Smith     if (snes->errorifnotconverged && snes->reason < 0) SETERRQ(((PetscObject)snes)->comm,PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged");
2882efd51863SBarry Smith     if (grid <  snes->gridsequence) {
2883efd51863SBarry Smith       DM  fine;
2884efd51863SBarry Smith       Vec xnew;
2885efd51863SBarry Smith       Mat interp;
2886efd51863SBarry Smith 
2887efd51863SBarry Smith       ierr = DMRefine(snes->dm,((PetscObject)snes)->comm,&fine);CHKERRQ(ierr);
2888efd51863SBarry Smith       ierr = DMGetInterpolation(snes->dm,fine,&interp,PETSC_NULL);CHKERRQ(ierr);
2889efd51863SBarry Smith       ierr = DMCreateGlobalVector(fine,&xnew);CHKERRQ(ierr);
2890efd51863SBarry Smith       ierr = MatInterpolate(interp,x,xnew);CHKERRQ(ierr);
2891efd51863SBarry Smith       ierr = MatDestroy(&interp);CHKERRQ(ierr);
2892efd51863SBarry Smith       x    = xnew;
2893efd51863SBarry Smith 
2894efd51863SBarry Smith       ierr = SNESReset(snes);CHKERRQ(ierr);
2895efd51863SBarry Smith       ierr = SNESSetDM(snes,fine);CHKERRQ(ierr);
2896efd51863SBarry Smith       ierr = DMDestroy(&fine);CHKERRQ(ierr);
2897a8248277SBarry Smith       ierr = PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(((PetscObject)snes)->comm));CHKERRQ(ierr);
2898efd51863SBarry Smith     }
2899efd51863SBarry Smith   }
29003a40ed3dSBarry Smith   PetscFunctionReturn(0);
29019b94acceSBarry Smith }
29029b94acceSBarry Smith 
29039b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */
29049b94acceSBarry Smith 
29054a2ae208SSatish Balay #undef __FUNCT__
29064a2ae208SSatish Balay #define __FUNCT__ "SNESSetType"
290782bf6240SBarry Smith /*@C
29084b0e389bSBarry Smith    SNESSetType - Sets the method for the nonlinear solver.
29099b94acceSBarry Smith 
2910fee21e36SBarry Smith    Collective on SNES
2911fee21e36SBarry Smith 
2912c7afd0dbSLois Curfman McInnes    Input Parameters:
2913c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2914454a90a3SBarry Smith -  type - a known method
2915c7afd0dbSLois Curfman McInnes 
2916c7afd0dbSLois Curfman McInnes    Options Database Key:
2917454a90a3SBarry Smith .  -snes_type <type> - Sets the method; use -help for a list
2918c7afd0dbSLois Curfman McInnes    of available methods (for instance, ls or tr)
2919ae12b187SLois Curfman McInnes 
29209b94acceSBarry Smith    Notes:
2921e090d566SSatish Balay    See "petsc/include/petscsnes.h" for available methods (for instance)
29224b27c08aSLois Curfman McInnes +    SNESLS - Newton's method with line search
2923c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
29244b27c08aSLois Curfman McInnes .    SNESTR - Newton's method with trust region
2925c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
29269b94acceSBarry Smith 
2927ae12b187SLois Curfman McInnes   Normally, it is best to use the SNESSetFromOptions() command and then
2928ae12b187SLois Curfman McInnes   set the SNES solver type from the options database rather than by using
2929ae12b187SLois Curfman McInnes   this routine.  Using the options database provides the user with
2930ae12b187SLois Curfman McInnes   maximum flexibility in evaluating the many nonlinear solvers.
2931ae12b187SLois Curfman McInnes   The SNESSetType() routine is provided for those situations where it
2932ae12b187SLois Curfman McInnes   is necessary to set the nonlinear solver independently of the command
2933ae12b187SLois Curfman McInnes   line or options database.  This might be the case, for example, when
2934ae12b187SLois Curfman McInnes   the choice of solver changes during the execution of the program,
2935ae12b187SLois Curfman McInnes   and the user's application is taking responsibility for choosing the
2936b0a32e0cSBarry Smith   appropriate method.
293736851e7fSLois Curfman McInnes 
293836851e7fSLois Curfman McInnes   Level: intermediate
2939a703fe33SLois Curfman McInnes 
2940454a90a3SBarry Smith .keywords: SNES, set, type
2941435da068SBarry Smith 
2942435da068SBarry Smith .seealso: SNESType, SNESCreate()
2943435da068SBarry Smith 
29449b94acceSBarry Smith @*/
29457087cfbeSBarry Smith PetscErrorCode  SNESSetType(SNES snes,const SNESType type)
29469b94acceSBarry Smith {
2947dfbe8321SBarry Smith   PetscErrorCode ierr,(*r)(SNES);
2948ace3abfcSBarry Smith   PetscBool      match;
29493a40ed3dSBarry Smith 
29503a40ed3dSBarry Smith   PetscFunctionBegin;
29510700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
29524482741eSBarry Smith   PetscValidCharPointer(type,2);
295382bf6240SBarry Smith 
29546831982aSBarry Smith   ierr = PetscTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr);
29550f5bd95cSBarry Smith   if (match) PetscFunctionReturn(0);
295692ff6ae8SBarry Smith 
29574b91b6eaSBarry Smith   ierr =  PetscFListFind(SNESList,((PetscObject)snes)->comm,type,PETSC_TRUE,(void (**)(void)) &r);CHKERRQ(ierr);
2958e32f2f54SBarry Smith   if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type);
295975396ef9SLisandro Dalcin   /* Destroy the previous private SNES context */
296075396ef9SLisandro Dalcin   if (snes->ops->destroy) { ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr); }
296175396ef9SLisandro Dalcin   /* Reinitialize function pointers in SNESOps structure */
296275396ef9SLisandro Dalcin   snes->ops->setup          = 0;
296375396ef9SLisandro Dalcin   snes->ops->solve          = 0;
296475396ef9SLisandro Dalcin   snes->ops->view           = 0;
296575396ef9SLisandro Dalcin   snes->ops->setfromoptions = 0;
296675396ef9SLisandro Dalcin   snes->ops->destroy        = 0;
296775396ef9SLisandro Dalcin   /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */
296875396ef9SLisandro Dalcin   snes->setupcalled = PETSC_FALSE;
2969454a90a3SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr);
297003bfa161SLisandro Dalcin   ierr = (*r)(snes);CHKERRQ(ierr);
29719fb22e1aSBarry Smith #if defined(PETSC_HAVE_AMS)
29729fb22e1aSBarry Smith   if (PetscAMSPublishAll) {
29739fb22e1aSBarry Smith     ierr = PetscObjectAMSPublish((PetscObject)snes);CHKERRQ(ierr);
29749fb22e1aSBarry Smith   }
29759fb22e1aSBarry Smith #endif
29763a40ed3dSBarry Smith   PetscFunctionReturn(0);
29779b94acceSBarry Smith }
29789b94acceSBarry Smith 
2979a847f771SSatish Balay 
29809b94acceSBarry Smith /* --------------------------------------------------------------------- */
29814a2ae208SSatish Balay #undef __FUNCT__
29824a2ae208SSatish Balay #define __FUNCT__ "SNESRegisterDestroy"
298352baeb72SSatish Balay /*@
29849b94acceSBarry Smith    SNESRegisterDestroy - Frees the list of nonlinear solvers that were
2985f1af5d2fSBarry Smith    registered by SNESRegisterDynamic().
29869b94acceSBarry Smith 
2987fee21e36SBarry Smith    Not Collective
2988fee21e36SBarry Smith 
298936851e7fSLois Curfman McInnes    Level: advanced
299036851e7fSLois Curfman McInnes 
29919b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy
29929b94acceSBarry Smith 
29939b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll()
29949b94acceSBarry Smith @*/
29957087cfbeSBarry Smith PetscErrorCode  SNESRegisterDestroy(void)
29969b94acceSBarry Smith {
2997dfbe8321SBarry Smith   PetscErrorCode ierr;
299882bf6240SBarry Smith 
29993a40ed3dSBarry Smith   PetscFunctionBegin;
30001441b1d3SBarry Smith   ierr = PetscFListDestroy(&SNESList);CHKERRQ(ierr);
30014c49b128SBarry Smith   SNESRegisterAllCalled = PETSC_FALSE;
30023a40ed3dSBarry Smith   PetscFunctionReturn(0);
30039b94acceSBarry Smith }
30049b94acceSBarry Smith 
30054a2ae208SSatish Balay #undef __FUNCT__
30064a2ae208SSatish Balay #define __FUNCT__ "SNESGetType"
30079b94acceSBarry Smith /*@C
30089a28b0a6SLois Curfman McInnes    SNESGetType - Gets the SNES method type and name (as a string).
30099b94acceSBarry Smith 
3010c7afd0dbSLois Curfman McInnes    Not Collective
3011c7afd0dbSLois Curfman McInnes 
30129b94acceSBarry Smith    Input Parameter:
30134b0e389bSBarry Smith .  snes - nonlinear solver context
30149b94acceSBarry Smith 
30159b94acceSBarry Smith    Output Parameter:
30163a7fca6bSBarry Smith .  type - SNES method (a character string)
30179b94acceSBarry Smith 
301836851e7fSLois Curfman McInnes    Level: intermediate
301936851e7fSLois Curfman McInnes 
3020454a90a3SBarry Smith .keywords: SNES, nonlinear, get, type, name
30219b94acceSBarry Smith @*/
30227087cfbeSBarry Smith PetscErrorCode  SNESGetType(SNES snes,const SNESType *type)
30239b94acceSBarry Smith {
30243a40ed3dSBarry Smith   PetscFunctionBegin;
30250700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
30264482741eSBarry Smith   PetscValidPointer(type,2);
30277adad957SLisandro Dalcin   *type = ((PetscObject)snes)->type_name;
30283a40ed3dSBarry Smith   PetscFunctionReturn(0);
30299b94acceSBarry Smith }
30309b94acceSBarry Smith 
30314a2ae208SSatish Balay #undef __FUNCT__
30324a2ae208SSatish Balay #define __FUNCT__ "SNESGetSolution"
303352baeb72SSatish Balay /*@
30349b94acceSBarry Smith    SNESGetSolution - Returns the vector where the approximate solution is
30359b94acceSBarry Smith    stored.
30369b94acceSBarry Smith 
3037c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
3038c7afd0dbSLois Curfman McInnes 
30399b94acceSBarry Smith    Input Parameter:
30409b94acceSBarry Smith .  snes - the SNES context
30419b94acceSBarry Smith 
30429b94acceSBarry Smith    Output Parameter:
30439b94acceSBarry Smith .  x - the solution
30449b94acceSBarry Smith 
304570e92668SMatthew Knepley    Level: intermediate
304636851e7fSLois Curfman McInnes 
30479b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution
30489b94acceSBarry Smith 
304985385478SLisandro Dalcin .seealso:  SNESGetSolutionUpdate(), SNESGetFunction()
30509b94acceSBarry Smith @*/
30517087cfbeSBarry Smith PetscErrorCode  SNESGetSolution(SNES snes,Vec *x)
30529b94acceSBarry Smith {
30533a40ed3dSBarry Smith   PetscFunctionBegin;
30540700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
30554482741eSBarry Smith   PetscValidPointer(x,2);
305685385478SLisandro Dalcin   *x = snes->vec_sol;
305770e92668SMatthew Knepley   PetscFunctionReturn(0);
305870e92668SMatthew Knepley }
305970e92668SMatthew Knepley 
306070e92668SMatthew Knepley #undef __FUNCT__
30614a2ae208SSatish Balay #define __FUNCT__ "SNESGetSolutionUpdate"
306252baeb72SSatish Balay /*@
30639b94acceSBarry Smith    SNESGetSolutionUpdate - Returns the vector where the solution update is
30649b94acceSBarry Smith    stored.
30659b94acceSBarry Smith 
3066c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
3067c7afd0dbSLois Curfman McInnes 
30689b94acceSBarry Smith    Input Parameter:
30699b94acceSBarry Smith .  snes - the SNES context
30709b94acceSBarry Smith 
30719b94acceSBarry Smith    Output Parameter:
30729b94acceSBarry Smith .  x - the solution update
30739b94acceSBarry Smith 
307436851e7fSLois Curfman McInnes    Level: advanced
307536851e7fSLois Curfman McInnes 
30769b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update
30779b94acceSBarry Smith 
307885385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction()
30799b94acceSBarry Smith @*/
30807087cfbeSBarry Smith PetscErrorCode  SNESGetSolutionUpdate(SNES snes,Vec *x)
30819b94acceSBarry Smith {
30823a40ed3dSBarry Smith   PetscFunctionBegin;
30830700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
30844482741eSBarry Smith   PetscValidPointer(x,2);
308585385478SLisandro Dalcin   *x = snes->vec_sol_update;
30863a40ed3dSBarry Smith   PetscFunctionReturn(0);
30879b94acceSBarry Smith }
30889b94acceSBarry Smith 
30894a2ae208SSatish Balay #undef __FUNCT__
30904a2ae208SSatish Balay #define __FUNCT__ "SNESGetFunction"
30919b94acceSBarry Smith /*@C
30923638b69dSLois Curfman McInnes    SNESGetFunction - Returns the vector where the function is stored.
30939b94acceSBarry Smith 
3094c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
3095c7afd0dbSLois Curfman McInnes 
30969b94acceSBarry Smith    Input Parameter:
30979b94acceSBarry Smith .  snes - the SNES context
30989b94acceSBarry Smith 
30999b94acceSBarry Smith    Output Parameter:
31007bf4e008SBarry Smith +  r - the function (or PETSC_NULL)
310170e92668SMatthew Knepley .  func - the function (or PETSC_NULL)
310270e92668SMatthew Knepley -  ctx - the function context (or PETSC_NULL)
31039b94acceSBarry Smith 
310436851e7fSLois Curfman McInnes    Level: advanced
310536851e7fSLois Curfman McInnes 
3106a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function
31079b94acceSBarry Smith 
31084b27c08aSLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution()
31099b94acceSBarry Smith @*/
31107087cfbeSBarry Smith PetscErrorCode  SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**func)(SNES,Vec,Vec,void*),void **ctx)
31119b94acceSBarry Smith {
31123a40ed3dSBarry Smith   PetscFunctionBegin;
31130700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
311485385478SLisandro Dalcin   if (r)    *r    = snes->vec_func;
3115e7788613SBarry Smith   if (func) *func = snes->ops->computefunction;
311670e92668SMatthew Knepley   if (ctx)  *ctx  = snes->funP;
31173a40ed3dSBarry Smith   PetscFunctionReturn(0);
31189b94acceSBarry Smith }
31199b94acceSBarry Smith 
31204a2ae208SSatish Balay #undef __FUNCT__
31214a2ae208SSatish Balay #define __FUNCT__ "SNESSetOptionsPrefix"
31223c7409f5SSatish Balay /*@C
31233c7409f5SSatish Balay    SNESSetOptionsPrefix - Sets the prefix used for searching for all
3124d850072dSLois Curfman McInnes    SNES options in the database.
31253c7409f5SSatish Balay 
31263f9fe445SBarry Smith    Logically Collective on SNES
3127fee21e36SBarry Smith 
3128c7afd0dbSLois Curfman McInnes    Input Parameter:
3129c7afd0dbSLois Curfman McInnes +  snes - the SNES context
3130c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
3131c7afd0dbSLois Curfman McInnes 
3132d850072dSLois Curfman McInnes    Notes:
3133a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
3134c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
3135d850072dSLois Curfman McInnes 
313636851e7fSLois Curfman McInnes    Level: advanced
313736851e7fSLois Curfman McInnes 
31383c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database
3139a86d99e1SLois Curfman McInnes 
3140a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions()
31413c7409f5SSatish Balay @*/
31427087cfbeSBarry Smith PetscErrorCode  SNESSetOptionsPrefix(SNES snes,const char prefix[])
31433c7409f5SSatish Balay {
3144dfbe8321SBarry Smith   PetscErrorCode ierr;
31453c7409f5SSatish Balay 
31463a40ed3dSBarry Smith   PetscFunctionBegin;
31470700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3148639f9d9dSBarry Smith   ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
31491cee3971SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
315094b7f48cSBarry Smith   ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
31513a40ed3dSBarry Smith   PetscFunctionReturn(0);
31523c7409f5SSatish Balay }
31533c7409f5SSatish Balay 
31544a2ae208SSatish Balay #undef __FUNCT__
31554a2ae208SSatish Balay #define __FUNCT__ "SNESAppendOptionsPrefix"
31563c7409f5SSatish Balay /*@C
3157f525115eSLois Curfman McInnes    SNESAppendOptionsPrefix - Appends to the prefix used for searching for all
3158d850072dSLois Curfman McInnes    SNES options in the database.
31593c7409f5SSatish Balay 
31603f9fe445SBarry Smith    Logically Collective on SNES
3161fee21e36SBarry Smith 
3162c7afd0dbSLois Curfman McInnes    Input Parameters:
3163c7afd0dbSLois Curfman McInnes +  snes - the SNES context
3164c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
3165c7afd0dbSLois Curfman McInnes 
3166d850072dSLois Curfman McInnes    Notes:
3167a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
3168c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
3169d850072dSLois Curfman McInnes 
317036851e7fSLois Curfman McInnes    Level: advanced
317136851e7fSLois Curfman McInnes 
31723c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database
3173a86d99e1SLois Curfman McInnes 
3174a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix()
31753c7409f5SSatish Balay @*/
31767087cfbeSBarry Smith PetscErrorCode  SNESAppendOptionsPrefix(SNES snes,const char prefix[])
31773c7409f5SSatish Balay {
3178dfbe8321SBarry Smith   PetscErrorCode ierr;
31793c7409f5SSatish Balay 
31803a40ed3dSBarry Smith   PetscFunctionBegin;
31810700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3182639f9d9dSBarry Smith   ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
31831cee3971SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
318494b7f48cSBarry Smith   ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
31853a40ed3dSBarry Smith   PetscFunctionReturn(0);
31863c7409f5SSatish Balay }
31873c7409f5SSatish Balay 
31884a2ae208SSatish Balay #undef __FUNCT__
31894a2ae208SSatish Balay #define __FUNCT__ "SNESGetOptionsPrefix"
31909ab63eb5SSatish Balay /*@C
31913c7409f5SSatish Balay    SNESGetOptionsPrefix - Sets the prefix used for searching for all
31923c7409f5SSatish Balay    SNES options in the database.
31933c7409f5SSatish Balay 
3194c7afd0dbSLois Curfman McInnes    Not Collective
3195c7afd0dbSLois Curfman McInnes 
31963c7409f5SSatish Balay    Input Parameter:
31973c7409f5SSatish Balay .  snes - the SNES context
31983c7409f5SSatish Balay 
31993c7409f5SSatish Balay    Output Parameter:
32003c7409f5SSatish Balay .  prefix - pointer to the prefix string used
32013c7409f5SSatish Balay 
32024ef407dbSRichard Tran Mills    Notes: On the fortran side, the user should pass in a string 'prefix' of
32039ab63eb5SSatish Balay    sufficient length to hold the prefix.
32049ab63eb5SSatish Balay 
320536851e7fSLois Curfman McInnes    Level: advanced
320636851e7fSLois Curfman McInnes 
32073c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database
3208a86d99e1SLois Curfman McInnes 
3209a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix()
32103c7409f5SSatish Balay @*/
32117087cfbeSBarry Smith PetscErrorCode  SNESGetOptionsPrefix(SNES snes,const char *prefix[])
32123c7409f5SSatish Balay {
3213dfbe8321SBarry Smith   PetscErrorCode ierr;
32143c7409f5SSatish Balay 
32153a40ed3dSBarry Smith   PetscFunctionBegin;
32160700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3217639f9d9dSBarry Smith   ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
32183a40ed3dSBarry Smith   PetscFunctionReturn(0);
32193c7409f5SSatish Balay }
32203c7409f5SSatish Balay 
3221b2002411SLois Curfman McInnes 
32224a2ae208SSatish Balay #undef __FUNCT__
32234a2ae208SSatish Balay #define __FUNCT__ "SNESRegister"
32243cea93caSBarry Smith /*@C
32253cea93caSBarry Smith   SNESRegister - See SNESRegisterDynamic()
32263cea93caSBarry Smith 
32277f6c08e0SMatthew Knepley   Level: advanced
32283cea93caSBarry Smith @*/
32297087cfbeSBarry Smith PetscErrorCode  SNESRegister(const char sname[],const char path[],const char name[],PetscErrorCode (*function)(SNES))
3230b2002411SLois Curfman McInnes {
3231e2d1d2b7SBarry Smith   char           fullname[PETSC_MAX_PATH_LEN];
3232dfbe8321SBarry Smith   PetscErrorCode ierr;
3233b2002411SLois Curfman McInnes 
3234b2002411SLois Curfman McInnes   PetscFunctionBegin;
3235b0a32e0cSBarry Smith   ierr = PetscFListConcat(path,name,fullname);CHKERRQ(ierr);
3236c134de8dSSatish Balay   ierr = PetscFListAdd(&SNESList,sname,fullname,(void (*)(void))function);CHKERRQ(ierr);
3237b2002411SLois Curfman McInnes   PetscFunctionReturn(0);
3238b2002411SLois Curfman McInnes }
3239da9b6338SBarry Smith 
3240da9b6338SBarry Smith #undef __FUNCT__
3241da9b6338SBarry Smith #define __FUNCT__ "SNESTestLocalMin"
32427087cfbeSBarry Smith PetscErrorCode  SNESTestLocalMin(SNES snes)
3243da9b6338SBarry Smith {
3244dfbe8321SBarry Smith   PetscErrorCode ierr;
324577431f27SBarry Smith   PetscInt       N,i,j;
3246da9b6338SBarry Smith   Vec            u,uh,fh;
3247da9b6338SBarry Smith   PetscScalar    value;
3248da9b6338SBarry Smith   PetscReal      norm;
3249da9b6338SBarry Smith 
3250da9b6338SBarry Smith   PetscFunctionBegin;
3251da9b6338SBarry Smith   ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr);
3252da9b6338SBarry Smith   ierr = VecDuplicate(u,&uh);CHKERRQ(ierr);
3253da9b6338SBarry Smith   ierr = VecDuplicate(u,&fh);CHKERRQ(ierr);
3254da9b6338SBarry Smith 
3255da9b6338SBarry Smith   /* currently only works for sequential */
3256da9b6338SBarry Smith   ierr = PetscPrintf(PETSC_COMM_WORLD,"Testing FormFunction() for local min\n");
3257da9b6338SBarry Smith   ierr = VecGetSize(u,&N);CHKERRQ(ierr);
3258da9b6338SBarry Smith   for (i=0; i<N; i++) {
3259da9b6338SBarry Smith     ierr = VecCopy(u,uh);CHKERRQ(ierr);
326077431f27SBarry Smith     ierr  = PetscPrintf(PETSC_COMM_WORLD,"i = %D\n",i);CHKERRQ(ierr);
3261da9b6338SBarry Smith     for (j=-10; j<11; j++) {
3262ccae9161SBarry Smith       value = PetscSign(j)*exp(PetscAbs(j)-10.0);
3263da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
32643ab0aad5SBarry Smith       ierr  = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr);
3265da9b6338SBarry Smith       ierr  = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr);
326677431f27SBarry Smith       ierr  = PetscPrintf(PETSC_COMM_WORLD,"       j norm %D %18.16e\n",j,norm);CHKERRQ(ierr);
3267da9b6338SBarry Smith       value = -value;
3268da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
3269da9b6338SBarry Smith     }
3270da9b6338SBarry Smith   }
32716bf464f9SBarry Smith   ierr = VecDestroy(&uh);CHKERRQ(ierr);
32726bf464f9SBarry Smith   ierr = VecDestroy(&fh);CHKERRQ(ierr);
3273da9b6338SBarry Smith   PetscFunctionReturn(0);
3274da9b6338SBarry Smith }
327571f87433Sdalcinl 
327671f87433Sdalcinl #undef __FUNCT__
3277fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPSetUseEW"
327871f87433Sdalcinl /*@
3279fa9f3622SBarry Smith    SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for
328071f87433Sdalcinl    computing relative tolerance for linear solvers within an inexact
328171f87433Sdalcinl    Newton method.
328271f87433Sdalcinl 
32833f9fe445SBarry Smith    Logically Collective on SNES
328471f87433Sdalcinl 
328571f87433Sdalcinl    Input Parameters:
328671f87433Sdalcinl +  snes - SNES context
328771f87433Sdalcinl -  flag - PETSC_TRUE or PETSC_FALSE
328871f87433Sdalcinl 
328964ba62caSBarry Smith     Options Database:
329064ba62caSBarry Smith +  -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence
329164ba62caSBarry Smith .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
329264ba62caSBarry Smith .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
329364ba62caSBarry Smith .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
329464ba62caSBarry Smith .  -snes_ksp_ew_gamma <gamma> - Sets gamma
329564ba62caSBarry Smith .  -snes_ksp_ew_alpha <alpha> - Sets alpha
329664ba62caSBarry Smith .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
329764ba62caSBarry Smith -  -snes_ksp_ew_threshold <threshold> - Sets threshold
329864ba62caSBarry Smith 
329971f87433Sdalcinl    Notes:
330071f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
330171f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
330271f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
330371f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
330471f87433Sdalcinl    solver.
330571f87433Sdalcinl 
330671f87433Sdalcinl    Level: advanced
330771f87433Sdalcinl 
330871f87433Sdalcinl    Reference:
330971f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
331071f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
331171f87433Sdalcinl 
331271f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
331371f87433Sdalcinl 
3314fa9f3622SBarry Smith .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
331571f87433Sdalcinl @*/
33167087cfbeSBarry Smith PetscErrorCode  SNESKSPSetUseEW(SNES snes,PetscBool  flag)
331771f87433Sdalcinl {
331871f87433Sdalcinl   PetscFunctionBegin;
33190700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3320acfcf0e5SJed Brown   PetscValidLogicalCollectiveBool(snes,flag,2);
332171f87433Sdalcinl   snes->ksp_ewconv = flag;
332271f87433Sdalcinl   PetscFunctionReturn(0);
332371f87433Sdalcinl }
332471f87433Sdalcinl 
332571f87433Sdalcinl #undef __FUNCT__
3326fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPGetUseEW"
332771f87433Sdalcinl /*@
3328fa9f3622SBarry Smith    SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method
332971f87433Sdalcinl    for computing relative tolerance for linear solvers within an
333071f87433Sdalcinl    inexact Newton method.
333171f87433Sdalcinl 
333271f87433Sdalcinl    Not Collective
333371f87433Sdalcinl 
333471f87433Sdalcinl    Input Parameter:
333571f87433Sdalcinl .  snes - SNES context
333671f87433Sdalcinl 
333771f87433Sdalcinl    Output Parameter:
333871f87433Sdalcinl .  flag - PETSC_TRUE or PETSC_FALSE
333971f87433Sdalcinl 
334071f87433Sdalcinl    Notes:
334171f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
334271f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
334371f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
334471f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
334571f87433Sdalcinl    solver.
334671f87433Sdalcinl 
334771f87433Sdalcinl    Level: advanced
334871f87433Sdalcinl 
334971f87433Sdalcinl    Reference:
335071f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
335171f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
335271f87433Sdalcinl 
335371f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
335471f87433Sdalcinl 
3355fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
335671f87433Sdalcinl @*/
33577087cfbeSBarry Smith PetscErrorCode  SNESKSPGetUseEW(SNES snes, PetscBool  *flag)
335871f87433Sdalcinl {
335971f87433Sdalcinl   PetscFunctionBegin;
33600700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
336171f87433Sdalcinl   PetscValidPointer(flag,2);
336271f87433Sdalcinl   *flag = snes->ksp_ewconv;
336371f87433Sdalcinl   PetscFunctionReturn(0);
336471f87433Sdalcinl }
336571f87433Sdalcinl 
336671f87433Sdalcinl #undef __FUNCT__
3367fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPSetParametersEW"
336871f87433Sdalcinl /*@
3369fa9f3622SBarry Smith    SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker
337071f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
337171f87433Sdalcinl    Newton method.
337271f87433Sdalcinl 
33733f9fe445SBarry Smith    Logically Collective on SNES
337471f87433Sdalcinl 
337571f87433Sdalcinl    Input Parameters:
337671f87433Sdalcinl +    snes - SNES context
337771f87433Sdalcinl .    version - version 1, 2 (default is 2) or 3
337871f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
337971f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
338071f87433Sdalcinl .    gamma - multiplicative factor for version 2 rtol computation
338171f87433Sdalcinl              (0 <= gamma2 <= 1)
338271f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
338371f87433Sdalcinl .    alpha2 - power for safeguard
338471f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
338571f87433Sdalcinl 
338671f87433Sdalcinl    Note:
338771f87433Sdalcinl    Version 3 was contributed by Luis Chacon, June 2006.
338871f87433Sdalcinl 
338971f87433Sdalcinl    Use PETSC_DEFAULT to retain the default for any of the parameters.
339071f87433Sdalcinl 
339171f87433Sdalcinl    Level: advanced
339271f87433Sdalcinl 
339371f87433Sdalcinl    Reference:
339471f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
339571f87433Sdalcinl    inexact Newton method", Utah State University Math. Stat. Dept. Res.
339671f87433Sdalcinl    Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput.
339771f87433Sdalcinl 
339871f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, set, parameters
339971f87433Sdalcinl 
3400fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW()
340171f87433Sdalcinl @*/
34027087cfbeSBarry Smith PetscErrorCode  SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,
340371f87433Sdalcinl 							    PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold)
340471f87433Sdalcinl {
3405fa9f3622SBarry Smith   SNESKSPEW *kctx;
340671f87433Sdalcinl   PetscFunctionBegin;
34070700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3408fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
3409e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
3410c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,version,2);
3411c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol_0,3);
3412c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol_max,4);
3413c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,gamma,5);
3414c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,alpha,6);
3415c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,alpha2,7);
3416c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,threshold,8);
341771f87433Sdalcinl 
341871f87433Sdalcinl   if (version != PETSC_DEFAULT)   kctx->version   = version;
341971f87433Sdalcinl   if (rtol_0 != PETSC_DEFAULT)    kctx->rtol_0    = rtol_0;
342071f87433Sdalcinl   if (rtol_max != PETSC_DEFAULT)  kctx->rtol_max  = rtol_max;
342171f87433Sdalcinl   if (gamma != PETSC_DEFAULT)     kctx->gamma     = gamma;
342271f87433Sdalcinl   if (alpha != PETSC_DEFAULT)     kctx->alpha     = alpha;
342371f87433Sdalcinl   if (alpha2 != PETSC_DEFAULT)    kctx->alpha2    = alpha2;
342471f87433Sdalcinl   if (threshold != PETSC_DEFAULT) kctx->threshold = threshold;
342571f87433Sdalcinl 
342671f87433Sdalcinl   if (kctx->version < 1 || kctx->version > 3) {
3427e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 and 3 are supported: %D",kctx->version);
342871f87433Sdalcinl   }
342971f87433Sdalcinl   if (kctx->rtol_0 < 0.0 || kctx->rtol_0 >= 1.0) {
3430e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_0 < 1.0: %G",kctx->rtol_0);
343171f87433Sdalcinl   }
343271f87433Sdalcinl   if (kctx->rtol_max < 0.0 || kctx->rtol_max >= 1.0) {
3433e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_max (%G) < 1.0\n",kctx->rtol_max);
343471f87433Sdalcinl   }
343571f87433Sdalcinl   if (kctx->gamma < 0.0 || kctx->gamma > 1.0) {
3436e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= gamma (%G) <= 1.0\n",kctx->gamma);
343771f87433Sdalcinl   }
343871f87433Sdalcinl   if (kctx->alpha <= 1.0 || kctx->alpha > 2.0) {
3439e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"1.0 < alpha (%G) <= 2.0\n",kctx->alpha);
344071f87433Sdalcinl   }
344171f87433Sdalcinl   if (kctx->threshold <= 0.0 || kctx->threshold >= 1.0) {
3442e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 < threshold (%G) < 1.0\n",kctx->threshold);
344371f87433Sdalcinl   }
344471f87433Sdalcinl   PetscFunctionReturn(0);
344571f87433Sdalcinl }
344671f87433Sdalcinl 
344771f87433Sdalcinl #undef __FUNCT__
3448fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPGetParametersEW"
344971f87433Sdalcinl /*@
3450fa9f3622SBarry Smith    SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker
345171f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
345271f87433Sdalcinl    Newton method.
345371f87433Sdalcinl 
345471f87433Sdalcinl    Not Collective
345571f87433Sdalcinl 
345671f87433Sdalcinl    Input Parameters:
345771f87433Sdalcinl      snes - SNES context
345871f87433Sdalcinl 
345971f87433Sdalcinl    Output Parameters:
346071f87433Sdalcinl +    version - version 1, 2 (default is 2) or 3
346171f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
346271f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
346371f87433Sdalcinl .    gamma - multiplicative factor for version 2 rtol computation
346471f87433Sdalcinl              (0 <= gamma2 <= 1)
346571f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
346671f87433Sdalcinl .    alpha2 - power for safeguard
346771f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
346871f87433Sdalcinl 
346971f87433Sdalcinl    Level: advanced
347071f87433Sdalcinl 
347171f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, get, parameters
347271f87433Sdalcinl 
3473fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW()
347471f87433Sdalcinl @*/
34757087cfbeSBarry Smith PetscErrorCode  SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,
347671f87433Sdalcinl 							    PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold)
347771f87433Sdalcinl {
3478fa9f3622SBarry Smith   SNESKSPEW *kctx;
347971f87433Sdalcinl   PetscFunctionBegin;
34800700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3481fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
3482e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
348371f87433Sdalcinl   if(version)   *version   = kctx->version;
348471f87433Sdalcinl   if(rtol_0)    *rtol_0    = kctx->rtol_0;
348571f87433Sdalcinl   if(rtol_max)  *rtol_max  = kctx->rtol_max;
348671f87433Sdalcinl   if(gamma)     *gamma     = kctx->gamma;
348771f87433Sdalcinl   if(alpha)     *alpha     = kctx->alpha;
348871f87433Sdalcinl   if(alpha2)    *alpha2    = kctx->alpha2;
348971f87433Sdalcinl   if(threshold) *threshold = kctx->threshold;
349071f87433Sdalcinl   PetscFunctionReturn(0);
349171f87433Sdalcinl }
349271f87433Sdalcinl 
349371f87433Sdalcinl #undef __FUNCT__
3494fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPEW_PreSolve"
3495fa9f3622SBarry Smith static PetscErrorCode SNESKSPEW_PreSolve(SNES snes, KSP ksp, Vec b, Vec x)
349671f87433Sdalcinl {
349771f87433Sdalcinl   PetscErrorCode ierr;
3498fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
349971f87433Sdalcinl   PetscReal      rtol=PETSC_DEFAULT,stol;
350071f87433Sdalcinl 
350171f87433Sdalcinl   PetscFunctionBegin;
3502e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context exists");
350371f87433Sdalcinl   if (!snes->iter) { /* first time in, so use the original user rtol */
350471f87433Sdalcinl     rtol = kctx->rtol_0;
350571f87433Sdalcinl   } else {
350671f87433Sdalcinl     if (kctx->version == 1) {
350771f87433Sdalcinl       rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last;
350871f87433Sdalcinl       if (rtol < 0.0) rtol = -rtol;
350971f87433Sdalcinl       stol = pow(kctx->rtol_last,kctx->alpha2);
351071f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
351171f87433Sdalcinl     } else if (kctx->version == 2) {
351271f87433Sdalcinl       rtol = kctx->gamma * pow(snes->norm/kctx->norm_last,kctx->alpha);
351371f87433Sdalcinl       stol = kctx->gamma * pow(kctx->rtol_last,kctx->alpha);
351471f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
351571f87433Sdalcinl     } else if (kctx->version == 3) {/* contributed by Luis Chacon, June 2006. */
351671f87433Sdalcinl       rtol = kctx->gamma * pow(snes->norm/kctx->norm_last,kctx->alpha);
351771f87433Sdalcinl       /* safeguard: avoid sharp decrease of rtol */
351871f87433Sdalcinl       stol = kctx->gamma*pow(kctx->rtol_last,kctx->alpha);
351971f87433Sdalcinl       stol = PetscMax(rtol,stol);
352071f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
352171f87433Sdalcinl       /* safeguard: avoid oversolving */
352271f87433Sdalcinl       stol = kctx->gamma*(snes->ttol)/snes->norm;
352371f87433Sdalcinl       stol = PetscMax(rtol,stol);
352471f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
3525e32f2f54SBarry Smith     } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version);
352671f87433Sdalcinl   }
352771f87433Sdalcinl   /* safeguard: avoid rtol greater than one */
352871f87433Sdalcinl   rtol = PetscMin(rtol,kctx->rtol_max);
352971f87433Sdalcinl   ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr);
353071f87433Sdalcinl   ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%G\n",snes->iter,kctx->version,rtol);CHKERRQ(ierr);
353171f87433Sdalcinl   PetscFunctionReturn(0);
353271f87433Sdalcinl }
353371f87433Sdalcinl 
353471f87433Sdalcinl #undef __FUNCT__
3535fa9f3622SBarry Smith #define __FUNCT__ "SNESKSPEW_PostSolve"
3536fa9f3622SBarry Smith static PetscErrorCode SNESKSPEW_PostSolve(SNES snes, KSP ksp, Vec b, Vec x)
353771f87433Sdalcinl {
353871f87433Sdalcinl   PetscErrorCode ierr;
3539fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
354071f87433Sdalcinl   PCSide         pcside;
354171f87433Sdalcinl   Vec            lres;
354271f87433Sdalcinl 
354371f87433Sdalcinl   PetscFunctionBegin;
3544e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context exists");
354571f87433Sdalcinl   ierr = KSPGetTolerances(ksp,&kctx->rtol_last,0,0,0);CHKERRQ(ierr);
354671f87433Sdalcinl   ierr = SNESGetFunctionNorm(snes,&kctx->norm_last);CHKERRQ(ierr);
354771f87433Sdalcinl   if (kctx->version == 1) {
3548b037da10SBarry Smith     ierr = KSPGetPCSide(ksp,&pcside);CHKERRQ(ierr);
354971f87433Sdalcinl     if (pcside == PC_RIGHT) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */
355071f87433Sdalcinl       /* KSP residual is true linear residual */
355171f87433Sdalcinl       ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr);
355271f87433Sdalcinl     } else {
355371f87433Sdalcinl       /* KSP residual is preconditioned residual */
355471f87433Sdalcinl       /* compute true linear residual norm */
355571f87433Sdalcinl       ierr = VecDuplicate(b,&lres);CHKERRQ(ierr);
355671f87433Sdalcinl       ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr);
355771f87433Sdalcinl       ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr);
355871f87433Sdalcinl       ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr);
35596bf464f9SBarry Smith       ierr = VecDestroy(&lres);CHKERRQ(ierr);
356071f87433Sdalcinl     }
356171f87433Sdalcinl   }
356271f87433Sdalcinl   PetscFunctionReturn(0);
356371f87433Sdalcinl }
356471f87433Sdalcinl 
356571f87433Sdalcinl #undef __FUNCT__
356671f87433Sdalcinl #define __FUNCT__ "SNES_KSPSolve"
356771f87433Sdalcinl PetscErrorCode SNES_KSPSolve(SNES snes, KSP ksp, Vec b, Vec x)
356871f87433Sdalcinl {
356971f87433Sdalcinl   PetscErrorCode ierr;
357071f87433Sdalcinl 
357171f87433Sdalcinl   PetscFunctionBegin;
3572fa9f3622SBarry Smith   if (snes->ksp_ewconv) { ierr = SNESKSPEW_PreSolve(snes,ksp,b,x);CHKERRQ(ierr);  }
357371f87433Sdalcinl   ierr = KSPSolve(ksp,b,x);CHKERRQ(ierr);
3574fa9f3622SBarry Smith   if (snes->ksp_ewconv) { ierr = SNESKSPEW_PostSolve(snes,ksp,b,x);CHKERRQ(ierr); }
357571f87433Sdalcinl   PetscFunctionReturn(0);
357671f87433Sdalcinl }
35776c699258SBarry Smith 
35786c699258SBarry Smith #undef __FUNCT__
35796c699258SBarry Smith #define __FUNCT__ "SNESSetDM"
35806c699258SBarry Smith /*@
35816c699258SBarry Smith    SNESSetDM - Sets the DM that may be used by some preconditioners
35826c699258SBarry Smith 
35833f9fe445SBarry Smith    Logically Collective on SNES
35846c699258SBarry Smith 
35856c699258SBarry Smith    Input Parameters:
35866c699258SBarry Smith +  snes - the preconditioner context
35876c699258SBarry Smith -  dm - the dm
35886c699258SBarry Smith 
35896c699258SBarry Smith    Level: intermediate
35906c699258SBarry Smith 
35916c699258SBarry Smith 
35926c699258SBarry Smith .seealso: SNESGetDM(), KSPSetDM(), KSPGetDM()
35936c699258SBarry Smith @*/
35947087cfbeSBarry Smith PetscErrorCode  SNESSetDM(SNES snes,DM dm)
35956c699258SBarry Smith {
35966c699258SBarry Smith   PetscErrorCode ierr;
3597345fed2cSBarry Smith   KSP            ksp;
35986c699258SBarry Smith 
35996c699258SBarry Smith   PetscFunctionBegin;
36000700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3601d0660788SBarry Smith   if (dm) {ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr);}
36026bf464f9SBarry Smith   ierr = DMDestroy(&snes->dm);CHKERRQ(ierr);
36036c699258SBarry Smith   snes->dm = dm;
3604345fed2cSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
3605345fed2cSBarry Smith   ierr = KSPSetDM(ksp,dm);CHKERRQ(ierr);
3606f22f69f0SBarry Smith   ierr = KSPSetDMActive(ksp,PETSC_FALSE);CHKERRQ(ierr);
36072c155ee1SBarry Smith   if (snes->pc) {
36082c155ee1SBarry Smith     ierr = SNESSetDM(snes->pc, snes->dm);CHKERRQ(ierr);
36092c155ee1SBarry Smith   }
36106c699258SBarry Smith   PetscFunctionReturn(0);
36116c699258SBarry Smith }
36126c699258SBarry Smith 
36136c699258SBarry Smith #undef __FUNCT__
36146c699258SBarry Smith #define __FUNCT__ "SNESGetDM"
36156c699258SBarry Smith /*@
36166c699258SBarry Smith    SNESGetDM - Gets the DM that may be used by some preconditioners
36176c699258SBarry Smith 
36183f9fe445SBarry Smith    Not Collective but DM obtained is parallel on SNES
36196c699258SBarry Smith 
36206c699258SBarry Smith    Input Parameter:
36216c699258SBarry Smith . snes - the preconditioner context
36226c699258SBarry Smith 
36236c699258SBarry Smith    Output Parameter:
36246c699258SBarry Smith .  dm - the dm
36256c699258SBarry Smith 
36266c699258SBarry Smith    Level: intermediate
36276c699258SBarry Smith 
36286c699258SBarry Smith 
36296c699258SBarry Smith .seealso: SNESSetDM(), KSPSetDM(), KSPGetDM()
36306c699258SBarry Smith @*/
36317087cfbeSBarry Smith PetscErrorCode  SNESGetDM(SNES snes,DM *dm)
36326c699258SBarry Smith {
36336c699258SBarry Smith   PetscFunctionBegin;
36340700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
36356c699258SBarry Smith   *dm = snes->dm;
36366c699258SBarry Smith   PetscFunctionReturn(0);
36376c699258SBarry Smith }
36380807856dSBarry Smith 
363931823bd8SMatthew G Knepley #undef __FUNCT__
364031823bd8SMatthew G Knepley #define __FUNCT__ "SNESSetPC"
364131823bd8SMatthew G Knepley /*@
3642fd4b34e9SJed Brown   SNESSetPC - Sets the nonlinear preconditioner to be used.
364331823bd8SMatthew G Knepley 
364431823bd8SMatthew G Knepley   Collective on SNES
364531823bd8SMatthew G Knepley 
364631823bd8SMatthew G Knepley   Input Parameters:
364731823bd8SMatthew G Knepley + snes - iterative context obtained from SNESCreate()
364831823bd8SMatthew G Knepley - pc   - the preconditioner object
364931823bd8SMatthew G Knepley 
365031823bd8SMatthew G Knepley   Notes:
365131823bd8SMatthew G Knepley   Use SNESGetPC() to retrieve the preconditioner context (for example,
365231823bd8SMatthew G Knepley   to configure it using the API).
365331823bd8SMatthew G Knepley 
365431823bd8SMatthew G Knepley   Level: developer
365531823bd8SMatthew G Knepley 
365631823bd8SMatthew G Knepley .keywords: SNES, set, precondition
365731823bd8SMatthew G Knepley .seealso: SNESGetPC()
365831823bd8SMatthew G Knepley @*/
365931823bd8SMatthew G Knepley PetscErrorCode SNESSetPC(SNES snes, SNES pc)
366031823bd8SMatthew G Knepley {
366131823bd8SMatthew G Knepley   PetscErrorCode ierr;
366231823bd8SMatthew G Knepley 
366331823bd8SMatthew G Knepley   PetscFunctionBegin;
366431823bd8SMatthew G Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
366531823bd8SMatthew G Knepley   PetscValidHeaderSpecific(pc, SNES_CLASSID, 2);
366631823bd8SMatthew G Knepley   PetscCheckSameComm(snes, 1, pc, 2);
366731823bd8SMatthew G Knepley   ierr = PetscObjectReference((PetscObject) pc);CHKERRQ(ierr);
3668bf11d3b6SBarry Smith   ierr = SNESDestroy(&snes->pc);CHKERRQ(ierr);
366931823bd8SMatthew G Knepley   snes->pc = pc;
367031823bd8SMatthew G Knepley   ierr = PetscLogObjectParent(snes, snes->pc);CHKERRQ(ierr);
367131823bd8SMatthew G Knepley   PetscFunctionReturn(0);
367231823bd8SMatthew G Knepley }
367331823bd8SMatthew G Knepley 
367431823bd8SMatthew G Knepley #undef __FUNCT__
367531823bd8SMatthew G Knepley #define __FUNCT__ "SNESGetPC"
367631823bd8SMatthew G Knepley /*@
3677fd4b34e9SJed Brown   SNESGetPC - Returns a pointer to the nonlinear preconditioning context set with SNESSetPC().
367831823bd8SMatthew G Knepley 
367931823bd8SMatthew G Knepley   Not Collective
368031823bd8SMatthew G Knepley 
368131823bd8SMatthew G Knepley   Input Parameter:
368231823bd8SMatthew G Knepley . snes - iterative context obtained from SNESCreate()
368331823bd8SMatthew G Knepley 
368431823bd8SMatthew G Knepley   Output Parameter:
368531823bd8SMatthew G Knepley . pc - preconditioner context
368631823bd8SMatthew G Knepley 
368731823bd8SMatthew G Knepley   Level: developer
368831823bd8SMatthew G Knepley 
368931823bd8SMatthew G Knepley .keywords: SNES, get, preconditioner
369031823bd8SMatthew G Knepley .seealso: SNESSetPC()
369131823bd8SMatthew G Knepley @*/
369231823bd8SMatthew G Knepley PetscErrorCode SNESGetPC(SNES snes, SNES *pc)
369331823bd8SMatthew G Knepley {
369431823bd8SMatthew G Knepley   PetscErrorCode ierr;
369531823bd8SMatthew G Knepley 
369631823bd8SMatthew G Knepley   PetscFunctionBegin;
369731823bd8SMatthew G Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
369831823bd8SMatthew G Knepley   PetscValidPointer(pc, 2);
369931823bd8SMatthew G Knepley   if (!snes->pc) {
370031823bd8SMatthew G Knepley     ierr = SNESCreate(((PetscObject) snes)->comm, &snes->pc);CHKERRQ(ierr);
37014a0c5b0cSMatthew G Knepley     ierr = PetscObjectIncrementTabLevel((PetscObject) snes->pc, (PetscObject) snes, 1);CHKERRQ(ierr);
370231823bd8SMatthew G Knepley     ierr = PetscLogObjectParent(snes, snes->pc);CHKERRQ(ierr);
370331823bd8SMatthew G Knepley   }
370431823bd8SMatthew G Knepley   *pc = snes->pc;
370531823bd8SMatthew G Knepley   PetscFunctionReturn(0);
370631823bd8SMatthew G Knepley }
370731823bd8SMatthew G Knepley 
370869b4f73cSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
3709c6db04a5SJed Brown #include <mex.h>
371069b4f73cSBarry Smith 
37118f6e6473SBarry Smith typedef struct {char *funcname; mxArray *ctx;} SNESMatlabContext;
37128f6e6473SBarry Smith 
37130807856dSBarry Smith #undef __FUNCT__
37140807856dSBarry Smith #define __FUNCT__ "SNESComputeFunction_Matlab"
37150807856dSBarry Smith /*
37160807856dSBarry Smith    SNESComputeFunction_Matlab - Calls the function that has been set with
37170807856dSBarry Smith                          SNESSetFunctionMatlab().
37180807856dSBarry Smith 
37190807856dSBarry Smith    Collective on SNES
37200807856dSBarry Smith 
37210807856dSBarry Smith    Input Parameters:
37220807856dSBarry Smith +  snes - the SNES context
37230807856dSBarry Smith -  x - input vector
37240807856dSBarry Smith 
37250807856dSBarry Smith    Output Parameter:
37260807856dSBarry Smith .  y - function vector, as set by SNESSetFunction()
37270807856dSBarry Smith 
37280807856dSBarry Smith    Notes:
37290807856dSBarry Smith    SNESComputeFunction() is typically used within nonlinear solvers
37300807856dSBarry Smith    implementations, so most users would not generally call this routine
37310807856dSBarry Smith    themselves.
37320807856dSBarry Smith 
37330807856dSBarry Smith    Level: developer
37340807856dSBarry Smith 
37350807856dSBarry Smith .keywords: SNES, nonlinear, compute, function
37360807856dSBarry Smith 
37370807856dSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction()
373861b2408cSBarry Smith */
37397087cfbeSBarry Smith PetscErrorCode  SNESComputeFunction_Matlab(SNES snes,Vec x,Vec y, void *ctx)
37400807856dSBarry Smith {
3741e650e774SBarry Smith   PetscErrorCode    ierr;
37428f6e6473SBarry Smith   SNESMatlabContext *sctx = (SNESMatlabContext *)ctx;
37438f6e6473SBarry Smith   int               nlhs = 1,nrhs = 5;
37448f6e6473SBarry Smith   mxArray	    *plhs[1],*prhs[5];
374591621f2eSBarry Smith   long long int     lx = 0,ly = 0,ls = 0;
3746e650e774SBarry Smith 
37470807856dSBarry Smith   PetscFunctionBegin;
37480807856dSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
37490807856dSBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
37500807856dSBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
37510807856dSBarry Smith   PetscCheckSameComm(snes,1,x,2);
37520807856dSBarry Smith   PetscCheckSameComm(snes,1,y,3);
37530807856dSBarry Smith 
37540807856dSBarry Smith   /* call Matlab function in ctx with arguments x and y */
3755e650e774SBarry Smith 
375691621f2eSBarry Smith   ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr);
3757e650e774SBarry Smith   ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr);
3758e650e774SBarry Smith   ierr = PetscMemcpy(&ly,&y,sizeof(x));CHKERRQ(ierr);
375991621f2eSBarry Smith   prhs[0] =  mxCreateDoubleScalar((double)ls);
376091621f2eSBarry Smith   prhs[1] =  mxCreateDoubleScalar((double)lx);
376191621f2eSBarry Smith   prhs[2] =  mxCreateDoubleScalar((double)ly);
37628f6e6473SBarry Smith   prhs[3] =  mxCreateString(sctx->funcname);
37638f6e6473SBarry Smith   prhs[4] =  sctx->ctx;
3764b807a863SBarry Smith   ierr    =  mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeFunctionInternal");CHKERRQ(ierr);
3765e650e774SBarry Smith   ierr    =  mxGetScalar(plhs[0]);CHKERRQ(ierr);
3766e650e774SBarry Smith   mxDestroyArray(prhs[0]);
3767e650e774SBarry Smith   mxDestroyArray(prhs[1]);
3768e650e774SBarry Smith   mxDestroyArray(prhs[2]);
37698f6e6473SBarry Smith   mxDestroyArray(prhs[3]);
3770e650e774SBarry Smith   mxDestroyArray(plhs[0]);
37710807856dSBarry Smith   PetscFunctionReturn(0);
37720807856dSBarry Smith }
37730807856dSBarry Smith 
37740807856dSBarry Smith 
37750807856dSBarry Smith #undef __FUNCT__
37760807856dSBarry Smith #define __FUNCT__ "SNESSetFunctionMatlab"
377761b2408cSBarry Smith /*
37780807856dSBarry Smith    SNESSetFunctionMatlab - Sets the function evaluation routine and function
37790807856dSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
3780e3c5b3baSBarry Smith    equations from MATLAB. Here the function is a string containing the name of a MATLAB function
37810807856dSBarry Smith 
37820807856dSBarry Smith    Logically Collective on SNES
37830807856dSBarry Smith 
37840807856dSBarry Smith    Input Parameters:
37850807856dSBarry Smith +  snes - the SNES context
37860807856dSBarry Smith .  r - vector to store function value
37870807856dSBarry Smith -  func - function evaluation routine
37880807856dSBarry Smith 
37890807856dSBarry Smith    Calling sequence of func:
379061b2408cSBarry Smith $    func (SNES snes,Vec x,Vec f,void *ctx);
37910807856dSBarry Smith 
37920807856dSBarry Smith 
37930807856dSBarry Smith    Notes:
37940807856dSBarry Smith    The Newton-like methods typically solve linear systems of the form
37950807856dSBarry Smith $      f'(x) x = -f(x),
37960807856dSBarry Smith    where f'(x) denotes the Jacobian matrix and f(x) is the function.
37970807856dSBarry Smith 
37980807856dSBarry Smith    Level: beginner
37990807856dSBarry Smith 
38000807856dSBarry Smith .keywords: SNES, nonlinear, set, function
38010807856dSBarry Smith 
38020807856dSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
380361b2408cSBarry Smith */
38047087cfbeSBarry Smith PetscErrorCode  SNESSetFunctionMatlab(SNES snes,Vec r,const char *func,mxArray *ctx)
38050807856dSBarry Smith {
38060807856dSBarry Smith   PetscErrorCode    ierr;
38078f6e6473SBarry Smith   SNESMatlabContext *sctx;
38080807856dSBarry Smith 
38090807856dSBarry Smith   PetscFunctionBegin;
38108f6e6473SBarry Smith   /* currently sctx is memory bleed */
38118f6e6473SBarry Smith   ierr = PetscMalloc(sizeof(SNESMatlabContext),&sctx);CHKERRQ(ierr);
38128f6e6473SBarry Smith   ierr = PetscStrallocpy(func,&sctx->funcname);CHKERRQ(ierr);
38138f6e6473SBarry Smith   /*
38148f6e6473SBarry Smith      This should work, but it doesn't
38158f6e6473SBarry Smith   sctx->ctx = ctx;
38168f6e6473SBarry Smith   mexMakeArrayPersistent(sctx->ctx);
38178f6e6473SBarry Smith   */
38188f6e6473SBarry Smith   sctx->ctx = mxDuplicateArray(ctx);
38198f6e6473SBarry Smith   ierr = SNESSetFunction(snes,r,SNESComputeFunction_Matlab,sctx);CHKERRQ(ierr);
38200807856dSBarry Smith   PetscFunctionReturn(0);
38210807856dSBarry Smith }
382269b4f73cSBarry Smith 
382361b2408cSBarry Smith #undef __FUNCT__
382461b2408cSBarry Smith #define __FUNCT__ "SNESComputeJacobian_Matlab"
382561b2408cSBarry Smith /*
382661b2408cSBarry Smith    SNESComputeJacobian_Matlab - Calls the function that has been set with
382761b2408cSBarry Smith                          SNESSetJacobianMatlab().
382861b2408cSBarry Smith 
382961b2408cSBarry Smith    Collective on SNES
383061b2408cSBarry Smith 
383161b2408cSBarry Smith    Input Parameters:
383261b2408cSBarry Smith +  snes - the SNES context
383361b2408cSBarry Smith .  x - input vector
383461b2408cSBarry Smith .  A, B - the matrices
383561b2408cSBarry Smith -  ctx - user context
383661b2408cSBarry Smith 
383761b2408cSBarry Smith    Output Parameter:
383861b2408cSBarry Smith .  flag - structure of the matrix
383961b2408cSBarry Smith 
384061b2408cSBarry Smith    Level: developer
384161b2408cSBarry Smith 
384261b2408cSBarry Smith .keywords: SNES, nonlinear, compute, function
384361b2408cSBarry Smith 
384461b2408cSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction()
384561b2408cSBarry Smith @*/
38467087cfbeSBarry Smith PetscErrorCode  SNESComputeJacobian_Matlab(SNES snes,Vec x,Mat *A,Mat *B,MatStructure *flag, void *ctx)
384761b2408cSBarry Smith {
384861b2408cSBarry Smith   PetscErrorCode    ierr;
384961b2408cSBarry Smith   SNESMatlabContext *sctx = (SNESMatlabContext *)ctx;
385061b2408cSBarry Smith   int               nlhs = 2,nrhs = 6;
385161b2408cSBarry Smith   mxArray	    *plhs[2],*prhs[6];
385261b2408cSBarry Smith   long long int     lx = 0,lA = 0,ls = 0, lB = 0;
385361b2408cSBarry Smith 
385461b2408cSBarry Smith   PetscFunctionBegin;
385561b2408cSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
385661b2408cSBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
385761b2408cSBarry Smith 
385861b2408cSBarry Smith   /* call Matlab function in ctx with arguments x and y */
385961b2408cSBarry Smith 
386061b2408cSBarry Smith   ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr);
386161b2408cSBarry Smith   ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr);
386261b2408cSBarry Smith   ierr = PetscMemcpy(&lA,A,sizeof(x));CHKERRQ(ierr);
386361b2408cSBarry Smith   ierr = PetscMemcpy(&lB,B,sizeof(x));CHKERRQ(ierr);
386461b2408cSBarry Smith   prhs[0] =  mxCreateDoubleScalar((double)ls);
386561b2408cSBarry Smith   prhs[1] =  mxCreateDoubleScalar((double)lx);
386661b2408cSBarry Smith   prhs[2] =  mxCreateDoubleScalar((double)lA);
386761b2408cSBarry Smith   prhs[3] =  mxCreateDoubleScalar((double)lB);
386861b2408cSBarry Smith   prhs[4] =  mxCreateString(sctx->funcname);
386961b2408cSBarry Smith   prhs[5] =  sctx->ctx;
3870b807a863SBarry Smith   ierr    =  mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeJacobianInternal");CHKERRQ(ierr);
387161b2408cSBarry Smith   ierr    =  mxGetScalar(plhs[0]);CHKERRQ(ierr);
387261b2408cSBarry Smith   *flag   =  (MatStructure) mxGetScalar(plhs[1]);CHKERRQ(ierr);
387361b2408cSBarry Smith   mxDestroyArray(prhs[0]);
387461b2408cSBarry Smith   mxDestroyArray(prhs[1]);
387561b2408cSBarry Smith   mxDestroyArray(prhs[2]);
387661b2408cSBarry Smith   mxDestroyArray(prhs[3]);
387761b2408cSBarry Smith   mxDestroyArray(prhs[4]);
387861b2408cSBarry Smith   mxDestroyArray(plhs[0]);
387961b2408cSBarry Smith   mxDestroyArray(plhs[1]);
388061b2408cSBarry Smith   PetscFunctionReturn(0);
388161b2408cSBarry Smith }
388261b2408cSBarry Smith 
388361b2408cSBarry Smith 
388461b2408cSBarry Smith #undef __FUNCT__
388561b2408cSBarry Smith #define __FUNCT__ "SNESSetJacobianMatlab"
388661b2408cSBarry Smith /*
388761b2408cSBarry Smith    SNESSetJacobianMatlab - Sets the Jacobian function evaluation routine and two empty Jacobian matrices
388861b2408cSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
3889e3c5b3baSBarry Smith    equations from MATLAB. Here the function is a string containing the name of a MATLAB function
389061b2408cSBarry Smith 
389161b2408cSBarry Smith    Logically Collective on SNES
389261b2408cSBarry Smith 
389361b2408cSBarry Smith    Input Parameters:
389461b2408cSBarry Smith +  snes - the SNES context
389561b2408cSBarry Smith .  A,B - Jacobian matrices
389661b2408cSBarry Smith .  func - function evaluation routine
389761b2408cSBarry Smith -  ctx - user context
389861b2408cSBarry Smith 
389961b2408cSBarry Smith    Calling sequence of func:
390061b2408cSBarry Smith $    flag = func (SNES snes,Vec x,Mat A,Mat B,void *ctx);
390161b2408cSBarry Smith 
390261b2408cSBarry Smith 
390361b2408cSBarry Smith    Level: developer
390461b2408cSBarry Smith 
390561b2408cSBarry Smith .keywords: SNES, nonlinear, set, function
390661b2408cSBarry Smith 
390761b2408cSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
390861b2408cSBarry Smith */
39097087cfbeSBarry Smith PetscErrorCode  SNESSetJacobianMatlab(SNES snes,Mat A,Mat B,const char *func,mxArray *ctx)
391061b2408cSBarry Smith {
391161b2408cSBarry Smith   PetscErrorCode    ierr;
391261b2408cSBarry Smith   SNESMatlabContext *sctx;
391361b2408cSBarry Smith 
391461b2408cSBarry Smith   PetscFunctionBegin;
391561b2408cSBarry Smith   /* currently sctx is memory bleed */
391661b2408cSBarry Smith   ierr = PetscMalloc(sizeof(SNESMatlabContext),&sctx);CHKERRQ(ierr);
391761b2408cSBarry Smith   ierr = PetscStrallocpy(func,&sctx->funcname);CHKERRQ(ierr);
391861b2408cSBarry Smith   /*
391961b2408cSBarry Smith      This should work, but it doesn't
392061b2408cSBarry Smith   sctx->ctx = ctx;
392161b2408cSBarry Smith   mexMakeArrayPersistent(sctx->ctx);
392261b2408cSBarry Smith   */
392361b2408cSBarry Smith   sctx->ctx = mxDuplicateArray(ctx);
392461b2408cSBarry Smith   ierr = SNESSetJacobian(snes,A,B,SNESComputeJacobian_Matlab,sctx);CHKERRQ(ierr);
392561b2408cSBarry Smith   PetscFunctionReturn(0);
392661b2408cSBarry Smith }
392769b4f73cSBarry Smith 
3928f9eb7ae2SShri Abhyankar #undef __FUNCT__
3929f9eb7ae2SShri Abhyankar #define __FUNCT__ "SNESMonitor_Matlab"
3930f9eb7ae2SShri Abhyankar /*
3931f9eb7ae2SShri Abhyankar    SNESMonitor_Matlab - Calls the function that has been set with SNESMonitorSetMatlab().
3932f9eb7ae2SShri Abhyankar 
3933f9eb7ae2SShri Abhyankar    Collective on SNES
3934f9eb7ae2SShri Abhyankar 
3935f9eb7ae2SShri Abhyankar .seealso: SNESSetFunction(), SNESGetFunction()
3936f9eb7ae2SShri Abhyankar @*/
39377087cfbeSBarry Smith PetscErrorCode  SNESMonitor_Matlab(SNES snes,PetscInt it, PetscReal fnorm, void *ctx)
3938f9eb7ae2SShri Abhyankar {
3939f9eb7ae2SShri Abhyankar   PetscErrorCode  ierr;
394048f37e70SShri Abhyankar   SNESMatlabContext *sctx = (SNESMatlabContext *)ctx;
3941f9eb7ae2SShri Abhyankar   int             nlhs = 1,nrhs = 6;
3942f9eb7ae2SShri Abhyankar   mxArray	  *plhs[1],*prhs[6];
3943f9eb7ae2SShri Abhyankar   long long int   lx = 0,ls = 0;
3944f9eb7ae2SShri Abhyankar   Vec             x=snes->vec_sol;
3945f9eb7ae2SShri Abhyankar 
3946f9eb7ae2SShri Abhyankar   PetscFunctionBegin;
3947f9eb7ae2SShri Abhyankar   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3948f9eb7ae2SShri Abhyankar 
3949f9eb7ae2SShri Abhyankar   ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr);
3950f9eb7ae2SShri Abhyankar   ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr);
3951f9eb7ae2SShri Abhyankar   prhs[0] =  mxCreateDoubleScalar((double)ls);
3952f9eb7ae2SShri Abhyankar   prhs[1] =  mxCreateDoubleScalar((double)it);
3953f9eb7ae2SShri Abhyankar   prhs[2] =  mxCreateDoubleScalar((double)fnorm);
3954f9eb7ae2SShri Abhyankar   prhs[3] =  mxCreateDoubleScalar((double)lx);
3955f9eb7ae2SShri Abhyankar   prhs[4] =  mxCreateString(sctx->funcname);
3956f9eb7ae2SShri Abhyankar   prhs[5] =  sctx->ctx;
3957f9eb7ae2SShri Abhyankar   ierr    =  mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESMonitorInternal");CHKERRQ(ierr);
3958f9eb7ae2SShri Abhyankar   ierr    =  mxGetScalar(plhs[0]);CHKERRQ(ierr);
3959f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[0]);
3960f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[1]);
3961f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[2]);
3962f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[3]);
3963f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[4]);
3964f9eb7ae2SShri Abhyankar   mxDestroyArray(plhs[0]);
3965f9eb7ae2SShri Abhyankar   PetscFunctionReturn(0);
3966f9eb7ae2SShri Abhyankar }
3967f9eb7ae2SShri Abhyankar 
3968f9eb7ae2SShri Abhyankar 
3969f9eb7ae2SShri Abhyankar #undef __FUNCT__
3970f9eb7ae2SShri Abhyankar #define __FUNCT__ "SNESMonitorSetMatlab"
3971f9eb7ae2SShri Abhyankar /*
3972e3c5b3baSBarry Smith    SNESMonitorSetMatlab - Sets the monitor function from MATLAB
3973f9eb7ae2SShri Abhyankar 
3974f9eb7ae2SShri Abhyankar    Level: developer
3975f9eb7ae2SShri Abhyankar 
3976f9eb7ae2SShri Abhyankar .keywords: SNES, nonlinear, set, function
3977f9eb7ae2SShri Abhyankar 
3978f9eb7ae2SShri Abhyankar .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
3979f9eb7ae2SShri Abhyankar */
39807087cfbeSBarry Smith PetscErrorCode  SNESMonitorSetMatlab(SNES snes,const char *func,mxArray *ctx)
3981f9eb7ae2SShri Abhyankar {
3982f9eb7ae2SShri Abhyankar   PetscErrorCode    ierr;
3983f9eb7ae2SShri Abhyankar   SNESMatlabContext *sctx;
3984f9eb7ae2SShri Abhyankar 
3985f9eb7ae2SShri Abhyankar   PetscFunctionBegin;
3986f9eb7ae2SShri Abhyankar   /* currently sctx is memory bleed */
3987f9eb7ae2SShri Abhyankar   ierr = PetscMalloc(sizeof(SNESMatlabContext),&sctx);CHKERRQ(ierr);
3988f9eb7ae2SShri Abhyankar   ierr = PetscStrallocpy(func,&sctx->funcname);CHKERRQ(ierr);
3989f9eb7ae2SShri Abhyankar   /*
3990f9eb7ae2SShri Abhyankar      This should work, but it doesn't
3991f9eb7ae2SShri Abhyankar   sctx->ctx = ctx;
3992f9eb7ae2SShri Abhyankar   mexMakeArrayPersistent(sctx->ctx);
3993f9eb7ae2SShri Abhyankar   */
3994f9eb7ae2SShri Abhyankar   sctx->ctx = mxDuplicateArray(ctx);
3995f9eb7ae2SShri Abhyankar   ierr = SNESMonitorSet(snes,SNESMonitor_Matlab,sctx,PETSC_NULL);CHKERRQ(ierr);
3996f9eb7ae2SShri Abhyankar   PetscFunctionReturn(0);
3997f9eb7ae2SShri Abhyankar }
3998f9eb7ae2SShri Abhyankar 
399969b4f73cSBarry Smith #endif
4000