testTensorVoting.cpp

Go to the documentation of this file.

#include <iostream>
#include "DGtal/base/Common.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/base/BasicFunctors.h"
#include "DGtal/graph/GraphVisitorRange.h"
#include "DGtal/io/boards/Board2D.h"
#include "DGtal/io/Color.h"
#include "DGtal/io/colormaps/GradientColorMap.h"
#include "DGtal/shapes/Shapes.h"
#include "DGtal/topology/CanonicSCellEmbedder.h"
#include "DGtal/graph/DistanceBreadthFirstVisitor.h"
#include "DGtal/geometry/volumes/distance/LpMetric.h"
#include "DGtal/geometry/surfaces/estimation/LocalEstimatorFromSurfelFunctorAdapter.h"
#include "DGtal/geometry/surfaces/estimation/estimationFunctors/BasicEstimatorFromSurfelsFunctors.h"
#include "DGtal/topology/LightImplicitDigitalSurface.h"
#include "DGtal/geometry/surfaces/estimation/estimationFunctors/TensorVotingFeatureExtraction.h"
 
#include "DGtal/shapes/implicit/ImplicitHyperCube.h"
#include "DGtal/shapes/implicit/ImplicitBall.h"
#include "DGtal/shapes/GaussDigitizer.h"
 
using namespace std;
using namespace DGtal;

// Functions for testing class LocalEstimatorFromFunctorAdapter.

template<typename Shape>
bool testLocalEstimatorFromFunctorAdapter()
{
  unsigned int nbok = 0;
  unsigned int nb = 0;
  trace.beginBlock ( "Testing init ..." );
 
  using namespace Z3i;
 
  typedef GaussDigitizer<Space,Shape> Gauss;
 
  typedef LightImplicitDigitalSurface<KSpace,Gauss> SurfaceContainer;
  typedef DigitalSurface<SurfaceContainer> Surface;
  typedef typename Surface::Surfel Surfel;
 
 
  trace.beginBlock("Creating Surface");
  Point p1( -10, -10, -10 );
  Point p2( 10, 10, 10 );
  KSpace K;
  nbok += K.init( p1, p2, true ) ? 1 : 0;
  nb++;
  trace.info() << "(" << nbok << "/" << nb << ") "
               << "K.init() is ok" << std::endl;
 
  //Shape
  Shape shape(RealPoint::diagonal(0.0), 6.0 );
  Gauss gauss;
  gauss.attach(shape);
  gauss.init(p1,p2,1.0);
 
  //Surface
  Surfel bel = Surfaces<KSpace>::findABel( K, gauss, 10000 );
  SurfaceContainer* surfaceContainer = new SurfaceContainer
    ( K, gauss, SurfelAdjacency<KSpace::dimension>( true ), bel );
  Surface surface( surfaceContainer ); // acquired
  trace.endBlock();
 
  trace.beginBlock("Creating  adapters");
  typedef functors::TensorVotingFeatureExtraction<Surfel, CanonicSCellEmbedder<KSpace> > FunctorVoting;
 
  typedef functors::GaussianKernel ConvFunctor;
  typedef LocalEstimatorFromSurfelFunctorAdapter<SurfaceContainer, LpMetric<Z3i::Space>, FunctorVoting, ConvFunctor> Reporter;
 
  CanonicSCellEmbedder<KSpace> embedder(surface.container().space());
  FunctorVoting estimator(embedder,1);
 
  ConvFunctor convFunc(1.0);
  Reporter reporter;
  reporter.attach(surface);
  LpMetric<Z3i::Space> l2double(2.0);
  reporter.setParams(l2double, estimator , convFunc, 2.0);
 
  reporter.init(1, surface.begin(), surface.end());
 
  typename FunctorVoting::Quantity val = reporter.eval( surface.begin());
  trace.info() << "probing at "<< *(surface.begin())<<std::endl;
 
  trace.info() << "Voting = "<<val <<std::endl;
 
  trace.endBlock();
  trace.endBlock();
 
  nbok += true ? 1 : 0;
  nb++;
  trace.info() << "(" << nbok << "/" << nb << ") "
               << "true == true" << std::endl;
 
  return nbok == nb;
}

bool testCube()
{
  unsigned int nbok = 0;
  unsigned int nb = 0;
  trace.beginBlock ( "Testing cube ..." );
 
  using namespace Z3i;
  typedef ImplicitHyperCube<Space> Shape;
  typedef GaussDigitizer<Space,Shape> Gauss;
 
  typedef LightImplicitDigitalSurface<KSpace,Gauss> SurfaceContainer;
  typedef DigitalSurface<SurfaceContainer> Surface;
  typedef  Surface::Surfel Surfel;
 
 
  trace.beginBlock("Creating Surface");
  Point p1( -10, -10, -10 );
  Point p2( 10, 10, 10 );
  KSpace K;
  nbok += K.init( p1, p2, true ) ? 1 : 0;
  nb++;
  trace.info() << "(" << nbok << "/" << nb << ") "
               << "K.init() is ok" << std::endl;
 
  //Shape
  Shape shape(RealPoint::diagonal(0.0), 6.0 );
  Gauss gauss;
  gauss.attach(shape);
  gauss.init(p1,p2,1.0);
 
  //Surface
  Surfel bel = Surfaces<KSpace>::findABel( K, gauss, 10000 );
  SurfaceContainer* surfaceContainer = new SurfaceContainer
    ( K, gauss, SurfelAdjacency<KSpace::dimension>( true ), bel );
  Surface surface( surfaceContainer ); // acquired
  trace.endBlock();
 
  trace.beginBlock("Creating  adapters");
  typedef functors::TensorVotingFeatureExtraction<Surfel, CanonicSCellEmbedder<KSpace> > FunctorVoting;
 
  typedef functors::GaussianKernel ConvFunctor;
  typedef LocalEstimatorFromSurfelFunctorAdapter<SurfaceContainer, LpMetric<Z3i::Space>, FunctorVoting, ConvFunctor> Reporter;
 
  CanonicSCellEmbedder<KSpace> embedder(surface.container().space());
  FunctorVoting estimator(embedder,1);
 
  LpMetric<Z3i::Space> l2(2.0);
 
  ConvFunctor convFunc(1.0);
  Reporter reporter(surface, l2, estimator , convFunc);
  reporter.attach(surface);
  reporter.setParams(l2, estimator , convFunc, 2.0);
  reporter.init(1, surface.begin(),surface.end());
  trace.endBlock();
 
  trace.beginBlock("Probing ...");
  Z3i::KSpace::Point p(11,-12,1);
  for(Surface::ConstIterator it = surface.begin(), itend = surface.end(); it!= itend; ++it)
    {
      if (K.sKCoords(*it) == p)
        {
          Reporter::Quantity val = reporter.eval( it );
          trace.info() << "probing at "<< *it<<std::endl;
          trace.info() << "Voting = "<<val <<std::endl;
        }
    }
  trace.endBlock();
  trace.endBlock();
 
  return nbok == nb;
}

// Standard services - public :
 
int main( int argc, char** argv )
{
  trace.beginBlock ( "Testing class TensorVotingFeatureExtraction" );
  trace.info() << "Args:";
  for ( int i = 0; i < argc; ++i )
    trace.info() << " " << argv[ i ];
  trace.info() << endl;
 
  bool res = testLocalEstimatorFromFunctorAdapter<ImplicitHyperCube<Z3i::Space> >()
    && testLocalEstimatorFromFunctorAdapter<ImplicitBall<Z3i::Space> >()
  && testCube();// && ... other tests
  trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
  trace.endBlock();
  return res ? 0 : 1;
}
//                                                                           //