xref: /petsc/src/dm/tests/ex2.c (revision c4762a1b19cd2af06abeed90e8f9d34fb975dd94) 
1*c4762a1bSJed Brown 
2*c4762a1bSJed Brown static char help[] = "Tests various 1-dimensional DMDA routines.\n\n";
3*c4762a1bSJed Brown 
4*c4762a1bSJed Brown #include <petscdm.h>
5*c4762a1bSJed Brown #include <petscdmda.h>
6*c4762a1bSJed Brown #include <petscdraw.h>
7*c4762a1bSJed Brown 
8*c4762a1bSJed Brown int main(int argc,char **argv)
9*c4762a1bSJed Brown {
10*c4762a1bSJed Brown   PetscMPIInt            rank;
11*c4762a1bSJed Brown   PetscInt               M  = 13,s=1,dof=1;
12*c4762a1bSJed Brown   DMBoundaryType         bx = DM_BOUNDARY_PERIODIC;
13*c4762a1bSJed Brown   PetscErrorCode         ierr;
14*c4762a1bSJed Brown   DM                     da;
15*c4762a1bSJed Brown   PetscViewer            viewer;
16*c4762a1bSJed Brown   Vec                    local,global;
17*c4762a1bSJed Brown   PetscScalar            value;
18*c4762a1bSJed Brown   PetscDraw              draw;
19*c4762a1bSJed Brown   PetscBool              flg = PETSC_FALSE;
20*c4762a1bSJed Brown   ISLocalToGlobalMapping is;
21*c4762a1bSJed Brown 
22*c4762a1bSJed Brown   ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr;
23*c4762a1bSJed Brown   ierr = PetscViewerDrawOpen(PETSC_COMM_WORLD,0,"",280,480,600,200,&viewer);CHKERRQ(ierr);
24*c4762a1bSJed Brown   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
25*c4762a1bSJed Brown   ierr = PetscDrawSetDoubleBuffer(draw);CHKERRQ(ierr);
26*c4762a1bSJed Brown 
27*c4762a1bSJed Brown   /* Readoptions */
28*c4762a1bSJed Brown   ierr = PetscOptionsGetInt(NULL,NULL,"-M",&M,NULL);CHKERRQ(ierr);
29*c4762a1bSJed Brown   ierr = PetscOptionsGetEnum(NULL,NULL,"-wrap",DMBoundaryTypes,(PetscEnum*)&bx,NULL);CHKERRQ(ierr);
30*c4762a1bSJed Brown   ierr = PetscOptionsGetInt(NULL,NULL,"-dof",&dof,NULL);CHKERRQ(ierr);
31*c4762a1bSJed Brown   ierr = PetscOptionsGetInt(NULL,NULL,"-s",&s,NULL);CHKERRQ(ierr);
32*c4762a1bSJed Brown 
33*c4762a1bSJed Brown   /* Create distributed array and get vectors */
34*c4762a1bSJed Brown   ierr = DMDACreate1d(PETSC_COMM_WORLD,bx,M,dof,s,NULL,&da);CHKERRQ(ierr);
35*c4762a1bSJed Brown   ierr = DMSetFromOptions(da);CHKERRQ(ierr);
36*c4762a1bSJed Brown   ierr = DMSetUp(da);CHKERRQ(ierr);
37*c4762a1bSJed Brown   ierr = DMView(da,viewer);CHKERRQ(ierr);
38*c4762a1bSJed Brown   ierr = DMCreateGlobalVector(da,&global);CHKERRQ(ierr);
39*c4762a1bSJed Brown   ierr = DMCreateLocalVector(da,&local);CHKERRQ(ierr);
40*c4762a1bSJed Brown 
41*c4762a1bSJed Brown   value = 1;
42*c4762a1bSJed Brown   ierr  = VecSet(global,value);CHKERRQ(ierr);
43*c4762a1bSJed Brown 
44*c4762a1bSJed Brown   ierr  = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
45*c4762a1bSJed Brown   value = rank+1;
46*c4762a1bSJed Brown   ierr  = VecScale(global,value);CHKERRQ(ierr);
47*c4762a1bSJed Brown 
48*c4762a1bSJed Brown   ierr = VecView(global,viewer);CHKERRQ(ierr);
49*c4762a1bSJed Brown   ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"\nGlobal Vector:\n");CHKERRQ(ierr);
50*c4762a1bSJed Brown   ierr = VecView(global,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
51*c4762a1bSJed Brown   ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"\n");CHKERRQ(ierr);
52*c4762a1bSJed Brown 
53*c4762a1bSJed Brown   /* Send ghost points to local vectors */
54*c4762a1bSJed Brown   ierr = DMGlobalToLocalBegin(da,global,INSERT_VALUES,local);CHKERRQ(ierr);
55*c4762a1bSJed Brown   ierr = DMGlobalToLocalEnd(da,global,INSERT_VALUES,local);CHKERRQ(ierr);
56*c4762a1bSJed Brown 
57*c4762a1bSJed Brown   ierr = PetscOptionsGetBool(NULL,NULL,"-local_print",&flg,NULL);CHKERRQ(ierr);
58*c4762a1bSJed Brown   if (flg) {
59*c4762a1bSJed Brown     PetscViewer            sviewer;
60*c4762a1bSJed Brown 
61*c4762a1bSJed Brown     ierr = PetscViewerASCIIPushSynchronized(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
62*c4762a1bSJed Brown     ierr = PetscViewerASCIISynchronizedPrintf(PETSC_VIEWER_STDOUT_WORLD,"\nLocal Vector: processor %d\n",rank);CHKERRQ(ierr);
63*c4762a1bSJed Brown     ierr = PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr);
64*c4762a1bSJed Brown     ierr = VecView(local,sviewer);CHKERRQ(ierr);
65*c4762a1bSJed Brown     ierr = PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr);
66*c4762a1bSJed Brown     ierr = PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
67*c4762a1bSJed Brown     ierr = PetscViewerASCIIPopSynchronized(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
68*c4762a1bSJed Brown   }
69*c4762a1bSJed Brown   ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"\nLocal to global mapping\n");CHKERRQ(ierr);
70*c4762a1bSJed Brown   ierr = DMGetLocalToGlobalMapping(da,&is);CHKERRQ(ierr);
71*c4762a1bSJed Brown   ierr = ISLocalToGlobalMappingView(is,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr);
72*c4762a1bSJed Brown 
73*c4762a1bSJed Brown   /* Free memory */
74*c4762a1bSJed Brown   ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
75*c4762a1bSJed Brown   ierr = VecDestroy(&global);CHKERRQ(ierr);
76*c4762a1bSJed Brown   ierr = VecDestroy(&local);CHKERRQ(ierr);
77*c4762a1bSJed Brown   ierr = DMDestroy(&da);CHKERRQ(ierr);
78*c4762a1bSJed Brown   ierr = PetscFinalize();
79*c4762a1bSJed Brown   return ierr;
80*c4762a1bSJed Brown }
81*c4762a1bSJed Brown 
82*c4762a1bSJed Brown /*TEST
83*c4762a1bSJed Brown 
84*c4762a1bSJed Brown    test:
85*c4762a1bSJed Brown       nsize: 2
86*c4762a1bSJed Brown       args: -nox
87*c4762a1bSJed Brown       filter: grep -v "MPI processes"
88*c4762a1bSJed Brown       output_file: output/ex2_1.out
89*c4762a1bSJed Brown       requires: x
90*c4762a1bSJed Brown 
91*c4762a1bSJed Brown    test:
92*c4762a1bSJed Brown       suffix: 2
93*c4762a1bSJed Brown       nsize: 3
94*c4762a1bSJed Brown       args: -wrap none -local_print -nox
95*c4762a1bSJed Brown       filter: grep -v "Vec Object: Vec"
96*c4762a1bSJed Brown       requires: x
97*c4762a1bSJed Brown 
98*c4762a1bSJed Brown    test:
99*c4762a1bSJed Brown       suffix: 3
100*c4762a1bSJed Brown       nsize: 3
101*c4762a1bSJed Brown       args: -wrap ghosted -local_print -nox
102*c4762a1bSJed Brown       filter: grep -v "Vec Object: Vec"
103*c4762a1bSJed Brown       requires: x
104*c4762a1bSJed Brown 
105*c4762a1bSJed Brown TEST*/
106