Materializer.hpp

Go to the documentation of this file.

 
/*
* Materializer.hpp
*
*  @date 17.07.2017
*  @author Jan Philipp Vogtherr <jvogtherr@uni-osnabrueck.de>
*  @author Kristin Schmidt <krschmidt@uni-osnabrueck.de>
*/
 
#ifndef LVR2_ALGORITHM_MATERIALIZER_H_
#define LVR2_ALGORITHM_MATERIALIZER_H_
 
#include <boost/shared_ptr.hpp>
#include <boost/smart_ptr/make_shared.hpp>
#include <boost/optional.hpp>
 
#include "lvr2/geometry/BaseMesh.hpp"
#include "lvr2/geometry/BaseVector.hpp"
#include "lvr2/geometry/Handles.hpp"
#include "lvr2/geometry/Normal.hpp"
#include "lvr2/reconstruction/PointsetSurface.hpp"
#include "lvr2/texture/ClusterTexCoordMapping.hpp"
#include "lvr2/texture/Texture.hpp"
#include "lvr2/util/ClusterBiMap.hpp"
#include "lvr2/texture/Material.hpp"
#include "lvr2/geometry/BoundingRectangle.hpp"
#include "lvr2/algorithm/Texturizer.hpp"
#include "lvr2/algorithm/ColorAlgorithms.hpp"
 
 
#include "lvr2/io/Progress.hpp"
#include "lvr2/io/Timestamp.hpp"
#include <unordered_map>
#include <unordered_set>
 
namespace lvr2
{
 
template<typename BaseVecT>
struct MaterializerResult
{
    DenseClusterMap<Material> m_clusterMaterials;
 
    boost::optional<StableVector<TextureHandle, Texture>> m_textures;
 
    boost::optional<SparseVertexMap<ClusterTexCoordMapping>> m_vertexTexCoords;
 
    boost::optional<std::unordered_map<BaseVecT, std::vector<float>>> m_keypoints;
 
    MaterializerResult(
        DenseClusterMap<Material> clusterMaterials
    )   :
        m_clusterMaterials(clusterMaterials)
    {
    }
 
    MaterializerResult(
        DenseClusterMap<Material> clusterMaterials,
        StableVector<TextureHandle, Texture> textures,
        SparseVertexMap<ClusterTexCoordMapping> vertexTexCoords,
        std::unordered_map<BaseVecT, std::vector<float>> keypoints
    ) :
        m_clusterMaterials(clusterMaterials),
        m_textures(textures),
        m_vertexTexCoords(vertexTexCoords),
        m_keypoints(keypoints)
    {
    }
 
};
 
template<typename BaseVecT>
class Materializer
{
 
public:
 
    Materializer(
        const BaseMesh<BaseVecT>& mesh,
        const ClusterBiMap<FaceHandle>& cluster,
        const FaceMap<Normal<typename BaseVecT::CoordType>>& normals,
        const PointsetSurface<BaseVecT>& surface
    );
 
    void setTexturizer(Texturizer<BaseVecT>& texturizer);
 
    MaterializerResult<BaseVecT> generateMaterials();
 
    void saveTextures();
 
private:
 
    const BaseMesh<BaseVecT>& m_mesh;
    const ClusterBiMap<FaceHandle>& m_cluster;
    const FaceMap<Normal<typename BaseVecT::CoordType>>& m_normals;
    const PointsetSurface<BaseVecT>& m_surface;
 
    boost::optional<Texturizer<BaseVecT>&> m_texturizer;
 
};
 
} // namespace lvr2
 
#include "lvr2/algorithm/Materializer.tcc"
 
#endif /* LVR2_ALGORITHM_MATERIALIZER_H_ */