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