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