steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

heaps: fix slurp-heap to allow modifying the heap while slurping.

db4
John Benediktsson 2014-12-15 08:53:48 -08:00
parent fb9cf1219f
commit fe377ba02a
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
basis/heaps/heaps.factor
View File

@ -147,13 +147,14 @@ M: heap heap-pop*
M: heap heap-pop
[ data-first >entry< ] [ heap-pop* ] bi ;
: heap-pop-all ( heap -- alist )
check-heap [ heap-size ] keep
'[ _ heap-pop swap 2array ] replicate ;
: slurp-heap ( ... heap quot: ( ... value key -- ... ) -- ... )
[ check-heap ] dip
[ drop '[ _ heap-empty? ] ]
[ '[ _ heap-pop @ ] until ] 2bi ; inline
: slurp-heap ( heap quot: ( value key -- ) -- )
[ check-heap [ heap-size ] keep ] dip
'[ _ heap-pop @ ] times ; inline
: heap-pop-all ( heap -- alist )
[ heap-size <vector> ] keep
[ swap 2array suffix! ] slurp-heap { } like ;
ERROR: bad-heap-delete ;