19b94acceSBarry Smith #ifndef lint 2*c4fc05e7SBarry Smith static char vcid[] = "$Id: snes.c,v 1.49 1996/02/01 18:54:09 curfman 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 11052efed2SBarry Smith extern int SNESGetTypeFromOptions_Private(SNES,SNESType*,int*); 124b0e389bSBarry Smith extern int SNESPrintTypes_Private(char*,char*); 139b94acceSBarry Smith 149b94acceSBarry Smith /*@ 159b94acceSBarry Smith SNESView - Prints the SNES data structure. 169b94acceSBarry Smith 179b94acceSBarry Smith Input Parameters: 189b94acceSBarry Smith . SNES - the SNES context 199b94acceSBarry Smith . viewer - visualization context 209b94acceSBarry Smith 219b94acceSBarry Smith Options Database Key: 229b94acceSBarry Smith $ -snes_view : calls SNESView() at end of SNESSolve() 239b94acceSBarry Smith 249b94acceSBarry Smith Notes: 259b94acceSBarry Smith The available visualization contexts include 269b94acceSBarry Smith $ STDOUT_VIEWER_SELF - standard output (default) 279b94acceSBarry Smith $ STDOUT_VIEWER_WORLD - synchronized standard 289b94acceSBarry Smith $ output where only the first processor opens 299b94acceSBarry Smith $ the file. All other processors send their 309b94acceSBarry Smith $ data to the first processor to print. 319b94acceSBarry Smith 329b94acceSBarry Smith The user can open alternative vistualization contexts with 33dbb450caSBarry Smith $ ViewerFileOpenASCII() - output to a specified file 349b94acceSBarry Smith 359b94acceSBarry Smith .keywords: SNES, view 369b94acceSBarry Smith 37dbb450caSBarry Smith .seealso: ViewerFileOpenASCII() 389b94acceSBarry Smith @*/ 399b94acceSBarry Smith int SNESView(SNES snes,Viewer viewer) 409b94acceSBarry Smith { 419b94acceSBarry Smith PetscObject vobj = (PetscObject) viewer; 429b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 439b94acceSBarry Smith FILE *fd; 449b94acceSBarry Smith int ierr; 459b94acceSBarry Smith SLES sles; 469b94acceSBarry Smith char *method; 479b94acceSBarry Smith 489b94acceSBarry Smith if (vobj->cookie == VIEWER_COOKIE && (vobj->type == ASCII_FILE_VIEWER || 499b94acceSBarry Smith vobj->type == ASCII_FILES_VIEWER)) { 50227d817aSBarry Smith ierr = ViewerFileGetPointer(viewer,&fd); CHKERRQ(ierr); 519b94acceSBarry Smith MPIU_fprintf(snes->comm,fd,"SNES Object:\n"); 524b0e389bSBarry Smith SNESGetType(snes,PETSC_NULL,&method); 539b94acceSBarry Smith MPIU_fprintf(snes->comm,fd," method: %s\n",method); 549b94acceSBarry Smith if (snes->view) (*snes->view)((PetscObject)snes,viewer); 559b94acceSBarry Smith MPIU_fprintf(snes->comm,fd, 569b94acceSBarry Smith " maximum iterations=%d, maximum function evaluations=%d\n", 579b94acceSBarry Smith snes->max_its,snes->max_funcs); 589b94acceSBarry Smith MPIU_fprintf(snes->comm,fd, 599b94acceSBarry Smith " tolerances: relative=%g, absolute=%g, truncation=%g, solution=%g\n", 609b94acceSBarry Smith snes->rtol, snes->atol, snes->trunctol, snes->xtol); 619b94acceSBarry Smith if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) 629b94acceSBarry Smith MPIU_fprintf(snes->comm,fd," min function tolerance=%g\n",snes->fmin); 639b94acceSBarry Smith if (snes->ksp_ewconv) { 649b94acceSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 659b94acceSBarry Smith if (kctx) { 669b94acceSBarry Smith MPIU_fprintf(snes->comm,fd, 679b94acceSBarry Smith " Eisenstat-Walker computation of KSP relative tolerance (version %d)\n", 689b94acceSBarry Smith kctx->version); 699b94acceSBarry Smith MPIU_fprintf(snes->comm,fd, 709b94acceSBarry Smith " rtol_0=%g, rtol_max=%g, threshold=%g\n",kctx->rtol_0, 719b94acceSBarry Smith kctx->rtol_max,kctx->threshold); 729b94acceSBarry Smith MPIU_fprintf(snes->comm,fd," gamma=%g, alpha=%g, alpha2=%g\n", 739b94acceSBarry Smith kctx->gamma,kctx->alpha,kctx->alpha2); 749b94acceSBarry Smith } 759b94acceSBarry Smith } 769b94acceSBarry Smith SNESGetSLES(snes,&sles); 779b94acceSBarry Smith ierr = SLESView(sles,viewer); CHKERRQ(ierr); 789b94acceSBarry Smith } 799b94acceSBarry Smith return 0; 809b94acceSBarry Smith } 819b94acceSBarry Smith 829b94acceSBarry Smith /*@ 83682d7d0cSBarry Smith SNESSetFromOptions - Sets various SNES and SLES parameters from user options. 849b94acceSBarry Smith 859b94acceSBarry Smith Input Parameter: 869b94acceSBarry Smith . snes - the SNES context 879b94acceSBarry Smith 889b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 899b94acceSBarry Smith 909b94acceSBarry Smith .seealso: SNESPrintHelp() 919b94acceSBarry Smith @*/ 929b94acceSBarry Smith int SNESSetFromOptions(SNES snes) 939b94acceSBarry Smith { 944b0e389bSBarry Smith SNESType method; 959b94acceSBarry Smith double tmp; 969b94acceSBarry Smith SLES sles; 973c7409f5SSatish Balay int ierr, flg; 983c7409f5SSatish Balay int version = PETSC_DEFAULT; 999b94acceSBarry Smith double rtol_0 = PETSC_DEFAULT; 1009b94acceSBarry Smith double rtol_max = PETSC_DEFAULT; 1019b94acceSBarry Smith double gamma2 = PETSC_DEFAULT; 1029b94acceSBarry Smith double alpha = PETSC_DEFAULT; 1039b94acceSBarry Smith double alpha2 = PETSC_DEFAULT; 1049b94acceSBarry Smith double threshold = PETSC_DEFAULT; 1059b94acceSBarry Smith 1069b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 1074b0e389bSBarry Smith if (snes->setup_called)SETERRQ(1,"SNESSetFromOptions:Must call prior to SNESSetUp!"); 108052efed2SBarry Smith ierr = SNESGetTypeFromOptions_Private(snes,&method,&flg); CHKERRQ(ierr); 109052efed2SBarry Smith if (flg) { 110052efed2SBarry Smith ierr = SNESSetType(snes,method); CHKERRQ(ierr); 1119b94acceSBarry Smith } 1125334005bSBarry Smith else if (!snes->set_method_called) { 1135334005bSBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1145334005bSBarry Smith ierr = SNESSetType(snes,SNES_EQ_NLS); CHKERRQ(ierr); 1155334005bSBarry Smith } 1165334005bSBarry Smith else { 1175334005bSBarry Smith ierr = SNESSetType(snes,SNES_UM_NTR); CHKERRQ(ierr); 1185334005bSBarry Smith } 1195334005bSBarry Smith } 1205334005bSBarry Smith 1213c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 1223c7409f5SSatish Balay if (flg) { SNESPrintHelp(snes); } 1233c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_stol",&tmp, &flg); CHKERRQ(ierr); 1243c7409f5SSatish Balay if (flg) { SNESSetSolutionTolerance(snes,tmp); } 1253c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ttol",&tmp, &flg); CHKERRQ(ierr); 1263c7409f5SSatish Balay if (flg) { SNESSetTruncationTolerance(snes,tmp); } 1273c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_atol",&tmp, &flg); CHKERRQ(ierr); 1283c7409f5SSatish Balay if (flg) { SNESSetAbsoluteTolerance(snes,tmp); } 1293c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_trtol",&tmp, &flg); CHKERRQ(ierr); 1303c7409f5SSatish Balay if (flg) { SNESSetTrustRegionTolerance(snes,tmp); } 1313c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_rtol",&tmp, &flg); CHKERRQ(ierr); 1323c7409f5SSatish Balay if (flg) { SNESSetRelativeTolerance(snes,tmp); } 1333c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_fmin",&tmp, &flg); CHKERRQ(ierr); 1343c7409f5SSatish Balay if (flg) { SNESSetMinFunctionTolerance(snes,tmp); } 1353c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_it",&snes->max_its, &flg); CHKERRQ(ierr); 1363c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_funcs",&snes->max_funcs, &flg);CHKERRQ(ierr); 1373c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_ksp_ew_conv", &flg); CHKERRQ(ierr); 1383c7409f5SSatish Balay if (flg) { snes->ksp_ewconv = 1; } 1393c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_ksp_ew_version",&version, \ 1403c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1413c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtol0",&rtol_0, \ 1423c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1433c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtolmax",&rtol_max, \ 1443c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1453c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_gamma",&gamma2, \ 1463c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1473c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha",&alpha, \ 1483c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1493c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha2",&alpha2, \ 1503c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1513c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_threshold",&threshold, \ 1523c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1539b94acceSBarry Smith ierr = SNES_KSP_SetParametersEW(snes,version,rtol_0,rtol_max,gamma2,alpha, 1549b94acceSBarry Smith alpha2,threshold); CHKERRQ(ierr); 1553c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_monitor", &flg); CHKERRQ(ierr); 1563c7409f5SSatish Balay if (flg) { SNESSetMonitor(snes,SNESDefaultMonitor,0); } 1573c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_smonitor", &flg); CHKERRQ(ierr); 1583c7409f5SSatish Balay if (flg) { SNESSetMonitor(snes,SNESDefaultSMonitor,0); } 1593c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_xmonitor", &flg); CHKERRQ(ierr); 1603c7409f5SSatish Balay if (flg) { 16117699dbbSLois Curfman McInnes int rank = 0; 162d7e8b826SBarry Smith DrawLG lg; 16317699dbbSLois Curfman McInnes MPI_Initialized(&rank); 16417699dbbSLois Curfman McInnes if (rank) MPI_Comm_rank(snes->comm,&rank); 16517699dbbSLois Curfman McInnes if (!rank) { 1669b94acceSBarry Smith ierr = SNESLGMonitorCreate(0,0,0,0,300,300,&lg); CHKERRQ(ierr); 1679b94acceSBarry Smith ierr = SNESSetMonitor(snes,SNESLGMonitor,(void *)lg); CHKERRQ(ierr); 1689b94acceSBarry Smith PLogObjectParent(snes,lg); 1699b94acceSBarry Smith } 1709b94acceSBarry Smith } 1713c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_fd", &flg); CHKERRQ(ierr); 1723c7409f5SSatish Balay if( flg && snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1739b94acceSBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre, 1749b94acceSBarry Smith SNESDefaultComputeJacobian,snes->funP); CHKERRQ(ierr); 1759b94acceSBarry Smith } 1769b94acceSBarry Smith ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 1779b94acceSBarry Smith ierr = SLESSetFromOptions(sles); CHKERRQ(ierr); 1789b94acceSBarry Smith if (!snes->setfromoptions) return 0; 1799b94acceSBarry Smith return (*snes->setfromoptions)(snes); 1809b94acceSBarry Smith } 1819b94acceSBarry Smith 1829b94acceSBarry Smith /*@ 1839b94acceSBarry Smith SNESPrintHelp - Prints all options for the SNES component. 1849b94acceSBarry Smith 1859b94acceSBarry Smith Input Parameter: 1869b94acceSBarry Smith . snes - the SNES context 1879b94acceSBarry Smith 188a703fe33SLois Curfman McInnes Options Database Keys: 189a703fe33SLois Curfman McInnes $ -help, -h 190a703fe33SLois Curfman McInnes 1919b94acceSBarry Smith .keywords: SNES, nonlinear, help 1929b94acceSBarry Smith 193682d7d0cSBarry Smith .seealso: SNESSetFromOptions() 1949b94acceSBarry Smith @*/ 1959b94acceSBarry Smith int SNESPrintHelp(SNES snes) 1969b94acceSBarry Smith { 1976daaf66cSBarry Smith char p[64]; 1986daaf66cSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 1996daaf66cSBarry Smith 2009b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 2016daaf66cSBarry Smith 2026daaf66cSBarry Smith PetscStrcpy(p,"-"); 2036daaf66cSBarry Smith if (snes->prefix) PetscStrcat(p, snes->prefix); 2046daaf66cSBarry Smith 2056daaf66cSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 2066daaf66cSBarry Smith 2079b94acceSBarry Smith MPIU_printf(snes->comm,"SNES options ----------------------------\n"); 2086daaf66cSBarry Smith SNESPrintTypes_Private(p,"snes_type"); 2096daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_monitor: use default SNES monitor\n",p); 2106daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_view: view SNES info after each nonlinear solve\n",p); 2116daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_max_it its (default %d)\n",p,snes->max_its); 2126daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_stol tol (default %g)\n",p,snes->xtol); 2136daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_atol tol (default %g)\n",p,snes->atol); 2146daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_rtol tol (default %g)\n",p,snes->rtol); 2156daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_ttol tol (default %g)\n",p,snes->trunctol); 2169b94acceSBarry Smith MPIU_printf(snes->comm, 2179b94acceSBarry Smith " options for solving systems of nonlinear equations only:\n"); 2186daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_fd: use finite differences for Jacobian\n",p); 2196daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_mf: use matrix-free Jacobian\n",p); 2206daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_ksp_ew_conv: use Eisenstat-Walker computation of KSP rtol. Params are:\n",p); 2219b94acceSBarry Smith MPIU_printf(snes->comm, 2229b94acceSBarry Smith " %ssnes_ksp_ew_version version (1 or 2, default is %d)\n", 2236daaf66cSBarry Smith p,kctx->version); 2249b94acceSBarry Smith MPIU_printf(snes->comm, 2259b94acceSBarry Smith " %ssnes_ksp_ew_rtol0 rtol0 (0 <= rtol0 < 1, default %g)\n", 2266daaf66cSBarry Smith p,kctx->rtol_0); 2279b94acceSBarry Smith MPIU_printf(snes->comm, 2289b94acceSBarry Smith " %ssnes_ksp_ew_rtolmax rtolmax (0 <= rtolmax < 1, default %g)\n", 2296daaf66cSBarry Smith p,kctx->rtol_max); 2309b94acceSBarry Smith MPIU_printf(snes->comm, 2319b94acceSBarry Smith " %ssnes_ksp_ew_gamma gamma (0 <= gamma <= 1, default %g)\n", 2326daaf66cSBarry Smith p,kctx->gamma); 2339b94acceSBarry Smith MPIU_printf(snes->comm, 2349b94acceSBarry Smith " %ssnes_ksp_ew_alpha alpha (1 < alpha <= 2, default %g)\n", 2356daaf66cSBarry Smith p,kctx->alpha); 2369b94acceSBarry Smith MPIU_printf(snes->comm, 2379b94acceSBarry Smith " %ssnes_ksp_ew_alpha2 alpha2 (default %g)\n", 2386daaf66cSBarry Smith p,kctx->alpha2); 2399b94acceSBarry Smith MPIU_printf(snes->comm, 2409b94acceSBarry Smith " %ssnes_ksp_ew_threshold threshold (0 < threshold < 1, default %g)\n", 2416daaf66cSBarry Smith p,kctx->threshold); 2429b94acceSBarry Smith MPIU_printf(snes->comm, 2439b94acceSBarry Smith " options for solving unconstrained minimization problems only:\n"); 2446daaf66cSBarry Smith MPIU_printf(snes->comm," %ssnes_fmin tol (default %g)\n",p,snes->fmin); 2456daaf66cSBarry Smith MPIU_printf(snes->comm," Run program with %ssnes_type method -help for help on ",p); 2469b94acceSBarry Smith MPIU_printf(snes->comm,"a particular method\n"); 2476daaf66cSBarry Smith if (snes->printhelp) (*snes->printhelp)(snes,p); 2489b94acceSBarry Smith return 0; 2499b94acceSBarry Smith } 2509b94acceSBarry Smith /*@ 2519b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 2529b94acceSBarry Smith the nonlinear solvers. 2539b94acceSBarry Smith 2549b94acceSBarry Smith Input Parameters: 2559b94acceSBarry Smith . snes - the SNES context 2569b94acceSBarry Smith . usrP - optional user context 2579b94acceSBarry Smith 2589b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 2599b94acceSBarry Smith 2609b94acceSBarry Smith .seealso: SNESGetApplicationContext() 2619b94acceSBarry Smith @*/ 2629b94acceSBarry Smith int SNESSetApplicationContext(SNES snes,void *usrP) 2639b94acceSBarry Smith { 2649b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 2659b94acceSBarry Smith snes->user = usrP; 2669b94acceSBarry Smith return 0; 2679b94acceSBarry Smith } 2689b94acceSBarry Smith /*@C 2699b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 2709b94acceSBarry Smith nonlinear solvers. 2719b94acceSBarry Smith 2729b94acceSBarry Smith Input Parameter: 2739b94acceSBarry Smith . snes - SNES context 2749b94acceSBarry Smith 2759b94acceSBarry Smith Output Parameter: 2769b94acceSBarry Smith . usrP - user context 2779b94acceSBarry Smith 2789b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 2799b94acceSBarry Smith 2809b94acceSBarry Smith .seealso: SNESSetApplicationContext() 2819b94acceSBarry Smith @*/ 2829b94acceSBarry Smith int SNESGetApplicationContext( SNES snes, void **usrP ) 2839b94acceSBarry Smith { 2849b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 2859b94acceSBarry Smith *usrP = snes->user; 2869b94acceSBarry Smith return 0; 2879b94acceSBarry Smith } 2889b94acceSBarry Smith /*@ 2899b94acceSBarry Smith SNESGetIterationNumber - Gets the current iteration number of the 2909b94acceSBarry Smith nonlinear solver. 2919b94acceSBarry Smith 2929b94acceSBarry Smith Input Parameter: 2939b94acceSBarry Smith . snes - SNES context 2949b94acceSBarry Smith 2959b94acceSBarry Smith Output Parameter: 2969b94acceSBarry Smith . iter - iteration number 2979b94acceSBarry Smith 2989b94acceSBarry Smith .keywords: SNES, nonlinear, get, iteration, number 2999b94acceSBarry Smith @*/ 3009b94acceSBarry Smith int SNESGetIterationNumber(SNES snes,int* iter) 3019b94acceSBarry Smith { 3029b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3039b94acceSBarry Smith *iter = snes->iter; 3049b94acceSBarry Smith return 0; 3059b94acceSBarry Smith } 3069b94acceSBarry Smith /*@ 3079b94acceSBarry Smith SNESGetFunctionNorm - Gets the norm of the current function that was set 3089b94acceSBarry Smith with SNESSSetFunction(). 3099b94acceSBarry Smith 3109b94acceSBarry Smith Input Parameter: 3119b94acceSBarry Smith . snes - SNES context 3129b94acceSBarry Smith 3139b94acceSBarry Smith Output Parameter: 3149b94acceSBarry Smith . fnorm - 2-norm of function 3159b94acceSBarry Smith 3169b94acceSBarry Smith Note: 3179b94acceSBarry Smith SNESGetFunctionNorm() is valid for SNES_NONLINEAR_EQUATIONS methods only. 3189b94acceSBarry Smith 3199b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm 3209b94acceSBarry Smith @*/ 3219b94acceSBarry Smith int SNESGetFunctionNorm(SNES snes,Scalar *fnorm) 3229b94acceSBarry Smith { 3239b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3249b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 32548d91487SBarry Smith "SNESGetFunctionNorm:For SNES_NONLINEAR_EQUATIONS only"); 3269b94acceSBarry Smith *fnorm = snes->norm; 3279b94acceSBarry Smith return 0; 3289b94acceSBarry Smith } 3299b94acceSBarry Smith /*@ 3309b94acceSBarry Smith SNESGetGradientNorm - Gets the norm of the current gradient that was set 3319b94acceSBarry Smith with SNESSSetGradient(). 3329b94acceSBarry Smith 3339b94acceSBarry Smith Input Parameter: 3349b94acceSBarry Smith . snes - SNES context 3359b94acceSBarry Smith 3369b94acceSBarry Smith Output Parameter: 3379b94acceSBarry Smith . fnorm - 2-norm of gradient 3389b94acceSBarry Smith 3399b94acceSBarry Smith Note: 3409b94acceSBarry Smith SNESGetGradientNorm() is valid for SNES_UNCONSTRAINED_MINIMIZATION 3419b94acceSBarry Smith methods only. 3429b94acceSBarry Smith 3439b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient, norm 3449b94acceSBarry Smith @*/ 3459b94acceSBarry Smith int SNESGetGradientNorm(SNES snes,Scalar *gnorm) 3469b94acceSBarry Smith { 3479b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3489b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 34948d91487SBarry Smith "SNESGetGradientNorm:For SNES_UNCONSTRAINED_MINIMIZATION only"); 3509b94acceSBarry Smith *gnorm = snes->norm; 3519b94acceSBarry Smith return 0; 3529b94acceSBarry Smith } 3539b94acceSBarry Smith /*@ 3549b94acceSBarry Smith SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps 3559b94acceSBarry Smith attempted by the nonlinear solver. 3569b94acceSBarry Smith 3579b94acceSBarry Smith Input Parameter: 3589b94acceSBarry Smith . snes - SNES context 3599b94acceSBarry Smith 3609b94acceSBarry Smith Output Parameter: 3619b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 3629b94acceSBarry Smith 3639b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 3649b94acceSBarry Smith @*/ 3659b94acceSBarry Smith int SNESGetNumberUnsuccessfulSteps(SNES snes,int* nfails) 3669b94acceSBarry Smith { 3679b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3689b94acceSBarry Smith *nfails = snes->nfailures; 3699b94acceSBarry Smith return 0; 3709b94acceSBarry Smith } 3719b94acceSBarry Smith /*@C 3729b94acceSBarry Smith SNESGetSLES - Returns the SLES context for a SNES solver. 3739b94acceSBarry Smith 3749b94acceSBarry Smith Input Parameter: 3759b94acceSBarry Smith . snes - the SNES context 3769b94acceSBarry Smith 3779b94acceSBarry Smith Output Parameter: 3789b94acceSBarry Smith . sles - the SLES context 3799b94acceSBarry Smith 3809b94acceSBarry Smith Notes: 3819b94acceSBarry Smith The user can then directly manipulate the SLES context to set various 3829b94acceSBarry Smith options, etc. Likewise, the user can then extract and manipulate the 3839b94acceSBarry Smith KSP and PC contexts as well. 3849b94acceSBarry Smith 3859b94acceSBarry Smith .keywords: SNES, nonlinear, get, SLES, context 3869b94acceSBarry Smith 3879b94acceSBarry Smith .seealso: SLESGetPC(), SLESGetKSP() 3889b94acceSBarry Smith @*/ 3899b94acceSBarry Smith int SNESGetSLES(SNES snes,SLES *sles) 3909b94acceSBarry Smith { 3919b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 3929b94acceSBarry Smith *sles = snes->sles; 3939b94acceSBarry Smith return 0; 3949b94acceSBarry Smith } 3959b94acceSBarry Smith /* -----------------------------------------------------------*/ 3969b94acceSBarry Smith /*@C 3979b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 3989b94acceSBarry Smith 3999b94acceSBarry Smith Input Parameter: 4009b94acceSBarry Smith . comm - MPI communicator 4019b94acceSBarry Smith . type - type of method, one of 4029b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS 4039b94acceSBarry Smith $ (for systems of nonlinear equations) 4049b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION 4059b94acceSBarry Smith $ (for unconstrained minimization) 4069b94acceSBarry Smith 4079b94acceSBarry Smith Output Parameter: 4089b94acceSBarry Smith . outsnes - the new SNES context 4099b94acceSBarry Smith 4109b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 4119b94acceSBarry Smith 41263a78c88SLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy() 4139b94acceSBarry Smith @*/ 4144b0e389bSBarry Smith int SNESCreate(MPI_Comm comm,SNESProblemType type,SNES *outsnes) 4159b94acceSBarry Smith { 4169b94acceSBarry Smith int ierr; 4179b94acceSBarry Smith SNES snes; 4189b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 41937fcc0dbSBarry Smith 4209b94acceSBarry Smith *outsnes = 0; 4212a0acf01SLois Curfman McInnes if (type != SNES_UNCONSTRAINED_MINIMIZATION && type != SNES_NONLINEAR_EQUATIONS) 4222a0acf01SLois Curfman McInnes SETERRQ(1,"SNESCreate:incorrect method type"); 4230452661fSBarry Smith PetscHeaderCreate(snes,_SNES,SNES_COOKIE,SNES_UNKNOWN_METHOD,comm); 4249b94acceSBarry Smith PLogObjectCreate(snes); 4259b94acceSBarry Smith snes->max_its = 50; 4269b94acceSBarry Smith snes->max_funcs = 1000; 4279b94acceSBarry Smith snes->norm = 0.0; 4285a2d0531SBarry Smith if (type == SNES_UNCONSTRAINED_MINIMIZATION) { 4292a0acf01SLois Curfman McInnes snes->rtol = 1.e-08; 4309b94acceSBarry Smith snes->atol = 1.e-10; 4315a2d0531SBarry Smith } 432b4874afaSBarry Smith else { 433b4874afaSBarry Smith snes->rtol = 1.e-8; 434b4874afaSBarry Smith snes->ttol = 0.0; 435b4874afaSBarry Smith snes->atol = 1.e-50; 436b4874afaSBarry Smith } 4379b94acceSBarry Smith snes->xtol = 1.e-8; 4389b94acceSBarry Smith snes->trunctol = 1.e-12; 4399b94acceSBarry Smith snes->nfuncs = 0; 4409b94acceSBarry Smith snes->nfailures = 0; 4419b94acceSBarry Smith snes->monitor = 0; 4429b94acceSBarry Smith snes->data = 0; 4439b94acceSBarry Smith snes->view = 0; 4449b94acceSBarry Smith snes->computeumfunction = 0; 4459b94acceSBarry Smith snes->umfunP = 0; 4469b94acceSBarry Smith snes->fc = 0; 4479b94acceSBarry Smith snes->deltatol = 1.e-12; 4489b94acceSBarry Smith snes->fmin = -1.e30; 4499b94acceSBarry Smith snes->method_class = type; 4509b94acceSBarry Smith snes->set_method_called = 0; 451a703fe33SLois Curfman McInnes snes->setup_called = 0; 4529b94acceSBarry Smith snes->ksp_ewconv = 0; 4539b94acceSBarry Smith 4549b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 4550452661fSBarry Smith kctx = PetscNew(SNES_KSP_EW_ConvCtx); CHKPTRQ(kctx); 4569b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 4579b94acceSBarry Smith kctx->version = 2; 4589b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 4599b94acceSBarry Smith this was too large for some test cases */ 4609b94acceSBarry Smith kctx->rtol_last = 0; 4619b94acceSBarry Smith kctx->rtol_max = .9; 4629b94acceSBarry Smith kctx->gamma = 1.0; 4639b94acceSBarry Smith kctx->alpha2 = .5*(1.0 + sqrt(5.0)); 4649b94acceSBarry Smith kctx->alpha = kctx->alpha2; 4659b94acceSBarry Smith kctx->threshold = .1; 4669b94acceSBarry Smith kctx->lresid_last = 0; 4679b94acceSBarry Smith kctx->norm_last = 0; 4689b94acceSBarry Smith 4699b94acceSBarry Smith ierr = SLESCreate(comm,&snes->sles); CHKERRQ(ierr); 4709b94acceSBarry Smith PLogObjectParent(snes,snes->sles) 4715334005bSBarry Smith 4729b94acceSBarry Smith *outsnes = snes; 4739b94acceSBarry Smith return 0; 4749b94acceSBarry Smith } 4759b94acceSBarry Smith 4769b94acceSBarry Smith /* --------------------------------------------------------------- */ 4779b94acceSBarry Smith /*@C 4789b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 4799b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 4809b94acceSBarry Smith equations. 4819b94acceSBarry Smith 4829b94acceSBarry Smith Input Parameters: 4839b94acceSBarry Smith . snes - the SNES context 4849b94acceSBarry Smith . func - function evaluation routine 4859b94acceSBarry Smith . ctx - optional user-defined function context 4869b94acceSBarry Smith . r - vector to store function value 4879b94acceSBarry Smith 4889b94acceSBarry Smith Calling sequence of func: 4899b94acceSBarry Smith . func (SNES, Vec x, Vec f, void *ctx); 4909b94acceSBarry Smith 4919b94acceSBarry Smith . x - input vector 4925334005bSBarry Smith . f - vector function 4939b94acceSBarry Smith . ctx - optional user-defined context for private data for the 4949b94acceSBarry Smith function evaluation routine (may be null) 4959b94acceSBarry Smith 4969b94acceSBarry Smith Notes: 4979b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 4989b94acceSBarry Smith $ f'(x) x = -f(x), 4999b94acceSBarry Smith $ where f'(x) denotes the Jacobian matrix and f(x) is the function. 5009b94acceSBarry Smith 5019b94acceSBarry Smith SNESSetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 5029b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 5039b94acceSBarry Smith SNESSetMinimizationFunction() and SNESSetGradient(); 5049b94acceSBarry Smith 5059b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 5069b94acceSBarry Smith 5078ddd3da0SLois Curfman McInnes .seealso: SNESGetFunction(), SNESSetJacobian() 5089b94acceSBarry Smith @*/ 5095334005bSBarry Smith int SNESSetFunction( SNES snes, Vec r, int (*func)(SNES,Vec,Vec,void*),void *ctx) 5109b94acceSBarry Smith { 5119b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 5129b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 51348d91487SBarry Smith "SNESSetFunction:For SNES_NONLINEAR_EQUATIONS only"); 5149b94acceSBarry Smith snes->computefunction = func; 5159b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 5169b94acceSBarry Smith snes->funP = ctx; 5179b94acceSBarry Smith return 0; 5189b94acceSBarry Smith } 5199b94acceSBarry Smith 5209b94acceSBarry Smith /*@ 5219b94acceSBarry Smith SNESComputeFunction - Computes the function that has been set with 5229b94acceSBarry Smith SNESSetFunction(). 5239b94acceSBarry Smith 5249b94acceSBarry Smith Input Parameters: 5259b94acceSBarry Smith . snes - the SNES context 5269b94acceSBarry Smith . x - input vector 5279b94acceSBarry Smith 5289b94acceSBarry Smith Output Parameter: 5293638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 5309b94acceSBarry Smith 5319b94acceSBarry Smith Notes: 5329b94acceSBarry Smith SNESComputeFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 5339b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 5349b94acceSBarry Smith SNESComputeMinimizationFunction() and SNESComputeGradient(); 5359b94acceSBarry Smith 5369b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 5379b94acceSBarry Smith 5389b94acceSBarry Smith .seealso: SNESSetFunction() 5399b94acceSBarry Smith @*/ 5409b94acceSBarry Smith int SNESComputeFunction(SNES snes,Vec x, Vec y) 5419b94acceSBarry Smith { 5429b94acceSBarry Smith int ierr; 5439b94acceSBarry Smith 5449b94acceSBarry Smith PLogEventBegin(SNES_FunctionEval,snes,x,y,0); 5459b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 5469b94acceSBarry Smith PLogEventEnd(SNES_FunctionEval,snes,x,y,0); 5479b94acceSBarry Smith return 0; 5489b94acceSBarry Smith } 5499b94acceSBarry Smith 5509b94acceSBarry Smith /*@C 5519b94acceSBarry Smith SNESSetMinimizationFunction - Sets the function evaluation routine for 5529b94acceSBarry Smith unconstrained minimization. 5539b94acceSBarry Smith 5549b94acceSBarry Smith Input Parameters: 5559b94acceSBarry Smith . snes - the SNES context 5569b94acceSBarry Smith . func - function evaluation routine 5579b94acceSBarry Smith . ctx - optional user-defined function context 5589b94acceSBarry Smith 5599b94acceSBarry Smith Calling sequence of func: 5609b94acceSBarry Smith . func (SNES snes,Vec x,double *f,void *ctx); 5619b94acceSBarry Smith 5629b94acceSBarry Smith . x - input vector 5639b94acceSBarry Smith . f - function 5649b94acceSBarry Smith . ctx - optional user-defined context for private data for the 5659b94acceSBarry Smith function evaluation routine (may be null) 5669b94acceSBarry Smith 5679b94acceSBarry Smith Notes: 5689b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 5699b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 5709b94acceSBarry Smith SNESSetFunction(). 5719b94acceSBarry Smith 5729b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimization, function 5739b94acceSBarry Smith 5749b94acceSBarry Smith .seealso: SNESGetMinimizationFunction(), SNESSetHessian(), SNESSetGradient(), 5759b94acceSBarry Smith @*/ 5769b94acceSBarry Smith int SNESSetMinimizationFunction(SNES snes,int (*func)(SNES,Vec,double*,void*), 5779b94acceSBarry Smith void *ctx) 5789b94acceSBarry Smith { 5799b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 5809b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 58148d91487SBarry Smith "SNESSetMinimizationFunction:Only for SNES_UNCONSTRAINED_MINIMIZATION"); 5829b94acceSBarry Smith snes->computeumfunction = func; 5839b94acceSBarry Smith snes->umfunP = ctx; 5849b94acceSBarry Smith return 0; 5859b94acceSBarry Smith } 5869b94acceSBarry Smith 5879b94acceSBarry Smith /*@ 5889b94acceSBarry Smith SNESComputeMinimizationFunction - Computes the function that has been 5899b94acceSBarry Smith set with SNESSetMinimizationFunction(). 5909b94acceSBarry Smith 5919b94acceSBarry Smith Input Parameters: 5929b94acceSBarry Smith . snes - the SNES context 5939b94acceSBarry Smith . x - input vector 5949b94acceSBarry Smith 5959b94acceSBarry Smith Output Parameter: 5969b94acceSBarry Smith . y - function value 5979b94acceSBarry Smith 5989b94acceSBarry Smith Notes: 5999b94acceSBarry Smith SNESComputeMinimizationFunction() is valid only for 6009b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 6019b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 6029b94acceSBarry Smith @*/ 6039b94acceSBarry Smith int SNESComputeMinimizationFunction(SNES snes,Vec x,double *y) 6049b94acceSBarry Smith { 6059b94acceSBarry Smith int ierr; 6069b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 60748d91487SBarry Smith "SNESComputeMinimizationFunction:Only for SNES_UNCONSTRAINED_MINIMIZATION"); 6089b94acceSBarry Smith PLogEventBegin(SNES_MinimizationFunctionEval,snes,x,y,0); 6099b94acceSBarry Smith ierr = (*snes->computeumfunction)(snes,x,y,snes->umfunP); CHKERRQ(ierr); 6109b94acceSBarry Smith PLogEventEnd(SNES_MinimizationFunctionEval,snes,x,y,0); 6119b94acceSBarry Smith return 0; 6129b94acceSBarry Smith } 6139b94acceSBarry Smith 6149b94acceSBarry Smith /*@C 6159b94acceSBarry Smith SNESSetGradient - Sets the gradient evaluation routine and gradient 6169b94acceSBarry Smith vector for use by the SNES routines. 6179b94acceSBarry Smith 6189b94acceSBarry Smith Input Parameters: 6199b94acceSBarry Smith . snes - the SNES context 6209b94acceSBarry Smith . func - function evaluation routine 6219b94acceSBarry Smith . ctx - optional user-defined function context 6229b94acceSBarry Smith . r - vector to store gradient value 6239b94acceSBarry Smith 6249b94acceSBarry Smith Calling sequence of func: 6259b94acceSBarry Smith . func (SNES, Vec x, Vec g, void *ctx); 6269b94acceSBarry Smith 6279b94acceSBarry Smith . x - input vector 6289b94acceSBarry Smith . g - gradient vector 6299b94acceSBarry Smith . ctx - optional user-defined context for private data for the 6309b94acceSBarry Smith function evaluation routine (may be null) 6319b94acceSBarry Smith 6329b94acceSBarry Smith Notes: 6339b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 6349b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 6359b94acceSBarry Smith SNESSetFunction(). 6369b94acceSBarry Smith 6379b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 6389b94acceSBarry Smith 6399b94acceSBarry Smith .seealso: SNESGetGradient(), SNESSetHessian(), SNESSetMinimizationFunction(), 6409b94acceSBarry Smith @*/ 6419b94acceSBarry Smith int SNESSetGradient(SNES snes,Vec r,int (*func)(SNES,Vec,Vec,void*), 6429b94acceSBarry Smith void *ctx) 6439b94acceSBarry Smith { 6449b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 6459b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 64648d91487SBarry Smith "SNESSetGradient:For SNES_UNCONSTRAINED_MINIMIZATION only"); 6479b94acceSBarry Smith snes->computefunction = func; 6489b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 6499b94acceSBarry Smith snes->funP = ctx; 6509b94acceSBarry Smith return 0; 6519b94acceSBarry Smith } 6529b94acceSBarry Smith 6539b94acceSBarry Smith /*@ 6549b94acceSBarry Smith SNESComputeGradient - Computes the gradient that has been 6559b94acceSBarry Smith set with SNESSetGradient(). 6569b94acceSBarry Smith 6579b94acceSBarry Smith Input Parameters: 6589b94acceSBarry Smith . snes - the SNES context 6599b94acceSBarry Smith . x - input vector 6609b94acceSBarry Smith 6619b94acceSBarry Smith Output Parameter: 6629b94acceSBarry Smith . y - gradient vector 6639b94acceSBarry Smith 6649b94acceSBarry Smith Notes: 6659b94acceSBarry Smith SNESComputeGradient() is valid only for 6669b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 6679b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 6689b94acceSBarry Smith @*/ 6699b94acceSBarry Smith int SNESComputeGradient(SNES snes,Vec x, Vec y) 6709b94acceSBarry Smith { 6719b94acceSBarry Smith int ierr; 6729b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 67348d91487SBarry Smith "SNESComputeGradient:For SNES_UNCONSTRAINED_MINIMIZATION only"); 6749b94acceSBarry Smith PLogEventBegin(SNES_GradientEval,snes,x,y,0); 6759b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 6769b94acceSBarry Smith PLogEventEnd(SNES_GradientEval,snes,x,y,0); 6779b94acceSBarry Smith return 0; 6789b94acceSBarry Smith } 6799b94acceSBarry Smith 6809b94acceSBarry Smith int SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 6819b94acceSBarry Smith { 6829b94acceSBarry Smith int ierr; 6839b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 68448d91487SBarry Smith "SNESComputeJacobian: For SNES_NONLINEAR_EQUATIONS only"); 6859b94acceSBarry Smith if (!snes->computejacobian) return 0; 6869b94acceSBarry Smith PLogEventBegin(SNES_JacobianEval,snes,X,*A,*B); 687*c4fc05e7SBarry Smith *flg = DIFFERENT_NONZERO_PATTERN; 6889b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 6899b94acceSBarry Smith PLogEventEnd(SNES_JacobianEval,snes,X,*A,*B); 6909b94acceSBarry Smith return 0; 6919b94acceSBarry Smith } 6929b94acceSBarry Smith 6939b94acceSBarry Smith int SNESComputeHessian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 6949b94acceSBarry Smith { 6959b94acceSBarry Smith int ierr; 6969b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 69748d91487SBarry Smith "SNESComputeHessian:For SNES_UNCONSTRAINED_MINIMIZATION only"); 6989b94acceSBarry Smith if (!snes->computejacobian) return 0; 6999b94acceSBarry Smith PLogEventBegin(SNES_HessianEval,snes,X,*A,*B); 7009b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 7019b94acceSBarry Smith PLogEventEnd(SNES_HessianEval,snes,X,*A,*B); 7029b94acceSBarry Smith return 0; 7039b94acceSBarry Smith } 7049b94acceSBarry Smith 7059b94acceSBarry Smith /*@C 7069b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 7079b94acceSBarry Smith location to store it. 7089b94acceSBarry Smith 7099b94acceSBarry Smith Input Parameters: 7109b94acceSBarry Smith . snes - the SNES context 7119b94acceSBarry Smith . A - Jacobian matrix 7129b94acceSBarry Smith . B - preconditioner matrix (usually same as the Jacobian) 7139b94acceSBarry Smith . func - Jacobian evaluation routine 7149b94acceSBarry Smith . ctx - optional user-defined context for private data for the 7159b94acceSBarry Smith Jacobian evaluation routine (may be null) 7169b94acceSBarry Smith 7179b94acceSBarry Smith Calling sequence of func: 7189b94acceSBarry Smith . func (SNES,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 7199b94acceSBarry Smith 7209b94acceSBarry Smith . x - input vector 7219b94acceSBarry Smith . A - Jacobian matrix 7229b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 7239b94acceSBarry Smith . flag - flag indicating information about matrix structure, 7249b94acceSBarry Smith same as flag in SLESSetOperators() 7259b94acceSBarry Smith . ctx - optional user-defined Jacobian context 7269b94acceSBarry Smith 7279b94acceSBarry Smith Notes: 7289b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 7299b94acceSBarry Smith This allows the Jacobian evaluation routine to replace A and/or B with a 7309b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 7319b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 7329b94acceSBarry Smith throughout the global iterations. 7339b94acceSBarry Smith 7349b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 7359b94acceSBarry Smith 7368ddd3da0SLois Curfman McInnes .seealso: SNESSetFunction() 7379b94acceSBarry Smith @*/ 7389b94acceSBarry Smith int SNESSetJacobian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 7399b94acceSBarry Smith MatStructure*,void*),void *ctx) 7409b94acceSBarry Smith { 7419b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 7429b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 74348d91487SBarry Smith "SNESSetJacobian:For SNES_NONLINEAR_EQUATIONS only"); 7449b94acceSBarry Smith snes->computejacobian = func; 7459b94acceSBarry Smith snes->jacP = ctx; 7469b94acceSBarry Smith snes->jacobian = A; 7479b94acceSBarry Smith snes->jacobian_pre = B; 7489b94acceSBarry Smith return 0; 7499b94acceSBarry Smith } 750b4fd4287SBarry Smith /*@ 751b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 752b4fd4287SBarry Smith provided context for evaluating the Jacobian. 753b4fd4287SBarry Smith 754b4fd4287SBarry Smith Input Parameter: 755b4fd4287SBarry Smith . snes - the nonlinear solver context 756b4fd4287SBarry Smith 757b4fd4287SBarry Smith Output Parameters: 758b4fd4287SBarry Smith . A - location to stash Jacobian matrix (or PETSC_NULL) 759b4fd4287SBarry Smith . B - location to stash preconditioner matrix (or PETSC_NULL) 760b4fd4287SBarry Smith . ctx - location to stash Jacobian ctx (or PETSC_NULL) 761b4fd4287SBarry Smith 762b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian() 763b4fd4287SBarry Smith @*/ 764b4fd4287SBarry Smith int SNESGetJacobian(SNES snes,Mat *A,Mat *B, void **ctx) 765b4fd4287SBarry Smith { 766b4fd4287SBarry Smith if (A) *A = snes->jacobian; 767b4fd4287SBarry Smith if (B) *B = snes->jacobian_pre; 768b4fd4287SBarry Smith if (ctx) *ctx = snes->jacP; 769b4fd4287SBarry Smith return 0; 770b4fd4287SBarry Smith } 771b4fd4287SBarry Smith 7729b94acceSBarry Smith /*@C 7739b94acceSBarry Smith SNESSetHessian - Sets the function to compute Hessian as well as the 7749b94acceSBarry Smith location to store it. 7759b94acceSBarry Smith 7769b94acceSBarry Smith Input Parameters: 7779b94acceSBarry Smith . snes - the SNES context 7789b94acceSBarry Smith . A - Hessian matrix 7799b94acceSBarry Smith . B - preconditioner matrix (usually same as the Hessian) 7809b94acceSBarry Smith . func - Jacobian evaluation routine 7819b94acceSBarry Smith . ctx - optional user-defined context for private data for the 7829b94acceSBarry Smith Hessian evaluation routine (may be null) 7839b94acceSBarry Smith 7849b94acceSBarry Smith Calling sequence of func: 7859b94acceSBarry Smith . func (SNES,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 7869b94acceSBarry Smith 7879b94acceSBarry Smith . x - input vector 7889b94acceSBarry Smith . A - Hessian matrix 7899b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 7909b94acceSBarry Smith . flag - flag indicating information about matrix structure, 7919b94acceSBarry Smith same as flag in SLESSetOperators() 7929b94acceSBarry Smith . ctx - optional user-defined Hessian context 7939b94acceSBarry Smith 7949b94acceSBarry Smith Notes: 7959b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 7969b94acceSBarry Smith This allows the Hessian evaluation routine to replace A and/or B with a 7979b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 7989b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 7999b94acceSBarry Smith throughout the global iterations. 8009b94acceSBarry Smith 8019b94acceSBarry Smith .keywords: SNES, nonlinear, set, Hessian, matrix 8029b94acceSBarry Smith 8038ddd3da0SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESSetGradient() 8049b94acceSBarry Smith @*/ 8059b94acceSBarry Smith int SNESSetHessian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 8069b94acceSBarry Smith MatStructure*,void*),void *ctx) 8079b94acceSBarry Smith { 8089b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 8099b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 81048d91487SBarry Smith "SNESSetHessian:For SNES_UNCONSTRAINED_MINIMIZATION only"); 8119b94acceSBarry Smith snes->computejacobian = func; 8129b94acceSBarry Smith snes->jacP = ctx; 8139b94acceSBarry Smith snes->jacobian = A; 8149b94acceSBarry Smith snes->jacobian_pre = B; 8159b94acceSBarry Smith return 0; 8169b94acceSBarry Smith } 8179b94acceSBarry Smith 8189b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */ 8199b94acceSBarry Smith 8209b94acceSBarry Smith /*@ 8219b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 822272ac6f2SLois Curfman McInnes of a nonlinear solver. 8239b94acceSBarry Smith 8249b94acceSBarry Smith Input Parameter: 8259b94acceSBarry Smith . snes - the SNES context 8268ddd3da0SLois Curfman McInnes . x - the solution vector 8279b94acceSBarry Smith 828272ac6f2SLois Curfman McInnes Notes: 829272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 830272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 831272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 832272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 833272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 834272ac6f2SLois Curfman McInnes 8359b94acceSBarry Smith .keywords: SNES, nonlinear, setup 8369b94acceSBarry Smith 8379b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 8389b94acceSBarry Smith @*/ 8398ddd3da0SLois Curfman McInnes int SNESSetUp(SNES snes,Vec x) 8409b94acceSBarry Smith { 841272ac6f2SLois Curfman McInnes int ierr, flg; 8429b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 8438ddd3da0SLois Curfman McInnes PETSCVALIDHEADERSPECIFIC(x,VEC_COOKIE); 8448ddd3da0SLois Curfman McInnes snes->vec_sol = snes->vec_sol_always = x; 8459b94acceSBarry Smith 846272ac6f2SLois Curfman McInnes ierr = OptionsHasName(snes->prefix,"-snes_mf", &flg); CHKERRQ(ierr); 847272ac6f2SLois Curfman McInnes if (flg && snes->method_class == SNES_NONLINEAR_EQUATIONS) { 848272ac6f2SLois Curfman McInnes Mat J; 849272ac6f2SLois Curfman McInnes ierr = SNESDefaultMatrixFreeMatCreate(snes,snes->vec_sol,&J);CHKERRQ(ierr); 850272ac6f2SLois Curfman McInnes ierr = SNESSetJacobian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 851272ac6f2SLois Curfman McInnes PLogObjectParent(snes,J); 852272ac6f2SLois Curfman McInnes snes->mfshell = J; 853272ac6f2SLois Curfman McInnes } 8549b94acceSBarry Smith if ((snes->method_class == SNES_NONLINEAR_EQUATIONS)) { 85537fcc0dbSBarry Smith if (!snes->vec_func) SETERRQ(1,"SNESSetUp:Must call SNESSetFunction() first"); 85637fcc0dbSBarry Smith if (!snes->computefunction) SETERRQ(1,"SNESSetUp:Must call SNESSetFunction() first"); 85737fcc0dbSBarry Smith if (!snes->jacobian) SETERRQ(1,"SNESSetUp:Must call SNESSetJacobian() first"); 858a703fe33SLois Curfman McInnes 859a703fe33SLois Curfman McInnes /* Set the KSP stopping criterion to use the Eisenstat-Walker method */ 860a703fe33SLois Curfman McInnes if (snes->ksp_ewconv && snes->type != SNES_EQ_NTR) { 861a703fe33SLois Curfman McInnes SLES sles; KSP ksp; 862a703fe33SLois Curfman McInnes ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 863a703fe33SLois Curfman McInnes ierr = SLESGetKSP(sles,&ksp); CHKERRQ(ierr); 864a703fe33SLois Curfman McInnes ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private, 865a703fe33SLois Curfman McInnes (void *)snes); CHKERRQ(ierr); 866a703fe33SLois Curfman McInnes } 8679b94acceSBarry Smith } else if ((snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION)) { 86837fcc0dbSBarry Smith if (!snes->vec_func) SETERRQ(1,"SNESSetUp:Must call SNESSetGradient() first"); 86937fcc0dbSBarry Smith if (!snes->computefunction) SETERRQ(1,"SNESSetUp:Must call SNESSetGradient() first"); 87037fcc0dbSBarry Smith if (!snes->computeumfunction) 87137fcc0dbSBarry Smith SETERRQ(1,"SNESSetUp:Must call SNESSetMinimizationFunction() first"); 87237fcc0dbSBarry Smith if (!snes->jacobian) SETERRQ(1,"SNESSetUp:Must call SNESSetHessian() first"); 8739b94acceSBarry Smith } else SETERRQ(1,"SNESSetUp:Unknown method class"); 874a703fe33SLois Curfman McInnes if (snes->setup) {ierr = (*snes->setup)(snes); CHKERRQ(ierr);} 875a703fe33SLois Curfman McInnes snes->setup_called = 1; 876a703fe33SLois Curfman McInnes return 0; 8779b94acceSBarry Smith } 8789b94acceSBarry Smith 8799b94acceSBarry Smith /*@C 8809b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 8819b94acceSBarry Smith with SNESCreate(). 8829b94acceSBarry Smith 8839b94acceSBarry Smith Input Parameter: 8849b94acceSBarry Smith . snes - the SNES context 8859b94acceSBarry Smith 8869b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 8879b94acceSBarry Smith 88863a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 8899b94acceSBarry Smith @*/ 8909b94acceSBarry Smith int SNESDestroy(SNES snes) 8919b94acceSBarry Smith { 8929b94acceSBarry Smith int ierr; 8939b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 8949b94acceSBarry Smith ierr = (*(snes)->destroy)((PetscObject)snes); CHKERRQ(ierr); 8950452661fSBarry Smith if (snes->kspconvctx) PetscFree(snes->kspconvctx); 8969b94acceSBarry Smith if (snes->mfshell) MatDestroy(snes->mfshell); 8979b94acceSBarry Smith ierr = SLESDestroy(snes->sles); CHKERRQ(ierr); 8989b94acceSBarry Smith PLogObjectDestroy((PetscObject)snes); 8990452661fSBarry Smith PetscHeaderDestroy((PetscObject)snes); 9009b94acceSBarry Smith return 0; 9019b94acceSBarry Smith } 9029b94acceSBarry Smith 9039b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 9049b94acceSBarry Smith 9059b94acceSBarry Smith /*@ 9069b94acceSBarry Smith SNESSetMaxIterations - Sets the maximum number of global iterations to use. 9079b94acceSBarry Smith 9089b94acceSBarry Smith Input Parameters: 9099b94acceSBarry Smith . snes - the SNES context 9109b94acceSBarry Smith . maxits - maximum number of iterations to use 9119b94acceSBarry Smith 9129b94acceSBarry Smith Options Database Key: 9139b94acceSBarry Smith $ -snes_max_it maxits 9149b94acceSBarry Smith 9159b94acceSBarry Smith Note: 9169b94acceSBarry Smith The default maximum number of iterations is 50. 9179b94acceSBarry Smith 9189b94acceSBarry Smith .keywords: SNES, nonlinear, set, maximum, iterations 9199b94acceSBarry Smith 9209b94acceSBarry Smith .seealso: SNESSetMaxFunctionEvaluations() 9219b94acceSBarry Smith @*/ 9229b94acceSBarry Smith int SNESSetMaxIterations(SNES snes,int maxits) 9239b94acceSBarry Smith { 9249b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 9259b94acceSBarry Smith snes->max_its = maxits; 9269b94acceSBarry Smith return 0; 9279b94acceSBarry Smith } 9289b94acceSBarry Smith 9299b94acceSBarry Smith /*@ 9309b94acceSBarry Smith SNESSetMaxFunctionEvaluations - Sets the maximum number of function 9319b94acceSBarry Smith evaluations to use. 9329b94acceSBarry Smith 9339b94acceSBarry Smith Input Parameters: 9349b94acceSBarry Smith . snes - the SNES context 9359b94acceSBarry Smith . maxf - maximum number of function evaluations 9369b94acceSBarry Smith 9379b94acceSBarry Smith Options Database Key: 9389b94acceSBarry Smith $ -snes_max_funcs maxf 9399b94acceSBarry Smith 9409b94acceSBarry Smith Note: 9419b94acceSBarry Smith The default maximum number of function evaluations is 1000. 9429b94acceSBarry Smith 9439b94acceSBarry Smith .keywords: SNES, nonlinear, set, maximum, function, evaluations 9449b94acceSBarry Smith 9459b94acceSBarry Smith .seealso: SNESSetMaxIterations() 9469b94acceSBarry Smith @*/ 9479b94acceSBarry Smith int SNESSetMaxFunctionEvaluations(SNES snes,int maxf) 9489b94acceSBarry Smith { 9499b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 9509b94acceSBarry Smith snes->max_funcs = maxf; 9519b94acceSBarry Smith return 0; 9529b94acceSBarry Smith } 9539b94acceSBarry Smith 9549b94acceSBarry Smith /*@ 9559b94acceSBarry Smith SNESSetRelativeTolerance - Sets the relative convergence tolerance. 9569b94acceSBarry Smith 9579b94acceSBarry Smith Input Parameters: 9589b94acceSBarry Smith . snes - the SNES context 9599b94acceSBarry Smith . rtol - tolerance 9609b94acceSBarry Smith 9619b94acceSBarry Smith Options Database Key: 9629b94acceSBarry Smith $ -snes_rtol tol 9639b94acceSBarry Smith 9649b94acceSBarry Smith .keywords: SNES, nonlinear, set, relative, convergence, tolerance 9659b94acceSBarry Smith 9669b94acceSBarry Smith .seealso: SNESSetAbsoluteTolerance(), SNESSetSolutionTolerance(), 9679b94acceSBarry Smith SNESSetTruncationTolerance() 9689b94acceSBarry Smith @*/ 9699b94acceSBarry Smith int SNESSetRelativeTolerance(SNES snes,double rtol) 9709b94acceSBarry Smith { 9719b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 9729b94acceSBarry Smith snes->rtol = rtol; 9739b94acceSBarry Smith return 0; 9749b94acceSBarry Smith } 9759b94acceSBarry Smith 9769b94acceSBarry Smith /*@ 9779b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 9789b94acceSBarry Smith 9799b94acceSBarry Smith Input Parameters: 9809b94acceSBarry Smith . snes - the SNES context 9819b94acceSBarry Smith . tol - tolerance 9829b94acceSBarry Smith 9839b94acceSBarry Smith Options Database Key: 9849b94acceSBarry Smith $ -snes_trtol tol 9859b94acceSBarry Smith 9869b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 9879b94acceSBarry Smith 9889b94acceSBarry Smith .seealso: SNESSetAbsoluteTolerance(), SNESSetSolutionTolerance(), 9899b94acceSBarry Smith SNESSetTruncationTolerance() 9909b94acceSBarry Smith @*/ 9919b94acceSBarry Smith int SNESSetTrustRegionTolerance(SNES snes,double tol) 9929b94acceSBarry Smith { 9939b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 9949b94acceSBarry Smith snes->deltatol = tol; 9959b94acceSBarry Smith return 0; 9969b94acceSBarry Smith } 9979b94acceSBarry Smith 9989b94acceSBarry Smith /*@ 9999b94acceSBarry Smith SNESSetAbsoluteTolerance - Sets the absolute convergence tolerance. 10009b94acceSBarry Smith 10019b94acceSBarry Smith Input Parameters: 10029b94acceSBarry Smith . snes - the SNES context 10039b94acceSBarry Smith . atol - tolerance 10049b94acceSBarry Smith 10059b94acceSBarry Smith Options Database Key: 10069b94acceSBarry Smith $ -snes_atol tol 10079b94acceSBarry Smith 10089b94acceSBarry Smith .keywords: SNES, nonlinear, set, absolute, convergence, tolerance 10099b94acceSBarry Smith 10109b94acceSBarry Smith .seealso: SNESSetRelativeTolerance(), SNESSetSolutionTolerance(), 10119b94acceSBarry Smith SNESSetTruncationTolerance() 10129b94acceSBarry Smith @*/ 10139b94acceSBarry Smith int SNESSetAbsoluteTolerance(SNES snes,double atol) 10149b94acceSBarry Smith { 10159b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 10169b94acceSBarry Smith snes->atol = atol; 10179b94acceSBarry Smith return 0; 10189b94acceSBarry Smith } 10199b94acceSBarry Smith 10209b94acceSBarry Smith /*@ 10219b94acceSBarry Smith SNESSetTruncationTolerance - Sets the tolerance that may be used by the 10229b94acceSBarry Smith step routines to control the accuracy of the step computation. 10239b94acceSBarry Smith 10249b94acceSBarry Smith Input Parameters: 10259b94acceSBarry Smith . snes - the SNES context 10269b94acceSBarry Smith . tol - tolerance 10279b94acceSBarry Smith 10289b94acceSBarry Smith Options Database Key: 10299b94acceSBarry Smith $ -snes_ttol tol 10309b94acceSBarry Smith 10319b94acceSBarry Smith Notes: 10329b94acceSBarry Smith If the step computation involves an application of the inverse 10339b94acceSBarry Smith Jacobian (or Hessian), this parameter may be used to control the 1034a703fe33SLois Curfman McInnes accuracy of that application. 10359b94acceSBarry Smith 10369b94acceSBarry Smith .keywords: SNES, nonlinear, set, truncation, tolerance 10379b94acceSBarry Smith 10389b94acceSBarry Smith .seealso: SNESSetRelativeTolerance(), SNESSetSolutionTolerance(), 10399b94acceSBarry Smith SNESSetAbsoluteTolerance() 10409b94acceSBarry Smith @*/ 10419b94acceSBarry Smith int SNESSetTruncationTolerance(SNES snes,double tol) 10429b94acceSBarry Smith { 10439b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 10449b94acceSBarry Smith snes->trunctol = tol; 10459b94acceSBarry Smith return 0; 10469b94acceSBarry Smith } 10479b94acceSBarry Smith 10489b94acceSBarry Smith /*@ 10499b94acceSBarry Smith SNESSetSolutionTolerance - Sets the convergence tolerance in terms of 10509b94acceSBarry Smith the norm of the change in the solution between steps. 10519b94acceSBarry Smith 10529b94acceSBarry Smith Input Parameters: 10539b94acceSBarry Smith . snes - the SNES context 10549b94acceSBarry Smith . tol - tolerance 10559b94acceSBarry Smith 10569b94acceSBarry Smith Options Database Key: 10579b94acceSBarry Smith $ -snes_stol tol 10589b94acceSBarry Smith 10599b94acceSBarry Smith .keywords: SNES, nonlinear, set, solution, tolerance 10609b94acceSBarry Smith 10619b94acceSBarry Smith .seealso: SNESSetTruncationTolerance(), SNESSetRelativeTolerance(), 10629b94acceSBarry Smith SNESSetAbsoluteTolerance() 10639b94acceSBarry Smith @*/ 10649b94acceSBarry Smith int SNESSetSolutionTolerance( SNES snes, double tol ) 10659b94acceSBarry Smith { 10669b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 10679b94acceSBarry Smith snes->xtol = tol; 10689b94acceSBarry Smith return 0; 10699b94acceSBarry Smith } 10709b94acceSBarry Smith 10719b94acceSBarry Smith /*@ 10729b94acceSBarry Smith SNESSetMinFunctionTolerance - Sets the minimum allowable function tolerance 10739b94acceSBarry Smith for unconstrained minimization solvers. 10749b94acceSBarry Smith 10759b94acceSBarry Smith Input Parameters: 10769b94acceSBarry Smith . snes - the SNES context 10779b94acceSBarry Smith . ftol - minimum function tolerance 10789b94acceSBarry Smith 10799b94acceSBarry Smith Options Database Key: 10809b94acceSBarry Smith $ -snes_fmin ftol 10819b94acceSBarry Smith 10829b94acceSBarry Smith Note: 10839b94acceSBarry Smith SNESSetMinFunctionTolerance() is valid for SNES_UNCONSTRAINED_MINIMIZATION 10849b94acceSBarry Smith methods only. 10859b94acceSBarry Smith 10869b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimum, convergence, function, tolerance 10879b94acceSBarry Smith 10889b94acceSBarry Smith .seealso: SNESSetRelativeTolerance(), SNESSetSolutionTolerance(), 10899b94acceSBarry Smith SNESSetTruncationTolerance() 10909b94acceSBarry Smith @*/ 10919b94acceSBarry Smith int SNESSetMinFunctionTolerance(SNES snes,double ftol) 10929b94acceSBarry Smith { 10939b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 10949b94acceSBarry Smith snes->fmin = ftol; 10959b94acceSBarry Smith return 0; 10969b94acceSBarry Smith } 10979b94acceSBarry Smith 10989b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 10999b94acceSBarry Smith 11009b94acceSBarry Smith /*@C 11019b94acceSBarry Smith SNESSetMonitor - Sets the function that is to be used at every 11029b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 11039b94acceSBarry Smith progress. 11049b94acceSBarry Smith 11059b94acceSBarry Smith Input Parameters: 11069b94acceSBarry Smith . snes - the SNES context 11079b94acceSBarry Smith . func - monitoring routine 11089b94acceSBarry Smith . mctx - optional user-defined context for private data for the 11099b94acceSBarry Smith monitor routine (may be null) 11109b94acceSBarry Smith 11119b94acceSBarry Smith Calling sequence of func: 1112682d7d0cSBarry Smith int func(SNES snes,int its, Vec x,Vec f,double norm,void *mctx) 11139b94acceSBarry Smith 11149b94acceSBarry Smith $ snes - the SNES context 11159b94acceSBarry Smith $ its - iteration number 11169b94acceSBarry Smith $ mctx - optional monitoring context 11179b94acceSBarry Smith $ 11189b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods: 11199b94acceSBarry Smith $ norm - 2-norm function value (may be estimated) 11209b94acceSBarry Smith $ 11219b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods: 11229b94acceSBarry Smith $ norm - 2-norm gradient value (may be estimated) 11239b94acceSBarry Smith 11249b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 11259b94acceSBarry Smith 11269b94acceSBarry Smith .seealso: SNESDefaultMonitor() 11279b94acceSBarry Smith @*/ 11289b94acceSBarry Smith int SNESSetMonitor( SNES snes, int (*func)(SNES,int,double,void*), 11299b94acceSBarry Smith void *mctx ) 11309b94acceSBarry Smith { 11319b94acceSBarry Smith snes->monitor = func; 11329b94acceSBarry Smith snes->monP = (void*)mctx; 11339b94acceSBarry Smith return 0; 11349b94acceSBarry Smith } 11359b94acceSBarry Smith 11369b94acceSBarry Smith /*@C 11379b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 11389b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 11399b94acceSBarry Smith 11409b94acceSBarry Smith Input Parameters: 11419b94acceSBarry Smith . snes - the SNES context 11429b94acceSBarry Smith . func - routine to test for convergence 11439b94acceSBarry Smith . cctx - optional context for private data for the convergence routine 11449b94acceSBarry Smith (may be null) 11459b94acceSBarry Smith 11469b94acceSBarry Smith Calling sequence of func: 11479b94acceSBarry Smith int func (SNES snes,double xnorm,double gnorm, 11489b94acceSBarry Smith double f,void *cctx) 11499b94acceSBarry Smith 11509b94acceSBarry Smith $ snes - the SNES context 11519b94acceSBarry Smith $ cctx - optional convergence context 11529b94acceSBarry Smith $ xnorm - 2-norm of current iterate 11539b94acceSBarry Smith $ 11549b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods: 11559b94acceSBarry Smith $ gnorm - 2-norm of current step 11569b94acceSBarry Smith $ f - 2-norm of function 11579b94acceSBarry Smith $ 11589b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods: 11599b94acceSBarry Smith $ gnorm - 2-norm of current gradient 11609b94acceSBarry Smith $ f - function value 11619b94acceSBarry Smith 11629b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 11639b94acceSBarry Smith 11649b94acceSBarry Smith .seealso: SNESDefaultConverged() 11659b94acceSBarry Smith @*/ 11669b94acceSBarry Smith int SNESSetConvergenceTest(SNES snes, 11679b94acceSBarry Smith int (*func)(SNES,double,double,double,void*),void *cctx) 11689b94acceSBarry Smith { 11699b94acceSBarry Smith (snes)->converged = func; 11709b94acceSBarry Smith (snes)->cnvP = cctx; 11719b94acceSBarry Smith return 0; 11729b94acceSBarry Smith } 11739b94acceSBarry Smith 11749b94acceSBarry Smith /* 11759b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 11769b94acceSBarry Smith positive parameter delta. 11779b94acceSBarry Smith 11789b94acceSBarry Smith Input Parameters: 11799b94acceSBarry Smith . snes - the SNES context 11809b94acceSBarry Smith . y - approximate solution of linear system 11819b94acceSBarry Smith . fnorm - 2-norm of current function 11829b94acceSBarry Smith . delta - trust region size 11839b94acceSBarry Smith 11849b94acceSBarry Smith Output Parameters: 11859b94acceSBarry Smith . gpnorm - predicted function norm at the new point, assuming local 11869b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 11879b94acceSBarry Smith region, and exceeds zero otherwise. 11889b94acceSBarry Smith . ynorm - 2-norm of the step 11899b94acceSBarry Smith 11909b94acceSBarry Smith Note: 1191ddd12767SBarry Smith For non-trust region methods such as SNES_EQ_NLS, the parameter delta 11929b94acceSBarry Smith is set to be the maximum allowable step size. 11939b94acceSBarry Smith 11949b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 11959b94acceSBarry Smith */ 11969b94acceSBarry Smith int SNESScaleStep_Private(SNES snes,Vec y,double *fnorm,double *delta, 11979b94acceSBarry Smith double *gpnorm,double *ynorm) 11989b94acceSBarry Smith { 11999b94acceSBarry Smith double norm; 12009b94acceSBarry Smith Scalar cnorm; 1201cddf8d76SBarry Smith VecNorm(y,NORM_2, &norm ); 12029b94acceSBarry Smith if (norm > *delta) { 12039b94acceSBarry Smith norm = *delta/norm; 12049b94acceSBarry Smith *gpnorm = (1.0 - norm)*(*fnorm); 12059b94acceSBarry Smith cnorm = norm; 12069b94acceSBarry Smith VecScale( &cnorm, y ); 12079b94acceSBarry Smith *ynorm = *delta; 12089b94acceSBarry Smith } else { 12099b94acceSBarry Smith *gpnorm = 0.0; 12109b94acceSBarry Smith *ynorm = norm; 12119b94acceSBarry Smith } 12129b94acceSBarry Smith return 0; 12139b94acceSBarry Smith } 12149b94acceSBarry Smith 12159b94acceSBarry Smith /*@ 12169b94acceSBarry Smith SNESSolve - Solves a nonlinear system. Call SNESSolve after calling 121763a78c88SLois Curfman McInnes SNESCreate() and optional routines of the form SNESSetXXX(). 12189b94acceSBarry Smith 12199b94acceSBarry Smith Input Parameter: 12209b94acceSBarry Smith . snes - the SNES context 12218ddd3da0SLois Curfman McInnes . x - the solution vector 12229b94acceSBarry Smith 12239b94acceSBarry Smith Output Parameter: 12249b94acceSBarry Smith its - number of iterations until termination 12259b94acceSBarry Smith 12268ddd3da0SLois Curfman McInnes Note: 12278ddd3da0SLois Curfman McInnes The user should initialize the vector, x, with the initial guess 12288ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 12298ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 12308ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 12318ddd3da0SLois Curfman McInnes 12329b94acceSBarry Smith .keywords: SNES, nonlinear, solve 12339b94acceSBarry Smith 123463a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESDestroy() 12359b94acceSBarry Smith @*/ 12368ddd3da0SLois Curfman McInnes int SNESSolve(SNES snes,Vec x,int *its) 12379b94acceSBarry Smith { 12383c7409f5SSatish Balay int ierr, flg; 1239052efed2SBarry Smith 124037fcc0dbSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 1241*c4fc05e7SBarry Smith if (!snes->setup_called) {ierr = SNESSetUp(snes,x); CHKERRQ(ierr);} 1242*c4fc05e7SBarry Smith else {snes->vec_sol = snes->vec_sol_always = x;} 12439b94acceSBarry Smith PLogEventBegin(SNES_Solve,snes,0,0,0); 12449b94acceSBarry Smith ierr = (*(snes)->solve)(snes,its); CHKERRQ(ierr); 12459b94acceSBarry Smith PLogEventEnd(SNES_Solve,snes,0,0,0); 12463c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-snes_view", &flg); CHKERRQ(ierr); 12473c7409f5SSatish Balay if (flg) { ierr = SNESView(snes,STDOUT_VIEWER_WORLD); CHKERRQ(ierr); } 12489b94acceSBarry Smith return 0; 12499b94acceSBarry Smith } 12509b94acceSBarry Smith 12519b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 125237fcc0dbSBarry Smith static NRList *__SNESList = 0; 12539b94acceSBarry Smith 12549b94acceSBarry Smith /*@ 12554b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 12569b94acceSBarry Smith 12579b94acceSBarry Smith Input Parameters: 12589b94acceSBarry Smith . snes - the SNES context 12599b94acceSBarry Smith . method - a known method 12609b94acceSBarry Smith 12619b94acceSBarry Smith Notes: 12629b94acceSBarry Smith See "petsc/include/snes.h" for available methods (for instance) 12639b94acceSBarry Smith $ Systems of nonlinear equations: 1264ddd12767SBarry Smith $ SNES_EQ_NLS - Newton's method with line search 1265ddd12767SBarry Smith $ SNES_EQ_NTR - Newton's method with trust region 12669b94acceSBarry Smith $ Unconstrained minimization: 12679b94acceSBarry Smith $ SNES_UM_NTR - Newton's method with trust region 1268a703fe33SLois Curfman McInnes $ SNES_UM_NLS - Newton's method with line search 12699b94acceSBarry Smith 12709b94acceSBarry Smith Options Database Command: 12714b0e389bSBarry Smith $ -snes_type <method> 12729b94acceSBarry Smith $ Use -help for a list of available methods 12739b94acceSBarry Smith $ (for instance, ls or tr) 1274a703fe33SLois Curfman McInnes 1275a703fe33SLois Curfman McInnes .keysords: SNES, set, method 12769b94acceSBarry Smith @*/ 12774b0e389bSBarry Smith int SNESSetType(SNES snes,SNESType method) 12789b94acceSBarry Smith { 12799b94acceSBarry Smith int (*r)(SNES); 12809b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 12819b94acceSBarry Smith /* Get the function pointers for the iterative method requested */ 128237fcc0dbSBarry Smith if (!__SNESList) {SNESRegisterAll();} 128337fcc0dbSBarry Smith if (!__SNESList) {SETERRQ(1,"SNESSetType:Could not get methods");} 128437fcc0dbSBarry Smith r = (int (*)(SNES))NRFindRoutine( __SNESList, (int)method, (char *)0 ); 12854b0e389bSBarry Smith if (!r) {SETERRQ(1,"SNESSetType:Unknown method");} 12860452661fSBarry Smith if (snes->data) PetscFree(snes->data); 12879b94acceSBarry Smith snes->set_method_called = 1; 12889b94acceSBarry Smith return (*r)(snes); 12899b94acceSBarry Smith } 12909b94acceSBarry Smith 12919b94acceSBarry Smith /* --------------------------------------------------------------------- */ 12929b94acceSBarry Smith /*@C 12939b94acceSBarry Smith SNESRegister - Adds the method to the nonlinear solver package, given 12944b0e389bSBarry Smith a function pointer and a nonlinear solver name of the type SNESType. 12959b94acceSBarry Smith 12969b94acceSBarry Smith Input Parameters: 1297ddd12767SBarry Smith . name - for instance SNES_EQ_NLS, SNES_EQ_NTR, ... 12989b94acceSBarry Smith . sname - corfunPonding string for name 12999b94acceSBarry Smith . create - routine to create method context 13009b94acceSBarry Smith 13019b94acceSBarry Smith .keywords: SNES, nonlinear, register 13029b94acceSBarry Smith 13039b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy() 13049b94acceSBarry Smith @*/ 13059b94acceSBarry Smith int SNESRegister(int name, char *sname, int (*create)(SNES)) 13069b94acceSBarry Smith { 13079b94acceSBarry Smith int ierr; 130837fcc0dbSBarry Smith if (!__SNESList) {ierr = NRCreate(&__SNESList); CHKERRQ(ierr);} 130937fcc0dbSBarry Smith NRRegister( __SNESList, name, sname, (int (*)(void*))create ); 13109b94acceSBarry Smith return 0; 13119b94acceSBarry Smith } 13129b94acceSBarry Smith /* --------------------------------------------------------------------- */ 13139b94acceSBarry Smith /*@C 13149b94acceSBarry Smith SNESRegisterDestroy - Frees the list of nonlinear solvers that were 13159b94acceSBarry Smith registered by SNESRegister(). 13169b94acceSBarry Smith 13179b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy 13189b94acceSBarry Smith 13199b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll() 13209b94acceSBarry Smith @*/ 13219b94acceSBarry Smith int SNESRegisterDestroy() 13229b94acceSBarry Smith { 132337fcc0dbSBarry Smith if (__SNESList) { 132437fcc0dbSBarry Smith NRDestroy( __SNESList ); 132537fcc0dbSBarry Smith __SNESList = 0; 13269b94acceSBarry Smith } 13279b94acceSBarry Smith return 0; 13289b94acceSBarry Smith } 13299b94acceSBarry Smith 13309b94acceSBarry Smith /* 13314b0e389bSBarry Smith SNESGetTypeFromOptions_Private - Sets the selected method from the 13329b94acceSBarry Smith options database. 13339b94acceSBarry Smith 13349b94acceSBarry Smith Input Parameter: 13359b94acceSBarry Smith . ctx - the SNES context 13369b94acceSBarry Smith 13379b94acceSBarry Smith Output Parameter: 13389b94acceSBarry Smith . method - solver method 13399b94acceSBarry Smith 13409b94acceSBarry Smith Returns: 13419b94acceSBarry Smith Returns 1 if the method is found; 0 otherwise. 13429b94acceSBarry Smith 13439b94acceSBarry Smith Options Database Key: 13444b0e389bSBarry Smith $ -snes_type method 13459b94acceSBarry Smith */ 1346052efed2SBarry Smith int SNESGetTypeFromOptions_Private(SNES ctx,SNESType *method,int *flg) 13479b94acceSBarry Smith { 1348052efed2SBarry Smith int ierr; 13499b94acceSBarry Smith char sbuf[50]; 1350052efed2SBarry Smith ierr = OptionsGetString(ctx->prefix,"-snes_type", sbuf, 50, flg); CHKERRQ(ierr); 1351052efed2SBarry Smith if (*flg) { 1352052efed2SBarry Smith if (!__SNESList) {ierr = SNESRegisterAll(); CHKERRQ(ierr);} 135337fcc0dbSBarry Smith *method = (SNESType)NRFindID( __SNESList, sbuf ); 13549b94acceSBarry Smith } 13559b94acceSBarry Smith return 0; 13569b94acceSBarry Smith } 13579b94acceSBarry Smith 13589b94acceSBarry Smith /*@C 13599a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 13609b94acceSBarry Smith 13619b94acceSBarry Smith Input Parameter: 13624b0e389bSBarry Smith . snes - nonlinear solver context 13639b94acceSBarry Smith 13649b94acceSBarry Smith Output Parameter: 13659a28b0a6SLois Curfman McInnes . method - SNES method (or use PETSC_NULL) 13669a28b0a6SLois Curfman McInnes . name - name of SNES method (or use PETSC_NULL) 13679b94acceSBarry Smith 13689b94acceSBarry Smith .keywords: SNES, nonlinear, get, method, name 13699b94acceSBarry Smith @*/ 13704b0e389bSBarry Smith int SNESGetType(SNES snes, SNESType *method,char **name) 13719b94acceSBarry Smith { 137206a9b0adSLois Curfman McInnes int ierr; 137337fcc0dbSBarry Smith if (!__SNESList) {ierr = SNESRegisterAll(); CHKERRQ(ierr);} 13744b0e389bSBarry Smith if (method) *method = (SNESType) snes->type; 137537fcc0dbSBarry Smith if (name) *name = NRFindName( __SNESList, (int) snes->type ); 13769b94acceSBarry Smith return 0; 13779b94acceSBarry Smith } 13789b94acceSBarry Smith 13799b94acceSBarry Smith #include <stdio.h> 13809b94acceSBarry Smith /* 13814b0e389bSBarry Smith SNESPrintTypes_Private - Prints the SNES methods available from the 13829b94acceSBarry Smith options database. 13839b94acceSBarry Smith 13849b94acceSBarry Smith Input Parameters: 13859b94acceSBarry Smith . prefix - prefix (usually "-") 13864b0e389bSBarry Smith . name - the options database name (by default "snes_type") 13879b94acceSBarry Smith */ 13884b0e389bSBarry Smith int SNESPrintTypes_Private(char* prefix,char *name) 13899b94acceSBarry Smith { 13909b94acceSBarry Smith FuncList *entry; 139137fcc0dbSBarry Smith if (!__SNESList) {SNESRegisterAll();} 139237fcc0dbSBarry Smith entry = __SNESList->head; 13939b94acceSBarry Smith fprintf(stderr," %s%s (one of)",prefix,name); 13949b94acceSBarry Smith while (entry) { 13959b94acceSBarry Smith fprintf(stderr," %s",entry->name); 13969b94acceSBarry Smith entry = entry->next; 13979b94acceSBarry Smith } 13989b94acceSBarry Smith fprintf(stderr,"\n"); 13999b94acceSBarry Smith return 0; 14009b94acceSBarry Smith } 14019b94acceSBarry Smith 14029b94acceSBarry Smith /*@C 14039b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 14049b94acceSBarry Smith stored. 14059b94acceSBarry Smith 14069b94acceSBarry Smith Input Parameter: 14079b94acceSBarry Smith . snes - the SNES context 14089b94acceSBarry Smith 14099b94acceSBarry Smith Output Parameter: 14109b94acceSBarry Smith . x - the solution 14119b94acceSBarry Smith 14129b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 14139b94acceSBarry Smith 14148ddd3da0SLois Curfman McInnes .seealso: SNESGetFunction(), SNESGetSolutionUpdate() 14159b94acceSBarry Smith @*/ 14169b94acceSBarry Smith int SNESGetSolution(SNES snes,Vec *x) 14179b94acceSBarry Smith { 14189b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 14199b94acceSBarry Smith *x = snes->vec_sol_always; 14209b94acceSBarry Smith return 0; 14219b94acceSBarry Smith } 14229b94acceSBarry Smith 14239b94acceSBarry Smith /*@C 14249b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 14259b94acceSBarry Smith stored. 14269b94acceSBarry Smith 14279b94acceSBarry Smith Input Parameter: 14289b94acceSBarry Smith . snes - the SNES context 14299b94acceSBarry Smith 14309b94acceSBarry Smith Output Parameter: 14319b94acceSBarry Smith . x - the solution update 14329b94acceSBarry Smith 14339b94acceSBarry Smith Notes: 14349b94acceSBarry Smith This vector is implementation dependent. 14359b94acceSBarry Smith 14369b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 14379b94acceSBarry Smith 14389b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction 14399b94acceSBarry Smith @*/ 14409b94acceSBarry Smith int SNESGetSolutionUpdate(SNES snes,Vec *x) 14419b94acceSBarry Smith { 14429b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 14439b94acceSBarry Smith *x = snes->vec_sol_update_always; 14449b94acceSBarry Smith return 0; 14459b94acceSBarry Smith } 14469b94acceSBarry Smith 14479b94acceSBarry Smith /*@C 14483638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 14499b94acceSBarry Smith 14509b94acceSBarry Smith Input Parameter: 14519b94acceSBarry Smith . snes - the SNES context 14529b94acceSBarry Smith 14539b94acceSBarry Smith Output Parameter: 14543638b69dSLois Curfman McInnes . r - the function 14559b94acceSBarry Smith 14569b94acceSBarry Smith Notes: 14579b94acceSBarry Smith SNESGetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only 14589b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 14599b94acceSBarry Smith SNESGetMinimizationFunction() and SNESGetGradient(); 14609b94acceSBarry Smith 14619b94acceSBarry Smith .keywords: SNES, nonlinear, get function 14629b94acceSBarry Smith 14639b94acceSBarry Smith .seealso: SNESSetFunction(), SNESGetSolution() 14649b94acceSBarry Smith @*/ 14659b94acceSBarry Smith int SNESGetFunction(SNES snes,Vec *r) 14669b94acceSBarry Smith { 14679b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 14689b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 146948d91487SBarry Smith "SNESGetFunction:For SNES_NONLINEAR_EQUATIONS only"); 14709b94acceSBarry Smith *r = snes->vec_func_always; 14719b94acceSBarry Smith return 0; 14729b94acceSBarry Smith } 14739b94acceSBarry Smith 14749b94acceSBarry Smith /*@C 14753638b69dSLois Curfman McInnes SNESGetGradient - Returns the vector where the gradient is stored. 14769b94acceSBarry Smith 14779b94acceSBarry Smith Input Parameter: 14789b94acceSBarry Smith . snes - the SNES context 14799b94acceSBarry Smith 14809b94acceSBarry Smith Output Parameter: 14819b94acceSBarry Smith . r - the gradient 14829b94acceSBarry Smith 14839b94acceSBarry Smith Notes: 14849b94acceSBarry Smith SNESGetGradient() is valid for SNES_UNCONSTRAINED_MINIMIZATION methods 14859b94acceSBarry Smith only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 14869b94acceSBarry Smith SNESGetFunction(). 14879b94acceSBarry Smith 14889b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient 14899b94acceSBarry Smith 14909b94acceSBarry Smith .seealso: SNESGetMinimizationFunction(), SNESGetSolution() 14919b94acceSBarry Smith @*/ 14929b94acceSBarry Smith int SNESGetGradient(SNES snes,Vec *r) 14939b94acceSBarry Smith { 14949b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 14959b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 149648d91487SBarry Smith "SNESGetGradient:For SNES_UNCONSTRAINED_MINIMIZATION only"); 14979b94acceSBarry Smith *r = snes->vec_func_always; 14989b94acceSBarry Smith return 0; 14999b94acceSBarry Smith } 15009b94acceSBarry Smith 15019b94acceSBarry Smith /*@ 15029b94acceSBarry Smith SNESGetMinimizationFunction - Returns the scalar function value for 15039b94acceSBarry Smith unconstrained minimization problems. 15049b94acceSBarry Smith 15059b94acceSBarry Smith Input Parameter: 15069b94acceSBarry Smith . snes - the SNES context 15079b94acceSBarry Smith 15089b94acceSBarry Smith Output Parameter: 15099b94acceSBarry Smith . r - the function 15109b94acceSBarry Smith 15119b94acceSBarry Smith Notes: 15129b94acceSBarry Smith SNESGetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 15139b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 15149b94acceSBarry Smith SNESGetFunction(). 15159b94acceSBarry Smith 15169b94acceSBarry Smith .keywords: SNES, nonlinear, get, function 15179b94acceSBarry Smith 15189b94acceSBarry Smith .seealso: SNESGetGradient(), SNESGetSolution() 15199b94acceSBarry Smith @*/ 15209b94acceSBarry Smith int SNESGetMinimizationFunction(SNES snes,double *r) 15219b94acceSBarry Smith { 15229b94acceSBarry Smith PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 15239b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 152448d91487SBarry Smith "SNESGetMinimizationFunction:For SNES_UNCONSTRAINED_MINIMIZATION only"); 15259b94acceSBarry Smith *r = snes->fc; 15269b94acceSBarry Smith return 0; 15279b94acceSBarry Smith } 15289b94acceSBarry Smith 15299b94acceSBarry Smith 15303c7409f5SSatish Balay /*@C 15313c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 15323c7409f5SSatish Balay SNES options in the database. 15333c7409f5SSatish Balay 15343c7409f5SSatish Balay Input Parameter: 15353c7409f5SSatish Balay . snes - the SNES context 15363c7409f5SSatish Balay . prefix - the prefix to prepend to all option names 15373c7409f5SSatish Balay 15383c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 15393c7409f5SSatish Balay @*/ 15403c7409f5SSatish Balay int SNESSetOptionsPrefix(SNES snes,char *prefix) 15413c7409f5SSatish Balay { 15423c7409f5SSatish Balay int ierr; 15433c7409f5SSatish Balay 15443c7409f5SSatish Balay PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 15453c7409f5SSatish Balay ierr = PetscObjectSetPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 15463c7409f5SSatish Balay ierr = SLESSetOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 15473c7409f5SSatish Balay return 0; 15483c7409f5SSatish Balay } 15493c7409f5SSatish Balay 15503c7409f5SSatish Balay /*@C 15513c7409f5SSatish Balay SNESAppendOptionsPrefix - Append to the prefix used for searching for all 15523c7409f5SSatish Balay SNES options in the database. 15533c7409f5SSatish Balay 15543c7409f5SSatish Balay Input Parameter: 15553c7409f5SSatish Balay . snes - the SNES context 15563c7409f5SSatish Balay . prefix - the prefix to prepend to all option names 15573c7409f5SSatish Balay 15583c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 15593c7409f5SSatish Balay @*/ 15603c7409f5SSatish Balay int SNESAppendOptionsPrefix(SNES snes,char *prefix) 15613c7409f5SSatish Balay { 15623c7409f5SSatish Balay int ierr; 15633c7409f5SSatish Balay 15643c7409f5SSatish Balay PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 15653c7409f5SSatish Balay ierr = PetscObjectAppendPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 15663c7409f5SSatish Balay ierr = SLESAppendOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 15673c7409f5SSatish Balay return 0; 15683c7409f5SSatish Balay } 15693c7409f5SSatish Balay 1570c83590e2SSatish Balay /*@ 15713c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 15723c7409f5SSatish Balay SNES options in the database. 15733c7409f5SSatish Balay 15743c7409f5SSatish Balay Input Parameter: 15753c7409f5SSatish Balay . snes - the SNES context 15763c7409f5SSatish Balay 15773c7409f5SSatish Balay Output Parameter: 15783c7409f5SSatish Balay . prefix - pointer to the prefix string used 15793c7409f5SSatish Balay 15803c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 15813c7409f5SSatish Balay @*/ 15823c7409f5SSatish Balay int SNESGetOptionsPrefix(SNES snes,char **prefix) 15833c7409f5SSatish Balay { 15843c7409f5SSatish Balay int ierr; 15853c7409f5SSatish Balay 15863c7409f5SSatish Balay PETSCVALIDHEADERSPECIFIC(snes,SNES_COOKIE); 15873c7409f5SSatish Balay ierr = PetscObjectGetPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 15883c7409f5SSatish Balay return 0; 15893c7409f5SSatish Balay } 15903c7409f5SSatish Balay 15913c7409f5SSatish Balay 15929b94acceSBarry Smith 15939b94acceSBarry Smith 15949b94acceSBarry Smith 1595