1 #include <petsc/private/dmpatchimpl.h> /*I "petscdmpatch.h" I*/ 2 #include <petscdmda.h> 3 4 PetscErrorCode DMSetFromOptions_Patch(PetscOptionItems *PetscOptionsObject,DM dm) 5 { 6 /* DM_Patch *mesh = (DM_Patch*) dm->data; */ 7 PetscErrorCode ierr; 8 9 PetscFunctionBegin; 10 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 11 ierr = PetscOptionsHead(PetscOptionsObject,"DMPatch Options");CHKERRQ(ierr); 12 /* Handle associated vectors */ 13 /* Handle viewing */ 14 ierr = PetscOptionsTail();CHKERRQ(ierr); 15 PetscFunctionReturn(0); 16 } 17 18 /* External function declarations here */ 19 extern PetscErrorCode DMSetUp_Patch(DM dm); 20 extern PetscErrorCode DMView_Patch(DM dm, PetscViewer viewer); 21 extern PetscErrorCode DMCreateGlobalVector_Patch(DM dm, Vec *g); 22 extern PetscErrorCode DMCreateLocalVector_Patch(DM dm, Vec *l); 23 extern PetscErrorCode DMDestroy_Patch(DM dm); 24 extern PetscErrorCode DMCreateSubDM_Patch(DM dm, PetscInt numFields, const PetscInt fields[], IS *is, DM *subdm); 25 26 PetscErrorCode DMInitialize_Patch(DM dm) 27 { 28 PetscFunctionBegin; 29 dm->ops->view = DMView_Patch; 30 dm->ops->setfromoptions = DMSetFromOptions_Patch; 31 dm->ops->setup = DMSetUp_Patch; 32 dm->ops->createglobalvector = DMCreateGlobalVector_Patch; 33 dm->ops->createlocalvector = DMCreateLocalVector_Patch; 34 dm->ops->getlocaltoglobalmapping = NULL; 35 dm->ops->createfieldis = NULL; 36 dm->ops->getcoloring = 0; 37 dm->ops->creatematrix = 0; 38 dm->ops->createinterpolation = 0; 39 dm->ops->createinjection = 0; 40 dm->ops->refine = 0; 41 dm->ops->coarsen = 0; 42 dm->ops->refinehierarchy = 0; 43 dm->ops->coarsenhierarchy = 0; 44 dm->ops->globaltolocalbegin = NULL; 45 dm->ops->globaltolocalend = NULL; 46 dm->ops->localtoglobalbegin = NULL; 47 dm->ops->localtoglobalend = NULL; 48 dm->ops->destroy = DMDestroy_Patch; 49 dm->ops->createsubdm = DMCreateSubDM_Patch; 50 PetscFunctionReturn(0); 51 } 52 53 PETSC_EXTERN PetscErrorCode DMCreate_Patch(DM dm) 54 { 55 DM_Patch *mesh; 56 PetscErrorCode ierr; 57 58 PetscFunctionBegin; 59 PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 60 ierr = PetscNewLog(dm,&mesh);CHKERRQ(ierr); 61 dm->data = mesh; 62 63 mesh->refct = 1; 64 mesh->dmCoarse = NULL; 65 mesh->patchSize.i = 0; 66 mesh->patchSize.j = 0; 67 mesh->patchSize.k = 0; 68 mesh->patchSize.c = 0; 69 70 ierr = DMInitialize_Patch(dm);CHKERRQ(ierr); 71 PetscFunctionReturn(0); 72 } 73 74 /*@ 75 DMPatchCreate - Creates a DMPatch object, which is a collections of DMs called patches. 76 77 Collective 78 79 Input Parameter: 80 . comm - The communicator for the DMPatch object 81 82 Output Parameter: 83 . mesh - The DMPatch object 84 85 Level: beginner 86 87 @*/ 88 PetscErrorCode DMPatchCreate(MPI_Comm comm, DM *mesh) 89 { 90 PetscErrorCode ierr; 91 92 PetscFunctionBegin; 93 PetscValidPointer(mesh,2); 94 ierr = DMCreate(comm, mesh);CHKERRQ(ierr); 95 ierr = DMSetType(*mesh, DMPATCH);CHKERRQ(ierr); 96 PetscFunctionReturn(0); 97 } 98 99 PetscErrorCode DMPatchCreateGrid(MPI_Comm comm, PetscInt dim, MatStencil patchSize, MatStencil commSize, MatStencil gridSize, DM *dm) 100 { 101 DM_Patch *mesh; 102 DM da; 103 PetscInt dof = 1, width = 1; 104 PetscErrorCode ierr; 105 106 PetscFunctionBegin; 107 ierr = DMPatchCreate(comm, dm);CHKERRQ(ierr); 108 mesh = (DM_Patch*) (*dm)->data; 109 if (dim < 2) { 110 gridSize.j = 1; 111 patchSize.j = 1; 112 } 113 if (dim < 3) { 114 gridSize.k = 1; 115 patchSize.k = 1; 116 } 117 ierr = DMCreate(comm, &da);CHKERRQ(ierr); 118 ierr = DMSetType(da, DMDA);CHKERRQ(ierr); 119 ierr = DMSetDimension(da, dim);CHKERRQ(ierr); 120 ierr = DMDASetSizes(da, gridSize.i, gridSize.j, gridSize.k);CHKERRQ(ierr); 121 ierr = DMDASetBoundaryType(da, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE);CHKERRQ(ierr); 122 ierr = DMDASetDof(da, dof);CHKERRQ(ierr); 123 ierr = DMDASetStencilType(da, DMDA_STENCIL_BOX);CHKERRQ(ierr); 124 ierr = DMDASetStencilWidth(da, width);CHKERRQ(ierr); 125 126 mesh->dmCoarse = da; 127 128 ierr = DMPatchSetPatchSize(*dm, patchSize);CHKERRQ(ierr); 129 ierr = DMPatchSetCommSize(*dm, commSize);CHKERRQ(ierr); 130 PetscFunctionReturn(0); 131 } 132