147c6ae99SBarry Smith 2b45d2f2cSJed Brown #include <petsc-private/daimpl.h> /*I "petscdmda.h" I*/ 347c6ae99SBarry Smith 47087cfbeSBarry Smith extern PetscErrorCode DMSetUp_DA_1D(DM); 57087cfbeSBarry Smith extern PetscErrorCode DMSetUp_DA_2D(DM); 67087cfbeSBarry Smith extern PetscErrorCode DMSetUp_DA_3D(DM); 747c6ae99SBarry Smith 847c6ae99SBarry Smith #undef __FUNCT__ 99a42bb27SBarry Smith #define __FUNCT__ "DMSetUp_DA" 107087cfbeSBarry Smith PetscErrorCode DMSetUp_DA(DM da) 1147c6ae99SBarry Smith { 1247c6ae99SBarry Smith PetscErrorCode ierr; 13d7bf68aeSBarry Smith DM_DA *dd = (DM_DA*)da->data; 1447c6ae99SBarry Smith 1547c6ae99SBarry Smith PetscFunctionBegin; 1647c6ae99SBarry Smith PetscValidHeaderSpecific(da, DM_CLASSID,1); 17*ce94432eSBarry Smith 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*ce94432eSBarry Smith if (dd->s < 0) SETERRQ1(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Stencil width cannot be negative: %D",dd->s); 19109c9344SBarry Smith 20109c9344SBarry Smith ierr = PetscMalloc(dd->w*sizeof(char*),&dd->fieldname);CHKERRQ(ierr); 21109c9344SBarry Smith ierr = PetscMemzero(dd->fieldname,dd->w*sizeof(char*));CHKERRQ(ierr); 22109c9344SBarry Smith ierr = PetscMalloc(dd->dim*sizeof(char*),&dd->coordinatename);CHKERRQ(ierr); 23109c9344SBarry Smith ierr = PetscMemzero(dd->coordinatename,dd->dim*sizeof(char*));CHKERRQ(ierr); 24d7bf68aeSBarry Smith if (dd->dim == 1) { 259a42bb27SBarry Smith ierr = DMSetUp_DA_1D(da);CHKERRQ(ierr); 26d7bf68aeSBarry Smith } else if (dd->dim == 2) { 279a42bb27SBarry Smith ierr = DMSetUp_DA_2D(da);CHKERRQ(ierr); 28d7bf68aeSBarry Smith } else if (dd->dim == 3) { 299a42bb27SBarry Smith ierr = DMSetUp_DA_3D(da);CHKERRQ(ierr); 30*ce94432eSBarry Smith } else SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_SUP,"DMs only supported for 1, 2, and 3d"); 3147c6ae99SBarry Smith PetscFunctionReturn(0); 3247c6ae99SBarry Smith } 3347c6ae99SBarry Smith 34