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
ProcessOptions(MPI_Comm comm,AppCtx * options)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) {
44 options->patchSize.i = patchSize;
45 options->commSize.i = commSize;
46 options->gridSize.i = gridSize;
47 }
48 if (options->dim > 1) {
49 options->patchSize.j = patchSize;
50 options->commSize.j = commSize;
51 options->gridSize.j = gridSize;
52 }
53 if (options->dim > 2) {
54 options->patchSize.k = patchSize;
55 options->commSize.k = commSize;
56 options->gridSize.k = gridSize;
57 }
58 PetscFunctionReturn(PETSC_SUCCESS);
59 }
60
main(int argc,char ** argv)61 int main(int argc, char **argv)
62 {
63 DM dm;
64 AppCtx user; /* user-defined work context */
65
66 PetscFunctionBeginUser;
67 PetscCall(PetscInitialize(&argc, &argv, NULL, help));
68 PetscCall(ProcessOptions(PETSC_COMM_WORLD, &user));
69 PetscCall(DMPatchCreateGrid(PETSC_COMM_WORLD, user.dim, user.patchSize, user.commSize, user.gridSize, &dm));
70 PetscCall(PetscObjectSetName((PetscObject)dm, "Patch Mesh"));
71 PetscCall(DMSetFromOptions(dm));
72 PetscCall(DMSetUp(dm));
73 PetscCall(DMView(dm, PETSC_VIEWER_STDOUT_WORLD));
74 PetscCall(DMPatchSolve(dm));
75 PetscCall(DMDestroy(&dm));
76 PetscCall(PetscFinalize());
77 return 0;
78 }
79
80 /*TEST
81
82 test:
83
84 TEST*/
85