SurfaceMesh.h

 
#pragma once
 
#if defined(SurfaceMesh_RECURSES)
#error Recursive header files inclusion detected in SurfaceMesh.h
#else // defined(SurfaceMesh_RECURSES)
#define SurfaceMesh_RECURSES
 
#if !defined SurfaceMesh_h
#define SurfaceMesh_h
 
// Inclusions
#include <iostream>
#include <sstream>
#include <string>
#include "DGtal/base/Common.h"
#include "DGtal/base/IntegerSequenceIterator.h"
#include "DGtal/helpers/StdDefs.h"
 
namespace DGtal
{
  // template class SurfaceMesh
  template < typename TRealPoint, typename TRealVector >
  struct SurfaceMesh
  {
    typedef TRealPoint                              RealPoint;
    typedef TRealVector                             RealVector;
    typedef SurfaceMesh< RealPoint, RealVector >    Self;
    
    static const Dimension dimension = RealPoint::dimension;
    BOOST_STATIC_ASSERT( ( dimension == 3 ) );
    
    typedef typename RealVector::Component          Scalar;
    typedef std::vector<Scalar>                     Scalars;
    typedef std::size_t                             Size;
    typedef std::size_t                             Index;
    typedef Index                                   Face;
    typedef Index                                   Edge;
    typedef Index                                   Vertex;
    typedef std::pair< Edge, Scalar >               WeightedEdge;
    typedef std::pair< Face, Scalar >               WeightedFace;
    typedef std::vector< Vertex >                   Vertices;
    typedef std::vector< Edge >                     Edges;
    typedef std::vector< WeightedEdge >             WeightedEdges;
    typedef std::vector< Face >                     Faces;
    typedef std::vector< WeightedFace >             WeightedFaces;
    typedef std::pair< Vertex, Vertex >             VertexPair;
 
    // Required by CUndirectedSimpleLocalGraph
    typedef std::set<Vertex>                   VertexSet;
    template <typename Value> struct           VertexMap {
      typedef typename std::map<Vertex, Value> Type;
    };
 
    // Required by CUndirectedSimpleGraph
    
    typedef IntegerSequenceIterator< Vertex >       ConstIterator;
    
    //---------------------------------------------------------------------------
  public:
    
    ~SurfaceMesh() = default;
    SurfaceMesh() = default;
    SurfaceMesh( const Self& other ) = default;
    SurfaceMesh( Self&& other ) = default;
    Self& operator=( const Self& other ) = default;
 
    template <typename RealPointIterator, typename VerticesIterator>
    SurfaceMesh( RealPointIterator itPos, RealPointIterator itPosEnd,
                 VerticesIterator itVertices, VerticesIterator itVerticesEnd );
 
    template <typename RealPointIterator, typename VerticesIterator>
    bool init( RealPointIterator itPos, RealPointIterator itPosEnd,
               VerticesIterator itVertices, VerticesIterator itVerticesEnd );
 
    void clear();
 
    
    //---------------------------------------------------------------------------
  public:
    
    template <typename RealVectorIterator>
    bool setVertexNormals( RealVectorIterator itN, RealVectorIterator itNEnd );
 
    template <typename RealVectorIterator>
    bool setFaceNormals( RealVectorIterator itN, RealVectorIterator itNEnd );
 
    void computeFaceNormalsFromPositions();
 
    void computeFaceNormalsFromVertexNormals();
 
    void computeVertexNormalsFromFaceNormals();
    
    void computeVertexNormalsFromFaceNormalsWithMaxWeights();
 
    template <typename AnyRing>
    std::vector<AnyRing> computeFaceValuesFromVertexValues
    ( const std::vector<AnyRing>& vvalues ) const;
    
    template <typename AnyRing>
    std::vector<AnyRing> computeVertexValuesFromFaceValues
    ( const std::vector<AnyRing>& fvalues ) const;
 
    std::vector<RealVector> computeFaceUnitVectorsFromVertexUnitVectors
    ( const std::vector<RealVector>& vuvectors ) const;
 
    std::vector<RealVector> computeVertexUnitVectorsFromFaceUnitVectors
    ( const std::vector<RealVector>& fuvectors ) const;
    
 
    //---------------------------------------------------------------------------
  public:
 
    Size nbVertices() const
    { return myIncidentFaces.size(); }
 
    Size nbEdges() const
    { return myEdgeVertices.size(); }
 
    Size nbFaces() const
    { return myIncidentVertices.size(); }
 
    long Euler() const
    { return nbVertices() - nbEdges() + nbFaces(); }
    
    Edge makeEdge( Vertex i, Vertex j ) const;
 
    const Vertices&  incidentVertices( Face f ) const
    { return myIncidentVertices[ f ]; }
 
    const Faces& incidentFaces( Vertex v ) const
    { return myIncidentFaces[ v ]; }
    
    const Faces& neighborFaces( Face f ) const
    { return myNeighborFaces[ f ]; }
 
    const Vertices& neighborVertices( Vertex v ) const
    { return myNeighborVertices[ v ]; }
 
    const VertexPair& edgeVertices( Edge e ) const
    { return myEdgeVertices[ e ]; }
    
    const Faces& edgeFaces( Edge e ) const
    { return myEdgeFaces[ e ]; }
 
    const Faces& edgeRightFaces( Edge e ) const 
    { return myEdgeRightFaces[ e ]; }
 
    const Faces& edgeLeftFaces( Edge e ) const 
    { return myEdgeLeftFaces[ e ]; }
 
    const std::vector< Vertices >& allIncidentVertices() const
    { return myIncidentVertices; }
 
    const std::vector< Faces >& allIncidentFaces() const
    { return myIncidentFaces; }
    
    const std::vector< Faces >& allNeighborFaces() const
    { return myNeighborFaces; }
 
    const std::vector< Vertices >& allNeighborVertices() const
    { return myNeighborVertices; }
 
    const std::vector< VertexPair >& allEdgeVertices() const
    { return myEdgeVertices; }
    
    const std::vector< Faces >& allEdgeFaces() const
    { return myEdgeFaces; }
 
    const std::vector< Faces >& allEdgeRightFaces() const 
    { return myEdgeRightFaces; }
 
    const std::vector< Faces >& allEdgeLeftFaces() const 
    { return myEdgeLeftFaces; }
    
 
    //---------------------------------------------------------------------------
  public:
 
    Edges computeManifoldBoundaryEdges() const;
    Edges computeManifoldInnerEdges() const;
    Edges computeManifoldInnerConsistentEdges() const;
    Edges computeManifoldInnerUnconsistentEdges() const;
    Edges computeNonManifoldEdges() const;
    
    
    // ----------------------- Undirected simple graph services ----------------------
  public:
    
    Size size() const
    { return nbVertices(); }
 
    Size bestCapacity() const
    { return 8; }
      
    Size degree( const Vertex & v ) const
    { return myNeighborVertices[ v ].size(); }
 
    template <typename OutputIterator>
    void
    writeNeighbors( OutputIterator &it ,
                    const Vertex & v ) const
    {
      for ( auto&& nv : myNeighborVertices[ v ] )
        *it++ = nv;
    }
 
    template <typename OutputIterator, typename VertexPredicate>
    void
    writeNeighbors( OutputIterator &it ,
                    const Vertex & v,
                    const VertexPredicate & pred) const
    {
      for ( auto&& nv : myNeighborVertices[ v ] )
        if ( pred( nv ) ) *it++ = nv;
    }
 
    ConstIterator begin() const
    { return ConstIterator( 0 ); }
 
    ConstIterator end() const
    { return ConstIterator( nbVertices() ); }
 
    
    //---------------------------------------------------------------------------
  public:
 
    const std::vector< RealPoint >& positions() const
    { return myPositions; }
 
    RealPoint& position( Vertex v )
    { return myPositions[ v ]; }
    
    const RealPoint& position( Vertex v ) const
    { return myPositions[ v ]; }
      
    const std::vector< RealVector >& vertexNormals() const
    { return myVertexNormals; }
 
    std::vector< RealVector >& vertexNormals() 
    { return myVertexNormals; }
 
    RealVector& vertexNormal( Vertex v )
    { return myVertexNormals[ v ]; }
    
    const RealVector& vertexNormal( Vertex v ) const
    { return myVertexNormals[ v ]; }
    
    const std::vector< RealVector >& faceNormals() const
    { return myFaceNormals; }
 
    std::vector< RealVector >& faceNormals() 
    { return myFaceNormals; }
 
    RealVector& faceNormal( Face f )
    { return myFaceNormals[ f ]; }
    
    const RealVector& faceNormal( Face f ) const
    { return myFaceNormals[ f ]; }
    
    Scalar averageEdgeLength() const;
 
    Scalar localWindow( Face f ) const;
 
    void perturbateWithUniformRandomNoise( Scalar p );
 
    void perturbateWithAdaptiveUniformRandomNoise( Scalar p );
 
    RealPoint edgeCentroid( Index e ) const;
 
    RealPoint faceCentroid( Index f ) const;
 
    Scalar faceArea( Index f ) const;
 
    Scalars getMaxWeights( Index v ) const;
    
    WeightedFaces
    computeFacesInclusionsInBall( Scalar r, Index f ) const;
    
    std::tuple< Vertices, WeightedEdges, WeightedFaces >
    computeCellsInclusionsInBall( Scalar r, Index f ) const;
    
    Scalar faceInclusionRatio( RealPoint p, Scalar r, Index f ) const;
 
    Scalar edgeInclusionRatio( RealPoint p, Scalar r, Index e ) const;
 
    Scalar vertexInclusionRatio( RealPoint p, Scalar r, Index v ) const;
 
    
    // ----------------------- Interface --------------------------------------
  public:
 
    void selfDisplay ( std::ostream & out ) const;
 
    bool isValid() const;
    
    // ------------------------- Protected Datas ------------------------------
  protected:
    std::vector< Vertices >     myIncidentVertices;
    std::vector< Faces >        myIncidentFaces;
    std::vector< RealPoint >    myPositions;
    std::vector< RealVector >   myVertexNormals;
    std::vector< RealVector >   myFaceNormals;
    std::vector< Faces >        myNeighborFaces;
    std::vector< Vertices >     myNeighborVertices;
    std::vector< VertexPair >   myEdgeVertices;
    std::vector< Faces >        myEdgeFaces;
    std::vector< Faces >        myEdgeRightFaces;
    std::vector< Faces >        myEdgeLeftFaces;
 
    // ------------------------- Private Datas --------------------------------
  private:
 
 
    // ------------------------- Internals ------------------------------------
  protected:
 
    void computeNeighbors();
    void computeEdges();
 
    static Scalar rand01()
    { return (Scalar) rand() / (Scalar) RAND_MAX; }
    
  }; // end of class SurfaceMesh
 
  template < typename TRealPoint, typename TRealVector >
  std::ostream&
  operator<< ( std::ostream & out,
               const SurfaceMesh<TRealPoint, TRealVector> & object );  
  
} // namespace DGtal
 
// Includes inline functions.
#include "SurfaceMesh.ih"
//                                                                           //
 
#endif // !defined SurfaceMesh_h
 
#undef SurfaceMesh_RECURSES
#endif // else defined(SurfaceMesh_RECURSES)