Actual source code: ex11.c
2: static char help[] = "Tests various 2-dimensional DMDA routines.\n\n";
4: #include <petscdmda.h>
5: #include <petscdraw.h>
7: int main(int argc,char **argv)
8: {
9: PetscInt M = 10,N = 8,dof=1,s=1,bx=0,by=0,i,n,j,k,m,wrap,xs,ys;
10: DM da,dac;
11: PetscViewer viewer;
12: Vec local,global,coors;
13: PetscScalar ***xy,***aglobal;
14: PetscDraw draw;
15: char fname[32];
17: PetscInitialize(&argc,&argv,(char*)0,help);
18: /* Create viewers */
19: PetscViewerDrawOpen(PETSC_COMM_WORLD,0,"",PETSC_DECIDE,PETSC_DECIDE,600,200,&viewer);
20: PetscViewerDrawGetDraw(viewer,0,&draw);
21: PetscDrawSetDoubleBuffer(draw);
23: /* Read options */
24: PetscOptionsGetInt(NULL,NULL,"-M",&M,NULL);
25: PetscOptionsGetInt(NULL,NULL,"-N",&N,NULL);
26: PetscOptionsGetInt(NULL,NULL,"-dof",&dof,NULL);
27: PetscOptionsGetInt(NULL,NULL,"-s",&s,NULL);
28: PetscOptionsGetInt(NULL,NULL,"-periodic_x",&wrap,NULL);
29: PetscOptionsGetInt(NULL,NULL,"-periodic_y",&wrap,NULL);
31: /* Create distributed array and get vectors */
32: DMDACreate2d(PETSC_COMM_WORLD,(DMBoundaryType)bx,(DMBoundaryType)by,DMDA_STENCIL_BOX,M,N,PETSC_DECIDE,PETSC_DECIDE,dof,s,NULL,NULL,&da);
33: DMSetFromOptions(da);
34: DMSetUp(da);
35: DMDASetUniformCoordinates(da,0.0,1.0,0.0,1.0,0.0,0.0);
36: for (i=0; i<dof; i++) {
37: sprintf(fname,"Field %d",(int)i);
38: DMDASetFieldName(da,i,fname);
39: }
41: DMView(da,viewer);
42: DMCreateGlobalVector(da,&global);
43: DMCreateLocalVector(da,&local);
44: DMGetCoordinates(da,&coors);
45: DMGetCoordinateDM(da,&dac);
47: /* Set values into global vectors */
48: DMDAVecGetArrayDOFRead(dac,coors,&xy);
49: DMDAVecGetArrayDOF(da,global,&aglobal);
50: DMDAGetCorners(da,&xs,&ys,0,&m,&n,0);
51: for (k=0; k<dof; k++) {
52: for (j=ys; j<ys+n; j++) {
53: for (i=xs; i<xs+m; i++) {
54: aglobal[j][i][k] = PetscSinScalar(2.0*PETSC_PI*(k+1)*xy[j][i][0]);
55: }
56: }
57: }
58: DMDAVecRestoreArrayDOF(da,global,&aglobal);
59: DMDAVecRestoreArrayDOFRead(dac,coors,&xy);
60: DMGlobalToLocalBegin(da,global,INSERT_VALUES,local);
61: DMGlobalToLocalEnd(da,global,INSERT_VALUES,local);
63: VecSet(global,0.0);
64: DMLocalToGlobalBegin(da,local,INSERT_VALUES,global);
65: DMLocalToGlobalEnd(da,local,INSERT_VALUES,global);
66: VecView(global,PETSC_VIEWER_STDOUT_WORLD);
67: VecView(global,viewer);
69: /* Free memory */
70: PetscViewerDestroy(&viewer);
71: VecDestroy(&global);
72: VecDestroy(&local);
73: DMDestroy(&da);
74: PetscFinalize();
75: return 0;
76: }
78: /*TEST
80: test:
81: args: -dof 2
82: filter: grep -v -i Object
83: requires: x
85: test:
86: suffix: 2
87: nsize: 2
88: args: -dof 2
89: filter: grep -v -i Object
90: requires: x
92: test:
93: suffix: 3
94: nsize: 3
95: args: -dof 2
96: filter: grep -v -i Object
97: requires: x
99: TEST*/