1 ! Copyright (C) 2008, 2009 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: accessors alien alien.c-types alien.data alien.parser
4 assocs byte-arrays classes compiler.units functors kernel lexer
5 libc math math.vectors math.vectors.private
6 math.vectors.specialization namespaces
7 parser prettyprint.custom sequences sequences.private strings
8 summary vocabs vocabs.loader vocabs.parser vocabs.generated
9 words fry combinators make ;
10 IN: specialized-arrays
12 MIXIN: specialized-array
14 INSTANCE: specialized-array sequence
16 GENERIC: direct-array-syntax ( obj -- word )
18 ERROR: bad-byte-array-length byte-array type ;
20 M: bad-byte-array-length summary
21 drop "Byte array length doesn't divide type width" ;
23 : (underlying) ( n c-type -- array )
24 heap-size * (byte-array) ; inline
26 : <underlying> ( n type -- array )
27 heap-size * <byte-array> ; inline
31 FUNCTOR: define-array ( T -- )
33 A DEFINES-CLASS ${T}-array
34 S DEFINES-CLASS ${T}-sequence
37 <direct-A> DEFINES <direct-${A}>
38 malloc-A DEFINES malloc-${A}
40 byte-array>A DEFINES byte-array>${A}
45 NTH [ T dup c-type-getter-boxer array-accessor ]
46 SET-NTH [ T dup c-setter array-accessor ]
53 { underlying c-ptr read-only }
54 { length array-capacity read-only } ;
56 : <direct-A> ( alien len -- specialized-array ) A boa ; inline
58 : <A> ( n -- specialized-array )
59 [ \ T <underlying> ] keep <direct-A> ; inline
61 : (A) ( n -- specialized-array )
62 [ \ T (underlying) ] keep <direct-A> ; inline
64 : malloc-A ( len -- specialized-array )
65 [ \ T heap-size calloc ] keep <direct-A> ; inline
67 : byte-array>A ( byte-array -- specialized-array )
68 >c-ptr dup length \ T heap-size /mod 0 =
69 [ drop \ T bad-byte-array-length ] unless
72 M: A new-underlying drop byte-array>A ;
74 M: A clone [ underlying>> clone ] [ length>> ] bi <direct-A> ; inline
76 M: A length length>> ; inline
78 M: A nth-unsafe underlying>> NTH call ; inline
80 M: A set-nth-unsafe underlying>> SET-NTH call ; inline
82 : >A ( seq -- specialized-array ) A new clone-like ;
84 M: A like drop dup A instance? [ >A ] unless ; inline
86 M: A new-sequence drop (A) ; inline
88 M: A equal? over A instance? [ sequence= ] [ 2drop f ] if ;
92 [ \ T heap-size * ] [ underlying>> ] bi*
97 M: A byte-length length \ T heap-size * ; inline
99 M: A element-type drop \ T ; inline
101 M: A direct-array-syntax drop \ A@ ;
103 M: A pprint-delims drop \ A{ \ } ;
105 M: A >pprint-sequence ;
107 SYNTAX: A{ \ } [ >A ] parse-literal ;
108 SYNTAX: A@ scan-object scan-object <direct-A> suffix! ;
110 INSTANCE: A specialized-array
112 A T c-type-boxed-class f specialize-vector-words
116 GENERIC: (underlying-type) ( c-type -- c-type' )
118 M: string (underlying-type) c-types get at ;
119 M: word (underlying-type) "c-type" word-prop ;
121 : underlying-type ( c-type -- c-type' )
122 dup (underlying-type) {
123 { [ dup not ] [ drop no-c-type ] }
124 { [ dup c-type-name? ] [ nip underlying-type ] }
128 : specialized-array-vocab ( c-type -- vocab )
130 "specialized-arrays.instances." %
131 [ vocabulary>> % "." % ]
138 : define-array-vocab ( type -- vocab )
140 [ specialized-array-vocab ] [ '[ _ define-array ] ] bi
143 M: c-type-name require-c-array define-array-vocab drop ;
145 ERROR: specialized-array-vocab-not-loaded c-type ;
147 M: c-type-name c-array-constructor
149 dup [ name>> "<" "-array>" surround ] [ specialized-array-vocab ] bi lookup
150 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
152 M: c-type-name c-(array)-constructor
154 dup [ name>> "(" "-array)" surround ] [ specialized-array-vocab ] bi lookup
155 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
157 M: c-type-name c-direct-array-constructor
159 dup [ name>> "<direct-" "-array>" surround ] [ specialized-array-vocab ] bi lookup
160 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
162 SYNTAX: SPECIALIZED-ARRAYS:
163 ";" parse-tokens [ parse-c-type define-array-vocab use-vocab ] each ;
165 SYNTAX: SPECIALIZED-ARRAY:
166 scan-c-type define-array-vocab use-vocab ;
168 "prettyprint" vocab [
169 "specialized-arrays.prettyprint" require
173 "specialized-arrays.mirrors" require