Actual source code: ex14.c
petsc-3.3-p6 2013-02-11
2: static char help[] = "Tests saving DM vectors to files.\n\n";
4: #include <petscdmda.h>
8: int main(int argc,char **argv)
9: {
10: PetscMPIInt rank;
11: PetscInt M = 10,N = 8,m = PETSC_DECIDE,n = PETSC_DECIDE, dof = 1;
13: DM da;
14: Vec local,global,natural;
15: PetscScalar value;
16: PetscViewer bviewer;
18: PetscInitialize(&argc,&argv,(char*)0,help);
20: /* Read options */
21: PetscOptionsGetInt(PETSC_NULL,"-M",&M,PETSC_NULL);
22: PetscOptionsGetInt(PETSC_NULL,"-N",&N,PETSC_NULL);
23: PetscOptionsGetInt(PETSC_NULL,"-m",&m,PETSC_NULL);
24: PetscOptionsGetInt(PETSC_NULL,"-n",&n,PETSC_NULL);
25: PetscOptionsGetInt(PETSC_NULL,"-dof",&dof,PETSC_NULL);
27: /* Create distributed array and get vectors */
28: DMDACreate2d(PETSC_COMM_WORLD, DMDA_BOUNDARY_NONE, DMDA_BOUNDARY_NONE,DMDA_STENCIL_BOX,
29: M,N,m,n,dof,1,PETSC_NULL,PETSC_NULL,&da);
30: DMCreateGlobalVector(da,&global);
31: DMCreateLocalVector(da,&local);
33: value = -3.0;
34: VecSet(global,value);
35: DMGlobalToLocalBegin(da,global,INSERT_VALUES,local);
36: DMGlobalToLocalEnd(da,global,INSERT_VALUES,local);
38: MPI_Comm_rank(PETSC_COMM_WORLD,&rank);
39: value = rank+1;
40: VecScale(local,value);
41: DMLocalToGlobalBegin(da,local,ADD_VALUES,global);
42: DMLocalToGlobalEnd(da,local,ADD_VALUES,global);
44: DMDACreateNaturalVector(da,&natural);
45: DMDAGlobalToNaturalBegin(da,global,INSERT_VALUES,natural);
46: DMDAGlobalToNaturalEnd(da,global,INSERT_VALUES,natural);
48: DMDASetFieldName(da,0,"First field");
49: /* VecView(global,PETSC_VIEWER_DRAW_WORLD); */
51: PetscViewerBinaryOpen(PETSC_COMM_WORLD,"daoutput",FILE_MODE_WRITE,&bviewer);
52: DMView(da,bviewer);
53: VecView(global,bviewer);
54: PetscViewerDestroy(&bviewer);
56: /* Free memory */
57: VecDestroy(&local);
58: VecDestroy(&global);
59: VecDestroy(&natural);
60: DMDestroy(&da);
61: PetscFinalize();
62: return 0;
63: }
64: