1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER 2*40a0c1c6SLois Curfman McInnes static char vcid[] = "$Id: snes.c,v 1.168 1999/02/02 02:47:20 curfman Exp curfman $"; 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 253e24b481bSBarry Smith 2543c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_fd", &flg); CHKERRQ(ierr); 2553c7409f5SSatish Balay if (flg && snes->method_class == SNES_NONLINEAR_EQUATIONS) { 2569b94acceSBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre, 2579b94acceSBarry Smith SNESDefaultComputeJacobian,snes->funP); CHKERRQ(ierr); 258981c4779SBarry Smith PLogInfo(snes,"SNESSetFromOptions: Setting default finite difference Jacobian matrix\n"); 259981c4779SBarry Smith } else if (flg && snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 26031615d04SLois Curfman McInnes ierr = SNESSetHessian(snes,snes->jacobian,snes->jacobian_pre, 26131615d04SLois Curfman McInnes SNESDefaultComputeHessian,snes->funP); CHKERRQ(ierr); 262d64ed03dSBarry Smith PLogInfo(snes,"SNESSetFromOptions: Setting default finite difference Hessian matrix\n"); 2639b94acceSBarry Smith } 264639f9d9dSBarry Smith 265639f9d9dSBarry Smith for ( i=0; i<numberofsetfromoptions; i++ ) { 266639f9d9dSBarry Smith ierr = (*othersetfromoptions[i])(snes); CHKERRQ(ierr); 267639f9d9dSBarry Smith } 268639f9d9dSBarry Smith 2699b94acceSBarry Smith ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 2709b94acceSBarry Smith ierr = SLESSetFromOptions(sles); CHKERRQ(ierr); 27193993e2dSLois Curfman McInnes 272e24b481bSBarry Smith /* set the special KSP monitor for matrix-free application */ 273e24b481bSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_mf_ksp_monitor",&flg); CHKERRQ(ierr); 274e24b481bSBarry Smith if (flg) { 275e24b481bSBarry Smith KSP ksp; 276e24b481bSBarry Smith ierr = SLESGetKSP(sles,&ksp);CHKERRQ(ierr); 277e24b481bSBarry Smith ierr = KSPSetMonitor(ksp,MatSNESFDMFKSPMonitor,PETSC_NULL);CHKERRQ(ierr); 278e24b481bSBarry Smith } 279e24b481bSBarry Smith 28082bf6240SBarry Smith /* 28193993e2dSLois Curfman McInnes Since the private setfromoptions requires the type to have 28293993e2dSLois Curfman McInnes been set already, we make sure a type is set by this time. 28382bf6240SBarry Smith */ 28482bf6240SBarry Smith if (!snes->type_name) { 28582bf6240SBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 28682bf6240SBarry Smith ierr = SNESSetType(snes,SNES_EQ_LS); CHKERRQ(ierr); 28782bf6240SBarry Smith } else { 28882bf6240SBarry Smith ierr = SNESSetType(snes,SNES_UM_TR); CHKERRQ(ierr); 28982bf6240SBarry Smith } 29082bf6240SBarry Smith } 29182bf6240SBarry Smith 29282bf6240SBarry Smith ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 29382bf6240SBarry Smith if (flg) { ierr = SNESPrintHelp(snes); CHKERRQ(ierr);} 29482bf6240SBarry Smith 2953a40ed3dSBarry Smith if (snes->setfromoptions) { 2963a40ed3dSBarry Smith ierr = (*snes->setfromoptions)(snes);CHKERRQ(ierr); 2973a40ed3dSBarry Smith } 2983a40ed3dSBarry Smith PetscFunctionReturn(0); 2999b94acceSBarry Smith } 3009b94acceSBarry Smith 301a847f771SSatish Balay 3025615d1e5SSatish Balay #undef __FUNC__ 303d4bb536fSBarry Smith #define __FUNC__ "SNESSetApplicationContext" 3049b94acceSBarry Smith /*@ 3059b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 3069b94acceSBarry Smith the nonlinear solvers. 3079b94acceSBarry Smith 308fee21e36SBarry Smith Collective on SNES 309fee21e36SBarry Smith 310c7afd0dbSLois Curfman McInnes Input Parameters: 311c7afd0dbSLois Curfman McInnes + snes - the SNES context 312c7afd0dbSLois Curfman McInnes - usrP - optional user context 313c7afd0dbSLois Curfman McInnes 31436851e7fSLois Curfman McInnes Level: intermediate 31536851e7fSLois Curfman McInnes 3169b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 3179b94acceSBarry Smith 3189b94acceSBarry Smith .seealso: SNESGetApplicationContext() 3199b94acceSBarry Smith @*/ 3209b94acceSBarry Smith int SNESSetApplicationContext(SNES snes,void *usrP) 3219b94acceSBarry Smith { 3223a40ed3dSBarry Smith PetscFunctionBegin; 32377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3249b94acceSBarry Smith snes->user = usrP; 3253a40ed3dSBarry Smith PetscFunctionReturn(0); 3269b94acceSBarry Smith } 32774679c65SBarry Smith 3285615d1e5SSatish Balay #undef __FUNC__ 329d4bb536fSBarry Smith #define __FUNC__ "SNESGetApplicationContext" 3309b94acceSBarry Smith /*@C 3319b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 3329b94acceSBarry Smith nonlinear solvers. 3339b94acceSBarry Smith 334c7afd0dbSLois Curfman McInnes Not Collective 335c7afd0dbSLois Curfman McInnes 3369b94acceSBarry Smith Input Parameter: 3379b94acceSBarry Smith . snes - SNES context 3389b94acceSBarry Smith 3399b94acceSBarry Smith Output Parameter: 3409b94acceSBarry Smith . usrP - user context 3419b94acceSBarry Smith 34236851e7fSLois Curfman McInnes Level: intermediate 34336851e7fSLois Curfman McInnes 3449b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 3459b94acceSBarry Smith 3469b94acceSBarry Smith .seealso: SNESSetApplicationContext() 3479b94acceSBarry Smith @*/ 3489b94acceSBarry Smith int SNESGetApplicationContext( SNES snes, void **usrP ) 3499b94acceSBarry Smith { 3503a40ed3dSBarry Smith PetscFunctionBegin; 35177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3529b94acceSBarry Smith *usrP = snes->user; 3533a40ed3dSBarry Smith PetscFunctionReturn(0); 3549b94acceSBarry Smith } 35574679c65SBarry Smith 3565615d1e5SSatish Balay #undef __FUNC__ 357d4bb536fSBarry Smith #define __FUNC__ "SNESGetIterationNumber" 3589b94acceSBarry Smith /*@ 3599b94acceSBarry Smith SNESGetIterationNumber - Gets the current iteration number of the 3609b94acceSBarry Smith nonlinear solver. 3619b94acceSBarry Smith 362c7afd0dbSLois Curfman McInnes Not Collective 363c7afd0dbSLois Curfman McInnes 3649b94acceSBarry Smith Input Parameter: 3659b94acceSBarry Smith . snes - SNES context 3669b94acceSBarry Smith 3679b94acceSBarry Smith Output Parameter: 3689b94acceSBarry Smith . iter - iteration number 3699b94acceSBarry Smith 37036851e7fSLois Curfman McInnes Level: intermediate 37136851e7fSLois Curfman McInnes 3729b94acceSBarry Smith .keywords: SNES, nonlinear, get, iteration, number 3739b94acceSBarry Smith @*/ 3749b94acceSBarry Smith int SNESGetIterationNumber(SNES snes,int* iter) 3759b94acceSBarry Smith { 3763a40ed3dSBarry Smith PetscFunctionBegin; 37777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 37874679c65SBarry Smith PetscValidIntPointer(iter); 3799b94acceSBarry Smith *iter = snes->iter; 3803a40ed3dSBarry Smith PetscFunctionReturn(0); 3819b94acceSBarry Smith } 38274679c65SBarry Smith 3835615d1e5SSatish Balay #undef __FUNC__ 3845615d1e5SSatish Balay #define __FUNC__ "SNESGetFunctionNorm" 3859b94acceSBarry Smith /*@ 3869b94acceSBarry Smith SNESGetFunctionNorm - Gets the norm of the current function that was set 3879b94acceSBarry Smith with SNESSSetFunction(). 3889b94acceSBarry Smith 389c7afd0dbSLois Curfman McInnes Collective on SNES 390c7afd0dbSLois Curfman McInnes 3919b94acceSBarry Smith Input Parameter: 3929b94acceSBarry Smith . snes - SNES context 3939b94acceSBarry Smith 3949b94acceSBarry Smith Output Parameter: 3959b94acceSBarry Smith . fnorm - 2-norm of function 3969b94acceSBarry Smith 3979b94acceSBarry Smith Note: 3989b94acceSBarry Smith SNESGetFunctionNorm() is valid for SNES_NONLINEAR_EQUATIONS methods only. 399a86d99e1SLois Curfman McInnes A related routine for SNES_UNCONSTRAINED_MINIMIZATION methods is 400a86d99e1SLois Curfman McInnes SNESGetGradientNorm(). 4019b94acceSBarry Smith 40236851e7fSLois Curfman McInnes Level: intermediate 40336851e7fSLois Curfman McInnes 4049b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm 405a86d99e1SLois Curfman McInnes 406a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction() 4079b94acceSBarry Smith @*/ 4089b94acceSBarry Smith int SNESGetFunctionNorm(SNES snes,Scalar *fnorm) 4099b94acceSBarry Smith { 4103a40ed3dSBarry Smith PetscFunctionBegin; 41177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 41274679c65SBarry Smith PetscValidScalarPointer(fnorm); 41374679c65SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 414d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_NONLINEAR_EQUATIONS only"); 41574679c65SBarry Smith } 4169b94acceSBarry Smith *fnorm = snes->norm; 4173a40ed3dSBarry Smith PetscFunctionReturn(0); 4189b94acceSBarry Smith } 41974679c65SBarry Smith 4205615d1e5SSatish Balay #undef __FUNC__ 4215615d1e5SSatish Balay #define __FUNC__ "SNESGetGradientNorm" 4229b94acceSBarry Smith /*@ 4239b94acceSBarry Smith SNESGetGradientNorm - Gets the norm of the current gradient that was set 4249b94acceSBarry Smith with SNESSSetGradient(). 4259b94acceSBarry Smith 426c7afd0dbSLois Curfman McInnes Collective on SNES 427c7afd0dbSLois Curfman McInnes 4289b94acceSBarry Smith Input Parameter: 4299b94acceSBarry Smith . snes - SNES context 4309b94acceSBarry Smith 4319b94acceSBarry Smith Output Parameter: 4329b94acceSBarry Smith . fnorm - 2-norm of gradient 4339b94acceSBarry Smith 4349b94acceSBarry Smith Note: 4359b94acceSBarry Smith SNESGetGradientNorm() is valid for SNES_UNCONSTRAINED_MINIMIZATION 436a86d99e1SLois Curfman McInnes methods only. A related routine for SNES_NONLINEAR_EQUATIONS methods 437a86d99e1SLois Curfman McInnes is SNESGetFunctionNorm(). 4389b94acceSBarry Smith 43936851e7fSLois Curfman McInnes Level: intermediate 44036851e7fSLois Curfman McInnes 4419b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient, norm 442a86d99e1SLois Curfman McInnes 443a86d99e1SLois Curfman McInnes .seelso: SNESSetGradient() 4449b94acceSBarry Smith @*/ 4459b94acceSBarry Smith int SNESGetGradientNorm(SNES snes,Scalar *gnorm) 4469b94acceSBarry Smith { 4473a40ed3dSBarry Smith PetscFunctionBegin; 44877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 44974679c65SBarry Smith PetscValidScalarPointer(gnorm); 45074679c65SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 451d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 45274679c65SBarry Smith } 4539b94acceSBarry Smith *gnorm = snes->norm; 4543a40ed3dSBarry Smith PetscFunctionReturn(0); 4559b94acceSBarry Smith } 45674679c65SBarry Smith 4575615d1e5SSatish Balay #undef __FUNC__ 458d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberUnsuccessfulSteps" 4599b94acceSBarry Smith /*@ 4609b94acceSBarry Smith SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps 4619b94acceSBarry Smith attempted by the nonlinear solver. 4629b94acceSBarry Smith 463c7afd0dbSLois Curfman McInnes Not Collective 464c7afd0dbSLois Curfman McInnes 4659b94acceSBarry Smith Input Parameter: 4669b94acceSBarry Smith . snes - SNES context 4679b94acceSBarry Smith 4689b94acceSBarry Smith Output Parameter: 4699b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 4709b94acceSBarry Smith 471c96a6f78SLois Curfman McInnes Notes: 472c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 473c96a6f78SLois Curfman McInnes 47436851e7fSLois Curfman McInnes Level: intermediate 47536851e7fSLois Curfman McInnes 4769b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 4779b94acceSBarry Smith @*/ 4789b94acceSBarry Smith int SNESGetNumberUnsuccessfulSteps(SNES snes,int* nfails) 4799b94acceSBarry Smith { 4803a40ed3dSBarry Smith PetscFunctionBegin; 48177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 48274679c65SBarry Smith PetscValidIntPointer(nfails); 4839b94acceSBarry Smith *nfails = snes->nfailures; 4843a40ed3dSBarry Smith PetscFunctionReturn(0); 4859b94acceSBarry Smith } 486a847f771SSatish Balay 4875615d1e5SSatish Balay #undef __FUNC__ 488d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberLinearIterations" 489c96a6f78SLois Curfman McInnes /*@ 490c96a6f78SLois Curfman McInnes SNESGetNumberLinearIterations - Gets the total number of linear iterations 491c96a6f78SLois Curfman McInnes used by the nonlinear solver. 492c96a6f78SLois Curfman McInnes 493c7afd0dbSLois Curfman McInnes Not Collective 494c7afd0dbSLois Curfman McInnes 495c96a6f78SLois Curfman McInnes Input Parameter: 496c96a6f78SLois Curfman McInnes . snes - SNES context 497c96a6f78SLois Curfman McInnes 498c96a6f78SLois Curfman McInnes Output Parameter: 499c96a6f78SLois Curfman McInnes . lits - number of linear iterations 500c96a6f78SLois Curfman McInnes 501c96a6f78SLois Curfman McInnes Notes: 502c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 503c96a6f78SLois Curfman McInnes 50436851e7fSLois Curfman McInnes Level: intermediate 50536851e7fSLois Curfman McInnes 506c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations 507c96a6f78SLois Curfman McInnes @*/ 50886bddb7dSBarry Smith int SNESGetNumberLinearIterations(SNES snes,int* lits) 509c96a6f78SLois Curfman McInnes { 5103a40ed3dSBarry Smith PetscFunctionBegin; 511c96a6f78SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 512c96a6f78SLois Curfman McInnes PetscValidIntPointer(lits); 513c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 5143a40ed3dSBarry Smith PetscFunctionReturn(0); 515c96a6f78SLois Curfman McInnes } 516c96a6f78SLois Curfman McInnes 517c96a6f78SLois Curfman McInnes #undef __FUNC__ 518d4bb536fSBarry Smith #define __FUNC__ "SNESGetSLES" 5199b94acceSBarry Smith /*@C 5209b94acceSBarry Smith SNESGetSLES - Returns the SLES context for a SNES solver. 5219b94acceSBarry Smith 522c7afd0dbSLois Curfman McInnes Not Collective, but if SNES object is parallel, then SLES object is parallel 523c7afd0dbSLois Curfman McInnes 5249b94acceSBarry Smith Input Parameter: 5259b94acceSBarry Smith . snes - the SNES context 5269b94acceSBarry Smith 5279b94acceSBarry Smith Output Parameter: 5289b94acceSBarry Smith . sles - the SLES context 5299b94acceSBarry Smith 5309b94acceSBarry Smith Notes: 5319b94acceSBarry Smith The user can then directly manipulate the SLES context to set various 5329b94acceSBarry Smith options, etc. Likewise, the user can then extract and manipulate the 5339b94acceSBarry Smith KSP and PC contexts as well. 5349b94acceSBarry Smith 53536851e7fSLois Curfman McInnes Level: beginner 53636851e7fSLois Curfman McInnes 5379b94acceSBarry Smith .keywords: SNES, nonlinear, get, SLES, context 5389b94acceSBarry Smith 5399b94acceSBarry Smith .seealso: SLESGetPC(), SLESGetKSP() 5409b94acceSBarry Smith @*/ 5419b94acceSBarry Smith int SNESGetSLES(SNES snes,SLES *sles) 5429b94acceSBarry Smith { 5433a40ed3dSBarry Smith PetscFunctionBegin; 54477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 5459b94acceSBarry Smith *sles = snes->sles; 5463a40ed3dSBarry Smith PetscFunctionReturn(0); 5479b94acceSBarry Smith } 54882bf6240SBarry Smith 549e24b481bSBarry Smith #undef __FUNC__ 550e24b481bSBarry Smith #define __FUNC__ "SNESPublish_Petsc" 551e24b481bSBarry Smith static int SNESPublish_Petsc(PetscObject object) 552e24b481bSBarry Smith { 553e24b481bSBarry Smith #if defined(HAVE_AMS) 554e24b481bSBarry Smith SNES v = (SNES) object; 555e24b481bSBarry Smith int ierr; 556e24b481bSBarry Smith 557e24b481bSBarry Smith PetscFunctionBegin; 558e24b481bSBarry Smith 559e24b481bSBarry Smith /* if it is already published then return */ 560e24b481bSBarry Smith if (v->amem >=0 ) PetscFunctionReturn(0); 561e24b481bSBarry Smith 5623f1db9ecSBarry Smith ierr = PetscObjectPublishBaseBegin(object);CHKERRQ(ierr); 563e24b481bSBarry Smith ierr = AMS_Memory_add_field((AMS_Memory)v->amem,"Iteration",&v->iter,1,AMS_INT,AMS_READ, 564e24b481bSBarry Smith AMS_COMMON,AMS_REDUCT_UNDEF);CHKERRQ(ierr); 565e24b481bSBarry Smith ierr = AMS_Memory_add_field((AMS_Memory)v->amem,"Residual",&v->norm,1,AMS_DOUBLE,AMS_READ, 566e24b481bSBarry Smith AMS_COMMON,AMS_REDUCT_UNDEF);CHKERRQ(ierr); 567e24b481bSBarry Smith ierr = PetscObjectPublishBaseEnd(object);CHKERRQ(ierr); 568e24b481bSBarry Smith #else 569e24b481bSBarry Smith PetscFunctionBegin; 570e24b481bSBarry Smith #endif 571e24b481bSBarry Smith PetscFunctionReturn(0); 572e24b481bSBarry Smith } 573e24b481bSBarry Smith 5749b94acceSBarry Smith /* -----------------------------------------------------------*/ 5755615d1e5SSatish Balay #undef __FUNC__ 5765615d1e5SSatish Balay #define __FUNC__ "SNESCreate" 5779b94acceSBarry Smith /*@C 5789b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 5799b94acceSBarry Smith 580c7afd0dbSLois Curfman McInnes Collective on MPI_Comm 581c7afd0dbSLois Curfman McInnes 582c7afd0dbSLois Curfman McInnes Input Parameters: 583c7afd0dbSLois Curfman McInnes + comm - MPI communicator 584c7afd0dbSLois Curfman McInnes - type - type of method, either 585c7afd0dbSLois Curfman McInnes SNES_NONLINEAR_EQUATIONS (for systems of nonlinear equations) 586c7afd0dbSLois Curfman McInnes or SNES_UNCONSTRAINED_MINIMIZATION (for unconstrained minimization) 5879b94acceSBarry Smith 5889b94acceSBarry Smith Output Parameter: 5899b94acceSBarry Smith . outsnes - the new SNES context 5909b94acceSBarry Smith 591c7afd0dbSLois Curfman McInnes Options Database Keys: 592c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 593c7afd0dbSLois Curfman McInnes and no preconditioning matrix 594c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 595c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 596c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 597c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 598c1f60f51SBarry Smith 59936851e7fSLois Curfman McInnes Level: beginner 60036851e7fSLois Curfman McInnes 6019b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 6029b94acceSBarry Smith 60363a78c88SLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy() 6049b94acceSBarry Smith @*/ 6054b0e389bSBarry Smith int SNESCreate(MPI_Comm comm,SNESProblemType type,SNES *outsnes) 6069b94acceSBarry Smith { 6079b94acceSBarry Smith int ierr; 6089b94acceSBarry Smith SNES snes; 6099b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 61037fcc0dbSBarry Smith 6113a40ed3dSBarry Smith PetscFunctionBegin; 6129b94acceSBarry Smith *outsnes = 0; 613d64ed03dSBarry Smith if (type != SNES_UNCONSTRAINED_MINIMIZATION && type != SNES_NONLINEAR_EQUATIONS){ 614d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"incorrect method type"); 615d64ed03dSBarry Smith } 6163f1db9ecSBarry Smith PetscHeaderCreate(snes,_p_SNES,int,SNES_COOKIE,0,"SNES",comm,SNESDestroy,SNESView); 6179b94acceSBarry Smith PLogObjectCreate(snes); 618e24b481bSBarry Smith snes->bops->publish = SNESPublish_Petsc; 6199b94acceSBarry Smith snes->max_its = 50; 6209750a799SBarry Smith snes->max_funcs = 10000; 6219b94acceSBarry Smith snes->norm = 0.0; 6225a2d0531SBarry Smith if (type == SNES_UNCONSTRAINED_MINIMIZATION) { 623b18e04deSLois Curfman McInnes snes->rtol = 1.e-8; 624b18e04deSLois Curfman McInnes snes->ttol = 0.0; 6259b94acceSBarry Smith snes->atol = 1.e-10; 626d64ed03dSBarry Smith } else { 627b4874afaSBarry Smith snes->rtol = 1.e-8; 628b4874afaSBarry Smith snes->ttol = 0.0; 629b4874afaSBarry Smith snes->atol = 1.e-50; 630b4874afaSBarry Smith } 6319b94acceSBarry Smith snes->xtol = 1.e-8; 632b18e04deSLois Curfman McInnes snes->trunctol = 1.e-12; /* no longer used */ 6339b94acceSBarry Smith snes->nfuncs = 0; 6349b94acceSBarry Smith snes->nfailures = 0; 6357a00f4a9SLois Curfman McInnes snes->linear_its = 0; 636639f9d9dSBarry Smith snes->numbermonitors = 0; 6379b94acceSBarry Smith snes->data = 0; 6389b94acceSBarry Smith snes->view = 0; 6399b94acceSBarry Smith snes->computeumfunction = 0; 6409b94acceSBarry Smith snes->umfunP = 0; 6419b94acceSBarry Smith snes->fc = 0; 6429b94acceSBarry Smith snes->deltatol = 1.e-12; 6439b94acceSBarry Smith snes->fmin = -1.e30; 6449b94acceSBarry Smith snes->method_class = type; 6459b94acceSBarry Smith snes->set_method_called = 0; 64682bf6240SBarry Smith snes->setupcalled = 0; 6479b94acceSBarry Smith snes->ksp_ewconv = 0; 6486f24a144SLois Curfman McInnes snes->xmonitor = 0; 6496f24a144SLois Curfman McInnes snes->vwork = 0; 6506f24a144SLois Curfman McInnes snes->nwork = 0; 651c9005455SLois Curfman McInnes snes->conv_hist_size = 0; 652c9005455SLois Curfman McInnes snes->conv_act_size = 0; 653c9005455SLois Curfman McInnes snes->conv_hist = 0; 6549b94acceSBarry Smith 6559b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 6560452661fSBarry Smith kctx = PetscNew(SNES_KSP_EW_ConvCtx); CHKPTRQ(kctx); 657eed86810SBarry Smith PLogObjectMemory(snes,sizeof(SNES_KSP_EW_ConvCtx)); 6589b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 6599b94acceSBarry Smith kctx->version = 2; 6609b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 6619b94acceSBarry Smith this was too large for some test cases */ 6629b94acceSBarry Smith kctx->rtol_last = 0; 6639b94acceSBarry Smith kctx->rtol_max = .9; 6649b94acceSBarry Smith kctx->gamma = 1.0; 6659b94acceSBarry Smith kctx->alpha2 = .5*(1.0 + sqrt(5.0)); 6669b94acceSBarry Smith kctx->alpha = kctx->alpha2; 6679b94acceSBarry Smith kctx->threshold = .1; 6689b94acceSBarry Smith kctx->lresid_last = 0; 6699b94acceSBarry Smith kctx->norm_last = 0; 6709b94acceSBarry Smith 6719b94acceSBarry Smith ierr = SLESCreate(comm,&snes->sles); CHKERRQ(ierr); 6729b94acceSBarry Smith PLogObjectParent(snes,snes->sles) 6735334005bSBarry Smith 6749b94acceSBarry Smith *outsnes = snes; 675e24b481bSBarry Smith PetscPublishAll(snes); 6763a40ed3dSBarry Smith PetscFunctionReturn(0); 6779b94acceSBarry Smith } 6789b94acceSBarry Smith 6799b94acceSBarry Smith /* --------------------------------------------------------------- */ 6805615d1e5SSatish Balay #undef __FUNC__ 681d4bb536fSBarry Smith #define __FUNC__ "SNESSetFunction" 6829b94acceSBarry Smith /*@C 6839b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 6849b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 6859b94acceSBarry Smith equations. 6869b94acceSBarry Smith 687fee21e36SBarry Smith Collective on SNES 688fee21e36SBarry Smith 689c7afd0dbSLois Curfman McInnes Input Parameters: 690c7afd0dbSLois Curfman McInnes + snes - the SNES context 691c7afd0dbSLois Curfman McInnes . func - function evaluation routine 692c7afd0dbSLois Curfman McInnes . r - vector to store function value 693c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 694c7afd0dbSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 6959b94acceSBarry Smith 696c7afd0dbSLois Curfman McInnes Calling sequence of func: 6978d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Vec f,void *ctx); 698c7afd0dbSLois Curfman McInnes 699313e4042SLois Curfman McInnes . f - function vector 700c7afd0dbSLois Curfman McInnes - ctx - optional user-defined function context 7019b94acceSBarry Smith 7029b94acceSBarry Smith Notes: 7039b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 7049b94acceSBarry Smith $ f'(x) x = -f(x), 705c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 7069b94acceSBarry Smith 7079b94acceSBarry Smith SNESSetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 7089b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 7099b94acceSBarry Smith SNESSetMinimizationFunction() and SNESSetGradient(); 7109b94acceSBarry Smith 71136851e7fSLois Curfman McInnes Level: beginner 71236851e7fSLois Curfman McInnes 7139b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 7149b94acceSBarry Smith 715a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 7169b94acceSBarry Smith @*/ 7175334005bSBarry Smith int SNESSetFunction( SNES snes, Vec r, int (*func)(SNES,Vec,Vec,void*),void *ctx) 7189b94acceSBarry Smith { 7193a40ed3dSBarry Smith PetscFunctionBegin; 72077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 721a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 722a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 723a8c6a408SBarry Smith } 7249b94acceSBarry Smith snes->computefunction = func; 7259b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 7269b94acceSBarry Smith snes->funP = ctx; 7273a40ed3dSBarry Smith PetscFunctionReturn(0); 7289b94acceSBarry Smith } 7299b94acceSBarry Smith 7305615d1e5SSatish Balay #undef __FUNC__ 7315615d1e5SSatish Balay #define __FUNC__ "SNESComputeFunction" 7329b94acceSBarry Smith /*@ 73336851e7fSLois Curfman McInnes SNESComputeFunction - Calls the function that has been set with 7349b94acceSBarry Smith SNESSetFunction(). 7359b94acceSBarry Smith 736c7afd0dbSLois Curfman McInnes Collective on SNES 737c7afd0dbSLois Curfman McInnes 7389b94acceSBarry Smith Input Parameters: 739c7afd0dbSLois Curfman McInnes + snes - the SNES context 740c7afd0dbSLois Curfman McInnes - x - input vector 7419b94acceSBarry Smith 7429b94acceSBarry Smith Output Parameter: 7433638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 7449b94acceSBarry Smith 7451bffabb2SLois Curfman McInnes Notes: 7469b94acceSBarry Smith SNESComputeFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 7479b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 7489b94acceSBarry Smith SNESComputeMinimizationFunction() and SNESComputeGradient(); 7499b94acceSBarry Smith 75036851e7fSLois Curfman McInnes SNESComputeFunction() is typically used within nonlinear solvers 75136851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 75236851e7fSLois Curfman McInnes themselves. 75336851e7fSLois Curfman McInnes 75436851e7fSLois Curfman McInnes Level: developer 75536851e7fSLois Curfman McInnes 7569b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 7579b94acceSBarry Smith 758a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 7599b94acceSBarry Smith @*/ 7609b94acceSBarry Smith int SNESComputeFunction(SNES snes,Vec x, Vec y) 7619b94acceSBarry Smith { 7629b94acceSBarry Smith int ierr; 7639b94acceSBarry Smith 7643a40ed3dSBarry Smith PetscFunctionBegin; 765d4bb536fSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 766a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 767d4bb536fSBarry Smith } 7689b94acceSBarry Smith PLogEventBegin(SNES_FunctionEval,snes,x,y,0); 769d64ed03dSBarry Smith PetscStackPush("SNES user function"); 7709b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 771d64ed03dSBarry Smith PetscStackPop; 772ae3c334cSLois Curfman McInnes snes->nfuncs++; 7739b94acceSBarry Smith PLogEventEnd(SNES_FunctionEval,snes,x,y,0); 7743a40ed3dSBarry Smith PetscFunctionReturn(0); 7759b94acceSBarry Smith } 7769b94acceSBarry Smith 7775615d1e5SSatish Balay #undef __FUNC__ 778d4bb536fSBarry Smith #define __FUNC__ "SNESSetMinimizationFunction" 7799b94acceSBarry Smith /*@C 7809b94acceSBarry Smith SNESSetMinimizationFunction - Sets the function evaluation routine for 7819b94acceSBarry Smith unconstrained minimization. 7829b94acceSBarry Smith 783fee21e36SBarry Smith Collective on SNES 784fee21e36SBarry Smith 785c7afd0dbSLois Curfman McInnes Input Parameters: 786c7afd0dbSLois Curfman McInnes + snes - the SNES context 787c7afd0dbSLois Curfman McInnes . func - function evaluation routine 788c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 789c7afd0dbSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 7909b94acceSBarry Smith 791c7afd0dbSLois Curfman McInnes Calling sequence of func: 7928d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,double *f,void *ctx); 793c7afd0dbSLois Curfman McInnes 794c7afd0dbSLois Curfman McInnes + x - input vector 7959b94acceSBarry Smith . f - function 796c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined function context 7979b94acceSBarry Smith 79836851e7fSLois Curfman McInnes Level: beginner 79936851e7fSLois Curfman McInnes 8009b94acceSBarry Smith Notes: 8019b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 8029b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 8039b94acceSBarry Smith SNESSetFunction(). 8049b94acceSBarry Smith 8059b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimization, function 8069b94acceSBarry Smith 807a86d99e1SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESComputeMinimizationFunction(), 808a86d99e1SLois Curfman McInnes SNESSetHessian(), SNESSetGradient() 8099b94acceSBarry Smith @*/ 8109b94acceSBarry Smith int SNESSetMinimizationFunction(SNES snes,int (*func)(SNES,Vec,double*,void*), 8119b94acceSBarry Smith void *ctx) 8129b94acceSBarry Smith { 8133a40ed3dSBarry Smith PetscFunctionBegin; 81477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 815a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 816a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 817a8c6a408SBarry Smith } 8189b94acceSBarry Smith snes->computeumfunction = func; 8199b94acceSBarry Smith snes->umfunP = ctx; 8203a40ed3dSBarry Smith PetscFunctionReturn(0); 8219b94acceSBarry Smith } 8229b94acceSBarry Smith 8235615d1e5SSatish Balay #undef __FUNC__ 8245615d1e5SSatish Balay #define __FUNC__ "SNESComputeMinimizationFunction" 8259b94acceSBarry Smith /*@ 8269b94acceSBarry Smith SNESComputeMinimizationFunction - Computes the function that has been 8279b94acceSBarry Smith set with SNESSetMinimizationFunction(). 8289b94acceSBarry Smith 829c7afd0dbSLois Curfman McInnes Collective on SNES 830c7afd0dbSLois Curfman McInnes 8319b94acceSBarry Smith Input Parameters: 832c7afd0dbSLois Curfman McInnes + snes - the SNES context 833c7afd0dbSLois Curfman McInnes - x - input vector 8349b94acceSBarry Smith 8359b94acceSBarry Smith Output Parameter: 8369b94acceSBarry Smith . y - function value 8379b94acceSBarry Smith 8389b94acceSBarry Smith Notes: 8399b94acceSBarry Smith SNESComputeMinimizationFunction() is valid only for 8409b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 8419b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 842a86d99e1SLois Curfman McInnes 84336851e7fSLois Curfman McInnes SNESComputeMinimizationFunction() is typically used within minimization 84436851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 84536851e7fSLois Curfman McInnes themselves. 84636851e7fSLois Curfman McInnes 84736851e7fSLois Curfman McInnes Level: developer 84836851e7fSLois Curfman McInnes 849a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, minimization, function 850a86d99e1SLois Curfman McInnes 851a86d99e1SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESGetMinimizationFunction(), 852a86d99e1SLois Curfman McInnes SNESComputeGradient(), SNESComputeHessian() 8539b94acceSBarry Smith @*/ 8549b94acceSBarry Smith int SNESComputeMinimizationFunction(SNES snes,Vec x,double *y) 8559b94acceSBarry Smith { 8569b94acceSBarry Smith int ierr; 8573a40ed3dSBarry Smith 8583a40ed3dSBarry Smith PetscFunctionBegin; 859a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 860a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 861a8c6a408SBarry Smith } 8629b94acceSBarry Smith PLogEventBegin(SNES_MinimizationFunctionEval,snes,x,y,0); 863d64ed03dSBarry Smith PetscStackPush("SNES user minimzation function"); 8649b94acceSBarry Smith ierr = (*snes->computeumfunction)(snes,x,y,snes->umfunP); CHKERRQ(ierr); 865d64ed03dSBarry Smith PetscStackPop; 866ae3c334cSLois Curfman McInnes snes->nfuncs++; 8679b94acceSBarry Smith PLogEventEnd(SNES_MinimizationFunctionEval,snes,x,y,0); 8683a40ed3dSBarry Smith PetscFunctionReturn(0); 8699b94acceSBarry Smith } 8709b94acceSBarry Smith 8715615d1e5SSatish Balay #undef __FUNC__ 872d4bb536fSBarry Smith #define __FUNC__ "SNESSetGradient" 8739b94acceSBarry Smith /*@C 8749b94acceSBarry Smith SNESSetGradient - Sets the gradient evaluation routine and gradient 8759b94acceSBarry Smith vector for use by the SNES routines. 8769b94acceSBarry Smith 877c7afd0dbSLois Curfman McInnes Collective on SNES 878c7afd0dbSLois Curfman McInnes 8799b94acceSBarry Smith Input Parameters: 880c7afd0dbSLois Curfman McInnes + snes - the SNES context 8819b94acceSBarry Smith . func - function evaluation routine 882044dda88SLois Curfman McInnes . ctx - optional user-defined context for private data for the 883044dda88SLois Curfman McInnes gradient evaluation routine (may be PETSC_NULL) 884c7afd0dbSLois Curfman McInnes - r - vector to store gradient value 885fee21e36SBarry Smith 8869b94acceSBarry Smith Calling sequence of func: 8878d76a1e5SLois Curfman McInnes $ func (SNES, Vec x, Vec g, void *ctx); 8889b94acceSBarry Smith 889c7afd0dbSLois Curfman McInnes + x - input vector 8909b94acceSBarry Smith . g - gradient vector 891c7afd0dbSLois Curfman McInnes - ctx - optional user-defined gradient context 8929b94acceSBarry Smith 8939b94acceSBarry Smith Notes: 8949b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 8959b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 8969b94acceSBarry Smith SNESSetFunction(). 8979b94acceSBarry Smith 89836851e7fSLois Curfman McInnes Level: beginner 89936851e7fSLois Curfman McInnes 9009b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 9019b94acceSBarry Smith 902a86d99e1SLois Curfman McInnes .seealso: SNESGetGradient(), SNESComputeGradient(), SNESSetHessian(), 903a86d99e1SLois Curfman McInnes SNESSetMinimizationFunction(), 9049b94acceSBarry Smith @*/ 90574679c65SBarry Smith int SNESSetGradient(SNES snes,Vec r,int (*func)(SNES,Vec,Vec,void*),void *ctx) 9069b94acceSBarry Smith { 9073a40ed3dSBarry Smith PetscFunctionBegin; 90877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 909a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 910a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 911a8c6a408SBarry Smith } 9129b94acceSBarry Smith snes->computefunction = func; 9139b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 9149b94acceSBarry Smith snes->funP = ctx; 9153a40ed3dSBarry Smith PetscFunctionReturn(0); 9169b94acceSBarry Smith } 9179b94acceSBarry Smith 9185615d1e5SSatish Balay #undef __FUNC__ 9195615d1e5SSatish Balay #define __FUNC__ "SNESComputeGradient" 9209b94acceSBarry Smith /*@ 921a86d99e1SLois Curfman McInnes SNESComputeGradient - Computes the gradient that has been set with 922a86d99e1SLois Curfman McInnes SNESSetGradient(). 9239b94acceSBarry Smith 924c7afd0dbSLois Curfman McInnes Collective on SNES 925c7afd0dbSLois Curfman McInnes 9269b94acceSBarry Smith Input Parameters: 927c7afd0dbSLois Curfman McInnes + snes - the SNES context 928c7afd0dbSLois Curfman McInnes - x - input vector 9299b94acceSBarry Smith 9309b94acceSBarry Smith Output Parameter: 9319b94acceSBarry Smith . y - gradient vector 9329b94acceSBarry Smith 9339b94acceSBarry Smith Notes: 9349b94acceSBarry Smith SNESComputeGradient() is valid only for 9359b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 9369b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 937a86d99e1SLois Curfman McInnes 93836851e7fSLois Curfman McInnes SNESComputeGradient() is typically used within minimization 93936851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 94036851e7fSLois Curfman McInnes themselves. 94136851e7fSLois Curfman McInnes 94236851e7fSLois Curfman McInnes Level: developer 94336851e7fSLois Curfman McInnes 944a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, gradient 945a86d99e1SLois Curfman McInnes 946a86d99e1SLois Curfman McInnes .seealso: SNESSetGradient(), SNESGetGradient(), 947a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction(), SNESComputeHessian() 9489b94acceSBarry Smith @*/ 9499b94acceSBarry Smith int SNESComputeGradient(SNES snes,Vec x, Vec y) 9509b94acceSBarry Smith { 9519b94acceSBarry Smith int ierr; 9523a40ed3dSBarry Smith 9533a40ed3dSBarry Smith PetscFunctionBegin; 9543a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 955a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 9563a40ed3dSBarry Smith } 9579b94acceSBarry Smith PLogEventBegin(SNES_GradientEval,snes,x,y,0); 958d64ed03dSBarry Smith PetscStackPush("SNES user gradient function"); 9599b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 960d64ed03dSBarry Smith PetscStackPop; 9619b94acceSBarry Smith PLogEventEnd(SNES_GradientEval,snes,x,y,0); 9623a40ed3dSBarry Smith PetscFunctionReturn(0); 9639b94acceSBarry Smith } 9649b94acceSBarry Smith 9655615d1e5SSatish Balay #undef __FUNC__ 9665615d1e5SSatish Balay #define __FUNC__ "SNESComputeJacobian" 96762fef451SLois Curfman McInnes /*@ 96862fef451SLois Curfman McInnes SNESComputeJacobian - Computes the Jacobian matrix that has been 96962fef451SLois Curfman McInnes set with SNESSetJacobian(). 97062fef451SLois Curfman McInnes 971c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 972c7afd0dbSLois Curfman McInnes 97362fef451SLois Curfman McInnes Input Parameters: 974c7afd0dbSLois Curfman McInnes + snes - the SNES context 975c7afd0dbSLois Curfman McInnes - x - input vector 97662fef451SLois Curfman McInnes 97762fef451SLois Curfman McInnes Output Parameters: 978c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 97962fef451SLois Curfman McInnes . B - optional preconditioning matrix 980c7afd0dbSLois Curfman McInnes - flag - flag indicating matrix structure 981fee21e36SBarry Smith 98262fef451SLois Curfman McInnes Notes: 98362fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 98462fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 98562fef451SLois Curfman McInnes 986dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 987dc5a77f8SLois Curfman McInnes flag parameter. 98862fef451SLois Curfman McInnes 98962fef451SLois Curfman McInnes SNESComputeJacobian() is valid only for SNES_NONLINEAR_EQUATIONS 99062fef451SLois Curfman McInnes methods. An analogous routine for SNES_UNCONSTRAINED_MINIMIZATION 991005c665bSBarry Smith methods is SNESComputeHessian(). 99262fef451SLois Curfman McInnes 99336851e7fSLois Curfman McInnes SNESComputeJacobian() is typically used within nonlinear solver 99436851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 99536851e7fSLois Curfman McInnes themselves. 99636851e7fSLois Curfman McInnes 99736851e7fSLois Curfman McInnes Level: developer 99836851e7fSLois Curfman McInnes 99962fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 100062fef451SLois Curfman McInnes 100162fef451SLois Curfman McInnes .seealso: SNESSetJacobian(), SLESSetOperators() 100262fef451SLois Curfman McInnes @*/ 10039b94acceSBarry Smith int SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 10049b94acceSBarry Smith { 10059b94acceSBarry Smith int ierr; 10063a40ed3dSBarry Smith 10073a40ed3dSBarry Smith PetscFunctionBegin; 10083a40ed3dSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1009a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 10103a40ed3dSBarry Smith } 10113a40ed3dSBarry Smith if (!snes->computejacobian) PetscFunctionReturn(0); 10129b94acceSBarry Smith PLogEventBegin(SNES_JacobianEval,snes,X,*A,*B); 1013c4fc05e7SBarry Smith *flg = DIFFERENT_NONZERO_PATTERN; 1014d64ed03dSBarry Smith PetscStackPush("SNES user Jacobian function"); 10159b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 1016d64ed03dSBarry Smith PetscStackPop; 10179b94acceSBarry Smith PLogEventEnd(SNES_JacobianEval,snes,X,*A,*B); 10186d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 101977c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 102077c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 10213a40ed3dSBarry Smith PetscFunctionReturn(0); 10229b94acceSBarry Smith } 10239b94acceSBarry Smith 10245615d1e5SSatish Balay #undef __FUNC__ 10255615d1e5SSatish Balay #define __FUNC__ "SNESComputeHessian" 102662fef451SLois Curfman McInnes /*@ 102762fef451SLois Curfman McInnes SNESComputeHessian - Computes the Hessian matrix that has been 102862fef451SLois Curfman McInnes set with SNESSetHessian(). 102962fef451SLois Curfman McInnes 1030c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1031c7afd0dbSLois Curfman McInnes 103262fef451SLois Curfman McInnes Input Parameters: 1033c7afd0dbSLois Curfman McInnes + snes - the SNES context 1034c7afd0dbSLois Curfman McInnes - x - input vector 103562fef451SLois Curfman McInnes 103662fef451SLois Curfman McInnes Output Parameters: 1037c7afd0dbSLois Curfman McInnes + A - Hessian matrix 103862fef451SLois Curfman McInnes . B - optional preconditioning matrix 1039c7afd0dbSLois Curfman McInnes - flag - flag indicating matrix structure 1040fee21e36SBarry Smith 104162fef451SLois Curfman McInnes Notes: 104262fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 104362fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 104462fef451SLois Curfman McInnes 1045dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 1046dc5a77f8SLois Curfman McInnes flag parameter. 104762fef451SLois Curfman McInnes 104862fef451SLois Curfman McInnes SNESComputeHessian() is valid only for 104962fef451SLois Curfman McInnes SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 105062fef451SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods is SNESComputeJacobian(). 105162fef451SLois Curfman McInnes 105236851e7fSLois Curfman McInnes SNESComputeHessian() is typically used within minimization 105336851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 105436851e7fSLois Curfman McInnes themselves. 105536851e7fSLois Curfman McInnes 105636851e7fSLois Curfman McInnes Level: developer 105736851e7fSLois Curfman McInnes 105862fef451SLois Curfman McInnes .keywords: SNES, compute, Hessian, matrix 105962fef451SLois Curfman McInnes 1060a86d99e1SLois Curfman McInnes .seealso: SNESSetHessian(), SLESSetOperators(), SNESComputeGradient(), 1061a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction() 106262fef451SLois Curfman McInnes @*/ 106362fef451SLois Curfman McInnes int SNESComputeHessian(SNES snes,Vec x,Mat *A,Mat *B,MatStructure *flag) 10649b94acceSBarry Smith { 10659b94acceSBarry Smith int ierr; 10663a40ed3dSBarry Smith 10673a40ed3dSBarry Smith PetscFunctionBegin; 10683a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1069a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 10703a40ed3dSBarry Smith } 10713a40ed3dSBarry Smith if (!snes->computejacobian) PetscFunctionReturn(0); 107262fef451SLois Curfman McInnes PLogEventBegin(SNES_HessianEval,snes,x,*A,*B); 10730f4a323eSLois Curfman McInnes *flag = DIFFERENT_NONZERO_PATTERN; 1074d64ed03dSBarry Smith PetscStackPush("SNES user Hessian function"); 107562fef451SLois Curfman McInnes ierr = (*snes->computejacobian)(snes,x,A,B,flag,snes->jacP); CHKERRQ(ierr); 1076d64ed03dSBarry Smith PetscStackPop; 107762fef451SLois Curfman McInnes PLogEventEnd(SNES_HessianEval,snes,x,*A,*B); 10780f4a323eSLois Curfman McInnes /* make sure user returned a correct Jacobian and preconditioner */ 107977c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 108077c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 10813a40ed3dSBarry Smith PetscFunctionReturn(0); 10829b94acceSBarry Smith } 10839b94acceSBarry Smith 10845615d1e5SSatish Balay #undef __FUNC__ 1085d4bb536fSBarry Smith #define __FUNC__ "SNESSetJacobian" 10869b94acceSBarry Smith /*@C 10879b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 1088044dda88SLois Curfman McInnes location to store the matrix. 10899b94acceSBarry Smith 1090c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1091c7afd0dbSLois Curfman McInnes 10929b94acceSBarry Smith Input Parameters: 1093c7afd0dbSLois Curfman McInnes + snes - the SNES context 10949b94acceSBarry Smith . A - Jacobian matrix 10959b94acceSBarry Smith . B - preconditioner matrix (usually same as the Jacobian) 10969b94acceSBarry Smith . func - Jacobian evaluation routine 1097c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 10982cd2dfdcSLois Curfman McInnes Jacobian evaluation routine (may be PETSC_NULL) 10999b94acceSBarry Smith 11009b94acceSBarry Smith Calling sequence of func: 11018d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 11029b94acceSBarry Smith 1103c7afd0dbSLois Curfman McInnes + x - input vector 11049b94acceSBarry Smith . A - Jacobian matrix 11059b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1106ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 1107ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 1108c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined Jacobian context 11099b94acceSBarry Smith 11109b94acceSBarry Smith Notes: 1111dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 11122cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1113ac21db08SLois Curfman McInnes 1114ac21db08SLois Curfman McInnes The routine func() takes Mat * as the matrix arguments rather than Mat. 11159b94acceSBarry Smith This allows the Jacobian evaluation routine to replace A and/or B with a 11169b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 11179b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 11189b94acceSBarry Smith throughout the global iterations. 11199b94acceSBarry Smith 112036851e7fSLois Curfman McInnes Level: beginner 112136851e7fSLois Curfman McInnes 11229b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 11239b94acceSBarry Smith 1124ac21db08SLois Curfman McInnes .seealso: SLESSetOperators(), SNESSetFunction() 11259b94acceSBarry Smith @*/ 11269b94acceSBarry Smith int SNESSetJacobian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 11279b94acceSBarry Smith MatStructure*,void*),void *ctx) 11289b94acceSBarry Smith { 11293a40ed3dSBarry Smith PetscFunctionBegin; 113077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1131a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1132a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 1133a8c6a408SBarry Smith } 11349b94acceSBarry Smith snes->computejacobian = func; 11359b94acceSBarry Smith snes->jacP = ctx; 11369b94acceSBarry Smith snes->jacobian = A; 11379b94acceSBarry Smith snes->jacobian_pre = B; 11383a40ed3dSBarry Smith PetscFunctionReturn(0); 11399b94acceSBarry Smith } 114062fef451SLois Curfman McInnes 11415615d1e5SSatish Balay #undef __FUNC__ 1142d4bb536fSBarry Smith #define __FUNC__ "SNESGetJacobian" 1143b4fd4287SBarry Smith /*@ 1144b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 1145b4fd4287SBarry Smith provided context for evaluating the Jacobian. 1146b4fd4287SBarry Smith 1147c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 1148c7afd0dbSLois Curfman McInnes 1149b4fd4287SBarry Smith Input Parameter: 1150b4fd4287SBarry Smith . snes - the nonlinear solver context 1151b4fd4287SBarry Smith 1152b4fd4287SBarry Smith Output Parameters: 1153c7afd0dbSLois Curfman McInnes + A - location to stash Jacobian matrix (or PETSC_NULL) 1154b4fd4287SBarry Smith . B - location to stash preconditioner matrix (or PETSC_NULL) 1155c7afd0dbSLois Curfman McInnes - ctx - location to stash Jacobian ctx (or PETSC_NULL) 1156fee21e36SBarry Smith 115736851e7fSLois Curfman McInnes Level: advanced 115836851e7fSLois Curfman McInnes 1159b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian() 1160b4fd4287SBarry Smith @*/ 1161b4fd4287SBarry Smith int SNESGetJacobian(SNES snes,Mat *A,Mat *B, void **ctx) 1162b4fd4287SBarry Smith { 11633a40ed3dSBarry Smith PetscFunctionBegin; 11643a40ed3dSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1165a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 11663a40ed3dSBarry Smith } 1167b4fd4287SBarry Smith if (A) *A = snes->jacobian; 1168b4fd4287SBarry Smith if (B) *B = snes->jacobian_pre; 1169b4fd4287SBarry Smith if (ctx) *ctx = snes->jacP; 11703a40ed3dSBarry Smith PetscFunctionReturn(0); 1171b4fd4287SBarry Smith } 1172b4fd4287SBarry Smith 11735615d1e5SSatish Balay #undef __FUNC__ 1174d4bb536fSBarry Smith #define __FUNC__ "SNESSetHessian" 11759b94acceSBarry Smith /*@C 11769b94acceSBarry Smith SNESSetHessian - Sets the function to compute Hessian as well as the 1177044dda88SLois Curfman McInnes location to store the matrix. 11789b94acceSBarry Smith 1179c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1180c7afd0dbSLois Curfman McInnes 11819b94acceSBarry Smith Input Parameters: 1182c7afd0dbSLois Curfman McInnes + snes - the SNES context 11839b94acceSBarry Smith . A - Hessian matrix 11849b94acceSBarry Smith . B - preconditioner matrix (usually same as the Hessian) 11859b94acceSBarry Smith . func - Jacobian evaluation routine 1186c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 1187313e4042SLois Curfman McInnes Hessian evaluation routine (may be PETSC_NULL) 11889b94acceSBarry Smith 11899b94acceSBarry Smith Calling sequence of func: 11908d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 11919b94acceSBarry Smith 1192c7afd0dbSLois Curfman McInnes + x - input vector 11939b94acceSBarry Smith . A - Hessian matrix 11949b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1195ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 1196ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 1197c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined Hessian context 11989b94acceSBarry Smith 11999b94acceSBarry Smith Notes: 1200dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 12012cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1202ac21db08SLois Curfman McInnes 12039b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 12049b94acceSBarry Smith This allows the Hessian evaluation routine to replace A and/or B with a 12059b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 12069b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 12079b94acceSBarry Smith throughout the global iterations. 12089b94acceSBarry Smith 120936851e7fSLois Curfman McInnes Level: beginner 121036851e7fSLois Curfman McInnes 12119b94acceSBarry Smith .keywords: SNES, nonlinear, set, Hessian, matrix 12129b94acceSBarry Smith 1213ac21db08SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESSetGradient(), SLESSetOperators() 12149b94acceSBarry Smith @*/ 12159b94acceSBarry Smith int SNESSetHessian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 12169b94acceSBarry Smith MatStructure*,void*),void *ctx) 12179b94acceSBarry Smith { 12183a40ed3dSBarry Smith PetscFunctionBegin; 121977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1220d4bb536fSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1221a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 1222d4bb536fSBarry Smith } 12239b94acceSBarry Smith snes->computejacobian = func; 12249b94acceSBarry Smith snes->jacP = ctx; 12259b94acceSBarry Smith snes->jacobian = A; 12269b94acceSBarry Smith snes->jacobian_pre = B; 12273a40ed3dSBarry Smith PetscFunctionReturn(0); 12289b94acceSBarry Smith } 12299b94acceSBarry Smith 12305615d1e5SSatish Balay #undef __FUNC__ 1231d4bb536fSBarry Smith #define __FUNC__ "SNESGetHessian" 123262fef451SLois Curfman McInnes /*@ 123362fef451SLois Curfman McInnes SNESGetHessian - Returns the Hessian matrix and optionally the user 123462fef451SLois Curfman McInnes provided context for evaluating the Hessian. 123562fef451SLois Curfman McInnes 1236c7afd0dbSLois Curfman McInnes Not Collective, but Mat object is parallel if SNES object is parallel 1237c7afd0dbSLois Curfman McInnes 123862fef451SLois Curfman McInnes Input Parameter: 123962fef451SLois Curfman McInnes . snes - the nonlinear solver context 124062fef451SLois Curfman McInnes 124162fef451SLois Curfman McInnes Output Parameters: 1242c7afd0dbSLois Curfman McInnes + A - location to stash Hessian matrix (or PETSC_NULL) 124362fef451SLois Curfman McInnes . B - location to stash preconditioner matrix (or PETSC_NULL) 1244c7afd0dbSLois Curfman McInnes - ctx - location to stash Hessian ctx (or PETSC_NULL) 1245fee21e36SBarry Smith 124636851e7fSLois Curfman McInnes Level: advanced 124736851e7fSLois Curfman McInnes 124862fef451SLois Curfman McInnes .seealso: SNESSetHessian(), SNESComputeHessian() 1249c7afd0dbSLois Curfman McInnes 1250c7afd0dbSLois Curfman McInnes .keywords: SNES, get, Hessian 125162fef451SLois Curfman McInnes @*/ 125262fef451SLois Curfman McInnes int SNESGetHessian(SNES snes,Mat *A,Mat *B, void **ctx) 125362fef451SLois Curfman McInnes { 12543a40ed3dSBarry Smith PetscFunctionBegin; 12553a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION){ 1256a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 12573a40ed3dSBarry Smith } 125862fef451SLois Curfman McInnes if (A) *A = snes->jacobian; 125962fef451SLois Curfman McInnes if (B) *B = snes->jacobian_pre; 126062fef451SLois Curfman McInnes if (ctx) *ctx = snes->jacP; 12613a40ed3dSBarry Smith PetscFunctionReturn(0); 126262fef451SLois Curfman McInnes } 126362fef451SLois Curfman McInnes 12649b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */ 12659b94acceSBarry Smith 12665615d1e5SSatish Balay #undef __FUNC__ 12675615d1e5SSatish Balay #define __FUNC__ "SNESSetUp" 12689b94acceSBarry Smith /*@ 12699b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 1270272ac6f2SLois Curfman McInnes of a nonlinear solver. 12719b94acceSBarry Smith 1272fee21e36SBarry Smith Collective on SNES 1273fee21e36SBarry Smith 1274c7afd0dbSLois Curfman McInnes Input Parameters: 1275c7afd0dbSLois Curfman McInnes + snes - the SNES context 1276c7afd0dbSLois Curfman McInnes - x - the solution vector 1277c7afd0dbSLois Curfman McInnes 1278272ac6f2SLois Curfman McInnes Notes: 1279272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 1280272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 1281272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 1282272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 1283272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 1284272ac6f2SLois Curfman McInnes 128536851e7fSLois Curfman McInnes Level: advanced 128636851e7fSLois Curfman McInnes 12879b94acceSBarry Smith .keywords: SNES, nonlinear, setup 12889b94acceSBarry Smith 12899b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 12909b94acceSBarry Smith @*/ 12918ddd3da0SLois Curfman McInnes int SNESSetUp(SNES snes,Vec x) 12929b94acceSBarry Smith { 1293272ac6f2SLois Curfman McInnes int ierr, flg; 12943a40ed3dSBarry Smith 12953a40ed3dSBarry Smith PetscFunctionBegin; 129677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 129777c4ece6SBarry Smith PetscValidHeaderSpecific(x,VEC_COOKIE); 12988ddd3da0SLois Curfman McInnes snes->vec_sol = snes->vec_sol_always = x; 12999b94acceSBarry Smith 1300c1f60f51SBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_mf_operator", &flg); CHKERRQ(ierr); 1301c1f60f51SBarry Smith /* 1302c1f60f51SBarry Smith This version replaces the user provided Jacobian matrix with a 1303dfa02198SLois Curfman McInnes matrix-free version but still employs the user-provided preconditioner matrix 1304c1f60f51SBarry Smith */ 1305c1f60f51SBarry Smith if (flg) { 1306c1f60f51SBarry Smith Mat J; 1307e24b481bSBarry Smith ierr = MatCreateSNESFDMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1308c1f60f51SBarry Smith PLogObjectParent(snes,J); 1309c1f60f51SBarry Smith snes->mfshell = J; 1310c1f60f51SBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1311c1f60f51SBarry Smith snes->jacobian = J; 1312c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Jacobian routines\n"); 1313d64ed03dSBarry Smith } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 1314c1f60f51SBarry Smith snes->jacobian = J; 1315c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Hessian routines\n"); 1316d4bb536fSBarry Smith } else { 1317a8c6a408SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"Method class doesn't support matrix-free operator option"); 1318d4bb536fSBarry Smith } 1319e24b481bSBarry Smith ierr = MatSNESFDMFSetFromOptions(J);CHKERRQ(ierr); 1320c1f60f51SBarry Smith } 1321272ac6f2SLois Curfman McInnes ierr = OptionsHasName(snes->prefix,"-snes_mf", &flg); CHKERRQ(ierr); 1322c1f60f51SBarry Smith /* 1323dfa02198SLois Curfman McInnes This version replaces both the user-provided Jacobian and the user- 1324c1f60f51SBarry Smith provided preconditioner matrix with the default matrix free version. 1325c1f60f51SBarry Smith */ 132631615d04SLois Curfman McInnes if (flg) { 1327272ac6f2SLois Curfman McInnes Mat J; 1328e24b481bSBarry Smith ierr = MatCreateSNESFDMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1329272ac6f2SLois Curfman McInnes PLogObjectParent(snes,J); 1330272ac6f2SLois Curfman McInnes snes->mfshell = J; 133183e56afdSLois Curfman McInnes if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 133283e56afdSLois Curfman McInnes ierr = SNESSetJacobian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 133394a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Jacobian routines\n"); 1334d64ed03dSBarry Smith } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 133583e56afdSLois Curfman McInnes ierr = SNESSetHessian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 133694a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Hessian routines\n"); 1337d4bb536fSBarry Smith } else { 1338a8c6a408SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"Method class doesn't support matrix-free option"); 1339d4bb536fSBarry Smith } 1340e24b481bSBarry Smith ierr = MatSNESFDMFSetFromOptions(J);CHKERRQ(ierr); 1341272ac6f2SLois Curfman McInnes } 13429b94acceSBarry Smith if ((snes->method_class == SNES_NONLINEAR_EQUATIONS)) { 1343a8c6a408SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetFunction() first"); 1344a8c6a408SBarry Smith if (!snes->computefunction) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetFunction() first"); 1345a8c6a408SBarry Smith if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetJacobian() first"); 1346a8c6a408SBarry Smith if (snes->vec_func == snes->vec_sol) { 1347a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_IDN,0,"Solution vector cannot be function vector"); 1348a8c6a408SBarry Smith } 1349a703fe33SLois Curfman McInnes 1350a703fe33SLois Curfman McInnes /* Set the KSP stopping criterion to use the Eisenstat-Walker method */ 135182bf6240SBarry Smith if (snes->ksp_ewconv && PetscStrcmp(snes->type_name,SNES_EQ_TR)) { 1352a703fe33SLois Curfman McInnes SLES sles; KSP ksp; 1353a703fe33SLois Curfman McInnes ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 1354a703fe33SLois Curfman McInnes ierr = SLESGetKSP(sles,&ksp); CHKERRQ(ierr); 1355a703fe33SLois Curfman McInnes ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private, 1356a703fe33SLois Curfman McInnes (void *)snes); CHKERRQ(ierr); 1357a703fe33SLois Curfman McInnes } 13589b94acceSBarry Smith } else if ((snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION)) { 1359a8c6a408SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetGradient() first"); 1360a8c6a408SBarry Smith if (!snes->computefunction) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetGradient() first"); 1361a8c6a408SBarry Smith if (!snes->computeumfunction) { 1362a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetMinimizationFunction() first"); 1363a8c6a408SBarry Smith } 1364a8c6a408SBarry Smith if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetHessian() first"); 1365d4bb536fSBarry Smith } else { 1366a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Unknown method class"); 1367d4bb536fSBarry Smith } 1368a703fe33SLois Curfman McInnes if (snes->setup) {ierr = (*snes->setup)(snes); CHKERRQ(ierr);} 136982bf6240SBarry Smith snes->setupcalled = 1; 13703a40ed3dSBarry Smith PetscFunctionReturn(0); 13719b94acceSBarry Smith } 13729b94acceSBarry Smith 13735615d1e5SSatish Balay #undef __FUNC__ 1374d4bb536fSBarry Smith #define __FUNC__ "SNESDestroy" 13759b94acceSBarry Smith /*@C 13769b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 13779b94acceSBarry Smith with SNESCreate(). 13789b94acceSBarry Smith 1379c7afd0dbSLois Curfman McInnes Collective on SNES 1380c7afd0dbSLois Curfman McInnes 13819b94acceSBarry Smith Input Parameter: 13829b94acceSBarry Smith . snes - the SNES context 13839b94acceSBarry Smith 138436851e7fSLois Curfman McInnes Level: beginner 138536851e7fSLois Curfman McInnes 13869b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 13879b94acceSBarry Smith 138863a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 13899b94acceSBarry Smith @*/ 13909b94acceSBarry Smith int SNESDestroy(SNES snes) 13919b94acceSBarry Smith { 13929b94acceSBarry Smith int ierr; 13933a40ed3dSBarry Smith 13943a40ed3dSBarry Smith PetscFunctionBegin; 139577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 13963a40ed3dSBarry Smith if (--snes->refct > 0) PetscFunctionReturn(0); 1397d4bb536fSBarry Smith 1398e1311b90SBarry Smith if (snes->destroy) {ierr = (*(snes)->destroy)(snes); CHKERRQ(ierr);} 13990452661fSBarry Smith if (snes->kspconvctx) PetscFree(snes->kspconvctx); 1400522c5e43SBarry Smith if (snes->mfshell) {ierr = MatDestroy(snes->mfshell);CHKERRQ(ierr);} 14019b94acceSBarry Smith ierr = SLESDestroy(snes->sles); CHKERRQ(ierr); 1402522c5e43SBarry Smith if (snes->xmonitor) {ierr = SNESLGMonitorDestroy(snes->xmonitor);CHKERRQ(ierr);} 1403522c5e43SBarry Smith if (snes->vwork) {ierr = VecDestroyVecs(snes->vwork,snes->nvwork);CHKERRQ(ierr);} 14049b94acceSBarry Smith PLogObjectDestroy((PetscObject)snes); 14050452661fSBarry Smith PetscHeaderDestroy((PetscObject)snes); 14063a40ed3dSBarry Smith PetscFunctionReturn(0); 14079b94acceSBarry Smith } 14089b94acceSBarry Smith 14099b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 14109b94acceSBarry Smith 14115615d1e5SSatish Balay #undef __FUNC__ 14125615d1e5SSatish Balay #define __FUNC__ "SNESSetTolerances" 14139b94acceSBarry Smith /*@ 1414d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 14159b94acceSBarry Smith 1416c7afd0dbSLois Curfman McInnes Collective on SNES 1417c7afd0dbSLois Curfman McInnes 14189b94acceSBarry Smith Input Parameters: 1419c7afd0dbSLois Curfman McInnes + snes - the SNES context 142033174efeSLois Curfman McInnes . atol - absolute convergence tolerance 142133174efeSLois Curfman McInnes . rtol - relative convergence tolerance 142233174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 142333174efeSLois Curfman McInnes of the change in the solution between steps 142433174efeSLois Curfman McInnes . maxit - maximum number of iterations 1425c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1426fee21e36SBarry Smith 142733174efeSLois Curfman McInnes Options Database Keys: 1428c7afd0dbSLois Curfman McInnes + -snes_atol <atol> - Sets atol 1429c7afd0dbSLois Curfman McInnes . -snes_rtol <rtol> - Sets rtol 1430c7afd0dbSLois Curfman McInnes . -snes_stol <stol> - Sets stol 1431c7afd0dbSLois Curfman McInnes . -snes_max_it <maxit> - Sets maxit 1432c7afd0dbSLois Curfman McInnes - -snes_max_funcs <maxf> - Sets maxf 14339b94acceSBarry Smith 1434d7a720efSLois Curfman McInnes Notes: 14359b94acceSBarry Smith The default maximum number of iterations is 50. 14369b94acceSBarry Smith The default maximum number of function evaluations is 1000. 14379b94acceSBarry Smith 143836851e7fSLois Curfman McInnes Level: intermediate 143936851e7fSLois Curfman McInnes 144033174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances 14419b94acceSBarry Smith 1442d7a720efSLois Curfman McInnes .seealso: SNESSetTrustRegionTolerance(), SNESSetMinimizationFunctionTolerance() 14439b94acceSBarry Smith @*/ 1444d7a720efSLois Curfman McInnes int SNESSetTolerances(SNES snes,double atol,double rtol,double stol,int maxit,int maxf) 14459b94acceSBarry Smith { 14463a40ed3dSBarry Smith PetscFunctionBegin; 144777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1448d7a720efSLois Curfman McInnes if (atol != PETSC_DEFAULT) snes->atol = atol; 1449d7a720efSLois Curfman McInnes if (rtol != PETSC_DEFAULT) snes->rtol = rtol; 1450d7a720efSLois Curfman McInnes if (stol != PETSC_DEFAULT) snes->xtol = stol; 1451d7a720efSLois Curfman McInnes if (maxit != PETSC_DEFAULT) snes->max_its = maxit; 1452d7a720efSLois Curfman McInnes if (maxf != PETSC_DEFAULT) snes->max_funcs = maxf; 14533a40ed3dSBarry Smith PetscFunctionReturn(0); 14549b94acceSBarry Smith } 14559b94acceSBarry Smith 14565615d1e5SSatish Balay #undef __FUNC__ 14575615d1e5SSatish Balay #define __FUNC__ "SNESGetTolerances" 14589b94acceSBarry Smith /*@ 145933174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 146033174efeSLois Curfman McInnes 1461c7afd0dbSLois Curfman McInnes Not Collective 1462c7afd0dbSLois Curfman McInnes 146333174efeSLois Curfman McInnes Input Parameters: 1464c7afd0dbSLois Curfman McInnes + snes - the SNES context 146533174efeSLois Curfman McInnes . atol - absolute convergence tolerance 146633174efeSLois Curfman McInnes . rtol - relative convergence tolerance 146733174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 146833174efeSLois Curfman McInnes of the change in the solution between steps 146933174efeSLois Curfman McInnes . maxit - maximum number of iterations 1470c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1471fee21e36SBarry Smith 147233174efeSLois Curfman McInnes Notes: 147333174efeSLois Curfman McInnes The user can specify PETSC_NULL for any parameter that is not needed. 147433174efeSLois Curfman McInnes 147536851e7fSLois Curfman McInnes Level: intermediate 147636851e7fSLois Curfman McInnes 147733174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances 147833174efeSLois Curfman McInnes 147933174efeSLois Curfman McInnes .seealso: SNESSetTolerances() 148033174efeSLois Curfman McInnes @*/ 148133174efeSLois Curfman McInnes int SNESGetTolerances(SNES snes,double *atol,double *rtol,double *stol,int *maxit,int *maxf) 148233174efeSLois Curfman McInnes { 14833a40ed3dSBarry Smith PetscFunctionBegin; 148433174efeSLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 148533174efeSLois Curfman McInnes if (atol) *atol = snes->atol; 148633174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 148733174efeSLois Curfman McInnes if (stol) *stol = snes->xtol; 148833174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 148933174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 14903a40ed3dSBarry Smith PetscFunctionReturn(0); 149133174efeSLois Curfman McInnes } 149233174efeSLois Curfman McInnes 14935615d1e5SSatish Balay #undef __FUNC__ 14945615d1e5SSatish Balay #define __FUNC__ "SNESSetTrustRegionTolerance" 149533174efeSLois Curfman McInnes /*@ 14969b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 14979b94acceSBarry Smith 1498fee21e36SBarry Smith Collective on SNES 1499fee21e36SBarry Smith 1500c7afd0dbSLois Curfman McInnes Input Parameters: 1501c7afd0dbSLois Curfman McInnes + snes - the SNES context 1502c7afd0dbSLois Curfman McInnes - tol - tolerance 1503c7afd0dbSLois Curfman McInnes 15049b94acceSBarry Smith Options Database Key: 1505c7afd0dbSLois Curfman McInnes . -snes_trtol <tol> - Sets tol 15069b94acceSBarry Smith 150736851e7fSLois Curfman McInnes Level: intermediate 150836851e7fSLois Curfman McInnes 15099b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 15109b94acceSBarry Smith 1511d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetMinimizationFunctionTolerance() 15129b94acceSBarry Smith @*/ 15139b94acceSBarry Smith int SNESSetTrustRegionTolerance(SNES snes,double tol) 15149b94acceSBarry Smith { 15153a40ed3dSBarry Smith PetscFunctionBegin; 151677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15179b94acceSBarry Smith snes->deltatol = tol; 15183a40ed3dSBarry Smith PetscFunctionReturn(0); 15199b94acceSBarry Smith } 15209b94acceSBarry Smith 15215615d1e5SSatish Balay #undef __FUNC__ 15225615d1e5SSatish Balay #define __FUNC__ "SNESSetMinimizationFunctionTolerance" 15239b94acceSBarry Smith /*@ 152477c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance - Sets the minimum allowable function tolerance 15259b94acceSBarry Smith for unconstrained minimization solvers. 15269b94acceSBarry Smith 1527fee21e36SBarry Smith Collective on SNES 1528fee21e36SBarry Smith 1529c7afd0dbSLois Curfman McInnes Input Parameters: 1530c7afd0dbSLois Curfman McInnes + snes - the SNES context 1531c7afd0dbSLois Curfman McInnes - ftol - minimum function tolerance 1532c7afd0dbSLois Curfman McInnes 15339b94acceSBarry Smith Options Database Key: 1534c7afd0dbSLois Curfman McInnes . -snes_fmin <ftol> - Sets ftol 15359b94acceSBarry Smith 15369b94acceSBarry Smith Note: 153777c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance() is valid for SNES_UNCONSTRAINED_MINIMIZATION 15389b94acceSBarry Smith methods only. 15399b94acceSBarry Smith 154036851e7fSLois Curfman McInnes Level: intermediate 154136851e7fSLois Curfman McInnes 15429b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimum, convergence, function, tolerance 15439b94acceSBarry Smith 1544d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetTrustRegionTolerance() 15459b94acceSBarry Smith @*/ 154677c4ece6SBarry Smith int SNESSetMinimizationFunctionTolerance(SNES snes,double ftol) 15479b94acceSBarry Smith { 15483a40ed3dSBarry Smith PetscFunctionBegin; 154977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15509b94acceSBarry Smith snes->fmin = ftol; 15513a40ed3dSBarry Smith PetscFunctionReturn(0); 15529b94acceSBarry Smith } 15539b94acceSBarry Smith 15549b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 15559b94acceSBarry Smith 15565615d1e5SSatish Balay #undef __FUNC__ 1557d4bb536fSBarry Smith #define __FUNC__ "SNESSetMonitor" 15589b94acceSBarry Smith /*@C 15595cd90555SBarry Smith SNESSetMonitor - Sets an ADDITIONAL function that is to be used at every 15609b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 15619b94acceSBarry Smith progress. 15629b94acceSBarry Smith 1563fee21e36SBarry Smith Collective on SNES 1564fee21e36SBarry Smith 1565c7afd0dbSLois Curfman McInnes Input Parameters: 1566c7afd0dbSLois Curfman McInnes + snes - the SNES context 1567c7afd0dbSLois Curfman McInnes . func - monitoring routine 1568c7afd0dbSLois Curfman McInnes - mctx - [optional] user-defined context for private data for the 1569c7afd0dbSLois Curfman McInnes monitor routine (may be PETSC_NULL) 15709b94acceSBarry Smith 1571c7afd0dbSLois Curfman McInnes Calling sequence of func: 1572*40a0c1c6SLois Curfman McInnes $ int func(SNES snes,int its, double norm,void *mctx) 1573c7afd0dbSLois Curfman McInnes 1574c7afd0dbSLois Curfman McInnes + snes - the SNES context 1575c7afd0dbSLois Curfman McInnes . its - iteration number 1576c7afd0dbSLois Curfman McInnes . norm - 2-norm function value (may be estimated) 1577c7afd0dbSLois Curfman McInnes for SNES_NONLINEAR_EQUATIONS methods 1578*40a0c1c6SLois Curfman McInnes . norm - 2-norm gradient value (may be estimated) 1579c7afd0dbSLois Curfman McInnes for SNES_UNCONSTRAINED_MINIMIZATION methods 1580*40a0c1c6SLois Curfman McInnes - mctx - [optional] monitoring context 15819b94acceSBarry Smith 15829665c990SLois Curfman McInnes Options Database Keys: 1583c7afd0dbSLois Curfman McInnes + -snes_monitor - sets SNESDefaultMonitor() 1584c7afd0dbSLois Curfman McInnes . -snes_xmonitor - sets line graph monitor, 1585c7afd0dbSLois Curfman McInnes uses SNESLGMonitorCreate() 1586c7afd0dbSLois Curfman McInnes _ -snes_cancelmonitors - cancels all monitors that have 1587c7afd0dbSLois Curfman McInnes been hardwired into a code by 1588c7afd0dbSLois Curfman McInnes calls to SNESSetMonitor(), but 1589c7afd0dbSLois Curfman McInnes does not cancel those set via 1590c7afd0dbSLois Curfman McInnes the options database. 15919665c990SLois Curfman McInnes 1592639f9d9dSBarry Smith Notes: 15936bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 15946bc08f3fSLois Curfman McInnes SNESSetMonitor() multiple times; all will be called in the 15956bc08f3fSLois Curfman McInnes order in which they were set. 1596639f9d9dSBarry Smith 159736851e7fSLois Curfman McInnes Level: intermediate 159836851e7fSLois Curfman McInnes 15999b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 16009b94acceSBarry Smith 16015cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESClearMonitor() 16029b94acceSBarry Smith @*/ 160374679c65SBarry Smith int SNESSetMonitor( SNES snes, int (*func)(SNES,int,double,void*),void *mctx ) 16049b94acceSBarry Smith { 16053a40ed3dSBarry Smith PetscFunctionBegin; 1606639f9d9dSBarry Smith if (snes->numbermonitors >= MAXSNESMONITORS) { 1607a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Too many monitors set"); 1608639f9d9dSBarry Smith } 1609639f9d9dSBarry Smith 1610639f9d9dSBarry Smith snes->monitor[snes->numbermonitors] = func; 1611639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 16123a40ed3dSBarry Smith PetscFunctionReturn(0); 16139b94acceSBarry Smith } 16149b94acceSBarry Smith 16155615d1e5SSatish Balay #undef __FUNC__ 16165cd90555SBarry Smith #define __FUNC__ "SNESClearMonitor" 16175cd90555SBarry Smith /*@C 16185cd90555SBarry Smith SNESClearMonitor - Clears all the monitor functions for a SNES object. 16195cd90555SBarry Smith 1620c7afd0dbSLois Curfman McInnes Collective on SNES 1621c7afd0dbSLois Curfman McInnes 16225cd90555SBarry Smith Input Parameters: 16235cd90555SBarry Smith . snes - the SNES context 16245cd90555SBarry Smith 16255cd90555SBarry Smith Options Database: 1626c7afd0dbSLois Curfman McInnes . -snes_cancelmonitors - cancels all monitors that have been hardwired 1627c7afd0dbSLois Curfman McInnes into a code by calls to SNESSetMonitor(), but does not cancel those 1628c7afd0dbSLois Curfman McInnes set via the options database 16295cd90555SBarry Smith 16305cd90555SBarry Smith Notes: 16315cd90555SBarry Smith There is no way to clear one specific monitor from a SNES object. 16325cd90555SBarry Smith 163336851e7fSLois Curfman McInnes Level: intermediate 163436851e7fSLois Curfman McInnes 16355cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor 16365cd90555SBarry Smith 16375cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESSetMonitor() 16385cd90555SBarry Smith @*/ 16395cd90555SBarry Smith int SNESClearMonitor( SNES snes ) 16405cd90555SBarry Smith { 16415cd90555SBarry Smith PetscFunctionBegin; 16425cd90555SBarry Smith snes->numbermonitors = 0; 16435cd90555SBarry Smith PetscFunctionReturn(0); 16445cd90555SBarry Smith } 16455cd90555SBarry Smith 16465cd90555SBarry Smith #undef __FUNC__ 1647d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceTest" 16489b94acceSBarry Smith /*@C 16499b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 16509b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 16519b94acceSBarry Smith 1652fee21e36SBarry Smith Collective on SNES 1653fee21e36SBarry Smith 1654c7afd0dbSLois Curfman McInnes Input Parameters: 1655c7afd0dbSLois Curfman McInnes + snes - the SNES context 1656c7afd0dbSLois Curfman McInnes . func - routine to test for convergence 1657c7afd0dbSLois Curfman McInnes - cctx - [optional] context for private data for the convergence routine 1658c7afd0dbSLois Curfman McInnes (may be PETSC_NULL) 16599b94acceSBarry Smith 1660c7afd0dbSLois Curfman McInnes Calling sequence of func: 16618d76a1e5SLois Curfman McInnes $ int func (SNES snes,double xnorm,double gnorm,double f,void *cctx) 1662c7afd0dbSLois Curfman McInnes 1663c7afd0dbSLois Curfman McInnes + snes - the SNES context 1664c7afd0dbSLois Curfman McInnes . cctx - [optional] convergence context 1665c7afd0dbSLois Curfman McInnes . xnorm - 2-norm of current iterate 1666c7afd0dbSLois Curfman McInnes . gnorm - 2-norm of current step (SNES_NONLINEAR_EQUATIONS methods) 1667c7afd0dbSLois Curfman McInnes . f - 2-norm of function (SNES_NONLINEAR_EQUATIONS methods) 1668c7afd0dbSLois Curfman McInnes . gnorm - 2-norm of current gradient (SNES_UNCONSTRAINED_MINIMIZATION methods) 1669c7afd0dbSLois Curfman McInnes - f - function value (SNES_UNCONSTRAINED_MINIMIZATION methods) 16709b94acceSBarry Smith 167136851e7fSLois Curfman McInnes Level: advanced 167236851e7fSLois Curfman McInnes 16739b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 16749b94acceSBarry Smith 167540191667SLois Curfman McInnes .seealso: SNESConverged_EQ_LS(), SNESConverged_EQ_TR(), 167640191667SLois Curfman McInnes SNESConverged_UM_LS(), SNESConverged_UM_TR() 16779b94acceSBarry Smith @*/ 167874679c65SBarry Smith int SNESSetConvergenceTest(SNES snes,int (*func)(SNES,double,double,double,void*),void *cctx) 16799b94acceSBarry Smith { 16803a40ed3dSBarry Smith PetscFunctionBegin; 16819b94acceSBarry Smith (snes)->converged = func; 16829b94acceSBarry Smith (snes)->cnvP = cctx; 16833a40ed3dSBarry Smith PetscFunctionReturn(0); 16849b94acceSBarry Smith } 16859b94acceSBarry Smith 16865615d1e5SSatish Balay #undef __FUNC__ 1687d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceHistory" 1688c9005455SLois Curfman McInnes /*@ 1689c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 1690c9005455SLois Curfman McInnes 1691fee21e36SBarry Smith Collective on SNES 1692fee21e36SBarry Smith 1693c7afd0dbSLois Curfman McInnes Input Parameters: 1694c7afd0dbSLois Curfman McInnes + snes - iterative context obtained from SNESCreate() 1695c7afd0dbSLois Curfman McInnes . a - array to hold history 1696c7afd0dbSLois Curfman McInnes - na - size of a 1697c7afd0dbSLois Curfman McInnes 1698c9005455SLois Curfman McInnes Notes: 1699c9005455SLois Curfman McInnes If set, this array will contain the function norms (for 1700c9005455SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods) or gradient norms 1701c9005455SLois Curfman McInnes (for SNES_UNCONSTRAINED_MINIMIZATION methods) computed 1702c9005455SLois Curfman McInnes at each step. 1703c9005455SLois Curfman McInnes 1704c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 1705c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 1706c9005455SLois Curfman McInnes during the section of code that is being timed. 1707c9005455SLois Curfman McInnes 170836851e7fSLois Curfman McInnes Level: intermediate 170936851e7fSLois Curfman McInnes 1710c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history 1711c9005455SLois Curfman McInnes @*/ 1712c9005455SLois Curfman McInnes int SNESSetConvergenceHistory(SNES snes, double *a, int na) 1713c9005455SLois Curfman McInnes { 17143a40ed3dSBarry Smith PetscFunctionBegin; 1715c9005455SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 1716c9005455SLois Curfman McInnes if (na) PetscValidScalarPointer(a); 1717c9005455SLois Curfman McInnes snes->conv_hist = a; 1718c9005455SLois Curfman McInnes snes->conv_hist_size = na; 17193a40ed3dSBarry Smith PetscFunctionReturn(0); 1720c9005455SLois Curfman McInnes } 1721c9005455SLois Curfman McInnes 1722c9005455SLois Curfman McInnes #undef __FUNC__ 17235615d1e5SSatish Balay #define __FUNC__ "SNESScaleStep_Private" 17249b94acceSBarry Smith /* 17259b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 17269b94acceSBarry Smith positive parameter delta. 17279b94acceSBarry Smith 17289b94acceSBarry Smith Input Parameters: 1729c7afd0dbSLois Curfman McInnes + snes - the SNES context 17309b94acceSBarry Smith . y - approximate solution of linear system 17319b94acceSBarry Smith . fnorm - 2-norm of current function 1732c7afd0dbSLois Curfman McInnes - delta - trust region size 17339b94acceSBarry Smith 17349b94acceSBarry Smith Output Parameters: 1735c7afd0dbSLois Curfman McInnes + gpnorm - predicted function norm at the new point, assuming local 17369b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 17379b94acceSBarry Smith region, and exceeds zero otherwise. 1738c7afd0dbSLois Curfman McInnes - ynorm - 2-norm of the step 17399b94acceSBarry Smith 17409b94acceSBarry Smith Note: 174140191667SLois Curfman McInnes For non-trust region methods such as SNES_EQ_LS, the parameter delta 17429b94acceSBarry Smith is set to be the maximum allowable step size. 17439b94acceSBarry Smith 17449b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 17459b94acceSBarry Smith */ 17469b94acceSBarry Smith int SNESScaleStep_Private(SNES snes,Vec y,double *fnorm,double *delta, 17479b94acceSBarry Smith double *gpnorm,double *ynorm) 17489b94acceSBarry Smith { 17499b94acceSBarry Smith double norm; 17509b94acceSBarry Smith Scalar cnorm; 17513a40ed3dSBarry Smith int ierr; 17523a40ed3dSBarry Smith 17533a40ed3dSBarry Smith PetscFunctionBegin; 17543a40ed3dSBarry Smith ierr = VecNorm(y,NORM_2, &norm );CHKERRQ(ierr); 17559b94acceSBarry Smith if (norm > *delta) { 17569b94acceSBarry Smith norm = *delta/norm; 17579b94acceSBarry Smith *gpnorm = (1.0 - norm)*(*fnorm); 17589b94acceSBarry Smith cnorm = norm; 17599b94acceSBarry Smith VecScale( &cnorm, y ); 17609b94acceSBarry Smith *ynorm = *delta; 17619b94acceSBarry Smith } else { 17629b94acceSBarry Smith *gpnorm = 0.0; 17639b94acceSBarry Smith *ynorm = norm; 17649b94acceSBarry Smith } 17653a40ed3dSBarry Smith PetscFunctionReturn(0); 17669b94acceSBarry Smith } 17679b94acceSBarry Smith 17685615d1e5SSatish Balay #undef __FUNC__ 17695615d1e5SSatish Balay #define __FUNC__ "SNESSolve" 17709b94acceSBarry Smith /*@ 17719b94acceSBarry Smith SNESSolve - Solves a nonlinear system. Call SNESSolve after calling 177263a78c88SLois Curfman McInnes SNESCreate() and optional routines of the form SNESSetXXX(). 17739b94acceSBarry Smith 1774c7afd0dbSLois Curfman McInnes Collective on SNES 1775c7afd0dbSLois Curfman McInnes 1776b2002411SLois Curfman McInnes Input Parameters: 1777c7afd0dbSLois Curfman McInnes + snes - the SNES context 1778c7afd0dbSLois Curfman McInnes - x - the solution vector 17799b94acceSBarry Smith 17809b94acceSBarry Smith Output Parameter: 1781b2002411SLois Curfman McInnes . its - number of iterations until termination 17829b94acceSBarry Smith 1783b2002411SLois Curfman McInnes Notes: 17848ddd3da0SLois Curfman McInnes The user should initialize the vector, x, with the initial guess 17858ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 17868ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 17878ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 17888ddd3da0SLois Curfman McInnes 178936851e7fSLois Curfman McInnes Level: beginner 179036851e7fSLois Curfman McInnes 17919b94acceSBarry Smith .keywords: SNES, nonlinear, solve 17929b94acceSBarry Smith 179363a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESDestroy() 17949b94acceSBarry Smith @*/ 17958ddd3da0SLois Curfman McInnes int SNESSolve(SNES snes,Vec x,int *its) 17969b94acceSBarry Smith { 17973c7409f5SSatish Balay int ierr, flg; 1798052efed2SBarry Smith 17993a40ed3dSBarry Smith PetscFunctionBegin; 180077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 180174679c65SBarry Smith PetscValidIntPointer(its); 180282bf6240SBarry Smith if (!snes->setupcalled) {ierr = SNESSetUp(snes,x); CHKERRQ(ierr);} 1803c4fc05e7SBarry Smith else {snes->vec_sol = snes->vec_sol_always = x;} 18049b94acceSBarry Smith PLogEventBegin(SNES_Solve,snes,0,0,0); 1805c96a6f78SLois Curfman McInnes snes->nfuncs = 0; snes->linear_its = 0; snes->nfailures = 0; 18069b94acceSBarry Smith ierr = (*(snes)->solve)(snes,its); CHKERRQ(ierr); 18079b94acceSBarry Smith PLogEventEnd(SNES_Solve,snes,0,0,0); 18083c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-snes_view", &flg); CHKERRQ(ierr); 18096d4a8577SBarry Smith if (flg) { ierr = SNESView(snes,VIEWER_STDOUT_WORLD); CHKERRQ(ierr); } 18103a40ed3dSBarry Smith PetscFunctionReturn(0); 18119b94acceSBarry Smith } 18129b94acceSBarry Smith 18139b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 18149b94acceSBarry Smith 18155615d1e5SSatish Balay #undef __FUNC__ 18165615d1e5SSatish Balay #define __FUNC__ "SNESSetType" 181782bf6240SBarry Smith /*@C 18184b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 18199b94acceSBarry Smith 1820fee21e36SBarry Smith Collective on SNES 1821fee21e36SBarry Smith 1822c7afd0dbSLois Curfman McInnes Input Parameters: 1823c7afd0dbSLois Curfman McInnes + snes - the SNES context 1824c7afd0dbSLois Curfman McInnes - method - a known method 1825c7afd0dbSLois Curfman McInnes 1826c7afd0dbSLois Curfman McInnes Options Database Key: 1827c7afd0dbSLois Curfman McInnes . -snes_type <method> - Sets the method; use -help for a list 1828c7afd0dbSLois Curfman McInnes of available methods (for instance, ls or tr) 1829ae12b187SLois Curfman McInnes 18309b94acceSBarry Smith Notes: 18319b94acceSBarry Smith See "petsc/include/snes.h" for available methods (for instance) 183236851e7fSLois Curfman McInnes + SNES_EQ_LS - Newton's method with line search 1833c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 1834c7afd0dbSLois Curfman McInnes . SNES_EQ_TR - Newton's method with trust region 1835c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 1836c7afd0dbSLois Curfman McInnes . SNES_UM_TR - Newton's method with trust region 1837c7afd0dbSLois Curfman McInnes (unconstrained minimization) 183836851e7fSLois Curfman McInnes - SNES_UM_LS - Newton's method with line search 1839c7afd0dbSLois Curfman McInnes (unconstrained minimization) 18409b94acceSBarry Smith 1841ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 1842ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 1843ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 1844ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 1845ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 1846ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 1847ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 1848ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 1849ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 185036851e7fSLois Curfman McInnes appropriate method. In other words, this routine is not for beginners. 185136851e7fSLois Curfman McInnes 185236851e7fSLois Curfman McInnes Level: intermediate 1853a703fe33SLois Curfman McInnes 1854f536c699SSatish Balay .keywords: SNES, set, method 18559b94acceSBarry Smith @*/ 18564b0e389bSBarry Smith int SNESSetType(SNES snes,SNESType method) 18579b94acceSBarry Smith { 185884cb2905SBarry Smith int ierr; 18599b94acceSBarry Smith int (*r)(SNES); 18603a40ed3dSBarry Smith 18613a40ed3dSBarry Smith PetscFunctionBegin; 186277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 186382bf6240SBarry Smith 18643f1db9ecSBarry Smith if (PetscTypeCompare(snes->type_name,method)) PetscFunctionReturn(0); 186582bf6240SBarry Smith 186682bf6240SBarry Smith if (snes->setupcalled) { 1867e1311b90SBarry Smith ierr = (*(snes)->destroy)(snes);CHKERRQ(ierr); 186882bf6240SBarry Smith snes->data = 0; 186982bf6240SBarry Smith } 18707d1a2b2bSBarry Smith 18719b94acceSBarry Smith /* Get the function pointers for the iterative method requested */ 187282bf6240SBarry Smith if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL); CHKERRQ(ierr);} 187382bf6240SBarry Smith 1874488ecbafSBarry Smith ierr = FListFind(snes->comm, SNESList, method,(int (**)(void *)) &r );CHKERRQ(ierr); 187582bf6240SBarry Smith 1876596552b5SBarry Smith if (!r) SETERRQ1(1,1,"Unable to find requested SNES type %s",method); 18771548b14aSBarry Smith 18780452661fSBarry Smith if (snes->data) PetscFree(snes->data); 187982bf6240SBarry Smith snes->data = 0; 18803a40ed3dSBarry Smith ierr = (*r)(snes); CHKERRQ(ierr); 188182bf6240SBarry Smith 188282bf6240SBarry Smith if (snes->type_name) PetscFree(snes->type_name); 188382bf6240SBarry Smith snes->type_name = (char *) PetscMalloc((PetscStrlen(method)+1)*sizeof(char));CHKPTRQ(snes->type_name); 188482bf6240SBarry Smith PetscStrcpy(snes->type_name,method); 188582bf6240SBarry Smith snes->set_method_called = 1; 188682bf6240SBarry Smith 18873a40ed3dSBarry Smith PetscFunctionReturn(0); 18889b94acceSBarry Smith } 18899b94acceSBarry Smith 1890a847f771SSatish Balay 18919b94acceSBarry Smith /* --------------------------------------------------------------------- */ 18925615d1e5SSatish Balay #undef __FUNC__ 1893d4bb536fSBarry Smith #define __FUNC__ "SNESRegisterDestroy" 18949b94acceSBarry Smith /*@C 18959b94acceSBarry Smith SNESRegisterDestroy - Frees the list of nonlinear solvers that were 18969b94acceSBarry Smith registered by SNESRegister(). 18979b94acceSBarry Smith 1898fee21e36SBarry Smith Not Collective 1899fee21e36SBarry Smith 190036851e7fSLois Curfman McInnes Level: advanced 190136851e7fSLois Curfman McInnes 19029b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy 19039b94acceSBarry Smith 19049b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll() 19059b94acceSBarry Smith @*/ 1906cf256101SBarry Smith int SNESRegisterDestroy(void) 19079b94acceSBarry Smith { 190882bf6240SBarry Smith int ierr; 190982bf6240SBarry Smith 19103a40ed3dSBarry Smith PetscFunctionBegin; 191182bf6240SBarry Smith if (SNESList) { 1912488ecbafSBarry Smith ierr = FListDestroy( SNESList );CHKERRQ(ierr); 191382bf6240SBarry Smith SNESList = 0; 19149b94acceSBarry Smith } 191584cb2905SBarry Smith SNESRegisterAllCalled = 0; 19163a40ed3dSBarry Smith PetscFunctionReturn(0); 19179b94acceSBarry Smith } 19189b94acceSBarry Smith 19195615d1e5SSatish Balay #undef __FUNC__ 1920d4bb536fSBarry Smith #define __FUNC__ "SNESGetType" 19219b94acceSBarry Smith /*@C 19229a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 19239b94acceSBarry Smith 1924c7afd0dbSLois Curfman McInnes Not Collective 1925c7afd0dbSLois Curfman McInnes 19269b94acceSBarry Smith Input Parameter: 19274b0e389bSBarry Smith . snes - nonlinear solver context 19289b94acceSBarry Smith 19299b94acceSBarry Smith Output Parameter: 193082bf6240SBarry Smith . method - SNES method (a charactor string) 19319b94acceSBarry Smith 193236851e7fSLois Curfman McInnes Level: intermediate 193336851e7fSLois Curfman McInnes 19349b94acceSBarry Smith .keywords: SNES, nonlinear, get, method, name 19359b94acceSBarry Smith @*/ 193682bf6240SBarry Smith int SNESGetType(SNES snes, SNESType *method) 19379b94acceSBarry Smith { 19383a40ed3dSBarry Smith PetscFunctionBegin; 193982bf6240SBarry Smith *method = snes->type_name; 19403a40ed3dSBarry Smith PetscFunctionReturn(0); 19419b94acceSBarry Smith } 19429b94acceSBarry Smith 19435615d1e5SSatish Balay #undef __FUNC__ 1944d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolution" 19459b94acceSBarry Smith /*@C 19469b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 19479b94acceSBarry Smith stored. 19489b94acceSBarry Smith 1949c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 1950c7afd0dbSLois Curfman McInnes 19519b94acceSBarry Smith Input Parameter: 19529b94acceSBarry Smith . snes - the SNES context 19539b94acceSBarry Smith 19549b94acceSBarry Smith Output Parameter: 19559b94acceSBarry Smith . x - the solution 19569b94acceSBarry Smith 195736851e7fSLois Curfman McInnes Level: advanced 195836851e7fSLois Curfman McInnes 19599b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 19609b94acceSBarry Smith 1961a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESGetGradient(), SNESGetSolutionUpdate() 19629b94acceSBarry Smith @*/ 19639b94acceSBarry Smith int SNESGetSolution(SNES snes,Vec *x) 19649b94acceSBarry Smith { 19653a40ed3dSBarry Smith PetscFunctionBegin; 196677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 19679b94acceSBarry Smith *x = snes->vec_sol_always; 19683a40ed3dSBarry Smith PetscFunctionReturn(0); 19699b94acceSBarry Smith } 19709b94acceSBarry Smith 19715615d1e5SSatish Balay #undef __FUNC__ 1972d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolutionUpdate" 19739b94acceSBarry Smith /*@C 19749b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 19759b94acceSBarry Smith stored. 19769b94acceSBarry Smith 1977c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 1978c7afd0dbSLois Curfman McInnes 19799b94acceSBarry Smith Input Parameter: 19809b94acceSBarry Smith . snes - the SNES context 19819b94acceSBarry Smith 19829b94acceSBarry Smith Output Parameter: 19839b94acceSBarry Smith . x - the solution update 19849b94acceSBarry Smith 198536851e7fSLois Curfman McInnes Level: advanced 198636851e7fSLois Curfman McInnes 19879b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 19889b94acceSBarry Smith 19899b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction 19909b94acceSBarry Smith @*/ 19919b94acceSBarry Smith int SNESGetSolutionUpdate(SNES snes,Vec *x) 19929b94acceSBarry Smith { 19933a40ed3dSBarry Smith PetscFunctionBegin; 199477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 19959b94acceSBarry Smith *x = snes->vec_sol_update_always; 19963a40ed3dSBarry Smith PetscFunctionReturn(0); 19979b94acceSBarry Smith } 19989b94acceSBarry Smith 19995615d1e5SSatish Balay #undef __FUNC__ 2000d4bb536fSBarry Smith #define __FUNC__ "SNESGetFunction" 20019b94acceSBarry Smith /*@C 20023638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 20039b94acceSBarry Smith 2004c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2005c7afd0dbSLois Curfman McInnes 20069b94acceSBarry Smith Input Parameter: 20079b94acceSBarry Smith . snes - the SNES context 20089b94acceSBarry Smith 20099b94acceSBarry Smith Output Parameter: 20103638b69dSLois Curfman McInnes . r - the function 20119b94acceSBarry Smith 20129b94acceSBarry Smith Notes: 20139b94acceSBarry Smith SNESGetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only 20149b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 20159b94acceSBarry Smith SNESGetMinimizationFunction() and SNESGetGradient(); 20169b94acceSBarry Smith 201736851e7fSLois Curfman McInnes Level: advanced 201836851e7fSLois Curfman McInnes 2019a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 20209b94acceSBarry Smith 202131615d04SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution(), SNESGetMinimizationFunction(), 202231615d04SLois Curfman McInnes SNESGetGradient() 20239b94acceSBarry Smith @*/ 20249b94acceSBarry Smith int SNESGetFunction(SNES snes,Vec *r) 20259b94acceSBarry Smith { 20263a40ed3dSBarry Smith PetscFunctionBegin; 202777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2028a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 2029a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 2030a8c6a408SBarry Smith } 20319b94acceSBarry Smith *r = snes->vec_func_always; 20323a40ed3dSBarry Smith PetscFunctionReturn(0); 20339b94acceSBarry Smith } 20349b94acceSBarry Smith 20355615d1e5SSatish Balay #undef __FUNC__ 2036d4bb536fSBarry Smith #define __FUNC__ "SNESGetGradient" 20379b94acceSBarry Smith /*@C 20383638b69dSLois Curfman McInnes SNESGetGradient - Returns the vector where the gradient is stored. 20399b94acceSBarry Smith 2040c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2041c7afd0dbSLois Curfman McInnes 20429b94acceSBarry Smith Input Parameter: 20439b94acceSBarry Smith . snes - the SNES context 20449b94acceSBarry Smith 20459b94acceSBarry Smith Output Parameter: 20469b94acceSBarry Smith . r - the gradient 20479b94acceSBarry Smith 20489b94acceSBarry Smith Notes: 20499b94acceSBarry Smith SNESGetGradient() is valid for SNES_UNCONSTRAINED_MINIMIZATION methods 20509b94acceSBarry Smith only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 20519b94acceSBarry Smith SNESGetFunction(). 20529b94acceSBarry Smith 205336851e7fSLois Curfman McInnes Level: advanced 205436851e7fSLois Curfman McInnes 20559b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient 20569b94acceSBarry Smith 205731615d04SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESGetSolution(), SNESGetFunction() 20589b94acceSBarry Smith @*/ 20599b94acceSBarry Smith int SNESGetGradient(SNES snes,Vec *r) 20609b94acceSBarry Smith { 20613a40ed3dSBarry Smith PetscFunctionBegin; 206277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 20633a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 2064a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 20653a40ed3dSBarry Smith } 20669b94acceSBarry Smith *r = snes->vec_func_always; 20673a40ed3dSBarry Smith PetscFunctionReturn(0); 20689b94acceSBarry Smith } 20699b94acceSBarry Smith 20705615d1e5SSatish Balay #undef __FUNC__ 2071d4bb536fSBarry Smith #define __FUNC__ "SNESGetMinimizationFunction" 20729b94acceSBarry Smith /*@ 20739b94acceSBarry Smith SNESGetMinimizationFunction - Returns the scalar function value for 20749b94acceSBarry Smith unconstrained minimization problems. 20759b94acceSBarry Smith 2076c7afd0dbSLois Curfman McInnes Not Collective 2077c7afd0dbSLois Curfman McInnes 20789b94acceSBarry Smith Input Parameter: 20799b94acceSBarry Smith . snes - the SNES context 20809b94acceSBarry Smith 20819b94acceSBarry Smith Output Parameter: 20829b94acceSBarry Smith . r - the function 20839b94acceSBarry Smith 20849b94acceSBarry Smith Notes: 20859b94acceSBarry Smith SNESGetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 20869b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 20879b94acceSBarry Smith SNESGetFunction(). 20889b94acceSBarry Smith 208936851e7fSLois Curfman McInnes Level: advanced 209036851e7fSLois Curfman McInnes 20919b94acceSBarry Smith .keywords: SNES, nonlinear, get, function 20929b94acceSBarry Smith 209331615d04SLois Curfman McInnes .seealso: SNESGetGradient(), SNESGetSolution(), SNESGetFunction() 20949b94acceSBarry Smith @*/ 20959b94acceSBarry Smith int SNESGetMinimizationFunction(SNES snes,double *r) 20969b94acceSBarry Smith { 20973a40ed3dSBarry Smith PetscFunctionBegin; 209877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 209974679c65SBarry Smith PetscValidScalarPointer(r); 21003a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 2101a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 21023a40ed3dSBarry Smith } 21039b94acceSBarry Smith *r = snes->fc; 21043a40ed3dSBarry Smith PetscFunctionReturn(0); 21059b94acceSBarry Smith } 21069b94acceSBarry Smith 21075615d1e5SSatish Balay #undef __FUNC__ 2108d4bb536fSBarry Smith #define __FUNC__ "SNESSetOptionsPrefix" 21093c7409f5SSatish Balay /*@C 21103c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 2111d850072dSLois Curfman McInnes SNES options in the database. 21123c7409f5SSatish Balay 2113fee21e36SBarry Smith Collective on SNES 2114fee21e36SBarry Smith 2115c7afd0dbSLois Curfman McInnes Input Parameter: 2116c7afd0dbSLois Curfman McInnes + snes - the SNES context 2117c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2118c7afd0dbSLois Curfman McInnes 2119d850072dSLois Curfman McInnes Notes: 2120a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2121c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2122d850072dSLois Curfman McInnes 212336851e7fSLois Curfman McInnes Level: advanced 212436851e7fSLois Curfman McInnes 21253c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 2126a86d99e1SLois Curfman McInnes 2127a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 21283c7409f5SSatish Balay @*/ 21293c7409f5SSatish Balay int SNESSetOptionsPrefix(SNES snes,char *prefix) 21303c7409f5SSatish Balay { 21313c7409f5SSatish Balay int ierr; 21323c7409f5SSatish Balay 21333a40ed3dSBarry Smith PetscFunctionBegin; 213477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2135639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 21363c7409f5SSatish Balay ierr = SLESSetOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 21373a40ed3dSBarry Smith PetscFunctionReturn(0); 21383c7409f5SSatish Balay } 21393c7409f5SSatish Balay 21405615d1e5SSatish Balay #undef __FUNC__ 2141d4bb536fSBarry Smith #define __FUNC__ "SNESAppendOptionsPrefix" 21423c7409f5SSatish Balay /*@C 2143f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 2144d850072dSLois Curfman McInnes SNES options in the database. 21453c7409f5SSatish Balay 2146fee21e36SBarry Smith Collective on SNES 2147fee21e36SBarry Smith 2148c7afd0dbSLois Curfman McInnes Input Parameters: 2149c7afd0dbSLois Curfman McInnes + snes - the SNES context 2150c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2151c7afd0dbSLois Curfman McInnes 2152d850072dSLois Curfman McInnes Notes: 2153a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2154c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2155d850072dSLois Curfman McInnes 215636851e7fSLois Curfman McInnes Level: advanced 215736851e7fSLois Curfman McInnes 21583c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 2159a86d99e1SLois Curfman McInnes 2160a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 21613c7409f5SSatish Balay @*/ 21623c7409f5SSatish Balay int SNESAppendOptionsPrefix(SNES snes,char *prefix) 21633c7409f5SSatish Balay { 21643c7409f5SSatish Balay int ierr; 21653c7409f5SSatish Balay 21663a40ed3dSBarry Smith PetscFunctionBegin; 216777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2168639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 21693c7409f5SSatish Balay ierr = SLESAppendOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 21703a40ed3dSBarry Smith PetscFunctionReturn(0); 21713c7409f5SSatish Balay } 21723c7409f5SSatish Balay 21735615d1e5SSatish Balay #undef __FUNC__ 2174d4bb536fSBarry Smith #define __FUNC__ "SNESGetOptionsPrefix" 2175c83590e2SSatish Balay /*@ 21763c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 21773c7409f5SSatish Balay SNES options in the database. 21783c7409f5SSatish Balay 2179c7afd0dbSLois Curfman McInnes Not Collective 2180c7afd0dbSLois Curfman McInnes 21813c7409f5SSatish Balay Input Parameter: 21823c7409f5SSatish Balay . snes - the SNES context 21833c7409f5SSatish Balay 21843c7409f5SSatish Balay Output Parameter: 21853c7409f5SSatish Balay . prefix - pointer to the prefix string used 21863c7409f5SSatish Balay 218736851e7fSLois Curfman McInnes Level: advanced 218836851e7fSLois Curfman McInnes 21893c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 2190a86d99e1SLois Curfman McInnes 2191a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 21923c7409f5SSatish Balay @*/ 21933c7409f5SSatish Balay int SNESGetOptionsPrefix(SNES snes,char **prefix) 21943c7409f5SSatish Balay { 21953c7409f5SSatish Balay int ierr; 21963c7409f5SSatish Balay 21973a40ed3dSBarry Smith PetscFunctionBegin; 219877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2199639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 22003a40ed3dSBarry Smith PetscFunctionReturn(0); 22013c7409f5SSatish Balay } 22023c7409f5SSatish Balay 2203ca161407SBarry Smith #undef __FUNC__ 2204ca161407SBarry Smith #define __FUNC__ "SNESPrintHelp" 2205ca161407SBarry Smith /*@ 2206ca161407SBarry Smith SNESPrintHelp - Prints all options for the SNES component. 2207ca161407SBarry Smith 2208c7afd0dbSLois Curfman McInnes Collective on SNES 2209c7afd0dbSLois Curfman McInnes 2210ca161407SBarry Smith Input Parameter: 2211ca161407SBarry Smith . snes - the SNES context 2212ca161407SBarry Smith 2213ca161407SBarry Smith Options Database Keys: 2214c7afd0dbSLois Curfman McInnes + -help - Prints SNES options 2215c7afd0dbSLois Curfman McInnes - -h - Prints SNES options 2216ca161407SBarry Smith 221736851e7fSLois Curfman McInnes Level: beginner 221836851e7fSLois Curfman McInnes 2219ca161407SBarry Smith .keywords: SNES, nonlinear, help 2220ca161407SBarry Smith 2221ca161407SBarry Smith .seealso: SNESSetFromOptions() 2222ca161407SBarry Smith @*/ 2223ca161407SBarry Smith int SNESPrintHelp(SNES snes) 2224ca161407SBarry Smith { 2225ca161407SBarry Smith char p[64]; 2226ca161407SBarry Smith SNES_KSP_EW_ConvCtx *kctx; 2227ca161407SBarry Smith int ierr; 2228ca161407SBarry Smith 2229ca161407SBarry Smith PetscFunctionBegin; 2230ca161407SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2231ca161407SBarry Smith 2232ca161407SBarry Smith PetscStrcpy(p,"-"); 2233ca161407SBarry Smith if (snes->prefix) PetscStrcat(p, snes->prefix); 2234ca161407SBarry Smith 2235ca161407SBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 2236ca161407SBarry Smith 223776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm,"SNES options ------------------------------------------------\n"); 2238488ecbafSBarry Smith ierr = FListPrintTypes(snes->comm,stdout,snes->prefix,"snes_type",SNESList);CHKERRQ(ierr); 223976be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_view: view SNES info after each nonlinear solve\n",p); 224076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_max_it <its>: max iterations (default %d)\n",p,snes->max_its); 224176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_max_funcs <maxf>: max function evals (default %d)\n",p,snes->max_funcs); 224276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_stol <stol>: successive step tolerance (default %g)\n",p,snes->xtol); 224376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_atol <atol>: absolute tolerance (default %g)\n",p,snes->atol); 224476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_rtol <rtol>: relative tolerance (default %g)\n",p,snes->rtol); 224576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_trtol <trtol>: trust region parameter tolerance (default %g)\n",p,snes->deltatol); 224676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," SNES Monitoring Options: Choose any of the following\n"); 224776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_cancelmonitors: cancels all monitors hardwired in code\n",p); 224876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_monitor: use default SNES convergence monitor, prints\n\ 2249ca161407SBarry Smith residual norm at each iteration.\n",p); 225076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_smonitor: same as the above, but prints fewer digits of the\n\ 2251ca161407SBarry Smith residual norm for small residual norms. This is useful to conceal\n\ 2252ca161407SBarry Smith meaningless digits that may be different on different machines.\n",p); 225376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_xmonitor [x,y,w,h]: use X graphics convergence monitor\n",p); 2254ca161407SBarry Smith if (snes->type == SNES_NONLINEAR_EQUATIONS) { 225576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2256ca161407SBarry Smith " Options for solving systems of nonlinear equations only:\n"); 225776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fd: use finite differences for Jacobian\n",p); 225876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf: use matrix-free Jacobian\n",p); 225976be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf_operator:use matrix-free Jacobian and user-provided preconditioning matrix\n",p); 2260e24b481bSBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf_ksp_monitor - if using matrix-free multiply then prints h at each KSP iteration\n",p); 226176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_no_convergence_test: Do not test for convergence, always run to SNES max its\n",p); 226276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_ksp_ew_conv: use Eisenstat-Walker computation of KSP rtol. Params are:\n",p); 226376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2264ca161407SBarry Smith " %ssnes_ksp_ew_version <version> (1 or 2, default is %d)\n",p,kctx->version); 226576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2266ca161407SBarry Smith " %ssnes_ksp_ew_rtol0 <rtol0> (0 <= rtol0 < 1, default %g)\n",p,kctx->rtol_0); 226776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2268ca161407SBarry Smith " %ssnes_ksp_ew_rtolmax <rtolmax> (0 <= rtolmax < 1, default %g)\n",p,kctx->rtol_max); 226976be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2270ca161407SBarry Smith " %ssnes_ksp_ew_gamma <gamma> (0 <= gamma <= 1, default %g)\n",p,kctx->gamma); 227176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2272ca161407SBarry Smith " %ssnes_ksp_ew_alpha <alpha> (1 < alpha <= 2, default %g)\n",p,kctx->alpha); 227376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2274ca161407SBarry Smith " %ssnes_ksp_ew_alpha2 <alpha2> (default %g)\n",p,kctx->alpha2); 227576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2276ca161407SBarry Smith " %ssnes_ksp_ew_threshold <threshold> (0 < threshold < 1, default %g)\n",p,kctx->threshold); 2277ca161407SBarry Smith } else if (snes->type == SNES_UNCONSTRAINED_MINIMIZATION) { 227876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2279ca161407SBarry Smith " Options for solving unconstrained minimization problems only:\n"); 228076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fmin <ftol>: minimum function tolerance (default %g)\n",p,snes->fmin); 228176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fd: use finite differences for Hessian\n",p); 228276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf: use matrix-free Hessian\n",p); 2283ca161407SBarry Smith } 228476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," Run program with -help %ssnes_type <method> for help on ",p); 228576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm,"a particular method\n"); 2286ca161407SBarry Smith if (snes->printhelp) { 2287ca161407SBarry Smith ierr = (*snes->printhelp)(snes,p);CHKERRQ(ierr); 2288ca161407SBarry Smith } 2289ca161407SBarry Smith PetscFunctionReturn(0); 2290ca161407SBarry Smith } 22913c7409f5SSatish Balay 2292acb85ae6SSatish Balay /*MC 2293b2002411SLois Curfman McInnes SNESRegister - Adds a method to the nonlinear solver package. 22949b94acceSBarry Smith 2295b2002411SLois Curfman McInnes Synopsis: 2296b2002411SLois Curfman McInnes SNESRegister(char *name_solver,char *path,char *name_create,int (*routine_create)(SNES)) 22979b94acceSBarry Smith 22988d76a1e5SLois Curfman McInnes Not collective 22998d76a1e5SLois Curfman McInnes 2300b2002411SLois Curfman McInnes Input Parameters: 2301c7afd0dbSLois Curfman McInnes + name_solver - name of a new user-defined solver 2302b2002411SLois Curfman McInnes . path - path (either absolute or relative) the library containing this solver 2303b2002411SLois Curfman McInnes . name_create - name of routine to create method context 2304c7afd0dbSLois Curfman McInnes - routine_create - routine to create method context 23059b94acceSBarry Smith 2306b2002411SLois Curfman McInnes Notes: 2307b2002411SLois Curfman McInnes SNESRegister() may be called multiple times to add several user-defined solvers. 23083a40ed3dSBarry Smith 2309b2002411SLois Curfman McInnes If dynamic libraries are used, then the fourth input argument (routine_create) 2310b2002411SLois Curfman McInnes is ignored. 2311b2002411SLois Curfman McInnes 2312b2002411SLois Curfman McInnes Sample usage: 231369225d78SLois Curfman McInnes .vb 2314b2002411SLois Curfman McInnes SNESRegister("my_solver",/home/username/my_lib/lib/libg/solaris/mylib.a, 2315b2002411SLois Curfman McInnes "MySolverCreate",MySolverCreate); 231669225d78SLois Curfman McInnes .ve 2317b2002411SLois Curfman McInnes 2318b2002411SLois Curfman McInnes Then, your solver can be chosen with the procedural interface via 2319b2002411SLois Curfman McInnes $ SNESSetType(snes,"my_solver") 2320b2002411SLois Curfman McInnes or at runtime via the option 2321b2002411SLois Curfman McInnes $ -snes_type my_solver 2322b2002411SLois Curfman McInnes 232336851e7fSLois Curfman McInnes Level: advanced 232436851e7fSLois Curfman McInnes 2325b2002411SLois Curfman McInnes .keywords: SNES, nonlinear, register 2326b2002411SLois Curfman McInnes 2327b2002411SLois Curfman McInnes .seealso: SNESRegisterAll(), SNESRegisterDestroy() 2328b2002411SLois Curfman McInnes M*/ 2329b2002411SLois Curfman McInnes 2330b2002411SLois Curfman McInnes #undef __FUNC__ 2331b2002411SLois Curfman McInnes #define __FUNC__ "SNESRegister_Private" 2332b2002411SLois Curfman McInnes int SNESRegister_Private(char *sname,char *path,char *name,int (*function)(SNES)) 2333b2002411SLois Curfman McInnes { 2334b2002411SLois Curfman McInnes char fullname[256]; 2335b2002411SLois Curfman McInnes int ierr; 2336b2002411SLois Curfman McInnes 2337b2002411SLois Curfman McInnes PetscFunctionBegin; 2338b2002411SLois Curfman McInnes PetscStrcpy(fullname,path); PetscStrcat(fullname,":");PetscStrcat(fullname,name); 2339488ecbafSBarry Smith ierr = FListAdd_Private(&SNESList,sname,fullname, (int (*)(void*))function);CHKERRQ(ierr); 2340b2002411SLois Curfman McInnes PetscFunctionReturn(0); 2341b2002411SLois Curfman McInnes } 2342