chiark / gitweb /
Added new type UNBOXED-VECTOR
[clg] / gffi / vectors.lisp
CommitLineData
e5426fae 1;; Common Lisp bindings for GTK+ 2.x
2;; Copyright 1999-2006 Espen S. Johnsen <espen@users.sf.net>
3;;
4;; Permission is hereby granted, free of charge, to any person obtaining
5;; a copy of this software and associated documentation files (the
6;; "Software"), to deal in the Software without restriction, including
7;; without limitation the rights to use, copy, modify, merge, publish,
8;; distribute, sublicense, and/or sell copies of the Software, and to
9;; permit persons to whom the Software is furnished to do so, subject to
10;; the following conditions:
11;;
12;; The above copyright notice and this permission notice shall be
13;; included in all copies or substantial portions of the Software.
14;;
15;; THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16;; EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
17;; MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
18;; IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
19;; CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
20;; TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
21;; SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
22
9cbf3858 23;; $Id: vectors.lisp,v 1.7 2008-04-29 22:16:28 espen Exp $
e5426fae 24
25
26(in-package "GFFI")
27
2c708568 28
29;;; Accessor functions for raw memory access
30
31(define-memory-accessor int-16)
32(define-memory-accessor int-32)
33(define-memory-accessor int-64)
34(define-memory-accessor uint-16)
35(define-memory-accessor uint-32)
36(define-memory-accessor uint-64)
37(define-memory-accessor single-float)
38(define-memory-accessor double-float)
39
40
e5426fae 41;;; Vector
42
43(defun make-c-vector (type length &key content location temp)
44 (let* ((element-size (size-of type))
45 (location (or location (allocate-memory (* element-size length))))
46 (writer (writer-function type :temp temp)))
47 (etypecase content
48 (vector
49 (loop
50 for element across content
51 for i below length
52 for offset by element-size
53 do (funcall writer element location offset)))
54 (list
55 (loop
56 for element in content
57 for i below length
58 for offset by element-size
59 do (funcall writer element location offset))))
60 location))
61
62
63(defun map-c-vector (seqtype function location element-type length
64 &optional (ref :read))
65 (let ((reader (reader-function element-type :ref ref))
66 (element-size (size-of element-type)))
67 (case seqtype
68 ((nil)
69 (loop
70 for i below length
71 for offset by element-size
72 do (funcall function (funcall reader location offset))))
73 (list
74 (loop
75 for i below length
76 for offset by element-size
77 collect (funcall function (funcall reader location offset))))
78 (t
79 (loop
80 with sequence = (make-sequence seqtype length)
81 for i below length
82 for offset by element-size
83 do (setf
84 (elt sequence i)
85 (funcall function (funcall reader location offset)))
86 finally (return sequence))))))
87
88
89(defun unset-c-vector (location element-type length &optional temp-p)
90 (loop
91 with destroy = (destroy-function element-type :temp temp-p)
92 with element-size = (size-of element-type)
93 for i below length
94 for offset by element-size
95 do (funcall destroy location offset)))
96
97
98(defun destroy-c-vector (location element-type length &optional temp-p)
99 (unset-c-vector location element-type length temp-p)
100 (deallocate-memory location))
101
102
103(defmacro with-c-vector (var type content &body body)
104 (let ((length (make-symbol "LENGTH")))
105 `(let ((,length (length ,content)))
106 (with-memory (,var (* ,(size-of type) ,length))
107 (make-c-vector ',type ,length :content ,content :location ,var :temp t)
108 (unwind-protect
109 (progn ,@body)
110 (unset-c-vector ,var ',type ,length t))))))
111
112
113(define-type-method alien-type ((type vector))
114 (declare (ignore type))
115 (alien-type 'pointer))
116
f9a67e67 117(define-type-method argument-type ((type vector))
118 (declare (ignore type))
119 'sequence)
120
121(define-type-method return-type ((type vector))
2c708568 122 (destructuring-bind (element-type &optional (length '*))
123 (rest (type-expand-to 'vector type))
124 (if (constantp length)
125 `(vector ,(return-type element-type) ,length)
126 `(vector ,(return-type element-type) *))))
127
e5426fae 128(define-type-method size-of ((type vector) &key inlined)
129 (if inlined
130 (destructuring-bind (element-type &optional (length '*))
131 (rest (type-expand-to 'vector type))
132 (if (eq length '*)
133 (error "Can't inline vector with variable size: ~A" type)
134 (* (size-of element-type) length)))
135 (size-of 'pointer)))
136
90e8bbf6 137(define-type-method type-alignment ((type vector) &key inlined)
138 (if inlined
139 (destructuring-bind (element-type &optional (length '*))
140 (rest (type-expand-to 'vector type))
141 (if (eq length '*)
142 (error "Can't inline vector with variable size: ~A" type)
143 (* (type-alignment element-type) length)))
144 (type-alignment 'pointer)))
145
e5426fae 146(define-type-method alien-arg-wrapper ((type vector) var vector style form &optional copy-in-p)
147 (destructuring-bind (element-type &optional (length '*))
148 (rest (type-expand-to 'vector type))
149 (when (and (eq length '*) (out-arg-p style))
150 (error "Can't use vector with variable size as return type"))
151 (cond
152 ((and (in-arg-p style) copy-in-p)
153 `(with-pointer (,var `(make-c-vector ',element-type
154 ,(if (eq length '*) `(length ,vector) length)
155 :content ,vector))
156 ,form))
157 ((and (in-arg-p style) (not (out-arg-p style)))
158 `(with-memory (,var ,(if (eq length '*)
159 `(* ,(size-of element-type)
160 (length ,vector))
161 `(* ,(size-of element-type) ,length)))
162 (make-c-vector ',element-type
163 ,(if (eq length '*) `(length ,vector) length)
164 :content ,vector :location ,var :temp t)
165 (unwind-protect
166 ,form
167 (unset-c-vector ,var ',element-type
168 ,(if (eq length '*) `(length ,vector) length) t))))
169 ((and (in-arg-p style) (out-arg-p style))
170 (let ((c-vector (make-symbol "C-VECTOR")))
171 `(with-memory (,c-vector (* ,(size-of element-type) length))
172 (make-c-vector ',element-type ,length
173 :content ,vector :location ,c-vector :temp t)
174 (with-pointer (,var ,c-vector)
175 (unwind-protect
176 ,form
177 (unset-c-vector ,c-vector ',element-type ,length t))))))
178 ((and (out-arg-p style) (not (in-arg-p style)))
179 `(with-pointer (,var)
180 ,form)))))
181
182;; This will enable us specify vectors with variable length in C callbacks
183(define-type-method callback-wrapper ((type vector) var vector form)
184 (funcall (find-applicable-type-method 'callback-wrapper t) type var vector form))
185
186(define-type-method to-alien-form ((type vector) vector &optional copy-p)
187 (declare (ignore copy-p))
188 (destructuring-bind (element-type &optional (length '*))
189 (rest (type-expand-to 'vector type))
190 `(make-c-vector ',element-type
191 ,(if (eq length '*) `(length ,vector) length) :content ,vector)))
192
193
194(define-type-method from-alien-form ((type vector) form &key (ref :free))
195 (destructuring-bind (element-type &optional (length '*))
196 (rest (type-expand-to 'vector type))
197 (if (eq length '*)
198 (error "Can't use vector of variable size as return type")
199 `(let ((c-vector ,form))
200 (prog1
201 (map-c-vector 'vector #'identity c-vector ',element-type ,length
202 ,(ecase ref (:free :get) ((:static :temp) :peek) (:copy :read)))
203 ,(when (eq ref :free)
204 `(deallocate-memory c-vector)))))))
205
206
207(define-type-method writer-function ((type vector) &key temp inlined)
208 (destructuring-bind (element-type &optional (length '*))
209 (rest (type-expand-to 'vector type))
210 (if inlined
211 (if (eq length '*)
212 (error "Can't inline vector with variable size: ~A" type)
213 #'(lambda (vector location &optional (offset 0))
214 (make-c-vector element-type length
215 :location (pointer+ location offset)
216 :content vector :temp temp)))
217 #'(lambda (vector location &optional (offset 0))
218 (setf
219 (ref-pointer location offset)
220 (make-c-vector element-type length :content vector :temp temp))))))
221
222(define-type-method reader-function ((type vector) &key (ref :read) inlined)
223 (destructuring-bind (element-type &optional (length '*))
224 (rest (type-expand-to 'vector type))
225 (cond
226 ((eq length '*)
227 (error "Can't create reader function for vector with variable size"))
228 (inlined
229 #'(lambda (location &optional (offset 0))
230 (map-c-vector 'vector #'identity (pointer+ location offset)
231 element-type length ref)))
232 (t
233 (ecase ref
234 ((:read :peek)
235 #'(lambda (location &optional (offset 0))
236 (unless (null-pointer-p (ref-pointer location offset))
237 (map-c-vector 'vector #'identity (ref-pointer location offset)
238 element-type length ref))))
239 (:get
240 #'(lambda (location &optional (offset 0))
241 (unless (null-pointer-p (ref-pointer location offset))
242 (prog1
243 (map-c-vector 'vector #'identity
244 (ref-pointer location offset) element-type length :get)
245 (deallocate-memory (ref-pointer location offset))
246 (setf (ref-pointer location offset) (make-pointer 0)))))))))))
247
248(define-type-method destroy-function ((type vector) &key temp inlined)
249 (destructuring-bind (element-type &optional (length '*))
250 (rest (type-expand-to 'vector type))
251 (cond
252 ((eq length '*)
253 (error "Can't create destroy function for vector with variable size"))
254 (inlined
255 #'(lambda (location &optional (offset 0))
256 (unset-c-vector (pointer+ location offset)
257 element-type length temp)))
258 (t
259 #'(lambda (location &optional (offset 0))
260 (unless (null-pointer-p (ref-pointer location offset))
261 (destroy-c-vector (ref-pointer location offset)
262 element-type length temp)
263 (setf (ref-pointer location offset) (make-pointer 0))))))))
264
265(define-type-method copy-function ((type vector) &key inlined)
266 (destructuring-bind (element-type &optional (length '*))
267 (rest (type-expand-to 'vector type))
268 (cond
269 ((eq length '*) (error "Can't copy vector with variable size: ~A" type))
270 (inlined
271 (let ((copy-element (copy-function element-type))
272 (element-size (size-of element-type)))
273 #'(lambda (from to &optional (offset 0))
274 (loop
275 repeat length
276 for element from offset by element-size
277 do (funcall copy-element from to element)))))
278 (t
279 (let ((size (* length (size-of element-type)))
280 (copy-content (copy-function type :inlined t)))
281 #'(lambda (from to &optional (offset 0))
282 (unless (null-pointer-p (ref-pointer from offset))
283 (let ((vector (allocate-memory size)))
284 (setf (ref-pointer to offset) vector)
285 (funcall copy-content (ref-pointer from offset) vector)))))))))
286
9cbf3858 287;;;; Unboxed vector
288
289(deftype unboxed-vector (element-type &optional (length '*))
290 `(simple-array ,element-type (,length)))
291
292(define-type-method argument-type ((type unboxed-vector))
293 type)
294
295(define-type-method return-type ((type unboxed-vector))
296 (destructuring-bind (element-type &optional (length '*))
297 (rest (type-expand-to 'unboxed-vector type))
298 (if (constantp length)
299 `(unboxed-vector ,(return-type element-type) ,length)
300 `(unboxed-vector ,(return-type element-type) *))))
301
302(defun check-unboxed-vector (type)
303 #+(or sbcl cmu)
304 (unless (subtypep type 'simple-unboxed-array)
305 (error "~A is not a subtype of ~A" type 'simple-unboxed-array)))
306
307#+(or sbcl cmu)
308(progn
309 (define-type-method alien-arg-wrapper ((type unboxed-vector) var vector style form &optional copy-in-p)
310 (destructuring-bind (element-type &optional (length '*))
311 (rest (type-expand-to 'unboxed-vector type))
312 (check-unboxed-vector type)
313 (when (and (eq length '*) (out-arg-p style))
314 (error "Can't use vector with variable size as return type"))
315 (cond
316 ((and (in-arg-p style) copy-in-p)
317 `(with-pointer (,var (with-pinned-objects (,vector)
318 (copy-memory (vector-sap ,vector)
319 (* (length ,vector) ,(size-of element-type)))))
320 ,form))
321 ((in-arg-p style)
322 `(with-pinned-objects (,vector)
323 (let ((,var (vector-sap ,vector)))
324 ,form)))
325 ((out-arg-p style)
326 `(with-pointer (,var)
327 ,form)))))
328
329 (define-type-method to-alien-form ((type unboxed-vector) vector &optional copy-p)
330 (declare (ignore copy-p))
331 (destructuring-bind (element-type &optional (length '*))
332 (rest (type-expand-to 'unboxed-vector type))
333 (check-unboxed-vector type)
334 `(with-pinned-objects (,vector)
335 (copy-memory
336 (vector-sap ,vector)
337 (* ,(if (eq length '*) `(length ,vector) length)
338 ,(size-of element-type))))))
339
340
341 (define-type-method from-alien-form ((type unboxed-vector) form &key (ref :free))
342 (destructuring-bind (element-type &optional (length '*))
343 (rest (type-expand-to 'unboxed-vector type))
344 (check-unboxed-vector type)
345 (when (eq length '*)
346 (error "Can't use vector of variable size as return type"))
347 `(let ((c-vector ,form)
348 (vector (make-array ,length :element-type ',element-type)))
349 (with-pinned-objects (vector)
350 (copy-memory c-vector (* ,length ,(size-of element-type)) (vector-sap vector))
351 ,(when (eq ref :free)
352 `(deallocate-memory c-vector))
353 vector))))
354
355 (define-type-method writer-function ((type unboxed-vector) &key temp inlined)
356 (declare (ignore temp))
357 (destructuring-bind (element-type &optional (length '*))
358 (rest (type-expand-to 'unboxed-vector type))
359 (check-unboxed-vector type)
360 (if inlined
361 (if (eq length '*)
362 (error "Can't inline vector with variable size: ~A" type)
363 #'(lambda (vector location &optional (offset 0))
364 (with-pinned-objects (vector)
365 (copy-memory
366 (vector-sap vector)
367 (* length (size-of element-type))
368 (pointer+ location offset)))))
369 #'(lambda (vector location &optional (offset 0))
370 (setf
371 (ref-pointer location offset)
372 (with-pinned-objects (vector)
373 (copy-memory (vector-sap vector)
374 (* (length vector) (size-of element-type)))))))))
375
376 (define-type-method reader-function ((type unboxed-vector) &key (ref :read) inlined)
377 (destructuring-bind (element-type &optional (length '*))
378 (rest (type-expand-to 'unboxed-vector type))
379 (check-unboxed-vector type)
380 (cond
381 ((eq length '*)
382 (error "Can't create reader function for vector with variable size"))
383 (inlined
384 #'(lambda (location &optional (offset 0))
385 (let ((vector (make-array length :element-type element-type)))
386 (with-pinned-objects (vector)
387 (copy-memory
388 (pointer+ location offset)
389 (* length (size-of element-type))
390 (vector-sap vector))
391 vector))))
392 (t
393 #'(lambda (location &optional (offset 0))
394 (let ((vector (make-array length :element-type element-type)))
395 (unless (null-pointer-p (ref-pointer location offset))
396 (with-pinned-objects (vector)
397 (copy-memory
398 (ref-pointer location offset)
399 (* (length vector) (size-of element-type))
400 (vector-sap vector)))
401 (when (eq ref :get)
402 (deallocate-memory (ref-pointer location offset))
403 (setf (ref-pointer location offset) (make-pointer 0)))
404 vector))))))))
405
406(define-type-method destroy-function ((type unboxed-vector) &key temp inlined)
407 (declare (ignore temp))
408 (destructuring-bind (element-type &optional (length '*))
409 (rest (type-expand-to 'unboxed-vector type))
410 (check-unboxed-vector type)
411 (cond
412 #+sbcl
413 ((eq length '*)
414 (error "Can't create destroy function for vector with variable size"))
415 (inlined
416 #'(lambda (location &optional (offset 0))
417 (clear-memory location (* length (size-of element-type)) offset)))
418 (t
419 #'(lambda (location &optional (offset 0))
420 (unless (null-pointer-p (ref-pointer location offset))
421 (deallocate-memory (ref-pointer location offset))
422 (setf (ref-pointer location offset) (make-pointer 0))))))))
423
e5426fae 424
425;;;; Null terminated vector
426
427(defun make-0-vector (type &key content location temp)
428 (let* ((element-size (size-of type))
429 (length (length content))
430 (location (or location (allocate-memory (* element-size (1+ length))))))
431 (make-c-vector type length :content content :location location :temp temp)))
432
433
434(defun map-0-vector (seqtype function location element-type &optional (ref :read))
435 (let ((reader (reader-function element-type :ref ref))
436 (element-size (size-of element-type)))
437 (case seqtype
438 ((nil)
439 (loop
440 for offset by element-size
441 until (memory-clear-p (pointer+ location offset) element-size)
442 do (funcall function (funcall reader location offset))))
443 (list
444 (loop
445 for offset by element-size
446 until (memory-clear-p (pointer+ location offset) element-size)
447 collect (funcall function (funcall reader location offset))))
448 (t
449 (coerce
450 (loop
451 for offset by element-size
452 until (memory-clear-p (pointer+ location offset) element-size)
453 collect (funcall function (funcall reader location offset)))
454 seqtype)))))
455
456
457(defun unset-0-vector (location element-type &optional temp-p)
458 (loop
925b76cc 459 with destroy = (destroy-function element-type :temp temp-p)
e5426fae 460 with element-size = (size-of element-type)
461 for offset by element-size
462 until (memory-clear-p (pointer+ location offset) element-size)
463 do (funcall destroy location offset)))
464
465(defun destroy-0-vector (location element-type &optional temp-p)
466 (unset-0-vector location element-type temp-p)
467 (deallocate-memory location))
468
469
470(deftype vector0 (element-type) `(vector ,element-type))
471(deftype null-terminated-vector (element-type) `(vector0 ,element-type))
472
473(define-type-method alien-type ((type vector0))
474 (declare (ignore type))
475 (alien-type 'pointer))
476
477(define-type-method size-of ((type vector0) &key inlined)
478 (assert-not-inlined type inlined)
479 (size-of 'pointer))
480
90e8bbf6 481(define-type-method type-alignment ((type vector0) &key inlined)
482 (assert-not-inlined type inlined)
483 (type-alignment 'pointer))
484
e5426fae 485(define-type-method alien-arg-wrapper ((type vector0) var vector style form &optional copy-in-p)
486 (destructuring-bind (element-type) (rest (type-expand-to 'vector0 type))
487 (cond
488 ((and (in-arg-p style) copy-in-p)
489 `(with-pointer (,var (make-0-vector ',element-type :content ,vector))
490 ,form))
491 ((and (in-arg-p style) (not (out-arg-p style)))
492 `(with-memory (,var (* ,(size-of element-type) (1+ (length ,vector))))
493 (make-0-vector ',element-type :content ,vector :location ,var :temp t)
494 (unwind-protect
495 ,form
496 (unset-0-vector ,var ',element-type t))))
497 ((and (in-arg-p style) (out-arg-p style))
498 (let ((c-vector (make-symbol "C-VECTOR")))
499 `(with-memory (,c-vector (* ,(size-of element-type) (1+ (length ,vector))))
500 (make-0-vector ',element-type :content ,vector :location ,c-vector :temp t)
501 (with-pointer (,var ,c-vector)
502 (unwind-protect
503 ,form
504 (unset-0-vector ,c-vector ',element-type t))))))
505 ((and (out-arg-p style) (not (in-arg-p style)))
506 `(with-pointer (,var)
507 ,form)))))
508
509
510(define-type-method to-alien-form ((type vector0) vector &optional copy-p)
511 (declare (ignore copy-p))
512 (destructuring-bind (element-type) (rest (type-expand-to 'vector0 type))
513 `(make-0-vector ',element-type :content ,vector)))
514
515(define-type-method from-alien-form ((type vector0) form &key (ref :free))
516 (destructuring-bind (element-type) (rest (type-expand-to 'vector0 type))
517 `(let ((c-vector ,form))
518 (prog1
519 (map-0-vector 'vector #'identity c-vector ',element-type
520 ,(ecase ref (:free :get) ((:static :temp) :peek) (:copy :read)))
521 ,(when (eq ref :free)
522 `(deallocate-memory c-vector))))))
523
524
525(define-type-method writer-function ((type vector0) &key temp inlined)
526 (assert-not-inlined type inlined)
527 (destructuring-bind (element-type) (rest (type-expand-to 'vector0 type))
528 #'(lambda (vector location &optional (offset 0))
529 (setf
530 (ref-pointer location offset)
531 (make-0-vector element-type :content vector :temp temp)))))
532
533(define-type-method reader-function ((type vector0) &key (ref :read) inlined)
534 (assert-not-inlined type inlined)
535 (destructuring-bind (element-type) (rest (type-expand-to 'vector0 type))
536 (ecase ref
537 ((:read :peek)
538 #'(lambda (location &optional (offset 0))
539 (unless (null-pointer-p (ref-pointer location offset))
540 (map-0-vector 'vector #'identity (ref-pointer location offset)
541 element-type ref))))
542 (:get
543 #'(lambda (location &optional (offset 0))
544 (unless (null-pointer-p (ref-pointer location offset))
545 (prog1
546 (map-0-vector 'vector #'identity (ref-pointer location offset)
547 element-type :get)
548 (deallocate-memory (ref-pointer location offset))
549 (setf (ref-pointer location offset) (make-pointer 0)))))))))
550
551
552(define-type-method destroy-function ((type vector0) &key temp inlined)
553 (assert-not-inlined type inlined)
554 (destructuring-bind (element-type) (rest (type-expand-to 'vector0 type))
555 #'(lambda (location &optional (offset 0))
556 (unless (null-pointer-p (ref-pointer location offset))
557 (destroy-0-vector
558 (ref-pointer location offset) element-type temp)
559 (setf (ref-pointer location offset) (make-pointer 0))))))
560
561(define-type-method copy-function ((type vector0) &key inlined)
562 (assert-not-inlined type inlined)
563 (destructuring-bind (element-type) (rest (type-expand-to 'vector0 type))
564 (let ((copy-element (copy-function element-type))
565 (element-size (size-of element-type)))
566 #'(lambda (from to &optional (offset 0))
567 (unless (null-pointer-p (ref-pointer from offset))
568 (let* ((from-vector (ref-pointer from offset))
569 (length
570 (loop
571 for length from 0
572 for element by element-size
573 until (memory-clear-p from-vector element-size element)
574 finally (return length)))
110bd96c 575 (to-vector
576 (setf (ref-pointer to offset)
577 (allocate-memory (* (1+ length) element-size)))))
e5426fae 578 (loop
579 repeat length
580 for element by element-size
581 do (funcall copy-element from-vector to-vector element))))))))
582
583(define-type-method unbound-value ((type vector0))
584 (declare (ignore type))
585 nil)
586
587
588
589;;;; Counted vector
590
591(defun make-counted-vector (type &key content location (counter-type 'unsigned-int) temp)
592 (let* ((element-size (size-of type))
593 (length (length content))
594 (location (or
595 location
596 (allocate-memory
597 (+ (size-of counter-type) (* element-size length))))))
598 (funcall (writer-function counter-type :temp temp) length location)
599 (make-c-vector type length :content content :location (pointer+ location (size-of counter-type)))
600 location))
601
602(defun map-counted-vector (seqtype function location element-type &optional (counter-type 'unsigned-int) (ref :read))
603 (let ((length (funcall (reader-function counter-type) location :ref ref)))
604 (map-c-vector
605 seqtype function (pointer+ location (size-of counter-type))
606 element-type length)))
607
608(defun unset-counted-vector (location element-type &optional (counter-type 'unsigned-int) temp-p)
609 (let ((length (funcall (reader-function counter-type) location)))
610 (unset-c-vector
611 (pointer+ location (size-of counter-type)) element-type length temp-p)))
612
613(defun destroy-counted-vector (location element-type &optional (counter-type 'unsigned-int) temp-p)
614 (unset-counted-vector location element-type counter-type temp-p)
615 (deallocate-memory location))
616
617
618(deftype counted-vector (element-type &optional counter-type)
619 (declare (ignore counter-type))
620 `(vector ,element-type))
621
622(define-type-method alien-type ((type counted-vector))
623 (declare (ignore type))
624 (alien-type 'pointer))
625
626(define-type-method size-of ((type counted-vector) &key inlined)
627 (assert-not-inlined type inlined)
628 (size-of 'pointer))
629
90e8bbf6 630(define-type-method type-alignment ((type counted-vector) &key inlined)
631 (assert-not-inlined type inlined)
632 (type-alignment 'pointer))
633
e5426fae 634(define-type-method alien-arg-wrapper ((type counted-vector) var vector style form &optional copy-in-p)
635 (destructuring-bind (element-type &optional (counter-type 'unsigned-int))
636 (rest (type-expand-to 'counted-vector type))
637 (cond
638 ((and (in-arg-p style) copy-in-p)
639 `(with-pointer (,var (make-counted-vector ',element-type
640 :content ,vector :counter-type ',counter-type))
641 ,form))
642 ((and (in-arg-p style) (not (out-arg-p style)))
643 `(with-memory (,var (+ (* ,(size-of element-type) (length ,vector)) ,(size-of counter-type)))
644 (make-counted-vector ',element-type :content ,vector
645 :location ,var :counter-type ',counter-type :temp t)
646 (unwind-protect
647 ,form
648 (unset-counted-vector ,var ',element-type ',counter-type t))))
649 ((and (in-arg-p style) (out-arg-p style))
650 (let ((c-vector (make-symbol "C-VECTOR")))
651 `(with-memory (,c-vector (+ (* ,(size-of element-type) (length ,vector)) ,(size-of counter-type)))
652 (make-counted-vector ',element-type :content ,vector ,c-vector
653 :counter-type ',counter-type :temp t)
654 (with-pointer (,var ,c-vector)
655 (unwind-protect
656 ,form
657 (unset-counted-vector ,c-vector ',element-type ',counter-type t))))))
658 ((and (out-arg-p style) (not (in-arg-p style)))
659 `(with-pointer (,var)
660 ,form)))))
661
662
663(define-type-method to-alien-form ((type counted-vector) vector &optional copy-p)
664 (declare (ignore copy-p))
665 (destructuring-bind (element-type &optional (counter-type 'unsigned-int))
666 (rest (type-expand-to 'counted-vector type))
667 `(make-counted-vector ',element-type
668 :content ,vector :counter-type ',counter-type)))
669
670(define-type-method from-alien-form ((type counted-vector) form &key (ref :free))
671 (destructuring-bind (element-type &optional (counter-type 'unsigned-int))
672 (rest (type-expand-to 'counted-vector type))
673 `(let ((c-vector ,form))
674 (prog1
675 (map-counted-vector 'vector #'identity c-vector ',element-type ',counter-type
676 ,(ecase ref (:free :get) ((:static :temp) :peek) (:copy :read)))
677 ,(when (eq ref :free)
678 `(deallocate c-vector))))))
679
680(define-type-method writer-function ((type counted-vector) &key temp inlined)
681 (assert-not-inlined type inlined)
682 (destructuring-bind (element-type &optional (counter-type 'unsigned-int))
683 (rest (type-expand-to 'counted-vector type))
684 #'(lambda (vector location &optional (offset 0))
685 (setf
686 (ref-pointer location offset)
687 (make-counted-vector element-type :content vector
688 :counter-type counter-type :temp temp)))))
689
690(define-type-method reader-function ((type counted-vector) &key (ref :read) inlined)
691 (assert-not-inlined type inlined)
692 (destructuring-bind (element-type &optional (counter-type 'unsigned-int))
693 (rest (type-expand-to 'counted-vector type))
694 (ecase ref
695 ((:read :peek)
696 #'(lambda (location &optional (offset 0))
697 (unless (null-pointer-p (ref-pointer location offset))
698 (map-counted-vector 'vector #'identity
699 (ref-pointer location offset) element-type counter-type ref))))
700 (:get
701 #'(lambda (location &optional (offset 0))
702 (unless (null-pointer-p (ref-pointer location offset))
703 (prog1
704 (map-counted-vector 'vector #'identity
705 (ref-pointer location offset) element-type counter-type :get)
706 (deallocate-memory (ref-pointer location offset))
707 (setf (ref-pointer location offset) (make-pointer 0)))))))))
708
709(define-type-method destroy-function ((type counted-vector) &key temp inlined)
710 (assert-not-inlined type inlined)
711 (destructuring-bind (element-type &optional (counter-type 'unsigned-int))
712 (rest (type-expand-to 'counted-vector type))
713 #'(lambda (location &optional (offset 0))
714 (unless (null-pointer-p (ref-pointer location offset))
715 (destroy-counted-vector (ref-pointer location offset)
716 element-type counter-type temp)
717 (setf (ref-pointer location offset) (make-pointer 0))))))
718
719(define-type-method copy-function ((type counted-vector) &key inlined)
720 (assert-not-inlined type inlined)
721 (destructuring-bind (element-type &optional (counter-type 'unsigned-int))
722 (rest (type-expand-to 'counted-vector type))
723 (let ((vector-length (reader-function counter-type))
724 (counter-size (size-of counter-type))
725 (copy-element (copy-function element-type))
726 (element-size (size-of element-type)))
727 #'(lambda (from to &optional (offset 0))
728 (unless (null-pointer-p (ref-pointer from offset))
729 (let* ((from-vector (ref-pointer from offset))
730 (length (funcall vector-length from-vector))
731 (to-vector (setf
732 (ref-pointer to offset)
733 (allocate-memory (+ counter-size (* length element-size))))))
734 (copy-memory from-vector counter-size to-vector)
735 (loop
736 repeat length
737 for element from counter-size by element-size
738 do (funcall copy-element from-vector to-vector element))))))))
2c708568 739
740
741;;;; Accessor functions for raw memory access
742
743(defun vector-reader-function (type &key (start 0) end)
744 "Returns a function for reading values from raw C vectors"
745 (let ((element-size (size-of type))
746 (reader (reader-function type)))
747 #'(lambda (vector index)
748 (assert (and (>= index start) (or (not end) (< index end))))
749 (funcall reader vector (* index element-size)))))
750
751(defun vector-writer-function (type &key (start 0) end)
752 "Returns a function for writing values to raw C vectors"
753 (let ((element-size (size-of type))
754 (writer (writer-function type)))
755 #'(lambda (value vector index)
756 (assert (and (>= index start) (or (not end) (< index end))))
757 (funcall writer value vector (* index element-size)))))
758
759
760(defmacro define-vector-accessor (type)
761 (let ((name (intern (format nil "VECTOR-REF-~A" type)))
762 (ref (intern (format nil "REF-~A" type))))
763 `(progn
764 (declaim
765 (ftype (function (pointer fixnum) ,type) ,name)
766 (inline ,name))
767 (defun ,name (vector index)
768 (,ref vector (* ,(size-of type) index)))
769 (declaim
770 (ftype (function (,type pointer fixnum) ,type) (setf ,name))
771 (inline (setf ,name)))
772 (defun (setf ,name) (value vector index)
773 (setf (,ref vector (* ,(size-of type) index)) value)))))
774
775(define-vector-accessor int-8)
776(define-vector-accessor uint-8)
777(define-vector-accessor int-16)
778(define-vector-accessor uint-16)
779(define-vector-accessor int-32)
780(define-vector-accessor uint-32)
781(define-vector-accessor int-64)
782(define-vector-accessor uint-64)
783(define-vector-accessor double-float)
784(define-vector-accessor single-float)
785