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

cursors: generalized -ncontainer-

release
Joe Groff 2010-03-24 16:41:52 -07:00
parent 56c89c0510
commit c17eb80b90
1 changed files with 29 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
extra/cursors/cursors.factor
View File

@ -1,6 +1,7 @@
! (c)2010 Joe Groff bsd license ! (c)2010 Joe Groff bsd license
USING: accessors arrays assocs combinators.short-circuit fry USING: accessors arrays assocs combinators.short-circuit fry
hashtables kernel locals math math.functions math.order sequences ; hashtables kernel locals macros math math.functions math.order
generalizations sequences ;
FROM: sequences.private => nth-unsafe set-nth-unsafe ; FROM: sequences.private => nth-unsafe set-nth-unsafe ;
FROM: hashtables.private => tombstone? ; FROM: hashtables.private => tombstone? ;
IN: cursors IN: cursors
@ -376,7 +377,7 @@ GENERIC: cursor-key-value ( cursor -- key value )
INSTANCE: input-cursor assoc-cursor INSTANCE: input-cursor assoc-cursor
M: input-cursor cursor-key-value M: input-cursor cursor-key-value
cursor-value first2 ; inline cursor-value-unsafe first2 ; inline
! !
! hashtable cursor ! hashtable cursor
@ -422,7 +423,7 @@ M: hashtable-cursor cursor-key-value
INSTANCE: hashtable-cursor input-cursor INSTANCE: hashtable-cursor input-cursor
M: hashtable-cursor cursor-value M: hashtable-cursor cursor-value-unsafe
cursor-key-value 2array ; inline cursor-key-value 2array ; inline
INSTANCE: hashtable container INSTANCE: hashtable container
@ -468,7 +469,7 @@ M: zip-cursor inc-cursor ( cursor -- cursor' )
INSTANCE: zip-cursor assoc-cursor INSTANCE: zip-cursor assoc-cursor
M: zip-cursor cursor-key-value M: zip-cursor cursor-key-value
[ keys>> cursor-value ] [ values>> cursor-value ] bi ; inline [ keys>> cursor-value-unsafe ] [ values>> cursor-value-unsafe ] bi ; inline
: zip-cursors ( a-begin a-end b-begin b-end -- begin end ) : zip-cursors ( a-begin a-end b-begin b-end -- begin end )
[ <zip-cursor> ] bi-curry@ bi* ; inline [ <zip-cursor> ] bi-curry@ bi* ; inline
@ -492,3 +493,27 @@ ALIAS: -2container- -assoc-
: 2map ( ... a b quot: ( ... x y -- ... z ) -- ... c ) : 2map ( ... a b quot: ( ... x y -- ... z ) -- ... c )
pick 2map-as ; inline pick 2map-as ; inline
!
! generalized zips
!
: -unzip- ( quot -- quot' )
'[ [ keys>> cursor-value-unsafe ] [ values>> ] bi @ ] ; inline
MACRO: nzip-cursors ( n -- ) 1 - [ zip-cursors ] n*quot ;
: nall ( seqs... n -- begin end ) [ [ all ] swap napply ] [ nzip-cursors ] bi ; inline
: nall- ( seqs... quot n -- begin end quot ) swap [ nall ] dip ; inline
MACRO: -ncontainer- ( n -- )
1 - [ -unzip- ] n*quot [ -container- ] prepend ;
: ncontainer- ( seqs... quot n -- begin end quot ) [ nall- ] [ -ncontainer- ] bi ; inline
: neach ( seqs... quot n -- ) ncontainer- -each ; inline
: nmap-as ( seqs... quot exemplar n -- newseq )
swap [ ncontainer- ] dip -map-as ; inline
: nmap ( seqs... quot n -- newseq )
dup [ npick ] curry [ dip swap ] curry dip nmap-as ; inline