1 ! Copyright (C) 2004, 2010 Slava Pestov, Daniel Ehrenberg.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: fry accessors alien alien.accessors arrays byte-arrays
4 classes continuations.private effects generic hashtables
5 hashtables.private io io.backend io.files io.files.private
6 io.streams.c kernel kernel.private math math.private
7 math.parser.private memory memory.private namespaces
8 namespaces.private parser quotations quotations.private sbufs
9 sbufs.private sequences sequences.private slots.private strings
10 strings.private system threads.private classes.tuple
11 classes.tuple.private vectors vectors.private words
12 words.private definitions assocs summary compiler.units
13 system.private combinators combinators.short-circuit locals
14 locals.backend locals.types combinators.private
15 stack-checker.values generic.single generic.single.private
16 alien.libraries tools.dispatch.private tools.profiler.private
22 stack-checker.branches
23 stack-checker.transforms
24 stack-checker.dependencies
25 stack-checker.recursive-state
26 stack-checker.row-polymorphism ;
27 IN: stack-checker.known-words
29 : infer-primitive ( word -- )
31 [ "input-classes" word-prop ]
32 [ "default-output-classes" word-prop ] bi <effect>
37 { 2drop (( x y -- )) }
38 { 3drop (( x y z -- )) }
39 { dup (( x -- x x )) }
40 { 2dup (( x y -- x y x y )) }
41 { 3dup (( x y z -- x y z x y z )) }
42 { rot (( x y z -- y z x )) }
43 { -rot (( x y z -- z x y )) }
44 { dupd (( x y -- x x y )) }
45 { swapd (( x y z -- y x z )) }
46 { nip (( x y -- y )) }
47 { 2nip (( x y z -- z )) }
48 { over (( x y -- x y x )) }
49 { pick (( x y z -- x y z x )) }
50 { swap (( x y -- y x )) }
51 } [ "shuffle" set-word-prop ] assoc-each
53 : infer-shuffle ( shuffle -- )
54 [ in>> length consume-d ] keep ! inputs shuffle
55 [ drop ] [ shuffle dup copy-values dup output-d ] 2bi ! inputs outputs copies
56 [ nip f f ] [ swap zip ] 2bi ! in-d out-d in-r out-r mapping
59 : infer-shuffle-word ( word -- )
60 "shuffle" word-prop infer-shuffle ;
62 : check-declaration ( declaration -- declaration )
63 dup { [ array? ] [ [ class? ] all? ] } 1&&
64 [ bad-declaration-error ] unless ;
66 : infer-declare ( -- )
67 pop-literal nip check-declaration
68 [ length ensure-d ] keep zip
71 \ declare [ infer-declare ] "special" set-word-prop
73 GENERIC: infer-call* ( value known -- )
75 : (infer-call) ( value -- ) dup known infer-call* ;
77 : infer-call ( -- ) pop-d (infer-call) ;
79 \ call [ infer-call ] "special" set-word-prop
81 \ (call) [ infer-call ] "special" set-word-prop
83 M: literal infer-call*
84 [ 1array #drop, ] [ infer-literal-quot ] bi* ;
86 M: curried infer-call*
88 [ uncurry ] infer-quot-here
89 [ quot>> known pop-d [ set-known ] keep ]
90 [ obj>> known pop-d [ set-known ] keep ] bi
93 M: composed infer-call*
95 [ uncompose ] infer-quot-here
96 [ quot2>> known pop-d [ set-known ] keep ]
97 [ quot1>> known pop-d [ set-known ] keep ] bi
100 terminated? get [ 1 infer-r> infer-call ] unless ;
102 M: declared-effect infer-call*
103 [ [ known>> infer-call* ] keep ] with-effect-here check-declared-effect ;
105 M: input-parameter infer-call* \ call unknown-macro-input ;
106 M: object infer-call* \ call bad-macro-input ;
108 : infer-ndip ( word n -- )
109 [ literals get ] 2dip
110 [ '[ _ def>> infer-quot-here ] ]
111 [ '[ _ [ pop ] dip [ infer->r infer-quot-here ] [ infer-r> ] bi ] ] bi*
114 : infer-dip ( -- ) \ dip 1 infer-ndip ;
116 \ dip [ infer-dip ] "special" set-word-prop
118 : infer-2dip ( -- ) \ 2dip 2 infer-ndip ;
120 \ 2dip [ infer-2dip ] "special" set-word-prop
122 : infer-3dip ( -- ) \ 3dip 3 infer-ndip ;
124 \ 3dip [ infer-3dip ] "special" set-word-prop
126 : infer-builder ( quot word -- )
129 [ dup first2 ] dip call make-known
130 [ push-d ] [ 1array ] bi
131 ] dip #call, ; inline
133 : infer-curry ( -- ) [ <curried> ] \ curry infer-builder ;
135 \ curry [ infer-curry ] "special" set-word-prop
137 : infer-compose ( -- ) [ <composed> ] \ compose infer-builder ;
139 \ compose [ infer-compose ] "special" set-word-prop
141 ERROR: bad-executable obj ;
143 M: bad-executable summary
144 drop "execute must be given a word" ;
146 : infer-execute ( -- )
151 \ bad-executable boa time-bomb
154 \ execute [ infer-execute ] "special" set-word-prop
156 \ (execute) [ infer-execute ] "special" set-word-prop
158 : infer-<tuple-boa> ( -- )
160 peek-d literal value>> second 1 + "obj" <array> { tuple } <effect>
163 \ <tuple-boa> [ infer-<tuple-boa> ] "special" set-word-prop
165 \ <tuple-boa> t "flushable" set-word-prop
167 : infer-effect-unsafe ( word -- )
172 : infer-execute-effect-unsafe ( -- )
173 \ (execute) infer-effect-unsafe ;
175 \ execute-effect-unsafe [ infer-execute-effect-unsafe ] "special" set-word-prop
177 : infer-call-effect-unsafe ( -- )
178 \ call infer-effect-unsafe ;
180 \ call-effect-unsafe [ infer-call-effect-unsafe ] "special" set-word-prop
183 \ exit (( n -- * )) apply-word/effect ;
185 \ exit [ infer-exit ] "special" set-word-prop
187 : infer-load-locals ( -- )
189 consume-d dup copy-values dup output-r
190 [ [ f f ] dip ] [ swap zip ] 2bi #shuffle, ;
192 \ load-locals [ infer-load-locals ] "special" set-word-prop
194 : infer-load-local ( -- )
197 \ load-local [ infer-load-local ] "special" set-word-prop
199 :: infer-get-local ( -- )
200 pop-literal nip 1 swap - :> n
202 in-r first copy-value 1array :> out-d
203 in-r copy-values :> out-r
208 out-r in-r zip out-d first in-r first 2array suffix
211 \ get-local [ infer-get-local ] "special" set-word-prop
213 : infer-drop-locals ( -- )
214 f f pop-literal nip consume-r f f #shuffle, ;
216 \ drop-locals [ infer-drop-locals ] "special" set-word-prop
218 : infer-call-effect ( word -- )
219 1 ensure-d first literal value>>
220 add-effect-input add-effect-input
223 { call-effect execute-effect } [
224 dup t "no-compile" set-word-prop
225 dup '[ _ infer-call-effect ] "special" set-word-prop
228 \ if [ infer-if ] "special" set-word-prop
229 \ dispatch [ infer-dispatch ] "special" set-word-prop
231 \ alien-invoke [ infer-alien-invoke ] "special" set-word-prop
232 \ alien-indirect [ infer-alien-indirect ] "special" set-word-prop
233 \ alien-assembly [ infer-alien-assembly ] "special" set-word-prop
234 \ alien-callback [ infer-alien-callback ] "special" set-word-prop
241 inline-cache-miss-tail
250 } [ dup '[ _ do-not-compile ] "special" set-word-prop ] each
252 : infer-special ( word -- )
253 [ current-word set ] [ "special" word-prop call( -- ) ] bi ;
255 : infer-local-reader ( word -- )
256 (( -- value )) apply-word/effect ;
258 : infer-local-writer ( word -- )
259 (( value -- )) apply-word/effect ;
261 : infer-local-word ( word -- )
262 "local-word-def" word-prop infer-quot-here ;
265 declare call (call) dip 2dip 3dip curry compose
266 execute (execute) call-effect-unsafe execute-effect-unsafe if
267 dispatch <tuple-boa> exit load-local load-locals get-local
268 drop-locals do-primitive alien-invoke alien-indirect
270 } [ t "no-compile" set-word-prop ] each
272 ! Exceptions to the above
273 \ curry f "no-compile" set-word-prop
274 \ compose f "no-compile" set-word-prop
276 ! More words not to compile
277 \ clear t "no-compile" set-word-prop
279 : non-inline-word ( word -- )
280 dup depends-on-effect
282 { [ dup "shuffle" word-prop ] [ infer-shuffle-word ] }
283 { [ dup "special" word-prop ] [ infer-special ] }
284 { [ dup "primitive" word-prop ] [ infer-primitive ] }
285 { [ dup "transform-quot" word-prop ] [ apply-transform ] }
286 { [ dup "macro" word-prop ] [ apply-macro ] }
287 { [ dup local? ] [ infer-local-reader ] }
288 { [ dup local-reader? ] [ infer-local-reader ] }
289 { [ dup local-writer? ] [ infer-local-writer ] }
290 { [ dup local-word? ] [ infer-local-word ] }
294 : define-primitive ( word inputs outputs -- )
295 [ 2drop t "primitive" set-word-prop ]
296 [ drop "input-classes" set-word-prop ]
297 [ nip "default-output-classes" set-word-prop ]
300 ! Stack effects for all primitives
301 \ fixnum< { fixnum fixnum } { object } define-primitive
302 \ fixnum< make-foldable
304 \ fixnum<= { fixnum fixnum } { object } define-primitive
305 \ fixnum<= make-foldable
307 \ fixnum> { fixnum fixnum } { object } define-primitive
308 \ fixnum> make-foldable
310 \ fixnum>= { fixnum fixnum } { object } define-primitive
311 \ fixnum>= make-foldable
313 \ eq? { object object } { object } define-primitive
316 \ bignum>fixnum { bignum } { fixnum } define-primitive
317 \ bignum>fixnum make-foldable
319 \ float>fixnum { float } { fixnum } define-primitive
320 \ bignum>fixnum make-foldable
322 \ fixnum>bignum { fixnum } { bignum } define-primitive
323 \ fixnum>bignum make-foldable
325 \ float>bignum { float } { bignum } define-primitive
326 \ float>bignum make-foldable
328 \ fixnum>float { fixnum } { float } define-primitive
329 \ fixnum>float make-foldable
331 \ bignum>float { bignum } { float } define-primitive
332 \ bignum>float make-foldable
334 \ (float>string) { float } { byte-array } define-primitive
335 \ (float>string) make-foldable
337 \ float>bits { real } { integer } define-primitive
338 \ float>bits make-foldable
340 \ double>bits { real } { integer } define-primitive
341 \ double>bits make-foldable
343 \ bits>float { integer } { float } define-primitive
344 \ bits>float make-foldable
346 \ bits>double { integer } { float } define-primitive
347 \ bits>double make-foldable
349 \ both-fixnums? { object object } { object } define-primitive
351 \ fixnum+ { fixnum fixnum } { integer } define-primitive
352 \ fixnum+ make-foldable
354 \ fixnum+fast { fixnum fixnum } { fixnum } define-primitive
355 \ fixnum+fast make-foldable
357 \ fixnum- { fixnum fixnum } { integer } define-primitive
358 \ fixnum- make-foldable
360 \ fixnum-fast { fixnum fixnum } { fixnum } define-primitive
361 \ fixnum-fast make-foldable
363 \ fixnum* { fixnum fixnum } { integer } define-primitive
364 \ fixnum* make-foldable
366 \ fixnum*fast { fixnum fixnum } { fixnum } define-primitive
367 \ fixnum*fast make-foldable
369 \ fixnum/i { fixnum fixnum } { integer } define-primitive
370 \ fixnum/i make-foldable
372 \ fixnum/i-fast { fixnum fixnum } { fixnum } define-primitive
373 \ fixnum/i-fast make-foldable
375 \ fixnum-mod { fixnum fixnum } { fixnum } define-primitive
376 \ fixnum-mod make-foldable
378 \ fixnum/mod { fixnum fixnum } { integer fixnum } define-primitive
379 \ fixnum/mod make-foldable
381 \ fixnum/mod-fast { fixnum fixnum } { fixnum fixnum } define-primitive
382 \ fixnum/mod-fast make-foldable
384 \ fixnum-bitand { fixnum fixnum } { fixnum } define-primitive
385 \ fixnum-bitand make-foldable
387 \ fixnum-bitor { fixnum fixnum } { fixnum } define-primitive
388 \ fixnum-bitor make-foldable
390 \ fixnum-bitxor { fixnum fixnum } { fixnum } define-primitive
391 \ fixnum-bitxor make-foldable
393 \ fixnum-bitnot { fixnum } { fixnum } define-primitive
394 \ fixnum-bitnot make-foldable
396 \ fixnum-shift { fixnum fixnum } { integer } define-primitive
397 \ fixnum-shift make-foldable
399 \ fixnum-shift-fast { fixnum fixnum } { fixnum } define-primitive
400 \ fixnum-shift-fast make-foldable
402 \ bignum= { bignum bignum } { object } define-primitive
403 \ bignum= make-foldable
405 \ bignum+ { bignum bignum } { bignum } define-primitive
406 \ bignum+ make-foldable
408 \ bignum- { bignum bignum } { bignum } define-primitive
409 \ bignum- make-foldable
411 \ bignum* { bignum bignum } { bignum } define-primitive
412 \ bignum* make-foldable
414 \ bignum/i { bignum bignum } { bignum } define-primitive
415 \ bignum/i make-foldable
417 \ bignum-mod { bignum bignum } { bignum } define-primitive
418 \ bignum-mod make-foldable
420 \ bignum/mod { bignum bignum } { bignum bignum } define-primitive
421 \ bignum/mod make-foldable
423 \ bignum-bitand { bignum bignum } { bignum } define-primitive
424 \ bignum-bitand make-foldable
426 \ bignum-bitor { bignum bignum } { bignum } define-primitive
427 \ bignum-bitor make-foldable
429 \ bignum-bitxor { bignum bignum } { bignum } define-primitive
430 \ bignum-bitxor make-foldable
432 \ bignum-bitnot { bignum } { bignum } define-primitive
433 \ bignum-bitnot make-foldable
435 \ bignum-shift { bignum fixnum } { bignum } define-primitive
436 \ bignum-shift make-foldable
438 \ bignum< { bignum bignum } { object } define-primitive
439 \ bignum< make-foldable
441 \ bignum<= { bignum bignum } { object } define-primitive
442 \ bignum<= make-foldable
444 \ bignum> { bignum bignum } { object } define-primitive
445 \ bignum> make-foldable
447 \ bignum>= { bignum bignum } { object } define-primitive
448 \ bignum>= make-foldable
450 \ bignum-bit? { bignum integer } { object } define-primitive
451 \ bignum-bit? make-foldable
453 \ bignum-log2 { bignum } { bignum } define-primitive
454 \ bignum-log2 make-foldable
456 \ byte-array>bignum { byte-array } { bignum } define-primitive
457 \ byte-array>bignum make-foldable
459 \ float= { float float } { object } define-primitive
460 \ float= make-foldable
462 \ float+ { float float } { float } define-primitive
463 \ float+ make-foldable
465 \ float- { float float } { float } define-primitive
466 \ float- make-foldable
468 \ float* { float float } { float } define-primitive
469 \ float* make-foldable
471 \ float/f { float float } { float } define-primitive
472 \ float/f make-foldable
474 \ float-mod { float float } { float } define-primitive
475 \ float-mod make-foldable
477 \ float< { float float } { object } define-primitive
478 \ float< make-foldable
480 \ float<= { float float } { object } define-primitive
481 \ float<= make-foldable
483 \ float> { float float } { object } define-primitive
484 \ float> make-foldable
486 \ float>= { float float } { object } define-primitive
487 \ float>= make-foldable
489 \ float-u< { float float } { object } define-primitive
490 \ float-u< make-foldable
492 \ float-u<= { float float } { object } define-primitive
493 \ float-u<= make-foldable
495 \ float-u> { float float } { object } define-primitive
496 \ float-u> make-foldable
498 \ float-u>= { float float } { object } define-primitive
499 \ float-u>= make-foldable
501 \ (word) { object object object } { word } define-primitive
502 \ (word) make-flushable
504 \ word-code { word } { integer integer } define-primitive
505 \ word-code make-flushable
507 \ special-object { fixnum } { object } define-primitive
508 \ special-object make-flushable
510 \ set-special-object { object fixnum } { } define-primitive
512 \ context-object { fixnum } { object } define-primitive
513 \ context-object make-flushable
515 \ set-context-object { object fixnum } { } define-primitive
517 \ (exists?) { string } { object } define-primitive
519 \ minor-gc { } { } define-primitive
521 \ gc { } { } define-primitive
523 \ compact-gc { } { } define-primitive
525 \ (save-image) { byte-array byte-array } { } define-primitive
527 \ (save-image-and-exit) { byte-array byte-array } { } define-primitive
529 \ data-room { } { byte-array } define-primitive
530 \ data-room make-flushable
532 \ (code-blocks) { } { array } define-primitive
533 \ (code-blocks) make-flushable
535 \ code-room { } { byte-array } define-primitive
536 \ code-room make-flushable
538 \ system-micros { } { integer } define-primitive
539 \ system-micros make-flushable
541 \ nano-count { } { integer } define-primitive
542 \ nano-count make-flushable
544 \ tag { object } { fixnum } define-primitive
547 \ (dlopen) { byte-array } { dll } define-primitive
549 \ (dlsym) { byte-array object } { c-ptr } define-primitive
551 \ dlclose { dll } { } define-primitive
553 \ <byte-array> { integer } { byte-array } define-primitive
554 \ <byte-array> make-flushable
556 \ (byte-array) { integer } { byte-array } define-primitive
557 \ (byte-array) make-flushable
559 \ <displaced-alien> { integer c-ptr } { c-ptr } define-primitive
560 \ <displaced-alien> make-flushable
562 \ alien-signed-cell { c-ptr integer } { integer } define-primitive
563 \ alien-signed-cell make-flushable
565 \ set-alien-signed-cell { integer c-ptr integer } { } define-primitive
567 \ alien-unsigned-cell { c-ptr integer } { integer } define-primitive
568 \ alien-unsigned-cell make-flushable
570 \ set-alien-unsigned-cell { integer c-ptr integer } { } define-primitive
572 \ alien-signed-8 { c-ptr integer } { integer } define-primitive
573 \ alien-signed-8 make-flushable
575 \ set-alien-signed-8 { integer c-ptr integer } { } define-primitive
577 \ alien-unsigned-8 { c-ptr integer } { integer } define-primitive
578 \ alien-unsigned-8 make-flushable
580 \ set-alien-unsigned-8 { integer c-ptr integer } { } define-primitive
582 \ alien-signed-4 { c-ptr integer } { integer } define-primitive
583 \ alien-signed-4 make-flushable
585 \ set-alien-signed-4 { integer c-ptr integer } { } define-primitive
587 \ alien-unsigned-4 { c-ptr integer } { integer } define-primitive
588 \ alien-unsigned-4 make-flushable
590 \ set-alien-unsigned-4 { integer c-ptr integer } { } define-primitive
592 \ alien-signed-2 { c-ptr integer } { fixnum } define-primitive
593 \ alien-signed-2 make-flushable
595 \ set-alien-signed-2 { integer c-ptr integer } { } define-primitive
597 \ alien-unsigned-2 { c-ptr integer } { fixnum } define-primitive
598 \ alien-unsigned-2 make-flushable
600 \ set-alien-unsigned-2 { integer c-ptr integer } { } define-primitive
602 \ alien-signed-1 { c-ptr integer } { fixnum } define-primitive
603 \ alien-signed-1 make-flushable
605 \ set-alien-signed-1 { integer c-ptr integer } { } define-primitive
607 \ alien-unsigned-1 { c-ptr integer } { fixnum } define-primitive
608 \ alien-unsigned-1 make-flushable
610 \ set-alien-unsigned-1 { integer c-ptr integer } { } define-primitive
612 \ alien-float { c-ptr integer } { float } define-primitive
613 \ alien-float make-flushable
615 \ set-alien-float { float c-ptr integer } { } define-primitive
617 \ alien-double { c-ptr integer } { float } define-primitive
618 \ alien-double make-flushable
620 \ set-alien-double { float c-ptr integer } { } define-primitive
622 \ alien-cell { c-ptr integer } { pinned-c-ptr } define-primitive
623 \ alien-cell make-flushable
625 \ set-alien-cell { c-ptr c-ptr integer } { } define-primitive
627 \ alien-address { alien } { integer } define-primitive
628 \ alien-address make-flushable
630 \ slot { object fixnum } { object } define-primitive
631 \ slot make-flushable
633 \ set-slot { object object fixnum } { } define-primitive
635 \ string-nth { fixnum string } { fixnum } define-primitive
636 \ string-nth make-flushable
638 \ set-string-nth-slow { fixnum fixnum string } { } define-primitive
639 \ set-string-nth-fast { fixnum fixnum string } { } define-primitive
641 \ resize-array { integer array } { array } define-primitive
642 \ resize-array make-flushable
644 \ resize-byte-array { integer byte-array } { byte-array } define-primitive
645 \ resize-byte-array make-flushable
647 \ resize-string { integer string } { string } define-primitive
648 \ resize-string make-flushable
650 \ <array> { integer object } { array } define-primitive
651 \ <array> make-flushable
653 \ all-instances { } { array } define-primitive
655 \ size { object } { fixnum } define-primitive
656 \ size make-flushable
658 \ die { } { } define-primitive
660 \ (fopen) { byte-array byte-array } { alien } define-primitive
662 \ fgetc { alien } { object } define-primitive
664 \ fwrite { c-ptr integer alien } { } define-primitive
666 \ fputc { object alien } { } define-primitive
668 \ fread { integer alien } { object } define-primitive
670 \ fflush { alien } { } define-primitive
672 \ fseek { integer integer alien } { } define-primitive
674 \ ftell { alien } { integer } define-primitive
676 \ fclose { alien } { } define-primitive
678 \ <wrapper> { object } { wrapper } define-primitive
679 \ <wrapper> make-foldable
681 \ (clone) { object } { object } define-primitive
682 \ (clone) make-flushable
684 \ <string> { integer integer } { string } define-primitive
685 \ <string> make-flushable
687 \ array>quotation { array } { quotation } define-primitive
688 \ array>quotation make-flushable
690 \ quotation-code { quotation } { integer integer } define-primitive
691 \ quotation-code make-flushable
693 \ <tuple> { tuple-layout } { tuple } define-primitive
694 \ <tuple> make-flushable
696 \ datastack { } { array } define-primitive
697 \ datastack make-flushable
699 \ check-datastack { array integer integer } { object } define-primitive
700 \ check-datastack make-flushable
702 \ retainstack { } { array } define-primitive
703 \ retainstack make-flushable
705 \ callstack { } { callstack } define-primitive
706 \ callstack make-flushable
708 \ callstack>array { callstack } { array } define-primitive
709 \ callstack>array make-flushable
711 \ (sleep) { integer } { } define-primitive
713 \ become { array array } { } define-primitive
715 \ innermost-frame-executing { callstack } { object } define-primitive
717 \ innermost-frame-scan { callstack } { fixnum } define-primitive
719 \ set-innermost-frame-quot { quotation callstack } { } define-primitive
721 \ dll-valid? { object } { object } define-primitive
723 \ modify-code-heap { array object object } { } define-primitive
725 \ unimplemented { } { } define-primitive
727 \ jit-compile { quotation } { } define-primitive
729 \ lookup-method { object array } { word } define-primitive
731 \ reset-dispatch-stats { } { } define-primitive
732 \ dispatch-stats { } { byte-array } define-primitive
734 \ optimized? { word } { object } define-primitive
736 \ strip-stack-traces { } { } define-primitive
738 \ <callback> { integer word } { alien } define-primitive
740 \ enable-gc-events { } { } define-primitive
741 \ disable-gc-events { } { object } define-primitive
743 \ profiling { object } { } define-primitive
745 \ (identity-hashcode) { object } { fixnum } define-primitive
747 \ compute-identity-hashcode { object } { } define-primitive
749 \ (exit) { integer } { } define-primitive
751 \ quot-compiled? { quotation } { object } define-primitive