xref: /petsc/src/dm/impls/da/dareg.c (revision d71ae5a4db6382e7f06317b8d368875286fe9008)
1af0996ceSBarry Smith #include <petsc/private/dmdaimpl.h> /*I "petscdmda.h"  I*/
247c6ae99SBarry Smith 
37087cfbeSBarry Smith extern PetscErrorCode DMSetUp_DA_1D(DM);
47087cfbeSBarry Smith extern PetscErrorCode DMSetUp_DA_2D(DM);
57087cfbeSBarry Smith extern PetscErrorCode DMSetUp_DA_3D(DM);
647c6ae99SBarry Smith 
7*d71ae5a4SJacob Faibussowitsch PetscErrorCode DMSetUp_DA(DM da)
8*d71ae5a4SJacob Faibussowitsch {
9d7bf68aeSBarry Smith   DM_DA *dd = (DM_DA *)da->data;
1047c6ae99SBarry Smith 
1147c6ae99SBarry Smith   PetscFunctionBegin;
1247c6ae99SBarry Smith   PetscValidHeaderSpecific(da, DM_CLASSID, 1);
1363a3b9bcSJacob Faibussowitsch   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);
1463a3b9bcSJacob Faibussowitsch   PetscCheck(dd->s >= 0, PetscObjectComm((PetscObject)da), PETSC_ERR_ARG_OUTOFRANGE, "Stencil width cannot be negative: %" PetscInt_FMT, dd->s);
15109c9344SBarry Smith 
169566063dSJacob Faibussowitsch   PetscCall(PetscCalloc1(dd->w + 1, &dd->fieldname));
179566063dSJacob Faibussowitsch   PetscCall(PetscCalloc1(da->dim, &dd->coordinatename));
18c73cfb54SMatthew G. Knepley   if (da->dim == 1) {
199566063dSJacob Faibussowitsch     PetscCall(DMSetUp_DA_1D(da));
20c73cfb54SMatthew G. Knepley   } else if (da->dim == 2) {
219566063dSJacob Faibussowitsch     PetscCall(DMSetUp_DA_2D(da));
22c73cfb54SMatthew G. Knepley   } else if (da->dim == 3) {
239566063dSJacob Faibussowitsch     PetscCall(DMSetUp_DA_3D(da));
24ce94432eSBarry Smith   } else SETERRQ(PetscObjectComm((PetscObject)da), PETSC_ERR_SUP, "DMs only supported for 1, 2, and 3d");
259566063dSJacob Faibussowitsch   PetscCall(DMViewFromOptions(da, NULL, "-dm_view"));
2647c6ae99SBarry Smith   PetscFunctionReturn(0);
2747c6ae99SBarry Smith }
28