591 lines
17 KiB
Factor
591 lines
17 KiB
Factor
! Copyright (C) 2004, 2008 Slava Pestov.
|
|
! See http://factorcode.org/license.txt for BSD license.
|
|
USING: fry accessors alien alien.accessors arrays byte-arrays
|
|
classes sequences.private continuations.private effects generic
|
|
hashtables hashtables.private io io.backend io.files
|
|
io.files.private io.streams.c kernel kernel.private math
|
|
math.private memory namespaces namespaces.private parser
|
|
prettyprint quotations quotations.private sbufs sbufs.private
|
|
sequences sequences.private slots.private strings
|
|
strings.private system threads.private classes.tuple
|
|
classes.tuple.private vectors vectors.private words definitions
|
|
words.private assocs summary compiler.units system.private
|
|
combinators locals.backend words.private quotations.private
|
|
stack-checker.state
|
|
stack-checker.backend
|
|
stack-checker.branches
|
|
stack-checker.errors
|
|
stack-checker.transforms
|
|
stack-checker.visitor
|
|
stack-checker.alien ;
|
|
IN: stack-checker.known-words
|
|
|
|
: infer-primitive ( word -- )
|
|
dup
|
|
[ "input-classes" word-prop ]
|
|
[ "default-output-classes" word-prop ] bi <effect>
|
|
apply-word/effect ;
|
|
|
|
{
|
|
{ drop (( x -- )) }
|
|
{ 2drop (( x y -- )) }
|
|
{ 3drop (( x y z -- )) }
|
|
{ dup (( x -- x x )) }
|
|
{ 2dup (( x y -- x y x y )) }
|
|
{ 3dup (( x y z -- x y z x y z )) }
|
|
{ rot (( x y z -- y z x )) }
|
|
{ -rot (( x y z -- z x y )) }
|
|
{ dupd (( x y -- x x y )) }
|
|
{ swapd (( x y z -- y x z )) }
|
|
{ nip (( x y -- y )) }
|
|
{ 2nip (( x y z -- z )) }
|
|
{ tuck (( x y -- y x y )) }
|
|
{ over (( x y -- x y x )) }
|
|
{ pick (( x y z -- x y z x )) }
|
|
{ swap (( x y -- y x )) }
|
|
} [ "shuffle" set-word-prop ] assoc-each
|
|
|
|
: infer-shuffle ( shuffle -- )
|
|
[ in>> length consume-d ] keep ! inputs shuffle
|
|
[ drop ] [ shuffle dup copy-values dup output-d ] 2bi ! inputs outputs copies
|
|
[ nip ] [ swap zip ] 2bi ! inputs copies mapping
|
|
#shuffle, ;
|
|
|
|
: infer-shuffle-word ( word -- )
|
|
"shuffle" word-prop infer-shuffle ;
|
|
|
|
: infer-declare ( -- )
|
|
pop-literal nip
|
|
[ length ensure-d ] keep zip
|
|
#declare, ;
|
|
|
|
GENERIC: infer-call* ( value known -- )
|
|
|
|
: infer-call ( value -- ) dup known infer-call* ;
|
|
|
|
M: literal infer-call*
|
|
[ 1array #drop, ] [ infer-literal-quot ] bi* ;
|
|
|
|
M: curried infer-call*
|
|
swap push-d
|
|
[ uncurry ] infer-quot-here
|
|
[ quot>> known pop-d [ set-known ] keep ]
|
|
[ obj>> known pop-d [ set-known ] keep ] bi
|
|
push-d infer-call ;
|
|
|
|
M: composed infer-call*
|
|
swap push-d
|
|
[ uncompose ] infer-quot-here
|
|
[ quot2>> known pop-d [ set-known ] keep ]
|
|
[ quot1>> known pop-d [ set-known ] keep ] bi
|
|
push-d push-d
|
|
1 infer->r pop-d infer-call
|
|
terminated? get [ 1 infer-r> pop-d infer-call ] unless ;
|
|
|
|
M: object infer-call*
|
|
\ literal-expected inference-warning ;
|
|
|
|
: infer-curry ( -- )
|
|
2 consume-d
|
|
dup first2 <curried> make-known
|
|
[ push-d ] [ 1array ] bi
|
|
\ curry #call, ;
|
|
|
|
: infer-compose ( -- )
|
|
2 consume-d
|
|
dup first2 <composed> make-known
|
|
[ push-d ] [ 1array ] bi
|
|
\ compose #call, ;
|
|
|
|
: infer-execute ( -- )
|
|
pop-literal nip
|
|
dup word? [
|
|
apply-object
|
|
] [
|
|
drop
|
|
"execute must be given a word" time-bomb
|
|
] if ;
|
|
|
|
: infer-<tuple-boa> ( -- )
|
|
\ <tuple-boa>
|
|
peek-d literal value>> second 1+ { tuple } <effect>
|
|
apply-word/effect ;
|
|
|
|
: infer-(throw) ( -- )
|
|
\ (throw)
|
|
peek-d literal value>> 2 + f <effect> t >>terminated?
|
|
apply-word/effect ;
|
|
|
|
: infer-exit ( -- )
|
|
\ exit
|
|
{ integer } { } t >>terminated? <effect>
|
|
apply-word/effect ;
|
|
|
|
: infer-load-locals ( -- )
|
|
pop-literal nip
|
|
[ dup reverse <effect> infer-shuffle ]
|
|
[ infer->r ]
|
|
bi ;
|
|
|
|
: infer-get-local ( -- )
|
|
pop-literal nip
|
|
[ infer-r> ]
|
|
[ dup 0 prefix <effect> infer-shuffle ]
|
|
[ infer->r ]
|
|
tri ;
|
|
|
|
: infer-drop-locals ( -- )
|
|
pop-literal nip
|
|
[ infer-r> ]
|
|
[ { } <effect> infer-shuffle ] bi ;
|
|
|
|
: infer-special ( word -- )
|
|
{
|
|
{ \ >r [ 1 infer->r ] }
|
|
{ \ r> [ 1 infer-r> ] }
|
|
{ \ declare [ infer-declare ] }
|
|
{ \ call [ pop-d infer-call ] }
|
|
{ \ (call) [ pop-d infer-call ] }
|
|
{ \ curry [ infer-curry ] }
|
|
{ \ compose [ infer-compose ] }
|
|
{ \ execute [ infer-execute ] }
|
|
{ \ (execute) [ infer-execute ] }
|
|
{ \ if [ infer-if ] }
|
|
{ \ dispatch [ infer-dispatch ] }
|
|
{ \ <tuple-boa> [ infer-<tuple-boa> ] }
|
|
{ \ (throw) [ infer-(throw) ] }
|
|
{ \ exit [ infer-exit ] }
|
|
{ \ load-locals [ infer-load-locals ] }
|
|
{ \ get-local [ infer-get-local ] }
|
|
{ \ drop-locals [ infer-drop-locals ] }
|
|
{ \ do-primitive [ \ do-primitive cannot-infer-effect ] }
|
|
{ \ alien-invoke [ infer-alien-invoke ] }
|
|
{ \ alien-indirect [ infer-alien-indirect ] }
|
|
{ \ alien-callback [ infer-alien-callback ] }
|
|
} case ;
|
|
|
|
{
|
|
>r r> declare call (call) curry compose execute (execute) if
|
|
dispatch <tuple-boa> (throw) load-locals get-local drop-locals
|
|
do-primitive alien-invoke alien-indirect alien-callback
|
|
} [ t "special" set-word-prop ] each
|
|
|
|
{ call execute dispatch load-locals get-local drop-locals }
|
|
[ t "no-compile" set-word-prop ] each
|
|
|
|
: non-inline-word ( word -- )
|
|
dup called-dependency depends-on
|
|
{
|
|
{ [ dup "shuffle" word-prop ] [ infer-shuffle-word ] }
|
|
{ [ dup "special" word-prop ] [ infer-special ] }
|
|
{ [ dup "primitive" word-prop ] [ infer-primitive ] }
|
|
{ [ dup "transform-quot" word-prop ] [ apply-transform ] }
|
|
{ [ dup "macro" word-prop ] [ apply-macro ] }
|
|
{ [ dup "cannot-infer" word-prop ] [ cannot-infer-effect ] }
|
|
{ [ dup "inferred-effect" word-prop ] [ cached-infer ] }
|
|
{ [ dup recursive-label ] [ call-recursive-word ] }
|
|
[ dup infer-word apply-word/effect ]
|
|
} cond ;
|
|
|
|
: define-primitive ( word inputs outputs -- )
|
|
[ 2drop t "primitive" set-word-prop ]
|
|
[ drop "input-classes" set-word-prop ]
|
|
[ nip "default-output-classes" set-word-prop ]
|
|
3tri ;
|
|
|
|
! Stack effects for all primitives
|
|
\ fixnum< { fixnum fixnum } { object } define-primitive
|
|
\ fixnum< make-foldable
|
|
|
|
\ fixnum<= { fixnum fixnum } { object } define-primitive
|
|
\ fixnum<= make-foldable
|
|
|
|
\ fixnum> { fixnum fixnum } { object } define-primitive
|
|
\ fixnum> make-foldable
|
|
|
|
\ fixnum>= { fixnum fixnum } { object } define-primitive
|
|
\ fixnum>= make-foldable
|
|
|
|
\ eq? { object object } { object } define-primitive
|
|
\ eq? make-foldable
|
|
|
|
\ bignum>fixnum { bignum } { fixnum } define-primitive
|
|
\ bignum>fixnum make-foldable
|
|
|
|
\ float>fixnum { float } { fixnum } define-primitive
|
|
\ bignum>fixnum make-foldable
|
|
|
|
\ fixnum>bignum { fixnum } { bignum } define-primitive
|
|
\ fixnum>bignum make-foldable
|
|
|
|
\ float>bignum { float } { bignum } define-primitive
|
|
\ float>bignum make-foldable
|
|
|
|
\ fixnum>float { fixnum } { float } define-primitive
|
|
\ fixnum>float make-foldable
|
|
|
|
\ bignum>float { bignum } { float } define-primitive
|
|
\ bignum>float make-foldable
|
|
|
|
\ <ratio> { integer integer } { ratio } define-primitive
|
|
\ <ratio> make-foldable
|
|
|
|
\ string>float { string } { float } define-primitive
|
|
\ string>float make-foldable
|
|
|
|
\ float>string { float } { string } define-primitive
|
|
\ float>string make-foldable
|
|
|
|
\ float>bits { real } { integer } define-primitive
|
|
\ float>bits make-foldable
|
|
|
|
\ double>bits { real } { integer } define-primitive
|
|
\ double>bits make-foldable
|
|
|
|
\ bits>float { integer } { float } define-primitive
|
|
\ bits>float make-foldable
|
|
|
|
\ bits>double { integer } { float } define-primitive
|
|
\ bits>double make-foldable
|
|
|
|
\ <complex> { real real } { complex } define-primitive
|
|
\ <complex> make-foldable
|
|
|
|
\ fixnum+ { fixnum fixnum } { integer } define-primitive
|
|
\ fixnum+ make-foldable
|
|
|
|
\ fixnum+fast { fixnum fixnum } { fixnum } define-primitive
|
|
\ fixnum+fast make-foldable
|
|
|
|
\ fixnum- { fixnum fixnum } { integer } define-primitive
|
|
\ fixnum- make-foldable
|
|
|
|
\ fixnum-fast { fixnum fixnum } { fixnum } define-primitive
|
|
\ fixnum-fast make-foldable
|
|
|
|
\ fixnum* { fixnum fixnum } { integer } define-primitive
|
|
\ fixnum* make-foldable
|
|
|
|
\ fixnum*fast { fixnum fixnum } { fixnum } define-primitive
|
|
\ fixnum*fast make-foldable
|
|
|
|
\ fixnum/i { fixnum fixnum } { integer } define-primitive
|
|
\ fixnum/i make-foldable
|
|
|
|
\ fixnum-mod { fixnum fixnum } { fixnum } define-primitive
|
|
\ fixnum-mod make-foldable
|
|
|
|
\ fixnum/mod { fixnum fixnum } { integer fixnum } define-primitive
|
|
\ fixnum/mod make-foldable
|
|
|
|
\ fixnum-bitand { fixnum fixnum } { fixnum } define-primitive
|
|
\ fixnum-bitand make-foldable
|
|
|
|
\ fixnum-bitor { fixnum fixnum } { fixnum } define-primitive
|
|
\ fixnum-bitor make-foldable
|
|
|
|
\ fixnum-bitxor { fixnum fixnum } { fixnum } define-primitive
|
|
\ fixnum-bitxor make-foldable
|
|
|
|
\ fixnum-bitnot { fixnum } { fixnum } define-primitive
|
|
\ fixnum-bitnot make-foldable
|
|
|
|
\ fixnum-shift { fixnum fixnum } { integer } define-primitive
|
|
\ fixnum-shift make-foldable
|
|
|
|
\ fixnum-shift-fast { fixnum fixnum } { fixnum } define-primitive
|
|
\ fixnum-shift-fast make-foldable
|
|
|
|
\ bignum= { bignum bignum } { object } define-primitive
|
|
\ bignum= make-foldable
|
|
|
|
\ bignum+ { bignum bignum } { bignum } define-primitive
|
|
\ bignum+ make-foldable
|
|
|
|
\ bignum- { bignum bignum } { bignum } define-primitive
|
|
\ bignum- make-foldable
|
|
|
|
\ bignum* { bignum bignum } { bignum } define-primitive
|
|
\ bignum* make-foldable
|
|
|
|
\ bignum/i { bignum bignum } { bignum } define-primitive
|
|
\ bignum/i make-foldable
|
|
|
|
\ bignum-mod { bignum bignum } { bignum } define-primitive
|
|
\ bignum-mod make-foldable
|
|
|
|
\ bignum/mod { bignum bignum } { bignum bignum } define-primitive
|
|
\ bignum/mod make-foldable
|
|
|
|
\ bignum-bitand { bignum bignum } { bignum } define-primitive
|
|
\ bignum-bitand make-foldable
|
|
|
|
\ bignum-bitor { bignum bignum } { bignum } define-primitive
|
|
\ bignum-bitor make-foldable
|
|
|
|
\ bignum-bitxor { bignum bignum } { bignum } define-primitive
|
|
\ bignum-bitxor make-foldable
|
|
|
|
\ bignum-bitnot { bignum } { bignum } define-primitive
|
|
\ bignum-bitnot make-foldable
|
|
|
|
\ bignum-shift { bignum fixnum } { bignum } define-primitive
|
|
\ bignum-shift make-foldable
|
|
|
|
\ bignum< { bignum bignum } { object } define-primitive
|
|
\ bignum< make-foldable
|
|
|
|
\ bignum<= { bignum bignum } { object } define-primitive
|
|
\ bignum<= make-foldable
|
|
|
|
\ bignum> { bignum bignum } { object } define-primitive
|
|
\ bignum> make-foldable
|
|
|
|
\ bignum>= { bignum bignum } { object } define-primitive
|
|
\ bignum>= make-foldable
|
|
|
|
\ bignum-bit? { bignum integer } { object } define-primitive
|
|
\ bignum-bit? make-foldable
|
|
|
|
\ bignum-log2 { bignum } { bignum } define-primitive
|
|
\ bignum-log2 make-foldable
|
|
|
|
\ byte-array>bignum { byte-array } { bignum } define-primitive
|
|
\ byte-array>bignum make-foldable
|
|
|
|
\ float= { float float } { object } define-primitive
|
|
\ float= make-foldable
|
|
|
|
\ float+ { float float } { float } define-primitive
|
|
\ float+ make-foldable
|
|
|
|
\ float- { float float } { float } define-primitive
|
|
\ float- make-foldable
|
|
|
|
\ float* { float float } { float } define-primitive
|
|
\ float* make-foldable
|
|
|
|
\ float/f { float float } { float } define-primitive
|
|
\ float/f make-foldable
|
|
|
|
\ float< { float float } { object } define-primitive
|
|
\ float< make-foldable
|
|
|
|
\ float-mod { float float } { float } define-primitive
|
|
\ float-mod make-foldable
|
|
|
|
\ float<= { float float } { object } define-primitive
|
|
\ float<= make-foldable
|
|
|
|
\ float> { float float } { object } define-primitive
|
|
\ float> make-foldable
|
|
|
|
\ float>= { float float } { object } define-primitive
|
|
\ float>= make-foldable
|
|
|
|
\ <word> { object object } { word } define-primitive
|
|
\ <word> make-flushable
|
|
|
|
\ word-xt { word } { integer integer } define-primitive
|
|
\ word-xt make-flushable
|
|
|
|
\ getenv { fixnum } { object } define-primitive
|
|
\ getenv make-flushable
|
|
|
|
\ setenv { object fixnum } { } define-primitive
|
|
|
|
\ (exists?) { string } { object } define-primitive
|
|
|
|
\ gc { } { } define-primitive
|
|
|
|
\ gc-stats { } { array } define-primitive
|
|
|
|
\ save-image { string } { } define-primitive
|
|
|
|
\ save-image-and-exit { string } { } define-primitive
|
|
|
|
\ data-room { } { integer integer array } define-primitive
|
|
\ data-room make-flushable
|
|
|
|
\ code-room { } { integer integer integer integer } define-primitive
|
|
\ code-room make-flushable
|
|
|
|
\ millis { } { integer } define-primitive
|
|
\ millis make-flushable
|
|
|
|
\ tag { object } { fixnum } define-primitive
|
|
\ tag make-foldable
|
|
|
|
\ dlopen { string } { dll } define-primitive
|
|
|
|
\ dlsym { string object } { c-ptr } define-primitive
|
|
|
|
\ dlclose { dll } { } define-primitive
|
|
|
|
\ <byte-array> { integer } { byte-array } define-primitive
|
|
\ <byte-array> make-flushable
|
|
|
|
\ <displaced-alien> { integer c-ptr } { c-ptr } define-primitive
|
|
\ <displaced-alien> make-flushable
|
|
|
|
\ alien-signed-cell { c-ptr integer } { integer } define-primitive
|
|
\ alien-signed-cell make-flushable
|
|
|
|
\ set-alien-signed-cell { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-unsigned-cell { c-ptr integer } { integer } define-primitive
|
|
\ alien-unsigned-cell make-flushable
|
|
|
|
\ set-alien-unsigned-cell { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-signed-8 { c-ptr integer } { integer } define-primitive
|
|
\ alien-signed-8 make-flushable
|
|
|
|
\ set-alien-signed-8 { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-unsigned-8 { c-ptr integer } { integer } define-primitive
|
|
\ alien-unsigned-8 make-flushable
|
|
|
|
\ set-alien-unsigned-8 { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-signed-4 { c-ptr integer } { integer } define-primitive
|
|
\ alien-signed-4 make-flushable
|
|
|
|
\ set-alien-signed-4 { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-unsigned-4 { c-ptr integer } { integer } define-primitive
|
|
\ alien-unsigned-4 make-flushable
|
|
|
|
\ set-alien-unsigned-4 { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-signed-2 { c-ptr integer } { fixnum } define-primitive
|
|
\ alien-signed-2 make-flushable
|
|
|
|
\ set-alien-signed-2 { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-unsigned-2 { c-ptr integer } { fixnum } define-primitive
|
|
\ alien-unsigned-2 make-flushable
|
|
|
|
\ set-alien-unsigned-2 { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-signed-1 { c-ptr integer } { fixnum } define-primitive
|
|
\ alien-signed-1 make-flushable
|
|
|
|
\ set-alien-signed-1 { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-unsigned-1 { c-ptr integer } { fixnum } define-primitive
|
|
\ alien-unsigned-1 make-flushable
|
|
|
|
\ set-alien-unsigned-1 { integer c-ptr integer } { } define-primitive
|
|
|
|
\ alien-float { c-ptr integer } { float } define-primitive
|
|
\ alien-float make-flushable
|
|
|
|
\ set-alien-float { float c-ptr integer } { } define-primitive
|
|
|
|
\ alien-double { c-ptr integer } { float } define-primitive
|
|
\ alien-double make-flushable
|
|
|
|
\ set-alien-double { float c-ptr integer } { } define-primitive
|
|
|
|
\ alien-cell { c-ptr integer } { simple-c-ptr } define-primitive
|
|
\ alien-cell make-flushable
|
|
|
|
\ set-alien-cell { c-ptr c-ptr integer } { } define-primitive
|
|
|
|
\ alien-address { alien } { integer } define-primitive
|
|
\ alien-address make-flushable
|
|
|
|
\ slot { object fixnum } { object } define-primitive
|
|
\ slot make-flushable
|
|
|
|
\ set-slot { object object fixnum } { } define-primitive
|
|
|
|
\ string-nth { fixnum string } { fixnum } define-primitive
|
|
\ string-nth make-flushable
|
|
|
|
\ set-string-nth { fixnum fixnum string } { } define-primitive
|
|
|
|
\ resize-array { integer array } { array } define-primitive
|
|
\ resize-array make-flushable
|
|
|
|
\ resize-byte-array { integer byte-array } { byte-array } define-primitive
|
|
\ resize-byte-array make-flushable
|
|
|
|
\ resize-string { integer string } { string } define-primitive
|
|
\ resize-string make-flushable
|
|
|
|
\ <array> { integer object } { array } define-primitive
|
|
\ <array> make-flushable
|
|
|
|
\ begin-scan { } { } define-primitive
|
|
|
|
\ next-object { } { object } define-primitive
|
|
|
|
\ end-scan { } { } define-primitive
|
|
|
|
\ size { object } { fixnum } define-primitive
|
|
\ size make-flushable
|
|
|
|
\ die { } { } define-primitive
|
|
|
|
\ fopen { string string } { alien } define-primitive
|
|
|
|
\ fgetc { alien } { object } define-primitive
|
|
|
|
\ fwrite { string alien } { } define-primitive
|
|
|
|
\ fputc { object alien } { } define-primitive
|
|
|
|
\ fread { integer string } { object } define-primitive
|
|
|
|
\ fflush { alien } { } define-primitive
|
|
|
|
\ fclose { alien } { } define-primitive
|
|
|
|
\ <wrapper> { object } { wrapper } define-primitive
|
|
\ <wrapper> make-foldable
|
|
|
|
\ (clone) { object } { object } define-primitive
|
|
\ (clone) make-flushable
|
|
|
|
\ <string> { integer integer } { string } define-primitive
|
|
\ <string> make-flushable
|
|
|
|
\ array>quotation { array } { quotation } define-primitive
|
|
\ array>quotation make-flushable
|
|
|
|
\ quotation-xt { quotation } { integer } define-primitive
|
|
\ quotation-xt make-flushable
|
|
|
|
\ <tuple> { tuple-layout } { tuple } define-primitive
|
|
\ <tuple> make-flushable
|
|
|
|
\ datastack { } { array } define-primitive
|
|
\ datastack make-flushable
|
|
|
|
\ retainstack { } { array } define-primitive
|
|
\ retainstack make-flushable
|
|
|
|
\ callstack { } { callstack } define-primitive
|
|
\ callstack make-flushable
|
|
|
|
\ callstack>array { callstack } { array } define-primitive
|
|
\ callstack>array make-flushable
|
|
|
|
\ (sleep) { integer } { } define-primitive
|
|
|
|
\ become { array array } { } define-primitive
|
|
|
|
\ innermost-frame-quot { callstack } { quotation } define-primitive
|
|
|
|
\ innermost-frame-scan { callstack } { fixnum } define-primitive
|
|
|
|
\ set-innermost-frame-quot { quotation callstack } { } define-primitive
|
|
|
|
\ dll-valid? { object } { object } define-primitive
|
|
|
|
\ modify-code-heap { array object } { } define-primitive
|
|
|
|
\ unimplemented { } { } define-primitive
|