DGtal 1.4.0
Loading...
Searching...
No Matches
testTensorVotingViewer.cpp
1
31#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
58using namespace std;
59using namespace DGtal;
60
62// Functions for testing class LocalEstimatorFromFunctorAdapter.
64
69template<typename Shape>
70bool 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;
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
169int 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
179
180 trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
181 trace.endBlock();
182 return res ? 0 : 1;
183}
184// //
Structure representing an RGB triple with alpha component.
Definition Color.h:68
static const Color Black
Definition Color.h:413
Aim: Represents a set of n-1-cells in a nD space, together with adjacency relation between these cell...
DigitalSurfaceContainer::Surfel Surfel
DigitalSurfaceContainer::SurfelConstIterator ConstIterator
Aim: A class for computing the Gauss digitization of some Euclidean shape, i.e. its intersection with...
Aim: This class template may be used to (linearly) convert scalar values in a given range into a colo...
Aim: model of CEuclideanOrientedShape and CEuclideanBoundedShape concepts to create a ball in nD....
Aim: This class is a model of CCellularGridSpaceND. It represents the cubical grid as a cell complex,...
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 model of CDigitalSurfaceContainer which defines the digital surface as the boundary of an impl...
Aim: this class adapts any local functor on digital surface element to define a local estimator....
Aim: implements l_p metrics.
Definition LpMetric.h:75
static Self diagonal(Component val=1)
static SCell findABel(const KSpace &K, const PointPredicate &pp, unsigned int nbtries=1000)
Aim: Represent adjacencies between surfel elements, telling if it follows an interior to exterior ord...
void beginBlock(const std::string &keyword="")
std::ostream & emphase()
std::ostream & info()
double endBlock()
Aim: Implements a functor to detect feature points from normal tensor voting strategy.
CountedPtr< SH3::DigitalSurface > surface
SH3::DigitalSurface Surface
DGtal is the top-level namespace which contains all DGtal functions and types.
Trace trace
Definition Common.h:153
STL namespace.
Aim: A trivial embedder for signed cell, which corresponds to the canonic injection of cell centroids...
Modifier class in a Display3D stream. Useful to choose your own mode for a given class....
Represents a signed cell in a cellular grid space by its Khalimsky coordinates and a boolean value.
Aim: defines a functor on double numbers which corresponds to a Gaussian convolution kernel....
int main()
Definition testBits.cpp:56
MyPointD Point
KSpace K
bool testLocalEstimatorFromFunctorAdapter()