1 /* 2 Defines a SNES that can consist of a collection of SNESes on patches of the domain 3 */ 4 #include <petsc/private/snesimpl.h> /*I "petscsnes.h" I*/ 5 #include <petsc/private/pcpatchimpl.h> /* We need internal access to PCPatch right now, until that part is moved to Plex */ 6 #include <petscsf.h> 7 8 typedef struct { 9 PC pc; /* The linear patch preconditioner */ 10 } SNES_Patch; 11 12 static PetscErrorCode SNESPatchComputeResidual_Private(SNES snes, Vec x, Vec F, void *ctx) 13 { 14 PC pc = (PC) ctx; 15 PC_PATCH *pcpatch = (PC_PATCH *) pc->data; 16 PetscInt pt, size; 17 const PetscInt *indices; 18 const PetscScalar *X; 19 PetscScalar *XWithArtificial; 20 PetscErrorCode ierr; 21 22 PetscFunctionBegin; 23 24 /* scatter from x to patch->patchStateWithArtificial[pt] */ 25 pt = pcpatch->currentPatch; 26 ierr = ISGetSize(pcpatch->dofMappingWithoutToWithArtificial[pt], &size);CHKERRQ(ierr); 27 28 ierr = ISGetIndices(pcpatch->dofMappingWithoutToWithArtificial[pt], &indices);CHKERRQ(ierr); 29 ierr = VecGetArrayRead(x, &X);CHKERRQ(ierr); 30 ierr = VecGetArray(pcpatch->patchStateWithArtificial[pt], &XWithArtificial);CHKERRQ(ierr); 31 32 for (PetscInt i = 0; i < size; ++i) { 33 XWithArtificial[indices[i]] = X[i]; 34 } 35 36 ierr = VecRestoreArray(pcpatch->patchStateWithArtificial[pt], &XWithArtificial);CHKERRQ(ierr); 37 ierr = VecRestoreArrayRead(x, &X);CHKERRQ(ierr); 38 ierr = ISRestoreIndices(pcpatch->dofMappingWithoutToWithArtificial[pt], &indices);CHKERRQ(ierr); 39 40 ierr = PCPatchComputeFunction_Internal(pc, pcpatch->patchStateWithArtificial[pt], F, pt);CHKERRQ(ierr); 41 PetscFunctionReturn(0); 42 } 43 44 static PetscErrorCode SNESPatchComputeJacobian_Private(SNES snes, Vec x, Mat J, Mat M, void *ctx) 45 { 46 PC pc = (PC) ctx; 47 PC_PATCH *pcpatch = (PC_PATCH *) pc->data; 48 PetscInt pt, size; 49 const PetscInt *indices; 50 const PetscScalar *X; 51 PetscScalar *XWithArtificial; 52 PetscErrorCode ierr; 53 54 PetscFunctionBegin; 55 /* scatter from x to patch->patchStateWithArtificial[pt] */ 56 pt = pcpatch->currentPatch; 57 ierr = ISGetSize(pcpatch->dofMappingWithoutToWithArtificial[pt], &size);CHKERRQ(ierr); 58 59 ierr = ISGetIndices(pcpatch->dofMappingWithoutToWithArtificial[pt], &indices);CHKERRQ(ierr); 60 ierr = VecGetArrayRead(x, &X);CHKERRQ(ierr); 61 ierr = VecGetArray(pcpatch->patchStateWithArtificial[pt], &XWithArtificial);CHKERRQ(ierr); 62 63 for (PetscInt i = 0; i < size; ++i) { 64 XWithArtificial[indices[i]] = X[i]; 65 } 66 67 ierr = VecRestoreArray(pcpatch->patchStateWithArtificial[pt], &XWithArtificial);CHKERRQ(ierr); 68 ierr = VecRestoreArrayRead(x, &X);CHKERRQ(ierr); 69 ierr = ISRestoreIndices(pcpatch->dofMappingWithoutToWithArtificial[pt], &indices);CHKERRQ(ierr); 70 71 ierr = PCPatchComputeOperator_Internal(pc, pcpatch->patchStateWithArtificial[pt], M, pcpatch->currentPatch, PETSC_FALSE);CHKERRQ(ierr); 72 PetscFunctionReturn(0); 73 } 74 75 static PetscErrorCode PCSetUp_PATCH_Nonlinear(PC pc) 76 { 77 PC_PATCH *patch = (PC_PATCH *) pc->data; 78 const char *prefix; 79 PetscInt i, pStart, dof; 80 PetscErrorCode ierr; 81 82 PetscFunctionBegin; 83 if (!pc->setupcalled) { 84 ierr = PetscMalloc1(patch->npatch, &patch->solver);CHKERRQ(ierr); 85 ierr = PCGetOptionsPrefix(pc, &prefix);CHKERRQ(ierr); 86 ierr = PetscSectionGetChart(patch->gtolCounts, &pStart, NULL);CHKERRQ(ierr); 87 for (i = 0; i < patch->npatch; ++i) { 88 SNES snes; 89 KSP subksp; 90 91 ierr = SNESCreate(PETSC_COMM_SELF, &snes);CHKERRQ(ierr); 92 ierr = SNESSetOptionsPrefix(snes, prefix);CHKERRQ(ierr); 93 ierr = SNESAppendOptionsPrefix(snes, "sub_");CHKERRQ(ierr); 94 ierr = PetscObjectIncrementTabLevel((PetscObject) snes, (PetscObject) pc, 2);CHKERRQ(ierr); 95 ierr = SNESGetKSP(snes, &subksp);CHKERRQ(ierr); 96 ierr = PetscObjectIncrementTabLevel((PetscObject) subksp, (PetscObject) pc, 2);CHKERRQ(ierr); 97 ierr = PetscLogObjectParent((PetscObject) pc, (PetscObject) snes);CHKERRQ(ierr); 98 patch->solver[i] = (PetscObject) snes; 99 } 100 101 ierr = PetscMalloc1(patch->npatch, &patch->patchResidual);CHKERRQ(ierr); 102 ierr = PetscMalloc1(patch->npatch, &patch->patchState);CHKERRQ(ierr); 103 ierr = PetscMalloc1(patch->npatch, &patch->patchStateWithArtificial);CHKERRQ(ierr); 104 for (i = 0; i < patch->npatch; ++i) { 105 ierr = VecDuplicate(patch->patchRHS[i], &patch->patchResidual[i]);CHKERRQ(ierr); 106 ierr = VecDuplicate(patch->patchUpdate[i], &patch->patchState[i]);CHKERRQ(ierr); 107 108 ierr = PetscSectionGetDof(patch->gtolCountsWithArtificial, i+pStart, &dof);CHKERRQ(ierr); 109 ierr = VecCreateSeq(PETSC_COMM_SELF, dof, &patch->patchStateWithArtificial[i]);CHKERRQ(ierr); 110 ierr = VecSetUp(patch->patchStateWithArtificial[i]);CHKERRQ(ierr); 111 } 112 ierr = VecDuplicate(patch->localUpdate, &patch->localState);CHKERRQ(ierr); 113 } 114 for (i = 0; i < patch->npatch; ++i) { 115 SNES snes = (SNES) patch->solver[i]; 116 117 ierr = SNESSetFunction(snes, patch->patchResidual[i], SNESPatchComputeResidual_Private, pc);CHKERRQ(ierr); 118 ierr = SNESSetJacobian(snes, patch->mat[i], patch->mat[i], SNESPatchComputeJacobian_Private, pc);CHKERRQ(ierr); 119 } 120 if (!pc->setupcalled && patch->optionsSet) for (i = 0; i < patch->npatch; ++i) {ierr = SNESSetFromOptions((SNES) patch->solver[i]);CHKERRQ(ierr);} 121 PetscFunctionReturn(0); 122 } 123 124 static PetscErrorCode PCApply_PATCH_Nonlinear(PC pc, PetscInt i, Vec patchRHS, Vec patchUpdate) 125 { 126 PC_PATCH *patch = (PC_PATCH *) pc->data; 127 PetscInt pStart; 128 PetscErrorCode ierr; 129 130 PetscFunctionBegin; 131 patch->currentPatch = i; 132 ierr = PetscLogEventBegin(PC_Patch_Solve, pc, 0, 0, 0);CHKERRQ(ierr); 133 134 /* Scatter the overlapped global state to our patch state vector */ 135 ierr = PetscSectionGetChart(patch->gtolCounts, &pStart, NULL);CHKERRQ(ierr); 136 ierr = PCPatch_ScatterLocal_Private(pc, i+pStart, patch->localState, patch->patchState[i], INSERT_VALUES, SCATTER_FORWARD, PETSC_FALSE);CHKERRQ(ierr); 137 ierr = PCPatch_ScatterLocal_Private(pc, i+pStart, patch->localState, patch->patchStateWithArtificial[i], INSERT_VALUES, SCATTER_FORWARD, PETSC_TRUE);CHKERRQ(ierr); 138 139 /* Set initial guess to be current state*/ 140 ierr = VecCopy(patch->patchState[i], patchUpdate);CHKERRQ(ierr); 141 /* Solve for new state */ 142 ierr = SNESSolve((SNES) patch->solver[i], patchRHS, patchUpdate);CHKERRQ(ierr); 143 /* To compute update, subtract off previous state */ 144 ierr = VecAXPY(patchUpdate, -1.0, patch->patchState[i]);CHKERRQ(ierr); 145 146 ierr = PetscLogEventEnd(PC_Patch_Solve, pc, 0, 0, 0);CHKERRQ(ierr); 147 PetscFunctionReturn(0); 148 } 149 150 static PetscErrorCode PCReset_PATCH_Nonlinear(PC pc) 151 { 152 PC_PATCH *patch = (PC_PATCH *) pc->data; 153 PetscInt i; 154 PetscErrorCode ierr; 155 156 PetscFunctionBegin; 157 158 if (patch->solver) { 159 for (i = 0; i < patch->npatch; ++i) {ierr = SNESReset((SNES) patch->solver[i]);CHKERRQ(ierr);} 160 } 161 162 if (patch->patchResidual) { 163 for (i = 0; i < patch->npatch; ++i) {ierr = VecDestroy(&patch->patchResidual[i]);CHKERRQ(ierr);} 164 ierr = PetscFree(patch->patchResidual);CHKERRQ(ierr); 165 } 166 167 if (patch->patchState) { 168 for (i = 0; i < patch->npatch; ++i) {ierr = VecDestroy(&patch->patchState[i]);CHKERRQ(ierr);} 169 ierr = PetscFree(patch->patchState);CHKERRQ(ierr); 170 } 171 172 if (patch->patchStateWithArtificial) { 173 for (i = 0; i < patch->npatch; ++i) {ierr = VecDestroy(&patch->patchStateWithArtificial[i]);CHKERRQ(ierr);} 174 ierr = PetscFree(patch->patchStateWithArtificial);CHKERRQ(ierr); 175 } 176 177 ierr = VecDestroy(&patch->localState);CHKERRQ(ierr); 178 179 PetscFunctionReturn(0); 180 } 181 182 static PetscErrorCode PCDestroy_PATCH_Nonlinear(PC pc) 183 { 184 PC_PATCH *patch = (PC_PATCH *) pc->data; 185 PetscInt i; 186 PetscErrorCode ierr; 187 188 PetscFunctionBegin; 189 if (patch->solver) { 190 for (i = 0; i < patch->npatch; ++i) {ierr = SNESDestroy((SNES *) &patch->solver[i]);CHKERRQ(ierr);} 191 ierr = PetscFree(patch->solver);CHKERRQ(ierr); 192 } 193 PetscFunctionReturn(0); 194 } 195 196 static PetscErrorCode PCUpdateMultiplicative_PATCH_Nonlinear(PC pc, PetscInt i, PetscInt pStart) 197 { 198 PC_PATCH *patch = (PC_PATCH *) pc->data; 199 PetscErrorCode ierr; 200 201 PetscFunctionBegin; 202 ierr = PCPatch_ScatterLocal_Private(pc, i + pStart, patch->patchUpdate[i], patch->localState, ADD_VALUES, SCATTER_REVERSE, PETSC_FALSE);CHKERRQ(ierr); 203 PetscFunctionReturn(0); 204 } 205 206 static PetscErrorCode SNESSetUp_Patch(SNES snes) 207 { 208 SNES_Patch *patch = (SNES_Patch *) snes->data; 209 DM dm; 210 Mat dummy; 211 Vec F; 212 PetscInt n, N; 213 PetscErrorCode ierr; 214 215 PetscFunctionBegin; 216 ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 217 ierr = PCSetDM(patch->pc, dm);CHKERRQ(ierr); 218 ierr = SNESGetFunction(snes, &F, NULL, NULL);CHKERRQ(ierr); 219 ierr = VecGetLocalSize(F, &n);CHKERRQ(ierr); 220 ierr = VecGetSize(F, &N);CHKERRQ(ierr); 221 ierr = MatCreateShell(PetscObjectComm((PetscObject) snes), n, n, N, N, (void *) snes, &dummy);CHKERRQ(ierr); 222 ierr = PCSetOperators(patch->pc, dummy, dummy);CHKERRQ(ierr); 223 ierr = MatDestroy(&dummy);CHKERRQ(ierr); 224 ierr = PCSetUp(patch->pc);CHKERRQ(ierr); 225 /* allocate workspace */ 226 PetscFunctionReturn(0); 227 } 228 229 static PetscErrorCode SNESReset_Patch(SNES snes) 230 { 231 SNES_Patch *patch = (SNES_Patch *) snes->data; 232 PetscErrorCode ierr; 233 234 PetscFunctionBegin; 235 ierr = PCReset(patch->pc);CHKERRQ(ierr); 236 PetscFunctionReturn(0); 237 } 238 239 static PetscErrorCode SNESDestroy_Patch(SNES snes) 240 { 241 SNES_Patch *patch = (SNES_Patch *) snes->data; 242 PetscErrorCode ierr; 243 244 PetscFunctionBegin; 245 ierr = SNESReset_Patch(snes);CHKERRQ(ierr); 246 ierr = PCDestroy(&patch->pc);CHKERRQ(ierr); 247 ierr = PetscFree(snes->data);CHKERRQ(ierr); 248 PetscFunctionReturn(0); 249 } 250 251 static PetscErrorCode SNESSetFromOptions_Patch(PetscOptionItems *PetscOptionsObject, SNES snes) 252 { 253 SNES_Patch *patch = (SNES_Patch *) snes->data; 254 const char *prefix; 255 PetscErrorCode ierr; 256 257 PetscFunctionBegin; 258 ierr = PetscObjectGetOptionsPrefix((PetscObject)snes, &prefix);CHKERRQ(ierr); 259 ierr = PetscObjectSetOptionsPrefix((PetscObject)patch->pc, prefix);CHKERRQ(ierr); 260 ierr = PCSetFromOptions(patch->pc);CHKERRQ(ierr); 261 PetscFunctionReturn(0); 262 } 263 264 static PetscErrorCode SNESView_Patch(SNES snes,PetscViewer viewer) 265 { 266 SNES_Patch *patch = (SNES_Patch *) snes->data; 267 PetscBool iascii; 268 PetscErrorCode ierr; 269 270 PetscFunctionBegin; 271 ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERASCII, &iascii);CHKERRQ(ierr); 272 if (iascii) { 273 ierr = PetscViewerASCIIPrintf(viewer,"SNESPATCH\n");CHKERRQ(ierr); 274 } 275 ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 276 ierr = PCView(patch->pc, viewer);CHKERRQ(ierr); 277 ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 278 PetscFunctionReturn(0); 279 } 280 281 static PetscErrorCode SNESSolve_Patch(SNES snes) 282 { 283 SNES_Patch *patch = (SNES_Patch *) snes->data; 284 PC_PATCH *pcpatch = (PC_PATCH *) patch->pc->data; 285 SNESLineSearch ls; 286 Vec rhs, update, state, residual; 287 const PetscScalar *globalState = NULL; 288 PetscScalar *localState = NULL; 289 PetscInt its = 0; 290 PetscReal xnorm = 0.0, ynorm = 0.0, fnorm = 0.0; 291 PetscErrorCode ierr; 292 293 PetscFunctionBegin; 294 295 ierr = SNESGetSolution(snes, &state);CHKERRQ(ierr); 296 ierr = SNESGetSolutionUpdate(snes, &update);CHKERRQ(ierr); 297 ierr = SNESGetRhs(snes, &rhs);CHKERRQ(ierr); 298 299 ierr = SNESGetFunction(snes, &residual, NULL, NULL);CHKERRQ(ierr); 300 ierr = SNESGetLineSearch(snes, &ls);CHKERRQ(ierr); 301 302 ierr = SNESSetConvergedReason(snes, SNES_CONVERGED_ITERATING);CHKERRQ(ierr); 303 ierr = VecSet(update, 0.0);CHKERRQ(ierr); 304 ierr = SNESComputeFunction(snes, state, residual);CHKERRQ(ierr); 305 306 ierr = VecNorm(state, NORM_2, &xnorm);CHKERRQ(ierr); 307 ierr = VecNorm(residual, NORM_2, &fnorm);CHKERRQ(ierr); 308 snes->ttol = fnorm*snes->rtol; 309 310 if (snes->ops->converged) { 311 ierr = (*snes->ops->converged)(snes,its,xnorm,ynorm,fnorm,&snes->reason,snes->cnvP);CHKERRQ(ierr); 312 } else { 313 ierr = SNESConvergedSkip(snes,its,xnorm,ynorm,fnorm,&snes->reason,0);CHKERRQ(ierr); 314 } 315 ierr = SNESLogConvergenceHistory(snes, fnorm, 0);CHKERRQ(ierr); /* should we count lits from the patches? */ 316 ierr = SNESMonitor(snes, its, fnorm);CHKERRQ(ierr); 317 318 /* The main solver loop */ 319 for (its = 0; its < snes->max_its; its++) { 320 321 ierr = SNESSetIterationNumber(snes, its);CHKERRQ(ierr); 322 323 /* Scatter state vector to overlapped vector on all patches. 324 The vector pcpatch->localState is scattered to each patch 325 in PCApply_PATCH_Nonlinear. */ 326 ierr = VecGetArrayRead(state, &globalState);CHKERRQ(ierr); 327 ierr = VecGetArray(pcpatch->localState, &localState);CHKERRQ(ierr); 328 ierr = PetscSFBcastBegin(pcpatch->defaultSF, MPIU_SCALAR, globalState, localState);CHKERRQ(ierr); 329 ierr = PetscSFBcastEnd(pcpatch->defaultSF, MPIU_SCALAR, globalState, localState);CHKERRQ(ierr); 330 ierr = VecRestoreArray(pcpatch->localState, &localState);CHKERRQ(ierr); 331 ierr = VecRestoreArrayRead(state, &globalState);CHKERRQ(ierr); 332 333 /* The looping over patches happens here */ 334 ierr = PCApply(patch->pc, rhs, update); 335 336 /* Apply a line search. This will often be basic with 337 damping = 1/(max number of patches a dof can be in), 338 but not always */ 339 ierr = VecScale(update, -1.0);CHKERRQ(ierr); 340 ierr = SNESLineSearchApply(ls, state, residual, &fnorm, update);CHKERRQ(ierr); 341 342 ierr = VecNorm(state, NORM_2, &xnorm);CHKERRQ(ierr); 343 ierr = VecNorm(update, NORM_2, &ynorm);CHKERRQ(ierr); 344 345 if (snes->ops->converged) { 346 ierr = (*snes->ops->converged)(snes,its,xnorm,ynorm,fnorm,&snes->reason,snes->cnvP);CHKERRQ(ierr); 347 } else { 348 ierr = SNESConvergedSkip(snes,its,xnorm,ynorm,fnorm,&snes->reason,0);CHKERRQ(ierr); 349 } 350 ierr = SNESLogConvergenceHistory(snes, fnorm, 0);CHKERRQ(ierr); /* FIXME: should we count lits? */ 351 ierr = SNESMonitor(snes, its, fnorm);CHKERRQ(ierr); 352 } 353 354 if (its == snes->max_its) { ierr = SNESSetConvergedReason(snes, SNES_DIVERGED_MAX_IT);CHKERRQ(ierr); } 355 PetscFunctionReturn(0); 356 } 357 358 /*MC 359 SNESPATCH - Solve a nonlinear problem by composing together many nonlinear solvers on patches 360 361 Level: intermediate 362 363 Concepts: composing solvers 364 365 .seealso: SNESCreate(), SNESSetType(), SNESType (for list of available types), SNES, 366 PCPATCH 367 368 References: 369 . 1. - Peter R. Brune, Matthew G. Knepley, Barry F. Smith, and Xuemin Tu, "Composing Scalable Nonlinear Algebraic Solvers", SIAM Review, 57(4), 2015 370 371 M*/ 372 PETSC_EXTERN PetscErrorCode SNESCreate_Patch(SNES snes) 373 { 374 PetscErrorCode ierr; 375 SNES_Patch *patch; 376 PC_PATCH *patchpc; 377 378 PetscFunctionBegin; 379 ierr = PetscNewLog(snes, &patch);CHKERRQ(ierr); 380 381 snes->ops->solve = SNESSolve_Patch; 382 snes->ops->setup = SNESSetUp_Patch; 383 snes->ops->reset = SNESReset_Patch; 384 snes->ops->destroy = SNESDestroy_Patch; 385 snes->ops->setfromoptions = SNESSetFromOptions_Patch; 386 snes->ops->view = SNESView_Patch; 387 388 snes->alwayscomputesfinalresidual = PETSC_FALSE; 389 390 snes->data = (void *) patch; 391 ierr = PCCreate(PetscObjectComm((PetscObject) snes), &patch->pc);CHKERRQ(ierr); 392 ierr = PCSetType(patch->pc, PCPATCH);CHKERRQ(ierr); 393 394 patchpc = (PC_PATCH*) patch->pc->data; 395 patchpc->classname = "snes"; 396 397 patchpc->setupsolver = PCSetUp_PATCH_Nonlinear; 398 patchpc->applysolver = PCApply_PATCH_Nonlinear; 399 patchpc->resetsolver = PCReset_PATCH_Nonlinear; 400 patchpc->destroysolver = PCDestroy_PATCH_Nonlinear; 401 patchpc->updatemultiplicative = PCUpdateMultiplicative_PATCH_Nonlinear; 402 403 PetscFunctionReturn(0); 404 } 405 406 PetscErrorCode SNESPatchSetDiscretisationInfo(SNES snes, PetscInt nsubspaces, DM *dms, PetscInt *bs, PetscInt *nodesPerCell, const PetscInt **cellNodeMap, 407 const PetscInt *subspaceOffsets, PetscInt numGhostBcs, const PetscInt *ghostBcNodes, PetscInt numGlobalBcs, const PetscInt *globalBcNodes) 408 { 409 SNES_Patch *patch = (SNES_Patch *) snes->data; 410 PetscErrorCode ierr; 411 DM dm; 412 413 PetscFunctionBegin; 414 ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 415 if (!dm) SETERRQ(PetscObjectComm((PetscObject)snes), PETSC_ERR_ARG_WRONGSTATE, "DM not yet set on patch SNES\n"); 416 ierr = PCSetDM(patch->pc, dm);CHKERRQ(ierr); 417 ierr = PCPatchSetDiscretisationInfo(patch->pc, nsubspaces, dms, bs, nodesPerCell, cellNodeMap, subspaceOffsets, numGhostBcs, ghostBcNodes, numGlobalBcs, globalBcNodes);CHKERRQ(ierr); 418 PetscFunctionReturn(0); 419 } 420 421 PetscErrorCode SNESPatchSetComputeOperator(SNES snes, PetscErrorCode (*func)(PC, PetscInt, Vec, Mat, IS, PetscInt, const PetscInt *, const PetscInt *, void *), void *ctx) 422 { 423 SNES_Patch *patch = (SNES_Patch *) snes->data; 424 PetscErrorCode ierr; 425 426 PetscFunctionBegin; 427 ierr = PCPatchSetComputeOperator(patch->pc, func, ctx);CHKERRQ(ierr); 428 PetscFunctionReturn(0); 429 } 430 431 PetscErrorCode SNESPatchSetComputeFunction(SNES snes, PetscErrorCode (*func)(PC, PetscInt, Vec, Vec, IS, PetscInt, const PetscInt *, const PetscInt *, void *), void *ctx) 432 { 433 SNES_Patch *patch = (SNES_Patch *) snes->data; 434 PetscErrorCode ierr; 435 436 PetscFunctionBegin; 437 ierr = PCPatchSetComputeFunction(patch->pc, func, ctx);CHKERRQ(ierr); 438 PetscFunctionReturn(0); 439 } 440 441 PetscErrorCode SNESPatchSetConstructType(SNES snes, PCPatchConstructType ctype, PetscErrorCode (*func)(PC, PetscInt *, IS **, IS *, void *), void *ctx) 442 { 443 SNES_Patch *patch = (SNES_Patch *) snes->data; 444 PetscErrorCode ierr; 445 446 PetscFunctionBegin; 447 ierr = PCPatchSetConstructType(patch->pc, ctype, func, ctx);CHKERRQ(ierr); 448 PetscFunctionReturn(0); 449 } 450 451 PetscErrorCode SNESPatchSetCellNumbering(SNES snes, PetscSection cellNumbering) 452 { 453 SNES_Patch *patch = (SNES_Patch *) snes->data; 454 PetscErrorCode ierr; 455 456 PetscFunctionBegin; 457 ierr = PCPatchSetCellNumbering(patch->pc, cellNumbering);CHKERRQ(ierr); 458 PetscFunctionReturn(0); 459 } 460