1c4762a1bSJed Brown static const char help[] = "Test DMDAGetOwnershipRanges()\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown #include <petscdm.h> 4c4762a1bSJed Brown #include <petscdmda.h> 5c4762a1bSJed Brown 6c4762a1bSJed Brown int main(int argc,char *argv[]) 7c4762a1bSJed Brown { 8c4762a1bSJed Brown DM da; 9c4762a1bSJed Brown PetscViewer vw; 10c4762a1bSJed Brown PetscInt dim = 2,m,n,p; 11c4762a1bSJed Brown const PetscInt *lx,*ly,*lz; 12c4762a1bSJed Brown PetscMPIInt rank; 13c4762a1bSJed Brown 14*b122ec5aSJacob Faibussowitsch CHKERRQ(PetscInitialize(&argc,&argv,0,help)); 155f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsGetInt(NULL,0,"-dim",&dim,0)); 16c4762a1bSJed Brown switch (dim) { 17c4762a1bSJed Brown case 2: 185f80ce2aSJacob Faibussowitsch CHKERRQ(DMDACreate2d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE, DM_BOUNDARY_NONE,DMDA_STENCIL_STAR, 3,5,PETSC_DECIDE,PETSC_DECIDE,2,1,NULL,NULL,&da)); 19c4762a1bSJed Brown break; 20c4762a1bSJed Brown case 3: 215f80ce2aSJacob Faibussowitsch CHKERRQ(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)); 22c4762a1bSJed Brown break; 2398921bdaSJacob Faibussowitsch default: SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"No support for %D dimensions",dim); 24c4762a1bSJed Brown } 255f80ce2aSJacob Faibussowitsch CHKERRQ(DMSetFromOptions(da)); 265f80ce2aSJacob Faibussowitsch CHKERRQ(DMSetUp(da)); 275f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAGetInfo(da, 0, 0,0,0, &m,&n,&p, 0,0, 0,0,0,0)); 285f80ce2aSJacob Faibussowitsch CHKERRQ(DMDAGetOwnershipRanges(da,&lx,&ly,&lz)); 295f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank)); 30c4762a1bSJed Brown 315f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&vw)); 325f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPrintf(vw,"[%d] lx ly%s\n",rank,dim>2 ? " lz" : "")); 335f80ce2aSJacob Faibussowitsch CHKERRQ(PetscIntView(m,lx,vw)); 345f80ce2aSJacob Faibussowitsch CHKERRQ(PetscIntView(n,ly,vw)); 355f80ce2aSJacob Faibussowitsch if (dim > 2) CHKERRQ(PetscIntView(n,lz,vw)); 365f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&vw)); 375f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD)); 38c4762a1bSJed Brown 395f80ce2aSJacob Faibussowitsch CHKERRQ(DMDestroy(&da)); 40*b122ec5aSJacob Faibussowitsch CHKERRQ(PetscFinalize()); 41*b122ec5aSJacob Faibussowitsch return 0; 42c4762a1bSJed Brown } 43c4762a1bSJed Brown 44c4762a1bSJed Brown /*TEST 45c4762a1bSJed Brown 46c4762a1bSJed Brown test: 47c4762a1bSJed Brown nsize: 12 48c4762a1bSJed Brown args: -dm_view -dim 3 -da_grid_x 11 -da_grid_y 5 -da_grid_z 7 49c4762a1bSJed Brown 50c4762a1bSJed Brown TEST*/ 51