factor/core/generic/tuple.factor

115 lines
3.0 KiB
Factor

! Copyright (C) 2005, 2006 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
IN: generic
USING: arrays definitions errors hashtables kernel
kernel-internals math namespaces parser sequences
sequences-internals strings vectors words ;
IN: kernel-internals
: tuple= ( tuple1 tuple2 -- ? )
2dup [ array-capacity ] 2apply number= [
dup array-capacity
[ 2dup swap array-nth >r pick array-nth r> = ] all? 2nip
] [
2drop f
] if ;
: <tuple> ( class n -- tuple )
f <array> [ 2 set-slot ] keep tuple-type become ;
IN: generic
: class ( object -- class )
dup tuple? [ 2 slot ] [ type type>class ] if ; inline
: tuple-predicate ( class -- )
dup predicate-word [
[ dup tuple? ] %
[ [ 2 slot ] % over literalize , \ eq? , ] [ ] make ,
[ [ drop f ] if ] %
] [ ] make define-predicate ;
: forget-tuple ( class -- )
dup forget
"predicate" word-prop
dup empty? [ drop ] [ first forget ] if ;
: check-shape ( class slots -- )
>r in get lookup dup [
dup tuple-size r> length 2 + =
[ drop ] [ forget-tuple ] if
] [
r> 2drop
] if ;
: delegate-slots { { 3 object delegate set-delegate } } ;
: define-tuple-slots ( class slots -- )
2dup "slot-names" set-word-prop
2dup length 2 + "tuple-size" set-word-prop
dupd 4 simple-slots
2dup delegate-slots swap append "slots" set-word-prop
define-slots ;
TUPLE: check-tuple class ;
: check-tuple ( class -- )
dup tuple-class? [ drop ] [ <check-tuple> throw ] if ;
: define-constructor ( word class def -- )
over check-tuple [
>r dup literalize , tuple-size , \ <tuple> , r> %
] [ ] make
define-compound ;
: default-constructor-quot ( class -- quot )
"slots" word-prop 1 tail <reversed> [ fourth ] map
[ length \ tuck <array> ] keep 2array
flip concat >quotation ;
: default-constructor ( class -- )
dup create-constructor
dup reset-generic
2dup "constructor" set-word-prop
dup rot dup default-constructor-quot define-constructor
t "inline" set-word-prop ;
: define-tuple-class ( class slots -- )
2dup check-shape
>r create-in
dup tuple-predicate
\ tuple bootstrap-word over set-superclass
dup define-class
dup r> define-tuple-slots
default-constructor ;
M: tuple clone
(clone) dup delegate clone over set-delegate ;
M: tuple hashcode 2 slot hashcode ;
M: tuple equal?
over tuple? [ tuple= ] [ 2drop f ] if ;
: (delegates) ( obj -- )
[ dup , delegate (delegates) ] when* ;
: delegates ( obj -- seq ) [ (delegates) ] { } make ;
: is? ( obj quot -- ? ) >r delegates r> contains? ; inline
: >tuple ( seq -- tuple )
>vector dup first tuple-size over set-length
>array tuple-type become ;
GENERIC: tuple>array ( tuple -- array )
M: tuple tuple>array (clone) array-type become ;
: tuple-slots ( tuple -- seq ) tuple>array 2 tail ;
! Definition protocol
M: tuple-class forget
dup "constructor" word-prop forget forget-class ;