ReducedMedialAxis.h

#pragma once

#if defined(ReducedMedialAxis_RECURSES)
#error Recursive header files inclusion detected in ReducedMedialAxis.h
#else // defined(ReducedMedialAxis_RECURSES)

#define ReducedMedialAxis_RECURSES

#if !defined ReducedMedialAxis_h

#define ReducedMedialAxis_h

// Inclusions
#include <iostream>
#include <vector>
#include "DGtal/base/Common.h"
#include "DGtal/kernel/NumberTraits.h"
#include "DGtal/geometry/volumes/distance/CPowerSeparableMetric.h"
#include "DGtal/geometry/volumes/distance/PowerMap.h"
#include "DGtal/images/DefaultConstImageRange.h"
#include "DGtal/kernel/domains/HyperRectDomain.h"
#include "DGtal/images/ImageContainerBySTLMap.h"
#include "DGtal/images/CImage.h"
#include "DGtal/images/Image.h"

namespace DGtal
{

  // template class ReducedMedialAxis
  template <typename TPowerMap,
            typename TImageContainer =  ImageContainerBySTLMap<typename TPowerMap::Domain,
                                                               typename TPowerMap::PowerSeparableMetric::Value> >
  struct ReducedMedialAxis
  {
    //MA Container
    typedef Image<TImageContainer> Type;

    static
    Type getReducedMedialAxisFromPowerMap(const TPowerMap &aPowerMap)
    {
      TImageContainer *computedMA = new TImageContainer( aPowerMap.domain() );

      for (typename TPowerMap::Domain::ConstIterator it = aPowerMap.domain().begin(),
             itend = aPowerMap.domain().end(); it != itend; ++it)
        {
          const auto v  = aPowerMap( *it );
          const auto pv = aPowerMap.projectPoint( v );

          if ( aPowerMap.metricPtr()->powerDistance( *it, v, aPowerMap.weightImagePtr()->operator()( pv ) )
                      < NumberTraits<typename TPowerMap::PowerSeparableMetric::Value>::ZERO )
            computedMA->setValue( v, aPowerMap.weightImagePtr()->operator()( pv ) );
        }

      return Type( computedMA );
    }
  }; // end of class ReducedMedialAxis



} // namespace DGtal

//                                                                           //

#endif // !defined ReducedMedialAxis_h

#undef ReducedMedialAxis_RECURSES
#endif // else defined(ReducedMedialAxisdesign pa_RECURSES)