1 /* 2 User interface for the nonlinear solvers package. 3 */ 4 #if !defined(__PETSCSNES_H) 5 #define __PETSCSNES_H 6 #include "petscksp.h" 7 #include "petscdmda.h" 8 PETSC_EXTERN_CXX_BEGIN 9 10 /*S 11 SNES - Abstract PETSc object that manages all nonlinear solves 12 13 Level: beginner 14 15 Concepts: nonlinear solvers 16 17 .seealso: SNESCreate(), SNESSetType(), SNESType, TS, KSP, KSP, PC 18 S*/ 19 typedef struct _p_SNES* SNES; 20 21 /*J 22 SNESType - String with the name of a PETSc SNES method or the creation function 23 with an optional dynamic library name, for example 24 http://www.mcs.anl.gov/petsc/lib.a:mysnescreate() 25 26 Level: beginner 27 28 .seealso: SNESSetType(), SNES 29 J*/ 30 #define SNESType char* 31 #define SNESLS "ls" 32 #define SNESTR "tr" 33 #define SNESPYTHON "python" 34 #define SNESTEST "test" 35 #define SNESNRICHARDSON "nrichardson" 36 #define SNESKSPONLY "ksponly" 37 #define SNESVIRS "virs" 38 #define SNESVISS "viss" 39 #define SNESNGMRES "ngmres" 40 #define SNESQN "qn" 41 #define SNESSHELL "shell" 42 #define SNESGS "gs" 43 #define SNESNCG "ncg" 44 #define SNESSORQN "sorqn" 45 #define SNESFAS "fas" 46 #define SNESMS "ms" 47 48 /* Logging support */ 49 extern PetscClassId SNES_CLASSID; 50 51 extern PetscErrorCode SNESInitializePackage(const char[]); 52 53 extern PetscErrorCode SNESCreate(MPI_Comm,SNES*); 54 extern PetscErrorCode SNESReset(SNES); 55 extern PetscErrorCode SNESDestroy(SNES*); 56 extern PetscErrorCode SNESSetType(SNES,const SNESType); 57 extern PetscErrorCode SNESMonitor(SNES,PetscInt,PetscReal); 58 extern PetscErrorCode SNESMonitorSet(SNES,PetscErrorCode(*)(SNES,PetscInt,PetscReal,void*),void *,PetscErrorCode (*)(void**)); 59 extern PetscErrorCode SNESMonitorCancel(SNES); 60 extern PetscErrorCode SNESSetConvergenceHistory(SNES,PetscReal[],PetscInt[],PetscInt,PetscBool ); 61 extern PetscErrorCode SNESGetConvergenceHistory(SNES,PetscReal*[],PetscInt *[],PetscInt *); 62 extern PetscErrorCode SNESSetUp(SNES); 63 extern PetscErrorCode SNESSolve(SNES,Vec,Vec); 64 extern PetscErrorCode SNESSetErrorIfNotConverged(SNES,PetscBool ); 65 extern PetscErrorCode SNESGetErrorIfNotConverged(SNES,PetscBool *); 66 67 68 extern PetscErrorCode SNESAddOptionsChecker(PetscErrorCode (*)(SNES)); 69 70 extern PetscErrorCode SNESSetUpdate(SNES, PetscErrorCode (*)(SNES, PetscInt)); 71 extern PetscErrorCode SNESDefaultUpdate(SNES, PetscInt); 72 73 extern PetscFList SNESList; 74 extern PetscErrorCode SNESRegisterDestroy(void); 75 extern PetscErrorCode SNESRegisterAll(const char[]); 76 77 extern PetscErrorCode SNESRegister(const char[],const char[],const char[],PetscErrorCode (*)(SNES)); 78 79 /*MC 80 SNESRegisterDynamic - Adds a method to the nonlinear solver package. 81 82 Synopsis: 83 PetscErrorCode SNESRegisterDynamic(const char *name_solver,const char *path,const char *name_create,PetscErrorCode (*routine_create)(SNES)) 84 85 Not collective 86 87 Input Parameters: 88 + name_solver - name of a new user-defined solver 89 . path - path (either absolute or relative) the library containing this solver 90 . name_create - name of routine to create method context 91 - routine_create - routine to create method context 92 93 Notes: 94 SNESRegisterDynamic() may be called multiple times to add several user-defined solvers. 95 96 If dynamic libraries are used, then the fourth input argument (routine_create) 97 is ignored. 98 99 Environmental variables such as ${PETSC_ARCH}, ${PETSC_DIR}, ${PETSC_LIB_DIR}, 100 and others of the form ${any_environmental_variable} occuring in pathname will be 101 replaced with appropriate values. 102 103 Sample usage: 104 .vb 105 SNESRegisterDynamic("my_solver",/home/username/my_lib/lib/libg/solaris/mylib.a, 106 "MySolverCreate",MySolverCreate); 107 .ve 108 109 Then, your solver can be chosen with the procedural interface via 110 $ SNESSetType(snes,"my_solver") 111 or at runtime via the option 112 $ -snes_type my_solver 113 114 Level: advanced 115 116 Note: If your function is not being put into a shared library then use SNESRegister() instead 117 118 .keywords: SNES, nonlinear, register 119 120 .seealso: SNESRegisterAll(), SNESRegisterDestroy() 121 M*/ 122 #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 123 #define SNESRegisterDynamic(a,b,c,d) SNESRegister(a,b,c,0) 124 #else 125 #define SNESRegisterDynamic(a,b,c,d) SNESRegister(a,b,c,d) 126 #endif 127 128 extern PetscErrorCode SNESGetKSP(SNES,KSP*); 129 extern PetscErrorCode SNESSetKSP(SNES,KSP); 130 extern PetscErrorCode SNESGetSolution(SNES,Vec*); 131 extern PetscErrorCode SNESGetSolutionUpdate(SNES,Vec*); 132 extern PetscErrorCode SNESGetRhs(SNES,Vec*); 133 extern PetscErrorCode SNESView(SNES,PetscViewer); 134 135 extern PetscErrorCode SNESSetOptionsPrefix(SNES,const char[]); 136 extern PetscErrorCode SNESAppendOptionsPrefix(SNES,const char[]); 137 extern PetscErrorCode SNESGetOptionsPrefix(SNES,const char*[]); 138 extern PetscErrorCode SNESSetFromOptions(SNES); 139 extern PetscErrorCode SNESDefaultGetWork(SNES,PetscInt); 140 141 extern PetscErrorCode MatCreateSNESMF(SNES,Mat*); 142 extern PetscErrorCode MatMFFDComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 143 144 extern PetscErrorCode MatDAADSetSNES(Mat,SNES); 145 146 extern PetscErrorCode SNESGetType(SNES,const SNESType*); 147 extern PetscErrorCode SNESMonitorDefault(SNES,PetscInt,PetscReal,void *); 148 extern PetscErrorCode SNESMonitorRange(SNES,PetscInt,PetscReal,void *); 149 extern PetscErrorCode SNESMonitorRatio(SNES,PetscInt,PetscReal,void *); 150 extern PetscErrorCode SNESMonitorSetRatio(SNES,PetscViewer); 151 extern PetscErrorCode SNESMonitorSolution(SNES,PetscInt,PetscReal,void *); 152 extern PetscErrorCode SNESMonitorResidual(SNES,PetscInt,PetscReal,void *); 153 extern PetscErrorCode SNESMonitorSolutionUpdate(SNES,PetscInt,PetscReal,void *); 154 extern PetscErrorCode SNESMonitorDefaultShort(SNES,PetscInt,PetscReal,void *); 155 extern PetscErrorCode SNESSetTolerances(SNES,PetscReal,PetscReal,PetscReal,PetscInt,PetscInt); 156 extern PetscErrorCode SNESGetTolerances(SNES,PetscReal*,PetscReal*,PetscReal*,PetscInt*,PetscInt*); 157 extern PetscErrorCode SNESSetTrustRegionTolerance(SNES,PetscReal); 158 extern PetscErrorCode SNESGetFunctionNorm(SNES,PetscReal*); 159 extern PetscErrorCode SNESSetFunctionNorm(SNES,PetscReal); 160 extern PetscErrorCode SNESGetIterationNumber(SNES,PetscInt*); 161 extern PetscErrorCode SNESSetIterationNumber(SNES,PetscInt); 162 163 extern PetscErrorCode SNESGetNonlinearStepFailures(SNES,PetscInt*); 164 extern PetscErrorCode SNESSetMaxNonlinearStepFailures(SNES,PetscInt); 165 extern PetscErrorCode SNESGetMaxNonlinearStepFailures(SNES,PetscInt*); 166 extern PetscErrorCode SNESGetNumberFunctionEvals(SNES,PetscInt*); 167 168 extern PetscErrorCode SNESSetLagPreconditioner(SNES,PetscInt); 169 extern PetscErrorCode SNESGetLagPreconditioner(SNES,PetscInt*); 170 extern PetscErrorCode SNESSetLagJacobian(SNES,PetscInt); 171 extern PetscErrorCode SNESGetLagJacobian(SNES,PetscInt*); 172 extern PetscErrorCode SNESSetGridSequence(SNES,PetscInt); 173 174 extern PetscErrorCode SNESGetLinearSolveIterations(SNES,PetscInt*); 175 extern PetscErrorCode SNESGetLinearSolveFailures(SNES,PetscInt*); 176 extern PetscErrorCode SNESSetMaxLinearSolveFailures(SNES,PetscInt); 177 extern PetscErrorCode SNESGetMaxLinearSolveFailures(SNES,PetscInt*); 178 179 extern PetscErrorCode SNESKSPSetUseEW(SNES,PetscBool ); 180 extern PetscErrorCode SNESKSPGetUseEW(SNES,PetscBool *); 181 extern PetscErrorCode SNESKSPSetParametersEW(SNES,PetscInt,PetscReal,PetscReal,PetscReal,PetscReal,PetscReal,PetscReal); 182 extern PetscErrorCode SNESKSPGetParametersEW(SNES,PetscInt*,PetscReal*,PetscReal*,PetscReal*,PetscReal*,PetscReal*,PetscReal*); 183 184 extern PetscErrorCode SNESMonitorLGCreate(const char[],const char[],int,int,int,int,PetscDrawLG*); 185 extern PetscErrorCode SNESMonitorLG(SNES,PetscInt,PetscReal,void*); 186 extern PetscErrorCode SNESMonitorLGDestroy(PetscDrawLG*); 187 extern PetscErrorCode SNESMonitorLGRangeCreate(const char[],const char[],int,int,int,int,PetscDrawLG*); 188 extern PetscErrorCode SNESMonitorLGRange(SNES,PetscInt,PetscReal,void*); 189 extern PetscErrorCode SNESMonitorLGRangeDestroy(PetscDrawLG*); 190 191 extern PetscErrorCode SNESSetApplicationContext(SNES,void *); 192 extern PetscErrorCode SNESGetApplicationContext(SNES,void *); 193 extern PetscErrorCode SNESSetComputeApplicationContext(SNES,PetscErrorCode (*)(SNES,void**),PetscErrorCode (*)(void**)); 194 195 extern PetscErrorCode SNESPythonSetType(SNES,const char[]); 196 197 extern PetscErrorCode SNESSetFunctionDomainError(SNES); 198 extern PetscErrorCode SNESGetFunctionDomainError(SNES, PetscBool *); 199 200 /*E 201 SNESConvergedReason - reason a SNES method was said to 202 have converged or diverged 203 204 Level: beginner 205 206 The two most common reasons for divergence are 207 $ 1) an incorrectly coded or computed Jacobian or 208 $ 2) failure or lack of convergence in the linear system (in this case we recommend 209 $ testing with -pc_type lu to eliminate the linear solver as the cause of the problem). 210 211 Diverged Reasons: 212 . SNES_DIVERGED_LOCAL_MIN - this can only occur when using the line-search variant of SNES. 213 The line search wants to minimize Q(alpha) = 1/2 || F(x + alpha s) ||^2_2 this occurs 214 at Q'(alpha) = s^T F'(x+alpha s)^T F(x+alpha s) = 0. If s is the Newton direction - F'(x)^(-1)F(x) then 215 you get Q'(alpha) = -F(x)^T F'(x)^(-1)^T F'(x+alpha s)F(x+alpha s); when alpha = 0 216 Q'(0) = - ||F(x)||^2_2 which is always NEGATIVE if F'(x) is invertible. This means the Newton 217 direction is a descent direction and the line search should succeed if alpha is small enough. 218 219 If F'(x) is NOT invertible AND F'(x)^T F(x) = 0 then Q'(0) = 0 and the Newton direction 220 is NOT a descent direction so the line search will fail. All one can do at this point 221 is change the initial guess and try again. 222 223 An alternative explanation: Newton's method can be regarded as replacing the function with 224 its linear approximation and minimizing the 2-norm of that. That is F(x+s) approx F(x) + F'(x)s 225 so we minimize || F(x) + F'(x) s ||^2_2; do this using Least Squares. If F'(x) is invertible then 226 s = - F'(x)^(-1)F(x) otherwise F'(x)^T F'(x) s = -F'(x)^T F(x). If F'(x)^T F(x) is NOT zero then there 227 exists a nontrival (that is F'(x)s != 0) solution to the equation and this direction is 228 s = - [F'(x)^T F'(x)]^(-1) F'(x)^T F(x) so Q'(0) = - F(x)^T F'(x) [F'(x)^T F'(x)]^(-T) F'(x)^T F(x) 229 = - (F'(x)^T F(x)) [F'(x)^T F'(x)]^(-T) (F'(x)^T F(x)). Since we are assuming (F'(x)^T F(x)) != 0 230 and F'(x)^T F'(x) has no negative eigenvalues Q'(0) < 0 so s is a descent direction and the line 231 search should succeed for small enough alpha. 232 233 Note that this RARELY happens in practice. Far more likely the linear system is not being solved 234 (well enough?) or the Jacobian is wrong. 235 236 SNES_DIVERGED_MAX_IT means that the solver reached the maximum number of iterations without satisfying any 237 convergence criteria. SNES_CONVERGED_ITS means that SNESSkipConverged() was chosen as the convergence test; 238 thus the usual convergence criteria have not been checked and may or may not be satisfied. 239 240 Developer Notes: this must match finclude/petscsnes.h 241 242 The string versions of these are in SNESConvergedReason, if you change any value here you must 243 also adjust that array. 244 245 Each reason has its own manual page. 246 247 .seealso: SNESSolve(), SNESGetConvergedReason(), KSPConvergedReason, SNESSetConvergenceTest() 248 E*/ 249 typedef enum {/* converged */ 250 SNES_CONVERGED_FNORM_ABS = 2, /* ||F|| < atol */ 251 SNES_CONVERGED_FNORM_RELATIVE = 3, /* ||F|| < rtol*||F_initial|| */ 252 SNES_CONVERGED_PNORM_RELATIVE = 4, /* Newton computed step size small; || delta x || < stol */ 253 SNES_CONVERGED_ITS = 5, /* maximum iterations reached */ 254 SNES_CONVERGED_TR_DELTA = 7, 255 /* diverged */ 256 SNES_DIVERGED_FUNCTION_DOMAIN = -1, /* the new x location passed the function is not in the domain of F */ 257 SNES_DIVERGED_FUNCTION_COUNT = -2, 258 SNES_DIVERGED_LINEAR_SOLVE = -3, /* the linear solve failed */ 259 SNES_DIVERGED_FNORM_NAN = -4, 260 SNES_DIVERGED_MAX_IT = -5, 261 SNES_DIVERGED_LINE_SEARCH = -6, /* the line search failed */ 262 SNES_DIVERGED_INNER = -7, /* inner solve failed */ 263 SNES_DIVERGED_LOCAL_MIN = -8, /* || J^T b || is small, implies converged to local minimum of F() */ 264 SNES_CONVERGED_ITERATING = 0} SNESConvergedReason; 265 extern const char *const*SNESConvergedReasons; 266 267 /*MC 268 SNES_CONVERGED_FNORM_ABS - 2-norm(F) <= abstol 269 270 Level: beginner 271 272 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 273 274 M*/ 275 276 /*MC 277 SNES_CONVERGED_FNORM_RELATIVE - 2-norm(F) <= rtol*2-norm(F(x_0)) where x_0 is the initial guess 278 279 Level: beginner 280 281 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 282 283 M*/ 284 285 /*MC 286 SNES_CONVERGED_PNORM_RELATIVE - The 2-norm of the last step <= stol * 2-norm(x) where x is the current 287 solution and stol is the 4th argument to SNESSetTolerances() 288 289 Level: beginner 290 291 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 292 293 M*/ 294 295 /*MC 296 SNES_DIVERGED_FUNCTION_COUNT - The user provided function has been called more times then the final 297 argument to SNESSetTolerances() 298 299 Level: beginner 300 301 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 302 303 M*/ 304 305 /*MC 306 SNES_DIVERGED_FNORM_NAN - the 2-norm of the current function evaluation is not-a-number (NaN), this 307 is usually caused by a division of 0 by 0. 308 309 Level: beginner 310 311 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 312 313 M*/ 314 315 /*MC 316 SNES_DIVERGED_MAX_IT - SNESSolve() has reached the maximum number of iterations requested 317 318 Level: beginner 319 320 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 321 322 M*/ 323 324 /*MC 325 SNES_DIVERGED_LINE_SEARCH - The line search has failed. This only occurs for a SNESType of SNESLS 326 327 Level: beginner 328 329 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 330 331 M*/ 332 333 /*MC 334 SNES_DIVERGED_LOCAL_MIN - the algorithm seems to have stagnated at a local minimum that is not zero. 335 See the manual page for SNESConvergedReason for more details 336 337 Level: beginner 338 339 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 340 341 M*/ 342 343 /*MC 344 SNES_CONERGED_ITERATING - this only occurs if SNESGetConvergedReason() is called during the SNESSolve() 345 346 Level: beginner 347 348 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 349 350 M*/ 351 352 extern PetscErrorCode SNESSetConvergenceTest(SNES,PetscErrorCode (*)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void*,PetscErrorCode (*)(void*)); 353 extern PetscErrorCode SNESDefaultConverged(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*); 354 extern PetscErrorCode SNESSkipConverged(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*); 355 extern PetscErrorCode SNESGetConvergedReason(SNES,SNESConvergedReason*); 356 357 extern PetscErrorCode SNESDMDAComputeFunction(SNES,Vec,Vec,void*); 358 extern PetscErrorCode SNESDMDAComputeJacobianWithAdic(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 359 extern PetscErrorCode SNESDMDAComputeJacobianWithAdifor(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 360 extern PetscErrorCode SNESDMDAComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 361 362 extern PetscErrorCode SNESDMMeshComputeFunction(SNES,Vec,Vec,void*); 363 extern PetscErrorCode SNESDMMeshComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 364 extern PetscErrorCode SNESDMComplexComputeFunction(SNES,Vec,Vec,void *); 365 extern PetscErrorCode SNESDMComplexComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 366 367 /* --------- Solving systems of nonlinear equations --------------- */ 368 typedef PetscErrorCode (*SNESFunction)(SNES,Vec,Vec,void*); 369 typedef PetscErrorCode (*SNESJacobian)(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 370 typedef PetscErrorCode (*SNESGSFunction)(SNES,Vec,Vec,void*); 371 extern PetscErrorCode SNESSetFunction(SNES,Vec,SNESFunction,void*); 372 extern PetscErrorCode SNESGetFunction(SNES,Vec*,SNESFunction*,void**); 373 extern PetscErrorCode SNESComputeFunction(SNES,Vec,Vec); 374 extern PetscErrorCode SNESSetJacobian(SNES,Mat,Mat,SNESJacobian,void*); 375 extern PetscErrorCode SNESGetJacobian(SNES,Mat*,Mat*,SNESJacobian*,void**); 376 extern PetscErrorCode SNESDefaultComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 377 extern PetscErrorCode SNESDefaultComputeJacobianColor(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 378 extern PetscErrorCode SNESSetComputeInitialGuess(SNES,PetscErrorCode (*)(SNES,Vec,void*),void*); 379 extern PetscErrorCode SNESSetPicard(SNES,Vec,SNESFunction,Mat,Mat,SNESJacobian,void*); 380 extern PetscErrorCode SNESGetPicard(SNES,Vec*,SNESFunction*,Mat*,SNESJacobian*,void**); 381 382 extern PetscErrorCode SNESSetGS(SNES,SNESGSFunction,void*); 383 extern PetscErrorCode SNESGetGS(SNES,SNESGSFunction*,void**); 384 extern PetscErrorCode SNESSetUseGS(SNES,PetscBool); 385 extern PetscErrorCode SNESGetUseGS(SNES,PetscBool *); 386 extern PetscErrorCode SNESSetGSSweeps(SNES,PetscInt); 387 extern PetscErrorCode SNESGetGSSweeps(SNES,PetscInt *); 388 extern PetscErrorCode SNESComputeGS(SNES,Vec,Vec); 389 390 /* --------- Routines specifically for line search methods --------------- */ 391 /*E 392 SNESLineSearchType - type of line search used in Newton's method as well as VI solvers and Richardson solvers 393 394 Level: beginner 395 396 .seealso: SNESSetFromOptions(), SNESLineSearchSet() 397 E*/ 398 typedef enum {SNES_LS_BASIC, SNES_LS_BASIC_NONORMS, SNES_LS_QUADRATIC, SNES_LS_CUBIC, SNES_LS_EXACT, SNES_LS_TEST, SNES_LS_CRITICAL, SNES_LS_USER_DEFINED} SNESLineSearchType; 399 extern const char *const SNESLineSearchTypes[]; 400 extern const char *SNESLineSearchTypeName(SNESLineSearchType); /* Does bounds checking, use this for viewing */ 401 402 extern PetscErrorCode SNESLineSearchSet(SNES,PetscErrorCode(*)(SNES,void*,Vec,Vec,Vec,PetscReal,PetscReal,Vec,Vec,PetscReal*,PetscReal*,PetscBool *),void*); 403 extern PetscErrorCode SNESLineSearchSetType(SNES,SNESLineSearchType); 404 extern PetscErrorCode SNESLineSearchNo(SNES,void*,Vec,Vec,Vec,PetscReal,PetscReal,Vec,Vec,PetscReal*,PetscReal*,PetscBool *); 405 extern PetscErrorCode SNESLineSearchNoNorms(SNES,void*,Vec,Vec,Vec,PetscReal,PetscReal,Vec,Vec,PetscReal*,PetscReal*,PetscBool *); 406 extern PetscErrorCode SNESLineSearchQuadratic(SNES,void*,Vec,Vec,Vec,PetscReal,PetscReal,Vec,Vec,PetscReal*,PetscReal*,PetscBool *); 407 extern PetscErrorCode SNESLineSearchCubic(SNES,void*,Vec,Vec,Vec,PetscReal,PetscReal,Vec,Vec,PetscReal*,PetscReal*,PetscBool *); 408 extern PetscErrorCode SNESLineSearchCriticalSecant(SNES,void*,Vec,Vec,Vec,PetscReal,PetscReal,Vec,Vec,PetscReal*,PetscReal*,PetscBool *); 409 extern PetscErrorCode SNESLineSearchQuadraticSecant(SNES,void*,Vec,Vec,Vec,PetscReal,PetscReal,Vec,Vec,PetscReal*,PetscReal*,PetscBool *); 410 411 extern PetscErrorCode SNESLineSearchApply(SNES,Vec,Vec,Vec,PetscReal,PetscReal,Vec,Vec,PetscReal*,PetscReal*,PetscBool *); 412 extern PetscErrorCode SNESLineSearchPreCheckApply(SNES,Vec,Vec,PetscBool*); 413 extern PetscErrorCode SNESLineSearchPostCheckApply(SNES,Vec,Vec,Vec,PetscBool*,PetscBool*); 414 415 extern PetscErrorCode SNESLineSearchSetPostCheck(SNES,PetscErrorCode(*)(SNES,Vec,Vec,Vec,void*,PetscBool *,PetscBool *),void*); 416 extern PetscErrorCode SNESLineSearchSetPreCheck(SNES,PetscErrorCode(*)(SNES,Vec,Vec,void*,PetscBool *),void*); 417 extern PetscErrorCode SNESLineSearchPreCheckPicard(SNES,Vec,Vec,void*,PetscBool*); 418 extern PetscErrorCode SNESLineSearchSetParams(SNES,PetscReal,PetscReal,PetscReal); 419 extern PetscErrorCode SNESLineSearchGetParams(SNES,PetscReal*,PetscReal*,PetscReal*); 420 extern PetscErrorCode SNESLineSearchSetMonitor(SNES,PetscBool ); 421 422 extern PetscErrorCode SNESShellGetContext(SNES,void**); 423 extern PetscErrorCode SNESShellSetContext(SNES,void*); 424 extern PetscErrorCode SNESShellSetSolve(SNES,PetscErrorCode (*)(SNES,Vec)); 425 426 typedef struct _p_LineSearch* PetscLineSearch; 427 428 /* 429 User interface for Line Searches 430 */ 431 432 #define PetscLineSearchType char* 433 #define PETSCLINESEARCHBT "bt" 434 #define PETSCLINESEARCHBASIC "basic" 435 #define PETSCLINESEARCHL2 "l2" 436 #define PETSCLINESEARCHCP "cp" 437 #define PETSCLINESEARCHSHELL "shell" 438 439 extern PetscClassId PETSCLINESEARCH_CLASSID; 440 extern PetscBool PetscLineSearchRegisterAllCalled; 441 extern PetscFList PetscLineSearchList; 442 extern PetscLogEvent PetscLineSearch_Apply; 443 444 typedef PetscErrorCode (*PetscLineSearchPreCheckFunc)(PetscLineSearch,Vec,Vec,PetscBool *); 445 typedef PetscErrorCode (*PetscLineSearchVIProjectFunc)(SNES,Vec); 446 typedef PetscErrorCode (*PetscLineSearchVINormFunc)(SNES,Vec,Vec,PetscReal *); 447 typedef PetscErrorCode (*PetscLineSearchPostCheckFunc)(PetscLineSearch,Vec,Vec,Vec,PetscBool *,PetscBool *); 448 typedef PetscErrorCode (*PetscLineSearchApplyFunc)(PetscLineSearch); 449 typedef PetscErrorCode (*PetscLineSearchUserFunc)(PetscLineSearch, void *); 450 451 extern PetscErrorCode PetscLineSearchCreate(MPI_Comm, PetscLineSearch*); 452 extern PetscErrorCode PetscLineSearchReset(PetscLineSearch); 453 extern PetscErrorCode PetscLineSearchView(PetscLineSearch); 454 extern PetscErrorCode PetscLineSearchDestroy(PetscLineSearch *); 455 extern PetscErrorCode PetscLineSearchSetType(PetscLineSearch, const PetscLineSearchType); 456 extern PetscErrorCode PetscLineSearchSetFromOptions(PetscLineSearch); 457 extern PetscErrorCode PetscLineSearchSetUp(PetscLineSearch); 458 extern PetscErrorCode PetscLineSearchApply(PetscLineSearch, Vec, Vec, PetscReal *, Vec); 459 extern PetscErrorCode PetscLineSearchPreCheck(PetscLineSearch, PetscBool *); 460 extern PetscErrorCode PetscLineSearchPostCheck(PetscLineSearch, PetscBool *, PetscBool *); 461 extern PetscErrorCode PetscLineSearchGetWork(PetscLineSearch, PetscInt); 462 463 /* set the functions for precheck and postcheck */ 464 465 extern PetscErrorCode PetscLineSearchSetPreCheck(PetscLineSearch, PetscLineSearchPreCheckFunc, void *ctx); 466 extern PetscErrorCode PetscLineSearchSetPostCheck(PetscLineSearch, PetscLineSearchPostCheckFunc, void *ctx); 467 468 extern PetscErrorCode PetscLineSearchGetPreCheck(PetscLineSearch, PetscLineSearchPreCheckFunc*, void **ctx); 469 extern PetscErrorCode PetscLineSearchGetPostCheck(PetscLineSearch, PetscLineSearchPostCheckFunc*, void **ctx); 470 471 /* set the functions for VI-specific line search operations */ 472 473 extern PetscErrorCode PetscLineSearchSetVIFunctions(PetscLineSearch, PetscLineSearchVIProjectFunc, PetscLineSearchVINormFunc); 474 extern PetscErrorCode PetscLineSearchGetVIFunctions(PetscLineSearch, PetscLineSearchVIProjectFunc*, PetscLineSearchVINormFunc*); 475 476 /* pointers to the associated SNES in order to be able to get the function evaluation out */ 477 extern PetscErrorCode PetscLineSearchSetSNES(PetscLineSearch,SNES); 478 extern PetscErrorCode PetscLineSearchGetSNES(PetscLineSearch,SNES*); 479 480 /* set and get the parameters and vectors */ 481 extern PetscErrorCode PetscLineSearchGetTolerances(PetscLineSearch,PetscReal*,PetscReal*,PetscReal*,PetscReal*,PetscReal*,PetscInt*); 482 extern PetscErrorCode PetscLineSearchSetTolerances(PetscLineSearch,PetscReal,PetscReal,PetscReal,PetscReal,PetscReal,PetscInt); 483 484 extern PetscErrorCode PetscLineSearchPreCheckPicard(PetscLineSearch,Vec,Vec,PetscBool*); 485 486 extern PetscErrorCode PetscLineSearchGetLambda(PetscLineSearch,PetscReal*); 487 extern PetscErrorCode PetscLineSearchSetLambda(PetscLineSearch,PetscReal); 488 489 extern PetscErrorCode PetscLineSearchGetDamping(PetscLineSearch,PetscReal*); 490 extern PetscErrorCode PetscLineSearchSetDamping(PetscLineSearch,PetscReal); 491 492 extern PetscErrorCode PetscLineSearchGetSuccess(PetscLineSearch, PetscBool*); 493 extern PetscErrorCode PetscLineSearchSetSuccess(PetscLineSearch, PetscBool); 494 495 extern PetscErrorCode PetscLineSearchGetVecs(PetscLineSearch,Vec*,Vec*,Vec*,Vec*,Vec*); 496 extern PetscErrorCode PetscLineSearchSetVecs(PetscLineSearch,Vec,Vec,Vec,Vec,Vec); 497 498 extern PetscErrorCode PetscLineSearchGetNorms(PetscLineSearch, PetscReal *, PetscReal *, PetscReal *); 499 extern PetscErrorCode PetscLineSearchSetNorms(PetscLineSearch, PetscReal, PetscReal, PetscReal); 500 extern PetscErrorCode PetscLineSearchComputeNorms(PetscLineSearch); 501 502 extern PetscErrorCode PetscLineSearchSetMonitor(PetscLineSearch, PetscBool); 503 extern PetscErrorCode PetscLineSearchGetMonitor(PetscLineSearch, PetscViewer*); 504 505 extern PetscErrorCode PetscLineSearchAppendOptionsPrefix(PetscLineSearch, const char prefix[]); 506 extern PetscErrorCode PetscLineSearchGetOptionsPrefix(PetscLineSearch, const char *prefix[]); 507 508 509 /* Shell interface functions */ 510 extern PetscErrorCode PetscLineSearchShellSetUserFunc(PetscLineSearch,PetscLineSearchUserFunc,void*); 511 extern PetscErrorCode PetscLineSearchShellGetUserFunc(PetscLineSearch,PetscLineSearchUserFunc*,void**); 512 513 /*register line search types */ 514 extern PetscErrorCode PetscLineSearchRegister(const char[],const char[],const char[],PetscErrorCode(*)(PetscLineSearch)); 515 extern PetscErrorCode PetscLineSearchRegisterAll(const char path[]); 516 extern PetscErrorCode PetscLineSearchRegisterDestroy(void); 517 518 #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 519 #define PetscLineSearchRegisterDynamic(a,b,c,d) PetscLineSearchRegister(a,b,c,0) 520 #else 521 #define PetscLineSearchRegisterDynamic(a,b,c,d) PetscLineSearchRegister(a,b,c,d) 522 #endif 523 524 /* Routines for VI solver */ 525 extern PetscErrorCode SNESVISetVariableBounds(SNES,Vec,Vec); 526 extern PetscErrorCode SNESVISetComputeVariableBounds(SNES, PetscErrorCode (*)(SNES,Vec,Vec)); 527 extern PetscErrorCode SNESVIGetInactiveSet(SNES,IS*); 528 extern PetscErrorCode SNESVIGetActiveSetIS(SNES,Vec,Vec,IS*); 529 extern PetscErrorCode SNESVIComputeInactiveSetFnorm(SNES,Vec,Vec,PetscReal*); 530 extern PetscErrorCode SNESVISetRedundancyCheck(SNES,PetscErrorCode(*)(SNES,IS,IS*,void*),void*); 531 #define SNES_VI_INF 1.0e20 532 #define SNES_VI_NINF -1.0e20 533 534 extern PetscErrorCode SNESTestLocalMin(SNES); 535 536 /* Should this routine be private? */ 537 extern PetscErrorCode SNESComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*); 538 539 extern PetscErrorCode SNESSetDM(SNES,DM); 540 extern PetscErrorCode SNESGetDM(SNES,DM*); 541 extern PetscErrorCode SNESSetPC(SNES,SNES); 542 extern PetscErrorCode SNESGetPC(SNES,SNES*); 543 extern PetscErrorCode SNESSetPetscLineSearch(SNES,PetscLineSearch); 544 extern PetscErrorCode SNESGetPetscLineSearch(SNES,PetscLineSearch*); 545 extern PetscErrorCode SNESRestrictHookAdd(SNES,PetscErrorCode (*)(SNES,SNES,void*),void*); 546 extern PetscErrorCode SNESRestrictHooksRun(SNES,SNES); 547 548 extern PetscErrorCode SNESSetUpMatrices(SNES); 549 extern PetscErrorCode DMSNESSetFunction(DM,PetscErrorCode(*)(SNES,Vec,Vec,void*),void*); 550 extern PetscErrorCode DMSNESGetFunction(DM,PetscErrorCode(**)(SNES,Vec,Vec,void*),void**); 551 extern PetscErrorCode DMSNESSetGS(DM,PetscErrorCode(*)(SNES,Vec,Vec,void*),void*); 552 extern PetscErrorCode DMSNESGetGS(DM,PetscErrorCode(**)(SNES,Vec,Vec,void*),void**); 553 extern PetscErrorCode DMSNESSetJacobian(DM,PetscErrorCode(*)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void*); 554 extern PetscErrorCode DMSNESGetJacobian(DM,PetscErrorCode(**)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void**); 555 556 extern PetscErrorCode DMDASNESSetFunctionLocal(DM,PetscErrorCode(*)(DMDALocalInfo*,void*,void*,void*),void*); 557 extern PetscErrorCode DMDASNESSetJacobianLocal(DM,PetscErrorCode(*)(DMDALocalInfo*,void*,Mat,Mat,MatStructure*,void*),void*); 558 559 /* Routines for Multiblock solver */ 560 extern PetscErrorCode SNESMultiblockSetFields(SNES, const char [], PetscInt, const PetscInt *); 561 extern PetscErrorCode SNESMultiblockSetIS(SNES, const char [], IS); 562 extern PetscErrorCode SNESMultiblockSetBlockSize(SNES, PetscInt); 563 extern PetscErrorCode SNESMultiblockSetType(SNES, PCCompositeType); 564 565 /*J 566 SNESMSType - String with the name of a PETSc SNESMS method. 567 568 Level: intermediate 569 570 .seealso: SNESMSSetType(), SNES 571 J*/ 572 #define SNESMSType char* 573 #define SNESMSM62 "m62" 574 #define SNESMSEULER "euler" 575 #define SNESMSJAMESON83 "jameson83" 576 #define SNESMSVLTP21 "vltp21" 577 #define SNESMSVLTP31 "vltp31" 578 #define SNESMSVLTP41 "vltp41" 579 #define SNESMSVLTP51 "vltp51" 580 #define SNESMSVLTP61 "vltp61" 581 582 extern PetscErrorCode SNESMSRegister(const SNESMSType,PetscInt,PetscInt,PetscReal,const PetscReal[],const PetscReal[],const PetscReal[]); 583 extern PetscErrorCode SNESMSSetType(SNES,const SNESMSType); 584 extern PetscErrorCode SNESMSFinalizePackage(void); 585 extern PetscErrorCode SNESMSInitializePackage(const char path[]); 586 extern PetscErrorCode SNESMSRegisterDestroy(void); 587 extern PetscErrorCode SNESMSRegisterAll(void); 588 589 PETSC_EXTERN_CXX_END 590 #endif 591