testSphericalHoughNormalVectorEstimator.cpp

Go to the documentation of this file.

#include <iostream>
#include "DGtal/base/Common.h"
#include "ConfigTest.h"
#include "DGtalCatch.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/shapes/implicit/ImplicitHyperCube.h"
#include "DGtal/shapes/implicit/ImplicitBall.h"
#include "DGtal/shapes/GaussDigitizer.h"
#include "DGtal/topology/LightImplicitDigitalSurface.h"
#include "DGtal/topology/DigitalSurface.h"
#include "DGtal/topology/CanonicSCellEmbedder.h"
#include "DGtal/base/BasicFunctors.h"
#include "DGtal/geometry/volumes/distance/LpMetric.h"
#include "DGtal/geometry/surfaces/estimation/LocalEstimatorFromSurfelFunctorAdapter.h"
#include "DGtal/geometry/surfaces/estimation/estimationFunctors/SphericalHoughNormalVectorEstimator.h"

#ifdef WITH_VISU3D_QGLVIEWER
#include "DGtal/io/viewers/Viewer3D.h"
#endif
#include "DGtal/io/boards/Board3D.h"


using namespace std;
using namespace DGtal;
using namespace Z3i;

// Functions for testing class SphericalHoughNormalVectorEstimator.

TEST_CASE( "Testing SphericalHoughNormalVectorEstimator" )
{
  typedef ImplicitBall<Space> Shape;
  typedef GaussDigitizer<Space,Shape> Gauss;
  
  typedef LightImplicitDigitalSurface<KSpace,Gauss> SurfaceContainer;
  typedef DigitalSurface<SurfaceContainer> Surface;
  typedef Surface::Surfel Surfel;
  
  
  Point p1( -50, -50, -50 );
  Point p2( 50, 50, 50 );
  KSpace K;
  REQUIRE( K.init( p1, p2, true ) );
  

  //Shape
  Shape shape(RealPoint::diagonal(0.0), 10.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

  
  typedef functors::SphericalHoughNormalVectorEstimator<Surfel, CanonicSCellEmbedder<KSpace> > SphericalHough;
  typedef functors::ConstValue<double> ConstFunctor;
  typedef LocalEstimatorFromSurfelFunctorAdapter<SurfaceContainer, LpMetric<Z3i::Space>, SphericalHough, ConstFunctor> Reporter;
  
  CanonicSCellEmbedder<KSpace> embedder(surface.container().space());
  SphericalHough estimator(embedder,1.0 , 0.001, 1000 , 10, 1);
  
  LpMetric<Z3i::Space> l2(2.0);
  ConstFunctor constFunc(1.0);
  Reporter reporter(surface, l2, estimator , constFunc);
  reporter.attach(surface);
  reporter.setParams(l2, estimator , constFunc, 10.0);
  reporter.init(1, surface.begin(),surface.end());

  REQUIRE( reporter.isValid() );
  
  typedef  Reporter::Quantity Quantity;
  
  Quantity result = reporter.eval( surface.begin() );
  trace.info() << "Result at begin = "<< result <<std::endl;
 
  //true normal (implicitBall)
  RealPoint res = embedder( *(surface.begin()) ).getNormalized();
  trace.info() << "Expecting  = "<< res <<std::endl;
  
  REQUIRE( std::abs(result.dot(res)) > 0.9 );
  
  
#ifdef WITH_VISU3D_QGLVIEWER_TESTS
  int argc=0;
  char **argv;
  QApplication application(argc,argv);
  Viewer3D<Z3i::Space, KSpace> viewer(K);
  viewer.show();
  for(Surface::ConstIterator it = surface.begin(), itend=surface.end(); it != itend ;
      ++it)
  {
    Quantity normal = reporter.eval( it );
    viewer << CustomColors3D( DGtal::Color(static_cast<unsigned char>(255*abs(normal[0])),
                                           static_cast<unsigned char>(255*abs(normal[1])),
                                           static_cast<unsigned char>(255*abs(normal[2]))),
                              DGtal::Color(static_cast<unsigned char>(255*abs(normal[0])),
                                           static_cast<unsigned char>(255*abs(normal[1])),
                                           static_cast<unsigned char>(255*abs(normal[2]))));

    viewer << K.unsigns(*it) ;
    
    Point center = K.sCoords ( *it );
    }
  viewer<< Viewer3D<>::updateDisplay;
  application.exec();
#endif
}