Point Cloud Library (PCL)  1.15.1-dev
sac_model.h
1 /*
2  * Software License Agreement (BSD License)
3  *
4  * Point Cloud Library (PCL) - www.pointclouds.org
5  * Copyright (c) 2010-2011, Willow Garage, Inc.
6  * Copyright (c) 2012-, Open Perception, Inc.
7  *
8  * All rights reserved.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  *
14  * * Redistributions of source code must retain the above copyright
15  * notice, this list of conditions and the following disclaimer.
16  * * Redistributions in binary form must reproduce the above
17  * copyright notice, this list of conditions and the following
18  * disclaimer in the documentation and/or other materials provided
19  * with the distribution.
20  * * Neither the name of the copyright holder(s) nor the names of its
21  * contributors may be used to endorse or promote products derived
22  * from this software without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
27  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
28  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
29  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
30  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
31  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
32  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
34  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
35  * POSSIBILITY OF SUCH DAMAGE.
36  *
37  * $Id$
38  *
39  */
40 
41 #pragma once
42 
43 #include <ctime>
44 #include <limits>
45 #include <memory>
46 #include <set>
47 #include <boost/random/mersenne_twister.hpp> // for mt19937
48 #include <boost/random/uniform_int.hpp> // for uniform_int
49 #include <boost/random/variate_generator.hpp> // for variate_generator
50 #include <random>
51 #include <numeric> // for iota
52 
53 #include <pcl/memory.h>
54 #include <pcl/console/print.h>
55 #include <pcl/point_cloud.h>
56 #include <pcl/types.h> // for index_t, Indices
57 #include <pcl/sample_consensus/model_types.h>
58 
59 #include <pcl/search/search.h>
60 
61 namespace pcl
62 {
63  template<class T> class ProgressiveSampleConsensus;
64 
65  /** \brief @b SampleConsensusModel represents the base model class. All sample consensus models must inherit
66  * from this class.
67  * \author Radu B. Rusu
68  * \ingroup sample_consensus
69  */
70  template <typename PointT>
72  {
73  public:
76  using PointCloudPtr = typename PointCloud::Ptr;
78 
79  using Ptr = shared_ptr<SampleConsensusModel<PointT> >;
80  using ConstPtr = shared_ptr<const SampleConsensusModel<PointT> >;
81 
82  protected:
83  /** \brief Empty constructor for base SampleConsensusModel.
84  * \param[in] random if true set the random seed to the current time, else set to 12345 (default: false)
85  */
86  SampleConsensusModel (bool random = false)
87  : input_ ()
88  , radius_min_ (-std::numeric_limits<double>::max ())
89  , radius_max_ (std::numeric_limits<double>::max ())
90  , samples_radius_ (0.)
92  , rng_dist_ (new boost::uniform_int<> (0, std::numeric_limits<int>::max ()))
93  , custom_model_constraints_ ([](auto){return true;})
94  {
95  // Create a random number generator object
96  if (random)
97  rng_alg_.seed (std::random_device()());
98  else
99  rng_alg_.seed (12345u);
100 
101  rng_gen_.reset (new boost::variate_generator<boost::mt19937&, boost::uniform_int<> > (rng_alg_, *rng_dist_));
102  }
103 
104  public:
105  /** \brief Constructor for base SampleConsensusModel.
106  * \param[in] cloud the input point cloud dataset
107  * \param[in] indices a vector of point indices to be used from \a cloud
108  * \param[in] random if true set the random seed to the current time, else set to 12345 (default: false)
109  */
110  SampleConsensusModel (const PointCloudConstPtr &cloud, const Indices &indices, const bool random = false)
111  : input_ (cloud)
112  , indices_(new Indices(indices))
113  , radius_min_ (-std::numeric_limits<double>::max ())
114  , radius_max_ (std::numeric_limits<double>::max ())
115  , samples_radius_ (0.)
117  , rng_dist_ (new boost::uniform_int<> (0, std::numeric_limits<int>::max ()))
118  , custom_model_constraints_ ([](auto){return true;})
119  {
120  if (random)
121  rng_alg_.seed (std::random_device()());
122  else
123  rng_alg_.seed (12345u);
124 
125  // If no indices are provided, use all points
126  if (indices_->empty())
127  {
128  indices_->resize(cloud->size());
129  std::iota(indices_->begin(), indices_->end(), 0);
130  }
131 
132  if (indices_->size () > input_->size ())
133  {
134  PCL_ERROR("[pcl::SampleConsensusModel] Invalid index vector given with size "
135  "%zu while the input PointCloud has size %zu!\n",
136  indices_->size(),
137  static_cast<std::size_t>(input_->size()));
138  indices_->clear ();
139  }
141 
142  // Create a random number generator object
143  rng_gen_.reset (new boost::variate_generator<boost::mt19937&, boost::uniform_int<> > (rng_alg_, *rng_dist_));
144  }
145 
146  /** \brief Constructor for base SampleConsensusModel.
147  * \param[in] cloud the input point cloud dataset
148  * \param[in] random if true set the random seed to the current time, else set to 12345 (default: false)
149  */
150  SampleConsensusModel (const PointCloudConstPtr &cloud, const bool random = false)
151  : SampleConsensusModel(cloud, Indices(), random) {};
152 
153  /** \brief Destructor for base SampleConsensusModel. */
154  virtual ~SampleConsensusModel () = default;
155 
156  /** \brief Get a set of random data samples and return them as point
157  * indices.
158  * \param[out] iterations the internal number of iterations used by SAC methods
159  * \param[out] samples the resultant model samples
160  */
161  virtual void
162  getSamples (int &iterations, Indices &samples)
163  {
164  // We're assuming that indices_ have already been set in the constructor
165  if (indices_->size () < getSampleSize ())
166  {
167  PCL_ERROR ("[pcl::SampleConsensusModel::getSamples] Can not select %lu unique points out of %lu!\n",
168  samples.size (), indices_->size ());
169  // one of these will make it stop :)
170  samples.clear ();
171  iterations = std::numeric_limits<int>::max() - 1;
172  return;
173  }
174 
175  // Get a second point which is different than the first
176  samples.resize (getSampleSize ());
177  for (unsigned int iter = 0; iter < getMaxSampleChecks(); ++iter)
178  {
179  // Choose the random indices
180  if (samples_radius_ < std::numeric_limits<double>::epsilon ())
182  else
184 
185  // If it's a good sample, stop here
186  if (isSampleGood (samples))
187  {
188  PCL_DEBUG ("[pcl::SampleConsensusModel::getSamples] Selected %lu samples.\n", samples.size ());
189  return;
190  }
191  }
192  PCL_DEBUG("[pcl::SampleConsensusModel::getSamples] WARNING: Could not select "
193  "%d sample points in %d iterations!\n",
194  getSampleSize(),
196  samples.clear ();
197  }
198 
199  /** \brief Check whether the given index samples can form a valid model,
200  * compute the model coefficients from these samples and store them
201  * in model_coefficients. Pure virtual.
202  * Implementations of this function must be thread-safe.
203  * \param[in] samples the point indices found as possible good candidates
204  * for creating a valid model
205  * \param[out] model_coefficients the computed model coefficients
206  */
207  virtual bool
209  Eigen::VectorXf &model_coefficients) const = 0;
210 
211  /** \brief Recompute the model coefficients using the given inlier set
212  * and return them to the user. Pure virtual.
213  *
214  * @note: these are the coefficients of the model after refinement
215  * (e.g., after a least-squares optimization)
216  *
217  * \param[in] inliers the data inliers supporting the model
218  * \param[in] model_coefficients the initial guess for the model coefficients
219  * \param[out] optimized_coefficients the resultant recomputed coefficients after non-linear optimization
220  */
221  virtual void
223  const Eigen::VectorXf &model_coefficients,
224  Eigen::VectorXf &optimized_coefficients) const = 0;
225 
226  /** \brief Compute all distances from the cloud data to a given model. Pure virtual.
227  *
228  * \param[in] model_coefficients the coefficients of a model that we need to compute distances to
229  * \param[out] distances the resultant estimated distances
230  */
231  virtual void
232  getDistancesToModel (const Eigen::VectorXf &model_coefficients,
233  std::vector<double> &distances) const = 0;
234 
235  /** \brief Select all the points which respect the given model
236  * coefficients as inliers. Pure virtual.
237  *
238  * \param[in] model_coefficients the coefficients of a model that we need to compute distances to
239  * \param[in] threshold a maximum admissible distance threshold for determining the inliers from
240  * the outliers
241  * \param[out] inliers the resultant model inliers
242  */
243  virtual void
244  selectWithinDistance (const Eigen::VectorXf &model_coefficients,
245  const double threshold,
246  Indices &inliers) = 0;
247 
248  /** \brief Count all the points which respect the given model
249  * coefficients as inliers. Pure virtual.
250  * Implementations of this function must be thread-safe.
251  * \param[in] model_coefficients the coefficients of a model that we need to
252  * compute distances to
253  * \param[in] threshold a maximum admissible distance threshold for
254  * determining the inliers from the outliers
255  * \return the resultant number of inliers
256  */
257  virtual std::size_t
258  countWithinDistance (const Eigen::VectorXf &model_coefficients,
259  const double threshold) const = 0;
260 
261  /** \brief Create a new point cloud with inliers projected onto the model. Pure virtual.
262  * \param[in] inliers the data inliers that we want to project on the model
263  * \param[in] model_coefficients the coefficients of a model
264  * \param[out] projected_points the resultant projected points
265  * \param[in] copy_data_fields set to true (default) if we want the \a
266  * projected_points cloud to be an exact copy of the input dataset minus
267  * the point projections on the plane model
268  */
269  virtual void
270  projectPoints (const Indices &inliers,
271  const Eigen::VectorXf &model_coefficients,
272  PointCloud &projected_points,
273  bool copy_data_fields = true) const = 0;
274 
275  /** \brief Verify whether a subset of indices verifies a given set of
276  * model coefficients. Pure virtual.
277  *
278  * \param[in] indices the data indices that need to be tested against the model
279  * \param[in] model_coefficients the set of model coefficients
280  * \param[in] threshold a maximum admissible distance threshold for
281  * determining the inliers from the outliers
282  */
283  virtual bool
284  doSamplesVerifyModel (const std::set<index_t> &indices,
285  const Eigen::VectorXf &model_coefficients,
286  const double threshold) const = 0;
287 
288  /** \brief Provide a pointer to the input dataset
289  * \param[in] cloud the const boost shared pointer to a PointCloud message
290  */
291  inline virtual void
293  {
294  input_ = cloud;
295  if (!indices_)
296  indices_.reset (new Indices ());
297  if (indices_->empty ())
298  {
299  // Prepare a set of indices to be used (entire cloud)
300  indices_->resize (cloud->size ());
301  for (std::size_t i = 0; i < cloud->size (); ++i)
302  (*indices_)[i] = static_cast<index_t> (i);
303  }
305  }
306 
307  /** \brief Get a pointer to the input point cloud dataset. */
308  inline PointCloudConstPtr
309  getInputCloud () const { return (input_); }
310 
311  /** \brief Provide a pointer to the vector of indices that represents the input data.
312  * \param[in] indices a pointer to the vector of indices that represents the input data.
313  */
314  inline void
315  setIndices (const IndicesPtr &indices)
316  {
317  indices_ = indices;
319  }
320 
321  /** \brief Provide the vector of indices that represents the input data.
322  * \param[out] indices the vector of indices that represents the input data.
323  */
324  inline void
325  setIndices (const Indices &indices)
326  {
327  indices_.reset (new Indices (indices));
328  shuffled_indices_ = indices;
329  }
330 
331  /** \brief Get a pointer to the vector of indices used. */
332  inline IndicesPtr
333  getIndices () const { return (indices_); }
334 
335  /** \brief Return a unique id for each type of model employed. */
336  virtual SacModel
337  getModelType () const = 0;
338 
339  /** \brief Get a string representation of the name of this class. */
340  inline const std::string&
341  getClassName () const
342  {
343  return (model_name_);
344  }
345 
346  /** \brief Return the size of a sample from which the model is computed. */
347  inline unsigned int
348  getSampleSize () const
349  {
350  return sample_size_;
351  }
352 
353  static unsigned int
355  {
356  /** The maximum number of samples to try until we get a good one */
357  static const unsigned int max_sample_checks_ = 1000;
358  return max_sample_checks_;
359  }
360 
361  /** \brief Return the number of coefficients in the model. */
362  inline unsigned int
363  getModelSize () const
364  {
365  return model_size_;
366  }
367 
368  /** \brief Set the minimum and maximum allowable radius limits for the
369  * model (applicable to models that estimate a radius)
370  * \param[in] min_radius the minimum radius model
371  * \param[in] max_radius the maximum radius model
372  * \todo change this to set limits on the entire model
373  */
374  inline void
375  setRadiusLimits (const double &min_radius, const double &max_radius)
376  {
377  radius_min_ = min_radius;
378  radius_max_ = max_radius;
379  }
380 
381  /** \brief Get the minimum and maximum allowable radius limits for the
382  * model as set by the user.
383  *
384  * \param[out] min_radius the resultant minimum radius model
385  * \param[out] max_radius the resultant maximum radius model
386  */
387  inline void
388  getRadiusLimits (double &min_radius, double &max_radius) const
389  {
390  min_radius = radius_min_;
391  max_radius = radius_max_;
392  }
393 
394  /** \brief This can be used to impose any kind of constraint on the model,
395  * e.g. that it has a specific direction, size, or anything else.
396  * \param[in] function A function that gets model coefficients and returns whether the model is acceptable or not.
397  */
398  inline void
399  setModelConstraints (std::function<bool(const Eigen::VectorXf &)> function)
400  {
401  if (!function)
402  {
403  PCL_ERROR ("[pcl::SampleConsensusModel::setModelConstraints] The given function is empty (i.e. does not contain a callable target)!\n");
404  return;
405  }
406  custom_model_constraints_ = std::move (function);
407  }
408 
409  /** \brief Set the maximum distance allowed when drawing random samples
410  * \param[in] radius the maximum distance (L2 norm)
411  * \param search
412  */
413  inline void
414  setSamplesMaxDist (const double &radius, SearchPtr search)
415  {
416  samples_radius_ = radius;
417  samples_radius_search_ = search;
418  }
419 
420  /** \brief Get maximum distance allowed when drawing random samples
421  *
422  * \param[out] radius the maximum distance (L2 norm)
423  */
424  inline void
425  getSamplesMaxDist (double &radius) const
426  {
427  radius = samples_radius_;
428  }
429 
430  friend class ProgressiveSampleConsensus<PointT>;
431 
432  /** \brief Compute the variance of the errors to the model.
433  * \param[in] error_sqr_dists a vector holding the distances
434  */
435  inline double
436  computeVariance (const std::vector<double> &error_sqr_dists) const
437  {
438  std::vector<double> dists (error_sqr_dists);
439  const std::size_t medIdx = dists.size () >> 1;
440  std::nth_element (dists.begin (), dists.begin () + medIdx, dists.end ());
441  double median_error_sqr = dists[medIdx];
442  return (2.1981 * median_error_sqr);
443  }
444 
445  /** \brief Compute the variance of the errors to the model from the internally
446  * estimated vector of distances. The model must be computed first (or at least
447  * selectWithinDistance must be called).
448  */
449  inline double
451  {
452  if (error_sqr_dists_.empty ())
453  {
454  PCL_ERROR ("[pcl::SampleConsensusModel::computeVariance] The variance of the Sample Consensus model distances cannot be estimated, as the model has not been computed yet. Please compute the model first or at least run selectWithinDistance before continuing. Returning NAN!\n");
455  return (std::numeric_limits<double>::quiet_NaN ());
456  }
458  }
459 
460  protected:
461 
462  /** \brief Fills a sample array with random samples from the indices_ vector
463  * \param[out] sample the set of indices of target_ to analyze
464  */
465  inline void
467  {
468  std::size_t sample_size = sample.size ();
469  std::size_t index_size = shuffled_indices_.size ();
470  for (std::size_t i = 0; i < sample_size; ++i)
471  // The 1/(RAND_MAX+1.0) trick is when the random numbers are not uniformly distributed and for small modulo
472  // elements, that does not matter (and nowadays, random number generators are good)
473  //std::swap (shuffled_indices_[i], shuffled_indices_[i + (rand () % (index_size - i))]);
474  std::swap (shuffled_indices_[i], shuffled_indices_[i + (rnd () % (index_size - i))]);
475  std::copy (shuffled_indices_.cbegin (), shuffled_indices_.cbegin () + sample_size, sample.begin ());
476  }
477 
478  /** \brief Fills a sample array with one random sample from the indices_ vector
479  * and other random samples that are closer than samples_radius_
480  * \param[out] sample the set of indices of target_ to analyze
481  */
482  inline void
484  {
485  std::size_t sample_size = sample.size ();
486  std::size_t index_size = shuffled_indices_.size ();
487 
488  std::swap (shuffled_indices_[0], shuffled_indices_[0 + (rnd () % (index_size - 0))]);
489  //const PointT& pt0 = (*input_)[shuffled_indices_[0]];
490 
491  Indices indices;
492  std::vector<float> sqr_dists;
493 
494  // If indices have been set when the search object was constructed,
495  // radiusSearch() expects an index into the indices vector as its
496  // first parameter. This can't be determined efficiently, so we use
497  // the point instead of the index.
498  // Returned indices are converted automatically.
499  samples_radius_search_->radiusSearch (input_->at(shuffled_indices_[0]),
500  samples_radius_, indices, sqr_dists );
501 
502  if (indices.size () < sample_size - 1)
503  {
504  // radius search failed, make an invalid model
505  for(std::size_t i = 1; i < sample_size; ++i)
507  }
508  else
509  {
510  for (std::size_t i = 0; i < sample_size-1; ++i)
511  std::swap (indices[i], indices[i + (rnd () % (indices.size () - i))]);
512  for (std::size_t i = 1; i < sample_size; ++i)
513  shuffled_indices_[i] = indices[i-1];
514  }
515 
516  std::copy (shuffled_indices_.cbegin (), shuffled_indices_.cbegin () + sample_size, sample.begin ());
517  }
518 
519  /** \brief Check whether a model is valid given the user constraints.
520  *
521  * Default implementation verifies that the number of coefficients in the supplied model is as expected for this
522  * SAC model type. Specific SAC models should extend this function by checking the user constraints (if any).
523  *
524  * \param[in] model_coefficients the set of model coefficients
525  */
526  virtual bool
527  isModelValid (const Eigen::VectorXf &model_coefficients) const
528  {
529  if (model_coefficients.size () != model_size_)
530  {
531  PCL_ERROR ("[pcl::%s::isModelValid] Invalid number of model coefficients given (is %lu, should be %lu)!\n", getClassName ().c_str (), model_coefficients.size (), model_size_);
532  return (false);
533  }
534  if (!custom_model_constraints_(model_coefficients))
535  {
536  PCL_DEBUG ("[pcl::%s::isModelValid] The user defined isModelValid function returned false.\n", getClassName ().c_str ());
537  return (false);
538  }
539  return (true);
540  }
541 
542  /** \brief Check if a sample of indices results in a good sample of points
543  * indices. Pure virtual.
544  * \param[in] samples the resultant index samples
545  */
546  virtual bool
547  isSampleGood (const Indices &samples) const = 0;
548 
549  /** \brief The model name. */
550  std::string model_name_;
551 
552  /** \brief A boost shared pointer to the point cloud data array. */
554 
555  /** \brief A pointer to the vector of point indices to use. */
557 
558  /** The maximum number of samples to try until we get a good one */
559  PCL_DEPRECATED(1, 18, "Use getMaxSampleChecks() instead.")
560  static const unsigned int max_sample_checks_ = 1000;
561 
562  /** \brief The minimum and maximum radius limits for the model.
563  * Applicable to all models that estimate a radius.
564  */
566 
567  /** \brief The maximum distance of subsequent samples from the first (radius search) */
569 
570  /** \brief The search object for picking subsequent samples using radius search */
572 
573  /** Data containing a shuffled version of the indices. This is used and modified when drawing samples. */
575 
576  /** \brief Boost-based random number generator algorithm. */
577  boost::mt19937 rng_alg_;
578 
579  /** \brief Boost-based random number generator distribution. */
580  std::shared_ptr<boost::uniform_int<> > rng_dist_;
581 
582  /** \brief Boost-based random number generator. */
583  std::shared_ptr<boost::variate_generator< boost::mt19937&, boost::uniform_int<> > > rng_gen_;
584 
585  /** \brief A vector holding the distances to the computed model. Used internally. */
586  std::vector<double> error_sqr_dists_;
587 
588  /** \brief The size of a sample from which the model is computed. Every subclass should initialize this appropriately. */
589  unsigned int sample_size_;
590 
591  /** \brief The number of coefficients in the model. Every subclass should initialize this appropriately. */
592  unsigned int model_size_;
593 
594  /** \brief Boost-based random number generator. */
595  inline int
596  rnd ()
597  {
598  return ((*rng_gen_) ());
599  }
600 
601  /** \brief A user defined function that takes model coefficients and returns whether the model is acceptable or not. */
602  std::function<bool(const Eigen::VectorXf &)> custom_model_constraints_;
603  public:
605  };
606 
607  /** \brief @b SampleConsensusModelFromNormals represents the base model class
608  * for models that require the use of surface normals for estimation.
609  * \ingroup sample_consensus
610  */
611  template <typename PointT, typename PointNT>
612  class SampleConsensusModelFromNormals //: public SampleConsensusModel<PointT>
613  {
614  public:
617 
618  using Ptr = shared_ptr<SampleConsensusModelFromNormals<PointT, PointNT> >;
619  using ConstPtr = shared_ptr<const SampleConsensusModelFromNormals<PointT, PointNT> >;
620 
621  /** \brief Empty constructor for base SampleConsensusModelFromNormals. */
623 
624  /** \brief Destructor. */
625  virtual ~SampleConsensusModelFromNormals () = default;
626 
627  /** \brief Set the normal angular distance weight.
628  * \param[in] w the relative weight (between 0 and 1) to give to the angular
629  * distance (0 to pi/2) between point normals and the plane normal.
630  * (The Euclidean distance will have weight 1-w.)
631  */
632  inline void
633  setNormalDistanceWeight (const double w)
634  {
635  if (w < 0.0 || w > 1.0)
636  {
637  PCL_ERROR ("[pcl::SampleConsensusModel::setNormalDistanceWeight] w is %g, but should be in [0; 1]. Weight will not be set.", w);
638  return;
639  }
641  }
642 
643  /** \brief Get the normal angular distance weight. */
644  inline double
646 
647  /** \brief Provide a pointer to the input dataset that contains the point
648  * normals of the XYZ dataset.
649  *
650  * \param[in] normals the const boost shared pointer to a PointCloud message
651  */
652  inline void
654  {
655  normals_ = normals;
656  }
657 
658  /** \brief Get a pointer to the normals of the input XYZ point cloud dataset. */
659  inline PointCloudNConstPtr
660  getInputNormals () const { return (normals_); }
661 
662  protected:
663  /** \brief The relative weight (between 0 and 1) to give to the angular
664  * distance (0 to pi/2) between point normals and the plane normal.
665  */
667 
668  /** \brief A pointer to the input dataset that contains the point normals
669  * of the XYZ dataset.
670  */
672  };
673 
674  /** Base functor all the models that need non linear optimization must
675  * define their own one and implement operator() (const Eigen::VectorXd& x, Eigen::VectorXd& fvec)
676  * or operator() (const Eigen::VectorXf& x, Eigen::VectorXf& fvec) depending on the chosen _Scalar
677  */
678  template<typename _Scalar, int NX=Eigen::Dynamic, int NY=Eigen::Dynamic>
679  struct Functor
680  {
681  using Scalar = _Scalar;
682  enum
683  {
686  };
687 
688  using ValueType = Eigen::Matrix<Scalar,ValuesAtCompileTime,1>;
689  using InputType = Eigen::Matrix<Scalar,InputsAtCompileTime,1>;
690  using JacobianType = Eigen::Matrix<Scalar,ValuesAtCompileTime,InputsAtCompileTime>;
691 
692  /** \brief Empty Constructor. */
693  Functor () : m_data_points_ (ValuesAtCompileTime) {}
694 
695  /** \brief Constructor
696  * \param[in] m_data_points number of data points to evaluate.
697  */
698  Functor (int m_data_points) : m_data_points_ (m_data_points) {}
699 
700  virtual ~Functor () = default;
701 
702  /** \brief Get the number of values. */
703  int
704  values () const { return (m_data_points_); }
705 
706  private:
707  const int m_data_points_;
708  };
709 }
PointCloud represents the base class in PCL for storing collections of 3D points.
Definition: point_cloud.h:174
shared_ptr< PointCloud< PointT > > Ptr
Definition: point_cloud.h:414
shared_ptr< const PointCloud< PointT > > ConstPtr
Definition: point_cloud.h:415
ProgressiveSampleConsensus represents an implementation of the PROSAC (PROgressive SAmple Consensus) ...
Definition: prosac.h:56
SampleConsensusModelFromNormals represents the base model class for models that require the use of su...
Definition: sac_model.h:613
void setNormalDistanceWeight(const double w)
Set the normal angular distance weight.
Definition: sac_model.h:633
PointCloudNConstPtr normals_
A pointer to the input dataset that contains the point normals of the XYZ dataset.
Definition: sac_model.h:671
typename pcl::PointCloud< PointNT >::ConstPtr PointCloudNConstPtr
Definition: sac_model.h:615
void setInputNormals(const PointCloudNConstPtr &normals)
Provide a pointer to the input dataset that contains the point normals of the XYZ dataset.
Definition: sac_model.h:653
double getNormalDistanceWeight() const
Get the normal angular distance weight.
Definition: sac_model.h:645
SampleConsensusModelFromNormals()
Empty constructor for base SampleConsensusModelFromNormals.
Definition: sac_model.h:622
virtual ~SampleConsensusModelFromNormals()=default
Destructor.
shared_ptr< const SampleConsensusModelFromNormals< PointT, PointNT > > ConstPtr
Definition: sac_model.h:619
typename pcl::PointCloud< PointNT >::Ptr PointCloudNPtr
Definition: sac_model.h:616
double normal_distance_weight_
The relative weight (between 0 and 1) to give to the angular distance (0 to pi/2) between point norma...
Definition: sac_model.h:666
PointCloudNConstPtr getInputNormals() const
Get a pointer to the normals of the input XYZ point cloud dataset.
Definition: sac_model.h:660
shared_ptr< SampleConsensusModelFromNormals< PointT, PointNT > > Ptr
Definition: sac_model.h:618
SampleConsensusModel represents the base model class.
Definition: sac_model.h:72
SampleConsensusModel(const PointCloudConstPtr &cloud, const bool random=false)
Constructor for base SampleConsensusModel.
Definition: sac_model.h:150
virtual void getSamples(int &iterations, Indices &samples)
Get a set of random data samples and return them as point indices.
Definition: sac_model.h:162
static const unsigned int max_sample_checks_
The maximum number of samples to try until we get a good one.
Definition: sac_model.h:560
virtual bool isSampleGood(const Indices &samples) const =0
Check if a sample of indices results in a good sample of points indices.
virtual bool computeModelCoefficients(const Indices &samples, Eigen::VectorXf &model_coefficients) const =0
Check whether the given index samples can form a valid model, compute the model coefficients from the...
virtual SacModel getModelType() const =0
Return a unique id for each type of model employed.
void drawIndexSampleRadius(Indices &sample)
Fills a sample array with one random sample from the indices_ vector and other random samples that ar...
Definition: sac_model.h:483
unsigned int getModelSize() const
Return the number of coefficients in the model.
Definition: sac_model.h:363
double radius_min_
The minimum and maximum radius limits for the model.
Definition: sac_model.h:565
std::function< bool(const Eigen::VectorXf &)> custom_model_constraints_
A user defined function that takes model coefficients and returns whether the model is acceptable or ...
Definition: sac_model.h:602
void setRadiusLimits(const double &min_radius, const double &max_radius)
Set the minimum and maximum allowable radius limits for the model (applicable to models that estimate...
Definition: sac_model.h:375
void getSamplesMaxDist(double &radius) const
Get maximum distance allowed when drawing random samples.
Definition: sac_model.h:425
PointCloudConstPtr getInputCloud() const
Get a pointer to the input point cloud dataset.
Definition: sac_model.h:309
virtual void optimizeModelCoefficients(const Indices &inliers, const Eigen::VectorXf &model_coefficients, Eigen::VectorXf &optimized_coefficients) const =0
Recompute the model coefficients using the given inlier set and return them to the user.
shared_ptr< SampleConsensusModel< PointT > > Ptr
Definition: sac_model.h:79
const std::string & getClassName() const
Get a string representation of the name of this class.
Definition: sac_model.h:341
SearchPtr samples_radius_search_
The search object for picking subsequent samples using radius search.
Definition: sac_model.h:571
virtual std::size_t countWithinDistance(const Eigen::VectorXf &model_coefficients, const double threshold) const =0
Count all the points which respect the given model coefficients as inliers.
double computeVariance(const std::vector< double > &error_sqr_dists) const
Compute the variance of the errors to the model.
Definition: sac_model.h:436
unsigned int sample_size_
The size of a sample from which the model is computed.
Definition: sac_model.h:589
typename PointCloud::ConstPtr PointCloudConstPtr
Definition: sac_model.h:75
IndicesPtr getIndices() const
Get a pointer to the vector of indices used.
Definition: sac_model.h:333
std::shared_ptr< boost::variate_generator< boost::mt19937 &, boost::uniform_int<> > > rng_gen_
Boost-based random number generator.
Definition: sac_model.h:583
IndicesPtr indices_
A pointer to the vector of point indices to use.
Definition: sac_model.h:556
double computeVariance() const
Compute the variance of the errors to the model from the internally estimated vector of distances.
Definition: sac_model.h:450
virtual void projectPoints(const Indices &inliers, const Eigen::VectorXf &model_coefficients, PointCloud &projected_points, bool copy_data_fields=true) const =0
Create a new point cloud with inliers projected onto the model.
void setModelConstraints(std::function< bool(const Eigen::VectorXf &)> function)
This can be used to impose any kind of constraint on the model, e.g.
Definition: sac_model.h:399
Indices shuffled_indices_
Data containing a shuffled version of the indices.
Definition: sac_model.h:574
boost::mt19937 rng_alg_
Boost-based random number generator algorithm.
Definition: sac_model.h:577
PointCloudConstPtr input_
A boost shared pointer to the point cloud data array.
Definition: sac_model.h:553
virtual bool isModelValid(const Eigen::VectorXf &model_coefficients) const
Check whether a model is valid given the user constraints.
Definition: sac_model.h:527
void setIndices(const IndicesPtr &indices)
Provide a pointer to the vector of indices that represents the input data.
Definition: sac_model.h:315
virtual bool doSamplesVerifyModel(const std::set< index_t > &indices, const Eigen::VectorXf &model_coefficients, const double threshold) const =0
Verify whether a subset of indices verifies a given set of model coefficients.
SampleConsensusModel(const PointCloudConstPtr &cloud, const Indices &indices, const bool random=false)
Constructor for base SampleConsensusModel.
Definition: sac_model.h:110
SampleConsensusModel(bool random=false)
Empty constructor for base SampleConsensusModel.
Definition: sac_model.h:86
virtual ~SampleConsensusModel()=default
Destructor for base SampleConsensusModel.
std::shared_ptr< boost::uniform_int<> > rng_dist_
Boost-based random number generator distribution.
Definition: sac_model.h:580
void setIndices(const Indices &indices)
Provide the vector of indices that represents the input data.
Definition: sac_model.h:325
double samples_radius_
The maximum distance of subsequent samples from the first (radius search)
Definition: sac_model.h:568
virtual void setInputCloud(const PointCloudConstPtr &cloud)
Provide a pointer to the input dataset.
Definition: sac_model.h:292
std::string model_name_
The model name.
Definition: sac_model.h:550
unsigned int model_size_
The number of coefficients in the model.
Definition: sac_model.h:592
int rnd()
Boost-based random number generator.
Definition: sac_model.h:596
void getRadiusLimits(double &min_radius, double &max_radius) const
Get the minimum and maximum allowable radius limits for the model as set by the user.
Definition: sac_model.h:388
typename pcl::search::Search< PointT >::Ptr SearchPtr
Definition: sac_model.h:77
void drawIndexSample(Indices &sample)
Fills a sample array with random samples from the indices_ vector.
Definition: sac_model.h:466
typename PointCloud::Ptr PointCloudPtr
Definition: sac_model.h:76
static unsigned int getMaxSampleChecks()
Definition: sac_model.h:354
shared_ptr< const SampleConsensusModel< PointT > > ConstPtr
Definition: sac_model.h:80
std::vector< double > error_sqr_dists_
A vector holding the distances to the computed model.
Definition: sac_model.h:586
void setSamplesMaxDist(const double &radius, SearchPtr search)
Set the maximum distance allowed when drawing random samples.
Definition: sac_model.h:414
virtual void getDistancesToModel(const Eigen::VectorXf &model_coefficients, std::vector< double > &distances) const =0
Compute all distances from the cloud data to a given model.
virtual void selectWithinDistance(const Eigen::VectorXf &model_coefficients, const double threshold, Indices &inliers)=0
Select all the points which respect the given model coefficients as inliers.
unsigned int getSampleSize() const
Return the size of a sample from which the model is computed.
Definition: sac_model.h:348
shared_ptr< pcl::search::Search< PointT > > Ptr
Definition: search.h:81
#define PCL_MAKE_ALIGNED_OPERATOR_NEW
Macro to signal a class requires a custom allocator.
Definition: memory.h:86
Defines functions, macros and traits for allocating and using memory.
detail::int_type_t< detail::index_type_size, detail::index_type_signed > index_t
Type used for an index in PCL.
Definition: types.h:112
SacModel
Definition: model_types.h:46
IndicesAllocator<> Indices
Type used for indices in PCL.
Definition: types.h:133
shared_ptr< Indices > IndicesPtr
Definition: pcl_base.h:58
#define PCL_DEPRECATED(Major, Minor, Message)
macro for compatibility across compilers and help remove old deprecated items for the Major....
Definition: pcl_macros.h:156
Base functor all the models that need non linear optimization must define their own one and implement...
Definition: sac_model.h:680
virtual ~Functor()=default
_Scalar Scalar
Definition: sac_model.h:681
int values() const
Get the number of values.
Definition: sac_model.h:704
Functor()
Empty Constructor.
Definition: sac_model.h:693
Eigen::Matrix< Scalar, InputsAtCompileTime, 1 > InputType
Definition: sac_model.h:689
@ InputsAtCompileTime
Definition: sac_model.h:684
@ ValuesAtCompileTime
Definition: sac_model.h:685
Eigen::Matrix< Scalar, ValuesAtCompileTime, 1 > ValueType
Definition: sac_model.h:688
Eigen::Matrix< Scalar, ValuesAtCompileTime, InputsAtCompileTime > JacobianType
Definition: sac_model.h:690
Functor(int m_data_points)
Constructor.
Definition: sac_model.h:698
A point structure representing Euclidean xyz coordinates, and the RGB color.
Defines basic non-point types used by PCL.