1c4762a1bSJed Brown static char help[] = "Tests saving DMDA vectors to files.\n\n";
2c4762a1bSJed Brown
3c4762a1bSJed Brown /*
4c4762a1bSJed Brown ex13.c reads in the DMDA and vector written by this program.
5c4762a1bSJed Brown */
6c4762a1bSJed Brown
7c4762a1bSJed Brown #include <petscdm.h>
8c4762a1bSJed Brown #include <petscdmda.h>
9c4762a1bSJed Brown
main(int argc,char ** argv)10d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv)
11d71ae5a4SJacob Faibussowitsch {
12c4762a1bSJed Brown PetscMPIInt rank;
13c4762a1bSJed Brown PetscInt M = 10, N = 8, m = PETSC_DECIDE, n = PETSC_DECIDE, dof = 1;
14c4762a1bSJed Brown DM da;
15c4762a1bSJed Brown Vec local, global, natural;
16c4762a1bSJed Brown PetscScalar value;
17c4762a1bSJed Brown PetscViewer bviewer;
18c4762a1bSJed Brown
19327415f7SBarry Smith PetscFunctionBeginUser;
20*c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &argv, NULL, help));
219566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL));
229566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL));
239566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL));
249566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-n", &n, NULL));
259566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-dof", &dof, NULL));
26c4762a1bSJed Brown
27c4762a1bSJed Brown /* Create distributed array and get vectors */
289566063dSJacob Faibussowitsch PetscCall(DMDACreate2d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_BOX, M, N, m, n, dof, 1, NULL, NULL, &da));
299566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(da));
309566063dSJacob Faibussowitsch PetscCall(DMSetUp(da));
319566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(da, &global));
329566063dSJacob Faibussowitsch PetscCall(DMCreateLocalVector(da, &local));
33c4762a1bSJed Brown
34c4762a1bSJed Brown value = -3.0;
359566063dSJacob Faibussowitsch PetscCall(VecSet(global, value));
369566063dSJacob Faibussowitsch PetscCall(DMGlobalToLocalBegin(da, global, INSERT_VALUES, local));
379566063dSJacob Faibussowitsch PetscCall(DMGlobalToLocalEnd(da, global, INSERT_VALUES, local));
38c4762a1bSJed Brown
399566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
40c4762a1bSJed Brown value = rank + 1;
419566063dSJacob Faibussowitsch PetscCall(VecScale(local, value));
429566063dSJacob Faibussowitsch PetscCall(DMLocalToGlobalBegin(da, local, ADD_VALUES, global));
439566063dSJacob Faibussowitsch PetscCall(DMLocalToGlobalEnd(da, local, ADD_VALUES, global));
44c4762a1bSJed Brown
459566063dSJacob Faibussowitsch PetscCall(DMDACreateNaturalVector(da, &natural));
469566063dSJacob Faibussowitsch PetscCall(DMDAGlobalToNaturalBegin(da, global, INSERT_VALUES, natural));
479566063dSJacob Faibussowitsch PetscCall(DMDAGlobalToNaturalEnd(da, global, INSERT_VALUES, natural));
48c4762a1bSJed Brown
499566063dSJacob Faibussowitsch PetscCall(DMDASetFieldName(da, 0, "First field"));
509566063dSJacob Faibussowitsch /* PetscCall(VecView(global,PETSC_VIEWER_DRAW_WORLD)); */
51c4762a1bSJed Brown
529566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "daoutput", FILE_MODE_WRITE, &bviewer));
539566063dSJacob Faibussowitsch PetscCall(DMView(da, bviewer));
549566063dSJacob Faibussowitsch PetscCall(VecView(global, bviewer));
559566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&bviewer));
56c4762a1bSJed Brown
57c4762a1bSJed Brown /* Free memory */
589566063dSJacob Faibussowitsch PetscCall(VecDestroy(&local));
599566063dSJacob Faibussowitsch PetscCall(VecDestroy(&global));
609566063dSJacob Faibussowitsch PetscCall(VecDestroy(&natural));
619566063dSJacob Faibussowitsch PetscCall(DMDestroy(&da));
629566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
63b122ec5aSJacob Faibussowitsch return 0;
64c4762a1bSJed Brown }
65