xref: /petsc/src/snes/interface/snes.c (revision 7d1a2b2bfed25720a23dbf396311effb795425c3)
19b94acceSBarry Smith #ifndef lint
2*7d1a2b2bSBarry Smith static char vcid[] = "$Id: snes.c,v 1.70 1996/03/26 00:10:05 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*);
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) {
11740191667SLois Curfman McInnes       ierr = SNESSetType(snes,SNES_EQ_LS); CHKERRQ(ierr);
1185334005bSBarry Smith     }
1195334005bSBarry Smith     else {
12040191667SLois Curfman McInnes       ierr = SNESSetType(snes,SNES_UM_TR); 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);
127d7a720efSLois 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);
129d7a720efSLois 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);
131d7a720efSLois 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);
134d7a720efSLois Curfman McInnes   ierr = OptionsGetDouble(snes->prefix,"-snes_trtol",&tmp, &flg); CHKERRQ(ierr);
135d7a720efSLois Curfman McInnes   if (flg) { SNESSetTrustRegionTolerance(snes,tmp); }
136d7a720efSLois Curfman McInnes   ierr = OptionsGetDouble(snes->prefix,"-snes_fmin",&tmp, &flg); CHKERRQ(ierr);
137d7a720efSLois Curfman McInnes   if (flg) { SNESSetMinimizationFunctionTolerance(snes,tmp); }
1383c7409f5SSatish Balay   ierr = OptionsHasName(snes->prefix,"-snes_ksp_ew_conv", &flg); CHKERRQ(ierr);
1393c7409f5SSatish Balay   if (flg) { snes->ksp_ewconv = 1; }
140b18e04deSLois Curfman McInnes   ierr = OptionsGetInt(snes->prefix,"-snes_ksp_ew_version",&version,&flg); CHKERRQ(ierr);
141b18e04deSLois Curfman McInnes   ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtol0",&rtol_0,&flg); CHKERRQ(ierr);
142b18e04deSLois Curfman McInnes   ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtolmax",&rtol_max,&flg); CHKERRQ(ierr);
143b18e04deSLois Curfman McInnes   ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_gamma",&gamma2,&flg); CHKERRQ(ierr);
144b18e04deSLois Curfman McInnes   ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha",&alpha,&flg); CHKERRQ(ierr);
145b18e04deSLois Curfman McInnes   ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha2",&alpha2,&flg); CHKERRQ(ierr);
146b18e04deSLois Curfman McInnes   ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_threshold",&threshold,&flg);  CHKERRQ(ierr);
1479b94acceSBarry Smith   ierr = SNES_KSP_SetParametersEW(snes,version,rtol_0,rtol_max,gamma2,alpha,
1489b94acceSBarry Smith                                   alpha2,threshold); CHKERRQ(ierr);
1493c7409f5SSatish Balay   ierr = OptionsHasName(snes->prefix,"-snes_monitor", &flg);  CHKERRQ(ierr);
1503c7409f5SSatish Balay   if (flg) { SNESSetMonitor(snes,SNESDefaultMonitor,0); }
1513c7409f5SSatish Balay   ierr = OptionsHasName(snes->prefix,"-snes_smonitor", &flg);  CHKERRQ(ierr);
1523c7409f5SSatish Balay    if (flg) { SNESSetMonitor(snes,SNESDefaultSMonitor,0); }
1533c7409f5SSatish Balay   ierr = OptionsHasName(snes->prefix,"-snes_xmonitor", &flg);  CHKERRQ(ierr);
1543c7409f5SSatish Balay   if (flg) {
15517699dbbSLois Curfman McInnes     int       rank = 0;
156d7e8b826SBarry Smith     DrawLG lg;
15717699dbbSLois Curfman McInnes     MPI_Initialized(&rank);
15817699dbbSLois Curfman McInnes     if (rank) MPI_Comm_rank(snes->comm,&rank);
15917699dbbSLois Curfman McInnes     if (!rank) {
1609b94acceSBarry Smith       ierr = SNESLGMonitorCreate(0,0,0,0,300,300,&lg); CHKERRQ(ierr);
1619b94acceSBarry Smith       ierr = SNESSetMonitor(snes,SNESLGMonitor,(void *)lg); CHKERRQ(ierr);
162f8c826e1SBarry Smith       snes->xmonitor = lg;
1639b94acceSBarry Smith       PLogObjectParent(snes,lg);
1649b94acceSBarry Smith     }
1659b94acceSBarry Smith   }
1663c7409f5SSatish Balay   ierr = OptionsHasName(snes->prefix,"-snes_fd", &flg);  CHKERRQ(ierr);
1673c7409f5SSatish Balay   if (flg && snes->method_class == SNES_NONLINEAR_EQUATIONS) {
1689b94acceSBarry Smith     ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,
1699b94acceSBarry Smith                  SNESDefaultComputeJacobian,snes->funP); CHKERRQ(ierr);
17094a424c1SBarry Smith     PLogInfo(snes,
17131615d04SLois Curfman McInnes       "SNESSetFromOptions: Setting default finite difference Jacobian matrix\n");
17231615d04SLois Curfman McInnes   }
17331615d04SLois Curfman McInnes   else if (flg && snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) {
17431615d04SLois Curfman McInnes     ierr = SNESSetHessian(snes,snes->jacobian,snes->jacobian_pre,
17531615d04SLois Curfman McInnes                  SNESDefaultComputeHessian,snes->funP); CHKERRQ(ierr);
17694a424c1SBarry Smith     PLogInfo(snes,
17731615d04SLois Curfman McInnes       "SNESSetFromOptions: Setting default finite difference Hessian matrix\n");
1789b94acceSBarry Smith   }
1799b94acceSBarry Smith   ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr);
1809b94acceSBarry Smith   ierr = SLESSetFromOptions(sles); CHKERRQ(ierr);
1819b94acceSBarry Smith   if (!snes->setfromoptions) return 0;
1829b94acceSBarry Smith   return (*snes->setfromoptions)(snes);
1839b94acceSBarry Smith }
1849b94acceSBarry Smith 
1859b94acceSBarry Smith /*@
1869b94acceSBarry Smith    SNESPrintHelp - Prints all options for the SNES component.
1879b94acceSBarry Smith 
1889b94acceSBarry Smith    Input Parameter:
1899b94acceSBarry Smith .  snes - the SNES context
1909b94acceSBarry Smith 
191a703fe33SLois Curfman McInnes    Options Database Keys:
192a703fe33SLois Curfman McInnes $  -help, -h
193a703fe33SLois Curfman McInnes 
1949b94acceSBarry Smith .keywords: SNES, nonlinear, help
1959b94acceSBarry Smith 
196682d7d0cSBarry Smith .seealso: SNESSetFromOptions()
1979b94acceSBarry Smith @*/
1989b94acceSBarry Smith int SNESPrintHelp(SNES snes)
1999b94acceSBarry Smith {
2006daaf66cSBarry Smith   char                p[64];
2016daaf66cSBarry Smith   SNES_KSP_EW_ConvCtx *kctx;
2026daaf66cSBarry Smith 
20377c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
2046daaf66cSBarry Smith 
2056daaf66cSBarry Smith   PetscStrcpy(p,"-");
2066daaf66cSBarry Smith   if (snes->prefix) PetscStrcat(p, snes->prefix);
2076daaf66cSBarry Smith 
2086daaf66cSBarry Smith   kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx;
2096daaf66cSBarry Smith 
21077c4ece6SBarry Smith   PetscPrintf(snes->comm,"SNES options ----------------------------\n");
21133455573SSatish Balay   SNESPrintTypes_Private(snes->comm,p,"snes_type");
21277c4ece6SBarry Smith   PetscPrintf(snes->comm," %ssnes_monitor: use default SNES monitor\n",p);
21377c4ece6SBarry Smith   PetscPrintf(snes->comm," %ssnes_view: view SNES info after each nonlinear solve\n",p);
214b18e04deSLois Curfman McInnes   PetscPrintf(snes->comm," %ssnes_max_it <its>: max iterations (default %d)\n",p,snes->max_its);
215b18e04deSLois Curfman McInnes   PetscPrintf(snes->comm," %ssnes_max_funcs <maxf>: max function evals (default %d)\n",p,snes->max_funcs);
216b18e04deSLois Curfman McInnes   PetscPrintf(snes->comm," %ssnes_stol <stol>: successive step tolerance (default %g)\n",p,snes->xtol);
217b18e04deSLois Curfman McInnes   PetscPrintf(snes->comm," %ssnes_atol <atol>: absolute tolerance (default %g)\n",p,snes->atol);
218b18e04deSLois Curfman McInnes   PetscPrintf(snes->comm," %ssnes_rtol <rtol>: relative tolerance (default %g)\n",p,snes->rtol);
219b18e04deSLois Curfman McInnes   PetscPrintf(snes->comm," %ssnes_trtol <trtol>: trust region parameter tolerance (default %g)\n",p,snes->deltatol);
220b18e04deSLois Curfman McInnes   if (snes->type == SNES_NONLINEAR_EQUATIONS) {
22177c4ece6SBarry Smith     PetscPrintf(snes->comm,
2229b94acceSBarry Smith      " options for solving systems of nonlinear equations only:\n");
22377c4ece6SBarry Smith     PetscPrintf(snes->comm,"   %ssnes_fd: use finite differences for Jacobian\n",p);
22477c4ece6SBarry Smith     PetscPrintf(snes->comm,"   %ssnes_mf: use matrix-free Jacobian\n",p);
22577c4ece6SBarry Smith     PetscPrintf(snes->comm,"   %ssnes_ksp_ew_conv: use Eisenstat-Walker computation of KSP rtol. Params are:\n",p);
22677c4ece6SBarry Smith     PetscPrintf(snes->comm,
227b18e04deSLois Curfman McInnes      "     %ssnes_ksp_ew_version <version> (1 or 2, default is %d)\n",p,kctx->version);
22877c4ece6SBarry Smith     PetscPrintf(snes->comm,
229b18e04deSLois Curfman McInnes      "     %ssnes_ksp_ew_rtol0 <rtol0> (0 <= rtol0 < 1, default %g)\n",p,kctx->rtol_0);
23077c4ece6SBarry Smith     PetscPrintf(snes->comm,
231b18e04deSLois Curfman McInnes      "     %ssnes_ksp_ew_rtolmax <rtolmax> (0 <= rtolmax < 1, default %g)\n",p,kctx->rtol_max);
23277c4ece6SBarry Smith     PetscPrintf(snes->comm,
233b18e04deSLois Curfman McInnes      "     %ssnes_ksp_ew_gamma <gamma> (0 <= gamma <= 1, default %g)\n",p,kctx->gamma);
23477c4ece6SBarry Smith     PetscPrintf(snes->comm,
235b18e04deSLois Curfman McInnes      "     %ssnes_ksp_ew_alpha <alpha> (1 < alpha <= 2, default %g)\n",p,kctx->alpha);
23677c4ece6SBarry Smith     PetscPrintf(snes->comm,
237b18e04deSLois Curfman McInnes      "     %ssnes_ksp_ew_alpha2 <alpha2> (default %g)\n",p,kctx->alpha2);
23877c4ece6SBarry Smith     PetscPrintf(snes->comm,
239b18e04deSLois Curfman McInnes      "     %ssnes_ksp_ew_threshold <threshold> (0 < threshold < 1, default %g)\n",p,kctx->threshold);
240b18e04deSLois Curfman McInnes   }
241b18e04deSLois Curfman McInnes   else if (snes->type == SNES_UNCONSTRAINED_MINIMIZATION) {
24277c4ece6SBarry Smith     PetscPrintf(snes->comm,
2439b94acceSBarry Smith      " options for solving unconstrained minimization problems only:\n");
244b18e04deSLois Curfman McInnes     PetscPrintf(snes->comm,"   %ssnes_fmin <ftol>: minimum function tolerance (default %g)\n",p,snes->fmin);
24577c4ece6SBarry Smith     PetscPrintf(snes->comm,"   %ssnes_fd: use finite differences for Hessian\n",p);
24677c4ece6SBarry Smith     PetscPrintf(snes->comm,"   %ssnes_mf: use matrix-free Hessian\n",p);
247b18e04deSLois Curfman McInnes   }
248b18e04deSLois Curfman McInnes   PetscPrintf(snes->comm," Run program with %ssnes_type method -help for help on ",p);
24977c4ece6SBarry Smith   PetscPrintf(snes->comm,"a particular method\n");
2506daaf66cSBarry Smith   if (snes->printhelp) (*snes->printhelp)(snes,p);
2519b94acceSBarry Smith   return 0;
2529b94acceSBarry Smith }
2539b94acceSBarry Smith /*@
2549b94acceSBarry Smith    SNESSetApplicationContext - Sets the optional user-defined context for
2559b94acceSBarry Smith    the nonlinear solvers.
2569b94acceSBarry Smith 
2579b94acceSBarry Smith    Input Parameters:
2589b94acceSBarry Smith .  snes - the SNES context
2599b94acceSBarry Smith .  usrP - optional user context
2609b94acceSBarry Smith 
2619b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context
2629b94acceSBarry Smith 
2639b94acceSBarry Smith .seealso: SNESGetApplicationContext()
2649b94acceSBarry Smith @*/
2659b94acceSBarry Smith int SNESSetApplicationContext(SNES snes,void *usrP)
2669b94acceSBarry Smith {
26777c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
2689b94acceSBarry Smith   snes->user		= usrP;
2699b94acceSBarry Smith   return 0;
2709b94acceSBarry Smith }
2719b94acceSBarry Smith /*@C
2729b94acceSBarry Smith    SNESGetApplicationContext - Gets the user-defined context for the
2739b94acceSBarry Smith    nonlinear solvers.
2749b94acceSBarry Smith 
2759b94acceSBarry Smith    Input Parameter:
2769b94acceSBarry Smith .  snes - SNES context
2779b94acceSBarry Smith 
2789b94acceSBarry Smith    Output Parameter:
2799b94acceSBarry Smith .  usrP - user context
2809b94acceSBarry Smith 
2819b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context
2829b94acceSBarry Smith 
2839b94acceSBarry Smith .seealso: SNESSetApplicationContext()
2849b94acceSBarry Smith @*/
2859b94acceSBarry Smith int SNESGetApplicationContext( SNES snes,  void **usrP )
2869b94acceSBarry Smith {
28777c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
2889b94acceSBarry Smith   *usrP = snes->user;
2899b94acceSBarry Smith   return 0;
2909b94acceSBarry Smith }
2919b94acceSBarry Smith /*@
2929b94acceSBarry Smith    SNESGetIterationNumber - Gets the current iteration number of the
2939b94acceSBarry Smith    nonlinear solver.
2949b94acceSBarry Smith 
2959b94acceSBarry Smith    Input Parameter:
2969b94acceSBarry Smith .  snes - SNES context
2979b94acceSBarry Smith 
2989b94acceSBarry Smith    Output Parameter:
2999b94acceSBarry Smith .  iter - iteration number
3009b94acceSBarry Smith 
3019b94acceSBarry Smith .keywords: SNES, nonlinear, get, iteration, number
3029b94acceSBarry Smith @*/
3039b94acceSBarry Smith int SNESGetIterationNumber(SNES snes,int* iter)
3049b94acceSBarry Smith {
30577c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
3069b94acceSBarry Smith   *iter = snes->iter;
3079b94acceSBarry Smith   return 0;
3089b94acceSBarry Smith }
3099b94acceSBarry Smith /*@
3109b94acceSBarry Smith    SNESGetFunctionNorm - Gets the norm of the current function that was set
3119b94acceSBarry Smith    with SNESSSetFunction().
3129b94acceSBarry Smith 
3139b94acceSBarry Smith    Input Parameter:
3149b94acceSBarry Smith .  snes - SNES context
3159b94acceSBarry Smith 
3169b94acceSBarry Smith    Output Parameter:
3179b94acceSBarry Smith .  fnorm - 2-norm of function
3189b94acceSBarry Smith 
3199b94acceSBarry Smith    Note:
3209b94acceSBarry Smith    SNESGetFunctionNorm() is valid for SNES_NONLINEAR_EQUATIONS methods only.
321a86d99e1SLois Curfman McInnes    A related routine for SNES_UNCONSTRAINED_MINIMIZATION methods is
322a86d99e1SLois Curfman McInnes    SNESGetGradientNorm().
3239b94acceSBarry Smith 
3249b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm
325a86d99e1SLois Curfman McInnes 
326a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction()
3279b94acceSBarry Smith @*/
3289b94acceSBarry Smith int SNESGetFunctionNorm(SNES snes,Scalar *fnorm)
3299b94acceSBarry Smith {
33077c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
3319b94acceSBarry Smith   if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,
33248d91487SBarry Smith     "SNESGetFunctionNorm:For SNES_NONLINEAR_EQUATIONS only");
3339b94acceSBarry Smith   *fnorm = snes->norm;
3349b94acceSBarry Smith   return 0;
3359b94acceSBarry Smith }
3369b94acceSBarry Smith /*@
3379b94acceSBarry Smith    SNESGetGradientNorm - Gets the norm of the current gradient that was set
3389b94acceSBarry Smith    with SNESSSetGradient().
3399b94acceSBarry Smith 
3409b94acceSBarry Smith    Input Parameter:
3419b94acceSBarry Smith .  snes - SNES context
3429b94acceSBarry Smith 
3439b94acceSBarry Smith    Output Parameter:
3449b94acceSBarry Smith .  fnorm - 2-norm of gradient
3459b94acceSBarry Smith 
3469b94acceSBarry Smith    Note:
3479b94acceSBarry Smith    SNESGetGradientNorm() is valid for SNES_UNCONSTRAINED_MINIMIZATION
348a86d99e1SLois Curfman McInnes    methods only.  A related routine for SNES_NONLINEAR_EQUATIONS methods
349a86d99e1SLois Curfman McInnes    is SNESGetFunctionNorm().
3509b94acceSBarry Smith 
3519b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient, norm
352a86d99e1SLois Curfman McInnes 
353a86d99e1SLois Curfman McInnes .seelso: SNESSetGradient()
3549b94acceSBarry Smith @*/
3559b94acceSBarry Smith int SNESGetGradientNorm(SNES snes,Scalar *gnorm)
3569b94acceSBarry Smith {
35777c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
3589b94acceSBarry Smith   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
35948d91487SBarry Smith     "SNESGetGradientNorm:For SNES_UNCONSTRAINED_MINIMIZATION only");
3609b94acceSBarry Smith   *gnorm = snes->norm;
3619b94acceSBarry Smith   return 0;
3629b94acceSBarry Smith }
3639b94acceSBarry Smith /*@
3649b94acceSBarry Smith    SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps
3659b94acceSBarry Smith    attempted by the nonlinear solver.
3669b94acceSBarry Smith 
3679b94acceSBarry Smith    Input Parameter:
3689b94acceSBarry Smith .  snes - SNES context
3699b94acceSBarry Smith 
3709b94acceSBarry Smith    Output Parameter:
3719b94acceSBarry Smith .  nfails - number of unsuccessful steps attempted
3729b94acceSBarry Smith 
3739b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps
3749b94acceSBarry Smith @*/
3759b94acceSBarry Smith int SNESGetNumberUnsuccessfulSteps(SNES snes,int* nfails)
3769b94acceSBarry Smith {
37777c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
3789b94acceSBarry Smith   *nfails = snes->nfailures;
3799b94acceSBarry Smith   return 0;
3809b94acceSBarry Smith }
3819b94acceSBarry Smith /*@C
3829b94acceSBarry Smith    SNESGetSLES - Returns the SLES context for a SNES solver.
3839b94acceSBarry Smith 
3849b94acceSBarry Smith    Input Parameter:
3859b94acceSBarry Smith .  snes - the SNES context
3869b94acceSBarry Smith 
3879b94acceSBarry Smith    Output Parameter:
3889b94acceSBarry Smith .  sles - the SLES context
3899b94acceSBarry Smith 
3909b94acceSBarry Smith    Notes:
3919b94acceSBarry Smith    The user can then directly manipulate the SLES context to set various
3929b94acceSBarry Smith    options, etc.  Likewise, the user can then extract and manipulate the
3939b94acceSBarry Smith    KSP and PC contexts as well.
3949b94acceSBarry Smith 
3959b94acceSBarry Smith .keywords: SNES, nonlinear, get, SLES, context
3969b94acceSBarry Smith 
3979b94acceSBarry Smith .seealso: SLESGetPC(), SLESGetKSP()
3989b94acceSBarry Smith @*/
3999b94acceSBarry Smith int SNESGetSLES(SNES snes,SLES *sles)
4009b94acceSBarry Smith {
40177c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
4029b94acceSBarry Smith   *sles = snes->sles;
4039b94acceSBarry Smith   return 0;
4049b94acceSBarry Smith }
4059b94acceSBarry Smith /* -----------------------------------------------------------*/
4069b94acceSBarry Smith /*@C
4079b94acceSBarry Smith    SNESCreate - Creates a nonlinear solver context.
4089b94acceSBarry Smith 
4099b94acceSBarry Smith    Input Parameter:
4109b94acceSBarry Smith .  comm - MPI communicator
4119b94acceSBarry Smith .  type - type of method, one of
4129b94acceSBarry Smith $    SNES_NONLINEAR_EQUATIONS
4139b94acceSBarry Smith $      (for systems of nonlinear equations)
4149b94acceSBarry Smith $    SNES_UNCONSTRAINED_MINIMIZATION
4159b94acceSBarry Smith $      (for unconstrained minimization)
4169b94acceSBarry Smith 
4179b94acceSBarry Smith    Output Parameter:
4189b94acceSBarry Smith .  outsnes - the new SNES context
4199b94acceSBarry Smith 
4209b94acceSBarry Smith .keywords: SNES, nonlinear, create, context
4219b94acceSBarry Smith 
42263a78c88SLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy()
4239b94acceSBarry Smith @*/
4244b0e389bSBarry Smith int SNESCreate(MPI_Comm comm,SNESProblemType type,SNES *outsnes)
4259b94acceSBarry Smith {
4269b94acceSBarry Smith   int                 ierr;
4279b94acceSBarry Smith   SNES                snes;
4289b94acceSBarry Smith   SNES_KSP_EW_ConvCtx *kctx;
42937fcc0dbSBarry Smith 
4309b94acceSBarry Smith   *outsnes = 0;
4312a0acf01SLois Curfman McInnes   if (type != SNES_UNCONSTRAINED_MINIMIZATION && type != SNES_NONLINEAR_EQUATIONS)
4322a0acf01SLois Curfman McInnes     SETERRQ(1,"SNESCreate:incorrect method type");
4330452661fSBarry Smith   PetscHeaderCreate(snes,_SNES,SNES_COOKIE,SNES_UNKNOWN_METHOD,comm);
4349b94acceSBarry Smith   PLogObjectCreate(snes);
4359b94acceSBarry Smith   snes->max_its           = 50;
4369b94acceSBarry Smith   snes->max_funcs	  = 1000;
4379b94acceSBarry Smith   snes->norm		  = 0.0;
4385a2d0531SBarry Smith   if (type == SNES_UNCONSTRAINED_MINIMIZATION) {
439b18e04deSLois Curfman McInnes     snes->rtol		  = 1.e-8;
440b18e04deSLois Curfman McInnes     snes->ttol            = 0.0;
4419b94acceSBarry Smith     snes->atol		  = 1.e-10;
4425a2d0531SBarry Smith   }
443b4874afaSBarry Smith   else {
444b4874afaSBarry Smith     snes->rtol		  = 1.e-8;
445b4874afaSBarry Smith     snes->ttol            = 0.0;
446b4874afaSBarry Smith     snes->atol		  = 1.e-50;
447b4874afaSBarry Smith   }
4489b94acceSBarry Smith   snes->xtol		  = 1.e-8;
449b18e04deSLois Curfman McInnes   snes->trunctol	  = 1.e-12; /* no longer used */
4509b94acceSBarry Smith   snes->nfuncs            = 0;
4519b94acceSBarry Smith   snes->nfailures         = 0;
4529b94acceSBarry Smith   snes->monitor           = 0;
4539b94acceSBarry Smith   snes->data              = 0;
4549b94acceSBarry Smith   snes->view              = 0;
4559b94acceSBarry Smith   snes->computeumfunction = 0;
4569b94acceSBarry Smith   snes->umfunP            = 0;
4579b94acceSBarry Smith   snes->fc                = 0;
4589b94acceSBarry Smith   snes->deltatol          = 1.e-12;
4599b94acceSBarry Smith   snes->fmin              = -1.e30;
4609b94acceSBarry Smith   snes->method_class      = type;
4619b94acceSBarry Smith   snes->set_method_called = 0;
462a703fe33SLois Curfman McInnes   snes->setup_called      = 0;
4639b94acceSBarry Smith   snes->ksp_ewconv        = 0;
464*7d1a2b2bSBarry Smith   snes->type              = -1;
4659b94acceSBarry Smith 
4669b94acceSBarry Smith   /* Create context to compute Eisenstat-Walker relative tolerance for KSP */
4670452661fSBarry Smith   kctx = PetscNew(SNES_KSP_EW_ConvCtx); CHKPTRQ(kctx);
4689b94acceSBarry Smith   snes->kspconvctx  = (void*)kctx;
4699b94acceSBarry Smith   kctx->version     = 2;
4709b94acceSBarry Smith   kctx->rtol_0      = .3; /* Eisenstat and Walker suggest rtol_0=.5, but
4719b94acceSBarry Smith                              this was too large for some test cases */
4729b94acceSBarry Smith   kctx->rtol_last   = 0;
4739b94acceSBarry Smith   kctx->rtol_max    = .9;
4749b94acceSBarry Smith   kctx->gamma       = 1.0;
4759b94acceSBarry Smith   kctx->alpha2      = .5*(1.0 + sqrt(5.0));
4769b94acceSBarry Smith   kctx->alpha       = kctx->alpha2;
4779b94acceSBarry Smith   kctx->threshold   = .1;
4789b94acceSBarry Smith   kctx->lresid_last = 0;
4799b94acceSBarry Smith   kctx->norm_last   = 0;
4809b94acceSBarry Smith 
4819b94acceSBarry Smith   ierr = SLESCreate(comm,&snes->sles); CHKERRQ(ierr);
4829b94acceSBarry Smith   PLogObjectParent(snes,snes->sles)
4835334005bSBarry Smith 
4849b94acceSBarry Smith   *outsnes = snes;
4859b94acceSBarry Smith   return 0;
4869b94acceSBarry Smith }
4879b94acceSBarry Smith 
4889b94acceSBarry Smith /* --------------------------------------------------------------- */
4899b94acceSBarry Smith /*@C
4909b94acceSBarry Smith    SNESSetFunction - Sets the function evaluation routine and function
4919b94acceSBarry Smith    vector for use by the SNES routines in solving systems of nonlinear
4929b94acceSBarry Smith    equations.
4939b94acceSBarry Smith 
4949b94acceSBarry Smith    Input Parameters:
4959b94acceSBarry Smith .  snes - the SNES context
4969b94acceSBarry Smith .  func - function evaluation routine
4979b94acceSBarry Smith .  ctx - optional user-defined function context
4989b94acceSBarry Smith .  r - vector to store function value
4999b94acceSBarry Smith 
5009b94acceSBarry Smith    Calling sequence of func:
5019b94acceSBarry Smith .  func (SNES, Vec x, Vec f, void *ctx);
5029b94acceSBarry Smith 
5039b94acceSBarry Smith .  x - input vector
5045334005bSBarry Smith .  f - vector function
5059b94acceSBarry Smith .  ctx - optional user-defined context for private data for the
5069b94acceSBarry Smith          function evaluation routine (may be null)
5079b94acceSBarry Smith 
5089b94acceSBarry Smith    Notes:
5099b94acceSBarry Smith    The Newton-like methods typically solve linear systems of the form
5109b94acceSBarry Smith $      f'(x) x = -f(x),
5119b94acceSBarry Smith $  where f'(x) denotes the Jacobian matrix and f(x) is the function.
5129b94acceSBarry Smith 
5139b94acceSBarry Smith    SNESSetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only.
5149b94acceSBarry Smith    Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are
5159b94acceSBarry Smith    SNESSetMinimizationFunction() and SNESSetGradient();
5169b94acceSBarry Smith 
5179b94acceSBarry Smith .keywords: SNES, nonlinear, set, function
5189b94acceSBarry Smith 
519a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian()
5209b94acceSBarry Smith @*/
5215334005bSBarry Smith int SNESSetFunction( SNES snes, Vec r, int (*func)(SNES,Vec,Vec,void*),void *ctx)
5229b94acceSBarry Smith {
52377c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
5249b94acceSBarry Smith   if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,
52548d91487SBarry Smith     "SNESSetFunction:For SNES_NONLINEAR_EQUATIONS only");
5269b94acceSBarry Smith   snes->computefunction     = func;
5279b94acceSBarry Smith   snes->vec_func            = snes->vec_func_always = r;
5289b94acceSBarry Smith   snes->funP                = ctx;
5299b94acceSBarry Smith   return 0;
5309b94acceSBarry Smith }
5319b94acceSBarry Smith 
5329b94acceSBarry Smith /*@
5339b94acceSBarry Smith    SNESComputeFunction - Computes the function that has been set with
5349b94acceSBarry Smith    SNESSetFunction().
5359b94acceSBarry Smith 
5369b94acceSBarry Smith    Input Parameters:
5379b94acceSBarry Smith .  snes - the SNES context
5389b94acceSBarry Smith .  x - input vector
5399b94acceSBarry Smith 
5409b94acceSBarry Smith    Output Parameter:
5413638b69dSLois Curfman McInnes .  y - function vector, as set by SNESSetFunction()
5429b94acceSBarry Smith 
543a86d99e1SLois Curfman McInnes n   Notes:
5449b94acceSBarry Smith    SNESComputeFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only.
5459b94acceSBarry Smith    Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are
5469b94acceSBarry Smith    SNESComputeMinimizationFunction() and SNESComputeGradient();
5479b94acceSBarry Smith 
5489b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function
5499b94acceSBarry Smith 
550a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction()
5519b94acceSBarry Smith @*/
5529b94acceSBarry Smith int SNESComputeFunction(SNES snes,Vec x, Vec y)
5539b94acceSBarry Smith {
5549b94acceSBarry Smith   int    ierr;
5559b94acceSBarry Smith 
556a86d99e1SLois Curfman McInnes   if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,
557a86d99e1SLois Curfman McInnes     "SNESComputeFunction: For SNES_NONLINEAR_EQUATIONS only");
5589b94acceSBarry Smith   PLogEventBegin(SNES_FunctionEval,snes,x,y,0);
5599b94acceSBarry Smith   ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr);
5609b94acceSBarry Smith   PLogEventEnd(SNES_FunctionEval,snes,x,y,0);
5619b94acceSBarry Smith   return 0;
5629b94acceSBarry Smith }
5639b94acceSBarry Smith 
5649b94acceSBarry Smith /*@C
5659b94acceSBarry Smith    SNESSetMinimizationFunction - Sets the function evaluation routine for
5669b94acceSBarry Smith    unconstrained minimization.
5679b94acceSBarry Smith 
5689b94acceSBarry Smith    Input Parameters:
5699b94acceSBarry Smith .  snes - the SNES context
5709b94acceSBarry Smith .  func - function evaluation routine
5719b94acceSBarry Smith .  ctx - optional user-defined function context
5729b94acceSBarry Smith 
5739b94acceSBarry Smith    Calling sequence of func:
5749b94acceSBarry Smith .  func (SNES snes,Vec x,double *f,void *ctx);
5759b94acceSBarry Smith 
5769b94acceSBarry Smith .  x - input vector
5779b94acceSBarry Smith .  f - function
5789b94acceSBarry Smith .  ctx - optional user-defined context for private data for the
5799b94acceSBarry Smith          function evaluation routine (may be null)
5809b94acceSBarry Smith 
5819b94acceSBarry Smith    Notes:
5829b94acceSBarry Smith    SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION
5839b94acceSBarry Smith    methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is
5849b94acceSBarry Smith    SNESSetFunction().
5859b94acceSBarry Smith 
5869b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimization, function
5879b94acceSBarry Smith 
588a86d99e1SLois Curfman McInnes .seealso:  SNESGetMinimizationFunction(), SNESComputeMinimizationFunction(),
589a86d99e1SLois Curfman McInnes            SNESSetHessian(), SNESSetGradient()
5909b94acceSBarry Smith @*/
5919b94acceSBarry Smith int SNESSetMinimizationFunction(SNES snes,int (*func)(SNES,Vec,double*,void*),
5929b94acceSBarry Smith                       void *ctx)
5939b94acceSBarry Smith {
59477c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
5959b94acceSBarry Smith   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
59648d91487SBarry Smith     "SNESSetMinimizationFunction:Only for SNES_UNCONSTRAINED_MINIMIZATION");
5979b94acceSBarry Smith   snes->computeumfunction   = func;
5989b94acceSBarry Smith   snes->umfunP              = ctx;
5999b94acceSBarry Smith   return 0;
6009b94acceSBarry Smith }
6019b94acceSBarry Smith 
6029b94acceSBarry Smith /*@
6039b94acceSBarry Smith    SNESComputeMinimizationFunction - Computes the function that has been
6049b94acceSBarry Smith    set with SNESSetMinimizationFunction().
6059b94acceSBarry Smith 
6069b94acceSBarry Smith    Input Parameters:
6079b94acceSBarry Smith .  snes - the SNES context
6089b94acceSBarry Smith .  x - input vector
6099b94acceSBarry Smith 
6109b94acceSBarry Smith    Output Parameter:
6119b94acceSBarry Smith .  y - function value
6129b94acceSBarry Smith 
6139b94acceSBarry Smith    Notes:
6149b94acceSBarry Smith    SNESComputeMinimizationFunction() is valid only for
6159b94acceSBarry Smith    SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for
6169b94acceSBarry Smith    SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction().
617a86d99e1SLois Curfman McInnes 
618a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, minimization, function
619a86d99e1SLois Curfman McInnes 
620a86d99e1SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESGetMinimizationFunction(),
621a86d99e1SLois Curfman McInnes           SNESComputeGradient(), SNESComputeHessian()
6229b94acceSBarry Smith @*/
6239b94acceSBarry Smith int SNESComputeMinimizationFunction(SNES snes,Vec x,double *y)
6249b94acceSBarry Smith {
6259b94acceSBarry Smith   int    ierr;
6269b94acceSBarry Smith   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
62748d91487SBarry Smith     "SNESComputeMinimizationFunction:Only for SNES_UNCONSTRAINED_MINIMIZATION");
6289b94acceSBarry Smith   PLogEventBegin(SNES_MinimizationFunctionEval,snes,x,y,0);
6299b94acceSBarry Smith   ierr = (*snes->computeumfunction)(snes,x,y,snes->umfunP); CHKERRQ(ierr);
6309b94acceSBarry Smith   PLogEventEnd(SNES_MinimizationFunctionEval,snes,x,y,0);
6319b94acceSBarry Smith   return 0;
6329b94acceSBarry Smith }
6339b94acceSBarry Smith 
6349b94acceSBarry Smith /*@C
6359b94acceSBarry Smith    SNESSetGradient - Sets the gradient evaluation routine and gradient
6369b94acceSBarry Smith    vector for use by the SNES routines.
6379b94acceSBarry Smith 
6389b94acceSBarry Smith    Input Parameters:
6399b94acceSBarry Smith .  snes - the SNES context
6409b94acceSBarry Smith .  func - function evaluation routine
6419b94acceSBarry Smith .  ctx - optional user-defined function context
6429b94acceSBarry Smith .  r - vector to store gradient value
6439b94acceSBarry Smith 
6449b94acceSBarry Smith    Calling sequence of func:
6459b94acceSBarry Smith .  func (SNES, Vec x, Vec g, void *ctx);
6469b94acceSBarry Smith 
6479b94acceSBarry Smith .  x - input vector
6489b94acceSBarry Smith .  g - gradient vector
6499b94acceSBarry Smith .  ctx - optional user-defined context for private data for the
6509b94acceSBarry Smith          function evaluation routine (may be null)
6519b94acceSBarry Smith 
6529b94acceSBarry Smith    Notes:
6539b94acceSBarry Smith    SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION
6549b94acceSBarry Smith    methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is
6559b94acceSBarry Smith    SNESSetFunction().
6569b94acceSBarry Smith 
6579b94acceSBarry Smith .keywords: SNES, nonlinear, set, function
6589b94acceSBarry Smith 
659a86d99e1SLois Curfman McInnes .seealso: SNESGetGradient(), SNESComputeGradient(), SNESSetHessian(),
660a86d99e1SLois Curfman McInnes           SNESSetMinimizationFunction(),
6619b94acceSBarry Smith @*/
6629b94acceSBarry Smith int SNESSetGradient(SNES snes,Vec r,int (*func)(SNES,Vec,Vec,void*),
6639b94acceSBarry Smith                      void *ctx)
6649b94acceSBarry Smith {
66577c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
6669b94acceSBarry Smith   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
66748d91487SBarry Smith     "SNESSetGradient:For SNES_UNCONSTRAINED_MINIMIZATION only");
6689b94acceSBarry Smith   snes->computefunction     = func;
6699b94acceSBarry Smith   snes->vec_func            = snes->vec_func_always = r;
6709b94acceSBarry Smith   snes->funP                = ctx;
6719b94acceSBarry Smith   return 0;
6729b94acceSBarry Smith }
6739b94acceSBarry Smith 
6749b94acceSBarry Smith /*@
675a86d99e1SLois Curfman McInnes    SNESComputeGradient - Computes the gradient that has been set with
676a86d99e1SLois Curfman McInnes    SNESSetGradient().
6779b94acceSBarry Smith 
6789b94acceSBarry Smith    Input Parameters:
6799b94acceSBarry Smith .  snes - the SNES context
6809b94acceSBarry Smith .  x - input vector
6819b94acceSBarry Smith 
6829b94acceSBarry Smith    Output Parameter:
6839b94acceSBarry Smith .  y - gradient vector
6849b94acceSBarry Smith 
6859b94acceSBarry Smith    Notes:
6869b94acceSBarry Smith    SNESComputeGradient() is valid only for
6879b94acceSBarry Smith    SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for
6889b94acceSBarry Smith    SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction().
689a86d99e1SLois Curfman McInnes 
690a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, gradient
691a86d99e1SLois Curfman McInnes 
692a86d99e1SLois Curfman McInnes .seealso:  SNESSetGradient(), SNESGetGradient(),
693a86d99e1SLois Curfman McInnes            SNESComputeMinimizationFunction(), SNESComputeHessian()
6949b94acceSBarry Smith @*/
6959b94acceSBarry Smith int SNESComputeGradient(SNES snes,Vec x, Vec y)
6969b94acceSBarry Smith {
6979b94acceSBarry Smith   int    ierr;
6989b94acceSBarry Smith   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
69948d91487SBarry Smith     "SNESComputeGradient:For SNES_UNCONSTRAINED_MINIMIZATION only");
7009b94acceSBarry Smith   PLogEventBegin(SNES_GradientEval,snes,x,y,0);
7019b94acceSBarry Smith   ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr);
7029b94acceSBarry Smith   PLogEventEnd(SNES_GradientEval,snes,x,y,0);
7039b94acceSBarry Smith   return 0;
7049b94acceSBarry Smith }
7059b94acceSBarry Smith 
70662fef451SLois Curfman McInnes /*@
70762fef451SLois Curfman McInnes    SNESComputeJacobian - Computes the Jacobian matrix that has been
70862fef451SLois Curfman McInnes    set with SNESSetJacobian().
70962fef451SLois Curfman McInnes 
71062fef451SLois Curfman McInnes    Input Parameters:
71162fef451SLois Curfman McInnes .  snes - the SNES context
71262fef451SLois Curfman McInnes .  x - input vector
71362fef451SLois Curfman McInnes 
71462fef451SLois Curfman McInnes    Output Parameters:
71562fef451SLois Curfman McInnes .  A - Jacobian matrix
71662fef451SLois Curfman McInnes .  B - optional preconditioning matrix
71762fef451SLois Curfman McInnes .  flag - flag indicating matrix structure
71862fef451SLois Curfman McInnes 
71962fef451SLois Curfman McInnes    Notes:
72062fef451SLois Curfman McInnes    Most users should not need to explicitly call this routine, as it
72162fef451SLois Curfman McInnes    is used internally within the nonlinear solvers.
72262fef451SLois Curfman McInnes 
72362fef451SLois Curfman McInnes    See SLESSetOperators() for information about setting the flag parameter.
72462fef451SLois Curfman McInnes 
72562fef451SLois Curfman McInnes    SNESComputeJacobian() is valid only for SNES_NONLINEAR_EQUATIONS
72662fef451SLois Curfman McInnes    methods. An analogous routine for SNES_UNCONSTRAINED_MINIMIZATION
72762fef451SLois Curfman McInnes    methods is SNESComputeJacobian().
72862fef451SLois Curfman McInnes 
72962fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix
73062fef451SLois Curfman McInnes 
73162fef451SLois Curfman McInnes .seealso:  SNESSetJacobian(), SLESSetOperators()
73262fef451SLois Curfman McInnes @*/
7339b94acceSBarry Smith int SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg)
7349b94acceSBarry Smith {
7359b94acceSBarry Smith   int    ierr;
7369b94acceSBarry Smith   if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,
73748d91487SBarry Smith     "SNESComputeJacobian: For SNES_NONLINEAR_EQUATIONS only");
7389b94acceSBarry Smith   if (!snes->computejacobian) return 0;
7399b94acceSBarry Smith   PLogEventBegin(SNES_JacobianEval,snes,X,*A,*B);
740c4fc05e7SBarry Smith   *flg = DIFFERENT_NONZERO_PATTERN;
7419b94acceSBarry Smith   ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr);
7429b94acceSBarry Smith   PLogEventEnd(SNES_JacobianEval,snes,X,*A,*B);
7436d84be18SBarry Smith   /* make sure user returned a correct Jacobian and preconditioner */
74477c4ece6SBarry Smith   PetscValidHeaderSpecific(*A,MAT_COOKIE);
74577c4ece6SBarry Smith   PetscValidHeaderSpecific(*B,MAT_COOKIE);
7469b94acceSBarry Smith   return 0;
7479b94acceSBarry Smith }
7489b94acceSBarry Smith 
74962fef451SLois Curfman McInnes /*@
75062fef451SLois Curfman McInnes    SNESComputeHessian - Computes the Hessian matrix that has been
75162fef451SLois Curfman McInnes    set with SNESSetHessian().
75262fef451SLois Curfman McInnes 
75362fef451SLois Curfman McInnes    Input Parameters:
75462fef451SLois Curfman McInnes .  snes - the SNES context
75562fef451SLois Curfman McInnes .  x - input vector
75662fef451SLois Curfman McInnes 
75762fef451SLois Curfman McInnes    Output Parameters:
75862fef451SLois Curfman McInnes .  A - Hessian matrix
75962fef451SLois Curfman McInnes .  B - optional preconditioning matrix
76062fef451SLois Curfman McInnes .  flag - flag indicating matrix structure
76162fef451SLois Curfman McInnes 
76262fef451SLois Curfman McInnes    Notes:
76362fef451SLois Curfman McInnes    Most users should not need to explicitly call this routine, as it
76462fef451SLois Curfman McInnes    is used internally within the nonlinear solvers.
76562fef451SLois Curfman McInnes 
76662fef451SLois Curfman McInnes    See SLESSetOperators() for information about setting the flag parameter.
76762fef451SLois Curfman McInnes 
76862fef451SLois Curfman McInnes    SNESComputeHessian() is valid only for
76962fef451SLois Curfman McInnes    SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for
77062fef451SLois Curfman McInnes    SNES_NONLINEAR_EQUATIONS methods is SNESComputeJacobian().
77162fef451SLois Curfman McInnes 
77262fef451SLois Curfman McInnes .keywords: SNES, compute, Hessian, matrix
77362fef451SLois Curfman McInnes 
774a86d99e1SLois Curfman McInnes .seealso:  SNESSetHessian(), SLESSetOperators(), SNESComputeGradient(),
775a86d99e1SLois Curfman McInnes            SNESComputeMinimizationFunction()
77662fef451SLois Curfman McInnes @*/
77762fef451SLois Curfman McInnes int SNESComputeHessian(SNES snes,Vec x,Mat *A,Mat *B,MatStructure *flag)
7789b94acceSBarry Smith {
7799b94acceSBarry Smith   int    ierr;
7809b94acceSBarry Smith   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
78148d91487SBarry Smith     "SNESComputeHessian:For SNES_UNCONSTRAINED_MINIMIZATION only");
7829b94acceSBarry Smith   if (!snes->computejacobian) return 0;
78362fef451SLois Curfman McInnes   PLogEventBegin(SNES_HessianEval,snes,x,*A,*B);
7840f4a323eSLois Curfman McInnes   *flag = DIFFERENT_NONZERO_PATTERN;
78562fef451SLois Curfman McInnes   ierr = (*snes->computejacobian)(snes,x,A,B,flag,snes->jacP); CHKERRQ(ierr);
78662fef451SLois Curfman McInnes   PLogEventEnd(SNES_HessianEval,snes,x,*A,*B);
7870f4a323eSLois Curfman McInnes   /* make sure user returned a correct Jacobian and preconditioner */
78877c4ece6SBarry Smith   PetscValidHeaderSpecific(*A,MAT_COOKIE);
78977c4ece6SBarry Smith   PetscValidHeaderSpecific(*B,MAT_COOKIE);
7909b94acceSBarry Smith   return 0;
7919b94acceSBarry Smith }
7929b94acceSBarry Smith 
7939b94acceSBarry Smith /*@C
7949b94acceSBarry Smith    SNESSetJacobian - Sets the function to compute Jacobian as well as the
7959b94acceSBarry Smith    location to store it.
7969b94acceSBarry Smith 
7979b94acceSBarry Smith    Input Parameters:
7989b94acceSBarry Smith .  snes - the SNES context
7999b94acceSBarry Smith .  A - Jacobian matrix
8009b94acceSBarry Smith .  B - preconditioner matrix (usually same as the Jacobian)
8019b94acceSBarry Smith .  func - Jacobian evaluation routine
8029b94acceSBarry Smith .  ctx - optional user-defined context for private data for the
8039b94acceSBarry Smith          Jacobian evaluation routine (may be null)
8049b94acceSBarry Smith 
8059b94acceSBarry Smith    Calling sequence of func:
8069b94acceSBarry Smith .  func (SNES,Vec x,Mat *A,Mat *B,int *flag,void *ctx);
8079b94acceSBarry Smith 
8089b94acceSBarry Smith .  x - input vector
8099b94acceSBarry Smith .  A - Jacobian matrix
8109b94acceSBarry Smith .  B - preconditioner matrix, usually the same as A
811ac21db08SLois Curfman McInnes .  flag - flag indicating information about the preconditioner matrix
812ac21db08SLois Curfman McInnes    structure (same as flag in SLESSetOperators())
8139b94acceSBarry Smith .  ctx - optional user-defined Jacobian context
8149b94acceSBarry Smith 
8159b94acceSBarry Smith    Notes:
816ac21db08SLois Curfman McInnes    See SLESSetOperators() for information about setting the flag input
817ac21db08SLois Curfman McInnes    parameter in the routine func().  Be sure to read this information!
818ac21db08SLois Curfman McInnes 
819ac21db08SLois Curfman McInnes    The routine func() takes Mat * as the matrix arguments rather than Mat.
8209b94acceSBarry Smith    This allows the Jacobian evaluation routine to replace A and/or B with a
8219b94acceSBarry Smith    completely new new matrix structure (not just different matrix elements)
8229b94acceSBarry Smith    when appropriate, for instance, if the nonzero structure is changing
8239b94acceSBarry Smith    throughout the global iterations.
8249b94acceSBarry Smith 
8259b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix
8269b94acceSBarry Smith 
827ac21db08SLois Curfman McInnes .seealso: SLESSetOperators(), SNESSetFunction()
8289b94acceSBarry Smith @*/
8299b94acceSBarry Smith int SNESSetJacobian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*,
8309b94acceSBarry Smith                     MatStructure*,void*),void *ctx)
8319b94acceSBarry Smith {
83277c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
8339b94acceSBarry Smith   if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,
83448d91487SBarry Smith     "SNESSetJacobian:For SNES_NONLINEAR_EQUATIONS only");
8359b94acceSBarry Smith   snes->computejacobian = func;
8369b94acceSBarry Smith   snes->jacP            = ctx;
8379b94acceSBarry Smith   snes->jacobian        = A;
8389b94acceSBarry Smith   snes->jacobian_pre    = B;
8399b94acceSBarry Smith   return 0;
8409b94acceSBarry Smith }
84162fef451SLois Curfman McInnes 
842b4fd4287SBarry Smith /*@
843b4fd4287SBarry Smith    SNESGetJacobian - Returns the Jacobian matrix and optionally the user
844b4fd4287SBarry Smith    provided context for evaluating the Jacobian.
845b4fd4287SBarry Smith 
846b4fd4287SBarry Smith    Input Parameter:
847b4fd4287SBarry Smith .  snes - the nonlinear solver context
848b4fd4287SBarry Smith 
849b4fd4287SBarry Smith    Output Parameters:
850b4fd4287SBarry Smith .  A - location to stash Jacobian matrix (or PETSC_NULL)
851b4fd4287SBarry Smith .  B - location to stash preconditioner matrix (or PETSC_NULL)
852b4fd4287SBarry Smith .  ctx - location to stash Jacobian ctx (or PETSC_NULL)
853b4fd4287SBarry Smith 
854b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian()
855b4fd4287SBarry Smith @*/
856b4fd4287SBarry Smith int SNESGetJacobian(SNES snes,Mat *A,Mat *B, void **ctx)
857b4fd4287SBarry Smith {
85862fef451SLois Curfman McInnes   if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,
85962fef451SLois Curfman McInnes     "SNESSetJacobian:For SNES_NONLINEAR_EQUATIONS only");
860b4fd4287SBarry Smith   if (A)   *A = snes->jacobian;
861b4fd4287SBarry Smith   if (B)   *B = snes->jacobian_pre;
862b4fd4287SBarry Smith   if (ctx) *ctx = snes->jacP;
863b4fd4287SBarry Smith   return 0;
864b4fd4287SBarry Smith }
865b4fd4287SBarry Smith 
8669b94acceSBarry Smith /*@C
8679b94acceSBarry Smith    SNESSetHessian - Sets the function to compute Hessian as well as the
8689b94acceSBarry Smith    location to store it.
8699b94acceSBarry Smith 
8709b94acceSBarry Smith    Input Parameters:
8719b94acceSBarry Smith .  snes - the SNES context
8729b94acceSBarry Smith .  A - Hessian matrix
8739b94acceSBarry Smith .  B - preconditioner matrix (usually same as the Hessian)
8749b94acceSBarry Smith .  func - Jacobian evaluation routine
8759b94acceSBarry Smith .  ctx - optional user-defined context for private data for the
8769b94acceSBarry Smith          Hessian evaluation routine (may be null)
8779b94acceSBarry Smith 
8789b94acceSBarry Smith    Calling sequence of func:
8799b94acceSBarry Smith .  func (SNES,Vec x,Mat *A,Mat *B,int *flag,void *ctx);
8809b94acceSBarry Smith 
8819b94acceSBarry Smith .  x - input vector
8829b94acceSBarry Smith .  A - Hessian matrix
8839b94acceSBarry Smith .  B - preconditioner matrix, usually the same as A
884ac21db08SLois Curfman McInnes .  flag - flag indicating information about the preconditioner matrix
885ac21db08SLois Curfman McInnes    structure (same as flag in SLESSetOperators())
8869b94acceSBarry Smith .  ctx - optional user-defined Hessian context
8879b94acceSBarry Smith 
8889b94acceSBarry Smith    Notes:
889ac21db08SLois Curfman McInnes    See SLESSetOperators() for information about setting the flag input
890ac21db08SLois Curfman McInnes    parameter in the routine func().  Be sure to read this information!
891ac21db08SLois Curfman McInnes 
8929b94acceSBarry Smith    The function func() takes Mat * as the matrix arguments rather than Mat.
8939b94acceSBarry Smith    This allows the Hessian evaluation routine to replace A and/or B with a
8949b94acceSBarry Smith    completely new new matrix structure (not just different matrix elements)
8959b94acceSBarry Smith    when appropriate, for instance, if the nonzero structure is changing
8969b94acceSBarry Smith    throughout the global iterations.
8979b94acceSBarry Smith 
8989b94acceSBarry Smith .keywords: SNES, nonlinear, set, Hessian, matrix
8999b94acceSBarry Smith 
900ac21db08SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESSetGradient(), SLESSetOperators()
9019b94acceSBarry Smith @*/
9029b94acceSBarry Smith int SNESSetHessian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*,
9039b94acceSBarry Smith                     MatStructure*,void*),void *ctx)
9049b94acceSBarry Smith {
90577c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
9069b94acceSBarry Smith   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
90748d91487SBarry Smith     "SNESSetHessian:For SNES_UNCONSTRAINED_MINIMIZATION only");
9089b94acceSBarry Smith   snes->computejacobian = func;
9099b94acceSBarry Smith   snes->jacP            = ctx;
9109b94acceSBarry Smith   snes->jacobian        = A;
9119b94acceSBarry Smith   snes->jacobian_pre    = B;
9129b94acceSBarry Smith   return 0;
9139b94acceSBarry Smith }
9149b94acceSBarry Smith 
91562fef451SLois Curfman McInnes /*@
91662fef451SLois Curfman McInnes    SNESGetHessian - Returns the Hessian matrix and optionally the user
91762fef451SLois Curfman McInnes    provided context for evaluating the Hessian.
91862fef451SLois Curfman McInnes 
91962fef451SLois Curfman McInnes    Input Parameter:
92062fef451SLois Curfman McInnes .  snes - the nonlinear solver context
92162fef451SLois Curfman McInnes 
92262fef451SLois Curfman McInnes    Output Parameters:
92362fef451SLois Curfman McInnes .  A - location to stash Hessian matrix (or PETSC_NULL)
92462fef451SLois Curfman McInnes .  B - location to stash preconditioner matrix (or PETSC_NULL)
92562fef451SLois Curfman McInnes .  ctx - location to stash Hessian ctx (or PETSC_NULL)
92662fef451SLois Curfman McInnes 
92762fef451SLois Curfman McInnes .seealso: SNESSetHessian(), SNESComputeHessian()
92862fef451SLois Curfman McInnes @*/
92962fef451SLois Curfman McInnes int SNESGetHessian(SNES snes,Mat *A,Mat *B, void **ctx)
93062fef451SLois Curfman McInnes {
93162fef451SLois Curfman McInnes   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
93262fef451SLois Curfman McInnes     "SNESSetHessian:For SNES_UNCONSTRAINED_MINIMIZATION only");
93362fef451SLois Curfman McInnes   if (A)   *A = snes->jacobian;
93462fef451SLois Curfman McInnes   if (B)   *B = snes->jacobian_pre;
93562fef451SLois Curfman McInnes   if (ctx) *ctx = snes->jacP;
93662fef451SLois Curfman McInnes   return 0;
93762fef451SLois Curfman McInnes }
93862fef451SLois Curfman McInnes 
9399b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */
9409b94acceSBarry Smith 
9419b94acceSBarry Smith /*@
9429b94acceSBarry Smith    SNESSetUp - Sets up the internal data structures for the later use
943272ac6f2SLois Curfman McInnes    of a nonlinear solver.
9449b94acceSBarry Smith 
9459b94acceSBarry Smith    Input Parameter:
9469b94acceSBarry Smith .  snes - the SNES context
9478ddd3da0SLois Curfman McInnes .  x - the solution vector
9489b94acceSBarry Smith 
949272ac6f2SLois Curfman McInnes    Notes:
950272ac6f2SLois Curfman McInnes    For basic use of the SNES solvers the user need not explicitly call
951272ac6f2SLois Curfman McInnes    SNESSetUp(), since these actions will automatically occur during
952272ac6f2SLois Curfman McInnes    the call to SNESSolve().  However, if one wishes to control this
953272ac6f2SLois Curfman McInnes    phase separately, SNESSetUp() should be called after SNESCreate()
954272ac6f2SLois Curfman McInnes    and optional routines of the form SNESSetXXX(), but before SNESSolve().
955272ac6f2SLois Curfman McInnes 
9569b94acceSBarry Smith .keywords: SNES, nonlinear, setup
9579b94acceSBarry Smith 
9589b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy()
9599b94acceSBarry Smith @*/
9608ddd3da0SLois Curfman McInnes int SNESSetUp(SNES snes,Vec x)
9619b94acceSBarry Smith {
962272ac6f2SLois Curfman McInnes   int ierr, flg;
96377c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
96477c4ece6SBarry Smith   PetscValidHeaderSpecific(x,VEC_COOKIE);
9658ddd3da0SLois Curfman McInnes   snes->vec_sol = snes->vec_sol_always = x;
9669b94acceSBarry Smith 
967272ac6f2SLois Curfman McInnes   ierr = OptionsHasName(snes->prefix,"-snes_mf", &flg);  CHKERRQ(ierr);
96831615d04SLois Curfman McInnes   if (flg) {
969272ac6f2SLois Curfman McInnes     Mat J;
970272ac6f2SLois Curfman McInnes     ierr = SNESDefaultMatrixFreeMatCreate(snes,snes->vec_sol,&J);CHKERRQ(ierr);
971272ac6f2SLois Curfman McInnes     PLogObjectParent(snes,J);
972272ac6f2SLois Curfman McInnes     snes->mfshell = J;
97383e56afdSLois Curfman McInnes     if (snes->method_class == SNES_NONLINEAR_EQUATIONS) {
97483e56afdSLois Curfman McInnes       ierr = SNESSetJacobian(snes,J,J,0,snes->funP); CHKERRQ(ierr);
97594a424c1SBarry Smith       PLogInfo(snes,"SNESSetUp: Setting default matrix-free Jacobian routines\n");
97683e56afdSLois Curfman McInnes     }
97783e56afdSLois Curfman McInnes     else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) {
97883e56afdSLois Curfman McInnes       ierr = SNESSetHessian(snes,J,J,0,snes->funP); CHKERRQ(ierr);
97994a424c1SBarry Smith       PLogInfo(snes,"SNESSetUp: Setting default matrix-free Hessian routines\n");
98083e56afdSLois Curfman McInnes     } else SETERRQ(1,"SNESSetUp:Method class doesn't support matrix-free option");
981272ac6f2SLois Curfman McInnes   }
9829b94acceSBarry Smith   if ((snes->method_class == SNES_NONLINEAR_EQUATIONS)) {
98337fcc0dbSBarry Smith     if (!snes->vec_func) SETERRQ(1,"SNESSetUp:Must call SNESSetFunction() first");
98437fcc0dbSBarry Smith     if (!snes->computefunction) SETERRQ(1,"SNESSetUp:Must call SNESSetFunction() first");
98537fcc0dbSBarry Smith     if (!snes->jacobian) SETERRQ(1,"SNESSetUp:Must call SNESSetJacobian() first");
986d804570eSBarry Smith     if (snes->vec_func == snes->vec_sol) SETERRQ(1,"SNESSetUp:Solution vector cannot be function vector");
987a703fe33SLois Curfman McInnes 
988a703fe33SLois Curfman McInnes     /* Set the KSP stopping criterion to use the Eisenstat-Walker method */
98940191667SLois Curfman McInnes     if (snes->ksp_ewconv && snes->type != SNES_EQ_TR) {
990a703fe33SLois Curfman McInnes       SLES sles; KSP ksp;
991a703fe33SLois Curfman McInnes       ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr);
992a703fe33SLois Curfman McInnes       ierr = SLESGetKSP(sles,&ksp); CHKERRQ(ierr);
993a703fe33SLois Curfman McInnes       ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private,
994a703fe33SLois Curfman McInnes              (void *)snes); CHKERRQ(ierr);
995a703fe33SLois Curfman McInnes     }
9969b94acceSBarry Smith   } else if ((snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION)) {
99737fcc0dbSBarry Smith     if (!snes->vec_func) SETERRQ(1,"SNESSetUp:Must call SNESSetGradient() first");
99837fcc0dbSBarry Smith     if (!snes->computefunction) SETERRQ(1,"SNESSetUp:Must call SNESSetGradient() first");
99937fcc0dbSBarry Smith     if (!snes->computeumfunction)
100037fcc0dbSBarry Smith       SETERRQ(1,"SNESSetUp:Must call SNESSetMinimizationFunction() first");
100137fcc0dbSBarry Smith     if (!snes->jacobian) SETERRQ(1,"SNESSetUp:Must call SNESSetHessian() first");
10029b94acceSBarry Smith   } else SETERRQ(1,"SNESSetUp:Unknown method class");
1003a703fe33SLois Curfman McInnes   if (snes->setup) {ierr = (*snes->setup)(snes); CHKERRQ(ierr);}
1004a703fe33SLois Curfman McInnes   snes->setup_called = 1;
1005a703fe33SLois Curfman McInnes   return 0;
10069b94acceSBarry Smith }
10079b94acceSBarry Smith 
10089b94acceSBarry Smith /*@C
10099b94acceSBarry Smith    SNESDestroy - Destroys the nonlinear solver context that was created
10109b94acceSBarry Smith    with SNESCreate().
10119b94acceSBarry Smith 
10129b94acceSBarry Smith    Input Parameter:
10139b94acceSBarry Smith .  snes - the SNES context
10149b94acceSBarry Smith 
10159b94acceSBarry Smith .keywords: SNES, nonlinear, destroy
10169b94acceSBarry Smith 
101763a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve()
10189b94acceSBarry Smith @*/
10199b94acceSBarry Smith int SNESDestroy(SNES snes)
10209b94acceSBarry Smith {
10219b94acceSBarry Smith   int ierr;
102277c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
10239b94acceSBarry Smith   ierr = (*(snes)->destroy)((PetscObject)snes); CHKERRQ(ierr);
10240452661fSBarry Smith   if (snes->kspconvctx) PetscFree(snes->kspconvctx);
10259b94acceSBarry Smith   if (snes->mfshell) MatDestroy(snes->mfshell);
10269b94acceSBarry Smith   ierr = SLESDestroy(snes->sles); CHKERRQ(ierr);
10273e2e452bSBarry Smith   if (snes->xmonitor) SNESLGMonitorDestroy(snes->xmonitor);
10289b94acceSBarry Smith   PLogObjectDestroy((PetscObject)snes);
10290452661fSBarry Smith   PetscHeaderDestroy((PetscObject)snes);
10309b94acceSBarry Smith   return 0;
10319b94acceSBarry Smith }
10329b94acceSBarry Smith 
10339b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */
10349b94acceSBarry Smith 
10359b94acceSBarry Smith /*@
1036d7a720efSLois Curfman McInnes    SNESSetTolerances - Sets various parameters used in convergence tests.
10379b94acceSBarry Smith 
10389b94acceSBarry Smith    Input Parameters:
10399b94acceSBarry Smith .  snes - the SNES context
1040d7a720efSLois Curfman McInnes .  atol - tolerance
10419b94acceSBarry Smith 
10429b94acceSBarry Smith    Options Database Key:
1043d7a720efSLois Curfman McInnes $    -snes_atol <atol> - absolute convergence tolerance
1044d7a720efSLois Curfman McInnes $    -snes_rtol <rtol> - relative convergence tolerance
1045d7a720efSLois Curfman McInnes $    -snes_stol <stol> - convergence tolerance in terms of
1046d7a720efSLois Curfman McInnes $          the norm of the change in the solution between steps
1047d7a720efSLois Curfman McInnes $    -snes_max_it <maxit> - maximum number of iterations
1048d7a720efSLois Curfman McInnes $    -snes_max_funcs <maxf> - maximum number of function evaluations
10499b94acceSBarry Smith 
1050d7a720efSLois Curfman McInnes    Notes:
10519b94acceSBarry Smith    The default maximum number of iterations is 50.
10529b94acceSBarry Smith    The default maximum number of function evaluations is 1000.
10539b94acceSBarry Smith 
1054d7a720efSLois Curfman McInnes .keywords: SNES, nonlinear, set, absolute, convergence, tolerance
10559b94acceSBarry Smith 
1056d7a720efSLois Curfman McInnes .seealso: SNESSetTrustRegionTolerance(), SNESSetMinimizationFunctionTolerance()
10579b94acceSBarry Smith @*/
1058d7a720efSLois Curfman McInnes int SNESSetTolerances(SNES snes,double atol,double rtol,double stol,int maxit,int maxf)
10599b94acceSBarry Smith {
106077c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
1061d7a720efSLois Curfman McInnes   if (atol != PETSC_DEFAULT)  snes->atol      = atol;
1062d7a720efSLois Curfman McInnes   if (rtol != PETSC_DEFAULT)  snes->rtol      = rtol;
1063d7a720efSLois Curfman McInnes   if (stol != PETSC_DEFAULT)  snes->xtol      = stol;
1064d7a720efSLois Curfman McInnes   if (maxit != PETSC_DEFAULT) snes->max_its   = maxit;
1065d7a720efSLois Curfman McInnes   if (maxf != PETSC_DEFAULT)  snes->max_funcs = maxf;
10669b94acceSBarry Smith   return 0;
10679b94acceSBarry Smith }
10689b94acceSBarry Smith 
10699b94acceSBarry Smith /*@
10709b94acceSBarry Smith    SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance.
10719b94acceSBarry Smith 
10729b94acceSBarry Smith    Input Parameters:
10739b94acceSBarry Smith .  snes - the SNES context
10749b94acceSBarry Smith .  tol - tolerance
10759b94acceSBarry Smith 
10769b94acceSBarry Smith    Options Database Key:
1077d7a720efSLois Curfman McInnes $    -snes_trtol <tol>
10789b94acceSBarry Smith 
10799b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance
10809b94acceSBarry Smith 
1081d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetMinimizationFunctionTolerance()
10829b94acceSBarry Smith @*/
10839b94acceSBarry Smith int SNESSetTrustRegionTolerance(SNES snes,double tol)
10849b94acceSBarry Smith {
108577c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
10869b94acceSBarry Smith   snes->deltatol = tol;
10879b94acceSBarry Smith   return 0;
10889b94acceSBarry Smith }
10899b94acceSBarry Smith 
10909b94acceSBarry Smith /*@
109177c4ece6SBarry Smith    SNESSetMinimizationFunctionTolerance - Sets the minimum allowable function tolerance
10929b94acceSBarry Smith    for unconstrained minimization solvers.
10939b94acceSBarry Smith 
10949b94acceSBarry Smith    Input Parameters:
10959b94acceSBarry Smith .  snes - the SNES context
10969b94acceSBarry Smith .  ftol - minimum function tolerance
10979b94acceSBarry Smith 
10989b94acceSBarry Smith    Options Database Key:
1099d7a720efSLois Curfman McInnes $    -snes_fmin <ftol>
11009b94acceSBarry Smith 
11019b94acceSBarry Smith    Note:
110277c4ece6SBarry Smith    SNESSetMinimizationFunctionTolerance() is valid for SNES_UNCONSTRAINED_MINIMIZATION
11039b94acceSBarry Smith    methods only.
11049b94acceSBarry Smith 
11059b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimum, convergence, function, tolerance
11069b94acceSBarry Smith 
1107d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetTrustRegionTolerance()
11089b94acceSBarry Smith @*/
110977c4ece6SBarry Smith int SNESSetMinimizationFunctionTolerance(SNES snes,double ftol)
11109b94acceSBarry Smith {
111177c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
11129b94acceSBarry Smith   snes->fmin = ftol;
11139b94acceSBarry Smith   return 0;
11149b94acceSBarry Smith }
11159b94acceSBarry Smith 
11169b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */
11179b94acceSBarry Smith 
11189b94acceSBarry Smith /*@C
11199b94acceSBarry Smith    SNESSetMonitor - Sets the function that is to be used at every
11209b94acceSBarry Smith    iteration of the nonlinear solver to display the iteration's
11219b94acceSBarry Smith    progress.
11229b94acceSBarry Smith 
11239b94acceSBarry Smith    Input Parameters:
11249b94acceSBarry Smith .  snes - the SNES context
11259b94acceSBarry Smith .  func - monitoring routine
11269b94acceSBarry Smith .  mctx - optional user-defined context for private data for the
11279b94acceSBarry Smith           monitor routine (may be null)
11289b94acceSBarry Smith 
11299b94acceSBarry Smith    Calling sequence of func:
1130682d7d0cSBarry Smith    int func(SNES snes,int its, Vec x,Vec f,double norm,void *mctx)
11319b94acceSBarry Smith 
11329b94acceSBarry Smith $    snes - the SNES context
11339b94acceSBarry Smith $    its - iteration number
11349b94acceSBarry Smith $    mctx - optional monitoring context
11359b94acceSBarry Smith $
11369b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods:
11379b94acceSBarry Smith $    norm - 2-norm function value (may be estimated)
11389b94acceSBarry Smith $
11399b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods:
11409b94acceSBarry Smith $    norm - 2-norm gradient value (may be estimated)
11419b94acceSBarry Smith 
11429b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor
11439b94acceSBarry Smith 
11449b94acceSBarry Smith .seealso: SNESDefaultMonitor()
11459b94acceSBarry Smith @*/
11469b94acceSBarry Smith int SNESSetMonitor( SNES snes, int (*func)(SNES,int,double,void*),
11479b94acceSBarry Smith                     void *mctx )
11489b94acceSBarry Smith {
11499b94acceSBarry Smith   snes->monitor = func;
11509b94acceSBarry Smith   snes->monP    = (void*)mctx;
11519b94acceSBarry Smith   return 0;
11529b94acceSBarry Smith }
11539b94acceSBarry Smith 
11549b94acceSBarry Smith /*@C
11559b94acceSBarry Smith    SNESSetConvergenceTest - Sets the function that is to be used
11569b94acceSBarry Smith    to test for convergence of the nonlinear iterative solution.
11579b94acceSBarry Smith 
11589b94acceSBarry Smith    Input Parameters:
11599b94acceSBarry Smith .  snes - the SNES context
11609b94acceSBarry Smith .  func - routine to test for convergence
11619b94acceSBarry Smith .  cctx - optional context for private data for the convergence routine
11629b94acceSBarry Smith           (may be null)
11639b94acceSBarry Smith 
11649b94acceSBarry Smith    Calling sequence of func:
11659b94acceSBarry Smith    int func (SNES snes,double xnorm,double gnorm,
11669b94acceSBarry Smith              double f,void *cctx)
11679b94acceSBarry Smith 
11689b94acceSBarry Smith $    snes - the SNES context
11699b94acceSBarry Smith $    cctx - optional convergence context
11709b94acceSBarry Smith $    xnorm - 2-norm of current iterate
11719b94acceSBarry Smith $
11729b94acceSBarry Smith $ SNES_NONLINEAR_EQUATIONS methods:
11739b94acceSBarry Smith $    gnorm - 2-norm of current step
11749b94acceSBarry Smith $    f - 2-norm of function
11759b94acceSBarry Smith $
11769b94acceSBarry Smith $ SNES_UNCONSTRAINED_MINIMIZATION methods:
11779b94acceSBarry Smith $    gnorm - 2-norm of current gradient
11789b94acceSBarry Smith $    f - function value
11799b94acceSBarry Smith 
11809b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test
11819b94acceSBarry Smith 
118240191667SLois Curfman McInnes .seealso: SNESConverged_EQ_LS(), SNESConverged_EQ_TR(),
118340191667SLois Curfman McInnes           SNESConverged_UM_LS(), SNESConverged_UM_TR()
11849b94acceSBarry Smith @*/
11859b94acceSBarry Smith int SNESSetConvergenceTest(SNES snes,
11869b94acceSBarry Smith           int (*func)(SNES,double,double,double,void*),void *cctx)
11879b94acceSBarry Smith {
11889b94acceSBarry Smith   (snes)->converged = func;
11899b94acceSBarry Smith   (snes)->cnvP      = cctx;
11909b94acceSBarry Smith   return 0;
11919b94acceSBarry Smith }
11929b94acceSBarry Smith 
11939b94acceSBarry Smith /*
11949b94acceSBarry Smith    SNESScaleStep_Private - Scales a step so that its length is less than the
11959b94acceSBarry Smith    positive parameter delta.
11969b94acceSBarry Smith 
11979b94acceSBarry Smith     Input Parameters:
11989b94acceSBarry Smith .   snes - the SNES context
11999b94acceSBarry Smith .   y - approximate solution of linear system
12009b94acceSBarry Smith .   fnorm - 2-norm of current function
12019b94acceSBarry Smith .   delta - trust region size
12029b94acceSBarry Smith 
12039b94acceSBarry Smith     Output Parameters:
12049b94acceSBarry Smith .   gpnorm - predicted function norm at the new point, assuming local
12059b94acceSBarry Smith     linearization.  The value is zero if the step lies within the trust
12069b94acceSBarry Smith     region, and exceeds zero otherwise.
12079b94acceSBarry Smith .   ynorm - 2-norm of the step
12089b94acceSBarry Smith 
12099b94acceSBarry Smith     Note:
121040191667SLois Curfman McInnes     For non-trust region methods such as SNES_EQ_LS, the parameter delta
12119b94acceSBarry Smith     is set to be the maximum allowable step size.
12129b94acceSBarry Smith 
12139b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step
12149b94acceSBarry Smith */
12159b94acceSBarry Smith int SNESScaleStep_Private(SNES snes,Vec y,double *fnorm,double *delta,
12169b94acceSBarry Smith                   double *gpnorm,double *ynorm)
12179b94acceSBarry Smith {
12189b94acceSBarry Smith   double norm;
12199b94acceSBarry Smith   Scalar cnorm;
1220cddf8d76SBarry Smith   VecNorm(y,NORM_2, &norm );
12219b94acceSBarry Smith   if (norm > *delta) {
12229b94acceSBarry Smith      norm = *delta/norm;
12239b94acceSBarry Smith      *gpnorm = (1.0 - norm)*(*fnorm);
12249b94acceSBarry Smith      cnorm = norm;
12259b94acceSBarry Smith      VecScale( &cnorm, y );
12269b94acceSBarry Smith      *ynorm = *delta;
12279b94acceSBarry Smith   } else {
12289b94acceSBarry Smith      *gpnorm = 0.0;
12299b94acceSBarry Smith      *ynorm = norm;
12309b94acceSBarry Smith   }
12319b94acceSBarry Smith   return 0;
12329b94acceSBarry Smith }
12339b94acceSBarry Smith 
12349b94acceSBarry Smith /*@
12359b94acceSBarry Smith    SNESSolve - Solves a nonlinear system.  Call SNESSolve after calling
123663a78c88SLois Curfman McInnes    SNESCreate() and optional routines of the form SNESSetXXX().
12379b94acceSBarry Smith 
12389b94acceSBarry Smith    Input Parameter:
12399b94acceSBarry Smith .  snes - the SNES context
12408ddd3da0SLois Curfman McInnes .  x - the solution vector
12419b94acceSBarry Smith 
12429b94acceSBarry Smith    Output Parameter:
12439b94acceSBarry Smith    its - number of iterations until termination
12449b94acceSBarry Smith 
12458ddd3da0SLois Curfman McInnes    Note:
12468ddd3da0SLois Curfman McInnes    The user should initialize the vector, x, with the initial guess
12478ddd3da0SLois Curfman McInnes    for the nonlinear solve prior to calling SNESSolve.  In particular,
12488ddd3da0SLois Curfman McInnes    to employ an initial guess of zero, the user should explicitly set
12498ddd3da0SLois Curfman McInnes    this vector to zero by calling VecSet().
12508ddd3da0SLois Curfman McInnes 
12519b94acceSBarry Smith .keywords: SNES, nonlinear, solve
12529b94acceSBarry Smith 
125363a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESDestroy()
12549b94acceSBarry Smith @*/
12558ddd3da0SLois Curfman McInnes int SNESSolve(SNES snes,Vec x,int *its)
12569b94acceSBarry Smith {
12573c7409f5SSatish Balay   int ierr, flg;
1258052efed2SBarry Smith 
125977c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
1260c4fc05e7SBarry Smith   if (!snes->setup_called) {ierr = SNESSetUp(snes,x); CHKERRQ(ierr);}
1261c4fc05e7SBarry Smith   else {snes->vec_sol = snes->vec_sol_always = x;}
12629b94acceSBarry Smith   PLogEventBegin(SNES_Solve,snes,0,0,0);
12639b94acceSBarry Smith   ierr = (*(snes)->solve)(snes,its); CHKERRQ(ierr);
12649b94acceSBarry Smith   PLogEventEnd(SNES_Solve,snes,0,0,0);
12653c7409f5SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-snes_view", &flg); CHKERRQ(ierr);
12663c7409f5SSatish Balay   if (flg) { ierr = SNESView(snes,STDOUT_VIEWER_WORLD); CHKERRQ(ierr); }
12679b94acceSBarry Smith   return 0;
12689b94acceSBarry Smith }
12699b94acceSBarry Smith 
12709b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */
127137fcc0dbSBarry Smith static NRList *__SNESList = 0;
12729b94acceSBarry Smith 
12739b94acceSBarry Smith /*@
12744b0e389bSBarry Smith    SNESSetType - Sets the method for the nonlinear solver.
12759b94acceSBarry Smith 
12769b94acceSBarry Smith    Input Parameters:
12779b94acceSBarry Smith .  snes - the SNES context
12789b94acceSBarry Smith .  method - a known method
12799b94acceSBarry Smith 
12809b94acceSBarry Smith    Notes:
12819b94acceSBarry Smith    See "petsc/include/snes.h" for available methods (for instance)
12829b94acceSBarry Smith $  Systems of nonlinear equations:
128340191667SLois Curfman McInnes $    SNES_EQ_LS - Newton's method with line search
128440191667SLois Curfman McInnes $    SNES_EQ_TR - Newton's method with trust region
12859b94acceSBarry Smith $  Unconstrained minimization:
128640191667SLois Curfman McInnes $    SNES_UM_TR - Newton's method with trust region
128740191667SLois Curfman McInnes $    SNES_UM_LS - Newton's method with line search
12889b94acceSBarry Smith 
12899b94acceSBarry Smith   Options Database Command:
12904b0e389bSBarry Smith $ -snes_type  <method>
12919b94acceSBarry Smith $    Use -help for a list of available methods
12929b94acceSBarry Smith $    (for instance, ls or tr)
1293a703fe33SLois Curfman McInnes 
1294a703fe33SLois Curfman McInnes .keysords: SNES, set, method
12959b94acceSBarry Smith @*/
12964b0e389bSBarry Smith int SNESSetType(SNES snes,SNESType method)
12979b94acceSBarry Smith {
12989b94acceSBarry Smith   int (*r)(SNES);
129977c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
1300*7d1a2b2bSBarry Smith   if (snes->type == (int) method) return 0;
1301*7d1a2b2bSBarry Smith 
13029b94acceSBarry Smith   /* Get the function pointers for the iterative method requested */
130337fcc0dbSBarry Smith   if (!__SNESList) {SNESRegisterAll();}
130437fcc0dbSBarry Smith   if (!__SNESList) {SETERRQ(1,"SNESSetType:Could not get methods");}
130537fcc0dbSBarry Smith   r =  (int (*)(SNES))NRFindRoutine( __SNESList, (int)method, (char *)0 );
13064b0e389bSBarry Smith   if (!r) {SETERRQ(1,"SNESSetType:Unknown method");}
13070452661fSBarry Smith   if (snes->data) PetscFree(snes->data);
13089b94acceSBarry Smith   snes->set_method_called = 1;
13099b94acceSBarry Smith   return (*r)(snes);
13109b94acceSBarry Smith }
13119b94acceSBarry Smith 
13129b94acceSBarry Smith /* --------------------------------------------------------------------- */
13139b94acceSBarry Smith /*@C
13149b94acceSBarry Smith    SNESRegister - Adds the method to the nonlinear solver package, given
13154b0e389bSBarry Smith    a function pointer and a nonlinear solver name of the type SNESType.
13169b94acceSBarry Smith 
13179b94acceSBarry Smith    Input Parameters:
131840191667SLois Curfman McInnes .  name - for instance SNES_EQ_LS, SNES_EQ_TR, ...
131940191667SLois Curfman McInnes .  sname - corresponding string for name
13209b94acceSBarry Smith .  create - routine to create method context
13219b94acceSBarry Smith 
13229b94acceSBarry Smith .keywords: SNES, nonlinear, register
13239b94acceSBarry Smith 
13249b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy()
13259b94acceSBarry Smith @*/
13269b94acceSBarry Smith int SNESRegister(int name, char *sname, int (*create)(SNES))
13279b94acceSBarry Smith {
13289b94acceSBarry Smith   int ierr;
132937fcc0dbSBarry Smith   if (!__SNESList) {ierr = NRCreate(&__SNESList); CHKERRQ(ierr);}
133037fcc0dbSBarry Smith   NRRegister( __SNESList, name, sname, (int (*)(void*))create );
13319b94acceSBarry Smith   return 0;
13329b94acceSBarry Smith }
13339b94acceSBarry Smith /* --------------------------------------------------------------------- */
13349b94acceSBarry Smith /*@C
13359b94acceSBarry Smith    SNESRegisterDestroy - Frees the list of nonlinear solvers that were
13369b94acceSBarry Smith    registered by SNESRegister().
13379b94acceSBarry Smith 
13389b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy
13399b94acceSBarry Smith 
13409b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll()
13419b94acceSBarry Smith @*/
13429b94acceSBarry Smith int SNESRegisterDestroy()
13439b94acceSBarry Smith {
134437fcc0dbSBarry Smith   if (__SNESList) {
134537fcc0dbSBarry Smith     NRDestroy( __SNESList );
134637fcc0dbSBarry Smith     __SNESList = 0;
13479b94acceSBarry Smith   }
13489b94acceSBarry Smith   return 0;
13499b94acceSBarry Smith }
13509b94acceSBarry Smith 
13519b94acceSBarry Smith /*
13524b0e389bSBarry Smith    SNESGetTypeFromOptions_Private - Sets the selected method from the
13539b94acceSBarry Smith    options database.
13549b94acceSBarry Smith 
13559b94acceSBarry Smith    Input Parameter:
13569b94acceSBarry Smith .  ctx - the SNES context
13579b94acceSBarry Smith 
13589b94acceSBarry Smith    Output Parameter:
13599b94acceSBarry Smith .  method -  solver method
13609b94acceSBarry Smith 
13619b94acceSBarry Smith    Returns:
13629b94acceSBarry Smith    Returns 1 if the method is found; 0 otherwise.
13639b94acceSBarry Smith 
13649b94acceSBarry Smith    Options Database Key:
13654b0e389bSBarry Smith $  -snes_type  method
13669b94acceSBarry Smith */
1367052efed2SBarry Smith int SNESGetTypeFromOptions_Private(SNES ctx,SNESType *method,int *flg)
13689b94acceSBarry Smith {
1369052efed2SBarry Smith   int ierr;
13709b94acceSBarry Smith   char sbuf[50];
1371052efed2SBarry Smith   ierr = OptionsGetString(ctx->prefix,"-snes_type", sbuf, 50, flg); CHKERRQ(ierr);
1372052efed2SBarry Smith   if (*flg) {
1373052efed2SBarry Smith     if (!__SNESList) {ierr = SNESRegisterAll(); CHKERRQ(ierr);}
137437fcc0dbSBarry Smith     *method = (SNESType)NRFindID( __SNESList, sbuf );
13759b94acceSBarry Smith   }
13769b94acceSBarry Smith   return 0;
13779b94acceSBarry Smith }
13789b94acceSBarry Smith 
13799b94acceSBarry Smith /*@C
13809a28b0a6SLois Curfman McInnes    SNESGetType - Gets the SNES method type and name (as a string).
13819b94acceSBarry Smith 
13829b94acceSBarry Smith    Input Parameter:
13834b0e389bSBarry Smith .  snes - nonlinear solver context
13849b94acceSBarry Smith 
13859b94acceSBarry Smith    Output Parameter:
13869a28b0a6SLois Curfman McInnes .  method - SNES method (or use PETSC_NULL)
13879a28b0a6SLois Curfman McInnes .  name - name of SNES method (or use PETSC_NULL)
13889b94acceSBarry Smith 
13899b94acceSBarry Smith .keywords: SNES, nonlinear, get, method, name
13909b94acceSBarry Smith @*/
13914b0e389bSBarry Smith int SNESGetType(SNES snes, SNESType *method,char **name)
13929b94acceSBarry Smith {
139306a9b0adSLois Curfman McInnes   int ierr;
139437fcc0dbSBarry Smith   if (!__SNESList) {ierr = SNESRegisterAll(); CHKERRQ(ierr);}
13954b0e389bSBarry Smith   if (method) *method = (SNESType) snes->type;
139637fcc0dbSBarry Smith   if (name)  *name = NRFindName( __SNESList, (int) snes->type );
13979b94acceSBarry Smith   return 0;
13989b94acceSBarry Smith }
13999b94acceSBarry Smith 
14009b94acceSBarry Smith #include <stdio.h>
14019b94acceSBarry Smith /*
14024b0e389bSBarry Smith    SNESPrintTypes_Private - Prints the SNES methods available from the
14039b94acceSBarry Smith    options database.
14049b94acceSBarry Smith 
14059b94acceSBarry Smith    Input Parameters:
140633455573SSatish Balay .  comm   - communicator (usually MPI_COMM_WORLD)
14079b94acceSBarry Smith .  prefix - prefix (usually "-")
14084b0e389bSBarry Smith .  name   - the options database name (by default "snes_type")
14099b94acceSBarry Smith */
141033455573SSatish Balay int SNESPrintTypes_Private(MPI_Comm comm,char* prefix,char *name)
14119b94acceSBarry Smith {
14129b94acceSBarry Smith   FuncList *entry;
141337fcc0dbSBarry Smith   if (!__SNESList) {SNESRegisterAll();}
141437fcc0dbSBarry Smith   entry = __SNESList->head;
141577c4ece6SBarry Smith   PetscPrintf(comm," %s%s (one of)",prefix,name);
14169b94acceSBarry Smith   while (entry) {
141777c4ece6SBarry Smith     PetscPrintf(comm," %s",entry->name);
14189b94acceSBarry Smith     entry = entry->next;
14199b94acceSBarry Smith   }
142077c4ece6SBarry Smith   PetscPrintf(comm,"\n");
14219b94acceSBarry Smith   return 0;
14229b94acceSBarry Smith }
14239b94acceSBarry Smith 
14249b94acceSBarry Smith /*@C
14259b94acceSBarry Smith    SNESGetSolution - Returns the vector where the approximate solution is
14269b94acceSBarry Smith    stored.
14279b94acceSBarry Smith 
14289b94acceSBarry Smith    Input Parameter:
14299b94acceSBarry Smith .  snes - the SNES context
14309b94acceSBarry Smith 
14319b94acceSBarry Smith    Output Parameter:
14329b94acceSBarry Smith .  x - the solution
14339b94acceSBarry Smith 
14349b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution
14359b94acceSBarry Smith 
1436a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESGetGradient(), SNESGetSolutionUpdate()
14379b94acceSBarry Smith @*/
14389b94acceSBarry Smith int SNESGetSolution(SNES snes,Vec *x)
14399b94acceSBarry Smith {
144077c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
14419b94acceSBarry Smith   *x = snes->vec_sol_always;
14429b94acceSBarry Smith   return 0;
14439b94acceSBarry Smith }
14449b94acceSBarry Smith 
14459b94acceSBarry Smith /*@C
14469b94acceSBarry Smith    SNESGetSolutionUpdate - Returns the vector where the solution update is
14479b94acceSBarry Smith    stored.
14489b94acceSBarry Smith 
14499b94acceSBarry Smith    Input Parameter:
14509b94acceSBarry Smith .  snes - the SNES context
14519b94acceSBarry Smith 
14529b94acceSBarry Smith    Output Parameter:
14539b94acceSBarry Smith .  x - the solution update
14549b94acceSBarry Smith 
14559b94acceSBarry Smith    Notes:
14569b94acceSBarry Smith    This vector is implementation dependent.
14579b94acceSBarry Smith 
14589b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update
14599b94acceSBarry Smith 
14609b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction
14619b94acceSBarry Smith @*/
14629b94acceSBarry Smith int SNESGetSolutionUpdate(SNES snes,Vec *x)
14639b94acceSBarry Smith {
146477c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
14659b94acceSBarry Smith   *x = snes->vec_sol_update_always;
14669b94acceSBarry Smith   return 0;
14679b94acceSBarry Smith }
14689b94acceSBarry Smith 
14699b94acceSBarry Smith /*@C
14703638b69dSLois Curfman McInnes    SNESGetFunction - Returns the vector where the function is stored.
14719b94acceSBarry Smith 
14729b94acceSBarry Smith    Input Parameter:
14739b94acceSBarry Smith .  snes - the SNES context
14749b94acceSBarry Smith 
14759b94acceSBarry Smith    Output Parameter:
14763638b69dSLois Curfman McInnes .  r - the function
14779b94acceSBarry Smith 
14789b94acceSBarry Smith    Notes:
14799b94acceSBarry Smith    SNESGetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only
14809b94acceSBarry Smith    Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are
14819b94acceSBarry Smith    SNESGetMinimizationFunction() and SNESGetGradient();
14829b94acceSBarry Smith 
1483a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function
14849b94acceSBarry Smith 
148531615d04SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution(), SNESGetMinimizationFunction(),
148631615d04SLois Curfman McInnes           SNESGetGradient()
14879b94acceSBarry Smith @*/
14889b94acceSBarry Smith int SNESGetFunction(SNES snes,Vec *r)
14899b94acceSBarry Smith {
149077c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
14919b94acceSBarry Smith   if (snes->method_class != SNES_NONLINEAR_EQUATIONS) SETERRQ(1,
149248d91487SBarry Smith     "SNESGetFunction:For SNES_NONLINEAR_EQUATIONS only");
14939b94acceSBarry Smith   *r = snes->vec_func_always;
14949b94acceSBarry Smith   return 0;
14959b94acceSBarry Smith }
14969b94acceSBarry Smith 
14979b94acceSBarry Smith /*@C
14983638b69dSLois Curfman McInnes    SNESGetGradient - Returns the vector where the gradient is stored.
14999b94acceSBarry Smith 
15009b94acceSBarry Smith    Input Parameter:
15019b94acceSBarry Smith .  snes - the SNES context
15029b94acceSBarry Smith 
15039b94acceSBarry Smith    Output Parameter:
15049b94acceSBarry Smith .  r - the gradient
15059b94acceSBarry Smith 
15069b94acceSBarry Smith    Notes:
15079b94acceSBarry Smith    SNESGetGradient() is valid for SNES_UNCONSTRAINED_MINIMIZATION methods
15089b94acceSBarry Smith    only.  An analogous routine for SNES_NONLINEAR_EQUATIONS methods is
15099b94acceSBarry Smith    SNESGetFunction().
15109b94acceSBarry Smith 
15119b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient
15129b94acceSBarry Smith 
151331615d04SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESGetSolution(), SNESGetFunction()
15149b94acceSBarry Smith @*/
15159b94acceSBarry Smith int SNESGetGradient(SNES snes,Vec *r)
15169b94acceSBarry Smith {
151777c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
15189b94acceSBarry Smith   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
151948d91487SBarry Smith     "SNESGetGradient:For SNES_UNCONSTRAINED_MINIMIZATION only");
15209b94acceSBarry Smith   *r = snes->vec_func_always;
15219b94acceSBarry Smith   return 0;
15229b94acceSBarry Smith }
15239b94acceSBarry Smith 
15249b94acceSBarry Smith /*@
15259b94acceSBarry Smith    SNESGetMinimizationFunction - Returns the scalar function value for
15269b94acceSBarry Smith    unconstrained minimization problems.
15279b94acceSBarry Smith 
15289b94acceSBarry Smith    Input Parameter:
15299b94acceSBarry Smith .  snes - the SNES context
15309b94acceSBarry Smith 
15319b94acceSBarry Smith    Output Parameter:
15329b94acceSBarry Smith .  r - the function
15339b94acceSBarry Smith 
15349b94acceSBarry Smith    Notes:
15359b94acceSBarry Smith    SNESGetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION
15369b94acceSBarry Smith    methods only.  An analogous routine for SNES_NONLINEAR_EQUATIONS methods is
15379b94acceSBarry Smith    SNESGetFunction().
15389b94acceSBarry Smith 
15399b94acceSBarry Smith .keywords: SNES, nonlinear, get, function
15409b94acceSBarry Smith 
154131615d04SLois Curfman McInnes .seealso: SNESGetGradient(), SNESGetSolution(), SNESGetFunction()
15429b94acceSBarry Smith @*/
15439b94acceSBarry Smith int SNESGetMinimizationFunction(SNES snes,double *r)
15449b94acceSBarry Smith {
154577c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
15469b94acceSBarry Smith   if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) SETERRQ(1,
154748d91487SBarry Smith     "SNESGetMinimizationFunction:For SNES_UNCONSTRAINED_MINIMIZATION only");
15489b94acceSBarry Smith   *r = snes->fc;
15499b94acceSBarry Smith   return 0;
15509b94acceSBarry Smith }
15519b94acceSBarry Smith 
15523c7409f5SSatish Balay /*@C
15533c7409f5SSatish Balay    SNESSetOptionsPrefix - Sets the prefix used for searching for all
15543c7409f5SSatish Balay    SNES options in the database.
15553c7409f5SSatish Balay 
15563c7409f5SSatish Balay    Input Parameter:
15573c7409f5SSatish Balay .  snes - the SNES context
15583c7409f5SSatish Balay .  prefix - the prefix to prepend to all option names
15593c7409f5SSatish Balay 
15603c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database
1561a86d99e1SLois Curfman McInnes 
1562a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions()
15633c7409f5SSatish Balay @*/
15643c7409f5SSatish Balay int SNESSetOptionsPrefix(SNES snes,char *prefix)
15653c7409f5SSatish Balay {
15663c7409f5SSatish Balay   int ierr;
15673c7409f5SSatish Balay 
156877c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
15693c7409f5SSatish Balay   ierr = PetscObjectSetPrefix((PetscObject)snes, prefix); CHKERRQ(ierr);
15703c7409f5SSatish Balay   ierr = SLESSetOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr);
15713c7409f5SSatish Balay   return 0;
15723c7409f5SSatish Balay }
15733c7409f5SSatish Balay 
15743c7409f5SSatish Balay /*@C
15753c7409f5SSatish Balay    SNESAppendOptionsPrefix - Append to the prefix used for searching for all
15763c7409f5SSatish Balay    SNES options in the database.
15773c7409f5SSatish Balay 
15783c7409f5SSatish Balay    Input Parameter:
15793c7409f5SSatish Balay .  snes - the SNES context
15803c7409f5SSatish Balay .  prefix - the prefix to prepend to all option names
15813c7409f5SSatish Balay 
15823c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database
1583a86d99e1SLois Curfman McInnes 
1584a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix()
15853c7409f5SSatish Balay @*/
15863c7409f5SSatish Balay int SNESAppendOptionsPrefix(SNES snes,char *prefix)
15873c7409f5SSatish Balay {
15883c7409f5SSatish Balay   int ierr;
15893c7409f5SSatish Balay 
159077c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
15913c7409f5SSatish Balay   ierr = PetscObjectAppendPrefix((PetscObject)snes, prefix); CHKERRQ(ierr);
15923c7409f5SSatish Balay   ierr = SLESAppendOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr);
15933c7409f5SSatish Balay   return 0;
15943c7409f5SSatish Balay }
15953c7409f5SSatish Balay 
1596c83590e2SSatish Balay /*@
15973c7409f5SSatish Balay    SNESGetOptionsPrefix - Sets the prefix used for searching for all
15983c7409f5SSatish Balay    SNES options in the database.
15993c7409f5SSatish Balay 
16003c7409f5SSatish Balay    Input Parameter:
16013c7409f5SSatish Balay .  snes - the SNES context
16023c7409f5SSatish Balay 
16033c7409f5SSatish Balay    Output Parameter:
16043c7409f5SSatish Balay .  prefix - pointer to the prefix string used
16053c7409f5SSatish Balay 
16063c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database
1607a86d99e1SLois Curfman McInnes 
1608a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix()
16093c7409f5SSatish Balay @*/
16103c7409f5SSatish Balay int SNESGetOptionsPrefix(SNES snes,char **prefix)
16113c7409f5SSatish Balay {
16123c7409f5SSatish Balay   int ierr;
16133c7409f5SSatish Balay 
161477c4ece6SBarry Smith   PetscValidHeaderSpecific(snes,SNES_COOKIE);
16153c7409f5SSatish Balay   ierr = PetscObjectGetPrefix((PetscObject)snes, prefix); CHKERRQ(ierr);
16163c7409f5SSatish Balay   return 0;
16173c7409f5SSatish Balay }
16183c7409f5SSatish Balay 
16193c7409f5SSatish Balay 
16209b94acceSBarry Smith 
16219b94acceSBarry Smith 
16229b94acceSBarry Smith 
1623