FinalizeAlgorithms.hpp

Go to the documentation of this file.

/*
 * FinalizeAlgorithms.hpp
 *
 *  @date 13.06.2017
 *  @author Johan M. von Behren <johan@vonbehren.eu>
 */

#ifndef LVR2_ALGORITHM_FINALIZEALGORITHM_H_
#define LVR2_ALGORITHM_FINALIZEALGORITHM_H_

#include <boost/shared_ptr.hpp>
#include <boost/smart_ptr/make_shared.hpp>
#include <boost/optional.hpp>



#include "lvr2/io/MeshBuffer.hpp"
#include "lvr2/geometry/Normal.hpp"
#include "lvr2/attrmaps/AttrMaps.hpp"
#include "lvr2/algorithm/ColorAlgorithms.hpp"
#include "lvr2/util/ClusterBiMap.hpp"
#include "lvr2/texture/Texture.hpp"
#include "lvr2/texture/Material.hpp"
#include "lvr2/algorithm/ClusterPainter.hpp"

#include "lvr2/io/ObjIO.hpp"

namespace lvr2
{


// Forward declaration
template<typename BaseVecT>
class MaterializerResult;

template<typename BaseVecT>
class SimpleFinalizer
{
private:
    boost::optional<const VertexMap<Rgb8Color>&> m_colorData;
    boost::optional<const VertexMap<Normal<typename BaseVecT::CoordType>>&> m_normalData;

public:
    SimpleFinalizer() {};

    MeshBufferPtr apply(const BaseMesh<BaseVecT>& mesh);

    void setColorData(const VertexMap<Rgb8Color>& colorData);

    void setNormalData(const VertexMap<Normal<typename BaseVecT::CoordType>>& normalData);
};

template<typename BaseVecT>
class TextureFinalizer
{
public:
    TextureFinalizer(const ClusterBiMap<FaceHandle>& cluster);

    void setVertexNormals(const VertexMap<Normal<typename BaseVecT::CoordType>>& normals);

    void setClusterColors(const ClusterMap<Rgb8Color>& colors);

    void setVertexColors(const VertexMap<Rgb8Color>& vertexColors);

    void setMaterializerResult(const MaterializerResult<BaseVecT>& materializerResult);

    MeshBufferPtr apply(const BaseMesh<BaseVecT>& mesh);

private:

    // Clusters of mesh (mandatory)
    const ClusterBiMap<FaceHandle>& m_cluster;

    // Normals (optional)
    boost::optional<const VertexMap<Normal<typename BaseVecT::CoordType>>&> m_vertexNormals;

    // Basic colors
    // Cluster colors will color each vertex in the color of its corresponding cluster
    // These have lower priority when cluster colors, as only one mode can be used
    boost::optional<const ClusterMap<Rgb8Color>&> m_clusterColors;

    // Vertex colors will color each vertex individually
    // These have a higher priority than cluster colors
    boost::optional<const VertexMap<Rgb8Color>&> m_vertexColors;

    // Materials and textures
    boost::optional<const MaterializerResult<BaseVecT>&> m_materializerResult;
};

} // namespace lvr2

#include "lvr2/algorithm/FinalizeAlgorithms.tcc"

#endif /* LVR2_ALGORITHM_FINALIZEALGORITHM_H_ */