Actual source code: ex44.c
1: static char help[] = "Tests various DMComposite routines.\n\n";
3: #include <petscdm.h>
4: #include <petscdmda.h>
5: #include <petscdmcomposite.h>
7: int main(int argc, char **argv)
8: {
9: PetscMPIInt rank;
10: DM da1, da2, packer;
11: Vec local, global, globals[2], buffer;
12: PetscScalar value, *array;
13: PetscInt n;
14: PetscViewer viewer;
16: PetscFunctionBeginUser;
17: PetscCall(PetscInitialize(&argc, &argv, NULL, help));
19: PetscCall(DMCompositeCreate(PETSC_COMM_WORLD, &packer));
20: PetscCall(DMDACreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, 8, 1, 1, NULL, &da1));
21: PetscCall(DMSetFromOptions(da1));
22: PetscCall(DMSetUp(da1));
23: PetscCall(DMCompositeAddDM(packer, da1));
24: PetscCall(DMDACreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, 6, 1, 1, NULL, &da2));
25: PetscCall(DMSetFromOptions(da2));
26: PetscCall(DMSetUp(da2));
27: PetscCall(DMCompositeAddDM(packer, da2));
29: PetscCall(DMCreateGlobalVector(packer, &global));
30: PetscCall(DMCreateLocalVector(packer, &local));
31: PetscCall(DMCreateLocalVector(packer, &buffer));
33: PetscCall(DMCompositeGetAccessArray(packer, global, 2, NULL, globals));
34: value = 1;
35: PetscCall(VecSet(globals[0], value));
36: value = -1;
37: PetscCall(VecSet(globals[1], value));
38: PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
39: value = rank + 1;
40: PetscCall(VecGetLocalSize(globals[0], &n));
41: PetscCall(VecGetArray(globals[0], &array));
42: for (PetscInt i = 0; i < n; i++) array[i] *= value;
43: PetscCall(VecRestoreArray(globals[0], &array));
44: PetscCall(VecGetLocalSize(globals[1], &n));
45: PetscCall(VecGetArray(globals[1], &array));
46: for (PetscInt i = 0; i < n; i++) array[i] *= value;
47: PetscCall(VecRestoreArray(globals[1], &array));
48: PetscCall(DMCompositeRestoreAccessArray(packer, global, 2, NULL, globals));
50: /* Test GlobalToLocal in insert mode */
51: PetscCall(DMGlobalToLocalBegin(packer, global, INSERT_VALUES, local));
52: PetscCall(DMGlobalToLocalEnd(packer, global, INSERT_VALUES, local));
54: PetscCall(PetscViewerASCIIPushSynchronized(PETSC_VIEWER_STDOUT_WORLD));
55: PetscCall(PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &viewer));
56: PetscCall(PetscViewerASCIIPrintf(viewer, "\nLocal Vector: processor %d\n", rank));
57: PetscCall(VecView(local, viewer));
58: PetscCall(PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &viewer));
59: PetscCall(PetscViewerASCIIPopSynchronized(PETSC_VIEWER_STDOUT_WORLD));
61: /* Test LocalToGlobal in insert mode */
62: PetscCall(DMLocalToGlobalBegin(packer, local, INSERT_VALUES, global));
63: PetscCall(DMLocalToGlobalEnd(packer, local, INSERT_VALUES, global));
65: PetscCall(VecView(global, PETSC_VIEWER_STDOUT_WORLD));
67: /* Test LocalToLocal in insert mode */
68: PetscCall(DMLocalToLocalBegin(packer, local, INSERT_VALUES, buffer));
69: PetscCall(DMLocalToLocalEnd(packer, local, INSERT_VALUES, buffer));
71: PetscCall(PetscViewerASCIIPushSynchronized(PETSC_VIEWER_STDOUT_WORLD));
72: PetscCall(PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &viewer));
73: PetscCall(PetscViewerASCIIPrintf(viewer, "\nLocal Vector: processor %d\n", rank));
74: PetscCall(VecView(buffer, viewer));
75: PetscCall(PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &viewer));
76: PetscCall(PetscViewerASCIIPopSynchronized(PETSC_VIEWER_STDOUT_WORLD));
78: PetscCall(VecDestroy(&buffer));
79: PetscCall(VecDestroy(&local));
80: PetscCall(VecDestroy(&global));
81: PetscCall(DMDestroy(&packer));
82: PetscCall(DMDestroy(&da2));
83: PetscCall(DMDestroy(&da1));
85: PetscCall(PetscFinalize());
86: return 0;
87: }
89: /*TEST
91: test:
92: nsize: 3
94: TEST*/