Actual source code: ex40.c
2: static char help[] = "Tests mirror boundary conditions in 2-d.\n\n";
4: #include <petscdm.h>
5: #include <petscdmda.h>
7: int main(int argc,char **argv)
8: {
9: PetscInt M = 8, N = 8,stencil_width = 1, dof = 1,m,n,xstart,ystart,i,j,c;
10: DM da;
11: Vec global,local;
12: PetscScalar ***vglobal;
13: PetscViewer sview;
15: PetscInitialize(&argc,&argv,(char*)0,help);
16: PetscOptionsGetInt(NULL,0,"-stencil_width",&stencil_width,0);
17: PetscOptionsGetInt(NULL,0,"-dof",&dof,0);
19: DMDACreate2d(PETSC_COMM_WORLD,DM_BOUNDARY_MIRROR,DM_BOUNDARY_MIRROR,DMDA_STENCIL_STAR,M,N,PETSC_DECIDE,PETSC_DECIDE,dof,stencil_width,NULL,NULL,&da);
20: DMSetFromOptions(da);
21: DMSetUp(da);
22: DMDAGetCorners(da,&xstart,&ystart,0,&m,&n,0);
24: DMCreateGlobalVector(da,&global);
25: DMDAVecGetArrayDOF(da,global,&vglobal);
26: for (j=ystart; j<ystart+n; j++) {
27: for (i=xstart; i<xstart+m; i++) {
28: for (c=0; c<dof; c++) {
29: vglobal[j][i][c] = 100*j + 10*(i+1) + c;
30: }
31: }
32: }
33: DMDAVecRestoreArrayDOF(da,global,&vglobal);
35: DMCreateLocalVector(da,&local);
36: DMGlobalToLocalBegin(da,global,INSERT_VALUES,local);
37: DMGlobalToLocalEnd(da,global,INSERT_VALUES,local);
39: PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sview);
40: VecView(local,sview);
41: PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sview);
42: PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);
43: VecView(global,PETSC_VIEWER_STDOUT_WORLD);
45: DMDestroy(&da);
46: VecDestroy(&local);
47: VecDestroy(&global);
49: PetscFinalize();
50: return 0;
51: }
53: /*TEST
55: test:
57: test:
58: suffix: 2
59: nsize: 4
60: filter: grep -v "Vec Object"
62: TEST*/