hdf5_utils.hpp

Go to the documentation of this file.

#ifndef HDF5_UTILS_HPP
#define HDF5_UTILS_HPP 1

#include <string>
#include <vector>
#include <H5Cpp.h>

using std::string;
using std::vector;
using std::pair;
using H5::Group;
using H5::H5File;
using H5::PredType;
using H5::DataSpace;
using H5::DSetCreatPropList;
using H5::DataSet;
using H5::DataType;

template<class T> void write_std_vector_to_hdf5
(const Group& file,
 const vector<T>& data,
 const string& caption,
 const DataType& dt)
{
  hsize_t dimsf[1];
  dimsf[0] = static_cast<hsize_t>(data.size());
  DataSpace dataspace(1, dimsf);

  DSetCreatPropList plist;
  if(dimsf[0]>100000)
    dimsf[0] = 100000;
  plist.setChunk(1,dimsf);
  plist.setDeflate(6);

  DataSet dataset = file.createDataSet
    (H5std_string(caption),
     dt,
     dataspace,
     plist);
  dataset.write(&data[0],dt);
}

void write_std_vector_to_hdf5
(const Group& file,
 const vector<double>& data,
 const string& caption);

void write_std_vector_to_hdf5
(const Group& file,
 const vector<int>& data,
 const string& caption);

class HDF5Shortcut
{
public:

  explicit HDF5Shortcut(const string& fname);

  HDF5Shortcut& operator()(const string& field_name,
                           const vector<double>& data_set);
  
  HDF5Shortcut& operator()(const string& field_name,
                           const vector<int>& data_set);

  ~HDF5Shortcut(void);

private:
  const string fname_;
  vector<pair<string,vector<double> > > double_data_;
  vector<pair<string,vector<int> > > int_data_;
};

#endif // HDF5_UTILS_HPP