1 files changed, 76 insertions, 0 deletions
diff --git a/src/GudhUI/utils/Vertex_collapsor.h b/src/GudhUI/utils/Vertex_collapsor.h
new file mode 100644
index 00000000..030e4bb0
--- /dev/null
+++ b/src/GudhUI/utils/Vertex_collapsor.h
@@ -0,0 +1,76 @@
+/*    This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT.
+ *    See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details.
+ *    Author(s):       David Salinas
+ *
+ *    Copyright (C) 2014 Inria
+ *
+ *    Modification(s):
+ *      - YYYY/MM Author: Description of the modification
+ */
+
+#ifndef UTILS_VERTEX_COLLAPSOR_H_
+#define UTILS_VERTEX_COLLAPSOR_H_
+
+#include <list>
+
+#include "utils/Edge_contractor.h"
+#include "utils/Furthest_point_epsilon_net.h"
+#include "utils/UI_utils.h"
+
+/**
+ * Iteratively puts every vertex at the center of its neighbors
+ */
+template<typename SkBlComplex> class Vertex_collapsor {
+ private:
+  SkBlComplex& complex_;
+  size_t num_collapses_;
+
+ public:
+  typedef typename SkBlComplex::Vertex_handle Vertex_handle;
+  typedef typename SkBlComplex::Edge_handle Edge_handle;
+
+  /**
+   * @brief Modify complex to be the expansion of the k-nearest neighbor
+   * symetric graph.
+   */
+  Vertex_collapsor(SkBlComplex& complex, size_t num_collapses) :
+      complex_(complex), num_collapses_(num_collapses) {
+    // std::list<Vertex_handle> vertices;
+    // vertices.insert(vertices.begin(),complex_.vertex_range().begin(),complex_.vertex_range().end());
+    // UIDBG("Collapse vertices");
+    // collapse_vertices(vertices);
+
+    std::list<Vertex_handle> vertices;
+
+    UIDBG("Compute eps net");
+    Furthest_point_epsilon_net<Complex> eps_net(complex_);
+
+    for (auto vh : eps_net.net_filtration_)
+      vertices.push_back(vh.vertex_handle);
+
+    UIDBG("Collapse vertices");
+    collapse_vertices(vertices);
+  }
+
+ private:
+  void collapse_vertices(std::list<Vertex_handle>& vertices) {
+    while (!vertices.empty() && num_collapses_--) {
+      Vertex_handle current_vertex = vertices.front();
+      vertices.pop_front();
+      if (is_link_reducible(current_vertex))
+        complex_.remove_vertex(current_vertex);
+    }
+  }
+
+  bool is_link_reducible(Vertex_handle v) {
+    auto link = complex_.link(v);
+    if (link.empty()) return false;
+    if (link.is_cone()) return true;
+    if (link.num_connected_components() > 1) return false;
+    Edge_contractor<Complex> contractor(link, link.num_vertices() - 1);
+    (void)contractor;
+    return (link.num_vertices() == 1);
+  }
+};
+
+#endif  // UTILS_VERTEX_COLLAPSOR_H_