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_ */