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
main(int argc,char ** argv)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