Bug 13689 fixed: There were errors in reading enum from hdf5 files
[scilab.git] / scilab / modules / hdf5 / src / cpp / H5EnumData.hxx
index 7be2c18..e510f96 100644 (file)
 #ifndef __H5ENUMDATA_HXX__
 #define __H5ENUMDATA_HXX__
 
+#include <map>
+#include <string>
+#include <vector>
+
 #include "H5Data.hxx"
 
 namespace org_modules_hdf5
 {
 
-class H5EnumData : public H5BasicData<unsigned int>
+template<typename T>
+class H5EnumData : public H5BasicData<T>
 {
-
-    const int nmembers;
+    typedef std::map<T, std::string> MapType;
+    const unsigned int nmembers;
     const std::string * names;
+    MapType map;
 
 public:
 
-    H5EnumData(H5Object & _parent, const hsize_t _totalSize, const hsize_t _dataSize, const hsize_t _ndims, const hsize_t * _dims, unsigned int * _data, const int _nmembers, const std::string * _names, const hsize_t _stride, const size_t _offset, const bool _dataOwner) : H5BasicData<unsigned int>(_parent, _totalSize, _dataSize, _ndims, _dims, _data, _stride, _offset, _dataOwner), nmembers(_nmembers), names(_names)
+    H5EnumData(H5Object & _parent, const hsize_t _totalSize, const hsize_t _dataSize, const hsize_t _ndims, const hsize_t * _dims, T * _data, hid_t enumType, hid_t nativeType, const unsigned int _nmembers, const std::string * _names, const hsize_t _stride, const size_t _offset, const bool _dataOwner) : H5BasicData<T>(_parent, _totalSize, _dataSize, _ndims, _dims, _data, _stride, _offset, _dataOwner), nmembers(_nmembers), names(_names)
     {
-
+        hid_t super = H5Tget_super(enumType);
+        for (unsigned long long i = 0; i < nmembers; ++i)
+        {
+            T value = 0;
+            H5Tget_member_value(enumType, i, &value);
+            H5Tconvert(super, nativeType, 1, &value, 0, H5P_DEFAULT);
+            map.insert(std::pair<T, std::string>(value, names[i]));
+        }
     }
 
     virtual ~H5EnumData()
@@ -38,6 +51,12 @@ public:
             delete[] names;
         }
     }
+
+    virtual void printData(std::ostream & os, const unsigned int pos, const unsigned int indentLevel) const
+    {
+        typename MapType::const_iterator i = map.find(static_cast<T *>(H5BasicData<T>::getData())[pos]);
+        os << i->second;
+    }
 };
 }