xref: /petsc/src/mat/impls/mffd/mffd.c (revision cd5eaeb712f0bc77bdce843018b541db19340d1f)
1 
2 #include <petsc/private/matimpl.h>
3 #include <../src/mat/impls/mffd/mffdimpl.h>   /*I  "petscmat.h"   I*/
4 
5 PetscFunctionList MatMFFDList              = 0;
6 PetscBool         MatMFFDRegisterAllCalled = PETSC_FALSE;
7 
8 PetscClassId  MATMFFD_CLASSID;
9 PetscLogEvent MATMFFD_Mult;
10 
11 static PetscBool MatMFFDPackageInitialized = PETSC_FALSE;
12 #undef __FUNCT__
13 #define __FUNCT__ "MatMFFDFinalizePackage"
14 /*@C
15   MatMFFDFinalizePackage - This function destroys everything in the MatMFFD package. It is
16   called from PetscFinalize().
17 
18   Level: developer
19 
20 .keywords: Petsc, destroy, package
21 .seealso: PetscFinalize()
22 @*/
23 PetscErrorCode  MatMFFDFinalizePackage(void)
24 {
25   PetscErrorCode ierr;
26 
27   PetscFunctionBegin;
28   ierr = PetscFunctionListDestroy(&MatMFFDList);CHKERRQ(ierr);
29   MatMFFDPackageInitialized = PETSC_FALSE;
30   MatMFFDRegisterAllCalled  = PETSC_FALSE;
31   PetscFunctionReturn(0);
32 }
33 
34 #undef __FUNCT__
35 #define __FUNCT__ "MatMFFDInitializePackage"
36 /*@C
37   MatMFFDInitializePackage - This function initializes everything in the MatMFFD package. It is called
38   from PetscDLLibraryRegister() when using dynamic libraries, and on the first call to MatCreate_MFFD()
39   when using static libraries.
40 
41   Level: developer
42 
43 .keywords: Vec, initialize, package
44 .seealso: PetscInitialize()
45 @*/
46 PetscErrorCode  MatMFFDInitializePackage(void)
47 {
48   char           logList[256];
49   char           *className;
50   PetscBool      opt;
51   PetscErrorCode ierr;
52 
53   PetscFunctionBegin;
54   if (MatMFFDPackageInitialized) PetscFunctionReturn(0);
55   MatMFFDPackageInitialized = PETSC_TRUE;
56   /* Register Classes */
57   ierr = PetscClassIdRegister("MatMFFD",&MATMFFD_CLASSID);CHKERRQ(ierr);
58   /* Register Constructors */
59   ierr = MatMFFDRegisterAll();CHKERRQ(ierr);
60   /* Register Events */
61   ierr = PetscLogEventRegister("MatMult MF",          MATMFFD_CLASSID,&MATMFFD_Mult);CHKERRQ(ierr);
62 
63   /* Process info exclusions */
64   ierr = PetscOptionsGetString(NULL, "-info_exclude", logList, 256, &opt);CHKERRQ(ierr);
65   if (opt) {
66     ierr = PetscStrstr(logList, "matmffd", &className);CHKERRQ(ierr);
67     if (className) {
68       ierr = PetscInfoDeactivateClass(MATMFFD_CLASSID);CHKERRQ(ierr);
69     }
70   }
71   /* Process summary exclusions */
72   ierr = PetscOptionsGetString(NULL, "-log_summary_exclude", logList, 256, &opt);CHKERRQ(ierr);
73   if (opt) {
74     ierr = PetscStrstr(logList, "matmffd", &className);CHKERRQ(ierr);
75     if (className) {
76       ierr = PetscLogEventDeactivateClass(MATMFFD_CLASSID);CHKERRQ(ierr);
77     }
78   }
79   ierr = PetscRegisterFinalize(MatMFFDFinalizePackage);CHKERRQ(ierr);
80   PetscFunctionReturn(0);
81 }
82 
83 #undef __FUNCT__
84 #define __FUNCT__ "MatMFFDSetType"
85 /*@C
86     MatMFFDSetType - Sets the method that is used to compute the
87     differencing parameter for finite differene matrix-free formulations.
88 
89     Input Parameters:
90 +   mat - the "matrix-free" matrix created via MatCreateSNESMF(), or MatCreateMFFD()
91           or MatSetType(mat,MATMFFD);
92 -   ftype - the type requested, either MATMFFD_WP or MATMFFD_DS
93 
94     Level: advanced
95 
96     Notes:
97     For example, such routines can compute h for use in
98     Jacobian-vector products of the form
99 
100                         F(x+ha) - F(x)
101           F'(u)a  ~=  ----------------
102                               h
103 
104 .seealso: MatCreateSNESMF(), MatMFFDRegister(), MatMFFDSetFunction()
105 @*/
106 PetscErrorCode  MatMFFDSetType(Mat mat,MatMFFDType ftype)
107 {
108   PetscErrorCode ierr,(*r)(MatMFFD);
109   MatMFFD        ctx = (MatMFFD)mat->data;
110   PetscBool      match;
111 
112   PetscFunctionBegin;
113   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
114   PetscValidCharPointer(ftype,2);
115 
116   ierr = PetscObjectTypeCompare((PetscObject)mat,MATMFFD,&match);CHKERRQ(ierr);
117   if (!match) PetscFunctionReturn(0);
118 
119   /* already set, so just return */
120   ierr = PetscObjectTypeCompare((PetscObject)ctx,ftype,&match);CHKERRQ(ierr);
121   if (match) PetscFunctionReturn(0);
122 
123   /* destroy the old one if it exists */
124   if (ctx->ops->destroy) {
125     ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr);
126   }
127 
128   ierr =  PetscFunctionListFind(MatMFFDList,ftype,&r);CHKERRQ(ierr);
129   if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unknown MatMFFD type %s given",ftype);
130   ierr = (*r)(ctx);CHKERRQ(ierr);
131   ierr = PetscObjectChangeTypeName((PetscObject)ctx,ftype);CHKERRQ(ierr);
132   PetscFunctionReturn(0);
133 }
134 
135 typedef PetscErrorCode (*FCN1)(void*,Vec); /* force argument to next function to not be extern C*/
136 #undef __FUNCT__
137 #define __FUNCT__ "MatMFFDSetFunctioniBase_MFFD"
138 PetscErrorCode  MatMFFDSetFunctioniBase_MFFD(Mat mat,FCN1 func)
139 {
140   MatMFFD ctx = (MatMFFD)mat->data;
141 
142   PetscFunctionBegin;
143   ctx->funcisetbase = func;
144   PetscFunctionReturn(0);
145 }
146 
147 typedef PetscErrorCode (*FCN2)(void*,PetscInt,Vec,PetscScalar*); /* force argument to next function to not be extern C*/
148 #undef __FUNCT__
149 #define __FUNCT__ "MatMFFDSetFunctioni_MFFD"
150 PetscErrorCode  MatMFFDSetFunctioni_MFFD(Mat mat,FCN2 funci)
151 {
152   MatMFFD ctx = (MatMFFD)mat->data;
153 
154   PetscFunctionBegin;
155   ctx->funci = funci;
156   PetscFunctionReturn(0);
157 }
158 
159 #undef __FUNCT__
160 #define __FUNCT__ "MatMFFDResetHHistory_MFFD"
161 PetscErrorCode  MatMFFDResetHHistory_MFFD(Mat J)
162 {
163   MatMFFD ctx = (MatMFFD)J->data;
164 
165   PetscFunctionBegin;
166   ctx->ncurrenth = 0;
167   PetscFunctionReturn(0);
168 }
169 
170 #undef __FUNCT__
171 #define __FUNCT__ "MatMFFDRegister"
172 /*@C
173    MatMFFDRegister - Adds a method to the MatMFFD registry.
174 
175    Not Collective
176 
177    Input Parameters:
178 +  name_solver - name of a new user-defined compute-h module
179 -  routine_create - routine to create method context
180 
181    Level: developer
182 
183    Notes:
184    MatMFFDRegister() may be called multiple times to add several user-defined solvers.
185 
186    Sample usage:
187 .vb
188    MatMFFDRegister("my_h",MyHCreate);
189 .ve
190 
191    Then, your solver can be chosen with the procedural interface via
192 $     MatMFFDSetType(mfctx,"my_h")
193    or at runtime via the option
194 $     -mat_mffd_type my_h
195 
196 .keywords: MatMFFD, register
197 
198 .seealso: MatMFFDRegisterAll(), MatMFFDRegisterDestroy()
199  @*/
200 PetscErrorCode  MatMFFDRegister(const char sname[],PetscErrorCode (*function)(MatMFFD))
201 {
202   PetscErrorCode ierr;
203 
204   PetscFunctionBegin;
205   ierr = PetscFunctionListAdd(&MatMFFDList,sname,function);CHKERRQ(ierr);
206   PetscFunctionReturn(0);
207 }
208 
209 /* ----------------------------------------------------------------------------------------*/
210 #undef __FUNCT__
211 #define __FUNCT__ "MatDestroy_MFFD"
212 PetscErrorCode MatDestroy_MFFD(Mat mat)
213 {
214   PetscErrorCode ierr;
215   MatMFFD        ctx = (MatMFFD)mat->data;
216 
217   PetscFunctionBegin;
218   ierr = VecDestroy(&ctx->w);CHKERRQ(ierr);
219   ierr = VecDestroy(&ctx->drscale);CHKERRQ(ierr);
220   ierr = VecDestroy(&ctx->dlscale);CHKERRQ(ierr);
221   ierr = VecDestroy(&ctx->dshift);CHKERRQ(ierr);
222   if (ctx->current_f_allocated) {
223     ierr = VecDestroy(&ctx->current_f);CHKERRQ(ierr);
224   }
225   if (ctx->ops->destroy) {ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr);}
226   ierr      = PetscHeaderDestroy(&ctx);CHKERRQ(ierr);
227   mat->data = 0;
228 
229   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetBase_C",NULL);CHKERRQ(ierr);
230   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctioniBase_C",NULL);CHKERRQ(ierr);
231   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctioni_C",NULL);CHKERRQ(ierr);
232   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunction_C",NULL);CHKERRQ(ierr);
233   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetFunctionError_C",NULL);CHKERRQ(ierr);
234   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetCheckh_C",NULL);CHKERRQ(ierr);
235   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDSetPeriod_C",NULL);CHKERRQ(ierr);
236   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMFFDResetHHistory_C",NULL);CHKERRQ(ierr);
237   PetscFunctionReturn(0);
238 }
239 
240 #undef __FUNCT__
241 #define __FUNCT__ "MatView_MFFD"
242 /*
243    MatMFFDView_MFFD - Views matrix-free parameters.
244 
245 */
246 PetscErrorCode MatView_MFFD(Mat J,PetscViewer viewer)
247 {
248   PetscErrorCode ierr;
249   MatMFFD        ctx = (MatMFFD)J->data;
250   PetscBool      iascii, viewbase, viewfunction;
251   const char     *prefix;
252 
253   PetscFunctionBegin;
254   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
255   if (iascii) {
256     ierr = PetscViewerASCIIPrintf(viewer,"Matrix-free approximation:\n");CHKERRQ(ierr);
257     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
258     ierr = PetscViewerASCIIPrintf(viewer,"err=%g (relative error in function evaluation)\n",(double)ctx->error_rel);CHKERRQ(ierr);
259     if (!((PetscObject)ctx)->type_name) {
260       ierr = PetscViewerASCIIPrintf(viewer,"The compute h routine has not yet been set\n");CHKERRQ(ierr);
261     } else {
262       ierr = PetscViewerASCIIPrintf(viewer,"Using %s compute h routine\n",((PetscObject)ctx)->type_name);CHKERRQ(ierr);
263     }
264     if (ctx->ops->view) {
265       ierr = (*ctx->ops->view)(ctx,viewer);CHKERRQ(ierr);
266     }
267     ierr = PetscObjectGetOptionsPrefix((PetscObject)J, &prefix);CHKERRQ(ierr);
268 
269     ierr = PetscOptionsHasName(prefix, "-mat_mffd_view_base", &viewbase);CHKERRQ(ierr);
270     if (viewbase) {
271       ierr = PetscViewerASCIIPrintf(viewer, "Base:\n");CHKERRQ(ierr);
272       ierr = VecView(ctx->current_u, viewer);CHKERRQ(ierr);
273     }
274     ierr = PetscOptionsHasName(prefix, "-mat_mffd_view_function", &viewfunction);CHKERRQ(ierr);
275     if (viewfunction) {
276       ierr = PetscViewerASCIIPrintf(viewer, "Function:\n");CHKERRQ(ierr);
277       ierr = VecView(ctx->current_f, viewer);CHKERRQ(ierr);
278     }
279     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
280   }
281   PetscFunctionReturn(0);
282 }
283 
284 #undef __FUNCT__
285 #define __FUNCT__ "MatAssemblyEnd_MFFD"
286 /*
287    MatAssemblyEnd_MFFD - Resets the ctx->ncurrenth to zero. This
288    allows the user to indicate the beginning of a new linear solve by calling
289    MatAssemblyXXX() on the matrix free matrix. This then allows the
290    MatCreateMFFD_WP() to properly compute ||U|| only the first time
291    in the linear solver rather than every time.
292 
293    This function is referenced directly from MatAssemblyEnd_SNESMF(), which may be in a different shared library.
294 */
295 PETSC_EXTERN PetscErrorCode MatAssemblyEnd_MFFD(Mat J,MatAssemblyType mt)
296 {
297   PetscErrorCode ierr;
298   MatMFFD        j = (MatMFFD)J->data;
299 
300   PetscFunctionBegin;
301   ierr      = MatMFFDResetHHistory(J);CHKERRQ(ierr);
302   j->vshift = 0.0;
303   j->vscale = 1.0;
304   PetscFunctionReturn(0);
305 }
306 
307 #undef __FUNCT__
308 #define __FUNCT__ "MatMult_MFFD"
309 /*
310   MatMult_MFFD - Default matrix-free form for Jacobian-vector product, y = F'(u)*a:
311 
312         y ~= (F(u + ha) - F(u))/h,
313   where F = nonlinear function, as set by SNESSetFunction()
314         u = current iterate
315         h = difference interval
316 */
317 PetscErrorCode MatMult_MFFD(Mat mat,Vec a,Vec y)
318 {
319   MatMFFD        ctx = (MatMFFD)mat->data;
320   PetscScalar    h;
321   Vec            w,U,F;
322   PetscErrorCode ierr;
323   PetscBool      zeroa;
324 
325   PetscFunctionBegin;
326   if (!ctx->current_u) SETERRQ(PetscObjectComm((PetscObject)mat),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");
327   /* We log matrix-free matrix-vector products separately, so that we can
328      separate the performance monitoring from the cases that use conventional
329      storage.  We may eventually modify event logging to associate events
330      with particular objects, hence alleviating the more general problem. */
331   ierr = PetscLogEventBegin(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr);
332 
333   w = ctx->w;
334   U = ctx->current_u;
335   F = ctx->current_f;
336   /*
337       Compute differencing parameter
338   */
339   if (!ctx->ops->compute) {
340     ierr = MatMFFDSetType(mat,MATMFFD_WP);CHKERRQ(ierr);
341     ierr = MatSetFromOptions(mat);CHKERRQ(ierr);
342   }
343   ierr = (*ctx->ops->compute)(ctx,U,a,&h,&zeroa);CHKERRQ(ierr);
344   if (zeroa) {
345     ierr = VecSet(y,0.0);CHKERRQ(ierr);
346     PetscFunctionReturn(0);
347   }
348 
349   if (mat->erroriffpe && PetscIsInfOrNanScalar(h)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Computed Nan differencing parameter h");
350   if (ctx->checkh) {
351     ierr = (*ctx->checkh)(ctx->checkhctx,U,a,&h);CHKERRQ(ierr);
352   }
353 
354   /* keep a record of the current differencing parameter h */
355   ctx->currenth = h;
356 #if defined(PETSC_USE_COMPLEX)
357   ierr = PetscInfo2(mat,"Current differencing parameter: %g + %g i\n",(double)PetscRealPart(h),(double)PetscImaginaryPart(h));CHKERRQ(ierr);
358 #else
359   ierr = PetscInfo1(mat,"Current differencing parameter: %15.12e\n",h);CHKERRQ(ierr);
360 #endif
361   if (ctx->historyh && ctx->ncurrenth < ctx->maxcurrenth) {
362     ctx->historyh[ctx->ncurrenth] = h;
363   }
364   ctx->ncurrenth++;
365 
366   /* w = u + ha */
367   if (ctx->drscale) {
368     ierr = VecPointwiseMult(ctx->drscale,a,U);CHKERRQ(ierr);
369     ierr = VecAYPX(U,h,w);CHKERRQ(ierr);
370   } else {
371     ierr = VecWAXPY(w,h,a,U);CHKERRQ(ierr);
372   }
373 
374   /* compute func(U) as base for differencing; only needed first time in and not when provided by user */
375   if (ctx->ncurrenth == 1 && ctx->current_f_allocated) {
376     ierr = (*ctx->func)(ctx->funcctx,U,F);CHKERRQ(ierr);
377   }
378   ierr = (*ctx->func)(ctx->funcctx,w,y);CHKERRQ(ierr);
379 
380   ierr = VecAXPY(y,-1.0,F);CHKERRQ(ierr);
381   ierr = VecScale(y,1.0/h);CHKERRQ(ierr);
382 
383   if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) {
384     ierr = VecAXPBY(y,ctx->vshift,ctx->vscale,a);CHKERRQ(ierr);
385   }
386   if (ctx->dlscale) {
387     ierr = VecPointwiseMult(y,ctx->dlscale,y);CHKERRQ(ierr);
388   }
389   if (ctx->dshift) {
390     ierr = VecPointwiseMult(ctx->dshift,a,U);CHKERRQ(ierr);
391     ierr = VecAXPY(y,1.0,U);CHKERRQ(ierr);
392   }
393 
394   if (mat->nullsp) {ierr = MatNullSpaceRemove(mat->nullsp,y);CHKERRQ(ierr);}
395 
396   ierr = PetscLogEventEnd(MATMFFD_Mult,a,y,0,0);CHKERRQ(ierr);
397   PetscFunctionReturn(0);
398 }
399 
400 #undef __FUNCT__
401 #define __FUNCT__ "MatGetDiagonal_MFFD"
402 /*
403   MatGetDiagonal_MFFD - Gets the diagonal for a matrix free matrix
404 
405         y ~= (F(u + ha) - F(u))/h,
406   where F = nonlinear function, as set by SNESSetFunction()
407         u = current iterate
408         h = difference interval
409 */
410 PetscErrorCode MatGetDiagonal_MFFD(Mat mat,Vec a)
411 {
412   MatMFFD        ctx = (MatMFFD)mat->data;
413   PetscScalar    h,*aa,*ww,v;
414   PetscReal      epsilon = PETSC_SQRT_MACHINE_EPSILON,umin = 100.0*PETSC_SQRT_MACHINE_EPSILON;
415   Vec            w,U;
416   PetscErrorCode ierr;
417   PetscInt       i,rstart,rend;
418 
419   PetscFunctionBegin;
420   if (!ctx->funci) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Requires calling MatMFFDSetFunctioni() first");
421 
422   w    = ctx->w;
423   U    = ctx->current_u;
424   ierr = (*ctx->func)(ctx->funcctx,U,a);CHKERRQ(ierr);
425   ierr = (*ctx->funcisetbase)(ctx->funcctx,U);CHKERRQ(ierr);
426   ierr = VecCopy(U,w);CHKERRQ(ierr);
427 
428   ierr = VecGetOwnershipRange(a,&rstart,&rend);CHKERRQ(ierr);
429   ierr = VecGetArray(a,&aa);CHKERRQ(ierr);
430   for (i=rstart; i<rend; i++) {
431     ierr = VecGetArray(w,&ww);CHKERRQ(ierr);
432     h    = ww[i-rstart];
433     if (h == 0.0) h = 1.0;
434     if (PetscAbsScalar(h) < umin && PetscRealPart(h) >= 0.0)     h = umin;
435     else if (PetscRealPart(h) < 0.0 && PetscAbsScalar(h) < umin) h = -umin;
436     h *= epsilon;
437 
438     ww[i-rstart] += h;
439     ierr          = VecRestoreArray(w,&ww);CHKERRQ(ierr);
440     ierr          = (*ctx->funci)(ctx->funcctx,i,w,&v);CHKERRQ(ierr);
441     aa[i-rstart]  = (v - aa[i-rstart])/h;
442 
443     /* possibly shift and scale result */
444     if ((ctx->vshift != 0.0) || (ctx->vscale != 1.0)) {
445       aa[i - rstart] = ctx->vshift + ctx->vscale*aa[i-rstart];
446     }
447 
448     ierr          = VecGetArray(w,&ww);CHKERRQ(ierr);
449     ww[i-rstart] -= h;
450     ierr          = VecRestoreArray(w,&ww);CHKERRQ(ierr);
451   }
452   ierr = VecRestoreArray(a,&aa);CHKERRQ(ierr);
453   PetscFunctionReturn(0);
454 }
455 
456 #undef __FUNCT__
457 #define __FUNCT__ "MatDiagonalScale_MFFD"
458 PetscErrorCode MatDiagonalScale_MFFD(Mat mat,Vec ll,Vec rr)
459 {
460   MatMFFD        aij = (MatMFFD)mat->data;
461   PetscErrorCode ierr;
462 
463   PetscFunctionBegin;
464   if (ll && !aij->dlscale) {
465     ierr = VecDuplicate(ll,&aij->dlscale);CHKERRQ(ierr);
466   }
467   if (rr && !aij->drscale) {
468     ierr = VecDuplicate(rr,&aij->drscale);CHKERRQ(ierr);
469   }
470   if (ll) {
471     ierr = VecCopy(ll,aij->dlscale);CHKERRQ(ierr);
472   }
473   if (rr) {
474     ierr = VecCopy(rr,aij->drscale);CHKERRQ(ierr);
475   }
476   PetscFunctionReturn(0);
477 }
478 
479 #undef __FUNCT__
480 #define __FUNCT__ "MatDiagonalSet_MFFD"
481 PetscErrorCode MatDiagonalSet_MFFD(Mat mat,Vec ll,InsertMode mode)
482 {
483   MatMFFD        aij = (MatMFFD)mat->data;
484   PetscErrorCode ierr;
485 
486   PetscFunctionBegin;
487   if (mode == INSERT_VALUES) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"No diagonal set with INSERT_VALUES");
488   if (!aij->dshift) {
489     ierr = VecDuplicate(ll,&aij->dshift);CHKERRQ(ierr);
490   }
491   ierr = VecAXPY(aij->dshift,1.0,ll);CHKERRQ(ierr);
492   PetscFunctionReturn(0);
493 }
494 
495 #undef __FUNCT__
496 #define __FUNCT__ "MatShift_MFFD"
497 PetscErrorCode MatShift_MFFD(Mat Y,PetscScalar a)
498 {
499   MatMFFD shell = (MatMFFD)Y->data;
500 
501   PetscFunctionBegin;
502   shell->vshift += a;
503   PetscFunctionReturn(0);
504 }
505 
506 #undef __FUNCT__
507 #define __FUNCT__ "MatScale_MFFD"
508 PetscErrorCode MatScale_MFFD(Mat Y,PetscScalar a)
509 {
510   MatMFFD shell = (MatMFFD)Y->data;
511 
512   PetscFunctionBegin;
513   shell->vscale *= a;
514   PetscFunctionReturn(0);
515 }
516 
517 #undef __FUNCT__
518 #define __FUNCT__ "MatMFFDSetBase_MFFD"
519 PETSC_EXTERN PetscErrorCode MatMFFDSetBase_MFFD(Mat J,Vec U,Vec F)
520 {
521   PetscErrorCode ierr;
522   MatMFFD        ctx = (MatMFFD)J->data;
523 
524   PetscFunctionBegin;
525   ierr = MatMFFDResetHHistory(J);CHKERRQ(ierr);
526 
527   ctx->current_u = U;
528   if (F) {
529     if (ctx->current_f_allocated) {ierr = VecDestroy(&ctx->current_f);CHKERRQ(ierr);}
530     ctx->current_f           = F;
531     ctx->current_f_allocated = PETSC_FALSE;
532   } else if (!ctx->current_f_allocated) {
533     ierr = MatCreateVecs(J,NULL,&ctx->current_f);CHKERRQ(ierr);
534 
535     ctx->current_f_allocated = PETSC_TRUE;
536   }
537   if (!ctx->w) {
538     ierr = VecDuplicate(ctx->current_u, &ctx->w);CHKERRQ(ierr);
539   }
540   J->assembled = PETSC_TRUE;
541   PetscFunctionReturn(0);
542 }
543 
544 typedef PetscErrorCode (*FCN3)(void*,Vec,Vec,PetscScalar*); /* force argument to next function to not be extern C*/
545 
546 #undef __FUNCT__
547 #define __FUNCT__ "MatMFFDSetCheckh_MFFD"
548 PetscErrorCode  MatMFFDSetCheckh_MFFD(Mat J,FCN3 fun,void *ectx)
549 {
550   MatMFFD ctx = (MatMFFD)J->data;
551 
552   PetscFunctionBegin;
553   ctx->checkh    = fun;
554   ctx->checkhctx = ectx;
555   PetscFunctionReturn(0);
556 }
557 
558 #undef __FUNCT__
559 #define __FUNCT__ "MatMFFDSetOptionsPrefix"
560 /*@C
561    MatMFFDSetOptionsPrefix - Sets the prefix used for searching for all
562    MatMFFD options in the database.
563 
564    Collective on Mat
565 
566    Input Parameter:
567 +  A - the Mat context
568 -  prefix - the prefix to prepend to all option names
569 
570    Notes:
571    A hyphen (-) must NOT be given at the beginning of the prefix name.
572    The first character of all runtime options is AUTOMATICALLY the hyphen.
573 
574    Level: advanced
575 
576 .keywords: SNES, matrix-free, parameters
577 
578 .seealso: MatSetFromOptions(), MatCreateSNESMF()
579 @*/
580 PetscErrorCode  MatMFFDSetOptionsPrefix(Mat mat,const char prefix[])
581 
582 {
583   MatMFFD        mfctx = mat ? (MatMFFD)mat->data : (MatMFFD)NULL;
584   PetscErrorCode ierr;
585 
586   PetscFunctionBegin;
587   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
588   PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1);
589   ierr = PetscObjectSetOptionsPrefix((PetscObject)mfctx,prefix);CHKERRQ(ierr);
590   PetscFunctionReturn(0);
591 }
592 
593 #undef __FUNCT__
594 #define __FUNCT__ "MatSetFromOptions_MFFD"
595 PetscErrorCode  MatSetFromOptions_MFFD(PetscOptions *PetscOptionsObject,Mat mat)
596 {
597   MatMFFD        mfctx = (MatMFFD)mat->data;
598   PetscErrorCode ierr;
599   PetscBool      flg;
600   char           ftype[256];
601 
602   PetscFunctionBegin;
603   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
604   PetscValidHeaderSpecific(mfctx,MATMFFD_CLASSID,1);
605   ierr = PetscObjectOptionsBegin((PetscObject)mfctx);CHKERRQ(ierr);
606   ierr = PetscOptionsFList("-mat_mffd_type","Matrix free type","MatMFFDSetType",MatMFFDList,((PetscObject)mfctx)->type_name,ftype,256,&flg);CHKERRQ(ierr);
607   if (flg) {
608     ierr = MatMFFDSetType(mat,ftype);CHKERRQ(ierr);
609   }
610 
611   ierr = PetscOptionsReal("-mat_mffd_err","set sqrt relative error in function","MatMFFDSetFunctionError",mfctx->error_rel,&mfctx->error_rel,0);CHKERRQ(ierr);
612   ierr = PetscOptionsInt("-mat_mffd_period","how often h is recomputed","MatMFFDSetPeriod",mfctx->recomputeperiod,&mfctx->recomputeperiod,0);CHKERRQ(ierr);
613 
614   flg  = PETSC_FALSE;
615   ierr = PetscOptionsBool("-mat_mffd_check_positivity","Insure that U + h*a is nonnegative","MatMFFDSetCheckh",flg,&flg,NULL);CHKERRQ(ierr);
616   if (flg) {
617     ierr = MatMFFDSetCheckh(mat,MatMFFDCheckPositivity,0);CHKERRQ(ierr);
618   }
619   if (mfctx->ops->setfromoptions) {
620     ierr = (*mfctx->ops->setfromoptions)(PetscOptionsObject,mfctx);CHKERRQ(ierr);
621   }
622   ierr = PetscOptionsEnd();CHKERRQ(ierr);
623   PetscFunctionReturn(0);
624 }
625 
626 #undef __FUNCT__
627 #define __FUNCT__ "MatMFFDSetPeriod_MFFD"
628 PetscErrorCode  MatMFFDSetPeriod_MFFD(Mat mat,PetscInt period)
629 {
630   MatMFFD ctx = (MatMFFD)mat->data;
631 
632   PetscFunctionBegin;
633   PetscValidLogicalCollectiveInt(mat,period,2);
634   ctx->recomputeperiod = period;
635   PetscFunctionReturn(0);
636 }
637 
638 #undef __FUNCT__
639 #define __FUNCT__ "MatMFFDSetFunction_MFFD"
640 PetscErrorCode  MatMFFDSetFunction_MFFD(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx)
641 {
642   MatMFFD ctx = (MatMFFD)mat->data;
643 
644   PetscFunctionBegin;
645   ctx->func    = func;
646   ctx->funcctx = funcctx;
647   PetscFunctionReturn(0);
648 }
649 
650 #undef __FUNCT__
651 #define __FUNCT__ "MatMFFDSetFunctionError_MFFD"
652 PetscErrorCode  MatMFFDSetFunctionError_MFFD(Mat mat,PetscReal error)
653 {
654   MatMFFD ctx = (MatMFFD)mat->data;
655 
656   PetscFunctionBegin;
657   PetscValidLogicalCollectiveReal(mat,error,2);
658   if (error != PETSC_DEFAULT) ctx->error_rel = error;
659   PetscFunctionReturn(0);
660 }
661 
662 /*MC
663   MATMFFD - MATMFFD = "mffd" - A matrix free matrix type.
664 
665   Level: advanced
666 
667 .seealso: MatCreateMFFD(), MatCreateSNESMF(), MatMFFDSetFunction()
668 M*/
669 #undef __FUNCT__
670 #define __FUNCT__ "MatCreate_MFFD"
671 PETSC_EXTERN PetscErrorCode MatCreate_MFFD(Mat A)
672 {
673   MatMFFD        mfctx;
674   PetscErrorCode ierr;
675 
676   PetscFunctionBegin;
677   ierr = MatMFFDInitializePackage();CHKERRQ(ierr);
678 
679   ierr = PetscHeaderCreate(mfctx,MATMFFD_CLASSID,"MatMFFD","Matrix-free Finite Differencing","Mat",PetscObjectComm((PetscObject)A),MatDestroy_MFFD,MatView_MFFD);CHKERRQ(ierr);
680 
681   mfctx->error_rel                = PETSC_SQRT_MACHINE_EPSILON;
682   mfctx->recomputeperiod          = 1;
683   mfctx->count                    = 0;
684   mfctx->currenth                 = 0.0;
685   mfctx->historyh                 = NULL;
686   mfctx->ncurrenth                = 0;
687   mfctx->maxcurrenth              = 0;
688   ((PetscObject)mfctx)->type_name = 0;
689 
690   mfctx->vshift = 0.0;
691   mfctx->vscale = 1.0;
692 
693   /*
694      Create the empty data structure to contain compute-h routines.
695      These will be filled in below from the command line options or
696      a later call with MatMFFDSetType() or if that is not called
697      then it will default in the first use of MatMult_MFFD()
698   */
699   mfctx->ops->compute        = 0;
700   mfctx->ops->destroy        = 0;
701   mfctx->ops->view           = 0;
702   mfctx->ops->setfromoptions = 0;
703   mfctx->hctx                = 0;
704 
705   mfctx->func    = 0;
706   mfctx->funcctx = 0;
707   mfctx->w       = NULL;
708 
709   A->data = mfctx;
710 
711   A->ops->mult           = MatMult_MFFD;
712   A->ops->destroy        = MatDestroy_MFFD;
713   A->ops->view           = MatView_MFFD;
714   A->ops->assemblyend    = MatAssemblyEnd_MFFD;
715   A->ops->getdiagonal    = MatGetDiagonal_MFFD;
716   A->ops->scale          = MatScale_MFFD;
717   A->ops->shift          = MatShift_MFFD;
718   A->ops->diagonalscale  = MatDiagonalScale_MFFD;
719   A->ops->diagonalset    = MatDiagonalSet_MFFD;
720   A->ops->setfromoptions = MatSetFromOptions_MFFD;
721   A->assembled           = PETSC_TRUE;
722 
723   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
724   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
725 
726   ierr = PetscObjectComposeFunction((PetscObject)A,"MatMFFDSetBase_C",MatMFFDSetBase_MFFD);CHKERRQ(ierr);
727   ierr = PetscObjectComposeFunction((PetscObject)A,"MatMFFDSetFunctioniBase_C",MatMFFDSetFunctioniBase_MFFD);CHKERRQ(ierr);
728   ierr = PetscObjectComposeFunction((PetscObject)A,"MatMFFDSetFunctioni_C",MatMFFDSetFunctioni_MFFD);CHKERRQ(ierr);
729   ierr = PetscObjectComposeFunction((PetscObject)A,"MatMFFDSetFunction_C",MatMFFDSetFunction_MFFD);CHKERRQ(ierr);
730   ierr = PetscObjectComposeFunction((PetscObject)A,"MatMFFDSetCheckh_C",MatMFFDSetCheckh_MFFD);CHKERRQ(ierr);
731   ierr = PetscObjectComposeFunction((PetscObject)A,"MatMFFDSetPeriod_C",MatMFFDSetPeriod_MFFD);CHKERRQ(ierr);
732   ierr = PetscObjectComposeFunction((PetscObject)A,"MatMFFDSetFunctionError_C",MatMFFDSetFunctionError_MFFD);CHKERRQ(ierr);
733   ierr = PetscObjectComposeFunction((PetscObject)A,"MatMFFDResetHHistory_C",MatMFFDResetHHistory_MFFD);CHKERRQ(ierr);
734 
735   mfctx->mat = A;
736 
737   ierr = PetscObjectChangeTypeName((PetscObject)A,MATMFFD);CHKERRQ(ierr);
738   PetscFunctionReturn(0);
739 }
740 
741 #undef __FUNCT__
742 #define __FUNCT__ "MatCreateMFFD"
743 /*@
744    MatCreateMFFD - Creates a matrix-free matrix. See also MatCreateSNESMF()
745 
746    Collective on Vec
747 
748    Input Parameters:
749 +  comm - MPI communicator
750 .  m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
751            This value should be the same as the local size used in creating the
752            y vector for the matrix-vector product y = Ax.
753 .  n - This value should be the same as the local size used in creating the
754        x vector for the matrix-vector product y = Ax. (or PETSC_DECIDE to have
755        calculated if N is given) For square matrices n is almost always m.
756 .  M - number of global rows (or PETSC_DETERMINE to have calculated if m is given)
757 -  N - number of global columns (or PETSC_DETERMINE to have calculated if n is given)
758 
759 
760    Output Parameter:
761 .  J - the matrix-free matrix
762 
763    Options Database Keys: call MatSetFromOptions() to trigger these
764 +  -mat_mffd_type - wp or ds (see MATMFFD_WP or MATMFFD_DS)
765 -  -mat_mffd_err - square root of estimated relative error in function evaluation
766 -  -mat_mffd_period - how often h is recomputed, defaults to 1, everytime
767 
768 
769    Level: advanced
770 
771    Notes:
772    The matrix-free matrix context merely contains the function pointers
773    and work space for performing finite difference approximations of
774    Jacobian-vector products, F'(u)*a,
775 
776    The default code uses the following approach to compute h
777 
778 .vb
779      F'(u)*a = [F(u+h*a) - F(u)]/h where
780      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
781        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   otherwise
782  where
783      error_rel = square root of relative error in function evaluation
784      umin = minimum iterate parameter
785 .ve
786 
787    You can call SNESSetJacobian() with MatMFFDComputeJacobian() if you are using matrix and not a different
788    preconditioner matrix
789 
790    The user can set the error_rel via MatMFFDSetFunctionError() and
791    umin via MatMFFDDSSetUmin(); see Users-Manual: ch_snes for details.
792 
793    The user should call MatDestroy() when finished with the matrix-free
794    matrix context.
795 
796    Options Database Keys:
797 +  -mat_mffd_err <error_rel> - Sets error_rel
798 .  -mat_mffd_unim <umin> - Sets umin (for default PETSc routine that computes h only)
799 -  -mat_mffd_check_positivity
800 
801 .keywords: default, matrix-free, create, matrix
802 
803 .seealso: MatDestroy(), MatMFFDSetFunctionError(), MatMFFDDSSetUmin(), MatMFFDSetFunction()
804           MatMFFDSetHHistory(), MatMFFDResetHHistory(), MatCreateSNESMF(),
805           MatMFFDGetH(), MatMFFDRegister(), MatMFFDComputeJacobian()
806 
807 @*/
808 PetscErrorCode  MatCreateMFFD(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,Mat *J)
809 {
810   PetscErrorCode ierr;
811 
812   PetscFunctionBegin;
813   ierr = MatCreate(comm,J);CHKERRQ(ierr);
814   ierr = MatSetSizes(*J,m,n,M,N);CHKERRQ(ierr);
815   ierr = MatSetType(*J,MATMFFD);CHKERRQ(ierr);
816   ierr = MatSetUp(*J);CHKERRQ(ierr);
817   PetscFunctionReturn(0);
818 }
819 
820 
821 #undef __FUNCT__
822 #define __FUNCT__ "MatMFFDGetH"
823 /*@
824    MatMFFDGetH - Gets the last value that was used as the differencing
825    parameter.
826 
827    Not Collective
828 
829    Input Parameters:
830 .  mat - the matrix obtained with MatCreateSNESMF()
831 
832    Output Paramter:
833 .  h - the differencing step size
834 
835    Level: advanced
836 
837 .keywords: SNES, matrix-free, parameters
838 
839 .seealso: MatCreateSNESMF(),MatMFFDSetHHistory(), MatCreateMFFD(), MATMFFD, MatMFFDResetHHistory()
840 @*/
841 PetscErrorCode  MatMFFDGetH(Mat mat,PetscScalar *h)
842 {
843   MatMFFD        ctx = (MatMFFD)mat->data;
844   PetscErrorCode ierr;
845   PetscBool      match;
846 
847   PetscFunctionBegin;
848   ierr = PetscObjectTypeCompare((PetscObject)mat,MATMFFD,&match);CHKERRQ(ierr);
849   if (!match) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONG,"Not a MFFD matrix");
850 
851   *h = ctx->currenth;
852   PetscFunctionReturn(0);
853 }
854 
855 #undef __FUNCT__
856 #define __FUNCT__ "MatMFFDSetFunction"
857 /*@C
858    MatMFFDSetFunction - Sets the function used in applying the matrix free.
859 
860    Logically Collective on Mat
861 
862    Input Parameters:
863 +  mat - the matrix free matrix created via MatCreateSNESMF() or MatCreateMFFD()
864 .  func - the function to use
865 -  funcctx - optional function context passed to function
866 
867    Calling Sequence of func:
868 $     func (void *funcctx, Vec x, Vec f)
869 
870 +  funcctx - user provided context
871 .  x - input vector
872 -  f - computed output function
873 
874    Level: advanced
875 
876    Notes:
877     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
878     matrix inside your compute Jacobian routine
879 
880     If this is not set then it will use the function set with SNESSetFunction() if MatCreateSNESMF() was used.
881 
882 .keywords: SNES, matrix-free, function
883 
884 .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD,
885           MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
886 @*/
887 PetscErrorCode  MatMFFDSetFunction(Mat mat,PetscErrorCode (*func)(void*,Vec,Vec),void *funcctx)
888 {
889   PetscErrorCode ierr;
890 
891   PetscFunctionBegin;
892   ierr = PetscTryMethod(mat,"MatMFFDSetFunction_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec),void*),(mat,func,funcctx));CHKERRQ(ierr);
893   PetscFunctionReturn(0);
894 }
895 
896 #undef __FUNCT__
897 #define __FUNCT__ "MatMFFDSetFunctioni"
898 /*@C
899    MatMFFDSetFunctioni - Sets the function for a single component
900 
901    Logically Collective on Mat
902 
903    Input Parameters:
904 +  mat - the matrix free matrix created via MatCreateSNESMF()
905 -  funci - the function to use
906 
907    Level: advanced
908 
909    Notes:
910     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
911     matrix inside your compute Jacobian routine
912 
913 
914 .keywords: SNES, matrix-free, function
915 
916 .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
917 
918 @*/
919 PetscErrorCode  MatMFFDSetFunctioni(Mat mat,PetscErrorCode (*funci)(void*,PetscInt,Vec,PetscScalar*))
920 {
921   PetscErrorCode ierr;
922 
923   PetscFunctionBegin;
924   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
925   ierr = PetscTryMethod(mat,"MatMFFDSetFunctioni_C",(Mat,PetscErrorCode (*)(void*,PetscInt,Vec,PetscScalar*)),(mat,funci));CHKERRQ(ierr);
926   PetscFunctionReturn(0);
927 }
928 
929 
930 #undef __FUNCT__
931 #define __FUNCT__ "MatMFFDSetFunctioniBase"
932 /*@C
933    MatMFFDSetFunctioniBase - Sets the base vector for a single component function evaluation
934 
935    Logically Collective on Mat
936 
937    Input Parameters:
938 +  mat - the matrix free matrix created via MatCreateSNESMF()
939 -  func - the function to use
940 
941    Level: advanced
942 
943    Notes:
944     If you use this you MUST call MatAssemblyBegin()/MatAssemblyEnd() on the matrix free
945     matrix inside your compute Jacobian routine
946 
947 
948 .keywords: SNES, matrix-free, function
949 
950 .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD
951           MatMFFDSetHHistory(), MatMFFDResetHHistory(), SNESetFunction()
952 @*/
953 PetscErrorCode  MatMFFDSetFunctioniBase(Mat mat,PetscErrorCode (*func)(void*,Vec))
954 {
955   PetscErrorCode ierr;
956 
957   PetscFunctionBegin;
958   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
959   ierr = PetscTryMethod(mat,"MatMFFDSetFunctioniBase_C",(Mat,PetscErrorCode (*)(void*,Vec)),(mat,func));CHKERRQ(ierr);
960   PetscFunctionReturn(0);
961 }
962 
963 #undef __FUNCT__
964 #define __FUNCT__ "MatMFFDSetPeriod"
965 /*@
966    MatMFFDSetPeriod - Sets how often h is recomputed, by default it is everytime
967 
968    Logically Collective on Mat
969 
970    Input Parameters:
971 +  mat - the matrix free matrix created via MatCreateSNESMF()
972 -  period - 1 for everytime, 2 for every second etc
973 
974    Options Database Keys:
975 +  -mat_mffd_period <period>
976 
977    Level: advanced
978 
979 
980 .keywords: SNES, matrix-free, parameters
981 
982 .seealso: MatCreateSNESMF(),MatMFFDGetH(),
983           MatMFFDSetHHistory(), MatMFFDResetHHistory()
984 @*/
985 PetscErrorCode  MatMFFDSetPeriod(Mat mat,PetscInt period)
986 {
987   PetscErrorCode ierr;
988 
989   PetscFunctionBegin;
990   ierr = PetscTryMethod(mat,"MatMFFDSetPeriod_C",(Mat,PetscInt),(mat,period));CHKERRQ(ierr);
991   PetscFunctionReturn(0);
992 }
993 
994 #undef __FUNCT__
995 #define __FUNCT__ "MatMFFDSetFunctionError"
996 /*@
997    MatMFFDSetFunctionError - Sets the error_rel for the approximation of
998    matrix-vector products using finite differences.
999 
1000    Logically Collective on Mat
1001 
1002    Input Parameters:
1003 +  mat - the matrix free matrix created via MatCreateMFFD() or MatCreateSNESMF()
1004 -  error_rel - relative error (should be set to the square root of
1005                the relative error in the function evaluations)
1006 
1007    Options Database Keys:
1008 +  -mat_mffd_err <error_rel> - Sets error_rel
1009 
1010    Level: advanced
1011 
1012    Notes:
1013    The default matrix-free matrix-vector product routine computes
1014 .vb
1015      F'(u)*a = [F(u+h*a) - F(u)]/h where
1016      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
1017        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   else
1018 .ve
1019 
1020 .keywords: SNES, matrix-free, parameters
1021 
1022 .seealso: MatCreateSNESMF(),MatMFFDGetH(), MatCreateMFFD(), MATMFFD
1023           MatMFFDSetHHistory(), MatMFFDResetHHistory()
1024 @*/
1025 PetscErrorCode  MatMFFDSetFunctionError(Mat mat,PetscReal error)
1026 {
1027   PetscErrorCode ierr;
1028 
1029   PetscFunctionBegin;
1030   ierr = PetscTryMethod(mat,"MatMFFDSetFunctionError_C",(Mat,PetscReal),(mat,error));CHKERRQ(ierr);
1031   PetscFunctionReturn(0);
1032 }
1033 
1034 #undef __FUNCT__
1035 #define __FUNCT__ "MatMFFDSetHHistory"
1036 /*@
1037    MatMFFDSetHHistory - Sets an array to collect a history of the
1038    differencing values (h) computed for the matrix-free product.
1039 
1040    Logically Collective on Mat
1041 
1042    Input Parameters:
1043 +  J - the matrix-free matrix context
1044 .  histroy - space to hold the history
1045 -  nhistory - number of entries in history, if more entries are generated than
1046               nhistory, then the later ones are discarded
1047 
1048    Level: advanced
1049 
1050    Notes:
1051    Use MatMFFDResetHHistory() to reset the history counter and collect
1052    a new batch of differencing parameters, h.
1053 
1054 .keywords: SNES, matrix-free, h history, differencing history
1055 
1056 .seealso: MatMFFDGetH(), MatCreateSNESMF(),
1057           MatMFFDResetHHistory(), MatMFFDSetFunctionError()
1058 
1059 @*/
1060 PetscErrorCode  MatMFFDSetHHistory(Mat J,PetscScalar history[],PetscInt nhistory)
1061 {
1062   MatMFFD        ctx = (MatMFFD)J->data;
1063   PetscErrorCode ierr;
1064   PetscBool      match;
1065 
1066   PetscFunctionBegin;
1067   ierr = PetscObjectTypeCompare((PetscObject)J,MATMFFD,&match);CHKERRQ(ierr);
1068   if (!match) SETERRQ(PetscObjectComm((PetscObject)J),PETSC_ERR_ARG_WRONG,"Not a MFFD matrix");
1069   ctx->historyh    = history;
1070   ctx->maxcurrenth = nhistory;
1071   ctx->currenth    = 0.;
1072   PetscFunctionReturn(0);
1073 }
1074 
1075 
1076 #undef __FUNCT__
1077 #define __FUNCT__ "MatMFFDResetHHistory"
1078 /*@
1079    MatMFFDResetHHistory - Resets the counter to zero to begin
1080    collecting a new set of differencing histories.
1081 
1082    Logically Collective on Mat
1083 
1084    Input Parameters:
1085 .  J - the matrix-free matrix context
1086 
1087    Level: advanced
1088 
1089    Notes:
1090    Use MatMFFDSetHHistory() to create the original history counter.
1091 
1092 .keywords: SNES, matrix-free, h history, differencing history
1093 
1094 .seealso: MatMFFDGetH(), MatCreateSNESMF(),
1095           MatMFFDSetHHistory(), MatMFFDSetFunctionError()
1096 
1097 @*/
1098 PetscErrorCode  MatMFFDResetHHistory(Mat J)
1099 {
1100   PetscErrorCode ierr;
1101 
1102   PetscFunctionBegin;
1103   ierr = PetscTryMethod(J,"MatMFFDResetHHistory_C",(Mat),(J));CHKERRQ(ierr);
1104   PetscFunctionReturn(0);
1105 }
1106 
1107 
1108 #undef __FUNCT__
1109 #define __FUNCT__ "MatMFFDSetBase"
1110 /*@
1111     MatMFFDSetBase - Sets the vector U at which matrix vector products of the
1112         Jacobian are computed
1113 
1114     Logically Collective on Mat
1115 
1116     Input Parameters:
1117 +   J - the MatMFFD matrix
1118 .   U - the vector
1119 -   F - (optional) vector that contains F(u) if it has been already computed
1120 
1121     Notes: This is rarely used directly
1122 
1123     If F is provided then it is not recomputed. Otherwise the function is evaluated at the base
1124     point during the first MatMult() after each call to MatMFFDSetBase().
1125 
1126     Level: advanced
1127 
1128 @*/
1129 PetscErrorCode  MatMFFDSetBase(Mat J,Vec U,Vec F)
1130 {
1131   PetscErrorCode ierr;
1132 
1133   PetscFunctionBegin;
1134   PetscValidHeaderSpecific(J,MAT_CLASSID,1);
1135   PetscValidHeaderSpecific(U,VEC_CLASSID,2);
1136   if (F) PetscValidHeaderSpecific(F,VEC_CLASSID,3);
1137   ierr = PetscTryMethod(J,"MatMFFDSetBase_C",(Mat,Vec,Vec),(J,U,F));CHKERRQ(ierr);
1138   PetscFunctionReturn(0);
1139 }
1140 
1141 #undef __FUNCT__
1142 #define __FUNCT__ "MatMFFDSetCheckh"
1143 /*@C
1144     MatMFFDSetCheckh - Sets a function that checks the computed h and adjusts
1145         it to satisfy some criteria
1146 
1147     Logically Collective on Mat
1148 
1149     Input Parameters:
1150 +   J - the MatMFFD matrix
1151 .   fun - the function that checks h
1152 -   ctx - any context needed by the function
1153 
1154     Options Database Keys:
1155 .   -mat_mffd_check_positivity
1156 
1157     Level: advanced
1158 
1159     Notes: For example, MatMFFDSetCheckPositivity() insures that all entries
1160        of U + h*a are non-negative
1161 
1162 .seealso:  MatMFFDSetCheckPositivity()
1163 @*/
1164 PetscErrorCode  MatMFFDSetCheckh(Mat J,PetscErrorCode (*fun)(void*,Vec,Vec,PetscScalar*),void *ctx)
1165 {
1166   PetscErrorCode ierr;
1167 
1168   PetscFunctionBegin;
1169   PetscValidHeaderSpecific(J,MAT_CLASSID,1);
1170   ierr = PetscTryMethod(J,"MatMFFDSetCheckh_C",(Mat,PetscErrorCode (*)(void*,Vec,Vec,PetscScalar*),void*),(J,fun,ctx));CHKERRQ(ierr);
1171   PetscFunctionReturn(0);
1172 }
1173 
1174 #undef __FUNCT__
1175 #define __FUNCT__ "MatMFFDSetCheckPositivity"
1176 /*@
1177     MatMFFDCheckPositivity - Checks that all entries in U + h*a are positive or
1178         zero, decreases h until this is satisfied.
1179 
1180     Logically Collective on Vec
1181 
1182     Input Parameters:
1183 +   U - base vector that is added to
1184 .   a - vector that is added
1185 .   h - scaling factor on a
1186 -   dummy - context variable (unused)
1187 
1188     Options Database Keys:
1189 .   -mat_mffd_check_positivity
1190 
1191     Level: advanced
1192 
1193     Notes: This is rarely used directly, rather it is passed as an argument to
1194            MatMFFDSetCheckh()
1195 
1196 .seealso:  MatMFFDSetCheckh()
1197 @*/
1198 PetscErrorCode  MatMFFDCheckPositivity(void *dummy,Vec U,Vec a,PetscScalar *h)
1199 {
1200   PetscReal      val, minval;
1201   PetscScalar    *u_vec, *a_vec;
1202   PetscErrorCode ierr;
1203   PetscInt       i,n;
1204   MPI_Comm       comm;
1205 
1206   PetscFunctionBegin;
1207   ierr   = PetscObjectGetComm((PetscObject)U,&comm);CHKERRQ(ierr);
1208   ierr   = VecGetArray(U,&u_vec);CHKERRQ(ierr);
1209   ierr   = VecGetArray(a,&a_vec);CHKERRQ(ierr);
1210   ierr   = VecGetLocalSize(U,&n);CHKERRQ(ierr);
1211   minval = PetscAbsScalar(*h*1.01);
1212   for (i=0; i<n; i++) {
1213     if (PetscRealPart(u_vec[i] + *h*a_vec[i]) <= 0.0) {
1214       val = PetscAbsScalar(u_vec[i]/a_vec[i]);
1215       if (val < minval) minval = val;
1216     }
1217   }
1218   ierr = VecRestoreArray(U,&u_vec);CHKERRQ(ierr);
1219   ierr = VecRestoreArray(a,&a_vec);CHKERRQ(ierr);
1220   ierr = MPI_Allreduce(&minval,&val,1,MPIU_REAL,MPIU_MIN,comm);CHKERRQ(ierr);
1221   if (val <= PetscAbsScalar(*h)) {
1222     ierr = PetscInfo2(U,"Scaling back h from %g to %g\n",(double)PetscRealPart(*h),(double)(.99*val));CHKERRQ(ierr);
1223     if (PetscRealPart(*h) > 0.0) *h =  0.99*val;
1224     else                         *h = -0.99*val;
1225   }
1226   PetscFunctionReturn(0);
1227 }
1228 
1229 
1230 
1231 
1232 
1233 
1234 
1235