xref: /petsc/src/snes/mf/snesmfj.c (revision 606d414c6e034e02e67059b83ebaefc3ebe99698)
1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER
2*606d414cSSatish Balay static char vcid[] = "$Id: snesmfj.c,v 1.88 1999/06/30 22:51:53 bsmith Exp balay $";
381e6777dSBarry Smith #endif
481e6777dSBarry Smith 
59a6cb015SBarry Smith #include "src/snes/snesimpl.h"
69a6cb015SBarry Smith #include "src/snes/mf/snesmfj.h"   /*I  "snes.h"   I*/
781e6777dSBarry Smith 
85a655dc6SBarry Smith FList MatSNESMFList              = 0;
95a655dc6SBarry Smith int   MatSNESMFRegisterAllCalled = 0;
10a4d4d686SBarry Smith 
115615d1e5SSatish Balay #undef __FUNC__
125a655dc6SBarry Smith #define __FUNC__ "MatSNESMFSetType"
13fd4bdd07SBarry Smith /*@C
1465f2ba5bSLois Curfman McInnes     MatSNESMFSetType - Sets the method that is used to compute the
1565f2ba5bSLois Curfman McInnes     differencing parameter for finite difference matrix-free formulations.
169a6cb015SBarry Smith 
179a6cb015SBarry Smith     Input Parameters:
18ef4ad1fdSLois Curfman McInnes +   mat - the "matrix-free" matrix created via MatCreateSNESMF()
199a6cb015SBarry Smith -   ftype - the type requested
209a6cb015SBarry Smith 
2115091d37SBarry Smith     Level: advanced
2215091d37SBarry Smith 
2365f2ba5bSLois Curfman McInnes     Notes:
2465f2ba5bSLois Curfman McInnes     For example, such routines can compute h for use in
2565f2ba5bSLois Curfman McInnes     Jacobian-vector products of the form
2665f2ba5bSLois Curfman McInnes 
2765f2ba5bSLois Curfman McInnes                         F(x+ha) - F(x)
28ef4ad1fdSLois Curfman McInnes           F'(u)a  ~=  ----------------
2965f2ba5bSLois Curfman McInnes                               h
3065f2ba5bSLois Curfman McInnes 
315a655dc6SBarry Smith .seealso: MatCreateSNESMF(), MatSNESMFRegister()
329a6cb015SBarry Smith @*/
335a655dc6SBarry Smith int MatSNESMFSetType(Mat mat,char *ftype)
34b9fa9cd0SBarry Smith {
355a655dc6SBarry Smith   int          ierr, (*r)(MatSNESMFCtx);
365a655dc6SBarry Smith   MatSNESMFCtx ctx;
37a4d4d686SBarry Smith 
38a4d4d686SBarry Smith   PetscFunctionBegin;
39a4d4d686SBarry Smith   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
40a4d4d686SBarry Smith 
419a6cb015SBarry Smith   /* already set, so just return */
423f1db9ecSBarry Smith   if (PetscTypeCompare(ctx->type_name,ftype)) PetscFunctionReturn(0);
43a4d4d686SBarry Smith 
449a6cb015SBarry Smith   /* destroy the old one if it exists */
459a6cb015SBarry Smith   if (ctx->ops->destroy) {
469a6cb015SBarry Smith     ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr);
479a6cb015SBarry Smith   }
489a6cb015SBarry Smith 
4965f2ba5bSLois Curfman McInnes   /* Get the function pointers for the requrested method */
505a655dc6SBarry Smith   if (!MatSNESMFRegisterAllCalled) {ierr = MatSNESMFRegisterAll(PETSC_NULL);CHKERRQ(ierr);}
519a6cb015SBarry Smith 
525a655dc6SBarry Smith   ierr =  FListFind(ctx->comm, MatSNESMFList, ftype,(int (**)(void *)) &r );CHKERRQ(ierr);
539a6cb015SBarry Smith 
545a655dc6SBarry Smith   if (!r) SETERRQ(1,1,"Unknown MatSNESMF type given");
559a6cb015SBarry Smith 
569a6cb015SBarry Smith   ierr = (*r)(ctx);CHKERRQ(ierr);
579a6cb015SBarry Smith   ierr = PetscStrncpy(ctx->type_name,ftype,256);CHKERRQ(ierr);
589a6cb015SBarry Smith 
599a6cb015SBarry Smith   PetscFunctionReturn(0);
609a6cb015SBarry Smith }
619a6cb015SBarry Smith 
629a6cb015SBarry Smith /*MC
6365f2ba5bSLois Curfman McInnes    MatSNESMFRegister - Adds a method to the MatSNESMF registry.
649a6cb015SBarry Smith 
659a6cb015SBarry Smith    Synopsis:
665a655dc6SBarry Smith    MatSNESMFRegister(char *name_solver,char *path,char *name_create,int (*routine_create)(MatSNESMF))
679a6cb015SBarry Smith 
689a6cb015SBarry Smith    Not Collective
699a6cb015SBarry Smith 
709a6cb015SBarry Smith    Input Parameters:
719a6cb015SBarry Smith +  name_solver - name of a new user-defined compute-h module
729a6cb015SBarry Smith .  path - path (either absolute or relative) the library containing this solver
739a6cb015SBarry Smith .  name_create - name of routine to create method context
749a6cb015SBarry Smith -  routine_create - routine to create method context
759a6cb015SBarry Smith 
7615091d37SBarry Smith    Level: developer
7715091d37SBarry Smith 
789a6cb015SBarry Smith    Notes:
795a655dc6SBarry Smith    MatSNESMFRegister() may be called multiple times to add several user-defined solvers.
809a6cb015SBarry Smith 
819a6cb015SBarry Smith    If dynamic libraries are used, then the fourth input argument (routine_create)
829a6cb015SBarry Smith    is ignored.
839a6cb015SBarry Smith 
849a6cb015SBarry Smith    Sample usage:
859a6cb015SBarry Smith .vb
865a655dc6SBarry Smith    MatSNESMFRegister("my_h",/home/username/my_lib/lib/libO/solaris/mylib.a,
879a6cb015SBarry Smith                "MyHCreate",MyHCreate);
889a6cb015SBarry Smith .ve
899a6cb015SBarry Smith 
909a6cb015SBarry Smith    Then, your solver can be chosen with the procedural interface via
915a655dc6SBarry Smith $     MatSNESMFSetType(mfctx,"my_h")
929a6cb015SBarry Smith    or at runtime via the option
939a6cb015SBarry Smith $     -snes_mf_type my_h
949a6cb015SBarry Smith 
955a655dc6SBarry Smith .keywords: MatSNESMF, register
969a6cb015SBarry Smith 
975a655dc6SBarry Smith .seealso: MatSNESMFRegisterAll(), MatSNESMFRegisterDestroy()
989a6cb015SBarry Smith M*/
999a6cb015SBarry Smith 
1009a6cb015SBarry Smith #undef __FUNC__
1015a655dc6SBarry Smith #define __FUNC__ "MatSNESMFRegister_Private"
1025a655dc6SBarry Smith int MatSNESMFRegister_Private(char *sname,char *path,char *name,int (*function)(MatSNESMFCtx))
1039a6cb015SBarry Smith {
1049a6cb015SBarry Smith   int ierr;
1059a6cb015SBarry Smith   char fullname[256];
1069a6cb015SBarry Smith 
1079a6cb015SBarry Smith   PetscFunctionBegin;
108549d3d68SSatish Balay   ierr = PetscStrcpy(fullname,path);CHKERRQ(ierr);
109549d3d68SSatish Balay   PetscStrcat(fullname,":");PetscStrcat(fullname,name);
1105a655dc6SBarry Smith   ierr = FListAdd_Private(&MatSNESMFList,sname,fullname,(int (*)(void*))function);CHKERRQ(ierr);
1119a6cb015SBarry Smith   PetscFunctionReturn(0);
1129a6cb015SBarry Smith }
1139a6cb015SBarry Smith 
1149a6cb015SBarry Smith 
1159a6cb015SBarry Smith #undef __FUNC__
1165a655dc6SBarry Smith #define __FUNC__ "MatSNESMFRegisterDestroy"
1179a6cb015SBarry Smith /*@C
1185a655dc6SBarry Smith    MatSNESMFRegisterDestroy - Frees the list of MatSNESMF methods that were
1195a655dc6SBarry Smith    registered by MatSNESMFRegister().
1209a6cb015SBarry Smith 
1219a6cb015SBarry Smith    Not Collective
1229a6cb015SBarry Smith 
12315091d37SBarry Smith    Level: developer
12415091d37SBarry Smith 
1255a655dc6SBarry Smith .keywords: MatSNESMF, register, destroy
1269a6cb015SBarry Smith 
1275a655dc6SBarry Smith .seealso: MatSNESMFRegister(), MatSNESMFRegisterAll()
1289a6cb015SBarry Smith @*/
1295a655dc6SBarry Smith int MatSNESMFRegisterDestroy(void)
1309a6cb015SBarry Smith {
1319a6cb015SBarry Smith   int ierr;
1329a6cb015SBarry Smith 
1339a6cb015SBarry Smith   PetscFunctionBegin;
1345a655dc6SBarry Smith   if (MatSNESMFList) {
1355a655dc6SBarry Smith     ierr = FListDestroy( MatSNESMFList );CHKERRQ(ierr);
1365a655dc6SBarry Smith     MatSNESMFList = 0;
1379a6cb015SBarry Smith   }
1385a655dc6SBarry Smith   MatSNESMFRegisterAllCalled = 0;
1399a6cb015SBarry Smith   PetscFunctionReturn(0);
1409a6cb015SBarry Smith }
1419a6cb015SBarry Smith 
1429a6cb015SBarry Smith /* ----------------------------------------------------------------------------------------*/
143a4d4d686SBarry Smith #undef __FUNC__
1445a655dc6SBarry Smith #define __FUNC__ "MatSNESMFDestroy_Private"
1455a655dc6SBarry Smith int MatSNESMFDestroy_Private(Mat mat)
146a4d4d686SBarry Smith {
147a4d4d686SBarry Smith   int          ierr;
1485a655dc6SBarry Smith   MatSNESMFCtx ctx;
149fae171e0SBarry Smith 
1503a40ed3dSBarry Smith   PetscFunctionBegin;
1510a25c783SBarry Smith   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
152b9fa9cd0SBarry Smith   ierr = VecDestroy(ctx->w);CHKERRQ(ierr);
1539a6cb015SBarry Smith   if (ctx->ops->destroy) {ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr);}
154b4fd4287SBarry Smith   if (ctx->sp) {ierr = PCNullSpaceDestroy(ctx->sp);CHKERRQ(ierr);}
155*606d414cSSatish Balay   ierr = PetscFree(ctx->ops);CHKERRQ(ierr);
156*606d414cSSatish Balay   ierr = PetscFree(ctx);CHKERRQ(ierr);
1573a40ed3dSBarry Smith   PetscFunctionReturn(0);
158b9fa9cd0SBarry Smith }
15950361f65SLois Curfman McInnes 
1605615d1e5SSatish Balay #undef __FUNC__
1615a655dc6SBarry Smith #define __FUNC__ "MatSNESMFView_Private"
16239e2f89bSBarry Smith /*
1635a655dc6SBarry Smith    MatSNESMFView_Private - Views matrix-free parameters.
1648f6e3e37SBarry Smith 
16539e2f89bSBarry Smith */
1665a655dc6SBarry Smith int MatSNESMFView_Private(Mat J,Viewer viewer)
167eb9086c3SLois Curfman McInnes {
168eb9086c3SLois Curfman McInnes   int          ierr;
1695a655dc6SBarry Smith   MatSNESMFCtx ctx;
170eb9086c3SLois Curfman McInnes   MPI_Comm     comm;
171eb9086c3SLois Curfman McInnes   FILE         *fd;
172eb9086c3SLois Curfman McInnes   ViewerType   vtype;
173eb9086c3SLois Curfman McInnes 
1743a40ed3dSBarry Smith   PetscFunctionBegin;
1752d0c0e3bSBarry Smith   ierr = PetscObjectGetComm((PetscObject)J,&comm);CHKERRQ(ierr);
176eb9086c3SLois Curfman McInnes   ierr = MatShellGetContext(J,(void **)&ctx);CHKERRQ(ierr);
177eb9086c3SLois Curfman McInnes   ierr = ViewerGetType(viewer,&vtype);CHKERRQ(ierr);
178eb9086c3SLois Curfman McInnes   ierr = ViewerASCIIGetPointer(viewer,&fd);CHKERRQ(ierr);
1793f1db9ecSBarry Smith   if (PetscTypeCompare(vtype,ASCII_VIEWER)) {
180d132466eSBarry Smith      ierr = PetscFPrintf(comm,fd,"  SNES matrix-free approximation:\n");CHKERRQ(ierr);
181d132466eSBarry Smith      ierr = PetscFPrintf(comm,fd,"    err=%g (relative error in function evaluation)\n",ctx->error_rel);CHKERRQ(ierr);
182d132466eSBarry Smith      ierr = PetscFPrintf(ctx->comm,fd,"    Using %s compute h routine\n",ctx->type_name);CHKERRQ(ierr);
1839a6cb015SBarry Smith      if (ctx->ops->view) {
1849a6cb015SBarry Smith        ierr = (*ctx->ops->view)(ctx,viewer);CHKERRQ(ierr);
1859a6cb015SBarry Smith      }
1865cd90555SBarry Smith   } else {
1875cd90555SBarry Smith     SETERRQ(1,1,"Viewer type not supported for this object");
188eb9086c3SLois Curfman McInnes   }
1893a40ed3dSBarry Smith   PetscFunctionReturn(0);
190eb9086c3SLois Curfman McInnes }
191eb9086c3SLois Curfman McInnes 
192be726c96SBarry Smith #undef __FUNC__
1935a655dc6SBarry Smith #define __FUNC__ "MatSNESMFAssemblyEnd_Private"
194be726c96SBarry Smith /*
1955a655dc6SBarry Smith    MatSNESMFAssemblyEnd_Private - Resets the ctx->ncurrenth to zero. This
19665f2ba5bSLois Curfman McInnes    allows the user to indicate the beginning of a new linear solve by calling
197be726c96SBarry Smith    MatAssemblyXXX() on the matrix free matrix. This then allows the
19865f2ba5bSLois Curfman McInnes    MatSNESMFCreate_WP() to properly compute ||U|| only the first time
19965f2ba5bSLois Curfman McInnes    in the linear solver rather than every time.
200be726c96SBarry Smith */
2015a655dc6SBarry Smith int MatSNESMFAssemblyEnd_Private(Mat J)
202be726c96SBarry Smith {
203be726c96SBarry Smith   int            ierr;
204be726c96SBarry Smith 
205be726c96SBarry Smith   PetscFunctionBegin;
2065a655dc6SBarry Smith   ierr = MatSNESMFResetHHistory(J);CHKERRQ(ierr);
207be726c96SBarry Smith   PetscFunctionReturn(0);
208be726c96SBarry Smith }
209be726c96SBarry Smith 
210c481317fSBarry Smith 
2115615d1e5SSatish Balay #undef __FUNC__
2125a655dc6SBarry Smith #define __FUNC__ "MatSNESMFMult_Private"
213eb9086c3SLois Curfman McInnes /*
2145a655dc6SBarry Smith   MatSNESMFMult_Private - Default matrix-free form for Jacobian-vector
215eb9086c3SLois Curfman McInnes   product, y = F'(u)*a:
216a4d4d686SBarry Smith 
2179a6cb015SBarry Smith         y ~= ( F(u + ha) - F(u) )/h,
218eb9086c3SLois Curfman McInnes   where F = nonlinear function, as set by SNESSetFunction()
219eb9086c3SLois Curfman McInnes         u = current iterate
220eb9086c3SLois Curfman McInnes         h = difference interval
221eb9086c3SLois Curfman McInnes */
2225a655dc6SBarry Smith int MatSNESMFMult_Private(Mat mat,Vec a,Vec y)
22339e2f89bSBarry Smith {
2245a655dc6SBarry Smith   MatSNESMFCtx ctx;
225fae171e0SBarry Smith   SNES           snes;
226a4d4d686SBarry Smith   Scalar         h, mone = -1.0;
227fae171e0SBarry Smith   Vec            w,U,F;
228a305c92eSSatish Balay   int            ierr, (*eval_fct)(SNES,Vec,Vec)=0;
22939e2f89bSBarry Smith 
2303a40ed3dSBarry Smith   PetscFunctionBegin;
2319a6cb015SBarry Smith   /* We log matrix-free matrix-vector products separately, so that we can
2329a6cb015SBarry Smith      separate the performance monitoring from the cases that use conventional
2339a6cb015SBarry Smith      storage.  We may eventually modify event logging to associate events
2349a6cb015SBarry Smith      with particular objects, hence alleviating the more general problem. */
23556cd22aeSBarry Smith   PLogEventBegin(MAT_MatrixFreeMult,a,y,0,0);
23656cd22aeSBarry Smith 
2376e38a044SBarry Smith   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
238fae171e0SBarry Smith   snes = ctx->snes;
239fae171e0SBarry Smith   w    = ctx->w;
24078b31e54SBarry Smith   ierr = SNESGetSolution(snes,&U);CHKERRQ(ierr);
24150361f65SLois Curfman McInnes 
24285614651SBarry Smith   /*
24385614651SBarry Smith       Compute differencing parameter
24485614651SBarry Smith   */
2459a6cb015SBarry Smith   if (!ctx->ops->compute) {
2465a655dc6SBarry Smith     ierr = MatSNESMFSetType(mat,"default");CHKERRQ(ierr);
2475a655dc6SBarry Smith     ierr = MatSNESMFSetFromOptions(mat);CHKERRQ(ierr);
2489a6cb015SBarry Smith   }
2499a6cb015SBarry Smith   ierr = (*ctx->ops->compute)(ctx,U,a,&h);CHKERRQ(ierr);
250a4d4d686SBarry Smith 
251a4d4d686SBarry Smith   /* keep a record of the current differencing parameter h */
252a4d4d686SBarry Smith   ctx->currenth = h;
253aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
254a4d4d686SBarry Smith   PLogInfo(mat,"Current differencing parameter: %g + %g i\n",PetscReal(h),PetscImaginary(h));
255a4d4d686SBarry Smith #else
256a4d4d686SBarry Smith   PLogInfo(mat,"Current differencing parameter: %g\n",h);
257a4d4d686SBarry Smith #endif
258a4d4d686SBarry Smith   if (ctx->historyh && ctx->ncurrenth < ctx->maxcurrenth) {
25985614651SBarry Smith     ctx->historyh[ctx->ncurrenth] = h;
260a4d4d686SBarry Smith   }
26185614651SBarry Smith   ctx->ncurrenth++;
262a4d4d686SBarry Smith 
26385614651SBarry Smith   /* w = u + ha */
264a4d4d686SBarry Smith   ierr = VecWAXPY(&h,a,U,w);CHKERRQ(ierr);
26585614651SBarry Smith 
26685614651SBarry Smith 
26785614651SBarry Smith   if (!ctx->func) {
26885614651SBarry Smith     if (snes->method_class == SNES_NONLINEAR_EQUATIONS) {
26985614651SBarry Smith       eval_fct = SNESComputeFunction;
27085614651SBarry Smith       ierr = SNESGetFunction(snes,&F);CHKERRQ(ierr);
27185614651SBarry Smith     } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) {
27285614651SBarry Smith       eval_fct = SNESComputeGradient;
27385614651SBarry Smith       ierr = SNESGetGradient(snes,&F);CHKERRQ(ierr);
27485614651SBarry Smith     } else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Invalid method class");
275a4d4d686SBarry Smith     ierr = eval_fct(snes,w,y);CHKERRQ(ierr);
27685614651SBarry Smith   } else {
27785614651SBarry Smith     F = ctx->funcvec;
27885614651SBarry Smith     /* compute func(U) as base for differencing */
27985614651SBarry Smith     if (ctx->ncurrenth == 1) {
28085614651SBarry Smith       ierr            = (*ctx->func)(snes,U,F,ctx->funcctx);CHKERRQ(ierr);
28185614651SBarry Smith     }
28285614651SBarry Smith     ierr = (*ctx->func)(snes,w,y,ctx->funcctx);CHKERRQ(ierr);
28385614651SBarry Smith   }
284a4d4d686SBarry Smith 
285a4d4d686SBarry Smith   ierr = VecAXPY(&mone,F,y);CHKERRQ(ierr);
286a4d4d686SBarry Smith   h    = 1.0/h;
287a4d4d686SBarry Smith   ierr = VecScale(&h,y);CHKERRQ(ierr);
288a4d4d686SBarry Smith   if (ctx->sp) {ierr = PCNullSpaceRemove(ctx->sp,y);CHKERRQ(ierr);}
289a4d4d686SBarry Smith 
290a4d4d686SBarry Smith   PLogEventEnd(MAT_MatrixFreeMult,a,y,0,0);
291a4d4d686SBarry Smith   PetscFunctionReturn(0);
292a4d4d686SBarry Smith }
293a4d4d686SBarry Smith 
294a4d4d686SBarry Smith #undef __FUNC__
2955a655dc6SBarry Smith #define __FUNC__ "MatCreateSNESMF"
296a4d4d686SBarry Smith /*@C
29765f2ba5bSLois Curfman McInnes    MatCreateSNESMF - Creates a matrix-free matrix context for use with
29865f2ba5bSLois Curfman McInnes    a SNES solver.  This matrix can be used as the Jacobian argument for
29965f2ba5bSLois Curfman McInnes    the routine SNESSetJacobian().
300a4d4d686SBarry Smith 
301a4d4d686SBarry Smith    Collective on SNES and Vec
302a4d4d686SBarry Smith 
303a4d4d686SBarry Smith    Input Parameters:
304a4d4d686SBarry Smith +  snes - the SNES context
305a4d4d686SBarry Smith -  x - vector where SNES solution is to be stored.
306a4d4d686SBarry Smith 
307a4d4d686SBarry Smith    Output Parameter:
308a4d4d686SBarry Smith .  J - the matrix-free matrix
309a4d4d686SBarry Smith 
31015091d37SBarry Smith    Level: advanced
31115091d37SBarry Smith 
312a4d4d686SBarry Smith    Notes:
313a4d4d686SBarry Smith    The matrix-free matrix context merely contains the function pointers
314a4d4d686SBarry Smith    and work space for performing finite difference approximations of
31565f2ba5bSLois Curfman McInnes    Jacobian-vector products, F'(u)*a,
3169a6cb015SBarry Smith 
3179a6cb015SBarry Smith    The default code uses the following approach to compute h
318a4d4d686SBarry Smith 
319a4d4d686SBarry Smith .vb
32065f2ba5bSLois Curfman McInnes      F'(u)*a = [F(u+h*a) - F(u)]/h where
321a4d4d686SBarry Smith      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
322a4d4d686SBarry Smith        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   otherwise
323a4d4d686SBarry Smith  where
324a4d4d686SBarry Smith      error_rel = square root of relative error in function evaluation
325a4d4d686SBarry Smith      umin = minimum iterate parameter
326a4d4d686SBarry Smith .ve
327a4d4d686SBarry Smith 
3285a655dc6SBarry Smith    The user can set the error_rel via MatSNESMFSetFunctionError() and
32965f2ba5bSLois Curfman McInnes    umin via MatSNESMFDefaultSetUmin(); see the nonlinear solvers chapter
33065f2ba5bSLois Curfman McInnes    of the users manual for details.
331a4d4d686SBarry Smith 
332a4d4d686SBarry Smith    The user should call MatDestroy() when finished with the matrix-free
333a4d4d686SBarry Smith    matrix context.
334a4d4d686SBarry Smith 
335a4d4d686SBarry Smith    Options Database Keys:
336a4d4d686SBarry Smith +  -snes_mf_err <error_rel> - Sets error_rel
3379a6cb015SBarry Smith .  -snes_mf_unim <umin> - Sets umin (for default PETSc routine that computes h only)
338a4d4d686SBarry Smith -  -snes_mf_ksp_monitor - KSP monitor routine that prints differencing h
339a4d4d686SBarry Smith 
340a4d4d686SBarry Smith .keywords: SNES, default, matrix-free, create, matrix
341a4d4d686SBarry Smith 
3425a655dc6SBarry Smith .seealso: MatDestroy(), MatSNESMFSetFunctionError(), MatSNESMFDefaultSetUmin()
3435a655dc6SBarry Smith           MatSNESMFSetHHistory(), MatSNESMFResetHHistory(),
3445a655dc6SBarry Smith           MatSNESMFGetH(),MatSNESMFKSPMonitor(), MatSNESMFRegister()
345a4d4d686SBarry Smith 
346a4d4d686SBarry Smith @*/
3475a655dc6SBarry Smith int MatCreateSNESMF(SNES snes,Vec x, Mat *J)
348a4d4d686SBarry Smith {
349a4d4d686SBarry Smith   MPI_Comm     comm;
3505a655dc6SBarry Smith   MatSNESMFCtx mfctx;
3519a6cb015SBarry Smith   int          n, nloc, ierr;
352a4d4d686SBarry Smith 
353a4d4d686SBarry Smith   PetscFunctionBegin;
3545a655dc6SBarry Smith   mfctx = (MatSNESMFCtx) PetscMalloc(sizeof(struct _p_MatSNESMFCtx));CHKPTRQ(mfctx);
3555a655dc6SBarry Smith   PLogObjectMemory(snes,sizeof(MatSNESMFCtx));
3569a6cb015SBarry Smith   mfctx->comm         = snes->comm;
357a4d4d686SBarry Smith   mfctx->sp           = 0;
358a4d4d686SBarry Smith   mfctx->snes         = snes;
359a4d4d686SBarry Smith   mfctx->error_rel    = 1.e-8; /* assumes double precision */
360a4d4d686SBarry Smith   mfctx->currenth     = 0.0;
361a4d4d686SBarry Smith   mfctx->historyh     = PETSC_NULL;
362a4d4d686SBarry Smith   mfctx->ncurrenth    = 0;
363a4d4d686SBarry Smith   mfctx->maxcurrenth  = 0;
364a4d4d686SBarry Smith 
3659a6cb015SBarry Smith   /*
3669a6cb015SBarry Smith      Create the empty data structure to contain compute-h routines.
3679a6cb015SBarry Smith      These will be filled in below from the command line options or
3685a655dc6SBarry Smith      a later call with MatSNESMFSetType() or if that is not called
3695a655dc6SBarry Smith      then it will default in the first use of MatSNESMFMult_private()
3709a6cb015SBarry Smith   */
3719a6cb015SBarry Smith   mfctx->ops                 = (MFOps *)PetscMalloc(sizeof(MFOps));CHKPTRQ(mfctx->ops);
3729a6cb015SBarry Smith   mfctx->ops->compute        = 0;
3739a6cb015SBarry Smith   mfctx->ops->destroy        = 0;
3749a6cb015SBarry Smith   mfctx->ops->view           = 0;
3759a6cb015SBarry Smith   mfctx->ops->printhelp      = 0;
3769a6cb015SBarry Smith   mfctx->ops->setfromoptions = 0;
3779a6cb015SBarry Smith   mfctx->hctx                = 0;
3789a6cb015SBarry Smith 
37985614651SBarry Smith   mfctx->func                = 0;
38085614651SBarry Smith   mfctx->funcctx             = 0;
38185614651SBarry Smith   mfctx->funcvec             = 0;
38285614651SBarry Smith 
383a4d4d686SBarry Smith   ierr = VecDuplicate(x,&mfctx->w);CHKERRQ(ierr);
384a4d4d686SBarry Smith   ierr = PetscObjectGetComm((PetscObject)x,&comm);CHKERRQ(ierr);
385a4d4d686SBarry Smith   ierr = VecGetSize(x,&n);CHKERRQ(ierr);
386a4d4d686SBarry Smith   ierr = VecGetLocalSize(x,&nloc);CHKERRQ(ierr);
387ffa0fd9eSBarry Smith   ierr = MatCreateShell(comm,nloc,nloc,n,n,mfctx,J);CHKERRQ(ierr);
3885a655dc6SBarry Smith   ierr = MatShellSetOperation(*J,MATOP_MULT,(void*)MatSNESMFMult_Private);CHKERRQ(ierr);
3895a655dc6SBarry Smith   ierr = MatShellSetOperation(*J,MATOP_DESTROY,(void *)MatSNESMFDestroy_Private);CHKERRQ(ierr);
3905a655dc6SBarry Smith   ierr = MatShellSetOperation(*J,MATOP_VIEW,(void *)MatSNESMFView_Private);CHKERRQ(ierr);
3915a655dc6SBarry Smith   ierr = MatShellSetOperation(*J,MATOP_ASSEMBLY_END,(void *)MatSNESMFAssemblyEnd_Private);CHKERRQ(ierr);
392a4d4d686SBarry Smith   PLogObjectParent(*J,mfctx->w);
393a4d4d686SBarry Smith   PLogObjectParent(snes,*J);
3949a6cb015SBarry Smith 
3959a6cb015SBarry Smith   mfctx->mat = *J;
3969a6cb015SBarry Smith 
3979a6cb015SBarry Smith 
3989a6cb015SBarry Smith   PetscFunctionReturn(0);
3999a6cb015SBarry Smith }
4009a6cb015SBarry Smith 
4019a6cb015SBarry Smith #undef __FUNC__
4025a655dc6SBarry Smith #define __FUNC__ "MatSNESMFSetFromOptions"
4039a6cb015SBarry Smith /*@
4045a655dc6SBarry Smith    MatSNESMFSetFromOptions - Sets the MatSNESMF options from the command line
4059a6cb015SBarry Smith    parameter.
4069a6cb015SBarry Smith 
4079a6cb015SBarry Smith    Collective on Mat
4089a6cb015SBarry Smith 
4099a6cb015SBarry Smith    Input Parameters:
4105a655dc6SBarry Smith .  mat - the matrix obtained with MatCreateSNESMF()
4115a655dc6SBarry Smith 
4125a655dc6SBarry Smith    Options Database Keys:
4135a655dc6SBarry Smith +  -snes_mf_type - <default,wp>
4145a655dc6SBarry Smith -  -snes_mf_err - square root of estimated relative error in function evaluation
4159a6cb015SBarry Smith 
41615091d37SBarry Smith    Level: advanced
41715091d37SBarry Smith 
4189a6cb015SBarry Smith .keywords: SNES, matrix-free, parameters
4199a6cb015SBarry Smith 
4205a655dc6SBarry Smith .seealso: MatCreateSNESMF(),MatSNESMFSetHHistory(),
4215a655dc6SBarry Smith           MatSNESMFResetHHistory(), MatSNESMFKSPMonitor()
4229a6cb015SBarry Smith @*/
4235a655dc6SBarry Smith int MatSNESMFSetFromOptions(Mat mat)
4249a6cb015SBarry Smith {
4255a655dc6SBarry Smith   MatSNESMFCtx mfctx;
4269a6cb015SBarry Smith   int          ierr,flg;
4279a6cb015SBarry Smith   char         ftype[256],p[64];
4289a6cb015SBarry Smith 
4299a6cb015SBarry Smith   PetscFunctionBegin;
4309a6cb015SBarry Smith   ierr = MatShellGetContext(mat,(void **)&mfctx);CHKERRQ(ierr);
4319a6cb015SBarry Smith   if (mfctx) {
4329a6cb015SBarry Smith     /* allow user to set the type */
4339a6cb015SBarry Smith     ierr = OptionsGetString(mfctx->snes->prefix,"-snes_mf_type",ftype,256,&flg);CHKERRQ(ierr);
4349a6cb015SBarry Smith     if (flg) {
4355a655dc6SBarry Smith       ierr = MatSNESMFSetType(mat,ftype);CHKERRQ(ierr);
4369a6cb015SBarry Smith     }
4379a6cb015SBarry Smith 
4389a6cb015SBarry Smith     ierr = OptionsGetDouble(mfctx->snes->prefix,"-snes_mf_err",&mfctx->error_rel,&flg);CHKERRQ(ierr);
4399a6cb015SBarry Smith     if (mfctx->ops->setfromoptions) {
4409a6cb015SBarry Smith       ierr = (*mfctx->ops->setfromoptions)(mfctx);CHKERRQ(ierr);
4419a6cb015SBarry Smith     }
4429a6cb015SBarry Smith 
4439a6cb015SBarry Smith     ierr = OptionsHasName(PETSC_NULL,"-help",&flg);CHKERRQ(ierr);
444d132466eSBarry Smith     ierr = PetscStrcpy(p,"-");CHKERRQ(ierr);
445d132466eSBarry Smith     if (mfctx->snes->prefix) {ierr = PetscStrcat(p,mfctx->snes->prefix);CHKERRQ(ierr);}
4469a6cb015SBarry Smith     if (flg) {
447d132466eSBarry Smith       ierr = (*PetscHelpPrintf)(mfctx->snes->comm,"   %ssnes_mf_err <err>: set sqrt rel error in function (default %g)\n",p,mfctx->error_rel);CHKERRQ(ierr);
4489a6cb015SBarry Smith       if (mfctx->ops->printhelp) {
449d132466eSBarry Smith         ierr = (*mfctx->ops->printhelp)(mfctx);CHKERRQ(ierr);
4509a6cb015SBarry Smith       }
4519a6cb015SBarry Smith     }
4529a6cb015SBarry Smith   }
453a4d4d686SBarry Smith   PetscFunctionReturn(0);
454a4d4d686SBarry Smith }
455a4d4d686SBarry Smith 
456a4d4d686SBarry Smith #undef __FUNC__
4575a655dc6SBarry Smith #define __FUNC__ "MatSNESMFGetH"
458a4d4d686SBarry Smith /*@
45965f2ba5bSLois Curfman McInnes    MatSNESMFGetH - Gets the last value that was used as the differencing
460a4d4d686SBarry Smith    parameter.
461a4d4d686SBarry Smith 
462a4d4d686SBarry Smith    Not Collective
463a4d4d686SBarry Smith 
464a4d4d686SBarry Smith    Input Parameters:
4655a655dc6SBarry Smith .  mat - the matrix obtained with MatCreateSNESMF()
466a4d4d686SBarry Smith 
467a4d4d686SBarry Smith    Output Paramter:
468a4d4d686SBarry Smith .  h - the differencing step size
469a4d4d686SBarry Smith 
47015091d37SBarry Smith    Level: advanced
47115091d37SBarry Smith 
472a4d4d686SBarry Smith .keywords: SNES, matrix-free, parameters
473a4d4d686SBarry Smith 
4745a655dc6SBarry Smith .seealso: MatCreateSNESMF(),MatSNESMFSetHHistory(),
4755a655dc6SBarry Smith           MatSNESMFResetHHistory(),MatSNESMFKSPMonitor()
476a4d4d686SBarry Smith @*/
4775a655dc6SBarry Smith int MatSNESMFGetH(Mat mat,Scalar *h)
478a4d4d686SBarry Smith {
4795a655dc6SBarry Smith   MatSNESMFCtx ctx;
480a4d4d686SBarry Smith   int          ierr;
481a4d4d686SBarry Smith 
482a4d4d686SBarry Smith   PetscFunctionBegin;
483a4d4d686SBarry Smith   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
484a4d4d686SBarry Smith   if (ctx) {
485a4d4d686SBarry Smith     *h = ctx->currenth;
486a4d4d686SBarry Smith   }
487a4d4d686SBarry Smith   PetscFunctionReturn(0);
488a4d4d686SBarry Smith }
489a4d4d686SBarry Smith 
490a4d4d686SBarry Smith #undef __FUNC__
4915a655dc6SBarry Smith #define __FUNC__ "MatSNESMFKSPMonitor"
492a4d4d686SBarry Smith /*
4935a655dc6SBarry Smith    MatSNESMFKSPMonitor - A KSP monitor for use with the default PETSc
49465f2ba5bSLois Curfman McInnes    SNES matrix free routines. Prints the differencing parameter used at
49565f2ba5bSLois Curfman McInnes    each step.
496a4d4d686SBarry Smith */
4975a655dc6SBarry Smith int MatSNESMFKSPMonitor(KSP ksp,int n,double rnorm,void *dummy)
498a4d4d686SBarry Smith {
499a4d4d686SBarry Smith   PC             pc;
5005a655dc6SBarry Smith   MatSNESMFCtx   ctx;
501a4d4d686SBarry Smith   int            ierr;
502a4d4d686SBarry Smith   Mat            mat;
503a4d4d686SBarry Smith   MPI_Comm       comm;
504a4d4d686SBarry Smith   PetscTruth     nonzeroinitialguess;
505a4d4d686SBarry Smith 
506a4d4d686SBarry Smith   PetscFunctionBegin;
507a4d4d686SBarry Smith   ierr = PetscObjectGetComm((PetscObject)ksp,&comm);CHKERRQ(ierr);
508a4d4d686SBarry Smith   ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
509a4d4d686SBarry Smith   ierr = KSPGetInitialGuessNonzero(ksp,&nonzeroinitialguess);CHKERRQ(ierr);
510a4d4d686SBarry Smith   ierr = PCGetOperators(pc,&mat,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);
511a4d4d686SBarry Smith   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
5129a6cb015SBarry Smith   if (!ctx) {
5139a6cb015SBarry Smith     SETERRQ(1,1,"Matrix is not a matrix free shell matrix");
5149a6cb015SBarry Smith   }
515a4d4d686SBarry Smith   if (n > 0 || nonzeroinitialguess) {
516aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
517d132466eSBarry Smith     ierr = PetscPrintf(comm,"%d KSP Residual norm %14.12e h %g + %g i\n",n,rnorm,
518d132466eSBarry Smith                 PetscReal(ctx->currenth),PetscImaginary(ctx->currenth));CHKERRQ(ierr);
519a4d4d686SBarry Smith #else
520d132466eSBarry Smith     ierr = PetscPrintf(comm,"%d KSP Residual norm %14.12e h %g \n",n,rnorm,ctx->currenth);CHKERRQ(ierr);
521a4d4d686SBarry Smith #endif
522a4d4d686SBarry Smith   } else {
523d132466eSBarry Smith     ierr = PetscPrintf(comm,"%d KSP Residual norm %14.12e\n",n,rnorm);CHKERRQ(ierr);
524a4d4d686SBarry Smith   }
525a4d4d686SBarry Smith   PetscFunctionReturn(0);
526a4d4d686SBarry Smith }
527a4d4d686SBarry Smith 
528a4d4d686SBarry Smith #undef __FUNC__
52985614651SBarry Smith #define __FUNC__ "MatSNESMFSetFunction"
53085614651SBarry Smith /*@C
53185614651SBarry Smith    MatSNESMFSetFunction - Sets the function used in applying the matrix free.
53285614651SBarry Smith 
53385614651SBarry Smith    Collective on Mat
53485614651SBarry Smith 
53585614651SBarry Smith    Input Parameters:
53685614651SBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
53785614651SBarry Smith .  v   - workspace vector
53885614651SBarry Smith .  func - the function to use
53985614651SBarry Smith -  funcctx - optional function context passed to function
54085614651SBarry Smith 
54185614651SBarry Smith    Level: advanced
54285614651SBarry Smith 
54385614651SBarry Smith    Notes:
54485614651SBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
54585614651SBarry Smith     matrix inside your compute Jacobian routine
54685614651SBarry Smith 
54785614651SBarry Smith     If this is not set then it will use the function set with SNESSetFunction()
54885614651SBarry Smith 
54985614651SBarry Smith .keywords: SNES, matrix-free, function
55085614651SBarry Smith 
55185614651SBarry Smith .seealso: MatCreateSNESMF(),MatSNESMFGetH(),
55285614651SBarry Smith           MatSNESMFSetHHistory(), MatSNESMFResetHHistory(),
55385614651SBarry Smith           MatSNESMFKSPMonitor(), SNESetFunction()
55485614651SBarry Smith @*/
55585614651SBarry Smith int MatSNESMFSetFunction(Mat mat,Vec v,int (*func)(SNES,Vec,Vec,void *),void *funcctx)
55685614651SBarry Smith {
55785614651SBarry Smith   MatSNESMFCtx ctx;
55885614651SBarry Smith   int          ierr;
55985614651SBarry Smith 
56085614651SBarry Smith   PetscFunctionBegin;
56185614651SBarry Smith   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
56285614651SBarry Smith   if (ctx) {
56385614651SBarry Smith     ctx->func    = func;
56485614651SBarry Smith     ctx->funcctx = funcctx;
56585614651SBarry Smith     ctx->funcvec = v;
56685614651SBarry Smith   }
56785614651SBarry Smith   PetscFunctionReturn(0);
56885614651SBarry Smith }
56985614651SBarry Smith 
57085614651SBarry Smith 
57185614651SBarry Smith #undef __FUNC__
5725a655dc6SBarry Smith #define __FUNC__ "MatSNESMFSetFunctionError"
573a4d4d686SBarry Smith /*@
5745a655dc6SBarry Smith    MatSNESMFSetFunctionError - Sets the error_rel for the approximation of
575a4d4d686SBarry Smith    matrix-vector products using finite differences.
576a4d4d686SBarry Smith 
577a4d4d686SBarry Smith    Collective on Mat
578a4d4d686SBarry Smith 
579a4d4d686SBarry Smith    Input Parameters:
5805a655dc6SBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
5819a6cb015SBarry Smith -  error_rel - relative error (should be set to the square root of
582a4d4d686SBarry Smith                the relative error in the function evaluations)
583a4d4d686SBarry Smith 
58415091d37SBarry Smith    Options Database Keys:
58515091d37SBarry Smith +  -snes_mf_err <error_rel> - Sets error_rel
58615091d37SBarry Smith 
58715091d37SBarry Smith    Level: advanced
58815091d37SBarry Smith 
589a4d4d686SBarry Smith    Notes:
590a4d4d686SBarry Smith    The default matrix-free matrix-vector product routine computes
591a4d4d686SBarry Smith .vb
59265f2ba5bSLois Curfman McInnes      F'(u)*a = [F(u+h*a) - F(u)]/h where
593a4d4d686SBarry Smith      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
594a4d4d686SBarry Smith        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   else
595a4d4d686SBarry Smith .ve
596a4d4d686SBarry Smith 
597a4d4d686SBarry Smith .keywords: SNES, matrix-free, parameters
598a4d4d686SBarry Smith 
5995a655dc6SBarry Smith .seealso: MatCreateSNESMF(),MatSNESMFGetH(),
6005a655dc6SBarry Smith           MatSNESMFSetHHistory(), MatSNESMFResetHHistory(),
6015a655dc6SBarry Smith           MatSNESMFKSPMonitor()
602a4d4d686SBarry Smith @*/
6035a655dc6SBarry Smith int MatSNESMFSetFunctionError(Mat mat,double error)
604a4d4d686SBarry Smith {
6055a655dc6SBarry Smith   MatSNESMFCtx ctx;
606a4d4d686SBarry Smith   int          ierr;
607a4d4d686SBarry Smith 
608a4d4d686SBarry Smith   PetscFunctionBegin;
609a4d4d686SBarry Smith   ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr);
610a4d4d686SBarry Smith   if (ctx) {
611a4d4d686SBarry Smith     if (error != PETSC_DEFAULT) ctx->error_rel = error;
612a4d4d686SBarry Smith   }
613a4d4d686SBarry Smith   PetscFunctionReturn(0);
614a4d4d686SBarry Smith }
615a4d4d686SBarry Smith 
616a4d4d686SBarry Smith #undef __FUNC__
6175a655dc6SBarry Smith #define __FUNC__ "MatSNESMFAddNullSpace"
618a4d4d686SBarry Smith /*@
61965f2ba5bSLois Curfman McInnes    MatSNESMFAddNullSpace - Provides a null space that an operator is
62065f2ba5bSLois Curfman McInnes    supposed to have.  Since roundoff will create a small component in
62165f2ba5bSLois Curfman McInnes    the null space, if you know the null space you may have it
62265f2ba5bSLois Curfman McInnes    automatically removed.
623a4d4d686SBarry Smith 
624a4d4d686SBarry Smith    Collective on Mat
625a4d4d686SBarry Smith 
626a4d4d686SBarry Smith    Input Parameters:
627a4d4d686SBarry Smith +  J - the matrix-free matrix context
62885614651SBarry Smith -  nullsp - object created with PCNullSpaceCreate()
629a4d4d686SBarry Smith 
63015091d37SBarry Smith    Level: advanced
63115091d37SBarry Smith 
632a4d4d686SBarry Smith .keywords: SNES, matrix-free, null space
633a4d4d686SBarry Smith 
6345a655dc6SBarry Smith .seealso: MatSNESMFGetH(), MatCreateSNESMF(),
6355a655dc6SBarry Smith           MatSNESMFSetHHistory(), MatSNESMFResetHHistory(),
6365a655dc6SBarry Smith           MatSNESMFKSPMonitor(), MatSNESMFErrorRel()
637a4d4d686SBarry Smith 
638a4d4d686SBarry Smith @*/
63985614651SBarry Smith int MatSNESMFAddNullSpace(Mat J,PCNullSpace nullsp)
640a4d4d686SBarry Smith {
641a4d4d686SBarry Smith   int          ierr;
6425a655dc6SBarry Smith   MatSNESMFCtx ctx;
643a4d4d686SBarry Smith   MPI_Comm     comm;
644a4d4d686SBarry Smith 
645a4d4d686SBarry Smith   PetscFunctionBegin;
6462d0c0e3bSBarry Smith   ierr = PetscObjectGetComm((PetscObject)J,&comm);CHKERRQ(ierr);
647a4d4d686SBarry Smith 
648a4d4d686SBarry Smith   ierr = MatShellGetContext(J,(void **)&ctx);CHKERRQ(ierr);
649a4d4d686SBarry Smith   /* no context indicates that it is not the "matrix free" matrix type */
650a4d4d686SBarry Smith   if (!ctx) PetscFunctionReturn(0);
65185614651SBarry Smith   ctx->sp = nullsp;
65285614651SBarry Smith   ierr = PetscObjectReference((PetscObject)nullsp);CHKERRQ(ierr);
653a4d4d686SBarry Smith   PetscFunctionReturn(0);
654a4d4d686SBarry Smith }
655a4d4d686SBarry Smith 
656a4d4d686SBarry Smith #undef __FUNC__
6575a655dc6SBarry Smith #define __FUNC__ "MatSNESMFSetHHistory"
658a4d4d686SBarry Smith /*@
65965f2ba5bSLois Curfman McInnes    MatSNESMFSetHHistory - Sets an array to collect a history of the
66065f2ba5bSLois Curfman McInnes    differencing values (h) computed for the matrix-free product.
661a4d4d686SBarry Smith 
662a4d4d686SBarry Smith    Collective on Mat
663a4d4d686SBarry Smith 
664a4d4d686SBarry Smith    Input Parameters:
665a4d4d686SBarry Smith +  J - the matrix-free matrix context
66665f2ba5bSLois Curfman McInnes .  histroy - space to hold the history
66765f2ba5bSLois Curfman McInnes -  nhistory - number of entries in history, if more entries are generated than
66865f2ba5bSLois Curfman McInnes               nhistory, then the later ones are discarded
669a4d4d686SBarry Smith 
67015091d37SBarry Smith    Level: advanced
67115091d37SBarry Smith 
672a4d4d686SBarry Smith    Notes:
67365f2ba5bSLois Curfman McInnes    Use MatSNESMFResetHHistory() to reset the history counter and collect
67465f2ba5bSLois Curfman McInnes    a new batch of differencing parameters, h.
675a4d4d686SBarry Smith 
676a4d4d686SBarry Smith .keywords: SNES, matrix-free, h history, differencing history
677a4d4d686SBarry Smith 
6785a655dc6SBarry Smith .seealso: MatSNESMFGetH(), MatCreateSNESMF(),
6795a655dc6SBarry Smith           MatSNESMFResetHHistory(),
6805a655dc6SBarry Smith           MatSNESMFKSPMonitor(), MatSNESMFSetFunctionError()
681a4d4d686SBarry Smith 
682a4d4d686SBarry Smith @*/
6835a655dc6SBarry Smith int MatSNESMFSetHHistory(Mat J,Scalar *history,int nhistory)
684a4d4d686SBarry Smith {
685a4d4d686SBarry Smith   int          ierr;
6865a655dc6SBarry Smith   MatSNESMFCtx ctx;
687a4d4d686SBarry Smith 
688a4d4d686SBarry Smith   PetscFunctionBegin;
689a4d4d686SBarry Smith 
690a4d4d686SBarry Smith   ierr = MatShellGetContext(J,(void **)&ctx);CHKERRQ(ierr);
691a4d4d686SBarry Smith   /* no context indicates that it is not the "matrix free" matrix type */
692a4d4d686SBarry Smith   if (!ctx) PetscFunctionReturn(0);
693a4d4d686SBarry Smith   ctx->historyh    = history;
694a4d4d686SBarry Smith   ctx->maxcurrenth = nhistory;
695a4d4d686SBarry Smith   ctx->currenth    = 0;
696a4d4d686SBarry Smith 
697a4d4d686SBarry Smith   PetscFunctionReturn(0);
698a4d4d686SBarry Smith }
699a4d4d686SBarry Smith 
700a4d4d686SBarry Smith #undef __FUNC__
7015a655dc6SBarry Smith #define __FUNC__ "MatSNESMFResetHHistory"
702a4d4d686SBarry Smith /*@
7035a655dc6SBarry Smith    MatSNESMFResetHHistory - Resets the counter to zero to begin
704a4d4d686SBarry Smith    collecting a new set of differencing histories.
705a4d4d686SBarry Smith 
706a4d4d686SBarry Smith    Collective on Mat
707a4d4d686SBarry Smith 
708a4d4d686SBarry Smith    Input Parameters:
709a4d4d686SBarry Smith .  J - the matrix-free matrix context
710a4d4d686SBarry Smith 
71115091d37SBarry Smith    Level: advanced
71215091d37SBarry Smith 
713a4d4d686SBarry Smith    Notes:
71465f2ba5bSLois Curfman McInnes    Use MatSNESMFSetHHistory() to create the original history counter.
715a4d4d686SBarry Smith 
716a4d4d686SBarry Smith .keywords: SNES, matrix-free, h history, differencing history
717a4d4d686SBarry Smith 
7185a655dc6SBarry Smith .seealso: MatSNESMFGetH(), MatCreateSNESMF(),
7195a655dc6SBarry Smith           MatSNESMFSetHHistory(),
7205a655dc6SBarry Smith           MatSNESMFKSPMonitor(), MatSNESMFSetFunctionError()
721a4d4d686SBarry Smith 
722a4d4d686SBarry Smith @*/
7235a655dc6SBarry Smith int MatSNESMFResetHHistory(Mat J)
724a4d4d686SBarry Smith {
725a4d4d686SBarry Smith   int          ierr;
7265a655dc6SBarry Smith   MatSNESMFCtx ctx;
727a4d4d686SBarry Smith 
728a4d4d686SBarry Smith   PetscFunctionBegin;
729a4d4d686SBarry Smith 
730a4d4d686SBarry Smith   ierr = MatShellGetContext(J,(void **)&ctx);CHKERRQ(ierr);
731a4d4d686SBarry Smith   /* no context indicates that it is not the "matrix free" matrix type */
732a4d4d686SBarry Smith   if (!ctx) PetscFunctionReturn(0);
733be726c96SBarry Smith   ctx->ncurrenth    = 0;
734a4d4d686SBarry Smith 
735a4d4d686SBarry Smith   PetscFunctionReturn(0);
736a4d4d686SBarry Smith }
737a4d4d686SBarry Smith 
738