216 lines
8.0 KiB
C++
216 lines
8.0 KiB
C++
/*
|
|
* Software License Agreement (BSD License)
|
|
*
|
|
* Point Cloud Library (PCL) - www.pointclouds.org
|
|
* Copyright (c) 2013-, Open Perception Inc.
|
|
*
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above
|
|
* copyright notice, this list of conditions and the following
|
|
* disclaimer in the documentation and/or other materials provided
|
|
* with the distribution.
|
|
* * Neither the name of Willow Garage, Inc. nor the names of its
|
|
* contributors may be used to endorse or promote products derived
|
|
* from this software without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
|
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
|
|
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
|
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <pcl/keypoints/keypoint.h>
|
|
#include <pcl/common/intensity.h>
|
|
|
|
namespace pcl
|
|
{
|
|
/** \brief TrajkovicKeypoint3D implements Trajkovic and Hedley corner detector on
|
|
* point cloud using geometric information.
|
|
* It uses first order statistics to find variation of normals.
|
|
* This work is part of Nizar Sallem PhD thesis.
|
|
*
|
|
* \author Nizar Sallem
|
|
* \ingroup keypoints
|
|
*/
|
|
template <typename PointInT, typename PointOutT, typename NormalT = pcl::Normal>
|
|
class TrajkovicKeypoint3D : public Keypoint<PointInT, PointOutT>
|
|
{
|
|
public:
|
|
using Ptr = shared_ptr<TrajkovicKeypoint3D<PointInT, PointOutT, NormalT> >;
|
|
using ConstPtr = shared_ptr<const TrajkovicKeypoint3D<PointInT, PointOutT, NormalT> >;
|
|
using PointCloudIn = typename Keypoint<PointInT, PointOutT>::PointCloudIn;
|
|
using PointCloudOut = typename Keypoint<PointInT, PointOutT>::PointCloudOut;
|
|
using PointCloudInConstPtr = typename PointCloudIn::ConstPtr;
|
|
using Normals = pcl::PointCloud<NormalT>;
|
|
using NormalsPtr = typename Normals::Ptr;
|
|
using NormalsConstPtr = typename Normals::ConstPtr;
|
|
|
|
using Keypoint<PointInT, PointOutT>::name_;
|
|
using Keypoint<PointInT, PointOutT>::input_;
|
|
using Keypoint<PointInT, PointOutT>::indices_;
|
|
using Keypoint<PointInT, PointOutT>::keypoints_indices_;
|
|
using Keypoint<PointInT, PointOutT>::initCompute;
|
|
|
|
enum ComputationMethod { FOUR_CORNERS, EIGHT_CORNERS };
|
|
|
|
/** \brief Constructor
|
|
* \param[in] method the method to be used to determine the corner responses
|
|
* \param[in] window_size
|
|
* \param[in] first_threshold the threshold used in the simple cornerness test.
|
|
* \param[in] second_threshold the threshold used to reject weak corners.
|
|
*/
|
|
TrajkovicKeypoint3D (ComputationMethod method = FOUR_CORNERS,
|
|
int window_size = 3,
|
|
float first_threshold = 0.00046,
|
|
float second_threshold = 0.03589)
|
|
: method_ (method)
|
|
, window_size_ (window_size)
|
|
, first_threshold_ (first_threshold)
|
|
, second_threshold_ (second_threshold)
|
|
, threads_ (1)
|
|
{
|
|
name_ = "TrajkovicKeypoint3D";
|
|
}
|
|
|
|
/** \brief set the method of the response to be calculated.
|
|
* \param[in] method either 4 corners or 8 corners
|
|
*/
|
|
inline void
|
|
setMethod (ComputationMethod method) { method_ = method; }
|
|
|
|
/// \brief \return the computation method
|
|
inline ComputationMethod
|
|
getMethod () const { return (method_); }
|
|
|
|
/// \brief Set window size
|
|
inline void
|
|
setWindowSize (int window_size) { window_size_= window_size; }
|
|
|
|
/// \brief \return window size i.e. window width or height
|
|
inline int
|
|
getWindowSize () const { return (window_size_); }
|
|
|
|
/** \brief set the first_threshold to reject corners in the simple cornerness
|
|
* computation stage.
|
|
* \param[in] threshold
|
|
*/
|
|
inline void
|
|
setFirstThreshold (float threshold) { first_threshold_= threshold; }
|
|
|
|
/// \brief \return first threshold
|
|
inline float
|
|
getFirstThreshold () const { return (first_threshold_); }
|
|
|
|
/** \brief set the second threshold to reject corners in the final cornerness
|
|
* computation stage.
|
|
* \param[in] threshold
|
|
*/
|
|
inline void
|
|
setSecondThreshold (float threshold) { second_threshold_= threshold; }
|
|
|
|
/// \brief \return second threshold
|
|
inline float
|
|
getSecondThreshold () const { return (second_threshold_); }
|
|
|
|
/** \brief Set normals if precalculated normals are available.
|
|
* \param normals
|
|
*/
|
|
inline void
|
|
setNormals (const NormalsConstPtr &normals) { normals_ = normals; }
|
|
|
|
/// \brief \return points normals as calculated or given
|
|
inline void
|
|
getNormals () const { return (normals_); }
|
|
|
|
/** \brief Initialize the scheduler and set the number of threads to use.
|
|
* \param nr_threads the number of hardware threads to use, 0 for automatic.
|
|
*/
|
|
inline void
|
|
setNumberOfThreads (unsigned int nr_threads = 0) { threads_ = nr_threads; }
|
|
|
|
/// \brief \return the number of threads
|
|
inline unsigned int
|
|
getNumberOfThreads () const { return (threads_); }
|
|
|
|
protected:
|
|
bool
|
|
initCompute () override;
|
|
|
|
void
|
|
detectKeypoints (PointCloudOut &output) override;
|
|
|
|
private:
|
|
/** Return a const reference to the normal at (i,j) if it is finite else return
|
|
* a reference to a null normal.
|
|
* If the returned normal is valid \a counter is incremented.
|
|
*/
|
|
inline const NormalT&
|
|
getNormalOrNull (int i, int j, int& counter) const
|
|
{
|
|
static const NormalT null;
|
|
if (!isFinite ((*normals_) (i,j))) return (null);
|
|
++counter;
|
|
return ((*normals_) (i,j));
|
|
}
|
|
/// \return difference of two normals vectors
|
|
inline float
|
|
normalsDiff (const NormalT& a, const NormalT& b) const
|
|
{
|
|
double nx = a.normal_x; double ny = a.normal_y; double nz = a.normal_z;
|
|
double mx = b.normal_x; double my = b.normal_y; double mz = b.normal_z;
|
|
return (static_cast<float> (1.0 - (nx*mx + ny*my + nz*mz)));
|
|
}
|
|
/// \return squared difference of two normals vectors
|
|
inline float
|
|
squaredNormalsDiff (const NormalT& a, const NormalT& b) const
|
|
{
|
|
float diff = normalsDiff (a,b);
|
|
return (diff * diff);
|
|
}
|
|
/** Comparator for responses intensity
|
|
* \return true if \a response_ at index \aa is greater than response at index \ab
|
|
*/
|
|
inline bool
|
|
greaterCornernessAtIndices (int a, int b) const
|
|
{
|
|
return (response_->points [a] > response_->points [b]);
|
|
}
|
|
/// computation method
|
|
ComputationMethod method_;
|
|
/// window size
|
|
int window_size_;
|
|
/// half window size
|
|
int half_window_size_;
|
|
/// first threshold for quick rejection
|
|
float first_threshold_;
|
|
/// second threshold for corner evaluation
|
|
float second_threshold_;
|
|
/// number of threads to be used
|
|
unsigned int threads_;
|
|
/// point cloud normals
|
|
NormalsConstPtr normals_;
|
|
/// point cloud response
|
|
pcl::PointCloud<float>::Ptr response_;
|
|
};
|
|
}
|
|
|
|
#include <pcl/keypoints/impl/trajkovic_3d.hpp>
|