Add H5 attribute functions

spinicist · Feb 7, 2024 · 71aea0e · 71aea0e
1 parent 741cd89
commit 71aea0e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/src/io/reader.cpp b/src/io/reader.cpp
@@ -258,5 +258,23 @@ auto Reader::readMeta() const -> std::map<std::string, float>
   return meta;
 }
 
+template <>
+auto Reader::readAttribute<float>(std::string const &ds_name, std::string const &attr_name) const -> float
+{
+  float val;
+  CheckedCall(H5LTget_attribute_float(handle_, ds_name.c_str(), attr_name.c_str(), &val),
+              fmt::format("reading attribute {} from {}", attr_name, ds_name));
+  return val;
+}
+
+template <>
+auto Reader::readAttribute<int>(std::string const &ds_name, std::string const &attr_name) const -> int
+{
+  int val;
+  CheckedCall(H5LTget_attribute_int(handle_, ds_name.c_str(), attr_name.c_str(), &val),
+              fmt::format("reading attribute {} from {}", attr_name, ds_name));
+  return val;
+}
+
 } // namespace HD5
 } // namespace rl
diff --git a/src/io/reader.hpp b/src/io/reader.hpp
@@ -28,7 +28,7 @@ struct Reader
   auto readMeta() const -> std::map<std::string, float>;                         // Read meta-data group
 
   template <typename T>
-  auto readAttribute(std::string const &dataset, std::string const &attribute) -> T;
+  auto readAttribute(std::string const &dataset, std::string const &attribute) const -> T;
 
   template <typename T>
   auto readTensor(std::string const &label) const -> T;