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 9 /*S 10 SNES - Abstract PETSc object that manages all nonlinear solves 11 12 Level: beginner 13 14 Concepts: nonlinear solvers 15 16 .seealso: SNESCreate(), SNESSetType(), SNESType, TS, KSP, KSP, PC 17 S*/ 18 typedef struct _p_SNES* SNES; 19 20 /*J 21 SNESType - String with the name of a PETSc SNES method or the creation function 22 with an optional dynamic library name, for example 23 http://www.mcs.anl.gov/petsc/lib.a:mysnescreate() 24 25 Level: beginner 26 27 .seealso: SNESSetType(), SNES 28 J*/ 29 typedef const char* SNESType; 30 #define SNESLS "ls" 31 #define SNESTR "tr" 32 #define SNESPYTHON "python" 33 #define SNESTEST "test" 34 #define SNESNRICHARDSON "nrichardson" 35 #define SNESKSPONLY "ksponly" 36 #define SNESVIRS "virs" 37 #define SNESVISS "viss" 38 #define SNESNGMRES "ngmres" 39 #define SNESQN "qn" 40 #define SNESSHELL "shell" 41 #define SNESGS "gs" 42 #define SNESNCG "ncg" 43 #define SNESFAS "fas" 44 #define SNESMS "ms" 45 #define SNESNASM "nasm" 46 47 /* Logging support */ 48 PETSC_EXTERN PetscClassId SNES_CLASSID; 49 50 PETSC_EXTERN PetscErrorCode SNESInitializePackage(const char[]); 51 52 PETSC_EXTERN PetscErrorCode SNESCreate(MPI_Comm,SNES*); 53 PETSC_EXTERN PetscErrorCode SNESReset(SNES); 54 PETSC_EXTERN PetscErrorCode SNESDestroy(SNES*); 55 PETSC_EXTERN PetscErrorCode SNESSetType(SNES,SNESType); 56 PETSC_EXTERN PetscErrorCode SNESMonitor(SNES,PetscInt,PetscReal); 57 PETSC_EXTERN PetscErrorCode SNESMonitorSet(SNES,PetscErrorCode(*)(SNES,PetscInt,PetscReal,void*),void *,PetscErrorCode (*)(void**)); 58 PETSC_EXTERN PetscErrorCode SNESMonitorCancel(SNES); 59 PETSC_EXTERN PetscErrorCode SNESSetConvergenceHistory(SNES,PetscReal[],PetscInt[],PetscInt,PetscBool ); 60 PETSC_EXTERN PetscErrorCode SNESGetConvergenceHistory(SNES,PetscReal*[],PetscInt *[],PetscInt *); 61 PETSC_EXTERN PetscErrorCode SNESSetUp(SNES); 62 PETSC_EXTERN PetscErrorCode SNESSolve(SNES,Vec,Vec); 63 PETSC_EXTERN PetscErrorCode SNESSetErrorIfNotConverged(SNES,PetscBool ); 64 PETSC_EXTERN PetscErrorCode SNESGetErrorIfNotConverged(SNES,PetscBool *); 65 66 67 PETSC_EXTERN PetscErrorCode SNESAddOptionsChecker(PetscErrorCode (*)(SNES)); 68 69 PETSC_EXTERN PetscErrorCode SNESSetUpdate(SNES, PetscErrorCode (*)(SNES, PetscInt)); 70 PETSC_EXTERN PetscErrorCode SNESDefaultUpdate(SNES, PetscInt); 71 72 PETSC_EXTERN PetscFList SNESList; 73 PETSC_EXTERN PetscErrorCode SNESRegisterDestroy(void); 74 PETSC_EXTERN PetscErrorCode SNESRegisterAll(const char[]); 75 76 PETSC_EXTERN PetscErrorCode SNESRegister(const char[],const char[],const char[],PetscErrorCode (*)(SNES)); 77 78 /*MC 79 SNESRegisterDynamic - Adds a method to the nonlinear solver package. 80 81 Synopsis: 82 PetscErrorCode SNESRegisterDynamic(const char *name_solver,const char *path,const char *name_create,PetscErrorCode (*routine_create)(SNES)) 83 84 Not collective 85 86 Input Parameters: 87 + name_solver - name of a new user-defined solver 88 . path - path (either absolute or relative) the library containing this solver 89 . name_create - name of routine to create method context 90 - routine_create - routine to create method context 91 92 Notes: 93 SNESRegisterDynamic() may be called multiple times to add several user-defined solvers. 94 95 If dynamic libraries are used, then the fourth input argument (routine_create) 96 is ignored. 97 98 Environmental variables such as ${PETSC_ARCH}, ${PETSC_DIR}, ${PETSC_LIB_DIR}, 99 and others of the form ${any_environmental_variable} occuring in pathname will be 100 replaced with appropriate values. 101 102 Sample usage: 103 .vb 104 SNESRegisterDynamic("my_solver",/home/username/my_lib/lib/libg/solaris/mylib.a, 105 "MySolverCreate",MySolverCreate); 106 .ve 107 108 Then, your solver can be chosen with the procedural interface via 109 $ SNESSetType(snes,"my_solver") 110 or at runtime via the option 111 $ -snes_type my_solver 112 113 Level: advanced 114 115 Note: If your function is not being put into a shared library then use SNESRegister() instead 116 117 .keywords: SNES, nonlinear, register 118 119 .seealso: SNESRegisterAll(), SNESRegisterDestroy() 120 M*/ 121 #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 122 #define SNESRegisterDynamic(a,b,c,d) SNESRegister(a,b,c,0) 123 #else 124 #define SNESRegisterDynamic(a,b,c,d) SNESRegister(a,b,c,d) 125 #endif 126 127 PETSC_EXTERN PetscErrorCode SNESGetKSP(SNES,KSP*); 128 PETSC_EXTERN PetscErrorCode SNESSetKSP(SNES,KSP); 129 PETSC_EXTERN PetscErrorCode SNESGetSolution(SNES,Vec*); 130 PETSC_EXTERN PetscErrorCode SNESGetSolutionUpdate(SNES,Vec*); 131 PETSC_EXTERN PetscErrorCode SNESGetRhs(SNES,Vec*); 132 PETSC_EXTERN PetscErrorCode SNESView(SNES,PetscViewer); 133 PETSC_EXTERN PetscErrorCode SNESLoad(SNES,PetscViewer); 134 135 #define SNES_FILE_CLASSID 1211224 136 137 PETSC_EXTERN PetscErrorCode SNESSetOptionsPrefix(SNES,const char[]); 138 PETSC_EXTERN PetscErrorCode SNESAppendOptionsPrefix(SNES,const char[]); 139 PETSC_EXTERN PetscErrorCode SNESGetOptionsPrefix(SNES,const char*[]); 140 PETSC_EXTERN PetscErrorCode SNESSetFromOptions(SNES); 141 PETSC_EXTERN PetscErrorCode SNESDefaultGetWork(SNES,PetscInt); 142 143 PETSC_EXTERN PetscErrorCode MatCreateSNESMF(SNES,Mat*); 144 PETSC_EXTERN PetscErrorCode MatMFFDComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 145 146 PETSC_EXTERN PetscErrorCode MatDAADSetSNES(Mat,SNES); 147 148 PETSC_EXTERN PetscErrorCode SNESGetType(SNES,SNESType*); 149 PETSC_EXTERN PetscErrorCode SNESMonitorDefault(SNES,PetscInt,PetscReal,void *); 150 PETSC_EXTERN PetscErrorCode SNESMonitorRange(SNES,PetscInt,PetscReal,void *); 151 PETSC_EXTERN PetscErrorCode SNESMonitorRatio(SNES,PetscInt,PetscReal,void *); 152 PETSC_EXTERN PetscErrorCode SNESMonitorSetRatio(SNES,PetscViewer); 153 PETSC_EXTERN PetscErrorCode SNESMonitorSolution(SNES,PetscInt,PetscReal,void *); 154 PETSC_EXTERN PetscErrorCode SNESMonitorResidual(SNES,PetscInt,PetscReal,void *); 155 PETSC_EXTERN PetscErrorCode SNESMonitorSolutionUpdate(SNES,PetscInt,PetscReal,void *); 156 PETSC_EXTERN PetscErrorCode SNESMonitorDefaultShort(SNES,PetscInt,PetscReal,void *); 157 PETSC_EXTERN PetscErrorCode SNESMonitorJacUpdateSpectrum(SNES,PetscInt,PetscReal,void *); 158 PETSC_EXTERN PetscErrorCode SNESSetTolerances(SNES,PetscReal,PetscReal,PetscReal,PetscInt,PetscInt); 159 PETSC_EXTERN PetscErrorCode SNESGetTolerances(SNES,PetscReal*,PetscReal*,PetscReal*,PetscInt*,PetscInt*); 160 PETSC_EXTERN PetscErrorCode SNESSetTrustRegionTolerance(SNES,PetscReal); 161 PETSC_EXTERN PetscErrorCode SNESGetFunctionNorm(SNES,PetscReal*); 162 PETSC_EXTERN PetscErrorCode SNESSetFunctionNorm(SNES,PetscReal); 163 PETSC_EXTERN PetscErrorCode SNESGetIterationNumber(SNES,PetscInt*); 164 PETSC_EXTERN PetscErrorCode SNESSetIterationNumber(SNES,PetscInt); 165 166 PETSC_EXTERN PetscErrorCode SNESGetNonlinearStepFailures(SNES,PetscInt*); 167 PETSC_EXTERN PetscErrorCode SNESSetMaxNonlinearStepFailures(SNES,PetscInt); 168 PETSC_EXTERN PetscErrorCode SNESGetMaxNonlinearStepFailures(SNES,PetscInt*); 169 PETSC_EXTERN PetscErrorCode SNESGetNumberFunctionEvals(SNES,PetscInt*); 170 171 PETSC_EXTERN PetscErrorCode SNESSetLagPreconditioner(SNES,PetscInt); 172 PETSC_EXTERN PetscErrorCode SNESGetLagPreconditioner(SNES,PetscInt*); 173 PETSC_EXTERN PetscErrorCode SNESSetLagJacobian(SNES,PetscInt); 174 PETSC_EXTERN PetscErrorCode SNESGetLagJacobian(SNES,PetscInt*); 175 PETSC_EXTERN PetscErrorCode SNESSetGridSequence(SNES,PetscInt); 176 177 PETSC_EXTERN PetscErrorCode SNESGetLinearSolveIterations(SNES,PetscInt*); 178 PETSC_EXTERN PetscErrorCode SNESGetLinearSolveFailures(SNES,PetscInt*); 179 PETSC_EXTERN PetscErrorCode SNESSetMaxLinearSolveFailures(SNES,PetscInt); 180 PETSC_EXTERN PetscErrorCode SNESGetMaxLinearSolveFailures(SNES,PetscInt*); 181 182 PETSC_EXTERN PetscErrorCode SNESKSPSetUseEW(SNES,PetscBool ); 183 PETSC_EXTERN PetscErrorCode SNESKSPGetUseEW(SNES,PetscBool *); 184 PETSC_EXTERN PetscErrorCode SNESKSPSetParametersEW(SNES,PetscInt,PetscReal,PetscReal,PetscReal,PetscReal,PetscReal,PetscReal); 185 PETSC_EXTERN PetscErrorCode SNESKSPGetParametersEW(SNES,PetscInt*,PetscReal*,PetscReal*,PetscReal*,PetscReal*,PetscReal*,PetscReal*); 186 187 PETSC_EXTERN PetscErrorCode SNESMonitorLGCreate(const char[],const char[],int,int,int,int,PetscDrawLG*); 188 PETSC_EXTERN PetscErrorCode SNESMonitorLGResidualNorm(SNES,PetscInt,PetscReal,void*); 189 PETSC_EXTERN PetscErrorCode SNESMonitorLGDestroy(PetscDrawLG*); 190 PETSC_EXTERN PetscErrorCode SNESMonitorLGRange(SNES,PetscInt,PetscReal,void*); 191 192 PETSC_EXTERN PetscErrorCode SNESSetApplicationContext(SNES,void *); 193 PETSC_EXTERN PetscErrorCode SNESGetApplicationContext(SNES,void *); 194 PETSC_EXTERN PetscErrorCode SNESSetComputeApplicationContext(SNES,PetscErrorCode (*)(SNES,void**),PetscErrorCode (*)(void**)); 195 196 PETSC_EXTERN PetscErrorCode SNESPythonSetType(SNES,const char[]); 197 198 PETSC_EXTERN PetscErrorCode SNESSetFunctionDomainError(SNES); 199 PETSC_EXTERN PetscErrorCode SNESGetFunctionDomainError(SNES, PetscBool *); 200 201 /*E 202 SNESConvergedReason - reason a SNES method was said to 203 have converged or diverged 204 205 Level: beginner 206 207 The two most common reasons for divergence are 208 $ 1) an incorrectly coded or computed Jacobian or 209 $ 2) failure or lack of convergence in the linear system (in this case we recommend 210 $ testing with -pc_type lu to eliminate the linear solver as the cause of the problem). 211 212 Diverged Reasons: 213 . SNES_DIVERGED_LOCAL_MIN - this can only occur when using the line-search variant of SNES. 214 The line search wants to minimize Q(alpha) = 1/2 || F(x + alpha s) ||^2_2 this occurs 215 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 216 you get Q'(alpha) = -F(x)^T F'(x)^(-1)^T F'(x+alpha s)F(x+alpha s); when alpha = 0 217 Q'(0) = - ||F(x)||^2_2 which is always NEGATIVE if F'(x) is invertible. This means the Newton 218 direction is a descent direction and the line search should succeed if alpha is small enough. 219 220 If F'(x) is NOT invertible AND F'(x)^T F(x) = 0 then Q'(0) = 0 and the Newton direction 221 is NOT a descent direction so the line search will fail. All one can do at this point 222 is change the initial guess and try again. 223 224 An alternative explanation: Newton's method can be regarded as replacing the function with 225 its linear approximation and minimizing the 2-norm of that. That is F(x+s) approx F(x) + F'(x)s 226 so we minimize || F(x) + F'(x) s ||^2_2; do this using Least Squares. If F'(x) is invertible then 227 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 228 exists a nontrival (that is F'(x)s != 0) solution to the equation and this direction is 229 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) 230 = - (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 231 and F'(x)^T F'(x) has no negative eigenvalues Q'(0) < 0 so s is a descent direction and the line 232 search should succeed for small enough alpha. 233 234 Note that this RARELY happens in practice. Far more likely the linear system is not being solved 235 (well enough?) or the Jacobian is wrong. 236 237 SNES_DIVERGED_MAX_IT means that the solver reached the maximum number of iterations without satisfying any 238 convergence criteria. SNES_CONVERGED_ITS means that SNESSkipConverged() was chosen as the convergence test; 239 thus the usual convergence criteria have not been checked and may or may not be satisfied. 240 241 Developer Notes: this must match finclude/petscsnes.h 242 243 The string versions of these are in SNESConvergedReason, if you change any value here you must 244 also adjust that array. 245 246 Each reason has its own manual page. 247 248 .seealso: SNESSolve(), SNESGetConvergedReason(), KSPConvergedReason, SNESSetConvergenceTest() 249 E*/ 250 typedef enum {/* converged */ 251 SNES_CONVERGED_FNORM_ABS = 2, /* ||F|| < atol */ 252 SNES_CONVERGED_FNORM_RELATIVE = 3, /* ||F|| < rtol*||F_initial|| */ 253 SNES_CONVERGED_SNORM_RELATIVE = 4, /* Newton computed step size small; || delta x || < stol */ 254 SNES_CONVERGED_ITS = 5, /* maximum iterations reached */ 255 SNES_CONVERGED_TR_DELTA = 7, 256 /* diverged */ 257 SNES_DIVERGED_FUNCTION_DOMAIN = -1, /* the new x location passed the function is not in the domain of F */ 258 SNES_DIVERGED_FUNCTION_COUNT = -2, 259 SNES_DIVERGED_LINEAR_SOLVE = -3, /* the linear solve failed */ 260 SNES_DIVERGED_FNORM_NAN = -4, 261 SNES_DIVERGED_MAX_IT = -5, 262 SNES_DIVERGED_LINE_SEARCH = -6, /* the line search failed */ 263 SNES_DIVERGED_INNER = -7, /* inner solve failed */ 264 SNES_DIVERGED_LOCAL_MIN = -8, /* || J^T b || is small, implies converged to local minimum of F() */ 265 SNES_CONVERGED_ITERATING = 0} SNESConvergedReason; 266 PETSC_EXTERN const char *const*SNESConvergedReasons; 267 268 /*MC 269 SNES_CONVERGED_FNORM_ABS - 2-norm(F) <= abstol 270 271 Level: beginner 272 273 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 274 275 M*/ 276 277 /*MC 278 SNES_CONVERGED_FNORM_RELATIVE - 2-norm(F) <= rtol*2-norm(F(x_0)) where x_0 is the initial guess 279 280 Level: beginner 281 282 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 283 284 M*/ 285 286 /*MC 287 SNES_CONVERGED_SNORM_RELATIVE - The 2-norm of the last step <= stol * 2-norm(x) where x is the current 288 solution and stol is the 4th argument to SNESSetTolerances() 289 290 Options Database Keys: 291 -snes_stol <stol> - the step tolerance 292 293 Level: beginner 294 295 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 296 297 M*/ 298 299 /*MC 300 SNES_DIVERGED_FUNCTION_COUNT - The user provided function has been called more times then the final 301 argument to SNESSetTolerances() 302 303 Level: beginner 304 305 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 306 307 M*/ 308 309 /*MC 310 SNES_DIVERGED_FNORM_NAN - the 2-norm of the current function evaluation is not-a-number (NaN), this 311 is usually caused by a division of 0 by 0. 312 313 Level: beginner 314 315 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 316 317 M*/ 318 319 /*MC 320 SNES_DIVERGED_MAX_IT - SNESSolve() has reached the maximum number of iterations requested 321 322 Level: beginner 323 324 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 325 326 M*/ 327 328 /*MC 329 SNES_DIVERGED_LINE_SEARCH - The line search has failed. This only occurs for a SNESType of SNESLS 330 331 Level: beginner 332 333 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 334 335 M*/ 336 337 /*MC 338 SNES_DIVERGED_LOCAL_MIN - the algorithm seems to have stagnated at a local minimum that is not zero. 339 See the manual page for SNESConvergedReason for more details 340 341 Level: beginner 342 343 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 344 345 M*/ 346 347 /*MC 348 SNES_CONERGED_ITERATING - this only occurs if SNESGetConvergedReason() is called during the SNESSolve() 349 350 Level: beginner 351 352 .seealso: SNESSolve(), SNESGetConvergedReason(), SNESConvergedReason, SNESSetTolerances() 353 354 M*/ 355 356 PETSC_EXTERN PetscErrorCode SNESSetConvergenceTest(SNES,PetscErrorCode (*)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void*,PetscErrorCode (*)(void*)); 357 PETSC_EXTERN PetscErrorCode SNESDefaultConverged(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*); 358 PETSC_EXTERN PetscErrorCode SNESSkipConverged(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*); 359 PETSC_EXTERN PetscErrorCode SNESGetConvergedReason(SNES,SNESConvergedReason*); 360 361 PETSC_EXTERN PetscErrorCode SNESDMMeshComputeFunction(SNES,Vec,Vec,void*); 362 PETSC_EXTERN PetscErrorCode SNESDMMeshComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 363 364 /* --------- Solving systems of nonlinear equations --------------- */ 365 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESFunction)(SNES,Vec,Vec,void*); 366 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESJacobian)(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 367 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESObjective)(SNES,Vec,PetscReal *,void*); 368 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESGSFunction)(SNES,Vec,Vec,void*); 369 PETSC_EXTERN PetscErrorCode SNESSetFunction(SNES,Vec,SNESFunction,void*); 370 PETSC_EXTERN PetscErrorCode SNESGetFunction(SNES,Vec*,SNESFunction*,void**); 371 PETSC_EXTERN PetscErrorCode SNESComputeFunction(SNES,Vec,Vec); 372 PETSC_EXTERN PetscErrorCode SNESSetJacobian(SNES,Mat,Mat,SNESJacobian,void*); 373 PETSC_EXTERN PetscErrorCode SNESGetJacobian(SNES,Mat*,Mat*,SNESJacobian*,void**); 374 PETSC_EXTERN PetscErrorCode SNESDefaultObjectiveComputeFunctionFD(SNES,Vec,Vec,void*); 375 PETSC_EXTERN PetscErrorCode SNESDefaultComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 376 PETSC_EXTERN PetscErrorCode SNESDefaultComputeJacobianColor(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 377 PETSC_EXTERN PetscErrorCode SNESSetComputeInitialGuess(SNES,PetscErrorCode (*)(SNES,Vec,void*),void*); 378 PETSC_EXTERN PetscErrorCode SNESSetPicard(SNES,Vec,SNESFunction,Mat,Mat,SNESJacobian,void*); 379 PETSC_EXTERN PetscErrorCode SNESGetPicard(SNES,Vec*,SNESFunction*,Mat*,Mat*,SNESJacobian*,void**); 380 PETSC_EXTERN PetscErrorCode SNESSetInitialFunction(SNES,Vec); 381 PETSC_EXTERN PetscErrorCode SNESSetInitialFunctionNorm(SNES,PetscReal); 382 383 PETSC_EXTERN PetscErrorCode SNESSetObjective(SNES,SNESObjective,void*); 384 PETSC_EXTERN PetscErrorCode SNESGetObjective(SNES,SNESObjective*,void**); 385 PETSC_EXTERN PetscErrorCode SNESComputeObjective(SNES,Vec,PetscReal *); 386 387 /*E 388 SNESNormType - Norm that is passed in the Krylov convergence 389 test routines. 390 391 Level: advanced 392 393 Support for these is highly dependent on the solver. 394 395 Notes: 396 This is primarily used to turn off extra norm and function computation 397 when the solvers are composed. 398 399 .seealso: SNESSolve(), SNESGetConvergedReason(), KSPSetNormType(), 400 KSPSetConvergenceTest(), KSPSetPCSide() 401 E*/ 402 403 typedef enum {SNES_NORM_DEFAULT = -1, 404 SNES_NORM_NONE = 0, 405 SNES_NORM_FUNCTION = 1, 406 SNES_NORM_INITIAL_ONLY = 2, 407 SNES_NORM_FINAL_ONLY = 3, 408 SNES_NORM_INITIAL_FINAL_ONLY = 4} SNESNormType; 409 PETSC_EXTERN const char *const*const SNESNormTypes; 410 /*MC 411 SNES_NORM_NONE - Don't compute function and its L2 norm. 412 413 Level: advanced 414 415 Notes: 416 This is most useful for stationary solvers with a fixed number of iterations used as smoothers. 417 418 .seealso: SNESNormType, SNESSetNormType(), SNES_NORM_DEFAULT 419 M*/ 420 421 /*MC 422 SNES_NORM_FUNCTION - Compute the function and its L2 norm at each iteration. 423 424 Level: advanced 425 426 Notes: 427 Most solvers will use this no matter what norm type is passed to them. 428 429 .seealso: SNESNormType, SNESSetNormType(), SNES_NORM_NONE 430 M*/ 431 432 /*MC 433 SNES_NORM_INITIAL_ONLY - Compute the function and its L2 at iteration 0, but do not update it. 434 435 Level: advanced 436 437 Notes: 438 This method is useful in composed methods, when a true solution might actually be found before SNESSolve() is called. 439 This option enables the solve to abort on the zeroth iteration if this is the case. 440 441 For solvers that require the computation of the L2 norm of the function as part of the method, this merely cancels 442 the norm computation at the last iteration (if possible). 443 444 .seealso: SNESNormType, SNESSetNormType(), SNES_NORM_FINAL_ONLY, SNES_NORM_INITIAL_FINAL_ONLY 445 M*/ 446 447 /*MC 448 SNES_NORM_FINAL_ONLY - Compute the function and its L2 norm on only the final iteration. 449 450 Level: advanced 451 452 Notes: 453 For solvers that require the computation of the L2 norm of the function as part of the method, behaves 454 exactly as SNES_NORM_DEFAULT. This method is useful when the function is gotten after SNESSolve and 455 used in subsequent computation for methods that do not need the norm computed during the rest of the 456 solution procedure. 457 458 .seealso: SNESNormType, SNESSetNormType(), SNES_NORM_INITIAL_ONLY, SNES_NORM_INITIAL_FINAL_ONLY 459 M*/ 460 461 /*MC 462 SNES_NORM_INITIAL_FINAL_ONLY - Compute the function and its L2 norm on only the initial and final iterations. 463 464 Level: advanced 465 466 Notes: 467 This method combines the benefits of SNES_NORM_INITIAL_ONLY and SNES_NORM_FINAL_ONLY. 468 469 .seealso: SNESNormType, SNESSetNormType(), SNES_NORM_SNES_NORM_INITIAL_ONLY, SNES_NORM_FINAL_ONLY 470 M*/ 471 472 473 PETSC_EXTERN PetscErrorCode SNESSetNormType(SNES,SNESNormType); 474 PETSC_EXTERN PetscErrorCode SNESGetNormType(SNES,SNESNormType*); 475 476 PETSC_EXTERN PetscErrorCode SNESSetGS(SNES,SNESGSFunction,void*); 477 PETSC_EXTERN PetscErrorCode SNESGetGS(SNES,SNESGSFunction*,void**); 478 PETSC_EXTERN PetscErrorCode SNESSetUseGS(SNES,PetscBool); 479 PETSC_EXTERN PetscErrorCode SNESGetUseGS(SNES,PetscBool *); 480 PETSC_EXTERN PetscErrorCode SNESSetGSSweeps(SNES,PetscInt); 481 PETSC_EXTERN PetscErrorCode SNESGetGSSweeps(SNES,PetscInt *); 482 PETSC_EXTERN PetscErrorCode SNESComputeGS(SNES,Vec,Vec); 483 484 PETSC_EXTERN PetscErrorCode SNESShellGetContext(SNES,void**); 485 PETSC_EXTERN PetscErrorCode SNESShellSetContext(SNES,void*); 486 PETSC_EXTERN PetscErrorCode SNESShellSetSolve(SNES,PetscErrorCode (*)(SNES,Vec)); 487 488 /* --------- Routines specifically for line search methods --------------- */ 489 490 typedef struct _p_LineSearch* SNESLineSearch; 491 492 /*S 493 SNESLineSearch - Abstract PETSc object that manages line-search operations 494 495 Level: beginner 496 497 Concepts: nonlinear solvers, line search 498 499 .seealso: SNESLineSearchCreate(), SNESLineSearchSetType(), SNES 500 S*/ 501 502 /*J 503 SNESLineSearchType - String with the name of a PETSc line search method 504 505 Level: beginner 506 507 .seealso: SNESLineSearchSetType(), SNES 508 J*/ 509 510 typedef const char* SNESLineSearchType; 511 #define SNESLINESEARCHBT "bt" 512 #define SNESLINESEARCHBASIC "basic" 513 #define SNESLINESEARCHL2 "l2" 514 #define SNESLINESEARCHCP "cp" 515 #define SNESLINESEARCHSHELL "shell" 516 517 PETSC_EXTERN PetscClassId SNESLINESEARCH_CLASSID; 518 PETSC_EXTERN PetscBool SNESLineSearchRegisterAllCalled; 519 PETSC_EXTERN PetscFList SNESLineSearchList; 520 PETSC_EXTERN PetscLogEvent SNESLineSearch_Apply; 521 522 #define SNES_LINESEARCH_ORDER_LINEAR 1 523 #define SNES_LINESEARCH_ORDER_QUADRATIC 2 524 #define SNES_LINESEARCH_ORDER_CUBIC 3 525 526 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESLineSearchPreCheckFunc)(SNESLineSearch,Vec,Vec,PetscBool*,void*); 527 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESLineSearchVIProjectFunc)(SNES,Vec); 528 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESLineSearchVINormFunc)(SNES,Vec,Vec,PetscReal *); 529 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESLineSearchPostCheckFunc)(SNESLineSearch,Vec,Vec,Vec,PetscBool *,PetscBool *,void*); 530 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESLineSearchApplyFunc)(SNESLineSearch); 531 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*SNESLineSearchUserFunc)(SNESLineSearch, void *); 532 533 PETSC_EXTERN PetscErrorCode SNESLineSearchCreate(MPI_Comm, SNESLineSearch*); 534 PETSC_EXTERN PetscErrorCode SNESLineSearchReset(SNESLineSearch); 535 PETSC_EXTERN PetscErrorCode SNESLineSearchView(SNESLineSearch,PetscViewer); 536 PETSC_EXTERN PetscErrorCode SNESLineSearchDestroy(SNESLineSearch *); 537 PETSC_EXTERN PetscErrorCode SNESLineSearchSetType(SNESLineSearch, SNESLineSearchType); 538 PETSC_EXTERN PetscErrorCode SNESLineSearchSetFromOptions(SNESLineSearch); 539 PETSC_EXTERN PetscErrorCode SNESLineSearchSetUp(SNESLineSearch); 540 PETSC_EXTERN PetscErrorCode SNESLineSearchApply(SNESLineSearch, Vec, Vec, PetscReal *, Vec); 541 PETSC_EXTERN PetscErrorCode SNESLineSearchPreCheck(SNESLineSearch,Vec,Vec,PetscBool *); 542 PETSC_EXTERN PetscErrorCode SNESLineSearchPostCheck(SNESLineSearch,Vec,Vec,Vec,PetscBool *,PetscBool *); 543 PETSC_EXTERN PetscErrorCode SNESLineSearchGetWork(SNESLineSearch, PetscInt); 544 545 /* set the functions for precheck and postcheck */ 546 547 PETSC_EXTERN PetscErrorCode SNESLineSearchSetPreCheck(SNESLineSearch, SNESLineSearchPreCheckFunc, void *ctx); 548 PETSC_EXTERN PetscErrorCode SNESLineSearchSetPostCheck(SNESLineSearch, SNESLineSearchPostCheckFunc, void *ctx); 549 550 PETSC_EXTERN PetscErrorCode SNESLineSearchGetPreCheck(SNESLineSearch, SNESLineSearchPreCheckFunc*, void **ctx); 551 PETSC_EXTERN PetscErrorCode SNESLineSearchGetPostCheck(SNESLineSearch, SNESLineSearchPostCheckFunc*, void **ctx); 552 553 /* set the functions for VI-specific line search operations */ 554 555 PETSC_EXTERN PetscErrorCode SNESLineSearchSetVIFunctions(SNESLineSearch, SNESLineSearchVIProjectFunc, SNESLineSearchVINormFunc); 556 PETSC_EXTERN PetscErrorCode SNESLineSearchGetVIFunctions(SNESLineSearch, SNESLineSearchVIProjectFunc*, SNESLineSearchVINormFunc*); 557 558 /* pointers to the associated SNES in order to be able to get the function evaluation out */ 559 PETSC_EXTERN PetscErrorCode SNESLineSearchSetSNES(SNESLineSearch,SNES); 560 PETSC_EXTERN PetscErrorCode SNESLineSearchGetSNES(SNESLineSearch,SNES*); 561 562 /* set and get the parameters and vectors */ 563 PETSC_EXTERN PetscErrorCode SNESLineSearchGetTolerances(SNESLineSearch,PetscReal*,PetscReal*,PetscReal*,PetscReal*,PetscReal*,PetscInt*); 564 PETSC_EXTERN PetscErrorCode SNESLineSearchSetTolerances(SNESLineSearch,PetscReal,PetscReal,PetscReal,PetscReal,PetscReal,PetscInt); 565 566 PETSC_EXTERN PetscErrorCode SNESLineSearchPreCheckPicard(SNESLineSearch,Vec,Vec,PetscBool*,void*); 567 568 PETSC_EXTERN PetscErrorCode SNESLineSearchGetLambda(SNESLineSearch,PetscReal*); 569 PETSC_EXTERN PetscErrorCode SNESLineSearchSetLambda(SNESLineSearch,PetscReal); 570 571 PETSC_EXTERN PetscErrorCode SNESLineSearchGetDamping(SNESLineSearch,PetscReal*); 572 PETSC_EXTERN PetscErrorCode SNESLineSearchSetDamping(SNESLineSearch,PetscReal); 573 574 PETSC_EXTERN PetscErrorCode SNESLineSearchGetOrder(SNESLineSearch,PetscInt *order); 575 PETSC_EXTERN PetscErrorCode SNESLineSearchSetOrder(SNESLineSearch,PetscInt order); 576 577 PETSC_EXTERN PetscErrorCode SNESLineSearchGetSuccess(SNESLineSearch, PetscBool*); 578 PETSC_EXTERN PetscErrorCode SNESLineSearchSetSuccess(SNESLineSearch, PetscBool); 579 580 PETSC_EXTERN PetscErrorCode SNESLineSearchGetVecs(SNESLineSearch,Vec*,Vec*,Vec*,Vec*,Vec*); 581 PETSC_EXTERN PetscErrorCode SNESLineSearchSetVecs(SNESLineSearch,Vec,Vec,Vec,Vec,Vec); 582 583 PETSC_EXTERN PetscErrorCode SNESLineSearchGetNorms(SNESLineSearch, PetscReal *, PetscReal *, PetscReal *); 584 PETSC_EXTERN PetscErrorCode SNESLineSearchSetNorms(SNESLineSearch, PetscReal, PetscReal, PetscReal); 585 PETSC_EXTERN PetscErrorCode SNESLineSearchComputeNorms(SNESLineSearch); 586 PETSC_EXTERN PetscErrorCode SNESLineSearchSetComputeNorms(SNESLineSearch, PetscBool); 587 588 PETSC_EXTERN PetscErrorCode SNESLineSearchSetMonitor(SNESLineSearch, PetscBool); 589 PETSC_EXTERN PetscErrorCode SNESLineSearchGetMonitor(SNESLineSearch, PetscViewer*); 590 591 PETSC_EXTERN PetscErrorCode SNESLineSearchAppendOptionsPrefix(SNESLineSearch, const char prefix[]); 592 PETSC_EXTERN PetscErrorCode SNESLineSearchGetOptionsPrefix(SNESLineSearch, const char *prefix[]); 593 594 595 /* Shell interface functions */ 596 PETSC_EXTERN PetscErrorCode SNESLineSearchShellSetUserFunc(SNESLineSearch,SNESLineSearchUserFunc,void*); 597 PETSC_EXTERN PetscErrorCode SNESLineSearchShellGetUserFunc(SNESLineSearch,SNESLineSearchUserFunc*,void**); 598 599 /* BT interface functions */ 600 PETSC_EXTERN PetscErrorCode SNESLineSearchBTSetAlpha(SNESLineSearch, PetscReal); 601 PETSC_EXTERN PetscErrorCode SNESLineSearchBTGetAlpha(SNESLineSearch, PetscReal*); 602 603 /*register line search types */ 604 PETSC_EXTERN PetscErrorCode SNESLineSearchRegister(const char[],const char[],const char[],PetscErrorCode(*)(SNESLineSearch)); 605 PETSC_EXTERN PetscErrorCode SNESLineSearchRegisterAll(const char path[]); 606 PETSC_EXTERN PetscErrorCode SNESLineSearchRegisterDestroy(void); 607 608 #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 609 #define SNESLineSearchRegisterDynamic(a,b,c,d) SNESLineSearchRegister(a,b,c,0) 610 #else 611 #define SNESLineSearchRegisterDynamic(a,b,c,d) SNESLineSearchRegister(a,b,c,d) 612 #endif 613 614 /* Routines for VI solver */ 615 PETSC_EXTERN PetscErrorCode SNESVISetVariableBounds(SNES,Vec,Vec); 616 PETSC_EXTERN PetscErrorCode SNESVISetComputeVariableBounds(SNES, PetscErrorCode (*)(SNES,Vec,Vec)); 617 PETSC_EXTERN PetscErrorCode SNESVIGetInactiveSet(SNES,IS*); 618 PETSC_EXTERN PetscErrorCode SNESVIGetActiveSetIS(SNES,Vec,Vec,IS*); 619 PETSC_EXTERN PetscErrorCode SNESVIComputeInactiveSetFnorm(SNES,Vec,Vec,PetscReal*); 620 PETSC_EXTERN PetscErrorCode SNESVISetRedundancyCheck(SNES,PetscErrorCode(*)(SNES,IS,IS*,void*),void*); 621 #define SNES_VI_INF 1.0e20 622 #define SNES_VI_NINF -1.0e20 623 624 PETSC_EXTERN PetscErrorCode SNESTestLocalMin(SNES); 625 626 /* Should this routine be private? */ 627 PETSC_EXTERN PetscErrorCode SNESComputeJacobian(SNES,Vec,Mat*,Mat*,MatStructure*); 628 629 PETSC_EXTERN PetscErrorCode SNESSetDM(SNES,DM); 630 PETSC_EXTERN PetscErrorCode SNESGetDM(SNES,DM*); 631 PETSC_EXTERN PetscErrorCode SNESSetPC(SNES,SNES); 632 PETSC_EXTERN PetscErrorCode SNESGetPC(SNES,SNES*); 633 PETSC_EXTERN PetscErrorCode SNESSetPCSide(SNES,PCSide); 634 PETSC_EXTERN PetscErrorCode SNESGetPCSide(SNES,PCSide*); 635 PETSC_EXTERN PetscErrorCode SNESSetSNESLineSearch(SNES,SNESLineSearch); 636 PETSC_EXTERN PetscErrorCode SNESGetSNESLineSearch(SNES,SNESLineSearch*); 637 PETSC_EXTERN PetscErrorCode SNESRestrictHookAdd(SNES,PetscErrorCode (*)(SNES,SNES,void*),void*); 638 PETSC_EXTERN PetscErrorCode SNESRestrictHooksRun(SNES,SNES); 639 640 PETSC_EXTERN PetscErrorCode SNESSetUpMatrices(SNES); 641 PETSC_EXTERN PetscErrorCode DMSNESSetFunction(DM,PetscErrorCode(*)(SNES,Vec,Vec,void*),void*); 642 PETSC_EXTERN PetscErrorCode DMSNESGetFunction(DM,PetscErrorCode(**)(SNES,Vec,Vec,void*),void**); 643 PETSC_EXTERN PetscErrorCode DMSNESSetGS(DM,PetscErrorCode(*)(SNES,Vec,Vec,void*),void*); 644 PETSC_EXTERN PetscErrorCode DMSNESGetGS(DM,PetscErrorCode(**)(SNES,Vec,Vec,void*),void**); 645 PETSC_EXTERN PetscErrorCode DMSNESSetJacobian(DM,PetscErrorCode(*)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void*); 646 PETSC_EXTERN PetscErrorCode DMSNESGetJacobian(DM,PetscErrorCode(**)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void**); 647 PETSC_EXTERN PetscErrorCode DMSNESSetPicard(DM,PetscErrorCode(*)(SNES,Vec,Vec,void*),PetscErrorCode(*)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void*); 648 PETSC_EXTERN PetscErrorCode DMSNESGetPicard(DM,PetscErrorCode(**)(SNES,Vec,Vec,void*),PetscErrorCode(**)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void**); 649 PETSC_EXTERN PetscErrorCode DMSNESSetObjective(DM,SNESObjective,void*); 650 PETSC_EXTERN PetscErrorCode DMSNESGetObjective(DM,SNESObjective*,void**); 651 652 PETSC_EXTERN PetscErrorCode DMSNESSetBlockFunction(DM,PetscErrorCode(*)(SNES,Vec,Vec,void*),void*); 653 PETSC_EXTERN PetscErrorCode DMSNESGetBlockFunction(DM,PetscErrorCode(**)(SNES,Vec,Vec,void*),void**); 654 PETSC_EXTERN PetscErrorCode DMSNESSetBlockJacobian(DM,PetscErrorCode(*)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void*); 655 PETSC_EXTERN PetscErrorCode DMSNESGetBlockJacobian(DM,PetscErrorCode(**)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void**); 656 657 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*DMDASNESFunction)(DMDALocalInfo*,void*,void*,void*); 658 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*DMDASNESJacobian)(DMDALocalInfo*,void*,Mat,Mat,MatStructure*,void*); 659 PETSC_EXTERN_TYPEDEF typedef PetscErrorCode (*DMDASNESObjective)(DMDALocalInfo*,void*,PetscReal*,void*); 660 661 PETSC_EXTERN PetscErrorCode DMDASNESSetFunctionLocal(DM,InsertMode,DMDASNESFunction,void*); 662 PETSC_EXTERN PetscErrorCode DMDASNESSetJacobianLocal(DM,DMDASNESJacobian,void*); 663 PETSC_EXTERN PetscErrorCode DMDASNESSetObjectiveLocal(DM,DMDASNESObjective,void*); 664 PETSC_EXTERN PetscErrorCode DMDASNESSetPicardLocal(DM,InsertMode,PetscErrorCode (*)(DMDALocalInfo*,void*,void*,void*),PetscErrorCode (*)(DMDALocalInfo*,void*,Mat,Mat,MatStructure*,void*),void*); 665 666 PETSC_EXTERN PetscErrorCode DMSNESSetFunctionLocal(DM,PetscErrorCode (*)(DM,Vec,Vec,void*),void*); 667 PETSC_EXTERN PetscErrorCode DMSNESSetJacobianLocal(DM,PetscErrorCode (*)(DM,Vec,Mat,Mat,MatStructure*,void*),void*); 668 669 PETSC_EXTERN PetscErrorCode SNESComputeLocalBlockFunction_DMDA(SNES,Vec,Vec,void *); 670 PETSC_EXTERN PetscErrorCode SNESComputeLocalBlockJacobian_DMDA(SNES,Vec,Mat*,Mat*,MatStructure*,void*); 671 672 /* Routines for Multiblock solver */ 673 PETSC_EXTERN PetscErrorCode SNESMultiblockSetFields(SNES, const char [], PetscInt, const PetscInt *); 674 PETSC_EXTERN PetscErrorCode SNESMultiblockSetIS(SNES, const char [], IS); 675 PETSC_EXTERN PetscErrorCode SNESMultiblockSetBlockSize(SNES, PetscInt); 676 PETSC_EXTERN PetscErrorCode SNESMultiblockSetType(SNES, PCCompositeType); 677 678 /*J 679 SNESMSType - String with the name of a PETSc SNESMS method. 680 681 Level: intermediate 682 683 .seealso: SNESMSSetType(), SNES 684 J*/ 685 typedef const char* SNESMSType; 686 #define SNESMSM62 "m62" 687 #define SNESMSEULER "euler" 688 #define SNESMSJAMESON83 "jameson83" 689 #define SNESMSVLTP21 "vltp21" 690 #define SNESMSVLTP31 "vltp31" 691 #define SNESMSVLTP41 "vltp41" 692 #define SNESMSVLTP51 "vltp51" 693 #define SNESMSVLTP61 "vltp61" 694 695 PETSC_EXTERN PetscErrorCode SNESMSRegister(SNESMSType,PetscInt,PetscInt,PetscReal,const PetscReal[],const PetscReal[],const PetscReal[]); 696 PETSC_EXTERN PetscErrorCode SNESMSSetType(SNES,SNESMSType); 697 PETSC_EXTERN PetscErrorCode SNESMSFinalizePackage(void); 698 PETSC_EXTERN PetscErrorCode SNESMSInitializePackage(const char path[]); 699 PETSC_EXTERN PetscErrorCode SNESMSRegisterDestroy(void); 700 PETSC_EXTERN PetscErrorCode SNESMSRegisterAll(void); 701 702 /* routines for NGMRES solver */ 703 704 typedef enum { 705 SNES_NGMRES_RESTART_NONE = 0, 706 SNES_NGMRES_RESTART_PERIODIC = 1, 707 SNES_NGMRES_RESTART_DIFFERENCE = 2} SNESNGMRESRestartType; 708 PETSC_EXTERN const char *const SNESNGMRESRestartTypes[]; 709 710 typedef enum { 711 SNES_NGMRES_SELECT_NONE = 0, 712 SNES_NGMRES_SELECT_DIFFERENCE = 1, 713 SNES_NGMRES_SELECT_LINESEARCH = 2} SNESNGMRESSelectType; 714 PETSC_EXTERN const char *const SNESNGMRESSelectTypes[]; 715 716 PETSC_EXTERN PetscErrorCode SNESNGMRESSetRestartType(SNES, SNESNGMRESRestartType); 717 PETSC_EXTERN PetscErrorCode SNESNGMRESSetSelectType(SNES, SNESNGMRESSelectType); 718 719 /* routines for NCG solver */ 720 721 typedef enum { 722 SNES_NCG_FR = 0, 723 SNES_NCG_PRP = 1, 724 SNES_NCG_HS = 2, 725 SNES_NCG_DY = 3, 726 SNES_NCG_CD = 4} SNESNCGType; 727 PETSC_EXTERN const char *const SNESNCGTypes[]; 728 729 PETSC_EXTERN PetscErrorCode SNESNCGSetType(SNES, SNESNCGType); 730 731 typedef enum {SNES_QN_SCALE_NONE = 0, 732 SNES_QN_SCALE_SHANNO = 1, 733 SNES_QN_SCALE_LINESEARCH = 2, 734 SNES_QN_SCALE_JACOBIAN = 3} SNESQNScaleType; 735 PETSC_EXTERN const char *const SNESQNScaleTypes[]; 736 typedef enum {SNES_QN_RESTART_NONE = 0, 737 SNES_QN_RESTART_POWELL = 1, 738 SNES_QN_RESTART_PERIODIC = 2} SNESQNRestartType; 739 PETSC_EXTERN const char *const SNESQNRestartTypes[]; 740 741 PETSC_EXTERN PetscErrorCode SNESQNSetScaleType(SNES, SNESQNScaleType); 742 PETSC_EXTERN PetscErrorCode SNESQNSetRestartType(SNES, SNESQNRestartType); 743 744 745 #endif 746