hdf5/ChunkIO.hpp

Go to the documentation of this file.

#pragma once
#ifndef LVR2_IO_HDF5_CHUNKIO_HPP
#define LVR2_IO_HDF5_CHUNKIO_HPP

#include "ArrayIO.hpp"
#include "MeshIO.hpp"
#include "PointCloudIO.hpp"
#include "lvr2/geometry/BaseVector.hpp"
#include "lvr2/io/hdf5/HDF5FeatureBase.hpp"
#include "lvr2/io/Model.hpp"

namespace lvr2
{

namespace hdf5features
{

template <typename Derived, typename T>
struct IOType;

template <typename Derived>
struct IOType<Derived, MeshBufferPtr> {
    using io_type = MeshIO<Derived>;
};

template <typename Derived>
struct IOType<Derived, PointBufferPtr> {
    using io_type = PointCloudIO<Derived>;
};

template <typename Derived>
class ChunkIO
{
  public:
    void saveAmount(BaseVector<std::size_t> amount);

    void saveChunkSize(float chunkSize);

    void saveBoundingBox(BoundingBox<BaseVector<float>> boundingBox);

    void save(BaseVector<std::size_t> amount,
              float chunkSize,
              BoundingBox<BaseVector<float>> boundingBox);

    template <typename T>
    void saveChunk(T data, std::string layer, int x, int y, int z);

    BaseVector<size_t> loadAmount();

    float loadChunkSize();

    BoundingBox<BaseVector<float>> loadBoundingBox();

    template <typename T>
    T loadChunk(std::string layer, int x, int y, int z);

  protected:
    Derived* m_file_access                 = static_cast<Derived*>(this);
    ArrayIO<Derived>* m_array_io           = static_cast<ArrayIO<Derived>*>(m_file_access);

  private:
    const std::string m_chunkName       = "chunks";
    const std::string m_amountName      = "amount";
    const std::string m_chunkSizeName   = "size";
    const std::string m_boundingBoxName = "bounding_box";
};

} // namespace hdf5features

template <typename Derived>
struct Hdf5Construct<hdf5features::ChunkIO, Derived>
{
    // DEPS
    using dep1 = typename Hdf5Construct<hdf5features::ArrayIO, Derived>::type;
    using dep2 = typename Hdf5Construct<hdf5features::MeshIO, Derived>::type;
    using dep3 = typename Hdf5Construct<hdf5features::PointCloudIO, Derived>::type;
    using deps = typename dep1::template Merge<dep2>::template Merge<dep3>;

    // add actual feature
    using type = typename deps::template add_features<hdf5features::ChunkIO>::type;
};

} // namespace lvr2

#include "ChunkIO.tcc"

#endif