Actual source code: ex11.c
1: static char help[] = "Tests various 2-dimensional DMDA routines.\n\n";
3: #include <petscdmda.h>
4: #include <petscdraw.h>
6: int main(int argc, char **argv)
7: {
8: PetscInt M = 10, N = 8, dof = 1, s = 1, bx = 0, by = 0, i, n, j, k, m, wrap, xs, ys;
9: DM da, dac;
10: PetscViewer viewer;
11: Vec local, global, coors;
12: PetscScalar ***xy, ***aglobal;
13: PetscDraw draw;
14: char fname[32];
16: PetscFunctionBeginUser;
17: PetscCall(PetscInitialize(&argc, &argv, (char *)0, help));
18: /* Create viewers */
19: PetscCall(PetscViewerDrawOpen(PETSC_COMM_WORLD, 0, "", PETSC_DECIDE, PETSC_DECIDE, 600, 200, &viewer));
20: PetscCall(PetscViewerDrawGetDraw(viewer, 0, &draw));
21: PetscCall(PetscDrawSetDoubleBuffer(draw));
23: /* Read options */
24: PetscCall(PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL));
25: PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL));
26: PetscCall(PetscOptionsGetInt(NULL, NULL, "-dof", &dof, NULL));
27: PetscCall(PetscOptionsGetInt(NULL, NULL, "-s", &s, NULL));
28: PetscCall(PetscOptionsGetInt(NULL, NULL, "-periodic_x", &wrap, NULL));
29: PetscCall(PetscOptionsGetInt(NULL, NULL, "-periodic_y", &wrap, NULL));
31: /* Create distributed array and get vectors */
32: PetscCall(DMDACreate2d(PETSC_COMM_WORLD, (DMBoundaryType)bx, (DMBoundaryType)by, DMDA_STENCIL_BOX, M, N, PETSC_DECIDE, PETSC_DECIDE, dof, s, NULL, NULL, &da));
33: PetscCall(DMSetFromOptions(da));
34: PetscCall(DMSetUp(da));
35: PetscCall(DMDASetUniformCoordinates(da, 0.0, 1.0, 0.0, 1.0, 0.0, 0.0));
36: for (i = 0; i < dof; i++) {
37: PetscCall(PetscSNPrintf(fname, PETSC_STATIC_ARRAY_LENGTH(fname), "Field %" PetscInt_FMT, i));
38: PetscCall(DMDASetFieldName(da, i, fname));
39: }
41: PetscCall(DMView(da, viewer));
42: PetscCall(DMCreateGlobalVector(da, &global));
43: PetscCall(DMCreateLocalVector(da, &local));
44: PetscCall(DMGetCoordinates(da, &coors));
45: PetscCall(DMGetCoordinateDM(da, &dac));
47: /* Set values into global vectors */
48: PetscCall(DMDAVecGetArrayDOFRead(dac, coors, &xy));
49: PetscCall(DMDAVecGetArrayDOF(da, global, &aglobal));
50: PetscCall(DMDAGetCorners(da, &xs, &ys, 0, &m, &n, 0));
51: for (k = 0; k < dof; k++) {
52: for (j = ys; j < ys + n; j++) {
53: for (i = xs; i < xs + m; i++) aglobal[j][i][k] = PetscSinScalar(2.0 * PETSC_PI * (k + 1) * xy[j][i][0]);
54: }
55: }
56: PetscCall(DMDAVecRestoreArrayDOF(da, global, &aglobal));
57: PetscCall(DMDAVecRestoreArrayDOFRead(dac, coors, &xy));
58: PetscCall(DMGlobalToLocalBegin(da, global, INSERT_VALUES, local));
59: PetscCall(DMGlobalToLocalEnd(da, global, INSERT_VALUES, local));
61: PetscCall(VecSet(global, 0.0));
62: PetscCall(DMLocalToGlobalBegin(da, local, INSERT_VALUES, global));
63: PetscCall(DMLocalToGlobalEnd(da, local, INSERT_VALUES, global));
64: PetscCall(VecView(global, PETSC_VIEWER_STDOUT_WORLD));
65: PetscCall(VecView(global, viewer));
67: /* Free memory */
68: PetscCall(PetscViewerDestroy(&viewer));
69: PetscCall(VecDestroy(&global));
70: PetscCall(VecDestroy(&local));
71: PetscCall(DMDestroy(&da));
72: PetscCall(PetscFinalize());
73: return 0;
74: }
76: /*TEST
78: test:
79: args: -dof 2
80: filter: grep -v -i Object
81: requires: x
83: test:
84: suffix: 2
85: nsize: 2
86: args: -dof 2
87: filter: grep -v -i Object
88: requires: x
90: test:
91: suffix: 3
92: nsize: 3
93: args: -dof 2
94: filter: grep -v -i Object
95: requires: x
97: TEST*/