Actual source code: ex16.c
1: static char help[] = "Test DMStag refinement and coarsening\n\n";
3: #include <petscdm.h>
4: #include <petscdmstag.h>
6: int main(int argc, char **argv)
7: {
8: DM dm, dmCoarsened, dmRefined;
9: PetscInt dim;
10: PetscBool flg;
12: /* Create a DMStag object */
14: PetscInitialize(&argc, &argv, (char *)0, help);
15: PetscOptionsGetInt(NULL, NULL, "-dim", &dim, &flg);
16: if (!flg) {
17: PetscPrintf(PETSC_COMM_WORLD, "Supply -dim option\n");
18: return 1;
19: }
20: if (dim == 1) {
21: DMStagCreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, 8, 2, 3, DMSTAG_STENCIL_BOX, 1, NULL, &dm);
22: } else if (dim == 2) {
23: DMStagCreate2d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, 4, 6, PETSC_DECIDE, PETSC_DECIDE, 2, 1, 1, DMSTAG_STENCIL_BOX, 1, NULL, NULL, &dm);
24: } else if (dim == 3) {
25: DMStagCreate3d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, 4, 4, 6, PETSC_DECIDE, PETSC_DECIDE, PETSC_DECIDE, 1, 1, 1, 1, DMSTAG_STENCIL_BOX, 1, NULL, NULL, NULL, &dm);
26: } else {
27: PetscPrintf(PETSC_COMM_WORLD, "Supply -dim option with value 1, 2, or 3\n");
28: return 1;
29: }
30: DMSetFromOptions(dm);
31: DMSetUp(dm);
32: DMView(dm, PETSC_VIEWER_STDOUT_WORLD);
34: /* Create a refined DMStag object */
35: DMRefine(dm, PetscObjectComm((PetscObject)dm), &dmRefined);
36: DMView(dmRefined, PETSC_VIEWER_STDOUT_WORLD);
38: /* Create a coarsened DMStag object */
39: DMCoarsen(dm, PetscObjectComm((PetscObject)dm), &dmCoarsened);
40: DMView(dmCoarsened, PETSC_VIEWER_STDOUT_WORLD);
42: DMDestroy(&dmCoarsened);
43: DMDestroy(&dmRefined);
44: DMDestroy(&dm);
45: PetscFinalize();
46: return 0;
47: }
49: /*TEST
51: test:
52: suffix: 1
53: nsize: 1
54: args: -dim 1 -stag_grid_x 2
56: test:
57: suffix: 2
58: nsize: 1
59: args: -dim 2 -stag_grid_x 6 -stag_grid_y 4
61: test:
62: suffix: 3
63: nsize: 6
64: args: -dim 3 -stag_grid_x 6 -stag_grid_y 4 -stag_grid_z 4
66: test:
67: suffix: 4
68: nsize: 2
69: args: -dim 1 -stag_grid_x 8
71: test:
72: suffix: 5
73: nsize: 4
74: args: -dim 2 -stag_grid_x 4 -stag_grid_y 8
76: test:
77: suffix: 6
78: nsize: 12
79: args: -dim 3 -stag_grid_x 4 -stag_grid_y 4 -stag_grid_z 12
81: TEST*/