LambdaMST3D.h

 
#pragma once
 
#if defined(LAMBDAMST3D_RECURSES)
#error Recursive header files inclusion detected in LambdaMST3D.h
#else // defined(LAMBDAMST3D_RECURSES)
#define LAMBDAMST3D_RECURSES
 
#if !defined LAMBDAMST3D_h
#define LAMBDAMST3D_h
 
#include <algorithm>
#include <iterator>
#include <cmath>
#include <vector>
#include <map>
#include <DGtal/base/Common.h>
#include <DGtal/helpers/StdDefs.h>
#include "DGtal/kernel/CSpace.h"
#include "DGtal/kernel/PointVector.h"
#include "DGtal/geometry/curves/estimation/FunctorsLambdaMST.h"
#include "DGtal/geometry/curves/CForwardSegmentComputer.h"
#include "DGtal/geometry/curves/estimation/CLMSTTangentFromDSS.h"
#include "DGtal/geometry/curves/estimation/CLMSTDSSFilter.h"
 
namespace DGtal {
  template < typename TSpace, typename TSegmentation, typename Functor, typename DSSFilter >
  class LambdaMST3DEstimator
  {
  public: 
    //Checking concepts
    BOOST_CONCEPT_ASSERT(( concepts::CSpace < TSpace > ));
    BOOST_STATIC_ASSERT(( TSpace::dimension == 3 ));
    BOOST_CONCEPT_ASSERT(( concepts::CLMSTTangentFromDSS < Functor > ));
    BOOST_CONCEPT_ASSERT(( concepts::CLMSTDSSFilter < DSSFilter > ));
    BOOST_CONCEPT_ASSERT(( concepts::CForwardSegmentComputer< typename TSegmentation::SegmentComputer > ));
    // ----------------------- Types ------------------------------
  public:
    typedef TSegmentation Segmentation;
    typedef typename TSegmentation::SegmentComputer SegmentComputer;
    typedef typename SegmentComputer::ConstIterator ConstIterator;
    typedef typename Functor::Value Value;
    typedef typename TSpace::RealVector RealVector;
    typedef typename TSpace::Point Point;
    
    // ----------------------- Standard services ------------------------------
  public:
    LambdaMST3DEstimator ( );
    
    void init ( ConstIterator itb, ConstIterator ite );
    
    void attach ( Alias<TSegmentation> segmentComputer );
    
    bool isValid ( ) const;
    
    RealVector eval ( const Point & p );
    
    template <typename OutputIterator>
    OutputIterator eval ( ConstIterator itb, ConstIterator ite, OutputIterator result );
 
    DSSFilter & getDSSFilter ( );
    
    // ------------------------- Internals ------------------------------------
  protected:
 
      typedef typename std::vector<SegmentComputer >::const_iterator OrphanDSSIterator;
    
    template <typename OutputIterator>
    void accumulate ( std::multimap < Point, Value > & outValues, ConstIterator itb, ConstIterator ite, OutputIterator & result );
 
    Value treatOrphan(OrphanDSSIterator begin, OrphanDSSIterator end, const Point &p);
    template < typename DSSesIterator, typename OrphanIterator >
 
    void treatOrphans(DSSesIterator begin, DSSesIterator end, OrphanIterator obegin, OrphanIterator oend,
                       std::multimap<Point, Value> &outValues);
 
 
    // ------------------------- Private Datas --------------------------------
  private:
    ConstIterator myBegin;
    ConstIterator myEnd;
    TSegmentation * dssSegments;
    Functor myFunctor;
 
    DSSFilter myDSSFilter;
 
  }; // end of class LambdaTangentFromDSSEstimator
  
  //-------------------------------------------------------------------------------------------
  
  // Template class LambdaMST3D
  template < typename DSSSegmentationComputer, typename LambdaFunction = functors::Lambda64Function,
             typename DSSFilter = DSSMuteFilter < typename DSSSegmentationComputer::SegmentComputer > >
  class LambdaMST3D:
  public LambdaMST3DEstimator<Z3i::Space, DSSSegmentationComputer,
    TangentFromDSS3DFunctor< typename DSSSegmentationComputer::SegmentComputer, LambdaFunction >, DSSFilter >
    {
      typedef 
      LambdaMST3DEstimator<Z3i::Space, DSSSegmentationComputer,
      TangentFromDSS3DFunctor< typename DSSSegmentationComputer::SegmentComputer, LambdaFunction >, DSSFilter > Super;
      
    public: 
      LambdaMST3D() : Super() {}
    };
}// namespace DGtal
 
// Includes inline functions
#include "DGtal/geometry/curves/estimation/LambdaMST3D.ih"
//                                                                           //
 
#endif // !defined LAMBDAMST3D_h
 
#undef LAMBDAMST3D_RECURSES
#endif // else defined(LAMBDAMST3D_RECURSES)