Lines Matching refs:dm
44 static PetscErrorCode DMDASNESGetContext(DM dm, DMSNES sdm, DMSNES_DA **dmdasnes) in DMDASNESGetContext() argument
59 DM dm; in SNESComputeFunction_DMDA() local
70 PetscCall(SNESGetDM(snes, &dm)); in SNESComputeFunction_DMDA()
71 PetscCall(DMGetLocalVector(dm, &Xloc)); in SNESComputeFunction_DMDA()
72 PetscCall(DMGlobalToLocalBegin(dm, X, INSERT_VALUES, Xloc)); in SNESComputeFunction_DMDA()
73 PetscCall(DMGlobalToLocalEnd(dm, X, INSERT_VALUES, Xloc)); in SNESComputeFunction_DMDA()
74 PetscCall(DMDAGetLocalInfo(dm, &info)); in SNESComputeFunction_DMDA()
81 PetscCall(DMDAVecGetArray(dm, Xloc, &x)); in SNESComputeFunction_DMDA()
82 PetscCall(DMDAVecGetArray(dm, F, &f)); in SNESComputeFunction_DMDA()
84 PetscCall(DMDAVecRestoreArray(dm, Xloc, &x)); in SNESComputeFunction_DMDA()
85 PetscCall(DMDAVecRestoreArray(dm, F, &f)); in SNESComputeFunction_DMDA()
91 PetscCall(DMGetLocalVector(dm, &Floc)); in SNESComputeFunction_DMDA()
96 PetscCall(DMDAVecGetArray(dm, Xloc, &x)); in SNESComputeFunction_DMDA()
97 PetscCall(DMDAVecGetArray(dm, Floc, &f)); in SNESComputeFunction_DMDA()
99 PetscCall(DMDAVecRestoreArray(dm, Xloc, &x)); in SNESComputeFunction_DMDA()
100 PetscCall(DMDAVecRestoreArray(dm, Floc, &f)); in SNESComputeFunction_DMDA()
104 PetscCall(DMLocalToGlobalBegin(dm, Floc, ADD_VALUES, F)); in SNESComputeFunction_DMDA()
105 PetscCall(DMLocalToGlobalEnd(dm, Floc, ADD_VALUES, F)); in SNESComputeFunction_DMDA()
106 PetscCall(DMRestoreLocalVector(dm, &Floc)); in SNESComputeFunction_DMDA()
111 PetscCall(DMRestoreLocalVector(dm, &Xloc)); in SNESComputeFunction_DMDA()
118 DM dm; in SNESComputeObjective_DMDA() local
129 PetscCall(SNESGetDM(snes, &dm)); in SNESComputeObjective_DMDA()
130 PetscCall(DMGetLocalVector(dm, &Xloc)); in SNESComputeObjective_DMDA()
131 PetscCall(DMGlobalToLocalBegin(dm, X, INSERT_VALUES, Xloc)); in SNESComputeObjective_DMDA()
132 PetscCall(DMGlobalToLocalEnd(dm, X, INSERT_VALUES, Xloc)); in SNESComputeObjective_DMDA()
133 PetscCall(DMDAGetLocalInfo(dm, &info)); in SNESComputeObjective_DMDA()
137 PetscCall(DMDAVecGetArray(dm, Xloc, &x)); in SNESComputeObjective_DMDA()
139 PetscCall(DMDAVecRestoreArray(dm, Xloc, &x)); in SNESComputeObjective_DMDA()
141 PetscCall(DMRestoreLocalVector(dm, &Xloc)); in SNESComputeObjective_DMDA()
149 DM dm; in SNESComputeJacobian_DMDA() local
157 PetscCall(SNESGetDM(snes, &dm)); in SNESComputeJacobian_DMDA()
160 PetscCall(DMGetLocalVector(dm, &Xloc)); in SNESComputeJacobian_DMDA()
161 PetscCall(DMGlobalToLocalBegin(dm, X, INSERT_VALUES, Xloc)); in SNESComputeJacobian_DMDA()
162 PetscCall(DMGlobalToLocalEnd(dm, X, INSERT_VALUES, Xloc)); in SNESComputeJacobian_DMDA()
163 PetscCall(DMDAGetLocalInfo(dm, &info)); in SNESComputeJacobian_DMDA()
166 PetscCall(DMDAVecGetArray(dm, Xloc, &x)); in SNESComputeJacobian_DMDA()
168 PetscCall(DMDAVecRestoreArray(dm, Xloc, &x)); in SNESComputeJacobian_DMDA()
170 PetscCall(DMRestoreLocalVector(dm, &Xloc)); in SNESComputeJacobian_DMDA()
173 PetscCall(PetscObjectQuery((PetscObject)dm, "DMDASNES_FDCOLORING", (PetscObject *)&fdcoloring)); in SNESComputeJacobian_DMDA()
177 PetscCall(DMCreateColoring(dm, dm->coloringtype, &coloring)); in SNESComputeJacobian_DMDA()
179 switch (dm->coloringtype) { in SNESComputeJacobian_DMDA()
184 …ject)snes), PETSC_ERR_SUP, "No support for coloring type '%s'", ISColoringTypes[dm->coloringtype]); in SNESComputeJacobian_DMDA()
186 PetscCall(PetscObjectSetOptionsPrefix((PetscObject)fdcoloring, ((PetscObject)dm)->prefix)); in SNESComputeJacobian_DMDA()
190 … PetscCall(PetscObjectCompose((PetscObject)dm, "DMDASNES_FDCOLORING", (PetscObject)fdcoloring)); in SNESComputeJacobian_DMDA()
199 PetscCall(PetscObjectDereference((PetscObject)dm)); in SNESComputeJacobian_DMDA()
232 PetscErrorCode DMDASNESSetFunctionLocal(DM dm, InsertMode imode, PetscErrorCode (*func)(DMDALocalIn… in DMDASNESSetFunctionLocal() argument
238 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMDASNESSetFunctionLocal()
239 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMDASNESSetFunctionLocal()
240 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetFunctionLocal()
246 PetscCall(DMSNESSetFunction(dm, SNESComputeFunction_DMDA, dmdasnes)); in DMDASNESSetFunctionLocal()
248 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMDA, dmdasnes)); in DMDASNESSetFunctionLocal()
274 PetscErrorCode DMDASNESSetFunctionLocalVec(DM dm, InsertMode imode, PetscErrorCode (*func)(DMDALoca… in DMDASNESSetFunctionLocalVec() argument
280 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMDASNESSetFunctionLocalVec()
281 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMDASNESSetFunctionLocalVec()
282 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetFunctionLocalVec()
288 PetscCall(DMSNESSetFunction(dm, SNESComputeFunction_DMDA, dmdasnes)); in DMDASNESSetFunctionLocalVec()
290 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMDA, dmdasnes)); in DMDASNESSetFunctionLocalVec()
319 PetscErrorCode DMDASNESSetJacobianLocal(DM dm, PetscErrorCode (*func)(DMDALocalInfo *info, void *x,… in DMDASNESSetJacobianLocal() argument
325 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMDASNESSetJacobianLocal()
326 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMDASNESSetJacobianLocal()
327 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetJacobianLocal()
332 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMDA, dmdasnes)); in DMDASNESSetJacobianLocal()
357 PetscErrorCode DMDASNESSetJacobianLocalVec(DM dm, PetscErrorCode (*func)(DMDALocalInfo *info, Vec x… in DMDASNESSetJacobianLocalVec() argument
363 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMDASNESSetJacobianLocalVec()
364 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMDASNESSetJacobianLocalVec()
365 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetJacobianLocalVec()
370 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMDA, dmdasnes)); in DMDASNESSetJacobianLocalVec()
394 PetscErrorCode DMDASNESSetObjectiveLocal(DM dm, PetscErrorCode (*func)(DMDALocalInfo *info, void *x… in DMDASNESSetObjectiveLocal() argument
400 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMDASNESSetObjectiveLocal()
401 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMDASNESSetObjectiveLocal()
402 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetObjectiveLocal()
407 PetscCall(DMSNESSetObjective(dm, SNESComputeObjective_DMDA, dmdasnes)); in DMDASNESSetObjectiveLocal()
431 PetscErrorCode DMDASNESSetObjectiveLocalVec(DM dm, PetscErrorCode (*func)(DMDALocalInfo *info, Vec … in DMDASNESSetObjectiveLocalVec() argument
437 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMDASNESSetObjectiveLocalVec()
438 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMDASNESSetObjectiveLocalVec()
439 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetObjectiveLocalVec()
444 PetscCall(DMSNESSetObjective(dm, SNESComputeObjective_DMDA, dmdasnes)); in DMDASNESSetObjectiveLocalVec()
450 DM dm; in SNESComputePicard_DMDA() local
461 PetscCall(SNESGetDM(snes, &dm)); in SNESComputePicard_DMDA()
462 PetscCall(DMGetLocalVector(dm, &Xloc)); in SNESComputePicard_DMDA()
463 PetscCall(DMGlobalToLocalBegin(dm, X, INSERT_VALUES, Xloc)); in SNESComputePicard_DMDA()
464 PetscCall(DMGlobalToLocalEnd(dm, X, INSERT_VALUES, Xloc)); in SNESComputePicard_DMDA()
465 PetscCall(DMDAGetLocalInfo(dm, &info)); in SNESComputePicard_DMDA()
466 PetscCall(DMDAVecGetArray(dm, Xloc, &x)); in SNESComputePicard_DMDA()
469 PetscCall(DMDAVecGetArray(dm, F, &f)); in SNESComputePicard_DMDA()
471 PetscCall(DMDAVecRestoreArray(dm, F, &f)); in SNESComputePicard_DMDA()
475 PetscCall(DMGetLocalVector(dm, &Floc)); in SNESComputePicard_DMDA()
477 PetscCall(DMDAVecGetArray(dm, Floc, &f)); in SNESComputePicard_DMDA()
479 PetscCall(DMDAVecRestoreArray(dm, Floc, &f)); in SNESComputePicard_DMDA()
481 PetscCall(DMLocalToGlobalBegin(dm, Floc, ADD_VALUES, F)); in SNESComputePicard_DMDA()
482 PetscCall(DMLocalToGlobalEnd(dm, Floc, ADD_VALUES, F)); in SNESComputePicard_DMDA()
483 PetscCall(DMRestoreLocalVector(dm, &Floc)); in SNESComputePicard_DMDA()
488 PetscCall(DMDAVecRestoreArray(dm, Xloc, &x)); in SNESComputePicard_DMDA()
489 PetscCall(DMRestoreLocalVector(dm, &Xloc)); in SNESComputePicard_DMDA()
495 DM dm; in SNESComputePicardJacobian_DMDA() local
503 PetscCall(SNESGetDM(snes, &dm)); in SNESComputePicardJacobian_DMDA()
505 PetscCall(DMGetLocalVector(dm, &Xloc)); in SNESComputePicardJacobian_DMDA()
506 PetscCall(DMGlobalToLocalBegin(dm, X, INSERT_VALUES, Xloc)); in SNESComputePicardJacobian_DMDA()
507 PetscCall(DMGlobalToLocalEnd(dm, X, INSERT_VALUES, Xloc)); in SNESComputePicardJacobian_DMDA()
508 PetscCall(DMDAGetLocalInfo(dm, &info)); in SNESComputePicardJacobian_DMDA()
509 PetscCall(DMDAVecGetArray(dm, Xloc, &x)); in SNESComputePicardJacobian_DMDA()
511 PetscCall(DMDAVecRestoreArray(dm, Xloc, &x)); in SNESComputePicardJacobian_DMDA()
512 PetscCall(DMRestoreLocalVector(dm, &Xloc)); in SNESComputePicardJacobian_DMDA()
553 PetscErrorCode DMDASNESSetPicardLocal(DM dm, InsertMode imode, PetscErrorCode (*func)(DMDALocalInfo… in DMDASNESSetPicardLocal() argument
559 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); in DMDASNESSetPicardLocal()
560 PetscCall(DMGetDMSNESWrite(dm, &sdm)); in DMDASNESSetPicardLocal()
561 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetPicardLocal()
568 PetscCall(DMSNESSetPicard(dm, SNESComputePicard_DMDA, SNESComputePicardJacobian_DMDA, dmdasnes)); in DMDASNESSetPicardLocal()
569 PetscCall(DMSNESSetMFFunction(dm, SNESComputeFunction_DMDA, dmdasnes)); in DMDASNESSetPicardLocal()