Actual source code: ex14.c
2: static char help[] = "Tests saving DMDA vectors to files.\n\n";
4: /*
5: ex13.c reads in the DMDA and vector written by this program.
6: */
8: #include <petscdm.h>
9: #include <petscdmda.h>
11: int main(int argc, char **argv)
12: {
13: PetscMPIInt rank;
14: PetscInt M = 10, N = 8, m = PETSC_DECIDE, n = PETSC_DECIDE, dof = 1;
15: DM da;
16: Vec local, global, natural;
17: PetscScalar value;
18: PetscViewer bviewer;
20: PetscFunctionBeginUser;
21: PetscCall(PetscInitialize(&argc, &argv, (char *)0, help));
22: PetscCall(PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL));
23: PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL));
24: PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL));
25: PetscCall(PetscOptionsGetInt(NULL, NULL, "-n", &n, NULL));
26: PetscCall(PetscOptionsGetInt(NULL, NULL, "-dof", &dof, NULL));
28: /* Create distributed array and get vectors */
29: PetscCall(DMDACreate2d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_BOX, M, N, m, n, dof, 1, NULL, NULL, &da));
30: PetscCall(DMSetFromOptions(da));
31: PetscCall(DMSetUp(da));
32: PetscCall(DMCreateGlobalVector(da, &global));
33: PetscCall(DMCreateLocalVector(da, &local));
35: value = -3.0;
36: PetscCall(VecSet(global, value));
37: PetscCall(DMGlobalToLocalBegin(da, global, INSERT_VALUES, local));
38: PetscCall(DMGlobalToLocalEnd(da, global, INSERT_VALUES, local));
40: PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
41: value = rank + 1;
42: PetscCall(VecScale(local, value));
43: PetscCall(DMLocalToGlobalBegin(da, local, ADD_VALUES, global));
44: PetscCall(DMLocalToGlobalEnd(da, local, ADD_VALUES, global));
46: PetscCall(DMDACreateNaturalVector(da, &natural));
47: PetscCall(DMDAGlobalToNaturalBegin(da, global, INSERT_VALUES, natural));
48: PetscCall(DMDAGlobalToNaturalEnd(da, global, INSERT_VALUES, natural));
50: PetscCall(DMDASetFieldName(da, 0, "First field"));
51: /* PetscCall(VecView(global,PETSC_VIEWER_DRAW_WORLD)); */
53: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "daoutput", FILE_MODE_WRITE, &bviewer));
54: PetscCall(DMView(da, bviewer));
55: PetscCall(VecView(global, bviewer));
56: PetscCall(PetscViewerDestroy(&bviewer));
58: /* Free memory */
59: PetscCall(VecDestroy(&local));
60: PetscCall(VecDestroy(&global));
61: PetscCall(VecDestroy(&natural));
62: PetscCall(DMDestroy(&da));
63: PetscCall(PetscFinalize());
64: return 0;
65: }