1*c4762a1bSJed Brown 2*c4762a1bSJed Brown static char help[] = "Tests DMLocalToGlobal() for dof > 1\n\n"; 3*c4762a1bSJed Brown 4*c4762a1bSJed Brown #include <petscdm.h> 5*c4762a1bSJed Brown #include <petscdmda.h> 6*c4762a1bSJed Brown 7*c4762a1bSJed Brown int main(int argc,char **argv) 8*c4762a1bSJed Brown { 9*c4762a1bSJed Brown PetscInt M = 6,N = 5,m = PETSC_DECIDE,n = PETSC_DECIDE,i,j,is,js,in,jen; 10*c4762a1bSJed Brown PetscErrorCode ierr; 11*c4762a1bSJed Brown DM da; 12*c4762a1bSJed Brown Vec local,global; 13*c4762a1bSJed Brown PetscScalar ***l; 14*c4762a1bSJed Brown 15*c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 16*c4762a1bSJed Brown /* Create distributed array and get vectors */ 17*c4762a1bSJed Brown ierr = DMDACreate2d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE,DMDA_STENCIL_BOX,M,N,m,n,3,1,NULL,NULL,&da);CHKERRQ(ierr); 18*c4762a1bSJed Brown ierr = DMSetFromOptions(da);CHKERRQ(ierr); 19*c4762a1bSJed Brown ierr = DMSetUp(da);CHKERRQ(ierr); 20*c4762a1bSJed Brown ierr = DMCreateGlobalVector(da,&global);CHKERRQ(ierr); 21*c4762a1bSJed Brown ierr = DMCreateLocalVector(da,&local);CHKERRQ(ierr); 22*c4762a1bSJed Brown 23*c4762a1bSJed Brown ierr = DMDAGetCorners(da,&is,&js,0,&in,&jen,0);CHKERRQ(ierr); 24*c4762a1bSJed Brown ierr = DMDAVecGetArrayDOF(da,local,&l);CHKERRQ(ierr); 25*c4762a1bSJed Brown for (i=is; i<is+in; i++) { 26*c4762a1bSJed Brown for (j=js; j<js+jen; j++) { 27*c4762a1bSJed Brown l[j][i][0] = 3*(i + j*M); 28*c4762a1bSJed Brown l[j][i][1] = 3*(i + j*M) + 1; 29*c4762a1bSJed Brown l[j][i][2] = 3*(i + j*M) + 2; 30*c4762a1bSJed Brown } 31*c4762a1bSJed Brown } 32*c4762a1bSJed Brown ierr = DMDAVecRestoreArrayDOF(da,local,&l);CHKERRQ(ierr); 33*c4762a1bSJed Brown ierr = DMLocalToGlobalBegin(da,local,ADD_VALUES,global);CHKERRQ(ierr); 34*c4762a1bSJed Brown ierr = DMLocalToGlobalEnd(da,local,ADD_VALUES,global);CHKERRQ(ierr); 35*c4762a1bSJed Brown 36*c4762a1bSJed Brown ierr = VecView(global,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 37*c4762a1bSJed Brown 38*c4762a1bSJed Brown /* Free memory */ 39*c4762a1bSJed Brown ierr = VecDestroy(&local);CHKERRQ(ierr); 40*c4762a1bSJed Brown ierr = VecDestroy(&global);CHKERRQ(ierr); 41*c4762a1bSJed Brown ierr = DMDestroy(&da);CHKERRQ(ierr); 42*c4762a1bSJed Brown ierr = PetscFinalize(); 43*c4762a1bSJed Brown return ierr; 44*c4762a1bSJed Brown } 45*c4762a1bSJed Brown 46*c4762a1bSJed Brown 47*c4762a1bSJed Brown 48*c4762a1bSJed Brown /*TEST 49*c4762a1bSJed Brown 50*c4762a1bSJed Brown test: 51*c4762a1bSJed Brown filter: grep -v -i Process 52*c4762a1bSJed Brown output_file: output/ex24_1.out 53*c4762a1bSJed Brown 54*c4762a1bSJed Brown test: 55*c4762a1bSJed Brown suffix: 2 56*c4762a1bSJed Brown nsize: 2 57*c4762a1bSJed Brown filter: grep -v -i Process 58*c4762a1bSJed Brown output_file: output/ex24_2.out 59*c4762a1bSJed Brown 60*c4762a1bSJed Brown test: 61*c4762a1bSJed Brown suffix: 3 62*c4762a1bSJed Brown nsize: 3 63*c4762a1bSJed Brown filter: grep -v -i Process 64*c4762a1bSJed Brown output_file: output/ex24_2.out 65*c4762a1bSJed Brown 66*c4762a1bSJed Brown test: 67*c4762a1bSJed Brown suffix: 4 68*c4762a1bSJed Brown nsize: 4 69*c4762a1bSJed Brown filter: grep -v -i Process 70*c4762a1bSJed Brown output_file: output/ex24_2.out 71*c4762a1bSJed Brown 72*c4762a1bSJed Brown test: 73*c4762a1bSJed Brown suffix: 5 74*c4762a1bSJed Brown nsize: 5 75*c4762a1bSJed Brown filter: grep -v -i Process 76*c4762a1bSJed Brown output_file: output/ex24_2.out 77*c4762a1bSJed Brown 78*c4762a1bSJed Brown test: 79*c4762a1bSJed Brown suffix: 6 80*c4762a1bSJed Brown nsize: 6 81*c4762a1bSJed Brown filter: grep -v -i Process 82*c4762a1bSJed Brown output_file: output/ex24_2.out 83*c4762a1bSJed Brown 84*c4762a1bSJed Brown TEST*/ 85