1 ! Copyright (C) 2008, 2010 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 namespaces
6 parser prettyprint.custom sequences sequences.private strings
7 summary vocabs vocabs.loader vocabs.parser vocabs.generated
8 words fry combinators make ;
11 MIXIN: specialized-array
13 INSTANCE: specialized-array sequence
15 GENERIC: direct-array-syntax ( obj -- word )
17 ERROR: bad-byte-array-length byte-array type ;
19 M: bad-byte-array-length summary
20 drop "Byte array length doesn't divide type width" ;
22 ERROR: not-a-byte-array alien ;
24 M: not-a-byte-array summary
25 drop "Not a byte array" ;
27 : (underlying) ( n c-type -- array )
28 heap-size * (byte-array) ; inline
30 : <underlying> ( n type -- array )
31 heap-size * <byte-array> ; inline
35 FUNCTOR: define-array ( T -- )
37 A DEFINES-CLASS ${T}-array
40 <direct-A> DEFINES <direct-${A}>
41 malloc-A DEFINES malloc-${A}
43 byte-array>A DEFINES byte-array>${A}
48 NTH [ T dup c-type-getter-boxer array-accessor ]
49 SET-NTH [ T dup c-setter array-accessor ]
54 { underlying c-ptr read-only }
55 { length array-capacity read-only } ; final
57 : <direct-A> ( alien len -- specialized-array ) A boa ; inline
59 : <A> ( n -- specialized-array )
60 [ \ T <underlying> ] keep <direct-A> ; inline
62 : (A) ( n -- specialized-array )
63 [ \ T (underlying) ] keep <direct-A> ; inline
65 : malloc-A ( len -- specialized-array )
66 [ \ T heap-size calloc ] keep <direct-A> ; inline
68 : byte-array>A ( byte-array -- specialized-array )
69 >c-ptr dup byte-array? [
70 dup length \ T heap-size /mod 0 =
72 [ drop \ T bad-byte-array-length ] if
73 ] [ not-a-byte-array ] if ; inline
75 M: A clone [ underlying>> clone ] [ length>> ] bi <direct-A> ; inline
77 M: A length length>> ; inline
79 M: A nth-unsafe underlying>> NTH call ; inline
81 M: A set-nth-unsafe underlying>> SET-NTH call ; inline
83 : >A ( seq -- specialized-array ) A new clone-like ;
85 M: A like drop dup A instance? [ >A ] unless ; inline
87 M: A new-sequence drop (A) ; inline
89 M: A equal? over A instance? [ sequence= ] [ 2drop f ] if ;
93 [ \ T heap-size * ] [ underlying>> ] bi*
98 M: A byte-length length \ T heap-size * ; inline
100 M: A direct-array-syntax drop \ A@ ;
102 M: A pprint-delims drop \ A{ \ } ;
104 M: A >pprint-sequence ;
106 SYNTAX: A{ \ } [ >A ] parse-literal ;
107 SYNTAX: A@ scan-object scan-object <direct-A> suffix! ;
109 INSTANCE: A specialized-array
111 M: A vs+ [ + \ T c-type-clamp ] 2map ; inline
112 M: A vs- [ - \ T c-type-clamp ] 2map ; inline
113 M: A vs* [ * \ T c-type-clamp ] 2map ; inline
115 M: A v*high [ * \ T heap-size neg shift ] 2map ; inline
119 GENERIC: underlying-type ( c-type -- c-type' )
121 M: c-type-word underlying-type
122 dup "c-type" word-prop {
123 { [ dup not ] [ drop no-c-type ] }
124 { [ dup pointer? ] [ 2drop void* ] }
125 { [ dup c-type-word? ] [ nip underlying-type ] }
129 M: pointer underlying-type
132 : specialized-array-vocab ( c-type -- vocab )
134 "specialized-arrays.instances." %
135 [ vocabulary>> % "." % ]
142 : define-array-vocab ( type -- vocab )
144 [ specialized-array-vocab ] [ '[ _ define-array ] ] bi
147 ERROR: specialized-array-vocab-not-loaded c-type ;
149 M: c-type-word c-array-constructor
151 dup [ name>> "<" "-array>" surround ] [ specialized-array-vocab ] bi lookup
152 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
154 M: pointer c-array-constructor drop void* c-array-constructor ;
156 M: c-type-word c-(array)-constructor
158 dup [ name>> "(" "-array)" surround ] [ specialized-array-vocab ] bi lookup
159 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
161 M: pointer c-(array)-constructor drop void* c-(array)-constructor ;
163 M: c-type-word c-direct-array-constructor
165 dup [ name>> "<direct-" "-array>" surround ] [ specialized-array-vocab ] bi lookup
166 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
168 M: pointer c-direct-array-constructor drop void* c-direct-array-constructor ;
170 SYNTAX: SPECIALIZED-ARRAYS:
171 ";" parse-tokens [ parse-c-type define-array-vocab use-vocab ] each ;
173 SYNTAX: SPECIALIZED-ARRAY:
174 scan-c-type define-array-vocab use-vocab ;
176 "prettyprint" vocab [
177 "specialized-arrays.prettyprint" require
181 "specialized-arrays.mirrors" require