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