diff --git a/GenericIO.cxx b/GenericIO.cxx index bcae435f6735aa18650b426d730e57e737c5313e..908dc8cae1544895d1c96fa07355050d17354407 100644 --- a/GenericIO.cxx +++ b/GenericIO.cxx @@ -1830,6 +1830,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 32414150b42f9c4756527df5da9815581508dd82..6c2c817c16897d08b8954208ea472b9200a4e0de 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 64bc989327285faf1fc6e315f4c06c0b05dae675..f51c3247900687bb55072c772a05a74bae566df2 100644 --- a/python/genericio.cpp +++ b/python/genericio.cpp @@ -372,6 +372,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)