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