NearestNeighbors.h
1 /*********************************************************************
2 * Software License Agreement (BSD License)
3 *
4 * Copyright (c) 2008, Willow Garage, Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * * Redistributions in binary form must reproduce the above
14 * copyright notice, this list of conditions and the following
15 * disclaimer in the documentation and/or other materials provided
16 * with the distribution.
17 * * Neither the name of the Willow Garage nor the names of its
18 * contributors may be used to endorse or promote products derived
19 * from this software without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
25 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
27 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
29 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
31 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32 * POSSIBILITY OF SUCH DAMAGE.
33 *********************************************************************/
34 
35 /* Author: Ioan Sucan */
36 
37 #ifndef OMPL_DATASTRUCTURES_NEAREST_NEIGHBORS_
38 #define OMPL_DATASTRUCTURES_NEAREST_NEIGHBORS_
39 
40 #include <vector>
41 #include <functional>
42 
43 namespace ompl
44 {
46  template <typename _T>
47  class NearestNeighbors
48  {
49  public:
51  using DistanceFunction = std::function<double(const _T &, const _T &)>;
52 
53  NearestNeighbors() = default;
54 
55  virtual ~NearestNeighbors() = default;
56 
58  virtual void setDistanceFunction(const DistanceFunction &distFun)
59  {
60  distFun_ = distFun;
61  }
62 
65  {
66  return distFun_;
67  }
68 
71  virtual bool reportsSortedResults() const = 0;
72 
74  virtual void clear() = 0;
75 
77  virtual void add(const _T &data) = 0;
78 
80  virtual void add(const std::vector<_T> &data)
81  {
82  for (const auto &elt : data)
83  add(elt);
84  }
85 
87  virtual bool remove(const _T &data) = 0;
88 
90  virtual _T nearest(const _T &data) const = 0;
91 
97  virtual void nearestK(const _T &data, std::size_t k, std::vector<_T> &nbh) const = 0;
98 
104  virtual void nearestR(const _T &data, double radius, std::vector<_T> &nbh) const = 0;
105 
107  virtual std::size_t size() const = 0;
108 
110  virtual void list(std::vector<_T> &data) const = 0;
111 
112  protected:
115  };
116 }
117 
118 #endif
virtual _T nearest(const _T &data) const =0
Get the nearest neighbor of a point.
virtual void add(const _T &data)=0
Add an element to the datastructure.
virtual bool reportsSortedResults() const =0
Return true if the solutions reported by this data structure are sorted, when calling nearestK / near...
virtual bool remove(const _T &data)=0
Remove an element from the datastructure.
const DistanceFunction & getDistanceFunction() const
Get the distance function used.
std::function< double(const _T &, const _T &)> DistanceFunction
The definition of a distance function.
virtual void nearestK(const _T &data, std::size_t k, std::vector< _T > &nbh) const =0
Get the k-nearest neighbors of a point.
DistanceFunction distFun_
The used distance function.
virtual void nearestR(const _T &data, double radius, std::vector< _T > &nbh) const =0
Get the nearest neighbors of a point, within a specified radius.
virtual void list(std::vector< _T > &data) const =0
Get all the elements in the datastructure.
virtual void clear()=0
Clear the datastructure.
virtual std::size_t size() const =0
Get the number of elements in the datastructure.
Main namespace. Contains everything in this library.
virtual void setDistanceFunction(const DistanceFunction &distFun)
Set the distance function to use.