#include <hierarchical_clustering_index.h>
Classes | |
struct | Node |
struct | PointInfo |
Public Types | |
typedef NNIndex< Distance > | BaseClass |
typedef Distance::ResultType | DistanceType |
typedef Distance::ElementType | ElementType |
Public Member Functions | |
void | addPoints (const Matrix< ElementType > &points, float rebuild_threshold=2) |
Incrementally add points to the index. | |
BaseClass * | clone () const |
void | findNeighbors (ResultSet< DistanceType > &result, const ElementType *vec, const SearchParams &searchParams) const |
flann_algorithm_t | getType () const |
HierarchicalClusteringIndex (const IndexParams &index_params=HierarchicalClusteringIndexParams(), Distance d=Distance()) | |
HierarchicalClusteringIndex (const Matrix< ElementType > &inputData, const IndexParams &index_params=HierarchicalClusteringIndexParams(), Distance d=Distance()) | |
HierarchicalClusteringIndex (const HierarchicalClusteringIndex &other) | |
void | initCenterChooser () |
void | loadIndex (FILE *stream) |
HierarchicalClusteringIndex & | operator= (HierarchicalClusteringIndex other) |
void | saveIndex (FILE *stream) |
template<typename Archive > | |
void | serialize (Archive &ar) |
int | usedMemory () const |
virtual | ~HierarchicalClusteringIndex () |
Protected Member Functions | |
void | buildIndexImpl () |
Private Types | |
typedef BranchStruct< NodePtr, DistanceType > | BranchSt |
typedef Node * | NodePtr |
Private Member Functions | |
void | addPointToTree (NodePtr node, size_t index) |
void | computeClustering (NodePtr node, int *indices, int indices_length) |
void | computeLabels (int *indices, int indices_length, int *centers, int centers_length, int *labels, DistanceType &cost) |
void | copyTree (NodePtr &dst, const NodePtr &src) |
template<bool with_removed> | |
void | findNeighborsWithRemoved (ResultSet< DistanceType > &result, const ElementType *vec, const SearchParams &searchParams) const |
template<bool with_removed> | |
void | findNN (NodePtr node, ResultSet< DistanceType > &result, const ElementType *vec, int &checks, int maxChecks, Heap< BranchSt > *heap, DynamicBitset &checked) const |
void | freeIndex () |
void | swap (HierarchicalClusteringIndex &other) |
Private Attributes | |
int | branching_ |
flann_centers_init_t | centers_init_ |
CenterChooser< Distance > * | chooseCenters_ |
int | leaf_max_size_ |
int | memoryCounter_ |
PooledAllocator | pool_ |
std::vector< Node * > | tree_roots_ |
int | trees_ |
Hierarchical index
Contains a tree constructed through a hierarchical clustering and other information for indexing a set of points for nearest-neighbour matching.
Definition at line 86 of file hierarchical_clustering_index.h.
typedef NNIndex<Distance> rtflann::HierarchicalClusteringIndex< Distance >::BaseClass |
Definition at line 92 of file hierarchical_clustering_index.h.
typedef BranchStruct<NodePtr, DistanceType> rtflann::HierarchicalClusteringIndex< Distance >::BranchSt [private] |
Alias definition for a nicer syntax.
Definition at line 429 of file hierarchical_clustering_index.h.
typedef Distance::ResultType rtflann::HierarchicalClusteringIndex< Distance >::DistanceType |
Reimplemented from rtflann::NNIndex< Distance >.
Definition at line 90 of file hierarchical_clustering_index.h.
typedef Distance::ElementType rtflann::HierarchicalClusteringIndex< Distance >::ElementType |
Reimplemented from rtflann::NNIndex< Distance >.
Definition at line 89 of file hierarchical_clustering_index.h.
typedef Node* rtflann::HierarchicalClusteringIndex< Distance >::NodePtr [private] |
Definition at line 422 of file hierarchical_clustering_index.h.
rtflann::HierarchicalClusteringIndex< Distance >::HierarchicalClusteringIndex | ( | const IndexParams & | index_params = HierarchicalClusteringIndexParams() , |
Distance | d = Distance() |
||
) | [inline] |
Constructor.
index_params | |
d |
Definition at line 100 of file hierarchical_clustering_index.h.
rtflann::HierarchicalClusteringIndex< Distance >::HierarchicalClusteringIndex | ( | const Matrix< ElementType > & | inputData, |
const IndexParams & | index_params = HierarchicalClusteringIndexParams() , |
||
Distance | d = Distance() |
||
) | [inline] |
Index constructor
Params: inputData = dataset with the input features params = parameters passed to the hierarchical k-means algorithm
Definition at line 121 of file hierarchical_clustering_index.h.
rtflann::HierarchicalClusteringIndex< Distance >::HierarchicalClusteringIndex | ( | const HierarchicalClusteringIndex< Distance > & | other | ) | [inline] |
Definition at line 140 of file hierarchical_clustering_index.h.
virtual rtflann::HierarchicalClusteringIndex< Distance >::~HierarchicalClusteringIndex | ( | ) | [inline, virtual] |
Index destructor.
Release the memory used by the index.
Definition at line 187 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::addPoints | ( | const Matrix< ElementType > & | points, |
float | rebuild_threshold = 2 |
||
) | [inline, virtual] |
Incrementally add points to the index.
points | Matrix with points to be added |
rebuild_threshold |
Reimplemented from rtflann::NNIndex< Distance >.
Definition at line 209 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::addPointToTree | ( | NodePtr | node, |
size_t | index | ||
) | [inline, private] |
Definition at line 622 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::buildIndexImpl | ( | ) | [inline, protected, virtual] |
Builds the index
Implements rtflann::NNIndex< Distance >.
Definition at line 305 of file hierarchical_clustering_index.h.
BaseClass* rtflann::HierarchicalClusteringIndex< Distance >::clone | ( | ) | const [inline, virtual] |
Implements rtflann::NNIndex< Distance >.
Definition at line 193 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::computeClustering | ( | NodePtr | node, |
int * | indices, | ||
int | indices_length | ||
) | [inline, private] |
The method responsible with actually doing the recursive hierarchical clustering
Params: node = the node to cluster indices = indices of the points belonging to the current node branching = the branching factor to use in the clustering
Definition at line 490 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::computeLabels | ( | int * | indices, |
int | indices_length, | ||
int * | centers, | ||
int | centers_length, | ||
int * | labels, | ||
DistanceType & | cost | ||
) | [inline, private] |
Definition at line 462 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::copyTree | ( | NodePtr & | dst, |
const NodePtr & | src | ||
) | [inline, private] |
Definition at line 443 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::findNeighbors | ( | ResultSet< DistanceType > & | result, |
const ElementType * | vec, | ||
const SearchParams & | searchParams | ||
) | const [inline, virtual] |
Find set of nearest neighbors to vec. Their indices are stored inside the result object.
Params: result = the result object in which the indices of the nearest-neighbors are stored vec = the vector for which to search the nearest neighbors searchParams = parameters that influence the search algorithm (checks)
Implements rtflann::NNIndex< Distance >.
Definition at line 290 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::findNeighborsWithRemoved | ( | ResultSet< DistanceType > & | result, |
const ElementType * | vec, | ||
const SearchParams & | searchParams | ||
) | const [inline, private] |
Definition at line 546 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::findNN | ( | NodePtr | node, |
ResultSet< DistanceType > & | result, | ||
const ElementType * | vec, | ||
int & | checks, | ||
int | maxChecks, | ||
Heap< BranchSt > * | heap, | ||
DynamicBitset & | checked | ||
) | const [inline, private] |
Performs one descent in the hierarchical k-means tree. The branches not visited are stored in a priority queue.
Params: node = node to explore result = container for the k-nearest neighbors found vec = query points checks = how many points in the dataset have been checked so far maxChecks = maximum dataset points to checks
Definition at line 582 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::freeIndex | ( | ) | [inline, private, virtual] |
Clears Node tree calling Node destructor explicitly
Implements rtflann::NNIndex< Distance >.
Definition at line 436 of file hierarchical_clustering_index.h.
flann_algorithm_t rtflann::HierarchicalClusteringIndex< Distance >::getType | ( | ) | const [inline, virtual] |
Implements rtflann::IndexBase.
Definition at line 229 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::initCenterChooser | ( | ) | [inline] |
Definition at line 162 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::loadIndex | ( | FILE * | stream | ) | [inline, virtual] |
Implements rtflann::IndexBase.
Definition at line 273 of file hierarchical_clustering_index.h.
HierarchicalClusteringIndex& rtflann::HierarchicalClusteringIndex< Distance >::operator= | ( | HierarchicalClusteringIndex< Distance > | other | ) | [inline] |
Definition at line 155 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::saveIndex | ( | FILE * | stream | ) | [inline, virtual] |
Implements rtflann::IndexBase.
Definition at line 266 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::serialize | ( | Archive & | ar | ) | [inline] |
Reimplemented from rtflann::NNIndex< Distance >.
Definition at line 236 of file hierarchical_clustering_index.h.
void rtflann::HierarchicalClusteringIndex< Distance >::swap | ( | HierarchicalClusteringIndex< Distance > & | other | ) | [inline, private] |
Definition at line 658 of file hierarchical_clustering_index.h.
int rtflann::HierarchicalClusteringIndex< Distance >::usedMemory | ( | ) | const [inline, virtual] |
Computes the inde memory usage Returns: memory used by the index
Implements rtflann::IndexBase.
Definition at line 202 of file hierarchical_clustering_index.h.
int rtflann::HierarchicalClusteringIndex< Distance >::branching_ [private] |
index parameters Branching factor to use for clustering
Definition at line 697 of file hierarchical_clustering_index.h.
flann_centers_init_t rtflann::HierarchicalClusteringIndex< Distance >::centers_init_ [private] |
Algorithm to use for choosing cluster centers
Definition at line 707 of file hierarchical_clustering_index.h.
CenterChooser<Distance>* rtflann::HierarchicalClusteringIndex< Distance >::chooseCenters_ [private] |
Algorithm used to choose initial centers
Definition at line 717 of file hierarchical_clustering_index.h.
int rtflann::HierarchicalClusteringIndex< Distance >::leaf_max_size_ [private] |
Max size of leaf nodes
Definition at line 712 of file hierarchical_clustering_index.h.
int rtflann::HierarchicalClusteringIndex< Distance >::memoryCounter_ [private] |
Memory occupied by the index.
Definition at line 691 of file hierarchical_clustering_index.h.
PooledAllocator rtflann::HierarchicalClusteringIndex< Distance >::pool_ [private] |
Pooled memory allocator.
Using a pooled memory allocator is more efficient than allocating memory directly when there is a large number small of memory allocations.
Definition at line 686 of file hierarchical_clustering_index.h.
std::vector<Node*> rtflann::HierarchicalClusteringIndex< Distance >::tree_roots_ [private] |
The root nodes in the tree.
Definition at line 677 of file hierarchical_clustering_index.h.
int rtflann::HierarchicalClusteringIndex< Distance >::trees_ [private] |
How many parallel trees to build
Definition at line 702 of file hierarchical_clustering_index.h.