19b94acceSBarry Smith #ifndef lint 2*d7a720efSLois Curfman McInnes static char vcid[] = "$Id: snes.c,v 1.64 1996/03/23 18:29:58 curfman Exp curfman $"; 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*); 1233455573SSatish Balay extern int SNESPrintTypes_Private(MPI_Comm,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 SNES_KSP_EW_ConvCtx *kctx; 429b94acceSBarry Smith FILE *fd; 439b94acceSBarry Smith int ierr; 449b94acceSBarry Smith SLES sles; 459b94acceSBarry Smith char *method; 4619bcc07fSBarry Smith ViewerType vtype; 479b94acceSBarry Smith 4819bcc07fSBarry Smith ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr); 4919bcc07fSBarry Smith if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER) { 5090ace30eSBarry Smith ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr); 5177c4ece6SBarry Smith PetscFPrintf(snes->comm,fd,"SNES Object:\n"); 524b0e389bSBarry Smith SNESGetType(snes,PETSC_NULL,&method); 5377c4ece6SBarry Smith PetscFPrintf(snes->comm,fd," method: %s\n",method); 549b94acceSBarry Smith if (snes->view) (*snes->view)((PetscObject)snes,viewer); 5577c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 569b94acceSBarry Smith " maximum iterations=%d, maximum function evaluations=%d\n", 579b94acceSBarry Smith snes->max_its,snes->max_funcs); 5877c4ece6SBarry Smith PetscFPrintf(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) 6277c4ece6SBarry Smith PetscFPrintf(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) { 6677c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 679b94acceSBarry Smith " Eisenstat-Walker computation of KSP relative tolerance (version %d)\n", 689b94acceSBarry Smith kctx->version); 6977c4ece6SBarry Smith PetscFPrintf(snes->comm,fd, 709b94acceSBarry Smith " rtol_0=%g, rtol_max=%g, threshold=%g\n",kctx->rtol_0, 719b94acceSBarry Smith kctx->rtol_max,kctx->threshold); 7277c4ece6SBarry Smith PetscFPrintf(snes->comm,fd," gamma=%g, alpha=%g, alpha2=%g\n", 739b94acceSBarry Smith kctx->gamma,kctx->alpha,kctx->alpha2); 749b94acceSBarry Smith } 759b94acceSBarry Smith } 76c30f285eSLois Curfman McInnes } else if (vtype == STRING_VIEWER) { 77c30f285eSLois Curfman McInnes SNESGetType(snes,PETSC_NULL,&method); 78c30f285eSLois Curfman McInnes ViewerStringSPrintf(viewer," %-3.3s",method); 7919bcc07fSBarry Smith } 809b94acceSBarry Smith SNESGetSLES(snes,&sles); 819b94acceSBarry Smith ierr = SLESView(sles,viewer); CHKERRQ(ierr); 829b94acceSBarry Smith return 0; 839b94acceSBarry Smith } 849b94acceSBarry Smith 859b94acceSBarry Smith /*@ 86682d7d0cSBarry Smith SNESSetFromOptions - Sets various SNES and SLES parameters from user options. 879b94acceSBarry Smith 889b94acceSBarry Smith Input Parameter: 899b94acceSBarry Smith . snes - the SNES context 909b94acceSBarry Smith 919b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 929b94acceSBarry Smith 93a86d99e1SLois Curfman McInnes .seealso: SNESPrintHelp(), SNESSetOptionsPrefix() 949b94acceSBarry Smith @*/ 959b94acceSBarry Smith int SNESSetFromOptions(SNES snes) 969b94acceSBarry Smith { 974b0e389bSBarry Smith SNESType method; 989b94acceSBarry Smith double tmp; 999b94acceSBarry Smith SLES sles; 1003c7409f5SSatish Balay int ierr, flg; 1013c7409f5SSatish Balay int version = PETSC_DEFAULT; 1029b94acceSBarry Smith double rtol_0 = PETSC_DEFAULT; 1039b94acceSBarry Smith double rtol_max = PETSC_DEFAULT; 1049b94acceSBarry Smith double gamma2 = PETSC_DEFAULT; 1059b94acceSBarry Smith double alpha = PETSC_DEFAULT; 1069b94acceSBarry Smith double alpha2 = PETSC_DEFAULT; 1079b94acceSBarry Smith double threshold = PETSC_DEFAULT; 1089b94acceSBarry Smith 10977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1104b0e389bSBarry Smith if (snes->setup_called)SETERRQ(1,"SNESSetFromOptions:Must call prior to SNESSetUp!"); 111052efed2SBarry Smith ierr = SNESGetTypeFromOptions_Private(snes,&method,&flg); CHKERRQ(ierr); 112052efed2SBarry Smith if (flg) { 113052efed2SBarry Smith ierr = SNESSetType(snes,method); CHKERRQ(ierr); 1149b94acceSBarry Smith } 1155334005bSBarry Smith else if (!snes->set_method_called) { 1165334005bSBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1175334005bSBarry Smith ierr = SNESSetType(snes,SNES_EQ_NLS); CHKERRQ(ierr); 1185334005bSBarry Smith } 1195334005bSBarry Smith else { 1205334005bSBarry Smith ierr = SNESSetType(snes,SNES_UM_NTR); CHKERRQ(ierr); 1215334005bSBarry Smith } 1225334005bSBarry Smith } 1235334005bSBarry Smith 1243c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 1253c7409f5SSatish Balay if (flg) { SNESPrintHelp(snes); } 1263c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_stol",&tmp, &flg); CHKERRQ(ierr); 127*d7a720efSLois Curfman McInnes if (flg) { SNESSetTolerances(snes,PETSC_DEFAULT,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT); } 1283c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_atol",&tmp, &flg); CHKERRQ(ierr); 129*d7a720efSLois Curfman McInnes if (flg) { SNESSetTolerances(snes,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT); } 1303c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_rtol",&tmp, &flg); CHKERRQ(ierr); 131*d7a720efSLois Curfman McInnes if (flg) { SNESSetTolerances(snes,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT); } 1323c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_it",&snes->max_its, &flg); CHKERRQ(ierr); 1333c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_funcs",&snes->max_funcs, &flg);CHKERRQ(ierr); 134*d7a720efSLois Curfman McInnes 135*d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_trtol",&tmp, &flg); CHKERRQ(ierr); 136*d7a720efSLois Curfman McInnes if (flg) { SNESSetTrustRegionTolerance(snes,tmp); } 137*d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_fmin",&tmp, &flg); CHKERRQ(ierr); 138*d7a720efSLois Curfman McInnes if (flg) { SNESSetMinimizationFunctionTolerance(snes,tmp); } 1393c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_ksp_ew_conv", &flg); CHKERRQ(ierr); 1403c7409f5SSatish Balay if (flg) { snes->ksp_ewconv = 1; } 1413c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_ksp_ew_version",&version, \ 1423c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1433c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtol0",&rtol_0, \ 1443c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1453c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtolmax",&rtol_max, \ 1463c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1473c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_gamma",&gamma2, \ 1483c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1493c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha",&alpha, \ 1503c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1513c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha2",&alpha2, \ 1523c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1533c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_threshold",&threshold, \ 1543c7409f5SSatish Balay &flg); CHKERRQ(ierr); 1559b94acceSBarry Smith ierr = SNES_KSP_SetParametersEW(snes,version,rtol_0,rtol_max,gamma2,alpha, 1569b94acceSBarry Smith alpha2,threshold); CHKERRQ(ierr); 1573c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_monitor", &flg); CHKERRQ(ierr); 1583c7409f5SSatish Balay if (flg) { SNESSetMonitor(snes,SNESDefaultMonitor,0); } 1593c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_smonitor", &flg); CHKERRQ(ierr); 1603c7409f5SSatish Balay if (flg) { SNESSetMonitor(snes,SNESDefaultSMonitor,0); } 1613c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_xmonitor", &flg); CHKERRQ(ierr); 1623c7409f5SSatish Balay if (flg) { 16317699dbbSLois Curfman McInnes int rank = 0; 164d7e8b826SBarry Smith DrawLG lg; 16517699dbbSLois Curfman McInnes MPI_Initialized(&rank); 16617699dbbSLois Curfman McInnes if (rank) MPI_Comm_rank(snes->comm,&rank); 16717699dbbSLois Curfman McInnes if (!rank) { 1689b94acceSBarry Smith ierr = SNESLGMonitorCreate(0,0,0,0,300,300,&lg); CHKERRQ(ierr); 1699b94acceSBarry Smith ierr = SNESSetMonitor(snes,SNESLGMonitor,(void *)lg); CHKERRQ(ierr); 1709b94acceSBarry Smith PLogObjectParent(snes,lg); 1719b94acceSBarry Smith } 1729b94acceSBarry Smith } 1733c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_fd", &flg); CHKERRQ(ierr); 1743c7409f5SSatish Balay if (flg && snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1759b94acceSBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre, 1769b94acceSBarry Smith SNESDefaultComputeJacobian,snes->funP); CHKERRQ(ierr); 17731615d04SLois Curfman McInnes PLogInfo((PetscObject)snes, 17831615d04SLois Curfman McInnes "SNESSetFromOptions: Setting default finite difference Jacobian matrix\n"); 17931615d04SLois Curfman McInnes } 18031615d04SLois Curfman McInnes else if (flg && snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 18131615d04SLois Curfman McInnes ierr = SNESSetHessian(snes,snes->jacobian,snes->jacobian_pre, 18231615d04SLois Curfman McInnes SNESDefaultComputeHessian,snes->funP); CHKERRQ(ierr); 18331615d04SLois Curfman McInnes PLogInfo((PetscObject)snes, 18431615d04SLois Curfman McInnes "SNESSetFromOptions: Setting default finite difference Hessian matrix\n"); 1859b94acceSBarry Smith } 1869b94acceSBarry Smith ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 1879b94acceSBarry Smith ierr = SLESSetFromOptions(sles); CHKERRQ(ierr); 1889b94acceSBarry Smith if (!snes->setfromoptions) return 0; 1899b94acceSBarry Smith return (*snes->setfromoptions)(snes); 1909b94acceSBarry Smith } 1919b94acceSBarry Smith 1929b94acceSBarry Smith /*@ 1939b94acceSBarry Smith SNESPrintHelp - Prints all options for the SNES component. 1949b94acceSBarry Smith 1959b94acceSBarry Smith Input Parameter: 1969b94acceSBarry Smith . snes - the SNES context 1979b94acceSBarry Smith 198a703fe33SLois Curfman McInnes Options Database Keys: 199a703fe33SLois Curfman McInnes $ -help, -h 200a703fe33SLois Curfman McInnes 2019b94acceSBarry Smith .keywords: SNES, nonlinear, help 2029b94acceSBarry Smith 203682d7d0cSBarry Smith .seealso: SNESSetFromOptions() 2049b94acceSBarry Smith @*/ 2059b94acceSBarry Smith int SNESPrintHelp(SNES snes) 2069b94acceSBarry Smith { 2076daaf66cSBarry Smith char p[64]; 2086daaf66cSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 2096daaf66cSBarry Smith 21077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2116daaf66cSBarry Smith 2126daaf66cSBarry Smith PetscStrcpy(p,"-"); 2136daaf66cSBarry Smith if (snes->prefix) PetscStrcat(p, snes->prefix); 2146daaf66cSBarry Smith 2156daaf66cSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 2166daaf66cSBarry Smith 21777c4ece6SBarry Smith PetscPrintf(snes->comm,"SNES options ----------------------------\n"); 21833455573SSatish Balay SNESPrintTypes_Private(snes->comm,p,"snes_type"); 21977c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_monitor: use default SNES monitor\n",p); 22077c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_view: view SNES info after each nonlinear solve\n",p); 22177c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_max_it its (default %d)\n",p,snes->max_its); 22277c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_stol tol (default %g)\n",p,snes->xtol); 22377c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_atol tol (default %g)\n",p,snes->atol); 22477c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_rtol tol (default %g)\n",p,snes->rtol); 22577c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_ttol tol (default %g)\n",p,snes->trunctol); 22677c4ece6SBarry Smith PetscPrintf(snes->comm, 2279b94acceSBarry Smith " options for solving systems of nonlinear equations only:\n"); 22877c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_fd: use finite differences for Jacobian\n",p); 22977c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_mf: use matrix-free Jacobian\n",p); 23077c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_ksp_ew_conv: use Eisenstat-Walker computation of KSP rtol. Params are:\n",p); 23177c4ece6SBarry Smith PetscPrintf(snes->comm, 2329b94acceSBarry Smith " %ssnes_ksp_ew_version version (1 or 2, default is %d)\n", 2336daaf66cSBarry Smith p,kctx->version); 23477c4ece6SBarry Smith PetscPrintf(snes->comm, 2359b94acceSBarry Smith " %ssnes_ksp_ew_rtol0 rtol0 (0 <= rtol0 < 1, default %g)\n", 2366daaf66cSBarry Smith p,kctx->rtol_0); 23777c4ece6SBarry Smith PetscPrintf(snes->comm, 2389b94acceSBarry Smith " %ssnes_ksp_ew_rtolmax rtolmax (0 <= rtolmax < 1, default %g)\n", 2396daaf66cSBarry Smith p,kctx->rtol_max); 24077c4ece6SBarry Smith PetscPrintf(snes->comm, 2419b94acceSBarry Smith " %ssnes_ksp_ew_gamma gamma (0 <= gamma <= 1, default %g)\n", 2426daaf66cSBarry Smith p,kctx->gamma); 24377c4ece6SBarry Smith PetscPrintf(snes->comm, 2449b94acceSBarry Smith " %ssnes_ksp_ew_alpha alpha (1 < alpha <= 2, default %g)\n", 2456daaf66cSBarry Smith p,kctx->alpha); 24677c4ece6SBarry Smith PetscPrintf(snes->comm, 2479b94acceSBarry Smith " %ssnes_ksp_ew_alpha2 alpha2 (default %g)\n", 2486daaf66cSBarry Smith p,kctx->alpha2); 24977c4ece6SBarry Smith PetscPrintf(snes->comm, 2509b94acceSBarry Smith " %ssnes_ksp_ew_threshold threshold (0 < threshold < 1, default %g)\n", 2516daaf66cSBarry Smith p,kctx->threshold); 25277c4ece6SBarry Smith PetscPrintf(snes->comm, 2539b94acceSBarry Smith " options for solving unconstrained minimization problems only:\n"); 25477c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_fmin tol (default %g)\n",p,snes->fmin); 25577c4ece6SBarry Smith PetscPrintf(snes->comm," Run program with %ssnes_type method -help for help on ",p); 25677c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_fd: use finite differences for Hessian\n",p); 25777c4ece6SBarry Smith PetscPrintf(snes->comm," %ssnes_mf: use matrix-free Hessian\n",p); 25877c4ece6SBarry Smith PetscPrintf(snes->comm,"a particular method\n"); 2596daaf66cSBarry Smith if (snes->printhelp) (*snes->printhelp)(snes,p); 2609b94acceSBarry Smith return 0; 2619b94acceSBarry Smith } 2629b94acceSBarry Smith /*@ 2639b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 2649b94acceSBarry Smith the nonlinear solvers. 2659b94acceSBarry Smith 2669b94acceSBarry Smith Input Parameters: 2679b94acceSBarry Smith . snes - the SNES context 2689b94acceSBarry Smith . usrP - optional user context 2699b94acceSBarry Smith 2709b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 2719b94acceSBarry Smith 2729b94acceSBarry Smith .seealso: SNESGetApplicationContext() 2739b94acceSBarry Smith @*/ 2749b94acceSBarry Smith int SNESSetApplicationContext(SNES snes,void *usrP) 2759b94acceSBarry Smith { 27677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2779b94acceSBarry Smith snes->user = usrP; 2789b94acceSBarry Smith return 0; 2799b94acceSBarry Smith } 2809b94acceSBarry Smith /*@C 2819b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 2829b94acceSBarry Smith nonlinear solvers. 2839b94acceSBarry Smith 2849b94acceSBarry Smith Input Parameter: 2859b94acceSBarry Smith . snes - SNES context 2869b94acceSBarry Smith 2879b94acceSBarry Smith Output Parameter: 2889b94acceSBarry Smith . usrP - user context 2899b94acceSBarry Smith 2909b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 2919b94acceSBarry Smith 2929b94acceSBarry Smith .seealso: SNESSetApplicationContext() 2939b94acceSBarry Smith @*/ 2949b94acceSBarry Smith int SNESGetApplicationContext( SNES snes, void **usrP ) 2959b94acceSBarry Smith { 29677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2979b94acceSBarry Smith *usrP = snes->user; 2989b94acceSBarry Smith return 0; 2999b94acceSBarry Smith } 3009b94acceSBarry Smith /*@ 3019b94acceSBarry Smith SNESGetIterationNumber - Gets the current iteration number of the 3029b94acceSBarry Smith nonlinear solver. 3039b94acceSBarry Smith 3049b94acceSBarry Smith Input Parameter: 3059b94acceSBarry Smith . snes - SNES context 3069b94acceSBarry Smith 3079b94acceSBarry Smith Output Parameter: 3089b94acceSBarry Smith . iter - iteration number 3099b94acceSBarry Smith 3109b94acceSBarry Smith .keywords: SNES, nonlinear, get, iteration, number 3119b94acceSBarry Smith @*/ 3129b94acceSBarry Smith int SNESGetIterationNumber(SNES snes,int* iter) 3139b94acceSBarry Smith { 31477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3159b94acceSBarry Smith *iter = snes->iter; 3169b94acceSBarry Smith return 0; 3179b94acceSBarry Smith } 3189b94acceSBarry Smith /*@ 3199b94acceSBarry Smith SNESGetFunctionNorm - Gets the norm of the current function that was set 3209b94acceSBarry Smith with SNESSSetFunction(). 3219b94acceSBarry Smith 3229b94acceSBarry Smith Input Parameter: 3239b94acceSBarry Smith . snes - SNES context 3249b94acceSBarry Smith 3259b94acceSBarry Smith Output Parameter: 3269b94acceSBarry Smith . fnorm - 2-norm of function 3279b94acceSBarry Smith 3289b94acceSBarry Smith Note: 3299b94acceSBarry Smith SNESGetFunctionNorm() is valid for SNES_NONLINEAR_EQUATIONS methods only. 330a86d99e1SLois Curfman McInnes A related routine for SNES_UNCONSTRAINED_MINIMIZATION methods is 331a86d99e1SLois Curfman McInnes SNESGetGradientNorm(). 3329b94acceSBarry Smith 3339b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm 334a86d99e1SLois Curfman McInnes 335a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction() 3369b94acceSBarry Smith @*/ 3379b94acceSBarry Smith int SNESGetFunctionNorm(SNES snes,Scalar *fnorm) 3389b94acceSBarry Smith { 33977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3409b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 34148d91487SBarry Smith "SNESGetFunctionNorm:For SNES_NONLINEAR_EQUATIONS only"); 3429b94acceSBarry Smith *fnorm = snes->norm; 3439b94acceSBarry Smith return 0; 3449b94acceSBarry Smith } 3459b94acceSBarry Smith /*@ 3469b94acceSBarry Smith SNESGetGradientNorm - Gets the norm of the current gradient that was set 3479b94acceSBarry Smith with SNESSSetGradient(). 3489b94acceSBarry Smith 3499b94acceSBarry Smith Input Parameter: 3509b94acceSBarry Smith . snes - SNES context 3519b94acceSBarry Smith 3529b94acceSBarry Smith Output Parameter: 3539b94acceSBarry Smith . fnorm - 2-norm of gradient 3549b94acceSBarry Smith 3559b94acceSBarry Smith Note: 3569b94acceSBarry Smith SNESGetGradientNorm() is valid for SNES_UNCONSTRAINED_MINIMIZATION 357a86d99e1SLois Curfman McInnes methods only. A related routine for SNES_NONLINEAR_EQUATIONS methods 358a86d99e1SLois Curfman McInnes is SNESGetFunctionNorm(). 3599b94acceSBarry Smith 3609b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient, norm 361a86d99e1SLois Curfman McInnes 362a86d99e1SLois Curfman McInnes .seelso: SNESSetGradient() 3639b94acceSBarry Smith @*/ 3649b94acceSBarry Smith int SNESGetGradientNorm(SNES snes,Scalar *gnorm) 3659b94acceSBarry Smith { 36677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3679b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 36848d91487SBarry Smith "SNESGetGradientNorm:For SNES_UNCONSTRAINED_MINIMIZATION only"); 3699b94acceSBarry Smith *gnorm = snes->norm; 3709b94acceSBarry Smith return 0; 3719b94acceSBarry Smith } 3729b94acceSBarry Smith /*@ 3739b94acceSBarry Smith SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps 3749b94acceSBarry Smith attempted by the nonlinear solver. 3759b94acceSBarry Smith 3769b94acceSBarry Smith Input Parameter: 3779b94acceSBarry Smith . snes - SNES context 3789b94acceSBarry Smith 3799b94acceSBarry Smith Output Parameter: 3809b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 3819b94acceSBarry Smith 3829b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 3839b94acceSBarry Smith @*/ 3849b94acceSBarry Smith int SNESGetNumberUnsuccessfulSteps(SNES snes,int* nfails) 3859b94acceSBarry Smith { 38677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3879b94acceSBarry Smith *nfails = snes->nfailures; 3889b94acceSBarry Smith return 0; 3899b94acceSBarry Smith } 3909b94acceSBarry Smith /*@C 3919b94acceSBarry Smith SNESGetSLES - Returns the SLES context for a SNES solver. 3929b94acceSBarry Smith 3939b94acceSBarry Smith Input Parameter: 3949b94acceSBarry Smith . snes - the SNES context 3959b94acceSBarry Smith 3969b94acceSBarry Smith Output Parameter: 3979b94acceSBarry Smith . sles - the SLES context 3989b94acceSBarry Smith 3999b94acceSBarry Smith Notes: 4009b94acceSBarry Smith The user can then directly manipulate the SLES context to set various 4019b94acceSBarry Smith options, etc. Likewise, the user can then extract and manipulate the 4029b94acceSBarry Smith KSP and PC contexts as well. 4039b94acceSBarry Smith 4049b94acceSBarry Smith .keywords: SNES, nonlinear, get, SLES, context 4059b94acceSBarry Smith 4069b94acceSBarry Smith .seealso: SLESGetPC(), SLESGetKSP() 4079b94acceSBarry Smith @*/ 4089b94acceSBarry Smith int SNESGetSLES(SNES snes,SLES *sles) 4099b94acceSBarry Smith { 41077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 4119b94acceSBarry Smith *sles = snes->sles; 4129b94acceSBarry Smith return 0; 4139b94acceSBarry Smith } 4149b94acceSBarry Smith /* -----------------------------------------------------------*/ 4159b94acceSBarry Smith /*@C 4169b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 4179b94acceSBarry Smith 4189b94acceSBarry Smith Input Parameter: 4199b94acceSBarry Smith . comm - MPI communicator 4209b94acceSBarry Smith . type - type of method, one of 4219b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS 4229b94acceSBarry Smith $ (for systems of nonlinear equations) 4239b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION 4249b94acceSBarry Smith $ (for unconstrained minimization) 4259b94acceSBarry Smith 4269b94acceSBarry Smith Output Parameter: 4279b94acceSBarry Smith . outsnes - the new SNES context 4289b94acceSBarry Smith 4299b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 4309b94acceSBarry Smith 43163a78c88SLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy() 4329b94acceSBarry Smith @*/ 4334b0e389bSBarry Smith int SNESCreate(MPI_Comm comm,SNESProblemType type,SNES *outsnes) 4349b94acceSBarry Smith { 4359b94acceSBarry Smith int ierr; 4369b94acceSBarry Smith SNES snes; 4379b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 43837fcc0dbSBarry Smith 4399b94acceSBarry Smith *outsnes = 0; 4402a0acf01SLois Curfman McInnes if (type != SNES_UNCONSTRAINED_MINIMIZATION && type != SNES_NONLINEAR_EQUATIONS) 4412a0acf01SLois Curfman McInnes SETERRQ(1,"SNESCreate:incorrect method type"); 4420452661fSBarry Smith PetscHeaderCreate(snes,_SNES,SNES_COOKIE,SNES_UNKNOWN_METHOD,comm); 4439b94acceSBarry Smith PLogObjectCreate(snes); 4449b94acceSBarry Smith snes->max_its = 50; 4459b94acceSBarry Smith snes->max_funcs = 1000; 4469b94acceSBarry Smith snes->norm = 0.0; 4475a2d0531SBarry Smith if (type == SNES_UNCONSTRAINED_MINIMIZATION) { 4482a0acf01SLois Curfman McInnes snes->rtol = 1.e-08; 4499b94acceSBarry Smith snes->atol = 1.e-10; 4505a2d0531SBarry Smith } 451b4874afaSBarry Smith else { 452b4874afaSBarry Smith snes->rtol = 1.e-8; 453b4874afaSBarry Smith snes->ttol = 0.0; 454b4874afaSBarry Smith snes->atol = 1.e-50; 455b4874afaSBarry Smith } 4569b94acceSBarry Smith snes->xtol = 1.e-8; 4579b94acceSBarry Smith snes->trunctol = 1.e-12; 4589b94acceSBarry Smith snes->nfuncs = 0; 4599b94acceSBarry Smith snes->nfailures = 0; 4609b94acceSBarry Smith snes->monitor = 0; 4619b94acceSBarry Smith snes->data = 0; 4629b94acceSBarry Smith snes->view = 0; 4639b94acceSBarry Smith snes->computeumfunction = 0; 4649b94acceSBarry Smith snes->umfunP = 0; 4659b94acceSBarry Smith snes->fc = 0; 4669b94acceSBarry Smith snes->deltatol = 1.e-12; 4679b94acceSBarry Smith snes->fmin = -1.e30; 4689b94acceSBarry Smith snes->method_class = type; 4699b94acceSBarry Smith snes->set_method_called = 0; 470a703fe33SLois Curfman McInnes snes->setup_called = 0; 4719b94acceSBarry Smith snes->ksp_ewconv = 0; 4729b94acceSBarry Smith 4739b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 4740452661fSBarry Smith kctx = PetscNew(SNES_KSP_EW_ConvCtx); CHKPTRQ(kctx); 4759b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 4769b94acceSBarry Smith kctx->version = 2; 4779b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 4789b94acceSBarry Smith this was too large for some test cases */ 4799b94acceSBarry Smith kctx->rtol_last = 0; 4809b94acceSBarry Smith kctx->rtol_max = .9; 4819b94acceSBarry Smith kctx->gamma = 1.0; 4829b94acceSBarry Smith kctx->alpha2 = .5*(1.0 + sqrt(5.0)); 4839b94acceSBarry Smith kctx->alpha = kctx->alpha2; 4849b94acceSBarry Smith kctx->threshold = .1; 4859b94acceSBarry Smith kctx->lresid_last = 0; 4869b94acceSBarry Smith kctx->norm_last = 0; 4879b94acceSBarry Smith 4889b94acceSBarry Smith ierr = SLESCreate(comm,&snes->sles); CHKERRQ(ierr); 4899b94acceSBarry Smith PLogObjectParent(snes,snes->sles) 4905334005bSBarry Smith 4919b94acceSBarry Smith *outsnes = snes; 4929b94acceSBarry Smith return 0; 4939b94acceSBarry Smith } 4949b94acceSBarry Smith 4959b94acceSBarry Smith /* --------------------------------------------------------------- */ 4969b94acceSBarry Smith /*@C 4979b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 4989b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 4999b94acceSBarry Smith equations. 5009b94acceSBarry Smith 5019b94acceSBarry Smith Input Parameters: 5029b94acceSBarry Smith . snes - the SNES context 5039b94acceSBarry Smith . func - function evaluation routine 5049b94acceSBarry Smith . ctx - optional user-defined function context 5059b94acceSBarry Smith . r - vector to store function value 5069b94acceSBarry Smith 5079b94acceSBarry Smith Calling sequence of func: 5089b94acceSBarry Smith . func (SNES, Vec x, Vec f, void *ctx); 5099b94acceSBarry Smith 5109b94acceSBarry Smith . x - input vector 5115334005bSBarry Smith . f - vector function 5129b94acceSBarry Smith . ctx - optional user-defined context for private data for the 5139b94acceSBarry Smith function evaluation routine (may be null) 5149b94acceSBarry Smith 5159b94acceSBarry Smith Notes: 5169b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 5179b94acceSBarry Smith $ f'(x) x = -f(x), 5189b94acceSBarry Smith $ where f'(x) denotes the Jacobian matrix and f(x) is the function. 5199b94acceSBarry Smith 5209b94acceSBarry Smith SNESSetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 5219b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 5229b94acceSBarry Smith SNESSetMinimizationFunction() and SNESSetGradient(); 5239b94acceSBarry Smith 5249b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 5259b94acceSBarry Smith 526a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 5279b94acceSBarry Smith @*/ 5285334005bSBarry Smith int SNESSetFunction( SNES snes, Vec r, int (*func)(SNES,Vec,Vec,void*),void *ctx) 5299b94acceSBarry Smith { 53077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 5319b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 53248d91487SBarry Smith "SNESSetFunction:For SNES_NONLINEAR_EQUATIONS only"); 5339b94acceSBarry Smith snes->computefunction = func; 5349b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 5359b94acceSBarry Smith snes->funP = ctx; 5369b94acceSBarry Smith return 0; 5379b94acceSBarry Smith } 5389b94acceSBarry Smith 5399b94acceSBarry Smith /*@ 5409b94acceSBarry Smith SNESComputeFunction - Computes the function that has been set with 5419b94acceSBarry Smith SNESSetFunction(). 5429b94acceSBarry Smith 5439b94acceSBarry Smith Input Parameters: 5449b94acceSBarry Smith . snes - the SNES context 5459b94acceSBarry Smith . x - input vector 5469b94acceSBarry Smith 5479b94acceSBarry Smith Output Parameter: 5483638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 5499b94acceSBarry Smith 550a86d99e1SLois Curfman McInnes n Notes: 5519b94acceSBarry Smith SNESComputeFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 5529b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 5539b94acceSBarry Smith SNESComputeMinimizationFunction() and SNESComputeGradient(); 5549b94acceSBarry Smith 5559b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 5569b94acceSBarry Smith 557a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 5589b94acceSBarry Smith @*/ 5599b94acceSBarry Smith int SNESComputeFunction(SNES snes,Vec x, Vec y) 5609b94acceSBarry Smith { 5619b94acceSBarry Smith int ierr; 5629b94acceSBarry Smith 563a86d99e1SLois Curfman McInnes if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 564a86d99e1SLois Curfman McInnes "SNESComputeFunction: For SNES_NONLINEAR_EQUATIONS only"); 5659b94acceSBarry Smith PLogEventBegin(SNES_FunctionEval,snes,x,y,0); 5669b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 5679b94acceSBarry Smith PLogEventEnd(SNES_FunctionEval,snes,x,y,0); 5689b94acceSBarry Smith return 0; 5699b94acceSBarry Smith } 5709b94acceSBarry Smith 5719b94acceSBarry Smith /*@C 5729b94acceSBarry Smith SNESSetMinimizationFunction - Sets the function evaluation routine for 5739b94acceSBarry Smith unconstrained minimization. 5749b94acceSBarry Smith 5759b94acceSBarry Smith Input Parameters: 5769b94acceSBarry Smith . snes - the SNES context 5779b94acceSBarry Smith . func - function evaluation routine 5789b94acceSBarry Smith . ctx - optional user-defined function context 5799b94acceSBarry Smith 5809b94acceSBarry Smith Calling sequence of func: 5819b94acceSBarry Smith . func (SNES snes,Vec x,double *f,void *ctx); 5829b94acceSBarry Smith 5839b94acceSBarry Smith . x - input vector 5849b94acceSBarry Smith . f - function 5859b94acceSBarry Smith . ctx - optional user-defined context for private data for the 5869b94acceSBarry Smith function evaluation routine (may be null) 5879b94acceSBarry Smith 5889b94acceSBarry Smith Notes: 5899b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 5909b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 5919b94acceSBarry Smith SNESSetFunction(). 5929b94acceSBarry Smith 5939b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimization, function 5949b94acceSBarry Smith 595a86d99e1SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESComputeMinimizationFunction(), 596a86d99e1SLois Curfman McInnes SNESSetHessian(), SNESSetGradient() 5979b94acceSBarry Smith @*/ 5989b94acceSBarry Smith int SNESSetMinimizationFunction(SNES snes,int (*func)(SNES,Vec,double*,void*), 5999b94acceSBarry Smith void *ctx) 6009b94acceSBarry Smith { 60177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 6029b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 60348d91487SBarry Smith "SNESSetMinimizationFunction:Only for SNES_UNCONSTRAINED_MINIMIZATION"); 6049b94acceSBarry Smith snes->computeumfunction = func; 6059b94acceSBarry Smith snes->umfunP = ctx; 6069b94acceSBarry Smith return 0; 6079b94acceSBarry Smith } 6089b94acceSBarry Smith 6099b94acceSBarry Smith /*@ 6109b94acceSBarry Smith SNESComputeMinimizationFunction - Computes the function that has been 6119b94acceSBarry Smith set with SNESSetMinimizationFunction(). 6129b94acceSBarry Smith 6139b94acceSBarry Smith Input Parameters: 6149b94acceSBarry Smith . snes - the SNES context 6159b94acceSBarry Smith . x - input vector 6169b94acceSBarry Smith 6179b94acceSBarry Smith Output Parameter: 6189b94acceSBarry Smith . y - function value 6199b94acceSBarry Smith 6209b94acceSBarry Smith Notes: 6219b94acceSBarry Smith SNESComputeMinimizationFunction() is valid only for 6229b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 6239b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 624a86d99e1SLois Curfman McInnes 625a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, minimization, function 626a86d99e1SLois Curfman McInnes 627a86d99e1SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESGetMinimizationFunction(), 628a86d99e1SLois Curfman McInnes SNESComputeGradient(), SNESComputeHessian() 6299b94acceSBarry Smith @*/ 6309b94acceSBarry Smith int SNESComputeMinimizationFunction(SNES snes,Vec x,double *y) 6319b94acceSBarry Smith { 6329b94acceSBarry Smith int ierr; 6339b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 63448d91487SBarry Smith "SNESComputeMinimizationFunction:Only for SNES_UNCONSTRAINED_MINIMIZATION"); 6359b94acceSBarry Smith PLogEventBegin(SNES_MinimizationFunctionEval,snes,x,y,0); 6369b94acceSBarry Smith ierr = (*snes->computeumfunction)(snes,x,y,snes->umfunP); CHKERRQ(ierr); 6379b94acceSBarry Smith PLogEventEnd(SNES_MinimizationFunctionEval,snes,x,y,0); 6389b94acceSBarry Smith return 0; 6399b94acceSBarry Smith } 6409b94acceSBarry Smith 6419b94acceSBarry Smith /*@C 6429b94acceSBarry Smith SNESSetGradient - Sets the gradient evaluation routine and gradient 6439b94acceSBarry Smith vector for use by the SNES routines. 6449b94acceSBarry Smith 6459b94acceSBarry Smith Input Parameters: 6469b94acceSBarry Smith . snes - the SNES context 6479b94acceSBarry Smith . func - function evaluation routine 6489b94acceSBarry Smith . ctx - optional user-defined function context 6499b94acceSBarry Smith . r - vector to store gradient value 6509b94acceSBarry Smith 6519b94acceSBarry Smith Calling sequence of func: 6529b94acceSBarry Smith . func (SNES, Vec x, Vec g, void *ctx); 6539b94acceSBarry Smith 6549b94acceSBarry Smith . x - input vector 6559b94acceSBarry Smith . g - gradient vector 6569b94acceSBarry Smith . ctx - optional user-defined context for private data for the 6579b94acceSBarry Smith function evaluation routine (may be null) 6589b94acceSBarry Smith 6599b94acceSBarry Smith Notes: 6609b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 6619b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 6629b94acceSBarry Smith SNESSetFunction(). 6639b94acceSBarry Smith 6649b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 6659b94acceSBarry Smith 666a86d99e1SLois Curfman McInnes .seealso: SNESGetGradient(), SNESComputeGradient(), SNESSetHessian(), 667a86d99e1SLois Curfman McInnes SNESSetMinimizationFunction(), 6689b94acceSBarry Smith @*/ 6699b94acceSBarry Smith int SNESSetGradient(SNES snes,Vec r,int (*func)(SNES,Vec,Vec,void*), 6709b94acceSBarry Smith void *ctx) 6719b94acceSBarry Smith { 67277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 6739b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 67448d91487SBarry Smith "SNESSetGradient:For SNES_UNCONSTRAINED_MINIMIZATION only"); 6759b94acceSBarry Smith snes->computefunction = func; 6769b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 6779b94acceSBarry Smith snes->funP = ctx; 6789b94acceSBarry Smith return 0; 6799b94acceSBarry Smith } 6809b94acceSBarry Smith 6819b94acceSBarry Smith /*@ 682a86d99e1SLois Curfman McInnes SNESComputeGradient - Computes the gradient that has been set with 683a86d99e1SLois Curfman McInnes SNESSetGradient(). 6849b94acceSBarry Smith 6859b94acceSBarry Smith Input Parameters: 6869b94acceSBarry Smith . snes - the SNES context 6879b94acceSBarry Smith . x - input vector 6889b94acceSBarry Smith 6899b94acceSBarry Smith Output Parameter: 6909b94acceSBarry Smith . y - gradient vector 6919b94acceSBarry Smith 6929b94acceSBarry Smith Notes: 6939b94acceSBarry Smith SNESComputeGradient() is valid only for 6949b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 6959b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 696a86d99e1SLois Curfman McInnes 697a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, gradient 698a86d99e1SLois Curfman McInnes 699a86d99e1SLois Curfman McInnes .seealso: SNESSetGradient(), SNESGetGradient(), 700a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction(), SNESComputeHessian() 7019b94acceSBarry Smith @*/ 7029b94acceSBarry Smith int SNESComputeGradient(SNES snes,Vec x, Vec y) 7039b94acceSBarry Smith { 7049b94acceSBarry Smith int ierr; 7059b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 70648d91487SBarry Smith "SNESComputeGradient:For SNES_UNCONSTRAINED_MINIMIZATION only"); 7079b94acceSBarry Smith PLogEventBegin(SNES_GradientEval,snes,x,y,0); 7089b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 7099b94acceSBarry Smith PLogEventEnd(SNES_GradientEval,snes,x,y,0); 7109b94acceSBarry Smith return 0; 7119b94acceSBarry Smith } 7129b94acceSBarry Smith 71362fef451SLois Curfman McInnes /*@ 71462fef451SLois Curfman McInnes SNESComputeJacobian - Computes the Jacobian matrix that has been 71562fef451SLois Curfman McInnes set with SNESSetJacobian(). 71662fef451SLois Curfman McInnes 71762fef451SLois Curfman McInnes Input Parameters: 71862fef451SLois Curfman McInnes . snes - the SNES context 71962fef451SLois Curfman McInnes . x - input vector 72062fef451SLois Curfman McInnes 72162fef451SLois Curfman McInnes Output Parameters: 72262fef451SLois Curfman McInnes . A - Jacobian matrix 72362fef451SLois Curfman McInnes . B - optional preconditioning matrix 72462fef451SLois Curfman McInnes . flag - flag indicating matrix structure 72562fef451SLois Curfman McInnes 72662fef451SLois Curfman McInnes Notes: 72762fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 72862fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 72962fef451SLois Curfman McInnes 73062fef451SLois Curfman McInnes See SLESSetOperators() for information about setting the flag parameter. 73162fef451SLois Curfman McInnes 73262fef451SLois Curfman McInnes SNESComputeJacobian() is valid only for SNES_NONLINEAR_EQUATIONS 73362fef451SLois Curfman McInnes methods. An analogous routine for SNES_UNCONSTRAINED_MINIMIZATION 73462fef451SLois Curfman McInnes methods is SNESComputeJacobian(). 73562fef451SLois Curfman McInnes 73662fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 73762fef451SLois Curfman McInnes 73862fef451SLois Curfman McInnes .seealso: SNESSetJacobian(), SLESSetOperators() 73962fef451SLois Curfman McInnes @*/ 7409b94acceSBarry Smith int SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 7419b94acceSBarry Smith { 7429b94acceSBarry Smith int ierr; 7439b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 74448d91487SBarry Smith "SNESComputeJacobian: For SNES_NONLINEAR_EQUATIONS only"); 7459b94acceSBarry Smith if (!snes->computejacobian) return 0; 7469b94acceSBarry Smith PLogEventBegin(SNES_JacobianEval,snes,X,*A,*B); 747c4fc05e7SBarry Smith *flg = DIFFERENT_NONZERO_PATTERN; 7489b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 7499b94acceSBarry Smith PLogEventEnd(SNES_JacobianEval,snes,X,*A,*B); 7506d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 75177c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 75277c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 7539b94acceSBarry Smith return 0; 7549b94acceSBarry Smith } 7559b94acceSBarry Smith 75662fef451SLois Curfman McInnes /*@ 75762fef451SLois Curfman McInnes SNESComputeHessian - Computes the Hessian matrix that has been 75862fef451SLois Curfman McInnes set with SNESSetHessian(). 75962fef451SLois Curfman McInnes 76062fef451SLois Curfman McInnes Input Parameters: 76162fef451SLois Curfman McInnes . snes - the SNES context 76262fef451SLois Curfman McInnes . x - input vector 76362fef451SLois Curfman McInnes 76462fef451SLois Curfman McInnes Output Parameters: 76562fef451SLois Curfman McInnes . A - Hessian matrix 76662fef451SLois Curfman McInnes . B - optional preconditioning matrix 76762fef451SLois Curfman McInnes . flag - flag indicating matrix structure 76862fef451SLois Curfman McInnes 76962fef451SLois Curfman McInnes Notes: 77062fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 77162fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 77262fef451SLois Curfman McInnes 77362fef451SLois Curfman McInnes See SLESSetOperators() for information about setting the flag parameter. 77462fef451SLois Curfman McInnes 77562fef451SLois Curfman McInnes SNESComputeHessian() is valid only for 77662fef451SLois Curfman McInnes SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 77762fef451SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods is SNESComputeJacobian(). 77862fef451SLois Curfman McInnes 77962fef451SLois Curfman McInnes .keywords: SNES, compute, Hessian, matrix 78062fef451SLois Curfman McInnes 781a86d99e1SLois Curfman McInnes .seealso: SNESSetHessian(), SLESSetOperators(), SNESComputeGradient(), 782a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction() 78362fef451SLois Curfman McInnes @*/ 78462fef451SLois Curfman McInnes int SNESComputeHessian(SNES snes,Vec x,Mat *A,Mat *B,MatStructure *flag) 7859b94acceSBarry Smith { 7869b94acceSBarry Smith int ierr; 7879b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 78848d91487SBarry Smith "SNESComputeHessian:For SNES_UNCONSTRAINED_MINIMIZATION only"); 7899b94acceSBarry Smith if (!snes->computejacobian) return 0; 79062fef451SLois Curfman McInnes PLogEventBegin(SNES_HessianEval,snes,x,*A,*B); 7910f4a323eSLois Curfman McInnes *flag = DIFFERENT_NONZERO_PATTERN; 79262fef451SLois Curfman McInnes ierr = (*snes->computejacobian)(snes,x,A,B,flag,snes->jacP); CHKERRQ(ierr); 79362fef451SLois Curfman McInnes PLogEventEnd(SNES_HessianEval,snes,x,*A,*B); 7940f4a323eSLois Curfman McInnes /* make sure user returned a correct Jacobian and preconditioner */ 79577c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 79677c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 7979b94acceSBarry Smith return 0; 7989b94acceSBarry Smith } 7999b94acceSBarry Smith 8009b94acceSBarry Smith /*@C 8019b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 8029b94acceSBarry Smith location to store it. 8039b94acceSBarry Smith 8049b94acceSBarry Smith Input Parameters: 8059b94acceSBarry Smith . snes - the SNES context 8069b94acceSBarry Smith . A - Jacobian matrix 8079b94acceSBarry Smith . B - preconditioner matrix (usually same as the Jacobian) 8089b94acceSBarry Smith . func - Jacobian evaluation routine 8099b94acceSBarry Smith . ctx - optional user-defined context for private data for the 8109b94acceSBarry Smith Jacobian evaluation routine (may be null) 8119b94acceSBarry Smith 8129b94acceSBarry Smith Calling sequence of func: 8139b94acceSBarry Smith . func (SNES,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 8149b94acceSBarry Smith 8159b94acceSBarry Smith . x - input vector 8169b94acceSBarry Smith . A - Jacobian matrix 8179b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 818ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 819ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 8209b94acceSBarry Smith . ctx - optional user-defined Jacobian context 8219b94acceSBarry Smith 8229b94acceSBarry Smith Notes: 823ac21db08SLois Curfman McInnes See SLESSetOperators() for information about setting the flag input 824ac21db08SLois Curfman McInnes parameter in the routine func(). Be sure to read this information! 825ac21db08SLois Curfman McInnes 826ac21db08SLois Curfman McInnes The routine func() takes Mat * as the matrix arguments rather than Mat. 8279b94acceSBarry Smith This allows the Jacobian evaluation routine to replace A and/or B with a 8289b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 8299b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 8309b94acceSBarry Smith throughout the global iterations. 8319b94acceSBarry Smith 8329b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 8339b94acceSBarry Smith 834ac21db08SLois Curfman McInnes .seealso: SLESSetOperators(), SNESSetFunction() 8359b94acceSBarry Smith @*/ 8369b94acceSBarry Smith int SNESSetJacobian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 8379b94acceSBarry Smith MatStructure*,void*),void *ctx) 8389b94acceSBarry Smith { 83977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 8409b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 84148d91487SBarry Smith "SNESSetJacobian:For SNES_NONLINEAR_EQUATIONS only"); 8429b94acceSBarry Smith snes->computejacobian = func; 8439b94acceSBarry Smith snes->jacP = ctx; 8449b94acceSBarry Smith snes->jacobian = A; 8459b94acceSBarry Smith snes->jacobian_pre = B; 8469b94acceSBarry Smith return 0; 8479b94acceSBarry Smith } 84862fef451SLois Curfman McInnes 849b4fd4287SBarry Smith /*@ 850b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 851b4fd4287SBarry Smith provided context for evaluating the Jacobian. 852b4fd4287SBarry Smith 853b4fd4287SBarry Smith Input Parameter: 854b4fd4287SBarry Smith . snes - the nonlinear solver context 855b4fd4287SBarry Smith 856b4fd4287SBarry Smith Output Parameters: 857b4fd4287SBarry Smith . A - location to stash Jacobian matrix (or PETSC_NULL) 858b4fd4287SBarry Smith . B - location to stash preconditioner matrix (or PETSC_NULL) 859b4fd4287SBarry Smith . ctx - location to stash Jacobian ctx (or PETSC_NULL) 860b4fd4287SBarry Smith 861b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian() 862b4fd4287SBarry Smith @*/ 863b4fd4287SBarry Smith int SNESGetJacobian(SNES snes,Mat *A,Mat *B, void **ctx) 864b4fd4287SBarry Smith { 86562fef451SLois Curfman McInnes if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 86662fef451SLois Curfman McInnes "SNESSetJacobian:For SNES_NONLINEAR_EQUATIONS only"); 867b4fd4287SBarry Smith if (A) *A = snes->jacobian; 868b4fd4287SBarry Smith if (B) *B = snes->jacobian_pre; 869b4fd4287SBarry Smith if (ctx) *ctx = snes->jacP; 870b4fd4287SBarry Smith return 0; 871b4fd4287SBarry Smith } 872b4fd4287SBarry Smith 8739b94acceSBarry Smith /*@C 8749b94acceSBarry Smith SNESSetHessian - Sets the function to compute Hessian as well as the 8759b94acceSBarry Smith location to store it. 8769b94acceSBarry Smith 8779b94acceSBarry Smith Input Parameters: 8789b94acceSBarry Smith . snes - the SNES context 8799b94acceSBarry Smith . A - Hessian matrix 8809b94acceSBarry Smith . B - preconditioner matrix (usually same as the Hessian) 8819b94acceSBarry Smith . func - Jacobian evaluation routine 8829b94acceSBarry Smith . ctx - optional user-defined context for private data for the 8839b94acceSBarry Smith Hessian evaluation routine (may be null) 8849b94acceSBarry Smith 8859b94acceSBarry Smith Calling sequence of func: 8869b94acceSBarry Smith . func (SNES,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 8879b94acceSBarry Smith 8889b94acceSBarry Smith . x - input vector 8899b94acceSBarry Smith . A - Hessian matrix 8909b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 891ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 892ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 8939b94acceSBarry Smith . ctx - optional user-defined Hessian context 8949b94acceSBarry Smith 8959b94acceSBarry Smith Notes: 896ac21db08SLois Curfman McInnes See SLESSetOperators() for information about setting the flag input 897ac21db08SLois Curfman McInnes parameter in the routine func(). Be sure to read this information! 898ac21db08SLois Curfman McInnes 8999b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 9009b94acceSBarry Smith This allows the Hessian evaluation routine to replace A and/or B with a 9019b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 9029b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 9039b94acceSBarry Smith throughout the global iterations. 9049b94acceSBarry Smith 9059b94acceSBarry Smith .keywords: SNES, nonlinear, set, Hessian, matrix 9069b94acceSBarry Smith 907ac21db08SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESSetGradient(), SLESSetOperators() 9089b94acceSBarry Smith @*/ 9099b94acceSBarry Smith int SNESSetHessian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 9109b94acceSBarry Smith MatStructure*,void*),void *ctx) 9119b94acceSBarry Smith { 91277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 9139b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 91448d91487SBarry Smith "SNESSetHessian:For SNES_UNCONSTRAINED_MINIMIZATION only"); 9159b94acceSBarry Smith snes->computejacobian = func; 9169b94acceSBarry Smith snes->jacP = ctx; 9179b94acceSBarry Smith snes->jacobian = A; 9189b94acceSBarry Smith snes->jacobian_pre = B; 9199b94acceSBarry Smith return 0; 9209b94acceSBarry Smith } 9219b94acceSBarry Smith 92262fef451SLois Curfman McInnes /*@ 92362fef451SLois Curfman McInnes SNESGetHessian - Returns the Hessian matrix and optionally the user 92462fef451SLois Curfman McInnes provided context for evaluating the Hessian. 92562fef451SLois Curfman McInnes 92662fef451SLois Curfman McInnes Input Parameter: 92762fef451SLois Curfman McInnes . snes - the nonlinear solver context 92862fef451SLois Curfman McInnes 92962fef451SLois Curfman McInnes Output Parameters: 93062fef451SLois Curfman McInnes . A - location to stash Hessian matrix (or PETSC_NULL) 93162fef451SLois Curfman McInnes . B - location to stash preconditioner matrix (or PETSC_NULL) 93262fef451SLois Curfman McInnes . ctx - location to stash Hessian ctx (or PETSC_NULL) 93362fef451SLois Curfman McInnes 93462fef451SLois Curfman McInnes .seealso: SNESSetHessian(), SNESComputeHessian() 93562fef451SLois Curfman McInnes @*/ 93662fef451SLois Curfman McInnes int SNESGetHessian(SNES snes,Mat *A,Mat *B, void **ctx) 93762fef451SLois Curfman McInnes { 93862fef451SLois Curfman McInnes if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 93962fef451SLois Curfman McInnes "SNESSetHessian:For SNES_UNCONSTRAINED_MINIMIZATION only"); 94062fef451SLois Curfman McInnes if (A) *A = snes->jacobian; 94162fef451SLois Curfman McInnes if (B) *B = snes->jacobian_pre; 94262fef451SLois Curfman McInnes if (ctx) *ctx = snes->jacP; 94362fef451SLois Curfman McInnes return 0; 94462fef451SLois Curfman McInnes } 94562fef451SLois Curfman McInnes 9469b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */ 9479b94acceSBarry Smith 9489b94acceSBarry Smith /*@ 9499b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 950272ac6f2SLois Curfman McInnes of a nonlinear solver. 9519b94acceSBarry Smith 9529b94acceSBarry Smith Input Parameter: 9539b94acceSBarry Smith . snes - the SNES context 9548ddd3da0SLois Curfman McInnes . x - the solution vector 9559b94acceSBarry Smith 956272ac6f2SLois Curfman McInnes Notes: 957272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 958272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 959272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 960272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 961272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 962272ac6f2SLois Curfman McInnes 9639b94acceSBarry Smith .keywords: SNES, nonlinear, setup 9649b94acceSBarry Smith 9659b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 9669b94acceSBarry Smith @*/ 9678ddd3da0SLois Curfman McInnes int SNESSetUp(SNES snes,Vec x) 9689b94acceSBarry Smith { 969272ac6f2SLois Curfman McInnes int ierr, flg; 97077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 97177c4ece6SBarry Smith PetscValidHeaderSpecific(x,VEC_COOKIE); 9728ddd3da0SLois Curfman McInnes snes->vec_sol = snes->vec_sol_always = x; 9739b94acceSBarry Smith 974272ac6f2SLois Curfman McInnes ierr = OptionsHasName(snes->prefix,"-snes_mf", &flg); CHKERRQ(ierr); 97531615d04SLois Curfman McInnes if (flg) { 976272ac6f2SLois Curfman McInnes Mat J; 977272ac6f2SLois Curfman McInnes ierr = SNESDefaultMatrixFreeMatCreate(snes,snes->vec_sol,&J);CHKERRQ(ierr); 978272ac6f2SLois Curfman McInnes PLogObjectParent(snes,J); 979272ac6f2SLois Curfman McInnes snes->mfshell = J; 98083e56afdSLois Curfman McInnes if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 98183e56afdSLois Curfman McInnes ierr = SNESSetJacobian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 98283e56afdSLois Curfman McInnes PLogInfo((PetscObject)snes,"SNESSetUp: Setting default matrix-free Jacobian routines\n"); 98383e56afdSLois Curfman McInnes } 98483e56afdSLois Curfman McInnes else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 98583e56afdSLois Curfman McInnes ierr = SNESSetHessian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 98683e56afdSLois Curfman McInnes PLogInfo((PetscObject)snes,"SNESSetUp: Setting default matrix-free Hessian routines\n"); 98783e56afdSLois Curfman McInnes } else SETERRQ(1,"SNESSetUp:Method class doesn't support matrix-free option"); 988272ac6f2SLois Curfman McInnes } 9899b94acceSBarry Smith if ((snes->method_class == SNES_NONLINEAR_EQUATIONS)) { 99037fcc0dbSBarry Smith if (!snes->vec_func) SETERRQ(1,"SNESSetUp:Must call SNESSetFunction() first"); 99137fcc0dbSBarry Smith if (!snes->computefunction) SETERRQ(1,"SNESSetUp:Must call SNESSetFunction() first"); 99237fcc0dbSBarry Smith if (!snes->jacobian) SETERRQ(1,"SNESSetUp:Must call SNESSetJacobian() first"); 993d804570eSBarry Smith if (snes->vec_func == snes->vec_sol) SETERRQ(1,"SNESSetUp:Solution vector cannot be function vector"); 994a703fe33SLois Curfman McInnes 995a703fe33SLois Curfman McInnes /* Set the KSP stopping criterion to use the Eisenstat-Walker method */ 996a703fe33SLois Curfman McInnes if (snes->ksp_ewconv && snes->type != SNES_EQ_NTR) { 997a703fe33SLois Curfman McInnes SLES sles; KSP ksp; 998a703fe33SLois Curfman McInnes ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 999a703fe33SLois Curfman McInnes ierr = SLESGetKSP(sles,&ksp); CHKERRQ(ierr); 1000a703fe33SLois Curfman McInnes ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private, 1001a703fe33SLois Curfman McInnes (void *)snes); CHKERRQ(ierr); 1002a703fe33SLois Curfman McInnes } 10039b94acceSBarry Smith } else if ((snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION)) { 100437fcc0dbSBarry Smith if (!snes->vec_func) SETERRQ(1,"SNESSetUp:Must call SNESSetGradient() first"); 100537fcc0dbSBarry Smith if (!snes->computefunction) SETERRQ(1,"SNESSetUp:Must call SNESSetGradient() first"); 100637fcc0dbSBarry Smith if (!snes->computeumfunction) 100737fcc0dbSBarry Smith SETERRQ(1,"SNESSetUp:Must call SNESSetMinimizationFunction() first"); 100837fcc0dbSBarry Smith if (!snes->jacobian) SETERRQ(1,"SNESSetUp:Must call SNESSetHessian() first"); 10099b94acceSBarry Smith } else SETERRQ(1,"SNESSetUp:Unknown method class"); 1010a703fe33SLois Curfman McInnes if (snes->setup) {ierr = (*snes->setup)(snes); CHKERRQ(ierr);} 1011a703fe33SLois Curfman McInnes snes->setup_called = 1; 1012a703fe33SLois Curfman McInnes return 0; 10139b94acceSBarry Smith } 10149b94acceSBarry Smith 10159b94acceSBarry Smith /*@C 10169b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 10179b94acceSBarry Smith with SNESCreate(). 10189b94acceSBarry Smith 10199b94acceSBarry Smith Input Parameter: 10209b94acceSBarry Smith . snes - the SNES context 10219b94acceSBarry Smith 10229b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 10239b94acceSBarry Smith 102463a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 10259b94acceSBarry Smith @*/ 10269b94acceSBarry Smith int SNESDestroy(SNES snes) 10279b94acceSBarry Smith { 10289b94acceSBarry Smith int ierr; 102977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 10309b94acceSBarry Smith ierr = (*(snes)->destroy)((PetscObject)snes); CHKERRQ(ierr); 10310452661fSBarry Smith if (snes->kspconvctx) PetscFree(snes->kspconvctx); 10329b94acceSBarry Smith if (snes->mfshell) MatDestroy(snes->mfshell); 10339b94acceSBarry Smith ierr = SLESDestroy(snes->sles); CHKERRQ(ierr); 10349b94acceSBarry Smith PLogObjectDestroy((PetscObject)snes); 10350452661fSBarry Smith PetscHeaderDestroy((PetscObject)snes); 10369b94acceSBarry Smith return 0; 10379b94acceSBarry Smith } 10389b94acceSBarry Smith 10399b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 10409b94acceSBarry Smith 10419b94acceSBarry Smith /*@ 1042*d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 10439b94acceSBarry Smith 10449b94acceSBarry Smith Input Parameters: 10459b94acceSBarry Smith . snes - the SNES context 1046*d7a720efSLois Curfman McInnes . atol - tolerance 10479b94acceSBarry Smith 10489b94acceSBarry Smith Options Database Key: 1049*d7a720efSLois Curfman McInnes $ -snes_atol <atol> - absolute convergence tolerance 1050*d7a720efSLois Curfman McInnes $ -snes_rtol <rtol> - relative convergence tolerance 1051*d7a720efSLois Curfman McInnes $ -snes_stol <stol> - convergence tolerance in terms of 1052*d7a720efSLois Curfman McInnes $ the norm of the change in the solution between steps 1053*d7a720efSLois Curfman McInnes $ -snes_max_it <maxit> - maximum number of iterations 1054*d7a720efSLois Curfman McInnes $ -snes_max_funcs <maxf> - maximum number of function evaluations 10559b94acceSBarry Smith 1056*d7a720efSLois Curfman McInnes Notes: 10579b94acceSBarry Smith The default maximum number of iterations is 50. 10589b94acceSBarry Smith The default maximum number of function evaluations is 1000. 10599b94acceSBarry Smith 1060*d7a720efSLois Curfman McInnes .keywords: SNES, nonlinear, set, absolute, convergence, tolerance 10619b94acceSBarry Smith 1062*d7a720efSLois Curfman McInnes .seealso: SNESSetTrustRegionTolerance(), SNESSetMinimizationFunctionTolerance() 10639b94acceSBarry Smith @*/ 1064*d7a720efSLois Curfman McInnes int SNESSetTolerances(SNES snes,double atol,double rtol,double stol,int maxit,int maxf) 10659b94acceSBarry Smith { 106677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1067*d7a720efSLois Curfman McInnes if (atol != PETSC_DEFAULT) snes->atol = atol; 1068*d7a720efSLois Curfman McInnes if (rtol != PETSC_DEFAULT) snes->rtol = rtol; 1069*d7a720efSLois Curfman McInnes if (stol != PETSC_DEFAULT) snes->xtol = stol; 1070*d7a720efSLois Curfman McInnes if (maxit != PETSC_DEFAULT) snes->max_its = maxit; 1071*d7a720efSLois Curfman McInnes if (maxf != PETSC_DEFAULT) snes->max_funcs = maxf; 10729b94acceSBarry Smith return 0; 10739b94acceSBarry Smith } 10749b94acceSBarry Smith 10759b94acceSBarry Smith /*@ 10769b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 10779b94acceSBarry Smith 10789b94acceSBarry Smith Input Parameters: 10799b94acceSBarry Smith . snes - the SNES context 10809b94acceSBarry Smith . tol - tolerance 10819b94acceSBarry Smith 10829b94acceSBarry Smith Options Database Key: 1083*d7a720efSLois Curfman McInnes $ -snes_trtol <tol> 10849b94acceSBarry Smith 10859b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 10869b94acceSBarry Smith 1087*d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetMinimizationFunctionTolerance() 10889b94acceSBarry Smith @*/ 10899b94acceSBarry Smith int SNESSetTrustRegionTolerance(SNES snes,double tol) 10909b94acceSBarry Smith { 109177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 10929b94acceSBarry Smith snes->deltatol = tol; 10939b94acceSBarry Smith return 0; 10949b94acceSBarry Smith } 10959b94acceSBarry Smith 10969b94acceSBarry Smith /*@ 109777c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance - Sets the minimum allowable function tolerance 10989b94acceSBarry Smith for unconstrained minimization solvers. 10999b94acceSBarry Smith 11009b94acceSBarry Smith Input Parameters: 11019b94acceSBarry Smith . snes - the SNES context 11029b94acceSBarry Smith . ftol - minimum function tolerance 11039b94acceSBarry Smith 11049b94acceSBarry Smith Options Database Key: 1105*d7a720efSLois Curfman McInnes $ -snes_fmin <ftol> 11069b94acceSBarry Smith 11079b94acceSBarry Smith Note: 110877c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance() is valid for SNES_UNCONSTRAINED_MINIMIZATION 11099b94acceSBarry Smith methods only. 11109b94acceSBarry Smith 11119b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimum, convergence, function, tolerance 11129b94acceSBarry Smith 1113*d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetTrustRegionTolerance() 11149b94acceSBarry Smith @*/ 111577c4ece6SBarry Smith int SNESSetMinimizationFunctionTolerance(SNES snes,double ftol) 11169b94acceSBarry Smith { 111777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 11189b94acceSBarry Smith snes->fmin = ftol; 11199b94acceSBarry Smith return 0; 11209b94acceSBarry Smith } 11219b94acceSBarry Smith 11229b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 11239b94acceSBarry Smith 11249b94acceSBarry Smith /*@C 11259b94acceSBarry Smith SNESSetMonitor - Sets the function that is to be used at every 11269b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 11279b94acceSBarry Smith progress. 11289b94acceSBarry Smith 11299b94acceSBarry Smith Input Parameters: 11309b94acceSBarry Smith . snes - the SNES context 11319b94acceSBarry Smith . func - monitoring routine 11329b94acceSBarry Smith . mctx - optional user-defined context for private data for the 11339b94acceSBarry Smith monitor routine (may be null) 11349b94acceSBarry Smith 11359b94acceSBarry Smith Calling sequence of func: 1136682d7d0cSBarry Smith int func(SNES snes,int its, Vec x,Vec f,double norm,void *mctx) 11379b94acceSBarry Smith 11389b94acceSBarry Smith $ snes - the SNES context 11399b94acceSBarry Smith $ its - iteration number 11409b94acceSBarry Smith $ mctx - optional monitoring context 11419b94acceSBarry Smith $ 11429b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods: 11439b94acceSBarry Smith $ norm - 2-norm function value (may be estimated) 11449b94acceSBarry Smith $ 11459b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods: 11469b94acceSBarry Smith $ norm - 2-norm gradient value (may be estimated) 11479b94acceSBarry Smith 11489b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 11499b94acceSBarry Smith 11509b94acceSBarry Smith .seealso: SNESDefaultMonitor() 11519b94acceSBarry Smith @*/ 11529b94acceSBarry Smith int SNESSetMonitor( SNES snes, int (*func)(SNES,int,double,void*), 11539b94acceSBarry Smith void *mctx ) 11549b94acceSBarry Smith { 11559b94acceSBarry Smith snes->monitor = func; 11569b94acceSBarry Smith snes->monP = (void*)mctx; 11579b94acceSBarry Smith return 0; 11589b94acceSBarry Smith } 11599b94acceSBarry Smith 11609b94acceSBarry Smith /*@C 11619b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 11629b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 11639b94acceSBarry Smith 11649b94acceSBarry Smith Input Parameters: 11659b94acceSBarry Smith . snes - the SNES context 11669b94acceSBarry Smith . func - routine to test for convergence 11679b94acceSBarry Smith . cctx - optional context for private data for the convergence routine 11689b94acceSBarry Smith (may be null) 11699b94acceSBarry Smith 11709b94acceSBarry Smith Calling sequence of func: 11719b94acceSBarry Smith int func (SNES snes,double xnorm,double gnorm, 11729b94acceSBarry Smith double f,void *cctx) 11739b94acceSBarry Smith 11749b94acceSBarry Smith $ snes - the SNES context 11759b94acceSBarry Smith $ cctx - optional convergence context 11769b94acceSBarry Smith $ xnorm - 2-norm of current iterate 11779b94acceSBarry Smith $ 11789b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods: 11799b94acceSBarry Smith $ gnorm - 2-norm of current step 11809b94acceSBarry Smith $ f - 2-norm of function 11819b94acceSBarry Smith $ 11829b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods: 11839b94acceSBarry Smith $ gnorm - 2-norm of current gradient 11849b94acceSBarry Smith $ f - function value 11859b94acceSBarry Smith 11869b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 11879b94acceSBarry Smith 11889b94acceSBarry Smith .seealso: SNESDefaultConverged() 11899b94acceSBarry Smith @*/ 11909b94acceSBarry Smith int SNESSetConvergenceTest(SNES snes, 11919b94acceSBarry Smith int (*func)(SNES,double,double,double,void*),void *cctx) 11929b94acceSBarry Smith { 11939b94acceSBarry Smith (snes)->converged = func; 11949b94acceSBarry Smith (snes)->cnvP = cctx; 11959b94acceSBarry Smith return 0; 11969b94acceSBarry Smith } 11979b94acceSBarry Smith 11989b94acceSBarry Smith /* 11999b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 12009b94acceSBarry Smith positive parameter delta. 12019b94acceSBarry Smith 12029b94acceSBarry Smith Input Parameters: 12039b94acceSBarry Smith . snes - the SNES context 12049b94acceSBarry Smith . y - approximate solution of linear system 12059b94acceSBarry Smith . fnorm - 2-norm of current function 12069b94acceSBarry Smith . delta - trust region size 12079b94acceSBarry Smith 12089b94acceSBarry Smith Output Parameters: 12099b94acceSBarry Smith . gpnorm - predicted function norm at the new point, assuming local 12109b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 12119b94acceSBarry Smith region, and exceeds zero otherwise. 12129b94acceSBarry Smith . ynorm - 2-norm of the step 12139b94acceSBarry Smith 12149b94acceSBarry Smith Note: 1215ddd12767SBarry Smith For non-trust region methods such as SNES_EQ_NLS, the parameter delta 12169b94acceSBarry Smith is set to be the maximum allowable step size. 12179b94acceSBarry Smith 12189b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 12199b94acceSBarry Smith */ 12209b94acceSBarry Smith int SNESScaleStep_Private(SNES snes,Vec y,double *fnorm,double *delta, 12219b94acceSBarry Smith double *gpnorm,double *ynorm) 12229b94acceSBarry Smith { 12239b94acceSBarry Smith double norm; 12249b94acceSBarry Smith Scalar cnorm; 1225cddf8d76SBarry Smith VecNorm(y,NORM_2, &norm ); 12269b94acceSBarry Smith if (norm > *delta) { 12279b94acceSBarry Smith norm = *delta/norm; 12289b94acceSBarry Smith *gpnorm = (1.0 - norm)*(*fnorm); 12299b94acceSBarry Smith cnorm = norm; 12309b94acceSBarry Smith VecScale( &cnorm, y ); 12319b94acceSBarry Smith *ynorm = *delta; 12329b94acceSBarry Smith } else { 12339b94acceSBarry Smith *gpnorm = 0.0; 12349b94acceSBarry Smith *ynorm = norm; 12359b94acceSBarry Smith } 12369b94acceSBarry Smith return 0; 12379b94acceSBarry Smith } 12389b94acceSBarry Smith 12399b94acceSBarry Smith /*@ 12409b94acceSBarry Smith SNESSolve - Solves a nonlinear system. Call SNESSolve after calling 124163a78c88SLois Curfman McInnes SNESCreate() and optional routines of the form SNESSetXXX(). 12429b94acceSBarry Smith 12439b94acceSBarry Smith Input Parameter: 12449b94acceSBarry Smith . snes - the SNES context 12458ddd3da0SLois Curfman McInnes . x - the solution vector 12469b94acceSBarry Smith 12479b94acceSBarry Smith Output Parameter: 12489b94acceSBarry Smith its - number of iterations until termination 12499b94acceSBarry Smith 12508ddd3da0SLois Curfman McInnes Note: 12518ddd3da0SLois Curfman McInnes The user should initialize the vector, x, with the initial guess 12528ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 12538ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 12548ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 12558ddd3da0SLois Curfman McInnes 12569b94acceSBarry Smith .keywords: SNES, nonlinear, solve 12579b94acceSBarry Smith 125863a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESDestroy() 12599b94acceSBarry Smith @*/ 12608ddd3da0SLois Curfman McInnes int SNESSolve(SNES snes,Vec x,int *its) 12619b94acceSBarry Smith { 12623c7409f5SSatish Balay int ierr, flg; 1263052efed2SBarry Smith 126477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1265c4fc05e7SBarry Smith if (!snes->setup_called) {ierr = SNESSetUp(snes,x); CHKERRQ(ierr);} 1266c4fc05e7SBarry Smith else {snes->vec_sol = snes->vec_sol_always = x;} 12679b94acceSBarry Smith PLogEventBegin(SNES_Solve,snes,0,0,0); 12689b94acceSBarry Smith ierr = (*(snes)->solve)(snes,its); CHKERRQ(ierr); 12699b94acceSBarry Smith PLogEventEnd(SNES_Solve,snes,0,0,0); 12703c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-snes_view", &flg); CHKERRQ(ierr); 12713c7409f5SSatish Balay if (flg) { ierr = SNESView(snes,STDOUT_VIEWER_WORLD); CHKERRQ(ierr); } 12729b94acceSBarry Smith return 0; 12739b94acceSBarry Smith } 12749b94acceSBarry Smith 12759b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 127637fcc0dbSBarry Smith static NRList *__SNESList = 0; 12779b94acceSBarry Smith 12789b94acceSBarry Smith /*@ 12794b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 12809b94acceSBarry Smith 12819b94acceSBarry Smith Input Parameters: 12829b94acceSBarry Smith . snes - the SNES context 12839b94acceSBarry Smith . method - a known method 12849b94acceSBarry Smith 12859b94acceSBarry Smith Notes: 12869b94acceSBarry Smith See "petsc/include/snes.h" for available methods (for instance) 12879b94acceSBarry Smith $ Systems of nonlinear equations: 1288ddd12767SBarry Smith $ SNES_EQ_NLS - Newton's method with line search 1289ddd12767SBarry Smith $ SNES_EQ_NTR - Newton's method with trust region 12909b94acceSBarry Smith $ Unconstrained minimization: 12919b94acceSBarry Smith $ SNES_UM_NTR - Newton's method with trust region 1292a703fe33SLois Curfman McInnes $ SNES_UM_NLS - Newton's method with line search 12939b94acceSBarry Smith 12949b94acceSBarry Smith Options Database Command: 12954b0e389bSBarry Smith $ -snes_type <method> 12969b94acceSBarry Smith $ Use -help for a list of available methods 12979b94acceSBarry Smith $ (for instance, ls or tr) 1298a703fe33SLois Curfman McInnes 1299a703fe33SLois Curfman McInnes .keysords: SNES, set, method 13009b94acceSBarry Smith @*/ 13014b0e389bSBarry Smith int SNESSetType(SNES snes,SNESType method) 13029b94acceSBarry Smith { 13039b94acceSBarry Smith int (*r)(SNES); 130477c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 13059b94acceSBarry Smith /* Get the function pointers for the iterative method requested */ 130637fcc0dbSBarry Smith if (!__SNESList) {SNESRegisterAll();} 130737fcc0dbSBarry Smith if (!__SNESList) {SETERRQ(1,"SNESSetType:Could not get methods");} 130837fcc0dbSBarry Smith r = (int (*)(SNES))NRFindRoutine( __SNESList, (int)method, (char *)0 ); 13094b0e389bSBarry Smith if (!r) {SETERRQ(1,"SNESSetType:Unknown method");} 13100452661fSBarry Smith if (snes->data) PetscFree(snes->data); 13119b94acceSBarry Smith snes->set_method_called = 1; 13129b94acceSBarry Smith return (*r)(snes); 13139b94acceSBarry Smith } 13149b94acceSBarry Smith 13159b94acceSBarry Smith /* --------------------------------------------------------------------- */ 13169b94acceSBarry Smith /*@C 13179b94acceSBarry Smith SNESRegister - Adds the method to the nonlinear solver package, given 13184b0e389bSBarry Smith a function pointer and a nonlinear solver name of the type SNESType. 13199b94acceSBarry Smith 13209b94acceSBarry Smith Input Parameters: 1321ddd12767SBarry Smith . name - for instance SNES_EQ_NLS, SNES_EQ_NTR, ... 13229b94acceSBarry Smith . sname - corfunPonding string for name 13239b94acceSBarry Smith . create - routine to create method context 13249b94acceSBarry Smith 13259b94acceSBarry Smith .keywords: SNES, nonlinear, register 13269b94acceSBarry Smith 13279b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy() 13289b94acceSBarry Smith @*/ 13299b94acceSBarry Smith int SNESRegister(int name, char *sname, int (*create)(SNES)) 13309b94acceSBarry Smith { 13319b94acceSBarry Smith int ierr; 133237fcc0dbSBarry Smith if (!__SNESList) {ierr = NRCreate(&__SNESList); CHKERRQ(ierr);} 133337fcc0dbSBarry Smith NRRegister( __SNESList, name, sname, (int (*)(void*))create ); 13349b94acceSBarry Smith return 0; 13359b94acceSBarry Smith } 13369b94acceSBarry Smith /* --------------------------------------------------------------------- */ 13379b94acceSBarry Smith /*@C 13389b94acceSBarry Smith SNESRegisterDestroy - Frees the list of nonlinear solvers that were 13399b94acceSBarry Smith registered by SNESRegister(). 13409b94acceSBarry Smith 13419b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy 13429b94acceSBarry Smith 13439b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll() 13449b94acceSBarry Smith @*/ 13459b94acceSBarry Smith int SNESRegisterDestroy() 13469b94acceSBarry Smith { 134737fcc0dbSBarry Smith if (__SNESList) { 134837fcc0dbSBarry Smith NRDestroy( __SNESList ); 134937fcc0dbSBarry Smith __SNESList = 0; 13509b94acceSBarry Smith } 13519b94acceSBarry Smith return 0; 13529b94acceSBarry Smith } 13539b94acceSBarry Smith 13549b94acceSBarry Smith /* 13554b0e389bSBarry Smith SNESGetTypeFromOptions_Private - Sets the selected method from the 13569b94acceSBarry Smith options database. 13579b94acceSBarry Smith 13589b94acceSBarry Smith Input Parameter: 13599b94acceSBarry Smith . ctx - the SNES context 13609b94acceSBarry Smith 13619b94acceSBarry Smith Output Parameter: 13629b94acceSBarry Smith . method - solver method 13639b94acceSBarry Smith 13649b94acceSBarry Smith Returns: 13659b94acceSBarry Smith Returns 1 if the method is found; 0 otherwise. 13669b94acceSBarry Smith 13679b94acceSBarry Smith Options Database Key: 13684b0e389bSBarry Smith $ -snes_type method 13699b94acceSBarry Smith */ 1370052efed2SBarry Smith int SNESGetTypeFromOptions_Private(SNES ctx,SNESType *method,int *flg) 13719b94acceSBarry Smith { 1372052efed2SBarry Smith int ierr; 13739b94acceSBarry Smith char sbuf[50]; 1374052efed2SBarry Smith ierr = OptionsGetString(ctx->prefix,"-snes_type", sbuf, 50, flg); CHKERRQ(ierr); 1375052efed2SBarry Smith if (*flg) { 1376052efed2SBarry Smith if (!__SNESList) {ierr = SNESRegisterAll(); CHKERRQ(ierr);} 137737fcc0dbSBarry Smith *method = (SNESType)NRFindID( __SNESList, sbuf ); 13789b94acceSBarry Smith } 13799b94acceSBarry Smith return 0; 13809b94acceSBarry Smith } 13819b94acceSBarry Smith 13829b94acceSBarry Smith /*@C 13839a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 13849b94acceSBarry Smith 13859b94acceSBarry Smith Input Parameter: 13864b0e389bSBarry Smith . snes - nonlinear solver context 13879b94acceSBarry Smith 13889b94acceSBarry Smith Output Parameter: 13899a28b0a6SLois Curfman McInnes . method - SNES method (or use PETSC_NULL) 13909a28b0a6SLois Curfman McInnes . name - name of SNES method (or use PETSC_NULL) 13919b94acceSBarry Smith 13929b94acceSBarry Smith .keywords: SNES, nonlinear, get, method, name 13939b94acceSBarry Smith @*/ 13944b0e389bSBarry Smith int SNESGetType(SNES snes, SNESType *method,char **name) 13959b94acceSBarry Smith { 139606a9b0adSLois Curfman McInnes int ierr; 139737fcc0dbSBarry Smith if (!__SNESList) {ierr = SNESRegisterAll(); CHKERRQ(ierr);} 13984b0e389bSBarry Smith if (method) *method = (SNESType) snes->type; 139937fcc0dbSBarry Smith if (name) *name = NRFindName( __SNESList, (int) snes->type ); 14009b94acceSBarry Smith return 0; 14019b94acceSBarry Smith } 14029b94acceSBarry Smith 14039b94acceSBarry Smith #include <stdio.h> 14049b94acceSBarry Smith /* 14054b0e389bSBarry Smith SNESPrintTypes_Private - Prints the SNES methods available from the 14069b94acceSBarry Smith options database. 14079b94acceSBarry Smith 14089b94acceSBarry Smith Input Parameters: 140933455573SSatish Balay . comm - communicator (usually MPI_COMM_WORLD) 14109b94acceSBarry Smith . prefix - prefix (usually "-") 14114b0e389bSBarry Smith . name - the options database name (by default "snes_type") 14129b94acceSBarry Smith */ 141333455573SSatish Balay int SNESPrintTypes_Private(MPI_Comm comm,char* prefix,char *name) 14149b94acceSBarry Smith { 14159b94acceSBarry Smith FuncList *entry; 141637fcc0dbSBarry Smith if (!__SNESList) {SNESRegisterAll();} 141737fcc0dbSBarry Smith entry = __SNESList->head; 141877c4ece6SBarry Smith PetscPrintf(comm," %s%s (one of)",prefix,name); 14199b94acceSBarry Smith while (entry) { 142077c4ece6SBarry Smith PetscPrintf(comm," %s",entry->name); 14219b94acceSBarry Smith entry = entry->next; 14229b94acceSBarry Smith } 142377c4ece6SBarry Smith PetscPrintf(comm,"\n"); 14249b94acceSBarry Smith return 0; 14259b94acceSBarry Smith } 14269b94acceSBarry Smith 14279b94acceSBarry Smith /*@C 14289b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 14299b94acceSBarry Smith stored. 14309b94acceSBarry Smith 14319b94acceSBarry Smith Input Parameter: 14329b94acceSBarry Smith . snes - the SNES context 14339b94acceSBarry Smith 14349b94acceSBarry Smith Output Parameter: 14359b94acceSBarry Smith . x - the solution 14369b94acceSBarry Smith 14379b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 14389b94acceSBarry Smith 1439a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESGetGradient(), SNESGetSolutionUpdate() 14409b94acceSBarry Smith @*/ 14419b94acceSBarry Smith int SNESGetSolution(SNES snes,Vec *x) 14429b94acceSBarry Smith { 144377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 14449b94acceSBarry Smith *x = snes->vec_sol_always; 14459b94acceSBarry Smith return 0; 14469b94acceSBarry Smith } 14479b94acceSBarry Smith 14489b94acceSBarry Smith /*@C 14499b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 14509b94acceSBarry Smith stored. 14519b94acceSBarry Smith 14529b94acceSBarry Smith Input Parameter: 14539b94acceSBarry Smith . snes - the SNES context 14549b94acceSBarry Smith 14559b94acceSBarry Smith Output Parameter: 14569b94acceSBarry Smith . x - the solution update 14579b94acceSBarry Smith 14589b94acceSBarry Smith Notes: 14599b94acceSBarry Smith This vector is implementation dependent. 14609b94acceSBarry Smith 14619b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 14629b94acceSBarry Smith 14639b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction 14649b94acceSBarry Smith @*/ 14659b94acceSBarry Smith int SNESGetSolutionUpdate(SNES snes,Vec *x) 14669b94acceSBarry Smith { 146777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 14689b94acceSBarry Smith *x = snes->vec_sol_update_always; 14699b94acceSBarry Smith return 0; 14709b94acceSBarry Smith } 14719b94acceSBarry Smith 14729b94acceSBarry Smith /*@C 14733638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 14749b94acceSBarry Smith 14759b94acceSBarry Smith Input Parameter: 14769b94acceSBarry Smith . snes - the SNES context 14779b94acceSBarry Smith 14789b94acceSBarry Smith Output Parameter: 14793638b69dSLois Curfman McInnes . r - the function 14809b94acceSBarry Smith 14819b94acceSBarry Smith Notes: 14829b94acceSBarry Smith SNESGetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only 14839b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 14849b94acceSBarry Smith SNESGetMinimizationFunction() and SNESGetGradient(); 14859b94acceSBarry Smith 1486a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 14879b94acceSBarry Smith 148831615d04SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution(), SNESGetMinimizationFunction(), 148931615d04SLois Curfman McInnes SNESGetGradient() 14909b94acceSBarry Smith @*/ 14919b94acceSBarry Smith int SNESGetFunction(SNES snes,Vec *r) 14929b94acceSBarry Smith { 149377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 14949b94acceSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1, 149548d91487SBarry Smith "SNESGetFunction:For SNES_NONLINEAR_EQUATIONS only"); 14969b94acceSBarry Smith *r = snes->vec_func_always; 14979b94acceSBarry Smith return 0; 14989b94acceSBarry Smith } 14999b94acceSBarry Smith 15009b94acceSBarry Smith /*@C 15013638b69dSLois Curfman McInnes SNESGetGradient - Returns the vector where the gradient is stored. 15029b94acceSBarry Smith 15039b94acceSBarry Smith Input Parameter: 15049b94acceSBarry Smith . snes - the SNES context 15059b94acceSBarry Smith 15069b94acceSBarry Smith Output Parameter: 15079b94acceSBarry Smith . r - the gradient 15089b94acceSBarry Smith 15099b94acceSBarry Smith Notes: 15109b94acceSBarry Smith SNESGetGradient() is valid for SNES_UNCONSTRAINED_MINIMIZATION methods 15119b94acceSBarry Smith only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 15129b94acceSBarry Smith SNESGetFunction(). 15139b94acceSBarry Smith 15149b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient 15159b94acceSBarry Smith 151631615d04SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESGetSolution(), SNESGetFunction() 15179b94acceSBarry Smith @*/ 15189b94acceSBarry Smith int SNESGetGradient(SNES snes,Vec *r) 15199b94acceSBarry Smith { 152077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15219b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 152248d91487SBarry Smith "SNESGetGradient:For SNES_UNCONSTRAINED_MINIMIZATION only"); 15239b94acceSBarry Smith *r = snes->vec_func_always; 15249b94acceSBarry Smith return 0; 15259b94acceSBarry Smith } 15269b94acceSBarry Smith 15279b94acceSBarry Smith /*@ 15289b94acceSBarry Smith SNESGetMinimizationFunction - Returns the scalar function value for 15299b94acceSBarry Smith unconstrained minimization problems. 15309b94acceSBarry Smith 15319b94acceSBarry Smith Input Parameter: 15329b94acceSBarry Smith . snes - the SNES context 15339b94acceSBarry Smith 15349b94acceSBarry Smith Output Parameter: 15359b94acceSBarry Smith . r - the function 15369b94acceSBarry Smith 15379b94acceSBarry Smith Notes: 15389b94acceSBarry Smith SNESGetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 15399b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 15409b94acceSBarry Smith SNESGetFunction(). 15419b94acceSBarry Smith 15429b94acceSBarry Smith .keywords: SNES, nonlinear, get, function 15439b94acceSBarry Smith 154431615d04SLois Curfman McInnes .seealso: SNESGetGradient(), SNESGetSolution(), SNESGetFunction() 15459b94acceSBarry Smith @*/ 15469b94acceSBarry Smith int SNESGetMinimizationFunction(SNES snes,double *r) 15479b94acceSBarry Smith { 154877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15499b94acceSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1, 155048d91487SBarry Smith "SNESGetMinimizationFunction:For SNES_UNCONSTRAINED_MINIMIZATION only"); 15519b94acceSBarry Smith *r = snes->fc; 15529b94acceSBarry Smith return 0; 15539b94acceSBarry Smith } 15549b94acceSBarry Smith 15553c7409f5SSatish Balay /*@C 15563c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 15573c7409f5SSatish Balay SNES options in the database. 15583c7409f5SSatish Balay 15593c7409f5SSatish Balay Input Parameter: 15603c7409f5SSatish Balay . snes - the SNES context 15613c7409f5SSatish Balay . prefix - the prefix to prepend to all option names 15623c7409f5SSatish Balay 15633c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 1564a86d99e1SLois Curfman McInnes 1565a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 15663c7409f5SSatish Balay @*/ 15673c7409f5SSatish Balay int SNESSetOptionsPrefix(SNES snes,char *prefix) 15683c7409f5SSatish Balay { 15693c7409f5SSatish Balay int ierr; 15703c7409f5SSatish Balay 157177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15723c7409f5SSatish Balay ierr = PetscObjectSetPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 15733c7409f5SSatish Balay ierr = SLESSetOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 15743c7409f5SSatish Balay return 0; 15753c7409f5SSatish Balay } 15763c7409f5SSatish Balay 15773c7409f5SSatish Balay /*@C 15783c7409f5SSatish Balay SNESAppendOptionsPrefix - Append to the prefix used for searching for all 15793c7409f5SSatish Balay SNES options in the database. 15803c7409f5SSatish Balay 15813c7409f5SSatish Balay Input Parameter: 15823c7409f5SSatish Balay . snes - the SNES context 15833c7409f5SSatish Balay . prefix - the prefix to prepend to all option names 15843c7409f5SSatish Balay 15853c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 1586a86d99e1SLois Curfman McInnes 1587a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 15883c7409f5SSatish Balay @*/ 15893c7409f5SSatish Balay int SNESAppendOptionsPrefix(SNES snes,char *prefix) 15903c7409f5SSatish Balay { 15913c7409f5SSatish Balay int ierr; 15923c7409f5SSatish Balay 159377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15943c7409f5SSatish Balay ierr = PetscObjectAppendPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 15953c7409f5SSatish Balay ierr = SLESAppendOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 15963c7409f5SSatish Balay return 0; 15973c7409f5SSatish Balay } 15983c7409f5SSatish Balay 1599c83590e2SSatish Balay /*@ 16003c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 16013c7409f5SSatish Balay SNES options in the database. 16023c7409f5SSatish Balay 16033c7409f5SSatish Balay Input Parameter: 16043c7409f5SSatish Balay . snes - the SNES context 16053c7409f5SSatish Balay 16063c7409f5SSatish Balay Output Parameter: 16073c7409f5SSatish Balay . prefix - pointer to the prefix string used 16083c7409f5SSatish Balay 16093c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 1610a86d99e1SLois Curfman McInnes 1611a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 16123c7409f5SSatish Balay @*/ 16133c7409f5SSatish Balay int SNESGetOptionsPrefix(SNES snes,char **prefix) 16143c7409f5SSatish Balay { 16153c7409f5SSatish Balay int ierr; 16163c7409f5SSatish Balay 161777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 16183c7409f5SSatish Balay ierr = PetscObjectGetPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 16193c7409f5SSatish Balay return 0; 16203c7409f5SSatish Balay } 16213c7409f5SSatish Balay 16223c7409f5SSatish Balay 16239b94acceSBarry Smith 16249b94acceSBarry Smith 16259b94acceSBarry Smith 1626