wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-24 13:20:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix heap.pyx to work when infinite values are pushed.

Browse Source
This commit is contained in:
Zachary Pincus
2009-12-30 14:31:08 -05:00
parent ad141b680d
commit 07fbf605b2
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scikits/image/graph/heap.pyx
+3 -3
View File
@@ -350,13 +350,13 @@ cdef class BinaryHeap:
cdef int levels = self.levels
# search tree (using absolute indices)
for level in range(1, levels):
if values[i] < values[i+1]:
if values[i] <= values[i+1]:
i = i*2+1 # CalcNextAbs
else:
i = (i+1)*2+1 # CalcNextAbs
# select best one in last level
if values[i] < values[i+1]:
if values[i] <= values[i+1]:
i = i
else:
i = i+1
@@ -367,7 +367,7 @@ cdef class BinaryHeap:
self._popped_ref = self._references[ir]
# remove it
if value != inf:
if self.count:
self._remove(i)
# return