Actual source code: dareg.c

petsc-3.4.5 2014-06-29
  2: #include <petsc-private/dmdaimpl.h>    /*I "petscdmda.h"  I*/

  4: extern PetscErrorCode  DMSetUp_DA_1D(DM);
  5: extern PetscErrorCode  DMSetUp_DA_2D(DM);
  6: extern PetscErrorCode  DMSetUp_DA_3D(DM);

 10: PetscErrorCode  DMSetUp_DA(DM da)
 11: {
 13:   DM_DA          *dd = (DM_DA*)da->data;

 17:   if (dd->w < 1) SETERRQ1(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Must have 1 or more degrees of freedom per node: %D",dd->w);
 18:   if (dd->s < 0) SETERRQ1(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Stencil width cannot be negative: %D",dd->s);

 20:   PetscMalloc(dd->w*sizeof(char*),&dd->fieldname);
 21:   PetscMemzero(dd->fieldname,dd->w*sizeof(char*));
 22:   PetscMalloc(dd->dim*sizeof(char*),&dd->coordinatename);
 23:   PetscMemzero(dd->coordinatename,dd->dim*sizeof(char*));
 24:   if (dd->dim == 1) {
 25:     DMSetUp_DA_1D(da);
 26:   } else if (dd->dim == 2) {
 27:     DMSetUp_DA_2D(da);
 28:   } else if (dd->dim == 3) {
 29:     DMSetUp_DA_3D(da);
 30:   } else SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_SUP,"DMs only supported for 1, 2, and 3d");
 31:   return(0);
 32: }