MRSurfaceDistanceBuilder.h

Go to the documentation of this file.

#pragma once
 
#include "MRId.h"
#include "MRVector.h"
#include "MRVector3.h"
#include <cfloat>
#include <optional>
#include <queue>
 
namespace MR
{
 
 
struct VertDistance
{
    VertId vert;
    float distance = FLT_MAX;
 
    VertDistance() = default;
    VertDistance( VertId v, float d ) : vert( v ), distance( d ) { }
};
 
inline bool operator <( const VertDistance & a, const VertDistance & b )
{
    return a.distance > b.distance;
}
 
class SurfaceDistanceBuilder
{
public:
    MRMESH_API SurfaceDistanceBuilder( const Mesh & mesh, const VertBitSet* region );
    SurfaceDistanceBuilder( const Mesh & mesh, const Vector3f & target, const VertBitSet* region ) :
        SurfaceDistanceBuilder( mesh, region) { target_ = target; }
    MRMESH_API void addStartRegion( const VertBitSet & region, float startDistance );
    MRMESH_API void addStartVertices( const HashMap<VertId, float>& startVertices );
    MRMESH_API void addStart( const MeshTriPoint & start );
 
    MRMESH_API void setMaxVertUpdates( int v );
 
    MRMESH_API VertId growOne();
    VertScalars takeDistanceMap() { return std::move( vertDistanceMap_ ); }
 
public:
    bool done() const { return nextVerts_.empty(); }
    float doneDistance() const { return nextVerts_.empty() ? FLT_MAX : nextVerts_.top().distance; }
 
private:
    const Mesh & mesh_;
    const VertBitSet* region_{nullptr};
    VertScalars vertDistanceMap_;
    Vector<char,VertId> vertUpdatedTimes_;
    std::priority_queue<VertDistance> nextVerts_;
    std::optional<Vector3f> target_;
    int maxVertUpdates_ = 3;
 
    bool suggestVertDistance_( VertDistance c );
    void suggestDistancesAround_( VertId v );
    void considerLeftTriPath_( EdgeId e );
    float metricToPenalty_( float metric, VertId v ) const;
};
 
 
} // namespace MR