xref: /petsc/src/mat/impls/mffd/mffd.c (revision 5fd668637986a8d8518383a9159eebc368e1d5b4)
1e884886fSBarry Smith 
2b45d2f2cSJed Brown #include <petsc-private/matimpl.h>
3c6db04a5SJed Brown #include <../src/mat/impls/mffd/mffdimpl.h>   /*I  "petscmat.h"   I*/
4e884886fSBarry Smith 
5140e18c1SBarry Smith PetscFunctionList MatMFFDList        = 0;
6ace3abfcSBarry Smith PetscBool         MatMFFDRegisterAllCalled = PETSC_FALSE;
7e884886fSBarry Smith 
87087cfbeSBarry Smith PetscClassId  MATMFFD_CLASSID;
9166c7f25SBarry Smith PetscLogEvent MATMFFD_Mult;
10e884886fSBarry Smith 
11ace3abfcSBarry Smith static PetscBool  MatMFFDPackageInitialized = PETSC_FALSE;
12b022a5c1SBarry Smith #undef __FUNCT__
13b022a5c1SBarry Smith #define __FUNCT__ "MatMFFDFinalizePackage"
14b022a5c1SBarry Smith /*@C
152390153bSJed Brown   MatMFFDFinalizePackage - This function destroys everything in the MatMFFD package. It is
16b022a5c1SBarry Smith   called from PetscFinalize().
17b022a5c1SBarry Smith 
18b022a5c1SBarry Smith   Level: developer
19b022a5c1SBarry Smith 
202390153bSJed Brown .keywords: Petsc, destroy, package
21b022a5c1SBarry Smith .seealso: PetscFinalize()
22b022a5c1SBarry Smith @*/
237087cfbeSBarry Smith PetscErrorCode  MatMFFDFinalizePackage(void)
24b022a5c1SBarry Smith {
25b022a5c1SBarry Smith   PetscFunctionBegin;
26b022a5c1SBarry Smith   MatMFFDPackageInitialized = PETSC_FALSE;
27b022a5c1SBarry Smith   MatMFFDRegisterAllCalled  = PETSC_FALSE;
28b022a5c1SBarry Smith   MatMFFDList               = PETSC_NULL;
29b022a5c1SBarry Smith   PetscFunctionReturn(0);
30b022a5c1SBarry Smith }
31b022a5c1SBarry Smith 
32e884886fSBarry Smith #undef __FUNCT__
333ec795f1SBarry Smith #define __FUNCT__ "MatMFFDInitializePackage"
343ec795f1SBarry Smith /*@C
353ec795f1SBarry Smith   MatMFFDInitializePackage - This function initializes everything in the MatMFFD package. It is called
363ec795f1SBarry Smith   from PetscDLLibraryRegister() when using dynamic libraries, and on the first call to MatCreate_MFFD()
373ec795f1SBarry Smith   when using static libraries.
383ec795f1SBarry Smith 
393ec795f1SBarry Smith   Input Parameter:
403ec795f1SBarry Smith . path - The dynamic library path, or PETSC_NULL
413ec795f1SBarry Smith 
423ec795f1SBarry Smith   Level: developer
433ec795f1SBarry Smith 
443ec795f1SBarry Smith .keywords: Vec, initialize, package
453ec795f1SBarry Smith .seealso: PetscInitialize()
463ec795f1SBarry Smith @*/
477087cfbeSBarry Smith PetscErrorCode  MatMFFDInitializePackage(const char path[])
483ec795f1SBarry Smith {
493ec795f1SBarry Smith   char              logList[256];
503ec795f1SBarry Smith   char              *className;
51ace3abfcSBarry Smith   PetscBool         opt;
523ec795f1SBarry Smith   PetscErrorCode    ierr;
533ec795f1SBarry Smith 
543ec795f1SBarry Smith   PetscFunctionBegin;
55b022a5c1SBarry Smith   if (MatMFFDPackageInitialized) PetscFunctionReturn(0);
56b022a5c1SBarry Smith   MatMFFDPackageInitialized = PETSC_TRUE;
573ec795f1SBarry Smith   /* Register Classes */
580700a824SBarry Smith   ierr = PetscClassIdRegister("MatMFFD",&MATMFFD_CLASSID);CHKERRQ(ierr);
593ec795f1SBarry Smith   /* Register Constructors */
603ec795f1SBarry Smith   ierr = MatMFFDRegisterAll(path);CHKERRQ(ierr);
613ec795f1SBarry Smith   /* Register Events */
620700a824SBarry Smith   ierr = PetscLogEventRegister("MatMult MF",          MATMFFD_CLASSID,&MATMFFD_Mult);CHKERRQ(ierr);
633ec795f1SBarry Smith 
643ec795f1SBarry Smith   /* Process info exclusions */
653ec795f1SBarry Smith   ierr = PetscOptionsGetString(PETSC_NULL, "-info_exclude", logList, 256, &opt);CHKERRQ(ierr);
663ec795f1SBarry Smith   if (opt) {
673ec795f1SBarry Smith     ierr = PetscStrstr(logList, "matmffd", &className);CHKERRQ(ierr);
683ec795f1SBarry Smith     if (className) {
690700a824SBarry Smith       ierr = PetscInfoDeactivateClass(MATMFFD_CLASSID);CHKERRQ(ierr);
703ec795f1SBarry Smith     }
713ec795f1SBarry Smith   }
723ec795f1SBarry Smith   /* Process summary exclusions */
733ec795f1SBarry Smith   ierr = PetscOptionsGetString(PETSC_NULL, "-log_summary_exclude", logList, 256, &opt);CHKERRQ(ierr);
743ec795f1SBarry Smith   if (opt) {
753ec795f1SBarry Smith     ierr = PetscStrstr(logList, "matmffd", &className);CHKERRQ(ierr);
763ec795f1SBarry Smith     if (className) {
770700a824SBarry Smith       ierr = PetscLogEventDeactivateClass(MATMFFD_CLASSID);CHKERRQ(ierr);
783ec795f1SBarry Smith     }
793ec795f1SBarry Smith   }
80b022a5c1SBarry Smith   ierr = PetscRegisterFinalize(MatMFFDFinalizePackage);CHKERRQ(ierr);
813ec795f1SBarry Smith   PetscFunctionReturn(0);
823ec795f1SBarry Smith }
833ec795f1SBarry Smith 
843ec795f1SBarry Smith #undef __FUNCT__
85e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetType"
86e884886fSBarry Smith /*@C
87e884886fSBarry Smith     MatMFFDSetType - Sets the method that is used to compute the
88e884886fSBarry Smith     differencing parameter for finite differene matrix-free formulations.
89e884886fSBarry Smith 
90e884886fSBarry Smith     Input Parameters:
91e884886fSBarry Smith +   mat - the "matrix-free" matrix created via MatCreateSNESMF(), or MatCreateMFFD()
92e884886fSBarry Smith           or MatSetType(mat,MATMFFD);
93e884886fSBarry Smith -   ftype - the type requested, either MATMFFD_WP or MATMFFD_DS
94e884886fSBarry Smith 
95e884886fSBarry Smith     Level: advanced
96e884886fSBarry Smith 
97e884886fSBarry Smith     Notes:
98e884886fSBarry Smith     For example, such routines can compute h for use in
99e884886fSBarry Smith     Jacobian-vector products of the form
100e884886fSBarry Smith 
101e884886fSBarry Smith                         F(x+ha) - F(x)
102e884886fSBarry Smith           F'(u)a  ~=  ----------------
103e884886fSBarry Smith                               h
104e884886fSBarry Smith 
105d2d6cebeSBarry Smith .seealso: MatCreateSNESMF(), MatMFFDRegisterDynamic(), MatMFFDSetFunction()
106e884886fSBarry Smith @*/
10719fd82e9SBarry Smith PetscErrorCode  MatMFFDSetType(Mat mat,MatMFFDType ftype)
108e884886fSBarry Smith {
109e884886fSBarry Smith   PetscErrorCode ierr,(*r)(MatMFFD);
110e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
111ace3abfcSBarry Smith   PetscBool      match;
112e884886fSBarry Smith 
113e884886fSBarry Smith   PetscFunctionBegin;
1140700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
115e884886fSBarry Smith   PetscValidCharPointer(ftype,2);
116e884886fSBarry Smith 
117251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)mat,MATMFFD,&match);CHKERRQ(ierr);
1189a13eae7SJed Brown   if (!match) PetscFunctionReturn(0);
1199a13eae7SJed Brown 
120e884886fSBarry Smith   /* already set, so just return */
121251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)ctx,ftype,&match);CHKERRQ(ierr);
122e884886fSBarry Smith   if (match) PetscFunctionReturn(0);
123e884886fSBarry Smith 
124e884886fSBarry Smith   /* destroy the old one if it exists */
125e884886fSBarry Smith   if (ctx->ops->destroy) {
126e884886fSBarry Smith     ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr);
127e884886fSBarry Smith   }
128e884886fSBarry Smith 
129140e18c1SBarry Smith   ierr =  PetscFunctionListFind(((PetscObject)ctx)->comm,MatMFFDList,ftype,PETSC_TRUE,(void (**)(void)) &r);CHKERRQ(ierr);
130e32f2f54SBarry Smith   if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unknown MatMFFD type %s given",ftype);
131e884886fSBarry Smith   ierr = (*r)(ctx);CHKERRQ(ierr);
132e884886fSBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)ctx,ftype);CHKERRQ(ierr);
133e884886fSBarry Smith   PetscFunctionReturn(0);
134e884886fSBarry Smith }
135e884886fSBarry Smith 
136e884886fSBarry Smith typedef PetscErrorCode (*FCN1)(void*,Vec); /* force argument to next function to not be extern C*/
137e884886fSBarry Smith EXTERN_C_BEGIN
138e884886fSBarry Smith #undef __FUNCT__
139c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioniBase_MFFD"
1407087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctioniBase_MFFD(Mat mat,FCN1 func)
141e884886fSBarry Smith {
142e884886fSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
143e884886fSBarry Smith 
144e884886fSBarry Smith   PetscFunctionBegin;
145e884886fSBarry Smith   ctx->funcisetbase = func;
146e884886fSBarry Smith   PetscFunctionReturn(0);
147e884886fSBarry Smith }
148e884886fSBarry Smith EXTERN_C_END
149e884886fSBarry Smith 
150e884886fSBarry Smith typedef PetscErrorCode (*FCN2)(void*,PetscInt,Vec,PetscScalar*); /* force argument to next function to not be extern C*/
151e884886fSBarry Smith EXTERN_C_BEGIN
152e884886fSBarry Smith #undef __FUNCT__
153c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioni_MFFD"
1547087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctioni_MFFD(Mat mat,FCN2 funci)
155e884886fSBarry Smith {
156e884886fSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
157e884886fSBarry Smith 
158e884886fSBarry Smith   PetscFunctionBegin;
159e884886fSBarry Smith   ctx->funci = funci;
160e884886fSBarry Smith   PetscFunctionReturn(0);
161e884886fSBarry Smith }
162e884886fSBarry Smith EXTERN_C_END
163e884886fSBarry Smith 
164bc0f08ceSBarry Smith EXTERN_C_BEGIN
165bc0f08ceSBarry Smith #undef __FUNCT__
166bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDResetHHistory_MFFD"
167bc0f08ceSBarry Smith PetscErrorCode  MatMFFDResetHHistory_MFFD(Mat J)
168bc0f08ceSBarry Smith {
169bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)J->data;
170bc0f08ceSBarry Smith 
171bc0f08ceSBarry Smith   PetscFunctionBegin;
172bc0f08ceSBarry Smith   ctx->ncurrenth    = 0;
173bc0f08ceSBarry Smith   PetscFunctionReturn(0);
174bc0f08ceSBarry Smith }
175bc0f08ceSBarry Smith EXTERN_C_END
176e884886fSBarry Smith 
177e884886fSBarry Smith #undef __FUNCT__
178e884886fSBarry Smith #define __FUNCT__ "MatMFFDRegister"
1797087cfbeSBarry Smith PetscErrorCode  MatMFFDRegister(const char sname[],const char path[],const char name[],PetscErrorCode (*function)(MatMFFD))
180e884886fSBarry Smith {
181e884886fSBarry Smith   PetscErrorCode ierr;
182e884886fSBarry Smith   char           fullname[PETSC_MAX_PATH_LEN];
183e884886fSBarry Smith 
184e884886fSBarry Smith   PetscFunctionBegin;
185140e18c1SBarry Smith   ierr = PetscFunctionListConcat(path,name,fullname);CHKERRQ(ierr);
186140e18c1SBarry Smith   ierr = PetscFunctionListAdd(PETSC_COMM_WORLD,&MatMFFDList,sname,fullname,(void (*)(void))function);CHKERRQ(ierr);
187e884886fSBarry Smith   PetscFunctionReturn(0);
188e884886fSBarry Smith }
189e884886fSBarry Smith 
190e884886fSBarry Smith 
191e884886fSBarry Smith #undef __FUNCT__
192e884886fSBarry Smith #define __FUNCT__ "MatMFFDRegisterDestroy"
193e884886fSBarry Smith /*@C
194e884886fSBarry Smith    MatMFFDRegisterDestroy - Frees the list of MatMFFD methods that were
195e884886fSBarry Smith    registered by MatMFFDRegisterDynamic).
196e884886fSBarry Smith 
197e884886fSBarry Smith    Not Collective
198e884886fSBarry Smith 
199e884886fSBarry Smith    Level: developer
200e884886fSBarry Smith 
201e884886fSBarry Smith .keywords: MatMFFD, register, destroy
202e884886fSBarry Smith 
203e884886fSBarry Smith .seealso: MatMFFDRegisterDynamic), MatMFFDRegisterAll()
204e884886fSBarry Smith @*/
2057087cfbeSBarry Smith PetscErrorCode  MatMFFDRegisterDestroy(void)
206e884886fSBarry Smith {
207e884886fSBarry Smith   PetscErrorCode ierr;
208e884886fSBarry Smith 
209e884886fSBarry Smith   PetscFunctionBegin;
210140e18c1SBarry Smith   ierr = PetscFunctionListDestroy(&MatMFFDList);CHKERRQ(ierr);
211e884886fSBarry Smith   MatMFFDRegisterAllCalled = PETSC_FALSE;
212e884886fSBarry Smith   PetscFunctionReturn(0);
213e884886fSBarry Smith }
214e884886fSBarry Smith 
215bc0f08ceSBarry Smith EXTERN_C_BEGIN
216bc0f08ceSBarry Smith #undef __FUNCT__
217bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDAddNullSpace_MFFD"
218bc0f08ceSBarry Smith PetscErrorCode  MatMFFDAddNullSpace_MFFD(Mat J,MatNullSpace nullsp)
219bc0f08ceSBarry Smith {
220bc0f08ceSBarry Smith   PetscErrorCode ierr;
221bc0f08ceSBarry Smith   MatMFFD      ctx = (MatMFFD)J->data;
222bc0f08ceSBarry Smith 
223bc0f08ceSBarry Smith   PetscFunctionBegin;
224bc0f08ceSBarry Smith   ierr = PetscObjectReference((PetscObject)nullsp);CHKERRQ(ierr);
225bc0f08ceSBarry Smith   if (ctx->sp) { ierr = MatNullSpaceDestroy(&ctx->sp);CHKERRQ(ierr); }
226bc0f08ceSBarry Smith   ctx->sp = nullsp;
227bc0f08ceSBarry Smith   PetscFunctionReturn(0);
228bc0f08ceSBarry Smith }
229bc0f08ceSBarry Smith EXTERN_C_END
230bc0f08ceSBarry Smith 
231e884886fSBarry Smith /* ----------------------------------------------------------------------------------------*/
232e884886fSBarry Smith #undef __FUNCT__
233e884886fSBarry Smith #define __FUNCT__ "MatDestroy_MFFD"
234e884886fSBarry Smith PetscErrorCode MatDestroy_MFFD(Mat mat)
235e884886fSBarry Smith {
236e884886fSBarry Smith   PetscErrorCode ierr;
237e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
238e884886fSBarry Smith 
239e884886fSBarry Smith   PetscFunctionBegin;
2406bf464f9SBarry Smith   ierr = VecDestroy(&ctx->w);CHKERRQ(ierr);
2416bf464f9SBarry Smith   ierr = VecDestroy(&ctx->drscale);CHKERRQ(ierr);
2426bf464f9SBarry Smith   ierr = VecDestroy(&ctx->dlscale);CHKERRQ(ierr);
2436bf464f9SBarry Smith   ierr = VecDestroy(&ctx->dshift);CHKERRQ(ierr);
244cfe22f5eSBarry Smith   if (ctx->current_f_allocated) {
2456bf464f9SBarry Smith     ierr = VecDestroy(&ctx->current_f);CHKERRQ(ierr);
246cfe22f5eSBarry Smith   }
247e884886fSBarry Smith   if (ctx->ops->destroy) {ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr);}
2486bf464f9SBarry Smith   ierr = MatNullSpaceDestroy(&ctx->sp);CHKERRQ(ierr);
2496bf464f9SBarry Smith   ierr = PetscHeaderDestroy(&ctx);CHKERRQ(ierr);
250bf0cc555SLisandro Dalcin   mat->data = 0;
251e884886fSBarry Smith 
252e884886fSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetBase_C","",PETSC_NULL);CHKERRQ(ierr);
253e884886fSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctioniBase_C","",PETSC_NULL);CHKERRQ(ierr);
254e884886fSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctioni_C","",PETSC_NULL);CHKERRQ(ierr);
255bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunction_C","",PETSC_NULL);CHKERRQ(ierr);
256bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctionError_C","",PETSC_NULL);CHKERRQ(ierr);
257e884886fSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetCheckh_C","",PETSC_NULL);CHKERRQ(ierr);
258bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetPeriod_C","",PETSC_NULL);CHKERRQ(ierr);
259bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDResetHHistory_C","",PETSC_NULL);CHKERRQ(ierr);
260bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDAddNullSpace_C","",PETSC_NULL);CHKERRQ(ierr);
261e884886fSBarry Smith 
262e884886fSBarry Smith   PetscFunctionReturn(0);
263e884886fSBarry Smith }
264e884886fSBarry Smith 
265e884886fSBarry Smith #undef __FUNCT__
266e884886fSBarry Smith #define __FUNCT__ "MatView_MFFD"
267e884886fSBarry Smith /*
268e884886fSBarry Smith    MatMFFDView_MFFD - Views matrix-free parameters.
269e884886fSBarry Smith 
270e884886fSBarry Smith */
271e884886fSBarry Smith PetscErrorCode MatView_MFFD(Mat J,PetscViewer viewer)
272e884886fSBarry Smith {
273e884886fSBarry Smith   PetscErrorCode ierr;
274e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)J->data;
275a283635eSDmitry Karpeev   PetscBool      iascii, viewbase, viewfunction;
276a283635eSDmitry Karpeev   const char*    prefix;
277e884886fSBarry Smith 
278e884886fSBarry Smith   PetscFunctionBegin;
279251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
280e884886fSBarry Smith   if (iascii) {
281a283635eSDmitry Karpeev     ierr = PetscViewerASCIIPrintf(viewer,"Matrix-free approximation:\n");CHKERRQ(ierr);
282a283635eSDmitry Karpeev     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
283e884886fSBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"err=%G (relative error in function evaluation)\n",ctx->error_rel);CHKERRQ(ierr);
2847adad957SLisandro Dalcin     if (!((PetscObject)ctx)->type_name) {
285e884886fSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"The compute h routine has not yet been set\n");CHKERRQ(ierr);
286e884886fSBarry Smith     } else {
2877adad957SLisandro Dalcin       ierr = PetscViewerASCIIPrintf(viewer,"Using %s compute h routine\n",((PetscObject)ctx)->type_name);CHKERRQ(ierr);
288e884886fSBarry Smith     }
289e884886fSBarry Smith     if (ctx->ops->view) {
290e884886fSBarry Smith       ierr = (*ctx->ops->view)(ctx,viewer);CHKERRQ(ierr);
291e884886fSBarry Smith     }
292a283635eSDmitry Karpeev     ierr = PetscObjectGetOptionsPrefix((PetscObject)J, &prefix);CHKERRQ(ierr);
293a283635eSDmitry Karpeev 
294a283635eSDmitry Karpeev     ierr = PetscOptionsHasName(prefix, "-mat_mffd_view_base", &viewbase);CHKERRQ(ierr);
295a283635eSDmitry Karpeev     if (viewbase) {
296a283635eSDmitry Karpeev       ierr = PetscViewerASCIIPrintf(viewer, "Base:\n");CHKERRQ(ierr);
297a283635eSDmitry Karpeev       ierr = VecView(ctx->current_u, viewer);CHKERRQ(ierr);
298a283635eSDmitry Karpeev     }
299a283635eSDmitry Karpeev     ierr = PetscOptionsHasName(prefix, "-mat_mffd_view_function", &viewfunction);CHKERRQ(ierr);
300a283635eSDmitry Karpeev     if (viewfunction) {
301a283635eSDmitry Karpeev       ierr = PetscViewerASCIIPrintf(viewer, "Function:\n");CHKERRQ(ierr);
302a283635eSDmitry Karpeev       ierr = VecView(ctx->current_f, viewer);CHKERRQ(ierr);
303a283635eSDmitry Karpeev     }
304a283635eSDmitry Karpeev     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
30511aeaf0aSBarry Smith   }
306e884886fSBarry Smith   PetscFunctionReturn(0);
307e884886fSBarry Smith }
308e884886fSBarry Smith 
309e884886fSBarry Smith #undef __FUNCT__
310e884886fSBarry Smith #define __FUNCT__ "MatAssemblyEnd_MFFD"
311e884886fSBarry Smith /*
312e884886fSBarry Smith    MatAssemblyEnd_MFFD - Resets the ctx->ncurrenth to zero. This
313e884886fSBarry Smith    allows the user to indicate the beginning of a new linear solve by calling
314e884886fSBarry Smith    MatAssemblyXXX() on the matrix free matrix. This then allows the
3151d0fab5eSBarry Smith    MatCreateMFFD_WP() to properly compute ||U|| only the first time
316e884886fSBarry Smith    in the linear solver rather than every time.
317e884886fSBarry Smith */
318e884886fSBarry Smith PetscErrorCode MatAssemblyEnd_MFFD(Mat J,MatAssemblyType mt)
319e884886fSBarry Smith {
320e884886fSBarry Smith   PetscErrorCode ierr;
321e884886fSBarry Smith   MatMFFD        j = (MatMFFD)J->data;
322e884886fSBarry Smith 
323e884886fSBarry Smith   PetscFunctionBegin;
324e884886fSBarry Smith   ierr      = MatMFFDResetHHistory(J);CHKERRQ(ierr);
325e884886fSBarry Smith   j->vshift = 0.0;
326e884886fSBarry Smith   j->vscale = 1.0;
327e884886fSBarry Smith   PetscFunctionReturn(0);
328e884886fSBarry Smith }
329e884886fSBarry Smith 
330e884886fSBarry Smith #undef __FUNCT__
331e884886fSBarry Smith #define __FUNCT__ "MatMult_MFFD"
332e884886fSBarry Smith /*
333e884886fSBarry Smith   MatMult_MFFD - Default matrix-free form for Jacobian-vector product, y = F'(u)*a:
334e884886fSBarry Smith 
335e884886fSBarry Smith         y ~= (F(u + ha) - F(u))/h,
336e884886fSBarry Smith   where F = nonlinear function, as set by SNESSetFunction()
337e884886fSBarry Smith         u = current iterate
338e884886fSBarry Smith         h = difference interval
339e884886fSBarry Smith */
340e884886fSBarry Smith PetscErrorCode MatMult_MFFD(Mat mat,Vec a,Vec y)
341e884886fSBarry Smith {
342e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
343e884886fSBarry Smith   PetscScalar    h;
344e884886fSBarry Smith   Vec            w,U,F;
345e884886fSBarry Smith   PetscErrorCode ierr;
346ace3abfcSBarry Smith   PetscBool      zeroa;
347e884886fSBarry Smith 
348e884886fSBarry Smith   PetscFunctionBegin;
34984da1bf7SJed Brown   if (!ctx->current_u) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_ARG_WRONGSTATE,"MatMFFDSetBase() has not been called, this is often caused by forgetting to call \n\t\tMatAssemblyBegin/End on the first Mat in the SNES compute function");
350e884886fSBarry Smith   /* We log matrix-free matrix-vector products separately, so that we can
351e884886fSBarry Smith      separate the performance monitoring from the cases that use conventional
352e884886fSBarry Smith      storage.  We may eventually modify event logging to associate events
353e884886fSBarry Smith      with particular objects, hence alleviating the more general problem. */
354e884886fSBarry Smith   ierr = PetscLogEventBegin(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr);
355e884886fSBarry Smith 
356e884886fSBarry Smith   w    = ctx->w;
357e884886fSBarry Smith   U    = ctx->current_u;
3583ec795f1SBarry Smith   F    = ctx->current_f;
359e884886fSBarry Smith   /*
360e884886fSBarry Smith       Compute differencing parameter
361e884886fSBarry Smith   */
362e884886fSBarry Smith   if (!ctx->ops->compute) {
363e884886fSBarry Smith     ierr = MatMFFDSetType(mat,MATMFFD_WP);CHKERRQ(ierr);
3649c6ac3b3SBarry Smith     ierr = MatSetFromOptions(mat);CHKERRQ(ierr);
365e884886fSBarry Smith   }
366e884886fSBarry Smith   ierr = (*ctx->ops->compute)(ctx,U,a,&h,&zeroa);CHKERRQ(ierr);
367e884886fSBarry Smith   if (zeroa) {
368e884886fSBarry Smith     ierr = VecSet(y,0.0);CHKERRQ(ierr);
369e884886fSBarry Smith     PetscFunctionReturn(0);
370e884886fSBarry Smith   }
371e884886fSBarry Smith 
372e32f2f54SBarry Smith   if (PetscIsInfOrNanScalar(h)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Computed Nan differencing parameter h");
373e884886fSBarry Smith   if (ctx->checkh) {
374e884886fSBarry Smith     ierr = (*ctx->checkh)(ctx->checkhctx,U,a,&h);CHKERRQ(ierr);
375e884886fSBarry Smith   }
376e884886fSBarry Smith 
377e884886fSBarry Smith   /* keep a record of the current differencing parameter h */
378e884886fSBarry Smith   ctx->currenth = h;
379e884886fSBarry Smith #if defined(PETSC_USE_COMPLEX)
380e884886fSBarry Smith   ierr = PetscInfo2(mat,"Current differencing parameter: %G + %G i\n",PetscRealPart(h),PetscImaginaryPart(h));CHKERRQ(ierr);
381e884886fSBarry Smith #else
382e884886fSBarry Smith   ierr = PetscInfo1(mat,"Current differencing parameter: %15.12e\n",h);CHKERRQ(ierr);
383e884886fSBarry Smith #endif
384e884886fSBarry Smith   if (ctx->historyh && ctx->ncurrenth < ctx->maxcurrenth) {
385e884886fSBarry Smith     ctx->historyh[ctx->ncurrenth] = h;
386e884886fSBarry Smith   }
387e884886fSBarry Smith   ctx->ncurrenth++;
388e884886fSBarry Smith 
389e884886fSBarry Smith   /* w = u + ha */
390c3bb7e23SBarry Smith   if (ctx->drscale) {
391c3bb7e23SBarry Smith     ierr = VecPointwiseMult(ctx->drscale,a,U);CHKERRQ(ierr);
392c3bb7e23SBarry Smith     ierr = VecAYPX(U,h,w);CHKERRQ(ierr);
393c3bb7e23SBarry Smith   } else {
394e884886fSBarry Smith     ierr = VecWAXPY(w,h,a,U);CHKERRQ(ierr);
395c3bb7e23SBarry Smith   }
396e884886fSBarry Smith 
3971b797266SDmitry Karpeev   /* compute func(U) as base for differencing; only needed first time in and not when provided by user */
3981b797266SDmitry Karpeev   if (ctx->ncurrenth == 1 && ctx->current_f_allocated) {
399e884886fSBarry Smith     ierr = (*ctx->func)(ctx->funcctx,U,F);CHKERRQ(ierr);
40052121784SBarry Smith   }
401e884886fSBarry Smith   ierr = (*ctx->func)(ctx->funcctx,w,y);CHKERRQ(ierr);
402e884886fSBarry Smith 
403e884886fSBarry Smith   ierr = VecAXPY(y,-1.0,F);CHKERRQ(ierr);
404e884886fSBarry Smith   ierr = VecScale(y,1.0/h);CHKERRQ(ierr);
405e884886fSBarry Smith 
406c35ec66cSMatthew G Knepley   if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) {
407e884886fSBarry Smith     ierr = VecAXPBY(y,ctx->vshift,ctx->vscale,a);CHKERRQ(ierr);
408c3bb7e23SBarry Smith   }
409c3bb7e23SBarry Smith   if (ctx->dlscale) {
410c3bb7e23SBarry Smith     ierr = VecPointwiseMult(y,ctx->dlscale,y);CHKERRQ(ierr);
411c3bb7e23SBarry Smith   }
412c3bb7e23SBarry Smith   if (ctx->dshift) {
413c3bb7e23SBarry Smith     ierr = VecPointwiseMult(ctx->dshift,a,U);CHKERRQ(ierr);
414c3bb7e23SBarry Smith     ierr = VecAXPY(y,1.0,U);CHKERRQ(ierr);
415c3bb7e23SBarry Smith   }
416e884886fSBarry Smith 
417e884886fSBarry Smith   if (ctx->sp) {ierr = MatNullSpaceRemove(ctx->sp,y,PETSC_NULL);CHKERRQ(ierr);}
418e884886fSBarry Smith 
419e884886fSBarry Smith   ierr = PetscLogEventEnd(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr);
420e884886fSBarry Smith   PetscFunctionReturn(0);
421e884886fSBarry Smith }
422e884886fSBarry Smith 
423e884886fSBarry Smith #undef __FUNCT__
424e884886fSBarry Smith #define __FUNCT__ "MatGetDiagonal_MFFD"
425e884886fSBarry Smith /*
426e884886fSBarry Smith   MatGetDiagonal_MFFD - Gets the diagonal for a matrix free matrix
427e884886fSBarry Smith 
428e884886fSBarry Smith         y ~= (F(u + ha) - F(u))/h,
429e884886fSBarry Smith   where F = nonlinear function, as set by SNESSetFunction()
430e884886fSBarry Smith         u = current iterate
431e884886fSBarry Smith         h = difference interval
432e884886fSBarry Smith */
433e884886fSBarry Smith PetscErrorCode MatGetDiagonal_MFFD(Mat mat,Vec a)
434e884886fSBarry Smith {
435e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
436e884886fSBarry Smith   PetscScalar    h,*aa,*ww,v;
437e884886fSBarry Smith   PetscReal      epsilon = PETSC_SQRT_MACHINE_EPSILON,umin = 100.0*PETSC_SQRT_MACHINE_EPSILON;
438e884886fSBarry Smith   Vec            w,U;
439e884886fSBarry Smith   PetscErrorCode ierr;
440e884886fSBarry Smith   PetscInt       i,rstart,rend;
441e884886fSBarry Smith 
442e884886fSBarry Smith   PetscFunctionBegin;
443e7e72b3dSBarry Smith   if (!ctx->funci) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Requires calling MatMFFDSetFunctioni() first");
444e884886fSBarry Smith 
445e884886fSBarry Smith   w    = ctx->w;
446e884886fSBarry Smith   U    = ctx->current_u;
447e884886fSBarry Smith   ierr = (*ctx->func)(ctx->funcctx,U,a);CHKERRQ(ierr);
448e884886fSBarry Smith   ierr = (*ctx->funcisetbase)(ctx->funcctx,U);CHKERRQ(ierr);
449e884886fSBarry Smith   ierr = VecCopy(U,w);CHKERRQ(ierr);
450e884886fSBarry Smith 
451e884886fSBarry Smith   ierr = VecGetOwnershipRange(a,&rstart,&rend);CHKERRQ(ierr);
452e884886fSBarry Smith   ierr = VecGetArray(a,&aa);CHKERRQ(ierr);
453e884886fSBarry Smith   for (i=rstart; i<rend; i++) {
454e884886fSBarry Smith     ierr = VecGetArray(w,&ww);CHKERRQ(ierr);
455e884886fSBarry Smith     h  = ww[i-rstart];
456e884886fSBarry Smith     if (h == 0.0) h = 1.0;
457e884886fSBarry Smith     if (PetscAbsScalar(h) < umin && PetscRealPart(h) >= 0.0)     h = umin;
458e884886fSBarry Smith     else if (PetscRealPart(h) < 0.0 && PetscAbsScalar(h) < umin) h = -umin;
459e884886fSBarry Smith     h     *= epsilon;
460e884886fSBarry Smith 
461e884886fSBarry Smith     ww[i-rstart] += h;
462e884886fSBarry Smith     ierr = VecRestoreArray(w,&ww);CHKERRQ(ierr);
463e884886fSBarry Smith     ierr          = (*ctx->funci)(ctx->funcctx,i,w,&v);CHKERRQ(ierr);
464e884886fSBarry Smith     aa[i-rstart]  = (v - aa[i-rstart])/h;
465e884886fSBarry Smith 
466e884886fSBarry Smith     /* possibly shift and scale result */
467c35ec66cSMatthew G Knepley     if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) {
468e884886fSBarry Smith       aa[i - rstart] = ctx->vshift + ctx->vscale*aa[i-rstart];
469c3bb7e23SBarry Smith     }
470e884886fSBarry Smith 
471e884886fSBarry Smith     ierr = VecGetArray(w,&ww);CHKERRQ(ierr);
472e884886fSBarry Smith     ww[i-rstart] -= h;
473e884886fSBarry Smith     ierr = VecRestoreArray(w,&ww);CHKERRQ(ierr);
474e884886fSBarry Smith   }
475e884886fSBarry Smith   ierr = VecRestoreArray(a,&aa);CHKERRQ(ierr);
476e884886fSBarry Smith   PetscFunctionReturn(0);
477e884886fSBarry Smith }
478e884886fSBarry Smith 
479e884886fSBarry Smith #undef __FUNCT__
480c3bb7e23SBarry Smith #define __FUNCT__ "MatDiagonalScale_MFFD"
481c3bb7e23SBarry Smith PetscErrorCode MatDiagonalScale_MFFD(Mat mat,Vec ll,Vec rr)
482c3bb7e23SBarry Smith {
483c3bb7e23SBarry Smith   MatMFFD        aij = (MatMFFD)mat->data;
484c3bb7e23SBarry Smith   PetscErrorCode ierr;
485c3bb7e23SBarry Smith 
486c3bb7e23SBarry Smith   PetscFunctionBegin;
487c3bb7e23SBarry Smith   if (ll && !aij->dlscale) {
488c3bb7e23SBarry Smith     ierr = VecDuplicate(ll,&aij->dlscale);CHKERRQ(ierr);
489c3bb7e23SBarry Smith   }
490c3bb7e23SBarry Smith   if (rr && !aij->drscale) {
491c3bb7e23SBarry Smith     ierr = VecDuplicate(rr,&aij->drscale);CHKERRQ(ierr);
492c3bb7e23SBarry Smith   }
493c3bb7e23SBarry Smith   if (ll) {
494c3bb7e23SBarry Smith     ierr = VecCopy(ll,aij->dlscale);CHKERRQ(ierr);
495c3bb7e23SBarry Smith   }
496c3bb7e23SBarry Smith   if (rr) {
497c3bb7e23SBarry Smith     ierr = VecCopy(rr,aij->drscale);CHKERRQ(ierr);
498c3bb7e23SBarry Smith   }
499c3bb7e23SBarry Smith   PetscFunctionReturn(0);
500c3bb7e23SBarry Smith }
501c3bb7e23SBarry Smith 
502c3bb7e23SBarry Smith #undef __FUNCT__
503c3bb7e23SBarry Smith #define __FUNCT__ "MatDiagonalSet_MFFD"
504c3bb7e23SBarry Smith PetscErrorCode MatDiagonalSet_MFFD(Mat mat,Vec ll,InsertMode mode)
505c3bb7e23SBarry Smith {
506c3bb7e23SBarry Smith   MatMFFD        aij = (MatMFFD)mat->data;
507c3bb7e23SBarry Smith   PetscErrorCode ierr;
508c3bb7e23SBarry Smith 
509c3bb7e23SBarry Smith   PetscFunctionBegin;
510c3bb7e23SBarry Smith   if (mode == INSERT_VALUES) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_SUP,"No diagonal set with INSERT_VALUES");
511c3bb7e23SBarry Smith   if (!aij->dshift) {
512c3bb7e23SBarry Smith     ierr = VecDuplicate(ll,&aij->dshift);CHKERRQ(ierr);
513c3bb7e23SBarry Smith   }
514c3bb7e23SBarry Smith   ierr = VecAXPY(aij->dshift,1.0,ll);CHKERRQ(ierr);
515c3bb7e23SBarry Smith   PetscFunctionReturn(0);
516c3bb7e23SBarry Smith }
517c3bb7e23SBarry Smith 
518c3bb7e23SBarry Smith #undef __FUNCT__
519e884886fSBarry Smith #define __FUNCT__ "MatShift_MFFD"
520e884886fSBarry Smith PetscErrorCode MatShift_MFFD(Mat Y,PetscScalar a)
521e884886fSBarry Smith {
522e884886fSBarry Smith   MatMFFD shell = (MatMFFD)Y->data;
523*5fd66863SKarl Rupp 
524e884886fSBarry Smith   PetscFunctionBegin;
525e884886fSBarry Smith   shell->vshift += a;
526e884886fSBarry Smith   PetscFunctionReturn(0);
527e884886fSBarry Smith }
528e884886fSBarry Smith 
529e884886fSBarry Smith #undef __FUNCT__
530e884886fSBarry Smith #define __FUNCT__ "MatScale_MFFD"
531e884886fSBarry Smith PetscErrorCode MatScale_MFFD(Mat Y,PetscScalar a)
532e884886fSBarry Smith {
533e884886fSBarry Smith   MatMFFD shell = (MatMFFD)Y->data;
534*5fd66863SKarl Rupp 
535e884886fSBarry Smith   PetscFunctionBegin;
536e884886fSBarry Smith   shell->vscale *= a;
537e884886fSBarry Smith   PetscFunctionReturn(0);
538e884886fSBarry Smith }
539e884886fSBarry Smith 
540e884886fSBarry Smith EXTERN_C_BEGIN
541e884886fSBarry Smith #undef __FUNCT__
542c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetBase_MFFD"
5437087cfbeSBarry Smith PetscErrorCode  MatMFFDSetBase_MFFD(Mat J,Vec U,Vec F)
544e884886fSBarry Smith {
545e884886fSBarry Smith   PetscErrorCode ierr;
546e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)J->data;
547e884886fSBarry Smith 
548e884886fSBarry Smith   PetscFunctionBegin;
549e884886fSBarry Smith   ierr = MatMFFDResetHHistory(J);CHKERRQ(ierr);
550e884886fSBarry Smith   ctx->current_u = U;
55152121784SBarry Smith   if (F) {
5526bf464f9SBarry Smith     if (ctx->current_f_allocated) {ierr = VecDestroy(&ctx->current_f);CHKERRQ(ierr);}
5533ec795f1SBarry Smith     ctx->current_f           = F;
554cfe22f5eSBarry Smith     ctx->current_f_allocated = PETSC_FALSE;
555cfe22f5eSBarry Smith   } else if (!ctx->current_f_allocated) {
55652121784SBarry Smith     ierr = VecDuplicate(ctx->current_u, &ctx->current_f);CHKERRQ(ierr);
557cfe22f5eSBarry Smith     ctx->current_f_allocated = PETSC_TRUE;
55852121784SBarry Smith   }
559e884886fSBarry Smith   if (!ctx->w) {
560e884886fSBarry Smith     ierr = VecDuplicate(ctx->current_u, &ctx->w);CHKERRQ(ierr);
561e884886fSBarry Smith   }
562e884886fSBarry Smith   J->assembled = PETSC_TRUE;
563e884886fSBarry Smith   PetscFunctionReturn(0);
564e884886fSBarry Smith }
565e884886fSBarry Smith EXTERN_C_END
566e884886fSBarry Smith typedef PetscErrorCode (*FCN3)(void*,Vec,Vec,PetscScalar*); /* force argument to next function to not be extern C*/
567e884886fSBarry Smith EXTERN_C_BEGIN
568e884886fSBarry Smith #undef __FUNCT__
569c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetCheckh_MFFD"
5707087cfbeSBarry Smith PetscErrorCode  MatMFFDSetCheckh_MFFD(Mat J,FCN3 fun,void*ectx)
571e884886fSBarry Smith {
572e884886fSBarry Smith   MatMFFD ctx = (MatMFFD)J->data;
573e884886fSBarry Smith 
574e884886fSBarry Smith   PetscFunctionBegin;
575e884886fSBarry Smith   ctx->checkh    = fun;
576e884886fSBarry Smith   ctx->checkhctx = ectx;
577e884886fSBarry Smith   PetscFunctionReturn(0);
578e884886fSBarry Smith }
579e884886fSBarry Smith EXTERN_C_END
580e884886fSBarry Smith 
581e884886fSBarry Smith #undef __FUNCT__
5826aa9148fSLisandro Dalcin #define __FUNCT__ "MatMFFDSetOptionsPrefix"
5836aa9148fSLisandro Dalcin /*@C
5846aa9148fSLisandro Dalcin    MatMFFDSetOptionsPrefix - Sets the prefix used for searching for all
5856aa9148fSLisandro Dalcin    MatMFFD options in the database.
5866aa9148fSLisandro Dalcin 
5876aa9148fSLisandro Dalcin    Collective on Mat
5886aa9148fSLisandro Dalcin 
5896aa9148fSLisandro Dalcin    Input Parameter:
5906aa9148fSLisandro Dalcin +  A - the Mat context
5916aa9148fSLisandro Dalcin -  prefix - the prefix to prepend to all option names
5926aa9148fSLisandro Dalcin 
5936aa9148fSLisandro Dalcin    Notes:
5946aa9148fSLisandro Dalcin    A hyphen (-) must NOT be given at the beginning of the prefix name.
5956aa9148fSLisandro Dalcin    The first character of all runtime options is AUTOMATICALLY the hyphen.
5966aa9148fSLisandro Dalcin 
5976aa9148fSLisandro Dalcin    Level: advanced
5986aa9148fSLisandro Dalcin 
5996aa9148fSLisandro Dalcin .keywords: SNES, matrix-free, parameters
6006aa9148fSLisandro Dalcin 
6019c6ac3b3SBarry Smith .seealso: MatSetFromOptions(), MatCreateSNESMF()
6026aa9148fSLisandro Dalcin @*/
6037087cfbeSBarry Smith PetscErrorCode  MatMFFDSetOptionsPrefix(Mat mat,const char prefix[])
6046aa9148fSLisandro Dalcin 
6056aa9148fSLisandro Dalcin {
6061db2f0e5SJed Brown   MatMFFD        mfctx = mat ? (MatMFFD)mat->data : (MatMFFD)PETSC_NULL;
6076aa9148fSLisandro Dalcin   PetscErrorCode ierr;
608*5fd66863SKarl Rupp 
6096aa9148fSLisandro Dalcin   PetscFunctionBegin;
6100700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6110700a824SBarry Smith   PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1);
6126aa9148fSLisandro Dalcin   ierr = PetscObjectSetOptionsPrefix((PetscObject)mfctx,prefix);CHKERRQ(ierr);
6136aa9148fSLisandro Dalcin   PetscFunctionReturn(0);
6146aa9148fSLisandro Dalcin }
6156aa9148fSLisandro Dalcin 
6166aa9148fSLisandro Dalcin #undef __FUNCT__
6179c6ac3b3SBarry Smith #define __FUNCT__ "MatSetFromOptions_MFFD"
6189c6ac3b3SBarry Smith PetscErrorCode  MatSetFromOptions_MFFD(Mat mat)
619e884886fSBarry Smith {
62071f08665SBarry Smith   MatMFFD        mfctx = (MatMFFD)mat->data;
621e884886fSBarry Smith   PetscErrorCode ierr;
622ace3abfcSBarry Smith   PetscBool      flg;
623e884886fSBarry Smith   char           ftype[256];
624e884886fSBarry Smith 
625e884886fSBarry Smith   PetscFunctionBegin;
6260700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6270700a824SBarry Smith   PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1);
6283194b578SJed Brown   ierr = PetscObjectOptionsBegin((PetscObject)mfctx);CHKERRQ(ierr);
629b022a5c1SBarry Smith   ierr = PetscOptionsList("-mat_mffd_type","Matrix free type","MatMFFDSetType",MatMFFDList,((PetscObject)mfctx)->type_name,ftype,256,&flg);CHKERRQ(ierr);
630e884886fSBarry Smith   if (flg) {
631e884886fSBarry Smith     ierr = MatMFFDSetType(mat,ftype);CHKERRQ(ierr);
632e884886fSBarry Smith   }
633e884886fSBarry Smith 
634e884886fSBarry Smith   ierr = PetscOptionsReal("-mat_mffd_err","set sqrt relative error in function","MatMFFDSetFunctionError",mfctx->error_rel,&mfctx->error_rel,0);CHKERRQ(ierr);
635e884886fSBarry Smith   ierr = PetscOptionsInt("-mat_mffd_period","how often h is recomputed","MatMFFDSetPeriod",mfctx->recomputeperiod,&mfctx->recomputeperiod,0);CHKERRQ(ierr);
636e884886fSBarry Smith 
63790d69ab7SBarry Smith   flg  = PETSC_FALSE;
638acfcf0e5SJed Brown   ierr = PetscOptionsBool("-mat_mffd_check_positivity","Insure that U + h*a is nonnegative","MatMFFDSetCheckh",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
639e884886fSBarry Smith   if (flg) {
640e884886fSBarry Smith     ierr = MatMFFDSetCheckh(mat,MatMFFDCheckPositivity,0);CHKERRQ(ierr);
641e884886fSBarry Smith   }
642e884886fSBarry Smith   if (mfctx->ops->setfromoptions) {
643e884886fSBarry Smith     ierr = (*mfctx->ops->setfromoptions)(mfctx);CHKERRQ(ierr);
644e884886fSBarry Smith   }
645e884886fSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
646e884886fSBarry Smith   PetscFunctionReturn(0);
647e884886fSBarry Smith }
648e884886fSBarry Smith 
649bc0f08ceSBarry Smith EXTERN_C_BEGIN
650bc0f08ceSBarry Smith #undef __FUNCT__
651bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetPeriod_MFFD"
652bc0f08ceSBarry Smith PetscErrorCode  MatMFFDSetPeriod_MFFD(Mat mat,PetscInt period)
653bc0f08ceSBarry Smith {
654bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
655bc0f08ceSBarry Smith 
656bc0f08ceSBarry Smith   PetscFunctionBegin;
657bc0f08ceSBarry Smith   PetscValidLogicalCollectiveInt(mat,period,2);
658bc0f08ceSBarry Smith   ctx->recomputeperiod = period;
659bc0f08ceSBarry Smith   PetscFunctionReturn(0);
660bc0f08ceSBarry Smith }
661bc0f08ceSBarry Smith EXTERN_C_END
662bc0f08ceSBarry Smith 
663bc0f08ceSBarry Smith EXTERN_C_BEGIN
664bc0f08ceSBarry Smith #undef __FUNCT__
665bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetFunction_MFFD"
666bc0f08ceSBarry Smith PetscErrorCode  MatMFFDSetFunction_MFFD(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx)
667bc0f08ceSBarry Smith {
668bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
669bc0f08ceSBarry Smith 
670bc0f08ceSBarry Smith   PetscFunctionBegin;
671bc0f08ceSBarry Smith   ctx->func    = func;
672bc0f08ceSBarry Smith   ctx->funcctx = funcctx;
673bc0f08ceSBarry Smith   PetscFunctionReturn(0);
674bc0f08ceSBarry Smith }
675bc0f08ceSBarry Smith EXTERN_C_END
676bc0f08ceSBarry Smith 
677bc0f08ceSBarry Smith EXTERN_C_BEGIN
678bc0f08ceSBarry Smith #undef __FUNCT__
679bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetFunctionError_MFFD"
680bc0f08ceSBarry Smith PetscErrorCode  MatMFFDSetFunctionError_MFFD(Mat mat,PetscReal error)
681bc0f08ceSBarry Smith {
682bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
683bc0f08ceSBarry Smith 
684bc0f08ceSBarry Smith   PetscFunctionBegin;
685bc0f08ceSBarry Smith   PetscValidLogicalCollectiveReal(mat,error,2);
686bc0f08ceSBarry Smith   if (error != PETSC_DEFAULT) ctx->error_rel = error;
687bc0f08ceSBarry Smith   PetscFunctionReturn(0);
688bc0f08ceSBarry Smith }
689bc0f08ceSBarry Smith EXTERN_C_END
690bc0f08ceSBarry Smith 
691e884886fSBarry Smith /*MC
692e884886fSBarry Smith   MATMFFD - MATMFFD = "mffd" - A matrix free matrix type.
693e884886fSBarry Smith 
694e884886fSBarry Smith   Level: advanced
695e884886fSBarry Smith 
696d2d6cebeSBarry Smith .seealso: MatCreateMFFD(), MatCreateSNESMF(), MatMFFDSetFunction()
697e884886fSBarry Smith M*/
698e884886fSBarry Smith EXTERN_C_BEGIN
699e884886fSBarry Smith #undef __FUNCT__
700e884886fSBarry Smith #define __FUNCT__ "MatCreate_MFFD"
7017087cfbeSBarry Smith PetscErrorCode  MatCreate_MFFD(Mat A)
702e884886fSBarry Smith {
703e884886fSBarry Smith   MatMFFD         mfctx;
704e884886fSBarry Smith   PetscErrorCode  ierr;
705e884886fSBarry Smith 
706e884886fSBarry Smith   PetscFunctionBegin;
707519f805aSKarl Rupp #if !defined(PETSC_USE_DYNAMIC_LIBRARIES)
7083ec795f1SBarry Smith   ierr = MatMFFDInitializePackage(PETSC_NULL);CHKERRQ(ierr);
7093ec795f1SBarry Smith #endif
710e884886fSBarry Smith 
7113194b578SJed Brown   ierr = PetscHeaderCreate(mfctx,_p_MatMFFD,struct _MFOps,MATMFFD_CLASSID,0,"MatMFFD","Matrix-free Finite Differencing","Mat",((PetscObject)A)->comm,MatDestroy_MFFD,MatView_MFFD);CHKERRQ(ierr);
712e884886fSBarry Smith   mfctx->sp              = 0;
713e884886fSBarry Smith   mfctx->error_rel       = PETSC_SQRT_MACHINE_EPSILON;
714e884886fSBarry Smith   mfctx->recomputeperiod = 1;
715e884886fSBarry Smith   mfctx->count           = 0;
716e884886fSBarry Smith   mfctx->currenth        = 0.0;
717e884886fSBarry Smith   mfctx->historyh        = PETSC_NULL;
718e884886fSBarry Smith   mfctx->ncurrenth       = 0;
719e884886fSBarry Smith   mfctx->maxcurrenth     = 0;
7207adad957SLisandro Dalcin   ((PetscObject)mfctx)->type_name       = 0;
721e884886fSBarry Smith 
722e884886fSBarry Smith   mfctx->vshift          = 0.0;
723e884886fSBarry Smith   mfctx->vscale          = 1.0;
724e884886fSBarry Smith 
725e884886fSBarry Smith   /*
726e884886fSBarry Smith      Create the empty data structure to contain compute-h routines.
727e884886fSBarry Smith      These will be filled in below from the command line options or
728e884886fSBarry Smith      a later call with MatMFFDSetType() or if that is not called
729e884886fSBarry Smith      then it will default in the first use of MatMult_MFFD()
730e884886fSBarry Smith   */
731e884886fSBarry Smith   mfctx->ops->compute        = 0;
732e884886fSBarry Smith   mfctx->ops->destroy        = 0;
733e884886fSBarry Smith   mfctx->ops->view           = 0;
734e884886fSBarry Smith   mfctx->ops->setfromoptions = 0;
735e884886fSBarry Smith   mfctx->hctx                = 0;
736e884886fSBarry Smith 
737e884886fSBarry Smith   mfctx->func                = 0;
738e884886fSBarry Smith   mfctx->funcctx             = 0;
739e884886fSBarry Smith   mfctx->w                   = PETSC_NULL;
740e884886fSBarry Smith 
741e884886fSBarry Smith   A->data                = mfctx;
742e884886fSBarry Smith 
743e884886fSBarry Smith   A->ops->mult           = MatMult_MFFD;
744e884886fSBarry Smith   A->ops->destroy        = MatDestroy_MFFD;
745e884886fSBarry Smith   A->ops->view           = MatView_MFFD;
746e884886fSBarry Smith   A->ops->assemblyend    = MatAssemblyEnd_MFFD;
747e884886fSBarry Smith   A->ops->getdiagonal    = MatGetDiagonal_MFFD;
748e884886fSBarry Smith   A->ops->scale          = MatScale_MFFD;
749e884886fSBarry Smith   A->ops->shift          = MatShift_MFFD;
750c3bb7e23SBarry Smith   A->ops->diagonalscale  = MatDiagonalScale_MFFD;
751c3bb7e23SBarry Smith   A->ops->diagonalset    = MatDiagonalSet_MFFD;
7529c6ac3b3SBarry Smith   A->ops->setfromoptions = MatSetFromOptions_MFFD;
753e884886fSBarry Smith   A->assembled = PETSC_TRUE;
754e884886fSBarry Smith 
75526283091SBarry Smith   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
75626283091SBarry Smith   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
757ee1cef2cSJed Brown 
758c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetBase_C","MatMFFDSetBase_MFFD",MatMFFDSetBase_MFFD);CHKERRQ(ierr);
759c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctioniBase_C","MatMFFDSetFunctioniBase_MFFD",MatMFFDSetFunctioniBase_MFFD);CHKERRQ(ierr);
760c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctioni_C","MatMFFDSetFunctioni_MFFD",MatMFFDSetFunctioni_MFFD);CHKERRQ(ierr);
761bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunction_C","MatMFFDSetFunction_MFFD",MatMFFDSetFunction_MFFD);CHKERRQ(ierr);
762c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetCheckh_C","MatMFFDSetCheckh_MFFD",MatMFFDSetCheckh_MFFD);CHKERRQ(ierr);
763bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetPeriod_C","MatMFFDSetPeriod_MFFD",MatMFFDSetPeriod_MFFD);CHKERRQ(ierr);
764bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctionError_C","MatMFFDSetFunctionError_MFFD",MatMFFDSetFunctionError_MFFD);CHKERRQ(ierr);
765bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDResetHHistory_C","MatMFFDResetHHistory_MFFD",MatMFFDResetHHistory_MFFD);CHKERRQ(ierr);
766bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDAddNullSpace_C","MatMFFDAddNullSpace_MFFD",MatMFFDAddNullSpace_MFFD);CHKERRQ(ierr);
767e884886fSBarry Smith   mfctx->mat = A;
768e884886fSBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)A,MATMFFD);CHKERRQ(ierr);
769e884886fSBarry Smith   PetscFunctionReturn(0);
770e884886fSBarry Smith }
771e884886fSBarry Smith EXTERN_C_END
772e884886fSBarry Smith 
773e884886fSBarry Smith #undef __FUNCT__
774e884886fSBarry Smith #define __FUNCT__ "MatCreateMFFD"
775e884886fSBarry Smith /*@
776e884886fSBarry Smith    MatCreateMFFD - Creates a matrix-free matrix. See also MatCreateSNESMF()
777e884886fSBarry Smith 
778e884886fSBarry Smith    Collective on Vec
779e884886fSBarry Smith 
780e884886fSBarry Smith    Input Parameters:
781fef1beadSBarry Smith +  comm - MPI communicator
782fef1beadSBarry Smith .  m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
783fef1beadSBarry Smith            This value should be the same as the local size used in creating the
784fef1beadSBarry Smith            y vector for the matrix-vector product y = Ax.
785fef1beadSBarry Smith .  n - This value should be the same as the local size used in creating the
786fef1beadSBarry Smith        x vector for the matrix-vector product y = Ax. (or PETSC_DECIDE to have
787fef1beadSBarry Smith        calculated if N is given) For square matrices n is almost always m.
788fef1beadSBarry Smith .  M - number of global rows (or PETSC_DETERMINE to have calculated if m is given)
789fef1beadSBarry Smith -  N - number of global columns (or PETSC_DETERMINE to have calculated if n is given)
790fef1beadSBarry Smith 
791e884886fSBarry Smith 
792e884886fSBarry Smith    Output Parameter:
793e884886fSBarry Smith .  J - the matrix-free matrix
794e884886fSBarry Smith 
7959c6ac3b3SBarry Smith    Options Database Keys: call MatSetFromOptions() to trigger these
7969c6ac3b3SBarry Smith +  -mat_mffd_type - wp or ds (see MATMFFD_WP or MATMFFD_DS)
7979c6ac3b3SBarry Smith -  -mat_mffd_err - square root of estimated relative error in function evaluation
7989c6ac3b3SBarry Smith -  -mat_mffd_period - how often h is recomputed, defaults to 1, everytime
7999c6ac3b3SBarry Smith 
8009c6ac3b3SBarry Smith 
801e884886fSBarry Smith    Level: advanced
802e884886fSBarry Smith 
803e884886fSBarry Smith    Notes:
804e884886fSBarry Smith    The matrix-free matrix context merely contains the function pointers
805e884886fSBarry Smith    and work space for performing finite difference approximations of
806e884886fSBarry Smith    Jacobian-vector products, F'(u)*a,
807e884886fSBarry Smith 
808e884886fSBarry Smith    The default code uses the following approach to compute h
809e884886fSBarry Smith 
810e884886fSBarry Smith .vb
811e884886fSBarry Smith      F'(u)*a = [F(u+h*a) - F(u)]/h where
812e884886fSBarry Smith      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
813e884886fSBarry Smith        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   otherwise
814e884886fSBarry Smith  where
815e884886fSBarry Smith      error_rel = square root of relative error in function evaluation
816e884886fSBarry Smith      umin = minimum iterate parameter
817e884886fSBarry Smith .ve
818e884886fSBarry Smith 
819e884886fSBarry Smith    The user can set the error_rel via MatMFFDSetFunctionError() and
8200598bfebSBarry Smith    umin via MatMFFDDSSetUmin(); see the <A href="../../docs/manual.pdf#nameddest=ch_snes">SNES chapter of the users manual</A> for details.
821e884886fSBarry Smith 
822e884886fSBarry Smith    The user should call MatDestroy() when finished with the matrix-free
823e884886fSBarry Smith    matrix context.
824e884886fSBarry Smith 
825e884886fSBarry Smith    Options Database Keys:
826e884886fSBarry Smith +  -mat_mffd_err <error_rel> - Sets error_rel
827e884886fSBarry Smith .  -mat_mffd_unim <umin> - Sets umin (for default PETSc routine that computes h only)
828e884886fSBarry Smith -  -mat_mffd_check_positivity
829e884886fSBarry Smith 
830e884886fSBarry Smith .keywords: default, matrix-free, create, matrix
831e884886fSBarry Smith 
8321d0fab5eSBarry Smith .seealso: MatDestroy(), MatMFFDSetFunctionError(), MatMFFDDSSetUmin(), MatMFFDSetFunction()
833e884886fSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory(), MatCreateSNESMF(),
8341d0fab5eSBarry Smith           MatMFFDGetH(), MatMFFDRegisterDynamic), MatMFFDComputeJacobian()
835e884886fSBarry Smith 
836e884886fSBarry Smith @*/
8377087cfbeSBarry Smith PetscErrorCode  MatCreateMFFD(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,Mat *J)
838e884886fSBarry Smith {
839e884886fSBarry Smith   PetscErrorCode ierr;
840e884886fSBarry Smith 
841e884886fSBarry Smith   PetscFunctionBegin;
842e884886fSBarry Smith   ierr = MatCreate(comm,J);CHKERRQ(ierr);
843fef1beadSBarry Smith   ierr = MatSetSizes(*J,m,n,M,N);CHKERRQ(ierr);
844e884886fSBarry Smith   ierr = MatSetType(*J,MATMFFD);CHKERRQ(ierr);
845be7c243fSJed Brown   ierr = MatSetUp(*J);CHKERRQ(ierr);
846e884886fSBarry Smith   PetscFunctionReturn(0);
847e884886fSBarry Smith }
848e884886fSBarry Smith 
849e884886fSBarry Smith 
850e884886fSBarry Smith #undef __FUNCT__
851e884886fSBarry Smith #define __FUNCT__ "MatMFFDGetH"
852e884886fSBarry Smith /*@
853e884886fSBarry Smith    MatMFFDGetH - Gets the last value that was used as the differencing
854e884886fSBarry Smith    parameter.
855e884886fSBarry Smith 
856e884886fSBarry Smith    Not Collective
857e884886fSBarry Smith 
858e884886fSBarry Smith    Input Parameters:
859e884886fSBarry Smith .  mat - the matrix obtained with MatCreateSNESMF()
860e884886fSBarry Smith 
861e884886fSBarry Smith    Output Paramter:
862e884886fSBarry Smith .  h - the differencing step size
863e884886fSBarry Smith 
864e884886fSBarry Smith    Level: advanced
865e884886fSBarry Smith 
866e884886fSBarry Smith .keywords: SNES, matrix-free, parameters
867e884886fSBarry Smith 
8681d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDSetHHistory(), MatCreateMFFD(), MATMFFD, MatMFFDResetHHistory()
869e884886fSBarry Smith @*/
8707087cfbeSBarry Smith PetscErrorCode  MatMFFDGetH(Mat mat,PetscScalar *h)
871e884886fSBarry Smith {
872e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
873bc0f08ceSBarry Smith   PetscErrorCode ierr;
874bc0f08ceSBarry Smith   PetscBool      match;
875e884886fSBarry Smith 
876e884886fSBarry Smith   PetscFunctionBegin;
877251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)mat,MATMFFD,&match);CHKERRQ(ierr);
878bc0f08ceSBarry Smith   if (!match) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_ARG_WRONG,"Not a MFFD matrix");
879bc0f08ceSBarry Smith 
880e884886fSBarry Smith   *h = ctx->currenth;
881e884886fSBarry Smith   PetscFunctionReturn(0);
882e884886fSBarry Smith }
883e884886fSBarry Smith 
884e884886fSBarry Smith #undef __FUNCT__
885e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunction"
886e884886fSBarry Smith /*@C
887e884886fSBarry Smith    MatMFFDSetFunction - Sets the function used in applying the matrix free.
888e884886fSBarry Smith 
8893f9fe445SBarry Smith    Logically Collective on Mat
890e884886fSBarry Smith 
891e884886fSBarry Smith    Input Parameters:
892e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
893e884886fSBarry Smith .  func - the function to use
894e884886fSBarry Smith -  funcctx - optional function context passed to function
895e884886fSBarry Smith 
896e884886fSBarry Smith    Level: advanced
897e884886fSBarry Smith 
898e884886fSBarry Smith    Notes:
899e884886fSBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
900e884886fSBarry Smith     matrix inside your compute Jacobian routine
901e884886fSBarry Smith 
9023ec795f1SBarry Smith     If this is not set then it will use the function set with SNESSetFunction() if MatCreateSNESMF() was used.
903e884886fSBarry Smith 
904e884886fSBarry Smith .keywords: SNES, matrix-free, function
905e884886fSBarry Smith 
9061d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD,
9071d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
908e884886fSBarry Smith @*/
9097087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunction(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx)
910e884886fSBarry Smith {
911bc0f08ceSBarry Smith   PetscErrorCode ierr;
912e884886fSBarry Smith 
913e884886fSBarry Smith   PetscFunctionBegin;
914bc0f08ceSBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunction_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec),void*),(mat,func,funcctx));CHKERRQ(ierr);
915e884886fSBarry Smith   PetscFunctionReturn(0);
916e884886fSBarry Smith }
917e884886fSBarry Smith 
918e884886fSBarry Smith #undef __FUNCT__
919e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioni"
920e884886fSBarry Smith /*@C
921e884886fSBarry Smith    MatMFFDSetFunctioni - Sets the function for a single component
922e884886fSBarry Smith 
9233f9fe445SBarry Smith    Logically Collective on Mat
924e884886fSBarry Smith 
925e884886fSBarry Smith    Input Parameters:
926e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
927e884886fSBarry Smith -  funci - the function to use
928e884886fSBarry Smith 
929e884886fSBarry Smith    Level: advanced
930e884886fSBarry Smith 
931e884886fSBarry Smith    Notes:
932e884886fSBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
933e884886fSBarry Smith     matrix inside your compute Jacobian routine
934e884886fSBarry Smith 
935e884886fSBarry Smith 
936e884886fSBarry Smith .keywords: SNES, matrix-free, function
937e884886fSBarry Smith 
9381d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
9391d0fab5eSBarry Smith 
940e884886fSBarry Smith @*/
9417087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctioni(Mat mat,PetscErrorCode (*funci)(void*,PetscInt,Vec,PetscScalar*))
942e884886fSBarry Smith {
9434ac538c5SBarry Smith   PetscErrorCode ierr;
944e884886fSBarry Smith 
945e884886fSBarry Smith   PetscFunctionBegin;
9460700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9474ac538c5SBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunctioni_C",(Mat,PetscErrorCode (*)(void*,PetscInt,Vec,PetscScalar*)),(mat,funci));CHKERRQ(ierr);
948e884886fSBarry Smith   PetscFunctionReturn(0);
949e884886fSBarry Smith }
950e884886fSBarry Smith 
951e884886fSBarry Smith 
952e884886fSBarry Smith #undef __FUNCT__
953e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioniBase"
954e884886fSBarry Smith /*@C
955e884886fSBarry Smith    MatMFFDSetFunctioniBase - Sets the base vector for a single component function evaluation
956e884886fSBarry Smith 
9573f9fe445SBarry Smith    Logically Collective on Mat
958e884886fSBarry Smith 
959e884886fSBarry Smith    Input Parameters:
960e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
961e884886fSBarry Smith -  func - the function to use
962e884886fSBarry Smith 
963e884886fSBarry Smith    Level: advanced
964e884886fSBarry Smith 
965e884886fSBarry Smith    Notes:
966e884886fSBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
967e884886fSBarry Smith     matrix inside your compute Jacobian routine
968e884886fSBarry Smith 
969e884886fSBarry Smith 
970e884886fSBarry Smith .keywords: SNES, matrix-free, function
971e884886fSBarry Smith 
972e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD
9731d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
974e884886fSBarry Smith @*/
9757087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctioniBase(Mat mat,PetscErrorCode (*func)(void*,Vec))
976e884886fSBarry Smith {
9774ac538c5SBarry Smith   PetscErrorCode ierr;
978e884886fSBarry Smith 
979e884886fSBarry Smith   PetscFunctionBegin;
9800700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9814ac538c5SBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunctioniBase_C",(Mat,PetscErrorCode (*)(void*,Vec)),(mat,func));CHKERRQ(ierr);
982e884886fSBarry Smith   PetscFunctionReturn(0);
983e884886fSBarry Smith }
984e884886fSBarry Smith 
985e884886fSBarry Smith #undef __FUNCT__
986e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetPeriod"
987e884886fSBarry Smith /*@
988e884886fSBarry Smith    MatMFFDSetPeriod - Sets how often h is recomputed, by default it is everytime
989e884886fSBarry Smith 
9903f9fe445SBarry Smith    Logically Collective on Mat
991e884886fSBarry Smith 
992e884886fSBarry Smith    Input Parameters:
993e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
994e884886fSBarry Smith -  period - 1 for everytime, 2 for every second etc
995e884886fSBarry Smith 
996e884886fSBarry Smith    Options Database Keys:
997e884886fSBarry Smith +  -mat_mffd_period <period>
998e884886fSBarry Smith 
999e884886fSBarry Smith    Level: advanced
1000e884886fSBarry Smith 
1001e884886fSBarry Smith 
1002e884886fSBarry Smith .keywords: SNES, matrix-free, parameters
1003e884886fSBarry Smith 
1004e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(),
10051d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory()
1006e884886fSBarry Smith @*/
10077087cfbeSBarry Smith PetscErrorCode  MatMFFDSetPeriod(Mat mat,PetscInt period)
1008e884886fSBarry Smith {
1009bc0f08ceSBarry Smith   PetscErrorCode ierr;
1010e884886fSBarry Smith 
1011e884886fSBarry Smith   PetscFunctionBegin;
1012bc0f08ceSBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetPeriod_C",(Mat,PetscInt),(mat,period));CHKERRQ(ierr);
1013e884886fSBarry Smith   PetscFunctionReturn(0);
1014e884886fSBarry Smith }
1015e884886fSBarry Smith 
1016e884886fSBarry Smith #undef __FUNCT__
1017e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctionError"
1018e884886fSBarry Smith /*@
1019e884886fSBarry Smith    MatMFFDSetFunctionError - Sets the error_rel for the approximation of
1020e884886fSBarry Smith    matrix-vector products using finite differences.
1021e884886fSBarry Smith 
10223f9fe445SBarry Smith    Logically Collective on Mat
1023e884886fSBarry Smith 
1024e884886fSBarry Smith    Input Parameters:
1025e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateMFFD() or MatCreateSNESMF()
1026e884886fSBarry Smith -  error_rel - relative error (should be set to the square root of
1027e884886fSBarry Smith                the relative error in the function evaluations)
1028e884886fSBarry Smith 
1029e884886fSBarry Smith    Options Database Keys:
1030e884886fSBarry Smith +  -mat_mffd_err <error_rel> - Sets error_rel
1031e884886fSBarry Smith 
1032e884886fSBarry Smith    Level: advanced
1033e884886fSBarry Smith 
1034e884886fSBarry Smith    Notes:
1035e884886fSBarry Smith    The default matrix-free matrix-vector product routine computes
1036e884886fSBarry Smith .vb
1037e884886fSBarry Smith      F'(u)*a = [F(u+h*a) - F(u)]/h where
1038e884886fSBarry Smith      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
1039e884886fSBarry Smith        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   else
1040e884886fSBarry Smith .ve
1041e884886fSBarry Smith 
1042e884886fSBarry Smith .keywords: SNES, matrix-free, parameters
1043e884886fSBarry Smith 
1044e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD
10451d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory()
1046e884886fSBarry Smith @*/
10477087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctionError(Mat mat,PetscReal error)
1048e884886fSBarry Smith {
1049bc0f08ceSBarry Smith   PetscErrorCode ierr;
1050e884886fSBarry Smith 
1051e884886fSBarry Smith   PetscFunctionBegin;
1052bc0f08ceSBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunctionError_C",(Mat,PetscReal),(mat,error));CHKERRQ(ierr);
1053e884886fSBarry Smith   PetscFunctionReturn(0);
1054e884886fSBarry Smith }
1055e884886fSBarry Smith 
1056e884886fSBarry Smith #undef __FUNCT__
1057e884886fSBarry Smith #define __FUNCT__ "MatMFFDAddNullSpace"
1058e884886fSBarry Smith /*@
1059e884886fSBarry Smith    MatMFFDAddNullSpace - Provides a null space that an operator is
1060e884886fSBarry Smith    supposed to have.  Since roundoff will create a small component in
1061e884886fSBarry Smith    the null space, if you know the null space you may have it
1062e884886fSBarry Smith    automatically removed.
1063e884886fSBarry Smith 
10643f9fe445SBarry Smith    Logically Collective on Mat
1065e884886fSBarry Smith 
1066e884886fSBarry Smith    Input Parameters:
1067e884886fSBarry Smith +  J - the matrix-free matrix context
1068e884886fSBarry Smith -  nullsp - object created with MatNullSpaceCreate()
1069e884886fSBarry Smith 
1070e884886fSBarry Smith    Level: advanced
1071e884886fSBarry Smith 
1072e884886fSBarry Smith .keywords: SNES, matrix-free, null space
1073e884886fSBarry Smith 
1074e884886fSBarry Smith .seealso: MatNullSpaceCreate(), MatMFFDGetH(), MatCreateSNESMF(), MatCreateMFFD(), MATMFFD
10751d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory()
1076e884886fSBarry Smith @*/
10777087cfbeSBarry Smith PetscErrorCode  MatMFFDAddNullSpace(Mat J,MatNullSpace nullsp)
1078e884886fSBarry Smith {
1079e884886fSBarry Smith   PetscErrorCode ierr;
1080e884886fSBarry Smith 
1081e884886fSBarry Smith   PetscFunctionBegin;
1082bc0f08ceSBarry Smith   ierr = PetscTryMethod(J,"MatMFFDAddNullSpace_C",(Mat,MatNullSpace),(J,nullsp));CHKERRQ(ierr);
1083e884886fSBarry Smith   PetscFunctionReturn(0);
1084e884886fSBarry Smith }
1085e884886fSBarry Smith 
1086e884886fSBarry Smith #undef __FUNCT__
1087e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetHHistory"
1088e884886fSBarry Smith /*@
1089e884886fSBarry Smith    MatMFFDSetHHistory - Sets an array to collect a history of the
1090e884886fSBarry Smith    differencing values (h) computed for the matrix-free product.
1091e884886fSBarry Smith 
10923f9fe445SBarry Smith    Logically Collective on Mat
1093e884886fSBarry Smith 
1094e884886fSBarry Smith    Input Parameters:
1095e884886fSBarry Smith +  J - the matrix-free matrix context
1096e884886fSBarry Smith .  histroy - space to hold the history
1097e884886fSBarry Smith -  nhistory - number of entries in history, if more entries are generated than
1098e884886fSBarry Smith               nhistory, then the later ones are discarded
1099e884886fSBarry Smith 
1100e884886fSBarry Smith    Level: advanced
1101e884886fSBarry Smith 
1102e884886fSBarry Smith    Notes:
1103e884886fSBarry Smith    Use MatMFFDResetHHistory() to reset the history counter and collect
1104e884886fSBarry Smith    a new batch of differencing parameters, h.
1105e884886fSBarry Smith 
1106e884886fSBarry Smith .keywords: SNES, matrix-free, h history, differencing history
1107e884886fSBarry Smith 
1108e884886fSBarry Smith .seealso: MatMFFDGetH(), MatCreateSNESMF(),
11091d0fab5eSBarry Smith           MatMFFDResetHHistory(), MatMFFDSetFunctionError()
1110e884886fSBarry Smith 
1111e884886fSBarry Smith @*/
11127087cfbeSBarry Smith PetscErrorCode  MatMFFDSetHHistory(Mat J,PetscScalar history[],PetscInt nhistory)
1113e884886fSBarry Smith {
1114e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)J->data;
1115bc0f08ceSBarry Smith   PetscErrorCode ierr;
1116bc0f08ceSBarry Smith   PetscBool      match;
1117e884886fSBarry Smith 
1118e884886fSBarry Smith   PetscFunctionBegin;
1119251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)J,MATMFFD,&match);CHKERRQ(ierr);
1120bc0f08ceSBarry Smith   if (!match) SETERRQ(((PetscObject)J)->comm,PETSC_ERR_ARG_WRONG,"Not a MFFD matrix");
1121e884886fSBarry Smith   ctx->historyh    = history;
1122e884886fSBarry Smith   ctx->maxcurrenth = nhistory;
112375567043SBarry Smith   ctx->currenth    = 0.;
1124e884886fSBarry Smith   PetscFunctionReturn(0);
1125e884886fSBarry Smith }
1126e884886fSBarry Smith 
1127bc0f08ceSBarry Smith 
1128e884886fSBarry Smith #undef __FUNCT__
1129e884886fSBarry Smith #define __FUNCT__ "MatMFFDResetHHistory"
1130e884886fSBarry Smith /*@
1131e884886fSBarry Smith    MatMFFDResetHHistory - Resets the counter to zero to begin
1132e884886fSBarry Smith    collecting a new set of differencing histories.
1133e884886fSBarry Smith 
11343f9fe445SBarry Smith    Logically Collective on Mat
1135e884886fSBarry Smith 
1136e884886fSBarry Smith    Input Parameters:
1137e884886fSBarry Smith .  J - the matrix-free matrix context
1138e884886fSBarry Smith 
1139e884886fSBarry Smith    Level: advanced
1140e884886fSBarry Smith 
1141e884886fSBarry Smith    Notes:
1142e884886fSBarry Smith    Use MatMFFDSetHHistory() to create the original history counter.
1143e884886fSBarry Smith 
1144e884886fSBarry Smith .keywords: SNES, matrix-free, h history, differencing history
1145e884886fSBarry Smith 
1146e884886fSBarry Smith .seealso: MatMFFDGetH(), MatCreateSNESMF(),
11471d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDSetFunctionError()
1148e884886fSBarry Smith 
1149e884886fSBarry Smith @*/
11507087cfbeSBarry Smith PetscErrorCode  MatMFFDResetHHistory(Mat J)
1151e884886fSBarry Smith {
1152bc0f08ceSBarry Smith   PetscErrorCode ierr;
1153e884886fSBarry Smith 
1154e884886fSBarry Smith   PetscFunctionBegin;
1155bc0f08ceSBarry Smith   ierr = PetscTryMethod(J,"MatMFFDResetHHistory_C",(Mat),(J));CHKERRQ(ierr);
1156e884886fSBarry Smith   PetscFunctionReturn(0);
1157e884886fSBarry Smith }
1158e884886fSBarry Smith 
1159e884886fSBarry Smith 
1160e884886fSBarry Smith #undef __FUNCT__
1161e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetBase"
1162e884886fSBarry Smith /*@
1163e884886fSBarry Smith     MatMFFDSetBase - Sets the vector U at which matrix vector products of the
1164e884886fSBarry Smith         Jacobian are computed
1165e884886fSBarry Smith 
11663f9fe445SBarry Smith     Logically Collective on Mat
1167e884886fSBarry Smith 
1168e884886fSBarry Smith     Input Parameters:
1169e884886fSBarry Smith +   J - the MatMFFD matrix
11703ec795f1SBarry Smith .   U - the vector
1171bcddec3dSBarry Smith -   F - (optional) vector that contains F(u) if it has been already computed
1172e884886fSBarry Smith 
1173e884886fSBarry Smith     Notes: This is rarely used directly
1174e884886fSBarry Smith 
11758af5ae88SBarry Smith     If F is provided then it is not recomputed. Otherwise the function is evaluated at the base
11768af5ae88SBarry Smith     point during the first MatMult() after each call to MatMFFDSetBase().
1177dff2f722SBarry Smith 
1178e884886fSBarry Smith     Level: advanced
1179e884886fSBarry Smith 
1180e884886fSBarry Smith @*/
11817087cfbeSBarry Smith PetscErrorCode  MatMFFDSetBase(Mat J,Vec U,Vec F)
1182e884886fSBarry Smith {
11834ac538c5SBarry Smith   PetscErrorCode ierr;
1184e884886fSBarry Smith 
1185e884886fSBarry Smith   PetscFunctionBegin;
11860700a824SBarry Smith   PetscValidHeaderSpecific(J,MAT_CLASSID,1);
11870700a824SBarry Smith   PetscValidHeaderSpecific(U,VEC_CLASSID,2);
11880700a824SBarry Smith   if (F) PetscValidHeaderSpecific(F,VEC_CLASSID,3);
11894ac538c5SBarry Smith   ierr = PetscTryMethod(J,"MatMFFDSetBase_C",(Mat,Vec,Vec),(J,U,F));CHKERRQ(ierr);
1190e884886fSBarry Smith   PetscFunctionReturn(0);
1191e884886fSBarry Smith }
1192e884886fSBarry Smith 
1193e884886fSBarry Smith #undef __FUNCT__
1194e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetCheckh"
1195e884886fSBarry Smith /*@C
1196e884886fSBarry Smith     MatMFFDSetCheckh - Sets a function that checks the computed h and adjusts
1197e884886fSBarry Smith         it to satisfy some criteria
1198e884886fSBarry Smith 
11993f9fe445SBarry Smith     Logically Collective on Mat
1200e884886fSBarry Smith 
1201e884886fSBarry Smith     Input Parameters:
1202e884886fSBarry Smith +   J - the MatMFFD matrix
1203e884886fSBarry Smith .   fun - the function that checks h
1204e884886fSBarry Smith -   ctx - any context needed by the function
1205e884886fSBarry Smith 
1206e884886fSBarry Smith     Options Database Keys:
1207e884886fSBarry Smith .   -mat_mffd_check_positivity
1208e884886fSBarry Smith 
1209e884886fSBarry Smith     Level: advanced
1210e884886fSBarry Smith 
1211e884886fSBarry Smith     Notes: For example, MatMFFDSetCheckPositivity() insures that all entries
1212e884886fSBarry Smith        of U + h*a are non-negative
1213e884886fSBarry Smith 
1214e884886fSBarry Smith .seealso:  MatMFFDSetCheckPositivity()
1215e884886fSBarry Smith @*/
12167087cfbeSBarry Smith PetscErrorCode  MatMFFDSetCheckh(Mat J,PetscErrorCode (*fun)(void*,Vec,Vec,PetscScalar*),void* ctx)
1217e884886fSBarry Smith {
12184ac538c5SBarry Smith   PetscErrorCode ierr;
1219e884886fSBarry Smith 
1220e884886fSBarry Smith   PetscFunctionBegin;
12210700a824SBarry Smith   PetscValidHeaderSpecific(J,MAT_CLASSID,1);
12224ac538c5SBarry Smith   ierr = PetscTryMethod(J,"MatMFFDSetCheckh_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec,PetscScalar*),void*),(J,fun,ctx));CHKERRQ(ierr);
1223e884886fSBarry Smith   PetscFunctionReturn(0);
1224e884886fSBarry Smith }
1225e884886fSBarry Smith 
1226e884886fSBarry Smith #undef __FUNCT__
1227e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetCheckPositivity"
1228e884886fSBarry Smith /*@
1229e884886fSBarry Smith     MatMFFDCheckPositivity - Checks that all entries in U + h*a are positive or
1230e884886fSBarry Smith         zero, decreases h until this is satisfied.
1231e884886fSBarry Smith 
12323f9fe445SBarry Smith     Logically Collective on Vec
1233e884886fSBarry Smith 
1234e884886fSBarry Smith     Input Parameters:
1235e884886fSBarry Smith +   U - base vector that is added to
1236e884886fSBarry Smith .   a - vector that is added
1237e884886fSBarry Smith .   h - scaling factor on a
1238e884886fSBarry Smith -   dummy - context variable (unused)
1239e884886fSBarry Smith 
1240e884886fSBarry Smith     Options Database Keys:
1241e884886fSBarry Smith .   -mat_mffd_check_positivity
1242e884886fSBarry Smith 
1243e884886fSBarry Smith     Level: advanced
1244e884886fSBarry Smith 
1245e884886fSBarry Smith     Notes: This is rarely used directly, rather it is passed as an argument to
1246e884886fSBarry Smith            MatMFFDSetCheckh()
1247e884886fSBarry Smith 
1248e884886fSBarry Smith .seealso:  MatMFFDSetCheckh()
1249e884886fSBarry Smith @*/
12507087cfbeSBarry Smith PetscErrorCode  MatMFFDCheckPositivity(void* dummy,Vec U,Vec a,PetscScalar *h)
1251e884886fSBarry Smith {
1252e884886fSBarry Smith   PetscReal      val, minval;
1253e884886fSBarry Smith   PetscScalar    *u_vec, *a_vec;
1254e884886fSBarry Smith   PetscErrorCode ierr;
1255e884886fSBarry Smith   PetscInt       i,n;
1256e884886fSBarry Smith   MPI_Comm       comm;
1257e884886fSBarry Smith 
1258e884886fSBarry Smith   PetscFunctionBegin;
1259e884886fSBarry Smith   ierr = PetscObjectGetComm((PetscObject)U,&comm);CHKERRQ(ierr);
1260e884886fSBarry Smith   ierr = VecGetArray(U,&u_vec);CHKERRQ(ierr);
1261e884886fSBarry Smith   ierr = VecGetArray(a,&a_vec);CHKERRQ(ierr);
1262e884886fSBarry Smith   ierr = VecGetLocalSize(U,&n);CHKERRQ(ierr);
1263e884886fSBarry Smith   minval = PetscAbsScalar(*h*1.01);
1264e884886fSBarry Smith   for (i=0;i<n;i++) {
1265e884886fSBarry Smith     if (PetscRealPart(u_vec[i] + *h*a_vec[i]) <= 0.0) {
1266e884886fSBarry Smith       val = PetscAbsScalar(u_vec[i]/a_vec[i]);
1267e884886fSBarry Smith       if (val < minval) minval = val;
1268e884886fSBarry Smith     }
1269e884886fSBarry Smith   }
1270e884886fSBarry Smith   ierr = VecRestoreArray(U,&u_vec);CHKERRQ(ierr);
1271e884886fSBarry Smith   ierr = VecRestoreArray(a,&a_vec);CHKERRQ(ierr);
1272d9822059SBarry Smith   ierr = MPI_Allreduce(&minval,&val,1,MPIU_REAL,MPIU_MIN,comm);CHKERRQ(ierr);
1273e884886fSBarry Smith   if (val <= PetscAbsScalar(*h)) {
1274e884886fSBarry Smith     ierr = PetscInfo2(U,"Scaling back h from %G to %G\n",PetscRealPart(*h),.99*val);CHKERRQ(ierr);
1275e884886fSBarry Smith     if (PetscRealPart(*h) > 0.0) *h =  0.99*val;
1276e884886fSBarry Smith     else                         *h = -0.99*val;
1277e884886fSBarry Smith   }
1278e884886fSBarry Smith   PetscFunctionReturn(0);
1279e884886fSBarry Smith }
1280e884886fSBarry Smith 
1281e884886fSBarry Smith 
1282e884886fSBarry Smith 
1283e884886fSBarry Smith 
1284e884886fSBarry Smith 
1285e884886fSBarry Smith 
1286e884886fSBarry Smith 
1287