factor/basis/compiler/cfg/utilities/utilities.factor

80 lines
2.4 KiB
Factor

! Copyright (C) 2008, 2010 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
USING: accessors assocs combinators combinators.short-circuit
cpu.architecture kernel layouts locals make math namespaces sequences
sets vectors fry arrays compiler.cfg compiler.cfg.instructions
compiler.cfg.rpo compiler.utilities ;
IN: compiler.cfg.utilities
: back-edge? ( from to -- ? )
[ number>> ] bi@ >= ;
: loop-entry? ( bb -- ? )
dup predecessors>> [ swap back-edge? ] with any? ;
: empty-block? ( bb -- ? )
instructions>> {
[ length 1 = ]
[ first ##branch? ]
} 1&& ;
: (skip-empty-blocks) ( visited bb -- visited' bb' )
dup pick in? [
dup empty-block? [
dup pick adjoin
successors>> first (skip-empty-blocks)
] when
] unless ; inline recursive
: skip-empty-blocks ( bb -- bb' )
[ HS{ } clone ] dip (skip-empty-blocks) nip ;
:: update-predecessors ( from to bb -- )
! Whenever 'from' appears in the list of predecessors of 'to'
! replace it with 'bb'.
to predecessors>> [ dup from eq? [ drop bb ] when ] map! drop ;
:: update-successors ( from to bb -- )
! Whenever 'to' appears in the list of successors of 'from'
! replace it with 'bb'.
from successors>> [ dup to eq? [ drop bb ] when ] map! drop ;
:: insert-basic-block ( from to insns -- )
! Insert basic block on the edge between 'from' and 'to'.
<basic-block> :> bb
insns V{ } like bb instructions<<
V{ from } bb predecessors<<
V{ to } bb successors<<
from to bb update-predecessors
from to bb update-successors ;
: add-instructions ( bb quot -- )
[ instructions>> building ] dip '[
building get pop
[ @ ] dip
,
] with-variable ; inline
: has-phis? ( bb -- ? )
instructions>> first ##phi? ;
: cfg-has-phis? ( cfg -- ? )
post-order [ has-phis? ] any? ;
: if-has-phis ( ..a bb quot: ( ..a bb -- ..b ) -- ..b )
[ dup has-phis? ] dip [ drop ] if ; inline
: each-phi ( ... bb quot: ( ... ##phi -- ... ) -- ... )
[ instructions>> ] dip
'[ dup ##phi? [ @ t ] [ drop f ] if ] all? drop ; inline
: each-non-phi ( ... bb quot: ( ... insn -- ... ) -- ... )
[ instructions>> ] dip
'[ dup ##phi? [ drop ] _ if ] each ; inline
: predecessor ( bb -- pred )
predecessors>> first ; inline
: <copy> ( dst src -- insn )
any-rep ##copy new-insn ;