1 static char help[] = "Demonstrates HDF5 vector input/ouput\n\n"; 2 3 #include <petscsys.h> 4 #include <petscdm.h> 5 #include <petscdmda.h> 6 #include <petscviewerhdf5.h> 7 8 int main(int argc,char **argv) 9 { 10 PetscErrorCode ierr; 11 PetscViewer viewer; 12 DM da; 13 Vec global,local,global2; 14 PetscMPIInt rank; 15 PetscBool flg; 16 PetscInt ndof; 17 18 /* 19 Every PETSc routine should begin with the PetscInitialize() routine. 20 argc, argv - These command line arguments are taken to extract the options 21 supplied to PETSc and options supplied to MPI. 22 help - When PETSc executable is invoked with the option -help, 23 it prints the various options that can be applied at 24 runtime. The user can use the "help" variable place 25 additional help messages in this printout. 26 */ 27 PetscCall(PetscInitialize(&argc,&argv,(char*)0,help)); 28 /* Get number of DOF's from command line */ 29 ierr = PetscOptionsBegin(PETSC_COMM_WORLD,NULL,"DMDA VecView/VecLoad example","");PetscCall(ierr); 30 { 31 ndof = 1; 32 PetscOptionsBoundedInt("-ndof","Number of DOF's in DMDA","",ndof,&ndof,NULL,1); 33 } 34 ierr = PetscOptionsEnd();PetscCall(ierr); 35 36 /* Create a DMDA and an associated vector */ 37 PetscCall(DMDACreate2d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE,DMDA_STENCIL_BOX,100,90,PETSC_DECIDE,PETSC_DECIDE,ndof,1,NULL,NULL,&da)); 38 PetscCall(DMSetFromOptions(da)); 39 PetscCall(DMSetUp(da)); 40 PetscCall(DMCreateGlobalVector(da,&global)); 41 PetscCall(DMCreateLocalVector(da,&local)); 42 PetscCall(VecSet(global,-1.0)); 43 PetscCall(DMGlobalToLocalBegin(da,global,INSERT_VALUES,local)); 44 PetscCall(DMGlobalToLocalEnd(da,global,INSERT_VALUES,local)); 45 PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank)); 46 PetscCall(VecScale(local,rank+1)); 47 PetscCall(DMLocalToGlobalBegin(da,local,ADD_VALUES,global)); 48 PetscCall(DMLocalToGlobalEnd(da,local,ADD_VALUES,global)); 49 50 /* Create the HDF5 viewer for writing */ 51 PetscCall(PetscViewerHDF5Open(PETSC_COMM_WORLD,"hdf5output.h5",FILE_MODE_WRITE,&viewer)); 52 PetscCall(PetscViewerSetFromOptions(viewer)); 53 54 /* Write the Vec without one extra dimension for BS */ 55 PetscCall(PetscViewerHDF5SetBaseDimension2(viewer, PETSC_FALSE)); 56 PetscCall(PetscObjectSetName((PetscObject) global, "noBsDim")); 57 PetscCall(VecView(global,viewer)); 58 59 /* Write the Vec with one extra, 1-sized, dimension for BS */ 60 PetscCall(PetscViewerHDF5SetBaseDimension2(viewer, PETSC_TRUE)); 61 PetscCall(PetscObjectSetName((PetscObject) global, "bsDim")); 62 PetscCall(VecView(global,viewer)); 63 64 PetscCall(PetscViewerDestroy(&viewer)); 65 PetscCallMPI(MPI_Barrier(PETSC_COMM_WORLD)); 66 PetscCall(VecDuplicate(global,&global2)); 67 68 /* Create the HDF5 viewer for reading */ 69 PetscCall(PetscViewerHDF5Open(PETSC_COMM_WORLD,"hdf5output.h5",FILE_MODE_READ,&viewer)); 70 PetscCall(PetscViewerSetFromOptions(viewer)); 71 72 /* Load the Vec without the BS dim and compare */ 73 PetscCall(PetscObjectSetName((PetscObject) global2, "noBsDim")); 74 PetscCall(VecLoad(global2,viewer)); 75 76 PetscCall(VecEqual(global,global2,&flg)); 77 if (!flg) { 78 PetscCall(PetscPrintf(PETSC_COMM_WORLD,"Error: Vectors are not equal\n")); 79 } 80 81 /* Load the Vec with one extra, 1-sized, BS dim and compare */ 82 PetscCall(PetscObjectSetName((PetscObject) global2, "bsDim")); 83 PetscCall(VecLoad(global2,viewer)); 84 PetscCall(PetscViewerDestroy(&viewer)); 85 86 PetscCall(VecEqual(global,global2,&flg)); 87 if (!flg) { 88 PetscCall(PetscPrintf(PETSC_COMM_WORLD,"Error: Vectors are not equal\n")); 89 } 90 91 /* clean up and exit */ 92 PetscCall(VecDestroy(&local)); 93 PetscCall(VecDestroy(&global)); 94 PetscCall(VecDestroy(&global2)); 95 PetscCall(DMDestroy(&da)); 96 97 PetscCall(PetscFinalize()); 98 return 0; 99 } 100 101 /*TEST 102 103 build: 104 requires: hdf5 105 106 test: 107 nsize: 4 108 109 test: 110 nsize: 4 111 suffix: 2 112 args: -ndof 2 113 output_file: output/ex9_1.out 114 115 TEST*/ 116