From 3e0123356820c00bb197db02fbabe7399364cad8 Mon Sep 17 00:00:00 2001
From: Edward 'Ned' Coltman <edward.coltman@iws.uni-stuttgart.de>
Date: Tue, 29 Sep 2020 22:43:30 +0200
Subject: [PATCH] [vtk][io] Add function to query if a data array exists
---
dumux/io/vtk/vtkreader.hh | 21 +++++++++++++++++++++
1 file changed, 21 insertions(+)
diff --git a/dumux/io/vtk/vtkreader.hh b/dumux/io/vtk/vtkreader.hh
index a69daab58a..ea65e39f3e 100644
--- a/dumux/io/vtk/vtkreader.hh
+++ b/dumux/io/vtk/vtkreader.hh
@@ -76,6 +76,27 @@ public:
DUNE_THROW(Dune::IOError, "Couldn't get 'Piece' node in " << fileName_ << ".");
}
+ /*!
+ * \brief Reviews data from the vtk file to check if there is a data array with a specified name
+ * \param name the name attribute of the data array to read
+ * \param type the data array type
+ */
+ bool hasData(const std::string& name, const DataType& type) const
+ {
+ using namespace tinyxml2;
+
+ const XMLElement* pieceNode = getPieceNode_();
+ const XMLElement* dataNode = getDataNode_(pieceNode, type);
+ if (dataNode == nullptr)
+ return false;
+
+ const XMLElement* dataArray = findDataArray_(dataNode, name);
+ if (dataArray == nullptr)
+ return false;
+
+ return true;
+ }
+
/*!
* \brief read data from the vtk file to a container, e.g. std::vector<double>
* \tparam Container a container type that has begin(), end(), push_back(), e.g. std::vector<>
--
GitLab