Lines Matching refs:dmdasnes
44 static PetscErrorCode DMDASNESGetContext(DM dm, DMSNES sdm, DMSNES_DA **dmdasnes) in DMDASNESGetContext() argument
47 *dmdasnes = NULL; in DMDASNESGetContext()
53 *dmdasnes = (DMSNES_DA *)sdm->data; in DMDASNESGetContext()
60 DMSNES_DA *dmdasnes = (DMSNES_DA *)ctx; in SNESComputeFunction_DMDA() local
69 …PetscCheck(dmdasnes->residuallocal || dmdasnes->residuallocalvec, PetscObjectComm((PetscObject)sne… in SNESComputeFunction_DMDA()
75 rctx = dmdasnes->residuallocalctx ? dmdasnes->residuallocalctx : snes->ctx; in SNESComputeFunction_DMDA()
76 switch (dmdasnes->residuallocalimode) { in SNESComputeFunction_DMDA()
79 …if (dmdasnes->residuallocalvec) PetscCallBack("SNES DMDA local callback function", (*dmdasnes->res… in SNESComputeFunction_DMDA()
83 … PetscCallBack("SNES DMDA local callback function", (*dmdasnes->residuallocal)(&info, x, f, rctx)); in SNESComputeFunction_DMDA()
94 …if (dmdasnes->residuallocalvec) PetscCallBack("SNES DMDA local callback function", (*dmdasnes->res… in SNESComputeFunction_DMDA()
98 … PetscCallBack("SNES DMDA local callback function", (*dmdasnes->residuallocal)(&info, x, f, rctx)); in SNESComputeFunction_DMDA()
109 …m((PetscObject)snes), PETSC_ERR_ARG_INCOMP, "Cannot use imode=%d", (int)dmdasnes->residuallocalimo… in SNESComputeFunction_DMDA()
119 DMSNES_DA *dmdasnes = (DMSNES_DA *)ctx; in SNESComputeObjective_DMDA() local
128 …PetscCheck(dmdasnes->objectivelocal || dmdasnes->objectivelocalvec, PetscObjectComm((PetscObject)s… in SNESComputeObjective_DMDA()
134 octx = dmdasnes->objectivelocalctx ? dmdasnes->objectivelocalctx : snes->ctx; in SNESComputeObjective_DMDA()
135 …if (dmdasnes->objectivelocalvec) PetscCallBack("SNES DMDA local callback objective", (*dmdasnes->o… in SNESComputeObjective_DMDA()
138 …PetscCallBack("SNES DMDA local callback objective", (*dmdasnes->objectivelocal)(&info, x, ob, octx… in SNESComputeObjective_DMDA()
150 DMSNES_DA *dmdasnes = (DMSNES_DA *)ctx; in SNESComputeJacobian_DMDA() local
156 …PetscCheck(dmdasnes->residuallocal || dmdasnes->residuallocalvec, PetscObjectComm((PetscObject)sne… in SNESComputeJacobian_DMDA()
158 jctx = dmdasnes->jacobianlocalctx ? dmdasnes->jacobianlocalctx : snes->ctx; in SNESComputeJacobian_DMDA()
159 if (dmdasnes->jacobianlocal || dmdasnes->jacobianlocalvec) { in SNESComputeJacobian_DMDA()
164 …if (dmdasnes->jacobianlocalvec) PetscCallBack("SNES DMDA local callback Jacobian", (*dmdasnes->jac… in SNESComputeJacobian_DMDA()
167 …PetscCallBack("SNES DMDA local callback Jacobian", (*dmdasnes->jacobianlocal)(&info, x, A, B, jctx… in SNESComputeJacobian_DMDA()
181 …cCall(MatFDColoringSetFunction(fdcoloring, (MatFDColoringFn *)SNESComputeFunction_DMDA, dmdasnes)); in SNESComputeJacobian_DMDA()
235 DMSNES_DA *dmdasnes; in DMDASNESSetFunctionLocal() local
240 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetFunctionLocal()
242 dmdasnes->residuallocalimode = imode; in DMDASNESSetFunctionLocal()
243 dmdasnes->residuallocal = func; in DMDASNESSetFunctionLocal()
244 dmdasnes->residuallocalctx = ctx; in DMDASNESSetFunctionLocal()
246 PetscCall(DMSNESSetFunction(dm, SNESComputeFunction_DMDA, dmdasnes)); in DMDASNESSetFunctionLocal()
248 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMDA, dmdasnes)); in DMDASNESSetFunctionLocal()
277 DMSNES_DA *dmdasnes; in DMDASNESSetFunctionLocalVec() local
282 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetFunctionLocalVec()
284 dmdasnes->residuallocalimode = imode; in DMDASNESSetFunctionLocalVec()
285 dmdasnes->residuallocalvec = func; in DMDASNESSetFunctionLocalVec()
286 dmdasnes->residuallocalctx = ctx; in DMDASNESSetFunctionLocalVec()
288 PetscCall(DMSNESSetFunction(dm, SNESComputeFunction_DMDA, dmdasnes)); in DMDASNESSetFunctionLocalVec()
290 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMDA, dmdasnes)); in DMDASNESSetFunctionLocalVec()
322 DMSNES_DA *dmdasnes; in DMDASNESSetJacobianLocal() local
327 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetJacobianLocal()
329 dmdasnes->jacobianlocal = func; in DMDASNESSetJacobianLocal()
330 dmdasnes->jacobianlocalctx = ctx; in DMDASNESSetJacobianLocal()
332 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMDA, dmdasnes)); in DMDASNESSetJacobianLocal()
360 DMSNES_DA *dmdasnes; in DMDASNESSetJacobianLocalVec() local
365 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetJacobianLocalVec()
367 dmdasnes->jacobianlocalvec = func; in DMDASNESSetJacobianLocalVec()
368 dmdasnes->jacobianlocalctx = ctx; in DMDASNESSetJacobianLocalVec()
370 PetscCall(DMSNESSetJacobian(dm, SNESComputeJacobian_DMDA, dmdasnes)); in DMDASNESSetJacobianLocalVec()
397 DMSNES_DA *dmdasnes; in DMDASNESSetObjectiveLocal() local
402 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetObjectiveLocal()
404 dmdasnes->objectivelocal = func; in DMDASNESSetObjectiveLocal()
405 dmdasnes->objectivelocalctx = ctx; in DMDASNESSetObjectiveLocal()
407 PetscCall(DMSNESSetObjective(dm, SNESComputeObjective_DMDA, dmdasnes)); in DMDASNESSetObjectiveLocal()
434 DMSNES_DA *dmdasnes; in DMDASNESSetObjectiveLocalVec() local
439 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetObjectiveLocalVec()
441 dmdasnes->objectivelocalvec = func; in DMDASNESSetObjectiveLocalVec()
442 dmdasnes->objectivelocalctx = ctx; in DMDASNESSetObjectiveLocalVec()
444 PetscCall(DMSNESSetObjective(dm, SNESComputeObjective_DMDA, dmdasnes)); in DMDASNESSetObjectiveLocalVec()
451 DMSNES_DA *dmdasnes = (DMSNES_DA *)ctx; in SNESComputePicard_DMDA() local
460 …PetscCheck(dmdasnes->rhsplocal, PetscObjectComm((PetscObject)snes), PETSC_ERR_PLIB, "Corrupt conte… in SNESComputePicard_DMDA()
467 switch (dmdasnes->residuallocalimode) { in SNESComputePicard_DMDA()
470 …CallBack("SNES Picard DMDA local callback function", (*dmdasnes->rhsplocal)(&info, x, f, dmdasnes-… in SNESComputePicard_DMDA()
478 …CallBack("SNES Picard DMDA local callback function", (*dmdasnes->rhsplocal)(&info, x, f, dmdasnes-… in SNESComputePicard_DMDA()
486 …m((PetscObject)snes), PETSC_ERR_ARG_INCOMP, "Cannot use imode=%d", (int)dmdasnes->residuallocalimo… in SNESComputePicard_DMDA()
496 DMSNES_DA *dmdasnes = (DMSNES_DA *)ctx; in SNESComputePicardJacobian_DMDA() local
502 …PetscCheck(dmdasnes->jacobianplocal, PetscObjectComm((PetscObject)snes), PETSC_ERR_PLIB, "Corrupt … in SNESComputePicardJacobian_DMDA()
510 …Back("SNES Picard DMDA local callback Jacobian", (*dmdasnes->jacobianplocal)(&info, x, A, B, dmdas… in SNESComputePicardJacobian_DMDA()
556 DMSNES_DA *dmdasnes; in DMDASNESSetPicardLocal() local
561 PetscCall(DMDASNESGetContext(dm, sdm, &dmdasnes)); in DMDASNESSetPicardLocal()
563 dmdasnes->residuallocalimode = imode; in DMDASNESSetPicardLocal()
564 dmdasnes->rhsplocal = func; in DMDASNESSetPicardLocal()
565 dmdasnes->jacobianplocal = jac; in DMDASNESSetPicardLocal()
566 dmdasnes->picardlocalctx = ctx; in DMDASNESSetPicardLocal()
568 PetscCall(DMSNESSetPicard(dm, SNESComputePicard_DMDA, SNESComputePicardJacobian_DMDA, dmdasnes)); in DMDASNESSetPicardLocal()
569 PetscCall(DMSNESSetMFFunction(dm, SNESComputeFunction_DMDA, dmdasnes)); in DMDASNESSetPicardLocal()