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/base/Common.h"
#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 getVoronoiVector(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)