xref: /petsc/src/dm/tests/ex39.c (revision d71ae5a4db6382e7f06317b8d368875286fe9008)
1c4762a1bSJed Brown 
2c4762a1bSJed Brown static char help[] = "Tests mirror boundary conditions in 1-d.\n\n";
3c4762a1bSJed Brown 
4c4762a1bSJed Brown #include <petscdm.h>
5c4762a1bSJed Brown #include <petscdmda.h>
6c4762a1bSJed Brown 
7*d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv)
8*d71ae5a4SJacob Faibussowitsch {
9c4762a1bSJed Brown   PetscInt      M = 6, stencil_width = 1, dof = 1, m, xstart, i, j;
10c4762a1bSJed Brown   DM            da;
11c4762a1bSJed Brown   Vec           global, local;
12c4762a1bSJed Brown   PetscScalar **vglobal;
13c4762a1bSJed Brown   PetscViewer   sviewer;
14c4762a1bSJed Brown 
15327415f7SBarry Smith   PetscFunctionBeginUser;
169566063dSJacob Faibussowitsch   PetscCall(PetscInitialize(&argc, &argv, (char *)0, help));
179566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL, 0, "-stencil_width", &stencil_width, 0));
189566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetInt(NULL, 0, "-dof", &dof, 0));
19c4762a1bSJed Brown 
209566063dSJacob Faibussowitsch   PetscCall(DMDACreate1d(PETSC_COMM_WORLD, DM_BOUNDARY_MIRROR, M, dof, stencil_width, NULL, &da));
219566063dSJacob Faibussowitsch   PetscCall(DMSetFromOptions(da));
229566063dSJacob Faibussowitsch   PetscCall(DMSetUp(da));
239566063dSJacob Faibussowitsch   PetscCall(DMDAGetCorners(da, &xstart, 0, 0, &m, 0, 0));
24c4762a1bSJed Brown 
259566063dSJacob Faibussowitsch   PetscCall(DMCreateGlobalVector(da, &global));
269566063dSJacob Faibussowitsch   PetscCall(DMDAVecGetArrayDOF(da, global, &vglobal));
27c4762a1bSJed Brown   for (i = xstart; i < xstart + m; i++) {
28ad540459SPierre Jolivet     for (j = 0; j < dof; j++) vglobal[i][j] = 100 * (i + 1) + j;
29c4762a1bSJed Brown   }
309566063dSJacob Faibussowitsch   PetscCall(DMDAVecRestoreArrayDOF(da, global, &vglobal));
31c4762a1bSJed Brown 
329566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector(da, &local));
339566063dSJacob Faibussowitsch   PetscCall(DMGlobalToLocalBegin(da, global, INSERT_VALUES, local));
349566063dSJacob Faibussowitsch   PetscCall(DMGlobalToLocalEnd(da, global, INSERT_VALUES, local));
35c4762a1bSJed Brown 
369566063dSJacob Faibussowitsch   PetscCall(PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &sviewer));
379566063dSJacob Faibussowitsch   PetscCall(VecView(local, sviewer));
389566063dSJacob Faibussowitsch   PetscCall(PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD, PETSC_COMM_SELF, &sviewer));
399566063dSJacob Faibussowitsch   PetscCall(PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD));
409566063dSJacob Faibussowitsch   PetscCall(VecView(global, PETSC_VIEWER_STDOUT_WORLD));
41c4762a1bSJed Brown 
429566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&da));
439566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&local));
449566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&global));
45c4762a1bSJed Brown 
469566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
47b122ec5aSJacob Faibussowitsch   return 0;
48c4762a1bSJed Brown }
49c4762a1bSJed Brown 
50c4762a1bSJed Brown /*TEST
51c4762a1bSJed Brown 
52c4762a1bSJed Brown    test:
53c4762a1bSJed Brown 
54c4762a1bSJed Brown    test:
55c4762a1bSJed Brown       suffix: 2
56c4762a1bSJed Brown       nsize: 2
57c4762a1bSJed Brown       filter: grep -v "Vec Object"
58c4762a1bSJed Brown 
59c4762a1bSJed Brown TEST*/
60