Actual source code: ex2.c
2: static char help[] = "Tests various 1-dimensional DA routines.\n\n";
4: #include petscda.h
8: int main(int argc,char **argv)
9: {
10: PetscMPIInt rank;
11: PetscInt M = 13,s=1,dof=1;
12: DAPeriodicType wrap = DA_XPERIODIC;
14: DA da;
15: PetscViewer viewer;
16: Vec local,global;
17: PetscScalar value;
18: PetscDraw draw;
19: PetscTruth flg = PETSC_FALSE;
21: PetscInitialize(&argc,&argv,(char*)0,help);
23: /* Create viewers */
24: PetscViewerDrawOpen(PETSC_COMM_WORLD,0,"",280,480,600,200,&viewer);
25: PetscViewerDrawGetDraw(viewer,0,&draw);
26: PetscDrawSetDoubleBuffer(draw);
28: /* Readoptions */
29: PetscOptionsGetInt(PETSC_NULL,"-M",&M,PETSC_NULL);
30: PetscOptionsGetEnum(PETSC_NULL,"-wrap",DAPeriodicTypes,(PetscEnum*)&wrap,PETSC_NULL);
31: PetscOptionsGetInt(PETSC_NULL,"-dof",&dof,PETSC_NULL);
32: PetscOptionsGetInt(PETSC_NULL,"-s",&s,PETSC_NULL);
34: /* Create distributed array and get vectors */
35: DACreate1d(PETSC_COMM_WORLD,wrap,M,dof,s,PETSC_NULL,&da);
36: DAView(da,viewer);
37: DACreateGlobalVector(da,&global);
38: DACreateLocalVector(da,&local);
40: /* Set global vector; send ghost points to local vectors */
41: value = 1;
42: VecSet(global,value);
43: DAGlobalToLocalBegin(da,global,INSERT_VALUES,local);
44: DAGlobalToLocalEnd(da,global,INSERT_VALUES,local);
46: /* Scale local vectors according to processor rank; pass to global vector */
47: MPI_Comm_rank(PETSC_COMM_WORLD,&rank);
48: value = rank+1;
49: VecScale(local,value);
50: DALocalToGlobal(da,local,INSERT_VALUES,global);
52: VecView(global,viewer);
53: PetscPrintf(PETSC_COMM_WORLD,"\nGlobal Vector:\n");
54: VecView(global,PETSC_VIEWER_STDOUT_WORLD);
55: PetscPrintf(PETSC_COMM_WORLD,"\n");
57: /* Send ghost points to local vectors */
58: DAGlobalToLocalBegin(da,global,INSERT_VALUES,local);
59: DAGlobalToLocalEnd(da,global,INSERT_VALUES,local);
61: PetscOptionsGetTruth(PETSC_NULL,"-local_print",&flg,PETSC_NULL);
62: if (flg) {
63: PetscViewer sviewer;
64: ISLocalToGlobalMapping is;
66: PetscSynchronizedPrintf(PETSC_COMM_WORLD,"\nLocal Vector: processor %d\n",rank);
67: PetscViewerGetSingleton(PETSC_VIEWER_STDOUT_WORLD,&sviewer);
68: VecView(local,sviewer);
69: PetscViewerRestoreSingleton(PETSC_VIEWER_STDOUT_WORLD,&sviewer);
70: PetscSynchronizedFlush(PETSC_COMM_WORLD);
72: PetscSynchronizedPrintf(PETSC_COMM_WORLD,"\nLocal to global mapping: processor %d\n",rank);
73: PetscViewerGetSingleton(PETSC_VIEWER_STDOUT_WORLD,&sviewer);
74: DAGetISLocalToGlobalMapping(da,&is);
75: ISLocalToGlobalMappingView(is,sviewer);
76: PetscViewerRestoreSingleton(PETSC_VIEWER_STDOUT_WORLD,&sviewer);
77: PetscSynchronizedFlush(PETSC_COMM_WORLD);
78: }
80: /* Free memory */
81: PetscViewerDestroy(viewer);
82: VecDestroy(global);
83: VecDestroy(local);
84: DADestroy(da);
85: PetscFinalize();
86: return 0;
87: }
88: