1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER 2*0c4c9dddSBarry Smith static char vcid[] = "$Id: snes.c,v 1.172 1999/02/27 04:48:56 bsmith Exp bsmith $"; 39b94acceSBarry Smith #endif 49b94acceSBarry Smith 570f55243SBarry Smith #include "src/snes/snesimpl.h" /*I "snes.h" I*/ 6f5eb4b81SSatish Balay #include "src/sys/nreg.h" 79b94acceSBarry Smith 884cb2905SBarry Smith int SNESRegisterAllCalled = 0; 9488ecbafSBarry Smith FList SNESList = 0; 1082bf6240SBarry Smith 115615d1e5SSatish Balay #undef __FUNC__ 12d4bb536fSBarry Smith #define __FUNC__ "SNESView" 139b94acceSBarry Smith /*@ 149b94acceSBarry Smith SNESView - Prints the SNES data structure. 159b94acceSBarry Smith 16fee21e36SBarry Smith Collective on SNES, unless Viewer is VIEWER_STDOUT_SELF 17fee21e36SBarry Smith 18c7afd0dbSLois Curfman McInnes Input Parameters: 19c7afd0dbSLois Curfman McInnes + SNES - the SNES context 20c7afd0dbSLois Curfman McInnes - viewer - visualization context 21c7afd0dbSLois Curfman McInnes 229b94acceSBarry Smith Options Database Key: 23c8a8ba5cSLois Curfman McInnes . -snes_view - Calls SNESView() at end of SNESSolve() 249b94acceSBarry Smith 259b94acceSBarry Smith Notes: 269b94acceSBarry Smith The available visualization contexts include 27c7afd0dbSLois Curfman McInnes + VIEWER_STDOUT_SELF - standard output (default) 28c7afd0dbSLois Curfman McInnes - VIEWER_STDOUT_WORLD - synchronized standard 29c8a8ba5cSLois Curfman McInnes output where only the first processor opens 30c8a8ba5cSLois Curfman McInnes the file. All other processors send their 31c8a8ba5cSLois Curfman McInnes data to the first processor to print. 329b94acceSBarry Smith 333e081fefSLois Curfman McInnes The user can open an alternative visualization context with 3477ed5343SBarry Smith ViewerASCIIOpen() - output to a specified file. 359b94acceSBarry Smith 3636851e7fSLois Curfman McInnes Level: beginner 3736851e7fSLois Curfman McInnes 389b94acceSBarry Smith .keywords: SNES, view 399b94acceSBarry Smith 4077ed5343SBarry Smith .seealso: ViewerASCIIOpen() 419b94acceSBarry Smith @*/ 429b94acceSBarry Smith int SNESView(SNES snes,Viewer viewer) 439b94acceSBarry Smith { 449b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 459b94acceSBarry Smith int ierr; 469b94acceSBarry Smith SLES sles; 479b94acceSBarry Smith char *method; 4819bcc07fSBarry Smith ViewerType vtype; 499b94acceSBarry Smith 503a40ed3dSBarry Smith PetscFunctionBegin; 5174679c65SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 5274679c65SBarry Smith if (viewer) {PetscValidHeader(viewer);} 5374679c65SBarry Smith else { viewer = VIEWER_STDOUT_SELF; } 5474679c65SBarry Smith 5519bcc07fSBarry Smith ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr); 563f1db9ecSBarry Smith if (PetscTypeCompare(vtype,ASCII_VIEWER)) { 570ef38995SBarry Smith ViewerASCIIPrintf(viewer,"SNES Object:\n"); 5882bf6240SBarry Smith SNESGetType(snes,&method); 590ef38995SBarry Smith ViewerASCIIPrintf(viewer," method: %s\n",method); 600ef38995SBarry Smith if (snes->view) { 610ef38995SBarry Smith ierr = ViewerASCIIPushTab(viewer);CHKERRQ(ierr); 620ef38995SBarry Smith ierr = (*snes->view)(snes,viewer);CHKERRQ(ierr); 630ef38995SBarry Smith ierr = ViewerASCIIPopTab(viewer);CHKERRQ(ierr); 640ef38995SBarry Smith } 650ef38995SBarry Smith ViewerASCIIPrintf(viewer," maximum iterations=%d, maximum function evaluations=%d\n",snes->max_its,snes->max_funcs); 660ef38995SBarry Smith ViewerASCIIPrintf(viewer," tolerances: relative=%g, absolute=%g, truncation=%g, solution=%g\n", 679b94acceSBarry Smith snes->rtol, snes->atol, snes->trunctol, snes->xtol); 680ef38995SBarry Smith ViewerASCIIPrintf(viewer," total number of linear solver iterations=%d\n",snes->linear_its); 690ef38995SBarry Smith ViewerASCIIPrintf(viewer," total number of function evaluations=%d\n",snes->nfuncs); 700ef38995SBarry Smith if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 710ef38995SBarry Smith ViewerASCIIPrintf(viewer," min function tolerance=%g\n",snes->fmin); 720ef38995SBarry Smith } 739b94acceSBarry Smith if (snes->ksp_ewconv) { 749b94acceSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 759b94acceSBarry Smith if (kctx) { 760ef38995SBarry Smith ViewerASCIIPrintf(viewer," Eisenstat-Walker computation of KSP relative tolerance (version %d)\n",kctx->version); 770ef38995SBarry Smith ViewerASCIIPrintf(viewer," rtol_0=%g, rtol_max=%g, threshold=%g\n",kctx->rtol_0,kctx->rtol_max,kctx->threshold); 780ef38995SBarry Smith ViewerASCIIPrintf(viewer," gamma=%g, alpha=%g, alpha2=%g\n",kctx->gamma,kctx->alpha,kctx->alpha2); 799b94acceSBarry Smith } 809b94acceSBarry Smith } 813f1db9ecSBarry Smith } else if (PetscTypeCompare(vtype,STRING_VIEWER)) { 820ef38995SBarry Smith ierr = SNESGetType(snes,&method);CHKERRQ(ierr); 830ef38995SBarry Smith ierr = ViewerStringSPrintf(viewer," %-3.3s",method);CHKERRQ(ierr); 8419bcc07fSBarry Smith } 8577ed5343SBarry Smith ierr = SNESGetSLES(snes,&sles);CHKERRQ(ierr); 860ef38995SBarry Smith ierr = ViewerASCIIPushTab(viewer);CHKERRQ(ierr); 879b94acceSBarry Smith ierr = SLESView(sles,viewer); CHKERRQ(ierr); 880ef38995SBarry Smith ierr = ViewerASCIIPopTab(viewer);CHKERRQ(ierr); 893a40ed3dSBarry Smith PetscFunctionReturn(0); 909b94acceSBarry Smith } 919b94acceSBarry Smith 92639f9d9dSBarry Smith /* 93639f9d9dSBarry Smith We retain a list of functions that also take SNES command 94639f9d9dSBarry Smith line options. These are called at the end SNESSetFromOptions() 95639f9d9dSBarry Smith */ 96639f9d9dSBarry Smith #define MAXSETFROMOPTIONS 5 97639f9d9dSBarry Smith static int numberofsetfromoptions; 98639f9d9dSBarry Smith static int (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES); 99639f9d9dSBarry Smith 1005615d1e5SSatish Balay #undef __FUNC__ 101d4bb536fSBarry Smith #define __FUNC__ "SNESAddOptionsChecker" 102639f9d9dSBarry Smith /*@ 103639f9d9dSBarry Smith SNESAddOptionsChecker - Adds an additional function to check for SNES options. 104639f9d9dSBarry Smith 105c7afd0dbSLois Curfman McInnes Not Collective 106c7afd0dbSLois Curfman McInnes 107639f9d9dSBarry Smith Input Parameter: 108639f9d9dSBarry Smith . snescheck - function that checks for options 109639f9d9dSBarry Smith 11036851e7fSLois Curfman McInnes Level: developer 11136851e7fSLois Curfman McInnes 112639f9d9dSBarry Smith .seealso: SNESSetFromOptions() 113639f9d9dSBarry Smith @*/ 114639f9d9dSBarry Smith int SNESAddOptionsChecker(int (*snescheck)(SNES) ) 115639f9d9dSBarry Smith { 1163a40ed3dSBarry Smith PetscFunctionBegin; 117639f9d9dSBarry Smith if (numberofsetfromoptions >= MAXSETFROMOPTIONS) { 118a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Too many options checkers, only 5 allowed"); 119639f9d9dSBarry Smith } 120639f9d9dSBarry Smith 121639f9d9dSBarry Smith othersetfromoptions[numberofsetfromoptions++] = snescheck; 1223a40ed3dSBarry Smith PetscFunctionReturn(0); 123639f9d9dSBarry Smith } 124639f9d9dSBarry Smith 1255615d1e5SSatish Balay #undef __FUNC__ 1265615d1e5SSatish Balay #define __FUNC__ "SNESSetFromOptions" 1279b94acceSBarry Smith /*@ 128682d7d0cSBarry Smith SNESSetFromOptions - Sets various SNES and SLES parameters from user options. 1299b94acceSBarry Smith 130c7afd0dbSLois Curfman McInnes Collective on SNES 131c7afd0dbSLois Curfman McInnes 1329b94acceSBarry Smith Input Parameter: 1339b94acceSBarry Smith . snes - the SNES context 1349b94acceSBarry Smith 13536851e7fSLois Curfman McInnes Options Database Keys: 136b39c3a46SLois Curfman McInnes + -snes_type <type> - SNES_EQ_LS, SNES_EQ_TR, SNES_UM_TR, SNES_UM_LS etc 13782738288SBarry Smith . -snes_stol - convergence tolerance in terms of the norm 13882738288SBarry Smith of the change in the solution between steps 139b39c3a46SLois Curfman McInnes . -snes_atol <atol> - absolute tolerance of residual norm 140b39c3a46SLois Curfman McInnes . -snes_rtol <rtol> - relative decrease in tolerance norm from initial 141b39c3a46SLois Curfman McInnes . -snes_max_it <max_it> - maximum number of iterations 142b39c3a46SLois Curfman McInnes . -snes_max_funcs <max_funcs> - maximum number of function evaluations 143b39c3a46SLois Curfman McInnes . -snes_trtol <trtol> - trust region tolerance 14482738288SBarry Smith . -snes_no_convergence_test - skip convergence test in nonlinear or minimization 14582738288SBarry Smith solver; hence iterations will continue until max_it 14682738288SBarry Smith or some other criteria is reached. Saves expense 14782738288SBarry Smith of convergence test 14882738288SBarry Smith . -snes_monitor - prints residual norm at each iteration 14982738288SBarry Smith . -snes_xmonitor - plots residual norm at each iteration 150e24b481bSBarry Smith . -snes_fd - use finite differences to compute Jacobian; very slow, only for testing 15136851e7fSLois Curfman McInnes - -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration 15282738288SBarry Smith 15382738288SBarry Smith Options Database for Eisenstat-Walker method: 15482738288SBarry Smith + -snes_ksp_eq_conv - use Eisenstat-Walker method for determining linear system convergence 15582738288SBarry Smith . -snes_ksp_eq_version ver - version of Eisenstat-Walker method 15636851e7fSLois Curfman McInnes . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 15736851e7fSLois Curfman McInnes . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 15836851e7fSLois Curfman McInnes . -snes_ksp_ew_gamma <gamma> - Sets gamma 15936851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha <alpha> - Sets alpha 16036851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 16136851e7fSLois Curfman McInnes - -snes_ksp_ew_threshold <threshold> - Sets threshold 16282738288SBarry Smith 16311ca99fdSLois Curfman McInnes Notes: 16411ca99fdSLois Curfman McInnes To see all options, run your program with the -help option or consult 16511ca99fdSLois Curfman McInnes the users manual. 16683e2fdc7SBarry Smith 16736851e7fSLois Curfman McInnes Level: beginner 16836851e7fSLois Curfman McInnes 1699b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 1709b94acceSBarry Smith 171a86d99e1SLois Curfman McInnes .seealso: SNESPrintHelp(), SNESSetOptionsPrefix() 1729b94acceSBarry Smith @*/ 1739b94acceSBarry Smith int SNESSetFromOptions(SNES snes) 1749b94acceSBarry Smith { 17582bf6240SBarry Smith char method[256]; 1769b94acceSBarry Smith double tmp; 1779b94acceSBarry Smith SLES sles; 17881f57ec7SBarry Smith int ierr, flg,i,loc[4],nmax = 4; 1793c7409f5SSatish Balay int version = PETSC_DEFAULT; 1809b94acceSBarry Smith double rtol_0 = PETSC_DEFAULT; 1819b94acceSBarry Smith double rtol_max = PETSC_DEFAULT; 1829b94acceSBarry Smith double gamma2 = PETSC_DEFAULT; 1839b94acceSBarry Smith double alpha = PETSC_DEFAULT; 1849b94acceSBarry Smith double alpha2 = PETSC_DEFAULT; 1859b94acceSBarry Smith double threshold = PETSC_DEFAULT; 1869b94acceSBarry Smith 1873a40ed3dSBarry Smith PetscFunctionBegin; 18881f57ec7SBarry Smith loc[0] = PETSC_DECIDE; loc[1] = PETSC_DECIDE; loc[2] = 300; loc[3] = 300; 18981f57ec7SBarry Smith 19077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 191596552b5SBarry Smith if (snes->setupcalled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call prior to SNESSetUp()"); 192ca161407SBarry Smith 19382bf6240SBarry Smith if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL);CHKERRQ(ierr);} 19482bf6240SBarry Smith ierr = OptionsGetString(snes->prefix,"-snes_type",method,256,&flg); 195052efed2SBarry Smith if (flg) { 19682bf6240SBarry Smith ierr = SNESSetType(snes,(SNESType) method); CHKERRQ(ierr); 1975334005bSBarry Smith } 1983c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_stol",&tmp, &flg); CHKERRQ(ierr); 199d64ed03dSBarry Smith if (flg) { 200d64ed03dSBarry Smith ierr = SNESSetTolerances(snes,PETSC_DEFAULT,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 201d64ed03dSBarry Smith } 2023c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_atol",&tmp, &flg); CHKERRQ(ierr); 203d64ed03dSBarry Smith if (flg) { 204d64ed03dSBarry Smith ierr = SNESSetTolerances(snes,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 205d64ed03dSBarry Smith } 2063c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_rtol",&tmp, &flg); CHKERRQ(ierr); 207d64ed03dSBarry Smith if (flg) { 208d64ed03dSBarry Smith ierr = SNESSetTolerances(snes,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 209d64ed03dSBarry Smith } 2103c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_it",&snes->max_its, &flg); CHKERRQ(ierr); 2113c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_funcs",&snes->max_funcs, &flg);CHKERRQ(ierr); 212d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_trtol",&tmp, &flg); CHKERRQ(ierr); 213d64ed03dSBarry Smith if (flg) { ierr = SNESSetTrustRegionTolerance(snes,tmp); CHKERRQ(ierr); } 214d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_fmin",&tmp, &flg); CHKERRQ(ierr); 215d64ed03dSBarry Smith if (flg) { ierr = SNESSetMinimizationFunctionTolerance(snes,tmp); CHKERRQ(ierr);} 2163c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_ksp_ew_conv", &flg); CHKERRQ(ierr); 2173c7409f5SSatish Balay if (flg) { snes->ksp_ewconv = 1; } 218b18e04deSLois Curfman McInnes ierr = OptionsGetInt(snes->prefix,"-snes_ksp_ew_version",&version,&flg); CHKERRQ(ierr); 219b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtol0",&rtol_0,&flg); CHKERRQ(ierr); 220b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtolmax",&rtol_max,&flg);CHKERRQ(ierr); 221b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_gamma",&gamma2,&flg); CHKERRQ(ierr); 222b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha",&alpha,&flg); CHKERRQ(ierr); 223b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha2",&alpha2,&flg); CHKERRQ(ierr); 224b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_threshold",&threshold,&flg);CHKERRQ(ierr); 22593c39befSBarry Smith 22693c39befSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_no_convergence_test",&flg); CHKERRQ(ierr); 22793c39befSBarry Smith if (flg) {snes->converged = 0;} 22893c39befSBarry Smith 2299b94acceSBarry Smith ierr = SNES_KSP_SetParametersEW(snes,version,rtol_0,rtol_max,gamma2,alpha, 2309b94acceSBarry Smith alpha2,threshold); CHKERRQ(ierr); 2315bbad29bSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_cancelmonitors",&flg); CHKERRQ(ierr); 2325cd90555SBarry Smith if (flg) {ierr = SNESClearMonitor(snes);CHKERRQ(ierr);} 2333c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_monitor",&flg); CHKERRQ(ierr); 234639f9d9dSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESDefaultMonitor,0);CHKERRQ(ierr);} 2353c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_smonitor",&flg); CHKERRQ(ierr); 2363f1db9ecSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESDefaultSMonitor,0); CHKERRQ(ierr);} 2373f1db9ecSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_vecmonitor",&flg); CHKERRQ(ierr); 2383f1db9ecSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESVecViewMonitor,0); CHKERRQ(ierr);} 23981f57ec7SBarry Smith ierr = OptionsGetIntArray(snes->prefix,"-snes_xmonitor",loc,&nmax,&flg);CHKERRQ(ierr); 2403c7409f5SSatish Balay if (flg) { 24117699dbbSLois Curfman McInnes int rank = 0; 242d7e8b826SBarry Smith DrawLG lg; 24317699dbbSLois Curfman McInnes MPI_Initialized(&rank); 24417699dbbSLois Curfman McInnes if (rank) MPI_Comm_rank(snes->comm,&rank); 24517699dbbSLois Curfman McInnes if (!rank) { 24681f57ec7SBarry Smith ierr = SNESLGMonitorCreate(0,0,loc[0],loc[1],loc[2],loc[3],&lg); CHKERRQ(ierr); 2479b94acceSBarry Smith ierr = SNESSetMonitor(snes,SNESLGMonitor,(void *)lg); CHKERRQ(ierr); 248f8c826e1SBarry Smith snes->xmonitor = lg; 2499b94acceSBarry Smith PLogObjectParent(snes,lg); 2509b94acceSBarry Smith } 2519b94acceSBarry Smith } 252e24b481bSBarry Smith 2533c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_fd", &flg); CHKERRQ(ierr); 2543c7409f5SSatish Balay if (flg && snes->method_class == SNES_NONLINEAR_EQUATIONS) { 2559b94acceSBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre, 2569b94acceSBarry Smith SNESDefaultComputeJacobian,snes->funP); CHKERRQ(ierr); 257981c4779SBarry Smith PLogInfo(snes,"SNESSetFromOptions: Setting default finite difference Jacobian matrix\n"); 258981c4779SBarry Smith } else if (flg && snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 25931615d04SLois Curfman McInnes ierr = SNESSetHessian(snes,snes->jacobian,snes->jacobian_pre, 26031615d04SLois Curfman McInnes SNESDefaultComputeHessian,snes->funP); CHKERRQ(ierr); 261d64ed03dSBarry Smith PLogInfo(snes,"SNESSetFromOptions: Setting default finite difference Hessian matrix\n"); 2629b94acceSBarry Smith } 263639f9d9dSBarry Smith 264639f9d9dSBarry Smith for ( i=0; i<numberofsetfromoptions; i++ ) { 265639f9d9dSBarry Smith ierr = (*othersetfromoptions[i])(snes); CHKERRQ(ierr); 266639f9d9dSBarry Smith } 267639f9d9dSBarry Smith 2689b94acceSBarry Smith ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 2699b94acceSBarry Smith ierr = SLESSetFromOptions(sles); CHKERRQ(ierr); 27093993e2dSLois Curfman McInnes 271e24b481bSBarry Smith /* set the special KSP monitor for matrix-free application */ 272e24b481bSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_mf_ksp_monitor",&flg); CHKERRQ(ierr); 273e24b481bSBarry Smith if (flg) { 274e24b481bSBarry Smith KSP ksp; 275e24b481bSBarry Smith ierr = SLESGetKSP(sles,&ksp);CHKERRQ(ierr); 276e24b481bSBarry Smith ierr = KSPSetMonitor(ksp,MatSNESFDMFKSPMonitor,PETSC_NULL);CHKERRQ(ierr); 277e24b481bSBarry Smith } 278e24b481bSBarry Smith 27982bf6240SBarry Smith /* 28093993e2dSLois Curfman McInnes Since the private setfromoptions requires the type to have 28193993e2dSLois Curfman McInnes been set already, we make sure a type is set by this time. 28282bf6240SBarry Smith */ 28382bf6240SBarry Smith if (!snes->type_name) { 28482bf6240SBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 28582bf6240SBarry Smith ierr = SNESSetType(snes,SNES_EQ_LS); CHKERRQ(ierr); 28682bf6240SBarry Smith } else { 28782bf6240SBarry Smith ierr = SNESSetType(snes,SNES_UM_TR); CHKERRQ(ierr); 28882bf6240SBarry Smith } 28982bf6240SBarry Smith } 29082bf6240SBarry Smith 29182bf6240SBarry Smith ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 29282bf6240SBarry Smith if (flg) { ierr = SNESPrintHelp(snes); CHKERRQ(ierr);} 29382bf6240SBarry Smith 2943a40ed3dSBarry Smith if (snes->setfromoptions) { 2953a40ed3dSBarry Smith ierr = (*snes->setfromoptions)(snes);CHKERRQ(ierr); 2963a40ed3dSBarry Smith } 2973a40ed3dSBarry Smith PetscFunctionReturn(0); 2989b94acceSBarry Smith } 2999b94acceSBarry Smith 300a847f771SSatish Balay 3015615d1e5SSatish Balay #undef __FUNC__ 302d4bb536fSBarry Smith #define __FUNC__ "SNESSetApplicationContext" 3039b94acceSBarry Smith /*@ 3049b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 3059b94acceSBarry Smith the nonlinear solvers. 3069b94acceSBarry Smith 307fee21e36SBarry Smith Collective on SNES 308fee21e36SBarry Smith 309c7afd0dbSLois Curfman McInnes Input Parameters: 310c7afd0dbSLois Curfman McInnes + snes - the SNES context 311c7afd0dbSLois Curfman McInnes - usrP - optional user context 312c7afd0dbSLois Curfman McInnes 31336851e7fSLois Curfman McInnes Level: intermediate 31436851e7fSLois Curfman McInnes 3159b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 3169b94acceSBarry Smith 3179b94acceSBarry Smith .seealso: SNESGetApplicationContext() 3189b94acceSBarry Smith @*/ 3199b94acceSBarry Smith int SNESSetApplicationContext(SNES snes,void *usrP) 3209b94acceSBarry Smith { 3213a40ed3dSBarry Smith PetscFunctionBegin; 32277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3239b94acceSBarry Smith snes->user = usrP; 3243a40ed3dSBarry Smith PetscFunctionReturn(0); 3259b94acceSBarry Smith } 32674679c65SBarry Smith 3275615d1e5SSatish Balay #undef __FUNC__ 328d4bb536fSBarry Smith #define __FUNC__ "SNESGetApplicationContext" 3299b94acceSBarry Smith /*@C 3309b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 3319b94acceSBarry Smith nonlinear solvers. 3329b94acceSBarry Smith 333c7afd0dbSLois Curfman McInnes Not Collective 334c7afd0dbSLois Curfman McInnes 3359b94acceSBarry Smith Input Parameter: 3369b94acceSBarry Smith . snes - SNES context 3379b94acceSBarry Smith 3389b94acceSBarry Smith Output Parameter: 3399b94acceSBarry Smith . usrP - user context 3409b94acceSBarry Smith 34136851e7fSLois Curfman McInnes Level: intermediate 34236851e7fSLois Curfman McInnes 3439b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 3449b94acceSBarry Smith 3459b94acceSBarry Smith .seealso: SNESSetApplicationContext() 3469b94acceSBarry Smith @*/ 3479b94acceSBarry Smith int SNESGetApplicationContext( SNES snes, void **usrP ) 3489b94acceSBarry Smith { 3493a40ed3dSBarry Smith PetscFunctionBegin; 35077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3519b94acceSBarry Smith *usrP = snes->user; 3523a40ed3dSBarry Smith PetscFunctionReturn(0); 3539b94acceSBarry Smith } 35474679c65SBarry Smith 3555615d1e5SSatish Balay #undef __FUNC__ 356d4bb536fSBarry Smith #define __FUNC__ "SNESGetIterationNumber" 3579b94acceSBarry Smith /*@ 358c8228a4eSBarry Smith SNESGetIterationNumber - Gets the number of nonlinear iterations completed 359c8228a4eSBarry Smith at this time. 3609b94acceSBarry Smith 361c7afd0dbSLois Curfman McInnes Not Collective 362c7afd0dbSLois Curfman McInnes 3639b94acceSBarry Smith Input Parameter: 3649b94acceSBarry Smith . snes - SNES context 3659b94acceSBarry Smith 3669b94acceSBarry Smith Output Parameter: 3679b94acceSBarry Smith . iter - iteration number 3689b94acceSBarry Smith 369c8228a4eSBarry Smith Notes: 370c8228a4eSBarry Smith For example, during the computation of iteration 2 this would return 1. 371c8228a4eSBarry Smith 372c8228a4eSBarry Smith This is useful for using lagged Jacobians (where one does not recompute the 373c8228a4eSBarry Smith Jacobian at each SNES iteration). For example the code 374c8228a4eSBarry Smith $ ierr = SNESGetIterationNumber(snes,&it); 375c8228a4eSBarry Smith $ if (!(it % 2)) { 376c8228a4eSBarry Smith $ compute Jacobian 377c8228a4eSBarry Smith $ } 378c8228a4eSBarry Smith can be used in your ComputeJacobian() function to cause the Jacobian to be 379c8228a4eSBarry Smith recomputed every second SNES iteration 380c8228a4eSBarry Smith 38136851e7fSLois Curfman McInnes Level: intermediate 38236851e7fSLois Curfman McInnes 3839b94acceSBarry Smith .keywords: SNES, nonlinear, get, iteration, number 3849b94acceSBarry Smith @*/ 3859b94acceSBarry Smith int SNESGetIterationNumber(SNES snes,int* iter) 3869b94acceSBarry Smith { 3873a40ed3dSBarry Smith PetscFunctionBegin; 38877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 38974679c65SBarry Smith PetscValidIntPointer(iter); 3909b94acceSBarry Smith *iter = snes->iter; 3913a40ed3dSBarry Smith PetscFunctionReturn(0); 3929b94acceSBarry Smith } 39374679c65SBarry Smith 3945615d1e5SSatish Balay #undef __FUNC__ 3955615d1e5SSatish Balay #define __FUNC__ "SNESGetFunctionNorm" 3969b94acceSBarry Smith /*@ 3979b94acceSBarry Smith SNESGetFunctionNorm - Gets the norm of the current function that was set 3989b94acceSBarry Smith with SNESSSetFunction(). 3999b94acceSBarry Smith 400c7afd0dbSLois Curfman McInnes Collective on SNES 401c7afd0dbSLois Curfman McInnes 4029b94acceSBarry Smith Input Parameter: 4039b94acceSBarry Smith . snes - SNES context 4049b94acceSBarry Smith 4059b94acceSBarry Smith Output Parameter: 4069b94acceSBarry Smith . fnorm - 2-norm of function 4079b94acceSBarry Smith 4089b94acceSBarry Smith Note: 4099b94acceSBarry Smith SNESGetFunctionNorm() is valid for SNES_NONLINEAR_EQUATIONS methods only. 410a86d99e1SLois Curfman McInnes A related routine for SNES_UNCONSTRAINED_MINIMIZATION methods is 411a86d99e1SLois Curfman McInnes SNESGetGradientNorm(). 4129b94acceSBarry Smith 41336851e7fSLois Curfman McInnes Level: intermediate 41436851e7fSLois Curfman McInnes 4159b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm 416a86d99e1SLois Curfman McInnes 417a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction() 4189b94acceSBarry Smith @*/ 4199b94acceSBarry Smith int SNESGetFunctionNorm(SNES snes,Scalar *fnorm) 4209b94acceSBarry Smith { 4213a40ed3dSBarry Smith PetscFunctionBegin; 42277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 42374679c65SBarry Smith PetscValidScalarPointer(fnorm); 42474679c65SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 425d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_NONLINEAR_EQUATIONS only"); 42674679c65SBarry Smith } 4279b94acceSBarry Smith *fnorm = snes->norm; 4283a40ed3dSBarry Smith PetscFunctionReturn(0); 4299b94acceSBarry Smith } 43074679c65SBarry Smith 4315615d1e5SSatish Balay #undef __FUNC__ 4325615d1e5SSatish Balay #define __FUNC__ "SNESGetGradientNorm" 4339b94acceSBarry Smith /*@ 4349b94acceSBarry Smith SNESGetGradientNorm - Gets the norm of the current gradient that was set 4359b94acceSBarry Smith with SNESSSetGradient(). 4369b94acceSBarry Smith 437c7afd0dbSLois Curfman McInnes Collective on SNES 438c7afd0dbSLois Curfman McInnes 4399b94acceSBarry Smith Input Parameter: 4409b94acceSBarry Smith . snes - SNES context 4419b94acceSBarry Smith 4429b94acceSBarry Smith Output Parameter: 4439b94acceSBarry Smith . fnorm - 2-norm of gradient 4449b94acceSBarry Smith 4459b94acceSBarry Smith Note: 4469b94acceSBarry Smith SNESGetGradientNorm() is valid for SNES_UNCONSTRAINED_MINIMIZATION 447a86d99e1SLois Curfman McInnes methods only. A related routine for SNES_NONLINEAR_EQUATIONS methods 448a86d99e1SLois Curfman McInnes is SNESGetFunctionNorm(). 4499b94acceSBarry Smith 45036851e7fSLois Curfman McInnes Level: intermediate 45136851e7fSLois Curfman McInnes 4529b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient, norm 453a86d99e1SLois Curfman McInnes 454a86d99e1SLois Curfman McInnes .seelso: SNESSetGradient() 4559b94acceSBarry Smith @*/ 4569b94acceSBarry Smith int SNESGetGradientNorm(SNES snes,Scalar *gnorm) 4579b94acceSBarry Smith { 4583a40ed3dSBarry Smith PetscFunctionBegin; 45977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 46074679c65SBarry Smith PetscValidScalarPointer(gnorm); 46174679c65SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 462d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 46374679c65SBarry Smith } 4649b94acceSBarry Smith *gnorm = snes->norm; 4653a40ed3dSBarry Smith PetscFunctionReturn(0); 4669b94acceSBarry Smith } 46774679c65SBarry Smith 4685615d1e5SSatish Balay #undef __FUNC__ 469d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberUnsuccessfulSteps" 4709b94acceSBarry Smith /*@ 4719b94acceSBarry Smith SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps 4729b94acceSBarry Smith attempted by the nonlinear solver. 4739b94acceSBarry Smith 474c7afd0dbSLois Curfman McInnes Not Collective 475c7afd0dbSLois Curfman McInnes 4769b94acceSBarry Smith Input Parameter: 4779b94acceSBarry Smith . snes - SNES context 4789b94acceSBarry Smith 4799b94acceSBarry Smith Output Parameter: 4809b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 4819b94acceSBarry Smith 482c96a6f78SLois Curfman McInnes Notes: 483c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 484c96a6f78SLois Curfman McInnes 48536851e7fSLois Curfman McInnes Level: intermediate 48636851e7fSLois Curfman McInnes 4879b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 4889b94acceSBarry Smith @*/ 4899b94acceSBarry Smith int SNESGetNumberUnsuccessfulSteps(SNES snes,int* nfails) 4909b94acceSBarry Smith { 4913a40ed3dSBarry Smith PetscFunctionBegin; 49277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 49374679c65SBarry Smith PetscValidIntPointer(nfails); 4949b94acceSBarry Smith *nfails = snes->nfailures; 4953a40ed3dSBarry Smith PetscFunctionReturn(0); 4969b94acceSBarry Smith } 497a847f771SSatish Balay 4985615d1e5SSatish Balay #undef __FUNC__ 499d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberLinearIterations" 500c96a6f78SLois Curfman McInnes /*@ 501c96a6f78SLois Curfman McInnes SNESGetNumberLinearIterations - Gets the total number of linear iterations 502c96a6f78SLois Curfman McInnes used by the nonlinear solver. 503c96a6f78SLois Curfman McInnes 504c7afd0dbSLois Curfman McInnes Not Collective 505c7afd0dbSLois Curfman McInnes 506c96a6f78SLois Curfman McInnes Input Parameter: 507c96a6f78SLois Curfman McInnes . snes - SNES context 508c96a6f78SLois Curfman McInnes 509c96a6f78SLois Curfman McInnes Output Parameter: 510c96a6f78SLois Curfman McInnes . lits - number of linear iterations 511c96a6f78SLois Curfman McInnes 512c96a6f78SLois Curfman McInnes Notes: 513c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 514c96a6f78SLois Curfman McInnes 51536851e7fSLois Curfman McInnes Level: intermediate 51636851e7fSLois Curfman McInnes 517c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations 518c96a6f78SLois Curfman McInnes @*/ 51986bddb7dSBarry Smith int SNESGetNumberLinearIterations(SNES snes,int* lits) 520c96a6f78SLois Curfman McInnes { 5213a40ed3dSBarry Smith PetscFunctionBegin; 522c96a6f78SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 523c96a6f78SLois Curfman McInnes PetscValidIntPointer(lits); 524c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 5253a40ed3dSBarry Smith PetscFunctionReturn(0); 526c96a6f78SLois Curfman McInnes } 527c96a6f78SLois Curfman McInnes 528c96a6f78SLois Curfman McInnes #undef __FUNC__ 529d4bb536fSBarry Smith #define __FUNC__ "SNESGetSLES" 5309b94acceSBarry Smith /*@C 5319b94acceSBarry Smith SNESGetSLES - Returns the SLES context for a SNES solver. 5329b94acceSBarry Smith 533c7afd0dbSLois Curfman McInnes Not Collective, but if SNES object is parallel, then SLES object is parallel 534c7afd0dbSLois Curfman McInnes 5359b94acceSBarry Smith Input Parameter: 5369b94acceSBarry Smith . snes - the SNES context 5379b94acceSBarry Smith 5389b94acceSBarry Smith Output Parameter: 5399b94acceSBarry Smith . sles - the SLES context 5409b94acceSBarry Smith 5419b94acceSBarry Smith Notes: 5429b94acceSBarry Smith The user can then directly manipulate the SLES context to set various 5439b94acceSBarry Smith options, etc. Likewise, the user can then extract and manipulate the 5449b94acceSBarry Smith KSP and PC contexts as well. 5459b94acceSBarry Smith 54636851e7fSLois Curfman McInnes Level: beginner 54736851e7fSLois Curfman McInnes 5489b94acceSBarry Smith .keywords: SNES, nonlinear, get, SLES, context 5499b94acceSBarry Smith 5509b94acceSBarry Smith .seealso: SLESGetPC(), SLESGetKSP() 5519b94acceSBarry Smith @*/ 5529b94acceSBarry Smith int SNESGetSLES(SNES snes,SLES *sles) 5539b94acceSBarry Smith { 5543a40ed3dSBarry Smith PetscFunctionBegin; 55577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 5569b94acceSBarry Smith *sles = snes->sles; 5573a40ed3dSBarry Smith PetscFunctionReturn(0); 5589b94acceSBarry Smith } 55982bf6240SBarry Smith 560e24b481bSBarry Smith #undef __FUNC__ 561e24b481bSBarry Smith #define __FUNC__ "SNESPublish_Petsc" 562e24b481bSBarry Smith static int SNESPublish_Petsc(PetscObject object) 563e24b481bSBarry Smith { 564e24b481bSBarry Smith #if defined(HAVE_AMS) 565e24b481bSBarry Smith SNES v = (SNES) object; 566e24b481bSBarry Smith int ierr; 567e24b481bSBarry Smith 568e24b481bSBarry Smith PetscFunctionBegin; 569e24b481bSBarry Smith 570e24b481bSBarry Smith /* if it is already published then return */ 571e24b481bSBarry Smith if (v->amem >=0 ) PetscFunctionReturn(0); 572e24b481bSBarry Smith 5733f1db9ecSBarry Smith ierr = PetscObjectPublishBaseBegin(object);CHKERRQ(ierr); 574e24b481bSBarry Smith ierr = AMS_Memory_add_field((AMS_Memory)v->amem,"Iteration",&v->iter,1,AMS_INT,AMS_READ, 575e24b481bSBarry Smith AMS_COMMON,AMS_REDUCT_UNDEF);CHKERRQ(ierr); 576e24b481bSBarry Smith ierr = AMS_Memory_add_field((AMS_Memory)v->amem,"Residual",&v->norm,1,AMS_DOUBLE,AMS_READ, 577e24b481bSBarry Smith AMS_COMMON,AMS_REDUCT_UNDEF);CHKERRQ(ierr); 578e24b481bSBarry Smith ierr = PetscObjectPublishBaseEnd(object);CHKERRQ(ierr); 579e24b481bSBarry Smith #else 580e24b481bSBarry Smith PetscFunctionBegin; 581e24b481bSBarry Smith #endif 582e24b481bSBarry Smith PetscFunctionReturn(0); 583e24b481bSBarry Smith } 584e24b481bSBarry Smith 5859b94acceSBarry Smith /* -----------------------------------------------------------*/ 5865615d1e5SSatish Balay #undef __FUNC__ 5875615d1e5SSatish Balay #define __FUNC__ "SNESCreate" 5889b94acceSBarry Smith /*@C 5899b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 5909b94acceSBarry Smith 591c7afd0dbSLois Curfman McInnes Collective on MPI_Comm 592c7afd0dbSLois Curfman McInnes 593c7afd0dbSLois Curfman McInnes Input Parameters: 594c7afd0dbSLois Curfman McInnes + comm - MPI communicator 595c7afd0dbSLois Curfman McInnes - type - type of method, either 596c7afd0dbSLois Curfman McInnes SNES_NONLINEAR_EQUATIONS (for systems of nonlinear equations) 597c7afd0dbSLois Curfman McInnes or SNES_UNCONSTRAINED_MINIMIZATION (for unconstrained minimization) 5989b94acceSBarry Smith 5999b94acceSBarry Smith Output Parameter: 6009b94acceSBarry Smith . outsnes - the new SNES context 6019b94acceSBarry Smith 602c7afd0dbSLois Curfman McInnes Options Database Keys: 603c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 604c7afd0dbSLois Curfman McInnes and no preconditioning matrix 605c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 606c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 607c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 608c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 609c1f60f51SBarry Smith 61036851e7fSLois Curfman McInnes Level: beginner 61136851e7fSLois Curfman McInnes 6129b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 6139b94acceSBarry Smith 61463a78c88SLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy() 6159b94acceSBarry Smith @*/ 6164b0e389bSBarry Smith int SNESCreate(MPI_Comm comm,SNESProblemType type,SNES *outsnes) 6179b94acceSBarry Smith { 6189b94acceSBarry Smith int ierr; 6199b94acceSBarry Smith SNES snes; 6209b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 62137fcc0dbSBarry Smith 6223a40ed3dSBarry Smith PetscFunctionBegin; 6239b94acceSBarry Smith *outsnes = 0; 624d64ed03dSBarry Smith if (type != SNES_UNCONSTRAINED_MINIMIZATION && type != SNES_NONLINEAR_EQUATIONS){ 625d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"incorrect method type"); 626d64ed03dSBarry Smith } 6273f1db9ecSBarry Smith PetscHeaderCreate(snes,_p_SNES,int,SNES_COOKIE,0,"SNES",comm,SNESDestroy,SNESView); 6289b94acceSBarry Smith PLogObjectCreate(snes); 629e24b481bSBarry Smith snes->bops->publish = SNESPublish_Petsc; 6309b94acceSBarry Smith snes->max_its = 50; 6319750a799SBarry Smith snes->max_funcs = 10000; 6329b94acceSBarry Smith snes->norm = 0.0; 6335a2d0531SBarry Smith if (type == SNES_UNCONSTRAINED_MINIMIZATION) { 634b18e04deSLois Curfman McInnes snes->rtol = 1.e-8; 635b18e04deSLois Curfman McInnes snes->ttol = 0.0; 6369b94acceSBarry Smith snes->atol = 1.e-10; 637d64ed03dSBarry Smith } else { 638b4874afaSBarry Smith snes->rtol = 1.e-8; 639b4874afaSBarry Smith snes->ttol = 0.0; 640b4874afaSBarry Smith snes->atol = 1.e-50; 641b4874afaSBarry Smith } 6429b94acceSBarry Smith snes->xtol = 1.e-8; 643b18e04deSLois Curfman McInnes snes->trunctol = 1.e-12; /* no longer used */ 6449b94acceSBarry Smith snes->nfuncs = 0; 6459b94acceSBarry Smith snes->nfailures = 0; 6467a00f4a9SLois Curfman McInnes snes->linear_its = 0; 647639f9d9dSBarry Smith snes->numbermonitors = 0; 6489b94acceSBarry Smith snes->data = 0; 6499b94acceSBarry Smith snes->view = 0; 6509b94acceSBarry Smith snes->computeumfunction = 0; 6519b94acceSBarry Smith snes->umfunP = 0; 6529b94acceSBarry Smith snes->fc = 0; 6539b94acceSBarry Smith snes->deltatol = 1.e-12; 6549b94acceSBarry Smith snes->fmin = -1.e30; 6559b94acceSBarry Smith snes->method_class = type; 6569b94acceSBarry Smith snes->set_method_called = 0; 65782bf6240SBarry Smith snes->setupcalled = 0; 6589b94acceSBarry Smith snes->ksp_ewconv = 0; 6596f24a144SLois Curfman McInnes snes->xmonitor = 0; 6606f24a144SLois Curfman McInnes snes->vwork = 0; 6616f24a144SLois Curfman McInnes snes->nwork = 0; 662758f92a0SBarry Smith snes->conv_hist_len = 0; 663758f92a0SBarry Smith snes->conv_hist_max = 0; 664758f92a0SBarry Smith snes->conv_hist = PETSC_NULL; 665758f92a0SBarry Smith snes->conv_hist_its = PETSC_NULL; 666758f92a0SBarry Smith snes->conv_hist_reset = PETSC_TRUE; 6679b94acceSBarry Smith 6689b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 6690452661fSBarry Smith kctx = PetscNew(SNES_KSP_EW_ConvCtx); CHKPTRQ(kctx); 670eed86810SBarry Smith PLogObjectMemory(snes,sizeof(SNES_KSP_EW_ConvCtx)); 6719b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 6729b94acceSBarry Smith kctx->version = 2; 6739b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 6749b94acceSBarry Smith this was too large for some test cases */ 6759b94acceSBarry Smith kctx->rtol_last = 0; 6769b94acceSBarry Smith kctx->rtol_max = .9; 6779b94acceSBarry Smith kctx->gamma = 1.0; 6789b94acceSBarry Smith kctx->alpha2 = .5*(1.0 + sqrt(5.0)); 6799b94acceSBarry Smith kctx->alpha = kctx->alpha2; 6809b94acceSBarry Smith kctx->threshold = .1; 6819b94acceSBarry Smith kctx->lresid_last = 0; 6829b94acceSBarry Smith kctx->norm_last = 0; 6839b94acceSBarry Smith 6849b94acceSBarry Smith ierr = SLESCreate(comm,&snes->sles); CHKERRQ(ierr); 6859b94acceSBarry Smith PLogObjectParent(snes,snes->sles) 6865334005bSBarry Smith 6879b94acceSBarry Smith *outsnes = snes; 688e24b481bSBarry Smith PetscPublishAll(snes); 6893a40ed3dSBarry Smith PetscFunctionReturn(0); 6909b94acceSBarry Smith } 6919b94acceSBarry Smith 6929b94acceSBarry Smith /* --------------------------------------------------------------- */ 6935615d1e5SSatish Balay #undef __FUNC__ 694d4bb536fSBarry Smith #define __FUNC__ "SNESSetFunction" 6959b94acceSBarry Smith /*@C 6969b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 6979b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 6989b94acceSBarry Smith equations. 6999b94acceSBarry Smith 700fee21e36SBarry Smith Collective on SNES 701fee21e36SBarry Smith 702c7afd0dbSLois Curfman McInnes Input Parameters: 703c7afd0dbSLois Curfman McInnes + snes - the SNES context 704c7afd0dbSLois Curfman McInnes . func - function evaluation routine 705c7afd0dbSLois Curfman McInnes . r - vector to store function value 706c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 707c7afd0dbSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 7089b94acceSBarry Smith 709c7afd0dbSLois Curfman McInnes Calling sequence of func: 7108d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Vec f,void *ctx); 711c7afd0dbSLois Curfman McInnes 712313e4042SLois Curfman McInnes . f - function vector 713c7afd0dbSLois Curfman McInnes - ctx - optional user-defined function context 7149b94acceSBarry Smith 7159b94acceSBarry Smith Notes: 7169b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 7179b94acceSBarry Smith $ f'(x) x = -f(x), 718c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 7199b94acceSBarry Smith 7209b94acceSBarry Smith SNESSetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 7219b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 7229b94acceSBarry Smith SNESSetMinimizationFunction() and SNESSetGradient(); 7239b94acceSBarry Smith 72436851e7fSLois Curfman McInnes Level: beginner 72536851e7fSLois Curfman McInnes 7269b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 7279b94acceSBarry Smith 728a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 7299b94acceSBarry Smith @*/ 7305334005bSBarry Smith int SNESSetFunction( SNES snes, Vec r, int (*func)(SNES,Vec,Vec,void*),void *ctx) 7319b94acceSBarry Smith { 7323a40ed3dSBarry Smith PetscFunctionBegin; 73377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 734a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 735a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 736a8c6a408SBarry Smith } 7379b94acceSBarry Smith snes->computefunction = func; 7389b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 7399b94acceSBarry Smith snes->funP = ctx; 7403a40ed3dSBarry Smith PetscFunctionReturn(0); 7419b94acceSBarry Smith } 7429b94acceSBarry Smith 7435615d1e5SSatish Balay #undef __FUNC__ 7445615d1e5SSatish Balay #define __FUNC__ "SNESComputeFunction" 7459b94acceSBarry Smith /*@ 74636851e7fSLois Curfman McInnes SNESComputeFunction - Calls the function that has been set with 7479b94acceSBarry Smith SNESSetFunction(). 7489b94acceSBarry Smith 749c7afd0dbSLois Curfman McInnes Collective on SNES 750c7afd0dbSLois Curfman McInnes 7519b94acceSBarry Smith Input Parameters: 752c7afd0dbSLois Curfman McInnes + snes - the SNES context 753c7afd0dbSLois Curfman McInnes - x - input vector 7549b94acceSBarry Smith 7559b94acceSBarry Smith Output Parameter: 7563638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 7579b94acceSBarry Smith 7581bffabb2SLois Curfman McInnes Notes: 7599b94acceSBarry Smith SNESComputeFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 7609b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 7619b94acceSBarry Smith SNESComputeMinimizationFunction() and SNESComputeGradient(); 7629b94acceSBarry Smith 76336851e7fSLois Curfman McInnes SNESComputeFunction() is typically used within nonlinear solvers 76436851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 76536851e7fSLois Curfman McInnes themselves. 76636851e7fSLois Curfman McInnes 76736851e7fSLois Curfman McInnes Level: developer 76836851e7fSLois Curfman McInnes 7699b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 7709b94acceSBarry Smith 771a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 7729b94acceSBarry Smith @*/ 7739b94acceSBarry Smith int SNESComputeFunction(SNES snes,Vec x, Vec y) 7749b94acceSBarry Smith { 7759b94acceSBarry Smith int ierr; 7769b94acceSBarry Smith 7773a40ed3dSBarry Smith PetscFunctionBegin; 778d4bb536fSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 779a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 780d4bb536fSBarry Smith } 7819b94acceSBarry Smith PLogEventBegin(SNES_FunctionEval,snes,x,y,0); 782d64ed03dSBarry Smith PetscStackPush("SNES user function"); 7839b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 784d64ed03dSBarry Smith PetscStackPop; 785ae3c334cSLois Curfman McInnes snes->nfuncs++; 7869b94acceSBarry Smith PLogEventEnd(SNES_FunctionEval,snes,x,y,0); 7873a40ed3dSBarry Smith PetscFunctionReturn(0); 7889b94acceSBarry Smith } 7899b94acceSBarry Smith 7905615d1e5SSatish Balay #undef __FUNC__ 791d4bb536fSBarry Smith #define __FUNC__ "SNESSetMinimizationFunction" 7929b94acceSBarry Smith /*@C 7939b94acceSBarry Smith SNESSetMinimizationFunction - Sets the function evaluation routine for 7949b94acceSBarry Smith unconstrained minimization. 7959b94acceSBarry Smith 796fee21e36SBarry Smith Collective on SNES 797fee21e36SBarry Smith 798c7afd0dbSLois Curfman McInnes Input Parameters: 799c7afd0dbSLois Curfman McInnes + snes - the SNES context 800c7afd0dbSLois Curfman McInnes . func - function evaluation routine 801c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 802c7afd0dbSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 8039b94acceSBarry Smith 804c7afd0dbSLois Curfman McInnes Calling sequence of func: 8058d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,double *f,void *ctx); 806c7afd0dbSLois Curfman McInnes 807c7afd0dbSLois Curfman McInnes + x - input vector 8089b94acceSBarry Smith . f - function 809c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined function context 8109b94acceSBarry Smith 81136851e7fSLois Curfman McInnes Level: beginner 81236851e7fSLois Curfman McInnes 8139b94acceSBarry Smith Notes: 8149b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 8159b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 8169b94acceSBarry Smith SNESSetFunction(). 8179b94acceSBarry Smith 8189b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimization, function 8199b94acceSBarry Smith 820a86d99e1SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESComputeMinimizationFunction(), 821a86d99e1SLois Curfman McInnes SNESSetHessian(), SNESSetGradient() 8229b94acceSBarry Smith @*/ 8239b94acceSBarry Smith int SNESSetMinimizationFunction(SNES snes,int (*func)(SNES,Vec,double*,void*), 8249b94acceSBarry Smith void *ctx) 8259b94acceSBarry Smith { 8263a40ed3dSBarry Smith PetscFunctionBegin; 82777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 828a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 829a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 830a8c6a408SBarry Smith } 8319b94acceSBarry Smith snes->computeumfunction = func; 8329b94acceSBarry Smith snes->umfunP = ctx; 8333a40ed3dSBarry Smith PetscFunctionReturn(0); 8349b94acceSBarry Smith } 8359b94acceSBarry Smith 8365615d1e5SSatish Balay #undef __FUNC__ 8375615d1e5SSatish Balay #define __FUNC__ "SNESComputeMinimizationFunction" 8389b94acceSBarry Smith /*@ 8399b94acceSBarry Smith SNESComputeMinimizationFunction - Computes the function that has been 8409b94acceSBarry Smith set with SNESSetMinimizationFunction(). 8419b94acceSBarry Smith 842c7afd0dbSLois Curfman McInnes Collective on SNES 843c7afd0dbSLois Curfman McInnes 8449b94acceSBarry Smith Input Parameters: 845c7afd0dbSLois Curfman McInnes + snes - the SNES context 846c7afd0dbSLois Curfman McInnes - x - input vector 8479b94acceSBarry Smith 8489b94acceSBarry Smith Output Parameter: 8499b94acceSBarry Smith . y - function value 8509b94acceSBarry Smith 8519b94acceSBarry Smith Notes: 8529b94acceSBarry Smith SNESComputeMinimizationFunction() is valid only for 8539b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 8549b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 855a86d99e1SLois Curfman McInnes 85636851e7fSLois Curfman McInnes SNESComputeMinimizationFunction() is typically used within minimization 85736851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 85836851e7fSLois Curfman McInnes themselves. 85936851e7fSLois Curfman McInnes 86036851e7fSLois Curfman McInnes Level: developer 86136851e7fSLois Curfman McInnes 862a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, minimization, function 863a86d99e1SLois Curfman McInnes 864a86d99e1SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESGetMinimizationFunction(), 865a86d99e1SLois Curfman McInnes SNESComputeGradient(), SNESComputeHessian() 8669b94acceSBarry Smith @*/ 8679b94acceSBarry Smith int SNESComputeMinimizationFunction(SNES snes,Vec x,double *y) 8689b94acceSBarry Smith { 8699b94acceSBarry Smith int ierr; 8703a40ed3dSBarry Smith 8713a40ed3dSBarry Smith PetscFunctionBegin; 872a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 873a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 874a8c6a408SBarry Smith } 8759b94acceSBarry Smith PLogEventBegin(SNES_MinimizationFunctionEval,snes,x,y,0); 876d64ed03dSBarry Smith PetscStackPush("SNES user minimzation function"); 8779b94acceSBarry Smith ierr = (*snes->computeumfunction)(snes,x,y,snes->umfunP); CHKERRQ(ierr); 878d64ed03dSBarry Smith PetscStackPop; 879ae3c334cSLois Curfman McInnes snes->nfuncs++; 8809b94acceSBarry Smith PLogEventEnd(SNES_MinimizationFunctionEval,snes,x,y,0); 8813a40ed3dSBarry Smith PetscFunctionReturn(0); 8829b94acceSBarry Smith } 8839b94acceSBarry Smith 8845615d1e5SSatish Balay #undef __FUNC__ 885d4bb536fSBarry Smith #define __FUNC__ "SNESSetGradient" 8869b94acceSBarry Smith /*@C 8879b94acceSBarry Smith SNESSetGradient - Sets the gradient evaluation routine and gradient 8889b94acceSBarry Smith vector for use by the SNES routines. 8899b94acceSBarry Smith 890c7afd0dbSLois Curfman McInnes Collective on SNES 891c7afd0dbSLois Curfman McInnes 8929b94acceSBarry Smith Input Parameters: 893c7afd0dbSLois Curfman McInnes + snes - the SNES context 8949b94acceSBarry Smith . func - function evaluation routine 895044dda88SLois Curfman McInnes . ctx - optional user-defined context for private data for the 896044dda88SLois Curfman McInnes gradient evaluation routine (may be PETSC_NULL) 897c7afd0dbSLois Curfman McInnes - r - vector to store gradient value 898fee21e36SBarry Smith 8999b94acceSBarry Smith Calling sequence of func: 9008d76a1e5SLois Curfman McInnes $ func (SNES, Vec x, Vec g, void *ctx); 9019b94acceSBarry Smith 902c7afd0dbSLois Curfman McInnes + x - input vector 9039b94acceSBarry Smith . g - gradient vector 904c7afd0dbSLois Curfman McInnes - ctx - optional user-defined gradient context 9059b94acceSBarry Smith 9069b94acceSBarry Smith Notes: 9079b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 9089b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 9099b94acceSBarry Smith SNESSetFunction(). 9109b94acceSBarry Smith 91136851e7fSLois Curfman McInnes Level: beginner 91236851e7fSLois Curfman McInnes 9139b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 9149b94acceSBarry Smith 915a86d99e1SLois Curfman McInnes .seealso: SNESGetGradient(), SNESComputeGradient(), SNESSetHessian(), 916a86d99e1SLois Curfman McInnes SNESSetMinimizationFunction(), 9179b94acceSBarry Smith @*/ 91874679c65SBarry Smith int SNESSetGradient(SNES snes,Vec r,int (*func)(SNES,Vec,Vec,void*),void *ctx) 9199b94acceSBarry Smith { 9203a40ed3dSBarry Smith PetscFunctionBegin; 92177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 922a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 923a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 924a8c6a408SBarry Smith } 9259b94acceSBarry Smith snes->computefunction = func; 9269b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 9279b94acceSBarry Smith snes->funP = ctx; 9283a40ed3dSBarry Smith PetscFunctionReturn(0); 9299b94acceSBarry Smith } 9309b94acceSBarry Smith 9315615d1e5SSatish Balay #undef __FUNC__ 9325615d1e5SSatish Balay #define __FUNC__ "SNESComputeGradient" 9339b94acceSBarry Smith /*@ 934a86d99e1SLois Curfman McInnes SNESComputeGradient - Computes the gradient that has been set with 935a86d99e1SLois Curfman McInnes SNESSetGradient(). 9369b94acceSBarry Smith 937c7afd0dbSLois Curfman McInnes Collective on SNES 938c7afd0dbSLois Curfman McInnes 9399b94acceSBarry Smith Input Parameters: 940c7afd0dbSLois Curfman McInnes + snes - the SNES context 941c7afd0dbSLois Curfman McInnes - x - input vector 9429b94acceSBarry Smith 9439b94acceSBarry Smith Output Parameter: 9449b94acceSBarry Smith . y - gradient vector 9459b94acceSBarry Smith 9469b94acceSBarry Smith Notes: 9479b94acceSBarry Smith SNESComputeGradient() is valid only for 9489b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 9499b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 950a86d99e1SLois Curfman McInnes 95136851e7fSLois Curfman McInnes SNESComputeGradient() is typically used within minimization 95236851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 95336851e7fSLois Curfman McInnes themselves. 95436851e7fSLois Curfman McInnes 95536851e7fSLois Curfman McInnes Level: developer 95636851e7fSLois Curfman McInnes 957a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, gradient 958a86d99e1SLois Curfman McInnes 959a86d99e1SLois Curfman McInnes .seealso: SNESSetGradient(), SNESGetGradient(), 960a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction(), SNESComputeHessian() 9619b94acceSBarry Smith @*/ 9629b94acceSBarry Smith int SNESComputeGradient(SNES snes,Vec x, Vec y) 9639b94acceSBarry Smith { 9649b94acceSBarry Smith int ierr; 9653a40ed3dSBarry Smith 9663a40ed3dSBarry Smith PetscFunctionBegin; 9673a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 968a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 9693a40ed3dSBarry Smith } 9709b94acceSBarry Smith PLogEventBegin(SNES_GradientEval,snes,x,y,0); 971d64ed03dSBarry Smith PetscStackPush("SNES user gradient function"); 9729b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 973d64ed03dSBarry Smith PetscStackPop; 9749b94acceSBarry Smith PLogEventEnd(SNES_GradientEval,snes,x,y,0); 9753a40ed3dSBarry Smith PetscFunctionReturn(0); 9769b94acceSBarry Smith } 9779b94acceSBarry Smith 9785615d1e5SSatish Balay #undef __FUNC__ 9795615d1e5SSatish Balay #define __FUNC__ "SNESComputeJacobian" 98062fef451SLois Curfman McInnes /*@ 98162fef451SLois Curfman McInnes SNESComputeJacobian - Computes the Jacobian matrix that has been 98262fef451SLois Curfman McInnes set with SNESSetJacobian(). 98362fef451SLois Curfman McInnes 984c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 985c7afd0dbSLois Curfman McInnes 98662fef451SLois Curfman McInnes Input Parameters: 987c7afd0dbSLois Curfman McInnes + snes - the SNES context 988c7afd0dbSLois Curfman McInnes - x - input vector 98962fef451SLois Curfman McInnes 99062fef451SLois Curfman McInnes Output Parameters: 991c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 99262fef451SLois Curfman McInnes . B - optional preconditioning matrix 993c7afd0dbSLois Curfman McInnes - flag - flag indicating matrix structure 994fee21e36SBarry Smith 99562fef451SLois Curfman McInnes Notes: 99662fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 99762fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 99862fef451SLois Curfman McInnes 999dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 1000dc5a77f8SLois Curfman McInnes flag parameter. 100162fef451SLois Curfman McInnes 100262fef451SLois Curfman McInnes SNESComputeJacobian() is valid only for SNES_NONLINEAR_EQUATIONS 100362fef451SLois Curfman McInnes methods. An analogous routine for SNES_UNCONSTRAINED_MINIMIZATION 1004005c665bSBarry Smith methods is SNESComputeHessian(). 100562fef451SLois Curfman McInnes 100636851e7fSLois Curfman McInnes SNESComputeJacobian() is typically used within nonlinear solver 100736851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 100836851e7fSLois Curfman McInnes themselves. 100936851e7fSLois Curfman McInnes 101036851e7fSLois Curfman McInnes Level: developer 101136851e7fSLois Curfman McInnes 101262fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 101362fef451SLois Curfman McInnes 101462fef451SLois Curfman McInnes .seealso: SNESSetJacobian(), SLESSetOperators() 101562fef451SLois Curfman McInnes @*/ 10169b94acceSBarry Smith int SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 10179b94acceSBarry Smith { 10189b94acceSBarry Smith int ierr; 10193a40ed3dSBarry Smith 10203a40ed3dSBarry Smith PetscFunctionBegin; 10213a40ed3dSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1022a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 10233a40ed3dSBarry Smith } 10243a40ed3dSBarry Smith if (!snes->computejacobian) PetscFunctionReturn(0); 10259b94acceSBarry Smith PLogEventBegin(SNES_JacobianEval,snes,X,*A,*B); 1026c4fc05e7SBarry Smith *flg = DIFFERENT_NONZERO_PATTERN; 1027d64ed03dSBarry Smith PetscStackPush("SNES user Jacobian function"); 10289b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 1029d64ed03dSBarry Smith PetscStackPop; 10309b94acceSBarry Smith PLogEventEnd(SNES_JacobianEval,snes,X,*A,*B); 10316d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 103277c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 103377c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 10343a40ed3dSBarry Smith PetscFunctionReturn(0); 10359b94acceSBarry Smith } 10369b94acceSBarry Smith 10375615d1e5SSatish Balay #undef __FUNC__ 10385615d1e5SSatish Balay #define __FUNC__ "SNESComputeHessian" 103962fef451SLois Curfman McInnes /*@ 104062fef451SLois Curfman McInnes SNESComputeHessian - Computes the Hessian matrix that has been 104162fef451SLois Curfman McInnes set with SNESSetHessian(). 104262fef451SLois Curfman McInnes 1043c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1044c7afd0dbSLois Curfman McInnes 104562fef451SLois Curfman McInnes Input Parameters: 1046c7afd0dbSLois Curfman McInnes + snes - the SNES context 1047c7afd0dbSLois Curfman McInnes - x - input vector 104862fef451SLois Curfman McInnes 104962fef451SLois Curfman McInnes Output Parameters: 1050c7afd0dbSLois Curfman McInnes + A - Hessian matrix 105162fef451SLois Curfman McInnes . B - optional preconditioning matrix 1052c7afd0dbSLois Curfman McInnes - flag - flag indicating matrix structure 1053fee21e36SBarry Smith 105462fef451SLois Curfman McInnes Notes: 105562fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 105662fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 105762fef451SLois Curfman McInnes 1058dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 1059dc5a77f8SLois Curfman McInnes flag parameter. 106062fef451SLois Curfman McInnes 106162fef451SLois Curfman McInnes SNESComputeHessian() is valid only for 106262fef451SLois Curfman McInnes SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 106362fef451SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods is SNESComputeJacobian(). 106462fef451SLois Curfman McInnes 106536851e7fSLois Curfman McInnes SNESComputeHessian() is typically used within minimization 106636851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 106736851e7fSLois Curfman McInnes themselves. 106836851e7fSLois Curfman McInnes 106936851e7fSLois Curfman McInnes Level: developer 107036851e7fSLois Curfman McInnes 107162fef451SLois Curfman McInnes .keywords: SNES, compute, Hessian, matrix 107262fef451SLois Curfman McInnes 1073a86d99e1SLois Curfman McInnes .seealso: SNESSetHessian(), SLESSetOperators(), SNESComputeGradient(), 1074a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction() 107562fef451SLois Curfman McInnes @*/ 107662fef451SLois Curfman McInnes int SNESComputeHessian(SNES snes,Vec x,Mat *A,Mat *B,MatStructure *flag) 10779b94acceSBarry Smith { 10789b94acceSBarry Smith int ierr; 10793a40ed3dSBarry Smith 10803a40ed3dSBarry Smith PetscFunctionBegin; 10813a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1082a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 10833a40ed3dSBarry Smith } 10843a40ed3dSBarry Smith if (!snes->computejacobian) PetscFunctionReturn(0); 108562fef451SLois Curfman McInnes PLogEventBegin(SNES_HessianEval,snes,x,*A,*B); 10860f4a323eSLois Curfman McInnes *flag = DIFFERENT_NONZERO_PATTERN; 1087d64ed03dSBarry Smith PetscStackPush("SNES user Hessian function"); 108862fef451SLois Curfman McInnes ierr = (*snes->computejacobian)(snes,x,A,B,flag,snes->jacP); CHKERRQ(ierr); 1089d64ed03dSBarry Smith PetscStackPop; 109062fef451SLois Curfman McInnes PLogEventEnd(SNES_HessianEval,snes,x,*A,*B); 10910f4a323eSLois Curfman McInnes /* make sure user returned a correct Jacobian and preconditioner */ 109277c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 109377c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 10943a40ed3dSBarry Smith PetscFunctionReturn(0); 10959b94acceSBarry Smith } 10969b94acceSBarry Smith 10975615d1e5SSatish Balay #undef __FUNC__ 1098d4bb536fSBarry Smith #define __FUNC__ "SNESSetJacobian" 10999b94acceSBarry Smith /*@C 11009b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 1101044dda88SLois Curfman McInnes location to store the matrix. 11029b94acceSBarry Smith 1103c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1104c7afd0dbSLois Curfman McInnes 11059b94acceSBarry Smith Input Parameters: 1106c7afd0dbSLois Curfman McInnes + snes - the SNES context 11079b94acceSBarry Smith . A - Jacobian matrix 11089b94acceSBarry Smith . B - preconditioner matrix (usually same as the Jacobian) 11099b94acceSBarry Smith . func - Jacobian evaluation routine 1110c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 11112cd2dfdcSLois Curfman McInnes Jacobian evaluation routine (may be PETSC_NULL) 11129b94acceSBarry Smith 11139b94acceSBarry Smith Calling sequence of func: 11148d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 11159b94acceSBarry Smith 1116c7afd0dbSLois Curfman McInnes + x - input vector 11179b94acceSBarry Smith . A - Jacobian matrix 11189b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1119ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 1120ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 1121c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined Jacobian context 11229b94acceSBarry Smith 11239b94acceSBarry Smith Notes: 1124dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 11252cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1126ac21db08SLois Curfman McInnes 1127ac21db08SLois Curfman McInnes The routine func() takes Mat * as the matrix arguments rather than Mat. 11289b94acceSBarry Smith This allows the Jacobian evaluation routine to replace A and/or B with a 11299b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 11309b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 11319b94acceSBarry Smith throughout the global iterations. 11329b94acceSBarry Smith 113336851e7fSLois Curfman McInnes Level: beginner 113436851e7fSLois Curfman McInnes 11359b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 11369b94acceSBarry Smith 1137ac21db08SLois Curfman McInnes .seealso: SLESSetOperators(), SNESSetFunction() 11389b94acceSBarry Smith @*/ 11399b94acceSBarry Smith int SNESSetJacobian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 11409b94acceSBarry Smith MatStructure*,void*),void *ctx) 11419b94acceSBarry Smith { 11423a40ed3dSBarry Smith PetscFunctionBegin; 114377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1144a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1145a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 1146a8c6a408SBarry Smith } 11479b94acceSBarry Smith snes->computejacobian = func; 11489b94acceSBarry Smith snes->jacP = ctx; 11499b94acceSBarry Smith snes->jacobian = A; 11509b94acceSBarry Smith snes->jacobian_pre = B; 11513a40ed3dSBarry Smith PetscFunctionReturn(0); 11529b94acceSBarry Smith } 115362fef451SLois Curfman McInnes 11545615d1e5SSatish Balay #undef __FUNC__ 1155d4bb536fSBarry Smith #define __FUNC__ "SNESGetJacobian" 1156b4fd4287SBarry Smith /*@ 1157b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 1158b4fd4287SBarry Smith provided context for evaluating the Jacobian. 1159b4fd4287SBarry Smith 1160c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 1161c7afd0dbSLois Curfman McInnes 1162b4fd4287SBarry Smith Input Parameter: 1163b4fd4287SBarry Smith . snes - the nonlinear solver context 1164b4fd4287SBarry Smith 1165b4fd4287SBarry Smith Output Parameters: 1166c7afd0dbSLois Curfman McInnes + A - location to stash Jacobian matrix (or PETSC_NULL) 1167b4fd4287SBarry Smith . B - location to stash preconditioner matrix (or PETSC_NULL) 1168c7afd0dbSLois Curfman McInnes - ctx - location to stash Jacobian ctx (or PETSC_NULL) 1169fee21e36SBarry Smith 117036851e7fSLois Curfman McInnes Level: advanced 117136851e7fSLois Curfman McInnes 1172b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian() 1173b4fd4287SBarry Smith @*/ 1174b4fd4287SBarry Smith int SNESGetJacobian(SNES snes,Mat *A,Mat *B, void **ctx) 1175b4fd4287SBarry Smith { 11763a40ed3dSBarry Smith PetscFunctionBegin; 11773a40ed3dSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1178a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 11793a40ed3dSBarry Smith } 1180b4fd4287SBarry Smith if (A) *A = snes->jacobian; 1181b4fd4287SBarry Smith if (B) *B = snes->jacobian_pre; 1182b4fd4287SBarry Smith if (ctx) *ctx = snes->jacP; 11833a40ed3dSBarry Smith PetscFunctionReturn(0); 1184b4fd4287SBarry Smith } 1185b4fd4287SBarry Smith 11865615d1e5SSatish Balay #undef __FUNC__ 1187d4bb536fSBarry Smith #define __FUNC__ "SNESSetHessian" 11889b94acceSBarry Smith /*@C 11899b94acceSBarry Smith SNESSetHessian - Sets the function to compute Hessian as well as the 1190044dda88SLois Curfman McInnes location to store the matrix. 11919b94acceSBarry Smith 1192c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1193c7afd0dbSLois Curfman McInnes 11949b94acceSBarry Smith Input Parameters: 1195c7afd0dbSLois Curfman McInnes + snes - the SNES context 11969b94acceSBarry Smith . A - Hessian matrix 11979b94acceSBarry Smith . B - preconditioner matrix (usually same as the Hessian) 11989b94acceSBarry Smith . func - Jacobian evaluation routine 1199c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 1200313e4042SLois Curfman McInnes Hessian evaluation routine (may be PETSC_NULL) 12019b94acceSBarry Smith 12029b94acceSBarry Smith Calling sequence of func: 12038d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 12049b94acceSBarry Smith 1205c7afd0dbSLois Curfman McInnes + x - input vector 12069b94acceSBarry Smith . A - Hessian matrix 12079b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1208ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 1209ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 1210c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined Hessian context 12119b94acceSBarry Smith 12129b94acceSBarry Smith Notes: 1213dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 12142cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1215ac21db08SLois Curfman McInnes 12169b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 12179b94acceSBarry Smith This allows the Hessian evaluation routine to replace A and/or B with a 12189b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 12199b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 12209b94acceSBarry Smith throughout the global iterations. 12219b94acceSBarry Smith 122236851e7fSLois Curfman McInnes Level: beginner 122336851e7fSLois Curfman McInnes 12249b94acceSBarry Smith .keywords: SNES, nonlinear, set, Hessian, matrix 12259b94acceSBarry Smith 1226ac21db08SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESSetGradient(), SLESSetOperators() 12279b94acceSBarry Smith @*/ 12289b94acceSBarry Smith int SNESSetHessian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 12299b94acceSBarry Smith MatStructure*,void*),void *ctx) 12309b94acceSBarry Smith { 12313a40ed3dSBarry Smith PetscFunctionBegin; 123277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1233d4bb536fSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1234a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 1235d4bb536fSBarry Smith } 12369b94acceSBarry Smith snes->computejacobian = func; 12379b94acceSBarry Smith snes->jacP = ctx; 12389b94acceSBarry Smith snes->jacobian = A; 12399b94acceSBarry Smith snes->jacobian_pre = B; 12403a40ed3dSBarry Smith PetscFunctionReturn(0); 12419b94acceSBarry Smith } 12429b94acceSBarry Smith 12435615d1e5SSatish Balay #undef __FUNC__ 1244d4bb536fSBarry Smith #define __FUNC__ "SNESGetHessian" 124562fef451SLois Curfman McInnes /*@ 124662fef451SLois Curfman McInnes SNESGetHessian - Returns the Hessian matrix and optionally the user 124762fef451SLois Curfman McInnes provided context for evaluating the Hessian. 124862fef451SLois Curfman McInnes 1249c7afd0dbSLois Curfman McInnes Not Collective, but Mat object is parallel if SNES object is parallel 1250c7afd0dbSLois Curfman McInnes 125162fef451SLois Curfman McInnes Input Parameter: 125262fef451SLois Curfman McInnes . snes - the nonlinear solver context 125362fef451SLois Curfman McInnes 125462fef451SLois Curfman McInnes Output Parameters: 1255c7afd0dbSLois Curfman McInnes + A - location to stash Hessian matrix (or PETSC_NULL) 125662fef451SLois Curfman McInnes . B - location to stash preconditioner matrix (or PETSC_NULL) 1257c7afd0dbSLois Curfman McInnes - ctx - location to stash Hessian ctx (or PETSC_NULL) 1258fee21e36SBarry Smith 125936851e7fSLois Curfman McInnes Level: advanced 126036851e7fSLois Curfman McInnes 126162fef451SLois Curfman McInnes .seealso: SNESSetHessian(), SNESComputeHessian() 1262c7afd0dbSLois Curfman McInnes 1263c7afd0dbSLois Curfman McInnes .keywords: SNES, get, Hessian 126462fef451SLois Curfman McInnes @*/ 126562fef451SLois Curfman McInnes int SNESGetHessian(SNES snes,Mat *A,Mat *B, void **ctx) 126662fef451SLois Curfman McInnes { 12673a40ed3dSBarry Smith PetscFunctionBegin; 12683a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION){ 1269a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 12703a40ed3dSBarry Smith } 127162fef451SLois Curfman McInnes if (A) *A = snes->jacobian; 127262fef451SLois Curfman McInnes if (B) *B = snes->jacobian_pre; 127362fef451SLois Curfman McInnes if (ctx) *ctx = snes->jacP; 12743a40ed3dSBarry Smith PetscFunctionReturn(0); 127562fef451SLois Curfman McInnes } 127662fef451SLois Curfman McInnes 12779b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */ 12789b94acceSBarry Smith 12795615d1e5SSatish Balay #undef __FUNC__ 12805615d1e5SSatish Balay #define __FUNC__ "SNESSetUp" 12819b94acceSBarry Smith /*@ 12829b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 1283272ac6f2SLois Curfman McInnes of a nonlinear solver. 12849b94acceSBarry Smith 1285fee21e36SBarry Smith Collective on SNES 1286fee21e36SBarry Smith 1287c7afd0dbSLois Curfman McInnes Input Parameters: 1288c7afd0dbSLois Curfman McInnes + snes - the SNES context 1289c7afd0dbSLois Curfman McInnes - x - the solution vector 1290c7afd0dbSLois Curfman McInnes 1291272ac6f2SLois Curfman McInnes Notes: 1292272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 1293272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 1294272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 1295272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 1296272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 1297272ac6f2SLois Curfman McInnes 129836851e7fSLois Curfman McInnes Level: advanced 129936851e7fSLois Curfman McInnes 13009b94acceSBarry Smith .keywords: SNES, nonlinear, setup 13019b94acceSBarry Smith 13029b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 13039b94acceSBarry Smith @*/ 13048ddd3da0SLois Curfman McInnes int SNESSetUp(SNES snes,Vec x) 13059b94acceSBarry Smith { 1306272ac6f2SLois Curfman McInnes int ierr, flg; 13073a40ed3dSBarry Smith 13083a40ed3dSBarry Smith PetscFunctionBegin; 130977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 131077c4ece6SBarry Smith PetscValidHeaderSpecific(x,VEC_COOKIE); 13118ddd3da0SLois Curfman McInnes snes->vec_sol = snes->vec_sol_always = x; 13129b94acceSBarry Smith 1313c1f60f51SBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_mf_operator", &flg); CHKERRQ(ierr); 1314c1f60f51SBarry Smith /* 1315c1f60f51SBarry Smith This version replaces the user provided Jacobian matrix with a 1316dfa02198SLois Curfman McInnes matrix-free version but still employs the user-provided preconditioner matrix 1317c1f60f51SBarry Smith */ 1318c1f60f51SBarry Smith if (flg) { 1319c1f60f51SBarry Smith Mat J; 1320e24b481bSBarry Smith ierr = MatCreateSNESFDMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1321c1f60f51SBarry Smith PLogObjectParent(snes,J); 1322c1f60f51SBarry Smith snes->mfshell = J; 1323c1f60f51SBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1324c1f60f51SBarry Smith snes->jacobian = J; 1325c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Jacobian routines\n"); 1326d64ed03dSBarry Smith } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 1327c1f60f51SBarry Smith snes->jacobian = J; 1328c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Hessian routines\n"); 1329d4bb536fSBarry Smith } else { 1330a8c6a408SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"Method class doesn't support matrix-free operator option"); 1331d4bb536fSBarry Smith } 1332e24b481bSBarry Smith ierr = MatSNESFDMFSetFromOptions(J);CHKERRQ(ierr); 1333c1f60f51SBarry Smith } 1334272ac6f2SLois Curfman McInnes ierr = OptionsHasName(snes->prefix,"-snes_mf", &flg); CHKERRQ(ierr); 1335c1f60f51SBarry Smith /* 1336dfa02198SLois Curfman McInnes This version replaces both the user-provided Jacobian and the user- 1337c1f60f51SBarry Smith provided preconditioner matrix with the default matrix free version. 1338c1f60f51SBarry Smith */ 133931615d04SLois Curfman McInnes if (flg) { 1340272ac6f2SLois Curfman McInnes Mat J; 1341e24b481bSBarry Smith ierr = MatCreateSNESFDMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1342272ac6f2SLois Curfman McInnes PLogObjectParent(snes,J); 1343272ac6f2SLois Curfman McInnes snes->mfshell = J; 134483e56afdSLois Curfman McInnes if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 134583e56afdSLois Curfman McInnes ierr = SNESSetJacobian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 134694a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Jacobian routines\n"); 1347d64ed03dSBarry Smith } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 134883e56afdSLois Curfman McInnes ierr = SNESSetHessian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 134994a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Hessian routines\n"); 1350d4bb536fSBarry Smith } else { 1351a8c6a408SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"Method class doesn't support matrix-free option"); 1352d4bb536fSBarry Smith } 1353e24b481bSBarry Smith ierr = MatSNESFDMFSetFromOptions(J);CHKERRQ(ierr); 1354272ac6f2SLois Curfman McInnes } 13559b94acceSBarry Smith if ((snes->method_class == SNES_NONLINEAR_EQUATIONS)) { 1356a8c6a408SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetFunction() first"); 1357a8c6a408SBarry Smith if (!snes->computefunction) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetFunction() first"); 1358a8c6a408SBarry Smith if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetJacobian() first"); 1359a8c6a408SBarry Smith if (snes->vec_func == snes->vec_sol) { 1360a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_IDN,0,"Solution vector cannot be function vector"); 1361a8c6a408SBarry Smith } 1362a703fe33SLois Curfman McInnes 1363a703fe33SLois Curfman McInnes /* Set the KSP stopping criterion to use the Eisenstat-Walker method */ 136482bf6240SBarry Smith if (snes->ksp_ewconv && PetscStrcmp(snes->type_name,SNES_EQ_TR)) { 1365a703fe33SLois Curfman McInnes SLES sles; KSP ksp; 1366a703fe33SLois Curfman McInnes ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 1367a703fe33SLois Curfman McInnes ierr = SLESGetKSP(sles,&ksp); CHKERRQ(ierr); 1368a703fe33SLois Curfman McInnes ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private, 1369a703fe33SLois Curfman McInnes (void *)snes); CHKERRQ(ierr); 1370a703fe33SLois Curfman McInnes } 13719b94acceSBarry Smith } else if ((snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION)) { 1372a8c6a408SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetGradient() first"); 1373a8c6a408SBarry Smith if (!snes->computefunction) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetGradient() first"); 1374a8c6a408SBarry Smith if (!snes->computeumfunction) { 1375a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetMinimizationFunction() first"); 1376a8c6a408SBarry Smith } 1377a8c6a408SBarry Smith if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetHessian() first"); 1378d4bb536fSBarry Smith } else { 1379a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Unknown method class"); 1380d4bb536fSBarry Smith } 1381a703fe33SLois Curfman McInnes if (snes->setup) {ierr = (*snes->setup)(snes); CHKERRQ(ierr);} 138282bf6240SBarry Smith snes->setupcalled = 1; 13833a40ed3dSBarry Smith PetscFunctionReturn(0); 13849b94acceSBarry Smith } 13859b94acceSBarry Smith 13865615d1e5SSatish Balay #undef __FUNC__ 1387d4bb536fSBarry Smith #define __FUNC__ "SNESDestroy" 13889b94acceSBarry Smith /*@C 13899b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 13909b94acceSBarry Smith with SNESCreate(). 13919b94acceSBarry Smith 1392c7afd0dbSLois Curfman McInnes Collective on SNES 1393c7afd0dbSLois Curfman McInnes 13949b94acceSBarry Smith Input Parameter: 13959b94acceSBarry Smith . snes - the SNES context 13969b94acceSBarry Smith 139736851e7fSLois Curfman McInnes Level: beginner 139836851e7fSLois Curfman McInnes 13999b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 14009b94acceSBarry Smith 140163a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 14029b94acceSBarry Smith @*/ 14039b94acceSBarry Smith int SNESDestroy(SNES snes) 14049b94acceSBarry Smith { 14059b94acceSBarry Smith int ierr; 14063a40ed3dSBarry Smith 14073a40ed3dSBarry Smith PetscFunctionBegin; 140877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 14093a40ed3dSBarry Smith if (--snes->refct > 0) PetscFunctionReturn(0); 1410d4bb536fSBarry Smith 1411e1311b90SBarry Smith if (snes->destroy) {ierr = (*(snes)->destroy)(snes); CHKERRQ(ierr);} 14120452661fSBarry Smith if (snes->kspconvctx) PetscFree(snes->kspconvctx); 1413522c5e43SBarry Smith if (snes->mfshell) {ierr = MatDestroy(snes->mfshell);CHKERRQ(ierr);} 14149b94acceSBarry Smith ierr = SLESDestroy(snes->sles); CHKERRQ(ierr); 1415522c5e43SBarry Smith if (snes->xmonitor) {ierr = SNESLGMonitorDestroy(snes->xmonitor);CHKERRQ(ierr);} 1416522c5e43SBarry Smith if (snes->vwork) {ierr = VecDestroyVecs(snes->vwork,snes->nvwork);CHKERRQ(ierr);} 14179b94acceSBarry Smith PLogObjectDestroy((PetscObject)snes); 14180452661fSBarry Smith PetscHeaderDestroy((PetscObject)snes); 14193a40ed3dSBarry Smith PetscFunctionReturn(0); 14209b94acceSBarry Smith } 14219b94acceSBarry Smith 14229b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 14239b94acceSBarry Smith 14245615d1e5SSatish Balay #undef __FUNC__ 14255615d1e5SSatish Balay #define __FUNC__ "SNESSetTolerances" 14269b94acceSBarry Smith /*@ 1427d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 14289b94acceSBarry Smith 1429c7afd0dbSLois Curfman McInnes Collective on SNES 1430c7afd0dbSLois Curfman McInnes 14319b94acceSBarry Smith Input Parameters: 1432c7afd0dbSLois Curfman McInnes + snes - the SNES context 143333174efeSLois Curfman McInnes . atol - absolute convergence tolerance 143433174efeSLois Curfman McInnes . rtol - relative convergence tolerance 143533174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 143633174efeSLois Curfman McInnes of the change in the solution between steps 143733174efeSLois Curfman McInnes . maxit - maximum number of iterations 1438c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1439fee21e36SBarry Smith 144033174efeSLois Curfman McInnes Options Database Keys: 1441c7afd0dbSLois Curfman McInnes + -snes_atol <atol> - Sets atol 1442c7afd0dbSLois Curfman McInnes . -snes_rtol <rtol> - Sets rtol 1443c7afd0dbSLois Curfman McInnes . -snes_stol <stol> - Sets stol 1444c7afd0dbSLois Curfman McInnes . -snes_max_it <maxit> - Sets maxit 1445c7afd0dbSLois Curfman McInnes - -snes_max_funcs <maxf> - Sets maxf 14469b94acceSBarry Smith 1447d7a720efSLois Curfman McInnes Notes: 14489b94acceSBarry Smith The default maximum number of iterations is 50. 14499b94acceSBarry Smith The default maximum number of function evaluations is 1000. 14509b94acceSBarry Smith 145136851e7fSLois Curfman McInnes Level: intermediate 145236851e7fSLois Curfman McInnes 145333174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances 14549b94acceSBarry Smith 1455d7a720efSLois Curfman McInnes .seealso: SNESSetTrustRegionTolerance(), SNESSetMinimizationFunctionTolerance() 14569b94acceSBarry Smith @*/ 1457d7a720efSLois Curfman McInnes int SNESSetTolerances(SNES snes,double atol,double rtol,double stol,int maxit,int maxf) 14589b94acceSBarry Smith { 14593a40ed3dSBarry Smith PetscFunctionBegin; 146077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1461d7a720efSLois Curfman McInnes if (atol != PETSC_DEFAULT) snes->atol = atol; 1462d7a720efSLois Curfman McInnes if (rtol != PETSC_DEFAULT) snes->rtol = rtol; 1463d7a720efSLois Curfman McInnes if (stol != PETSC_DEFAULT) snes->xtol = stol; 1464d7a720efSLois Curfman McInnes if (maxit != PETSC_DEFAULT) snes->max_its = maxit; 1465d7a720efSLois Curfman McInnes if (maxf != PETSC_DEFAULT) snes->max_funcs = maxf; 14663a40ed3dSBarry Smith PetscFunctionReturn(0); 14679b94acceSBarry Smith } 14689b94acceSBarry Smith 14695615d1e5SSatish Balay #undef __FUNC__ 14705615d1e5SSatish Balay #define __FUNC__ "SNESGetTolerances" 14719b94acceSBarry Smith /*@ 147233174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 147333174efeSLois Curfman McInnes 1474c7afd0dbSLois Curfman McInnes Not Collective 1475c7afd0dbSLois Curfman McInnes 147633174efeSLois Curfman McInnes Input Parameters: 1477c7afd0dbSLois Curfman McInnes + snes - the SNES context 147833174efeSLois Curfman McInnes . atol - absolute convergence tolerance 147933174efeSLois Curfman McInnes . rtol - relative convergence tolerance 148033174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 148133174efeSLois Curfman McInnes of the change in the solution between steps 148233174efeSLois Curfman McInnes . maxit - maximum number of iterations 1483c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1484fee21e36SBarry Smith 148533174efeSLois Curfman McInnes Notes: 148633174efeSLois Curfman McInnes The user can specify PETSC_NULL for any parameter that is not needed. 148733174efeSLois Curfman McInnes 148836851e7fSLois Curfman McInnes Level: intermediate 148936851e7fSLois Curfman McInnes 149033174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances 149133174efeSLois Curfman McInnes 149233174efeSLois Curfman McInnes .seealso: SNESSetTolerances() 149333174efeSLois Curfman McInnes @*/ 149433174efeSLois Curfman McInnes int SNESGetTolerances(SNES snes,double *atol,double *rtol,double *stol,int *maxit,int *maxf) 149533174efeSLois Curfman McInnes { 14963a40ed3dSBarry Smith PetscFunctionBegin; 149733174efeSLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 149833174efeSLois Curfman McInnes if (atol) *atol = snes->atol; 149933174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 150033174efeSLois Curfman McInnes if (stol) *stol = snes->xtol; 150133174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 150233174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 15033a40ed3dSBarry Smith PetscFunctionReturn(0); 150433174efeSLois Curfman McInnes } 150533174efeSLois Curfman McInnes 15065615d1e5SSatish Balay #undef __FUNC__ 15075615d1e5SSatish Balay #define __FUNC__ "SNESSetTrustRegionTolerance" 150833174efeSLois Curfman McInnes /*@ 15099b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 15109b94acceSBarry Smith 1511fee21e36SBarry Smith Collective on SNES 1512fee21e36SBarry Smith 1513c7afd0dbSLois Curfman McInnes Input Parameters: 1514c7afd0dbSLois Curfman McInnes + snes - the SNES context 1515c7afd0dbSLois Curfman McInnes - tol - tolerance 1516c7afd0dbSLois Curfman McInnes 15179b94acceSBarry Smith Options Database Key: 1518c7afd0dbSLois Curfman McInnes . -snes_trtol <tol> - Sets tol 15199b94acceSBarry Smith 152036851e7fSLois Curfman McInnes Level: intermediate 152136851e7fSLois Curfman McInnes 15229b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 15239b94acceSBarry Smith 1524d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetMinimizationFunctionTolerance() 15259b94acceSBarry Smith @*/ 15269b94acceSBarry Smith int SNESSetTrustRegionTolerance(SNES snes,double tol) 15279b94acceSBarry Smith { 15283a40ed3dSBarry Smith PetscFunctionBegin; 152977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15309b94acceSBarry Smith snes->deltatol = tol; 15313a40ed3dSBarry Smith PetscFunctionReturn(0); 15329b94acceSBarry Smith } 15339b94acceSBarry Smith 15345615d1e5SSatish Balay #undef __FUNC__ 15355615d1e5SSatish Balay #define __FUNC__ "SNESSetMinimizationFunctionTolerance" 15369b94acceSBarry Smith /*@ 153777c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance - Sets the minimum allowable function tolerance 15389b94acceSBarry Smith for unconstrained minimization solvers. 15399b94acceSBarry Smith 1540fee21e36SBarry Smith Collective on SNES 1541fee21e36SBarry Smith 1542c7afd0dbSLois Curfman McInnes Input Parameters: 1543c7afd0dbSLois Curfman McInnes + snes - the SNES context 1544c7afd0dbSLois Curfman McInnes - ftol - minimum function tolerance 1545c7afd0dbSLois Curfman McInnes 15469b94acceSBarry Smith Options Database Key: 1547c7afd0dbSLois Curfman McInnes . -snes_fmin <ftol> - Sets ftol 15489b94acceSBarry Smith 15499b94acceSBarry Smith Note: 155077c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance() is valid for SNES_UNCONSTRAINED_MINIMIZATION 15519b94acceSBarry Smith methods only. 15529b94acceSBarry Smith 155336851e7fSLois Curfman McInnes Level: intermediate 155436851e7fSLois Curfman McInnes 15559b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimum, convergence, function, tolerance 15569b94acceSBarry Smith 1557d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetTrustRegionTolerance() 15589b94acceSBarry Smith @*/ 155977c4ece6SBarry Smith int SNESSetMinimizationFunctionTolerance(SNES snes,double ftol) 15609b94acceSBarry Smith { 15613a40ed3dSBarry Smith PetscFunctionBegin; 156277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15639b94acceSBarry Smith snes->fmin = ftol; 15643a40ed3dSBarry Smith PetscFunctionReturn(0); 15659b94acceSBarry Smith } 15669b94acceSBarry Smith 1567ce1608b8SBarry Smith #undef __FUNC__ 1568ce1608b8SBarry Smith #define __FUNC__ "SNESLGMonitor" 1569ce1608b8SBarry Smith int SNESLGMonitor(SNES snes,int it,double norm,void *ctx) 1570ce1608b8SBarry Smith { 1571ce1608b8SBarry Smith int ierr; 1572ce1608b8SBarry Smith 1573ce1608b8SBarry Smith PetscFunctionBegin; 1574ce1608b8SBarry Smith ierr = KSPLGMonitor((KSP)snes,it,norm,ctx);CHKERRQ(ierr); 1575ce1608b8SBarry Smith PetscFunctionReturn(0); 1576ce1608b8SBarry Smith } 1577ce1608b8SBarry Smith 15789b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 15799b94acceSBarry Smith 15805615d1e5SSatish Balay #undef __FUNC__ 1581d4bb536fSBarry Smith #define __FUNC__ "SNESSetMonitor" 15829b94acceSBarry Smith /*@C 15835cd90555SBarry Smith SNESSetMonitor - Sets an ADDITIONAL function that is to be used at every 15849b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 15859b94acceSBarry Smith progress. 15869b94acceSBarry Smith 1587fee21e36SBarry Smith Collective on SNES 1588fee21e36SBarry Smith 1589c7afd0dbSLois Curfman McInnes Input Parameters: 1590c7afd0dbSLois Curfman McInnes + snes - the SNES context 1591c7afd0dbSLois Curfman McInnes . func - monitoring routine 1592c7afd0dbSLois Curfman McInnes - mctx - [optional] user-defined context for private data for the 1593c7afd0dbSLois Curfman McInnes monitor routine (may be PETSC_NULL) 15949b94acceSBarry Smith 1595c7afd0dbSLois Curfman McInnes Calling sequence of func: 159640a0c1c6SLois Curfman McInnes $ int func(SNES snes,int its, double norm,void *mctx) 1597c7afd0dbSLois Curfman McInnes 1598c7afd0dbSLois Curfman McInnes + snes - the SNES context 1599c7afd0dbSLois Curfman McInnes . its - iteration number 1600c7afd0dbSLois Curfman McInnes . norm - 2-norm function value (may be estimated) 1601c7afd0dbSLois Curfman McInnes for SNES_NONLINEAR_EQUATIONS methods 160240a0c1c6SLois Curfman McInnes . norm - 2-norm gradient value (may be estimated) 1603c7afd0dbSLois Curfman McInnes for SNES_UNCONSTRAINED_MINIMIZATION methods 160440a0c1c6SLois Curfman McInnes - mctx - [optional] monitoring context 16059b94acceSBarry Smith 16069665c990SLois Curfman McInnes Options Database Keys: 1607c7afd0dbSLois Curfman McInnes + -snes_monitor - sets SNESDefaultMonitor() 1608c7afd0dbSLois Curfman McInnes . -snes_xmonitor - sets line graph monitor, 1609c7afd0dbSLois Curfman McInnes uses SNESLGMonitorCreate() 1610c7afd0dbSLois Curfman McInnes _ -snes_cancelmonitors - cancels all monitors that have 1611c7afd0dbSLois Curfman McInnes been hardwired into a code by 1612c7afd0dbSLois Curfman McInnes calls to SNESSetMonitor(), but 1613c7afd0dbSLois Curfman McInnes does not cancel those set via 1614c7afd0dbSLois Curfman McInnes the options database. 16159665c990SLois Curfman McInnes 1616639f9d9dSBarry Smith Notes: 16176bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 16186bc08f3fSLois Curfman McInnes SNESSetMonitor() multiple times; all will be called in the 16196bc08f3fSLois Curfman McInnes order in which they were set. 1620639f9d9dSBarry Smith 162136851e7fSLois Curfman McInnes Level: intermediate 162236851e7fSLois Curfman McInnes 16239b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 16249b94acceSBarry Smith 16255cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESClearMonitor() 16269b94acceSBarry Smith @*/ 162774679c65SBarry Smith int SNESSetMonitor( SNES snes, int (*func)(SNES,int,double,void*),void *mctx ) 16289b94acceSBarry Smith { 16293a40ed3dSBarry Smith PetscFunctionBegin; 1630639f9d9dSBarry Smith if (snes->numbermonitors >= MAXSNESMONITORS) { 1631a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Too many monitors set"); 1632639f9d9dSBarry Smith } 1633639f9d9dSBarry Smith 1634639f9d9dSBarry Smith snes->monitor[snes->numbermonitors] = func; 1635639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 16363a40ed3dSBarry Smith PetscFunctionReturn(0); 16379b94acceSBarry Smith } 16389b94acceSBarry Smith 16395615d1e5SSatish Balay #undef __FUNC__ 16405cd90555SBarry Smith #define __FUNC__ "SNESClearMonitor" 16415cd90555SBarry Smith /*@C 16425cd90555SBarry Smith SNESClearMonitor - Clears all the monitor functions for a SNES object. 16435cd90555SBarry Smith 1644c7afd0dbSLois Curfman McInnes Collective on SNES 1645c7afd0dbSLois Curfman McInnes 16465cd90555SBarry Smith Input Parameters: 16475cd90555SBarry Smith . snes - the SNES context 16485cd90555SBarry Smith 16495cd90555SBarry Smith Options Database: 1650c7afd0dbSLois Curfman McInnes . -snes_cancelmonitors - cancels all monitors that have been hardwired 1651c7afd0dbSLois Curfman McInnes into a code by calls to SNESSetMonitor(), but does not cancel those 1652c7afd0dbSLois Curfman McInnes set via the options database 16535cd90555SBarry Smith 16545cd90555SBarry Smith Notes: 16555cd90555SBarry Smith There is no way to clear one specific monitor from a SNES object. 16565cd90555SBarry Smith 165736851e7fSLois Curfman McInnes Level: intermediate 165836851e7fSLois Curfman McInnes 16595cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor 16605cd90555SBarry Smith 16615cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESSetMonitor() 16625cd90555SBarry Smith @*/ 16635cd90555SBarry Smith int SNESClearMonitor( SNES snes ) 16645cd90555SBarry Smith { 16655cd90555SBarry Smith PetscFunctionBegin; 16665cd90555SBarry Smith snes->numbermonitors = 0; 16675cd90555SBarry Smith PetscFunctionReturn(0); 16685cd90555SBarry Smith } 16695cd90555SBarry Smith 16705cd90555SBarry Smith #undef __FUNC__ 1671d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceTest" 16729b94acceSBarry Smith /*@C 16739b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 16749b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 16759b94acceSBarry Smith 1676fee21e36SBarry Smith Collective on SNES 1677fee21e36SBarry Smith 1678c7afd0dbSLois Curfman McInnes Input Parameters: 1679c7afd0dbSLois Curfman McInnes + snes - the SNES context 1680c7afd0dbSLois Curfman McInnes . func - routine to test for convergence 1681c7afd0dbSLois Curfman McInnes - cctx - [optional] context for private data for the convergence routine 1682c7afd0dbSLois Curfman McInnes (may be PETSC_NULL) 16839b94acceSBarry Smith 1684c7afd0dbSLois Curfman McInnes Calling sequence of func: 16858d76a1e5SLois Curfman McInnes $ int func (SNES snes,double xnorm,double gnorm,double f,void *cctx) 1686c7afd0dbSLois Curfman McInnes 1687c7afd0dbSLois Curfman McInnes + snes - the SNES context 1688c7afd0dbSLois Curfman McInnes . cctx - [optional] convergence context 1689c7afd0dbSLois Curfman McInnes . xnorm - 2-norm of current iterate 1690c7afd0dbSLois Curfman McInnes . gnorm - 2-norm of current step (SNES_NONLINEAR_EQUATIONS methods) 1691c7afd0dbSLois Curfman McInnes . f - 2-norm of function (SNES_NONLINEAR_EQUATIONS methods) 1692c7afd0dbSLois Curfman McInnes . gnorm - 2-norm of current gradient (SNES_UNCONSTRAINED_MINIMIZATION methods) 1693c7afd0dbSLois Curfman McInnes - f - function value (SNES_UNCONSTRAINED_MINIMIZATION methods) 16949b94acceSBarry Smith 169536851e7fSLois Curfman McInnes Level: advanced 169636851e7fSLois Curfman McInnes 16979b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 16989b94acceSBarry Smith 169940191667SLois Curfman McInnes .seealso: SNESConverged_EQ_LS(), SNESConverged_EQ_TR(), 170040191667SLois Curfman McInnes SNESConverged_UM_LS(), SNESConverged_UM_TR() 17019b94acceSBarry Smith @*/ 170274679c65SBarry Smith int SNESSetConvergenceTest(SNES snes,int (*func)(SNES,double,double,double,void*),void *cctx) 17039b94acceSBarry Smith { 17043a40ed3dSBarry Smith PetscFunctionBegin; 17059b94acceSBarry Smith (snes)->converged = func; 17069b94acceSBarry Smith (snes)->cnvP = cctx; 17073a40ed3dSBarry Smith PetscFunctionReturn(0); 17089b94acceSBarry Smith } 17099b94acceSBarry Smith 17105615d1e5SSatish Balay #undef __FUNC__ 1711d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceHistory" 1712c9005455SLois Curfman McInnes /*@ 1713c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 1714c9005455SLois Curfman McInnes 1715fee21e36SBarry Smith Collective on SNES 1716fee21e36SBarry Smith 1717c7afd0dbSLois Curfman McInnes Input Parameters: 1718c7afd0dbSLois Curfman McInnes + snes - iterative context obtained from SNESCreate() 1719c7afd0dbSLois Curfman McInnes . a - array to hold history 1720758f92a0SBarry Smith . its - integer array holds the number of linear iterations (or 1721758f92a0SBarry Smith negative if not converged) for each solve. 1722758f92a0SBarry Smith . na - size of a and its 1723758f92a0SBarry Smith - reset - PETSC_TRUTH indicates each new nonlinear solve resets the history counter to zero, 1724758f92a0SBarry Smith else it continues storing new values for new nonlinear solves after the old ones 1725c7afd0dbSLois Curfman McInnes 1726c9005455SLois Curfman McInnes Notes: 1727c9005455SLois Curfman McInnes If set, this array will contain the function norms (for 1728c9005455SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods) or gradient norms 1729c9005455SLois Curfman McInnes (for SNES_UNCONSTRAINED_MINIMIZATION methods) computed 1730c9005455SLois Curfman McInnes at each step. 1731c9005455SLois Curfman McInnes 1732c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 1733c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 1734c9005455SLois Curfman McInnes during the section of code that is being timed. 1735c9005455SLois Curfman McInnes 173636851e7fSLois Curfman McInnes Level: intermediate 173736851e7fSLois Curfman McInnes 1738c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history 1739758f92a0SBarry Smith 1740758f92a0SBarry Smith .seealso: SNESGetConvergencHistory() 1741758f92a0SBarry Smith 1742c9005455SLois Curfman McInnes @*/ 1743758f92a0SBarry Smith int SNESSetConvergenceHistory(SNES snes, double *a, int *its,int na,PetscTruth reset) 1744c9005455SLois Curfman McInnes { 17453a40ed3dSBarry Smith PetscFunctionBegin; 1746c9005455SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 1747c9005455SLois Curfman McInnes if (na) PetscValidScalarPointer(a); 1748c9005455SLois Curfman McInnes snes->conv_hist = a; 1749758f92a0SBarry Smith snes->conv_hist_its = its; 1750758f92a0SBarry Smith snes->conv_hist_max = na; 1751758f92a0SBarry Smith snes->conv_hist_reset = reset; 1752758f92a0SBarry Smith PetscFunctionReturn(0); 1753758f92a0SBarry Smith } 1754758f92a0SBarry Smith 1755758f92a0SBarry Smith #undef __FUNC__ 1756758f92a0SBarry Smith #define __FUNC__ "SNESGetConvergenceHistory" 1757*0c4c9dddSBarry Smith /*@C 1758758f92a0SBarry Smith SNESGetConvergenceHistory - Gets the array used to hold the convergence history. 1759758f92a0SBarry Smith 1760758f92a0SBarry Smith Collective on SNES 1761758f92a0SBarry Smith 1762758f92a0SBarry Smith Input Parameter: 1763758f92a0SBarry Smith . snes - iterative context obtained from SNESCreate() 1764758f92a0SBarry Smith 1765758f92a0SBarry Smith Output Parameters: 1766758f92a0SBarry Smith . a - array to hold history 1767758f92a0SBarry Smith . its - integer array holds the number of linear iterations (or 1768758f92a0SBarry Smith negative if not converged) for each solve. 1769758f92a0SBarry Smith - na - size of a and its 1770758f92a0SBarry Smith 1771758f92a0SBarry Smith Notes: 1772758f92a0SBarry Smith The calling sequence for this routine in Fortran is 1773758f92a0SBarry Smith $ call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr) 1774758f92a0SBarry Smith 1775758f92a0SBarry Smith This routine is useful, e.g., when running a code for purposes 1776758f92a0SBarry Smith of accurate performance monitoring, when no I/O should be done 1777758f92a0SBarry Smith during the section of code that is being timed. 1778758f92a0SBarry Smith 1779758f92a0SBarry Smith Level: intermediate 1780758f92a0SBarry Smith 1781758f92a0SBarry Smith .keywords: SNES, get, convergence, history 1782758f92a0SBarry Smith 1783758f92a0SBarry Smith .seealso: SNESSetConvergencHistory() 1784758f92a0SBarry Smith 1785758f92a0SBarry Smith @*/ 1786758f92a0SBarry Smith int SNESGetConvergenceHistory(SNES snes, double **a, int **its,int *na) 1787758f92a0SBarry Smith { 1788758f92a0SBarry Smith PetscFunctionBegin; 1789758f92a0SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1790758f92a0SBarry Smith if (a) *a = snes->conv_hist; 1791758f92a0SBarry Smith if (its) *its = snes->conv_hist_its; 1792758f92a0SBarry Smith if (na) *na = snes->conv_hist_len; 17933a40ed3dSBarry Smith PetscFunctionReturn(0); 1794c9005455SLois Curfman McInnes } 1795c9005455SLois Curfman McInnes 1796c9005455SLois Curfman McInnes #undef __FUNC__ 17975615d1e5SSatish Balay #define __FUNC__ "SNESScaleStep_Private" 17989b94acceSBarry Smith /* 17999b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 18009b94acceSBarry Smith positive parameter delta. 18019b94acceSBarry Smith 18029b94acceSBarry Smith Input Parameters: 1803c7afd0dbSLois Curfman McInnes + snes - the SNES context 18049b94acceSBarry Smith . y - approximate solution of linear system 18059b94acceSBarry Smith . fnorm - 2-norm of current function 1806c7afd0dbSLois Curfman McInnes - delta - trust region size 18079b94acceSBarry Smith 18089b94acceSBarry Smith Output Parameters: 1809c7afd0dbSLois Curfman McInnes + gpnorm - predicted function norm at the new point, assuming local 18109b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 18119b94acceSBarry Smith region, and exceeds zero otherwise. 1812c7afd0dbSLois Curfman McInnes - ynorm - 2-norm of the step 18139b94acceSBarry Smith 18149b94acceSBarry Smith Note: 181540191667SLois Curfman McInnes For non-trust region methods such as SNES_EQ_LS, the parameter delta 18169b94acceSBarry Smith is set to be the maximum allowable step size. 18179b94acceSBarry Smith 18189b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 18199b94acceSBarry Smith */ 18209b94acceSBarry Smith int SNESScaleStep_Private(SNES snes,Vec y,double *fnorm,double *delta, 18219b94acceSBarry Smith double *gpnorm,double *ynorm) 18229b94acceSBarry Smith { 18239b94acceSBarry Smith double norm; 18249b94acceSBarry Smith Scalar cnorm; 18253a40ed3dSBarry Smith int ierr; 18263a40ed3dSBarry Smith 18273a40ed3dSBarry Smith PetscFunctionBegin; 18283a40ed3dSBarry Smith ierr = VecNorm(y,NORM_2, &norm );CHKERRQ(ierr); 18299b94acceSBarry Smith if (norm > *delta) { 18309b94acceSBarry Smith norm = *delta/norm; 18319b94acceSBarry Smith *gpnorm = (1.0 - norm)*(*fnorm); 18329b94acceSBarry Smith cnorm = norm; 18339b94acceSBarry Smith VecScale( &cnorm, y ); 18349b94acceSBarry Smith *ynorm = *delta; 18359b94acceSBarry Smith } else { 18369b94acceSBarry Smith *gpnorm = 0.0; 18379b94acceSBarry Smith *ynorm = norm; 18389b94acceSBarry Smith } 18393a40ed3dSBarry Smith PetscFunctionReturn(0); 18409b94acceSBarry Smith } 18419b94acceSBarry Smith 18425615d1e5SSatish Balay #undef __FUNC__ 18435615d1e5SSatish Balay #define __FUNC__ "SNESSolve" 18449b94acceSBarry Smith /*@ 18459b94acceSBarry Smith SNESSolve - Solves a nonlinear system. Call SNESSolve after calling 184663a78c88SLois Curfman McInnes SNESCreate() and optional routines of the form SNESSetXXX(). 18479b94acceSBarry Smith 1848c7afd0dbSLois Curfman McInnes Collective on SNES 1849c7afd0dbSLois Curfman McInnes 1850b2002411SLois Curfman McInnes Input Parameters: 1851c7afd0dbSLois Curfman McInnes + snes - the SNES context 1852c7afd0dbSLois Curfman McInnes - x - the solution vector 18539b94acceSBarry Smith 18549b94acceSBarry Smith Output Parameter: 1855b2002411SLois Curfman McInnes . its - number of iterations until termination 18569b94acceSBarry Smith 1857b2002411SLois Curfman McInnes Notes: 18588ddd3da0SLois Curfman McInnes The user should initialize the vector, x, with the initial guess 18598ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 18608ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 18618ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 18628ddd3da0SLois Curfman McInnes 186336851e7fSLois Curfman McInnes Level: beginner 186436851e7fSLois Curfman McInnes 18659b94acceSBarry Smith .keywords: SNES, nonlinear, solve 18669b94acceSBarry Smith 186763a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESDestroy() 18689b94acceSBarry Smith @*/ 18698ddd3da0SLois Curfman McInnes int SNESSolve(SNES snes,Vec x,int *its) 18709b94acceSBarry Smith { 18713c7409f5SSatish Balay int ierr, flg; 1872052efed2SBarry Smith 18733a40ed3dSBarry Smith PetscFunctionBegin; 187477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 187574679c65SBarry Smith PetscValidIntPointer(its); 187682bf6240SBarry Smith if (!snes->setupcalled) {ierr = SNESSetUp(snes,x); CHKERRQ(ierr);} 1877c4fc05e7SBarry Smith else {snes->vec_sol = snes->vec_sol_always = x;} 1878758f92a0SBarry Smith if (snes->conv_hist_reset == PETSC_TRUE) snes->conv_hist_len = 0; 18799b94acceSBarry Smith PLogEventBegin(SNES_Solve,snes,0,0,0); 1880c96a6f78SLois Curfman McInnes snes->nfuncs = 0; snes->linear_its = 0; snes->nfailures = 0; 18819b94acceSBarry Smith ierr = (*(snes)->solve)(snes,its); CHKERRQ(ierr); 18829b94acceSBarry Smith PLogEventEnd(SNES_Solve,snes,0,0,0); 18833c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-snes_view", &flg); CHKERRQ(ierr); 18846d4a8577SBarry Smith if (flg) { ierr = SNESView(snes,VIEWER_STDOUT_WORLD); CHKERRQ(ierr); } 18853a40ed3dSBarry Smith PetscFunctionReturn(0); 18869b94acceSBarry Smith } 18879b94acceSBarry Smith 18889b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 18899b94acceSBarry Smith 18905615d1e5SSatish Balay #undef __FUNC__ 18915615d1e5SSatish Balay #define __FUNC__ "SNESSetType" 189282bf6240SBarry Smith /*@C 18934b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 18949b94acceSBarry Smith 1895fee21e36SBarry Smith Collective on SNES 1896fee21e36SBarry Smith 1897c7afd0dbSLois Curfman McInnes Input Parameters: 1898c7afd0dbSLois Curfman McInnes + snes - the SNES context 1899c7afd0dbSLois Curfman McInnes - method - a known method 1900c7afd0dbSLois Curfman McInnes 1901c7afd0dbSLois Curfman McInnes Options Database Key: 1902c7afd0dbSLois Curfman McInnes . -snes_type <method> - Sets the method; use -help for a list 1903c7afd0dbSLois Curfman McInnes of available methods (for instance, ls or tr) 1904ae12b187SLois Curfman McInnes 19059b94acceSBarry Smith Notes: 19069b94acceSBarry Smith See "petsc/include/snes.h" for available methods (for instance) 190736851e7fSLois Curfman McInnes + SNES_EQ_LS - Newton's method with line search 1908c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 1909c7afd0dbSLois Curfman McInnes . SNES_EQ_TR - Newton's method with trust region 1910c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 1911c7afd0dbSLois Curfman McInnes . SNES_UM_TR - Newton's method with trust region 1912c7afd0dbSLois Curfman McInnes (unconstrained minimization) 191336851e7fSLois Curfman McInnes - SNES_UM_LS - Newton's method with line search 1914c7afd0dbSLois Curfman McInnes (unconstrained minimization) 19159b94acceSBarry Smith 1916ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 1917ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 1918ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 1919ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 1920ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 1921ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 1922ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 1923ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 1924ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 192536851e7fSLois Curfman McInnes appropriate method. In other words, this routine is not for beginners. 192636851e7fSLois Curfman McInnes 192736851e7fSLois Curfman McInnes Level: intermediate 1928a703fe33SLois Curfman McInnes 1929f536c699SSatish Balay .keywords: SNES, set, method 19309b94acceSBarry Smith @*/ 19314b0e389bSBarry Smith int SNESSetType(SNES snes,SNESType method) 19329b94acceSBarry Smith { 193384cb2905SBarry Smith int ierr; 19349b94acceSBarry Smith int (*r)(SNES); 19353a40ed3dSBarry Smith 19363a40ed3dSBarry Smith PetscFunctionBegin; 193777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 193882bf6240SBarry Smith 19393f1db9ecSBarry Smith if (PetscTypeCompare(snes->type_name,method)) PetscFunctionReturn(0); 194082bf6240SBarry Smith 194182bf6240SBarry Smith if (snes->setupcalled) { 1942e1311b90SBarry Smith ierr = (*(snes)->destroy)(snes);CHKERRQ(ierr); 194382bf6240SBarry Smith snes->data = 0; 194482bf6240SBarry Smith } 19457d1a2b2bSBarry Smith 19469b94acceSBarry Smith /* Get the function pointers for the iterative method requested */ 194782bf6240SBarry Smith if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL); CHKERRQ(ierr);} 194882bf6240SBarry Smith 1949488ecbafSBarry Smith ierr = FListFind(snes->comm, SNESList, method,(int (**)(void *)) &r );CHKERRQ(ierr); 195082bf6240SBarry Smith 1951596552b5SBarry Smith if (!r) SETERRQ1(1,1,"Unable to find requested SNES type %s",method); 19521548b14aSBarry Smith 19530452661fSBarry Smith if (snes->data) PetscFree(snes->data); 195482bf6240SBarry Smith snes->data = 0; 19553a40ed3dSBarry Smith ierr = (*r)(snes); CHKERRQ(ierr); 195682bf6240SBarry Smith 195782bf6240SBarry Smith if (snes->type_name) PetscFree(snes->type_name); 195882bf6240SBarry Smith snes->type_name = (char *) PetscMalloc((PetscStrlen(method)+1)*sizeof(char));CHKPTRQ(snes->type_name); 195982bf6240SBarry Smith PetscStrcpy(snes->type_name,method); 196082bf6240SBarry Smith snes->set_method_called = 1; 196182bf6240SBarry Smith 19623a40ed3dSBarry Smith PetscFunctionReturn(0); 19639b94acceSBarry Smith } 19649b94acceSBarry Smith 1965a847f771SSatish Balay 19669b94acceSBarry Smith /* --------------------------------------------------------------------- */ 19675615d1e5SSatish Balay #undef __FUNC__ 1968d4bb536fSBarry Smith #define __FUNC__ "SNESRegisterDestroy" 19699b94acceSBarry Smith /*@C 19709b94acceSBarry Smith SNESRegisterDestroy - Frees the list of nonlinear solvers that were 19719b94acceSBarry Smith registered by SNESRegister(). 19729b94acceSBarry Smith 1973fee21e36SBarry Smith Not Collective 1974fee21e36SBarry Smith 197536851e7fSLois Curfman McInnes Level: advanced 197636851e7fSLois Curfman McInnes 19779b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy 19789b94acceSBarry Smith 19799b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll() 19809b94acceSBarry Smith @*/ 1981cf256101SBarry Smith int SNESRegisterDestroy(void) 19829b94acceSBarry Smith { 198382bf6240SBarry Smith int ierr; 198482bf6240SBarry Smith 19853a40ed3dSBarry Smith PetscFunctionBegin; 198682bf6240SBarry Smith if (SNESList) { 1987488ecbafSBarry Smith ierr = FListDestroy( SNESList );CHKERRQ(ierr); 198882bf6240SBarry Smith SNESList = 0; 19899b94acceSBarry Smith } 199084cb2905SBarry Smith SNESRegisterAllCalled = 0; 19913a40ed3dSBarry Smith PetscFunctionReturn(0); 19929b94acceSBarry Smith } 19939b94acceSBarry Smith 19945615d1e5SSatish Balay #undef __FUNC__ 1995d4bb536fSBarry Smith #define __FUNC__ "SNESGetType" 19969b94acceSBarry Smith /*@C 19979a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 19989b94acceSBarry Smith 1999c7afd0dbSLois Curfman McInnes Not Collective 2000c7afd0dbSLois Curfman McInnes 20019b94acceSBarry Smith Input Parameter: 20024b0e389bSBarry Smith . snes - nonlinear solver context 20039b94acceSBarry Smith 20049b94acceSBarry Smith Output Parameter: 200582bf6240SBarry Smith . method - SNES method (a charactor string) 20069b94acceSBarry Smith 200736851e7fSLois Curfman McInnes Level: intermediate 200836851e7fSLois Curfman McInnes 20099b94acceSBarry Smith .keywords: SNES, nonlinear, get, method, name 20109b94acceSBarry Smith @*/ 201182bf6240SBarry Smith int SNESGetType(SNES snes, SNESType *method) 20129b94acceSBarry Smith { 20133a40ed3dSBarry Smith PetscFunctionBegin; 201482bf6240SBarry Smith *method = snes->type_name; 20153a40ed3dSBarry Smith PetscFunctionReturn(0); 20169b94acceSBarry Smith } 20179b94acceSBarry Smith 20185615d1e5SSatish Balay #undef __FUNC__ 2019d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolution" 20209b94acceSBarry Smith /*@C 20219b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 20229b94acceSBarry Smith stored. 20239b94acceSBarry Smith 2024c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2025c7afd0dbSLois Curfman McInnes 20269b94acceSBarry Smith Input Parameter: 20279b94acceSBarry Smith . snes - the SNES context 20289b94acceSBarry Smith 20299b94acceSBarry Smith Output Parameter: 20309b94acceSBarry Smith . x - the solution 20319b94acceSBarry Smith 203236851e7fSLois Curfman McInnes Level: advanced 203336851e7fSLois Curfman McInnes 20349b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 20359b94acceSBarry Smith 2036a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESGetGradient(), SNESGetSolutionUpdate() 20379b94acceSBarry Smith @*/ 20389b94acceSBarry Smith int SNESGetSolution(SNES snes,Vec *x) 20399b94acceSBarry Smith { 20403a40ed3dSBarry Smith PetscFunctionBegin; 204177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 20429b94acceSBarry Smith *x = snes->vec_sol_always; 20433a40ed3dSBarry Smith PetscFunctionReturn(0); 20449b94acceSBarry Smith } 20459b94acceSBarry Smith 20465615d1e5SSatish Balay #undef __FUNC__ 2047d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolutionUpdate" 20489b94acceSBarry Smith /*@C 20499b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 20509b94acceSBarry Smith stored. 20519b94acceSBarry Smith 2052c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2053c7afd0dbSLois Curfman McInnes 20549b94acceSBarry Smith Input Parameter: 20559b94acceSBarry Smith . snes - the SNES context 20569b94acceSBarry Smith 20579b94acceSBarry Smith Output Parameter: 20589b94acceSBarry Smith . x - the solution update 20599b94acceSBarry Smith 206036851e7fSLois Curfman McInnes Level: advanced 206136851e7fSLois Curfman McInnes 20629b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 20639b94acceSBarry Smith 20649b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction 20659b94acceSBarry Smith @*/ 20669b94acceSBarry Smith int SNESGetSolutionUpdate(SNES snes,Vec *x) 20679b94acceSBarry Smith { 20683a40ed3dSBarry Smith PetscFunctionBegin; 206977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 20709b94acceSBarry Smith *x = snes->vec_sol_update_always; 20713a40ed3dSBarry Smith PetscFunctionReturn(0); 20729b94acceSBarry Smith } 20739b94acceSBarry Smith 20745615d1e5SSatish Balay #undef __FUNC__ 2075d4bb536fSBarry Smith #define __FUNC__ "SNESGetFunction" 20769b94acceSBarry Smith /*@C 20773638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 20789b94acceSBarry Smith 2079c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2080c7afd0dbSLois Curfman McInnes 20819b94acceSBarry Smith Input Parameter: 20829b94acceSBarry Smith . snes - the SNES context 20839b94acceSBarry Smith 20849b94acceSBarry Smith Output Parameter: 20853638b69dSLois Curfman McInnes . r - the function 20869b94acceSBarry Smith 20879b94acceSBarry Smith Notes: 20889b94acceSBarry Smith SNESGetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only 20899b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 20909b94acceSBarry Smith SNESGetMinimizationFunction() and SNESGetGradient(); 20919b94acceSBarry Smith 209236851e7fSLois Curfman McInnes Level: advanced 209336851e7fSLois Curfman McInnes 2094a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 20959b94acceSBarry Smith 209631615d04SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution(), SNESGetMinimizationFunction(), 209731615d04SLois Curfman McInnes SNESGetGradient() 20989b94acceSBarry Smith @*/ 20999b94acceSBarry Smith int SNESGetFunction(SNES snes,Vec *r) 21009b94acceSBarry Smith { 21013a40ed3dSBarry Smith PetscFunctionBegin; 210277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2103a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 2104a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 2105a8c6a408SBarry Smith } 21069b94acceSBarry Smith *r = snes->vec_func_always; 21073a40ed3dSBarry Smith PetscFunctionReturn(0); 21089b94acceSBarry Smith } 21099b94acceSBarry Smith 21105615d1e5SSatish Balay #undef __FUNC__ 2111d4bb536fSBarry Smith #define __FUNC__ "SNESGetGradient" 21129b94acceSBarry Smith /*@C 21133638b69dSLois Curfman McInnes SNESGetGradient - Returns the vector where the gradient is stored. 21149b94acceSBarry Smith 2115c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2116c7afd0dbSLois Curfman McInnes 21179b94acceSBarry Smith Input Parameter: 21189b94acceSBarry Smith . snes - the SNES context 21199b94acceSBarry Smith 21209b94acceSBarry Smith Output Parameter: 21219b94acceSBarry Smith . r - the gradient 21229b94acceSBarry Smith 21239b94acceSBarry Smith Notes: 21249b94acceSBarry Smith SNESGetGradient() is valid for SNES_UNCONSTRAINED_MINIMIZATION methods 21259b94acceSBarry Smith only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 21269b94acceSBarry Smith SNESGetFunction(). 21279b94acceSBarry Smith 212836851e7fSLois Curfman McInnes Level: advanced 212936851e7fSLois Curfman McInnes 21309b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient 21319b94acceSBarry Smith 213231615d04SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESGetSolution(), SNESGetFunction() 21339b94acceSBarry Smith @*/ 21349b94acceSBarry Smith int SNESGetGradient(SNES snes,Vec *r) 21359b94acceSBarry Smith { 21363a40ed3dSBarry Smith PetscFunctionBegin; 213777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 21383a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 2139a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 21403a40ed3dSBarry Smith } 21419b94acceSBarry Smith *r = snes->vec_func_always; 21423a40ed3dSBarry Smith PetscFunctionReturn(0); 21439b94acceSBarry Smith } 21449b94acceSBarry Smith 21455615d1e5SSatish Balay #undef __FUNC__ 2146d4bb536fSBarry Smith #define __FUNC__ "SNESGetMinimizationFunction" 21479b94acceSBarry Smith /*@ 21489b94acceSBarry Smith SNESGetMinimizationFunction - Returns the scalar function value for 21499b94acceSBarry Smith unconstrained minimization problems. 21509b94acceSBarry Smith 2151c7afd0dbSLois Curfman McInnes Not Collective 2152c7afd0dbSLois Curfman McInnes 21539b94acceSBarry Smith Input Parameter: 21549b94acceSBarry Smith . snes - the SNES context 21559b94acceSBarry Smith 21569b94acceSBarry Smith Output Parameter: 21579b94acceSBarry Smith . r - the function 21589b94acceSBarry Smith 21599b94acceSBarry Smith Notes: 21609b94acceSBarry Smith SNESGetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 21619b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 21629b94acceSBarry Smith SNESGetFunction(). 21639b94acceSBarry Smith 216436851e7fSLois Curfman McInnes Level: advanced 216536851e7fSLois Curfman McInnes 21669b94acceSBarry Smith .keywords: SNES, nonlinear, get, function 21679b94acceSBarry Smith 216831615d04SLois Curfman McInnes .seealso: SNESGetGradient(), SNESGetSolution(), SNESGetFunction() 21699b94acceSBarry Smith @*/ 21709b94acceSBarry Smith int SNESGetMinimizationFunction(SNES snes,double *r) 21719b94acceSBarry Smith { 21723a40ed3dSBarry Smith PetscFunctionBegin; 217377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 217474679c65SBarry Smith PetscValidScalarPointer(r); 21753a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 2176a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 21773a40ed3dSBarry Smith } 21789b94acceSBarry Smith *r = snes->fc; 21793a40ed3dSBarry Smith PetscFunctionReturn(0); 21809b94acceSBarry Smith } 21819b94acceSBarry Smith 21825615d1e5SSatish Balay #undef __FUNC__ 2183d4bb536fSBarry Smith #define __FUNC__ "SNESSetOptionsPrefix" 21843c7409f5SSatish Balay /*@C 21853c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 2186d850072dSLois Curfman McInnes SNES options in the database. 21873c7409f5SSatish Balay 2188fee21e36SBarry Smith Collective on SNES 2189fee21e36SBarry Smith 2190c7afd0dbSLois Curfman McInnes Input Parameter: 2191c7afd0dbSLois Curfman McInnes + snes - the SNES context 2192c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2193c7afd0dbSLois Curfman McInnes 2194d850072dSLois Curfman McInnes Notes: 2195a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2196c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2197d850072dSLois Curfman McInnes 219836851e7fSLois Curfman McInnes Level: advanced 219936851e7fSLois Curfman McInnes 22003c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 2201a86d99e1SLois Curfman McInnes 2202a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 22033c7409f5SSatish Balay @*/ 22043c7409f5SSatish Balay int SNESSetOptionsPrefix(SNES snes,char *prefix) 22053c7409f5SSatish Balay { 22063c7409f5SSatish Balay int ierr; 22073c7409f5SSatish Balay 22083a40ed3dSBarry Smith PetscFunctionBegin; 220977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2210639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 22113c7409f5SSatish Balay ierr = SLESSetOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 22123a40ed3dSBarry Smith PetscFunctionReturn(0); 22133c7409f5SSatish Balay } 22143c7409f5SSatish Balay 22155615d1e5SSatish Balay #undef __FUNC__ 2216d4bb536fSBarry Smith #define __FUNC__ "SNESAppendOptionsPrefix" 22173c7409f5SSatish Balay /*@C 2218f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 2219d850072dSLois Curfman McInnes SNES options in the database. 22203c7409f5SSatish Balay 2221fee21e36SBarry Smith Collective on SNES 2222fee21e36SBarry Smith 2223c7afd0dbSLois Curfman McInnes Input Parameters: 2224c7afd0dbSLois Curfman McInnes + snes - the SNES context 2225c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2226c7afd0dbSLois Curfman McInnes 2227d850072dSLois Curfman McInnes Notes: 2228a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2229c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2230d850072dSLois Curfman McInnes 223136851e7fSLois Curfman McInnes Level: advanced 223236851e7fSLois Curfman McInnes 22333c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 2234a86d99e1SLois Curfman McInnes 2235a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 22363c7409f5SSatish Balay @*/ 22373c7409f5SSatish Balay int SNESAppendOptionsPrefix(SNES snes,char *prefix) 22383c7409f5SSatish Balay { 22393c7409f5SSatish Balay int ierr; 22403c7409f5SSatish Balay 22413a40ed3dSBarry Smith PetscFunctionBegin; 224277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2243639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 22443c7409f5SSatish Balay ierr = SLESAppendOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 22453a40ed3dSBarry Smith PetscFunctionReturn(0); 22463c7409f5SSatish Balay } 22473c7409f5SSatish Balay 22485615d1e5SSatish Balay #undef __FUNC__ 2249d4bb536fSBarry Smith #define __FUNC__ "SNESGetOptionsPrefix" 2250c83590e2SSatish Balay /*@ 22513c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 22523c7409f5SSatish Balay SNES options in the database. 22533c7409f5SSatish Balay 2254c7afd0dbSLois Curfman McInnes Not Collective 2255c7afd0dbSLois Curfman McInnes 22563c7409f5SSatish Balay Input Parameter: 22573c7409f5SSatish Balay . snes - the SNES context 22583c7409f5SSatish Balay 22593c7409f5SSatish Balay Output Parameter: 22603c7409f5SSatish Balay . prefix - pointer to the prefix string used 22613c7409f5SSatish Balay 226236851e7fSLois Curfman McInnes Level: advanced 226336851e7fSLois Curfman McInnes 22643c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 2265a86d99e1SLois Curfman McInnes 2266a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 22673c7409f5SSatish Balay @*/ 22683c7409f5SSatish Balay int SNESGetOptionsPrefix(SNES snes,char **prefix) 22693c7409f5SSatish Balay { 22703c7409f5SSatish Balay int ierr; 22713c7409f5SSatish Balay 22723a40ed3dSBarry Smith PetscFunctionBegin; 227377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2274639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 22753a40ed3dSBarry Smith PetscFunctionReturn(0); 22763c7409f5SSatish Balay } 22773c7409f5SSatish Balay 2278ca161407SBarry Smith #undef __FUNC__ 2279ca161407SBarry Smith #define __FUNC__ "SNESPrintHelp" 2280ca161407SBarry Smith /*@ 2281ca161407SBarry Smith SNESPrintHelp - Prints all options for the SNES component. 2282ca161407SBarry Smith 2283c7afd0dbSLois Curfman McInnes Collective on SNES 2284c7afd0dbSLois Curfman McInnes 2285ca161407SBarry Smith Input Parameter: 2286ca161407SBarry Smith . snes - the SNES context 2287ca161407SBarry Smith 2288ca161407SBarry Smith Options Database Keys: 2289c7afd0dbSLois Curfman McInnes + -help - Prints SNES options 2290c7afd0dbSLois Curfman McInnes - -h - Prints SNES options 2291ca161407SBarry Smith 229236851e7fSLois Curfman McInnes Level: beginner 229336851e7fSLois Curfman McInnes 2294ca161407SBarry Smith .keywords: SNES, nonlinear, help 2295ca161407SBarry Smith 2296ca161407SBarry Smith .seealso: SNESSetFromOptions() 2297ca161407SBarry Smith @*/ 2298ca161407SBarry Smith int SNESPrintHelp(SNES snes) 2299ca161407SBarry Smith { 2300ca161407SBarry Smith char p[64]; 2301ca161407SBarry Smith SNES_KSP_EW_ConvCtx *kctx; 2302ca161407SBarry Smith int ierr; 2303ca161407SBarry Smith 2304ca161407SBarry Smith PetscFunctionBegin; 2305ca161407SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2306ca161407SBarry Smith 2307ca161407SBarry Smith PetscStrcpy(p,"-"); 2308ca161407SBarry Smith if (snes->prefix) PetscStrcat(p, snes->prefix); 2309ca161407SBarry Smith 2310ca161407SBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 2311ca161407SBarry Smith 231276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm,"SNES options ------------------------------------------------\n"); 2313488ecbafSBarry Smith ierr = FListPrintTypes(snes->comm,stdout,snes->prefix,"snes_type",SNESList);CHKERRQ(ierr); 231476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_view: view SNES info after each nonlinear solve\n",p); 231576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_max_it <its>: max iterations (default %d)\n",p,snes->max_its); 231676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_max_funcs <maxf>: max function evals (default %d)\n",p,snes->max_funcs); 231776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_stol <stol>: successive step tolerance (default %g)\n",p,snes->xtol); 231876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_atol <atol>: absolute tolerance (default %g)\n",p,snes->atol); 231976be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_rtol <rtol>: relative tolerance (default %g)\n",p,snes->rtol); 232076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_trtol <trtol>: trust region parameter tolerance (default %g)\n",p,snes->deltatol); 232176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," SNES Monitoring Options: Choose any of the following\n"); 232276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_cancelmonitors: cancels all monitors hardwired in code\n",p); 232376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_monitor: use default SNES convergence monitor, prints\n\ 2324ca161407SBarry Smith residual norm at each iteration.\n",p); 232576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_smonitor: same as the above, but prints fewer digits of the\n\ 2326ca161407SBarry Smith residual norm for small residual norms. This is useful to conceal\n\ 2327ca161407SBarry Smith meaningless digits that may be different on different machines.\n",p); 232876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_xmonitor [x,y,w,h]: use X graphics convergence monitor\n",p); 2329ca161407SBarry Smith if (snes->type == SNES_NONLINEAR_EQUATIONS) { 233076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2331ca161407SBarry Smith " Options for solving systems of nonlinear equations only:\n"); 233276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fd: use finite differences for Jacobian\n",p); 233376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf: use matrix-free Jacobian\n",p); 233476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf_operator:use matrix-free Jacobian and user-provided preconditioning matrix\n",p); 2335e24b481bSBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf_ksp_monitor - if using matrix-free multiply then prints h at each KSP iteration\n",p); 233676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_no_convergence_test: Do not test for convergence, always run to SNES max its\n",p); 233776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_ksp_ew_conv: use Eisenstat-Walker computation of KSP rtol. Params are:\n",p); 233876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2339ca161407SBarry Smith " %ssnes_ksp_ew_version <version> (1 or 2, default is %d)\n",p,kctx->version); 234076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2341ca161407SBarry Smith " %ssnes_ksp_ew_rtol0 <rtol0> (0 <= rtol0 < 1, default %g)\n",p,kctx->rtol_0); 234276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2343ca161407SBarry Smith " %ssnes_ksp_ew_rtolmax <rtolmax> (0 <= rtolmax < 1, default %g)\n",p,kctx->rtol_max); 234476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2345ca161407SBarry Smith " %ssnes_ksp_ew_gamma <gamma> (0 <= gamma <= 1, default %g)\n",p,kctx->gamma); 234676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2347ca161407SBarry Smith " %ssnes_ksp_ew_alpha <alpha> (1 < alpha <= 2, default %g)\n",p,kctx->alpha); 234876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2349ca161407SBarry Smith " %ssnes_ksp_ew_alpha2 <alpha2> (default %g)\n",p,kctx->alpha2); 235076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2351ca161407SBarry Smith " %ssnes_ksp_ew_threshold <threshold> (0 < threshold < 1, default %g)\n",p,kctx->threshold); 2352ca161407SBarry Smith } else if (snes->type == SNES_UNCONSTRAINED_MINIMIZATION) { 235376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2354ca161407SBarry Smith " Options for solving unconstrained minimization problems only:\n"); 235576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fmin <ftol>: minimum function tolerance (default %g)\n",p,snes->fmin); 235676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fd: use finite differences for Hessian\n",p); 235776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf: use matrix-free Hessian\n",p); 2358ca161407SBarry Smith } 235976be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," Run program with -help %ssnes_type <method> for help on ",p); 236076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm,"a particular method\n"); 2361ca161407SBarry Smith if (snes->printhelp) { 2362ca161407SBarry Smith ierr = (*snes->printhelp)(snes,p);CHKERRQ(ierr); 2363ca161407SBarry Smith } 2364ca161407SBarry Smith PetscFunctionReturn(0); 2365ca161407SBarry Smith } 23663c7409f5SSatish Balay 2367acb85ae6SSatish Balay /*MC 2368b2002411SLois Curfman McInnes SNESRegister - Adds a method to the nonlinear solver package. 23699b94acceSBarry Smith 2370b2002411SLois Curfman McInnes Synopsis: 2371b2002411SLois Curfman McInnes SNESRegister(char *name_solver,char *path,char *name_create,int (*routine_create)(SNES)) 23729b94acceSBarry Smith 23738d76a1e5SLois Curfman McInnes Not collective 23748d76a1e5SLois Curfman McInnes 2375b2002411SLois Curfman McInnes Input Parameters: 2376c7afd0dbSLois Curfman McInnes + name_solver - name of a new user-defined solver 2377b2002411SLois Curfman McInnes . path - path (either absolute or relative) the library containing this solver 2378b2002411SLois Curfman McInnes . name_create - name of routine to create method context 2379c7afd0dbSLois Curfman McInnes - routine_create - routine to create method context 23809b94acceSBarry Smith 2381b2002411SLois Curfman McInnes Notes: 2382b2002411SLois Curfman McInnes SNESRegister() may be called multiple times to add several user-defined solvers. 23833a40ed3dSBarry Smith 2384b2002411SLois Curfman McInnes If dynamic libraries are used, then the fourth input argument (routine_create) 2385b2002411SLois Curfman McInnes is ignored. 2386b2002411SLois Curfman McInnes 2387b2002411SLois Curfman McInnes Sample usage: 238869225d78SLois Curfman McInnes .vb 2389b2002411SLois Curfman McInnes SNESRegister("my_solver",/home/username/my_lib/lib/libg/solaris/mylib.a, 2390b2002411SLois Curfman McInnes "MySolverCreate",MySolverCreate); 239169225d78SLois Curfman McInnes .ve 2392b2002411SLois Curfman McInnes 2393b2002411SLois Curfman McInnes Then, your solver can be chosen with the procedural interface via 2394b2002411SLois Curfman McInnes $ SNESSetType(snes,"my_solver") 2395b2002411SLois Curfman McInnes or at runtime via the option 2396b2002411SLois Curfman McInnes $ -snes_type my_solver 2397b2002411SLois Curfman McInnes 239836851e7fSLois Curfman McInnes Level: advanced 239936851e7fSLois Curfman McInnes 2400b2002411SLois Curfman McInnes .keywords: SNES, nonlinear, register 2401b2002411SLois Curfman McInnes 2402b2002411SLois Curfman McInnes .seealso: SNESRegisterAll(), SNESRegisterDestroy() 2403b2002411SLois Curfman McInnes M*/ 2404b2002411SLois Curfman McInnes 2405b2002411SLois Curfman McInnes #undef __FUNC__ 2406b2002411SLois Curfman McInnes #define __FUNC__ "SNESRegister_Private" 2407b2002411SLois Curfman McInnes int SNESRegister_Private(char *sname,char *path,char *name,int (*function)(SNES)) 2408b2002411SLois Curfman McInnes { 2409b2002411SLois Curfman McInnes char fullname[256]; 2410b2002411SLois Curfman McInnes int ierr; 2411b2002411SLois Curfman McInnes 2412b2002411SLois Curfman McInnes PetscFunctionBegin; 2413b2002411SLois Curfman McInnes PetscStrcpy(fullname,path); PetscStrcat(fullname,":");PetscStrcat(fullname,name); 2414488ecbafSBarry Smith ierr = FListAdd_Private(&SNESList,sname,fullname, (int (*)(void*))function);CHKERRQ(ierr); 2415b2002411SLois Curfman McInnes PetscFunctionReturn(0); 2416b2002411SLois Curfman McInnes } 2417