Point Cloud Library (PCL)  1.12.0
rgb_plane_coefficient_comparator.h
1 /*
2  * Software License Agreement (BSD License)
3  *
4  * Point Cloud Library (PCL) - www.pointclouds.org
5  * Copyright (c) 2010-2012, Willow Garage, Inc.
6  *
7  * All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  *
13  * * Redistributions of source code must retain the above copyright
14  * notice, this list of conditions and the following disclaimer.
15  * * Redistributions in binary form must reproduce the above
16  * copyright notice, this list of conditions and the following
17  * disclaimer in the documentation and/or other materials provided
18  * with the distribution.
19  * * Neither the name of the copyright holder(s) nor the names of its
20  * contributors may be used to endorse or promote products derived
21  * from this software without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
26  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
27  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
28  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
29  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
30  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
31  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
33  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
34  * POSSIBILITY OF SUCH DAMAGE.
35  *
36  *
37  *
38  */
39 
40 #pragma once
41 
42 #include <pcl/segmentation/plane_coefficient_comparator.h>
43 
44 namespace pcl
45 {
46  /** \brief RGBPlaneCoefficientComparator is a Comparator that operates on plane coefficients,
47  * for use in planar segmentation. Also takes into account RGB, so we can segmented different colored co-planar regions.
48  * In conjunction with OrganizedConnectedComponentSegmentation, this allows planes to be segmented from organized data.
49  *
50  * \author Alex Trevor
51  */
52  template<typename PointT, typename PointNT>
54  {
55  public:
58 
60  using PointCloudNPtr = typename PointCloudN::Ptr;
62 
63  using Ptr = shared_ptr<RGBPlaneCoefficientComparator<PointT, PointNT> >;
64  using ConstPtr = shared_ptr<const RGBPlaneCoefficientComparator<PointT, PointNT> >;
65 
70 
71  /** \brief Empty constructor for RGBPlaneCoefficientComparator. */
73  : color_threshold_ (50.0f)
74  {
75  }
76 
77  /** \brief Constructor for RGBPlaneCoefficientComparator.
78  * \param[in] plane_coeff_d a reference to a vector of d coefficients of plane equations. Must be the same size as the input cloud and input normals. a, b, and c coefficients are in the input normals.
79  */
80  RGBPlaneCoefficientComparator (shared_ptr<std::vector<float> >& plane_coeff_d)
81  : PlaneCoefficientComparator<PointT, PointNT> (plane_coeff_d), color_threshold_ (50.0f)
82  {
83  }
84 
85  /** \brief Destructor for RGBPlaneCoefficientComparator. */
86 
88  {
89  }
90 
91  /** \brief Set the tolerance in color space between neighboring points, to be considered part of the same plane.
92  * \param[in] color_threshold The distance in color space
93  */
94  inline void
95  setColorThreshold (float color_threshold)
96  {
97  color_threshold_ = color_threshold * color_threshold;
98  }
99 
100  /** \brief Get the color threshold between neighboring points, to be considered part of the same plane. */
101  inline float
103  {
104  return (color_threshold_);
105  }
106 
107  /** \brief Compare two neighboring points, by using normal information, euclidean distance, and color information.
108  * \param[in] idx1 The index of the first point.
109  * \param[in] idx2 The index of the second point.
110  */
111  bool
112  compare (int idx1, int idx2) const override
113  {
114  float dx = (*input_)[idx1].x - (*input_)[idx2].x;
115  float dy = (*input_)[idx1].y - (*input_)[idx2].y;
116  float dz = (*input_)[idx1].z - (*input_)[idx2].z;
117  float dist = std::sqrt (dx*dx + dy*dy + dz*dz);
118  int dr = (*input_)[idx1].r - (*input_)[idx2].r;
119  int dg = (*input_)[idx1].g - (*input_)[idx2].g;
120  int db = (*input_)[idx1].b - (*input_)[idx2].b;
121  //Note: This is not the best metric for color comparisons, we should probably use HSV space.
122  float color_dist = static_cast<float> (dr*dr + dg*dg + db*db);
123  return ( (dist < distance_threshold_)
124  && ((*normals_)[idx1].getNormalVector3fMap ().dot ((*normals_)[idx2].getNormalVector3fMap () ) > angular_threshold_ )
125  && (color_dist < color_threshold_));
126  }
127 
128  protected:
130  };
131 }
Comparator is the base class for comparators that compare two points given some function.
Definition: comparator.h:55
shared_ptr< Comparator< PointT > > Ptr
Definition: comparator.h:61
typename PointCloud::ConstPtr PointCloudConstPtr
Definition: comparator.h:59
shared_ptr< const Comparator< PointT > > ConstPtr
Definition: comparator.h:62
PlaneCoefficientComparator is a Comparator that operates on plane coefficients, for use in planar seg...
typename PointCloudN::ConstPtr PointCloudNConstPtr
PointCloud represents the base class in PCL for storing collections of 3D points.
Definition: point_cloud.h:173
shared_ptr< PointCloud< PointNT > > Ptr
Definition: point_cloud.h:413
shared_ptr< const PointCloud< PointNT > > ConstPtr
Definition: point_cloud.h:414
RGBPlaneCoefficientComparator is a Comparator that operates on plane coefficients,...
bool compare(int idx1, int idx2) const override
Compare two neighboring points, by using normal information, euclidean distance, and color informatio...
RGBPlaneCoefficientComparator()
Empty constructor for RGBPlaneCoefficientComparator.
RGBPlaneCoefficientComparator(shared_ptr< std::vector< float > > &plane_coeff_d)
Constructor for RGBPlaneCoefficientComparator.
void setColorThreshold(float color_threshold)
Set the tolerance in color space between neighboring points, to be considered part of the same plane.
float getColorThreshold() const
Get the color threshold between neighboring points, to be considered part of the same plane.
~RGBPlaneCoefficientComparator()
Destructor for RGBPlaneCoefficientComparator.
A point structure representing Euclidean xyz coordinates, and the RGB color.