158 lines
4.3 KiB
Factor
158 lines
4.3 KiB
Factor
! Copyright (C) 2008 Slava Pestov.
|
|
! See http://factorcode.org/license.txt for BSD license.
|
|
USING: fry namespaces assocs kernel sequences words accessors
|
|
definitions math math.order effects classes arrays combinators
|
|
vectors hints
|
|
stack-checker.state
|
|
stack-checker.errors
|
|
stack-checker.values
|
|
stack-checker.visitor
|
|
stack-checker.backend
|
|
stack-checker.branches
|
|
stack-checker.known-words
|
|
stack-checker.recursive-state ;
|
|
IN: stack-checker.inlining
|
|
|
|
! Code to handle inline words. Much of the complexity stems from
|
|
! having to handle recursive inline words.
|
|
|
|
: infer-inline-word-def ( word label -- )
|
|
[ drop specialized-def ] [ add-inline-word ] 2bi infer-quot ;
|
|
|
|
TUPLE: inline-recursive < identity-tuple
|
|
id
|
|
word
|
|
enter-out enter-recursive
|
|
return calls
|
|
fixed-point
|
|
introductions
|
|
loop? ;
|
|
|
|
M: inline-recursive hashcode* id>> hashcode* ;
|
|
|
|
: inlined-block? ( word -- ? ) "inlined-block" word-prop ;
|
|
|
|
: <inline-recursive> ( word -- label )
|
|
inline-recursive new
|
|
gensym dup t "inlined-block" set-word-prop >>id
|
|
swap >>word ;
|
|
|
|
: quotation-param? ( obj -- ? )
|
|
dup pair? [ second effect? ] [ drop f ] if ;
|
|
|
|
: make-copies ( values effect-in -- values' )
|
|
[ length cut* ] keep
|
|
[ quotation-param? [ copy-value ] [ drop <value> ] if ] 2map
|
|
[ make-values ] dip append ;
|
|
|
|
SYMBOL: enter-in
|
|
SYMBOL: enter-out
|
|
|
|
: prepare-stack ( word -- )
|
|
required-stack-effect in>>
|
|
[ length ensure-d drop ] [
|
|
meta-d clone enter-in set
|
|
meta-d swap make-copies enter-out set
|
|
] bi ;
|
|
|
|
: emit-enter-recursive ( label -- )
|
|
enter-out get >>enter-out
|
|
enter-in get enter-out get #enter-recursive,
|
|
enter-out get >vector \ meta-d set ;
|
|
|
|
: entry-stack-height ( label -- stack )
|
|
enter-out>> length ;
|
|
|
|
: check-return ( word label -- )
|
|
2dup
|
|
[ stack-effect effect-height ]
|
|
[ entry-stack-height current-stack-height swap - ]
|
|
bi*
|
|
= [ 2drop ] [
|
|
terminated? get [ 2drop ] [
|
|
word>> current-stack-height
|
|
unbalanced-recursion-error inference-error
|
|
] if
|
|
] if ;
|
|
|
|
: end-recursive-word ( word label -- )
|
|
[ check-return ]
|
|
[ meta-d dup copy-values dup \ meta-d set #return-recursive, ]
|
|
bi ;
|
|
|
|
: recursive-word-inputs ( label -- n )
|
|
entry-stack-height d-in get + ;
|
|
|
|
: (inline-recursive-word) ( word -- label in out visitor terminated? )
|
|
dup prepare-stack
|
|
[
|
|
init-inference
|
|
nest-visitor
|
|
|
|
dup <inline-recursive>
|
|
[ dup emit-enter-recursive infer-inline-word-def ]
|
|
[ end-recursive-word ]
|
|
[ nip ]
|
|
2tri
|
|
|
|
dup recursive-word-inputs
|
|
meta-d
|
|
stack-visitor get
|
|
terminated? get
|
|
] with-scope ;
|
|
|
|
: inline-recursive-word ( word -- )
|
|
(inline-recursive-word)
|
|
[ [ consume-d ] [ output-d ] [ ] tri* #recursive, ] dip
|
|
[ terminate ] when ;
|
|
|
|
: check-call-height ( label -- )
|
|
dup entry-stack-height current-stack-height >
|
|
[ word>> diverging-recursion-error inference-error ] [ drop ] if ;
|
|
|
|
: trim-stack ( label seq -- stack )
|
|
swap word>> required-stack-effect in>> length tail* ;
|
|
|
|
: call-site-stack ( label -- stack )
|
|
meta-d trim-stack ;
|
|
|
|
: trimmed-enter-out ( label -- stack )
|
|
dup enter-out>> trim-stack ;
|
|
|
|
: check-call-site-stack ( label -- )
|
|
[ ] [ call-site-stack ] [ trimmed-enter-out ] tri
|
|
[ dup known [ [ known ] bi@ = ] [ 2drop t ] if ] 2all?
|
|
[ drop ] [ word>> inconsistent-recursive-call-error inference-error ] if ;
|
|
|
|
: check-call ( label -- )
|
|
[ check-call-height ] [ check-call-site-stack ] bi ;
|
|
|
|
: adjust-stack-effect ( effect -- effect' )
|
|
[ in>> ] [ out>> ] bi
|
|
meta-d length pick length [-]
|
|
object <repetition> '[ _ prepend ] bi@
|
|
<effect> ;
|
|
|
|
: call-recursive-inline-word ( word label -- )
|
|
over "recursive" word-prop [
|
|
[ required-stack-effect adjust-stack-effect ] dip
|
|
[ check-call ] [ '[ _ #call-recursive, ] consume/produce ] bi
|
|
] [ drop undeclared-recursion-error inference-error ] if ;
|
|
|
|
: inline-word ( word -- )
|
|
commit-literals
|
|
[ inlined-dependency depends-on ]
|
|
[
|
|
dup inline-recursive-label [
|
|
call-recursive-inline-word
|
|
] [
|
|
dup "recursive" word-prop
|
|
[ inline-recursive-word ]
|
|
[ dup infer-inline-word-def ]
|
|
if
|
|
] if*
|
|
] bi ;
|
|
|
|
M: word apply-object
|
|
dup inline? [ inline-word ] [ non-inline-word ] if ;
|