fixed memory leak in flann tests

HarshKhaiwal · Jul 12, 2016 · bcc9010 · bcc9010
1 parent 02aabcc
commit bcc9010
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/modules/flann/src/miniflann.cpp b/modules/flann/src/miniflann.cpp
@@ -318,7 +318,19 @@ buildIndex_(void*& index, const Mat& data, const IndexParams& params, const Dist
 
     ::cvflann::Matrix<ElementType> dataset((ElementType*)data.data, data.rows, data.cols);
     IndexType* _index = new IndexType(dataset, get_params(params), dist);
-    _index->buildIndex();
+
+    try
+    {
+        _index->buildIndex();
+    }
+    catch (...)
+    {
+        delete _index;
+        _index = NULL;
+
+        throw;
+    }
+
     index = _index;
 }