xref: /petsc/src/snes/interface/snes.c (revision 1fb7b255396822dd64a4ea9c656967637c985cf5)
1af0996ceSBarry Smith #include <petsc/private/snesimpl.h>      /*I "petscsnes.h"  I*/
207475bc1SBarry Smith #include <petscdmshell.h>
3d96771aaSLisandro Dalcin #include <petscdraw.h>
4a01aa210SMatthew G. Knepley #include <petscds.h>
534b4d3a8SMatthew G. Knepley #include <petscdmadaptor.h>
606fc46c8SMatthew G. Knepley #include <petscconvest.h>
79b94acceSBarry Smith 
8ace3abfcSBarry Smith PetscBool         SNESRegisterAllCalled = PETSC_FALSE;
90298fd71SBarry Smith PetscFunctionList SNESList              = NULL;
108ba1e511SMatthew Knepley 
118ba1e511SMatthew Knepley /* Logging support */
1222c6f798SBarry Smith PetscClassId  SNES_CLASSID, DMSNES_CLASSID;
13e3ed9ee7SBarry Smith PetscLogEvent SNES_Solve, SNES_Setup, SNES_FunctionEval, SNES_JacobianEval, SNES_NGSEval, SNES_NGSFuncEval, SNES_NPCSolve, SNES_ObjectiveEval;
14a09944afSBarry Smith 
15e113a28aSBarry Smith /*@
16e113a28aSBarry Smith    SNESSetErrorIfNotConverged - Causes SNESSolve() to generate an error if the solver has not converged.
17e113a28aSBarry Smith 
183f9fe445SBarry Smith    Logically Collective on SNES
19e113a28aSBarry Smith 
20e113a28aSBarry Smith    Input Parameters:
21e113a28aSBarry Smith +  snes - iterative context obtained from SNESCreate()
22e113a28aSBarry Smith -  flg - PETSC_TRUE indicates you want the error generated
23e113a28aSBarry Smith 
24e113a28aSBarry Smith    Options database keys:
25e113a28aSBarry Smith .  -snes_error_if_not_converged : this takes an optional truth value (0/1/no/yes/true/false)
26e113a28aSBarry Smith 
27e113a28aSBarry Smith    Level: intermediate
28e113a28aSBarry Smith 
29e113a28aSBarry Smith    Notes:
30e113a28aSBarry Smith     Normally PETSc continues if a linear solver fails to converge, you can call SNESGetConvergedReason() after a SNESSolve()
31e113a28aSBarry Smith     to determine if it has converged.
32e113a28aSBarry Smith 
33e113a28aSBarry Smith .seealso: SNESGetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged()
34e113a28aSBarry Smith @*/
357087cfbeSBarry Smith PetscErrorCode  SNESSetErrorIfNotConverged(SNES snes,PetscBool flg)
36e113a28aSBarry Smith {
37e113a28aSBarry Smith   PetscFunctionBegin;
38e113a28aSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
39acfcf0e5SJed Brown   PetscValidLogicalCollectiveBool(snes,flg,2);
40e113a28aSBarry Smith   snes->errorifnotconverged = flg;
41e113a28aSBarry Smith   PetscFunctionReturn(0);
42e113a28aSBarry Smith }
43e113a28aSBarry Smith 
44e113a28aSBarry Smith /*@
45e113a28aSBarry Smith    SNESGetErrorIfNotConverged - Will SNESSolve() generate an error if the solver does not converge?
46e113a28aSBarry Smith 
47e113a28aSBarry Smith    Not Collective
48e113a28aSBarry Smith 
49e113a28aSBarry Smith    Input Parameter:
50e113a28aSBarry Smith .  snes - iterative context obtained from SNESCreate()
51e113a28aSBarry Smith 
52e113a28aSBarry Smith    Output Parameter:
53e113a28aSBarry Smith .  flag - PETSC_TRUE if it will generate an error, else PETSC_FALSE
54e113a28aSBarry Smith 
55e113a28aSBarry Smith    Level: intermediate
56e113a28aSBarry Smith 
57e113a28aSBarry Smith .seealso:  SNESSetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged()
58e113a28aSBarry Smith @*/
597087cfbeSBarry Smith PetscErrorCode  SNESGetErrorIfNotConverged(SNES snes,PetscBool  *flag)
60e113a28aSBarry Smith {
61e113a28aSBarry Smith   PetscFunctionBegin;
62e113a28aSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
63534a8f05SLisandro Dalcin   PetscValidBoolPointer(flag,2);
64e113a28aSBarry Smith   *flag = snes->errorifnotconverged;
65e113a28aSBarry Smith   PetscFunctionReturn(0);
66e113a28aSBarry Smith }
67e113a28aSBarry Smith 
684fc747eaSLawrence Mitchell /*@
694fc747eaSLawrence Mitchell     SNESSetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution?
704fc747eaSLawrence Mitchell 
714fc747eaSLawrence Mitchell    Logically Collective on SNES
724fc747eaSLawrence Mitchell 
734fc747eaSLawrence Mitchell     Input Parameters:
744fc747eaSLawrence Mitchell +   snes - the shell SNES
754fc747eaSLawrence Mitchell -   flg - is the residual computed?
764fc747eaSLawrence Mitchell 
774fc747eaSLawrence Mitchell    Level: advanced
784fc747eaSLawrence Mitchell 
794fc747eaSLawrence Mitchell .seealso: SNESGetAlwaysComputesFinalResidual()
804fc747eaSLawrence Mitchell @*/
814fc747eaSLawrence Mitchell PetscErrorCode  SNESSetAlwaysComputesFinalResidual(SNES snes, PetscBool flg)
824fc747eaSLawrence Mitchell {
834fc747eaSLawrence Mitchell   PetscFunctionBegin;
844fc747eaSLawrence Mitchell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
854fc747eaSLawrence Mitchell   snes->alwayscomputesfinalresidual = flg;
864fc747eaSLawrence Mitchell   PetscFunctionReturn(0);
874fc747eaSLawrence Mitchell }
884fc747eaSLawrence Mitchell 
894fc747eaSLawrence Mitchell /*@
904fc747eaSLawrence Mitchell     SNESGetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution?
914fc747eaSLawrence Mitchell 
924fc747eaSLawrence Mitchell    Logically Collective on SNES
934fc747eaSLawrence Mitchell 
944fc747eaSLawrence Mitchell     Input Parameter:
954fc747eaSLawrence Mitchell .   snes - the shell SNES
964fc747eaSLawrence Mitchell 
974fc747eaSLawrence Mitchell     Output Parameter:
984fc747eaSLawrence Mitchell .   flg - is the residual computed?
994fc747eaSLawrence Mitchell 
1004fc747eaSLawrence Mitchell    Level: advanced
1014fc747eaSLawrence Mitchell 
1024fc747eaSLawrence Mitchell .seealso: SNESSetAlwaysComputesFinalResidual()
1034fc747eaSLawrence Mitchell @*/
1044fc747eaSLawrence Mitchell PetscErrorCode  SNESGetAlwaysComputesFinalResidual(SNES snes, PetscBool *flg)
1054fc747eaSLawrence Mitchell {
1064fc747eaSLawrence Mitchell   PetscFunctionBegin;
1074fc747eaSLawrence Mitchell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1084fc747eaSLawrence Mitchell   *flg = snes->alwayscomputesfinalresidual;
1094fc747eaSLawrence Mitchell   PetscFunctionReturn(0);
1104fc747eaSLawrence Mitchell }
1114fc747eaSLawrence Mitchell 
112e725d27bSBarry Smith /*@
113bf388a1fSBarry Smith    SNESSetFunctionDomainError - tells SNES that the input vector to your SNESFunction is not
1144936397dSBarry Smith      in the functions domain. For example, negative pressure.
1154936397dSBarry Smith 
1163f9fe445SBarry Smith    Logically Collective on SNES
1174936397dSBarry Smith 
1184936397dSBarry Smith    Input Parameters:
1196a388c36SPeter Brune .  snes - the SNES context
1204936397dSBarry Smith 
12128529972SSatish Balay    Level: advanced
1224936397dSBarry Smith 
123bf388a1fSBarry Smith .seealso: SNESCreate(), SNESSetFunction(), SNESFunction
1244936397dSBarry Smith @*/
1257087cfbeSBarry Smith PetscErrorCode  SNESSetFunctionDomainError(SNES snes)
1264936397dSBarry Smith {
1274936397dSBarry Smith   PetscFunctionBegin;
1280700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
129422a814eSBarry Smith   if (snes->errorifnotconverged) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates input vector is not in the function domain");
1304936397dSBarry Smith   snes->domainerror = PETSC_TRUE;
1314936397dSBarry Smith   PetscFunctionReturn(0);
1324936397dSBarry Smith }
1334936397dSBarry Smith 
1346a388c36SPeter Brune /*@
13507b62357SFande Kong    SNESSetJacobianDomainError - tells SNES that computeJacobian does not make sense any more. For example there is a negative element transformation.
13607b62357SFande Kong 
13707b62357SFande Kong    Logically Collective on SNES
13807b62357SFande Kong 
13907b62357SFande Kong    Input Parameters:
14007b62357SFande Kong .  snes - the SNES context
14107b62357SFande Kong 
14207b62357SFande Kong    Level: advanced
14307b62357SFande Kong 
14407b62357SFande Kong .seealso: SNESCreate(), SNESSetFunction(), SNESFunction(), SNESSetFunctionDomainError()
14507b62357SFande Kong @*/
14607b62357SFande Kong PetscErrorCode SNESSetJacobianDomainError(SNES snes)
14707b62357SFande Kong {
14807b62357SFande Kong   PetscFunctionBegin;
14907b62357SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
15007b62357SFande Kong   if (snes->errorifnotconverged) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates computeJacobian does not make sense");
15107b62357SFande Kong   snes->jacobiandomainerror = PETSC_TRUE;
15207b62357SFande Kong   PetscFunctionReturn(0);
15307b62357SFande Kong }
15407b62357SFande Kong 
15507b62357SFande Kong /*@
156b351a90bSFande Kong    SNESSetCheckJacobianDomainError - if or not to check jacobian domain error after each Jacobian evaluation. By default, we check Jacobian domain error
157b351a90bSFande Kong    in the debug mode, and do not check it in the optimized mode.
158b351a90bSFande Kong 
159b351a90bSFande Kong    Logically Collective on SNES
160b351a90bSFande Kong 
161b351a90bSFande Kong    Input Parameters:
162a2b725a8SWilliam Gropp +  snes - the SNES context
163a2b725a8SWilliam Gropp -  flg  - indicates if or not to check jacobian domain error after each Jacobian evaluation
164b351a90bSFande Kong 
165b351a90bSFande Kong    Level: advanced
166b351a90bSFande Kong 
1678383d7d7SFande Kong .seealso: SNESCreate(), SNESSetFunction(), SNESFunction(), SNESSetFunctionDomainError(), SNESGetCheckJacobianDomainError()
168b351a90bSFande Kong @*/
169b351a90bSFande Kong PetscErrorCode SNESSetCheckJacobianDomainError(SNES snes, PetscBool flg)
170b351a90bSFande Kong {
171b351a90bSFande Kong   PetscFunctionBegin;
172b351a90bSFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
173b351a90bSFande Kong   snes->checkjacdomainerror = flg;
174b351a90bSFande Kong   PetscFunctionReturn(0);
175b351a90bSFande Kong }
176b351a90bSFande Kong 
177b351a90bSFande Kong /*@
1788383d7d7SFande Kong    SNESGetCheckJacobianDomainError - Get an indicator whether or not we are checking Jacobian domain errors after each Jacobian evaluation.
1798383d7d7SFande Kong 
1808383d7d7SFande Kong    Logically Collective on SNES
1818383d7d7SFande Kong 
1828383d7d7SFande Kong    Input Parameters:
1838383d7d7SFande Kong .  snes - the SNES context
1848383d7d7SFande Kong 
1858383d7d7SFande Kong    Output Parameters:
1868383d7d7SFande Kong .  flg  - PETSC_FALSE indicates that we don't check jacobian domain errors after each Jacobian evaluation
1878383d7d7SFande Kong 
1888383d7d7SFande Kong    Level: advanced
1898383d7d7SFande Kong 
1908383d7d7SFande Kong .seealso: SNESCreate(), SNESSetFunction(), SNESFunction(), SNESSetFunctionDomainError(), SNESSetCheckJacobianDomainError()
1918383d7d7SFande Kong @*/
1928383d7d7SFande Kong PetscErrorCode SNESGetCheckJacobianDomainError(SNES snes, PetscBool *flg)
1938383d7d7SFande Kong {
1948383d7d7SFande Kong   PetscFunctionBegin;
1958383d7d7SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
196534a8f05SLisandro Dalcin   PetscValidBoolPointer(flg,2);
1978383d7d7SFande Kong   *flg = snes->checkjacdomainerror;
1988383d7d7SFande Kong   PetscFunctionReturn(0);
1998383d7d7SFande Kong }
2008383d7d7SFande Kong 
2018383d7d7SFande Kong /*@
202c77b2880SPeter Brune    SNESGetFunctionDomainError - Gets the status of the domain error after a call to SNESComputeFunction;
2036a388c36SPeter Brune 
2046a388c36SPeter Brune    Logically Collective on SNES
2056a388c36SPeter Brune 
2066a388c36SPeter Brune    Input Parameters:
2076a388c36SPeter Brune .  snes - the SNES context
2086a388c36SPeter Brune 
2096a388c36SPeter Brune    Output Parameters:
210bf388a1fSBarry Smith .  domainerror - Set to PETSC_TRUE if there's a domain error; PETSC_FALSE otherwise.
2116a388c36SPeter Brune 
2126a388c36SPeter Brune    Level: advanced
2136a388c36SPeter Brune 
214bf388a1fSBarry Smith .seealso: SNESSetFunctionDomainError(), SNESComputeFunction()
2156a388c36SPeter Brune @*/
2166a388c36SPeter Brune PetscErrorCode  SNESGetFunctionDomainError(SNES snes, PetscBool *domainerror)
2176a388c36SPeter Brune {
2186a388c36SPeter Brune   PetscFunctionBegin;
2196a388c36SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
220534a8f05SLisandro Dalcin   PetscValidBoolPointer(domainerror,2);
2216a388c36SPeter Brune   *domainerror = snes->domainerror;
2226a388c36SPeter Brune   PetscFunctionReturn(0);
2236a388c36SPeter Brune }
2246a388c36SPeter Brune 
22507b62357SFande Kong /*@
22607b62357SFande Kong    SNESGetJacobianDomainError - Gets the status of the Jacobian domain error after a call to SNESComputeJacobian;
22707b62357SFande Kong 
22807b62357SFande Kong    Logically Collective on SNES
22907b62357SFande Kong 
23007b62357SFande Kong    Input Parameters:
23107b62357SFande Kong .  snes - the SNES context
23207b62357SFande Kong 
23307b62357SFande Kong    Output Parameters:
23407b62357SFande Kong .  domainerror - Set to PETSC_TRUE if there's a jacobian domain error; PETSC_FALSE otherwise.
23507b62357SFande Kong 
23607b62357SFande Kong    Level: advanced
23707b62357SFande Kong 
23807b62357SFande Kong .seealso: SNESSetFunctionDomainError(), SNESComputeFunction(),SNESGetFunctionDomainError()
23907b62357SFande Kong @*/
24007b62357SFande Kong PetscErrorCode SNESGetJacobianDomainError(SNES snes, PetscBool *domainerror)
24107b62357SFande Kong {
24207b62357SFande Kong   PetscFunctionBegin;
24307b62357SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
244534a8f05SLisandro Dalcin   PetscValidBoolPointer(domainerror,2);
24507b62357SFande Kong   *domainerror = snes->jacobiandomainerror;
24607b62357SFande Kong   PetscFunctionReturn(0);
24707b62357SFande Kong }
24807b62357SFande Kong 
24955849f57SBarry Smith /*@C
25055849f57SBarry Smith   SNESLoad - Loads a SNES that has been stored in binary  with SNESView().
25155849f57SBarry Smith 
25255849f57SBarry Smith   Collective on PetscViewer
25355849f57SBarry Smith 
25455849f57SBarry Smith   Input Parameters:
25555849f57SBarry Smith + newdm - the newly loaded SNES, this needs to have been created with SNESCreate() or
25655849f57SBarry Smith            some related function before a call to SNESLoad().
25755849f57SBarry Smith - viewer - binary file viewer, obtained from PetscViewerBinaryOpen()
25855849f57SBarry Smith 
25955849f57SBarry Smith    Level: intermediate
26055849f57SBarry Smith 
26155849f57SBarry Smith   Notes:
26255849f57SBarry Smith    The type is determined by the data in the file, any type set into the SNES before this call is ignored.
26355849f57SBarry Smith 
26455849f57SBarry Smith   Notes for advanced users:
26555849f57SBarry Smith   Most users should not need to know the details of the binary storage
26655849f57SBarry Smith   format, since SNESLoad() and TSView() completely hide these details.
26755849f57SBarry Smith   But for anyone who's interested, the standard binary matrix storage
26855849f57SBarry Smith   format is
26955849f57SBarry Smith .vb
27055849f57SBarry Smith      has not yet been determined
27155849f57SBarry Smith .ve
27255849f57SBarry Smith 
27355849f57SBarry Smith .seealso: PetscViewerBinaryOpen(), SNESView(), MatLoad(), VecLoad()
27455849f57SBarry Smith @*/
2752d53ad75SBarry Smith PetscErrorCode  SNESLoad(SNES snes, PetscViewer viewer)
27655849f57SBarry Smith {
27755849f57SBarry Smith   PetscErrorCode ierr;
27855849f57SBarry Smith   PetscBool      isbinary;
279060da220SMatthew G. Knepley   PetscInt       classid;
28055849f57SBarry Smith   char           type[256];
28155849f57SBarry Smith   KSP            ksp;
2822d53ad75SBarry Smith   DM             dm;
2832d53ad75SBarry Smith   DMSNES         dmsnes;
28455849f57SBarry Smith 
28555849f57SBarry Smith   PetscFunctionBegin;
2862d53ad75SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
28755849f57SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
28855849f57SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
28955849f57SBarry Smith   if (!isbinary) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen()");
29055849f57SBarry Smith 
291060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,&classid,1,NULL,PETSC_INT);CHKERRQ(ierr);
292ce94432eSBarry Smith   if (classid != SNES_FILE_CLASSID) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_WRONG,"Not SNES next in file");
293060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,type,256,NULL,PETSC_CHAR);CHKERRQ(ierr);
2942d53ad75SBarry Smith   ierr = SNESSetType(snes, type);CHKERRQ(ierr);
2952d53ad75SBarry Smith   if (snes->ops->load) {
2962d53ad75SBarry Smith     ierr = (*snes->ops->load)(snes,viewer);CHKERRQ(ierr);
297f2c2a1b9SBarry Smith   }
2982d53ad75SBarry Smith   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2992d53ad75SBarry Smith   ierr = DMGetDMSNES(dm,&dmsnes);CHKERRQ(ierr);
3002d53ad75SBarry Smith   ierr = DMSNESLoad(dmsnes,viewer);CHKERRQ(ierr);
3012d53ad75SBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
30255849f57SBarry Smith   ierr = KSPLoad(ksp,viewer);CHKERRQ(ierr);
30355849f57SBarry Smith   PetscFunctionReturn(0);
30455849f57SBarry Smith }
3056a388c36SPeter Brune 
3069804daf3SBarry Smith #include <petscdraw.h>
307e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
308e04113cfSBarry Smith #include <petscviewersaws.h>
309bfb97211SBarry Smith #endif
3108404b7f3SBarry Smith 
311fe2efc57SMark /*@C
312fe2efc57SMark    SNESViewFromOptions - View from Options
313fe2efc57SMark 
314fe2efc57SMark    Collective on SNES
315fe2efc57SMark 
316fe2efc57SMark    Input Parameters:
317fe2efc57SMark +  A - the application ordering context
318736c3998SJose E. Roman .  obj - Optional object
319736c3998SJose E. Roman -  name - command line option
320fe2efc57SMark 
321fe2efc57SMark    Level: intermediate
322fe2efc57SMark .seealso:  SNES, SNESView, PetscObjectViewFromOptions(), SNESCreate()
323fe2efc57SMark @*/
324fe2efc57SMark PetscErrorCode  SNESViewFromOptions(SNES A,PetscObject obj,const char name[])
325fe2efc57SMark {
326fe2efc57SMark   PetscErrorCode ierr;
327fe2efc57SMark 
328fe2efc57SMark   PetscFunctionBegin;
329fe2efc57SMark   PetscValidHeaderSpecific(A,SNES_CLASSID,1);
330fe2efc57SMark   ierr = PetscObjectViewFromOptions((PetscObject)A,obj,name);CHKERRQ(ierr);
331fe2efc57SMark   PetscFunctionReturn(0);
332fe2efc57SMark }
333fe2efc57SMark 
334789d8953SBarry Smith PETSC_EXTERN PetscErrorCode SNESComputeJacobian_DMDA(SNES,Vec,Mat,Mat,void*);
335789d8953SBarry Smith 
3367e2c5f70SBarry Smith /*@C
3379b94acceSBarry Smith    SNESView - Prints the SNES data structure.
3389b94acceSBarry Smith 
3394c49b128SBarry Smith    Collective on SNES
340fee21e36SBarry Smith 
341c7afd0dbSLois Curfman McInnes    Input Parameters:
342c7afd0dbSLois Curfman McInnes +  SNES - the SNES context
343c7afd0dbSLois Curfman McInnes -  viewer - visualization context
344c7afd0dbSLois Curfman McInnes 
3459b94acceSBarry Smith    Options Database Key:
346c8a8ba5cSLois Curfman McInnes .  -snes_view - Calls SNESView() at end of SNESSolve()
3479b94acceSBarry Smith 
3489b94acceSBarry Smith    Notes:
3499b94acceSBarry Smith    The available visualization contexts include
350b0a32e0cSBarry Smith +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
351b0a32e0cSBarry Smith -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
352c8a8ba5cSLois Curfman McInnes          output where only the first processor opens
353c8a8ba5cSLois Curfman McInnes          the file.  All other processors send their
354c8a8ba5cSLois Curfman McInnes          data to the first processor to print.
3559b94acceSBarry Smith 
356052bf0daSPierre Jolivet    The available formats include
357052bf0daSPierre Jolivet +     PETSC_VIEWER_DEFAULT - standard output (default)
358052bf0daSPierre Jolivet -     PETSC_VIEWER_ASCII_INFO_DETAIL - more verbose output for SNESNASM
359052bf0daSPierre Jolivet 
3603e081fefSLois Curfman McInnes    The user can open an alternative visualization context with
361b0a32e0cSBarry Smith    PetscViewerASCIIOpen() - output to a specified file.
3629b94acceSBarry Smith 
363595c91d4SBarry Smith   In the debugger you can do "call SNESView(snes,0)" to display the SNES solver. (The same holds for any PETSc object viewer).
364595c91d4SBarry Smith 
36536851e7fSLois Curfman McInnes    Level: beginner
36636851e7fSLois Curfman McInnes 
367b0a32e0cSBarry Smith .seealso: PetscViewerASCIIOpen()
3689b94acceSBarry Smith @*/
3697087cfbeSBarry Smith PetscErrorCode  SNESView(SNES snes,PetscViewer viewer)
3709b94acceSBarry Smith {
371fa9f3622SBarry Smith   SNESKSPEW      *kctx;
372dfbe8321SBarry Smith   PetscErrorCode ierr;
37394b7f48cSBarry Smith   KSP            ksp;
3747f1410a3SPeter Brune   SNESLineSearch linesearch;
37572a02f06SBarry Smith   PetscBool      iascii,isstring,isbinary,isdraw;
3762d53ad75SBarry Smith   DMSNES         dmsnes;
377e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
378536b137fSBarry Smith   PetscBool      issaws;
379bfb97211SBarry Smith #endif
3809b94acceSBarry Smith 
3813a40ed3dSBarry Smith   PetscFunctionBegin;
3820700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3833050cee2SBarry Smith   if (!viewer) {
384ce94432eSBarry Smith     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&viewer);CHKERRQ(ierr);
3853050cee2SBarry Smith   }
3860700a824SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
387c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,viewer,2);
38874679c65SBarry Smith 
389251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
390251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr);
39155849f57SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
39272a02f06SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
393e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
394536b137fSBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSAWS,&issaws);CHKERRQ(ierr);
395bfb97211SBarry Smith #endif
39632077d6dSBarry Smith   if (iascii) {
397dc0571f2SMatthew G. Knepley     SNESNormSchedule normschedule;
3988404b7f3SBarry Smith     DM               dm;
3998404b7f3SBarry Smith     PetscErrorCode   (*cJ)(SNES,Vec,Mat,Mat,void*);
4008404b7f3SBarry Smith     void             *ctx;
401789d8953SBarry Smith     const char       *pre = "";
402dc0571f2SMatthew G. Knepley 
403dae58748SBarry Smith     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)snes,viewer);CHKERRQ(ierr);
404fce1e034SJed Brown     if (!snes->setupcalled) {
405fce1e034SJed Brown       ierr = PetscViewerASCIIPrintf(viewer,"  SNES has not been set up so information may be incomplete\n");CHKERRQ(ierr);
406fce1e034SJed Brown     }
407e7788613SBarry Smith     if (snes->ops->view) {
408b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
409e7788613SBarry Smith       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
410b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
4110ef38995SBarry Smith     }
41277431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr);
41357622a8eSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  tolerances: relative=%g, absolute=%g, solution=%g\n",(double)snes->rtol,(double)snes->abstol,(double)snes->stol);CHKERRQ(ierr);
414efd4aadfSBarry Smith     if (snes->usesksp) {
41577431f27SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr);
416efd4aadfSBarry Smith     }
41777431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr);
418dc0571f2SMatthew G. Knepley     ierr = SNESGetNormSchedule(snes, &normschedule);CHKERRQ(ierr);
419dc0571f2SMatthew G. Knepley     if (normschedule > 0) {ierr = PetscViewerASCIIPrintf(viewer,"  norm schedule %s\n",SNESNormSchedules[normschedule]);CHKERRQ(ierr);}
42017fe4bdfSPeter Brune     if (snes->gridsequence) {
42117fe4bdfSPeter Brune       ierr = PetscViewerASCIIPrintf(viewer,"  total number of grid sequence refinements=%D\n",snes->gridsequence);CHKERRQ(ierr);
42217fe4bdfSPeter Brune     }
4239b94acceSBarry Smith     if (snes->ksp_ewconv) {
424fa9f3622SBarry Smith       kctx = (SNESKSPEW*)snes->kspconvctx;
4259b94acceSBarry Smith       if (kctx) {
42677431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"  Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr);
42757622a8eSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"    rtol_0=%g, rtol_max=%g, threshold=%g\n",(double)kctx->rtol_0,(double)kctx->rtol_max,(double)kctx->threshold);CHKERRQ(ierr);
42857622a8eSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"    gamma=%g, alpha=%g, alpha2=%g\n",(double)kctx->gamma,(double)kctx->alpha,(double)kctx->alpha2);CHKERRQ(ierr);
4299b94acceSBarry Smith       }
4309b94acceSBarry Smith     }
431eb1f6c34SBarry Smith     if (snes->lagpreconditioner == -1) {
432eb1f6c34SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Preconditioned is never rebuilt\n");CHKERRQ(ierr);
433eb1f6c34SBarry Smith     } else if (snes->lagpreconditioner > 1) {
434eb1f6c34SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Preconditioned is rebuilt every %D new Jacobians\n",snes->lagpreconditioner);CHKERRQ(ierr);
435eb1f6c34SBarry Smith     }
436eb1f6c34SBarry Smith     if (snes->lagjacobian == -1) {
437eb1f6c34SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is never rebuilt\n");CHKERRQ(ierr);
438eb1f6c34SBarry Smith     } else if (snes->lagjacobian > 1) {
43942f4f86dSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is rebuilt every %D SNES iterations\n",snes->lagjacobian);CHKERRQ(ierr);
440eb1f6c34SBarry Smith     }
4418404b7f3SBarry Smith     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4428404b7f3SBarry Smith     ierr = DMSNESGetJacobian(dm,&cJ,&ctx);CHKERRQ(ierr);
443789d8953SBarry Smith     if (snes->mf_operator) {
444789d8953SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is applied matrix-free with differencing\n");CHKERRQ(ierr);
445789d8953SBarry Smith       pre  = "Preconditioning ";
446789d8953SBarry Smith     }
4478404b7f3SBarry Smith     if (cJ == SNESComputeJacobianDefault) {
448789d8953SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  %sJacobian is built using finite differences one column at a time\n",pre);CHKERRQ(ierr);
4498404b7f3SBarry Smith     } else if (cJ == SNESComputeJacobianDefaultColor) {
450789d8953SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  %sJacobian is built using finite differences with coloring\n",pre);CHKERRQ(ierr);
451789d8953SBarry Smith     /* it slightly breaks data encapsulation for access the DMDA information directly */
452789d8953SBarry Smith     } else if (cJ == SNESComputeJacobian_DMDA) {
453789d8953SBarry Smith       MatFDColoring fdcoloring;
454789d8953SBarry Smith       ierr = PetscObjectQuery((PetscObject)dm,"DMDASNES_FDCOLORING",(PetscObject*)&fdcoloring);CHKERRQ(ierr);
455789d8953SBarry Smith       if (fdcoloring) {
456789d8953SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"  %sJacobian is built using colored finite differences on a DMDA\n",pre);CHKERRQ(ierr);
457789d8953SBarry Smith       } else {
458789d8953SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"  %sJacobian is built using a DMDA local Jacobian\n",pre);CHKERRQ(ierr);
459789d8953SBarry Smith       }
460789d8953SBarry Smith     } else if (snes->mf) {
461789d8953SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is applied matrix-free with differencing, no explict Jacobian\n");CHKERRQ(ierr);
4628404b7f3SBarry Smith     }
4630f5bd95cSBarry Smith   } else if (isstring) {
464317d6ea6SBarry Smith     const char *type;
465454a90a3SBarry Smith     ierr = SNESGetType(snes,&type);CHKERRQ(ierr);
46636a9e3b9SBarry Smith     ierr = PetscViewerStringSPrintf(viewer," SNESType: %-7.7s",type);CHKERRQ(ierr);
46736a9e3b9SBarry Smith     if (snes->ops->view) {ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);}
46855849f57SBarry Smith   } else if (isbinary) {
46955849f57SBarry Smith     PetscInt    classid = SNES_FILE_CLASSID;
47055849f57SBarry Smith     MPI_Comm    comm;
47155849f57SBarry Smith     PetscMPIInt rank;
47255849f57SBarry Smith     char        type[256];
47355849f57SBarry Smith 
47455849f57SBarry Smith     ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr);
475ffc4695bSBarry Smith     ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr);
47655849f57SBarry Smith     if (!rank) {
477f253e43cSLisandro Dalcin       ierr = PetscViewerBinaryWrite(viewer,&classid,1,PETSC_INT);CHKERRQ(ierr);
47889d949e2SBarry Smith       ierr = PetscStrncpy(type,((PetscObject)snes)->type_name,sizeof(type));CHKERRQ(ierr);
479f253e43cSLisandro Dalcin       ierr = PetscViewerBinaryWrite(viewer,type,sizeof(type),PETSC_CHAR);CHKERRQ(ierr);
48055849f57SBarry Smith     }
48155849f57SBarry Smith     if (snes->ops->view) {
48255849f57SBarry Smith       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
48355849f57SBarry Smith     }
48472a02f06SBarry Smith   } else if (isdraw) {
48572a02f06SBarry Smith     PetscDraw draw;
48672a02f06SBarry Smith     char      str[36];
48789fd9fafSBarry Smith     PetscReal x,y,bottom,h;
48872a02f06SBarry Smith 
48972a02f06SBarry Smith     ierr   = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
49072a02f06SBarry Smith     ierr   = PetscDrawGetCurrentPoint(draw,&x,&y);CHKERRQ(ierr);
491a126751eSBarry Smith     ierr   = PetscStrncpy(str,"SNES: ",sizeof(str));CHKERRQ(ierr);
492a126751eSBarry Smith     ierr   = PetscStrlcat(str,((PetscObject)snes)->type_name,sizeof(str));CHKERRQ(ierr);
49351fa3d41SBarry Smith     ierr   = PetscDrawStringBoxed(draw,x,y,PETSC_DRAW_BLUE,PETSC_DRAW_BLACK,str,NULL,&h);CHKERRQ(ierr);
49489fd9fafSBarry Smith     bottom = y - h;
49572a02f06SBarry Smith     ierr   = PetscDrawPushCurrentPoint(draw,x,bottom);CHKERRQ(ierr);
496c4646bacSPeter Brune     if (snes->ops->view) {
497c4646bacSPeter Brune       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
498c4646bacSPeter Brune     }
499e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
500536b137fSBarry Smith   } else if (issaws) {
501d45a07a7SBarry Smith     PetscMPIInt rank;
5022657e9d9SBarry Smith     const char *name;
503d45a07a7SBarry Smith 
5042657e9d9SBarry Smith     ierr = PetscObjectGetName((PetscObject)snes,&name);CHKERRQ(ierr);
505ffc4695bSBarry Smith     ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr);
506d45a07a7SBarry Smith     if (!((PetscObject)snes)->amsmem && !rank) {
507d45a07a7SBarry Smith       char       dir[1024];
508d45a07a7SBarry Smith 
509e04113cfSBarry Smith       ierr = PetscObjectViewSAWs((PetscObject)snes,viewer);CHKERRQ(ierr);
5102657e9d9SBarry Smith       ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/its",name);CHKERRQ(ierr);
5112657e9d9SBarry Smith       PetscStackCallSAWs(SAWs_Register,(dir,&snes->iter,1,SAWs_READ,SAWs_INT));
512bfb97211SBarry Smith       if (!snes->conv_hist) {
513a0931e03SBarry Smith         ierr = SNESSetConvergenceHistory(snes,NULL,NULL,PETSC_DECIDE,PETSC_TRUE);CHKERRQ(ierr);
514bfb97211SBarry Smith       }
5152657e9d9SBarry Smith       ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/conv_hist",name);CHKERRQ(ierr);
5162657e9d9SBarry Smith       PetscStackCallSAWs(SAWs_Register,(dir,snes->conv_hist,10,SAWs_READ,SAWs_DOUBLE));
517f05ece33SBarry Smith     }
518bfb97211SBarry Smith #endif
51972a02f06SBarry Smith   }
52072a02f06SBarry Smith   if (snes->linesearch) {
5217601faf0SJed Brown     ierr = SNESGetLineSearch(snes, &linesearch);CHKERRQ(ierr);
52285928a98SStefano Zampini     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
52372a02f06SBarry Smith     ierr = SNESLineSearchView(linesearch, viewer);CHKERRQ(ierr);
52485928a98SStefano Zampini     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
52519bcc07fSBarry Smith   }
526efd4aadfSBarry Smith   if (snes->npc && snes->usesnpc) {
52785928a98SStefano Zampini     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
528efd4aadfSBarry Smith     ierr = SNESView(snes->npc, viewer);CHKERRQ(ierr);
52985928a98SStefano Zampini     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
5304a0c5b0cSMatthew G Knepley   }
5312d53ad75SBarry Smith   ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
5322d53ad75SBarry Smith   ierr = DMGetDMSNES(snes->dm,&dmsnes);CHKERRQ(ierr);
5332d53ad75SBarry Smith   ierr = DMSNESView(dmsnes, viewer);CHKERRQ(ierr);
5342d53ad75SBarry Smith   ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
5352c155ee1SBarry Smith   if (snes->usesksp) {
5362c155ee1SBarry Smith     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
53785928a98SStefano Zampini     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
53894b7f48cSBarry Smith     ierr = KSPView(ksp,viewer);CHKERRQ(ierr);
53985928a98SStefano Zampini     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
5402c155ee1SBarry Smith   }
54172a02f06SBarry Smith   if (isdraw) {
54272a02f06SBarry Smith     PetscDraw draw;
54372a02f06SBarry Smith     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
54472a02f06SBarry Smith     ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr);
5457f1410a3SPeter Brune   }
5463a40ed3dSBarry Smith   PetscFunctionReturn(0);
5479b94acceSBarry Smith }
5489b94acceSBarry Smith 
54976b2cf59SMatthew Knepley /*
55076b2cf59SMatthew Knepley   We retain a list of functions that also take SNES command
55176b2cf59SMatthew Knepley   line options. These are called at the end SNESSetFromOptions()
55276b2cf59SMatthew Knepley */
55376b2cf59SMatthew Knepley #define MAXSETFROMOPTIONS 5
554a7cc72afSBarry Smith static PetscInt numberofsetfromoptions;
5556849ba73SBarry Smith static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES);
55676b2cf59SMatthew Knepley 
557ac226902SBarry Smith /*@C
55876b2cf59SMatthew Knepley   SNESAddOptionsChecker - Adds an additional function to check for SNES options.
55976b2cf59SMatthew Knepley 
56076b2cf59SMatthew Knepley   Not Collective
56176b2cf59SMatthew Knepley 
56276b2cf59SMatthew Knepley   Input Parameter:
56376b2cf59SMatthew Knepley . snescheck - function that checks for options
56476b2cf59SMatthew Knepley 
56576b2cf59SMatthew Knepley   Level: developer
56676b2cf59SMatthew Knepley 
56776b2cf59SMatthew Knepley .seealso: SNESSetFromOptions()
56876b2cf59SMatthew Knepley @*/
5697087cfbeSBarry Smith PetscErrorCode  SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES))
57076b2cf59SMatthew Knepley {
57176b2cf59SMatthew Knepley   PetscFunctionBegin;
572f23aa3ddSBarry Smith   if (numberofsetfromoptions >= MAXSETFROMOPTIONS) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS);
57376b2cf59SMatthew Knepley   othersetfromoptions[numberofsetfromoptions++] = snescheck;
57476b2cf59SMatthew Knepley   PetscFunctionReturn(0);
57576b2cf59SMatthew Knepley }
57676b2cf59SMatthew Knepley 
57725acbd8eSLisandro Dalcin PETSC_INTERN PetscErrorCode SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*);
578aa3661deSLisandro Dalcin 
579ace3abfcSBarry Smith static PetscErrorCode SNESSetUpMatrixFree_Private(SNES snes, PetscBool hasOperator, PetscInt version)
580aa3661deSLisandro Dalcin {
581aa3661deSLisandro Dalcin   Mat            J;
582aa3661deSLisandro Dalcin   PetscErrorCode ierr;
583895c21f2SBarry Smith   MatNullSpace   nullsp;
584aa3661deSLisandro Dalcin 
585aa3661deSLisandro Dalcin   PetscFunctionBegin;
5860700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
587aa3661deSLisandro Dalcin 
58898613b67SLisandro Dalcin   if (!snes->vec_func && (snes->jacobian || snes->jacobian_pre)) {
58998613b67SLisandro Dalcin     Mat A = snes->jacobian, B = snes->jacobian_pre;
5902a7a6963SBarry Smith     ierr = MatCreateVecs(A ? A : B, NULL,&snes->vec_func);CHKERRQ(ierr);
59198613b67SLisandro Dalcin   }
59298613b67SLisandro Dalcin 
593aa3661deSLisandro Dalcin   if (version == 1) {
594aa3661deSLisandro Dalcin     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
59598613b67SLisandro Dalcin     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
5969c6ac3b3SBarry Smith     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
597aa3661deSLisandro Dalcin   } else if (version == 2) {
598e32f2f54SBarry Smith     if (!snes->vec_func) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"SNESSetFunction() must be called first");
599570b7f6dSBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_REAL_SINGLE) && !defined(PETSC_USE_REAL___FLOAT128) && !defined(PETSC_USE_REAL___FP16)
600aa3661deSLisandro Dalcin     ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_func,&J);CHKERRQ(ierr);
601aa3661deSLisandro Dalcin #else
6022479783cSJose E. Roman     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP, "matrix-free operator routines (version 2)");
603aa3661deSLisandro Dalcin #endif
6042479783cSJose E. Roman   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "matrix-free operator routines, only version 1 and 2");
605aa3661deSLisandro Dalcin 
606895c21f2SBarry Smith   /* attach any user provided null space that was on Amat to the newly created matrix free matrix */
607895c21f2SBarry Smith   if (snes->jacobian) {
608895c21f2SBarry Smith     ierr = MatGetNullSpace(snes->jacobian,&nullsp);CHKERRQ(ierr);
609895c21f2SBarry Smith     if (nullsp) {
610895c21f2SBarry Smith       ierr = MatSetNullSpace(J,nullsp);CHKERRQ(ierr);
611895c21f2SBarry Smith     }
612895c21f2SBarry Smith   }
613895c21f2SBarry Smith 
614aa3661deSLisandro Dalcin   ierr = PetscInfo1(snes,"Setting default matrix-free operator routines (version %D)\n", version);CHKERRQ(ierr);
615d3462f78SMatthew Knepley   if (hasOperator) {
6163232da50SPeter Brune 
617aa3661deSLisandro Dalcin     /* This version replaces the user provided Jacobian matrix with a
618aa3661deSLisandro Dalcin        matrix-free version but still employs the user-provided preconditioner matrix. */
6199e5d0892SLisandro Dalcin     ierr = SNESSetJacobian(snes,J,NULL,NULL,NULL);CHKERRQ(ierr);
620aa3661deSLisandro Dalcin   } else {
621aa3661deSLisandro Dalcin     /* This version replaces both the user-provided Jacobian and the user-
6223232da50SPeter Brune      provided preconditioner Jacobian with the default matrix free version. */
623efd4aadfSBarry Smith     if ((snes->npcside== PC_LEFT) && snes->npc) {
6249e5d0892SLisandro Dalcin       if (!snes->jacobian){ierr = SNESSetJacobian(snes,J,NULL,NULL,NULL);CHKERRQ(ierr);}
625172a4300SPeter Brune     } else {
626789d8953SBarry Smith       KSP       ksp;
627789d8953SBarry Smith       PC        pc;
628789d8953SBarry Smith       PetscBool match;
629789d8953SBarry Smith 
6309e5d0892SLisandro Dalcin       ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,NULL);CHKERRQ(ierr);
631aa3661deSLisandro Dalcin       /* Force no preconditioner */
632aa3661deSLisandro Dalcin       ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
633aa3661deSLisandro Dalcin       ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
634251f4c67SDmitry Karpeev       ierr = PetscObjectTypeCompare((PetscObject)pc,PCSHELL,&match);CHKERRQ(ierr);
635aa3661deSLisandro Dalcin       if (!match) {
636aa3661deSLisandro Dalcin         ierr = PetscInfo(snes,"Setting default matrix-free preconditioner routines\nThat is no preconditioner is being used\n");CHKERRQ(ierr);
637aa3661deSLisandro Dalcin         ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr);
638aa3661deSLisandro Dalcin       }
639aa3661deSLisandro Dalcin     }
640789d8953SBarry Smith   }
6416bf464f9SBarry Smith   ierr = MatDestroy(&J);CHKERRQ(ierr);
642aa3661deSLisandro Dalcin   PetscFunctionReturn(0);
643aa3661deSLisandro Dalcin }
644aa3661deSLisandro Dalcin 
645dfe15315SJed Brown static PetscErrorCode DMRestrictHook_SNESVecSol(DM dmfine,Mat Restrict,Vec Rscale,Mat Inject,DM dmcoarse,void *ctx)
646dfe15315SJed Brown {
647dfe15315SJed Brown   SNES           snes = (SNES)ctx;
648dfe15315SJed Brown   PetscErrorCode ierr;
6490298fd71SBarry Smith   Vec            Xfine,Xfine_named = NULL,Xcoarse;
650dfe15315SJed Brown 
651dfe15315SJed Brown   PetscFunctionBegin;
65216ebb321SJed Brown   if (PetscLogPrintInfo) {
65316ebb321SJed Brown     PetscInt finelevel,coarselevel,fineclevel,coarseclevel;
65416ebb321SJed Brown     ierr = DMGetRefineLevel(dmfine,&finelevel);CHKERRQ(ierr);
65516ebb321SJed Brown     ierr = DMGetCoarsenLevel(dmfine,&fineclevel);CHKERRQ(ierr);
65616ebb321SJed Brown     ierr = DMGetRefineLevel(dmcoarse,&coarselevel);CHKERRQ(ierr);
65716ebb321SJed Brown     ierr = DMGetCoarsenLevel(dmcoarse,&coarseclevel);CHKERRQ(ierr);
65816ebb321SJed Brown     ierr = PetscInfo4(dmfine,"Restricting SNES solution vector from level %D-%D to level %D-%D\n",finelevel,fineclevel,coarselevel,coarseclevel);CHKERRQ(ierr);
65916ebb321SJed Brown   }
660dfe15315SJed Brown   if (dmfine == snes->dm) Xfine = snes->vec_sol;
661dfe15315SJed Brown   else {
662dfe15315SJed Brown     ierr  = DMGetNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);
663dfe15315SJed Brown     Xfine = Xfine_named;
664dfe15315SJed Brown   }
665dfe15315SJed Brown   ierr = DMGetNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr);
666907f5c5aSLawrence Mitchell   if (Inject) {
667907f5c5aSLawrence Mitchell     ierr = MatRestrict(Inject,Xfine,Xcoarse);CHKERRQ(ierr);
668907f5c5aSLawrence Mitchell   } else {
669dfe15315SJed Brown     ierr = MatRestrict(Restrict,Xfine,Xcoarse);CHKERRQ(ierr);
670dfe15315SJed Brown     ierr = VecPointwiseMult(Xcoarse,Xcoarse,Rscale);CHKERRQ(ierr);
671907f5c5aSLawrence Mitchell   }
672dfe15315SJed Brown   ierr = DMRestoreNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr);
673dfe15315SJed Brown   if (Xfine_named) {ierr = DMRestoreNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);}
674dfe15315SJed Brown   PetscFunctionReturn(0);
675dfe15315SJed Brown }
676dfe15315SJed Brown 
67716ebb321SJed Brown static PetscErrorCode DMCoarsenHook_SNESVecSol(DM dm,DM dmc,void *ctx)
67816ebb321SJed Brown {
67916ebb321SJed Brown   PetscErrorCode ierr;
68016ebb321SJed Brown 
68116ebb321SJed Brown   PetscFunctionBegin;
68216ebb321SJed Brown   ierr = DMCoarsenHookAdd(dmc,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,ctx);CHKERRQ(ierr);
68316ebb321SJed Brown   PetscFunctionReturn(0);
68416ebb321SJed Brown }
68516ebb321SJed Brown 
686a6950cb2SJed Brown /* This may be called to rediscretize the operator on levels of linear multigrid. The DM shuffle is so the user can
687a6950cb2SJed Brown  * safely call SNESGetDM() in their residual evaluation routine. */
68823ee1639SBarry Smith static PetscErrorCode KSPComputeOperators_SNES(KSP ksp,Mat A,Mat B,void *ctx)
689caa4e7f2SJed Brown {
690caa4e7f2SJed Brown   SNES           snes = (SNES)ctx;
691caa4e7f2SJed Brown   PetscErrorCode ierr;
6920298fd71SBarry Smith   Vec            X,Xnamed = NULL;
693dfe15315SJed Brown   DM             dmsave;
6944e269d77SPeter Brune   void           *ctxsave;
69525ce1634SJed Brown   PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*) = NULL;
696caa4e7f2SJed Brown 
697caa4e7f2SJed Brown   PetscFunctionBegin;
698dfe15315SJed Brown   dmsave = snes->dm;
699dfe15315SJed Brown   ierr   = KSPGetDM(ksp,&snes->dm);CHKERRQ(ierr);
700dfe15315SJed Brown   if (dmsave == snes->dm) X = snes->vec_sol; /* We are on the finest level */
701dfe15315SJed Brown   else {                                     /* We are on a coarser level, this vec was initialized using a DM restrict hook */
702dfe15315SJed Brown     ierr = DMGetNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr);
703dfe15315SJed Brown     X    = Xnamed;
7040298fd71SBarry Smith     ierr = SNESGetJacobian(snes,NULL,NULL,&jac,&ctxsave);CHKERRQ(ierr);
7054e269d77SPeter Brune     /* If the DM's don't match up, the MatFDColoring context needed for the jacobian won't match up either -- fixit. */
7068d359177SBarry Smith     if (jac == SNESComputeJacobianDefaultColor) {
7079e5d0892SLisandro Dalcin       ierr = SNESSetJacobian(snes,NULL,NULL,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr);
708dfe15315SJed Brown     }
7094e269d77SPeter Brune   }
7104dde8bb0SMatthew G. Knepley   /* Make sure KSP DM has the Jacobian computation routine */
7114dde8bb0SMatthew G. Knepley   {
7124dde8bb0SMatthew G. Knepley     DMSNES sdm;
7134e269d77SPeter Brune 
7144dde8bb0SMatthew G. Knepley     ierr = DMGetDMSNES(snes->dm, &sdm);CHKERRQ(ierr);
7154dde8bb0SMatthew G. Knepley     if (!sdm->ops->computejacobian) {
7164dde8bb0SMatthew G. Knepley       ierr = DMCopyDMSNES(dmsave, snes->dm);CHKERRQ(ierr);
7174dde8bb0SMatthew G. Knepley     }
7184dde8bb0SMatthew G. Knepley   }
7192b93b426SMatthew G. Knepley   /* Compute the operators */
720d1e9a80fSBarry Smith   ierr = SNESComputeJacobian(snes,X,A,B);CHKERRQ(ierr);
7212b93b426SMatthew G. Knepley   /* Put the previous context back */
7228d359177SBarry Smith   if (snes->dm != dmsave && jac == SNESComputeJacobianDefaultColor) {
7230298fd71SBarry Smith     ierr = SNESSetJacobian(snes,NULL,NULL,jac,ctxsave);CHKERRQ(ierr);
7244e269d77SPeter Brune   }
7254e269d77SPeter Brune 
7262b93b426SMatthew G. Knepley   if (Xnamed) {ierr = DMRestoreNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr);}
727dfe15315SJed Brown   snes->dm = dmsave;
728caa4e7f2SJed Brown   PetscFunctionReturn(0);
729caa4e7f2SJed Brown }
730caa4e7f2SJed Brown 
7316cab3a1bSJed Brown /*@
7326cab3a1bSJed Brown    SNESSetUpMatrices - ensures that matrices are available for SNES, to be called by SNESSetUp_XXX()
7336cab3a1bSJed Brown 
7346cab3a1bSJed Brown    Collective
7356cab3a1bSJed Brown 
7366cab3a1bSJed Brown    Input Arguments:
7376cab3a1bSJed Brown .  snes - snes to configure
7386cab3a1bSJed Brown 
7396cab3a1bSJed Brown    Level: developer
7406cab3a1bSJed Brown 
7416cab3a1bSJed Brown .seealso: SNESSetUp()
7426cab3a1bSJed Brown @*/
7436cab3a1bSJed Brown PetscErrorCode SNESSetUpMatrices(SNES snes)
7446cab3a1bSJed Brown {
7456cab3a1bSJed Brown   PetscErrorCode ierr;
7466cab3a1bSJed Brown   DM             dm;
747942e3340SBarry Smith   DMSNES         sdm;
7486cab3a1bSJed Brown 
7496cab3a1bSJed Brown   PetscFunctionBegin;
7506cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
751942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
75258b371f3SBarry Smith   if (!snes->jacobian && snes->mf) {
7536cab3a1bSJed Brown     Mat  J;
7546cab3a1bSJed Brown     void *functx;
7556cab3a1bSJed Brown     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
7566cab3a1bSJed Brown     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
7576cab3a1bSJed Brown     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
7580298fd71SBarry Smith     ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr);
7599e5d0892SLisandro Dalcin     ierr = SNESSetJacobian(snes,J,J,NULL,NULL);CHKERRQ(ierr);
7606cab3a1bSJed Brown     ierr = MatDestroy(&J);CHKERRQ(ierr);
761caa4e7f2SJed Brown   } else if (snes->mf_operator && !snes->jacobian_pre && !snes->jacobian) {
7626cab3a1bSJed Brown     Mat J,B;
7636cab3a1bSJed Brown     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
7646cab3a1bSJed Brown     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
7656cab3a1bSJed Brown     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
766b412c318SBarry Smith     ierr = DMCreateMatrix(snes->dm,&B);CHKERRQ(ierr);
76706f20277SJed Brown     /* sdm->computejacobian was already set to reach here */
7680298fd71SBarry Smith     ierr = SNESSetJacobian(snes,J,B,NULL,NULL);CHKERRQ(ierr);
7696cab3a1bSJed Brown     ierr = MatDestroy(&J);CHKERRQ(ierr);
7706cab3a1bSJed Brown     ierr = MatDestroy(&B);CHKERRQ(ierr);
771caa4e7f2SJed Brown   } else if (!snes->jacobian_pre) {
7721ba9b98eSMatthew G. Knepley     PetscDS   prob;
7736cab3a1bSJed Brown     Mat       J, B;
7741ba9b98eSMatthew G. Knepley     PetscBool hasPrec   = PETSC_FALSE;
7751ba9b98eSMatthew G. Knepley 
7766cab3a1bSJed Brown     J    = snes->jacobian;
7771ba9b98eSMatthew G. Knepley     ierr = DMGetDS(dm, &prob);CHKERRQ(ierr);
7781ba9b98eSMatthew G. Knepley     if (prob) {ierr = PetscDSHasJacobianPreconditioner(prob, &hasPrec);CHKERRQ(ierr);}
779ec9a985fSMatthew G. Knepley     if (J)            {ierr = PetscObjectReference((PetscObject) J);CHKERRQ(ierr);}
780ec9a985fSMatthew G. Knepley     else if (hasPrec) {ierr = DMCreateMatrix(snes->dm, &J);CHKERRQ(ierr);}
781b412c318SBarry Smith     ierr = DMCreateMatrix(snes->dm, &B);CHKERRQ(ierr);
7820298fd71SBarry Smith     ierr = SNESSetJacobian(snes, J ? J : B, B, NULL, NULL);CHKERRQ(ierr);
7831ba9b98eSMatthew G. Knepley     ierr = MatDestroy(&J);CHKERRQ(ierr);
7846cab3a1bSJed Brown     ierr = MatDestroy(&B);CHKERRQ(ierr);
7856cab3a1bSJed Brown   }
786caa4e7f2SJed Brown   {
787caa4e7f2SJed Brown     KSP ksp;
788caa4e7f2SJed Brown     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
789caa4e7f2SJed Brown     ierr = KSPSetComputeOperators(ksp,KSPComputeOperators_SNES,snes);CHKERRQ(ierr);
79016ebb321SJed Brown     ierr = DMCoarsenHookAdd(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr);
791caa4e7f2SJed Brown   }
7926cab3a1bSJed Brown   PetscFunctionReturn(0);
7936cab3a1bSJed Brown }
7946cab3a1bSJed Brown 
795fde5950dSBarry Smith /*@C
796fde5950dSBarry Smith    SNESMonitorSetFromOptions - Sets a monitor function and viewer appropriate for the type indicated by the user
797fde5950dSBarry Smith 
798fde5950dSBarry Smith    Collective on SNES
799fde5950dSBarry Smith 
800fde5950dSBarry Smith    Input Parameters:
801fde5950dSBarry Smith +  snes - SNES object you wish to monitor
802fde5950dSBarry Smith .  name - the monitor type one is seeking
803fde5950dSBarry Smith .  help - message indicating what monitoring is done
804fde5950dSBarry Smith .  manual - manual page for the monitor
805fde5950dSBarry Smith .  monitor - the monitor function
806fde5950dSBarry Smith -  monitorsetup - a function that is called once ONLY if the user selected this monitor that may set additional features of the SNES or PetscViewer objects
807fde5950dSBarry Smith 
808fde5950dSBarry Smith    Level: developer
809fde5950dSBarry Smith 
810fde5950dSBarry Smith .seealso: PetscOptionsGetViewer(), PetscOptionsGetReal(), PetscOptionsHasName(), PetscOptionsGetString(),
811fde5950dSBarry Smith           PetscOptionsGetIntArray(), PetscOptionsGetRealArray(), PetscOptionsBool()
812fde5950dSBarry Smith           PetscOptionsInt(), PetscOptionsString(), PetscOptionsReal(), PetscOptionsBool(),
813fde5950dSBarry Smith           PetscOptionsName(), PetscOptionsBegin(), PetscOptionsEnd(), PetscOptionsHead(),
814fde5950dSBarry Smith           PetscOptionsStringArray(),PetscOptionsRealArray(), PetscOptionsScalar(),
815fde5950dSBarry Smith           PetscOptionsBoolGroupBegin(), PetscOptionsBoolGroup(), PetscOptionsBoolGroupEnd(),
816fde5950dSBarry Smith           PetscOptionsFList(), PetscOptionsEList()
817fde5950dSBarry Smith @*/
818d43b4f6eSBarry Smith PetscErrorCode  SNESMonitorSetFromOptions(SNES snes,const char name[],const char help[], const char manual[],PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,PetscViewerAndFormat*),PetscErrorCode (*monitorsetup)(SNES,PetscViewerAndFormat*))
819fde5950dSBarry Smith {
820fde5950dSBarry Smith   PetscErrorCode    ierr;
821fde5950dSBarry Smith   PetscViewer       viewer;
822fde5950dSBarry Smith   PetscViewerFormat format;
823fde5950dSBarry Smith   PetscBool         flg;
824fde5950dSBarry Smith 
825fde5950dSBarry Smith   PetscFunctionBegin;
82616413a6aSBarry Smith   ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,name,&viewer,&format,&flg);CHKERRQ(ierr);
827fde5950dSBarry Smith   if (flg) {
828d43b4f6eSBarry Smith     PetscViewerAndFormat *vf;
829d43b4f6eSBarry Smith     ierr = PetscViewerAndFormatCreate(viewer,format,&vf);CHKERRQ(ierr);
830d43b4f6eSBarry Smith     ierr = PetscObjectDereference((PetscObject)viewer);CHKERRQ(ierr);
831fde5950dSBarry Smith     if (monitorsetup) {
832d43b4f6eSBarry Smith       ierr = (*monitorsetup)(snes,vf);CHKERRQ(ierr);
833fde5950dSBarry Smith     }
834d43b4f6eSBarry Smith     ierr = SNESMonitorSet(snes,(PetscErrorCode (*)(SNES,PetscInt,PetscReal,void*))monitor,vf,(PetscErrorCode (*)(void**))PetscViewerAndFormatDestroy);CHKERRQ(ierr);
835fde5950dSBarry Smith   }
836fde5950dSBarry Smith   PetscFunctionReturn(0);
837fde5950dSBarry Smith }
838fde5950dSBarry Smith 
8399b94acceSBarry Smith /*@
84094b7f48cSBarry Smith    SNESSetFromOptions - Sets various SNES and KSP parameters from user options.
8419b94acceSBarry Smith 
842c7afd0dbSLois Curfman McInnes    Collective on SNES
843c7afd0dbSLois Curfman McInnes 
8449b94acceSBarry Smith    Input Parameter:
8459b94acceSBarry Smith .  snes - the SNES context
8469b94acceSBarry Smith 
84736851e7fSLois Curfman McInnes    Options Database Keys:
848722329fbSBarry Smith +  -snes_type <type> - newtonls, newtontr, ngmres, ncg, nrichardson, qn, vi, fas, SNESType for complete list
84982738288SBarry Smith .  -snes_stol - convergence tolerance in terms of the norm
85082738288SBarry Smith                 of the change in the solution between steps
85170441072SBarry Smith .  -snes_atol <abstol> - absolute tolerance of residual norm
852b39c3a46SLois Curfman McInnes .  -snes_rtol <rtol> - relative decrease in tolerance norm from initial
853e4d06f11SPatrick Farrell .  -snes_divergence_tolerance <divtol> - if the residual goes above divtol*rnorm0, exit with divergence
854be5caee7SBarry Smith .  -snes_force_iteration <force> - force SNESSolve() to take at least one iteration
855b39c3a46SLois Curfman McInnes .  -snes_max_it <max_it> - maximum number of iterations
856b39c3a46SLois Curfman McInnes .  -snes_max_funcs <max_funcs> - maximum number of function evaluations
8574839bfe8SBarry Smith .  -snes_max_fail <max_fail> - maximum number of line search failures allowed before stopping, default is none
858ddf469c8SBarry Smith .  -snes_max_linear_solve_fail - number of linear solver failures before SNESSolve() stops
859a8054027SBarry Smith .  -snes_lag_preconditioner <lag> - how often preconditioner is rebuilt (use -1 to never rebuild)
8603d5a8a6aSBarry Smith .  -snes_lag_preconditioner_persists <true,false> - retains the -snes_lag_preconditioner information across multiple SNESSolve()
861e35cf81dSBarry Smith .  -snes_lag_jacobian <lag> - how often Jacobian is rebuilt (use -1 to never rebuild)
8623d5a8a6aSBarry Smith .  -snes_lag_jacobian_persists <true,false> - retains the -snes_lag_jacobian information across multiple SNESSolve()
863b39c3a46SLois Curfman McInnes .  -snes_trtol <trtol> - trust region tolerance
8642492ecdbSBarry Smith .  -snes_no_convergence_test - skip convergence test in nonlinear
86582738288SBarry Smith                                solver; hence iterations will continue until max_it
8661fbbfb26SLois Curfman McInnes                                or some other criterion is reached. Saves expense
86782738288SBarry Smith                                of convergence test
868fde5950dSBarry Smith .  -snes_monitor [ascii][:filename][:viewer format] - prints residual norm at each iteration. if no filename given prints to stdout
869fde5950dSBarry Smith .  -snes_monitor_solution [ascii binary draw][:filename][:viewer format] - plots solution at each iteration
870fde5950dSBarry Smith .  -snes_monitor_residual [ascii binary draw][:filename][:viewer format] - plots residual (not its norm) at each iteration
871fde5950dSBarry Smith .  -snes_monitor_solution_update [ascii binary draw][:filename][:viewer format] - plots update to solution at each iteration
8724619e776SBarry Smith .  -snes_monitor_lg_residualnorm - plots residual norm at each iteration
873459f5d12SBarry Smith .  -snes_monitor_lg_range - plots residual norm at each iteration
874e24b481bSBarry Smith .  -snes_fd - use finite differences to compute Jacobian; very slow, only for testing
875e2e60de9SPeter Brune .  -snes_fd_color - use finite differences with coloring to compute Jacobian
8765968eb51SBarry Smith .  -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration
877b5badacbSBarry Smith .  -snes_converged_reason - print the reason for convergence/divergence after each solve
878e62ac41dSBarry Smith .  -npc_snes_type <type> - the SNES type to use as a nonlinear preconditioner
879e62ac41dSBarry Smith .   -snes_test_jacobian <optional threshold> - compare the user provided Jacobian with one computed via finite differences to check for errors.  If a threshold is given, display only those entries whose difference is greater than the threshold.
880e62ac41dSBarry Smith -   -snes_test_jacobian_view - display the user provided Jacobian, the finite difference Jacobian and the difference between them to help users detect the location of errors in the user provided Jacobian.
88182738288SBarry Smith 
88282738288SBarry Smith     Options Database for Eisenstat-Walker method:
883fa9f3622SBarry Smith +  -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence
8844b27c08aSLois Curfman McInnes .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
88536851e7fSLois Curfman McInnes .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
88636851e7fSLois Curfman McInnes .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
88736851e7fSLois Curfman McInnes .  -snes_ksp_ew_gamma <gamma> - Sets gamma
88836851e7fSLois Curfman McInnes .  -snes_ksp_ew_alpha <alpha> - Sets alpha
88936851e7fSLois Curfman McInnes .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
89036851e7fSLois Curfman McInnes -  -snes_ksp_ew_threshold <threshold> - Sets threshold
89182738288SBarry Smith 
89211ca99fdSLois Curfman McInnes    Notes:
893ec5066bdSBarry Smith    To see all options, run your program with the -help option or consult the users manual
894ec5066bdSBarry Smith 
895ec5066bdSBarry Smith    Notes:
896ec5066bdSBarry Smith       SNES supports three approaches for computing (approximate) Jacobians: user provided via SNESSetJacobian(), matrix free, and computing explictly with
897ec5066bdSBarry Smith       finite differences and coloring using MatFDColoring. It is also possible to use automatic differentiation and the MatFDColoring object.
89883e2fdc7SBarry Smith 
89936851e7fSLois Curfman McInnes    Level: beginner
90036851e7fSLois Curfman McInnes 
901ec5066bdSBarry Smith .seealso: SNESSetOptionsPrefix(), SNESResetFromOptions(), SNES, SNESCreate()
9029b94acceSBarry Smith @*/
9037087cfbeSBarry Smith PetscErrorCode  SNESSetFromOptions(SNES snes)
9049b94acceSBarry Smith {
9058afaa268SBarry Smith   PetscBool      flg,pcset,persist,set;
906d8f46077SPeter Brune   PetscInt       i,indx,lag,grids;
90704d7464bSBarry Smith   const char     *deft        = SNESNEWTONLS;
908649ef022SMatthew Knepley   const char     *convtests[] = {"default","skip","correct_pressure"};
90985385478SLisandro Dalcin   SNESKSPEW      *kctx        = NULL;
910e8105e01SRichard Katz   char           type[256], monfilename[PETSC_MAX_PATH_LEN];
91185385478SLisandro Dalcin   PetscErrorCode ierr;
912c40d0f55SPeter Brune   PCSide         pcside;
913a64e098fSPeter Brune   const char     *optionsprefix;
9149b94acceSBarry Smith 
9153a40ed3dSBarry Smith   PetscFunctionBegin;
9160700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
9170f51fdf8SToby Isaac   ierr = SNESRegisterAll();CHKERRQ(ierr);
9183194b578SJed Brown   ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr);
919639ff905SBarry Smith   if (((PetscObject)snes)->type_name) deft = ((PetscObject)snes)->type_name;
920a264d7a6SBarry Smith   ierr = PetscOptionsFList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr);
921d64ed03dSBarry Smith   if (flg) {
922186905e3SBarry Smith     ierr = SNESSetType(snes,type);CHKERRQ(ierr);
9237adad957SLisandro Dalcin   } else if (!((PetscObject)snes)->type_name) {
924186905e3SBarry Smith     ierr = SNESSetType(snes,deft);CHKERRQ(ierr);
925d64ed03dSBarry Smith   }
92694ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_stol","Stop if step length less than","SNESSetTolerances",snes->stol,&snes->stol,NULL);CHKERRQ(ierr);
92794ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_atol","Stop if function norm less than","SNESSetTolerances",snes->abstol,&snes->abstol,NULL);CHKERRQ(ierr);
928186905e3SBarry Smith 
92994ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less than","SNESSetTolerances",snes->rtol,&snes->rtol,NULL);CHKERRQ(ierr);
930e4d06f11SPatrick Farrell   ierr = PetscOptionsReal("-snes_divergence_tolerance","Stop if residual norm increases by this factor","SNESSetDivergenceTolerance",snes->divtol,&snes->divtol,NULL);CHKERRQ(ierr);
9310298fd71SBarry Smith   ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,NULL);CHKERRQ(ierr);
9320298fd71SBarry Smith   ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,NULL);CHKERRQ(ierr);
9330298fd71SBarry Smith   ierr = PetscOptionsInt("-snes_max_fail","Maximum nonlinear step failures","SNESSetMaxNonlinearStepFailures",snes->maxFailures,&snes->maxFailures,NULL);CHKERRQ(ierr);
9340298fd71SBarry Smith   ierr = PetscOptionsInt("-snes_max_linear_solve_fail","Maximum failures in linear solves allowed","SNESSetMaxLinearSolveFailures",snes->maxLinearSolveFailures,&snes->maxLinearSolveFailures,NULL);CHKERRQ(ierr);
9350298fd71SBarry Smith   ierr = PetscOptionsBool("-snes_error_if_not_converged","Generate error if solver does not converge","SNESSetErrorIfNotConverged",snes->errorifnotconverged,&snes->errorifnotconverged,NULL);CHKERRQ(ierr);
9360d6f27a8SBarry Smith   ierr = PetscOptionsBool("-snes_force_iteration","Force SNESSolve() to take at least one iteration","SNESSetForceIteration",snes->forceiteration,&snes->forceiteration,NULL);CHKERRQ(ierr);
937b351a90bSFande Kong   ierr = PetscOptionsBool("-snes_check_jacobian_domain_error","Check Jacobian domain error after Jacobian evaluation","SNESCheckJacobianDomainError",snes->checkjacdomainerror,&snes->checkjacdomainerror,NULL);CHKERRQ(ierr);
93885385478SLisandro Dalcin 
939a8054027SBarry Smith   ierr = PetscOptionsInt("-snes_lag_preconditioner","How often to rebuild preconditioner","SNESSetLagPreconditioner",snes->lagpreconditioner,&lag,&flg);CHKERRQ(ierr);
940a8054027SBarry Smith   if (flg) {
9413d5a8a6aSBarry Smith     if (lag == -1) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Cannot set the lag to -1 from the command line since the preconditioner must be built as least once, perhaps you mean -2");
942a8054027SBarry Smith     ierr = SNESSetLagPreconditioner(snes,lag);CHKERRQ(ierr);
943a8054027SBarry Smith   }
9449590daa0SBarry Smith   ierr = PetscOptionsBool("-snes_lag_preconditioner_persists","Preconditioner lagging through multiple SNES solves","SNESSetLagPreconditionerPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr);
94537ec4e1aSPeter Brune   if (flg) {
94637ec4e1aSPeter Brune     ierr = SNESSetLagPreconditionerPersists(snes,persist);CHKERRQ(ierr);
94737ec4e1aSPeter Brune   }
948e35cf81dSBarry Smith   ierr = PetscOptionsInt("-snes_lag_jacobian","How often to rebuild Jacobian","SNESSetLagJacobian",snes->lagjacobian,&lag,&flg);CHKERRQ(ierr);
949e35cf81dSBarry Smith   if (flg) {
9503d5a8a6aSBarry Smith     if (lag == -1) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Cannot set the lag to -1 from the command line since the Jacobian must be built as least once, perhaps you mean -2");
951e35cf81dSBarry Smith     ierr = SNESSetLagJacobian(snes,lag);CHKERRQ(ierr);
952e35cf81dSBarry Smith   }
9539590daa0SBarry Smith   ierr = PetscOptionsBool("-snes_lag_jacobian_persists","Jacobian lagging through multiple SNES solves","SNESSetLagJacobianPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr);
95437ec4e1aSPeter Brune   if (flg) {
95537ec4e1aSPeter Brune     ierr = SNESSetLagJacobianPersists(snes,persist);CHKERRQ(ierr);
95637ec4e1aSPeter Brune   }
95737ec4e1aSPeter Brune 
958efd51863SBarry Smith   ierr = PetscOptionsInt("-snes_grid_sequence","Use grid sequencing to generate initial guess","SNESSetGridSequence",snes->gridsequence,&grids,&flg);CHKERRQ(ierr);
959efd51863SBarry Smith   if (flg) {
960efd51863SBarry Smith     ierr = SNESSetGridSequence(snes,grids);CHKERRQ(ierr);
961efd51863SBarry Smith   }
962a8054027SBarry Smith 
963649ef022SMatthew Knepley   ierr = PetscOptionsEList("-snes_convergence_test","Convergence test","SNESSetConvergenceTest",convtests,sizeof(convtests)/sizeof(char*),"default",&indx,&flg);CHKERRQ(ierr);
96485385478SLisandro Dalcin   if (flg) {
96585385478SLisandro Dalcin     switch (indx) {
9668d359177SBarry Smith     case 0: ierr = SNESSetConvergenceTest(snes,SNESConvergedDefault,NULL,NULL);CHKERRQ(ierr); break;
967e2a6519dSDmitry Karpeev     case 1: ierr = SNESSetConvergenceTest(snes,SNESConvergedSkip,NULL,NULL);CHKERRQ(ierr); break;
968649ef022SMatthew Knepley     case 2: ierr = SNESSetConvergenceTest(snes,SNESConvergedCorrectPressure,NULL,NULL);CHKERRQ(ierr); break;
96985385478SLisandro Dalcin     }
97085385478SLisandro Dalcin   }
97185385478SLisandro Dalcin 
972365a6726SPeter Brune   ierr = PetscOptionsEList("-snes_norm_schedule","SNES Norm schedule","SNESSetNormSchedule",SNESNormSchedules,5,"function",&indx,&flg);CHKERRQ(ierr);
973365a6726SPeter Brune   if (flg) { ierr = SNESSetNormSchedule(snes,(SNESNormSchedule)indx);CHKERRQ(ierr); }
974fdacfa88SPeter Brune 
97547073ea2SPeter Brune   ierr = PetscOptionsEList("-snes_function_type","SNES Norm schedule","SNESSetFunctionType",SNESFunctionTypes,2,"unpreconditioned",&indx,&flg);CHKERRQ(ierr);
97647073ea2SPeter Brune   if (flg) { ierr = SNESSetFunctionType(snes,(SNESFunctionType)indx);CHKERRQ(ierr); }
977186905e3SBarry Smith 
97885385478SLisandro Dalcin   kctx = (SNESKSPEW*)snes->kspconvctx;
97985385478SLisandro Dalcin 
9800298fd71SBarry Smith   ierr = PetscOptionsBool("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,NULL);CHKERRQ(ierr);
981186905e3SBarry Smith 
98294ae4db5SBarry Smith   ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,NULL);CHKERRQ(ierr);
98394ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,NULL);CHKERRQ(ierr);
98494ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,NULL);CHKERRQ(ierr);
98594ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,NULL);CHKERRQ(ierr);
98694ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,NULL);CHKERRQ(ierr);
98794ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,NULL);CHKERRQ(ierr);
98894ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,NULL);CHKERRQ(ierr);
989186905e3SBarry Smith 
99090d69ab7SBarry Smith   flg  = PETSC_FALSE;
9918afaa268SBarry Smith   ierr = PetscOptionsBool("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",flg,&flg,&set);CHKERRQ(ierr);
9928afaa268SBarry Smith   if (set && flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);}
993eabae89aSBarry Smith 
994fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor","Monitor norm of function","SNESMonitorDefault",SNESMonitorDefault,NULL);CHKERRQ(ierr);
995fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_short","Monitor norm of function with fewer digits","SNESMonitorDefaultShort",SNESMonitorDefaultShort,NULL);CHKERRQ(ierr);
996fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_range","Monitor range of elements of function","SNESMonitorRange",SNESMonitorRange,NULL);CHKERRQ(ierr);
997eabae89aSBarry Smith 
998fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_ratio","Monitor ratios of the norm of function for consecutive steps","SNESMonitorRatio",SNESMonitorRatio,SNESMonitorRatioSetUp);CHKERRQ(ierr);
999fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_field","Monitor norm of function (split into fields)","SNESMonitorDefaultField",SNESMonitorDefaultField,NULL);CHKERRQ(ierr);
1000fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution","View solution at each iteration","SNESMonitorSolution",SNESMonitorSolution,NULL);CHKERRQ(ierr);
1001fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution_update","View correction at each iteration","SNESMonitorSolutionUpdate",SNESMonitorSolutionUpdate,NULL);CHKERRQ(ierr);
1002fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_residual","View residual at each iteration","SNESMonitorResidual",SNESMonitorResidual,NULL);CHKERRQ(ierr);
1003fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_jacupdate_spectrum","Print the change in the spectrum of the Jacobian","SNESMonitorJacUpdateSpectrum",SNESMonitorJacUpdateSpectrum,NULL);CHKERRQ(ierr);
1004fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_fields","Monitor norm of function per field","SNESMonitorSet",SNESMonitorFields,NULL);CHKERRQ(ierr);
10052db13446SMatthew G. Knepley 
1006589a23caSBarry Smith   ierr = PetscOptionsString("-snes_monitor_python","Use Python function","SNESMonitorSet",NULL,monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr);
10075180491cSLisandro Dalcin   if (flg) {ierr = PetscPythonMonitorSet((PetscObject)snes,monfilename);CHKERRQ(ierr);}
10085180491cSLisandro Dalcin 
100990d69ab7SBarry Smith   flg  = PETSC_FALSE;
10100298fd71SBarry Smith   ierr = PetscOptionsBool("-snes_monitor_lg_range","Plot function range at each iteration","SNESMonitorLGRange",flg,&flg,NULL);CHKERRQ(ierr);
1011459f5d12SBarry Smith   if (flg) {
1012459f5d12SBarry Smith     PetscViewer ctx;
1013e24b481bSBarry Smith 
10146ba87a44SLisandro Dalcin     ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr);
1015459f5d12SBarry Smith     ierr = SNESMonitorSet(snes,SNESMonitorLGRange,ctx,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
1016459f5d12SBarry Smith   }
10172e7541e6SPeter Brune 
101890d69ab7SBarry Smith   flg  = PETSC_FALSE;
1019c4421ceaSFande Kong   ierr = PetscOptionsBool("-snes_converged_reason_view_cancel","Remove all converged reason viewers","SNESConvergedReasonViewCancel",flg,&flg,&set);CHKERRQ(ierr);
1020c4421ceaSFande Kong   if (set && flg) {ierr = SNESConvergedReasonViewCancel(snes);CHKERRQ(ierr);}
1021c4421ceaSFande Kong 
1022c4421ceaSFande Kong   flg  = PETSC_FALSE;
10238d359177SBarry Smith   ierr = PetscOptionsBool("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESComputeJacobianDefault",flg,&flg,NULL);CHKERRQ(ierr);
10244b27c08aSLois Curfman McInnes   if (flg) {
10256cab3a1bSJed Brown     void    *functx;
1026b1f624c7SBarry Smith     DM      dm;
1027b1f624c7SBarry Smith     DMSNES  sdm;
1028b1f624c7SBarry Smith     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
1029b1f624c7SBarry Smith     ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
1030b1f624c7SBarry Smith     sdm->jacobianctx = NULL;
10310298fd71SBarry Smith     ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr);
10328d359177SBarry Smith     ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefault,functx);CHKERRQ(ierr);
1033ae15b995SBarry Smith     ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr);
10349b94acceSBarry Smith   }
1035639f9d9dSBarry Smith 
103644848bc4SPeter Brune   flg  = PETSC_FALSE;
10378d359177SBarry Smith   ierr = PetscOptionsBool("-snes_fd_function","Use finite differences (slow) to compute function from user objective","SNESObjectiveComputeFunctionDefaultFD",flg,&flg,NULL);CHKERRQ(ierr);
103897584545SPeter Brune   if (flg) {
10398d359177SBarry Smith     ierr = SNESSetFunction(snes,NULL,SNESObjectiveComputeFunctionDefaultFD,NULL);CHKERRQ(ierr);
104097584545SPeter Brune   }
104197584545SPeter Brune 
104297584545SPeter Brune   flg  = PETSC_FALSE;
10438d359177SBarry Smith   ierr = PetscOptionsBool("-snes_fd_color","Use finite differences with coloring to compute Jacobian","SNESComputeJacobianDefaultColor",flg,&flg,NULL);CHKERRQ(ierr);
104444848bc4SPeter Brune   if (flg) {
1045c52e227fSPeter Brune     DM             dm;
1046c52e227fSPeter Brune     DMSNES         sdm;
1047c52e227fSPeter Brune     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
1048aace71b7SPeter Brune     ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
1049aace71b7SPeter Brune     sdm->jacobianctx = NULL;
10509e5d0892SLisandro Dalcin     ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr);
105144848bc4SPeter Brune     ierr = PetscInfo(snes,"Setting default finite difference coloring Jacobian matrix\n");CHKERRQ(ierr);
105244848bc4SPeter Brune   }
105344848bc4SPeter Brune 
1054aa3661deSLisandro Dalcin   flg  = PETSC_FALSE;
1055f871313dSBarry Smith   ierr = PetscOptionsBool("-snes_mf_operator","Use a Matrix-Free Jacobian with user-provided preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf_operator,&flg);CHKERRQ(ierr);
1056d8f46077SPeter Brune   if (flg && snes->mf_operator) {
1057a8248277SBarry Smith     snes->mf_operator = PETSC_TRUE;
1058d8f46077SPeter Brune     snes->mf          = PETSC_TRUE;
1059a8248277SBarry Smith   }
1060aa3661deSLisandro Dalcin   flg  = PETSC_FALSE;
1061f871313dSBarry Smith   ierr = PetscOptionsBool("-snes_mf","Use a Matrix-Free Jacobian with no preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf,&flg);CHKERRQ(ierr);
1062d8f46077SPeter Brune   if (!flg && snes->mf_operator) snes->mf = PETSC_TRUE;
10639e5d0892SLisandro Dalcin   ierr = PetscOptionsInt("-snes_mf_version","Matrix-Free routines version 1 or 2","None",snes->mf_version,&snes->mf_version,NULL);CHKERRQ(ierr);
1064d28543b3SPeter Brune 
1065c40d0f55SPeter Brune   flg  = PETSC_FALSE;
1066be95d8f1SBarry Smith   ierr = SNESGetNPCSide(snes,&pcside);CHKERRQ(ierr);
1067be95d8f1SBarry Smith   ierr = PetscOptionsEnum("-snes_npc_side","SNES nonlinear preconditioner side","SNESSetNPCSide",PCSides,(PetscEnum)pcside,(PetscEnum*)&pcside,&flg);CHKERRQ(ierr);
1068be95d8f1SBarry Smith   if (flg) {ierr = SNESSetNPCSide(snes,pcside);CHKERRQ(ierr);}
1069c40d0f55SPeter Brune 
1070e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
10718a70d858SHong Zhang   /*
10728a70d858SHong Zhang     Publish convergence information using SAWs
10738a70d858SHong Zhang   */
10748a70d858SHong Zhang   flg  = PETSC_FALSE;
10758a70d858SHong Zhang   ierr = PetscOptionsBool("-snes_monitor_saws","Publish SNES progress using SAWs","SNESMonitorSet",flg,&flg,NULL);CHKERRQ(ierr);
10768a70d858SHong Zhang   if (flg) {
10778a70d858SHong Zhang     void *ctx;
10788a70d858SHong Zhang     ierr = SNESMonitorSAWsCreate(snes,&ctx);CHKERRQ(ierr);
10798a70d858SHong Zhang     ierr = SNESMonitorSet(snes,SNESMonitorSAWs,ctx,SNESMonitorSAWsDestroy);CHKERRQ(ierr);
10808a70d858SHong Zhang   }
10818a70d858SHong Zhang #endif
10828a70d858SHong Zhang #if defined(PETSC_HAVE_SAWS)
1083b90c6cbeSBarry Smith   {
1084b90c6cbeSBarry Smith   PetscBool set;
1085b90c6cbeSBarry Smith   flg  = PETSC_FALSE;
10868a70d858SHong Zhang   ierr = PetscOptionsBool("-snes_saws_block","Block for SAWs at end of SNESSolve","PetscObjectSAWsBlock",((PetscObject)snes)->amspublishblock,&flg,&set);CHKERRQ(ierr);
1087b90c6cbeSBarry Smith   if (set) {
1088e04113cfSBarry Smith     ierr = PetscObjectSAWsSetBlock((PetscObject)snes,flg);CHKERRQ(ierr);
1089b90c6cbeSBarry Smith   }
1090b90c6cbeSBarry Smith   }
1091b90c6cbeSBarry Smith #endif
1092b90c6cbeSBarry Smith 
109376b2cf59SMatthew Knepley   for (i = 0; i < numberofsetfromoptions; i++) {
109476b2cf59SMatthew Knepley     ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr);
109576b2cf59SMatthew Knepley   }
109676b2cf59SMatthew Knepley 
1097e7788613SBarry Smith   if (snes->ops->setfromoptions) {
1098e55864a3SBarry Smith     ierr = (*snes->ops->setfromoptions)(PetscOptionsObject,snes);CHKERRQ(ierr);
1099639f9d9dSBarry Smith   }
11005d973c19SBarry Smith 
11015d973c19SBarry Smith   /* process any options handlers added with PetscObjectAddOptionsHandler() */
11020633abcbSJed Brown   ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject)snes);CHKERRQ(ierr);
1103b0a32e0cSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
11044bbc92c1SBarry Smith 
1105d8d34be6SBarry Smith   if (snes->linesearch) {
11067601faf0SJed Brown     ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr);
1107f1c6b773SPeter Brune     ierr = SNESLineSearchSetFromOptions(snes->linesearch);CHKERRQ(ierr);
1108d8d34be6SBarry Smith   }
11099e764e56SPeter Brune 
11106aa5e7e9SBarry Smith   if (snes->usesksp) {
11116991f827SBarry Smith     if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
11126991f827SBarry Smith     ierr = KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre);CHKERRQ(ierr);
11136991f827SBarry Smith     ierr = KSPSetFromOptions(snes->ksp);CHKERRQ(ierr);
11146aa5e7e9SBarry Smith   }
11156991f827SBarry Smith 
1116b5badacbSBarry Smith   /* if user has set the SNES NPC type via options database, create it. */
111751e86f29SPeter Brune   ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr);
1118c5929fdfSBarry Smith   ierr = PetscOptionsHasName(((PetscObject)snes)->options,optionsprefix, "-npc_snes_type", &pcset);CHKERRQ(ierr);
1119efd4aadfSBarry Smith   if (pcset && (!snes->npc)) {
1120efd4aadfSBarry Smith     ierr = SNESGetNPC(snes, &snes->npc);CHKERRQ(ierr);
112151e86f29SPeter Brune   }
1122b5badacbSBarry Smith   if (snes->npc) {
1123b5badacbSBarry Smith     ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr);
1124b5badacbSBarry Smith   }
1125b3cd9a81SMatthew G. Knepley   snes->setfromoptionscalled++;
1126b3cd9a81SMatthew G. Knepley   PetscFunctionReturn(0);
1127b3cd9a81SMatthew G. Knepley }
1128b3cd9a81SMatthew G. Knepley 
1129b3cd9a81SMatthew G. Knepley /*@
1130b3cd9a81SMatthew G. Knepley    SNESResetFromOptions - Sets various SNES and KSP parameters from user options ONLY if the SNES was previously set from options
1131b3cd9a81SMatthew G. Knepley 
1132b3cd9a81SMatthew G. Knepley    Collective on SNES
1133b3cd9a81SMatthew G. Knepley 
1134b3cd9a81SMatthew G. Knepley    Input Parameter:
1135b3cd9a81SMatthew G. Knepley .  snes - the SNES context
1136b3cd9a81SMatthew G. Knepley 
1137b3cd9a81SMatthew G. Knepley    Level: beginner
1138b3cd9a81SMatthew G. Knepley 
1139b3cd9a81SMatthew G. Knepley .seealso: SNESSetFromOptions(), SNESSetOptionsPrefix()
1140b3cd9a81SMatthew G. Knepley @*/
1141b3cd9a81SMatthew G. Knepley PetscErrorCode SNESResetFromOptions(SNES snes)
1142b3cd9a81SMatthew G. Knepley {
1143b3cd9a81SMatthew G. Knepley   PetscErrorCode ierr;
1144b3cd9a81SMatthew G. Knepley 
1145b3cd9a81SMatthew G. Knepley   PetscFunctionBegin;
1146b3cd9a81SMatthew G. Knepley   if (snes->setfromoptionscalled) {ierr = SNESSetFromOptions(snes);CHKERRQ(ierr);}
11473a40ed3dSBarry Smith   PetscFunctionReturn(0);
11489b94acceSBarry Smith }
11499b94acceSBarry Smith 
1150bb9467b5SJed Brown /*@C
1151d25893d9SBarry Smith    SNESSetComputeApplicationContext - Sets an optional function to compute a user-defined context for
1152d25893d9SBarry Smith    the nonlinear solvers.
1153d25893d9SBarry Smith 
1154d25893d9SBarry Smith    Logically Collective on SNES
1155d25893d9SBarry Smith 
1156d25893d9SBarry Smith    Input Parameters:
1157d25893d9SBarry Smith +  snes - the SNES context
1158d25893d9SBarry Smith .  compute - function to compute the context
1159d25893d9SBarry Smith -  destroy - function to destroy the context
1160d25893d9SBarry Smith 
1161d25893d9SBarry Smith    Level: intermediate
1162d25893d9SBarry Smith 
1163bb9467b5SJed Brown    Notes:
1164bb9467b5SJed Brown    This function is currently not available from Fortran.
1165bb9467b5SJed Brown 
1166d25893d9SBarry Smith .seealso: SNESGetApplicationContext(), SNESSetComputeApplicationContext(), SNESGetApplicationContext()
1167d25893d9SBarry Smith @*/
1168d25893d9SBarry Smith PetscErrorCode  SNESSetComputeApplicationContext(SNES snes,PetscErrorCode (*compute)(SNES,void**),PetscErrorCode (*destroy)(void**))
1169d25893d9SBarry Smith {
1170d25893d9SBarry Smith   PetscFunctionBegin;
1171d25893d9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1172d25893d9SBarry Smith   snes->ops->usercompute = compute;
1173d25893d9SBarry Smith   snes->ops->userdestroy = destroy;
1174d25893d9SBarry Smith   PetscFunctionReturn(0);
1175d25893d9SBarry Smith }
1176a847f771SSatish Balay 
1177b07ff414SBarry Smith /*@
11789b94acceSBarry Smith    SNESSetApplicationContext - Sets the optional user-defined context for
11799b94acceSBarry Smith    the nonlinear solvers.
11809b94acceSBarry Smith 
11813f9fe445SBarry Smith    Logically Collective on SNES
1182fee21e36SBarry Smith 
1183c7afd0dbSLois Curfman McInnes    Input Parameters:
1184c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1185c7afd0dbSLois Curfman McInnes -  usrP - optional user context
1186c7afd0dbSLois Curfman McInnes 
118736851e7fSLois Curfman McInnes    Level: intermediate
118836851e7fSLois Curfman McInnes 
118995452b02SPatrick Sanan    Fortran Notes:
119095452b02SPatrick Sanan     To use this from Fortran you must write a Fortran interface definition for this
1191daf670e6SBarry Smith     function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument.
1192daf670e6SBarry Smith 
1193ecaffddaSVictor Eijkhout .seealso: SNESGetApplicationContext()
11949b94acceSBarry Smith @*/
11957087cfbeSBarry Smith PetscErrorCode  SNESSetApplicationContext(SNES snes,void *usrP)
11969b94acceSBarry Smith {
11971b2093e4SBarry Smith   PetscErrorCode ierr;
1198b07ff414SBarry Smith   KSP            ksp;
11991b2093e4SBarry Smith 
12003a40ed3dSBarry Smith   PetscFunctionBegin;
12010700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1202b07ff414SBarry Smith   ierr       = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
1203b07ff414SBarry Smith   ierr       = KSPSetApplicationContext(ksp,usrP);CHKERRQ(ierr);
12049b94acceSBarry Smith   snes->user = usrP;
12053a40ed3dSBarry Smith   PetscFunctionReturn(0);
12069b94acceSBarry Smith }
120774679c65SBarry Smith 
1208b07ff414SBarry Smith /*@
12099b94acceSBarry Smith    SNESGetApplicationContext - Gets the user-defined context for the
12109b94acceSBarry Smith    nonlinear solvers.
12119b94acceSBarry Smith 
1212c7afd0dbSLois Curfman McInnes    Not Collective
1213c7afd0dbSLois Curfman McInnes 
12149b94acceSBarry Smith    Input Parameter:
12159b94acceSBarry Smith .  snes - SNES context
12169b94acceSBarry Smith 
12179b94acceSBarry Smith    Output Parameter:
12189b94acceSBarry Smith .  usrP - user context
12199b94acceSBarry Smith 
122095452b02SPatrick Sanan    Fortran Notes:
122195452b02SPatrick Sanan     To use this from Fortran you must write a Fortran interface definition for this
1222daf670e6SBarry Smith     function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument.
1223daf670e6SBarry Smith 
122436851e7fSLois Curfman McInnes    Level: intermediate
122536851e7fSLois Curfman McInnes 
12269b94acceSBarry Smith .seealso: SNESSetApplicationContext()
12279b94acceSBarry Smith @*/
1228e71120c6SJed Brown PetscErrorCode  SNESGetApplicationContext(SNES snes,void *usrP)
12299b94acceSBarry Smith {
12303a40ed3dSBarry Smith   PetscFunctionBegin;
12310700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1232e71120c6SJed Brown   *(void**)usrP = snes->user;
12333a40ed3dSBarry Smith   PetscFunctionReturn(0);
12349b94acceSBarry Smith }
123574679c65SBarry Smith 
12369b94acceSBarry Smith /*@
1237ec5066bdSBarry Smith    SNESSetUseMatrixFree - indicates that SNES should use matrix free finite difference matrix vector products internally to apply the Jacobian.
12383565c898SBarry Smith 
12393565c898SBarry Smith    Collective on SNES
12403565c898SBarry Smith 
12413565c898SBarry Smith    Input Parameters:
12423565c898SBarry Smith +  snes - SNES context
12434ddffce6SLisandro Dalcin .  mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used
12444ddffce6SLisandro Dalcin -  mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored
12453565c898SBarry Smith 
12463565c898SBarry Smith    Options Database:
12473565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator
1248ec5066bdSBarry Smith . -snes_mf_operator - use matrix free only for the mat operator
1249ec5066bdSBarry Smith . -snes_fd_color - compute the Jacobian via coloring and finite differences.
1250ec5066bdSBarry Smith - -snes_fd - compute the Jacobian via finite differences (slow)
12513565c898SBarry Smith 
12523565c898SBarry Smith    Level: intermediate
12533565c898SBarry Smith 
1254ec5066bdSBarry Smith    Notes:
1255ec5066bdSBarry Smith       SNES supports three approaches for computing (approximate) Jacobians: user provided via SNESSetJacobian(), matrix free, and computing explictly with
1256ec5066bdSBarry Smith       finite differences and coloring using MatFDColoring. It is also possible to use automatic differentiation and the MatFDColoring object.
1257ec5066bdSBarry Smith 
1258ec5066bdSBarry Smith .seealso:   SNESGetUseMatrixFree(), MatCreateSNESMF(), SNESComputeJacobianDefaultColor()
12593565c898SBarry Smith @*/
12603565c898SBarry Smith PetscErrorCode  SNESSetUseMatrixFree(SNES snes,PetscBool mf_operator,PetscBool mf)
12613565c898SBarry Smith {
12623565c898SBarry Smith   PetscFunctionBegin;
12633565c898SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
126488b4c220SStefano Zampini   PetscValidLogicalCollectiveBool(snes,mf_operator,2);
126588b4c220SStefano Zampini   PetscValidLogicalCollectiveBool(snes,mf,3);
12664ddffce6SLisandro Dalcin   snes->mf          = mf_operator ? PETSC_TRUE : mf;
12673565c898SBarry Smith   snes->mf_operator = mf_operator;
12683565c898SBarry Smith   PetscFunctionReturn(0);
12693565c898SBarry Smith }
12703565c898SBarry Smith 
12713565c898SBarry Smith /*@
1272ec5066bdSBarry Smith    SNESGetUseMatrixFree - indicates if the SNES uses matrix free finite difference matrix vector products to apply the Jacobian.
12733565c898SBarry Smith 
12743565c898SBarry Smith    Collective on SNES
12753565c898SBarry Smith 
12763565c898SBarry Smith    Input Parameter:
12773565c898SBarry Smith .  snes - SNES context
12783565c898SBarry Smith 
12793565c898SBarry Smith    Output Parameters:
12804ddffce6SLisandro Dalcin +  mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used
12814ddffce6SLisandro Dalcin -  mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored
12823565c898SBarry Smith 
12833565c898SBarry Smith    Options Database:
12843565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator
12853565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator
12863565c898SBarry Smith 
12873565c898SBarry Smith    Level: intermediate
12883565c898SBarry Smith 
12893565c898SBarry Smith .seealso:   SNESSetUseMatrixFree(), MatCreateSNESMF()
12903565c898SBarry Smith @*/
12913565c898SBarry Smith PetscErrorCode  SNESGetUseMatrixFree(SNES snes,PetscBool *mf_operator,PetscBool *mf)
12923565c898SBarry Smith {
12933565c898SBarry Smith   PetscFunctionBegin;
12943565c898SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
12953565c898SBarry Smith   if (mf)          *mf          = snes->mf;
12963565c898SBarry Smith   if (mf_operator) *mf_operator = snes->mf_operator;
12973565c898SBarry Smith   PetscFunctionReturn(0);
12983565c898SBarry Smith }
12993565c898SBarry Smith 
13003565c898SBarry Smith /*@
1301c8228a4eSBarry Smith    SNESGetIterationNumber - Gets the number of nonlinear iterations completed
1302c8228a4eSBarry Smith    at this time.
13039b94acceSBarry Smith 
1304c7afd0dbSLois Curfman McInnes    Not Collective
1305c7afd0dbSLois Curfman McInnes 
13069b94acceSBarry Smith    Input Parameter:
13079b94acceSBarry Smith .  snes - SNES context
13089b94acceSBarry Smith 
13099b94acceSBarry Smith    Output Parameter:
13109b94acceSBarry Smith .  iter - iteration number
13119b94acceSBarry Smith 
1312c8228a4eSBarry Smith    Notes:
1313c8228a4eSBarry Smith    For example, during the computation of iteration 2 this would return 1.
1314c8228a4eSBarry Smith 
1315c8228a4eSBarry Smith    This is useful for using lagged Jacobians (where one does not recompute the
131608405cd6SLois Curfman McInnes    Jacobian at each SNES iteration). For example, the code
131708405cd6SLois Curfman McInnes .vb
131808405cd6SLois Curfman McInnes       ierr = SNESGetIterationNumber(snes,&it);
131908405cd6SLois Curfman McInnes       if (!(it % 2)) {
132008405cd6SLois Curfman McInnes         [compute Jacobian here]
132108405cd6SLois Curfman McInnes       }
132208405cd6SLois Curfman McInnes .ve
1323c8228a4eSBarry Smith    can be used in your ComputeJacobian() function to cause the Jacobian to be
132408405cd6SLois Curfman McInnes    recomputed every second SNES iteration.
1325c8228a4eSBarry Smith 
1326c04deec6SBarry Smith    After the SNES solve is complete this will return the number of nonlinear iterations used.
1327c04deec6SBarry Smith 
132836851e7fSLois Curfman McInnes    Level: intermediate
132936851e7fSLois Curfman McInnes 
133071dbe336SPeter Brune .seealso:   SNESGetLinearSolveIterations()
13319b94acceSBarry Smith @*/
13327087cfbeSBarry Smith PetscErrorCode  SNESGetIterationNumber(SNES snes,PetscInt *iter)
13339b94acceSBarry Smith {
13343a40ed3dSBarry Smith   PetscFunctionBegin;
13350700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
13364482741eSBarry Smith   PetscValidIntPointer(iter,2);
13379b94acceSBarry Smith   *iter = snes->iter;
13383a40ed3dSBarry Smith   PetscFunctionReturn(0);
13399b94acceSBarry Smith }
134074679c65SBarry Smith 
1341360c497dSPeter Brune /*@
1342360c497dSPeter Brune    SNESSetIterationNumber - Sets the current iteration number.
1343360c497dSPeter Brune 
1344360c497dSPeter Brune    Not Collective
1345360c497dSPeter Brune 
1346360c497dSPeter Brune    Input Parameter:
1347a2b725a8SWilliam Gropp +  snes - SNES context
1348a2b725a8SWilliam Gropp -  iter - iteration number
1349360c497dSPeter Brune 
1350360c497dSPeter Brune    Level: developer
1351360c497dSPeter Brune 
135271dbe336SPeter Brune .seealso:   SNESGetLinearSolveIterations()
1353360c497dSPeter Brune @*/
1354360c497dSPeter Brune PetscErrorCode  SNESSetIterationNumber(SNES snes,PetscInt iter)
1355360c497dSPeter Brune {
1356360c497dSPeter Brune   PetscErrorCode ierr;
1357360c497dSPeter Brune 
1358360c497dSPeter Brune   PetscFunctionBegin;
1359360c497dSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1360e04113cfSBarry Smith   ierr       = PetscObjectSAWsTakeAccess((PetscObject)snes);CHKERRQ(ierr);
1361360c497dSPeter Brune   snes->iter = iter;
1362e04113cfSBarry Smith   ierr       = PetscObjectSAWsGrantAccess((PetscObject)snes);CHKERRQ(ierr);
1363360c497dSPeter Brune   PetscFunctionReturn(0);
1364360c497dSPeter Brune }
1365360c497dSPeter Brune 
13669b94acceSBarry Smith /*@
1367b850b91aSLisandro Dalcin    SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps
13689b94acceSBarry Smith    attempted by the nonlinear solver.
13699b94acceSBarry Smith 
1370c7afd0dbSLois Curfman McInnes    Not Collective
1371c7afd0dbSLois Curfman McInnes 
13729b94acceSBarry Smith    Input Parameter:
13739b94acceSBarry Smith .  snes - SNES context
13749b94acceSBarry Smith 
13759b94acceSBarry Smith    Output Parameter:
13769b94acceSBarry Smith .  nfails - number of unsuccessful steps attempted
13779b94acceSBarry Smith 
1378c96a6f78SLois Curfman McInnes    Notes:
1379c96a6f78SLois Curfman McInnes    This counter is reset to zero for each successive call to SNESSolve().
1380c96a6f78SLois Curfman McInnes 
138136851e7fSLois Curfman McInnes    Level: intermediate
138236851e7fSLois Curfman McInnes 
1383e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
138458ebbce7SBarry Smith           SNESSetMaxNonlinearStepFailures(), SNESGetMaxNonlinearStepFailures()
13859b94acceSBarry Smith @*/
13867087cfbeSBarry Smith PetscErrorCode  SNESGetNonlinearStepFailures(SNES snes,PetscInt *nfails)
13879b94acceSBarry Smith {
13883a40ed3dSBarry Smith   PetscFunctionBegin;
13890700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
13904482741eSBarry Smith   PetscValidIntPointer(nfails,2);
139150ffb88aSMatthew Knepley   *nfails = snes->numFailures;
139250ffb88aSMatthew Knepley   PetscFunctionReturn(0);
139350ffb88aSMatthew Knepley }
139450ffb88aSMatthew Knepley 
139550ffb88aSMatthew Knepley /*@
1396b850b91aSLisandro Dalcin    SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps
139750ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
139850ffb88aSMatthew Knepley 
139950ffb88aSMatthew Knepley    Not Collective
140050ffb88aSMatthew Knepley 
140150ffb88aSMatthew Knepley    Input Parameters:
140250ffb88aSMatthew Knepley +  snes     - SNES context
140350ffb88aSMatthew Knepley -  maxFails - maximum of unsuccessful steps
140450ffb88aSMatthew Knepley 
140550ffb88aSMatthew Knepley    Level: intermediate
140650ffb88aSMatthew Knepley 
1407e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
140858ebbce7SBarry Smith           SNESGetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures()
140950ffb88aSMatthew Knepley @*/
14107087cfbeSBarry Smith PetscErrorCode  SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails)
141150ffb88aSMatthew Knepley {
141250ffb88aSMatthew Knepley   PetscFunctionBegin;
14130700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
141450ffb88aSMatthew Knepley   snes->maxFailures = maxFails;
141550ffb88aSMatthew Knepley   PetscFunctionReturn(0);
141650ffb88aSMatthew Knepley }
141750ffb88aSMatthew Knepley 
141850ffb88aSMatthew Knepley /*@
1419b850b91aSLisandro Dalcin    SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps
142050ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
142150ffb88aSMatthew Knepley 
142250ffb88aSMatthew Knepley    Not Collective
142350ffb88aSMatthew Knepley 
142450ffb88aSMatthew Knepley    Input Parameter:
142550ffb88aSMatthew Knepley .  snes     - SNES context
142650ffb88aSMatthew Knepley 
142750ffb88aSMatthew Knepley    Output Parameter:
142850ffb88aSMatthew Knepley .  maxFails - maximum of unsuccessful steps
142950ffb88aSMatthew Knepley 
143050ffb88aSMatthew Knepley    Level: intermediate
143150ffb88aSMatthew Knepley 
1432e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
143358ebbce7SBarry Smith           SNESSetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures()
143458ebbce7SBarry Smith 
143550ffb88aSMatthew Knepley @*/
14367087cfbeSBarry Smith PetscErrorCode  SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails)
143750ffb88aSMatthew Knepley {
143850ffb88aSMatthew Knepley   PetscFunctionBegin;
14390700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
14404482741eSBarry Smith   PetscValidIntPointer(maxFails,2);
144150ffb88aSMatthew Knepley   *maxFails = snes->maxFailures;
14423a40ed3dSBarry Smith   PetscFunctionReturn(0);
14439b94acceSBarry Smith }
1444a847f771SSatish Balay 
14452541af92SBarry Smith /*@
14462541af92SBarry Smith    SNESGetNumberFunctionEvals - Gets the number of user provided function evaluations
14472541af92SBarry Smith      done by SNES.
14482541af92SBarry Smith 
14492541af92SBarry Smith    Not Collective
14502541af92SBarry Smith 
14512541af92SBarry Smith    Input Parameter:
14522541af92SBarry Smith .  snes     - SNES context
14532541af92SBarry Smith 
14542541af92SBarry Smith    Output Parameter:
14552541af92SBarry Smith .  nfuncs - number of evaluations
14562541af92SBarry Smith 
14572541af92SBarry Smith    Level: intermediate
14582541af92SBarry Smith 
145995452b02SPatrick Sanan    Notes:
146095452b02SPatrick Sanan     Reset every time SNESSolve is called unless SNESSetCountersReset() is used.
1461971e163fSPeter Brune 
1462971e163fSPeter Brune .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), SNESSetCountersReset()
14632541af92SBarry Smith @*/
14647087cfbeSBarry Smith PetscErrorCode  SNESGetNumberFunctionEvals(SNES snes, PetscInt *nfuncs)
14652541af92SBarry Smith {
14662541af92SBarry Smith   PetscFunctionBegin;
14670700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
14682541af92SBarry Smith   PetscValidIntPointer(nfuncs,2);
14692541af92SBarry Smith   *nfuncs = snes->nfuncs;
14702541af92SBarry Smith   PetscFunctionReturn(0);
14712541af92SBarry Smith }
14722541af92SBarry Smith 
14733d4c4710SBarry Smith /*@
14743d4c4710SBarry Smith    SNESGetLinearSolveFailures - Gets the number of failed (non-converged)
14753d4c4710SBarry Smith    linear solvers.
14763d4c4710SBarry Smith 
14773d4c4710SBarry Smith    Not Collective
14783d4c4710SBarry Smith 
14793d4c4710SBarry Smith    Input Parameter:
14803d4c4710SBarry Smith .  snes - SNES context
14813d4c4710SBarry Smith 
14823d4c4710SBarry Smith    Output Parameter:
14833d4c4710SBarry Smith .  nfails - number of failed solves
14843d4c4710SBarry Smith 
14859d85da0cSMatthew G. Knepley    Level: intermediate
14869d85da0cSMatthew G. Knepley 
14879d85da0cSMatthew G. Knepley    Options Database Keys:
14889d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated
14899d85da0cSMatthew G. Knepley 
14903d4c4710SBarry Smith    Notes:
14913d4c4710SBarry Smith    This counter is reset to zero for each successive call to SNESSolve().
14923d4c4710SBarry Smith 
1493e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures()
14943d4c4710SBarry Smith @*/
14957087cfbeSBarry Smith PetscErrorCode  SNESGetLinearSolveFailures(SNES snes,PetscInt *nfails)
14963d4c4710SBarry Smith {
14973d4c4710SBarry Smith   PetscFunctionBegin;
14980700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
14993d4c4710SBarry Smith   PetscValidIntPointer(nfails,2);
15003d4c4710SBarry Smith   *nfails = snes->numLinearSolveFailures;
15013d4c4710SBarry Smith   PetscFunctionReturn(0);
15023d4c4710SBarry Smith }
15033d4c4710SBarry Smith 
15043d4c4710SBarry Smith /*@
15053d4c4710SBarry Smith    SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts
15063d4c4710SBarry Smith    allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE
15073d4c4710SBarry Smith 
15083f9fe445SBarry Smith    Logically Collective on SNES
15093d4c4710SBarry Smith 
15103d4c4710SBarry Smith    Input Parameters:
15113d4c4710SBarry Smith +  snes     - SNES context
15123d4c4710SBarry Smith -  maxFails - maximum allowed linear solve failures
15133d4c4710SBarry Smith 
15143d4c4710SBarry Smith    Level: intermediate
15153d4c4710SBarry Smith 
15169d85da0cSMatthew G. Knepley    Options Database Keys:
15179d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated
15189d85da0cSMatthew G. Knepley 
151995452b02SPatrick Sanan    Notes:
152095452b02SPatrick Sanan     By default this is 0; that is SNES returns on the first failed linear solve
15213d4c4710SBarry Smith 
152258ebbce7SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations()
15233d4c4710SBarry Smith @*/
15247087cfbeSBarry Smith PetscErrorCode  SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails)
15253d4c4710SBarry Smith {
15263d4c4710SBarry Smith   PetscFunctionBegin;
15270700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1528c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxFails,2);
15293d4c4710SBarry Smith   snes->maxLinearSolveFailures = maxFails;
15303d4c4710SBarry Smith   PetscFunctionReturn(0);
15313d4c4710SBarry Smith }
15323d4c4710SBarry Smith 
15333d4c4710SBarry Smith /*@
15343d4c4710SBarry Smith    SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that
15353d4c4710SBarry Smith      are allowed before SNES terminates
15363d4c4710SBarry Smith 
15373d4c4710SBarry Smith    Not Collective
15383d4c4710SBarry Smith 
15393d4c4710SBarry Smith    Input Parameter:
15403d4c4710SBarry Smith .  snes     - SNES context
15413d4c4710SBarry Smith 
15423d4c4710SBarry Smith    Output Parameter:
15433d4c4710SBarry Smith .  maxFails - maximum of unsuccessful solves allowed
15443d4c4710SBarry Smith 
15453d4c4710SBarry Smith    Level: intermediate
15463d4c4710SBarry Smith 
154795452b02SPatrick Sanan    Notes:
154895452b02SPatrick Sanan     By default this is 1; that is SNES returns on the first failed linear solve
15493d4c4710SBarry Smith 
1550e1c61ce8SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(),
15513d4c4710SBarry Smith @*/
15527087cfbeSBarry Smith PetscErrorCode  SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails)
15533d4c4710SBarry Smith {
15543d4c4710SBarry Smith   PetscFunctionBegin;
15550700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
15563d4c4710SBarry Smith   PetscValidIntPointer(maxFails,2);
15573d4c4710SBarry Smith   *maxFails = snes->maxLinearSolveFailures;
15583d4c4710SBarry Smith   PetscFunctionReturn(0);
15593d4c4710SBarry Smith }
15603d4c4710SBarry Smith 
1561c96a6f78SLois Curfman McInnes /*@
1562b850b91aSLisandro Dalcin    SNESGetLinearSolveIterations - Gets the total number of linear iterations
1563c96a6f78SLois Curfman McInnes    used by the nonlinear solver.
1564c96a6f78SLois Curfman McInnes 
1565c7afd0dbSLois Curfman McInnes    Not Collective
1566c7afd0dbSLois Curfman McInnes 
1567c96a6f78SLois Curfman McInnes    Input Parameter:
1568c96a6f78SLois Curfman McInnes .  snes - SNES context
1569c96a6f78SLois Curfman McInnes 
1570c96a6f78SLois Curfman McInnes    Output Parameter:
1571c96a6f78SLois Curfman McInnes .  lits - number of linear iterations
1572c96a6f78SLois Curfman McInnes 
1573c96a6f78SLois Curfman McInnes    Notes:
1574971e163fSPeter Brune    This counter is reset to zero for each successive call to SNESSolve() unless SNESSetCountersReset() is used.
1575c96a6f78SLois Curfman McInnes 
1576010be392SBarry Smith    If the linear solver fails inside the SNESSolve() the iterations for that call to the linear solver are not included. If you wish to count them
1577010be392SBarry Smith    then call KSPGetIterationNumber() after the failed solve.
1578010be392SBarry Smith 
157936851e7fSLois Curfman McInnes    Level: intermediate
158036851e7fSLois Curfman McInnes 
158171dbe336SPeter Brune .seealso:  SNESGetIterationNumber(), SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESSetCountersReset()
1582c96a6f78SLois Curfman McInnes @*/
15837087cfbeSBarry Smith PetscErrorCode  SNESGetLinearSolveIterations(SNES snes,PetscInt *lits)
1584c96a6f78SLois Curfman McInnes {
15853a40ed3dSBarry Smith   PetscFunctionBegin;
15860700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
15874482741eSBarry Smith   PetscValidIntPointer(lits,2);
1588c96a6f78SLois Curfman McInnes   *lits = snes->linear_its;
15893a40ed3dSBarry Smith   PetscFunctionReturn(0);
1590c96a6f78SLois Curfman McInnes }
1591c96a6f78SLois Curfman McInnes 
1592971e163fSPeter Brune /*@
1593971e163fSPeter Brune    SNESSetCountersReset - Sets whether or not the counters for linear iterations and function evaluations
1594971e163fSPeter Brune    are reset every time SNESSolve() is called.
1595971e163fSPeter Brune 
1596971e163fSPeter Brune    Logically Collective on SNES
1597971e163fSPeter Brune 
1598971e163fSPeter Brune    Input Parameter:
1599971e163fSPeter Brune +  snes - SNES context
1600971e163fSPeter Brune -  reset - whether to reset the counters or not
1601971e163fSPeter Brune 
1602971e163fSPeter Brune    Notes:
1603fa19ca70SBarry Smith    This defaults to PETSC_TRUE
1604971e163fSPeter Brune 
1605971e163fSPeter Brune    Level: developer
1606971e163fSPeter Brune 
1607734794cfSBarry Smith .seealso:  SNESGetNumberFunctionEvals(), SNESGetLinearSolveIterations(), SNESGetNPC()
1608971e163fSPeter Brune @*/
1609971e163fSPeter Brune PetscErrorCode  SNESSetCountersReset(SNES snes,PetscBool reset)
1610971e163fSPeter Brune {
1611971e163fSPeter Brune   PetscFunctionBegin;
1612971e163fSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1613971e163fSPeter Brune   PetscValidLogicalCollectiveBool(snes,reset,2);
1614971e163fSPeter Brune   snes->counters_reset = reset;
1615971e163fSPeter Brune   PetscFunctionReturn(0);
1616971e163fSPeter Brune }
1617971e163fSPeter Brune 
161882bf6240SBarry Smith 
16192999313aSBarry Smith /*@
16202999313aSBarry Smith    SNESSetKSP - Sets a KSP context for the SNES object to use
16212999313aSBarry Smith 
16222999313aSBarry Smith    Not Collective, but the SNES and KSP objects must live on the same MPI_Comm
16232999313aSBarry Smith 
16242999313aSBarry Smith    Input Parameters:
16252999313aSBarry Smith +  snes - the SNES context
16262999313aSBarry Smith -  ksp - the KSP context
16272999313aSBarry Smith 
16282999313aSBarry Smith    Notes:
16292999313aSBarry Smith    The SNES object already has its KSP object, you can obtain with SNESGetKSP()
16302999313aSBarry Smith    so this routine is rarely needed.
16312999313aSBarry Smith 
16322999313aSBarry Smith    The KSP object that is already in the SNES object has its reference count
16332999313aSBarry Smith    decreased by one.
16342999313aSBarry Smith 
16352999313aSBarry Smith    Level: developer
16362999313aSBarry Smith 
16372999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
16382999313aSBarry Smith @*/
16397087cfbeSBarry Smith PetscErrorCode  SNESSetKSP(SNES snes,KSP ksp)
16402999313aSBarry Smith {
16412999313aSBarry Smith   PetscErrorCode ierr;
16422999313aSBarry Smith 
16432999313aSBarry Smith   PetscFunctionBegin;
16440700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
16450700a824SBarry Smith   PetscValidHeaderSpecific(ksp,KSP_CLASSID,2);
16462999313aSBarry Smith   PetscCheckSameComm(snes,1,ksp,2);
16477dcf0eaaSdalcinl   ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr);
1648906ed7ccSBarry Smith   if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);}
16492999313aSBarry Smith   snes->ksp = ksp;
16502999313aSBarry Smith   PetscFunctionReturn(0);
16512999313aSBarry Smith }
16522999313aSBarry Smith 
16539b94acceSBarry Smith /* -----------------------------------------------------------*/
165452baeb72SSatish Balay /*@
16559b94acceSBarry Smith    SNESCreate - Creates a nonlinear solver context.
16569b94acceSBarry Smith 
1657d083f849SBarry Smith    Collective
1658c7afd0dbSLois Curfman McInnes 
1659c7afd0dbSLois Curfman McInnes    Input Parameters:
1660906ed7ccSBarry Smith .  comm - MPI communicator
16619b94acceSBarry Smith 
16629b94acceSBarry Smith    Output Parameter:
16639b94acceSBarry Smith .  outsnes - the new SNES context
16649b94acceSBarry Smith 
1665c7afd0dbSLois Curfman McInnes    Options Database Keys:
1666c7afd0dbSLois Curfman McInnes +   -snes_mf - Activates default matrix-free Jacobian-vector products,
1667c7afd0dbSLois Curfman McInnes                and no preconditioning matrix
1668c7afd0dbSLois Curfman McInnes .   -snes_mf_operator - Activates default matrix-free Jacobian-vector
1669c7afd0dbSLois Curfman McInnes                products, and a user-provided preconditioning matrix
1670c7afd0dbSLois Curfman McInnes                as set by SNESSetJacobian()
1671c7afd0dbSLois Curfman McInnes -   -snes_fd - Uses (slow!) finite differences to compute Jacobian
1672c1f60f51SBarry Smith 
167336851e7fSLois Curfman McInnes    Level: beginner
167436851e7fSLois Curfman McInnes 
167595452b02SPatrick Sanan    Developer Notes:
167695452b02SPatrick Sanan     SNES always creates a KSP object even though many SNES methods do not use it. This is
1677efd4aadfSBarry Smith                     unfortunate and should be fixed at some point. The flag snes->usesksp indicates if the
1678efd4aadfSBarry Smith                     particular method does use KSP and regulates if the information about the KSP is printed
1679efd4aadfSBarry Smith                     in SNESView(). TSSetFromOptions() does call SNESSetFromOptions() which can lead to users being confused
1680efd4aadfSBarry Smith                     by help messages about meaningless SNES options.
1681efd4aadfSBarry Smith 
1682efd4aadfSBarry Smith                     SNES always creates the snes->kspconvctx even though it is used by only one type. This should
1683efd4aadfSBarry Smith                     be fixed.
1684efd4aadfSBarry Smith 
16853d5a8a6aSBarry Smith .seealso: SNESSolve(), SNESDestroy(), SNES, SNESSetLagPreconditioner(), SNESSetLagJacobian()
1686a8054027SBarry Smith 
16879b94acceSBarry Smith @*/
16887087cfbeSBarry Smith PetscErrorCode  SNESCreate(MPI_Comm comm,SNES *outsnes)
16899b94acceSBarry Smith {
1690dfbe8321SBarry Smith   PetscErrorCode ierr;
16919b94acceSBarry Smith   SNES           snes;
1692fa9f3622SBarry Smith   SNESKSPEW      *kctx;
169337fcc0dbSBarry Smith 
16943a40ed3dSBarry Smith   PetscFunctionBegin;
1695ed1caa07SMatthew Knepley   PetscValidPointer(outsnes,2);
16960298fd71SBarry Smith   *outsnes = NULL;
1697607a6623SBarry Smith   ierr = SNESInitializePackage();CHKERRQ(ierr);
16988ba1e511SMatthew Knepley 
169973107ff1SLisandro Dalcin   ierr = PetscHeaderCreate(snes,SNES_CLASSID,"SNES","Nonlinear solver","SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr);
17007adad957SLisandro Dalcin 
17018d359177SBarry Smith   snes->ops->converged    = SNESConvergedDefault;
17022c155ee1SBarry Smith   snes->usesksp           = PETSC_TRUE;
170388976e71SPeter Brune   snes->tolerancesset     = PETSC_FALSE;
17049b94acceSBarry Smith   snes->max_its           = 50;
17059750a799SBarry Smith   snes->max_funcs         = 10000;
17069b94acceSBarry Smith   snes->norm              = 0.0;
1707c1e67a49SFande Kong   snes->xnorm             = 0.0;
1708c1e67a49SFande Kong   snes->ynorm             = 0.0;
1709365a6726SPeter Brune   snes->normschedule      = SNES_NORM_ALWAYS;
17106c67d002SPeter Brune   snes->functype          = SNES_FUNCTION_DEFAULT;
17113a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
17123a2046daSBarry Smith   snes->rtol              = 1.e-5;
17133a2046daSBarry Smith #else
1714b4874afaSBarry Smith   snes->rtol              = 1.e-8;
17153a2046daSBarry Smith #endif
1716b4874afaSBarry Smith   snes->ttol              = 0.0;
17173a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
17183a2046daSBarry Smith   snes->abstol            = 1.e-25;
17193a2046daSBarry Smith #else
172070441072SBarry Smith   snes->abstol            = 1.e-50;
17213a2046daSBarry Smith #endif
17227cd0ae37SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE)
17237cd0ae37SLisandro Dalcin   snes->stol              = 1.e-5;
17247cd0ae37SLisandro Dalcin #else
1725c60f73f4SPeter Brune   snes->stol              = 1.e-8;
17267cd0ae37SLisandro Dalcin #endif
17273a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
17283a2046daSBarry Smith   snes->deltatol          = 1.e-6;
17293a2046daSBarry Smith #else
17304b27c08aSLois Curfman McInnes   snes->deltatol          = 1.e-12;
17313a2046daSBarry Smith #endif
1732e37c518bSBarry Smith   snes->divtol            = 1.e4;
1733e37c518bSBarry Smith   snes->rnorm0            = 0;
17349b94acceSBarry Smith   snes->nfuncs            = 0;
173550ffb88aSMatthew Knepley   snes->numFailures       = 0;
173650ffb88aSMatthew Knepley   snes->maxFailures       = 1;
17377a00f4a9SLois Curfman McInnes   snes->linear_its        = 0;
1738e35cf81dSBarry Smith   snes->lagjacobian       = 1;
173937ec4e1aSPeter Brune   snes->jac_iter          = 0;
174037ec4e1aSPeter Brune   snes->lagjac_persist    = PETSC_FALSE;
1741a8054027SBarry Smith   snes->lagpreconditioner = 1;
174237ec4e1aSPeter Brune   snes->pre_iter          = 0;
174337ec4e1aSPeter Brune   snes->lagpre_persist    = PETSC_FALSE;
1744639f9d9dSBarry Smith   snes->numbermonitors    = 0;
1745c4421ceaSFande Kong   snes->numberreasonviews = 0;
17469e5d0892SLisandro Dalcin   snes->data              = NULL;
17474dc4c822SBarry Smith   snes->setupcalled       = PETSC_FALSE;
1748186905e3SBarry Smith   snes->ksp_ewconv        = PETSC_FALSE;
17496f24a144SLois Curfman McInnes   snes->nwork             = 0;
17509e5d0892SLisandro Dalcin   snes->work              = NULL;
175158c9b817SLisandro Dalcin   snes->nvwork            = 0;
17529e5d0892SLisandro Dalcin   snes->vwork             = NULL;
1753758f92a0SBarry Smith   snes->conv_hist_len     = 0;
1754758f92a0SBarry Smith   snes->conv_hist_max     = 0;
17550298fd71SBarry Smith   snes->conv_hist         = NULL;
17560298fd71SBarry Smith   snes->conv_hist_its     = NULL;
1757758f92a0SBarry Smith   snes->conv_hist_reset   = PETSC_TRUE;
1758971e163fSPeter Brune   snes->counters_reset    = PETSC_TRUE;
1759e4ed7901SPeter Brune   snes->vec_func_init_set = PETSC_FALSE;
1760184914b5SBarry Smith   snes->reason            = SNES_CONVERGED_ITERATING;
1761efd4aadfSBarry Smith   snes->npcside           = PC_RIGHT;
1762b3cd9a81SMatthew G. Knepley   snes->setfromoptionscalled = 0;
1763c40d0f55SPeter Brune 
1764d8f46077SPeter Brune   snes->mf          = PETSC_FALSE;
1765d8f46077SPeter Brune   snes->mf_operator = PETSC_FALSE;
1766d8f46077SPeter Brune   snes->mf_version  = 1;
1767d8f46077SPeter Brune 
17683d4c4710SBarry Smith   snes->numLinearSolveFailures = 0;
17693d4c4710SBarry Smith   snes->maxLinearSolveFailures = 1;
17703d4c4710SBarry Smith 
1771349187a7SBarry Smith   snes->vizerotolerance = 1.e-8;
177276bd3646SJed Brown   snes->checkjacdomainerror = PetscDefined(USE_DEBUG) ? PETSC_TRUE : PETSC_FALSE;
1773349187a7SBarry Smith 
17744fc747eaSLawrence Mitchell   /* Set this to true if the implementation of SNESSolve_XXX does compute the residual at the final solution. */
17754fc747eaSLawrence Mitchell   snes->alwayscomputesfinalresidual = PETSC_FALSE;
17764fc747eaSLawrence Mitchell 
17779b94acceSBarry Smith   /* Create context to compute Eisenstat-Walker relative tolerance for KSP */
1778b00a9115SJed Brown   ierr = PetscNewLog(snes,&kctx);CHKERRQ(ierr);
1779f5af7f23SKarl Rupp 
17809b94acceSBarry Smith   snes->kspconvctx  = (void*)kctx;
17819b94acceSBarry Smith   kctx->version     = 2;
17829b94acceSBarry Smith   kctx->rtol_0      = .3; /* Eisenstat and Walker suggest rtol_0=.5, but
17839b94acceSBarry Smith                              this was too large for some test cases */
178475567043SBarry Smith   kctx->rtol_last   = 0.0;
17859b94acceSBarry Smith   kctx->rtol_max    = .9;
17869b94acceSBarry Smith   kctx->gamma       = 1.0;
178762d1f40fSBarry Smith   kctx->alpha       = .5*(1.0 + PetscSqrtReal(5.0));
178871f87433Sdalcinl   kctx->alpha2      = kctx->alpha;
17899b94acceSBarry Smith   kctx->threshold   = .1;
179075567043SBarry Smith   kctx->lresid_last = 0.0;
179175567043SBarry Smith   kctx->norm_last   = 0.0;
17929b94acceSBarry Smith 
17939b94acceSBarry Smith   *outsnes = snes;
17943a40ed3dSBarry Smith   PetscFunctionReturn(0);
17959b94acceSBarry Smith }
17969b94acceSBarry Smith 
179788f0584fSBarry Smith /*MC
1798411c0326SBarry Smith     SNESFunction - Functional form used to convey the nonlinear function to be solved by SNES
179988f0584fSBarry Smith 
180088f0584fSBarry Smith      Synopsis:
1801411c0326SBarry Smith      #include "petscsnes.h"
1802411c0326SBarry Smith      PetscErrorCode SNESFunction(SNES snes,Vec x,Vec f,void *ctx);
180388f0584fSBarry Smith 
18041843f636SBarry Smith      Collective on snes
18051843f636SBarry Smith 
180688f0584fSBarry Smith      Input Parameters:
180788f0584fSBarry Smith +     snes - the SNES context
180888f0584fSBarry Smith .     x    - state at which to evaluate residual
180988f0584fSBarry Smith -     ctx     - optional user-defined function context, passed in with SNESSetFunction()
181088f0584fSBarry Smith 
181188f0584fSBarry Smith      Output Parameter:
181288f0584fSBarry Smith .     f  - vector to put residual (function value)
181388f0584fSBarry Smith 
1814878cb397SSatish Balay    Level: intermediate
1815878cb397SSatish Balay 
181688f0584fSBarry Smith .seealso:   SNESSetFunction(), SNESGetFunction()
181788f0584fSBarry Smith M*/
181888f0584fSBarry Smith 
18199b94acceSBarry Smith /*@C
18209b94acceSBarry Smith    SNESSetFunction - Sets the function evaluation routine and function
18219b94acceSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
18229b94acceSBarry Smith    equations.
18239b94acceSBarry Smith 
18243f9fe445SBarry Smith    Logically Collective on SNES
1825fee21e36SBarry Smith 
1826c7afd0dbSLois Curfman McInnes    Input Parameters:
1827c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1828c7afd0dbSLois Curfman McInnes .  r - vector to store function value
1829f8b49ee9SBarry Smith .  f - function evaluation routine; see SNESFunction for calling sequence details
1830c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
18310298fd71SBarry Smith          function evaluation routine (may be NULL)
18329b94acceSBarry Smith 
18339b94acceSBarry Smith    Notes:
18349b94acceSBarry Smith    The Newton-like methods typically solve linear systems of the form
18359b94acceSBarry Smith $      f'(x) x = -f(x),
1836c7afd0dbSLois Curfman McInnes    where f'(x) denotes the Jacobian matrix and f(x) is the function.
18379b94acceSBarry Smith 
183836851e7fSLois Curfman McInnes    Level: beginner
183936851e7fSLois Curfman McInnes 
1840bf388a1fSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetPicard(), SNESFunction
18419b94acceSBarry Smith @*/
1842f8b49ee9SBarry Smith PetscErrorCode  SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx)
18439b94acceSBarry Smith {
184485385478SLisandro Dalcin   PetscErrorCode ierr;
18456cab3a1bSJed Brown   DM             dm;
18466cab3a1bSJed Brown 
18473a40ed3dSBarry Smith   PetscFunctionBegin;
18480700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1849d2a683ecSLisandro Dalcin   if (r) {
1850d2a683ecSLisandro Dalcin     PetscValidHeaderSpecific(r,VEC_CLASSID,2);
1851d2a683ecSLisandro Dalcin     PetscCheckSameComm(snes,1,r,2);
185285385478SLisandro Dalcin     ierr = PetscObjectReference((PetscObject)r);CHKERRQ(ierr);
18536bf464f9SBarry Smith     ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr);
1854f5af7f23SKarl Rupp 
185585385478SLisandro Dalcin     snes->vec_func = r;
1856d2a683ecSLisandro Dalcin   }
18576cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
1858f8b49ee9SBarry Smith   ierr = DMSNESSetFunction(dm,f,ctx);CHKERRQ(ierr);
18593a40ed3dSBarry Smith   PetscFunctionReturn(0);
18609b94acceSBarry Smith }
18619b94acceSBarry Smith 
1862646217ecSPeter Brune 
1863e4ed7901SPeter Brune /*@C
1864e4ed7901SPeter Brune    SNESSetInitialFunction - Sets the function vector to be used as the
1865e4ed7901SPeter Brune    function norm at the initialization of the method.  In some
1866e4ed7901SPeter Brune    instances, the user has precomputed the function before calling
1867e4ed7901SPeter Brune    SNESSolve.  This function allows one to avoid a redundant call
1868e4ed7901SPeter Brune    to SNESComputeFunction in that case.
1869e4ed7901SPeter Brune 
1870e4ed7901SPeter Brune    Logically Collective on SNES
1871e4ed7901SPeter Brune 
1872e4ed7901SPeter Brune    Input Parameters:
1873e4ed7901SPeter Brune +  snes - the SNES context
1874e4ed7901SPeter Brune -  f - vector to store function value
1875e4ed7901SPeter Brune 
1876e4ed7901SPeter Brune    Notes:
1877e4ed7901SPeter Brune    This should not be modified during the solution procedure.
1878e4ed7901SPeter Brune 
1879e4ed7901SPeter Brune    This is used extensively in the SNESFAS hierarchy and in nonlinear preconditioning.
1880e4ed7901SPeter Brune 
1881e4ed7901SPeter Brune    Level: developer
1882e4ed7901SPeter Brune 
1883e4ed7901SPeter Brune .seealso: SNESSetFunction(), SNESComputeFunction(), SNESSetInitialFunctionNorm()
1884e4ed7901SPeter Brune @*/
1885e4ed7901SPeter Brune PetscErrorCode  SNESSetInitialFunction(SNES snes, Vec f)
1886e4ed7901SPeter Brune {
1887e4ed7901SPeter Brune   PetscErrorCode ierr;
1888e4ed7901SPeter Brune   Vec            vec_func;
1889e4ed7901SPeter Brune 
1890e4ed7901SPeter Brune   PetscFunctionBegin;
1891e4ed7901SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1892e4ed7901SPeter Brune   PetscValidHeaderSpecific(f,VEC_CLASSID,2);
1893e4ed7901SPeter Brune   PetscCheckSameComm(snes,1,f,2);
1894efd4aadfSBarry Smith   if (snes->npcside== PC_LEFT && snes->functype == SNES_FUNCTION_PRECONDITIONED) {
1895902f982fSPeter Brune     snes->vec_func_init_set = PETSC_FALSE;
1896902f982fSPeter Brune     PetscFunctionReturn(0);
1897902f982fSPeter Brune   }
18980298fd71SBarry Smith   ierr = SNESGetFunction(snes,&vec_func,NULL,NULL);CHKERRQ(ierr);
1899e4ed7901SPeter Brune   ierr = VecCopy(f, vec_func);CHKERRQ(ierr);
1900f5af7f23SKarl Rupp 
1901217b9c2eSPeter Brune   snes->vec_func_init_set = PETSC_TRUE;
1902e4ed7901SPeter Brune   PetscFunctionReturn(0);
1903e4ed7901SPeter Brune }
1904e4ed7901SPeter Brune 
1905534ebe21SPeter Brune /*@
1906365a6726SPeter Brune    SNESSetNormSchedule - Sets the SNESNormSchedule used in covergence and monitoring
1907534ebe21SPeter Brune    of the SNES method.
1908534ebe21SPeter Brune 
1909534ebe21SPeter Brune    Logically Collective on SNES
1910534ebe21SPeter Brune 
1911534ebe21SPeter Brune    Input Parameters:
1912534ebe21SPeter Brune +  snes - the SNES context
1913365a6726SPeter Brune -  normschedule - the frequency of norm computation
1914534ebe21SPeter Brune 
1915517f1916SMatthew G. Knepley    Options Database Key:
1916517f1916SMatthew G. Knepley .  -snes_norm_schedule <none, always, initialonly, finalonly, initalfinalonly>
1917517f1916SMatthew G. Knepley 
1918534ebe21SPeter Brune    Notes:
1919365a6726SPeter Brune    Only certain SNES methods support certain SNESNormSchedules.  Most require evaluation
1920534ebe21SPeter Brune    of the nonlinear function and the taking of its norm at every iteration to
1921534ebe21SPeter Brune    even ensure convergence at all.  However, methods such as custom Gauss-Seidel methods
1922be95d8f1SBarry Smith    (SNESNGS) and the like do not require the norm of the function to be computed, and therfore
1923534ebe21SPeter Brune    may either be monitored for convergence or not.  As these are often used as nonlinear
1924534ebe21SPeter Brune    preconditioners, monitoring the norm of their error is not a useful enterprise within
1925534ebe21SPeter Brune    their solution.
1926534ebe21SPeter Brune 
1927534ebe21SPeter Brune    Level: developer
1928534ebe21SPeter Brune 
1929365a6726SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1930534ebe21SPeter Brune @*/
1931365a6726SPeter Brune PetscErrorCode  SNESSetNormSchedule(SNES snes, SNESNormSchedule normschedule)
1932534ebe21SPeter Brune {
1933534ebe21SPeter Brune   PetscFunctionBegin;
1934534ebe21SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1935365a6726SPeter Brune   snes->normschedule = normschedule;
1936534ebe21SPeter Brune   PetscFunctionReturn(0);
1937534ebe21SPeter Brune }
1938534ebe21SPeter Brune 
1939534ebe21SPeter Brune 
1940534ebe21SPeter Brune /*@
1941365a6726SPeter Brune    SNESGetNormSchedule - Gets the SNESNormSchedule used in covergence and monitoring
1942534ebe21SPeter Brune    of the SNES method.
1943534ebe21SPeter Brune 
1944534ebe21SPeter Brune    Logically Collective on SNES
1945534ebe21SPeter Brune 
1946534ebe21SPeter Brune    Input Parameters:
1947534ebe21SPeter Brune +  snes - the SNES context
1948365a6726SPeter Brune -  normschedule - the type of the norm used
1949534ebe21SPeter Brune 
1950534ebe21SPeter Brune    Level: advanced
1951534ebe21SPeter Brune 
1952365a6726SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1953534ebe21SPeter Brune @*/
1954365a6726SPeter Brune PetscErrorCode  SNESGetNormSchedule(SNES snes, SNESNormSchedule *normschedule)
1955534ebe21SPeter Brune {
1956534ebe21SPeter Brune   PetscFunctionBegin;
1957534ebe21SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1958365a6726SPeter Brune   *normschedule = snes->normschedule;
1959534ebe21SPeter Brune   PetscFunctionReturn(0);
1960534ebe21SPeter Brune }
1961534ebe21SPeter Brune 
196247073ea2SPeter Brune 
1963c5ce4427SMatthew G. Knepley /*@
1964c5ce4427SMatthew G. Knepley   SNESSetFunctionNorm - Sets the last computed residual norm.
1965c5ce4427SMatthew G. Knepley 
1966c5ce4427SMatthew G. Knepley   Logically Collective on SNES
1967c5ce4427SMatthew G. Knepley 
1968c5ce4427SMatthew G. Knepley   Input Parameters:
1969c5ce4427SMatthew G. Knepley + snes - the SNES context
1970c5ce4427SMatthew G. Knepley 
1971c5ce4427SMatthew G. Knepley - normschedule - the frequency of norm computation
1972c5ce4427SMatthew G. Knepley 
1973c5ce4427SMatthew G. Knepley   Level: developer
1974c5ce4427SMatthew G. Knepley 
1975c5ce4427SMatthew G. Knepley .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1976c5ce4427SMatthew G. Knepley @*/
1977c5ce4427SMatthew G. Knepley PetscErrorCode SNESSetFunctionNorm(SNES snes, PetscReal norm)
1978c5ce4427SMatthew G. Knepley {
1979c5ce4427SMatthew G. Knepley   PetscFunctionBegin;
1980c5ce4427SMatthew G. Knepley   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1981c5ce4427SMatthew G. Knepley   snes->norm = norm;
1982c5ce4427SMatthew G. Knepley   PetscFunctionReturn(0);
1983c5ce4427SMatthew G. Knepley }
1984c5ce4427SMatthew G. Knepley 
1985c5ce4427SMatthew G. Knepley /*@
1986c5ce4427SMatthew G. Knepley   SNESGetFunctionNorm - Gets the last computed norm of the residual
1987c5ce4427SMatthew G. Knepley 
1988c5ce4427SMatthew G. Knepley   Not Collective
1989c5ce4427SMatthew G. Knepley 
1990c5ce4427SMatthew G. Knepley   Input Parameter:
1991c5ce4427SMatthew G. Knepley . snes - the SNES context
1992c5ce4427SMatthew G. Knepley 
1993c5ce4427SMatthew G. Knepley   Output Parameter:
1994c5ce4427SMatthew G. Knepley . norm - the last computed residual norm
1995c5ce4427SMatthew G. Knepley 
1996c5ce4427SMatthew G. Knepley   Level: developer
1997c5ce4427SMatthew G. Knepley 
1998c5ce4427SMatthew G. Knepley .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1999c5ce4427SMatthew G. Knepley @*/
2000c5ce4427SMatthew G. Knepley PetscErrorCode SNESGetFunctionNorm(SNES snes, PetscReal *norm)
2001c5ce4427SMatthew G. Knepley {
2002c5ce4427SMatthew G. Knepley   PetscFunctionBegin;
2003c5ce4427SMatthew G. Knepley   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2004c5ce4427SMatthew G. Knepley   PetscValidPointer(norm, 2);
2005c5ce4427SMatthew G. Knepley   *norm = snes->norm;
2006c5ce4427SMatthew G. Knepley   PetscFunctionReturn(0);
2007c5ce4427SMatthew G. Knepley }
2008c5ce4427SMatthew G. Knepley 
2009c1e67a49SFande Kong /*@
2010c1e67a49SFande Kong   SNESGetUpdateNorm - Gets the last computed norm of the Newton update
2011c1e67a49SFande Kong 
2012c1e67a49SFande Kong   Not Collective
2013c1e67a49SFande Kong 
2014c1e67a49SFande Kong   Input Parameter:
2015c1e67a49SFande Kong . snes - the SNES context
2016c1e67a49SFande Kong 
2017c1e67a49SFande Kong   Output Parameter:
2018c1e67a49SFande Kong . ynorm - the last computed update norm
2019c1e67a49SFande Kong 
2020c1e67a49SFande Kong   Level: developer
2021c1e67a49SFande Kong 
2022c1e67a49SFande Kong .seealso: SNESSetNormSchedule(), SNESComputeFunction(), SNESGetFunctionNorm()
2023c1e67a49SFande Kong @*/
2024c1e67a49SFande Kong PetscErrorCode SNESGetUpdateNorm(SNES snes, PetscReal *ynorm)
2025c1e67a49SFande Kong {
2026c1e67a49SFande Kong   PetscFunctionBegin;
2027c1e67a49SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2028c1e67a49SFande Kong   PetscValidPointer(ynorm, 2);
2029c1e67a49SFande Kong   *ynorm = snes->ynorm;
2030c1e67a49SFande Kong   PetscFunctionReturn(0);
2031c1e67a49SFande Kong }
2032c1e67a49SFande Kong 
2033c1e67a49SFande Kong /*@
20344591eaf2SFande Kong   SNESGetSolutionNorm - Gets the last computed norm of the solution
2035c1e67a49SFande Kong 
2036c1e67a49SFande Kong   Not Collective
2037c1e67a49SFande Kong 
2038c1e67a49SFande Kong   Input Parameter:
2039c1e67a49SFande Kong . snes - the SNES context
2040c1e67a49SFande Kong 
2041c1e67a49SFande Kong   Output Parameter:
2042c1e67a49SFande Kong . xnorm - the last computed solution norm
2043c1e67a49SFande Kong 
2044c1e67a49SFande Kong   Level: developer
2045c1e67a49SFande Kong 
2046c1e67a49SFande Kong .seealso: SNESSetNormSchedule(), SNESComputeFunction(), SNESGetFunctionNorm(), SNESGetUpdateNorm()
2047c1e67a49SFande Kong @*/
2048c1e67a49SFande Kong PetscErrorCode SNESGetSolutionNorm(SNES snes, PetscReal *xnorm)
2049c1e67a49SFande Kong {
2050c1e67a49SFande Kong   PetscFunctionBegin;
2051c1e67a49SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2052c1e67a49SFande Kong   PetscValidPointer(xnorm, 2);
2053c1e67a49SFande Kong   *xnorm = snes->xnorm;
2054c1e67a49SFande Kong   PetscFunctionReturn(0);
2055c1e67a49SFande Kong }
2056c1e67a49SFande Kong 
205747073ea2SPeter Brune /*@C
205847073ea2SPeter Brune    SNESSetFunctionType - Sets the SNESNormSchedule used in covergence and monitoring
205947073ea2SPeter Brune    of the SNES method.
206047073ea2SPeter Brune 
206147073ea2SPeter Brune    Logically Collective on SNES
206247073ea2SPeter Brune 
206347073ea2SPeter Brune    Input Parameters:
206447073ea2SPeter Brune +  snes - the SNES context
206547073ea2SPeter Brune -  normschedule - the frequency of norm computation
206647073ea2SPeter Brune 
206747073ea2SPeter Brune    Notes:
206847073ea2SPeter Brune    Only certain SNES methods support certain SNESNormSchedules.  Most require evaluation
206947073ea2SPeter Brune    of the nonlinear function and the taking of its norm at every iteration to
207047073ea2SPeter Brune    even ensure convergence at all.  However, methods such as custom Gauss-Seidel methods
2071be95d8f1SBarry Smith    (SNESNGS) and the like do not require the norm of the function to be computed, and therfore
207247073ea2SPeter Brune    may either be monitored for convergence or not.  As these are often used as nonlinear
207347073ea2SPeter Brune    preconditioners, monitoring the norm of their error is not a useful enterprise within
207447073ea2SPeter Brune    their solution.
207547073ea2SPeter Brune 
207647073ea2SPeter Brune    Level: developer
207747073ea2SPeter Brune 
207847073ea2SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
207947073ea2SPeter Brune @*/
208047073ea2SPeter Brune PetscErrorCode  SNESSetFunctionType(SNES snes, SNESFunctionType type)
208147073ea2SPeter Brune {
208247073ea2SPeter Brune   PetscFunctionBegin;
208347073ea2SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
208447073ea2SPeter Brune   snes->functype = type;
208547073ea2SPeter Brune   PetscFunctionReturn(0);
208647073ea2SPeter Brune }
208747073ea2SPeter Brune 
208847073ea2SPeter Brune 
208947073ea2SPeter Brune /*@C
209047073ea2SPeter Brune    SNESGetFunctionType - Gets the SNESNormSchedule used in covergence and monitoring
209147073ea2SPeter Brune    of the SNES method.
209247073ea2SPeter Brune 
209347073ea2SPeter Brune    Logically Collective on SNES
209447073ea2SPeter Brune 
209547073ea2SPeter Brune    Input Parameters:
209647073ea2SPeter Brune +  snes - the SNES context
209747073ea2SPeter Brune -  normschedule - the type of the norm used
209847073ea2SPeter Brune 
209947073ea2SPeter Brune    Level: advanced
210047073ea2SPeter Brune 
210147073ea2SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
210247073ea2SPeter Brune @*/
210347073ea2SPeter Brune PetscErrorCode  SNESGetFunctionType(SNES snes, SNESFunctionType *type)
210447073ea2SPeter Brune {
210547073ea2SPeter Brune   PetscFunctionBegin;
210647073ea2SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
210747073ea2SPeter Brune   *type = snes->functype;
2108534ebe21SPeter Brune   PetscFunctionReturn(0);
2109534ebe21SPeter Brune }
2110534ebe21SPeter Brune 
2111bf388a1fSBarry Smith /*MC
2112be95d8f1SBarry Smith     SNESNGSFunction - function used to convey a Gauss-Seidel sweep on the nonlinear function
2113bf388a1fSBarry Smith 
2114bf388a1fSBarry Smith      Synopsis:
2115aaa7dc30SBarry Smith      #include <petscsnes.h>
2116be95d8f1SBarry Smith $    SNESNGSFunction(SNES snes,Vec x,Vec b,void *ctx);
2117bf388a1fSBarry Smith 
21181843f636SBarry Smith      Collective on snes
21191843f636SBarry Smith 
21201843f636SBarry Smith      Input Parameters:
2121bf388a1fSBarry Smith +  X   - solution vector
2122bf388a1fSBarry Smith .  B   - RHS vector
2123bf388a1fSBarry Smith -  ctx - optional user-defined Gauss-Seidel context
2124bf388a1fSBarry Smith 
21251843f636SBarry Smith      Output Parameter:
21261843f636SBarry Smith .  X   - solution vector
21271843f636SBarry Smith 
2128878cb397SSatish Balay    Level: intermediate
2129878cb397SSatish Balay 
2130be95d8f1SBarry Smith .seealso:   SNESSetNGS(), SNESGetNGS()
2131bf388a1fSBarry Smith M*/
2132bf388a1fSBarry Smith 
2133c79ef259SPeter Brune /*@C
2134be95d8f1SBarry Smith    SNESSetNGS - Sets the user nonlinear Gauss-Seidel routine for
2135c79ef259SPeter Brune    use with composed nonlinear solvers.
2136c79ef259SPeter Brune 
2137c79ef259SPeter Brune    Input Parameters:
2138c79ef259SPeter Brune +  snes   - the SNES context
2139be95d8f1SBarry Smith .  f - function evaluation routine to apply Gauss-Seidel see SNESNGSFunction
2140c79ef259SPeter Brune -  ctx    - [optional] user-defined context for private data for the
21410298fd71SBarry Smith             smoother evaluation routine (may be NULL)
2142c79ef259SPeter Brune 
2143c79ef259SPeter Brune    Notes:
2144be95d8f1SBarry Smith    The NGS routines are used by the composed nonlinear solver to generate
2145c79ef259SPeter Brune     a problem appropriate update to the solution, particularly FAS.
2146c79ef259SPeter Brune 
2147d28543b3SPeter Brune    Level: intermediate
2148c79ef259SPeter Brune 
2149be95d8f1SBarry Smith .seealso: SNESGetFunction(), SNESComputeNGS()
2150c79ef259SPeter Brune @*/
2151be95d8f1SBarry Smith PetscErrorCode SNESSetNGS(SNES snes,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx)
21526cab3a1bSJed Brown {
21536cab3a1bSJed Brown   PetscErrorCode ierr;
21546cab3a1bSJed Brown   DM             dm;
21556cab3a1bSJed Brown 
2156646217ecSPeter Brune   PetscFunctionBegin;
21576cab3a1bSJed Brown   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
21586cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2159be95d8f1SBarry Smith   ierr = DMSNESSetNGS(dm,f,ctx);CHKERRQ(ierr);
2160646217ecSPeter Brune   PetscFunctionReturn(0);
2161646217ecSPeter Brune }
2162646217ecSPeter Brune 
216325acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeFunction(SNES snes,Vec x,Vec f,void *ctx)
21648b0a5094SBarry Smith {
21658b0a5094SBarry Smith   PetscErrorCode ierr;
2166e03ab78fSPeter Brune   DM             dm;
2167942e3340SBarry Smith   DMSNES         sdm;
21686cab3a1bSJed Brown 
21698b0a5094SBarry Smith   PetscFunctionBegin;
2170e03ab78fSPeter Brune   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2171942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
217225acbd8eSLisandro Dalcin   if (!sdm->ops->computepfunction) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard function.");
217325acbd8eSLisandro Dalcin   if (!sdm->ops->computepjacobian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard Jacobian.");
21748b0a5094SBarry Smith   /*  A(x)*x - b(x) */
217525acbd8eSLisandro Dalcin   PetscStackPush("SNES Picard user function");
217622c6f798SBarry Smith   ierr = (*sdm->ops->computepfunction)(snes,x,f,sdm->pctx);CHKERRQ(ierr);
217725acbd8eSLisandro Dalcin   PetscStackPop;
217825acbd8eSLisandro Dalcin   PetscStackPush("SNES Picard user Jacobian");
2179d1e9a80fSBarry Smith   ierr = (*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx);CHKERRQ(ierr);
218025acbd8eSLisandro Dalcin   PetscStackPop;
21818b0a5094SBarry Smith   ierr = VecScale(f,-1.0);CHKERRQ(ierr);
218295eabcedSBarry Smith   ierr = MatMultAdd(snes->jacobian,x,f,f);CHKERRQ(ierr);
21838b0a5094SBarry Smith   PetscFunctionReturn(0);
21848b0a5094SBarry Smith }
21858b0a5094SBarry Smith 
218625acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeJacobian(SNES snes,Vec x1,Mat J,Mat B,void *ctx)
21878b0a5094SBarry Smith {
21888b0a5094SBarry Smith   PetscFunctionBegin;
2189e03ab78fSPeter Brune   /* the jacobian matrix should be pre-filled in SNESPicardComputeFunction */
21908b0a5094SBarry Smith   PetscFunctionReturn(0);
21918b0a5094SBarry Smith }
21928b0a5094SBarry Smith 
21938b0a5094SBarry Smith /*@C
21940d04baf8SBarry Smith    SNESSetPicard - Use SNES to solve the semilinear-system A(x) x = b(x) via a Picard type iteration (Picard linearization)
21958b0a5094SBarry Smith 
21968b0a5094SBarry Smith    Logically Collective on SNES
21978b0a5094SBarry Smith 
21988b0a5094SBarry Smith    Input Parameters:
21998b0a5094SBarry Smith +  snes - the SNES context
22008b0a5094SBarry Smith .  r - vector to store function value
2201f8b49ee9SBarry Smith .  b - function evaluation routine
2202e5d3d808SBarry Smith .  Amat - matrix with which A(x) x - b(x) is to be computed
2203e5d3d808SBarry Smith .  Pmat - matrix from which preconditioner is computed (usually the same as Amat)
2204411c0326SBarry Smith .  J  - function to compute matrix value, see SNESJacobianFunction for details on its calling sequence
22058b0a5094SBarry Smith -  ctx - [optional] user-defined context for private data for the
22060298fd71SBarry Smith          function evaluation routine (may be NULL)
22078b0a5094SBarry Smith 
22088b0a5094SBarry Smith    Notes:
2209f450aa47SBarry Smith     We do not recomemend using this routine. It is far better to provide the nonlinear function F() and some approximation to the Jacobian and use
2210f450aa47SBarry Smith     an approximate Newton solver. This interface is provided to allow porting/testing a previous Picard based code in PETSc before converting it to approximate Newton.
2211f450aa47SBarry Smith 
22128b0a5094SBarry Smith     One can call SNESSetPicard() or SNESSetFunction() (and possibly SNESSetJacobian()) but cannot call both
22138b0a5094SBarry Smith 
22148b0a5094SBarry Smith $     Solves the equation A(x) x = b(x) via the defect correction algorithm A(x^{n}) (x^{n+1} - x^{n}) = b(x^{n}) - A(x^{n})x^{n}
22158b0a5094SBarry Smith $     Note that when an exact solver is used this corresponds to the "classic" Picard A(x^{n}) x^{n+1} = b(x^{n}) iteration.
22168b0a5094SBarry Smith 
22178b0a5094SBarry Smith      Run with -snes_mf_operator to solve the system with Newton's method using A(x^{n}) to construct the preconditioner.
22188b0a5094SBarry Smith 
22190d04baf8SBarry Smith    We implement the defect correction form of the Picard iteration because it converges much more generally when inexact linear solvers are used then
22200d04baf8SBarry Smith    the direct Picard iteration A(x^n) x^{n+1} = b(x^n)
22218b0a5094SBarry Smith 
22228b0a5094SBarry Smith    There is some controversity over the definition of a Picard iteration for nonlinear systems but almost everyone agrees that it involves a linear solve and some
22238b0a5094SBarry Smith    believe it is the iteration  A(x^{n}) x^{n+1} = b(x^{n}) hence we use the name Picard. If anyone has an authoritative  reference that defines the Picard iteration
22248b0a5094SBarry Smith    different please contact us at petsc-dev@mcs.anl.gov and we'll have an entirely new argument :-).
22258b0a5094SBarry Smith 
2226f450aa47SBarry Smith    Level: intermediate
22278b0a5094SBarry Smith 
2228411c0326SBarry Smith .seealso: SNESGetFunction(), SNESSetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESGetPicard(), SNESLineSearchPreCheckPicard(), SNESJacobianFunction
22298b0a5094SBarry Smith @*/
2230d1e9a80fSBarry Smith PetscErrorCode  SNESSetPicard(SNES snes,Vec r,PetscErrorCode (*b)(SNES,Vec,Vec,void*),Mat Amat, Mat Pmat, PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx)
22318b0a5094SBarry Smith {
22328b0a5094SBarry Smith   PetscErrorCode ierr;
2233e03ab78fSPeter Brune   DM             dm;
2234e03ab78fSPeter Brune 
22358b0a5094SBarry Smith   PetscFunctionBegin;
22368b0a5094SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2237e03ab78fSPeter Brune   ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
2238f8b49ee9SBarry Smith   ierr = DMSNESSetPicard(dm,b,J,ctx);CHKERRQ(ierr);
22398b0a5094SBarry Smith   ierr = SNESSetFunction(snes,r,SNESPicardComputeFunction,ctx);CHKERRQ(ierr);
2240e5d3d808SBarry Smith   ierr = SNESSetJacobian(snes,Amat,Pmat,SNESPicardComputeJacobian,ctx);CHKERRQ(ierr);
22418b0a5094SBarry Smith   PetscFunctionReturn(0);
22428b0a5094SBarry Smith }
22438b0a5094SBarry Smith 
22447971a8bfSPeter Brune /*@C
22457971a8bfSPeter Brune    SNESGetPicard - Returns the context for the Picard iteration
22467971a8bfSPeter Brune 
22477971a8bfSPeter Brune    Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet.
22487971a8bfSPeter Brune 
22497971a8bfSPeter Brune    Input Parameter:
22507971a8bfSPeter Brune .  snes - the SNES context
22517971a8bfSPeter Brune 
22527971a8bfSPeter Brune    Output Parameter:
22530298fd71SBarry Smith +  r - the function (or NULL)
2254f8b49ee9SBarry Smith .  f - the function (or NULL); see SNESFunction for calling sequence details
2255e4357dc4SBarry Smith .  Amat - the matrix used to defined the operation A(x) x - b(x) (or NULL)
2256e4357dc4SBarry Smith .  Pmat  - the matrix from which the preconditioner will be constructed (or NULL)
2257f8b49ee9SBarry Smith .  J - the function for matrix evaluation (or NULL); see SNESJacobianFunction for calling sequence details
22580298fd71SBarry Smith -  ctx - the function context (or NULL)
22597971a8bfSPeter Brune 
22607971a8bfSPeter Brune    Level: advanced
22617971a8bfSPeter Brune 
2262e4357dc4SBarry Smith .seealso: SNESSetPicard(), SNESGetFunction(), SNESGetJacobian(), SNESGetDM(), SNESFunction, SNESJacobianFunction
22637971a8bfSPeter Brune @*/
2264d1e9a80fSBarry Smith PetscErrorCode  SNESGetPicard(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),Mat *Amat, Mat *Pmat, PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx)
22657971a8bfSPeter Brune {
22667971a8bfSPeter Brune   PetscErrorCode ierr;
22677971a8bfSPeter Brune   DM             dm;
22687971a8bfSPeter Brune 
22697971a8bfSPeter Brune   PetscFunctionBegin;
22707971a8bfSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
22710298fd71SBarry Smith   ierr = SNESGetFunction(snes,r,NULL,NULL);CHKERRQ(ierr);
2272e4357dc4SBarry Smith   ierr = SNESGetJacobian(snes,Amat,Pmat,NULL,NULL);CHKERRQ(ierr);
22737971a8bfSPeter Brune   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2274f8b49ee9SBarry Smith   ierr = DMSNESGetPicard(dm,f,J,ctx);CHKERRQ(ierr);
22757971a8bfSPeter Brune   PetscFunctionReturn(0);
22767971a8bfSPeter Brune }
22777971a8bfSPeter Brune 
2278d25893d9SBarry Smith /*@C
2279d25893d9SBarry Smith    SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem
2280d25893d9SBarry Smith 
2281d25893d9SBarry Smith    Logically Collective on SNES
2282d25893d9SBarry Smith 
2283d25893d9SBarry Smith    Input Parameters:
2284d25893d9SBarry Smith +  snes - the SNES context
2285d25893d9SBarry Smith .  func - function evaluation routine
2286d25893d9SBarry Smith -  ctx - [optional] user-defined context for private data for the
22870298fd71SBarry Smith          function evaluation routine (may be NULL)
2288d25893d9SBarry Smith 
2289d25893d9SBarry Smith    Calling sequence of func:
2290d25893d9SBarry Smith $    func (SNES snes,Vec x,void *ctx);
2291d25893d9SBarry Smith 
2292d25893d9SBarry Smith .  f - function vector
2293d25893d9SBarry Smith -  ctx - optional user-defined function context
2294d25893d9SBarry Smith 
2295d25893d9SBarry Smith    Level: intermediate
2296d25893d9SBarry Smith 
2297d25893d9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian()
2298d25893d9SBarry Smith @*/
2299d25893d9SBarry Smith PetscErrorCode  SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx)
2300d25893d9SBarry Smith {
2301d25893d9SBarry Smith   PetscFunctionBegin;
2302d25893d9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2303d25893d9SBarry Smith   if (func) snes->ops->computeinitialguess = func;
2304d25893d9SBarry Smith   if (ctx)  snes->initialguessP            = ctx;
2305d25893d9SBarry Smith   PetscFunctionReturn(0);
2306d25893d9SBarry Smith }
2307d25893d9SBarry Smith 
23083ab0aad5SBarry Smith /* --------------------------------------------------------------- */
23091096aae1SMatthew Knepley /*@C
23101096aae1SMatthew Knepley    SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set
23111096aae1SMatthew Knepley    it assumes a zero right hand side.
23121096aae1SMatthew Knepley 
23133f9fe445SBarry Smith    Logically Collective on SNES
23141096aae1SMatthew Knepley 
23151096aae1SMatthew Knepley    Input Parameter:
23161096aae1SMatthew Knepley .  snes - the SNES context
23171096aae1SMatthew Knepley 
23181096aae1SMatthew Knepley    Output Parameter:
23190298fd71SBarry Smith .  rhs - the right hand side vector or NULL if the right hand side vector is null
23201096aae1SMatthew Knepley 
23211096aae1SMatthew Knepley    Level: intermediate
23221096aae1SMatthew Knepley 
232385385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
23241096aae1SMatthew Knepley @*/
23257087cfbeSBarry Smith PetscErrorCode  SNESGetRhs(SNES snes,Vec *rhs)
23261096aae1SMatthew Knepley {
23271096aae1SMatthew Knepley   PetscFunctionBegin;
23280700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
23291096aae1SMatthew Knepley   PetscValidPointer(rhs,2);
233085385478SLisandro Dalcin   *rhs = snes->vec_rhs;
23311096aae1SMatthew Knepley   PetscFunctionReturn(0);
23321096aae1SMatthew Knepley }
23331096aae1SMatthew Knepley 
23349b94acceSBarry Smith /*@
2335bf388a1fSBarry Smith    SNESComputeFunction - Calls the function that has been set with SNESSetFunction().
23369b94acceSBarry Smith 
2337c7afd0dbSLois Curfman McInnes    Collective on SNES
2338c7afd0dbSLois Curfman McInnes 
23399b94acceSBarry Smith    Input Parameters:
2340c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2341c7afd0dbSLois Curfman McInnes -  x - input vector
23429b94acceSBarry Smith 
23439b94acceSBarry Smith    Output Parameter:
23443638b69dSLois Curfman McInnes .  y - function vector, as set by SNESSetFunction()
23459b94acceSBarry Smith 
23461bffabb2SLois Curfman McInnes    Notes:
234736851e7fSLois Curfman McInnes    SNESComputeFunction() is typically used within nonlinear solvers
234836851e7fSLois Curfman McInnes    implementations, so most users would not generally call this routine
234936851e7fSLois Curfman McInnes    themselves.
235036851e7fSLois Curfman McInnes 
235136851e7fSLois Curfman McInnes    Level: developer
235236851e7fSLois Curfman McInnes 
2353a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction()
23549b94acceSBarry Smith @*/
23557087cfbeSBarry Smith PetscErrorCode  SNESComputeFunction(SNES snes,Vec x,Vec y)
23569b94acceSBarry Smith {
2357dfbe8321SBarry Smith   PetscErrorCode ierr;
23586cab3a1bSJed Brown   DM             dm;
2359942e3340SBarry Smith   DMSNES         sdm;
23609b94acceSBarry Smith 
23613a40ed3dSBarry Smith   PetscFunctionBegin;
23620700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
23630700a824SBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
23640700a824SBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2365c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,x,2);
2366c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,3);
236762796dfbSBarry Smith   ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr);
2368184914b5SBarry Smith 
23696cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2370942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
237132f3f7c2SPeter Brune   if (sdm->ops->computefunction) {
237294db00ebSBarry Smith     if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) {
2373ccf3c845SPeter Brune       ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
237494db00ebSBarry Smith     }
23758860a134SJunchao Zhang     ierr = VecLockReadPush(x);CHKERRQ(ierr);
2376d64ed03dSBarry Smith     PetscStackPush("SNES user function");
23778ddeebeaSSteve Benbow     /* ensure domainerror is false prior to computefunction evaluation (may not have been reset) */
23788ddeebeaSSteve Benbow     snes->domainerror = PETSC_FALSE;
237922c6f798SBarry Smith     ierr = (*sdm->ops->computefunction)(snes,x,y,sdm->functionctx);CHKERRQ(ierr);
2380d64ed03dSBarry Smith     PetscStackPop;
23818860a134SJunchao Zhang     ierr = VecLockReadPop(x);CHKERRQ(ierr);
238294db00ebSBarry Smith     if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) {
2383ccf3c845SPeter Brune       ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
238494db00ebSBarry Smith     }
2385c90fad12SPeter Brune   } else if (snes->vec_rhs) {
2386c90fad12SPeter Brune     ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr);
2387644e2e5bSBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve().");
238885385478SLisandro Dalcin   if (snes->vec_rhs) {
238985385478SLisandro Dalcin     ierr = VecAXPY(y,-1.0,snes->vec_rhs);CHKERRQ(ierr);
23903ab0aad5SBarry Smith   }
2391ae3c334cSLois Curfman McInnes   snes->nfuncs++;
2392422a814eSBarry Smith   /*
2393422a814eSBarry Smith      domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will
2394422a814eSBarry Smith      propagate the value to all processes
2395422a814eSBarry Smith   */
2396422a814eSBarry Smith   if (snes->domainerror) {
2397422a814eSBarry Smith     ierr = VecSetInf(y);CHKERRQ(ierr);
2398422a814eSBarry Smith   }
23993a40ed3dSBarry Smith   PetscFunctionReturn(0);
24009b94acceSBarry Smith }
24019b94acceSBarry Smith 
2402c79ef259SPeter Brune /*@
2403be95d8f1SBarry Smith    SNESComputeNGS - Calls the Gauss-Seidel function that has been set with  SNESSetNGS().
2404c79ef259SPeter Brune 
2405c79ef259SPeter Brune    Collective on SNES
2406c79ef259SPeter Brune 
2407c79ef259SPeter Brune    Input Parameters:
2408c79ef259SPeter Brune +  snes - the SNES context
2409c79ef259SPeter Brune .  x - input vector
2410c79ef259SPeter Brune -  b - rhs vector
2411c79ef259SPeter Brune 
2412c79ef259SPeter Brune    Output Parameter:
2413c79ef259SPeter Brune .  x - new solution vector
2414c79ef259SPeter Brune 
2415c79ef259SPeter Brune    Notes:
2416be95d8f1SBarry Smith    SNESComputeNGS() is typically used within composed nonlinear solver
2417c79ef259SPeter Brune    implementations, so most users would not generally call this routine
2418c79ef259SPeter Brune    themselves.
2419c79ef259SPeter Brune 
2420c79ef259SPeter Brune    Level: developer
2421c79ef259SPeter Brune 
2422be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESComputeFunction()
2423c79ef259SPeter Brune @*/
2424be95d8f1SBarry Smith PetscErrorCode  SNESComputeNGS(SNES snes,Vec b,Vec x)
2425646217ecSPeter Brune {
2426646217ecSPeter Brune   PetscErrorCode ierr;
24276cab3a1bSJed Brown   DM             dm;
2428942e3340SBarry Smith   DMSNES         sdm;
2429646217ecSPeter Brune 
2430646217ecSPeter Brune   PetscFunctionBegin;
2431646217ecSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2432646217ecSPeter Brune   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2433646217ecSPeter Brune   if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,3);
2434646217ecSPeter Brune   PetscCheckSameComm(snes,1,x,2);
2435646217ecSPeter Brune   if (b) PetscCheckSameComm(snes,1,b,3);
243662796dfbSBarry Smith   if (b) {ierr = VecValidValues(b,2,PETSC_TRUE);CHKERRQ(ierr);}
2437be95d8f1SBarry Smith   ierr = PetscLogEventBegin(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr);
24386cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2439942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
244022c6f798SBarry Smith   if (sdm->ops->computegs) {
24418860a134SJunchao Zhang     if (b) {ierr = VecLockReadPush(b);CHKERRQ(ierr);}
2442be95d8f1SBarry Smith     PetscStackPush("SNES user NGS");
244322c6f798SBarry Smith     ierr = (*sdm->ops->computegs)(snes,x,b,sdm->gsctx);CHKERRQ(ierr);
2444646217ecSPeter Brune     PetscStackPop;
24458860a134SJunchao Zhang     if (b) {ierr = VecLockReadPop(b);CHKERRQ(ierr);}
2446be95d8f1SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetNGS() before SNESComputeNGS(), likely called from SNESSolve().");
2447be95d8f1SBarry Smith   ierr = PetscLogEventEnd(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr);
2448646217ecSPeter Brune   PetscFunctionReturn(0);
2449646217ecSPeter Brune }
2450646217ecSPeter Brune 
2451e885f1abSBarry Smith PetscErrorCode SNESTestJacobian(SNES snes)
2452e885f1abSBarry Smith {
245312837594SBarry Smith   Mat               A,B,C,D,jacobian;
2454e885f1abSBarry Smith   Vec               x = snes->vec_sol,f = snes->vec_func;
2455e885f1abSBarry Smith   PetscErrorCode    ierr;
2456e885f1abSBarry Smith   PetscReal         nrm,gnorm;
245781e7118cSBarry Smith   PetscReal         threshold = 1.e-5;
24580e276705SLisandro Dalcin   MatType           mattype;
2459e885f1abSBarry Smith   PetscInt          m,n,M,N;
2460e885f1abSBarry Smith   void              *functx;
24612cd624f9SStefano Zampini   PetscBool         complete_print = PETSC_FALSE,threshold_print = PETSC_FALSE,test = PETSC_FALSE,flg,istranspose;
24623325ff46SBarry Smith   PetscViewer       viewer,mviewer;
2463e885f1abSBarry Smith   MPI_Comm          comm;
2464e885f1abSBarry Smith   PetscInt          tabs;
246512837594SBarry Smith   static PetscBool  directionsprinted = PETSC_FALSE;
24663325ff46SBarry Smith   PetscViewerFormat format;
2467e885f1abSBarry Smith 
2468e885f1abSBarry Smith   PetscFunctionBegin;
2469fc35ed60SBarry Smith   ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr);
247012837594SBarry Smith   ierr = PetscOptionsName("-snes_test_jacobian","Compare hand-coded and finite difference Jacobians","None",&test);CHKERRQ(ierr);
247112837594SBarry Smith   ierr = PetscOptionsReal("-snes_test_jacobian", "Threshold for element difference between hand-coded and finite difference being meaningful", "None", threshold, &threshold,NULL);CHKERRQ(ierr);
24723325ff46SBarry Smith   ierr = PetscOptionsViewer("-snes_test_jacobian_view","View difference between hand-coded and finite difference Jacobians element entries","None",&mviewer,&format,&complete_print);CHKERRQ(ierr);
247318d89885SKarl Rupp   if (!complete_print) {
2474455a5933SJed Brown     ierr = PetscOptionsDeprecated("-snes_test_jacobian_display","-snes_test_jacobian_view","3.13",NULL);CHKERRQ(ierr);
247518d89885SKarl Rupp     ierr = PetscOptionsViewer("-snes_test_jacobian_display","Display difference between hand-coded and finite difference Jacobians","None",&mviewer,&format,&complete_print);CHKERRQ(ierr);
247618d89885SKarl Rupp   }
247718d89885SKarl Rupp   /* for compatibility with PETSc 3.9 and older. */
2478455a5933SJed Brown   ierr = PetscOptionsDeprecated("-snes_test_jacobian_display_threshold","-snes_test_jacobian","3.13","-snes_test_jacobian accepts an optional threshold (since v3.10)");CHKERRQ(ierr);
247918d89885SKarl Rupp   ierr = PetscOptionsReal("-snes_test_jacobian_display_threshold", "Display difference between hand-coded and finite difference Jacobians which exceed input threshold", "None", threshold, &threshold, &threshold_print);CHKERRQ(ierr);
2480e885f1abSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
2481e885f1abSBarry Smith   if (!test) PetscFunctionReturn(0);
2482e885f1abSBarry Smith 
2483e885f1abSBarry Smith   ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr);
2484e885f1abSBarry Smith   ierr = PetscViewerASCIIGetStdout(comm,&viewer);CHKERRQ(ierr);
2485e885f1abSBarry Smith   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
2486e885f1abSBarry Smith   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)snes)->tablevel);CHKERRQ(ierr);
248712837594SBarry Smith   ierr = PetscViewerASCIIPrintf(viewer,"  ---------- Testing Jacobian -------------\n");CHKERRQ(ierr);
248812837594SBarry Smith   if (!complete_print && !directionsprinted) {
248912837594SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Run with -snes_test_jacobian_view and optionally -snes_test_jacobian <threshold> to show difference\n");CHKERRQ(ierr);
2490fc35ed60SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"    of hand-coded and finite difference Jacobian entries greater than <threshold>.\n");CHKERRQ(ierr);
249112837594SBarry Smith   }
249212837594SBarry Smith   if (!directionsprinted) {
249312837594SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Testing hand-coded Jacobian, if (for double precision runs) ||J - Jfd||_F/||J||_F is\n");CHKERRQ(ierr);
2494e885f1abSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"    O(1.e-8), the hand-coded Jacobian is probably correct.\n");CHKERRQ(ierr);
249512837594SBarry Smith     directionsprinted = PETSC_TRUE;
2496e885f1abSBarry Smith   }
24973325ff46SBarry Smith   if (complete_print) {
24983325ff46SBarry Smith     ierr = PetscViewerPushFormat(mviewer,format);CHKERRQ(ierr);
2499e885f1abSBarry Smith   }
2500e885f1abSBarry Smith 
250112837594SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)snes->jacobian,MATMFFD,&flg);CHKERRQ(ierr);
250212837594SBarry Smith   if (!flg) jacobian = snes->jacobian;
250312837594SBarry Smith   else jacobian = snes->jacobian_pre;
250412837594SBarry Smith 
2505a82339d0SMatthew G. Knepley   if (!x) {
2506a82339d0SMatthew G. Knepley     ierr = MatCreateVecs(jacobian, &x, NULL);CHKERRQ(ierr);
2507a82339d0SMatthew G. Knepley   } else {
2508a82339d0SMatthew G. Knepley     ierr = PetscObjectReference((PetscObject) x);CHKERRQ(ierr);
2509a82339d0SMatthew G. Knepley   }
2510a82339d0SMatthew G. Knepley   if (!f) {
2511a82339d0SMatthew G. Knepley     ierr = VecDuplicate(x, &f);CHKERRQ(ierr);
2512a82339d0SMatthew G. Knepley   } else {
2513a82339d0SMatthew G. Knepley     ierr = PetscObjectReference((PetscObject) f);CHKERRQ(ierr);
2514a82339d0SMatthew G. Knepley   }
2515a82339d0SMatthew G. Knepley   /* evaluate the function at this point because SNESComputeJacobianDefault() assumes that the function has been evaluated and put into snes->vec_func */
2516a82339d0SMatthew G. Knepley   ierr = SNESComputeFunction(snes,x,f);CHKERRQ(ierr);
2517a82339d0SMatthew G. Knepley   ierr = VecDestroy(&f);CHKERRQ(ierr);
25182cd624f9SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)snes,SNESKSPTRANSPOSEONLY,&istranspose);CHKERRQ(ierr);
251912837594SBarry Smith   while (jacobian) {
25202cd624f9SStefano Zampini     Mat JT = NULL, Jsave = NULL;
25212cd624f9SStefano Zampini 
25222cd624f9SStefano Zampini     if (istranspose) {
25232cd624f9SStefano Zampini       ierr = MatCreateTranspose(jacobian,&JT);CHKERRQ(ierr);
25242cd624f9SStefano Zampini       Jsave = jacobian;
25252cd624f9SStefano Zampini       jacobian = JT;
25262cd624f9SStefano Zampini     }
25270e276705SLisandro Dalcin     ierr = PetscObjectBaseTypeCompareAny((PetscObject)jacobian,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPISBAIJ,"");CHKERRQ(ierr);
252812837594SBarry Smith     if (flg) {
252912837594SBarry Smith       A    = jacobian;
253012837594SBarry Smith       ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
253112837594SBarry Smith     } else {
25320bacdadaSStefano Zampini       ierr = MatComputeOperator(jacobian,MATAIJ,&A);CHKERRQ(ierr);
253312837594SBarry Smith     }
2534e885f1abSBarry Smith 
25350e276705SLisandro Dalcin     ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
2536e885f1abSBarry Smith     ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
2537e885f1abSBarry Smith     ierr = MatGetLocalSize(A,&m,&n);CHKERRQ(ierr);
25380e276705SLisandro Dalcin     ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
25390e276705SLisandro Dalcin     ierr = MatSetType(B,mattype);CHKERRQ(ierr);
2540e885f1abSBarry Smith     ierr = MatSetSizes(B,m,n,M,N);CHKERRQ(ierr);
25410e276705SLisandro Dalcin     ierr = MatSetBlockSizesFromMats(B,A,A);CHKERRQ(ierr);
2542e885f1abSBarry Smith     ierr = MatSetUp(B);CHKERRQ(ierr);
2543e885f1abSBarry Smith     ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr);
2544e885f1abSBarry Smith 
2545e885f1abSBarry Smith     ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr);
2546e885f1abSBarry Smith     ierr = SNESComputeJacobianDefault(snes,x,B,B,functx);CHKERRQ(ierr);
254712837594SBarry Smith 
254812837594SBarry Smith     ierr = MatDuplicate(B,MAT_COPY_VALUES,&D);CHKERRQ(ierr);
254912837594SBarry Smith     ierr = MatAYPX(D,-1.0,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
255012837594SBarry Smith     ierr = MatNorm(D,NORM_FROBENIUS,&nrm);CHKERRQ(ierr);
2551e885f1abSBarry Smith     ierr = MatNorm(A,NORM_FROBENIUS,&gnorm);CHKERRQ(ierr);
255212837594SBarry Smith     ierr = MatDestroy(&D);CHKERRQ(ierr);
255312837594SBarry Smith     if (!gnorm) gnorm = 1; /* just in case */
255412837594SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  ||J - Jfd||_F/||J||_F = %g, ||J - Jfd||_F = %g\n",(double)(nrm/gnorm),(double)nrm);CHKERRQ(ierr);
255512837594SBarry Smith 
2556e885f1abSBarry Smith     if (complete_print) {
255712837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Hand-coded Jacobian ----------\n");CHKERRQ(ierr);
25581878987eSStefano Zampini       ierr = MatView(A,mviewer);CHKERRQ(ierr);
255912837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Finite difference Jacobian ----------\n");CHKERRQ(ierr);
25603325ff46SBarry Smith       ierr = MatView(B,mviewer);CHKERRQ(ierr);
2561e885f1abSBarry Smith     }
2562e885f1abSBarry Smith 
2563df10fb39SFande Kong     if (threshold_print || complete_print) {
2564e885f1abSBarry Smith       PetscInt          Istart, Iend, *ccols, bncols, cncols, j, row;
2565e885f1abSBarry Smith       PetscScalar       *cvals;
2566e885f1abSBarry Smith       const PetscInt    *bcols;
2567e885f1abSBarry Smith       const PetscScalar *bvals;
2568e885f1abSBarry Smith 
2569e885f1abSBarry Smith       ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
25700e276705SLisandro Dalcin       ierr = MatSetType(C,mattype);CHKERRQ(ierr);
2571e885f1abSBarry Smith       ierr = MatSetSizes(C,m,n,M,N);CHKERRQ(ierr);
25720e276705SLisandro Dalcin       ierr = MatSetBlockSizesFromMats(C,A,A);CHKERRQ(ierr);
2573e885f1abSBarry Smith       ierr = MatSetUp(C);CHKERRQ(ierr);
2574e885f1abSBarry Smith       ierr = MatSetOption(C,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr);
25750e276705SLisandro Dalcin 
25760e276705SLisandro Dalcin       ierr = MatAYPX(B,-1.0,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
2577e885f1abSBarry Smith       ierr = MatGetOwnershipRange(B,&Istart,&Iend);CHKERRQ(ierr);
2578e885f1abSBarry Smith 
2579e885f1abSBarry Smith       for (row = Istart; row < Iend; row++) {
2580e885f1abSBarry Smith         ierr = MatGetRow(B,row,&bncols,&bcols,&bvals);CHKERRQ(ierr);
2581e885f1abSBarry Smith         ierr = PetscMalloc2(bncols,&ccols,bncols,&cvals);CHKERRQ(ierr);
2582e885f1abSBarry Smith         for (j = 0, cncols = 0; j < bncols; j++) {
258323a52b1dSBarry Smith           if (PetscAbsScalar(bvals[j]) > threshold) {
2584e885f1abSBarry Smith             ccols[cncols] = bcols[j];
2585e885f1abSBarry Smith             cvals[cncols] = bvals[j];
2586e885f1abSBarry Smith             cncols += 1;
2587e885f1abSBarry Smith           }
2588e885f1abSBarry Smith         }
2589e885f1abSBarry Smith         if (cncols) {
2590e885f1abSBarry Smith           ierr = MatSetValues(C,1,&row,cncols,ccols,cvals,INSERT_VALUES);CHKERRQ(ierr);
2591e885f1abSBarry Smith         }
2592e885f1abSBarry Smith         ierr = MatRestoreRow(B,row,&bncols,&bcols,&bvals);CHKERRQ(ierr);
2593e885f1abSBarry Smith         ierr = PetscFree2(ccols,cvals);CHKERRQ(ierr);
2594e885f1abSBarry Smith       }
2595e885f1abSBarry Smith       ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2596e885f1abSBarry Smith       ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
259712837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Hand-coded minus finite-difference Jacobian with tolerance %g ----------\n",(double)threshold);CHKERRQ(ierr);
259818d89885SKarl Rupp       ierr = MatView(C,complete_print ? mviewer : viewer);CHKERRQ(ierr);
2599e885f1abSBarry Smith       ierr = MatDestroy(&C);CHKERRQ(ierr);
2600e885f1abSBarry Smith     }
260112837594SBarry Smith     ierr = MatDestroy(&A);CHKERRQ(ierr);
2602e885f1abSBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
26032cd624f9SStefano Zampini     ierr = MatDestroy(&JT);CHKERRQ(ierr);
26042cd624f9SStefano Zampini     if (Jsave) jacobian = Jsave;
260512837594SBarry Smith     if (jacobian != snes->jacobian_pre) {
260612837594SBarry Smith       jacobian = snes->jacobian_pre;
260712837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  ---------- Testing Jacobian for preconditioner -------------\n");CHKERRQ(ierr);
260812837594SBarry Smith     }
260912837594SBarry Smith     else jacobian = NULL;
261012837594SBarry Smith   }
2611a82339d0SMatthew G. Knepley   ierr = VecDestroy(&x);CHKERRQ(ierr);
26123325ff46SBarry Smith   if (complete_print) {
26133325ff46SBarry Smith     ierr = PetscViewerPopFormat(mviewer);CHKERRQ(ierr);
26143325ff46SBarry Smith   }
2615a0c90127SFande Kong   if (mviewer) { ierr = PetscViewerDestroy(&mviewer);CHKERRQ(ierr); }
2616e885f1abSBarry Smith   ierr = PetscViewerASCIISetTab(viewer,tabs);CHKERRQ(ierr);
2617e885f1abSBarry Smith   PetscFunctionReturn(0);
2618e885f1abSBarry Smith }
2619e885f1abSBarry Smith 
262062fef451SLois Curfman McInnes /*@
2621bf388a1fSBarry Smith    SNESComputeJacobian - Computes the Jacobian matrix that has been set with SNESSetJacobian().
262262fef451SLois Curfman McInnes 
2623d083f849SBarry Smith    Collective on SNES
2624c7afd0dbSLois Curfman McInnes 
262562fef451SLois Curfman McInnes    Input Parameters:
2626c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2627c7afd0dbSLois Curfman McInnes -  x - input vector
262862fef451SLois Curfman McInnes 
262962fef451SLois Curfman McInnes    Output Parameters:
2630c7afd0dbSLois Curfman McInnes +  A - Jacobian matrix
2631d1e9a80fSBarry Smith -  B - optional preconditioning matrix
2632fee21e36SBarry Smith 
2633e35cf81dSBarry Smith   Options Database Keys:
2634e35cf81dSBarry Smith +    -snes_lag_preconditioner <lag>
2635693365a8SJed Brown .    -snes_lag_jacobian <lag>
2636455a5933SJed Brown .    -snes_test_jacobian <optional threshold> - compare the user provided Jacobian with one compute via finite differences to check for errors.  If a threshold is given, display only those entries whose difference is greater than the threshold.
2637455a5933SJed Brown .    -snes_test_jacobian_view - display the user provided Jacobian, the finite difference Jacobian and the difference between them to help users detect the location of errors in the user provided Jacobian
2638693365a8SJed Brown .    -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences
2639693365a8SJed Brown .    -snes_compare_explicit_draw  - Compare the computed Jacobian to the finite difference Jacobian and draw the result
2640693365a8SJed Brown .    -snes_compare_explicit_contour  - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result
26414c30e9fbSJed Brown .    -snes_compare_operator  - Make the comparison options above use the operator instead of the preconditioning matrix
264294d6a431SBarry Smith .    -snes_compare_coloring - Compute the finite difference Jacobian using coloring and display norms of difference
2643c01495d3SJed Brown .    -snes_compare_coloring_display - Compute the finite differece Jacobian using coloring and display verbose differences
2644c01495d3SJed Brown .    -snes_compare_coloring_threshold - Display only those matrix entries that differ by more than a given threshold
2645c01495d3SJed Brown .    -snes_compare_coloring_threshold_atol - Absolute tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold
2646c01495d3SJed Brown .    -snes_compare_coloring_threshold_rtol - Relative tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold
26474c30e9fbSJed Brown .    -snes_compare_coloring_draw - Compute the finite differece Jacobian using coloring and draw differences
2648c01495d3SJed Brown -    -snes_compare_coloring_draw_contour - Compute the finite differece Jacobian using coloring and show contours of matrices and differences
2649c01495d3SJed Brown 
2650e35cf81dSBarry Smith 
265162fef451SLois Curfman McInnes    Notes:
265262fef451SLois Curfman McInnes    Most users should not need to explicitly call this routine, as it
265362fef451SLois Curfman McInnes    is used internally within the nonlinear solvers.
265462fef451SLois Curfman McInnes 
265595452b02SPatrick Sanan    Developer Notes:
265695452b02SPatrick Sanan     This has duplicative ways of checking the accuracy of the user provided Jacobian (see the options above). This is for historical reasons, the routine SNESTestJacobian() use to used
2657e885f1abSBarry Smith       for with the SNESType of test that has been removed.
2658e885f1abSBarry Smith 
265936851e7fSLois Curfman McInnes    Level: developer
266036851e7fSLois Curfman McInnes 
2661e35cf81dSBarry Smith .seealso:  SNESSetJacobian(), KSPSetOperators(), MatStructure, SNESSetLagPreconditioner(), SNESSetLagJacobian()
266262fef451SLois Curfman McInnes @*/
2663d1e9a80fSBarry Smith PetscErrorCode  SNESComputeJacobian(SNES snes,Vec X,Mat A,Mat B)
26649b94acceSBarry Smith {
2665dfbe8321SBarry Smith   PetscErrorCode ierr;
2666ace3abfcSBarry Smith   PetscBool      flag;
26676cab3a1bSJed Brown   DM             dm;
2668942e3340SBarry Smith   DMSNES         sdm;
2669e0e3a89bSBarry Smith   KSP            ksp;
26703a40ed3dSBarry Smith 
26713a40ed3dSBarry Smith   PetscFunctionBegin;
26720700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
26730700a824SBarry Smith   PetscValidHeaderSpecific(X,VEC_CLASSID,2);
2674c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,X,2);
267562796dfbSBarry Smith   ierr = VecValidValues(X,2,PETSC_TRUE);CHKERRQ(ierr);
26766cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2677942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
26783232da50SPeter Brune 
2679ce94432eSBarry Smith   if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Must call SNESSetJacobian(), DMSNESSetJacobian(), DMDASNESSetJacobianLocal(), etc");
2680ebd3b9afSBarry Smith 
2681ebd3b9afSBarry Smith   /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */
2682ebd3b9afSBarry Smith 
2683fe3ffe1eSBarry Smith   if (snes->lagjacobian == -2) {
2684fe3ffe1eSBarry Smith     snes->lagjacobian = -1;
2685f5af7f23SKarl Rupp 
2686fe3ffe1eSBarry Smith     ierr = PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n");CHKERRQ(ierr);
2687fe3ffe1eSBarry Smith   } else if (snes->lagjacobian == -1) {
2688e35cf81dSBarry Smith     ierr = PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n");CHKERRQ(ierr);
268994ab13aaSBarry Smith     ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr);
2690ebd3b9afSBarry Smith     if (flag) {
269194ab13aaSBarry Smith       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
269294ab13aaSBarry Smith       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2693ebd3b9afSBarry Smith     }
2694e35cf81dSBarry Smith     PetscFunctionReturn(0);
269537ec4e1aSPeter Brune   } else if (snes->lagjacobian > 1 && (snes->iter + snes->jac_iter) % snes->lagjacobian) {
2696e35cf81dSBarry Smith     ierr = PetscInfo2(snes,"Reusing Jacobian/preconditioner because lag is %D and SNES iteration is %D\n",snes->lagjacobian,snes->iter);CHKERRQ(ierr);
269794ab13aaSBarry Smith     ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr);
2698ebd3b9afSBarry Smith     if (flag) {
269994ab13aaSBarry Smith       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
270094ab13aaSBarry Smith       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2701ebd3b9afSBarry Smith     }
2702e35cf81dSBarry Smith     PetscFunctionReturn(0);
2703e35cf81dSBarry Smith   }
2704efd4aadfSBarry Smith   if (snes->npc && snes->npcside== PC_LEFT) {
270594ab13aaSBarry Smith     ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
270694ab13aaSBarry Smith     ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2707d728fb7dSPeter Brune     PetscFunctionReturn(0);
2708d728fb7dSPeter Brune   }
2709e35cf81dSBarry Smith 
271094ab13aaSBarry Smith   ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr);
27118860a134SJunchao Zhang   ierr = VecLockReadPush(X);CHKERRQ(ierr);
2712d64ed03dSBarry Smith   PetscStackPush("SNES user Jacobian function");
2713d1e9a80fSBarry Smith   ierr = (*sdm->ops->computejacobian)(snes,X,A,B,sdm->jacobianctx);CHKERRQ(ierr);
2714d64ed03dSBarry Smith   PetscStackPop;
27158860a134SJunchao Zhang   ierr = VecLockReadPop(X);CHKERRQ(ierr);
271694ab13aaSBarry Smith   ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr);
271728d58a37SPierre Jolivet 
271828d58a37SPierre Jolivet   /* attach latest linearization point to the preconditioning matrix */
271928d58a37SPierre Jolivet   ierr = PetscObjectCompose((PetscObject)B,"__SNES_latest_X",(PetscObject)X);CHKERRQ(ierr);
2720a8054027SBarry Smith 
2721e0e3a89bSBarry Smith   /* the next line ensures that snes->ksp exists */
2722e0e3a89bSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
27233b4f5425SBarry Smith   if (snes->lagpreconditioner == -2) {
27243b4f5425SBarry Smith     ierr = PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n");CHKERRQ(ierr);
2725d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr);
27263b4f5425SBarry Smith     snes->lagpreconditioner = -1;
27273b4f5425SBarry Smith   } else if (snes->lagpreconditioner == -1) {
2728a8054027SBarry Smith     ierr = PetscInfo(snes,"Reusing preconditioner because lag is -1\n");CHKERRQ(ierr);
2729d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr);
273037ec4e1aSPeter Brune   } else if (snes->lagpreconditioner > 1 && (snes->iter + snes->pre_iter) % snes->lagpreconditioner) {
2731a8054027SBarry Smith     ierr = PetscInfo2(snes,"Reusing preconditioner because lag is %D and SNES iteration is %D\n",snes->lagpreconditioner,snes->iter);CHKERRQ(ierr);
2732d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr);
2733d1e9a80fSBarry Smith   } else {
2734d1e9a80fSBarry Smith     ierr = PetscInfo(snes,"Rebuilding preconditioner\n");CHKERRQ(ierr);
2735d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr);
2736a8054027SBarry Smith   }
2737a8054027SBarry Smith 
2738e885f1abSBarry Smith   ierr = SNESTestJacobian(snes);CHKERRQ(ierr);
27396d84be18SBarry Smith   /* make sure user returned a correct Jacobian and preconditioner */
274094ab13aaSBarry Smith   /* PetscValidHeaderSpecific(A,MAT_CLASSID,3);
274194ab13aaSBarry Smith     PetscValidHeaderSpecific(B,MAT_CLASSID,4);   */
2742693365a8SJed Brown   {
2743693365a8SJed Brown     PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE;
274416413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit",NULL,NULL,&flag);CHKERRQ(ierr);
274516413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr);
274616413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr);
274716413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_operator",NULL,NULL,&flag_operator);CHKERRQ(ierr);
2748693365a8SJed Brown     if (flag || flag_draw || flag_contour) {
27490298fd71SBarry Smith       Mat          Bexp_mine = NULL,Bexp,FDexp;
2750693365a8SJed Brown       PetscViewer  vdraw,vstdout;
27516b3a5b13SJed Brown       PetscBool    flg;
2752693365a8SJed Brown       if (flag_operator) {
27530bacdadaSStefano Zampini         ierr = MatComputeOperator(A,MATAIJ,&Bexp_mine);CHKERRQ(ierr);
2754693365a8SJed Brown         Bexp = Bexp_mine;
2755693365a8SJed Brown       } else {
2756693365a8SJed Brown         /* See if the preconditioning matrix can be viewed and added directly */
2757b9e7e5c1SBarry Smith         ierr = PetscObjectBaseTypeCompareAny((PetscObject)B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr);
275894ab13aaSBarry Smith         if (flg) Bexp = B;
2759693365a8SJed Brown         else {
2760693365a8SJed Brown           /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */
27610bacdadaSStefano Zampini           ierr = MatComputeOperator(B,MATAIJ,&Bexp_mine);CHKERRQ(ierr);
2762693365a8SJed Brown           Bexp = Bexp_mine;
2763693365a8SJed Brown         }
2764693365a8SJed Brown       }
2765693365a8SJed Brown       ierr = MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp);CHKERRQ(ierr);
2766d1e9a80fSBarry Smith       ierr = SNESComputeJacobianDefault(snes,X,FDexp,FDexp,NULL);CHKERRQ(ierr);
2767ce94432eSBarry Smith       ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr);
2768693365a8SJed Brown       if (flag_draw || flag_contour) {
27699e5d0892SLisandro Dalcin         ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
2770693365a8SJed Brown         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
27710298fd71SBarry Smith       } else vdraw = NULL;
2772693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator ? "Jacobian" : "preconditioning Jacobian");CHKERRQ(ierr);
2773693365a8SJed Brown       if (flag) {ierr = MatView(Bexp,vstdout);CHKERRQ(ierr);}
2774693365a8SJed Brown       if (vdraw) {ierr = MatView(Bexp,vdraw);CHKERRQ(ierr);}
2775693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n");CHKERRQ(ierr);
2776693365a8SJed Brown       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
2777693365a8SJed Brown       if (vdraw) {ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);}
2778693365a8SJed Brown       ierr = MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
2779693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n");CHKERRQ(ierr);
2780693365a8SJed Brown       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
2781693365a8SJed Brown       if (vdraw) {              /* Always use contour for the difference */
2782693365a8SJed Brown         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
2783693365a8SJed Brown         ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);
2784693365a8SJed Brown         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
2785693365a8SJed Brown       }
2786693365a8SJed Brown       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
2787693365a8SJed Brown       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
2788693365a8SJed Brown       ierr = MatDestroy(&Bexp_mine);CHKERRQ(ierr);
2789693365a8SJed Brown       ierr = MatDestroy(&FDexp);CHKERRQ(ierr);
2790693365a8SJed Brown     }
2791693365a8SJed Brown   }
27924c30e9fbSJed Brown   {
27936719d8e4SJed Brown     PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE;
27946719d8e4SJed Brown     PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON;
279516413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring",NULL,NULL,&flag);CHKERRQ(ierr);
279616413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_display",NULL,NULL,&flag_display);CHKERRQ(ierr);
279716413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr);
279816413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr);
279916413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",NULL,NULL,&flag_threshold);CHKERRQ(ierr);
280027b0f280SBarry Smith     if (flag_threshold) {
2801c5929fdfSBarry Smith       ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,NULL);CHKERRQ(ierr);
2802c5929fdfSBarry Smith       ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,NULL);CHKERRQ(ierr);
280327b0f280SBarry Smith     }
28046719d8e4SJed Brown     if (flag || flag_display || flag_draw || flag_contour || flag_threshold) {
28054c30e9fbSJed Brown       Mat            Bfd;
28064c30e9fbSJed Brown       PetscViewer    vdraw,vstdout;
2807335efc43SPeter Brune       MatColoring    coloring;
28084c30e9fbSJed Brown       ISColoring     iscoloring;
28094c30e9fbSJed Brown       MatFDColoring  matfdcoloring;
28104c30e9fbSJed Brown       PetscErrorCode (*func)(SNES,Vec,Vec,void*);
28114c30e9fbSJed Brown       void           *funcctx;
28126719d8e4SJed Brown       PetscReal      norm1,norm2,normmax;
28134c30e9fbSJed Brown 
281494ab13aaSBarry Smith       ierr = MatDuplicate(B,MAT_DO_NOT_COPY_VALUES,&Bfd);CHKERRQ(ierr);
2815335efc43SPeter Brune       ierr = MatColoringCreate(Bfd,&coloring);CHKERRQ(ierr);
2816335efc43SPeter Brune       ierr = MatColoringSetType(coloring,MATCOLORINGSL);CHKERRQ(ierr);
2817335efc43SPeter Brune       ierr = MatColoringSetFromOptions(coloring);CHKERRQ(ierr);
2818335efc43SPeter Brune       ierr = MatColoringApply(coloring,&iscoloring);CHKERRQ(ierr);
2819335efc43SPeter Brune       ierr = MatColoringDestroy(&coloring);CHKERRQ(ierr);
28204c30e9fbSJed Brown       ierr = MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring);CHKERRQ(ierr);
2821f86b9fbaSHong Zhang       ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr);
2822f86b9fbaSHong Zhang       ierr = MatFDColoringSetUp(Bfd,iscoloring,matfdcoloring);CHKERRQ(ierr);
28234c30e9fbSJed Brown       ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr);
28244c30e9fbSJed Brown 
28254c30e9fbSJed Brown       /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */
28260298fd71SBarry Smith       ierr = SNESGetFunction(snes,NULL,&func,&funcctx);CHKERRQ(ierr);
28274c30e9fbSJed Brown       ierr = MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode (*)(void))func,funcctx);CHKERRQ(ierr);
28284c30e9fbSJed Brown       ierr = PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix);CHKERRQ(ierr);
28294c30e9fbSJed Brown       ierr = PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_");CHKERRQ(ierr);
28304c30e9fbSJed Brown       ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr);
2831d1e9a80fSBarry Smith       ierr = MatFDColoringApply(Bfd,matfdcoloring,X,snes);CHKERRQ(ierr);
28324c30e9fbSJed Brown       ierr = MatFDColoringDestroy(&matfdcoloring);CHKERRQ(ierr);
28334c30e9fbSJed Brown 
2834ce94432eSBarry Smith       ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr);
28354c30e9fbSJed Brown       if (flag_draw || flag_contour) {
28369e5d0892SLisandro Dalcin         ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
28374c30e9fbSJed Brown         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
28380298fd71SBarry Smith       } else vdraw = NULL;
28394c30e9fbSJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n");CHKERRQ(ierr);
284094ab13aaSBarry Smith       if (flag_display) {ierr = MatView(B,vstdout);CHKERRQ(ierr);}
284194ab13aaSBarry Smith       if (vdraw) {ierr = MatView(B,vdraw);CHKERRQ(ierr);}
28424c30e9fbSJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n");CHKERRQ(ierr);
28436719d8e4SJed Brown       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
28444c30e9fbSJed Brown       if (vdraw) {ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);}
284594ab13aaSBarry Smith       ierr = MatAYPX(Bfd,-1.0,B,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
28464c30e9fbSJed Brown       ierr = MatNorm(Bfd,NORM_1,&norm1);CHKERRQ(ierr);
28476719d8e4SJed Brown       ierr = MatNorm(Bfd,NORM_FROBENIUS,&norm2);CHKERRQ(ierr);
28484c30e9fbSJed Brown       ierr = MatNorm(Bfd,NORM_MAX,&normmax);CHKERRQ(ierr);
284957622a8eSBarry Smith       ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian, norm1=%g normFrob=%g normmax=%g\n",(double)norm1,(double)norm2,(double)normmax);CHKERRQ(ierr);
28506719d8e4SJed Brown       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
28514c30e9fbSJed Brown       if (vdraw) {              /* Always use contour for the difference */
28524c30e9fbSJed Brown         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
28534c30e9fbSJed Brown         ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);
28544c30e9fbSJed Brown         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
28554c30e9fbSJed Brown       }
28564c30e9fbSJed Brown       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
28576719d8e4SJed Brown 
28586719d8e4SJed Brown       if (flag_threshold) {
28596719d8e4SJed Brown         PetscInt bs,rstart,rend,i;
286094ab13aaSBarry Smith         ierr = MatGetBlockSize(B,&bs);CHKERRQ(ierr);
286194ab13aaSBarry Smith         ierr = MatGetOwnershipRange(B,&rstart,&rend);CHKERRQ(ierr);
28626719d8e4SJed Brown         for (i=rstart; i<rend; i++) {
28636719d8e4SJed Brown           const PetscScalar *ba,*ca;
28646719d8e4SJed Brown           const PetscInt    *bj,*cj;
28656719d8e4SJed Brown           PetscInt          bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1;
28666719d8e4SJed Brown           PetscReal         maxentry = 0,maxdiff = 0,maxrdiff = 0;
286794ab13aaSBarry Smith           ierr = MatGetRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr);
28686719d8e4SJed Brown           ierr = MatGetRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
286994ab13aaSBarry Smith           if (bn != cn) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold");
28706719d8e4SJed Brown           for (j=0; j<bn; j++) {
28716719d8e4SJed Brown             PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
28726719d8e4SJed Brown             if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) {
28736719d8e4SJed Brown               maxentrycol = bj[j];
28746719d8e4SJed Brown               maxentry    = PetscRealPart(ba[j]);
28756719d8e4SJed Brown             }
28766719d8e4SJed Brown             if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) {
28776719d8e4SJed Brown               maxdiffcol = bj[j];
28786719d8e4SJed Brown               maxdiff    = PetscRealPart(ca[j]);
28796719d8e4SJed Brown             }
28806719d8e4SJed Brown             if (rdiff > maxrdiff) {
28816719d8e4SJed Brown               maxrdiffcol = bj[j];
28826719d8e4SJed Brown               maxrdiff    = rdiff;
28836719d8e4SJed Brown             }
28846719d8e4SJed Brown           }
28856719d8e4SJed Brown           if (maxrdiff > 1) {
288657622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(vstdout,"row %D (maxentry=%g at %D, maxdiff=%g at %D, maxrdiff=%g at %D):",i,(double)maxentry,maxentrycol,(double)maxdiff,maxdiffcol,(double)maxrdiff,maxrdiffcol);CHKERRQ(ierr);
28876719d8e4SJed Brown             for (j=0; j<bn; j++) {
28886719d8e4SJed Brown               PetscReal rdiff;
28896719d8e4SJed Brown               rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
28906719d8e4SJed Brown               if (rdiff > 1) {
289157622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(vstdout," (%D,%g:%g)",bj[j],(double)PetscRealPart(ba[j]),(double)PetscRealPart(ca[j]));CHKERRQ(ierr);
28926719d8e4SJed Brown               }
28936719d8e4SJed Brown             }
28946719d8e4SJed Brown             ierr = PetscViewerASCIIPrintf(vstdout,"\n",i,maxentry,maxdiff,maxrdiff);CHKERRQ(ierr);
28956719d8e4SJed Brown           }
289694ab13aaSBarry Smith           ierr = MatRestoreRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr);
28976719d8e4SJed Brown           ierr = MatRestoreRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
28986719d8e4SJed Brown         }
28996719d8e4SJed Brown       }
29004c30e9fbSJed Brown       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
29014c30e9fbSJed Brown       ierr = MatDestroy(&Bfd);CHKERRQ(ierr);
29024c30e9fbSJed Brown     }
29034c30e9fbSJed Brown   }
29043a40ed3dSBarry Smith   PetscFunctionReturn(0);
29059b94acceSBarry Smith }
29069b94acceSBarry Smith 
2907bf388a1fSBarry Smith /*MC
2908411c0326SBarry Smith     SNESJacobianFunction - Function used to convey the nonlinear Jacobian of the function to be solved by SNES
2909bf388a1fSBarry Smith 
2910bf388a1fSBarry Smith      Synopsis:
2911411c0326SBarry Smith      #include "petscsnes.h"
2912411c0326SBarry Smith      PetscErrorCode SNESJacobianFunction(SNES snes,Vec x,Mat Amat,Mat Pmat,void *ctx);
2913bf388a1fSBarry Smith 
29141843f636SBarry Smith      Collective on snes
29151843f636SBarry Smith 
29161843f636SBarry Smith     Input Parameters:
29171843f636SBarry Smith +  x - input vector, the Jacobian is to be computed at this value
2918bf388a1fSBarry Smith -  ctx - [optional] user-defined Jacobian context
2919bf388a1fSBarry Smith 
29201843f636SBarry Smith     Output Parameters:
29211843f636SBarry Smith +  Amat - the matrix that defines the (approximate) Jacobian
29221843f636SBarry Smith -  Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat.
29231843f636SBarry Smith 
2924878cb397SSatish Balay    Level: intermediate
2925878cb397SSatish Balay 
2926bf388a1fSBarry Smith .seealso:   SNESSetFunction(), SNESGetFunction(), SNESSetJacobian(), SNESGetJacobian()
2927bf388a1fSBarry Smith M*/
2928bf388a1fSBarry Smith 
29299b94acceSBarry Smith /*@C
29309b94acceSBarry Smith    SNESSetJacobian - Sets the function to compute Jacobian as well as the
2931044dda88SLois Curfman McInnes    location to store the matrix.
29329b94acceSBarry Smith 
2933d083f849SBarry Smith    Logically Collective on SNES
2934c7afd0dbSLois Curfman McInnes 
29359b94acceSBarry Smith    Input Parameters:
2936c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2937e5d3d808SBarry Smith .  Amat - the matrix that defines the (approximate) Jacobian
2938e5d3d808SBarry Smith .  Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat.
2939411c0326SBarry Smith .  J - Jacobian evaluation routine (if NULL then SNES retains any previously set value), see SNESJacobianFunction for details
2940c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
29410298fd71SBarry Smith          Jacobian evaluation routine (may be NULL) (if NULL then SNES retains any previously set value)
29429b94acceSBarry Smith 
29439b94acceSBarry Smith    Notes:
2944e5d3d808SBarry Smith    If the Amat matrix and Pmat matrix are different you must call MatAssemblyBegin/End() on
294516913363SBarry Smith    each matrix.
294616913363SBarry Smith 
2947895c21f2SBarry Smith    If you know the operator Amat has a null space you can use MatSetNullSpace() and MatSetTransposeNullSpace() to supply the null
2948895c21f2SBarry Smith    space to Amat and the KSP solvers will automatically use that null space as needed during the solution process.
2949895c21f2SBarry Smith 
29508d359177SBarry Smith    If using SNESComputeJacobianDefaultColor() to assemble a Jacobian, the ctx argument
2951a8a26c1eSJed Brown    must be a MatFDColoring.
2952a8a26c1eSJed Brown 
2953c3cc8fd1SJed Brown    Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian.  One common
2954c3cc8fd1SJed Brown    example is to use the "Picard linearization" which only differentiates through the highest order parts of each term.
2955c3cc8fd1SJed Brown 
295636851e7fSLois Curfman McInnes    Level: beginner
295736851e7fSLois Curfman McInnes 
2958411c0326SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESComputeJacobianDefaultColor(), MatStructure, J,
2959411c0326SBarry Smith           SNESSetPicard(), SNESJacobianFunction
29609b94acceSBarry Smith @*/
2961d1e9a80fSBarry Smith PetscErrorCode  SNESSetJacobian(SNES snes,Mat Amat,Mat Pmat,PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx)
29629b94acceSBarry Smith {
2963dfbe8321SBarry Smith   PetscErrorCode ierr;
29646cab3a1bSJed Brown   DM             dm;
29653a7fca6bSBarry Smith 
29663a40ed3dSBarry Smith   PetscFunctionBegin;
29670700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2968e5d3d808SBarry Smith   if (Amat) PetscValidHeaderSpecific(Amat,MAT_CLASSID,2);
2969e5d3d808SBarry Smith   if (Pmat) PetscValidHeaderSpecific(Pmat,MAT_CLASSID,3);
2970e5d3d808SBarry Smith   if (Amat) PetscCheckSameComm(snes,1,Amat,2);
2971e5d3d808SBarry Smith   if (Pmat) PetscCheckSameComm(snes,1,Pmat,3);
29726cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2973f8b49ee9SBarry Smith   ierr = DMSNESSetJacobian(dm,J,ctx);CHKERRQ(ierr);
2974e5d3d808SBarry Smith   if (Amat) {
2975e5d3d808SBarry Smith     ierr = PetscObjectReference((PetscObject)Amat);CHKERRQ(ierr);
29766bf464f9SBarry Smith     ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
2977f5af7f23SKarl Rupp 
2978e5d3d808SBarry Smith     snes->jacobian = Amat;
29793a7fca6bSBarry Smith   }
2980e5d3d808SBarry Smith   if (Pmat) {
2981e5d3d808SBarry Smith     ierr = PetscObjectReference((PetscObject)Pmat);CHKERRQ(ierr);
29826bf464f9SBarry Smith     ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
2983f5af7f23SKarl Rupp 
2984e5d3d808SBarry Smith     snes->jacobian_pre = Pmat;
29853a7fca6bSBarry Smith   }
29863a40ed3dSBarry Smith   PetscFunctionReturn(0);
29879b94acceSBarry Smith }
298862fef451SLois Curfman McInnes 
2989c2aafc4cSSatish Balay /*@C
2990b4fd4287SBarry Smith    SNESGetJacobian - Returns the Jacobian matrix and optionally the user
2991b4fd4287SBarry Smith    provided context for evaluating the Jacobian.
2992b4fd4287SBarry Smith 
2993c7afd0dbSLois Curfman McInnes    Not Collective, but Mat object will be parallel if SNES object is
2994c7afd0dbSLois Curfman McInnes 
2995b4fd4287SBarry Smith    Input Parameter:
2996b4fd4287SBarry Smith .  snes - the nonlinear solver context
2997b4fd4287SBarry Smith 
2998b4fd4287SBarry Smith    Output Parameters:
2999e5d3d808SBarry Smith +  Amat - location to stash (approximate) Jacobian matrix (or NULL)
3000e5d3d808SBarry Smith .  Pmat - location to stash matrix used to compute the preconditioner (or NULL)
3001411c0326SBarry Smith .  J - location to put Jacobian function (or NULL), see SNESJacobianFunction for details on its calling sequence
30020298fd71SBarry Smith -  ctx - location to stash Jacobian ctx (or NULL)
3003fee21e36SBarry Smith 
300436851e7fSLois Curfman McInnes    Level: advanced
300536851e7fSLois Curfman McInnes 
3006411c0326SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian(), SNESJacobianFunction, SNESGetFunction()
3007b4fd4287SBarry Smith @*/
3008d1e9a80fSBarry Smith PetscErrorCode SNESGetJacobian(SNES snes,Mat *Amat,Mat *Pmat,PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx)
3009b4fd4287SBarry Smith {
30106cab3a1bSJed Brown   PetscErrorCode ierr;
30116cab3a1bSJed Brown   DM             dm;
3012942e3340SBarry Smith   DMSNES         sdm;
30136cab3a1bSJed Brown 
30143a40ed3dSBarry Smith   PetscFunctionBegin;
30150700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3016e5d3d808SBarry Smith   if (Amat) *Amat = snes->jacobian;
3017e5d3d808SBarry Smith   if (Pmat) *Pmat = snes->jacobian_pre;
30186cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
3019942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
3020f8b49ee9SBarry Smith   if (J) *J = sdm->ops->computejacobian;
30216cab3a1bSJed Brown   if (ctx) *ctx = sdm->jacobianctx;
30223a40ed3dSBarry Smith   PetscFunctionReturn(0);
3023b4fd4287SBarry Smith }
3024b4fd4287SBarry Smith 
302558b371f3SBarry Smith static PetscErrorCode SNESSetDefaultComputeJacobian(SNES snes)
302658b371f3SBarry Smith {
302758b371f3SBarry Smith   PetscErrorCode ierr;
302858b371f3SBarry Smith   DM             dm;
302958b371f3SBarry Smith   DMSNES         sdm;
303058b371f3SBarry Smith 
303158b371f3SBarry Smith   PetscFunctionBegin;
303258b371f3SBarry Smith   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
303358b371f3SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
303458b371f3SBarry Smith   if (!sdm->ops->computejacobian && snes->jacobian_pre) {
303558b371f3SBarry Smith     DM        dm;
303658b371f3SBarry Smith     PetscBool isdense,ismf;
303758b371f3SBarry Smith 
303858b371f3SBarry Smith     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
303958b371f3SBarry Smith     ierr = PetscObjectTypeCompareAny((PetscObject)snes->jacobian_pre,&isdense,MATSEQDENSE,MATMPIDENSE,MATDENSE,NULL);CHKERRQ(ierr);
304058b371f3SBarry Smith     ierr = PetscObjectTypeCompareAny((PetscObject)snes->jacobian_pre,&ismf,MATMFFD,MATSHELL,NULL);CHKERRQ(ierr);
304158b371f3SBarry Smith     if (isdense) {
304258b371f3SBarry Smith       ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefault,NULL);CHKERRQ(ierr);
304358b371f3SBarry Smith     } else if (!ismf) {
304458b371f3SBarry Smith       ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr);
304558b371f3SBarry Smith     }
304658b371f3SBarry Smith   }
304758b371f3SBarry Smith   PetscFunctionReturn(0);
304858b371f3SBarry Smith }
304958b371f3SBarry Smith 
30509b94acceSBarry Smith /*@
30519b94acceSBarry Smith    SNESSetUp - Sets up the internal data structures for the later use
3052272ac6f2SLois Curfman McInnes    of a nonlinear solver.
30539b94acceSBarry Smith 
3054fee21e36SBarry Smith    Collective on SNES
3055fee21e36SBarry Smith 
3056c7afd0dbSLois Curfman McInnes    Input Parameters:
305770e92668SMatthew Knepley .  snes - the SNES context
3058c7afd0dbSLois Curfman McInnes 
3059272ac6f2SLois Curfman McInnes    Notes:
3060272ac6f2SLois Curfman McInnes    For basic use of the SNES solvers the user need not explicitly call
3061272ac6f2SLois Curfman McInnes    SNESSetUp(), since these actions will automatically occur during
3062272ac6f2SLois Curfman McInnes    the call to SNESSolve().  However, if one wishes to control this
3063272ac6f2SLois Curfman McInnes    phase separately, SNESSetUp() should be called after SNESCreate()
3064272ac6f2SLois Curfman McInnes    and optional routines of the form SNESSetXXX(), but before SNESSolve().
3065272ac6f2SLois Curfman McInnes 
306636851e7fSLois Curfman McInnes    Level: advanced
306736851e7fSLois Curfman McInnes 
30689b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy()
30699b94acceSBarry Smith @*/
30707087cfbeSBarry Smith PetscErrorCode  SNESSetUp(SNES snes)
30719b94acceSBarry Smith {
3072dfbe8321SBarry Smith   PetscErrorCode ierr;
30736cab3a1bSJed Brown   DM             dm;
3074942e3340SBarry Smith   DMSNES         sdm;
3075c35f09e5SBarry Smith   SNESLineSearch linesearch, pclinesearch;
30766e2a1849SPeter Brune   void           *lsprectx,*lspostctx;
30776b2b7091SBarry Smith   PetscErrorCode (*precheck)(SNESLineSearch,Vec,Vec,PetscBool*,void*);
30786b2b7091SBarry Smith   PetscErrorCode (*postcheck)(SNESLineSearch,Vec,Vec,Vec,PetscBool*,PetscBool*,void*);
30796e2a1849SPeter Brune   PetscErrorCode (*func)(SNES,Vec,Vec,void*);
30806e2a1849SPeter Brune   Vec            f,fpc;
30816e2a1849SPeter Brune   void           *funcctx;
3082d1e9a80fSBarry Smith   PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*);
30831eb13d49SPeter Brune   void           *jacctx,*appctx;
308432b97717SPeter Brune   Mat            j,jpre;
30853a40ed3dSBarry Smith 
30863a40ed3dSBarry Smith   PetscFunctionBegin;
30870700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
30884dc4c822SBarry Smith   if (snes->setupcalled) PetscFunctionReturn(0);
3089e3ed9ee7SBarry Smith   ierr = PetscLogEventBegin(SNES_Setup,snes,0,0,0);CHKERRQ(ierr);
30909b94acceSBarry Smith 
30917adad957SLisandro Dalcin   if (!((PetscObject)snes)->type_name) {
309204d7464bSBarry Smith     ierr = SNESSetType(snes,SNESNEWTONLS);CHKERRQ(ierr);
309385385478SLisandro Dalcin   }
309485385478SLisandro Dalcin 
30950298fd71SBarry Smith   ierr = SNESGetFunction(snes,&snes->vec_func,NULL,NULL);CHKERRQ(ierr);
309658c9b817SLisandro Dalcin 
30976cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
3098942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
3099ce94432eSBarry Smith   if (!sdm->ops->computefunction) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE,"Function never provided to SNES object");
310058b371f3SBarry Smith   ierr = SNESSetDefaultComputeJacobian(snes);CHKERRQ(ierr);
310158b371f3SBarry Smith 
31026cab3a1bSJed Brown   if (!snes->vec_func) {
31036cab3a1bSJed Brown     ierr = DMCreateGlobalVector(dm,&snes->vec_func);CHKERRQ(ierr);
3104214df951SJed Brown   }
3105efd51863SBarry Smith 
310622d28d08SBarry Smith   if (!snes->ksp) {
310722d28d08SBarry Smith     ierr = SNESGetKSP(snes, &snes->ksp);CHKERRQ(ierr);
310822d28d08SBarry Smith   }
3109b710008aSBarry Smith 
3110d8d34be6SBarry Smith   if (snes->linesearch) {
31117601faf0SJed Brown     ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr);
3112ed07d7d7SPeter Brune     ierr = SNESLineSearchSetFunction(snes->linesearch,SNESComputeFunction);CHKERRQ(ierr);
3113d8d34be6SBarry Smith   }
31149e764e56SPeter Brune 
3115efd4aadfSBarry Smith   if (snes->npc && (snes->npcside== PC_LEFT)) {
3116172a4300SPeter Brune     snes->mf          = PETSC_TRUE;
3117172a4300SPeter Brune     snes->mf_operator = PETSC_FALSE;
3118172a4300SPeter Brune   }
3119d8f46077SPeter Brune 
3120efd4aadfSBarry Smith   if (snes->npc) {
31216e2a1849SPeter Brune     /* copy the DM over */
31226e2a1849SPeter Brune     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
3123efd4aadfSBarry Smith     ierr = SNESSetDM(snes->npc,dm);CHKERRQ(ierr);
31246e2a1849SPeter Brune 
31256e2a1849SPeter Brune     ierr = SNESGetFunction(snes,&f,&func,&funcctx);CHKERRQ(ierr);
31266e2a1849SPeter Brune     ierr = VecDuplicate(f,&fpc);CHKERRQ(ierr);
3127efd4aadfSBarry Smith     ierr = SNESSetFunction(snes->npc,fpc,func,funcctx);CHKERRQ(ierr);
312832b97717SPeter Brune     ierr = SNESGetJacobian(snes,&j,&jpre,&jac,&jacctx);CHKERRQ(ierr);
3129efd4aadfSBarry Smith     ierr = SNESSetJacobian(snes->npc,j,jpre,jac,jacctx);CHKERRQ(ierr);
31301eb13d49SPeter Brune     ierr = SNESGetApplicationContext(snes,&appctx);CHKERRQ(ierr);
3131efd4aadfSBarry Smith     ierr = SNESSetApplicationContext(snes->npc,appctx);CHKERRQ(ierr);
31326e2a1849SPeter Brune     ierr = VecDestroy(&fpc);CHKERRQ(ierr);
31336e2a1849SPeter Brune 
31346e2a1849SPeter Brune     /* copy the function pointers over */
3135efd4aadfSBarry Smith     ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr);
31366e2a1849SPeter Brune 
31376e2a1849SPeter Brune     /* default to 1 iteration */
3138efd4aadfSBarry Smith     ierr = SNESSetTolerances(snes->npc,0.0,0.0,0.0,1,snes->npc->max_funcs);CHKERRQ(ierr);
3139efd4aadfSBarry Smith     if (snes->npcside==PC_RIGHT) {
3140efd4aadfSBarry Smith       ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_FINAL_ONLY);CHKERRQ(ierr);
3141a9936a0cSPeter Brune     } else {
3142efd4aadfSBarry Smith       ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_NONE);CHKERRQ(ierr);
3143a9936a0cSPeter Brune     }
3144efd4aadfSBarry Smith     ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr);
31456e2a1849SPeter Brune 
31466e2a1849SPeter Brune     /* copy the line search context over */
3147d8d34be6SBarry Smith     if (snes->linesearch && snes->npc->linesearch) {
31487601faf0SJed Brown       ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr);
3149efd4aadfSBarry Smith       ierr = SNESGetLineSearch(snes->npc,&pclinesearch);CHKERRQ(ierr);
31506b2b7091SBarry Smith       ierr = SNESLineSearchGetPreCheck(linesearch,&precheck,&lsprectx);CHKERRQ(ierr);
31516b2b7091SBarry Smith       ierr = SNESLineSearchGetPostCheck(linesearch,&postcheck,&lspostctx);CHKERRQ(ierr);
31526b2b7091SBarry Smith       ierr = SNESLineSearchSetPreCheck(pclinesearch,precheck,lsprectx);CHKERRQ(ierr);
31536b2b7091SBarry Smith       ierr = SNESLineSearchSetPostCheck(pclinesearch,postcheck,lspostctx);CHKERRQ(ierr);
31546e2a1849SPeter Brune       ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)linesearch, (PetscObject)pclinesearch);CHKERRQ(ierr);
31556e2a1849SPeter Brune     }
3156d8d34be6SBarry Smith   }
315732b97717SPeter Brune   if (snes->mf) {
315832b97717SPeter Brune     ierr = SNESSetUpMatrixFree_Private(snes, snes->mf_operator, snes->mf_version);CHKERRQ(ierr);
315932b97717SPeter Brune   }
316032b97717SPeter Brune   if (snes->ops->usercompute && !snes->user) {
316132b97717SPeter Brune     ierr = (*snes->ops->usercompute)(snes,(void**)&snes->user);CHKERRQ(ierr);
316232b97717SPeter Brune   }
31636e2a1849SPeter Brune 
316437ec4e1aSPeter Brune   snes->jac_iter = 0;
316537ec4e1aSPeter Brune   snes->pre_iter = 0;
316637ec4e1aSPeter Brune 
3167410397dcSLisandro Dalcin   if (snes->ops->setup) {
3168410397dcSLisandro Dalcin     ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr);
3169410397dcSLisandro Dalcin   }
317058c9b817SLisandro Dalcin 
317158b371f3SBarry Smith   ierr = SNESSetDefaultComputeJacobian(snes);CHKERRQ(ierr);
317258b371f3SBarry Smith 
3173efd4aadfSBarry Smith   if (snes->npc && (snes->npcside== PC_LEFT)) {
31746c67d002SPeter Brune     if (snes->functype == SNES_FUNCTION_PRECONDITIONED) {
3175d8d34be6SBarry Smith       if (snes->linesearch){
317655d4788fSPeter Brune         ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr);
3177be95d8f1SBarry Smith         ierr = SNESLineSearchSetFunction(linesearch,SNESComputeFunctionDefaultNPC);CHKERRQ(ierr);
31786c67d002SPeter Brune       }
31796c67d002SPeter Brune     }
3180d8d34be6SBarry Smith   }
3181e3ed9ee7SBarry Smith   ierr = PetscLogEventEnd(SNES_Setup,snes,0,0,0);CHKERRQ(ierr);
31827aaed0d8SBarry Smith   snes->setupcalled = PETSC_TRUE;
31833a40ed3dSBarry Smith   PetscFunctionReturn(0);
31849b94acceSBarry Smith }
31859b94acceSBarry Smith 
318637596af1SLisandro Dalcin /*@
318737596af1SLisandro Dalcin    SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats
318837596af1SLisandro Dalcin 
318937596af1SLisandro Dalcin    Collective on SNES
319037596af1SLisandro Dalcin 
319137596af1SLisandro Dalcin    Input Parameter:
319237596af1SLisandro Dalcin .  snes - iterative context obtained from SNESCreate()
319337596af1SLisandro Dalcin 
3194d25893d9SBarry Smith    Level: intermediate
3195d25893d9SBarry Smith 
319695452b02SPatrick Sanan    Notes:
319795452b02SPatrick Sanan     Also calls the application context destroy routine set with SNESSetComputeApplicationContext()
319837596af1SLisandro Dalcin 
319937596af1SLisandro Dalcin .seealso: SNESCreate(), SNESSetUp(), SNESSolve()
320037596af1SLisandro Dalcin @*/
320137596af1SLisandro Dalcin PetscErrorCode  SNESReset(SNES snes)
320237596af1SLisandro Dalcin {
320337596af1SLisandro Dalcin   PetscErrorCode ierr;
320437596af1SLisandro Dalcin 
320537596af1SLisandro Dalcin   PetscFunctionBegin;
320637596af1SLisandro Dalcin   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3207d25893d9SBarry Smith   if (snes->ops->userdestroy && snes->user) {
3208d25893d9SBarry Smith     ierr       = (*snes->ops->userdestroy)((void**)&snes->user);CHKERRQ(ierr);
32090298fd71SBarry Smith     snes->user = NULL;
3210d25893d9SBarry Smith   }
3211efd4aadfSBarry Smith   if (snes->npc) {
3212efd4aadfSBarry Smith     ierr = SNESReset(snes->npc);CHKERRQ(ierr);
32138a23116dSBarry Smith   }
32148a23116dSBarry Smith 
321537596af1SLisandro Dalcin   if (snes->ops->reset) {
321637596af1SLisandro Dalcin     ierr = (*snes->ops->reset)(snes);CHKERRQ(ierr);
321737596af1SLisandro Dalcin   }
32189e764e56SPeter Brune   if (snes->ksp) {
32199e764e56SPeter Brune     ierr = KSPReset(snes->ksp);CHKERRQ(ierr);
32209e764e56SPeter Brune   }
32219e764e56SPeter Brune 
32229e764e56SPeter Brune   if (snes->linesearch) {
3223f1c6b773SPeter Brune     ierr = SNESLineSearchReset(snes->linesearch);CHKERRQ(ierr);
32249e764e56SPeter Brune   }
32259e764e56SPeter Brune 
32266bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
32276bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
32286bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_sol_update);CHKERRQ(ierr);
32296bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr);
32306bf464f9SBarry Smith   ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
32316bf464f9SBarry Smith   ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
3232c41d97abSJed Brown   ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr);
3233c41d97abSJed Brown   ierr = VecDestroyVecs(snes->nvwork,&snes->vwork);CHKERRQ(ierr);
3234f5af7f23SKarl Rupp 
323540fdac6aSLawrence Mitchell   snes->alwayscomputesfinalresidual = PETSC_FALSE;
323640fdac6aSLawrence Mitchell 
323737596af1SLisandro Dalcin   snes->nwork       = snes->nvwork = 0;
323837596af1SLisandro Dalcin   snes->setupcalled = PETSC_FALSE;
323937596af1SLisandro Dalcin   PetscFunctionReturn(0);
324037596af1SLisandro Dalcin }
324137596af1SLisandro Dalcin 
324252baeb72SSatish Balay /*@
3243c4421ceaSFande Kong    SNESConvergedReasonViewCancel - Clears all the reasonview functions for a SNES object.
3244c4421ceaSFande Kong 
3245c4421ceaSFande Kong    Collective on SNES
3246c4421ceaSFande Kong 
3247c4421ceaSFande Kong    Input Parameter:
3248c4421ceaSFande Kong .  snes - iterative context obtained from SNESCreate()
3249c4421ceaSFande Kong 
3250c4421ceaSFande Kong    Level: intermediate
3251c4421ceaSFande Kong 
3252c4421ceaSFande Kong .seealso: SNESCreate(), SNESDestroy(), SNESReset()
3253c4421ceaSFande Kong @*/
3254c4421ceaSFande Kong PetscErrorCode  SNESConvergedReasonViewCancel(SNES snes)
3255c4421ceaSFande Kong {
3256c4421ceaSFande Kong   PetscErrorCode ierr;
3257c4421ceaSFande Kong   PetscInt       i;
3258c4421ceaSFande Kong 
3259c4421ceaSFande Kong   PetscFunctionBegin;
3260c4421ceaSFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3261c4421ceaSFande Kong   for (i=0; i<snes->numberreasonviews; i++) {
3262c4421ceaSFande Kong     if (snes->reasonviewdestroy[i]) {
3263c4421ceaSFande Kong       ierr = (*snes->reasonviewdestroy[i])(&snes->reasonviewcontext[i]);CHKERRQ(ierr);
3264c4421ceaSFande Kong     }
3265c4421ceaSFande Kong   }
3266c4421ceaSFande Kong   snes->numberreasonviews = 0;
3267c4421ceaSFande Kong   PetscFunctionReturn(0);
3268c4421ceaSFande Kong }
3269c4421ceaSFande Kong 
3270*1fb7b255SJunchao Zhang /*@C
32719b94acceSBarry Smith    SNESDestroy - Destroys the nonlinear solver context that was created
32729b94acceSBarry Smith    with SNESCreate().
32739b94acceSBarry Smith 
3274c7afd0dbSLois Curfman McInnes    Collective on SNES
3275c7afd0dbSLois Curfman McInnes 
32769b94acceSBarry Smith    Input Parameter:
32779b94acceSBarry Smith .  snes - the SNES context
32789b94acceSBarry Smith 
327936851e7fSLois Curfman McInnes    Level: beginner
328036851e7fSLois Curfman McInnes 
328163a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve()
32829b94acceSBarry Smith @*/
32836bf464f9SBarry Smith PetscErrorCode  SNESDestroy(SNES *snes)
32849b94acceSBarry Smith {
32856849ba73SBarry Smith   PetscErrorCode ierr;
32863a40ed3dSBarry Smith 
32873a40ed3dSBarry Smith   PetscFunctionBegin;
32886bf464f9SBarry Smith   if (!*snes) PetscFunctionReturn(0);
32896bf464f9SBarry Smith   PetscValidHeaderSpecific((*snes),SNES_CLASSID,1);
32909e5d0892SLisandro Dalcin   if (--((PetscObject)(*snes))->refct > 0) {*snes = NULL; PetscFunctionReturn(0);}
3291d4bb536fSBarry Smith 
32926bf464f9SBarry Smith   ierr = SNESReset((*snes));CHKERRQ(ierr);
3293efd4aadfSBarry Smith   ierr = SNESDestroy(&(*snes)->npc);CHKERRQ(ierr);
32946b8b9a38SLisandro Dalcin 
3295e04113cfSBarry Smith   /* if memory was published with SAWs then destroy it */
3296e04113cfSBarry Smith   ierr = PetscObjectSAWsViewOff((PetscObject)*snes);CHKERRQ(ierr);
32976bf464f9SBarry Smith   if ((*snes)->ops->destroy) {ierr = (*((*snes))->ops->destroy)((*snes));CHKERRQ(ierr);}
32986d4c513bSLisandro Dalcin 
329933124788SMatthew G. Knepley   if ((*snes)->dm) {ierr = DMCoarsenHookRemove((*snes)->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,*snes);CHKERRQ(ierr);}
33006bf464f9SBarry Smith   ierr = DMDestroy(&(*snes)->dm);CHKERRQ(ierr);
33016bf464f9SBarry Smith   ierr = KSPDestroy(&(*snes)->ksp);CHKERRQ(ierr);
3302f1c6b773SPeter Brune   ierr = SNESLineSearchDestroy(&(*snes)->linesearch);CHKERRQ(ierr);
33036b8b9a38SLisandro Dalcin 
33046bf464f9SBarry Smith   ierr = PetscFree((*snes)->kspconvctx);CHKERRQ(ierr);
33056bf464f9SBarry Smith   if ((*snes)->ops->convergeddestroy) {
33066bf464f9SBarry Smith     ierr = (*(*snes)->ops->convergeddestroy)((*snes)->cnvP);CHKERRQ(ierr);
33076b8b9a38SLisandro Dalcin   }
3308071fcb05SBarry Smith   if ((*snes)->conv_hist_alloc) {
3309071fcb05SBarry Smith     ierr = PetscFree2((*snes)->conv_hist,(*snes)->conv_hist_its);CHKERRQ(ierr);
331058c9b817SLisandro Dalcin   }
33116bf464f9SBarry Smith   ierr = SNESMonitorCancel((*snes));CHKERRQ(ierr);
3312c4421ceaSFande Kong   ierr = SNESConvergedReasonViewCancel((*snes));CHKERRQ(ierr);
3313a79aaaedSSatish Balay   ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr);
33143a40ed3dSBarry Smith   PetscFunctionReturn(0);
33159b94acceSBarry Smith }
33169b94acceSBarry Smith 
33179b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */
33189b94acceSBarry Smith 
3319a8054027SBarry Smith /*@
3320a8054027SBarry Smith    SNESSetLagPreconditioner - Determines when the preconditioner is rebuilt in the nonlinear solve.
3321a8054027SBarry Smith 
33223f9fe445SBarry Smith    Logically Collective on SNES
3323a8054027SBarry Smith 
3324a8054027SBarry Smith    Input Parameters:
3325a8054027SBarry Smith +  snes - the SNES context
3326d8e291bfSBarry Smith -  lag - 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
33273b4f5425SBarry Smith          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
3328a8054027SBarry Smith 
3329a8054027SBarry Smith    Options Database Keys:
33303d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
33313d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
33323d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
33333d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag
3334a8054027SBarry Smith 
3335a8054027SBarry Smith    Notes:
3336a8054027SBarry Smith    The default is 1
33373d5a8a6aSBarry Smith    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 or SNESSetLagPreconditionerPersists() was called
3338d8e291bfSBarry Smith 
3339d8e291bfSBarry Smith    SNESSetLagPreconditionerPersists() allows using the same uniform lagging (for example every second solve) across multiple solves.
3340a8054027SBarry Smith 
3341a8054027SBarry Smith    Level: intermediate
3342a8054027SBarry Smith 
33433d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetLagPreconditionerPersists(),
33443d5a8a6aSBarry Smith           SNESSetLagJacobianPersists()
3345a8054027SBarry Smith 
3346a8054027SBarry Smith @*/
33477087cfbeSBarry Smith PetscErrorCode  SNESSetLagPreconditioner(SNES snes,PetscInt lag)
3348a8054027SBarry Smith {
3349a8054027SBarry Smith   PetscFunctionBegin;
33500700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3351e32f2f54SBarry Smith   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
3352e32f2f54SBarry Smith   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
3353c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,lag,2);
3354a8054027SBarry Smith   snes->lagpreconditioner = lag;
3355a8054027SBarry Smith   PetscFunctionReturn(0);
3356a8054027SBarry Smith }
3357a8054027SBarry Smith 
3358efd51863SBarry Smith /*@
3359efd51863SBarry Smith    SNESSetGridSequence - sets the number of steps of grid sequencing that SNES does
3360efd51863SBarry Smith 
3361efd51863SBarry Smith    Logically Collective on SNES
3362efd51863SBarry Smith 
3363efd51863SBarry Smith    Input Parameters:
3364efd51863SBarry Smith +  snes - the SNES context
3365efd51863SBarry Smith -  steps - the number of refinements to do, defaults to 0
3366efd51863SBarry Smith 
3367efd51863SBarry Smith    Options Database Keys:
3368efd51863SBarry Smith .    -snes_grid_sequence <steps>
3369efd51863SBarry Smith 
3370efd51863SBarry Smith    Level: intermediate
3371efd51863SBarry Smith 
3372c0df2a02SJed Brown    Notes:
3373c0df2a02SJed Brown    Use SNESGetSolution() to extract the fine grid solution after grid sequencing.
3374c0df2a02SJed Brown 
3375fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetGridSequence()
3376efd51863SBarry Smith 
3377efd51863SBarry Smith @*/
3378efd51863SBarry Smith PetscErrorCode  SNESSetGridSequence(SNES snes,PetscInt steps)
3379efd51863SBarry Smith {
3380efd51863SBarry Smith   PetscFunctionBegin;
3381efd51863SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3382efd51863SBarry Smith   PetscValidLogicalCollectiveInt(snes,steps,2);
3383efd51863SBarry Smith   snes->gridsequence = steps;
3384efd51863SBarry Smith   PetscFunctionReturn(0);
3385efd51863SBarry Smith }
3386efd51863SBarry Smith 
3387fa19ca70SBarry Smith /*@
3388fa19ca70SBarry Smith    SNESGetGridSequence - gets the number of steps of grid sequencing that SNES does
3389fa19ca70SBarry Smith 
3390fa19ca70SBarry Smith    Logically Collective on SNES
3391fa19ca70SBarry Smith 
3392fa19ca70SBarry Smith    Input Parameter:
3393fa19ca70SBarry Smith .  snes - the SNES context
3394fa19ca70SBarry Smith 
3395fa19ca70SBarry Smith    Output Parameter:
3396fa19ca70SBarry Smith .  steps - the number of refinements to do, defaults to 0
3397fa19ca70SBarry Smith 
3398fa19ca70SBarry Smith    Options Database Keys:
3399fa19ca70SBarry Smith .    -snes_grid_sequence <steps>
3400fa19ca70SBarry Smith 
3401fa19ca70SBarry Smith    Level: intermediate
3402fa19ca70SBarry Smith 
3403fa19ca70SBarry Smith    Notes:
3404fa19ca70SBarry Smith    Use SNESGetSolution() to extract the fine grid solution after grid sequencing.
3405fa19ca70SBarry Smith 
3406fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetGridSequence()
3407fa19ca70SBarry Smith 
3408fa19ca70SBarry Smith @*/
3409fa19ca70SBarry Smith PetscErrorCode  SNESGetGridSequence(SNES snes,PetscInt *steps)
3410fa19ca70SBarry Smith {
3411fa19ca70SBarry Smith   PetscFunctionBegin;
3412fa19ca70SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3413fa19ca70SBarry Smith   *steps = snes->gridsequence;
3414fa19ca70SBarry Smith   PetscFunctionReturn(0);
3415fa19ca70SBarry Smith }
3416fa19ca70SBarry Smith 
3417a8054027SBarry Smith /*@
3418a8054027SBarry Smith    SNESGetLagPreconditioner - Indicates how often the preconditioner is rebuilt
3419a8054027SBarry Smith 
34203f9fe445SBarry Smith    Not Collective
3421a8054027SBarry Smith 
3422a8054027SBarry Smith    Input Parameter:
3423a8054027SBarry Smith .  snes - the SNES context
3424a8054027SBarry Smith 
3425a8054027SBarry Smith    Output Parameter:
3426a8054027SBarry 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
34273b4f5425SBarry Smith          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
3428a8054027SBarry Smith 
3429a8054027SBarry Smith    Options Database Keys:
34303d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
34313d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
34323d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
34333d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag
3434a8054027SBarry Smith 
3435a8054027SBarry Smith    Notes:
3436a8054027SBarry Smith    The default is 1
3437a8054027SBarry Smith    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3438a8054027SBarry Smith 
3439a8054027SBarry Smith    Level: intermediate
3440a8054027SBarry Smith 
34413d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagPreconditioner(), SNESSetLagJacobianPersists(), SNESSetLagPreconditionerPersists()
3442a8054027SBarry Smith 
3443a8054027SBarry Smith @*/
34447087cfbeSBarry Smith PetscErrorCode  SNESGetLagPreconditioner(SNES snes,PetscInt *lag)
3445a8054027SBarry Smith {
3446a8054027SBarry Smith   PetscFunctionBegin;
34470700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3448a8054027SBarry Smith   *lag = snes->lagpreconditioner;
3449a8054027SBarry Smith   PetscFunctionReturn(0);
3450a8054027SBarry Smith }
3451a8054027SBarry Smith 
3452e35cf81dSBarry Smith /*@
3453e35cf81dSBarry Smith    SNESSetLagJacobian - Determines when the Jacobian is rebuilt in the nonlinear solve. See SNESSetLagPreconditioner() for determining how
3454e35cf81dSBarry Smith      often the preconditioner is rebuilt.
3455e35cf81dSBarry Smith 
34563f9fe445SBarry Smith    Logically Collective on SNES
3457e35cf81dSBarry Smith 
3458e35cf81dSBarry Smith    Input Parameters:
3459e35cf81dSBarry Smith +  snes - the SNES context
3460e35cf81dSBarry 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
3461fe3ffe1eSBarry Smith          the Jacobian is built etc. -2 means rebuild at next chance but then never again
3462e35cf81dSBarry Smith 
3463e35cf81dSBarry Smith    Options Database Keys:
34643d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
34653d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
34663d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
34673d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag.
3468e35cf81dSBarry Smith 
3469e35cf81dSBarry Smith    Notes:
3470e35cf81dSBarry Smith    The default is 1
3471e35cf81dSBarry Smith    The Jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3472fe3ffe1eSBarry 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
3473fe3ffe1eSBarry Smith    at the next Newton step but never again (unless it is reset to another value)
3474e35cf81dSBarry Smith 
3475e35cf81dSBarry Smith    Level: intermediate
3476e35cf81dSBarry Smith 
34773d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagPreconditioner(), SNESGetLagJacobianPersists(), SNESSetLagPreconditionerPersists()
3478e35cf81dSBarry Smith 
3479e35cf81dSBarry Smith @*/
34807087cfbeSBarry Smith PetscErrorCode  SNESSetLagJacobian(SNES snes,PetscInt lag)
3481e35cf81dSBarry Smith {
3482e35cf81dSBarry Smith   PetscFunctionBegin;
34830700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3484e32f2f54SBarry Smith   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
3485e32f2f54SBarry Smith   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
3486c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,lag,2);
3487e35cf81dSBarry Smith   snes->lagjacobian = lag;
3488e35cf81dSBarry Smith   PetscFunctionReturn(0);
3489e35cf81dSBarry Smith }
3490e35cf81dSBarry Smith 
3491e35cf81dSBarry Smith /*@
3492e35cf81dSBarry Smith    SNESGetLagJacobian - Indicates how often the Jacobian is rebuilt. See SNESGetLagPreconditioner() to determine when the preconditioner is rebuilt
3493e35cf81dSBarry Smith 
34943f9fe445SBarry Smith    Not Collective
3495e35cf81dSBarry Smith 
3496e35cf81dSBarry Smith    Input Parameter:
3497e35cf81dSBarry Smith .  snes - the SNES context
3498e35cf81dSBarry Smith 
3499e35cf81dSBarry Smith    Output Parameter:
3500e35cf81dSBarry 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
3501e35cf81dSBarry Smith          the Jacobian is built etc.
3502e35cf81dSBarry Smith 
3503e35cf81dSBarry Smith    Notes:
3504e35cf81dSBarry Smith    The default is 1
35053d5a8a6aSBarry Smith    The jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 or SNESSetLagJacobianPersists() was called.
3506e35cf81dSBarry Smith 
3507e35cf81dSBarry Smith    Level: intermediate
3508e35cf81dSBarry Smith 
35093d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagJacobian(), SNESSetLagPreconditioner(), SNESGetLagPreconditioner(), SNESSetLagJacobianPersists(), SNESSetLagPreconditionerPersists()
3510e35cf81dSBarry Smith 
3511e35cf81dSBarry Smith @*/
35127087cfbeSBarry Smith PetscErrorCode  SNESGetLagJacobian(SNES snes,PetscInt *lag)
3513e35cf81dSBarry Smith {
3514e35cf81dSBarry Smith   PetscFunctionBegin;
35150700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3516e35cf81dSBarry Smith   *lag = snes->lagjacobian;
3517e35cf81dSBarry Smith   PetscFunctionReturn(0);
3518e35cf81dSBarry Smith }
3519e35cf81dSBarry Smith 
352037ec4e1aSPeter Brune /*@
352137ec4e1aSPeter Brune    SNESSetLagJacobianPersists - Set whether or not the Jacobian lagging persists through multiple solves
352237ec4e1aSPeter Brune 
352337ec4e1aSPeter Brune    Logically collective on SNES
352437ec4e1aSPeter Brune 
352537ec4e1aSPeter Brune    Input Parameter:
352637ec4e1aSPeter Brune +  snes - the SNES context
35279d7e2deaSPeter Brune -   flg - jacobian lagging persists if true
352837ec4e1aSPeter Brune 
352937ec4e1aSPeter Brune    Options Database Keys:
35303d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
35313d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
35323d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
35333d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag
35343d5a8a6aSBarry Smith 
353537ec4e1aSPeter Brune 
353695452b02SPatrick Sanan    Notes:
353795452b02SPatrick Sanan     This is useful both for nonlinear preconditioning, where it's appropriate to have the Jacobian be stale by
353837ec4e1aSPeter Brune    several solves, and for implicit time-stepping, where Jacobian lagging in the inner nonlinear solve over several
353937ec4e1aSPeter Brune    timesteps may present huge efficiency gains.
354037ec4e1aSPeter Brune 
354137ec4e1aSPeter Brune    Level: developer
354237ec4e1aSPeter Brune 
35433d5a8a6aSBarry Smith .seealso: SNESSetLagPreconditionerPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC(), SNESSetLagJacobianPersists()
354437ec4e1aSPeter Brune 
354537ec4e1aSPeter Brune @*/
354637ec4e1aSPeter Brune PetscErrorCode  SNESSetLagJacobianPersists(SNES snes,PetscBool flg)
354737ec4e1aSPeter Brune {
354837ec4e1aSPeter Brune   PetscFunctionBegin;
354937ec4e1aSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
355037ec4e1aSPeter Brune   PetscValidLogicalCollectiveBool(snes,flg,2);
355137ec4e1aSPeter Brune   snes->lagjac_persist = flg;
355237ec4e1aSPeter Brune   PetscFunctionReturn(0);
355337ec4e1aSPeter Brune }
355437ec4e1aSPeter Brune 
355537ec4e1aSPeter Brune /*@
3556d8e291bfSBarry Smith    SNESSetLagPreconditionerPersists - Set whether or not the preconditioner lagging persists through multiple nonlinear solves
355737ec4e1aSPeter Brune 
355837ec4e1aSPeter Brune    Logically Collective on SNES
355937ec4e1aSPeter Brune 
356037ec4e1aSPeter Brune    Input Parameter:
356137ec4e1aSPeter Brune +  snes - the SNES context
35629d7e2deaSPeter Brune -   flg - preconditioner lagging persists if true
356337ec4e1aSPeter Brune 
356437ec4e1aSPeter Brune    Options Database Keys:
35653d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
35663d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
35673d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
35683d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag
356937ec4e1aSPeter Brune 
357095452b02SPatrick Sanan    Notes:
357195452b02SPatrick Sanan     This is useful both for nonlinear preconditioning, where it's appropriate to have the preconditioner be stale
357237ec4e1aSPeter Brune    by several solves, and for implicit time-stepping, where preconditioner lagging in the inner nonlinear solve over
357337ec4e1aSPeter Brune    several timesteps may present huge efficiency gains.
357437ec4e1aSPeter Brune 
357537ec4e1aSPeter Brune    Level: developer
357637ec4e1aSPeter Brune 
35773d5a8a6aSBarry Smith .seealso: SNESSetLagJacobianPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC(), SNESSetLagPreconditioner()
357837ec4e1aSPeter Brune 
357937ec4e1aSPeter Brune @*/
358037ec4e1aSPeter Brune PetscErrorCode  SNESSetLagPreconditionerPersists(SNES snes,PetscBool flg)
358137ec4e1aSPeter Brune {
358237ec4e1aSPeter Brune   PetscFunctionBegin;
358337ec4e1aSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
358437ec4e1aSPeter Brune   PetscValidLogicalCollectiveBool(snes,flg,2);
358537ec4e1aSPeter Brune   snes->lagpre_persist = flg;
358637ec4e1aSPeter Brune   PetscFunctionReturn(0);
358737ec4e1aSPeter Brune }
358837ec4e1aSPeter Brune 
35899b94acceSBarry Smith /*@
3590be5caee7SBarry Smith    SNESSetForceIteration - force SNESSolve() to take at least one iteration regardless of the initial residual norm
3591be5caee7SBarry Smith 
3592be5caee7SBarry Smith    Logically Collective on SNES
3593be5caee7SBarry Smith 
3594be5caee7SBarry Smith    Input Parameters:
3595be5caee7SBarry Smith +  snes - the SNES context
3596be5caee7SBarry Smith -  force - PETSC_TRUE require at least one iteration
3597be5caee7SBarry Smith 
3598be5caee7SBarry Smith    Options Database Keys:
3599be5caee7SBarry Smith .    -snes_force_iteration <force> - Sets forcing an iteration
3600be5caee7SBarry Smith 
3601be5caee7SBarry Smith    Notes:
3602be5caee7SBarry Smith    This is used sometimes with TS to prevent TS from detecting a false steady state solution
3603be5caee7SBarry Smith 
3604be5caee7SBarry Smith    Level: intermediate
3605be5caee7SBarry Smith 
3606be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance()
3607be5caee7SBarry Smith @*/
3608be5caee7SBarry Smith PetscErrorCode  SNESSetForceIteration(SNES snes,PetscBool force)
3609be5caee7SBarry Smith {
3610be5caee7SBarry Smith   PetscFunctionBegin;
3611be5caee7SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3612be5caee7SBarry Smith   snes->forceiteration = force;
3613be5caee7SBarry Smith   PetscFunctionReturn(0);
3614be5caee7SBarry Smith }
3615be5caee7SBarry Smith 
361685216dc7SFande Kong /*@
361785216dc7SFande Kong    SNESGetForceIteration - Whether or not to force SNESSolve() take at least one iteration regardless of the initial residual norm
361885216dc7SFande Kong 
361985216dc7SFande Kong    Logically Collective on SNES
362085216dc7SFande Kong 
362185216dc7SFande Kong    Input Parameters:
362285216dc7SFande Kong .  snes - the SNES context
362385216dc7SFande Kong 
362485216dc7SFande Kong    Output Parameter:
362585216dc7SFande Kong .  force - PETSC_TRUE requires at least one iteration.
362685216dc7SFande Kong 
362706dd6b0eSSatish Balay    Level: intermediate
362806dd6b0eSSatish Balay 
362985216dc7SFande Kong .seealso: SNESSetForceIteration(), SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance()
363085216dc7SFande Kong @*/
363185216dc7SFande Kong PetscErrorCode  SNESGetForceIteration(SNES snes,PetscBool *force)
363285216dc7SFande Kong {
363385216dc7SFande Kong   PetscFunctionBegin;
363485216dc7SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
363585216dc7SFande Kong   *force = snes->forceiteration;
363685216dc7SFande Kong   PetscFunctionReturn(0);
363785216dc7SFande Kong }
3638be5caee7SBarry Smith 
3639be5caee7SBarry Smith /*@
3640d7a720efSLois Curfman McInnes    SNESSetTolerances - Sets various parameters used in convergence tests.
36419b94acceSBarry Smith 
36423f9fe445SBarry Smith    Logically Collective on SNES
3643c7afd0dbSLois Curfman McInnes 
36449b94acceSBarry Smith    Input Parameters:
3645c7afd0dbSLois Curfman McInnes +  snes - the SNES context
364670441072SBarry Smith .  abstol - absolute convergence tolerance
364733174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
36485358d0d4SBarry Smith .  stol -  convergence tolerance in terms of the norm of the change in the solution between steps,  || delta x || < stol*|| x ||
364933174efeSLois Curfman McInnes .  maxit - maximum number of iterations
3650e71169deSBarry Smith -  maxf - maximum number of function evaluations (-1 indicates no limit)
3651fee21e36SBarry Smith 
365233174efeSLois Curfman McInnes    Options Database Keys:
365370441072SBarry Smith +    -snes_atol <abstol> - Sets abstol
3654c7afd0dbSLois Curfman McInnes .    -snes_rtol <rtol> - Sets rtol
3655c7afd0dbSLois Curfman McInnes .    -snes_stol <stol> - Sets stol
3656c7afd0dbSLois Curfman McInnes .    -snes_max_it <maxit> - Sets maxit
3657c7afd0dbSLois Curfman McInnes -    -snes_max_funcs <maxf> - Sets maxf
36589b94acceSBarry Smith 
3659d7a720efSLois Curfman McInnes    Notes:
36609b94acceSBarry Smith    The default maximum number of iterations is 50.
36619b94acceSBarry Smith    The default maximum number of function evaluations is 1000.
36629b94acceSBarry Smith 
366336851e7fSLois Curfman McInnes    Level: intermediate
366436851e7fSLois Curfman McInnes 
3665be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance(), SNESSetForceIteration()
36669b94acceSBarry Smith @*/
36677087cfbeSBarry Smith PetscErrorCode  SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf)
36689b94acceSBarry Smith {
36693a40ed3dSBarry Smith   PetscFunctionBegin;
36700700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3671c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,abstol,2);
3672c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol,3);
3673c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,stol,4);
3674c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxit,5);
3675c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxf,6);
3676c5eb9154SBarry Smith 
3677ab54825eSJed Brown   if (abstol != PETSC_DEFAULT) {
367857622a8eSBarry Smith     if (abstol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Absolute tolerance %g must be non-negative",(double)abstol);
3679ab54825eSJed Brown     snes->abstol = abstol;
3680ab54825eSJed Brown   }
3681ab54825eSJed Brown   if (rtol != PETSC_DEFAULT) {
368257622a8eSBarry Smith     if (rtol < 0.0 || 1.0 <= rtol) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Relative tolerance %g must be non-negative and less than 1.0",(double)rtol);
3683ab54825eSJed Brown     snes->rtol = rtol;
3684ab54825eSJed Brown   }
3685ab54825eSJed Brown   if (stol != PETSC_DEFAULT) {
368657622a8eSBarry Smith     if (stol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Step tolerance %g must be non-negative",(double)stol);
3687c60f73f4SPeter Brune     snes->stol = stol;
3688ab54825eSJed Brown   }
3689ab54825eSJed Brown   if (maxit != PETSC_DEFAULT) {
3690ce94432eSBarry Smith     if (maxit < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of iterations %D must be non-negative",maxit);
3691ab54825eSJed Brown     snes->max_its = maxit;
3692ab54825eSJed Brown   }
3693ab54825eSJed Brown   if (maxf != PETSC_DEFAULT) {
3694e71169deSBarry Smith     if (maxf < -1) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of function evaluations %D must be -1 or nonnegative",maxf);
3695ab54825eSJed Brown     snes->max_funcs = maxf;
3696ab54825eSJed Brown   }
369788976e71SPeter Brune   snes->tolerancesset = PETSC_TRUE;
36983a40ed3dSBarry Smith   PetscFunctionReturn(0);
36999b94acceSBarry Smith }
37009b94acceSBarry Smith 
3701e4d06f11SPatrick Farrell /*@
3702e4d06f11SPatrick Farrell    SNESSetDivergenceTolerance - Sets the divergence tolerance used for the SNES divergence test.
3703e4d06f11SPatrick Farrell 
3704e4d06f11SPatrick Farrell    Logically Collective on SNES
3705e4d06f11SPatrick Farrell 
3706e4d06f11SPatrick Farrell    Input Parameters:
3707e4d06f11SPatrick Farrell +  snes - the SNES context
3708e4d06f11SPatrick Farrell -  divtol - the divergence tolerance. Use -1 to deactivate the test.
3709e4d06f11SPatrick Farrell 
3710e4d06f11SPatrick Farrell    Options Database Keys:
3711a2b725a8SWilliam Gropp .    -snes_divergence_tolerance <divtol> - Sets divtol
3712e4d06f11SPatrick Farrell 
3713e4d06f11SPatrick Farrell    Notes:
3714e4d06f11SPatrick Farrell    The default divergence tolerance is 1e4.
3715e4d06f11SPatrick Farrell 
3716e4d06f11SPatrick Farrell    Level: intermediate
3717e4d06f11SPatrick Farrell 
3718e4d06f11SPatrick Farrell .seealso: SNESSetTolerances(), SNESGetDivergenceTolerance
3719e4d06f11SPatrick Farrell @*/
3720e4d06f11SPatrick Farrell PetscErrorCode  SNESSetDivergenceTolerance(SNES snes,PetscReal divtol)
3721e4d06f11SPatrick Farrell {
3722e4d06f11SPatrick Farrell   PetscFunctionBegin;
3723e4d06f11SPatrick Farrell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3724e4d06f11SPatrick Farrell   PetscValidLogicalCollectiveReal(snes,divtol,2);
3725e4d06f11SPatrick Farrell 
3726e4d06f11SPatrick Farrell   if (divtol != PETSC_DEFAULT) {
3727e4d06f11SPatrick Farrell     snes->divtol = divtol;
3728e4d06f11SPatrick Farrell   }
3729e4d06f11SPatrick Farrell   else {
3730e4d06f11SPatrick Farrell     snes->divtol = 1.0e4;
3731e4d06f11SPatrick Farrell   }
3732e4d06f11SPatrick Farrell   PetscFunctionReturn(0);
3733e4d06f11SPatrick Farrell }
3734e4d06f11SPatrick Farrell 
37359b94acceSBarry Smith /*@
373633174efeSLois Curfman McInnes    SNESGetTolerances - Gets various parameters used in convergence tests.
373733174efeSLois Curfman McInnes 
3738c7afd0dbSLois Curfman McInnes    Not Collective
3739c7afd0dbSLois Curfman McInnes 
374033174efeSLois Curfman McInnes    Input Parameters:
3741c7afd0dbSLois Curfman McInnes +  snes - the SNES context
374285385478SLisandro Dalcin .  atol - absolute convergence tolerance
374333174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
374433174efeSLois Curfman McInnes .  stol -  convergence tolerance in terms of the norm
374533174efeSLois Curfman McInnes            of the change in the solution between steps
374633174efeSLois Curfman McInnes .  maxit - maximum number of iterations
3747c7afd0dbSLois Curfman McInnes -  maxf - maximum number of function evaluations
3748fee21e36SBarry Smith 
374933174efeSLois Curfman McInnes    Notes:
37500298fd71SBarry Smith    The user can specify NULL for any parameter that is not needed.
375133174efeSLois Curfman McInnes 
375236851e7fSLois Curfman McInnes    Level: intermediate
375336851e7fSLois Curfman McInnes 
375433174efeSLois Curfman McInnes .seealso: SNESSetTolerances()
375533174efeSLois Curfman McInnes @*/
37567087cfbeSBarry Smith PetscErrorCode  SNESGetTolerances(SNES snes,PetscReal *atol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf)
375733174efeSLois Curfman McInnes {
37583a40ed3dSBarry Smith   PetscFunctionBegin;
37590700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
376085385478SLisandro Dalcin   if (atol)  *atol  = snes->abstol;
376133174efeSLois Curfman McInnes   if (rtol)  *rtol  = snes->rtol;
3762c60f73f4SPeter Brune   if (stol)  *stol  = snes->stol;
376333174efeSLois Curfman McInnes   if (maxit) *maxit = snes->max_its;
376433174efeSLois Curfman McInnes   if (maxf)  *maxf  = snes->max_funcs;
37653a40ed3dSBarry Smith   PetscFunctionReturn(0);
376633174efeSLois Curfman McInnes }
376733174efeSLois Curfman McInnes 
3768e4d06f11SPatrick Farrell /*@
3769e4d06f11SPatrick Farrell    SNESGetDivergenceTolerance - Gets divergence tolerance used in divergence test.
3770e4d06f11SPatrick Farrell 
3771e4d06f11SPatrick Farrell    Not Collective
3772e4d06f11SPatrick Farrell 
3773e4d06f11SPatrick Farrell    Input Parameters:
3774e4d06f11SPatrick Farrell +  snes - the SNES context
3775e4d06f11SPatrick Farrell -  divtol - divergence tolerance
3776e4d06f11SPatrick Farrell 
3777e4d06f11SPatrick Farrell    Level: intermediate
3778e4d06f11SPatrick Farrell 
3779e4d06f11SPatrick Farrell .seealso: SNESSetDivergenceTolerance()
3780e4d06f11SPatrick Farrell @*/
3781e4d06f11SPatrick Farrell PetscErrorCode  SNESGetDivergenceTolerance(SNES snes,PetscReal *divtol)
3782e4d06f11SPatrick Farrell {
3783e4d06f11SPatrick Farrell   PetscFunctionBegin;
3784e4d06f11SPatrick Farrell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3785e4d06f11SPatrick Farrell   if (divtol) *divtol = snes->divtol;
3786e4d06f11SPatrick Farrell   PetscFunctionReturn(0);
3787e4d06f11SPatrick Farrell }
3788e4d06f11SPatrick Farrell 
378933174efeSLois Curfman McInnes /*@
37909b94acceSBarry Smith    SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance.
37919b94acceSBarry Smith 
37923f9fe445SBarry Smith    Logically Collective on SNES
3793fee21e36SBarry Smith 
3794c7afd0dbSLois Curfman McInnes    Input Parameters:
3795c7afd0dbSLois Curfman McInnes +  snes - the SNES context
3796c7afd0dbSLois Curfman McInnes -  tol - tolerance
3797c7afd0dbSLois Curfman McInnes 
37989b94acceSBarry Smith    Options Database Key:
3799c7afd0dbSLois Curfman McInnes .  -snes_trtol <tol> - Sets tol
38009b94acceSBarry Smith 
380136851e7fSLois Curfman McInnes    Level: intermediate
380236851e7fSLois Curfman McInnes 
38032492ecdbSBarry Smith .seealso: SNESSetTolerances()
38049b94acceSBarry Smith @*/
38057087cfbeSBarry Smith PetscErrorCode  SNESSetTrustRegionTolerance(SNES snes,PetscReal tol)
38069b94acceSBarry Smith {
38073a40ed3dSBarry Smith   PetscFunctionBegin;
38080700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3809c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,tol,2);
38109b94acceSBarry Smith   snes->deltatol = tol;
38113a40ed3dSBarry Smith   PetscFunctionReturn(0);
38129b94acceSBarry Smith }
38139b94acceSBarry Smith 
38146ba87a44SLisandro Dalcin PETSC_INTERN PetscErrorCode  SNESMonitorRange_Private(SNES,PetscInt,PetscReal*);
38156ba87a44SLisandro Dalcin 
38167087cfbeSBarry Smith PetscErrorCode  SNESMonitorLGRange(SNES snes,PetscInt n,PetscReal rnorm,void *monctx)
3817b271bb04SBarry Smith {
3818b271bb04SBarry Smith   PetscDrawLG      lg;
3819b271bb04SBarry Smith   PetscErrorCode   ierr;
3820b271bb04SBarry Smith   PetscReal        x,y,per;
3821b271bb04SBarry Smith   PetscViewer      v = (PetscViewer)monctx;
3822b271bb04SBarry Smith   static PetscReal prev; /* should be in the context */
3823b271bb04SBarry Smith   PetscDraw        draw;
3824b271bb04SBarry Smith 
3825459f5d12SBarry Smith   PetscFunctionBegin;
38264d4332d5SBarry Smith   PetscValidHeaderSpecific(v,PETSC_VIEWER_CLASSID,4);
3827b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,0,&lg);CHKERRQ(ierr);
3828b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3829b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3830b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"Residual norm");CHKERRQ(ierr);
3831b271bb04SBarry Smith   x    = (PetscReal)n;
383277b4d14cSPeter Brune   if (rnorm > 0.0) y = PetscLog10Real(rnorm);
383394c9c6d3SKarl Rupp   else y = -15.0;
3834b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
38356934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3836b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
38376934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3838b271bb04SBarry Smith   }
3839b271bb04SBarry Smith 
3840b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,1,&lg);CHKERRQ(ierr);
3841b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3842b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3843b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"% elemts > .2*max elemt");CHKERRQ(ierr);
3844b271bb04SBarry Smith   ierr =  SNESMonitorRange_Private(snes,n,&per);CHKERRQ(ierr);
3845b271bb04SBarry Smith   x    = (PetscReal)n;
3846b271bb04SBarry Smith   y    = 100.0*per;
3847b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
38486934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3849b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
38506934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3851b271bb04SBarry Smith   }
3852b271bb04SBarry Smith 
3853b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,2,&lg);CHKERRQ(ierr);
3854b271bb04SBarry Smith   if (!n) {prev = rnorm;ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3855b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3856b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm");CHKERRQ(ierr);
3857b271bb04SBarry Smith   x    = (PetscReal)n;
3858b271bb04SBarry Smith   y    = (prev - rnorm)/prev;
3859b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
38606934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3861b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
38626934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3863b271bb04SBarry Smith   }
3864b271bb04SBarry Smith 
3865b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,3,&lg);CHKERRQ(ierr);
3866b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3867b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3868b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr);
3869b271bb04SBarry Smith   x    = (PetscReal)n;
3870b271bb04SBarry Smith   y    = (prev - rnorm)/(prev*per);
3871b271bb04SBarry Smith   if (n > 2) { /*skip initial crazy value */
3872b271bb04SBarry Smith     ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
3873b271bb04SBarry Smith   }
38746934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3875b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
38766934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3877b271bb04SBarry Smith   }
3878b271bb04SBarry Smith   prev = rnorm;
3879b271bb04SBarry Smith   PetscFunctionReturn(0);
3880b271bb04SBarry Smith }
3881b271bb04SBarry Smith 
3882228d79bcSJed Brown /*@
3883228d79bcSJed Brown    SNESMonitor - runs the user provided monitor routines, if they exist
3884228d79bcSJed Brown 
3885228d79bcSJed Brown    Collective on SNES
3886228d79bcSJed Brown 
3887228d79bcSJed Brown    Input Parameters:
3888228d79bcSJed Brown +  snes - nonlinear solver context obtained from SNESCreate()
3889228d79bcSJed Brown .  iter - iteration number
3890228d79bcSJed Brown -  rnorm - relative norm of the residual
3891228d79bcSJed Brown 
3892228d79bcSJed Brown    Notes:
3893228d79bcSJed Brown    This routine is called by the SNES implementations.
3894228d79bcSJed Brown    It does not typically need to be called by the user.
3895228d79bcSJed Brown 
3896228d79bcSJed Brown    Level: developer
3897228d79bcSJed Brown 
3898228d79bcSJed Brown .seealso: SNESMonitorSet()
3899228d79bcSJed Brown @*/
39007a03ce2fSLisandro Dalcin PetscErrorCode  SNESMonitor(SNES snes,PetscInt iter,PetscReal rnorm)
39017a03ce2fSLisandro Dalcin {
39027a03ce2fSLisandro Dalcin   PetscErrorCode ierr;
39037a03ce2fSLisandro Dalcin   PetscInt       i,n = snes->numbermonitors;
39047a03ce2fSLisandro Dalcin 
39057a03ce2fSLisandro Dalcin   PetscFunctionBegin;
39068860a134SJunchao Zhang   ierr = VecLockReadPush(snes->vec_sol);CHKERRQ(ierr);
39077a03ce2fSLisandro Dalcin   for (i=0; i<n; i++) {
39087a03ce2fSLisandro Dalcin     ierr = (*snes->monitor[i])(snes,iter,rnorm,snes->monitorcontext[i]);CHKERRQ(ierr);
39097a03ce2fSLisandro Dalcin   }
39108860a134SJunchao Zhang   ierr = VecLockReadPop(snes->vec_sol);CHKERRQ(ierr);
39117a03ce2fSLisandro Dalcin   PetscFunctionReturn(0);
39127a03ce2fSLisandro Dalcin }
39137a03ce2fSLisandro Dalcin 
39149b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */
39159b94acceSBarry Smith 
3916bf388a1fSBarry Smith /*MC
3917bf388a1fSBarry Smith     SNESMonitorFunction - functional form passed to SNESMonitorSet() to monitor convergence of nonlinear solver
3918bf388a1fSBarry Smith 
3919bf388a1fSBarry Smith      Synopsis:
3920aaa7dc30SBarry Smith      #include <petscsnes.h>
3921bf388a1fSBarry Smith $    PetscErrorCode SNESMonitorFunction(SNES snes,PetscInt its, PetscReal norm,void *mctx)
3922bf388a1fSBarry Smith 
39231843f636SBarry Smith      Collective on snes
39241843f636SBarry Smith 
39251843f636SBarry Smith     Input Parameters:
3926bf388a1fSBarry Smith +    snes - the SNES context
3927bf388a1fSBarry Smith .    its - iteration number
3928bf388a1fSBarry Smith .    norm - 2-norm function value (may be estimated)
3929bf388a1fSBarry Smith -    mctx - [optional] monitoring context
3930bf388a1fSBarry Smith 
3931878cb397SSatish Balay    Level: advanced
3932878cb397SSatish Balay 
3933bf388a1fSBarry Smith .seealso:   SNESMonitorSet(), SNESMonitorGet()
3934bf388a1fSBarry Smith M*/
3935bf388a1fSBarry Smith 
39369b94acceSBarry Smith /*@C
3937a6570f20SBarry Smith    SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every
39389b94acceSBarry Smith    iteration of the nonlinear solver to display the iteration's
39399b94acceSBarry Smith    progress.
39409b94acceSBarry Smith 
39413f9fe445SBarry Smith    Logically Collective on SNES
3942fee21e36SBarry Smith 
3943c7afd0dbSLois Curfman McInnes    Input Parameters:
3944c7afd0dbSLois Curfman McInnes +  snes - the SNES context
39456e4dcb14SBarry Smith .  f - the monitor function, see SNESMonitorFunction for the calling sequence
3946b8a78c4aSBarry Smith .  mctx - [optional] user-defined context for private data for the
39470298fd71SBarry Smith           monitor routine (use NULL if no context is desired)
3948b3006f0bSLois Curfman McInnes -  monitordestroy - [optional] routine that frees monitor context
39490298fd71SBarry Smith           (may be NULL)
39509b94acceSBarry Smith 
39519665c990SLois Curfman McInnes    Options Database Keys:
3952a6570f20SBarry Smith +    -snes_monitor        - sets SNESMonitorDefault()
3953798534f6SMatthew G. Knepley .    -snes_monitor draw::draw_lg - sets line graph monitor,
3954cca6129bSJed Brown -    -snes_monitor_cancel - cancels all monitors that have
3955c7afd0dbSLois Curfman McInnes                             been hardwired into a code by
3956a6570f20SBarry Smith                             calls to SNESMonitorSet(), but
3957c7afd0dbSLois Curfman McInnes                             does not cancel those set via
3958c7afd0dbSLois Curfman McInnes                             the options database.
39599665c990SLois Curfman McInnes 
3960639f9d9dSBarry Smith    Notes:
39616bc08f3fSLois Curfman McInnes    Several different monitoring routines may be set by calling
3962a6570f20SBarry Smith    SNESMonitorSet() multiple times; all will be called in the
39636bc08f3fSLois Curfman McInnes    order in which they were set.
3964639f9d9dSBarry Smith 
396595452b02SPatrick Sanan    Fortran Notes:
396695452b02SPatrick Sanan     Only a single monitor function can be set for each SNES object
3967025f1a04SBarry Smith 
396836851e7fSLois Curfman McInnes    Level: intermediate
396936851e7fSLois Curfman McInnes 
3970bf388a1fSBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorCancel(), SNESMonitorFunction
39719b94acceSBarry Smith @*/
39726e4dcb14SBarry Smith PetscErrorCode  SNESMonitorSet(SNES snes,PetscErrorCode (*f)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void**))
39739b94acceSBarry Smith {
3974b90d0a6eSBarry Smith   PetscInt       i;
3975649052a6SBarry Smith   PetscErrorCode ierr;
397678064530SBarry Smith   PetscBool      identical;
3977b90d0a6eSBarry Smith 
39783a40ed3dSBarry Smith   PetscFunctionBegin;
39790700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3980b90d0a6eSBarry Smith   for (i=0; i<snes->numbermonitors;i++) {
398178064530SBarry Smith     ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,mctx,monitordestroy,(PetscErrorCode (*)(void))snes->monitor[i],snes->monitorcontext[i],snes->monitordestroy[i],&identical);CHKERRQ(ierr);
398278064530SBarry Smith     if (identical) PetscFunctionReturn(0);
3983649052a6SBarry Smith   }
398478064530SBarry Smith   if (snes->numbermonitors >= MAXSNESMONITORS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set");
39856e4dcb14SBarry Smith   snes->monitor[snes->numbermonitors]          = f;
3986b8a78c4aSBarry Smith   snes->monitordestroy[snes->numbermonitors]   = monitordestroy;
3987639f9d9dSBarry Smith   snes->monitorcontext[snes->numbermonitors++] = (void*)mctx;
39883a40ed3dSBarry Smith   PetscFunctionReturn(0);
39899b94acceSBarry Smith }
39909b94acceSBarry Smith 
3991a278d85bSSatish Balay /*@
3992a6570f20SBarry Smith    SNESMonitorCancel - Clears all the monitor functions for a SNES object.
39935cd90555SBarry Smith 
39943f9fe445SBarry Smith    Logically Collective on SNES
3995c7afd0dbSLois Curfman McInnes 
39965cd90555SBarry Smith    Input Parameters:
39975cd90555SBarry Smith .  snes - the SNES context
39985cd90555SBarry Smith 
39991a480d89SAdministrator    Options Database Key:
4000a6570f20SBarry Smith .  -snes_monitor_cancel - cancels all monitors that have been hardwired
4001a6570f20SBarry Smith     into a code by calls to SNESMonitorSet(), but does not cancel those
4002c7afd0dbSLois Curfman McInnes     set via the options database
40035cd90555SBarry Smith 
40045cd90555SBarry Smith    Notes:
40055cd90555SBarry Smith    There is no way to clear one specific monitor from a SNES object.
40065cd90555SBarry Smith 
400736851e7fSLois Curfman McInnes    Level: intermediate
400836851e7fSLois Curfman McInnes 
4009a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorSet()
40105cd90555SBarry Smith @*/
40117087cfbeSBarry Smith PetscErrorCode  SNESMonitorCancel(SNES snes)
40125cd90555SBarry Smith {
4013d952e501SBarry Smith   PetscErrorCode ierr;
4014d952e501SBarry Smith   PetscInt       i;
4015d952e501SBarry Smith 
40165cd90555SBarry Smith   PetscFunctionBegin;
40170700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4018d952e501SBarry Smith   for (i=0; i<snes->numbermonitors; i++) {
4019d952e501SBarry Smith     if (snes->monitordestroy[i]) {
40203c4aec1bSBarry Smith       ierr = (*snes->monitordestroy[i])(&snes->monitorcontext[i]);CHKERRQ(ierr);
4021d952e501SBarry Smith     }
4022d952e501SBarry Smith   }
40235cd90555SBarry Smith   snes->numbermonitors = 0;
40245cd90555SBarry Smith   PetscFunctionReturn(0);
40255cd90555SBarry Smith }
40265cd90555SBarry Smith 
4027bf388a1fSBarry Smith /*MC
4028bf388a1fSBarry Smith     SNESConvergenceTestFunction - functional form used for testing of convergence of nonlinear solver
4029bf388a1fSBarry Smith 
4030bf388a1fSBarry Smith      Synopsis:
4031aaa7dc30SBarry Smith      #include <petscsnes.h>
4032bf388a1fSBarry Smith $     PetscErrorCode SNESConvergenceTest(SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx)
4033bf388a1fSBarry Smith 
40341843f636SBarry Smith      Collective on snes
40351843f636SBarry Smith 
40361843f636SBarry Smith     Input Parameters:
4037bf388a1fSBarry Smith +    snes - the SNES context
4038bf388a1fSBarry Smith .    it - current iteration (0 is the first and is before any Newton step)
4039bf388a1fSBarry Smith .    xnorm - 2-norm of current iterate
4040bf388a1fSBarry Smith .    gnorm - 2-norm of current step
40411843f636SBarry Smith .    f - 2-norm of function
40421843f636SBarry Smith -    cctx - [optional] convergence context
40431843f636SBarry Smith 
40441843f636SBarry Smith     Output Parameter:
40451843f636SBarry Smith .    reason - reason for convergence/divergence, only needs to be set when convergence or divergence is detected
4046bf388a1fSBarry Smith 
4047878cb397SSatish Balay    Level: intermediate
4048bf388a1fSBarry Smith 
4049bf388a1fSBarry Smith .seealso:   SNESSetConvergenceTest(), SNESGetConvergenceTest()
4050bf388a1fSBarry Smith M*/
4051bf388a1fSBarry Smith 
40529b94acceSBarry Smith /*@C
40539b94acceSBarry Smith    SNESSetConvergenceTest - Sets the function that is to be used
40549b94acceSBarry Smith    to test for convergence of the nonlinear iterative solution.
40559b94acceSBarry Smith 
40563f9fe445SBarry Smith    Logically Collective on SNES
4057fee21e36SBarry Smith 
4058c7afd0dbSLois Curfman McInnes    Input Parameters:
4059c7afd0dbSLois Curfman McInnes +  snes - the SNES context
4060bf388a1fSBarry Smith .  SNESConvergenceTestFunction - routine to test for convergence
40610298fd71SBarry Smith .  cctx - [optional] context for private data for the convergence routine  (may be NULL)
4062cf90aa19SBarry Smith -  destroy - [optional] destructor for the context (may be NULL; PETSC_NULL_FUNCTION in Fortran)
40639b94acceSBarry Smith 
406436851e7fSLois Curfman McInnes    Level: advanced
406536851e7fSLois Curfman McInnes 
4066e2a6519dSDmitry Karpeev .seealso: SNESConvergedDefault(), SNESConvergedSkip(), SNESConvergenceTestFunction
40679b94acceSBarry Smith @*/
4068bf388a1fSBarry Smith PetscErrorCode  SNESSetConvergenceTest(SNES snes,PetscErrorCode (*SNESConvergenceTestFunction)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx,PetscErrorCode (*destroy)(void*))
40699b94acceSBarry Smith {
40707f7931b9SBarry Smith   PetscErrorCode ierr;
40717f7931b9SBarry Smith 
40723a40ed3dSBarry Smith   PetscFunctionBegin;
40730700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4074e2a6519dSDmitry Karpeev   if (!SNESConvergenceTestFunction) SNESConvergenceTestFunction = SNESConvergedSkip;
40757f7931b9SBarry Smith   if (snes->ops->convergeddestroy) {
40767f7931b9SBarry Smith     ierr = (*snes->ops->convergeddestroy)(snes->cnvP);CHKERRQ(ierr);
40777f7931b9SBarry Smith   }
4078bf388a1fSBarry Smith   snes->ops->converged        = SNESConvergenceTestFunction;
40797f7931b9SBarry Smith   snes->ops->convergeddestroy = destroy;
408085385478SLisandro Dalcin   snes->cnvP                  = cctx;
40813a40ed3dSBarry Smith   PetscFunctionReturn(0);
40829b94acceSBarry Smith }
40839b94acceSBarry Smith 
408452baeb72SSatish Balay /*@
4085184914b5SBarry Smith    SNESGetConvergedReason - Gets the reason the SNES iteration was stopped.
4086184914b5SBarry Smith 
4087184914b5SBarry Smith    Not Collective
4088184914b5SBarry Smith 
4089184914b5SBarry Smith    Input Parameter:
4090184914b5SBarry Smith .  snes - the SNES context
4091184914b5SBarry Smith 
4092184914b5SBarry Smith    Output Parameter:
40934d0a8057SBarry Smith .  reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the
4094184914b5SBarry Smith             manual pages for the individual convergence tests for complete lists
4095184914b5SBarry Smith 
40966a4d7782SBarry Smith    Options Database:
40976a4d7782SBarry Smith .   -snes_converged_reason - prints the reason to standard out
40986a4d7782SBarry Smith 
4099184914b5SBarry Smith    Level: intermediate
4100184914b5SBarry Smith 
410195452b02SPatrick Sanan    Notes:
410295452b02SPatrick Sanan     Should only be called after the call the SNESSolve() is complete, if it is called earlier it returns the value SNES__CONVERGED_ITERATING.
4103184914b5SBarry Smith 
410433866048SMatthew G. Knepley .seealso: SNESSetConvergenceTest(), SNESSetConvergedReason(), SNESConvergedReason
4105184914b5SBarry Smith @*/
41067087cfbeSBarry Smith PetscErrorCode SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason)
4107184914b5SBarry Smith {
4108184914b5SBarry Smith   PetscFunctionBegin;
41090700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
41104482741eSBarry Smith   PetscValidPointer(reason,2);
4111184914b5SBarry Smith   *reason = snes->reason;
4112184914b5SBarry Smith   PetscFunctionReturn(0);
4113184914b5SBarry Smith }
4114184914b5SBarry Smith 
4115c4421ceaSFande Kong /*@C
4116c4421ceaSFande Kong    SNESGetConvergedReasonString - Return a human readable string for snes converged reason
4117c4421ceaSFande Kong 
4118c4421ceaSFande Kong    Not Collective
4119c4421ceaSFande Kong 
4120c4421ceaSFande Kong    Input Parameter:
4121c4421ceaSFande Kong .  snes - the SNES context
4122c4421ceaSFande Kong 
4123c4421ceaSFande Kong    Output Parameter:
4124c4421ceaSFande Kong .  strreason - a human readable string that describes SNES converged reason
4125c4421ceaSFande Kong 
412699c90e12SSatish Balay    Level: beginner
4127c4421ceaSFande Kong 
4128c4421ceaSFande Kong .seealso: SNESGetConvergedReason()
4129c4421ceaSFande Kong @*/
4130c4421ceaSFande Kong PetscErrorCode SNESGetConvergedReasonString(SNES snes, const char** strreason)
4131c4421ceaSFande Kong {
4132c4421ceaSFande Kong   PetscFunctionBegin;
4133c4421ceaSFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4134c4421ceaSFande Kong   PetscValidCharPointer(strreason,2);
4135c4421ceaSFande Kong   *strreason = SNESConvergedReasons[snes->reason];
4136c4421ceaSFande Kong   PetscFunctionReturn(0);
4137c4421ceaSFande Kong }
4138c4421ceaSFande Kong 
413933866048SMatthew G. Knepley /*@
414033866048SMatthew G. Knepley    SNESSetConvergedReason - Sets the reason the SNES iteration was stopped.
414133866048SMatthew G. Knepley 
414233866048SMatthew G. Knepley    Not Collective
414333866048SMatthew G. Knepley 
414433866048SMatthew G. Knepley    Input Parameters:
414533866048SMatthew G. Knepley +  snes - the SNES context
414633866048SMatthew G. Knepley -  reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the
414733866048SMatthew G. Knepley             manual pages for the individual convergence tests for complete lists
414833866048SMatthew G. Knepley 
414933866048SMatthew G. Knepley    Level: intermediate
415033866048SMatthew G. Knepley 
415133866048SMatthew G. Knepley .seealso: SNESGetConvergedReason(), SNESSetConvergenceTest(), SNESConvergedReason
415233866048SMatthew G. Knepley @*/
415333866048SMatthew G. Knepley PetscErrorCode SNESSetConvergedReason(SNES snes,SNESConvergedReason reason)
415433866048SMatthew G. Knepley {
415533866048SMatthew G. Knepley   PetscFunctionBegin;
415633866048SMatthew G. Knepley   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
415733866048SMatthew G. Knepley   snes->reason = reason;
415833866048SMatthew G. Knepley   PetscFunctionReturn(0);
415933866048SMatthew G. Knepley }
416033866048SMatthew G. Knepley 
4161c9005455SLois Curfman McInnes /*@
4162c9005455SLois Curfman McInnes    SNESSetConvergenceHistory - Sets the array used to hold the convergence history.
4163c9005455SLois Curfman McInnes 
41643f9fe445SBarry Smith    Logically Collective on SNES
4165fee21e36SBarry Smith 
4166c7afd0dbSLois Curfman McInnes    Input Parameters:
4167c7afd0dbSLois Curfman McInnes +  snes - iterative context obtained from SNESCreate()
41688c7482ecSBarry Smith .  a   - array to hold history, this array will contain the function norms computed at each step
4169cd5578b5SBarry Smith .  its - integer array holds the number of linear iterations for each solve.
4170758f92a0SBarry Smith .  na  - size of a and its
417164731454SLois Curfman McInnes -  reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero,
4172758f92a0SBarry Smith            else it continues storing new values for new nonlinear solves after the old ones
4173c7afd0dbSLois Curfman McInnes 
4174308dcc3eSBarry Smith    Notes:
41750298fd71SBarry Smith    If 'a' and 'its' are NULL then space is allocated for the history. If 'na' PETSC_DECIDE or PETSC_DEFAULT then a
4176308dcc3eSBarry Smith    default array of length 10000 is allocated.
4177308dcc3eSBarry Smith 
4178c9005455SLois Curfman McInnes    This routine is useful, e.g., when running a code for purposes
4179c9005455SLois Curfman McInnes    of accurate performance monitoring, when no I/O should be done
4180c9005455SLois Curfman McInnes    during the section of code that is being timed.
4181c9005455SLois Curfman McInnes 
418236851e7fSLois Curfman McInnes    Level: intermediate
418336851e7fSLois Curfman McInnes 
418408405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory()
4185758f92a0SBarry Smith 
4186c9005455SLois Curfman McInnes @*/
41877087cfbeSBarry Smith PetscErrorCode  SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscBool reset)
4188c9005455SLois Curfman McInnes {
4189308dcc3eSBarry Smith   PetscErrorCode ierr;
4190308dcc3eSBarry Smith 
41913a40ed3dSBarry Smith   PetscFunctionBegin;
41920700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
41937a1ec6d4SBarry Smith   if (a) PetscValidScalarPointer(a,2);
4194a562a398SLisandro Dalcin   if (its) PetscValidIntPointer(its,3);
41957a1ec6d4SBarry Smith   if (!a) {
4196308dcc3eSBarry Smith     if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
4197071fcb05SBarry Smith     ierr = PetscCalloc2(na,&a,na,&its);CHKERRQ(ierr);
4198071fcb05SBarry Smith     snes->conv_hist_alloc = PETSC_TRUE;
4199308dcc3eSBarry Smith   }
4200c9005455SLois Curfman McInnes   snes->conv_hist       = a;
4201758f92a0SBarry Smith   snes->conv_hist_its   = its;
4202758f92a0SBarry Smith   snes->conv_hist_max   = na;
4203a12bc48fSLisandro Dalcin   snes->conv_hist_len   = 0;
4204758f92a0SBarry Smith   snes->conv_hist_reset = reset;
4205758f92a0SBarry Smith   PetscFunctionReturn(0);
4206758f92a0SBarry Smith }
4207758f92a0SBarry Smith 
4208308dcc3eSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
4209c6db04a5SJed Brown #include <engine.h>   /* MATLAB include file */
4210c6db04a5SJed Brown #include <mex.h>      /* MATLAB include file */
421199e0435eSBarry Smith 
42128cc058d9SJed Brown PETSC_EXTERN mxArray *SNESGetConvergenceHistoryMatlab(SNES snes)
4213308dcc3eSBarry Smith {
4214308dcc3eSBarry Smith   mxArray   *mat;
4215308dcc3eSBarry Smith   PetscInt  i;
4216308dcc3eSBarry Smith   PetscReal *ar;
4217308dcc3eSBarry Smith 
4218308dcc3eSBarry Smith   PetscFunctionBegin;
4219308dcc3eSBarry Smith   mat = mxCreateDoubleMatrix(snes->conv_hist_len,1,mxREAL);
4220308dcc3eSBarry Smith   ar  = (PetscReal*) mxGetData(mat);
4221f5af7f23SKarl Rupp   for (i=0; i<snes->conv_hist_len; i++) ar[i] = snes->conv_hist[i];
4222308dcc3eSBarry Smith   PetscFunctionReturn(mat);
4223308dcc3eSBarry Smith }
4224308dcc3eSBarry Smith #endif
4225308dcc3eSBarry Smith 
42260c4c9dddSBarry Smith /*@C
4227758f92a0SBarry Smith    SNESGetConvergenceHistory - Gets the array used to hold the convergence history.
4228758f92a0SBarry Smith 
42293f9fe445SBarry Smith    Not Collective
4230758f92a0SBarry Smith 
4231758f92a0SBarry Smith    Input Parameter:
4232758f92a0SBarry Smith .  snes - iterative context obtained from SNESCreate()
4233758f92a0SBarry Smith 
4234758f92a0SBarry Smith    Output Parameters:
4235a2b725a8SWilliam Gropp +  a   - array to hold history
4236758f92a0SBarry Smith .  its - integer array holds the number of linear iterations (or
4237758f92a0SBarry Smith          negative if not converged) for each solve.
4238758f92a0SBarry Smith -  na  - size of a and its
4239758f92a0SBarry Smith 
4240758f92a0SBarry Smith    Notes:
4241758f92a0SBarry Smith     The calling sequence for this routine in Fortran is
4242758f92a0SBarry Smith $   call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr)
4243758f92a0SBarry Smith 
4244758f92a0SBarry Smith    This routine is useful, e.g., when running a code for purposes
4245758f92a0SBarry Smith    of accurate performance monitoring, when no I/O should be done
4246758f92a0SBarry Smith    during the section of code that is being timed.
4247758f92a0SBarry Smith 
4248758f92a0SBarry Smith    Level: intermediate
4249758f92a0SBarry Smith 
42505ea7661aSPierre Jolivet .seealso: SNESSetConvergenceHistory()
4251758f92a0SBarry Smith 
4252758f92a0SBarry Smith @*/
42537087cfbeSBarry Smith PetscErrorCode  SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na)
4254758f92a0SBarry Smith {
4255758f92a0SBarry Smith   PetscFunctionBegin;
42560700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4257758f92a0SBarry Smith   if (a)   *a   = snes->conv_hist;
4258758f92a0SBarry Smith   if (its) *its = snes->conv_hist_its;
4259758f92a0SBarry Smith   if (na)  *na  = snes->conv_hist_len;
42603a40ed3dSBarry Smith   PetscFunctionReturn(0);
4261c9005455SLois Curfman McInnes }
4262c9005455SLois Curfman McInnes 
4263ac226902SBarry Smith /*@C
426476b2cf59SMatthew Knepley   SNESSetUpdate - Sets the general-purpose update function called
4265eec8f646SJed Brown   at the beginning of every iteration of the nonlinear solve. Specifically
42667e4bb74cSBarry Smith   it is called just before the Jacobian is "evaluated".
426776b2cf59SMatthew Knepley 
42683f9fe445SBarry Smith   Logically Collective on SNES
426976b2cf59SMatthew Knepley 
427076b2cf59SMatthew Knepley   Input Parameters:
4271a2b725a8SWilliam Gropp + snes - The nonlinear solver context
4272a2b725a8SWilliam Gropp - func - The function
427376b2cf59SMatthew Knepley 
427476b2cf59SMatthew Knepley   Calling sequence of func:
4275a2b725a8SWilliam Gropp $ func (SNES snes, PetscInt step);
427676b2cf59SMatthew Knepley 
427776b2cf59SMatthew Knepley . step - The current step of the iteration
427876b2cf59SMatthew Knepley 
4279fe97e370SBarry Smith   Level: advanced
4280fe97e370SBarry Smith 
4281fe97e370SBarry 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()
4282fe97e370SBarry Smith         This is not used by most users.
428376b2cf59SMatthew Knepley 
42848d359177SBarry Smith .seealso SNESSetJacobian(), SNESSolve()
428576b2cf59SMatthew Knepley @*/
42867087cfbeSBarry Smith PetscErrorCode  SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt))
428776b2cf59SMatthew Knepley {
428876b2cf59SMatthew Knepley   PetscFunctionBegin;
42890700a824SBarry Smith   PetscValidHeaderSpecific(snes, SNES_CLASSID,1);
4290e7788613SBarry Smith   snes->ops->update = func;
429176b2cf59SMatthew Knepley   PetscFunctionReturn(0);
429276b2cf59SMatthew Knepley }
429376b2cf59SMatthew Knepley 
42949b94acceSBarry Smith /*
42959b94acceSBarry Smith    SNESScaleStep_Private - Scales a step so that its length is less than the
42969b94acceSBarry Smith    positive parameter delta.
42979b94acceSBarry Smith 
42989b94acceSBarry Smith     Input Parameters:
4299c7afd0dbSLois Curfman McInnes +   snes - the SNES context
43009b94acceSBarry Smith .   y - approximate solution of linear system
43019b94acceSBarry Smith .   fnorm - 2-norm of current function
4302c7afd0dbSLois Curfman McInnes -   delta - trust region size
43039b94acceSBarry Smith 
43049b94acceSBarry Smith     Output Parameters:
4305c7afd0dbSLois Curfman McInnes +   gpnorm - predicted function norm at the new point, assuming local
43069b94acceSBarry Smith     linearization.  The value is zero if the step lies within the trust
43079b94acceSBarry Smith     region, and exceeds zero otherwise.
4308c7afd0dbSLois Curfman McInnes -   ynorm - 2-norm of the step
43099b94acceSBarry Smith 
43109b94acceSBarry Smith     Note:
431104d7464bSBarry Smith     For non-trust region methods such as SNESNEWTONLS, the parameter delta
43129b94acceSBarry Smith     is set to be the maximum allowable step size.
43139b94acceSBarry Smith 
43149b94acceSBarry Smith */
4315dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm)
43169b94acceSBarry Smith {
4317064f8208SBarry Smith   PetscReal      nrm;
4318ea709b57SSatish Balay   PetscScalar    cnorm;
4319dfbe8321SBarry Smith   PetscErrorCode ierr;
43203a40ed3dSBarry Smith 
43213a40ed3dSBarry Smith   PetscFunctionBegin;
43220700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
43230700a824SBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,2);
4324c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,2);
4325184914b5SBarry Smith 
4326064f8208SBarry Smith   ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr);
4327064f8208SBarry Smith   if (nrm > *delta) {
4328064f8208SBarry Smith     nrm     = *delta/nrm;
4329064f8208SBarry Smith     *gpnorm = (1.0 - nrm)*(*fnorm);
4330064f8208SBarry Smith     cnorm   = nrm;
43312dcb1b2aSMatthew Knepley     ierr    = VecScale(y,cnorm);CHKERRQ(ierr);
43329b94acceSBarry Smith     *ynorm  = *delta;
43339b94acceSBarry Smith   } else {
43349b94acceSBarry Smith     *gpnorm = 0.0;
4335064f8208SBarry Smith     *ynorm  = nrm;
43369b94acceSBarry Smith   }
43373a40ed3dSBarry Smith   PetscFunctionReturn(0);
43389b94acceSBarry Smith }
43399b94acceSBarry Smith 
434091f3e32bSBarry Smith /*@C
434119a666eeSBarry Smith    SNESConvergedReasonView - Displays the reason a SNES solve converged or diverged to a viewer
43422a359c20SBarry Smith 
43432a359c20SBarry Smith    Collective on SNES
43442a359c20SBarry Smith 
43452a359c20SBarry Smith    Parameter:
43462a359c20SBarry Smith +  snes - iterative context obtained from SNESCreate()
43472a359c20SBarry Smith -  viewer - the viewer to display the reason
43482a359c20SBarry Smith 
43492a359c20SBarry Smith 
43502a359c20SBarry Smith    Options Database Keys:
4351ee300463SSatish Balay +  -snes_converged_reason - print reason for converged or diverged, also prints number of iterations
4352ee300463SSatish Balay -  -snes_converged_reason ::failed - only print reason and number of iterations when diverged
4353eafd5ff0SAlex Lindsay 
435419a666eeSBarry Smith   Notes:
435519a666eeSBarry Smith      To change the format of the output call PetscViewerPushFormat(viewer,format) before this call. Use PETSC_VIEWER_DEFAULT for the default,
435619a666eeSBarry Smith      use PETSC_VIEWER_FAILED to only display a reason if it fails.
43572a359c20SBarry Smith 
43582a359c20SBarry Smith    Level: beginner
43592a359c20SBarry Smith 
436019a666eeSBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault(), SNESGetConvergedReason(), SNESConvergedReasonViewFromOptions(),
436119a666eeSBarry Smith           PetscViewerPushFormat(), PetscViewerPopFormat()
43622a359c20SBarry Smith 
43632a359c20SBarry Smith @*/
436419a666eeSBarry Smith PetscErrorCode  SNESConvergedReasonView(SNES snes,PetscViewer viewer)
43652a359c20SBarry Smith {
436675cca76cSMatthew G. Knepley   PetscViewerFormat format;
43672a359c20SBarry Smith   PetscBool         isAscii;
436875cca76cSMatthew G. Knepley   PetscErrorCode    ierr;
43692a359c20SBarry Smith 
43702a359c20SBarry Smith   PetscFunctionBegin;
437119a666eeSBarry Smith   if (!viewer) viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes));
43722a359c20SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isAscii);CHKERRQ(ierr);
43732a359c20SBarry Smith   if (isAscii) {
437475cca76cSMatthew G. Knepley     ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
43752a359c20SBarry Smith     ierr = PetscViewerASCIIAddTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr);
437675cca76cSMatthew G. Knepley     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
437775cca76cSMatthew G. Knepley       DM              dm;
437875cca76cSMatthew G. Knepley       Vec             u;
437975cca76cSMatthew G. Knepley       PetscDS         prob;
438075cca76cSMatthew G. Knepley       PetscInt        Nf, f;
438195cbbfd3SMatthew G. Knepley       PetscErrorCode (**exactSol)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar[], void *);
438295cbbfd3SMatthew G. Knepley       void            **exactCtx;
438375cca76cSMatthew G. Knepley       PetscReal       error;
438475cca76cSMatthew G. Knepley 
438575cca76cSMatthew G. Knepley       ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
438675cca76cSMatthew G. Knepley       ierr = SNESGetSolution(snes, &u);CHKERRQ(ierr);
438775cca76cSMatthew G. Knepley       ierr = DMGetDS(dm, &prob);CHKERRQ(ierr);
438875cca76cSMatthew G. Knepley       ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr);
438995cbbfd3SMatthew G. Knepley       ierr = PetscMalloc2(Nf, &exactSol, Nf, &exactCtx);CHKERRQ(ierr);
439095cbbfd3SMatthew G. Knepley       for (f = 0; f < Nf; ++f) {ierr = PetscDSGetExactSolution(prob, f, &exactSol[f], &exactCtx[f]);CHKERRQ(ierr);}
439195cbbfd3SMatthew G. Knepley       ierr = DMComputeL2Diff(dm, 0.0, exactSol, exactCtx, u, &error);CHKERRQ(ierr);
439295cbbfd3SMatthew G. Knepley       ierr = PetscFree2(exactSol, exactCtx);CHKERRQ(ierr);
439375cca76cSMatthew G. Knepley       if (error < 1.0e-11) {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: < 1.0e-11\n");CHKERRQ(ierr);}
439475cca76cSMatthew G. Knepley       else                 {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: %g\n", error);CHKERRQ(ierr);}
439575cca76cSMatthew G. Knepley     }
4396eafd5ff0SAlex Lindsay     if (snes->reason > 0 && format != PETSC_VIEWER_FAILED) {
43972a359c20SBarry Smith       if (((PetscObject) snes)->prefix) {
43982a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve converged due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
43992a359c20SBarry Smith       } else {
44002a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve converged due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
44012a359c20SBarry Smith       }
4402eafd5ff0SAlex Lindsay     } else if (snes->reason <= 0) {
44032a359c20SBarry Smith       if (((PetscObject) snes)->prefix) {
44042a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve did not converge due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
44052a359c20SBarry Smith       } else {
44062a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve did not converge due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
44072a359c20SBarry Smith       }
44082a359c20SBarry Smith     }
44092a359c20SBarry Smith     ierr = PetscViewerASCIISubtractTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr);
44102a359c20SBarry Smith   }
44112a359c20SBarry Smith   PetscFunctionReturn(0);
44122a359c20SBarry Smith }
44132a359c20SBarry Smith 
4414c4421ceaSFande Kong /*@C
4415c4421ceaSFande Kong    SNESConvergedReasonViewSet - Sets an ADDITIONAL function that is to be used at the
4416c4421ceaSFande Kong     end of the nonlinear solver to display the conver reason of the nonlinear solver.
4417c4421ceaSFande Kong 
4418c4421ceaSFande Kong    Logically Collective on SNES
4419c4421ceaSFande Kong 
4420c4421ceaSFande Kong    Input Parameters:
4421c4421ceaSFande Kong +  snes - the SNES context
4422c4421ceaSFande Kong .  f - the snes converged reason view function
4423c4421ceaSFande Kong .  vctx - [optional] user-defined context for private data for the
4424c4421ceaSFande Kong           snes converged reason view routine (use NULL if no context is desired)
4425c4421ceaSFande Kong -  reasonviewdestroy - [optional] routine that frees reasonview context
4426c4421ceaSFande Kong           (may be NULL)
4427c4421ceaSFande Kong 
4428c4421ceaSFande Kong    Options Database Keys:
4429c4421ceaSFande Kong +    -snes_converged_reason        - sets a default SNESConvergedReasonView()
4430c4421ceaSFande Kong -    -snes_converged_reason_view_cancel - cancels all converged reason viewers that have
4431c4421ceaSFande Kong                             been hardwired into a code by
4432c4421ceaSFande Kong                             calls to SNESConvergedReasonViewSet(), but
4433c4421ceaSFande Kong                             does not cancel those set via
4434c4421ceaSFande Kong                             the options database.
4435c4421ceaSFande Kong 
4436c4421ceaSFande Kong    Notes:
4437c4421ceaSFande Kong    Several different converged reason view routines may be set by calling
4438c4421ceaSFande Kong    SNESConvergedReasonViewSet() multiple times; all will be called in the
4439c4421ceaSFande Kong    order in which they were set.
4440c4421ceaSFande Kong 
4441c4421ceaSFande Kong    Level: intermediate
4442c4421ceaSFande Kong 
4443c4421ceaSFande Kong .seealso: SNESConvergedReasonView(), SNESConvergedReasonViewCancel()
4444c4421ceaSFande Kong @*/
4445c4421ceaSFande Kong PetscErrorCode  SNESConvergedReasonViewSet(SNES snes,PetscErrorCode (*f)(SNES,void*),void *vctx,PetscErrorCode (*reasonviewdestroy)(void**))
4446c4421ceaSFande Kong {
4447c4421ceaSFande Kong   PetscInt       i;
4448c4421ceaSFande Kong   PetscErrorCode ierr;
4449c4421ceaSFande Kong   PetscBool      identical;
4450c4421ceaSFande Kong 
4451c4421ceaSFande Kong   PetscFunctionBegin;
4452c4421ceaSFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4453c4421ceaSFande Kong   for (i=0; i<snes->numberreasonviews;i++) {
4454c4421ceaSFande Kong     ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,vctx,reasonviewdestroy,(PetscErrorCode (*)(void))snes->reasonview[i],snes->reasonviewcontext[i],snes->reasonviewdestroy[i],&identical);CHKERRQ(ierr);
4455c4421ceaSFande Kong     if (identical) PetscFunctionReturn(0);
4456c4421ceaSFande Kong   }
4457c4421ceaSFande Kong   if (snes->numberreasonviews >= MAXSNESREASONVIEWS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many SNES reasonview set");
4458c4421ceaSFande Kong   snes->reasonview[snes->numberreasonviews]          = f;
4459c4421ceaSFande Kong   snes->reasonviewdestroy[snes->numberreasonviews]   = reasonviewdestroy;
4460c4421ceaSFande Kong   snes->reasonviewcontext[snes->numberreasonviews++] = (void*)vctx;
4461c4421ceaSFande Kong   PetscFunctionReturn(0);
4462c4421ceaSFande Kong }
4463c4421ceaSFande Kong 
446491f3e32bSBarry Smith /*@
446519a666eeSBarry Smith   SNESConvergedReasonViewFromOptions - Processes command line options to determine if/how a SNESReason is to be viewed.
4466c4421ceaSFande Kong                                        All the user-provided convergedReasonView routines will be involved as well, if they exist.
44672a359c20SBarry Smith 
44682a359c20SBarry Smith   Collective on SNES
44692a359c20SBarry Smith 
44702a359c20SBarry Smith   Input Parameters:
44712a359c20SBarry Smith . snes   - the SNES object
44722a359c20SBarry Smith 
44732a359c20SBarry Smith   Level: intermediate
44742a359c20SBarry Smith 
447519a666eeSBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault(), SNESGetConvergedReason(), SNESConvergedReasonView()
447619a666eeSBarry Smith 
44772a359c20SBarry Smith @*/
447819a666eeSBarry Smith PetscErrorCode SNESConvergedReasonViewFromOptions(SNES snes)
44792a359c20SBarry Smith {
44802a359c20SBarry Smith   PetscErrorCode    ierr;
44812a359c20SBarry Smith   PetscViewer       viewer;
44822a359c20SBarry Smith   PetscBool         flg;
44832a359c20SBarry Smith   static PetscBool  incall = PETSC_FALSE;
44842a359c20SBarry Smith   PetscViewerFormat format;
4485c4421ceaSFande Kong   PetscInt          i;
44862a359c20SBarry Smith 
44872a359c20SBarry Smith   PetscFunctionBegin;
44882a359c20SBarry Smith   if (incall) PetscFunctionReturn(0);
44892a359c20SBarry Smith   incall = PETSC_TRUE;
4490c4421ceaSFande Kong 
4491c4421ceaSFande Kong   /* All user-provided viewers are called first, if they exist. */
4492c4421ceaSFande Kong   for (i=0; i<snes->numberreasonviews; i++) {
4493c4421ceaSFande Kong     ierr = (*snes->reasonview[i])(snes,snes->reasonviewcontext[i]);CHKERRQ(ierr);
4494c4421ceaSFande Kong   }
4495c4421ceaSFande Kong 
4496c4421ceaSFande Kong   /* Call PETSc default routine if users ask for it */
449716413a6aSBarry Smith   ierr   = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_converged_reason",&viewer,&format,&flg);CHKERRQ(ierr);
44982a359c20SBarry Smith   if (flg) {
44992a359c20SBarry Smith     ierr = PetscViewerPushFormat(viewer,format);CHKERRQ(ierr);
450019a666eeSBarry Smith     ierr = SNESConvergedReasonView(snes,viewer);CHKERRQ(ierr);
45012a359c20SBarry Smith     ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
45022a359c20SBarry Smith     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
45032a359c20SBarry Smith   }
45042a359c20SBarry Smith   incall = PETSC_FALSE;
45052a359c20SBarry Smith   PetscFunctionReturn(0);
45062a359c20SBarry Smith }
45072a359c20SBarry Smith 
4508487a658cSBarry Smith /*@
4509f69a0ea3SMatthew Knepley    SNESSolve - Solves a nonlinear system F(x) = b.
4510f69a0ea3SMatthew Knepley    Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX().
45119b94acceSBarry Smith 
4512c7afd0dbSLois Curfman McInnes    Collective on SNES
4513c7afd0dbSLois Curfman McInnes 
4514b2002411SLois Curfman McInnes    Input Parameters:
4515c7afd0dbSLois Curfman McInnes +  snes - the SNES context
45160298fd71SBarry Smith .  b - the constant part of the equation F(x) = b, or NULL to use zero.
451785385478SLisandro Dalcin -  x - the solution vector.
45189b94acceSBarry Smith 
4519b2002411SLois Curfman McInnes    Notes:
45208ddd3da0SLois Curfman McInnes    The user should initialize the vector,x, with the initial guess
45218ddd3da0SLois Curfman McInnes    for the nonlinear solve prior to calling SNESSolve.  In particular,
45228ddd3da0SLois Curfman McInnes    to employ an initial guess of zero, the user should explicitly set
45238ddd3da0SLois Curfman McInnes    this vector to zero by calling VecSet().
45248ddd3da0SLois Curfman McInnes 
452536851e7fSLois Curfman McInnes    Level: beginner
452636851e7fSLois Curfman McInnes 
4527c0df2a02SJed Brown .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetGridSequence(), SNESGetSolution()
45289b94acceSBarry Smith @*/
45297087cfbeSBarry Smith PetscErrorCode  SNESSolve(SNES snes,Vec b,Vec x)
45309b94acceSBarry Smith {
4531dfbe8321SBarry Smith   PetscErrorCode    ierr;
4532ace3abfcSBarry Smith   PetscBool         flg;
4533efd51863SBarry Smith   PetscInt          grid;
45340298fd71SBarry Smith   Vec               xcreated = NULL;
4535caa4e7f2SJed Brown   DM                dm;
4536052efed2SBarry Smith 
45373a40ed3dSBarry Smith   PetscFunctionBegin;
45380700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4539a69afd8bSBarry Smith   if (x) PetscValidHeaderSpecific(x,VEC_CLASSID,3);
4540a69afd8bSBarry Smith   if (x) PetscCheckSameComm(snes,1,x,3);
45410700a824SBarry Smith   if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2);
454285385478SLisandro Dalcin   if (b) PetscCheckSameComm(snes,1,b,2);
454385385478SLisandro Dalcin 
454434b4d3a8SMatthew G. Knepley   /* High level operations using the nonlinear solver */
454506fc46c8SMatthew G. Knepley   {
454606fc46c8SMatthew G. Knepley     PetscViewer       viewer;
454706fc46c8SMatthew G. Knepley     PetscViewerFormat format;
45487c88af5aSMatthew G. Knepley     PetscInt          num;
454906fc46c8SMatthew G. Knepley     PetscBool         flg;
455006fc46c8SMatthew G. Knepley     static PetscBool  incall = PETSC_FALSE;
455106fc46c8SMatthew G. Knepley 
455206fc46c8SMatthew G. Knepley     if (!incall) {
455334b4d3a8SMatthew G. Knepley       /* Estimate the convergence rate of the discretization */
455416413a6aSBarry Smith       ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) snes),((PetscObject)snes)->options, ((PetscObject) snes)->prefix, "-snes_convergence_estimate", &viewer, &format, &flg);CHKERRQ(ierr);
455506fc46c8SMatthew G. Knepley       if (flg) {
455606fc46c8SMatthew G. Knepley         PetscConvEst conv;
455746079b62SMatthew G. Knepley         DM           dm;
455846079b62SMatthew G. Knepley         PetscReal   *alpha; /* Convergence rate of the solution error for each field in the L_2 norm */
455946079b62SMatthew G. Knepley         PetscInt     Nf;
456006fc46c8SMatthew G. Knepley 
456106fc46c8SMatthew G. Knepley         incall = PETSC_TRUE;
456246079b62SMatthew G. Knepley         ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
456346079b62SMatthew G. Knepley         ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr);
4564ca5a3622SMatthew G. Knepley         ierr = PetscCalloc1(Nf, &alpha);CHKERRQ(ierr);
456506fc46c8SMatthew G. Knepley         ierr = PetscConvEstCreate(PetscObjectComm((PetscObject) snes), &conv);CHKERRQ(ierr);
4566900f6b5bSMatthew G. Knepley         ierr = PetscConvEstSetSolver(conv, (PetscObject) snes);CHKERRQ(ierr);
456706fc46c8SMatthew G. Knepley         ierr = PetscConvEstSetFromOptions(conv);CHKERRQ(ierr);
45680955ed61SMatthew G. Knepley         ierr = PetscConvEstSetUp(conv);CHKERRQ(ierr);
456946079b62SMatthew G. Knepley         ierr = PetscConvEstGetConvRate(conv, alpha);CHKERRQ(ierr);
457006fc46c8SMatthew G. Knepley         ierr = PetscViewerPushFormat(viewer, format);CHKERRQ(ierr);
457106fc46c8SMatthew G. Knepley         ierr = PetscConvEstRateView(conv, alpha, viewer);CHKERRQ(ierr);
457206fc46c8SMatthew G. Knepley         ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
457306fc46c8SMatthew G. Knepley         ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
457406fc46c8SMatthew G. Knepley         ierr = PetscConvEstDestroy(&conv);CHKERRQ(ierr);
457546079b62SMatthew G. Knepley         ierr = PetscFree(alpha);CHKERRQ(ierr);
457606fc46c8SMatthew G. Knepley         incall = PETSC_FALSE;
457706fc46c8SMatthew G. Knepley       }
457834b4d3a8SMatthew G. Knepley       /* Adaptively refine the initial grid */
4579b2588ea6SMatthew G. Knepley       num  = 1;
4580b2588ea6SMatthew G. Knepley       ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_initial", &num, &flg);CHKERRQ(ierr);
458134b4d3a8SMatthew G. Knepley       if (flg) {
458234b4d3a8SMatthew G. Knepley         DMAdaptor adaptor;
458334b4d3a8SMatthew G. Knepley 
458434b4d3a8SMatthew G. Knepley         incall = PETSC_TRUE;
4585ea13f565SStefano Zampini         ierr = DMAdaptorCreate(PetscObjectComm((PetscObject)snes), &adaptor);CHKERRQ(ierr);
458634b4d3a8SMatthew G. Knepley         ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr);
4587b2588ea6SMatthew G. Knepley         ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr);
458834b4d3a8SMatthew G. Knepley         ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr);
458934b4d3a8SMatthew G. Knepley         ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr);
459034b4d3a8SMatthew G. Knepley         ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_INITIAL, &dm, &x);CHKERRQ(ierr);
459134b4d3a8SMatthew G. Knepley         ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr);
459234b4d3a8SMatthew G. Knepley         incall = PETSC_FALSE;
459334b4d3a8SMatthew G. Knepley       }
45947c88af5aSMatthew G. Knepley       /* Use grid sequencing to adapt */
45957c88af5aSMatthew G. Knepley       num  = 0;
45967c88af5aSMatthew G. Knepley       ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_sequence", &num, NULL);CHKERRQ(ierr);
45977c88af5aSMatthew G. Knepley       if (num) {
45987c88af5aSMatthew G. Knepley         DMAdaptor adaptor;
45997c88af5aSMatthew G. Knepley 
46007c88af5aSMatthew G. Knepley         incall = PETSC_TRUE;
4601ea13f565SStefano Zampini         ierr = DMAdaptorCreate(PetscObjectComm((PetscObject)snes), &adaptor);CHKERRQ(ierr);
46027c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr);
46037c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr);
46047c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr);
46057c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr);
46067c88af5aSMatthew G. Knepley         ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_SEQUENTIAL, &dm, &x);CHKERRQ(ierr);
46077c88af5aSMatthew G. Knepley         ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr);
46087c88af5aSMatthew G. Knepley         incall = PETSC_FALSE;
46097c88af5aSMatthew G. Knepley       }
461006fc46c8SMatthew G. Knepley     }
461106fc46c8SMatthew G. Knepley   }
4612caa4e7f2SJed Brown   if (!x) {
4613caa4e7f2SJed Brown     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4614caa4e7f2SJed Brown     ierr = DMCreateGlobalVector(dm,&xcreated);CHKERRQ(ierr);
4615a69afd8bSBarry Smith     x    = xcreated;
4616a69afd8bSBarry Smith   }
4617ce1779c8SBarry Smith   ierr = SNESViewFromOptions(snes,NULL,"-snes_view_pre");CHKERRQ(ierr);
4618f05ece33SBarry Smith 
4619ce94432eSBarry Smith   for (grid=0; grid<snes->gridsequence; grid++) {ierr = PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);}
4620efd51863SBarry Smith   for (grid=0; grid<snes->gridsequence+1; grid++) {
4621efd51863SBarry Smith 
462285385478SLisandro Dalcin     /* set solution vector */
4623efd51863SBarry Smith     if (!grid) {ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);}
46246bf464f9SBarry Smith     ierr          = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
462585385478SLisandro Dalcin     snes->vec_sol = x;
4626caa4e7f2SJed Brown     ierr          = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4627caa4e7f2SJed Brown 
4628caa4e7f2SJed Brown     /* set affine vector if provided */
462985385478SLisandro Dalcin     if (b) { ierr = PetscObjectReference((PetscObject)b);CHKERRQ(ierr); }
46306bf464f9SBarry Smith     ierr          = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
463185385478SLisandro Dalcin     snes->vec_rhs = b;
463285385478SLisandro Dalcin 
4633bfdd6862SPatrick Farrell     if (snes->vec_rhs && (snes->vec_func == snes->vec_rhs)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Right hand side vector cannot be function vector");
4634154060b5SMatthew G. Knepley     if (snes->vec_func == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector");
4635154060b5SMatthew G. Knepley     if (snes->vec_rhs  == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be right hand side vector");
4636154060b5SMatthew G. Knepley     if (!snes->vec_sol_update /* && snes->vec_sol */) {
4637154060b5SMatthew G. Knepley       ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr);
4638154060b5SMatthew G. Knepley       ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->vec_sol_update);CHKERRQ(ierr);
4639154060b5SMatthew G. Knepley     }
4640154060b5SMatthew G. Knepley     ierr = DMShellSetGlobalVector(dm,snes->vec_sol);CHKERRQ(ierr);
464170e92668SMatthew Knepley     ierr = SNESSetUp(snes);CHKERRQ(ierr);
46423f149594SLisandro Dalcin 
46437eee914bSBarry Smith     if (!grid) {
46447eee914bSBarry Smith       if (snes->ops->computeinitialguess) {
4645d25893d9SBarry Smith         ierr = (*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP);CHKERRQ(ierr);
4646d25893d9SBarry Smith       }
4647dd568438SSatish Balay     }
4648d25893d9SBarry Smith 
4649abc0a331SBarry Smith     if (snes->conv_hist_reset) snes->conv_hist_len = 0;
4650971e163fSPeter Brune     if (snes->counters_reset) {snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;}
4651d5e45103SBarry Smith 
46523f149594SLisandro Dalcin     ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
46534936397dSBarry Smith     ierr = (*snes->ops->solve)(snes);CHKERRQ(ierr);
465485385478SLisandro Dalcin     ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
465517186662SBarry Smith     if (!snes->reason) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason");
4656422a814eSBarry Smith     snes->domainerror = PETSC_FALSE; /* clear the flag if it has been set */
46573f149594SLisandro Dalcin 
465837ec4e1aSPeter Brune     if (snes->lagjac_persist) snes->jac_iter += snes->iter;
465937ec4e1aSPeter Brune     if (snes->lagpre_persist) snes->pre_iter += snes->iter;
466037ec4e1aSPeter Brune 
466116413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_test_local_min",NULL,NULL,&flg);CHKERRQ(ierr);
4662da9b6338SBarry Smith     if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); }
4663c4421ceaSFande Kong     /* Call converged reason views. This may involve user-provided viewers as well */
466419a666eeSBarry Smith     ierr = SNESConvergedReasonViewFromOptions(snes);CHKERRQ(ierr);
46655968eb51SBarry Smith 
4666ce94432eSBarry Smith     if (snes->errorifnotconverged && snes->reason < 0) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged");
46679c8e83a9SBarry Smith     if (snes->reason < 0) break;
4668efd51863SBarry Smith     if (grid <  snes->gridsequence) {
4669efd51863SBarry Smith       DM  fine;
4670efd51863SBarry Smith       Vec xnew;
4671efd51863SBarry Smith       Mat interp;
4672efd51863SBarry Smith 
4673ce94432eSBarry Smith       ierr = DMRefine(snes->dm,PetscObjectComm((PetscObject)snes),&fine);CHKERRQ(ierr);
4674ce94432eSBarry Smith       if (!fine) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"DMRefine() did not perform any refinement, cannot continue grid sequencing");
46750298fd71SBarry Smith       ierr = DMCreateInterpolation(snes->dm,fine,&interp,NULL);CHKERRQ(ierr);
4676efd51863SBarry Smith       ierr = DMCreateGlobalVector(fine,&xnew);CHKERRQ(ierr);
4677efd51863SBarry Smith       ierr = MatInterpolate(interp,x,xnew);CHKERRQ(ierr);
4678c833c3b5SJed Brown       ierr = DMInterpolate(snes->dm,interp,fine);CHKERRQ(ierr);
4679efd51863SBarry Smith       ierr = MatDestroy(&interp);CHKERRQ(ierr);
4680efd51863SBarry Smith       x    = xnew;
4681efd51863SBarry Smith 
4682efd51863SBarry Smith       ierr = SNESReset(snes);CHKERRQ(ierr);
4683efd51863SBarry Smith       ierr = SNESSetDM(snes,fine);CHKERRQ(ierr);
4684352f405dSMatthew G. Knepley       ierr = SNESResetFromOptions(snes);CHKERRQ(ierr);
4685efd51863SBarry Smith       ierr = DMDestroy(&fine);CHKERRQ(ierr);
4686ce94432eSBarry Smith       ierr = PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);
4687efd51863SBarry Smith     }
4688efd51863SBarry Smith   }
4689ce1779c8SBarry Smith   ierr = SNESViewFromOptions(snes,NULL,"-snes_view");CHKERRQ(ierr);
4690685405a1SBarry Smith   ierr = VecViewFromOptions(snes->vec_sol,(PetscObject)snes,"-snes_view_solution");CHKERRQ(ierr);
4691c0f0dcc3SMatthew G. Knepley   ierr = DMMonitor(snes->dm);CHKERRQ(ierr);
46923f7e2da0SPeter Brune 
4693a69afd8bSBarry Smith   ierr = VecDestroy(&xcreated);CHKERRQ(ierr);
4694e04113cfSBarry Smith   ierr = PetscObjectSAWsBlock((PetscObject)snes);CHKERRQ(ierr);
46953a40ed3dSBarry Smith   PetscFunctionReturn(0);
46969b94acceSBarry Smith }
46979b94acceSBarry Smith 
46989b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */
46999b94acceSBarry Smith 
470082bf6240SBarry Smith /*@C
47014b0e389bSBarry Smith    SNESSetType - Sets the method for the nonlinear solver.
47029b94acceSBarry Smith 
4703fee21e36SBarry Smith    Collective on SNES
4704fee21e36SBarry Smith 
4705c7afd0dbSLois Curfman McInnes    Input Parameters:
4706c7afd0dbSLois Curfman McInnes +  snes - the SNES context
4707454a90a3SBarry Smith -  type - a known method
4708c7afd0dbSLois Curfman McInnes 
4709c7afd0dbSLois Curfman McInnes    Options Database Key:
4710454a90a3SBarry Smith .  -snes_type <type> - Sets the method; use -help for a list
471104d7464bSBarry Smith    of available methods (for instance, newtonls or newtontr)
4712ae12b187SLois Curfman McInnes 
47139b94acceSBarry Smith    Notes:
4714e090d566SSatish Balay    See "petsc/include/petscsnes.h" for available methods (for instance)
471504d7464bSBarry Smith +    SNESNEWTONLS - Newton's method with line search
4716c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
4717a2b725a8SWilliam Gropp -    SNESNEWTONTR - Newton's method with trust region
4718c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
47199b94acceSBarry Smith 
4720ae12b187SLois Curfman McInnes   Normally, it is best to use the SNESSetFromOptions() command and then
4721ae12b187SLois Curfman McInnes   set the SNES solver type from the options database rather than by using
4722ae12b187SLois Curfman McInnes   this routine.  Using the options database provides the user with
4723ae12b187SLois Curfman McInnes   maximum flexibility in evaluating the many nonlinear solvers.
4724ae12b187SLois Curfman McInnes   The SNESSetType() routine is provided for those situations where it
4725ae12b187SLois Curfman McInnes   is necessary to set the nonlinear solver independently of the command
4726ae12b187SLois Curfman McInnes   line or options database.  This might be the case, for example, when
4727ae12b187SLois Curfman McInnes   the choice of solver changes during the execution of the program,
4728ae12b187SLois Curfman McInnes   and the user's application is taking responsibility for choosing the
4729b0a32e0cSBarry Smith   appropriate method.
473036851e7fSLois Curfman McInnes 
473195452b02SPatrick Sanan     Developer Notes:
473295452b02SPatrick Sanan     SNESRegister() adds a constructor for a new SNESType to SNESList, SNESSetType() locates
47338f6c3df8SBarry Smith     the constructor in that list and calls it to create the spexific object.
47348f6c3df8SBarry Smith 
473536851e7fSLois Curfman McInnes   Level: intermediate
4736a703fe33SLois Curfman McInnes 
47378f6c3df8SBarry Smith .seealso: SNESType, SNESCreate(), SNESDestroy(), SNESGetType(), SNESSetFromOptions()
4738435da068SBarry Smith 
47399b94acceSBarry Smith @*/
474019fd82e9SBarry Smith PetscErrorCode  SNESSetType(SNES snes,SNESType type)
47419b94acceSBarry Smith {
4742dfbe8321SBarry Smith   PetscErrorCode ierr,(*r)(SNES);
4743ace3abfcSBarry Smith   PetscBool      match;
47443a40ed3dSBarry Smith 
47453a40ed3dSBarry Smith   PetscFunctionBegin;
47460700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
47474482741eSBarry Smith   PetscValidCharPointer(type,2);
474882bf6240SBarry Smith 
4749251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr);
47500f5bd95cSBarry Smith   if (match) PetscFunctionReturn(0);
475192ff6ae8SBarry Smith 
47521c9cd337SJed Brown   ierr = PetscFunctionListFind(SNESList,type,&r);CHKERRQ(ierr);
4753e32f2f54SBarry Smith   if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type);
475475396ef9SLisandro Dalcin   /* Destroy the previous private SNES context */
4755b5c23020SJed Brown   if (snes->ops->destroy) {
4756b5c23020SJed Brown     ierr               = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr);
47570298fd71SBarry Smith     snes->ops->destroy = NULL;
4758b5c23020SJed Brown   }
475975396ef9SLisandro Dalcin   /* Reinitialize function pointers in SNESOps structure */
47609e5d0892SLisandro Dalcin   snes->ops->setup          = NULL;
47619e5d0892SLisandro Dalcin   snes->ops->solve          = NULL;
47629e5d0892SLisandro Dalcin   snes->ops->view           = NULL;
47639e5d0892SLisandro Dalcin   snes->ops->setfromoptions = NULL;
47649e5d0892SLisandro Dalcin   snes->ops->destroy        = NULL;
47657fe760d5SStefano Zampini 
47667fe760d5SStefano Zampini   /* It may happen the user has customized the line search before calling SNESSetType */
47677fe760d5SStefano Zampini   if (((PetscObject)snes)->type_name) {
4768d8d34be6SBarry Smith     ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr);
47697fe760d5SStefano Zampini   }
47707fe760d5SStefano Zampini 
477175396ef9SLisandro Dalcin   /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */
477275396ef9SLisandro Dalcin   snes->setupcalled = PETSC_FALSE;
4773f5af7f23SKarl Rupp 
4774454a90a3SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr);
477503bfa161SLisandro Dalcin   ierr = (*r)(snes);CHKERRQ(ierr);
47763a40ed3dSBarry Smith   PetscFunctionReturn(0);
47779b94acceSBarry Smith }
47789b94acceSBarry Smith 
47799b94acceSBarry Smith /*@C
47809a28b0a6SLois Curfman McInnes    SNESGetType - Gets the SNES method type and name (as a string).
47819b94acceSBarry Smith 
4782c7afd0dbSLois Curfman McInnes    Not Collective
4783c7afd0dbSLois Curfman McInnes 
47849b94acceSBarry Smith    Input Parameter:
47854b0e389bSBarry Smith .  snes - nonlinear solver context
47869b94acceSBarry Smith 
47879b94acceSBarry Smith    Output Parameter:
47883a7fca6bSBarry Smith .  type - SNES method (a character string)
47899b94acceSBarry Smith 
479036851e7fSLois Curfman McInnes    Level: intermediate
479136851e7fSLois Curfman McInnes 
47929b94acceSBarry Smith @*/
479319fd82e9SBarry Smith PetscErrorCode  SNESGetType(SNES snes,SNESType *type)
47949b94acceSBarry Smith {
47953a40ed3dSBarry Smith   PetscFunctionBegin;
47960700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
47974482741eSBarry Smith   PetscValidPointer(type,2);
47987adad957SLisandro Dalcin   *type = ((PetscObject)snes)->type_name;
47993a40ed3dSBarry Smith   PetscFunctionReturn(0);
48009b94acceSBarry Smith }
48019b94acceSBarry Smith 
48023cd8a7caSMatthew G. Knepley /*@
48033cd8a7caSMatthew G. Knepley   SNESSetSolution - Sets the solution vector for use by the SNES routines.
48043cd8a7caSMatthew G. Knepley 
4805d083f849SBarry Smith   Logically Collective on SNES
48063cd8a7caSMatthew G. Knepley 
48073cd8a7caSMatthew G. Knepley   Input Parameters:
48083cd8a7caSMatthew G. Knepley + snes - the SNES context obtained from SNESCreate()
48093cd8a7caSMatthew G. Knepley - u    - the solution vector
48103cd8a7caSMatthew G. Knepley 
48113cd8a7caSMatthew G. Knepley   Level: beginner
48123cd8a7caSMatthew G. Knepley 
48133cd8a7caSMatthew G. Knepley @*/
48143cd8a7caSMatthew G. Knepley PetscErrorCode SNESSetSolution(SNES snes, Vec u)
48153cd8a7caSMatthew G. Knepley {
48163cd8a7caSMatthew G. Knepley   DM             dm;
48173cd8a7caSMatthew G. Knepley   PetscErrorCode ierr;
48183cd8a7caSMatthew G. Knepley 
48193cd8a7caSMatthew G. Knepley   PetscFunctionBegin;
48203cd8a7caSMatthew G. Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
48213cd8a7caSMatthew G. Knepley   PetscValidHeaderSpecific(u, VEC_CLASSID, 2);
48223cd8a7caSMatthew G. Knepley   ierr = PetscObjectReference((PetscObject) u);CHKERRQ(ierr);
48233cd8a7caSMatthew G. Knepley   ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
48243cd8a7caSMatthew G. Knepley 
48253cd8a7caSMatthew G. Knepley   snes->vec_sol = u;
48263cd8a7caSMatthew G. Knepley 
48273cd8a7caSMatthew G. Knepley   ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
48283cd8a7caSMatthew G. Knepley   ierr = DMShellSetGlobalVector(dm, u);CHKERRQ(ierr);
48293cd8a7caSMatthew G. Knepley   PetscFunctionReturn(0);
48303cd8a7caSMatthew G. Knepley }
48313cd8a7caSMatthew G. Knepley 
483252baeb72SSatish Balay /*@
48339b94acceSBarry Smith    SNESGetSolution - Returns the vector where the approximate solution is
4834c0df2a02SJed Brown    stored. This is the fine grid solution when using SNESSetGridSequence().
48359b94acceSBarry Smith 
4836c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
4837c7afd0dbSLois Curfman McInnes 
48389b94acceSBarry Smith    Input Parameter:
48399b94acceSBarry Smith .  snes - the SNES context
48409b94acceSBarry Smith 
48419b94acceSBarry Smith    Output Parameter:
48429b94acceSBarry Smith .  x - the solution
48439b94acceSBarry Smith 
484470e92668SMatthew Knepley    Level: intermediate
484536851e7fSLois Curfman McInnes 
484685385478SLisandro Dalcin .seealso:  SNESGetSolutionUpdate(), SNESGetFunction()
48479b94acceSBarry Smith @*/
48487087cfbeSBarry Smith PetscErrorCode  SNESGetSolution(SNES snes,Vec *x)
48499b94acceSBarry Smith {
48503a40ed3dSBarry Smith   PetscFunctionBegin;
48510700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
48524482741eSBarry Smith   PetscValidPointer(x,2);
485385385478SLisandro Dalcin   *x = snes->vec_sol;
485470e92668SMatthew Knepley   PetscFunctionReturn(0);
485570e92668SMatthew Knepley }
485670e92668SMatthew Knepley 
485752baeb72SSatish Balay /*@
48589b94acceSBarry Smith    SNESGetSolutionUpdate - Returns the vector where the solution update is
48599b94acceSBarry Smith    stored.
48609b94acceSBarry Smith 
4861c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
4862c7afd0dbSLois Curfman McInnes 
48639b94acceSBarry Smith    Input Parameter:
48649b94acceSBarry Smith .  snes - the SNES context
48659b94acceSBarry Smith 
48669b94acceSBarry Smith    Output Parameter:
48679b94acceSBarry Smith .  x - the solution update
48689b94acceSBarry Smith 
486936851e7fSLois Curfman McInnes    Level: advanced
487036851e7fSLois Curfman McInnes 
487185385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction()
48729b94acceSBarry Smith @*/
48737087cfbeSBarry Smith PetscErrorCode  SNESGetSolutionUpdate(SNES snes,Vec *x)
48749b94acceSBarry Smith {
48753a40ed3dSBarry Smith   PetscFunctionBegin;
48760700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
48774482741eSBarry Smith   PetscValidPointer(x,2);
487885385478SLisandro Dalcin   *x = snes->vec_sol_update;
48793a40ed3dSBarry Smith   PetscFunctionReturn(0);
48809b94acceSBarry Smith }
48819b94acceSBarry Smith 
48829b94acceSBarry Smith /*@C
48833638b69dSLois Curfman McInnes    SNESGetFunction - Returns the vector where the function is stored.
48849b94acceSBarry Smith 
4885a63bb30eSJed Brown    Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet.
4886c7afd0dbSLois Curfman McInnes 
48879b94acceSBarry Smith    Input Parameter:
48889b94acceSBarry Smith .  snes - the SNES context
48899b94acceSBarry Smith 
48909b94acceSBarry Smith    Output Parameter:
48910298fd71SBarry Smith +  r - the vector that is used to store residuals (or NULL if you don't want it)
4892f8b49ee9SBarry Smith .  f - the function (or NULL if you don't want it); see SNESFunction for calling sequence details
48930298fd71SBarry Smith -  ctx - the function context (or NULL if you don't want it)
48949b94acceSBarry Smith 
489536851e7fSLois Curfman McInnes    Level: advanced
489636851e7fSLois Curfman McInnes 
489704edfde5SBarry Smith     Notes: The vector r DOES NOT, in general contain the current value of the SNES nonlinear function
489804edfde5SBarry Smith 
4899bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetSolution(), SNESFunction
49009b94acceSBarry Smith @*/
4901f8b49ee9SBarry Smith PetscErrorCode  SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),void **ctx)
49029b94acceSBarry Smith {
4903a63bb30eSJed Brown   PetscErrorCode ierr;
49046cab3a1bSJed Brown   DM             dm;
4905a63bb30eSJed Brown 
49063a40ed3dSBarry Smith   PetscFunctionBegin;
49070700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4908a63bb30eSJed Brown   if (r) {
4909a63bb30eSJed Brown     if (!snes->vec_func) {
4910a63bb30eSJed Brown       if (snes->vec_rhs) {
4911a63bb30eSJed Brown         ierr = VecDuplicate(snes->vec_rhs,&snes->vec_func);CHKERRQ(ierr);
4912a63bb30eSJed Brown       } else if (snes->vec_sol) {
4913a63bb30eSJed Brown         ierr = VecDuplicate(snes->vec_sol,&snes->vec_func);CHKERRQ(ierr);
4914a63bb30eSJed Brown       } else if (snes->dm) {
4915a63bb30eSJed Brown         ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr);
4916a63bb30eSJed Brown       }
4917a63bb30eSJed Brown     }
4918a63bb30eSJed Brown     *r = snes->vec_func;
4919a63bb30eSJed Brown   }
49206cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4921f8b49ee9SBarry Smith   ierr = DMSNESGetFunction(dm,f,ctx);CHKERRQ(ierr);
49223a40ed3dSBarry Smith   PetscFunctionReturn(0);
49239b94acceSBarry Smith }
49249b94acceSBarry Smith 
4925c79ef259SPeter Brune /*@C
4926be95d8f1SBarry Smith    SNESGetNGS - Returns the NGS function and context.
4927c79ef259SPeter Brune 
4928c79ef259SPeter Brune    Input Parameter:
4929c79ef259SPeter Brune .  snes - the SNES context
4930c79ef259SPeter Brune 
4931c79ef259SPeter Brune    Output Parameter:
4932be95d8f1SBarry Smith +  f - the function (or NULL) see SNESNGSFunction for details
49330298fd71SBarry Smith -  ctx    - the function context (or NULL)
4934c79ef259SPeter Brune 
4935c79ef259SPeter Brune    Level: advanced
4936c79ef259SPeter Brune 
4937be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetFunction()
4938c79ef259SPeter Brune @*/
4939c79ef259SPeter Brune 
4940be95d8f1SBarry Smith PetscErrorCode SNESGetNGS (SNES snes, PetscErrorCode (**f)(SNES, Vec, Vec, void*), void ** ctx)
4941646217ecSPeter Brune {
49426cab3a1bSJed Brown   PetscErrorCode ierr;
49436cab3a1bSJed Brown   DM             dm;
49446cab3a1bSJed Brown 
4945646217ecSPeter Brune   PetscFunctionBegin;
4946646217ecSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
49476cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4948be95d8f1SBarry Smith   ierr = DMSNESGetNGS(dm,f,ctx);CHKERRQ(ierr);
4949646217ecSPeter Brune   PetscFunctionReturn(0);
4950646217ecSPeter Brune }
4951646217ecSPeter Brune 
49523c7409f5SSatish Balay /*@C
49533c7409f5SSatish Balay    SNESSetOptionsPrefix - Sets the prefix used for searching for all
4954d850072dSLois Curfman McInnes    SNES options in the database.
49553c7409f5SSatish Balay 
49563f9fe445SBarry Smith    Logically Collective on SNES
4957fee21e36SBarry Smith 
4958c7afd0dbSLois Curfman McInnes    Input Parameter:
4959c7afd0dbSLois Curfman McInnes +  snes - the SNES context
4960c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
4961c7afd0dbSLois Curfman McInnes 
4962d850072dSLois Curfman McInnes    Notes:
4963a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
4964c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
4965d850072dSLois Curfman McInnes 
496636851e7fSLois Curfman McInnes    Level: advanced
496736851e7fSLois Curfman McInnes 
4968a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions()
49693c7409f5SSatish Balay @*/
49707087cfbeSBarry Smith PetscErrorCode  SNESSetOptionsPrefix(SNES snes,const char prefix[])
49713c7409f5SSatish Balay {
4972dfbe8321SBarry Smith   PetscErrorCode ierr;
49733c7409f5SSatish Balay 
49743a40ed3dSBarry Smith   PetscFunctionBegin;
49750700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4976639f9d9dSBarry Smith   ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
49771cee3971SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
497835f5d045SPeter Brune   if (snes->linesearch) {
49797601faf0SJed Brown     ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr);
498008b6c495SPeter Brune     ierr = PetscObjectSetOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr);
498135f5d045SPeter Brune   }
498235f5d045SPeter Brune   ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
49833a40ed3dSBarry Smith   PetscFunctionReturn(0);
49843c7409f5SSatish Balay }
49853c7409f5SSatish Balay 
49863c7409f5SSatish Balay /*@C
4987f525115eSLois Curfman McInnes    SNESAppendOptionsPrefix - Appends to the prefix used for searching for all
4988d850072dSLois Curfman McInnes    SNES options in the database.
49893c7409f5SSatish Balay 
49903f9fe445SBarry Smith    Logically Collective on SNES
4991fee21e36SBarry Smith 
4992c7afd0dbSLois Curfman McInnes    Input Parameters:
4993c7afd0dbSLois Curfman McInnes +  snes - the SNES context
4994c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
4995c7afd0dbSLois Curfman McInnes 
4996d850072dSLois Curfman McInnes    Notes:
4997a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
4998c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
4999d850072dSLois Curfman McInnes 
500036851e7fSLois Curfman McInnes    Level: advanced
500136851e7fSLois Curfman McInnes 
5002a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix()
50033c7409f5SSatish Balay @*/
50047087cfbeSBarry Smith PetscErrorCode  SNESAppendOptionsPrefix(SNES snes,const char prefix[])
50053c7409f5SSatish Balay {
5006dfbe8321SBarry Smith   PetscErrorCode ierr;
50073c7409f5SSatish Balay 
50083a40ed3dSBarry Smith   PetscFunctionBegin;
50090700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5010639f9d9dSBarry Smith   ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
50111cee3971SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
501235f5d045SPeter Brune   if (snes->linesearch) {
50137601faf0SJed Brown     ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr);
501408b6c495SPeter Brune     ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr);
501535f5d045SPeter Brune   }
501635f5d045SPeter Brune   ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
50173a40ed3dSBarry Smith   PetscFunctionReturn(0);
50183c7409f5SSatish Balay }
50193c7409f5SSatish Balay 
50209ab63eb5SSatish Balay /*@C
50213c7409f5SSatish Balay    SNESGetOptionsPrefix - Sets the prefix used for searching for all
50223c7409f5SSatish Balay    SNES options in the database.
50233c7409f5SSatish Balay 
5024c7afd0dbSLois Curfman McInnes    Not Collective
5025c7afd0dbSLois Curfman McInnes 
50263c7409f5SSatish Balay    Input Parameter:
50273c7409f5SSatish Balay .  snes - the SNES context
50283c7409f5SSatish Balay 
50293c7409f5SSatish Balay    Output Parameter:
50303c7409f5SSatish Balay .  prefix - pointer to the prefix string used
50313c7409f5SSatish Balay 
503295452b02SPatrick Sanan    Notes:
503395452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
50349ab63eb5SSatish Balay    sufficient length to hold the prefix.
50359ab63eb5SSatish Balay 
503636851e7fSLois Curfman McInnes    Level: advanced
503736851e7fSLois Curfman McInnes 
5038a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix()
50393c7409f5SSatish Balay @*/
50407087cfbeSBarry Smith PetscErrorCode  SNESGetOptionsPrefix(SNES snes,const char *prefix[])
50413c7409f5SSatish Balay {
5042dfbe8321SBarry Smith   PetscErrorCode ierr;
50433c7409f5SSatish Balay 
50443a40ed3dSBarry Smith   PetscFunctionBegin;
50450700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5046639f9d9dSBarry Smith   ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
50473a40ed3dSBarry Smith   PetscFunctionReturn(0);
50483c7409f5SSatish Balay }
50493c7409f5SSatish Balay 
5050b2002411SLois Curfman McInnes 
50513cea93caSBarry Smith /*@C
50521c84c290SBarry Smith   SNESRegister - Adds a method to the nonlinear solver package.
50531c84c290SBarry Smith 
50541c84c290SBarry Smith    Not collective
50551c84c290SBarry Smith 
50561c84c290SBarry Smith    Input Parameters:
50571c84c290SBarry Smith +  name_solver - name of a new user-defined solver
50581c84c290SBarry Smith -  routine_create - routine to create method context
50591c84c290SBarry Smith 
50601c84c290SBarry Smith    Notes:
50611c84c290SBarry Smith    SNESRegister() may be called multiple times to add several user-defined solvers.
50621c84c290SBarry Smith 
50631c84c290SBarry Smith    Sample usage:
50641c84c290SBarry Smith .vb
5065bdf89e91SBarry Smith    SNESRegister("my_solver",MySolverCreate);
50661c84c290SBarry Smith .ve
50671c84c290SBarry Smith 
50681c84c290SBarry Smith    Then, your solver can be chosen with the procedural interface via
50691c84c290SBarry Smith $     SNESSetType(snes,"my_solver")
50701c84c290SBarry Smith    or at runtime via the option
50711c84c290SBarry Smith $     -snes_type my_solver
50721c84c290SBarry Smith 
50731c84c290SBarry Smith    Level: advanced
50741c84c290SBarry Smith 
50751c84c290SBarry Smith     Note: If your function is not being put into a shared library then use SNESRegister() instead
50761c84c290SBarry Smith 
50771c84c290SBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy()
50783cea93caSBarry Smith 
50797f6c08e0SMatthew Knepley   Level: advanced
50803cea93caSBarry Smith @*/
5081bdf89e91SBarry Smith PetscErrorCode  SNESRegister(const char sname[],PetscErrorCode (*function)(SNES))
5082b2002411SLois Curfman McInnes {
5083dfbe8321SBarry Smith   PetscErrorCode ierr;
5084b2002411SLois Curfman McInnes 
5085b2002411SLois Curfman McInnes   PetscFunctionBegin;
50861d36bdfdSBarry Smith   ierr = SNESInitializePackage();CHKERRQ(ierr);
5087a240a19fSJed Brown   ierr = PetscFunctionListAdd(&SNESList,sname,function);CHKERRQ(ierr);
5088b2002411SLois Curfman McInnes   PetscFunctionReturn(0);
5089b2002411SLois Curfman McInnes }
5090da9b6338SBarry Smith 
50917087cfbeSBarry Smith PetscErrorCode  SNESTestLocalMin(SNES snes)
5092da9b6338SBarry Smith {
5093dfbe8321SBarry Smith   PetscErrorCode ierr;
509477431f27SBarry Smith   PetscInt       N,i,j;
5095da9b6338SBarry Smith   Vec            u,uh,fh;
5096da9b6338SBarry Smith   PetscScalar    value;
5097da9b6338SBarry Smith   PetscReal      norm;
5098da9b6338SBarry Smith 
5099da9b6338SBarry Smith   PetscFunctionBegin;
5100da9b6338SBarry Smith   ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr);
5101da9b6338SBarry Smith   ierr = VecDuplicate(u,&uh);CHKERRQ(ierr);
5102da9b6338SBarry Smith   ierr = VecDuplicate(u,&fh);CHKERRQ(ierr);
5103da9b6338SBarry Smith 
5104da9b6338SBarry Smith   /* currently only works for sequential */
5105ea13f565SStefano Zampini   ierr = PetscPrintf(PetscObjectComm((PetscObject)snes),"Testing FormFunction() for local min\n");CHKERRQ(ierr);
5106da9b6338SBarry Smith   ierr = VecGetSize(u,&N);CHKERRQ(ierr);
5107da9b6338SBarry Smith   for (i=0; i<N; i++) {
5108da9b6338SBarry Smith     ierr = VecCopy(u,uh);CHKERRQ(ierr);
5109ea13f565SStefano Zampini     ierr = PetscPrintf(PetscObjectComm((PetscObject)snes),"i = %D\n",i);CHKERRQ(ierr);
5110da9b6338SBarry Smith     for (j=-10; j<11; j++) {
51118b49ba18SBarry Smith       value = PetscSign(j)*PetscExpReal(PetscAbs(j)-10.0);
5112da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
51133ab0aad5SBarry Smith       ierr  = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr);
5114da9b6338SBarry Smith       ierr  = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr);
5115ea13f565SStefano Zampini       ierr  = PetscPrintf(PetscObjectComm((PetscObject)snes),"       j norm %D %18.16e\n",j,norm);CHKERRQ(ierr);
5116da9b6338SBarry Smith       value = -value;
5117da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
5118da9b6338SBarry Smith     }
5119da9b6338SBarry Smith   }
51206bf464f9SBarry Smith   ierr = VecDestroy(&uh);CHKERRQ(ierr);
51216bf464f9SBarry Smith   ierr = VecDestroy(&fh);CHKERRQ(ierr);
5122da9b6338SBarry Smith   PetscFunctionReturn(0);
5123da9b6338SBarry Smith }
512471f87433Sdalcinl 
512571f87433Sdalcinl /*@
5126fa9f3622SBarry Smith    SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for
512771f87433Sdalcinl    computing relative tolerance for linear solvers within an inexact
512871f87433Sdalcinl    Newton method.
512971f87433Sdalcinl 
51303f9fe445SBarry Smith    Logically Collective on SNES
513171f87433Sdalcinl 
513271f87433Sdalcinl    Input Parameters:
513371f87433Sdalcinl +  snes - SNES context
513471f87433Sdalcinl -  flag - PETSC_TRUE or PETSC_FALSE
513571f87433Sdalcinl 
513664ba62caSBarry Smith     Options Database:
513764ba62caSBarry Smith +  -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence
513864ba62caSBarry Smith .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
513964ba62caSBarry Smith .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
514064ba62caSBarry Smith .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
514164ba62caSBarry Smith .  -snes_ksp_ew_gamma <gamma> - Sets gamma
514264ba62caSBarry Smith .  -snes_ksp_ew_alpha <alpha> - Sets alpha
514364ba62caSBarry Smith .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
514464ba62caSBarry Smith -  -snes_ksp_ew_threshold <threshold> - Sets threshold
514564ba62caSBarry Smith 
514671f87433Sdalcinl    Notes:
514771f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
514871f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
514971f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
515071f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
515171f87433Sdalcinl    solver.
515271f87433Sdalcinl 
515371f87433Sdalcinl    Level: advanced
515471f87433Sdalcinl 
515571f87433Sdalcinl    Reference:
515671f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
515771f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
515871f87433Sdalcinl 
5159fa9f3622SBarry Smith .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
516071f87433Sdalcinl @*/
51617087cfbeSBarry Smith PetscErrorCode  SNESKSPSetUseEW(SNES snes,PetscBool flag)
516271f87433Sdalcinl {
516371f87433Sdalcinl   PetscFunctionBegin;
51640700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5165acfcf0e5SJed Brown   PetscValidLogicalCollectiveBool(snes,flag,2);
516671f87433Sdalcinl   snes->ksp_ewconv = flag;
516771f87433Sdalcinl   PetscFunctionReturn(0);
516871f87433Sdalcinl }
516971f87433Sdalcinl 
517071f87433Sdalcinl /*@
5171fa9f3622SBarry Smith    SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method
517271f87433Sdalcinl    for computing relative tolerance for linear solvers within an
517371f87433Sdalcinl    inexact Newton method.
517471f87433Sdalcinl 
517571f87433Sdalcinl    Not Collective
517671f87433Sdalcinl 
517771f87433Sdalcinl    Input Parameter:
517871f87433Sdalcinl .  snes - SNES context
517971f87433Sdalcinl 
518071f87433Sdalcinl    Output Parameter:
518171f87433Sdalcinl .  flag - PETSC_TRUE or PETSC_FALSE
518271f87433Sdalcinl 
518371f87433Sdalcinl    Notes:
518471f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
518571f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
518671f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
518771f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
518871f87433Sdalcinl    solver.
518971f87433Sdalcinl 
519071f87433Sdalcinl    Level: advanced
519171f87433Sdalcinl 
519271f87433Sdalcinl    Reference:
519371f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
519471f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
519571f87433Sdalcinl 
5196fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
519771f87433Sdalcinl @*/
51987087cfbeSBarry Smith PetscErrorCode  SNESKSPGetUseEW(SNES snes, PetscBool  *flag)
519971f87433Sdalcinl {
520071f87433Sdalcinl   PetscFunctionBegin;
52010700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5202534a8f05SLisandro Dalcin   PetscValidBoolPointer(flag,2);
520371f87433Sdalcinl   *flag = snes->ksp_ewconv;
520471f87433Sdalcinl   PetscFunctionReturn(0);
520571f87433Sdalcinl }
520671f87433Sdalcinl 
520771f87433Sdalcinl /*@
5208fa9f3622SBarry Smith    SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker
520971f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
521071f87433Sdalcinl    Newton method.
521171f87433Sdalcinl 
52123f9fe445SBarry Smith    Logically Collective on SNES
521371f87433Sdalcinl 
521471f87433Sdalcinl    Input Parameters:
521571f87433Sdalcinl +    snes - SNES context
521671f87433Sdalcinl .    version - version 1, 2 (default is 2) or 3
521771f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
521871f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
521971f87433Sdalcinl .    gamma - multiplicative factor for version 2 rtol computation
522071f87433Sdalcinl              (0 <= gamma2 <= 1)
522171f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
522271f87433Sdalcinl .    alpha2 - power for safeguard
522371f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
522471f87433Sdalcinl 
522571f87433Sdalcinl    Note:
522671f87433Sdalcinl    Version 3 was contributed by Luis Chacon, June 2006.
522771f87433Sdalcinl 
522871f87433Sdalcinl    Use PETSC_DEFAULT to retain the default for any of the parameters.
522971f87433Sdalcinl 
523071f87433Sdalcinl    Level: advanced
523171f87433Sdalcinl 
523271f87433Sdalcinl    Reference:
523371f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
523471f87433Sdalcinl    inexact Newton method", Utah State University Math. Stat. Dept. Res.
523571f87433Sdalcinl    Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput.
523671f87433Sdalcinl 
5237fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW()
523871f87433Sdalcinl @*/
5239f5af7f23SKarl Rupp PetscErrorCode  SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold)
524071f87433Sdalcinl {
5241fa9f3622SBarry Smith   SNESKSPEW *kctx;
52425fd66863SKarl Rupp 
524371f87433Sdalcinl   PetscFunctionBegin;
52440700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5245fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
5246e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
5247c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,version,2);
5248c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol_0,3);
5249c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol_max,4);
5250c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,gamma,5);
5251c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,alpha,6);
5252c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,alpha2,7);
5253c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,threshold,8);
525471f87433Sdalcinl 
525571f87433Sdalcinl   if (version != PETSC_DEFAULT)   kctx->version   = version;
525671f87433Sdalcinl   if (rtol_0 != PETSC_DEFAULT)    kctx->rtol_0    = rtol_0;
525771f87433Sdalcinl   if (rtol_max != PETSC_DEFAULT)  kctx->rtol_max  = rtol_max;
525871f87433Sdalcinl   if (gamma != PETSC_DEFAULT)     kctx->gamma     = gamma;
525971f87433Sdalcinl   if (alpha != PETSC_DEFAULT)     kctx->alpha     = alpha;
526071f87433Sdalcinl   if (alpha2 != PETSC_DEFAULT)    kctx->alpha2    = alpha2;
526171f87433Sdalcinl   if (threshold != PETSC_DEFAULT) kctx->threshold = threshold;
526271f87433Sdalcinl 
5263f23aa3ddSBarry Smith   if (kctx->version < 1 || kctx->version > 3) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 and 3 are supported: %D",kctx->version);
526457622a8eSBarry Smith   if (kctx->rtol_0 < 0.0 || kctx->rtol_0 >= 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_0 < 1.0: %g",(double)kctx->rtol_0);
526557622a8eSBarry Smith   if (kctx->rtol_max < 0.0 || kctx->rtol_max >= 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_max (%g) < 1.0\n",(double)kctx->rtol_max);
526657622a8eSBarry Smith   if (kctx->gamma < 0.0 || kctx->gamma > 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= gamma (%g) <= 1.0\n",(double)kctx->gamma);
526757622a8eSBarry Smith   if (kctx->alpha <= 1.0 || kctx->alpha > 2.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"1.0 < alpha (%g) <= 2.0\n",(double)kctx->alpha);
526857622a8eSBarry Smith   if (kctx->threshold <= 0.0 || kctx->threshold >= 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 < threshold (%g) < 1.0\n",(double)kctx->threshold);
526971f87433Sdalcinl   PetscFunctionReturn(0);
527071f87433Sdalcinl }
527171f87433Sdalcinl 
527271f87433Sdalcinl /*@
5273fa9f3622SBarry Smith    SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker
527471f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
527571f87433Sdalcinl    Newton method.
527671f87433Sdalcinl 
527771f87433Sdalcinl    Not Collective
527871f87433Sdalcinl 
527971f87433Sdalcinl    Input Parameters:
528071f87433Sdalcinl      snes - SNES context
528171f87433Sdalcinl 
528271f87433Sdalcinl    Output Parameters:
528371f87433Sdalcinl +    version - version 1, 2 (default is 2) or 3
528471f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
528571f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
5286bf388a1fSBarry Smith .    gamma - multiplicative factor for version 2 rtol computation (0 <= gamma2 <= 1)
528771f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
528871f87433Sdalcinl .    alpha2 - power for safeguard
528971f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
529071f87433Sdalcinl 
529171f87433Sdalcinl    Level: advanced
529271f87433Sdalcinl 
5293fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW()
529471f87433Sdalcinl @*/
5295bf388a1fSBarry Smith PetscErrorCode  SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold)
529671f87433Sdalcinl {
5297fa9f3622SBarry Smith   SNESKSPEW *kctx;
52985fd66863SKarl Rupp 
529971f87433Sdalcinl   PetscFunctionBegin;
53000700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5301fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
5302e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
530371f87433Sdalcinl   if (version)   *version   = kctx->version;
530471f87433Sdalcinl   if (rtol_0)    *rtol_0    = kctx->rtol_0;
530571f87433Sdalcinl   if (rtol_max)  *rtol_max  = kctx->rtol_max;
530671f87433Sdalcinl   if (gamma)     *gamma     = kctx->gamma;
530771f87433Sdalcinl   if (alpha)     *alpha     = kctx->alpha;
530871f87433Sdalcinl   if (alpha2)    *alpha2    = kctx->alpha2;
530971f87433Sdalcinl   if (threshold) *threshold = kctx->threshold;
531071f87433Sdalcinl   PetscFunctionReturn(0);
531171f87433Sdalcinl }
531271f87433Sdalcinl 
5313d5378b5fSDmitry Karpeev  PetscErrorCode KSPPreSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes)
531471f87433Sdalcinl {
531571f87433Sdalcinl   PetscErrorCode ierr;
5316fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
531771f87433Sdalcinl   PetscReal      rtol  = PETSC_DEFAULT,stol;
531871f87433Sdalcinl 
531971f87433Sdalcinl   PetscFunctionBegin;
5320d4211eb9SBarry Smith   if (!snes->ksp_ewconv) PetscFunctionReturn(0);
532130058271SDmitry Karpeev   if (!snes->iter) {
532230058271SDmitry Karpeev     rtol = kctx->rtol_0; /* first time in, so use the original user rtol */
532330058271SDmitry Karpeev     ierr = VecNorm(snes->vec_func,NORM_2,&kctx->norm_first);CHKERRQ(ierr);
532430058271SDmitry Karpeev   }
5325f5af7f23SKarl Rupp   else {
532671f87433Sdalcinl     if (kctx->version == 1) {
532771f87433Sdalcinl       rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last;
532871f87433Sdalcinl       if (rtol < 0.0) rtol = -rtol;
532985ec1a3cSBarry Smith       stol = PetscPowReal(kctx->rtol_last,kctx->alpha2);
533071f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
533171f87433Sdalcinl     } else if (kctx->version == 2) {
533285ec1a3cSBarry Smith       rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha);
533385ec1a3cSBarry Smith       stol = kctx->gamma * PetscPowReal(kctx->rtol_last,kctx->alpha);
533471f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
533571f87433Sdalcinl     } else if (kctx->version == 3) { /* contributed by Luis Chacon, June 2006. */
533685ec1a3cSBarry Smith       rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha);
533771f87433Sdalcinl       /* safeguard: avoid sharp decrease of rtol */
533885ec1a3cSBarry Smith       stol = kctx->gamma*PetscPowReal(kctx->rtol_last,kctx->alpha);
533971f87433Sdalcinl       stol = PetscMax(rtol,stol);
534071f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
534171f87433Sdalcinl       /* safeguard: avoid oversolving */
534230058271SDmitry Karpeev       stol = kctx->gamma*(kctx->norm_first*snes->rtol)/snes->norm;
534371f87433Sdalcinl       stol = PetscMax(rtol,stol);
534471f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
5345e32f2f54SBarry Smith     } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version);
534671f87433Sdalcinl   }
534771f87433Sdalcinl   /* safeguard: avoid rtol greater than one */
534871f87433Sdalcinl   rtol = PetscMin(rtol,kctx->rtol_max);
534971f87433Sdalcinl   ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr);
535057622a8eSBarry Smith   ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%g\n",snes->iter,kctx->version,(double)rtol);CHKERRQ(ierr);
535171f87433Sdalcinl   PetscFunctionReturn(0);
535271f87433Sdalcinl }
535371f87433Sdalcinl 
5354d5378b5fSDmitry Karpeev PetscErrorCode KSPPostSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes)
535571f87433Sdalcinl {
535671f87433Sdalcinl   PetscErrorCode ierr;
5357fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
535871f87433Sdalcinl   PCSide         pcside;
535971f87433Sdalcinl   Vec            lres;
536071f87433Sdalcinl 
536171f87433Sdalcinl   PetscFunctionBegin;
5362d4211eb9SBarry Smith   if (!snes->ksp_ewconv) PetscFunctionReturn(0);
53639e5d0892SLisandro Dalcin   ierr = KSPGetTolerances(ksp,&kctx->rtol_last,NULL,NULL,NULL);CHKERRQ(ierr);
536471dbe336SPeter Brune   kctx->norm_last = snes->norm;
536571f87433Sdalcinl   if (kctx->version == 1) {
53664f00ce20SMatthew G. Knepley     PC        pc;
53674f00ce20SMatthew G. Knepley     PetscBool isNone;
53684f00ce20SMatthew G. Knepley 
53694f00ce20SMatthew G. Knepley     ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr);
53704f00ce20SMatthew G. Knepley     ierr = PetscObjectTypeCompare((PetscObject) pc, PCNONE, &isNone);CHKERRQ(ierr);
5371b037da10SBarry Smith     ierr = KSPGetPCSide(ksp,&pcside);CHKERRQ(ierr);
53724f00ce20SMatthew G. Knepley      if (pcside == PC_RIGHT || isNone) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */
537371f87433Sdalcinl       /* KSP residual is true linear residual */
537471f87433Sdalcinl       ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr);
537571f87433Sdalcinl     } else {
537671f87433Sdalcinl       /* KSP residual is preconditioned residual */
537771f87433Sdalcinl       /* compute true linear residual norm */
537871f87433Sdalcinl       ierr = VecDuplicate(b,&lres);CHKERRQ(ierr);
537971f87433Sdalcinl       ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr);
538071f87433Sdalcinl       ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr);
538171f87433Sdalcinl       ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr);
53826bf464f9SBarry Smith       ierr = VecDestroy(&lres);CHKERRQ(ierr);
538371f87433Sdalcinl     }
538471f87433Sdalcinl   }
538571f87433Sdalcinl   PetscFunctionReturn(0);
538671f87433Sdalcinl }
538771f87433Sdalcinl 
5388d4211eb9SBarry Smith /*@
5389d4211eb9SBarry Smith    SNESGetKSP - Returns the KSP context for a SNES solver.
5390d4211eb9SBarry Smith 
5391d4211eb9SBarry Smith    Not Collective, but if SNES object is parallel, then KSP object is parallel
5392d4211eb9SBarry Smith 
5393d4211eb9SBarry Smith    Input Parameter:
5394d4211eb9SBarry Smith .  snes - the SNES context
5395d4211eb9SBarry Smith 
5396d4211eb9SBarry Smith    Output Parameter:
5397d4211eb9SBarry Smith .  ksp - the KSP context
5398d4211eb9SBarry Smith 
5399d4211eb9SBarry Smith    Notes:
5400d4211eb9SBarry Smith    The user can then directly manipulate the KSP context to set various
5401d4211eb9SBarry Smith    options, etc.  Likewise, the user can then extract and manipulate the
5402d4211eb9SBarry Smith    PC contexts as well.
5403d4211eb9SBarry Smith 
5404d4211eb9SBarry Smith    Level: beginner
5405d4211eb9SBarry Smith 
5406d4211eb9SBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
5407d4211eb9SBarry Smith @*/
5408d4211eb9SBarry Smith PetscErrorCode  SNESGetKSP(SNES snes,KSP *ksp)
540971f87433Sdalcinl {
541071f87433Sdalcinl   PetscErrorCode ierr;
541171f87433Sdalcinl 
541271f87433Sdalcinl   PetscFunctionBegin;
5413d4211eb9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5414d4211eb9SBarry Smith   PetscValidPointer(ksp,2);
5415d4211eb9SBarry Smith 
5416d4211eb9SBarry Smith   if (!snes->ksp) {
5417d4211eb9SBarry Smith     ierr = KSPCreate(PetscObjectComm((PetscObject)snes),&snes->ksp);CHKERRQ(ierr);
5418d4211eb9SBarry Smith     ierr = PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1);CHKERRQ(ierr);
54193bb1ff40SBarry Smith     ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->ksp);CHKERRQ(ierr);
5420d4211eb9SBarry Smith 
5421d5378b5fSDmitry Karpeev     ierr = KSPSetPreSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_SNESEW,snes);CHKERRQ(ierr);
5422d5378b5fSDmitry Karpeev     ierr = KSPSetPostSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_SNESEW,snes);CHKERRQ(ierr);
5423a5c2985bSBarry Smith 
5424798534f6SMatthew G. Knepley     ierr = KSPMonitorSetFromOptions(snes->ksp, "-snes_monitor_ksp", "snes_preconditioned_residual", snes);CHKERRQ(ierr);
542516413a6aSBarry Smith     ierr = PetscObjectSetOptions((PetscObject)snes->ksp,((PetscObject)snes)->options);CHKERRQ(ierr);
5426d4211eb9SBarry Smith   }
5427d4211eb9SBarry Smith   *ksp = snes->ksp;
542871f87433Sdalcinl   PetscFunctionReturn(0);
542971f87433Sdalcinl }
54306c699258SBarry Smith 
5431d4211eb9SBarry Smith 
5432af0996ceSBarry Smith #include <petsc/private/dmimpl.h>
54336c699258SBarry Smith /*@
54342a808120SBarry Smith    SNESSetDM - Sets the DM that may be used by some nonlinear solvers or their underlying preconditioners
54356c699258SBarry Smith 
54363f9fe445SBarry Smith    Logically Collective on SNES
54376c699258SBarry Smith 
54386c699258SBarry Smith    Input Parameters:
54392a808120SBarry Smith +  snes - the nonlinear solver context
54402a808120SBarry Smith -  dm - the dm, cannot be NULL
54416c699258SBarry Smith 
5442e03a659cSJed Brown    Notes:
5443e03a659cSJed Brown    A DM can only be used for solving one problem at a time because information about the problem is stored on the DM,
5444e03a659cSJed Brown    even when not using interfaces like DMSNESSetFunction().  Use DMClone() to get a distinct DM when solving different
5445e03a659cSJed Brown    problems using the same function space.
5446e03a659cSJed Brown 
54476c699258SBarry Smith    Level: intermediate
54486c699258SBarry Smith 
54494c2026ceSFande Kong .seealso: SNESGetDM(), KSPSetDM(), KSPGetDM()
54506c699258SBarry Smith @*/
54517087cfbeSBarry Smith PetscErrorCode  SNESSetDM(SNES snes,DM dm)
54526c699258SBarry Smith {
54536c699258SBarry Smith   PetscErrorCode ierr;
5454345fed2cSBarry Smith   KSP            ksp;
5455942e3340SBarry Smith   DMSNES         sdm;
54566c699258SBarry Smith 
54576c699258SBarry Smith   PetscFunctionBegin;
54580700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
54592a808120SBarry Smith   PetscValidHeaderSpecific(dm,DM_CLASSID,2);
54602a808120SBarry Smith   ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr);
5461942e3340SBarry Smith   if (snes->dm) {               /* Move the DMSNES context over to the new DM unless the new DM already has one */
546251f4b3c7SToby Isaac     if (snes->dm->dmsnes && !dm->dmsnes) {
5463942e3340SBarry Smith       ierr = DMCopyDMSNES(snes->dm,dm);CHKERRQ(ierr);
5464942e3340SBarry Smith       ierr = DMGetDMSNES(snes->dm,&sdm);CHKERRQ(ierr);
5465f5af7f23SKarl Rupp       if (sdm->originaldm == snes->dm) sdm->originaldm = dm; /* Grant write privileges to the replacement DM */
54666cab3a1bSJed Brown     }
5467dc822a44SJed Brown     ierr = DMCoarsenHookRemove(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr);
54686bf464f9SBarry Smith     ierr = DMDestroy(&snes->dm);CHKERRQ(ierr);
54696cab3a1bSJed Brown   }
54706c699258SBarry Smith   snes->dm     = dm;
5471116d1032SJed Brown   snes->dmAuto = PETSC_FALSE;
5472f5af7f23SKarl Rupp 
5473345fed2cSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
5474345fed2cSBarry Smith   ierr = KSPSetDM(ksp,dm);CHKERRQ(ierr);
5475f22f69f0SBarry Smith   ierr = KSPSetDMActive(ksp,PETSC_FALSE);CHKERRQ(ierr);
5476efd4aadfSBarry Smith   if (snes->npc) {
5477efd4aadfSBarry Smith     ierr = SNESSetDM(snes->npc, snes->dm);CHKERRQ(ierr);
5478efd4aadfSBarry Smith     ierr = SNESSetNPCSide(snes,snes->npcside);CHKERRQ(ierr);
54792c155ee1SBarry Smith   }
54806c699258SBarry Smith   PetscFunctionReturn(0);
54816c699258SBarry Smith }
54826c699258SBarry Smith 
54836c699258SBarry Smith /*@
54846c699258SBarry Smith    SNESGetDM - Gets the DM that may be used by some preconditioners
54856c699258SBarry Smith 
54863f9fe445SBarry Smith    Not Collective but DM obtained is parallel on SNES
54876c699258SBarry Smith 
54886c699258SBarry Smith    Input Parameter:
54896c699258SBarry Smith . snes - the preconditioner context
54906c699258SBarry Smith 
54916c699258SBarry Smith    Output Parameter:
54926c699258SBarry Smith .  dm - the dm
54936c699258SBarry Smith 
54946c699258SBarry Smith    Level: intermediate
54956c699258SBarry Smith 
54964c2026ceSFande Kong .seealso: SNESSetDM(), KSPSetDM(), KSPGetDM()
54976c699258SBarry Smith @*/
54987087cfbeSBarry Smith PetscErrorCode  SNESGetDM(SNES snes,DM *dm)
54996c699258SBarry Smith {
55006cab3a1bSJed Brown   PetscErrorCode ierr;
55016cab3a1bSJed Brown 
55026c699258SBarry Smith   PetscFunctionBegin;
55030700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
55046cab3a1bSJed Brown   if (!snes->dm) {
5505ce94432eSBarry Smith     ierr         = DMShellCreate(PetscObjectComm((PetscObject)snes),&snes->dm);CHKERRQ(ierr);
5506116d1032SJed Brown     snes->dmAuto = PETSC_TRUE;
55076cab3a1bSJed Brown   }
55086c699258SBarry Smith   *dm = snes->dm;
55096c699258SBarry Smith   PetscFunctionReturn(0);
55106c699258SBarry Smith }
55110807856dSBarry Smith 
551231823bd8SMatthew G Knepley /*@
5513be95d8f1SBarry Smith   SNESSetNPC - Sets the nonlinear preconditioner to be used.
551431823bd8SMatthew G Knepley 
551531823bd8SMatthew G Knepley   Collective on SNES
551631823bd8SMatthew G Knepley 
551731823bd8SMatthew G Knepley   Input Parameters:
551831823bd8SMatthew G Knepley + snes - iterative context obtained from SNESCreate()
551931823bd8SMatthew G Knepley - pc   - the preconditioner object
552031823bd8SMatthew G Knepley 
552131823bd8SMatthew G Knepley   Notes:
5522be95d8f1SBarry Smith   Use SNESGetNPC() to retrieve the preconditioner context (for example,
552331823bd8SMatthew G Knepley   to configure it using the API).
552431823bd8SMatthew G Knepley 
552531823bd8SMatthew G Knepley   Level: developer
552631823bd8SMatthew G Knepley 
55273ad1a0b9SPatrick Farrell .seealso: SNESGetNPC(), SNESHasNPC()
552831823bd8SMatthew G Knepley @*/
5529be95d8f1SBarry Smith PetscErrorCode SNESSetNPC(SNES snes, SNES pc)
553031823bd8SMatthew G Knepley {
553131823bd8SMatthew G Knepley   PetscErrorCode ierr;
553231823bd8SMatthew G Knepley 
553331823bd8SMatthew G Knepley   PetscFunctionBegin;
553431823bd8SMatthew G Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
553531823bd8SMatthew G Knepley   PetscValidHeaderSpecific(pc, SNES_CLASSID, 2);
553631823bd8SMatthew G Knepley   PetscCheckSameComm(snes, 1, pc, 2);
553731823bd8SMatthew G Knepley   ierr     = PetscObjectReference((PetscObject) pc);CHKERRQ(ierr);
5538efd4aadfSBarry Smith   ierr     = SNESDestroy(&snes->npc);CHKERRQ(ierr);
5539efd4aadfSBarry Smith   snes->npc = pc;
5540efd4aadfSBarry Smith   ierr     = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->npc);CHKERRQ(ierr);
554131823bd8SMatthew G Knepley   PetscFunctionReturn(0);
554231823bd8SMatthew G Knepley }
554331823bd8SMatthew G Knepley 
554431823bd8SMatthew G Knepley /*@
5545be95d8f1SBarry Smith   SNESGetNPC - Creates a nonlinear preconditioning solver (SNES) to be used to precondition the nonlinear solver.
554631823bd8SMatthew G Knepley 
5547951fe5abSBarry Smith   Not Collective; but any changes to the obtained SNES object must be applied collectively
554831823bd8SMatthew G Knepley 
554931823bd8SMatthew G Knepley   Input Parameter:
555031823bd8SMatthew G Knepley . snes - iterative context obtained from SNESCreate()
555131823bd8SMatthew G Knepley 
555231823bd8SMatthew G Knepley   Output Parameter:
555331823bd8SMatthew G Knepley . pc - preconditioner context
555431823bd8SMatthew G Knepley 
5555b5badacbSBarry Smith   Options Database:
5556b5badacbSBarry Smith . -npc_snes_type <type> - set the type of the SNES to use as the nonlinear preconditioner
5557b5badacbSBarry Smith 
555895452b02SPatrick Sanan   Notes:
5559b5badacbSBarry Smith     If a SNES was previously set with SNESSetNPC() then that SNES is returned, otherwise a new SNES object is created.
5560be95d8f1SBarry Smith 
5561951fe5abSBarry Smith     The (preconditioner) SNES returned automatically inherits the same nonlinear function and Jacobian supplied to the original
5562951fe5abSBarry Smith     SNES during SNESSetUp()
5563951fe5abSBarry Smith 
556431823bd8SMatthew G Knepley   Level: developer
556531823bd8SMatthew G Knepley 
5566951fe5abSBarry Smith .seealso: SNESSetNPC(), SNESHasNPC(), SNES, SNESCreate()
556731823bd8SMatthew G Knepley @*/
5568be95d8f1SBarry Smith PetscErrorCode SNESGetNPC(SNES snes, SNES *pc)
556931823bd8SMatthew G Knepley {
557031823bd8SMatthew G Knepley   PetscErrorCode ierr;
5571a64e098fSPeter Brune   const char     *optionsprefix;
557231823bd8SMatthew G Knepley 
557331823bd8SMatthew G Knepley   PetscFunctionBegin;
557431823bd8SMatthew G Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
557531823bd8SMatthew G Knepley   PetscValidPointer(pc, 2);
5576efd4aadfSBarry Smith   if (!snes->npc) {
5577efd4aadfSBarry Smith     ierr = SNESCreate(PetscObjectComm((PetscObject)snes),&snes->npc);CHKERRQ(ierr);
5578efd4aadfSBarry Smith     ierr = PetscObjectIncrementTabLevel((PetscObject)snes->npc,(PetscObject)snes,1);CHKERRQ(ierr);
5579efd4aadfSBarry Smith     ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr);
5580a64e098fSPeter Brune     ierr = SNESGetOptionsPrefix(snes,&optionsprefix);CHKERRQ(ierr);
5581efd4aadfSBarry Smith     ierr = SNESSetOptionsPrefix(snes->npc,optionsprefix);CHKERRQ(ierr);
5582efd4aadfSBarry Smith     ierr = SNESAppendOptionsPrefix(snes->npc,"npc_");CHKERRQ(ierr);
5583efd4aadfSBarry Smith     ierr = SNESSetCountersReset(snes->npc,PETSC_FALSE);CHKERRQ(ierr);
558431823bd8SMatthew G Knepley   }
5585efd4aadfSBarry Smith   *pc = snes->npc;
558631823bd8SMatthew G Knepley   PetscFunctionReturn(0);
558731823bd8SMatthew G Knepley }
558831823bd8SMatthew G Knepley 
55893ad1a0b9SPatrick Farrell /*@
55903ad1a0b9SPatrick Farrell   SNESHasNPC - Returns whether a nonlinear preconditioner exists
55913ad1a0b9SPatrick Farrell 
55923ad1a0b9SPatrick Farrell   Not Collective
55933ad1a0b9SPatrick Farrell 
55943ad1a0b9SPatrick Farrell   Input Parameter:
55953ad1a0b9SPatrick Farrell . snes - iterative context obtained from SNESCreate()
55963ad1a0b9SPatrick Farrell 
55973ad1a0b9SPatrick Farrell   Output Parameter:
55983ad1a0b9SPatrick Farrell . has_npc - whether the SNES has an NPC or not
55993ad1a0b9SPatrick Farrell 
56003ad1a0b9SPatrick Farrell   Level: developer
56013ad1a0b9SPatrick Farrell 
56023ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESGetNPC()
56033ad1a0b9SPatrick Farrell @*/
56043ad1a0b9SPatrick Farrell PetscErrorCode SNESHasNPC(SNES snes, PetscBool *has_npc)
56053ad1a0b9SPatrick Farrell {
56063ad1a0b9SPatrick Farrell   PetscFunctionBegin;
56073ad1a0b9SPatrick Farrell   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5608efd4aadfSBarry Smith   *has_npc = (PetscBool) (snes->npc ? PETSC_TRUE : PETSC_FALSE);
56093ad1a0b9SPatrick Farrell   PetscFunctionReturn(0);
56103ad1a0b9SPatrick Farrell }
56113ad1a0b9SPatrick Farrell 
5612c40d0f55SPeter Brune /*@
5613be95d8f1SBarry Smith     SNESSetNPCSide - Sets the preconditioning side.
5614c40d0f55SPeter Brune 
5615c40d0f55SPeter Brune     Logically Collective on SNES
5616c40d0f55SPeter Brune 
5617c40d0f55SPeter Brune     Input Parameter:
5618c40d0f55SPeter Brune .   snes - iterative context obtained from SNESCreate()
5619c40d0f55SPeter Brune 
5620c40d0f55SPeter Brune     Output Parameter:
5621c40d0f55SPeter Brune .   side - the preconditioning side, where side is one of
5622c40d0f55SPeter Brune .vb
56232d547940SBarry Smith       PC_LEFT - left preconditioning
56242d547940SBarry Smith       PC_RIGHT - right preconditioning (default for most nonlinear solvers)
5625c40d0f55SPeter Brune .ve
5626c40d0f55SPeter Brune 
5627c40d0f55SPeter Brune     Options Database Keys:
5628c40d0f55SPeter Brune .   -snes_pc_side <right,left>
5629c40d0f55SPeter Brune 
563095452b02SPatrick Sanan     Notes:
563195452b02SPatrick Sanan     SNESNRICHARDSON and SNESNCG only support left preconditioning.
56322d547940SBarry Smith 
5633c40d0f55SPeter Brune     Level: intermediate
5634c40d0f55SPeter Brune 
5635be95d8f1SBarry Smith .seealso: SNESGetNPCSide(), KSPSetPCSide()
5636c40d0f55SPeter Brune @*/
5637be95d8f1SBarry Smith PetscErrorCode  SNESSetNPCSide(SNES snes,PCSide side)
5638c40d0f55SPeter Brune {
5639c40d0f55SPeter Brune   PetscFunctionBegin;
5640c40d0f55SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5641c40d0f55SPeter Brune   PetscValidLogicalCollectiveEnum(snes,side,2);
5642efd4aadfSBarry Smith   snes->npcside= side;
5643c40d0f55SPeter Brune   PetscFunctionReturn(0);
5644c40d0f55SPeter Brune }
5645c40d0f55SPeter Brune 
5646c40d0f55SPeter Brune /*@
5647be95d8f1SBarry Smith     SNESGetNPCSide - Gets the preconditioning side.
5648c40d0f55SPeter Brune 
5649c40d0f55SPeter Brune     Not Collective
5650c40d0f55SPeter Brune 
5651c40d0f55SPeter Brune     Input Parameter:
5652c40d0f55SPeter Brune .   snes - iterative context obtained from SNESCreate()
5653c40d0f55SPeter Brune 
5654c40d0f55SPeter Brune     Output Parameter:
5655c40d0f55SPeter Brune .   side - the preconditioning side, where side is one of
5656c40d0f55SPeter Brune .vb
56572d547940SBarry Smith       PC_LEFT - left preconditioning
56582d547940SBarry Smith       PC_RIGHT - right preconditioning (default for most nonlinear solvers)
5659c40d0f55SPeter Brune .ve
5660c40d0f55SPeter Brune 
5661c40d0f55SPeter Brune     Level: intermediate
5662c40d0f55SPeter Brune 
5663be95d8f1SBarry Smith .seealso: SNESSetNPCSide(), KSPGetPCSide()
5664c40d0f55SPeter Brune @*/
5665be95d8f1SBarry Smith PetscErrorCode  SNESGetNPCSide(SNES snes,PCSide *side)
5666c40d0f55SPeter Brune {
5667c40d0f55SPeter Brune   PetscFunctionBegin;
5668c40d0f55SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5669c40d0f55SPeter Brune   PetscValidPointer(side,2);
5670efd4aadfSBarry Smith   *side = snes->npcside;
5671c40d0f55SPeter Brune   PetscFunctionReturn(0);
5672c40d0f55SPeter Brune }
5673c40d0f55SPeter Brune 
56749e764e56SPeter Brune /*@
56757601faf0SJed Brown   SNESSetLineSearch - Sets the linesearch on the SNES instance.
56769e764e56SPeter Brune 
56779e764e56SPeter Brune   Collective on SNES
56789e764e56SPeter Brune 
56799e764e56SPeter Brune   Input Parameters:
56809e764e56SPeter Brune + snes - iterative context obtained from SNESCreate()
56819e764e56SPeter Brune - linesearch   - the linesearch object
56829e764e56SPeter Brune 
56839e764e56SPeter Brune   Notes:
56847601faf0SJed Brown   Use SNESGetLineSearch() to retrieve the preconditioner context (for example,
56859e764e56SPeter Brune   to configure it using the API).
56869e764e56SPeter Brune 
56879e764e56SPeter Brune   Level: developer
56889e764e56SPeter Brune 
56897601faf0SJed Brown .seealso: SNESGetLineSearch()
56909e764e56SPeter Brune @*/
56917601faf0SJed Brown PetscErrorCode SNESSetLineSearch(SNES snes, SNESLineSearch linesearch)
56929e764e56SPeter Brune {
56939e764e56SPeter Brune   PetscErrorCode ierr;
56949e764e56SPeter Brune 
56959e764e56SPeter Brune   PetscFunctionBegin;
56969e764e56SPeter Brune   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5697f1c6b773SPeter Brune   PetscValidHeaderSpecific(linesearch, SNESLINESEARCH_CLASSID, 2);
56989e764e56SPeter Brune   PetscCheckSameComm(snes, 1, linesearch, 2);
56999e764e56SPeter Brune   ierr = PetscObjectReference((PetscObject) linesearch);CHKERRQ(ierr);
5700f1c6b773SPeter Brune   ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr);
5701f5af7f23SKarl Rupp 
57029e764e56SPeter Brune   snes->linesearch = linesearch;
5703f5af7f23SKarl Rupp 
57043bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr);
57059e764e56SPeter Brune   PetscFunctionReturn(0);
57069e764e56SPeter Brune }
57079e764e56SPeter Brune 
5708a34ceb2aSJed Brown /*@
57097601faf0SJed Brown   SNESGetLineSearch - Returns a pointer to the line search context set with SNESSetLineSearch()
57108141a3b9SPeter Brune   or creates a default line search instance associated with the SNES and returns it.
57119e764e56SPeter Brune 
57129e764e56SPeter Brune   Not Collective
57139e764e56SPeter Brune 
57149e764e56SPeter Brune   Input Parameter:
57159e764e56SPeter Brune . snes - iterative context obtained from SNESCreate()
57169e764e56SPeter Brune 
57179e764e56SPeter Brune   Output Parameter:
57189e764e56SPeter Brune . linesearch - linesearch context
57199e764e56SPeter Brune 
5720162e0bf5SPeter Brune   Level: beginner
57219e764e56SPeter Brune 
5722162e0bf5SPeter Brune .seealso: SNESSetLineSearch(), SNESLineSearchCreate()
57239e764e56SPeter Brune @*/
57247601faf0SJed Brown PetscErrorCode SNESGetLineSearch(SNES snes, SNESLineSearch *linesearch)
57259e764e56SPeter Brune {
57269e764e56SPeter Brune   PetscErrorCode ierr;
57279e764e56SPeter Brune   const char     *optionsprefix;
57289e764e56SPeter Brune 
57299e764e56SPeter Brune   PetscFunctionBegin;
57309e764e56SPeter Brune   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
57319e764e56SPeter Brune   PetscValidPointer(linesearch, 2);
57329e764e56SPeter Brune   if (!snes->linesearch) {
57339e764e56SPeter Brune     ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr);
573482f516ccSBarry Smith     ierr = SNESLineSearchCreate(PetscObjectComm((PetscObject)snes), &snes->linesearch);CHKERRQ(ierr);
5735f1c6b773SPeter Brune     ierr = SNESLineSearchSetSNES(snes->linesearch, snes);CHKERRQ(ierr);
5736b1dca40bSPeter Brune     ierr = SNESLineSearchAppendOptionsPrefix(snes->linesearch, optionsprefix);CHKERRQ(ierr);
57379e764e56SPeter Brune     ierr = PetscObjectIncrementTabLevel((PetscObject) snes->linesearch, (PetscObject) snes, 1);CHKERRQ(ierr);
57383bb1ff40SBarry Smith     ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr);
57399e764e56SPeter Brune   }
57409e764e56SPeter Brune   *linesearch = snes->linesearch;
57419e764e56SPeter Brune   PetscFunctionReturn(0);
57429e764e56SPeter Brune }
5743