From ac72aa576010b3861604fcc70a93abf60d963d54 Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 17 Jan 2018 14:30:29 +0100
Subject: [PATCH] Add dataset type resolution to helper

---
 src/h5_utils.cpp | 38 ++++++++++++++++++++++++++++++++++++--
 src/h5_utils.hpp |  1 +
 2 files changed, 37 insertions(+), 2 deletions(-)

diff --git a/src/h5_utils.cpp b/src/h5_utils.cpp
index 1e59cbf..7762146 100644
--- a/src/h5_utils.cpp
+++ b/src/h5_utils.cpp
@@ -4,6 +4,7 @@
 #include <iostream>
 
 #include "h5_utils.hpp"
+#include "config.hpp"
 
 using namespace std;
 
@@ -16,7 +17,7 @@ hsize_t h5_utils::expand_dataset(const H5::DataSet& dataset, hsize_t frame_index
     dataset_dimension[0] = frame_index + dataset_increase_step;
 
     #ifdef DEBUG_OUTPUT
-        cout << "[expand_dataset] Expanding dataspace to size (";
+        cout << "[h5_utils::expand_dataset] Expanding dataspace to size (";
         for (hsize_t i=0; i<dataset_rank; ++i) {
             cout << dataset_dimension[i] << ",";
         }
@@ -37,7 +38,7 @@ void h5_utils::compact_dataset(const H5::DataSet& dataset, hsize_t max_frame_ind
     dataset_dimension[0] = max_frame_index + 1;
 
     #ifdef DEBUG_OUTPUT
-        cout << "[compact_dataset] Compacting dataspace to size (";
+        cout << "[h5_utils::compact_dataset] Compacting dataspace to size (";
         for (hsize_t i=0; i<dataset_rank; ++i) {
             cout << dataset_dimension[i] << ",";
         }
@@ -72,6 +73,39 @@ boost::any h5_utils::get_value_from_reference(string& dataset_name, boost::any v
     }
 }
 
+H5::PredType h5_utils::get_dataset_data_type(string& type){
+
+    #ifdef DEBUG_OUTPUT
+        cout << "[h5_utils::get_dataset_data_type] Getting dataset type for received frame type " << type << endl;
+    #endif
+
+    if (type == "uint8") {
+        return H5::PredType::NATIVE_UINT8;
+
+    } else if (type == "uint16") {
+        return H5::PredType::NATIVE_UINT16;
+
+    } else if (type == "uint32") {
+        return H5::PredType::NATIVE_UINT32;
+
+    }if (type == "int8") {
+        return H5::PredType::NATIVE_INT8;
+        
+    } else if (type == "int16") {
+        return H5::PredType::NATIVE_INT16;
+
+    } else if (type == "int32") {
+        return H5::PredType::NATIVE_INT32;
+
+    } else {
+        // We cannot really convert this attribute.
+        stringstream error_message;
+        error_message << "Unsupported array data_type " << type << endl;
+
+        throw runtime_error(error_message.str());
+    }
+}
+
 H5::DataSet h5_utils::write_dataset(H5::Group& target, h5_dataset& dataset, map<string, boost::any>& values){
     string name = dataset.name;
     boost::any value;
diff --git a/src/h5_utils.hpp b/src/h5_utils.hpp
index 53ff599..c5d89a5 100644
--- a/src/h5_utils.hpp
+++ b/src/h5_utils.hpp
@@ -72,6 +72,7 @@ namespace h5_utils{
     void compact_dataset(const H5::DataSet& dataset, hsize_t max_frame_index);
 
     H5::Group create_group(H5::CommonFG& target, std::string name);
+    H5::PredType get_dataset_data_type(std::string& type);
 
     H5::DataSet write_dataset(H5::Group& target, h5_dataset& dataset, std::map<std::string, boost::any>& values);
     H5::DataSet write_dataset(H5::Group& target, std::string name, double value);