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 CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank)); 29 CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size)); 30 31 CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-M",&M,NULL)); 32 CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-N",&N,NULL)); 33 CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-P",&P,NULL)); 34 CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-dof",&dof,NULL)); 35 CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-stencil_width",&stencil_width,NULL)); 36 CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-periodic",&pt,NULL)); 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 CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-stencil_type",&st,NULL)); 42 stencil_type = (DMDAStencilType) st; 43 44 CHKERRQ(PetscOptionsHasName(NULL,NULL,"-oned",&flg2)); 45 CHKERRQ(PetscOptionsHasName(NULL,NULL,"-twod",&flg2)); 46 CHKERRQ(PetscOptionsHasName(NULL,NULL,"-threed",&flg3)); 47 48 CHKERRQ(PetscOptionsHasName(NULL,NULL,"-binary",&isbinary)); 49 #if defined(PETSC_HAVE_HDF5) 50 CHKERRQ(PetscOptionsHasName(NULL,NULL,"-hdf5",&ishdf5)); 51 #endif 52 CHKERRQ(PetscOptionsHasName(NULL,NULL,"-mpiio",&mpiio)); 53 if (flg2) { 54 CHKERRQ(DMDACreate2d(PETSC_COMM_WORLD,bx,by,stencil_type,M,N,PETSC_DECIDE,PETSC_DECIDE,dof,stencil_width,0,0,&da)); 55 } else if (flg3) { 56 CHKERRQ(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)); 57 } else { 58 CHKERRQ(DMDACreate1d(PETSC_COMM_WORLD,bx,M,dof,stencil_width,NULL,&da)); 59 } 60 CHKERRQ(DMSetFromOptions(da)); 61 CHKERRQ(DMSetUp(da)); 62 63 CHKERRQ(DMCreateGlobalVector(da,&global1)); 64 CHKERRQ(PetscObjectSetName((PetscObject)global1,"Test_Vec")); 65 CHKERRQ(PetscRandomCreate(PETSC_COMM_WORLD,&rdm)); 66 CHKERRQ(PetscRandomSetFromOptions(rdm)); 67 CHKERRQ(VecSetRandom(global1,rdm)); 68 if (isbinary) { 69 if (mpiio) { 70 CHKERRQ(PetscOptionsSetValue(NULL,"-viewer_binary_mpiio","")); 71 } 72 CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,"temp",FILE_MODE_WRITE,&viewer)); 73 #if defined(PETSC_HAVE_HDF5) 74 } else if (ishdf5) { 75 CHKERRQ(PetscViewerHDF5Open(PETSC_COMM_WORLD,"temp",FILE_MODE_WRITE,&viewer)); 76 #endif 77 } else SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"Invalid Viewer : Run with -binary or -hdf5 option"); 78 CHKERRQ(VecView(global1,viewer)); 79 CHKERRQ(PetscViewerDestroy(&viewer)); 80 81 CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"Global vector written to temp file is \n")); 82 CHKERRQ(VecView(global1,PETSC_VIEWER_STDOUT_WORLD)); 83 84 if (flg2) { 85 CHKERRQ(DMDACreate2d(PETSC_COMM_WORLD,bx,by,stencil_type,M,N,PETSC_DECIDE,PETSC_DECIDE,dof,stencil_width,0,0,&da2)); 86 } else if (flg3) { 87 CHKERRQ(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)); 88 } else { 89 CHKERRQ(DMDACreate1d(PETSC_COMM_WORLD,bx,M,dof,stencil_width,NULL,&da2)); 90 } 91 CHKERRQ(DMSetFromOptions(da2)); 92 CHKERRQ(DMSetUp(da2)); 93 94 if (isbinary) { 95 CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,"temp",FILE_MODE_READ,&viewer)); 96 #if defined(PETSC_HAVE_HDF5) 97 } else if (ishdf5) { 98 CHKERRQ(PetscViewerHDF5Open(PETSC_COMM_WORLD,"temp",FILE_MODE_READ,&viewer)); 99 #endif 100 } else SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"Invalid Viewer : Run with -binary or -hdf5 option"); 101 102 CHKERRQ(DMCreateGlobalVector(da2,&global2)); 103 CHKERRQ(PetscObjectSetName((PetscObject)global2,"Test_Vec")); 104 CHKERRQ(VecLoad(global2,viewer)); 105 CHKERRQ(PetscViewerDestroy(&viewer)); 106 107 CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"Global vector read from temp file is \n")); 108 CHKERRQ(VecView(global2,PETSC_VIEWER_STDOUT_WORLD)); 109 CHKERRQ(VecAXPY(global2,mone,global1)); 110 CHKERRQ(VecNorm(global2,NORM_MAX,&norm)); 111 if (norm != 0.0) { 112 CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"ex23: Norm of difference %g should be zero\n",(double)norm)); 113 CHKERRQ(PetscPrintf(PETSC_COMM_WORLD," Number of processors %d\n",size)); 114 CHKERRQ(PetscPrintf(PETSC_COMM_WORLD," M,N,P,dof %D %D %D %D\n",M,N,P,dof)); 115 CHKERRQ(PetscPrintf(PETSC_COMM_WORLD," stencil_width %D stencil_type %d periodic %d\n",stencil_width,(int)stencil_type,(int)pt)); 116 CHKERRQ(PetscPrintf(PETSC_COMM_WORLD," dimension %d\n",1 + (int) flg2 + (int) flg3)); 117 } 118 119 CHKERRQ(PetscRandomDestroy(&rdm)); 120 CHKERRQ(DMDestroy(&da)); 121 CHKERRQ(DMDestroy(&da2)); 122 CHKERRQ(VecDestroy(&global1)); 123 CHKERRQ(VecDestroy(&global2)); 124 ierr = PetscFinalize(); 125 return ierr; 126 } 127