MROutlierPoints.h

Go to the documentation of this file.

#pragma once
#include "MRMeshFwd.h"
#include "MRMesh/MRExpected.h"
#include "MRUnionFind.h"
#include "MRBitSet.h"
#include "MRConstants.h"
#include "MRFlagOperators.h"
 
namespace MR
{
 
struct OutlierParams
{
    int maxClusterSize = 20;
    int maxNeighbors = 7;
    float minHeight = 0.3f;
    float minAngle = PI_F / 3.f;
};
 
enum class OutlierTypeMask
{
    SmallComponents = 1 << 0, 
    WeaklyConnected = 1 << 1, 
    FarSurface = 1 << 2, 
    AwayNormal = 1 << 3, 
    All = SmallComponents | WeaklyConnected | FarSurface | AwayNormal
};
MR_MAKE_FLAG_OPERATORS( OutlierTypeMask )
 
 
class MRMESH_CLASS OutliersDetector
{
public:
    OutliersDetector() = default;
 
    MRMESH_API Expected<void> prepare( const PointCloud& pc, float radius, OutlierTypeMask mask, ProgressCallback progress = {} ); // calculate caches
 
    MRMESH_API void setParams( const OutlierParams& params );
    MRMESH_API const OutlierParams& getParams() const { return params_; }
 
    MRMESH_API Expected<VertBitSet> find( OutlierTypeMask mask, ProgressCallback progress = {} ); // unite and calculate actual outliers
 
    MRMESH_API const std::vector<uint8_t>& getWeaklyConnectedStat() { return weaklyConnectedStat_; }
 
private:
    Expected<VertBitSet> findSmallComponents( ProgressCallback progress = {} );
    Expected<VertBitSet> findWeaklyConnected( ProgressCallback progress = {} );
    Expected<VertBitSet> findFarSurface( ProgressCallback progress = {} );
    Expected<VertBitSet> findAwayNormal( ProgressCallback progress = {} );
 
    float radius_ = 1.f;
    OutlierParams params_;
 
    // Cached data
    UnionFind<VertId> unionFindStructure_; // SmallComponents
    std::vector<uint8_t> weaklyConnectedStat_; // WeaklyConnected
    std::vector<float> farSurfaceStat_; // FarSurface
    std::vector<float> badNormalStat_; // AwayNormal
 
    OutlierTypeMask maskCached_ = OutlierTypeMask( 0 ); // true means valid cache
 
    VertBitSet validPoints_;
};
 
struct FindOutliersParams
{
    OutlierParams finderParams; 
    float radius = 1.f; 
 
    OutlierTypeMask mask = OutlierTypeMask::All; 
 
    ProgressCallback progress = {}; 
};
 
MRMESH_API Expected<VertBitSet> findOutliers( const PointCloud& pc, const FindOutliersParams& params );
 
}