19b94acceSBarry Smith #ifndef lint 2*6daaf66cSBarry Smith static char vcid[] = "$Id: snes.c,v 1.34 1996/01/11 20:14:56 bsmith Exp bsmith $"; 39b94acceSBarry Smith #endif 49b94acceSBarry Smith 59b94acceSBarry Smith #include "draw.h" /*I "draw.h" I*/ 69b94acceSBarry Smith #include "snesimpl.h" /*I "snes.h" I*/ 7682d7d0cSBarry Smith #include "sys/nreg.h" 89b94acceSBarry Smith #include "pinclude/pviewer.h" 99b94acceSBarry Smith #include <math.h> 109b94acceSBarry Smith 114b0e389bSBarry Smith extern int SNESGetTypeFromOptions_Private(SNES,SNESType*); 124b0e389bSBarry Smith extern int SNESPrintTypes_Private(char*,char*); 139b94acceSBarry Smith 14*6daaf66cSBarry Smith /*@C 15*6daaf66cSBarry Smith SLESSetOptionsPrefix - Sets the prefix used for searching for all 16*6daaf66cSBarry Smith SLES options in the database. 17*6daaf66cSBarry Smith 18*6daaf66cSBarry Smith Input Parameter: 19*6daaf66cSBarry Smith . snes - the SNES context 20*6daaf66cSBarry Smith . prefix - the prefix to prepend to all option names 21*6daaf66cSBarry Smith 22*6daaf66cSBarry Smith .keywords: SNES, set, options, prefix, database 23*6daaf66cSBarry Smith @*/ 24*6daaf66cSBarry Smith int SNESSetOptionsPrefix(SNES snes,char *prefix) 25*6daaf66cSBarry Smith { 26*6daaf66cSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 27*6daaf66cSBarry Smith 28*6daaf66cSBarry Smith snes->prefix = (char*) PetscMalloc((1+PetscStrlen(prefix))*sizeof(char)); 29*6daaf66cSBarry Smith CHKPTRQ(snes->prefix); 30*6daaf66cSBarry Smith PetscStrcpy(snes->prefix,prefix); 31*6daaf66cSBarry Smith return 0; 32*6daaf66cSBarry Smith } 33*6daaf66cSBarry Smith 349b94acceSBarry Smith /*@ 359b94acceSBarry Smith SNESView - Prints the SNES data structure. 369b94acceSBarry Smith 379b94acceSBarry Smith Input Parameters: 389b94acceSBarry Smith . SNES - the SNES context 399b94acceSBarry Smith . viewer - visualization context 409b94acceSBarry Smith 419b94acceSBarry Smith Options Database Key: 429b94acceSBarry Smith $ -snes_view : calls SNESView() at end of SNESSolve() 439b94acceSBarry Smith 449b94acceSBarry Smith Notes: 459b94acceSBarry Smith The available visualization contexts include 469b94acceSBarry Smith $ STDOUT_VIEWER_SELF - standard output (default) 479b94acceSBarry Smith $ STDOUT_VIEWER_WORLD - synchronized standard 489b94acceSBarry Smith $ output where only the first processor opens 499b94acceSBarry Smith $ the file. All other processors send their 509b94acceSBarry Smith $ data to the first processor to print. 519b94acceSBarry Smith 529b94acceSBarry Smith The user can open alternative vistualization contexts with 53dbb450caSBarry Smith $ ViewerFileOpenASCII() - output to a specified file 549b94acceSBarry Smith 559b94acceSBarry Smith .keywords: SNES, view 569b94acceSBarry Smith 57dbb450caSBarry Smith .seealso: ViewerFileOpenASCII() 589b94acceSBarry Smith @*/ 599b94acceSBarry Smith int SNESView(SNES snes,Viewer viewer) 609b94acceSBarry Smith { 619b94acceSBarry Smith PetscObject vobj = (PetscObject) viewer; 629b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 639b94acceSBarry Smith FILE *fd; 649b94acceSBarry Smith int ierr; 659b94acceSBarry Smith SLES sles; 669b94acceSBarry Smith char *method; 679b94acceSBarry Smith 689b94acceSBarry Smith if (vobj->cookie == VIEWER_COOKIE && (vobj->type == ASCII_FILE_VIEWER || 699b94acceSBarry Smith vobj->type == ASCII_FILES_VIEWER)) { 709b94acceSBarry Smith ierr = ViewerFileGetPointer_Private(viewer,&fd); CHKERRQ(ierr); 719b94acceSBarry Smith MPIU_fprintf(snes->comm,fd,"SNES Object:\n"); 724b0e389bSBarry Smith SNESGetType(snes,PETSC_NULL,&method); 739b94acceSBarry Smith MPIU_fprintf(snes->comm,fd," method: %s\n",method); 749b94acceSBarry Smith if (snes->view) (*snes->view)((PetscObject)snes,viewer); 759b94acceSBarry Smith MPIU_fprintf(snes->comm,fd, 769b94acceSBarry Smith " maximum iterations=%d, maximum function evaluations=%d\n", 779b94acceSBarry Smith snes->max_its,snes->max_funcs); 789b94acceSBarry Smith MPIU_fprintf(snes->comm,fd, 799b94acceSBarry Smith " tolerances: relative=%g, absolute=%g, truncation=%g, solution=%g\n", 809b94acceSBarry Smith snes->rtol, snes->atol, snes->trunctol, snes->xtol); 819b94acceSBarry Smith if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) 829b94acceSBarry Smith MPIU_fprintf(snes->comm,fd," min function tolerance=%g\n",snes->fmin); 839b94acceSBarry Smith if (snes->ksp_ewconv) { 849b94acceSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 859b94acceSBarry Smith if (kctx) { 869b94acceSBarry Smith MPIU_fprintf(snes->comm,fd, 879b94acceSBarry Smith " Eisenstat-Walker computation of KSP relative tolerance (version %d)\n", 889b94acceSBarry Smith kctx->version); 899b94acceSBarry Smith MPIU_fprintf(snes->comm,fd, 909b94acceSBarry Smith " rtol_0=%g, rtol_max=%g, threshold=%g\n",kctx->rtol_0, 919b94acceSBarry Smith kctx->rtol_max,kctx->threshold); 929b94acceSBarry Smith MPIU_fprintf(snes->comm,fd," gamma=%g, alpha=%g, alpha2=%g\n", 939b94acceSBarry Smith kctx->gamma,kctx->alpha,kctx->alpha2); 949b94acceSBarry Smith } 959b94acceSBarry Smith } 969b94acceSBarry Smith SNESGetSLES(snes,&sles); 979b94acceSBarry Smith ierr = SLESView(sles,viewer); CHKERRQ(ierr); 989b94acceSBarry Smith } 999b94acceSBarry Smith return 0; 1009b94acceSBarry Smith } 1019b94acceSBarry Smith 1029b94acceSBarry Smith /*@ 103682d7d0cSBarry Smith SNESSetFromOptions - Sets various SNES and SLES parameters from user options. 1049b94acceSBarry Smith 1059b94acceSBarry Smith Input Parameter: 1069b94acceSBarry Smith . snes - the SNES context 1079b94acceSBarry Smith 1089b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 1099b94acceSBarry Smith 1109b94acceSBarry Smith .seealso: SNESPrintHelp() 1119b94acceSBarry Smith @*/ 1129b94acceSBarry Smith int SNESSetFromOptions(SNES snes) 1139b94acceSBarry Smith { 1144b0e389bSBarry Smith SNESType method; 1159b94acceSBarry Smith double tmp; 1169b94acceSBarry Smith SLES sles; 1174b0e389bSBarry Smith int ierr,version = PETSC_DEFAULT; 1189b94acceSBarry Smith double rtol_0 = PETSC_DEFAULT; 1199b94acceSBarry Smith double rtol_max = PETSC_DEFAULT; 1209b94acceSBarry Smith double gamma2 = PETSC_DEFAULT; 1219b94acceSBarry Smith double alpha = PETSC_DEFAULT; 1229b94acceSBarry Smith double alpha2 = PETSC_DEFAULT; 1239b94acceSBarry Smith double threshold = PETSC_DEFAULT; 1249b94acceSBarry Smith 1259b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 1264b0e389bSBarry Smith if (snes->setup_called)SETERRQ(1,"SNESSetFromOptions:Must call prior to SNESSetUp!"); 1274b0e389bSBarry Smith if (SNESGetTypeFromOptions_Private(snes,&method)) { 1284b0e389bSBarry Smith SNESSetType(snes,method); 1299b94acceSBarry Smith } 1305334005bSBarry Smith else if (!snes->set_method_called) { 1315334005bSBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1325334005bSBarry Smith ierr = SNESSetType(snes,SNES_EQ_NLS); CHKERRQ(ierr); 1335334005bSBarry Smith } 1345334005bSBarry Smith else { 1355334005bSBarry Smith ierr = SNESSetType(snes,SNES_UM_NTR); CHKERRQ(ierr); 1365334005bSBarry Smith } 1375334005bSBarry Smith } 1385334005bSBarry Smith 139b4fd4287SBarry Smith if (OptionsHasName(PETSC_NULL,"-help")) SNESPrintHelp(snes); 1409b94acceSBarry Smith if (OptionsGetDouble(snes->prefix,"-snes_stol",&tmp)) { 1419b94acceSBarry Smith SNESSetSolutionTolerance(snes,tmp); 1429b94acceSBarry Smith } 1439b94acceSBarry Smith if (OptionsGetDouble(snes->prefix,"-snes_ttol",&tmp)) { 1449b94acceSBarry Smith SNESSetTruncationTolerance(snes,tmp); 1459b94acceSBarry Smith } 1469b94acceSBarry Smith if (OptionsGetDouble(snes->prefix,"-snes_atol",&tmp)) { 1479b94acceSBarry Smith SNESSetAbsoluteTolerance(snes,tmp); 1489b94acceSBarry Smith } 1499b94acceSBarry Smith if (OptionsGetDouble(snes->prefix,"-snes_trtol",&tmp)) { 1509b94acceSBarry Smith SNESSetTrustRegionTolerance(snes,tmp); 1519b94acceSBarry Smith } 1529b94acceSBarry Smith if (OptionsGetDouble(snes->prefix,"-snes_rtol",&tmp)) { 1539b94acceSBarry Smith SNESSetRelativeTolerance(snes,tmp); 1549b94acceSBarry Smith } 1559b94acceSBarry Smith if (OptionsGetDouble(snes->prefix,"-snes_fmin",&tmp)) { 1569b94acceSBarry Smith SNESSetMinFunctionTolerance(snes,tmp); 1579b94acceSBarry Smith } 1589b94acceSBarry Smith OptionsGetInt(snes->prefix,"-snes_max_it",&snes->max_its); 1599b94acceSBarry Smith OptionsGetInt(snes->prefix,"-snes_max_funcs",&snes->max_funcs); 1609b94acceSBarry Smith if (OptionsHasName(snes->prefix,"-snes_ksp_ew_conv")) { 1619b94acceSBarry Smith snes->ksp_ewconv = 1; 1629b94acceSBarry Smith } 1639b94acceSBarry Smith OptionsGetInt(snes->prefix,"-snes_ksp_ew_version",&version); 1649b94acceSBarry Smith OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtol0",&rtol_0); 1659b94acceSBarry Smith OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtolmax",&rtol_max); 1669b94acceSBarry Smith OptionsGetDouble(snes->prefix,"-snes_ksp_ew_gamma",&gamma2); 1679b94acceSBarry Smith OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha",&alpha); 1689b94acceSBarry Smith OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha2",&alpha2); 1699b94acceSBarry Smith OptionsGetDouble(snes->prefix,"-snes_ksp_ew_threshold",&threshold); 1709b94acceSBarry Smith ierr = SNES_KSP_SetParametersEW(snes,version,rtol_0,rtol_max,gamma2,alpha, 1719b94acceSBarry Smith alpha2,threshold); CHKERRQ(ierr); 1729b94acceSBarry Smith if (OptionsHasName(snes->prefix,"-snes_monitor")) { 1739b94acceSBarry Smith SNESSetMonitor(snes,SNESDefaultMonitor,0); 1749b94acceSBarry Smith } 1759b94acceSBarry Smith if (OptionsHasName(snes->prefix,"-snes_smonitor")) { 1769b94acceSBarry Smith SNESSetMonitor(snes,SNESDefaultSMonitor,0); 1779b94acceSBarry Smith } 1789b94acceSBarry Smith if (OptionsHasName(snes->prefix,"-snes_xmonitor")){ 17917699dbbSLois Curfman McInnes int rank = 0; 180d7e8b826SBarry Smith DrawLG lg; 18117699dbbSLois Curfman McInnes MPI_Initialized(&rank); 18217699dbbSLois Curfman McInnes if (rank) MPI_Comm_rank(snes->comm,&rank); 18317699dbbSLois Curfman McInnes if (!rank) { 1849b94acceSBarry Smith ierr = SNESLGMonitorCreate(0,0,0,0,300,300,&lg); CHKERRQ(ierr); 1859b94acceSBarry Smith ierr = SNESSetMonitor(snes,SNESLGMonitor,(void *)lg); CHKERRQ(ierr); 1869b94acceSBarry Smith PLogObjectParent(snes,lg); 1879b94acceSBarry Smith } 1889b94acceSBarry Smith } 1899b94acceSBarry Smith if (OptionsHasName(snes->prefix,"-snes_fd") && 1909b94acceSBarry Smith snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1919b94acceSBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre, 1929b94acceSBarry Smith SNESDefaultComputeJacobian,snes->funP); CHKERRQ(ierr); 1939b94acceSBarry Smith } 1949b94acceSBarry Smith if (OptionsHasName(snes->prefix,"-snes_mf") && 1959b94acceSBarry Smith snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1969b94acceSBarry Smith Mat J; 1979b94acceSBarry Smith ierr = SNESDefaultMatrixFreeMatCreate(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1989b94acceSBarry Smith ierr = SNESSetJacobian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 1999b94acceSBarry Smith PLogObjectParent(snes,J); 2009b94acceSBarry Smith snes->mfshell = J; 2019b94acceSBarry Smith } 2029b94acceSBarry Smith ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 2039b94acceSBarry Smith ierr = SLESSetFromOptions(sles); CHKERRQ(ierr); 2049b94acceSBarry Smith if (!snes->setfromoptions) return 0; 2059b94acceSBarry Smith return (*snes->setfromoptions)(snes); 2069b94acceSBarry Smith } 2079b94acceSBarry Smith 2089b94acceSBarry Smith /*@ 2099b94acceSBarry Smith SNESPrintHelp - Prints all options for the SNES component. 2109b94acceSBarry Smith 2119b94acceSBarry Smith Input Parameter: 2129b94acceSBarry Smith . snes - the SNES context 2139b94acceSBarry Smith 214a703fe33SLois Curfman McInnes Options Database Keys: 215a703fe33SLois Curfman McInnes $ -help, -h 216a703fe33SLois Curfman McInnes 2179b94acceSBarry Smith .keywords: SNES, nonlinear, help 2189b94acceSBarry Smith 219682d7d0cSBarry Smith .seealso: SNESSetFromOptions() 2209b94acceSBarry Smith @*/ 2219b94acceSBarry Smith int SNESPrintHelp(SNES snes) 2229b94acceSBarry Smith { 223*6daaf66cSBarry Smith char p[64]; 224*6daaf66cSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 225*6daaf66cSBarry Smith 2269b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 227*6daaf66cSBarry Smith 228*6daaf66cSBarry Smith PetscStrcpy(p,"-"); 229*6daaf66cSBarry Smith if (snes->prefix) PetscStrcat(p, snes->prefix); 230*6daaf66cSBarry Smith 231*6daaf66cSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 232*6daaf66cSBarry Smith 2339b94acceSBarry Smith MPIU_printf(snes->comm,"SNES options ----------------------------\n"); 234*6daaf66cSBarry Smith SNESPrintTypes_Private(p,"snes_type"); 235*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_monitor: use default SNES monitor\n",p); 236*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_view: view SNES info after each nonlinear solve\n",p); 237*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_max_it its (default %d)\n",p,snes->max_its); 238*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_stol tol (default %g)\n",p,snes->xtol); 239*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_atol tol (default %g)\n",p,snes->atol); 240*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_rtol tol (default %g)\n",p,snes->rtol); 241*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_ttol tol (default %g)\n",p,snes->trunctol); 2429b94acceSBarry Smith MPIU_printf(snes->comm, 2439b94acceSBarry Smith " options for solving systems of nonlinear equations only:\n"); 244*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_fd: use finite differences for Jacobian\n",p); 245*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_mf: use matrix-free Jacobian\n",p); 246*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_ksp_ew_conv: use Eisenstat-Walker computation of KSP rtol. Params are:\n",p); 2479b94acceSBarry Smith MPIU_printf(snes->comm, 2489b94acceSBarry Smith " %ssnes_ksp_ew_version version (1 or 2, default is %d)\n", 249*6daaf66cSBarry Smith p,kctx->version); 2509b94acceSBarry Smith MPIU_printf(snes->comm, 2519b94acceSBarry Smith " %ssnes_ksp_ew_rtol0 rtol0 (0 <= rtol0 < 1, default %g)\n", 252*6daaf66cSBarry Smith p,kctx->rtol_0); 2539b94acceSBarry Smith MPIU_printf(snes->comm, 2549b94acceSBarry Smith " %ssnes_ksp_ew_rtolmax rtolmax (0 <= rtolmax < 1, default %g)\n", 255*6daaf66cSBarry Smith p,kctx->rtol_max); 2569b94acceSBarry Smith MPIU_printf(snes->comm, 2579b94acceSBarry Smith " %ssnes_ksp_ew_gamma gamma (0 <= gamma <= 1, default %g)\n", 258*6daaf66cSBarry Smith p,kctx->gamma); 2599b94acceSBarry Smith MPIU_printf(snes->comm, 2609b94acceSBarry Smith " %ssnes_ksp_ew_alpha alpha (1 < alpha <= 2, default %g)\n", 261*6daaf66cSBarry Smith p,kctx->alpha); 2629b94acceSBarry Smith MPIU_printf(snes->comm, 2639b94acceSBarry Smith " %ssnes_ksp_ew_alpha2 alpha2 (default %g)\n", 264*6daaf66cSBarry Smith p,kctx->alpha2); 2659b94acceSBarry Smith MPIU_printf(snes->comm, 2669b94acceSBarry Smith " %ssnes_ksp_ew_threshold threshold (0 < threshold < 1, default %g)\n", 267*6daaf66cSBarry Smith p,kctx->threshold); 2689b94acceSBarry Smith MPIU_printf(snes->comm, 2699b94acceSBarry Smith " options for solving unconstrained minimization problems only:\n"); 270*6daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_fmin tol (default %g)\n",p,snes->fmin); 271*6daaf66cSBarry Smith MPIU_printf(snes->comm," Run program with %ssnes_type method -help for help on ",p); 2729b94acceSBarry Smith MPIU_printf(snes->comm,"a particular method\n"); 273*6daaf66cSBarry Smith if (snes->printhelp) (*snes->printhelp)(snes,p); 2749b94acceSBarry Smith return 0; 2759b94acceSBarry Smith } 2769b94acceSBarry Smith /*@ 2779b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 2789b94acceSBarry Smith the nonlinear solvers. 2799b94acceSBarry Smith 2809b94acceSBarry Smith Input Parameters: 2819b94acceSBarry Smith . snes - the SNES context 2829b94acceSBarry Smith . usrP - optional user context 2839b94acceSBarry Smith 2849b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 2859b94acceSBarry Smith 2869b94acceSBarry Smith .seealso: SNESGetApplicationContext() 2879b94acceSBarry Smith @*/ 2889b94acceSBarry Smith int SNESSetApplicationContext(SNES snes,void *usrP) 2899b94acceSBarry Smith { 2909b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 2919b94acceSBarry Smith snes->user = usrP; 2929b94acceSBarry Smith return 0; 2939b94acceSBarry Smith } 2949b94acceSBarry Smith /*@C 2959b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 2969b94acceSBarry Smith nonlinear solvers. 2979b94acceSBarry Smith 2989b94acceSBarry Smith Input Parameter: 2999b94acceSBarry Smith . snes - SNES context 3009b94acceSBarry Smith 3019b94acceSBarry Smith Output Parameter: 3029b94acceSBarry Smith . usrP - user context 3039b94acceSBarry Smith 3049b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 3059b94acceSBarry Smith 3069b94acceSBarry Smith .seealso: SNESSetApplicationContext() 3079b94acceSBarry Smith @*/ 3089b94acceSBarry Smith int SNESGetApplicationContext( SNES snes, void **usrP ) 3099b94acceSBarry Smith { 3109b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3119b94acceSBarry Smith *usrP = snes->user; 3129b94acceSBarry Smith return 0; 3139b94acceSBarry Smith } 3149b94acceSBarry Smith /*@ 3159b94acceSBarry Smith SNESGetIterationNumber - Gets the current iteration number of the 3169b94acceSBarry Smith nonlinear solver. 3179b94acceSBarry Smith 3189b94acceSBarry Smith Input Parameter: 3199b94acceSBarry Smith . snes - SNES context 3209b94acceSBarry Smith 3219b94acceSBarry Smith Output Parameter: 3229b94acceSBarry Smith . iter - iteration number 3239b94acceSBarry Smith 3249b94acceSBarry Smith .keywords: SNES, nonlinear, get, iteration, number 3259b94acceSBarry Smith @*/ 3269b94acceSBarry Smith int SNESGetIterationNumber(SNES snes,int* iter) 3279b94acceSBarry Smith { 3289b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3299b94acceSBarry Smith *iter = snes->iter; 3309b94acceSBarry Smith return 0; 3319b94acceSBarry Smith } 3329b94acceSBarry Smith /*@ 3339b94acceSBarry Smith SNESGetFunctionNorm - Gets the norm of the current function that was set 3349b94acceSBarry Smith with SNESSSetFunction(). 3359b94acceSBarry Smith 3369b94acceSBarry Smith Input Parameter: 3379b94acceSBarry Smith . snes - SNES context 3389b94acceSBarry Smith 3399b94acceSBarry Smith Output Parameter: 3409b94acceSBarry Smith . fnorm - 2-norm of function 3419b94acceSBarry Smith 3429b94acceSBarry Smith Note: 3439b94acceSBarry Smith SNESGetFunctionNorm() is valid for SNES_NONLINEAR_EQUATIONS methods only. 3449b94acceSBarry Smith 3459b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm 3469b94acceSBarry Smith @*/ 3479b94acceSBarry Smith int SNESGetFunctionNorm(SNES snes,Scalar *fnorm) 3489b94acceSBarry Smith { 3499b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3509b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 35148d91487SBarry Smith "SNESGetFunctionNorm:For SNES_NONLINEAR_EQUATIONS only"); 3529b94acceSBarry Smith *fnorm = snes->norm; 3539b94acceSBarry Smith return 0; 3549b94acceSBarry Smith } 3559b94acceSBarry Smith /*@ 3569b94acceSBarry Smith SNESGetGradientNorm - Gets the norm of the current gradient that was set 3579b94acceSBarry Smith with SNESSSetGradient(). 3589b94acceSBarry Smith 3599b94acceSBarry Smith Input Parameter: 3609b94acceSBarry Smith . snes - SNES context 3619b94acceSBarry Smith 3629b94acceSBarry Smith Output Parameter: 3639b94acceSBarry Smith . fnorm - 2-norm of gradient 3649b94acceSBarry Smith 3659b94acceSBarry Smith Note: 3669b94acceSBarry Smith SNESGetGradientNorm() is valid for SNES_UNCONSTRAINED_MINIMIZATION 3679b94acceSBarry Smith methods only. 3689b94acceSBarry Smith 3699b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient, norm 3709b94acceSBarry Smith @*/ 3719b94acceSBarry Smith int SNESGetGradientNorm(SNES snes,Scalar *gnorm) 3729b94acceSBarry Smith { 3739b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3749b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 37548d91487SBarry Smith "SNESGetGradientNorm:For SNES_UNCONSTRAINED_MINIMIZATION only"); 3769b94acceSBarry Smith *gnorm = snes->norm; 3779b94acceSBarry Smith return 0; 3789b94acceSBarry Smith } 3799b94acceSBarry Smith /*@ 3809b94acceSBarry Smith SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps 3819b94acceSBarry Smith attempted by the nonlinear solver. 3829b94acceSBarry Smith 3839b94acceSBarry Smith Input Parameter: 3849b94acceSBarry Smith . snes - SNES context 3859b94acceSBarry Smith 3869b94acceSBarry Smith Output Parameter: 3879b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 3889b94acceSBarry Smith 3899b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 3909b94acceSBarry Smith @*/ 3919b94acceSBarry Smith int SNESGetNumberUnsuccessfulSteps(SNES snes,int* nfails) 3929b94acceSBarry Smith { 3939b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3949b94acceSBarry Smith *nfails = snes->nfailures; 3959b94acceSBarry Smith return 0; 3969b94acceSBarry Smith } 3979b94acceSBarry Smith /*@C 3989b94acceSBarry Smith SNESGetSLES - Returns the SLES context for a SNES solver. 3999b94acceSBarry Smith 4009b94acceSBarry Smith Input Parameter: 4019b94acceSBarry Smith . snes - the SNES context 4029b94acceSBarry Smith 4039b94acceSBarry Smith Output Parameter: 4049b94acceSBarry Smith . sles - the SLES context 4059b94acceSBarry Smith 4069b94acceSBarry Smith Notes: 4079b94acceSBarry Smith The user can then directly manipulate the SLES context to set various 4089b94acceSBarry Smith options, etc. Likewise, the user can then extract and manipulate the 4099b94acceSBarry Smith KSP and PC contexts as well. 4109b94acceSBarry Smith 4119b94acceSBarry Smith .keywords: SNES, nonlinear, get, SLES, context 4129b94acceSBarry Smith 4139b94acceSBarry Smith .seealso: SLESGetPC(), SLESGetKSP() 4149b94acceSBarry Smith @*/ 4159b94acceSBarry Smith int SNESGetSLES(SNES snes,SLES *sles) 4169b94acceSBarry Smith { 4179b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 4189b94acceSBarry Smith *sles = snes->sles; 4199b94acceSBarry Smith return 0; 4209b94acceSBarry Smith } 4219b94acceSBarry Smith /* -----------------------------------------------------------*/ 4229b94acceSBarry Smith /*@C 4239b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 4249b94acceSBarry Smith 4259b94acceSBarry Smith Input Parameter: 4269b94acceSBarry Smith . comm - MPI communicator 4279b94acceSBarry Smith . type - type of method, one of 4289b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS 4299b94acceSBarry Smith $ (for systems of nonlinear equations) 4309b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION 4319b94acceSBarry Smith $ (for unconstrained minimization) 4329b94acceSBarry Smith 4339b94acceSBarry Smith Output Parameter: 4349b94acceSBarry Smith . outsnes - the new SNES context 4359b94acceSBarry Smith 4369b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 4379b94acceSBarry Smith 4389b94acceSBarry Smith .seealso: SNESSetUp(), SNESSolve(), SNESDestroy() 4399b94acceSBarry Smith @*/ 4404b0e389bSBarry Smith int SNESCreate(MPI_Comm comm,SNESProblemType type,SNES *outsnes) 4419b94acceSBarry Smith { 4429b94acceSBarry Smith int ierr; 4439b94acceSBarry Smith SNES snes; 4449b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 44537fcc0dbSBarry Smith 4469b94acceSBarry Smith *outsnes = 0; 4470452661fSBarry Smith PetscHeaderCreate(snes,_SNES,SNES_COOKIE,SNES_UNKNOWN_METHOD,comm); 4489b94acceSBarry Smith PLogObjectCreate(snes); 4499b94acceSBarry Smith snes->max_its = 50; 4509b94acceSBarry Smith snes->max_funcs = 1000; 4519b94acceSBarry Smith snes->norm = 0.0; 4529b94acceSBarry Smith snes->rtol = 1.e-8; 4539b94acceSBarry Smith snes->atol = 1.e-10; 4549b94acceSBarry Smith snes->xtol = 1.e-8; 4559b94acceSBarry Smith snes->trunctol = 1.e-12; 4569b94acceSBarry Smith snes->nfuncs = 0; 4579b94acceSBarry Smith snes->nfailures = 0; 4589b94acceSBarry Smith snes->monitor = 0; 4599b94acceSBarry Smith snes->data = 0; 4609b94acceSBarry Smith snes->view = 0; 4619b94acceSBarry Smith snes->computeumfunction = 0; 4629b94acceSBarry Smith snes->umfunP = 0; 4639b94acceSBarry Smith snes->fc = 0; 4649b94acceSBarry Smith snes->deltatol = 1.e-12; 4659b94acceSBarry Smith snes->fmin = -1.e30; 4669b94acceSBarry Smith snes->method_class = type; 4679b94acceSBarry Smith snes->set_method_called = 0; 468a703fe33SLois Curfman McInnes snes->setup_called = 0; 4699b94acceSBarry Smith snes->ksp_ewconv = 0; 4709b94acceSBarry Smith 4719b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 4720452661fSBarry Smith kctx = PetscNew(SNES_KSP_EW_ConvCtx); CHKPTRQ(kctx); 4739b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 4749b94acceSBarry Smith kctx->version = 2; 4759b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 4769b94acceSBarry Smith this was too large for some test cases */ 4779b94acceSBarry Smith kctx->rtol_last = 0; 4789b94acceSBarry Smith kctx->rtol_max = .9; 4799b94acceSBarry Smith kctx->gamma = 1.0; 4809b94acceSBarry Smith kctx->alpha2 = .5*(1.0 + sqrt(5.0)); 4819b94acceSBarry Smith kctx->alpha = kctx->alpha2; 4829b94acceSBarry Smith kctx->threshold = .1; 4839b94acceSBarry Smith kctx->lresid_last = 0; 4849b94acceSBarry Smith kctx->norm_last = 0; 4859b94acceSBarry Smith 4869b94acceSBarry Smith ierr = SLESCreate(comm,&snes->sles); CHKERRQ(ierr); 4879b94acceSBarry Smith PLogObjectParent(snes,snes->sles) 4885334005bSBarry Smith 4895334005bSBarry Smith 4909b94acceSBarry Smith *outsnes = snes; 4919b94acceSBarry Smith return 0; 4929b94acceSBarry Smith } 4939b94acceSBarry Smith 4949b94acceSBarry Smith /* --------------------------------------------------------------- */ 4959b94acceSBarry Smith /*@C 4969b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 4979b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 4989b94acceSBarry Smith equations. 4999b94acceSBarry Smith 5009b94acceSBarry Smith Input Parameters: 5019b94acceSBarry Smith . snes - the SNES context 5029b94acceSBarry Smith . func - function evaluation routine 5039b94acceSBarry Smith . ctx - optional user-defined function context 5049b94acceSBarry Smith . r - vector to store function value 5059b94acceSBarry Smith 5069b94acceSBarry Smith Calling sequence of func: 5079b94acceSBarry Smith . func (SNES, Vec x, Vec f, void *ctx); 5089b94acceSBarry Smith 5099b94acceSBarry Smith . x - input vector 5105334005bSBarry Smith . f - vector function 5119b94acceSBarry Smith . ctx - optional user-defined context for private data for the 5129b94acceSBarry Smith function evaluation routine (may be null) 5139b94acceSBarry Smith 5149b94acceSBarry Smith Notes: 5159b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 5169b94acceSBarry Smith $ f'(x) x = -f(x), 5179b94acceSBarry Smith $ where f'(x) denotes the Jacobian matrix and f(x) is the function. 5189b94acceSBarry Smith 5199b94acceSBarry Smith SNESSetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 5209b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 5219b94acceSBarry Smith SNESSetMinimizationFunction() and SNESSetGradient(); 5229b94acceSBarry Smith 5239b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 5249b94acceSBarry Smith 5259b94acceSBarry Smith .seealso: SNESGetFunction(), SNESSetJacobian(), SNESSetSolution() 5269b94acceSBarry Smith @*/ 5275334005bSBarry Smith int SNESSetFunction( SNES snes, Vec r, int (*func)(SNES,Vec,Vec,void*),void *ctx) 5289b94acceSBarry Smith { 5299b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 5309b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 53148d91487SBarry Smith "SNESSetFunction:For SNES_NONLINEAR_EQUATIONS only"); 5329b94acceSBarry Smith snes->computefunction = func; 5339b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 5349b94acceSBarry Smith snes->funP = ctx; 5359b94acceSBarry Smith return 0; 5369b94acceSBarry Smith } 5379b94acceSBarry Smith 5389b94acceSBarry Smith /*@ 5399b94acceSBarry Smith SNESComputeFunction - Computes the function that has been set with 5409b94acceSBarry Smith SNESSetFunction(). 5419b94acceSBarry Smith 5429b94acceSBarry Smith Input Parameters: 5439b94acceSBarry Smith . snes - the SNES context 5449b94acceSBarry Smith . x - input vector 5459b94acceSBarry Smith 5469b94acceSBarry Smith Output Parameter: 5479b94acceSBarry Smith . y - function vector or its negative, as set by SNESSetFunction() 5489b94acceSBarry Smith 5499b94acceSBarry Smith Notes: 5509b94acceSBarry Smith SNESComputeFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 5519b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 5529b94acceSBarry Smith SNESComputeMinimizationFunction() and SNESComputeGradient(); 5539b94acceSBarry Smith 5549b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 5559b94acceSBarry Smith 5569b94acceSBarry Smith .seealso: SNESSetFunction() 5579b94acceSBarry Smith @*/ 5589b94acceSBarry Smith int SNESComputeFunction(SNES snes,Vec x, Vec y) 5599b94acceSBarry Smith { 5609b94acceSBarry Smith int ierr; 5619b94acceSBarry Smith 5629b94acceSBarry Smith PLogEventBegin(SNES_FunctionEval,snes,x,y,0); 5639b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 5649b94acceSBarry Smith PLogEventEnd(SNES_FunctionEval,snes,x,y,0); 5659b94acceSBarry Smith return 0; 5669b94acceSBarry Smith } 5679b94acceSBarry Smith 5689b94acceSBarry Smith /*@C 5699b94acceSBarry Smith SNESSetMinimizationFunction - Sets the function evaluation routine for 5709b94acceSBarry Smith unconstrained minimization. 5719b94acceSBarry Smith 5729b94acceSBarry Smith Input Parameters: 5739b94acceSBarry Smith . snes - the SNES context 5749b94acceSBarry Smith . func - function evaluation routine 5759b94acceSBarry Smith . ctx - optional user-defined function context 5769b94acceSBarry Smith 5779b94acceSBarry Smith Calling sequence of func: 5789b94acceSBarry Smith . func (SNES snes,Vec x,double *f,void *ctx); 5799b94acceSBarry Smith 5809b94acceSBarry Smith . x - input vector 5819b94acceSBarry Smith . f - function 5829b94acceSBarry Smith . ctx - optional user-defined context for private data for the 5839b94acceSBarry Smith function evaluation routine (may be null) 5849b94acceSBarry Smith 5859b94acceSBarry Smith Notes: 5869b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 5879b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 5889b94acceSBarry Smith SNESSetFunction(). 5899b94acceSBarry Smith 5909b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimization, function 5919b94acceSBarry Smith 5929b94acceSBarry Smith .seealso: SNESGetMinimizationFunction(), SNESSetHessian(), SNESSetGradient(), 5939b94acceSBarry Smith SNESSetSolution() 5949b94acceSBarry Smith @*/ 5959b94acceSBarry Smith int SNESSetMinimizationFunction(SNES snes,int (*func)(SNES,Vec,double*,void*), 5969b94acceSBarry Smith void *ctx) 5979b94acceSBarry Smith { 5989b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 5999b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 60048d91487SBarry Smith "SNESSetMinimizationFunction:Only for SNES_UNCONSTRAINED_MINIMIZATION"); 6019b94acceSBarry Smith snes->computeumfunction = func; 6029b94acceSBarry Smith snes->umfunP = ctx; 6039b94acceSBarry Smith return 0; 6049b94acceSBarry Smith } 6059b94acceSBarry Smith 6069b94acceSBarry Smith /*@ 6079b94acceSBarry Smith SNESComputeMinimizationFunction - Computes the function that has been 6089b94acceSBarry Smith set with SNESSetMinimizationFunction(). 6099b94acceSBarry Smith 6109b94acceSBarry Smith Input Parameters: 6119b94acceSBarry Smith . snes - the SNES context 6129b94acceSBarry Smith . x - input vector 6139b94acceSBarry Smith 6149b94acceSBarry Smith Output Parameter: 6159b94acceSBarry Smith . y - function value 6169b94acceSBarry Smith 6179b94acceSBarry Smith Notes: 6189b94acceSBarry Smith SNESComputeMinimizationFunction() is valid only for 6199b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 6209b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 6219b94acceSBarry Smith @*/ 6229b94acceSBarry Smith int SNESComputeMinimizationFunction(SNES snes,Vec x,double *y) 6239b94acceSBarry Smith { 6249b94acceSBarry Smith int ierr; 6259b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 62648d91487SBarry Smith "SNESComputeMinimizationFunction:Only for SNES_UNCONSTRAINED_MINIMIZATION"); 6279b94acceSBarry Smith PLogEventBegin(SNES_MinimizationFunctionEval,snes,x,y,0); 6289b94acceSBarry Smith ierr = (*snes->computeumfunction)(snes,x,y,snes->umfunP); CHKERRQ(ierr); 6299b94acceSBarry Smith PLogEventEnd(SNES_MinimizationFunctionEval,snes,x,y,0); 6309b94acceSBarry Smith return 0; 6319b94acceSBarry Smith } 6329b94acceSBarry Smith 6339b94acceSBarry Smith /*@C 6349b94acceSBarry Smith SNESSetGradient - Sets the gradient evaluation routine and gradient 6359b94acceSBarry Smith vector for use by the SNES routines. 6369b94acceSBarry Smith 6379b94acceSBarry Smith Input Parameters: 6389b94acceSBarry Smith . snes - the SNES context 6399b94acceSBarry Smith . func - function evaluation routine 6409b94acceSBarry Smith . ctx - optional user-defined function context 6419b94acceSBarry Smith . r - vector to store gradient value 6429b94acceSBarry Smith 6439b94acceSBarry Smith Calling sequence of func: 6449b94acceSBarry Smith . func (SNES, Vec x, Vec g, void *ctx); 6459b94acceSBarry Smith 6469b94acceSBarry Smith . x - input vector 6479b94acceSBarry Smith . g - gradient vector 6489b94acceSBarry Smith . ctx - optional user-defined context for private data for the 6499b94acceSBarry Smith function evaluation routine (may be null) 6509b94acceSBarry Smith 6519b94acceSBarry Smith Notes: 6529b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 6539b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 6549b94acceSBarry Smith SNESSetFunction(). 6559b94acceSBarry Smith 6569b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 6579b94acceSBarry Smith 6589b94acceSBarry Smith .seealso: SNESGetGradient(), SNESSetHessian(), SNESSetMinimizationFunction(), 6599b94acceSBarry Smith SNESSetSolution() 6609b94acceSBarry Smith @*/ 6619b94acceSBarry Smith int SNESSetGradient(SNES snes,Vec r,int (*func)(SNES,Vec,Vec,void*), 6629b94acceSBarry Smith void *ctx) 6639b94acceSBarry Smith { 6649b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 6659b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 66648d91487SBarry Smith "SNESSetGradient:For SNES_UNCONSTRAINED_MINIMIZATION only"); 6679b94acceSBarry Smith snes->computefunction = func; 6689b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 6699b94acceSBarry Smith snes->funP = ctx; 6709b94acceSBarry Smith return 0; 6719b94acceSBarry Smith } 6729b94acceSBarry Smith 6739b94acceSBarry Smith /*@ 6749b94acceSBarry Smith SNESComputeGradient - Computes the gradient that has been 6759b94acceSBarry Smith set with SNESSetGradient(). 6769b94acceSBarry Smith 6779b94acceSBarry Smith Input Parameters: 6789b94acceSBarry Smith . snes - the SNES context 6799b94acceSBarry Smith . x - input vector 6809b94acceSBarry Smith 6819b94acceSBarry Smith Output Parameter: 6829b94acceSBarry Smith . y - gradient vector 6839b94acceSBarry Smith 6849b94acceSBarry Smith Notes: 6859b94acceSBarry Smith SNESComputeGradient() is valid only for 6869b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 6879b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 6889b94acceSBarry Smith @*/ 6899b94acceSBarry Smith int SNESComputeGradient(SNES snes,Vec x, Vec y) 6909b94acceSBarry Smith { 6919b94acceSBarry Smith int ierr; 6929b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 69348d91487SBarry Smith "SNESComputeGradient:For SNES_UNCONSTRAINED_MINIMIZATION only"); 6949b94acceSBarry Smith PLogEventBegin(SNES_GradientEval,snes,x,y,0); 6959b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 6969b94acceSBarry Smith PLogEventEnd(SNES_GradientEval,snes,x,y,0); 6979b94acceSBarry Smith return 0; 6989b94acceSBarry Smith } 6999b94acceSBarry Smith 7009b94acceSBarry Smith int SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 7019b94acceSBarry Smith { 7029b94acceSBarry Smith int ierr; 7039b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 70448d91487SBarry Smith "SNESComputeJacobian: For SNES_NONLINEAR_EQUATIONS only"); 7059b94acceSBarry Smith if (!snes->computejacobian) return 0; 7069b94acceSBarry Smith PLogEventBegin(SNES_JacobianEval,snes,X,*A,*B); 707b4fd4287SBarry Smith *flg = ALLMAT_DIFFERENT_NONZERO_PATTERN; 7089b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 7099b94acceSBarry Smith PLogEventEnd(SNES_JacobianEval,snes,X,*A,*B); 7109b94acceSBarry Smith return 0; 7119b94acceSBarry Smith } 7129b94acceSBarry Smith 7139b94acceSBarry Smith int SNESComputeHessian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 7149b94acceSBarry Smith { 7159b94acceSBarry Smith int ierr; 7169b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 71748d91487SBarry Smith "SNESComputeHessian:For SNES_UNCONSTRAINED_MINIMIZATION only"); 7189b94acceSBarry Smith if (!snes->computejacobian) return 0; 7199b94acceSBarry Smith PLogEventBegin(SNES_HessianEval,snes,X,*A,*B); 7209b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 7219b94acceSBarry Smith PLogEventEnd(SNES_HessianEval,snes,X,*A,*B); 7229b94acceSBarry Smith return 0; 7239b94acceSBarry Smith } 7249b94acceSBarry Smith 7259b94acceSBarry Smith /*@C 7269b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 7279b94acceSBarry Smith location to store it. 7289b94acceSBarry Smith 7299b94acceSBarry Smith Input Parameters: 7309b94acceSBarry Smith . snes - the SNES context 7319b94acceSBarry Smith . A - Jacobian matrix 7329b94acceSBarry Smith . B - preconditioner matrix (usually same as the Jacobian) 7339b94acceSBarry Smith . func - Jacobian evaluation routine 7349b94acceSBarry Smith . ctx - optional user-defined context for private data for the 7359b94acceSBarry Smith Jacobian evaluation routine (may be null) 7369b94acceSBarry Smith 7379b94acceSBarry Smith Calling sequence of func: 7389b94acceSBarry Smith . func (SNES,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 7399b94acceSBarry Smith 7409b94acceSBarry Smith . x - input vector 7419b94acceSBarry Smith . A - Jacobian matrix 7429b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 7439b94acceSBarry Smith . flag - flag indicating information about matrix structure, 7449b94acceSBarry Smith same as flag in SLESSetOperators() 7459b94acceSBarry Smith . ctx - optional user-defined Jacobian context 7469b94acceSBarry Smith 7479b94acceSBarry Smith Notes: 7489b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 7499b94acceSBarry Smith This allows the Jacobian evaluation routine to replace A and/or B with a 7509b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 7519b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 7529b94acceSBarry Smith throughout the global iterations. 7539b94acceSBarry Smith 7549b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 7559b94acceSBarry Smith 7569b94acceSBarry Smith .seealso: SNESSetFunction(), SNESSetSolution() 7579b94acceSBarry Smith @*/ 7589b94acceSBarry Smith int SNESSetJacobian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 7599b94acceSBarry Smith MatStructure*,void*),void *ctx) 7609b94acceSBarry Smith { 7619b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 7629b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 76348d91487SBarry Smith "SNESSetJacobian:For SNES_NONLINEAR_EQUATIONS only"); 7649b94acceSBarry Smith snes->computejacobian = func; 7659b94acceSBarry Smith snes->jacP = ctx; 7669b94acceSBarry Smith snes->jacobian = A; 7679b94acceSBarry Smith snes->jacobian_pre = B; 7689b94acceSBarry Smith return 0; 7699b94acceSBarry Smith } 770b4fd4287SBarry Smith /*@ 771b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 772b4fd4287SBarry Smith provided context for evaluating the Jacobian. 773b4fd4287SBarry Smith 774b4fd4287SBarry Smith Input Parameter: 775b4fd4287SBarry Smith . snes - the nonlinear solver context 776b4fd4287SBarry Smith 777b4fd4287SBarry Smith Output Parameters: 778b4fd4287SBarry Smith . A - location to stash Jacobian matrix (or PETSC_NULL) 779b4fd4287SBarry Smith . B - location to stash preconditioner matrix (or PETSC_NULL) 780b4fd4287SBarry Smith . ctx - location to stash Jacobian ctx (or PETSC_NULL) 781b4fd4287SBarry Smith 782b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian() 783b4fd4287SBarry Smith @*/ 784b4fd4287SBarry Smith int SNESGetJacobian(SNES snes,Mat *A,Mat *B, void **ctx) 785b4fd4287SBarry Smith { 786b4fd4287SBarry Smith if (A) *A = snes->jacobian; 787b4fd4287SBarry Smith if (B) *B = snes->jacobian_pre; 788b4fd4287SBarry Smith if (ctx) *ctx = snes->jacP; 789b4fd4287SBarry Smith return 0; 790b4fd4287SBarry Smith } 791b4fd4287SBarry Smith 7929b94acceSBarry Smith /*@C 7939b94acceSBarry Smith SNESSetHessian - Sets the function to compute Hessian as well as the 7949b94acceSBarry Smith location to store it. 7959b94acceSBarry Smith 7969b94acceSBarry Smith Input Parameters: 7979b94acceSBarry Smith . snes - the SNES context 7989b94acceSBarry Smith . A - Hessian matrix 7999b94acceSBarry Smith . B - preconditioner matrix (usually same as the Hessian) 8009b94acceSBarry Smith . func - Jacobian evaluation routine 8019b94acceSBarry Smith . ctx - optional user-defined context for private data for the 8029b94acceSBarry Smith Hessian evaluation routine (may be null) 8039b94acceSBarry Smith 8049b94acceSBarry Smith Calling sequence of func: 8059b94acceSBarry Smith . func (SNES,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 8069b94acceSBarry Smith 8079b94acceSBarry Smith . x - input vector 8089b94acceSBarry Smith . A - Hessian matrix 8099b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 8109b94acceSBarry Smith . flag - flag indicating information about matrix structure, 8119b94acceSBarry Smith same as flag in SLESSetOperators() 8129b94acceSBarry Smith . ctx - optional user-defined Hessian context 8139b94acceSBarry Smith 8149b94acceSBarry Smith Notes: 8159b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 8169b94acceSBarry Smith This allows the Hessian evaluation routine to replace A and/or B with a 8179b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 8189b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 8199b94acceSBarry Smith throughout the global iterations. 8209b94acceSBarry Smith 8219b94acceSBarry Smith .keywords: SNES, nonlinear, set, Hessian, matrix 8229b94acceSBarry Smith 8239b94acceSBarry Smith .seealso: SNESSetMinimizationFunction(), SNESSetSolution(), SNESSetGradient() 8249b94acceSBarry Smith @*/ 8259b94acceSBarry Smith int SNESSetHessian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 8269b94acceSBarry Smith MatStructure*,void*),void *ctx) 8279b94acceSBarry Smith { 8289b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 8299b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 83048d91487SBarry Smith "SNESSetHessian:For SNES_UNCONSTRAINED_MINIMIZATION only"); 8319b94acceSBarry Smith snes->computejacobian = func; 8329b94acceSBarry Smith snes->jacP = ctx; 8339b94acceSBarry Smith snes->jacobian = A; 8349b94acceSBarry Smith snes->jacobian_pre = B; 8359b94acceSBarry Smith return 0; 8369b94acceSBarry Smith } 8379b94acceSBarry Smith 8389b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */ 8399b94acceSBarry Smith 8409b94acceSBarry Smith /*@ 8419b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 8429b94acceSBarry Smith of a nonlinear solver. Call SNESSetUp() after calling SNESCreate() 8439b94acceSBarry Smith and optional routines of the form SNESSetXXX(), but before calling 8449b94acceSBarry Smith SNESSolve(). 8459b94acceSBarry Smith 8469b94acceSBarry Smith Input Parameter: 8479b94acceSBarry Smith . snes - the SNES context 8489b94acceSBarry Smith 8499b94acceSBarry Smith .keywords: SNES, nonlinear, setup 8509b94acceSBarry Smith 8519b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 8529b94acceSBarry Smith @*/ 8539b94acceSBarry Smith int SNESSetUp(SNES snes) 8549b94acceSBarry Smith { 8559b94acceSBarry Smith int ierr; 8569b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 85737fcc0dbSBarry Smith if (!snes->vec_sol) SETERRQ(1,"SNESSetUp:Must call SNESSetSolution() first"); 8589b94acceSBarry Smith 8599b94acceSBarry Smith if ((snes->method_class == SNES_NONLINEAR_EQUATIONS)) { 86037fcc0dbSBarry Smith if (!snes->vec_func) SETERRQ(1,"SNESSetUp:Must call SNESSetFunction() first"); 86137fcc0dbSBarry Smith if (!snes->computefunction) SETERRQ(1,"SNESSetUp:Must call SNESSetFunction() first"); 86237fcc0dbSBarry Smith if (!snes->jacobian) SETERRQ(1,"SNESSetUp:Must call SNESSetJacobian() first"); 863a703fe33SLois Curfman McInnes 864a703fe33SLois Curfman McInnes /* Set the KSP stopping criterion to use the Eisenstat-Walker method */ 865a703fe33SLois Curfman McInnes if (snes->ksp_ewconv && snes->type != SNES_EQ_NTR) { 866a703fe33SLois Curfman McInnes SLES sles; KSP ksp; 867a703fe33SLois Curfman McInnes ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 868a703fe33SLois Curfman McInnes ierr = SLESGetKSP(sles,&ksp); CHKERRQ(ierr); 869a703fe33SLois Curfman McInnes ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private, 870a703fe33SLois Curfman McInnes (void *)snes); CHKERRQ(ierr); 871a703fe33SLois Curfman McInnes } 8729b94acceSBarry Smith } else if ((snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION)) { 87337fcc0dbSBarry Smith if (!snes->vec_func) SETERRQ(1,"SNESSetUp:Must call SNESSetGradient() first"); 87437fcc0dbSBarry Smith if (!snes->computefunction) SETERRQ(1,"SNESSetUp:Must call SNESSetGradient() first"); 87537fcc0dbSBarry Smith if (!snes->computeumfunction) 87637fcc0dbSBarry Smith SETERRQ(1,"SNESSetUp:Must call SNESSetMinimizationFunction() first"); 87737fcc0dbSBarry Smith if (!snes->jacobian) SETERRQ(1,"SNESSetUp:Must call SNESSetHessian() first"); 8789b94acceSBarry Smith } else SETERRQ(1,"SNESSetUp:Unknown method class"); 879a703fe33SLois Curfman McInnes if (snes->setup) {ierr = (*snes->setup)(snes); CHKERRQ(ierr);} 880a703fe33SLois Curfman McInnes snes->setup_called = 1; 881a703fe33SLois Curfman McInnes return 0; 8829b94acceSBarry Smith } 8839b94acceSBarry Smith 8849b94acceSBarry Smith /*@C 8859b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 8869b94acceSBarry Smith with SNESCreate(). 8879b94acceSBarry Smith 8889b94acceSBarry Smith Input Parameter: 8899b94acceSBarry Smith . snes - the SNES context 8909b94acceSBarry Smith 8919b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 8929b94acceSBarry Smith 8939b94acceSBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESSolve() 8949b94acceSBarry Smith @*/ 8959b94acceSBarry Smith int SNESDestroy(SNES snes) 8969b94acceSBarry Smith { 8979b94acceSBarry Smith int ierr; 8989b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 8999b94acceSBarry Smith ierr = (*(snes)->destroy)((PetscObject)snes); CHKERRQ(ierr); 9000452661fSBarry Smith if (snes->kspconvctx) PetscFree(snes->kspconvctx); 9019b94acceSBarry Smith if (snes->mfshell) MatDestroy(snes->mfshell); 9029b94acceSBarry Smith ierr = SLESDestroy(snes->sles); CHKERRQ(ierr); 9039b94acceSBarry Smith PLogObjectDestroy((PetscObject)snes); 9040452661fSBarry Smith PetscHeaderDestroy((PetscObject)snes); 9059b94acceSBarry Smith return 0; 9069b94acceSBarry Smith } 9079b94acceSBarry Smith 9089b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 9099b94acceSBarry Smith 9109b94acceSBarry Smith /*@ 9119b94acceSBarry Smith SNESSetMaxIterations - Sets the maximum number of global iterations to use. 9129b94acceSBarry Smith 9139b94acceSBarry Smith Input Parameters: 9149b94acceSBarry Smith . snes - the SNES context 9159b94acceSBarry Smith . maxits - maximum number of iterations to use 9169b94acceSBarry Smith 9179b94acceSBarry Smith Options Database Key: 9189b94acceSBarry Smith $ -snes_max_it maxits 9199b94acceSBarry Smith 9209b94acceSBarry Smith Note: 9219b94acceSBarry Smith The default maximum number of iterations is 50. 9229b94acceSBarry Smith 9239b94acceSBarry Smith .keywords: SNES, nonlinear, set, maximum, iterations 9249b94acceSBarry Smith 9259b94acceSBarry Smith .seealso: SNESSetMaxFunctionEvaluations() 9269b94acceSBarry Smith @*/ 9279b94acceSBarry Smith int SNESSetMaxIterations(SNES snes,int maxits) 9289b94acceSBarry Smith { 9299b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 9309b94acceSBarry Smith snes->max_its = maxits; 9319b94acceSBarry Smith return 0; 9329b94acceSBarry Smith } 9339b94acceSBarry Smith 9349b94acceSBarry Smith /*@ 9359b94acceSBarry Smith SNESSetMaxFunctionEvaluations - Sets the maximum number of function 9369b94acceSBarry Smith evaluations to use. 9379b94acceSBarry Smith 9389b94acceSBarry Smith Input Parameters: 9399b94acceSBarry Smith . snes - the SNES context 9409b94acceSBarry Smith . maxf - maximum number of function evaluations 9419b94acceSBarry Smith 9429b94acceSBarry Smith Options Database Key: 9439b94acceSBarry Smith $ -snes_max_funcs maxf 9449b94acceSBarry Smith 9459b94acceSBarry Smith Note: 9469b94acceSBarry Smith The default maximum number of function evaluations is 1000. 9479b94acceSBarry Smith 9489b94acceSBarry Smith .keywords: SNES, nonlinear, set, maximum, function, evaluations 9499b94acceSBarry Smith 9509b94acceSBarry Smith .seealso: SNESSetMaxIterations() 9519b94acceSBarry Smith @*/ 9529b94acceSBarry Smith int SNESSetMaxFunctionEvaluations(SNES snes,int maxf) 9539b94acceSBarry Smith { 9549b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 9559b94acceSBarry Smith snes->max_funcs = maxf; 9569b94acceSBarry Smith return 0; 9579b94acceSBarry Smith } 9589b94acceSBarry Smith 9599b94acceSBarry Smith /*@ 9609b94acceSBarry Smith SNESSetRelativeTolerance - Sets the relative convergence tolerance. 9619b94acceSBarry Smith 9629b94acceSBarry Smith Input Parameters: 9639b94acceSBarry Smith . snes - the SNES context 9649b94acceSBarry Smith . rtol - tolerance 9659b94acceSBarry Smith 9669b94acceSBarry Smith Options Database Key: 9679b94acceSBarry Smith $ -snes_rtol tol 9689b94acceSBarry Smith 9699b94acceSBarry Smith .keywords: SNES, nonlinear, set, relative, convergence, tolerance 9709b94acceSBarry Smith 9719b94acceSBarry Smith .seealso: SNESSetAbsoluteTolerance(), SNESSetSolutionTolerance(), 9729b94acceSBarry Smith SNESSetTruncationTolerance() 9739b94acceSBarry Smith @*/ 9749b94acceSBarry Smith int SNESSetRelativeTolerance(SNES snes,double rtol) 9759b94acceSBarry Smith { 9769b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 9779b94acceSBarry Smith snes->rtol = rtol; 9789b94acceSBarry Smith return 0; 9799b94acceSBarry Smith } 9809b94acceSBarry Smith 9819b94acceSBarry Smith /*@ 9829b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 9839b94acceSBarry Smith 9849b94acceSBarry Smith Input Parameters: 9859b94acceSBarry Smith . snes - the SNES context 9869b94acceSBarry Smith . tol - tolerance 9879b94acceSBarry Smith 9889b94acceSBarry Smith Options Database Key: 9899b94acceSBarry Smith $ -snes_trtol tol 9909b94acceSBarry Smith 9919b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 9929b94acceSBarry Smith 9939b94acceSBarry Smith .seealso: SNESSetAbsoluteTolerance(), SNESSetSolutionTolerance(), 9949b94acceSBarry Smith SNESSetTruncationTolerance() 9959b94acceSBarry Smith @*/ 9969b94acceSBarry Smith int SNESSetTrustRegionTolerance(SNES snes,double tol) 9979b94acceSBarry Smith { 9989b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 9999b94acceSBarry Smith snes->deltatol = tol; 10009b94acceSBarry Smith return 0; 10019b94acceSBarry Smith } 10029b94acceSBarry Smith 10039b94acceSBarry Smith /*@ 10049b94acceSBarry Smith SNESSetAbsoluteTolerance - Sets the absolute convergence tolerance. 10059b94acceSBarry Smith 10069b94acceSBarry Smith Input Parameters: 10079b94acceSBarry Smith . snes - the SNES context 10089b94acceSBarry Smith . atol - tolerance 10099b94acceSBarry Smith 10109b94acceSBarry Smith Options Database Key: 10119b94acceSBarry Smith $ -snes_atol tol 10129b94acceSBarry Smith 10139b94acceSBarry Smith .keywords: SNES, nonlinear, set, absolute, convergence, tolerance 10149b94acceSBarry Smith 10159b94acceSBarry Smith .seealso: SNESSetRelativeTolerance(), SNESSetSolutionTolerance(), 10169b94acceSBarry Smith SNESSetTruncationTolerance() 10179b94acceSBarry Smith @*/ 10189b94acceSBarry Smith int SNESSetAbsoluteTolerance(SNES snes,double atol) 10199b94acceSBarry Smith { 10209b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 10219b94acceSBarry Smith snes->atol = atol; 10229b94acceSBarry Smith return 0; 10239b94acceSBarry Smith } 10249b94acceSBarry Smith 10259b94acceSBarry Smith /*@ 10269b94acceSBarry Smith SNESSetTruncationTolerance - Sets the tolerance that may be used by the 10279b94acceSBarry Smith step routines to control the accuracy of the step computation. 10289b94acceSBarry Smith 10299b94acceSBarry Smith Input Parameters: 10309b94acceSBarry Smith . snes - the SNES context 10319b94acceSBarry Smith . tol - tolerance 10329b94acceSBarry Smith 10339b94acceSBarry Smith Options Database Key: 10349b94acceSBarry Smith $ -snes_ttol tol 10359b94acceSBarry Smith 10369b94acceSBarry Smith Notes: 10379b94acceSBarry Smith If the step computation involves an application of the inverse 10389b94acceSBarry Smith Jacobian (or Hessian), this parameter may be used to control the 1039a703fe33SLois Curfman McInnes accuracy of that application. 10409b94acceSBarry Smith 10419b94acceSBarry Smith .keywords: SNES, nonlinear, set, truncation, tolerance 10429b94acceSBarry Smith 10439b94acceSBarry Smith .seealso: SNESSetRelativeTolerance(), SNESSetSolutionTolerance(), 10449b94acceSBarry Smith SNESSetAbsoluteTolerance() 10459b94acceSBarry Smith @*/ 10469b94acceSBarry Smith int SNESSetTruncationTolerance(SNES snes,double tol) 10479b94acceSBarry Smith { 10489b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 10499b94acceSBarry Smith snes->trunctol = tol; 10509b94acceSBarry Smith return 0; 10519b94acceSBarry Smith } 10529b94acceSBarry Smith 10539b94acceSBarry Smith /*@ 10549b94acceSBarry Smith SNESSetSolutionTolerance - Sets the convergence tolerance in terms of 10559b94acceSBarry Smith the norm of the change in the solution between steps. 10569b94acceSBarry Smith 10579b94acceSBarry Smith Input Parameters: 10589b94acceSBarry Smith . snes - the SNES context 10599b94acceSBarry Smith . tol - tolerance 10609b94acceSBarry Smith 10619b94acceSBarry Smith Options Database Key: 10629b94acceSBarry Smith $ -snes_stol tol 10639b94acceSBarry Smith 10649b94acceSBarry Smith .keywords: SNES, nonlinear, set, solution, tolerance 10659b94acceSBarry Smith 10669b94acceSBarry Smith .seealso: SNESSetTruncationTolerance(), SNESSetRelativeTolerance(), 10679b94acceSBarry Smith SNESSetAbsoluteTolerance() 10689b94acceSBarry Smith @*/ 10699b94acceSBarry Smith int SNESSetSolutionTolerance( SNES snes, double tol ) 10709b94acceSBarry Smith { 10719b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 10729b94acceSBarry Smith snes->xtol = tol; 10739b94acceSBarry Smith return 0; 10749b94acceSBarry Smith } 10759b94acceSBarry Smith 10769b94acceSBarry Smith /*@ 10779b94acceSBarry Smith SNESSetMinFunctionTolerance - Sets the minimum allowable function tolerance 10789b94acceSBarry Smith for unconstrained minimization solvers. 10799b94acceSBarry Smith 10809b94acceSBarry Smith Input Parameters: 10819b94acceSBarry Smith . snes - the SNES context 10829b94acceSBarry Smith . ftol - minimum function tolerance 10839b94acceSBarry Smith 10849b94acceSBarry Smith Options Database Key: 10859b94acceSBarry Smith $ -snes_fmin ftol 10869b94acceSBarry Smith 10879b94acceSBarry Smith Note: 10889b94acceSBarry Smith SNESSetMinFunctionTolerance() is valid for SNES_UNCONSTRAINED_MINIMIZATION 10899b94acceSBarry Smith methods only. 10909b94acceSBarry Smith 10919b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimum, convergence, function, tolerance 10929b94acceSBarry Smith 10939b94acceSBarry Smith .seealso: SNESSetRelativeTolerance(), SNESSetSolutionTolerance(), 10949b94acceSBarry Smith SNESSetTruncationTolerance() 10959b94acceSBarry Smith @*/ 10969b94acceSBarry Smith int SNESSetMinFunctionTolerance(SNES snes,double ftol) 10979b94acceSBarry Smith { 10989b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 10999b94acceSBarry Smith snes->fmin = ftol; 11009b94acceSBarry Smith return 0; 11019b94acceSBarry Smith } 11029b94acceSBarry Smith 11039b94acceSBarry Smith 11049b94acceSBarry Smith 11059b94acceSBarry Smith /* ---------- Routines to set various aspects of nonlinear solver --------- */ 11069b94acceSBarry Smith 11079b94acceSBarry Smith /*@C 11089b94acceSBarry Smith SNESSetSolution - Sets the initial guess routine and solution vector 11099b94acceSBarry Smith for use by the SNES routines. 11109b94acceSBarry Smith 11119b94acceSBarry Smith Input Parameters: 11129b94acceSBarry Smith . snes - the SNES context 11139b94acceSBarry Smith . x - the solution vector 11149b94acceSBarry Smith . func - optional routine to compute an initial guess (may be null) 11159b94acceSBarry Smith . ctx - optional user-defined context for private data for the 11169b94acceSBarry Smith initial guess routine (may be null) 11179b94acceSBarry Smith 11189b94acceSBarry Smith Calling sequence of func: 1119682d7d0cSBarry Smith int guess(SNES, Vec x, void *ctx) 11209b94acceSBarry Smith 11219b94acceSBarry Smith . x - input vector 11229b94acceSBarry Smith . ctx - optional user-defined initial guess context 11239b94acceSBarry Smith 11249b94acceSBarry Smith Note: 11259b94acceSBarry Smith If no initial guess routine is indicated, an initial guess of zero 11269b94acceSBarry Smith will be used. 11279b94acceSBarry Smith 11289b94acceSBarry Smith .keywords: SNES, nonlinear, set, solution, initial guess 11299b94acceSBarry Smith 11309b94acceSBarry Smith .seealso: SNESGetSolution(), SNESSetJacobian(), SNESSetFunction() 11319b94acceSBarry Smith @*/ 11329b94acceSBarry Smith int SNESSetSolution(SNES snes,Vec x,int (*func)(SNES,Vec,void*),void *ctx) 11339b94acceSBarry Smith { 11349b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 11359b94acceSBarry Smith snes->vec_sol = snes->vec_sol_always = x; 11369b94acceSBarry Smith snes->computeinitialguess = func; 11379b94acceSBarry Smith snes->gusP = ctx; 11389b94acceSBarry Smith return 0; 11399b94acceSBarry Smith } 11409b94acceSBarry Smith 11419b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 11429b94acceSBarry Smith 11439b94acceSBarry Smith /*@C 11449b94acceSBarry Smith SNESSetMonitor - Sets the function that is to be used at every 11459b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 11469b94acceSBarry Smith progress. 11479b94acceSBarry Smith 11489b94acceSBarry Smith Input Parameters: 11499b94acceSBarry Smith . snes - the SNES context 11509b94acceSBarry Smith . func - monitoring routine 11519b94acceSBarry Smith . mctx - optional user-defined context for private data for the 11529b94acceSBarry Smith monitor routine (may be null) 11539b94acceSBarry Smith 11549b94acceSBarry Smith Calling sequence of func: 1155682d7d0cSBarry Smith int func(SNES snes,int its, Vec x,Vec f,double norm,void *mctx) 11569b94acceSBarry Smith 11579b94acceSBarry Smith $ snes - the SNES context 11589b94acceSBarry Smith $ its - iteration number 11599b94acceSBarry Smith $ mctx - optional monitoring context 11609b94acceSBarry Smith $ 11619b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods: 11629b94acceSBarry Smith $ norm - 2-norm function value (may be estimated) 11639b94acceSBarry Smith $ 11649b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods: 11659b94acceSBarry Smith $ norm - 2-norm gradient value (may be estimated) 11669b94acceSBarry Smith 11679b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 11689b94acceSBarry Smith 11699b94acceSBarry Smith .seealso: SNESDefaultMonitor() 11709b94acceSBarry Smith @*/ 11719b94acceSBarry Smith int SNESSetMonitor( SNES snes, int (*func)(SNES,int,double,void*), 11729b94acceSBarry Smith void *mctx ) 11739b94acceSBarry Smith { 11749b94acceSBarry Smith snes->monitor = func; 11759b94acceSBarry Smith snes->monP = (void*)mctx; 11769b94acceSBarry Smith return 0; 11779b94acceSBarry Smith } 11789b94acceSBarry Smith 11799b94acceSBarry Smith /*@C 11809b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 11819b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 11829b94acceSBarry Smith 11839b94acceSBarry Smith Input Parameters: 11849b94acceSBarry Smith . snes - the SNES context 11859b94acceSBarry Smith . func - routine to test for convergence 11869b94acceSBarry Smith . cctx - optional context for private data for the convergence routine 11879b94acceSBarry Smith (may be null) 11889b94acceSBarry Smith 11899b94acceSBarry Smith Calling sequence of func: 11909b94acceSBarry Smith int func (SNES snes,double xnorm,double gnorm, 11919b94acceSBarry Smith double f,void *cctx) 11929b94acceSBarry Smith 11939b94acceSBarry Smith $ snes - the SNES context 11949b94acceSBarry Smith $ cctx - optional convergence context 11959b94acceSBarry Smith $ xnorm - 2-norm of current iterate 11969b94acceSBarry Smith $ 11979b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods: 11989b94acceSBarry Smith $ gnorm - 2-norm of current step 11999b94acceSBarry Smith $ f - 2-norm of function 12009b94acceSBarry Smith $ 12019b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods: 12029b94acceSBarry Smith $ gnorm - 2-norm of current gradient 12039b94acceSBarry Smith $ f - function value 12049b94acceSBarry Smith 12059b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 12069b94acceSBarry Smith 12079b94acceSBarry Smith .seealso: SNESDefaultConverged() 12089b94acceSBarry Smith @*/ 12099b94acceSBarry Smith int SNESSetConvergenceTest(SNES snes, 12109b94acceSBarry Smith int (*func)(SNES,double,double,double,void*),void *cctx) 12119b94acceSBarry Smith { 12129b94acceSBarry Smith (snes)->converged = func; 12139b94acceSBarry Smith (snes)->cnvP = cctx; 12149b94acceSBarry Smith return 0; 12159b94acceSBarry Smith } 12169b94acceSBarry Smith 12179b94acceSBarry Smith /* 12189b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 12199b94acceSBarry Smith positive parameter delta. 12209b94acceSBarry Smith 12219b94acceSBarry Smith Input Parameters: 12229b94acceSBarry Smith . snes - the SNES context 12239b94acceSBarry Smith . y - approximate solution of linear system 12249b94acceSBarry Smith . fnorm - 2-norm of current function 12259b94acceSBarry Smith . delta - trust region size 12269b94acceSBarry Smith 12279b94acceSBarry Smith Output Parameters: 12289b94acceSBarry Smith . gpnorm - predicted function norm at the new point, assuming local 12299b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 12309b94acceSBarry Smith region, and exceeds zero otherwise. 12319b94acceSBarry Smith . ynorm - 2-norm of the step 12329b94acceSBarry Smith 12339b94acceSBarry Smith Note: 1234ddd12767SBarry Smith For non-trust region methods such as SNES_EQ_NLS, the parameter delta 12359b94acceSBarry Smith is set to be the maximum allowable step size. 12369b94acceSBarry Smith 12379b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 12389b94acceSBarry Smith */ 12399b94acceSBarry Smith int SNESScaleStep_Private(SNES snes,Vec y,double *fnorm,double *delta, 12409b94acceSBarry Smith double *gpnorm,double *ynorm) 12419b94acceSBarry Smith { 12429b94acceSBarry Smith double norm; 12439b94acceSBarry Smith Scalar cnorm; 1244cddf8d76SBarry Smith VecNorm(y,NORM_2, &norm ); 12459b94acceSBarry Smith if (norm > *delta) { 12469b94acceSBarry Smith norm = *delta/norm; 12479b94acceSBarry Smith *gpnorm = (1.0 - norm)*(*fnorm); 12489b94acceSBarry Smith cnorm = norm; 12499b94acceSBarry Smith VecScale( &cnorm, y ); 12509b94acceSBarry Smith *ynorm = *delta; 12519b94acceSBarry Smith } else { 12529b94acceSBarry Smith *gpnorm = 0.0; 12539b94acceSBarry Smith *ynorm = norm; 12549b94acceSBarry Smith } 12559b94acceSBarry Smith return 0; 12569b94acceSBarry Smith } 12579b94acceSBarry Smith 12589b94acceSBarry Smith /*@ 12599b94acceSBarry Smith SNESSolve - Solves a nonlinear system. Call SNESSolve after calling 12609b94acceSBarry Smith SNESCreate(), optional routines of the form SNESSetXXX(), and SNESSetUp(). 12619b94acceSBarry Smith 12629b94acceSBarry Smith Input Parameter: 12639b94acceSBarry Smith . snes - the SNES context 12649b94acceSBarry Smith 12659b94acceSBarry Smith Output Parameter: 12669b94acceSBarry Smith its - number of iterations until termination 12679b94acceSBarry Smith 12689b94acceSBarry Smith .keywords: SNES, nonlinear, solve 12699b94acceSBarry Smith 12709b94acceSBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy() 12719b94acceSBarry Smith @*/ 12729b94acceSBarry Smith int SNESSolve(SNES snes,int *its) 12739b94acceSBarry Smith { 12749b94acceSBarry Smith int ierr; 127537fcc0dbSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 12769b94acceSBarry Smith PLogEventBegin(SNES_Solve,snes,0,0,0); 12779b94acceSBarry Smith ierr = (*(snes)->solve)(snes,its); CHKERRQ(ierr); 12789b94acceSBarry Smith PLogEventEnd(SNES_Solve,snes,0,0,0); 1279b4fd4287SBarry Smith if (OptionsHasName(PETSC_NULL,"-snes_view")) { 128037fcc0dbSBarry Smith ierr = SNESView(snes,STDOUT_VIEWER_WORLD); CHKERRQ(ierr); 12819b94acceSBarry Smith } 12829b94acceSBarry Smith return 0; 12839b94acceSBarry Smith } 12849b94acceSBarry Smith 12859b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 12869b94acceSBarry Smith 12879b94acceSBarry Smith /* 12889b94acceSBarry Smith SNESComputeInitialGuess - Manages computation of initial approximation. 12899b94acceSBarry Smith */ 12909b94acceSBarry Smith int SNESComputeInitialGuess( SNES snes,Vec x ) 12919b94acceSBarry Smith { 12929b94acceSBarry Smith int ierr; 12939b94acceSBarry Smith Scalar zero = 0.0; 12949b94acceSBarry Smith if (snes->computeinitialguess) { 12959b94acceSBarry Smith ierr = (*snes->computeinitialguess)(snes, x, snes->gusP); CHKERRQ(ierr); 12969b94acceSBarry Smith } 12979b94acceSBarry Smith else { 12989b94acceSBarry Smith ierr = VecSet(&zero,x); CHKERRQ(ierr); 12999b94acceSBarry Smith } 13009b94acceSBarry Smith return 0; 13019b94acceSBarry Smith } 13029b94acceSBarry Smith 13039b94acceSBarry Smith /* ------------------------------------------------------------------ */ 13049b94acceSBarry Smith 130537fcc0dbSBarry Smith static NRList *__SNESList = 0; 13069b94acceSBarry Smith 13079b94acceSBarry Smith /*@ 13084b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 13099b94acceSBarry Smith 13109b94acceSBarry Smith Input Parameters: 13119b94acceSBarry Smith . snes - the SNES context 13129b94acceSBarry Smith . method - a known method 13139b94acceSBarry Smith 13149b94acceSBarry Smith Notes: 13159b94acceSBarry Smith See "petsc/include/snes.h" for available methods (for instance) 13169b94acceSBarry Smith $ Systems of nonlinear equations: 1317ddd12767SBarry Smith $ SNES_EQ_NLS - Newton's method with line search 1318ddd12767SBarry Smith $ SNES_EQ_NTR - Newton's method with trust region 13199b94acceSBarry Smith $ Unconstrained minimization: 13209b94acceSBarry Smith $ SNES_UM_NTR - Newton's method with trust region 1321a703fe33SLois Curfman McInnes $ SNES_UM_NLS - Newton's method with line search 13229b94acceSBarry Smith 13239b94acceSBarry Smith Options Database Command: 13244b0e389bSBarry Smith $ -snes_type <method> 13259b94acceSBarry Smith $ Use -help for a list of available methods 13269b94acceSBarry Smith $ (for instance, ls or tr) 1327a703fe33SLois Curfman McInnes 1328a703fe33SLois Curfman McInnes .keysords: SNES, set, method 13299b94acceSBarry Smith @*/ 13304b0e389bSBarry Smith int SNESSetType(SNES snes,SNESType method) 13319b94acceSBarry Smith { 13329b94acceSBarry Smith int (*r)(SNES); 13339b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 13349b94acceSBarry Smith /* Get the function pointers for the iterative method requested */ 133537fcc0dbSBarry Smith if (!__SNESList) {SNESRegisterAll();} 133637fcc0dbSBarry Smith if (!__SNESList) {SETERRQ(1,"SNESSetType:Could not get methods");} 133737fcc0dbSBarry Smith r = (int (*)(SNES))NRFindRoutine( __SNESList, (int)method, (char *)0 ); 13384b0e389bSBarry Smith if (!r) {SETERRQ(1,"SNESSetType:Unknown method");} 13390452661fSBarry Smith if (snes->data) PetscFree(snes->data); 13409b94acceSBarry Smith snes->set_method_called = 1; 13419b94acceSBarry Smith return (*r)(snes); 13429b94acceSBarry Smith } 13439b94acceSBarry Smith 13449b94acceSBarry Smith /* --------------------------------------------------------------------- */ 13459b94acceSBarry Smith /*@C 13469b94acceSBarry Smith SNESRegister - Adds the method to the nonlinear solver package, given 13474b0e389bSBarry Smith a function pointer and a nonlinear solver name of the type SNESType. 13489b94acceSBarry Smith 13499b94acceSBarry Smith Input Parameters: 1350ddd12767SBarry Smith . name - for instance SNES_EQ_NLS, SNES_EQ_NTR, ... 13519b94acceSBarry Smith . sname - corfunPonding string for name 13529b94acceSBarry Smith . create - routine to create method context 13539b94acceSBarry Smith 13549b94acceSBarry Smith .keywords: SNES, nonlinear, register 13559b94acceSBarry Smith 13569b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy() 13579b94acceSBarry Smith @*/ 13589b94acceSBarry Smith int SNESRegister(int name, char *sname, int (*create)(SNES)) 13599b94acceSBarry Smith { 13609b94acceSBarry Smith int ierr; 136137fcc0dbSBarry Smith if (!__SNESList) {ierr = NRCreate(&__SNESList); CHKERRQ(ierr);} 136237fcc0dbSBarry Smith NRRegister( __SNESList, name, sname, (int (*)(void*))create ); 13639b94acceSBarry Smith return 0; 13649b94acceSBarry Smith } 13659b94acceSBarry Smith /* --------------------------------------------------------------------- */ 13669b94acceSBarry Smith /*@C 13679b94acceSBarry Smith SNESRegisterDestroy - Frees the list of nonlinear solvers that were 13689b94acceSBarry Smith registered by SNESRegister(). 13699b94acceSBarry Smith 13709b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy 13719b94acceSBarry Smith 13729b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll() 13739b94acceSBarry Smith @*/ 13749b94acceSBarry Smith int SNESRegisterDestroy() 13759b94acceSBarry Smith { 137637fcc0dbSBarry Smith if (__SNESList) { 137737fcc0dbSBarry Smith NRDestroy( __SNESList ); 137837fcc0dbSBarry Smith __SNESList = 0; 13799b94acceSBarry Smith } 13809b94acceSBarry Smith return 0; 13819b94acceSBarry Smith } 13829b94acceSBarry Smith 13839b94acceSBarry Smith /* 13844b0e389bSBarry Smith SNESGetTypeFromOptions_Private - Sets the selected method from the 13859b94acceSBarry Smith options database. 13869b94acceSBarry Smith 13879b94acceSBarry Smith Input Parameter: 13889b94acceSBarry Smith . ctx - the SNES context 13899b94acceSBarry Smith 13909b94acceSBarry Smith Output Parameter: 13919b94acceSBarry Smith . method - solver method 13929b94acceSBarry Smith 13939b94acceSBarry Smith Returns: 13949b94acceSBarry Smith Returns 1 if the method is found; 0 otherwise. 13959b94acceSBarry Smith 13969b94acceSBarry Smith Options Database Key: 13974b0e389bSBarry Smith $ -snes_type method 13989b94acceSBarry Smith */ 13994b0e389bSBarry Smith int SNESGetTypeFromOptions_Private(SNES ctx,SNESType *method) 14009b94acceSBarry Smith { 14019b94acceSBarry Smith char sbuf[50]; 14024b0e389bSBarry Smith if (OptionsGetString(ctx->prefix,"-snes_type", sbuf, 50 )) { 140337fcc0dbSBarry Smith if (!__SNESList) SNESRegisterAll(); 140437fcc0dbSBarry Smith *method = (SNESType)NRFindID( __SNESList, sbuf ); 14059b94acceSBarry Smith return 1; 14069b94acceSBarry Smith } 14079b94acceSBarry Smith return 0; 14089b94acceSBarry Smith } 14099b94acceSBarry Smith 14109b94acceSBarry Smith /*@C 14119a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 14129b94acceSBarry Smith 14139b94acceSBarry Smith Input Parameter: 14144b0e389bSBarry Smith . snes - nonlinear solver context 14159b94acceSBarry Smith 14169b94acceSBarry Smith Output Parameter: 14179a28b0a6SLois Curfman McInnes . method - SNES method (or use PETSC_NULL) 14189a28b0a6SLois Curfman McInnes . name - name of SNES method (or use PETSC_NULL) 14199b94acceSBarry Smith 14209b94acceSBarry Smith .keywords: SNES, nonlinear, get, method, name 14219b94acceSBarry Smith @*/ 14224b0e389bSBarry Smith int SNESGetType(SNES snes, SNESType *method,char **name) 14239b94acceSBarry Smith { 142406a9b0adSLois Curfman McInnes int ierr; 142537fcc0dbSBarry Smith if (!__SNESList) {ierr = SNESRegisterAll(); CHKERRQ(ierr);} 14264b0e389bSBarry Smith if (method) *method = (SNESType) snes->type; 142737fcc0dbSBarry Smith if (name) *name = NRFindName( __SNESList, (int) snes->type ); 14289b94acceSBarry Smith return 0; 14299b94acceSBarry Smith } 14309b94acceSBarry Smith 14319b94acceSBarry Smith #include <stdio.h> 14329b94acceSBarry Smith /* 14334b0e389bSBarry Smith SNESPrintTypes_Private - Prints the SNES methods available from the 14349b94acceSBarry Smith options database. 14359b94acceSBarry Smith 14369b94acceSBarry Smith Input Parameters: 14379b94acceSBarry Smith . prefix - prefix (usually "-") 14384b0e389bSBarry Smith . name - the options database name (by default "snes_type") 14399b94acceSBarry Smith */ 14404b0e389bSBarry Smith int SNESPrintTypes_Private(char* prefix,char *name) 14419b94acceSBarry Smith { 14429b94acceSBarry Smith FuncList *entry; 144337fcc0dbSBarry Smith if (!__SNESList) {SNESRegisterAll();} 144437fcc0dbSBarry Smith entry = __SNESList->head; 14459b94acceSBarry Smith fprintf(stderr," %s%s (one of)",prefix,name); 14469b94acceSBarry Smith while (entry) { 14479b94acceSBarry Smith fprintf(stderr," %s",entry->name); 14489b94acceSBarry Smith entry = entry->next; 14499b94acceSBarry Smith } 14509b94acceSBarry Smith fprintf(stderr,"\n"); 14519b94acceSBarry Smith return 0; 14529b94acceSBarry Smith } 14539b94acceSBarry Smith 14549b94acceSBarry Smith /*@C 14559b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 14569b94acceSBarry Smith stored. 14579b94acceSBarry Smith 14589b94acceSBarry Smith Input Parameter: 14599b94acceSBarry Smith . snes - the SNES context 14609b94acceSBarry Smith 14619b94acceSBarry Smith Output Parameter: 14629b94acceSBarry Smith . x - the solution 14639b94acceSBarry Smith 14649b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 14659b94acceSBarry Smith 14669b94acceSBarry Smith .seealso: SNESSetSolution(), SNESGetFunction(), SNESGetSolutionUpdate() 14679b94acceSBarry Smith @*/ 14689b94acceSBarry Smith int SNESGetSolution(SNES snes,Vec *x) 14699b94acceSBarry Smith { 14709b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 14719b94acceSBarry Smith *x = snes->vec_sol_always; 14729b94acceSBarry Smith return 0; 14739b94acceSBarry Smith } 14749b94acceSBarry Smith 14759b94acceSBarry Smith /*@C 14769b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 14779b94acceSBarry Smith stored. 14789b94acceSBarry Smith 14799b94acceSBarry Smith Input Parameter: 14809b94acceSBarry Smith . snes - the SNES context 14819b94acceSBarry Smith 14829b94acceSBarry Smith Output Parameter: 14839b94acceSBarry Smith . x - the solution update 14849b94acceSBarry Smith 14859b94acceSBarry Smith Notes: 14869b94acceSBarry Smith This vector is implementation dependent. 14879b94acceSBarry Smith 14889b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 14899b94acceSBarry Smith 14909b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction 14919b94acceSBarry Smith @*/ 14929b94acceSBarry Smith int SNESGetSolutionUpdate(SNES snes,Vec *x) 14939b94acceSBarry Smith { 14949b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 14959b94acceSBarry Smith *x = snes->vec_sol_update_always; 14969b94acceSBarry Smith return 0; 14979b94acceSBarry Smith } 14989b94acceSBarry Smith 14999b94acceSBarry Smith /*@C 15009b94acceSBarry Smith SNESGetFunction - Returns the vector where the function is 15019b94acceSBarry Smith stored. Actually usually returns the vector where the negative of 15029b94acceSBarry Smith the function is stored. 15039b94acceSBarry Smith 15049b94acceSBarry Smith Input Parameter: 15059b94acceSBarry Smith . snes - the SNES context 15069b94acceSBarry Smith 15079b94acceSBarry Smith Output Parameter: 15089b94acceSBarry Smith . r - the function (or its negative) 15099b94acceSBarry Smith 15109b94acceSBarry Smith Notes: 15119b94acceSBarry Smith SNESGetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only 15129b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 15139b94acceSBarry Smith SNESGetMinimizationFunction() and SNESGetGradient(); 15149b94acceSBarry Smith 15159b94acceSBarry Smith .keywords: SNES, nonlinear, get function 15169b94acceSBarry Smith 15179b94acceSBarry Smith .seealso: SNESSetFunction(), SNESGetSolution() 15189b94acceSBarry Smith @*/ 15199b94acceSBarry Smith int SNESGetFunction(SNES snes,Vec *r) 15209b94acceSBarry Smith { 15219b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 15229b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 152348d91487SBarry Smith "SNESGetFunction:For SNES_NONLINEAR_EQUATIONS only"); 15249b94acceSBarry Smith *r = snes->vec_func_always; 15259b94acceSBarry Smith return 0; 15269b94acceSBarry Smith } 15279b94acceSBarry Smith 15289b94acceSBarry Smith /*@C 15299b94acceSBarry Smith SNESGetGradient - Returns the vector where the gradient is 15309b94acceSBarry Smith stored. Actually usually returns the vector where the negative of 15319b94acceSBarry Smith the function is stored. 15329b94acceSBarry Smith 15339b94acceSBarry Smith Input Parameter: 15349b94acceSBarry Smith . snes - the SNES context 15359b94acceSBarry Smith 15369b94acceSBarry Smith Output Parameter: 15379b94acceSBarry Smith . r - the gradient 15389b94acceSBarry Smith 15399b94acceSBarry Smith Notes: 15409b94acceSBarry Smith SNESGetGradient() is valid for SNES_UNCONSTRAINED_MINIMIZATION methods 15419b94acceSBarry Smith only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 15429b94acceSBarry Smith SNESGetFunction(). 15439b94acceSBarry Smith 15449b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient 15459b94acceSBarry Smith 15469b94acceSBarry Smith .seealso: SNESGetMinimizationFunction(), SNESGetSolution() 15479b94acceSBarry Smith @*/ 15489b94acceSBarry Smith int SNESGetGradient(SNES snes,Vec *r) 15499b94acceSBarry Smith { 15509b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 15519b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 155248d91487SBarry Smith "SNESGetGradient:For SNES_UNCONSTRAINED_MINIMIZATION only"); 15539b94acceSBarry Smith *r = snes->vec_func_always; 15549b94acceSBarry Smith return 0; 15559b94acceSBarry Smith } 15569b94acceSBarry Smith 15579b94acceSBarry Smith /*@ 15589b94acceSBarry Smith SNESGetMinimizationFunction - Returns the scalar function value for 15599b94acceSBarry Smith unconstrained minimization problems. 15609b94acceSBarry Smith 15619b94acceSBarry Smith Input Parameter: 15629b94acceSBarry Smith . snes - the SNES context 15639b94acceSBarry Smith 15649b94acceSBarry Smith Output Parameter: 15659b94acceSBarry Smith . r - the function 15669b94acceSBarry Smith 15679b94acceSBarry Smith Notes: 15689b94acceSBarry Smith SNESGetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 15699b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 15709b94acceSBarry Smith SNESGetFunction(). 15719b94acceSBarry Smith 15729b94acceSBarry Smith .keywords: SNES, nonlinear, get, function 15739b94acceSBarry Smith 15749b94acceSBarry Smith .seealso: SNESGetGradient(), SNESGetSolution() 15759b94acceSBarry Smith @*/ 15769b94acceSBarry Smith int SNESGetMinimizationFunction(SNES snes,double *r) 15779b94acceSBarry Smith { 15789b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 15799b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 158048d91487SBarry Smith "SNESGetMinimizationFunction:For SNES_UNCONSTRAINED_MINIMIZATION only"); 15819b94acceSBarry Smith *r = snes->fc; 15829b94acceSBarry Smith return 0; 15839b94acceSBarry Smith } 15849b94acceSBarry Smith 15859b94acceSBarry Smith 15869b94acceSBarry Smith 15879b94acceSBarry Smith 15889b94acceSBarry Smith 1589