Actual source code: ex34.c
petsc-3.8.0 2017-09-26
1: static const char help[] = "Test DMDAGetOwnershipRanges()\n";
3: #include <petscdm.h>
4: #include <petscdmda.h>
6: int main(int argc,char *argv[])
7: {
9: DM da;
10: PetscInt dim = 2,m,n,p,i;
11: const PetscInt *lx,*ly,*lz;
12: PetscMPIInt rank,size;
14: PetscInitialize(&argc,&argv,0,help);if (ierr) return ierr;
15: PetscOptionsGetInt(NULL,0,"-dim",&dim,0);
16: switch (dim) {
17: case 2:
18: DMDACreate2d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE, DM_BOUNDARY_NONE,DMDA_STENCIL_STAR, 3,5,PETSC_DECIDE,PETSC_DECIDE,2,1,NULL,NULL,&da);
19: break;
20: case 3:
21: 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);
22: break;
23: default: SETERRQ1(PETSC_COMM_WORLD,PETSC_ERR_SUP,"No support for %D dimensions",dim);
24: }
25: DMSetFromOptions(da);
26: DMSetUp(da);
27: DMDAGetInfo(da, 0, 0,0,0, &m,&n,&p, 0,0, 0,0,0,0);
28: DMDAGetOwnershipRanges(da,&lx,&ly,&lz);
29: MPI_Comm_size(PETSC_COMM_WORLD,&size);
30: MPI_Comm_rank(PETSC_COMM_WORLD,&rank);
31: for (i=0; i<size; i++) {
32: PetscViewerFlush(PETSC_VIEWER_STDOUT_SELF);
33: if (i == rank) {
34: PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_SELF,"[%d] lx ly%s\n",rank,dim>2 ? " lz" : "");
35: PetscIntView(m,lx,PETSC_VIEWER_STDOUT_SELF);
36: PetscIntView(n,ly,PETSC_VIEWER_STDOUT_SELF);
37: if (dim > 2) {PetscIntView(n,lz,PETSC_VIEWER_STDOUT_SELF);}
38: }
39: MPI_Barrier(PETSC_COMM_WORLD);
40: }
41: DMDestroy(&da);
42: PetscFinalize();
43: return ierr;
44: }