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 25 PetscFunctionBegin; 26 options->debug = 0; 27 options->dim = 2; 28 patchSize = 0; 29 commSize = 0; 30 gridSize = 1; 31 32 PetscOptionsBegin(comm, "", "Patch Test Options", "DMPATCH"); 33 PetscCall(PetscOptionsBoundedInt("-debug", "The debugging level", "ex1.c", options->debug, &options->debug, NULL,0)); 34 PetscCall(PetscOptionsRangeInt("-dim", "The spatial dimension", "ex1.c", options->dim, &options->dim, NULL,1,3)); 35 PetscCall(PetscOptionsBoundedInt("-patch_size", "The patch size in each dimension", "ex1.c", patchSize, &patchSize, NULL,0)); 36 PetscCall(PetscOptionsBoundedInt("-comm_size", "The comm size in each dimension", "ex1.c", commSize, &commSize, NULL,0)); 37 PetscCall(PetscOptionsBoundedInt("-grid_size", "The grid size in each dimension", "ex1.c", gridSize, &gridSize, NULL,1)); 38 PetscOptionsEnd(); 39 40 options->patchSize.i = options->patchSize.j = options->patchSize.k = 1; 41 options->commSize.i = options->commSize.j = options->commSize.k = 1; 42 options->gridSize.i = options->gridSize.j = options->gridSize.k = 1; 43 if (options->dim > 0) {options->patchSize.i = patchSize; options->commSize.i = commSize; options->gridSize.i = gridSize;} 44 if (options->dim > 1) {options->patchSize.j = patchSize; options->commSize.j = commSize; options->gridSize.j = gridSize;} 45 if (options->dim > 2) {options->patchSize.k = patchSize; options->commSize.k = commSize; options->gridSize.k = gridSize;} 46 PetscFunctionReturn(0); 47 } 48 49 int main(int argc, char **argv) 50 { 51 DM dm; 52 AppCtx user; /* user-defined work context */ 53 54 PetscFunctionBeginUser; 55 PetscCall(PetscInitialize(&argc, &argv, NULL, help)); 56 PetscCall(ProcessOptions(PETSC_COMM_WORLD, &user)); 57 PetscCall(DMPatchCreateGrid(PETSC_COMM_WORLD, user.dim, user.patchSize, user.commSize, user.gridSize, &dm)); 58 PetscCall(PetscObjectSetName((PetscObject) dm, "Patch Mesh")); 59 PetscCall(DMSetFromOptions(dm)); 60 PetscCall(DMSetUp(dm)); 61 PetscCall(DMView(dm, PETSC_VIEWER_STDOUT_WORLD)); 62 PetscCall(DMPatchSolve(dm)); 63 PetscCall(DMDestroy(&dm)); 64 PetscCall(PetscFinalize()); 65 return 0; 66 } 67 68 /*TEST 69 70 test: 71 72 TEST*/ 73