xref: /petsc/src/mat/impls/mffd/mffd.c (revision bf0cc55543cd83e035744be2f77202b216d1436e)
1e884886fSBarry Smith 
2c6db04a5SJed Brown #include <private/matimpl.h>
3c6db04a5SJed Brown #include <../src/mat/impls/mffd/mffdimpl.h>   /*I  "petscmat.h"   I*/
4e884886fSBarry Smith 
5b022a5c1SBarry Smith PetscFList 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 @*/
1077087cfbeSBarry Smith PetscErrorCode  MatMFFDSetType(Mat mat,const 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 
1179a13eae7SJed Brown   ierr = PetscTypeCompare((PetscObject)mat,MATMFFD,&match);CHKERRQ(ierr);
1189a13eae7SJed Brown   if (!match) PetscFunctionReturn(0);
1199a13eae7SJed Brown 
120e884886fSBarry Smith   /* already set, so just return */
121e884886fSBarry Smith   ierr = PetscTypeCompare((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 
1294b91b6eaSBarry Smith   ierr =  PetscFListFind(MatMFFDList,((PetscObject)ctx)->comm,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;
185e884886fSBarry Smith   ierr = PetscFListConcat(path,name,fullname);CHKERRQ(ierr);
186b022a5c1SBarry Smith   ierr = PetscFListAdd(&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;
210b022a5c1SBarry Smith   ierr = PetscFListDestroy(&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);
250*bf0cc555SLisandro 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;
275ace3abfcSBarry Smith   PetscBool      iascii;
276e884886fSBarry Smith 
277e884886fSBarry Smith   PetscFunctionBegin;
2782692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
279e884886fSBarry Smith   if (iascii) {
280e884886fSBarry Smith      ierr = PetscViewerASCIIPrintf(viewer,"  matrix-free approximation:\n");CHKERRQ(ierr);
281e884886fSBarry Smith      ierr = PetscViewerASCIIPrintf(viewer,"    err=%G (relative error in function evaluation)\n",ctx->error_rel);CHKERRQ(ierr);
2827adad957SLisandro Dalcin      if (!((PetscObject)ctx)->type_name) {
283e884886fSBarry Smith        ierr = PetscViewerASCIIPrintf(viewer,"    The compute h routine has not yet been set\n");CHKERRQ(ierr);
284e884886fSBarry Smith      } else {
2857adad957SLisandro Dalcin        ierr = PetscViewerASCIIPrintf(viewer,"    Using %s compute h routine\n",((PetscObject)ctx)->type_name);CHKERRQ(ierr);
286e884886fSBarry Smith      }
287e884886fSBarry Smith      if (ctx->ops->view) {
288e884886fSBarry Smith        ierr = (*ctx->ops->view)(ctx,viewer);CHKERRQ(ierr);
289e884886fSBarry Smith      }
290e884886fSBarry Smith   } else {
291e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for matrix-free matrix",((PetscObject)viewer)->type_name);
292e884886fSBarry Smith   }
293e884886fSBarry Smith   PetscFunctionReturn(0);
294e884886fSBarry Smith }
295e884886fSBarry Smith 
296e884886fSBarry Smith #undef __FUNCT__
297e884886fSBarry Smith #define __FUNCT__ "MatAssemblyEnd_MFFD"
298e884886fSBarry Smith /*
299e884886fSBarry Smith    MatAssemblyEnd_MFFD - Resets the ctx->ncurrenth to zero. This
300e884886fSBarry Smith    allows the user to indicate the beginning of a new linear solve by calling
301e884886fSBarry Smith    MatAssemblyXXX() on the matrix free matrix. This then allows the
3021d0fab5eSBarry Smith    MatCreateMFFD_WP() to properly compute ||U|| only the first time
303e884886fSBarry Smith    in the linear solver rather than every time.
304e884886fSBarry Smith */
305e884886fSBarry Smith PetscErrorCode MatAssemblyEnd_MFFD(Mat J,MatAssemblyType mt)
306e884886fSBarry Smith {
307e884886fSBarry Smith   PetscErrorCode ierr;
308e884886fSBarry Smith   MatMFFD        j = (MatMFFD)J->data;
309e884886fSBarry Smith 
310e884886fSBarry Smith   PetscFunctionBegin;
311e884886fSBarry Smith   ierr      = MatMFFDResetHHistory(J);CHKERRQ(ierr);
312e884886fSBarry Smith   j->vshift = 0.0;
313e884886fSBarry Smith   j->vscale = 1.0;
314e884886fSBarry Smith   PetscFunctionReturn(0);
315e884886fSBarry Smith }
316e884886fSBarry Smith 
317e884886fSBarry Smith #undef __FUNCT__
318e884886fSBarry Smith #define __FUNCT__ "MatMult_MFFD"
319e884886fSBarry Smith /*
320e884886fSBarry Smith   MatMult_MFFD - Default matrix-free form for Jacobian-vector product, y = F'(u)*a:
321e884886fSBarry Smith 
322e884886fSBarry Smith         y ~= (F(u + ha) - F(u))/h,
323e884886fSBarry Smith   where F = nonlinear function, as set by SNESSetFunction()
324e884886fSBarry Smith         u = current iterate
325e884886fSBarry Smith         h = difference interval
326e884886fSBarry Smith */
327e884886fSBarry Smith PetscErrorCode MatMult_MFFD(Mat mat,Vec a,Vec y)
328e884886fSBarry Smith {
329e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
330e884886fSBarry Smith   PetscScalar    h;
331e884886fSBarry Smith   Vec            w,U,F;
332e884886fSBarry Smith   PetscErrorCode ierr;
333ace3abfcSBarry Smith   PetscBool      zeroa;
334e884886fSBarry Smith 
335e884886fSBarry Smith   PetscFunctionBegin;
33684da1bf7SJed 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");
337e884886fSBarry Smith   /* We log matrix-free matrix-vector products separately, so that we can
338e884886fSBarry Smith      separate the performance monitoring from the cases that use conventional
339e884886fSBarry Smith      storage.  We may eventually modify event logging to associate events
340e884886fSBarry Smith      with particular objects, hence alleviating the more general problem. */
341e884886fSBarry Smith   ierr = PetscLogEventBegin(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr);
342e884886fSBarry Smith 
343e884886fSBarry Smith   w    = ctx->w;
344e884886fSBarry Smith   U    = ctx->current_u;
3453ec795f1SBarry Smith   F    = ctx->current_f;
346e884886fSBarry Smith   /*
347e884886fSBarry Smith       Compute differencing parameter
348e884886fSBarry Smith   */
349e884886fSBarry Smith   if (!ctx->ops->compute) {
350e884886fSBarry Smith     ierr = MatMFFDSetType(mat,MATMFFD_WP);CHKERRQ(ierr);
3519c6ac3b3SBarry Smith     ierr = MatSetFromOptions(mat);CHKERRQ(ierr);
352e884886fSBarry Smith   }
353e884886fSBarry Smith   ierr = (*ctx->ops->compute)(ctx,U,a,&h,&zeroa);CHKERRQ(ierr);
354e884886fSBarry Smith   if (zeroa) {
355e884886fSBarry Smith     ierr = VecSet(y,0.0);CHKERRQ(ierr);
356e884886fSBarry Smith     PetscFunctionReturn(0);
357e884886fSBarry Smith   }
358e884886fSBarry Smith 
359e32f2f54SBarry Smith   if (PetscIsInfOrNanScalar(h)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Computed Nan differencing parameter h");
360e884886fSBarry Smith   if (ctx->checkh) {
361e884886fSBarry Smith     ierr = (*ctx->checkh)(ctx->checkhctx,U,a,&h);CHKERRQ(ierr);
362e884886fSBarry Smith   }
363e884886fSBarry Smith 
364e884886fSBarry Smith   /* keep a record of the current differencing parameter h */
365e884886fSBarry Smith   ctx->currenth = h;
366e884886fSBarry Smith #if defined(PETSC_USE_COMPLEX)
367e884886fSBarry Smith   ierr = PetscInfo2(mat,"Current differencing parameter: %G + %G i\n",PetscRealPart(h),PetscImaginaryPart(h));CHKERRQ(ierr);
368e884886fSBarry Smith #else
369e884886fSBarry Smith   ierr = PetscInfo1(mat,"Current differencing parameter: %15.12e\n",h);CHKERRQ(ierr);
370e884886fSBarry Smith #endif
371e884886fSBarry Smith   if (ctx->historyh && ctx->ncurrenth < ctx->maxcurrenth) {
372e884886fSBarry Smith     ctx->historyh[ctx->ncurrenth] = h;
373e884886fSBarry Smith   }
374e884886fSBarry Smith   ctx->ncurrenth++;
375e884886fSBarry Smith 
376e884886fSBarry Smith   /* w = u + ha */
377c3bb7e23SBarry Smith   if (ctx->drscale) {
378c3bb7e23SBarry Smith     ierr = VecPointwiseMult(ctx->drscale,a,U);CHKERRQ(ierr);
379c3bb7e23SBarry Smith     ierr = VecAYPX(U,h,w);CHKERRQ(ierr);
380c3bb7e23SBarry Smith   } else {
381e884886fSBarry Smith     ierr = VecWAXPY(w,h,a,U);CHKERRQ(ierr);
382c3bb7e23SBarry Smith   }
383e884886fSBarry Smith 
384bcddec3dSBarry Smith   /* compute func(U) as base for differencing; only needed first time in and not when provided by user */
385bcddec3dSBarry Smith   if (ctx->ncurrenth == 1 && ctx->current_f_allocated) {
386e884886fSBarry Smith     ierr = (*ctx->func)(ctx->funcctx,U,F);CHKERRQ(ierr);
38752121784SBarry Smith   }
388e884886fSBarry Smith   ierr = (*ctx->func)(ctx->funcctx,w,y);CHKERRQ(ierr);
389e884886fSBarry Smith 
390e884886fSBarry Smith   ierr = VecAXPY(y,-1.0,F);CHKERRQ(ierr);
391e884886fSBarry Smith   ierr = VecScale(y,1.0/h);CHKERRQ(ierr);
392e884886fSBarry Smith 
393c35ec66cSMatthew G Knepley   if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) {
394e884886fSBarry Smith     ierr = VecAXPBY(y,ctx->vshift,ctx->vscale,a);CHKERRQ(ierr);
395c3bb7e23SBarry Smith   }
396c3bb7e23SBarry Smith   if (ctx->dlscale) {
397c3bb7e23SBarry Smith     ierr = VecPointwiseMult(y,ctx->dlscale,y);CHKERRQ(ierr);
398c3bb7e23SBarry Smith   }
399c3bb7e23SBarry Smith   if (ctx->dshift) {
400c3bb7e23SBarry Smith     ierr = VecPointwiseMult(ctx->dshift,a,U);CHKERRQ(ierr);
401c3bb7e23SBarry Smith     ierr = VecAXPY(y,1.0,U);CHKERRQ(ierr);
402c3bb7e23SBarry Smith   }
403e884886fSBarry Smith 
404e884886fSBarry Smith   if (ctx->sp) {ierr = MatNullSpaceRemove(ctx->sp,y,PETSC_NULL);CHKERRQ(ierr);}
405e884886fSBarry Smith 
406e884886fSBarry Smith   ierr = PetscLogEventEnd(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr);
407e884886fSBarry Smith   PetscFunctionReturn(0);
408e884886fSBarry Smith }
409e884886fSBarry Smith 
410e884886fSBarry Smith #undef __FUNCT__
411e884886fSBarry Smith #define __FUNCT__ "MatGetDiagonal_MFFD"
412e884886fSBarry Smith /*
413e884886fSBarry Smith   MatGetDiagonal_MFFD - Gets the diagonal for a matrix free matrix
414e884886fSBarry Smith 
415e884886fSBarry Smith         y ~= (F(u + ha) - F(u))/h,
416e884886fSBarry Smith   where F = nonlinear function, as set by SNESSetFunction()
417e884886fSBarry Smith         u = current iterate
418e884886fSBarry Smith         h = difference interval
419e884886fSBarry Smith */
420e884886fSBarry Smith PetscErrorCode MatGetDiagonal_MFFD(Mat mat,Vec a)
421e884886fSBarry Smith {
422e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
423e884886fSBarry Smith   PetscScalar    h,*aa,*ww,v;
424e884886fSBarry Smith   PetscReal      epsilon = PETSC_SQRT_MACHINE_EPSILON,umin = 100.0*PETSC_SQRT_MACHINE_EPSILON;
425e884886fSBarry Smith   Vec            w,U;
426e884886fSBarry Smith   PetscErrorCode ierr;
427e884886fSBarry Smith   PetscInt       i,rstart,rend;
428e884886fSBarry Smith 
429e884886fSBarry Smith   PetscFunctionBegin;
430e7e72b3dSBarry Smith   if (!ctx->funci) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Requires calling MatMFFDSetFunctioni() first");
431e884886fSBarry Smith 
432e884886fSBarry Smith   w    = ctx->w;
433e884886fSBarry Smith   U    = ctx->current_u;
434e884886fSBarry Smith   ierr = (*ctx->func)(ctx->funcctx,U,a);CHKERRQ(ierr);
435e884886fSBarry Smith   ierr = (*ctx->funcisetbase)(ctx->funcctx,U);CHKERRQ(ierr);
436e884886fSBarry Smith   ierr = VecCopy(U,w);CHKERRQ(ierr);
437e884886fSBarry Smith 
438e884886fSBarry Smith   ierr = VecGetOwnershipRange(a,&rstart,&rend);CHKERRQ(ierr);
439e884886fSBarry Smith   ierr = VecGetArray(a,&aa);CHKERRQ(ierr);
440e884886fSBarry Smith   for (i=rstart; i<rend; i++) {
441e884886fSBarry Smith     ierr = VecGetArray(w,&ww);CHKERRQ(ierr);
442e884886fSBarry Smith     h  = ww[i-rstart];
443e884886fSBarry Smith     if (h == 0.0) h = 1.0;
444e884886fSBarry Smith #if !defined(PETSC_USE_COMPLEX)
445e884886fSBarry Smith     if (h < umin && h >= 0.0)      h = umin;
446e884886fSBarry Smith     else if (h < 0.0 && h > -umin) h = -umin;
447e884886fSBarry Smith #else
448e884886fSBarry Smith     if (PetscAbsScalar(h) < umin && PetscRealPart(h) >= 0.0)     h = umin;
449e884886fSBarry Smith     else if (PetscRealPart(h) < 0.0 && PetscAbsScalar(h) < umin) h = -umin;
450e884886fSBarry Smith #endif
451e884886fSBarry Smith     h     *= epsilon;
452e884886fSBarry Smith 
453e884886fSBarry Smith     ww[i-rstart] += h;
454e884886fSBarry Smith     ierr = VecRestoreArray(w,&ww);CHKERRQ(ierr);
455e884886fSBarry Smith     ierr          = (*ctx->funci)(ctx->funcctx,i,w,&v);CHKERRQ(ierr);
456e884886fSBarry Smith     aa[i-rstart]  = (v - aa[i-rstart])/h;
457e884886fSBarry Smith 
458e884886fSBarry Smith     /* possibly shift and scale result */
459c35ec66cSMatthew G Knepley     if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) {
460e884886fSBarry Smith       aa[i - rstart] = ctx->vshift + ctx->vscale*aa[i-rstart];
461c3bb7e23SBarry Smith     }
462e884886fSBarry Smith 
463e884886fSBarry Smith     ierr = VecGetArray(w,&ww);CHKERRQ(ierr);
464e884886fSBarry Smith     ww[i-rstart] -= h;
465e884886fSBarry Smith     ierr = VecRestoreArray(w,&ww);CHKERRQ(ierr);
466e884886fSBarry Smith   }
467e884886fSBarry Smith   ierr = VecRestoreArray(a,&aa);CHKERRQ(ierr);
468e884886fSBarry Smith   PetscFunctionReturn(0);
469e884886fSBarry Smith }
470e884886fSBarry Smith 
471e884886fSBarry Smith #undef __FUNCT__
472c3bb7e23SBarry Smith #define __FUNCT__ "MatDiagonalScale_MFFD"
473c3bb7e23SBarry Smith PetscErrorCode MatDiagonalScale_MFFD(Mat mat,Vec ll,Vec rr)
474c3bb7e23SBarry Smith {
475c3bb7e23SBarry Smith   MatMFFD        aij = (MatMFFD)mat->data;
476c3bb7e23SBarry Smith   PetscErrorCode ierr;
477c3bb7e23SBarry Smith 
478c3bb7e23SBarry Smith   PetscFunctionBegin;
479c3bb7e23SBarry Smith   if (ll && !aij->dlscale) {
480c3bb7e23SBarry Smith     ierr = VecDuplicate(ll,&aij->dlscale);CHKERRQ(ierr);
481c3bb7e23SBarry Smith   }
482c3bb7e23SBarry Smith   if (rr && !aij->drscale) {
483c3bb7e23SBarry Smith     ierr = VecDuplicate(rr,&aij->drscale);CHKERRQ(ierr);
484c3bb7e23SBarry Smith   }
485c3bb7e23SBarry Smith   if (ll) {
486c3bb7e23SBarry Smith     ierr = VecCopy(ll,aij->dlscale);CHKERRQ(ierr);
487c3bb7e23SBarry Smith   }
488c3bb7e23SBarry Smith   if (rr) {
489c3bb7e23SBarry Smith     ierr = VecCopy(rr,aij->drscale);CHKERRQ(ierr);
490c3bb7e23SBarry Smith   }
491c3bb7e23SBarry Smith   PetscFunctionReturn(0);
492c3bb7e23SBarry Smith }
493c3bb7e23SBarry Smith 
494c3bb7e23SBarry Smith #undef __FUNCT__
495c3bb7e23SBarry Smith #define __FUNCT__ "MatDiagonalSet_MFFD"
496c3bb7e23SBarry Smith PetscErrorCode MatDiagonalSet_MFFD(Mat mat,Vec ll,InsertMode mode)
497c3bb7e23SBarry Smith {
498c3bb7e23SBarry Smith   MatMFFD        aij = (MatMFFD)mat->data;
499c3bb7e23SBarry Smith   PetscErrorCode ierr;
500c3bb7e23SBarry Smith 
501c3bb7e23SBarry Smith   PetscFunctionBegin;
502c3bb7e23SBarry Smith   if (mode == INSERT_VALUES) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_SUP,"No diagonal set with INSERT_VALUES");
503c3bb7e23SBarry Smith   if (!aij->dshift) {
504c3bb7e23SBarry Smith     ierr = VecDuplicate(ll,&aij->dshift);CHKERRQ(ierr);
505c3bb7e23SBarry Smith   }
506c3bb7e23SBarry Smith   ierr = VecAXPY(aij->dshift,1.0,ll);CHKERRQ(ierr);
507c3bb7e23SBarry Smith   PetscFunctionReturn(0);
508c3bb7e23SBarry Smith }
509c3bb7e23SBarry Smith 
510c3bb7e23SBarry Smith #undef __FUNCT__
511e884886fSBarry Smith #define __FUNCT__ "MatShift_MFFD"
512e884886fSBarry Smith PetscErrorCode MatShift_MFFD(Mat Y,PetscScalar a)
513e884886fSBarry Smith {
514e884886fSBarry Smith   MatMFFD shell = (MatMFFD)Y->data;
515e884886fSBarry Smith   PetscFunctionBegin;
516e884886fSBarry Smith   shell->vshift += a;
517e884886fSBarry Smith   PetscFunctionReturn(0);
518e884886fSBarry Smith }
519e884886fSBarry Smith 
520e884886fSBarry Smith #undef __FUNCT__
521e884886fSBarry Smith #define __FUNCT__ "MatScale_MFFD"
522e884886fSBarry Smith PetscErrorCode MatScale_MFFD(Mat Y,PetscScalar a)
523e884886fSBarry Smith {
524e884886fSBarry Smith   MatMFFD shell = (MatMFFD)Y->data;
525e884886fSBarry Smith   PetscFunctionBegin;
526e884886fSBarry Smith   shell->vscale *= a;
527e884886fSBarry Smith   PetscFunctionReturn(0);
528e884886fSBarry Smith }
529e884886fSBarry Smith 
530e884886fSBarry Smith EXTERN_C_BEGIN
531e884886fSBarry Smith #undef __FUNCT__
532c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetBase_MFFD"
5337087cfbeSBarry Smith PetscErrorCode  MatMFFDSetBase_MFFD(Mat J,Vec U,Vec F)
534e884886fSBarry Smith {
535e884886fSBarry Smith   PetscErrorCode ierr;
536e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)J->data;
537e884886fSBarry Smith 
538e884886fSBarry Smith   PetscFunctionBegin;
539e884886fSBarry Smith   ierr = MatMFFDResetHHistory(J);CHKERRQ(ierr);
540e884886fSBarry Smith   ctx->current_u = U;
54152121784SBarry Smith   if (F) {
5426bf464f9SBarry Smith     if (ctx->current_f_allocated) {ierr = VecDestroy(&ctx->current_f);CHKERRQ(ierr);}
5433ec795f1SBarry Smith     ctx->current_f           = F;
544cfe22f5eSBarry Smith     ctx->current_f_allocated = PETSC_FALSE;
545cfe22f5eSBarry Smith   } else if (!ctx->current_f_allocated) {
54652121784SBarry Smith     ierr = VecDuplicate(ctx->current_u, &ctx->current_f);CHKERRQ(ierr);
547cfe22f5eSBarry Smith     ctx->current_f_allocated = PETSC_TRUE;
54852121784SBarry Smith   }
549e884886fSBarry Smith   if (!ctx->w) {
550e884886fSBarry Smith     ierr = VecDuplicate(ctx->current_u, &ctx->w);CHKERRQ(ierr);
551e884886fSBarry Smith   }
552e884886fSBarry Smith   J->assembled = PETSC_TRUE;
553e884886fSBarry Smith   PetscFunctionReturn(0);
554e884886fSBarry Smith }
555e884886fSBarry Smith EXTERN_C_END
556e884886fSBarry Smith typedef PetscErrorCode (*FCN3)(void*,Vec,Vec,PetscScalar*); /* force argument to next function to not be extern C*/
557e884886fSBarry Smith EXTERN_C_BEGIN
558e884886fSBarry Smith #undef __FUNCT__
559c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetCheckh_MFFD"
5607087cfbeSBarry Smith PetscErrorCode  MatMFFDSetCheckh_MFFD(Mat J,FCN3 fun,void*ectx)
561e884886fSBarry Smith {
562e884886fSBarry Smith   MatMFFD ctx = (MatMFFD)J->data;
563e884886fSBarry Smith 
564e884886fSBarry Smith   PetscFunctionBegin;
565e884886fSBarry Smith   ctx->checkh    = fun;
566e884886fSBarry Smith   ctx->checkhctx = ectx;
567e884886fSBarry Smith   PetscFunctionReturn(0);
568e884886fSBarry Smith }
569e884886fSBarry Smith EXTERN_C_END
570e884886fSBarry Smith 
571e884886fSBarry Smith #undef __FUNCT__
5726aa9148fSLisandro Dalcin #define __FUNCT__ "MatMFFDSetOptionsPrefix"
5736aa9148fSLisandro Dalcin /*@C
5746aa9148fSLisandro Dalcin    MatMFFDSetOptionsPrefix - Sets the prefix used for searching for all
5756aa9148fSLisandro Dalcin    MatMFFD options in the database.
5766aa9148fSLisandro Dalcin 
5776aa9148fSLisandro Dalcin    Collective on Mat
5786aa9148fSLisandro Dalcin 
5796aa9148fSLisandro Dalcin    Input Parameter:
5806aa9148fSLisandro Dalcin +  A - the Mat context
5816aa9148fSLisandro Dalcin -  prefix - the prefix to prepend to all option names
5826aa9148fSLisandro Dalcin 
5836aa9148fSLisandro Dalcin    Notes:
5846aa9148fSLisandro Dalcin    A hyphen (-) must NOT be given at the beginning of the prefix name.
5856aa9148fSLisandro Dalcin    The first character of all runtime options is AUTOMATICALLY the hyphen.
5866aa9148fSLisandro Dalcin 
5876aa9148fSLisandro Dalcin    Level: advanced
5886aa9148fSLisandro Dalcin 
5896aa9148fSLisandro Dalcin .keywords: SNES, matrix-free, parameters
5906aa9148fSLisandro Dalcin 
5919c6ac3b3SBarry Smith .seealso: MatSetFromOptions(), MatCreateSNESMF()
5926aa9148fSLisandro Dalcin @*/
5937087cfbeSBarry Smith PetscErrorCode  MatMFFDSetOptionsPrefix(Mat mat,const char prefix[])
5946aa9148fSLisandro Dalcin 
5956aa9148fSLisandro Dalcin {
5966aa9148fSLisandro Dalcin   MatMFFD        mfctx = mat ? (MatMFFD)mat->data : PETSC_NULL;
5976aa9148fSLisandro Dalcin   PetscErrorCode ierr;
5986aa9148fSLisandro Dalcin   PetscFunctionBegin;
5990700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6000700a824SBarry Smith   PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1);
6016aa9148fSLisandro Dalcin   ierr = PetscObjectSetOptionsPrefix((PetscObject)mfctx,prefix);CHKERRQ(ierr);
6026aa9148fSLisandro Dalcin   PetscFunctionReturn(0);
6036aa9148fSLisandro Dalcin }
6046aa9148fSLisandro Dalcin 
6056aa9148fSLisandro Dalcin #undef __FUNCT__
6069c6ac3b3SBarry Smith #define __FUNCT__ "MatSetFromOptions_MFFD"
6079c6ac3b3SBarry Smith PetscErrorCode  MatSetFromOptions_MFFD(Mat mat)
608e884886fSBarry Smith {
60971f08665SBarry Smith   MatMFFD        mfctx = (MatMFFD)mat->data;
610e884886fSBarry Smith   PetscErrorCode ierr;
611ace3abfcSBarry Smith   PetscBool      flg;
612e884886fSBarry Smith   char           ftype[256];
613e884886fSBarry Smith 
614e884886fSBarry Smith   PetscFunctionBegin;
6150700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6160700a824SBarry Smith   PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1);
6177adad957SLisandro Dalcin   ierr = PetscOptionsBegin(((PetscObject)mfctx)->comm,((PetscObject)mfctx)->prefix,"Set matrix free computation parameters","MatMFFD");CHKERRQ(ierr);
618b022a5c1SBarry Smith   ierr = PetscOptionsList("-mat_mffd_type","Matrix free type","MatMFFDSetType",MatMFFDList,((PetscObject)mfctx)->type_name,ftype,256,&flg);CHKERRQ(ierr);
619e884886fSBarry Smith   if (flg) {
620e884886fSBarry Smith     ierr = MatMFFDSetType(mat,ftype);CHKERRQ(ierr);
621e884886fSBarry Smith   }
622e884886fSBarry Smith 
623e884886fSBarry Smith   ierr = PetscOptionsReal("-mat_mffd_err","set sqrt relative error in function","MatMFFDSetFunctionError",mfctx->error_rel,&mfctx->error_rel,0);CHKERRQ(ierr);
624e884886fSBarry Smith   ierr = PetscOptionsInt("-mat_mffd_period","how often h is recomputed","MatMFFDSetPeriod",mfctx->recomputeperiod,&mfctx->recomputeperiod,0);CHKERRQ(ierr);
625e884886fSBarry Smith 
62690d69ab7SBarry Smith   flg  = PETSC_FALSE;
627acfcf0e5SJed Brown   ierr = PetscOptionsBool("-mat_mffd_check_positivity","Insure that U + h*a is nonnegative","MatMFFDSetCheckh",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
628e884886fSBarry Smith   if (flg) {
629e884886fSBarry Smith     ierr = MatMFFDSetCheckh(mat,MatMFFDCheckPositivity,0);CHKERRQ(ierr);
630e884886fSBarry Smith   }
631e884886fSBarry Smith   if (mfctx->ops->setfromoptions) {
632e884886fSBarry Smith     ierr = (*mfctx->ops->setfromoptions)(mfctx);CHKERRQ(ierr);
633e884886fSBarry Smith   }
634e884886fSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
635e884886fSBarry Smith   PetscFunctionReturn(0);
636e884886fSBarry Smith }
637e884886fSBarry Smith 
638bc0f08ceSBarry Smith EXTERN_C_BEGIN
639bc0f08ceSBarry Smith #undef __FUNCT__
640bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetPeriod_MFFD"
641bc0f08ceSBarry Smith PetscErrorCode  MatMFFDSetPeriod_MFFD(Mat mat,PetscInt period)
642bc0f08ceSBarry Smith {
643bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
644bc0f08ceSBarry Smith 
645bc0f08ceSBarry Smith   PetscFunctionBegin;
646bc0f08ceSBarry Smith   PetscValidLogicalCollectiveInt(mat,period,2);
647bc0f08ceSBarry Smith   ctx->recomputeperiod = period;
648bc0f08ceSBarry Smith   PetscFunctionReturn(0);
649bc0f08ceSBarry Smith }
650bc0f08ceSBarry Smith EXTERN_C_END
651bc0f08ceSBarry Smith 
652bc0f08ceSBarry Smith EXTERN_C_BEGIN
653bc0f08ceSBarry Smith #undef __FUNCT__
654bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetFunction_MFFD"
655bc0f08ceSBarry Smith PetscErrorCode  MatMFFDSetFunction_MFFD(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx)
656bc0f08ceSBarry Smith {
657bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
658bc0f08ceSBarry Smith 
659bc0f08ceSBarry Smith   PetscFunctionBegin;
660bc0f08ceSBarry Smith   ctx->func    = func;
661bc0f08ceSBarry Smith   ctx->funcctx = funcctx;
662bc0f08ceSBarry Smith   PetscFunctionReturn(0);
663bc0f08ceSBarry Smith }
664bc0f08ceSBarry Smith EXTERN_C_END
665bc0f08ceSBarry Smith 
666bc0f08ceSBarry Smith EXTERN_C_BEGIN
667bc0f08ceSBarry Smith #undef __FUNCT__
668bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetFunctionError_MFFD"
669bc0f08ceSBarry Smith PetscErrorCode  MatMFFDSetFunctionError_MFFD(Mat mat,PetscReal error)
670bc0f08ceSBarry Smith {
671bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
672bc0f08ceSBarry Smith 
673bc0f08ceSBarry Smith   PetscFunctionBegin;
674bc0f08ceSBarry Smith   PetscValidLogicalCollectiveReal(mat,error,2);
675bc0f08ceSBarry Smith   if (error != PETSC_DEFAULT) ctx->error_rel = error;
676bc0f08ceSBarry Smith   PetscFunctionReturn(0);
677bc0f08ceSBarry Smith }
678bc0f08ceSBarry Smith EXTERN_C_END
679bc0f08ceSBarry Smith 
680e884886fSBarry Smith /*MC
681e884886fSBarry Smith   MATMFFD - MATMFFD = "mffd" - A matrix free matrix type.
682e884886fSBarry Smith 
683e884886fSBarry Smith   Level: advanced
684e884886fSBarry Smith 
685d2d6cebeSBarry Smith .seealso: MatCreateMFFD(), MatCreateSNESMF(), MatMFFDSetFunction()
686e884886fSBarry Smith M*/
687e884886fSBarry Smith EXTERN_C_BEGIN
688e884886fSBarry Smith #undef __FUNCT__
689e884886fSBarry Smith #define __FUNCT__ "MatCreate_MFFD"
6907087cfbeSBarry Smith PetscErrorCode  MatCreate_MFFD(Mat A)
691e884886fSBarry Smith {
692e884886fSBarry Smith   MatMFFD         mfctx;
693e884886fSBarry Smith   PetscErrorCode  ierr;
694e884886fSBarry Smith 
695e884886fSBarry Smith   PetscFunctionBegin;
6963ec795f1SBarry Smith #ifndef PETSC_USE_DYNAMIC_LIBRARIES
6973ec795f1SBarry Smith   ierr = MatMFFDInitializePackage(PETSC_NULL);CHKERRQ(ierr);
6983ec795f1SBarry Smith #endif
699e884886fSBarry Smith 
7000700a824SBarry Smith   ierr = PetscHeaderCreate(mfctx,_p_MatMFFD,struct _MFOps,MATMFFD_CLASSID,0,"MatMFFD",((PetscObject)A)->comm,MatDestroy_MFFD,MatView_MFFD);CHKERRQ(ierr);
701e884886fSBarry Smith   mfctx->sp              = 0;
702e884886fSBarry Smith   mfctx->error_rel       = PETSC_SQRT_MACHINE_EPSILON;
703e884886fSBarry Smith   mfctx->recomputeperiod = 1;
704e884886fSBarry Smith   mfctx->count           = 0;
705e884886fSBarry Smith   mfctx->currenth        = 0.0;
706e884886fSBarry Smith   mfctx->historyh        = PETSC_NULL;
707e884886fSBarry Smith   mfctx->ncurrenth       = 0;
708e884886fSBarry Smith   mfctx->maxcurrenth     = 0;
7097adad957SLisandro Dalcin   ((PetscObject)mfctx)->type_name       = 0;
710e884886fSBarry Smith 
711e884886fSBarry Smith   mfctx->vshift          = 0.0;
712e884886fSBarry Smith   mfctx->vscale          = 1.0;
713e884886fSBarry Smith 
714e884886fSBarry Smith   /*
715e884886fSBarry Smith      Create the empty data structure to contain compute-h routines.
716e884886fSBarry Smith      These will be filled in below from the command line options or
717e884886fSBarry Smith      a later call with MatMFFDSetType() or if that is not called
718e884886fSBarry Smith      then it will default in the first use of MatMult_MFFD()
719e884886fSBarry Smith   */
720e884886fSBarry Smith   mfctx->ops->compute        = 0;
721e884886fSBarry Smith   mfctx->ops->destroy        = 0;
722e884886fSBarry Smith   mfctx->ops->view           = 0;
723e884886fSBarry Smith   mfctx->ops->setfromoptions = 0;
724e884886fSBarry Smith   mfctx->hctx                = 0;
725e884886fSBarry Smith 
726e884886fSBarry Smith   mfctx->func                = 0;
727e884886fSBarry Smith   mfctx->funcctx             = 0;
728e884886fSBarry Smith   mfctx->w                   = PETSC_NULL;
729e884886fSBarry Smith 
730e884886fSBarry Smith   A->data                = mfctx;
731e884886fSBarry Smith 
732e884886fSBarry Smith   A->ops->mult           = MatMult_MFFD;
733e884886fSBarry Smith   A->ops->destroy        = MatDestroy_MFFD;
734e884886fSBarry Smith   A->ops->view           = MatView_MFFD;
735e884886fSBarry Smith   A->ops->assemblyend    = MatAssemblyEnd_MFFD;
736e884886fSBarry Smith   A->ops->getdiagonal    = MatGetDiagonal_MFFD;
737e884886fSBarry Smith   A->ops->scale          = MatScale_MFFD;
738e884886fSBarry Smith   A->ops->shift          = MatShift_MFFD;
739c3bb7e23SBarry Smith   A->ops->diagonalscale  = MatDiagonalScale_MFFD;
740c3bb7e23SBarry Smith   A->ops->diagonalset    = MatDiagonalSet_MFFD;
7419c6ac3b3SBarry Smith   A->ops->setfromoptions = MatSetFromOptions_MFFD;
742e884886fSBarry Smith   A->assembled = PETSC_TRUE;
743e884886fSBarry Smith 
74426283091SBarry Smith   ierr = PetscLayoutSetBlockSize(A->rmap,1);CHKERRQ(ierr);
74526283091SBarry Smith   ierr = PetscLayoutSetBlockSize(A->cmap,1);CHKERRQ(ierr);
74626283091SBarry Smith   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
74726283091SBarry Smith   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
748ee1cef2cSJed Brown 
749c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetBase_C","MatMFFDSetBase_MFFD",MatMFFDSetBase_MFFD);CHKERRQ(ierr);
750c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctioniBase_C","MatMFFDSetFunctioniBase_MFFD",MatMFFDSetFunctioniBase_MFFD);CHKERRQ(ierr);
751c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctioni_C","MatMFFDSetFunctioni_MFFD",MatMFFDSetFunctioni_MFFD);CHKERRQ(ierr);
752bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunction_C","MatMFFDSetFunction_MFFD",MatMFFDSetFunction_MFFD);CHKERRQ(ierr);
753c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetCheckh_C","MatMFFDSetCheckh_MFFD",MatMFFDSetCheckh_MFFD);CHKERRQ(ierr);
754bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetPeriod_C","MatMFFDSetPeriod_MFFD",MatMFFDSetPeriod_MFFD);CHKERRQ(ierr);
755bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctionError_C","MatMFFDSetFunctionError_MFFD",MatMFFDSetFunctionError_MFFD);CHKERRQ(ierr);
756bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDResetHHistory_C","MatMFFDResetHHistory_MFFD",MatMFFDResetHHistory_MFFD);CHKERRQ(ierr);
757bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDAddNullSpace_C","MatMFFDAddNullSpace_MFFD",MatMFFDAddNullSpace_MFFD);CHKERRQ(ierr);
758e884886fSBarry Smith   mfctx->mat = A;
759e884886fSBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)A,MATMFFD);CHKERRQ(ierr);
760e884886fSBarry Smith   PetscFunctionReturn(0);
761e884886fSBarry Smith }
762e884886fSBarry Smith EXTERN_C_END
763e884886fSBarry Smith 
764e884886fSBarry Smith #undef __FUNCT__
765e884886fSBarry Smith #define __FUNCT__ "MatCreateMFFD"
766e884886fSBarry Smith /*@
767e884886fSBarry Smith    MatCreateMFFD - Creates a matrix-free matrix. See also MatCreateSNESMF()
768e884886fSBarry Smith 
769e884886fSBarry Smith    Collective on Vec
770e884886fSBarry Smith 
771e884886fSBarry Smith    Input Parameters:
772fef1beadSBarry Smith +  comm - MPI communicator
773fef1beadSBarry Smith .  m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
774fef1beadSBarry Smith            This value should be the same as the local size used in creating the
775fef1beadSBarry Smith            y vector for the matrix-vector product y = Ax.
776fef1beadSBarry Smith .  n - This value should be the same as the local size used in creating the
777fef1beadSBarry Smith        x vector for the matrix-vector product y = Ax. (or PETSC_DECIDE to have
778fef1beadSBarry Smith        calculated if N is given) For square matrices n is almost always m.
779fef1beadSBarry Smith .  M - number of global rows (or PETSC_DETERMINE to have calculated if m is given)
780fef1beadSBarry Smith -  N - number of global columns (or PETSC_DETERMINE to have calculated if n is given)
781fef1beadSBarry Smith 
782e884886fSBarry Smith 
783e884886fSBarry Smith    Output Parameter:
784e884886fSBarry Smith .  J - the matrix-free matrix
785e884886fSBarry Smith 
7869c6ac3b3SBarry Smith    Options Database Keys: call MatSetFromOptions() to trigger these
7879c6ac3b3SBarry Smith +  -mat_mffd_type - wp or ds (see MATMFFD_WP or MATMFFD_DS)
7889c6ac3b3SBarry Smith -  -mat_mffd_err - square root of estimated relative error in function evaluation
7899c6ac3b3SBarry Smith -  -mat_mffd_period - how often h is recomputed, defaults to 1, everytime
7909c6ac3b3SBarry Smith 
7919c6ac3b3SBarry Smith 
792e884886fSBarry Smith    Level: advanced
793e884886fSBarry Smith 
794e884886fSBarry Smith    Notes:
795e884886fSBarry Smith    The matrix-free matrix context merely contains the function pointers
796e884886fSBarry Smith    and work space for performing finite difference approximations of
797e884886fSBarry Smith    Jacobian-vector products, F'(u)*a,
798e884886fSBarry Smith 
799e884886fSBarry Smith    The default code uses the following approach to compute h
800e884886fSBarry Smith 
801e884886fSBarry Smith .vb
802e884886fSBarry Smith      F'(u)*a = [F(u+h*a) - F(u)]/h where
803e884886fSBarry Smith      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
804e884886fSBarry Smith        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   otherwise
805e884886fSBarry Smith  where
806e884886fSBarry Smith      error_rel = square root of relative error in function evaluation
807e884886fSBarry Smith      umin = minimum iterate parameter
808e884886fSBarry Smith .ve
809e884886fSBarry Smith 
810e884886fSBarry Smith    The user can set the error_rel via MatMFFDSetFunctionError() and
8110598bfebSBarry Smith    umin via MatMFFDDSSetUmin(); see the <A href="../../docs/manual.pdf#nameddest=ch_snes">SNES chapter of the users manual</A> for details.
812e884886fSBarry Smith 
813e884886fSBarry Smith    The user should call MatDestroy() when finished with the matrix-free
814e884886fSBarry Smith    matrix context.
815e884886fSBarry Smith 
816e884886fSBarry Smith    Options Database Keys:
817e884886fSBarry Smith +  -mat_mffd_err <error_rel> - Sets error_rel
818e884886fSBarry Smith .  -mat_mffd_unim <umin> - Sets umin (for default PETSc routine that computes h only)
819e884886fSBarry Smith -  -mat_mffd_check_positivity
820e884886fSBarry Smith 
821e884886fSBarry Smith .keywords: default, matrix-free, create, matrix
822e884886fSBarry Smith 
8231d0fab5eSBarry Smith .seealso: MatDestroy(), MatMFFDSetFunctionError(), MatMFFDDSSetUmin(), MatMFFDSetFunction()
824e884886fSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory(), MatCreateSNESMF(),
8251d0fab5eSBarry Smith           MatMFFDGetH(), MatMFFDRegisterDynamic), MatMFFDComputeJacobian()
826e884886fSBarry Smith 
827e884886fSBarry Smith @*/
8287087cfbeSBarry Smith PetscErrorCode  MatCreateMFFD(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,Mat *J)
829e884886fSBarry Smith {
830e884886fSBarry Smith   PetscErrorCode ierr;
831e884886fSBarry Smith 
832e884886fSBarry Smith   PetscFunctionBegin;
833e884886fSBarry Smith   ierr = MatCreate(comm,J);CHKERRQ(ierr);
834fef1beadSBarry Smith   ierr = MatSetSizes(*J,m,n,M,N);CHKERRQ(ierr);
835e884886fSBarry Smith   ierr = MatSetType(*J,MATMFFD);CHKERRQ(ierr);
836e884886fSBarry Smith   PetscFunctionReturn(0);
837e884886fSBarry Smith }
838e884886fSBarry Smith 
839e884886fSBarry Smith 
840e884886fSBarry Smith #undef __FUNCT__
841e884886fSBarry Smith #define __FUNCT__ "MatMFFDGetH"
842e884886fSBarry Smith /*@
843e884886fSBarry Smith    MatMFFDGetH - Gets the last value that was used as the differencing
844e884886fSBarry Smith    parameter.
845e884886fSBarry Smith 
846e884886fSBarry Smith    Not Collective
847e884886fSBarry Smith 
848e884886fSBarry Smith    Input Parameters:
849e884886fSBarry Smith .  mat - the matrix obtained with MatCreateSNESMF()
850e884886fSBarry Smith 
851e884886fSBarry Smith    Output Paramter:
852e884886fSBarry Smith .  h - the differencing step size
853e884886fSBarry Smith 
854e884886fSBarry Smith    Level: advanced
855e884886fSBarry Smith 
856e884886fSBarry Smith .keywords: SNES, matrix-free, parameters
857e884886fSBarry Smith 
8581d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDSetHHistory(), MatCreateMFFD(), MATMFFD, MatMFFDResetHHistory()
859e884886fSBarry Smith @*/
8607087cfbeSBarry Smith PetscErrorCode  MatMFFDGetH(Mat mat,PetscScalar *h)
861e884886fSBarry Smith {
862e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
863bc0f08ceSBarry Smith   PetscErrorCode ierr;
864bc0f08ceSBarry Smith   PetscBool      match;
865e884886fSBarry Smith 
866e884886fSBarry Smith   PetscFunctionBegin;
867bc0f08ceSBarry Smith   ierr = PetscTypeCompare((PetscObject)mat,MATMFFD,&match);CHKERRQ(ierr);
868bc0f08ceSBarry Smith   if (!match) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_ARG_WRONG,"Not a MFFD matrix");
869bc0f08ceSBarry Smith 
870e884886fSBarry Smith   *h = ctx->currenth;
871e884886fSBarry Smith   PetscFunctionReturn(0);
872e884886fSBarry Smith }
873e884886fSBarry Smith 
874e884886fSBarry Smith #undef __FUNCT__
875e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunction"
876e884886fSBarry Smith /*@C
877e884886fSBarry Smith    MatMFFDSetFunction - Sets the function used in applying the matrix free.
878e884886fSBarry Smith 
8793f9fe445SBarry Smith    Logically Collective on Mat
880e884886fSBarry Smith 
881e884886fSBarry Smith    Input Parameters:
882e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
883e884886fSBarry Smith .  func - the function to use
884e884886fSBarry Smith -  funcctx - optional function context passed to function
885e884886fSBarry Smith 
886e884886fSBarry Smith    Level: advanced
887e884886fSBarry Smith 
888e884886fSBarry Smith    Notes:
889e884886fSBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
890e884886fSBarry Smith     matrix inside your compute Jacobian routine
891e884886fSBarry Smith 
8923ec795f1SBarry Smith     If this is not set then it will use the function set with SNESSetFunction() if MatCreateSNESMF() was used.
893e884886fSBarry Smith 
894e884886fSBarry Smith .keywords: SNES, matrix-free, function
895e884886fSBarry Smith 
8961d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD,
8971d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
898e884886fSBarry Smith @*/
8997087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunction(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx)
900e884886fSBarry Smith {
901bc0f08ceSBarry Smith   PetscErrorCode ierr;
902e884886fSBarry Smith 
903e884886fSBarry Smith   PetscFunctionBegin;
904bc0f08ceSBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunction_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec),void*),(mat,func,funcctx));CHKERRQ(ierr);
905e884886fSBarry Smith   PetscFunctionReturn(0);
906e884886fSBarry Smith }
907e884886fSBarry Smith 
908e884886fSBarry Smith #undef __FUNCT__
909e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioni"
910e884886fSBarry Smith /*@C
911e884886fSBarry Smith    MatMFFDSetFunctioni - Sets the function for a single component
912e884886fSBarry Smith 
9133f9fe445SBarry Smith    Logically Collective on Mat
914e884886fSBarry Smith 
915e884886fSBarry Smith    Input Parameters:
916e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
917e884886fSBarry Smith -  funci - the function to use
918e884886fSBarry Smith 
919e884886fSBarry Smith    Level: advanced
920e884886fSBarry Smith 
921e884886fSBarry Smith    Notes:
922e884886fSBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
923e884886fSBarry Smith     matrix inside your compute Jacobian routine
924e884886fSBarry Smith 
925e884886fSBarry Smith 
926e884886fSBarry Smith .keywords: SNES, matrix-free, function
927e884886fSBarry Smith 
9281d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
9291d0fab5eSBarry Smith 
930e884886fSBarry Smith @*/
9317087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctioni(Mat mat,PetscErrorCode (*funci)(void*,PetscInt,Vec,PetscScalar*))
932e884886fSBarry Smith {
9334ac538c5SBarry Smith   PetscErrorCode ierr;
934e884886fSBarry Smith 
935e884886fSBarry Smith   PetscFunctionBegin;
9360700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9374ac538c5SBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunctioni_C",(Mat,PetscErrorCode (*)(void*,PetscInt,Vec,PetscScalar*)),(mat,funci));CHKERRQ(ierr);
938e884886fSBarry Smith   PetscFunctionReturn(0);
939e884886fSBarry Smith }
940e884886fSBarry Smith 
941e884886fSBarry Smith 
942e884886fSBarry Smith #undef __FUNCT__
943e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioniBase"
944e884886fSBarry Smith /*@C
945e884886fSBarry Smith    MatMFFDSetFunctioniBase - Sets the base vector for a single component function evaluation
946e884886fSBarry Smith 
9473f9fe445SBarry Smith    Logically Collective on Mat
948e884886fSBarry Smith 
949e884886fSBarry Smith    Input Parameters:
950e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
951e884886fSBarry Smith -  func - the function to use
952e884886fSBarry Smith 
953e884886fSBarry Smith    Level: advanced
954e884886fSBarry Smith 
955e884886fSBarry Smith    Notes:
956e884886fSBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
957e884886fSBarry Smith     matrix inside your compute Jacobian routine
958e884886fSBarry Smith 
959e884886fSBarry Smith 
960e884886fSBarry Smith .keywords: SNES, matrix-free, function
961e884886fSBarry Smith 
962e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD
9631d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
964e884886fSBarry Smith @*/
9657087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctioniBase(Mat mat,PetscErrorCode (*func)(void*,Vec))
966e884886fSBarry Smith {
9674ac538c5SBarry Smith   PetscErrorCode ierr;
968e884886fSBarry Smith 
969e884886fSBarry Smith   PetscFunctionBegin;
9700700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9714ac538c5SBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunctioniBase_C",(Mat,PetscErrorCode (*)(void*,Vec)),(mat,func));CHKERRQ(ierr);
972e884886fSBarry Smith   PetscFunctionReturn(0);
973e884886fSBarry Smith }
974e884886fSBarry Smith 
975e884886fSBarry Smith #undef __FUNCT__
976e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetPeriod"
977e884886fSBarry Smith /*@
978e884886fSBarry Smith    MatMFFDSetPeriod - Sets how often h is recomputed, by default it is everytime
979e884886fSBarry Smith 
9803f9fe445SBarry Smith    Logically Collective on Mat
981e884886fSBarry Smith 
982e884886fSBarry Smith    Input Parameters:
983e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
984e884886fSBarry Smith -  period - 1 for everytime, 2 for every second etc
985e884886fSBarry Smith 
986e884886fSBarry Smith    Options Database Keys:
987e884886fSBarry Smith +  -mat_mffd_period <period>
988e884886fSBarry Smith 
989e884886fSBarry Smith    Level: advanced
990e884886fSBarry Smith 
991e884886fSBarry Smith 
992e884886fSBarry Smith .keywords: SNES, matrix-free, parameters
993e884886fSBarry Smith 
994e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(),
9951d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory()
996e884886fSBarry Smith @*/
9977087cfbeSBarry Smith PetscErrorCode  MatMFFDSetPeriod(Mat mat,PetscInt period)
998e884886fSBarry Smith {
999bc0f08ceSBarry Smith   PetscErrorCode ierr;
1000e884886fSBarry Smith 
1001e884886fSBarry Smith   PetscFunctionBegin;
1002bc0f08ceSBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetPeriod_C",(Mat,PetscInt),(mat,period));CHKERRQ(ierr);
1003e884886fSBarry Smith   PetscFunctionReturn(0);
1004e884886fSBarry Smith }
1005e884886fSBarry Smith 
1006e884886fSBarry Smith #undef __FUNCT__
1007e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctionError"
1008e884886fSBarry Smith /*@
1009e884886fSBarry Smith    MatMFFDSetFunctionError - Sets the error_rel for the approximation of
1010e884886fSBarry Smith    matrix-vector products using finite differences.
1011e884886fSBarry Smith 
10123f9fe445SBarry Smith    Logically Collective on Mat
1013e884886fSBarry Smith 
1014e884886fSBarry Smith    Input Parameters:
1015e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateMFFD() or MatCreateSNESMF()
1016e884886fSBarry Smith -  error_rel - relative error (should be set to the square root of
1017e884886fSBarry Smith                the relative error in the function evaluations)
1018e884886fSBarry Smith 
1019e884886fSBarry Smith    Options Database Keys:
1020e884886fSBarry Smith +  -mat_mffd_err <error_rel> - Sets error_rel
1021e884886fSBarry Smith 
1022e884886fSBarry Smith    Level: advanced
1023e884886fSBarry Smith 
1024e884886fSBarry Smith    Notes:
1025e884886fSBarry Smith    The default matrix-free matrix-vector product routine computes
1026e884886fSBarry Smith .vb
1027e884886fSBarry Smith      F'(u)*a = [F(u+h*a) - F(u)]/h where
1028e884886fSBarry Smith      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
1029e884886fSBarry Smith        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   else
1030e884886fSBarry Smith .ve
1031e884886fSBarry Smith 
1032e884886fSBarry Smith .keywords: SNES, matrix-free, parameters
1033e884886fSBarry Smith 
1034e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD
10351d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory()
1036e884886fSBarry Smith @*/
10377087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctionError(Mat mat,PetscReal error)
1038e884886fSBarry Smith {
1039bc0f08ceSBarry Smith   PetscErrorCode ierr;
1040e884886fSBarry Smith 
1041e884886fSBarry Smith   PetscFunctionBegin;
1042bc0f08ceSBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunctionError_C",(Mat,PetscReal),(mat,error));CHKERRQ(ierr);
1043e884886fSBarry Smith   PetscFunctionReturn(0);
1044e884886fSBarry Smith }
1045e884886fSBarry Smith 
1046e884886fSBarry Smith #undef __FUNCT__
1047e884886fSBarry Smith #define __FUNCT__ "MatMFFDAddNullSpace"
1048e884886fSBarry Smith /*@
1049e884886fSBarry Smith    MatMFFDAddNullSpace - Provides a null space that an operator is
1050e884886fSBarry Smith    supposed to have.  Since roundoff will create a small component in
1051e884886fSBarry Smith    the null space, if you know the null space you may have it
1052e884886fSBarry Smith    automatically removed.
1053e884886fSBarry Smith 
10543f9fe445SBarry Smith    Logically Collective on Mat
1055e884886fSBarry Smith 
1056e884886fSBarry Smith    Input Parameters:
1057e884886fSBarry Smith +  J - the matrix-free matrix context
1058e884886fSBarry Smith -  nullsp - object created with MatNullSpaceCreate()
1059e884886fSBarry Smith 
1060e884886fSBarry Smith    Level: advanced
1061e884886fSBarry Smith 
1062e884886fSBarry Smith .keywords: SNES, matrix-free, null space
1063e884886fSBarry Smith 
1064e884886fSBarry Smith .seealso: MatNullSpaceCreate(), MatMFFDGetH(), MatCreateSNESMF(), MatCreateMFFD(), MATMFFD
10651d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory()
1066e884886fSBarry Smith @*/
10677087cfbeSBarry Smith PetscErrorCode  MatMFFDAddNullSpace(Mat J,MatNullSpace nullsp)
1068e884886fSBarry Smith {
1069e884886fSBarry Smith   PetscErrorCode ierr;
1070e884886fSBarry Smith 
1071e884886fSBarry Smith   PetscFunctionBegin;
1072bc0f08ceSBarry Smith   ierr = PetscTryMethod(J,"MatMFFDAddNullSpace_C",(Mat,MatNullSpace),(J,nullsp));CHKERRQ(ierr);
1073e884886fSBarry Smith   PetscFunctionReturn(0);
1074e884886fSBarry Smith }
1075e884886fSBarry Smith 
1076e884886fSBarry Smith #undef __FUNCT__
1077e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetHHistory"
1078e884886fSBarry Smith /*@
1079e884886fSBarry Smith    MatMFFDSetHHistory - Sets an array to collect a history of the
1080e884886fSBarry Smith    differencing values (h) computed for the matrix-free product.
1081e884886fSBarry Smith 
10823f9fe445SBarry Smith    Logically Collective on Mat
1083e884886fSBarry Smith 
1084e884886fSBarry Smith    Input Parameters:
1085e884886fSBarry Smith +  J - the matrix-free matrix context
1086e884886fSBarry Smith .  histroy - space to hold the history
1087e884886fSBarry Smith -  nhistory - number of entries in history, if more entries are generated than
1088e884886fSBarry Smith               nhistory, then the later ones are discarded
1089e884886fSBarry Smith 
1090e884886fSBarry Smith    Level: advanced
1091e884886fSBarry Smith 
1092e884886fSBarry Smith    Notes:
1093e884886fSBarry Smith    Use MatMFFDResetHHistory() to reset the history counter and collect
1094e884886fSBarry Smith    a new batch of differencing parameters, h.
1095e884886fSBarry Smith 
1096e884886fSBarry Smith .keywords: SNES, matrix-free, h history, differencing history
1097e884886fSBarry Smith 
1098e884886fSBarry Smith .seealso: MatMFFDGetH(), MatCreateSNESMF(),
10991d0fab5eSBarry Smith           MatMFFDResetHHistory(), MatMFFDSetFunctionError()
1100e884886fSBarry Smith 
1101e884886fSBarry Smith @*/
11027087cfbeSBarry Smith PetscErrorCode  MatMFFDSetHHistory(Mat J,PetscScalar history[],PetscInt nhistory)
1103e884886fSBarry Smith {
1104e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)J->data;
1105bc0f08ceSBarry Smith   PetscErrorCode ierr;
1106bc0f08ceSBarry Smith   PetscBool      match;
1107e884886fSBarry Smith 
1108e884886fSBarry Smith   PetscFunctionBegin;
1109bc0f08ceSBarry Smith   ierr = PetscTypeCompare((PetscObject)J,MATMFFD,&match);CHKERRQ(ierr);
1110bc0f08ceSBarry Smith   if (!match) SETERRQ(((PetscObject)J)->comm,PETSC_ERR_ARG_WRONG,"Not a MFFD matrix");
1111e884886fSBarry Smith   ctx->historyh    = history;
1112e884886fSBarry Smith   ctx->maxcurrenth = nhistory;
111375567043SBarry Smith   ctx->currenth    = 0.;
1114e884886fSBarry Smith   PetscFunctionReturn(0);
1115e884886fSBarry Smith }
1116e884886fSBarry Smith 
1117bc0f08ceSBarry Smith 
1118e884886fSBarry Smith #undef __FUNCT__
1119e884886fSBarry Smith #define __FUNCT__ "MatMFFDResetHHistory"
1120e884886fSBarry Smith /*@
1121e884886fSBarry Smith    MatMFFDResetHHistory - Resets the counter to zero to begin
1122e884886fSBarry Smith    collecting a new set of differencing histories.
1123e884886fSBarry Smith 
11243f9fe445SBarry Smith    Logically Collective on Mat
1125e884886fSBarry Smith 
1126e884886fSBarry Smith    Input Parameters:
1127e884886fSBarry Smith .  J - the matrix-free matrix context
1128e884886fSBarry Smith 
1129e884886fSBarry Smith    Level: advanced
1130e884886fSBarry Smith 
1131e884886fSBarry Smith    Notes:
1132e884886fSBarry Smith    Use MatMFFDSetHHistory() to create the original history counter.
1133e884886fSBarry Smith 
1134e884886fSBarry Smith .keywords: SNES, matrix-free, h history, differencing history
1135e884886fSBarry Smith 
1136e884886fSBarry Smith .seealso: MatMFFDGetH(), MatCreateSNESMF(),
11371d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDSetFunctionError()
1138e884886fSBarry Smith 
1139e884886fSBarry Smith @*/
11407087cfbeSBarry Smith PetscErrorCode  MatMFFDResetHHistory(Mat J)
1141e884886fSBarry Smith {
1142bc0f08ceSBarry Smith   PetscErrorCode ierr;
1143e884886fSBarry Smith 
1144e884886fSBarry Smith   PetscFunctionBegin;
1145bc0f08ceSBarry Smith   ierr = PetscTryMethod(J,"MatMFFDResetHHistory_C",(Mat),(J));CHKERRQ(ierr);
1146e884886fSBarry Smith   PetscFunctionReturn(0);
1147e884886fSBarry Smith }
1148e884886fSBarry Smith 
1149e884886fSBarry Smith 
1150e884886fSBarry Smith #undef __FUNCT__
1151e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetBase"
1152e884886fSBarry Smith /*@
1153e884886fSBarry Smith     MatMFFDSetBase - Sets the vector U at which matrix vector products of the
1154e884886fSBarry Smith         Jacobian are computed
1155e884886fSBarry Smith 
11563f9fe445SBarry Smith     Logically Collective on Mat
1157e884886fSBarry Smith 
1158e884886fSBarry Smith     Input Parameters:
1159e884886fSBarry Smith +   J - the MatMFFD matrix
11603ec795f1SBarry Smith .   U - the vector
1161bcddec3dSBarry Smith -   F - (optional) vector that contains F(u) if it has been already computed
1162e884886fSBarry Smith 
1163e884886fSBarry Smith     Notes: This is rarely used directly
1164e884886fSBarry Smith 
11658af5ae88SBarry Smith     If F is provided then it is not recomputed. Otherwise the function is evaluated at the base
11668af5ae88SBarry Smith     point during the first MatMult() after each call to MatMFFDSetBase().
1167dff2f722SBarry Smith 
1168e884886fSBarry Smith     Level: advanced
1169e884886fSBarry Smith 
1170e884886fSBarry Smith @*/
11717087cfbeSBarry Smith PetscErrorCode  MatMFFDSetBase(Mat J,Vec U,Vec F)
1172e884886fSBarry Smith {
11734ac538c5SBarry Smith   PetscErrorCode ierr;
1174e884886fSBarry Smith 
1175e884886fSBarry Smith   PetscFunctionBegin;
11760700a824SBarry Smith   PetscValidHeaderSpecific(J,MAT_CLASSID,1);
11770700a824SBarry Smith   PetscValidHeaderSpecific(U,VEC_CLASSID,2);
11780700a824SBarry Smith   if (F) PetscValidHeaderSpecific(F,VEC_CLASSID,3);
11794ac538c5SBarry Smith   ierr = PetscTryMethod(J,"MatMFFDSetBase_C",(Mat,Vec,Vec),(J,U,F));CHKERRQ(ierr);
1180e884886fSBarry Smith   PetscFunctionReturn(0);
1181e884886fSBarry Smith }
1182e884886fSBarry Smith 
1183e884886fSBarry Smith #undef __FUNCT__
1184e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetCheckh"
1185e884886fSBarry Smith /*@C
1186e884886fSBarry Smith     MatMFFDSetCheckh - Sets a function that checks the computed h and adjusts
1187e884886fSBarry Smith         it to satisfy some criteria
1188e884886fSBarry Smith 
11893f9fe445SBarry Smith     Logically Collective on Mat
1190e884886fSBarry Smith 
1191e884886fSBarry Smith     Input Parameters:
1192e884886fSBarry Smith +   J - the MatMFFD matrix
1193e884886fSBarry Smith .   fun - the function that checks h
1194e884886fSBarry Smith -   ctx - any context needed by the function
1195e884886fSBarry Smith 
1196e884886fSBarry Smith     Options Database Keys:
1197e884886fSBarry Smith .   -mat_mffd_check_positivity
1198e884886fSBarry Smith 
1199e884886fSBarry Smith     Level: advanced
1200e884886fSBarry Smith 
1201e884886fSBarry Smith     Notes: For example, MatMFFDSetCheckPositivity() insures that all entries
1202e884886fSBarry Smith        of U + h*a are non-negative
1203e884886fSBarry Smith 
1204e884886fSBarry Smith .seealso:  MatMFFDSetCheckPositivity()
1205e884886fSBarry Smith @*/
12067087cfbeSBarry Smith PetscErrorCode  MatMFFDSetCheckh(Mat J,PetscErrorCode (*fun)(void*,Vec,Vec,PetscScalar*),void* ctx)
1207e884886fSBarry Smith {
12084ac538c5SBarry Smith   PetscErrorCode ierr;
1209e884886fSBarry Smith 
1210e884886fSBarry Smith   PetscFunctionBegin;
12110700a824SBarry Smith   PetscValidHeaderSpecific(J,MAT_CLASSID,1);
12124ac538c5SBarry Smith   ierr = PetscTryMethod(J,"MatMFFDSetCheckh_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec,PetscScalar*),void*),(J,fun,ctx));CHKERRQ(ierr);
1213e884886fSBarry Smith   PetscFunctionReturn(0);
1214e884886fSBarry Smith }
1215e884886fSBarry Smith 
1216e884886fSBarry Smith #undef __FUNCT__
1217e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetCheckPositivity"
1218e884886fSBarry Smith /*@
1219e884886fSBarry Smith     MatMFFDCheckPositivity - Checks that all entries in U + h*a are positive or
1220e884886fSBarry Smith         zero, decreases h until this is satisfied.
1221e884886fSBarry Smith 
12223f9fe445SBarry Smith     Logically Collective on Vec
1223e884886fSBarry Smith 
1224e884886fSBarry Smith     Input Parameters:
1225e884886fSBarry Smith +   U - base vector that is added to
1226e884886fSBarry Smith .   a - vector that is added
1227e884886fSBarry Smith .   h - scaling factor on a
1228e884886fSBarry Smith -   dummy - context variable (unused)
1229e884886fSBarry Smith 
1230e884886fSBarry Smith     Options Database Keys:
1231e884886fSBarry Smith .   -mat_mffd_check_positivity
1232e884886fSBarry Smith 
1233e884886fSBarry Smith     Level: advanced
1234e884886fSBarry Smith 
1235e884886fSBarry Smith     Notes: This is rarely used directly, rather it is passed as an argument to
1236e884886fSBarry Smith            MatMFFDSetCheckh()
1237e884886fSBarry Smith 
1238e884886fSBarry Smith .seealso:  MatMFFDSetCheckh()
1239e884886fSBarry Smith @*/
12407087cfbeSBarry Smith PetscErrorCode  MatMFFDCheckPositivity(void* dummy,Vec U,Vec a,PetscScalar *h)
1241e884886fSBarry Smith {
1242e884886fSBarry Smith   PetscReal      val, minval;
1243e884886fSBarry Smith   PetscScalar    *u_vec, *a_vec;
1244e884886fSBarry Smith   PetscErrorCode ierr;
1245e884886fSBarry Smith   PetscInt       i,n;
1246e884886fSBarry Smith   MPI_Comm       comm;
1247e884886fSBarry Smith 
1248e884886fSBarry Smith   PetscFunctionBegin;
1249e884886fSBarry Smith   ierr = PetscObjectGetComm((PetscObject)U,&comm);CHKERRQ(ierr);
1250e884886fSBarry Smith   ierr = VecGetArray(U,&u_vec);CHKERRQ(ierr);
1251e884886fSBarry Smith   ierr = VecGetArray(a,&a_vec);CHKERRQ(ierr);
1252e884886fSBarry Smith   ierr = VecGetLocalSize(U,&n);CHKERRQ(ierr);
1253e884886fSBarry Smith   minval = PetscAbsScalar(*h*1.01);
1254e884886fSBarry Smith   for(i=0;i<n;i++) {
1255e884886fSBarry Smith     if (PetscRealPart(u_vec[i] + *h*a_vec[i]) <= 0.0) {
1256e884886fSBarry Smith       val = PetscAbsScalar(u_vec[i]/a_vec[i]);
1257e884886fSBarry Smith       if (val < minval) minval = val;
1258e884886fSBarry Smith     }
1259e884886fSBarry Smith   }
1260e884886fSBarry Smith   ierr = VecRestoreArray(U,&u_vec);CHKERRQ(ierr);
1261e884886fSBarry Smith   ierr = VecRestoreArray(a,&a_vec);CHKERRQ(ierr);
1262d9822059SBarry Smith   ierr = MPI_Allreduce(&minval,&val,1,MPIU_REAL,MPIU_MIN,comm);CHKERRQ(ierr);
1263e884886fSBarry Smith   if (val <= PetscAbsScalar(*h)) {
1264e884886fSBarry Smith     ierr = PetscInfo2(U,"Scaling back h from %G to %G\n",PetscRealPart(*h),.99*val);CHKERRQ(ierr);
1265e884886fSBarry Smith     if (PetscRealPart(*h) > 0.0) *h =  0.99*val;
1266e884886fSBarry Smith     else                         *h = -0.99*val;
1267e884886fSBarry Smith   }
1268e884886fSBarry Smith   PetscFunctionReturn(0);
1269e884886fSBarry Smith }
1270e884886fSBarry Smith 
1271e884886fSBarry Smith 
1272e884886fSBarry Smith 
1273e884886fSBarry Smith 
1274e884886fSBarry Smith 
1275e884886fSBarry Smith 
1276e884886fSBarry Smith 
1277