DistanceTransformation.h

 
#pragma once
 
#if defined(DistanceTransformation_RECURSES)
#error Recursive header files inclusion detected in DistanceTransformation.h
#else // defined(DistanceTransformation_RECURSES)
#define DistanceTransformation_RECURSES
 
#if !defined DistanceTransformation_h
#define DistanceTransformation_h
 
// Inclusions
#include <iostream>
#include <vector>
#include "DGtal/kernel/NumberTraits.h"
#include "DGtal/kernel/CPointPredicate.h"
#include "DGtal/geometry/volumes/distance/CSeparableMetric.h"
#include "DGtal/geometry/volumes/distance/VoronoiMap.h"
#include "DGtal/images/DefaultConstImageRange.h"
#include "DGtal/kernel/domains/HyperRectDomain.h"
#include "DGtal/base/ConstAlias.h"
 
namespace DGtal
{
 
  // template class DistanceTransformation
  template < typename TSpace,
             typename TPointPredicate,
             typename TSeparableMetric,
             typename TImageContainer =
               ImageContainerBySTLVector< HyperRectDomain<TSpace>,
                                          typename TSpace::Vector >
           >
  class DistanceTransformation
    : public VoronoiMap<TSpace,TPointPredicate, TSeparableMetric, TImageContainer>
  {
 
  public:
    BOOST_CONCEPT_ASSERT(( concepts::CSpace< TSpace > ));
    BOOST_CONCEPT_ASSERT(( concepts::CPointPredicate<TPointPredicate> ));
    BOOST_CONCEPT_ASSERT(( concepts::CSeparableMetric<TSeparableMetric> ));
 
    typedef TSeparableMetric SeparableMetric;
 
    typedef TSpace  Space;
 
    typedef typename TSpace::Vector  Vector;
 
    typedef TPointPredicate PointPredicate;
 
    typedef  typename SeparableMetric::Value Value;
 
    typedef  typename SeparableMetric::Point Point;
    BOOST_STATIC_ASSERT((boost::is_same< typename Space::Point,
                         typename SeparableMetric::Point>::value));
 
    typedef  DistanceTransformation<TSpace,TPointPredicate,TSeparableMetric> Self;
 
    typedef VoronoiMap<TSpace,TPointPredicate,TSeparableMetric> Parent;
 
    typedef  DefaultConstImageRange<Self> ConstRange;
 
 
    typedef typename VoronoiMap<TSpace,TPointPredicate,
            TSeparableMetric,TImageContainer>::Domain  Domain;
 
    DistanceTransformation(ConstAlias<Domain> aDomain,
                           ConstAlias<PointPredicate> predicate,
                           ConstAlias<SeparableMetric> aMetric):
      VoronoiMap<TSpace,TPointPredicate,TSeparableMetric,TImageContainer>(aDomain,
                                                                          predicate,
                                                                          aMetric)
    {}
 
    DistanceTransformation(ConstAlias<Domain> aDomain,
                           ConstAlias<PointPredicate> predicate,
                           ConstAlias<SeparableMetric> aMetric,
                           typename Parent::PeriodicitySpec const & aPeriodicitySpec)
      : VoronoiMap<TSpace,TPointPredicate,TSeparableMetric,TImageContainer>(aDomain,
                                                                            predicate,
                                                                            aMetric,
                                                                            aPeriodicitySpec)
    {}
 
    ~DistanceTransformation() {};
 
    // ------------------- Private functions ------------------------
  public:
 
    const Domain & domain() const
    {
      return Parent::domain();
    }
 
    ConstRange constRange() const
    {
      return ConstRange(*this);
    }
 
    Value operator()(const Point &aPoint) const
    {
      return this->myMetricPtr->operator()(aPoint,
                                           this->myImagePtr->operator()(aPoint));
    }
 
    Vector getVoronoiSite(const Point &aPoint) const
    {
      return this->myImagePtr->operator()(aPoint);
    }
 
    const SeparableMetric* metric() const
    {
      return Parent::metric();
    }
 
    void selfDisplay ( std::ostream & out ) const
    {
      out << "[DistanceTransformation] underlying VoronoiMap={";
      Parent::selfDisplay(out);
      out << "}";
    }
 
    // ------------------- protected methods ------------------------
  protected:
 
    DistanceTransformation();
 
 
    // ------------------- Private members ------------------------
  private:
 
  }; // end of class DistanceTransformation
 
 
// //                                                                           //
// ///////////////////////////////////////////////////////////////////////////////
 
  template <typename S,typename P,typename TSep>
  inline
  std::ostream&
  operator<< ( std::ostream & out,
               const DistanceTransformation<S,P,TSep> & object )
  {
    object.selfDisplay( out );
    return out;
  }
 
 
 
} // namespace DGtal
 
//                                                                           //
 
#endif // !defined DistanceTransformation_h
 
#undef DistanceTransformation_RECURSES
#endif // else defined(DistanceTransformation_RECURSES)