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