DGtal  0.9.2
testSphereFitting.cpp
1 
30 #include <iostream>
32 #include "DGtal/base/Common.h"
33 #include "DGtal/helpers/StdDefs.h"
34 #include "DGtal/base/BasicFunctors.h"
35 #include "DGtal/graph/GraphVisitorRange.h"
36 #include "DGtal/io/boards/Board2D.h"
37 #include "DGtal/io/Color.h"
38 #include "DGtal/io/colormaps/GradientColorMap.h"
39 
40 #include "DGtal/shapes/Shapes.h"
41 #include "DGtal/shapes/implicit/ImplicitBall.h"
42 #include "DGtal/shapes/GaussDigitizer.h"
43 
44 #include "DGtal/topology/CanonicSCellEmbedder.h"
45 #include "DGtal/graph/DistanceBreadthFirstVisitor.h"
46 #include "DGtal/geometry/volumes/distance/ExactPredicateLpSeparableMetric.h"
47 #include "DGtal/geometry/surfaces/estimation/LocalEstimatorFromSurfelFunctorAdapter.h"
48 #include "DGtal/geometry/surfaces/estimation/estimationFunctors/BasicEstimatorFromSurfelsFunctors.h"
49 #include "DGtal/topology/LightImplicitDigitalSurface.h"
50 #include "DGtal/geometry/surfaces/estimation/estimationFunctors/SphereFittingEstimator.h"
51 #include "DGtal/geometry/surfaces/estimation/estimationFunctors/ElementaryConvolutionNormalVectorEstimator.h"
52 #include "DGtal/geometry/surfaces/estimation/EstimatorCache.h"
54 
55 using namespace std;
56 using namespace DGtal;
57 
59 // Functions for testing class LocalEstimatorFromFunctorAdapter.
61 
62 
67 bool testFitting()
68 {
69  unsigned int nbok = 0;
70  unsigned int nb = 0;
71  trace.beginBlock ( "Testing init ..." );
72 
73  using namespace Z3i;
74 
75  trace.beginBlock("Creating Surface");
76  Point p1( -20, -20, -20 );
77  Point p2( 20, 20, 20 );
78 
79  ImplicitBall<Z3i::Space> shape( RealPoint(6.0,0,0), 4);
81  Gauss gauss;
82  gauss.attach(shape);
83  gauss.init(p1, p2, 1);
84 
85  typedef LightImplicitDigitalSurface<KSpace, Gauss > SurfaceContainer;
86  typedef DigitalSurface<SurfaceContainer> Surface;
87  typedef Surface::Surfel Surfel;
88 
89 
90  KSpace K;
91  nbok += K.init( p1, p2, true ) ? 1 : 0;
92  nb++;
93  trace.info() << "(" << nbok << "/" << nb << ") "
94  << "K.init() is ok" << std::endl;
95  Surfel bel = Surfaces<KSpace>::findABel( K, gauss, 10000 );
96  SurfaceContainer* surfaceContainer = new SurfaceContainer
97  ( K, gauss, SurfelAdjacency<KSpace::dimension>( true ), bel );
98  Surface surface( surfaceContainer ); // acquired
99  CanonicSCellEmbedder<KSpace> embedder(surface.container().space());
100  trace.endBlock();
101 
102  trace.beginBlock("Normal vector field computation");
104  typedef LocalEstimatorFromSurfelFunctorAdapter<SurfaceContainer, Z3i::L2Metric,
105  FunctorNormal,
106  DGtal::functors::GaussianKernel> ReporterNormal;
107  typedef EstimatorCache<ReporterNormal> NormalCache;
108 
109  //estimator
110  DGtal::functors::GaussianKernel gaussKernelFunc(5.0);
111  FunctorNormal functorNormal(embedder, 1.0);
112  ReporterNormal reporterNormal;
113  reporterNormal.attach(surface);
114  reporterNormal.setParams(l2Metric, functorNormal, gaussKernelFunc, 5.0);
115 
116  //caching normal field
117  NormalCache normalCache(reporterNormal);
118  normalCache.init( 1, surface.begin(), surface.end());
119  trace.info() << "Normal vector field cached... "<< normalCache << std::endl;
120  trace.endBlock();
121 
122  trace.beginBlock("Creating sphere fitting adapter from normal vector field");
124  typedef functors::ConstValue< double > ConvFunctor;
126 
127  Functor fitter(embedder,1.0, 5.0, normalCache);
128  ConvFunctor convFunc(1.0);
129  Reporter reporter;
130  reporter.attach(surface);
131  reporter.setParams(l2Metric, fitter , convFunc, 15.0);
132 
133  reporter.init(1, surface.begin(), surface.end());
134  for(Surface::ConstIterator it = surface.begin(), ite=surface.end(); it!=ite; ++it)
135  {
136  Functor::Quantity val = reporter.eval( it );
137  trace.info() << "Fitting = "<<val.center <<" rad="<<val.radius<<std::endl;
138  }
139  trace.endBlock();
140 
141 
142  trace.endBlock();
143 
144  nbok += true ? 1 : 0;
145  nb++;
146  trace.info() << "(" << nbok << "/" << nb << ") "
147  << "true == true" << std::endl;
148 
149  return nbok == nb;
150 }
151 
153 // Standard services - public :
154 
155 int main( int argc, char** argv )
156 {
157  trace.beginBlock ( "Testing class SphereFitting" );
158  trace.info() << "Args:";
159  for ( int i = 0; i < argc; ++i )
160  trace.info() << " " << argv[ i ];
161  trace.info() << endl;
162 
163  bool res = testFitting(); // && ... other tests
164  trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
165  trace.endBlock();
166  return res ? 0 : 1;
167 }
168 // //
void beginBlock(const std::string &keyword="")
Aim: defines a functor on double numbers which corresponds to a Gaussian convolution kernel...
Aim: Define a simple functor that returns a constant value (0 by default).
Aim: Estimates normal vector by convolution of elementary normal vector to adjacent surfel...
Aim: A model of CDigitalSurfaceContainer which defines the digital surface as the boundary of an impl...
Trace trace
Definition: Common.h:130
Aim: A utility class for constructing surfaces (i.e. set of (n-1)-cells).
Definition: Surfaces.h:78
KhalimskySpaceND< 2, Integer > KSpace
Definition: StdDefs.h:77
STL namespace.
double endBlock()
Aim: Represents a set of n-1-cells in a nD space, together with adjacency relation between these cell...
void attach(const EuclideanShape &shape)
Aim: implements separable l_p metrics with exact predicates.
std::ostream & emphase()
Aim: A class for computing the Gauss digitization of some Euclidean shape, i.e. its intersection with...
Aim: model of CEuclideanOrientedShape and CEuclideanBoundedShape concepts to create a ball in nD...
Definition: ImplicitBall.h:64
DGtal is the top-level namespace which contains all DGtal functions and types.
std::ostream & info()
Aim: this class adapts any local functor on digital surface element to define a local estimator...
Aim: this class adapts any local surface estimator to cache the estimated values in a associative con...
Space::RealPoint RealPoint
Definition: StdDefs.h:97
Aim: Use Patate library to perform a local sphere fitting.