1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER 2*e24b481bSBarry Smith static char vcid[] = "$Id: snes.c,v 1.160 1998/10/24 02:25:31 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 1084cb2905SBarry Smith int SNESRegisterAllCalled = 0; 11488ecbafSBarry Smith FList SNESList = 0; 1282bf6240SBarry Smith 135615d1e5SSatish Balay #undef __FUNC__ 14d4bb536fSBarry Smith #define __FUNC__ "SNESView" 159b94acceSBarry Smith /*@ 169b94acceSBarry Smith SNESView - Prints the SNES data structure. 179b94acceSBarry Smith 18fee21e36SBarry Smith Collective on SNES, unless Viewer is VIEWER_STDOUT_SELF 19fee21e36SBarry Smith 20c7afd0dbSLois Curfman McInnes Input Parameters: 21c7afd0dbSLois Curfman McInnes + SNES - the SNES context 22c7afd0dbSLois Curfman McInnes - viewer - visualization context 23c7afd0dbSLois Curfman McInnes 249b94acceSBarry Smith Options Database Key: 25c8a8ba5cSLois Curfman McInnes . -snes_view - Calls SNESView() at end of SNESSolve() 269b94acceSBarry Smith 279b94acceSBarry Smith Notes: 289b94acceSBarry Smith The available visualization contexts include 29c7afd0dbSLois Curfman McInnes + VIEWER_STDOUT_SELF - standard output (default) 30c7afd0dbSLois Curfman McInnes - VIEWER_STDOUT_WORLD - synchronized standard 31c8a8ba5cSLois Curfman McInnes output where only the first processor opens 32c8a8ba5cSLois Curfman McInnes the file. All other processors send their 33c8a8ba5cSLois Curfman McInnes data to the first processor to print. 349b94acceSBarry Smith 353e081fefSLois Curfman McInnes The user can open an alternative visualization context with 361edb8f11SLois Curfman McInnes 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 513a40ed3dSBarry Smith PetscFunctionBegin; 5274679c65SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 5374679c65SBarry Smith if (viewer) {PetscValidHeader(viewer);} 5474679c65SBarry Smith else { viewer = VIEWER_STDOUT_SELF; } 5574679c65SBarry Smith 5619bcc07fSBarry Smith ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr); 5719bcc07fSBarry Smith if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER) { 5890ace30eSBarry Smith ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr); 5977c4ece6SBarry Smith PetscFPrintf(snes->comm,fd,"SNES Object:\n"); 6082bf6240SBarry Smith SNESGetType(snes,&method); 6177c4ece6SBarry Smith PetscFPrintf(snes->comm,fd," method: %s\n",method); 62e1311b90SBarry Smith if (snes->view) (*snes->view)(snes,viewer); 6377c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 649b94acceSBarry Smith " maximum iterations=%d, maximum function evaluations=%d\n", 659b94acceSBarry Smith snes->max_its,snes->max_funcs); 6677c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 679b94acceSBarry Smith " tolerances: relative=%g, absolute=%g, truncation=%g, solution=%g\n", 689b94acceSBarry Smith snes->rtol, snes->atol, snes->trunctol, snes->xtol); 697a00f4a9SLois Curfman McInnes PetscFPrintf(snes->comm,fd, 707a00f4a9SLois Curfman McInnes " total number of linear solver iterations=%d\n",snes->linear_its); 71ae3c334cSLois Curfman McInnes PetscFPrintf(snes->comm,fd, 7268632166SLois Curfman McInnes " total number of function evaluations=%d\n",snes->nfuncs); 739b94acceSBarry Smith if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) 7477c4ece6SBarry Smith PetscFPrintf(snes->comm,fd," min function tolerance=%g\n",snes->fmin); 759b94acceSBarry Smith if (snes->ksp_ewconv) { 769b94acceSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 779b94acceSBarry Smith if (kctx) { 7877c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 799b94acceSBarry Smith " Eisenstat-Walker computation of KSP relative tolerance (version %d)\n", 809b94acceSBarry Smith kctx->version); 8177c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 829b94acceSBarry Smith " rtol_0=%g, rtol_max=%g, threshold=%g\n",kctx->rtol_0, 839b94acceSBarry Smith kctx->rtol_max,kctx->threshold); 8477c4ece6SBarry Smith PetscFPrintf(snes->comm,fd," gamma=%g, alpha=%g, alpha2=%g\n", 859b94acceSBarry Smith kctx->gamma,kctx->alpha,kctx->alpha2); 869b94acceSBarry Smith } 879b94acceSBarry Smith } 88c30f285eSLois Curfman McInnes } else if (vtype == STRING_VIEWER) { 8982bf6240SBarry Smith SNESGetType(snes,&method); 90c30f285eSLois Curfman McInnes ViewerStringSPrintf(viewer," %-3.3s",method); 9119bcc07fSBarry Smith } 929b94acceSBarry Smith SNESGetSLES(snes,&sles); 939b94acceSBarry Smith ierr = SLESView(sles,viewer); CHKERRQ(ierr); 943a40ed3dSBarry Smith PetscFunctionReturn(0); 959b94acceSBarry Smith } 969b94acceSBarry Smith 97639f9d9dSBarry Smith /* 98639f9d9dSBarry Smith We retain a list of functions that also take SNES command 99639f9d9dSBarry Smith line options. These are called at the end SNESSetFromOptions() 100639f9d9dSBarry Smith */ 101639f9d9dSBarry Smith #define MAXSETFROMOPTIONS 5 102639f9d9dSBarry Smith static int numberofsetfromoptions; 103639f9d9dSBarry Smith static int (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES); 104639f9d9dSBarry Smith 1055615d1e5SSatish Balay #undef __FUNC__ 106d4bb536fSBarry Smith #define __FUNC__ "SNESAddOptionsChecker" 107639f9d9dSBarry Smith /*@ 108639f9d9dSBarry Smith SNESAddOptionsChecker - Adds an additional function to check for SNES options. 109639f9d9dSBarry Smith 110c7afd0dbSLois Curfman McInnes Not Collective 111c7afd0dbSLois Curfman McInnes 112639f9d9dSBarry Smith Input Parameter: 113639f9d9dSBarry Smith . snescheck - function that checks for options 114639f9d9dSBarry Smith 115639f9d9dSBarry Smith .seealso: SNESSetFromOptions() 116639f9d9dSBarry Smith @*/ 117639f9d9dSBarry Smith int SNESAddOptionsChecker(int (*snescheck)(SNES) ) 118639f9d9dSBarry Smith { 1193a40ed3dSBarry Smith PetscFunctionBegin; 120639f9d9dSBarry Smith if (numberofsetfromoptions >= MAXSETFROMOPTIONS) { 121a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Too many options checkers, only 5 allowed"); 122639f9d9dSBarry Smith } 123639f9d9dSBarry Smith 124639f9d9dSBarry Smith othersetfromoptions[numberofsetfromoptions++] = snescheck; 1253a40ed3dSBarry Smith PetscFunctionReturn(0); 126639f9d9dSBarry Smith } 127639f9d9dSBarry Smith 1285615d1e5SSatish Balay #undef __FUNC__ 1295615d1e5SSatish Balay #define __FUNC__ "SNESSetFromOptions" 1309b94acceSBarry Smith /*@ 131682d7d0cSBarry Smith SNESSetFromOptions - Sets various SNES and SLES parameters from user options. 1329b94acceSBarry Smith 133c7afd0dbSLois Curfman McInnes Collective on SNES 134c7afd0dbSLois Curfman McInnes 1359b94acceSBarry Smith Input Parameter: 1369b94acceSBarry Smith . snes - the SNES context 1379b94acceSBarry Smith 13882738288SBarry Smith Options Database: 13982738288SBarry Smith + -snes_type type - SNES_EQ_LS, SNES_EQ_TR, SNES_UM_TR, SNES_UM_LS etc 14082738288SBarry Smith . -snes_stol - convergence tolerance in terms of the norm 14182738288SBarry Smith of the change in the solution between steps 14282738288SBarry Smith . -snes_atol atol - absolute tolerance of residual norm 14382738288SBarry Smith . -snes_rtol rtol - relative decrease in tolerance norm from initial 14482738288SBarry Smith . -snes_max_it max_it - maximum number of iterations 14582738288SBarry Smith . -snes_max_funcs max_funcs - maximum number of function evaluations 14682738288SBarry Smith . -snes_trtol trtol - trust region tolerance 14782738288SBarry Smith . -snes_no_convergence_test - skip convergence test in nonlinear or minimization 14882738288SBarry Smith solver; hence iterations will continue until max_it 14982738288SBarry Smith or some other criteria is reached. Saves expense 15082738288SBarry Smith of convergence test 15182738288SBarry Smith . -snes_monitor - prints residual norm at each iteration 15282738288SBarry Smith . -snes_xmonitor - plots residual norm at each iteration 153*e24b481bSBarry Smith . -snes_fd - use finite differences to compute Jacobian; very slow, only for testing 154*e24b481bSBarry Smith - -snes_mf_ksp_monitor - if using matrix-free multiply then prints h at each KSP iteration 15582738288SBarry Smith 15682738288SBarry Smith Options Database for Eisenstat-Walker method: 15782738288SBarry Smith + -snes_ksp_eq_conv - use Eisenstat-Walker method for determining linear system convergence 15882738288SBarry Smith . -snes_ksp_eq_version ver - version of Eisenstat-Walker method 15982738288SBarry Smith . -snes_ksp_ew_rtol0 rtol0 - 16082738288SBarry Smith . -snes_ksp_ew_rtolmax rtolmax - 16182738288SBarry Smith . -snes_ksp_ew_gamma gamma - 16282738288SBarry Smith . -snes_ksp_ew_alpha alpha - 16382738288SBarry Smith . -snes_ksp_ew_alpha2 alpha2 - 16482738288SBarry Smith - -snes_ksp_ew_threshold threshold - 16582738288SBarry Smith 16611ca99fdSLois Curfman McInnes Notes: 16711ca99fdSLois Curfman McInnes To see all options, run your program with the -help option or consult 16811ca99fdSLois Curfman McInnes the users manual. 16983e2fdc7SBarry Smith 1709b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 1719b94acceSBarry Smith 172a86d99e1SLois Curfman McInnes .seealso: SNESPrintHelp(), SNESSetOptionsPrefix() 1739b94acceSBarry Smith @*/ 1749b94acceSBarry Smith int SNESSetFromOptions(SNES snes) 1759b94acceSBarry Smith { 17682bf6240SBarry Smith char method[256]; 1779b94acceSBarry Smith double tmp; 1789b94acceSBarry Smith SLES sles; 17981f57ec7SBarry Smith int ierr, flg,i,loc[4],nmax = 4; 1803c7409f5SSatish Balay int version = PETSC_DEFAULT; 1819b94acceSBarry Smith double rtol_0 = PETSC_DEFAULT; 1829b94acceSBarry Smith double rtol_max = PETSC_DEFAULT; 1839b94acceSBarry Smith double gamma2 = PETSC_DEFAULT; 1849b94acceSBarry Smith double alpha = PETSC_DEFAULT; 1859b94acceSBarry Smith double alpha2 = PETSC_DEFAULT; 1869b94acceSBarry Smith double threshold = PETSC_DEFAULT; 1879b94acceSBarry Smith 1883a40ed3dSBarry Smith PetscFunctionBegin; 18981f57ec7SBarry Smith loc[0] = PETSC_DECIDE; loc[1] = PETSC_DECIDE; loc[2] = 300; loc[3] = 300; 19081f57ec7SBarry Smith 19177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 19282bf6240SBarry Smith if (snes->setupcalled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call prior to SNESSetUp"); 193ca161407SBarry Smith 19482bf6240SBarry Smith if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL);CHKERRQ(ierr);} 19582bf6240SBarry Smith ierr = OptionsGetString(snes->prefix,"-snes_type",method,256,&flg); 196052efed2SBarry Smith if (flg) { 19782bf6240SBarry Smith ierr = SNESSetType(snes,(SNESType) method); CHKERRQ(ierr); 1985334005bSBarry Smith } 1993c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_stol",&tmp, &flg); CHKERRQ(ierr); 200d64ed03dSBarry Smith if (flg) { 201d64ed03dSBarry Smith ierr = SNESSetTolerances(snes,PETSC_DEFAULT,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 202d64ed03dSBarry Smith } 2033c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_atol",&tmp, &flg); CHKERRQ(ierr); 204d64ed03dSBarry Smith if (flg) { 205d64ed03dSBarry Smith ierr = SNESSetTolerances(snes,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 206d64ed03dSBarry Smith } 2073c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_rtol",&tmp, &flg); CHKERRQ(ierr); 208d64ed03dSBarry Smith if (flg) { 209d64ed03dSBarry Smith ierr = SNESSetTolerances(snes,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 210d64ed03dSBarry Smith } 2113c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_it",&snes->max_its, &flg); CHKERRQ(ierr); 2123c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_funcs",&snes->max_funcs, &flg);CHKERRQ(ierr); 213d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_trtol",&tmp, &flg); CHKERRQ(ierr); 214d64ed03dSBarry Smith if (flg) { ierr = SNESSetTrustRegionTolerance(snes,tmp); CHKERRQ(ierr); } 215d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_fmin",&tmp, &flg); CHKERRQ(ierr); 216d64ed03dSBarry Smith if (flg) { ierr = SNESSetMinimizationFunctionTolerance(snes,tmp); CHKERRQ(ierr);} 2173c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_ksp_ew_conv", &flg); CHKERRQ(ierr); 2183c7409f5SSatish Balay if (flg) { snes->ksp_ewconv = 1; } 219b18e04deSLois Curfman McInnes ierr = OptionsGetInt(snes->prefix,"-snes_ksp_ew_version",&version,&flg); CHKERRQ(ierr); 220b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtol0",&rtol_0,&flg); CHKERRQ(ierr); 221b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtolmax",&rtol_max,&flg);CHKERRQ(ierr); 222b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_gamma",&gamma2,&flg); CHKERRQ(ierr); 223b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha",&alpha,&flg); CHKERRQ(ierr); 224b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha2",&alpha2,&flg); CHKERRQ(ierr); 225b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_threshold",&threshold,&flg);CHKERRQ(ierr); 22693c39befSBarry Smith 22793c39befSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_no_convergence_test",&flg); CHKERRQ(ierr); 22893c39befSBarry Smith if (flg) {snes->converged = 0;} 22993c39befSBarry Smith 2309b94acceSBarry Smith ierr = SNES_KSP_SetParametersEW(snes,version,rtol_0,rtol_max,gamma2,alpha, 2319b94acceSBarry Smith alpha2,threshold); CHKERRQ(ierr); 2325bbad29bSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_cancelmonitors",&flg); CHKERRQ(ierr); 2335cd90555SBarry Smith if (flg) {ierr = SNESClearMonitor(snes);CHKERRQ(ierr);} 2343c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_monitor",&flg); CHKERRQ(ierr); 235639f9d9dSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESDefaultMonitor,0);CHKERRQ(ierr);} 2363c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_smonitor",&flg); CHKERRQ(ierr); 237d31a3109SLois Curfman McInnes if (flg) { 238639f9d9dSBarry Smith ierr = SNESSetMonitor(snes,SNESDefaultSMonitor,0); CHKERRQ(ierr); 239d31a3109SLois Curfman McInnes } 24081f57ec7SBarry Smith ierr = OptionsGetIntArray(snes->prefix,"-snes_xmonitor",loc,&nmax,&flg);CHKERRQ(ierr); 2413c7409f5SSatish Balay if (flg) { 24217699dbbSLois Curfman McInnes int rank = 0; 243d7e8b826SBarry Smith DrawLG lg; 24417699dbbSLois Curfman McInnes MPI_Initialized(&rank); 24517699dbbSLois Curfman McInnes if (rank) MPI_Comm_rank(snes->comm,&rank); 24617699dbbSLois Curfman McInnes if (!rank) { 24781f57ec7SBarry Smith ierr = SNESLGMonitorCreate(0,0,loc[0],loc[1],loc[2],loc[3],&lg); CHKERRQ(ierr); 2489b94acceSBarry Smith ierr = SNESSetMonitor(snes,SNESLGMonitor,(void *)lg); CHKERRQ(ierr); 249f8c826e1SBarry Smith snes->xmonitor = lg; 2509b94acceSBarry Smith PLogObjectParent(snes,lg); 2519b94acceSBarry Smith } 2529b94acceSBarry Smith } 253*e24b481bSBarry Smith 254*e24b481bSBarry Smith 2553c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_fd", &flg); CHKERRQ(ierr); 2563c7409f5SSatish Balay if (flg && snes->method_class == SNES_NONLINEAR_EQUATIONS) { 2579b94acceSBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre, 2589b94acceSBarry Smith SNESDefaultComputeJacobian,snes->funP); CHKERRQ(ierr); 259981c4779SBarry Smith PLogInfo(snes,"SNESSetFromOptions: Setting default finite difference Jacobian matrix\n"); 260981c4779SBarry Smith } else if (flg && snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 26131615d04SLois Curfman McInnes ierr = SNESSetHessian(snes,snes->jacobian,snes->jacobian_pre, 26231615d04SLois Curfman McInnes SNESDefaultComputeHessian,snes->funP); CHKERRQ(ierr); 263d64ed03dSBarry Smith PLogInfo(snes,"SNESSetFromOptions: Setting default finite difference Hessian matrix\n"); 2649b94acceSBarry Smith } 265639f9d9dSBarry Smith 266639f9d9dSBarry Smith for ( i=0; i<numberofsetfromoptions; i++ ) { 267639f9d9dSBarry Smith ierr = (*othersetfromoptions[i])(snes); CHKERRQ(ierr); 268639f9d9dSBarry Smith } 269639f9d9dSBarry Smith 2709b94acceSBarry Smith ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 2719b94acceSBarry Smith ierr = SLESSetFromOptions(sles); CHKERRQ(ierr); 27293993e2dSLois Curfman McInnes 273*e24b481bSBarry Smith /* set the special KSP monitor for matrix-free application */ 274*e24b481bSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_mf_ksp_monitor",&flg); CHKERRQ(ierr); 275*e24b481bSBarry Smith if (flg) { 276*e24b481bSBarry Smith KSP ksp; 277*e24b481bSBarry Smith ierr = SLESGetKSP(sles,&ksp);CHKERRQ(ierr); 278*e24b481bSBarry Smith ierr = KSPSetMonitor(ksp,MatSNESFDMFKSPMonitor,PETSC_NULL);CHKERRQ(ierr); 279*e24b481bSBarry Smith } 280*e24b481bSBarry Smith 28182bf6240SBarry Smith /* 28293993e2dSLois Curfman McInnes Since the private setfromoptions requires the type to have 28393993e2dSLois Curfman McInnes been set already, we make sure a type is set by this time. 28482bf6240SBarry Smith */ 28582bf6240SBarry Smith if (!snes->type_name) { 28682bf6240SBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 28782bf6240SBarry Smith ierr = SNESSetType(snes,SNES_EQ_LS); CHKERRQ(ierr); 28882bf6240SBarry Smith } else { 28982bf6240SBarry Smith ierr = SNESSetType(snes,SNES_UM_TR); CHKERRQ(ierr); 29082bf6240SBarry Smith } 29182bf6240SBarry Smith } 29282bf6240SBarry Smith 29382bf6240SBarry Smith ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 29482bf6240SBarry Smith if (flg) { ierr = SNESPrintHelp(snes); CHKERRQ(ierr);} 29582bf6240SBarry Smith 2963a40ed3dSBarry Smith if (snes->setfromoptions) { 2973a40ed3dSBarry Smith ierr = (*snes->setfromoptions)(snes);CHKERRQ(ierr); 2983a40ed3dSBarry Smith } 2993a40ed3dSBarry Smith PetscFunctionReturn(0); 3009b94acceSBarry Smith } 3019b94acceSBarry Smith 302a847f771SSatish Balay 3035615d1e5SSatish Balay #undef __FUNC__ 304d4bb536fSBarry Smith #define __FUNC__ "SNESSetApplicationContext" 3059b94acceSBarry Smith /*@ 3069b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 3079b94acceSBarry Smith the nonlinear solvers. 3089b94acceSBarry Smith 309fee21e36SBarry Smith Collective on SNES 310fee21e36SBarry Smith 311c7afd0dbSLois Curfman McInnes Input Parameters: 312c7afd0dbSLois Curfman McInnes + snes - the SNES context 313c7afd0dbSLois Curfman McInnes - usrP - optional user context 314c7afd0dbSLois Curfman McInnes 3159b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 3169b94acceSBarry Smith 3179b94acceSBarry Smith .seealso: SNESGetApplicationContext() 3189b94acceSBarry Smith @*/ 3199b94acceSBarry Smith int SNESSetApplicationContext(SNES snes,void *usrP) 3209b94acceSBarry Smith { 3213a40ed3dSBarry Smith PetscFunctionBegin; 32277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3239b94acceSBarry Smith snes->user = usrP; 3243a40ed3dSBarry Smith PetscFunctionReturn(0); 3259b94acceSBarry Smith } 32674679c65SBarry Smith 3275615d1e5SSatish Balay #undef __FUNC__ 328d4bb536fSBarry Smith #define __FUNC__ "SNESGetApplicationContext" 3299b94acceSBarry Smith /*@C 3309b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 3319b94acceSBarry Smith nonlinear solvers. 3329b94acceSBarry Smith 333c7afd0dbSLois Curfman McInnes Not Collective 334c7afd0dbSLois Curfman McInnes 3359b94acceSBarry Smith Input Parameter: 3369b94acceSBarry Smith . snes - SNES context 3379b94acceSBarry Smith 3389b94acceSBarry Smith Output Parameter: 3399b94acceSBarry Smith . usrP - user context 3409b94acceSBarry Smith 3419b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 3429b94acceSBarry Smith 3439b94acceSBarry Smith .seealso: SNESSetApplicationContext() 3449b94acceSBarry Smith @*/ 3459b94acceSBarry Smith int SNESGetApplicationContext( SNES snes, void **usrP ) 3469b94acceSBarry Smith { 3473a40ed3dSBarry Smith PetscFunctionBegin; 34877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3499b94acceSBarry Smith *usrP = snes->user; 3503a40ed3dSBarry Smith PetscFunctionReturn(0); 3519b94acceSBarry Smith } 35274679c65SBarry Smith 3535615d1e5SSatish Balay #undef __FUNC__ 354d4bb536fSBarry Smith #define __FUNC__ "SNESGetIterationNumber" 3559b94acceSBarry Smith /*@ 3569b94acceSBarry Smith SNESGetIterationNumber - Gets the current iteration number of the 3579b94acceSBarry Smith nonlinear solver. 3589b94acceSBarry Smith 359c7afd0dbSLois Curfman McInnes Not Collective 360c7afd0dbSLois Curfman McInnes 3619b94acceSBarry Smith Input Parameter: 3629b94acceSBarry Smith . snes - SNES context 3639b94acceSBarry Smith 3649b94acceSBarry Smith Output Parameter: 3659b94acceSBarry Smith . iter - iteration number 3669b94acceSBarry Smith 3679b94acceSBarry Smith .keywords: SNES, nonlinear, get, iteration, number 3689b94acceSBarry Smith @*/ 3699b94acceSBarry Smith int SNESGetIterationNumber(SNES snes,int* iter) 3709b94acceSBarry Smith { 3713a40ed3dSBarry Smith PetscFunctionBegin; 37277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 37374679c65SBarry Smith PetscValidIntPointer(iter); 3749b94acceSBarry Smith *iter = snes->iter; 3753a40ed3dSBarry Smith PetscFunctionReturn(0); 3769b94acceSBarry Smith } 37774679c65SBarry Smith 3785615d1e5SSatish Balay #undef __FUNC__ 3795615d1e5SSatish Balay #define __FUNC__ "SNESGetFunctionNorm" 3809b94acceSBarry Smith /*@ 3819b94acceSBarry Smith SNESGetFunctionNorm - Gets the norm of the current function that was set 3829b94acceSBarry Smith with SNESSSetFunction(). 3839b94acceSBarry Smith 384c7afd0dbSLois Curfman McInnes Collective on SNES 385c7afd0dbSLois Curfman McInnes 3869b94acceSBarry Smith Input Parameter: 3879b94acceSBarry Smith . snes - SNES context 3889b94acceSBarry Smith 3899b94acceSBarry Smith Output Parameter: 3909b94acceSBarry Smith . fnorm - 2-norm of function 3919b94acceSBarry Smith 3929b94acceSBarry Smith Note: 3939b94acceSBarry Smith SNESGetFunctionNorm() is valid for SNES_NONLINEAR_EQUATIONS methods only. 394a86d99e1SLois Curfman McInnes A related routine for SNES_UNCONSTRAINED_MINIMIZATION methods is 395a86d99e1SLois Curfman McInnes SNESGetGradientNorm(). 3969b94acceSBarry Smith 3979b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm 398a86d99e1SLois Curfman McInnes 399a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction() 4009b94acceSBarry Smith @*/ 4019b94acceSBarry Smith int SNESGetFunctionNorm(SNES snes,Scalar *fnorm) 4029b94acceSBarry Smith { 4033a40ed3dSBarry Smith PetscFunctionBegin; 40477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 40574679c65SBarry Smith PetscValidScalarPointer(fnorm); 40674679c65SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 407d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_NONLINEAR_EQUATIONS only"); 40874679c65SBarry Smith } 4099b94acceSBarry Smith *fnorm = snes->norm; 4103a40ed3dSBarry Smith PetscFunctionReturn(0); 4119b94acceSBarry Smith } 41274679c65SBarry Smith 4135615d1e5SSatish Balay #undef __FUNC__ 4145615d1e5SSatish Balay #define __FUNC__ "SNESGetGradientNorm" 4159b94acceSBarry Smith /*@ 4169b94acceSBarry Smith SNESGetGradientNorm - Gets the norm of the current gradient that was set 4179b94acceSBarry Smith with SNESSSetGradient(). 4189b94acceSBarry Smith 419c7afd0dbSLois Curfman McInnes Collective on SNES 420c7afd0dbSLois Curfman McInnes 4219b94acceSBarry Smith Input Parameter: 4229b94acceSBarry Smith . snes - SNES context 4239b94acceSBarry Smith 4249b94acceSBarry Smith Output Parameter: 4259b94acceSBarry Smith . fnorm - 2-norm of gradient 4269b94acceSBarry Smith 4279b94acceSBarry Smith Note: 4289b94acceSBarry Smith SNESGetGradientNorm() is valid for SNES_UNCONSTRAINED_MINIMIZATION 429a86d99e1SLois Curfman McInnes methods only. A related routine for SNES_NONLINEAR_EQUATIONS methods 430a86d99e1SLois Curfman McInnes is SNESGetFunctionNorm(). 4319b94acceSBarry Smith 4329b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient, norm 433a86d99e1SLois Curfman McInnes 434a86d99e1SLois Curfman McInnes .seelso: SNESSetGradient() 4359b94acceSBarry Smith @*/ 4369b94acceSBarry Smith int SNESGetGradientNorm(SNES snes,Scalar *gnorm) 4379b94acceSBarry Smith { 4383a40ed3dSBarry Smith PetscFunctionBegin; 43977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 44074679c65SBarry Smith PetscValidScalarPointer(gnorm); 44174679c65SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 442d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 44374679c65SBarry Smith } 4449b94acceSBarry Smith *gnorm = snes->norm; 4453a40ed3dSBarry Smith PetscFunctionReturn(0); 4469b94acceSBarry Smith } 44774679c65SBarry Smith 4485615d1e5SSatish Balay #undef __FUNC__ 449d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberUnsuccessfulSteps" 4509b94acceSBarry Smith /*@ 4519b94acceSBarry Smith SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps 4529b94acceSBarry Smith attempted by the nonlinear solver. 4539b94acceSBarry Smith 454c7afd0dbSLois Curfman McInnes Not Collective 455c7afd0dbSLois Curfman McInnes 4569b94acceSBarry Smith Input Parameter: 4579b94acceSBarry Smith . snes - SNES context 4589b94acceSBarry Smith 4599b94acceSBarry Smith Output Parameter: 4609b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 4619b94acceSBarry Smith 462c96a6f78SLois Curfman McInnes Notes: 463c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 464c96a6f78SLois Curfman McInnes 4659b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 4669b94acceSBarry Smith @*/ 4679b94acceSBarry Smith int SNESGetNumberUnsuccessfulSteps(SNES snes,int* nfails) 4689b94acceSBarry Smith { 4693a40ed3dSBarry Smith PetscFunctionBegin; 47077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 47174679c65SBarry Smith PetscValidIntPointer(nfails); 4729b94acceSBarry Smith *nfails = snes->nfailures; 4733a40ed3dSBarry Smith PetscFunctionReturn(0); 4749b94acceSBarry Smith } 475a847f771SSatish Balay 4765615d1e5SSatish Balay #undef __FUNC__ 477d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberLinearIterations" 478c96a6f78SLois Curfman McInnes /*@ 479c96a6f78SLois Curfman McInnes SNESGetNumberLinearIterations - Gets the total number of linear iterations 480c96a6f78SLois Curfman McInnes used by the nonlinear solver. 481c96a6f78SLois Curfman McInnes 482c7afd0dbSLois Curfman McInnes Not Collective 483c7afd0dbSLois Curfman McInnes 484c96a6f78SLois Curfman McInnes Input Parameter: 485c96a6f78SLois Curfman McInnes . snes - SNES context 486c96a6f78SLois Curfman McInnes 487c96a6f78SLois Curfman McInnes Output Parameter: 488c96a6f78SLois Curfman McInnes . lits - number of linear iterations 489c96a6f78SLois Curfman McInnes 490c96a6f78SLois Curfman McInnes Notes: 491c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 492c96a6f78SLois Curfman McInnes 493c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations 494c96a6f78SLois Curfman McInnes @*/ 49586bddb7dSBarry Smith int SNESGetNumberLinearIterations(SNES snes,int* lits) 496c96a6f78SLois Curfman McInnes { 4973a40ed3dSBarry Smith PetscFunctionBegin; 498c96a6f78SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 499c96a6f78SLois Curfman McInnes PetscValidIntPointer(lits); 500c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 5013a40ed3dSBarry Smith PetscFunctionReturn(0); 502c96a6f78SLois Curfman McInnes } 503c96a6f78SLois Curfman McInnes 504c96a6f78SLois Curfman McInnes #undef __FUNC__ 505d4bb536fSBarry Smith #define __FUNC__ "SNESGetSLES" 5069b94acceSBarry Smith /*@C 5079b94acceSBarry Smith SNESGetSLES - Returns the SLES context for a SNES solver. 5089b94acceSBarry Smith 509c7afd0dbSLois Curfman McInnes Not Collective, but if SNES object is parallel, then SLES object is parallel 510c7afd0dbSLois Curfman McInnes 5119b94acceSBarry Smith Input Parameter: 5129b94acceSBarry Smith . snes - the SNES context 5139b94acceSBarry Smith 5149b94acceSBarry Smith Output Parameter: 5159b94acceSBarry Smith . sles - the SLES context 5169b94acceSBarry Smith 5179b94acceSBarry Smith Notes: 5189b94acceSBarry Smith The user can then directly manipulate the SLES context to set various 5199b94acceSBarry Smith options, etc. Likewise, the user can then extract and manipulate the 5209b94acceSBarry Smith KSP and PC contexts as well. 5219b94acceSBarry Smith 5229b94acceSBarry Smith .keywords: SNES, nonlinear, get, SLES, context 5239b94acceSBarry Smith 5249b94acceSBarry Smith .seealso: SLESGetPC(), SLESGetKSP() 5259b94acceSBarry Smith @*/ 5269b94acceSBarry Smith int SNESGetSLES(SNES snes,SLES *sles) 5279b94acceSBarry Smith { 5283a40ed3dSBarry Smith PetscFunctionBegin; 52977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 5309b94acceSBarry Smith *sles = snes->sles; 5313a40ed3dSBarry Smith PetscFunctionReturn(0); 5329b94acceSBarry Smith } 53382bf6240SBarry Smith 534*e24b481bSBarry Smith #undef __FUNC__ 535*e24b481bSBarry Smith #define __FUNC__ "SNESPublish_Petsc" 536*e24b481bSBarry Smith static int SNESPublish_Petsc(PetscObject object) 537*e24b481bSBarry Smith { 538*e24b481bSBarry Smith #if defined(HAVE_AMS) 539*e24b481bSBarry Smith SNES v = (SNES) object; 540*e24b481bSBarry Smith int ierr; 541*e24b481bSBarry Smith 542*e24b481bSBarry Smith PetscFunctionBegin; 543*e24b481bSBarry Smith 544*e24b481bSBarry Smith /* if it is already published then return */ 545*e24b481bSBarry Smith if (v->amem >=0 ) PetscFunctionReturn(0); 546*e24b481bSBarry Smith 547*e24b481bSBarry Smith ierr = PetscObjectPublishBaseBegin(object,"SNES");CHKERRQ(ierr); 548*e24b481bSBarry Smith ierr = AMS_Memory_add_field((AMS_Memory)v->amem,"Iteration",&v->iter,1,AMS_INT,AMS_READ, 549*e24b481bSBarry Smith AMS_COMMON,AMS_REDUCT_UNDEF);CHKERRQ(ierr); 550*e24b481bSBarry Smith ierr = AMS_Memory_add_field((AMS_Memory)v->amem,"Residual",&v->norm,1,AMS_DOUBLE,AMS_READ, 551*e24b481bSBarry Smith AMS_COMMON,AMS_REDUCT_UNDEF);CHKERRQ(ierr); 552*e24b481bSBarry Smith ierr = PetscObjectPublishBaseEnd(object);CHKERRQ(ierr); 553*e24b481bSBarry Smith 554*e24b481bSBarry Smith /* publish children of object */ 555*e24b481bSBarry Smith ierr = PetscObjectPublish((PetscObject)v->sles);CHKERRQ(ierr); 556*e24b481bSBarry Smith 557*e24b481bSBarry Smith #else 558*e24b481bSBarry Smith PetscFunctionBegin; 559*e24b481bSBarry Smith #endif 560*e24b481bSBarry Smith PetscFunctionReturn(0); 561*e24b481bSBarry Smith } 562*e24b481bSBarry Smith 5639b94acceSBarry Smith /* -----------------------------------------------------------*/ 5645615d1e5SSatish Balay #undef __FUNC__ 5655615d1e5SSatish Balay #define __FUNC__ "SNESCreate" 5669b94acceSBarry Smith /*@C 5679b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 5689b94acceSBarry Smith 569c7afd0dbSLois Curfman McInnes Collective on MPI_Comm 570c7afd0dbSLois Curfman McInnes 571c7afd0dbSLois Curfman McInnes Input Parameters: 572c7afd0dbSLois Curfman McInnes + comm - MPI communicator 573c7afd0dbSLois Curfman McInnes - type - type of method, either 574c7afd0dbSLois Curfman McInnes SNES_NONLINEAR_EQUATIONS (for systems of nonlinear equations) 575c7afd0dbSLois Curfman McInnes or SNES_UNCONSTRAINED_MINIMIZATION (for unconstrained minimization) 5769b94acceSBarry Smith 5779b94acceSBarry Smith Output Parameter: 5789b94acceSBarry Smith . outsnes - the new SNES context 5799b94acceSBarry Smith 580c7afd0dbSLois Curfman McInnes Options Database Keys: 581c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 582c7afd0dbSLois Curfman McInnes and no preconditioning matrix 583c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 584c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 585c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 586c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 587c1f60f51SBarry Smith 5889b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 5899b94acceSBarry Smith 59063a78c88SLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy() 5919b94acceSBarry Smith @*/ 5924b0e389bSBarry Smith int SNESCreate(MPI_Comm comm,SNESProblemType type,SNES *outsnes) 5939b94acceSBarry Smith { 5949b94acceSBarry Smith int ierr; 5959b94acceSBarry Smith SNES snes; 5969b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 59737fcc0dbSBarry Smith 5983a40ed3dSBarry Smith PetscFunctionBegin; 5999b94acceSBarry Smith *outsnes = 0; 600d64ed03dSBarry Smith if (type != SNES_UNCONSTRAINED_MINIMIZATION && type != SNES_NONLINEAR_EQUATIONS){ 601d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"incorrect method type"); 602d64ed03dSBarry Smith } 603f830108cSBarry Smith PetscHeaderCreate(snes,_p_SNES,int,SNES_COOKIE,0,comm,SNESDestroy,SNESView); 6049b94acceSBarry Smith PLogObjectCreate(snes); 605*e24b481bSBarry Smith snes->bops->publish = SNESPublish_Petsc; 6069b94acceSBarry Smith snes->max_its = 50; 6079750a799SBarry Smith snes->max_funcs = 10000; 6089b94acceSBarry Smith snes->norm = 0.0; 6095a2d0531SBarry Smith if (type == SNES_UNCONSTRAINED_MINIMIZATION) { 610b18e04deSLois Curfman McInnes snes->rtol = 1.e-8; 611b18e04deSLois Curfman McInnes snes->ttol = 0.0; 6129b94acceSBarry Smith snes->atol = 1.e-10; 613d64ed03dSBarry Smith } else { 614b4874afaSBarry Smith snes->rtol = 1.e-8; 615b4874afaSBarry Smith snes->ttol = 0.0; 616b4874afaSBarry Smith snes->atol = 1.e-50; 617b4874afaSBarry Smith } 6189b94acceSBarry Smith snes->xtol = 1.e-8; 619b18e04deSLois Curfman McInnes snes->trunctol = 1.e-12; /* no longer used */ 6209b94acceSBarry Smith snes->nfuncs = 0; 6219b94acceSBarry Smith snes->nfailures = 0; 6227a00f4a9SLois Curfman McInnes snes->linear_its = 0; 623639f9d9dSBarry Smith snes->numbermonitors = 0; 6249b94acceSBarry Smith snes->data = 0; 6259b94acceSBarry Smith snes->view = 0; 6269b94acceSBarry Smith snes->computeumfunction = 0; 6279b94acceSBarry Smith snes->umfunP = 0; 6289b94acceSBarry Smith snes->fc = 0; 6299b94acceSBarry Smith snes->deltatol = 1.e-12; 6309b94acceSBarry Smith snes->fmin = -1.e30; 6319b94acceSBarry Smith snes->method_class = type; 6329b94acceSBarry Smith snes->set_method_called = 0; 63382bf6240SBarry Smith snes->setupcalled = 0; 6349b94acceSBarry Smith snes->ksp_ewconv = 0; 6356f24a144SLois Curfman McInnes snes->xmonitor = 0; 6366f24a144SLois Curfman McInnes snes->vwork = 0; 6376f24a144SLois Curfman McInnes snes->nwork = 0; 638c9005455SLois Curfman McInnes snes->conv_hist_size = 0; 639c9005455SLois Curfman McInnes snes->conv_act_size = 0; 640c9005455SLois Curfman McInnes snes->conv_hist = 0; 6419b94acceSBarry Smith 6429b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 6430452661fSBarry Smith kctx = PetscNew(SNES_KSP_EW_ConvCtx); CHKPTRQ(kctx); 644eed86810SBarry Smith PLogObjectMemory(snes,sizeof(SNES_KSP_EW_ConvCtx)); 6459b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 6469b94acceSBarry Smith kctx->version = 2; 6479b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 6489b94acceSBarry Smith this was too large for some test cases */ 6499b94acceSBarry Smith kctx->rtol_last = 0; 6509b94acceSBarry Smith kctx->rtol_max = .9; 6519b94acceSBarry Smith kctx->gamma = 1.0; 6529b94acceSBarry Smith kctx->alpha2 = .5*(1.0 + sqrt(5.0)); 6539b94acceSBarry Smith kctx->alpha = kctx->alpha2; 6549b94acceSBarry Smith kctx->threshold = .1; 6559b94acceSBarry Smith kctx->lresid_last = 0; 6569b94acceSBarry Smith kctx->norm_last = 0; 6579b94acceSBarry Smith 6589b94acceSBarry Smith ierr = SLESCreate(comm,&snes->sles); CHKERRQ(ierr); 6599b94acceSBarry Smith PLogObjectParent(snes,snes->sles) 6605334005bSBarry Smith 6619b94acceSBarry Smith *outsnes = snes; 662*e24b481bSBarry Smith PetscPublishAll(snes); 6633a40ed3dSBarry Smith PetscFunctionReturn(0); 6649b94acceSBarry Smith } 6659b94acceSBarry Smith 6669b94acceSBarry Smith /* --------------------------------------------------------------- */ 6675615d1e5SSatish Balay #undef __FUNC__ 668d4bb536fSBarry Smith #define __FUNC__ "SNESSetFunction" 6699b94acceSBarry Smith /*@C 6709b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 6719b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 6729b94acceSBarry Smith equations. 6739b94acceSBarry Smith 674fee21e36SBarry Smith Collective on SNES 675fee21e36SBarry Smith 676c7afd0dbSLois Curfman McInnes Input Parameters: 677c7afd0dbSLois Curfman McInnes + snes - the SNES context 678c7afd0dbSLois Curfman McInnes . func - function evaluation routine 679c7afd0dbSLois Curfman McInnes . r - vector to store function value 680c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 681c7afd0dbSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 6829b94acceSBarry Smith 683c7afd0dbSLois Curfman McInnes Calling sequence of func: 6848d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Vec f,void *ctx); 685c7afd0dbSLois Curfman McInnes 686313e4042SLois Curfman McInnes . f - function vector 687c7afd0dbSLois Curfman McInnes - ctx - optional user-defined function context 6889b94acceSBarry Smith 6899b94acceSBarry Smith Notes: 6909b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 6919b94acceSBarry Smith $ f'(x) x = -f(x), 692c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 6939b94acceSBarry Smith 6949b94acceSBarry Smith SNESSetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 6959b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 6969b94acceSBarry Smith SNESSetMinimizationFunction() and SNESSetGradient(); 6979b94acceSBarry Smith 6989b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 6999b94acceSBarry Smith 700a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 7019b94acceSBarry Smith @*/ 7025334005bSBarry Smith int SNESSetFunction( SNES snes, Vec r, int (*func)(SNES,Vec,Vec,void*),void *ctx) 7039b94acceSBarry Smith { 7043a40ed3dSBarry Smith PetscFunctionBegin; 70577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 706a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 707a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 708a8c6a408SBarry Smith } 7099b94acceSBarry Smith snes->computefunction = func; 7109b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 7119b94acceSBarry Smith snes->funP = ctx; 7123a40ed3dSBarry Smith PetscFunctionReturn(0); 7139b94acceSBarry Smith } 7149b94acceSBarry Smith 7155615d1e5SSatish Balay #undef __FUNC__ 7165615d1e5SSatish Balay #define __FUNC__ "SNESComputeFunction" 7179b94acceSBarry Smith /*@ 7189b94acceSBarry Smith SNESComputeFunction - Computes the function that has been set with 7199b94acceSBarry Smith SNESSetFunction(). 7209b94acceSBarry Smith 721c7afd0dbSLois Curfman McInnes Collective on SNES 722c7afd0dbSLois Curfman McInnes 7239b94acceSBarry Smith Input Parameters: 724c7afd0dbSLois Curfman McInnes + snes - the SNES context 725c7afd0dbSLois Curfman McInnes - x - input vector 7269b94acceSBarry Smith 7279b94acceSBarry Smith Output Parameter: 7283638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 7299b94acceSBarry Smith 7301bffabb2SLois Curfman McInnes Notes: 7319b94acceSBarry Smith SNESComputeFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 7329b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 7339b94acceSBarry Smith SNESComputeMinimizationFunction() and SNESComputeGradient(); 7349b94acceSBarry Smith 7359b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 7369b94acceSBarry Smith 737a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 7389b94acceSBarry Smith @*/ 7399b94acceSBarry Smith int SNESComputeFunction(SNES snes,Vec x, Vec y) 7409b94acceSBarry Smith { 7419b94acceSBarry Smith int ierr; 7429b94acceSBarry Smith 7433a40ed3dSBarry Smith PetscFunctionBegin; 744d4bb536fSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 745a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 746d4bb536fSBarry Smith } 7479b94acceSBarry Smith PLogEventBegin(SNES_FunctionEval,snes,x,y,0); 748d64ed03dSBarry Smith PetscStackPush("SNES user function"); 7499b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 750d64ed03dSBarry Smith PetscStackPop; 751ae3c334cSLois Curfman McInnes snes->nfuncs++; 7529b94acceSBarry Smith PLogEventEnd(SNES_FunctionEval,snes,x,y,0); 7533a40ed3dSBarry Smith PetscFunctionReturn(0); 7549b94acceSBarry Smith } 7559b94acceSBarry Smith 7565615d1e5SSatish Balay #undef __FUNC__ 757d4bb536fSBarry Smith #define __FUNC__ "SNESSetMinimizationFunction" 7589b94acceSBarry Smith /*@C 7599b94acceSBarry Smith SNESSetMinimizationFunction - Sets the function evaluation routine for 7609b94acceSBarry Smith unconstrained minimization. 7619b94acceSBarry Smith 762fee21e36SBarry Smith Collective on SNES 763fee21e36SBarry Smith 764c7afd0dbSLois Curfman McInnes Input Parameters: 765c7afd0dbSLois Curfman McInnes + snes - the SNES context 766c7afd0dbSLois Curfman McInnes . func - function evaluation routine 767c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 768c7afd0dbSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 7699b94acceSBarry Smith 770c7afd0dbSLois Curfman McInnes Calling sequence of func: 7718d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,double *f,void *ctx); 772c7afd0dbSLois Curfman McInnes 773c7afd0dbSLois Curfman McInnes + x - input vector 7749b94acceSBarry Smith . f - function 775c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined function context 7769b94acceSBarry Smith 7779b94acceSBarry Smith Notes: 7789b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 7799b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 7809b94acceSBarry Smith SNESSetFunction(). 7819b94acceSBarry Smith 7829b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimization, function 7839b94acceSBarry Smith 784a86d99e1SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESComputeMinimizationFunction(), 785a86d99e1SLois Curfman McInnes SNESSetHessian(), SNESSetGradient() 7869b94acceSBarry Smith @*/ 7879b94acceSBarry Smith int SNESSetMinimizationFunction(SNES snes,int (*func)(SNES,Vec,double*,void*), 7889b94acceSBarry Smith void *ctx) 7899b94acceSBarry Smith { 7903a40ed3dSBarry Smith PetscFunctionBegin; 79177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 792a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 793a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 794a8c6a408SBarry Smith } 7959b94acceSBarry Smith snes->computeumfunction = func; 7969b94acceSBarry Smith snes->umfunP = ctx; 7973a40ed3dSBarry Smith PetscFunctionReturn(0); 7989b94acceSBarry Smith } 7999b94acceSBarry Smith 8005615d1e5SSatish Balay #undef __FUNC__ 8015615d1e5SSatish Balay #define __FUNC__ "SNESComputeMinimizationFunction" 8029b94acceSBarry Smith /*@ 8039b94acceSBarry Smith SNESComputeMinimizationFunction - Computes the function that has been 8049b94acceSBarry Smith set with SNESSetMinimizationFunction(). 8059b94acceSBarry Smith 806c7afd0dbSLois Curfman McInnes Collective on SNES 807c7afd0dbSLois Curfman McInnes 8089b94acceSBarry Smith Input Parameters: 809c7afd0dbSLois Curfman McInnes + snes - the SNES context 810c7afd0dbSLois Curfman McInnes - x - input vector 8119b94acceSBarry Smith 8129b94acceSBarry Smith Output Parameter: 8139b94acceSBarry Smith . y - function value 8149b94acceSBarry Smith 8159b94acceSBarry Smith Notes: 8169b94acceSBarry Smith SNESComputeMinimizationFunction() is valid only for 8179b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 8189b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 819a86d99e1SLois Curfman McInnes 820a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, minimization, function 821a86d99e1SLois Curfman McInnes 822a86d99e1SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESGetMinimizationFunction(), 823a86d99e1SLois Curfman McInnes SNESComputeGradient(), SNESComputeHessian() 8249b94acceSBarry Smith @*/ 8259b94acceSBarry Smith int SNESComputeMinimizationFunction(SNES snes,Vec x,double *y) 8269b94acceSBarry Smith { 8279b94acceSBarry Smith int ierr; 8283a40ed3dSBarry Smith 8293a40ed3dSBarry Smith PetscFunctionBegin; 830a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 831a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 832a8c6a408SBarry Smith } 8339b94acceSBarry Smith PLogEventBegin(SNES_MinimizationFunctionEval,snes,x,y,0); 834d64ed03dSBarry Smith PetscStackPush("SNES user minimzation function"); 8359b94acceSBarry Smith ierr = (*snes->computeumfunction)(snes,x,y,snes->umfunP); CHKERRQ(ierr); 836d64ed03dSBarry Smith PetscStackPop; 837ae3c334cSLois Curfman McInnes snes->nfuncs++; 8389b94acceSBarry Smith PLogEventEnd(SNES_MinimizationFunctionEval,snes,x,y,0); 8393a40ed3dSBarry Smith PetscFunctionReturn(0); 8409b94acceSBarry Smith } 8419b94acceSBarry Smith 8425615d1e5SSatish Balay #undef __FUNC__ 843d4bb536fSBarry Smith #define __FUNC__ "SNESSetGradient" 8449b94acceSBarry Smith /*@C 8459b94acceSBarry Smith SNESSetGradient - Sets the gradient evaluation routine and gradient 8469b94acceSBarry Smith vector for use by the SNES routines. 8479b94acceSBarry Smith 848c7afd0dbSLois Curfman McInnes Collective on SNES 849c7afd0dbSLois Curfman McInnes 8509b94acceSBarry Smith Input Parameters: 851c7afd0dbSLois Curfman McInnes + snes - the SNES context 8529b94acceSBarry Smith . func - function evaluation routine 853044dda88SLois Curfman McInnes . ctx - optional user-defined context for private data for the 854044dda88SLois Curfman McInnes gradient evaluation routine (may be PETSC_NULL) 855c7afd0dbSLois Curfman McInnes - r - vector to store gradient value 856fee21e36SBarry Smith 8579b94acceSBarry Smith Calling sequence of func: 8588d76a1e5SLois Curfman McInnes $ func (SNES, Vec x, Vec g, void *ctx); 8599b94acceSBarry Smith 860c7afd0dbSLois Curfman McInnes + x - input vector 8619b94acceSBarry Smith . g - gradient vector 862c7afd0dbSLois Curfman McInnes - ctx - optional user-defined gradient context 8639b94acceSBarry Smith 8649b94acceSBarry Smith Notes: 8659b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 8669b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 8679b94acceSBarry Smith SNESSetFunction(). 8689b94acceSBarry Smith 8699b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 8709b94acceSBarry Smith 871a86d99e1SLois Curfman McInnes .seealso: SNESGetGradient(), SNESComputeGradient(), SNESSetHessian(), 872a86d99e1SLois Curfman McInnes SNESSetMinimizationFunction(), 8739b94acceSBarry Smith @*/ 87474679c65SBarry Smith int SNESSetGradient(SNES snes,Vec r,int (*func)(SNES,Vec,Vec,void*),void *ctx) 8759b94acceSBarry Smith { 8763a40ed3dSBarry Smith PetscFunctionBegin; 87777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 878a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 879a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 880a8c6a408SBarry Smith } 8819b94acceSBarry Smith snes->computefunction = func; 8829b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 8839b94acceSBarry Smith snes->funP = ctx; 8843a40ed3dSBarry Smith PetscFunctionReturn(0); 8859b94acceSBarry Smith } 8869b94acceSBarry Smith 8875615d1e5SSatish Balay #undef __FUNC__ 8885615d1e5SSatish Balay #define __FUNC__ "SNESComputeGradient" 8899b94acceSBarry Smith /*@ 890a86d99e1SLois Curfman McInnes SNESComputeGradient - Computes the gradient that has been set with 891a86d99e1SLois Curfman McInnes SNESSetGradient(). 8929b94acceSBarry Smith 893c7afd0dbSLois Curfman McInnes Collective on SNES 894c7afd0dbSLois Curfman McInnes 8959b94acceSBarry Smith Input Parameters: 896c7afd0dbSLois Curfman McInnes + snes - the SNES context 897c7afd0dbSLois Curfman McInnes - x - input vector 8989b94acceSBarry Smith 8999b94acceSBarry Smith Output Parameter: 9009b94acceSBarry Smith . y - gradient vector 9019b94acceSBarry Smith 9029b94acceSBarry Smith Notes: 9039b94acceSBarry Smith SNESComputeGradient() is valid only for 9049b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 9059b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 906a86d99e1SLois Curfman McInnes 907a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, gradient 908a86d99e1SLois Curfman McInnes 909a86d99e1SLois Curfman McInnes .seealso: SNESSetGradient(), SNESGetGradient(), 910a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction(), SNESComputeHessian() 9119b94acceSBarry Smith @*/ 9129b94acceSBarry Smith int SNESComputeGradient(SNES snes,Vec x, Vec y) 9139b94acceSBarry Smith { 9149b94acceSBarry Smith int ierr; 9153a40ed3dSBarry Smith 9163a40ed3dSBarry Smith PetscFunctionBegin; 9173a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 918a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 9193a40ed3dSBarry Smith } 9209b94acceSBarry Smith PLogEventBegin(SNES_GradientEval,snes,x,y,0); 921d64ed03dSBarry Smith PetscStackPush("SNES user gradient function"); 9229b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 923d64ed03dSBarry Smith PetscStackPop; 9249b94acceSBarry Smith PLogEventEnd(SNES_GradientEval,snes,x,y,0); 9253a40ed3dSBarry Smith PetscFunctionReturn(0); 9269b94acceSBarry Smith } 9279b94acceSBarry Smith 9285615d1e5SSatish Balay #undef __FUNC__ 9295615d1e5SSatish Balay #define __FUNC__ "SNESComputeJacobian" 93062fef451SLois Curfman McInnes /*@ 93162fef451SLois Curfman McInnes SNESComputeJacobian - Computes the Jacobian matrix that has been 93262fef451SLois Curfman McInnes set with SNESSetJacobian(). 93362fef451SLois Curfman McInnes 934c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 935c7afd0dbSLois Curfman McInnes 93662fef451SLois Curfman McInnes Input Parameters: 937c7afd0dbSLois Curfman McInnes + snes - the SNES context 938c7afd0dbSLois Curfman McInnes - x - input vector 93962fef451SLois Curfman McInnes 94062fef451SLois Curfman McInnes Output Parameters: 941c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 94262fef451SLois Curfman McInnes . B - optional preconditioning matrix 943c7afd0dbSLois Curfman McInnes - flag - flag indicating matrix structure 944fee21e36SBarry Smith 94562fef451SLois Curfman McInnes Notes: 94662fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 94762fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 94862fef451SLois Curfman McInnes 949dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 950dc5a77f8SLois Curfman McInnes flag parameter. 95162fef451SLois Curfman McInnes 95262fef451SLois Curfman McInnes SNESComputeJacobian() is valid only for SNES_NONLINEAR_EQUATIONS 95362fef451SLois Curfman McInnes methods. An analogous routine for SNES_UNCONSTRAINED_MINIMIZATION 954005c665bSBarry Smith methods is SNESComputeHessian(). 95562fef451SLois Curfman McInnes 95662fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 95762fef451SLois Curfman McInnes 95862fef451SLois Curfman McInnes .seealso: SNESSetJacobian(), SLESSetOperators() 95962fef451SLois Curfman McInnes @*/ 9609b94acceSBarry Smith int SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 9619b94acceSBarry Smith { 9629b94acceSBarry Smith int ierr; 9633a40ed3dSBarry Smith 9643a40ed3dSBarry Smith PetscFunctionBegin; 9653a40ed3dSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 966a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 9673a40ed3dSBarry Smith } 9683a40ed3dSBarry Smith if (!snes->computejacobian) PetscFunctionReturn(0); 9699b94acceSBarry Smith PLogEventBegin(SNES_JacobianEval,snes,X,*A,*B); 970c4fc05e7SBarry Smith *flg = DIFFERENT_NONZERO_PATTERN; 971d64ed03dSBarry Smith PetscStackPush("SNES user Jacobian function"); 9729b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 973d64ed03dSBarry Smith PetscStackPop; 9749b94acceSBarry Smith PLogEventEnd(SNES_JacobianEval,snes,X,*A,*B); 9756d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 97677c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 97777c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 9783a40ed3dSBarry Smith PetscFunctionReturn(0); 9799b94acceSBarry Smith } 9809b94acceSBarry Smith 9815615d1e5SSatish Balay #undef __FUNC__ 9825615d1e5SSatish Balay #define __FUNC__ "SNESComputeHessian" 98362fef451SLois Curfman McInnes /*@ 98462fef451SLois Curfman McInnes SNESComputeHessian - Computes the Hessian matrix that has been 98562fef451SLois Curfman McInnes set with SNESSetHessian(). 98662fef451SLois Curfman McInnes 987c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 988c7afd0dbSLois Curfman McInnes 98962fef451SLois Curfman McInnes Input Parameters: 990c7afd0dbSLois Curfman McInnes + snes - the SNES context 991c7afd0dbSLois Curfman McInnes - x - input vector 99262fef451SLois Curfman McInnes 99362fef451SLois Curfman McInnes Output Parameters: 994c7afd0dbSLois Curfman McInnes + A - Hessian matrix 99562fef451SLois Curfman McInnes . B - optional preconditioning matrix 996c7afd0dbSLois Curfman McInnes - flag - flag indicating matrix structure 997fee21e36SBarry Smith 99862fef451SLois Curfman McInnes Notes: 99962fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 100062fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 100162fef451SLois Curfman McInnes 1002dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 1003dc5a77f8SLois Curfman McInnes flag parameter. 100462fef451SLois Curfman McInnes 100562fef451SLois Curfman McInnes SNESComputeHessian() is valid only for 100662fef451SLois Curfman McInnes SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 100762fef451SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods is SNESComputeJacobian(). 100862fef451SLois Curfman McInnes 100962fef451SLois Curfman McInnes .keywords: SNES, compute, Hessian, matrix 101062fef451SLois Curfman McInnes 1011a86d99e1SLois Curfman McInnes .seealso: SNESSetHessian(), SLESSetOperators(), SNESComputeGradient(), 1012a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction() 101362fef451SLois Curfman McInnes @*/ 101462fef451SLois Curfman McInnes int SNESComputeHessian(SNES snes,Vec x,Mat *A,Mat *B,MatStructure *flag) 10159b94acceSBarry Smith { 10169b94acceSBarry Smith int ierr; 10173a40ed3dSBarry Smith 10183a40ed3dSBarry Smith PetscFunctionBegin; 10193a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1020a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 10213a40ed3dSBarry Smith } 10223a40ed3dSBarry Smith if (!snes->computejacobian) PetscFunctionReturn(0); 102362fef451SLois Curfman McInnes PLogEventBegin(SNES_HessianEval,snes,x,*A,*B); 10240f4a323eSLois Curfman McInnes *flag = DIFFERENT_NONZERO_PATTERN; 1025d64ed03dSBarry Smith PetscStackPush("SNES user Hessian function"); 102662fef451SLois Curfman McInnes ierr = (*snes->computejacobian)(snes,x,A,B,flag,snes->jacP); CHKERRQ(ierr); 1027d64ed03dSBarry Smith PetscStackPop; 102862fef451SLois Curfman McInnes PLogEventEnd(SNES_HessianEval,snes,x,*A,*B); 10290f4a323eSLois Curfman McInnes /* make sure user returned a correct Jacobian and preconditioner */ 103077c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 103177c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 10323a40ed3dSBarry Smith PetscFunctionReturn(0); 10339b94acceSBarry Smith } 10349b94acceSBarry Smith 10355615d1e5SSatish Balay #undef __FUNC__ 1036d4bb536fSBarry Smith #define __FUNC__ "SNESSetJacobian" 10379b94acceSBarry Smith /*@C 10389b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 1039044dda88SLois Curfman McInnes location to store the matrix. 10409b94acceSBarry Smith 1041c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1042c7afd0dbSLois Curfman McInnes 10439b94acceSBarry Smith Input Parameters: 1044c7afd0dbSLois Curfman McInnes + snes - the SNES context 10459b94acceSBarry Smith . A - Jacobian matrix 10469b94acceSBarry Smith . B - preconditioner matrix (usually same as the Jacobian) 10479b94acceSBarry Smith . func - Jacobian evaluation routine 1048c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 10492cd2dfdcSLois Curfman McInnes Jacobian evaluation routine (may be PETSC_NULL) 10509b94acceSBarry Smith 10519b94acceSBarry Smith Calling sequence of func: 10528d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 10539b94acceSBarry Smith 1054c7afd0dbSLois Curfman McInnes + x - input vector 10559b94acceSBarry Smith . A - Jacobian matrix 10569b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1057ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 1058ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 1059c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined Jacobian context 10609b94acceSBarry Smith 10619b94acceSBarry Smith Notes: 1062dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 10632cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1064ac21db08SLois Curfman McInnes 1065ac21db08SLois Curfman McInnes The routine func() takes Mat * as the matrix arguments rather than Mat. 10669b94acceSBarry Smith This allows the Jacobian evaluation routine to replace A and/or B with a 10679b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 10689b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 10699b94acceSBarry Smith throughout the global iterations. 10709b94acceSBarry Smith 10719b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 10729b94acceSBarry Smith 1073ac21db08SLois Curfman McInnes .seealso: SLESSetOperators(), SNESSetFunction() 10749b94acceSBarry Smith @*/ 10759b94acceSBarry Smith int SNESSetJacobian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 10769b94acceSBarry Smith MatStructure*,void*),void *ctx) 10779b94acceSBarry Smith { 10783a40ed3dSBarry Smith PetscFunctionBegin; 107977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1080a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1081a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 1082a8c6a408SBarry Smith } 10839b94acceSBarry Smith snes->computejacobian = func; 10849b94acceSBarry Smith snes->jacP = ctx; 10859b94acceSBarry Smith snes->jacobian = A; 10869b94acceSBarry Smith snes->jacobian_pre = B; 10873a40ed3dSBarry Smith PetscFunctionReturn(0); 10889b94acceSBarry Smith } 108962fef451SLois Curfman McInnes 10905615d1e5SSatish Balay #undef __FUNC__ 1091d4bb536fSBarry Smith #define __FUNC__ "SNESGetJacobian" 1092b4fd4287SBarry Smith /*@ 1093b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 1094b4fd4287SBarry Smith provided context for evaluating the Jacobian. 1095b4fd4287SBarry Smith 1096c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 1097c7afd0dbSLois Curfman McInnes 1098b4fd4287SBarry Smith Input Parameter: 1099b4fd4287SBarry Smith . snes - the nonlinear solver context 1100b4fd4287SBarry Smith 1101b4fd4287SBarry Smith Output Parameters: 1102c7afd0dbSLois Curfman McInnes + A - location to stash Jacobian matrix (or PETSC_NULL) 1103b4fd4287SBarry Smith . B - location to stash preconditioner matrix (or PETSC_NULL) 1104c7afd0dbSLois Curfman McInnes - ctx - location to stash Jacobian ctx (or PETSC_NULL) 1105fee21e36SBarry Smith 1106b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian() 1107b4fd4287SBarry Smith @*/ 1108b4fd4287SBarry Smith int SNESGetJacobian(SNES snes,Mat *A,Mat *B, void **ctx) 1109b4fd4287SBarry Smith { 11103a40ed3dSBarry Smith PetscFunctionBegin; 11113a40ed3dSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1112a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 11133a40ed3dSBarry Smith } 1114b4fd4287SBarry Smith if (A) *A = snes->jacobian; 1115b4fd4287SBarry Smith if (B) *B = snes->jacobian_pre; 1116b4fd4287SBarry Smith if (ctx) *ctx = snes->jacP; 11173a40ed3dSBarry Smith PetscFunctionReturn(0); 1118b4fd4287SBarry Smith } 1119b4fd4287SBarry Smith 11205615d1e5SSatish Balay #undef __FUNC__ 1121d4bb536fSBarry Smith #define __FUNC__ "SNESSetHessian" 11229b94acceSBarry Smith /*@C 11239b94acceSBarry Smith SNESSetHessian - Sets the function to compute Hessian as well as the 1124044dda88SLois Curfman McInnes location to store the matrix. 11259b94acceSBarry Smith 1126c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1127c7afd0dbSLois Curfman McInnes 11289b94acceSBarry Smith Input Parameters: 1129c7afd0dbSLois Curfman McInnes + snes - the SNES context 11309b94acceSBarry Smith . A - Hessian matrix 11319b94acceSBarry Smith . B - preconditioner matrix (usually same as the Hessian) 11329b94acceSBarry Smith . func - Jacobian evaluation routine 1133c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 1134313e4042SLois Curfman McInnes Hessian evaluation routine (may be PETSC_NULL) 11359b94acceSBarry Smith 11369b94acceSBarry Smith Calling sequence of func: 11378d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 11389b94acceSBarry Smith 1139c7afd0dbSLois Curfman McInnes + x - input vector 11409b94acceSBarry Smith . A - Hessian matrix 11419b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1142ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 1143ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 1144c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined Hessian context 11459b94acceSBarry Smith 11469b94acceSBarry Smith Notes: 1147dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 11482cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1149ac21db08SLois Curfman McInnes 11509b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 11519b94acceSBarry Smith This allows the Hessian evaluation routine to replace A and/or B with a 11529b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 11539b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 11549b94acceSBarry Smith throughout the global iterations. 11559b94acceSBarry Smith 11569b94acceSBarry Smith .keywords: SNES, nonlinear, set, Hessian, matrix 11579b94acceSBarry Smith 1158ac21db08SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESSetGradient(), SLESSetOperators() 11599b94acceSBarry Smith @*/ 11609b94acceSBarry Smith int SNESSetHessian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 11619b94acceSBarry Smith MatStructure*,void*),void *ctx) 11629b94acceSBarry Smith { 11633a40ed3dSBarry Smith PetscFunctionBegin; 116477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1165d4bb536fSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1166a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 1167d4bb536fSBarry Smith } 11689b94acceSBarry Smith snes->computejacobian = func; 11699b94acceSBarry Smith snes->jacP = ctx; 11709b94acceSBarry Smith snes->jacobian = A; 11719b94acceSBarry Smith snes->jacobian_pre = B; 11723a40ed3dSBarry Smith PetscFunctionReturn(0); 11739b94acceSBarry Smith } 11749b94acceSBarry Smith 11755615d1e5SSatish Balay #undef __FUNC__ 1176d4bb536fSBarry Smith #define __FUNC__ "SNESGetHessian" 117762fef451SLois Curfman McInnes /*@ 117862fef451SLois Curfman McInnes SNESGetHessian - Returns the Hessian matrix and optionally the user 117962fef451SLois Curfman McInnes provided context for evaluating the Hessian. 118062fef451SLois Curfman McInnes 1181c7afd0dbSLois Curfman McInnes Not Collective, but Mat object is parallel if SNES object is parallel 1182c7afd0dbSLois Curfman McInnes 118362fef451SLois Curfman McInnes Input Parameter: 118462fef451SLois Curfman McInnes . snes - the nonlinear solver context 118562fef451SLois Curfman McInnes 118662fef451SLois Curfman McInnes Output Parameters: 1187c7afd0dbSLois Curfman McInnes + A - location to stash Hessian matrix (or PETSC_NULL) 118862fef451SLois Curfman McInnes . B - location to stash preconditioner matrix (or PETSC_NULL) 1189c7afd0dbSLois Curfman McInnes - ctx - location to stash Hessian ctx (or PETSC_NULL) 1190fee21e36SBarry Smith 119162fef451SLois Curfman McInnes .seealso: SNESSetHessian(), SNESComputeHessian() 1192c7afd0dbSLois Curfman McInnes 1193c7afd0dbSLois Curfman McInnes .keywords: SNES, get, Hessian 119462fef451SLois Curfman McInnes @*/ 119562fef451SLois Curfman McInnes int SNESGetHessian(SNES snes,Mat *A,Mat *B, void **ctx) 119662fef451SLois Curfman McInnes { 11973a40ed3dSBarry Smith PetscFunctionBegin; 11983a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION){ 1199a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 12003a40ed3dSBarry Smith } 120162fef451SLois Curfman McInnes if (A) *A = snes->jacobian; 120262fef451SLois Curfman McInnes if (B) *B = snes->jacobian_pre; 120362fef451SLois Curfman McInnes if (ctx) *ctx = snes->jacP; 12043a40ed3dSBarry Smith PetscFunctionReturn(0); 120562fef451SLois Curfman McInnes } 120662fef451SLois Curfman McInnes 12079b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */ 12089b94acceSBarry Smith 12095615d1e5SSatish Balay #undef __FUNC__ 12105615d1e5SSatish Balay #define __FUNC__ "SNESSetUp" 12119b94acceSBarry Smith /*@ 12129b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 1213272ac6f2SLois Curfman McInnes of a nonlinear solver. 12149b94acceSBarry Smith 1215fee21e36SBarry Smith Collective on SNES 1216fee21e36SBarry Smith 1217c7afd0dbSLois Curfman McInnes Input Parameters: 1218c7afd0dbSLois Curfman McInnes + snes - the SNES context 1219c7afd0dbSLois Curfman McInnes - x - the solution vector 1220c7afd0dbSLois Curfman McInnes 1221272ac6f2SLois Curfman McInnes Notes: 1222272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 1223272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 1224272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 1225272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 1226272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 1227272ac6f2SLois Curfman McInnes 12289b94acceSBarry Smith .keywords: SNES, nonlinear, setup 12299b94acceSBarry Smith 12309b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 12319b94acceSBarry Smith @*/ 12328ddd3da0SLois Curfman McInnes int SNESSetUp(SNES snes,Vec x) 12339b94acceSBarry Smith { 1234272ac6f2SLois Curfman McInnes int ierr, flg; 12353a40ed3dSBarry Smith 12363a40ed3dSBarry Smith PetscFunctionBegin; 123777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 123877c4ece6SBarry Smith PetscValidHeaderSpecific(x,VEC_COOKIE); 12398ddd3da0SLois Curfman McInnes snes->vec_sol = snes->vec_sol_always = x; 12409b94acceSBarry Smith 1241c1f60f51SBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_mf_operator", &flg); CHKERRQ(ierr); 1242c1f60f51SBarry Smith /* 1243c1f60f51SBarry Smith This version replaces the user provided Jacobian matrix with a 1244dfa02198SLois Curfman McInnes matrix-free version but still employs the user-provided preconditioner matrix 1245c1f60f51SBarry Smith */ 1246c1f60f51SBarry Smith if (flg) { 1247c1f60f51SBarry Smith Mat J; 1248*e24b481bSBarry Smith ierr = MatCreateSNESFDMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1249c1f60f51SBarry Smith PLogObjectParent(snes,J); 1250c1f60f51SBarry Smith snes->mfshell = J; 1251c1f60f51SBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1252c1f60f51SBarry Smith snes->jacobian = J; 1253c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Jacobian routines\n"); 1254d64ed03dSBarry Smith } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 1255c1f60f51SBarry Smith snes->jacobian = J; 1256c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Hessian routines\n"); 1257d4bb536fSBarry Smith } else { 1258a8c6a408SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"Method class doesn't support matrix-free operator option"); 1259d4bb536fSBarry Smith } 1260*e24b481bSBarry Smith ierr = MatSNESFDMFSetFromOptions(J);CHKERRQ(ierr); 1261c1f60f51SBarry Smith } 1262272ac6f2SLois Curfman McInnes ierr = OptionsHasName(snes->prefix,"-snes_mf", &flg); CHKERRQ(ierr); 1263c1f60f51SBarry Smith /* 1264dfa02198SLois Curfman McInnes This version replaces both the user-provided Jacobian and the user- 1265c1f60f51SBarry Smith provided preconditioner matrix with the default matrix free version. 1266c1f60f51SBarry Smith */ 126731615d04SLois Curfman McInnes if (flg) { 1268272ac6f2SLois Curfman McInnes Mat J; 1269*e24b481bSBarry Smith ierr = MatCreateSNESFDMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1270272ac6f2SLois Curfman McInnes PLogObjectParent(snes,J); 1271272ac6f2SLois Curfman McInnes snes->mfshell = J; 127283e56afdSLois Curfman McInnes if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 127383e56afdSLois Curfman McInnes ierr = SNESSetJacobian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 127494a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Jacobian routines\n"); 1275d64ed03dSBarry Smith } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 127683e56afdSLois Curfman McInnes ierr = SNESSetHessian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 127794a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Hessian routines\n"); 1278d4bb536fSBarry Smith } else { 1279a8c6a408SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"Method class doesn't support matrix-free option"); 1280d4bb536fSBarry Smith } 1281*e24b481bSBarry Smith ierr = MatSNESFDMFSetFromOptions(J);CHKERRQ(ierr); 1282272ac6f2SLois Curfman McInnes } 12839b94acceSBarry Smith if ((snes->method_class == SNES_NONLINEAR_EQUATIONS)) { 1284a8c6a408SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetFunction() first"); 1285a8c6a408SBarry Smith if (!snes->computefunction) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetFunction() first"); 1286a8c6a408SBarry Smith if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetJacobian() first"); 1287a8c6a408SBarry Smith if (snes->vec_func == snes->vec_sol) { 1288a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_IDN,0,"Solution vector cannot be function vector"); 1289a8c6a408SBarry Smith } 1290a703fe33SLois Curfman McInnes 1291a703fe33SLois Curfman McInnes /* Set the KSP stopping criterion to use the Eisenstat-Walker method */ 129282bf6240SBarry Smith if (snes->ksp_ewconv && PetscStrcmp(snes->type_name,SNES_EQ_TR)) { 1293a703fe33SLois Curfman McInnes SLES sles; KSP ksp; 1294a703fe33SLois Curfman McInnes ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 1295a703fe33SLois Curfman McInnes ierr = SLESGetKSP(sles,&ksp); CHKERRQ(ierr); 1296a703fe33SLois Curfman McInnes ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private, 1297a703fe33SLois Curfman McInnes (void *)snes); CHKERRQ(ierr); 1298a703fe33SLois Curfman McInnes } 12999b94acceSBarry Smith } else if ((snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION)) { 1300a8c6a408SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetGradient() first"); 1301a8c6a408SBarry Smith if (!snes->computefunction) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetGradient() first"); 1302a8c6a408SBarry Smith if (!snes->computeumfunction) { 1303a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetMinimizationFunction() first"); 1304a8c6a408SBarry Smith } 1305a8c6a408SBarry Smith if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetHessian() first"); 1306d4bb536fSBarry Smith } else { 1307a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Unknown method class"); 1308d4bb536fSBarry Smith } 1309a703fe33SLois Curfman McInnes if (snes->setup) {ierr = (*snes->setup)(snes); CHKERRQ(ierr);} 131082bf6240SBarry Smith snes->setupcalled = 1; 13113a40ed3dSBarry Smith PetscFunctionReturn(0); 13129b94acceSBarry Smith } 13139b94acceSBarry Smith 13145615d1e5SSatish Balay #undef __FUNC__ 1315d4bb536fSBarry Smith #define __FUNC__ "SNESDestroy" 13169b94acceSBarry Smith /*@C 13179b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 13189b94acceSBarry Smith with SNESCreate(). 13199b94acceSBarry Smith 1320c7afd0dbSLois Curfman McInnes Collective on SNES 1321c7afd0dbSLois Curfman McInnes 13229b94acceSBarry Smith Input Parameter: 13239b94acceSBarry Smith . snes - the SNES context 13249b94acceSBarry Smith 13259b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 13269b94acceSBarry Smith 132763a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 13289b94acceSBarry Smith @*/ 13299b94acceSBarry Smith int SNESDestroy(SNES snes) 13309b94acceSBarry Smith { 13319b94acceSBarry Smith int ierr; 13323a40ed3dSBarry Smith 13333a40ed3dSBarry Smith PetscFunctionBegin; 133477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 13353a40ed3dSBarry Smith if (--snes->refct > 0) PetscFunctionReturn(0); 1336d4bb536fSBarry Smith 1337e1311b90SBarry Smith if (snes->destroy) {ierr = (*(snes)->destroy)(snes); CHKERRQ(ierr);} 13380452661fSBarry Smith if (snes->kspconvctx) PetscFree(snes->kspconvctx); 1339522c5e43SBarry Smith if (snes->mfshell) {ierr = MatDestroy(snes->mfshell);CHKERRQ(ierr);} 13409b94acceSBarry Smith ierr = SLESDestroy(snes->sles); CHKERRQ(ierr); 1341522c5e43SBarry Smith if (snes->xmonitor) {ierr = SNESLGMonitorDestroy(snes->xmonitor);CHKERRQ(ierr);} 1342522c5e43SBarry Smith if (snes->vwork) {ierr = VecDestroyVecs(snes->vwork,snes->nvwork);CHKERRQ(ierr);} 13439b94acceSBarry Smith PLogObjectDestroy((PetscObject)snes); 13440452661fSBarry Smith PetscHeaderDestroy((PetscObject)snes); 13453a40ed3dSBarry Smith PetscFunctionReturn(0); 13469b94acceSBarry Smith } 13479b94acceSBarry Smith 13489b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 13499b94acceSBarry Smith 13505615d1e5SSatish Balay #undef __FUNC__ 13515615d1e5SSatish Balay #define __FUNC__ "SNESSetTolerances" 13529b94acceSBarry Smith /*@ 1353d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 13549b94acceSBarry Smith 1355c7afd0dbSLois Curfman McInnes Collective on SNES 1356c7afd0dbSLois Curfman McInnes 13579b94acceSBarry Smith Input Parameters: 1358c7afd0dbSLois Curfman McInnes + snes - the SNES context 135933174efeSLois Curfman McInnes . atol - absolute convergence tolerance 136033174efeSLois Curfman McInnes . rtol - relative convergence tolerance 136133174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 136233174efeSLois Curfman McInnes of the change in the solution between steps 136333174efeSLois Curfman McInnes . maxit - maximum number of iterations 1364c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1365fee21e36SBarry Smith 136633174efeSLois Curfman McInnes Options Database Keys: 1367c7afd0dbSLois Curfman McInnes + -snes_atol <atol> - Sets atol 1368c7afd0dbSLois Curfman McInnes . -snes_rtol <rtol> - Sets rtol 1369c7afd0dbSLois Curfman McInnes . -snes_stol <stol> - Sets stol 1370c7afd0dbSLois Curfman McInnes . -snes_max_it <maxit> - Sets maxit 1371c7afd0dbSLois Curfman McInnes - -snes_max_funcs <maxf> - Sets maxf 13729b94acceSBarry Smith 1373d7a720efSLois Curfman McInnes Notes: 13749b94acceSBarry Smith The default maximum number of iterations is 50. 13759b94acceSBarry Smith The default maximum number of function evaluations is 1000. 13769b94acceSBarry Smith 137733174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances 13789b94acceSBarry Smith 1379d7a720efSLois Curfman McInnes .seealso: SNESSetTrustRegionTolerance(), SNESSetMinimizationFunctionTolerance() 13809b94acceSBarry Smith @*/ 1381d7a720efSLois Curfman McInnes int SNESSetTolerances(SNES snes,double atol,double rtol,double stol,int maxit,int maxf) 13829b94acceSBarry Smith { 13833a40ed3dSBarry Smith PetscFunctionBegin; 138477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1385d7a720efSLois Curfman McInnes if (atol != PETSC_DEFAULT) snes->atol = atol; 1386d7a720efSLois Curfman McInnes if (rtol != PETSC_DEFAULT) snes->rtol = rtol; 1387d7a720efSLois Curfman McInnes if (stol != PETSC_DEFAULT) snes->xtol = stol; 1388d7a720efSLois Curfman McInnes if (maxit != PETSC_DEFAULT) snes->max_its = maxit; 1389d7a720efSLois Curfman McInnes if (maxf != PETSC_DEFAULT) snes->max_funcs = maxf; 13903a40ed3dSBarry Smith PetscFunctionReturn(0); 13919b94acceSBarry Smith } 13929b94acceSBarry Smith 13935615d1e5SSatish Balay #undef __FUNC__ 13945615d1e5SSatish Balay #define __FUNC__ "SNESGetTolerances" 13959b94acceSBarry Smith /*@ 139633174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 139733174efeSLois Curfman McInnes 1398c7afd0dbSLois Curfman McInnes Not Collective 1399c7afd0dbSLois Curfman McInnes 140033174efeSLois Curfman McInnes Input Parameters: 1401c7afd0dbSLois Curfman McInnes + snes - the SNES context 140233174efeSLois Curfman McInnes . atol - absolute convergence tolerance 140333174efeSLois Curfman McInnes . rtol - relative convergence tolerance 140433174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 140533174efeSLois Curfman McInnes of the change in the solution between steps 140633174efeSLois Curfman McInnes . maxit - maximum number of iterations 1407c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1408fee21e36SBarry Smith 140933174efeSLois Curfman McInnes Notes: 141033174efeSLois Curfman McInnes The user can specify PETSC_NULL for any parameter that is not needed. 141133174efeSLois Curfman McInnes 141233174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances 141333174efeSLois Curfman McInnes 141433174efeSLois Curfman McInnes .seealso: SNESSetTolerances() 141533174efeSLois Curfman McInnes @*/ 141633174efeSLois Curfman McInnes int SNESGetTolerances(SNES snes,double *atol,double *rtol,double *stol,int *maxit,int *maxf) 141733174efeSLois Curfman McInnes { 14183a40ed3dSBarry Smith PetscFunctionBegin; 141933174efeSLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 142033174efeSLois Curfman McInnes if (atol) *atol = snes->atol; 142133174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 142233174efeSLois Curfman McInnes if (stol) *stol = snes->xtol; 142333174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 142433174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 14253a40ed3dSBarry Smith PetscFunctionReturn(0); 142633174efeSLois Curfman McInnes } 142733174efeSLois Curfman McInnes 14285615d1e5SSatish Balay #undef __FUNC__ 14295615d1e5SSatish Balay #define __FUNC__ "SNESSetTrustRegionTolerance" 143033174efeSLois Curfman McInnes /*@ 14319b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 14329b94acceSBarry Smith 1433fee21e36SBarry Smith Collective on SNES 1434fee21e36SBarry Smith 1435c7afd0dbSLois Curfman McInnes Input Parameters: 1436c7afd0dbSLois Curfman McInnes + snes - the SNES context 1437c7afd0dbSLois Curfman McInnes - tol - tolerance 1438c7afd0dbSLois Curfman McInnes 14399b94acceSBarry Smith Options Database Key: 1440c7afd0dbSLois Curfman McInnes . -snes_trtol <tol> - Sets tol 14419b94acceSBarry Smith 14429b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 14439b94acceSBarry Smith 1444d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetMinimizationFunctionTolerance() 14459b94acceSBarry Smith @*/ 14469b94acceSBarry Smith int SNESSetTrustRegionTolerance(SNES snes,double tol) 14479b94acceSBarry Smith { 14483a40ed3dSBarry Smith PetscFunctionBegin; 144977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 14509b94acceSBarry Smith snes->deltatol = tol; 14513a40ed3dSBarry Smith PetscFunctionReturn(0); 14529b94acceSBarry Smith } 14539b94acceSBarry Smith 14545615d1e5SSatish Balay #undef __FUNC__ 14555615d1e5SSatish Balay #define __FUNC__ "SNESSetMinimizationFunctionTolerance" 14569b94acceSBarry Smith /*@ 145777c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance - Sets the minimum allowable function tolerance 14589b94acceSBarry Smith for unconstrained minimization solvers. 14599b94acceSBarry Smith 1460fee21e36SBarry Smith Collective on SNES 1461fee21e36SBarry Smith 1462c7afd0dbSLois Curfman McInnes Input Parameters: 1463c7afd0dbSLois Curfman McInnes + snes - the SNES context 1464c7afd0dbSLois Curfman McInnes - ftol - minimum function tolerance 1465c7afd0dbSLois Curfman McInnes 14669b94acceSBarry Smith Options Database Key: 1467c7afd0dbSLois Curfman McInnes . -snes_fmin <ftol> - Sets ftol 14689b94acceSBarry Smith 14699b94acceSBarry Smith Note: 147077c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance() is valid for SNES_UNCONSTRAINED_MINIMIZATION 14719b94acceSBarry Smith methods only. 14729b94acceSBarry Smith 14739b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimum, convergence, function, tolerance 14749b94acceSBarry Smith 1475d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetTrustRegionTolerance() 14769b94acceSBarry Smith @*/ 147777c4ece6SBarry Smith int SNESSetMinimizationFunctionTolerance(SNES snes,double ftol) 14789b94acceSBarry Smith { 14793a40ed3dSBarry Smith PetscFunctionBegin; 148077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 14819b94acceSBarry Smith snes->fmin = ftol; 14823a40ed3dSBarry Smith PetscFunctionReturn(0); 14839b94acceSBarry Smith } 14849b94acceSBarry Smith 14859b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 14869b94acceSBarry Smith 14875615d1e5SSatish Balay #undef __FUNC__ 1488d4bb536fSBarry Smith #define __FUNC__ "SNESSetMonitor" 14899b94acceSBarry Smith /*@C 14905cd90555SBarry Smith SNESSetMonitor - Sets an ADDITIONAL function that is to be used at every 14919b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 14929b94acceSBarry Smith progress. 14939b94acceSBarry Smith 1494fee21e36SBarry Smith Collective on SNES 1495fee21e36SBarry Smith 1496c7afd0dbSLois Curfman McInnes Input Parameters: 1497c7afd0dbSLois Curfman McInnes + snes - the SNES context 1498c7afd0dbSLois Curfman McInnes . func - monitoring routine 1499c7afd0dbSLois Curfman McInnes - mctx - [optional] user-defined context for private data for the 1500c7afd0dbSLois Curfman McInnes monitor routine (may be PETSC_NULL) 15019b94acceSBarry Smith 1502c7afd0dbSLois Curfman McInnes Calling sequence of func: 15038d76a1e5SLois Curfman McInnes $ int func(SNES snes,int its, Vec x,Vec f,double norm,void *mctx) 1504c7afd0dbSLois Curfman McInnes 1505c7afd0dbSLois Curfman McInnes + snes - the SNES context 1506c7afd0dbSLois Curfman McInnes . its - iteration number 1507c7afd0dbSLois Curfman McInnes . mctx - [optional] monitoring context 1508c7afd0dbSLois Curfman McInnes . norm - 2-norm function value (may be estimated) 1509c7afd0dbSLois Curfman McInnes for SNES_NONLINEAR_EQUATIONS methods 1510c7afd0dbSLois Curfman McInnes - norm - 2-norm gradient value (may be estimated) 1511c7afd0dbSLois Curfman McInnes for SNES_UNCONSTRAINED_MINIMIZATION methods 15129b94acceSBarry Smith 15139665c990SLois Curfman McInnes Options Database Keys: 1514c7afd0dbSLois Curfman McInnes + -snes_monitor - sets SNESDefaultMonitor() 1515c7afd0dbSLois Curfman McInnes . -snes_xmonitor - sets line graph monitor, 1516c7afd0dbSLois Curfman McInnes uses SNESLGMonitorCreate() 1517c7afd0dbSLois Curfman McInnes _ -snes_cancelmonitors - cancels all monitors that have 1518c7afd0dbSLois Curfman McInnes been hardwired into a code by 1519c7afd0dbSLois Curfman McInnes calls to SNESSetMonitor(), but 1520c7afd0dbSLois Curfman McInnes does not cancel those set via 1521c7afd0dbSLois Curfman McInnes the options database. 15229665c990SLois Curfman McInnes 1523639f9d9dSBarry Smith Notes: 15246bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 15256bc08f3fSLois Curfman McInnes SNESSetMonitor() multiple times; all will be called in the 15266bc08f3fSLois Curfman McInnes order in which they were set. 1527639f9d9dSBarry Smith 15289b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 15299b94acceSBarry Smith 15305cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESClearMonitor() 15319b94acceSBarry Smith @*/ 153274679c65SBarry Smith int SNESSetMonitor( SNES snes, int (*func)(SNES,int,double,void*),void *mctx ) 15339b94acceSBarry Smith { 15343a40ed3dSBarry Smith PetscFunctionBegin; 1535639f9d9dSBarry Smith if (snes->numbermonitors >= MAXSNESMONITORS) { 1536a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Too many monitors set"); 1537639f9d9dSBarry Smith } 1538639f9d9dSBarry Smith 1539639f9d9dSBarry Smith snes->monitor[snes->numbermonitors] = func; 1540639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 15413a40ed3dSBarry Smith PetscFunctionReturn(0); 15429b94acceSBarry Smith } 15439b94acceSBarry Smith 15445615d1e5SSatish Balay #undef __FUNC__ 15455cd90555SBarry Smith #define __FUNC__ "SNESClearMonitor" 15465cd90555SBarry Smith /*@C 15475cd90555SBarry Smith SNESClearMonitor - Clears all the monitor functions for a SNES object. 15485cd90555SBarry Smith 1549c7afd0dbSLois Curfman McInnes Collective on SNES 1550c7afd0dbSLois Curfman McInnes 15515cd90555SBarry Smith Input Parameters: 15525cd90555SBarry Smith . snes - the SNES context 15535cd90555SBarry Smith 15545cd90555SBarry Smith Options Database: 1555c7afd0dbSLois Curfman McInnes . -snes_cancelmonitors - cancels all monitors that have been hardwired 1556c7afd0dbSLois Curfman McInnes into a code by calls to SNESSetMonitor(), but does not cancel those 1557c7afd0dbSLois Curfman McInnes set via the options database 15585cd90555SBarry Smith 15595cd90555SBarry Smith Notes: 15605cd90555SBarry Smith There is no way to clear one specific monitor from a SNES object. 15615cd90555SBarry Smith 15625cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor 15635cd90555SBarry Smith 15645cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESSetMonitor() 15655cd90555SBarry Smith @*/ 15665cd90555SBarry Smith int SNESClearMonitor( SNES snes ) 15675cd90555SBarry Smith { 15685cd90555SBarry Smith PetscFunctionBegin; 15695cd90555SBarry Smith snes->numbermonitors = 0; 15705cd90555SBarry Smith PetscFunctionReturn(0); 15715cd90555SBarry Smith } 15725cd90555SBarry Smith 15735cd90555SBarry Smith #undef __FUNC__ 1574d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceTest" 15759b94acceSBarry Smith /*@C 15769b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 15779b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 15789b94acceSBarry Smith 1579fee21e36SBarry Smith Collective on SNES 1580fee21e36SBarry Smith 1581c7afd0dbSLois Curfman McInnes Input Parameters: 1582c7afd0dbSLois Curfman McInnes + snes - the SNES context 1583c7afd0dbSLois Curfman McInnes . func - routine to test for convergence 1584c7afd0dbSLois Curfman McInnes - cctx - [optional] context for private data for the convergence routine 1585c7afd0dbSLois Curfman McInnes (may be PETSC_NULL) 15869b94acceSBarry Smith 1587c7afd0dbSLois Curfman McInnes Calling sequence of func: 15888d76a1e5SLois Curfman McInnes $ int func (SNES snes,double xnorm,double gnorm,double f,void *cctx) 1589c7afd0dbSLois Curfman McInnes 1590c7afd0dbSLois Curfman McInnes + snes - the SNES context 1591c7afd0dbSLois Curfman McInnes . cctx - [optional] convergence context 1592c7afd0dbSLois Curfman McInnes . xnorm - 2-norm of current iterate 1593c7afd0dbSLois Curfman McInnes . gnorm - 2-norm of current step (SNES_NONLINEAR_EQUATIONS methods) 1594c7afd0dbSLois Curfman McInnes . f - 2-norm of function (SNES_NONLINEAR_EQUATIONS methods) 1595c7afd0dbSLois Curfman McInnes . gnorm - 2-norm of current gradient (SNES_UNCONSTRAINED_MINIMIZATION methods) 1596c7afd0dbSLois Curfman McInnes - f - function value (SNES_UNCONSTRAINED_MINIMIZATION methods) 15979b94acceSBarry Smith 15989b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 15999b94acceSBarry Smith 160040191667SLois Curfman McInnes .seealso: SNESConverged_EQ_LS(), SNESConverged_EQ_TR(), 160140191667SLois Curfman McInnes SNESConverged_UM_LS(), SNESConverged_UM_TR() 16029b94acceSBarry Smith @*/ 160374679c65SBarry Smith int SNESSetConvergenceTest(SNES snes,int (*func)(SNES,double,double,double,void*),void *cctx) 16049b94acceSBarry Smith { 16053a40ed3dSBarry Smith PetscFunctionBegin; 16069b94acceSBarry Smith (snes)->converged = func; 16079b94acceSBarry Smith (snes)->cnvP = cctx; 16083a40ed3dSBarry Smith PetscFunctionReturn(0); 16099b94acceSBarry Smith } 16109b94acceSBarry Smith 16115615d1e5SSatish Balay #undef __FUNC__ 1612d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceHistory" 1613c9005455SLois Curfman McInnes /*@ 1614c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 1615c9005455SLois Curfman McInnes 1616fee21e36SBarry Smith Collective on SNES 1617fee21e36SBarry Smith 1618c7afd0dbSLois Curfman McInnes Input Parameters: 1619c7afd0dbSLois Curfman McInnes + snes - iterative context obtained from SNESCreate() 1620c7afd0dbSLois Curfman McInnes . a - array to hold history 1621c7afd0dbSLois Curfman McInnes - na - size of a 1622c7afd0dbSLois Curfman McInnes 1623c9005455SLois Curfman McInnes Notes: 1624c9005455SLois Curfman McInnes If set, this array will contain the function norms (for 1625c9005455SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods) or gradient norms 1626c9005455SLois Curfman McInnes (for SNES_UNCONSTRAINED_MINIMIZATION methods) computed 1627c9005455SLois Curfman McInnes at each step. 1628c9005455SLois Curfman McInnes 1629c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 1630c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 1631c9005455SLois Curfman McInnes during the section of code that is being timed. 1632c9005455SLois Curfman McInnes 1633c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history 1634c9005455SLois Curfman McInnes @*/ 1635c9005455SLois Curfman McInnes int SNESSetConvergenceHistory(SNES snes, double *a, int na) 1636c9005455SLois Curfman McInnes { 16373a40ed3dSBarry Smith PetscFunctionBegin; 1638c9005455SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 1639c9005455SLois Curfman McInnes if (na) PetscValidScalarPointer(a); 1640c9005455SLois Curfman McInnes snes->conv_hist = a; 1641c9005455SLois Curfman McInnes snes->conv_hist_size = na; 16423a40ed3dSBarry Smith PetscFunctionReturn(0); 1643c9005455SLois Curfman McInnes } 1644c9005455SLois Curfman McInnes 1645c9005455SLois Curfman McInnes #undef __FUNC__ 16465615d1e5SSatish Balay #define __FUNC__ "SNESScaleStep_Private" 16479b94acceSBarry Smith /* 16489b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 16499b94acceSBarry Smith positive parameter delta. 16509b94acceSBarry Smith 16519b94acceSBarry Smith Input Parameters: 1652c7afd0dbSLois Curfman McInnes + snes - the SNES context 16539b94acceSBarry Smith . y - approximate solution of linear system 16549b94acceSBarry Smith . fnorm - 2-norm of current function 1655c7afd0dbSLois Curfman McInnes - delta - trust region size 16569b94acceSBarry Smith 16579b94acceSBarry Smith Output Parameters: 1658c7afd0dbSLois Curfman McInnes + gpnorm - predicted function norm at the new point, assuming local 16599b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 16609b94acceSBarry Smith region, and exceeds zero otherwise. 1661c7afd0dbSLois Curfman McInnes - ynorm - 2-norm of the step 16629b94acceSBarry Smith 16639b94acceSBarry Smith Note: 166440191667SLois Curfman McInnes For non-trust region methods such as SNES_EQ_LS, the parameter delta 16659b94acceSBarry Smith is set to be the maximum allowable step size. 16669b94acceSBarry Smith 16679b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 16689b94acceSBarry Smith */ 16699b94acceSBarry Smith int SNESScaleStep_Private(SNES snes,Vec y,double *fnorm,double *delta, 16709b94acceSBarry Smith double *gpnorm,double *ynorm) 16719b94acceSBarry Smith { 16729b94acceSBarry Smith double norm; 16739b94acceSBarry Smith Scalar cnorm; 16743a40ed3dSBarry Smith int ierr; 16753a40ed3dSBarry Smith 16763a40ed3dSBarry Smith PetscFunctionBegin; 16773a40ed3dSBarry Smith ierr = VecNorm(y,NORM_2, &norm );CHKERRQ(ierr); 16789b94acceSBarry Smith if (norm > *delta) { 16799b94acceSBarry Smith norm = *delta/norm; 16809b94acceSBarry Smith *gpnorm = (1.0 - norm)*(*fnorm); 16819b94acceSBarry Smith cnorm = norm; 16829b94acceSBarry Smith VecScale( &cnorm, y ); 16839b94acceSBarry Smith *ynorm = *delta; 16849b94acceSBarry Smith } else { 16859b94acceSBarry Smith *gpnorm = 0.0; 16869b94acceSBarry Smith *ynorm = norm; 16879b94acceSBarry Smith } 16883a40ed3dSBarry Smith PetscFunctionReturn(0); 16899b94acceSBarry Smith } 16909b94acceSBarry Smith 16915615d1e5SSatish Balay #undef __FUNC__ 16925615d1e5SSatish Balay #define __FUNC__ "SNESSolve" 16939b94acceSBarry Smith /*@ 16949b94acceSBarry Smith SNESSolve - Solves a nonlinear system. Call SNESSolve after calling 169563a78c88SLois Curfman McInnes SNESCreate() and optional routines of the form SNESSetXXX(). 16969b94acceSBarry Smith 1697c7afd0dbSLois Curfman McInnes Collective on SNES 1698c7afd0dbSLois Curfman McInnes 1699b2002411SLois Curfman McInnes Input Parameters: 1700c7afd0dbSLois Curfman McInnes + snes - the SNES context 1701c7afd0dbSLois Curfman McInnes - x - the solution vector 17029b94acceSBarry Smith 17039b94acceSBarry Smith Output Parameter: 1704b2002411SLois Curfman McInnes . its - number of iterations until termination 17059b94acceSBarry Smith 1706b2002411SLois Curfman McInnes Notes: 17078ddd3da0SLois Curfman McInnes The user should initialize the vector, x, with the initial guess 17088ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 17098ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 17108ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 17118ddd3da0SLois Curfman McInnes 17129b94acceSBarry Smith .keywords: SNES, nonlinear, solve 17139b94acceSBarry Smith 171463a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESDestroy() 17159b94acceSBarry Smith @*/ 17168ddd3da0SLois Curfman McInnes int SNESSolve(SNES snes,Vec x,int *its) 17179b94acceSBarry Smith { 17183c7409f5SSatish Balay int ierr, flg; 1719052efed2SBarry Smith 17203a40ed3dSBarry Smith PetscFunctionBegin; 172177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 172274679c65SBarry Smith PetscValidIntPointer(its); 172382bf6240SBarry Smith if (!snes->setupcalled) {ierr = SNESSetUp(snes,x); CHKERRQ(ierr);} 1724c4fc05e7SBarry Smith else {snes->vec_sol = snes->vec_sol_always = x;} 17259b94acceSBarry Smith PLogEventBegin(SNES_Solve,snes,0,0,0); 1726c96a6f78SLois Curfman McInnes snes->nfuncs = 0; snes->linear_its = 0; snes->nfailures = 0; 17279b94acceSBarry Smith ierr = (*(snes)->solve)(snes,its); CHKERRQ(ierr); 17289b94acceSBarry Smith PLogEventEnd(SNES_Solve,snes,0,0,0); 17293c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-snes_view", &flg); CHKERRQ(ierr); 17306d4a8577SBarry Smith if (flg) { ierr = SNESView(snes,VIEWER_STDOUT_WORLD); CHKERRQ(ierr); } 17313a40ed3dSBarry Smith PetscFunctionReturn(0); 17329b94acceSBarry Smith } 17339b94acceSBarry Smith 17349b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 17359b94acceSBarry Smith 17365615d1e5SSatish Balay #undef __FUNC__ 17375615d1e5SSatish Balay #define __FUNC__ "SNESSetType" 173882bf6240SBarry Smith /*@C 17394b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 17409b94acceSBarry Smith 1741fee21e36SBarry Smith Collective on SNES 1742fee21e36SBarry Smith 1743c7afd0dbSLois Curfman McInnes Input Parameters: 1744c7afd0dbSLois Curfman McInnes + snes - the SNES context 1745c7afd0dbSLois Curfman McInnes - method - a known method 1746c7afd0dbSLois Curfman McInnes 1747c7afd0dbSLois Curfman McInnes Options Database Key: 1748c7afd0dbSLois Curfman McInnes . -snes_type <method> - Sets the method; use -help for a list 1749c7afd0dbSLois Curfman McInnes of available methods (for instance, ls or tr) 1750ae12b187SLois Curfman McInnes 17519b94acceSBarry Smith Notes: 17529b94acceSBarry Smith See "petsc/include/snes.h" for available methods (for instance) 1753c7afd0dbSLois Curfman McInnes . SNES_EQ_LS - Newton's method with line search 1754c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 1755c7afd0dbSLois Curfman McInnes . SNES_EQ_TR - Newton's method with trust region 1756c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 1757c7afd0dbSLois Curfman McInnes . SNES_UM_TR - Newton's method with trust region 1758c7afd0dbSLois Curfman McInnes (unconstrained minimization) 1759c7afd0dbSLois Curfman McInnes . SNES_UM_LS - Newton's method with line search 1760c7afd0dbSLois Curfman McInnes (unconstrained minimization) 17619b94acceSBarry Smith 1762ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 1763ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 1764ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 1765ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 1766ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 1767ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 1768ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 1769ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 1770ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 1771ae12b187SLois Curfman McInnes appropriate method. In other words, this routine is for the advanced user. 1772a703fe33SLois Curfman McInnes 1773f536c699SSatish Balay .keywords: SNES, set, method 17749b94acceSBarry Smith @*/ 17754b0e389bSBarry Smith int SNESSetType(SNES snes,SNESType method) 17769b94acceSBarry Smith { 177784cb2905SBarry Smith int ierr; 17789b94acceSBarry Smith int (*r)(SNES); 17793a40ed3dSBarry Smith 17803a40ed3dSBarry Smith PetscFunctionBegin; 178177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 178282bf6240SBarry Smith 178382bf6240SBarry Smith if (!PetscStrcmp(snes->type_name,method)) PetscFunctionReturn(0); 178482bf6240SBarry Smith 178582bf6240SBarry Smith if (snes->setupcalled) { 1786e1311b90SBarry Smith ierr = (*(snes)->destroy)(snes);CHKERRQ(ierr); 178782bf6240SBarry Smith snes->data = 0; 178882bf6240SBarry Smith } 17897d1a2b2bSBarry Smith 17909b94acceSBarry Smith /* Get the function pointers for the iterative method requested */ 179182bf6240SBarry Smith if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL); CHKERRQ(ierr);} 179282bf6240SBarry Smith 1793488ecbafSBarry Smith ierr = FListFind(snes->comm, SNESList, method,(int (**)(void *)) &r );CHKERRQ(ierr); 179482bf6240SBarry Smith 17951548b14aSBarry Smith if (!r) SETERRQ(1,1,"Unable to find requested SNES type"); 17961548b14aSBarry Smith 17970452661fSBarry Smith if (snes->data) PetscFree(snes->data); 179882bf6240SBarry Smith snes->data = 0; 17993a40ed3dSBarry Smith ierr = (*r)(snes); CHKERRQ(ierr); 180082bf6240SBarry Smith 180182bf6240SBarry Smith if (snes->type_name) PetscFree(snes->type_name); 180282bf6240SBarry Smith snes->type_name = (char *) PetscMalloc((PetscStrlen(method)+1)*sizeof(char));CHKPTRQ(snes->type_name); 180382bf6240SBarry Smith PetscStrcpy(snes->type_name,method); 180482bf6240SBarry Smith snes->set_method_called = 1; 180582bf6240SBarry Smith 18063a40ed3dSBarry Smith PetscFunctionReturn(0); 18079b94acceSBarry Smith } 18089b94acceSBarry Smith 1809a847f771SSatish Balay 18109b94acceSBarry Smith /* --------------------------------------------------------------------- */ 18115615d1e5SSatish Balay #undef __FUNC__ 1812d4bb536fSBarry Smith #define __FUNC__ "SNESRegisterDestroy" 18139b94acceSBarry Smith /*@C 18149b94acceSBarry Smith SNESRegisterDestroy - Frees the list of nonlinear solvers that were 18159b94acceSBarry Smith registered by SNESRegister(). 18169b94acceSBarry Smith 1817fee21e36SBarry Smith Not Collective 1818fee21e36SBarry Smith 18199b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy 18209b94acceSBarry Smith 18219b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll() 18229b94acceSBarry Smith @*/ 1823cf256101SBarry Smith int SNESRegisterDestroy(void) 18249b94acceSBarry Smith { 182582bf6240SBarry Smith int ierr; 182682bf6240SBarry Smith 18273a40ed3dSBarry Smith PetscFunctionBegin; 182882bf6240SBarry Smith if (SNESList) { 1829488ecbafSBarry Smith ierr = FListDestroy( SNESList );CHKERRQ(ierr); 183082bf6240SBarry Smith SNESList = 0; 18319b94acceSBarry Smith } 183284cb2905SBarry Smith SNESRegisterAllCalled = 0; 18333a40ed3dSBarry Smith PetscFunctionReturn(0); 18349b94acceSBarry Smith } 18359b94acceSBarry Smith 18365615d1e5SSatish Balay #undef __FUNC__ 1837d4bb536fSBarry Smith #define __FUNC__ "SNESGetType" 18389b94acceSBarry Smith /*@C 18399a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 18409b94acceSBarry Smith 1841c7afd0dbSLois Curfman McInnes Not Collective 1842c7afd0dbSLois Curfman McInnes 18439b94acceSBarry Smith Input Parameter: 18444b0e389bSBarry Smith . snes - nonlinear solver context 18459b94acceSBarry Smith 18469b94acceSBarry Smith Output Parameter: 184782bf6240SBarry Smith . method - SNES method (a charactor string) 18489b94acceSBarry Smith 18499b94acceSBarry Smith .keywords: SNES, nonlinear, get, method, name 18509b94acceSBarry Smith @*/ 185182bf6240SBarry Smith int SNESGetType(SNES snes, SNESType *method) 18529b94acceSBarry Smith { 18533a40ed3dSBarry Smith PetscFunctionBegin; 185482bf6240SBarry Smith *method = snes->type_name; 18553a40ed3dSBarry Smith PetscFunctionReturn(0); 18569b94acceSBarry Smith } 18579b94acceSBarry Smith 18585615d1e5SSatish Balay #undef __FUNC__ 1859d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolution" 18609b94acceSBarry Smith /*@C 18619b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 18629b94acceSBarry Smith stored. 18639b94acceSBarry Smith 1864c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 1865c7afd0dbSLois Curfman McInnes 18669b94acceSBarry Smith Input Parameter: 18679b94acceSBarry Smith . snes - the SNES context 18689b94acceSBarry Smith 18699b94acceSBarry Smith Output Parameter: 18709b94acceSBarry Smith . x - the solution 18719b94acceSBarry Smith 18729b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 18739b94acceSBarry Smith 1874a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESGetGradient(), SNESGetSolutionUpdate() 18759b94acceSBarry Smith @*/ 18769b94acceSBarry Smith int SNESGetSolution(SNES snes,Vec *x) 18779b94acceSBarry Smith { 18783a40ed3dSBarry Smith PetscFunctionBegin; 187977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 18809b94acceSBarry Smith *x = snes->vec_sol_always; 18813a40ed3dSBarry Smith PetscFunctionReturn(0); 18829b94acceSBarry Smith } 18839b94acceSBarry Smith 18845615d1e5SSatish Balay #undef __FUNC__ 1885d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolutionUpdate" 18869b94acceSBarry Smith /*@C 18879b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 18889b94acceSBarry Smith stored. 18899b94acceSBarry Smith 1890c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 1891c7afd0dbSLois Curfman McInnes 18929b94acceSBarry Smith Input Parameter: 18939b94acceSBarry Smith . snes - the SNES context 18949b94acceSBarry Smith 18959b94acceSBarry Smith Output Parameter: 18969b94acceSBarry Smith . x - the solution update 18979b94acceSBarry Smith 18989b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 18999b94acceSBarry Smith 19009b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction 19019b94acceSBarry Smith @*/ 19029b94acceSBarry Smith int SNESGetSolutionUpdate(SNES snes,Vec *x) 19039b94acceSBarry Smith { 19043a40ed3dSBarry Smith PetscFunctionBegin; 190577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 19069b94acceSBarry Smith *x = snes->vec_sol_update_always; 19073a40ed3dSBarry Smith PetscFunctionReturn(0); 19089b94acceSBarry Smith } 19099b94acceSBarry Smith 19105615d1e5SSatish Balay #undef __FUNC__ 1911d4bb536fSBarry Smith #define __FUNC__ "SNESGetFunction" 19129b94acceSBarry Smith /*@C 19133638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 19149b94acceSBarry Smith 1915c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 1916c7afd0dbSLois Curfman McInnes 19179b94acceSBarry Smith Input Parameter: 19189b94acceSBarry Smith . snes - the SNES context 19199b94acceSBarry Smith 19209b94acceSBarry Smith Output Parameter: 19213638b69dSLois Curfman McInnes . r - the function 19229b94acceSBarry Smith 19239b94acceSBarry Smith Notes: 19249b94acceSBarry Smith SNESGetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only 19259b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 19269b94acceSBarry Smith SNESGetMinimizationFunction() and SNESGetGradient(); 19279b94acceSBarry Smith 1928a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 19299b94acceSBarry Smith 193031615d04SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution(), SNESGetMinimizationFunction(), 193131615d04SLois Curfman McInnes SNESGetGradient() 19329b94acceSBarry Smith @*/ 19339b94acceSBarry Smith int SNESGetFunction(SNES snes,Vec *r) 19349b94acceSBarry Smith { 19353a40ed3dSBarry Smith PetscFunctionBegin; 193677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1937a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1938a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 1939a8c6a408SBarry Smith } 19409b94acceSBarry Smith *r = snes->vec_func_always; 19413a40ed3dSBarry Smith PetscFunctionReturn(0); 19429b94acceSBarry Smith } 19439b94acceSBarry Smith 19445615d1e5SSatish Balay #undef __FUNC__ 1945d4bb536fSBarry Smith #define __FUNC__ "SNESGetGradient" 19469b94acceSBarry Smith /*@C 19473638b69dSLois Curfman McInnes SNESGetGradient - Returns the vector where the gradient is stored. 19489b94acceSBarry Smith 1949c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 1950c7afd0dbSLois Curfman McInnes 19519b94acceSBarry Smith Input Parameter: 19529b94acceSBarry Smith . snes - the SNES context 19539b94acceSBarry Smith 19549b94acceSBarry Smith Output Parameter: 19559b94acceSBarry Smith . r - the gradient 19569b94acceSBarry Smith 19579b94acceSBarry Smith Notes: 19589b94acceSBarry Smith SNESGetGradient() is valid for SNES_UNCONSTRAINED_MINIMIZATION methods 19599b94acceSBarry Smith only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 19609b94acceSBarry Smith SNESGetFunction(). 19619b94acceSBarry Smith 19629b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient 19639b94acceSBarry Smith 196431615d04SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESGetSolution(), SNESGetFunction() 19659b94acceSBarry Smith @*/ 19669b94acceSBarry Smith int SNESGetGradient(SNES snes,Vec *r) 19679b94acceSBarry Smith { 19683a40ed3dSBarry Smith PetscFunctionBegin; 196977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 19703a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1971a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 19723a40ed3dSBarry Smith } 19739b94acceSBarry Smith *r = snes->vec_func_always; 19743a40ed3dSBarry Smith PetscFunctionReturn(0); 19759b94acceSBarry Smith } 19769b94acceSBarry Smith 19775615d1e5SSatish Balay #undef __FUNC__ 1978d4bb536fSBarry Smith #define __FUNC__ "SNESGetMinimizationFunction" 19799b94acceSBarry Smith /*@ 19809b94acceSBarry Smith SNESGetMinimizationFunction - Returns the scalar function value for 19819b94acceSBarry Smith unconstrained minimization problems. 19829b94acceSBarry Smith 1983c7afd0dbSLois Curfman McInnes Not Collective 1984c7afd0dbSLois Curfman McInnes 19859b94acceSBarry Smith Input Parameter: 19869b94acceSBarry Smith . snes - the SNES context 19879b94acceSBarry Smith 19889b94acceSBarry Smith Output Parameter: 19899b94acceSBarry Smith . r - the function 19909b94acceSBarry Smith 19919b94acceSBarry Smith Notes: 19929b94acceSBarry Smith SNESGetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 19939b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 19949b94acceSBarry Smith SNESGetFunction(). 19959b94acceSBarry Smith 19969b94acceSBarry Smith .keywords: SNES, nonlinear, get, function 19979b94acceSBarry Smith 199831615d04SLois Curfman McInnes .seealso: SNESGetGradient(), SNESGetSolution(), SNESGetFunction() 19999b94acceSBarry Smith @*/ 20009b94acceSBarry Smith int SNESGetMinimizationFunction(SNES snes,double *r) 20019b94acceSBarry Smith { 20023a40ed3dSBarry Smith PetscFunctionBegin; 200377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 200474679c65SBarry Smith PetscValidScalarPointer(r); 20053a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 2006a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 20073a40ed3dSBarry Smith } 20089b94acceSBarry Smith *r = snes->fc; 20093a40ed3dSBarry Smith PetscFunctionReturn(0); 20109b94acceSBarry Smith } 20119b94acceSBarry Smith 20125615d1e5SSatish Balay #undef __FUNC__ 2013d4bb536fSBarry Smith #define __FUNC__ "SNESSetOptionsPrefix" 20143c7409f5SSatish Balay /*@C 20153c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 2016d850072dSLois Curfman McInnes SNES options in the database. 20173c7409f5SSatish Balay 2018fee21e36SBarry Smith Collective on SNES 2019fee21e36SBarry Smith 2020c7afd0dbSLois Curfman McInnes Input Parameter: 2021c7afd0dbSLois Curfman McInnes + snes - the SNES context 2022c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2023c7afd0dbSLois Curfman McInnes 2024d850072dSLois Curfman McInnes Notes: 2025a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2026c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2027d850072dSLois Curfman McInnes 20283c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 2029a86d99e1SLois Curfman McInnes 2030a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 20313c7409f5SSatish Balay @*/ 20323c7409f5SSatish Balay int SNESSetOptionsPrefix(SNES snes,char *prefix) 20333c7409f5SSatish Balay { 20343c7409f5SSatish Balay int ierr; 20353c7409f5SSatish Balay 20363a40ed3dSBarry Smith PetscFunctionBegin; 203777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2038639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 20393c7409f5SSatish Balay ierr = SLESSetOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 20403a40ed3dSBarry Smith PetscFunctionReturn(0); 20413c7409f5SSatish Balay } 20423c7409f5SSatish Balay 20435615d1e5SSatish Balay #undef __FUNC__ 2044d4bb536fSBarry Smith #define __FUNC__ "SNESAppendOptionsPrefix" 20453c7409f5SSatish Balay /*@C 2046f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 2047d850072dSLois Curfman McInnes SNES options in the database. 20483c7409f5SSatish Balay 2049fee21e36SBarry Smith Collective on SNES 2050fee21e36SBarry Smith 2051c7afd0dbSLois Curfman McInnes Input Parameters: 2052c7afd0dbSLois Curfman McInnes + snes - the SNES context 2053c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2054c7afd0dbSLois Curfman McInnes 2055d850072dSLois Curfman McInnes Notes: 2056a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2057c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2058d850072dSLois Curfman McInnes 20593c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 2060a86d99e1SLois Curfman McInnes 2061a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 20623c7409f5SSatish Balay @*/ 20633c7409f5SSatish Balay int SNESAppendOptionsPrefix(SNES snes,char *prefix) 20643c7409f5SSatish Balay { 20653c7409f5SSatish Balay int ierr; 20663c7409f5SSatish Balay 20673a40ed3dSBarry Smith PetscFunctionBegin; 206877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2069639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 20703c7409f5SSatish Balay ierr = SLESAppendOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 20713a40ed3dSBarry Smith PetscFunctionReturn(0); 20723c7409f5SSatish Balay } 20733c7409f5SSatish Balay 20745615d1e5SSatish Balay #undef __FUNC__ 2075d4bb536fSBarry Smith #define __FUNC__ "SNESGetOptionsPrefix" 2076c83590e2SSatish Balay /*@ 20773c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 20783c7409f5SSatish Balay SNES options in the database. 20793c7409f5SSatish Balay 2080c7afd0dbSLois Curfman McInnes Not Collective 2081c7afd0dbSLois Curfman McInnes 20823c7409f5SSatish Balay Input Parameter: 20833c7409f5SSatish Balay . snes - the SNES context 20843c7409f5SSatish Balay 20853c7409f5SSatish Balay Output Parameter: 20863c7409f5SSatish Balay . prefix - pointer to the prefix string used 20873c7409f5SSatish Balay 20883c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 2089a86d99e1SLois Curfman McInnes 2090a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 20913c7409f5SSatish Balay @*/ 20923c7409f5SSatish Balay int SNESGetOptionsPrefix(SNES snes,char **prefix) 20933c7409f5SSatish Balay { 20943c7409f5SSatish Balay int ierr; 20953c7409f5SSatish Balay 20963a40ed3dSBarry Smith PetscFunctionBegin; 209777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2098639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 20993a40ed3dSBarry Smith PetscFunctionReturn(0); 21003c7409f5SSatish Balay } 21013c7409f5SSatish Balay 2102ca161407SBarry Smith #undef __FUNC__ 2103ca161407SBarry Smith #define __FUNC__ "SNESPrintHelp" 2104ca161407SBarry Smith /*@ 2105ca161407SBarry Smith SNESPrintHelp - Prints all options for the SNES component. 2106ca161407SBarry Smith 2107c7afd0dbSLois Curfman McInnes Collective on SNES 2108c7afd0dbSLois Curfman McInnes 2109ca161407SBarry Smith Input Parameter: 2110ca161407SBarry Smith . snes - the SNES context 2111ca161407SBarry Smith 2112ca161407SBarry Smith Options Database Keys: 2113c7afd0dbSLois Curfman McInnes + -help - Prints SNES options 2114c7afd0dbSLois Curfman McInnes - -h - Prints SNES options 2115ca161407SBarry Smith 2116ca161407SBarry Smith .keywords: SNES, nonlinear, help 2117ca161407SBarry Smith 2118ca161407SBarry Smith .seealso: SNESSetFromOptions() 2119ca161407SBarry Smith @*/ 2120ca161407SBarry Smith int SNESPrintHelp(SNES snes) 2121ca161407SBarry Smith { 2122ca161407SBarry Smith char p[64]; 2123ca161407SBarry Smith SNES_KSP_EW_ConvCtx *kctx; 2124ca161407SBarry Smith int ierr; 2125ca161407SBarry Smith 2126ca161407SBarry Smith PetscFunctionBegin; 2127ca161407SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2128ca161407SBarry Smith 2129ca161407SBarry Smith PetscStrcpy(p,"-"); 2130ca161407SBarry Smith if (snes->prefix) PetscStrcat(p, snes->prefix); 2131ca161407SBarry Smith 2132ca161407SBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 2133ca161407SBarry Smith 213476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm,"SNES options ------------------------------------------------\n"); 2135488ecbafSBarry Smith ierr = FListPrintTypes(snes->comm,stdout,snes->prefix,"snes_type",SNESList);CHKERRQ(ierr); 213676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_view: view SNES info after each nonlinear solve\n",p); 213776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_max_it <its>: max iterations (default %d)\n",p,snes->max_its); 213876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_max_funcs <maxf>: max function evals (default %d)\n",p,snes->max_funcs); 213976be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_stol <stol>: successive step tolerance (default %g)\n",p,snes->xtol); 214076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_atol <atol>: absolute tolerance (default %g)\n",p,snes->atol); 214176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_rtol <rtol>: relative tolerance (default %g)\n",p,snes->rtol); 214276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_trtol <trtol>: trust region parameter tolerance (default %g)\n",p,snes->deltatol); 214376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," SNES Monitoring Options: Choose any of the following\n"); 214476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_cancelmonitors: cancels all monitors hardwired in code\n",p); 214576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_monitor: use default SNES convergence monitor, prints\n\ 2146ca161407SBarry Smith residual norm at each iteration.\n",p); 214776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_smonitor: same as the above, but prints fewer digits of the\n\ 2148ca161407SBarry Smith residual norm for small residual norms. This is useful to conceal\n\ 2149ca161407SBarry Smith meaningless digits that may be different on different machines.\n",p); 215076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_xmonitor [x,y,w,h]: use X graphics convergence monitor\n",p); 2151ca161407SBarry Smith if (snes->type == SNES_NONLINEAR_EQUATIONS) { 215276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2153ca161407SBarry Smith " Options for solving systems of nonlinear equations only:\n"); 215476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fd: use finite differences for Jacobian\n",p); 215576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf: use matrix-free Jacobian\n",p); 215676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf_operator:use matrix-free Jacobian and user-provided preconditioning matrix\n",p); 2157*e24b481bSBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf_ksp_monitor - if using matrix-free multiply then prints h at each KSP iteration\n",p); 215876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_no_convergence_test: Do not test for convergence, always run to SNES max its\n",p); 215976be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_ksp_ew_conv: use Eisenstat-Walker computation of KSP rtol. Params are:\n",p); 216076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2161ca161407SBarry Smith " %ssnes_ksp_ew_version <version> (1 or 2, default is %d)\n",p,kctx->version); 216276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2163ca161407SBarry Smith " %ssnes_ksp_ew_rtol0 <rtol0> (0 <= rtol0 < 1, default %g)\n",p,kctx->rtol_0); 216476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2165ca161407SBarry Smith " %ssnes_ksp_ew_rtolmax <rtolmax> (0 <= rtolmax < 1, default %g)\n",p,kctx->rtol_max); 216676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2167ca161407SBarry Smith " %ssnes_ksp_ew_gamma <gamma> (0 <= gamma <= 1, default %g)\n",p,kctx->gamma); 216876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2169ca161407SBarry Smith " %ssnes_ksp_ew_alpha <alpha> (1 < alpha <= 2, default %g)\n",p,kctx->alpha); 217076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2171ca161407SBarry Smith " %ssnes_ksp_ew_alpha2 <alpha2> (default %g)\n",p,kctx->alpha2); 217276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2173ca161407SBarry Smith " %ssnes_ksp_ew_threshold <threshold> (0 < threshold < 1, default %g)\n",p,kctx->threshold); 2174ca161407SBarry Smith } else if (snes->type == SNES_UNCONSTRAINED_MINIMIZATION) { 217576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2176ca161407SBarry Smith " Options for solving unconstrained minimization problems only:\n"); 217776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fmin <ftol>: minimum function tolerance (default %g)\n",p,snes->fmin); 217876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fd: use finite differences for Hessian\n",p); 217976be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf: use matrix-free Hessian\n",p); 2180ca161407SBarry Smith } 218176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," Run program with -help %ssnes_type <method> for help on ",p); 218276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm,"a particular method\n"); 2183ca161407SBarry Smith if (snes->printhelp) { 2184ca161407SBarry Smith ierr = (*snes->printhelp)(snes,p);CHKERRQ(ierr); 2185ca161407SBarry Smith } 2186ca161407SBarry Smith PetscFunctionReturn(0); 2187ca161407SBarry Smith } 21883c7409f5SSatish Balay 2189acb85ae6SSatish Balay /*MC 2190b2002411SLois Curfman McInnes SNESRegister - Adds a method to the nonlinear solver package. 21919b94acceSBarry Smith 2192b2002411SLois Curfman McInnes Synopsis: 2193b2002411SLois Curfman McInnes SNESRegister(char *name_solver,char *path,char *name_create,int (*routine_create)(SNES)) 21949b94acceSBarry Smith 21958d76a1e5SLois Curfman McInnes Not collective 21968d76a1e5SLois Curfman McInnes 2197b2002411SLois Curfman McInnes Input Parameters: 2198c7afd0dbSLois Curfman McInnes + name_solver - name of a new user-defined solver 2199b2002411SLois Curfman McInnes . path - path (either absolute or relative) the library containing this solver 2200b2002411SLois Curfman McInnes . name_create - name of routine to create method context 2201c7afd0dbSLois Curfman McInnes - routine_create - routine to create method context 22029b94acceSBarry Smith 2203b2002411SLois Curfman McInnes Notes: 2204b2002411SLois Curfman McInnes SNESRegister() may be called multiple times to add several user-defined solvers. 22053a40ed3dSBarry Smith 2206b2002411SLois Curfman McInnes If dynamic libraries are used, then the fourth input argument (routine_create) 2207b2002411SLois Curfman McInnes is ignored. 2208b2002411SLois Curfman McInnes 2209b2002411SLois Curfman McInnes Sample usage: 221069225d78SLois Curfman McInnes .vb 2211b2002411SLois Curfman McInnes SNESRegister("my_solver",/home/username/my_lib/lib/libg/solaris/mylib.a, 2212b2002411SLois Curfman McInnes "MySolverCreate",MySolverCreate); 221369225d78SLois Curfman McInnes .ve 2214b2002411SLois Curfman McInnes 2215b2002411SLois Curfman McInnes Then, your solver can be chosen with the procedural interface via 2216b2002411SLois Curfman McInnes $ SNESSetType(snes,"my_solver") 2217b2002411SLois Curfman McInnes or at runtime via the option 2218b2002411SLois Curfman McInnes $ -snes_type my_solver 2219b2002411SLois Curfman McInnes 2220b2002411SLois Curfman McInnes .keywords: SNES, nonlinear, register 2221b2002411SLois Curfman McInnes 2222b2002411SLois Curfman McInnes .seealso: SNESRegisterAll(), SNESRegisterDestroy() 2223b2002411SLois Curfman McInnes M*/ 2224b2002411SLois Curfman McInnes 2225b2002411SLois Curfman McInnes #undef __FUNC__ 2226b2002411SLois Curfman McInnes #define __FUNC__ "SNESRegister_Private" 2227b2002411SLois Curfman McInnes int SNESRegister_Private(char *sname,char *path,char *name,int (*function)(SNES)) 2228b2002411SLois Curfman McInnes { 2229b2002411SLois Curfman McInnes char fullname[256]; 2230b2002411SLois Curfman McInnes int ierr; 2231b2002411SLois Curfman McInnes 2232b2002411SLois Curfman McInnes PetscFunctionBegin; 2233b2002411SLois Curfman McInnes PetscStrcpy(fullname,path); PetscStrcat(fullname,":");PetscStrcat(fullname,name); 2234488ecbafSBarry Smith ierr = FListAdd_Private(&SNESList,sname,fullname, (int (*)(void*))function);CHKERRQ(ierr); 2235b2002411SLois Curfman McInnes PetscFunctionReturn(0); 2236b2002411SLois Curfman McInnes } 2237