1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER 2*d4bb536fSBarry Smith static char vcid[] = "$Id: snes.c,v 1.127 1997/08/13 22:26:08 bsmith Exp bsmith $"; 39b94acceSBarry Smith #endif 49b94acceSBarry Smith 570f55243SBarry Smith #include "src/snes/snesimpl.h" /*I "snes.h" I*/ 6f5eb4b81SSatish Balay #include "src/sys/nreg.h" 79b94acceSBarry Smith #include "pinclude/pviewer.h" 89b94acceSBarry Smith #include <math.h> 99b94acceSBarry Smith 10052efed2SBarry Smith extern int SNESGetTypeFromOptions_Private(SNES,SNESType*,int*); 1133455573SSatish Balay extern int SNESPrintTypes_Private(MPI_Comm,char*,char*); 129b94acceSBarry Smith 1384cb2905SBarry Smith int SNESRegisterAllCalled = 0; 1484cb2905SBarry Smith 155615d1e5SSatish Balay #undef __FUNC__ 16*d4bb536fSBarry Smith #define __FUNC__ "SNESView" 179b94acceSBarry Smith /*@ 189b94acceSBarry Smith SNESView - Prints the SNES data structure. 199b94acceSBarry Smith 209b94acceSBarry Smith Input Parameters: 219b94acceSBarry Smith . SNES - the SNES context 229b94acceSBarry Smith . viewer - visualization context 239b94acceSBarry Smith 249b94acceSBarry Smith Options Database Key: 259b94acceSBarry Smith $ -snes_view : calls SNESView() at end of SNESSolve() 269b94acceSBarry Smith 279b94acceSBarry Smith Notes: 289b94acceSBarry Smith The available visualization contexts include 296d4a8577SBarry Smith $ VIEWER_STDOUT_SELF - standard output (default) 306d4a8577SBarry Smith $ VIEWER_STDOUT_WORLD - synchronized standard 319b94acceSBarry Smith $ output where only the first processor opens 329b94acceSBarry Smith $ the file. All other processors send their 339b94acceSBarry Smith $ data to the first processor to print. 349b94acceSBarry Smith 359b94acceSBarry Smith The user can open alternative vistualization contexts with 36dbb450caSBarry Smith $ ViewerFileOpenASCII() - output to a specified file 379b94acceSBarry Smith 389b94acceSBarry Smith .keywords: SNES, view 399b94acceSBarry Smith 40dbb450caSBarry Smith .seealso: ViewerFileOpenASCII() 419b94acceSBarry Smith @*/ 429b94acceSBarry Smith int SNESView(SNES snes,Viewer viewer) 439b94acceSBarry Smith { 449b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 459b94acceSBarry Smith FILE *fd; 469b94acceSBarry Smith int ierr; 479b94acceSBarry Smith SLES sles; 489b94acceSBarry Smith char *method; 4919bcc07fSBarry Smith ViewerType vtype; 509b94acceSBarry Smith 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); 5619bcc07fSBarry Smith if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER) { 5790ace30eSBarry Smith ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr); 5877c4ece6SBarry Smith PetscFPrintf(snes->comm,fd,"SNES Object:\n"); 594b0e389bSBarry Smith SNESGetType(snes,PETSC_NULL,&method); 6077c4ece6SBarry Smith PetscFPrintf(snes->comm,fd," method: %s\n",method); 619b94acceSBarry Smith if (snes->view) (*snes->view)((PetscObject)snes,viewer); 6277c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 639b94acceSBarry Smith " maximum iterations=%d, maximum function evaluations=%d\n", 649b94acceSBarry Smith snes->max_its,snes->max_funcs); 6577c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 669b94acceSBarry Smith " tolerances: relative=%g, absolute=%g, truncation=%g, solution=%g\n", 679b94acceSBarry Smith snes->rtol, snes->atol, snes->trunctol, snes->xtol); 687a00f4a9SLois Curfman McInnes PetscFPrintf(snes->comm,fd, 697a00f4a9SLois Curfman McInnes " total number of linear solver iterations=%d\n",snes->linear_its); 70ae3c334cSLois Curfman McInnes PetscFPrintf(snes->comm,fd, 7168632166SLois Curfman McInnes " total number of function evaluations=%d\n",snes->nfuncs); 729b94acceSBarry Smith if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) 7377c4ece6SBarry Smith PetscFPrintf(snes->comm,fd," min function tolerance=%g\n",snes->fmin); 749b94acceSBarry Smith if (snes->ksp_ewconv) { 759b94acceSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 769b94acceSBarry Smith if (kctx) { 7777c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 789b94acceSBarry Smith " Eisenstat-Walker computation of KSP relative tolerance (version %d)\n", 799b94acceSBarry Smith kctx->version); 8077c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 819b94acceSBarry Smith " rtol_0=%g, rtol_max=%g, threshold=%g\n",kctx->rtol_0, 829b94acceSBarry Smith kctx->rtol_max,kctx->threshold); 8377c4ece6SBarry Smith PetscFPrintf(snes->comm,fd," gamma=%g, alpha=%g, alpha2=%g\n", 849b94acceSBarry Smith kctx->gamma,kctx->alpha,kctx->alpha2); 859b94acceSBarry Smith } 869b94acceSBarry Smith } 87c30f285eSLois Curfman McInnes } else if (vtype == STRING_VIEWER) { 88c30f285eSLois Curfman McInnes SNESGetType(snes,PETSC_NULL,&method); 89c30f285eSLois Curfman McInnes ViewerStringSPrintf(viewer," %-3.3s",method); 9019bcc07fSBarry Smith } 919b94acceSBarry Smith SNESGetSLES(snes,&sles); 929b94acceSBarry Smith ierr = SLESView(sles,viewer); CHKERRQ(ierr); 939b94acceSBarry Smith return 0; 949b94acceSBarry Smith } 959b94acceSBarry Smith 96639f9d9dSBarry Smith /* 97639f9d9dSBarry Smith We retain a list of functions that also take SNES command 98639f9d9dSBarry Smith line options. These are called at the end SNESSetFromOptions() 99639f9d9dSBarry Smith */ 100639f9d9dSBarry Smith #define MAXSETFROMOPTIONS 5 101639f9d9dSBarry Smith static int numberofsetfromoptions; 102639f9d9dSBarry Smith static int (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES); 103639f9d9dSBarry Smith 1045615d1e5SSatish Balay #undef __FUNC__ 105*d4bb536fSBarry Smith #define __FUNC__ "SNESAddOptionsChecker" 106639f9d9dSBarry Smith /*@ 107639f9d9dSBarry Smith SNESAddOptionsChecker - Adds an additional function to check for SNES options. 108639f9d9dSBarry Smith 109639f9d9dSBarry Smith Input Parameter: 110639f9d9dSBarry Smith . snescheck - function that checks for options 111639f9d9dSBarry Smith 112639f9d9dSBarry Smith .seealso: SNESSetFromOptions() 113639f9d9dSBarry Smith @*/ 114639f9d9dSBarry Smith int SNESAddOptionsChecker(int (*snescheck)(SNES) ) 115639f9d9dSBarry Smith { 116639f9d9dSBarry Smith if (numberofsetfromoptions >= MAXSETFROMOPTIONS) { 117e3372554SBarry Smith SETERRQ(1,0,"Too many options checkers, only 5 allowed"); 118639f9d9dSBarry Smith } 119639f9d9dSBarry Smith 120639f9d9dSBarry Smith othersetfromoptions[numberofsetfromoptions++] = snescheck; 121639f9d9dSBarry Smith return 0; 122639f9d9dSBarry Smith } 123639f9d9dSBarry Smith 1245615d1e5SSatish Balay #undef __FUNC__ 1255615d1e5SSatish Balay #define __FUNC__ "SNESSetFromOptions" 1269b94acceSBarry Smith /*@ 127682d7d0cSBarry Smith SNESSetFromOptions - Sets various SNES and SLES parameters from user options. 1289b94acceSBarry Smith 1299b94acceSBarry Smith Input Parameter: 1309b94acceSBarry Smith . snes - the SNES context 1319b94acceSBarry Smith 13283e2fdc7SBarry Smith Notes: To see all options, run your program with the -help option; 13383e2fdc7SBarry Smith or consult the users manual. 13483e2fdc7SBarry Smith 1359b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 1369b94acceSBarry Smith 137a86d99e1SLois Curfman McInnes .seealso: SNESPrintHelp(), SNESSetOptionsPrefix() 1389b94acceSBarry Smith @*/ 1399b94acceSBarry Smith int SNESSetFromOptions(SNES snes) 1409b94acceSBarry Smith { 1414b0e389bSBarry Smith SNESType method; 1429b94acceSBarry Smith double tmp; 1439b94acceSBarry Smith SLES sles; 14481f57ec7SBarry Smith int ierr, flg,i,loc[4],nmax = 4; 1453c7409f5SSatish Balay int version = PETSC_DEFAULT; 1469b94acceSBarry Smith double rtol_0 = PETSC_DEFAULT; 1479b94acceSBarry Smith double rtol_max = PETSC_DEFAULT; 1489b94acceSBarry Smith double gamma2 = PETSC_DEFAULT; 1499b94acceSBarry Smith double alpha = PETSC_DEFAULT; 1509b94acceSBarry Smith double alpha2 = PETSC_DEFAULT; 1519b94acceSBarry Smith double threshold = PETSC_DEFAULT; 1529b94acceSBarry Smith 15381f57ec7SBarry Smith loc[0] = PETSC_DECIDE; loc[1] = PETSC_DECIDE; loc[2] = 300; loc[3] = 300; 15481f57ec7SBarry Smith 15581f57ec7SBarry Smith 15677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 157e3372554SBarry Smith if (snes->setup_called) SETERRQ(1,0,"Must call prior to SNESSetUp"); 158052efed2SBarry Smith ierr = SNESGetTypeFromOptions_Private(snes,&method,&flg); CHKERRQ(ierr); 159052efed2SBarry Smith if (flg) { 160052efed2SBarry Smith ierr = SNESSetType(snes,method); CHKERRQ(ierr); 1619b94acceSBarry Smith } 1625334005bSBarry Smith else if (!snes->set_method_called) { 1635334005bSBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 16440191667SLois Curfman McInnes ierr = SNESSetType(snes,SNES_EQ_LS); CHKERRQ(ierr); 1655334005bSBarry Smith } 1665334005bSBarry Smith else { 16740191667SLois Curfman McInnes ierr = SNESSetType(snes,SNES_UM_TR); CHKERRQ(ierr); 1685334005bSBarry Smith } 1695334005bSBarry Smith } 1705334005bSBarry Smith 1713c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 1723c7409f5SSatish Balay if (flg) { SNESPrintHelp(snes); } 1733c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_stol",&tmp, &flg); CHKERRQ(ierr); 174d7a720efSLois Curfman McInnes if (flg) {SNESSetTolerances(snes,PETSC_DEFAULT,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT);} 1753c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_atol",&tmp, &flg); CHKERRQ(ierr); 176d7a720efSLois Curfman McInnes if (flg) {SNESSetTolerances(snes,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);} 1773c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_rtol",&tmp, &flg); CHKERRQ(ierr); 178d7a720efSLois Curfman McInnes if (flg) {SNESSetTolerances(snes,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT); } 1793c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_it",&snes->max_its, &flg); CHKERRQ(ierr); 1803c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_funcs",&snes->max_funcs, &flg);CHKERRQ(ierr); 181d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_trtol",&tmp, &flg); CHKERRQ(ierr); 182d7a720efSLois Curfman McInnes if (flg) { SNESSetTrustRegionTolerance(snes,tmp); } 183d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_fmin",&tmp, &flg); CHKERRQ(ierr); 184d7a720efSLois Curfman McInnes if (flg) { SNESSetMinimizationFunctionTolerance(snes,tmp); } 1853c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_ksp_ew_conv", &flg); CHKERRQ(ierr); 1863c7409f5SSatish Balay if (flg) { snes->ksp_ewconv = 1; } 187b18e04deSLois Curfman McInnes ierr = OptionsGetInt(snes->prefix,"-snes_ksp_ew_version",&version,&flg); CHKERRQ(ierr); 188b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtol0",&rtol_0,&flg); CHKERRQ(ierr); 189b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtolmax",&rtol_max,&flg);CHKERRQ(ierr); 190b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_gamma",&gamma2,&flg); CHKERRQ(ierr); 191b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha",&alpha,&flg); CHKERRQ(ierr); 192b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha2",&alpha2,&flg); CHKERRQ(ierr); 193b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_threshold",&threshold,&flg);CHKERRQ(ierr); 19493c39befSBarry Smith 19593c39befSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_no_convergence_test",&flg); CHKERRQ(ierr); 19693c39befSBarry Smith if (flg) {snes->converged = 0;} 19793c39befSBarry Smith 1989b94acceSBarry Smith ierr = SNES_KSP_SetParametersEW(snes,version,rtol_0,rtol_max,gamma2,alpha, 1999b94acceSBarry Smith alpha2,threshold); CHKERRQ(ierr); 2005bbad29bSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_cancelmonitors",&flg); CHKERRQ(ierr); 2015bbad29bSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,0,0);CHKERRQ(ierr);} 2023c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_monitor",&flg); CHKERRQ(ierr); 203639f9d9dSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESDefaultMonitor,0);CHKERRQ(ierr);} 2043c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_smonitor",&flg); CHKERRQ(ierr); 205d31a3109SLois Curfman McInnes if (flg) { 206639f9d9dSBarry Smith ierr = SNESSetMonitor(snes,SNESDefaultSMonitor,0); CHKERRQ(ierr); 207d31a3109SLois Curfman McInnes } 20881f57ec7SBarry Smith ierr = OptionsGetIntArray(snes->prefix,"-snes_xmonitor",loc,&nmax,&flg);CHKERRQ(ierr); 2093c7409f5SSatish Balay if (flg) { 21017699dbbSLois Curfman McInnes int rank = 0; 211d7e8b826SBarry Smith DrawLG lg; 21217699dbbSLois Curfman McInnes MPI_Initialized(&rank); 21317699dbbSLois Curfman McInnes if (rank) MPI_Comm_rank(snes->comm,&rank); 21417699dbbSLois Curfman McInnes if (!rank) { 21581f57ec7SBarry Smith ierr = SNESLGMonitorCreate(0,0,loc[0],loc[1],loc[2],loc[3],&lg); CHKERRQ(ierr); 2169b94acceSBarry Smith ierr = SNESSetMonitor(snes,SNESLGMonitor,(void *)lg); CHKERRQ(ierr); 217f8c826e1SBarry Smith snes->xmonitor = lg; 2189b94acceSBarry Smith PLogObjectParent(snes,lg); 2199b94acceSBarry Smith } 2209b94acceSBarry Smith } 2213c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_fd", &flg); CHKERRQ(ierr); 2223c7409f5SSatish Balay if (flg && snes->method_class == SNES_NONLINEAR_EQUATIONS) { 2239b94acceSBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre, 2249b94acceSBarry Smith SNESDefaultComputeJacobian,snes->funP); CHKERRQ(ierr); 22594a424c1SBarry Smith PLogInfo(snes, 22631615d04SLois Curfman McInnes "SNESSetFromOptions: Setting default finite difference Jacobian matrix\n"); 22731615d04SLois Curfman McInnes } 22831615d04SLois Curfman McInnes else if (flg && snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 22931615d04SLois Curfman McInnes ierr = SNESSetHessian(snes,snes->jacobian,snes->jacobian_pre, 23031615d04SLois Curfman McInnes SNESDefaultComputeHessian,snes->funP); CHKERRQ(ierr); 23194a424c1SBarry Smith PLogInfo(snes, 23231615d04SLois Curfman McInnes "SNESSetFromOptions: Setting default finite difference Hessian matrix\n"); 2339b94acceSBarry Smith } 234639f9d9dSBarry Smith 235639f9d9dSBarry Smith for ( i=0; i<numberofsetfromoptions; i++ ) { 236639f9d9dSBarry Smith ierr = (*othersetfromoptions[i])(snes); CHKERRQ(ierr); 237639f9d9dSBarry Smith } 238639f9d9dSBarry Smith 2399b94acceSBarry Smith ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 2409b94acceSBarry Smith ierr = SLESSetFromOptions(sles); CHKERRQ(ierr); 2419b94acceSBarry Smith if (!snes->setfromoptions) return 0; 2429b94acceSBarry Smith return (*snes->setfromoptions)(snes); 2439b94acceSBarry Smith } 2449b94acceSBarry Smith 2455615d1e5SSatish Balay #undef __FUNC__ 246*d4bb536fSBarry Smith #define __FUNC__ "SNESPrintHelp" 2479b94acceSBarry Smith /*@ 2489b94acceSBarry Smith SNESPrintHelp - Prints all options for the SNES component. 2499b94acceSBarry Smith 2509b94acceSBarry Smith Input Parameter: 2519b94acceSBarry Smith . snes - the SNES context 2529b94acceSBarry Smith 253a703fe33SLois Curfman McInnes Options Database Keys: 254a703fe33SLois Curfman McInnes $ -help, -h 255a703fe33SLois Curfman McInnes 2569b94acceSBarry Smith .keywords: SNES, nonlinear, help 2579b94acceSBarry Smith 258682d7d0cSBarry Smith .seealso: SNESSetFromOptions() 2599b94acceSBarry Smith @*/ 2609b94acceSBarry Smith int SNESPrintHelp(SNES snes) 2619b94acceSBarry Smith { 2626daaf66cSBarry Smith char p[64]; 2636daaf66cSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 2646daaf66cSBarry Smith 26577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2666daaf66cSBarry Smith 2676daaf66cSBarry Smith PetscStrcpy(p,"-"); 2686daaf66cSBarry Smith if (snes->prefix) PetscStrcat(p, snes->prefix); 2696daaf66cSBarry Smith 2706daaf66cSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 2716daaf66cSBarry Smith 272d31a3109SLois Curfman McInnes PetscPrintf(snes->comm,"SNES options ------------------------------------------------\n"); 27333455573SSatish Balay SNESPrintTypes_Private(snes->comm,p,"snes_type"); 27477c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_view: view SNES info after each nonlinear solve\n",p); 275b18e04deSLois Curfman McInnes PetscPrintf(snes->comm," %ssnes_max_it <its>: max iterations (default %d)\n",p,snes->max_its); 276b18e04deSLois Curfman McInnes PetscPrintf(snes->comm," %ssnes_max_funcs <maxf>: max function evals (default %d)\n",p,snes->max_funcs); 277b18e04deSLois Curfman McInnes PetscPrintf(snes->comm," %ssnes_stol <stol>: successive step tolerance (default %g)\n",p,snes->xtol); 278b18e04deSLois Curfman McInnes PetscPrintf(snes->comm," %ssnes_atol <atol>: absolute tolerance (default %g)\n",p,snes->atol); 279b18e04deSLois Curfman McInnes PetscPrintf(snes->comm," %ssnes_rtol <rtol>: relative tolerance (default %g)\n",p,snes->rtol); 280b18e04deSLois Curfman McInnes PetscPrintf(snes->comm," %ssnes_trtol <trtol>: trust region parameter tolerance (default %g)\n",p,snes->deltatol); 2815bbad29bSBarry Smith PetscPrintf(snes->comm," SNES Monitoring Options: Choose any of the following\n"); 2825bbad29bSBarry Smith PetscPrintf(snes->comm," %ssnes_cancelmonitors: cancels all monitors hardwired in code\n",p); 28383e2fdc7SBarry Smith PetscPrintf(snes->comm," %ssnes_monitor: use default SNES convergence monitor, prints\n\ 28483e2fdc7SBarry Smith residual norm at each iteration.\n",p); 28583e2fdc7SBarry Smith PetscPrintf(snes->comm," %ssnes_smonitor: same as the above, but prints fewer digits of the\n\ 28683e2fdc7SBarry Smith residual norm for small residual norms. This is useful to conceal\n\ 28783e2fdc7SBarry Smith meaningless digits that may be different on different machines.\n",p); 288d31a3109SLois Curfman McInnes PetscPrintf(snes->comm," %ssnes_xmonitor [x,y,w,h]: use X graphics convergence monitor\n",p); 289b18e04deSLois Curfman McInnes if (snes->type == SNES_NONLINEAR_EQUATIONS) { 29077c4ece6SBarry Smith PetscPrintf(snes->comm, 291d31a3109SLois Curfman McInnes " Options for solving systems of nonlinear equations only:\n"); 29277c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_fd: use finite differences for Jacobian\n",p); 29377c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_mf: use matrix-free Jacobian\n",p); 294ef1dfb11SLois Curfman McInnes PetscPrintf(snes->comm," %ssnes_mf_operator:use matrix-free Jacobian and user-provided preconditioning matrix\n",p); 29593c39befSBarry Smith PetscPrintf(snes->comm," %ssnes_no_convergence_test: Do not test for convergence, always run to SNES max its\n",p); 29677c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_ksp_ew_conv: use Eisenstat-Walker computation of KSP rtol. Params are:\n",p); 29777c4ece6SBarry Smith PetscPrintf(snes->comm, 298b18e04deSLois Curfman McInnes " %ssnes_ksp_ew_version <version> (1 or 2, default is %d)\n",p,kctx->version); 29977c4ece6SBarry Smith PetscPrintf(snes->comm, 300b18e04deSLois Curfman McInnes " %ssnes_ksp_ew_rtol0 <rtol0> (0 <= rtol0 < 1, default %g)\n",p,kctx->rtol_0); 30177c4ece6SBarry Smith PetscPrintf(snes->comm, 302b18e04deSLois Curfman McInnes " %ssnes_ksp_ew_rtolmax <rtolmax> (0 <= rtolmax < 1, default %g)\n",p,kctx->rtol_max); 30377c4ece6SBarry Smith PetscPrintf(snes->comm, 304b18e04deSLois Curfman McInnes " %ssnes_ksp_ew_gamma <gamma> (0 <= gamma <= 1, default %g)\n",p,kctx->gamma); 30577c4ece6SBarry Smith PetscPrintf(snes->comm, 306b18e04deSLois Curfman McInnes " %ssnes_ksp_ew_alpha <alpha> (1 < alpha <= 2, default %g)\n",p,kctx->alpha); 30777c4ece6SBarry Smith PetscPrintf(snes->comm, 308b18e04deSLois Curfman McInnes " %ssnes_ksp_ew_alpha2 <alpha2> (default %g)\n",p,kctx->alpha2); 30977c4ece6SBarry Smith PetscPrintf(snes->comm, 310b18e04deSLois Curfman McInnes " %ssnes_ksp_ew_threshold <threshold> (0 < threshold < 1, default %g)\n",p,kctx->threshold); 311b18e04deSLois Curfman McInnes } 312b18e04deSLois Curfman McInnes else if (snes->type == SNES_UNCONSTRAINED_MINIMIZATION) { 31377c4ece6SBarry Smith PetscPrintf(snes->comm, 314d31a3109SLois Curfman McInnes " Options for solving unconstrained minimization problems only:\n"); 315b18e04deSLois Curfman McInnes PetscPrintf(snes->comm," %ssnes_fmin <ftol>: minimum function tolerance (default %g)\n",p,snes->fmin); 31677c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_fd: use finite differences for Hessian\n",p); 31777c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_mf: use matrix-free Hessian\n",p); 318b18e04deSLois Curfman McInnes } 3194537a946SLois Curfman McInnes PetscPrintf(snes->comm," Run program with -help %ssnes_type <method> for help on ",p); 32077c4ece6SBarry Smith PetscPrintf(snes->comm,"a particular method\n"); 3216daaf66cSBarry Smith if (snes->printhelp) (*snes->printhelp)(snes,p); 3229b94acceSBarry Smith return 0; 3239b94acceSBarry Smith } 324a847f771SSatish Balay 3255615d1e5SSatish Balay #undef __FUNC__ 326*d4bb536fSBarry Smith #define __FUNC__ "SNESSetApplicationContext" 3279b94acceSBarry Smith /*@ 3289b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 3299b94acceSBarry Smith the nonlinear solvers. 3309b94acceSBarry Smith 3319b94acceSBarry Smith Input Parameters: 3329b94acceSBarry Smith . snes - the SNES context 3339b94acceSBarry Smith . usrP - optional user context 3349b94acceSBarry Smith 3359b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 3369b94acceSBarry Smith 3379b94acceSBarry Smith .seealso: SNESGetApplicationContext() 3389b94acceSBarry Smith @*/ 3399b94acceSBarry Smith int SNESSetApplicationContext(SNES snes,void *usrP) 3409b94acceSBarry Smith { 34177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3429b94acceSBarry Smith snes->user = usrP; 3439b94acceSBarry Smith return 0; 3449b94acceSBarry Smith } 34574679c65SBarry Smith 3465615d1e5SSatish Balay #undef __FUNC__ 347*d4bb536fSBarry Smith #define __FUNC__ "SNESGetApplicationContext" 3489b94acceSBarry Smith /*@C 3499b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 3509b94acceSBarry Smith nonlinear solvers. 3519b94acceSBarry Smith 3529b94acceSBarry Smith Input Parameter: 3539b94acceSBarry Smith . snes - SNES context 3549b94acceSBarry Smith 3559b94acceSBarry Smith Output Parameter: 3569b94acceSBarry Smith . usrP - user context 3579b94acceSBarry Smith 3589b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 3599b94acceSBarry Smith 3609b94acceSBarry Smith .seealso: SNESSetApplicationContext() 3619b94acceSBarry Smith @*/ 3629b94acceSBarry Smith int SNESGetApplicationContext( SNES snes, void **usrP ) 3639b94acceSBarry Smith { 36477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3659b94acceSBarry Smith *usrP = snes->user; 3669b94acceSBarry Smith return 0; 3679b94acceSBarry Smith } 36874679c65SBarry Smith 3695615d1e5SSatish Balay #undef __FUNC__ 370*d4bb536fSBarry Smith #define __FUNC__ "SNESGetIterationNumber" 3719b94acceSBarry Smith /*@ 3729b94acceSBarry Smith SNESGetIterationNumber - Gets the current iteration number of the 3739b94acceSBarry Smith nonlinear solver. 3749b94acceSBarry Smith 3759b94acceSBarry Smith Input Parameter: 3769b94acceSBarry Smith . snes - SNES context 3779b94acceSBarry Smith 3789b94acceSBarry Smith Output Parameter: 3799b94acceSBarry Smith . iter - iteration number 3809b94acceSBarry Smith 3819b94acceSBarry Smith .keywords: SNES, nonlinear, get, iteration, number 3829b94acceSBarry Smith @*/ 3839b94acceSBarry Smith int SNESGetIterationNumber(SNES snes,int* iter) 3849b94acceSBarry Smith { 38577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 38674679c65SBarry Smith PetscValidIntPointer(iter); 3879b94acceSBarry Smith *iter = snes->iter; 3889b94acceSBarry Smith return 0; 3899b94acceSBarry Smith } 39074679c65SBarry Smith 3915615d1e5SSatish Balay #undef __FUNC__ 3925615d1e5SSatish Balay #define __FUNC__ "SNESGetFunctionNorm" 3939b94acceSBarry Smith /*@ 3949b94acceSBarry Smith SNESGetFunctionNorm - Gets the norm of the current function that was set 3959b94acceSBarry Smith with SNESSSetFunction(). 3969b94acceSBarry Smith 3979b94acceSBarry Smith Input Parameter: 3989b94acceSBarry Smith . snes - SNES context 3999b94acceSBarry Smith 4009b94acceSBarry Smith Output Parameter: 4019b94acceSBarry Smith . fnorm - 2-norm of function 4029b94acceSBarry Smith 4039b94acceSBarry Smith Note: 4049b94acceSBarry Smith SNESGetFunctionNorm() is valid for SNES_NONLINEAR_EQUATIONS methods only. 405a86d99e1SLois Curfman McInnes A related routine for SNES_UNCONSTRAINED_MINIMIZATION methods is 406a86d99e1SLois Curfman McInnes SNESGetGradientNorm(). 4079b94acceSBarry Smith 4089b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm 409a86d99e1SLois Curfman McInnes 410a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction() 4119b94acceSBarry Smith @*/ 4129b94acceSBarry Smith int SNESGetFunctionNorm(SNES snes,Scalar *fnorm) 4139b94acceSBarry Smith { 41477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 41574679c65SBarry Smith PetscValidScalarPointer(fnorm); 41674679c65SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 417d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_NONLINEAR_EQUATIONS only"); 41874679c65SBarry Smith } 4199b94acceSBarry Smith *fnorm = snes->norm; 4209b94acceSBarry Smith return 0; 4219b94acceSBarry Smith } 42274679c65SBarry Smith 4235615d1e5SSatish Balay #undef __FUNC__ 4245615d1e5SSatish Balay #define __FUNC__ "SNESGetGradientNorm" 4259b94acceSBarry Smith /*@ 4269b94acceSBarry Smith SNESGetGradientNorm - Gets the norm of the current gradient that was set 4279b94acceSBarry Smith with SNESSSetGradient(). 4289b94acceSBarry Smith 4299b94acceSBarry Smith Input Parameter: 4309b94acceSBarry Smith . snes - SNES context 4319b94acceSBarry Smith 4329b94acceSBarry Smith Output Parameter: 4339b94acceSBarry Smith . fnorm - 2-norm of gradient 4349b94acceSBarry Smith 4359b94acceSBarry Smith Note: 4369b94acceSBarry Smith SNESGetGradientNorm() is valid for SNES_UNCONSTRAINED_MINIMIZATION 437a86d99e1SLois Curfman McInnes methods only. A related routine for SNES_NONLINEAR_EQUATIONS methods 438a86d99e1SLois Curfman McInnes is SNESGetFunctionNorm(). 4399b94acceSBarry Smith 4409b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient, norm 441a86d99e1SLois Curfman McInnes 442a86d99e1SLois Curfman McInnes .seelso: SNESSetGradient() 4439b94acceSBarry Smith @*/ 4449b94acceSBarry Smith int SNESGetGradientNorm(SNES snes,Scalar *gnorm) 4459b94acceSBarry Smith { 44677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 44774679c65SBarry Smith PetscValidScalarPointer(gnorm); 44874679c65SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 449d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 45074679c65SBarry Smith } 4519b94acceSBarry Smith *gnorm = snes->norm; 4529b94acceSBarry Smith return 0; 4539b94acceSBarry Smith } 45474679c65SBarry Smith 4555615d1e5SSatish Balay #undef __FUNC__ 456*d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberUnsuccessfulSteps" 4579b94acceSBarry Smith /*@ 4589b94acceSBarry Smith SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps 4599b94acceSBarry Smith attempted by the nonlinear solver. 4609b94acceSBarry Smith 4619b94acceSBarry Smith Input Parameter: 4629b94acceSBarry Smith . snes - SNES context 4639b94acceSBarry Smith 4649b94acceSBarry Smith Output Parameter: 4659b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 4669b94acceSBarry Smith 467c96a6f78SLois Curfman McInnes Notes: 468c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 469c96a6f78SLois Curfman McInnes 4709b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 4719b94acceSBarry Smith @*/ 4729b94acceSBarry Smith int SNESGetNumberUnsuccessfulSteps(SNES snes,int* nfails) 4739b94acceSBarry Smith { 47477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 47574679c65SBarry Smith PetscValidIntPointer(nfails); 4769b94acceSBarry Smith *nfails = snes->nfailures; 4779b94acceSBarry Smith return 0; 4789b94acceSBarry Smith } 479a847f771SSatish Balay 4805615d1e5SSatish Balay #undef __FUNC__ 481*d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberLinearIterations" 482c96a6f78SLois Curfman McInnes /*@ 483c96a6f78SLois Curfman McInnes SNESGetNumberLinearIterations - Gets the total number of linear iterations 484c96a6f78SLois Curfman McInnes used by the nonlinear solver. 485c96a6f78SLois Curfman McInnes 486c96a6f78SLois Curfman McInnes Input Parameter: 487c96a6f78SLois Curfman McInnes . snes - SNES context 488c96a6f78SLois Curfman McInnes 489c96a6f78SLois Curfman McInnes Output Parameter: 490c96a6f78SLois Curfman McInnes . lits - number of linear iterations 491c96a6f78SLois Curfman McInnes 492c96a6f78SLois Curfman McInnes Notes: 493c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 494c96a6f78SLois Curfman McInnes 495c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations 496c96a6f78SLois Curfman McInnes @*/ 49786bddb7dSBarry Smith int SNESGetNumberLinearIterations(SNES snes,int* lits) 498c96a6f78SLois Curfman McInnes { 499c96a6f78SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 500c96a6f78SLois Curfman McInnes PetscValidIntPointer(lits); 501c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 502c96a6f78SLois Curfman McInnes return 0; 503c96a6f78SLois Curfman McInnes } 504c96a6f78SLois Curfman McInnes 505c96a6f78SLois Curfman McInnes #undef __FUNC__ 506*d4bb536fSBarry Smith #define __FUNC__ "SNESGetSLES" 5079b94acceSBarry Smith /*@C 5089b94acceSBarry Smith SNESGetSLES - Returns the SLES context for a SNES solver. 5099b94acceSBarry Smith 5109b94acceSBarry Smith Input Parameter: 5119b94acceSBarry Smith . snes - the SNES context 5129b94acceSBarry Smith 5139b94acceSBarry Smith Output Parameter: 5149b94acceSBarry Smith . sles - the SLES context 5159b94acceSBarry Smith 5169b94acceSBarry Smith Notes: 5179b94acceSBarry Smith The user can then directly manipulate the SLES context to set various 5189b94acceSBarry Smith options, etc. Likewise, the user can then extract and manipulate the 5199b94acceSBarry Smith KSP and PC contexts as well. 5209b94acceSBarry Smith 5219b94acceSBarry Smith .keywords: SNES, nonlinear, get, SLES, context 5229b94acceSBarry Smith 5239b94acceSBarry Smith .seealso: SLESGetPC(), SLESGetKSP() 5249b94acceSBarry Smith @*/ 5259b94acceSBarry Smith int SNESGetSLES(SNES snes,SLES *sles) 5269b94acceSBarry Smith { 52777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 5289b94acceSBarry Smith *sles = snes->sles; 5299b94acceSBarry Smith return 0; 5309b94acceSBarry Smith } 5319b94acceSBarry Smith /* -----------------------------------------------------------*/ 5325615d1e5SSatish Balay #undef __FUNC__ 5335615d1e5SSatish Balay #define __FUNC__ "SNESCreate" 5349b94acceSBarry Smith /*@C 5359b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 5369b94acceSBarry Smith 5379b94acceSBarry Smith Input Parameter: 5389b94acceSBarry Smith . comm - MPI communicator 5399b94acceSBarry Smith . type - type of method, one of 5409b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS 5419b94acceSBarry Smith $ (for systems of nonlinear equations) 5429b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION 5439b94acceSBarry Smith $ (for unconstrained minimization) 5449b94acceSBarry Smith 5459b94acceSBarry Smith Output Parameter: 5469b94acceSBarry Smith . outsnes - the new SNES context 5479b94acceSBarry Smith 548c1f60f51SBarry Smith Options Database Key: 54919bd6747SLois Curfman McInnes $ -snes_mf - use default matrix-free Jacobian-vector products, 55019bd6747SLois Curfman McInnes $ and no preconditioning matrix 55119bd6747SLois Curfman McInnes $ -snes_mf_operator - use default matrix-free Jacobian-vector 55219bd6747SLois Curfman McInnes $ products, and a user-provided preconditioning matrix 55319bd6747SLois Curfman McInnes $ as set by SNESSetJacobian() 55419bd6747SLois Curfman McInnes $ -snes_fd - use (slow!) finite differences to compute Jacobian 555c1f60f51SBarry Smith 5569b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 5579b94acceSBarry Smith 55863a78c88SLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy() 5599b94acceSBarry Smith @*/ 5604b0e389bSBarry Smith int SNESCreate(MPI_Comm comm,SNESProblemType type,SNES *outsnes) 5619b94acceSBarry Smith { 5629b94acceSBarry Smith int ierr; 5639b94acceSBarry Smith SNES snes; 5649b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 56537fcc0dbSBarry Smith 5669b94acceSBarry Smith *outsnes = 0; 5672a0acf01SLois Curfman McInnes if (type != SNES_UNCONSTRAINED_MINIMIZATION && type != SNES_NONLINEAR_EQUATIONS) 568d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"incorrect method type"); 569*d4bb536fSBarry Smith PetscHeaderCreate(snes,_p_SNES,SNES_COOKIE,SNES_UNKNOWN_METHOD,comm,SNESDestroy,SNESView); 5709b94acceSBarry Smith PLogObjectCreate(snes); 5719b94acceSBarry Smith snes->max_its = 50; 5729750a799SBarry Smith snes->max_funcs = 10000; 5739b94acceSBarry Smith snes->norm = 0.0; 5745a2d0531SBarry Smith if (type == SNES_UNCONSTRAINED_MINIMIZATION) { 575b18e04deSLois Curfman McInnes snes->rtol = 1.e-8; 576b18e04deSLois Curfman McInnes snes->ttol = 0.0; 5779b94acceSBarry Smith snes->atol = 1.e-10; 5785a2d0531SBarry Smith } 579b4874afaSBarry Smith else { 580b4874afaSBarry Smith snes->rtol = 1.e-8; 581b4874afaSBarry Smith snes->ttol = 0.0; 582b4874afaSBarry Smith snes->atol = 1.e-50; 583b4874afaSBarry Smith } 5849b94acceSBarry Smith snes->xtol = 1.e-8; 585b18e04deSLois Curfman McInnes snes->trunctol = 1.e-12; /* no longer used */ 5869b94acceSBarry Smith snes->nfuncs = 0; 5879b94acceSBarry Smith snes->nfailures = 0; 5887a00f4a9SLois Curfman McInnes snes->linear_its = 0; 589639f9d9dSBarry Smith snes->numbermonitors = 0; 5909b94acceSBarry Smith snes->data = 0; 5919b94acceSBarry Smith snes->view = 0; 5929b94acceSBarry Smith snes->computeumfunction = 0; 5939b94acceSBarry Smith snes->umfunP = 0; 5949b94acceSBarry Smith snes->fc = 0; 5959b94acceSBarry Smith snes->deltatol = 1.e-12; 5969b94acceSBarry Smith snes->fmin = -1.e30; 5979b94acceSBarry Smith snes->method_class = type; 5989b94acceSBarry Smith snes->set_method_called = 0; 599a703fe33SLois Curfman McInnes snes->setup_called = 0; 6009b94acceSBarry Smith snes->ksp_ewconv = 0; 6017d1a2b2bSBarry Smith snes->type = -1; 6026f24a144SLois Curfman McInnes snes->xmonitor = 0; 6036f24a144SLois Curfman McInnes snes->vwork = 0; 6046f24a144SLois Curfman McInnes snes->nwork = 0; 605c9005455SLois Curfman McInnes snes->conv_hist_size = 0; 606c9005455SLois Curfman McInnes snes->conv_act_size = 0; 607c9005455SLois Curfman McInnes snes->conv_hist = 0; 6089b94acceSBarry Smith 6099b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 6100452661fSBarry Smith kctx = PetscNew(SNES_KSP_EW_ConvCtx); CHKPTRQ(kctx); 611eed86810SBarry Smith PLogObjectMemory(snes,sizeof(SNES_KSP_EW_ConvCtx)); 6129b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 6139b94acceSBarry Smith kctx->version = 2; 6149b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 6159b94acceSBarry Smith this was too large for some test cases */ 6169b94acceSBarry Smith kctx->rtol_last = 0; 6179b94acceSBarry Smith kctx->rtol_max = .9; 6189b94acceSBarry Smith kctx->gamma = 1.0; 6199b94acceSBarry Smith kctx->alpha2 = .5*(1.0 + sqrt(5.0)); 6209b94acceSBarry Smith kctx->alpha = kctx->alpha2; 6219b94acceSBarry Smith kctx->threshold = .1; 6229b94acceSBarry Smith kctx->lresid_last = 0; 6239b94acceSBarry Smith kctx->norm_last = 0; 6249b94acceSBarry Smith 6259b94acceSBarry Smith ierr = SLESCreate(comm,&snes->sles); CHKERRQ(ierr); 6269b94acceSBarry Smith PLogObjectParent(snes,snes->sles) 6275334005bSBarry Smith 6289b94acceSBarry Smith *outsnes = snes; 6299b94acceSBarry Smith return 0; 6309b94acceSBarry Smith } 6319b94acceSBarry Smith 6329b94acceSBarry Smith /* --------------------------------------------------------------- */ 6335615d1e5SSatish Balay #undef __FUNC__ 634*d4bb536fSBarry Smith #define __FUNC__ "SNESSetFunction" 6359b94acceSBarry Smith /*@C 6369b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 6379b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 6389b94acceSBarry Smith equations. 6399b94acceSBarry Smith 6409b94acceSBarry Smith Input Parameters: 6419b94acceSBarry Smith . snes - the SNES context 6429b94acceSBarry Smith . func - function evaluation routine 6439b94acceSBarry Smith . r - vector to store function value 6442cd2dfdcSLois Curfman McInnes . ctx - [optional] user-defined context for private data for the 6452cd2dfdcSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 6469b94acceSBarry Smith 6479b94acceSBarry Smith Calling sequence of func: 648313e4042SLois Curfman McInnes . func (SNES snes,Vec x,Vec f,void *ctx); 6499b94acceSBarry Smith 6509b94acceSBarry Smith . x - input vector 651313e4042SLois Curfman McInnes . f - function vector 6522cd2dfdcSLois Curfman McInnes . ctx - optional user-defined function context 6539b94acceSBarry Smith 6549b94acceSBarry Smith Notes: 6559b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 6569b94acceSBarry Smith $ f'(x) x = -f(x), 6579b94acceSBarry Smith $ where f'(x) denotes the Jacobian matrix and f(x) is the function. 6589b94acceSBarry Smith 6599b94acceSBarry Smith SNESSetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 6609b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 6619b94acceSBarry Smith SNESSetMinimizationFunction() and SNESSetGradient(); 6629b94acceSBarry Smith 6639b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 6649b94acceSBarry Smith 665a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 6669b94acceSBarry Smith @*/ 6675334005bSBarry Smith int SNESSetFunction( SNES snes, Vec r, int (*func)(SNES,Vec,Vec,void*),void *ctx) 6689b94acceSBarry Smith { 66977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 670e3372554SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,0,"For SNES_NONLINEAR_EQUATIONS only"); 6719b94acceSBarry Smith snes->computefunction = func; 6729b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 6739b94acceSBarry Smith snes->funP = ctx; 6749b94acceSBarry Smith return 0; 6759b94acceSBarry Smith } 6769b94acceSBarry Smith 6775615d1e5SSatish Balay #undef __FUNC__ 6785615d1e5SSatish Balay #define __FUNC__ "SNESComputeFunction" 6799b94acceSBarry Smith /*@ 6809b94acceSBarry Smith SNESComputeFunction - Computes the function that has been set with 6819b94acceSBarry Smith SNESSetFunction(). 6829b94acceSBarry Smith 6839b94acceSBarry Smith Input Parameters: 6849b94acceSBarry Smith . snes - the SNES context 6859b94acceSBarry Smith . x - input vector 6869b94acceSBarry Smith 6879b94acceSBarry Smith Output Parameter: 6883638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 6899b94acceSBarry Smith 6901bffabb2SLois Curfman McInnes Notes: 6919b94acceSBarry Smith SNESComputeFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 6929b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 6939b94acceSBarry Smith SNESComputeMinimizationFunction() and SNESComputeGradient(); 6949b94acceSBarry Smith 6959b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 6969b94acceSBarry Smith 697a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 6989b94acceSBarry Smith @*/ 6999b94acceSBarry Smith int SNESComputeFunction(SNES snes,Vec x, Vec y) 7009b94acceSBarry Smith { 7019b94acceSBarry Smith int ierr; 7029b94acceSBarry Smith 703*d4bb536fSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 704e3372554SBarry Smith SETERRQ(1,0,"For SNES_NONLINEAR_EQUATIONS only"); 705*d4bb536fSBarry Smith } 7069b94acceSBarry Smith PLogEventBegin(SNES_FunctionEval,snes,x,y,0); 7079b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 708ae3c334cSLois Curfman McInnes snes->nfuncs++; 7099b94acceSBarry Smith PLogEventEnd(SNES_FunctionEval,snes,x,y,0); 7109b94acceSBarry Smith return 0; 7119b94acceSBarry Smith } 7129b94acceSBarry Smith 7135615d1e5SSatish Balay #undef __FUNC__ 714*d4bb536fSBarry Smith #define __FUNC__ "SNESSetMinimizationFunction" 7159b94acceSBarry Smith /*@C 7169b94acceSBarry Smith SNESSetMinimizationFunction - Sets the function evaluation routine for 7179b94acceSBarry Smith unconstrained minimization. 7189b94acceSBarry Smith 7199b94acceSBarry Smith Input Parameters: 7209b94acceSBarry Smith . snes - the SNES context 7219b94acceSBarry Smith . func - function evaluation routine 722044dda88SLois Curfman McInnes . ctx - [optional] user-defined context for private data for the 723044dda88SLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 7249b94acceSBarry Smith 7259b94acceSBarry Smith Calling sequence of func: 7269b94acceSBarry Smith . func (SNES snes,Vec x,double *f,void *ctx); 7279b94acceSBarry Smith 7289b94acceSBarry Smith . x - input vector 7299b94acceSBarry Smith . f - function 730044dda88SLois Curfman McInnes . ctx - [optional] user-defined function context 7319b94acceSBarry Smith 7329b94acceSBarry Smith Notes: 7339b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 7349b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 7359b94acceSBarry Smith SNESSetFunction(). 7369b94acceSBarry Smith 7379b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimization, function 7389b94acceSBarry Smith 739a86d99e1SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESComputeMinimizationFunction(), 740a86d99e1SLois Curfman McInnes SNESSetHessian(), SNESSetGradient() 7419b94acceSBarry Smith @*/ 7429b94acceSBarry Smith int SNESSetMinimizationFunction(SNES snes,int (*func)(SNES,Vec,double*,void*), 7439b94acceSBarry Smith void *ctx) 7449b94acceSBarry Smith { 74577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 746e3372554SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 7479b94acceSBarry Smith snes->computeumfunction = func; 7489b94acceSBarry Smith snes->umfunP = ctx; 7499b94acceSBarry Smith return 0; 7509b94acceSBarry Smith } 7519b94acceSBarry Smith 7525615d1e5SSatish Balay #undef __FUNC__ 7535615d1e5SSatish Balay #define __FUNC__ "SNESComputeMinimizationFunction" 7549b94acceSBarry Smith /*@ 7559b94acceSBarry Smith SNESComputeMinimizationFunction - Computes the function that has been 7569b94acceSBarry Smith set with SNESSetMinimizationFunction(). 7579b94acceSBarry Smith 7589b94acceSBarry Smith Input Parameters: 7599b94acceSBarry Smith . snes - the SNES context 7609b94acceSBarry Smith . x - input vector 7619b94acceSBarry Smith 7629b94acceSBarry Smith Output Parameter: 7639b94acceSBarry Smith . y - function value 7649b94acceSBarry Smith 7659b94acceSBarry Smith Notes: 7669b94acceSBarry Smith SNESComputeMinimizationFunction() is valid only for 7679b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 7689b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 769a86d99e1SLois Curfman McInnes 770a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, minimization, function 771a86d99e1SLois Curfman McInnes 772a86d99e1SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESGetMinimizationFunction(), 773a86d99e1SLois Curfman McInnes SNESComputeGradient(), SNESComputeHessian() 7749b94acceSBarry Smith @*/ 7759b94acceSBarry Smith int SNESComputeMinimizationFunction(SNES snes,Vec x,double *y) 7769b94acceSBarry Smith { 7779b94acceSBarry Smith int ierr; 778e3372554SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 7799b94acceSBarry Smith PLogEventBegin(SNES_MinimizationFunctionEval,snes,x,y,0); 7809b94acceSBarry Smith ierr = (*snes->computeumfunction)(snes,x,y,snes->umfunP); CHKERRQ(ierr); 781ae3c334cSLois Curfman McInnes snes->nfuncs++; 7829b94acceSBarry Smith PLogEventEnd(SNES_MinimizationFunctionEval,snes,x,y,0); 7839b94acceSBarry Smith return 0; 7849b94acceSBarry Smith } 7859b94acceSBarry Smith 7865615d1e5SSatish Balay #undef __FUNC__ 787*d4bb536fSBarry Smith #define __FUNC__ "SNESSetGradient" 7889b94acceSBarry Smith /*@C 7899b94acceSBarry Smith SNESSetGradient - Sets the gradient evaluation routine and gradient 7909b94acceSBarry Smith vector for use by the SNES routines. 7919b94acceSBarry Smith 7929b94acceSBarry Smith Input Parameters: 7939b94acceSBarry Smith . snes - the SNES context 7949b94acceSBarry Smith . func - function evaluation routine 795044dda88SLois Curfman McInnes . ctx - optional user-defined context for private data for the 796044dda88SLois Curfman McInnes gradient evaluation routine (may be PETSC_NULL) 7979b94acceSBarry Smith . r - vector to store gradient value 7989b94acceSBarry Smith 7999b94acceSBarry Smith Calling sequence of func: 8009b94acceSBarry Smith . func (SNES, Vec x, Vec g, void *ctx); 8019b94acceSBarry Smith 8029b94acceSBarry Smith . x - input vector 8039b94acceSBarry Smith . g - gradient vector 804044dda88SLois Curfman McInnes . ctx - optional user-defined gradient context 8059b94acceSBarry Smith 8069b94acceSBarry Smith Notes: 8079b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 8089b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 8099b94acceSBarry Smith SNESSetFunction(). 8109b94acceSBarry Smith 8119b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 8129b94acceSBarry Smith 813a86d99e1SLois Curfman McInnes .seealso: SNESGetGradient(), SNESComputeGradient(), SNESSetHessian(), 814a86d99e1SLois Curfman McInnes SNESSetMinimizationFunction(), 8159b94acceSBarry Smith @*/ 81674679c65SBarry Smith int SNESSetGradient(SNES snes,Vec r,int (*func)(SNES,Vec,Vec,void*),void *ctx) 8179b94acceSBarry Smith { 81877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 819e3372554SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 8209b94acceSBarry Smith snes->computefunction = func; 8219b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 8229b94acceSBarry Smith snes->funP = ctx; 8239b94acceSBarry Smith return 0; 8249b94acceSBarry Smith } 8259b94acceSBarry Smith 8265615d1e5SSatish Balay #undef __FUNC__ 8275615d1e5SSatish Balay #define __FUNC__ "SNESComputeGradient" 8289b94acceSBarry Smith /*@ 829a86d99e1SLois Curfman McInnes SNESComputeGradient - Computes the gradient that has been set with 830a86d99e1SLois Curfman McInnes SNESSetGradient(). 8319b94acceSBarry Smith 8329b94acceSBarry Smith Input Parameters: 8339b94acceSBarry Smith . snes - the SNES context 8349b94acceSBarry Smith . x - input vector 8359b94acceSBarry Smith 8369b94acceSBarry Smith Output Parameter: 8379b94acceSBarry Smith . y - gradient vector 8389b94acceSBarry Smith 8399b94acceSBarry Smith Notes: 8409b94acceSBarry Smith SNESComputeGradient() is valid only for 8419b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 8429b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 843a86d99e1SLois Curfman McInnes 844a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, gradient 845a86d99e1SLois Curfman McInnes 846a86d99e1SLois Curfman McInnes .seealso: SNESSetGradient(), SNESGetGradient(), 847a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction(), SNESComputeHessian() 8489b94acceSBarry Smith @*/ 8499b94acceSBarry Smith int SNESComputeGradient(SNES snes,Vec x, Vec y) 8509b94acceSBarry Smith { 8519b94acceSBarry Smith int ierr; 85274679c65SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) 853e3372554SBarry Smith SETERRQ(1,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 8549b94acceSBarry Smith PLogEventBegin(SNES_GradientEval,snes,x,y,0); 8559b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 8569b94acceSBarry Smith PLogEventEnd(SNES_GradientEval,snes,x,y,0); 8579b94acceSBarry Smith return 0; 8589b94acceSBarry Smith } 8599b94acceSBarry Smith 8605615d1e5SSatish Balay #undef __FUNC__ 8615615d1e5SSatish Balay #define __FUNC__ "SNESComputeJacobian" 86262fef451SLois Curfman McInnes /*@ 86362fef451SLois Curfman McInnes SNESComputeJacobian - Computes the Jacobian matrix that has been 86462fef451SLois Curfman McInnes set with SNESSetJacobian(). 86562fef451SLois Curfman McInnes 86662fef451SLois Curfman McInnes Input Parameters: 86762fef451SLois Curfman McInnes . snes - the SNES context 86862fef451SLois Curfman McInnes . x - input vector 86962fef451SLois Curfman McInnes 87062fef451SLois Curfman McInnes Output Parameters: 87162fef451SLois Curfman McInnes . A - Jacobian matrix 87262fef451SLois Curfman McInnes . B - optional preconditioning matrix 87362fef451SLois Curfman McInnes . flag - flag indicating matrix structure 87462fef451SLois Curfman McInnes 87562fef451SLois Curfman McInnes Notes: 87662fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 87762fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 87862fef451SLois Curfman McInnes 879dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 880dc5a77f8SLois Curfman McInnes flag parameter. 88162fef451SLois Curfman McInnes 88262fef451SLois Curfman McInnes SNESComputeJacobian() is valid only for SNES_NONLINEAR_EQUATIONS 88362fef451SLois Curfman McInnes methods. An analogous routine for SNES_UNCONSTRAINED_MINIMIZATION 884005c665bSBarry Smith methods is SNESComputeHessian(). 88562fef451SLois Curfman McInnes 88662fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 88762fef451SLois Curfman McInnes 88862fef451SLois Curfman McInnes .seealso: SNESSetJacobian(), SLESSetOperators() 88962fef451SLois Curfman McInnes @*/ 8909b94acceSBarry Smith int SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 8919b94acceSBarry Smith { 8929b94acceSBarry Smith int ierr; 89374679c65SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) 894e3372554SBarry Smith SETERRQ(1,0,"For SNES_NONLINEAR_EQUATIONS only"); 8959b94acceSBarry Smith if (!snes->computejacobian) return 0; 8969b94acceSBarry Smith PLogEventBegin(SNES_JacobianEval,snes,X,*A,*B); 897c4fc05e7SBarry Smith *flg = DIFFERENT_NONZERO_PATTERN; 8989b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 8999b94acceSBarry Smith PLogEventEnd(SNES_JacobianEval,snes,X,*A,*B); 9006d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 90177c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 90277c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 9039b94acceSBarry Smith return 0; 9049b94acceSBarry Smith } 9059b94acceSBarry Smith 9065615d1e5SSatish Balay #undef __FUNC__ 9075615d1e5SSatish Balay #define __FUNC__ "SNESComputeHessian" 90862fef451SLois Curfman McInnes /*@ 90962fef451SLois Curfman McInnes SNESComputeHessian - Computes the Hessian matrix that has been 91062fef451SLois Curfman McInnes set with SNESSetHessian(). 91162fef451SLois Curfman McInnes 91262fef451SLois Curfman McInnes Input Parameters: 91362fef451SLois Curfman McInnes . snes - the SNES context 91462fef451SLois Curfman McInnes . x - input vector 91562fef451SLois Curfman McInnes 91662fef451SLois Curfman McInnes Output Parameters: 91762fef451SLois Curfman McInnes . A - Hessian matrix 91862fef451SLois Curfman McInnes . B - optional preconditioning matrix 91962fef451SLois Curfman McInnes . flag - flag indicating matrix structure 92062fef451SLois Curfman McInnes 92162fef451SLois Curfman McInnes Notes: 92262fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 92362fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 92462fef451SLois Curfman McInnes 925dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 926dc5a77f8SLois Curfman McInnes flag parameter. 92762fef451SLois Curfman McInnes 92862fef451SLois Curfman McInnes SNESComputeHessian() is valid only for 92962fef451SLois Curfman McInnes SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 93062fef451SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods is SNESComputeJacobian(). 93162fef451SLois Curfman McInnes 93262fef451SLois Curfman McInnes .keywords: SNES, compute, Hessian, matrix 93362fef451SLois Curfman McInnes 934a86d99e1SLois Curfman McInnes .seealso: SNESSetHessian(), SLESSetOperators(), SNESComputeGradient(), 935a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction() 93662fef451SLois Curfman McInnes @*/ 93762fef451SLois Curfman McInnes int SNESComputeHessian(SNES snes,Vec x,Mat *A,Mat *B,MatStructure *flag) 9389b94acceSBarry Smith { 9399b94acceSBarry Smith int ierr; 94074679c65SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) 941e3372554SBarry Smith SETERRQ(1,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 9429b94acceSBarry Smith if (!snes->computejacobian) return 0; 94362fef451SLois Curfman McInnes PLogEventBegin(SNES_HessianEval,snes,x,*A,*B); 9440f4a323eSLois Curfman McInnes *flag = DIFFERENT_NONZERO_PATTERN; 94562fef451SLois Curfman McInnes ierr = (*snes->computejacobian)(snes,x,A,B,flag,snes->jacP); CHKERRQ(ierr); 94662fef451SLois Curfman McInnes PLogEventEnd(SNES_HessianEval,snes,x,*A,*B); 9470f4a323eSLois Curfman McInnes /* make sure user returned a correct Jacobian and preconditioner */ 94877c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 94977c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 9509b94acceSBarry Smith return 0; 9519b94acceSBarry Smith } 9529b94acceSBarry Smith 9535615d1e5SSatish Balay #undef __FUNC__ 954*d4bb536fSBarry Smith #define __FUNC__ "SNESSetJacobian" 9559b94acceSBarry Smith /*@C 9569b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 957044dda88SLois Curfman McInnes location to store the matrix. 9589b94acceSBarry Smith 9599b94acceSBarry Smith Input Parameters: 9609b94acceSBarry Smith . snes - the SNES context 9619b94acceSBarry Smith . A - Jacobian matrix 9629b94acceSBarry Smith . B - preconditioner matrix (usually same as the Jacobian) 9639b94acceSBarry Smith . func - Jacobian evaluation routine 9642cd2dfdcSLois Curfman McInnes . ctx - [optional] user-defined context for private data for the 9652cd2dfdcSLois Curfman McInnes Jacobian evaluation routine (may be PETSC_NULL) 9669b94acceSBarry Smith 9679b94acceSBarry Smith Calling sequence of func: 968313e4042SLois Curfman McInnes . func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 9699b94acceSBarry Smith 9709b94acceSBarry Smith . x - input vector 9719b94acceSBarry Smith . A - Jacobian matrix 9729b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 973ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 974ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 9752cd2dfdcSLois Curfman McInnes . ctx - [optional] user-defined Jacobian context 9769b94acceSBarry Smith 9779b94acceSBarry Smith Notes: 978dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 9792cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 980ac21db08SLois Curfman McInnes 981ac21db08SLois Curfman McInnes The routine func() takes Mat * as the matrix arguments rather than Mat. 9829b94acceSBarry Smith This allows the Jacobian evaluation routine to replace A and/or B with a 9839b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 9849b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 9859b94acceSBarry Smith throughout the global iterations. 9869b94acceSBarry Smith 9879b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 9889b94acceSBarry Smith 989ac21db08SLois Curfman McInnes .seealso: SLESSetOperators(), SNESSetFunction() 9909b94acceSBarry Smith @*/ 9919b94acceSBarry Smith int SNESSetJacobian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 9929b94acceSBarry Smith MatStructure*,void*),void *ctx) 9939b94acceSBarry Smith { 99477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 995*d4bb536fSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,0,"For SNES_NONLINEAR_EQUATIONS only"); 9969b94acceSBarry Smith snes->computejacobian = func; 9979b94acceSBarry Smith snes->jacP = ctx; 9989b94acceSBarry Smith snes->jacobian = A; 9999b94acceSBarry Smith snes->jacobian_pre = B; 10009b94acceSBarry Smith return 0; 10019b94acceSBarry Smith } 100262fef451SLois Curfman McInnes 10035615d1e5SSatish Balay #undef __FUNC__ 1004*d4bb536fSBarry Smith #define __FUNC__ "SNESGetJacobian" 1005b4fd4287SBarry Smith /*@ 1006b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 1007b4fd4287SBarry Smith provided context for evaluating the Jacobian. 1008b4fd4287SBarry Smith 1009b4fd4287SBarry Smith Input Parameter: 1010b4fd4287SBarry Smith . snes - the nonlinear solver context 1011b4fd4287SBarry Smith 1012b4fd4287SBarry Smith Output Parameters: 1013b4fd4287SBarry Smith . A - location to stash Jacobian matrix (or PETSC_NULL) 1014b4fd4287SBarry Smith . B - location to stash preconditioner matrix (or PETSC_NULL) 1015b4fd4287SBarry Smith . ctx - location to stash Jacobian ctx (or PETSC_NULL) 1016b4fd4287SBarry Smith 1017b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian() 1018b4fd4287SBarry Smith @*/ 1019b4fd4287SBarry Smith int SNESGetJacobian(SNES snes,Mat *A,Mat *B, void **ctx) 1020b4fd4287SBarry Smith { 102174679c65SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) 1022e3372554SBarry Smith SETERRQ(1,0,"For SNES_NONLINEAR_EQUATIONS only"); 1023b4fd4287SBarry Smith if (A) *A = snes->jacobian; 1024b4fd4287SBarry Smith if (B) *B = snes->jacobian_pre; 1025b4fd4287SBarry Smith if (ctx) *ctx = snes->jacP; 1026b4fd4287SBarry Smith return 0; 1027b4fd4287SBarry Smith } 1028b4fd4287SBarry Smith 10295615d1e5SSatish Balay #undef __FUNC__ 1030*d4bb536fSBarry Smith #define __FUNC__ "SNESSetHessian" 10319b94acceSBarry Smith /*@C 10329b94acceSBarry Smith SNESSetHessian - Sets the function to compute Hessian as well as the 1033044dda88SLois Curfman McInnes location to store the matrix. 10349b94acceSBarry Smith 10359b94acceSBarry Smith Input Parameters: 10369b94acceSBarry Smith . snes - the SNES context 10379b94acceSBarry Smith . A - Hessian matrix 10389b94acceSBarry Smith . B - preconditioner matrix (usually same as the Hessian) 10399b94acceSBarry Smith . func - Jacobian evaluation routine 1040313e4042SLois Curfman McInnes . ctx - [optional] user-defined context for private data for the 1041313e4042SLois Curfman McInnes Hessian evaluation routine (may be PETSC_NULL) 10429b94acceSBarry Smith 10439b94acceSBarry Smith Calling sequence of func: 1044313e4042SLois Curfman McInnes . func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 10459b94acceSBarry Smith 10469b94acceSBarry Smith . x - input vector 10479b94acceSBarry Smith . A - Hessian matrix 10489b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1049ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 1050ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 10512cd2dfdcSLois Curfman McInnes . ctx - [optional] user-defined Hessian context 10529b94acceSBarry Smith 10539b94acceSBarry Smith Notes: 1054dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 10552cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1056ac21db08SLois Curfman McInnes 10579b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 10589b94acceSBarry Smith This allows the Hessian evaluation routine to replace A and/or B with a 10599b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 10609b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 10619b94acceSBarry Smith throughout the global iterations. 10629b94acceSBarry Smith 10639b94acceSBarry Smith .keywords: SNES, nonlinear, set, Hessian, matrix 10649b94acceSBarry Smith 1065ac21db08SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESSetGradient(), SLESSetOperators() 10669b94acceSBarry Smith @*/ 10679b94acceSBarry Smith int SNESSetHessian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 10689b94acceSBarry Smith MatStructure*,void*),void *ctx) 10699b94acceSBarry Smith { 107077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1071*d4bb536fSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1072e3372554SBarry Smith SETERRQ(1,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 1073*d4bb536fSBarry Smith } 10749b94acceSBarry Smith snes->computejacobian = func; 10759b94acceSBarry Smith snes->jacP = ctx; 10769b94acceSBarry Smith snes->jacobian = A; 10779b94acceSBarry Smith snes->jacobian_pre = B; 10789b94acceSBarry Smith return 0; 10799b94acceSBarry Smith } 10809b94acceSBarry Smith 10815615d1e5SSatish Balay #undef __FUNC__ 1082*d4bb536fSBarry Smith #define __FUNC__ "SNESGetHessian" 108362fef451SLois Curfman McInnes /*@ 108462fef451SLois Curfman McInnes SNESGetHessian - Returns the Hessian matrix and optionally the user 108562fef451SLois Curfman McInnes provided context for evaluating the Hessian. 108662fef451SLois Curfman McInnes 108762fef451SLois Curfman McInnes Input Parameter: 108862fef451SLois Curfman McInnes . snes - the nonlinear solver context 108962fef451SLois Curfman McInnes 109062fef451SLois Curfman McInnes Output Parameters: 109162fef451SLois Curfman McInnes . A - location to stash Hessian matrix (or PETSC_NULL) 109262fef451SLois Curfman McInnes . B - location to stash preconditioner matrix (or PETSC_NULL) 109362fef451SLois Curfman McInnes . ctx - location to stash Hessian ctx (or PETSC_NULL) 109462fef451SLois Curfman McInnes 109562fef451SLois Curfman McInnes .seealso: SNESSetHessian(), SNESComputeHessian() 109662fef451SLois Curfman McInnes @*/ 109762fef451SLois Curfman McInnes int SNESGetHessian(SNES snes,Mat *A,Mat *B, void **ctx) 109862fef451SLois Curfman McInnes { 109974679c65SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) 1100e3372554SBarry Smith SETERRQ(1,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 110162fef451SLois Curfman McInnes if (A) *A = snes->jacobian; 110262fef451SLois Curfman McInnes if (B) *B = snes->jacobian_pre; 110362fef451SLois Curfman McInnes if (ctx) *ctx = snes->jacP; 110462fef451SLois Curfman McInnes return 0; 110562fef451SLois Curfman McInnes } 110662fef451SLois Curfman McInnes 11079b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */ 11089b94acceSBarry Smith 11095615d1e5SSatish Balay #undef __FUNC__ 11105615d1e5SSatish Balay #define __FUNC__ "SNESSetUp" 11119b94acceSBarry Smith /*@ 11129b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 1113272ac6f2SLois Curfman McInnes of a nonlinear solver. 11149b94acceSBarry Smith 11159b94acceSBarry Smith Input Parameter: 11169b94acceSBarry Smith . snes - the SNES context 11178ddd3da0SLois Curfman McInnes . x - the solution vector 11189b94acceSBarry Smith 1119272ac6f2SLois Curfman McInnes Notes: 1120272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 1121272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 1122272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 1123272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 1124272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 1125272ac6f2SLois Curfman McInnes 11269b94acceSBarry Smith .keywords: SNES, nonlinear, setup 11279b94acceSBarry Smith 11289b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 11299b94acceSBarry Smith @*/ 11308ddd3da0SLois Curfman McInnes int SNESSetUp(SNES snes,Vec x) 11319b94acceSBarry Smith { 1132272ac6f2SLois Curfman McInnes int ierr, flg; 113377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 113477c4ece6SBarry Smith PetscValidHeaderSpecific(x,VEC_COOKIE); 11358ddd3da0SLois Curfman McInnes snes->vec_sol = snes->vec_sol_always = x; 11369b94acceSBarry Smith 1137c1f60f51SBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_mf_operator", &flg); CHKERRQ(ierr); 1138c1f60f51SBarry Smith /* 1139c1f60f51SBarry Smith This version replaces the user provided Jacobian matrix with a 1140dfa02198SLois Curfman McInnes matrix-free version but still employs the user-provided preconditioner matrix 1141c1f60f51SBarry Smith */ 1142c1f60f51SBarry Smith if (flg) { 1143c1f60f51SBarry Smith Mat J; 1144c1f60f51SBarry Smith ierr = SNESDefaultMatrixFreeMatCreate(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1145c1f60f51SBarry Smith PLogObjectParent(snes,J); 1146c1f60f51SBarry Smith snes->mfshell = J; 1147c1f60f51SBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1148c1f60f51SBarry Smith snes->jacobian = J; 1149c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Jacobian routines\n"); 1150c1f60f51SBarry Smith } 1151c1f60f51SBarry Smith else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 1152c1f60f51SBarry Smith snes->jacobian = J; 1153c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Hessian routines\n"); 1154*d4bb536fSBarry Smith } else { 1155*d4bb536fSBarry Smith SETERRQ(1,0,"Method class doesn't support matrix-free operator option"); 1156*d4bb536fSBarry Smith } 1157c1f60f51SBarry Smith } 1158272ac6f2SLois Curfman McInnes ierr = OptionsHasName(snes->prefix,"-snes_mf", &flg); CHKERRQ(ierr); 1159c1f60f51SBarry Smith /* 1160dfa02198SLois Curfman McInnes This version replaces both the user-provided Jacobian and the user- 1161c1f60f51SBarry Smith provided preconditioner matrix with the default matrix free version. 1162c1f60f51SBarry Smith */ 116331615d04SLois Curfman McInnes if (flg) { 1164272ac6f2SLois Curfman McInnes Mat J; 1165272ac6f2SLois Curfman McInnes ierr = SNESDefaultMatrixFreeMatCreate(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1166272ac6f2SLois Curfman McInnes PLogObjectParent(snes,J); 1167272ac6f2SLois Curfman McInnes snes->mfshell = J; 116883e56afdSLois Curfman McInnes if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 116983e56afdSLois Curfman McInnes ierr = SNESSetJacobian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 117094a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Jacobian routines\n"); 117183e56afdSLois Curfman McInnes } 117283e56afdSLois Curfman McInnes else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 117383e56afdSLois Curfman McInnes ierr = SNESSetHessian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 117494a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Hessian routines\n"); 1175*d4bb536fSBarry Smith } else { 1176*d4bb536fSBarry Smith SETERRQ(1,0,"Method class doesn't support matrix-free option"); 1177*d4bb536fSBarry Smith } 1178272ac6f2SLois Curfman McInnes } 11799b94acceSBarry Smith if ((snes->method_class == SNES_NONLINEAR_EQUATIONS)) { 1180e3372554SBarry Smith if (!snes->vec_func) SETERRQ(1,0,"Must call SNESSetFunction() first"); 1181e3372554SBarry Smith if (!snes->computefunction) SETERRQ(1,0,"Must call SNESSetFunction() first"); 1182e3372554SBarry Smith if (!snes->jacobian) SETERRQ(1,0,"Must call SNESSetJacobian() first"); 1183e3372554SBarry Smith if (snes->vec_func == snes->vec_sol) SETERRQ(1,0,"Solution vector cannot be function vector"); 1184a703fe33SLois Curfman McInnes 1185a703fe33SLois Curfman McInnes /* Set the KSP stopping criterion to use the Eisenstat-Walker method */ 118640191667SLois Curfman McInnes if (snes->ksp_ewconv && snes->type != SNES_EQ_TR) { 1187a703fe33SLois Curfman McInnes SLES sles; KSP ksp; 1188a703fe33SLois Curfman McInnes ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 1189a703fe33SLois Curfman McInnes ierr = SLESGetKSP(sles,&ksp); CHKERRQ(ierr); 1190a703fe33SLois Curfman McInnes ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private, 1191a703fe33SLois Curfman McInnes (void *)snes); CHKERRQ(ierr); 1192a703fe33SLois Curfman McInnes } 11939b94acceSBarry Smith } else if ((snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION)) { 1194e3372554SBarry Smith if (!snes->vec_func) SETERRQ(1,0,"Must call SNESSetGradient() first"); 1195e3372554SBarry Smith if (!snes->computefunction) SETERRQ(1,0,"Must call SNESSetGradient() first"); 1196*d4bb536fSBarry Smith if (!snes->computeumfunction) SETERRQ(1,0,"Must call SNESSetMinimizationFunction() first"); 1197e3372554SBarry Smith if (!snes->jacobian) SETERRQ(1,0,"Must call SNESSetHessian() first"); 1198*d4bb536fSBarry Smith } else { 1199*d4bb536fSBarry Smith SETERRQ(1,0,"Unknown method class"); 1200*d4bb536fSBarry Smith } 1201a703fe33SLois Curfman McInnes if (snes->setup) {ierr = (*snes->setup)(snes); CHKERRQ(ierr);} 1202a703fe33SLois Curfman McInnes snes->setup_called = 1; 1203a703fe33SLois Curfman McInnes return 0; 12049b94acceSBarry Smith } 12059b94acceSBarry Smith 12065615d1e5SSatish Balay #undef __FUNC__ 1207*d4bb536fSBarry Smith #define __FUNC__ "SNESDestroy" 12089b94acceSBarry Smith /*@C 12099b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 12109b94acceSBarry Smith with SNESCreate(). 12119b94acceSBarry Smith 12129b94acceSBarry Smith Input Parameter: 12139b94acceSBarry Smith . snes - the SNES context 12149b94acceSBarry Smith 12159b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 12169b94acceSBarry Smith 121763a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 12189b94acceSBarry Smith @*/ 12199b94acceSBarry Smith int SNESDestroy(SNES snes) 12209b94acceSBarry Smith { 12219b94acceSBarry Smith int ierr; 122277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1223*d4bb536fSBarry Smith if (--snes->refct > 0) return 0; 1224*d4bb536fSBarry Smith 12259750a799SBarry Smith if (snes->destroy) {ierr = (*(snes)->destroy)((PetscObject)snes); CHKERRQ(ierr);} 12260452661fSBarry Smith if (snes->kspconvctx) PetscFree(snes->kspconvctx); 12279b94acceSBarry Smith if (snes->mfshell) MatDestroy(snes->mfshell); 12289b94acceSBarry Smith ierr = SLESDestroy(snes->sles); CHKERRQ(ierr); 12293e2e452bSBarry Smith if (snes->xmonitor) SNESLGMonitorDestroy(snes->xmonitor); 12306f24a144SLois Curfman McInnes if (snes->vwork) VecDestroyVecs(snes->vwork,snes->nvwork); 12319b94acceSBarry Smith PLogObjectDestroy((PetscObject)snes); 12320452661fSBarry Smith PetscHeaderDestroy((PetscObject)snes); 12339b94acceSBarry Smith return 0; 12349b94acceSBarry Smith } 12359b94acceSBarry Smith 12369b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 12379b94acceSBarry Smith 12385615d1e5SSatish Balay #undef __FUNC__ 12395615d1e5SSatish Balay #define __FUNC__ "SNESSetTolerances" 12409b94acceSBarry Smith /*@ 1241d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 12429b94acceSBarry Smith 12439b94acceSBarry Smith Input Parameters: 12449b94acceSBarry Smith . snes - the SNES context 124533174efeSLois Curfman McInnes . atol - absolute convergence tolerance 124633174efeSLois Curfman McInnes . rtol - relative convergence tolerance 124733174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 124833174efeSLois Curfman McInnes of the change in the solution between steps 124933174efeSLois Curfman McInnes . maxit - maximum number of iterations 125033174efeSLois Curfman McInnes . maxf - maximum number of function evaluations 12519b94acceSBarry Smith 125233174efeSLois Curfman McInnes Options Database Keys: 125333174efeSLois Curfman McInnes $ -snes_atol <atol> 125433174efeSLois Curfman McInnes $ -snes_rtol <rtol> 125533174efeSLois Curfman McInnes $ -snes_stol <stol> 125633174efeSLois Curfman McInnes $ -snes_max_it <maxit> 125733174efeSLois Curfman McInnes $ -snes_max_funcs <maxf> 12589b94acceSBarry Smith 1259d7a720efSLois Curfman McInnes Notes: 12609b94acceSBarry Smith The default maximum number of iterations is 50. 12619b94acceSBarry Smith The default maximum number of function evaluations is 1000. 12629b94acceSBarry Smith 126333174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances 12649b94acceSBarry Smith 1265d7a720efSLois Curfman McInnes .seealso: SNESSetTrustRegionTolerance(), SNESSetMinimizationFunctionTolerance() 12669b94acceSBarry Smith @*/ 1267d7a720efSLois Curfman McInnes int SNESSetTolerances(SNES snes,double atol,double rtol,double stol,int maxit,int maxf) 12689b94acceSBarry Smith { 126977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1270d7a720efSLois Curfman McInnes if (atol != PETSC_DEFAULT) snes->atol = atol; 1271d7a720efSLois Curfman McInnes if (rtol != PETSC_DEFAULT) snes->rtol = rtol; 1272d7a720efSLois Curfman McInnes if (stol != PETSC_DEFAULT) snes->xtol = stol; 1273d7a720efSLois Curfman McInnes if (maxit != PETSC_DEFAULT) snes->max_its = maxit; 1274d7a720efSLois Curfman McInnes if (maxf != PETSC_DEFAULT) snes->max_funcs = maxf; 12759b94acceSBarry Smith return 0; 12769b94acceSBarry Smith } 12779b94acceSBarry Smith 12785615d1e5SSatish Balay #undef __FUNC__ 12795615d1e5SSatish Balay #define __FUNC__ "SNESGetTolerances" 12809b94acceSBarry Smith /*@ 128133174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 128233174efeSLois Curfman McInnes 128333174efeSLois Curfman McInnes Input Parameters: 128433174efeSLois Curfman McInnes . snes - the SNES context 128533174efeSLois Curfman McInnes . atol - absolute convergence tolerance 128633174efeSLois Curfman McInnes . rtol - relative convergence tolerance 128733174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 128833174efeSLois Curfman McInnes of the change in the solution between steps 128933174efeSLois Curfman McInnes . maxit - maximum number of iterations 129033174efeSLois Curfman McInnes . maxf - maximum number of function evaluations 129133174efeSLois Curfman McInnes 129233174efeSLois Curfman McInnes Notes: 129333174efeSLois Curfman McInnes The user can specify PETSC_NULL for any parameter that is not needed. 129433174efeSLois Curfman McInnes 129533174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances 129633174efeSLois Curfman McInnes 129733174efeSLois Curfman McInnes .seealso: SNESSetTolerances() 129833174efeSLois Curfman McInnes @*/ 129933174efeSLois Curfman McInnes int SNESGetTolerances(SNES snes,double *atol,double *rtol,double *stol,int *maxit,int *maxf) 130033174efeSLois Curfman McInnes { 130133174efeSLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 130233174efeSLois Curfman McInnes if (atol) *atol = snes->atol; 130333174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 130433174efeSLois Curfman McInnes if (stol) *stol = snes->xtol; 130533174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 130633174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 130733174efeSLois Curfman McInnes return 0; 130833174efeSLois Curfman McInnes } 130933174efeSLois Curfman McInnes 13105615d1e5SSatish Balay #undef __FUNC__ 13115615d1e5SSatish Balay #define __FUNC__ "SNESSetTrustRegionTolerance" 131233174efeSLois Curfman McInnes /*@ 13139b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 13149b94acceSBarry Smith 13159b94acceSBarry Smith Input Parameters: 13169b94acceSBarry Smith . snes - the SNES context 13179b94acceSBarry Smith . tol - tolerance 13189b94acceSBarry Smith 13199b94acceSBarry Smith Options Database Key: 1320d7a720efSLois Curfman McInnes $ -snes_trtol <tol> 13219b94acceSBarry Smith 13229b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 13239b94acceSBarry Smith 1324d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetMinimizationFunctionTolerance() 13259b94acceSBarry Smith @*/ 13269b94acceSBarry Smith int SNESSetTrustRegionTolerance(SNES snes,double tol) 13279b94acceSBarry Smith { 132877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 13299b94acceSBarry Smith snes->deltatol = tol; 13309b94acceSBarry Smith return 0; 13319b94acceSBarry Smith } 13329b94acceSBarry Smith 13335615d1e5SSatish Balay #undef __FUNC__ 13345615d1e5SSatish Balay #define __FUNC__ "SNESSetMinimizationFunctionTolerance" 13359b94acceSBarry Smith /*@ 133677c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance - Sets the minimum allowable function tolerance 13379b94acceSBarry Smith for unconstrained minimization solvers. 13389b94acceSBarry Smith 13399b94acceSBarry Smith Input Parameters: 13409b94acceSBarry Smith . snes - the SNES context 13419b94acceSBarry Smith . ftol - minimum function tolerance 13429b94acceSBarry Smith 13439b94acceSBarry Smith Options Database Key: 1344d7a720efSLois Curfman McInnes $ -snes_fmin <ftol> 13459b94acceSBarry Smith 13469b94acceSBarry Smith Note: 134777c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance() is valid for SNES_UNCONSTRAINED_MINIMIZATION 13489b94acceSBarry Smith methods only. 13499b94acceSBarry Smith 13509b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimum, convergence, function, tolerance 13519b94acceSBarry Smith 1352d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetTrustRegionTolerance() 13539b94acceSBarry Smith @*/ 135477c4ece6SBarry Smith int SNESSetMinimizationFunctionTolerance(SNES snes,double ftol) 13559b94acceSBarry Smith { 135677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 13579b94acceSBarry Smith snes->fmin = ftol; 13589b94acceSBarry Smith return 0; 13599b94acceSBarry Smith } 13609b94acceSBarry Smith 13619b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 13629b94acceSBarry Smith 13635615d1e5SSatish Balay #undef __FUNC__ 1364*d4bb536fSBarry Smith #define __FUNC__ "SNESSetMonitor" 13659b94acceSBarry Smith /*@C 13669b94acceSBarry Smith SNESSetMonitor - Sets the function that is to be used at every 13679b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 13689b94acceSBarry Smith progress. 13699b94acceSBarry Smith 13709b94acceSBarry Smith Input Parameters: 13719b94acceSBarry Smith . snes - the SNES context 13729b94acceSBarry Smith . func - monitoring routine 1373044dda88SLois Curfman McInnes . mctx - [optional] user-defined context for private data for the 1374044dda88SLois Curfman McInnes monitor routine (may be PETSC_NULL) 13759b94acceSBarry Smith 13769b94acceSBarry Smith Calling sequence of func: 1377682d7d0cSBarry Smith int func(SNES snes,int its, Vec x,Vec f,double norm,void *mctx) 13789b94acceSBarry Smith 13799b94acceSBarry Smith $ snes - the SNES context 13809b94acceSBarry Smith $ its - iteration number 1381044dda88SLois Curfman McInnes $ mctx - [optional] monitoring context 13829b94acceSBarry Smith $ 13839b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods: 13849b94acceSBarry Smith $ norm - 2-norm function value (may be estimated) 13859b94acceSBarry Smith $ 13869b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods: 13879b94acceSBarry Smith $ norm - 2-norm gradient value (may be estimated) 13889b94acceSBarry Smith 13899665c990SLois Curfman McInnes Options Database Keys: 13909665c990SLois Curfman McInnes $ -snes_monitor : sets SNESDefaultMonitor() 13919665c990SLois Curfman McInnes $ -snes_xmonitor : sets line graph monitor, 13929665c990SLois Curfman McInnes $ uses SNESLGMonitorCreate() 13939665c990SLois Curfman McInnes $ -snes_cancelmonitors : cancels all monitors that have 13949665c990SLois Curfman McInnes $ been hardwired into a code by 13959665c990SLois Curfman McInnes $ calls to SNESSetMonitor(), but 13969665c990SLois Curfman McInnes $ does not cancel those set via 13979665c990SLois Curfman McInnes $ the options database. 13989665c990SLois Curfman McInnes 13999665c990SLois Curfman McInnes 1400639f9d9dSBarry Smith Notes: 14016bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 14026bc08f3fSLois Curfman McInnes SNESSetMonitor() multiple times; all will be called in the 14036bc08f3fSLois Curfman McInnes order in which they were set. 1404639f9d9dSBarry Smith 14059b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 14069b94acceSBarry Smith 14079b94acceSBarry Smith .seealso: SNESDefaultMonitor() 14089b94acceSBarry Smith @*/ 140974679c65SBarry Smith int SNESSetMonitor( SNES snes, int (*func)(SNES,int,double,void*),void *mctx ) 14109b94acceSBarry Smith { 1411639f9d9dSBarry Smith if (!func) { 1412639f9d9dSBarry Smith snes->numbermonitors = 0; 1413639f9d9dSBarry Smith return 0; 1414639f9d9dSBarry Smith } 1415639f9d9dSBarry Smith if (snes->numbermonitors >= MAXSNESMONITORS) { 1416e3372554SBarry Smith SETERRQ(1,0,"Too many monitors set"); 1417639f9d9dSBarry Smith } 1418639f9d9dSBarry Smith 1419639f9d9dSBarry Smith snes->monitor[snes->numbermonitors] = func; 1420639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 14219b94acceSBarry Smith return 0; 14229b94acceSBarry Smith } 14239b94acceSBarry Smith 14245615d1e5SSatish Balay #undef __FUNC__ 1425*d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceTest" 14269b94acceSBarry Smith /*@C 14279b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 14289b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 14299b94acceSBarry Smith 14309b94acceSBarry Smith Input Parameters: 14319b94acceSBarry Smith . snes - the SNES context 14329b94acceSBarry Smith . func - routine to test for convergence 1433044dda88SLois Curfman McInnes . cctx - [optional] context for private data for the convergence routine 1434044dda88SLois Curfman McInnes (may be PETSC_NULL) 14359b94acceSBarry Smith 14369b94acceSBarry Smith Calling sequence of func: 14379b94acceSBarry Smith int func (SNES snes,double xnorm,double gnorm, 14389b94acceSBarry Smith double f,void *cctx) 14399b94acceSBarry Smith 14409b94acceSBarry Smith $ snes - the SNES context 1441044dda88SLois Curfman McInnes $ cctx - [optional] convergence context 14429b94acceSBarry Smith $ xnorm - 2-norm of current iterate 14439b94acceSBarry Smith $ 14449b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods: 14459b94acceSBarry Smith $ gnorm - 2-norm of current step 14469b94acceSBarry Smith $ f - 2-norm of function 14479b94acceSBarry Smith $ 14489b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods: 14499b94acceSBarry Smith $ gnorm - 2-norm of current gradient 14509b94acceSBarry Smith $ f - function value 14519b94acceSBarry Smith 14529b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 14539b94acceSBarry Smith 145440191667SLois Curfman McInnes .seealso: SNESConverged_EQ_LS(), SNESConverged_EQ_TR(), 145540191667SLois Curfman McInnes SNESConverged_UM_LS(), SNESConverged_UM_TR() 14569b94acceSBarry Smith @*/ 145774679c65SBarry Smith int SNESSetConvergenceTest(SNES snes,int (*func)(SNES,double,double,double,void*),void *cctx) 14589b94acceSBarry Smith { 14599b94acceSBarry Smith (snes)->converged = func; 14609b94acceSBarry Smith (snes)->cnvP = cctx; 14619b94acceSBarry Smith return 0; 14629b94acceSBarry Smith } 14639b94acceSBarry Smith 14645615d1e5SSatish Balay #undef __FUNC__ 1465*d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceHistory" 1466c9005455SLois Curfman McInnes /*@ 1467c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 1468c9005455SLois Curfman McInnes 1469c9005455SLois Curfman McInnes Input Parameters: 1470c9005455SLois Curfman McInnes . snes - iterative context obtained from SNESCreate() 1471c9005455SLois Curfman McInnes . a - array to hold history 1472c9005455SLois Curfman McInnes . na - size of a 1473c9005455SLois Curfman McInnes 1474c9005455SLois Curfman McInnes Notes: 1475c9005455SLois Curfman McInnes If set, this array will contain the function norms (for 1476c9005455SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods) or gradient norms 1477c9005455SLois Curfman McInnes (for SNES_UNCONSTRAINED_MINIMIZATION methods) computed 1478c9005455SLois Curfman McInnes at each step. 1479c9005455SLois Curfman McInnes 1480c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 1481c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 1482c9005455SLois Curfman McInnes during the section of code that is being timed. 1483c9005455SLois Curfman McInnes 1484c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history 1485c9005455SLois Curfman McInnes @*/ 1486c9005455SLois Curfman McInnes int SNESSetConvergenceHistory(SNES snes, double *a, int na) 1487c9005455SLois Curfman McInnes { 1488c9005455SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 1489c9005455SLois Curfman McInnes if (na) PetscValidScalarPointer(a); 1490c9005455SLois Curfman McInnes snes->conv_hist = a; 1491c9005455SLois Curfman McInnes snes->conv_hist_size = na; 1492c9005455SLois Curfman McInnes return 0; 1493c9005455SLois Curfman McInnes } 1494c9005455SLois Curfman McInnes 1495c9005455SLois Curfman McInnes #undef __FUNC__ 14965615d1e5SSatish Balay #define __FUNC__ "SNESScaleStep_Private" 14979b94acceSBarry Smith /* 14989b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 14999b94acceSBarry Smith positive parameter delta. 15009b94acceSBarry Smith 15019b94acceSBarry Smith Input Parameters: 15029b94acceSBarry Smith . snes - the SNES context 15039b94acceSBarry Smith . y - approximate solution of linear system 15049b94acceSBarry Smith . fnorm - 2-norm of current function 15059b94acceSBarry Smith . delta - trust region size 15069b94acceSBarry Smith 15079b94acceSBarry Smith Output Parameters: 15089b94acceSBarry Smith . gpnorm - predicted function norm at the new point, assuming local 15099b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 15109b94acceSBarry Smith region, and exceeds zero otherwise. 15119b94acceSBarry Smith . ynorm - 2-norm of the step 15129b94acceSBarry Smith 15139b94acceSBarry Smith Note: 151440191667SLois Curfman McInnes For non-trust region methods such as SNES_EQ_LS, the parameter delta 15159b94acceSBarry Smith is set to be the maximum allowable step size. 15169b94acceSBarry Smith 15179b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 15189b94acceSBarry Smith */ 15199b94acceSBarry Smith int SNESScaleStep_Private(SNES snes,Vec y,double *fnorm,double *delta, 15209b94acceSBarry Smith double *gpnorm,double *ynorm) 15219b94acceSBarry Smith { 15229b94acceSBarry Smith double norm; 15239b94acceSBarry Smith Scalar cnorm; 1524cddf8d76SBarry Smith VecNorm(y,NORM_2, &norm ); 15259b94acceSBarry Smith if (norm > *delta) { 15269b94acceSBarry Smith norm = *delta/norm; 15279b94acceSBarry Smith *gpnorm = (1.0 - norm)*(*fnorm); 15289b94acceSBarry Smith cnorm = norm; 15299b94acceSBarry Smith VecScale( &cnorm, y ); 15309b94acceSBarry Smith *ynorm = *delta; 15319b94acceSBarry Smith } else { 15329b94acceSBarry Smith *gpnorm = 0.0; 15339b94acceSBarry Smith *ynorm = norm; 15349b94acceSBarry Smith } 15359b94acceSBarry Smith return 0; 15369b94acceSBarry Smith } 15379b94acceSBarry Smith 15385615d1e5SSatish Balay #undef __FUNC__ 15395615d1e5SSatish Balay #define __FUNC__ "SNESSolve" 15409b94acceSBarry Smith /*@ 15419b94acceSBarry Smith SNESSolve - Solves a nonlinear system. Call SNESSolve after calling 154263a78c88SLois Curfman McInnes SNESCreate() and optional routines of the form SNESSetXXX(). 15439b94acceSBarry Smith 15449b94acceSBarry Smith Input Parameter: 15459b94acceSBarry Smith . snes - the SNES context 15468ddd3da0SLois Curfman McInnes . x - the solution vector 15479b94acceSBarry Smith 15489b94acceSBarry Smith Output Parameter: 15499b94acceSBarry Smith its - number of iterations until termination 15509b94acceSBarry Smith 15518ddd3da0SLois Curfman McInnes Note: 15528ddd3da0SLois Curfman McInnes The user should initialize the vector, x, with the initial guess 15538ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 15548ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 15558ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 15568ddd3da0SLois Curfman McInnes 15579b94acceSBarry Smith .keywords: SNES, nonlinear, solve 15589b94acceSBarry Smith 155963a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESDestroy() 15609b94acceSBarry Smith @*/ 15618ddd3da0SLois Curfman McInnes int SNESSolve(SNES snes,Vec x,int *its) 15629b94acceSBarry Smith { 15633c7409f5SSatish Balay int ierr, flg; 1564052efed2SBarry Smith 156577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 156674679c65SBarry Smith PetscValidIntPointer(its); 1567c4fc05e7SBarry Smith if (!snes->setup_called) {ierr = SNESSetUp(snes,x); CHKERRQ(ierr);} 1568c4fc05e7SBarry Smith else {snes->vec_sol = snes->vec_sol_always = x;} 15699b94acceSBarry Smith PLogEventBegin(SNES_Solve,snes,0,0,0); 1570c96a6f78SLois Curfman McInnes snes->nfuncs = 0; snes->linear_its = 0; snes->nfailures = 0; 15719b94acceSBarry Smith ierr = (*(snes)->solve)(snes,its); CHKERRQ(ierr); 15729b94acceSBarry Smith PLogEventEnd(SNES_Solve,snes,0,0,0); 15733c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-snes_view", &flg); CHKERRQ(ierr); 15746d4a8577SBarry Smith if (flg) { ierr = SNESView(snes,VIEWER_STDOUT_WORLD); CHKERRQ(ierr); } 15759b94acceSBarry Smith return 0; 15769b94acceSBarry Smith } 15779b94acceSBarry Smith 15789b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 157937fcc0dbSBarry Smith static NRList *__SNESList = 0; 15809b94acceSBarry Smith 15815615d1e5SSatish Balay #undef __FUNC__ 15825615d1e5SSatish Balay #define __FUNC__ "SNESSetType" 15839b94acceSBarry Smith /*@ 15844b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 15859b94acceSBarry Smith 15869b94acceSBarry Smith Input Parameters: 15879b94acceSBarry Smith . snes - the SNES context 15889b94acceSBarry Smith . method - a known method 15899b94acceSBarry Smith 1590ae12b187SLois Curfman McInnes Options Database Command: 1591ae12b187SLois Curfman McInnes $ -snes_type <method> 1592ae12b187SLois Curfman McInnes $ Use -help for a list of available methods 1593ae12b187SLois Curfman McInnes $ (for instance, ls or tr) 1594ae12b187SLois Curfman McInnes 15959b94acceSBarry Smith Notes: 15969b94acceSBarry Smith See "petsc/include/snes.h" for available methods (for instance) 15979b94acceSBarry Smith $ Systems of nonlinear equations: 159840191667SLois Curfman McInnes $ SNES_EQ_LS - Newton's method with line search 159940191667SLois Curfman McInnes $ SNES_EQ_TR - Newton's method with trust region 16009b94acceSBarry Smith $ Unconstrained minimization: 160140191667SLois Curfman McInnes $ SNES_UM_TR - Newton's method with trust region 160240191667SLois Curfman McInnes $ SNES_UM_LS - Newton's method with line search 16039b94acceSBarry Smith 1604ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 1605ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 1606ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 1607ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 1608ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 1609ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 1610ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 1611ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 1612ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 1613ae12b187SLois Curfman McInnes appropriate method. In other words, this routine is for the advanced user. 1614a703fe33SLois Curfman McInnes 1615f536c699SSatish Balay .keywords: SNES, set, method 16169b94acceSBarry Smith @*/ 16174b0e389bSBarry Smith int SNESSetType(SNES snes,SNESType method) 16189b94acceSBarry Smith { 161984cb2905SBarry Smith int ierr; 16209b94acceSBarry Smith int (*r)(SNES); 162177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 16227d1a2b2bSBarry Smith if (snes->type == (int) method) return 0; 16237d1a2b2bSBarry Smith 16249b94acceSBarry Smith /* Get the function pointers for the iterative method requested */ 162584cb2905SBarry Smith if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(); CHKERRQ(ierr);} 1626e3372554SBarry Smith if (!__SNESList) {SETERRQ(1,0,"Could not get methods");} 162737fcc0dbSBarry Smith r = (int (*)(SNES))NRFindRoutine( __SNESList, (int)method, (char *)0 ); 1628e3372554SBarry Smith if (!r) {SETERRQ(1,0,"Unknown method");} 16290452661fSBarry Smith if (snes->data) PetscFree(snes->data); 16309b94acceSBarry Smith snes->set_method_called = 1; 16319b94acceSBarry Smith return (*r)(snes); 16329b94acceSBarry Smith } 16339b94acceSBarry Smith 16349b94acceSBarry Smith /* --------------------------------------------------------------------- */ 16355615d1e5SSatish Balay #undef __FUNC__ 1636*d4bb536fSBarry Smith #define __FUNC__ "SNESRegister" 16379b94acceSBarry Smith /*@C 16389b94acceSBarry Smith SNESRegister - Adds the method to the nonlinear solver package, given 16394b0e389bSBarry Smith a function pointer and a nonlinear solver name of the type SNESType. 16409b94acceSBarry Smith 16419b94acceSBarry Smith Input Parameters: 16422d872ea7SLois Curfman McInnes . name - either a predefined name such as SNES_EQ_LS, or SNES_NEW 16432d872ea7SLois Curfman McInnes to indicate a new user-defined solver 164440191667SLois Curfman McInnes . sname - corresponding string for name 16459b94acceSBarry Smith . create - routine to create method context 16469b94acceSBarry Smith 164784cb2905SBarry Smith Output Parameter: 164884cb2905SBarry Smith . oname - type associated with this new solver 164984cb2905SBarry Smith 16502d872ea7SLois Curfman McInnes Notes: 16512d872ea7SLois Curfman McInnes Multiple user-defined nonlinear solvers can be added by calling 16522d872ea7SLois Curfman McInnes SNESRegister() with the input parameter "name" set to be SNES_NEW; 16532d872ea7SLois Curfman McInnes each call will return a unique solver type in the output 16542d872ea7SLois Curfman McInnes parameter "oname". 16552d872ea7SLois Curfman McInnes 16569b94acceSBarry Smith .keywords: SNES, nonlinear, register 16579b94acceSBarry Smith 16589b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy() 16599b94acceSBarry Smith @*/ 166084cb2905SBarry Smith int SNESRegister(SNESType name,SNESType *oname, char *sname, int (*create)(SNES)) 16619b94acceSBarry Smith { 16629b94acceSBarry Smith int ierr; 166384cb2905SBarry Smith static int numberregistered = 0; 166484cb2905SBarry Smith 1665d252947aSBarry Smith if (name == SNES_NEW) name = (SNESType) ((int) SNES_NEW + numberregistered++); 166684cb2905SBarry Smith 166784cb2905SBarry Smith if (oname) *oname = name; 166837fcc0dbSBarry Smith if (!__SNESList) {ierr = NRCreate(&__SNESList); CHKERRQ(ierr);} 166984cb2905SBarry Smith NRRegister( __SNESList, (int) name, sname, (int (*)(void*))create ); 16709b94acceSBarry Smith return 0; 16719b94acceSBarry Smith } 1672a847f771SSatish Balay 16739b94acceSBarry Smith /* --------------------------------------------------------------------- */ 16745615d1e5SSatish Balay #undef __FUNC__ 1675*d4bb536fSBarry Smith #define __FUNC__ "SNESRegisterDestroy" 16769b94acceSBarry Smith /*@C 16779b94acceSBarry Smith SNESRegisterDestroy - Frees the list of nonlinear solvers that were 16789b94acceSBarry Smith registered by SNESRegister(). 16799b94acceSBarry Smith 16809b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy 16819b94acceSBarry Smith 16829b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll() 16839b94acceSBarry Smith @*/ 16849b94acceSBarry Smith int SNESRegisterDestroy() 16859b94acceSBarry Smith { 168637fcc0dbSBarry Smith if (__SNESList) { 168737fcc0dbSBarry Smith NRDestroy( __SNESList ); 168837fcc0dbSBarry Smith __SNESList = 0; 16899b94acceSBarry Smith } 169084cb2905SBarry Smith SNESRegisterAllCalled = 0; 16919b94acceSBarry Smith return 0; 16929b94acceSBarry Smith } 16939b94acceSBarry Smith 16945615d1e5SSatish Balay #undef __FUNC__ 1695*d4bb536fSBarry Smith #define __FUNC__ "SNESGetTypeFromOptions_Private" 16969b94acceSBarry Smith /* 16974b0e389bSBarry Smith SNESGetTypeFromOptions_Private - Sets the selected method from the 16989b94acceSBarry Smith options database. 16999b94acceSBarry Smith 17009b94acceSBarry Smith Input Parameter: 17019b94acceSBarry Smith . ctx - the SNES context 17029b94acceSBarry Smith 17039b94acceSBarry Smith Output Parameter: 17049b94acceSBarry Smith . method - solver method 17059b94acceSBarry Smith 17069b94acceSBarry Smith Returns: 17079b94acceSBarry Smith Returns 1 if the method is found; 0 otherwise. 17089b94acceSBarry Smith 17099b94acceSBarry Smith Options Database Key: 17104b0e389bSBarry Smith $ -snes_type method 17119b94acceSBarry Smith */ 1712052efed2SBarry Smith int SNESGetTypeFromOptions_Private(SNES ctx,SNESType *method,int *flg) 17139b94acceSBarry Smith { 1714052efed2SBarry Smith int ierr; 17159b94acceSBarry Smith char sbuf[50]; 17165baf8537SBarry Smith 1717052efed2SBarry Smith ierr = OptionsGetString(ctx->prefix,"-snes_type",sbuf,50,flg);CHKERRQ(ierr); 1718052efed2SBarry Smith if (*flg) { 1719052efed2SBarry Smith if (!__SNESList) {ierr = SNESRegisterAll(); CHKERRQ(ierr);} 172037fcc0dbSBarry Smith *method = (SNESType)NRFindID( __SNESList, sbuf ); 17219b94acceSBarry Smith } 17229b94acceSBarry Smith return 0; 17239b94acceSBarry Smith } 17249b94acceSBarry Smith 17255615d1e5SSatish Balay #undef __FUNC__ 1726*d4bb536fSBarry Smith #define __FUNC__ "SNESGetType" 17279b94acceSBarry Smith /*@C 17289a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 17299b94acceSBarry Smith 17309b94acceSBarry Smith Input Parameter: 17314b0e389bSBarry Smith . snes - nonlinear solver context 17329b94acceSBarry Smith 17339b94acceSBarry Smith Output Parameter: 17349a28b0a6SLois Curfman McInnes . method - SNES method (or use PETSC_NULL) 17359a28b0a6SLois Curfman McInnes . name - name of SNES method (or use PETSC_NULL) 17369b94acceSBarry Smith 17379b94acceSBarry Smith .keywords: SNES, nonlinear, get, method, name 17389b94acceSBarry Smith @*/ 17394b0e389bSBarry Smith int SNESGetType(SNES snes, SNESType *method,char **name) 17409b94acceSBarry Smith { 174106a9b0adSLois Curfman McInnes int ierr; 174237fcc0dbSBarry Smith if (!__SNESList) {ierr = SNESRegisterAll(); CHKERRQ(ierr);} 17434b0e389bSBarry Smith if (method) *method = (SNESType) snes->type; 174437fcc0dbSBarry Smith if (name) *name = NRFindName( __SNESList, (int) snes->type ); 17459b94acceSBarry Smith return 0; 17469b94acceSBarry Smith } 17479b94acceSBarry Smith 17489b94acceSBarry Smith #include <stdio.h> 17495615d1e5SSatish Balay #undef __FUNC__ 1750*d4bb536fSBarry Smith #define __FUNC__ "SNESPrintTypes_Private" 17519b94acceSBarry Smith /* 17524b0e389bSBarry Smith SNESPrintTypes_Private - Prints the SNES methods available from the 17539b94acceSBarry Smith options database. 17549b94acceSBarry Smith 17559b94acceSBarry Smith Input Parameters: 175633455573SSatish Balay . comm - communicator (usually MPI_COMM_WORLD) 17579b94acceSBarry Smith . prefix - prefix (usually "-") 17584b0e389bSBarry Smith . name - the options database name (by default "snes_type") 17599b94acceSBarry Smith */ 176033455573SSatish Balay int SNESPrintTypes_Private(MPI_Comm comm,char* prefix,char *name) 17619b94acceSBarry Smith { 17629b94acceSBarry Smith FuncList *entry; 176337fcc0dbSBarry Smith if (!__SNESList) {SNESRegisterAll();} 176437fcc0dbSBarry Smith entry = __SNESList->head; 176577c4ece6SBarry Smith PetscPrintf(comm," %s%s (one of)",prefix,name); 17669b94acceSBarry Smith while (entry) { 176777c4ece6SBarry Smith PetscPrintf(comm," %s",entry->name); 17689b94acceSBarry Smith entry = entry->next; 17699b94acceSBarry Smith } 177077c4ece6SBarry Smith PetscPrintf(comm,"\n"); 17719b94acceSBarry Smith return 0; 17729b94acceSBarry Smith } 17739b94acceSBarry Smith 17745615d1e5SSatish Balay #undef __FUNC__ 1775*d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolution" 17769b94acceSBarry Smith /*@C 17779b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 17789b94acceSBarry Smith stored. 17799b94acceSBarry Smith 17809b94acceSBarry Smith Input Parameter: 17819b94acceSBarry Smith . snes - the SNES context 17829b94acceSBarry Smith 17839b94acceSBarry Smith Output Parameter: 17849b94acceSBarry Smith . x - the solution 17859b94acceSBarry Smith 17869b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 17879b94acceSBarry Smith 1788a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESGetGradient(), SNESGetSolutionUpdate() 17899b94acceSBarry Smith @*/ 17909b94acceSBarry Smith int SNESGetSolution(SNES snes,Vec *x) 17919b94acceSBarry Smith { 179277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 17939b94acceSBarry Smith *x = snes->vec_sol_always; 17949b94acceSBarry Smith return 0; 17959b94acceSBarry Smith } 17969b94acceSBarry Smith 17975615d1e5SSatish Balay #undef __FUNC__ 1798*d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolutionUpdate" 17999b94acceSBarry Smith /*@C 18009b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 18019b94acceSBarry Smith stored. 18029b94acceSBarry Smith 18039b94acceSBarry Smith Input Parameter: 18049b94acceSBarry Smith . snes - the SNES context 18059b94acceSBarry Smith 18069b94acceSBarry Smith Output Parameter: 18079b94acceSBarry Smith . x - the solution update 18089b94acceSBarry Smith 18099b94acceSBarry Smith Notes: 18109b94acceSBarry Smith This vector is implementation dependent. 18119b94acceSBarry Smith 18129b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 18139b94acceSBarry Smith 18149b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction 18159b94acceSBarry Smith @*/ 18169b94acceSBarry Smith int SNESGetSolutionUpdate(SNES snes,Vec *x) 18179b94acceSBarry Smith { 181877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 18199b94acceSBarry Smith *x = snes->vec_sol_update_always; 18209b94acceSBarry Smith return 0; 18219b94acceSBarry Smith } 18229b94acceSBarry Smith 18235615d1e5SSatish Balay #undef __FUNC__ 1824*d4bb536fSBarry Smith #define __FUNC__ "SNESGetFunction" 18259b94acceSBarry Smith /*@C 18263638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 18279b94acceSBarry Smith 18289b94acceSBarry Smith Input Parameter: 18299b94acceSBarry Smith . snes - the SNES context 18309b94acceSBarry Smith 18319b94acceSBarry Smith Output Parameter: 18323638b69dSLois Curfman McInnes . r - the function 18339b94acceSBarry Smith 18349b94acceSBarry Smith Notes: 18359b94acceSBarry Smith SNESGetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only 18369b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 18379b94acceSBarry Smith SNESGetMinimizationFunction() and SNESGetGradient(); 18389b94acceSBarry Smith 1839a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 18409b94acceSBarry Smith 184131615d04SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution(), SNESGetMinimizationFunction(), 184231615d04SLois Curfman McInnes SNESGetGradient() 18439b94acceSBarry Smith @*/ 18449b94acceSBarry Smith int SNESGetFunction(SNES snes,Vec *r) 18459b94acceSBarry Smith { 184677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1847e3372554SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,0, 18487c792091SSatish Balay "For SNES_NONLINEAR_EQUATIONS only"); 18499b94acceSBarry Smith *r = snes->vec_func_always; 18509b94acceSBarry Smith return 0; 18519b94acceSBarry Smith } 18529b94acceSBarry Smith 18535615d1e5SSatish Balay #undef __FUNC__ 1854*d4bb536fSBarry Smith #define __FUNC__ "SNESGetGradient" 18559b94acceSBarry Smith /*@C 18563638b69dSLois Curfman McInnes SNESGetGradient - Returns the vector where the gradient is stored. 18579b94acceSBarry Smith 18589b94acceSBarry Smith Input Parameter: 18599b94acceSBarry Smith . snes - the SNES context 18609b94acceSBarry Smith 18619b94acceSBarry Smith Output Parameter: 18629b94acceSBarry Smith . r - the gradient 18639b94acceSBarry Smith 18649b94acceSBarry Smith Notes: 18659b94acceSBarry Smith SNESGetGradient() is valid for SNES_UNCONSTRAINED_MINIMIZATION methods 18669b94acceSBarry Smith only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 18679b94acceSBarry Smith SNESGetFunction(). 18689b94acceSBarry Smith 18699b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient 18709b94acceSBarry Smith 187131615d04SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESGetSolution(), SNESGetFunction() 18729b94acceSBarry Smith @*/ 18739b94acceSBarry Smith int SNESGetGradient(SNES snes,Vec *r) 18749b94acceSBarry Smith { 187577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1876e3372554SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,0, 187763c41f6aSSatish Balay "For SNES_UNCONSTRAINED_MINIMIZATION only"); 18789b94acceSBarry Smith *r = snes->vec_func_always; 18799b94acceSBarry Smith return 0; 18809b94acceSBarry Smith } 18819b94acceSBarry Smith 18825615d1e5SSatish Balay #undef __FUNC__ 1883*d4bb536fSBarry Smith #define __FUNC__ "SNESGetMinimizationFunction" 18849b94acceSBarry Smith /*@ 18859b94acceSBarry Smith SNESGetMinimizationFunction - Returns the scalar function value for 18869b94acceSBarry Smith unconstrained minimization problems. 18879b94acceSBarry Smith 18889b94acceSBarry Smith Input Parameter: 18899b94acceSBarry Smith . snes - the SNES context 18909b94acceSBarry Smith 18919b94acceSBarry Smith Output Parameter: 18929b94acceSBarry Smith . r - the function 18939b94acceSBarry Smith 18949b94acceSBarry Smith Notes: 18959b94acceSBarry Smith SNESGetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 18969b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 18979b94acceSBarry Smith SNESGetFunction(). 18989b94acceSBarry Smith 18999b94acceSBarry Smith .keywords: SNES, nonlinear, get, function 19009b94acceSBarry Smith 190131615d04SLois Curfman McInnes .seealso: SNESGetGradient(), SNESGetSolution(), SNESGetFunction() 19029b94acceSBarry Smith @*/ 19039b94acceSBarry Smith int SNESGetMinimizationFunction(SNES snes,double *r) 19049b94acceSBarry Smith { 190577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 190674679c65SBarry Smith PetscValidScalarPointer(r); 1907e3372554SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,0, 190863c41f6aSSatish Balay "For SNES_UNCONSTRAINED_MINIMIZATION only"); 19099b94acceSBarry Smith *r = snes->fc; 19109b94acceSBarry Smith return 0; 19119b94acceSBarry Smith } 19129b94acceSBarry Smith 19135615d1e5SSatish Balay #undef __FUNC__ 1914*d4bb536fSBarry Smith #define __FUNC__ "SNESSetOptionsPrefix" 19153c7409f5SSatish Balay /*@C 19163c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 1917d850072dSLois Curfman McInnes SNES options in the database. 19183c7409f5SSatish Balay 19193c7409f5SSatish Balay Input Parameter: 19203c7409f5SSatish Balay . snes - the SNES context 19213c7409f5SSatish Balay . prefix - the prefix to prepend to all option names 19223c7409f5SSatish Balay 1923d850072dSLois Curfman McInnes Notes: 1924a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 1925a83b1b31SSatish Balay The first character of all runtime options is AUTOMATICALLY the 1926a83b1b31SSatish Balay hyphen. 1927d850072dSLois Curfman McInnes 19283c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 1929a86d99e1SLois Curfman McInnes 1930a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 19313c7409f5SSatish Balay @*/ 19323c7409f5SSatish Balay int SNESSetOptionsPrefix(SNES snes,char *prefix) 19333c7409f5SSatish Balay { 19343c7409f5SSatish Balay int ierr; 19353c7409f5SSatish Balay 193677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1937639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 19383c7409f5SSatish Balay ierr = SLESSetOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 19393c7409f5SSatish Balay return 0; 19403c7409f5SSatish Balay } 19413c7409f5SSatish Balay 19425615d1e5SSatish Balay #undef __FUNC__ 1943*d4bb536fSBarry Smith #define __FUNC__ "SNESAppendOptionsPrefix" 19443c7409f5SSatish Balay /*@C 1945f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 1946d850072dSLois Curfman McInnes SNES options in the database. 19473c7409f5SSatish Balay 19483c7409f5SSatish Balay Input Parameter: 19493c7409f5SSatish Balay . snes - the SNES context 19503c7409f5SSatish Balay . prefix - the prefix to prepend to all option names 19513c7409f5SSatish Balay 1952d850072dSLois Curfman McInnes Notes: 1953a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 1954a83b1b31SSatish Balay The first character of all runtime options is AUTOMATICALLY the 1955a83b1b31SSatish Balay hyphen. 1956d850072dSLois Curfman McInnes 19573c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 1958a86d99e1SLois Curfman McInnes 1959a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 19603c7409f5SSatish Balay @*/ 19613c7409f5SSatish Balay int SNESAppendOptionsPrefix(SNES snes,char *prefix) 19623c7409f5SSatish Balay { 19633c7409f5SSatish Balay int ierr; 19643c7409f5SSatish Balay 196577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1966639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 19673c7409f5SSatish Balay ierr = SLESAppendOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 19683c7409f5SSatish Balay return 0; 19693c7409f5SSatish Balay } 19703c7409f5SSatish Balay 19715615d1e5SSatish Balay #undef __FUNC__ 1972*d4bb536fSBarry Smith #define __FUNC__ "SNESGetOptionsPrefix" 1973c83590e2SSatish Balay /*@ 19743c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 19753c7409f5SSatish Balay SNES options in the database. 19763c7409f5SSatish Balay 19773c7409f5SSatish Balay Input Parameter: 19783c7409f5SSatish Balay . snes - the SNES context 19793c7409f5SSatish Balay 19803c7409f5SSatish Balay Output Parameter: 19813c7409f5SSatish Balay . prefix - pointer to the prefix string used 19823c7409f5SSatish Balay 19833c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 1984a86d99e1SLois Curfman McInnes 1985a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 19863c7409f5SSatish Balay @*/ 19873c7409f5SSatish Balay int SNESGetOptionsPrefix(SNES snes,char **prefix) 19883c7409f5SSatish Balay { 19893c7409f5SSatish Balay int ierr; 19903c7409f5SSatish Balay 199177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1992639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 19933c7409f5SSatish Balay return 0; 19943c7409f5SSatish Balay } 19953c7409f5SSatish Balay 19963c7409f5SSatish Balay 19979b94acceSBarry Smith 19989b94acceSBarry Smith 19999b94acceSBarry Smith 2000