1 ! Copyright (C) 2008, 2009 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: accessors alien alien.c-types assocs byte-arrays classes
4 compiler.units functors io kernel lexer libc math
5 math.vectors.specialization namespaces parser
6 prettyprint.custom sequences sequences.private strings summary
7 vocabs vocabs.loader vocabs.parser words ;
10 MIXIN: specialized-array
12 INSTANCE: specialized-array sequence
14 GENERIC: direct-array-syntax ( obj -- word )
16 ERROR: bad-byte-array-length byte-array type ;
18 M: bad-byte-array-length summary
19 drop "Byte array length doesn't divide type width" ;
21 : (underlying) ( n c-type -- array )
22 heap-size * (byte-array) ; inline
24 : <underlying> ( n type -- array )
25 heap-size * <byte-array> ; inline
29 FUNCTOR: define-array ( T -- )
31 A DEFINES-CLASS ${T}-array
32 S DEFINES-CLASS ${T}-sequence
35 <direct-A> DEFINES <direct-${A}>
36 malloc-A DEFINES malloc-${A}
38 byte-array>A DEFINES byte-array>${A}
43 NTH [ T dup c-type-getter-boxer array-accessor ]
44 SET-NTH [ T dup c-setter array-accessor ]
51 { underlying c-ptr read-only }
52 { length array-capacity read-only } ;
54 : <direct-A> ( alien len -- specialized-array ) A boa ; inline
56 : <A> ( n -- specialized-array ) [ T <underlying> ] keep <direct-A> ; inline
58 : (A) ( n -- specialized-array ) [ T (underlying) ] keep <direct-A> ; inline
60 : malloc-A ( len -- specialized-array ) [ T heap-size calloc ] keep <direct-A> ; inline
62 : byte-array>A ( byte-array -- specialized-array )
63 dup length T heap-size /mod 0 = [ drop T bad-byte-array-length ] unless
66 M: A clone [ underlying>> clone ] [ length>> ] bi <direct-A> ; inline
68 M: A length length>> ; inline
70 M: A nth-unsafe underlying>> NTH call ; inline
72 M: A set-nth-unsafe underlying>> SET-NTH call ; inline
74 : >A ( seq -- specialized-array ) A new clone-like ;
76 M: A like drop dup A instance? [ >A ] unless ; inline
78 M: A new-sequence drop (A) ; inline
80 M: A equal? over A instance? [ sequence= ] [ 2drop f ] if ;
84 [ T heap-size * ] [ underlying>> ] bi*
89 M: A byte-length underlying>> length ; inline
90 M: A pprint-delims drop \ A{ \ } ;
91 M: A >pprint-sequence ;
93 SYNTAX: A{ \ } [ >A ] parse-literal ;
94 SYNTAX: A@ scan-object scan-object <direct-A> parsed ;
96 INSTANCE: A specialized-array
98 A T c-type-boxed-class f specialize-vector-words
102 : underlying-type ( c-type -- c-type' )
103 dup c-types get at string? [
104 c-types get at underlying-type
107 : specialized-array-vocab ( c-type -- vocab )
108 "specialized-arrays.instances." prepend ;
110 : defining-array-message ( type -- )
111 "quiet" get [ drop ] [
112 "Generating specialized " " arrays..." surround print
117 : define-array-vocab ( type -- vocab )
119 dup specialized-array-vocab vocab
121 [ defining-array-message ]
124 dup specialized-array-vocab
125 [ define-array ] with-current-vocab
126 ] with-compilation-unit
128 [ specialized-array-vocab ]
132 M: string require-c-array define-array-vocab drop ;
134 ERROR: specialized-array-vocab-not-loaded c-type ;
136 M: string c-array-constructor
138 dup [ "<" "-array>" surround ] [ specialized-array-vocab ] bi lookup
139 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
141 M: string c-(array)-constructor
143 dup [ "(" "-array)" surround ] [ specialized-array-vocab ] bi lookup
144 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
146 M: string c-direct-array-constructor
148 dup [ "<direct-" "-array>" surround ] [ specialized-array-vocab ] bi lookup
149 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
151 SYNTAX: SPECIALIZED-ARRAY:
152 scan define-array-vocab use-vocab ;
154 "prettyprint" vocab [
155 "specialized-arrays.prettyprint" require