Actual source code: ex10.c

  1: static char help[] = "Tests I/O of vectors for different data formats (binary,HDF5) and illustrates the use of user-defined event logging\n\n";

  3: #include <petscvec.h>
  4: #include <petscviewerhdf5.h>

  6: /* Note:  Most applications would not read and write a vector within
  7:   the same program.  This example is intended only to demonstrate
  8:   both input and output and is written for use with either 1,2,or 4 processors. */

 10: int main(int argc, char **args)
 11: {
 12:   PetscMPIInt rank, size;
 13:   PetscInt    i, m = 20, low, high, ldim, iglobal, lsize;
 14:   PetscScalar v;
 15:   Vec         u;
 16:   PetscViewer viewer;
 17:   PetscBool   vstage2, vstage3, mpiio_use, isbinary = PETSC_FALSE;
 18: #if defined(PETSC_HAVE_HDF5)
 19:   PetscBool ishdf5 = PETSC_FALSE;
 20: #endif
 21: #if defined(PETSC_HAVE_ADIOS)
 22:   PetscBool isadios = PETSC_FALSE;
 23: #endif
 24:   PetscScalar const *values;
 25:   PetscLogEvent      VECTOR_GENERATE, VECTOR_READ;

 27:   PetscFunctionBeginUser;
 28:   PetscCall(PetscInitialize(&argc, &args, (char *)0, help));
 29:   mpiio_use = vstage2 = vstage3 = PETSC_FALSE;

 31:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-binary", &isbinary, NULL));
 32: #if defined(PETSC_HAVE_HDF5)
 33:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-hdf5", &ishdf5, NULL));
 34: #endif
 35: #if defined(PETSC_HAVE_ADIOS)
 36:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-adios", &isadios, NULL));
 37: #endif
 38:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-mpiio", &mpiio_use, NULL));
 39:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-sizes_set", &vstage2, NULL));
 40:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-type_set", &vstage3, NULL));

 42:   PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
 43:   PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
 44:   PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL));

 46:   /* PART 1:  Generate vector, then write it in the given data format */

 48:   PetscCall(PetscLogEventRegister("Generate Vector", VEC_CLASSID, &VECTOR_GENERATE));
 49:   PetscCall(PetscLogEventBegin(VECTOR_GENERATE, 0, 0, 0, 0));
 50:   /* Generate vector */
 51:   PetscCall(VecCreate(PETSC_COMM_WORLD, &u));
 52:   PetscCall(PetscObjectSetName((PetscObject)u, "Test_Vec"));
 53:   PetscCall(VecSetSizes(u, PETSC_DECIDE, m));
 54:   PetscCall(VecSetFromOptions(u));
 55:   PetscCall(VecGetOwnershipRange(u, &low, &high));
 56:   PetscCall(VecGetLocalSize(u, &ldim));
 57:   for (i = 0; i < ldim; i++) {
 58:     iglobal = i + low;
 59:     v       = (PetscScalar)(i + low);
 60:     PetscCall(VecSetValues(u, 1, &iglobal, &v, INSERT_VALUES));
 61:   }
 62:   PetscCall(VecAssemblyBegin(u));
 63:   PetscCall(VecAssemblyEnd(u));
 64:   PetscCall(VecView(u, PETSC_VIEWER_STDOUT_WORLD));

 66:   if (isbinary) {
 67:     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "writing vector in binary to vector.dat ...\n"));
 68:     PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "vector.dat", FILE_MODE_WRITE, &viewer));
 69: #if defined(PETSC_HAVE_HDF5)
 70:   } else if (ishdf5) {
 71:     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "writing vector in hdf5 to vector.dat ...\n"));
 72:     PetscCall(PetscViewerHDF5Open(PETSC_COMM_WORLD, "vector.dat", FILE_MODE_WRITE, &viewer));
 73: #endif
 74: #if defined(PETSC_HAVE_ADIOS)
 75:   } else if (isadios) {
 76:     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "writing vector in adios to vector.dat ...\n"));
 77:     PetscCall(PetscViewerADIOSOpen(PETSC_COMM_WORLD, "vector.dat", FILE_MODE_WRITE, &viewer));
 78: #endif
 79:   } else SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_SUP, "No data format specified, run with one of -binary -hdf5 -adios options");
 80:   PetscCall(VecView(u, viewer));
 81:   PetscCall(PetscViewerDestroy(&viewer));
 82:   PetscCall(VecDestroy(&u));

 84:   PetscCall(PetscLogEventEnd(VECTOR_GENERATE, 0, 0, 0, 0));

 86:   /* PART 2:  Read in vector in binary format */

 88:   /* Read new vector in binary format */
 89:   PetscCall(PetscLogEventRegister("Read Vector", VEC_CLASSID, &VECTOR_READ));
 90:   PetscCall(PetscLogEventBegin(VECTOR_READ, 0, 0, 0, 0));
 91:   if (mpiio_use) {
 92:     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Using MPI IO for reading the vector\n"));
 93:     PetscCall(PetscOptionsSetValue(NULL, "-viewer_binary_mpiio", ""));
 94:   }
 95:   if (isbinary) {
 96:     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "reading vector in binary from vector.dat ...\n"));
 97:     PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "vector.dat", FILE_MODE_READ, &viewer));
 98:     PetscCall(PetscViewerBinarySetFlowControl(viewer, 2));
 99: #if defined(PETSC_HAVE_HDF5)
100:   } else if (ishdf5) {
101:     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "reading vector in hdf5 from vector.dat ...\n"));
102:     PetscCall(PetscViewerHDF5Open(PETSC_COMM_WORLD, "vector.dat", FILE_MODE_READ, &viewer));
103: #endif
104: #if defined(PETSC_HAVE_ADIOS)
105:   } else if (isadios) {
106:     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "reading vector in adios from vector.dat ...\n"));
107:     PetscCall(PetscViewerADIOSOpen(PETSC_COMM_WORLD, "vector.dat", FILE_MODE_READ, &viewer));
108: #endif
109:   }
110:   PetscCall(VecCreate(PETSC_COMM_WORLD, &u));
111:   PetscCall(PetscObjectSetName((PetscObject)u, "Test_Vec"));

113:   if (vstage2) {
114:     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Setting vector sizes...\n"));
115:     if (size > 1) {
116:       if (rank == 0) {
117:         lsize = m / size + size;
118:         PetscCall(VecSetSizes(u, lsize, m));
119:       } else if (rank == size - 1) {
120:         lsize = m / size - size;
121:         PetscCall(VecSetSizes(u, lsize, m));
122:       } else {
123:         lsize = m / size;
124:         PetscCall(VecSetSizes(u, lsize, m));
125:       }
126:     } else {
127:       PetscCall(VecSetSizes(u, m, m));
128:     }
129:   }

131:   if (vstage3) {
132:     PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Setting vector type...\n"));
133:     PetscCall(VecSetType(u, VECMPI));
134:   }
135:   PetscCall(VecLoad(u, viewer));
136:   PetscCall(PetscViewerDestroy(&viewer));
137:   PetscCall(PetscLogEventEnd(VECTOR_READ, 0, 0, 0, 0));
138:   PetscCall(VecView(u, PETSC_VIEWER_STDOUT_WORLD));
139:   PetscCall(VecGetArrayRead(u, &values));
140:   PetscCall(VecGetLocalSize(u, &ldim));
141:   PetscCall(VecGetOwnershipRange(u, &low, NULL));
142:   for (i = 0; i < ldim; i++) PetscCheck(values[i] == (PetscScalar)(i + low), PETSC_COMM_WORLD, PETSC_ERR_SUP, "Data check failed!");
143:   PetscCall(VecRestoreArrayRead(u, &values));

145:   /* Free data structures */
146:   PetscCall(VecDestroy(&u));
147:   PetscCall(PetscFinalize());
148:   return 0;
149: }

151: /*TEST

153:      test:
154:        nsize: 2
155:        args: -binary

157:      test:
158:        suffix: 2
159:        nsize: 3
160:        args: -binary

162:      test:
163:        suffix: 3
164:        nsize: 5
165:        args: -binary

167:      test:
168:        suffix: 4
169:        requires: hdf5
170:        nsize: 2
171:        args: -hdf5

173:      test:
174:        suffix: 5
175:        nsize: 4
176:        args: -binary -sizes_set

178:      test:
179:        suffix: 6
180:        requires: hdf5
181:        nsize: 4
182:        args: -hdf5 -sizes_set

184: TEST*/