BigVolumen.hpp

Go to the documentation of this file.

 
/*
 * BigVolumen.hpp
 *
 *  Created on: Jul 17, 2017
 *      Author: Isaak Mitschke
 */
 
#ifndef LAS_VEGAS_BigVolumen_HPP
#define LAS_VEGAS_BigVolumen_HPP
 
#include "lvr2/geometry/BoundingBox.hpp"
#include "lvr2/io/DataStruct.hpp"
 
#include <boost/archive/binary_iarchive.hpp>
#include <boost/archive/binary_oarchive.hpp>
#include <boost/iostreams/device/mapped_file.hpp>
#include <fstream>
#include <sstream>
#include <string>
#include <unordered_map>
#include <utility>
 
#ifndef __APPLE__
#include <omp.h>
#endif
 
namespace lvr2
{
 
template <typename BaseVecT>
class BigVolumen
{
 
  public:
    struct VolumeCellInfo
    {
        VolumeCellInfo() : size(0), overlapping_size(0) {}
        size_t size;
        size_t overlapping_size;
        size_t ix;
        size_t iy;
        size_t iz;
        std::string path;
        BoundingBox<BaseVecT> bb;
        std::ofstream ofs_points;
        std::ofstream ofs_normals;
        std::ofstream ofs_colors;
    };
 
    BigVolumen(std::vector<std::string>,
               float voxelsize,
               float overlapping_size = 0,
               float scale = 1);
 
    size_t size();
 
    size_t pointSize();
 
    BoundingBox<BaseVecT>& getBB() { return m_bb; }
 
    virtual ~BigVolumen();
 
    inline size_t hashValue(size_t i, size_t j, size_t k)
    {
        return i * m_maxIndexSquare + j * m_maxIndex + k;
    }
 
    inline bool hasColors() { return m_has_color; }
    inline bool hasNormals() { return m_has_normal; }
 
    inline std::unordered_map<size_t, VolumeCellInfo>* getCellinfo() { return &m_gridNumPoints; };
 
  private:
    inline int calcIndex(float f) { return f < 0 ? f - .5 : f + .5; }
 
    size_t m_maxIndexSquare;
    size_t m_maxIndex;
    size_t m_maxIndexX;
    size_t m_maxIndexY;
    size_t m_maxIndexZ;
    size_t m_numPoints;
    float m_voxelSize;
    bool m_extrude;
 
    bool m_has_normal;
    bool m_has_color;
 
    boost::iostreams::mapped_file m_PointFile;
    boost::iostreams::mapped_file m_NomralFile;
    boost::iostreams::mapped_file m_ColorFile;
    BoundingBox<BaseVecT> m_bb;
    std::unordered_map<size_t, VolumeCellInfo> m_gridNumPoints;
    float m_scale;
};
 
} // namespace lvr2
 
#include "lvr2/reconstruction/BigVolumen.tcc"
 
#endif // LAS_VEGAS_BigVolumen_HPP