47 lines
1.3 KiB
Factor
47 lines
1.3 KiB
Factor
! Copyright (C) 2008 Slava Pestov.
|
|
! See http://factorcode.org/license.txt for BSD license.
|
|
USING: fry kernel accessors sequences sequences.deep
|
|
compiler.tree ;
|
|
IN: compiler.tree.combinators
|
|
|
|
: each-node ( nodes quot: ( node -- ) -- )
|
|
dup dup '[
|
|
, [
|
|
dup #branch? [
|
|
children>> [ , each-node ] each
|
|
] [
|
|
dup #recursive? [
|
|
child>> , each-node
|
|
] [ drop ] if
|
|
] if
|
|
] bi
|
|
] each ; inline recursive
|
|
|
|
: map-nodes ( nodes quot: ( node -- node' ) -- nodes )
|
|
dup dup '[
|
|
@
|
|
dup #branch? [
|
|
[ [ , map-nodes ] map ] change-children
|
|
] [
|
|
dup #recursive? [
|
|
[ , map-nodes ] change-child
|
|
] when
|
|
] if
|
|
] map flatten ; inline recursive
|
|
|
|
: contains-node? ( nodes quot: ( node -- ? ) -- ? )
|
|
dup dup '[
|
|
, keep swap [ drop t ] [
|
|
dup #branch? [
|
|
children>> [ , contains-node? ] contains?
|
|
] [
|
|
dup #recursive? [
|
|
child>> , contains-node?
|
|
] [ drop f ] if
|
|
] if
|
|
] if
|
|
] contains? ; inline recursive
|
|
|
|
: select-children ( seq flags -- seq' )
|
|
[ [ drop f ] unless ] 2map ;
|