basis/compiler/utilities/utilities.factor

   1 ! Copyright (C) 2008, 2009 Slava Pestov.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: kernel sequences sequences.private arrays vectors fry
   4 math.order namespaces assocs ;
   5 IN: compiler.utilities
   6
   7 : flattener ( seq quot -- seq vector quot' )
   8     over length <vector> [
   9         dup
  10         '[
  11             @ [
  12                 dup array?
  13                 [ _ push-all ] [ _ push ] if
  14             ] when*
  15         ]
  16     ] keep ; inline
  17
  18 : flattening ( seq quot combinator -- seq' )
  19     [ flattener ] dip dip { } like ; inline
  20
  21 : map-flat ( seq quot -- seq' ) [ each ] flattening ; inline
  22
  23 : 2map-flat ( seq quot -- seq' ) [ 2each ] flattening ; inline
  24
  25 SYMBOL: yield-hook
  26
  27 yield-hook global [ [ ] or ] change-at