basis/sequences/next/next.factor

   1 USING: kernel sequences sequences.private math ;
   2 IN: sequences.next
   3
   4 <PRIVATE
   5
   6 : iterate-seq ( seq quot -- i seq quot )
   7     [ [ length ] keep ] dip ; inline
   8
   9 : (map-next) ( i seq quot -- )
  10     ! this uses O(n) more bounds checks than is really necessary
  11     [ [ [ 1+ ] dip ?nth ] 2keep nth-unsafe ] dip call ; inline
  12
  13 PRIVATE>
  14
  15 : each-next ( seq quot: ( next-elt elt -- ) -- )
  16     iterate-seq [ (map-next) ] 2curry each-integer ; inline
  17
  18 : map-next ( seq quot: ( next-elt elt -- newelt ) -- newseq )
  19     over dup length swap new-sequence [
  20         iterate-seq [ (map-next) ] 2curry
  21     ] dip [ collect ] keep ; inline