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