xref: /petsc/src/snes/interface/snes.c (revision c1e67a491dcd3fce18efcb05d6f137bc2aa8e6a9)
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;
1394db00ebSBarry Smith PetscLogEvent SNES_Solve, 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 .keywords: SNES, set, initial guess, nonzero
34e113a28aSBarry Smith 
35e113a28aSBarry Smith .seealso: SNESGetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged()
36e113a28aSBarry Smith @*/
377087cfbeSBarry Smith PetscErrorCode  SNESSetErrorIfNotConverged(SNES snes,PetscBool flg)
38e113a28aSBarry Smith {
39e113a28aSBarry Smith   PetscFunctionBegin;
40e113a28aSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
41acfcf0e5SJed Brown   PetscValidLogicalCollectiveBool(snes,flg,2);
42e113a28aSBarry Smith   snes->errorifnotconverged = flg;
43e113a28aSBarry Smith   PetscFunctionReturn(0);
44e113a28aSBarry Smith }
45e113a28aSBarry Smith 
46e113a28aSBarry Smith /*@
47e113a28aSBarry Smith    SNESGetErrorIfNotConverged - Will SNESSolve() generate an error if the solver does not converge?
48e113a28aSBarry Smith 
49e113a28aSBarry Smith    Not Collective
50e113a28aSBarry Smith 
51e113a28aSBarry Smith    Input Parameter:
52e113a28aSBarry Smith .  snes - iterative context obtained from SNESCreate()
53e113a28aSBarry Smith 
54e113a28aSBarry Smith    Output Parameter:
55e113a28aSBarry Smith .  flag - PETSC_TRUE if it will generate an error, else PETSC_FALSE
56e113a28aSBarry Smith 
57e113a28aSBarry Smith    Level: intermediate
58e113a28aSBarry Smith 
59e113a28aSBarry Smith .keywords: SNES, set, initial guess, nonzero
60e113a28aSBarry Smith 
61e113a28aSBarry Smith .seealso:  SNESSetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged()
62e113a28aSBarry Smith @*/
637087cfbeSBarry Smith PetscErrorCode  SNESGetErrorIfNotConverged(SNES snes,PetscBool  *flag)
64e113a28aSBarry Smith {
65e113a28aSBarry Smith   PetscFunctionBegin;
66e113a28aSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
67e113a28aSBarry Smith   PetscValidPointer(flag,2);
68e113a28aSBarry Smith   *flag = snes->errorifnotconverged;
69e113a28aSBarry Smith   PetscFunctionReturn(0);
70e113a28aSBarry Smith }
71e113a28aSBarry Smith 
724fc747eaSLawrence Mitchell /*@
734fc747eaSLawrence Mitchell     SNESSetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution?
744fc747eaSLawrence Mitchell 
754fc747eaSLawrence Mitchell    Logically Collective on SNES
764fc747eaSLawrence Mitchell 
774fc747eaSLawrence Mitchell     Input Parameters:
784fc747eaSLawrence Mitchell +   snes - the shell SNES
794fc747eaSLawrence Mitchell -   flg - is the residual computed?
804fc747eaSLawrence Mitchell 
814fc747eaSLawrence Mitchell    Level: advanced
824fc747eaSLawrence Mitchell 
834fc747eaSLawrence Mitchell .seealso: SNESGetAlwaysComputesFinalResidual()
844fc747eaSLawrence Mitchell @*/
854fc747eaSLawrence Mitchell PetscErrorCode  SNESSetAlwaysComputesFinalResidual(SNES snes, PetscBool flg)
864fc747eaSLawrence Mitchell {
874fc747eaSLawrence Mitchell   PetscFunctionBegin;
884fc747eaSLawrence Mitchell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
894fc747eaSLawrence Mitchell   snes->alwayscomputesfinalresidual = flg;
904fc747eaSLawrence Mitchell   PetscFunctionReturn(0);
914fc747eaSLawrence Mitchell }
924fc747eaSLawrence Mitchell 
934fc747eaSLawrence Mitchell /*@
944fc747eaSLawrence Mitchell     SNESGetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution?
954fc747eaSLawrence Mitchell 
964fc747eaSLawrence Mitchell    Logically Collective on SNES
974fc747eaSLawrence Mitchell 
984fc747eaSLawrence Mitchell     Input Parameter:
994fc747eaSLawrence Mitchell .   snes - the shell SNES
1004fc747eaSLawrence Mitchell 
1014fc747eaSLawrence Mitchell     Output Parameter:
1024fc747eaSLawrence Mitchell .   flg - is the residual computed?
1034fc747eaSLawrence Mitchell 
1044fc747eaSLawrence Mitchell    Level: advanced
1054fc747eaSLawrence Mitchell 
1064fc747eaSLawrence Mitchell .seealso: SNESSetAlwaysComputesFinalResidual()
1074fc747eaSLawrence Mitchell @*/
1084fc747eaSLawrence Mitchell PetscErrorCode  SNESGetAlwaysComputesFinalResidual(SNES snes, PetscBool *flg)
1094fc747eaSLawrence Mitchell {
1104fc747eaSLawrence Mitchell   PetscFunctionBegin;
1114fc747eaSLawrence Mitchell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1124fc747eaSLawrence Mitchell   *flg = snes->alwayscomputesfinalresidual;
1134fc747eaSLawrence Mitchell   PetscFunctionReturn(0);
1144fc747eaSLawrence Mitchell }
1154fc747eaSLawrence Mitchell 
116e725d27bSBarry Smith /*@
117bf388a1fSBarry Smith    SNESSetFunctionDomainError - tells SNES that the input vector to your SNESFunction is not
1184936397dSBarry Smith      in the functions domain. For example, negative pressure.
1194936397dSBarry Smith 
1203f9fe445SBarry Smith    Logically Collective on SNES
1214936397dSBarry Smith 
1224936397dSBarry Smith    Input Parameters:
1236a388c36SPeter Brune .  snes - the SNES context
1244936397dSBarry Smith 
12528529972SSatish Balay    Level: advanced
1264936397dSBarry Smith 
1274936397dSBarry Smith .keywords: SNES, view
1284936397dSBarry Smith 
129bf388a1fSBarry Smith .seealso: SNESCreate(), SNESSetFunction(), SNESFunction
1304936397dSBarry Smith @*/
1317087cfbeSBarry Smith PetscErrorCode  SNESSetFunctionDomainError(SNES snes)
1324936397dSBarry Smith {
1334936397dSBarry Smith   PetscFunctionBegin;
1340700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
135422a814eSBarry Smith   if (snes->errorifnotconverged) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates input vector is not in the function domain");
1364936397dSBarry Smith   snes->domainerror = PETSC_TRUE;
1374936397dSBarry Smith   PetscFunctionReturn(0);
1384936397dSBarry Smith }
1394936397dSBarry Smith 
1406a388c36SPeter Brune /*@
141c77b2880SPeter Brune    SNESGetFunctionDomainError - Gets the status of the domain error after a call to SNESComputeFunction;
1426a388c36SPeter Brune 
1436a388c36SPeter Brune    Logically Collective on SNES
1446a388c36SPeter Brune 
1456a388c36SPeter Brune    Input Parameters:
1466a388c36SPeter Brune .  snes - the SNES context
1476a388c36SPeter Brune 
1486a388c36SPeter Brune    Output Parameters:
149bf388a1fSBarry Smith .  domainerror - Set to PETSC_TRUE if there's a domain error; PETSC_FALSE otherwise.
1506a388c36SPeter Brune 
1516a388c36SPeter Brune    Level: advanced
1526a388c36SPeter Brune 
1536a388c36SPeter Brune .keywords: SNES, view
1546a388c36SPeter Brune 
155bf388a1fSBarry Smith .seealso: SNESSetFunctionDomainError(), SNESComputeFunction()
1566a388c36SPeter Brune @*/
1576a388c36SPeter Brune PetscErrorCode  SNESGetFunctionDomainError(SNES snes, PetscBool *domainerror)
1586a388c36SPeter Brune {
1596a388c36SPeter Brune   PetscFunctionBegin;
1606a388c36SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1616a388c36SPeter Brune   PetscValidPointer(domainerror, 2);
1626a388c36SPeter Brune   *domainerror = snes->domainerror;
1636a388c36SPeter Brune   PetscFunctionReturn(0);
1646a388c36SPeter Brune }
1656a388c36SPeter Brune 
16655849f57SBarry Smith /*@C
16755849f57SBarry Smith   SNESLoad - Loads a SNES that has been stored in binary  with SNESView().
16855849f57SBarry Smith 
16955849f57SBarry Smith   Collective on PetscViewer
17055849f57SBarry Smith 
17155849f57SBarry Smith   Input Parameters:
17255849f57SBarry Smith + newdm - the newly loaded SNES, this needs to have been created with SNESCreate() or
17355849f57SBarry Smith            some related function before a call to SNESLoad().
17455849f57SBarry Smith - viewer - binary file viewer, obtained from PetscViewerBinaryOpen()
17555849f57SBarry Smith 
17655849f57SBarry Smith    Level: intermediate
17755849f57SBarry Smith 
17855849f57SBarry Smith   Notes:
17955849f57SBarry Smith    The type is determined by the data in the file, any type set into the SNES before this call is ignored.
18055849f57SBarry Smith 
18155849f57SBarry Smith   Notes for advanced users:
18255849f57SBarry Smith   Most users should not need to know the details of the binary storage
18355849f57SBarry Smith   format, since SNESLoad() and TSView() completely hide these details.
18455849f57SBarry Smith   But for anyone who's interested, the standard binary matrix storage
18555849f57SBarry Smith   format is
18655849f57SBarry Smith .vb
18755849f57SBarry Smith      has not yet been determined
18855849f57SBarry Smith .ve
18955849f57SBarry Smith 
19055849f57SBarry Smith .seealso: PetscViewerBinaryOpen(), SNESView(), MatLoad(), VecLoad()
19155849f57SBarry Smith @*/
1922d53ad75SBarry Smith PetscErrorCode  SNESLoad(SNES snes, PetscViewer viewer)
19355849f57SBarry Smith {
19455849f57SBarry Smith   PetscErrorCode ierr;
19555849f57SBarry Smith   PetscBool      isbinary;
196060da220SMatthew G. Knepley   PetscInt       classid;
19755849f57SBarry Smith   char           type[256];
19855849f57SBarry Smith   KSP            ksp;
1992d53ad75SBarry Smith   DM             dm;
2002d53ad75SBarry Smith   DMSNES         dmsnes;
20155849f57SBarry Smith 
20255849f57SBarry Smith   PetscFunctionBegin;
2032d53ad75SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
20455849f57SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
20555849f57SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
20655849f57SBarry Smith   if (!isbinary) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen()");
20755849f57SBarry Smith 
208060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,&classid,1,NULL,PETSC_INT);CHKERRQ(ierr);
209ce94432eSBarry Smith   if (classid != SNES_FILE_CLASSID) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_WRONG,"Not SNES next in file");
210060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,type,256,NULL,PETSC_CHAR);CHKERRQ(ierr);
2112d53ad75SBarry Smith   ierr = SNESSetType(snes, type);CHKERRQ(ierr);
2122d53ad75SBarry Smith   if (snes->ops->load) {
2132d53ad75SBarry Smith     ierr = (*snes->ops->load)(snes,viewer);CHKERRQ(ierr);
214f2c2a1b9SBarry Smith   }
2152d53ad75SBarry Smith   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2162d53ad75SBarry Smith   ierr = DMGetDMSNES(dm,&dmsnes);CHKERRQ(ierr);
2172d53ad75SBarry Smith   ierr = DMSNESLoad(dmsnes,viewer);CHKERRQ(ierr);
2182d53ad75SBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
21955849f57SBarry Smith   ierr = KSPLoad(ksp,viewer);CHKERRQ(ierr);
22055849f57SBarry Smith   PetscFunctionReturn(0);
22155849f57SBarry Smith }
2226a388c36SPeter Brune 
2239804daf3SBarry Smith #include <petscdraw.h>
224e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
225e04113cfSBarry Smith #include <petscviewersaws.h>
226bfb97211SBarry Smith #endif
2278404b7f3SBarry Smith 
2287e2c5f70SBarry Smith /*@C
2299b94acceSBarry Smith    SNESView - Prints the SNES data structure.
2309b94acceSBarry Smith 
2314c49b128SBarry Smith    Collective on SNES
232fee21e36SBarry Smith 
233c7afd0dbSLois Curfman McInnes    Input Parameters:
234c7afd0dbSLois Curfman McInnes +  SNES - the SNES context
235c7afd0dbSLois Curfman McInnes -  viewer - visualization context
236c7afd0dbSLois Curfman McInnes 
2379b94acceSBarry Smith    Options Database Key:
238c8a8ba5cSLois Curfman McInnes .  -snes_view - Calls SNESView() at end of SNESSolve()
2399b94acceSBarry Smith 
2409b94acceSBarry Smith    Notes:
2419b94acceSBarry Smith    The available visualization contexts include
242b0a32e0cSBarry Smith +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
243b0a32e0cSBarry Smith -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
244c8a8ba5cSLois Curfman McInnes          output where only the first processor opens
245c8a8ba5cSLois Curfman McInnes          the file.  All other processors send their
246c8a8ba5cSLois Curfman McInnes          data to the first processor to print.
2479b94acceSBarry Smith 
2483e081fefSLois Curfman McInnes    The user can open an alternative visualization context with
249b0a32e0cSBarry Smith    PetscViewerASCIIOpen() - output to a specified file.
2509b94acceSBarry Smith 
25136851e7fSLois Curfman McInnes    Level: beginner
25236851e7fSLois Curfman McInnes 
2539b94acceSBarry Smith .keywords: SNES, view
2549b94acceSBarry Smith 
255b0a32e0cSBarry Smith .seealso: PetscViewerASCIIOpen()
2569b94acceSBarry Smith @*/
2577087cfbeSBarry Smith PetscErrorCode  SNESView(SNES snes,PetscViewer viewer)
2589b94acceSBarry Smith {
259fa9f3622SBarry Smith   SNESKSPEW      *kctx;
260dfbe8321SBarry Smith   PetscErrorCode ierr;
26194b7f48cSBarry Smith   KSP            ksp;
2627f1410a3SPeter Brune   SNESLineSearch linesearch;
26372a02f06SBarry Smith   PetscBool      iascii,isstring,isbinary,isdraw;
2642d53ad75SBarry Smith   DMSNES         dmsnes;
265e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
266536b137fSBarry Smith   PetscBool      issaws;
267bfb97211SBarry Smith #endif
2689b94acceSBarry Smith 
2693a40ed3dSBarry Smith   PetscFunctionBegin;
2700700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2713050cee2SBarry Smith   if (!viewer) {
272ce94432eSBarry Smith     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&viewer);CHKERRQ(ierr);
2733050cee2SBarry Smith   }
2740700a824SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
275c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,viewer,2);
27674679c65SBarry Smith 
277251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
278251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr);
27955849f57SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
28072a02f06SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
281e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
282536b137fSBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSAWS,&issaws);CHKERRQ(ierr);
283bfb97211SBarry Smith #endif
28432077d6dSBarry Smith   if (iascii) {
285dc0571f2SMatthew G. Knepley     SNESNormSchedule normschedule;
2868404b7f3SBarry Smith     DM               dm;
2878404b7f3SBarry Smith     PetscErrorCode   (*cJ)(SNES,Vec,Mat,Mat,void*);
2888404b7f3SBarry Smith     void             *ctx;
289dc0571f2SMatthew G. Knepley 
290dae58748SBarry Smith     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)snes,viewer);CHKERRQ(ierr);
291fce1e034SJed Brown     if (!snes->setupcalled) {
292fce1e034SJed Brown       ierr = PetscViewerASCIIPrintf(viewer,"  SNES has not been set up so information may be incomplete\n");CHKERRQ(ierr);
293fce1e034SJed Brown     }
294e7788613SBarry Smith     if (snes->ops->view) {
295b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
296e7788613SBarry Smith       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
297b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
2980ef38995SBarry Smith     }
29977431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr);
30057622a8eSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  tolerances: relative=%g, absolute=%g, solution=%g\n",(double)snes->rtol,(double)snes->abstol,(double)snes->stol);CHKERRQ(ierr);
301efd4aadfSBarry Smith     if (snes->usesksp) {
30277431f27SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr);
303efd4aadfSBarry Smith     }
30477431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr);
305dc0571f2SMatthew G. Knepley     ierr = SNESGetNormSchedule(snes, &normschedule);CHKERRQ(ierr);
306dc0571f2SMatthew G. Knepley     if (normschedule > 0) {ierr = PetscViewerASCIIPrintf(viewer,"  norm schedule %s\n",SNESNormSchedules[normschedule]);CHKERRQ(ierr);}
30717fe4bdfSPeter Brune     if (snes->gridsequence) {
30817fe4bdfSPeter Brune       ierr = PetscViewerASCIIPrintf(viewer,"  total number of grid sequence refinements=%D\n",snes->gridsequence);CHKERRQ(ierr);
30917fe4bdfSPeter Brune     }
3109b94acceSBarry Smith     if (snes->ksp_ewconv) {
311fa9f3622SBarry Smith       kctx = (SNESKSPEW*)snes->kspconvctx;
3129b94acceSBarry Smith       if (kctx) {
31377431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"  Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr);
31457622a8eSBarry 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);
31557622a8eSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"    gamma=%g, alpha=%g, alpha2=%g\n",(double)kctx->gamma,(double)kctx->alpha,(double)kctx->alpha2);CHKERRQ(ierr);
3169b94acceSBarry Smith       }
3179b94acceSBarry Smith     }
318eb1f6c34SBarry Smith     if (snes->lagpreconditioner == -1) {
319eb1f6c34SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Preconditioned is never rebuilt\n");CHKERRQ(ierr);
320eb1f6c34SBarry Smith     } else if (snes->lagpreconditioner > 1) {
321eb1f6c34SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Preconditioned is rebuilt every %D new Jacobians\n",snes->lagpreconditioner);CHKERRQ(ierr);
322eb1f6c34SBarry Smith     }
323eb1f6c34SBarry Smith     if (snes->lagjacobian == -1) {
324eb1f6c34SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is never rebuilt\n");CHKERRQ(ierr);
325eb1f6c34SBarry Smith     } else if (snes->lagjacobian > 1) {
32642f4f86dSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is rebuilt every %D SNES iterations\n",snes->lagjacobian);CHKERRQ(ierr);
327eb1f6c34SBarry Smith     }
3288404b7f3SBarry Smith     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
3298404b7f3SBarry Smith     ierr = DMSNESGetJacobian(dm,&cJ,&ctx);CHKERRQ(ierr);
3308404b7f3SBarry Smith     if (cJ == SNESComputeJacobianDefault) {
3318404b7f3SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is built using finite differences one column at a time\n");CHKERRQ(ierr);
3328404b7f3SBarry Smith     } else if (cJ == SNESComputeJacobianDefaultColor) {
3338404b7f3SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Jacobian is built using finite differences with coloring\n");CHKERRQ(ierr);
3348404b7f3SBarry Smith     }
3350f5bd95cSBarry Smith   } else if (isstring) {
336317d6ea6SBarry Smith     const char *type;
337454a90a3SBarry Smith     ierr = SNESGetType(snes,&type);CHKERRQ(ierr);
338b0a32e0cSBarry Smith     ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr);
33955849f57SBarry Smith   } else if (isbinary) {
34055849f57SBarry Smith     PetscInt    classid = SNES_FILE_CLASSID;
34155849f57SBarry Smith     MPI_Comm    comm;
34255849f57SBarry Smith     PetscMPIInt rank;
34355849f57SBarry Smith     char        type[256];
34455849f57SBarry Smith 
34555849f57SBarry Smith     ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr);
34655849f57SBarry Smith     ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
34755849f57SBarry Smith     if (!rank) {
34855849f57SBarry Smith       ierr = PetscViewerBinaryWrite(viewer,&classid,1,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr);
34989d949e2SBarry Smith       ierr = PetscStrncpy(type,((PetscObject)snes)->type_name,sizeof(type));CHKERRQ(ierr);
35089d949e2SBarry Smith       ierr = PetscViewerBinaryWrite(viewer,type,sizeof(type),PETSC_CHAR,PETSC_FALSE);CHKERRQ(ierr);
35155849f57SBarry Smith     }
35255849f57SBarry Smith     if (snes->ops->view) {
35355849f57SBarry Smith       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
35455849f57SBarry Smith     }
35572a02f06SBarry Smith   } else if (isdraw) {
35672a02f06SBarry Smith     PetscDraw draw;
35772a02f06SBarry Smith     char      str[36];
35889fd9fafSBarry Smith     PetscReal x,y,bottom,h;
35972a02f06SBarry Smith 
36072a02f06SBarry Smith     ierr   = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
36172a02f06SBarry Smith     ierr   = PetscDrawGetCurrentPoint(draw,&x,&y);CHKERRQ(ierr);
362a126751eSBarry Smith     ierr   = PetscStrncpy(str,"SNES: ",sizeof(str));CHKERRQ(ierr);
363a126751eSBarry Smith     ierr   = PetscStrlcat(str,((PetscObject)snes)->type_name,sizeof(str));CHKERRQ(ierr);
36451fa3d41SBarry Smith     ierr   = PetscDrawStringBoxed(draw,x,y,PETSC_DRAW_BLUE,PETSC_DRAW_BLACK,str,NULL,&h);CHKERRQ(ierr);
36589fd9fafSBarry Smith     bottom = y - h;
36672a02f06SBarry Smith     ierr   = PetscDrawPushCurrentPoint(draw,x,bottom);CHKERRQ(ierr);
367c4646bacSPeter Brune     if (snes->ops->view) {
368c4646bacSPeter Brune       ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);
369c4646bacSPeter Brune     }
370e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
371536b137fSBarry Smith   } else if (issaws) {
372d45a07a7SBarry Smith     PetscMPIInt rank;
3732657e9d9SBarry Smith     const char *name;
374d45a07a7SBarry Smith 
3752657e9d9SBarry Smith     ierr = PetscObjectGetName((PetscObject)snes,&name);CHKERRQ(ierr);
376d45a07a7SBarry Smith     ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
377d45a07a7SBarry Smith     if (!((PetscObject)snes)->amsmem && !rank) {
378d45a07a7SBarry Smith       char       dir[1024];
379d45a07a7SBarry Smith 
380e04113cfSBarry Smith       ierr = PetscObjectViewSAWs((PetscObject)snes,viewer);CHKERRQ(ierr);
3812657e9d9SBarry Smith       ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/its",name);CHKERRQ(ierr);
3822657e9d9SBarry Smith       PetscStackCallSAWs(SAWs_Register,(dir,&snes->iter,1,SAWs_READ,SAWs_INT));
383bfb97211SBarry Smith       if (!snes->conv_hist) {
384a0931e03SBarry Smith         ierr = SNESSetConvergenceHistory(snes,NULL,NULL,PETSC_DECIDE,PETSC_TRUE);CHKERRQ(ierr);
385bfb97211SBarry Smith       }
3862657e9d9SBarry Smith       ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/conv_hist",name);CHKERRQ(ierr);
3872657e9d9SBarry Smith       PetscStackCallSAWs(SAWs_Register,(dir,snes->conv_hist,10,SAWs_READ,SAWs_DOUBLE));
388f05ece33SBarry Smith     }
389bfb97211SBarry Smith #endif
39072a02f06SBarry Smith   }
39172a02f06SBarry Smith   if (snes->linesearch) {
3927601faf0SJed Brown     ierr = SNESGetLineSearch(snes, &linesearch);CHKERRQ(ierr);
39385928a98SStefano Zampini     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
39472a02f06SBarry Smith     ierr = SNESLineSearchView(linesearch, viewer);CHKERRQ(ierr);
39585928a98SStefano Zampini     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
39619bcc07fSBarry Smith   }
397efd4aadfSBarry Smith   if (snes->npc && snes->usesnpc) {
39885928a98SStefano Zampini     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
399efd4aadfSBarry Smith     ierr = SNESView(snes->npc, viewer);CHKERRQ(ierr);
40085928a98SStefano Zampini     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
4014a0c5b0cSMatthew G Knepley   }
4022d53ad75SBarry Smith   ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
4032d53ad75SBarry Smith   ierr = DMGetDMSNES(snes->dm,&dmsnes);CHKERRQ(ierr);
4042d53ad75SBarry Smith   ierr = DMSNESView(dmsnes, viewer);CHKERRQ(ierr);
4052d53ad75SBarry Smith   ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
4062c155ee1SBarry Smith   if (snes->usesksp) {
4072c155ee1SBarry Smith     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
40885928a98SStefano Zampini     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
40994b7f48cSBarry Smith     ierr = KSPView(ksp,viewer);CHKERRQ(ierr);
41085928a98SStefano Zampini     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
4112c155ee1SBarry Smith   }
41272a02f06SBarry Smith   if (isdraw) {
41372a02f06SBarry Smith     PetscDraw draw;
41472a02f06SBarry Smith     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
41572a02f06SBarry Smith     ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr);
4167f1410a3SPeter Brune   }
4173a40ed3dSBarry Smith   PetscFunctionReturn(0);
4189b94acceSBarry Smith }
4199b94acceSBarry Smith 
42076b2cf59SMatthew Knepley /*
42176b2cf59SMatthew Knepley   We retain a list of functions that also take SNES command
42276b2cf59SMatthew Knepley   line options. These are called at the end SNESSetFromOptions()
42376b2cf59SMatthew Knepley */
42476b2cf59SMatthew Knepley #define MAXSETFROMOPTIONS 5
425a7cc72afSBarry Smith static PetscInt numberofsetfromoptions;
4266849ba73SBarry Smith static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES);
42776b2cf59SMatthew Knepley 
428ac226902SBarry Smith /*@C
42976b2cf59SMatthew Knepley   SNESAddOptionsChecker - Adds an additional function to check for SNES options.
43076b2cf59SMatthew Knepley 
43176b2cf59SMatthew Knepley   Not Collective
43276b2cf59SMatthew Knepley 
43376b2cf59SMatthew Knepley   Input Parameter:
43476b2cf59SMatthew Knepley . snescheck - function that checks for options
43576b2cf59SMatthew Knepley 
43676b2cf59SMatthew Knepley   Level: developer
43776b2cf59SMatthew Knepley 
43876b2cf59SMatthew Knepley .seealso: SNESSetFromOptions()
43976b2cf59SMatthew Knepley @*/
4407087cfbeSBarry Smith PetscErrorCode  SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES))
44176b2cf59SMatthew Knepley {
44276b2cf59SMatthew Knepley   PetscFunctionBegin;
443f23aa3ddSBarry Smith   if (numberofsetfromoptions >= MAXSETFROMOPTIONS) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS);
44476b2cf59SMatthew Knepley   othersetfromoptions[numberofsetfromoptions++] = snescheck;
44576b2cf59SMatthew Knepley   PetscFunctionReturn(0);
44676b2cf59SMatthew Knepley }
44776b2cf59SMatthew Knepley 
44825acbd8eSLisandro Dalcin PETSC_INTERN PetscErrorCode SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*);
449aa3661deSLisandro Dalcin 
450ace3abfcSBarry Smith static PetscErrorCode SNESSetUpMatrixFree_Private(SNES snes, PetscBool hasOperator, PetscInt version)
451aa3661deSLisandro Dalcin {
452aa3661deSLisandro Dalcin   Mat            J;
453aa3661deSLisandro Dalcin   KSP            ksp;
454aa3661deSLisandro Dalcin   PC             pc;
455ace3abfcSBarry Smith   PetscBool      match;
456aa3661deSLisandro Dalcin   PetscErrorCode ierr;
457895c21f2SBarry Smith   MatNullSpace   nullsp;
458aa3661deSLisandro Dalcin 
459aa3661deSLisandro Dalcin   PetscFunctionBegin;
4600700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
461aa3661deSLisandro Dalcin 
46298613b67SLisandro Dalcin   if (!snes->vec_func && (snes->jacobian || snes->jacobian_pre)) {
46398613b67SLisandro Dalcin     Mat A = snes->jacobian, B = snes->jacobian_pre;
4642a7a6963SBarry Smith     ierr = MatCreateVecs(A ? A : B, NULL,&snes->vec_func);CHKERRQ(ierr);
46598613b67SLisandro Dalcin   }
46698613b67SLisandro Dalcin 
467aa3661deSLisandro Dalcin   if (version == 1) {
468aa3661deSLisandro Dalcin     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
46998613b67SLisandro Dalcin     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
4709c6ac3b3SBarry Smith     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
471aa3661deSLisandro Dalcin   } else if (version == 2) {
472e32f2f54SBarry Smith     if (!snes->vec_func) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"SNESSetFunction() must be called first");
473570b7f6dSBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_REAL_SINGLE) && !defined(PETSC_USE_REAL___FLOAT128) && !defined(PETSC_USE_REAL___FP16)
474aa3661deSLisandro Dalcin     ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_func,&J);CHKERRQ(ierr);
475aa3661deSLisandro Dalcin #else
476e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP, "matrix-free operator rutines (version 2)");
477aa3661deSLisandro Dalcin #endif
478a8248277SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "matrix-free operator rutines, only version 1 and 2");
479aa3661deSLisandro Dalcin 
480895c21f2SBarry Smith   /* attach any user provided null space that was on Amat to the newly created matrix free matrix */
481895c21f2SBarry Smith   if (snes->jacobian) {
482895c21f2SBarry Smith     ierr = MatGetNullSpace(snes->jacobian,&nullsp);CHKERRQ(ierr);
483895c21f2SBarry Smith     if (nullsp) {
484895c21f2SBarry Smith       ierr = MatSetNullSpace(J,nullsp);CHKERRQ(ierr);
485895c21f2SBarry Smith     }
486895c21f2SBarry Smith   }
487895c21f2SBarry Smith 
488aa3661deSLisandro Dalcin   ierr = PetscInfo1(snes,"Setting default matrix-free operator routines (version %D)\n", version);CHKERRQ(ierr);
489d3462f78SMatthew Knepley   if (hasOperator) {
4903232da50SPeter Brune 
491aa3661deSLisandro Dalcin     /* This version replaces the user provided Jacobian matrix with a
492aa3661deSLisandro Dalcin        matrix-free version but still employs the user-provided preconditioner matrix. */
493aa3661deSLisandro Dalcin     ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);
494aa3661deSLisandro Dalcin   } else {
495aa3661deSLisandro Dalcin     /* This version replaces both the user-provided Jacobian and the user-
4963232da50SPeter Brune      provided preconditioner Jacobian with the default matrix free version. */
497efd4aadfSBarry Smith     if ((snes->npcside== PC_LEFT) && snes->npc) {
498d728fb7dSPeter Brune       if (!snes->jacobian){ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);}
499172a4300SPeter Brune     } else {
50028a52e04SBarry Smith       ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,0);CHKERRQ(ierr);
501172a4300SPeter Brune     }
502aa3661deSLisandro Dalcin     /* Force no preconditioner */
503aa3661deSLisandro Dalcin     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
504aa3661deSLisandro Dalcin     ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
505251f4c67SDmitry Karpeev     ierr = PetscObjectTypeCompare((PetscObject)pc,PCSHELL,&match);CHKERRQ(ierr);
506aa3661deSLisandro Dalcin     if (!match) {
507aa3661deSLisandro Dalcin       ierr = PetscInfo(snes,"Setting default matrix-free preconditioner routines\nThat is no preconditioner is being used\n");CHKERRQ(ierr);
508aa3661deSLisandro Dalcin       ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr);
509aa3661deSLisandro Dalcin     }
510aa3661deSLisandro Dalcin   }
5116bf464f9SBarry Smith   ierr = MatDestroy(&J);CHKERRQ(ierr);
512aa3661deSLisandro Dalcin   PetscFunctionReturn(0);
513aa3661deSLisandro Dalcin }
514aa3661deSLisandro Dalcin 
515dfe15315SJed Brown static PetscErrorCode DMRestrictHook_SNESVecSol(DM dmfine,Mat Restrict,Vec Rscale,Mat Inject,DM dmcoarse,void *ctx)
516dfe15315SJed Brown {
517dfe15315SJed Brown   SNES           snes = (SNES)ctx;
518dfe15315SJed Brown   PetscErrorCode ierr;
5190298fd71SBarry Smith   Vec            Xfine,Xfine_named = NULL,Xcoarse;
520dfe15315SJed Brown 
521dfe15315SJed Brown   PetscFunctionBegin;
52216ebb321SJed Brown   if (PetscLogPrintInfo) {
52316ebb321SJed Brown     PetscInt finelevel,coarselevel,fineclevel,coarseclevel;
52416ebb321SJed Brown     ierr = DMGetRefineLevel(dmfine,&finelevel);CHKERRQ(ierr);
52516ebb321SJed Brown     ierr = DMGetCoarsenLevel(dmfine,&fineclevel);CHKERRQ(ierr);
52616ebb321SJed Brown     ierr = DMGetRefineLevel(dmcoarse,&coarselevel);CHKERRQ(ierr);
52716ebb321SJed Brown     ierr = DMGetCoarsenLevel(dmcoarse,&coarseclevel);CHKERRQ(ierr);
52816ebb321SJed Brown     ierr = PetscInfo4(dmfine,"Restricting SNES solution vector from level %D-%D to level %D-%D\n",finelevel,fineclevel,coarselevel,coarseclevel);CHKERRQ(ierr);
52916ebb321SJed Brown   }
530dfe15315SJed Brown   if (dmfine == snes->dm) Xfine = snes->vec_sol;
531dfe15315SJed Brown   else {
532dfe15315SJed Brown     ierr  = DMGetNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);
533dfe15315SJed Brown     Xfine = Xfine_named;
534dfe15315SJed Brown   }
535dfe15315SJed Brown   ierr = DMGetNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr);
536907f5c5aSLawrence Mitchell   if (Inject) {
537907f5c5aSLawrence Mitchell     ierr = MatRestrict(Inject,Xfine,Xcoarse);CHKERRQ(ierr);
538907f5c5aSLawrence Mitchell   } else {
539dfe15315SJed Brown     ierr = MatRestrict(Restrict,Xfine,Xcoarse);CHKERRQ(ierr);
540dfe15315SJed Brown     ierr = VecPointwiseMult(Xcoarse,Xcoarse,Rscale);CHKERRQ(ierr);
541907f5c5aSLawrence Mitchell   }
542dfe15315SJed Brown   ierr = DMRestoreNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr);
543dfe15315SJed Brown   if (Xfine_named) {ierr = DMRestoreNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);}
544dfe15315SJed Brown   PetscFunctionReturn(0);
545dfe15315SJed Brown }
546dfe15315SJed Brown 
54716ebb321SJed Brown static PetscErrorCode DMCoarsenHook_SNESVecSol(DM dm,DM dmc,void *ctx)
54816ebb321SJed Brown {
54916ebb321SJed Brown   PetscErrorCode ierr;
55016ebb321SJed Brown 
55116ebb321SJed Brown   PetscFunctionBegin;
55216ebb321SJed Brown   ierr = DMCoarsenHookAdd(dmc,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,ctx);CHKERRQ(ierr);
55316ebb321SJed Brown   PetscFunctionReturn(0);
55416ebb321SJed Brown }
55516ebb321SJed Brown 
556a6950cb2SJed Brown /* This may be called to rediscretize the operator on levels of linear multigrid. The DM shuffle is so the user can
557a6950cb2SJed Brown  * safely call SNESGetDM() in their residual evaluation routine. */
55823ee1639SBarry Smith static PetscErrorCode KSPComputeOperators_SNES(KSP ksp,Mat A,Mat B,void *ctx)
559caa4e7f2SJed Brown {
560caa4e7f2SJed Brown   SNES           snes = (SNES)ctx;
561caa4e7f2SJed Brown   PetscErrorCode ierr;
5620298fd71SBarry Smith   Vec            X,Xnamed = NULL;
563dfe15315SJed Brown   DM             dmsave;
5644e269d77SPeter Brune   void           *ctxsave;
56525ce1634SJed Brown   PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*) = NULL;
566caa4e7f2SJed Brown 
567caa4e7f2SJed Brown   PetscFunctionBegin;
568dfe15315SJed Brown   dmsave = snes->dm;
569dfe15315SJed Brown   ierr   = KSPGetDM(ksp,&snes->dm);CHKERRQ(ierr);
570dfe15315SJed Brown   if (dmsave == snes->dm) X = snes->vec_sol; /* We are on the finest level */
571dfe15315SJed Brown   else {                                     /* We are on a coarser level, this vec was initialized using a DM restrict hook */
572dfe15315SJed Brown     ierr = DMGetNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr);
573dfe15315SJed Brown     X    = Xnamed;
5740298fd71SBarry Smith     ierr = SNESGetJacobian(snes,NULL,NULL,&jac,&ctxsave);CHKERRQ(ierr);
5754e269d77SPeter Brune     /* If the DM's don't match up, the MatFDColoring context needed for the jacobian won't match up either -- fixit. */
5768d359177SBarry Smith     if (jac == SNESComputeJacobianDefaultColor) {
5778d359177SBarry Smith       ierr = SNESSetJacobian(snes,NULL,NULL,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr);
578dfe15315SJed Brown     }
5794e269d77SPeter Brune   }
5804dde8bb0SMatthew G. Knepley   /* Make sure KSP DM has the Jacobian computation routine */
5814dde8bb0SMatthew G. Knepley   {
5824dde8bb0SMatthew G. Knepley     DMSNES sdm;
5834e269d77SPeter Brune 
5844dde8bb0SMatthew G. Knepley     ierr = DMGetDMSNES(snes->dm, &sdm);CHKERRQ(ierr);
5854dde8bb0SMatthew G. Knepley     if (!sdm->ops->computejacobian) {
5864dde8bb0SMatthew G. Knepley       ierr = DMCopyDMSNES(dmsave, snes->dm);CHKERRQ(ierr);
5874dde8bb0SMatthew G. Knepley     }
5884dde8bb0SMatthew G. Knepley   }
5892b93b426SMatthew G. Knepley   /* Compute the operators */
590d1e9a80fSBarry Smith   ierr = SNESComputeJacobian(snes,X,A,B);CHKERRQ(ierr);
5912b93b426SMatthew G. Knepley   /* Put the previous context back */
5928d359177SBarry Smith   if (snes->dm != dmsave && jac == SNESComputeJacobianDefaultColor) {
5930298fd71SBarry Smith     ierr = SNESSetJacobian(snes,NULL,NULL,jac,ctxsave);CHKERRQ(ierr);
5944e269d77SPeter Brune   }
5954e269d77SPeter Brune 
5962b93b426SMatthew G. Knepley   if (Xnamed) {ierr = DMRestoreNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr);}
597dfe15315SJed Brown   snes->dm = dmsave;
598caa4e7f2SJed Brown   PetscFunctionReturn(0);
599caa4e7f2SJed Brown }
600caa4e7f2SJed Brown 
6016cab3a1bSJed Brown /*@
6026cab3a1bSJed Brown    SNESSetUpMatrices - ensures that matrices are available for SNES, to be called by SNESSetUp_XXX()
6036cab3a1bSJed Brown 
6046cab3a1bSJed Brown    Collective
6056cab3a1bSJed Brown 
6066cab3a1bSJed Brown    Input Arguments:
6076cab3a1bSJed Brown .  snes - snes to configure
6086cab3a1bSJed Brown 
6096cab3a1bSJed Brown    Level: developer
6106cab3a1bSJed Brown 
6116cab3a1bSJed Brown .seealso: SNESSetUp()
6126cab3a1bSJed Brown @*/
6136cab3a1bSJed Brown PetscErrorCode SNESSetUpMatrices(SNES snes)
6146cab3a1bSJed Brown {
6156cab3a1bSJed Brown   PetscErrorCode ierr;
6166cab3a1bSJed Brown   DM             dm;
617942e3340SBarry Smith   DMSNES         sdm;
6186cab3a1bSJed Brown 
6196cab3a1bSJed Brown   PetscFunctionBegin;
6206cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
621942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
622ce94432eSBarry Smith   if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_PLIB,"DMSNES not properly configured");
623f5af7f23SKarl Rupp   else if (!snes->jacobian && snes->mf) {
6246cab3a1bSJed Brown     Mat  J;
6256cab3a1bSJed Brown     void *functx;
6266cab3a1bSJed Brown     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
6276cab3a1bSJed Brown     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
6286cab3a1bSJed Brown     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
6290298fd71SBarry Smith     ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr);
6303232da50SPeter Brune     ierr = SNESSetJacobian(snes,J,J,0,0);CHKERRQ(ierr);
6316cab3a1bSJed Brown     ierr = MatDestroy(&J);CHKERRQ(ierr);
632caa4e7f2SJed Brown   } else if (snes->mf_operator && !snes->jacobian_pre && !snes->jacobian) {
6336cab3a1bSJed Brown     Mat J,B;
6346cab3a1bSJed Brown     ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr);
6356cab3a1bSJed Brown     ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr);
6366cab3a1bSJed Brown     ierr = MatSetFromOptions(J);CHKERRQ(ierr);
637b412c318SBarry Smith     ierr = DMCreateMatrix(snes->dm,&B);CHKERRQ(ierr);
63806f20277SJed Brown     /* sdm->computejacobian was already set to reach here */
6390298fd71SBarry Smith     ierr = SNESSetJacobian(snes,J,B,NULL,NULL);CHKERRQ(ierr);
6406cab3a1bSJed Brown     ierr = MatDestroy(&J);CHKERRQ(ierr);
6416cab3a1bSJed Brown     ierr = MatDestroy(&B);CHKERRQ(ierr);
642caa4e7f2SJed Brown   } else if (!snes->jacobian_pre) {
6434f3e65b0SMatthew G. Knepley     PetscErrorCode (*nspconstr)(DM, PetscInt, MatNullSpace *);
6441ba9b98eSMatthew G. Knepley     PetscDS          prob;
6456cab3a1bSJed Brown     Mat              J, B;
6464f3e65b0SMatthew G. Knepley     MatNullSpace     nullspace = NULL;
6471ba9b98eSMatthew G. Knepley     PetscBool        hasPrec   = PETSC_FALSE;
6484f3e65b0SMatthew G. Knepley     PetscInt         Nf;
6491ba9b98eSMatthew G. Knepley 
6506cab3a1bSJed Brown     J    = snes->jacobian;
6511ba9b98eSMatthew G. Knepley     ierr = DMGetDS(dm, &prob);CHKERRQ(ierr);
6521ba9b98eSMatthew G. Knepley     if (prob) {ierr = PetscDSHasJacobianPreconditioner(prob, &hasPrec);CHKERRQ(ierr);}
653ec9a985fSMatthew G. Knepley     if (J)            {ierr = PetscObjectReference((PetscObject) J);CHKERRQ(ierr);}
654ec9a985fSMatthew G. Knepley     else if (hasPrec) {ierr = DMCreateMatrix(snes->dm, &J);CHKERRQ(ierr);}
655b412c318SBarry Smith     ierr = DMCreateMatrix(snes->dm, &B);CHKERRQ(ierr);
6564f3e65b0SMatthew G. Knepley     ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr);
6574f3e65b0SMatthew G. Knepley     ierr = DMGetNullSpaceConstructor(snes->dm, Nf, &nspconstr);CHKERRQ(ierr);
6584f3e65b0SMatthew G. Knepley     if (nspconstr) (*nspconstr)(snes->dm, -1, &nullspace);
6594f3e65b0SMatthew G. Knepley     ierr = MatSetNullSpace(B, nullspace);CHKERRQ(ierr);
6604f3e65b0SMatthew G. Knepley     ierr = MatNullSpaceDestroy(&nullspace);CHKERRQ(ierr);
6610298fd71SBarry Smith     ierr = SNESSetJacobian(snes, J ? J : B, B, NULL, NULL);CHKERRQ(ierr);
6621ba9b98eSMatthew G. Knepley     ierr = MatDestroy(&J);CHKERRQ(ierr);
6636cab3a1bSJed Brown     ierr = MatDestroy(&B);CHKERRQ(ierr);
6646cab3a1bSJed Brown   }
665caa4e7f2SJed Brown   {
666caa4e7f2SJed Brown     KSP ksp;
667caa4e7f2SJed Brown     ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
668caa4e7f2SJed Brown     ierr = KSPSetComputeOperators(ksp,KSPComputeOperators_SNES,snes);CHKERRQ(ierr);
66916ebb321SJed Brown     ierr = DMCoarsenHookAdd(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr);
670caa4e7f2SJed Brown   }
6716cab3a1bSJed Brown   PetscFunctionReturn(0);
6726cab3a1bSJed Brown }
6736cab3a1bSJed Brown 
674fde5950dSBarry Smith /*@C
675fde5950dSBarry Smith    SNESMonitorSetFromOptions - Sets a monitor function and viewer appropriate for the type indicated by the user
676fde5950dSBarry Smith 
677fde5950dSBarry Smith    Collective on SNES
678fde5950dSBarry Smith 
679fde5950dSBarry Smith    Input Parameters:
680fde5950dSBarry Smith +  snes - SNES object you wish to monitor
681fde5950dSBarry Smith .  name - the monitor type one is seeking
682fde5950dSBarry Smith .  help - message indicating what monitoring is done
683fde5950dSBarry Smith .  manual - manual page for the monitor
684fde5950dSBarry Smith .  monitor - the monitor function
685fde5950dSBarry 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
686fde5950dSBarry Smith 
687fde5950dSBarry Smith    Level: developer
688fde5950dSBarry Smith 
689fde5950dSBarry Smith .seealso: PetscOptionsGetViewer(), PetscOptionsGetReal(), PetscOptionsHasName(), PetscOptionsGetString(),
690fde5950dSBarry Smith           PetscOptionsGetIntArray(), PetscOptionsGetRealArray(), PetscOptionsBool()
691fde5950dSBarry Smith           PetscOptionsInt(), PetscOptionsString(), PetscOptionsReal(), PetscOptionsBool(),
692fde5950dSBarry Smith           PetscOptionsName(), PetscOptionsBegin(), PetscOptionsEnd(), PetscOptionsHead(),
693fde5950dSBarry Smith           PetscOptionsStringArray(),PetscOptionsRealArray(), PetscOptionsScalar(),
694fde5950dSBarry Smith           PetscOptionsBoolGroupBegin(), PetscOptionsBoolGroup(), PetscOptionsBoolGroupEnd(),
695fde5950dSBarry Smith           PetscOptionsFList(), PetscOptionsEList()
696fde5950dSBarry Smith @*/
697d43b4f6eSBarry Smith PetscErrorCode  SNESMonitorSetFromOptions(SNES snes,const char name[],const char help[], const char manual[],PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,PetscViewerAndFormat*),PetscErrorCode (*monitorsetup)(SNES,PetscViewerAndFormat*))
698fde5950dSBarry Smith {
699fde5950dSBarry Smith   PetscErrorCode    ierr;
700fde5950dSBarry Smith   PetscViewer       viewer;
701fde5950dSBarry Smith   PetscViewerFormat format;
702fde5950dSBarry Smith   PetscBool         flg;
703fde5950dSBarry Smith 
704fde5950dSBarry Smith   PetscFunctionBegin;
70516413a6aSBarry Smith   ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,name,&viewer,&format,&flg);CHKERRQ(ierr);
706fde5950dSBarry Smith   if (flg) {
707d43b4f6eSBarry Smith     PetscViewerAndFormat *vf;
708d43b4f6eSBarry Smith     ierr = PetscViewerAndFormatCreate(viewer,format,&vf);CHKERRQ(ierr);
709d43b4f6eSBarry Smith     ierr = PetscObjectDereference((PetscObject)viewer);CHKERRQ(ierr);
710fde5950dSBarry Smith     if (monitorsetup) {
711d43b4f6eSBarry Smith       ierr = (*monitorsetup)(snes,vf);CHKERRQ(ierr);
712fde5950dSBarry Smith     }
713d43b4f6eSBarry Smith     ierr = SNESMonitorSet(snes,(PetscErrorCode (*)(SNES,PetscInt,PetscReal,void*))monitor,vf,(PetscErrorCode (*)(void**))PetscViewerAndFormatDestroy);CHKERRQ(ierr);
714fde5950dSBarry Smith   }
715fde5950dSBarry Smith   PetscFunctionReturn(0);
716fde5950dSBarry Smith }
717fde5950dSBarry Smith 
7189b94acceSBarry Smith /*@
71994b7f48cSBarry Smith    SNESSetFromOptions - Sets various SNES and KSP parameters from user options.
7209b94acceSBarry Smith 
721c7afd0dbSLois Curfman McInnes    Collective on SNES
722c7afd0dbSLois Curfman McInnes 
7239b94acceSBarry Smith    Input Parameter:
7249b94acceSBarry Smith .  snes - the SNES context
7259b94acceSBarry Smith 
72636851e7fSLois Curfman McInnes    Options Database Keys:
727722329fbSBarry Smith +  -snes_type <type> - newtonls, newtontr, ngmres, ncg, nrichardson, qn, vi, fas, SNESType for complete list
72882738288SBarry Smith .  -snes_stol - convergence tolerance in terms of the norm
72982738288SBarry Smith                 of the change in the solution between steps
73070441072SBarry Smith .  -snes_atol <abstol> - absolute tolerance of residual norm
731b39c3a46SLois Curfman McInnes .  -snes_rtol <rtol> - relative decrease in tolerance norm from initial
732e4d06f11SPatrick Farrell .  -snes_divergence_tolerance <divtol> - if the residual goes above divtol*rnorm0, exit with divergence
733be5caee7SBarry Smith .  -snes_force_iteration <force> - force SNESSolve() to take at least one iteration
734b39c3a46SLois Curfman McInnes .  -snes_max_it <max_it> - maximum number of iterations
735b39c3a46SLois Curfman McInnes .  -snes_max_funcs <max_funcs> - maximum number of function evaluations
7364839bfe8SBarry Smith .  -snes_max_fail <max_fail> - maximum number of line search failures allowed before stopping, default is none
737ddf469c8SBarry Smith .  -snes_max_linear_solve_fail - number of linear solver failures before SNESSolve() stops
738a8054027SBarry Smith .  -snes_lag_preconditioner <lag> - how often preconditioner is rebuilt (use -1 to never rebuild)
739e35cf81dSBarry Smith .  -snes_lag_jacobian <lag> - how often Jacobian is rebuilt (use -1 to never rebuild)
740b39c3a46SLois Curfman McInnes .  -snes_trtol <trtol> - trust region tolerance
7412492ecdbSBarry Smith .  -snes_no_convergence_test - skip convergence test in nonlinear
74282738288SBarry Smith                                solver; hence iterations will continue until max_it
7431fbbfb26SLois Curfman McInnes                                or some other criterion is reached. Saves expense
74482738288SBarry Smith                                of convergence test
745fde5950dSBarry Smith .  -snes_monitor [ascii][:filename][:viewer format] - prints residual norm at each iteration. if no filename given prints to stdout
746fde5950dSBarry Smith .  -snes_monitor_solution [ascii binary draw][:filename][:viewer format] - plots solution at each iteration
747fde5950dSBarry Smith .  -snes_monitor_residual [ascii binary draw][:filename][:viewer format] - plots residual (not its norm) at each iteration
748fde5950dSBarry Smith .  -snes_monitor_solution_update [ascii binary draw][:filename][:viewer format] - plots update to solution at each iteration
7494619e776SBarry Smith .  -snes_monitor_lg_residualnorm - plots residual norm at each iteration
750459f5d12SBarry Smith .  -snes_monitor_lg_range - plots residual norm at each iteration
751e24b481bSBarry Smith .  -snes_fd - use finite differences to compute Jacobian; very slow, only for testing
752e2e60de9SPeter Brune .  -snes_fd_color - use finite differences with coloring to compute Jacobian
7535968eb51SBarry Smith .  -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration
754fee2055bSBarry Smith -  -snes_converged_reason - print the reason for convergence/divergence after each solve
75582738288SBarry Smith 
75682738288SBarry Smith     Options Database for Eisenstat-Walker method:
757fa9f3622SBarry Smith +  -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence
7584b27c08aSLois Curfman McInnes .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
75936851e7fSLois Curfman McInnes .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
76036851e7fSLois Curfman McInnes .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
76136851e7fSLois Curfman McInnes .  -snes_ksp_ew_gamma <gamma> - Sets gamma
76236851e7fSLois Curfman McInnes .  -snes_ksp_ew_alpha <alpha> - Sets alpha
76336851e7fSLois Curfman McInnes .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
76436851e7fSLois Curfman McInnes -  -snes_ksp_ew_threshold <threshold> - Sets threshold
76582738288SBarry Smith 
76611ca99fdSLois Curfman McInnes    Notes:
76711ca99fdSLois Curfman McInnes    To see all options, run your program with the -help option or consult
768a7f22e61SSatish Balay    Users-Manual: ch_snes
76983e2fdc7SBarry Smith 
77036851e7fSLois Curfman McInnes    Level: beginner
77136851e7fSLois Curfman McInnes 
7729b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database
7739b94acceSBarry Smith 
774b3cd9a81SMatthew G. Knepley .seealso: SNESSetOptionsPrefix(), SNESResetFromOptions()
7759b94acceSBarry Smith @*/
7767087cfbeSBarry Smith PetscErrorCode  SNESSetFromOptions(SNES snes)
7779b94acceSBarry Smith {
7788afaa268SBarry Smith   PetscBool      flg,pcset,persist,set;
779d8f46077SPeter Brune   PetscInt       i,indx,lag,grids;
78004d7464bSBarry Smith   const char     *deft        = SNESNEWTONLS;
78185385478SLisandro Dalcin   const char     *convtests[] = {"default","skip"};
78285385478SLisandro Dalcin   SNESKSPEW      *kctx        = NULL;
783e8105e01SRichard Katz   char           type[256], monfilename[PETSC_MAX_PATH_LEN];
78485385478SLisandro Dalcin   PetscErrorCode ierr;
785c40d0f55SPeter Brune   PCSide         pcside;
786a64e098fSPeter Brune   const char     *optionsprefix;
7879b94acceSBarry Smith 
7883a40ed3dSBarry Smith   PetscFunctionBegin;
7890700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
7900f51fdf8SToby Isaac   ierr = SNESRegisterAll();CHKERRQ(ierr);
7913194b578SJed Brown   ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr);
792639ff905SBarry Smith   if (((PetscObject)snes)->type_name) deft = ((PetscObject)snes)->type_name;
793a264d7a6SBarry Smith   ierr = PetscOptionsFList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr);
794d64ed03dSBarry Smith   if (flg) {
795186905e3SBarry Smith     ierr = SNESSetType(snes,type);CHKERRQ(ierr);
7967adad957SLisandro Dalcin   } else if (!((PetscObject)snes)->type_name) {
797186905e3SBarry Smith     ierr = SNESSetType(snes,deft);CHKERRQ(ierr);
798d64ed03dSBarry Smith   }
79994ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_stol","Stop if step length less than","SNESSetTolerances",snes->stol,&snes->stol,NULL);CHKERRQ(ierr);
80094ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_atol","Stop if function norm less than","SNESSetTolerances",snes->abstol,&snes->abstol,NULL);CHKERRQ(ierr);
801186905e3SBarry Smith 
80294ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less than","SNESSetTolerances",snes->rtol,&snes->rtol,NULL);CHKERRQ(ierr);
803e4d06f11SPatrick Farrell   ierr = PetscOptionsReal("-snes_divergence_tolerance","Stop if residual norm increases by this factor","SNESSetDivergenceTolerance",snes->divtol,&snes->divtol,NULL);CHKERRQ(ierr);
8040298fd71SBarry Smith   ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,NULL);CHKERRQ(ierr);
8050298fd71SBarry Smith   ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,NULL);CHKERRQ(ierr);
8060298fd71SBarry Smith   ierr = PetscOptionsInt("-snes_max_fail","Maximum nonlinear step failures","SNESSetMaxNonlinearStepFailures",snes->maxFailures,&snes->maxFailures,NULL);CHKERRQ(ierr);
8070298fd71SBarry Smith   ierr = PetscOptionsInt("-snes_max_linear_solve_fail","Maximum failures in linear solves allowed","SNESSetMaxLinearSolveFailures",snes->maxLinearSolveFailures,&snes->maxLinearSolveFailures,NULL);CHKERRQ(ierr);
8080298fd71SBarry Smith   ierr = PetscOptionsBool("-snes_error_if_not_converged","Generate error if solver does not converge","SNESSetErrorIfNotConverged",snes->errorifnotconverged,&snes->errorifnotconverged,NULL);CHKERRQ(ierr);
8090d6f27a8SBarry Smith   ierr = PetscOptionsBool("-snes_force_iteration","Force SNESSolve() to take at least one iteration","SNESSetForceIteration",snes->forceiteration,&snes->forceiteration,NULL);CHKERRQ(ierr);
81085385478SLisandro Dalcin 
811a8054027SBarry Smith   ierr = PetscOptionsInt("-snes_lag_preconditioner","How often to rebuild preconditioner","SNESSetLagPreconditioner",snes->lagpreconditioner,&lag,&flg);CHKERRQ(ierr);
812a8054027SBarry Smith   if (flg) {
813a8054027SBarry Smith     ierr = SNESSetLagPreconditioner(snes,lag);CHKERRQ(ierr);
814a8054027SBarry Smith   }
81537ec4e1aSPeter Brune   ierr = PetscOptionsBool("-snes_lag_preconditioner_persists","Preconditioner lagging through multiple solves","SNESSetLagPreconditionerPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr);
81637ec4e1aSPeter Brune   if (flg) {
81737ec4e1aSPeter Brune     ierr = SNESSetLagPreconditionerPersists(snes,persist);CHKERRQ(ierr);
81837ec4e1aSPeter Brune   }
819e35cf81dSBarry Smith   ierr = PetscOptionsInt("-snes_lag_jacobian","How often to rebuild Jacobian","SNESSetLagJacobian",snes->lagjacobian,&lag,&flg);CHKERRQ(ierr);
820e35cf81dSBarry Smith   if (flg) {
821e35cf81dSBarry Smith     ierr = SNESSetLagJacobian(snes,lag);CHKERRQ(ierr);
822e35cf81dSBarry Smith   }
82337ec4e1aSPeter Brune   ierr = PetscOptionsBool("-snes_lag_jacobian_persists","Jacobian lagging through multiple solves","SNESSetLagJacobianPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr);
82437ec4e1aSPeter Brune   if (flg) {
82537ec4e1aSPeter Brune     ierr = SNESSetLagJacobianPersists(snes,persist);CHKERRQ(ierr);
82637ec4e1aSPeter Brune   }
82737ec4e1aSPeter Brune 
828efd51863SBarry Smith   ierr = PetscOptionsInt("-snes_grid_sequence","Use grid sequencing to generate initial guess","SNESSetGridSequence",snes->gridsequence,&grids,&flg);CHKERRQ(ierr);
829efd51863SBarry Smith   if (flg) {
830efd51863SBarry Smith     ierr = SNESSetGridSequence(snes,grids);CHKERRQ(ierr);
831efd51863SBarry Smith   }
832a8054027SBarry Smith 
83385385478SLisandro Dalcin   ierr = PetscOptionsEList("-snes_convergence_test","Convergence test","SNESSetConvergenceTest",convtests,2,"default",&indx,&flg);CHKERRQ(ierr);
83485385478SLisandro Dalcin   if (flg) {
83585385478SLisandro Dalcin     switch (indx) {
8368d359177SBarry Smith     case 0: ierr = SNESSetConvergenceTest(snes,SNESConvergedDefault,NULL,NULL);CHKERRQ(ierr); break;
837e2a6519dSDmitry Karpeev     case 1: ierr = SNESSetConvergenceTest(snes,SNESConvergedSkip,NULL,NULL);CHKERRQ(ierr);    break;
83885385478SLisandro Dalcin     }
83985385478SLisandro Dalcin   }
84085385478SLisandro Dalcin 
841365a6726SPeter Brune   ierr = PetscOptionsEList("-snes_norm_schedule","SNES Norm schedule","SNESSetNormSchedule",SNESNormSchedules,5,"function",&indx,&flg);CHKERRQ(ierr);
842365a6726SPeter Brune   if (flg) { ierr = SNESSetNormSchedule(snes,(SNESNormSchedule)indx);CHKERRQ(ierr); }
843fdacfa88SPeter Brune 
84447073ea2SPeter Brune   ierr = PetscOptionsEList("-snes_function_type","SNES Norm schedule","SNESSetFunctionType",SNESFunctionTypes,2,"unpreconditioned",&indx,&flg);CHKERRQ(ierr);
84547073ea2SPeter Brune   if (flg) { ierr = SNESSetFunctionType(snes,(SNESFunctionType)indx);CHKERRQ(ierr); }
846186905e3SBarry Smith 
84785385478SLisandro Dalcin   kctx = (SNESKSPEW*)snes->kspconvctx;
84885385478SLisandro Dalcin 
8490298fd71SBarry Smith   ierr = PetscOptionsBool("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,NULL);CHKERRQ(ierr);
850186905e3SBarry Smith 
85194ae4db5SBarry Smith   ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,NULL);CHKERRQ(ierr);
85294ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,NULL);CHKERRQ(ierr);
85394ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,NULL);CHKERRQ(ierr);
85494ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,NULL);CHKERRQ(ierr);
85594ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,NULL);CHKERRQ(ierr);
85694ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,NULL);CHKERRQ(ierr);
85794ae4db5SBarry Smith   ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,NULL);CHKERRQ(ierr);
858186905e3SBarry Smith 
85990d69ab7SBarry Smith   flg  = PETSC_FALSE;
8608afaa268SBarry Smith   ierr = PetscOptionsBool("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",flg,&flg,&set);CHKERRQ(ierr);
8618afaa268SBarry Smith   if (set && flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);}
862eabae89aSBarry Smith 
863fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor","Monitor norm of function","SNESMonitorDefault",SNESMonitorDefault,NULL);CHKERRQ(ierr);
864fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_short","Monitor norm of function with fewer digits","SNESMonitorDefaultShort",SNESMonitorDefaultShort,NULL);CHKERRQ(ierr);
865fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_range","Monitor range of elements of function","SNESMonitorRange",SNESMonitorRange,NULL);CHKERRQ(ierr);
866eabae89aSBarry Smith 
867fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_ratio","Monitor ratios of the norm of function for consecutive steps","SNESMonitorRatio",SNESMonitorRatio,SNESMonitorRatioSetUp);CHKERRQ(ierr);
868fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_field","Monitor norm of function (split into fields)","SNESMonitorDefaultField",SNESMonitorDefaultField,NULL);CHKERRQ(ierr);
869fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution","View solution at each iteration","SNESMonitorSolution",SNESMonitorSolution,NULL);CHKERRQ(ierr);
870fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution_update","View correction at each iteration","SNESMonitorSolutionUpdate",SNESMonitorSolutionUpdate,NULL);CHKERRQ(ierr);
871fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_residual","View residual at each iteration","SNESMonitorResidual",SNESMonitorResidual,NULL);CHKERRQ(ierr);
872fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_jacupdate_spectrum","Print the change in the spectrum of the Jacobian","SNESMonitorJacUpdateSpectrum",SNESMonitorJacUpdateSpectrum,NULL);CHKERRQ(ierr);
873fde5950dSBarry Smith   ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_fields","Monitor norm of function per field","SNESMonitorSet",SNESMonitorFields,NULL);CHKERRQ(ierr);
8742db13446SMatthew G. Knepley 
8755180491cSLisandro Dalcin   ierr = PetscOptionsString("-snes_monitor_python","Use Python function","SNESMonitorSet",0,monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
8765180491cSLisandro Dalcin   if (flg) {ierr = PetscPythonMonitorSet((PetscObject)snes,monfilename);CHKERRQ(ierr);}
8775180491cSLisandro Dalcin 
878fde5950dSBarry Smith 
87990d69ab7SBarry Smith   flg  = PETSC_FALSE;
8800298fd71SBarry Smith   ierr = PetscOptionsBool("-snes_monitor_lg_residualnorm","Plot function norm at each iteration","SNESMonitorLGResidualNorm",flg,&flg,NULL);CHKERRQ(ierr);
881459f5d12SBarry Smith   if (flg) {
882d96771aaSLisandro Dalcin     PetscDrawLG ctx;
883459f5d12SBarry Smith 
8846ba87a44SLisandro Dalcin     ierr = SNESMonitorLGCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr);
885d96771aaSLisandro Dalcin     ierr = SNESMonitorSet(snes,SNESMonitorLGResidualNorm,ctx,(PetscErrorCode (*)(void**))PetscDrawLGDestroy);CHKERRQ(ierr);
886459f5d12SBarry Smith   }
88790d69ab7SBarry Smith   flg  = PETSC_FALSE;
8880298fd71SBarry Smith   ierr = PetscOptionsBool("-snes_monitor_lg_range","Plot function range at each iteration","SNESMonitorLGRange",flg,&flg,NULL);CHKERRQ(ierr);
889459f5d12SBarry Smith   if (flg) {
890459f5d12SBarry Smith     PetscViewer ctx;
891e24b481bSBarry Smith 
8926ba87a44SLisandro Dalcin     ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr);
893459f5d12SBarry Smith     ierr = SNESMonitorSet(snes,SNESMonitorLGRange,ctx,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
894459f5d12SBarry Smith   }
8952e7541e6SPeter Brune 
8962e7541e6SPeter Brune 
897cc0c4584SMatthew G. Knepley 
89890d69ab7SBarry Smith   flg  = PETSC_FALSE;
8998d359177SBarry Smith   ierr = PetscOptionsBool("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESComputeJacobianDefault",flg,&flg,NULL);CHKERRQ(ierr);
9004b27c08aSLois Curfman McInnes   if (flg) {
9016cab3a1bSJed Brown     void    *functx;
902b1f624c7SBarry Smith     DM      dm;
903b1f624c7SBarry Smith     DMSNES  sdm;
904b1f624c7SBarry Smith     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
905b1f624c7SBarry Smith     ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
906b1f624c7SBarry Smith     sdm->jacobianctx = NULL;
9070298fd71SBarry Smith     ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr);
9088d359177SBarry Smith     ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefault,functx);CHKERRQ(ierr);
909ae15b995SBarry Smith     ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr);
9109b94acceSBarry Smith   }
911639f9d9dSBarry Smith 
91244848bc4SPeter Brune   flg  = PETSC_FALSE;
9138d359177SBarry Smith   ierr = PetscOptionsBool("-snes_fd_function","Use finite differences (slow) to compute function from user objective","SNESObjectiveComputeFunctionDefaultFD",flg,&flg,NULL);CHKERRQ(ierr);
91497584545SPeter Brune   if (flg) {
9158d359177SBarry Smith     ierr = SNESSetFunction(snes,NULL,SNESObjectiveComputeFunctionDefaultFD,NULL);CHKERRQ(ierr);
91697584545SPeter Brune   }
91797584545SPeter Brune 
91897584545SPeter Brune   flg  = PETSC_FALSE;
9198d359177SBarry Smith   ierr = PetscOptionsBool("-snes_fd_color","Use finite differences with coloring to compute Jacobian","SNESComputeJacobianDefaultColor",flg,&flg,NULL);CHKERRQ(ierr);
92044848bc4SPeter Brune   if (flg) {
921c52e227fSPeter Brune     DM             dm;
922c52e227fSPeter Brune     DMSNES         sdm;
923c52e227fSPeter Brune     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
924aace71b7SPeter Brune     ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
925aace71b7SPeter Brune     sdm->jacobianctx = NULL;
9268d359177SBarry Smith     ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr);
92744848bc4SPeter Brune     ierr = PetscInfo(snes,"Setting default finite difference coloring Jacobian matrix\n");CHKERRQ(ierr);
92844848bc4SPeter Brune   }
92944848bc4SPeter Brune 
930aa3661deSLisandro Dalcin   flg  = PETSC_FALSE;
931f871313dSBarry 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);
932d8f46077SPeter Brune   if (flg && snes->mf_operator) {
933a8248277SBarry Smith     snes->mf_operator = PETSC_TRUE;
934d8f46077SPeter Brune     snes->mf          = PETSC_TRUE;
935a8248277SBarry Smith   }
936aa3661deSLisandro Dalcin   flg  = PETSC_FALSE;
937f871313dSBarry Smith   ierr = PetscOptionsBool("-snes_mf","Use a Matrix-Free Jacobian with no preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf,&flg);CHKERRQ(ierr);
938d8f46077SPeter Brune   if (!flg && snes->mf_operator) snes->mf = PETSC_TRUE;
939d8f46077SPeter Brune   ierr = PetscOptionsInt("-snes_mf_version","Matrix-Free routines version 1 or 2","None",snes->mf_version,&snes->mf_version,0);CHKERRQ(ierr);
940d28543b3SPeter Brune 
941c40d0f55SPeter Brune   flg  = PETSC_FALSE;
942be95d8f1SBarry Smith   ierr = SNESGetNPCSide(snes,&pcside);CHKERRQ(ierr);
943be95d8f1SBarry Smith   ierr = PetscOptionsEnum("-snes_npc_side","SNES nonlinear preconditioner side","SNESSetNPCSide",PCSides,(PetscEnum)pcside,(PetscEnum*)&pcside,&flg);CHKERRQ(ierr);
944be95d8f1SBarry Smith   if (flg) {ierr = SNESSetNPCSide(snes,pcside);CHKERRQ(ierr);}
945c40d0f55SPeter Brune 
946e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS)
9478a70d858SHong Zhang   /*
9488a70d858SHong Zhang     Publish convergence information using SAWs
9498a70d858SHong Zhang   */
9508a70d858SHong Zhang   flg  = PETSC_FALSE;
9518a70d858SHong Zhang   ierr = PetscOptionsBool("-snes_monitor_saws","Publish SNES progress using SAWs","SNESMonitorSet",flg,&flg,NULL);CHKERRQ(ierr);
9528a70d858SHong Zhang   if (flg) {
9538a70d858SHong Zhang     void *ctx;
9548a70d858SHong Zhang     ierr = SNESMonitorSAWsCreate(snes,&ctx);CHKERRQ(ierr);
9558a70d858SHong Zhang     ierr = SNESMonitorSet(snes,SNESMonitorSAWs,ctx,SNESMonitorSAWsDestroy);CHKERRQ(ierr);
9568a70d858SHong Zhang   }
9578a70d858SHong Zhang #endif
9588a70d858SHong Zhang #if defined(PETSC_HAVE_SAWS)
959b90c6cbeSBarry Smith   {
960b90c6cbeSBarry Smith   PetscBool set;
961b90c6cbeSBarry Smith   flg  = PETSC_FALSE;
9628a70d858SHong Zhang   ierr = PetscOptionsBool("-snes_saws_block","Block for SAWs at end of SNESSolve","PetscObjectSAWsBlock",((PetscObject)snes)->amspublishblock,&flg,&set);CHKERRQ(ierr);
963b90c6cbeSBarry Smith   if (set) {
964e04113cfSBarry Smith     ierr = PetscObjectSAWsSetBlock((PetscObject)snes,flg);CHKERRQ(ierr);
965b90c6cbeSBarry Smith   }
966b90c6cbeSBarry Smith   }
967b90c6cbeSBarry Smith #endif
968b90c6cbeSBarry Smith 
96976b2cf59SMatthew Knepley   for (i = 0; i < numberofsetfromoptions; i++) {
97076b2cf59SMatthew Knepley     ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr);
97176b2cf59SMatthew Knepley   }
97276b2cf59SMatthew Knepley 
973e7788613SBarry Smith   if (snes->ops->setfromoptions) {
974e55864a3SBarry Smith     ierr = (*snes->ops->setfromoptions)(PetscOptionsObject,snes);CHKERRQ(ierr);
975639f9d9dSBarry Smith   }
9765d973c19SBarry Smith 
9775d973c19SBarry Smith   /* process any options handlers added with PetscObjectAddOptionsHandler() */
9780633abcbSJed Brown   ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject)snes);CHKERRQ(ierr);
979b0a32e0cSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
9804bbc92c1SBarry Smith 
9819e764e56SPeter Brune   if (!snes->linesearch) {
9827601faf0SJed Brown     ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr);
9839e764e56SPeter Brune   }
984f1c6b773SPeter Brune   ierr = SNESLineSearchSetFromOptions(snes->linesearch);CHKERRQ(ierr);
9859e764e56SPeter Brune 
9866991f827SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
9876991f827SBarry Smith   ierr = KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre);CHKERRQ(ierr);
9886991f827SBarry Smith   ierr = KSPSetFromOptions(snes->ksp);CHKERRQ(ierr);
9896991f827SBarry Smith 
990be95d8f1SBarry Smith   /* if someone has set the SNES NPC type, create it. */
99151e86f29SPeter Brune   ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr);
992c5929fdfSBarry Smith   ierr = PetscOptionsHasName(((PetscObject)snes)->options,optionsprefix, "-npc_snes_type", &pcset);CHKERRQ(ierr);
993efd4aadfSBarry Smith   if (pcset && (!snes->npc)) {
994efd4aadfSBarry Smith     ierr = SNESGetNPC(snes, &snes->npc);CHKERRQ(ierr);
99551e86f29SPeter Brune   }
996b3cd9a81SMatthew G. Knepley   snes->setfromoptionscalled++;
997b3cd9a81SMatthew G. Knepley   PetscFunctionReturn(0);
998b3cd9a81SMatthew G. Knepley }
999b3cd9a81SMatthew G. Knepley 
1000b3cd9a81SMatthew G. Knepley /*@
1001b3cd9a81SMatthew G. Knepley    SNESResetFromOptions - Sets various SNES and KSP parameters from user options ONLY if the SNES was previously set from options
1002b3cd9a81SMatthew G. Knepley 
1003b3cd9a81SMatthew G. Knepley    Collective on SNES
1004b3cd9a81SMatthew G. Knepley 
1005b3cd9a81SMatthew G. Knepley    Input Parameter:
1006b3cd9a81SMatthew G. Knepley .  snes - the SNES context
1007b3cd9a81SMatthew G. Knepley 
1008b3cd9a81SMatthew G. Knepley    Level: beginner
1009b3cd9a81SMatthew G. Knepley 
1010b3cd9a81SMatthew G. Knepley .keywords: SNES, nonlinear, set, options, database
1011b3cd9a81SMatthew G. Knepley 
1012b3cd9a81SMatthew G. Knepley .seealso: SNESSetFromOptions(), SNESSetOptionsPrefix()
1013b3cd9a81SMatthew G. Knepley @*/
1014b3cd9a81SMatthew G. Knepley PetscErrorCode SNESResetFromOptions(SNES snes)
1015b3cd9a81SMatthew G. Knepley {
1016b3cd9a81SMatthew G. Knepley   PetscErrorCode ierr;
1017b3cd9a81SMatthew G. Knepley 
1018b3cd9a81SMatthew G. Knepley   PetscFunctionBegin;
1019b3cd9a81SMatthew G. Knepley   if (snes->setfromoptionscalled) {ierr = SNESSetFromOptions(snes);CHKERRQ(ierr);}
10203a40ed3dSBarry Smith   PetscFunctionReturn(0);
10219b94acceSBarry Smith }
10229b94acceSBarry Smith 
1023bb9467b5SJed Brown /*@C
1024d25893d9SBarry Smith    SNESSetComputeApplicationContext - Sets an optional function to compute a user-defined context for
1025d25893d9SBarry Smith    the nonlinear solvers.
1026d25893d9SBarry Smith 
1027d25893d9SBarry Smith    Logically Collective on SNES
1028d25893d9SBarry Smith 
1029d25893d9SBarry Smith    Input Parameters:
1030d25893d9SBarry Smith +  snes - the SNES context
1031d25893d9SBarry Smith .  compute - function to compute the context
1032d25893d9SBarry Smith -  destroy - function to destroy the context
1033d25893d9SBarry Smith 
1034d25893d9SBarry Smith    Level: intermediate
1035d25893d9SBarry Smith 
1036bb9467b5SJed Brown    Notes:
1037bb9467b5SJed Brown    This function is currently not available from Fortran.
1038bb9467b5SJed Brown 
1039d25893d9SBarry Smith .keywords: SNES, nonlinear, set, application, context
1040d25893d9SBarry Smith 
1041d25893d9SBarry Smith .seealso: SNESGetApplicationContext(), SNESSetComputeApplicationContext(), SNESGetApplicationContext()
1042d25893d9SBarry Smith @*/
1043d25893d9SBarry Smith PetscErrorCode  SNESSetComputeApplicationContext(SNES snes,PetscErrorCode (*compute)(SNES,void**),PetscErrorCode (*destroy)(void**))
1044d25893d9SBarry Smith {
1045d25893d9SBarry Smith   PetscFunctionBegin;
1046d25893d9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1047d25893d9SBarry Smith   snes->ops->usercompute = compute;
1048d25893d9SBarry Smith   snes->ops->userdestroy = destroy;
1049d25893d9SBarry Smith   PetscFunctionReturn(0);
1050d25893d9SBarry Smith }
1051a847f771SSatish Balay 
1052b07ff414SBarry Smith /*@
10539b94acceSBarry Smith    SNESSetApplicationContext - Sets the optional user-defined context for
10549b94acceSBarry Smith    the nonlinear solvers.
10559b94acceSBarry Smith 
10563f9fe445SBarry Smith    Logically Collective on SNES
1057fee21e36SBarry Smith 
1058c7afd0dbSLois Curfman McInnes    Input Parameters:
1059c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1060c7afd0dbSLois Curfman McInnes -  usrP - optional user context
1061c7afd0dbSLois Curfman McInnes 
106236851e7fSLois Curfman McInnes    Level: intermediate
106336851e7fSLois Curfman McInnes 
106495452b02SPatrick Sanan    Fortran Notes:
106595452b02SPatrick Sanan     To use this from Fortran you must write a Fortran interface definition for this
1066daf670e6SBarry Smith     function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument.
1067daf670e6SBarry Smith 
10689b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context
10699b94acceSBarry Smith 
1070ecaffddaSVictor Eijkhout .seealso: SNESGetApplicationContext()
10719b94acceSBarry Smith @*/
10727087cfbeSBarry Smith PetscErrorCode  SNESSetApplicationContext(SNES snes,void *usrP)
10739b94acceSBarry Smith {
10741b2093e4SBarry Smith   PetscErrorCode ierr;
1075b07ff414SBarry Smith   KSP            ksp;
10761b2093e4SBarry Smith 
10773a40ed3dSBarry Smith   PetscFunctionBegin;
10780700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1079b07ff414SBarry Smith   ierr       = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
1080b07ff414SBarry Smith   ierr       = KSPSetApplicationContext(ksp,usrP);CHKERRQ(ierr);
10819b94acceSBarry Smith   snes->user = usrP;
10823a40ed3dSBarry Smith   PetscFunctionReturn(0);
10839b94acceSBarry Smith }
108474679c65SBarry Smith 
1085b07ff414SBarry Smith /*@
10869b94acceSBarry Smith    SNESGetApplicationContext - Gets the user-defined context for the
10879b94acceSBarry Smith    nonlinear solvers.
10889b94acceSBarry Smith 
1089c7afd0dbSLois Curfman McInnes    Not Collective
1090c7afd0dbSLois Curfman McInnes 
10919b94acceSBarry Smith    Input Parameter:
10929b94acceSBarry Smith .  snes - SNES context
10939b94acceSBarry Smith 
10949b94acceSBarry Smith    Output Parameter:
10959b94acceSBarry Smith .  usrP - user context
10969b94acceSBarry Smith 
109795452b02SPatrick Sanan    Fortran Notes:
109895452b02SPatrick Sanan     To use this from Fortran you must write a Fortran interface definition for this
1099daf670e6SBarry Smith     function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument.
1100daf670e6SBarry Smith 
110136851e7fSLois Curfman McInnes    Level: intermediate
110236851e7fSLois Curfman McInnes 
11039b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context
11049b94acceSBarry Smith 
11059b94acceSBarry Smith .seealso: SNESSetApplicationContext()
11069b94acceSBarry Smith @*/
1107e71120c6SJed Brown PetscErrorCode  SNESGetApplicationContext(SNES snes,void *usrP)
11089b94acceSBarry Smith {
11093a40ed3dSBarry Smith   PetscFunctionBegin;
11100700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1111e71120c6SJed Brown   *(void**)usrP = snes->user;
11123a40ed3dSBarry Smith   PetscFunctionReturn(0);
11139b94acceSBarry Smith }
111474679c65SBarry Smith 
11159b94acceSBarry Smith /*@
11163565c898SBarry Smith    SNESSetUseMatrixFree - indicates that SNES should use matrix free finite difference matrix vector products internally to apply
11173565c898SBarry Smith                           the Jacobian.
11183565c898SBarry Smith 
11193565c898SBarry Smith    Collective on SNES
11203565c898SBarry Smith 
11213565c898SBarry Smith    Input Parameters:
11223565c898SBarry Smith +  snes - SNES context
11233565c898SBarry Smith .  mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored
11243565c898SBarry Smith -  mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used
11253565c898SBarry Smith 
11263565c898SBarry Smith    Options Database:
11273565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator
11283565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator
11293565c898SBarry Smith 
11303565c898SBarry Smith    Level: intermediate
11313565c898SBarry Smith 
11323565c898SBarry Smith .keywords: SNES, nonlinear, get, iteration, number,
11333565c898SBarry Smith 
11343565c898SBarry Smith .seealso:   SNESGetUseMatrixFree(), MatCreateSNESMF()
11353565c898SBarry Smith @*/
11363565c898SBarry Smith PetscErrorCode  SNESSetUseMatrixFree(SNES snes,PetscBool mf_operator,PetscBool mf)
11373565c898SBarry Smith {
11383565c898SBarry Smith   PetscFunctionBegin;
11393565c898SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
114088b4c220SStefano Zampini   PetscValidLogicalCollectiveBool(snes,mf_operator,2);
114188b4c220SStefano Zampini   PetscValidLogicalCollectiveBool(snes,mf,3);
11423565c898SBarry Smith   if (mf && !mf_operator) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"If using mf must also use mf_operator");
11433565c898SBarry Smith   snes->mf          = mf;
11443565c898SBarry Smith   snes->mf_operator = mf_operator;
11453565c898SBarry Smith   PetscFunctionReturn(0);
11463565c898SBarry Smith }
11473565c898SBarry Smith 
11483565c898SBarry Smith /*@
11493565c898SBarry Smith    SNESGetUseMatrixFree - indicates if the SNES uses matrix free finite difference matrix vector products to apply
11503565c898SBarry Smith                           the Jacobian.
11513565c898SBarry Smith 
11523565c898SBarry Smith    Collective on SNES
11533565c898SBarry Smith 
11543565c898SBarry Smith    Input Parameter:
11553565c898SBarry Smith .  snes - SNES context
11563565c898SBarry Smith 
11573565c898SBarry Smith    Output Parameters:
11583565c898SBarry Smith +  mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored
11593565c898SBarry Smith -  mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used
11603565c898SBarry Smith 
11613565c898SBarry Smith    Options Database:
11623565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator
11633565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator
11643565c898SBarry Smith 
11653565c898SBarry Smith    Level: intermediate
11663565c898SBarry Smith 
11673565c898SBarry Smith .keywords: SNES, nonlinear, get, iteration, number,
11683565c898SBarry Smith 
11693565c898SBarry Smith .seealso:   SNESSetUseMatrixFree(), MatCreateSNESMF()
11703565c898SBarry Smith @*/
11713565c898SBarry Smith PetscErrorCode  SNESGetUseMatrixFree(SNES snes,PetscBool *mf_operator,PetscBool *mf)
11723565c898SBarry Smith {
11733565c898SBarry Smith   PetscFunctionBegin;
11743565c898SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
11753565c898SBarry Smith   if (mf)          *mf          = snes->mf;
11763565c898SBarry Smith   if (mf_operator) *mf_operator = snes->mf_operator;
11773565c898SBarry Smith   PetscFunctionReturn(0);
11783565c898SBarry Smith }
11793565c898SBarry Smith 
11803565c898SBarry Smith /*@
1181c8228a4eSBarry Smith    SNESGetIterationNumber - Gets the number of nonlinear iterations completed
1182c8228a4eSBarry Smith    at this time.
11839b94acceSBarry Smith 
1184c7afd0dbSLois Curfman McInnes    Not Collective
1185c7afd0dbSLois Curfman McInnes 
11869b94acceSBarry Smith    Input Parameter:
11879b94acceSBarry Smith .  snes - SNES context
11889b94acceSBarry Smith 
11899b94acceSBarry Smith    Output Parameter:
11909b94acceSBarry Smith .  iter - iteration number
11919b94acceSBarry Smith 
1192c8228a4eSBarry Smith    Notes:
1193c8228a4eSBarry Smith    For example, during the computation of iteration 2 this would return 1.
1194c8228a4eSBarry Smith 
1195c8228a4eSBarry Smith    This is useful for using lagged Jacobians (where one does not recompute the
119608405cd6SLois Curfman McInnes    Jacobian at each SNES iteration). For example, the code
119708405cd6SLois Curfman McInnes .vb
119808405cd6SLois Curfman McInnes       ierr = SNESGetIterationNumber(snes,&it);
119908405cd6SLois Curfman McInnes       if (!(it % 2)) {
120008405cd6SLois Curfman McInnes         [compute Jacobian here]
120108405cd6SLois Curfman McInnes       }
120208405cd6SLois Curfman McInnes .ve
1203c8228a4eSBarry Smith    can be used in your ComputeJacobian() function to cause the Jacobian to be
120408405cd6SLois Curfman McInnes    recomputed every second SNES iteration.
1205c8228a4eSBarry Smith 
1206c04deec6SBarry Smith    After the SNES solve is complete this will return the number of nonlinear iterations used.
1207c04deec6SBarry Smith 
120836851e7fSLois Curfman McInnes    Level: intermediate
120936851e7fSLois Curfman McInnes 
12102b668275SBarry Smith .keywords: SNES, nonlinear, get, iteration, number,
12112b668275SBarry Smith 
121271dbe336SPeter Brune .seealso:   SNESGetLinearSolveIterations()
12139b94acceSBarry Smith @*/
12147087cfbeSBarry Smith PetscErrorCode  SNESGetIterationNumber(SNES snes,PetscInt *iter)
12159b94acceSBarry Smith {
12163a40ed3dSBarry Smith   PetscFunctionBegin;
12170700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
12184482741eSBarry Smith   PetscValidIntPointer(iter,2);
12199b94acceSBarry Smith   *iter = snes->iter;
12203a40ed3dSBarry Smith   PetscFunctionReturn(0);
12219b94acceSBarry Smith }
122274679c65SBarry Smith 
1223360c497dSPeter Brune /*@
1224360c497dSPeter Brune    SNESSetIterationNumber - Sets the current iteration number.
1225360c497dSPeter Brune 
1226360c497dSPeter Brune    Not Collective
1227360c497dSPeter Brune 
1228360c497dSPeter Brune    Input Parameter:
1229360c497dSPeter Brune .  snes - SNES context
1230360c497dSPeter Brune .  iter - iteration number
1231360c497dSPeter Brune 
1232360c497dSPeter Brune    Level: developer
1233360c497dSPeter Brune 
1234360c497dSPeter Brune .keywords: SNES, nonlinear, set, iteration, number,
1235360c497dSPeter Brune 
123671dbe336SPeter Brune .seealso:   SNESGetLinearSolveIterations()
1237360c497dSPeter Brune @*/
1238360c497dSPeter Brune PetscErrorCode  SNESSetIterationNumber(SNES snes,PetscInt iter)
1239360c497dSPeter Brune {
1240360c497dSPeter Brune   PetscErrorCode ierr;
1241360c497dSPeter Brune 
1242360c497dSPeter Brune   PetscFunctionBegin;
1243360c497dSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1244e04113cfSBarry Smith   ierr       = PetscObjectSAWsTakeAccess((PetscObject)snes);CHKERRQ(ierr);
1245360c497dSPeter Brune   snes->iter = iter;
1246e04113cfSBarry Smith   ierr       = PetscObjectSAWsGrantAccess((PetscObject)snes);CHKERRQ(ierr);
1247360c497dSPeter Brune   PetscFunctionReturn(0);
1248360c497dSPeter Brune }
1249360c497dSPeter Brune 
12509b94acceSBarry Smith /*@
1251b850b91aSLisandro Dalcin    SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps
12529b94acceSBarry Smith    attempted by the nonlinear solver.
12539b94acceSBarry Smith 
1254c7afd0dbSLois Curfman McInnes    Not Collective
1255c7afd0dbSLois Curfman McInnes 
12569b94acceSBarry Smith    Input Parameter:
12579b94acceSBarry Smith .  snes - SNES context
12589b94acceSBarry Smith 
12599b94acceSBarry Smith    Output Parameter:
12609b94acceSBarry Smith .  nfails - number of unsuccessful steps attempted
12619b94acceSBarry Smith 
1262c96a6f78SLois Curfman McInnes    Notes:
1263c96a6f78SLois Curfman McInnes    This counter is reset to zero for each successive call to SNESSolve().
1264c96a6f78SLois Curfman McInnes 
126536851e7fSLois Curfman McInnes    Level: intermediate
126636851e7fSLois Curfman McInnes 
12679b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps
126858ebbce7SBarry Smith 
1269e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
127058ebbce7SBarry Smith           SNESSetMaxNonlinearStepFailures(), SNESGetMaxNonlinearStepFailures()
12719b94acceSBarry Smith @*/
12727087cfbeSBarry Smith PetscErrorCode  SNESGetNonlinearStepFailures(SNES snes,PetscInt *nfails)
12739b94acceSBarry Smith {
12743a40ed3dSBarry Smith   PetscFunctionBegin;
12750700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
12764482741eSBarry Smith   PetscValidIntPointer(nfails,2);
127750ffb88aSMatthew Knepley   *nfails = snes->numFailures;
127850ffb88aSMatthew Knepley   PetscFunctionReturn(0);
127950ffb88aSMatthew Knepley }
128050ffb88aSMatthew Knepley 
128150ffb88aSMatthew Knepley /*@
1282b850b91aSLisandro Dalcin    SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps
128350ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
128450ffb88aSMatthew Knepley 
128550ffb88aSMatthew Knepley    Not Collective
128650ffb88aSMatthew Knepley 
128750ffb88aSMatthew Knepley    Input Parameters:
128850ffb88aSMatthew Knepley +  snes     - SNES context
128950ffb88aSMatthew Knepley -  maxFails - maximum of unsuccessful steps
129050ffb88aSMatthew Knepley 
129150ffb88aSMatthew Knepley    Level: intermediate
129250ffb88aSMatthew Knepley 
129350ffb88aSMatthew Knepley .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
129458ebbce7SBarry Smith 
1295e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
129658ebbce7SBarry Smith           SNESGetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures()
129750ffb88aSMatthew Knepley @*/
12987087cfbeSBarry Smith PetscErrorCode  SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails)
129950ffb88aSMatthew Knepley {
130050ffb88aSMatthew Knepley   PetscFunctionBegin;
13010700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
130250ffb88aSMatthew Knepley   snes->maxFailures = maxFails;
130350ffb88aSMatthew Knepley   PetscFunctionReturn(0);
130450ffb88aSMatthew Knepley }
130550ffb88aSMatthew Knepley 
130650ffb88aSMatthew Knepley /*@
1307b850b91aSLisandro Dalcin    SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps
130850ffb88aSMatthew Knepley    attempted by the nonlinear solver before it gives up.
130950ffb88aSMatthew Knepley 
131050ffb88aSMatthew Knepley    Not Collective
131150ffb88aSMatthew Knepley 
131250ffb88aSMatthew Knepley    Input Parameter:
131350ffb88aSMatthew Knepley .  snes     - SNES context
131450ffb88aSMatthew Knepley 
131550ffb88aSMatthew Knepley    Output Parameter:
131650ffb88aSMatthew Knepley .  maxFails - maximum of unsuccessful steps
131750ffb88aSMatthew Knepley 
131850ffb88aSMatthew Knepley    Level: intermediate
131950ffb88aSMatthew Knepley 
132050ffb88aSMatthew Knepley .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
132158ebbce7SBarry Smith 
1322e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(),
132358ebbce7SBarry Smith           SNESSetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures()
132458ebbce7SBarry Smith 
132550ffb88aSMatthew Knepley @*/
13267087cfbeSBarry Smith PetscErrorCode  SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails)
132750ffb88aSMatthew Knepley {
132850ffb88aSMatthew Knepley   PetscFunctionBegin;
13290700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
13304482741eSBarry Smith   PetscValidIntPointer(maxFails,2);
133150ffb88aSMatthew Knepley   *maxFails = snes->maxFailures;
13323a40ed3dSBarry Smith   PetscFunctionReturn(0);
13339b94acceSBarry Smith }
1334a847f771SSatish Balay 
13352541af92SBarry Smith /*@
13362541af92SBarry Smith    SNESGetNumberFunctionEvals - Gets the number of user provided function evaluations
13372541af92SBarry Smith      done by SNES.
13382541af92SBarry Smith 
13392541af92SBarry Smith    Not Collective
13402541af92SBarry Smith 
13412541af92SBarry Smith    Input Parameter:
13422541af92SBarry Smith .  snes     - SNES context
13432541af92SBarry Smith 
13442541af92SBarry Smith    Output Parameter:
13452541af92SBarry Smith .  nfuncs - number of evaluations
13462541af92SBarry Smith 
13472541af92SBarry Smith    Level: intermediate
13482541af92SBarry Smith 
134995452b02SPatrick Sanan    Notes:
135095452b02SPatrick Sanan     Reset every time SNESSolve is called unless SNESSetCountersReset() is used.
1351971e163fSPeter Brune 
13522541af92SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
135358ebbce7SBarry Smith 
1354971e163fSPeter Brune .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), SNESSetCountersReset()
13552541af92SBarry Smith @*/
13567087cfbeSBarry Smith PetscErrorCode  SNESGetNumberFunctionEvals(SNES snes, PetscInt *nfuncs)
13572541af92SBarry Smith {
13582541af92SBarry Smith   PetscFunctionBegin;
13590700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
13602541af92SBarry Smith   PetscValidIntPointer(nfuncs,2);
13612541af92SBarry Smith   *nfuncs = snes->nfuncs;
13622541af92SBarry Smith   PetscFunctionReturn(0);
13632541af92SBarry Smith }
13642541af92SBarry Smith 
13653d4c4710SBarry Smith /*@
13663d4c4710SBarry Smith    SNESGetLinearSolveFailures - Gets the number of failed (non-converged)
13673d4c4710SBarry Smith    linear solvers.
13683d4c4710SBarry Smith 
13693d4c4710SBarry Smith    Not Collective
13703d4c4710SBarry Smith 
13713d4c4710SBarry Smith    Input Parameter:
13723d4c4710SBarry Smith .  snes - SNES context
13733d4c4710SBarry Smith 
13743d4c4710SBarry Smith    Output Parameter:
13753d4c4710SBarry Smith .  nfails - number of failed solves
13763d4c4710SBarry Smith 
13779d85da0cSMatthew G. Knepley    Level: intermediate
13789d85da0cSMatthew G. Knepley 
13799d85da0cSMatthew G. Knepley    Options Database Keys:
13809d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated
13819d85da0cSMatthew G. Knepley 
13823d4c4710SBarry Smith    Notes:
13833d4c4710SBarry Smith    This counter is reset to zero for each successive call to SNESSolve().
13843d4c4710SBarry Smith 
13853d4c4710SBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps
138658ebbce7SBarry Smith 
1387e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures()
13883d4c4710SBarry Smith @*/
13897087cfbeSBarry Smith PetscErrorCode  SNESGetLinearSolveFailures(SNES snes,PetscInt *nfails)
13903d4c4710SBarry Smith {
13913d4c4710SBarry Smith   PetscFunctionBegin;
13920700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
13933d4c4710SBarry Smith   PetscValidIntPointer(nfails,2);
13943d4c4710SBarry Smith   *nfails = snes->numLinearSolveFailures;
13953d4c4710SBarry Smith   PetscFunctionReturn(0);
13963d4c4710SBarry Smith }
13973d4c4710SBarry Smith 
13983d4c4710SBarry Smith /*@
13993d4c4710SBarry Smith    SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts
14003d4c4710SBarry Smith    allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE
14013d4c4710SBarry Smith 
14023f9fe445SBarry Smith    Logically Collective on SNES
14033d4c4710SBarry Smith 
14043d4c4710SBarry Smith    Input Parameters:
14053d4c4710SBarry Smith +  snes     - SNES context
14063d4c4710SBarry Smith -  maxFails - maximum allowed linear solve failures
14073d4c4710SBarry Smith 
14083d4c4710SBarry Smith    Level: intermediate
14093d4c4710SBarry Smith 
14109d85da0cSMatthew G. Knepley    Options Database Keys:
14119d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated
14129d85da0cSMatthew G. Knepley 
141395452b02SPatrick Sanan    Notes:
141495452b02SPatrick Sanan     By default this is 0; that is SNES returns on the first failed linear solve
14153d4c4710SBarry Smith 
14163d4c4710SBarry Smith .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps
14173d4c4710SBarry Smith 
141858ebbce7SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations()
14193d4c4710SBarry Smith @*/
14207087cfbeSBarry Smith PetscErrorCode  SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails)
14213d4c4710SBarry Smith {
14223d4c4710SBarry Smith   PetscFunctionBegin;
14230700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1424c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxFails,2);
14253d4c4710SBarry Smith   snes->maxLinearSolveFailures = maxFails;
14263d4c4710SBarry Smith   PetscFunctionReturn(0);
14273d4c4710SBarry Smith }
14283d4c4710SBarry Smith 
14293d4c4710SBarry Smith /*@
14303d4c4710SBarry Smith    SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that
14313d4c4710SBarry Smith      are allowed before SNES terminates
14323d4c4710SBarry Smith 
14333d4c4710SBarry Smith    Not Collective
14343d4c4710SBarry Smith 
14353d4c4710SBarry Smith    Input Parameter:
14363d4c4710SBarry Smith .  snes     - SNES context
14373d4c4710SBarry Smith 
14383d4c4710SBarry Smith    Output Parameter:
14393d4c4710SBarry Smith .  maxFails - maximum of unsuccessful solves allowed
14403d4c4710SBarry Smith 
14413d4c4710SBarry Smith    Level: intermediate
14423d4c4710SBarry Smith 
144395452b02SPatrick Sanan    Notes:
144495452b02SPatrick Sanan     By default this is 1; that is SNES returns on the first failed linear solve
14453d4c4710SBarry Smith 
14463d4c4710SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps
14473d4c4710SBarry Smith 
1448e1c61ce8SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(),
14493d4c4710SBarry Smith @*/
14507087cfbeSBarry Smith PetscErrorCode  SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails)
14513d4c4710SBarry Smith {
14523d4c4710SBarry Smith   PetscFunctionBegin;
14530700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
14543d4c4710SBarry Smith   PetscValidIntPointer(maxFails,2);
14553d4c4710SBarry Smith   *maxFails = snes->maxLinearSolveFailures;
14563d4c4710SBarry Smith   PetscFunctionReturn(0);
14573d4c4710SBarry Smith }
14583d4c4710SBarry Smith 
1459c96a6f78SLois Curfman McInnes /*@
1460b850b91aSLisandro Dalcin    SNESGetLinearSolveIterations - Gets the total number of linear iterations
1461c96a6f78SLois Curfman McInnes    used by the nonlinear solver.
1462c96a6f78SLois Curfman McInnes 
1463c7afd0dbSLois Curfman McInnes    Not Collective
1464c7afd0dbSLois Curfman McInnes 
1465c96a6f78SLois Curfman McInnes    Input Parameter:
1466c96a6f78SLois Curfman McInnes .  snes - SNES context
1467c96a6f78SLois Curfman McInnes 
1468c96a6f78SLois Curfman McInnes    Output Parameter:
1469c96a6f78SLois Curfman McInnes .  lits - number of linear iterations
1470c96a6f78SLois Curfman McInnes 
1471c96a6f78SLois Curfman McInnes    Notes:
1472971e163fSPeter Brune    This counter is reset to zero for each successive call to SNESSolve() unless SNESSetCountersReset() is used.
1473c96a6f78SLois Curfman McInnes 
1474010be392SBarry 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
1475010be392SBarry Smith    then call KSPGetIterationNumber() after the failed solve.
1476010be392SBarry Smith 
147736851e7fSLois Curfman McInnes    Level: intermediate
147836851e7fSLois Curfman McInnes 
1479c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations
14802b668275SBarry Smith 
148171dbe336SPeter Brune .seealso:  SNESGetIterationNumber(), SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESSetCountersReset()
1482c96a6f78SLois Curfman McInnes @*/
14837087cfbeSBarry Smith PetscErrorCode  SNESGetLinearSolveIterations(SNES snes,PetscInt *lits)
1484c96a6f78SLois Curfman McInnes {
14853a40ed3dSBarry Smith   PetscFunctionBegin;
14860700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
14874482741eSBarry Smith   PetscValidIntPointer(lits,2);
1488c96a6f78SLois Curfman McInnes   *lits = snes->linear_its;
14893a40ed3dSBarry Smith   PetscFunctionReturn(0);
1490c96a6f78SLois Curfman McInnes }
1491c96a6f78SLois Curfman McInnes 
1492971e163fSPeter Brune /*@
1493971e163fSPeter Brune    SNESSetCountersReset - Sets whether or not the counters for linear iterations and function evaluations
1494971e163fSPeter Brune    are reset every time SNESSolve() is called.
1495971e163fSPeter Brune 
1496971e163fSPeter Brune    Logically Collective on SNES
1497971e163fSPeter Brune 
1498971e163fSPeter Brune    Input Parameter:
1499971e163fSPeter Brune +  snes - SNES context
1500971e163fSPeter Brune -  reset - whether to reset the counters or not
1501971e163fSPeter Brune 
1502971e163fSPeter Brune    Notes:
1503fa19ca70SBarry Smith    This defaults to PETSC_TRUE
1504971e163fSPeter Brune 
1505971e163fSPeter Brune    Level: developer
1506971e163fSPeter Brune 
1507971e163fSPeter Brune .keywords: SNES, nonlinear, set, reset, number, linear, iterations
1508971e163fSPeter Brune 
1509734794cfSBarry Smith .seealso:  SNESGetNumberFunctionEvals(), SNESGetLinearSolveIterations(), SNESGetNPC()
1510971e163fSPeter Brune @*/
1511971e163fSPeter Brune PetscErrorCode  SNESSetCountersReset(SNES snes,PetscBool reset)
1512971e163fSPeter Brune {
1513971e163fSPeter Brune   PetscFunctionBegin;
1514971e163fSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1515971e163fSPeter Brune   PetscValidLogicalCollectiveBool(snes,reset,2);
1516971e163fSPeter Brune   snes->counters_reset = reset;
1517971e163fSPeter Brune   PetscFunctionReturn(0);
1518971e163fSPeter Brune }
1519971e163fSPeter Brune 
152082bf6240SBarry Smith 
15212999313aSBarry Smith /*@
15222999313aSBarry Smith    SNESSetKSP - Sets a KSP context for the SNES object to use
15232999313aSBarry Smith 
15242999313aSBarry Smith    Not Collective, but the SNES and KSP objects must live on the same MPI_Comm
15252999313aSBarry Smith 
15262999313aSBarry Smith    Input Parameters:
15272999313aSBarry Smith +  snes - the SNES context
15282999313aSBarry Smith -  ksp - the KSP context
15292999313aSBarry Smith 
15302999313aSBarry Smith    Notes:
15312999313aSBarry Smith    The SNES object already has its KSP object, you can obtain with SNESGetKSP()
15322999313aSBarry Smith    so this routine is rarely needed.
15332999313aSBarry Smith 
15342999313aSBarry Smith    The KSP object that is already in the SNES object has its reference count
15352999313aSBarry Smith    decreased by one.
15362999313aSBarry Smith 
15372999313aSBarry Smith    Level: developer
15382999313aSBarry Smith 
15392999313aSBarry Smith .keywords: SNES, nonlinear, get, KSP, context
15402999313aSBarry Smith 
15412999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
15422999313aSBarry Smith @*/
15437087cfbeSBarry Smith PetscErrorCode  SNESSetKSP(SNES snes,KSP ksp)
15442999313aSBarry Smith {
15452999313aSBarry Smith   PetscErrorCode ierr;
15462999313aSBarry Smith 
15472999313aSBarry Smith   PetscFunctionBegin;
15480700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
15490700a824SBarry Smith   PetscValidHeaderSpecific(ksp,KSP_CLASSID,2);
15502999313aSBarry Smith   PetscCheckSameComm(snes,1,ksp,2);
15517dcf0eaaSdalcinl   ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr);
1552906ed7ccSBarry Smith   if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);}
15532999313aSBarry Smith   snes->ksp = ksp;
15542999313aSBarry Smith   PetscFunctionReturn(0);
15552999313aSBarry Smith }
15562999313aSBarry Smith 
15579b94acceSBarry Smith /* -----------------------------------------------------------*/
155852baeb72SSatish Balay /*@
15599b94acceSBarry Smith    SNESCreate - Creates a nonlinear solver context.
15609b94acceSBarry Smith 
1561c7afd0dbSLois Curfman McInnes    Collective on MPI_Comm
1562c7afd0dbSLois Curfman McInnes 
1563c7afd0dbSLois Curfman McInnes    Input Parameters:
1564906ed7ccSBarry Smith .  comm - MPI communicator
15659b94acceSBarry Smith 
15669b94acceSBarry Smith    Output Parameter:
15679b94acceSBarry Smith .  outsnes - the new SNES context
15689b94acceSBarry Smith 
1569c7afd0dbSLois Curfman McInnes    Options Database Keys:
1570c7afd0dbSLois Curfman McInnes +   -snes_mf - Activates default matrix-free Jacobian-vector products,
1571c7afd0dbSLois Curfman McInnes                and no preconditioning matrix
1572c7afd0dbSLois Curfman McInnes .   -snes_mf_operator - Activates default matrix-free Jacobian-vector
1573c7afd0dbSLois Curfman McInnes                products, and a user-provided preconditioning matrix
1574c7afd0dbSLois Curfman McInnes                as set by SNESSetJacobian()
1575c7afd0dbSLois Curfman McInnes -   -snes_fd - Uses (slow!) finite differences to compute Jacobian
1576c1f60f51SBarry Smith 
157736851e7fSLois Curfman McInnes    Level: beginner
157836851e7fSLois Curfman McInnes 
157995452b02SPatrick Sanan    Developer Notes:
158095452b02SPatrick Sanan     SNES always creates a KSP object even though many SNES methods do not use it. This is
1581efd4aadfSBarry Smith                     unfortunate and should be fixed at some point. The flag snes->usesksp indicates if the
1582efd4aadfSBarry Smith                     particular method does use KSP and regulates if the information about the KSP is printed
1583efd4aadfSBarry Smith                     in SNESView(). TSSetFromOptions() does call SNESSetFromOptions() which can lead to users being confused
1584efd4aadfSBarry Smith                     by help messages about meaningless SNES options.
1585efd4aadfSBarry Smith 
1586efd4aadfSBarry Smith                     SNES always creates the snes->kspconvctx even though it is used by only one type. This should
1587efd4aadfSBarry Smith                     be fixed.
1588efd4aadfSBarry Smith 
15899b94acceSBarry Smith .keywords: SNES, nonlinear, create, context
15909b94acceSBarry Smith 
1591a8054027SBarry Smith .seealso: SNESSolve(), SNESDestroy(), SNES, SNESSetLagPreconditioner()
1592a8054027SBarry Smith 
15939b94acceSBarry Smith @*/
15947087cfbeSBarry Smith PetscErrorCode  SNESCreate(MPI_Comm comm,SNES *outsnes)
15959b94acceSBarry Smith {
1596dfbe8321SBarry Smith   PetscErrorCode ierr;
15979b94acceSBarry Smith   SNES           snes;
1598fa9f3622SBarry Smith   SNESKSPEW      *kctx;
159937fcc0dbSBarry Smith 
16003a40ed3dSBarry Smith   PetscFunctionBegin;
1601ed1caa07SMatthew Knepley   PetscValidPointer(outsnes,2);
16020298fd71SBarry Smith   *outsnes = NULL;
1603607a6623SBarry Smith   ierr = SNESInitializePackage();CHKERRQ(ierr);
16048ba1e511SMatthew Knepley 
160573107ff1SLisandro Dalcin   ierr = PetscHeaderCreate(snes,SNES_CLASSID,"SNES","Nonlinear solver","SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr);
16067adad957SLisandro Dalcin 
16078d359177SBarry Smith   snes->ops->converged    = SNESConvergedDefault;
16082c155ee1SBarry Smith   snes->usesksp           = PETSC_TRUE;
160988976e71SPeter Brune   snes->tolerancesset     = PETSC_FALSE;
16109b94acceSBarry Smith   snes->max_its           = 50;
16119750a799SBarry Smith   snes->max_funcs         = 10000;
16129b94acceSBarry Smith   snes->norm              = 0.0;
1613*c1e67a49SFande Kong   snes->xnorm             = 0.0;
1614*c1e67a49SFande Kong   snes->ynorm             = 0.0;
1615365a6726SPeter Brune   snes->normschedule      = SNES_NORM_ALWAYS;
16166c67d002SPeter Brune   snes->functype          = SNES_FUNCTION_DEFAULT;
16173a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
16183a2046daSBarry Smith   snes->rtol              = 1.e-5;
16193a2046daSBarry Smith #else
1620b4874afaSBarry Smith   snes->rtol              = 1.e-8;
16213a2046daSBarry Smith #endif
1622b4874afaSBarry Smith   snes->ttol              = 0.0;
16233a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
16243a2046daSBarry Smith   snes->abstol            = 1.e-25;
16253a2046daSBarry Smith #else
162670441072SBarry Smith   snes->abstol            = 1.e-50;
16273a2046daSBarry Smith #endif
16287cd0ae37SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE)
16297cd0ae37SLisandro Dalcin   snes->stol              = 1.e-5;
16307cd0ae37SLisandro Dalcin #else
1631c60f73f4SPeter Brune   snes->stol              = 1.e-8;
16327cd0ae37SLisandro Dalcin #endif
16333a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
16343a2046daSBarry Smith   snes->deltatol          = 1.e-6;
16353a2046daSBarry Smith #else
16364b27c08aSLois Curfman McInnes   snes->deltatol          = 1.e-12;
16373a2046daSBarry Smith #endif
1638e37c518bSBarry Smith   snes->divtol            = 1.e4;
1639e37c518bSBarry Smith   snes->rnorm0            = 0;
16409b94acceSBarry Smith   snes->nfuncs            = 0;
164150ffb88aSMatthew Knepley   snes->numFailures       = 0;
164250ffb88aSMatthew Knepley   snes->maxFailures       = 1;
16437a00f4a9SLois Curfman McInnes   snes->linear_its        = 0;
1644e35cf81dSBarry Smith   snes->lagjacobian       = 1;
164537ec4e1aSPeter Brune   snes->jac_iter          = 0;
164637ec4e1aSPeter Brune   snes->lagjac_persist    = PETSC_FALSE;
1647a8054027SBarry Smith   snes->lagpreconditioner = 1;
164837ec4e1aSPeter Brune   snes->pre_iter          = 0;
164937ec4e1aSPeter Brune   snes->lagpre_persist    = PETSC_FALSE;
1650639f9d9dSBarry Smith   snes->numbermonitors    = 0;
16519b94acceSBarry Smith   snes->data              = 0;
16524dc4c822SBarry Smith   snes->setupcalled       = PETSC_FALSE;
1653186905e3SBarry Smith   snes->ksp_ewconv        = PETSC_FALSE;
16546f24a144SLois Curfman McInnes   snes->nwork             = 0;
165558c9b817SLisandro Dalcin   snes->work              = 0;
165658c9b817SLisandro Dalcin   snes->nvwork            = 0;
165758c9b817SLisandro Dalcin   snes->vwork             = 0;
1658758f92a0SBarry Smith   snes->conv_hist_len     = 0;
1659758f92a0SBarry Smith   snes->conv_hist_max     = 0;
16600298fd71SBarry Smith   snes->conv_hist         = NULL;
16610298fd71SBarry Smith   snes->conv_hist_its     = NULL;
1662758f92a0SBarry Smith   snes->conv_hist_reset   = PETSC_TRUE;
1663971e163fSPeter Brune   snes->counters_reset    = PETSC_TRUE;
1664e4ed7901SPeter Brune   snes->vec_func_init_set = PETSC_FALSE;
1665184914b5SBarry Smith   snes->reason            = SNES_CONVERGED_ITERATING;
1666efd4aadfSBarry Smith   snes->npcside           = PC_RIGHT;
1667b3cd9a81SMatthew G. Knepley   snes->setfromoptionscalled = 0;
1668c40d0f55SPeter Brune 
1669d8f46077SPeter Brune   snes->mf          = PETSC_FALSE;
1670d8f46077SPeter Brune   snes->mf_operator = PETSC_FALSE;
1671d8f46077SPeter Brune   snes->mf_version  = 1;
1672d8f46077SPeter Brune 
16733d4c4710SBarry Smith   snes->numLinearSolveFailures = 0;
16743d4c4710SBarry Smith   snes->maxLinearSolveFailures = 1;
16753d4c4710SBarry Smith 
1676349187a7SBarry Smith   snes->vizerotolerance = 1.e-8;
1677349187a7SBarry Smith 
16784fc747eaSLawrence Mitchell   /* Set this to true if the implementation of SNESSolve_XXX does compute the residual at the final solution. */
16794fc747eaSLawrence Mitchell   snes->alwayscomputesfinalresidual = PETSC_FALSE;
16804fc747eaSLawrence Mitchell 
16819b94acceSBarry Smith   /* Create context to compute Eisenstat-Walker relative tolerance for KSP */
1682b00a9115SJed Brown   ierr = PetscNewLog(snes,&kctx);CHKERRQ(ierr);
1683f5af7f23SKarl Rupp 
16849b94acceSBarry Smith   snes->kspconvctx  = (void*)kctx;
16859b94acceSBarry Smith   kctx->version     = 2;
16869b94acceSBarry Smith   kctx->rtol_0      = .3; /* Eisenstat and Walker suggest rtol_0=.5, but
16879b94acceSBarry Smith                              this was too large for some test cases */
168875567043SBarry Smith   kctx->rtol_last   = 0.0;
16899b94acceSBarry Smith   kctx->rtol_max    = .9;
16909b94acceSBarry Smith   kctx->gamma       = 1.0;
169162d1f40fSBarry Smith   kctx->alpha       = .5*(1.0 + PetscSqrtReal(5.0));
169271f87433Sdalcinl   kctx->alpha2      = kctx->alpha;
16939b94acceSBarry Smith   kctx->threshold   = .1;
169475567043SBarry Smith   kctx->lresid_last = 0.0;
169575567043SBarry Smith   kctx->norm_last   = 0.0;
16969b94acceSBarry Smith 
16979b94acceSBarry Smith   *outsnes = snes;
16983a40ed3dSBarry Smith   PetscFunctionReturn(0);
16999b94acceSBarry Smith }
17009b94acceSBarry Smith 
170188f0584fSBarry Smith /*MC
1702411c0326SBarry Smith     SNESFunction - Functional form used to convey the nonlinear function to be solved by SNES
170388f0584fSBarry Smith 
170488f0584fSBarry Smith      Synopsis:
1705411c0326SBarry Smith      #include "petscsnes.h"
1706411c0326SBarry Smith      PetscErrorCode SNESFunction(SNES snes,Vec x,Vec f,void *ctx);
170788f0584fSBarry Smith 
170888f0584fSBarry Smith      Input Parameters:
170988f0584fSBarry Smith +     snes - the SNES context
171088f0584fSBarry Smith .     x    - state at which to evaluate residual
171188f0584fSBarry Smith -     ctx     - optional user-defined function context, passed in with SNESSetFunction()
171288f0584fSBarry Smith 
171388f0584fSBarry Smith      Output Parameter:
171488f0584fSBarry Smith .     f  - vector to put residual (function value)
171588f0584fSBarry Smith 
1716878cb397SSatish Balay    Level: intermediate
1717878cb397SSatish Balay 
171888f0584fSBarry Smith .seealso:   SNESSetFunction(), SNESGetFunction()
171988f0584fSBarry Smith M*/
172088f0584fSBarry Smith 
17219b94acceSBarry Smith /*@C
17229b94acceSBarry Smith    SNESSetFunction - Sets the function evaluation routine and function
17239b94acceSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
17249b94acceSBarry Smith    equations.
17259b94acceSBarry Smith 
17263f9fe445SBarry Smith    Logically Collective on SNES
1727fee21e36SBarry Smith 
1728c7afd0dbSLois Curfman McInnes    Input Parameters:
1729c7afd0dbSLois Curfman McInnes +  snes - the SNES context
1730c7afd0dbSLois Curfman McInnes .  r - vector to store function value
1731f8b49ee9SBarry Smith .  f - function evaluation routine; see SNESFunction for calling sequence details
1732c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
17330298fd71SBarry Smith          function evaluation routine (may be NULL)
17349b94acceSBarry Smith 
17359b94acceSBarry Smith    Notes:
17369b94acceSBarry Smith    The Newton-like methods typically solve linear systems of the form
17379b94acceSBarry Smith $      f'(x) x = -f(x),
1738c7afd0dbSLois Curfman McInnes    where f'(x) denotes the Jacobian matrix and f(x) is the function.
17399b94acceSBarry Smith 
174036851e7fSLois Curfman McInnes    Level: beginner
174136851e7fSLois Curfman McInnes 
17429b94acceSBarry Smith .keywords: SNES, nonlinear, set, function
17439b94acceSBarry Smith 
1744bf388a1fSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetPicard(), SNESFunction
17459b94acceSBarry Smith @*/
1746f8b49ee9SBarry Smith PetscErrorCode  SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx)
17479b94acceSBarry Smith {
174885385478SLisandro Dalcin   PetscErrorCode ierr;
17496cab3a1bSJed Brown   DM             dm;
17506cab3a1bSJed Brown 
17513a40ed3dSBarry Smith   PetscFunctionBegin;
17520700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1753d2a683ecSLisandro Dalcin   if (r) {
1754d2a683ecSLisandro Dalcin     PetscValidHeaderSpecific(r,VEC_CLASSID,2);
1755d2a683ecSLisandro Dalcin     PetscCheckSameComm(snes,1,r,2);
175685385478SLisandro Dalcin     ierr = PetscObjectReference((PetscObject)r);CHKERRQ(ierr);
17576bf464f9SBarry Smith     ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr);
1758f5af7f23SKarl Rupp 
175985385478SLisandro Dalcin     snes->vec_func = r;
1760d2a683ecSLisandro Dalcin   }
17616cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
1762f8b49ee9SBarry Smith   ierr = DMSNESSetFunction(dm,f,ctx);CHKERRQ(ierr);
17633a40ed3dSBarry Smith   PetscFunctionReturn(0);
17649b94acceSBarry Smith }
17659b94acceSBarry Smith 
1766646217ecSPeter Brune 
1767e4ed7901SPeter Brune /*@C
1768e4ed7901SPeter Brune    SNESSetInitialFunction - Sets the function vector to be used as the
1769e4ed7901SPeter Brune    function norm at the initialization of the method.  In some
1770e4ed7901SPeter Brune    instances, the user has precomputed the function before calling
1771e4ed7901SPeter Brune    SNESSolve.  This function allows one to avoid a redundant call
1772e4ed7901SPeter Brune    to SNESComputeFunction in that case.
1773e4ed7901SPeter Brune 
1774e4ed7901SPeter Brune    Logically Collective on SNES
1775e4ed7901SPeter Brune 
1776e4ed7901SPeter Brune    Input Parameters:
1777e4ed7901SPeter Brune +  snes - the SNES context
1778e4ed7901SPeter Brune -  f - vector to store function value
1779e4ed7901SPeter Brune 
1780e4ed7901SPeter Brune    Notes:
1781e4ed7901SPeter Brune    This should not be modified during the solution procedure.
1782e4ed7901SPeter Brune 
1783e4ed7901SPeter Brune    This is used extensively in the SNESFAS hierarchy and in nonlinear preconditioning.
1784e4ed7901SPeter Brune 
1785e4ed7901SPeter Brune    Level: developer
1786e4ed7901SPeter Brune 
1787e4ed7901SPeter Brune .keywords: SNES, nonlinear, set, function
1788e4ed7901SPeter Brune 
1789e4ed7901SPeter Brune .seealso: SNESSetFunction(), SNESComputeFunction(), SNESSetInitialFunctionNorm()
1790e4ed7901SPeter Brune @*/
1791e4ed7901SPeter Brune PetscErrorCode  SNESSetInitialFunction(SNES snes, Vec f)
1792e4ed7901SPeter Brune {
1793e4ed7901SPeter Brune   PetscErrorCode ierr;
1794e4ed7901SPeter Brune   Vec            vec_func;
1795e4ed7901SPeter Brune 
1796e4ed7901SPeter Brune   PetscFunctionBegin;
1797e4ed7901SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1798e4ed7901SPeter Brune   PetscValidHeaderSpecific(f,VEC_CLASSID,2);
1799e4ed7901SPeter Brune   PetscCheckSameComm(snes,1,f,2);
1800efd4aadfSBarry Smith   if (snes->npcside== PC_LEFT && snes->functype == SNES_FUNCTION_PRECONDITIONED) {
1801902f982fSPeter Brune     snes->vec_func_init_set = PETSC_FALSE;
1802902f982fSPeter Brune     PetscFunctionReturn(0);
1803902f982fSPeter Brune   }
18040298fd71SBarry Smith   ierr = SNESGetFunction(snes,&vec_func,NULL,NULL);CHKERRQ(ierr);
1805e4ed7901SPeter Brune   ierr = VecCopy(f, vec_func);CHKERRQ(ierr);
1806f5af7f23SKarl Rupp 
1807217b9c2eSPeter Brune   snes->vec_func_init_set = PETSC_TRUE;
1808e4ed7901SPeter Brune   PetscFunctionReturn(0);
1809e4ed7901SPeter Brune }
1810e4ed7901SPeter Brune 
1811534ebe21SPeter Brune /*@
1812365a6726SPeter Brune    SNESSetNormSchedule - Sets the SNESNormSchedule used in covergence and monitoring
1813534ebe21SPeter Brune    of the SNES method.
1814534ebe21SPeter Brune 
1815534ebe21SPeter Brune    Logically Collective on SNES
1816534ebe21SPeter Brune 
1817534ebe21SPeter Brune    Input Parameters:
1818534ebe21SPeter Brune +  snes - the SNES context
1819365a6726SPeter Brune -  normschedule - the frequency of norm computation
1820534ebe21SPeter Brune 
1821517f1916SMatthew G. Knepley    Options Database Key:
1822517f1916SMatthew G. Knepley .  -snes_norm_schedule <none, always, initialonly, finalonly, initalfinalonly>
1823517f1916SMatthew G. Knepley 
1824534ebe21SPeter Brune    Notes:
1825365a6726SPeter Brune    Only certain SNES methods support certain SNESNormSchedules.  Most require evaluation
1826534ebe21SPeter Brune    of the nonlinear function and the taking of its norm at every iteration to
1827534ebe21SPeter Brune    even ensure convergence at all.  However, methods such as custom Gauss-Seidel methods
1828be95d8f1SBarry Smith    (SNESNGS) and the like do not require the norm of the function to be computed, and therfore
1829534ebe21SPeter Brune    may either be monitored for convergence or not.  As these are often used as nonlinear
1830534ebe21SPeter Brune    preconditioners, monitoring the norm of their error is not a useful enterprise within
1831534ebe21SPeter Brune    their solution.
1832534ebe21SPeter Brune 
1833534ebe21SPeter Brune    Level: developer
1834534ebe21SPeter Brune 
1835534ebe21SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type
1836534ebe21SPeter Brune 
1837365a6726SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1838534ebe21SPeter Brune @*/
1839365a6726SPeter Brune PetscErrorCode  SNESSetNormSchedule(SNES snes, SNESNormSchedule normschedule)
1840534ebe21SPeter Brune {
1841534ebe21SPeter Brune   PetscFunctionBegin;
1842534ebe21SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1843365a6726SPeter Brune   snes->normschedule = normschedule;
1844534ebe21SPeter Brune   PetscFunctionReturn(0);
1845534ebe21SPeter Brune }
1846534ebe21SPeter Brune 
1847534ebe21SPeter Brune 
1848534ebe21SPeter Brune /*@
1849365a6726SPeter Brune    SNESGetNormSchedule - Gets the SNESNormSchedule used in covergence and monitoring
1850534ebe21SPeter Brune    of the SNES method.
1851534ebe21SPeter Brune 
1852534ebe21SPeter Brune    Logically Collective on SNES
1853534ebe21SPeter Brune 
1854534ebe21SPeter Brune    Input Parameters:
1855534ebe21SPeter Brune +  snes - the SNES context
1856365a6726SPeter Brune -  normschedule - the type of the norm used
1857534ebe21SPeter Brune 
1858534ebe21SPeter Brune    Level: advanced
1859534ebe21SPeter Brune 
1860534ebe21SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type
1861534ebe21SPeter Brune 
1862365a6726SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1863534ebe21SPeter Brune @*/
1864365a6726SPeter Brune PetscErrorCode  SNESGetNormSchedule(SNES snes, SNESNormSchedule *normschedule)
1865534ebe21SPeter Brune {
1866534ebe21SPeter Brune   PetscFunctionBegin;
1867534ebe21SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1868365a6726SPeter Brune   *normschedule = snes->normschedule;
1869534ebe21SPeter Brune   PetscFunctionReturn(0);
1870534ebe21SPeter Brune }
1871534ebe21SPeter Brune 
187247073ea2SPeter Brune 
1873c5ce4427SMatthew G. Knepley /*@
1874c5ce4427SMatthew G. Knepley   SNESSetFunctionNorm - Sets the last computed residual norm.
1875c5ce4427SMatthew G. Knepley 
1876c5ce4427SMatthew G. Knepley   Logically Collective on SNES
1877c5ce4427SMatthew G. Knepley 
1878c5ce4427SMatthew G. Knepley   Input Parameters:
1879c5ce4427SMatthew G. Knepley + snes - the SNES context
1880c5ce4427SMatthew G. Knepley 
1881c5ce4427SMatthew G. Knepley - normschedule - the frequency of norm computation
1882c5ce4427SMatthew G. Knepley 
1883c5ce4427SMatthew G. Knepley   Level: developer
1884c5ce4427SMatthew G. Knepley 
1885c5ce4427SMatthew G. Knepley .keywords: SNES, nonlinear, set, function, norm, type
1886c5ce4427SMatthew G. Knepley .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1887c5ce4427SMatthew G. Knepley @*/
1888c5ce4427SMatthew G. Knepley PetscErrorCode SNESSetFunctionNorm(SNES snes, PetscReal norm)
1889c5ce4427SMatthew G. Knepley {
1890c5ce4427SMatthew G. Knepley   PetscFunctionBegin;
1891c5ce4427SMatthew G. Knepley   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1892c5ce4427SMatthew G. Knepley   snes->norm = norm;
1893c5ce4427SMatthew G. Knepley   PetscFunctionReturn(0);
1894c5ce4427SMatthew G. Knepley }
1895c5ce4427SMatthew G. Knepley 
1896c5ce4427SMatthew G. Knepley /*@
1897c5ce4427SMatthew G. Knepley   SNESGetFunctionNorm - Gets the last computed norm of the residual
1898c5ce4427SMatthew G. Knepley 
1899c5ce4427SMatthew G. Knepley   Not Collective
1900c5ce4427SMatthew G. Knepley 
1901c5ce4427SMatthew G. Knepley   Input Parameter:
1902c5ce4427SMatthew G. Knepley . snes - the SNES context
1903c5ce4427SMatthew G. Knepley 
1904c5ce4427SMatthew G. Knepley   Output Parameter:
1905c5ce4427SMatthew G. Knepley . norm - the last computed residual norm
1906c5ce4427SMatthew G. Knepley 
1907c5ce4427SMatthew G. Knepley   Level: developer
1908c5ce4427SMatthew G. Knepley 
1909c5ce4427SMatthew G. Knepley .keywords: SNES, nonlinear, set, function, norm, type
1910c5ce4427SMatthew G. Knepley .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
1911c5ce4427SMatthew G. Knepley @*/
1912c5ce4427SMatthew G. Knepley PetscErrorCode SNESGetFunctionNorm(SNES snes, PetscReal *norm)
1913c5ce4427SMatthew G. Knepley {
1914c5ce4427SMatthew G. Knepley   PetscFunctionBegin;
1915c5ce4427SMatthew G. Knepley   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1916c5ce4427SMatthew G. Knepley   PetscValidPointer(norm, 2);
1917c5ce4427SMatthew G. Knepley   *norm = snes->norm;
1918c5ce4427SMatthew G. Knepley   PetscFunctionReturn(0);
1919c5ce4427SMatthew G. Knepley }
1920c5ce4427SMatthew G. Knepley 
1921*c1e67a49SFande Kong /*@
1922*c1e67a49SFande Kong   SNESGetUpdateNorm - Gets the last computed norm of the Newton update
1923*c1e67a49SFande Kong 
1924*c1e67a49SFande Kong   Not Collective
1925*c1e67a49SFande Kong 
1926*c1e67a49SFande Kong   Input Parameter:
1927*c1e67a49SFande Kong . snes - the SNES context
1928*c1e67a49SFande Kong 
1929*c1e67a49SFande Kong   Output Parameter:
1930*c1e67a49SFande Kong . ynorm - the last computed update norm
1931*c1e67a49SFande Kong 
1932*c1e67a49SFande Kong   Level: developer
1933*c1e67a49SFande Kong 
1934*c1e67a49SFande Kong .keywords: SNES, nonlinear, set, function, norm, type
1935*c1e67a49SFande Kong .seealso: SNESSetNormSchedule(), SNESComputeFunction(), SNESGetFunctionNorm()
1936*c1e67a49SFande Kong @*/
1937*c1e67a49SFande Kong PetscErrorCode SNESGetUpdateNorm(SNES snes, PetscReal *ynorm)
1938*c1e67a49SFande Kong {
1939*c1e67a49SFande Kong   PetscFunctionBegin;
1940*c1e67a49SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1941*c1e67a49SFande Kong   PetscValidPointer(ynorm, 2);
1942*c1e67a49SFande Kong   *ynorm = snes->ynorm;
1943*c1e67a49SFande Kong   PetscFunctionReturn(0);
1944*c1e67a49SFande Kong }
1945*c1e67a49SFande Kong 
1946*c1e67a49SFande Kong /*@
1947*c1e67a49SFande Kong   SNESGetFunctionNorm - Gets the last computed norm of the solution
1948*c1e67a49SFande Kong 
1949*c1e67a49SFande Kong   Not Collective
1950*c1e67a49SFande Kong 
1951*c1e67a49SFande Kong   Input Parameter:
1952*c1e67a49SFande Kong . snes - the SNES context
1953*c1e67a49SFande Kong 
1954*c1e67a49SFande Kong   Output Parameter:
1955*c1e67a49SFande Kong . xnorm - the last computed solution norm
1956*c1e67a49SFande Kong 
1957*c1e67a49SFande Kong   Level: developer
1958*c1e67a49SFande Kong 
1959*c1e67a49SFande Kong .keywords: SNES, nonlinear, set, function, norm, type
1960*c1e67a49SFande Kong .seealso: SNESSetNormSchedule(), SNESComputeFunction(), SNESGetFunctionNorm(), SNESGetUpdateNorm()
1961*c1e67a49SFande Kong @*/
1962*c1e67a49SFande Kong PetscErrorCode SNESGetSolutionNorm(SNES snes, PetscReal *xnorm)
1963*c1e67a49SFande Kong {
1964*c1e67a49SFande Kong   PetscFunctionBegin;
1965*c1e67a49SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1966*c1e67a49SFande Kong   PetscValidPointer(xnorm, 2);
1967*c1e67a49SFande Kong   *xnorm = snes->xnorm;
1968*c1e67a49SFande Kong   PetscFunctionReturn(0);
1969*c1e67a49SFande Kong }
1970*c1e67a49SFande Kong 
197147073ea2SPeter Brune /*@C
197247073ea2SPeter Brune    SNESSetFunctionType - Sets the SNESNormSchedule used in covergence and monitoring
197347073ea2SPeter Brune    of the SNES method.
197447073ea2SPeter Brune 
197547073ea2SPeter Brune    Logically Collective on SNES
197647073ea2SPeter Brune 
197747073ea2SPeter Brune    Input Parameters:
197847073ea2SPeter Brune +  snes - the SNES context
197947073ea2SPeter Brune -  normschedule - the frequency of norm computation
198047073ea2SPeter Brune 
198147073ea2SPeter Brune    Notes:
198247073ea2SPeter Brune    Only certain SNES methods support certain SNESNormSchedules.  Most require evaluation
198347073ea2SPeter Brune    of the nonlinear function and the taking of its norm at every iteration to
198447073ea2SPeter Brune    even ensure convergence at all.  However, methods such as custom Gauss-Seidel methods
1985be95d8f1SBarry Smith    (SNESNGS) and the like do not require the norm of the function to be computed, and therfore
198647073ea2SPeter Brune    may either be monitored for convergence or not.  As these are often used as nonlinear
198747073ea2SPeter Brune    preconditioners, monitoring the norm of their error is not a useful enterprise within
198847073ea2SPeter Brune    their solution.
198947073ea2SPeter Brune 
199047073ea2SPeter Brune    Level: developer
199147073ea2SPeter Brune 
199247073ea2SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type
199347073ea2SPeter Brune 
199447073ea2SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
199547073ea2SPeter Brune @*/
199647073ea2SPeter Brune PetscErrorCode  SNESSetFunctionType(SNES snes, SNESFunctionType type)
199747073ea2SPeter Brune {
199847073ea2SPeter Brune   PetscFunctionBegin;
199947073ea2SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
200047073ea2SPeter Brune   snes->functype = type;
200147073ea2SPeter Brune   PetscFunctionReturn(0);
200247073ea2SPeter Brune }
200347073ea2SPeter Brune 
200447073ea2SPeter Brune 
200547073ea2SPeter Brune /*@C
200647073ea2SPeter Brune    SNESGetFunctionType - Gets the SNESNormSchedule used in covergence and monitoring
200747073ea2SPeter Brune    of the SNES method.
200847073ea2SPeter Brune 
200947073ea2SPeter Brune    Logically Collective on SNES
201047073ea2SPeter Brune 
201147073ea2SPeter Brune    Input Parameters:
201247073ea2SPeter Brune +  snes - the SNES context
201347073ea2SPeter Brune -  normschedule - the type of the norm used
201447073ea2SPeter Brune 
201547073ea2SPeter Brune    Level: advanced
201647073ea2SPeter Brune 
201747073ea2SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type
201847073ea2SPeter Brune 
201947073ea2SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule
202047073ea2SPeter Brune @*/
202147073ea2SPeter Brune PetscErrorCode  SNESGetFunctionType(SNES snes, SNESFunctionType *type)
202247073ea2SPeter Brune {
202347073ea2SPeter Brune   PetscFunctionBegin;
202447073ea2SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
202547073ea2SPeter Brune   *type = snes->functype;
2026534ebe21SPeter Brune   PetscFunctionReturn(0);
2027534ebe21SPeter Brune }
2028534ebe21SPeter Brune 
2029bf388a1fSBarry Smith /*MC
2030be95d8f1SBarry Smith     SNESNGSFunction - function used to convey a Gauss-Seidel sweep on the nonlinear function
2031bf388a1fSBarry Smith 
2032bf388a1fSBarry Smith      Synopsis:
2033aaa7dc30SBarry Smith      #include <petscsnes.h>
2034be95d8f1SBarry Smith $    SNESNGSFunction(SNES snes,Vec x,Vec b,void *ctx);
2035bf388a1fSBarry Smith 
2036bf388a1fSBarry Smith +  X   - solution vector
2037bf388a1fSBarry Smith .  B   - RHS vector
2038bf388a1fSBarry Smith -  ctx - optional user-defined Gauss-Seidel context
2039bf388a1fSBarry Smith 
2040878cb397SSatish Balay    Level: intermediate
2041878cb397SSatish Balay 
2042be95d8f1SBarry Smith .seealso:   SNESSetNGS(), SNESGetNGS()
2043bf388a1fSBarry Smith M*/
2044bf388a1fSBarry Smith 
2045c79ef259SPeter Brune /*@C
2046be95d8f1SBarry Smith    SNESSetNGS - Sets the user nonlinear Gauss-Seidel routine for
2047c79ef259SPeter Brune    use with composed nonlinear solvers.
2048c79ef259SPeter Brune 
2049c79ef259SPeter Brune    Input Parameters:
2050c79ef259SPeter Brune +  snes   - the SNES context
2051be95d8f1SBarry Smith .  f - function evaluation routine to apply Gauss-Seidel see SNESNGSFunction
2052c79ef259SPeter Brune -  ctx    - [optional] user-defined context for private data for the
20530298fd71SBarry Smith             smoother evaluation routine (may be NULL)
2054c79ef259SPeter Brune 
2055c79ef259SPeter Brune    Notes:
2056be95d8f1SBarry Smith    The NGS routines are used by the composed nonlinear solver to generate
2057c79ef259SPeter Brune     a problem appropriate update to the solution, particularly FAS.
2058c79ef259SPeter Brune 
2059d28543b3SPeter Brune    Level: intermediate
2060c79ef259SPeter Brune 
2061d28543b3SPeter Brune .keywords: SNES, nonlinear, set, Gauss-Seidel
2062c79ef259SPeter Brune 
2063be95d8f1SBarry Smith .seealso: SNESGetFunction(), SNESComputeNGS()
2064c79ef259SPeter Brune @*/
2065be95d8f1SBarry Smith PetscErrorCode SNESSetNGS(SNES snes,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx)
20666cab3a1bSJed Brown {
20676cab3a1bSJed Brown   PetscErrorCode ierr;
20686cab3a1bSJed Brown   DM             dm;
20696cab3a1bSJed Brown 
2070646217ecSPeter Brune   PetscFunctionBegin;
20716cab3a1bSJed Brown   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
20726cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2073be95d8f1SBarry Smith   ierr = DMSNESSetNGS(dm,f,ctx);CHKERRQ(ierr);
2074646217ecSPeter Brune   PetscFunctionReturn(0);
2075646217ecSPeter Brune }
2076646217ecSPeter Brune 
207725acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeFunction(SNES snes,Vec x,Vec f,void *ctx)
20788b0a5094SBarry Smith {
20798b0a5094SBarry Smith   PetscErrorCode ierr;
2080e03ab78fSPeter Brune   DM             dm;
2081942e3340SBarry Smith   DMSNES         sdm;
20826cab3a1bSJed Brown 
20838b0a5094SBarry Smith   PetscFunctionBegin;
2084e03ab78fSPeter Brune   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2085942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
208625acbd8eSLisandro Dalcin   if (!sdm->ops->computepfunction) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard function.");
208725acbd8eSLisandro Dalcin   if (!sdm->ops->computepjacobian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard Jacobian.");
20888b0a5094SBarry Smith   /*  A(x)*x - b(x) */
208925acbd8eSLisandro Dalcin   PetscStackPush("SNES Picard user function");
209022c6f798SBarry Smith   ierr = (*sdm->ops->computepfunction)(snes,x,f,sdm->pctx);CHKERRQ(ierr);
209125acbd8eSLisandro Dalcin   PetscStackPop;
209225acbd8eSLisandro Dalcin   PetscStackPush("SNES Picard user Jacobian");
2093d1e9a80fSBarry Smith   ierr = (*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx);CHKERRQ(ierr);
209425acbd8eSLisandro Dalcin   PetscStackPop;
20958b0a5094SBarry Smith   ierr = VecScale(f,-1.0);CHKERRQ(ierr);
209695eabcedSBarry Smith   ierr = MatMultAdd(snes->jacobian,x,f,f);CHKERRQ(ierr);
20978b0a5094SBarry Smith   PetscFunctionReturn(0);
20988b0a5094SBarry Smith }
20998b0a5094SBarry Smith 
210025acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeJacobian(SNES snes,Vec x1,Mat J,Mat B,void *ctx)
21018b0a5094SBarry Smith {
21028b0a5094SBarry Smith   PetscFunctionBegin;
2103e03ab78fSPeter Brune   /* the jacobian matrix should be pre-filled in SNESPicardComputeFunction */
21048b0a5094SBarry Smith   PetscFunctionReturn(0);
21058b0a5094SBarry Smith }
21068b0a5094SBarry Smith 
21078b0a5094SBarry Smith /*@C
21080d04baf8SBarry Smith    SNESSetPicard - Use SNES to solve the semilinear-system A(x) x = b(x) via a Picard type iteration (Picard linearization)
21098b0a5094SBarry Smith 
21108b0a5094SBarry Smith    Logically Collective on SNES
21118b0a5094SBarry Smith 
21128b0a5094SBarry Smith    Input Parameters:
21138b0a5094SBarry Smith +  snes - the SNES context
21148b0a5094SBarry Smith .  r - vector to store function value
2115f8b49ee9SBarry Smith .  b - function evaluation routine
2116e5d3d808SBarry Smith .  Amat - matrix with which A(x) x - b(x) is to be computed
2117e5d3d808SBarry Smith .  Pmat - matrix from which preconditioner is computed (usually the same as Amat)
2118411c0326SBarry Smith .  J  - function to compute matrix value, see SNESJacobianFunction for details on its calling sequence
21198b0a5094SBarry Smith -  ctx - [optional] user-defined context for private data for the
21200298fd71SBarry Smith          function evaluation routine (may be NULL)
21218b0a5094SBarry Smith 
21228b0a5094SBarry Smith    Notes:
2123f450aa47SBarry 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
2124f450aa47SBarry 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.
2125f450aa47SBarry Smith 
21268b0a5094SBarry Smith     One can call SNESSetPicard() or SNESSetFunction() (and possibly SNESSetJacobian()) but cannot call both
21278b0a5094SBarry Smith 
21288b0a5094SBarry 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}
21298b0a5094SBarry 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.
21308b0a5094SBarry Smith 
21318b0a5094SBarry Smith      Run with -snes_mf_operator to solve the system with Newton's method using A(x^{n}) to construct the preconditioner.
21328b0a5094SBarry Smith 
21330d04baf8SBarry Smith    We implement the defect correction form of the Picard iteration because it converges much more generally when inexact linear solvers are used then
21340d04baf8SBarry Smith    the direct Picard iteration A(x^n) x^{n+1} = b(x^n)
21358b0a5094SBarry Smith 
21368b0a5094SBarry 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
21378b0a5094SBarry 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
21388b0a5094SBarry Smith    different please contact us at petsc-dev@mcs.anl.gov and we'll have an entirely new argument :-).
21398b0a5094SBarry Smith 
2140f450aa47SBarry Smith    Level: intermediate
21418b0a5094SBarry Smith 
21428b0a5094SBarry Smith .keywords: SNES, nonlinear, set, function
21438b0a5094SBarry Smith 
2144411c0326SBarry Smith .seealso: SNESGetFunction(), SNESSetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESGetPicard(), SNESLineSearchPreCheckPicard(), SNESJacobianFunction
21458b0a5094SBarry Smith @*/
2146d1e9a80fSBarry Smith PetscErrorCode  SNESSetPicard(SNES snes,Vec r,PetscErrorCode (*b)(SNES,Vec,Vec,void*),Mat Amat, Mat Pmat, PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx)
21478b0a5094SBarry Smith {
21488b0a5094SBarry Smith   PetscErrorCode ierr;
2149e03ab78fSPeter Brune   DM             dm;
2150e03ab78fSPeter Brune 
21518b0a5094SBarry Smith   PetscFunctionBegin;
21528b0a5094SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2153e03ab78fSPeter Brune   ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
2154f8b49ee9SBarry Smith   ierr = DMSNESSetPicard(dm,b,J,ctx);CHKERRQ(ierr);
21558b0a5094SBarry Smith   ierr = SNESSetFunction(snes,r,SNESPicardComputeFunction,ctx);CHKERRQ(ierr);
2156e5d3d808SBarry Smith   ierr = SNESSetJacobian(snes,Amat,Pmat,SNESPicardComputeJacobian,ctx);CHKERRQ(ierr);
21578b0a5094SBarry Smith   PetscFunctionReturn(0);
21588b0a5094SBarry Smith }
21598b0a5094SBarry Smith 
21607971a8bfSPeter Brune /*@C
21617971a8bfSPeter Brune    SNESGetPicard - Returns the context for the Picard iteration
21627971a8bfSPeter Brune 
21637971a8bfSPeter Brune    Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet.
21647971a8bfSPeter Brune 
21657971a8bfSPeter Brune    Input Parameter:
21667971a8bfSPeter Brune .  snes - the SNES context
21677971a8bfSPeter Brune 
21687971a8bfSPeter Brune    Output Parameter:
21690298fd71SBarry Smith +  r - the function (or NULL)
2170f8b49ee9SBarry Smith .  f - the function (or NULL); see SNESFunction for calling sequence details
2171e4357dc4SBarry Smith .  Amat - the matrix used to defined the operation A(x) x - b(x) (or NULL)
2172e4357dc4SBarry Smith .  Pmat  - the matrix from which the preconditioner will be constructed (or NULL)
2173f8b49ee9SBarry Smith .  J - the function for matrix evaluation (or NULL); see SNESJacobianFunction for calling sequence details
21740298fd71SBarry Smith -  ctx - the function context (or NULL)
21757971a8bfSPeter Brune 
21767971a8bfSPeter Brune    Level: advanced
21777971a8bfSPeter Brune 
21787971a8bfSPeter Brune .keywords: SNES, nonlinear, get, function
21797971a8bfSPeter Brune 
2180e4357dc4SBarry Smith .seealso: SNESSetPicard(), SNESGetFunction(), SNESGetJacobian(), SNESGetDM(), SNESFunction, SNESJacobianFunction
21817971a8bfSPeter Brune @*/
2182d1e9a80fSBarry 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)
21837971a8bfSPeter Brune {
21847971a8bfSPeter Brune   PetscErrorCode ierr;
21857971a8bfSPeter Brune   DM             dm;
21867971a8bfSPeter Brune 
21877971a8bfSPeter Brune   PetscFunctionBegin;
21887971a8bfSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
21890298fd71SBarry Smith   ierr = SNESGetFunction(snes,r,NULL,NULL);CHKERRQ(ierr);
2190e4357dc4SBarry Smith   ierr = SNESGetJacobian(snes,Amat,Pmat,NULL,NULL);CHKERRQ(ierr);
21917971a8bfSPeter Brune   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2192f8b49ee9SBarry Smith   ierr = DMSNESGetPicard(dm,f,J,ctx);CHKERRQ(ierr);
21937971a8bfSPeter Brune   PetscFunctionReturn(0);
21947971a8bfSPeter Brune }
21957971a8bfSPeter Brune 
2196d25893d9SBarry Smith /*@C
2197d25893d9SBarry Smith    SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem
2198d25893d9SBarry Smith 
2199d25893d9SBarry Smith    Logically Collective on SNES
2200d25893d9SBarry Smith 
2201d25893d9SBarry Smith    Input Parameters:
2202d25893d9SBarry Smith +  snes - the SNES context
2203d25893d9SBarry Smith .  func - function evaluation routine
2204d25893d9SBarry Smith -  ctx - [optional] user-defined context for private data for the
22050298fd71SBarry Smith          function evaluation routine (may be NULL)
2206d25893d9SBarry Smith 
2207d25893d9SBarry Smith    Calling sequence of func:
2208d25893d9SBarry Smith $    func (SNES snes,Vec x,void *ctx);
2209d25893d9SBarry Smith 
2210d25893d9SBarry Smith .  f - function vector
2211d25893d9SBarry Smith -  ctx - optional user-defined function context
2212d25893d9SBarry Smith 
2213d25893d9SBarry Smith    Level: intermediate
2214d25893d9SBarry Smith 
2215d25893d9SBarry Smith .keywords: SNES, nonlinear, set, function
2216d25893d9SBarry Smith 
2217d25893d9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian()
2218d25893d9SBarry Smith @*/
2219d25893d9SBarry Smith PetscErrorCode  SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx)
2220d25893d9SBarry Smith {
2221d25893d9SBarry Smith   PetscFunctionBegin;
2222d25893d9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2223d25893d9SBarry Smith   if (func) snes->ops->computeinitialguess = func;
2224d25893d9SBarry Smith   if (ctx)  snes->initialguessP            = ctx;
2225d25893d9SBarry Smith   PetscFunctionReturn(0);
2226d25893d9SBarry Smith }
2227d25893d9SBarry Smith 
22283ab0aad5SBarry Smith /* --------------------------------------------------------------- */
22291096aae1SMatthew Knepley /*@C
22301096aae1SMatthew Knepley    SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set
22311096aae1SMatthew Knepley    it assumes a zero right hand side.
22321096aae1SMatthew Knepley 
22333f9fe445SBarry Smith    Logically Collective on SNES
22341096aae1SMatthew Knepley 
22351096aae1SMatthew Knepley    Input Parameter:
22361096aae1SMatthew Knepley .  snes - the SNES context
22371096aae1SMatthew Knepley 
22381096aae1SMatthew Knepley    Output Parameter:
22390298fd71SBarry Smith .  rhs - the right hand side vector or NULL if the right hand side vector is null
22401096aae1SMatthew Knepley 
22411096aae1SMatthew Knepley    Level: intermediate
22421096aae1SMatthew Knepley 
22431096aae1SMatthew Knepley .keywords: SNES, nonlinear, get, function, right hand side
22441096aae1SMatthew Knepley 
224585385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
22461096aae1SMatthew Knepley @*/
22477087cfbeSBarry Smith PetscErrorCode  SNESGetRhs(SNES snes,Vec *rhs)
22481096aae1SMatthew Knepley {
22491096aae1SMatthew Knepley   PetscFunctionBegin;
22500700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
22511096aae1SMatthew Knepley   PetscValidPointer(rhs,2);
225285385478SLisandro Dalcin   *rhs = snes->vec_rhs;
22531096aae1SMatthew Knepley   PetscFunctionReturn(0);
22541096aae1SMatthew Knepley }
22551096aae1SMatthew Knepley 
22569b94acceSBarry Smith /*@
2257bf388a1fSBarry Smith    SNESComputeFunction - Calls the function that has been set with SNESSetFunction().
22589b94acceSBarry Smith 
2259c7afd0dbSLois Curfman McInnes    Collective on SNES
2260c7afd0dbSLois Curfman McInnes 
22619b94acceSBarry Smith    Input Parameters:
2262c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2263c7afd0dbSLois Curfman McInnes -  x - input vector
22649b94acceSBarry Smith 
22659b94acceSBarry Smith    Output Parameter:
22663638b69dSLois Curfman McInnes .  y - function vector, as set by SNESSetFunction()
22679b94acceSBarry Smith 
22681bffabb2SLois Curfman McInnes    Notes:
226936851e7fSLois Curfman McInnes    SNESComputeFunction() is typically used within nonlinear solvers
227036851e7fSLois Curfman McInnes    implementations, so most users would not generally call this routine
227136851e7fSLois Curfman McInnes    themselves.
227236851e7fSLois Curfman McInnes 
227336851e7fSLois Curfman McInnes    Level: developer
227436851e7fSLois Curfman McInnes 
22759b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function
22769b94acceSBarry Smith 
2277a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction()
22789b94acceSBarry Smith @*/
22797087cfbeSBarry Smith PetscErrorCode  SNESComputeFunction(SNES snes,Vec x,Vec y)
22809b94acceSBarry Smith {
2281dfbe8321SBarry Smith   PetscErrorCode ierr;
22826cab3a1bSJed Brown   DM             dm;
2283942e3340SBarry Smith   DMSNES         sdm;
22849b94acceSBarry Smith 
22853a40ed3dSBarry Smith   PetscFunctionBegin;
22860700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
22870700a824SBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
22880700a824SBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2289c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,x,2);
2290c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,3);
229162796dfbSBarry Smith   ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr);
2292184914b5SBarry Smith 
22936cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2294942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
229532f3f7c2SPeter Brune   if (sdm->ops->computefunction) {
229694db00ebSBarry Smith     if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) {
2297ccf3c845SPeter Brune       ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
229894db00ebSBarry Smith     }
22995edff71fSBarry Smith     ierr = VecLockPush(x);CHKERRQ(ierr);
2300d64ed03dSBarry Smith     PetscStackPush("SNES user function");
230122c6f798SBarry Smith     ierr = (*sdm->ops->computefunction)(snes,x,y,sdm->functionctx);CHKERRQ(ierr);
2302d64ed03dSBarry Smith     PetscStackPop;
23035edff71fSBarry Smith     ierr = VecLockPop(x);CHKERRQ(ierr);
230494db00ebSBarry Smith     if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) {
2305ccf3c845SPeter Brune       ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr);
230694db00ebSBarry Smith     }
2307c90fad12SPeter Brune   } else if (snes->vec_rhs) {
2308c90fad12SPeter Brune     ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr);
2309644e2e5bSBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve().");
231085385478SLisandro Dalcin   if (snes->vec_rhs) {
231185385478SLisandro Dalcin     ierr = VecAXPY(y,-1.0,snes->vec_rhs);CHKERRQ(ierr);
23123ab0aad5SBarry Smith   }
2313ae3c334cSLois Curfman McInnes   snes->nfuncs++;
2314422a814eSBarry Smith   /*
2315422a814eSBarry Smith      domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will
2316422a814eSBarry Smith      propagate the value to all processes
2317422a814eSBarry Smith   */
2318422a814eSBarry Smith   if (snes->domainerror) {
2319422a814eSBarry Smith     ierr = VecSetInf(y);CHKERRQ(ierr);
2320422a814eSBarry Smith   }
23213a40ed3dSBarry Smith   PetscFunctionReturn(0);
23229b94acceSBarry Smith }
23239b94acceSBarry Smith 
2324c79ef259SPeter Brune /*@
2325be95d8f1SBarry Smith    SNESComputeNGS - Calls the Gauss-Seidel function that has been set with  SNESSetNGS().
2326c79ef259SPeter Brune 
2327c79ef259SPeter Brune    Collective on SNES
2328c79ef259SPeter Brune 
2329c79ef259SPeter Brune    Input Parameters:
2330c79ef259SPeter Brune +  snes - the SNES context
2331c79ef259SPeter Brune .  x - input vector
2332c79ef259SPeter Brune -  b - rhs vector
2333c79ef259SPeter Brune 
2334c79ef259SPeter Brune    Output Parameter:
2335c79ef259SPeter Brune .  x - new solution vector
2336c79ef259SPeter Brune 
2337c79ef259SPeter Brune    Notes:
2338be95d8f1SBarry Smith    SNESComputeNGS() is typically used within composed nonlinear solver
2339c79ef259SPeter Brune    implementations, so most users would not generally call this routine
2340c79ef259SPeter Brune    themselves.
2341c79ef259SPeter Brune 
2342c79ef259SPeter Brune    Level: developer
2343c79ef259SPeter Brune 
2344c79ef259SPeter Brune .keywords: SNES, nonlinear, compute, function
2345c79ef259SPeter Brune 
2346be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESComputeFunction()
2347c79ef259SPeter Brune @*/
2348be95d8f1SBarry Smith PetscErrorCode  SNESComputeNGS(SNES snes,Vec b,Vec x)
2349646217ecSPeter Brune {
2350646217ecSPeter Brune   PetscErrorCode ierr;
23516cab3a1bSJed Brown   DM             dm;
2352942e3340SBarry Smith   DMSNES         sdm;
2353646217ecSPeter Brune 
2354646217ecSPeter Brune   PetscFunctionBegin;
2355646217ecSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2356646217ecSPeter Brune   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2357646217ecSPeter Brune   if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,3);
2358646217ecSPeter Brune   PetscCheckSameComm(snes,1,x,2);
2359646217ecSPeter Brune   if (b) PetscCheckSameComm(snes,1,b,3);
236062796dfbSBarry Smith   if (b) {ierr = VecValidValues(b,2,PETSC_TRUE);CHKERRQ(ierr);}
2361be95d8f1SBarry Smith   ierr = PetscLogEventBegin(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr);
23626cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2363942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
236422c6f798SBarry Smith   if (sdm->ops->computegs) {
23655edff71fSBarry Smith     if (b) {ierr = VecLockPush(b);CHKERRQ(ierr);}
2366be95d8f1SBarry Smith     PetscStackPush("SNES user NGS");
236722c6f798SBarry Smith     ierr = (*sdm->ops->computegs)(snes,x,b,sdm->gsctx);CHKERRQ(ierr);
2368646217ecSPeter Brune     PetscStackPop;
23695edff71fSBarry Smith     if (b) {ierr = VecLockPop(b);CHKERRQ(ierr);}
2370be95d8f1SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetNGS() before SNESComputeNGS(), likely called from SNESSolve().");
2371be95d8f1SBarry Smith   ierr = PetscLogEventEnd(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr);
2372646217ecSPeter Brune   PetscFunctionReturn(0);
2373646217ecSPeter Brune }
2374646217ecSPeter Brune 
2375e885f1abSBarry Smith PetscErrorCode SNESTestJacobian(SNES snes)
2376e885f1abSBarry Smith {
237712837594SBarry Smith   Mat               A,B,C,D,jacobian;
2378e885f1abSBarry Smith   Vec               x = snes->vec_sol,f = snes->vec_func;
2379e885f1abSBarry Smith   PetscErrorCode    ierr;
2380e885f1abSBarry Smith   PetscReal         nrm,gnorm;
238181e7118cSBarry Smith   PetscReal         threshold = 1.e-5;
2382e885f1abSBarry Smith   PetscInt          m,n,M,N;
2383e885f1abSBarry Smith   void              *functx;
238418d89885SKarl Rupp   PetscBool         complete_print = PETSC_FALSE,threshold_print = PETSC_FALSE,test = PETSC_FALSE,flg;
23853325ff46SBarry Smith   PetscViewer       viewer,mviewer;
2386e885f1abSBarry Smith   MPI_Comm          comm;
2387e885f1abSBarry Smith   PetscInt          tabs;
238812837594SBarry Smith   static PetscBool  directionsprinted = PETSC_FALSE;
23893325ff46SBarry Smith   PetscViewerFormat format;
2390e885f1abSBarry Smith 
2391e885f1abSBarry Smith   PetscFunctionBegin;
2392fc35ed60SBarry Smith   ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr);
239312837594SBarry Smith   ierr = PetscOptionsName("-snes_test_jacobian","Compare hand-coded and finite difference Jacobians","None",&test);CHKERRQ(ierr);
239412837594SBarry Smith   ierr = PetscOptionsReal("-snes_test_jacobian", "Threshold for element difference between hand-coded and finite difference being meaningful", "None", threshold, &threshold,NULL);CHKERRQ(ierr);
23953325ff46SBarry 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);
239618d89885SKarl Rupp   if (!complete_print) {
239718d89885SKarl Rupp     ierr = PetscOptionsViewer("-snes_test_jacobian_display","Display difference between hand-coded and finite difference Jacobians","None",&mviewer,&format,&complete_print);CHKERRQ(ierr);
239818d89885SKarl Rupp   }
239918d89885SKarl Rupp   /* for compatibility with PETSc 3.9 and older. */
240018d89885SKarl 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);
2401e885f1abSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
2402e885f1abSBarry Smith   if (!test) PetscFunctionReturn(0);
2403e885f1abSBarry Smith 
2404e885f1abSBarry Smith   ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr);
2405e885f1abSBarry Smith   ierr = PetscViewerASCIIGetStdout(comm,&viewer);CHKERRQ(ierr);
2406e885f1abSBarry Smith   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
2407e885f1abSBarry Smith   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)snes)->tablevel);CHKERRQ(ierr);
240812837594SBarry Smith   ierr = PetscViewerASCIIPrintf(viewer,"  ---------- Testing Jacobian -------------\n");CHKERRQ(ierr);
240912837594SBarry Smith   if (!complete_print && !directionsprinted) {
241012837594SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Run with -snes_test_jacobian_view and optionally -snes_test_jacobian <threshold> to show difference\n");CHKERRQ(ierr);
2411fc35ed60SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"    of hand-coded and finite difference Jacobian entries greater than <threshold>.\n");CHKERRQ(ierr);
241212837594SBarry Smith   }
241312837594SBarry Smith   if (!directionsprinted) {
241412837594SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Testing hand-coded Jacobian, if (for double precision runs) ||J - Jfd||_F/||J||_F is\n");CHKERRQ(ierr);
2415e885f1abSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"    O(1.e-8), the hand-coded Jacobian is probably correct.\n");CHKERRQ(ierr);
241612837594SBarry Smith     directionsprinted = PETSC_TRUE;
2417e885f1abSBarry Smith   }
24183325ff46SBarry Smith   if (complete_print) {
24193325ff46SBarry Smith     ierr = PetscViewerPushFormat(mviewer,format);CHKERRQ(ierr);
2420e885f1abSBarry Smith   }
2421e885f1abSBarry Smith 
242212837594SBarry Smith   /* evaluate the function at this point because SNESComputeJacobianDefault() assumes that the function has been evaluated and put into snes->vec_func */
2423e885f1abSBarry Smith   ierr = SNESComputeFunction(snes,x,f);CHKERRQ(ierr);
242412837594SBarry Smith 
242512837594SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)snes->jacobian,MATMFFD,&flg);CHKERRQ(ierr);
242612837594SBarry Smith   if (!flg) jacobian = snes->jacobian;
242712837594SBarry Smith   else jacobian = snes->jacobian_pre;
242812837594SBarry Smith 
242912837594SBarry Smith   while (jacobian) {
243012837594SBarry Smith     ierr = PetscObjectTypeCompareAny((PetscObject)jacobian,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr);
243112837594SBarry Smith     if (flg) {
243212837594SBarry Smith       A    = jacobian;
243312837594SBarry Smith       ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
243412837594SBarry Smith     } else {
243512837594SBarry Smith       ierr = MatComputeExplicitOperator(jacobian,&A);CHKERRQ(ierr);
243612837594SBarry Smith     }
2437e885f1abSBarry Smith 
2438e885f1abSBarry Smith     ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
2439e885f1abSBarry Smith     ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
2440e885f1abSBarry Smith     ierr = MatGetLocalSize(A,&m,&n);CHKERRQ(ierr);
2441e885f1abSBarry Smith     ierr = MatSetSizes(B,m,n,M,N);CHKERRQ(ierr);
2442e885f1abSBarry Smith     ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr);
2443e885f1abSBarry Smith     ierr = MatSetUp(B);CHKERRQ(ierr);
2444e885f1abSBarry Smith     ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr);
2445e885f1abSBarry Smith 
2446e885f1abSBarry Smith     ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr);
2447e885f1abSBarry Smith     ierr = SNESComputeJacobianDefault(snes,x,B,B,functx);CHKERRQ(ierr);
244812837594SBarry Smith 
244912837594SBarry Smith     ierr = MatDuplicate(B,MAT_COPY_VALUES,&D);CHKERRQ(ierr);
245012837594SBarry Smith     ierr = MatAYPX(D,-1.0,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
245112837594SBarry Smith     ierr = MatNorm(D,NORM_FROBENIUS,&nrm);CHKERRQ(ierr);
2452e885f1abSBarry Smith     ierr = MatNorm(A,NORM_FROBENIUS,&gnorm);CHKERRQ(ierr);
245312837594SBarry Smith     ierr = MatDestroy(&D);CHKERRQ(ierr);
245412837594SBarry Smith     if (!gnorm) gnorm = 1; /* just in case */
245512837594SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  ||J - Jfd||_F/||J||_F = %g, ||J - Jfd||_F = %g\n",(double)(nrm/gnorm),(double)nrm);CHKERRQ(ierr);
245612837594SBarry Smith 
2457e885f1abSBarry Smith     if (complete_print) {
245812837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Hand-coded Jacobian ----------\n");CHKERRQ(ierr);
24593325ff46SBarry Smith       ierr = MatView(jacobian,mviewer);CHKERRQ(ierr);
246012837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Finite difference Jacobian ----------\n");CHKERRQ(ierr);
24613325ff46SBarry Smith       ierr = MatView(B,mviewer);CHKERRQ(ierr);
2462e885f1abSBarry Smith     }
2463e885f1abSBarry Smith 
2464df10fb39SFande Kong     if (threshold_print || complete_print) {
2465e885f1abSBarry Smith       PetscInt          Istart, Iend, *ccols, bncols, cncols, j, row;
2466e885f1abSBarry Smith       PetscScalar       *cvals;
2467e885f1abSBarry Smith       const PetscInt    *bcols;
2468e885f1abSBarry Smith       const PetscScalar *bvals;
2469e885f1abSBarry Smith 
247012837594SBarry Smith       ierr = MatAYPX(B,-1.0,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
2471e885f1abSBarry Smith       ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
2472e885f1abSBarry Smith       ierr = MatSetSizes(C,m,n,M,N);CHKERRQ(ierr);
2473e885f1abSBarry Smith       ierr = MatSetType(C,((PetscObject)A)->type_name);CHKERRQ(ierr);
2474e885f1abSBarry Smith       ierr = MatSetUp(C);CHKERRQ(ierr);
2475e885f1abSBarry Smith       ierr = MatSetOption(C,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr);
2476e885f1abSBarry Smith       ierr = MatGetOwnershipRange(B,&Istart,&Iend);CHKERRQ(ierr);
2477e885f1abSBarry Smith 
2478e885f1abSBarry Smith       for (row = Istart; row < Iend; row++) {
2479e885f1abSBarry Smith         ierr = MatGetRow(B,row,&bncols,&bcols,&bvals);CHKERRQ(ierr);
2480e885f1abSBarry Smith         ierr = PetscMalloc2(bncols,&ccols,bncols,&cvals);CHKERRQ(ierr);
2481e885f1abSBarry Smith         for (j = 0, cncols = 0; j < bncols; j++) {
248223a52b1dSBarry Smith           if (PetscAbsScalar(bvals[j]) > threshold) {
2483e885f1abSBarry Smith             ccols[cncols] = bcols[j];
2484e885f1abSBarry Smith             cvals[cncols] = bvals[j];
2485e885f1abSBarry Smith             cncols += 1;
2486e885f1abSBarry Smith           }
2487e885f1abSBarry Smith         }
2488e885f1abSBarry Smith         if (cncols) {
2489e885f1abSBarry Smith           ierr = MatSetValues(C,1,&row,cncols,ccols,cvals,INSERT_VALUES);CHKERRQ(ierr);
2490e885f1abSBarry Smith         }
2491e885f1abSBarry Smith         ierr = MatRestoreRow(B,row,&bncols,&bcols,&bvals);CHKERRQ(ierr);
2492e885f1abSBarry Smith         ierr = PetscFree2(ccols,cvals);CHKERRQ(ierr);
2493e885f1abSBarry Smith       }
2494e885f1abSBarry Smith       ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2495e885f1abSBarry Smith       ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
249612837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Hand-coded minus finite-difference Jacobian with tolerance %g ----------\n",(double)threshold);CHKERRQ(ierr);
249718d89885SKarl Rupp       ierr = MatView(C,complete_print ? mviewer : viewer);CHKERRQ(ierr);
2498e885f1abSBarry Smith       ierr = MatDestroy(&C);CHKERRQ(ierr);
2499e885f1abSBarry Smith     }
250012837594SBarry Smith     ierr = MatDestroy(&A);CHKERRQ(ierr);
2501e885f1abSBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
250212837594SBarry Smith 
250312837594SBarry Smith     if (jacobian != snes->jacobian_pre) {
250412837594SBarry Smith       jacobian = snes->jacobian_pre;
250512837594SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  ---------- Testing Jacobian for preconditioner -------------\n");CHKERRQ(ierr);
250612837594SBarry Smith     }
250712837594SBarry Smith     else jacobian = NULL;
250812837594SBarry Smith   }
25093325ff46SBarry Smith   if (complete_print) {
25103325ff46SBarry Smith     ierr = PetscViewerPopFormat(mviewer);CHKERRQ(ierr);
25113325ff46SBarry Smith   }
2512a0c90127SFande Kong   if (mviewer) { ierr = PetscViewerDestroy(&mviewer);CHKERRQ(ierr); }
2513e885f1abSBarry Smith   ierr = PetscViewerASCIISetTab(viewer,tabs);CHKERRQ(ierr);
2514e885f1abSBarry Smith   PetscFunctionReturn(0);
2515e885f1abSBarry Smith }
2516e885f1abSBarry Smith 
251762fef451SLois Curfman McInnes /*@
2518bf388a1fSBarry Smith    SNESComputeJacobian - Computes the Jacobian matrix that has been set with SNESSetJacobian().
251962fef451SLois Curfman McInnes 
2520c7afd0dbSLois Curfman McInnes    Collective on SNES and Mat
2521c7afd0dbSLois Curfman McInnes 
252262fef451SLois Curfman McInnes    Input Parameters:
2523c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2524c7afd0dbSLois Curfman McInnes -  x - input vector
252562fef451SLois Curfman McInnes 
252662fef451SLois Curfman McInnes    Output Parameters:
2527c7afd0dbSLois Curfman McInnes +  A - Jacobian matrix
2528d1e9a80fSBarry Smith -  B - optional preconditioning matrix
2529fee21e36SBarry Smith 
2530e35cf81dSBarry Smith   Options Database Keys:
2531e35cf81dSBarry Smith +    -snes_lag_preconditioner <lag>
2532693365a8SJed Brown .    -snes_lag_jacobian <lag>
2533e885f1abSBarry Smith .    -snes_test_jacobian - compare the user provided Jacobian with one compute via finite differences to check for errors
2534e885f1abSBarry Smith .    -snes_test_jacobian_display - 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
2535e885f1abSBarry Smith .    -snes_test_jacobian_display_threshold <numerical value>  - display entries in the difference between the user provided Jacobian and finite difference Jacobian that are greater than a certain value to help users detect errors
2536693365a8SJed Brown .    -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences
2537693365a8SJed Brown .    -snes_compare_explicit_draw  - Compare the computed Jacobian to the finite difference Jacobian and draw the result
2538693365a8SJed Brown .    -snes_compare_explicit_contour  - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result
25394c30e9fbSJed Brown .    -snes_compare_operator  - Make the comparison options above use the operator instead of the preconditioning matrix
254094d6a431SBarry Smith .    -snes_compare_coloring - Compute the finite difference Jacobian using coloring and display norms of difference
2541c01495d3SJed Brown .    -snes_compare_coloring_display - Compute the finite differece Jacobian using coloring and display verbose differences
2542c01495d3SJed Brown .    -snes_compare_coloring_threshold - Display only those matrix entries that differ by more than a given threshold
2543c01495d3SJed Brown .    -snes_compare_coloring_threshold_atol - Absolute tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold
2544c01495d3SJed Brown .    -snes_compare_coloring_threshold_rtol - Relative tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold
25454c30e9fbSJed Brown .    -snes_compare_coloring_draw - Compute the finite differece Jacobian using coloring and draw differences
2546c01495d3SJed Brown -    -snes_compare_coloring_draw_contour - Compute the finite differece Jacobian using coloring and show contours of matrices and differences
2547c01495d3SJed Brown 
2548e35cf81dSBarry Smith 
254962fef451SLois Curfman McInnes    Notes:
255062fef451SLois Curfman McInnes    Most users should not need to explicitly call this routine, as it
255162fef451SLois Curfman McInnes    is used internally within the nonlinear solvers.
255262fef451SLois Curfman McInnes 
255395452b02SPatrick Sanan    Developer Notes:
255495452b02SPatrick 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
2555e885f1abSBarry Smith       for with the SNESType of test that has been removed.
2556e885f1abSBarry Smith 
255736851e7fSLois Curfman McInnes    Level: developer
255836851e7fSLois Curfman McInnes 
255962fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix
256062fef451SLois Curfman McInnes 
2561e35cf81dSBarry Smith .seealso:  SNESSetJacobian(), KSPSetOperators(), MatStructure, SNESSetLagPreconditioner(), SNESSetLagJacobian()
256262fef451SLois Curfman McInnes @*/
2563d1e9a80fSBarry Smith PetscErrorCode  SNESComputeJacobian(SNES snes,Vec X,Mat A,Mat B)
25649b94acceSBarry Smith {
2565dfbe8321SBarry Smith   PetscErrorCode ierr;
2566ace3abfcSBarry Smith   PetscBool      flag;
25676cab3a1bSJed Brown   DM             dm;
2568942e3340SBarry Smith   DMSNES         sdm;
2569e0e3a89bSBarry Smith   KSP            ksp;
25703a40ed3dSBarry Smith 
25713a40ed3dSBarry Smith   PetscFunctionBegin;
25720700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
25730700a824SBarry Smith   PetscValidHeaderSpecific(X,VEC_CLASSID,2);
2574c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,X,2);
257562796dfbSBarry Smith   ierr = VecValidValues(X,2,PETSC_TRUE);CHKERRQ(ierr);
25766cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2577942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
25783232da50SPeter Brune 
2579ce94432eSBarry Smith   if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Must call SNESSetJacobian(), DMSNESSetJacobian(), DMDASNESSetJacobianLocal(), etc");
2580ebd3b9afSBarry Smith 
2581ebd3b9afSBarry Smith   /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */
2582ebd3b9afSBarry Smith 
2583fe3ffe1eSBarry Smith   if (snes->lagjacobian == -2) {
2584fe3ffe1eSBarry Smith     snes->lagjacobian = -1;
2585f5af7f23SKarl Rupp 
2586fe3ffe1eSBarry Smith     ierr = PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n");CHKERRQ(ierr);
2587fe3ffe1eSBarry Smith   } else if (snes->lagjacobian == -1) {
2588e35cf81dSBarry Smith     ierr = PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n");CHKERRQ(ierr);
258994ab13aaSBarry Smith     ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr);
2590ebd3b9afSBarry Smith     if (flag) {
259194ab13aaSBarry Smith       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
259294ab13aaSBarry Smith       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2593ebd3b9afSBarry Smith     }
2594e35cf81dSBarry Smith     PetscFunctionReturn(0);
259537ec4e1aSPeter Brune   } else if (snes->lagjacobian > 1 && (snes->iter + snes->jac_iter) % snes->lagjacobian) {
2596e35cf81dSBarry Smith     ierr = PetscInfo2(snes,"Reusing Jacobian/preconditioner because lag is %D and SNES iteration is %D\n",snes->lagjacobian,snes->iter);CHKERRQ(ierr);
259794ab13aaSBarry Smith     ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr);
2598ebd3b9afSBarry Smith     if (flag) {
259994ab13aaSBarry Smith       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
260094ab13aaSBarry Smith       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2601ebd3b9afSBarry Smith     }
2602e35cf81dSBarry Smith     PetscFunctionReturn(0);
2603e35cf81dSBarry Smith   }
2604efd4aadfSBarry Smith   if (snes->npc && snes->npcside== PC_LEFT) {
260594ab13aaSBarry Smith       ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
260694ab13aaSBarry Smith       ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2607d728fb7dSPeter Brune       PetscFunctionReturn(0);
2608d728fb7dSPeter Brune   }
2609e35cf81dSBarry Smith 
261094ab13aaSBarry Smith   ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr);
26115edff71fSBarry Smith   ierr = VecLockPush(X);CHKERRQ(ierr);
2612d64ed03dSBarry Smith   PetscStackPush("SNES user Jacobian function");
2613d1e9a80fSBarry Smith   ierr = (*sdm->ops->computejacobian)(snes,X,A,B,sdm->jacobianctx);CHKERRQ(ierr);
2614d64ed03dSBarry Smith   PetscStackPop;
26155edff71fSBarry Smith   ierr = VecLockPop(X);CHKERRQ(ierr);
261694ab13aaSBarry Smith   ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr);
2617a8054027SBarry Smith 
2618e0e3a89bSBarry Smith   /* the next line ensures that snes->ksp exists */
2619e0e3a89bSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
26203b4f5425SBarry Smith   if (snes->lagpreconditioner == -2) {
26213b4f5425SBarry Smith     ierr = PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n");CHKERRQ(ierr);
2622d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr);
26233b4f5425SBarry Smith     snes->lagpreconditioner = -1;
26243b4f5425SBarry Smith   } else if (snes->lagpreconditioner == -1) {
2625a8054027SBarry Smith     ierr = PetscInfo(snes,"Reusing preconditioner because lag is -1\n");CHKERRQ(ierr);
2626d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr);
262737ec4e1aSPeter Brune   } else if (snes->lagpreconditioner > 1 && (snes->iter + snes->pre_iter) % snes->lagpreconditioner) {
2628a8054027SBarry Smith     ierr = PetscInfo2(snes,"Reusing preconditioner because lag is %D and SNES iteration is %D\n",snes->lagpreconditioner,snes->iter);CHKERRQ(ierr);
2629d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr);
2630d1e9a80fSBarry Smith   } else {
2631d1e9a80fSBarry Smith     ierr = PetscInfo(snes,"Rebuilding preconditioner\n");CHKERRQ(ierr);
2632d1e9a80fSBarry Smith     ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr);
2633a8054027SBarry Smith   }
2634a8054027SBarry Smith 
2635e885f1abSBarry Smith   ierr = SNESTestJacobian(snes);CHKERRQ(ierr);
26366d84be18SBarry Smith   /* make sure user returned a correct Jacobian and preconditioner */
263794ab13aaSBarry Smith   /* PetscValidHeaderSpecific(A,MAT_CLASSID,3);
263894ab13aaSBarry Smith     PetscValidHeaderSpecific(B,MAT_CLASSID,4);   */
2639693365a8SJed Brown   {
2640693365a8SJed Brown     PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE;
264116413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit",NULL,NULL,&flag);CHKERRQ(ierr);
264216413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr);
264316413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr);
264416413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_operator",NULL,NULL,&flag_operator);CHKERRQ(ierr);
2645693365a8SJed Brown     if (flag || flag_draw || flag_contour) {
26460298fd71SBarry Smith       Mat          Bexp_mine = NULL,Bexp,FDexp;
2647693365a8SJed Brown       PetscViewer  vdraw,vstdout;
26486b3a5b13SJed Brown       PetscBool    flg;
2649693365a8SJed Brown       if (flag_operator) {
265094ab13aaSBarry Smith         ierr = MatComputeExplicitOperator(A,&Bexp_mine);CHKERRQ(ierr);
2651693365a8SJed Brown         Bexp = Bexp_mine;
2652693365a8SJed Brown       } else {
2653693365a8SJed Brown         /* See if the preconditioning matrix can be viewed and added directly */
265494ab13aaSBarry Smith         ierr = PetscObjectTypeCompareAny((PetscObject)B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr);
265594ab13aaSBarry Smith         if (flg) Bexp = B;
2656693365a8SJed Brown         else {
2657693365a8SJed Brown           /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */
265894ab13aaSBarry Smith           ierr = MatComputeExplicitOperator(B,&Bexp_mine);CHKERRQ(ierr);
2659693365a8SJed Brown           Bexp = Bexp_mine;
2660693365a8SJed Brown         }
2661693365a8SJed Brown       }
2662693365a8SJed Brown       ierr = MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp);CHKERRQ(ierr);
2663d1e9a80fSBarry Smith       ierr = SNESComputeJacobianDefault(snes,X,FDexp,FDexp,NULL);CHKERRQ(ierr);
2664ce94432eSBarry Smith       ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr);
2665693365a8SJed Brown       if (flag_draw || flag_contour) {
2666ce94432eSBarry Smith         ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
2667693365a8SJed Brown         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
26680298fd71SBarry Smith       } else vdraw = NULL;
2669693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator ? "Jacobian" : "preconditioning Jacobian");CHKERRQ(ierr);
2670693365a8SJed Brown       if (flag) {ierr = MatView(Bexp,vstdout);CHKERRQ(ierr);}
2671693365a8SJed Brown       if (vdraw) {ierr = MatView(Bexp,vdraw);CHKERRQ(ierr);}
2672693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n");CHKERRQ(ierr);
2673693365a8SJed Brown       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
2674693365a8SJed Brown       if (vdraw) {ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);}
2675693365a8SJed Brown       ierr = MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
2676693365a8SJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n");CHKERRQ(ierr);
2677693365a8SJed Brown       if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);}
2678693365a8SJed Brown       if (vdraw) {              /* Always use contour for the difference */
2679693365a8SJed Brown         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
2680693365a8SJed Brown         ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);
2681693365a8SJed Brown         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
2682693365a8SJed Brown       }
2683693365a8SJed Brown       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
2684693365a8SJed Brown       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
2685693365a8SJed Brown       ierr = MatDestroy(&Bexp_mine);CHKERRQ(ierr);
2686693365a8SJed Brown       ierr = MatDestroy(&FDexp);CHKERRQ(ierr);
2687693365a8SJed Brown     }
2688693365a8SJed Brown   }
26894c30e9fbSJed Brown   {
26906719d8e4SJed Brown     PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE;
26916719d8e4SJed Brown     PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON;
269216413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring",NULL,NULL,&flag);CHKERRQ(ierr);
269316413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_display",NULL,NULL,&flag_display);CHKERRQ(ierr);
269416413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr);
269516413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr);
269616413a6aSBarry Smith     ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",NULL,NULL,&flag_threshold);CHKERRQ(ierr);
269727b0f280SBarry Smith     if (flag_threshold) {
2698c5929fdfSBarry Smith       ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,NULL);CHKERRQ(ierr);
2699c5929fdfSBarry Smith       ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,NULL);CHKERRQ(ierr);
270027b0f280SBarry Smith     }
27016719d8e4SJed Brown     if (flag || flag_display || flag_draw || flag_contour || flag_threshold) {
27024c30e9fbSJed Brown       Mat            Bfd;
27034c30e9fbSJed Brown       PetscViewer    vdraw,vstdout;
2704335efc43SPeter Brune       MatColoring    coloring;
27054c30e9fbSJed Brown       ISColoring     iscoloring;
27064c30e9fbSJed Brown       MatFDColoring  matfdcoloring;
27074c30e9fbSJed Brown       PetscErrorCode (*func)(SNES,Vec,Vec,void*);
27084c30e9fbSJed Brown       void           *funcctx;
27096719d8e4SJed Brown       PetscReal      norm1,norm2,normmax;
27104c30e9fbSJed Brown 
271194ab13aaSBarry Smith       ierr = MatDuplicate(B,MAT_DO_NOT_COPY_VALUES,&Bfd);CHKERRQ(ierr);
2712335efc43SPeter Brune       ierr = MatColoringCreate(Bfd,&coloring);CHKERRQ(ierr);
2713335efc43SPeter Brune       ierr = MatColoringSetType(coloring,MATCOLORINGSL);CHKERRQ(ierr);
2714335efc43SPeter Brune       ierr = MatColoringSetFromOptions(coloring);CHKERRQ(ierr);
2715335efc43SPeter Brune       ierr = MatColoringApply(coloring,&iscoloring);CHKERRQ(ierr);
2716335efc43SPeter Brune       ierr = MatColoringDestroy(&coloring);CHKERRQ(ierr);
27174c30e9fbSJed Brown       ierr = MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring);CHKERRQ(ierr);
2718f86b9fbaSHong Zhang       ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr);
2719f86b9fbaSHong Zhang       ierr = MatFDColoringSetUp(Bfd,iscoloring,matfdcoloring);CHKERRQ(ierr);
27204c30e9fbSJed Brown       ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr);
27214c30e9fbSJed Brown 
27224c30e9fbSJed Brown       /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */
27230298fd71SBarry Smith       ierr = SNESGetFunction(snes,NULL,&func,&funcctx);CHKERRQ(ierr);
27244c30e9fbSJed Brown       ierr = MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode (*)(void))func,funcctx);CHKERRQ(ierr);
27254c30e9fbSJed Brown       ierr = PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix);CHKERRQ(ierr);
27264c30e9fbSJed Brown       ierr = PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_");CHKERRQ(ierr);
27274c30e9fbSJed Brown       ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr);
2728d1e9a80fSBarry Smith       ierr = MatFDColoringApply(Bfd,matfdcoloring,X,snes);CHKERRQ(ierr);
27294c30e9fbSJed Brown       ierr = MatFDColoringDestroy(&matfdcoloring);CHKERRQ(ierr);
27304c30e9fbSJed Brown 
2731ce94432eSBarry Smith       ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr);
27324c30e9fbSJed Brown       if (flag_draw || flag_contour) {
2733ce94432eSBarry Smith         ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr);
27344c30e9fbSJed Brown         if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);}
27350298fd71SBarry Smith       } else vdraw = NULL;
27364c30e9fbSJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n");CHKERRQ(ierr);
273794ab13aaSBarry Smith       if (flag_display) {ierr = MatView(B,vstdout);CHKERRQ(ierr);}
273894ab13aaSBarry Smith       if (vdraw) {ierr = MatView(B,vdraw);CHKERRQ(ierr);}
27394c30e9fbSJed Brown       ierr = PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n");CHKERRQ(ierr);
27406719d8e4SJed Brown       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
27414c30e9fbSJed Brown       if (vdraw) {ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);}
274294ab13aaSBarry Smith       ierr = MatAYPX(Bfd,-1.0,B,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
27434c30e9fbSJed Brown       ierr = MatNorm(Bfd,NORM_1,&norm1);CHKERRQ(ierr);
27446719d8e4SJed Brown       ierr = MatNorm(Bfd,NORM_FROBENIUS,&norm2);CHKERRQ(ierr);
27454c30e9fbSJed Brown       ierr = MatNorm(Bfd,NORM_MAX,&normmax);CHKERRQ(ierr);
274657622a8eSBarry 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);
27476719d8e4SJed Brown       if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);}
27484c30e9fbSJed Brown       if (vdraw) {              /* Always use contour for the difference */
27494c30e9fbSJed Brown         ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);
27504c30e9fbSJed Brown         ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);
27514c30e9fbSJed Brown         ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);
27524c30e9fbSJed Brown       }
27534c30e9fbSJed Brown       if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);}
27546719d8e4SJed Brown 
27556719d8e4SJed Brown       if (flag_threshold) {
27566719d8e4SJed Brown         PetscInt bs,rstart,rend,i;
275794ab13aaSBarry Smith         ierr = MatGetBlockSize(B,&bs);CHKERRQ(ierr);
275894ab13aaSBarry Smith         ierr = MatGetOwnershipRange(B,&rstart,&rend);CHKERRQ(ierr);
27596719d8e4SJed Brown         for (i=rstart; i<rend; i++) {
27606719d8e4SJed Brown           const PetscScalar *ba,*ca;
27616719d8e4SJed Brown           const PetscInt    *bj,*cj;
27626719d8e4SJed Brown           PetscInt          bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1;
27636719d8e4SJed Brown           PetscReal         maxentry = 0,maxdiff = 0,maxrdiff = 0;
276494ab13aaSBarry Smith           ierr = MatGetRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr);
27656719d8e4SJed Brown           ierr = MatGetRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
276694ab13aaSBarry Smith           if (bn != cn) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold");
27676719d8e4SJed Brown           for (j=0; j<bn; j++) {
27686719d8e4SJed Brown             PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
27696719d8e4SJed Brown             if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) {
27706719d8e4SJed Brown               maxentrycol = bj[j];
27716719d8e4SJed Brown               maxentry    = PetscRealPart(ba[j]);
27726719d8e4SJed Brown             }
27736719d8e4SJed Brown             if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) {
27746719d8e4SJed Brown               maxdiffcol = bj[j];
27756719d8e4SJed Brown               maxdiff    = PetscRealPart(ca[j]);
27766719d8e4SJed Brown             }
27776719d8e4SJed Brown             if (rdiff > maxrdiff) {
27786719d8e4SJed Brown               maxrdiffcol = bj[j];
27796719d8e4SJed Brown               maxrdiff    = rdiff;
27806719d8e4SJed Brown             }
27816719d8e4SJed Brown           }
27826719d8e4SJed Brown           if (maxrdiff > 1) {
278357622a8eSBarry 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);
27846719d8e4SJed Brown             for (j=0; j<bn; j++) {
27856719d8e4SJed Brown               PetscReal rdiff;
27866719d8e4SJed Brown               rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j]));
27876719d8e4SJed Brown               if (rdiff > 1) {
278857622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(vstdout," (%D,%g:%g)",bj[j],(double)PetscRealPart(ba[j]),(double)PetscRealPart(ca[j]));CHKERRQ(ierr);
27896719d8e4SJed Brown               }
27906719d8e4SJed Brown             }
27916719d8e4SJed Brown             ierr = PetscViewerASCIIPrintf(vstdout,"\n",i,maxentry,maxdiff,maxrdiff);CHKERRQ(ierr);
27926719d8e4SJed Brown           }
279394ab13aaSBarry Smith           ierr = MatRestoreRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr);
27946719d8e4SJed Brown           ierr = MatRestoreRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr);
27956719d8e4SJed Brown         }
27966719d8e4SJed Brown       }
27974c30e9fbSJed Brown       ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr);
27984c30e9fbSJed Brown       ierr = MatDestroy(&Bfd);CHKERRQ(ierr);
27994c30e9fbSJed Brown     }
28004c30e9fbSJed Brown   }
28013a40ed3dSBarry Smith   PetscFunctionReturn(0);
28029b94acceSBarry Smith }
28039b94acceSBarry Smith 
2804bf388a1fSBarry Smith /*MC
2805411c0326SBarry Smith     SNESJacobianFunction - Function used to convey the nonlinear Jacobian of the function to be solved by SNES
2806bf388a1fSBarry Smith 
2807bf388a1fSBarry Smith      Synopsis:
2808411c0326SBarry Smith      #include "petscsnes.h"
2809411c0326SBarry Smith      PetscErrorCode SNESJacobianFunction(SNES snes,Vec x,Mat Amat,Mat Pmat,void *ctx);
2810bf388a1fSBarry Smith 
2811bf388a1fSBarry Smith +  x - input vector
2812e5d3d808SBarry Smith .  Amat - the matrix that defines the (approximate) Jacobian
2813e5d3d808SBarry Smith .  Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat.
2814bf388a1fSBarry Smith -  ctx - [optional] user-defined Jacobian context
2815bf388a1fSBarry Smith 
2816878cb397SSatish Balay    Level: intermediate
2817878cb397SSatish Balay 
2818bf388a1fSBarry Smith .seealso:   SNESSetFunction(), SNESGetFunction(), SNESSetJacobian(), SNESGetJacobian()
2819bf388a1fSBarry Smith M*/
2820bf388a1fSBarry Smith 
28219b94acceSBarry Smith /*@C
28229b94acceSBarry Smith    SNESSetJacobian - Sets the function to compute Jacobian as well as the
2823044dda88SLois Curfman McInnes    location to store the matrix.
28249b94acceSBarry Smith 
28253f9fe445SBarry Smith    Logically Collective on SNES and Mat
2826c7afd0dbSLois Curfman McInnes 
28279b94acceSBarry Smith    Input Parameters:
2828c7afd0dbSLois Curfman McInnes +  snes - the SNES context
2829e5d3d808SBarry Smith .  Amat - the matrix that defines the (approximate) Jacobian
2830e5d3d808SBarry Smith .  Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat.
2831411c0326SBarry Smith .  J - Jacobian evaluation routine (if NULL then SNES retains any previously set value), see SNESJacobianFunction for details
2832c7afd0dbSLois Curfman McInnes -  ctx - [optional] user-defined context for private data for the
28330298fd71SBarry Smith          Jacobian evaluation routine (may be NULL) (if NULL then SNES retains any previously set value)
28349b94acceSBarry Smith 
28359b94acceSBarry Smith    Notes:
2836e5d3d808SBarry Smith    If the Amat matrix and Pmat matrix are different you must call MatAssemblyBegin/End() on
283716913363SBarry Smith    each matrix.
283816913363SBarry Smith 
2839895c21f2SBarry Smith    If you know the operator Amat has a null space you can use MatSetNullSpace() and MatSetTransposeNullSpace() to supply the null
2840895c21f2SBarry Smith    space to Amat and the KSP solvers will automatically use that null space as needed during the solution process.
2841895c21f2SBarry Smith 
28428d359177SBarry Smith    If using SNESComputeJacobianDefaultColor() to assemble a Jacobian, the ctx argument
2843a8a26c1eSJed Brown    must be a MatFDColoring.
2844a8a26c1eSJed Brown 
2845c3cc8fd1SJed Brown    Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian.  One common
2846c3cc8fd1SJed Brown    example is to use the "Picard linearization" which only differentiates through the highest order parts of each term.
2847c3cc8fd1SJed Brown 
284836851e7fSLois Curfman McInnes    Level: beginner
284936851e7fSLois Curfman McInnes 
28509b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix
28519b94acceSBarry Smith 
2852411c0326SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESComputeJacobianDefaultColor(), MatStructure, J,
2853411c0326SBarry Smith           SNESSetPicard(), SNESJacobianFunction
28549b94acceSBarry Smith @*/
2855d1e9a80fSBarry Smith PetscErrorCode  SNESSetJacobian(SNES snes,Mat Amat,Mat Pmat,PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx)
28569b94acceSBarry Smith {
2857dfbe8321SBarry Smith   PetscErrorCode ierr;
28586cab3a1bSJed Brown   DM             dm;
28593a7fca6bSBarry Smith 
28603a40ed3dSBarry Smith   PetscFunctionBegin;
28610700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2862e5d3d808SBarry Smith   if (Amat) PetscValidHeaderSpecific(Amat,MAT_CLASSID,2);
2863e5d3d808SBarry Smith   if (Pmat) PetscValidHeaderSpecific(Pmat,MAT_CLASSID,3);
2864e5d3d808SBarry Smith   if (Amat) PetscCheckSameComm(snes,1,Amat,2);
2865e5d3d808SBarry Smith   if (Pmat) PetscCheckSameComm(snes,1,Pmat,3);
28666cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2867f8b49ee9SBarry Smith   ierr = DMSNESSetJacobian(dm,J,ctx);CHKERRQ(ierr);
2868e5d3d808SBarry Smith   if (Amat) {
2869e5d3d808SBarry Smith     ierr = PetscObjectReference((PetscObject)Amat);CHKERRQ(ierr);
28706bf464f9SBarry Smith     ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
2871f5af7f23SKarl Rupp 
2872e5d3d808SBarry Smith     snes->jacobian = Amat;
28733a7fca6bSBarry Smith   }
2874e5d3d808SBarry Smith   if (Pmat) {
2875e5d3d808SBarry Smith     ierr = PetscObjectReference((PetscObject)Pmat);CHKERRQ(ierr);
28766bf464f9SBarry Smith     ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
2877f5af7f23SKarl Rupp 
2878e5d3d808SBarry Smith     snes->jacobian_pre = Pmat;
28793a7fca6bSBarry Smith   }
28803a40ed3dSBarry Smith   PetscFunctionReturn(0);
28819b94acceSBarry Smith }
288262fef451SLois Curfman McInnes 
2883c2aafc4cSSatish Balay /*@C
2884b4fd4287SBarry Smith    SNESGetJacobian - Returns the Jacobian matrix and optionally the user
2885b4fd4287SBarry Smith    provided context for evaluating the Jacobian.
2886b4fd4287SBarry Smith 
2887c7afd0dbSLois Curfman McInnes    Not Collective, but Mat object will be parallel if SNES object is
2888c7afd0dbSLois Curfman McInnes 
2889b4fd4287SBarry Smith    Input Parameter:
2890b4fd4287SBarry Smith .  snes - the nonlinear solver context
2891b4fd4287SBarry Smith 
2892b4fd4287SBarry Smith    Output Parameters:
2893e5d3d808SBarry Smith +  Amat - location to stash (approximate) Jacobian matrix (or NULL)
2894e5d3d808SBarry Smith .  Pmat - location to stash matrix used to compute the preconditioner (or NULL)
2895411c0326SBarry Smith .  J - location to put Jacobian function (or NULL), see SNESJacobianFunction for details on its calling sequence
28960298fd71SBarry Smith -  ctx - location to stash Jacobian ctx (or NULL)
2897fee21e36SBarry Smith 
289836851e7fSLois Curfman McInnes    Level: advanced
289936851e7fSLois Curfman McInnes 
2900411c0326SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian(), SNESJacobianFunction, SNESGetFunction()
2901b4fd4287SBarry Smith @*/
2902d1e9a80fSBarry Smith PetscErrorCode SNESGetJacobian(SNES snes,Mat *Amat,Mat *Pmat,PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx)
2903b4fd4287SBarry Smith {
29046cab3a1bSJed Brown   PetscErrorCode ierr;
29056cab3a1bSJed Brown   DM             dm;
2906942e3340SBarry Smith   DMSNES         sdm;
29076cab3a1bSJed Brown 
29083a40ed3dSBarry Smith   PetscFunctionBegin;
29090700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
2910e5d3d808SBarry Smith   if (Amat) *Amat = snes->jacobian;
2911e5d3d808SBarry Smith   if (Pmat) *Pmat = snes->jacobian_pre;
29126cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2913942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
2914f8b49ee9SBarry Smith   if (J) *J = sdm->ops->computejacobian;
29156cab3a1bSJed Brown   if (ctx) *ctx = sdm->jacobianctx;
29163a40ed3dSBarry Smith   PetscFunctionReturn(0);
2917b4fd4287SBarry Smith }
2918b4fd4287SBarry Smith 
29199b94acceSBarry Smith /*@
29209b94acceSBarry Smith    SNESSetUp - Sets up the internal data structures for the later use
2921272ac6f2SLois Curfman McInnes    of a nonlinear solver.
29229b94acceSBarry Smith 
2923fee21e36SBarry Smith    Collective on SNES
2924fee21e36SBarry Smith 
2925c7afd0dbSLois Curfman McInnes    Input Parameters:
292670e92668SMatthew Knepley .  snes - the SNES context
2927c7afd0dbSLois Curfman McInnes 
2928272ac6f2SLois Curfman McInnes    Notes:
2929272ac6f2SLois Curfman McInnes    For basic use of the SNES solvers the user need not explicitly call
2930272ac6f2SLois Curfman McInnes    SNESSetUp(), since these actions will automatically occur during
2931272ac6f2SLois Curfman McInnes    the call to SNESSolve().  However, if one wishes to control this
2932272ac6f2SLois Curfman McInnes    phase separately, SNESSetUp() should be called after SNESCreate()
2933272ac6f2SLois Curfman McInnes    and optional routines of the form SNESSetXXX(), but before SNESSolve().
2934272ac6f2SLois Curfman McInnes 
293536851e7fSLois Curfman McInnes    Level: advanced
293636851e7fSLois Curfman McInnes 
29379b94acceSBarry Smith .keywords: SNES, nonlinear, setup
29389b94acceSBarry Smith 
29399b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy()
29409b94acceSBarry Smith @*/
29417087cfbeSBarry Smith PetscErrorCode  SNESSetUp(SNES snes)
29429b94acceSBarry Smith {
2943dfbe8321SBarry Smith   PetscErrorCode ierr;
29446cab3a1bSJed Brown   DM             dm;
2945942e3340SBarry Smith   DMSNES         sdm;
2946c35f09e5SBarry Smith   SNESLineSearch linesearch, pclinesearch;
29476e2a1849SPeter Brune   void           *lsprectx,*lspostctx;
29486b2b7091SBarry Smith   PetscErrorCode (*precheck)(SNESLineSearch,Vec,Vec,PetscBool*,void*);
29496b2b7091SBarry Smith   PetscErrorCode (*postcheck)(SNESLineSearch,Vec,Vec,Vec,PetscBool*,PetscBool*,void*);
29506e2a1849SPeter Brune   PetscErrorCode (*func)(SNES,Vec,Vec,void*);
29516e2a1849SPeter Brune   Vec            f,fpc;
29526e2a1849SPeter Brune   void           *funcctx;
2953d1e9a80fSBarry Smith   PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*);
29541eb13d49SPeter Brune   void           *jacctx,*appctx;
295532b97717SPeter Brune   Mat            j,jpre;
29563a40ed3dSBarry Smith 
29573a40ed3dSBarry Smith   PetscFunctionBegin;
29580700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
29594dc4c822SBarry Smith   if (snes->setupcalled) PetscFunctionReturn(0);
29609b94acceSBarry Smith 
29617adad957SLisandro Dalcin   if (!((PetscObject)snes)->type_name) {
296204d7464bSBarry Smith     ierr = SNESSetType(snes,SNESNEWTONLS);CHKERRQ(ierr);
296385385478SLisandro Dalcin   }
296485385478SLisandro Dalcin 
29650298fd71SBarry Smith   ierr = SNESGetFunction(snes,&snes->vec_func,NULL,NULL);CHKERRQ(ierr);
296658c9b817SLisandro Dalcin 
29676cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2968942e3340SBarry Smith   ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr);
2969ce94432eSBarry Smith   if (!sdm->ops->computefunction) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE,"Function never provided to SNES object");
297022c6f798SBarry Smith   if (!sdm->ops->computejacobian) {
29718d359177SBarry Smith     ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr);
297219f7a02aSBarry Smith   }
29736cab3a1bSJed Brown   if (!snes->vec_func) {
29746cab3a1bSJed Brown     ierr = DMCreateGlobalVector(dm,&snes->vec_func);CHKERRQ(ierr);
2975214df951SJed Brown   }
2976efd51863SBarry Smith 
297722d28d08SBarry Smith   if (!snes->ksp) {
297822d28d08SBarry Smith     ierr = SNESGetKSP(snes, &snes->ksp);CHKERRQ(ierr);
297922d28d08SBarry Smith   }
2980b710008aSBarry Smith 
298122d28d08SBarry Smith   if (!snes->linesearch) {
29827601faf0SJed Brown     ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr);
298322d28d08SBarry Smith   }
2984ed07d7d7SPeter Brune   ierr = SNESLineSearchSetFunction(snes->linesearch,SNESComputeFunction);CHKERRQ(ierr);
29859e764e56SPeter Brune 
2986efd4aadfSBarry Smith   if (snes->npc && (snes->npcside== PC_LEFT)) {
2987172a4300SPeter Brune     snes->mf          = PETSC_TRUE;
2988172a4300SPeter Brune     snes->mf_operator = PETSC_FALSE;
2989172a4300SPeter Brune   }
2990d8f46077SPeter Brune 
2991efd4aadfSBarry Smith   if (snes->npc) {
29926e2a1849SPeter Brune     /* copy the DM over */
29936e2a1849SPeter Brune     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
2994efd4aadfSBarry Smith     ierr = SNESSetDM(snes->npc,dm);CHKERRQ(ierr);
29956e2a1849SPeter Brune 
29966e2a1849SPeter Brune     ierr = SNESGetFunction(snes,&f,&func,&funcctx);CHKERRQ(ierr);
29976e2a1849SPeter Brune     ierr = VecDuplicate(f,&fpc);CHKERRQ(ierr);
2998efd4aadfSBarry Smith     ierr = SNESSetFunction(snes->npc,fpc,func,funcctx);CHKERRQ(ierr);
299932b97717SPeter Brune     ierr = SNESGetJacobian(snes,&j,&jpre,&jac,&jacctx);CHKERRQ(ierr);
3000efd4aadfSBarry Smith     ierr = SNESSetJacobian(snes->npc,j,jpre,jac,jacctx);CHKERRQ(ierr);
30011eb13d49SPeter Brune     ierr = SNESGetApplicationContext(snes,&appctx);CHKERRQ(ierr);
3002efd4aadfSBarry Smith     ierr = SNESSetApplicationContext(snes->npc,appctx);CHKERRQ(ierr);
30036e2a1849SPeter Brune     ierr = VecDestroy(&fpc);CHKERRQ(ierr);
30046e2a1849SPeter Brune 
30056e2a1849SPeter Brune     /* copy the function pointers over */
3006efd4aadfSBarry Smith     ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr);
30076e2a1849SPeter Brune 
30086e2a1849SPeter Brune     /* default to 1 iteration */
3009efd4aadfSBarry Smith     ierr = SNESSetTolerances(snes->npc,0.0,0.0,0.0,1,snes->npc->max_funcs);CHKERRQ(ierr);
3010efd4aadfSBarry Smith     if (snes->npcside==PC_RIGHT) {
3011efd4aadfSBarry Smith       ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_FINAL_ONLY);CHKERRQ(ierr);
3012a9936a0cSPeter Brune     } else {
3013efd4aadfSBarry Smith       ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_NONE);CHKERRQ(ierr);
3014a9936a0cSPeter Brune     }
3015efd4aadfSBarry Smith     ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr);
30166e2a1849SPeter Brune 
30176e2a1849SPeter Brune     /* copy the line search context over */
30187601faf0SJed Brown     ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr);
3019efd4aadfSBarry Smith     ierr = SNESGetLineSearch(snes->npc,&pclinesearch);CHKERRQ(ierr);
30206b2b7091SBarry Smith     ierr = SNESLineSearchGetPreCheck(linesearch,&precheck,&lsprectx);CHKERRQ(ierr);
30216b2b7091SBarry Smith     ierr = SNESLineSearchGetPostCheck(linesearch,&postcheck,&lspostctx);CHKERRQ(ierr);
30226b2b7091SBarry Smith     ierr = SNESLineSearchSetPreCheck(pclinesearch,precheck,lsprectx);CHKERRQ(ierr);
30236b2b7091SBarry Smith     ierr = SNESLineSearchSetPostCheck(pclinesearch,postcheck,lspostctx);CHKERRQ(ierr);
30246e2a1849SPeter Brune     ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)linesearch, (PetscObject)pclinesearch);CHKERRQ(ierr);
30256e2a1849SPeter Brune   }
302632b97717SPeter Brune   if (snes->mf) {
302732b97717SPeter Brune     ierr = SNESSetUpMatrixFree_Private(snes, snes->mf_operator, snes->mf_version);CHKERRQ(ierr);
302832b97717SPeter Brune   }
302932b97717SPeter Brune   if (snes->ops->usercompute && !snes->user) {
303032b97717SPeter Brune     ierr = (*snes->ops->usercompute)(snes,(void**)&snes->user);CHKERRQ(ierr);
303132b97717SPeter Brune   }
30326e2a1849SPeter Brune 
303337ec4e1aSPeter Brune   snes->jac_iter = 0;
303437ec4e1aSPeter Brune   snes->pre_iter = 0;
303537ec4e1aSPeter Brune 
3036410397dcSLisandro Dalcin   if (snes->ops->setup) {
3037410397dcSLisandro Dalcin     ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr);
3038410397dcSLisandro Dalcin   }
303958c9b817SLisandro Dalcin 
3040efd4aadfSBarry Smith   if (snes->npc && (snes->npcside== PC_LEFT)) {
30416c67d002SPeter Brune     if (snes->functype == SNES_FUNCTION_PRECONDITIONED) {
304255d4788fSPeter Brune       ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr);
3043be95d8f1SBarry Smith       ierr = SNESLineSearchSetFunction(linesearch,SNESComputeFunctionDefaultNPC);CHKERRQ(ierr);
30446c67d002SPeter Brune     }
30456c67d002SPeter Brune   }
30466c67d002SPeter Brune 
30477aaed0d8SBarry Smith   snes->setupcalled = PETSC_TRUE;
30483a40ed3dSBarry Smith   PetscFunctionReturn(0);
30499b94acceSBarry Smith }
30509b94acceSBarry Smith 
305137596af1SLisandro Dalcin /*@
305237596af1SLisandro Dalcin    SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats
305337596af1SLisandro Dalcin 
305437596af1SLisandro Dalcin    Collective on SNES
305537596af1SLisandro Dalcin 
305637596af1SLisandro Dalcin    Input Parameter:
305737596af1SLisandro Dalcin .  snes - iterative context obtained from SNESCreate()
305837596af1SLisandro Dalcin 
3059d25893d9SBarry Smith    Level: intermediate
3060d25893d9SBarry Smith 
306195452b02SPatrick Sanan    Notes:
306295452b02SPatrick Sanan     Also calls the application context destroy routine set with SNESSetComputeApplicationContext()
306337596af1SLisandro Dalcin 
306437596af1SLisandro Dalcin .keywords: SNES, destroy
306537596af1SLisandro Dalcin 
306637596af1SLisandro Dalcin .seealso: SNESCreate(), SNESSetUp(), SNESSolve()
306737596af1SLisandro Dalcin @*/
306837596af1SLisandro Dalcin PetscErrorCode  SNESReset(SNES snes)
306937596af1SLisandro Dalcin {
307037596af1SLisandro Dalcin   PetscErrorCode ierr;
307137596af1SLisandro Dalcin 
307237596af1SLisandro Dalcin   PetscFunctionBegin;
307337596af1SLisandro Dalcin   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3074d25893d9SBarry Smith   if (snes->ops->userdestroy && snes->user) {
3075d25893d9SBarry Smith     ierr       = (*snes->ops->userdestroy)((void**)&snes->user);CHKERRQ(ierr);
30760298fd71SBarry Smith     snes->user = NULL;
3077d25893d9SBarry Smith   }
3078efd4aadfSBarry Smith   if (snes->npc) {
3079efd4aadfSBarry Smith     ierr = SNESReset(snes->npc);CHKERRQ(ierr);
30808a23116dSBarry Smith   }
30818a23116dSBarry Smith 
308237596af1SLisandro Dalcin   if (snes->ops->reset) {
308337596af1SLisandro Dalcin     ierr = (*snes->ops->reset)(snes);CHKERRQ(ierr);
308437596af1SLisandro Dalcin   }
30859e764e56SPeter Brune   if (snes->ksp) {
30869e764e56SPeter Brune     ierr = KSPReset(snes->ksp);CHKERRQ(ierr);
30879e764e56SPeter Brune   }
30889e764e56SPeter Brune 
30899e764e56SPeter Brune   if (snes->linesearch) {
3090f1c6b773SPeter Brune     ierr = SNESLineSearchReset(snes->linesearch);CHKERRQ(ierr);
30919e764e56SPeter Brune   }
30929e764e56SPeter Brune 
30936bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
30946bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
30956bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_sol_update);CHKERRQ(ierr);
30966bf464f9SBarry Smith   ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr);
30976bf464f9SBarry Smith   ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr);
30986bf464f9SBarry Smith   ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr);
3099c41d97abSJed Brown   ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr);
3100c41d97abSJed Brown   ierr = VecDestroyVecs(snes->nvwork,&snes->vwork);CHKERRQ(ierr);
3101f5af7f23SKarl Rupp 
310240fdac6aSLawrence Mitchell   snes->alwayscomputesfinalresidual = PETSC_FALSE;
310340fdac6aSLawrence Mitchell 
310437596af1SLisandro Dalcin   snes->nwork       = snes->nvwork = 0;
310537596af1SLisandro Dalcin   snes->setupcalled = PETSC_FALSE;
310637596af1SLisandro Dalcin   PetscFunctionReturn(0);
310737596af1SLisandro Dalcin }
310837596af1SLisandro Dalcin 
310952baeb72SSatish Balay /*@
31109b94acceSBarry Smith    SNESDestroy - Destroys the nonlinear solver context that was created
31119b94acceSBarry Smith    with SNESCreate().
31129b94acceSBarry Smith 
3113c7afd0dbSLois Curfman McInnes    Collective on SNES
3114c7afd0dbSLois Curfman McInnes 
31159b94acceSBarry Smith    Input Parameter:
31169b94acceSBarry Smith .  snes - the SNES context
31179b94acceSBarry Smith 
311836851e7fSLois Curfman McInnes    Level: beginner
311936851e7fSLois Curfman McInnes 
31209b94acceSBarry Smith .keywords: SNES, nonlinear, destroy
31219b94acceSBarry Smith 
312263a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve()
31239b94acceSBarry Smith @*/
31246bf464f9SBarry Smith PetscErrorCode  SNESDestroy(SNES *snes)
31259b94acceSBarry Smith {
31266849ba73SBarry Smith   PetscErrorCode ierr;
31273a40ed3dSBarry Smith 
31283a40ed3dSBarry Smith   PetscFunctionBegin;
31296bf464f9SBarry Smith   if (!*snes) PetscFunctionReturn(0);
31306bf464f9SBarry Smith   PetscValidHeaderSpecific((*snes),SNES_CLASSID,1);
31316bf464f9SBarry Smith   if (--((PetscObject)(*snes))->refct > 0) {*snes = 0; PetscFunctionReturn(0);}
3132d4bb536fSBarry Smith 
31336bf464f9SBarry Smith   ierr = SNESReset((*snes));CHKERRQ(ierr);
3134efd4aadfSBarry Smith   ierr = SNESDestroy(&(*snes)->npc);CHKERRQ(ierr);
31356b8b9a38SLisandro Dalcin 
3136e04113cfSBarry Smith   /* if memory was published with SAWs then destroy it */
3137e04113cfSBarry Smith   ierr = PetscObjectSAWsViewOff((PetscObject)*snes);CHKERRQ(ierr);
31386bf464f9SBarry Smith   if ((*snes)->ops->destroy) {ierr = (*((*snes))->ops->destroy)((*snes));CHKERRQ(ierr);}
31396d4c513bSLisandro Dalcin 
314033124788SMatthew G. Knepley   if ((*snes)->dm) {ierr = DMCoarsenHookRemove((*snes)->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,*snes);CHKERRQ(ierr);}
31416bf464f9SBarry Smith   ierr = DMDestroy(&(*snes)->dm);CHKERRQ(ierr);
31426bf464f9SBarry Smith   ierr = KSPDestroy(&(*snes)->ksp);CHKERRQ(ierr);
3143f1c6b773SPeter Brune   ierr = SNESLineSearchDestroy(&(*snes)->linesearch);CHKERRQ(ierr);
31446b8b9a38SLisandro Dalcin 
31456bf464f9SBarry Smith   ierr = PetscFree((*snes)->kspconvctx);CHKERRQ(ierr);
31466bf464f9SBarry Smith   if ((*snes)->ops->convergeddestroy) {
31476bf464f9SBarry Smith     ierr = (*(*snes)->ops->convergeddestroy)((*snes)->cnvP);CHKERRQ(ierr);
31486b8b9a38SLisandro Dalcin   }
31496bf464f9SBarry Smith   if ((*snes)->conv_malloc) {
31506bf464f9SBarry Smith     ierr = PetscFree((*snes)->conv_hist);CHKERRQ(ierr);
31516bf464f9SBarry Smith     ierr = PetscFree((*snes)->conv_hist_its);CHKERRQ(ierr);
315258c9b817SLisandro Dalcin   }
31536bf464f9SBarry Smith   ierr = SNESMonitorCancel((*snes));CHKERRQ(ierr);
3154a79aaaedSSatish Balay   ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr);
31553a40ed3dSBarry Smith   PetscFunctionReturn(0);
31569b94acceSBarry Smith }
31579b94acceSBarry Smith 
31589b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */
31599b94acceSBarry Smith 
3160a8054027SBarry Smith /*@
3161a8054027SBarry Smith    SNESSetLagPreconditioner - Determines when the preconditioner is rebuilt in the nonlinear solve.
3162a8054027SBarry Smith 
31633f9fe445SBarry Smith    Logically Collective on SNES
3164a8054027SBarry Smith 
3165a8054027SBarry Smith    Input Parameters:
3166a8054027SBarry Smith +  snes - the SNES context
3167a8054027SBarry 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
31683b4f5425SBarry Smith          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
3169a8054027SBarry Smith 
3170a8054027SBarry Smith    Options Database Keys:
3171a8054027SBarry Smith .    -snes_lag_preconditioner <lag>
3172a8054027SBarry Smith 
3173a8054027SBarry Smith    Notes:
3174a8054027SBarry Smith    The default is 1
3175a8054027SBarry Smith    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3176a8054027SBarry Smith    If  -1 is used before the very first nonlinear solve the preconditioner is still built because there is no previous preconditioner to use
3177a8054027SBarry Smith 
3178a8054027SBarry Smith    Level: intermediate
3179a8054027SBarry Smith 
3180a8054027SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
3181a8054027SBarry Smith 
3182e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian()
3183a8054027SBarry Smith 
3184a8054027SBarry Smith @*/
31857087cfbeSBarry Smith PetscErrorCode  SNESSetLagPreconditioner(SNES snes,PetscInt lag)
3186a8054027SBarry Smith {
3187a8054027SBarry Smith   PetscFunctionBegin;
31880700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3189e32f2f54SBarry Smith   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
3190e32f2f54SBarry Smith   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
3191c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,lag,2);
3192a8054027SBarry Smith   snes->lagpreconditioner = lag;
3193a8054027SBarry Smith   PetscFunctionReturn(0);
3194a8054027SBarry Smith }
3195a8054027SBarry Smith 
3196efd51863SBarry Smith /*@
3197efd51863SBarry Smith    SNESSetGridSequence - sets the number of steps of grid sequencing that SNES does
3198efd51863SBarry Smith 
3199efd51863SBarry Smith    Logically Collective on SNES
3200efd51863SBarry Smith 
3201efd51863SBarry Smith    Input Parameters:
3202efd51863SBarry Smith +  snes - the SNES context
3203efd51863SBarry Smith -  steps - the number of refinements to do, defaults to 0
3204efd51863SBarry Smith 
3205efd51863SBarry Smith    Options Database Keys:
3206efd51863SBarry Smith .    -snes_grid_sequence <steps>
3207efd51863SBarry Smith 
3208efd51863SBarry Smith    Level: intermediate
3209efd51863SBarry Smith 
3210c0df2a02SJed Brown    Notes:
3211c0df2a02SJed Brown    Use SNESGetSolution() to extract the fine grid solution after grid sequencing.
3212c0df2a02SJed Brown 
3213efd51863SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
3214efd51863SBarry Smith 
3215fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetGridSequence()
3216efd51863SBarry Smith 
3217efd51863SBarry Smith @*/
3218efd51863SBarry Smith PetscErrorCode  SNESSetGridSequence(SNES snes,PetscInt steps)
3219efd51863SBarry Smith {
3220efd51863SBarry Smith   PetscFunctionBegin;
3221efd51863SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3222efd51863SBarry Smith   PetscValidLogicalCollectiveInt(snes,steps,2);
3223efd51863SBarry Smith   snes->gridsequence = steps;
3224efd51863SBarry Smith   PetscFunctionReturn(0);
3225efd51863SBarry Smith }
3226efd51863SBarry Smith 
3227fa19ca70SBarry Smith /*@
3228fa19ca70SBarry Smith    SNESGetGridSequence - gets the number of steps of grid sequencing that SNES does
3229fa19ca70SBarry Smith 
3230fa19ca70SBarry Smith    Logically Collective on SNES
3231fa19ca70SBarry Smith 
3232fa19ca70SBarry Smith    Input Parameter:
3233fa19ca70SBarry Smith .  snes - the SNES context
3234fa19ca70SBarry Smith 
3235fa19ca70SBarry Smith    Output Parameter:
3236fa19ca70SBarry Smith .  steps - the number of refinements to do, defaults to 0
3237fa19ca70SBarry Smith 
3238fa19ca70SBarry Smith    Options Database Keys:
3239fa19ca70SBarry Smith .    -snes_grid_sequence <steps>
3240fa19ca70SBarry Smith 
3241fa19ca70SBarry Smith    Level: intermediate
3242fa19ca70SBarry Smith 
3243fa19ca70SBarry Smith    Notes:
3244fa19ca70SBarry Smith    Use SNESGetSolution() to extract the fine grid solution after grid sequencing.
3245fa19ca70SBarry Smith 
3246fa19ca70SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
3247fa19ca70SBarry Smith 
3248fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetGridSequence()
3249fa19ca70SBarry Smith 
3250fa19ca70SBarry Smith @*/
3251fa19ca70SBarry Smith PetscErrorCode  SNESGetGridSequence(SNES snes,PetscInt *steps)
3252fa19ca70SBarry Smith {
3253fa19ca70SBarry Smith   PetscFunctionBegin;
3254fa19ca70SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3255fa19ca70SBarry Smith   *steps = snes->gridsequence;
3256fa19ca70SBarry Smith   PetscFunctionReturn(0);
3257fa19ca70SBarry Smith }
3258fa19ca70SBarry Smith 
3259a8054027SBarry Smith /*@
3260a8054027SBarry Smith    SNESGetLagPreconditioner - Indicates how often the preconditioner is rebuilt
3261a8054027SBarry Smith 
32623f9fe445SBarry Smith    Not Collective
3263a8054027SBarry Smith 
3264a8054027SBarry Smith    Input Parameter:
3265a8054027SBarry Smith .  snes - the SNES context
3266a8054027SBarry Smith 
3267a8054027SBarry Smith    Output Parameter:
3268a8054027SBarry 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
32693b4f5425SBarry Smith          the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that
3270a8054027SBarry Smith 
3271a8054027SBarry Smith    Options Database Keys:
3272a8054027SBarry Smith .    -snes_lag_preconditioner <lag>
3273a8054027SBarry Smith 
3274a8054027SBarry Smith    Notes:
3275a8054027SBarry Smith    The default is 1
3276a8054027SBarry Smith    The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3277a8054027SBarry Smith 
3278a8054027SBarry Smith    Level: intermediate
3279a8054027SBarry Smith 
3280a8054027SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
3281a8054027SBarry Smith 
3282a8054027SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagPreconditioner()
3283a8054027SBarry Smith 
3284a8054027SBarry Smith @*/
32857087cfbeSBarry Smith PetscErrorCode  SNESGetLagPreconditioner(SNES snes,PetscInt *lag)
3286a8054027SBarry Smith {
3287a8054027SBarry Smith   PetscFunctionBegin;
32880700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3289a8054027SBarry Smith   *lag = snes->lagpreconditioner;
3290a8054027SBarry Smith   PetscFunctionReturn(0);
3291a8054027SBarry Smith }
3292a8054027SBarry Smith 
3293e35cf81dSBarry Smith /*@
3294e35cf81dSBarry Smith    SNESSetLagJacobian - Determines when the Jacobian is rebuilt in the nonlinear solve. See SNESSetLagPreconditioner() for determining how
3295e35cf81dSBarry Smith      often the preconditioner is rebuilt.
3296e35cf81dSBarry Smith 
32973f9fe445SBarry Smith    Logically Collective on SNES
3298e35cf81dSBarry Smith 
3299e35cf81dSBarry Smith    Input Parameters:
3300e35cf81dSBarry Smith +  snes - the SNES context
3301e35cf81dSBarry 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
3302fe3ffe1eSBarry Smith          the Jacobian is built etc. -2 means rebuild at next chance but then never again
3303e35cf81dSBarry Smith 
3304e35cf81dSBarry Smith    Options Database Keys:
3305e35cf81dSBarry Smith .    -snes_lag_jacobian <lag>
3306e35cf81dSBarry Smith 
3307e35cf81dSBarry Smith    Notes:
3308e35cf81dSBarry Smith    The default is 1
3309e35cf81dSBarry Smith    The Jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3310fe3ffe1eSBarry 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
3311fe3ffe1eSBarry Smith    at the next Newton step but never again (unless it is reset to another value)
3312e35cf81dSBarry Smith 
3313e35cf81dSBarry Smith    Level: intermediate
3314e35cf81dSBarry Smith 
3315e35cf81dSBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
3316e35cf81dSBarry Smith 
3317e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagPreconditioner(), SNESGetLagJacobian()
3318e35cf81dSBarry Smith 
3319e35cf81dSBarry Smith @*/
33207087cfbeSBarry Smith PetscErrorCode  SNESSetLagJacobian(SNES snes,PetscInt lag)
3321e35cf81dSBarry Smith {
3322e35cf81dSBarry Smith   PetscFunctionBegin;
33230700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3324e32f2f54SBarry Smith   if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater");
3325e32f2f54SBarry Smith   if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0");
3326c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,lag,2);
3327e35cf81dSBarry Smith   snes->lagjacobian = lag;
3328e35cf81dSBarry Smith   PetscFunctionReturn(0);
3329e35cf81dSBarry Smith }
3330e35cf81dSBarry Smith 
3331e35cf81dSBarry Smith /*@
3332e35cf81dSBarry Smith    SNESGetLagJacobian - Indicates how often the Jacobian is rebuilt. See SNESGetLagPreconditioner() to determine when the preconditioner is rebuilt
3333e35cf81dSBarry Smith 
33343f9fe445SBarry Smith    Not Collective
3335e35cf81dSBarry Smith 
3336e35cf81dSBarry Smith    Input Parameter:
3337e35cf81dSBarry Smith .  snes - the SNES context
3338e35cf81dSBarry Smith 
3339e35cf81dSBarry Smith    Output Parameter:
3340e35cf81dSBarry 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
3341e35cf81dSBarry Smith          the Jacobian is built etc.
3342e35cf81dSBarry Smith 
3343e35cf81dSBarry Smith    Options Database Keys:
3344e35cf81dSBarry Smith .    -snes_lag_jacobian <lag>
3345e35cf81dSBarry Smith 
3346e35cf81dSBarry Smith    Notes:
3347e35cf81dSBarry Smith    The default is 1
3348e35cf81dSBarry Smith    The jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1
3349e35cf81dSBarry Smith 
3350e35cf81dSBarry Smith    Level: intermediate
3351e35cf81dSBarry Smith 
3352e35cf81dSBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
3353e35cf81dSBarry Smith 
3354e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagJacobian(), SNESSetLagPreconditioner(), SNESGetLagPreconditioner()
3355e35cf81dSBarry Smith 
3356e35cf81dSBarry Smith @*/
33577087cfbeSBarry Smith PetscErrorCode  SNESGetLagJacobian(SNES snes,PetscInt *lag)
3358e35cf81dSBarry Smith {
3359e35cf81dSBarry Smith   PetscFunctionBegin;
33600700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3361e35cf81dSBarry Smith   *lag = snes->lagjacobian;
3362e35cf81dSBarry Smith   PetscFunctionReturn(0);
3363e35cf81dSBarry Smith }
3364e35cf81dSBarry Smith 
336537ec4e1aSPeter Brune /*@
336637ec4e1aSPeter Brune    SNESSetLagJacobianPersists - Set whether or not the Jacobian lagging persists through multiple solves
336737ec4e1aSPeter Brune 
336837ec4e1aSPeter Brune    Logically collective on SNES
336937ec4e1aSPeter Brune 
337037ec4e1aSPeter Brune    Input Parameter:
337137ec4e1aSPeter Brune +  snes - the SNES context
33729d7e2deaSPeter Brune -   flg - jacobian lagging persists if true
337337ec4e1aSPeter Brune 
337437ec4e1aSPeter Brune    Options Database Keys:
337537ec4e1aSPeter Brune .    -snes_lag_jacobian_persists <flg>
337637ec4e1aSPeter Brune 
337795452b02SPatrick Sanan    Notes:
337895452b02SPatrick Sanan     This is useful both for nonlinear preconditioning, where it's appropriate to have the Jacobian be stale by
337937ec4e1aSPeter Brune    several solves, and for implicit time-stepping, where Jacobian lagging in the inner nonlinear solve over several
338037ec4e1aSPeter Brune    timesteps may present huge efficiency gains.
338137ec4e1aSPeter Brune 
338237ec4e1aSPeter Brune    Level: developer
338337ec4e1aSPeter Brune 
3384be95d8f1SBarry Smith .keywords: SNES, nonlinear, lag
338537ec4e1aSPeter Brune 
3386be95d8f1SBarry Smith .seealso: SNESSetLagPreconditionerPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC()
338737ec4e1aSPeter Brune 
338837ec4e1aSPeter Brune @*/
338937ec4e1aSPeter Brune PetscErrorCode  SNESSetLagJacobianPersists(SNES snes,PetscBool flg)
339037ec4e1aSPeter Brune {
339137ec4e1aSPeter Brune   PetscFunctionBegin;
339237ec4e1aSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
339337ec4e1aSPeter Brune   PetscValidLogicalCollectiveBool(snes,flg,2);
339437ec4e1aSPeter Brune   snes->lagjac_persist = flg;
339537ec4e1aSPeter Brune   PetscFunctionReturn(0);
339637ec4e1aSPeter Brune }
339737ec4e1aSPeter Brune 
339837ec4e1aSPeter Brune /*@
339937ec4e1aSPeter Brune    SNESSetLagPreconditionerPersists - Set whether or not the preconditioner lagging persists through multiple solves
340037ec4e1aSPeter Brune 
340137ec4e1aSPeter Brune    Logically Collective on SNES
340237ec4e1aSPeter Brune 
340337ec4e1aSPeter Brune    Input Parameter:
340437ec4e1aSPeter Brune +  snes - the SNES context
34059d7e2deaSPeter Brune -   flg - preconditioner lagging persists if true
340637ec4e1aSPeter Brune 
340737ec4e1aSPeter Brune    Options Database Keys:
340837ec4e1aSPeter Brune .    -snes_lag_jacobian_persists <flg>
340937ec4e1aSPeter Brune 
341095452b02SPatrick Sanan    Notes:
341195452b02SPatrick Sanan     This is useful both for nonlinear preconditioning, where it's appropriate to have the preconditioner be stale
341237ec4e1aSPeter Brune    by several solves, and for implicit time-stepping, where preconditioner lagging in the inner nonlinear solve over
341337ec4e1aSPeter Brune    several timesteps may present huge efficiency gains.
341437ec4e1aSPeter Brune 
341537ec4e1aSPeter Brune    Level: developer
341637ec4e1aSPeter Brune 
3417be95d8f1SBarry Smith .keywords: SNES, nonlinear, lag
341837ec4e1aSPeter Brune 
3419be95d8f1SBarry Smith .seealso: SNESSetLagJacobianPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC()
342037ec4e1aSPeter Brune 
342137ec4e1aSPeter Brune @*/
342237ec4e1aSPeter Brune PetscErrorCode  SNESSetLagPreconditionerPersists(SNES snes,PetscBool flg)
342337ec4e1aSPeter Brune {
342437ec4e1aSPeter Brune   PetscFunctionBegin;
342537ec4e1aSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
342637ec4e1aSPeter Brune   PetscValidLogicalCollectiveBool(snes,flg,2);
342737ec4e1aSPeter Brune   snes->lagpre_persist = flg;
342837ec4e1aSPeter Brune   PetscFunctionReturn(0);
342937ec4e1aSPeter Brune }
343037ec4e1aSPeter Brune 
34319b94acceSBarry Smith /*@
3432be5caee7SBarry Smith    SNESSetForceIteration - force SNESSolve() to take at least one iteration regardless of the initial residual norm
3433be5caee7SBarry Smith 
3434be5caee7SBarry Smith    Logically Collective on SNES
3435be5caee7SBarry Smith 
3436be5caee7SBarry Smith    Input Parameters:
3437be5caee7SBarry Smith +  snes - the SNES context
3438be5caee7SBarry Smith -  force - PETSC_TRUE require at least one iteration
3439be5caee7SBarry Smith 
3440be5caee7SBarry Smith    Options Database Keys:
3441be5caee7SBarry Smith .    -snes_force_iteration <force> - Sets forcing an iteration
3442be5caee7SBarry Smith 
3443be5caee7SBarry Smith    Notes:
3444be5caee7SBarry Smith    This is used sometimes with TS to prevent TS from detecting a false steady state solution
3445be5caee7SBarry Smith 
3446be5caee7SBarry Smith    Level: intermediate
3447be5caee7SBarry Smith 
3448be5caee7SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances
3449be5caee7SBarry Smith 
3450be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance()
3451be5caee7SBarry Smith @*/
3452be5caee7SBarry Smith PetscErrorCode  SNESSetForceIteration(SNES snes,PetscBool force)
3453be5caee7SBarry Smith {
3454be5caee7SBarry Smith   PetscFunctionBegin;
3455be5caee7SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3456be5caee7SBarry Smith   snes->forceiteration = force;
3457be5caee7SBarry Smith   PetscFunctionReturn(0);
3458be5caee7SBarry Smith }
3459be5caee7SBarry Smith 
346085216dc7SFande Kong /*@
346185216dc7SFande Kong    SNESGetForceIteration - Whether or not to force SNESSolve() take at least one iteration regardless of the initial residual norm
346285216dc7SFande Kong 
346385216dc7SFande Kong    Logically Collective on SNES
346485216dc7SFande Kong 
346585216dc7SFande Kong    Input Parameters:
346685216dc7SFande Kong .  snes - the SNES context
346785216dc7SFande Kong 
346885216dc7SFande Kong    Output Parameter:
346985216dc7SFande Kong .  force - PETSC_TRUE requires at least one iteration.
347085216dc7SFande Kong 
347185216dc7SFande Kong .keywords: SNES, nonlinear, set, convergence, tolerances
347285216dc7SFande Kong 
347306dd6b0eSSatish Balay    Level: intermediate
347406dd6b0eSSatish Balay 
347585216dc7SFande Kong .seealso: SNESSetForceIteration(), SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance()
347685216dc7SFande Kong @*/
347785216dc7SFande Kong PetscErrorCode  SNESGetForceIteration(SNES snes,PetscBool *force)
347885216dc7SFande Kong {
347985216dc7SFande Kong   PetscFunctionBegin;
348085216dc7SFande Kong   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
348185216dc7SFande Kong   *force = snes->forceiteration;
348285216dc7SFande Kong   PetscFunctionReturn(0);
348385216dc7SFande Kong }
3484be5caee7SBarry Smith 
3485be5caee7SBarry Smith /*@
3486d7a720efSLois Curfman McInnes    SNESSetTolerances - Sets various parameters used in convergence tests.
34879b94acceSBarry Smith 
34883f9fe445SBarry Smith    Logically Collective on SNES
3489c7afd0dbSLois Curfman McInnes 
34909b94acceSBarry Smith    Input Parameters:
3491c7afd0dbSLois Curfman McInnes +  snes - the SNES context
349270441072SBarry Smith .  abstol - absolute convergence tolerance
349333174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
34945358d0d4SBarry Smith .  stol -  convergence tolerance in terms of the norm of the change in the solution between steps,  || delta x || < stol*|| x ||
349533174efeSLois Curfman McInnes .  maxit - maximum number of iterations
3496e71169deSBarry Smith -  maxf - maximum number of function evaluations (-1 indicates no limit)
3497fee21e36SBarry Smith 
349833174efeSLois Curfman McInnes    Options Database Keys:
349970441072SBarry Smith +    -snes_atol <abstol> - Sets abstol
3500c7afd0dbSLois Curfman McInnes .    -snes_rtol <rtol> - Sets rtol
3501c7afd0dbSLois Curfman McInnes .    -snes_stol <stol> - Sets stol
3502c7afd0dbSLois Curfman McInnes .    -snes_max_it <maxit> - Sets maxit
3503c7afd0dbSLois Curfman McInnes -    -snes_max_funcs <maxf> - Sets maxf
35049b94acceSBarry Smith 
3505d7a720efSLois Curfman McInnes    Notes:
35069b94acceSBarry Smith    The default maximum number of iterations is 50.
35079b94acceSBarry Smith    The default maximum number of function evaluations is 1000.
35089b94acceSBarry Smith 
350936851e7fSLois Curfman McInnes    Level: intermediate
351036851e7fSLois Curfman McInnes 
351133174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances
35129b94acceSBarry Smith 
3513be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance(), SNESSetForceIteration()
35149b94acceSBarry Smith @*/
35157087cfbeSBarry Smith PetscErrorCode  SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf)
35169b94acceSBarry Smith {
35173a40ed3dSBarry Smith   PetscFunctionBegin;
35180700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3519c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,abstol,2);
3520c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol,3);
3521c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,stol,4);
3522c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxit,5);
3523c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,maxf,6);
3524c5eb9154SBarry Smith 
3525ab54825eSJed Brown   if (abstol != PETSC_DEFAULT) {
352657622a8eSBarry Smith     if (abstol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Absolute tolerance %g must be non-negative",(double)abstol);
3527ab54825eSJed Brown     snes->abstol = abstol;
3528ab54825eSJed Brown   }
3529ab54825eSJed Brown   if (rtol != PETSC_DEFAULT) {
353057622a8eSBarry 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);
3531ab54825eSJed Brown     snes->rtol = rtol;
3532ab54825eSJed Brown   }
3533ab54825eSJed Brown   if (stol != PETSC_DEFAULT) {
353457622a8eSBarry Smith     if (stol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Step tolerance %g must be non-negative",(double)stol);
3535c60f73f4SPeter Brune     snes->stol = stol;
3536ab54825eSJed Brown   }
3537ab54825eSJed Brown   if (maxit != PETSC_DEFAULT) {
3538ce94432eSBarry Smith     if (maxit < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of iterations %D must be non-negative",maxit);
3539ab54825eSJed Brown     snes->max_its = maxit;
3540ab54825eSJed Brown   }
3541ab54825eSJed Brown   if (maxf != PETSC_DEFAULT) {
3542e71169deSBarry Smith     if (maxf < -1) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of function evaluations %D must be -1 or nonnegative",maxf);
3543ab54825eSJed Brown     snes->max_funcs = maxf;
3544ab54825eSJed Brown   }
354588976e71SPeter Brune   snes->tolerancesset = PETSC_TRUE;
35463a40ed3dSBarry Smith   PetscFunctionReturn(0);
35479b94acceSBarry Smith }
35489b94acceSBarry Smith 
3549e4d06f11SPatrick Farrell /*@
3550e4d06f11SPatrick Farrell    SNESSetDivergenceTolerance - Sets the divergence tolerance used for the SNES divergence test.
3551e4d06f11SPatrick Farrell 
3552e4d06f11SPatrick Farrell    Logically Collective on SNES
3553e4d06f11SPatrick Farrell 
3554e4d06f11SPatrick Farrell    Input Parameters:
3555e4d06f11SPatrick Farrell +  snes - the SNES context
3556e4d06f11SPatrick Farrell -  divtol - the divergence tolerance. Use -1 to deactivate the test.
3557e4d06f11SPatrick Farrell 
3558e4d06f11SPatrick Farrell    Options Database Keys:
3559e4d06f11SPatrick Farrell +    -snes_divergence_tolerance <divtol> - Sets divtol
3560e4d06f11SPatrick Farrell 
3561e4d06f11SPatrick Farrell    Notes:
3562e4d06f11SPatrick Farrell    The default divergence tolerance is 1e4.
3563e4d06f11SPatrick Farrell 
3564e4d06f11SPatrick Farrell    Level: intermediate
3565e4d06f11SPatrick Farrell 
3566e4d06f11SPatrick Farrell .keywords: SNES, nonlinear, set, divergence, tolerance
3567e4d06f11SPatrick Farrell 
3568e4d06f11SPatrick Farrell .seealso: SNESSetTolerances(), SNESGetDivergenceTolerance
3569e4d06f11SPatrick Farrell @*/
3570e4d06f11SPatrick Farrell PetscErrorCode  SNESSetDivergenceTolerance(SNES snes,PetscReal divtol)
3571e4d06f11SPatrick Farrell {
3572e4d06f11SPatrick Farrell   PetscFunctionBegin;
3573e4d06f11SPatrick Farrell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3574e4d06f11SPatrick Farrell   PetscValidLogicalCollectiveReal(snes,divtol,2);
3575e4d06f11SPatrick Farrell 
3576e4d06f11SPatrick Farrell   if (divtol != PETSC_DEFAULT) {
3577e4d06f11SPatrick Farrell     snes->divtol = divtol;
3578e4d06f11SPatrick Farrell   }
3579e4d06f11SPatrick Farrell   else {
3580e4d06f11SPatrick Farrell     snes->divtol = 1.0e4;
3581e4d06f11SPatrick Farrell   }
3582e4d06f11SPatrick Farrell   PetscFunctionReturn(0);
3583e4d06f11SPatrick Farrell }
3584e4d06f11SPatrick Farrell 
35859b94acceSBarry Smith /*@
358633174efeSLois Curfman McInnes    SNESGetTolerances - Gets various parameters used in convergence tests.
358733174efeSLois Curfman McInnes 
3588c7afd0dbSLois Curfman McInnes    Not Collective
3589c7afd0dbSLois Curfman McInnes 
359033174efeSLois Curfman McInnes    Input Parameters:
3591c7afd0dbSLois Curfman McInnes +  snes - the SNES context
359285385478SLisandro Dalcin .  atol - absolute convergence tolerance
359333174efeSLois Curfman McInnes .  rtol - relative convergence tolerance
359433174efeSLois Curfman McInnes .  stol -  convergence tolerance in terms of the norm
359533174efeSLois Curfman McInnes            of the change in the solution between steps
359633174efeSLois Curfman McInnes .  maxit - maximum number of iterations
3597c7afd0dbSLois Curfman McInnes -  maxf - maximum number of function evaluations
3598fee21e36SBarry Smith 
359933174efeSLois Curfman McInnes    Notes:
36000298fd71SBarry Smith    The user can specify NULL for any parameter that is not needed.
360133174efeSLois Curfman McInnes 
360236851e7fSLois Curfman McInnes    Level: intermediate
360336851e7fSLois Curfman McInnes 
360433174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances
360533174efeSLois Curfman McInnes 
360633174efeSLois Curfman McInnes .seealso: SNESSetTolerances()
360733174efeSLois Curfman McInnes @*/
36087087cfbeSBarry Smith PetscErrorCode  SNESGetTolerances(SNES snes,PetscReal *atol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf)
360933174efeSLois Curfman McInnes {
36103a40ed3dSBarry Smith   PetscFunctionBegin;
36110700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
361285385478SLisandro Dalcin   if (atol)  *atol  = snes->abstol;
361333174efeSLois Curfman McInnes   if (rtol)  *rtol  = snes->rtol;
3614c60f73f4SPeter Brune   if (stol)  *stol  = snes->stol;
361533174efeSLois Curfman McInnes   if (maxit) *maxit = snes->max_its;
361633174efeSLois Curfman McInnes   if (maxf)  *maxf  = snes->max_funcs;
36173a40ed3dSBarry Smith   PetscFunctionReturn(0);
361833174efeSLois Curfman McInnes }
361933174efeSLois Curfman McInnes 
3620e4d06f11SPatrick Farrell /*@
3621e4d06f11SPatrick Farrell    SNESGetDivergenceTolerance - Gets divergence tolerance used in divergence test.
3622e4d06f11SPatrick Farrell 
3623e4d06f11SPatrick Farrell    Not Collective
3624e4d06f11SPatrick Farrell 
3625e4d06f11SPatrick Farrell    Input Parameters:
3626e4d06f11SPatrick Farrell +  snes - the SNES context
3627e4d06f11SPatrick Farrell -  divtol - divergence tolerance
3628e4d06f11SPatrick Farrell 
3629e4d06f11SPatrick Farrell    Level: intermediate
3630e4d06f11SPatrick Farrell 
3631e4d06f11SPatrick Farrell .keywords: SNES, nonlinear, get, divergence, tolerance
3632e4d06f11SPatrick Farrell 
3633e4d06f11SPatrick Farrell .seealso: SNESSetDivergenceTolerance()
3634e4d06f11SPatrick Farrell @*/
3635e4d06f11SPatrick Farrell PetscErrorCode  SNESGetDivergenceTolerance(SNES snes,PetscReal *divtol)
3636e4d06f11SPatrick Farrell {
3637e4d06f11SPatrick Farrell   PetscFunctionBegin;
3638e4d06f11SPatrick Farrell   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3639e4d06f11SPatrick Farrell   if (divtol) *divtol = snes->divtol;
3640e4d06f11SPatrick Farrell   PetscFunctionReturn(0);
3641e4d06f11SPatrick Farrell }
3642e4d06f11SPatrick Farrell 
364333174efeSLois Curfman McInnes /*@
36449b94acceSBarry Smith    SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance.
36459b94acceSBarry Smith 
36463f9fe445SBarry Smith    Logically Collective on SNES
3647fee21e36SBarry Smith 
3648c7afd0dbSLois Curfman McInnes    Input Parameters:
3649c7afd0dbSLois Curfman McInnes +  snes - the SNES context
3650c7afd0dbSLois Curfman McInnes -  tol - tolerance
3651c7afd0dbSLois Curfman McInnes 
36529b94acceSBarry Smith    Options Database Key:
3653c7afd0dbSLois Curfman McInnes .  -snes_trtol <tol> - Sets tol
36549b94acceSBarry Smith 
365536851e7fSLois Curfman McInnes    Level: intermediate
365636851e7fSLois Curfman McInnes 
36579b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance
36589b94acceSBarry Smith 
36592492ecdbSBarry Smith .seealso: SNESSetTolerances()
36609b94acceSBarry Smith @*/
36617087cfbeSBarry Smith PetscErrorCode  SNESSetTrustRegionTolerance(SNES snes,PetscReal tol)
36629b94acceSBarry Smith {
36633a40ed3dSBarry Smith   PetscFunctionBegin;
36640700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3665c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,tol,2);
36669b94acceSBarry Smith   snes->deltatol = tol;
36673a40ed3dSBarry Smith   PetscFunctionReturn(0);
36689b94acceSBarry Smith }
36699b94acceSBarry Smith 
3670df9fa365SBarry Smith /*
3671df9fa365SBarry Smith    Duplicate the lg monitors for SNES from KSP; for some reason with
3672df9fa365SBarry Smith    dynamic libraries things don't work under Sun4 if we just use
3673df9fa365SBarry Smith    macros instead of functions
3674df9fa365SBarry Smith */
3675d96771aaSLisandro Dalcin PetscErrorCode  SNESMonitorLGResidualNorm(SNES snes,PetscInt it,PetscReal norm,void *ctx)
3676ce1608b8SBarry Smith {
3677dfbe8321SBarry Smith   PetscErrorCode ierr;
3678ce1608b8SBarry Smith 
3679ce1608b8SBarry Smith   PetscFunctionBegin;
36800700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3681d96771aaSLisandro Dalcin   ierr = KSPMonitorLGResidualNorm((KSP)snes,it,norm,ctx);CHKERRQ(ierr);
3682ce1608b8SBarry Smith   PetscFunctionReturn(0);
3683ce1608b8SBarry Smith }
3684ce1608b8SBarry Smith 
3685d96771aaSLisandro Dalcin PetscErrorCode  SNESMonitorLGCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *lgctx)
3686df9fa365SBarry Smith {
3687dfbe8321SBarry Smith   PetscErrorCode ierr;
3688df9fa365SBarry Smith 
3689df9fa365SBarry Smith   PetscFunctionBegin;
3690d96771aaSLisandro Dalcin   ierr = KSPMonitorLGResidualNormCreate(comm,host,label,x,y,m,n,lgctx);CHKERRQ(ierr);
3691df9fa365SBarry Smith   PetscFunctionReturn(0);
3692df9fa365SBarry Smith }
3693df9fa365SBarry Smith 
36946ba87a44SLisandro Dalcin PETSC_INTERN PetscErrorCode  SNESMonitorRange_Private(SNES,PetscInt,PetscReal*);
36956ba87a44SLisandro Dalcin 
36967087cfbeSBarry Smith PetscErrorCode  SNESMonitorLGRange(SNES snes,PetscInt n,PetscReal rnorm,void *monctx)
3697b271bb04SBarry Smith {
3698b271bb04SBarry Smith   PetscDrawLG      lg;
3699b271bb04SBarry Smith   PetscErrorCode   ierr;
3700b271bb04SBarry Smith   PetscReal        x,y,per;
3701b271bb04SBarry Smith   PetscViewer      v = (PetscViewer)monctx;
3702b271bb04SBarry Smith   static PetscReal prev; /* should be in the context */
3703b271bb04SBarry Smith   PetscDraw        draw;
3704b271bb04SBarry Smith 
3705459f5d12SBarry Smith   PetscFunctionBegin;
37064d4332d5SBarry Smith   PetscValidHeaderSpecific(v,PETSC_VIEWER_CLASSID,4);
3707b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,0,&lg);CHKERRQ(ierr);
3708b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3709b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3710b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"Residual norm");CHKERRQ(ierr);
3711b271bb04SBarry Smith   x    = (PetscReal)n;
371277b4d14cSPeter Brune   if (rnorm > 0.0) y = PetscLog10Real(rnorm);
371394c9c6d3SKarl Rupp   else y = -15.0;
3714b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
37156934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3716b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
37176934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3718b271bb04SBarry Smith   }
3719b271bb04SBarry Smith 
3720b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,1,&lg);CHKERRQ(ierr);
3721b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3722b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3723b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"% elemts > .2*max elemt");CHKERRQ(ierr);
3724b271bb04SBarry Smith   ierr =  SNESMonitorRange_Private(snes,n,&per);CHKERRQ(ierr);
3725b271bb04SBarry Smith   x    = (PetscReal)n;
3726b271bb04SBarry Smith   y    = 100.0*per;
3727b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
37286934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3729b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
37306934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3731b271bb04SBarry Smith   }
3732b271bb04SBarry Smith 
3733b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,2,&lg);CHKERRQ(ierr);
3734b271bb04SBarry Smith   if (!n) {prev = rnorm;ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3735b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3736b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm");CHKERRQ(ierr);
3737b271bb04SBarry Smith   x    = (PetscReal)n;
3738b271bb04SBarry Smith   y    = (prev - rnorm)/prev;
3739b271bb04SBarry Smith   ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
37406934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3741b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
37426934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3743b271bb04SBarry Smith   }
3744b271bb04SBarry Smith 
3745b271bb04SBarry Smith   ierr = PetscViewerDrawGetDrawLG(v,3,&lg);CHKERRQ(ierr);
3746b271bb04SBarry Smith   if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);}
3747b271bb04SBarry Smith   ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr);
3748b271bb04SBarry Smith   ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr);
3749b271bb04SBarry Smith   x    = (PetscReal)n;
3750b271bb04SBarry Smith   y    = (prev - rnorm)/(prev*per);
3751b271bb04SBarry Smith   if (n > 2) { /*skip initial crazy value */
3752b271bb04SBarry Smith     ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr);
3753b271bb04SBarry Smith   }
37546934998bSLisandro Dalcin   if (n < 20 || !(n % 5) || snes->reason) {
3755b271bb04SBarry Smith     ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr);
37566934998bSLisandro Dalcin     ierr = PetscDrawLGSave(lg);CHKERRQ(ierr);
3757b271bb04SBarry Smith   }
3758b271bb04SBarry Smith   prev = rnorm;
3759b271bb04SBarry Smith   PetscFunctionReturn(0);
3760b271bb04SBarry Smith }
3761b271bb04SBarry Smith 
3762228d79bcSJed Brown /*@
3763228d79bcSJed Brown    SNESMonitor - runs the user provided monitor routines, if they exist
3764228d79bcSJed Brown 
3765228d79bcSJed Brown    Collective on SNES
3766228d79bcSJed Brown 
3767228d79bcSJed Brown    Input Parameters:
3768228d79bcSJed Brown +  snes - nonlinear solver context obtained from SNESCreate()
3769228d79bcSJed Brown .  iter - iteration number
3770228d79bcSJed Brown -  rnorm - relative norm of the residual
3771228d79bcSJed Brown 
3772228d79bcSJed Brown    Notes:
3773228d79bcSJed Brown    This routine is called by the SNES implementations.
3774228d79bcSJed Brown    It does not typically need to be called by the user.
3775228d79bcSJed Brown 
3776228d79bcSJed Brown    Level: developer
3777228d79bcSJed Brown 
3778228d79bcSJed Brown .seealso: SNESMonitorSet()
3779228d79bcSJed Brown @*/
37807a03ce2fSLisandro Dalcin PetscErrorCode  SNESMonitor(SNES snes,PetscInt iter,PetscReal rnorm)
37817a03ce2fSLisandro Dalcin {
37827a03ce2fSLisandro Dalcin   PetscErrorCode ierr;
37837a03ce2fSLisandro Dalcin   PetscInt       i,n = snes->numbermonitors;
37847a03ce2fSLisandro Dalcin 
37857a03ce2fSLisandro Dalcin   PetscFunctionBegin;
37865edff71fSBarry Smith   ierr = VecLockPush(snes->vec_sol);CHKERRQ(ierr);
37877a03ce2fSLisandro Dalcin   for (i=0; i<n; i++) {
37887a03ce2fSLisandro Dalcin     ierr = (*snes->monitor[i])(snes,iter,rnorm,snes->monitorcontext[i]);CHKERRQ(ierr);
37897a03ce2fSLisandro Dalcin   }
37905edff71fSBarry Smith   ierr = VecLockPop(snes->vec_sol);CHKERRQ(ierr);
37917a03ce2fSLisandro Dalcin   PetscFunctionReturn(0);
37927a03ce2fSLisandro Dalcin }
37937a03ce2fSLisandro Dalcin 
37949b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */
37959b94acceSBarry Smith 
3796bf388a1fSBarry Smith /*MC
3797bf388a1fSBarry Smith     SNESMonitorFunction - functional form passed to SNESMonitorSet() to monitor convergence of nonlinear solver
3798bf388a1fSBarry Smith 
3799bf388a1fSBarry Smith      Synopsis:
3800aaa7dc30SBarry Smith      #include <petscsnes.h>
3801bf388a1fSBarry Smith $    PetscErrorCode SNESMonitorFunction(SNES snes,PetscInt its, PetscReal norm,void *mctx)
3802bf388a1fSBarry Smith 
3803bf388a1fSBarry Smith +    snes - the SNES context
3804bf388a1fSBarry Smith .    its - iteration number
3805bf388a1fSBarry Smith .    norm - 2-norm function value (may be estimated)
3806bf388a1fSBarry Smith -    mctx - [optional] monitoring context
3807bf388a1fSBarry Smith 
3808878cb397SSatish Balay    Level: advanced
3809878cb397SSatish Balay 
3810bf388a1fSBarry Smith .seealso:   SNESMonitorSet(), SNESMonitorGet()
3811bf388a1fSBarry Smith M*/
3812bf388a1fSBarry Smith 
38139b94acceSBarry Smith /*@C
3814a6570f20SBarry Smith    SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every
38159b94acceSBarry Smith    iteration of the nonlinear solver to display the iteration's
38169b94acceSBarry Smith    progress.
38179b94acceSBarry Smith 
38183f9fe445SBarry Smith    Logically Collective on SNES
3819fee21e36SBarry Smith 
3820c7afd0dbSLois Curfman McInnes    Input Parameters:
3821c7afd0dbSLois Curfman McInnes +  snes - the SNES context
38226e4dcb14SBarry Smith .  f - the monitor function, see SNESMonitorFunction for the calling sequence
3823b8a78c4aSBarry Smith .  mctx - [optional] user-defined context for private data for the
38240298fd71SBarry Smith           monitor routine (use NULL if no context is desired)
3825b3006f0bSLois Curfman McInnes -  monitordestroy - [optional] routine that frees monitor context
38260298fd71SBarry Smith           (may be NULL)
38279b94acceSBarry Smith 
38289665c990SLois Curfman McInnes    Options Database Keys:
3829a6570f20SBarry Smith +    -snes_monitor        - sets SNESMonitorDefault()
38304619e776SBarry Smith .    -snes_monitor_lg_residualnorm    - sets line graph monitor,
3831a6570f20SBarry Smith                             uses SNESMonitorLGCreate()
3832cca6129bSJed Brown -    -snes_monitor_cancel - cancels all monitors that have
3833c7afd0dbSLois Curfman McInnes                             been hardwired into a code by
3834a6570f20SBarry Smith                             calls to SNESMonitorSet(), but
3835c7afd0dbSLois Curfman McInnes                             does not cancel those set via
3836c7afd0dbSLois Curfman McInnes                             the options database.
38379665c990SLois Curfman McInnes 
3838639f9d9dSBarry Smith    Notes:
38396bc08f3fSLois Curfman McInnes    Several different monitoring routines may be set by calling
3840a6570f20SBarry Smith    SNESMonitorSet() multiple times; all will be called in the
38416bc08f3fSLois Curfman McInnes    order in which they were set.
3842639f9d9dSBarry Smith 
384395452b02SPatrick Sanan    Fortran Notes:
384495452b02SPatrick Sanan     Only a single monitor function can be set for each SNES object
3845025f1a04SBarry Smith 
384636851e7fSLois Curfman McInnes    Level: intermediate
384736851e7fSLois Curfman McInnes 
38489b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor
38499b94acceSBarry Smith 
3850bf388a1fSBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorCancel(), SNESMonitorFunction
38519b94acceSBarry Smith @*/
38526e4dcb14SBarry Smith PetscErrorCode  SNESMonitorSet(SNES snes,PetscErrorCode (*f)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void**))
38539b94acceSBarry Smith {
3854b90d0a6eSBarry Smith   PetscInt       i;
3855649052a6SBarry Smith   PetscErrorCode ierr;
385678064530SBarry Smith   PetscBool      identical;
3857b90d0a6eSBarry Smith 
38583a40ed3dSBarry Smith   PetscFunctionBegin;
38590700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3860b90d0a6eSBarry Smith   for (i=0; i<snes->numbermonitors;i++) {
386178064530SBarry Smith     ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,mctx,monitordestroy,(PetscErrorCode (*)(void))snes->monitor[i],snes->monitorcontext[i],snes->monitordestroy[i],&identical);CHKERRQ(ierr);
386278064530SBarry Smith     if (identical) PetscFunctionReturn(0);
3863649052a6SBarry Smith   }
386478064530SBarry Smith   if (snes->numbermonitors >= MAXSNESMONITORS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set");
38656e4dcb14SBarry Smith   snes->monitor[snes->numbermonitors]          = f;
3866b8a78c4aSBarry Smith   snes->monitordestroy[snes->numbermonitors]   = monitordestroy;
3867639f9d9dSBarry Smith   snes->monitorcontext[snes->numbermonitors++] = (void*)mctx;
38683a40ed3dSBarry Smith   PetscFunctionReturn(0);
38699b94acceSBarry Smith }
38709b94acceSBarry Smith 
3871a278d85bSSatish Balay /*@
3872a6570f20SBarry Smith    SNESMonitorCancel - Clears all the monitor functions for a SNES object.
38735cd90555SBarry Smith 
38743f9fe445SBarry Smith    Logically Collective on SNES
3875c7afd0dbSLois Curfman McInnes 
38765cd90555SBarry Smith    Input Parameters:
38775cd90555SBarry Smith .  snes - the SNES context
38785cd90555SBarry Smith 
38791a480d89SAdministrator    Options Database Key:
3880a6570f20SBarry Smith .  -snes_monitor_cancel - cancels all monitors that have been hardwired
3881a6570f20SBarry Smith     into a code by calls to SNESMonitorSet(), but does not cancel those
3882c7afd0dbSLois Curfman McInnes     set via the options database
38835cd90555SBarry Smith 
38845cd90555SBarry Smith    Notes:
38855cd90555SBarry Smith    There is no way to clear one specific monitor from a SNES object.
38865cd90555SBarry Smith 
388736851e7fSLois Curfman McInnes    Level: intermediate
388836851e7fSLois Curfman McInnes 
38895cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor
38905cd90555SBarry Smith 
3891a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorSet()
38925cd90555SBarry Smith @*/
38937087cfbeSBarry Smith PetscErrorCode  SNESMonitorCancel(SNES snes)
38945cd90555SBarry Smith {
3895d952e501SBarry Smith   PetscErrorCode ierr;
3896d952e501SBarry Smith   PetscInt       i;
3897d952e501SBarry Smith 
38985cd90555SBarry Smith   PetscFunctionBegin;
38990700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3900d952e501SBarry Smith   for (i=0; i<snes->numbermonitors; i++) {
3901d952e501SBarry Smith     if (snes->monitordestroy[i]) {
39023c4aec1bSBarry Smith       ierr = (*snes->monitordestroy[i])(&snes->monitorcontext[i]);CHKERRQ(ierr);
3903d952e501SBarry Smith     }
3904d952e501SBarry Smith   }
39055cd90555SBarry Smith   snes->numbermonitors = 0;
39065cd90555SBarry Smith   PetscFunctionReturn(0);
39075cd90555SBarry Smith }
39085cd90555SBarry Smith 
3909bf388a1fSBarry Smith /*MC
3910bf388a1fSBarry Smith     SNESConvergenceTestFunction - functional form used for testing of convergence of nonlinear solver
3911bf388a1fSBarry Smith 
3912bf388a1fSBarry Smith      Synopsis:
3913aaa7dc30SBarry Smith      #include <petscsnes.h>
3914bf388a1fSBarry Smith $     PetscErrorCode SNESConvergenceTest(SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx)
3915bf388a1fSBarry Smith 
3916bf388a1fSBarry Smith +    snes - the SNES context
3917bf388a1fSBarry Smith .    it - current iteration (0 is the first and is before any Newton step)
3918bf388a1fSBarry Smith .    cctx - [optional] convergence context
3919bf388a1fSBarry Smith .    reason - reason for convergence/divergence
3920bf388a1fSBarry Smith .    xnorm - 2-norm of current iterate
3921bf388a1fSBarry Smith .    gnorm - 2-norm of current step
3922bf388a1fSBarry Smith -    f - 2-norm of function
3923bf388a1fSBarry Smith 
3924878cb397SSatish Balay    Level: intermediate
3925bf388a1fSBarry Smith 
3926bf388a1fSBarry Smith .seealso:   SNESSetConvergenceTest(), SNESGetConvergenceTest()
3927bf388a1fSBarry Smith M*/
3928bf388a1fSBarry Smith 
39299b94acceSBarry Smith /*@C
39309b94acceSBarry Smith    SNESSetConvergenceTest - Sets the function that is to be used
39319b94acceSBarry Smith    to test for convergence of the nonlinear iterative solution.
39329b94acceSBarry Smith 
39333f9fe445SBarry Smith    Logically Collective on SNES
3934fee21e36SBarry Smith 
3935c7afd0dbSLois Curfman McInnes    Input Parameters:
3936c7afd0dbSLois Curfman McInnes +  snes - the SNES context
3937bf388a1fSBarry Smith .  SNESConvergenceTestFunction - routine to test for convergence
39380298fd71SBarry Smith .  cctx - [optional] context for private data for the convergence routine  (may be NULL)
3939cf90aa19SBarry Smith -  destroy - [optional] destructor for the context (may be NULL; PETSC_NULL_FUNCTION in Fortran)
39409b94acceSBarry Smith 
394136851e7fSLois Curfman McInnes    Level: advanced
394236851e7fSLois Curfman McInnes 
39439b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test
39449b94acceSBarry Smith 
3945e2a6519dSDmitry Karpeev .seealso: SNESConvergedDefault(), SNESConvergedSkip(), SNESConvergenceTestFunction
39469b94acceSBarry Smith @*/
3947bf388a1fSBarry Smith PetscErrorCode  SNESSetConvergenceTest(SNES snes,PetscErrorCode (*SNESConvergenceTestFunction)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx,PetscErrorCode (*destroy)(void*))
39489b94acceSBarry Smith {
39497f7931b9SBarry Smith   PetscErrorCode ierr;
39507f7931b9SBarry Smith 
39513a40ed3dSBarry Smith   PetscFunctionBegin;
39520700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
3953e2a6519dSDmitry Karpeev   if (!SNESConvergenceTestFunction) SNESConvergenceTestFunction = SNESConvergedSkip;
39547f7931b9SBarry Smith   if (snes->ops->convergeddestroy) {
39557f7931b9SBarry Smith     ierr = (*snes->ops->convergeddestroy)(snes->cnvP);CHKERRQ(ierr);
39567f7931b9SBarry Smith   }
3957bf388a1fSBarry Smith   snes->ops->converged        = SNESConvergenceTestFunction;
39587f7931b9SBarry Smith   snes->ops->convergeddestroy = destroy;
395985385478SLisandro Dalcin   snes->cnvP                  = cctx;
39603a40ed3dSBarry Smith   PetscFunctionReturn(0);
39619b94acceSBarry Smith }
39629b94acceSBarry Smith 
396352baeb72SSatish Balay /*@
3964184914b5SBarry Smith    SNESGetConvergedReason - Gets the reason the SNES iteration was stopped.
3965184914b5SBarry Smith 
3966184914b5SBarry Smith    Not Collective
3967184914b5SBarry Smith 
3968184914b5SBarry Smith    Input Parameter:
3969184914b5SBarry Smith .  snes - the SNES context
3970184914b5SBarry Smith 
3971184914b5SBarry Smith    Output Parameter:
39724d0a8057SBarry Smith .  reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the
3973184914b5SBarry Smith             manual pages for the individual convergence tests for complete lists
3974184914b5SBarry Smith 
39756a4d7782SBarry Smith    Options Database:
39766a4d7782SBarry Smith .   -snes_converged_reason - prints the reason to standard out
39776a4d7782SBarry Smith 
3978184914b5SBarry Smith    Level: intermediate
3979184914b5SBarry Smith 
398095452b02SPatrick Sanan    Notes:
398195452b02SPatrick Sanan     Should only be called after the call the SNESSolve() is complete, if it is called earlier it returns the value SNES__CONVERGED_ITERATING.
3982184914b5SBarry Smith 
3983184914b5SBarry Smith .keywords: SNES, nonlinear, set, convergence, test
3984184914b5SBarry Smith 
398533866048SMatthew G. Knepley .seealso: SNESSetConvergenceTest(), SNESSetConvergedReason(), SNESConvergedReason
3986184914b5SBarry Smith @*/
39877087cfbeSBarry Smith PetscErrorCode SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason)
3988184914b5SBarry Smith {
3989184914b5SBarry Smith   PetscFunctionBegin;
39900700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
39914482741eSBarry Smith   PetscValidPointer(reason,2);
3992184914b5SBarry Smith   *reason = snes->reason;
3993184914b5SBarry Smith   PetscFunctionReturn(0);
3994184914b5SBarry Smith }
3995184914b5SBarry Smith 
399633866048SMatthew G. Knepley /*@
399733866048SMatthew G. Knepley    SNESSetConvergedReason - Sets the reason the SNES iteration was stopped.
399833866048SMatthew G. Knepley 
399933866048SMatthew G. Knepley    Not Collective
400033866048SMatthew G. Knepley 
400133866048SMatthew G. Knepley    Input Parameters:
400233866048SMatthew G. Knepley +  snes - the SNES context
400333866048SMatthew G. Knepley -  reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the
400433866048SMatthew G. Knepley             manual pages for the individual convergence tests for complete lists
400533866048SMatthew G. Knepley 
400633866048SMatthew G. Knepley    Level: intermediate
400733866048SMatthew G. Knepley 
400833866048SMatthew G. Knepley .keywords: SNES, nonlinear, set, convergence, test
400933866048SMatthew G. Knepley .seealso: SNESGetConvergedReason(), SNESSetConvergenceTest(), SNESConvergedReason
401033866048SMatthew G. Knepley @*/
401133866048SMatthew G. Knepley PetscErrorCode SNESSetConvergedReason(SNES snes,SNESConvergedReason reason)
401233866048SMatthew G. Knepley {
401333866048SMatthew G. Knepley   PetscFunctionBegin;
401433866048SMatthew G. Knepley   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
401533866048SMatthew G. Knepley   snes->reason = reason;
401633866048SMatthew G. Knepley   PetscFunctionReturn(0);
401733866048SMatthew G. Knepley }
401833866048SMatthew G. Knepley 
4019c9005455SLois Curfman McInnes /*@
4020c9005455SLois Curfman McInnes    SNESSetConvergenceHistory - Sets the array used to hold the convergence history.
4021c9005455SLois Curfman McInnes 
40223f9fe445SBarry Smith    Logically Collective on SNES
4023fee21e36SBarry Smith 
4024c7afd0dbSLois Curfman McInnes    Input Parameters:
4025c7afd0dbSLois Curfman McInnes +  snes - iterative context obtained from SNESCreate()
40268c7482ecSBarry Smith .  a   - array to hold history, this array will contain the function norms computed at each step
4027cd5578b5SBarry Smith .  its - integer array holds the number of linear iterations for each solve.
4028758f92a0SBarry Smith .  na  - size of a and its
402964731454SLois Curfman McInnes -  reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero,
4030758f92a0SBarry Smith            else it continues storing new values for new nonlinear solves after the old ones
4031c7afd0dbSLois Curfman McInnes 
4032308dcc3eSBarry Smith    Notes:
40330298fd71SBarry Smith    If 'a' and 'its' are NULL then space is allocated for the history. If 'na' PETSC_DECIDE or PETSC_DEFAULT then a
4034308dcc3eSBarry Smith    default array of length 10000 is allocated.
4035308dcc3eSBarry Smith 
4036c9005455SLois Curfman McInnes    This routine is useful, e.g., when running a code for purposes
4037c9005455SLois Curfman McInnes    of accurate performance monitoring, when no I/O should be done
4038c9005455SLois Curfman McInnes    during the section of code that is being timed.
4039c9005455SLois Curfman McInnes 
404036851e7fSLois Curfman McInnes    Level: intermediate
404136851e7fSLois Curfman McInnes 
4042c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history
4043758f92a0SBarry Smith 
404408405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory()
4045758f92a0SBarry Smith 
4046c9005455SLois Curfman McInnes @*/
40477087cfbeSBarry Smith PetscErrorCode  SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscBool reset)
4048c9005455SLois Curfman McInnes {
4049308dcc3eSBarry Smith   PetscErrorCode ierr;
4050308dcc3eSBarry Smith 
40513a40ed3dSBarry Smith   PetscFunctionBegin;
40520700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
40537a1ec6d4SBarry Smith   if (a) PetscValidScalarPointer(a,2);
4054a562a398SLisandro Dalcin   if (its) PetscValidIntPointer(its,3);
40557a1ec6d4SBarry Smith   if (!a) {
4056308dcc3eSBarry Smith     if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
40577fdeb8b9SBarry Smith     ierr = PetscCalloc1(na,&a);CHKERRQ(ierr);
40587fdeb8b9SBarry Smith     ierr = PetscCalloc1(na,&its);CHKERRQ(ierr);
4059f5af7f23SKarl Rupp 
4060308dcc3eSBarry Smith     snes->conv_malloc = PETSC_TRUE;
4061308dcc3eSBarry Smith   }
4062c9005455SLois Curfman McInnes   snes->conv_hist       = a;
4063758f92a0SBarry Smith   snes->conv_hist_its   = its;
4064758f92a0SBarry Smith   snes->conv_hist_max   = na;
4065a12bc48fSLisandro Dalcin   snes->conv_hist_len   = 0;
4066758f92a0SBarry Smith   snes->conv_hist_reset = reset;
4067758f92a0SBarry Smith   PetscFunctionReturn(0);
4068758f92a0SBarry Smith }
4069758f92a0SBarry Smith 
4070308dcc3eSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
4071c6db04a5SJed Brown #include <engine.h>   /* MATLAB include file */
4072c6db04a5SJed Brown #include <mex.h>      /* MATLAB include file */
407399e0435eSBarry Smith 
40748cc058d9SJed Brown PETSC_EXTERN mxArray *SNESGetConvergenceHistoryMatlab(SNES snes)
4075308dcc3eSBarry Smith {
4076308dcc3eSBarry Smith   mxArray   *mat;
4077308dcc3eSBarry Smith   PetscInt  i;
4078308dcc3eSBarry Smith   PetscReal *ar;
4079308dcc3eSBarry Smith 
4080308dcc3eSBarry Smith   PetscFunctionBegin;
4081308dcc3eSBarry Smith   mat = mxCreateDoubleMatrix(snes->conv_hist_len,1,mxREAL);
4082308dcc3eSBarry Smith   ar  = (PetscReal*) mxGetData(mat);
4083f5af7f23SKarl Rupp   for (i=0; i<snes->conv_hist_len; i++) ar[i] = snes->conv_hist[i];
4084308dcc3eSBarry Smith   PetscFunctionReturn(mat);
4085308dcc3eSBarry Smith }
4086308dcc3eSBarry Smith #endif
4087308dcc3eSBarry Smith 
40880c4c9dddSBarry Smith /*@C
4089758f92a0SBarry Smith    SNESGetConvergenceHistory - Gets the array used to hold the convergence history.
4090758f92a0SBarry Smith 
40913f9fe445SBarry Smith    Not Collective
4092758f92a0SBarry Smith 
4093758f92a0SBarry Smith    Input Parameter:
4094758f92a0SBarry Smith .  snes - iterative context obtained from SNESCreate()
4095758f92a0SBarry Smith 
4096758f92a0SBarry Smith    Output Parameters:
4097758f92a0SBarry Smith .  a   - array to hold history
4098758f92a0SBarry Smith .  its - integer array holds the number of linear iterations (or
4099758f92a0SBarry Smith          negative if not converged) for each solve.
4100758f92a0SBarry Smith -  na  - size of a and its
4101758f92a0SBarry Smith 
4102758f92a0SBarry Smith    Notes:
4103758f92a0SBarry Smith     The calling sequence for this routine in Fortran is
4104758f92a0SBarry Smith $   call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr)
4105758f92a0SBarry Smith 
4106758f92a0SBarry Smith    This routine is useful, e.g., when running a code for purposes
4107758f92a0SBarry Smith    of accurate performance monitoring, when no I/O should be done
4108758f92a0SBarry Smith    during the section of code that is being timed.
4109758f92a0SBarry Smith 
4110758f92a0SBarry Smith    Level: intermediate
4111758f92a0SBarry Smith 
4112758f92a0SBarry Smith .keywords: SNES, get, convergence, history
4113758f92a0SBarry Smith 
4114758f92a0SBarry Smith .seealso: SNESSetConvergencHistory()
4115758f92a0SBarry Smith 
4116758f92a0SBarry Smith @*/
41177087cfbeSBarry Smith PetscErrorCode  SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na)
4118758f92a0SBarry Smith {
4119758f92a0SBarry Smith   PetscFunctionBegin;
41200700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4121758f92a0SBarry Smith   if (a)   *a   = snes->conv_hist;
4122758f92a0SBarry Smith   if (its) *its = snes->conv_hist_its;
4123758f92a0SBarry Smith   if (na)  *na  = snes->conv_hist_len;
41243a40ed3dSBarry Smith   PetscFunctionReturn(0);
4125c9005455SLois Curfman McInnes }
4126c9005455SLois Curfman McInnes 
4127ac226902SBarry Smith /*@C
412876b2cf59SMatthew Knepley   SNESSetUpdate - Sets the general-purpose update function called
4129eec8f646SJed Brown   at the beginning of every iteration of the nonlinear solve. Specifically
41307e4bb74cSBarry Smith   it is called just before the Jacobian is "evaluated".
413176b2cf59SMatthew Knepley 
41323f9fe445SBarry Smith   Logically Collective on SNES
413376b2cf59SMatthew Knepley 
413476b2cf59SMatthew Knepley   Input Parameters:
413576b2cf59SMatthew Knepley . snes - The nonlinear solver context
413676b2cf59SMatthew Knepley . func - The function
413776b2cf59SMatthew Knepley 
413876b2cf59SMatthew Knepley   Calling sequence of func:
4139b5d30489SBarry Smith . func (SNES snes, PetscInt step);
414076b2cf59SMatthew Knepley 
414176b2cf59SMatthew Knepley . step - The current step of the iteration
414276b2cf59SMatthew Knepley 
4143fe97e370SBarry Smith   Level: advanced
4144fe97e370SBarry Smith 
4145fe97e370SBarry 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()
4146fe97e370SBarry Smith         This is not used by most users.
414776b2cf59SMatthew Knepley 
414876b2cf59SMatthew Knepley .keywords: SNES, update
4149b5d30489SBarry Smith 
41508d359177SBarry Smith .seealso SNESSetJacobian(), SNESSolve()
415176b2cf59SMatthew Knepley @*/
41527087cfbeSBarry Smith PetscErrorCode  SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt))
415376b2cf59SMatthew Knepley {
415476b2cf59SMatthew Knepley   PetscFunctionBegin;
41550700a824SBarry Smith   PetscValidHeaderSpecific(snes, SNES_CLASSID,1);
4156e7788613SBarry Smith   snes->ops->update = func;
415776b2cf59SMatthew Knepley   PetscFunctionReturn(0);
415876b2cf59SMatthew Knepley }
415976b2cf59SMatthew Knepley 
41609b94acceSBarry Smith /*
41619b94acceSBarry Smith    SNESScaleStep_Private - Scales a step so that its length is less than the
41629b94acceSBarry Smith    positive parameter delta.
41639b94acceSBarry Smith 
41649b94acceSBarry Smith     Input Parameters:
4165c7afd0dbSLois Curfman McInnes +   snes - the SNES context
41669b94acceSBarry Smith .   y - approximate solution of linear system
41679b94acceSBarry Smith .   fnorm - 2-norm of current function
4168c7afd0dbSLois Curfman McInnes -   delta - trust region size
41699b94acceSBarry Smith 
41709b94acceSBarry Smith     Output Parameters:
4171c7afd0dbSLois Curfman McInnes +   gpnorm - predicted function norm at the new point, assuming local
41729b94acceSBarry Smith     linearization.  The value is zero if the step lies within the trust
41739b94acceSBarry Smith     region, and exceeds zero otherwise.
4174c7afd0dbSLois Curfman McInnes -   ynorm - 2-norm of the step
41759b94acceSBarry Smith 
41769b94acceSBarry Smith     Note:
417704d7464bSBarry Smith     For non-trust region methods such as SNESNEWTONLS, the parameter delta
41789b94acceSBarry Smith     is set to be the maximum allowable step size.
41799b94acceSBarry Smith 
41809b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step
41819b94acceSBarry Smith */
4182dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm)
41839b94acceSBarry Smith {
4184064f8208SBarry Smith   PetscReal      nrm;
4185ea709b57SSatish Balay   PetscScalar    cnorm;
4186dfbe8321SBarry Smith   PetscErrorCode ierr;
41873a40ed3dSBarry Smith 
41883a40ed3dSBarry Smith   PetscFunctionBegin;
41890700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
41900700a824SBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,2);
4191c9780b6fSBarry Smith   PetscCheckSameComm(snes,1,y,2);
4192184914b5SBarry Smith 
4193064f8208SBarry Smith   ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr);
4194064f8208SBarry Smith   if (nrm > *delta) {
4195064f8208SBarry Smith     nrm     = *delta/nrm;
4196064f8208SBarry Smith     *gpnorm = (1.0 - nrm)*(*fnorm);
4197064f8208SBarry Smith     cnorm   = nrm;
41982dcb1b2aSMatthew Knepley     ierr    = VecScale(y,cnorm);CHKERRQ(ierr);
41999b94acceSBarry Smith     *ynorm  = *delta;
42009b94acceSBarry Smith   } else {
42019b94acceSBarry Smith     *gpnorm = 0.0;
4202064f8208SBarry Smith     *ynorm  = nrm;
42039b94acceSBarry Smith   }
42043a40ed3dSBarry Smith   PetscFunctionReturn(0);
42059b94acceSBarry Smith }
42069b94acceSBarry Smith 
42072a359c20SBarry Smith /*@
42082a359c20SBarry Smith    SNESReasonView - Displays the reason a SNES solve converged or diverged to a viewer
42092a359c20SBarry Smith 
42102a359c20SBarry Smith    Collective on SNES
42112a359c20SBarry Smith 
42122a359c20SBarry Smith    Parameter:
42132a359c20SBarry Smith +  snes - iterative context obtained from SNESCreate()
42142a359c20SBarry Smith -  viewer - the viewer to display the reason
42152a359c20SBarry Smith 
42162a359c20SBarry Smith 
42172a359c20SBarry Smith    Options Database Keys:
42182a359c20SBarry Smith .  -snes_converged_reason - print reason for converged or diverged, also prints number of iterations
42192a359c20SBarry Smith 
42202a359c20SBarry Smith    Level: beginner
42212a359c20SBarry Smith 
42222a359c20SBarry Smith .keywords: SNES, solve, linear system
42232a359c20SBarry Smith 
42242a359c20SBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault()
42252a359c20SBarry Smith 
42262a359c20SBarry Smith @*/
42272a359c20SBarry Smith PetscErrorCode  SNESReasonView(SNES snes,PetscViewer viewer)
42282a359c20SBarry Smith {
422975cca76cSMatthew G. Knepley   PetscViewerFormat format;
42302a359c20SBarry Smith   PetscBool         isAscii;
423175cca76cSMatthew G. Knepley   PetscErrorCode    ierr;
42322a359c20SBarry Smith 
42332a359c20SBarry Smith   PetscFunctionBegin;
42342a359c20SBarry Smith   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isAscii);CHKERRQ(ierr);
42352a359c20SBarry Smith   if (isAscii) {
423675cca76cSMatthew G. Knepley     ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
42372a359c20SBarry Smith     ierr = PetscViewerASCIIAddTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr);
423875cca76cSMatthew G. Knepley     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
423975cca76cSMatthew G. Knepley       DM                dm;
424075cca76cSMatthew G. Knepley       Vec               u;
424175cca76cSMatthew G. Knepley       PetscDS           prob;
424275cca76cSMatthew G. Knepley       PetscInt          Nf, f;
424375cca76cSMatthew G. Knepley       PetscErrorCode (**exactFuncs)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar[], void *);
424475cca76cSMatthew G. Knepley       PetscReal         error;
424575cca76cSMatthew G. Knepley 
424675cca76cSMatthew G. Knepley       ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
424775cca76cSMatthew G. Knepley       ierr = SNESGetSolution(snes, &u);CHKERRQ(ierr);
424875cca76cSMatthew G. Knepley       ierr = DMGetDS(dm, &prob);CHKERRQ(ierr);
424975cca76cSMatthew G. Knepley       ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr);
425075cca76cSMatthew G. Knepley       ierr = PetscMalloc1(Nf, &exactFuncs);CHKERRQ(ierr);
425175cca76cSMatthew G. Knepley       for (f = 0; f < Nf; ++f) {ierr = PetscDSGetExactSolution(prob, f, &exactFuncs[f]);CHKERRQ(ierr);}
425275cca76cSMatthew G. Knepley       ierr = DMComputeL2Diff(dm, 0.0, exactFuncs, NULL, u, &error);CHKERRQ(ierr);
425375cca76cSMatthew G. Knepley       ierr = PetscFree(exactFuncs);CHKERRQ(ierr);
425475cca76cSMatthew G. Knepley       if (error < 1.0e-11) {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: < 1.0e-11\n");CHKERRQ(ierr);}
425575cca76cSMatthew G. Knepley       else                 {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: %g\n", error);CHKERRQ(ierr);}
425675cca76cSMatthew G. Knepley     }
42572a359c20SBarry Smith     if (snes->reason > 0) {
42582a359c20SBarry Smith       if (((PetscObject) snes)->prefix) {
42592a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve converged due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
42602a359c20SBarry Smith       } else {
42612a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve converged due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
42622a359c20SBarry Smith       }
42632a359c20SBarry Smith     } else {
42642a359c20SBarry Smith       if (((PetscObject) snes)->prefix) {
42652a359c20SBarry 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);
42662a359c20SBarry Smith       } else {
42672a359c20SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve did not converge due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr);
42682a359c20SBarry Smith       }
42692a359c20SBarry Smith     }
42702a359c20SBarry Smith     ierr = PetscViewerASCIISubtractTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr);
42712a359c20SBarry Smith   }
42722a359c20SBarry Smith   PetscFunctionReturn(0);
42732a359c20SBarry Smith }
42742a359c20SBarry Smith 
42752a359c20SBarry Smith /*@C
42762a359c20SBarry Smith   SNESReasonViewFromOptions - Processes command line options to determine if/how a SNESReason is to be viewed.
42772a359c20SBarry Smith 
42782a359c20SBarry Smith   Collective on SNES
42792a359c20SBarry Smith 
42802a359c20SBarry Smith   Input Parameters:
42812a359c20SBarry Smith . snes   - the SNES object
42822a359c20SBarry Smith 
42832a359c20SBarry Smith   Level: intermediate
42842a359c20SBarry Smith 
42852a359c20SBarry Smith @*/
42862a359c20SBarry Smith PetscErrorCode SNESReasonViewFromOptions(SNES snes)
42872a359c20SBarry Smith {
42882a359c20SBarry Smith   PetscErrorCode    ierr;
42892a359c20SBarry Smith   PetscViewer       viewer;
42902a359c20SBarry Smith   PetscBool         flg;
42912a359c20SBarry Smith   static PetscBool  incall = PETSC_FALSE;
42922a359c20SBarry Smith   PetscViewerFormat format;
42932a359c20SBarry Smith 
42942a359c20SBarry Smith   PetscFunctionBegin;
42952a359c20SBarry Smith   if (incall) PetscFunctionReturn(0);
42962a359c20SBarry Smith   incall = PETSC_TRUE;
429716413a6aSBarry Smith   ierr   = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_converged_reason",&viewer,&format,&flg);CHKERRQ(ierr);
42982a359c20SBarry Smith   if (flg) {
42992a359c20SBarry Smith     ierr = PetscViewerPushFormat(viewer,format);CHKERRQ(ierr);
43002a359c20SBarry Smith     ierr = SNESReasonView(snes,viewer);CHKERRQ(ierr);
43012a359c20SBarry Smith     ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
43022a359c20SBarry Smith     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
43032a359c20SBarry Smith   }
43042a359c20SBarry Smith   incall = PETSC_FALSE;
43052a359c20SBarry Smith   PetscFunctionReturn(0);
43062a359c20SBarry Smith }
43072a359c20SBarry Smith 
4308487a658cSBarry Smith /*@
4309f69a0ea3SMatthew Knepley    SNESSolve - Solves a nonlinear system F(x) = b.
4310f69a0ea3SMatthew Knepley    Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX().
43119b94acceSBarry Smith 
4312c7afd0dbSLois Curfman McInnes    Collective on SNES
4313c7afd0dbSLois Curfman McInnes 
4314b2002411SLois Curfman McInnes    Input Parameters:
4315c7afd0dbSLois Curfman McInnes +  snes - the SNES context
43160298fd71SBarry Smith .  b - the constant part of the equation F(x) = b, or NULL to use zero.
431785385478SLisandro Dalcin -  x - the solution vector.
43189b94acceSBarry Smith 
4319b2002411SLois Curfman McInnes    Notes:
43208ddd3da0SLois Curfman McInnes    The user should initialize the vector,x, with the initial guess
43218ddd3da0SLois Curfman McInnes    for the nonlinear solve prior to calling SNESSolve.  In particular,
43228ddd3da0SLois Curfman McInnes    to employ an initial guess of zero, the user should explicitly set
43238ddd3da0SLois Curfman McInnes    this vector to zero by calling VecSet().
43248ddd3da0SLois Curfman McInnes 
432536851e7fSLois Curfman McInnes    Level: beginner
432636851e7fSLois Curfman McInnes 
43279b94acceSBarry Smith .keywords: SNES, nonlinear, solve
43289b94acceSBarry Smith 
4329c0df2a02SJed Brown .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetGridSequence(), SNESGetSolution()
43309b94acceSBarry Smith @*/
43317087cfbeSBarry Smith PetscErrorCode  SNESSolve(SNES snes,Vec b,Vec x)
43329b94acceSBarry Smith {
4333dfbe8321SBarry Smith   PetscErrorCode    ierr;
4334ace3abfcSBarry Smith   PetscBool         flg;
4335efd51863SBarry Smith   PetscInt          grid;
43360298fd71SBarry Smith   Vec               xcreated = NULL;
4337caa4e7f2SJed Brown   DM                dm;
4338052efed2SBarry Smith 
43393a40ed3dSBarry Smith   PetscFunctionBegin;
43400700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4341a69afd8bSBarry Smith   if (x) PetscValidHeaderSpecific(x,VEC_CLASSID,3);
4342a69afd8bSBarry Smith   if (x) PetscCheckSameComm(snes,1,x,3);
43430700a824SBarry Smith   if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2);
434485385478SLisandro Dalcin   if (b) PetscCheckSameComm(snes,1,b,2);
434585385478SLisandro Dalcin 
434634b4d3a8SMatthew G. Knepley   /* High level operations using the nonlinear solver */
434706fc46c8SMatthew G. Knepley   {
434806fc46c8SMatthew G. Knepley     PetscViewer       viewer;
434906fc46c8SMatthew G. Knepley     PetscViewerFormat format;
43507c88af5aSMatthew G. Knepley     PetscInt          num;
435106fc46c8SMatthew G. Knepley     PetscBool         flg;
435206fc46c8SMatthew G. Knepley     static PetscBool  incall = PETSC_FALSE;
435306fc46c8SMatthew G. Knepley 
435406fc46c8SMatthew G. Knepley     if (!incall) {
435534b4d3a8SMatthew G. Knepley       /* Estimate the convergence rate of the discretization */
435616413a6aSBarry Smith       ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) snes),((PetscObject)snes)->options, ((PetscObject) snes)->prefix, "-snes_convergence_estimate", &viewer, &format, &flg);CHKERRQ(ierr);
435706fc46c8SMatthew G. Knepley       if (flg) {
435806fc46c8SMatthew G. Knepley         PetscConvEst conv;
435946079b62SMatthew G. Knepley         DM           dm;
436046079b62SMatthew G. Knepley         PetscReal   *alpha; /* Convergence rate of the solution error for each field in the L_2 norm */
436146079b62SMatthew G. Knepley         PetscInt     Nf;
436206fc46c8SMatthew G. Knepley 
436306fc46c8SMatthew G. Knepley         incall = PETSC_TRUE;
436446079b62SMatthew G. Knepley         ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
436546079b62SMatthew G. Knepley         ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr);
436646079b62SMatthew G. Knepley         ierr = PetscMalloc1(Nf, &alpha);CHKERRQ(ierr);
436706fc46c8SMatthew G. Knepley         ierr = PetscConvEstCreate(PetscObjectComm((PetscObject) snes), &conv);CHKERRQ(ierr);
436806fc46c8SMatthew G. Knepley         ierr = PetscConvEstSetSolver(conv, snes);CHKERRQ(ierr);
436906fc46c8SMatthew G. Knepley         ierr = PetscConvEstSetFromOptions(conv);CHKERRQ(ierr);
43700955ed61SMatthew G. Knepley         ierr = PetscConvEstSetUp(conv);CHKERRQ(ierr);
437146079b62SMatthew G. Knepley         ierr = PetscConvEstGetConvRate(conv, alpha);CHKERRQ(ierr);
437206fc46c8SMatthew G. Knepley         ierr = PetscViewerPushFormat(viewer, format);CHKERRQ(ierr);
437306fc46c8SMatthew G. Knepley         ierr = PetscConvEstRateView(conv, alpha, viewer);CHKERRQ(ierr);
437406fc46c8SMatthew G. Knepley         ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
437506fc46c8SMatthew G. Knepley         ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
437606fc46c8SMatthew G. Knepley         ierr = PetscConvEstDestroy(&conv);CHKERRQ(ierr);
437746079b62SMatthew G. Knepley         ierr = PetscFree(alpha);CHKERRQ(ierr);
437806fc46c8SMatthew G. Knepley         incall = PETSC_FALSE;
437906fc46c8SMatthew G. Knepley       }
438034b4d3a8SMatthew G. Knepley       /* Adaptively refine the initial grid */
4381b2588ea6SMatthew G. Knepley       num  = 1;
4382b2588ea6SMatthew G. Knepley       ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_initial", &num, &flg);CHKERRQ(ierr);
438334b4d3a8SMatthew G. Knepley       if (flg) {
438434b4d3a8SMatthew G. Knepley         DMAdaptor adaptor;
438534b4d3a8SMatthew G. Knepley 
438634b4d3a8SMatthew G. Knepley         incall = PETSC_TRUE;
438734b4d3a8SMatthew G. Knepley         ierr = DMAdaptorCreate(PETSC_COMM_WORLD, &adaptor);CHKERRQ(ierr);
438834b4d3a8SMatthew G. Knepley         ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr);
4389b2588ea6SMatthew G. Knepley         ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr);
439034b4d3a8SMatthew G. Knepley         ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr);
439134b4d3a8SMatthew G. Knepley         ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr);
439234b4d3a8SMatthew G. Knepley         ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_INITIAL, &dm, &x);CHKERRQ(ierr);
439334b4d3a8SMatthew G. Knepley         ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr);
439434b4d3a8SMatthew G. Knepley         incall = PETSC_FALSE;
439534b4d3a8SMatthew G. Knepley       }
43967c88af5aSMatthew G. Knepley       /* Use grid sequencing to adapt */
43977c88af5aSMatthew G. Knepley       num  = 0;
43987c88af5aSMatthew G. Knepley       ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_sequence", &num, NULL);CHKERRQ(ierr);
43997c88af5aSMatthew G. Knepley       if (num) {
44007c88af5aSMatthew G. Knepley         DMAdaptor adaptor;
44017c88af5aSMatthew G. Knepley 
44027c88af5aSMatthew G. Knepley         incall = PETSC_TRUE;
44037c88af5aSMatthew G. Knepley         ierr = DMAdaptorCreate(PETSC_COMM_WORLD, &adaptor);CHKERRQ(ierr);
44047c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr);
44057c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr);
44067c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr);
44077c88af5aSMatthew G. Knepley         ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr);
44087c88af5aSMatthew G. Knepley         ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_SEQUENTIAL, &dm, &x);CHKERRQ(ierr);
44097c88af5aSMatthew G. Knepley         ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr);
44107c88af5aSMatthew G. Knepley         incall = PETSC_FALSE;
44117c88af5aSMatthew G. Knepley       }
441206fc46c8SMatthew G. Knepley     }
441306fc46c8SMatthew G. Knepley   }
4414caa4e7f2SJed Brown   if (!x) {
4415caa4e7f2SJed Brown     ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4416caa4e7f2SJed Brown     ierr = DMCreateGlobalVector(dm,&xcreated);CHKERRQ(ierr);
4417a69afd8bSBarry Smith     x    = xcreated;
4418a69afd8bSBarry Smith   }
4419ce1779c8SBarry Smith   ierr = SNESViewFromOptions(snes,NULL,"-snes_view_pre");CHKERRQ(ierr);
4420f05ece33SBarry Smith 
4421ce94432eSBarry Smith   for (grid=0; grid<snes->gridsequence; grid++) {ierr = PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);}
4422efd51863SBarry Smith   for (grid=0; grid<snes->gridsequence+1; grid++) {
4423efd51863SBarry Smith 
442485385478SLisandro Dalcin     /* set solution vector */
4425efd51863SBarry Smith     if (!grid) {ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);}
44266bf464f9SBarry Smith     ierr          = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
442785385478SLisandro Dalcin     snes->vec_sol = x;
4428caa4e7f2SJed Brown     ierr          = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4429caa4e7f2SJed Brown 
4430caa4e7f2SJed Brown     /* set affine vector if provided */
443185385478SLisandro Dalcin     if (b) { ierr = PetscObjectReference((PetscObject)b);CHKERRQ(ierr); }
44326bf464f9SBarry Smith     ierr          = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr);
443385385478SLisandro Dalcin     snes->vec_rhs = b;
443485385478SLisandro Dalcin 
4435154060b5SMatthew G. Knepley     if (snes->vec_func == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector");
4436154060b5SMatthew 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");
4437154060b5SMatthew G. Knepley     if (!snes->vec_sol_update /* && snes->vec_sol */) {
4438154060b5SMatthew G. Knepley       ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr);
4439154060b5SMatthew G. Knepley       ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->vec_sol_update);CHKERRQ(ierr);
4440154060b5SMatthew G. Knepley     }
4441154060b5SMatthew G. Knepley     ierr = DMShellSetGlobalVector(dm,snes->vec_sol);CHKERRQ(ierr);
444270e92668SMatthew Knepley     ierr = SNESSetUp(snes);CHKERRQ(ierr);
44433f149594SLisandro Dalcin 
44447eee914bSBarry Smith     if (!grid) {
44457eee914bSBarry Smith       if (snes->ops->computeinitialguess) {
4446d25893d9SBarry Smith         ierr = (*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP);CHKERRQ(ierr);
4447d25893d9SBarry Smith       }
4448dd568438SSatish Balay     }
4449d25893d9SBarry Smith 
4450abc0a331SBarry Smith     if (snes->conv_hist_reset) snes->conv_hist_len = 0;
4451971e163fSPeter Brune     if (snes->counters_reset) {snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;}
4452d5e45103SBarry Smith 
44533f149594SLisandro Dalcin     ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
44544936397dSBarry Smith     ierr = (*snes->ops->solve)(snes);CHKERRQ(ierr);
445585385478SLisandro Dalcin     ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr);
445617186662SBarry Smith     if (!snes->reason) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason");
4457422a814eSBarry Smith     snes->domainerror = PETSC_FALSE; /* clear the flag if it has been set */
44583f149594SLisandro Dalcin 
445937ec4e1aSPeter Brune     if (snes->lagjac_persist) snes->jac_iter += snes->iter;
446037ec4e1aSPeter Brune     if (snes->lagpre_persist) snes->pre_iter += snes->iter;
446137ec4e1aSPeter Brune 
446216413a6aSBarry Smith     ierr   = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_test_local_min",NULL,NULL,&flg);CHKERRQ(ierr);
4463da9b6338SBarry Smith     if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); }
44642a359c20SBarry Smith     ierr = SNESReasonViewFromOptions(snes);CHKERRQ(ierr);
44655968eb51SBarry Smith 
4466ce94432eSBarry Smith     if (snes->errorifnotconverged && snes->reason < 0) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged");
44679c8e83a9SBarry Smith     if (snes->reason < 0) break;
4468efd51863SBarry Smith     if (grid <  snes->gridsequence) {
4469efd51863SBarry Smith       DM  fine;
4470efd51863SBarry Smith       Vec xnew;
4471efd51863SBarry Smith       Mat interp;
4472efd51863SBarry Smith 
4473ce94432eSBarry Smith       ierr = DMRefine(snes->dm,PetscObjectComm((PetscObject)snes),&fine);CHKERRQ(ierr);
4474ce94432eSBarry Smith       if (!fine) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"DMRefine() did not perform any refinement, cannot continue grid sequencing");
44750298fd71SBarry Smith       ierr = DMCreateInterpolation(snes->dm,fine,&interp,NULL);CHKERRQ(ierr);
4476efd51863SBarry Smith       ierr = DMCreateGlobalVector(fine,&xnew);CHKERRQ(ierr);
4477efd51863SBarry Smith       ierr = MatInterpolate(interp,x,xnew);CHKERRQ(ierr);
4478c833c3b5SJed Brown       ierr = DMInterpolate(snes->dm,interp,fine);CHKERRQ(ierr);
4479efd51863SBarry Smith       ierr = MatDestroy(&interp);CHKERRQ(ierr);
4480efd51863SBarry Smith       x    = xnew;
4481efd51863SBarry Smith 
4482efd51863SBarry Smith       ierr = SNESReset(snes);CHKERRQ(ierr);
4483efd51863SBarry Smith       ierr = SNESSetDM(snes,fine);CHKERRQ(ierr);
4484352f405dSMatthew G. Knepley       ierr = SNESResetFromOptions(snes);CHKERRQ(ierr);
4485efd51863SBarry Smith       ierr = DMDestroy(&fine);CHKERRQ(ierr);
4486ce94432eSBarry Smith       ierr = PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);
4487efd51863SBarry Smith     }
4488efd51863SBarry Smith   }
4489ce1779c8SBarry Smith   ierr = SNESViewFromOptions(snes,NULL,"-snes_view");CHKERRQ(ierr);
4490685405a1SBarry Smith   ierr = VecViewFromOptions(snes->vec_sol,(PetscObject)snes,"-snes_view_solution");CHKERRQ(ierr);
44913f7e2da0SPeter Brune 
4492a69afd8bSBarry Smith   ierr = VecDestroy(&xcreated);CHKERRQ(ierr);
4493e04113cfSBarry Smith   ierr = PetscObjectSAWsBlock((PetscObject)snes);CHKERRQ(ierr);
44943a40ed3dSBarry Smith   PetscFunctionReturn(0);
44959b94acceSBarry Smith }
44969b94acceSBarry Smith 
44979b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */
44989b94acceSBarry Smith 
449982bf6240SBarry Smith /*@C
45004b0e389bSBarry Smith    SNESSetType - Sets the method for the nonlinear solver.
45019b94acceSBarry Smith 
4502fee21e36SBarry Smith    Collective on SNES
4503fee21e36SBarry Smith 
4504c7afd0dbSLois Curfman McInnes    Input Parameters:
4505c7afd0dbSLois Curfman McInnes +  snes - the SNES context
4506454a90a3SBarry Smith -  type - a known method
4507c7afd0dbSLois Curfman McInnes 
4508c7afd0dbSLois Curfman McInnes    Options Database Key:
4509454a90a3SBarry Smith .  -snes_type <type> - Sets the method; use -help for a list
451004d7464bSBarry Smith    of available methods (for instance, newtonls or newtontr)
4511ae12b187SLois Curfman McInnes 
45129b94acceSBarry Smith    Notes:
4513e090d566SSatish Balay    See "petsc/include/petscsnes.h" for available methods (for instance)
451404d7464bSBarry Smith +    SNESNEWTONLS - Newton's method with line search
4515c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
451604d7464bSBarry Smith .    SNESNEWTONTR - Newton's method with trust region
4517c7afd0dbSLois Curfman McInnes      (systems of nonlinear equations)
45189b94acceSBarry Smith 
4519ae12b187SLois Curfman McInnes   Normally, it is best to use the SNESSetFromOptions() command and then
4520ae12b187SLois Curfman McInnes   set the SNES solver type from the options database rather than by using
4521ae12b187SLois Curfman McInnes   this routine.  Using the options database provides the user with
4522ae12b187SLois Curfman McInnes   maximum flexibility in evaluating the many nonlinear solvers.
4523ae12b187SLois Curfman McInnes   The SNESSetType() routine is provided for those situations where it
4524ae12b187SLois Curfman McInnes   is necessary to set the nonlinear solver independently of the command
4525ae12b187SLois Curfman McInnes   line or options database.  This might be the case, for example, when
4526ae12b187SLois Curfman McInnes   the choice of solver changes during the execution of the program,
4527ae12b187SLois Curfman McInnes   and the user's application is taking responsibility for choosing the
4528b0a32e0cSBarry Smith   appropriate method.
452936851e7fSLois Curfman McInnes 
453095452b02SPatrick Sanan     Developer Notes:
453195452b02SPatrick Sanan     SNESRegister() adds a constructor for a new SNESType to SNESList, SNESSetType() locates
45328f6c3df8SBarry Smith     the constructor in that list and calls it to create the spexific object.
45338f6c3df8SBarry Smith 
453436851e7fSLois Curfman McInnes   Level: intermediate
4535a703fe33SLois Curfman McInnes 
4536454a90a3SBarry Smith .keywords: SNES, set, type
4537435da068SBarry Smith 
45388f6c3df8SBarry Smith .seealso: SNESType, SNESCreate(), SNESDestroy(), SNESGetType(), SNESSetFromOptions()
4539435da068SBarry Smith 
45409b94acceSBarry Smith @*/
454119fd82e9SBarry Smith PetscErrorCode  SNESSetType(SNES snes,SNESType type)
45429b94acceSBarry Smith {
4543dfbe8321SBarry Smith   PetscErrorCode ierr,(*r)(SNES);
4544ace3abfcSBarry Smith   PetscBool      match;
45453a40ed3dSBarry Smith 
45463a40ed3dSBarry Smith   PetscFunctionBegin;
45470700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
45484482741eSBarry Smith   PetscValidCharPointer(type,2);
454982bf6240SBarry Smith 
4550251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr);
45510f5bd95cSBarry Smith   if (match) PetscFunctionReturn(0);
455292ff6ae8SBarry Smith 
45531c9cd337SJed Brown   ierr =  PetscFunctionListFind(SNESList,type,&r);CHKERRQ(ierr);
4554e32f2f54SBarry Smith   if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type);
455575396ef9SLisandro Dalcin   /* Destroy the previous private SNES context */
4556b5c23020SJed Brown   if (snes->ops->destroy) {
4557b5c23020SJed Brown     ierr               = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr);
45580298fd71SBarry Smith     snes->ops->destroy = NULL;
4559b5c23020SJed Brown   }
456075396ef9SLisandro Dalcin   /* Reinitialize function pointers in SNESOps structure */
456175396ef9SLisandro Dalcin   snes->ops->setup          = 0;
456275396ef9SLisandro Dalcin   snes->ops->solve          = 0;
456375396ef9SLisandro Dalcin   snes->ops->view           = 0;
456475396ef9SLisandro Dalcin   snes->ops->setfromoptions = 0;
456575396ef9SLisandro Dalcin   snes->ops->destroy        = 0;
456675396ef9SLisandro Dalcin   /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */
456775396ef9SLisandro Dalcin   snes->setupcalled = PETSC_FALSE;
4568f5af7f23SKarl Rupp 
4569454a90a3SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr);
457003bfa161SLisandro Dalcin   ierr = (*r)(snes);CHKERRQ(ierr);
45713a40ed3dSBarry Smith   PetscFunctionReturn(0);
45729b94acceSBarry Smith }
45739b94acceSBarry Smith 
45749b94acceSBarry Smith /*@C
45759a28b0a6SLois Curfman McInnes    SNESGetType - Gets the SNES method type and name (as a string).
45769b94acceSBarry Smith 
4577c7afd0dbSLois Curfman McInnes    Not Collective
4578c7afd0dbSLois Curfman McInnes 
45799b94acceSBarry Smith    Input Parameter:
45804b0e389bSBarry Smith .  snes - nonlinear solver context
45819b94acceSBarry Smith 
45829b94acceSBarry Smith    Output Parameter:
45833a7fca6bSBarry Smith .  type - SNES method (a character string)
45849b94acceSBarry Smith 
458536851e7fSLois Curfman McInnes    Level: intermediate
458636851e7fSLois Curfman McInnes 
4587454a90a3SBarry Smith .keywords: SNES, nonlinear, get, type, name
45889b94acceSBarry Smith @*/
458919fd82e9SBarry Smith PetscErrorCode  SNESGetType(SNES snes,SNESType *type)
45909b94acceSBarry Smith {
45913a40ed3dSBarry Smith   PetscFunctionBegin;
45920700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
45934482741eSBarry Smith   PetscValidPointer(type,2);
45947adad957SLisandro Dalcin   *type = ((PetscObject)snes)->type_name;
45953a40ed3dSBarry Smith   PetscFunctionReturn(0);
45969b94acceSBarry Smith }
45979b94acceSBarry Smith 
45983cd8a7caSMatthew G. Knepley /*@
45993cd8a7caSMatthew G. Knepley   SNESSetSolution - Sets the solution vector for use by the SNES routines.
46003cd8a7caSMatthew G. Knepley 
46013cd8a7caSMatthew G. Knepley   Logically Collective on SNES and Vec
46023cd8a7caSMatthew G. Knepley 
46033cd8a7caSMatthew G. Knepley   Input Parameters:
46043cd8a7caSMatthew G. Knepley + snes - the SNES context obtained from SNESCreate()
46053cd8a7caSMatthew G. Knepley - u    - the solution vector
46063cd8a7caSMatthew G. Knepley 
46073cd8a7caSMatthew G. Knepley   Level: beginner
46083cd8a7caSMatthew G. Knepley 
46093cd8a7caSMatthew G. Knepley .keywords: SNES, set, solution
46103cd8a7caSMatthew G. Knepley @*/
46113cd8a7caSMatthew G. Knepley PetscErrorCode SNESSetSolution(SNES snes, Vec u)
46123cd8a7caSMatthew G. Knepley {
46133cd8a7caSMatthew G. Knepley   DM             dm;
46143cd8a7caSMatthew G. Knepley   PetscErrorCode ierr;
46153cd8a7caSMatthew G. Knepley 
46163cd8a7caSMatthew G. Knepley   PetscFunctionBegin;
46173cd8a7caSMatthew G. Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
46183cd8a7caSMatthew G. Knepley   PetscValidHeaderSpecific(u, VEC_CLASSID, 2);
46193cd8a7caSMatthew G. Knepley   ierr = PetscObjectReference((PetscObject) u);CHKERRQ(ierr);
46203cd8a7caSMatthew G. Knepley   ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr);
46213cd8a7caSMatthew G. Knepley 
46223cd8a7caSMatthew G. Knepley   snes->vec_sol = u;
46233cd8a7caSMatthew G. Knepley 
46243cd8a7caSMatthew G. Knepley   ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
46253cd8a7caSMatthew G. Knepley   ierr = DMShellSetGlobalVector(dm, u);CHKERRQ(ierr);
46263cd8a7caSMatthew G. Knepley   PetscFunctionReturn(0);
46273cd8a7caSMatthew G. Knepley }
46283cd8a7caSMatthew G. Knepley 
462952baeb72SSatish Balay /*@
46309b94acceSBarry Smith    SNESGetSolution - Returns the vector where the approximate solution is
4631c0df2a02SJed Brown    stored. This is the fine grid solution when using SNESSetGridSequence().
46329b94acceSBarry Smith 
4633c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
4634c7afd0dbSLois Curfman McInnes 
46359b94acceSBarry Smith    Input Parameter:
46369b94acceSBarry Smith .  snes - the SNES context
46379b94acceSBarry Smith 
46389b94acceSBarry Smith    Output Parameter:
46399b94acceSBarry Smith .  x - the solution
46409b94acceSBarry Smith 
464170e92668SMatthew Knepley    Level: intermediate
464236851e7fSLois Curfman McInnes 
46439b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution
46449b94acceSBarry Smith 
464585385478SLisandro Dalcin .seealso:  SNESGetSolutionUpdate(), SNESGetFunction()
46469b94acceSBarry Smith @*/
46477087cfbeSBarry Smith PetscErrorCode  SNESGetSolution(SNES snes,Vec *x)
46489b94acceSBarry Smith {
46493a40ed3dSBarry Smith   PetscFunctionBegin;
46500700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
46514482741eSBarry Smith   PetscValidPointer(x,2);
465285385478SLisandro Dalcin   *x = snes->vec_sol;
465370e92668SMatthew Knepley   PetscFunctionReturn(0);
465470e92668SMatthew Knepley }
465570e92668SMatthew Knepley 
465652baeb72SSatish Balay /*@
46579b94acceSBarry Smith    SNESGetSolutionUpdate - Returns the vector where the solution update is
46589b94acceSBarry Smith    stored.
46599b94acceSBarry Smith 
4660c7afd0dbSLois Curfman McInnes    Not Collective, but Vec is parallel if SNES is parallel
4661c7afd0dbSLois Curfman McInnes 
46629b94acceSBarry Smith    Input Parameter:
46639b94acceSBarry Smith .  snes - the SNES context
46649b94acceSBarry Smith 
46659b94acceSBarry Smith    Output Parameter:
46669b94acceSBarry Smith .  x - the solution update
46679b94acceSBarry Smith 
466836851e7fSLois Curfman McInnes    Level: advanced
466936851e7fSLois Curfman McInnes 
46709b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update
46719b94acceSBarry Smith 
467285385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction()
46739b94acceSBarry Smith @*/
46747087cfbeSBarry Smith PetscErrorCode  SNESGetSolutionUpdate(SNES snes,Vec *x)
46759b94acceSBarry Smith {
46763a40ed3dSBarry Smith   PetscFunctionBegin;
46770700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
46784482741eSBarry Smith   PetscValidPointer(x,2);
467985385478SLisandro Dalcin   *x = snes->vec_sol_update;
46803a40ed3dSBarry Smith   PetscFunctionReturn(0);
46819b94acceSBarry Smith }
46829b94acceSBarry Smith 
46839b94acceSBarry Smith /*@C
46843638b69dSLois Curfman McInnes    SNESGetFunction - Returns the vector where the function is stored.
46859b94acceSBarry Smith 
4686a63bb30eSJed Brown    Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet.
4687c7afd0dbSLois Curfman McInnes 
46889b94acceSBarry Smith    Input Parameter:
46899b94acceSBarry Smith .  snes - the SNES context
46909b94acceSBarry Smith 
46919b94acceSBarry Smith    Output Parameter:
46920298fd71SBarry Smith +  r - the vector that is used to store residuals (or NULL if you don't want it)
4693f8b49ee9SBarry Smith .  f - the function (or NULL if you don't want it); see SNESFunction for calling sequence details
46940298fd71SBarry Smith -  ctx - the function context (or NULL if you don't want it)
46959b94acceSBarry Smith 
469636851e7fSLois Curfman McInnes    Level: advanced
469736851e7fSLois Curfman McInnes 
469804edfde5SBarry Smith     Notes: The vector r DOES NOT, in general contain the current value of the SNES nonlinear function
469904edfde5SBarry Smith 
4700a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function
47019b94acceSBarry Smith 
4702bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetSolution(), SNESFunction
47039b94acceSBarry Smith @*/
4704f8b49ee9SBarry Smith PetscErrorCode  SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),void **ctx)
47059b94acceSBarry Smith {
4706a63bb30eSJed Brown   PetscErrorCode ierr;
47076cab3a1bSJed Brown   DM             dm;
4708a63bb30eSJed Brown 
47093a40ed3dSBarry Smith   PetscFunctionBegin;
47100700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4711a63bb30eSJed Brown   if (r) {
4712a63bb30eSJed Brown     if (!snes->vec_func) {
4713a63bb30eSJed Brown       if (snes->vec_rhs) {
4714a63bb30eSJed Brown         ierr = VecDuplicate(snes->vec_rhs,&snes->vec_func);CHKERRQ(ierr);
4715a63bb30eSJed Brown       } else if (snes->vec_sol) {
4716a63bb30eSJed Brown         ierr = VecDuplicate(snes->vec_sol,&snes->vec_func);CHKERRQ(ierr);
4717a63bb30eSJed Brown       } else if (snes->dm) {
4718a63bb30eSJed Brown         ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr);
4719a63bb30eSJed Brown       }
4720a63bb30eSJed Brown     }
4721a63bb30eSJed Brown     *r = snes->vec_func;
4722a63bb30eSJed Brown   }
47236cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4724f8b49ee9SBarry Smith   ierr = DMSNESGetFunction(dm,f,ctx);CHKERRQ(ierr);
47253a40ed3dSBarry Smith   PetscFunctionReturn(0);
47269b94acceSBarry Smith }
47279b94acceSBarry Smith 
4728c79ef259SPeter Brune /*@C
4729be95d8f1SBarry Smith    SNESGetNGS - Returns the NGS function and context.
4730c79ef259SPeter Brune 
4731c79ef259SPeter Brune    Input Parameter:
4732c79ef259SPeter Brune .  snes - the SNES context
4733c79ef259SPeter Brune 
4734c79ef259SPeter Brune    Output Parameter:
4735be95d8f1SBarry Smith +  f - the function (or NULL) see SNESNGSFunction for details
47360298fd71SBarry Smith -  ctx    - the function context (or NULL)
4737c79ef259SPeter Brune 
4738c79ef259SPeter Brune    Level: advanced
4739c79ef259SPeter Brune 
4740c79ef259SPeter Brune .keywords: SNES, nonlinear, get, function
4741c79ef259SPeter Brune 
4742be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetFunction()
4743c79ef259SPeter Brune @*/
4744c79ef259SPeter Brune 
4745be95d8f1SBarry Smith PetscErrorCode SNESGetNGS (SNES snes, PetscErrorCode (**f)(SNES, Vec, Vec, void*), void ** ctx)
4746646217ecSPeter Brune {
47476cab3a1bSJed Brown   PetscErrorCode ierr;
47486cab3a1bSJed Brown   DM             dm;
47496cab3a1bSJed Brown 
4750646217ecSPeter Brune   PetscFunctionBegin;
4751646217ecSPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
47526cab3a1bSJed Brown   ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr);
4753be95d8f1SBarry Smith   ierr = DMSNESGetNGS(dm,f,ctx);CHKERRQ(ierr);
4754646217ecSPeter Brune   PetscFunctionReturn(0);
4755646217ecSPeter Brune }
4756646217ecSPeter Brune 
47573c7409f5SSatish Balay /*@C
47583c7409f5SSatish Balay    SNESSetOptionsPrefix - Sets the prefix used for searching for all
4759d850072dSLois Curfman McInnes    SNES options in the database.
47603c7409f5SSatish Balay 
47613f9fe445SBarry Smith    Logically Collective on SNES
4762fee21e36SBarry Smith 
4763c7afd0dbSLois Curfman McInnes    Input Parameter:
4764c7afd0dbSLois Curfman McInnes +  snes - the SNES context
4765c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
4766c7afd0dbSLois Curfman McInnes 
4767d850072dSLois Curfman McInnes    Notes:
4768a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
4769c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
4770d850072dSLois Curfman McInnes 
477136851e7fSLois Curfman McInnes    Level: advanced
477236851e7fSLois Curfman McInnes 
47733c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database
4774a86d99e1SLois Curfman McInnes 
4775a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions()
47763c7409f5SSatish Balay @*/
47777087cfbeSBarry Smith PetscErrorCode  SNESSetOptionsPrefix(SNES snes,const char prefix[])
47783c7409f5SSatish Balay {
4779dfbe8321SBarry Smith   PetscErrorCode ierr;
47803c7409f5SSatish Balay 
47813a40ed3dSBarry Smith   PetscFunctionBegin;
47820700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4783639f9d9dSBarry Smith   ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
47841cee3971SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
478535f5d045SPeter Brune   if (snes->linesearch) {
47867601faf0SJed Brown     ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr);
478708b6c495SPeter Brune     ierr = PetscObjectSetOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr);
478835f5d045SPeter Brune   }
478935f5d045SPeter Brune   ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
47903a40ed3dSBarry Smith   PetscFunctionReturn(0);
47913c7409f5SSatish Balay }
47923c7409f5SSatish Balay 
47933c7409f5SSatish Balay /*@C
4794f525115eSLois Curfman McInnes    SNESAppendOptionsPrefix - Appends to the prefix used for searching for all
4795d850072dSLois Curfman McInnes    SNES options in the database.
47963c7409f5SSatish Balay 
47973f9fe445SBarry Smith    Logically Collective on SNES
4798fee21e36SBarry Smith 
4799c7afd0dbSLois Curfman McInnes    Input Parameters:
4800c7afd0dbSLois Curfman McInnes +  snes - the SNES context
4801c7afd0dbSLois Curfman McInnes -  prefix - the prefix to prepend to all option names
4802c7afd0dbSLois Curfman McInnes 
4803d850072dSLois Curfman McInnes    Notes:
4804a83b1b31SSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
4805c7afd0dbSLois Curfman McInnes    The first character of all runtime options is AUTOMATICALLY the hyphen.
4806d850072dSLois Curfman McInnes 
480736851e7fSLois Curfman McInnes    Level: advanced
480836851e7fSLois Curfman McInnes 
48093c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database
4810a86d99e1SLois Curfman McInnes 
4811a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix()
48123c7409f5SSatish Balay @*/
48137087cfbeSBarry Smith PetscErrorCode  SNESAppendOptionsPrefix(SNES snes,const char prefix[])
48143c7409f5SSatish Balay {
4815dfbe8321SBarry Smith   PetscErrorCode ierr;
48163c7409f5SSatish Balay 
48173a40ed3dSBarry Smith   PetscFunctionBegin;
48180700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4819639f9d9dSBarry Smith   ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
48201cee3971SBarry Smith   if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);}
482135f5d045SPeter Brune   if (snes->linesearch) {
48227601faf0SJed Brown     ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr);
482308b6c495SPeter Brune     ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr);
482435f5d045SPeter Brune   }
482535f5d045SPeter Brune   ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr);
48263a40ed3dSBarry Smith   PetscFunctionReturn(0);
48273c7409f5SSatish Balay }
48283c7409f5SSatish Balay 
48299ab63eb5SSatish Balay /*@C
48303c7409f5SSatish Balay    SNESGetOptionsPrefix - Sets the prefix used for searching for all
48313c7409f5SSatish Balay    SNES options in the database.
48323c7409f5SSatish Balay 
4833c7afd0dbSLois Curfman McInnes    Not Collective
4834c7afd0dbSLois Curfman McInnes 
48353c7409f5SSatish Balay    Input Parameter:
48363c7409f5SSatish Balay .  snes - the SNES context
48373c7409f5SSatish Balay 
48383c7409f5SSatish Balay    Output Parameter:
48393c7409f5SSatish Balay .  prefix - pointer to the prefix string used
48403c7409f5SSatish Balay 
484195452b02SPatrick Sanan    Notes:
484295452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
48439ab63eb5SSatish Balay    sufficient length to hold the prefix.
48449ab63eb5SSatish Balay 
484536851e7fSLois Curfman McInnes    Level: advanced
484636851e7fSLois Curfman McInnes 
48473c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database
4848a86d99e1SLois Curfman McInnes 
4849a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix()
48503c7409f5SSatish Balay @*/
48517087cfbeSBarry Smith PetscErrorCode  SNESGetOptionsPrefix(SNES snes,const char *prefix[])
48523c7409f5SSatish Balay {
4853dfbe8321SBarry Smith   PetscErrorCode ierr;
48543c7409f5SSatish Balay 
48553a40ed3dSBarry Smith   PetscFunctionBegin;
48560700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4857639f9d9dSBarry Smith   ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr);
48583a40ed3dSBarry Smith   PetscFunctionReturn(0);
48593c7409f5SSatish Balay }
48603c7409f5SSatish Balay 
4861b2002411SLois Curfman McInnes 
48623cea93caSBarry Smith /*@C
48631c84c290SBarry Smith   SNESRegister - Adds a method to the nonlinear solver package.
48641c84c290SBarry Smith 
48651c84c290SBarry Smith    Not collective
48661c84c290SBarry Smith 
48671c84c290SBarry Smith    Input Parameters:
48681c84c290SBarry Smith +  name_solver - name of a new user-defined solver
48691c84c290SBarry Smith -  routine_create - routine to create method context
48701c84c290SBarry Smith 
48711c84c290SBarry Smith    Notes:
48721c84c290SBarry Smith    SNESRegister() may be called multiple times to add several user-defined solvers.
48731c84c290SBarry Smith 
48741c84c290SBarry Smith    Sample usage:
48751c84c290SBarry Smith .vb
4876bdf89e91SBarry Smith    SNESRegister("my_solver",MySolverCreate);
48771c84c290SBarry Smith .ve
48781c84c290SBarry Smith 
48791c84c290SBarry Smith    Then, your solver can be chosen with the procedural interface via
48801c84c290SBarry Smith $     SNESSetType(snes,"my_solver")
48811c84c290SBarry Smith    or at runtime via the option
48821c84c290SBarry Smith $     -snes_type my_solver
48831c84c290SBarry Smith 
48841c84c290SBarry Smith    Level: advanced
48851c84c290SBarry Smith 
48861c84c290SBarry Smith     Note: If your function is not being put into a shared library then use SNESRegister() instead
48871c84c290SBarry Smith 
48881c84c290SBarry Smith .keywords: SNES, nonlinear, register
48891c84c290SBarry Smith 
48901c84c290SBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy()
48913cea93caSBarry Smith 
48927f6c08e0SMatthew Knepley   Level: advanced
48933cea93caSBarry Smith @*/
4894bdf89e91SBarry Smith PetscErrorCode  SNESRegister(const char sname[],PetscErrorCode (*function)(SNES))
4895b2002411SLois Curfman McInnes {
4896dfbe8321SBarry Smith   PetscErrorCode ierr;
4897b2002411SLois Curfman McInnes 
4898b2002411SLois Curfman McInnes   PetscFunctionBegin;
48991d36bdfdSBarry Smith   ierr = SNESInitializePackage();CHKERRQ(ierr);
4900a240a19fSJed Brown   ierr = PetscFunctionListAdd(&SNESList,sname,function);CHKERRQ(ierr);
4901b2002411SLois Curfman McInnes   PetscFunctionReturn(0);
4902b2002411SLois Curfman McInnes }
4903da9b6338SBarry Smith 
49047087cfbeSBarry Smith PetscErrorCode  SNESTestLocalMin(SNES snes)
4905da9b6338SBarry Smith {
4906dfbe8321SBarry Smith   PetscErrorCode ierr;
490777431f27SBarry Smith   PetscInt       N,i,j;
4908da9b6338SBarry Smith   Vec            u,uh,fh;
4909da9b6338SBarry Smith   PetscScalar    value;
4910da9b6338SBarry Smith   PetscReal      norm;
4911da9b6338SBarry Smith 
4912da9b6338SBarry Smith   PetscFunctionBegin;
4913da9b6338SBarry Smith   ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr);
4914da9b6338SBarry Smith   ierr = VecDuplicate(u,&uh);CHKERRQ(ierr);
4915da9b6338SBarry Smith   ierr = VecDuplicate(u,&fh);CHKERRQ(ierr);
4916da9b6338SBarry Smith 
4917da9b6338SBarry Smith   /* currently only works for sequential */
491822d28d08SBarry Smith   ierr = PetscPrintf(PETSC_COMM_WORLD,"Testing FormFunction() for local min\n");CHKERRQ(ierr);
4919da9b6338SBarry Smith   ierr = VecGetSize(u,&N);CHKERRQ(ierr);
4920da9b6338SBarry Smith   for (i=0; i<N; i++) {
4921da9b6338SBarry Smith     ierr = VecCopy(u,uh);CHKERRQ(ierr);
492277431f27SBarry Smith     ierr = PetscPrintf(PETSC_COMM_WORLD,"i = %D\n",i);CHKERRQ(ierr);
4923da9b6338SBarry Smith     for (j=-10; j<11; j++) {
49248b49ba18SBarry Smith       value = PetscSign(j)*PetscExpReal(PetscAbs(j)-10.0);
4925da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
49263ab0aad5SBarry Smith       ierr  = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr);
4927da9b6338SBarry Smith       ierr  = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr);
492877431f27SBarry Smith       ierr  = PetscPrintf(PETSC_COMM_WORLD,"       j norm %D %18.16e\n",j,norm);CHKERRQ(ierr);
4929da9b6338SBarry Smith       value = -value;
4930da9b6338SBarry Smith       ierr  = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr);
4931da9b6338SBarry Smith     }
4932da9b6338SBarry Smith   }
49336bf464f9SBarry Smith   ierr = VecDestroy(&uh);CHKERRQ(ierr);
49346bf464f9SBarry Smith   ierr = VecDestroy(&fh);CHKERRQ(ierr);
4935da9b6338SBarry Smith   PetscFunctionReturn(0);
4936da9b6338SBarry Smith }
493771f87433Sdalcinl 
493871f87433Sdalcinl /*@
4939fa9f3622SBarry Smith    SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for
494071f87433Sdalcinl    computing relative tolerance for linear solvers within an inexact
494171f87433Sdalcinl    Newton method.
494271f87433Sdalcinl 
49433f9fe445SBarry Smith    Logically Collective on SNES
494471f87433Sdalcinl 
494571f87433Sdalcinl    Input Parameters:
494671f87433Sdalcinl +  snes - SNES context
494771f87433Sdalcinl -  flag - PETSC_TRUE or PETSC_FALSE
494871f87433Sdalcinl 
494964ba62caSBarry Smith     Options Database:
495064ba62caSBarry Smith +  -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence
495164ba62caSBarry Smith .  -snes_ksp_ew_version ver - version of  Eisenstat-Walker method
495264ba62caSBarry Smith .  -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0
495364ba62caSBarry Smith .  -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax
495464ba62caSBarry Smith .  -snes_ksp_ew_gamma <gamma> - Sets gamma
495564ba62caSBarry Smith .  -snes_ksp_ew_alpha <alpha> - Sets alpha
495664ba62caSBarry Smith .  -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2
495764ba62caSBarry Smith -  -snes_ksp_ew_threshold <threshold> - Sets threshold
495864ba62caSBarry Smith 
495971f87433Sdalcinl    Notes:
496071f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
496171f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
496271f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
496371f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
496471f87433Sdalcinl    solver.
496571f87433Sdalcinl 
496671f87433Sdalcinl    Level: advanced
496771f87433Sdalcinl 
496871f87433Sdalcinl    Reference:
496971f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
497071f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
497171f87433Sdalcinl 
497271f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
497371f87433Sdalcinl 
4974fa9f3622SBarry Smith .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
497571f87433Sdalcinl @*/
49767087cfbeSBarry Smith PetscErrorCode  SNESKSPSetUseEW(SNES snes,PetscBool flag)
497771f87433Sdalcinl {
497871f87433Sdalcinl   PetscFunctionBegin;
49790700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
4980acfcf0e5SJed Brown   PetscValidLogicalCollectiveBool(snes,flag,2);
498171f87433Sdalcinl   snes->ksp_ewconv = flag;
498271f87433Sdalcinl   PetscFunctionReturn(0);
498371f87433Sdalcinl }
498471f87433Sdalcinl 
498571f87433Sdalcinl /*@
4986fa9f3622SBarry Smith    SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method
498771f87433Sdalcinl    for computing relative tolerance for linear solvers within an
498871f87433Sdalcinl    inexact Newton method.
498971f87433Sdalcinl 
499071f87433Sdalcinl    Not Collective
499171f87433Sdalcinl 
499271f87433Sdalcinl    Input Parameter:
499371f87433Sdalcinl .  snes - SNES context
499471f87433Sdalcinl 
499571f87433Sdalcinl    Output Parameter:
499671f87433Sdalcinl .  flag - PETSC_TRUE or PETSC_FALSE
499771f87433Sdalcinl 
499871f87433Sdalcinl    Notes:
499971f87433Sdalcinl    Currently, the default is to use a constant relative tolerance for
500071f87433Sdalcinl    the inner linear solvers.  Alternatively, one can use the
500171f87433Sdalcinl    Eisenstat-Walker method, where the relative convergence tolerance
500271f87433Sdalcinl    is reset at each Newton iteration according progress of the nonlinear
500371f87433Sdalcinl    solver.
500471f87433Sdalcinl 
500571f87433Sdalcinl    Level: advanced
500671f87433Sdalcinl 
500771f87433Sdalcinl    Reference:
500871f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
500971f87433Sdalcinl    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
501071f87433Sdalcinl 
501171f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton
501271f87433Sdalcinl 
5013fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW()
501471f87433Sdalcinl @*/
50157087cfbeSBarry Smith PetscErrorCode  SNESKSPGetUseEW(SNES snes, PetscBool  *flag)
501671f87433Sdalcinl {
501771f87433Sdalcinl   PetscFunctionBegin;
50180700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
501971f87433Sdalcinl   PetscValidPointer(flag,2);
502071f87433Sdalcinl   *flag = snes->ksp_ewconv;
502171f87433Sdalcinl   PetscFunctionReturn(0);
502271f87433Sdalcinl }
502371f87433Sdalcinl 
502471f87433Sdalcinl /*@
5025fa9f3622SBarry Smith    SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker
502671f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
502771f87433Sdalcinl    Newton method.
502871f87433Sdalcinl 
50293f9fe445SBarry Smith    Logically Collective on SNES
503071f87433Sdalcinl 
503171f87433Sdalcinl    Input Parameters:
503271f87433Sdalcinl +    snes - SNES context
503371f87433Sdalcinl .    version - version 1, 2 (default is 2) or 3
503471f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
503571f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
503671f87433Sdalcinl .    gamma - multiplicative factor for version 2 rtol computation
503771f87433Sdalcinl              (0 <= gamma2 <= 1)
503871f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
503971f87433Sdalcinl .    alpha2 - power for safeguard
504071f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
504171f87433Sdalcinl 
504271f87433Sdalcinl    Note:
504371f87433Sdalcinl    Version 3 was contributed by Luis Chacon, June 2006.
504471f87433Sdalcinl 
504571f87433Sdalcinl    Use PETSC_DEFAULT to retain the default for any of the parameters.
504671f87433Sdalcinl 
504771f87433Sdalcinl    Level: advanced
504871f87433Sdalcinl 
504971f87433Sdalcinl    Reference:
505071f87433Sdalcinl    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
505171f87433Sdalcinl    inexact Newton method", Utah State University Math. Stat. Dept. Res.
505271f87433Sdalcinl    Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput.
505371f87433Sdalcinl 
505471f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, set, parameters
505571f87433Sdalcinl 
5056fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW()
505771f87433Sdalcinl @*/
5058f5af7f23SKarl Rupp PetscErrorCode  SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold)
505971f87433Sdalcinl {
5060fa9f3622SBarry Smith   SNESKSPEW *kctx;
50615fd66863SKarl Rupp 
506271f87433Sdalcinl   PetscFunctionBegin;
50630700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5064fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
5065e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
5066c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(snes,version,2);
5067c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol_0,3);
5068c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,rtol_max,4);
5069c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,gamma,5);
5070c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,alpha,6);
5071c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,alpha2,7);
5072c5eb9154SBarry Smith   PetscValidLogicalCollectiveReal(snes,threshold,8);
507371f87433Sdalcinl 
507471f87433Sdalcinl   if (version != PETSC_DEFAULT)   kctx->version   = version;
507571f87433Sdalcinl   if (rtol_0 != PETSC_DEFAULT)    kctx->rtol_0    = rtol_0;
507671f87433Sdalcinl   if (rtol_max != PETSC_DEFAULT)  kctx->rtol_max  = rtol_max;
507771f87433Sdalcinl   if (gamma != PETSC_DEFAULT)     kctx->gamma     = gamma;
507871f87433Sdalcinl   if (alpha != PETSC_DEFAULT)     kctx->alpha     = alpha;
507971f87433Sdalcinl   if (alpha2 != PETSC_DEFAULT)    kctx->alpha2    = alpha2;
508071f87433Sdalcinl   if (threshold != PETSC_DEFAULT) kctx->threshold = threshold;
508171f87433Sdalcinl 
5082f23aa3ddSBarry 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);
508357622a8eSBarry 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);
508457622a8eSBarry 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);
508557622a8eSBarry 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);
508657622a8eSBarry 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);
508757622a8eSBarry 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);
508871f87433Sdalcinl   PetscFunctionReturn(0);
508971f87433Sdalcinl }
509071f87433Sdalcinl 
509171f87433Sdalcinl /*@
5092fa9f3622SBarry Smith    SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker
509371f87433Sdalcinl    convergence criteria for the linear solvers within an inexact
509471f87433Sdalcinl    Newton method.
509571f87433Sdalcinl 
509671f87433Sdalcinl    Not Collective
509771f87433Sdalcinl 
509871f87433Sdalcinl    Input Parameters:
509971f87433Sdalcinl      snes - SNES context
510071f87433Sdalcinl 
510171f87433Sdalcinl    Output Parameters:
510271f87433Sdalcinl +    version - version 1, 2 (default is 2) or 3
510371f87433Sdalcinl .    rtol_0 - initial relative tolerance (0 <= rtol_0 < 1)
510471f87433Sdalcinl .    rtol_max - maximum relative tolerance (0 <= rtol_max < 1)
5105bf388a1fSBarry Smith .    gamma - multiplicative factor for version 2 rtol computation (0 <= gamma2 <= 1)
510671f87433Sdalcinl .    alpha - power for version 2 rtol computation (1 < alpha <= 2)
510771f87433Sdalcinl .    alpha2 - power for safeguard
510871f87433Sdalcinl -    threshold - threshold for imposing safeguard (0 < threshold < 1)
510971f87433Sdalcinl 
511071f87433Sdalcinl    Level: advanced
511171f87433Sdalcinl 
511271f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, get, parameters
511371f87433Sdalcinl 
5114fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW()
511571f87433Sdalcinl @*/
5116bf388a1fSBarry Smith PetscErrorCode  SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold)
511771f87433Sdalcinl {
5118fa9f3622SBarry Smith   SNESKSPEW *kctx;
51195fd66863SKarl Rupp 
512071f87433Sdalcinl   PetscFunctionBegin;
51210700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5122fa9f3622SBarry Smith   kctx = (SNESKSPEW*)snes->kspconvctx;
5123e32f2f54SBarry Smith   if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing");
512471f87433Sdalcinl   if (version)   *version   = kctx->version;
512571f87433Sdalcinl   if (rtol_0)    *rtol_0    = kctx->rtol_0;
512671f87433Sdalcinl   if (rtol_max)  *rtol_max  = kctx->rtol_max;
512771f87433Sdalcinl   if (gamma)     *gamma     = kctx->gamma;
512871f87433Sdalcinl   if (alpha)     *alpha     = kctx->alpha;
512971f87433Sdalcinl   if (alpha2)    *alpha2    = kctx->alpha2;
513071f87433Sdalcinl   if (threshold) *threshold = kctx->threshold;
513171f87433Sdalcinl   PetscFunctionReturn(0);
513271f87433Sdalcinl }
513371f87433Sdalcinl 
5134d5378b5fSDmitry Karpeev  PetscErrorCode KSPPreSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes)
513571f87433Sdalcinl {
513671f87433Sdalcinl   PetscErrorCode ierr;
5137fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
513871f87433Sdalcinl   PetscReal      rtol  = PETSC_DEFAULT,stol;
513971f87433Sdalcinl 
514071f87433Sdalcinl   PetscFunctionBegin;
5141d4211eb9SBarry Smith   if (!snes->ksp_ewconv) PetscFunctionReturn(0);
514230058271SDmitry Karpeev   if (!snes->iter) {
514330058271SDmitry Karpeev     rtol = kctx->rtol_0; /* first time in, so use the original user rtol */
514430058271SDmitry Karpeev     ierr = VecNorm(snes->vec_func,NORM_2,&kctx->norm_first);CHKERRQ(ierr);
514530058271SDmitry Karpeev   }
5146f5af7f23SKarl Rupp   else {
514771f87433Sdalcinl     if (kctx->version == 1) {
514871f87433Sdalcinl       rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last;
514971f87433Sdalcinl       if (rtol < 0.0) rtol = -rtol;
515085ec1a3cSBarry Smith       stol = PetscPowReal(kctx->rtol_last,kctx->alpha2);
515171f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
515271f87433Sdalcinl     } else if (kctx->version == 2) {
515385ec1a3cSBarry Smith       rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha);
515485ec1a3cSBarry Smith       stol = kctx->gamma * PetscPowReal(kctx->rtol_last,kctx->alpha);
515571f87433Sdalcinl       if (stol > kctx->threshold) rtol = PetscMax(rtol,stol);
515671f87433Sdalcinl     } else if (kctx->version == 3) { /* contributed by Luis Chacon, June 2006. */
515785ec1a3cSBarry Smith       rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha);
515871f87433Sdalcinl       /* safeguard: avoid sharp decrease of rtol */
515985ec1a3cSBarry Smith       stol = kctx->gamma*PetscPowReal(kctx->rtol_last,kctx->alpha);
516071f87433Sdalcinl       stol = PetscMax(rtol,stol);
516171f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
516271f87433Sdalcinl       /* safeguard: avoid oversolving */
516330058271SDmitry Karpeev       stol = kctx->gamma*(kctx->norm_first*snes->rtol)/snes->norm;
516471f87433Sdalcinl       stol = PetscMax(rtol,stol);
516571f87433Sdalcinl       rtol = PetscMin(kctx->rtol_0,stol);
5166e32f2f54SBarry Smith     } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version);
516771f87433Sdalcinl   }
516871f87433Sdalcinl   /* safeguard: avoid rtol greater than one */
516971f87433Sdalcinl   rtol = PetscMin(rtol,kctx->rtol_max);
517071f87433Sdalcinl   ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr);
517157622a8eSBarry Smith   ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%g\n",snes->iter,kctx->version,(double)rtol);CHKERRQ(ierr);
517271f87433Sdalcinl   PetscFunctionReturn(0);
517371f87433Sdalcinl }
517471f87433Sdalcinl 
5175d5378b5fSDmitry Karpeev PetscErrorCode KSPPostSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes)
517671f87433Sdalcinl {
517771f87433Sdalcinl   PetscErrorCode ierr;
5178fa9f3622SBarry Smith   SNESKSPEW      *kctx = (SNESKSPEW*)snes->kspconvctx;
517971f87433Sdalcinl   PCSide         pcside;
518071f87433Sdalcinl   Vec            lres;
518171f87433Sdalcinl 
518271f87433Sdalcinl   PetscFunctionBegin;
5183d4211eb9SBarry Smith   if (!snes->ksp_ewconv) PetscFunctionReturn(0);
518471f87433Sdalcinl   ierr = KSPGetTolerances(ksp,&kctx->rtol_last,0,0,0);CHKERRQ(ierr);
518571dbe336SPeter Brune   kctx->norm_last = snes->norm;
518671f87433Sdalcinl   if (kctx->version == 1) {
51874f00ce20SMatthew G. Knepley     PC        pc;
51884f00ce20SMatthew G. Knepley     PetscBool isNone;
51894f00ce20SMatthew G. Knepley 
51904f00ce20SMatthew G. Knepley     ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr);
51914f00ce20SMatthew G. Knepley     ierr = PetscObjectTypeCompare((PetscObject) pc, PCNONE, &isNone);CHKERRQ(ierr);
5192b037da10SBarry Smith     ierr = KSPGetPCSide(ksp,&pcside);CHKERRQ(ierr);
51934f00ce20SMatthew G. Knepley      if (pcside == PC_RIGHT || isNone) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */
519471f87433Sdalcinl       /* KSP residual is true linear residual */
519571f87433Sdalcinl       ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr);
519671f87433Sdalcinl     } else {
519771f87433Sdalcinl       /* KSP residual is preconditioned residual */
519871f87433Sdalcinl       /* compute true linear residual norm */
519971f87433Sdalcinl       ierr = VecDuplicate(b,&lres);CHKERRQ(ierr);
520071f87433Sdalcinl       ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr);
520171f87433Sdalcinl       ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr);
520271f87433Sdalcinl       ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr);
52036bf464f9SBarry Smith       ierr = VecDestroy(&lres);CHKERRQ(ierr);
520471f87433Sdalcinl     }
520571f87433Sdalcinl   }
520671f87433Sdalcinl   PetscFunctionReturn(0);
520771f87433Sdalcinl }
520871f87433Sdalcinl 
5209d4211eb9SBarry Smith /*@
5210d4211eb9SBarry Smith    SNESGetKSP - Returns the KSP context for a SNES solver.
5211d4211eb9SBarry Smith 
5212d4211eb9SBarry Smith    Not Collective, but if SNES object is parallel, then KSP object is parallel
5213d4211eb9SBarry Smith 
5214d4211eb9SBarry Smith    Input Parameter:
5215d4211eb9SBarry Smith .  snes - the SNES context
5216d4211eb9SBarry Smith 
5217d4211eb9SBarry Smith    Output Parameter:
5218d4211eb9SBarry Smith .  ksp - the KSP context
5219d4211eb9SBarry Smith 
5220d4211eb9SBarry Smith    Notes:
5221d4211eb9SBarry Smith    The user can then directly manipulate the KSP context to set various
5222d4211eb9SBarry Smith    options, etc.  Likewise, the user can then extract and manipulate the
5223d4211eb9SBarry Smith    PC contexts as well.
5224d4211eb9SBarry Smith 
5225d4211eb9SBarry Smith    Level: beginner
5226d4211eb9SBarry Smith 
5227d4211eb9SBarry Smith .keywords: SNES, nonlinear, get, KSP, context
5228d4211eb9SBarry Smith 
5229d4211eb9SBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP()
5230d4211eb9SBarry Smith @*/
5231d4211eb9SBarry Smith PetscErrorCode  SNESGetKSP(SNES snes,KSP *ksp)
523271f87433Sdalcinl {
523371f87433Sdalcinl   PetscErrorCode ierr;
523471f87433Sdalcinl 
523571f87433Sdalcinl   PetscFunctionBegin;
5236d4211eb9SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5237d4211eb9SBarry Smith   PetscValidPointer(ksp,2);
5238d4211eb9SBarry Smith 
5239d4211eb9SBarry Smith   if (!snes->ksp) {
5240a5c2985bSBarry Smith     PetscBool monitor = PETSC_FALSE;
5241a5c2985bSBarry Smith 
5242d4211eb9SBarry Smith     ierr = KSPCreate(PetscObjectComm((PetscObject)snes),&snes->ksp);CHKERRQ(ierr);
5243d4211eb9SBarry Smith     ierr = PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1);CHKERRQ(ierr);
52443bb1ff40SBarry Smith     ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->ksp);CHKERRQ(ierr);
5245d4211eb9SBarry Smith 
5246d5378b5fSDmitry Karpeev     ierr = KSPSetPreSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_SNESEW,snes);CHKERRQ(ierr);
5247d5378b5fSDmitry Karpeev     ierr = KSPSetPostSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_SNESEW,snes);CHKERRQ(ierr);
5248a5c2985bSBarry Smith 
5249c5929fdfSBarry Smith     ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes",&monitor,NULL);CHKERRQ(ierr);
5250a5c2985bSBarry Smith     if (monitor) {
5251a5c2985bSBarry Smith       ierr = KSPMonitorSet(snes->ksp,KSPMonitorSNES,snes,NULL);CHKERRQ(ierr);
5252a5c2985bSBarry Smith     }
5253e5f7ee39SBarry Smith     monitor = PETSC_FALSE;
5254c5929fdfSBarry Smith     ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes_lg",&monitor,NULL);CHKERRQ(ierr);
5255e5f7ee39SBarry Smith     if (monitor) {
5256e5f7ee39SBarry Smith       PetscObject *objs;
52578b0b5a47SLisandro Dalcin       ierr = KSPMonitorSNESLGResidualNormCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,600,600,&objs);CHKERRQ(ierr);
5258e5f7ee39SBarry Smith       objs[0] = (PetscObject) snes;
5259e5f7ee39SBarry Smith       ierr = KSPMonitorSet(snes->ksp,(PetscErrorCode (*)(KSP,PetscInt,PetscReal,void*))KSPMonitorSNESLGResidualNorm,objs,(PetscErrorCode (*)(void**))KSPMonitorSNESLGResidualNormDestroy);CHKERRQ(ierr);
5260e5f7ee39SBarry Smith     }
526116413a6aSBarry Smith     ierr = PetscObjectSetOptions((PetscObject)snes->ksp,((PetscObject)snes)->options);CHKERRQ(ierr);
5262d4211eb9SBarry Smith   }
5263d4211eb9SBarry Smith   *ksp = snes->ksp;
526471f87433Sdalcinl   PetscFunctionReturn(0);
526571f87433Sdalcinl }
52666c699258SBarry Smith 
5267d4211eb9SBarry Smith 
5268af0996ceSBarry Smith #include <petsc/private/dmimpl.h>
52696c699258SBarry Smith /*@
52702a808120SBarry Smith    SNESSetDM - Sets the DM that may be used by some nonlinear solvers or their underlying preconditioners
52716c699258SBarry Smith 
52723f9fe445SBarry Smith    Logically Collective on SNES
52736c699258SBarry Smith 
52746c699258SBarry Smith    Input Parameters:
52752a808120SBarry Smith +  snes - the nonlinear solver context
52762a808120SBarry Smith -  dm - the dm, cannot be NULL
52776c699258SBarry Smith 
5278e03a659cSJed Brown    Notes:
5279e03a659cSJed Brown    A DM can only be used for solving one problem at a time because information about the problem is stored on the DM,
5280e03a659cSJed Brown    even when not using interfaces like DMSNESSetFunction().  Use DMClone() to get a distinct DM when solving different
5281e03a659cSJed Brown    problems using the same function space.
5282e03a659cSJed Brown 
52836c699258SBarry Smith    Level: intermediate
52846c699258SBarry Smith 
52854c2026ceSFande Kong .seealso: SNESGetDM(), KSPSetDM(), KSPGetDM()
52866c699258SBarry Smith @*/
52877087cfbeSBarry Smith PetscErrorCode  SNESSetDM(SNES snes,DM dm)
52886c699258SBarry Smith {
52896c699258SBarry Smith   PetscErrorCode ierr;
5290345fed2cSBarry Smith   KSP            ksp;
5291942e3340SBarry Smith   DMSNES         sdm;
52926c699258SBarry Smith 
52936c699258SBarry Smith   PetscFunctionBegin;
52940700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
52952a808120SBarry Smith   PetscValidHeaderSpecific(dm,DM_CLASSID,2);
52962a808120SBarry Smith   ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr);
5297942e3340SBarry Smith   if (snes->dm) {               /* Move the DMSNES context over to the new DM unless the new DM already has one */
529851f4b3c7SToby Isaac     if (snes->dm->dmsnes && !dm->dmsnes) {
5299942e3340SBarry Smith       ierr = DMCopyDMSNES(snes->dm,dm);CHKERRQ(ierr);
5300942e3340SBarry Smith       ierr = DMGetDMSNES(snes->dm,&sdm);CHKERRQ(ierr);
5301f5af7f23SKarl Rupp       if (sdm->originaldm == snes->dm) sdm->originaldm = dm; /* Grant write privileges to the replacement DM */
53026cab3a1bSJed Brown     }
5303dc822a44SJed Brown     ierr = DMCoarsenHookRemove(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr);
53046bf464f9SBarry Smith     ierr = DMDestroy(&snes->dm);CHKERRQ(ierr);
53056cab3a1bSJed Brown   }
53066c699258SBarry Smith   snes->dm     = dm;
5307116d1032SJed Brown   snes->dmAuto = PETSC_FALSE;
5308f5af7f23SKarl Rupp 
5309345fed2cSBarry Smith   ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr);
5310345fed2cSBarry Smith   ierr = KSPSetDM(ksp,dm);CHKERRQ(ierr);
5311f22f69f0SBarry Smith   ierr = KSPSetDMActive(ksp,PETSC_FALSE);CHKERRQ(ierr);
5312efd4aadfSBarry Smith   if (snes->npc) {
5313efd4aadfSBarry Smith     ierr = SNESSetDM(snes->npc, snes->dm);CHKERRQ(ierr);
5314efd4aadfSBarry Smith     ierr = SNESSetNPCSide(snes,snes->npcside);CHKERRQ(ierr);
53152c155ee1SBarry Smith   }
53166c699258SBarry Smith   PetscFunctionReturn(0);
53176c699258SBarry Smith }
53186c699258SBarry Smith 
53196c699258SBarry Smith /*@
53206c699258SBarry Smith    SNESGetDM - Gets the DM that may be used by some preconditioners
53216c699258SBarry Smith 
53223f9fe445SBarry Smith    Not Collective but DM obtained is parallel on SNES
53236c699258SBarry Smith 
53246c699258SBarry Smith    Input Parameter:
53256c699258SBarry Smith . snes - the preconditioner context
53266c699258SBarry Smith 
53276c699258SBarry Smith    Output Parameter:
53286c699258SBarry Smith .  dm - the dm
53296c699258SBarry Smith 
53306c699258SBarry Smith    Level: intermediate
53316c699258SBarry Smith 
53324c2026ceSFande Kong .seealso: SNESSetDM(), KSPSetDM(), KSPGetDM()
53336c699258SBarry Smith @*/
53347087cfbeSBarry Smith PetscErrorCode  SNESGetDM(SNES snes,DM *dm)
53356c699258SBarry Smith {
53366cab3a1bSJed Brown   PetscErrorCode ierr;
53376cab3a1bSJed Brown 
53386c699258SBarry Smith   PetscFunctionBegin;
53390700a824SBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
53406cab3a1bSJed Brown   if (!snes->dm) {
5341ce94432eSBarry Smith     ierr         = DMShellCreate(PetscObjectComm((PetscObject)snes),&snes->dm);CHKERRQ(ierr);
5342116d1032SJed Brown     snes->dmAuto = PETSC_TRUE;
53436cab3a1bSJed Brown   }
53446c699258SBarry Smith   *dm = snes->dm;
53456c699258SBarry Smith   PetscFunctionReturn(0);
53466c699258SBarry Smith }
53470807856dSBarry Smith 
534831823bd8SMatthew G Knepley /*@
5349be95d8f1SBarry Smith   SNESSetNPC - Sets the nonlinear preconditioner to be used.
535031823bd8SMatthew G Knepley 
535131823bd8SMatthew G Knepley   Collective on SNES
535231823bd8SMatthew G Knepley 
535331823bd8SMatthew G Knepley   Input Parameters:
535431823bd8SMatthew G Knepley + snes - iterative context obtained from SNESCreate()
535531823bd8SMatthew G Knepley - pc   - the preconditioner object
535631823bd8SMatthew G Knepley 
535731823bd8SMatthew G Knepley   Notes:
5358be95d8f1SBarry Smith   Use SNESGetNPC() to retrieve the preconditioner context (for example,
535931823bd8SMatthew G Knepley   to configure it using the API).
536031823bd8SMatthew G Knepley 
536131823bd8SMatthew G Knepley   Level: developer
536231823bd8SMatthew G Knepley 
536331823bd8SMatthew G Knepley .keywords: SNES, set, precondition
53643ad1a0b9SPatrick Farrell .seealso: SNESGetNPC(), SNESHasNPC()
536531823bd8SMatthew G Knepley @*/
5366be95d8f1SBarry Smith PetscErrorCode SNESSetNPC(SNES snes, SNES pc)
536731823bd8SMatthew G Knepley {
536831823bd8SMatthew G Knepley   PetscErrorCode ierr;
536931823bd8SMatthew G Knepley 
537031823bd8SMatthew G Knepley   PetscFunctionBegin;
537131823bd8SMatthew G Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
537231823bd8SMatthew G Knepley   PetscValidHeaderSpecific(pc, SNES_CLASSID, 2);
537331823bd8SMatthew G Knepley   PetscCheckSameComm(snes, 1, pc, 2);
537431823bd8SMatthew G Knepley   ierr     = PetscObjectReference((PetscObject) pc);CHKERRQ(ierr);
5375efd4aadfSBarry Smith   ierr     = SNESDestroy(&snes->npc);CHKERRQ(ierr);
5376efd4aadfSBarry Smith   snes->npc = pc;
5377efd4aadfSBarry Smith   ierr     = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->npc);CHKERRQ(ierr);
537831823bd8SMatthew G Knepley   PetscFunctionReturn(0);
537931823bd8SMatthew G Knepley }
538031823bd8SMatthew G Knepley 
538131823bd8SMatthew G Knepley /*@
5382be95d8f1SBarry Smith   SNESGetNPC - Creates a nonlinear preconditioning solver (SNES) to be used to precondition the nonlinear solver.
538331823bd8SMatthew G Knepley 
538431823bd8SMatthew G Knepley   Not Collective
538531823bd8SMatthew G Knepley 
538631823bd8SMatthew G Knepley   Input Parameter:
538731823bd8SMatthew G Knepley . snes - iterative context obtained from SNESCreate()
538831823bd8SMatthew G Knepley 
538931823bd8SMatthew G Knepley   Output Parameter:
539031823bd8SMatthew G Knepley . pc - preconditioner context
539131823bd8SMatthew G Knepley 
539295452b02SPatrick Sanan   Notes:
539395452b02SPatrick Sanan     If a SNES was previously set with SNESSetNPC() then that SNES is returned.
5394be95d8f1SBarry Smith 
539531823bd8SMatthew G Knepley   Level: developer
539631823bd8SMatthew G Knepley 
539731823bd8SMatthew G Knepley .keywords: SNES, get, preconditioner
53983ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESHasNPC()
539931823bd8SMatthew G Knepley @*/
5400be95d8f1SBarry Smith PetscErrorCode SNESGetNPC(SNES snes, SNES *pc)
540131823bd8SMatthew G Knepley {
540231823bd8SMatthew G Knepley   PetscErrorCode ierr;
5403a64e098fSPeter Brune   const char     *optionsprefix;
540431823bd8SMatthew G Knepley 
540531823bd8SMatthew G Knepley   PetscFunctionBegin;
540631823bd8SMatthew G Knepley   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
540731823bd8SMatthew G Knepley   PetscValidPointer(pc, 2);
5408efd4aadfSBarry Smith   if (!snes->npc) {
5409efd4aadfSBarry Smith     ierr = SNESCreate(PetscObjectComm((PetscObject)snes),&snes->npc);CHKERRQ(ierr);
5410efd4aadfSBarry Smith     ierr = PetscObjectIncrementTabLevel((PetscObject)snes->npc,(PetscObject)snes,1);CHKERRQ(ierr);
5411efd4aadfSBarry Smith     ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr);
5412a64e098fSPeter Brune     ierr = SNESGetOptionsPrefix(snes,&optionsprefix);CHKERRQ(ierr);
5413efd4aadfSBarry Smith     ierr = SNESSetOptionsPrefix(snes->npc,optionsprefix);CHKERRQ(ierr);
5414efd4aadfSBarry Smith     ierr = SNESAppendOptionsPrefix(snes->npc,"npc_");CHKERRQ(ierr);
5415efd4aadfSBarry Smith     ierr = SNESSetCountersReset(snes->npc,PETSC_FALSE);CHKERRQ(ierr);
541631823bd8SMatthew G Knepley   }
5417efd4aadfSBarry Smith   *pc = snes->npc;
541831823bd8SMatthew G Knepley   PetscFunctionReturn(0);
541931823bd8SMatthew G Knepley }
542031823bd8SMatthew G Knepley 
54213ad1a0b9SPatrick Farrell /*@
54223ad1a0b9SPatrick Farrell   SNESHasNPC - Returns whether a nonlinear preconditioner exists
54233ad1a0b9SPatrick Farrell 
54243ad1a0b9SPatrick Farrell   Not Collective
54253ad1a0b9SPatrick Farrell 
54263ad1a0b9SPatrick Farrell   Input Parameter:
54273ad1a0b9SPatrick Farrell . snes - iterative context obtained from SNESCreate()
54283ad1a0b9SPatrick Farrell 
54293ad1a0b9SPatrick Farrell   Output Parameter:
54303ad1a0b9SPatrick Farrell . has_npc - whether the SNES has an NPC or not
54313ad1a0b9SPatrick Farrell 
54323ad1a0b9SPatrick Farrell   Level: developer
54333ad1a0b9SPatrick Farrell 
54343ad1a0b9SPatrick Farrell .keywords: SNES, has, preconditioner
54353ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESGetNPC()
54363ad1a0b9SPatrick Farrell @*/
54373ad1a0b9SPatrick Farrell PetscErrorCode SNESHasNPC(SNES snes, PetscBool *has_npc)
54383ad1a0b9SPatrick Farrell {
54393ad1a0b9SPatrick Farrell   PetscFunctionBegin;
54403ad1a0b9SPatrick Farrell   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5441efd4aadfSBarry Smith   *has_npc = (PetscBool) (snes->npc ? PETSC_TRUE : PETSC_FALSE);
54423ad1a0b9SPatrick Farrell   PetscFunctionReturn(0);
54433ad1a0b9SPatrick Farrell }
54443ad1a0b9SPatrick Farrell 
5445c40d0f55SPeter Brune /*@
5446be95d8f1SBarry Smith     SNESSetNPCSide - Sets the preconditioning side.
5447c40d0f55SPeter Brune 
5448c40d0f55SPeter Brune     Logically Collective on SNES
5449c40d0f55SPeter Brune 
5450c40d0f55SPeter Brune     Input Parameter:
5451c40d0f55SPeter Brune .   snes - iterative context obtained from SNESCreate()
5452c40d0f55SPeter Brune 
5453c40d0f55SPeter Brune     Output Parameter:
5454c40d0f55SPeter Brune .   side - the preconditioning side, where side is one of
5455c40d0f55SPeter Brune .vb
54562d547940SBarry Smith       PC_LEFT - left preconditioning
54572d547940SBarry Smith       PC_RIGHT - right preconditioning (default for most nonlinear solvers)
5458c40d0f55SPeter Brune .ve
5459c40d0f55SPeter Brune 
5460c40d0f55SPeter Brune     Options Database Keys:
5461c40d0f55SPeter Brune .   -snes_pc_side <right,left>
5462c40d0f55SPeter Brune 
546395452b02SPatrick Sanan     Notes:
546495452b02SPatrick Sanan     SNESNRICHARDSON and SNESNCG only support left preconditioning.
54652d547940SBarry Smith 
5466c40d0f55SPeter Brune     Level: intermediate
5467c40d0f55SPeter Brune 
5468c40d0f55SPeter Brune .keywords: SNES, set, right, left, side, preconditioner, flag
5469c40d0f55SPeter Brune 
5470be95d8f1SBarry Smith .seealso: SNESGetNPCSide(), KSPSetPCSide()
5471c40d0f55SPeter Brune @*/
5472be95d8f1SBarry Smith PetscErrorCode  SNESSetNPCSide(SNES snes,PCSide side)
5473c40d0f55SPeter Brune {
5474c40d0f55SPeter Brune   PetscFunctionBegin;
5475c40d0f55SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5476c40d0f55SPeter Brune   PetscValidLogicalCollectiveEnum(snes,side,2);
5477efd4aadfSBarry Smith   snes->npcside= side;
5478c40d0f55SPeter Brune   PetscFunctionReturn(0);
5479c40d0f55SPeter Brune }
5480c40d0f55SPeter Brune 
5481c40d0f55SPeter Brune /*@
5482be95d8f1SBarry Smith     SNESGetNPCSide - Gets the preconditioning side.
5483c40d0f55SPeter Brune 
5484c40d0f55SPeter Brune     Not Collective
5485c40d0f55SPeter Brune 
5486c40d0f55SPeter Brune     Input Parameter:
5487c40d0f55SPeter Brune .   snes - iterative context obtained from SNESCreate()
5488c40d0f55SPeter Brune 
5489c40d0f55SPeter Brune     Output Parameter:
5490c40d0f55SPeter Brune .   side - the preconditioning side, where side is one of
5491c40d0f55SPeter Brune .vb
54922d547940SBarry Smith       PC_LEFT - left preconditioning
54932d547940SBarry Smith       PC_RIGHT - right preconditioning (default for most nonlinear solvers)
5494c40d0f55SPeter Brune .ve
5495c40d0f55SPeter Brune 
5496c40d0f55SPeter Brune     Level: intermediate
5497c40d0f55SPeter Brune 
5498c40d0f55SPeter Brune .keywords: SNES, get, right, left, side, preconditioner, flag
5499c40d0f55SPeter Brune 
5500be95d8f1SBarry Smith .seealso: SNESSetNPCSide(), KSPGetPCSide()
5501c40d0f55SPeter Brune @*/
5502be95d8f1SBarry Smith PetscErrorCode  SNESGetNPCSide(SNES snes,PCSide *side)
5503c40d0f55SPeter Brune {
5504c40d0f55SPeter Brune   PetscFunctionBegin;
5505c40d0f55SPeter Brune   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5506c40d0f55SPeter Brune   PetscValidPointer(side,2);
5507efd4aadfSBarry Smith   *side = snes->npcside;
5508c40d0f55SPeter Brune   PetscFunctionReturn(0);
5509c40d0f55SPeter Brune }
5510c40d0f55SPeter Brune 
55119e764e56SPeter Brune /*@
55127601faf0SJed Brown   SNESSetLineSearch - Sets the linesearch on the SNES instance.
55139e764e56SPeter Brune 
55149e764e56SPeter Brune   Collective on SNES
55159e764e56SPeter Brune 
55169e764e56SPeter Brune   Input Parameters:
55179e764e56SPeter Brune + snes - iterative context obtained from SNESCreate()
55189e764e56SPeter Brune - linesearch   - the linesearch object
55199e764e56SPeter Brune 
55209e764e56SPeter Brune   Notes:
55217601faf0SJed Brown   Use SNESGetLineSearch() to retrieve the preconditioner context (for example,
55229e764e56SPeter Brune   to configure it using the API).
55239e764e56SPeter Brune 
55249e764e56SPeter Brune   Level: developer
55259e764e56SPeter Brune 
55269e764e56SPeter Brune .keywords: SNES, set, linesearch
55277601faf0SJed Brown .seealso: SNESGetLineSearch()
55289e764e56SPeter Brune @*/
55297601faf0SJed Brown PetscErrorCode SNESSetLineSearch(SNES snes, SNESLineSearch linesearch)
55309e764e56SPeter Brune {
55319e764e56SPeter Brune   PetscErrorCode ierr;
55329e764e56SPeter Brune 
55339e764e56SPeter Brune   PetscFunctionBegin;
55349e764e56SPeter Brune   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
5535f1c6b773SPeter Brune   PetscValidHeaderSpecific(linesearch, SNESLINESEARCH_CLASSID, 2);
55369e764e56SPeter Brune   PetscCheckSameComm(snes, 1, linesearch, 2);
55379e764e56SPeter Brune   ierr = PetscObjectReference((PetscObject) linesearch);CHKERRQ(ierr);
5538f1c6b773SPeter Brune   ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr);
5539f5af7f23SKarl Rupp 
55409e764e56SPeter Brune   snes->linesearch = linesearch;
5541f5af7f23SKarl Rupp 
55423bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr);
55439e764e56SPeter Brune   PetscFunctionReturn(0);
55449e764e56SPeter Brune }
55459e764e56SPeter Brune 
5546a34ceb2aSJed Brown /*@
55477601faf0SJed Brown   SNESGetLineSearch - Returns a pointer to the line search context set with SNESSetLineSearch()
55488141a3b9SPeter Brune   or creates a default line search instance associated with the SNES and returns it.
55499e764e56SPeter Brune 
55509e764e56SPeter Brune   Not Collective
55519e764e56SPeter Brune 
55529e764e56SPeter Brune   Input Parameter:
55539e764e56SPeter Brune . snes - iterative context obtained from SNESCreate()
55549e764e56SPeter Brune 
55559e764e56SPeter Brune   Output Parameter:
55569e764e56SPeter Brune . linesearch - linesearch context
55579e764e56SPeter Brune 
5558162e0bf5SPeter Brune   Level: beginner
55599e764e56SPeter Brune 
55609e764e56SPeter Brune .keywords: SNES, get, linesearch
5561162e0bf5SPeter Brune .seealso: SNESSetLineSearch(), SNESLineSearchCreate()
55629e764e56SPeter Brune @*/
55637601faf0SJed Brown PetscErrorCode SNESGetLineSearch(SNES snes, SNESLineSearch *linesearch)
55649e764e56SPeter Brune {
55659e764e56SPeter Brune   PetscErrorCode ierr;
55669e764e56SPeter Brune   const char     *optionsprefix;
55679e764e56SPeter Brune 
55689e764e56SPeter Brune   PetscFunctionBegin;
55699e764e56SPeter Brune   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
55709e764e56SPeter Brune   PetscValidPointer(linesearch, 2);
55719e764e56SPeter Brune   if (!snes->linesearch) {
55729e764e56SPeter Brune     ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr);
557382f516ccSBarry Smith     ierr = SNESLineSearchCreate(PetscObjectComm((PetscObject)snes), &snes->linesearch);CHKERRQ(ierr);
5574f1c6b773SPeter Brune     ierr = SNESLineSearchSetSNES(snes->linesearch, snes);CHKERRQ(ierr);
5575b1dca40bSPeter Brune     ierr = SNESLineSearchAppendOptionsPrefix(snes->linesearch, optionsprefix);CHKERRQ(ierr);
55769e764e56SPeter Brune     ierr = PetscObjectIncrementTabLevel((PetscObject) snes->linesearch, (PetscObject) snes, 1);CHKERRQ(ierr);
55773bb1ff40SBarry Smith     ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr);
55789e764e56SPeter Brune   }
55799e764e56SPeter Brune   *linesearch = snes->linesearch;
55809e764e56SPeter Brune   PetscFunctionReturn(0);
55819e764e56SPeter Brune }
55829e764e56SPeter Brune 
558369b4f73cSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
5584c6db04a5SJed Brown #include <mex.h>
558569b4f73cSBarry Smith 
55868f6e6473SBarry Smith typedef struct {char *funcname; mxArray *ctx;} SNESMatlabContext;
55878f6e6473SBarry Smith 
55880807856dSBarry Smith /*
5589bf388a1fSBarry Smith    SNESComputeFunction_Matlab - Calls the function that has been set with SNESSetFunctionMatlab().
55900807856dSBarry Smith 
55910807856dSBarry Smith    Collective on SNES
55920807856dSBarry Smith 
55930807856dSBarry Smith    Input Parameters:
55940807856dSBarry Smith +  snes - the SNES context
55950807856dSBarry Smith -  x - input vector
55960807856dSBarry Smith 
55970807856dSBarry Smith    Output Parameter:
55980807856dSBarry Smith .  y - function vector, as set by SNESSetFunction()
55990807856dSBarry Smith 
56000807856dSBarry Smith    Notes:
56010807856dSBarry Smith    SNESComputeFunction() is typically used within nonlinear solvers
56020807856dSBarry Smith    implementations, so most users would not generally call this routine
56030807856dSBarry Smith    themselves.
56040807856dSBarry Smith 
56050807856dSBarry Smith    Level: developer
56060807856dSBarry Smith 
56070807856dSBarry Smith .keywords: SNES, nonlinear, compute, function
56080807856dSBarry Smith 
56090807856dSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction()
561061b2408cSBarry Smith */
56117087cfbeSBarry Smith PetscErrorCode  SNESComputeFunction_Matlab(SNES snes,Vec x,Vec y, void *ctx)
56120807856dSBarry Smith {
5613e650e774SBarry Smith   PetscErrorCode    ierr;
56148f6e6473SBarry Smith   SNESMatlabContext *sctx = (SNESMatlabContext*)ctx;
56158f6e6473SBarry Smith   int               nlhs  = 1,nrhs = 5;
56168f6e6473SBarry Smith   mxArray           *plhs[1],*prhs[5];
561791621f2eSBarry Smith   long long int     lx = 0,ly = 0,ls = 0;
5618e650e774SBarry Smith 
56190807856dSBarry Smith   PetscFunctionBegin;
56200807856dSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
56210807856dSBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
56220807856dSBarry Smith   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
56230807856dSBarry Smith   PetscCheckSameComm(snes,1,x,2);
56240807856dSBarry Smith   PetscCheckSameComm(snes,1,y,3);
56250807856dSBarry Smith 
56260807856dSBarry Smith   /* call Matlab function in ctx with arguments x and y */
5627e650e774SBarry Smith 
562891621f2eSBarry Smith   ierr    = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr);
5629e650e774SBarry Smith   ierr    = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr);
5630e650e774SBarry Smith   ierr    = PetscMemcpy(&ly,&y,sizeof(x));CHKERRQ(ierr);
563191621f2eSBarry Smith   prhs[0] = mxCreateDoubleScalar((double)ls);
563291621f2eSBarry Smith   prhs[1] = mxCreateDoubleScalar((double)lx);
563391621f2eSBarry Smith   prhs[2] = mxCreateDoubleScalar((double)ly);
56348f6e6473SBarry Smith   prhs[3] = mxCreateString(sctx->funcname);
56358f6e6473SBarry Smith   prhs[4] = sctx->ctx;
5636b807a863SBarry Smith   ierr    = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeFunctionInternal");CHKERRQ(ierr);
5637e650e774SBarry Smith   ierr    = mxGetScalar(plhs[0]);CHKERRQ(ierr);
5638e650e774SBarry Smith   mxDestroyArray(prhs[0]);
5639e650e774SBarry Smith   mxDestroyArray(prhs[1]);
5640e650e774SBarry Smith   mxDestroyArray(prhs[2]);
56418f6e6473SBarry Smith   mxDestroyArray(prhs[3]);
5642e650e774SBarry Smith   mxDestroyArray(plhs[0]);
56430807856dSBarry Smith   PetscFunctionReturn(0);
56440807856dSBarry Smith }
56450807856dSBarry Smith 
564661b2408cSBarry Smith /*
56470807856dSBarry Smith    SNESSetFunctionMatlab - Sets the function evaluation routine and function
56480807856dSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
5649e3c5b3baSBarry Smith    equations from MATLAB. Here the function is a string containing the name of a MATLAB function
56500807856dSBarry Smith 
56510807856dSBarry Smith    Logically Collective on SNES
56520807856dSBarry Smith 
56530807856dSBarry Smith    Input Parameters:
56540807856dSBarry Smith +  snes - the SNES context
56550807856dSBarry Smith .  r - vector to store function value
5656f8b49ee9SBarry Smith -  f - function evaluation routine
56570807856dSBarry Smith 
56580807856dSBarry Smith    Notes:
56590807856dSBarry Smith    The Newton-like methods typically solve linear systems of the form
56600807856dSBarry Smith $      f'(x) x = -f(x),
56610807856dSBarry Smith    where f'(x) denotes the Jacobian matrix and f(x) is the function.
56620807856dSBarry Smith 
56630807856dSBarry Smith    Level: beginner
56640807856dSBarry Smith 
5665c5b75c40SBarry Smith    Developer Note:  This bleeds the allocated memory SNESMatlabContext *sctx;
5666c5b75c40SBarry Smith 
56670807856dSBarry Smith .keywords: SNES, nonlinear, set, function
56680807856dSBarry Smith 
56690807856dSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
567061b2408cSBarry Smith */
5671f8b49ee9SBarry Smith PetscErrorCode  SNESSetFunctionMatlab(SNES snes,Vec r,const char *f,mxArray *ctx)
56720807856dSBarry Smith {
56730807856dSBarry Smith   PetscErrorCode    ierr;
56748f6e6473SBarry Smith   SNESMatlabContext *sctx;
56750807856dSBarry Smith 
56760807856dSBarry Smith   PetscFunctionBegin;
56778f6e6473SBarry Smith   /* currently sctx is memory bleed */
5678854ce69bSBarry Smith   ierr = PetscNew(&sctx);CHKERRQ(ierr);
5679f8b49ee9SBarry Smith   ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr);
56808f6e6473SBarry Smith   /*
56818f6e6473SBarry Smith      This should work, but it doesn't
56828f6e6473SBarry Smith   sctx->ctx = ctx;
56838f6e6473SBarry Smith   mexMakeArrayPersistent(sctx->ctx);
56848f6e6473SBarry Smith   */
56858f6e6473SBarry Smith   sctx->ctx = mxDuplicateArray(ctx);
56868f6e6473SBarry Smith   ierr      = SNESSetFunction(snes,r,SNESComputeFunction_Matlab,sctx);CHKERRQ(ierr);
56870807856dSBarry Smith   PetscFunctionReturn(0);
56880807856dSBarry Smith }
568969b4f73cSBarry Smith 
569061b2408cSBarry Smith /*
5691bf388a1fSBarry Smith    SNESComputeJacobian_Matlab - Calls the function that has been set with SNESSetJacobianMatlab().
569261b2408cSBarry Smith 
569361b2408cSBarry Smith    Collective on SNES
569461b2408cSBarry Smith 
569561b2408cSBarry Smith    Input Parameters:
569661b2408cSBarry Smith +  snes - the SNES context
569761b2408cSBarry Smith .  x - input vector
569861b2408cSBarry Smith .  A, B - the matrices
569961b2408cSBarry Smith -  ctx - user context
570061b2408cSBarry Smith 
570161b2408cSBarry Smith    Level: developer
570261b2408cSBarry Smith 
570361b2408cSBarry Smith .keywords: SNES, nonlinear, compute, function
570461b2408cSBarry Smith 
570561b2408cSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction()
570661b2408cSBarry Smith @*/
5707f3229a78SSatish Balay PetscErrorCode  SNESComputeJacobian_Matlab(SNES snes,Vec x,Mat A,Mat B,void *ctx)
570861b2408cSBarry Smith {
570961b2408cSBarry Smith   PetscErrorCode    ierr;
571061b2408cSBarry Smith   SNESMatlabContext *sctx = (SNESMatlabContext*)ctx;
571161b2408cSBarry Smith   int               nlhs  = 2,nrhs = 6;
571261b2408cSBarry Smith   mxArray           *plhs[2],*prhs[6];
571361b2408cSBarry Smith   long long int     lx = 0,lA = 0,ls = 0, lB = 0;
571461b2408cSBarry Smith 
571561b2408cSBarry Smith   PetscFunctionBegin;
571661b2408cSBarry Smith   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
571761b2408cSBarry Smith   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
571861b2408cSBarry Smith 
571961b2408cSBarry Smith   /* call Matlab function in ctx with arguments x and y */
572061b2408cSBarry Smith 
572161b2408cSBarry Smith   ierr    = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr);
572261b2408cSBarry Smith   ierr    = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr);
572361b2408cSBarry Smith   ierr    = PetscMemcpy(&lA,A,sizeof(x));CHKERRQ(ierr);
572461b2408cSBarry Smith   ierr    = PetscMemcpy(&lB,B,sizeof(x));CHKERRQ(ierr);
572561b2408cSBarry Smith   prhs[0] = mxCreateDoubleScalar((double)ls);
572661b2408cSBarry Smith   prhs[1] = mxCreateDoubleScalar((double)lx);
572761b2408cSBarry Smith   prhs[2] = mxCreateDoubleScalar((double)lA);
572861b2408cSBarry Smith   prhs[3] = mxCreateDoubleScalar((double)lB);
572961b2408cSBarry Smith   prhs[4] = mxCreateString(sctx->funcname);
573061b2408cSBarry Smith   prhs[5] = sctx->ctx;
5731b807a863SBarry Smith   ierr    = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeJacobianInternal");CHKERRQ(ierr);
573261b2408cSBarry Smith   ierr    = mxGetScalar(plhs[0]);CHKERRQ(ierr);
573361b2408cSBarry Smith   mxDestroyArray(prhs[0]);
573461b2408cSBarry Smith   mxDestroyArray(prhs[1]);
573561b2408cSBarry Smith   mxDestroyArray(prhs[2]);
573661b2408cSBarry Smith   mxDestroyArray(prhs[3]);
573761b2408cSBarry Smith   mxDestroyArray(prhs[4]);
573861b2408cSBarry Smith   mxDestroyArray(plhs[0]);
573961b2408cSBarry Smith   mxDestroyArray(plhs[1]);
574061b2408cSBarry Smith   PetscFunctionReturn(0);
574161b2408cSBarry Smith }
574261b2408cSBarry Smith 
574361b2408cSBarry Smith /*
574461b2408cSBarry Smith    SNESSetJacobianMatlab - Sets the Jacobian function evaluation routine and two empty Jacobian matrices
574561b2408cSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
5746e3c5b3baSBarry Smith    equations from MATLAB. Here the function is a string containing the name of a MATLAB function
574761b2408cSBarry Smith 
574861b2408cSBarry Smith    Logically Collective on SNES
574961b2408cSBarry Smith 
575061b2408cSBarry Smith    Input Parameters:
575161b2408cSBarry Smith +  snes - the SNES context
575261b2408cSBarry Smith .  A,B - Jacobian matrices
5753f8b49ee9SBarry Smith .  J - function evaluation routine
575461b2408cSBarry Smith -  ctx - user context
575561b2408cSBarry Smith 
575661b2408cSBarry Smith    Level: developer
575761b2408cSBarry Smith 
5758c5b75c40SBarry Smith    Developer Note:  This bleeds the allocated memory SNESMatlabContext *sctx;
5759c5b75c40SBarry Smith 
576061b2408cSBarry Smith .keywords: SNES, nonlinear, set, function
576161b2408cSBarry Smith 
5762f8b49ee9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction(), J
576361b2408cSBarry Smith */
5764f8b49ee9SBarry Smith PetscErrorCode  SNESSetJacobianMatlab(SNES snes,Mat A,Mat B,const char *J,mxArray *ctx)
576561b2408cSBarry Smith {
576661b2408cSBarry Smith   PetscErrorCode    ierr;
576761b2408cSBarry Smith   SNESMatlabContext *sctx;
576861b2408cSBarry Smith 
576961b2408cSBarry Smith   PetscFunctionBegin;
577061b2408cSBarry Smith   /* currently sctx is memory bleed */
5771854ce69bSBarry Smith   ierr = PetscNew(&sctx);CHKERRQ(ierr);
5772f8b49ee9SBarry Smith   ierr = PetscStrallocpy(J,&sctx->funcname);CHKERRQ(ierr);
577361b2408cSBarry Smith   /*
577461b2408cSBarry Smith      This should work, but it doesn't
577561b2408cSBarry Smith   sctx->ctx = ctx;
577661b2408cSBarry Smith   mexMakeArrayPersistent(sctx->ctx);
577761b2408cSBarry Smith   */
577861b2408cSBarry Smith   sctx->ctx = mxDuplicateArray(ctx);
577961b2408cSBarry Smith   ierr      = SNESSetJacobian(snes,A,B,SNESComputeJacobian_Matlab,sctx);CHKERRQ(ierr);
578061b2408cSBarry Smith   PetscFunctionReturn(0);
578161b2408cSBarry Smith }
578269b4f73cSBarry Smith 
5783f9eb7ae2SShri Abhyankar /*
5784f9eb7ae2SShri Abhyankar    SNESMonitor_Matlab - Calls the function that has been set with SNESMonitorSetMatlab().
5785f9eb7ae2SShri Abhyankar 
5786f9eb7ae2SShri Abhyankar    Collective on SNES
5787f9eb7ae2SShri Abhyankar 
5788f9eb7ae2SShri Abhyankar .seealso: SNESSetFunction(), SNESGetFunction()
5789f9eb7ae2SShri Abhyankar @*/
57907087cfbeSBarry Smith PetscErrorCode  SNESMonitor_Matlab(SNES snes,PetscInt it, PetscReal fnorm, void *ctx)
5791f9eb7ae2SShri Abhyankar {
5792f9eb7ae2SShri Abhyankar   PetscErrorCode    ierr;
579348f37e70SShri Abhyankar   SNESMatlabContext *sctx = (SNESMatlabContext*)ctx;
5794f9eb7ae2SShri Abhyankar   int               nlhs  = 1,nrhs = 6;
5795f9eb7ae2SShri Abhyankar   mxArray           *plhs[1],*prhs[6];
5796f9eb7ae2SShri Abhyankar   long long int     lx = 0,ls = 0;
5797f9eb7ae2SShri Abhyankar   Vec               x  = snes->vec_sol;
5798f9eb7ae2SShri Abhyankar 
5799f9eb7ae2SShri Abhyankar   PetscFunctionBegin;
5800f9eb7ae2SShri Abhyankar   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
5801f9eb7ae2SShri Abhyankar 
5802f9eb7ae2SShri Abhyankar   ierr    = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr);
5803f9eb7ae2SShri Abhyankar   ierr    = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr);
5804f9eb7ae2SShri Abhyankar   prhs[0] = mxCreateDoubleScalar((double)ls);
5805f9eb7ae2SShri Abhyankar   prhs[1] = mxCreateDoubleScalar((double)it);
5806f9eb7ae2SShri Abhyankar   prhs[2] = mxCreateDoubleScalar((double)fnorm);
5807f9eb7ae2SShri Abhyankar   prhs[3] = mxCreateDoubleScalar((double)lx);
5808f9eb7ae2SShri Abhyankar   prhs[4] = mxCreateString(sctx->funcname);
5809f9eb7ae2SShri Abhyankar   prhs[5] = sctx->ctx;
5810f9eb7ae2SShri Abhyankar   ierr    = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESMonitorInternal");CHKERRQ(ierr);
5811f9eb7ae2SShri Abhyankar   ierr    = mxGetScalar(plhs[0]);CHKERRQ(ierr);
5812f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[0]);
5813f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[1]);
5814f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[2]);
5815f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[3]);
5816f9eb7ae2SShri Abhyankar   mxDestroyArray(prhs[4]);
5817f9eb7ae2SShri Abhyankar   mxDestroyArray(plhs[0]);
5818f9eb7ae2SShri Abhyankar   PetscFunctionReturn(0);
5819f9eb7ae2SShri Abhyankar }
5820f9eb7ae2SShri Abhyankar 
5821f9eb7ae2SShri Abhyankar /*
5822e3c5b3baSBarry Smith    SNESMonitorSetMatlab - Sets the monitor function from MATLAB
5823f9eb7ae2SShri Abhyankar 
5824f9eb7ae2SShri Abhyankar    Level: developer
5825f9eb7ae2SShri Abhyankar 
5826c5b75c40SBarry Smith    Developer Note:  This bleeds the allocated memory SNESMatlabContext *sctx;
5827c5b75c40SBarry Smith 
5828f9eb7ae2SShri Abhyankar .keywords: SNES, nonlinear, set, function
5829f9eb7ae2SShri Abhyankar 
5830f9eb7ae2SShri Abhyankar .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction()
5831f9eb7ae2SShri Abhyankar */
58326e4dcb14SBarry Smith PetscErrorCode  SNESMonitorSetMatlab(SNES snes,const char *f,mxArray *ctx)
5833f9eb7ae2SShri Abhyankar {
5834f9eb7ae2SShri Abhyankar   PetscErrorCode    ierr;
5835f9eb7ae2SShri Abhyankar   SNESMatlabContext *sctx;
5836f9eb7ae2SShri Abhyankar 
5837f9eb7ae2SShri Abhyankar   PetscFunctionBegin;
5838f9eb7ae2SShri Abhyankar   /* currently sctx is memory bleed */
5839854ce69bSBarry Smith   ierr = PetscNew(&sctx);CHKERRQ(ierr);
58406e4dcb14SBarry Smith   ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr);
5841f9eb7ae2SShri Abhyankar   /*
5842f9eb7ae2SShri Abhyankar      This should work, but it doesn't
5843f9eb7ae2SShri Abhyankar   sctx->ctx = ctx;
5844f9eb7ae2SShri Abhyankar   mexMakeArrayPersistent(sctx->ctx);
5845f9eb7ae2SShri Abhyankar   */
5846f9eb7ae2SShri Abhyankar   sctx->ctx = mxDuplicateArray(ctx);
58470298fd71SBarry Smith   ierr      = SNESMonitorSet(snes,SNESMonitor_Matlab,sctx,NULL);CHKERRQ(ierr);
5848f9eb7ae2SShri Abhyankar   PetscFunctionReturn(0);
5849f9eb7ae2SShri Abhyankar }
5850f9eb7ae2SShri Abhyankar 
585169b4f73cSBarry Smith #endif
5852