xref: /petsc/src/snes/utils/convest.c (revision 478db826a40dca33d2f318a94ee2bc67dcb6bd71)
16af0ca60SMatthew G. Knepley #include <petscconvest.h>            /*I "petscconvest.h" I*/
26af0ca60SMatthew G. Knepley #include <petscdmplex.h>
36af0ca60SMatthew G. Knepley #include <petscds.h>
46af0ca60SMatthew G. Knepley 
56af0ca60SMatthew G. Knepley #include <petsc/private/petscconvestimpl.h>
66af0ca60SMatthew G. Knepley 
76af0ca60SMatthew G. Knepley static PetscErrorCode zero_private(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nc, PetscScalar *u, void *ctx)
86af0ca60SMatthew G. Knepley {
96af0ca60SMatthew G. Knepley   PetscInt c;
106af0ca60SMatthew G. Knepley   for (c = 0; c < Nc; ++c) u[c] = 0.0;
116af0ca60SMatthew G. Knepley   return 0;
126af0ca60SMatthew G. Knepley }
136af0ca60SMatthew G. Knepley 
146af0ca60SMatthew G. Knepley /*@
156af0ca60SMatthew G. Knepley   PetscConvEstDestroy - Destroys a PetscConvEst object
166af0ca60SMatthew G. Knepley 
176af0ca60SMatthew G. Knepley   Collective on PetscConvEst
186af0ca60SMatthew G. Knepley 
196af0ca60SMatthew G. Knepley   Input Parameter:
206af0ca60SMatthew G. Knepley . ce - The PetscConvEst object
216af0ca60SMatthew G. Knepley 
226af0ca60SMatthew G. Knepley   Level: beginner
236af0ca60SMatthew G. Knepley 
246af0ca60SMatthew G. Knepley .seealso: PetscConvEstCreate(), PetscConvEstGetConvRate()
256af0ca60SMatthew G. Knepley @*/
266af0ca60SMatthew G. Knepley PetscErrorCode PetscConvEstDestroy(PetscConvEst *ce)
276af0ca60SMatthew G. Knepley {
286af0ca60SMatthew G. Knepley   PetscErrorCode ierr;
296af0ca60SMatthew G. Knepley 
306af0ca60SMatthew G. Knepley   PetscFunctionBegin;
316af0ca60SMatthew G. Knepley   if (!*ce) PetscFunctionReturn(0);
326af0ca60SMatthew G. Knepley   PetscValidHeaderSpecific((*ce),PETSC_OBJECT_CLASSID,1);
336af0ca60SMatthew G. Knepley   if (--((PetscObject)(*ce))->refct > 0) {
346af0ca60SMatthew G. Knepley     *ce = NULL;
356af0ca60SMatthew G. Knepley     PetscFunctionReturn(0);
366af0ca60SMatthew G. Knepley   }
3795cbbfd3SMatthew G. Knepley   ierr = PetscFree3((*ce)->initGuess, (*ce)->exactSol, (*ce)->ctxs);CHKERRQ(ierr);
386af0ca60SMatthew G. Knepley   ierr = PetscFree((*ce)->errors);CHKERRQ(ierr);
396af0ca60SMatthew G. Knepley   ierr = PetscHeaderDestroy(ce);CHKERRQ(ierr);
406af0ca60SMatthew G. Knepley   PetscFunctionReturn(0);
416af0ca60SMatthew G. Knepley }
426af0ca60SMatthew G. Knepley 
436af0ca60SMatthew G. Knepley /*@
446af0ca60SMatthew G. Knepley   PetscConvEstSetFromOptions - Sets a PetscConvEst object from options
456af0ca60SMatthew G. Knepley 
466af0ca60SMatthew G. Knepley   Collective on PetscConvEst
476af0ca60SMatthew G. Knepley 
486af0ca60SMatthew G. Knepley   Input Parameters:
496af0ca60SMatthew G. Knepley . ce - The PetscConvEst object
506af0ca60SMatthew G. Knepley 
516af0ca60SMatthew G. Knepley   Level: beginner
526af0ca60SMatthew G. Knepley 
536af0ca60SMatthew G. Knepley .seealso: PetscConvEstCreate(), PetscConvEstGetConvRate()
546af0ca60SMatthew G. Knepley @*/
556af0ca60SMatthew G. Knepley PetscErrorCode PetscConvEstSetFromOptions(PetscConvEst ce)
566af0ca60SMatthew G. Knepley {
576af0ca60SMatthew G. Knepley   PetscErrorCode ierr;
586af0ca60SMatthew G. Knepley 
596af0ca60SMatthew G. Knepley   PetscFunctionBegin;
606af0ca60SMatthew G. Knepley   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject) ce), "", "Convergence Estimator Options", "PetscConvEst");CHKERRQ(ierr);
6146079b62SMatthew G. Knepley   ierr = PetscOptionsInt("-convest_num_refine", "The number of refinements for the convergence check", "PetscConvEst", ce->Nr, &ce->Nr, NULL);CHKERRQ(ierr);
622e61be88SMatthew G. Knepley   ierr = PetscOptionsReal("-convest_refine_factor", "The increase in resolution in each dimension", "PetscConvEst", ce->r, &ce->r, NULL);CHKERRQ(ierr);
6346079b62SMatthew G. Knepley   ierr = PetscOptionsBool("-convest_monitor", "Monitor the error for each convergence check", "PetscConvEst", ce->monitor, &ce->monitor, NULL);CHKERRQ(ierr);
646af0ca60SMatthew G. Knepley   ierr = PetscOptionsEnd();
656af0ca60SMatthew G. Knepley   PetscFunctionReturn(0);
666af0ca60SMatthew G. Knepley }
676af0ca60SMatthew G. Knepley 
686af0ca60SMatthew G. Knepley /*@
696af0ca60SMatthew G. Knepley   PetscConvEstView - Views a PetscConvEst object
706af0ca60SMatthew G. Knepley 
716af0ca60SMatthew G. Knepley   Collective on PetscConvEst
726af0ca60SMatthew G. Knepley 
736af0ca60SMatthew G. Knepley   Input Parameters:
746af0ca60SMatthew G. Knepley + ce     - The PetscConvEst object
756af0ca60SMatthew G. Knepley - viewer - The PetscViewer object
766af0ca60SMatthew G. Knepley 
776af0ca60SMatthew G. Knepley   Level: beginner
786af0ca60SMatthew G. Knepley 
796af0ca60SMatthew G. Knepley .seealso: PetscConvEstCreate(), PetscConvEstGetConvRate()
806af0ca60SMatthew G. Knepley @*/
816af0ca60SMatthew G. Knepley PetscErrorCode PetscConvEstView(PetscConvEst ce, PetscViewer viewer)
826af0ca60SMatthew G. Knepley {
836af0ca60SMatthew G. Knepley   PetscErrorCode ierr;
846af0ca60SMatthew G. Knepley 
856af0ca60SMatthew G. Knepley   PetscFunctionBegin;
866af0ca60SMatthew G. Knepley   ierr = PetscObjectPrintClassNamePrefixType((PetscObject) ce, viewer);CHKERRQ(ierr);
876af0ca60SMatthew G. Knepley   ierr = PetscViewerASCIIPrintf(viewer, "ConvEst with %D levels\n", ce->Nr+1);CHKERRQ(ierr);
886af0ca60SMatthew G. Knepley   PetscFunctionReturn(0);
896af0ca60SMatthew G. Knepley }
906af0ca60SMatthew G. Knepley 
916af0ca60SMatthew G. Knepley /*@
926af0ca60SMatthew G. Knepley   PetscConvEstGetSolver - Gets the solver used to produce discrete solutions
936af0ca60SMatthew G. Knepley 
946af0ca60SMatthew G. Knepley   Not collective
956af0ca60SMatthew G. Knepley 
966af0ca60SMatthew G. Knepley   Input Parameter:
976af0ca60SMatthew G. Knepley . ce     - The PetscConvEst object
986af0ca60SMatthew G. Knepley 
996af0ca60SMatthew G. Knepley   Output Parameter:
100900f6b5bSMatthew G. Knepley . solver - The solver
1016af0ca60SMatthew G. Knepley 
1026af0ca60SMatthew G. Knepley   Level: intermediate
1036af0ca60SMatthew G. Knepley 
1046af0ca60SMatthew G. Knepley .seealso: PetscConvEstSetSolver(), PetscConvEstCreate(), PetscConvEstGetConvRate()
1056af0ca60SMatthew G. Knepley @*/
106900f6b5bSMatthew G. Knepley PetscErrorCode PetscConvEstGetSolver(PetscConvEst ce, PetscObject *solver)
1076af0ca60SMatthew G. Knepley {
1086af0ca60SMatthew G. Knepley   PetscFunctionBegin;
1096af0ca60SMatthew G. Knepley   PetscValidHeaderSpecific(ce, PETSC_OBJECT_CLASSID, 1);
110900f6b5bSMatthew G. Knepley   PetscValidPointer(solver, 2);
111900f6b5bSMatthew G. Knepley   *solver = ce->solver;
1126af0ca60SMatthew G. Knepley   PetscFunctionReturn(0);
1136af0ca60SMatthew G. Knepley }
1146af0ca60SMatthew G. Knepley 
1156af0ca60SMatthew G. Knepley /*@
1166af0ca60SMatthew G. Knepley   PetscConvEstSetSolver - Sets the solver used to produce discrete solutions
1176af0ca60SMatthew G. Knepley 
1186af0ca60SMatthew G. Knepley   Not collective
1196af0ca60SMatthew G. Knepley 
1206af0ca60SMatthew G. Knepley   Input Parameters:
1216af0ca60SMatthew G. Knepley + ce     - The PetscConvEst object
122900f6b5bSMatthew G. Knepley - solver - The solver
1236af0ca60SMatthew G. Knepley 
1246af0ca60SMatthew G. Knepley   Level: intermediate
1256af0ca60SMatthew G. Knepley 
1266af0ca60SMatthew G. Knepley   Note: The solver MUST have an attached DM/DS, so that we know the exact solution
1276af0ca60SMatthew G. Knepley 
128900f6b5bSMatthew G. Knepley .seealso: PetscConvEstGetSNES(), PetscConvEstCreate(), PetscConvEstGetConvRate()
1296af0ca60SMatthew G. Knepley @*/
130900f6b5bSMatthew G. Knepley PetscErrorCode PetscConvEstSetSolver(PetscConvEst ce, PetscObject solver)
1316af0ca60SMatthew G. Knepley {
1326af0ca60SMatthew G. Knepley   PetscErrorCode ierr;
1336af0ca60SMatthew G. Knepley 
1346af0ca60SMatthew G. Knepley   PetscFunctionBegin;
1356af0ca60SMatthew G. Knepley   PetscValidHeaderSpecific(ce, PETSC_OBJECT_CLASSID, 1);
136900f6b5bSMatthew G. Knepley   PetscValidHeader(solver, 2);
137900f6b5bSMatthew G. Knepley   ce->solver = solver;
138900f6b5bSMatthew G. Knepley   ierr = (*ce->ops->setsolver)(ce, solver);CHKERRQ(ierr);
1396af0ca60SMatthew G. Knepley   PetscFunctionReturn(0);
1406af0ca60SMatthew G. Knepley }
1416af0ca60SMatthew G. Knepley 
1426af0ca60SMatthew G. Knepley /*@
1436af0ca60SMatthew G. Knepley   PetscConvEstSetUp - After the solver is specified, we create structures for estimating convergence
1446af0ca60SMatthew G. Knepley 
1456af0ca60SMatthew G. Knepley   Collective on PetscConvEst
1466af0ca60SMatthew G. Knepley 
1476af0ca60SMatthew G. Knepley   Input Parameters:
1486af0ca60SMatthew G. Knepley . ce - The PetscConvEst object
1496af0ca60SMatthew G. Knepley 
1506af0ca60SMatthew G. Knepley   Level: beginner
1516af0ca60SMatthew G. Knepley 
1526af0ca60SMatthew G. Knepley .seealso: PetscConvEstCreate(), PetscConvEstGetConvRate()
1536af0ca60SMatthew G. Knepley @*/
1540955ed61SMatthew G. Knepley PetscErrorCode PetscConvEstSetUp(PetscConvEst ce)
1556af0ca60SMatthew G. Knepley {
156083401c6SMatthew G. Knepley   PetscInt       Nf, f, Nds, s;
1576af0ca60SMatthew G. Knepley   PetscErrorCode ierr;
1586af0ca60SMatthew G. Knepley 
1596af0ca60SMatthew G. Knepley   PetscFunctionBegin;
160083401c6SMatthew G. Knepley   ierr = DMGetNumFields(ce->idm, &Nf);CHKERRQ(ierr);
161900f6b5bSMatthew G. Knepley   ce->Nf = PetscMax(Nf, 1);
1626af0ca60SMatthew G. Knepley   ierr = PetscMalloc1((ce->Nr+1)*ce->Nf, &ce->errors);CHKERRQ(ierr);
163083401c6SMatthew G. Knepley   ierr = PetscCalloc3(ce->Nf, &ce->initGuess, ce->Nf, &ce->exactSol, ce->Nf, &ce->ctxs);CHKERRQ(ierr);
164900f6b5bSMatthew G. Knepley   for (f = 0; f < Nf; ++f) ce->initGuess[f] = zero_private;
165083401c6SMatthew G. Knepley   ierr = DMGetNumDS(ce->idm, &Nds);CHKERRQ(ierr);
166083401c6SMatthew G. Knepley   for (s = 0; s < Nds; ++s) {
167083401c6SMatthew G. Knepley     PetscDS         ds;
168083401c6SMatthew G. Knepley     DMLabel         label;
169083401c6SMatthew G. Knepley     IS              fieldIS;
170083401c6SMatthew G. Knepley     const PetscInt *fields;
171083401c6SMatthew G. Knepley     PetscInt        dsNf;
172083401c6SMatthew G. Knepley 
173083401c6SMatthew G. Knepley     ierr = DMGetRegionNumDS(ce->idm, s, &label, &fieldIS, &ds);CHKERRQ(ierr);
174083401c6SMatthew G. Knepley     ierr = PetscDSGetNumFields(ds, &dsNf);CHKERRQ(ierr);
175083401c6SMatthew G. Knepley     if (fieldIS) {ierr = ISGetIndices(fieldIS, &fields);CHKERRQ(ierr);}
176083401c6SMatthew G. Knepley     for (f = 0; f < dsNf; ++f) {
177083401c6SMatthew G. Knepley       const PetscInt field = fields[f];
178083401c6SMatthew G. Knepley       ierr = PetscDSGetExactSolution(ds, field, &ce->exactSol[field], &ce->ctxs[field]);CHKERRQ(ierr);
179083401c6SMatthew G. Knepley     }
180083401c6SMatthew G. Knepley     if (fieldIS) {ierr = ISRestoreIndices(fieldIS, &fields);CHKERRQ(ierr);}
181083401c6SMatthew G. Knepley   }
182900f6b5bSMatthew G. Knepley   for (f = 0; f < Nf; ++f) {
1836af0ca60SMatthew G. Knepley     if (!ce->exactSol[f]) SETERRQ1(PetscObjectComm((PetscObject) ce), PETSC_ERR_ARG_WRONG, "DS must contain exact solution functions in order to estimate convergence, missing for field %D", f);
1846af0ca60SMatthew G. Knepley   }
1856af0ca60SMatthew G. Knepley   PetscFunctionReturn(0);
1866af0ca60SMatthew G. Knepley }
1876af0ca60SMatthew G. Knepley 
188900f6b5bSMatthew G. Knepley PetscErrorCode PetscConvEstComputeInitialGuess(PetscConvEst ce, PetscInt r, DM dm, Vec u)
1896af0ca60SMatthew G. Knepley {
1906af0ca60SMatthew G. Knepley   PetscErrorCode ierr;
1916af0ca60SMatthew G. Knepley 
1926af0ca60SMatthew G. Knepley   PetscFunctionBegin;
193900f6b5bSMatthew G. Knepley   PetscValidHeaderSpecific(ce, PETSC_OBJECT_CLASSID, 1);
194900f6b5bSMatthew G. Knepley   if (dm) PetscValidHeaderSpecific(dm, DM_CLASSID, 3);
195900f6b5bSMatthew G. Knepley   PetscValidHeaderSpecific(u, VEC_CLASSID, 4);
196900f6b5bSMatthew G. Knepley   ierr = (*ce->ops->initguess)(ce, r, dm, u);CHKERRQ(ierr);
197900f6b5bSMatthew G. Knepley   PetscFunctionReturn(0);
198900f6b5bSMatthew G. Knepley }
199900f6b5bSMatthew G. Knepley 
200900f6b5bSMatthew G. Knepley PetscErrorCode PetscConvEstComputeError(PetscConvEst ce, PetscInt r, DM dm, Vec u, PetscReal errors[])
201900f6b5bSMatthew G. Knepley {
202900f6b5bSMatthew G. Knepley   PetscErrorCode ierr;
203900f6b5bSMatthew G. Knepley 
204900f6b5bSMatthew G. Knepley   PetscFunctionBegin;
205900f6b5bSMatthew G. Knepley   PetscValidHeaderSpecific(ce, PETSC_OBJECT_CLASSID, 1);
206900f6b5bSMatthew G. Knepley   if (dm) PetscValidHeaderSpecific(dm, DM_CLASSID, 3);
207900f6b5bSMatthew G. Knepley   PetscValidHeaderSpecific(u, VEC_CLASSID, 4);
208900f6b5bSMatthew G. Knepley   PetscValidRealPointer(errors, 5);
209900f6b5bSMatthew G. Knepley   ierr = (*ce->ops->computeerror)(ce, r, dm, u, errors);CHKERRQ(ierr);
210900f6b5bSMatthew G. Knepley   PetscFunctionReturn(0);
211900f6b5bSMatthew G. Knepley }
212900f6b5bSMatthew G. Knepley 
213f2cacb80SMatthew G. Knepley /*@
214f2cacb80SMatthew G. Knepley   PetscConvEstMonitorDefault - Monitors the convergence estimation loop
215f2cacb80SMatthew G. Knepley 
216f2cacb80SMatthew G. Knepley   Collective on PetscConvEst
217f2cacb80SMatthew G. Knepley 
218f2cacb80SMatthew G. Knepley   Input Parameter:
219f2cacb80SMatthew G. Knepley + ce - The PetscConvEst object
220f2cacb80SMatthew G. Knepley - r  - The refinement level
221f2cacb80SMatthew G. Knepley 
222f2cacb80SMatthew G. Knepley   Options database keys:
223ee300463SSatish Balay . -convest_monitor - Activate the monitor
224f2cacb80SMatthew G. Knepley 
225f2cacb80SMatthew G. Knepley   Level: intermediate
226f2cacb80SMatthew G. Knepley 
227f2cacb80SMatthew G. Knepley .seealso: PetscConvEstCreate(), PetscConvEstGetConvRate(), SNESSolve(), TSSolve()
228f2cacb80SMatthew G. Knepley @*/
229f2cacb80SMatthew G. Knepley PetscErrorCode PetscConvEstMonitorDefault(PetscConvEst ce, PetscInt r)
230900f6b5bSMatthew G. Knepley {
231900f6b5bSMatthew G. Knepley   MPI_Comm       comm;
232900f6b5bSMatthew G. Knepley   PetscInt       f;
233900f6b5bSMatthew G. Knepley   PetscErrorCode ierr;
234900f6b5bSMatthew G. Knepley 
235900f6b5bSMatthew G. Knepley   PetscFunctionBegin;
236900f6b5bSMatthew G. Knepley   if (ce->monitor) {
237900f6b5bSMatthew G. Knepley     PetscReal *errors = &ce->errors[r*ce->Nf];
238900f6b5bSMatthew G. Knepley 
2396af0ca60SMatthew G. Knepley     ierr = PetscObjectGetComm((PetscObject) ce, &comm);CHKERRQ(ierr);
240900f6b5bSMatthew G. Knepley     ierr = PetscPrintf(comm, "L_2 Error: ");CHKERRQ(ierr);
241900f6b5bSMatthew G. Knepley     if (ce->Nf > 1) {ierr = PetscPrintf(comm, "[");CHKERRQ(ierr);}
242900f6b5bSMatthew G. Knepley     for (f = 0; f < ce->Nf; ++f) {
243900f6b5bSMatthew G. Knepley       if (f > 0) {ierr = PetscPrintf(comm, ", ");CHKERRQ(ierr);}
244900f6b5bSMatthew G. Knepley       if (errors[f] < 1.0e-11) {ierr = PetscPrintf(comm, "< 1e-11");CHKERRQ(ierr);}
245900f6b5bSMatthew G. Knepley       else                     {ierr = PetscPrintf(comm, "%g", (double) errors[f]);CHKERRQ(ierr);}
246900f6b5bSMatthew G. Knepley     }
247900f6b5bSMatthew G. Knepley     if (ce->Nf > 1) {ierr = PetscPrintf(comm, "]");CHKERRQ(ierr);}
248900f6b5bSMatthew G. Knepley     ierr = PetscPrintf(comm, "\n");CHKERRQ(ierr);
249900f6b5bSMatthew G. Knepley   }
250900f6b5bSMatthew G. Knepley   PetscFunctionReturn(0);
251900f6b5bSMatthew G. Knepley }
252900f6b5bSMatthew G. Knepley 
253900f6b5bSMatthew G. Knepley static PetscErrorCode PetscConvEstSetSNES_Private(PetscConvEst ce, PetscObject solver)
254900f6b5bSMatthew G. Knepley {
255900f6b5bSMatthew G. Knepley   PetscClassId   id;
256900f6b5bSMatthew G. Knepley   PetscErrorCode ierr;
257900f6b5bSMatthew G. Knepley 
258900f6b5bSMatthew G. Knepley   PetscFunctionBegin;
259900f6b5bSMatthew G. Knepley   ierr = PetscObjectGetClassId(ce->solver, &id);CHKERRQ(ierr);
260900f6b5bSMatthew G. Knepley   if (id != SNES_CLASSID) SETERRQ(PetscObjectComm((PetscObject) ce), PETSC_ERR_ARG_WRONG, "Solver was not a SNES");
261900f6b5bSMatthew G. Knepley   ierr = SNESGetDM((SNES) ce->solver, &ce->idm);CHKERRQ(ierr);
262900f6b5bSMatthew G. Knepley   PetscFunctionReturn(0);
263900f6b5bSMatthew G. Knepley }
264900f6b5bSMatthew G. Knepley 
265900f6b5bSMatthew G. Knepley static PetscErrorCode PetscConvEstInitGuessSNES_Private(PetscConvEst ce, PetscInt r, DM dm, Vec u)
266900f6b5bSMatthew G. Knepley {
267900f6b5bSMatthew G. Knepley   PetscErrorCode ierr;
268900f6b5bSMatthew G. Knepley 
269900f6b5bSMatthew G. Knepley   PetscFunctionBegin;
270900f6b5bSMatthew G. Knepley   ierr = DMProjectFunction(dm, 0.0, ce->initGuess, ce->ctxs, INSERT_VALUES, u);CHKERRQ(ierr);
271900f6b5bSMatthew G. Knepley   PetscFunctionReturn(0);
272900f6b5bSMatthew G. Knepley }
273900f6b5bSMatthew G. Knepley 
274900f6b5bSMatthew G. Knepley static PetscErrorCode PetscConvEstComputeErrorSNES_Private(PetscConvEst ce, PetscInt r, DM dm, Vec u, PetscReal errors[])
275900f6b5bSMatthew G. Knepley {
276900f6b5bSMatthew G. Knepley   PetscErrorCode ierr;
277900f6b5bSMatthew G. Knepley 
278900f6b5bSMatthew G. Knepley   PetscFunctionBegin;
279900f6b5bSMatthew G. Knepley   ierr = DMComputeL2FieldDiff(dm, 0.0, ce->exactSol, ce->ctxs, u, errors);CHKERRQ(ierr);
280900f6b5bSMatthew G. Knepley   PetscFunctionReturn(0);
281900f6b5bSMatthew G. Knepley }
282900f6b5bSMatthew G. Knepley 
283*478db826SMatthew G. Knepley static PetscErrorCode PetscConvEstSetJacobianNullspace_Private(PetscConvEst ce, SNES snes)
284*478db826SMatthew G. Knepley {
285*478db826SMatthew G. Knepley   DM             dm;
286*478db826SMatthew G. Knepley   PetscInt       f;
287*478db826SMatthew G. Knepley   PetscErrorCode ierr;
288*478db826SMatthew G. Knepley 
289*478db826SMatthew G. Knepley   PetscFunctionBegin;
290*478db826SMatthew G. Knepley   ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr);
291*478db826SMatthew G. Knepley   for (f = 0; f < ce->Nf; ++f) {
292*478db826SMatthew G. Knepley     PetscErrorCode (*nspconstr)(DM, PetscInt, PetscInt, MatNullSpace *);
293*478db826SMatthew G. Knepley 
294*478db826SMatthew G. Knepley     ierr = DMGetNullSpaceConstructor(dm, f, &nspconstr);CHKERRQ(ierr);
295*478db826SMatthew G. Knepley     if (nspconstr) {
296*478db826SMatthew G. Knepley       MatNullSpace nullsp;
297*478db826SMatthew G. Knepley       Mat          J;
298*478db826SMatthew G. Knepley 
299*478db826SMatthew G. Knepley       ierr = (*nspconstr)(dm, f, f,&nullsp);CHKERRQ(ierr);
300*478db826SMatthew G. Knepley       ierr = SNESSetUp(snes);CHKERRQ(ierr);
301*478db826SMatthew G. Knepley       ierr = SNESGetJacobian(snes, &J, NULL, NULL, NULL);CHKERRQ(ierr);
302*478db826SMatthew G. Knepley       ierr = MatSetNullSpace(J, nullsp);CHKERRQ(ierr);
303*478db826SMatthew G. Knepley       ierr = MatNullSpaceDestroy(&nullsp);CHKERRQ(ierr);
304*478db826SMatthew G. Knepley       break;
305*478db826SMatthew G. Knepley     }
306*478db826SMatthew G. Knepley   }
307*478db826SMatthew G. Knepley   PetscFunctionReturn(0);
308*478db826SMatthew G. Knepley }
309*478db826SMatthew G. Knepley 
310900f6b5bSMatthew G. Knepley static PetscErrorCode PetscConvEstGetConvRateSNES_Private(PetscConvEst ce, PetscReal alpha[])
311900f6b5bSMatthew G. Knepley {
312900f6b5bSMatthew G. Knepley   SNES           snes = (SNES) ce->solver;
313900f6b5bSMatthew G. Knepley   DM            *dm;
314900f6b5bSMatthew G. Knepley   PetscObject    disc;
315900f6b5bSMatthew G. Knepley   PetscReal     *x, *y, slope, intercept;
316900f6b5bSMatthew G. Knepley   PetscInt      *dof, Nr = ce->Nr, r, f, dim, oldlevel, oldnlev;
317900f6b5bSMatthew G. Knepley   void          *ctx;
318900f6b5bSMatthew G. Knepley   PetscErrorCode ierr;
319900f6b5bSMatthew G. Knepley 
320900f6b5bSMatthew G. Knepley   PetscFunctionBegin;
3212e61be88SMatthew G. Knepley   if (ce->r != 2.0) SETERRQ1(PetscObjectComm((PetscObject) ce), PETSC_ERR_SUP, "Only refinement factor 2 is currently supported (not %g)", (double) ce->r);
3226af0ca60SMatthew G. Knepley   ierr = DMGetDimension(ce->idm, &dim);CHKERRQ(ierr);
3236af0ca60SMatthew G. Knepley   ierr = DMGetApplicationContext(ce->idm, &ctx);CHKERRQ(ierr);
3246af0ca60SMatthew G. Knepley   ierr = DMPlexSetRefinementUniform(ce->idm, PETSC_TRUE);CHKERRQ(ierr);
32573269098SMatthew G. Knepley   ierr = DMGetRefineLevel(ce->idm, &oldlevel);CHKERRQ(ierr);
326adfa7136SMatthew G. Knepley   ierr = PetscMalloc2((Nr+1), &dm, (Nr+1)*ce->Nf, &dof);CHKERRQ(ierr);
3276af0ca60SMatthew G. Knepley   /* Loop over meshes */
328900f6b5bSMatthew G. Knepley   dm[0] = ce->idm;
3296af0ca60SMatthew G. Knepley   for (r = 0; r <= Nr; ++r) {
330900f6b5bSMatthew G. Knepley     Vec           u;
331e5ed2c37SJose E. Roman #if defined(PETSC_USE_LOG)
332adfa7136SMatthew G. Knepley     PetscLogStage stage;
333e5ed2c37SJose E. Roman #endif
334adfa7136SMatthew G. Knepley     char          stageName[PETSC_MAX_PATH_LEN];
335900f6b5bSMatthew G. Knepley     const char   *dmname, *uname;
336adfa7136SMatthew G. Knepley 
337adfa7136SMatthew G. Knepley     ierr = PetscSNPrintf(stageName, PETSC_MAX_PATH_LEN-1, "ConvEst Refinement Level %D", r);CHKERRQ(ierr);
338adfa7136SMatthew G. Knepley     ierr = PetscLogStageRegister(stageName, &stage);CHKERRQ(ierr);
339adfa7136SMatthew G. Knepley     ierr = PetscLogStagePush(stage);CHKERRQ(ierr);
3406af0ca60SMatthew G. Knepley     if (r > 0) {
3416af0ca60SMatthew G. Knepley       ierr = DMRefine(dm[r-1], MPI_COMM_NULL, &dm[r]);CHKERRQ(ierr);
3426af0ca60SMatthew G. Knepley       ierr = DMSetCoarseDM(dm[r], dm[r-1]);CHKERRQ(ierr);
343ca3d3a14SMatthew G. Knepley       ierr = DMCopyTransform(ce->idm, dm[r]);CHKERRQ(ierr);
3446af0ca60SMatthew G. Knepley       ierr = PetscObjectGetName((PetscObject) dm[r-1], &dmname);CHKERRQ(ierr);
3456af0ca60SMatthew G. Knepley       ierr = PetscObjectSetName((PetscObject) dm[r], dmname);CHKERRQ(ierr);
346*478db826SMatthew G. Knepley       for (f = 0; f < ce->Nf; ++f) {
3478cda7954SMatthew G. Knepley         PetscErrorCode (*nspconstr)(DM, PetscInt, PetscInt, MatNullSpace *);
348900f6b5bSMatthew G. Knepley 
3498f892730SMatthew G. Knepley         ierr = DMGetNullSpaceConstructor(dm[r-1], f, &nspconstr);CHKERRQ(ierr);
3508f892730SMatthew G. Knepley         ierr = DMSetNullSpaceConstructor(dm[r],   f,  nspconstr);CHKERRQ(ierr);
3518f892730SMatthew G. Knepley       }
3526af0ca60SMatthew G. Knepley     }
3536af0ca60SMatthew G. Knepley     ierr = DMViewFromOptions(dm[r], NULL, "-conv_dm_view");CHKERRQ(ierr);
3546af0ca60SMatthew G. Knepley     /* Create solution */
3556af0ca60SMatthew G. Knepley     ierr = DMCreateGlobalVector(dm[r], &u);CHKERRQ(ierr);
356e5e52638SMatthew G. Knepley     ierr = DMGetField(dm[r], 0, NULL, &disc);CHKERRQ(ierr);
357239a0881SMatthew G. Knepley     ierr = PetscObjectGetName(disc, &uname);CHKERRQ(ierr);
3586af0ca60SMatthew G. Knepley     ierr = PetscObjectSetName((PetscObject) u, uname);CHKERRQ(ierr);
3596af0ca60SMatthew G. Knepley     /* Setup solver */
360900f6b5bSMatthew G. Knepley     ierr = SNESReset(snes);CHKERRQ(ierr);
361900f6b5bSMatthew G. Knepley     ierr = SNESSetDM(snes, dm[r]);CHKERRQ(ierr);
3626af0ca60SMatthew G. Knepley     ierr = DMPlexSetSNESLocalFEM(dm[r], ctx, ctx, ctx);CHKERRQ(ierr);
363900f6b5bSMatthew G. Knepley     ierr = SNESSetFromOptions(snes);CHKERRQ(ierr);
364*478db826SMatthew G. Knepley     /* Set nullspace for Jacobian */
365*478db826SMatthew G. Knepley     ierr = PetscConvEstSetJacobianNullspace_Private(ce, snes);CHKERRQ(ierr);
3666af0ca60SMatthew G. Knepley     /* Create initial guess */
367900f6b5bSMatthew G. Knepley     ierr = PetscConvEstComputeInitialGuess(ce, r, dm[r], u);CHKERRQ(ierr);
368900f6b5bSMatthew G. Knepley     ierr = SNESSolve(snes, NULL, u);CHKERRQ(ierr);
369900f6b5bSMatthew G. Knepley     ierr = PetscLogEventBegin(ce->event, ce, 0, 0, 0);CHKERRQ(ierr);
370900f6b5bSMatthew G. Knepley     ierr = PetscConvEstComputeError(ce, r, dm[r], u, &ce->errors[r*ce->Nf]);CHKERRQ(ierr);
371900f6b5bSMatthew G. Knepley     ierr = PetscLogEventEnd(ce->event, ce, 0, 0, 0);CHKERRQ(ierr);
372adfa7136SMatthew G. Knepley     for (f = 0; f < ce->Nf; ++f) {
373a56d3bf6SMatthew G. Knepley       PetscSection s, fs;
374a56d3bf6SMatthew G. Knepley       PetscInt     lsize;
375a56d3bf6SMatthew G. Knepley 
376a56d3bf6SMatthew G. Knepley       /* Could use DMGetOutputDM() to add in Dirichlet dofs */
37792fd8e1eSJed Brown       ierr = DMGetLocalSection(dm[r], &s);CHKERRQ(ierr);
378a56d3bf6SMatthew G. Knepley       ierr = PetscSectionGetField(s, f, &fs);CHKERRQ(ierr);
379a56d3bf6SMatthew G. Knepley       ierr = PetscSectionGetConstrainedStorageSize(fs, &lsize);CHKERRQ(ierr);
380900f6b5bSMatthew G. Knepley       ierr = MPI_Allreduce(&lsize, &dof[r*ce->Nf+f], 1, MPIU_INT, MPI_SUM, PetscObjectComm((PetscObject) snes));CHKERRQ(ierr);
381900f6b5bSMatthew G. Knepley       ierr = PetscLogEventSetDof(ce->event, f, dof[r*ce->Nf+f]);CHKERRQ(ierr);
382900f6b5bSMatthew G. Knepley       ierr = PetscLogEventSetError(ce->event, f, ce->errors[r*ce->Nf+f]);CHKERRQ(ierr);
383adfa7136SMatthew G. Knepley     }
3846af0ca60SMatthew G. Knepley     /* Monitor */
385f2cacb80SMatthew G. Knepley     ierr = PetscConvEstMonitorDefault(ce, r);CHKERRQ(ierr);
38673269098SMatthew G. Knepley     if (!r) {
38773269098SMatthew G. Knepley       /* PCReset() does not wipe out the level structure */
38873269098SMatthew G. Knepley       KSP ksp;
38973269098SMatthew G. Knepley       PC  pc;
39073269098SMatthew G. Knepley 
391900f6b5bSMatthew G. Knepley       ierr = SNESGetKSP(snes, &ksp);CHKERRQ(ierr);
39273269098SMatthew G. Knepley       ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr);
39373269098SMatthew G. Knepley       ierr = PCMGGetLevels(pc, &oldnlev);CHKERRQ(ierr);
39473269098SMatthew G. Knepley     }
3956af0ca60SMatthew G. Knepley     /* Cleanup */
3966af0ca60SMatthew G. Knepley     ierr = VecDestroy(&u);CHKERRQ(ierr);
397adfa7136SMatthew G. Knepley     ierr = PetscLogStagePop();CHKERRQ(ierr);
3986af0ca60SMatthew G. Knepley   }
3996af0ca60SMatthew G. Knepley   for (r = 1; r <= Nr; ++r) {
4006af0ca60SMatthew G. Knepley     ierr = DMDestroy(&dm[r]);CHKERRQ(ierr);
4016af0ca60SMatthew G. Knepley   }
4026af0ca60SMatthew G. Knepley   /* Fit convergence rate */
4036af0ca60SMatthew G. Knepley   ierr = PetscMalloc2(Nr+1, &x, Nr+1, &y);CHKERRQ(ierr);
40446079b62SMatthew G. Knepley   for (f = 0; f < ce->Nf; ++f) {
4056af0ca60SMatthew G. Knepley     for (r = 0; r <= Nr; ++r) {
406adfa7136SMatthew G. Knepley       x[r] = PetscLog10Real(dof[r*ce->Nf+f]);
40746079b62SMatthew G. Knepley       y[r] = PetscLog10Real(ce->errors[r*ce->Nf+f]);
4086af0ca60SMatthew G. Knepley     }
409bebf13c0SMatthew G. Knepley     ierr = PetscLinearRegression(Nr+1, x, y, &slope, &intercept);CHKERRQ(ierr);
4106af0ca60SMatthew G. Knepley     /* Since h^{-dim} = N, lg err = s lg N + b = -s dim lg h + b */
41146079b62SMatthew G. Knepley     alpha[f] = -slope * dim;
41246079b62SMatthew G. Knepley   }
41346079b62SMatthew G. Knepley   ierr = PetscFree2(x, y);CHKERRQ(ierr);
4146af0ca60SMatthew G. Knepley   ierr = PetscFree2(dm, dof);CHKERRQ(ierr);
4152cae373cSMatthew G. Knepley   /* Restore solver */
416900f6b5bSMatthew G. Knepley   ierr = SNESReset(snes);CHKERRQ(ierr);
41773269098SMatthew G. Knepley   {
41873269098SMatthew G. Knepley     /* PCReset() does not wipe out the level structure */
41973269098SMatthew G. Knepley     KSP ksp;
42073269098SMatthew G. Knepley     PC  pc;
42173269098SMatthew G. Knepley 
422900f6b5bSMatthew G. Knepley     ierr = SNESGetKSP(snes, &ksp);CHKERRQ(ierr);
42373269098SMatthew G. Knepley     ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr);
42473269098SMatthew G. Knepley     ierr = PCMGSetLevels(pc, oldnlev, NULL);CHKERRQ(ierr);
42573269098SMatthew G. Knepley     ierr = DMSetRefineLevel(ce->idm, oldlevel);CHKERRQ(ierr); /* The damn DMCoarsen() calls in PCMG can reset this */
42673269098SMatthew G. Knepley   }
427900f6b5bSMatthew G. Knepley   ierr = SNESSetDM(snes, ce->idm);CHKERRQ(ierr);
4282cae373cSMatthew G. Knepley   ierr = DMPlexSetSNESLocalFEM(ce->idm, ctx, ctx, ctx);CHKERRQ(ierr);
429900f6b5bSMatthew G. Knepley   ierr = SNESSetFromOptions(snes);CHKERRQ(ierr);
430*478db826SMatthew G. Knepley   ierr = PetscConvEstSetJacobianNullspace_Private(ce, snes);CHKERRQ(ierr);
431900f6b5bSMatthew G. Knepley   PetscFunctionReturn(0);
432900f6b5bSMatthew G. Knepley }
433900f6b5bSMatthew G. Knepley 
434900f6b5bSMatthew G. Knepley /*@
435900f6b5bSMatthew G. Knepley   PetscConvEstGetConvRate - Returns an estimate of the convergence rate for the discretization
436900f6b5bSMatthew G. Knepley 
437900f6b5bSMatthew G. Knepley   Not collective
438900f6b5bSMatthew G. Knepley 
439900f6b5bSMatthew G. Knepley   Input Parameter:
440900f6b5bSMatthew G. Knepley . ce   - The PetscConvEst object
441900f6b5bSMatthew G. Knepley 
442900f6b5bSMatthew G. Knepley   Output Parameter:
443900f6b5bSMatthew G. Knepley . alpha - The convergence rate for each field
444900f6b5bSMatthew G. Knepley 
445900f6b5bSMatthew G. Knepley   Note: The convergence rate alpha is defined by
446900f6b5bSMatthew G. Knepley $ || u_\Delta - u_exact || < C \Delta^alpha
447900f6b5bSMatthew G. Knepley where u_\Delta is the discrete solution, and Delta is a measure of the discretization size. We usually use h for the
448900f6b5bSMatthew G. Knepley spatial resolution and \Delta t for the temporal resolution.
449900f6b5bSMatthew G. Knepley 
450900f6b5bSMatthew G. Knepley We solve a series of problems using increasing resolution (refined meshes or decreased timesteps), calculate an error
451900f6b5bSMatthew G. Knepley based upon the exact solution in the DS, and then fit the result to our model above using linear regression.
452900f6b5bSMatthew G. Knepley 
453900f6b5bSMatthew G. Knepley   Options database keys:
454900f6b5bSMatthew G. Knepley + -snes_convergence_estimate : Execute convergence estimation inside SNESSolve() and print out the rate
455900f6b5bSMatthew G. Knepley - -ts_convergence_estimate : Execute convergence estimation inside TSSolve() and print out the rate
456900f6b5bSMatthew G. Knepley 
457900f6b5bSMatthew G. Knepley   Level: intermediate
458900f6b5bSMatthew G. Knepley 
459900f6b5bSMatthew G. Knepley .seealso: PetscConvEstSetSolver(), PetscConvEstCreate(), PetscConvEstGetConvRate(), SNESSolve(), TSSolve()
460900f6b5bSMatthew G. Knepley @*/
461900f6b5bSMatthew G. Knepley PetscErrorCode PetscConvEstGetConvRate(PetscConvEst ce, PetscReal alpha[])
462900f6b5bSMatthew G. Knepley {
463900f6b5bSMatthew G. Knepley   PetscInt       f;
464900f6b5bSMatthew G. Knepley   PetscErrorCode ierr;
465900f6b5bSMatthew G. Knepley 
466900f6b5bSMatthew G. Knepley   PetscFunctionBegin;
467900f6b5bSMatthew G. Knepley   if (ce->event < 0) {ierr = PetscLogEventRegister("ConvEst Error", PETSC_OBJECT_CLASSID, &ce->event);CHKERRQ(ierr);}
468900f6b5bSMatthew G. Knepley   for (f = 0; f < ce->Nf; ++f) alpha[f] = 0.0;
469900f6b5bSMatthew G. Knepley   ierr = (*ce->ops->getconvrate)(ce, alpha);CHKERRQ(ierr);
4706af0ca60SMatthew G. Knepley   PetscFunctionReturn(0);
4716af0ca60SMatthew G. Knepley }
4726af0ca60SMatthew G. Knepley 
4736af0ca60SMatthew G. Knepley /*@
4746af0ca60SMatthew G. Knepley   PetscConvEstRateView - Displays the convergence rate to a viewer
4756af0ca60SMatthew G. Knepley 
4766af0ca60SMatthew G. Knepley    Collective on SNES
4776af0ca60SMatthew G. Knepley 
4786af0ca60SMatthew G. Knepley    Parameter:
4796af0ca60SMatthew G. Knepley +  snes - iterative context obtained from SNESCreate()
48046079b62SMatthew G. Knepley .  alpha - the convergence rate for each field
4816af0ca60SMatthew G. Knepley -  viewer - the viewer to display the reason
4826af0ca60SMatthew G. Knepley 
4836af0ca60SMatthew G. Knepley    Options Database Keys:
4846af0ca60SMatthew G. Knepley .  -snes_convergence_estimate - print the convergence rate
4856af0ca60SMatthew G. Knepley 
4866af0ca60SMatthew G. Knepley    Level: developer
4876af0ca60SMatthew G. Knepley 
4886af0ca60SMatthew G. Knepley .seealso: PetscConvEstGetRate()
4896af0ca60SMatthew G. Knepley @*/
490a56d3bf6SMatthew G. Knepley PetscErrorCode PetscConvEstRateView(PetscConvEst ce, const PetscReal alpha[], PetscViewer viewer)
4916af0ca60SMatthew G. Knepley {
4926af0ca60SMatthew G. Knepley   PetscBool      isAscii;
4936af0ca60SMatthew G. Knepley   PetscErrorCode ierr;
4946af0ca60SMatthew G. Knepley 
4956af0ca60SMatthew G. Knepley   PetscFunctionBegin;
4966af0ca60SMatthew G. Knepley   ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERASCII, &isAscii);CHKERRQ(ierr);
4976af0ca60SMatthew G. Knepley   if (isAscii) {
498900f6b5bSMatthew G. Knepley     PetscInt Nf = ce->Nf, f;
499a56d3bf6SMatthew G. Knepley 
5006af0ca60SMatthew G. Knepley     ierr = PetscViewerASCIIAddTab(viewer, ((PetscObject) ce)->tablevel);CHKERRQ(ierr);
501adfa7136SMatthew G. Knepley     ierr = PetscViewerASCIIPrintf(viewer, "L_2 convergence rate: ");CHKERRQ(ierr);
502a56d3bf6SMatthew G. Knepley     if (Nf > 1) {ierr = PetscViewerASCIIPrintf(viewer, "[");CHKERRQ(ierr);}
503a56d3bf6SMatthew G. Knepley     for (f = 0; f < Nf; ++f) {
50446079b62SMatthew G. Knepley       if (f > 0) {ierr = PetscViewerASCIIPrintf(viewer, ", ");CHKERRQ(ierr);}
5055f9b3039SMatthew G. Knepley       ierr = PetscViewerASCIIPrintf(viewer, "%#.2g", (double) alpha[f]);CHKERRQ(ierr);
50646079b62SMatthew G. Knepley     }
507a56d3bf6SMatthew G. Knepley     if (Nf > 1) {ierr = PetscViewerASCIIPrintf(viewer, "]");CHKERRQ(ierr);}
508adfa7136SMatthew G. Knepley     ierr = PetscViewerASCIIPrintf(viewer, "\n");CHKERRQ(ierr);
5096af0ca60SMatthew G. Knepley     ierr = PetscViewerASCIISubtractTab(viewer, ((PetscObject) ce)->tablevel);CHKERRQ(ierr);
5106af0ca60SMatthew G. Knepley   }
5116af0ca60SMatthew G. Knepley   PetscFunctionReturn(0);
5126af0ca60SMatthew G. Knepley }
513900f6b5bSMatthew G. Knepley 
514900f6b5bSMatthew G. Knepley /*@
515900f6b5bSMatthew G. Knepley   PetscConvEstCreate - Create a PetscConvEst object
516900f6b5bSMatthew G. Knepley 
517900f6b5bSMatthew G. Knepley   Collective
518900f6b5bSMatthew G. Knepley 
519900f6b5bSMatthew G. Knepley   Input Parameter:
520900f6b5bSMatthew G. Knepley . comm - The communicator for the PetscConvEst object
521900f6b5bSMatthew G. Knepley 
522900f6b5bSMatthew G. Knepley   Output Parameter:
523900f6b5bSMatthew G. Knepley . ce   - The PetscConvEst object
524900f6b5bSMatthew G. Knepley 
525900f6b5bSMatthew G. Knepley   Level: beginner
526900f6b5bSMatthew G. Knepley 
527900f6b5bSMatthew G. Knepley .seealso: PetscConvEstDestroy(), PetscConvEstGetConvRate()
528900f6b5bSMatthew G. Knepley @*/
529900f6b5bSMatthew G. Knepley PetscErrorCode PetscConvEstCreate(MPI_Comm comm, PetscConvEst *ce)
530900f6b5bSMatthew G. Knepley {
531900f6b5bSMatthew G. Knepley   PetscErrorCode ierr;
532900f6b5bSMatthew G. Knepley 
533900f6b5bSMatthew G. Knepley   PetscFunctionBegin;
534900f6b5bSMatthew G. Knepley   PetscValidPointer(ce, 2);
535900f6b5bSMatthew G. Knepley   ierr = PetscSysInitializePackage();CHKERRQ(ierr);
536900f6b5bSMatthew G. Knepley   ierr = PetscHeaderCreate(*ce, PETSC_OBJECT_CLASSID, "PetscConvEst", "ConvergenceEstimator", "SNES", comm, PetscConvEstDestroy, PetscConvEstView);CHKERRQ(ierr);
537900f6b5bSMatthew G. Knepley   (*ce)->monitor = PETSC_FALSE;
5382e61be88SMatthew G. Knepley   (*ce)->r       = 2.0;
539900f6b5bSMatthew G. Knepley   (*ce)->Nr      = 4;
540900f6b5bSMatthew G. Knepley   (*ce)->event   = -1;
541900f6b5bSMatthew G. Knepley   (*ce)->ops->setsolver    = PetscConvEstSetSNES_Private;
542900f6b5bSMatthew G. Knepley   (*ce)->ops->initguess    = PetscConvEstInitGuessSNES_Private;
543900f6b5bSMatthew G. Knepley   (*ce)->ops->computeerror = PetscConvEstComputeErrorSNES_Private;
544900f6b5bSMatthew G. Knepley   (*ce)->ops->getconvrate  = PetscConvEstGetConvRateSNES_Private;
545900f6b5bSMatthew G. Knepley   PetscFunctionReturn(0);
546900f6b5bSMatthew G. Knepley }
547