xref: /petsc/src/mat/impls/mffd/mffd.c (revision bc0f08cebdae986a221953ddc93e818de7426e9c)
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 
129b022a5c1SBarry Smith   ierr =  PetscFListFind(MatMFFDList,((PetscObject)ctx)->comm,ftype,(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 
164*bc0f08ceSBarry Smith EXTERN_C_BEGIN
165*bc0f08ceSBarry Smith #undef __FUNCT__
166*bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDResetHHistory_MFFD"
167*bc0f08ceSBarry Smith PetscErrorCode  MatMFFDResetHHistory_MFFD(Mat J)
168*bc0f08ceSBarry Smith {
169*bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)J->data;
170*bc0f08ceSBarry Smith 
171*bc0f08ceSBarry Smith   PetscFunctionBegin;
172*bc0f08ceSBarry Smith   ctx->ncurrenth    = 0;
173*bc0f08ceSBarry Smith   PetscFunctionReturn(0);
174*bc0f08ceSBarry Smith }
175*bc0f08ceSBarry 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 
215*bc0f08ceSBarry Smith EXTERN_C_BEGIN
216*bc0f08ceSBarry Smith #undef __FUNCT__
217*bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDAddNullSpace_MFFD"
218*bc0f08ceSBarry Smith PetscErrorCode  MatMFFDAddNullSpace_MFFD(Mat J,MatNullSpace nullsp)
219*bc0f08ceSBarry Smith {
220*bc0f08ceSBarry Smith   PetscErrorCode ierr;
221*bc0f08ceSBarry Smith   MatMFFD      ctx = (MatMFFD)J->data;
222*bc0f08ceSBarry Smith 
223*bc0f08ceSBarry Smith   PetscFunctionBegin;
224*bc0f08ceSBarry Smith   ierr = PetscObjectReference((PetscObject)nullsp);CHKERRQ(ierr);
225*bc0f08ceSBarry Smith   if (ctx->sp) { ierr = MatNullSpaceDestroy(&ctx->sp);CHKERRQ(ierr); }
226*bc0f08ceSBarry Smith   ctx->sp = nullsp;
227*bc0f08ceSBarry Smith   PetscFunctionReturn(0);
228*bc0f08ceSBarry Smith }
229*bc0f08ceSBarry Smith EXTERN_C_END
230*bc0f08ceSBarry 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;
240e884886fSBarry Smith   if (ctx->w) {
241e884886fSBarry Smith     ierr = VecDestroy(ctx->w);CHKERRQ(ierr);
242e884886fSBarry Smith   }
243c3bb7e23SBarry Smith   if (ctx->drscale) {
244c3bb7e23SBarry Smith     ierr = VecDestroy(ctx->drscale);CHKERRQ(ierr);
245c3bb7e23SBarry Smith   }
246c3bb7e23SBarry Smith   if (ctx->dlscale) {
247c3bb7e23SBarry Smith     ierr = VecDestroy(ctx->dlscale);CHKERRQ(ierr);
248c3bb7e23SBarry Smith   }
249c3bb7e23SBarry Smith   if (ctx->dshift) {
250c3bb7e23SBarry Smith     ierr = VecDestroy(ctx->dshift);CHKERRQ(ierr);
251c3bb7e23SBarry Smith   }
252cfe22f5eSBarry Smith   if (ctx->current_f_allocated) {
253cfe22f5eSBarry Smith     ierr = VecDestroy(ctx->current_f);
254cfe22f5eSBarry Smith   }
255e884886fSBarry Smith   if (ctx->ops->destroy) {ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr);}
256d34fcf5fSBarry Smith   if (ctx->sp) {ierr = MatNullSpaceDestroy(&ctx->sp);CHKERRQ(ierr);}
257e884886fSBarry Smith   ierr = PetscHeaderDestroy(ctx);CHKERRQ(ierr);
258e884886fSBarry Smith 
259e884886fSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetBase_C","",PETSC_NULL);CHKERRQ(ierr);
260e884886fSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctioniBase_C","",PETSC_NULL);CHKERRQ(ierr);
261e884886fSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctioni_C","",PETSC_NULL);CHKERRQ(ierr);
262*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunction_C","",PETSC_NULL);CHKERRQ(ierr);
263*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctionError_C","",PETSC_NULL);CHKERRQ(ierr);
264e884886fSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetCheckh_C","",PETSC_NULL);CHKERRQ(ierr);
265*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetPeriod_C","",PETSC_NULL);CHKERRQ(ierr);
266*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDResetHHistory_C","",PETSC_NULL);CHKERRQ(ierr);
267*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDAddNullSpace_C","",PETSC_NULL);CHKERRQ(ierr);
268e884886fSBarry Smith 
269e884886fSBarry Smith   PetscFunctionReturn(0);
270e884886fSBarry Smith }
271e884886fSBarry Smith 
272e884886fSBarry Smith #undef __FUNCT__
273e884886fSBarry Smith #define __FUNCT__ "MatView_MFFD"
274e884886fSBarry Smith /*
275e884886fSBarry Smith    MatMFFDView_MFFD - Views matrix-free parameters.
276e884886fSBarry Smith 
277e884886fSBarry Smith */
278e884886fSBarry Smith PetscErrorCode MatView_MFFD(Mat J,PetscViewer viewer)
279e884886fSBarry Smith {
280e884886fSBarry Smith   PetscErrorCode ierr;
281e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)J->data;
282ace3abfcSBarry Smith   PetscBool      iascii;
283e884886fSBarry Smith 
284e884886fSBarry Smith   PetscFunctionBegin;
2852692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
286e884886fSBarry Smith   if (iascii) {
287e884886fSBarry Smith      ierr = PetscViewerASCIIPrintf(viewer,"  matrix-free approximation:\n");CHKERRQ(ierr);
288e884886fSBarry Smith      ierr = PetscViewerASCIIPrintf(viewer,"    err=%G (relative error in function evaluation)\n",ctx->error_rel);CHKERRQ(ierr);
2897adad957SLisandro Dalcin      if (!((PetscObject)ctx)->type_name) {
290e884886fSBarry Smith        ierr = PetscViewerASCIIPrintf(viewer,"    The compute h routine has not yet been set\n");CHKERRQ(ierr);
291e884886fSBarry Smith      } else {
2927adad957SLisandro Dalcin        ierr = PetscViewerASCIIPrintf(viewer,"    Using %s compute h routine\n",((PetscObject)ctx)->type_name);CHKERRQ(ierr);
293e884886fSBarry Smith      }
294e884886fSBarry Smith      if (ctx->ops->view) {
295e884886fSBarry Smith        ierr = (*ctx->ops->view)(ctx,viewer);CHKERRQ(ierr);
296e884886fSBarry Smith      }
297e884886fSBarry Smith   } else {
298e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for matrix-free matrix",((PetscObject)viewer)->type_name);
299e884886fSBarry Smith   }
300e884886fSBarry Smith   PetscFunctionReturn(0);
301e884886fSBarry Smith }
302e884886fSBarry Smith 
303e884886fSBarry Smith #undef __FUNCT__
304e884886fSBarry Smith #define __FUNCT__ "MatAssemblyEnd_MFFD"
305e884886fSBarry Smith /*
306e884886fSBarry Smith    MatAssemblyEnd_MFFD - Resets the ctx->ncurrenth to zero. This
307e884886fSBarry Smith    allows the user to indicate the beginning of a new linear solve by calling
308e884886fSBarry Smith    MatAssemblyXXX() on the matrix free matrix. This then allows the
3091d0fab5eSBarry Smith    MatCreateMFFD_WP() to properly compute ||U|| only the first time
310e884886fSBarry Smith    in the linear solver rather than every time.
311e884886fSBarry Smith */
312e884886fSBarry Smith PetscErrorCode MatAssemblyEnd_MFFD(Mat J,MatAssemblyType mt)
313e884886fSBarry Smith {
314e884886fSBarry Smith   PetscErrorCode ierr;
315e884886fSBarry Smith   MatMFFD        j = (MatMFFD)J->data;
316e884886fSBarry Smith 
317e884886fSBarry Smith   PetscFunctionBegin;
318e884886fSBarry Smith   ierr      = MatMFFDResetHHistory(J);CHKERRQ(ierr);
319e884886fSBarry Smith   j->vshift = 0.0;
320e884886fSBarry Smith   j->vscale = 1.0;
321e884886fSBarry Smith   PetscFunctionReturn(0);
322e884886fSBarry Smith }
323e884886fSBarry Smith 
324e884886fSBarry Smith #undef __FUNCT__
325e884886fSBarry Smith #define __FUNCT__ "MatMult_MFFD"
326e884886fSBarry Smith /*
327e884886fSBarry Smith   MatMult_MFFD - Default matrix-free form for Jacobian-vector product, y = F'(u)*a:
328e884886fSBarry Smith 
329e884886fSBarry Smith         y ~= (F(u + ha) - F(u))/h,
330e884886fSBarry Smith   where F = nonlinear function, as set by SNESSetFunction()
331e884886fSBarry Smith         u = current iterate
332e884886fSBarry Smith         h = difference interval
333e884886fSBarry Smith */
334e884886fSBarry Smith PetscErrorCode MatMult_MFFD(Mat mat,Vec a,Vec y)
335e884886fSBarry Smith {
336e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
337e884886fSBarry Smith   PetscScalar    h;
338e884886fSBarry Smith   Vec            w,U,F;
339e884886fSBarry Smith   PetscErrorCode ierr;
340ace3abfcSBarry Smith   PetscBool      zeroa;
341e884886fSBarry Smith 
342e884886fSBarry Smith   PetscFunctionBegin;
34384da1bf7SJed 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");
344e884886fSBarry Smith   /* We log matrix-free matrix-vector products separately, so that we can
345e884886fSBarry Smith      separate the performance monitoring from the cases that use conventional
346e884886fSBarry Smith      storage.  We may eventually modify event logging to associate events
347e884886fSBarry Smith      with particular objects, hence alleviating the more general problem. */
348e884886fSBarry Smith   ierr = PetscLogEventBegin(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr);
349e884886fSBarry Smith 
350e884886fSBarry Smith   w    = ctx->w;
351e884886fSBarry Smith   U    = ctx->current_u;
3523ec795f1SBarry Smith   F    = ctx->current_f;
353e884886fSBarry Smith   /*
354e884886fSBarry Smith       Compute differencing parameter
355e884886fSBarry Smith   */
356e884886fSBarry Smith   if (!ctx->ops->compute) {
357e884886fSBarry Smith     ierr = MatMFFDSetType(mat,MATMFFD_WP);CHKERRQ(ierr);
3589c6ac3b3SBarry Smith     ierr = MatSetFromOptions(mat);CHKERRQ(ierr);
359e884886fSBarry Smith   }
360e884886fSBarry Smith   ierr = (*ctx->ops->compute)(ctx,U,a,&h,&zeroa);CHKERRQ(ierr);
361e884886fSBarry Smith   if (zeroa) {
362e884886fSBarry Smith     ierr = VecSet(y,0.0);CHKERRQ(ierr);
363e884886fSBarry Smith     PetscFunctionReturn(0);
364e884886fSBarry Smith   }
365e884886fSBarry Smith 
366e32f2f54SBarry Smith   if (PetscIsInfOrNanScalar(h)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Computed Nan differencing parameter h");
367e884886fSBarry Smith   if (ctx->checkh) {
368e884886fSBarry Smith     ierr = (*ctx->checkh)(ctx->checkhctx,U,a,&h);CHKERRQ(ierr);
369e884886fSBarry Smith   }
370e884886fSBarry Smith 
371e884886fSBarry Smith   /* keep a record of the current differencing parameter h */
372e884886fSBarry Smith   ctx->currenth = h;
373e884886fSBarry Smith #if defined(PETSC_USE_COMPLEX)
374e884886fSBarry Smith   ierr = PetscInfo2(mat,"Current differencing parameter: %G + %G i\n",PetscRealPart(h),PetscImaginaryPart(h));CHKERRQ(ierr);
375e884886fSBarry Smith #else
376e884886fSBarry Smith   ierr = PetscInfo1(mat,"Current differencing parameter: %15.12e\n",h);CHKERRQ(ierr);
377e884886fSBarry Smith #endif
378e884886fSBarry Smith   if (ctx->historyh && ctx->ncurrenth < ctx->maxcurrenth) {
379e884886fSBarry Smith     ctx->historyh[ctx->ncurrenth] = h;
380e884886fSBarry Smith   }
381e884886fSBarry Smith   ctx->ncurrenth++;
382e884886fSBarry Smith 
383e884886fSBarry Smith   /* w = u + ha */
384c3bb7e23SBarry Smith   if (ctx->drscale) {
385c3bb7e23SBarry Smith     ierr = VecPointwiseMult(ctx->drscale,a,U);CHKERRQ(ierr);
386c3bb7e23SBarry Smith     ierr = VecAYPX(U,h,w);CHKERRQ(ierr);
387c3bb7e23SBarry Smith   } else {
388e884886fSBarry Smith     ierr = VecWAXPY(w,h,a,U);CHKERRQ(ierr);
389c3bb7e23SBarry Smith   }
390e884886fSBarry Smith 
391bcddec3dSBarry Smith   /* compute func(U) as base for differencing; only needed first time in and not when provided by user */
392bcddec3dSBarry Smith   if (ctx->ncurrenth == 1 && ctx->current_f_allocated) {
393e884886fSBarry Smith     ierr = (*ctx->func)(ctx->funcctx,U,F);CHKERRQ(ierr);
39452121784SBarry Smith   }
395e884886fSBarry Smith   ierr = (*ctx->func)(ctx->funcctx,w,y);CHKERRQ(ierr);
396e884886fSBarry Smith 
397e884886fSBarry Smith   ierr = VecAXPY(y,-1.0,F);CHKERRQ(ierr);
398e884886fSBarry Smith   ierr = VecScale(y,1.0/h);CHKERRQ(ierr);
399e884886fSBarry Smith 
400c35ec66cSMatthew G Knepley   if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) {
401e884886fSBarry Smith     ierr = VecAXPBY(y,ctx->vshift,ctx->vscale,a);CHKERRQ(ierr);
402c3bb7e23SBarry Smith   }
403c3bb7e23SBarry Smith   if (ctx->dlscale) {
404c3bb7e23SBarry Smith     ierr = VecPointwiseMult(y,ctx->dlscale,y);CHKERRQ(ierr);
405c3bb7e23SBarry Smith   }
406c3bb7e23SBarry Smith   if (ctx->dshift) {
407c3bb7e23SBarry Smith     ierr = VecPointwiseMult(ctx->dshift,a,U);CHKERRQ(ierr);
408c3bb7e23SBarry Smith     ierr = VecAXPY(y,1.0,U);CHKERRQ(ierr);
409c3bb7e23SBarry Smith   }
410e884886fSBarry Smith 
411e884886fSBarry Smith   if (ctx->sp) {ierr = MatNullSpaceRemove(ctx->sp,y,PETSC_NULL);CHKERRQ(ierr);}
412e884886fSBarry Smith 
413e884886fSBarry Smith   ierr = PetscLogEventEnd(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr);
414e884886fSBarry Smith   PetscFunctionReturn(0);
415e884886fSBarry Smith }
416e884886fSBarry Smith 
417e884886fSBarry Smith #undef __FUNCT__
418e884886fSBarry Smith #define __FUNCT__ "MatGetDiagonal_MFFD"
419e884886fSBarry Smith /*
420e884886fSBarry Smith   MatGetDiagonal_MFFD - Gets the diagonal for a matrix free matrix
421e884886fSBarry Smith 
422e884886fSBarry Smith         y ~= (F(u + ha) - F(u))/h,
423e884886fSBarry Smith   where F = nonlinear function, as set by SNESSetFunction()
424e884886fSBarry Smith         u = current iterate
425e884886fSBarry Smith         h = difference interval
426e884886fSBarry Smith */
427e884886fSBarry Smith PetscErrorCode MatGetDiagonal_MFFD(Mat mat,Vec a)
428e884886fSBarry Smith {
429e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
430e884886fSBarry Smith   PetscScalar    h,*aa,*ww,v;
431e884886fSBarry Smith   PetscReal      epsilon = PETSC_SQRT_MACHINE_EPSILON,umin = 100.0*PETSC_SQRT_MACHINE_EPSILON;
432e884886fSBarry Smith   Vec            w,U;
433e884886fSBarry Smith   PetscErrorCode ierr;
434e884886fSBarry Smith   PetscInt       i,rstart,rend;
435e884886fSBarry Smith 
436e884886fSBarry Smith   PetscFunctionBegin;
437e7e72b3dSBarry Smith   if (!ctx->funci) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Requires calling MatMFFDSetFunctioni() first");
438e884886fSBarry Smith 
439e884886fSBarry Smith   w    = ctx->w;
440e884886fSBarry Smith   U    = ctx->current_u;
441e884886fSBarry Smith   ierr = (*ctx->func)(ctx->funcctx,U,a);CHKERRQ(ierr);
442e884886fSBarry Smith   ierr = (*ctx->funcisetbase)(ctx->funcctx,U);CHKERRQ(ierr);
443e884886fSBarry Smith   ierr = VecCopy(U,w);CHKERRQ(ierr);
444e884886fSBarry Smith 
445e884886fSBarry Smith   ierr = VecGetOwnershipRange(a,&rstart,&rend);CHKERRQ(ierr);
446e884886fSBarry Smith   ierr = VecGetArray(a,&aa);CHKERRQ(ierr);
447e884886fSBarry Smith   for (i=rstart; i<rend; i++) {
448e884886fSBarry Smith     ierr = VecGetArray(w,&ww);CHKERRQ(ierr);
449e884886fSBarry Smith     h  = ww[i-rstart];
450e884886fSBarry Smith     if (h == 0.0) h = 1.0;
451e884886fSBarry Smith #if !defined(PETSC_USE_COMPLEX)
452e884886fSBarry Smith     if (h < umin && h >= 0.0)      h = umin;
453e884886fSBarry Smith     else if (h < 0.0 && h > -umin) h = -umin;
454e884886fSBarry Smith #else
455e884886fSBarry Smith     if (PetscAbsScalar(h) < umin && PetscRealPart(h) >= 0.0)     h = umin;
456e884886fSBarry Smith     else if (PetscRealPart(h) < 0.0 && PetscAbsScalar(h) < umin) h = -umin;
457e884886fSBarry Smith #endif
458e884886fSBarry Smith     h     *= epsilon;
459e884886fSBarry Smith 
460e884886fSBarry Smith     ww[i-rstart] += h;
461e884886fSBarry Smith     ierr = VecRestoreArray(w,&ww);CHKERRQ(ierr);
462e884886fSBarry Smith     ierr          = (*ctx->funci)(ctx->funcctx,i,w,&v);CHKERRQ(ierr);
463e884886fSBarry Smith     aa[i-rstart]  = (v - aa[i-rstart])/h;
464e884886fSBarry Smith 
465e884886fSBarry Smith     /* possibly shift and scale result */
466c35ec66cSMatthew G Knepley     if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) {
467e884886fSBarry Smith       aa[i - rstart] = ctx->vshift + ctx->vscale*aa[i-rstart];
468c3bb7e23SBarry Smith     }
469e884886fSBarry Smith 
470e884886fSBarry Smith     ierr = VecGetArray(w,&ww);CHKERRQ(ierr);
471e884886fSBarry Smith     ww[i-rstart] -= h;
472e884886fSBarry Smith     ierr = VecRestoreArray(w,&ww);CHKERRQ(ierr);
473e884886fSBarry Smith   }
474e884886fSBarry Smith   ierr = VecRestoreArray(a,&aa);CHKERRQ(ierr);
475e884886fSBarry Smith   PetscFunctionReturn(0);
476e884886fSBarry Smith }
477e884886fSBarry Smith 
478e884886fSBarry Smith #undef __FUNCT__
479c3bb7e23SBarry Smith #define __FUNCT__ "MatDiagonalScale_MFFD"
480c3bb7e23SBarry Smith PetscErrorCode MatDiagonalScale_MFFD(Mat mat,Vec ll,Vec rr)
481c3bb7e23SBarry Smith {
482c3bb7e23SBarry Smith   MatMFFD        aij = (MatMFFD)mat->data;
483c3bb7e23SBarry Smith   PetscErrorCode ierr;
484c3bb7e23SBarry Smith 
485c3bb7e23SBarry Smith   PetscFunctionBegin;
486c3bb7e23SBarry Smith   if (ll && !aij->dlscale) {
487c3bb7e23SBarry Smith     ierr = VecDuplicate(ll,&aij->dlscale);CHKERRQ(ierr);
488c3bb7e23SBarry Smith   }
489c3bb7e23SBarry Smith   if (rr && !aij->drscale) {
490c3bb7e23SBarry Smith     ierr = VecDuplicate(rr,&aij->drscale);CHKERRQ(ierr);
491c3bb7e23SBarry Smith   }
492c3bb7e23SBarry Smith   if (ll) {
493c3bb7e23SBarry Smith     ierr = VecCopy(ll,aij->dlscale);CHKERRQ(ierr);
494c3bb7e23SBarry Smith   }
495c3bb7e23SBarry Smith   if (rr) {
496c3bb7e23SBarry Smith     ierr = VecCopy(rr,aij->drscale);CHKERRQ(ierr);
497c3bb7e23SBarry Smith   }
498c3bb7e23SBarry Smith   PetscFunctionReturn(0);
499c3bb7e23SBarry Smith }
500c3bb7e23SBarry Smith 
501c3bb7e23SBarry Smith #undef __FUNCT__
502c3bb7e23SBarry Smith #define __FUNCT__ "MatDiagonalSet_MFFD"
503c3bb7e23SBarry Smith PetscErrorCode MatDiagonalSet_MFFD(Mat mat,Vec ll,InsertMode mode)
504c3bb7e23SBarry Smith {
505c3bb7e23SBarry Smith   MatMFFD        aij = (MatMFFD)mat->data;
506c3bb7e23SBarry Smith   PetscErrorCode ierr;
507c3bb7e23SBarry Smith 
508c3bb7e23SBarry Smith   PetscFunctionBegin;
509c3bb7e23SBarry Smith   if (mode == INSERT_VALUES) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_SUP,"No diagonal set with INSERT_VALUES");
510c3bb7e23SBarry Smith   if (!aij->dshift) {
511c3bb7e23SBarry Smith     ierr = VecDuplicate(ll,&aij->dshift);CHKERRQ(ierr);
512c3bb7e23SBarry Smith   }
513c3bb7e23SBarry Smith   ierr = VecAXPY(aij->dshift,1.0,ll);CHKERRQ(ierr);
514c3bb7e23SBarry Smith   PetscFunctionReturn(0);
515c3bb7e23SBarry Smith }
516c3bb7e23SBarry Smith 
517c3bb7e23SBarry Smith #undef __FUNCT__
518e884886fSBarry Smith #define __FUNCT__ "MatShift_MFFD"
519e884886fSBarry Smith PetscErrorCode MatShift_MFFD(Mat Y,PetscScalar a)
520e884886fSBarry Smith {
521e884886fSBarry Smith   MatMFFD shell = (MatMFFD)Y->data;
522e884886fSBarry Smith   PetscFunctionBegin;
523e884886fSBarry Smith   shell->vshift += a;
524e884886fSBarry Smith   PetscFunctionReturn(0);
525e884886fSBarry Smith }
526e884886fSBarry Smith 
527e884886fSBarry Smith #undef __FUNCT__
528e884886fSBarry Smith #define __FUNCT__ "MatScale_MFFD"
529e884886fSBarry Smith PetscErrorCode MatScale_MFFD(Mat Y,PetscScalar a)
530e884886fSBarry Smith {
531e884886fSBarry Smith   MatMFFD shell = (MatMFFD)Y->data;
532e884886fSBarry Smith   PetscFunctionBegin;
533e884886fSBarry Smith   shell->vscale *= a;
534e884886fSBarry Smith   PetscFunctionReturn(0);
535e884886fSBarry Smith }
536e884886fSBarry Smith 
537e884886fSBarry Smith EXTERN_C_BEGIN
538e884886fSBarry Smith #undef __FUNCT__
539c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetBase_MFFD"
5407087cfbeSBarry Smith PetscErrorCode  MatMFFDSetBase_MFFD(Mat J,Vec U,Vec F)
541e884886fSBarry Smith {
542e884886fSBarry Smith   PetscErrorCode ierr;
543e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)J->data;
544e884886fSBarry Smith 
545e884886fSBarry Smith   PetscFunctionBegin;
546e884886fSBarry Smith   ierr = MatMFFDResetHHistory(J);CHKERRQ(ierr);
547e884886fSBarry Smith   ctx->current_u = U;
54852121784SBarry Smith   if (F) {
549cfe22f5eSBarry Smith     if (ctx->current_f_allocated) {ierr = VecDestroy(ctx->current_f);CHKERRQ(ierr);}
5503ec795f1SBarry Smith     ctx->current_f           = F;
551cfe22f5eSBarry Smith     ctx->current_f_allocated = PETSC_FALSE;
552cfe22f5eSBarry Smith   } else if (!ctx->current_f_allocated) {
55352121784SBarry Smith     ierr = VecDuplicate(ctx->current_u, &ctx->current_f);CHKERRQ(ierr);
554cfe22f5eSBarry Smith     ctx->current_f_allocated = PETSC_TRUE;
55552121784SBarry Smith   }
556e884886fSBarry Smith   if (!ctx->w) {
557e884886fSBarry Smith     ierr = VecDuplicate(ctx->current_u, &ctx->w);CHKERRQ(ierr);
558e884886fSBarry Smith   }
559e884886fSBarry Smith   J->assembled = PETSC_TRUE;
560e884886fSBarry Smith   PetscFunctionReturn(0);
561e884886fSBarry Smith }
562e884886fSBarry Smith EXTERN_C_END
563e884886fSBarry Smith typedef PetscErrorCode (*FCN3)(void*,Vec,Vec,PetscScalar*); /* force argument to next function to not be extern C*/
564e884886fSBarry Smith EXTERN_C_BEGIN
565e884886fSBarry Smith #undef __FUNCT__
566c879e56bSBarry Smith #define __FUNCT__ "MatMFFDSetCheckh_MFFD"
5677087cfbeSBarry Smith PetscErrorCode  MatMFFDSetCheckh_MFFD(Mat J,FCN3 fun,void*ectx)
568e884886fSBarry Smith {
569e884886fSBarry Smith   MatMFFD ctx = (MatMFFD)J->data;
570e884886fSBarry Smith 
571e884886fSBarry Smith   PetscFunctionBegin;
572e884886fSBarry Smith   ctx->checkh    = fun;
573e884886fSBarry Smith   ctx->checkhctx = ectx;
574e884886fSBarry Smith   PetscFunctionReturn(0);
575e884886fSBarry Smith }
576e884886fSBarry Smith EXTERN_C_END
577e884886fSBarry Smith 
578e884886fSBarry Smith #undef __FUNCT__
5796aa9148fSLisandro Dalcin #define __FUNCT__ "MatMFFDSetOptionsPrefix"
5806aa9148fSLisandro Dalcin /*@C
5816aa9148fSLisandro Dalcin    MatMFFDSetOptionsPrefix - Sets the prefix used for searching for all
5826aa9148fSLisandro Dalcin    MatMFFD options in the database.
5836aa9148fSLisandro Dalcin 
5846aa9148fSLisandro Dalcin    Collective on Mat
5856aa9148fSLisandro Dalcin 
5866aa9148fSLisandro Dalcin    Input Parameter:
5876aa9148fSLisandro Dalcin +  A - the Mat context
5886aa9148fSLisandro Dalcin -  prefix - the prefix to prepend to all option names
5896aa9148fSLisandro Dalcin 
5906aa9148fSLisandro Dalcin    Notes:
5916aa9148fSLisandro Dalcin    A hyphen (-) must NOT be given at the beginning of the prefix name.
5926aa9148fSLisandro Dalcin    The first character of all runtime options is AUTOMATICALLY the hyphen.
5936aa9148fSLisandro Dalcin 
5946aa9148fSLisandro Dalcin    Level: advanced
5956aa9148fSLisandro Dalcin 
5966aa9148fSLisandro Dalcin .keywords: SNES, matrix-free, parameters
5976aa9148fSLisandro Dalcin 
5989c6ac3b3SBarry Smith .seealso: MatSetFromOptions(), MatCreateSNESMF()
5996aa9148fSLisandro Dalcin @*/
6007087cfbeSBarry Smith PetscErrorCode  MatMFFDSetOptionsPrefix(Mat mat,const char prefix[])
6016aa9148fSLisandro Dalcin 
6026aa9148fSLisandro Dalcin {
6036aa9148fSLisandro Dalcin   MatMFFD        mfctx = mat ? (MatMFFD)mat->data : PETSC_NULL;
6046aa9148fSLisandro Dalcin   PetscErrorCode ierr;
6056aa9148fSLisandro Dalcin   PetscFunctionBegin;
6060700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6070700a824SBarry Smith   PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1);
6086aa9148fSLisandro Dalcin   ierr = PetscObjectSetOptionsPrefix((PetscObject)mfctx,prefix);CHKERRQ(ierr);
6096aa9148fSLisandro Dalcin   PetscFunctionReturn(0);
6106aa9148fSLisandro Dalcin }
6116aa9148fSLisandro Dalcin 
6126aa9148fSLisandro Dalcin #undef __FUNCT__
6139c6ac3b3SBarry Smith #define __FUNCT__ "MatSetFromOptions_MFFD"
6149c6ac3b3SBarry Smith PetscErrorCode  MatSetFromOptions_MFFD(Mat mat)
615e884886fSBarry Smith {
61671f08665SBarry Smith   MatMFFD        mfctx = (MatMFFD)mat->data;
617e884886fSBarry Smith   PetscErrorCode ierr;
618ace3abfcSBarry Smith   PetscBool      flg;
619e884886fSBarry Smith   char           ftype[256];
620e884886fSBarry Smith 
621e884886fSBarry Smith   PetscFunctionBegin;
6220700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6230700a824SBarry Smith   PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1);
6247adad957SLisandro Dalcin   ierr = PetscOptionsBegin(((PetscObject)mfctx)->comm,((PetscObject)mfctx)->prefix,"Set matrix free computation parameters","MatMFFD");CHKERRQ(ierr);
625b022a5c1SBarry Smith   ierr = PetscOptionsList("-mat_mffd_type","Matrix free type","MatMFFDSetType",MatMFFDList,((PetscObject)mfctx)->type_name,ftype,256,&flg);CHKERRQ(ierr);
626e884886fSBarry Smith   if (flg) {
627e884886fSBarry Smith     ierr = MatMFFDSetType(mat,ftype);CHKERRQ(ierr);
628e884886fSBarry Smith   }
629e884886fSBarry Smith 
630e884886fSBarry Smith   ierr = PetscOptionsReal("-mat_mffd_err","set sqrt relative error in function","MatMFFDSetFunctionError",mfctx->error_rel,&mfctx->error_rel,0);CHKERRQ(ierr);
631e884886fSBarry Smith   ierr = PetscOptionsInt("-mat_mffd_period","how often h is recomputed","MatMFFDSetPeriod",mfctx->recomputeperiod,&mfctx->recomputeperiod,0);CHKERRQ(ierr);
632e884886fSBarry Smith 
63390d69ab7SBarry Smith   flg  = PETSC_FALSE;
634acfcf0e5SJed Brown   ierr = PetscOptionsBool("-mat_mffd_check_positivity","Insure that U + h*a is nonnegative","MatMFFDSetCheckh",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
635e884886fSBarry Smith   if (flg) {
636e884886fSBarry Smith     ierr = MatMFFDSetCheckh(mat,MatMFFDCheckPositivity,0);CHKERRQ(ierr);
637e884886fSBarry Smith   }
638e884886fSBarry Smith   if (mfctx->ops->setfromoptions) {
639e884886fSBarry Smith     ierr = (*mfctx->ops->setfromoptions)(mfctx);CHKERRQ(ierr);
640e884886fSBarry Smith   }
641e884886fSBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
642e884886fSBarry Smith   PetscFunctionReturn(0);
643e884886fSBarry Smith }
644e884886fSBarry Smith 
645*bc0f08ceSBarry Smith EXTERN_C_BEGIN
646*bc0f08ceSBarry Smith #undef __FUNCT__
647*bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetPeriod_MFFD"
648*bc0f08ceSBarry Smith PetscErrorCode  MatMFFDSetPeriod_MFFD(Mat mat,PetscInt period)
649*bc0f08ceSBarry Smith {
650*bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
651*bc0f08ceSBarry Smith 
652*bc0f08ceSBarry Smith   PetscFunctionBegin;
653*bc0f08ceSBarry Smith   PetscValidLogicalCollectiveInt(mat,period,2);
654*bc0f08ceSBarry Smith   ctx->recomputeperiod = period;
655*bc0f08ceSBarry Smith   PetscFunctionReturn(0);
656*bc0f08ceSBarry Smith }
657*bc0f08ceSBarry Smith EXTERN_C_END
658*bc0f08ceSBarry Smith 
659*bc0f08ceSBarry Smith EXTERN_C_BEGIN
660*bc0f08ceSBarry Smith #undef __FUNCT__
661*bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetFunction_MFFD"
662*bc0f08ceSBarry Smith PetscErrorCode  MatMFFDSetFunction_MFFD(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx)
663*bc0f08ceSBarry Smith {
664*bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
665*bc0f08ceSBarry Smith 
666*bc0f08ceSBarry Smith   PetscFunctionBegin;
667*bc0f08ceSBarry Smith   ctx->func    = func;
668*bc0f08ceSBarry Smith   ctx->funcctx = funcctx;
669*bc0f08ceSBarry Smith   PetscFunctionReturn(0);
670*bc0f08ceSBarry Smith }
671*bc0f08ceSBarry Smith EXTERN_C_END
672*bc0f08ceSBarry Smith 
673*bc0f08ceSBarry Smith EXTERN_C_BEGIN
674*bc0f08ceSBarry Smith #undef __FUNCT__
675*bc0f08ceSBarry Smith #define __FUNCT__ "MatMFFDSetFunctionError_MFFD"
676*bc0f08ceSBarry Smith PetscErrorCode  MatMFFDSetFunctionError_MFFD(Mat mat,PetscReal error)
677*bc0f08ceSBarry Smith {
678*bc0f08ceSBarry Smith   MatMFFD ctx = (MatMFFD)mat->data;
679*bc0f08ceSBarry Smith 
680*bc0f08ceSBarry Smith   PetscFunctionBegin;
681*bc0f08ceSBarry Smith   PetscValidLogicalCollectiveReal(mat,error,2);
682*bc0f08ceSBarry Smith   if (error != PETSC_DEFAULT) ctx->error_rel = error;
683*bc0f08ceSBarry Smith   PetscFunctionReturn(0);
684*bc0f08ceSBarry Smith }
685*bc0f08ceSBarry Smith EXTERN_C_END
686*bc0f08ceSBarry Smith 
687e884886fSBarry Smith /*MC
688e884886fSBarry Smith   MATMFFD - MATMFFD = "mffd" - A matrix free matrix type.
689e884886fSBarry Smith 
690e884886fSBarry Smith   Level: advanced
691e884886fSBarry Smith 
692d2d6cebeSBarry Smith .seealso: MatCreateMFFD(), MatCreateSNESMF(), MatMFFDSetFunction()
693e884886fSBarry Smith M*/
694e884886fSBarry Smith EXTERN_C_BEGIN
695e884886fSBarry Smith #undef __FUNCT__
696e884886fSBarry Smith #define __FUNCT__ "MatCreate_MFFD"
6977087cfbeSBarry Smith PetscErrorCode  MatCreate_MFFD(Mat A)
698e884886fSBarry Smith {
699e884886fSBarry Smith   MatMFFD         mfctx;
700e884886fSBarry Smith   PetscErrorCode  ierr;
701e884886fSBarry Smith 
702e884886fSBarry Smith   PetscFunctionBegin;
7033ec795f1SBarry Smith #ifndef PETSC_USE_DYNAMIC_LIBRARIES
7043ec795f1SBarry Smith   ierr = MatMFFDInitializePackage(PETSC_NULL);CHKERRQ(ierr);
7053ec795f1SBarry Smith #endif
706e884886fSBarry Smith 
7070700a824SBarry Smith   ierr = PetscHeaderCreate(mfctx,_p_MatMFFD,struct _MFOps,MATMFFD_CLASSID,0,"MatMFFD",((PetscObject)A)->comm,MatDestroy_MFFD,MatView_MFFD);CHKERRQ(ierr);
708e884886fSBarry Smith   mfctx->sp              = 0;
709e884886fSBarry Smith   mfctx->error_rel       = PETSC_SQRT_MACHINE_EPSILON;
710e884886fSBarry Smith   mfctx->recomputeperiod = 1;
711e884886fSBarry Smith   mfctx->count           = 0;
712e884886fSBarry Smith   mfctx->currenth        = 0.0;
713e884886fSBarry Smith   mfctx->historyh        = PETSC_NULL;
714e884886fSBarry Smith   mfctx->ncurrenth       = 0;
715e884886fSBarry Smith   mfctx->maxcurrenth     = 0;
7167adad957SLisandro Dalcin   ((PetscObject)mfctx)->type_name       = 0;
717e884886fSBarry Smith 
718e884886fSBarry Smith   mfctx->vshift          = 0.0;
719e884886fSBarry Smith   mfctx->vscale          = 1.0;
720e884886fSBarry Smith 
721e884886fSBarry Smith   /*
722e884886fSBarry Smith      Create the empty data structure to contain compute-h routines.
723e884886fSBarry Smith      These will be filled in below from the command line options or
724e884886fSBarry Smith      a later call with MatMFFDSetType() or if that is not called
725e884886fSBarry Smith      then it will default in the first use of MatMult_MFFD()
726e884886fSBarry Smith   */
727e884886fSBarry Smith   mfctx->ops->compute        = 0;
728e884886fSBarry Smith   mfctx->ops->destroy        = 0;
729e884886fSBarry Smith   mfctx->ops->view           = 0;
730e884886fSBarry Smith   mfctx->ops->setfromoptions = 0;
731e884886fSBarry Smith   mfctx->hctx                = 0;
732e884886fSBarry Smith 
733e884886fSBarry Smith   mfctx->func                = 0;
734e884886fSBarry Smith   mfctx->funcctx             = 0;
735e884886fSBarry Smith   mfctx->w                   = PETSC_NULL;
736e884886fSBarry Smith 
737e884886fSBarry Smith   A->data                = mfctx;
738e884886fSBarry Smith 
739e884886fSBarry Smith   A->ops->mult           = MatMult_MFFD;
740e884886fSBarry Smith   A->ops->destroy        = MatDestroy_MFFD;
741e884886fSBarry Smith   A->ops->view           = MatView_MFFD;
742e884886fSBarry Smith   A->ops->assemblyend    = MatAssemblyEnd_MFFD;
743e884886fSBarry Smith   A->ops->getdiagonal    = MatGetDiagonal_MFFD;
744e884886fSBarry Smith   A->ops->scale          = MatScale_MFFD;
745e884886fSBarry Smith   A->ops->shift          = MatShift_MFFD;
746c3bb7e23SBarry Smith   A->ops->diagonalscale  = MatDiagonalScale_MFFD;
747c3bb7e23SBarry Smith   A->ops->diagonalset    = MatDiagonalSet_MFFD;
7489c6ac3b3SBarry Smith   A->ops->setfromoptions = MatSetFromOptions_MFFD;
749e884886fSBarry Smith   A->assembled = PETSC_TRUE;
750e884886fSBarry Smith 
75126283091SBarry Smith   ierr = PetscLayoutSetBlockSize(A->rmap,1);CHKERRQ(ierr);
75226283091SBarry Smith   ierr = PetscLayoutSetBlockSize(A->cmap,1);CHKERRQ(ierr);
75326283091SBarry Smith   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
75426283091SBarry Smith   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
755ee1cef2cSJed Brown 
756c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetBase_C","MatMFFDSetBase_MFFD",MatMFFDSetBase_MFFD);CHKERRQ(ierr);
757c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctioniBase_C","MatMFFDSetFunctioniBase_MFFD",MatMFFDSetFunctioniBase_MFFD);CHKERRQ(ierr);
758c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctioni_C","MatMFFDSetFunctioni_MFFD",MatMFFDSetFunctioni_MFFD);CHKERRQ(ierr);
759*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunction_C","MatMFFDSetFunction_MFFD",MatMFFDSetFunction_MFFD);CHKERRQ(ierr);
760c879e56bSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetCheckh_C","MatMFFDSetCheckh_MFFD",MatMFFDSetCheckh_MFFD);CHKERRQ(ierr);
761*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetPeriod_C","MatMFFDSetPeriod_MFFD",MatMFFDSetPeriod_MFFD);CHKERRQ(ierr);
762*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDSetFunctionError_C","MatMFFDSetFunctionError_MFFD",MatMFFDSetFunctionError_MFFD);CHKERRQ(ierr);
763*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDResetHHistory_C","MatMFFDResetHHistory_MFFD",MatMFFDResetHHistory_MFFD);CHKERRQ(ierr);
764*bc0f08ceSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatMFFDAddNullSpace_C","MatMFFDAddNullSpace_MFFD",MatMFFDAddNullSpace_MFFD);CHKERRQ(ierr);
765e884886fSBarry Smith   mfctx->mat = A;
766e884886fSBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)A,MATMFFD);CHKERRQ(ierr);
767e884886fSBarry Smith   PetscFunctionReturn(0);
768e884886fSBarry Smith }
769e884886fSBarry Smith EXTERN_C_END
770e884886fSBarry Smith 
771e884886fSBarry Smith #undef __FUNCT__
772e884886fSBarry Smith #define __FUNCT__ "MatCreateMFFD"
773e884886fSBarry Smith /*@
774e884886fSBarry Smith    MatCreateMFFD - Creates a matrix-free matrix. See also MatCreateSNESMF()
775e884886fSBarry Smith 
776e884886fSBarry Smith    Collective on Vec
777e884886fSBarry Smith 
778e884886fSBarry Smith    Input Parameters:
779fef1beadSBarry Smith +  comm - MPI communicator
780fef1beadSBarry Smith .  m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
781fef1beadSBarry Smith            This value should be the same as the local size used in creating the
782fef1beadSBarry Smith            y vector for the matrix-vector product y = Ax.
783fef1beadSBarry Smith .  n - This value should be the same as the local size used in creating the
784fef1beadSBarry Smith        x vector for the matrix-vector product y = Ax. (or PETSC_DECIDE to have
785fef1beadSBarry Smith        calculated if N is given) For square matrices n is almost always m.
786fef1beadSBarry Smith .  M - number of global rows (or PETSC_DETERMINE to have calculated if m is given)
787fef1beadSBarry Smith -  N - number of global columns (or PETSC_DETERMINE to have calculated if n is given)
788fef1beadSBarry Smith 
789e884886fSBarry Smith 
790e884886fSBarry Smith    Output Parameter:
791e884886fSBarry Smith .  J - the matrix-free matrix
792e884886fSBarry Smith 
7939c6ac3b3SBarry Smith    Options Database Keys: call MatSetFromOptions() to trigger these
7949c6ac3b3SBarry Smith +  -mat_mffd_type - wp or ds (see MATMFFD_WP or MATMFFD_DS)
7959c6ac3b3SBarry Smith -  -mat_mffd_err - square root of estimated relative error in function evaluation
7969c6ac3b3SBarry Smith -  -mat_mffd_period - how often h is recomputed, defaults to 1, everytime
7979c6ac3b3SBarry Smith 
7989c6ac3b3SBarry Smith 
799e884886fSBarry Smith    Level: advanced
800e884886fSBarry Smith 
801e884886fSBarry Smith    Notes:
802e884886fSBarry Smith    The matrix-free matrix context merely contains the function pointers
803e884886fSBarry Smith    and work space for performing finite difference approximations of
804e884886fSBarry Smith    Jacobian-vector products, F'(u)*a,
805e884886fSBarry Smith 
806e884886fSBarry Smith    The default code uses the following approach to compute h
807e884886fSBarry Smith 
808e884886fSBarry Smith .vb
809e884886fSBarry Smith      F'(u)*a = [F(u+h*a) - F(u)]/h where
810e884886fSBarry Smith      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
811e884886fSBarry Smith        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   otherwise
812e884886fSBarry Smith  where
813e884886fSBarry Smith      error_rel = square root of relative error in function evaluation
814e884886fSBarry Smith      umin = minimum iterate parameter
815e884886fSBarry Smith .ve
816e884886fSBarry Smith 
817e884886fSBarry Smith    The user can set the error_rel via MatMFFDSetFunctionError() and
8180598bfebSBarry Smith    umin via MatMFFDDSSetUmin(); see the <A href="../../docs/manual.pdf#nameddest=ch_snes">SNES chapter of the users manual</A> for details.
819e884886fSBarry Smith 
820e884886fSBarry Smith    The user should call MatDestroy() when finished with the matrix-free
821e884886fSBarry Smith    matrix context.
822e884886fSBarry Smith 
823e884886fSBarry Smith    Options Database Keys:
824e884886fSBarry Smith +  -mat_mffd_err <error_rel> - Sets error_rel
825e884886fSBarry Smith .  -mat_mffd_unim <umin> - Sets umin (for default PETSc routine that computes h only)
826e884886fSBarry Smith -  -mat_mffd_check_positivity
827e884886fSBarry Smith 
828e884886fSBarry Smith .keywords: default, matrix-free, create, matrix
829e884886fSBarry Smith 
8301d0fab5eSBarry Smith .seealso: MatDestroy(), MatMFFDSetFunctionError(), MatMFFDDSSetUmin(), MatMFFDSetFunction()
831e884886fSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory(), MatCreateSNESMF(),
8321d0fab5eSBarry Smith           MatMFFDGetH(), MatMFFDRegisterDynamic), MatMFFDComputeJacobian()
833e884886fSBarry Smith 
834e884886fSBarry Smith @*/
8357087cfbeSBarry Smith PetscErrorCode  MatCreateMFFD(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,Mat *J)
836e884886fSBarry Smith {
837e884886fSBarry Smith   PetscErrorCode ierr;
838e884886fSBarry Smith 
839e884886fSBarry Smith   PetscFunctionBegin;
840e884886fSBarry Smith   ierr = MatCreate(comm,J);CHKERRQ(ierr);
841fef1beadSBarry Smith   ierr = MatSetSizes(*J,m,n,M,N);CHKERRQ(ierr);
842e884886fSBarry Smith   ierr = MatSetType(*J,MATMFFD);CHKERRQ(ierr);
843e884886fSBarry Smith   PetscFunctionReturn(0);
844e884886fSBarry Smith }
845e884886fSBarry Smith 
846e884886fSBarry Smith 
847e884886fSBarry Smith #undef __FUNCT__
848e884886fSBarry Smith #define __FUNCT__ "MatMFFDGetH"
849e884886fSBarry Smith /*@
850e884886fSBarry Smith    MatMFFDGetH - Gets the last value that was used as the differencing
851e884886fSBarry Smith    parameter.
852e884886fSBarry Smith 
853e884886fSBarry Smith    Not Collective
854e884886fSBarry Smith 
855e884886fSBarry Smith    Input Parameters:
856e884886fSBarry Smith .  mat - the matrix obtained with MatCreateSNESMF()
857e884886fSBarry Smith 
858e884886fSBarry Smith    Output Paramter:
859e884886fSBarry Smith .  h - the differencing step size
860e884886fSBarry Smith 
861e884886fSBarry Smith    Level: advanced
862e884886fSBarry Smith 
863e884886fSBarry Smith .keywords: SNES, matrix-free, parameters
864e884886fSBarry Smith 
8651d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDSetHHistory(), MatCreateMFFD(), MATMFFD, MatMFFDResetHHistory()
866e884886fSBarry Smith @*/
8677087cfbeSBarry Smith PetscErrorCode  MatMFFDGetH(Mat mat,PetscScalar *h)
868e884886fSBarry Smith {
869e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)mat->data;
870*bc0f08ceSBarry Smith   PetscErrorCode ierr;
871*bc0f08ceSBarry Smith   PetscBool      match;
872e884886fSBarry Smith 
873e884886fSBarry Smith   PetscFunctionBegin;
874*bc0f08ceSBarry Smith   ierr = PetscTypeCompare((PetscObject)mat,MATMFFD,&match);CHKERRQ(ierr);
875*bc0f08ceSBarry Smith   if (!match) SETERRQ(((PetscObject)mat)->comm,PETSC_ERR_ARG_WRONG,"Not a MFFD matrix");
876*bc0f08ceSBarry Smith 
877e884886fSBarry Smith   *h = ctx->currenth;
878e884886fSBarry Smith   PetscFunctionReturn(0);
879e884886fSBarry Smith }
880e884886fSBarry Smith 
881e884886fSBarry Smith #undef __FUNCT__
882e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunction"
883e884886fSBarry Smith /*@C
884e884886fSBarry Smith    MatMFFDSetFunction - Sets the function used in applying the matrix free.
885e884886fSBarry Smith 
8863f9fe445SBarry Smith    Logically Collective on Mat
887e884886fSBarry Smith 
888e884886fSBarry Smith    Input Parameters:
889e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
890e884886fSBarry Smith .  func - the function to use
891e884886fSBarry Smith -  funcctx - optional function context passed to function
892e884886fSBarry Smith 
893e884886fSBarry Smith    Level: advanced
894e884886fSBarry Smith 
895e884886fSBarry Smith    Notes:
896e884886fSBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
897e884886fSBarry Smith     matrix inside your compute Jacobian routine
898e884886fSBarry Smith 
8993ec795f1SBarry Smith     If this is not set then it will use the function set with SNESSetFunction() if MatCreateSNESMF() was used.
900e884886fSBarry Smith 
901e884886fSBarry Smith .keywords: SNES, matrix-free, function
902e884886fSBarry Smith 
9031d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD,
9041d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
905e884886fSBarry Smith @*/
9067087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunction(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx)
907e884886fSBarry Smith {
908*bc0f08ceSBarry Smith   PetscErrorCode ierr;
909e884886fSBarry Smith 
910e884886fSBarry Smith   PetscFunctionBegin;
911*bc0f08ceSBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunction_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec),void*),(mat,func,funcctx));CHKERRQ(ierr);
912e884886fSBarry Smith   PetscFunctionReturn(0);
913e884886fSBarry Smith }
914e884886fSBarry Smith 
915e884886fSBarry Smith #undef __FUNCT__
916e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioni"
917e884886fSBarry Smith /*@C
918e884886fSBarry Smith    MatMFFDSetFunctioni - Sets the function for a single component
919e884886fSBarry Smith 
9203f9fe445SBarry Smith    Logically Collective on Mat
921e884886fSBarry Smith 
922e884886fSBarry Smith    Input Parameters:
923e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
924e884886fSBarry Smith -  funci - the function to use
925e884886fSBarry Smith 
926e884886fSBarry Smith    Level: advanced
927e884886fSBarry Smith 
928e884886fSBarry Smith    Notes:
929e884886fSBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
930e884886fSBarry Smith     matrix inside your compute Jacobian routine
931e884886fSBarry Smith 
932e884886fSBarry Smith 
933e884886fSBarry Smith .keywords: SNES, matrix-free, function
934e884886fSBarry Smith 
9351d0fab5eSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
9361d0fab5eSBarry Smith 
937e884886fSBarry Smith @*/
9387087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctioni(Mat mat,PetscErrorCode (*funci)(void*,PetscInt,Vec,PetscScalar*))
939e884886fSBarry Smith {
9404ac538c5SBarry Smith   PetscErrorCode ierr;
941e884886fSBarry Smith 
942e884886fSBarry Smith   PetscFunctionBegin;
9430700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9444ac538c5SBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunctioni_C",(Mat,PetscErrorCode (*)(void*,PetscInt,Vec,PetscScalar*)),(mat,funci));CHKERRQ(ierr);
945e884886fSBarry Smith   PetscFunctionReturn(0);
946e884886fSBarry Smith }
947e884886fSBarry Smith 
948e884886fSBarry Smith 
949e884886fSBarry Smith #undef __FUNCT__
950e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctioniBase"
951e884886fSBarry Smith /*@C
952e884886fSBarry Smith    MatMFFDSetFunctioniBase - Sets the base vector for a single component function evaluation
953e884886fSBarry Smith 
9543f9fe445SBarry Smith    Logically Collective on Mat
955e884886fSBarry Smith 
956e884886fSBarry Smith    Input Parameters:
957e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
958e884886fSBarry Smith -  func - the function to use
959e884886fSBarry Smith 
960e884886fSBarry Smith    Level: advanced
961e884886fSBarry Smith 
962e884886fSBarry Smith    Notes:
963e884886fSBarry Smith     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
964e884886fSBarry Smith     matrix inside your compute Jacobian routine
965e884886fSBarry Smith 
966e884886fSBarry Smith 
967e884886fSBarry Smith .keywords: SNES, matrix-free, function
968e884886fSBarry Smith 
969e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD
9701d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
971e884886fSBarry Smith @*/
9727087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctioniBase(Mat mat,PetscErrorCode (*func)(void*,Vec))
973e884886fSBarry Smith {
9744ac538c5SBarry Smith   PetscErrorCode ierr;
975e884886fSBarry Smith 
976e884886fSBarry Smith   PetscFunctionBegin;
9770700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9784ac538c5SBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunctioniBase_C",(Mat,PetscErrorCode (*)(void*,Vec)),(mat,func));CHKERRQ(ierr);
979e884886fSBarry Smith   PetscFunctionReturn(0);
980e884886fSBarry Smith }
981e884886fSBarry Smith 
982e884886fSBarry Smith #undef __FUNCT__
983e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetPeriod"
984e884886fSBarry Smith /*@
985e884886fSBarry Smith    MatMFFDSetPeriod - Sets how often h is recomputed, by default it is everytime
986e884886fSBarry Smith 
9873f9fe445SBarry Smith    Logically Collective on Mat
988e884886fSBarry Smith 
989e884886fSBarry Smith    Input Parameters:
990e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateSNESMF()
991e884886fSBarry Smith -  period - 1 for everytime, 2 for every second etc
992e884886fSBarry Smith 
993e884886fSBarry Smith    Options Database Keys:
994e884886fSBarry Smith +  -mat_mffd_period <period>
995e884886fSBarry Smith 
996e884886fSBarry Smith    Level: advanced
997e884886fSBarry Smith 
998e884886fSBarry Smith 
999e884886fSBarry Smith .keywords: SNES, matrix-free, parameters
1000e884886fSBarry Smith 
1001e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(),
10021d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory()
1003e884886fSBarry Smith @*/
10047087cfbeSBarry Smith PetscErrorCode  MatMFFDSetPeriod(Mat mat,PetscInt period)
1005e884886fSBarry Smith {
1006*bc0f08ceSBarry Smith   PetscErrorCode ierr;
1007e884886fSBarry Smith 
1008e884886fSBarry Smith   PetscFunctionBegin;
1009*bc0f08ceSBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetPeriod_C",(Mat,PetscInt),(mat,period));CHKERRQ(ierr);
1010e884886fSBarry Smith   PetscFunctionReturn(0);
1011e884886fSBarry Smith }
1012e884886fSBarry Smith 
1013e884886fSBarry Smith #undef __FUNCT__
1014e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetFunctionError"
1015e884886fSBarry Smith /*@
1016e884886fSBarry Smith    MatMFFDSetFunctionError - Sets the error_rel for the approximation of
1017e884886fSBarry Smith    matrix-vector products using finite differences.
1018e884886fSBarry Smith 
10193f9fe445SBarry Smith    Logically Collective on Mat
1020e884886fSBarry Smith 
1021e884886fSBarry Smith    Input Parameters:
1022e884886fSBarry Smith +  mat - the matrix free matrix created via MatCreateMFFD() or MatCreateSNESMF()
1023e884886fSBarry Smith -  error_rel - relative error (should be set to the square root of
1024e884886fSBarry Smith                the relative error in the function evaluations)
1025e884886fSBarry Smith 
1026e884886fSBarry Smith    Options Database Keys:
1027e884886fSBarry Smith +  -mat_mffd_err <error_rel> - Sets error_rel
1028e884886fSBarry Smith 
1029e884886fSBarry Smith    Level: advanced
1030e884886fSBarry Smith 
1031e884886fSBarry Smith    Notes:
1032e884886fSBarry Smith    The default matrix-free matrix-vector product routine computes
1033e884886fSBarry Smith .vb
1034e884886fSBarry Smith      F'(u)*a = [F(u+h*a) - F(u)]/h where
1035e884886fSBarry Smith      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
1036e884886fSBarry Smith        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   else
1037e884886fSBarry Smith .ve
1038e884886fSBarry Smith 
1039e884886fSBarry Smith .keywords: SNES, matrix-free, parameters
1040e884886fSBarry Smith 
1041e884886fSBarry Smith .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD
10421d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory()
1043e884886fSBarry Smith @*/
10447087cfbeSBarry Smith PetscErrorCode  MatMFFDSetFunctionError(Mat mat,PetscReal error)
1045e884886fSBarry Smith {
1046*bc0f08ceSBarry Smith   PetscErrorCode ierr;
1047e884886fSBarry Smith 
1048e884886fSBarry Smith   PetscFunctionBegin;
1049*bc0f08ceSBarry Smith   ierr = PetscTryMethod(mat,"MatMFFDSetFunctionError_C",(Mat,PetscReal),(mat,error));CHKERRQ(ierr);
1050e884886fSBarry Smith   PetscFunctionReturn(0);
1051e884886fSBarry Smith }
1052e884886fSBarry Smith 
1053e884886fSBarry Smith #undef __FUNCT__
1054e884886fSBarry Smith #define __FUNCT__ "MatMFFDAddNullSpace"
1055e884886fSBarry Smith /*@
1056e884886fSBarry Smith    MatMFFDAddNullSpace - Provides a null space that an operator is
1057e884886fSBarry Smith    supposed to have.  Since roundoff will create a small component in
1058e884886fSBarry Smith    the null space, if you know the null space you may have it
1059e884886fSBarry Smith    automatically removed.
1060e884886fSBarry Smith 
10613f9fe445SBarry Smith    Logically Collective on Mat
1062e884886fSBarry Smith 
1063e884886fSBarry Smith    Input Parameters:
1064e884886fSBarry Smith +  J - the matrix-free matrix context
1065e884886fSBarry Smith -  nullsp - object created with MatNullSpaceCreate()
1066e884886fSBarry Smith 
1067e884886fSBarry Smith    Level: advanced
1068e884886fSBarry Smith 
1069e884886fSBarry Smith .keywords: SNES, matrix-free, null space
1070e884886fSBarry Smith 
1071e884886fSBarry Smith .seealso: MatNullSpaceCreate(), MatMFFDGetH(), MatCreateSNESMF(), MatCreateMFFD(), MATMFFD
10721d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDResetHHistory()
1073e884886fSBarry Smith @*/
10747087cfbeSBarry Smith PetscErrorCode  MatMFFDAddNullSpace(Mat J,MatNullSpace nullsp)
1075e884886fSBarry Smith {
1076e884886fSBarry Smith   PetscErrorCode ierr;
1077e884886fSBarry Smith 
1078e884886fSBarry Smith   PetscFunctionBegin;
1079*bc0f08ceSBarry Smith   ierr = PetscTryMethod(J,"MatMFFDAddNullSpace_C",(Mat,MatNullSpace),(J,nullsp));CHKERRQ(ierr);
1080e884886fSBarry Smith   PetscFunctionReturn(0);
1081e884886fSBarry Smith }
1082e884886fSBarry Smith 
1083e884886fSBarry Smith #undef __FUNCT__
1084e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetHHistory"
1085e884886fSBarry Smith /*@
1086e884886fSBarry Smith    MatMFFDSetHHistory - Sets an array to collect a history of the
1087e884886fSBarry Smith    differencing values (h) computed for the matrix-free product.
1088e884886fSBarry Smith 
10893f9fe445SBarry Smith    Logically Collective on Mat
1090e884886fSBarry Smith 
1091e884886fSBarry Smith    Input Parameters:
1092e884886fSBarry Smith +  J - the matrix-free matrix context
1093e884886fSBarry Smith .  histroy - space to hold the history
1094e884886fSBarry Smith -  nhistory - number of entries in history, if more entries are generated than
1095e884886fSBarry Smith               nhistory, then the later ones are discarded
1096e884886fSBarry Smith 
1097e884886fSBarry Smith    Level: advanced
1098e884886fSBarry Smith 
1099e884886fSBarry Smith    Notes:
1100e884886fSBarry Smith    Use MatMFFDResetHHistory() to reset the history counter and collect
1101e884886fSBarry Smith    a new batch of differencing parameters, h.
1102e884886fSBarry Smith 
1103e884886fSBarry Smith .keywords: SNES, matrix-free, h history, differencing history
1104e884886fSBarry Smith 
1105e884886fSBarry Smith .seealso: MatMFFDGetH(), MatCreateSNESMF(),
11061d0fab5eSBarry Smith           MatMFFDResetHHistory(), MatMFFDSetFunctionError()
1107e884886fSBarry Smith 
1108e884886fSBarry Smith @*/
11097087cfbeSBarry Smith PetscErrorCode  MatMFFDSetHHistory(Mat J,PetscScalar history[],PetscInt nhistory)
1110e884886fSBarry Smith {
1111e884886fSBarry Smith   MatMFFD        ctx = (MatMFFD)J->data;
1112*bc0f08ceSBarry Smith   PetscErrorCode ierr;
1113*bc0f08ceSBarry Smith   PetscBool      match;
1114e884886fSBarry Smith 
1115e884886fSBarry Smith   PetscFunctionBegin;
1116*bc0f08ceSBarry Smith   ierr = PetscTypeCompare((PetscObject)J,MATMFFD,&match);CHKERRQ(ierr);
1117*bc0f08ceSBarry Smith   if (!match) SETERRQ(((PetscObject)J)->comm,PETSC_ERR_ARG_WRONG,"Not a MFFD matrix");
1118e884886fSBarry Smith   ctx->historyh    = history;
1119e884886fSBarry Smith   ctx->maxcurrenth = nhistory;
112075567043SBarry Smith   ctx->currenth    = 0.;
1121e884886fSBarry Smith   PetscFunctionReturn(0);
1122e884886fSBarry Smith }
1123e884886fSBarry Smith 
1124*bc0f08ceSBarry Smith 
1125e884886fSBarry Smith #undef __FUNCT__
1126e884886fSBarry Smith #define __FUNCT__ "MatMFFDResetHHistory"
1127e884886fSBarry Smith /*@
1128e884886fSBarry Smith    MatMFFDResetHHistory - Resets the counter to zero to begin
1129e884886fSBarry Smith    collecting a new set of differencing histories.
1130e884886fSBarry Smith 
11313f9fe445SBarry Smith    Logically Collective on Mat
1132e884886fSBarry Smith 
1133e884886fSBarry Smith    Input Parameters:
1134e884886fSBarry Smith .  J - the matrix-free matrix context
1135e884886fSBarry Smith 
1136e884886fSBarry Smith    Level: advanced
1137e884886fSBarry Smith 
1138e884886fSBarry Smith    Notes:
1139e884886fSBarry Smith    Use MatMFFDSetHHistory() to create the original history counter.
1140e884886fSBarry Smith 
1141e884886fSBarry Smith .keywords: SNES, matrix-free, h history, differencing history
1142e884886fSBarry Smith 
1143e884886fSBarry Smith .seealso: MatMFFDGetH(), MatCreateSNESMF(),
11441d0fab5eSBarry Smith           MatMFFDSetHHistory(), MatMFFDSetFunctionError()
1145e884886fSBarry Smith 
1146e884886fSBarry Smith @*/
11477087cfbeSBarry Smith PetscErrorCode  MatMFFDResetHHistory(Mat J)
1148e884886fSBarry Smith {
1149*bc0f08ceSBarry Smith   PetscErrorCode ierr;
1150e884886fSBarry Smith 
1151e884886fSBarry Smith   PetscFunctionBegin;
1152*bc0f08ceSBarry Smith   ierr = PetscTryMethod(J,"MatMFFDResetHHistory_C",(Mat),(J));CHKERRQ(ierr);
1153e884886fSBarry Smith   PetscFunctionReturn(0);
1154e884886fSBarry Smith }
1155e884886fSBarry Smith 
1156e884886fSBarry Smith 
1157e884886fSBarry Smith #undef __FUNCT__
1158e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetBase"
1159e884886fSBarry Smith /*@
1160e884886fSBarry Smith     MatMFFDSetBase - Sets the vector U at which matrix vector products of the
1161e884886fSBarry Smith         Jacobian are computed
1162e884886fSBarry Smith 
11633f9fe445SBarry Smith     Logically Collective on Mat
1164e884886fSBarry Smith 
1165e884886fSBarry Smith     Input Parameters:
1166e884886fSBarry Smith +   J - the MatMFFD matrix
11673ec795f1SBarry Smith .   U - the vector
1168bcddec3dSBarry Smith -   F - (optional) vector that contains F(u) if it has been already computed
1169e884886fSBarry Smith 
1170e884886fSBarry Smith     Notes: This is rarely used directly
1171e884886fSBarry Smith 
11728af5ae88SBarry Smith     If F is provided then it is not recomputed. Otherwise the function is evaluated at the base
11738af5ae88SBarry Smith     point during the first MatMult() after each call to MatMFFDSetBase().
1174dff2f722SBarry Smith 
1175e884886fSBarry Smith     Level: advanced
1176e884886fSBarry Smith 
1177e884886fSBarry Smith @*/
11787087cfbeSBarry Smith PetscErrorCode  MatMFFDSetBase(Mat J,Vec U,Vec F)
1179e884886fSBarry Smith {
11804ac538c5SBarry Smith   PetscErrorCode ierr;
1181e884886fSBarry Smith 
1182e884886fSBarry Smith   PetscFunctionBegin;
11830700a824SBarry Smith   PetscValidHeaderSpecific(J,MAT_CLASSID,1);
11840700a824SBarry Smith   PetscValidHeaderSpecific(U,VEC_CLASSID,2);
11850700a824SBarry Smith   if (F) PetscValidHeaderSpecific(F,VEC_CLASSID,3);
11864ac538c5SBarry Smith   ierr = PetscTryMethod(J,"MatMFFDSetBase_C",(Mat,Vec,Vec),(J,U,F));CHKERRQ(ierr);
1187e884886fSBarry Smith   PetscFunctionReturn(0);
1188e884886fSBarry Smith }
1189e884886fSBarry Smith 
1190e884886fSBarry Smith #undef __FUNCT__
1191e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetCheckh"
1192e884886fSBarry Smith /*@C
1193e884886fSBarry Smith     MatMFFDSetCheckh - Sets a function that checks the computed h and adjusts
1194e884886fSBarry Smith         it to satisfy some criteria
1195e884886fSBarry Smith 
11963f9fe445SBarry Smith     Logically Collective on Mat
1197e884886fSBarry Smith 
1198e884886fSBarry Smith     Input Parameters:
1199e884886fSBarry Smith +   J - the MatMFFD matrix
1200e884886fSBarry Smith .   fun - the function that checks h
1201e884886fSBarry Smith -   ctx - any context needed by the function
1202e884886fSBarry Smith 
1203e884886fSBarry Smith     Options Database Keys:
1204e884886fSBarry Smith .   -mat_mffd_check_positivity
1205e884886fSBarry Smith 
1206e884886fSBarry Smith     Level: advanced
1207e884886fSBarry Smith 
1208e884886fSBarry Smith     Notes: For example, MatMFFDSetCheckPositivity() insures that all entries
1209e884886fSBarry Smith        of U + h*a are non-negative
1210e884886fSBarry Smith 
1211e884886fSBarry Smith .seealso:  MatMFFDSetCheckPositivity()
1212e884886fSBarry Smith @*/
12137087cfbeSBarry Smith PetscErrorCode  MatMFFDSetCheckh(Mat J,PetscErrorCode (*fun)(void*,Vec,Vec,PetscScalar*),void* ctx)
1214e884886fSBarry Smith {
12154ac538c5SBarry Smith   PetscErrorCode ierr;
1216e884886fSBarry Smith 
1217e884886fSBarry Smith   PetscFunctionBegin;
12180700a824SBarry Smith   PetscValidHeaderSpecific(J,MAT_CLASSID,1);
12194ac538c5SBarry Smith   ierr = PetscTryMethod(J,"MatMFFDSetCheckh_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec,PetscScalar*),void*),(J,fun,ctx));CHKERRQ(ierr);
1220e884886fSBarry Smith   PetscFunctionReturn(0);
1221e884886fSBarry Smith }
1222e884886fSBarry Smith 
1223e884886fSBarry Smith #undef __FUNCT__
1224e884886fSBarry Smith #define __FUNCT__ "MatMFFDSetCheckPositivity"
1225e884886fSBarry Smith /*@
1226e884886fSBarry Smith     MatMFFDCheckPositivity - Checks that all entries in U + h*a are positive or
1227e884886fSBarry Smith         zero, decreases h until this is satisfied.
1228e884886fSBarry Smith 
12293f9fe445SBarry Smith     Logically Collective on Vec
1230e884886fSBarry Smith 
1231e884886fSBarry Smith     Input Parameters:
1232e884886fSBarry Smith +   U - base vector that is added to
1233e884886fSBarry Smith .   a - vector that is added
1234e884886fSBarry Smith .   h - scaling factor on a
1235e884886fSBarry Smith -   dummy - context variable (unused)
1236e884886fSBarry Smith 
1237e884886fSBarry Smith     Options Database Keys:
1238e884886fSBarry Smith .   -mat_mffd_check_positivity
1239e884886fSBarry Smith 
1240e884886fSBarry Smith     Level: advanced
1241e884886fSBarry Smith 
1242e884886fSBarry Smith     Notes: This is rarely used directly, rather it is passed as an argument to
1243e884886fSBarry Smith            MatMFFDSetCheckh()
1244e884886fSBarry Smith 
1245e884886fSBarry Smith .seealso:  MatMFFDSetCheckh()
1246e884886fSBarry Smith @*/
12477087cfbeSBarry Smith PetscErrorCode  MatMFFDCheckPositivity(void* dummy,Vec U,Vec a,PetscScalar *h)
1248e884886fSBarry Smith {
1249e884886fSBarry Smith   PetscReal      val, minval;
1250e884886fSBarry Smith   PetscScalar    *u_vec, *a_vec;
1251e884886fSBarry Smith   PetscErrorCode ierr;
1252e884886fSBarry Smith   PetscInt       i,n;
1253e884886fSBarry Smith   MPI_Comm       comm;
1254e884886fSBarry Smith 
1255e884886fSBarry Smith   PetscFunctionBegin;
1256e884886fSBarry Smith   ierr = PetscObjectGetComm((PetscObject)U,&comm);CHKERRQ(ierr);
1257e884886fSBarry Smith   ierr = VecGetArray(U,&u_vec);CHKERRQ(ierr);
1258e884886fSBarry Smith   ierr = VecGetArray(a,&a_vec);CHKERRQ(ierr);
1259e884886fSBarry Smith   ierr = VecGetLocalSize(U,&n);CHKERRQ(ierr);
1260e884886fSBarry Smith   minval = PetscAbsScalar(*h*1.01);
1261e884886fSBarry Smith   for(i=0;i<n;i++) {
1262e884886fSBarry Smith     if (PetscRealPart(u_vec[i] + *h*a_vec[i]) <= 0.0) {
1263e884886fSBarry Smith       val = PetscAbsScalar(u_vec[i]/a_vec[i]);
1264e884886fSBarry Smith       if (val < minval) minval = val;
1265e884886fSBarry Smith     }
1266e884886fSBarry Smith   }
1267e884886fSBarry Smith   ierr = VecRestoreArray(U,&u_vec);CHKERRQ(ierr);
1268e884886fSBarry Smith   ierr = VecRestoreArray(a,&a_vec);CHKERRQ(ierr);
1269d9822059SBarry Smith   ierr = MPI_Allreduce(&minval,&val,1,MPIU_REAL,MPIU_MIN,comm);CHKERRQ(ierr);
1270e884886fSBarry Smith   if (val <= PetscAbsScalar(*h)) {
1271e884886fSBarry Smith     ierr = PetscInfo2(U,"Scaling back h from %G to %G\n",PetscRealPart(*h),.99*val);CHKERRQ(ierr);
1272e884886fSBarry Smith     if (PetscRealPart(*h) > 0.0) *h =  0.99*val;
1273e884886fSBarry Smith     else                         *h = -0.99*val;
1274e884886fSBarry Smith   }
1275e884886fSBarry Smith   PetscFunctionReturn(0);
1276e884886fSBarry Smith }
1277e884886fSBarry Smith 
1278e884886fSBarry Smith 
1279e884886fSBarry Smith 
1280e884886fSBarry Smith 
1281e884886fSBarry Smith 
1282e884886fSBarry Smith 
1283e884886fSBarry Smith 
1284