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