1 static const char help[] = "Test DMDAGetOwnershipRanges()\n";
2
3 #include <petscdm.h>
4 #include <petscdmda.h>
5
main(int argc,char * argv[])6 int main(int argc, char *argv[])
7 {
8 DM da;
9 PetscViewer vw;
10 PetscInt dim = 2, m, n, p;
11 const PetscInt *lx, *ly, *lz;
12 PetscMPIInt rank;
13
14 PetscFunctionBeginUser;
15 PetscCall(PetscInitialize(&argc, &argv, 0, help));
16 PetscCall(PetscOptionsGetInt(NULL, 0, "-dim", &dim, 0));
17 switch (dim) {
18 case 2:
19 PetscCall(DMDACreate2d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_STAR, 3, 5, PETSC_DECIDE, PETSC_DECIDE, 2, 1, NULL, NULL, &da));
20 break;
21 case 3:
22 PetscCall(DMDACreate3d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_STAR, 3, 5, 7, PETSC_DECIDE, PETSC_DECIDE, PETSC_DECIDE, 2, 1, NULL, NULL, NULL, &da));
23 break;
24 default:
25 SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_SUP, "No support for %" PetscInt_FMT " dimensions", dim);
26 }
27 PetscCall(DMSetFromOptions(da));
28 PetscCall(DMSetUp(da));
29 PetscCall(DMDAGetInfo(da, 0, 0, 0, 0, &m, &n, &p, 0, 0, 0, 0, 0, 0));
30 PetscCall(DMDAGetOwnershipRanges(da, &lx, &ly, &lz));
31 PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
32
33 PetscCall(PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &vw));
34 PetscCall(PetscViewerASCIIPrintf(vw, "[%d] lx ly%s\n", rank, dim > 2 ? " lz" : ""));
35 PetscCall(PetscIntView(m, lx, vw));
36 PetscCall(PetscIntView(n, ly, vw));
37 if (dim > 2) PetscCall(PetscIntView(n, lz, vw));
38 PetscCall(PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &vw));
39 PetscCall(DMDestroy(&da));
40 PetscCall(PetscFinalize());
41 return 0;
42 }
43
44 /*TEST
45
46 test:
47 nsize: 12
48 args: -dm_view -dim 3 -da_grid_x 11 -da_grid_y 5 -da_grid_z 7
49
50 TEST*/
51