1 ! Copyright (C) 2008, 2010 Slava Pestov.
2 ! See https://factorcode.org/license.txt for BSD license.
3 USING: accessors alien alien.c-types alien.data alien.parser
4 byte-arrays classes combinators fry functors kernel lexer locals
5 make math math.vectors parser prettyprint.custom sequences
6 sequences.private vocabs.generated vocabs.loader vocabs.parser
10 MIXIN: specialized-array
12 INSTANCE: specialized-array sequence
14 : (underlying) ( n c-type -- array )
15 heap-size * (byte-array) ; inline
17 : <underlying> ( n type -- array )
18 heap-size * <byte-array> ; inline
20 GENERIC: underlying-type ( c-type -- c-type' )
22 M: c-type-word underlying-type
23 dup "c-type" word-prop {
24 { [ dup not ] [ drop no-c-type ] }
25 { [ dup pointer? ] [ 2drop void* ] }
26 { [ dup c-type-word? ] [ nip underlying-type ] }
30 M: pointer underlying-type
35 GENERIC: nth-c-ptr ( n seq -- displaced-alien )
36 GENERIC: direct-like ( alien len exemplar -- seq )
38 M: byte-array nth-c-ptr <displaced-alien> ; inline
39 M: byte-array direct-like drop uchar <c-direct-array> ; inline
41 <FUNCTOR: define-array ( T -- )
43 A DEFINES-CLASS ${T}-array
46 <direct-A> DEFINES <direct-${A}>
52 { underlying c-ptr read-only }
53 { length array-capacity read-only } ; final
55 : <direct-A> ( alien len -- specialized-array ) A boa ; inline
57 M: A direct-like drop <direct-A> ; 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 M: A clone [ underlying>> clone ] [ length>> ] bi <direct-A> ; inline
67 M: A length length>> ; inline
69 M: A nth-unsafe [ integer>fixnum ] dip underlying>> \ T alien-element ; inline
71 M: A nth-c-ptr underlying>> \ T array-accessor drop swap <displaced-alien> ; inline
73 M: A set-nth-unsafe [ integer>fixnum ] dip underlying>> \ T set-alien-element ; inline
75 M: A like drop dup A instance? [ \ T >c-array ] unless ; inline
77 M: A new-sequence drop (A) ; inline
79 M: A equal? over A instance? [ sequence= ] [ 2drop f ] if ;
83 [ \ T heap-size * ] [ underlying>> ] bi*
88 M: A element-size drop \ T heap-size ; inline
90 M: A underlying-type drop \ T ;
92 M: A pprint-delims drop \ A{ \ } ;
94 M: A >pprint-sequence ;
96 SYNTAX: A{ \ } [ \ T >c-array ] parse-literal ;
98 INSTANCE: A specialized-array
100 M: A vs+ [ + \ T c-type-clamp ] 2map ; inline
101 M: A vs- [ - \ T c-type-clamp ] 2map ; inline
102 M: A vs* [ * \ T c-type-clamp ] 2map ; inline
104 M: A v*high [ * \ T heap-size neg shift ] 2map ; inline
108 : specialized-array-vocab ( c-type -- vocab )
110 "specialized-arrays.instances." %
111 [ vocabulary>> % "." % ]
116 :: direct-slice-unsafe ( from to seq -- seq' )
117 from seq nth-c-ptr to from - seq direct-like ; inline
121 : direct-slice ( from to seq -- seq' )
122 check-slice direct-slice-unsafe ; inline
124 : direct-head ( seq n -- seq' ) head-to-index direct-slice ; inline
125 : direct-tail ( seq n -- seq' ) index-to-tail direct-slice ; inline
126 : direct-head* ( seq n -- seq' ) from-tail direct-head ; inline
127 : direct-tail* ( seq n -- seq' ) from-tail direct-tail ; inline
129 : define-array-vocab ( type -- vocab )
131 [ specialized-array-vocab ] [ '[ _ define-array ] ] bi
134 ERROR: specialized-array-vocab-not-loaded c-type ;
136 M: c-type-word c-array-constructor
138 dup [ name>> "<" "-array>" surround ] [ specialized-array-vocab ] bi lookup-word
139 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
141 M: pointer c-array-constructor drop void* c-array-constructor ;
143 M: c-type-word c-(array)-constructor
145 dup [ name>> "(" "-array)" surround ] [ specialized-array-vocab ] bi lookup-word
146 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
148 M: pointer c-(array)-constructor drop void* c-(array)-constructor ;
150 M: c-type-word c-direct-array-constructor
152 dup [ name>> "<direct-" "-array>" surround ] [ specialized-array-vocab ] bi lookup-word
153 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
155 M: pointer c-direct-array-constructor drop void* c-direct-array-constructor ;
157 M: c-type-word c-array-type
159 dup [ name>> "-array" append ] [ specialized-array-vocab ] bi lookup-word
160 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
162 M: pointer c-array-type drop void* c-array-type ;
164 M: c-type-word c-array-type?
166 dup [ name>> "-array?" append ] [ specialized-array-vocab ] bi lookup-word
167 [ ] [ specialized-array-vocab-not-loaded ] ?if ; foldable
169 M: pointer c-array-type? drop void* c-array-type? ;
171 SYNTAX: SPECIALIZED-ARRAYS:
172 ";" [ parse-c-type define-array-vocab use-vocab ] each-token ;
174 SYNTAX: SPECIALIZED-ARRAY:
175 scan-c-type define-array-vocab use-vocab ;
177 { "specialized-arrays" "prettyprint" } "specialized-arrays.prettyprint" require-when
179 { "specialized-arrays" "mirrors" } "specialized-arrays.mirrors" require-when
181 uchar define-array-vocab drop