1 2 #include <petsc-private/daimpl.h> /*I "petscdmda.h" I*/ 3 4 extern PetscErrorCode DMSetUp_DA_1D(DM); 5 extern PetscErrorCode DMSetUp_DA_2D(DM); 6 extern PetscErrorCode DMSetUp_DA_3D(DM); 7 8 #undef __FUNCT__ 9 #define __FUNCT__ "DMSetUp_DA" 10 PetscErrorCode DMSetUp_DA(DM da) 11 { 12 PetscErrorCode ierr; 13 DM_DA *dd = (DM_DA*)da->data; 14 15 PetscFunctionBegin; 16 PetscValidHeaderSpecific(da, DM_CLASSID,1); 17 if (dd->w < 1) SETERRQ1(((PetscObject)da)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Must have 1 or more degrees of freedom per node: %D",dd->w); 18 if (dd->s < 0) SETERRQ1(((PetscObject)da)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Stencil width cannot be negative: %D",dd->s); 19 20 ierr = PetscMalloc(dd->w*sizeof(char*),&dd->fieldname);CHKERRQ(ierr); 21 ierr = PetscMemzero(dd->fieldname,dd->w*sizeof(char*));CHKERRQ(ierr); 22 ierr = PetscMalloc(dd->dim*sizeof(char*),&dd->coordinatename);CHKERRQ(ierr); 23 ierr = PetscMemzero(dd->coordinatename,dd->dim*sizeof(char*));CHKERRQ(ierr); 24 if (dd->dim == 1) { 25 ierr = DMSetUp_DA_1D(da);CHKERRQ(ierr); 26 } else if (dd->dim == 2) { 27 ierr = DMSetUp_DA_2D(da);CHKERRQ(ierr); 28 } else if (dd->dim == 3) { 29 ierr = DMSetUp_DA_3D(da);CHKERRQ(ierr); 30 } else SETERRQ(((PetscObject)da)->comm,PETSC_ERR_SUP,"DMs only supported for 1, 2, and 3d"); 31 PetscFunctionReturn(0); 32 } 33 34