1c4762a1bSJed Brown static char help[] = "Tests various DM routines.\n\n";
2c4762a1bSJed Brown
3c4762a1bSJed Brown #include <petscdm.h>
4c4762a1bSJed Brown #include <petscdmda.h>
5c4762a1bSJed Brown
main(int argc,char ** argv)6d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv)
7d71ae5a4SJacob Faibussowitsch {
8c4762a1bSJed Brown PetscMPIInt rank;
9c4762a1bSJed Brown PetscInt M = 10, N = 8, m = PETSC_DECIDE, n = PETSC_DECIDE;
10c4762a1bSJed Brown DM da;
11c4762a1bSJed Brown PetscViewer viewer;
12c4762a1bSJed Brown Vec local, global;
13c4762a1bSJed Brown PetscScalar value;
14c4762a1bSJed Brown
15327415f7SBarry Smith PetscFunctionBeginUser;
16*c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &argv, NULL, help));
179566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawOpen(PETSC_COMM_WORLD, 0, "", 300, 0, 300, 300, &viewer));
18c4762a1bSJed Brown
19c4762a1bSJed Brown /* Read options */
209566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL));
219566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL));
229566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL));
239566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-n", &n, NULL));
24c4762a1bSJed Brown
25c4762a1bSJed Brown /* Create distributed array and get vectors */
269566063dSJacob Faibussowitsch PetscCall(DMDACreate2d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_BOX, M, N, m, n, 1, 1, NULL, NULL, &da));
279566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(da));
289566063dSJacob Faibussowitsch PetscCall(DMSetUp(da));
299566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(da, &global));
309566063dSJacob Faibussowitsch PetscCall(DMCreateLocalVector(da, &local));
31c4762a1bSJed Brown
32c4762a1bSJed Brown value = -3.0;
339566063dSJacob Faibussowitsch PetscCall(VecSet(global, value));
349566063dSJacob Faibussowitsch PetscCall(DMGlobalToLocalBegin(da, global, INSERT_VALUES, local));
359566063dSJacob Faibussowitsch PetscCall(DMGlobalToLocalEnd(da, global, INSERT_VALUES, local));
36c4762a1bSJed Brown
379566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
38c4762a1bSJed Brown value = rank + 1;
399566063dSJacob Faibussowitsch PetscCall(VecScale(local, value));
409566063dSJacob Faibussowitsch PetscCall(DMLocalToGlobalBegin(da, local, ADD_VALUES, global));
419566063dSJacob Faibussowitsch PetscCall(DMLocalToGlobalEnd(da, local, ADD_VALUES, global));
42c4762a1bSJed Brown
439566063dSJacob Faibussowitsch PetscCall(VecView(global, PETSC_VIEWER_STDOUT_WORLD));
449566063dSJacob Faibussowitsch PetscCall(DMView(da, viewer));
45c4762a1bSJed Brown
46c4762a1bSJed Brown /* Free memory */
479566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer));
489566063dSJacob Faibussowitsch PetscCall(VecDestroy(&local));
499566063dSJacob Faibussowitsch PetscCall(VecDestroy(&global));
509566063dSJacob Faibussowitsch PetscCall(DMDestroy(&da));
519566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
52b122ec5aSJacob Faibussowitsch return 0;
53c4762a1bSJed Brown }
54c4762a1bSJed Brown
55c4762a1bSJed Brown /*TEST
56c4762a1bSJed Brown
57c4762a1bSJed Brown test:
58c4762a1bSJed Brown nsize: 2
59c4762a1bSJed Brown args: -nox
60c4762a1bSJed Brown filter: grep -v -i Object
61c4762a1bSJed Brown
62c4762a1bSJed Brown test:
63c4762a1bSJed Brown suffix: cuda1
64c4762a1bSJed Brown requires: cuda
65c4762a1bSJed Brown args: -dm_vec_type cuda -nox
66c4762a1bSJed Brown filter: grep -v -i Object
67c4762a1bSJed Brown
68c4762a1bSJed Brown test:
69c4762a1bSJed Brown suffix: cuda2
70c4762a1bSJed Brown nsize: 2
71c4762a1bSJed Brown requires: cuda
72c4762a1bSJed Brown args: -dm_vec_type cuda -nox
73c4762a1bSJed Brown filter: grep -v -i Object
74c4762a1bSJed Brown
75c4762a1bSJed Brown TEST*/
76