Actual source code: dareg.c
1: #include <petsc/private/dmdaimpl.h>
3: extern PetscErrorCode DMSetUp_DA_1D(DM);
4: extern PetscErrorCode DMSetUp_DA_2D(DM);
5: extern PetscErrorCode DMSetUp_DA_3D(DM);
7: PetscErrorCode DMSetUp_DA(DM da)
8: {
9: DM_DA *dd = (DM_DA *)da->data;
11: PetscFunctionBegin;
13: PetscCheck(dd->w >= 1, PetscObjectComm((PetscObject)da), PETSC_ERR_ARG_OUTOFRANGE, "Must have 1 or more degrees of freedom per node: %" PetscInt_FMT, dd->w);
14: PetscCheck(dd->s >= 0, PetscObjectComm((PetscObject)da), PETSC_ERR_ARG_OUTOFRANGE, "Stencil width cannot be negative: %" PetscInt_FMT, dd->s);
16: PetscCall(PetscCalloc1(dd->w + 1, &dd->fieldname));
17: PetscCall(PetscCalloc1(da->dim, &dd->coordinatename));
18: if (da->dim == 1) {
19: PetscCall(DMSetUp_DA_1D(da));
20: } else if (da->dim == 2) {
21: PetscCall(DMSetUp_DA_2D(da));
22: } else if (da->dim == 3) {
23: PetscCall(DMSetUp_DA_3D(da));
24: } else SETERRQ(PetscObjectComm((PetscObject)da), PETSC_ERR_SUP, "DMs only supported for 1, 2, and 3d");
25: PetscCall(DMViewFromOptions(da, NULL, "-dm_view"));
26: PetscFunctionReturn(PETSC_SUCCESS);
27: }