xref: /petsc/include/petscsnes.h (revision 61860be5939b0caf981177eaa35e3e4ec5310d8f)
173f4d377SMatthew Knepley /* $Id: petscsnes.h,v 1.111 2001/08/06 21:17:05 bsmith Exp $ */
2f26ada1bSBarry Smith /*
3eef9c623SLois Curfman McInnes     User interface for the nonlinear solvers package.
4f26ada1bSBarry Smith */
50a835dfdSSatish Balay #if !defined(__PETSCSNES_H)
60a835dfdSSatish Balay #define __PETSCSNES_H
70a835dfdSSatish Balay #include "petscsles.h"
8b1f5cb9dSBarry Smith 
9435da068SBarry Smith /*S
10435da068SBarry Smith      SNES - Abstract PETSc object that manages all nonlinear solves
11435da068SBarry Smith 
12435da068SBarry Smith    Level: beginner
13435da068SBarry Smith 
14435da068SBarry Smith   Concepts: nonlinear solvers
15435da068SBarry Smith 
16eef9c623SLois Curfman McInnes .seealso:  SNESCreate(), SNESSetType(), SNESType, TS, SLES, KSP, PC
17435da068SBarry Smith S*/
18f09e8eb9SSatish Balay typedef struct _p_SNES* SNES;
19435da068SBarry Smith 
20435da068SBarry Smith /*E
21435da068SBarry Smith     SNESType - String with the name of a PETSc SNES method or the creation function
22435da068SBarry Smith        with an optional dynamic library name, for example
23435da068SBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:mysnescreate()
24435da068SBarry Smith 
25435da068SBarry Smith    Level: beginner
26435da068SBarry Smith 
27435da068SBarry Smith .seealso: SNESSetType(), SNES
28435da068SBarry Smith E*/
29eef9c623SLois Curfman McInnes #define SNESLS          "ls"
30eef9c623SLois Curfman McInnes #define SNESTR          "tr"
31eef9c623SLois Curfman McInnes #define SNESTEST        "test"
3282bf6240SBarry Smith typedef char *SNESType;
33b1f5cb9dSBarry Smith 
34deeb6e72SMatthew Knepley /* Logging support */
35deeb6e72SMatthew Knepley extern int SNES_COOKIE;
36eef9c623SLois Curfman McInnes extern int SNES_Solve, SNES_LineSearch, SNES_FunctionEval, SNES_JacobianEval;
37eef9c623SLois Curfman McInnes 
38deeb6e72SMatthew Knepley 
398ba1e511SMatthew Knepley EXTERN int SNESInitializePackage(char *);
40deeb6e72SMatthew Knepley 
41eef9c623SLois Curfman McInnes EXTERN int SNESCreate(MPI_Comm,SNES*);
42ca44d042SBarry Smith EXTERN int SNESDestroy(SNES);
43ca44d042SBarry Smith EXTERN int SNESSetType(SNES,SNESType);
4487828ca2SBarry Smith EXTERN int SNESSetMonitor(SNES,int(*)(SNES,int,PetscReal,void*),void *,int (*)(void *));
45ca44d042SBarry Smith EXTERN int SNESClearMonitor(SNES);
4687828ca2SBarry Smith EXTERN int SNESSetConvergenceHistory(SNES,PetscReal*,int *,int,PetscTruth);
4787828ca2SBarry Smith EXTERN int SNESGetConvergenceHistory(SNES,PetscReal**,int **,int *);
48ca44d042SBarry Smith EXTERN int SNESSetUp(SNES,Vec);
49ca44d042SBarry Smith EXTERN int SNESSolve(SNES,Vec,int*);
5084cb2905SBarry Smith 
514d8f6ca9SMatthew Knepley EXTERN int SNESAddOptionsChecker(int (*)(SNES));
524d8f6ca9SMatthew Knepley 
534d8f6ca9SMatthew Knepley EXTERN int SNESSetRhsBC(SNES, int (*)(SNES, Vec, void *));
544d8f6ca9SMatthew Knepley EXTERN int SNESDefaultRhsBC(SNES, Vec, void *);
554d8f6ca9SMatthew Knepley EXTERN int SNESSetSolutionBC(SNES, int (*)(SNES, Vec, void *));
564d8f6ca9SMatthew Knepley EXTERN int SNESDefaultSolutionBC(SNES, Vec, void *);
574d8f6ca9SMatthew Knepley EXTERN int SNESSetUpdate(SNES, int (*)(SNES, int));
584d8f6ca9SMatthew Knepley EXTERN int SNESDefaultUpdate(SNES, int);
594d8f6ca9SMatthew Knepley 
60b0a32e0cSBarry Smith extern PetscFList SNESList;
61ca44d042SBarry Smith EXTERN int SNESRegisterDestroy(void);
62ca44d042SBarry Smith EXTERN int SNESRegisterAll(char *);
6384cb2905SBarry Smith 
64ca44d042SBarry Smith EXTERN int SNESRegister(char*,char*,char*,int(*)(SNES));
65aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
66f1af5d2fSBarry Smith #define SNESRegisterDynamic(a,b,c,d) SNESRegister(a,b,c,0)
67b2002411SLois Curfman McInnes #else
68f1af5d2fSBarry Smith #define SNESRegisterDynamic(a,b,c,d) SNESRegister(a,b,c,d)
69b2002411SLois Curfman McInnes #endif
70b2002411SLois Curfman McInnes 
71ca44d042SBarry Smith EXTERN int SNESGetSLES(SNES,SLES*);
72ca44d042SBarry Smith EXTERN int SNESGetSolution(SNES,Vec*);
73ca44d042SBarry Smith EXTERN int SNESGetSolutionUpdate(SNES,Vec*);
748439883dSBarry Smith EXTERN int SNESGetFunction(SNES,Vec*,void**,int(**)(SNES,Vec,Vec,void*));
75b0a32e0cSBarry Smith EXTERN int SNESView(SNES,PetscViewer);
767bc3d0afSSatish Balay 
77ca44d042SBarry Smith EXTERN int SNESSetOptionsPrefix(SNES,char*);
78ca44d042SBarry Smith EXTERN int SNESAppendOptionsPrefix(SNES,char*);
79ca44d042SBarry Smith EXTERN int SNESGetOptionsPrefix(SNES,char**);
80ca44d042SBarry Smith EXTERN int SNESSetFromOptions(SNES);
8140191667SLois Curfman McInnes 
82ca44d042SBarry Smith EXTERN int MatCreateSNESMF(SNES,Vec,Mat*);
831d1367b7SBarry Smith EXTERN int MatCreateMF(Vec,Mat*);
841d1367b7SBarry Smith EXTERN int MatSNESMFSetBase(Mat,Vec);
853a7fca6bSBarry Smith EXTERN int MatSNESMFComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*);
86ca44d042SBarry Smith EXTERN int MatSNESMFSetFunction(Mat,Vec,int(*)(SNES,Vec,Vec,void*),void*);
8787828ca2SBarry Smith EXTERN int MatSNESMFSetFunctioni(Mat,int (*)(int,Vec,PetscScalar*,void*));
888a124369SBarry Smith EXTERN int MatSNESMFSetFunctioniBase(Mat,int (*)(Vec,void*));
8974637425SBarry Smith EXTERN int MatSNESMFAddNullSpace(Mat,MatNullSpace);
9087828ca2SBarry Smith EXTERN int MatSNESMFSetHHistory(Mat,PetscScalar *,int);
91ca44d042SBarry Smith EXTERN int MatSNESMFResetHHistory(Mat);
9287828ca2SBarry Smith EXTERN int MatSNESMFSetFunctionError(Mat,PetscReal);
93ca44d042SBarry Smith EXTERN int MatSNESMFSetPeriod(Mat,int);
9487828ca2SBarry Smith EXTERN int MatSNESMFGetH(Mat,PetscScalar *);
9587828ca2SBarry Smith EXTERN int MatSNESMFKSPMonitor(KSP,int,PetscReal,void *);
96ca44d042SBarry Smith EXTERN int MatSNESMFSetFromOptions(Mat);
975b7f0c42SBarry Smith EXTERN int MatSNESMFCheckPositivity(Vec,Vec,PetscScalar*,void*);
98*61860be5SBarry Smith EXTERN int MatSNESMFSetCheckh(Mat,int (*)(Vec,Vec,PetscScalar*,void*),void*);
995b7f0c42SBarry Smith 
1005a655dc6SBarry Smith typedef struct _p_MatSNESMFCtx   *MatSNESMFCtx;
101435da068SBarry Smith 
102435da068SBarry Smith #define MATSNESMF_DEFAULT "default"
103435da068SBarry Smith #define MATSNESMF_WP      "wp"
104b7fd4e64SBarry Smith typedef char* MatSNESMFType;
105ca44d042SBarry Smith EXTERN int MatSNESMFSetType(Mat,MatSNESMFType);
106ca44d042SBarry Smith EXTERN int MatSNESMFRegister(char *,char *,char *,int (*)(MatSNESMFCtx));
107aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
108f1af5d2fSBarry Smith #define MatSNESMFRegisterDynamic(a,b,c,d) MatSNESMFRegister(a,b,c,0)
1099a6cb015SBarry Smith #else
110f1af5d2fSBarry Smith #define MatSNESMFRegisterDynamic(a,b,c,d) MatSNESMFRegister(a,b,c,d)
1119a6cb015SBarry Smith #endif
112ca44d042SBarry Smith EXTERN int MatSNESMFRegisterAll(char *);
113ca44d042SBarry Smith EXTERN int MatSNESMFRegisterDestroy(void);
1143a7fca6bSBarry Smith EXTERN int MatSNESMFDefaultSetUmin(Mat,PetscReal);
115ca44d042SBarry Smith EXTERN int MatSNESMFWPSetComputeNormA(Mat,PetscTruth);
116ca44d042SBarry Smith EXTERN int MatSNESMFWPSetComputeNormU(Mat,PetscTruth);
1178f6e3e37SBarry Smith 
1183a7fca6bSBarry Smith EXTERN int MatDAADSetSNES(Mat,SNES);
1193a7fca6bSBarry Smith 
120ca44d042SBarry Smith EXTERN int SNESGetType(SNES,SNESType*);
12187828ca2SBarry Smith EXTERN int SNESDefaultMonitor(SNES,int,PetscReal,void *);
12287828ca2SBarry Smith EXTERN int SNESRatioMonitor(SNES,int,PetscReal,void *);
1233a7fca6bSBarry Smith EXTERN int SNESSetRatioMonitor(SNES);
12487828ca2SBarry Smith EXTERN int SNESVecViewMonitor(SNES,int,PetscReal,void *);
1255ed2d596SBarry Smith EXTERN int SNESVecViewResidualMonitor(SNES,int,PetscReal,void *);
12687828ca2SBarry Smith EXTERN int SNESVecViewUpdateMonitor(SNES,int,PetscReal,void *);
12787828ca2SBarry Smith EXTERN int SNESDefaultSMonitor(SNES,int,PetscReal,void *);
12887828ca2SBarry Smith EXTERN int SNESSetTolerances(SNES,PetscReal,PetscReal,PetscReal,int,int);
12987828ca2SBarry Smith EXTERN int SNESGetTolerances(SNES,PetscReal*,PetscReal*,PetscReal*,int*,int*);
13087828ca2SBarry Smith EXTERN int SNESSetTrustRegionTolerance(SNES,PetscReal);
131ca44d042SBarry Smith EXTERN int SNESGetIterationNumber(SNES,int*);
13287828ca2SBarry Smith EXTERN int SNESGetFunctionNorm(SNES,PetscScalar*);
133ca44d042SBarry Smith EXTERN int SNESGetNumberUnsuccessfulSteps(SNES,int*);
1343064298fSMatthew Knepley EXTERN int SNESSetMaximumUnsuccessfulSteps(SNES,int);
1353064298fSMatthew Knepley EXTERN int SNESGetMaximumUnsuccessfulSteps(SNES,int*);
136ca44d042SBarry Smith EXTERN int SNESGetNumberLinearIterations(SNES,int*);
13787828ca2SBarry Smith EXTERN int SNES_KSP_SetParametersEW(SNES,int,PetscReal,PetscReal,PetscReal,PetscReal,PetscReal,PetscReal);
138ca44d042SBarry Smith EXTERN int SNES_KSP_SetConvergenceTestEW(SNES);
139eafb4bcbSBarry Smith 
1403369ce9aSBarry Smith /*
1413369ce9aSBarry Smith      Reuse the default KSP monitor routines for SNES
1423369ce9aSBarry Smith */
143b0a32e0cSBarry Smith EXTERN int SNESLGMonitorCreate(char*,char*,int,int,int,int,PetscDrawLG*);
14487828ca2SBarry Smith EXTERN int SNESLGMonitor(SNES,int,PetscReal,void*);
145b0a32e0cSBarry Smith EXTERN int SNESLGMonitorDestroy(PetscDrawLG);
146eafb4bcbSBarry Smith 
147ca44d042SBarry Smith EXTERN int SNESSetApplicationContext(SNES,void *);
148ca44d042SBarry Smith EXTERN int SNESGetApplicationContext(SNES,void **);
149184914b5SBarry Smith 
150435da068SBarry Smith /*E
151435da068SBarry Smith     SNESConvergedReason - reason a SNES method was said to
152435da068SBarry Smith          have converged or diverged
153435da068SBarry Smith 
154435da068SBarry Smith    Level: beginner
155435da068SBarry Smith 
156435da068SBarry Smith    Notes: this must match finclude/petscsnes.h
157435da068SBarry Smith 
158435da068SBarry Smith .seealso: SNESSolve(), SNESGetConvergedReason(), KSPConvergedReason, SNESSetConvergenceTest()
159435da068SBarry Smith E*/
160184914b5SBarry Smith typedef enum {/* converged */
161184914b5SBarry Smith               SNES_CONVERGED_FNORM_ABS         =  2, /* F < F_minabs */
162184914b5SBarry Smith               SNES_CONVERGED_FNORM_RELATIVE    =  3, /* F < F_mintol*F_initial */
163184914b5SBarry Smith               SNES_CONVERGED_PNORM_RELATIVE    =  4, /* step size small */
164184914b5SBarry Smith               SNES_CONVERGED_GNORM_ABS         =  5, /* grad F < grad F_min */
165184914b5SBarry Smith               SNES_CONVERGED_TR_REDUCTION      =  6,
166184914b5SBarry Smith               SNES_CONVERGED_TR_DELTA          =  7,
167184914b5SBarry Smith               /* diverged */
168184914b5SBarry Smith               SNES_DIVERGED_FUNCTION_COUNT     = -2,
169184914b5SBarry Smith               SNES_DIVERGED_FNORM_NAN          = -4,
170184914b5SBarry Smith               SNES_DIVERGED_MAX_IT             = -5,
171184914b5SBarry Smith               SNES_DIVERGED_LS_FAILURE         = -6,
172184914b5SBarry Smith               SNES_DIVERGED_TR_REDUCTION       = -7,
17358c775ebSBarry Smith               SNES_DIVERGED_LOCAL_MIN          = -8,  /* || J^T b || is small, implies converged to local minimum of F() */
174184914b5SBarry Smith               SNES_CONVERGED_ITERATING         =  0} SNESConvergedReason;
175184914b5SBarry Smith 
17687828ca2SBarry Smith EXTERN int SNESSetConvergenceTest(SNES,int (*)(SNES,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void*);
177eef9c623SLois Curfman McInnes EXTERN int SNESConverged_LS(SNES,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*);
178eef9c623SLois Curfman McInnes EXTERN int SNESConverged_TR(SNES,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*);
179ca44d042SBarry Smith EXTERN int SNESGetConvergedReason(SNES,SNESConvergedReason*);
180ddbbdb52SLois Curfman McInnes 
1813a7fca6bSBarry Smith EXTERN int SNESDAFormFunction(SNES,Vec,Vec,void*);
1823a7fca6bSBarry Smith EXTERN int SNESDAComputeJacobianWithAdic(SNES,Vec,Mat*,Mat*,MatStructure*,void*);
1833a7fca6bSBarry Smith EXTERN int SNESDAComputeJacobianWithAdifor(SNES,Vec,Mat*,Mat*,MatStructure*,void*);
1843a7fca6bSBarry Smith EXTERN int SNESDAComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*);
1853a7fca6bSBarry Smith 
186b67197daSBarry Smith /* --------- Solving systems of nonlinear equations --------------- */
187ca44d042SBarry Smith EXTERN int SNESSetFunction(SNES,Vec,int(*)(SNES,Vec,Vec,void*),void *);
188ca44d042SBarry Smith EXTERN int SNESComputeFunction(SNES,Vec,Vec);
189ca44d042SBarry Smith EXTERN int SNESSetJacobian(SNES,Mat,Mat,int(*)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void *);
1908439883dSBarry Smith EXTERN int SNESGetJacobian(SNES,Mat*,Mat*,void **,int(**)(SNES,Vec,Mat*,Mat*,MatStructure*,void*));
191ca44d042SBarry Smith EXTERN int SNESDefaultComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*);
192ca44d042SBarry Smith EXTERN int SNESDefaultComputeJacobianColor(SNES,Vec,Mat*,Mat*,MatStructure*,void*);
19387828ca2SBarry Smith EXTERN int SNESSetLineSearch(SNES,int(*)(SNES,void*,Vec,Vec,Vec,Vec,Vec,PetscReal,PetscReal*,PetscReal*,int*),void*);
19487828ca2SBarry Smith EXTERN int SNESNoLineSearch(SNES,void*,Vec,Vec,Vec,Vec,Vec,PetscReal,PetscReal*,PetscReal*,int*);
19587828ca2SBarry Smith EXTERN int SNESNoLineSearchNoNorms(SNES,void*,Vec,Vec,Vec,Vec,Vec,PetscReal,PetscReal*,PetscReal*,int*);
19687828ca2SBarry Smith EXTERN int SNESCubicLineSearch(SNES,void*,Vec,Vec,Vec,Vec,Vec,PetscReal,PetscReal*,PetscReal*,int*);
19787828ca2SBarry Smith EXTERN int SNESQuadraticLineSearch(SNES,void*,Vec,Vec,Vec,Vec,Vec,PetscReal,PetscReal*,PetscReal*,int*);
198ca44d042SBarry Smith EXTERN int SNESSetLineSearchCheck(SNES,int(*)(SNES,void*,Vec,PetscTruth*),void*);
19987828ca2SBarry Smith EXTERN int SNESSetLineSearchParams(SNES,PetscReal,PetscReal,PetscReal);
20087828ca2SBarry Smith EXTERN int SNESGetLineSearchParams(SNES,PetscReal*,PetscReal*,PetscReal*);
20140191667SLois Curfman McInnes 
202eef9c623SLois Curfman McInnes /* Should this routine be private? */
203ca44d042SBarry Smith EXTERN int SNESComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*);
204ddbbdb52SLois Curfman McInnes 
205ce3d82beSBarry Smith #endif
206b1f5cb9dSBarry Smith 
207