summaryrefslogtreecommitdiff
path: root/geom_matching/wasserstein/include/dnn
diff options
context:
space:
mode:
Diffstat (limited to 'geom_matching/wasserstein/include/dnn')
-rw-r--r--geom_matching/wasserstein/include/dnn/geometry/euclidean-fixed.h2
-rw-r--r--geom_matching/wasserstein/include/dnn/local/kd-tree.hpp9
-rw-r--r--geom_matching/wasserstein/include/dnn/parallel/tbb.h2
3 files changed, 5 insertions, 8 deletions
diff --git a/geom_matching/wasserstein/include/dnn/geometry/euclidean-fixed.h b/geom_matching/wasserstein/include/dnn/geometry/euclidean-fixed.h
index a6ccef7..e2c5b44 100644
--- a/geom_matching/wasserstein/include/dnn/geometry/euclidean-fixed.h
+++ b/geom_matching/wasserstein/include/dnn/geometry/euclidean-fixed.h
@@ -7,7 +7,7 @@
#include <boost/serialization/access.hpp>
#include <boost/serialization/base_object.hpp>
-#include <iostream>
+//#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
diff --git a/geom_matching/wasserstein/include/dnn/local/kd-tree.hpp b/geom_matching/wasserstein/include/dnn/local/kd-tree.hpp
index 151a4ad..6b0852c 100644
--- a/geom_matching/wasserstein/include/dnn/local/kd-tree.hpp
+++ b/geom_matching/wasserstein/include/dnn/local/kd-tree.hpp
@@ -6,6 +6,7 @@
#include <stack>
#include "../parallel/tbb.h"
+#include "def_debug_ws.h"
template<class T>
dnn::KDTree<T>::
@@ -127,13 +128,8 @@ search(PointHandle q, ResultsFunctor& rf) const
// TODO: use tbb::scalable_allocator for the queue
std::queue<KDTreeNode> nodes;
-
-
nodes.push(KDTreeNode(tree_.begin(), tree_.end(), 0));
-
- //std::cout << "started kdtree::search" << std::endl;
-
while (!nodes.empty())
{
HCIterator b, e; size_t i;
@@ -163,7 +159,6 @@ search(PointHandle q, ResultsFunctor& rf) const
nodes.push(KDTreeNode(b, m, i));
}
}
- //std::cout << "exited kdtree::search" << std::endl;
}
template<class T>
@@ -290,6 +285,7 @@ void
dnn::KDTree<T>::
printWeights(void)
{
+#ifndef FOR_R_TDA
std::cout << "weights_:" << std::endl;
for(const auto ph : indices_) {
std::cout << "idx = " << ph.second << ": (" << (ph.first)->at(0) << ", " << (ph.first)->at(1) << ") weight = " << weights_[ph.second] << std::endl;
@@ -298,6 +294,7 @@ printWeights(void)
for(size_t idx = 0; idx < subtree_weights_.size(); ++idx) {
std::cout << idx << " : " << subtree_weights_[idx] << std::endl;
}
+#endif
}
diff --git a/geom_matching/wasserstein/include/dnn/parallel/tbb.h b/geom_matching/wasserstein/include/dnn/parallel/tbb.h
index 4aa6805..64c59e0 100644
--- a/geom_matching/wasserstein/include/dnn/parallel/tbb.h
+++ b/geom_matching/wasserstein/include/dnn/parallel/tbb.h
@@ -1,7 +1,7 @@
#ifndef PARALLEL_H
#define PARALLEL_H
-#include <iostream>
+//#include <iostream>
#include <vector>
#include <boost/range.hpp>