1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "Tests DMGetGlobalVector() and DMRestoreGlobalVector().\n\n"; 3c4762a1bSJed Brown 4c4762a1bSJed Brown /* 5c4762a1bSJed Brown Use the options 6c4762a1bSJed Brown -da_grid_x <nx> - number of grid points in x direction, if M < 0 7c4762a1bSJed Brown -da_grid_y <ny> - number of grid points in y direction, if N < 0 8c4762a1bSJed Brown -da_processors_x <MX> number of processors in x directio 9c4762a1bSJed Brown -da_processors_y <MY> number of processors in x direction 10c4762a1bSJed Brown */ 11c4762a1bSJed Brown 12c4762a1bSJed Brown #include <petscdm.h> 13c4762a1bSJed Brown #include <petscdmda.h> 14c4762a1bSJed Brown 15c4762a1bSJed Brown int main(int argc,char **argv) 16c4762a1bSJed Brown { 17c4762a1bSJed Brown PetscInt M = 10,N = 8; 18c4762a1bSJed Brown PetscErrorCode ierr; 19c4762a1bSJed Brown PetscBool flg = PETSC_FALSE; 20c4762a1bSJed Brown DM da; 21c4762a1bSJed Brown Vec global1,global2,global3; 22c4762a1bSJed Brown DMBoundaryType bx = DM_BOUNDARY_NONE,by = DM_BOUNDARY_NONE; 23c4762a1bSJed Brown DMDAStencilType stype = DMDA_STENCIL_BOX; 24c4762a1bSJed Brown 25c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 26*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscOptionsGetBool(NULL,NULL,"-star_stencil",&flg,NULL)); 27c4762a1bSJed Brown if (flg) stype = DMDA_STENCIL_STAR; 28c4762a1bSJed Brown 29c4762a1bSJed Brown /* Create distributed array and get vectors */ 30*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDACreate2d(PETSC_COMM_WORLD,bx,by,stype,M,N,PETSC_DECIDE,PETSC_DECIDE,1,1,NULL,NULL,&da)); 31*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMSetFromOptions(da)); 32*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMSetUp(da)); 33*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(da,&global1)); 34*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(da,&global2)); 35*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(da,&global1)); 36*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(da,&global2)); 37*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(da,&global1)); 38*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(da,&global3)); 39*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(da,&global2)); 40*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(da,&global1)); 41*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(da,&global3)); 42*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(da,&global2)); 43*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(da,&global1)); 44*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(da,&global3)); 45*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMGetGlobalVector(da,&global2)); 46*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(da,&global1)); 47*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(da,&global3)); 48*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMRestoreGlobalVector(da,&global2)); 49*5f80ce2aSJacob Faibussowitsch CHKERRQ(DMDestroy(&da)); 50c4762a1bSJed Brown ierr = PetscFinalize(); 51c4762a1bSJed Brown return ierr; 52c4762a1bSJed Brown } 53c4762a1bSJed Brown 54c4762a1bSJed Brown /*TEST 55c4762a1bSJed Brown 56c4762a1bSJed Brown test: 57c4762a1bSJed Brown 58c4762a1bSJed Brown TEST*/ 59