1 ! Copyright (C) 2005, 2008 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: arrays definitions hashtables kernel kernel.private math
4 namespaces make sequences sequences.private strings vectors
5 words quotations memory combinators generic classes
6 classes.algebra classes.builtin classes.private slots.private
7 slots compiler.units math.private accessors assocs effects ;
10 PREDICATE: tuple-class < class
11 "metaclass" word-prop tuple-class eq? ;
13 ERROR: not-a-tuple object ;
15 : check-tuple ( object -- tuple )
16 dup tuple? [ not-a-tuple ] unless ; inline
18 : all-slots ( class -- slots )
19 superclasses [ "slots" word-prop ] map concat ;
21 PREDICATE: immutable-tuple-class < tuple-class ( class -- ? )
22 all-slots [ read-only>> ] all? ;
26 : tuple-layout ( class -- layout )
29 : layout-of ( tuple -- layout )
30 1 slot { array } declare ; inline
32 M: tuple class layout-of 2 slot { word } declare ;
34 : tuple-size ( tuple -- size )
35 layout-of second ; inline
37 : prepare-tuple>array ( tuple -- n tuple layout )
38 check-tuple [ tuple-size ] [ ] [ layout-of ] tri ;
40 : copy-tuple-slots ( n tuple -- array )
41 [ array-nth ] curry map ;
43 : check-slots ( seq class -- seq class )
46 class>> 2dup instance?
47 [ 2drop ] [ bad-slot-value ] if
49 ] if-bootstrapping ; inline
53 : initial-values ( class -- slots )
54 all-slots [ initial>> ] map ;
56 : pad-slots ( slots class -- slots' class )
57 [ initial-values over length tail append ] keep ; inline
59 : tuple>array ( tuple -- array )
61 [ copy-tuple-slots ] dip
64 : tuple-slots ( tuple -- seq )
65 prepare-tuple>array drop copy-tuple-slots ;
67 GENERIC: slots>tuple ( seq class -- tuple )
69 M: tuple-class slots>tuple
71 tuple-layout <tuple> [
73 [ [ set-array-nth ] curry ]
77 : >tuple ( seq -- tuple )
80 ERROR: bad-superclass class ;
82 : tuple= ( tuple1 tuple2 -- ? )
83 2dup [ tuple? ] both? [
84 2dup [ layout-of ] bi@ eq? [
86 [ [ [ drop array-nth ] [ nip array-nth ] 3bi = ] 2curry ]
89 ] [ 2drop f ] if ; inline
93 : tuple-predicate-quot/1 ( class -- quot )
94 #! Fast path for tuples with no superclass
95 [ ] curry [ layout-of 7 slot ] [ eq? ] surround 1quotation
96 [ dup tuple? ] [ [ drop f ] if ] surround ;
98 : tuple-instance? ( object class offset -- ? )
102 [ swap slot eq? ] [ 3drop f ] if
103 ] [ 3drop f ] if ; inline
105 : layout-class-offset ( echelon -- n )
108 : tuple-predicate-quot ( class echelon -- quot )
109 layout-class-offset [ tuple-instance? ] 2curry ;
111 : echelon-of ( class -- n )
114 : define-tuple-predicate ( class -- )
116 { 1 [ tuple-predicate-quot/1 ] }
117 [ tuple-predicate-quot ]
118 } case define-predicate ;
120 : class-size ( class -- n )
121 superclasses [ "slots" word-prop length ] sigma ;
123 : (instance-check-quot) ( class -- quot )
126 [ "predicate" word-prop % ]
127 [ [ literalize , \ bad-slot-value , ] [ ] make , ] bi
131 : (fixnum-check-quot) ( class -- quot )
132 (instance-check-quot) fixnum "coercer" word-prop prepend ;
134 : instance-check-quot ( class -- quot )
136 { [ dup object bootstrap-word eq? ] [ drop [ ] ] }
137 { [ dup "coercer" word-prop ] [ "coercer" word-prop ] }
138 { [ dup \ fixnum class<= ] [ (fixnum-check-quot) ] }
139 [ (instance-check-quot) ]
142 : boa-check-quot ( class -- quot )
143 all-slots [ class>> instance-check-quot ] map spread>quot
146 : define-boa-check ( class -- )
147 dup boa-check-quot "boa-check" set-word-prop ;
149 : tuple-prototype ( class -- prototype )
150 [ initial-values ] keep
151 over [ ] contains? [ slots>tuple ] [ 2drop f ] if ;
153 : define-tuple-prototype ( class -- )
154 dup tuple-prototype "prototype" set-word-prop ;
156 : prepare-slots ( slots superclass -- slots' )
157 [ make-slots ] [ class-size 2 + ] bi* finalize-slots ;
159 : define-tuple-slots ( class -- )
160 dup "slots" word-prop over superclass prepare-slots
163 : make-tuple-layout ( class -- layout )
167 [ [ superclass class-size ] [ "slots" word-prop length ] bi + , ]
168 [ superclasses length 1- , ]
169 [ superclasses [ [ , ] [ hashcode , ] bi ] each ]
173 : define-tuple-layout ( class -- )
174 dup make-tuple-layout "layout" set-word-prop ;
176 : compute-slot-permutation ( new-slots old-slots -- triples )
177 [ [ [ name>> ] map ] bi@ [ index ] curry map ]
178 [ drop [ class>> ] map ]
179 [ drop [ initial>> ] map ]
182 : update-slot ( old-values n class initial -- value )
184 [ [ swap nth dup ] dip instance? ] dip swap
186 ] [ [ 3drop ] dip ] if ;
188 : apply-slot-permutation ( old-values triples -- new-values )
189 [ first3 update-slot ] with map ;
191 : permute-slots ( old-values layout -- new-values )
192 [ first all-slots ] [ outdated-tuples get at ] bi
193 compute-slot-permutation
194 apply-slot-permutation ;
196 : update-tuple ( tuple -- newtuple )
197 [ tuple-slots ] [ layout-of ] bi
198 [ permute-slots ] [ first ] bi
201 : outdated-tuple? ( tuple assoc -- ? )
203 [ [ layout-of ] dip key? ]
204 [ drop class "forgotten" word-prop not ]
208 : update-tuples ( -- )
210 dup assoc-empty? [ drop ] [
211 [ outdated-tuple? ] curry instances
212 dup [ update-tuple ] map become
215 [ update-tuples ] update-tuples-hook set-global
217 : update-tuples-after ( class -- )
218 [ all-slots ] [ tuple-layout ] bi outdated-tuples get set-at ;
220 M: tuple-class update-class
223 [ define-tuple-layout ]
224 [ define-tuple-slots ]
225 [ define-tuple-predicate ]
226 [ define-tuple-prototype ]
229 : define-new-tuple-class ( class superclass slots -- )
230 [ drop f f tuple-class define-class ]
231 [ nip "slots" set-word-prop ]
232 [ 2drop update-classes ]
235 : subclasses ( class -- classes )
236 class-usages [ tuple-class? ] filter ;
238 : each-subclass ( class quot -- )
239 [ subclasses ] dip each ; inline
241 : redefine-tuple-class ( class superclass slots -- )
245 [ update-tuples-after ]
250 [ define-new-tuple-class ]
253 : tuple-class-unchanged? ( class superclass slots -- ? )
255 [ [ superclass ] [ bootstrap-word ] bi* = ]
256 [ [ "slots" word-prop ] dip = ] 2bi* and ;
258 : valid-superclass? ( class -- ? )
259 [ tuple-class? ] [ tuple eq? ] bi or ;
261 : check-superclass ( superclass -- )
262 dup valid-superclass? [ bad-superclass ] unless drop ;
264 GENERIC# (define-tuple-class) 2 ( class superclass slots -- )
268 : define-tuple-class ( class superclass slots -- )
269 over check-superclass
271 (define-tuple-class) ;
273 M: word (define-tuple-class)
274 define-new-tuple-class ;
276 M: tuple-class (define-tuple-class)
277 3dup tuple-class-unchanged?
278 [ 3drop ] [ redefine-tuple-class ] if ;
280 : thrower-effect ( slots -- effect )
281 [ dup array? [ first ] when ] map f <effect> t >>terminated? ;
283 : define-error-class ( class superclass slots -- )
284 [ define-tuple-class ]
285 [ 2drop reset-generic ]
287 [ dup [ boa throw ] curry ]
293 : boa-effect ( class -- effect )
294 [ all-slots [ name>> ] map ] [ name>> 1array ] bi <effect> ;
296 : define-boa-word ( word class -- )
297 [ [ boa ] curry ] [ boa-effect ] bi define-inline ;
299 M: tuple-class reset-class
301 dup "slots" word-prop [
303 [ reader-word method forget ]
304 [ writer-word method forget ] 2bi
308 [ { "layout" "slots" "boa-check" "prototype" } reset-props ]
312 M: tuple-class rank-class drop 0 ;
314 M: tuple-class instance?
315 dup echelon-of layout-class-offset tuple-instance? ;
317 M: tuple-class (flatten-class) dup set ;
319 M: tuple-class (classes-intersect?)
321 { [ over tuple eq? ] [ 2drop t ] }
322 { [ over builtin-class? ] [ 2drop f ] }
323 { [ over tuple-class? ] [ [ class<= ] [ swap class<= ] 2bi or ] }
324 [ swap classes-intersect? ]
327 M: tuple clone (clone) ;
329 M: tuple equal? over tuple? [ tuple= ] [ 2drop f ] if ;
331 GENERIC: tuple-hashcode ( n tuple -- x )
333 M: tuple tuple-hashcode
335 [ class hashcode ] [ tuple-size ] [ ] tri
337 swapd array-nth hashcode* sequence-hashcode-step
339 ] recursive-hashcode ;
341 M: tuple hashcode* tuple-hashcode ;
344 dup "prototype" word-prop
345 [ (clone) ] [ tuple-layout <tuple> ] ?if ;
348 [ "boa-check" word-prop [ call ] when* ]
352 M: tuple-class initial-value* new ;