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