1 static char help[] = "Make a 2D grid of patches and view them\n\n"; 2 3 /* 4 Serial Test 5 Parallel Test where all zooms are serials 6 Parallel Test where zooms are parallel 7 8 Return DMPatch from Zoom 9 Override refine from DMPatch to split cells 10 */ 11 #include <petscdmpatch.h> 12 13 typedef struct { 14 PetscInt debug; /* The debugging level */ 15 PetscInt dim; /* The spatial dimension */ 16 MatStencil patchSize; /* Size of patches */ 17 MatStencil gridSize; /* Size of patch grid */ 18 MatStencil commSize; /* Size of patch comm */ 19 } AppCtx; 20 21 PetscErrorCode ProcessOptions(MPI_Comm comm, AppCtx *options) 22 { 23 PetscInt patchSize, commSize, gridSize; 24 PetscErrorCode ierr; 25 26 PetscFunctionBegin; 27 options->debug = 0; 28 options->dim = 2; 29 patchSize = 0; 30 commSize = 0; 31 gridSize = 0; 32 33 ierr = PetscOptionsBegin(comm, "", "Patch Test Options", "DMPATCH");CHKERRQ(ierr); 34 ierr = PetscOptionsBoundedInt("-debug", "The debugging level", "ex1.c", options->debug, &options->debug, NULL,0);CHKERRQ(ierr); 35 ierr = PetscOptionsRangeInt("-dim", "The spatial dimension", "ex1.c", options->dim, &options->dim, NULL,1,3);CHKERRQ(ierr); 36 ierr = PetscOptionsBoundedInt("-patch_size", "The patch size in each dimension", "ex1.c", patchSize, &patchSize, NULL,0);CHKERRQ(ierr); 37 ierr = PetscOptionsBoundedInt("-comm_size", "The comm size in each dimension", "ex1.c", commSize, &commSize, NULL,0);CHKERRQ(ierr); 38 ierr = PetscOptionsBoundedInt("-grid_size", "The grid size in each dimension", "ex1.c", gridSize, &gridSize, NULL,1);CHKERRQ(ierr); 39 ierr = PetscOptionsEnd(); 40 41 options->patchSize.i = options->patchSize.j = options->patchSize.k = 1; 42 options->commSize.i = options->commSize.j = options->commSize.k = 1; 43 options->gridSize.i = options->gridSize.j = options->gridSize.k = 1; 44 if (options->dim > 0) {options->patchSize.i = patchSize; options->commSize.i = commSize; options->gridSize.i = gridSize;} 45 if (options->dim > 1) {options->patchSize.j = patchSize; options->commSize.j = commSize; options->gridSize.j = gridSize;} 46 if (options->dim > 2) {options->patchSize.k = patchSize; options->commSize.k = commSize; options->gridSize.k = gridSize;} 47 PetscFunctionReturn(0); 48 }; 49 50 int main(int argc, char **argv) 51 { 52 DM dm; 53 AppCtx user; /* user-defined work context */ 54 PetscErrorCode ierr; 55 56 ierr = PetscInitialize(&argc, &argv, NULL, help);if (ierr) return ierr; 57 ierr = ProcessOptions(PETSC_COMM_WORLD, &user);CHKERRQ(ierr); 58 ierr = DMPatchCreateGrid(PETSC_COMM_WORLD, user.dim, user.patchSize, user.commSize, user.gridSize, &dm);CHKERRQ(ierr); 59 ierr = PetscObjectSetName((PetscObject) dm, "Patch Mesh");CHKERRQ(ierr); 60 ierr = DMSetFromOptions(dm);CHKERRQ(ierr); 61 ierr = DMSetUp(dm);CHKERRQ(ierr); 62 ierr = DMView(dm, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 63 ierr = DMPatchSolve(dm);CHKERRQ(ierr); 64 ierr = DMDestroy(&dm);CHKERRQ(ierr); 65 ierr = PetscFinalize(); 66 return ierr; 67 } 68