summaryrefslogtreecommitdiff
path: root/src/Collapse/test
diff options
context:
space:
mode:
authorROUVREAU Vincent <vincent.rouvreau@inria.fr>2020-06-25 17:10:36 +0200
committerROUVREAU Vincent <vincent.rouvreau@inria.fr>2020-06-25 17:10:36 +0200
commitb522b330b10d11f0da640b8bba7ee689dea774d7 (patch)
treeb94adbfcb414b86433f2023a633374bfd2142d8d /src/Collapse/test
parent2610ce8092a3935e228065884bcbd70d910b40cd (diff)
Remove interface with boost graphs and use boost transform for data from graphs
Diffstat (limited to 'src/Collapse/test')
-rw-r--r--src/Collapse/test/collapse_unit_test.cpp13
1 files changed, 11 insertions, 2 deletions
diff --git a/src/Collapse/test/collapse_unit_test.cpp b/src/Collapse/test/collapse_unit_test.cpp
index 108f77e4..b5ad09c5 100644
--- a/src/Collapse/test/collapse_unit_test.cpp
+++ b/src/Collapse/test/collapse_unit_test.cpp
@@ -13,6 +13,7 @@
#define BOOST_TEST_MODULE "collapse"
#include <boost/test/unit_test.hpp>
#include <boost/mpl/list.hpp>
+#include <boost/range/adaptor/transformed.hpp>
#include <gudhi/Flag_complex_edge_collapser.h>
#include <gudhi/distance_functions.h>
@@ -49,7 +50,7 @@ void trace_and_check_collapse(const Filtered_edge_range& filtered_edges, const F
}
std::cout << "COLLAPSE - keep edges: " << std::endl;
- Flag_complex_edge_collapser edge_collapser(filtered_edges.begin(), filtered_edges.end());
+ Flag_complex_edge_collapser edge_collapser(filtered_edges);
Filtered_edge_list remaining_edges;
edge_collapser.process_edges(
[&remaining_edges](Vertex_handle u, Vertex_handle v, Filtration_value filtration) {
@@ -174,7 +175,15 @@ BOOST_AUTO_TEST_CASE(collapse_from_proximity_graph) {
Proximity_graph proximity_graph = Gudhi::compute_proximity_graph<Flag_complex_edge_collapser>(point_cloud,
threshold,
Gudhi::Euclidean_distance());
- Flag_complex_edge_collapser edge_collapser(proximity_graph);
+
+ Flag_complex_edge_collapser edge_collapser(
+ boost::adaptors::transform(edges(proximity_graph), [&](auto&&edge){
+ return std::make_tuple(source(edge, proximity_graph),
+ target(edge, proximity_graph),
+ get(Gudhi::edge_filtration_t(), proximity_graph, edge));
+ })
+ );
+
Filtered_edge_list remaining_edges;
edge_collapser.process_edges(
[&remaining_edges](Vertex_handle u, Vertex_handle v, Filtration_value filtration) {