diff options
| author | ROUVREAU Vincent <vincent.rouvreau@inria.fr> | 2021-03-23 16:10:24 +0100 |
|---|---|---|
| committer | ROUVREAU Vincent <vincent.rouvreau@inria.fr> | 2021-03-23 16:10:24 +0100 |
| commit | 0313c98f32363bfc75162613b3cfa9b7efa4081b (patch) | |
| tree | ba7cd142cdd99ba8681809e274161b09e6474e6c /src/python/gudhi | |
| parent | ecb217234aaedc8ed04c26b09764da151c9fc77a (diff) | |
Add simplex tree equality operator to be able to test alpha complex
Diffstat (limited to 'src/python/gudhi')
| -rw-r--r-- | src/python/gudhi/simplex_tree.pxd | 1 | ||||
| -rw-r--r-- | src/python/gudhi/simplex_tree.pyx | 9 |
2 files changed, 10 insertions, 0 deletions
diff --git a/src/python/gudhi/simplex_tree.pxd b/src/python/gudhi/simplex_tree.pxd index 000323af..2aa435b1 100644 --- a/src/python/gudhi/simplex_tree.pxd +++ b/src/python/gudhi/simplex_tree.pxd @@ -74,6 +74,7 @@ cdef extern from "Simplex_tree_interface.h" namespace "Gudhi": Simplex_tree_skeleton_iterator get_skeleton_iterator_begin(int dimension) nogil Simplex_tree_skeleton_iterator get_skeleton_iterator_end(int dimension) nogil pair[Simplex_tree_boundary_iterator, Simplex_tree_boundary_iterator] get_boundary_iterators(vector[int] simplex) nogil except + + bint operator==(Simplex_tree_interface_full_featured) nogil cdef extern from "Persistent_cohomology_interface.h" namespace "Gudhi": cdef cppclass Simplex_tree_persistence_interface "Gudhi::Persistent_cohomology_interface<Gudhi::Simplex_tree<Gudhi::Simplex_tree_options_full_featured>>": diff --git a/src/python/gudhi/simplex_tree.pyx b/src/python/gudhi/simplex_tree.pyx index d7991417..b5a938d5 100644 --- a/src/python/gudhi/simplex_tree.pyx +++ b/src/python/gudhi/simplex_tree.pyx @@ -639,3 +639,12 @@ cdef class SimplexTree: self.thisptr = <intptr_t>(ptr.collapse_edges(nb_iter)) # Delete old pointer del ptr + + def __eq__(self, other): + """Simplex tree equality operator using C++ depth first search operator== + + :returns: True if the 2 simplex trees are equal, False otherwise. + :rtype: bool + """ + cdef intptr_t other_int_ptr=other.thisptr + return dereference(self.get_ptr()) == dereference(<Simplex_tree_interface_full_featured*>other_int_ptr)
\ No newline at end of file |