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