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