1 ! Copyright (C) 2004, 2010 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: accessors alien alien.accessors arrays byte-arrays
4 classes combinators compiler.units cpu.architecture delegate
5 fry kernel layouts locals macros math math.order quotations
6 sequences system words words.symbol ;
24 TUPLE: abstract-c-type
25 { class class initial: object }
26 { boxed-class class initial: object }
27 { boxer-quot callable }
28 { unboxer-quot callable }
33 { align-first integer } ;
35 TUPLE: c-type < abstract-c-type
38 { rep initial: int-rep } ;
40 : <c-type> ( -- c-type )
43 ERROR: no-c-type name ;
46 GENERIC: c-type ( name -- c-type ) foldable
48 PREDICATE: c-type-word < word
51 TUPLE: pointer { to initial: void read-only } ;
57 : resolve-typedef ( name -- c-type )
58 dup void? [ no-c-type ] when
59 dup c-type-name? [ c-type ] when ;
62 dup "c-type" word-prop resolve-typedef
63 [ ] [ no-c-type ] ?if ;
65 GENERIC: c-type-class ( name -- class )
67 M: abstract-c-type c-type-class class>> ;
69 GENERIC: c-type-boxed-class ( name -- class )
71 M: abstract-c-type c-type-boxed-class boxed-class>> ;
73 GENERIC: c-type-boxer-quot ( name -- quot )
75 M: abstract-c-type c-type-boxer-quot boxer-quot>> ;
77 GENERIC: c-type-unboxer-quot ( name -- quot )
79 M: abstract-c-type c-type-unboxer-quot unboxer-quot>> ;
81 GENERIC: c-type-rep ( name -- rep )
83 M: c-type c-type-rep rep>> ;
85 GENERIC: c-type-getter ( name -- quot )
87 M: c-type c-type-getter getter>> ;
89 GENERIC: c-type-copier ( name -- quot )
91 M: c-type c-type-copier drop [ ] ;
93 GENERIC: c-type-setter ( name -- quot )
95 M: c-type c-type-setter setter>> ;
97 GENERIC: c-type-align ( name -- n ) foldable
99 M: abstract-c-type c-type-align align>> ;
101 GENERIC: c-type-align-first ( name -- n )
103 M: abstract-c-type c-type-align-first align-first>> ;
105 GENERIC: base-type ( c-type -- c-type )
107 M: c-type-name base-type c-type ;
109 M: c-type base-type ;
111 : little-endian? ( -- ? ) 1 int <ref> char deref 1 = ; foldable
113 GENERIC: heap-size ( name -- size )
115 M: abstract-c-type heap-size size>> ;
119 MACRO: alien-value ( c-type -- quot: ( c-ptr offset -- value ) )
120 [ c-type-getter ] [ c-type-boxer-quot ] bi append ;
122 MACRO: alien-copy-value ( c-type -- quot: ( c-ptr offset -- value ) )
123 [ c-type-getter ] [ c-type-copier ] [ c-type-boxer-quot ] tri 3append ;
125 MACRO: set-alien-value ( c-type -- quot: ( value c-ptr offset -- ) )
126 [ c-type-unboxer-quot [ [ ] ] [ '[ _ 2dip ] ] if-empty ]
130 : array-accessor ( n c-ptr c-type -- c-ptr offset c-type )
131 [ swapd heap-size * >fixnum ] keep ; inline
133 : alien-element ( n c-ptr c-type -- value )
134 array-accessor alien-value ; inline
136 : set-alien-element ( value n c-ptr c-type -- )
137 array-accessor set-alien-value ; inline
139 PROTOCOL: c-type-protocol
153 CONSULT: c-type-protocol c-type-name
156 PREDICATE: typedef-word < c-type-word
157 "c-type" word-prop [ c-type-name? ] [ array? ] bi or ;
159 : typedef ( old new -- )
161 [ nip define-symbol ]
162 [ swap "c-type" set-word-prop ]
165 TUPLE: long-long-type < c-type ;
167 : <long-long-type> ( -- c-type )
170 : if-void ( c-type true false -- )
171 pick void? [ drop nip call ] [ nip call ] if ; inline
174 ptrdiff_t intptr_t uintptr_t size_t
177 CONSTANT: primitive-types
189 : >c-bool ( ? -- int ) 1 0 ? ; inline
191 : c-bool> ( int -- ? ) 0 = not ; inline
195 : 8-byte-alignment ( c-type -- c-type )
197 { [ cpu ppc? os macosx? and ] [ 4 >>align 8 >>align-first ] }
198 { [ cpu x86.32? os windows? not and ] [ 4 >>align 4 >>align-first ] }
199 [ 8 >>align 8 >>align-first ]
202 : resolve-pointer-typedef ( type -- base-type )
203 dup "c-type" word-prop dup word?
204 [ nip resolve-pointer-typedef ] [
205 pointer? [ drop void* ] when
208 : primitive-pointer-type? ( type -- ? )
210 resolve-pointer-typedef [ void? ] [ primitive-types member? ] bi or
213 : (pointer-c-type) ( void* type -- void*' )
214 [ clone ] dip c-type-boxer-quot '[ _ [ f ] if* ] >>boxer-quot ;
219 [ \ void* c-type ] dip
220 to>> dup primitive-pointer-type? [ drop ] [ (pointer-c-type) ] if ;
226 [ alien-cell ] >>getter
227 [ set-alien-cell ] >>setter
228 bootstrap-cell >>size
229 bootstrap-cell >>align
230 bootstrap-cell >>align-first
231 [ >c-ptr ] >>unboxer-quot
232 "allot_alien" >>boxer
233 "alien_offset" >>unboxer
239 [ alien-signed-2 ] >>getter
240 [ set-alien-signed-2 ] >>setter
244 "from_signed_2" >>boxer
245 "to_signed_2" >>unboxer
246 [ >fixnum ] >>unboxer-quot
252 [ alien-unsigned-2 ] >>getter
253 [ set-alien-unsigned-2 ] >>setter
257 "from_unsigned_2" >>boxer
258 "to_unsigned_2" >>unboxer
259 [ >fixnum ] >>unboxer-quot
265 [ alien-signed-1 ] >>getter
266 [ set-alien-signed-1 ] >>setter
270 "from_signed_1" >>boxer
271 "to_signed_1" >>unboxer
272 [ >fixnum ] >>unboxer-quot
278 [ alien-unsigned-1 ] >>getter
279 [ set-alien-unsigned-1 ] >>setter
283 "from_unsigned_1" >>boxer
284 "to_unsigned_1" >>unboxer
285 [ >fixnum ] >>unboxer-quot
290 math:float >>boxed-class
291 [ alien-float ] >>getter
292 [ set-alien-float ] >>setter
299 [ >float ] >>unboxer-quot
304 math:float >>boxed-class
305 [ alien-double ] >>getter
306 [ set-alien-double ] >>setter
309 "from_double" >>boxer
310 "to_double" >>unboxer
312 [ >float ] >>unboxer-quot
319 [ alien-signed-4 ] >>getter
320 [ set-alien-signed-4 ] >>setter
324 "from_signed_4" >>boxer
325 "to_signed_4" >>unboxer
326 [ >fixnum ] >>unboxer-quot
332 [ alien-unsigned-4 ] >>getter
333 [ set-alien-unsigned-4 ] >>setter
337 "from_unsigned_4" >>boxer
338 "to_unsigned_4" >>unboxer
339 [ >fixnum ] >>unboxer-quot
344 integer >>boxed-class
345 [ alien-signed-cell ] >>getter
346 [ set-alien-signed-cell ] >>setter
350 "from_signed_cell" >>boxer
351 "to_fixnum" >>unboxer
352 [ >integer ] >>unboxer-quot
357 integer >>boxed-class
358 [ alien-unsigned-cell ] >>getter
359 [ set-alien-unsigned-cell ] >>setter
363 "from_unsigned_cell" >>boxer
365 [ >integer ] >>unboxer-quot
369 \ int c-type \ long typedef
370 \ uint c-type \ ulong typedef
372 \ longlong c-type \ long typedef
373 \ ulonglong c-type \ ulong typedef
376 \ longlong c-type \ ptrdiff_t typedef
377 \ longlong c-type \ intptr_t typedef
379 \ ulonglong c-type \ uintptr_t typedef
380 \ ulonglong c-type \ size_t typedef
384 integer >>boxed-class
385 [ alien-signed-cell ] >>getter
386 [ set-alien-signed-cell ] >>setter
390 "from_signed_cell" >>boxer
391 "to_fixnum" >>unboxer
392 [ >integer ] >>unboxer-quot
397 integer >>boxed-class
398 [ alien-unsigned-cell ] >>getter
399 [ set-alien-unsigned-cell ] >>setter
403 "from_unsigned_cell" >>boxer
405 [ >integer ] >>unboxer-quot
410 integer >>boxed-class
411 [ alien-signed-8 ] >>getter
412 [ set-alien-signed-8 ] >>setter
415 "from_signed_8" >>boxer
416 "to_signed_8" >>unboxer
417 [ >integer ] >>unboxer-quot
422 integer >>boxed-class
423 [ alien-unsigned-8 ] >>getter
424 [ set-alien-unsigned-8 ] >>setter
427 "from_unsigned_8" >>boxer
428 "to_unsigned_8" >>unboxer
429 [ >integer ] >>unboxer-quot
432 \ int c-type \ long typedef
433 \ uint c-type \ ulong typedef
435 \ int c-type \ ptrdiff_t typedef
436 \ int c-type \ intptr_t typedef
438 \ uint c-type \ uintptr_t typedef
439 \ uint c-type \ size_t typedef
442 cpu ppc? \ uint \ uchar ? c-type clone
443 [ >c-bool ] >>unboxer-quot
444 [ c-bool> ] >>boxer-quot
448 ] with-compilation-unit
450 M: char-16-rep rep-component-type drop char ;
451 M: uchar-16-rep rep-component-type drop uchar ;
452 M: short-8-rep rep-component-type drop short ;
453 M: ushort-8-rep rep-component-type drop ushort ;
454 M: int-4-rep rep-component-type drop int ;
455 M: uint-4-rep rep-component-type drop uint ;
456 M: longlong-2-rep rep-component-type drop longlong ;
457 M: ulonglong-2-rep rep-component-type drop ulonglong ;
458 M: float-4-rep rep-component-type drop float ;
459 M: double-2-rep rep-component-type drop double ;
461 : (unsigned-interval) ( bytes -- from to ) [ 0 ] dip 8 * 2^ 1 - ; foldable
462 : unsigned-interval ( c-type -- from to ) heap-size (unsigned-interval) ; foldable
463 : (signed-interval) ( bytes -- from to ) 8 * 1 - 2^ [ neg ] [ 1 - ] bi ; foldable
464 : signed-interval ( c-type -- from to ) heap-size (signed-interval) ; foldable
466 : c-type-interval ( c-type -- from to )
468 { [ dup { float double } member-eq? ] [ drop -1/0. 1/0. ] }
469 { [ dup { char short int long longlong } member-eq? ] [ signed-interval ] }
470 { [ dup { uchar ushort uint ulong ulonglong } member-eq? ] [ unsigned-interval ] }
473 : c-type-clamp ( value c-type -- value' )
474 dup { float double } member-eq?
475 [ drop ] [ c-type-interval clamp ] if ; inline
477 :: <ref> ( value c-type -- c-ptr )
478 c-type heap-size <byte-array> :> c-ptr
479 value c-ptr 0 c-type set-alien-value
482 : deref ( c-ptr c-type -- value )
483 [ 0 ] dip alien-value ; inline