ClusterAlgorithms.hpp

Go to the documentation of this file.

/*
 * ClusterAlgorithms.hpp
 *
 * Collection of algorithms for cluster modification.
 *
 * @date 24.07.2017
 * @author Johan M. von Behren <johan@vonbehren.eu>
 * @author Jan Philipp Vogtherr <jvogtherr@uni-osnabrueck.de>
 * @author Kristin Schmidt <krschmidt@uni-osnabrueck.de>
 */

#ifndef LVR2_ALGORITHM_CLUSTERALGORITHMS_H_
#define LVR2_ALGORITHM_CLUSTERALGORITHMS_H_

#include "lvr2/algorithm/Materializer.hpp"
#include "lvr2/geometry/BaseMesh.hpp"
#include "lvr2/geometry/HalfEdgeMesh.hpp"
#include "lvr2/geometry/Handles.hpp"
#include "lvr2/geometry/Plane.hpp"
#include "lvr2/io/MeshBuffer.hpp"
#include "lvr2/util/Cluster.hpp"
#include "lvr2/util/ClusterBiMap.hpp"
#include "lvr2/attrmaps/AttrMaps.hpp"
#include "lvr2/geometry/Line.hpp"

#include <vector>

using std::vector;


namespace lvr2
{

template<typename BaseVecT>
void removeDanglingCluster(BaseMesh<BaseVecT>& mesh, size_t sizeThreshold);

template<typename BaseVecT>
vector<vector<VertexHandle>> findContours(
    BaseMesh<BaseVecT>& mesh,
    const ClusterBiMap<FaceHandle>& clusters,
    ClusterHandle clusterH
);

template<typename BaseVecT>
vector<VertexHandle> simplifyContour(const BaseMesh<BaseVecT>& mesh,
                                     const vector<VertexHandle>& contour,
                                     float threshold
);


template<typename BaseVecT>
vector<VertexHandle> calculateClusterContourVertices(
    ClusterHandle clusterH,
    const BaseMesh<BaseVecT>& mesh,
    const ClusterBiMap<FaceHandle>& clusterBiMap
);

template<typename BaseVecT>
BoundingRectangle<typename BaseVecT::CoordType> calculateBoundingRectangle(
    const vector<VertexHandle>& contour,
    const BaseMesh<BaseVecT>& mesh,
    const Cluster<FaceHandle>& cluster,
    const FaceMap<Normal<typename BaseVecT::CoordType>>& normals,
    float texelSize,
    ClusterHandle clusterH
);

template<typename BaseVecT, typename Pred>
ClusterBiMap<FaceHandle> clusterGrowing(const BaseMesh<BaseVecT>& mesh, Pred pred);

template<typename BaseVecT>
ClusterBiMap<FaceHandle> planarClusterGrowing(
    const BaseMesh<BaseVecT>& mesh,
    const FaceMap<Normal<typename BaseVecT::CoordType>>& normals,
    float minSinAngle
);

template<typename BaseVecT>
ClusterBiMap<FaceHandle> iterativePlanarClusterGrowing(
    BaseMesh<BaseVecT>& mesh,
    FaceMap<Normal<typename BaseVecT::CoordType>>& normals,
    float minSinAngle,
    int numIterations,
    int minClusterSize
);

template<typename BaseVecT>
ClusterBiMap<FaceHandle> iterativePlanarClusterGrowingRANSAC(
    BaseMesh<BaseVecT>& mesh,
    FaceMap<Normal<typename BaseVecT::CoordType>>& normals,
    float minSinAngle,
    int numIterations,
    int minClusterSize,
    int ransacIterations = 100,
    int ransacSamples = 10
);

template<typename BaseVecT>
Plane<BaseVecT> calcRegressionPlane(
    const BaseMesh<BaseVecT>& mesh,
    const Cluster<FaceHandle>& cluster,
    const FaceMap<Normal<typename BaseVecT::CoordType>>& normals
);

template<typename BaseVecT>
Plane<BaseVecT> calcRegressionPlaneRANSAC(
    const BaseMesh<BaseVecT>& mesh,
    const Cluster<FaceHandle>& cluster,
    const FaceMap<Normal<typename BaseVecT::CoordType>>& normals,
    const int num_iterations = 100,
    const int num_samples = 10
);

template<typename BaseVecT>
Plane<BaseVecT> calcRegressionPlanePCA(
    const BaseMesh<BaseVecT>& mesh,
    const Cluster<FaceHandle>& cluster,
    const FaceMap<Normal<typename BaseVecT::CoordType>>& normals,
    const int num_iterations = 100,
    const int num_samples = 10
);

template<typename BaseVecT>
DenseClusterMap<Plane<BaseVecT>> calcRegressionPlanes(
    const BaseMesh<BaseVecT>& mesh,
    const ClusterBiMap<FaceHandle>& clusters,
    const FaceMap<Normal<typename BaseVecT::CoordType>>& normals,
    int minClusterSize
);

template<typename BaseVecT>
DenseClusterMap<Plane<BaseVecT>> calcRegressionPlanesRANSAC(
    const BaseMesh<BaseVecT>& mesh,
    const ClusterBiMap<FaceHandle>& clusters,
    const FaceMap<Normal<typename BaseVecT::CoordType>>& normals,
    int minClusterSize,
    int iterations = 100,
    int samples = 10
);

template<typename BaseVecT>
void dragToRegressionPlane(
    BaseMesh<BaseVecT>& mesh,
    const Cluster<FaceHandle>& cluster,
    const Plane<BaseVecT>& plane,
    FaceMap<Normal<typename BaseVecT::CoordType>>& normals
);

template<typename BaseVecT>
void dragToRegressionPlanes(
    BaseMesh<BaseVecT>& mesh,
    const ClusterBiMap<FaceHandle>& clusters,
    const ClusterMap<Plane<BaseVecT>>& planes,
    FaceMap<Normal<typename BaseVecT::CoordType>>& normals
);

template<typename BaseVecT>
void optimizePlaneIntersections(
    BaseMesh<BaseVecT>& mesh,
    const ClusterBiMap<FaceHandle>& clusters,
    const ClusterMap<Plane<BaseVecT>>& planes
);

template<typename BaseVecT>
void dragOntoIntersection(
    BaseMesh<BaseVecT>& mesh,
    const ClusterBiMap<FaceHandle>& clusters,
    const ClusterHandle& clusterH,
    const ClusterHandle& neighbourClusterH,
    const Line<BaseVecT>& intersection
);

template<typename BaseVecT>
void debugPlanes(
    const BaseMesh<BaseVecT>& mesh,
    const ClusterBiMap<FaceHandle>& clusters,
    const ClusterMap<Plane<BaseVecT>>& planes,
    string filename,
    size_t minClusterSize
);

template<typename BaseVecT>
void deleteSmallPlanarCluster(
    BaseMesh<BaseVecT>& mesh,
    ClusterBiMap<FaceHandle>& clusters,
    size_t smallClusterThreshold
);

} // namespace lvr2

#include "lvr2/algorithm/ClusterAlgorithms.tcc"

#endif /* LVR2_ALGORITHM_CLUSTERALGORITHMS_H_ */