steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

for typed words, put the specialized definition in a gensym, and check the input types and declare the output types in the inlined outer word so the checks can be cleared by the compiler when possible

Joe Groff 2009-09-02 18:45:08 -05:00
parent 35f0c31916
commit 5d24e48f8c
1 changed files with 32 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
extra/typed/typed.factor
View File

@ -1,6 +1,7 @@
! (c)Joe Groff bsd license ! (c)Joe Groff bsd license
USING: accessors combinators definitions effects fry hints USING: accessors combinators combinators.short-circuit
kernel kernel.private parser sequences words ; definitions effects fry hints kernel kernel.private namespaces
parser quotations see.private sequences words ;
IN: typed IN: typed
ERROR: type-mismatch-error word expected-types ; ERROR: type-mismatch-error word expected-types ;
@ -43,24 +44,41 @@ ERROR: output-mismatch-error < type-mismatch-error ;
! defining typed words ! defining typed words
PREDICATE: typed < word "typed" word-prop ; : typed-gensym-quot ( def word effect -- quot )
[ nip effect-in-types swap '[ _ declare @ ] ]
: typed-def ( word def effect -- quot )
[ swap ] dip
[ effect-in-types dup typed-stack-effect? [ typed-inputs ] [ 2drop ] if ]
[ effect-out-types dup typed-stack-effect? [ typed-outputs ] [ 2drop ] if ] 2bi ; [ effect-out-types dup typed-stack-effect? [ typed-outputs ] [ 2drop ] if ] 2bi ;
: define-typed-gensym ( word def effect -- gensym )
[ 3drop gensym dup ]
[ [ swap ] dip typed-gensym-quot ]
[ 2nip ] 3tri define-declared ;
PREDICATE: typed < word "typed-word" word-prop ;
: typed-quot ( quot word effect -- quot' )
[ effect-in-types dup typed-stack-effect? [ typed-inputs ] [ 2drop ] if ]
[ nip effect-out-types dup typed-stack-effect? [ '[ @ _ declare ] ] [ drop ] if ] 2bi ;
: (typed-def) ( word def effect -- quot )
[ define-typed-gensym ] 3keep
[ drop [ swap "typed-word" set-word-prop ] [ [ 1quotation ] dip ] 2bi ] dip
typed-quot ;
: typed-def ( word def effect -- quot )
dup {
[ effect-in-types typed-stack-effect? ]
[ effect-out-types typed-stack-effect? ]
} 1|| [ (typed-def) ] [ drop nip ] if ;
: define-typed ( word def effect -- ) : define-typed ( word def effect -- )
{ [ [ 2drop ] [ typed-def ] [ 2nip ] 3tri define-inline ]
[ [ 2drop ] [ typed-def ] [ 2nip ] 3tri define-declared ] [ drop "typed-def" set-word-prop ]
[ nip effect-in-types "input-classes" set-word-prop ] [ 2drop "typed-word" word-prop \ word set-global ] 3tri ;
[ nip effect-out-types "default-output-classes" set-word-prop ]
[ drop "typed" set-word-prop ]
} 3cleave ;
SYNTAX: TYPED: SYNTAX: TYPED:
(:) define-typed ; (:) define-typed ;
M: typed definer drop \ TYPED: \ ; ; M: typed definer drop \ TYPED: \ ; ;
M: typed definition "typed" word-prop ; M: typed definition "typed-def" word-prop ;
M: typed declarations. "typed-word" word-prop declarations. ;