bfknn.hpp

Go to the documentation of this file.

#pragma once
 
#include <cpp-toolbox/pcl/knn/base_knn.hpp>
#include <cpp-toolbox/metrics/metric_factory.hpp>
#include <cpp-toolbox/metrics/vector_metrics.hpp>
 
namespace toolbox::pcl
{
 
template<typename Element, typename Metric = toolbox::metrics::L2Metric<typename Element::value_type>>
class CPP_TOOLBOX_EXPORT bfknn_generic_t : public base_knn_generic_t<bfknn_generic_t<Element, Metric>, Element, Metric>
{
public:
  using base_type = base_knn_generic_t<bfknn_generic_t<Element, Metric>, Element, Metric>;
  using traits_type = typename base_type::traits_type;
  using element_type = typename traits_type::element_type;
  using metric_type = typename traits_type::metric_type;
  using distance_type = typename traits_type::distance_type;
  using container_type = typename base_type::container_type;
  using container_ptr = typename base_type::container_ptr;
  using value_type = typename Element::value_type;
 
  bfknn_generic_t() = default;
  ~bfknn_generic_t() = default;
 
  bfknn_generic_t(const bfknn_generic_t&) = delete;
  bfknn_generic_t& operator=(const bfknn_generic_t&) = delete;
  bfknn_generic_t(bfknn_generic_t&&) = delete;
  bfknn_generic_t& operator=(bfknn_generic_t&&) = delete;
 
  std::size_t set_input_impl(const container_type& data);
  
  std::size_t set_input_impl(const container_ptr& data);
  
  void set_metric_impl(const metric_type& metric);
  
  template<typename T = typename Element::value_type>
  void set_metric_impl(std::shared_ptr<toolbox::metrics::IMetric<T>> metric);
  
  template<typename T = typename Element::value_type>
  void set_metric_impl(std::unique_ptr<toolbox::metrics::IMetric<T>> metric);
 
  bool kneighbors_impl(const element_type& query,
                       std::size_t num_neighbors,
                       std::vector<std::size_t>& indices,
                       std::vector<distance_type>& distances);
 
  bool radius_neighbors_impl(const element_type& query,
                             distance_type radius,
                             std::vector<std::size_t>& indices,
                             std::vector<distance_type>& distances);
 
private:
  container_ptr m_data;  
  metric_type m_compile_time_metric;  
  std::shared_ptr<toolbox::metrics::IMetric<typename Element::value_type>> m_runtime_metric;  
  bool m_use_runtime_metric = false;  
};
 
template<typename DataType>
using bfknn_t = bfknn_generic_t<point_t<DataType>, toolbox::metrics::L2Metric<DataType>>;
 
}  // namespace toolbox::pcl
 
#include <cpp-toolbox/pcl/knn/impl/bfknn_impl.hpp>