xref: /petsc/src/snes/interface/snes.c (revision 0df40c3555541438f50e9ac13eaa6989b426b66a)
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);
185485385478SLisandro Dalcin     snes->vec_func = r;
1855d2a683ecSLisandro Dalcin   }
18566cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
1857f8b49ee9SBarry Smith   ierr = DMSNESSetFunction(dm,f,ctx);CHKERRQ(ierr);
1858bbc1464cSBarry Smith   if (f == SNESPicardComputeFunction) {
1859bbc1464cSBarry Smith     ierr = DMSNESSetMFFunction(dm,SNESPicardComputeMFFunction,ctx);CHKERRQ(ierr);
1860bbc1464cSBarry Smith   }
18613a40ed3dSBarry Smith   PetscFunctionReturn(0);
18629b94acceSBarry Smith }
18639b94acceSBarry Smith 
1864646217ecSPeter Brune 
1865e4ed7901SPeter Brune /*@C
1866e4ed7901SPeter Brune    SNESSetInitialFunction - Sets the function vector to be used as the
1867e4ed7901SPeter Brune    function norm at the initialization of the method.  In some
1868e4ed7901SPeter Brune    instances, the user has precomputed the function before calling
1869e4ed7901SPeter Brune    SNESSolve.  This function allows one to avoid a redundant call
1870e4ed7901SPeter Brune    to SNESComputeFunction in that case.
1871e4ed7901SPeter Brune 
1872e4ed7901SPeter Brune    Logically Collective on SNES
1873e4ed7901SPeter Brune 
1874e4ed7901SPeter Brune    Input Parameters:
1875e4ed7901SPeter Brune +  snes - the SNES context
1876e4ed7901SPeter Brune -  f - vector to store function value
1877e4ed7901SPeter Brune 
1878e4ed7901SPeter Brune    Notes:
1879e4ed7901SPeter Brune    This should not be modified during the solution procedure.
1880e4ed7901SPeter Brune 
1881e4ed7901SPeter Brune    This is used extensively in the SNESFAS hierarchy and in nonlinear preconditioning.
1882e4ed7901SPeter Brune 
1883e4ed7901SPeter Brune    Level: developer
1884e4ed7901SPeter Brune 
1885e4ed7901SPeter Brune .seealso: SNESSetFunction(), SNESComputeFunction(), SNESSetInitialFunctionNorm()
1886e4ed7901SPeter Brune @*/
1887e4ed7901SPeter Brune PetscErrorCode  SNESSetInitialFunction(SNES snes, Vec f)
1888e4ed7901SPeter Brune {
1889e4ed7901SPeter Brune   PetscErrorCode ierr;
1890e4ed7901SPeter Brune   Vec            vec_func;
1891e4ed7901SPeter Brune 
1892e4ed7901SPeter Brune   PetscFunctionBegin;
1893e4ed7901SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1894e4ed7901SPeter Brune   PetscValidHeaderSpecific(f,VEC_CLASSID,2);
1895e4ed7901SPeter Brune   PetscCheckSameComm(snes,1,f,2);
1896efd4aadfSBarry Smith   if (snes->npcside== PC_LEFT && snes->functype == SNES_FUNCTION_PRECONDITIONED) {
1897902f982fSPeter Brune     snes->vec_func_init_set = PETSC_FALSE;
1898902f982fSPeter Brune     PetscFunctionReturn(0);
1899902f982fSPeter Brune   }
19000298fd71SBarry Smith   ierr = SNESGetFunction(snes,&vec_func,NULL,NULL);CHKERRQ(ierr);
1901e4ed7901SPeter Brune   ierr = VecCopy(f, vec_func);CHKERRQ(ierr);
1902f5af7f23SKarl Rupp 
1903217b9c2eSPeter Brune   snes->vec_func_init_set = PETSC_TRUE;
1904e4ed7901SPeter Brune   PetscFunctionReturn(0);
1905e4ed7901SPeter Brune }
1906e4ed7901SPeter Brune 
1907534ebe21SPeter Brune /*@
1908365a6726SPeter Brune    SNESSetNormSchedule - Sets the SNESNormSchedule used in covergence and monitoring
1909534ebe21SPeter Brune    of the SNES method.
1910534ebe21SPeter Brune 
1911534ebe21SPeter Brune    Logically Collective on SNES
1912534ebe21SPeter Brune 
1913534ebe21SPeter Brune    Input Parameters:
1914534ebe21SPeter Brune +  snes - the SNES context
1915365a6726SPeter Brune -  normschedule - the frequency of norm computation
1916534ebe21SPeter Brune 
1917517f1916SMatthew G. Knepley    Options Database Key:
1918517f1916SMatthew G. Knepley .  -snes_norm_schedule <none, always, initialonly, finalonly, initalfinalonly>
1919517f1916SMatthew G. Knepley 
1920534ebe21SPeter Brune    Notes:
1921365a6726SPeter Brune    Only certain SNES methods support certain SNESNormSchedules.  Most require evaluation
1922534ebe21SPeter Brune    of the nonlinear function and the taking of its norm at every iteration to
1923534ebe21SPeter Brune    even ensure convergence at all.  However, methods such as custom Gauss-Seidel methods
1924be95d8f1SBarry Smith    (SNESNGS) and the like do not require the norm of the function to be computed, and therfore
1925534ebe21SPeter Brune    may either be monitored for convergence or not.  As these are often used as nonlinear
1926534ebe21SPeter Brune    preconditioners, monitoring the norm of their error is not a useful enterprise within
1927534ebe21SPeter Brune    their solution.
1928534ebe21SPeter Brune 
1929534ebe21SPeter Brune    Level: developer
1930534ebe21SPeter Brune 
1931365a6726SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1932534ebe21SPeter Brune @*/
1933365a6726SPeter Brune PetscErrorCode  SNESSetNormSchedule(SNES snes, SNESNormSchedule normschedule)
1934534ebe21SPeter Brune {
1935534ebe21SPeter Brune   PetscFunctionBegin;
1936534ebe21SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1937365a6726SPeter Brune   snes->normschedule = normschedule;
1938534ebe21SPeter Brune   PetscFunctionReturn(0);
1939534ebe21SPeter Brune }
1940534ebe21SPeter Brune 
1941534ebe21SPeter Brune 
1942534ebe21SPeter Brune /*@
1943365a6726SPeter Brune    SNESGetNormSchedule - Gets the SNESNormSchedule used in covergence and monitoring
1944534ebe21SPeter Brune    of the SNES method.
1945534ebe21SPeter Brune 
1946534ebe21SPeter Brune    Logically Collective on SNES
1947534ebe21SPeter Brune 
1948534ebe21SPeter Brune    Input Parameters:
1949534ebe21SPeter Brune +  snes - the SNES context
1950365a6726SPeter Brune -  normschedule - the type of the norm used
1951534ebe21SPeter Brune 
1952534ebe21SPeter Brune    Level: advanced
1953534ebe21SPeter Brune 
1954365a6726SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1955534ebe21SPeter Brune @*/
1956365a6726SPeter Brune PetscErrorCode  SNESGetNormSchedule(SNES snes, SNESNormSchedule *normschedule)
1957534ebe21SPeter Brune {
1958534ebe21SPeter Brune   PetscFunctionBegin;
1959534ebe21SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1960365a6726SPeter Brune   *normschedule = snes->normschedule;
1961534ebe21SPeter Brune   PetscFunctionReturn(0);
1962534ebe21SPeter Brune }
1963534ebe21SPeter Brune 
196447073ea2SPeter Brune 
1965c5ce4427SMatthew G. Knepley /*@
1966c5ce4427SMatthew G. Knepley   SNESSetFunctionNorm - Sets the last computed residual norm.
1967c5ce4427SMatthew G. Knepley 
1968c5ce4427SMatthew G. Knepley   Logically Collective on SNES
1969c5ce4427SMatthew G. Knepley 
1970c5ce4427SMatthew G. Knepley   Input Parameters:
1971c5ce4427SMatthew G. Knepley + snes - the SNES context
1972c5ce4427SMatthew G. Knepley 
1973c5ce4427SMatthew G. Knepley - normschedule - the frequency of norm computation
1974c5ce4427SMatthew G. Knepley 
1975c5ce4427SMatthew G. Knepley   Level: developer
1976c5ce4427SMatthew G. Knepley 
1977c5ce4427SMatthew G. Knepley .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1978c5ce4427SMatthew G. Knepley @*/
1979c5ce4427SMatthew G. Knepley PetscErrorCode SNESSetFunctionNorm(SNES snes, PetscReal norm)
1980c5ce4427SMatthew G. Knepley {
1981c5ce4427SMatthew G. Knepley   PetscFunctionBegin;
1982c5ce4427SMatthew G. Knepley   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1983c5ce4427SMatthew G. Knepley   snes->norm = norm;
1984c5ce4427SMatthew G. Knepley   PetscFunctionReturn(0);
1985c5ce4427SMatthew G. Knepley }
1986c5ce4427SMatthew G. Knepley 
1987c5ce4427SMatthew G. Knepley /*@
1988c5ce4427SMatthew G. Knepley   SNESGetFunctionNorm - Gets the last computed norm of the residual
1989c5ce4427SMatthew G. Knepley 
1990c5ce4427SMatthew G. Knepley   Not Collective
1991c5ce4427SMatthew G. Knepley 
1992c5ce4427SMatthew G. Knepley   Input Parameter:
1993c5ce4427SMatthew G. Knepley . snes - the SNES context
1994c5ce4427SMatthew G. Knepley 
1995c5ce4427SMatthew G. Knepley   Output Parameter:
1996c5ce4427SMatthew G. Knepley . norm - the last computed residual norm
1997c5ce4427SMatthew G. Knepley 
1998c5ce4427SMatthew G. Knepley   Level: developer
1999c5ce4427SMatthew G. Knepley 
2000c5ce4427SMatthew G. Knepley .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
2001c5ce4427SMatthew G. Knepley @*/
2002c5ce4427SMatthew G. Knepley PetscErrorCode SNESGetFunctionNorm(SNES snes, PetscReal *norm)
2003c5ce4427SMatthew G. Knepley {
2004c5ce4427SMatthew G. Knepley   PetscFunctionBegin;
2005c5ce4427SMatthew G. Knepley   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2006c5ce4427SMatthew G. Knepley   PetscValidPointer(norm, 2);
2007c5ce4427SMatthew G. Knepley   *norm = snes->norm;
2008c5ce4427SMatthew G. Knepley   PetscFunctionReturn(0);
2009c5ce4427SMatthew G. Knepley }
2010c5ce4427SMatthew G. Knepley 
2011c1e67a49SFande Kong /*@
2012c1e67a49SFande Kong   SNESGetUpdateNorm - Gets the last computed norm of the Newton update
2013c1e67a49SFande Kong 
2014c1e67a49SFande Kong   Not Collective
2015c1e67a49SFande Kong 
2016c1e67a49SFande Kong   Input Parameter:
2017c1e67a49SFande Kong . snes - the SNES context
2018c1e67a49SFande Kong 
2019c1e67a49SFande Kong   Output Parameter:
2020c1e67a49SFande Kong . ynorm - the last computed update norm
2021c1e67a49SFande Kong 
2022c1e67a49SFande Kong   Level: developer
2023c1e67a49SFande Kong 
2024c1e67a49SFande Kong .seealso: SNESSetNormSchedule(), SNESComputeFunction(), SNESGetFunctionNorm()
2025c1e67a49SFande Kong @*/
2026c1e67a49SFande Kong PetscErrorCode SNESGetUpdateNorm(SNES snes, PetscReal *ynorm)
2027c1e67a49SFande Kong {
2028c1e67a49SFande Kong   PetscFunctionBegin;
2029c1e67a49SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2030c1e67a49SFande Kong   PetscValidPointer(ynorm, 2);
2031c1e67a49SFande Kong   *ynorm = snes->ynorm;
2032c1e67a49SFande Kong   PetscFunctionReturn(0);
2033c1e67a49SFande Kong }
2034c1e67a49SFande Kong 
2035c1e67a49SFande Kong /*@
20364591eaf2SFande Kong   SNESGetSolutionNorm - Gets the last computed norm of the solution
2037c1e67a49SFande Kong 
2038c1e67a49SFande Kong   Not Collective
2039c1e67a49SFande Kong 
2040c1e67a49SFande Kong   Input Parameter:
2041c1e67a49SFande Kong . snes - the SNES context
2042c1e67a49SFande Kong 
2043c1e67a49SFande Kong   Output Parameter:
2044c1e67a49SFande Kong . xnorm - the last computed solution norm
2045c1e67a49SFande Kong 
2046c1e67a49SFande Kong   Level: developer
2047c1e67a49SFande Kong 
2048c1e67a49SFande Kong .seealso: SNESSetNormSchedule(), SNESComputeFunction(), SNESGetFunctionNorm(), SNESGetUpdateNorm()
2049c1e67a49SFande Kong @*/
2050c1e67a49SFande Kong PetscErrorCode SNESGetSolutionNorm(SNES snes, PetscReal *xnorm)
2051c1e67a49SFande Kong {
2052c1e67a49SFande Kong   PetscFunctionBegin;
2053c1e67a49SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2054c1e67a49SFande Kong   PetscValidPointer(xnorm, 2);
2055c1e67a49SFande Kong   *xnorm = snes->xnorm;
2056c1e67a49SFande Kong   PetscFunctionReturn(0);
2057c1e67a49SFande Kong }
2058c1e67a49SFande Kong 
205947073ea2SPeter Brune /*@C
206047073ea2SPeter Brune    SNESSetFunctionType - Sets the SNESNormSchedule used in covergence and monitoring
206147073ea2SPeter Brune    of the SNES method.
206247073ea2SPeter Brune 
206347073ea2SPeter Brune    Logically Collective on SNES
206447073ea2SPeter Brune 
206547073ea2SPeter Brune    Input Parameters:
206647073ea2SPeter Brune +  snes - the SNES context
206747073ea2SPeter Brune -  normschedule - the frequency of norm computation
206847073ea2SPeter Brune 
206947073ea2SPeter Brune    Notes:
207047073ea2SPeter Brune    Only certain SNES methods support certain SNESNormSchedules.  Most require evaluation
207147073ea2SPeter Brune    of the nonlinear function and the taking of its norm at every iteration to
207247073ea2SPeter Brune    even ensure convergence at all.  However, methods such as custom Gauss-Seidel methods
2073be95d8f1SBarry Smith    (SNESNGS) and the like do not require the norm of the function to be computed, and therfore
207447073ea2SPeter Brune    may either be monitored for convergence or not.  As these are often used as nonlinear
207547073ea2SPeter Brune    preconditioners, monitoring the norm of their error is not a useful enterprise within
207647073ea2SPeter Brune    their solution.
207747073ea2SPeter Brune 
207847073ea2SPeter Brune    Level: developer
207947073ea2SPeter Brune 
208047073ea2SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
208147073ea2SPeter Brune @*/
208247073ea2SPeter Brune PetscErrorCode  SNESSetFunctionType(SNES snes, SNESFunctionType type)
208347073ea2SPeter Brune {
208447073ea2SPeter Brune   PetscFunctionBegin;
208547073ea2SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
208647073ea2SPeter Brune   snes->functype = type;
208747073ea2SPeter Brune   PetscFunctionReturn(0);
208847073ea2SPeter Brune }
208947073ea2SPeter Brune 
209047073ea2SPeter Brune 
209147073ea2SPeter Brune /*@C
209247073ea2SPeter Brune    SNESGetFunctionType - Gets the SNESNormSchedule used in covergence and monitoring
209347073ea2SPeter Brune    of the SNES method.
209447073ea2SPeter Brune 
209547073ea2SPeter Brune    Logically Collective on SNES
209647073ea2SPeter Brune 
209747073ea2SPeter Brune    Input Parameters:
209847073ea2SPeter Brune +  snes - the SNES context
209947073ea2SPeter Brune -  normschedule - the type of the norm used
210047073ea2SPeter Brune 
210147073ea2SPeter Brune    Level: advanced
210247073ea2SPeter Brune 
210347073ea2SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
210447073ea2SPeter Brune @*/
210547073ea2SPeter Brune PetscErrorCode  SNESGetFunctionType(SNES snes, SNESFunctionType *type)
210647073ea2SPeter Brune {
210747073ea2SPeter Brune   PetscFunctionBegin;
210847073ea2SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
210947073ea2SPeter Brune   *type = snes->functype;
2110534ebe21SPeter Brune   PetscFunctionReturn(0);
2111534ebe21SPeter Brune }
2112534ebe21SPeter Brune 
2113bf388a1fSBarry Smith /*MC
2114be95d8f1SBarry Smith     SNESNGSFunction - function used to convey a Gauss-Seidel sweep on the nonlinear function
2115bf388a1fSBarry Smith 
2116bf388a1fSBarry Smith      Synopsis:
2117aaa7dc30SBarry Smith      #include <petscsnes.h>
2118be95d8f1SBarry Smith $    SNESNGSFunction(SNES snes,Vec x,Vec b,void *ctx);
2119bf388a1fSBarry Smith 
21201843f636SBarry Smith      Collective on snes
21211843f636SBarry Smith 
21221843f636SBarry Smith      Input Parameters:
2123bf388a1fSBarry Smith +  X   - solution vector
2124bf388a1fSBarry Smith .  B   - RHS vector
2125bf388a1fSBarry Smith -  ctx - optional user-defined Gauss-Seidel context
2126bf388a1fSBarry Smith 
21271843f636SBarry Smith      Output Parameter:
21281843f636SBarry Smith .  X   - solution vector
21291843f636SBarry Smith 
2130878cb397SSatish Balay    Level: intermediate
2131878cb397SSatish Balay 
2132be95d8f1SBarry Smith .seealso:   SNESSetNGS(), SNESGetNGS()
2133bf388a1fSBarry Smith M*/
2134bf388a1fSBarry Smith 
2135c79ef259SPeter Brune /*@C
2136be95d8f1SBarry Smith    SNESSetNGS - Sets the user nonlinear Gauss-Seidel routine for
2137c79ef259SPeter Brune    use with composed nonlinear solvers.
2138c79ef259SPeter Brune 
2139c79ef259SPeter Brune    Input Parameters:
2140c79ef259SPeter Brune +  snes   - the SNES context
2141be95d8f1SBarry Smith .  f - function evaluation routine to apply Gauss-Seidel see SNESNGSFunction
2142c79ef259SPeter Brune -  ctx    - [optional] user-defined context for private data for the
21430298fd71SBarry Smith             smoother evaluation routine (may be NULL)
2144c79ef259SPeter Brune 
2145c79ef259SPeter Brune    Notes:
2146be95d8f1SBarry Smith    The NGS routines are used by the composed nonlinear solver to generate
2147c79ef259SPeter Brune     a problem appropriate update to the solution, particularly FAS.
2148c79ef259SPeter Brune 
2149d28543b3SPeter Brune    Level: intermediate
2150c79ef259SPeter Brune 
2151be95d8f1SBarry Smith .seealso: SNESGetFunction(), SNESComputeNGS()
2152c79ef259SPeter Brune @*/
2153be95d8f1SBarry Smith PetscErrorCode SNESSetNGS(SNES snes,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx)
21546cab3a1bSJed Brown {
21556cab3a1bSJed Brown   PetscErrorCode ierr;
21566cab3a1bSJed Brown   DM             dm;
21576cab3a1bSJed Brown 
2158646217ecSPeter Brune   PetscFunctionBegin;
21596cab3a1bSJed Brown   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
21606cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2161be95d8f1SBarry Smith   ierr = DMSNESSetNGS(dm,f,ctx);CHKERRQ(ierr);
2162646217ecSPeter Brune   PetscFunctionReturn(0);
2163646217ecSPeter Brune }
2164646217ecSPeter Brune 
2165bbc1464cSBarry Smith /*
2166bbc1464cSBarry Smith      This is used for -snes_mf_operator; it uses a duplicate of snes->jacobian_pre because snes->jacobian_pre cannot be
2167bbc1464cSBarry Smith    changed during the KSPSolve()
2168bbc1464cSBarry Smith */
2169bbc1464cSBarry Smith PetscErrorCode SNESPicardComputeMFFunction(SNES snes,Vec x,Vec f,void *ctx)
2170bbc1464cSBarry Smith {
2171bbc1464cSBarry Smith   PetscErrorCode ierr;
2172bbc1464cSBarry Smith   DM             dm;
2173bbc1464cSBarry Smith   DMSNES         sdm;
2174bbc1464cSBarry Smith 
2175bbc1464cSBarry Smith   PetscFunctionBegin;
2176bbc1464cSBarry Smith   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2177bbc1464cSBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
2178bbc1464cSBarry Smith   if (!sdm->ops->computepjacobian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard Jacobian.");
2179bbc1464cSBarry Smith   /*  A(x)*x - b(x) */
2180bbc1464cSBarry Smith   if (sdm->ops->computepfunction) {
2181bbc1464cSBarry Smith     PetscStackPush("SNES Picard user function");
2182bbc1464cSBarry Smith     ierr = (*sdm->ops->computepfunction)(snes,x,f,sdm->pctx);CHKERRQ(ierr);
2183bbc1464cSBarry Smith     PetscStackPop;
2184bbc1464cSBarry Smith     ierr = VecScale(f,-1.0);CHKERRQ(ierr);
2185bbc1464cSBarry Smith     if (!snes->picard) {
2186*0df40c35SBarry Smith       /* Cannot share nonzero pattern because of the possible use of SNESComputeJacobianDefault() */
2187*0df40c35SBarry Smith       ierr = MatDuplicate(snes->jacobian_pre,MAT_DO_NOT_COPY_VALUES,&snes->picard);CHKERRQ(ierr);
2188bbc1464cSBarry Smith     }
2189bbc1464cSBarry Smith     PetscStackPush("SNES Picard user Jacobian");
2190bbc1464cSBarry Smith     ierr = (*sdm->ops->computepjacobian)(snes,x,snes->picard,snes->picard,sdm->pctx);CHKERRQ(ierr);
2191bbc1464cSBarry Smith     PetscStackPop;
2192bbc1464cSBarry Smith     ierr = MatMultAdd(snes->picard,x,f,f);CHKERRQ(ierr);
2193bbc1464cSBarry Smith   } else {
2194bbc1464cSBarry Smith     PetscStackPush("SNES Picard user Jacobian");
2195bbc1464cSBarry Smith     ierr = (*sdm->ops->computepjacobian)(snes,x,snes->picard,snes->picard,sdm->pctx);CHKERRQ(ierr);
2196bbc1464cSBarry Smith     PetscStackPop;
2197bbc1464cSBarry Smith     ierr = MatMult(snes->picard,x,f);CHKERRQ(ierr);
2198bbc1464cSBarry Smith   }
2199bbc1464cSBarry Smith   PetscFunctionReturn(0);
2200bbc1464cSBarry Smith }
2201bbc1464cSBarry Smith 
220225acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeFunction(SNES snes,Vec x,Vec f,void *ctx)
22038b0a5094SBarry Smith {
22048b0a5094SBarry Smith   PetscErrorCode ierr;
2205e03ab78fSPeter Brune   DM             dm;
2206942e3340SBarry Smith   DMSNES         sdm;
22076cab3a1bSJed Brown 
22088b0a5094SBarry Smith   PetscFunctionBegin;
2209e03ab78fSPeter Brune   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2210942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
221125acbd8eSLisandro Dalcin   if (!sdm->ops->computepjacobian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard Jacobian.");
22128b0a5094SBarry Smith   /*  A(x)*x - b(x) */
2213bbc1464cSBarry Smith   if (sdm->ops->computepfunction){
221425acbd8eSLisandro Dalcin     PetscStackPush("SNES Picard user function");
221522c6f798SBarry Smith     ierr = (*sdm->ops->computepfunction)(snes,x,f,sdm->pctx);CHKERRQ(ierr);
221625acbd8eSLisandro Dalcin     PetscStackPop;
2217bbc1464cSBarry Smith     ierr = VecScale(f,-1.0);CHKERRQ(ierr);
221825acbd8eSLisandro Dalcin     PetscStackPush("SNES Picard user Jacobian");
2219d1e9a80fSBarry Smith     ierr = (*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx);CHKERRQ(ierr);
222025acbd8eSLisandro Dalcin     PetscStackPop;
2221bbc1464cSBarry Smith     ierr = MatMultAdd(snes->jacobian_pre,x,f,f);CHKERRQ(ierr);
2222bbc1464cSBarry Smith   } else {
2223bbc1464cSBarry Smith     PetscStackPush("SNES Picard user Jacobian");
2224bbc1464cSBarry Smith     ierr = (*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx);CHKERRQ(ierr);
2225bbc1464cSBarry Smith     PetscStackPop;
2226bbc1464cSBarry Smith     ierr = MatMult(snes->jacobian_pre,x,f);CHKERRQ(ierr);
2227bbc1464cSBarry Smith   }
22288b0a5094SBarry Smith   PetscFunctionReturn(0);
22298b0a5094SBarry Smith }
22308b0a5094SBarry Smith 
223125acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeJacobian(SNES snes,Vec x1,Mat J,Mat B,void *ctx)
22328b0a5094SBarry Smith {
2233bbc1464cSBarry Smith   PetscErrorCode ierr;
22348b0a5094SBarry Smith   PetscFunctionBegin;
2235e03ab78fSPeter Brune   /* the jacobian matrix should be pre-filled in SNESPicardComputeFunction */
2236bbc1464cSBarry Smith   /* must assembly if matrix-free to get the last SNES solution */
2237bbc1464cSBarry Smith   ierr = MatAssemblyBegin(J,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2238bbc1464cSBarry Smith   ierr = MatAssemblyEnd(J,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
22398b0a5094SBarry Smith   PetscFunctionReturn(0);
22408b0a5094SBarry Smith }
22418b0a5094SBarry Smith 
22428b0a5094SBarry Smith /*@C
2243bbc1464cSBarry Smith    SNESSetPicard - Use SNES to solve the system A(x) x = bp(x) + b via a Picard type iteration (Picard linearization)
22448b0a5094SBarry Smith 
22458b0a5094SBarry Smith    Logically Collective on SNES
22468b0a5094SBarry Smith 
22478b0a5094SBarry Smith    Input Parameters:
22488b0a5094SBarry Smith +  snes - the SNES context
22498b0a5094SBarry Smith .  r - vector to store function value
2250bbc1464cSBarry Smith .  bp - function evaluation routine
2251e5d3d808SBarry Smith .  Amat - matrix with which A(x) x - b(x) is to be computed
2252e5d3d808SBarry Smith .  Pmat - matrix from which preconditioner is computed (usually the same as Amat)
2253411c0326SBarry Smith .  J  - function to compute matrix value, see SNESJacobianFunction for details on its calling sequence
22548b0a5094SBarry Smith -  ctx - [optional] user-defined context for private data for the
22550298fd71SBarry Smith          function evaluation routine (may be NULL)
22568b0a5094SBarry Smith 
22578b0a5094SBarry Smith    Notes:
2258f450aa47SBarry 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
2259f450aa47SBarry 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.
2260f450aa47SBarry Smith 
22618b0a5094SBarry Smith     One can call SNESSetPicard() or SNESSetFunction() (and possibly SNESSetJacobian()) but cannot call both
22628b0a5094SBarry Smith 
22638b0a5094SBarry 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}
22648b0a5094SBarry 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.
22658b0a5094SBarry Smith 
22668b0a5094SBarry Smith      Run with -snes_mf_operator to solve the system with Newton's method using A(x^{n}) to construct the preconditioner.
22678b0a5094SBarry Smith 
22680d04baf8SBarry Smith    We implement the defect correction form of the Picard iteration because it converges much more generally when inexact linear solvers are used then
22690d04baf8SBarry Smith    the direct Picard iteration A(x^n) x^{n+1} = b(x^n)
22708b0a5094SBarry Smith 
22718b0a5094SBarry 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
22728b0a5094SBarry 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
22738b0a5094SBarry Smith    different please contact us at petsc-dev@mcs.anl.gov and we'll have an entirely new argument :-).
22748b0a5094SBarry Smith 
2275bbc1464cSBarry Smith    When used with -snes_mf_operator this will run matrix-free Newton's method where the matrix-vector product is of the true Jacobian of A(x)x - b(x).
2276bbc1464cSBarry Smith 
2277f450aa47SBarry Smith    Level: intermediate
22788b0a5094SBarry Smith 
2279411c0326SBarry Smith .seealso: SNESGetFunction(), SNESSetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESGetPicard(), SNESLineSearchPreCheckPicard(), SNESJacobianFunction
22808b0a5094SBarry Smith @*/
2281bbc1464cSBarry Smith PetscErrorCode  SNESSetPicard(SNES snes,Vec r,PetscErrorCode (*bp)(SNES,Vec,Vec,void*),Mat Amat, Mat Pmat, PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx)
22828b0a5094SBarry Smith {
22838b0a5094SBarry Smith   PetscErrorCode ierr;
2284e03ab78fSPeter Brune   DM             dm;
2285e03ab78fSPeter Brune 
22868b0a5094SBarry Smith   PetscFunctionBegin;
22878b0a5094SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2288e03ab78fSPeter Brune   ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
2289bbc1464cSBarry Smith   ierr = DMSNESSetPicard(dm,bp,J,ctx);CHKERRQ(ierr);
2290bbc1464cSBarry Smith   ierr = DMSNESSetMFFunction(dm,SNESPicardComputeMFFunction,ctx);CHKERRQ(ierr);
22918b0a5094SBarry Smith   ierr = SNESSetFunction(snes,r,SNESPicardComputeFunction,ctx);CHKERRQ(ierr);
2292e5d3d808SBarry Smith   ierr = SNESSetJacobian(snes,Amat,Pmat,SNESPicardComputeJacobian,ctx);CHKERRQ(ierr);
22938b0a5094SBarry Smith   PetscFunctionReturn(0);
22948b0a5094SBarry Smith }
22958b0a5094SBarry Smith 
22967971a8bfSPeter Brune /*@C
22977971a8bfSPeter Brune    SNESGetPicard - Returns the context for the Picard iteration
22987971a8bfSPeter Brune 
22997971a8bfSPeter Brune    Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet.
23007971a8bfSPeter Brune 
23017971a8bfSPeter Brune    Input Parameter:
23027971a8bfSPeter Brune .  snes - the SNES context
23037971a8bfSPeter Brune 
23047971a8bfSPeter Brune    Output Parameter:
23050298fd71SBarry Smith +  r - the function (or NULL)
2306f8b49ee9SBarry Smith .  f - the function (or NULL); see SNESFunction for calling sequence details
2307e4357dc4SBarry Smith .  Amat - the matrix used to defined the operation A(x) x - b(x) (or NULL)
2308e4357dc4SBarry Smith .  Pmat  - the matrix from which the preconditioner will be constructed (or NULL)
2309f8b49ee9SBarry Smith .  J - the function for matrix evaluation (or NULL); see SNESJacobianFunction for calling sequence details
23100298fd71SBarry Smith -  ctx - the function context (or NULL)
23117971a8bfSPeter Brune 
23127971a8bfSPeter Brune    Level: advanced
23137971a8bfSPeter Brune 
2314e4357dc4SBarry Smith .seealso: SNESSetPicard(), SNESGetFunction(), SNESGetJacobian(), SNESGetDM(), SNESFunction, SNESJacobianFunction
23157971a8bfSPeter Brune @*/
2316d1e9a80fSBarry 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)
23177971a8bfSPeter Brune {
23187971a8bfSPeter Brune   PetscErrorCode ierr;
23197971a8bfSPeter Brune   DM             dm;
23207971a8bfSPeter Brune 
23217971a8bfSPeter Brune   PetscFunctionBegin;
23227971a8bfSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
23230298fd71SBarry Smith   ierr = SNESGetFunction(snes,r,NULL,NULL);CHKERRQ(ierr);
2324e4357dc4SBarry Smith   ierr = SNESGetJacobian(snes,Amat,Pmat,NULL,NULL);CHKERRQ(ierr);
23257971a8bfSPeter Brune   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2326f8b49ee9SBarry Smith   ierr = DMSNESGetPicard(dm,f,J,ctx);CHKERRQ(ierr);
23277971a8bfSPeter Brune   PetscFunctionReturn(0);
23287971a8bfSPeter Brune }
23297971a8bfSPeter Brune 
2330d25893d9SBarry Smith /*@C
2331d25893d9SBarry Smith    SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem
2332d25893d9SBarry Smith 
2333d25893d9SBarry Smith    Logically Collective on SNES
2334d25893d9SBarry Smith 
2335d25893d9SBarry Smith    Input Parameters:
2336d25893d9SBarry Smith +  snes - the SNES context
2337d25893d9SBarry Smith .  func - function evaluation routine
2338d25893d9SBarry Smith -  ctx - [optional] user-defined context for private data for the
23390298fd71SBarry Smith          function evaluation routine (may be NULL)
2340d25893d9SBarry Smith 
2341d25893d9SBarry Smith    Calling sequence of func:
2342d25893d9SBarry Smith $    func (SNES snes,Vec x,void *ctx);
2343d25893d9SBarry Smith 
2344d25893d9SBarry Smith .  f - function vector
2345d25893d9SBarry Smith -  ctx - optional user-defined function context
2346d25893d9SBarry Smith 
2347d25893d9SBarry Smith    Level: intermediate
2348d25893d9SBarry Smith 
2349d25893d9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian()
2350d25893d9SBarry Smith @*/
2351d25893d9SBarry Smith PetscErrorCode  SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx)
2352d25893d9SBarry Smith {
2353d25893d9SBarry Smith   PetscFunctionBegin;
2354d25893d9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2355d25893d9SBarry Smith   if (func) snes->ops->computeinitialguess = func;
2356d25893d9SBarry Smith   if (ctx)  snes->initialguessP            = ctx;
2357d25893d9SBarry Smith   PetscFunctionReturn(0);
2358d25893d9SBarry Smith }
2359d25893d9SBarry Smith 
23603ab0aad5SBarry Smith /* --------------------------------------------------------------- */
23611096aae1SMatthew Knepley /*@C
23621096aae1SMatthew Knepley    SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set
23631096aae1SMatthew Knepley    it assumes a zero right hand side.
23641096aae1SMatthew Knepley 
23653f9fe445SBarry Smith    Logically Collective on SNES
23661096aae1SMatthew Knepley 
23671096aae1SMatthew Knepley    Input Parameter:
23681096aae1SMatthew Knepley .  snes - the SNES context
23691096aae1SMatthew Knepley 
23701096aae1SMatthew Knepley    Output Parameter:
23710298fd71SBarry Smith .  rhs - the right hand side vector or NULL if the right hand side vector is null
23721096aae1SMatthew Knepley 
23731096aae1SMatthew Knepley    Level: intermediate
23741096aae1SMatthew Knepley 
237585385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
23761096aae1SMatthew Knepley @*/
23777087cfbeSBarry Smith PetscErrorCode  SNESGetRhs(SNES snes,Vec *rhs)
23781096aae1SMatthew Knepley {
23791096aae1SMatthew Knepley   PetscFunctionBegin;
23800700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
23811096aae1SMatthew Knepley   PetscValidPointer(rhs,2);
238285385478SLisandro Dalcin   *rhs = snes->vec_rhs;
23831096aae1SMatthew Knepley   PetscFunctionReturn(0);
23841096aae1SMatthew Knepley }
23851096aae1SMatthew Knepley 
23869b94acceSBarry Smith /*@
2387bf388a1fSBarry Smith    SNESComputeFunction - Calls the function that has been set with SNESSetFunction().
23889b94acceSBarry Smith 
2389c7afd0dbSLois Curfman McInnes    Collective on SNES
2390c7afd0dbSLois Curfman McInnes 
23919b94acceSBarry Smith    Input Parameters:
2392c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2393c7afd0dbSLois Curfman McInnes -  x - input vector
23949b94acceSBarry Smith 
23959b94acceSBarry Smith    Output Parameter:
23963638b69dSLois Curfman McInnes .  y - function vector, as set by SNESSetFunction()
23979b94acceSBarry Smith 
23981bffabb2SLois Curfman McInnes    Notes:
239936851e7fSLois Curfman McInnes    SNESComputeFunction() is typically used within nonlinear solvers
2400bbc1464cSBarry Smith    implementations, so users would not generally call this routine themselves.
240136851e7fSLois Curfman McInnes 
240236851e7fSLois Curfman McInnes    Level: developer
240336851e7fSLois Curfman McInnes 
2404bbc1464cSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction(), SNESComputeMFFunction()
24059b94acceSBarry Smith @*/
24067087cfbeSBarry Smith PetscErrorCode  SNESComputeFunction(SNES snes,Vec x,Vec y)
24079b94acceSBarry Smith {
2408dfbe8321SBarry Smith   PetscErrorCode ierr;
24096cab3a1bSJed Brown   DM             dm;
2410942e3340SBarry Smith   DMSNES         sdm;
24119b94acceSBarry Smith 
24123a40ed3dSBarry Smith   PetscFunctionBegin;
24130700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
24140700a824SBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
24150700a824SBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2416c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,x,2);
2417c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,3);
241862796dfbSBarry Smith   ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr);
2419184914b5SBarry Smith 
24206cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2421942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
242232f3f7c2SPeter Brune   if (sdm->ops->computefunction) {
242394db00ebSBarry Smith     if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) {
2424ccf3c845SPeter Brune       ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
242594db00ebSBarry Smith     }
24268860a134SJunchao Zhang     ierr = VecLockReadPush(x);CHKERRQ(ierr);
2427d64ed03dSBarry Smith     PetscStackPush("SNES user function");
24288ddeebeaSSteve Benbow     /* ensure domainerror is false prior to computefunction evaluation (may not have been reset) */
24298ddeebeaSSteve Benbow     snes->domainerror = PETSC_FALSE;
243022c6f798SBarry Smith     ierr = (*sdm->ops->computefunction)(snes,x,y,sdm->functionctx);CHKERRQ(ierr);
2431d64ed03dSBarry Smith     PetscStackPop;
24328860a134SJunchao Zhang     ierr = VecLockReadPop(x);CHKERRQ(ierr);
243394db00ebSBarry Smith     if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) {
2434ccf3c845SPeter Brune       ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
243594db00ebSBarry Smith     }
2436c90fad12SPeter Brune   } else if (snes->vec_rhs) {
2437c90fad12SPeter Brune     ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr);
2438644e2e5bSBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve().");
243985385478SLisandro Dalcin   if (snes->vec_rhs) {
244085385478SLisandro Dalcin     ierr = VecAXPY(y,-1.0,snes->vec_rhs);CHKERRQ(ierr);
24413ab0aad5SBarry Smith   }
2442ae3c334cSLois Curfman McInnes   snes->nfuncs++;
2443422a814eSBarry Smith   /*
2444422a814eSBarry Smith      domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will
2445422a814eSBarry Smith      propagate the value to all processes
2446422a814eSBarry Smith   */
2447422a814eSBarry Smith   if (snes->domainerror) {
2448422a814eSBarry Smith     ierr = VecSetInf(y);CHKERRQ(ierr);
2449422a814eSBarry Smith   }
24503a40ed3dSBarry Smith   PetscFunctionReturn(0);
24519b94acceSBarry Smith }
24529b94acceSBarry Smith 
2453c79ef259SPeter Brune /*@
2454bbc1464cSBarry Smith    SNESComputeMFFunction - Calls the function that has been set with SNESSetMFFunction().
2455bbc1464cSBarry Smith 
2456bbc1464cSBarry Smith    Collective on SNES
2457bbc1464cSBarry Smith 
2458bbc1464cSBarry Smith    Input Parameters:
2459bbc1464cSBarry Smith +  snes - the SNES context
2460bbc1464cSBarry Smith -  x - input vector
2461bbc1464cSBarry Smith 
2462bbc1464cSBarry Smith    Output Parameter:
2463bbc1464cSBarry Smith .  y - function vector, as set by SNESSetMFFunction()
2464bbc1464cSBarry Smith 
2465bbc1464cSBarry Smith    Notes:
2466bbc1464cSBarry Smith        SNESComputeMFFunction() is used within the matrix vector products called by the matrix created with MatCreateSNESMF()
2467bbc1464cSBarry Smith    so users would not generally call this routine themselves.
2468bbc1464cSBarry Smith 
2469bbc1464cSBarry Smith        Since this function is intended for use with finite differencing it does not subtract the right hand side vector provided with SNESSolve()
2470bbc1464cSBarry Smith     while SNESComputeFunction() does. As such, this routine cannot be used with  MatMFFDSetBase() with a provided F function value even if it applies the
2471bbc1464cSBarry Smith     same function as SNESComputeFunction() if a SNESSolve() right hand side vector is use because the two functions difference would include this right hand side function.
2472bbc1464cSBarry Smith 
2473bbc1464cSBarry Smith    Level: developer
2474bbc1464cSBarry Smith 
2475bbc1464cSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction(), SNESComputeFunction(), MatCreateSNESMF
2476bbc1464cSBarry Smith @*/
2477bbc1464cSBarry Smith PetscErrorCode  SNESComputeMFFunction(SNES snes,Vec x,Vec y)
2478bbc1464cSBarry Smith {
2479bbc1464cSBarry Smith   PetscErrorCode ierr;
2480bbc1464cSBarry Smith   DM             dm;
2481bbc1464cSBarry Smith   DMSNES         sdm;
2482bbc1464cSBarry Smith 
2483bbc1464cSBarry Smith   PetscFunctionBegin;
2484bbc1464cSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2485bbc1464cSBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2486bbc1464cSBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2487bbc1464cSBarry Smith   PetscCheckSameComm(snes,1,x,2);
2488bbc1464cSBarry Smith   PetscCheckSameComm(snes,1,y,3);
2489bbc1464cSBarry Smith   ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr);
2490bbc1464cSBarry Smith 
2491bbc1464cSBarry Smith   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2492bbc1464cSBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
2493bbc1464cSBarry Smith   ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
2494bbc1464cSBarry Smith   ierr = VecLockReadPush(x);CHKERRQ(ierr);
2495bbc1464cSBarry Smith   PetscStackPush("SNES user function");
2496bbc1464cSBarry Smith   /* ensure domainerror is false prior to computefunction evaluation (may not have been reset) */
2497bbc1464cSBarry Smith   snes->domainerror = PETSC_FALSE;
2498bbc1464cSBarry Smith   ierr = (*sdm->ops->computemffunction)(snes,x,y,sdm->mffunctionctx);CHKERRQ(ierr);
2499bbc1464cSBarry Smith   PetscStackPop;
2500bbc1464cSBarry Smith   ierr = VecLockReadPop(x);CHKERRQ(ierr);
2501bbc1464cSBarry Smith   ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
2502bbc1464cSBarry Smith   snes->nfuncs++;
2503bbc1464cSBarry Smith   /*
2504bbc1464cSBarry Smith      domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will
2505bbc1464cSBarry Smith      propagate the value to all processes
2506bbc1464cSBarry Smith   */
2507bbc1464cSBarry Smith   if (snes->domainerror) {
2508bbc1464cSBarry Smith     ierr = VecSetInf(y);CHKERRQ(ierr);
2509bbc1464cSBarry Smith   }
2510bbc1464cSBarry Smith   PetscFunctionReturn(0);
2511bbc1464cSBarry Smith }
2512bbc1464cSBarry Smith 
2513bbc1464cSBarry Smith /*@
2514be95d8f1SBarry Smith    SNESComputeNGS - Calls the Gauss-Seidel function that has been set with  SNESSetNGS().
2515c79ef259SPeter Brune 
2516c79ef259SPeter Brune    Collective on SNES
2517c79ef259SPeter Brune 
2518c79ef259SPeter Brune    Input Parameters:
2519c79ef259SPeter Brune +  snes - the SNES context
2520c79ef259SPeter Brune .  x - input vector
2521c79ef259SPeter Brune -  b - rhs vector
2522c79ef259SPeter Brune 
2523c79ef259SPeter Brune    Output Parameter:
2524c79ef259SPeter Brune .  x - new solution vector
2525c79ef259SPeter Brune 
2526c79ef259SPeter Brune    Notes:
2527be95d8f1SBarry Smith    SNESComputeNGS() is typically used within composed nonlinear solver
2528c79ef259SPeter Brune    implementations, so most users would not generally call this routine
2529c79ef259SPeter Brune    themselves.
2530c79ef259SPeter Brune 
2531c79ef259SPeter Brune    Level: developer
2532c79ef259SPeter Brune 
2533be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESComputeFunction()
2534c79ef259SPeter Brune @*/
2535be95d8f1SBarry Smith PetscErrorCode  SNESComputeNGS(SNES snes,Vec b,Vec x)
2536646217ecSPeter Brune {
2537646217ecSPeter Brune   PetscErrorCode ierr;
25386cab3a1bSJed Brown   DM             dm;
2539942e3340SBarry Smith   DMSNES         sdm;
2540646217ecSPeter Brune 
2541646217ecSPeter Brune   PetscFunctionBegin;
2542646217ecSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2543646217ecSPeter Brune   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2544646217ecSPeter Brune   if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,3);
2545646217ecSPeter Brune   PetscCheckSameComm(snes,1,x,2);
2546646217ecSPeter Brune   if (b) PetscCheckSameComm(snes,1,b,3);
254762796dfbSBarry Smith   if (b) {ierr = VecValidValues(b,2,PETSC_TRUE);CHKERRQ(ierr);}
2548be95d8f1SBarry Smith   ierr = PetscLogEventBegin(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr);
25496cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2550942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
255122c6f798SBarry Smith   if (sdm->ops->computegs) {
25528860a134SJunchao Zhang     if (b) {ierr = VecLockReadPush(b);CHKERRQ(ierr);}
2553be95d8f1SBarry Smith     PetscStackPush("SNES user NGS");
255422c6f798SBarry Smith     ierr = (*sdm->ops->computegs)(snes,x,b,sdm->gsctx);CHKERRQ(ierr);
2555646217ecSPeter Brune     PetscStackPop;
25568860a134SJunchao Zhang     if (b) {ierr = VecLockReadPop(b);CHKERRQ(ierr);}
2557be95d8f1SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetNGS() before SNESComputeNGS(), likely called from SNESSolve().");
2558be95d8f1SBarry Smith   ierr = PetscLogEventEnd(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr);
2559646217ecSPeter Brune   PetscFunctionReturn(0);
2560646217ecSPeter Brune }
2561646217ecSPeter Brune 
2562e885f1abSBarry Smith PetscErrorCode SNESTestJacobian(SNES snes)
2563e885f1abSBarry Smith {
256412837594SBarry Smith   Mat               A,B,C,D,jacobian;
2565e885f1abSBarry Smith   Vec               x = snes->vec_sol,f = snes->vec_func;
2566e885f1abSBarry Smith   PetscErrorCode    ierr;
2567e885f1abSBarry Smith   PetscReal         nrm,gnorm;
256881e7118cSBarry Smith   PetscReal         threshold = 1.e-5;
25690e276705SLisandro Dalcin   MatType           mattype;
2570e885f1abSBarry Smith   PetscInt          m,n,M,N;
2571e885f1abSBarry Smith   void              *functx;
25722cd624f9SStefano Zampini   PetscBool         complete_print = PETSC_FALSE,threshold_print = PETSC_FALSE,test = PETSC_FALSE,flg,istranspose;
25733325ff46SBarry Smith   PetscViewer       viewer,mviewer;
2574e885f1abSBarry Smith   MPI_Comm          comm;
2575e885f1abSBarry Smith   PetscInt          tabs;
257612837594SBarry Smith   static PetscBool  directionsprinted = PETSC_FALSE;
25773325ff46SBarry Smith   PetscViewerFormat format;
2578e885f1abSBarry Smith 
2579e885f1abSBarry Smith   PetscFunctionBegin;
2580fc35ed60SBarry Smith   ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr);
258112837594SBarry Smith   ierr = PetscOptionsName("-snes_test_jacobian","Compare hand-coded and finite difference Jacobians","None",&test);CHKERRQ(ierr);
258212837594SBarry Smith   ierr = PetscOptionsReal("-snes_test_jacobian", "Threshold for element difference between hand-coded and finite difference being meaningful", "None", threshold, &threshold,NULL);CHKERRQ(ierr);
25833325ff46SBarry 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);
258418d89885SKarl Rupp   if (!complete_print) {
2585455a5933SJed Brown     ierr = PetscOptionsDeprecated("-snes_test_jacobian_display","-snes_test_jacobian_view","3.13",NULL);CHKERRQ(ierr);
258618d89885SKarl Rupp     ierr = PetscOptionsViewer("-snes_test_jacobian_display","Display difference between hand-coded and finite difference Jacobians","None",&mviewer,&format,&complete_print);CHKERRQ(ierr);
258718d89885SKarl Rupp   }
258818d89885SKarl Rupp   /* for compatibility with PETSc 3.9 and older. */
2589455a5933SJed 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);
259018d89885SKarl 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);
2591e885f1abSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
2592e885f1abSBarry Smith   if (!test) PetscFunctionReturn(0);
2593e885f1abSBarry Smith 
2594e885f1abSBarry Smith   ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr);
2595e885f1abSBarry Smith   ierr = PetscViewerASCIIGetStdout(comm,&viewer);CHKERRQ(ierr);
2596e885f1abSBarry Smith   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
2597e885f1abSBarry Smith   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)snes)->tablevel);CHKERRQ(ierr);
259812837594SBarry Smith   ierr = PetscViewerASCIIPrintf(viewer,"  ---------- Testing Jacobian -------------\n");CHKERRQ(ierr);
259912837594SBarry Smith   if (!complete_print && !directionsprinted) {
260012837594SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Run with -snes_test_jacobian_view and optionally -snes_test_jacobian <threshold> to show difference\n");CHKERRQ(ierr);
2601fc35ed60SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"    of hand-coded and finite difference Jacobian entries greater than <threshold>.\n");CHKERRQ(ierr);
260212837594SBarry Smith   }
260312837594SBarry Smith   if (!directionsprinted) {
260412837594SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Testing hand-coded Jacobian, if (for double precision runs) ||J - Jfd||_F/||J||_F is\n");CHKERRQ(ierr);
2605e885f1abSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"    O(1.e-8), the hand-coded Jacobian is probably correct.\n");CHKERRQ(ierr);
260612837594SBarry Smith     directionsprinted = PETSC_TRUE;
2607e885f1abSBarry Smith   }
26083325ff46SBarry Smith   if (complete_print) {
26093325ff46SBarry Smith     ierr = PetscViewerPushFormat(mviewer,format);CHKERRQ(ierr);
2610e885f1abSBarry Smith   }
2611e885f1abSBarry Smith 
261212837594SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)snes->jacobian,MATMFFD,&flg);CHKERRQ(ierr);
261312837594SBarry Smith   if (!flg) jacobian = snes->jacobian;
261412837594SBarry Smith   else jacobian = snes->jacobian_pre;
261512837594SBarry Smith 
2616a82339d0SMatthew G. Knepley   if (!x) {
2617a82339d0SMatthew G. Knepley     ierr = MatCreateVecs(jacobian, &x, NULL);CHKERRQ(ierr);
2618a82339d0SMatthew G. Knepley   } else {
2619a82339d0SMatthew G. Knepley     ierr = PetscObjectReference((PetscObject) x);CHKERRQ(ierr);
2620a82339d0SMatthew G. Knepley   }
2621a82339d0SMatthew G. Knepley   if (!f) {
2622a82339d0SMatthew G. Knepley     ierr = VecDuplicate(x, &f);CHKERRQ(ierr);
2623a82339d0SMatthew G. Knepley   } else {
2624a82339d0SMatthew G. Knepley     ierr = PetscObjectReference((PetscObject) f);CHKERRQ(ierr);
2625a82339d0SMatthew G. Knepley   }
2626a82339d0SMatthew G. Knepley   /* evaluate the function at this point because SNESComputeJacobianDefault() assumes that the function has been evaluated and put into snes->vec_func */
2627a82339d0SMatthew G. Knepley   ierr = SNESComputeFunction(snes,x,f);CHKERRQ(ierr);
2628a82339d0SMatthew G. Knepley   ierr = VecDestroy(&f);CHKERRQ(ierr);
26292cd624f9SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)snes,SNESKSPTRANSPOSEONLY,&istranspose);CHKERRQ(ierr);
263012837594SBarry Smith   while (jacobian) {
26312cd624f9SStefano Zampini     Mat JT = NULL, Jsave = NULL;
26322cd624f9SStefano Zampini 
26332cd624f9SStefano Zampini     if (istranspose) {
26342cd624f9SStefano Zampini       ierr = MatCreateTranspose(jacobian,&JT);CHKERRQ(ierr);
26352cd624f9SStefano Zampini       Jsave = jacobian;
26362cd624f9SStefano Zampini       jacobian = JT;
26372cd624f9SStefano Zampini     }
26380e276705SLisandro Dalcin     ierr = PetscObjectBaseTypeCompareAny((PetscObject)jacobian,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPISBAIJ,"");CHKERRQ(ierr);
263912837594SBarry Smith     if (flg) {
264012837594SBarry Smith       A    = jacobian;
264112837594SBarry Smith       ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
264212837594SBarry Smith     } else {
26430bacdadaSStefano Zampini       ierr = MatComputeOperator(jacobian,MATAIJ,&A);CHKERRQ(ierr);
264412837594SBarry Smith     }
2645e885f1abSBarry Smith 
26460e276705SLisandro Dalcin     ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
2647e885f1abSBarry Smith     ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
2648e885f1abSBarry Smith     ierr = MatGetLocalSize(A,&m,&n);CHKERRQ(ierr);
26490e276705SLisandro Dalcin     ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
26500e276705SLisandro Dalcin     ierr = MatSetType(B,mattype);CHKERRQ(ierr);
2651e885f1abSBarry Smith     ierr = MatSetSizes(B,m,n,M,N);CHKERRQ(ierr);
26520e276705SLisandro Dalcin     ierr = MatSetBlockSizesFromMats(B,A,A);CHKERRQ(ierr);
2653e885f1abSBarry Smith     ierr = MatSetUp(B);CHKERRQ(ierr);
2654e885f1abSBarry Smith     ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr);
2655e885f1abSBarry Smith 
2656e885f1abSBarry Smith     ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr);
2657e885f1abSBarry Smith     ierr = SNESComputeJacobianDefault(snes,x,B,B,functx);CHKERRQ(ierr);
265812837594SBarry Smith 
265912837594SBarry Smith     ierr = MatDuplicate(B,MAT_COPY_VALUES,&D);CHKERRQ(ierr);
266012837594SBarry Smith     ierr = MatAYPX(D,-1.0,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
266112837594SBarry Smith     ierr = MatNorm(D,NORM_FROBENIUS,&nrm);CHKERRQ(ierr);
2662e885f1abSBarry Smith     ierr = MatNorm(A,NORM_FROBENIUS,&gnorm);CHKERRQ(ierr);
266312837594SBarry Smith     ierr = MatDestroy(&D);CHKERRQ(ierr);
266412837594SBarry Smith     if (!gnorm) gnorm = 1; /* just in case */
266512837594SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  ||J - Jfd||_F/||J||_F = %g, ||J - Jfd||_F = %g\n",(double)(nrm/gnorm),(double)nrm);CHKERRQ(ierr);
266612837594SBarry Smith 
2667e885f1abSBarry Smith     if (complete_print) {
266812837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Hand-coded Jacobian ----------\n");CHKERRQ(ierr);
26691878987eSStefano Zampini       ierr = MatView(A,mviewer);CHKERRQ(ierr);
267012837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Finite difference Jacobian ----------\n");CHKERRQ(ierr);
26713325ff46SBarry Smith       ierr = MatView(B,mviewer);CHKERRQ(ierr);
2672e885f1abSBarry Smith     }
2673e885f1abSBarry Smith 
2674df10fb39SFande Kong     if (threshold_print || complete_print) {
2675e885f1abSBarry Smith       PetscInt          Istart, Iend, *ccols, bncols, cncols, j, row;
2676e885f1abSBarry Smith       PetscScalar       *cvals;
2677e885f1abSBarry Smith       const PetscInt    *bcols;
2678e885f1abSBarry Smith       const PetscScalar *bvals;
2679e885f1abSBarry Smith 
2680e885f1abSBarry Smith       ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
26810e276705SLisandro Dalcin       ierr = MatSetType(C,mattype);CHKERRQ(ierr);
2682e885f1abSBarry Smith       ierr = MatSetSizes(C,m,n,M,N);CHKERRQ(ierr);
26830e276705SLisandro Dalcin       ierr = MatSetBlockSizesFromMats(C,A,A);CHKERRQ(ierr);
2684e885f1abSBarry Smith       ierr = MatSetUp(C);CHKERRQ(ierr);
2685e885f1abSBarry Smith       ierr = MatSetOption(C,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr);
26860e276705SLisandro Dalcin 
26870e276705SLisandro Dalcin       ierr = MatAYPX(B,-1.0,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
2688e885f1abSBarry Smith       ierr = MatGetOwnershipRange(B,&Istart,&Iend);CHKERRQ(ierr);
2689e885f1abSBarry Smith 
2690e885f1abSBarry Smith       for (row = Istart; row < Iend; row++) {
2691e885f1abSBarry Smith         ierr = MatGetRow(B,row,&bncols,&bcols,&bvals);CHKERRQ(ierr);
2692e885f1abSBarry Smith         ierr = PetscMalloc2(bncols,&ccols,bncols,&cvals);CHKERRQ(ierr);
2693e885f1abSBarry Smith         for (j = 0, cncols = 0; j < bncols; j++) {
269423a52b1dSBarry Smith           if (PetscAbsScalar(bvals[j]) > threshold) {
2695e885f1abSBarry Smith             ccols[cncols] = bcols[j];
2696e885f1abSBarry Smith             cvals[cncols] = bvals[j];
2697e885f1abSBarry Smith             cncols += 1;
2698e885f1abSBarry Smith           }
2699e885f1abSBarry Smith         }
2700e885f1abSBarry Smith         if (cncols) {
2701e885f1abSBarry Smith           ierr = MatSetValues(C,1,&row,cncols,ccols,cvals,INSERT_VALUES);CHKERRQ(ierr);
2702e885f1abSBarry Smith         }
2703e885f1abSBarry Smith         ierr = MatRestoreRow(B,row,&bncols,&bcols,&bvals);CHKERRQ(ierr);
2704e885f1abSBarry Smith         ierr = PetscFree2(ccols,cvals);CHKERRQ(ierr);
2705e885f1abSBarry Smith       }
2706e885f1abSBarry Smith       ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2707e885f1abSBarry Smith       ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
270812837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Hand-coded minus finite-difference Jacobian with tolerance %g ----------\n",(double)threshold);CHKERRQ(ierr);
270918d89885SKarl Rupp       ierr = MatView(C,complete_print ? mviewer : viewer);CHKERRQ(ierr);
2710e885f1abSBarry Smith       ierr = MatDestroy(&C);CHKERRQ(ierr);
2711e885f1abSBarry Smith     }
271212837594SBarry Smith     ierr = MatDestroy(&A);CHKERRQ(ierr);
2713e885f1abSBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
27142cd624f9SStefano Zampini     ierr = MatDestroy(&JT);CHKERRQ(ierr);
27152cd624f9SStefano Zampini     if (Jsave) jacobian = Jsave;
271612837594SBarry Smith     if (jacobian != snes->jacobian_pre) {
271712837594SBarry Smith       jacobian = snes->jacobian_pre;
271812837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  ---------- Testing Jacobian for preconditioner -------------\n");CHKERRQ(ierr);
271912837594SBarry Smith     }
272012837594SBarry Smith     else jacobian = NULL;
272112837594SBarry Smith   }
2722a82339d0SMatthew G. Knepley   ierr = VecDestroy(&x);CHKERRQ(ierr);
27233325ff46SBarry Smith   if (complete_print) {
27243325ff46SBarry Smith     ierr = PetscViewerPopFormat(mviewer);CHKERRQ(ierr);
27253325ff46SBarry Smith   }
2726a0c90127SFande Kong   if (mviewer) { ierr = PetscViewerDestroy(&mviewer);CHKERRQ(ierr); }
2727e885f1abSBarry Smith   ierr = PetscViewerASCIISetTab(viewer,tabs);CHKERRQ(ierr);
2728e885f1abSBarry Smith   PetscFunctionReturn(0);
2729e885f1abSBarry Smith }
2730e885f1abSBarry Smith 
273162fef451SLois Curfman McInnes /*@
2732bf388a1fSBarry Smith    SNESComputeJacobian - Computes the Jacobian matrix that has been set with SNESSetJacobian().
273362fef451SLois Curfman McInnes 
2734d083f849SBarry Smith    Collective on SNES
2735c7afd0dbSLois Curfman McInnes 
273662fef451SLois Curfman McInnes    Input Parameters:
2737c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2738c7afd0dbSLois Curfman McInnes -  x - input vector
273962fef451SLois Curfman McInnes 
274062fef451SLois Curfman McInnes    Output Parameters:
2741c7afd0dbSLois Curfman McInnes +  A - Jacobian matrix
2742d1e9a80fSBarry Smith -  B - optional preconditioning matrix
2743fee21e36SBarry Smith 
2744e35cf81dSBarry Smith   Options Database Keys:
2745e35cf81dSBarry Smith +    -snes_lag_preconditioner <lag>
2746693365a8SJed Brown .    -snes_lag_jacobian <lag>
2747455a5933SJed 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.
2748455a5933SJed 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
2749693365a8SJed Brown .    -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences
2750693365a8SJed Brown .    -snes_compare_explicit_draw  - Compare the computed Jacobian to the finite difference Jacobian and draw the result
2751693365a8SJed Brown .    -snes_compare_explicit_contour  - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result
27524c30e9fbSJed Brown .    -snes_compare_operator  - Make the comparison options above use the operator instead of the preconditioning matrix
275394d6a431SBarry Smith .    -snes_compare_coloring - Compute the finite difference Jacobian using coloring and display norms of difference
2754c01495d3SJed Brown .    -snes_compare_coloring_display - Compute the finite differece Jacobian using coloring and display verbose differences
2755c01495d3SJed Brown .    -snes_compare_coloring_threshold - Display only those matrix entries that differ by more than a given threshold
2756c01495d3SJed Brown .    -snes_compare_coloring_threshold_atol - Absolute tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold
2757c01495d3SJed Brown .    -snes_compare_coloring_threshold_rtol - Relative tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold
27584c30e9fbSJed Brown .    -snes_compare_coloring_draw - Compute the finite differece Jacobian using coloring and draw differences
2759c01495d3SJed Brown -    -snes_compare_coloring_draw_contour - Compute the finite differece Jacobian using coloring and show contours of matrices and differences
2760c01495d3SJed Brown 
2761e35cf81dSBarry Smith 
276262fef451SLois Curfman McInnes    Notes:
276362fef451SLois Curfman McInnes    Most users should not need to explicitly call this routine, as it
276462fef451SLois Curfman McInnes    is used internally within the nonlinear solvers.
276562fef451SLois Curfman McInnes 
276695452b02SPatrick Sanan    Developer Notes:
276795452b02SPatrick 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
2768e885f1abSBarry Smith       for with the SNESType of test that has been removed.
2769e885f1abSBarry Smith 
277036851e7fSLois Curfman McInnes    Level: developer
277136851e7fSLois Curfman McInnes 
2772e35cf81dSBarry Smith .seealso:  SNESSetJacobian(), KSPSetOperators(), MatStructure, SNESSetLagPreconditioner(), SNESSetLagJacobian()
277362fef451SLois Curfman McInnes @*/
2774d1e9a80fSBarry Smith PetscErrorCode  SNESComputeJacobian(SNES snes,Vec X,Mat A,Mat B)
27759b94acceSBarry Smith {
2776dfbe8321SBarry Smith   PetscErrorCode ierr;
2777ace3abfcSBarry Smith   PetscBool      flag;
27786cab3a1bSJed Brown   DM             dm;
2779942e3340SBarry Smith   DMSNES         sdm;
2780e0e3a89bSBarry Smith   KSP            ksp;
27813a40ed3dSBarry Smith 
27823a40ed3dSBarry Smith   PetscFunctionBegin;
27830700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
27840700a824SBarry Smith   PetscValidHeaderSpecific(X,VEC_CLASSID,2);
2785c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,X,2);
278662796dfbSBarry Smith   ierr = VecValidValues(X,2,PETSC_TRUE);CHKERRQ(ierr);
27876cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2788942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
27893232da50SPeter Brune 
2790ce94432eSBarry Smith   if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Must call SNESSetJacobian(), DMSNESSetJacobian(), DMDASNESSetJacobianLocal(), etc");
2791ebd3b9afSBarry Smith 
2792ebd3b9afSBarry Smith   /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */
2793ebd3b9afSBarry Smith 
2794fe3ffe1eSBarry Smith   if (snes->lagjacobian == -2) {
2795fe3ffe1eSBarry Smith     snes->lagjacobian = -1;
2796f5af7f23SKarl Rupp 
2797fe3ffe1eSBarry Smith     ierr = PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n");CHKERRQ(ierr);
2798fe3ffe1eSBarry Smith   } else if (snes->lagjacobian == -1) {
2799e35cf81dSBarry Smith     ierr = PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n");CHKERRQ(ierr);
280094ab13aaSBarry Smith     ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr);
2801ebd3b9afSBarry Smith     if (flag) {
280294ab13aaSBarry Smith       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
280394ab13aaSBarry Smith       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2804ebd3b9afSBarry Smith     }
2805e35cf81dSBarry Smith     PetscFunctionReturn(0);
280637ec4e1aSPeter Brune   } else if (snes->lagjacobian > 1 && (snes->iter + snes->jac_iter) % snes->lagjacobian) {
2807e35cf81dSBarry Smith     ierr = PetscInfo2(snes,"Reusing Jacobian/preconditioner because lag is %D and SNES iteration is %D\n",snes->lagjacobian,snes->iter);CHKERRQ(ierr);
280894ab13aaSBarry Smith     ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr);
2809ebd3b9afSBarry Smith     if (flag) {
281094ab13aaSBarry Smith       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
281194ab13aaSBarry Smith       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2812ebd3b9afSBarry Smith     }
2813e35cf81dSBarry Smith     PetscFunctionReturn(0);
2814e35cf81dSBarry Smith   }
2815efd4aadfSBarry Smith   if (snes->npc && snes->npcside== PC_LEFT) {
281694ab13aaSBarry Smith     ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
281794ab13aaSBarry Smith     ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2818d728fb7dSPeter Brune     PetscFunctionReturn(0);
2819d728fb7dSPeter Brune   }
2820e35cf81dSBarry Smith 
282194ab13aaSBarry Smith   ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr);
28228860a134SJunchao Zhang   ierr = VecLockReadPush(X);CHKERRQ(ierr);
2823d64ed03dSBarry Smith   PetscStackPush("SNES user Jacobian function");
2824d1e9a80fSBarry Smith   ierr = (*sdm->ops->computejacobian)(snes,X,A,B,sdm->jacobianctx);CHKERRQ(ierr);
2825d64ed03dSBarry Smith   PetscStackPop;
28268860a134SJunchao Zhang   ierr = VecLockReadPop(X);CHKERRQ(ierr);
282794ab13aaSBarry Smith   ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr);
282828d58a37SPierre Jolivet 
282928d58a37SPierre Jolivet   /* attach latest linearization point to the preconditioning matrix */
283028d58a37SPierre Jolivet   ierr = PetscObjectCompose((PetscObject)B,"__SNES_latest_X",(PetscObject)X);CHKERRQ(ierr);
2831a8054027SBarry Smith 
2832e0e3a89bSBarry Smith   /* the next line ensures that snes->ksp exists */
2833e0e3a89bSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
28343b4f5425SBarry Smith   if (snes->lagpreconditioner == -2) {
28353b4f5425SBarry Smith     ierr = PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n");CHKERRQ(ierr);
2836d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr);
28373b4f5425SBarry Smith     snes->lagpreconditioner = -1;
28383b4f5425SBarry Smith   } else if (snes->lagpreconditioner == -1) {
2839a8054027SBarry Smith     ierr = PetscInfo(snes,"Reusing preconditioner because lag is -1\n");CHKERRQ(ierr);
2840d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr);
284137ec4e1aSPeter Brune   } else if (snes->lagpreconditioner > 1 && (snes->iter + snes->pre_iter) % snes->lagpreconditioner) {
2842a8054027SBarry Smith     ierr = PetscInfo2(snes,"Reusing preconditioner because lag is %D and SNES iteration is %D\n",snes->lagpreconditioner,snes->iter);CHKERRQ(ierr);
2843d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr);
2844d1e9a80fSBarry Smith   } else {
2845d1e9a80fSBarry Smith     ierr = PetscInfo(snes,"Rebuilding preconditioner\n");CHKERRQ(ierr);
2846d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr);
2847a8054027SBarry Smith   }
2848a8054027SBarry Smith 
2849e885f1abSBarry Smith   ierr = SNESTestJacobian(snes);CHKERRQ(ierr);
28506d84be18SBarry Smith   /* make sure user returned a correct Jacobian and preconditioner */
285194ab13aaSBarry Smith   /* PetscValidHeaderSpecific(A,MAT_CLASSID,3);
285294ab13aaSBarry Smith     PetscValidHeaderSpecific(B,MAT_CLASSID,4);   */
2853693365a8SJed Brown   {
2854693365a8SJed Brown     PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE;
285516413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit",NULL,NULL,&flag);CHKERRQ(ierr);
285616413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr);
285716413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr);
285816413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_operator",NULL,NULL,&flag_operator);CHKERRQ(ierr);
2859693365a8SJed Brown     if (flag || flag_draw || flag_contour) {
28600298fd71SBarry Smith       Mat          Bexp_mine = NULL,Bexp,FDexp;
2861693365a8SJed Brown       PetscViewer  vdraw,vstdout;
28626b3a5b13SJed Brown       PetscBool    flg;
2863693365a8SJed Brown       if (flag_operator) {
28640bacdadaSStefano Zampini         ierr = MatComputeOperator(A,MATAIJ,&Bexp_mine);CHKERRQ(ierr);
2865693365a8SJed Brown         Bexp = Bexp_mine;
2866693365a8SJed Brown       } else {
2867693365a8SJed Brown         /* See if the preconditioning matrix can be viewed and added directly */
2868b9e7e5c1SBarry Smith         ierr = PetscObjectBaseTypeCompareAny((PetscObject)B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr);
286994ab13aaSBarry Smith         if (flg) Bexp = B;
2870693365a8SJed Brown         else {
2871693365a8SJed Brown           /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */
28720bacdadaSStefano Zampini           ierr = MatComputeOperator(B,MATAIJ,&Bexp_mine);CHKERRQ(ierr);
2873693365a8SJed Brown           Bexp = Bexp_mine;
2874693365a8SJed Brown         }
2875693365a8SJed Brown       }
2876693365a8SJed Brown       ierr = MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp);CHKERRQ(ierr);
2877d1e9a80fSBarry Smith       ierr = SNESComputeJacobianDefault(snes,X,FDexp,FDexp,NULL);CHKERRQ(ierr);
2878ce94432eSBarry Smith       ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr);
2879693365a8SJed Brown       if (flag_draw || flag_contour) {
28809e5d0892SLisandro Dalcin         ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
2881693365a8SJed Brown         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
28820298fd71SBarry Smith       } else vdraw = NULL;
2883693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator ? "Jacobian" : "preconditioning Jacobian");CHKERRQ(ierr);
2884693365a8SJed Brown       if (flag) {ierr = MatView(Bexp,vstdout);CHKERRQ(ierr);}
2885693365a8SJed Brown       if (vdraw) {ierr = MatView(Bexp,vdraw);CHKERRQ(ierr);}
2886693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n");CHKERRQ(ierr);
2887693365a8SJed Brown       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
2888693365a8SJed Brown       if (vdraw) {ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);}
2889693365a8SJed Brown       ierr = MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
2890693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n");CHKERRQ(ierr);
2891693365a8SJed Brown       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
2892693365a8SJed Brown       if (vdraw) {              /* Always use contour for the difference */
2893693365a8SJed Brown         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
2894693365a8SJed Brown         ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);
2895693365a8SJed Brown         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
2896693365a8SJed Brown       }
2897693365a8SJed Brown       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
2898693365a8SJed Brown       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
2899693365a8SJed Brown       ierr = MatDestroy(&Bexp_mine);CHKERRQ(ierr);
2900693365a8SJed Brown       ierr = MatDestroy(&FDexp);CHKERRQ(ierr);
2901693365a8SJed Brown     }
2902693365a8SJed Brown   }
29034c30e9fbSJed Brown   {
29046719d8e4SJed Brown     PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE;
29056719d8e4SJed Brown     PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON;
290616413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring",NULL,NULL,&flag);CHKERRQ(ierr);
290716413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_display",NULL,NULL,&flag_display);CHKERRQ(ierr);
290816413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr);
290916413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr);
291016413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",NULL,NULL,&flag_threshold);CHKERRQ(ierr);
291127b0f280SBarry Smith     if (flag_threshold) {
2912c5929fdfSBarry Smith       ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,NULL);CHKERRQ(ierr);
2913c5929fdfSBarry Smith       ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,NULL);CHKERRQ(ierr);
291427b0f280SBarry Smith     }
29156719d8e4SJed Brown     if (flag || flag_display || flag_draw || flag_contour || flag_threshold) {
29164c30e9fbSJed Brown       Mat            Bfd;
29174c30e9fbSJed Brown       PetscViewer    vdraw,vstdout;
2918335efc43SPeter Brune       MatColoring    coloring;
29194c30e9fbSJed Brown       ISColoring     iscoloring;
29204c30e9fbSJed Brown       MatFDColoring  matfdcoloring;
29214c30e9fbSJed Brown       PetscErrorCode (*func)(SNES,Vec,Vec,void*);
29224c30e9fbSJed Brown       void           *funcctx;
29236719d8e4SJed Brown       PetscReal      norm1,norm2,normmax;
29244c30e9fbSJed Brown 
292594ab13aaSBarry Smith       ierr = MatDuplicate(B,MAT_DO_NOT_COPY_VALUES,&Bfd);CHKERRQ(ierr);
2926335efc43SPeter Brune       ierr = MatColoringCreate(Bfd,&coloring);CHKERRQ(ierr);
2927335efc43SPeter Brune       ierr = MatColoringSetType(coloring,MATCOLORINGSL);CHKERRQ(ierr);
2928335efc43SPeter Brune       ierr = MatColoringSetFromOptions(coloring);CHKERRQ(ierr);
2929335efc43SPeter Brune       ierr = MatColoringApply(coloring,&iscoloring);CHKERRQ(ierr);
2930335efc43SPeter Brune       ierr = MatColoringDestroy(&coloring);CHKERRQ(ierr);
29314c30e9fbSJed Brown       ierr = MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring);CHKERRQ(ierr);
2932f86b9fbaSHong Zhang       ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr);
2933f86b9fbaSHong Zhang       ierr = MatFDColoringSetUp(Bfd,iscoloring,matfdcoloring);CHKERRQ(ierr);
29344c30e9fbSJed Brown       ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr);
29354c30e9fbSJed Brown 
29364c30e9fbSJed Brown       /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */
29370298fd71SBarry Smith       ierr = SNESGetFunction(snes,NULL,&func,&funcctx);CHKERRQ(ierr);
29384c30e9fbSJed Brown       ierr = MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode (*)(void))func,funcctx);CHKERRQ(ierr);
29394c30e9fbSJed Brown       ierr = PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix);CHKERRQ(ierr);
29404c30e9fbSJed Brown       ierr = PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_");CHKERRQ(ierr);
29414c30e9fbSJed Brown       ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr);
2942d1e9a80fSBarry Smith       ierr = MatFDColoringApply(Bfd,matfdcoloring,X,snes);CHKERRQ(ierr);
29434c30e9fbSJed Brown       ierr = MatFDColoringDestroy(&matfdcoloring);CHKERRQ(ierr);
29444c30e9fbSJed Brown 
2945ce94432eSBarry Smith       ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr);
29464c30e9fbSJed Brown       if (flag_draw || flag_contour) {
29479e5d0892SLisandro Dalcin         ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
29484c30e9fbSJed Brown         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
29490298fd71SBarry Smith       } else vdraw = NULL;
29504c30e9fbSJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n");CHKERRQ(ierr);
295194ab13aaSBarry Smith       if (flag_display) {ierr = MatView(B,vstdout);CHKERRQ(ierr);}
295294ab13aaSBarry Smith       if (vdraw) {ierr = MatView(B,vdraw);CHKERRQ(ierr);}
29534c30e9fbSJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n");CHKERRQ(ierr);
29546719d8e4SJed Brown       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
29554c30e9fbSJed Brown       if (vdraw) {ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);}
295694ab13aaSBarry Smith       ierr = MatAYPX(Bfd,-1.0,B,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
29574c30e9fbSJed Brown       ierr = MatNorm(Bfd,NORM_1,&norm1);CHKERRQ(ierr);
29586719d8e4SJed Brown       ierr = MatNorm(Bfd,NORM_FROBENIUS,&norm2);CHKERRQ(ierr);
29594c30e9fbSJed Brown       ierr = MatNorm(Bfd,NORM_MAX,&normmax);CHKERRQ(ierr);
296057622a8eSBarry 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);
29616719d8e4SJed Brown       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
29624c30e9fbSJed Brown       if (vdraw) {              /* Always use contour for the difference */
29634c30e9fbSJed Brown         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
29644c30e9fbSJed Brown         ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);
29654c30e9fbSJed Brown         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
29664c30e9fbSJed Brown       }
29674c30e9fbSJed Brown       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
29686719d8e4SJed Brown 
29696719d8e4SJed Brown       if (flag_threshold) {
29706719d8e4SJed Brown         PetscInt bs,rstart,rend,i;
297194ab13aaSBarry Smith         ierr = MatGetBlockSize(B,&bs);CHKERRQ(ierr);
297294ab13aaSBarry Smith         ierr = MatGetOwnershipRange(B,&rstart,&rend);CHKERRQ(ierr);
29736719d8e4SJed Brown         for (i=rstart; i<rend; i++) {
29746719d8e4SJed Brown           const PetscScalar *ba,*ca;
29756719d8e4SJed Brown           const PetscInt    *bj,*cj;
29766719d8e4SJed Brown           PetscInt          bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1;
29776719d8e4SJed Brown           PetscReal         maxentry = 0,maxdiff = 0,maxrdiff = 0;
297894ab13aaSBarry Smith           ierr = MatGetRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr);
29796719d8e4SJed Brown           ierr = MatGetRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
298094ab13aaSBarry Smith           if (bn != cn) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold");
29816719d8e4SJed Brown           for (j=0; j<bn; j++) {
29826719d8e4SJed Brown             PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
29836719d8e4SJed Brown             if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) {
29846719d8e4SJed Brown               maxentrycol = bj[j];
29856719d8e4SJed Brown               maxentry    = PetscRealPart(ba[j]);
29866719d8e4SJed Brown             }
29876719d8e4SJed Brown             if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) {
29886719d8e4SJed Brown               maxdiffcol = bj[j];
29896719d8e4SJed Brown               maxdiff    = PetscRealPart(ca[j]);
29906719d8e4SJed Brown             }
29916719d8e4SJed Brown             if (rdiff > maxrdiff) {
29926719d8e4SJed Brown               maxrdiffcol = bj[j];
29936719d8e4SJed Brown               maxrdiff    = rdiff;
29946719d8e4SJed Brown             }
29956719d8e4SJed Brown           }
29966719d8e4SJed Brown           if (maxrdiff > 1) {
299757622a8eSBarry 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);
29986719d8e4SJed Brown             for (j=0; j<bn; j++) {
29996719d8e4SJed Brown               PetscReal rdiff;
30006719d8e4SJed Brown               rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
30016719d8e4SJed Brown               if (rdiff > 1) {
300257622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(vstdout," (%D,%g:%g)",bj[j],(double)PetscRealPart(ba[j]),(double)PetscRealPart(ca[j]));CHKERRQ(ierr);
30036719d8e4SJed Brown               }
30046719d8e4SJed Brown             }
30056719d8e4SJed Brown             ierr = PetscViewerASCIIPrintf(vstdout,"\n",i,maxentry,maxdiff,maxrdiff);CHKERRQ(ierr);
30066719d8e4SJed Brown           }
300794ab13aaSBarry Smith           ierr = MatRestoreRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr);
30086719d8e4SJed Brown           ierr = MatRestoreRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
30096719d8e4SJed Brown         }
30106719d8e4SJed Brown       }
30114c30e9fbSJed Brown       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
30124c30e9fbSJed Brown       ierr = MatDestroy(&Bfd);CHKERRQ(ierr);
30134c30e9fbSJed Brown     }
30144c30e9fbSJed Brown   }
30153a40ed3dSBarry Smith   PetscFunctionReturn(0);
30169b94acceSBarry Smith }
30179b94acceSBarry Smith 
3018bf388a1fSBarry Smith /*MC
3019411c0326SBarry Smith     SNESJacobianFunction - Function used to convey the nonlinear Jacobian of the function to be solved by SNES
3020bf388a1fSBarry Smith 
3021bf388a1fSBarry Smith      Synopsis:
3022411c0326SBarry Smith      #include "petscsnes.h"
3023411c0326SBarry Smith      PetscErrorCode SNESJacobianFunction(SNES snes,Vec x,Mat Amat,Mat Pmat,void *ctx);
3024bf388a1fSBarry Smith 
30251843f636SBarry Smith      Collective on snes
30261843f636SBarry Smith 
30271843f636SBarry Smith     Input Parameters:
30281843f636SBarry Smith +  x - input vector, the Jacobian is to be computed at this value
3029bf388a1fSBarry Smith -  ctx - [optional] user-defined Jacobian context
3030bf388a1fSBarry Smith 
30311843f636SBarry Smith     Output Parameters:
30321843f636SBarry Smith +  Amat - the matrix that defines the (approximate) Jacobian
30331843f636SBarry Smith -  Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat.
30341843f636SBarry Smith 
3035878cb397SSatish Balay    Level: intermediate
3036878cb397SSatish Balay 
3037bf388a1fSBarry Smith .seealso:   SNESSetFunction(), SNESGetFunction(), SNESSetJacobian(), SNESGetJacobian()
3038bf388a1fSBarry Smith M*/
3039bf388a1fSBarry Smith 
30409b94acceSBarry Smith /*@C
30419b94acceSBarry Smith    SNESSetJacobian - Sets the function to compute Jacobian as well as the
3042044dda88SLois Curfman McInnes    location to store the matrix.
30439b94acceSBarry Smith 
3044d083f849SBarry Smith    Logically Collective on SNES
3045c7afd0dbSLois Curfman McInnes 
30469b94acceSBarry Smith    Input Parameters:
3047c7afd0dbSLois Curfman McInnes +  snes - the SNES context
3048e5d3d808SBarry Smith .  Amat - the matrix that defines the (approximate) Jacobian
3049e5d3d808SBarry Smith .  Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat.
3050411c0326SBarry Smith .  J - Jacobian evaluation routine (if NULL then SNES retains any previously set value), see SNESJacobianFunction for details
3051c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
30520298fd71SBarry Smith          Jacobian evaluation routine (may be NULL) (if NULL then SNES retains any previously set value)
30539b94acceSBarry Smith 
30549b94acceSBarry Smith    Notes:
3055e5d3d808SBarry Smith    If the Amat matrix and Pmat matrix are different you must call MatAssemblyBegin/End() on
305616913363SBarry Smith    each matrix.
305716913363SBarry Smith 
3058895c21f2SBarry Smith    If you know the operator Amat has a null space you can use MatSetNullSpace() and MatSetTransposeNullSpace() to supply the null
3059895c21f2SBarry Smith    space to Amat and the KSP solvers will automatically use that null space as needed during the solution process.
3060895c21f2SBarry Smith 
30618d359177SBarry Smith    If using SNESComputeJacobianDefaultColor() to assemble a Jacobian, the ctx argument
3062a8a26c1eSJed Brown    must be a MatFDColoring.
3063a8a26c1eSJed Brown 
3064c3cc8fd1SJed Brown    Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian.  One common
3065c3cc8fd1SJed Brown    example is to use the "Picard linearization" which only differentiates through the highest order parts of each term.
3066c3cc8fd1SJed Brown 
306736851e7fSLois Curfman McInnes    Level: beginner
306836851e7fSLois Curfman McInnes 
3069411c0326SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESComputeJacobianDefaultColor(), MatStructure, J,
3070411c0326SBarry Smith           SNESSetPicard(), SNESJacobianFunction
30719b94acceSBarry Smith @*/
3072d1e9a80fSBarry Smith PetscErrorCode  SNESSetJacobian(SNES snes,Mat Amat,Mat Pmat,PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx)
30739b94acceSBarry Smith {
3074dfbe8321SBarry Smith   PetscErrorCode ierr;
30756cab3a1bSJed Brown   DM             dm;
30763a7fca6bSBarry Smith 
30773a40ed3dSBarry Smith   PetscFunctionBegin;
30780700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3079e5d3d808SBarry Smith   if (Amat) PetscValidHeaderSpecific(Amat,MAT_CLASSID,2);
3080e5d3d808SBarry Smith   if (Pmat) PetscValidHeaderSpecific(Pmat,MAT_CLASSID,3);
3081e5d3d808SBarry Smith   if (Amat) PetscCheckSameComm(snes,1,Amat,2);
3082e5d3d808SBarry Smith   if (Pmat) PetscCheckSameComm(snes,1,Pmat,3);
30836cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
3084f8b49ee9SBarry Smith   ierr = DMSNESSetJacobian(dm,J,ctx);CHKERRQ(ierr);
3085e5d3d808SBarry Smith   if (Amat) {
3086e5d3d808SBarry Smith     ierr = PetscObjectReference((PetscObject)Amat);CHKERRQ(ierr);
30876bf464f9SBarry Smith     ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
3088f5af7f23SKarl Rupp 
3089e5d3d808SBarry Smith     snes->jacobian = Amat;
30903a7fca6bSBarry Smith   }
3091e5d3d808SBarry Smith   if (Pmat) {
3092e5d3d808SBarry Smith     ierr = PetscObjectReference((PetscObject)Pmat);CHKERRQ(ierr);
30936bf464f9SBarry Smith     ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
3094f5af7f23SKarl Rupp 
3095e5d3d808SBarry Smith     snes->jacobian_pre = Pmat;
30963a7fca6bSBarry Smith   }
30973a40ed3dSBarry Smith   PetscFunctionReturn(0);
30989b94acceSBarry Smith }
309962fef451SLois Curfman McInnes 
3100c2aafc4cSSatish Balay /*@C
3101b4fd4287SBarry Smith    SNESGetJacobian - Returns the Jacobian matrix and optionally the user
3102b4fd4287SBarry Smith    provided context for evaluating the Jacobian.
3103b4fd4287SBarry Smith 
3104c7afd0dbSLois Curfman McInnes    Not Collective, but Mat object will be parallel if SNES object is
3105c7afd0dbSLois Curfman McInnes 
3106b4fd4287SBarry Smith    Input Parameter:
3107b4fd4287SBarry Smith .  snes - the nonlinear solver context
3108b4fd4287SBarry Smith 
3109b4fd4287SBarry Smith    Output Parameters:
3110e5d3d808SBarry Smith +  Amat - location to stash (approximate) Jacobian matrix (or NULL)
3111e5d3d808SBarry Smith .  Pmat - location to stash matrix used to compute the preconditioner (or NULL)
3112411c0326SBarry Smith .  J - location to put Jacobian function (or NULL), see SNESJacobianFunction for details on its calling sequence
31130298fd71SBarry Smith -  ctx - location to stash Jacobian ctx (or NULL)
3114fee21e36SBarry Smith 
311536851e7fSLois Curfman McInnes    Level: advanced
311636851e7fSLois Curfman McInnes 
3117411c0326SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian(), SNESJacobianFunction, SNESGetFunction()
3118b4fd4287SBarry Smith @*/
3119d1e9a80fSBarry Smith PetscErrorCode SNESGetJacobian(SNES snes,Mat *Amat,Mat *Pmat,PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx)
3120b4fd4287SBarry Smith {
31216cab3a1bSJed Brown   PetscErrorCode ierr;
31226cab3a1bSJed Brown   DM             dm;
3123942e3340SBarry Smith   DMSNES         sdm;
31246cab3a1bSJed Brown 
31253a40ed3dSBarry Smith   PetscFunctionBegin;
31260700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3127e5d3d808SBarry Smith   if (Amat) *Amat = snes->jacobian;
3128e5d3d808SBarry Smith   if (Pmat) *Pmat = snes->jacobian_pre;
31296cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
3130942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
3131f8b49ee9SBarry Smith   if (J) *J = sdm->ops->computejacobian;
31326cab3a1bSJed Brown   if (ctx) *ctx = sdm->jacobianctx;
31333a40ed3dSBarry Smith   PetscFunctionReturn(0);
3134b4fd4287SBarry Smith }
3135b4fd4287SBarry Smith 
313658b371f3SBarry Smith static PetscErrorCode SNESSetDefaultComputeJacobian(SNES snes)
313758b371f3SBarry Smith {
313858b371f3SBarry Smith   PetscErrorCode ierr;
313958b371f3SBarry Smith   DM             dm;
314058b371f3SBarry Smith   DMSNES         sdm;
314158b371f3SBarry Smith 
314258b371f3SBarry Smith   PetscFunctionBegin;
314358b371f3SBarry Smith   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
314458b371f3SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
314558b371f3SBarry Smith   if (!sdm->ops->computejacobian && snes->jacobian_pre) {
314658b371f3SBarry Smith     DM        dm;
314758b371f3SBarry Smith     PetscBool isdense,ismf;
314858b371f3SBarry Smith 
314958b371f3SBarry Smith     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
315058b371f3SBarry Smith     ierr = PetscObjectTypeCompareAny((PetscObject)snes->jacobian_pre,&isdense,MATSEQDENSE,MATMPIDENSE,MATDENSE,NULL);CHKERRQ(ierr);
315158b371f3SBarry Smith     ierr = PetscObjectTypeCompareAny((PetscObject)snes->jacobian_pre,&ismf,MATMFFD,MATSHELL,NULL);CHKERRQ(ierr);
315258b371f3SBarry Smith     if (isdense) {
315358b371f3SBarry Smith       ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefault,NULL);CHKERRQ(ierr);
315458b371f3SBarry Smith     } else if (!ismf) {
315558b371f3SBarry Smith       ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr);
315658b371f3SBarry Smith     }
315758b371f3SBarry Smith   }
315858b371f3SBarry Smith   PetscFunctionReturn(0);
315958b371f3SBarry Smith }
316058b371f3SBarry Smith 
31619b94acceSBarry Smith /*@
31629b94acceSBarry Smith    SNESSetUp - Sets up the internal data structures for the later use
3163272ac6f2SLois Curfman McInnes    of a nonlinear solver.
31649b94acceSBarry Smith 
3165fee21e36SBarry Smith    Collective on SNES
3166fee21e36SBarry Smith 
3167c7afd0dbSLois Curfman McInnes    Input Parameters:
316870e92668SMatthew Knepley .  snes - the SNES context
3169c7afd0dbSLois Curfman McInnes 
3170272ac6f2SLois Curfman McInnes    Notes:
3171272ac6f2SLois Curfman McInnes    For basic use of the SNES solvers the user need not explicitly call
3172272ac6f2SLois Curfman McInnes    SNESSetUp(), since these actions will automatically occur during
3173272ac6f2SLois Curfman McInnes    the call to SNESSolve().  However, if one wishes to control this
3174272ac6f2SLois Curfman McInnes    phase separately, SNESSetUp() should be called after SNESCreate()
3175272ac6f2SLois Curfman McInnes    and optional routines of the form SNESSetXXX(), but before SNESSolve().
3176272ac6f2SLois Curfman McInnes 
317736851e7fSLois Curfman McInnes    Level: advanced
317836851e7fSLois Curfman McInnes 
31799b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy()
31809b94acceSBarry Smith @*/
31817087cfbeSBarry Smith PetscErrorCode  SNESSetUp(SNES snes)
31829b94acceSBarry Smith {
3183dfbe8321SBarry Smith   PetscErrorCode ierr;
31846cab3a1bSJed Brown   DM             dm;
3185942e3340SBarry Smith   DMSNES         sdm;
3186c35f09e5SBarry Smith   SNESLineSearch linesearch, pclinesearch;
31876e2a1849SPeter Brune   void           *lsprectx,*lspostctx;
31886b2b7091SBarry Smith   PetscErrorCode (*precheck)(SNESLineSearch,Vec,Vec,PetscBool*,void*);
31896b2b7091SBarry Smith   PetscErrorCode (*postcheck)(SNESLineSearch,Vec,Vec,Vec,PetscBool*,PetscBool*,void*);
31906e2a1849SPeter Brune   PetscErrorCode (*func)(SNES,Vec,Vec,void*);
31916e2a1849SPeter Brune   Vec            f,fpc;
31926e2a1849SPeter Brune   void           *funcctx;
3193d1e9a80fSBarry Smith   PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*);
31941eb13d49SPeter Brune   void           *jacctx,*appctx;
319532b97717SPeter Brune   Mat            j,jpre;
31963a40ed3dSBarry Smith 
31973a40ed3dSBarry Smith   PetscFunctionBegin;
31980700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
31994dc4c822SBarry Smith   if (snes->setupcalled) PetscFunctionReturn(0);
3200e3ed9ee7SBarry Smith   ierr = PetscLogEventBegin(SNES_Setup,snes,0,0,0);CHKERRQ(ierr);
32019b94acceSBarry Smith 
32027adad957SLisandro Dalcin   if (!((PetscObject)snes)->type_name) {
320304d7464bSBarry Smith     ierr = SNESSetType(snes,SNESNEWTONLS);CHKERRQ(ierr);
320485385478SLisandro Dalcin   }
320585385478SLisandro Dalcin 
32060298fd71SBarry Smith   ierr = SNESGetFunction(snes,&snes->vec_func,NULL,NULL);CHKERRQ(ierr);
320758c9b817SLisandro Dalcin 
32086cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
3209942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
3210ce94432eSBarry Smith   if (!sdm->ops->computefunction) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE,"Function never provided to SNES object");
321158b371f3SBarry Smith   ierr = SNESSetDefaultComputeJacobian(snes);CHKERRQ(ierr);
321258b371f3SBarry Smith 
32136cab3a1bSJed Brown   if (!snes->vec_func) {
32146cab3a1bSJed Brown     ierr = DMCreateGlobalVector(dm,&snes->vec_func);CHKERRQ(ierr);
3215214df951SJed Brown   }
3216efd51863SBarry Smith 
321722d28d08SBarry Smith   if (!snes->ksp) {
321822d28d08SBarry Smith     ierr = SNESGetKSP(snes, &snes->ksp);CHKERRQ(ierr);
321922d28d08SBarry Smith   }
3220b710008aSBarry Smith 
3221d8d34be6SBarry Smith   if (snes->linesearch) {
32227601faf0SJed Brown     ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr);
3223ed07d7d7SPeter Brune     ierr = SNESLineSearchSetFunction(snes->linesearch,SNESComputeFunction);CHKERRQ(ierr);
3224d8d34be6SBarry Smith   }
32259e764e56SPeter Brune 
3226efd4aadfSBarry Smith   if (snes->npc && (snes->npcside== PC_LEFT)) {
3227172a4300SPeter Brune     snes->mf          = PETSC_TRUE;
3228172a4300SPeter Brune     snes->mf_operator = PETSC_FALSE;
3229172a4300SPeter Brune   }
3230d8f46077SPeter Brune 
3231efd4aadfSBarry Smith   if (snes->npc) {
32326e2a1849SPeter Brune     /* copy the DM over */
32336e2a1849SPeter Brune     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
3234efd4aadfSBarry Smith     ierr = SNESSetDM(snes->npc,dm);CHKERRQ(ierr);
32356e2a1849SPeter Brune 
32366e2a1849SPeter Brune     ierr = SNESGetFunction(snes,&f,&func,&funcctx);CHKERRQ(ierr);
32376e2a1849SPeter Brune     ierr = VecDuplicate(f,&fpc);CHKERRQ(ierr);
3238efd4aadfSBarry Smith     ierr = SNESSetFunction(snes->npc,fpc,func,funcctx);CHKERRQ(ierr);
323932b97717SPeter Brune     ierr = SNESGetJacobian(snes,&j,&jpre,&jac,&jacctx);CHKERRQ(ierr);
3240efd4aadfSBarry Smith     ierr = SNESSetJacobian(snes->npc,j,jpre,jac,jacctx);CHKERRQ(ierr);
32411eb13d49SPeter Brune     ierr = SNESGetApplicationContext(snes,&appctx);CHKERRQ(ierr);
3242efd4aadfSBarry Smith     ierr = SNESSetApplicationContext(snes->npc,appctx);CHKERRQ(ierr);
32436e2a1849SPeter Brune     ierr = VecDestroy(&fpc);CHKERRQ(ierr);
32446e2a1849SPeter Brune 
32456e2a1849SPeter Brune     /* copy the function pointers over */
3246efd4aadfSBarry Smith     ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr);
32476e2a1849SPeter Brune 
32486e2a1849SPeter Brune     /* default to 1 iteration */
3249efd4aadfSBarry Smith     ierr = SNESSetTolerances(snes->npc,0.0,0.0,0.0,1,snes->npc->max_funcs);CHKERRQ(ierr);
3250efd4aadfSBarry Smith     if (snes->npcside==PC_RIGHT) {
3251efd4aadfSBarry Smith       ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_FINAL_ONLY);CHKERRQ(ierr);
3252a9936a0cSPeter Brune     } else {
3253efd4aadfSBarry Smith       ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_NONE);CHKERRQ(ierr);
3254a9936a0cSPeter Brune     }
3255efd4aadfSBarry Smith     ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr);
32566e2a1849SPeter Brune 
32576e2a1849SPeter Brune     /* copy the line search context over */
3258d8d34be6SBarry Smith     if (snes->linesearch && snes->npc->linesearch) {
32597601faf0SJed Brown       ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr);
3260efd4aadfSBarry Smith       ierr = SNESGetLineSearch(snes->npc,&pclinesearch);CHKERRQ(ierr);
32616b2b7091SBarry Smith       ierr = SNESLineSearchGetPreCheck(linesearch,&precheck,&lsprectx);CHKERRQ(ierr);
32626b2b7091SBarry Smith       ierr = SNESLineSearchGetPostCheck(linesearch,&postcheck,&lspostctx);CHKERRQ(ierr);
32636b2b7091SBarry Smith       ierr = SNESLineSearchSetPreCheck(pclinesearch,precheck,lsprectx);CHKERRQ(ierr);
32646b2b7091SBarry Smith       ierr = SNESLineSearchSetPostCheck(pclinesearch,postcheck,lspostctx);CHKERRQ(ierr);
32656e2a1849SPeter Brune       ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)linesearch, (PetscObject)pclinesearch);CHKERRQ(ierr);
32666e2a1849SPeter Brune     }
3267d8d34be6SBarry Smith   }
326832b97717SPeter Brune   if (snes->mf) {
326932b97717SPeter Brune     ierr = SNESSetUpMatrixFree_Private(snes, snes->mf_operator, snes->mf_version);CHKERRQ(ierr);
327032b97717SPeter Brune   }
327132b97717SPeter Brune   if (snes->ops->usercompute && !snes->user) {
327232b97717SPeter Brune     ierr = (*snes->ops->usercompute)(snes,(void**)&snes->user);CHKERRQ(ierr);
327332b97717SPeter Brune   }
32746e2a1849SPeter Brune 
327537ec4e1aSPeter Brune   snes->jac_iter = 0;
327637ec4e1aSPeter Brune   snes->pre_iter = 0;
327737ec4e1aSPeter Brune 
3278410397dcSLisandro Dalcin   if (snes->ops->setup) {
3279410397dcSLisandro Dalcin     ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr);
3280410397dcSLisandro Dalcin   }
328158c9b817SLisandro Dalcin 
328258b371f3SBarry Smith   ierr = SNESSetDefaultComputeJacobian(snes);CHKERRQ(ierr);
328358b371f3SBarry Smith 
3284efd4aadfSBarry Smith   if (snes->npc && (snes->npcside== PC_LEFT)) {
32856c67d002SPeter Brune     if (snes->functype == SNES_FUNCTION_PRECONDITIONED) {
3286d8d34be6SBarry Smith       if (snes->linesearch){
328755d4788fSPeter Brune         ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr);
3288be95d8f1SBarry Smith         ierr = SNESLineSearchSetFunction(linesearch,SNESComputeFunctionDefaultNPC);CHKERRQ(ierr);
32896c67d002SPeter Brune       }
32906c67d002SPeter Brune     }
3291d8d34be6SBarry Smith   }
3292e3ed9ee7SBarry Smith   ierr = PetscLogEventEnd(SNES_Setup,snes,0,0,0);CHKERRQ(ierr);
32937aaed0d8SBarry Smith   snes->setupcalled = PETSC_TRUE;
32943a40ed3dSBarry Smith   PetscFunctionReturn(0);
32959b94acceSBarry Smith }
32969b94acceSBarry Smith 
329737596af1SLisandro Dalcin /*@
329837596af1SLisandro Dalcin    SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats
329937596af1SLisandro Dalcin 
330037596af1SLisandro Dalcin    Collective on SNES
330137596af1SLisandro Dalcin 
330237596af1SLisandro Dalcin    Input Parameter:
330337596af1SLisandro Dalcin .  snes - iterative context obtained from SNESCreate()
330437596af1SLisandro Dalcin 
3305d25893d9SBarry Smith    Level: intermediate
3306d25893d9SBarry Smith 
330795452b02SPatrick Sanan    Notes:
330895452b02SPatrick Sanan     Also calls the application context destroy routine set with SNESSetComputeApplicationContext()
330937596af1SLisandro Dalcin 
331037596af1SLisandro Dalcin .seealso: SNESCreate(), SNESSetUp(), SNESSolve()
331137596af1SLisandro Dalcin @*/
331237596af1SLisandro Dalcin PetscErrorCode  SNESReset(SNES snes)
331337596af1SLisandro Dalcin {
331437596af1SLisandro Dalcin   PetscErrorCode ierr;
331537596af1SLisandro Dalcin 
331637596af1SLisandro Dalcin   PetscFunctionBegin;
331737596af1SLisandro Dalcin   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3318d25893d9SBarry Smith   if (snes->ops->userdestroy && snes->user) {
3319d25893d9SBarry Smith     ierr       = (*snes->ops->userdestroy)((void**)&snes->user);CHKERRQ(ierr);
33200298fd71SBarry Smith     snes->user = NULL;
3321d25893d9SBarry Smith   }
3322efd4aadfSBarry Smith   if (snes->npc) {
3323efd4aadfSBarry Smith     ierr = SNESReset(snes->npc);CHKERRQ(ierr);
33248a23116dSBarry Smith   }
33258a23116dSBarry Smith 
332637596af1SLisandro Dalcin   if (snes->ops->reset) {
332737596af1SLisandro Dalcin     ierr = (*snes->ops->reset)(snes);CHKERRQ(ierr);
332837596af1SLisandro Dalcin   }
33299e764e56SPeter Brune   if (snes->ksp) {
33309e764e56SPeter Brune     ierr = KSPReset(snes->ksp);CHKERRQ(ierr);
33319e764e56SPeter Brune   }
33329e764e56SPeter Brune 
33339e764e56SPeter Brune   if (snes->linesearch) {
3334f1c6b773SPeter Brune     ierr = SNESLineSearchReset(snes->linesearch);CHKERRQ(ierr);
33359e764e56SPeter Brune   }
33369e764e56SPeter Brune 
33376bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
33386bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
33396bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_sol_update);CHKERRQ(ierr);
33406bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr);
33416bf464f9SBarry Smith   ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
33426bf464f9SBarry Smith   ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
3343bbc1464cSBarry Smith   ierr = MatDestroy(&snes->picard);CHKERRQ(ierr);
3344c41d97abSJed Brown   ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr);
3345c41d97abSJed Brown   ierr = VecDestroyVecs(snes->nvwork,&snes->vwork);CHKERRQ(ierr);
3346f5af7f23SKarl Rupp 
334740fdac6aSLawrence Mitchell   snes->alwayscomputesfinalresidual = PETSC_FALSE;
334840fdac6aSLawrence Mitchell 
334937596af1SLisandro Dalcin   snes->nwork       = snes->nvwork = 0;
335037596af1SLisandro Dalcin   snes->setupcalled = PETSC_FALSE;
335137596af1SLisandro Dalcin   PetscFunctionReturn(0);
335237596af1SLisandro Dalcin }
335337596af1SLisandro Dalcin 
335452baeb72SSatish Balay /*@
3355c4421ceaSFande Kong    SNESConvergedReasonViewCancel - Clears all the reasonview functions for a SNES object.
3356c4421ceaSFande Kong 
3357c4421ceaSFande Kong    Collective on SNES
3358c4421ceaSFande Kong 
3359c4421ceaSFande Kong    Input Parameter:
3360c4421ceaSFande Kong .  snes - iterative context obtained from SNESCreate()
3361c4421ceaSFande Kong 
3362c4421ceaSFande Kong    Level: intermediate
3363c4421ceaSFande Kong 
3364c4421ceaSFande Kong .seealso: SNESCreate(), SNESDestroy(), SNESReset()
3365c4421ceaSFande Kong @*/
3366c4421ceaSFande Kong PetscErrorCode  SNESConvergedReasonViewCancel(SNES snes)
3367c4421ceaSFande Kong {
3368c4421ceaSFande Kong   PetscErrorCode ierr;
3369c4421ceaSFande Kong   PetscInt       i;
3370c4421ceaSFande Kong 
3371c4421ceaSFande Kong   PetscFunctionBegin;
3372c4421ceaSFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3373c4421ceaSFande Kong   for (i=0; i<snes->numberreasonviews; i++) {
3374c4421ceaSFande Kong     if (snes->reasonviewdestroy[i]) {
3375c4421ceaSFande Kong       ierr = (*snes->reasonviewdestroy[i])(&snes->reasonviewcontext[i]);CHKERRQ(ierr);
3376c4421ceaSFande Kong     }
3377c4421ceaSFande Kong   }
3378c4421ceaSFande Kong   snes->numberreasonviews = 0;
3379c4421ceaSFande Kong   PetscFunctionReturn(0);
3380c4421ceaSFande Kong }
3381c4421ceaSFande Kong 
33821fb7b255SJunchao Zhang /*@C
33839b94acceSBarry Smith    SNESDestroy - Destroys the nonlinear solver context that was created
33849b94acceSBarry Smith    with SNESCreate().
33859b94acceSBarry Smith 
3386c7afd0dbSLois Curfman McInnes    Collective on SNES
3387c7afd0dbSLois Curfman McInnes 
33889b94acceSBarry Smith    Input Parameter:
33899b94acceSBarry Smith .  snes - the SNES context
33909b94acceSBarry Smith 
339136851e7fSLois Curfman McInnes    Level: beginner
339236851e7fSLois Curfman McInnes 
339363a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve()
33949b94acceSBarry Smith @*/
33956bf464f9SBarry Smith PetscErrorCode  SNESDestroy(SNES *snes)
33969b94acceSBarry Smith {
33976849ba73SBarry Smith   PetscErrorCode ierr;
33983a40ed3dSBarry Smith 
33993a40ed3dSBarry Smith   PetscFunctionBegin;
34006bf464f9SBarry Smith   if (!*snes) PetscFunctionReturn(0);
34016bf464f9SBarry Smith   PetscValidHeaderSpecific((*snes),SNES_CLASSID,1);
34029e5d0892SLisandro Dalcin   if (--((PetscObject)(*snes))->refct > 0) {*snes = NULL; PetscFunctionReturn(0);}
3403d4bb536fSBarry Smith 
34046bf464f9SBarry Smith   ierr = SNESReset((*snes));CHKERRQ(ierr);
3405efd4aadfSBarry Smith   ierr = SNESDestroy(&(*snes)->npc);CHKERRQ(ierr);
34066b8b9a38SLisandro Dalcin 
3407e04113cfSBarry Smith   /* if memory was published with SAWs then destroy it */
3408e04113cfSBarry Smith   ierr = PetscObjectSAWsViewOff((PetscObject)*snes);CHKERRQ(ierr);
34096bf464f9SBarry Smith   if ((*snes)->ops->destroy) {ierr = (*((*snes))->ops->destroy)((*snes));CHKERRQ(ierr);}
34106d4c513bSLisandro Dalcin 
341133124788SMatthew G. Knepley   if ((*snes)->dm) {ierr = DMCoarsenHookRemove((*snes)->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,*snes);CHKERRQ(ierr);}
34126bf464f9SBarry Smith   ierr = DMDestroy(&(*snes)->dm);CHKERRQ(ierr);
34136bf464f9SBarry Smith   ierr = KSPDestroy(&(*snes)->ksp);CHKERRQ(ierr);
3414f1c6b773SPeter Brune   ierr = SNESLineSearchDestroy(&(*snes)->linesearch);CHKERRQ(ierr);
34156b8b9a38SLisandro Dalcin 
34166bf464f9SBarry Smith   ierr = PetscFree((*snes)->kspconvctx);CHKERRQ(ierr);
34176bf464f9SBarry Smith   if ((*snes)->ops->convergeddestroy) {
34186bf464f9SBarry Smith     ierr = (*(*snes)->ops->convergeddestroy)((*snes)->cnvP);CHKERRQ(ierr);
34196b8b9a38SLisandro Dalcin   }
3420071fcb05SBarry Smith   if ((*snes)->conv_hist_alloc) {
3421071fcb05SBarry Smith     ierr = PetscFree2((*snes)->conv_hist,(*snes)->conv_hist_its);CHKERRQ(ierr);
342258c9b817SLisandro Dalcin   }
34236bf464f9SBarry Smith   ierr = SNESMonitorCancel((*snes));CHKERRQ(ierr);
3424c4421ceaSFande Kong   ierr = SNESConvergedReasonViewCancel((*snes));CHKERRQ(ierr);
3425a79aaaedSSatish Balay   ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr);
34263a40ed3dSBarry Smith   PetscFunctionReturn(0);
34279b94acceSBarry Smith }
34289b94acceSBarry Smith 
34299b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */
34309b94acceSBarry Smith 
3431a8054027SBarry Smith /*@
3432a8054027SBarry Smith    SNESSetLagPreconditioner - Determines when the preconditioner is rebuilt in the nonlinear solve.
3433a8054027SBarry Smith 
34343f9fe445SBarry Smith    Logically Collective on SNES
3435a8054027SBarry Smith 
3436a8054027SBarry Smith    Input Parameters:
3437a8054027SBarry Smith +  snes - the SNES context
3438d8e291bfSBarry Smith -  lag - 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time
34393b4f5425SBarry Smith          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
3440a8054027SBarry Smith 
3441a8054027SBarry Smith    Options Database Keys:
34423d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
34433d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
34443d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
34453d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag
3446a8054027SBarry Smith 
3447a8054027SBarry Smith    Notes:
3448a8054027SBarry Smith    The default is 1
34493d5a8a6aSBarry Smith    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 or SNESSetLagPreconditionerPersists() was called
3450d8e291bfSBarry Smith 
3451d8e291bfSBarry Smith    SNESSetLagPreconditionerPersists() allows using the same uniform lagging (for example every second solve) across multiple solves.
3452a8054027SBarry Smith 
3453a8054027SBarry Smith    Level: intermediate
3454a8054027SBarry Smith 
34553d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetLagPreconditionerPersists(),
34563d5a8a6aSBarry Smith           SNESSetLagJacobianPersists()
3457a8054027SBarry Smith 
3458a8054027SBarry Smith @*/
34597087cfbeSBarry Smith PetscErrorCode  SNESSetLagPreconditioner(SNES snes,PetscInt lag)
3460a8054027SBarry Smith {
3461a8054027SBarry Smith   PetscFunctionBegin;
34620700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3463e32f2f54SBarry Smith   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
3464e32f2f54SBarry Smith   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
3465c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,lag,2);
3466a8054027SBarry Smith   snes->lagpreconditioner = lag;
3467a8054027SBarry Smith   PetscFunctionReturn(0);
3468a8054027SBarry Smith }
3469a8054027SBarry Smith 
3470efd51863SBarry Smith /*@
3471efd51863SBarry Smith    SNESSetGridSequence - sets the number of steps of grid sequencing that SNES does
3472efd51863SBarry Smith 
3473efd51863SBarry Smith    Logically Collective on SNES
3474efd51863SBarry Smith 
3475efd51863SBarry Smith    Input Parameters:
3476efd51863SBarry Smith +  snes - the SNES context
3477efd51863SBarry Smith -  steps - the number of refinements to do, defaults to 0
3478efd51863SBarry Smith 
3479efd51863SBarry Smith    Options Database Keys:
3480efd51863SBarry Smith .    -snes_grid_sequence <steps>
3481efd51863SBarry Smith 
3482efd51863SBarry Smith    Level: intermediate
3483efd51863SBarry Smith 
3484c0df2a02SJed Brown    Notes:
3485c0df2a02SJed Brown    Use SNESGetSolution() to extract the fine grid solution after grid sequencing.
3486c0df2a02SJed Brown 
3487fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetGridSequence()
3488efd51863SBarry Smith 
3489efd51863SBarry Smith @*/
3490efd51863SBarry Smith PetscErrorCode  SNESSetGridSequence(SNES snes,PetscInt steps)
3491efd51863SBarry Smith {
3492efd51863SBarry Smith   PetscFunctionBegin;
3493efd51863SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3494efd51863SBarry Smith   PetscValidLogicalCollectiveInt(snes,steps,2);
3495efd51863SBarry Smith   snes->gridsequence = steps;
3496efd51863SBarry Smith   PetscFunctionReturn(0);
3497efd51863SBarry Smith }
3498efd51863SBarry Smith 
3499fa19ca70SBarry Smith /*@
3500fa19ca70SBarry Smith    SNESGetGridSequence - gets the number of steps of grid sequencing that SNES does
3501fa19ca70SBarry Smith 
3502fa19ca70SBarry Smith    Logically Collective on SNES
3503fa19ca70SBarry Smith 
3504fa19ca70SBarry Smith    Input Parameter:
3505fa19ca70SBarry Smith .  snes - the SNES context
3506fa19ca70SBarry Smith 
3507fa19ca70SBarry Smith    Output Parameter:
3508fa19ca70SBarry Smith .  steps - the number of refinements to do, defaults to 0
3509fa19ca70SBarry Smith 
3510fa19ca70SBarry Smith    Options Database Keys:
3511fa19ca70SBarry Smith .    -snes_grid_sequence <steps>
3512fa19ca70SBarry Smith 
3513fa19ca70SBarry Smith    Level: intermediate
3514fa19ca70SBarry Smith 
3515fa19ca70SBarry Smith    Notes:
3516fa19ca70SBarry Smith    Use SNESGetSolution() to extract the fine grid solution after grid sequencing.
3517fa19ca70SBarry Smith 
3518fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetGridSequence()
3519fa19ca70SBarry Smith 
3520fa19ca70SBarry Smith @*/
3521fa19ca70SBarry Smith PetscErrorCode  SNESGetGridSequence(SNES snes,PetscInt *steps)
3522fa19ca70SBarry Smith {
3523fa19ca70SBarry Smith   PetscFunctionBegin;
3524fa19ca70SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3525fa19ca70SBarry Smith   *steps = snes->gridsequence;
3526fa19ca70SBarry Smith   PetscFunctionReturn(0);
3527fa19ca70SBarry Smith }
3528fa19ca70SBarry Smith 
3529a8054027SBarry Smith /*@
3530a8054027SBarry Smith    SNESGetLagPreconditioner - Indicates how often the preconditioner is rebuilt
3531a8054027SBarry Smith 
35323f9fe445SBarry Smith    Not Collective
3533a8054027SBarry Smith 
3534a8054027SBarry Smith    Input Parameter:
3535a8054027SBarry Smith .  snes - the SNES context
3536a8054027SBarry Smith 
3537a8054027SBarry Smith    Output Parameter:
3538a8054027SBarry 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
35393b4f5425SBarry Smith          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
3540a8054027SBarry Smith 
3541a8054027SBarry Smith    Options Database Keys:
35423d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
35433d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
35443d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
35453d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag
3546a8054027SBarry Smith 
3547a8054027SBarry Smith    Notes:
3548a8054027SBarry Smith    The default is 1
3549a8054027SBarry Smith    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3550a8054027SBarry Smith 
3551a8054027SBarry Smith    Level: intermediate
3552a8054027SBarry Smith 
35533d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagPreconditioner(), SNESSetLagJacobianPersists(), SNESSetLagPreconditionerPersists()
3554a8054027SBarry Smith 
3555a8054027SBarry Smith @*/
35567087cfbeSBarry Smith PetscErrorCode  SNESGetLagPreconditioner(SNES snes,PetscInt *lag)
3557a8054027SBarry Smith {
3558a8054027SBarry Smith   PetscFunctionBegin;
35590700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3560a8054027SBarry Smith   *lag = snes->lagpreconditioner;
3561a8054027SBarry Smith   PetscFunctionReturn(0);
3562a8054027SBarry Smith }
3563a8054027SBarry Smith 
3564e35cf81dSBarry Smith /*@
3565e35cf81dSBarry Smith    SNESSetLagJacobian - Determines when the Jacobian is rebuilt in the nonlinear solve. See SNESSetLagPreconditioner() for determining how
3566e35cf81dSBarry Smith      often the preconditioner is rebuilt.
3567e35cf81dSBarry Smith 
35683f9fe445SBarry Smith    Logically Collective on SNES
3569e35cf81dSBarry Smith 
3570e35cf81dSBarry Smith    Input Parameters:
3571e35cf81dSBarry Smith +  snes - the SNES context
3572e35cf81dSBarry 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
3573fe3ffe1eSBarry Smith          the Jacobian is built etc. -2 means rebuild at next chance but then never again
3574e35cf81dSBarry Smith 
3575e35cf81dSBarry Smith    Options Database Keys:
35763d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
35773d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
35783d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
35793d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag.
3580e35cf81dSBarry Smith 
3581e35cf81dSBarry Smith    Notes:
3582e35cf81dSBarry Smith    The default is 1
3583e35cf81dSBarry Smith    The Jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3584fe3ffe1eSBarry 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
3585fe3ffe1eSBarry Smith    at the next Newton step but never again (unless it is reset to another value)
3586e35cf81dSBarry Smith 
3587e35cf81dSBarry Smith    Level: intermediate
3588e35cf81dSBarry Smith 
35893d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagPreconditioner(), SNESGetLagJacobianPersists(), SNESSetLagPreconditionerPersists()
3590e35cf81dSBarry Smith 
3591e35cf81dSBarry Smith @*/
35927087cfbeSBarry Smith PetscErrorCode  SNESSetLagJacobian(SNES snes,PetscInt lag)
3593e35cf81dSBarry Smith {
3594e35cf81dSBarry Smith   PetscFunctionBegin;
35950700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3596e32f2f54SBarry Smith   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
3597e32f2f54SBarry Smith   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
3598c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,lag,2);
3599e35cf81dSBarry Smith   snes->lagjacobian = lag;
3600e35cf81dSBarry Smith   PetscFunctionReturn(0);
3601e35cf81dSBarry Smith }
3602e35cf81dSBarry Smith 
3603e35cf81dSBarry Smith /*@
3604e35cf81dSBarry Smith    SNESGetLagJacobian - Indicates how often the Jacobian is rebuilt. See SNESGetLagPreconditioner() to determine when the preconditioner is rebuilt
3605e35cf81dSBarry Smith 
36063f9fe445SBarry Smith    Not Collective
3607e35cf81dSBarry Smith 
3608e35cf81dSBarry Smith    Input Parameter:
3609e35cf81dSBarry Smith .  snes - the SNES context
3610e35cf81dSBarry Smith 
3611e35cf81dSBarry Smith    Output Parameter:
3612e35cf81dSBarry 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
3613e35cf81dSBarry Smith          the Jacobian is built etc.
3614e35cf81dSBarry Smith 
3615e35cf81dSBarry Smith    Notes:
3616e35cf81dSBarry Smith    The default is 1
36173d5a8a6aSBarry Smith    The jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 or SNESSetLagJacobianPersists() was called.
3618e35cf81dSBarry Smith 
3619e35cf81dSBarry Smith    Level: intermediate
3620e35cf81dSBarry Smith 
36213d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagJacobian(), SNESSetLagPreconditioner(), SNESGetLagPreconditioner(), SNESSetLagJacobianPersists(), SNESSetLagPreconditionerPersists()
3622e35cf81dSBarry Smith 
3623e35cf81dSBarry Smith @*/
36247087cfbeSBarry Smith PetscErrorCode  SNESGetLagJacobian(SNES snes,PetscInt *lag)
3625e35cf81dSBarry Smith {
3626e35cf81dSBarry Smith   PetscFunctionBegin;
36270700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3628e35cf81dSBarry Smith   *lag = snes->lagjacobian;
3629e35cf81dSBarry Smith   PetscFunctionReturn(0);
3630e35cf81dSBarry Smith }
3631e35cf81dSBarry Smith 
363237ec4e1aSPeter Brune /*@
363337ec4e1aSPeter Brune    SNESSetLagJacobianPersists - Set whether or not the Jacobian lagging persists through multiple solves
363437ec4e1aSPeter Brune 
363537ec4e1aSPeter Brune    Logically collective on SNES
363637ec4e1aSPeter Brune 
363737ec4e1aSPeter Brune    Input Parameter:
363837ec4e1aSPeter Brune +  snes - the SNES context
36399d7e2deaSPeter Brune -   flg - jacobian lagging persists if true
364037ec4e1aSPeter Brune 
364137ec4e1aSPeter Brune    Options Database Keys:
36423d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
36433d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
36443d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
36453d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag
36463d5a8a6aSBarry Smith 
364737ec4e1aSPeter Brune 
364895452b02SPatrick Sanan    Notes:
364995452b02SPatrick Sanan     This is useful both for nonlinear preconditioning, where it's appropriate to have the Jacobian be stale by
365037ec4e1aSPeter Brune    several solves, and for implicit time-stepping, where Jacobian lagging in the inner nonlinear solve over several
365137ec4e1aSPeter Brune    timesteps may present huge efficiency gains.
365237ec4e1aSPeter Brune 
365337ec4e1aSPeter Brune    Level: developer
365437ec4e1aSPeter Brune 
36553d5a8a6aSBarry Smith .seealso: SNESSetLagPreconditionerPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC(), SNESSetLagJacobianPersists()
365637ec4e1aSPeter Brune 
365737ec4e1aSPeter Brune @*/
365837ec4e1aSPeter Brune PetscErrorCode  SNESSetLagJacobianPersists(SNES snes,PetscBool flg)
365937ec4e1aSPeter Brune {
366037ec4e1aSPeter Brune   PetscFunctionBegin;
366137ec4e1aSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
366237ec4e1aSPeter Brune   PetscValidLogicalCollectiveBool(snes,flg,2);
366337ec4e1aSPeter Brune   snes->lagjac_persist = flg;
366437ec4e1aSPeter Brune   PetscFunctionReturn(0);
366537ec4e1aSPeter Brune }
366637ec4e1aSPeter Brune 
366737ec4e1aSPeter Brune /*@
3668d8e291bfSBarry Smith    SNESSetLagPreconditionerPersists - Set whether or not the preconditioner lagging persists through multiple nonlinear solves
366937ec4e1aSPeter Brune 
367037ec4e1aSPeter Brune    Logically Collective on SNES
367137ec4e1aSPeter Brune 
367237ec4e1aSPeter Brune    Input Parameter:
367337ec4e1aSPeter Brune +  snes - the SNES context
36749d7e2deaSPeter Brune -   flg - preconditioner lagging persists if true
367537ec4e1aSPeter Brune 
367637ec4e1aSPeter Brune    Options Database Keys:
36773d5a8a6aSBarry Smith +    -snes_lag_jacobian_persists <true,false> - sets the persistence
36783d5a8a6aSBarry Smith .    -snes_lag_jacobian <-2,1,2,...> - sets the lag
36793d5a8a6aSBarry Smith .    -snes_lag_preconditioner_persists <true,false> - sets the persistence
36803d5a8a6aSBarry Smith -    -snes_lag_preconditioner <-2,1,2,...> - sets the lag
368137ec4e1aSPeter Brune 
368295452b02SPatrick Sanan    Notes:
368395452b02SPatrick Sanan     This is useful both for nonlinear preconditioning, where it's appropriate to have the preconditioner be stale
368437ec4e1aSPeter Brune    by several solves, and for implicit time-stepping, where preconditioner lagging in the inner nonlinear solve over
368537ec4e1aSPeter Brune    several timesteps may present huge efficiency gains.
368637ec4e1aSPeter Brune 
368737ec4e1aSPeter Brune    Level: developer
368837ec4e1aSPeter Brune 
36893d5a8a6aSBarry Smith .seealso: SNESSetLagJacobianPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC(), SNESSetLagPreconditioner()
369037ec4e1aSPeter Brune 
369137ec4e1aSPeter Brune @*/
369237ec4e1aSPeter Brune PetscErrorCode  SNESSetLagPreconditionerPersists(SNES snes,PetscBool flg)
369337ec4e1aSPeter Brune {
369437ec4e1aSPeter Brune   PetscFunctionBegin;
369537ec4e1aSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
369637ec4e1aSPeter Brune   PetscValidLogicalCollectiveBool(snes,flg,2);
369737ec4e1aSPeter Brune   snes->lagpre_persist = flg;
369837ec4e1aSPeter Brune   PetscFunctionReturn(0);
369937ec4e1aSPeter Brune }
370037ec4e1aSPeter Brune 
37019b94acceSBarry Smith /*@
3702be5caee7SBarry Smith    SNESSetForceIteration - force SNESSolve() to take at least one iteration regardless of the initial residual norm
3703be5caee7SBarry Smith 
3704be5caee7SBarry Smith    Logically Collective on SNES
3705be5caee7SBarry Smith 
3706be5caee7SBarry Smith    Input Parameters:
3707be5caee7SBarry Smith +  snes - the SNES context
3708be5caee7SBarry Smith -  force - PETSC_TRUE require at least one iteration
3709be5caee7SBarry Smith 
3710be5caee7SBarry Smith    Options Database Keys:
3711be5caee7SBarry Smith .    -snes_force_iteration <force> - Sets forcing an iteration
3712be5caee7SBarry Smith 
3713be5caee7SBarry Smith    Notes:
3714be5caee7SBarry Smith    This is used sometimes with TS to prevent TS from detecting a false steady state solution
3715be5caee7SBarry Smith 
3716be5caee7SBarry Smith    Level: intermediate
3717be5caee7SBarry Smith 
3718be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance()
3719be5caee7SBarry Smith @*/
3720be5caee7SBarry Smith PetscErrorCode  SNESSetForceIteration(SNES snes,PetscBool force)
3721be5caee7SBarry Smith {
3722be5caee7SBarry Smith   PetscFunctionBegin;
3723be5caee7SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3724be5caee7SBarry Smith   snes->forceiteration = force;
3725be5caee7SBarry Smith   PetscFunctionReturn(0);
3726be5caee7SBarry Smith }
3727be5caee7SBarry Smith 
372885216dc7SFande Kong /*@
372985216dc7SFande Kong    SNESGetForceIteration - Whether or not to force SNESSolve() take at least one iteration regardless of the initial residual norm
373085216dc7SFande Kong 
373185216dc7SFande Kong    Logically Collective on SNES
373285216dc7SFande Kong 
373385216dc7SFande Kong    Input Parameters:
373485216dc7SFande Kong .  snes - the SNES context
373585216dc7SFande Kong 
373685216dc7SFande Kong    Output Parameter:
373785216dc7SFande Kong .  force - PETSC_TRUE requires at least one iteration.
373885216dc7SFande Kong 
373906dd6b0eSSatish Balay    Level: intermediate
374006dd6b0eSSatish Balay 
374185216dc7SFande Kong .seealso: SNESSetForceIteration(), SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance()
374285216dc7SFande Kong @*/
374385216dc7SFande Kong PetscErrorCode  SNESGetForceIteration(SNES snes,PetscBool *force)
374485216dc7SFande Kong {
374585216dc7SFande Kong   PetscFunctionBegin;
374685216dc7SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
374785216dc7SFande Kong   *force = snes->forceiteration;
374885216dc7SFande Kong   PetscFunctionReturn(0);
374985216dc7SFande Kong }
3750be5caee7SBarry Smith 
3751be5caee7SBarry Smith /*@
3752d7a720efSLois Curfman McInnes    SNESSetTolerances - Sets various parameters used in convergence tests.
37539b94acceSBarry Smith 
37543f9fe445SBarry Smith    Logically Collective on SNES
3755c7afd0dbSLois Curfman McInnes 
37569b94acceSBarry Smith    Input Parameters:
3757c7afd0dbSLois Curfman McInnes +  snes - the SNES context
375870441072SBarry Smith .  abstol - absolute convergence tolerance
375933174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
37605358d0d4SBarry Smith .  stol -  convergence tolerance in terms of the norm of the change in the solution between steps,  || delta x || < stol*|| x ||
376133174efeSLois Curfman McInnes .  maxit - maximum number of iterations
3762e71169deSBarry Smith -  maxf - maximum number of function evaluations (-1 indicates no limit)
3763fee21e36SBarry Smith 
376433174efeSLois Curfman McInnes    Options Database Keys:
376570441072SBarry Smith +    -snes_atol <abstol> - Sets abstol
3766c7afd0dbSLois Curfman McInnes .    -snes_rtol <rtol> - Sets rtol
3767c7afd0dbSLois Curfman McInnes .    -snes_stol <stol> - Sets stol
3768c7afd0dbSLois Curfman McInnes .    -snes_max_it <maxit> - Sets maxit
3769c7afd0dbSLois Curfman McInnes -    -snes_max_funcs <maxf> - Sets maxf
37709b94acceSBarry Smith 
3771d7a720efSLois Curfman McInnes    Notes:
37729b94acceSBarry Smith    The default maximum number of iterations is 50.
37739b94acceSBarry Smith    The default maximum number of function evaluations is 1000.
37749b94acceSBarry Smith 
377536851e7fSLois Curfman McInnes    Level: intermediate
377636851e7fSLois Curfman McInnes 
3777be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance(), SNESSetForceIteration()
37789b94acceSBarry Smith @*/
37797087cfbeSBarry Smith PetscErrorCode  SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf)
37809b94acceSBarry Smith {
37813a40ed3dSBarry Smith   PetscFunctionBegin;
37820700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3783c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,abstol,2);
3784c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol,3);
3785c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,stol,4);
3786c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxit,5);
3787c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxf,6);
3788c5eb9154SBarry Smith 
3789ab54825eSJed Brown   if (abstol != PETSC_DEFAULT) {
379057622a8eSBarry Smith     if (abstol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Absolute tolerance %g must be non-negative",(double)abstol);
3791ab54825eSJed Brown     snes->abstol = abstol;
3792ab54825eSJed Brown   }
3793ab54825eSJed Brown   if (rtol != PETSC_DEFAULT) {
379457622a8eSBarry 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);
3795ab54825eSJed Brown     snes->rtol = rtol;
3796ab54825eSJed Brown   }
3797ab54825eSJed Brown   if (stol != PETSC_DEFAULT) {
379857622a8eSBarry Smith     if (stol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Step tolerance %g must be non-negative",(double)stol);
3799c60f73f4SPeter Brune     snes->stol = stol;
3800ab54825eSJed Brown   }
3801ab54825eSJed Brown   if (maxit != PETSC_DEFAULT) {
3802ce94432eSBarry Smith     if (maxit < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of iterations %D must be non-negative",maxit);
3803ab54825eSJed Brown     snes->max_its = maxit;
3804ab54825eSJed Brown   }
3805ab54825eSJed Brown   if (maxf != PETSC_DEFAULT) {
3806e71169deSBarry Smith     if (maxf < -1) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of function evaluations %D must be -1 or nonnegative",maxf);
3807ab54825eSJed Brown     snes->max_funcs = maxf;
3808ab54825eSJed Brown   }
380988976e71SPeter Brune   snes->tolerancesset = PETSC_TRUE;
38103a40ed3dSBarry Smith   PetscFunctionReturn(0);
38119b94acceSBarry Smith }
38129b94acceSBarry Smith 
3813e4d06f11SPatrick Farrell /*@
3814e4d06f11SPatrick Farrell    SNESSetDivergenceTolerance - Sets the divergence tolerance used for the SNES divergence test.
3815e4d06f11SPatrick Farrell 
3816e4d06f11SPatrick Farrell    Logically Collective on SNES
3817e4d06f11SPatrick Farrell 
3818e4d06f11SPatrick Farrell    Input Parameters:
3819e4d06f11SPatrick Farrell +  snes - the SNES context
3820e4d06f11SPatrick Farrell -  divtol - the divergence tolerance. Use -1 to deactivate the test.
3821e4d06f11SPatrick Farrell 
3822e4d06f11SPatrick Farrell    Options Database Keys:
3823a2b725a8SWilliam Gropp .    -snes_divergence_tolerance <divtol> - Sets divtol
3824e4d06f11SPatrick Farrell 
3825e4d06f11SPatrick Farrell    Notes:
3826e4d06f11SPatrick Farrell    The default divergence tolerance is 1e4.
3827e4d06f11SPatrick Farrell 
3828e4d06f11SPatrick Farrell    Level: intermediate
3829e4d06f11SPatrick Farrell 
3830e4d06f11SPatrick Farrell .seealso: SNESSetTolerances(), SNESGetDivergenceTolerance
3831e4d06f11SPatrick Farrell @*/
3832e4d06f11SPatrick Farrell PetscErrorCode  SNESSetDivergenceTolerance(SNES snes,PetscReal divtol)
3833e4d06f11SPatrick Farrell {
3834e4d06f11SPatrick Farrell   PetscFunctionBegin;
3835e4d06f11SPatrick Farrell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3836e4d06f11SPatrick Farrell   PetscValidLogicalCollectiveReal(snes,divtol,2);
3837e4d06f11SPatrick Farrell 
3838e4d06f11SPatrick Farrell   if (divtol != PETSC_DEFAULT) {
3839e4d06f11SPatrick Farrell     snes->divtol = divtol;
3840e4d06f11SPatrick Farrell   }
3841e4d06f11SPatrick Farrell   else {
3842e4d06f11SPatrick Farrell     snes->divtol = 1.0e4;
3843e4d06f11SPatrick Farrell   }
3844e4d06f11SPatrick Farrell   PetscFunctionReturn(0);
3845e4d06f11SPatrick Farrell }
3846e4d06f11SPatrick Farrell 
38479b94acceSBarry Smith /*@
384833174efeSLois Curfman McInnes    SNESGetTolerances - Gets various parameters used in convergence tests.
384933174efeSLois Curfman McInnes 
3850c7afd0dbSLois Curfman McInnes    Not Collective
3851c7afd0dbSLois Curfman McInnes 
385233174efeSLois Curfman McInnes    Input Parameters:
3853c7afd0dbSLois Curfman McInnes +  snes - the SNES context
385485385478SLisandro Dalcin .  atol - absolute convergence tolerance
385533174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
385633174efeSLois Curfman McInnes .  stol -  convergence tolerance in terms of the norm
385733174efeSLois Curfman McInnes            of the change in the solution between steps
385833174efeSLois Curfman McInnes .  maxit - maximum number of iterations
3859c7afd0dbSLois Curfman McInnes -  maxf - maximum number of function evaluations
3860fee21e36SBarry Smith 
386133174efeSLois Curfman McInnes    Notes:
38620298fd71SBarry Smith    The user can specify NULL for any parameter that is not needed.
386333174efeSLois Curfman McInnes 
386436851e7fSLois Curfman McInnes    Level: intermediate
386536851e7fSLois Curfman McInnes 
386633174efeSLois Curfman McInnes .seealso: SNESSetTolerances()
386733174efeSLois Curfman McInnes @*/
38687087cfbeSBarry Smith PetscErrorCode  SNESGetTolerances(SNES snes,PetscReal *atol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf)
386933174efeSLois Curfman McInnes {
38703a40ed3dSBarry Smith   PetscFunctionBegin;
38710700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
387285385478SLisandro Dalcin   if (atol)  *atol  = snes->abstol;
387333174efeSLois Curfman McInnes   if (rtol)  *rtol  = snes->rtol;
3874c60f73f4SPeter Brune   if (stol)  *stol  = snes->stol;
387533174efeSLois Curfman McInnes   if (maxit) *maxit = snes->max_its;
387633174efeSLois Curfman McInnes   if (maxf)  *maxf  = snes->max_funcs;
38773a40ed3dSBarry Smith   PetscFunctionReturn(0);
387833174efeSLois Curfman McInnes }
387933174efeSLois Curfman McInnes 
3880e4d06f11SPatrick Farrell /*@
3881e4d06f11SPatrick Farrell    SNESGetDivergenceTolerance - Gets divergence tolerance used in divergence test.
3882e4d06f11SPatrick Farrell 
3883e4d06f11SPatrick Farrell    Not Collective
3884e4d06f11SPatrick Farrell 
3885e4d06f11SPatrick Farrell    Input Parameters:
3886e4d06f11SPatrick Farrell +  snes - the SNES context
3887e4d06f11SPatrick Farrell -  divtol - divergence tolerance
3888e4d06f11SPatrick Farrell 
3889e4d06f11SPatrick Farrell    Level: intermediate
3890e4d06f11SPatrick Farrell 
3891e4d06f11SPatrick Farrell .seealso: SNESSetDivergenceTolerance()
3892e4d06f11SPatrick Farrell @*/
3893e4d06f11SPatrick Farrell PetscErrorCode  SNESGetDivergenceTolerance(SNES snes,PetscReal *divtol)
3894e4d06f11SPatrick Farrell {
3895e4d06f11SPatrick Farrell   PetscFunctionBegin;
3896e4d06f11SPatrick Farrell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3897e4d06f11SPatrick Farrell   if (divtol) *divtol = snes->divtol;
3898e4d06f11SPatrick Farrell   PetscFunctionReturn(0);
3899e4d06f11SPatrick Farrell }
3900e4d06f11SPatrick Farrell 
390133174efeSLois Curfman McInnes /*@
39029b94acceSBarry Smith    SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance.
39039b94acceSBarry Smith 
39043f9fe445SBarry Smith    Logically Collective on SNES
3905fee21e36SBarry Smith 
3906c7afd0dbSLois Curfman McInnes    Input Parameters:
3907c7afd0dbSLois Curfman McInnes +  snes - the SNES context
3908c7afd0dbSLois Curfman McInnes -  tol - tolerance
3909c7afd0dbSLois Curfman McInnes 
39109b94acceSBarry Smith    Options Database Key:
3911c7afd0dbSLois Curfman McInnes .  -snes_trtol <tol> - Sets tol
39129b94acceSBarry Smith 
391336851e7fSLois Curfman McInnes    Level: intermediate
391436851e7fSLois Curfman McInnes 
39152492ecdbSBarry Smith .seealso: SNESSetTolerances()
39169b94acceSBarry Smith @*/
39177087cfbeSBarry Smith PetscErrorCode  SNESSetTrustRegionTolerance(SNES snes,PetscReal tol)
39189b94acceSBarry Smith {
39193a40ed3dSBarry Smith   PetscFunctionBegin;
39200700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3921c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,tol,2);
39229b94acceSBarry Smith   snes->deltatol = tol;
39233a40ed3dSBarry Smith   PetscFunctionReturn(0);
39249b94acceSBarry Smith }
39259b94acceSBarry Smith 
39266ba87a44SLisandro Dalcin PETSC_INTERN PetscErrorCode  SNESMonitorRange_Private(SNES,PetscInt,PetscReal*);
39276ba87a44SLisandro Dalcin 
39287087cfbeSBarry Smith PetscErrorCode  SNESMonitorLGRange(SNES snes,PetscInt n,PetscReal rnorm,void *monctx)
3929b271bb04SBarry Smith {
3930b271bb04SBarry Smith   PetscDrawLG      lg;
3931b271bb04SBarry Smith   PetscErrorCode   ierr;
3932b271bb04SBarry Smith   PetscReal        x,y,per;
3933b271bb04SBarry Smith   PetscViewer      v = (PetscViewer)monctx;
3934b271bb04SBarry Smith   static PetscReal prev; /* should be in the context */
3935b271bb04SBarry Smith   PetscDraw        draw;
3936b271bb04SBarry Smith 
3937459f5d12SBarry Smith   PetscFunctionBegin;
39384d4332d5SBarry Smith   PetscValidHeaderSpecific(v,PETSC_VIEWER_CLASSID,4);
3939b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,0,&lg);CHKERRQ(ierr);
3940b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3941b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3942b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"Residual norm");CHKERRQ(ierr);
3943b271bb04SBarry Smith   x    = (PetscReal)n;
394477b4d14cSPeter Brune   if (rnorm > 0.0) y = PetscLog10Real(rnorm);
394594c9c6d3SKarl Rupp   else y = -15.0;
3946b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
39476934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3948b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
39496934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3950b271bb04SBarry Smith   }
3951b271bb04SBarry Smith 
3952b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,1,&lg);CHKERRQ(ierr);
3953b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3954b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3955b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"% elemts > .2*max elemt");CHKERRQ(ierr);
3956b271bb04SBarry Smith   ierr =  SNESMonitorRange_Private(snes,n,&per);CHKERRQ(ierr);
3957b271bb04SBarry Smith   x    = (PetscReal)n;
3958b271bb04SBarry Smith   y    = 100.0*per;
3959b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
39606934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3961b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
39626934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3963b271bb04SBarry Smith   }
3964b271bb04SBarry Smith 
3965b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,2,&lg);CHKERRQ(ierr);
3966b271bb04SBarry Smith   if (!n) {prev = rnorm;ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3967b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3968b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm");CHKERRQ(ierr);
3969b271bb04SBarry Smith   x    = (PetscReal)n;
3970b271bb04SBarry Smith   y    = (prev - rnorm)/prev;
3971b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
39726934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3973b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
39746934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3975b271bb04SBarry Smith   }
3976b271bb04SBarry Smith 
3977b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,3,&lg);CHKERRQ(ierr);
3978b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3979b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3980b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr);
3981b271bb04SBarry Smith   x    = (PetscReal)n;
3982b271bb04SBarry Smith   y    = (prev - rnorm)/(prev*per);
3983b271bb04SBarry Smith   if (n > 2) { /*skip initial crazy value */
3984b271bb04SBarry Smith     ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
3985b271bb04SBarry Smith   }
39866934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3987b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
39886934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3989b271bb04SBarry Smith   }
3990b271bb04SBarry Smith   prev = rnorm;
3991b271bb04SBarry Smith   PetscFunctionReturn(0);
3992b271bb04SBarry Smith }
3993b271bb04SBarry Smith 
3994228d79bcSJed Brown /*@
3995228d79bcSJed Brown    SNESMonitor - runs the user provided monitor routines, if they exist
3996228d79bcSJed Brown 
3997228d79bcSJed Brown    Collective on SNES
3998228d79bcSJed Brown 
3999228d79bcSJed Brown    Input Parameters:
4000228d79bcSJed Brown +  snes - nonlinear solver context obtained from SNESCreate()
4001228d79bcSJed Brown .  iter - iteration number
4002228d79bcSJed Brown -  rnorm - relative norm of the residual
4003228d79bcSJed Brown 
4004228d79bcSJed Brown    Notes:
4005228d79bcSJed Brown    This routine is called by the SNES implementations.
4006228d79bcSJed Brown    It does not typically need to be called by the user.
4007228d79bcSJed Brown 
4008228d79bcSJed Brown    Level: developer
4009228d79bcSJed Brown 
4010228d79bcSJed Brown .seealso: SNESMonitorSet()
4011228d79bcSJed Brown @*/
40127a03ce2fSLisandro Dalcin PetscErrorCode  SNESMonitor(SNES snes,PetscInt iter,PetscReal rnorm)
40137a03ce2fSLisandro Dalcin {
40147a03ce2fSLisandro Dalcin   PetscErrorCode ierr;
40157a03ce2fSLisandro Dalcin   PetscInt       i,n = snes->numbermonitors;
40167a03ce2fSLisandro Dalcin 
40177a03ce2fSLisandro Dalcin   PetscFunctionBegin;
40188860a134SJunchao Zhang   ierr = VecLockReadPush(snes->vec_sol);CHKERRQ(ierr);
40197a03ce2fSLisandro Dalcin   for (i=0; i<n; i++) {
40207a03ce2fSLisandro Dalcin     ierr = (*snes->monitor[i])(snes,iter,rnorm,snes->monitorcontext[i]);CHKERRQ(ierr);
40217a03ce2fSLisandro Dalcin   }
40228860a134SJunchao Zhang   ierr = VecLockReadPop(snes->vec_sol);CHKERRQ(ierr);
40237a03ce2fSLisandro Dalcin   PetscFunctionReturn(0);
40247a03ce2fSLisandro Dalcin }
40257a03ce2fSLisandro Dalcin 
40269b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */
40279b94acceSBarry Smith 
4028bf388a1fSBarry Smith /*MC
4029bf388a1fSBarry Smith     SNESMonitorFunction - functional form passed to SNESMonitorSet() to monitor convergence of nonlinear solver
4030bf388a1fSBarry Smith 
4031bf388a1fSBarry Smith      Synopsis:
4032aaa7dc30SBarry Smith      #include <petscsnes.h>
4033bf388a1fSBarry Smith $    PetscErrorCode SNESMonitorFunction(SNES snes,PetscInt its, PetscReal norm,void *mctx)
4034bf388a1fSBarry Smith 
40351843f636SBarry Smith      Collective on snes
40361843f636SBarry Smith 
40371843f636SBarry Smith     Input Parameters:
4038bf388a1fSBarry Smith +    snes - the SNES context
4039bf388a1fSBarry Smith .    its - iteration number
4040bf388a1fSBarry Smith .    norm - 2-norm function value (may be estimated)
4041bf388a1fSBarry Smith -    mctx - [optional] monitoring context
4042bf388a1fSBarry Smith 
4043878cb397SSatish Balay    Level: advanced
4044878cb397SSatish Balay 
4045bf388a1fSBarry Smith .seealso:   SNESMonitorSet(), SNESMonitorGet()
4046bf388a1fSBarry Smith M*/
4047bf388a1fSBarry Smith 
40489b94acceSBarry Smith /*@C
4049a6570f20SBarry Smith    SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every
40509b94acceSBarry Smith    iteration of the nonlinear solver to display the iteration's
40519b94acceSBarry Smith    progress.
40529b94acceSBarry Smith 
40533f9fe445SBarry Smith    Logically Collective on SNES
4054fee21e36SBarry Smith 
4055c7afd0dbSLois Curfman McInnes    Input Parameters:
4056c7afd0dbSLois Curfman McInnes +  snes - the SNES context
40576e4dcb14SBarry Smith .  f - the monitor function, see SNESMonitorFunction for the calling sequence
4058b8a78c4aSBarry Smith .  mctx - [optional] user-defined context for private data for the
40590298fd71SBarry Smith           monitor routine (use NULL if no context is desired)
4060b3006f0bSLois Curfman McInnes -  monitordestroy - [optional] routine that frees monitor context
40610298fd71SBarry Smith           (may be NULL)
40629b94acceSBarry Smith 
40639665c990SLois Curfman McInnes    Options Database Keys:
4064a6570f20SBarry Smith +    -snes_monitor        - sets SNESMonitorDefault()
4065798534f6SMatthew G. Knepley .    -snes_monitor draw::draw_lg - sets line graph monitor,
4066cca6129bSJed Brown -    -snes_monitor_cancel - cancels all monitors that have
4067c7afd0dbSLois Curfman McInnes                             been hardwired into a code by
4068a6570f20SBarry Smith                             calls to SNESMonitorSet(), but
4069c7afd0dbSLois Curfman McInnes                             does not cancel those set via
4070c7afd0dbSLois Curfman McInnes                             the options database.
40719665c990SLois Curfman McInnes 
4072639f9d9dSBarry Smith    Notes:
40736bc08f3fSLois Curfman McInnes    Several different monitoring routines may be set by calling
4074a6570f20SBarry Smith    SNESMonitorSet() multiple times; all will be called in the
40756bc08f3fSLois Curfman McInnes    order in which they were set.
4076639f9d9dSBarry Smith 
407795452b02SPatrick Sanan    Fortran Notes:
407895452b02SPatrick Sanan     Only a single monitor function can be set for each SNES object
4079025f1a04SBarry Smith 
408036851e7fSLois Curfman McInnes    Level: intermediate
408136851e7fSLois Curfman McInnes 
4082bf388a1fSBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorCancel(), SNESMonitorFunction
40839b94acceSBarry Smith @*/
40846e4dcb14SBarry Smith PetscErrorCode  SNESMonitorSet(SNES snes,PetscErrorCode (*f)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void**))
40859b94acceSBarry Smith {
4086b90d0a6eSBarry Smith   PetscInt       i;
4087649052a6SBarry Smith   PetscErrorCode ierr;
408878064530SBarry Smith   PetscBool      identical;
4089b90d0a6eSBarry Smith 
40903a40ed3dSBarry Smith   PetscFunctionBegin;
40910700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4092b90d0a6eSBarry Smith   for (i=0; i<snes->numbermonitors;i++) {
409378064530SBarry Smith     ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,mctx,monitordestroy,(PetscErrorCode (*)(void))snes->monitor[i],snes->monitorcontext[i],snes->monitordestroy[i],&identical);CHKERRQ(ierr);
409478064530SBarry Smith     if (identical) PetscFunctionReturn(0);
4095649052a6SBarry Smith   }
409678064530SBarry Smith   if (snes->numbermonitors >= MAXSNESMONITORS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set");
40976e4dcb14SBarry Smith   snes->monitor[snes->numbermonitors]          = f;
4098b8a78c4aSBarry Smith   snes->monitordestroy[snes->numbermonitors]   = monitordestroy;
4099639f9d9dSBarry Smith   snes->monitorcontext[snes->numbermonitors++] = (void*)mctx;
41003a40ed3dSBarry Smith   PetscFunctionReturn(0);
41019b94acceSBarry Smith }
41029b94acceSBarry Smith 
4103a278d85bSSatish Balay /*@
4104a6570f20SBarry Smith    SNESMonitorCancel - Clears all the monitor functions for a SNES object.
41055cd90555SBarry Smith 
41063f9fe445SBarry Smith    Logically Collective on SNES
4107c7afd0dbSLois Curfman McInnes 
41085cd90555SBarry Smith    Input Parameters:
41095cd90555SBarry Smith .  snes - the SNES context
41105cd90555SBarry Smith 
41111a480d89SAdministrator    Options Database Key:
4112a6570f20SBarry Smith .  -snes_monitor_cancel - cancels all monitors that have been hardwired
4113a6570f20SBarry Smith     into a code by calls to SNESMonitorSet(), but does not cancel those
4114c7afd0dbSLois Curfman McInnes     set via the options database
41155cd90555SBarry Smith 
41165cd90555SBarry Smith    Notes:
41175cd90555SBarry Smith    There is no way to clear one specific monitor from a SNES object.
41185cd90555SBarry Smith 
411936851e7fSLois Curfman McInnes    Level: intermediate
412036851e7fSLois Curfman McInnes 
4121a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorSet()
41225cd90555SBarry Smith @*/
41237087cfbeSBarry Smith PetscErrorCode  SNESMonitorCancel(SNES snes)
41245cd90555SBarry Smith {
4125d952e501SBarry Smith   PetscErrorCode ierr;
4126d952e501SBarry Smith   PetscInt       i;
4127d952e501SBarry Smith 
41285cd90555SBarry Smith   PetscFunctionBegin;
41290700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4130d952e501SBarry Smith   for (i=0; i<snes->numbermonitors; i++) {
4131d952e501SBarry Smith     if (snes->monitordestroy[i]) {
41323c4aec1bSBarry Smith       ierr = (*snes->monitordestroy[i])(&snes->monitorcontext[i]);CHKERRQ(ierr);
4133d952e501SBarry Smith     }
4134d952e501SBarry Smith   }
41355cd90555SBarry Smith   snes->numbermonitors = 0;
41365cd90555SBarry Smith   PetscFunctionReturn(0);
41375cd90555SBarry Smith }
41385cd90555SBarry Smith 
4139bf388a1fSBarry Smith /*MC
4140bf388a1fSBarry Smith     SNESConvergenceTestFunction - functional form used for testing of convergence of nonlinear solver
4141bf388a1fSBarry Smith 
4142bf388a1fSBarry Smith      Synopsis:
4143aaa7dc30SBarry Smith      #include <petscsnes.h>
4144bf388a1fSBarry Smith $     PetscErrorCode SNESConvergenceTest(SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx)
4145bf388a1fSBarry Smith 
41461843f636SBarry Smith      Collective on snes
41471843f636SBarry Smith 
41481843f636SBarry Smith     Input Parameters:
4149bf388a1fSBarry Smith +    snes - the SNES context
4150bf388a1fSBarry Smith .    it - current iteration (0 is the first and is before any Newton step)
4151bf388a1fSBarry Smith .    xnorm - 2-norm of current iterate
4152bf388a1fSBarry Smith .    gnorm - 2-norm of current step
41531843f636SBarry Smith .    f - 2-norm of function
41541843f636SBarry Smith -    cctx - [optional] convergence context
41551843f636SBarry Smith 
41561843f636SBarry Smith     Output Parameter:
41571843f636SBarry Smith .    reason - reason for convergence/divergence, only needs to be set when convergence or divergence is detected
4158bf388a1fSBarry Smith 
4159878cb397SSatish Balay    Level: intermediate
4160bf388a1fSBarry Smith 
4161bf388a1fSBarry Smith .seealso:   SNESSetConvergenceTest(), SNESGetConvergenceTest()
4162bf388a1fSBarry Smith M*/
4163bf388a1fSBarry Smith 
41649b94acceSBarry Smith /*@C
41659b94acceSBarry Smith    SNESSetConvergenceTest - Sets the function that is to be used
41669b94acceSBarry Smith    to test for convergence of the nonlinear iterative solution.
41679b94acceSBarry Smith 
41683f9fe445SBarry Smith    Logically Collective on SNES
4169fee21e36SBarry Smith 
4170c7afd0dbSLois Curfman McInnes    Input Parameters:
4171c7afd0dbSLois Curfman McInnes +  snes - the SNES context
4172bf388a1fSBarry Smith .  SNESConvergenceTestFunction - routine to test for convergence
41730298fd71SBarry Smith .  cctx - [optional] context for private data for the convergence routine  (may be NULL)
4174cf90aa19SBarry Smith -  destroy - [optional] destructor for the context (may be NULL; PETSC_NULL_FUNCTION in Fortran)
41759b94acceSBarry Smith 
417636851e7fSLois Curfman McInnes    Level: advanced
417736851e7fSLois Curfman McInnes 
4178e2a6519dSDmitry Karpeev .seealso: SNESConvergedDefault(), SNESConvergedSkip(), SNESConvergenceTestFunction
41799b94acceSBarry Smith @*/
4180bf388a1fSBarry Smith PetscErrorCode  SNESSetConvergenceTest(SNES snes,PetscErrorCode (*SNESConvergenceTestFunction)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx,PetscErrorCode (*destroy)(void*))
41819b94acceSBarry Smith {
41827f7931b9SBarry Smith   PetscErrorCode ierr;
41837f7931b9SBarry Smith 
41843a40ed3dSBarry Smith   PetscFunctionBegin;
41850700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4186e2a6519dSDmitry Karpeev   if (!SNESConvergenceTestFunction) SNESConvergenceTestFunction = SNESConvergedSkip;
41877f7931b9SBarry Smith   if (snes->ops->convergeddestroy) {
41887f7931b9SBarry Smith     ierr = (*snes->ops->convergeddestroy)(snes->cnvP);CHKERRQ(ierr);
41897f7931b9SBarry Smith   }
4190bf388a1fSBarry Smith   snes->ops->converged        = SNESConvergenceTestFunction;
41917f7931b9SBarry Smith   snes->ops->convergeddestroy = destroy;
419285385478SLisandro Dalcin   snes->cnvP                  = cctx;
41933a40ed3dSBarry Smith   PetscFunctionReturn(0);
41949b94acceSBarry Smith }
41959b94acceSBarry Smith 
419652baeb72SSatish Balay /*@
4197184914b5SBarry Smith    SNESGetConvergedReason - Gets the reason the SNES iteration was stopped.
4198184914b5SBarry Smith 
4199184914b5SBarry Smith    Not Collective
4200184914b5SBarry Smith 
4201184914b5SBarry Smith    Input Parameter:
4202184914b5SBarry Smith .  snes - the SNES context
4203184914b5SBarry Smith 
4204184914b5SBarry Smith    Output Parameter:
42054d0a8057SBarry Smith .  reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the
4206184914b5SBarry Smith             manual pages for the individual convergence tests for complete lists
4207184914b5SBarry Smith 
42086a4d7782SBarry Smith    Options Database:
42096a4d7782SBarry Smith .   -snes_converged_reason - prints the reason to standard out
42106a4d7782SBarry Smith 
4211184914b5SBarry Smith    Level: intermediate
4212184914b5SBarry Smith 
421395452b02SPatrick Sanan    Notes:
421495452b02SPatrick Sanan     Should only be called after the call the SNESSolve() is complete, if it is called earlier it returns the value SNES__CONVERGED_ITERATING.
4215184914b5SBarry Smith 
421633866048SMatthew G. Knepley .seealso: SNESSetConvergenceTest(), SNESSetConvergedReason(), SNESConvergedReason
4217184914b5SBarry Smith @*/
42187087cfbeSBarry Smith PetscErrorCode SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason)
4219184914b5SBarry Smith {
4220184914b5SBarry Smith   PetscFunctionBegin;
42210700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
42224482741eSBarry Smith   PetscValidPointer(reason,2);
4223184914b5SBarry Smith   *reason = snes->reason;
4224184914b5SBarry Smith   PetscFunctionReturn(0);
4225184914b5SBarry Smith }
4226184914b5SBarry Smith 
4227c4421ceaSFande Kong /*@C
4228c4421ceaSFande Kong    SNESGetConvergedReasonString - Return a human readable string for snes converged reason
4229c4421ceaSFande Kong 
4230c4421ceaSFande Kong    Not Collective
4231c4421ceaSFande Kong 
4232c4421ceaSFande Kong    Input Parameter:
4233c4421ceaSFande Kong .  snes - the SNES context
4234c4421ceaSFande Kong 
4235c4421ceaSFande Kong    Output Parameter:
4236c4421ceaSFande Kong .  strreason - a human readable string that describes SNES converged reason
4237c4421ceaSFande Kong 
423899c90e12SSatish Balay    Level: beginner
4239c4421ceaSFande Kong 
4240c4421ceaSFande Kong .seealso: SNESGetConvergedReason()
4241c4421ceaSFande Kong @*/
4242c4421ceaSFande Kong PetscErrorCode SNESGetConvergedReasonString(SNES snes, const char** strreason)
4243c4421ceaSFande Kong {
4244c4421ceaSFande Kong   PetscFunctionBegin;
4245c4421ceaSFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4246c4421ceaSFande Kong   PetscValidCharPointer(strreason,2);
4247c4421ceaSFande Kong   *strreason = SNESConvergedReasons[snes->reason];
4248c4421ceaSFande Kong   PetscFunctionReturn(0);
4249c4421ceaSFande Kong }
4250c4421ceaSFande Kong 
425133866048SMatthew G. Knepley /*@
425233866048SMatthew G. Knepley    SNESSetConvergedReason - Sets the reason the SNES iteration was stopped.
425333866048SMatthew G. Knepley 
425433866048SMatthew G. Knepley    Not Collective
425533866048SMatthew G. Knepley 
425633866048SMatthew G. Knepley    Input Parameters:
425733866048SMatthew G. Knepley +  snes - the SNES context
425833866048SMatthew G. Knepley -  reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the
425933866048SMatthew G. Knepley             manual pages for the individual convergence tests for complete lists
426033866048SMatthew G. Knepley 
426133866048SMatthew G. Knepley    Level: intermediate
426233866048SMatthew G. Knepley 
426333866048SMatthew G. Knepley .seealso: SNESGetConvergedReason(), SNESSetConvergenceTest(), SNESConvergedReason
426433866048SMatthew G. Knepley @*/
426533866048SMatthew G. Knepley PetscErrorCode SNESSetConvergedReason(SNES snes,SNESConvergedReason reason)
426633866048SMatthew G. Knepley {
426733866048SMatthew G. Knepley   PetscFunctionBegin;
426833866048SMatthew G. Knepley   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
426933866048SMatthew G. Knepley   snes->reason = reason;
427033866048SMatthew G. Knepley   PetscFunctionReturn(0);
427133866048SMatthew G. Knepley }
427233866048SMatthew G. Knepley 
4273c9005455SLois Curfman McInnes /*@
4274c9005455SLois Curfman McInnes    SNESSetConvergenceHistory - Sets the array used to hold the convergence history.
4275c9005455SLois Curfman McInnes 
42763f9fe445SBarry Smith    Logically Collective on SNES
4277fee21e36SBarry Smith 
4278c7afd0dbSLois Curfman McInnes    Input Parameters:
4279c7afd0dbSLois Curfman McInnes +  snes - iterative context obtained from SNESCreate()
42808c7482ecSBarry Smith .  a   - array to hold history, this array will contain the function norms computed at each step
4281cd5578b5SBarry Smith .  its - integer array holds the number of linear iterations for each solve.
4282758f92a0SBarry Smith .  na  - size of a and its
428364731454SLois Curfman McInnes -  reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero,
4284758f92a0SBarry Smith            else it continues storing new values for new nonlinear solves after the old ones
4285c7afd0dbSLois Curfman McInnes 
4286308dcc3eSBarry Smith    Notes:
42870298fd71SBarry Smith    If 'a' and 'its' are NULL then space is allocated for the history. If 'na' PETSC_DECIDE or PETSC_DEFAULT then a
4288308dcc3eSBarry Smith    default array of length 10000 is allocated.
4289308dcc3eSBarry Smith 
4290c9005455SLois Curfman McInnes    This routine is useful, e.g., when running a code for purposes
4291c9005455SLois Curfman McInnes    of accurate performance monitoring, when no I/O should be done
4292c9005455SLois Curfman McInnes    during the section of code that is being timed.
4293c9005455SLois Curfman McInnes 
429436851e7fSLois Curfman McInnes    Level: intermediate
429536851e7fSLois Curfman McInnes 
429608405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory()
4297758f92a0SBarry Smith 
4298c9005455SLois Curfman McInnes @*/
42997087cfbeSBarry Smith PetscErrorCode  SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscBool reset)
4300c9005455SLois Curfman McInnes {
4301308dcc3eSBarry Smith   PetscErrorCode ierr;
4302308dcc3eSBarry Smith 
43033a40ed3dSBarry Smith   PetscFunctionBegin;
43040700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
43057a1ec6d4SBarry Smith   if (a) PetscValidScalarPointer(a,2);
4306a562a398SLisandro Dalcin   if (its) PetscValidIntPointer(its,3);
43077a1ec6d4SBarry Smith   if (!a) {
4308308dcc3eSBarry Smith     if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
4309071fcb05SBarry Smith     ierr = PetscCalloc2(na,&a,na,&its);CHKERRQ(ierr);
4310071fcb05SBarry Smith     snes->conv_hist_alloc = PETSC_TRUE;
4311308dcc3eSBarry Smith   }
4312c9005455SLois Curfman McInnes   snes->conv_hist       = a;
4313758f92a0SBarry Smith   snes->conv_hist_its   = its;
4314758f92a0SBarry Smith   snes->conv_hist_max   = na;
4315a12bc48fSLisandro Dalcin   snes->conv_hist_len   = 0;
4316758f92a0SBarry Smith   snes->conv_hist_reset = reset;
4317758f92a0SBarry Smith   PetscFunctionReturn(0);
4318758f92a0SBarry Smith }
4319758f92a0SBarry Smith 
4320308dcc3eSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
4321c6db04a5SJed Brown #include <engine.h>   /* MATLAB include file */
4322c6db04a5SJed Brown #include <mex.h>      /* MATLAB include file */
432399e0435eSBarry Smith 
43248cc058d9SJed Brown PETSC_EXTERN mxArray *SNESGetConvergenceHistoryMatlab(SNES snes)
4325308dcc3eSBarry Smith {
4326308dcc3eSBarry Smith   mxArray   *mat;
4327308dcc3eSBarry Smith   PetscInt  i;
4328308dcc3eSBarry Smith   PetscReal *ar;
4329308dcc3eSBarry Smith 
4330308dcc3eSBarry Smith   PetscFunctionBegin;
4331308dcc3eSBarry Smith   mat = mxCreateDoubleMatrix(snes->conv_hist_len,1,mxREAL);
4332308dcc3eSBarry Smith   ar  = (PetscReal*) mxGetData(mat);
4333f5af7f23SKarl Rupp   for (i=0; i<snes->conv_hist_len; i++) ar[i] = snes->conv_hist[i];
4334308dcc3eSBarry Smith   PetscFunctionReturn(mat);
4335308dcc3eSBarry Smith }
4336308dcc3eSBarry Smith #endif
4337308dcc3eSBarry Smith 
43380c4c9dddSBarry Smith /*@C
4339758f92a0SBarry Smith    SNESGetConvergenceHistory - Gets the array used to hold the convergence history.
4340758f92a0SBarry Smith 
43413f9fe445SBarry Smith    Not Collective
4342758f92a0SBarry Smith 
4343758f92a0SBarry Smith    Input Parameter:
4344758f92a0SBarry Smith .  snes - iterative context obtained from SNESCreate()
4345758f92a0SBarry Smith 
4346758f92a0SBarry Smith    Output Parameters:
4347a2b725a8SWilliam Gropp +  a   - array to hold history
4348758f92a0SBarry Smith .  its - integer array holds the number of linear iterations (or
4349758f92a0SBarry Smith          negative if not converged) for each solve.
4350758f92a0SBarry Smith -  na  - size of a and its
4351758f92a0SBarry Smith 
4352758f92a0SBarry Smith    Notes:
4353758f92a0SBarry Smith     The calling sequence for this routine in Fortran is
4354758f92a0SBarry Smith $   call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr)
4355758f92a0SBarry Smith 
4356758f92a0SBarry Smith    This routine is useful, e.g., when running a code for purposes
4357758f92a0SBarry Smith    of accurate performance monitoring, when no I/O should be done
4358758f92a0SBarry Smith    during the section of code that is being timed.
4359758f92a0SBarry Smith 
4360758f92a0SBarry Smith    Level: intermediate
4361758f92a0SBarry Smith 
43625ea7661aSPierre Jolivet .seealso: SNESSetConvergenceHistory()
4363758f92a0SBarry Smith 
4364758f92a0SBarry Smith @*/
43657087cfbeSBarry Smith PetscErrorCode  SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na)
4366758f92a0SBarry Smith {
4367758f92a0SBarry Smith   PetscFunctionBegin;
43680700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4369758f92a0SBarry Smith   if (a)   *a   = snes->conv_hist;
4370758f92a0SBarry Smith   if (its) *its = snes->conv_hist_its;
4371758f92a0SBarry Smith   if (na)  *na  = snes->conv_hist_len;
43723a40ed3dSBarry Smith   PetscFunctionReturn(0);
4373c9005455SLois Curfman McInnes }
4374c9005455SLois Curfman McInnes 
4375ac226902SBarry Smith /*@C
437676b2cf59SMatthew Knepley   SNESSetUpdate - Sets the general-purpose update function called
4377eec8f646SJed Brown   at the beginning of every iteration of the nonlinear solve. Specifically
43787e4bb74cSBarry Smith   it is called just before the Jacobian is "evaluated".
437976b2cf59SMatthew Knepley 
43803f9fe445SBarry Smith   Logically Collective on SNES
438176b2cf59SMatthew Knepley 
438276b2cf59SMatthew Knepley   Input Parameters:
4383a2b725a8SWilliam Gropp + snes - The nonlinear solver context
4384a2b725a8SWilliam Gropp - func - The function
438576b2cf59SMatthew Knepley 
438676b2cf59SMatthew Knepley   Calling sequence of func:
4387a2b725a8SWilliam Gropp $ func (SNES snes, PetscInt step);
438876b2cf59SMatthew Knepley 
438976b2cf59SMatthew Knepley . step - The current step of the iteration
439076b2cf59SMatthew Knepley 
4391fe97e370SBarry Smith   Level: advanced
4392fe97e370SBarry Smith 
4393fe97e370SBarry 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()
4394fe97e370SBarry Smith         This is not used by most users.
439576b2cf59SMatthew Knepley 
43968d359177SBarry Smith .seealso SNESSetJacobian(), SNESSolve()
439776b2cf59SMatthew Knepley @*/
43987087cfbeSBarry Smith PetscErrorCode  SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt))
439976b2cf59SMatthew Knepley {
440076b2cf59SMatthew Knepley   PetscFunctionBegin;
44010700a824SBarry Smith   PetscValidHeaderSpecific(snes, SNES_CLASSID,1);
4402e7788613SBarry Smith   snes->ops->update = func;
440376b2cf59SMatthew Knepley   PetscFunctionReturn(0);
440476b2cf59SMatthew Knepley }
440576b2cf59SMatthew Knepley 
44069b94acceSBarry Smith /*
44079b94acceSBarry Smith    SNESScaleStep_Private - Scales a step so that its length is less than the
44089b94acceSBarry Smith    positive parameter delta.
44099b94acceSBarry Smith 
44109b94acceSBarry Smith     Input Parameters:
4411c7afd0dbSLois Curfman McInnes +   snes - the SNES context
44129b94acceSBarry Smith .   y - approximate solution of linear system
44139b94acceSBarry Smith .   fnorm - 2-norm of current function
4414c7afd0dbSLois Curfman McInnes -   delta - trust region size
44159b94acceSBarry Smith 
44169b94acceSBarry Smith     Output Parameters:
4417c7afd0dbSLois Curfman McInnes +   gpnorm - predicted function norm at the new point, assuming local
44189b94acceSBarry Smith     linearization.  The value is zero if the step lies within the trust
44199b94acceSBarry Smith     region, and exceeds zero otherwise.
4420c7afd0dbSLois Curfman McInnes -   ynorm - 2-norm of the step
44219b94acceSBarry Smith 
44229b94acceSBarry Smith     Note:
442304d7464bSBarry Smith     For non-trust region methods such as SNESNEWTONLS, the parameter delta
44249b94acceSBarry Smith     is set to be the maximum allowable step size.
44259b94acceSBarry Smith 
44269b94acceSBarry Smith */
4427dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm)
44289b94acceSBarry Smith {
4429064f8208SBarry Smith   PetscReal      nrm;
4430ea709b57SSatish Balay   PetscScalar    cnorm;
4431dfbe8321SBarry Smith   PetscErrorCode ierr;
44323a40ed3dSBarry Smith 
44333a40ed3dSBarry Smith   PetscFunctionBegin;
44340700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
44350700a824SBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,2);
4436c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,2);
4437184914b5SBarry Smith 
4438064f8208SBarry Smith   ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr);
4439064f8208SBarry Smith   if (nrm > *delta) {
4440064f8208SBarry Smith     nrm     = *delta/nrm;
4441064f8208SBarry Smith     *gpnorm = (1.0 - nrm)*(*fnorm);
4442064f8208SBarry Smith     cnorm   = nrm;
44432dcb1b2aSMatthew Knepley     ierr    = VecScale(y,cnorm);CHKERRQ(ierr);
44449b94acceSBarry Smith     *ynorm  = *delta;
44459b94acceSBarry Smith   } else {
44469b94acceSBarry Smith     *gpnorm = 0.0;
4447064f8208SBarry Smith     *ynorm  = nrm;
44489b94acceSBarry Smith   }
44493a40ed3dSBarry Smith   PetscFunctionReturn(0);
44509b94acceSBarry Smith }
44519b94acceSBarry Smith 
445291f3e32bSBarry Smith /*@C
445319a666eeSBarry Smith    SNESConvergedReasonView - Displays the reason a SNES solve converged or diverged to a viewer
44542a359c20SBarry Smith 
44552a359c20SBarry Smith    Collective on SNES
44562a359c20SBarry Smith 
44572a359c20SBarry Smith    Parameter:
44582a359c20SBarry Smith +  snes - iterative context obtained from SNESCreate()
44592a359c20SBarry Smith -  viewer - the viewer to display the reason
44602a359c20SBarry Smith 
44612a359c20SBarry Smith 
44622a359c20SBarry Smith    Options Database Keys:
4463ee300463SSatish Balay +  -snes_converged_reason - print reason for converged or diverged, also prints number of iterations
4464ee300463SSatish Balay -  -snes_converged_reason ::failed - only print reason and number of iterations when diverged
4465eafd5ff0SAlex Lindsay 
446619a666eeSBarry Smith   Notes:
446719a666eeSBarry Smith      To change the format of the output call PetscViewerPushFormat(viewer,format) before this call. Use PETSC_VIEWER_DEFAULT for the default,
446819a666eeSBarry Smith      use PETSC_VIEWER_FAILED to only display a reason if it fails.
44692a359c20SBarry Smith 
44702a359c20SBarry Smith    Level: beginner
44712a359c20SBarry Smith 
447219a666eeSBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault(), SNESGetConvergedReason(), SNESConvergedReasonViewFromOptions(),
447319a666eeSBarry Smith           PetscViewerPushFormat(), PetscViewerPopFormat()
44742a359c20SBarry Smith 
44752a359c20SBarry Smith @*/
447619a666eeSBarry Smith PetscErrorCode  SNESConvergedReasonView(SNES snes,PetscViewer viewer)
44772a359c20SBarry Smith {
447875cca76cSMatthew G. Knepley   PetscViewerFormat format;
44792a359c20SBarry Smith   PetscBool         isAscii;
448075cca76cSMatthew G. Knepley   PetscErrorCode    ierr;
44812a359c20SBarry Smith 
44822a359c20SBarry Smith   PetscFunctionBegin;
448319a666eeSBarry Smith   if (!viewer) viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes));
44842a359c20SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isAscii);CHKERRQ(ierr);
44852a359c20SBarry Smith   if (isAscii) {
448675cca76cSMatthew G. Knepley     ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
44872a359c20SBarry Smith     ierr = PetscViewerASCIIAddTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr);
448875cca76cSMatthew G. Knepley     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
448975cca76cSMatthew G. Knepley       DM              dm;
449075cca76cSMatthew G. Knepley       Vec             u;
449175cca76cSMatthew G. Knepley       PetscDS         prob;
449275cca76cSMatthew G. Knepley       PetscInt        Nf, f;
449395cbbfd3SMatthew G. Knepley       PetscErrorCode (**exactSol)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar[], void *);
449495cbbfd3SMatthew G. Knepley       void            **exactCtx;
449575cca76cSMatthew G. Knepley       PetscReal       error;
449675cca76cSMatthew G. Knepley 
449775cca76cSMatthew G. Knepley       ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
449875cca76cSMatthew G. Knepley       ierr = SNESGetSolution(snes, &u);CHKERRQ(ierr);
449975cca76cSMatthew G. Knepley       ierr = DMGetDS(dm, &prob);CHKERRQ(ierr);
450075cca76cSMatthew G. Knepley       ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr);
450195cbbfd3SMatthew G. Knepley       ierr = PetscMalloc2(Nf, &exactSol, Nf, &exactCtx);CHKERRQ(ierr);
450295cbbfd3SMatthew G. Knepley       for (f = 0; f < Nf; ++f) {ierr = PetscDSGetExactSolution(prob, f, &exactSol[f], &exactCtx[f]);CHKERRQ(ierr);}
450395cbbfd3SMatthew G. Knepley       ierr = DMComputeL2Diff(dm, 0.0, exactSol, exactCtx, u, &error);CHKERRQ(ierr);
450495cbbfd3SMatthew G. Knepley       ierr = PetscFree2(exactSol, exactCtx);CHKERRQ(ierr);
450575cca76cSMatthew G. Knepley       if (error < 1.0e-11) {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: < 1.0e-11\n");CHKERRQ(ierr);}
450675cca76cSMatthew G. Knepley       else                 {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: %g\n", error);CHKERRQ(ierr);}
450775cca76cSMatthew G. Knepley     }
4508eafd5ff0SAlex Lindsay     if (snes->reason > 0 && format != PETSC_VIEWER_FAILED) {
45092a359c20SBarry Smith       if (((PetscObject) snes)->prefix) {
45102a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve converged due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
45112a359c20SBarry Smith       } else {
45122a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve converged due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
45132a359c20SBarry Smith       }
4514eafd5ff0SAlex Lindsay     } else if (snes->reason <= 0) {
45152a359c20SBarry Smith       if (((PetscObject) snes)->prefix) {
45162a359c20SBarry 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);
45172a359c20SBarry Smith       } else {
45182a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve did not converge due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
45192a359c20SBarry Smith       }
45202a359c20SBarry Smith     }
45212a359c20SBarry Smith     ierr = PetscViewerASCIISubtractTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr);
45222a359c20SBarry Smith   }
45232a359c20SBarry Smith   PetscFunctionReturn(0);
45242a359c20SBarry Smith }
45252a359c20SBarry Smith 
4526c4421ceaSFande Kong /*@C
4527c4421ceaSFande Kong    SNESConvergedReasonViewSet - Sets an ADDITIONAL function that is to be used at the
4528c4421ceaSFande Kong     end of the nonlinear solver to display the conver reason of the nonlinear solver.
4529c4421ceaSFande Kong 
4530c4421ceaSFande Kong    Logically Collective on SNES
4531c4421ceaSFande Kong 
4532c4421ceaSFande Kong    Input Parameters:
4533c4421ceaSFande Kong +  snes - the SNES context
4534c4421ceaSFande Kong .  f - the snes converged reason view function
4535c4421ceaSFande Kong .  vctx - [optional] user-defined context for private data for the
4536c4421ceaSFande Kong           snes converged reason view routine (use NULL if no context is desired)
4537c4421ceaSFande Kong -  reasonviewdestroy - [optional] routine that frees reasonview context
4538c4421ceaSFande Kong           (may be NULL)
4539c4421ceaSFande Kong 
4540c4421ceaSFande Kong    Options Database Keys:
4541c4421ceaSFande Kong +    -snes_converged_reason        - sets a default SNESConvergedReasonView()
4542c4421ceaSFande Kong -    -snes_converged_reason_view_cancel - cancels all converged reason viewers that have
4543c4421ceaSFande Kong                             been hardwired into a code by
4544c4421ceaSFande Kong                             calls to SNESConvergedReasonViewSet(), but
4545c4421ceaSFande Kong                             does not cancel those set via
4546c4421ceaSFande Kong                             the options database.
4547c4421ceaSFande Kong 
4548c4421ceaSFande Kong    Notes:
4549c4421ceaSFande Kong    Several different converged reason view routines may be set by calling
4550c4421ceaSFande Kong    SNESConvergedReasonViewSet() multiple times; all will be called in the
4551c4421ceaSFande Kong    order in which they were set.
4552c4421ceaSFande Kong 
4553c4421ceaSFande Kong    Level: intermediate
4554c4421ceaSFande Kong 
4555c4421ceaSFande Kong .seealso: SNESConvergedReasonView(), SNESConvergedReasonViewCancel()
4556c4421ceaSFande Kong @*/
4557c4421ceaSFande Kong PetscErrorCode  SNESConvergedReasonViewSet(SNES snes,PetscErrorCode (*f)(SNES,void*),void *vctx,PetscErrorCode (*reasonviewdestroy)(void**))
4558c4421ceaSFande Kong {
4559c4421ceaSFande Kong   PetscInt       i;
4560c4421ceaSFande Kong   PetscErrorCode ierr;
4561c4421ceaSFande Kong   PetscBool      identical;
4562c4421ceaSFande Kong 
4563c4421ceaSFande Kong   PetscFunctionBegin;
4564c4421ceaSFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4565c4421ceaSFande Kong   for (i=0; i<snes->numberreasonviews;i++) {
4566c4421ceaSFande Kong     ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,vctx,reasonviewdestroy,(PetscErrorCode (*)(void))snes->reasonview[i],snes->reasonviewcontext[i],snes->reasonviewdestroy[i],&identical);CHKERRQ(ierr);
4567c4421ceaSFande Kong     if (identical) PetscFunctionReturn(0);
4568c4421ceaSFande Kong   }
4569c4421ceaSFande Kong   if (snes->numberreasonviews >= MAXSNESREASONVIEWS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many SNES reasonview set");
4570c4421ceaSFande Kong   snes->reasonview[snes->numberreasonviews]          = f;
4571c4421ceaSFande Kong   snes->reasonviewdestroy[snes->numberreasonviews]   = reasonviewdestroy;
4572c4421ceaSFande Kong   snes->reasonviewcontext[snes->numberreasonviews++] = (void*)vctx;
4573c4421ceaSFande Kong   PetscFunctionReturn(0);
4574c4421ceaSFande Kong }
4575c4421ceaSFande Kong 
457691f3e32bSBarry Smith /*@
457719a666eeSBarry Smith   SNESConvergedReasonViewFromOptions - Processes command line options to determine if/how a SNESReason is to be viewed.
4578c4421ceaSFande Kong                                        All the user-provided convergedReasonView routines will be involved as well, if they exist.
45792a359c20SBarry Smith 
45802a359c20SBarry Smith   Collective on SNES
45812a359c20SBarry Smith 
45822a359c20SBarry Smith   Input Parameters:
45832a359c20SBarry Smith . snes   - the SNES object
45842a359c20SBarry Smith 
45852a359c20SBarry Smith   Level: intermediate
45862a359c20SBarry Smith 
458719a666eeSBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault(), SNESGetConvergedReason(), SNESConvergedReasonView()
458819a666eeSBarry Smith 
45892a359c20SBarry Smith @*/
459019a666eeSBarry Smith PetscErrorCode SNESConvergedReasonViewFromOptions(SNES snes)
45912a359c20SBarry Smith {
45922a359c20SBarry Smith   PetscErrorCode    ierr;
45932a359c20SBarry Smith   PetscViewer       viewer;
45942a359c20SBarry Smith   PetscBool         flg;
45952a359c20SBarry Smith   static PetscBool  incall = PETSC_FALSE;
45962a359c20SBarry Smith   PetscViewerFormat format;
4597c4421ceaSFande Kong   PetscInt          i;
45982a359c20SBarry Smith 
45992a359c20SBarry Smith   PetscFunctionBegin;
46002a359c20SBarry Smith   if (incall) PetscFunctionReturn(0);
46012a359c20SBarry Smith   incall = PETSC_TRUE;
4602c4421ceaSFande Kong 
4603c4421ceaSFande Kong   /* All user-provided viewers are called first, if they exist. */
4604c4421ceaSFande Kong   for (i=0; i<snes->numberreasonviews; i++) {
4605c4421ceaSFande Kong     ierr = (*snes->reasonview[i])(snes,snes->reasonviewcontext[i]);CHKERRQ(ierr);
4606c4421ceaSFande Kong   }
4607c4421ceaSFande Kong 
4608c4421ceaSFande Kong   /* Call PETSc default routine if users ask for it */
460916413a6aSBarry Smith   ierr   = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_converged_reason",&viewer,&format,&flg);CHKERRQ(ierr);
46102a359c20SBarry Smith   if (flg) {
46112a359c20SBarry Smith     ierr = PetscViewerPushFormat(viewer,format);CHKERRQ(ierr);
461219a666eeSBarry Smith     ierr = SNESConvergedReasonView(snes,viewer);CHKERRQ(ierr);
46132a359c20SBarry Smith     ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
46142a359c20SBarry Smith     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
46152a359c20SBarry Smith   }
46162a359c20SBarry Smith   incall = PETSC_FALSE;
46172a359c20SBarry Smith   PetscFunctionReturn(0);
46182a359c20SBarry Smith }
46192a359c20SBarry Smith 
4620487a658cSBarry Smith /*@
4621f69a0ea3SMatthew Knepley    SNESSolve - Solves a nonlinear system F(x) = b.
4622f69a0ea3SMatthew Knepley    Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX().
46239b94acceSBarry Smith 
4624c7afd0dbSLois Curfman McInnes    Collective on SNES
4625c7afd0dbSLois Curfman McInnes 
4626b2002411SLois Curfman McInnes    Input Parameters:
4627c7afd0dbSLois Curfman McInnes +  snes - the SNES context
46280298fd71SBarry Smith .  b - the constant part of the equation F(x) = b, or NULL to use zero.
462985385478SLisandro Dalcin -  x - the solution vector.
46309b94acceSBarry Smith 
4631b2002411SLois Curfman McInnes    Notes:
46328ddd3da0SLois Curfman McInnes    The user should initialize the vector,x, with the initial guess
46338ddd3da0SLois Curfman McInnes    for the nonlinear solve prior to calling SNESSolve.  In particular,
46348ddd3da0SLois Curfman McInnes    to employ an initial guess of zero, the user should explicitly set
46358ddd3da0SLois Curfman McInnes    this vector to zero by calling VecSet().
46368ddd3da0SLois Curfman McInnes 
463736851e7fSLois Curfman McInnes    Level: beginner
463836851e7fSLois Curfman McInnes 
4639c0df2a02SJed Brown .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetGridSequence(), SNESGetSolution()
46409b94acceSBarry Smith @*/
46417087cfbeSBarry Smith PetscErrorCode  SNESSolve(SNES snes,Vec b,Vec x)
46429b94acceSBarry Smith {
4643dfbe8321SBarry Smith   PetscErrorCode    ierr;
4644ace3abfcSBarry Smith   PetscBool         flg;
4645efd51863SBarry Smith   PetscInt          grid;
46460298fd71SBarry Smith   Vec               xcreated = NULL;
4647caa4e7f2SJed Brown   DM                dm;
4648052efed2SBarry Smith 
46493a40ed3dSBarry Smith   PetscFunctionBegin;
46500700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4651a69afd8bSBarry Smith   if (x) PetscValidHeaderSpecific(x,VEC_CLASSID,3);
4652a69afd8bSBarry Smith   if (x) PetscCheckSameComm(snes,1,x,3);
46530700a824SBarry Smith   if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2);
465485385478SLisandro Dalcin   if (b) PetscCheckSameComm(snes,1,b,2);
465585385478SLisandro Dalcin 
465634b4d3a8SMatthew G. Knepley   /* High level operations using the nonlinear solver */
465706fc46c8SMatthew G. Knepley   {
465806fc46c8SMatthew G. Knepley     PetscViewer       viewer;
465906fc46c8SMatthew G. Knepley     PetscViewerFormat format;
46607c88af5aSMatthew G. Knepley     PetscInt          num;
466106fc46c8SMatthew G. Knepley     PetscBool         flg;
466206fc46c8SMatthew G. Knepley     static PetscBool  incall = PETSC_FALSE;
466306fc46c8SMatthew G. Knepley 
466406fc46c8SMatthew G. Knepley     if (!incall) {
466534b4d3a8SMatthew G. Knepley       /* Estimate the convergence rate of the discretization */
466616413a6aSBarry Smith       ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) snes),((PetscObject)snes)->options, ((PetscObject) snes)->prefix, "-snes_convergence_estimate", &viewer, &format, &flg);CHKERRQ(ierr);
466706fc46c8SMatthew G. Knepley       if (flg) {
466806fc46c8SMatthew G. Knepley         PetscConvEst conv;
466946079b62SMatthew G. Knepley         DM           dm;
467046079b62SMatthew G. Knepley         PetscReal   *alpha; /* Convergence rate of the solution error for each field in the L_2 norm */
467146079b62SMatthew G. Knepley         PetscInt     Nf;
467206fc46c8SMatthew G. Knepley 
467306fc46c8SMatthew G. Knepley         incall = PETSC_TRUE;
467446079b62SMatthew G. Knepley         ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
467546079b62SMatthew G. Knepley         ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr);
4676ca5a3622SMatthew G. Knepley         ierr = PetscCalloc1(Nf, &alpha);CHKERRQ(ierr);
467706fc46c8SMatthew G. Knepley         ierr = PetscConvEstCreate(PetscObjectComm((PetscObject) snes), &conv);CHKERRQ(ierr);
4678900f6b5bSMatthew G. Knepley         ierr = PetscConvEstSetSolver(conv, (PetscObject) snes);CHKERRQ(ierr);
467906fc46c8SMatthew G. Knepley         ierr = PetscConvEstSetFromOptions(conv);CHKERRQ(ierr);
46800955ed61SMatthew G. Knepley         ierr = PetscConvEstSetUp(conv);CHKERRQ(ierr);
468146079b62SMatthew G. Knepley         ierr = PetscConvEstGetConvRate(conv, alpha);CHKERRQ(ierr);
468206fc46c8SMatthew G. Knepley         ierr = PetscViewerPushFormat(viewer, format);CHKERRQ(ierr);
468306fc46c8SMatthew G. Knepley         ierr = PetscConvEstRateView(conv, alpha, viewer);CHKERRQ(ierr);
468406fc46c8SMatthew G. Knepley         ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
468506fc46c8SMatthew G. Knepley         ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
468606fc46c8SMatthew G. Knepley         ierr = PetscConvEstDestroy(&conv);CHKERRQ(ierr);
468746079b62SMatthew G. Knepley         ierr = PetscFree(alpha);CHKERRQ(ierr);
468806fc46c8SMatthew G. Knepley         incall = PETSC_FALSE;
468906fc46c8SMatthew G. Knepley       }
469034b4d3a8SMatthew G. Knepley       /* Adaptively refine the initial grid */
4691b2588ea6SMatthew G. Knepley       num  = 1;
4692b2588ea6SMatthew G. Knepley       ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_initial", &num, &flg);CHKERRQ(ierr);
469334b4d3a8SMatthew G. Knepley       if (flg) {
469434b4d3a8SMatthew G. Knepley         DMAdaptor adaptor;
469534b4d3a8SMatthew G. Knepley 
469634b4d3a8SMatthew G. Knepley         incall = PETSC_TRUE;
4697ea13f565SStefano Zampini         ierr = DMAdaptorCreate(PetscObjectComm((PetscObject)snes), &adaptor);CHKERRQ(ierr);
469834b4d3a8SMatthew G. Knepley         ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr);
4699b2588ea6SMatthew G. Knepley         ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr);
470034b4d3a8SMatthew G. Knepley         ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr);
470134b4d3a8SMatthew G. Knepley         ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr);
470234b4d3a8SMatthew G. Knepley         ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_INITIAL, &dm, &x);CHKERRQ(ierr);
470334b4d3a8SMatthew G. Knepley         ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr);
470434b4d3a8SMatthew G. Knepley         incall = PETSC_FALSE;
470534b4d3a8SMatthew G. Knepley       }
47067c88af5aSMatthew G. Knepley       /* Use grid sequencing to adapt */
47077c88af5aSMatthew G. Knepley       num  = 0;
47087c88af5aSMatthew G. Knepley       ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_sequence", &num, NULL);CHKERRQ(ierr);
47097c88af5aSMatthew G. Knepley       if (num) {
47107c88af5aSMatthew G. Knepley         DMAdaptor adaptor;
47117c88af5aSMatthew G. Knepley 
47127c88af5aSMatthew G. Knepley         incall = PETSC_TRUE;
4713ea13f565SStefano Zampini         ierr = DMAdaptorCreate(PetscObjectComm((PetscObject)snes), &adaptor);CHKERRQ(ierr);
47147c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr);
47157c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr);
47167c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr);
47177c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr);
47187c88af5aSMatthew G. Knepley         ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_SEQUENTIAL, &dm, &x);CHKERRQ(ierr);
47197c88af5aSMatthew G. Knepley         ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr);
47207c88af5aSMatthew G. Knepley         incall = PETSC_FALSE;
47217c88af5aSMatthew G. Knepley       }
472206fc46c8SMatthew G. Knepley     }
472306fc46c8SMatthew G. Knepley   }
4724caa4e7f2SJed Brown   if (!x) {
4725caa4e7f2SJed Brown     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4726caa4e7f2SJed Brown     ierr = DMCreateGlobalVector(dm,&xcreated);CHKERRQ(ierr);
4727a69afd8bSBarry Smith     x    = xcreated;
4728a69afd8bSBarry Smith   }
4729ce1779c8SBarry Smith   ierr = SNESViewFromOptions(snes,NULL,"-snes_view_pre");CHKERRQ(ierr);
4730f05ece33SBarry Smith 
4731ce94432eSBarry Smith   for (grid=0; grid<snes->gridsequence; grid++) {ierr = PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);}
4732efd51863SBarry Smith   for (grid=0; grid<snes->gridsequence+1; grid++) {
4733efd51863SBarry Smith 
473485385478SLisandro Dalcin     /* set solution vector */
4735efd51863SBarry Smith     if (!grid) {ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);}
47366bf464f9SBarry Smith     ierr          = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
473785385478SLisandro Dalcin     snes->vec_sol = x;
4738caa4e7f2SJed Brown     ierr          = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4739caa4e7f2SJed Brown 
4740caa4e7f2SJed Brown     /* set affine vector if provided */
474185385478SLisandro Dalcin     if (b) { ierr = PetscObjectReference((PetscObject)b);CHKERRQ(ierr); }
47426bf464f9SBarry Smith     ierr          = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
474385385478SLisandro Dalcin     snes->vec_rhs = b;
474485385478SLisandro Dalcin 
4745bfdd6862SPatrick 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");
4746154060b5SMatthew G. Knepley     if (snes->vec_func == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector");
4747154060b5SMatthew 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");
4748154060b5SMatthew G. Knepley     if (!snes->vec_sol_update /* && snes->vec_sol */) {
4749154060b5SMatthew G. Knepley       ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr);
4750154060b5SMatthew G. Knepley       ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->vec_sol_update);CHKERRQ(ierr);
4751154060b5SMatthew G. Knepley     }
4752154060b5SMatthew G. Knepley     ierr = DMShellSetGlobalVector(dm,snes->vec_sol);CHKERRQ(ierr);
475370e92668SMatthew Knepley     ierr = SNESSetUp(snes);CHKERRQ(ierr);
47543f149594SLisandro Dalcin 
47557eee914bSBarry Smith     if (!grid) {
47567eee914bSBarry Smith       if (snes->ops->computeinitialguess) {
4757d25893d9SBarry Smith         ierr = (*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP);CHKERRQ(ierr);
4758d25893d9SBarry Smith       }
4759dd568438SSatish Balay     }
4760d25893d9SBarry Smith 
4761abc0a331SBarry Smith     if (snes->conv_hist_reset) snes->conv_hist_len = 0;
4762971e163fSPeter Brune     if (snes->counters_reset) {snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;}
4763d5e45103SBarry Smith 
47643f149594SLisandro Dalcin     ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
47654936397dSBarry Smith     ierr = (*snes->ops->solve)(snes);CHKERRQ(ierr);
476685385478SLisandro Dalcin     ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
476717186662SBarry Smith     if (!snes->reason) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason");
4768422a814eSBarry Smith     snes->domainerror = PETSC_FALSE; /* clear the flag if it has been set */
47693f149594SLisandro Dalcin 
477037ec4e1aSPeter Brune     if (snes->lagjac_persist) snes->jac_iter += snes->iter;
477137ec4e1aSPeter Brune     if (snes->lagpre_persist) snes->pre_iter += snes->iter;
477237ec4e1aSPeter Brune 
477316413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_test_local_min",NULL,NULL,&flg);CHKERRQ(ierr);
4774da9b6338SBarry Smith     if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); }
4775c4421ceaSFande Kong     /* Call converged reason views. This may involve user-provided viewers as well */
477619a666eeSBarry Smith     ierr = SNESConvergedReasonViewFromOptions(snes);CHKERRQ(ierr);
47775968eb51SBarry Smith 
4778ce94432eSBarry Smith     if (snes->errorifnotconverged && snes->reason < 0) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged");
47799c8e83a9SBarry Smith     if (snes->reason < 0) break;
4780efd51863SBarry Smith     if (grid <  snes->gridsequence) {
4781efd51863SBarry Smith       DM  fine;
4782efd51863SBarry Smith       Vec xnew;
4783efd51863SBarry Smith       Mat interp;
4784efd51863SBarry Smith 
4785ce94432eSBarry Smith       ierr = DMRefine(snes->dm,PetscObjectComm((PetscObject)snes),&fine);CHKERRQ(ierr);
4786ce94432eSBarry Smith       if (!fine) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"DMRefine() did not perform any refinement, cannot continue grid sequencing");
47870298fd71SBarry Smith       ierr = DMCreateInterpolation(snes->dm,fine,&interp,NULL);CHKERRQ(ierr);
4788efd51863SBarry Smith       ierr = DMCreateGlobalVector(fine,&xnew);CHKERRQ(ierr);
4789efd51863SBarry Smith       ierr = MatInterpolate(interp,x,xnew);CHKERRQ(ierr);
4790c833c3b5SJed Brown       ierr = DMInterpolate(snes->dm,interp,fine);CHKERRQ(ierr);
4791efd51863SBarry Smith       ierr = MatDestroy(&interp);CHKERRQ(ierr);
4792efd51863SBarry Smith       x    = xnew;
4793efd51863SBarry Smith 
4794efd51863SBarry Smith       ierr = SNESReset(snes);CHKERRQ(ierr);
4795efd51863SBarry Smith       ierr = SNESSetDM(snes,fine);CHKERRQ(ierr);
4796352f405dSMatthew G. Knepley       ierr = SNESResetFromOptions(snes);CHKERRQ(ierr);
4797efd51863SBarry Smith       ierr = DMDestroy(&fine);CHKERRQ(ierr);
4798ce94432eSBarry Smith       ierr = PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);
4799efd51863SBarry Smith     }
4800efd51863SBarry Smith   }
4801ce1779c8SBarry Smith   ierr = SNESViewFromOptions(snes,NULL,"-snes_view");CHKERRQ(ierr);
4802685405a1SBarry Smith   ierr = VecViewFromOptions(snes->vec_sol,(PetscObject)snes,"-snes_view_solution");CHKERRQ(ierr);
4803c0f0dcc3SMatthew G. Knepley   ierr = DMMonitor(snes->dm);CHKERRQ(ierr);
48043f7e2da0SPeter Brune 
4805a69afd8bSBarry Smith   ierr = VecDestroy(&xcreated);CHKERRQ(ierr);
4806e04113cfSBarry Smith   ierr = PetscObjectSAWsBlock((PetscObject)snes);CHKERRQ(ierr);
48073a40ed3dSBarry Smith   PetscFunctionReturn(0);
48089b94acceSBarry Smith }
48099b94acceSBarry Smith 
48109b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */
48119b94acceSBarry Smith 
481282bf6240SBarry Smith /*@C
48134b0e389bSBarry Smith    SNESSetType - Sets the method for the nonlinear solver.
48149b94acceSBarry Smith 
4815fee21e36SBarry Smith    Collective on SNES
4816fee21e36SBarry Smith 
4817c7afd0dbSLois Curfman McInnes    Input Parameters:
4818c7afd0dbSLois Curfman McInnes +  snes - the SNES context
4819454a90a3SBarry Smith -  type - a known method
4820c7afd0dbSLois Curfman McInnes 
4821c7afd0dbSLois Curfman McInnes    Options Database Key:
4822454a90a3SBarry Smith .  -snes_type <type> - Sets the method; use -help for a list
482304d7464bSBarry Smith    of available methods (for instance, newtonls or newtontr)
4824ae12b187SLois Curfman McInnes 
48259b94acceSBarry Smith    Notes:
4826e090d566SSatish Balay    See "petsc/include/petscsnes.h" for available methods (for instance)
482704d7464bSBarry Smith +    SNESNEWTONLS - Newton's method with line search
4828c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
4829a2b725a8SWilliam Gropp -    SNESNEWTONTR - Newton's method with trust region
4830c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
48319b94acceSBarry Smith 
4832ae12b187SLois Curfman McInnes   Normally, it is best to use the SNESSetFromOptions() command and then
4833ae12b187SLois Curfman McInnes   set the SNES solver type from the options database rather than by using
4834ae12b187SLois Curfman McInnes   this routine.  Using the options database provides the user with
4835ae12b187SLois Curfman McInnes   maximum flexibility in evaluating the many nonlinear solvers.
4836ae12b187SLois Curfman McInnes   The SNESSetType() routine is provided for those situations where it
4837ae12b187SLois Curfman McInnes   is necessary to set the nonlinear solver independently of the command
4838ae12b187SLois Curfman McInnes   line or options database.  This might be the case, for example, when
4839ae12b187SLois Curfman McInnes   the choice of solver changes during the execution of the program,
4840ae12b187SLois Curfman McInnes   and the user's application is taking responsibility for choosing the
4841b0a32e0cSBarry Smith   appropriate method.
484236851e7fSLois Curfman McInnes 
484395452b02SPatrick Sanan     Developer Notes:
484495452b02SPatrick Sanan     SNESRegister() adds a constructor for a new SNESType to SNESList, SNESSetType() locates
48458f6c3df8SBarry Smith     the constructor in that list and calls it to create the spexific object.
48468f6c3df8SBarry Smith 
484736851e7fSLois Curfman McInnes   Level: intermediate
4848a703fe33SLois Curfman McInnes 
48498f6c3df8SBarry Smith .seealso: SNESType, SNESCreate(), SNESDestroy(), SNESGetType(), SNESSetFromOptions()
4850435da068SBarry Smith 
48519b94acceSBarry Smith @*/
485219fd82e9SBarry Smith PetscErrorCode  SNESSetType(SNES snes,SNESType type)
48539b94acceSBarry Smith {
4854dfbe8321SBarry Smith   PetscErrorCode ierr,(*r)(SNES);
4855ace3abfcSBarry Smith   PetscBool      match;
48563a40ed3dSBarry Smith 
48573a40ed3dSBarry Smith   PetscFunctionBegin;
48580700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
48594482741eSBarry Smith   PetscValidCharPointer(type,2);
486082bf6240SBarry Smith 
4861251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr);
48620f5bd95cSBarry Smith   if (match) PetscFunctionReturn(0);
486392ff6ae8SBarry Smith 
48641c9cd337SJed Brown   ierr = PetscFunctionListFind(SNESList,type,&r);CHKERRQ(ierr);
4865e32f2f54SBarry Smith   if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type);
486675396ef9SLisandro Dalcin   /* Destroy the previous private SNES context */
4867b5c23020SJed Brown   if (snes->ops->destroy) {
4868b5c23020SJed Brown     ierr               = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr);
48690298fd71SBarry Smith     snes->ops->destroy = NULL;
4870b5c23020SJed Brown   }
487175396ef9SLisandro Dalcin   /* Reinitialize function pointers in SNESOps structure */
48729e5d0892SLisandro Dalcin   snes->ops->setup          = NULL;
48739e5d0892SLisandro Dalcin   snes->ops->solve          = NULL;
48749e5d0892SLisandro Dalcin   snes->ops->view           = NULL;
48759e5d0892SLisandro Dalcin   snes->ops->setfromoptions = NULL;
48769e5d0892SLisandro Dalcin   snes->ops->destroy        = NULL;
48777fe760d5SStefano Zampini 
48787fe760d5SStefano Zampini   /* It may happen the user has customized the line search before calling SNESSetType */
48797fe760d5SStefano Zampini   if (((PetscObject)snes)->type_name) {
4880d8d34be6SBarry Smith     ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr);
48817fe760d5SStefano Zampini   }
48827fe760d5SStefano Zampini 
488375396ef9SLisandro Dalcin   /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */
488475396ef9SLisandro Dalcin   snes->setupcalled = PETSC_FALSE;
4885f5af7f23SKarl Rupp 
4886454a90a3SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr);
488703bfa161SLisandro Dalcin   ierr = (*r)(snes);CHKERRQ(ierr);
48883a40ed3dSBarry Smith   PetscFunctionReturn(0);
48899b94acceSBarry Smith }
48909b94acceSBarry Smith 
48919b94acceSBarry Smith /*@C
48929a28b0a6SLois Curfman McInnes    SNESGetType - Gets the SNES method type and name (as a string).
48939b94acceSBarry Smith 
4894c7afd0dbSLois Curfman McInnes    Not Collective
4895c7afd0dbSLois Curfman McInnes 
48969b94acceSBarry Smith    Input Parameter:
48974b0e389bSBarry Smith .  snes - nonlinear solver context
48989b94acceSBarry Smith 
48999b94acceSBarry Smith    Output Parameter:
49003a7fca6bSBarry Smith .  type - SNES method (a character string)
49019b94acceSBarry Smith 
490236851e7fSLois Curfman McInnes    Level: intermediate
490336851e7fSLois Curfman McInnes 
49049b94acceSBarry Smith @*/
490519fd82e9SBarry Smith PetscErrorCode  SNESGetType(SNES snes,SNESType *type)
49069b94acceSBarry Smith {
49073a40ed3dSBarry Smith   PetscFunctionBegin;
49080700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
49094482741eSBarry Smith   PetscValidPointer(type,2);
49107adad957SLisandro Dalcin   *type = ((PetscObject)snes)->type_name;
49113a40ed3dSBarry Smith   PetscFunctionReturn(0);
49129b94acceSBarry Smith }
49139b94acceSBarry Smith 
49143cd8a7caSMatthew G. Knepley /*@
49153cd8a7caSMatthew G. Knepley   SNESSetSolution - Sets the solution vector for use by the SNES routines.
49163cd8a7caSMatthew G. Knepley 
4917d083f849SBarry Smith   Logically Collective on SNES
49183cd8a7caSMatthew G. Knepley 
49193cd8a7caSMatthew G. Knepley   Input Parameters:
49203cd8a7caSMatthew G. Knepley + snes - the SNES context obtained from SNESCreate()
49213cd8a7caSMatthew G. Knepley - u    - the solution vector
49223cd8a7caSMatthew G. Knepley 
49233cd8a7caSMatthew G. Knepley   Level: beginner
49243cd8a7caSMatthew G. Knepley 
49253cd8a7caSMatthew G. Knepley @*/
49263cd8a7caSMatthew G. Knepley PetscErrorCode SNESSetSolution(SNES snes, Vec u)
49273cd8a7caSMatthew G. Knepley {
49283cd8a7caSMatthew G. Knepley   DM             dm;
49293cd8a7caSMatthew G. Knepley   PetscErrorCode ierr;
49303cd8a7caSMatthew G. Knepley 
49313cd8a7caSMatthew G. Knepley   PetscFunctionBegin;
49323cd8a7caSMatthew G. Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
49333cd8a7caSMatthew G. Knepley   PetscValidHeaderSpecific(u, VEC_CLASSID, 2);
49343cd8a7caSMatthew G. Knepley   ierr = PetscObjectReference((PetscObject) u);CHKERRQ(ierr);
49353cd8a7caSMatthew G. Knepley   ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
49363cd8a7caSMatthew G. Knepley 
49373cd8a7caSMatthew G. Knepley   snes->vec_sol = u;
49383cd8a7caSMatthew G. Knepley 
49393cd8a7caSMatthew G. Knepley   ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
49403cd8a7caSMatthew G. Knepley   ierr = DMShellSetGlobalVector(dm, u);CHKERRQ(ierr);
49413cd8a7caSMatthew G. Knepley   PetscFunctionReturn(0);
49423cd8a7caSMatthew G. Knepley }
49433cd8a7caSMatthew G. Knepley 
494452baeb72SSatish Balay /*@
49459b94acceSBarry Smith    SNESGetSolution - Returns the vector where the approximate solution is
4946c0df2a02SJed Brown    stored. This is the fine grid solution when using SNESSetGridSequence().
49479b94acceSBarry Smith 
4948c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
4949c7afd0dbSLois Curfman McInnes 
49509b94acceSBarry Smith    Input Parameter:
49519b94acceSBarry Smith .  snes - the SNES context
49529b94acceSBarry Smith 
49539b94acceSBarry Smith    Output Parameter:
49549b94acceSBarry Smith .  x - the solution
49559b94acceSBarry Smith 
495670e92668SMatthew Knepley    Level: intermediate
495736851e7fSLois Curfman McInnes 
495885385478SLisandro Dalcin .seealso:  SNESGetSolutionUpdate(), SNESGetFunction()
49599b94acceSBarry Smith @*/
49607087cfbeSBarry Smith PetscErrorCode  SNESGetSolution(SNES snes,Vec *x)
49619b94acceSBarry Smith {
49623a40ed3dSBarry Smith   PetscFunctionBegin;
49630700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
49644482741eSBarry Smith   PetscValidPointer(x,2);
496585385478SLisandro Dalcin   *x = snes->vec_sol;
496670e92668SMatthew Knepley   PetscFunctionReturn(0);
496770e92668SMatthew Knepley }
496870e92668SMatthew Knepley 
496952baeb72SSatish Balay /*@
49709b94acceSBarry Smith    SNESGetSolutionUpdate - Returns the vector where the solution update is
49719b94acceSBarry Smith    stored.
49729b94acceSBarry Smith 
4973c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
4974c7afd0dbSLois Curfman McInnes 
49759b94acceSBarry Smith    Input Parameter:
49769b94acceSBarry Smith .  snes - the SNES context
49779b94acceSBarry Smith 
49789b94acceSBarry Smith    Output Parameter:
49799b94acceSBarry Smith .  x - the solution update
49809b94acceSBarry Smith 
498136851e7fSLois Curfman McInnes    Level: advanced
498236851e7fSLois Curfman McInnes 
498385385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction()
49849b94acceSBarry Smith @*/
49857087cfbeSBarry Smith PetscErrorCode  SNESGetSolutionUpdate(SNES snes,Vec *x)
49869b94acceSBarry Smith {
49873a40ed3dSBarry Smith   PetscFunctionBegin;
49880700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
49894482741eSBarry Smith   PetscValidPointer(x,2);
499085385478SLisandro Dalcin   *x = snes->vec_sol_update;
49913a40ed3dSBarry Smith   PetscFunctionReturn(0);
49929b94acceSBarry Smith }
49939b94acceSBarry Smith 
49949b94acceSBarry Smith /*@C
49953638b69dSLois Curfman McInnes    SNESGetFunction - Returns the vector where the function is stored.
49969b94acceSBarry Smith 
4997a63bb30eSJed Brown    Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet.
4998c7afd0dbSLois Curfman McInnes 
49999b94acceSBarry Smith    Input Parameter:
50009b94acceSBarry Smith .  snes - the SNES context
50019b94acceSBarry Smith 
50029b94acceSBarry Smith    Output Parameter:
50030298fd71SBarry Smith +  r - the vector that is used to store residuals (or NULL if you don't want it)
5004f8b49ee9SBarry Smith .  f - the function (or NULL if you don't want it); see SNESFunction for calling sequence details
50050298fd71SBarry Smith -  ctx - the function context (or NULL if you don't want it)
50069b94acceSBarry Smith 
500736851e7fSLois Curfman McInnes    Level: advanced
500836851e7fSLois Curfman McInnes 
500904edfde5SBarry Smith     Notes: The vector r DOES NOT, in general contain the current value of the SNES nonlinear function
501004edfde5SBarry Smith 
5011bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetSolution(), SNESFunction
50129b94acceSBarry Smith @*/
5013f8b49ee9SBarry Smith PetscErrorCode  SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),void **ctx)
50149b94acceSBarry Smith {
5015a63bb30eSJed Brown   PetscErrorCode ierr;
50166cab3a1bSJed Brown   DM             dm;
5017a63bb30eSJed Brown 
50183a40ed3dSBarry Smith   PetscFunctionBegin;
50190700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5020a63bb30eSJed Brown   if (r) {
5021a63bb30eSJed Brown     if (!snes->vec_func) {
5022a63bb30eSJed Brown       if (snes->vec_rhs) {
5023a63bb30eSJed Brown         ierr = VecDuplicate(snes->vec_rhs,&snes->vec_func);CHKERRQ(ierr);
5024a63bb30eSJed Brown       } else if (snes->vec_sol) {
5025a63bb30eSJed Brown         ierr = VecDuplicate(snes->vec_sol,&snes->vec_func);CHKERRQ(ierr);
5026a63bb30eSJed Brown       } else if (snes->dm) {
5027a63bb30eSJed Brown         ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr);
5028a63bb30eSJed Brown       }
5029a63bb30eSJed Brown     }
5030a63bb30eSJed Brown     *r = snes->vec_func;
5031a63bb30eSJed Brown   }
50326cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
5033f8b49ee9SBarry Smith   ierr = DMSNESGetFunction(dm,f,ctx);CHKERRQ(ierr);
50343a40ed3dSBarry Smith   PetscFunctionReturn(0);
50359b94acceSBarry Smith }
50369b94acceSBarry Smith 
5037c79ef259SPeter Brune /*@C
5038be95d8f1SBarry Smith    SNESGetNGS - Returns the NGS function and context.
5039c79ef259SPeter Brune 
5040c79ef259SPeter Brune    Input Parameter:
5041c79ef259SPeter Brune .  snes - the SNES context
5042c79ef259SPeter Brune 
5043c79ef259SPeter Brune    Output Parameter:
5044be95d8f1SBarry Smith +  f - the function (or NULL) see SNESNGSFunction for details
50450298fd71SBarry Smith -  ctx    - the function context (or NULL)
5046c79ef259SPeter Brune 
5047c79ef259SPeter Brune    Level: advanced
5048c79ef259SPeter Brune 
5049be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetFunction()
5050c79ef259SPeter Brune @*/
5051c79ef259SPeter Brune 
5052be95d8f1SBarry Smith PetscErrorCode SNESGetNGS (SNES snes, PetscErrorCode (**f)(SNES, Vec, Vec, void*), void ** ctx)
5053646217ecSPeter Brune {
50546cab3a1bSJed Brown   PetscErrorCode ierr;
50556cab3a1bSJed Brown   DM             dm;
50566cab3a1bSJed Brown 
5057646217ecSPeter Brune   PetscFunctionBegin;
5058646217ecSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
50596cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
5060be95d8f1SBarry Smith   ierr = DMSNESGetNGS(dm,f,ctx);CHKERRQ(ierr);
5061646217ecSPeter Brune   PetscFunctionReturn(0);
5062646217ecSPeter Brune }
5063646217ecSPeter Brune 
50643c7409f5SSatish Balay /*@C
50653c7409f5SSatish Balay    SNESSetOptionsPrefix - Sets the prefix used for searching for all
5066d850072dSLois Curfman McInnes    SNES options in the database.
50673c7409f5SSatish Balay 
50683f9fe445SBarry Smith    Logically Collective on SNES
5069fee21e36SBarry Smith 
5070c7afd0dbSLois Curfman McInnes    Input Parameter:
5071c7afd0dbSLois Curfman McInnes +  snes - the SNES context
5072c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
5073c7afd0dbSLois Curfman McInnes 
5074d850072dSLois Curfman McInnes    Notes:
5075a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
5076c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
5077d850072dSLois Curfman McInnes 
507836851e7fSLois Curfman McInnes    Level: advanced
507936851e7fSLois Curfman McInnes 
5080a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions()
50813c7409f5SSatish Balay @*/
50827087cfbeSBarry Smith PetscErrorCode  SNESSetOptionsPrefix(SNES snes,const char prefix[])
50833c7409f5SSatish Balay {
5084dfbe8321SBarry Smith   PetscErrorCode ierr;
50853c7409f5SSatish Balay 
50863a40ed3dSBarry Smith   PetscFunctionBegin;
50870700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5088639f9d9dSBarry Smith   ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
50891cee3971SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
509035f5d045SPeter Brune   if (snes->linesearch) {
50917601faf0SJed Brown     ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr);
509208b6c495SPeter Brune     ierr = PetscObjectSetOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr);
509335f5d045SPeter Brune   }
509435f5d045SPeter Brune   ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
50953a40ed3dSBarry Smith   PetscFunctionReturn(0);
50963c7409f5SSatish Balay }
50973c7409f5SSatish Balay 
50983c7409f5SSatish Balay /*@C
5099f525115eSLois Curfman McInnes    SNESAppendOptionsPrefix - Appends to the prefix used for searching for all
5100d850072dSLois Curfman McInnes    SNES options in the database.
51013c7409f5SSatish Balay 
51023f9fe445SBarry Smith    Logically Collective on SNES
5103fee21e36SBarry Smith 
5104c7afd0dbSLois Curfman McInnes    Input Parameters:
5105c7afd0dbSLois Curfman McInnes +  snes - the SNES context
5106c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
5107c7afd0dbSLois Curfman McInnes 
5108d850072dSLois Curfman McInnes    Notes:
5109a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
5110c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
5111d850072dSLois Curfman McInnes 
511236851e7fSLois Curfman McInnes    Level: advanced
511336851e7fSLois Curfman McInnes 
5114a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix()
51153c7409f5SSatish Balay @*/
51167087cfbeSBarry Smith PetscErrorCode  SNESAppendOptionsPrefix(SNES snes,const char prefix[])
51173c7409f5SSatish Balay {
5118dfbe8321SBarry Smith   PetscErrorCode ierr;
51193c7409f5SSatish Balay 
51203a40ed3dSBarry Smith   PetscFunctionBegin;
51210700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5122639f9d9dSBarry Smith   ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
51231cee3971SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
512435f5d045SPeter Brune   if (snes->linesearch) {
51257601faf0SJed Brown     ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr);
512608b6c495SPeter Brune     ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr);
512735f5d045SPeter Brune   }
512835f5d045SPeter Brune   ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
51293a40ed3dSBarry Smith   PetscFunctionReturn(0);
51303c7409f5SSatish Balay }
51313c7409f5SSatish Balay 
51329ab63eb5SSatish Balay /*@C
51333c7409f5SSatish Balay    SNESGetOptionsPrefix - Sets the prefix used for searching for all
51343c7409f5SSatish Balay    SNES options in the database.
51353c7409f5SSatish Balay 
5136c7afd0dbSLois Curfman McInnes    Not Collective
5137c7afd0dbSLois Curfman McInnes 
51383c7409f5SSatish Balay    Input Parameter:
51393c7409f5SSatish Balay .  snes - the SNES context
51403c7409f5SSatish Balay 
51413c7409f5SSatish Balay    Output Parameter:
51423c7409f5SSatish Balay .  prefix - pointer to the prefix string used
51433c7409f5SSatish Balay 
514495452b02SPatrick Sanan    Notes:
514595452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
51469ab63eb5SSatish Balay    sufficient length to hold the prefix.
51479ab63eb5SSatish Balay 
514836851e7fSLois Curfman McInnes    Level: advanced
514936851e7fSLois Curfman McInnes 
5150a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix()
51513c7409f5SSatish Balay @*/
51527087cfbeSBarry Smith PetscErrorCode  SNESGetOptionsPrefix(SNES snes,const char *prefix[])
51533c7409f5SSatish Balay {
5154dfbe8321SBarry Smith   PetscErrorCode ierr;
51553c7409f5SSatish Balay 
51563a40ed3dSBarry Smith   PetscFunctionBegin;
51570700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5158639f9d9dSBarry Smith   ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
51593a40ed3dSBarry Smith   PetscFunctionReturn(0);
51603c7409f5SSatish Balay }
51613c7409f5SSatish Balay 
5162b2002411SLois Curfman McInnes 
51633cea93caSBarry Smith /*@C
51641c84c290SBarry Smith   SNESRegister - Adds a method to the nonlinear solver package.
51651c84c290SBarry Smith 
51661c84c290SBarry Smith    Not collective
51671c84c290SBarry Smith 
51681c84c290SBarry Smith    Input Parameters:
51691c84c290SBarry Smith +  name_solver - name of a new user-defined solver
51701c84c290SBarry Smith -  routine_create - routine to create method context
51711c84c290SBarry Smith 
51721c84c290SBarry Smith    Notes:
51731c84c290SBarry Smith    SNESRegister() may be called multiple times to add several user-defined solvers.
51741c84c290SBarry Smith 
51751c84c290SBarry Smith    Sample usage:
51761c84c290SBarry Smith .vb
5177bdf89e91SBarry Smith    SNESRegister("my_solver",MySolverCreate);
51781c84c290SBarry Smith .ve
51791c84c290SBarry Smith 
51801c84c290SBarry Smith    Then, your solver can be chosen with the procedural interface via
51811c84c290SBarry Smith $     SNESSetType(snes,"my_solver")
51821c84c290SBarry Smith    or at runtime via the option
51831c84c290SBarry Smith $     -snes_type my_solver
51841c84c290SBarry Smith 
51851c84c290SBarry Smith    Level: advanced
51861c84c290SBarry Smith 
51871c84c290SBarry Smith     Note: If your function is not being put into a shared library then use SNESRegister() instead
51881c84c290SBarry Smith 
51891c84c290SBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy()
51903cea93caSBarry Smith 
51917f6c08e0SMatthew Knepley   Level: advanced
51923cea93caSBarry Smith @*/
5193bdf89e91SBarry Smith PetscErrorCode  SNESRegister(const char sname[],PetscErrorCode (*function)(SNES))
5194b2002411SLois Curfman McInnes {
5195dfbe8321SBarry Smith   PetscErrorCode ierr;
5196b2002411SLois Curfman McInnes 
5197b2002411SLois Curfman McInnes   PetscFunctionBegin;
51981d36bdfdSBarry Smith   ierr = SNESInitializePackage();CHKERRQ(ierr);
5199a240a19fSJed Brown   ierr = PetscFunctionListAdd(&SNESList,sname,function);CHKERRQ(ierr);
5200b2002411SLois Curfman McInnes   PetscFunctionReturn(0);
5201b2002411SLois Curfman McInnes }
5202da9b6338SBarry Smith 
52037087cfbeSBarry Smith PetscErrorCode  SNESTestLocalMin(SNES snes)
5204da9b6338SBarry Smith {
5205dfbe8321SBarry Smith   PetscErrorCode ierr;
520677431f27SBarry Smith   PetscInt       N,i,j;
5207da9b6338SBarry Smith   Vec            u,uh,fh;
5208da9b6338SBarry Smith   PetscScalar    value;
5209da9b6338SBarry Smith   PetscReal      norm;
5210da9b6338SBarry Smith 
5211da9b6338SBarry Smith   PetscFunctionBegin;
5212da9b6338SBarry Smith   ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr);
5213da9b6338SBarry Smith   ierr = VecDuplicate(u,&uh);CHKERRQ(ierr);
5214da9b6338SBarry Smith   ierr = VecDuplicate(u,&fh);CHKERRQ(ierr);
5215da9b6338SBarry Smith 
5216da9b6338SBarry Smith   /* currently only works for sequential */
5217ea13f565SStefano Zampini   ierr = PetscPrintf(PetscObjectComm((PetscObject)snes),"Testing FormFunction() for local min\n");CHKERRQ(ierr);
5218da9b6338SBarry Smith   ierr = VecGetSize(u,&N);CHKERRQ(ierr);
5219da9b6338SBarry Smith   for (i=0; i<N; i++) {
5220da9b6338SBarry Smith     ierr = VecCopy(u,uh);CHKERRQ(ierr);
5221ea13f565SStefano Zampini     ierr = PetscPrintf(PetscObjectComm((PetscObject)snes),"i = %D\n",i);CHKERRQ(ierr);
5222da9b6338SBarry Smith     for (j=-10; j<11; j++) {
52238b49ba18SBarry Smith       value = PetscSign(j)*PetscExpReal(PetscAbs(j)-10.0);
5224da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
52253ab0aad5SBarry Smith       ierr  = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr);
5226da9b6338SBarry Smith       ierr  = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr);
5227ea13f565SStefano Zampini       ierr  = PetscPrintf(PetscObjectComm((PetscObject)snes),"       j norm %D %18.16e\n",j,norm);CHKERRQ(ierr);
5228da9b6338SBarry Smith       value = -value;
5229da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
5230da9b6338SBarry Smith     }
5231da9b6338SBarry Smith   }
52326bf464f9SBarry Smith   ierr = VecDestroy(&uh);CHKERRQ(ierr);
52336bf464f9SBarry Smith   ierr = VecDestroy(&fh);CHKERRQ(ierr);
5234da9b6338SBarry Smith   PetscFunctionReturn(0);
5235da9b6338SBarry Smith }
523671f87433Sdalcinl 
523771f87433Sdalcinl /*@
5238fa9f3622SBarry Smith    SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for
523971f87433Sdalcinl    computing relative tolerance for linear solvers within an inexact
524071f87433Sdalcinl    Newton method.
524171f87433Sdalcinl 
52423f9fe445SBarry Smith    Logically Collective on SNES
524371f87433Sdalcinl 
524471f87433Sdalcinl    Input Parameters:
524571f87433Sdalcinl +  snes - SNES context
524671f87433Sdalcinl -  flag - PETSC_TRUE or PETSC_FALSE
524771f87433Sdalcinl 
524864ba62caSBarry Smith     Options Database:
524964ba62caSBarry Smith +  -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence
525064ba62caSBarry Smith .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
525164ba62caSBarry Smith .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
525264ba62caSBarry Smith .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
525364ba62caSBarry Smith .  -snes_ksp_ew_gamma <gamma> - Sets gamma
525464ba62caSBarry Smith .  -snes_ksp_ew_alpha <alpha> - Sets alpha
525564ba62caSBarry Smith .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
525664ba62caSBarry Smith -  -snes_ksp_ew_threshold <threshold> - Sets threshold
525764ba62caSBarry Smith 
525871f87433Sdalcinl    Notes:
525971f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
526071f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
526171f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
526271f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
526371f87433Sdalcinl    solver.
526471f87433Sdalcinl 
526571f87433Sdalcinl    Level: advanced
526671f87433Sdalcinl 
526771f87433Sdalcinl    Reference:
526871f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
526971f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
527071f87433Sdalcinl 
5271fa9f3622SBarry Smith .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
527271f87433Sdalcinl @*/
52737087cfbeSBarry Smith PetscErrorCode  SNESKSPSetUseEW(SNES snes,PetscBool flag)
527471f87433Sdalcinl {
527571f87433Sdalcinl   PetscFunctionBegin;
52760700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5277acfcf0e5SJed Brown   PetscValidLogicalCollectiveBool(snes,flag,2);
527871f87433Sdalcinl   snes->ksp_ewconv = flag;
527971f87433Sdalcinl   PetscFunctionReturn(0);
528071f87433Sdalcinl }
528171f87433Sdalcinl 
528271f87433Sdalcinl /*@
5283fa9f3622SBarry Smith    SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method
528471f87433Sdalcinl    for computing relative tolerance for linear solvers within an
528571f87433Sdalcinl    inexact Newton method.
528671f87433Sdalcinl 
528771f87433Sdalcinl    Not Collective
528871f87433Sdalcinl 
528971f87433Sdalcinl    Input Parameter:
529071f87433Sdalcinl .  snes - SNES context
529171f87433Sdalcinl 
529271f87433Sdalcinl    Output Parameter:
529371f87433Sdalcinl .  flag - PETSC_TRUE or PETSC_FALSE
529471f87433Sdalcinl 
529571f87433Sdalcinl    Notes:
529671f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
529771f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
529871f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
529971f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
530071f87433Sdalcinl    solver.
530171f87433Sdalcinl 
530271f87433Sdalcinl    Level: advanced
530371f87433Sdalcinl 
530471f87433Sdalcinl    Reference:
530571f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
530671f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
530771f87433Sdalcinl 
5308fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
530971f87433Sdalcinl @*/
53107087cfbeSBarry Smith PetscErrorCode  SNESKSPGetUseEW(SNES snes, PetscBool  *flag)
531171f87433Sdalcinl {
531271f87433Sdalcinl   PetscFunctionBegin;
53130700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5314534a8f05SLisandro Dalcin   PetscValidBoolPointer(flag,2);
531571f87433Sdalcinl   *flag = snes->ksp_ewconv;
531671f87433Sdalcinl   PetscFunctionReturn(0);
531771f87433Sdalcinl }
531871f87433Sdalcinl 
531971f87433Sdalcinl /*@
5320fa9f3622SBarry Smith    SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker
532171f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
532271f87433Sdalcinl    Newton method.
532371f87433Sdalcinl 
53243f9fe445SBarry Smith    Logically Collective on SNES
532571f87433Sdalcinl 
532671f87433Sdalcinl    Input Parameters:
532771f87433Sdalcinl +    snes - SNES context
532871f87433Sdalcinl .    version - version 1, 2 (default is 2) or 3
532971f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
533071f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
533171f87433Sdalcinl .    gamma - multiplicative factor for version 2 rtol computation
533271f87433Sdalcinl              (0 <= gamma2 <= 1)
533371f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
533471f87433Sdalcinl .    alpha2 - power for safeguard
533571f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
533671f87433Sdalcinl 
533771f87433Sdalcinl    Note:
533871f87433Sdalcinl    Version 3 was contributed by Luis Chacon, June 2006.
533971f87433Sdalcinl 
534071f87433Sdalcinl    Use PETSC_DEFAULT to retain the default for any of the parameters.
534171f87433Sdalcinl 
534271f87433Sdalcinl    Level: advanced
534371f87433Sdalcinl 
534471f87433Sdalcinl    Reference:
534571f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
534671f87433Sdalcinl    inexact Newton method", Utah State University Math. Stat. Dept. Res.
534771f87433Sdalcinl    Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput.
534871f87433Sdalcinl 
5349fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW()
535071f87433Sdalcinl @*/
5351f5af7f23SKarl Rupp PetscErrorCode  SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold)
535271f87433Sdalcinl {
5353fa9f3622SBarry Smith   SNESKSPEW *kctx;
53545fd66863SKarl Rupp 
535571f87433Sdalcinl   PetscFunctionBegin;
53560700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5357fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
5358e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
5359c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,version,2);
5360c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol_0,3);
5361c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol_max,4);
5362c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,gamma,5);
5363c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,alpha,6);
5364c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,alpha2,7);
5365c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,threshold,8);
536671f87433Sdalcinl 
536771f87433Sdalcinl   if (version != PETSC_DEFAULT)   kctx->version   = version;
536871f87433Sdalcinl   if (rtol_0 != PETSC_DEFAULT)    kctx->rtol_0    = rtol_0;
536971f87433Sdalcinl   if (rtol_max != PETSC_DEFAULT)  kctx->rtol_max  = rtol_max;
537071f87433Sdalcinl   if (gamma != PETSC_DEFAULT)     kctx->gamma     = gamma;
537171f87433Sdalcinl   if (alpha != PETSC_DEFAULT)     kctx->alpha     = alpha;
537271f87433Sdalcinl   if (alpha2 != PETSC_DEFAULT)    kctx->alpha2    = alpha2;
537371f87433Sdalcinl   if (threshold != PETSC_DEFAULT) kctx->threshold = threshold;
537471f87433Sdalcinl 
5375f23aa3ddSBarry 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);
537657622a8eSBarry 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);
537757622a8eSBarry 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);
537857622a8eSBarry 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);
537957622a8eSBarry 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);
538057622a8eSBarry 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);
538171f87433Sdalcinl   PetscFunctionReturn(0);
538271f87433Sdalcinl }
538371f87433Sdalcinl 
538471f87433Sdalcinl /*@
5385fa9f3622SBarry Smith    SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker
538671f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
538771f87433Sdalcinl    Newton method.
538871f87433Sdalcinl 
538971f87433Sdalcinl    Not Collective
539071f87433Sdalcinl 
539171f87433Sdalcinl    Input Parameters:
539271f87433Sdalcinl      snes - SNES context
539371f87433Sdalcinl 
539471f87433Sdalcinl    Output Parameters:
539571f87433Sdalcinl +    version - version 1, 2 (default is 2) or 3
539671f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
539771f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
5398bf388a1fSBarry Smith .    gamma - multiplicative factor for version 2 rtol computation (0 <= gamma2 <= 1)
539971f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
540071f87433Sdalcinl .    alpha2 - power for safeguard
540171f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
540271f87433Sdalcinl 
540371f87433Sdalcinl    Level: advanced
540471f87433Sdalcinl 
5405fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW()
540671f87433Sdalcinl @*/
5407bf388a1fSBarry Smith PetscErrorCode  SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold)
540871f87433Sdalcinl {
5409fa9f3622SBarry Smith   SNESKSPEW *kctx;
54105fd66863SKarl Rupp 
541171f87433Sdalcinl   PetscFunctionBegin;
54120700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5413fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
5414e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
541571f87433Sdalcinl   if (version)   *version   = kctx->version;
541671f87433Sdalcinl   if (rtol_0)    *rtol_0    = kctx->rtol_0;
541771f87433Sdalcinl   if (rtol_max)  *rtol_max  = kctx->rtol_max;
541871f87433Sdalcinl   if (gamma)     *gamma     = kctx->gamma;
541971f87433Sdalcinl   if (alpha)     *alpha     = kctx->alpha;
542071f87433Sdalcinl   if (alpha2)    *alpha2    = kctx->alpha2;
542171f87433Sdalcinl   if (threshold) *threshold = kctx->threshold;
542271f87433Sdalcinl   PetscFunctionReturn(0);
542371f87433Sdalcinl }
542471f87433Sdalcinl 
5425d5378b5fSDmitry Karpeev  PetscErrorCode KSPPreSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes)
542671f87433Sdalcinl {
542771f87433Sdalcinl   PetscErrorCode ierr;
5428fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
542971f87433Sdalcinl   PetscReal      rtol  = PETSC_DEFAULT,stol;
543071f87433Sdalcinl 
543171f87433Sdalcinl   PetscFunctionBegin;
5432d4211eb9SBarry Smith   if (!snes->ksp_ewconv) PetscFunctionReturn(0);
543330058271SDmitry Karpeev   if (!snes->iter) {
543430058271SDmitry Karpeev     rtol = kctx->rtol_0; /* first time in, so use the original user rtol */
543530058271SDmitry Karpeev     ierr = VecNorm(snes->vec_func,NORM_2,&kctx->norm_first);CHKERRQ(ierr);
543630058271SDmitry Karpeev   }
5437f5af7f23SKarl Rupp   else {
543871f87433Sdalcinl     if (kctx->version == 1) {
543971f87433Sdalcinl       rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last;
544071f87433Sdalcinl       if (rtol < 0.0) rtol = -rtol;
544185ec1a3cSBarry Smith       stol = PetscPowReal(kctx->rtol_last,kctx->alpha2);
544271f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
544371f87433Sdalcinl     } else if (kctx->version == 2) {
544485ec1a3cSBarry Smith       rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha);
544585ec1a3cSBarry Smith       stol = kctx->gamma * PetscPowReal(kctx->rtol_last,kctx->alpha);
544671f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
544771f87433Sdalcinl     } else if (kctx->version == 3) { /* contributed by Luis Chacon, June 2006. */
544885ec1a3cSBarry Smith       rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha);
544971f87433Sdalcinl       /* safeguard: avoid sharp decrease of rtol */
545085ec1a3cSBarry Smith       stol = kctx->gamma*PetscPowReal(kctx->rtol_last,kctx->alpha);
545171f87433Sdalcinl       stol = PetscMax(rtol,stol);
545271f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
545371f87433Sdalcinl       /* safeguard: avoid oversolving */
545430058271SDmitry Karpeev       stol = kctx->gamma*(kctx->norm_first*snes->rtol)/snes->norm;
545571f87433Sdalcinl       stol = PetscMax(rtol,stol);
545671f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
5457e32f2f54SBarry Smith     } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version);
545871f87433Sdalcinl   }
545971f87433Sdalcinl   /* safeguard: avoid rtol greater than one */
546071f87433Sdalcinl   rtol = PetscMin(rtol,kctx->rtol_max);
546171f87433Sdalcinl   ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr);
546257622a8eSBarry Smith   ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%g\n",snes->iter,kctx->version,(double)rtol);CHKERRQ(ierr);
546371f87433Sdalcinl   PetscFunctionReturn(0);
546471f87433Sdalcinl }
546571f87433Sdalcinl 
5466d5378b5fSDmitry Karpeev PetscErrorCode KSPPostSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes)
546771f87433Sdalcinl {
546871f87433Sdalcinl   PetscErrorCode ierr;
5469fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
547071f87433Sdalcinl   PCSide         pcside;
547171f87433Sdalcinl   Vec            lres;
547271f87433Sdalcinl 
547371f87433Sdalcinl   PetscFunctionBegin;
5474d4211eb9SBarry Smith   if (!snes->ksp_ewconv) PetscFunctionReturn(0);
54759e5d0892SLisandro Dalcin   ierr = KSPGetTolerances(ksp,&kctx->rtol_last,NULL,NULL,NULL);CHKERRQ(ierr);
547671dbe336SPeter Brune   kctx->norm_last = snes->norm;
547771f87433Sdalcinl   if (kctx->version == 1) {
54784f00ce20SMatthew G. Knepley     PC        pc;
54794f00ce20SMatthew G. Knepley     PetscBool isNone;
54804f00ce20SMatthew G. Knepley 
54814f00ce20SMatthew G. Knepley     ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr);
54824f00ce20SMatthew G. Knepley     ierr = PetscObjectTypeCompare((PetscObject) pc, PCNONE, &isNone);CHKERRQ(ierr);
5483b037da10SBarry Smith     ierr = KSPGetPCSide(ksp,&pcside);CHKERRQ(ierr);
54844f00ce20SMatthew G. Knepley      if (pcside == PC_RIGHT || isNone) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */
548571f87433Sdalcinl       /* KSP residual is true linear residual */
548671f87433Sdalcinl       ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr);
548771f87433Sdalcinl     } else {
548871f87433Sdalcinl       /* KSP residual is preconditioned residual */
548971f87433Sdalcinl       /* compute true linear residual norm */
549071f87433Sdalcinl       ierr = VecDuplicate(b,&lres);CHKERRQ(ierr);
549171f87433Sdalcinl       ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr);
549271f87433Sdalcinl       ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr);
549371f87433Sdalcinl       ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr);
54946bf464f9SBarry Smith       ierr = VecDestroy(&lres);CHKERRQ(ierr);
549571f87433Sdalcinl     }
549671f87433Sdalcinl   }
549771f87433Sdalcinl   PetscFunctionReturn(0);
549871f87433Sdalcinl }
549971f87433Sdalcinl 
5500d4211eb9SBarry Smith /*@
5501d4211eb9SBarry Smith    SNESGetKSP - Returns the KSP context for a SNES solver.
5502d4211eb9SBarry Smith 
5503d4211eb9SBarry Smith    Not Collective, but if SNES object is parallel, then KSP object is parallel
5504d4211eb9SBarry Smith 
5505d4211eb9SBarry Smith    Input Parameter:
5506d4211eb9SBarry Smith .  snes - the SNES context
5507d4211eb9SBarry Smith 
5508d4211eb9SBarry Smith    Output Parameter:
5509d4211eb9SBarry Smith .  ksp - the KSP context
5510d4211eb9SBarry Smith 
5511d4211eb9SBarry Smith    Notes:
5512d4211eb9SBarry Smith    The user can then directly manipulate the KSP context to set various
5513d4211eb9SBarry Smith    options, etc.  Likewise, the user can then extract and manipulate the
5514d4211eb9SBarry Smith    PC contexts as well.
5515d4211eb9SBarry Smith 
5516d4211eb9SBarry Smith    Level: beginner
5517d4211eb9SBarry Smith 
5518d4211eb9SBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
5519d4211eb9SBarry Smith @*/
5520d4211eb9SBarry Smith PetscErrorCode  SNESGetKSP(SNES snes,KSP *ksp)
552171f87433Sdalcinl {
552271f87433Sdalcinl   PetscErrorCode ierr;
552371f87433Sdalcinl 
552471f87433Sdalcinl   PetscFunctionBegin;
5525d4211eb9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5526d4211eb9SBarry Smith   PetscValidPointer(ksp,2);
5527d4211eb9SBarry Smith 
5528d4211eb9SBarry Smith   if (!snes->ksp) {
5529d4211eb9SBarry Smith     ierr = KSPCreate(PetscObjectComm((PetscObject)snes),&snes->ksp);CHKERRQ(ierr);
5530d4211eb9SBarry Smith     ierr = PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1);CHKERRQ(ierr);
55313bb1ff40SBarry Smith     ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->ksp);CHKERRQ(ierr);
5532d4211eb9SBarry Smith 
5533d5378b5fSDmitry Karpeev     ierr = KSPSetPreSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_SNESEW,snes);CHKERRQ(ierr);
5534d5378b5fSDmitry Karpeev     ierr = KSPSetPostSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_SNESEW,snes);CHKERRQ(ierr);
5535a5c2985bSBarry Smith 
5536798534f6SMatthew G. Knepley     ierr = KSPMonitorSetFromOptions(snes->ksp, "-snes_monitor_ksp", "snes_preconditioned_residual", snes);CHKERRQ(ierr);
553716413a6aSBarry Smith     ierr = PetscObjectSetOptions((PetscObject)snes->ksp,((PetscObject)snes)->options);CHKERRQ(ierr);
5538d4211eb9SBarry Smith   }
5539d4211eb9SBarry Smith   *ksp = snes->ksp;
554071f87433Sdalcinl   PetscFunctionReturn(0);
554171f87433Sdalcinl }
55426c699258SBarry Smith 
5543d4211eb9SBarry Smith 
5544af0996ceSBarry Smith #include <petsc/private/dmimpl.h>
55456c699258SBarry Smith /*@
55462a808120SBarry Smith    SNESSetDM - Sets the DM that may be used by some nonlinear solvers or their underlying preconditioners
55476c699258SBarry Smith 
55483f9fe445SBarry Smith    Logically Collective on SNES
55496c699258SBarry Smith 
55506c699258SBarry Smith    Input Parameters:
55512a808120SBarry Smith +  snes - the nonlinear solver context
55522a808120SBarry Smith -  dm - the dm, cannot be NULL
55536c699258SBarry Smith 
5554e03a659cSJed Brown    Notes:
5555e03a659cSJed Brown    A DM can only be used for solving one problem at a time because information about the problem is stored on the DM,
5556e03a659cSJed Brown    even when not using interfaces like DMSNESSetFunction().  Use DMClone() to get a distinct DM when solving different
5557e03a659cSJed Brown    problems using the same function space.
5558e03a659cSJed Brown 
55596c699258SBarry Smith    Level: intermediate
55606c699258SBarry Smith 
55614c2026ceSFande Kong .seealso: SNESGetDM(), KSPSetDM(), KSPGetDM()
55626c699258SBarry Smith @*/
55637087cfbeSBarry Smith PetscErrorCode  SNESSetDM(SNES snes,DM dm)
55646c699258SBarry Smith {
55656c699258SBarry Smith   PetscErrorCode ierr;
5566345fed2cSBarry Smith   KSP            ksp;
5567942e3340SBarry Smith   DMSNES         sdm;
55686c699258SBarry Smith 
55696c699258SBarry Smith   PetscFunctionBegin;
55700700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
55712a808120SBarry Smith   PetscValidHeaderSpecific(dm,DM_CLASSID,2);
55722a808120SBarry Smith   ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr);
5573942e3340SBarry Smith   if (snes->dm) {               /* Move the DMSNES context over to the new DM unless the new DM already has one */
557451f4b3c7SToby Isaac     if (snes->dm->dmsnes && !dm->dmsnes) {
5575942e3340SBarry Smith       ierr = DMCopyDMSNES(snes->dm,dm);CHKERRQ(ierr);
5576942e3340SBarry Smith       ierr = DMGetDMSNES(snes->dm,&sdm);CHKERRQ(ierr);
5577f5af7f23SKarl Rupp       if (sdm->originaldm == snes->dm) sdm->originaldm = dm; /* Grant write privileges to the replacement DM */
55786cab3a1bSJed Brown     }
5579dc822a44SJed Brown     ierr = DMCoarsenHookRemove(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr);
55806bf464f9SBarry Smith     ierr = DMDestroy(&snes->dm);CHKERRQ(ierr);
55816cab3a1bSJed Brown   }
55826c699258SBarry Smith   snes->dm     = dm;
5583116d1032SJed Brown   snes->dmAuto = PETSC_FALSE;
5584f5af7f23SKarl Rupp 
5585345fed2cSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
5586345fed2cSBarry Smith   ierr = KSPSetDM(ksp,dm);CHKERRQ(ierr);
5587f22f69f0SBarry Smith   ierr = KSPSetDMActive(ksp,PETSC_FALSE);CHKERRQ(ierr);
5588efd4aadfSBarry Smith   if (snes->npc) {
5589efd4aadfSBarry Smith     ierr = SNESSetDM(snes->npc, snes->dm);CHKERRQ(ierr);
5590efd4aadfSBarry Smith     ierr = SNESSetNPCSide(snes,snes->npcside);CHKERRQ(ierr);
55912c155ee1SBarry Smith   }
55926c699258SBarry Smith   PetscFunctionReturn(0);
55936c699258SBarry Smith }
55946c699258SBarry Smith 
55956c699258SBarry Smith /*@
55966c699258SBarry Smith    SNESGetDM - Gets the DM that may be used by some preconditioners
55976c699258SBarry Smith 
55983f9fe445SBarry Smith    Not Collective but DM obtained is parallel on SNES
55996c699258SBarry Smith 
56006c699258SBarry Smith    Input Parameter:
56016c699258SBarry Smith . snes - the preconditioner context
56026c699258SBarry Smith 
56036c699258SBarry Smith    Output Parameter:
56046c699258SBarry Smith .  dm - the dm
56056c699258SBarry Smith 
56066c699258SBarry Smith    Level: intermediate
56076c699258SBarry Smith 
56084c2026ceSFande Kong .seealso: SNESSetDM(), KSPSetDM(), KSPGetDM()
56096c699258SBarry Smith @*/
56107087cfbeSBarry Smith PetscErrorCode  SNESGetDM(SNES snes,DM *dm)
56116c699258SBarry Smith {
56126cab3a1bSJed Brown   PetscErrorCode ierr;
56136cab3a1bSJed Brown 
56146c699258SBarry Smith   PetscFunctionBegin;
56150700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
56166cab3a1bSJed Brown   if (!snes->dm) {
5617ce94432eSBarry Smith     ierr         = DMShellCreate(PetscObjectComm((PetscObject)snes),&snes->dm);CHKERRQ(ierr);
5618116d1032SJed Brown     snes->dmAuto = PETSC_TRUE;
56196cab3a1bSJed Brown   }
56206c699258SBarry Smith   *dm = snes->dm;
56216c699258SBarry Smith   PetscFunctionReturn(0);
56226c699258SBarry Smith }
56230807856dSBarry Smith 
562431823bd8SMatthew G Knepley /*@
5625be95d8f1SBarry Smith   SNESSetNPC - Sets the nonlinear preconditioner to be used.
562631823bd8SMatthew G Knepley 
562731823bd8SMatthew G Knepley   Collective on SNES
562831823bd8SMatthew G Knepley 
562931823bd8SMatthew G Knepley   Input Parameters:
563031823bd8SMatthew G Knepley + snes - iterative context obtained from SNESCreate()
563131823bd8SMatthew G Knepley - pc   - the preconditioner object
563231823bd8SMatthew G Knepley 
563331823bd8SMatthew G Knepley   Notes:
5634be95d8f1SBarry Smith   Use SNESGetNPC() to retrieve the preconditioner context (for example,
563531823bd8SMatthew G Knepley   to configure it using the API).
563631823bd8SMatthew G Knepley 
563731823bd8SMatthew G Knepley   Level: developer
563831823bd8SMatthew G Knepley 
56393ad1a0b9SPatrick Farrell .seealso: SNESGetNPC(), SNESHasNPC()
564031823bd8SMatthew G Knepley @*/
5641be95d8f1SBarry Smith PetscErrorCode SNESSetNPC(SNES snes, SNES pc)
564231823bd8SMatthew G Knepley {
564331823bd8SMatthew G Knepley   PetscErrorCode ierr;
564431823bd8SMatthew G Knepley 
564531823bd8SMatthew G Knepley   PetscFunctionBegin;
564631823bd8SMatthew G Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
564731823bd8SMatthew G Knepley   PetscValidHeaderSpecific(pc, SNES_CLASSID, 2);
564831823bd8SMatthew G Knepley   PetscCheckSameComm(snes, 1, pc, 2);
564931823bd8SMatthew G Knepley   ierr     = PetscObjectReference((PetscObject) pc);CHKERRQ(ierr);
5650efd4aadfSBarry Smith   ierr     = SNESDestroy(&snes->npc);CHKERRQ(ierr);
5651efd4aadfSBarry Smith   snes->npc = pc;
5652efd4aadfSBarry Smith   ierr     = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->npc);CHKERRQ(ierr);
565331823bd8SMatthew G Knepley   PetscFunctionReturn(0);
565431823bd8SMatthew G Knepley }
565531823bd8SMatthew G Knepley 
565631823bd8SMatthew G Knepley /*@
5657be95d8f1SBarry Smith   SNESGetNPC - Creates a nonlinear preconditioning solver (SNES) to be used to precondition the nonlinear solver.
565831823bd8SMatthew G Knepley 
5659951fe5abSBarry Smith   Not Collective; but any changes to the obtained SNES object must be applied collectively
566031823bd8SMatthew G Knepley 
566131823bd8SMatthew G Knepley   Input Parameter:
566231823bd8SMatthew G Knepley . snes - iterative context obtained from SNESCreate()
566331823bd8SMatthew G Knepley 
566431823bd8SMatthew G Knepley   Output Parameter:
566531823bd8SMatthew G Knepley . pc - preconditioner context
566631823bd8SMatthew G Knepley 
5667b5badacbSBarry Smith   Options Database:
5668b5badacbSBarry Smith . -npc_snes_type <type> - set the type of the SNES to use as the nonlinear preconditioner
5669b5badacbSBarry Smith 
567095452b02SPatrick Sanan   Notes:
5671b5badacbSBarry Smith     If a SNES was previously set with SNESSetNPC() then that SNES is returned, otherwise a new SNES object is created.
5672be95d8f1SBarry Smith 
5673951fe5abSBarry Smith     The (preconditioner) SNES returned automatically inherits the same nonlinear function and Jacobian supplied to the original
5674951fe5abSBarry Smith     SNES during SNESSetUp()
5675951fe5abSBarry Smith 
567631823bd8SMatthew G Knepley   Level: developer
567731823bd8SMatthew G Knepley 
5678951fe5abSBarry Smith .seealso: SNESSetNPC(), SNESHasNPC(), SNES, SNESCreate()
567931823bd8SMatthew G Knepley @*/
5680be95d8f1SBarry Smith PetscErrorCode SNESGetNPC(SNES snes, SNES *pc)
568131823bd8SMatthew G Knepley {
568231823bd8SMatthew G Knepley   PetscErrorCode ierr;
5683a64e098fSPeter Brune   const char     *optionsprefix;
568431823bd8SMatthew G Knepley 
568531823bd8SMatthew G Knepley   PetscFunctionBegin;
568631823bd8SMatthew G Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
568731823bd8SMatthew G Knepley   PetscValidPointer(pc, 2);
5688efd4aadfSBarry Smith   if (!snes->npc) {
5689efd4aadfSBarry Smith     ierr = SNESCreate(PetscObjectComm((PetscObject)snes),&snes->npc);CHKERRQ(ierr);
5690efd4aadfSBarry Smith     ierr = PetscObjectIncrementTabLevel((PetscObject)snes->npc,(PetscObject)snes,1);CHKERRQ(ierr);
5691efd4aadfSBarry Smith     ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr);
5692a64e098fSPeter Brune     ierr = SNESGetOptionsPrefix(snes,&optionsprefix);CHKERRQ(ierr);
5693efd4aadfSBarry Smith     ierr = SNESSetOptionsPrefix(snes->npc,optionsprefix);CHKERRQ(ierr);
5694efd4aadfSBarry Smith     ierr = SNESAppendOptionsPrefix(snes->npc,"npc_");CHKERRQ(ierr);
5695efd4aadfSBarry Smith     ierr = SNESSetCountersReset(snes->npc,PETSC_FALSE);CHKERRQ(ierr);
569631823bd8SMatthew G Knepley   }
5697efd4aadfSBarry Smith   *pc = snes->npc;
569831823bd8SMatthew G Knepley   PetscFunctionReturn(0);
569931823bd8SMatthew G Knepley }
570031823bd8SMatthew G Knepley 
57013ad1a0b9SPatrick Farrell /*@
57023ad1a0b9SPatrick Farrell   SNESHasNPC - Returns whether a nonlinear preconditioner exists
57033ad1a0b9SPatrick Farrell 
57043ad1a0b9SPatrick Farrell   Not Collective
57053ad1a0b9SPatrick Farrell 
57063ad1a0b9SPatrick Farrell   Input Parameter:
57073ad1a0b9SPatrick Farrell . snes - iterative context obtained from SNESCreate()
57083ad1a0b9SPatrick Farrell 
57093ad1a0b9SPatrick Farrell   Output Parameter:
57103ad1a0b9SPatrick Farrell . has_npc - whether the SNES has an NPC or not
57113ad1a0b9SPatrick Farrell 
57123ad1a0b9SPatrick Farrell   Level: developer
57133ad1a0b9SPatrick Farrell 
57143ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESGetNPC()
57153ad1a0b9SPatrick Farrell @*/
57163ad1a0b9SPatrick Farrell PetscErrorCode SNESHasNPC(SNES snes, PetscBool *has_npc)
57173ad1a0b9SPatrick Farrell {
57183ad1a0b9SPatrick Farrell   PetscFunctionBegin;
57193ad1a0b9SPatrick Farrell   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5720efd4aadfSBarry Smith   *has_npc = (PetscBool) (snes->npc ? PETSC_TRUE : PETSC_FALSE);
57213ad1a0b9SPatrick Farrell   PetscFunctionReturn(0);
57223ad1a0b9SPatrick Farrell }
57233ad1a0b9SPatrick Farrell 
5724c40d0f55SPeter Brune /*@
5725be95d8f1SBarry Smith     SNESSetNPCSide - Sets the preconditioning side.
5726c40d0f55SPeter Brune 
5727c40d0f55SPeter Brune     Logically Collective on SNES
5728c40d0f55SPeter Brune 
5729c40d0f55SPeter Brune     Input Parameter:
5730c40d0f55SPeter Brune .   snes - iterative context obtained from SNESCreate()
5731c40d0f55SPeter Brune 
5732c40d0f55SPeter Brune     Output Parameter:
5733c40d0f55SPeter Brune .   side - the preconditioning side, where side is one of
5734c40d0f55SPeter Brune .vb
57352d547940SBarry Smith       PC_LEFT - left preconditioning
57362d547940SBarry Smith       PC_RIGHT - right preconditioning (default for most nonlinear solvers)
5737c40d0f55SPeter Brune .ve
5738c40d0f55SPeter Brune 
5739c40d0f55SPeter Brune     Options Database Keys:
5740c40d0f55SPeter Brune .   -snes_pc_side <right,left>
5741c40d0f55SPeter Brune 
574295452b02SPatrick Sanan     Notes:
574395452b02SPatrick Sanan     SNESNRICHARDSON and SNESNCG only support left preconditioning.
57442d547940SBarry Smith 
5745c40d0f55SPeter Brune     Level: intermediate
5746c40d0f55SPeter Brune 
5747be95d8f1SBarry Smith .seealso: SNESGetNPCSide(), KSPSetPCSide()
5748c40d0f55SPeter Brune @*/
5749be95d8f1SBarry Smith PetscErrorCode  SNESSetNPCSide(SNES snes,PCSide side)
5750c40d0f55SPeter Brune {
5751c40d0f55SPeter Brune   PetscFunctionBegin;
5752c40d0f55SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5753c40d0f55SPeter Brune   PetscValidLogicalCollectiveEnum(snes,side,2);
5754efd4aadfSBarry Smith   snes->npcside= side;
5755c40d0f55SPeter Brune   PetscFunctionReturn(0);
5756c40d0f55SPeter Brune }
5757c40d0f55SPeter Brune 
5758c40d0f55SPeter Brune /*@
5759be95d8f1SBarry Smith     SNESGetNPCSide - Gets the preconditioning side.
5760c40d0f55SPeter Brune 
5761c40d0f55SPeter Brune     Not Collective
5762c40d0f55SPeter Brune 
5763c40d0f55SPeter Brune     Input Parameter:
5764c40d0f55SPeter Brune .   snes - iterative context obtained from SNESCreate()
5765c40d0f55SPeter Brune 
5766c40d0f55SPeter Brune     Output Parameter:
5767c40d0f55SPeter Brune .   side - the preconditioning side, where side is one of
5768c40d0f55SPeter Brune .vb
57692d547940SBarry Smith       PC_LEFT - left preconditioning
57702d547940SBarry Smith       PC_RIGHT - right preconditioning (default for most nonlinear solvers)
5771c40d0f55SPeter Brune .ve
5772c40d0f55SPeter Brune 
5773c40d0f55SPeter Brune     Level: intermediate
5774c40d0f55SPeter Brune 
5775be95d8f1SBarry Smith .seealso: SNESSetNPCSide(), KSPGetPCSide()
5776c40d0f55SPeter Brune @*/
5777be95d8f1SBarry Smith PetscErrorCode  SNESGetNPCSide(SNES snes,PCSide *side)
5778c40d0f55SPeter Brune {
5779c40d0f55SPeter Brune   PetscFunctionBegin;
5780c40d0f55SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5781c40d0f55SPeter Brune   PetscValidPointer(side,2);
5782efd4aadfSBarry Smith   *side = snes->npcside;
5783c40d0f55SPeter Brune   PetscFunctionReturn(0);
5784c40d0f55SPeter Brune }
5785c40d0f55SPeter Brune 
57869e764e56SPeter Brune /*@
57877601faf0SJed Brown   SNESSetLineSearch - Sets the linesearch on the SNES instance.
57889e764e56SPeter Brune 
57899e764e56SPeter Brune   Collective on SNES
57909e764e56SPeter Brune 
57919e764e56SPeter Brune   Input Parameters:
57929e764e56SPeter Brune + snes - iterative context obtained from SNESCreate()
57939e764e56SPeter Brune - linesearch   - the linesearch object
57949e764e56SPeter Brune 
57959e764e56SPeter Brune   Notes:
57967601faf0SJed Brown   Use SNESGetLineSearch() to retrieve the preconditioner context (for example,
57979e764e56SPeter Brune   to configure it using the API).
57989e764e56SPeter Brune 
57999e764e56SPeter Brune   Level: developer
58009e764e56SPeter Brune 
58017601faf0SJed Brown .seealso: SNESGetLineSearch()
58029e764e56SPeter Brune @*/
58037601faf0SJed Brown PetscErrorCode SNESSetLineSearch(SNES snes, SNESLineSearch linesearch)
58049e764e56SPeter Brune {
58059e764e56SPeter Brune   PetscErrorCode ierr;
58069e764e56SPeter Brune 
58079e764e56SPeter Brune   PetscFunctionBegin;
58089e764e56SPeter Brune   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5809f1c6b773SPeter Brune   PetscValidHeaderSpecific(linesearch, SNESLINESEARCH_CLASSID, 2);
58109e764e56SPeter Brune   PetscCheckSameComm(snes, 1, linesearch, 2);
58119e764e56SPeter Brune   ierr = PetscObjectReference((PetscObject) linesearch);CHKERRQ(ierr);
5812f1c6b773SPeter Brune   ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr);
5813f5af7f23SKarl Rupp 
58149e764e56SPeter Brune   snes->linesearch = linesearch;
5815f5af7f23SKarl Rupp 
58163bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr);
58179e764e56SPeter Brune   PetscFunctionReturn(0);
58189e764e56SPeter Brune }
58199e764e56SPeter Brune 
5820a34ceb2aSJed Brown /*@
58217601faf0SJed Brown   SNESGetLineSearch - Returns a pointer to the line search context set with SNESSetLineSearch()
58228141a3b9SPeter Brune   or creates a default line search instance associated with the SNES and returns it.
58239e764e56SPeter Brune 
58249e764e56SPeter Brune   Not Collective
58259e764e56SPeter Brune 
58269e764e56SPeter Brune   Input Parameter:
58279e764e56SPeter Brune . snes - iterative context obtained from SNESCreate()
58289e764e56SPeter Brune 
58299e764e56SPeter Brune   Output Parameter:
58309e764e56SPeter Brune . linesearch - linesearch context
58319e764e56SPeter Brune 
5832162e0bf5SPeter Brune   Level: beginner
58339e764e56SPeter Brune 
5834162e0bf5SPeter Brune .seealso: SNESSetLineSearch(), SNESLineSearchCreate()
58359e764e56SPeter Brune @*/
58367601faf0SJed Brown PetscErrorCode SNESGetLineSearch(SNES snes, SNESLineSearch *linesearch)
58379e764e56SPeter Brune {
58389e764e56SPeter Brune   PetscErrorCode ierr;
58399e764e56SPeter Brune   const char     *optionsprefix;
58409e764e56SPeter Brune 
58419e764e56SPeter Brune   PetscFunctionBegin;
58429e764e56SPeter Brune   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
58439e764e56SPeter Brune   PetscValidPointer(linesearch, 2);
58449e764e56SPeter Brune   if (!snes->linesearch) {
58459e764e56SPeter Brune     ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr);
584682f516ccSBarry Smith     ierr = SNESLineSearchCreate(PetscObjectComm((PetscObject)snes), &snes->linesearch);CHKERRQ(ierr);
5847f1c6b773SPeter Brune     ierr = SNESLineSearchSetSNES(snes->linesearch, snes);CHKERRQ(ierr);
5848b1dca40bSPeter Brune     ierr = SNESLineSearchAppendOptionsPrefix(snes->linesearch, optionsprefix);CHKERRQ(ierr);
58499e764e56SPeter Brune     ierr = PetscObjectIncrementTabLevel((PetscObject) snes->linesearch, (PetscObject) snes, 1);CHKERRQ(ierr);
58503bb1ff40SBarry Smith     ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr);
58519e764e56SPeter Brune   }
58529e764e56SPeter Brune   *linesearch = snes->linesearch;
58539e764e56SPeter Brune   PetscFunctionReturn(0);
58549e764e56SPeter Brune }
5855