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