hdf5/ImageIO.hpp

Go to the documentation of this file.

#pragma once
 
#ifndef LVR2_IO_HDF5_IMAGEIO_HPP
#define LVR2_IO_HDF5_IMAGEIO_HPP
 
#include <boost/optional.hpp>
#include <opencv2/core.hpp>
#include <opencv2/core/traits.hpp>
#include <highfive/H5DataSet.hpp>
#include <highfive/H5DataSpace.hpp>
#include <highfive/H5File.hpp>
 
namespace lvr2 {
 
namespace hdf5features {
 
template<typename Derived>
class ImageIO {
public:
 
    void save(std::string groupName,
        std::string datasetName,
        const cv::Mat& img
    );
 
    void save(HighFive::Group& group,
        std::string datasetName,
        const cv::Mat& img
    );
 
    boost::optional<cv::Mat> load(HighFive::Group& group,
        std::string datasetName);
 
    boost::optional<cv::Mat> load(std::string groupName,
        std::string datasetName);
 
    boost::optional<cv::Mat> loadImage(std::string groupName,
        std::string datasetName);
 
protected:
 
    template<typename T>
    cv::Mat createMat(std::vector<size_t>& dims);
 
    Derived* m_file_access = static_cast<Derived*>(this);
 
};
 
} // namespace hdf5features
 
} // namespace lvr2
 
#include "ImageIO.tcc"
 
#endif // LVR2_IO_HDF5_IMAGEIO_HPP