Lines Matching refs:dm
34 static PetscErrorCode DMLocalSNESGetContext(DM dm, DMSNES sdm, DMSNES_Local **dmlocalsnes) in DMLocalSNESGetContext() argument
51 DM dm; in SNESComputeObjective_DMLocal() local
58 PetscCall(SNESGetDM(snes, &dm)); in SNESComputeObjective_DMLocal()
59 PetscCall(DMGetLocalVector(dm, &Xloc)); in SNESComputeObjective_DMLocal()
62 …if (dmlocalsnes->boundarylocal) PetscCall((*dmlocalsnes->boundarylocal)(dm, Xloc, dmlocalsnes->bou… in SNESComputeObjective_DMLocal()
63 PetscCall(DMGlobalToLocalBegin(dm, X, INSERT_VALUES, Xloc)); in SNESComputeObjective_DMLocal()
64 PetscCall(DMGlobalToLocalEnd(dm, X, INSERT_VALUES, Xloc)); in SNESComputeObjective_DMLocal()
66 PetscCall(DMHasBasisTransform(dm, &transform)); in SNESComputeObjective_DMLocal()
67 …if (transform && dmlocalsnes->boundarylocal) PetscCall((*dmlocalsnes->boundarylocal)(dm, Xloc, dml… in SNESComputeObjective_DMLocal()
69 PetscCall((*dmlocalsnes->objectivelocal)(dm, Xloc, obj, dmlocalsnes->objectivelocalctx)); in SNESComputeObjective_DMLocal()
72 PetscCall(DMRestoreLocalVector(dm, &Xloc)); in SNESComputeObjective_DMLocal()
79 DM dm; in SNESComputeFunction_DMLocal() local
87 PetscCall(SNESGetDM(snes, &dm)); in SNESComputeFunction_DMLocal()
88 PetscCall(DMGetLocalVector(dm, &Xloc)); in SNESComputeFunction_DMLocal()
89 PetscCall(DMGetLocalVector(dm, &Floc)); in SNESComputeFunction_DMLocal()
93 …if (dmlocalsnes->boundarylocal) PetscCall((*dmlocalsnes->boundarylocal)(dm, Xloc, dmlocalsnes->bou… in SNESComputeFunction_DMLocal()
94 PetscCall(DMGlobalToLocalBegin(dm, X, INSERT_VALUES, Xloc)); in SNESComputeFunction_DMLocal()
95 PetscCall(DMGlobalToLocalEnd(dm, X, INSERT_VALUES, Xloc)); in SNESComputeFunction_DMLocal()
97 PetscCall(DMHasBasisTransform(dm, &transform)); in SNESComputeFunction_DMLocal()
98 …if (transform && dmlocalsnes->boundarylocal) PetscCall((*dmlocalsnes->boundarylocal)(dm, Xloc, dml… in SNESComputeFunction_DMLocal()
100 PetscCall((*dmlocalsnes->residuallocal)(dm, Xloc, Floc, dmlocalsnes->residuallocalctx)); in SNESComputeFunction_DMLocal()
103 PetscCall(DMLocalToGlobalBegin(dm, Floc, ADD_VALUES, F)); in SNESComputeFunction_DMLocal()
104 PetscCall(DMLocalToGlobalEnd(dm, Floc, ADD_VALUES, F)); in SNESComputeFunction_DMLocal()
105 PetscCall(DMRestoreLocalVector(dm, &Floc)); in SNESComputeFunction_DMLocal()
106 PetscCall(DMRestoreLocalVector(dm, &Xloc)); in SNESComputeFunction_DMLocal()
130 DM dm; in SNESComputeJacobian_DMLocal() local
135 PetscCall(SNESGetDM(snes, &dm)); in SNESComputeJacobian_DMLocal()
137 PetscCall(DMGetLocalVector(dm, &Xloc)); in SNESComputeJacobian_DMLocal()
140 …if (dmlocalsnes->boundarylocal) PetscCall((*dmlocalsnes->boundarylocal)(dm, Xloc, dmlocalsnes->bou… in SNESComputeJacobian_DMLocal()
141 PetscCall(DMGlobalToLocalBegin(dm, X, INSERT_VALUES, Xloc)); in SNESComputeJacobian_DMLocal()
142 PetscCall(DMGlobalToLocalEnd(dm, X, INSERT_VALUES, Xloc)); in SNESComputeJacobian_DMLocal()
144 PetscCall(DMHasBasisTransform(dm, &transform)); in SNESComputeJacobian_DMLocal()
145 …if (transform && dmlocalsnes->boundarylocal) PetscCall((*dmlocalsnes->boundarylocal)(dm, Xloc, dml… in SNESComputeJacobian_DMLocal()
147 PetscCall((*dmlocalsnes->jacobianlocal)(dm, Xloc, A, B, dmlocalsnes->jacobianlocalctx)); in SNESComputeJacobian_DMLocal()
149 PetscCall(DMRestoreLocalVector(dm, &Xloc)); in SNESComputeJacobian_DMLocal()
152 PetscCall(PetscObjectQuery((PetscObject)dm, "DMDASNES_FDCOLORING", (PetscObject *)&fdcoloring)); in SNESComputeJacobian_DMLocal()
156 PetscCall(DMCreateColoring(dm, dm->coloringtype, &coloring)); in SNESComputeJacobian_DMLocal()
159 switch (dm->coloringtype) { in SNESComputeJacobian_DMLocal()
164 …ject)snes), PETSC_ERR_SUP, "No support for coloring type '%s'", ISColoringTypes[dm->coloringtype]); in SNESComputeJacobian_DMLocal()
166 PetscCall(PetscObjectSetOptionsPrefix((PetscObject)fdcoloring, ((PetscObject)dm)->prefix)); in SNESComputeJacobian_DMLocal()
169 … PetscCall(PetscObjectCompose((PetscObject)dm, "DMDASNES_FDCOLORING", (PetscObject)fdcoloring)); in SNESComputeJacobian_DMLocal()
178 PetscCall(PetscObjectDereference((PetscObject)dm)); in SNESComputeJacobian_DMLocal()
206 PetscErrorCode DMSNESSetObjectiveLocal(DM dm, PetscErrorCode (*func)(DM, Vec, PetscReal *, void *),… in DMSNESSetObjectiveLocal() argument
212 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMSNESSetObjectiveLocal()
213 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMSNESSetObjectiveLocal()
214 PetscCall(DMLocalSNESGetContext(dm, sdm, &dmlocalsnes)); in DMSNESSetObjectiveLocal()
219 PetscCall(DMSNESSetObjective(dm, SNESComputeObjective_DMLocal, dmlocalsnes)); in DMSNESSetObjectiveLocal()
245 PetscErrorCode DMSNESSetFunctionLocal(DM dm, PetscErrorCode (*func)(DM dm, Vec x, Vec f, PetscCtx c… in DMSNESSetFunctionLocal() argument
251 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMSNESSetFunctionLocal()
252 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMSNESSetFunctionLocal()
253 PetscCall(DMLocalSNESGetContext(dm, sdm, &dmlocalsnes)); in DMSNESSetFunctionLocal()
258 PetscCall(DMSNESSetFunction(dm, SNESComputeFunction_DMLocal, dmlocalsnes)); in DMSNESSetFunctionLocal()
260 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMLocal, dmlocalsnes)); in DMSNESSetFunctionLocal()
284 PetscErrorCode DMSNESSetBoundaryLocal(DM dm, PetscErrorCode (*func)(DM dm, Vec X, PetscCtx ctx), Pe… in DMSNESSetBoundaryLocal() argument
290 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMSNESSetBoundaryLocal()
291 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMSNESSetBoundaryLocal()
292 PetscCall(DMLocalSNESGetContext(dm, sdm, &dmlocalsnes)); in DMSNESSetBoundaryLocal()
320 PetscErrorCode DMSNESSetJacobianLocal(DM dm, PetscErrorCode (*func)(DM dm, Vec X, Mat J, Mat Jp, Pe… in DMSNESSetJacobianLocal() argument
326 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMSNESSetJacobianLocal()
327 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMSNESSetJacobianLocal()
328 PetscCall(DMLocalSNESGetContext(dm, sdm, &dmlocalsnes)); in DMSNESSetJacobianLocal()
333 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMLocal, dmlocalsnes)); in DMSNESSetJacobianLocal()
353 PetscErrorCode DMSNESGetObjectiveLocal(DM dm, PetscErrorCode (**func)(DM, Vec, PetscReal *, void *)… in DMSNESGetObjectiveLocal() argument
359 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMSNESGetObjectiveLocal()
360 PetscCall(DMGetDMSNES(dm, &sdm)); in DMSNESGetObjectiveLocal()
361 PetscCall(DMLocalSNESGetContext(dm, sdm, &dmlocalsnes)); in DMSNESGetObjectiveLocal()
383 PetscErrorCode DMSNESGetFunctionLocal(DM dm, PetscErrorCode (**func)(DM, Vec, Vec, void *), PetscCt… in DMSNESGetFunctionLocal() argument
389 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMSNESGetFunctionLocal()
390 PetscCall(DMGetDMSNES(dm, &sdm)); in DMSNESGetFunctionLocal()
391 PetscCall(DMLocalSNESGetContext(dm, sdm, &dmlocalsnes)); in DMSNESGetFunctionLocal()
413 PetscErrorCode DMSNESGetBoundaryLocal(DM dm, PetscErrorCode (**func)(DM, Vec, void *), PetscCtxRt c… in DMSNESGetBoundaryLocal() argument
419 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMSNESGetBoundaryLocal()
420 PetscCall(DMGetDMSNES(dm, &sdm)); in DMSNESGetBoundaryLocal()
421 PetscCall(DMLocalSNESGetContext(dm, sdm, &dmlocalsnes)); in DMSNESGetBoundaryLocal()
443 PetscErrorCode DMSNESGetJacobianLocal(DM dm, PetscErrorCode (**func)(DM, Vec, Mat, Mat, void *), Pe… in DMSNESGetJacobianLocal() argument
449 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMSNESGetJacobianLocal()
450 PetscCall(DMGetDMSNES(dm, &sdm)); in DMSNESGetJacobianLocal()
451 PetscCall(DMLocalSNESGetContext(dm, sdm, &dmlocalsnes)); in DMSNESGetJacobianLocal()