DGtal  1.0.0
testTensorVotingViewer.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 #include "DGtal/shapes/Shapes.h"
40 #include "DGtal/topology/CanonicSCellEmbedder.h"
41 #include "DGtal/graph/DistanceBreadthFirstVisitor.h"
42 #include "DGtal/geometry/volumes/distance/ExactPredicateLpSeparableMetric.h"
43 #include "DGtal/geometry/surfaces/estimation/LocalEstimatorFromSurfelFunctorAdapter.h"
44 #include "DGtal/geometry/surfaces/estimation/estimationFunctors/BasicEstimatorFromSurfelsFunctors.h"
45 #include "DGtal/topology/LightImplicitDigitalSurface.h"
46 #include "DGtal/geometry/surfaces/estimation/estimationFunctors/TensorVotingFeatureExtraction.h"
47 #include "DGtal/geometry/volumes/distance/LpMetric.h"
48 
49 #include "DGtal/shapes/implicit/ImplicitHyperCube.h"
50 #include "DGtal/shapes/implicit/ImplicitBall.h"
51 #include "DGtal/shapes/GaussDigitizer.h"
52 
53 #include "DGtal/io/viewers/Viewer3D.h"
54 #include "DGtal/io/colormaps/GradientColorMap.h"
55 
57 
58 using namespace std;
59 using namespace DGtal;
60 
62 // Functions for testing class LocalEstimatorFromFunctorAdapter.
64 
69 template<typename Shape>
70 bool testLocalEstimatorFromFunctorAdapter(int argc, char **argv)
71 {
72  unsigned int nbok = 0;
73  unsigned int nb = 0;
74  trace.beginBlock ( "Testing init ..." );
75 
76  using namespace Z3i;
77 
78  typedef GaussDigitizer<Space,Shape> Gauss;
79 
80  typedef LightImplicitDigitalSurface<KSpace,Gauss> SurfaceContainer;
81  typedef DigitalSurface<SurfaceContainer> Surface;
82  //typedef Surface::SurfelConstIterator ConstIterator;
83  //typedef Surface::Tracker Tracker;
84  typedef typename Surface::Surfel Surfel;
85 
86 
87  trace.beginBlock("Creating Surface");
88  Point p1( -100, -100, -100 );
89  Point p2( 100, 100, 100 );
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 
96  //Shape
97  Shape shape(RealPoint::diagonal(0.0), 30.0 );
98  Gauss gauss;
99  gauss.attach(shape);
100  gauss.init(p1,p2,1.0);
101 
102  //Surface
103  Surfel bel = Surfaces<KSpace>::findABel( K, gauss, 10000 );
104  SurfaceContainer* surfaceContainer = new SurfaceContainer
105  ( K, gauss, SurfelAdjacency<KSpace::dimension>( true ), bel );
106  Surface surface( surfaceContainer ); // acquired
107  trace.endBlock();
108 
109  trace.beginBlock("Creating adapters");
111 
112  typedef functors::GaussianKernel ConvFunctor;
113  typedef LocalEstimatorFromSurfelFunctorAdapter<SurfaceContainer, LpMetric<Space>, FunctorVoting, ConvFunctor> Reporter;
114 
115  CanonicSCellEmbedder<KSpace> embedder(surface.container().space());
116  FunctorVoting estimator(embedder,1);
117 
118  LpMetric<Space> l2(2.0); // L2 metric in R^3 for surface propagation.
119 
120  ConvFunctor convFunc(4.0);
121  Reporter reporter;
122  reporter.attach(surface);
123  reporter.setParams(l2, estimator , convFunc, 5.0);
124 
125  reporter.init(1, surface.begin() , surface.end());
126 
127  std::vector<double> values;
128  reporter.eval( surface.begin(), surface.end(), std::back_insert_iterator<std::vector<double> >(values));
129 
130  double maxval = *std::max_element(values.begin(), values.end());
131  double minval = *std::min_element(values.begin(), values.end());
132  trace.info() << "Min/max= "<< minval<<"/"<<maxval<<std::endl;
133  QApplication application( argc, argv );
135  Viewer viewer( K );
136  viewer.setWindowTitle("Features from Tensor Voting");
137  viewer.show();
138 
139  typedef GradientColorMap< double > Gradient;
140  Gradient cmap_grad( minval, maxval );
141  cmap_grad.addColor( Color( 50, 50, 255 ) );
142  cmap_grad.addColor( Color( 255, 0, 0 ) );
143  cmap_grad.addColor( Color( 255, 255, 10 ) );
144 
145 
146  viewer << SetMode3D((*(surface.begin())).className(), "Basic" );
147  unsigned int i=0;
148 
149  for(typename Surface::ConstIterator it = surface.begin(), itend=surface.end();
150  it!= itend;
151  ++it, ++i)
152  {
153  viewer << CustomColors3D( Color::Black, cmap_grad( values[ i ] ))
154  << *it ;
155  }
156 
157 
158  viewer << Viewer3D<>::updateDisplay;
159 
160  trace.endBlock();
161  application.exec();
162 
163  return true;
164 }
165 
167 // Standard services - public :
168 
169 int main( int argc, char** argv )
170 {
171  trace.beginBlock ( "Testing class TensorVotingFeatureExtraction" );
172  trace.info() << "Args:";
173  for ( int i = 0; i < argc; ++i )
174  trace.info() << " " << argv[ i ];
175  trace.info() << endl;
176 
177  bool res = testLocalEstimatorFromFunctorAdapter<ImplicitHyperCube<Z3i::Space> >(argc,argv)
179 
180  trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
181  trace.endBlock();
182  return res ? 0 : 1;
183 }
184 // //
void beginBlock(const std::string &keyword="")
Aim: defines a functor on double numbers which corresponds to a Gaussian convolution kernel....
Aim: This class template may be used to (linearly) convert scalar values in a given range into a colo...
Aim: A model of CDigitalSurfaceContainer which defines the digital surface as the boundary of an impl...
MyDigitalSurface::ConstIterator ConstIterator
Aim: implements l_p metrics.
Definition: LpMetric.h:74
Trace trace
Definition: Common.h:144
static Self diagonal(Component val=1)
Aim: A utility class for constructing surfaces (i.e. set of (n-1)-cells).
Definition: Surfaces.h:78
double endBlock()
Aim: Represents a set of n-1-cells in a nD space, together with adjacency relation between these cell...
std::ostream & emphase()
bool init(const Point &lower, const Point &upper, bool isClosed)
Specifies the upper and lower bounds for the maximal cells in this space.
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
int main(int argc, char **argv)
bool testLocalEstimatorFromFunctorAdapter()
DGtal is the top-level namespace which contains all DGtal functions and types.
MyPointD Point
Definition: testClone2.cpp:383
std::ostream & info()
Aim: this class adapts any local functor on digital surface element to define a local estimator....
Structure representing an RGB triple with alpha component.
Definition: Color.h:66
Aim: Implements a functor to detect feature points from normal tensor voting strategy.
KSpace K
Aim: This class is a model of CCellularGridSpaceND. It represents the cubical grid as a cell complex,...
Modifier class in a Display3D stream. Useful to choose your own mode for a given class....