Actual source code: ex14.c
1: /*$Id: ex14.c,v 1.17 2001/08/07 03:04:42 balay Exp $*/
3: static char help[] = "Tests saving DA vectors to files.\n\n";
5: #include petscda.h
6: #include petscsys.h
10: int main(int argc,char **argv)
11: {
12: int rank,M = 10,N = 8,m = PETSC_DECIDE,n = PETSC_DECIDE,ierr;
13: int dof = 1;
14: DA da;
15: Vec local,global,natural;
16: PetscScalar value;
17: PetscViewer bviewer;
19: PetscInitialize(&argc,&argv,(char*)0,help);
21: /* Read options */
22: PetscOptionsGetInt(PETSC_NULL,"-M",&M,PETSC_NULL);
23: PetscOptionsGetInt(PETSC_NULL,"-N",&N,PETSC_NULL);
24: PetscOptionsGetInt(PETSC_NULL,"-m",&m,PETSC_NULL);
25: PetscOptionsGetInt(PETSC_NULL,"-n",&n,PETSC_NULL);
26: PetscOptionsGetInt(PETSC_NULL,"-dof",&dof,PETSC_NULL);
28: /* Create distributed array and get vectors */
29: DACreate2d(PETSC_COMM_WORLD,DA_NONPERIODIC,DA_STENCIL_BOX,
30: M,N,m,n,dof,1,PETSC_NULL,PETSC_NULL,&da);
31: DACreateGlobalVector(da,&global);
32: DACreateLocalVector(da,&local);
34: value = -3.0;
35: VecSet(&value,global);
36: DAGlobalToLocalBegin(da,global,INSERT_VALUES,local);
37: DAGlobalToLocalEnd(da,global,INSERT_VALUES,local);
39: MPI_Comm_rank(PETSC_COMM_WORLD,&rank);
40: value = rank+1;
41: VecScale(&value,local);
42: DALocalToGlobal(da,local,ADD_VALUES,global);
44: DACreateNaturalVector(da,&natural);
45: DAGlobalToNaturalBegin(da,global,INSERT_VALUES,natural);
46: DAGlobalToNaturalEnd(da,global,INSERT_VALUES,natural);
48: DASetFieldName(da,0,"First field");
49: VecView(global,PETSC_VIEWER_DRAW_WORLD);
51: PetscViewerBinaryOpen(PETSC_COMM_WORLD,"daoutput",PETSC_FILE_CREATE,&bviewer);
52: DAView(da,bviewer);
53: VecView(global,bviewer);
54: PetscViewerDestroy(bviewer);
56: /* Free memory */
57: VecDestroy(local);
58: VecDestroy(global);
59: VecDestroy(natural);
60: DADestroy(da);
61: PetscFinalize();
62: return 0;
63: }
64: