diff options
author | ROUVREAU Vincent <vincent.rouvreau@inria.fr> | 2020-08-03 08:32:48 +0200 |
---|---|---|
committer | ROUVREAU Vincent <vincent.rouvreau@inria.fr> | 2020-08-03 08:32:48 +0200 |
commit | c3a4fc7bdcc69a5d56761c67a77c7e5b6ff6d1ee (patch) | |
tree | 0a0e149249edac006d2b4c17ffeea99af32581d1 /src/python | |
parent | 39fba06ef758483bc237b9375413974c3bbc16e4 (diff) |
code review: simplify edge parsing
Diffstat (limited to 'src/python')
-rw-r--r-- | src/python/include/Simplex_tree_interface.h | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/src/python/include/Simplex_tree_interface.h b/src/python/include/Simplex_tree_interface.h index f786ad6e..e288a8cf 100644 --- a/src/python/include/Simplex_tree_interface.h +++ b/src/python/include/Simplex_tree_interface.h @@ -164,11 +164,12 @@ class Simplex_tree_interface : public Simplex_tree<SimplexTreeOptions> { using Filtered_edge = std::tuple<Vertex_handle, Vertex_handle, Filtration_value>; std::vector<Filtered_edge> edges; for (Simplex_handle sh : Base::skeleton_simplex_range(1)) { - typename Base::Simplex_vertex_range rg = Base::simplex_vertex_range(sh); - auto rg_begin = rg.begin(); - // We take only edges into account - if (std::distance(rg_begin, rg.end()) == 2) { - edges.emplace_back(*rg_begin, *std::next(rg_begin), Base::filtration(sh)); + if (Base::dimension(sh) == 1) { + typename Base::Simplex_vertex_range rg = Base::simplex_vertex_range(sh); + auto vit = rg.begin(); + Vertex_handle v = *vit; + Vertex_handle w = *++vit; + edges.emplace_back(v, w, Base::filtration(sh)); } } |