163dd3a1aSKris Buschelman #define PETSCSNES_DLL 29b94acceSBarry Smith 3e090d566SSatish Balay #include "src/snes/snesimpl.h" /*I "petscsnes.h" I*/ 49b94acceSBarry Smith 54c49b128SBarry Smith PetscTruth SNESRegisterAllCalled = PETSC_FALSE; 68ba1e511SMatthew Knepley PetscFList SNESList = PETSC_NULL; 78ba1e511SMatthew Knepley 88ba1e511SMatthew Knepley /* Logging support */ 963dd3a1aSKris Buschelman PetscCookie PETSCSNES_DLLEXPORT SNES_COOKIE = 0; 106849ba73SBarry Smith PetscEvent SNES_Solve = 0, SNES_LineSearch = 0, SNES_FunctionEval = 0, SNES_JacobianEval = 0; 11a09944afSBarry Smith 12a09944afSBarry Smith #undef __FUNCT__ 134a2ae208SSatish Balay #define __FUNCT__ "SNESView" 147e2c5f70SBarry Smith /*@C 159b94acceSBarry Smith SNESView - Prints the SNES data structure. 169b94acceSBarry Smith 174c49b128SBarry Smith Collective on SNES 18fee21e36SBarry Smith 19c7afd0dbSLois Curfman McInnes Input Parameters: 20c7afd0dbSLois Curfman McInnes + SNES - the SNES context 21c7afd0dbSLois Curfman McInnes - viewer - visualization context 22c7afd0dbSLois Curfman McInnes 239b94acceSBarry Smith Options Database Key: 24c8a8ba5cSLois Curfman McInnes . -snes_view - Calls SNESView() at end of SNESSolve() 259b94acceSBarry Smith 269b94acceSBarry Smith Notes: 279b94acceSBarry Smith The available visualization contexts include 28b0a32e0cSBarry Smith + PETSC_VIEWER_STDOUT_SELF - standard output (default) 29b0a32e0cSBarry Smith - PETSC_VIEWER_STDOUT_WORLD - synchronized standard 30c8a8ba5cSLois Curfman McInnes output where only the first processor opens 31c8a8ba5cSLois Curfman McInnes the file. All other processors send their 32c8a8ba5cSLois Curfman McInnes data to the first processor to print. 339b94acceSBarry Smith 343e081fefSLois Curfman McInnes The user can open an alternative visualization context with 35b0a32e0cSBarry Smith PetscViewerASCIIOpen() - output to a specified file. 369b94acceSBarry Smith 3736851e7fSLois Curfman McInnes Level: beginner 3836851e7fSLois Curfman McInnes 399b94acceSBarry Smith .keywords: SNES, view 409b94acceSBarry Smith 41b0a32e0cSBarry Smith .seealso: PetscViewerASCIIOpen() 429b94acceSBarry Smith @*/ 4363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESView(SNES snes,PetscViewer viewer) 449b94acceSBarry Smith { 459b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 46dfbe8321SBarry Smith PetscErrorCode ierr; 4794b7f48cSBarry Smith KSP ksp; 48e060cb09SBarry Smith SNESType type; 4932077d6dSBarry Smith PetscTruth iascii,isstring; 509b94acceSBarry Smith 513a40ed3dSBarry Smith PetscFunctionBegin; 524482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 53b0a32e0cSBarry Smith if (!viewer) viewer = PETSC_VIEWER_STDOUT_(snes->comm); 544482741eSBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_COOKIE,2); 55c9780b6fSBarry Smith PetscCheckSameComm(snes,1,viewer,2); 5674679c65SBarry Smith 5732077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 58b0a32e0cSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_STRING,&isstring);CHKERRQ(ierr); 5932077d6dSBarry Smith if (iascii) { 603a7fca6bSBarry Smith if (snes->prefix) { 613a7fca6bSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"SNES Object:(%s)\n",snes->prefix);CHKERRQ(ierr); 623a7fca6bSBarry Smith } else { 63b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"SNES Object:\n");CHKERRQ(ierr); 643a7fca6bSBarry Smith } 65454a90a3SBarry Smith ierr = SNESGetType(snes,&type);CHKERRQ(ierr); 66454a90a3SBarry Smith if (type) { 67b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," type: %s\n",type);CHKERRQ(ierr); 68184914b5SBarry Smith } else { 69b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," type: not set yet\n");CHKERRQ(ierr); 70184914b5SBarry Smith } 71e7788613SBarry Smith if (snes->ops->view) { 72b0a32e0cSBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 73e7788613SBarry Smith ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 74b0a32e0cSBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 750ef38995SBarry Smith } 7677431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr); 77a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," tolerances: relative=%G, absolute=%G, solution=%G\n", 7870441072SBarry Smith snes->rtol,snes->abstol,snes->xtol);CHKERRQ(ierr); 7977431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr); 8077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr); 819b94acceSBarry Smith if (snes->ksp_ewconv) { 829b94acceSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 839b94acceSBarry Smith if (kctx) { 8477431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr); 85a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," rtol_0=%G, rtol_max=%G, threshold=%G\n",kctx->rtol_0,kctx->rtol_max,kctx->threshold);CHKERRQ(ierr); 86a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," gamma=%G, alpha=%G, alpha2=%G\n",kctx->gamma,kctx->alpha,kctx->alpha2);CHKERRQ(ierr); 879b94acceSBarry Smith } 889b94acceSBarry Smith } 890f5bd95cSBarry Smith } else if (isstring) { 90454a90a3SBarry Smith ierr = SNESGetType(snes,&type);CHKERRQ(ierr); 91b0a32e0cSBarry Smith ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr); 9219bcc07fSBarry Smith } 9394b7f48cSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 94b0a32e0cSBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 9594b7f48cSBarry Smith ierr = KSPView(ksp,viewer);CHKERRQ(ierr); 96b0a32e0cSBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 973a40ed3dSBarry Smith PetscFunctionReturn(0); 989b94acceSBarry Smith } 999b94acceSBarry Smith 10076b2cf59SMatthew Knepley /* 10176b2cf59SMatthew Knepley We retain a list of functions that also take SNES command 10276b2cf59SMatthew Knepley line options. These are called at the end SNESSetFromOptions() 10376b2cf59SMatthew Knepley */ 10476b2cf59SMatthew Knepley #define MAXSETFROMOPTIONS 5 105a7cc72afSBarry Smith static PetscInt numberofsetfromoptions; 1066849ba73SBarry Smith static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES); 10776b2cf59SMatthew Knepley 108e74ef692SMatthew Knepley #undef __FUNCT__ 109e74ef692SMatthew Knepley #define __FUNCT__ "SNESAddOptionsChecker" 110ac226902SBarry Smith /*@C 11176b2cf59SMatthew Knepley SNESAddOptionsChecker - Adds an additional function to check for SNES options. 11276b2cf59SMatthew Knepley 11376b2cf59SMatthew Knepley Not Collective 11476b2cf59SMatthew Knepley 11576b2cf59SMatthew Knepley Input Parameter: 11676b2cf59SMatthew Knepley . snescheck - function that checks for options 11776b2cf59SMatthew Knepley 11876b2cf59SMatthew Knepley Level: developer 11976b2cf59SMatthew Knepley 12076b2cf59SMatthew Knepley .seealso: SNESSetFromOptions() 12176b2cf59SMatthew Knepley @*/ 12263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES)) 12376b2cf59SMatthew Knepley { 12476b2cf59SMatthew Knepley PetscFunctionBegin; 12576b2cf59SMatthew Knepley if (numberofsetfromoptions >= MAXSETFROMOPTIONS) { 12677431f27SBarry Smith SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS); 12776b2cf59SMatthew Knepley } 12876b2cf59SMatthew Knepley othersetfromoptions[numberofsetfromoptions++] = snescheck; 12976b2cf59SMatthew Knepley PetscFunctionReturn(0); 13076b2cf59SMatthew Knepley } 13176b2cf59SMatthew Knepley 1324a2ae208SSatish Balay #undef __FUNCT__ 1334a2ae208SSatish Balay #define __FUNCT__ "SNESSetFromOptions" 1349b94acceSBarry Smith /*@ 13594b7f48cSBarry Smith SNESSetFromOptions - Sets various SNES and KSP parameters from user options. 1369b94acceSBarry Smith 137c7afd0dbSLois Curfman McInnes Collective on SNES 138c7afd0dbSLois Curfman McInnes 1399b94acceSBarry Smith Input Parameter: 1409b94acceSBarry Smith . snes - the SNES context 1419b94acceSBarry Smith 14236851e7fSLois Curfman McInnes Options Database Keys: 1436831982aSBarry Smith + -snes_type <type> - ls, tr, umls, umtr, test 14482738288SBarry Smith . -snes_stol - convergence tolerance in terms of the norm 14582738288SBarry Smith of the change in the solution between steps 14670441072SBarry Smith . -snes_atol <abstol> - absolute tolerance of residual norm 147b39c3a46SLois Curfman McInnes . -snes_rtol <rtol> - relative decrease in tolerance norm from initial 148b39c3a46SLois Curfman McInnes . -snes_max_it <max_it> - maximum number of iterations 149b39c3a46SLois Curfman McInnes . -snes_max_funcs <max_funcs> - maximum number of function evaluations 15050ffb88aSMatthew Knepley . -snes_max_fail <max_fail> - maximum number of failures 151b39c3a46SLois Curfman McInnes . -snes_trtol <trtol> - trust region tolerance 1522492ecdbSBarry Smith . -snes_no_convergence_test - skip convergence test in nonlinear 15382738288SBarry Smith solver; hence iterations will continue until max_it 1541fbbfb26SLois Curfman McInnes or some other criterion is reached. Saves expense 15582738288SBarry Smith of convergence test 156e8105e01SRichard Katz . -snes_monitor <optional filename> - prints residual norm at each iteration. if no 157e8105e01SRichard Katz filename given prints to stdout 158d132466eSBarry Smith . -snes_vecmonitor - plots solution at each iteration 159d132466eSBarry Smith . -snes_vecmonitor_update - plots update to solution at each iteration 16082738288SBarry Smith . -snes_xmonitor - plots residual norm at each iteration 161e24b481bSBarry Smith . -snes_fd - use finite differences to compute Jacobian; very slow, only for testing 1625968eb51SBarry Smith . -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration 1635968eb51SBarry Smith - -snes_print_converged_reason - print the reason for convergence/divergence after each solve 16482738288SBarry Smith 16582738288SBarry Smith Options Database for Eisenstat-Walker method: 1664b27c08aSLois Curfman McInnes + -snes_ksp_ew_conv - use Eisenstat-Walker method for determining linear system convergence 1674b27c08aSLois Curfman McInnes . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 16836851e7fSLois Curfman McInnes . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 16936851e7fSLois Curfman McInnes . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 17036851e7fSLois Curfman McInnes . -snes_ksp_ew_gamma <gamma> - Sets gamma 17136851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha <alpha> - Sets alpha 17236851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 17336851e7fSLois Curfman McInnes - -snes_ksp_ew_threshold <threshold> - Sets threshold 17482738288SBarry Smith 17511ca99fdSLois Curfman McInnes Notes: 17611ca99fdSLois Curfman McInnes To see all options, run your program with the -help option or consult 17711ca99fdSLois Curfman McInnes the users manual. 17883e2fdc7SBarry Smith 17936851e7fSLois Curfman McInnes Level: beginner 18036851e7fSLois Curfman McInnes 1819b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 1829b94acceSBarry Smith 18369ed319cSSatish Balay .seealso: SNESSetOptionsPrefix() 1849b94acceSBarry Smith @*/ 18563dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetFromOptions(SNES snes) 1869b94acceSBarry Smith { 18794b7f48cSBarry Smith KSP ksp; 188186905e3SBarry Smith SNES_KSP_EW_ConvCtx *kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 189f1af5d2fSBarry Smith PetscTruth flg; 190dfbe8321SBarry Smith PetscErrorCode ierr; 19177431f27SBarry Smith PetscInt i; 1922fc52814SBarry Smith const char *deft; 193e8105e01SRichard Katz char type[256], monfilename[PETSC_MAX_PATH_LEN]; 194e8105e01SRichard Katz PetscViewer monviewer; 1959b94acceSBarry Smith 1963a40ed3dSBarry Smith PetscFunctionBegin; 1974482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 198ca161407SBarry Smith 199b0a32e0cSBarry Smith ierr = PetscOptionsBegin(snes->comm,snes->prefix,"Nonlinear solver (SNES) options","SNES");CHKERRQ(ierr); 200186905e3SBarry Smith if (snes->type_name) { 201186905e3SBarry Smith deft = snes->type_name; 202186905e3SBarry Smith } else { 2034b27c08aSLois Curfman McInnes deft = SNESLS; 204d64ed03dSBarry Smith } 2054bbc92c1SBarry Smith 206186905e3SBarry Smith if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL);CHKERRQ(ierr);} 207b0a32e0cSBarry Smith ierr = PetscOptionsList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr); 208d64ed03dSBarry Smith if (flg) { 209186905e3SBarry Smith ierr = SNESSetType(snes,type);CHKERRQ(ierr); 210186905e3SBarry Smith } else if (!snes->type_name) { 211186905e3SBarry Smith ierr = SNESSetType(snes,deft);CHKERRQ(ierr); 212d64ed03dSBarry Smith } 213909c8a9fSBarry Smith ierr = PetscOptionsName("-snes_view","Print detailed information on solver used","SNESView",0);CHKERRQ(ierr); 21493c39befSBarry Smith 21587828ca2SBarry Smith ierr = PetscOptionsReal("-snes_stol","Stop if step length less then","SNESSetTolerances",snes->xtol,&snes->xtol,0);CHKERRQ(ierr); 21670441072SBarry Smith ierr = PetscOptionsReal("-snes_atol","Stop if function norm less then","SNESSetTolerances",snes->abstol,&snes->abstol,0);CHKERRQ(ierr); 217186905e3SBarry Smith 21887828ca2SBarry Smith ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less then","SNESSetTolerances",snes->rtol,&snes->rtol,0);CHKERRQ(ierr); 219b0a32e0cSBarry Smith ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,PETSC_NULL);CHKERRQ(ierr); 220b0a32e0cSBarry Smith ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,PETSC_NULL);CHKERRQ(ierr); 22150ffb88aSMatthew Knepley ierr = PetscOptionsInt("-snes_max_fail","Maximum failures","SNESSetTolerances",snes->maxFailures,&snes->maxFailures,PETSC_NULL);CHKERRQ(ierr); 2225968eb51SBarry Smith ierr = PetscOptionsName("-snes_converged_reason","Print reason for converged or diverged","SNESSolve",&flg);CHKERRQ(ierr); 2235968eb51SBarry Smith if (flg) { 2245968eb51SBarry Smith snes->printreason = PETSC_TRUE; 2255968eb51SBarry Smith } 226186905e3SBarry Smith 2277aaed0d8SBarry Smith ierr = PetscOptionsTruth("-snes_ksp_ew_conv","Use Eisentat-Walker linear system convergence test","SNES_KSP_SetParametersEW",snes->ksp_ewconv,&snes->ksp_ewconv,PETSC_NULL);CHKERRQ(ierr); 228186905e3SBarry Smith 229cf502942SBarry Smith ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNES_KSP_SetParametersEW",kctx->version,&kctx->version,0);CHKERRQ(ierr); 23087828ca2SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNES_KSP_SetParametersEW",kctx->rtol_0,&kctx->rtol_0,0);CHKERRQ(ierr); 23187828ca2SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNES_KSP_SetParametersEW",kctx->rtol_max,&kctx->rtol_max,0);CHKERRQ(ierr); 23287828ca2SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNES_KSP_SetParametersEW",kctx->gamma,&kctx->gamma,0);CHKERRQ(ierr); 23387828ca2SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNES_KSP_SetParametersEW",kctx->alpha,&kctx->alpha,0);CHKERRQ(ierr); 23487828ca2SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNES_KSP_SetParametersEW",kctx->alpha2,&kctx->alpha2,0);CHKERRQ(ierr); 23587828ca2SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNES_KSP_SetParametersEW",kctx->threshold,&kctx->threshold,0);CHKERRQ(ierr); 236186905e3SBarry Smith 237b0a32e0cSBarry Smith ierr = PetscOptionsName("-snes_no_convergence_test","Don't test for convergence","None",&flg);CHKERRQ(ierr); 238e7788613SBarry Smith if (flg) {snes->ops->converged = 0;} 239b0a32e0cSBarry Smith ierr = PetscOptionsName("-snes_cancelmonitors","Remove all monitors","SNESClearMonitor",&flg);CHKERRQ(ierr); 2405cd90555SBarry Smith if (flg) {ierr = SNESClearMonitor(snes);CHKERRQ(ierr);} 241eabae89aSBarry Smith 242eabae89aSBarry Smith ierr = PetscOptionsString("-snes_monitor","Monitor norm of function","SNESSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr); 243e8105e01SRichard Katz if (flg) { 244e8105e01SRichard Katz ierr = PetscViewerASCIIOpen(snes->comm,monfilename,&monviewer);CHKERRQ(ierr); 245e8105e01SRichard Katz ierr = SNESSetMonitor(snes,SNESDefaultMonitor,monviewer,(PetscErrorCode (*)(void*))PetscViewerDestroy);CHKERRQ(ierr); 246e8105e01SRichard Katz } 247eabae89aSBarry Smith 248eabae89aSBarry Smith ierr = PetscOptionsString("-snes_ratiomonitor","Monitor ratios of norms of function","SNESSetRatioMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr); 249eabae89aSBarry Smith if (flg) { 250eabae89aSBarry Smith ierr = PetscViewerASCIIOpen(snes->comm,monfilename,&monviewer);CHKERRQ(ierr); 251eabae89aSBarry Smith ierr = SNESSetRatioMonitor(snes,monviewer);CHKERRQ(ierr); 252e8105e01SRichard Katz } 253eabae89aSBarry Smith 254eabae89aSBarry Smith ierr = PetscOptionsString("-snes_smonitor","Monitor norm of function (fewer digits)","SNESSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr); 255eabae89aSBarry Smith if (flg) { 256eabae89aSBarry Smith ierr = PetscViewerASCIIOpen(snes->comm,monfilename,&monviewer);CHKERRQ(ierr); 257eabae89aSBarry Smith ierr = SNESSetMonitor(snes,SNESDefaultSMonitor,monviewer,(PetscErrorCode (*)(void*))PetscViewerDestroy);CHKERRQ(ierr); 258eabae89aSBarry Smith } 259eabae89aSBarry Smith 260b0a32e0cSBarry Smith ierr = PetscOptionsName("-snes_vecmonitor","Plot solution at each iteration","SNESVecViewMonitor",&flg);CHKERRQ(ierr); 261b8a78c4aSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESVecViewMonitor,0,0);CHKERRQ(ierr);} 262b0a32e0cSBarry Smith ierr = PetscOptionsName("-snes_vecmonitor_update","Plot correction at each iteration","SNESVecViewUpdateMonitor",&flg);CHKERRQ(ierr); 2637c922b88SBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESVecViewUpdateMonitor,0,0);CHKERRQ(ierr);} 2645ed2d596SBarry Smith ierr = PetscOptionsName("-snes_vecmonitor_residual","Plot residual at each iteration","SNESVecViewResidualMonitor",&flg);CHKERRQ(ierr); 2655ed2d596SBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESVecViewResidualMonitor,0,0);CHKERRQ(ierr);} 266b0a32e0cSBarry Smith ierr = PetscOptionsName("-snes_xmonitor","Plot function norm at each iteration","SNESLGMonitor",&flg);CHKERRQ(ierr); 267186905e3SBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESLGMonitor,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);} 268e24b481bSBarry Smith 269b0a32e0cSBarry Smith ierr = PetscOptionsName("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESDefaultComputeJacobian",&flg);CHKERRQ(ierr); 2704b27c08aSLois Curfman McInnes if (flg) { 271186905e3SBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESDefaultComputeJacobian,snes->funP);CHKERRQ(ierr); 272ae15b995SBarry Smith ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr); 2739b94acceSBarry Smith } 274639f9d9dSBarry Smith 27576b2cf59SMatthew Knepley for(i = 0; i < numberofsetfromoptions; i++) { 27676b2cf59SMatthew Knepley ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr); 27776b2cf59SMatthew Knepley } 27876b2cf59SMatthew Knepley 279e7788613SBarry Smith if (snes->ops->setfromoptions) { 280e7788613SBarry Smith ierr = (*snes->ops->setfromoptions)(snes);CHKERRQ(ierr); 281639f9d9dSBarry Smith } 282b0a32e0cSBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 2834bbc92c1SBarry Smith 28494b7f48cSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 28594b7f48cSBarry Smith ierr = KSPSetFromOptions(ksp);CHKERRQ(ierr); 28693993e2dSLois Curfman McInnes 2873a40ed3dSBarry Smith PetscFunctionReturn(0); 2889b94acceSBarry Smith } 2899b94acceSBarry Smith 290a847f771SSatish Balay 2914a2ae208SSatish Balay #undef __FUNCT__ 2924a2ae208SSatish Balay #define __FUNCT__ "SNESSetApplicationContext" 2939b94acceSBarry Smith /*@ 2949b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 2959b94acceSBarry Smith the nonlinear solvers. 2969b94acceSBarry Smith 297fee21e36SBarry Smith Collective on SNES 298fee21e36SBarry Smith 299c7afd0dbSLois Curfman McInnes Input Parameters: 300c7afd0dbSLois Curfman McInnes + snes - the SNES context 301c7afd0dbSLois Curfman McInnes - usrP - optional user context 302c7afd0dbSLois Curfman McInnes 30336851e7fSLois Curfman McInnes Level: intermediate 30436851e7fSLois Curfman McInnes 3059b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 3069b94acceSBarry Smith 3079b94acceSBarry Smith .seealso: SNESGetApplicationContext() 3089b94acceSBarry Smith @*/ 30963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetApplicationContext(SNES snes,void *usrP) 3109b94acceSBarry Smith { 3113a40ed3dSBarry Smith PetscFunctionBegin; 3124482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 3139b94acceSBarry Smith snes->user = usrP; 3143a40ed3dSBarry Smith PetscFunctionReturn(0); 3159b94acceSBarry Smith } 31674679c65SBarry Smith 3174a2ae208SSatish Balay #undef __FUNCT__ 3184a2ae208SSatish Balay #define __FUNCT__ "SNESGetApplicationContext" 3199b94acceSBarry Smith /*@C 3209b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 3219b94acceSBarry Smith nonlinear solvers. 3229b94acceSBarry Smith 323c7afd0dbSLois Curfman McInnes Not Collective 324c7afd0dbSLois Curfman McInnes 3259b94acceSBarry Smith Input Parameter: 3269b94acceSBarry Smith . snes - SNES context 3279b94acceSBarry Smith 3289b94acceSBarry Smith Output Parameter: 3299b94acceSBarry Smith . usrP - user context 3309b94acceSBarry Smith 33136851e7fSLois Curfman McInnes Level: intermediate 33236851e7fSLois Curfman McInnes 3339b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 3349b94acceSBarry Smith 3359b94acceSBarry Smith .seealso: SNESSetApplicationContext() 3369b94acceSBarry Smith @*/ 33763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetApplicationContext(SNES snes,void **usrP) 3389b94acceSBarry Smith { 3393a40ed3dSBarry Smith PetscFunctionBegin; 3404482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 3419b94acceSBarry Smith *usrP = snes->user; 3423a40ed3dSBarry Smith PetscFunctionReturn(0); 3439b94acceSBarry Smith } 34474679c65SBarry Smith 3454a2ae208SSatish Balay #undef __FUNCT__ 3464a2ae208SSatish Balay #define __FUNCT__ "SNESGetIterationNumber" 3479b94acceSBarry Smith /*@ 348c8228a4eSBarry Smith SNESGetIterationNumber - Gets the number of nonlinear iterations completed 349c8228a4eSBarry Smith at this time. 3509b94acceSBarry Smith 351c7afd0dbSLois Curfman McInnes Not Collective 352c7afd0dbSLois Curfman McInnes 3539b94acceSBarry Smith Input Parameter: 3549b94acceSBarry Smith . snes - SNES context 3559b94acceSBarry Smith 3569b94acceSBarry Smith Output Parameter: 3579b94acceSBarry Smith . iter - iteration number 3589b94acceSBarry Smith 359c8228a4eSBarry Smith Notes: 360c8228a4eSBarry Smith For example, during the computation of iteration 2 this would return 1. 361c8228a4eSBarry Smith 362c8228a4eSBarry Smith This is useful for using lagged Jacobians (where one does not recompute the 36308405cd6SLois Curfman McInnes Jacobian at each SNES iteration). For example, the code 36408405cd6SLois Curfman McInnes .vb 36508405cd6SLois Curfman McInnes ierr = SNESGetIterationNumber(snes,&it); 36608405cd6SLois Curfman McInnes if (!(it % 2)) { 36708405cd6SLois Curfman McInnes [compute Jacobian here] 36808405cd6SLois Curfman McInnes } 36908405cd6SLois Curfman McInnes .ve 370c8228a4eSBarry Smith can be used in your ComputeJacobian() function to cause the Jacobian to be 37108405cd6SLois Curfman McInnes recomputed every second SNES iteration. 372c8228a4eSBarry Smith 37336851e7fSLois Curfman McInnes Level: intermediate 37436851e7fSLois Curfman McInnes 3752b668275SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 3762b668275SBarry Smith 3772b668275SBarry Smith .seealso: SNESGetFunctionNorm(), SNESGetNumberLinearIterations() 3789b94acceSBarry Smith @*/ 37963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetIterationNumber(SNES snes,PetscInt* iter) 3809b94acceSBarry Smith { 3813a40ed3dSBarry Smith PetscFunctionBegin; 3824482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 3834482741eSBarry Smith PetscValidIntPointer(iter,2); 3849b94acceSBarry Smith *iter = snes->iter; 3853a40ed3dSBarry Smith PetscFunctionReturn(0); 3869b94acceSBarry Smith } 38774679c65SBarry Smith 3884a2ae208SSatish Balay #undef __FUNCT__ 3894a2ae208SSatish Balay #define __FUNCT__ "SNESGetFunctionNorm" 3909b94acceSBarry Smith /*@ 3919b94acceSBarry Smith SNESGetFunctionNorm - Gets the norm of the current function that was set 3929b94acceSBarry Smith with SNESSSetFunction(). 3939b94acceSBarry Smith 394c7afd0dbSLois Curfman McInnes Collective on SNES 395c7afd0dbSLois Curfman McInnes 3969b94acceSBarry Smith Input Parameter: 3979b94acceSBarry Smith . snes - SNES context 3989b94acceSBarry Smith 3999b94acceSBarry Smith Output Parameter: 4009b94acceSBarry Smith . fnorm - 2-norm of function 4019b94acceSBarry Smith 40236851e7fSLois Curfman McInnes Level: intermediate 40336851e7fSLois Curfman McInnes 4049b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm 405a86d99e1SLois Curfman McInnes 4062b668275SBarry Smith .seealso: SNESGetFunction(), SNESGetIterationNumber(), SNESGetNumberLinearIterations() 4079b94acceSBarry Smith @*/ 40863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetFunctionNorm(SNES snes,PetscScalar *fnorm) 4099b94acceSBarry Smith { 4103a40ed3dSBarry Smith PetscFunctionBegin; 4114482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 4124482741eSBarry Smith PetscValidScalarPointer(fnorm,2); 4139b94acceSBarry Smith *fnorm = snes->norm; 4143a40ed3dSBarry Smith PetscFunctionReturn(0); 4159b94acceSBarry Smith } 41674679c65SBarry Smith 4174a2ae208SSatish Balay #undef __FUNCT__ 4184a2ae208SSatish Balay #define __FUNCT__ "SNESGetNumberUnsuccessfulSteps" 4199b94acceSBarry Smith /*@ 4209b94acceSBarry Smith SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps 4219b94acceSBarry Smith attempted by the nonlinear solver. 4229b94acceSBarry Smith 423c7afd0dbSLois Curfman McInnes Not Collective 424c7afd0dbSLois Curfman McInnes 4259b94acceSBarry Smith Input Parameter: 4269b94acceSBarry Smith . snes - SNES context 4279b94acceSBarry Smith 4289b94acceSBarry Smith Output Parameter: 4299b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 4309b94acceSBarry Smith 431c96a6f78SLois Curfman McInnes Notes: 432c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 433c96a6f78SLois Curfman McInnes 43436851e7fSLois Curfman McInnes Level: intermediate 43536851e7fSLois Curfman McInnes 4369b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 4379b94acceSBarry Smith @*/ 43863dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetNumberUnsuccessfulSteps(SNES snes,PetscInt* nfails) 4399b94acceSBarry Smith { 4403a40ed3dSBarry Smith PetscFunctionBegin; 4414482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 4424482741eSBarry Smith PetscValidIntPointer(nfails,2); 44350ffb88aSMatthew Knepley *nfails = snes->numFailures; 44450ffb88aSMatthew Knepley PetscFunctionReturn(0); 44550ffb88aSMatthew Knepley } 44650ffb88aSMatthew Knepley 44750ffb88aSMatthew Knepley #undef __FUNCT__ 44850ffb88aSMatthew Knepley #define __FUNCT__ "SNESSetMaximumUnsuccessfulSteps" 44950ffb88aSMatthew Knepley /*@ 45050ffb88aSMatthew Knepley SNESSetMaximumUnsuccessfulSteps - Sets the maximum number of unsuccessful steps 45150ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 45250ffb88aSMatthew Knepley 45350ffb88aSMatthew Knepley Not Collective 45450ffb88aSMatthew Knepley 45550ffb88aSMatthew Knepley Input Parameters: 45650ffb88aSMatthew Knepley + snes - SNES context 45750ffb88aSMatthew Knepley - maxFails - maximum of unsuccessful steps 45850ffb88aSMatthew Knepley 45950ffb88aSMatthew Knepley Level: intermediate 46050ffb88aSMatthew Knepley 46150ffb88aSMatthew Knepley .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps 46250ffb88aSMatthew Knepley @*/ 46363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetMaximumUnsuccessfulSteps(SNES snes, PetscInt maxFails) 46450ffb88aSMatthew Knepley { 46550ffb88aSMatthew Knepley PetscFunctionBegin; 4664482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 46750ffb88aSMatthew Knepley snes->maxFailures = maxFails; 46850ffb88aSMatthew Knepley PetscFunctionReturn(0); 46950ffb88aSMatthew Knepley } 47050ffb88aSMatthew Knepley 47150ffb88aSMatthew Knepley #undef __FUNCT__ 47250ffb88aSMatthew Knepley #define __FUNCT__ "SNESGetMaximumUnsuccessfulSteps" 47350ffb88aSMatthew Knepley /*@ 47450ffb88aSMatthew Knepley SNESGetMaximumUnsuccessfulSteps - Gets the maximum number of unsuccessful steps 47550ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 47650ffb88aSMatthew Knepley 47750ffb88aSMatthew Knepley Not Collective 47850ffb88aSMatthew Knepley 47950ffb88aSMatthew Knepley Input Parameter: 48050ffb88aSMatthew Knepley . snes - SNES context 48150ffb88aSMatthew Knepley 48250ffb88aSMatthew Knepley Output Parameter: 48350ffb88aSMatthew Knepley . maxFails - maximum of unsuccessful steps 48450ffb88aSMatthew Knepley 48550ffb88aSMatthew Knepley Level: intermediate 48650ffb88aSMatthew Knepley 48750ffb88aSMatthew Knepley .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 48850ffb88aSMatthew Knepley @*/ 48963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetMaximumUnsuccessfulSteps(SNES snes, PetscInt *maxFails) 49050ffb88aSMatthew Knepley { 49150ffb88aSMatthew Knepley PetscFunctionBegin; 4924482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 4934482741eSBarry Smith PetscValidIntPointer(maxFails,2); 49450ffb88aSMatthew Knepley *maxFails = snes->maxFailures; 4953a40ed3dSBarry Smith PetscFunctionReturn(0); 4969b94acceSBarry Smith } 497a847f771SSatish Balay 4984a2ae208SSatish Balay #undef __FUNCT__ 4993d4c4710SBarry Smith #define __FUNCT__ "SNESGetLinearSolveFailures" 5003d4c4710SBarry Smith /*@ 5013d4c4710SBarry Smith SNESGetLinearSolveFailures - Gets the number of failed (non-converged) 5023d4c4710SBarry Smith linear solvers. 5033d4c4710SBarry Smith 5043d4c4710SBarry Smith Not Collective 5053d4c4710SBarry Smith 5063d4c4710SBarry Smith Input Parameter: 5073d4c4710SBarry Smith . snes - SNES context 5083d4c4710SBarry Smith 5093d4c4710SBarry Smith Output Parameter: 5103d4c4710SBarry Smith . nfails - number of failed solves 5113d4c4710SBarry Smith 5123d4c4710SBarry Smith Notes: 5133d4c4710SBarry Smith This counter is reset to zero for each successive call to SNESSolve(). 5143d4c4710SBarry Smith 5153d4c4710SBarry Smith Level: intermediate 5163d4c4710SBarry Smith 5173d4c4710SBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 5183d4c4710SBarry Smith @*/ 5193d4c4710SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESGetLinearSolveFailures(SNES snes,PetscInt* nfails) 5203d4c4710SBarry Smith { 5213d4c4710SBarry Smith PetscFunctionBegin; 5223d4c4710SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 5233d4c4710SBarry Smith PetscValidIntPointer(nfails,2); 5243d4c4710SBarry Smith *nfails = snes->numLinearSolveFailures; 5253d4c4710SBarry Smith PetscFunctionReturn(0); 5263d4c4710SBarry Smith } 5273d4c4710SBarry Smith 5283d4c4710SBarry Smith #undef __FUNCT__ 5293d4c4710SBarry Smith #define __FUNCT__ "SNESSetMaxLinearSolveFailures" 5303d4c4710SBarry Smith /*@ 5313d4c4710SBarry Smith SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts 5323d4c4710SBarry Smith allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE 5333d4c4710SBarry Smith 5343d4c4710SBarry Smith Collective on SNES 5353d4c4710SBarry Smith 5363d4c4710SBarry Smith Input Parameters: 5373d4c4710SBarry Smith + snes - SNES context 5383d4c4710SBarry Smith - maxFails - maximum allowed linear solve failures 5393d4c4710SBarry Smith 5403d4c4710SBarry Smith Level: intermediate 5413d4c4710SBarry Smith 5423d4c4710SBarry Smith Notes: By default this is 1; that is SNES returns on the first failed linear solve 5433d4c4710SBarry Smith 5443d4c4710SBarry Smith .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps 5453d4c4710SBarry Smith 5463d4c4710SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures() 5473d4c4710SBarry Smith @*/ 5483d4c4710SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails) 5493d4c4710SBarry Smith { 5503d4c4710SBarry Smith PetscFunctionBegin; 5513d4c4710SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 5523d4c4710SBarry Smith snes->maxLinearSolveFailures = maxFails; 5533d4c4710SBarry Smith PetscFunctionReturn(0); 5543d4c4710SBarry Smith } 5553d4c4710SBarry Smith 5563d4c4710SBarry Smith #undef __FUNCT__ 5573d4c4710SBarry Smith #define __FUNCT__ "SNESGetMaxLinearSolveFailures" 5583d4c4710SBarry Smith /*@ 5593d4c4710SBarry Smith SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that 5603d4c4710SBarry Smith are allowed before SNES terminates 5613d4c4710SBarry Smith 5623d4c4710SBarry Smith Not Collective 5633d4c4710SBarry Smith 5643d4c4710SBarry Smith Input Parameter: 5653d4c4710SBarry Smith . snes - SNES context 5663d4c4710SBarry Smith 5673d4c4710SBarry Smith Output Parameter: 5683d4c4710SBarry Smith . maxFails - maximum of unsuccessful solves allowed 5693d4c4710SBarry Smith 5703d4c4710SBarry Smith Level: intermediate 5713d4c4710SBarry Smith 5723d4c4710SBarry Smith Notes: By default this is 1; that is SNES returns on the first failed linear solve 5733d4c4710SBarry Smith 5743d4c4710SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 5753d4c4710SBarry Smith 5763d4c4710SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures() 5773d4c4710SBarry Smith @*/ 5783d4c4710SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails) 5793d4c4710SBarry Smith { 5803d4c4710SBarry Smith PetscFunctionBegin; 5813d4c4710SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 5823d4c4710SBarry Smith PetscValidIntPointer(maxFails,2); 5833d4c4710SBarry Smith *maxFails = snes->maxLinearSolveFailures; 5843d4c4710SBarry Smith PetscFunctionReturn(0); 5853d4c4710SBarry Smith } 5863d4c4710SBarry Smith 5873d4c4710SBarry Smith #undef __FUNCT__ 5884a2ae208SSatish Balay #define __FUNCT__ "SNESGetNumberLinearIterations" 589c96a6f78SLois Curfman McInnes /*@ 590c96a6f78SLois Curfman McInnes SNESGetNumberLinearIterations - Gets the total number of linear iterations 591c96a6f78SLois Curfman McInnes used by the nonlinear solver. 592c96a6f78SLois Curfman McInnes 593c7afd0dbSLois Curfman McInnes Not Collective 594c7afd0dbSLois Curfman McInnes 595c96a6f78SLois Curfman McInnes Input Parameter: 596c96a6f78SLois Curfman McInnes . snes - SNES context 597c96a6f78SLois Curfman McInnes 598c96a6f78SLois Curfman McInnes Output Parameter: 599c96a6f78SLois Curfman McInnes . lits - number of linear iterations 600c96a6f78SLois Curfman McInnes 601c96a6f78SLois Curfman McInnes Notes: 602c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 603c96a6f78SLois Curfman McInnes 60436851e7fSLois Curfman McInnes Level: intermediate 60536851e7fSLois Curfman McInnes 606c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations 6072b668275SBarry Smith 6082b668275SBarry Smith .seealso: SNESGetIterationNumber(), SNESGetFunctionNorm() 609c96a6f78SLois Curfman McInnes @*/ 61063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetNumberLinearIterations(SNES snes,PetscInt* lits) 611c96a6f78SLois Curfman McInnes { 6123a40ed3dSBarry Smith PetscFunctionBegin; 6134482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 6144482741eSBarry Smith PetscValidIntPointer(lits,2); 615c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 6163a40ed3dSBarry Smith PetscFunctionReturn(0); 617c96a6f78SLois Curfman McInnes } 618c96a6f78SLois Curfman McInnes 6194a2ae208SSatish Balay #undef __FUNCT__ 62094b7f48cSBarry Smith #define __FUNCT__ "SNESGetKSP" 62152baeb72SSatish Balay /*@ 62294b7f48cSBarry Smith SNESGetKSP - Returns the KSP context for a SNES solver. 6239b94acceSBarry Smith 62494b7f48cSBarry Smith Not Collective, but if SNES object is parallel, then KSP object is parallel 625c7afd0dbSLois Curfman McInnes 6269b94acceSBarry Smith Input Parameter: 6279b94acceSBarry Smith . snes - the SNES context 6289b94acceSBarry Smith 6299b94acceSBarry Smith Output Parameter: 63094b7f48cSBarry Smith . ksp - the KSP context 6319b94acceSBarry Smith 6329b94acceSBarry Smith Notes: 63394b7f48cSBarry Smith The user can then directly manipulate the KSP context to set various 6349b94acceSBarry Smith options, etc. Likewise, the user can then extract and manipulate the 6352999313aSBarry Smith PC contexts as well. 6369b94acceSBarry Smith 63736851e7fSLois Curfman McInnes Level: beginner 63836851e7fSLois Curfman McInnes 63994b7f48cSBarry Smith .keywords: SNES, nonlinear, get, KSP, context 6409b94acceSBarry Smith 6412999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 6429b94acceSBarry Smith @*/ 64363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetKSP(SNES snes,KSP *ksp) 6449b94acceSBarry Smith { 6453a40ed3dSBarry Smith PetscFunctionBegin; 6464482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 6474482741eSBarry Smith PetscValidPointer(ksp,2); 64894b7f48cSBarry Smith *ksp = snes->ksp; 6493a40ed3dSBarry Smith PetscFunctionReturn(0); 6509b94acceSBarry Smith } 65182bf6240SBarry Smith 6524a2ae208SSatish Balay #undef __FUNCT__ 6532999313aSBarry Smith #define __FUNCT__ "SNESSetKSP" 6542999313aSBarry Smith /*@ 6552999313aSBarry Smith SNESSetKSP - Sets a KSP context for the SNES object to use 6562999313aSBarry Smith 6572999313aSBarry Smith Not Collective, but the SNES and KSP objects must live on the same MPI_Comm 6582999313aSBarry Smith 6592999313aSBarry Smith Input Parameters: 6602999313aSBarry Smith + snes - the SNES context 6612999313aSBarry Smith - ksp - the KSP context 6622999313aSBarry Smith 6632999313aSBarry Smith Notes: 6642999313aSBarry Smith The SNES object already has its KSP object, you can obtain with SNESGetKSP() 6652999313aSBarry Smith so this routine is rarely needed. 6662999313aSBarry Smith 6672999313aSBarry Smith The KSP object that is already in the SNES object has its reference count 6682999313aSBarry Smith decreased by one. 6692999313aSBarry Smith 6702999313aSBarry Smith Level: developer 6712999313aSBarry Smith 6722999313aSBarry Smith .keywords: SNES, nonlinear, get, KSP, context 6732999313aSBarry Smith 6742999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 6752999313aSBarry Smith @*/ 6762999313aSBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetKSP(SNES snes,KSP ksp) 6772999313aSBarry Smith { 6782999313aSBarry Smith PetscErrorCode ierr; 6792999313aSBarry Smith 6802999313aSBarry Smith PetscFunctionBegin; 6812999313aSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 6822999313aSBarry Smith PetscValidHeaderSpecific(ksp,KSP_COOKIE,2); 6832999313aSBarry Smith PetscCheckSameComm(snes,1,ksp,2); 684*906ed7ccSBarry Smith if (ksp) {ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr);} 685*906ed7ccSBarry Smith if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);} 6862999313aSBarry Smith snes->ksp = ksp; 6872999313aSBarry Smith PetscFunctionReturn(0); 6882999313aSBarry Smith } 6892999313aSBarry Smith 6902999313aSBarry Smith #undef __FUNCT__ 6914a2ae208SSatish Balay #define __FUNCT__ "SNESPublish_Petsc" 6926849ba73SBarry Smith static PetscErrorCode SNESPublish_Petsc(PetscObject obj) 693e24b481bSBarry Smith { 694e24b481bSBarry Smith PetscFunctionBegin; 695e24b481bSBarry Smith PetscFunctionReturn(0); 696e24b481bSBarry Smith } 697e24b481bSBarry Smith 6989b94acceSBarry Smith /* -----------------------------------------------------------*/ 6994a2ae208SSatish Balay #undef __FUNCT__ 7004a2ae208SSatish Balay #define __FUNCT__ "SNESCreate" 70152baeb72SSatish Balay /*@ 7029b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 7039b94acceSBarry Smith 704c7afd0dbSLois Curfman McInnes Collective on MPI_Comm 705c7afd0dbSLois Curfman McInnes 706c7afd0dbSLois Curfman McInnes Input Parameters: 707*906ed7ccSBarry Smith . comm - MPI communicator 7089b94acceSBarry Smith 7099b94acceSBarry Smith Output Parameter: 7109b94acceSBarry Smith . outsnes - the new SNES context 7119b94acceSBarry Smith 712c7afd0dbSLois Curfman McInnes Options Database Keys: 713c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 714c7afd0dbSLois Curfman McInnes and no preconditioning matrix 715c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 716c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 717c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 718c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 719c1f60f51SBarry Smith 72036851e7fSLois Curfman McInnes Level: beginner 72136851e7fSLois Curfman McInnes 7229b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 7239b94acceSBarry Smith 7244b27c08aSLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy(), SNES 7259b94acceSBarry Smith @*/ 72663dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESCreate(MPI_Comm comm,SNES *outsnes) 7279b94acceSBarry Smith { 728dfbe8321SBarry Smith PetscErrorCode ierr; 7299b94acceSBarry Smith SNES snes; 7309b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 73137fcc0dbSBarry Smith 7323a40ed3dSBarry Smith PetscFunctionBegin; 733ed1caa07SMatthew Knepley PetscValidPointer(outsnes,2); 7348ba1e511SMatthew Knepley *outsnes = PETSC_NULL; 7358ba1e511SMatthew Knepley #ifndef PETSC_USE_DYNAMIC_LIBRARIES 7368ba1e511SMatthew Knepley ierr = SNESInitializePackage(PETSC_NULL);CHKERRQ(ierr); 7378ba1e511SMatthew Knepley #endif 7388ba1e511SMatthew Knepley 739e7788613SBarry Smith ierr = PetscHeaderCreate(snes,_p_SNES,struct _SNESOps,SNES_COOKIE,0,"SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr); 740e24b481bSBarry Smith snes->bops->publish = SNESPublish_Petsc; 7419b94acceSBarry Smith snes->max_its = 50; 7429750a799SBarry Smith snes->max_funcs = 10000; 7439b94acceSBarry Smith snes->norm = 0.0; 744b4874afaSBarry Smith snes->rtol = 1.e-8; 745b4874afaSBarry Smith snes->ttol = 0.0; 74670441072SBarry Smith snes->abstol = 1.e-50; 7479b94acceSBarry Smith snes->xtol = 1.e-8; 7484b27c08aSLois Curfman McInnes snes->deltatol = 1.e-12; 7499b94acceSBarry Smith snes->nfuncs = 0; 75050ffb88aSMatthew Knepley snes->numFailures = 0; 75150ffb88aSMatthew Knepley snes->maxFailures = 1; 7527a00f4a9SLois Curfman McInnes snes->linear_its = 0; 753639f9d9dSBarry Smith snes->numbermonitors = 0; 7549b94acceSBarry Smith snes->data = 0; 755e7788613SBarry Smith snes->ops->view = 0; 7564dc4c822SBarry Smith snes->setupcalled = PETSC_FALSE; 757186905e3SBarry Smith snes->ksp_ewconv = PETSC_FALSE; 7586f24a144SLois Curfman McInnes snes->vwork = 0; 7596f24a144SLois Curfman McInnes snes->nwork = 0; 760758f92a0SBarry Smith snes->conv_hist_len = 0; 761758f92a0SBarry Smith snes->conv_hist_max = 0; 762758f92a0SBarry Smith snes->conv_hist = PETSC_NULL; 763758f92a0SBarry Smith snes->conv_hist_its = PETSC_NULL; 764758f92a0SBarry Smith snes->conv_hist_reset = PETSC_TRUE; 765184914b5SBarry Smith snes->reason = SNES_CONVERGED_ITERATING; 7669b94acceSBarry Smith 7673d4c4710SBarry Smith snes->numLinearSolveFailures = 0; 7683d4c4710SBarry Smith snes->maxLinearSolveFailures = 1; 7693d4c4710SBarry Smith 7709b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 771b0a32e0cSBarry Smith ierr = PetscNew(SNES_KSP_EW_ConvCtx,&kctx);CHKERRQ(ierr); 77252e6d16bSBarry Smith ierr = PetscLogObjectMemory(snes,sizeof(SNES_KSP_EW_ConvCtx));CHKERRQ(ierr); 7739b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 7749b94acceSBarry Smith kctx->version = 2; 7759b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 7769b94acceSBarry Smith this was too large for some test cases */ 7779b94acceSBarry Smith kctx->rtol_last = 0; 7789b94acceSBarry Smith kctx->rtol_max = .9; 7799b94acceSBarry Smith kctx->gamma = 1.0; 7809b94acceSBarry Smith kctx->alpha2 = .5*(1.0 + sqrt(5.0)); 7819b94acceSBarry Smith kctx->alpha = kctx->alpha2; 7829b94acceSBarry Smith kctx->threshold = .1; 7839b94acceSBarry Smith kctx->lresid_last = 0; 7849b94acceSBarry Smith kctx->norm_last = 0; 7859b94acceSBarry Smith 78694b7f48cSBarry Smith ierr = KSPCreate(comm,&snes->ksp);CHKERRQ(ierr); 78752e6d16bSBarry Smith ierr = PetscLogObjectParent(snes,snes->ksp);CHKERRQ(ierr); 7885334005bSBarry Smith 7899b94acceSBarry Smith *outsnes = snes; 79000036973SBarry Smith ierr = PetscPublishAll(snes);CHKERRQ(ierr); 7913a40ed3dSBarry Smith PetscFunctionReturn(0); 7929b94acceSBarry Smith } 7939b94acceSBarry Smith 7944a2ae208SSatish Balay #undef __FUNCT__ 7954a2ae208SSatish Balay #define __FUNCT__ "SNESSetFunction" 7969b94acceSBarry Smith /*@C 7979b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 7989b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 7999b94acceSBarry Smith equations. 8009b94acceSBarry Smith 801fee21e36SBarry Smith Collective on SNES 802fee21e36SBarry Smith 803c7afd0dbSLois Curfman McInnes Input Parameters: 804c7afd0dbSLois Curfman McInnes + snes - the SNES context 805c7afd0dbSLois Curfman McInnes . func - function evaluation routine 806c7afd0dbSLois Curfman McInnes . r - vector to store function value 807c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 808c7afd0dbSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 8099b94acceSBarry Smith 810c7afd0dbSLois Curfman McInnes Calling sequence of func: 8118d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Vec f,void *ctx); 812c7afd0dbSLois Curfman McInnes 813313e4042SLois Curfman McInnes . f - function vector 814c7afd0dbSLois Curfman McInnes - ctx - optional user-defined function context 8159b94acceSBarry Smith 8169b94acceSBarry Smith Notes: 8179b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 8189b94acceSBarry Smith $ f'(x) x = -f(x), 819c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 8209b94acceSBarry Smith 82136851e7fSLois Curfman McInnes Level: beginner 82236851e7fSLois Curfman McInnes 8239b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 8249b94acceSBarry Smith 825a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 8269b94acceSBarry Smith @*/ 82763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*func)(SNES,Vec,Vec,void*),void *ctx) 8289b94acceSBarry Smith { 8293a40ed3dSBarry Smith PetscFunctionBegin; 8304482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 8314482741eSBarry Smith PetscValidHeaderSpecific(r,VEC_COOKIE,2); 832c9780b6fSBarry Smith PetscCheckSameComm(snes,1,r,2); 833184914b5SBarry Smith 834e7788613SBarry Smith snes->ops->computefunction = func; 8359b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 8369b94acceSBarry Smith snes->funP = ctx; 8373a40ed3dSBarry Smith PetscFunctionReturn(0); 8389b94acceSBarry Smith } 8399b94acceSBarry Smith 8403ab0aad5SBarry Smith /* --------------------------------------------------------------- */ 8413ab0aad5SBarry Smith #undef __FUNCT__ 8423ab0aad5SBarry Smith #define __FUNCT__ "SNESSetRhs" 8433ab0aad5SBarry Smith /*@C 8443ab0aad5SBarry Smith SNESSetRhs - Sets the vector for solving F(x) = rhs. If rhs is not set 8453ab0aad5SBarry Smith it assumes a zero right hand side. 8463ab0aad5SBarry Smith 8473ab0aad5SBarry Smith Collective on SNES 8483ab0aad5SBarry Smith 8493ab0aad5SBarry Smith Input Parameters: 8503ab0aad5SBarry Smith + snes - the SNES context 8513ab0aad5SBarry Smith - rhs - the right hand side vector or PETSC_NULL for a zero right hand side 8523ab0aad5SBarry Smith 8533ab0aad5SBarry Smith Level: intermediate 8543ab0aad5SBarry Smith 8553ab0aad5SBarry Smith .keywords: SNES, nonlinear, set, function, right hand side 8563ab0aad5SBarry Smith 8571096aae1SMatthew Knepley .seealso: SNESGetRhs(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 8583ab0aad5SBarry Smith @*/ 85963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetRhs(SNES snes,Vec rhs) 8603ab0aad5SBarry Smith { 861dfbe8321SBarry Smith PetscErrorCode ierr; 8623ab0aad5SBarry Smith 8633ab0aad5SBarry Smith PetscFunctionBegin; 8643ab0aad5SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 8653ab0aad5SBarry Smith if (rhs) { 8663ab0aad5SBarry Smith PetscValidHeaderSpecific(rhs,VEC_COOKIE,2); 8673ab0aad5SBarry Smith PetscCheckSameComm(snes,1,rhs,2); 8683ab0aad5SBarry Smith ierr = PetscObjectReference((PetscObject)rhs);CHKERRQ(ierr); 8693ab0aad5SBarry Smith } 8703ab0aad5SBarry Smith if (snes->afine) { 8713ab0aad5SBarry Smith ierr = VecDestroy(snes->afine);CHKERRQ(ierr); 8723ab0aad5SBarry Smith } 8733ab0aad5SBarry Smith snes->afine = rhs; 8743ab0aad5SBarry Smith PetscFunctionReturn(0); 8753ab0aad5SBarry Smith } 8763ab0aad5SBarry Smith 8774a2ae208SSatish Balay #undef __FUNCT__ 8781096aae1SMatthew Knepley #define __FUNCT__ "SNESGetRhs" 8791096aae1SMatthew Knepley /*@C 8801096aae1SMatthew Knepley SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set 8811096aae1SMatthew Knepley it assumes a zero right hand side. 8821096aae1SMatthew Knepley 8831096aae1SMatthew Knepley Collective on SNES 8841096aae1SMatthew Knepley 8851096aae1SMatthew Knepley Input Parameter: 8861096aae1SMatthew Knepley . snes - the SNES context 8871096aae1SMatthew Knepley 8881096aae1SMatthew Knepley Output Parameter: 8891096aae1SMatthew Knepley . rhs - the right hand side vector or PETSC_NULL for a zero right hand side 8901096aae1SMatthew Knepley 8911096aae1SMatthew Knepley Level: intermediate 8921096aae1SMatthew Knepley 8931096aae1SMatthew Knepley .keywords: SNES, nonlinear, get, function, right hand side 8941096aae1SMatthew Knepley 8951096aae1SMatthew Knepley .seealso: SNESSetRhs(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 8961096aae1SMatthew Knepley @*/ 8971096aae1SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESGetRhs(SNES snes,Vec *rhs) 8981096aae1SMatthew Knepley { 8991096aae1SMatthew Knepley PetscFunctionBegin; 9001096aae1SMatthew Knepley PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 9011096aae1SMatthew Knepley PetscValidPointer(rhs,2); 9021096aae1SMatthew Knepley *rhs = snes->afine; 9031096aae1SMatthew Knepley PetscFunctionReturn(0); 9041096aae1SMatthew Knepley } 9051096aae1SMatthew Knepley 9061096aae1SMatthew Knepley #undef __FUNCT__ 9074a2ae208SSatish Balay #define __FUNCT__ "SNESComputeFunction" 9089b94acceSBarry Smith /*@ 90936851e7fSLois Curfman McInnes SNESComputeFunction - Calls the function that has been set with 9109b94acceSBarry Smith SNESSetFunction(). 9119b94acceSBarry Smith 912c7afd0dbSLois Curfman McInnes Collective on SNES 913c7afd0dbSLois Curfman McInnes 9149b94acceSBarry Smith Input Parameters: 915c7afd0dbSLois Curfman McInnes + snes - the SNES context 916c7afd0dbSLois Curfman McInnes - x - input vector 9179b94acceSBarry Smith 9189b94acceSBarry Smith Output Parameter: 9193638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 9209b94acceSBarry Smith 9211bffabb2SLois Curfman McInnes Notes: 92236851e7fSLois Curfman McInnes SNESComputeFunction() is typically used within nonlinear solvers 92336851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 92436851e7fSLois Curfman McInnes themselves. 92536851e7fSLois Curfman McInnes 92636851e7fSLois Curfman McInnes Level: developer 92736851e7fSLois Curfman McInnes 9289b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 9299b94acceSBarry Smith 930a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 9319b94acceSBarry Smith @*/ 93263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESComputeFunction(SNES snes,Vec x,Vec y) 9339b94acceSBarry Smith { 934dfbe8321SBarry Smith PetscErrorCode ierr; 9359b94acceSBarry Smith 9363a40ed3dSBarry Smith PetscFunctionBegin; 9374482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 9384482741eSBarry Smith PetscValidHeaderSpecific(x,VEC_COOKIE,2); 9394482741eSBarry Smith PetscValidHeaderSpecific(y,VEC_COOKIE,3); 940c9780b6fSBarry Smith PetscCheckSameComm(snes,1,x,2); 941c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,3); 942184914b5SBarry Smith 943d5ba7fb7SMatthew Knepley ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 944e7788613SBarry Smith if (snes->ops->computefunction) { 945d64ed03dSBarry Smith PetscStackPush("SNES user function"); 946e9a2bbcdSBarry Smith CHKMEMQ; 947e7788613SBarry Smith ierr = (*snes->ops->computefunction)(snes,x,y,snes->funP); 948e9a2bbcdSBarry Smith CHKMEMQ; 949d64ed03dSBarry Smith PetscStackPop; 950d5e45103SBarry Smith if (PetscExceptionValue(ierr)) { 95119717074SBarry Smith PetscErrorCode pierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(pierr); 95219717074SBarry Smith } 953d5e45103SBarry Smith CHKERRQ(ierr); 9541096aae1SMatthew Knepley } else if (snes->afine) { 9551096aae1SMatthew Knepley ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr); 9561096aae1SMatthew Knepley } else { 9571096aae1SMatthew Knepley SETERRQ(PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() before SNESComputeFunction(), likely called from SNESSolve()."); 9581096aae1SMatthew Knepley } 9593ab0aad5SBarry Smith if (snes->afine) { 960016dedfbSBarry Smith ierr = VecAXPY(y,-1.0,snes->afine);CHKERRQ(ierr); 9613ab0aad5SBarry Smith } 962ae3c334cSLois Curfman McInnes snes->nfuncs++; 963d5ba7fb7SMatthew Knepley ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 9643a40ed3dSBarry Smith PetscFunctionReturn(0); 9659b94acceSBarry Smith } 9669b94acceSBarry Smith 9674a2ae208SSatish Balay #undef __FUNCT__ 9684a2ae208SSatish Balay #define __FUNCT__ "SNESComputeJacobian" 96962fef451SLois Curfman McInnes /*@ 97062fef451SLois Curfman McInnes SNESComputeJacobian - Computes the Jacobian matrix that has been 97162fef451SLois Curfman McInnes set with SNESSetJacobian(). 97262fef451SLois Curfman McInnes 973c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 974c7afd0dbSLois Curfman McInnes 97562fef451SLois Curfman McInnes Input Parameters: 976c7afd0dbSLois Curfman McInnes + snes - the SNES context 977c7afd0dbSLois Curfman McInnes - x - input vector 97862fef451SLois Curfman McInnes 97962fef451SLois Curfman McInnes Output Parameters: 980c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 98162fef451SLois Curfman McInnes . B - optional preconditioning matrix 9822b668275SBarry Smith - flag - flag indicating matrix structure (one of, SAME_NONZERO_PATTERN,DIFFERENT_NONZERO_PATTERN,SAME_PRECONDITIONER) 983fee21e36SBarry Smith 98462fef451SLois Curfman McInnes Notes: 98562fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 98662fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 98762fef451SLois Curfman McInnes 98894b7f48cSBarry Smith See KSPSetOperators() for important information about setting the 989dc5a77f8SLois Curfman McInnes flag parameter. 99062fef451SLois Curfman McInnes 99136851e7fSLois Curfman McInnes Level: developer 99236851e7fSLois Curfman McInnes 99362fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 99462fef451SLois Curfman McInnes 9952b668275SBarry Smith .seealso: SNESSetJacobian(), KSPSetOperators(), MatStructure 99662fef451SLois Curfman McInnes @*/ 99763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 9989b94acceSBarry Smith { 999dfbe8321SBarry Smith PetscErrorCode ierr; 10003a40ed3dSBarry Smith 10013a40ed3dSBarry Smith PetscFunctionBegin; 10024482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 10034482741eSBarry Smith PetscValidHeaderSpecific(X,VEC_COOKIE,2); 10044482741eSBarry Smith PetscValidPointer(flg,5); 1005c9780b6fSBarry Smith PetscCheckSameComm(snes,1,X,2); 1006e7788613SBarry Smith if (!snes->ops->computejacobian) PetscFunctionReturn(0); 1007d5ba7fb7SMatthew Knepley ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,*A,*B);CHKERRQ(ierr); 1008c4fc05e7SBarry Smith *flg = DIFFERENT_NONZERO_PATTERN; 1009d64ed03dSBarry Smith PetscStackPush("SNES user Jacobian function"); 1010dc67937bSBarry Smith CHKMEMQ; 1011e7788613SBarry Smith ierr = (*snes->ops->computejacobian)(snes,X,A,B,flg,snes->jacP);CHKERRQ(ierr); 1012dc67937bSBarry Smith CHKMEMQ; 1013d64ed03dSBarry Smith PetscStackPop; 1014d5ba7fb7SMatthew Knepley ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,*A,*B);CHKERRQ(ierr); 10156d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 10166ce558aeSBarry Smith /* PetscValidHeaderSpecific(*A,MAT_COOKIE,3); 10176ce558aeSBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE,4); */ 10183a40ed3dSBarry Smith PetscFunctionReturn(0); 10199b94acceSBarry Smith } 10209b94acceSBarry Smith 10214a2ae208SSatish Balay #undef __FUNCT__ 10224a2ae208SSatish Balay #define __FUNCT__ "SNESSetJacobian" 10239b94acceSBarry Smith /*@C 10249b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 1025044dda88SLois Curfman McInnes location to store the matrix. 10269b94acceSBarry Smith 1027c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1028c7afd0dbSLois Curfman McInnes 10299b94acceSBarry Smith Input Parameters: 1030c7afd0dbSLois Curfman McInnes + snes - the SNES context 10319b94acceSBarry Smith . A - Jacobian matrix 10329b94acceSBarry Smith . B - preconditioner matrix (usually same as the Jacobian) 10339b94acceSBarry Smith . func - Jacobian evaluation routine 1034c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 10352cd2dfdcSLois Curfman McInnes Jacobian evaluation routine (may be PETSC_NULL) 10369b94acceSBarry Smith 10379b94acceSBarry Smith Calling sequence of func: 10388d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 10399b94acceSBarry Smith 1040c7afd0dbSLois Curfman McInnes + x - input vector 10419b94acceSBarry Smith . A - Jacobian matrix 10429b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1043ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 10442b668275SBarry Smith structure (same as flag in KSPSetOperators()), one of SAME_NONZERO_PATTERN,DIFFERENT_NONZERO_PATTERN,SAME_PRECONDITIONER 1045c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined Jacobian context 10469b94acceSBarry Smith 10479b94acceSBarry Smith Notes: 104894b7f48cSBarry Smith See KSPSetOperators() for important information about setting the flag 10492cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1050ac21db08SLois Curfman McInnes 1051ac21db08SLois Curfman McInnes The routine func() takes Mat * as the matrix arguments rather than Mat. 10529b94acceSBarry Smith This allows the Jacobian evaluation routine to replace A and/or B with a 10539b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 10549b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 10559b94acceSBarry Smith throughout the global iterations. 10569b94acceSBarry Smith 105736851e7fSLois Curfman McInnes Level: beginner 105836851e7fSLois Curfman McInnes 10599b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 10609b94acceSBarry Smith 10612b668275SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatSNESMFComputeJacobian(), SNESDefaultComputeJacobianColor(), MatStructure 10629b94acceSBarry Smith @*/ 106363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetJacobian(SNES snes,Mat A,Mat B,PetscErrorCode (*func)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void *ctx) 10649b94acceSBarry Smith { 1065dfbe8321SBarry Smith PetscErrorCode ierr; 10663a7fca6bSBarry Smith 10673a40ed3dSBarry Smith PetscFunctionBegin; 10684482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 10694482741eSBarry Smith if (A) PetscValidHeaderSpecific(A,MAT_COOKIE,2); 10704482741eSBarry Smith if (B) PetscValidHeaderSpecific(B,MAT_COOKIE,3); 1071c9780b6fSBarry Smith if (A) PetscCheckSameComm(snes,1,A,2); 1072c9780b6fSBarry Smith if (B) PetscCheckSameComm(snes,1,B,2); 1073e7788613SBarry Smith if (func) snes->ops->computejacobian = func; 10743a7fca6bSBarry Smith if (ctx) snes->jacP = ctx; 10753a7fca6bSBarry Smith if (A) { 10763a7fca6bSBarry Smith if (snes->jacobian) {ierr = MatDestroy(snes->jacobian);CHKERRQ(ierr);} 10779b94acceSBarry Smith snes->jacobian = A; 10783a7fca6bSBarry Smith ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 10793a7fca6bSBarry Smith } 10803a7fca6bSBarry Smith if (B) { 10813a7fca6bSBarry Smith if (snes->jacobian_pre) {ierr = MatDestroy(snes->jacobian_pre);CHKERRQ(ierr);} 10829b94acceSBarry Smith snes->jacobian_pre = B; 10833a7fca6bSBarry Smith ierr = PetscObjectReference((PetscObject)B);CHKERRQ(ierr); 10843a7fca6bSBarry Smith } 108569a612a9SBarry Smith ierr = KSPSetOperators(snes->ksp,A,B,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 10863a40ed3dSBarry Smith PetscFunctionReturn(0); 10879b94acceSBarry Smith } 108862fef451SLois Curfman McInnes 10894a2ae208SSatish Balay #undef __FUNCT__ 10904a2ae208SSatish Balay #define __FUNCT__ "SNESGetJacobian" 1091c2aafc4cSSatish Balay /*@C 1092b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 1093b4fd4287SBarry Smith provided context for evaluating the Jacobian. 1094b4fd4287SBarry Smith 1095c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 1096c7afd0dbSLois Curfman McInnes 1097b4fd4287SBarry Smith Input Parameter: 1098b4fd4287SBarry Smith . snes - the nonlinear solver context 1099b4fd4287SBarry Smith 1100b4fd4287SBarry Smith Output Parameters: 1101c7afd0dbSLois Curfman McInnes + A - location to stash Jacobian matrix (or PETSC_NULL) 1102b4fd4287SBarry Smith . B - location to stash preconditioner matrix (or PETSC_NULL) 110370e92668SMatthew Knepley . func - location to put Jacobian function (or PETSC_NULL) 110470e92668SMatthew Knepley - ctx - location to stash Jacobian ctx (or PETSC_NULL) 1105fee21e36SBarry Smith 110636851e7fSLois Curfman McInnes Level: advanced 110736851e7fSLois Curfman McInnes 1108b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian() 1109b4fd4287SBarry Smith @*/ 111070e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESGetJacobian(SNES snes,Mat *A,Mat *B,PetscErrorCode (**func)(SNES,Vec,Mat*,Mat*,MatStructure*,void*),void **ctx) 1111b4fd4287SBarry Smith { 11123a40ed3dSBarry Smith PetscFunctionBegin; 11134482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 1114b4fd4287SBarry Smith if (A) *A = snes->jacobian; 1115b4fd4287SBarry Smith if (B) *B = snes->jacobian_pre; 1116e7788613SBarry Smith if (func) *func = snes->ops->computejacobian; 111770e92668SMatthew Knepley if (ctx) *ctx = snes->jacP; 11183a40ed3dSBarry Smith PetscFunctionReturn(0); 1119b4fd4287SBarry Smith } 1120b4fd4287SBarry Smith 11219b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */ 112263dd3a1aSKris Buschelman EXTERN PetscErrorCode PETSCSNES_DLLEXPORT SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*); 11239b94acceSBarry Smith 11244a2ae208SSatish Balay #undef __FUNCT__ 11254a2ae208SSatish Balay #define __FUNCT__ "SNESSetUp" 11269b94acceSBarry Smith /*@ 11279b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 1128272ac6f2SLois Curfman McInnes of a nonlinear solver. 11299b94acceSBarry Smith 1130fee21e36SBarry Smith Collective on SNES 1131fee21e36SBarry Smith 1132c7afd0dbSLois Curfman McInnes Input Parameters: 113370e92668SMatthew Knepley . snes - the SNES context 1134c7afd0dbSLois Curfman McInnes 1135272ac6f2SLois Curfman McInnes Notes: 1136272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 1137272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 1138272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 1139272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 1140272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 1141272ac6f2SLois Curfman McInnes 114236851e7fSLois Curfman McInnes Level: advanced 114336851e7fSLois Curfman McInnes 11449b94acceSBarry Smith .keywords: SNES, nonlinear, setup 11459b94acceSBarry Smith 11469b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 11479b94acceSBarry Smith @*/ 114870e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESSetUp(SNES snes) 11499b94acceSBarry Smith { 1150dfbe8321SBarry Smith PetscErrorCode ierr; 11514b27c08aSLois Curfman McInnes PetscTruth flg, iseqtr; 11523a40ed3dSBarry Smith 11533a40ed3dSBarry Smith PetscFunctionBegin; 11544482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 11554dc4c822SBarry Smith if (snes->setupcalled) PetscFunctionReturn(0); 11569b94acceSBarry Smith 1157b0a32e0cSBarry Smith ierr = PetscOptionsHasName(snes->prefix,"-snes_mf_operator",&flg);CHKERRQ(ierr); 1158c1f60f51SBarry Smith /* 1159c1f60f51SBarry Smith This version replaces the user provided Jacobian matrix with a 1160dfa02198SLois Curfman McInnes matrix-free version but still employs the user-provided preconditioner matrix 1161c1f60f51SBarry Smith */ 1162c1f60f51SBarry Smith if (flg) { 1163c1f60f51SBarry Smith Mat J; 11645a655dc6SBarry Smith ierr = MatCreateSNESMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 11655a655dc6SBarry Smith ierr = MatSNESMFSetFromOptions(J);CHKERRQ(ierr); 1166ae15b995SBarry Smith ierr = PetscInfo(snes,"Setting default matrix-free operator routines\n");CHKERRQ(ierr); 11673a7fca6bSBarry Smith ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr); 11683a7fca6bSBarry Smith ierr = MatDestroy(J);CHKERRQ(ierr); 1169c1f60f51SBarry Smith } 117045fc7adcSBarry Smith 117103c60df9SBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_SINGLE) && !defined(PETSC_USE_MAT_SINGLE) && !defined(PETSC_USE_LONG_DOUBLE) && !defined(PETSC_USE_INT) 117245fc7adcSBarry Smith ierr = PetscOptionsHasName(snes->prefix,"-snes_mf_operator2",&flg);CHKERRQ(ierr); 117345fc7adcSBarry Smith if (flg) { 117445fc7adcSBarry Smith Mat J; 117545fc7adcSBarry Smith ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_sol,&J);CHKERRQ(ierr); 117645fc7adcSBarry Smith ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr); 117745fc7adcSBarry Smith ierr = MatDestroy(J);CHKERRQ(ierr); 117845fc7adcSBarry Smith } 117932a4b47aSMatthew Knepley #endif 118045fc7adcSBarry Smith 1181b0a32e0cSBarry Smith ierr = PetscOptionsHasName(snes->prefix,"-snes_mf",&flg);CHKERRQ(ierr); 1182c1f60f51SBarry Smith /* 1183dfa02198SLois Curfman McInnes This version replaces both the user-provided Jacobian and the user- 1184c1f60f51SBarry Smith provided preconditioner matrix with the default matrix free version. 1185c1f60f51SBarry Smith */ 118631615d04SLois Curfman McInnes if (flg) { 1187272ac6f2SLois Curfman McInnes Mat J; 1188b5d62d44SBarry Smith KSP ksp; 118994b7f48cSBarry Smith PC pc; 1190f3ef73ceSBarry Smith 11915a655dc6SBarry Smith ierr = MatCreateSNESMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 11923a7fca6bSBarry Smith ierr = MatSNESMFSetFromOptions(J);CHKERRQ(ierr); 1193ae15b995SBarry Smith ierr = PetscInfo(snes,"Setting default matrix-free operator and preconditioner routines;\nThat is no preconditioner is being used.\n");CHKERRQ(ierr); 11943a7fca6bSBarry Smith ierr = SNESSetJacobian(snes,J,J,MatSNESMFComputeJacobian,snes->funP);CHKERRQ(ierr); 11953a7fca6bSBarry Smith ierr = MatDestroy(J);CHKERRQ(ierr); 11963a7fca6bSBarry Smith 1197f3ef73ceSBarry Smith /* force no preconditioner */ 119894b7f48cSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 1199b5d62d44SBarry Smith ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr); 1200a9815358SBarry Smith ierr = PetscTypeCompare((PetscObject)pc,PCSHELL,&flg);CHKERRQ(ierr); 1201a9815358SBarry Smith if (!flg) { 1202f3ef73ceSBarry Smith ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr); 1203272ac6f2SLois Curfman McInnes } 1204a9815358SBarry Smith } 1205f3ef73ceSBarry Smith 12061096aae1SMatthew Knepley if (!snes->vec_func && !snes->afine) { 12071096aae1SMatthew Knepley SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetFunction() first"); 12081096aae1SMatthew Knepley } 1209e7788613SBarry Smith if (!snes->ops->computefunction && !snes->afine) { 12101096aae1SMatthew Knepley SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetFunction() first"); 12111096aae1SMatthew Knepley } 121229bbc08cSBarry Smith if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetJacobian() first \n or use -snes_mf option"); 1213a8c6a408SBarry Smith if (snes->vec_func == snes->vec_sol) { 121429bbc08cSBarry Smith SETERRQ(PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector"); 1215a8c6a408SBarry Smith } 1216a703fe33SLois Curfman McInnes 1217a703fe33SLois Curfman McInnes /* Set the KSP stopping criterion to use the Eisenstat-Walker method */ 12184b27c08aSLois Curfman McInnes ierr = PetscTypeCompare((PetscObject)snes,SNESTR,&iseqtr);CHKERRQ(ierr); 12196831982aSBarry Smith if (snes->ksp_ewconv && !iseqtr) { 122094b7f48cSBarry Smith KSP ksp; 122194b7f48cSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 1222186905e3SBarry Smith ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private,snes);CHKERRQ(ierr); 1223a703fe33SLois Curfman McInnes } 12244b27c08aSLois Curfman McInnes 1225e7788613SBarry Smith if (snes->ops->setup) {ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr);} 12267aaed0d8SBarry Smith snes->setupcalled = PETSC_TRUE; 12273a40ed3dSBarry Smith PetscFunctionReturn(0); 12289b94acceSBarry Smith } 12299b94acceSBarry Smith 12304a2ae208SSatish Balay #undef __FUNCT__ 12314a2ae208SSatish Balay #define __FUNCT__ "SNESDestroy" 123252baeb72SSatish Balay /*@ 12339b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 12349b94acceSBarry Smith with SNESCreate(). 12359b94acceSBarry Smith 1236c7afd0dbSLois Curfman McInnes Collective on SNES 1237c7afd0dbSLois Curfman McInnes 12389b94acceSBarry Smith Input Parameter: 12399b94acceSBarry Smith . snes - the SNES context 12409b94acceSBarry Smith 124136851e7fSLois Curfman McInnes Level: beginner 124236851e7fSLois Curfman McInnes 12439b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 12449b94acceSBarry Smith 124563a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 12469b94acceSBarry Smith @*/ 124763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESDestroy(SNES snes) 12489b94acceSBarry Smith { 12496849ba73SBarry Smith PetscErrorCode ierr; 12503a40ed3dSBarry Smith 12513a40ed3dSBarry Smith PetscFunctionBegin; 12524482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 12533a40ed3dSBarry Smith if (--snes->refct > 0) PetscFunctionReturn(0); 1254d4bb536fSBarry Smith 1255be0abb6dSBarry Smith /* if memory was published with AMS then destroy it */ 12560f5bd95cSBarry Smith ierr = PetscObjectDepublish(snes);CHKERRQ(ierr); 1257be0abb6dSBarry Smith 1258e7788613SBarry Smith if (snes->ops->destroy) {ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr);} 125905b42c5fSBarry Smith ierr = PetscFree(snes->kspconvctx);CHKERRQ(ierr); 12603a7fca6bSBarry Smith if (snes->jacobian) {ierr = MatDestroy(snes->jacobian);CHKERRQ(ierr);} 12613a7fca6bSBarry Smith if (snes->jacobian_pre) {ierr = MatDestroy(snes->jacobian_pre);CHKERRQ(ierr);} 12623ab0aad5SBarry Smith if (snes->afine) {ierr = VecDestroy(snes->afine);CHKERRQ(ierr);} 126394b7f48cSBarry Smith ierr = KSPDestroy(snes->ksp);CHKERRQ(ierr); 1264522c5e43SBarry Smith if (snes->vwork) {ierr = VecDestroyVecs(snes->vwork,snes->nvwork);CHKERRQ(ierr);} 1265d952e501SBarry Smith ierr = SNESClearMonitor(snes);CHKERRQ(ierr); 1266a79aaaedSSatish Balay ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr); 12673a40ed3dSBarry Smith PetscFunctionReturn(0); 12689b94acceSBarry Smith } 12699b94acceSBarry Smith 12709b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 12719b94acceSBarry Smith 12724a2ae208SSatish Balay #undef __FUNCT__ 12734a2ae208SSatish Balay #define __FUNCT__ "SNESSetTolerances" 12749b94acceSBarry Smith /*@ 1275d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 12769b94acceSBarry Smith 1277c7afd0dbSLois Curfman McInnes Collective on SNES 1278c7afd0dbSLois Curfman McInnes 12799b94acceSBarry Smith Input Parameters: 1280c7afd0dbSLois Curfman McInnes + snes - the SNES context 128170441072SBarry Smith . abstol - absolute convergence tolerance 128233174efeSLois Curfman McInnes . rtol - relative convergence tolerance 128333174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 128433174efeSLois Curfman McInnes of the change in the solution between steps 128533174efeSLois Curfman McInnes . maxit - maximum number of iterations 1286c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1287fee21e36SBarry Smith 128833174efeSLois Curfman McInnes Options Database Keys: 128970441072SBarry Smith + -snes_atol <abstol> - Sets abstol 1290c7afd0dbSLois Curfman McInnes . -snes_rtol <rtol> - Sets rtol 1291c7afd0dbSLois Curfman McInnes . -snes_stol <stol> - Sets stol 1292c7afd0dbSLois Curfman McInnes . -snes_max_it <maxit> - Sets maxit 1293c7afd0dbSLois Curfman McInnes - -snes_max_funcs <maxf> - Sets maxf 12949b94acceSBarry Smith 1295d7a720efSLois Curfman McInnes Notes: 12969b94acceSBarry Smith The default maximum number of iterations is 50. 12979b94acceSBarry Smith The default maximum number of function evaluations is 1000. 12989b94acceSBarry Smith 129936851e7fSLois Curfman McInnes Level: intermediate 130036851e7fSLois Curfman McInnes 130133174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances 13029b94acceSBarry Smith 13032492ecdbSBarry Smith .seealso: SNESSetTrustRegionTolerance() 13049b94acceSBarry Smith @*/ 130563dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf) 13069b94acceSBarry Smith { 13073a40ed3dSBarry Smith PetscFunctionBegin; 13084482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 130970441072SBarry Smith if (abstol != PETSC_DEFAULT) snes->abstol = abstol; 1310d7a720efSLois Curfman McInnes if (rtol != PETSC_DEFAULT) snes->rtol = rtol; 1311d7a720efSLois Curfman McInnes if (stol != PETSC_DEFAULT) snes->xtol = stol; 1312d7a720efSLois Curfman McInnes if (maxit != PETSC_DEFAULT) snes->max_its = maxit; 1313d7a720efSLois Curfman McInnes if (maxf != PETSC_DEFAULT) snes->max_funcs = maxf; 13143a40ed3dSBarry Smith PetscFunctionReturn(0); 13159b94acceSBarry Smith } 13169b94acceSBarry Smith 13174a2ae208SSatish Balay #undef __FUNCT__ 13184a2ae208SSatish Balay #define __FUNCT__ "SNESGetTolerances" 13199b94acceSBarry Smith /*@ 132033174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 132133174efeSLois Curfman McInnes 1322c7afd0dbSLois Curfman McInnes Not Collective 1323c7afd0dbSLois Curfman McInnes 132433174efeSLois Curfman McInnes Input Parameters: 1325c7afd0dbSLois Curfman McInnes + snes - the SNES context 132670441072SBarry Smith . abstol - absolute convergence tolerance 132733174efeSLois Curfman McInnes . rtol - relative convergence tolerance 132833174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 132933174efeSLois Curfman McInnes of the change in the solution between steps 133033174efeSLois Curfman McInnes . maxit - maximum number of iterations 1331c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1332fee21e36SBarry Smith 133333174efeSLois Curfman McInnes Notes: 133433174efeSLois Curfman McInnes The user can specify PETSC_NULL for any parameter that is not needed. 133533174efeSLois Curfman McInnes 133636851e7fSLois Curfman McInnes Level: intermediate 133736851e7fSLois Curfman McInnes 133833174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances 133933174efeSLois Curfman McInnes 134033174efeSLois Curfman McInnes .seealso: SNESSetTolerances() 134133174efeSLois Curfman McInnes @*/ 134263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetTolerances(SNES snes,PetscReal *abstol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf) 134333174efeSLois Curfman McInnes { 13443a40ed3dSBarry Smith PetscFunctionBegin; 13454482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 134670441072SBarry Smith if (abstol) *abstol = snes->abstol; 134733174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 134833174efeSLois Curfman McInnes if (stol) *stol = snes->xtol; 134933174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 135033174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 13513a40ed3dSBarry Smith PetscFunctionReturn(0); 135233174efeSLois Curfman McInnes } 135333174efeSLois Curfman McInnes 13544a2ae208SSatish Balay #undef __FUNCT__ 13554a2ae208SSatish Balay #define __FUNCT__ "SNESSetTrustRegionTolerance" 135633174efeSLois Curfman McInnes /*@ 13579b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 13589b94acceSBarry Smith 1359fee21e36SBarry Smith Collective on SNES 1360fee21e36SBarry Smith 1361c7afd0dbSLois Curfman McInnes Input Parameters: 1362c7afd0dbSLois Curfman McInnes + snes - the SNES context 1363c7afd0dbSLois Curfman McInnes - tol - tolerance 1364c7afd0dbSLois Curfman McInnes 13659b94acceSBarry Smith Options Database Key: 1366c7afd0dbSLois Curfman McInnes . -snes_trtol <tol> - Sets tol 13679b94acceSBarry Smith 136836851e7fSLois Curfman McInnes Level: intermediate 136936851e7fSLois Curfman McInnes 13709b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 13719b94acceSBarry Smith 13722492ecdbSBarry Smith .seealso: SNESSetTolerances() 13739b94acceSBarry Smith @*/ 137463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetTrustRegionTolerance(SNES snes,PetscReal tol) 13759b94acceSBarry Smith { 13763a40ed3dSBarry Smith PetscFunctionBegin; 13774482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 13789b94acceSBarry Smith snes->deltatol = tol; 13793a40ed3dSBarry Smith PetscFunctionReturn(0); 13809b94acceSBarry Smith } 13819b94acceSBarry Smith 1382df9fa365SBarry Smith /* 1383df9fa365SBarry Smith Duplicate the lg monitors for SNES from KSP; for some reason with 1384df9fa365SBarry Smith dynamic libraries things don't work under Sun4 if we just use 1385df9fa365SBarry Smith macros instead of functions 1386df9fa365SBarry Smith */ 13874a2ae208SSatish Balay #undef __FUNCT__ 13884a2ae208SSatish Balay #define __FUNCT__ "SNESLGMonitor" 138963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESLGMonitor(SNES snes,PetscInt it,PetscReal norm,void *ctx) 1390ce1608b8SBarry Smith { 1391dfbe8321SBarry Smith PetscErrorCode ierr; 1392ce1608b8SBarry Smith 1393ce1608b8SBarry Smith PetscFunctionBegin; 13944482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 1395ce1608b8SBarry Smith ierr = KSPLGMonitor((KSP)snes,it,norm,ctx);CHKERRQ(ierr); 1396ce1608b8SBarry Smith PetscFunctionReturn(0); 1397ce1608b8SBarry Smith } 1398ce1608b8SBarry Smith 13994a2ae208SSatish Balay #undef __FUNCT__ 14004a2ae208SSatish Balay #define __FUNCT__ "SNESLGMonitorCreate" 140163dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESLGMonitorCreate(const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *draw) 1402df9fa365SBarry Smith { 1403dfbe8321SBarry Smith PetscErrorCode ierr; 1404df9fa365SBarry Smith 1405df9fa365SBarry Smith PetscFunctionBegin; 1406df9fa365SBarry Smith ierr = KSPLGMonitorCreate(host,label,x,y,m,n,draw);CHKERRQ(ierr); 1407df9fa365SBarry Smith PetscFunctionReturn(0); 1408df9fa365SBarry Smith } 1409df9fa365SBarry Smith 14104a2ae208SSatish Balay #undef __FUNCT__ 14114a2ae208SSatish Balay #define __FUNCT__ "SNESLGMonitorDestroy" 141263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESLGMonitorDestroy(PetscDrawLG draw) 1413df9fa365SBarry Smith { 1414dfbe8321SBarry Smith PetscErrorCode ierr; 1415df9fa365SBarry Smith 1416df9fa365SBarry Smith PetscFunctionBegin; 1417df9fa365SBarry Smith ierr = KSPLGMonitorDestroy(draw);CHKERRQ(ierr); 1418df9fa365SBarry Smith PetscFunctionReturn(0); 1419df9fa365SBarry Smith } 1420df9fa365SBarry Smith 14219b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 14229b94acceSBarry Smith 14234a2ae208SSatish Balay #undef __FUNCT__ 14244a2ae208SSatish Balay #define __FUNCT__ "SNESSetMonitor" 14259b94acceSBarry Smith /*@C 14265cd90555SBarry Smith SNESSetMonitor - Sets an ADDITIONAL function that is to be used at every 14279b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 14289b94acceSBarry Smith progress. 14299b94acceSBarry Smith 1430fee21e36SBarry Smith Collective on SNES 1431fee21e36SBarry Smith 1432c7afd0dbSLois Curfman McInnes Input Parameters: 1433c7afd0dbSLois Curfman McInnes + snes - the SNES context 1434c7afd0dbSLois Curfman McInnes . func - monitoring routine 1435b8a78c4aSBarry Smith . mctx - [optional] user-defined context for private data for the 1436e8105e01SRichard Katz monitor routine (use PETSC_NULL if no context is desired) 1437b3006f0bSLois Curfman McInnes - monitordestroy - [optional] routine that frees monitor context 1438b3006f0bSLois Curfman McInnes (may be PETSC_NULL) 14399b94acceSBarry Smith 1440c7afd0dbSLois Curfman McInnes Calling sequence of func: 1441a7cc72afSBarry Smith $ int func(SNES snes,PetscInt its, PetscReal norm,void *mctx) 1442c7afd0dbSLois Curfman McInnes 1443c7afd0dbSLois Curfman McInnes + snes - the SNES context 1444c7afd0dbSLois Curfman McInnes . its - iteration number 1445c7afd0dbSLois Curfman McInnes . norm - 2-norm function value (may be estimated) 144640a0c1c6SLois Curfman McInnes - mctx - [optional] monitoring context 14479b94acceSBarry Smith 14489665c990SLois Curfman McInnes Options Database Keys: 1449c7afd0dbSLois Curfman McInnes + -snes_monitor - sets SNESDefaultMonitor() 1450c7afd0dbSLois Curfman McInnes . -snes_xmonitor - sets line graph monitor, 1451c7afd0dbSLois Curfman McInnes uses SNESLGMonitorCreate() 1452c7afd0dbSLois Curfman McInnes _ -snes_cancelmonitors - cancels all monitors that have 1453c7afd0dbSLois Curfman McInnes been hardwired into a code by 1454c7afd0dbSLois Curfman McInnes calls to SNESSetMonitor(), but 1455c7afd0dbSLois Curfman McInnes does not cancel those set via 1456c7afd0dbSLois Curfman McInnes the options database. 14579665c990SLois Curfman McInnes 1458639f9d9dSBarry Smith Notes: 14596bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 14606bc08f3fSLois Curfman McInnes SNESSetMonitor() multiple times; all will be called in the 14616bc08f3fSLois Curfman McInnes order in which they were set. 1462639f9d9dSBarry Smith 146336851e7fSLois Curfman McInnes Level: intermediate 146436851e7fSLois Curfman McInnes 14659b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 14669b94acceSBarry Smith 14675cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESClearMonitor() 14689b94acceSBarry Smith @*/ 146963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetMonitor(SNES snes,PetscErrorCode (*func)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void*)) 14709b94acceSBarry Smith { 14713a40ed3dSBarry Smith PetscFunctionBegin; 14724482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 1473639f9d9dSBarry Smith if (snes->numbermonitors >= MAXSNESMONITORS) { 147429bbc08cSBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set"); 1475639f9d9dSBarry Smith } 1476639f9d9dSBarry Smith snes->monitor[snes->numbermonitors] = func; 1477b8a78c4aSBarry Smith snes->monitordestroy[snes->numbermonitors] = monitordestroy; 1478639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 14793a40ed3dSBarry Smith PetscFunctionReturn(0); 14809b94acceSBarry Smith } 14819b94acceSBarry Smith 14824a2ae208SSatish Balay #undef __FUNCT__ 14834a2ae208SSatish Balay #define __FUNCT__ "SNESClearMonitor" 14845cd90555SBarry Smith /*@C 14855cd90555SBarry Smith SNESClearMonitor - Clears all the monitor functions for a SNES object. 14865cd90555SBarry Smith 1487c7afd0dbSLois Curfman McInnes Collective on SNES 1488c7afd0dbSLois Curfman McInnes 14895cd90555SBarry Smith Input Parameters: 14905cd90555SBarry Smith . snes - the SNES context 14915cd90555SBarry Smith 14921a480d89SAdministrator Options Database Key: 1493c7afd0dbSLois Curfman McInnes . -snes_cancelmonitors - cancels all monitors that have been hardwired 1494c7afd0dbSLois Curfman McInnes into a code by calls to SNESSetMonitor(), but does not cancel those 1495c7afd0dbSLois Curfman McInnes set via the options database 14965cd90555SBarry Smith 14975cd90555SBarry Smith Notes: 14985cd90555SBarry Smith There is no way to clear one specific monitor from a SNES object. 14995cd90555SBarry Smith 150036851e7fSLois Curfman McInnes Level: intermediate 150136851e7fSLois Curfman McInnes 15025cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor 15035cd90555SBarry Smith 15045cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESSetMonitor() 15055cd90555SBarry Smith @*/ 150663dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESClearMonitor(SNES snes) 15075cd90555SBarry Smith { 1508d952e501SBarry Smith PetscErrorCode ierr; 1509d952e501SBarry Smith PetscInt i; 1510d952e501SBarry Smith 15115cd90555SBarry Smith PetscFunctionBegin; 15124482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 1513d952e501SBarry Smith for (i=0; i<snes->numbermonitors; i++) { 1514d952e501SBarry Smith if (snes->monitordestroy[i]) { 1515d952e501SBarry Smith ierr = (*snes->monitordestroy[i])(snes->monitorcontext[i]);CHKERRQ(ierr); 1516d952e501SBarry Smith } 1517d952e501SBarry Smith } 15185cd90555SBarry Smith snes->numbermonitors = 0; 15195cd90555SBarry Smith PetscFunctionReturn(0); 15205cd90555SBarry Smith } 15215cd90555SBarry Smith 15224a2ae208SSatish Balay #undef __FUNCT__ 15234a2ae208SSatish Balay #define __FUNCT__ "SNESSetConvergenceTest" 15249b94acceSBarry Smith /*@C 15259b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 15269b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 15279b94acceSBarry Smith 1528fee21e36SBarry Smith Collective on SNES 1529fee21e36SBarry Smith 1530c7afd0dbSLois Curfman McInnes Input Parameters: 1531c7afd0dbSLois Curfman McInnes + snes - the SNES context 1532c7afd0dbSLois Curfman McInnes . func - routine to test for convergence 1533c7afd0dbSLois Curfman McInnes - cctx - [optional] context for private data for the convergence routine 1534c7afd0dbSLois Curfman McInnes (may be PETSC_NULL) 15359b94acceSBarry Smith 1536c7afd0dbSLois Curfman McInnes Calling sequence of func: 153706ee9f85SBarry Smith $ PetscErrorCode func (SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx) 1538c7afd0dbSLois Curfman McInnes 1539c7afd0dbSLois Curfman McInnes + snes - the SNES context 154006ee9f85SBarry Smith . it - current iteration (0 is the first and is before any Newton step) 1541c7afd0dbSLois Curfman McInnes . cctx - [optional] convergence context 1542184914b5SBarry Smith . reason - reason for convergence/divergence 1543c7afd0dbSLois Curfman McInnes . xnorm - 2-norm of current iterate 15444b27c08aSLois Curfman McInnes . gnorm - 2-norm of current step 15454b27c08aSLois Curfman McInnes - f - 2-norm of function 15469b94acceSBarry Smith 154736851e7fSLois Curfman McInnes Level: advanced 154836851e7fSLois Curfman McInnes 15499b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 15509b94acceSBarry Smith 15514b27c08aSLois Curfman McInnes .seealso: SNESConverged_LS(), SNESConverged_TR() 15529b94acceSBarry Smith @*/ 155306ee9f85SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetConvergenceTest(SNES snes,PetscErrorCode (*func)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx) 15549b94acceSBarry Smith { 15553a40ed3dSBarry Smith PetscFunctionBegin; 15564482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 1557e7788613SBarry Smith (snes)->ops->converged = func; 15589b94acceSBarry Smith (snes)->cnvP = cctx; 15593a40ed3dSBarry Smith PetscFunctionReturn(0); 15609b94acceSBarry Smith } 15619b94acceSBarry Smith 15624a2ae208SSatish Balay #undef __FUNCT__ 15634a2ae208SSatish Balay #define __FUNCT__ "SNESGetConvergedReason" 156452baeb72SSatish Balay /*@ 1565184914b5SBarry Smith SNESGetConvergedReason - Gets the reason the SNES iteration was stopped. 1566184914b5SBarry Smith 1567184914b5SBarry Smith Not Collective 1568184914b5SBarry Smith 1569184914b5SBarry Smith Input Parameter: 1570184914b5SBarry Smith . snes - the SNES context 1571184914b5SBarry Smith 1572184914b5SBarry Smith Output Parameter: 1573e090d566SSatish Balay . reason - negative value indicates diverged, positive value converged, see petscsnes.h or the 1574184914b5SBarry Smith manual pages for the individual convergence tests for complete lists 1575184914b5SBarry Smith 1576184914b5SBarry Smith Level: intermediate 1577184914b5SBarry Smith 1578184914b5SBarry Smith Notes: Can only be called after the call the SNESSolve() is complete. 1579184914b5SBarry Smith 1580184914b5SBarry Smith .keywords: SNES, nonlinear, set, convergence, test 1581184914b5SBarry Smith 15824b27c08aSLois Curfman McInnes .seealso: SNESSetConvergenceTest(), SNESConverged_LS(), SNESConverged_TR(), SNESConvergedReason 1583184914b5SBarry Smith @*/ 158463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason) 1585184914b5SBarry Smith { 1586184914b5SBarry Smith PetscFunctionBegin; 15874482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 15884482741eSBarry Smith PetscValidPointer(reason,2); 1589184914b5SBarry Smith *reason = snes->reason; 1590184914b5SBarry Smith PetscFunctionReturn(0); 1591184914b5SBarry Smith } 1592184914b5SBarry Smith 15934a2ae208SSatish Balay #undef __FUNCT__ 15944a2ae208SSatish Balay #define __FUNCT__ "SNESSetConvergenceHistory" 1595c9005455SLois Curfman McInnes /*@ 1596c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 1597c9005455SLois Curfman McInnes 1598fee21e36SBarry Smith Collective on SNES 1599fee21e36SBarry Smith 1600c7afd0dbSLois Curfman McInnes Input Parameters: 1601c7afd0dbSLois Curfman McInnes + snes - iterative context obtained from SNESCreate() 1602c7afd0dbSLois Curfman McInnes . a - array to hold history 1603cd5578b5SBarry Smith . its - integer array holds the number of linear iterations for each solve. 1604758f92a0SBarry Smith . na - size of a and its 160564731454SLois Curfman McInnes - reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero, 1606758f92a0SBarry Smith else it continues storing new values for new nonlinear solves after the old ones 1607c7afd0dbSLois Curfman McInnes 1608c9005455SLois Curfman McInnes Notes: 16094b27c08aSLois Curfman McInnes If set, this array will contain the function norms computed 1610c9005455SLois Curfman McInnes at each step. 1611c9005455SLois Curfman McInnes 1612c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 1613c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 1614c9005455SLois Curfman McInnes during the section of code that is being timed. 1615c9005455SLois Curfman McInnes 161636851e7fSLois Curfman McInnes Level: intermediate 161736851e7fSLois Curfman McInnes 1618c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history 1619758f92a0SBarry Smith 162008405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory() 1621758f92a0SBarry Smith 1622c9005455SLois Curfman McInnes @*/ 162363dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt *its,PetscInt na,PetscTruth reset) 1624c9005455SLois Curfman McInnes { 16253a40ed3dSBarry Smith PetscFunctionBegin; 16264482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 16274482741eSBarry Smith if (na) PetscValidScalarPointer(a,2); 1628c9005455SLois Curfman McInnes snes->conv_hist = a; 1629758f92a0SBarry Smith snes->conv_hist_its = its; 1630758f92a0SBarry Smith snes->conv_hist_max = na; 1631758f92a0SBarry Smith snes->conv_hist_reset = reset; 1632758f92a0SBarry Smith PetscFunctionReturn(0); 1633758f92a0SBarry Smith } 1634758f92a0SBarry Smith 16354a2ae208SSatish Balay #undef __FUNCT__ 16364a2ae208SSatish Balay #define __FUNCT__ "SNESGetConvergenceHistory" 16370c4c9dddSBarry Smith /*@C 1638758f92a0SBarry Smith SNESGetConvergenceHistory - Gets the array used to hold the convergence history. 1639758f92a0SBarry Smith 1640758f92a0SBarry Smith Collective on SNES 1641758f92a0SBarry Smith 1642758f92a0SBarry Smith Input Parameter: 1643758f92a0SBarry Smith . snes - iterative context obtained from SNESCreate() 1644758f92a0SBarry Smith 1645758f92a0SBarry Smith Output Parameters: 1646758f92a0SBarry Smith . a - array to hold history 1647758f92a0SBarry Smith . its - integer array holds the number of linear iterations (or 1648758f92a0SBarry Smith negative if not converged) for each solve. 1649758f92a0SBarry Smith - na - size of a and its 1650758f92a0SBarry Smith 1651758f92a0SBarry Smith Notes: 1652758f92a0SBarry Smith The calling sequence for this routine in Fortran is 1653758f92a0SBarry Smith $ call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr) 1654758f92a0SBarry Smith 1655758f92a0SBarry Smith This routine is useful, e.g., when running a code for purposes 1656758f92a0SBarry Smith of accurate performance monitoring, when no I/O should be done 1657758f92a0SBarry Smith during the section of code that is being timed. 1658758f92a0SBarry Smith 1659758f92a0SBarry Smith Level: intermediate 1660758f92a0SBarry Smith 1661758f92a0SBarry Smith .keywords: SNES, get, convergence, history 1662758f92a0SBarry Smith 1663758f92a0SBarry Smith .seealso: SNESSetConvergencHistory() 1664758f92a0SBarry Smith 1665758f92a0SBarry Smith @*/ 166663dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na) 1667758f92a0SBarry Smith { 1668758f92a0SBarry Smith PetscFunctionBegin; 16694482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 1670758f92a0SBarry Smith if (a) *a = snes->conv_hist; 1671758f92a0SBarry Smith if (its) *its = snes->conv_hist_its; 1672758f92a0SBarry Smith if (na) *na = snes->conv_hist_len; 16733a40ed3dSBarry Smith PetscFunctionReturn(0); 1674c9005455SLois Curfman McInnes } 1675c9005455SLois Curfman McInnes 1676e74ef692SMatthew Knepley #undef __FUNCT__ 1677e74ef692SMatthew Knepley #define __FUNCT__ "SNESSetUpdate" 1678ac226902SBarry Smith /*@C 167976b2cf59SMatthew Knepley SNESSetUpdate - Sets the general-purpose update function called 16807e4bb74cSBarry Smith at the beginning o every iteration of the nonlinear solve. Specifically 16817e4bb74cSBarry Smith it is called just before the Jacobian is "evaluated". 168276b2cf59SMatthew Knepley 168376b2cf59SMatthew Knepley Collective on SNES 168476b2cf59SMatthew Knepley 168576b2cf59SMatthew Knepley Input Parameters: 168676b2cf59SMatthew Knepley . snes - The nonlinear solver context 168776b2cf59SMatthew Knepley . func - The function 168876b2cf59SMatthew Knepley 168976b2cf59SMatthew Knepley Calling sequence of func: 1690b5d30489SBarry Smith . func (SNES snes, PetscInt step); 169176b2cf59SMatthew Knepley 169276b2cf59SMatthew Knepley . step - The current step of the iteration 169376b2cf59SMatthew Knepley 169476b2cf59SMatthew Knepley Level: intermediate 169576b2cf59SMatthew Knepley 169676b2cf59SMatthew Knepley .keywords: SNES, update 1697b5d30489SBarry Smith 169876b2cf59SMatthew Knepley .seealso SNESDefaultUpdate(), SNESSetRhsBC(), SNESSetSolutionBC() 169976b2cf59SMatthew Knepley @*/ 170063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt)) 170176b2cf59SMatthew Knepley { 170276b2cf59SMatthew Knepley PetscFunctionBegin; 17034482741eSBarry Smith PetscValidHeaderSpecific(snes, SNES_COOKIE,1); 1704e7788613SBarry Smith snes->ops->update = func; 170576b2cf59SMatthew Knepley PetscFunctionReturn(0); 170676b2cf59SMatthew Knepley } 170776b2cf59SMatthew Knepley 1708e74ef692SMatthew Knepley #undef __FUNCT__ 1709e74ef692SMatthew Knepley #define __FUNCT__ "SNESDefaultUpdate" 171076b2cf59SMatthew Knepley /*@ 171176b2cf59SMatthew Knepley SNESDefaultUpdate - The default update function which does nothing. 171276b2cf59SMatthew Knepley 171376b2cf59SMatthew Knepley Not collective 171476b2cf59SMatthew Knepley 171576b2cf59SMatthew Knepley Input Parameters: 171676b2cf59SMatthew Knepley . snes - The nonlinear solver context 171776b2cf59SMatthew Knepley . step - The current step of the iteration 171876b2cf59SMatthew Knepley 1719205452f4SMatthew Knepley Level: intermediate 1720205452f4SMatthew Knepley 172176b2cf59SMatthew Knepley .keywords: SNES, update 172276b2cf59SMatthew Knepley .seealso SNESSetUpdate(), SNESDefaultRhsBC(), SNESDefaultSolutionBC() 172376b2cf59SMatthew Knepley @*/ 172463dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESDefaultUpdate(SNES snes, PetscInt step) 172576b2cf59SMatthew Knepley { 172676b2cf59SMatthew Knepley PetscFunctionBegin; 172776b2cf59SMatthew Knepley PetscFunctionReturn(0); 172876b2cf59SMatthew Knepley } 172976b2cf59SMatthew Knepley 17304a2ae208SSatish Balay #undef __FUNCT__ 17314a2ae208SSatish Balay #define __FUNCT__ "SNESScaleStep_Private" 17329b94acceSBarry Smith /* 17339b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 17349b94acceSBarry Smith positive parameter delta. 17359b94acceSBarry Smith 17369b94acceSBarry Smith Input Parameters: 1737c7afd0dbSLois Curfman McInnes + snes - the SNES context 17389b94acceSBarry Smith . y - approximate solution of linear system 17399b94acceSBarry Smith . fnorm - 2-norm of current function 1740c7afd0dbSLois Curfman McInnes - delta - trust region size 17419b94acceSBarry Smith 17429b94acceSBarry Smith Output Parameters: 1743c7afd0dbSLois Curfman McInnes + gpnorm - predicted function norm at the new point, assuming local 17449b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 17459b94acceSBarry Smith region, and exceeds zero otherwise. 1746c7afd0dbSLois Curfman McInnes - ynorm - 2-norm of the step 17479b94acceSBarry Smith 17489b94acceSBarry Smith Note: 17494b27c08aSLois Curfman McInnes For non-trust region methods such as SNESLS, the parameter delta 17509b94acceSBarry Smith is set to be the maximum allowable step size. 17519b94acceSBarry Smith 17529b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 17539b94acceSBarry Smith */ 1754dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm) 17559b94acceSBarry Smith { 1756064f8208SBarry Smith PetscReal nrm; 1757ea709b57SSatish Balay PetscScalar cnorm; 1758dfbe8321SBarry Smith PetscErrorCode ierr; 17593a40ed3dSBarry Smith 17603a40ed3dSBarry Smith PetscFunctionBegin; 17614482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 17624482741eSBarry Smith PetscValidHeaderSpecific(y,VEC_COOKIE,2); 1763c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,2); 1764184914b5SBarry Smith 1765064f8208SBarry Smith ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr); 1766064f8208SBarry Smith if (nrm > *delta) { 1767064f8208SBarry Smith nrm = *delta/nrm; 1768064f8208SBarry Smith *gpnorm = (1.0 - nrm)*(*fnorm); 1769064f8208SBarry Smith cnorm = nrm; 17702dcb1b2aSMatthew Knepley ierr = VecScale(y,cnorm);CHKERRQ(ierr); 17719b94acceSBarry Smith *ynorm = *delta; 17729b94acceSBarry Smith } else { 17739b94acceSBarry Smith *gpnorm = 0.0; 1774064f8208SBarry Smith *ynorm = nrm; 17759b94acceSBarry Smith } 17763a40ed3dSBarry Smith PetscFunctionReturn(0); 17779b94acceSBarry Smith } 17789b94acceSBarry Smith 17794a2ae208SSatish Balay #undef __FUNCT__ 17804a2ae208SSatish Balay #define __FUNCT__ "SNESSolve" 17816ce558aeSBarry Smith /*@C 1782f69a0ea3SMatthew Knepley SNESSolve - Solves a nonlinear system F(x) = b. 1783f69a0ea3SMatthew Knepley Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX(). 17849b94acceSBarry Smith 1785c7afd0dbSLois Curfman McInnes Collective on SNES 1786c7afd0dbSLois Curfman McInnes 1787b2002411SLois Curfman McInnes Input Parameters: 1788c7afd0dbSLois Curfman McInnes + snes - the SNES context 1789f69a0ea3SMatthew Knepley . b - the constant part of the equation, or PETSC_NULL to use zero. 179070e92668SMatthew Knepley - x - the solution vector, or PETSC_NULL if it was set with SNESSetSolution() 17919b94acceSBarry Smith 1792b2002411SLois Curfman McInnes Notes: 17938ddd3da0SLois Curfman McInnes The user should initialize the vector,x, with the initial guess 17948ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 17958ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 17968ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 17978ddd3da0SLois Curfman McInnes 179836851e7fSLois Curfman McInnes Level: beginner 179936851e7fSLois Curfman McInnes 18009b94acceSBarry Smith .keywords: SNES, nonlinear, solve 18019b94acceSBarry Smith 180270e92668SMatthew Knepley .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetRhs(), SNESSetSolution() 18039b94acceSBarry Smith @*/ 1804f69a0ea3SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESSolve(SNES snes,Vec b,Vec x) 18059b94acceSBarry Smith { 1806dfbe8321SBarry Smith PetscErrorCode ierr; 1807f1af5d2fSBarry Smith PetscTruth flg; 1808eabae89aSBarry Smith char filename[PETSC_MAX_PATH_LEN]; 1809eabae89aSBarry Smith PetscViewer viewer; 1810052efed2SBarry Smith 18113a40ed3dSBarry Smith PetscFunctionBegin; 18124482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 1813e7788613SBarry Smith if (!snes->ops->solve) SETERRQ(PETSC_ERR_ORDER,"SNESSetType() or SNESSetFromOptions() must be called before SNESSolve()"); 181474637425SBarry Smith 1815f69a0ea3SMatthew Knepley if (b) { 1816f69a0ea3SMatthew Knepley ierr = SNESSetRhs(snes, b); CHKERRQ(ierr); 18171096aae1SMatthew Knepley if (!snes->vec_func) { 18181096aae1SMatthew Knepley Vec r; 18191096aae1SMatthew Knepley 18201096aae1SMatthew Knepley ierr = VecDuplicate(b, &r); CHKERRQ(ierr); 18211096aae1SMatthew Knepley ierr = SNESSetFunction(snes, r, PETSC_NULL, PETSC_NULL); CHKERRQ(ierr); 18221096aae1SMatthew Knepley } 1823f69a0ea3SMatthew Knepley } 182470e92668SMatthew Knepley if (x) { 1825f69a0ea3SMatthew Knepley PetscValidHeaderSpecific(x,VEC_COOKIE,3); 1826f69a0ea3SMatthew Knepley PetscCheckSameComm(snes,1,x,3); 182770e92668SMatthew Knepley } else { 182870e92668SMatthew Knepley ierr = SNESGetSolution(snes, &x); CHKERRQ(ierr); 182970e92668SMatthew Knepley if (!x) { 183070e92668SMatthew Knepley ierr = VecDuplicate(snes->vec_func_always, &x); CHKERRQ(ierr); 183170e92668SMatthew Knepley } 183270e92668SMatthew Knepley } 183370e92668SMatthew Knepley snes->vec_sol = snes->vec_sol_always = x; 183470e92668SMatthew Knepley if (!snes->setupcalled) { 183570e92668SMatthew Knepley ierr = SNESSetUp(snes);CHKERRQ(ierr); 183670e92668SMatthew Knepley } 1837abc0a331SBarry Smith if (snes->conv_hist_reset) snes->conv_hist_len = 0; 1838d5ba7fb7SMatthew Knepley ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 183950ffb88aSMatthew Knepley snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0; 1840d5e45103SBarry Smith 1841e7788613SBarry Smith ierr = PetscExceptionTry1((*(snes)->ops->solve)(snes),PETSC_ERR_ARG_DOMAIN); 1842d5e45103SBarry Smith if (PetscExceptionValue(ierr)) { 184338f152feSBarry Smith /* this means that a caller above me has also tryed this exception so I don't handle it here, pass it up */ 184419717074SBarry Smith PetscErrorCode pierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(pierr); 1845d5e45103SBarry Smith } else if (PetscExceptionCaught(ierr,PETSC_ERR_ARG_DOMAIN)) { 1846d5e45103SBarry Smith /* translate exception into SNES not converged reason */ 1847d5e45103SBarry Smith snes->reason = SNES_DIVERGED_FUNCTION_DOMAIN; 184838f152feSBarry Smith ierr = 0; 184938f152feSBarry Smith } 185038f152feSBarry Smith CHKERRQ(ierr); 1851d5e45103SBarry Smith 1852d5ba7fb7SMatthew Knepley ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 1853eabae89aSBarry Smith ierr = PetscOptionsGetString(snes->prefix,"-snes_view",filename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr); 1854eabae89aSBarry Smith if (flg && !PetscPreLoadingOn) { 1855eabae89aSBarry Smith ierr = PetscViewerASCIIOpen(snes->comm,filename,&viewer);CHKERRQ(ierr); 1856eabae89aSBarry Smith ierr = SNESView(snes,viewer);CHKERRQ(ierr); 1857eabae89aSBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 1858eabae89aSBarry Smith } 1859eabae89aSBarry Smith 1860da9b6338SBarry Smith ierr = PetscOptionsHasName(snes->prefix,"-snes_test_local_min",&flg);CHKERRQ(ierr); 1861da9b6338SBarry Smith if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); } 18625968eb51SBarry Smith if (snes->printreason) { 18635968eb51SBarry Smith if (snes->reason > 0) { 18649dcbbd2bSBarry Smith ierr = PetscPrintf(snes->comm,"Nonlinear solve converged due to %s\n",SNESConvergedReasons[snes->reason]);CHKERRQ(ierr); 18655968eb51SBarry Smith } else { 18669dcbbd2bSBarry Smith ierr = PetscPrintf(snes->comm,"Nonlinear solve did not converge due to %s\n",SNESConvergedReasons[snes->reason]);CHKERRQ(ierr); 18675968eb51SBarry Smith } 18685968eb51SBarry Smith } 18695968eb51SBarry Smith 18703a40ed3dSBarry Smith PetscFunctionReturn(0); 18719b94acceSBarry Smith } 18729b94acceSBarry Smith 18739b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 18749b94acceSBarry Smith 18754a2ae208SSatish Balay #undef __FUNCT__ 18764a2ae208SSatish Balay #define __FUNCT__ "SNESSetType" 187782bf6240SBarry Smith /*@C 18784b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 18799b94acceSBarry Smith 1880fee21e36SBarry Smith Collective on SNES 1881fee21e36SBarry Smith 1882c7afd0dbSLois Curfman McInnes Input Parameters: 1883c7afd0dbSLois Curfman McInnes + snes - the SNES context 1884454a90a3SBarry Smith - type - a known method 1885c7afd0dbSLois Curfman McInnes 1886c7afd0dbSLois Curfman McInnes Options Database Key: 1887454a90a3SBarry Smith . -snes_type <type> - Sets the method; use -help for a list 1888c7afd0dbSLois Curfman McInnes of available methods (for instance, ls or tr) 1889ae12b187SLois Curfman McInnes 18909b94acceSBarry Smith Notes: 1891e090d566SSatish Balay See "petsc/include/petscsnes.h" for available methods (for instance) 18924b27c08aSLois Curfman McInnes + SNESLS - Newton's method with line search 1893c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 18944b27c08aSLois Curfman McInnes . SNESTR - Newton's method with trust region 1895c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 18969b94acceSBarry Smith 1897ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 1898ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 1899ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 1900ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 1901ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 1902ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 1903ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 1904ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 1905ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 1906b0a32e0cSBarry Smith appropriate method. 190736851e7fSLois Curfman McInnes 190836851e7fSLois Curfman McInnes Level: intermediate 1909a703fe33SLois Curfman McInnes 1910454a90a3SBarry Smith .keywords: SNES, set, type 1911435da068SBarry Smith 1912435da068SBarry Smith .seealso: SNESType, SNESCreate() 1913435da068SBarry Smith 19149b94acceSBarry Smith @*/ 1915e060cb09SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESSetType(SNES snes,SNESType type) 19169b94acceSBarry Smith { 1917dfbe8321SBarry Smith PetscErrorCode ierr,(*r)(SNES); 19186831982aSBarry Smith PetscTruth match; 19193a40ed3dSBarry Smith 19203a40ed3dSBarry Smith PetscFunctionBegin; 19214482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 19224482741eSBarry Smith PetscValidCharPointer(type,2); 192382bf6240SBarry Smith 19246831982aSBarry Smith ierr = PetscTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr); 19250f5bd95cSBarry Smith if (match) PetscFunctionReturn(0); 192682bf6240SBarry Smith 192782bf6240SBarry Smith if (snes->setupcalled) { 19284dc4c822SBarry Smith snes->setupcalled = PETSC_FALSE; 1929e7788613SBarry Smith ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr); 193082bf6240SBarry Smith snes->data = 0; 193182bf6240SBarry Smith } 19327d1a2b2bSBarry Smith 19339b94acceSBarry Smith /* Get the function pointers for the iterative method requested */ 193482bf6240SBarry Smith if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL);CHKERRQ(ierr);} 1935b9617806SBarry Smith ierr = PetscFListFind(snes->comm,SNESList,type,(void (**)(void)) &r);CHKERRQ(ierr); 1936958c9bccSBarry Smith if (!r) SETERRQ1(PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type); 193705b42c5fSBarry Smith ierr = PetscFree(snes->data);CHKERRQ(ierr); 193882bf6240SBarry Smith snes->data = 0; 19393a40ed3dSBarry Smith ierr = (*r)(snes);CHKERRQ(ierr); 1940454a90a3SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr); 19413a40ed3dSBarry Smith PetscFunctionReturn(0); 19429b94acceSBarry Smith } 19439b94acceSBarry Smith 1944a847f771SSatish Balay 19459b94acceSBarry Smith /* --------------------------------------------------------------------- */ 19464a2ae208SSatish Balay #undef __FUNCT__ 19474a2ae208SSatish Balay #define __FUNCT__ "SNESRegisterDestroy" 194852baeb72SSatish Balay /*@ 19499b94acceSBarry Smith SNESRegisterDestroy - Frees the list of nonlinear solvers that were 1950f1af5d2fSBarry Smith registered by SNESRegisterDynamic(). 19519b94acceSBarry Smith 1952fee21e36SBarry Smith Not Collective 1953fee21e36SBarry Smith 195436851e7fSLois Curfman McInnes Level: advanced 195536851e7fSLois Curfman McInnes 19569b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy 19579b94acceSBarry Smith 19589b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll() 19599b94acceSBarry Smith @*/ 196063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESRegisterDestroy(void) 19619b94acceSBarry Smith { 1962dfbe8321SBarry Smith PetscErrorCode ierr; 196382bf6240SBarry Smith 19643a40ed3dSBarry Smith PetscFunctionBegin; 196582bf6240SBarry Smith if (SNESList) { 1966b0a32e0cSBarry Smith ierr = PetscFListDestroy(&SNESList);CHKERRQ(ierr); 196782bf6240SBarry Smith SNESList = 0; 19689b94acceSBarry Smith } 19694c49b128SBarry Smith SNESRegisterAllCalled = PETSC_FALSE; 19703a40ed3dSBarry Smith PetscFunctionReturn(0); 19719b94acceSBarry Smith } 19729b94acceSBarry Smith 19734a2ae208SSatish Balay #undef __FUNCT__ 19744a2ae208SSatish Balay #define __FUNCT__ "SNESGetType" 19759b94acceSBarry Smith /*@C 19769a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 19779b94acceSBarry Smith 1978c7afd0dbSLois Curfman McInnes Not Collective 1979c7afd0dbSLois Curfman McInnes 19809b94acceSBarry Smith Input Parameter: 19814b0e389bSBarry Smith . snes - nonlinear solver context 19829b94acceSBarry Smith 19839b94acceSBarry Smith Output Parameter: 19843a7fca6bSBarry Smith . type - SNES method (a character string) 19859b94acceSBarry Smith 198636851e7fSLois Curfman McInnes Level: intermediate 198736851e7fSLois Curfman McInnes 1988454a90a3SBarry Smith .keywords: SNES, nonlinear, get, type, name 19899b94acceSBarry Smith @*/ 199063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetType(SNES snes,SNESType *type) 19919b94acceSBarry Smith { 19923a40ed3dSBarry Smith PetscFunctionBegin; 19934482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 19944482741eSBarry Smith PetscValidPointer(type,2); 1995454a90a3SBarry Smith *type = snes->type_name; 19963a40ed3dSBarry Smith PetscFunctionReturn(0); 19979b94acceSBarry Smith } 19989b94acceSBarry Smith 19994a2ae208SSatish Balay #undef __FUNCT__ 20004a2ae208SSatish Balay #define __FUNCT__ "SNESGetSolution" 200152baeb72SSatish Balay /*@ 20029b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 20039b94acceSBarry Smith stored. 20049b94acceSBarry Smith 2005c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2006c7afd0dbSLois Curfman McInnes 20079b94acceSBarry Smith Input Parameter: 20089b94acceSBarry Smith . snes - the SNES context 20099b94acceSBarry Smith 20109b94acceSBarry Smith Output Parameter: 20119b94acceSBarry Smith . x - the solution 20129b94acceSBarry Smith 201370e92668SMatthew Knepley Level: intermediate 201436851e7fSLois Curfman McInnes 20159b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 20169b94acceSBarry Smith 201770e92668SMatthew Knepley .seealso: SNESSetSolution(), SNESGetFunction(), SNESGetSolutionUpdate() 20189b94acceSBarry Smith @*/ 201963dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetSolution(SNES snes,Vec *x) 20209b94acceSBarry Smith { 20213a40ed3dSBarry Smith PetscFunctionBegin; 20224482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 20234482741eSBarry Smith PetscValidPointer(x,2); 20249b94acceSBarry Smith *x = snes->vec_sol_always; 20253a40ed3dSBarry Smith PetscFunctionReturn(0); 20269b94acceSBarry Smith } 20279b94acceSBarry Smith 20284a2ae208SSatish Balay #undef __FUNCT__ 202970e92668SMatthew Knepley #define __FUNCT__ "SNESSetSolution" 20307e4bb74cSBarry Smith /*@ 203170e92668SMatthew Knepley SNESSetSolution - Sets the vector where the approximate solution is stored. 203270e92668SMatthew Knepley 203370e92668SMatthew Knepley Not Collective, but Vec is parallel if SNES is parallel 203470e92668SMatthew Knepley 203570e92668SMatthew Knepley Input Parameters: 203670e92668SMatthew Knepley + snes - the SNES context 203770e92668SMatthew Knepley - x - the solution 203870e92668SMatthew Knepley 203970e92668SMatthew Knepley Output Parameter: 204070e92668SMatthew Knepley 204170e92668SMatthew Knepley Level: intermediate 204270e92668SMatthew Knepley 204342219521SBarry Smith Notes: this is not normally used, rather one simply calls SNESSolve() with 204442219521SBarry Smith the appropriate solution vector. 204542219521SBarry Smith 204670e92668SMatthew Knepley .keywords: SNES, nonlinear, set, solution 204770e92668SMatthew Knepley 204870e92668SMatthew Knepley .seealso: SNESGetSolution(), SNESGetFunction(), SNESGetSolutionUpdate() 204970e92668SMatthew Knepley @*/ 205070e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESSetSolution(SNES snes,Vec x) 205170e92668SMatthew Knepley { 205270e92668SMatthew Knepley PetscFunctionBegin; 205370e92668SMatthew Knepley PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 205470e92668SMatthew Knepley PetscValidHeaderSpecific(x,VEC_COOKIE,2); 205570e92668SMatthew Knepley PetscCheckSameComm(snes,1,x,2); 205670e92668SMatthew Knepley snes->vec_sol_always = x; 205770e92668SMatthew Knepley PetscFunctionReturn(0); 205870e92668SMatthew Knepley } 205970e92668SMatthew Knepley 206070e92668SMatthew Knepley #undef __FUNCT__ 20614a2ae208SSatish Balay #define __FUNCT__ "SNESGetSolutionUpdate" 206252baeb72SSatish Balay /*@ 20639b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 20649b94acceSBarry Smith stored. 20659b94acceSBarry Smith 2066c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2067c7afd0dbSLois Curfman McInnes 20689b94acceSBarry Smith Input Parameter: 20699b94acceSBarry Smith . snes - the SNES context 20709b94acceSBarry Smith 20719b94acceSBarry Smith Output Parameter: 20729b94acceSBarry Smith . x - the solution update 20739b94acceSBarry Smith 207436851e7fSLois Curfman McInnes Level: advanced 207536851e7fSLois Curfman McInnes 20769b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 20779b94acceSBarry Smith 20789b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction 20799b94acceSBarry Smith @*/ 208063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESGetSolutionUpdate(SNES snes,Vec *x) 20819b94acceSBarry Smith { 20823a40ed3dSBarry Smith PetscFunctionBegin; 20834482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 20844482741eSBarry Smith PetscValidPointer(x,2); 20859b94acceSBarry Smith *x = snes->vec_sol_update_always; 20863a40ed3dSBarry Smith PetscFunctionReturn(0); 20879b94acceSBarry Smith } 20889b94acceSBarry Smith 20894a2ae208SSatish Balay #undef __FUNCT__ 20904a2ae208SSatish Balay #define __FUNCT__ "SNESGetFunction" 20919b94acceSBarry Smith /*@C 20923638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 20939b94acceSBarry Smith 2094c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2095c7afd0dbSLois Curfman McInnes 20969b94acceSBarry Smith Input Parameter: 20979b94acceSBarry Smith . snes - the SNES context 20989b94acceSBarry Smith 20999b94acceSBarry Smith Output Parameter: 21007bf4e008SBarry Smith + r - the function (or PETSC_NULL) 210170e92668SMatthew Knepley . func - the function (or PETSC_NULL) 210270e92668SMatthew Knepley - ctx - the function context (or PETSC_NULL) 21039b94acceSBarry Smith 210436851e7fSLois Curfman McInnes Level: advanced 210536851e7fSLois Curfman McInnes 2106a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 21079b94acceSBarry Smith 21084b27c08aSLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution() 21099b94acceSBarry Smith @*/ 211070e92668SMatthew Knepley PetscErrorCode PETSCSNES_DLLEXPORT SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**func)(SNES,Vec,Vec,void*),void **ctx) 21119b94acceSBarry Smith { 21123a40ed3dSBarry Smith PetscFunctionBegin; 21134482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 21147bf4e008SBarry Smith if (r) *r = snes->vec_func_always; 2115e7788613SBarry Smith if (func) *func = snes->ops->computefunction; 211670e92668SMatthew Knepley if (ctx) *ctx = snes->funP; 21173a40ed3dSBarry Smith PetscFunctionReturn(0); 21189b94acceSBarry Smith } 21199b94acceSBarry Smith 21204a2ae208SSatish Balay #undef __FUNCT__ 21214a2ae208SSatish Balay #define __FUNCT__ "SNESSetOptionsPrefix" 21223c7409f5SSatish Balay /*@C 21233c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 2124d850072dSLois Curfman McInnes SNES options in the database. 21253c7409f5SSatish Balay 2126fee21e36SBarry Smith Collective on SNES 2127fee21e36SBarry Smith 2128c7afd0dbSLois Curfman McInnes Input Parameter: 2129c7afd0dbSLois Curfman McInnes + snes - the SNES context 2130c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2131c7afd0dbSLois Curfman McInnes 2132d850072dSLois Curfman McInnes Notes: 2133a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2134c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2135d850072dSLois Curfman McInnes 213636851e7fSLois Curfman McInnes Level: advanced 213736851e7fSLois Curfman McInnes 21383c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 2139a86d99e1SLois Curfman McInnes 2140a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 21413c7409f5SSatish Balay @*/ 214263dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESSetOptionsPrefix(SNES snes,const char prefix[]) 21433c7409f5SSatish Balay { 2144dfbe8321SBarry Smith PetscErrorCode ierr; 21453c7409f5SSatish Balay 21463a40ed3dSBarry Smith PetscFunctionBegin; 21474482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 2148639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 214994b7f48cSBarry Smith ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 21503a40ed3dSBarry Smith PetscFunctionReturn(0); 21513c7409f5SSatish Balay } 21523c7409f5SSatish Balay 21534a2ae208SSatish Balay #undef __FUNCT__ 21544a2ae208SSatish Balay #define __FUNCT__ "SNESAppendOptionsPrefix" 21553c7409f5SSatish Balay /*@C 2156f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 2157d850072dSLois Curfman McInnes SNES options in the database. 21583c7409f5SSatish Balay 2159fee21e36SBarry Smith Collective on SNES 2160fee21e36SBarry Smith 2161c7afd0dbSLois Curfman McInnes Input Parameters: 2162c7afd0dbSLois Curfman McInnes + snes - the SNES context 2163c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2164c7afd0dbSLois Curfman McInnes 2165d850072dSLois Curfman McInnes Notes: 2166a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2167c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2168d850072dSLois Curfman McInnes 216936851e7fSLois Curfman McInnes Level: advanced 217036851e7fSLois Curfman McInnes 21713c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 2172a86d99e1SLois Curfman McInnes 2173a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 21743c7409f5SSatish Balay @*/ 217563dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESAppendOptionsPrefix(SNES snes,const char prefix[]) 21763c7409f5SSatish Balay { 2177dfbe8321SBarry Smith PetscErrorCode ierr; 21783c7409f5SSatish Balay 21793a40ed3dSBarry Smith PetscFunctionBegin; 21804482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 2181639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 218294b7f48cSBarry Smith ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 21833a40ed3dSBarry Smith PetscFunctionReturn(0); 21843c7409f5SSatish Balay } 21853c7409f5SSatish Balay 21864a2ae208SSatish Balay #undef __FUNCT__ 21874a2ae208SSatish Balay #define __FUNCT__ "SNESGetOptionsPrefix" 21889ab63eb5SSatish Balay /*@C 21893c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 21903c7409f5SSatish Balay SNES options in the database. 21913c7409f5SSatish Balay 2192c7afd0dbSLois Curfman McInnes Not Collective 2193c7afd0dbSLois Curfman McInnes 21943c7409f5SSatish Balay Input Parameter: 21953c7409f5SSatish Balay . snes - the SNES context 21963c7409f5SSatish Balay 21973c7409f5SSatish Balay Output Parameter: 21983c7409f5SSatish Balay . prefix - pointer to the prefix string used 21993c7409f5SSatish Balay 22009ab63eb5SSatish Balay Notes: On the fortran side, the user should pass in a string 'prifix' of 22019ab63eb5SSatish Balay sufficient length to hold the prefix. 22029ab63eb5SSatish Balay 220336851e7fSLois Curfman McInnes Level: advanced 220436851e7fSLois Curfman McInnes 22053c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 2206a86d99e1SLois Curfman McInnes 2207a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 22083c7409f5SSatish Balay @*/ 2209e060cb09SBarry Smith PetscErrorCode PETSCSNES_DLLEXPORT SNESGetOptionsPrefix(SNES snes,const char *prefix[]) 22103c7409f5SSatish Balay { 2211dfbe8321SBarry Smith PetscErrorCode ierr; 22123c7409f5SSatish Balay 22133a40ed3dSBarry Smith PetscFunctionBegin; 22144482741eSBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE,1); 2215639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 22163a40ed3dSBarry Smith PetscFunctionReturn(0); 22173c7409f5SSatish Balay } 22183c7409f5SSatish Balay 2219b2002411SLois Curfman McInnes 22204a2ae208SSatish Balay #undef __FUNCT__ 22214a2ae208SSatish Balay #define __FUNCT__ "SNESRegister" 22223cea93caSBarry Smith /*@C 22233cea93caSBarry Smith SNESRegister - See SNESRegisterDynamic() 22243cea93caSBarry Smith 22257f6c08e0SMatthew Knepley Level: advanced 22263cea93caSBarry Smith @*/ 222763dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESRegister(const char sname[],const char path[],const char name[],PetscErrorCode (*function)(SNES)) 2228b2002411SLois Curfman McInnes { 2229e2d1d2b7SBarry Smith char fullname[PETSC_MAX_PATH_LEN]; 2230dfbe8321SBarry Smith PetscErrorCode ierr; 2231b2002411SLois Curfman McInnes 2232b2002411SLois Curfman McInnes PetscFunctionBegin; 2233b0a32e0cSBarry Smith ierr = PetscFListConcat(path,name,fullname);CHKERRQ(ierr); 2234c134de8dSSatish Balay ierr = PetscFListAdd(&SNESList,sname,fullname,(void (*)(void))function);CHKERRQ(ierr); 2235b2002411SLois Curfman McInnes PetscFunctionReturn(0); 2236b2002411SLois Curfman McInnes } 2237da9b6338SBarry Smith 2238da9b6338SBarry Smith #undef __FUNCT__ 2239da9b6338SBarry Smith #define __FUNCT__ "SNESTestLocalMin" 224063dd3a1aSKris Buschelman PetscErrorCode PETSCSNES_DLLEXPORT SNESTestLocalMin(SNES snes) 2241da9b6338SBarry Smith { 2242dfbe8321SBarry Smith PetscErrorCode ierr; 224377431f27SBarry Smith PetscInt N,i,j; 2244da9b6338SBarry Smith Vec u,uh,fh; 2245da9b6338SBarry Smith PetscScalar value; 2246da9b6338SBarry Smith PetscReal norm; 2247da9b6338SBarry Smith 2248da9b6338SBarry Smith PetscFunctionBegin; 2249da9b6338SBarry Smith ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr); 2250da9b6338SBarry Smith ierr = VecDuplicate(u,&uh);CHKERRQ(ierr); 2251da9b6338SBarry Smith ierr = VecDuplicate(u,&fh);CHKERRQ(ierr); 2252da9b6338SBarry Smith 2253da9b6338SBarry Smith /* currently only works for sequential */ 2254da9b6338SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD,"Testing FormFunction() for local min\n"); 2255da9b6338SBarry Smith ierr = VecGetSize(u,&N);CHKERRQ(ierr); 2256da9b6338SBarry Smith for (i=0; i<N; i++) { 2257da9b6338SBarry Smith ierr = VecCopy(u,uh);CHKERRQ(ierr); 225877431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD,"i = %D\n",i);CHKERRQ(ierr); 2259da9b6338SBarry Smith for (j=-10; j<11; j++) { 2260ccae9161SBarry Smith value = PetscSign(j)*exp(PetscAbs(j)-10.0); 2261da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 22623ab0aad5SBarry Smith ierr = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr); 2263da9b6338SBarry Smith ierr = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr); 226477431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD," j norm %D %18.16e\n",j,norm);CHKERRQ(ierr); 2265da9b6338SBarry Smith value = -value; 2266da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 2267da9b6338SBarry Smith } 2268da9b6338SBarry Smith } 2269da9b6338SBarry Smith ierr = VecDestroy(uh);CHKERRQ(ierr); 2270da9b6338SBarry Smith ierr = VecDestroy(fh);CHKERRQ(ierr); 2271da9b6338SBarry Smith PetscFunctionReturn(0); 2272da9b6338SBarry Smith } 2273