xref: /petsc/src/dm/tests/ex39.c (revision 327415f76d85372a4417cf1aaa14db707d4d6c04)
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 
7c4762a1bSJed Brown int main(int argc,char **argv)
8c4762a1bSJed Brown {
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 
15*327415f7SBarry 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++) {
28c4762a1bSJed Brown     for (j=0; j<dof; j++) {
29c4762a1bSJed Brown       vglobal[i][j] = 100*(i+1) + j;
30c4762a1bSJed Brown     }
31c4762a1bSJed Brown   }
329566063dSJacob Faibussowitsch   PetscCall(DMDAVecRestoreArrayDOF(da,global,&vglobal));
33c4762a1bSJed Brown 
349566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector(da,&local));
359566063dSJacob Faibussowitsch   PetscCall(DMGlobalToLocalBegin(da,global,INSERT_VALUES,local));
369566063dSJacob Faibussowitsch   PetscCall(DMGlobalToLocalEnd(da,global,INSERT_VALUES,local));
37c4762a1bSJed Brown 
389566063dSJacob Faibussowitsch   PetscCall(PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer));
399566063dSJacob Faibussowitsch   PetscCall(VecView(local,sviewer));
409566063dSJacob Faibussowitsch   PetscCall(PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer));
419566063dSJacob Faibussowitsch   PetscCall(PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD));
429566063dSJacob Faibussowitsch   PetscCall(VecView(global,PETSC_VIEWER_STDOUT_WORLD));
43c4762a1bSJed Brown 
449566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&da));
459566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&local));
469566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&global));
47c4762a1bSJed Brown 
489566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
49b122ec5aSJacob Faibussowitsch   return 0;
50c4762a1bSJed Brown }
51c4762a1bSJed Brown 
52c4762a1bSJed Brown /*TEST
53c4762a1bSJed Brown 
54c4762a1bSJed Brown    test:
55c4762a1bSJed Brown 
56c4762a1bSJed Brown    test:
57c4762a1bSJed Brown       suffix: 2
58c4762a1bSJed Brown       nsize: 2
59c4762a1bSJed Brown       filter: grep -v "Vec Object"
60c4762a1bSJed Brown 
61c4762a1bSJed Brown TEST*/
62