MRPointsProject.h

Go to the documentation of this file.

#pragma once
 
#include "MRMeshFwd.h"
#include "MRExpected.h"
#include "MRId.h"
 
#include <cfloat>
 
namespace MR
{
 
struct PointsProjectionResult
{
    float distSq{ 0 };
    VertId vId;
 
    auto operator <=>(const PointsProjectionResult &) const = default;
};
 
 
[[nodiscard]] MRMESH_API PointsProjectionResult findProjectionOnPoints( const Vector3f& pt, const PointCloud& pc,
    float upDistLimitSq = FLT_MAX,
    const AffineXf3f* xf = nullptr,
    float loDistLimitSq = 0,
    VertPredicate skipCb = {} );
 
MRMESH_API void findFewClosestPoints( const Vector3f& pt, const PointCloud& pc, FewSmallest<PointsProjectionResult> & res,
    float upDistLimitSq = FLT_MAX,
    const AffineXf3f* xf = nullptr,
    float loDistLimitSq = 0 );
 
[[nodiscard]] MRMESH_API Buffer<VertId> findNClosestPointsPerPoint( const PointCloud& pc, int numNei, const ProgressCallback & progress = {} );
 
[[nodiscard]] MRMESH_API VertPair findTwoClosestPoints( const PointCloud& pc, const ProgressCallback & progress = {} );
 
struct FindProjectionOnPointsSettings
{
    const BitSet* valid = nullptr;
    const AffineXf3f* xf = nullptr;
    float upDistLimitSq = FLT_MAX;
    float loDistLimitSq = 0.f;
    bool skipSameIndex = false;
    ProgressCallback cb;
};
 
class IPointsProjector
{
public:
    virtual ~IPointsProjector() = default;
 
    virtual Expected<void> setPointCloud( const PointCloud& pointCloud ) = 0;
 
    virtual Expected<void> findProjections( std::vector<PointsProjectionResult>& results,
        const std::vector<Vector3f>& points, const FindProjectionOnPointsSettings& settings ) const = 0;
 
    virtual size_t projectionsHeapBytes( size_t numProjections ) const = 0;
};
 
class MRMESH_CLASS PointsProjector : public IPointsProjector
{
public:
    MRMESH_API Expected<void> setPointCloud( const PointCloud& pointCloud ) override;
 
    MRMESH_API Expected<void> findProjections( std::vector<PointsProjectionResult>& results,
        const std::vector<Vector3f>& points, const FindProjectionOnPointsSettings& settings ) const override;
 
    MRMESH_API virtual size_t projectionsHeapBytes( size_t numProjections ) const override;
private:
    const PointCloud* pointCloud_{ nullptr };
};
 
}