Update to 2.0.0 tree from current Fremantle build

[opencv] / tests / cv / src / akdtree.cpp

diff --git a/tests/cv/src/akdtree.cpp b/tests/cv/src/akdtree.cpp
new file mode 100644 (file)
index 0000000..8a5d7c1
--- /dev/null
+++ b/tests/cv/src/akdtree.cpp
@@ -0,0 +1,74 @@
+// 2009-01-14, Xavier Delacour <xavier.delacour@gmail.com>
+
+#include "cvtest.h"
+
+// * add test for cvFindFeaturesBoxed
+
+#include <algorithm>
+#include <vector>
+#include <iostream>
+
+class CV_KdtreeTest : public CvTest {
+public:
+  CV_KdtreeTest();
+  ~CV_KdtreeTest();
+protected:
+  virtual void run( int start_from );
+};
+
+CV_KdtreeTest::CV_KdtreeTest()
+: CvTest( "kd-tree", "cvFindFeatures-kd" ) {
+}
+
+CV_KdtreeTest::~CV_KdtreeTest() {
+}
+
+
+void CV_KdtreeTest::run( int /*start_from*/ ) {
+  int dims = 64;
+  int features = 2000;
+  int k = 1; // * should also test 2nd nn etc.?
+  int emax = 20;
+  double noise = .2;
+  int points = 1000;
+
+  CvRNG rng = cvRNG();
+  CvMat* desc = cvCreateMat(features, dims, CV_64FC1);
+  cvRandArr( &rng, desc, CV_RAND_UNI, cvRealScalar(0), cvRealScalar(1));
+
+  CvFeatureTree* tr = cvCreateKDTree(desc);
+  CvMat* results = cvCreateMat(points, k, CV_32SC1);
+  CvMat* dist = cvCreateMat(points, k, CV_64FC1);
+
+  CvMat* pts = cvCreateMat(points, dims, CV_64FC1);
+  std::vector<int> fmap(points);
+  for (int j = 0; j < points; ++j) {
+    int fi = cvRandInt(&rng) % features;
+    fmap[j] = fi;
+    double* f = (double*)cvPtr2D(desc, fi, 0);
+    double* p = (double*)cvPtr2D(pts, j, 0);
+    for (int k = 0; k < dims; ++ k)
+      p[k] = f[k] + cvRandReal(&rng) * noise;
+  }
+
+  cvFindFeatures(tr, pts, results, dist, k, emax);
+
+  int correct_matches = 0;
+  { // Aisle "j" to avoid error on MSVC6
+    for (int j = 0; j < points; ++j) {
+      int fi = (int)cvGetReal2D(results, j, 0);
+      if (fmap[j] == fi)
+        ++correct_matches;
+    }
+  }
+
+  double correct_perc = correct_matches / (double)points;
+  ts->printf( CvTS::LOG, "correct_perc = %d\n", correct_perc );
+  if (correct_perc < .8)
+    ts->set_failed_test_info(CvTS::FAIL_INVALID_OUTPUT);
+
+  cvReleaseFeatureTree(tr);
+}
+
+
+CV_KdtreeTest kdtree_test;