testDistanceTransformationND.cpp

Go to the documentation of this file.

#include <iostream>
#include "DGtal/base/Common.h"
#include "DGtal/base/Common.h"
#include "DGtal/kernel/SpaceND.h"
#include "DGtal/kernel/domains/HyperRectDomain.h"
#include "DGtal/images/ImageSelector.h"
#include "DGtal/geometry/volumes/distance/ExactPredicateLpSeparableMetric.h"
#include "DGtal/geometry/volumes/distance/DistanceTransformation.h"
#include "DGtal/io/colormaps/HueShadeColorMap.h"
#include "DGtal/io/colormaps/GrayscaleColorMap.h"
#include "DGtal/io/boards/Board2D.h"
#include "DGtal/images/SimpleThresholdForegroundPredicate.h"

using namespace std;
using namespace DGtal;
using namespace DGtal::functors;

// Functions for testing class DistanceTransformND.

bool testDistanceTransformND()
{
  unsigned int nbok = 0;
  unsigned int nb = 0;

  trace.beginBlock ( "Testing dT dim=5 ..." );

  typedef SpaceND<5> TSpace;
  typedef TSpace::Point Point;
  typedef HyperRectDomain<TSpace> Domain;
  TSpace::Integer t[5] = {0,0,0,0,0};
  Point a ( t );
  TSpace::Integer t2[5] = {15,15,15,15,15};
  Point b ( t2 );
  TSpace::Integer t3[5] = {3,3,3,3,3};
  Point c ( t3 );
  Point d;

  typedef ImageSelector<Domain, unsigned int>::Type Image;
  Domain domain(a,b);
  Image image(domain);

  //We create an object image with a signle background point (set to 0)
  for (Image::Iterator it=image.begin(),itend=image.end(); it!=itend; ++it)
    *it = 128;
  image.setValue( c , 0 );

  typedef SimpleThresholdForegroundPredicate<Image> Predicate;
  Predicate aPredicate(image,0);

  typedef ExactPredicateLpSeparableMetric<TSpace, 2> L2Metric;
  L2Metric l2;
  DistanceTransformation<TSpace,Predicate,L2Metric> dt(&domain,&aPredicate, &l2 );

  //We check the result
  bool res=true;
  for(Domain::ConstIterator itDom = domain.begin(), itDomend = domain.end();
      itDom != itDomend; ++itDom)
    {
      //distance from the point to the seed
      d = (*itDom) - c;
      L2Metric::RawValue norm2=0;
      for(Point::Iterator itd=d.begin(), itdend=d.end(); itd!=itdend; ++itd)
        norm2+= (*itd)*(*itd);

      if ( dt.metric()->rawDistance( (*itDom), dt.getVoronoiVector(*itDom) ) != norm2)
  {
    trace.error()<<"Error at "<<(*itDom)
                 << ": expected="<<norm2<<" and computed="
                 <<dt.metric()->rawDistance( (*itDom), dt.getVoronoiVector(*itDom) )
                 <<endl;
  res=false;
  }
    }
  nbok += res ? 1 : 0;
  nb++;
  trace.info() << "(" << nbok << "/" << nb << ") "
         << "true == true" << std::endl;
  trace.endBlock();

  return nbok == nb;
}

// Standard services - public :

int main( int argc, char** argv )
{
  trace.beginBlock ( "Testing class DistanceTransformND" );
  trace.info() << "Args:";
  for ( int i = 0; i < argc; ++i )
    trace.info() << " " << argv[ i ];
  trace.info() << endl;

  bool res = testDistanceTransformND(); // && ... other tests
  trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
  trace.endBlock();
  return res ? 0 : 1;
}
//                                                                           //