1c4762a1bSJed Brown static char help[] = "Tests DMGetGlobalVector() and DMRestoreGlobalVector().\n\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown /* 4c4762a1bSJed Brown Use the options 5c4762a1bSJed Brown -da_grid_x <nx> - number of grid points in x direction, if M < 0 6c4762a1bSJed Brown -da_grid_y <ny> - number of grid points in y direction, if N < 0 7c4762a1bSJed Brown -da_processors_x <MX> number of processors in x directio 8c4762a1bSJed Brown -da_processors_y <MY> number of processors in x direction 9c4762a1bSJed Brown */ 10c4762a1bSJed Brown 11c4762a1bSJed Brown #include <petscdm.h> 12c4762a1bSJed Brown #include <petscdmda.h> 13c4762a1bSJed Brown 14d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv) 15d71ae5a4SJacob Faibussowitsch { 16c4762a1bSJed Brown PetscInt M = 10, N = 8; 17c4762a1bSJed Brown PetscBool flg = PETSC_FALSE; 18c4762a1bSJed Brown DM da; 19c4762a1bSJed Brown Vec global1, global2, global3; 20c4762a1bSJed Brown DMBoundaryType bx = DM_BOUNDARY_NONE, by = DM_BOUNDARY_NONE; 21c4762a1bSJed Brown DMDAStencilType stype = DMDA_STENCIL_BOX; 22c4762a1bSJed Brown 23327415f7SBarry Smith PetscFunctionBeginUser; 24c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &argv, NULL, help)); 259566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(NULL, NULL, "-star_stencil", &flg, NULL)); 26c4762a1bSJed Brown if (flg) stype = DMDA_STENCIL_STAR; 27c4762a1bSJed Brown 28c4762a1bSJed Brown /* Create distributed array and get vectors */ 299566063dSJacob Faibussowitsch PetscCall(DMDACreate2d(PETSC_COMM_WORLD, bx, by, stype, M, N, PETSC_DECIDE, PETSC_DECIDE, 1, 1, NULL, NULL, &da)); 309566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(da)); 319566063dSJacob Faibussowitsch PetscCall(DMSetUp(da)); 329566063dSJacob Faibussowitsch PetscCall(DMGetGlobalVector(da, &global1)); 339566063dSJacob Faibussowitsch PetscCall(DMGetGlobalVector(da, &global2)); 349566063dSJacob Faibussowitsch PetscCall(DMRestoreGlobalVector(da, &global1)); 359566063dSJacob Faibussowitsch PetscCall(DMRestoreGlobalVector(da, &global2)); 369566063dSJacob Faibussowitsch PetscCall(DMGetGlobalVector(da, &global1)); 379566063dSJacob Faibussowitsch PetscCall(DMGetGlobalVector(da, &global3)); 389566063dSJacob Faibussowitsch PetscCall(DMGetGlobalVector(da, &global2)); 399566063dSJacob Faibussowitsch PetscCall(DMRestoreGlobalVector(da, &global1)); 409566063dSJacob Faibussowitsch PetscCall(DMRestoreGlobalVector(da, &global3)); 419566063dSJacob Faibussowitsch PetscCall(DMRestoreGlobalVector(da, &global2)); 429566063dSJacob Faibussowitsch PetscCall(DMGetGlobalVector(da, &global1)); 439566063dSJacob Faibussowitsch PetscCall(DMGetGlobalVector(da, &global3)); 449566063dSJacob Faibussowitsch PetscCall(DMGetGlobalVector(da, &global2)); 459566063dSJacob Faibussowitsch PetscCall(DMRestoreGlobalVector(da, &global1)); 469566063dSJacob Faibussowitsch PetscCall(DMRestoreGlobalVector(da, &global3)); 479566063dSJacob Faibussowitsch PetscCall(DMRestoreGlobalVector(da, &global2)); 489566063dSJacob Faibussowitsch PetscCall(DMDestroy(&da)); 499566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 50b122ec5aSJacob Faibussowitsch return 0; 51c4762a1bSJed Brown } 52c4762a1bSJed Brown 53c4762a1bSJed Brown /*TEST 54c4762a1bSJed Brown 55c4762a1bSJed Brown test: 56*3886731fSPierre Jolivet output_file: output/empty.out 57c4762a1bSJed Brown 58c4762a1bSJed Brown TEST*/ 59