1 2 static char help[] = "Tests VecView()/VecLoad() for DMDA vectors (this tests DMDAGlobalToNatural()).\n\n"; 3 4 #include <petscdm.h> 5 #include <petscdmda.h> 6 #include <petscviewerhdf5.h> 7 8 int main(int argc,char **argv) 9 { 10 PetscMPIInt rank,size; 11 PetscInt N = 6,M=8,P=5,dof=1; 12 PetscInt stencil_width=1,pt=0,st=0; 13 PetscErrorCode ierr; 14 PetscBool flg2,flg3,isbinary,mpiio; 15 DMBoundaryType bx = DM_BOUNDARY_NONE,by = DM_BOUNDARY_NONE,bz = DM_BOUNDARY_NONE; 16 DMDAStencilType stencil_type = DMDA_STENCIL_STAR; 17 DM da,da2; 18 Vec global1,global2; 19 PetscScalar mone = -1.0; 20 PetscReal norm; 21 PetscViewer viewer; 22 PetscRandom rdm; 23 #if defined(PETSC_HAVE_HDF5) 24 PetscBool ishdf5; 25 #endif 26 27 ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 28 ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr); 29 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr); 30 31 ierr = PetscOptionsGetInt(NULL,NULL,"-M",&M,NULL);CHKERRQ(ierr); 32 ierr = PetscOptionsGetInt(NULL,NULL,"-N",&N,NULL);CHKERRQ(ierr); 33 ierr = PetscOptionsGetInt(NULL,NULL,"-P",&P,NULL);CHKERRQ(ierr); 34 ierr = PetscOptionsGetInt(NULL,NULL,"-dof",&dof,NULL);CHKERRQ(ierr); 35 ierr = PetscOptionsGetInt(NULL,NULL,"-stencil_width",&stencil_width,NULL);CHKERRQ(ierr); 36 ierr = PetscOptionsGetInt(NULL,NULL,"-periodic",&pt,NULL);CHKERRQ(ierr); 37 if (pt == 1) bx = DM_BOUNDARY_PERIODIC; 38 if (pt == 2) by = DM_BOUNDARY_PERIODIC; 39 if (pt == 4) {bx = DM_BOUNDARY_PERIODIC; by = DM_BOUNDARY_PERIODIC;} 40 41 ierr = PetscOptionsGetInt(NULL,NULL,"-stencil_type",&st,NULL);CHKERRQ(ierr); 42 stencil_type = (DMDAStencilType) st; 43 44 ierr = PetscOptionsHasName(NULL,NULL,"-oned",&flg2);CHKERRQ(ierr); 45 ierr = PetscOptionsHasName(NULL,NULL,"-twod",&flg2);CHKERRQ(ierr); 46 ierr = PetscOptionsHasName(NULL,NULL,"-threed",&flg3);CHKERRQ(ierr); 47 48 ierr = PetscOptionsHasName(NULL,NULL,"-binary",&isbinary);CHKERRQ(ierr); 49 #if defined(PETSC_HAVE_HDF5) 50 ierr = PetscOptionsHasName(NULL,NULL,"-hdf5",&ishdf5);CHKERRQ(ierr); 51 #endif 52 ierr = PetscOptionsHasName(NULL,NULL,"-mpiio",&mpiio);CHKERRQ(ierr); 53 if (flg2) { 54 ierr = DMDACreate2d(PETSC_COMM_WORLD,bx,by,stencil_type,M,N,PETSC_DECIDE,PETSC_DECIDE,dof,stencil_width,0,0,&da);CHKERRQ(ierr); 55 } else if (flg3) { 56 ierr = DMDACreate3d(PETSC_COMM_WORLD,bx,by,bz,stencil_type,M,N,P,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,dof,stencil_width,0,0,0,&da);CHKERRQ(ierr); 57 } else { 58 ierr = DMDACreate1d(PETSC_COMM_WORLD,bx,M,dof,stencil_width,NULL,&da);CHKERRQ(ierr); 59 } 60 ierr = DMSetFromOptions(da);CHKERRQ(ierr); 61 ierr = DMSetUp(da);CHKERRQ(ierr); 62 63 ierr = DMCreateGlobalVector(da,&global1);CHKERRQ(ierr); 64 ierr = PetscObjectSetName((PetscObject)global1,"Test_Vec");CHKERRQ(ierr); 65 ierr = PetscRandomCreate(PETSC_COMM_WORLD,&rdm);CHKERRQ(ierr); 66 ierr = PetscRandomSetFromOptions(rdm);CHKERRQ(ierr); 67 ierr = VecSetRandom(global1,rdm);CHKERRQ(ierr); 68 if (isbinary) { 69 if (mpiio) { 70 ierr = PetscOptionsSetValue(NULL,"-viewer_binary_mpiio","");CHKERRQ(ierr); 71 } 72 ierr = PetscViewerBinaryOpen(PETSC_COMM_WORLD,"temp",FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 73 #if defined(PETSC_HAVE_HDF5) 74 } else if (ishdf5) { 75 ierr = PetscViewerHDF5Open(PETSC_COMM_WORLD,"temp",FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 76 #endif 77 } else SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"Invalid Viewer : Run with -binary or -hdf5 option\n"); 78 ierr = VecView(global1,viewer);CHKERRQ(ierr); 79 ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 80 81 ierr = PetscPrintf(PETSC_COMM_WORLD,"Global vector written to temp file is \n");CHKERRQ(ierr); 82 ierr = VecView(global1,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 83 84 if (flg2) { 85 ierr = DMDACreate2d(PETSC_COMM_WORLD,bx,by,stencil_type,M,N,PETSC_DECIDE,PETSC_DECIDE,dof,stencil_width,0,0,&da2);CHKERRQ(ierr); 86 } else if (flg3) { 87 ierr = DMDACreate3d(PETSC_COMM_WORLD,bx,by,bz,stencil_type,M,N,P,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,dof,stencil_width,0,0,0,&da2);CHKERRQ(ierr); 88 } else { 89 ierr = DMDACreate1d(PETSC_COMM_WORLD,bx,M,dof,stencil_width,NULL,&da2);CHKERRQ(ierr); 90 } 91 ierr = DMSetFromOptions(da2);CHKERRQ(ierr); 92 ierr = DMSetUp(da2);CHKERRQ(ierr); 93 94 if (isbinary) { 95 ierr = PetscViewerBinaryOpen(PETSC_COMM_WORLD,"temp",FILE_MODE_READ,&viewer);CHKERRQ(ierr); 96 #if defined(PETSC_HAVE_HDF5) 97 } else if (ishdf5) { 98 ierr = PetscViewerHDF5Open(PETSC_COMM_WORLD,"temp",FILE_MODE_READ,&viewer);CHKERRQ(ierr); 99 #endif 100 } else SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"Invalid Viewer : Run with -binary or -hdf5 option\n"); 101 102 ierr = DMCreateGlobalVector(da2,&global2);CHKERRQ(ierr); 103 ierr = PetscObjectSetName((PetscObject)global2,"Test_Vec");CHKERRQ(ierr); 104 ierr = VecLoad(global2,viewer);CHKERRQ(ierr); 105 ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 106 107 ierr = PetscPrintf(PETSC_COMM_WORLD,"Global vector read from temp file is \n");CHKERRQ(ierr); 108 ierr = VecView(global2,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 109 ierr = VecAXPY(global2,mone,global1);CHKERRQ(ierr); 110 ierr = VecNorm(global2,NORM_MAX,&norm);CHKERRQ(ierr); 111 if (norm != 0.0) { 112 ierr = PetscPrintf(PETSC_COMM_WORLD,"ex23: Norm of difference %g should be zero\n",(double)norm);CHKERRQ(ierr); 113 ierr = PetscPrintf(PETSC_COMM_WORLD," Number of processors %d\n",size);CHKERRQ(ierr); 114 ierr = PetscPrintf(PETSC_COMM_WORLD," M,N,P,dof %D %D %D %D\n",M,N,P,dof);CHKERRQ(ierr); 115 ierr = PetscPrintf(PETSC_COMM_WORLD," stencil_width %D stencil_type %d periodic %d\n",stencil_width,(int)stencil_type,(int)pt);CHKERRQ(ierr); 116 ierr = PetscPrintf(PETSC_COMM_WORLD," dimension %d\n",1 + (int) flg2 + (int) flg3);CHKERRQ(ierr); 117 } 118 119 ierr = PetscRandomDestroy(&rdm);CHKERRQ(ierr); 120 ierr = DMDestroy(&da);CHKERRQ(ierr); 121 ierr = DMDestroy(&da2);CHKERRQ(ierr); 122 ierr = VecDestroy(&global1);CHKERRQ(ierr); 123 ierr = VecDestroy(&global2);CHKERRQ(ierr); 124 ierr = PetscFinalize(); 125 return ierr; 126 } 127