diff --git a/GenericIO.cxx b/GenericIO.cxx index ac00f34cead32e317ad408ae62df1303abdcbf39..091a9bef87e7fe2d1255589406e4610105376d5b 100644 --- a/GenericIO.cxx +++ b/GenericIO.cxx @@ -1863,6 +1863,17 @@ void GenericIO::getVariableInfo(vector<VariableInfo> &VI) { } } +bool GenericIO::hasVariable(const std::string &Name) { + vector<VariableInfo> VI; + getVariableInfo(VI); + + for (size_t i = 0; i < VI.size(); ++i) + if (VI[i].Name == Name) + return true; + + return false; +} + void GenericIO::setNaturalDefaultPartition() { #ifdef __bgq__ DefaultPartition = MPIX_IO_link_id(); diff --git a/GenericIO.h b/GenericIO.h index 02bf68563c8d257d17ff35cc9e95ea49c46955d5..2b3df11441fd50c63b296bc4428e61276fa9814e 100644 --- a/GenericIO.h +++ b/GenericIO.h @@ -434,6 +434,8 @@ public: void getVariableInfo(std::vector<VariableInfo> &VI); + bool hasVariable(const std::string &Name); + std::size_t readNumElems(int EffRank = -1); void readCoords(int Coords[3], int EffRank = -1); int readGlobalRankNumber(int EffRank = -1); diff --git a/python/genericio.cpp b/python/genericio.cpp index a29d478f35d38ecbcf05d88bb8237410cc05dcaf..1ee9b54536bc837cc0898787e78521a65c24ad81 100644 --- a/python/genericio.cpp +++ b/python/genericio.cpp @@ -377,6 +377,7 @@ PYBIND11_MODULE(pygio, m) { py::arg("eff_rank")=-1) .def("read_nranks", (int (PyGenericIO::*)(void))(&PyGenericIO::readNRanks)) .def("read_variable_names", &PyGenericIO::read_variable_names) + .def("has_variable", &PyGenericIO::hasVariable) .def("read_variable_dtypes", &PyGenericIO::read_variable_dtypes) .def("get_source_ranks", &PyGenericIO::getSourceRanks) .def_static("setDefaultShouldCompress", &PyGenericIO::setDefaultShouldCompress)