Actual source code: mffddef.c

petsc-3.12.5 2020-03-29
Report Typos and Errors

  2: /*
  3:   Implements the DS PETSc approach for computing the h
  4:   parameter used with the finite difference based matrix-free
  5:   Jacobian-vector products.

  7:   To make your own: clone this file and modify for your needs.

  9:   Mandatory functions:
 10:   -------------------
 11:       MatMFFDCompute_  - for a given point and direction computes h

 13:       MatCreateMFFD _ - fills in the MatMFFD data structure
 14:                            for this particular implementation


 17:    Optional functions:
 18:    -------------------
 19:       MatMFFDView_ - prints information about the parameters being used.
 20:                        This is called when SNESView() or -snes_view is used.

 22:       MatMFFDSetFromOptions_ - checks the options database for options that
 23:                                apply to this method.

 25:       MatMFFDDestroy_ - frees any space allocated by the routines above

 27: */

 29: /*
 30:     This include file defines the data structure  MatMFFD that
 31:    includes information about the computation of h. It is shared by
 32:    all implementations that people provide
 33: */
 34:  #include <petsc/private/matimpl.h>
 35:  #include <../src/mat/impls/mffd/mffdimpl.h>

 37: /*
 38:       The  method has one parameter that is used to
 39:    "cutoff" very small values. This is stored in a data structure
 40:    that is only visible to this file. If your method has no parameters
 41:    it can omit this, if it has several simply reorganize the data structure.
 42:    The data structure is "hung-off" the MatMFFD data structure in
 43:    the void *hctx; field.
 44: */
 45: typedef struct {
 46:   PetscReal umin;          /* minimum allowable u'a value relative to |u|_1 */
 47: } MatMFFD_DS;

 49: /*
 50:    MatMFFDCompute_DS - Standard PETSc code for computing the
 51:    differencing paramter (h) for use with matrix-free finite differences.

 53:    Input Parameters:
 54: +  ctx - the matrix free context
 55: .  U - the location at which you want the Jacobian
 56: -  a - the direction you want the derivative


 59:    Output Parameter:
 60: .  h - the scale computed

 62: */
 63: static PetscErrorCode MatMFFDCompute_DS(MatMFFD ctx,Vec U,Vec a,PetscScalar *h,PetscBool  *zeroa)
 64: {
 65:   MatMFFD_DS     *hctx = (MatMFFD_DS*)ctx->hctx;
 66:   PetscReal      nrm,sum,umin = hctx->umin;
 67:   PetscScalar    dot;

 71:   if (!(ctx->count % ctx->recomputeperiod)) {
 72:     /*
 73:      This algorithm requires 2 norms and 1 inner product. Rather than
 74:      use directly the VecNorm() and VecDot() routines (and thus have
 75:      three separate collective operations, we use the VecxxxBegin/End() routines
 76:     */
 77:     VecDotBegin(U,a,&dot);
 78:     VecNormBegin(a,NORM_1,&sum);
 79:     VecNormBegin(a,NORM_2,&nrm);
 80:     VecDotEnd(U,a,&dot);
 81:     VecNormEnd(a,NORM_1,&sum);
 82:     VecNormEnd(a,NORM_2,&nrm);

 84:     if (nrm == 0.0) {
 85:       *zeroa = PETSC_TRUE;
 86:       return(0);
 87:     }
 88:     *zeroa = PETSC_FALSE;

 90:     /*
 91:       Safeguard for step sizes that are "too small"
 92:     */
 93:     if (PetscAbsScalar(dot) < umin*sum && PetscRealPart(dot) >= 0.0) dot = umin*sum;
 94:     else if (PetscAbsScalar(dot) < 0.0 && PetscRealPart(dot) > -umin*sum) dot = -umin*sum;
 95:     *h = ctx->error_rel*dot/(nrm*nrm);
 96:     if (PetscIsInfOrNanScalar(*h)) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Differencing parameter is not a number sum = %g dot = %g norm = %g",(double)sum,(double)PetscRealPart(dot),(double)nrm);
 97:   } else {
 98:     *h = ctx->currenth;
 99:   }
100:   ctx->count++;
101:   return(0);
102: }

104: /*
105:    MatMFFDView_DS - Prints information about this particular
106:    method for computing h. Note that this does not print the general
107:    information about the matrix-free method, as such info is printed
108:    by the calling routine.

110:    Input Parameters:
111: +  ctx - the matrix free context
112: -  viewer - the PETSc viewer
113: */
114: static PetscErrorCode MatMFFDView_DS(MatMFFD ctx,PetscViewer viewer)
115: {
116:   MatMFFD_DS     *hctx = (MatMFFD_DS*)ctx->hctx;
118:   PetscBool      iascii;

121:   /*
122:      Currently this only handles the ascii file viewers, others
123:      could be added, but for this type of object other viewers
124:      make less sense
125:   */
126:   PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);
127:   if (iascii) {
128:     PetscViewerASCIIPrintf(viewer,"    umin=%g (minimum iterate parameter)\n",(double)hctx->umin);
129:   }
130:   return(0);
131: }

133: /*
134:    MatMFFDSetFromOptions_DS - Looks in the options database for
135:    any options appropriate for this method.

137:    Input Parameter:
138: .  ctx - the matrix free context

140: */
141: static PetscErrorCode MatMFFDSetFromOptions_DS(PetscOptionItems *PetscOptionsObject,MatMFFD ctx)
142: {
144:   MatMFFD_DS     *hctx = (MatMFFD_DS*)ctx->hctx;

147:   PetscOptionsHead(PetscOptionsObject,"Finite difference matrix free parameters");
148:   PetscOptionsReal("-mat_mffd_umin","umin","MatMFFDDSSetUmin",hctx->umin,&hctx->umin,0);
149:   PetscOptionsTail();
150:   return(0);
151: }

153: /*
154:    MatMFFDDestroy_DS - Frees the space allocated by
155:    MatCreateMFFD_DS().

157:    Input Parameter:
158: .  ctx - the matrix free context

160:    Notes:
161:    Does not free the ctx, that is handled by the calling routine
162: */
163: static PetscErrorCode MatMFFDDestroy_DS(MatMFFD ctx)
164: {

168:   PetscFree(ctx->hctx);
169:   return(0);
170: }

172: /*
173:    The following two routines use the PetscObjectCompose() and PetscObjectQuery()
174:    mechanism to allow the user to change the Umin parameter used in this method.
175: */
176: PetscErrorCode MatMFFDDSSetUmin_DS(Mat mat,PetscReal umin)
177: {
178:   MatMFFD        ctx=NULL;
179:   MatMFFD_DS     *hctx;

183:   MatShellGetContext(mat,&ctx);
184:   if (!ctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"MatMFFDDSSetUmin() attached to non-shell matrix");
185:   hctx       = (MatMFFD_DS*)ctx->hctx;
186:   hctx->umin = umin;
187:   return(0);
188: }

190: /*@
191:     MatMFFDDSSetUmin - Sets the "umin" parameter used by the
192:     PETSc routine for computing the differencing parameter, h, which is used
193:     for matrix-free Jacobian-vector products.

195:    Input Parameters:
196: +  A - the matrix created with MatCreateSNESMF()
197: -  umin - the parameter

199:    Level: advanced

201:    Notes:
202:    See the manual page for MatCreateSNESMF() for a complete description of the
203:    algorithm used to compute h.

205: .seealso: MatMFFDSetFunctionError(), MatCreateSNESMF()

207: @*/
208: PetscErrorCode  MatMFFDDSSetUmin(Mat A,PetscReal umin)
209: {

214:   PetscTryMethod(A,"MatMFFDDSSetUmin_C",(Mat,PetscReal),(A,umin));
215:   return(0);
216: }

218: /*MC
219:      MATMFFD_DS - the code for compute the "h" used in the finite difference
220:             matrix-free matrix vector product.  This code
221:         implements the strategy in Dennis and Schnabel, "Numerical Methods for Unconstrained
222:         Optimization and Nonlinear Equations".

224:    Options Database Keys:
225: .  -mat_mffd_umin <umin> see MatMFFDDSSetUmin()

227:    Level: intermediate

229:    Notes:
230:     Requires 2 norms and 1 inner product, but they are computed together
231:        so only one parallel collective operation is needed. See MATMFFD_WP for a method
232:        (with GMRES) that requires NO collective operations.

234:    Formula used:
235:      F'(u)*a = [F(u+h*a) - F(u)]/h where
236:      h = error_rel*u'a/||a||^2                        if  |u'a| > umin*||a||_{1}
237:        = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2   otherwise
238:  where
239:      error_rel = square root of relative error in function evaluation
240:      umin = minimum iterate parameter

242: .seealso: MATMFFD, MatCreateMFFD(), MatCreateSNESMF(), MATMFFD_WP, MatMFFDDSSetUmin()

244: M*/
245: PETSC_EXTERN PetscErrorCode MatCreateMFFD_DS(MatMFFD ctx)
246: {
247:   MatMFFD_DS     *hctx;

251:   /* allocate my own private data structure */
252:   PetscNewLog(ctx,&hctx);
253:   ctx->hctx = (void*)hctx;
254:   /* set a default for my parameter */
255:   hctx->umin = 1.e-6;

257:   /* set the functions I am providing */
258:   ctx->ops->compute        = MatMFFDCompute_DS;
259:   ctx->ops->destroy        = MatMFFDDestroy_DS;
260:   ctx->ops->view           = MatMFFDView_DS;
261:   ctx->ops->setfromoptions = MatMFFDSetFromOptions_DS;

263:   PetscObjectComposeFunction((PetscObject)ctx->mat,"MatMFFDDSSetUmin_C",MatMFFDDSSetUmin_DS);
264:   return(0);
265: }