1 2 #include "private/daimpl.h" /*I "petscdm.h" I*/ 3 4 #undef __FUNCT__ 5 #define __FUNCT__ "DMSetFromOptions_DA" 6 PetscErrorCode DMSetFromOptions_DA(DM da) 7 { 8 PetscErrorCode ierr; 9 PetscBool flg; 10 char typeName[256]; 11 DM_DA *dd = (DM_DA*)da->data; 12 13 PetscFunctionBegin; 14 PetscValidHeaderSpecific(da,DM_CLASSID,1); 15 16 ierr = PetscOptionsBegin(((PetscObject)da)->comm,((PetscObject)da)->prefix,"DMDA Options","DMDA");CHKERRQ(ierr); 17 /* Handle DMDA grid sizes */ 18 if (dd->M < 0) { 19 PetscInt newM = -dd->M; 20 ierr = PetscOptionsInt("-da_grid_x","Number of grid points in x direction","DMDASetSizes",newM,&newM,PETSC_NULL);CHKERRQ(ierr); 21 dd->M = newM; 22 } 23 if (dd->dim > 1 && dd->N < 0) { 24 PetscInt newN = -dd->N; 25 ierr = PetscOptionsInt("-da_grid_y","Number of grid points in y direction","DMDASetSizes",newN,&newN,PETSC_NULL);CHKERRQ(ierr); 26 dd->N = newN; 27 } 28 if (dd->dim > 2 && dd->P < 0) { 29 PetscInt newP = -dd->P; 30 ierr = PetscOptionsInt("-da_grid_z","Number of grid points in z direction","DMDASetSizes",newP,&newP,PETSC_NULL);CHKERRQ(ierr); 31 dd->P = newP; 32 } 33 /* Handle DMDA parallel distibution */ 34 ierr = PetscOptionsInt("-da_processors_x","Number of processors in x direction","DMDASetNumProcs",dd->m,&dd->m,PETSC_NULL);CHKERRQ(ierr); 35 if (dd->dim > 1) {ierr = PetscOptionsInt("-da_processors_y","Number of processors in y direction","DMDASetNumProcs",dd->n,&dd->n,PETSC_NULL);CHKERRQ(ierr);} 36 if (dd->dim > 2) {ierr = PetscOptionsInt("-da_processors_z","Number of processors in z direction","DMDASetNumProcs",dd->p,&dd->p,PETSC_NULL);CHKERRQ(ierr);} 37 /* Handle DMDA refinement */ 38 ierr = PetscOptionsInt("-da_refine_x","Refinement ratio in x direction","DMDASetRefinementFactor",dd->refine_x,&dd->refine_x,PETSC_NULL);CHKERRQ(ierr); 39 if (dd->dim > 1) {ierr = PetscOptionsInt("-da_refine_y","Refinement ratio in y direction","DMDASetRefinementFactor",dd->refine_y,&dd->refine_y,PETSC_NULL);CHKERRQ(ierr);} 40 if (dd->dim > 2) {ierr = PetscOptionsInt("-da_refine_z","Refinement ratio in z direction","DMDASetRefinementFactor",dd->refine_z,&dd->refine_z,PETSC_NULL);CHKERRQ(ierr);} 41 42 if (!VecRegisterAllCalled) {ierr = VecRegisterAll(PETSC_NULL);CHKERRQ(ierr);} 43 ierr = PetscOptionsList("-da_vec_type","Vector type used for created vectors","DMSetVecType",VecList,da->vectype,typeName,256,&flg);CHKERRQ(ierr); 44 if (flg) { 45 ierr = DMSetVecType(da,typeName);CHKERRQ(ierr); 46 } 47 48 /* process any options handlers added with PetscObjectAddOptionsHandler() */ 49 ierr = PetscObjectProcessOptionsHandlers((PetscObject)da);CHKERRQ(ierr); 50 ierr = PetscOptionsEnd();CHKERRQ(ierr); 51 52 PetscFunctionReturn(0); 53 } 54 55 extern PetscErrorCode DMCreateGlobalVector_DA(DM,Vec*); 56 extern PetscErrorCode DMCreateLocalVector_DA(DM,Vec*); 57 extern PetscErrorCode DMGlobalToLocalBegin_DA(DM,Vec,InsertMode,Vec); 58 extern PetscErrorCode DMGlobalToLocalEnd_DA(DM,Vec,InsertMode,Vec); 59 extern PetscErrorCode DMLocalToGlobalBegin_DA(DM,Vec,InsertMode,Vec); 60 extern PetscErrorCode DMLocalToGlobalEnd_DA(DM,Vec,InsertMode,Vec); 61 extern PetscErrorCode DMGetInterpolation_DA(DM,DM,Mat*,Vec*); 62 extern PetscErrorCode DMGetColoring_DA(DM,ISColoringType,const MatType,ISColoring*); 63 extern PetscErrorCode DMGetElements_DA(DM,PetscInt*,PetscInt*,const PetscInt *[]); 64 extern PetscErrorCode DMGetMatrix_DA(DM,const MatType,Mat*); 65 extern PetscErrorCode DMRefine_DA(DM,MPI_Comm,DM*); 66 extern PetscErrorCode DMCoarsen_DA(DM,MPI_Comm,DM*); 67 extern PetscErrorCode DMRefineHierarchy_DA(DM,PetscInt,DM[]); 68 extern PetscErrorCode DMCoarsenHierarchy_DA(DM,PetscInt,DM[]); 69 extern PetscErrorCode DMGetInjection_DA(DM,DM,VecScatter*); 70 extern PetscErrorCode DMGetAggregates_DA(DM,DM,Mat*); 71 extern PetscErrorCode DMView_DA(DM,PetscViewer); 72 extern PetscErrorCode DMSetUp_DA(DM); 73 extern PetscErrorCode DMDestroy_DA(DM); 74 75 EXTERN_C_BEGIN 76 #undef __FUNCT__ 77 #define __FUNCT__ "DMCreate_DA" 78 PetscErrorCode DMCreate_DA(DM da) 79 { 80 PetscErrorCode ierr; 81 DM_DA *dd; 82 83 PetscFunctionBegin; 84 PetscValidPointer(da,1); 85 ierr = PetscNewLog(da,DM_DA,&dd);CHKERRQ(ierr); 86 da->data = dd; 87 88 dd->dim = -1; 89 dd->interptype = DMDA_Q1; 90 dd->refine_x = 2; 91 dd->refine_y = 2; 92 dd->refine_z = 2; 93 dd->fieldname = PETSC_NULL; 94 dd->nlocal = -1; 95 dd->Nlocal = -1; 96 dd->M = -1; 97 dd->N = -1; 98 dd->P = -1; 99 dd->m = -1; 100 dd->n = -1; 101 dd->p = -1; 102 dd->w = -1; 103 dd->s = -1; 104 dd->xs = -1; dd->xe = -1; dd->ys = -1; dd->ye = -1; dd->zs = -1; dd->ze = -1; 105 dd->Xs = -1; dd->Xe = -1; dd->Ys = -1; dd->Ye = -1; dd->Zs = -1; dd->Ze = -1; 106 107 dd->gtol = PETSC_NULL; 108 dd->ltog = PETSC_NULL; 109 dd->ltol = PETSC_NULL; 110 dd->ao = PETSC_NULL; 111 dd->base = -1; 112 dd->bx = DMDA_BOUNDARY_NONE; 113 dd->by = DMDA_BOUNDARY_NONE; 114 dd->bz = DMDA_BOUNDARY_NONE; 115 dd->stencil_type = DMDA_STENCIL_BOX; 116 dd->interptype = DMDA_Q1; 117 dd->idx = PETSC_NULL; 118 dd->Nl = -1; 119 dd->lx = PETSC_NULL; 120 dd->ly = PETSC_NULL; 121 dd->lz = PETSC_NULL; 122 123 dd->elementtype = DMDA_ELEMENT_Q1; 124 125 ierr = PetscStrallocpy(VECSTANDARD,&da->vectype);CHKERRQ(ierr); 126 da->ops->globaltolocalbegin = DMGlobalToLocalBegin_DA; 127 da->ops->globaltolocalend = DMGlobalToLocalEnd_DA; 128 da->ops->localtoglobalbegin = DMLocalToGlobalBegin_DA; 129 da->ops->localtoglobalend = DMLocalToGlobalEnd_DA; 130 da->ops->createglobalvector = DMCreateGlobalVector_DA; 131 da->ops->createlocalvector = DMCreateLocalVector_DA; 132 da->ops->getinterpolation = DMGetInterpolation_DA; 133 da->ops->getcoloring = DMGetColoring_DA; 134 da->ops->getelements = DMGetElements_DA; 135 da->ops->getmatrix = DMGetMatrix_DA; 136 da->ops->refine = DMRefine_DA; 137 da->ops->coarsen = DMCoarsen_DA; 138 da->ops->refinehierarchy = DMRefineHierarchy_DA; 139 da->ops->coarsenhierarchy = DMCoarsenHierarchy_DA; 140 da->ops->getinjection = DMGetInjection_DA; 141 da->ops->getaggregates = DMGetAggregates_DA; 142 da->ops->destroy = DMDestroy_DA; 143 da->ops->view = 0; 144 da->ops->setfromoptions = DMSetFromOptions_DA; 145 da->ops->setup = DMSetUp_DA; 146 PetscFunctionReturn(0); 147 } 148 EXTERN_C_END 149 150 #undef __FUNCT__ 151 #define __FUNCT__ "DMDACreate" 152 /*@ 153 DMDACreate - Creates a DMDA object. 154 155 Collective on MPI_Comm 156 157 Input Parameter: 158 . comm - The communicator for the DMDA object 159 160 Output Parameter: 161 . da - The DMDA object 162 163 Level: beginner 164 165 .keywords: DMDA, create 166 .seealso: DMDASetSizes(), DMDADuplicate() 167 @*/ 168 PetscErrorCode DMDACreate(MPI_Comm comm, DM *da) 169 { 170 PetscErrorCode ierr; 171 172 PetscFunctionBegin; 173 PetscValidPointer(da,2); 174 ierr = DMCreate(comm,da);CHKERRQ(ierr); 175 ierr = DMSetType(*da,DMDA);CHKERRQ(ierr); 176 PetscFunctionReturn(0); 177 } 178