miniflann.hpp 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158
  1. /*M///////////////////////////////////////////////////////////////////////////////////////
  2. //
  3. // IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
  4. //
  5. // By downloading, copying, installing or using the software you agree to this license.
  6. // If you do not agree to this license, do not download, install,
  7. // copy or use the software.
  8. //
  9. //
  10. // License Agreement
  11. // For Open Source Computer Vision Library
  12. //
  13. // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
  14. // Copyright (C) 2009, Willow Garage Inc., all rights reserved.
  15. // Third party copyrights are property of their respective owners.
  16. //
  17. // Redistribution and use in source and binary forms, with or without modification,
  18. // are permitted provided that the following conditions are met:
  19. //
  20. // * Redistribution's of source code must retain the above copyright notice,
  21. // this list of conditions and the following disclaimer.
  22. //
  23. // * Redistribution's in binary form must reproduce the above copyright notice,
  24. // this list of conditions and the following disclaimer in the documentation
  25. // and/or other materials provided with the distribution.
  26. //
  27. // * The name of the copyright holders may not be used to endorse or promote products
  28. // derived from this software without specific prior written permission.
  29. //
  30. // This software is provided by the copyright holders and contributors "as is" and
  31. // any express or implied warranties, including, but not limited to, the implied
  32. // warranties of merchantability and fitness for a particular purpose are disclaimed.
  33. // In no event shall the Intel Corporation or contributors be liable for any direct,
  34. // indirect, incidental, special, exemplary, or consequential damages
  35. // (including, but not limited to, procurement of substitute goods or services;
  36. // loss of use, data, or profits; or business interruption) however caused
  37. // and on any theory of liability, whether in contract, strict liability,
  38. // or tort (including negligence or otherwise) arising in any way out of
  39. // the use of this software, even if advised of the possibility of such damage.
  40. //
  41. //M*/
  42. #ifndef _OPENCV_MINIFLANN_HPP_
  43. #define _OPENCV_MINIFLANN_HPP_
  44. #include "opencv2/core.hpp"
  45. #include "opencv2/flann/defines.h"
  46. namespace cv
  47. {
  48. namespace flann
  49. {
  50. struct CV_EXPORTS IndexParams
  51. {
  52. IndexParams();
  53. ~IndexParams();
  54. String getString(const String& key, const String& defaultVal=String()) const;
  55. int getInt(const String& key, int defaultVal=-1) const;
  56. double getDouble(const String& key, double defaultVal=-1) const;
  57. void setString(const String& key, const String& value);
  58. void setInt(const String& key, int value);
  59. void setDouble(const String& key, double value);
  60. void setFloat(const String& key, float value);
  61. void setBool(const String& key, bool value);
  62. void setAlgorithm(int value);
  63. void getAll(std::vector<String>& names,
  64. std::vector<int>& types,
  65. std::vector<String>& strValues,
  66. std::vector<double>& numValues) const;
  67. void* params;
  68. };
  69. struct CV_EXPORTS KDTreeIndexParams : public IndexParams
  70. {
  71. KDTreeIndexParams(int trees=4);
  72. };
  73. struct CV_EXPORTS LinearIndexParams : public IndexParams
  74. {
  75. LinearIndexParams();
  76. };
  77. struct CV_EXPORTS CompositeIndexParams : public IndexParams
  78. {
  79. CompositeIndexParams(int trees = 4, int branching = 32, int iterations = 11,
  80. cvflann::flann_centers_init_t centers_init = cvflann::FLANN_CENTERS_RANDOM, float cb_index = 0.2f );
  81. };
  82. struct CV_EXPORTS AutotunedIndexParams : public IndexParams
  83. {
  84. AutotunedIndexParams(float target_precision = 0.8f, float build_weight = 0.01f,
  85. float memory_weight = 0, float sample_fraction = 0.1f);
  86. };
  87. struct CV_EXPORTS HierarchicalClusteringIndexParams : public IndexParams
  88. {
  89. HierarchicalClusteringIndexParams(int branching = 32,
  90. cvflann::flann_centers_init_t centers_init = cvflann::FLANN_CENTERS_RANDOM, int trees = 4, int leaf_size = 100 );
  91. };
  92. struct CV_EXPORTS KMeansIndexParams : public IndexParams
  93. {
  94. KMeansIndexParams(int branching = 32, int iterations = 11,
  95. cvflann::flann_centers_init_t centers_init = cvflann::FLANN_CENTERS_RANDOM, float cb_index = 0.2f );
  96. };
  97. struct CV_EXPORTS LshIndexParams : public IndexParams
  98. {
  99. LshIndexParams(int table_number, int key_size, int multi_probe_level);
  100. };
  101. struct CV_EXPORTS SavedIndexParams : public IndexParams
  102. {
  103. SavedIndexParams(const String& filename);
  104. };
  105. struct CV_EXPORTS SearchParams : public IndexParams
  106. {
  107. SearchParams( int checks = 32, float eps = 0, bool sorted = true );
  108. };
  109. class CV_EXPORTS_W Index
  110. {
  111. public:
  112. CV_WRAP Index();
  113. CV_WRAP Index(InputArray features, const IndexParams& params, cvflann::flann_distance_t distType=cvflann::FLANN_DIST_L2);
  114. virtual ~Index();
  115. CV_WRAP virtual void build(InputArray features, const IndexParams& params, cvflann::flann_distance_t distType=cvflann::FLANN_DIST_L2);
  116. CV_WRAP virtual void knnSearch(InputArray query, OutputArray indices,
  117. OutputArray dists, int knn, const SearchParams& params=SearchParams());
  118. CV_WRAP virtual int radiusSearch(InputArray query, OutputArray indices,
  119. OutputArray dists, double radius, int maxResults,
  120. const SearchParams& params=SearchParams());
  121. CV_WRAP virtual void save(const String& filename) const;
  122. CV_WRAP virtual bool load(InputArray features, const String& filename);
  123. CV_WRAP virtual void release();
  124. CV_WRAP cvflann::flann_distance_t getDistance() const;
  125. CV_WRAP cvflann::flann_algorithm_t getAlgorithm() const;
  126. protected:
  127. cvflann::flann_distance_t distType;
  128. cvflann::flann_algorithm_t algo;
  129. int featureType;
  130. void* index;
  131. };
  132. } } // namespace cv::flann
  133. #endif