mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-15 11:25:53 +08:00
Zachary Pincus
4741c113ca
cython classes for fast binary heaps (that store (value, reference) pairs) are now available from scikits.image.graph.heap
35 lines
1.0 KiB
Cython
35 lines
1.0 KiB
Cython
""" This is the definition file for heap.pyx.
|
|
It contains the definitions of the heap classes, such that
|
|
other cython modules can "cimport heap" and thus use the
|
|
C versions of pop(), push(), and value_of(): pop_fast(), push_fast() and
|
|
value_of_fast()
|
|
"""
|
|
|
|
# determine datatypes for heap
|
|
ctypedef double VALUE_T
|
|
ctypedef int REFERENCE_T
|
|
|
|
|
|
cdef class BinaryHeap:
|
|
cdef readonly int count, levels, min_levels
|
|
cdef VALUE_T *_values
|
|
cdef REFERENCE_T *_references
|
|
cdef int _popped_ref
|
|
|
|
cdef void _add_or_remove_level(self, int add_or_remove)
|
|
cdef void _update(self)
|
|
cdef void _update_one(self, int i)
|
|
cdef void _remove(self, int i)
|
|
|
|
cdef int push_fast(self, double value, int reference)
|
|
cdef float pop_fast(self)
|
|
|
|
cdef class FastUpdateBinaryHeap(BinaryHeap):
|
|
cdef readonly int max_reference
|
|
cdef REFERENCE_T *_crossref
|
|
cdef int _invalid_ref
|
|
cdef int _pushed
|
|
|
|
cdef float value_of_fast(self, int reference)
|
|
cdef int push_if_lower_fast(self, double value, int reference)
|
|
|