1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#! /usr/bin/env python
# Shamelessly stolen from pyopengl-ctypes on 2015-06-21.
#
# Original file here:
# http://bazaar.launchpad.net/~mcfletch/pyopengl/trunk/view/head:/OpenGL/arrays/_buffers.py
"""Python 3.x buffer-handling (currently just for bytes/bytearray types)
"""
import ctypes
import sys
if sys.version_info[:2] < (2, 6):
raise ImportError('Buffer interface only usable on Python 2.6+')
PyBUF_SIMPLE = 0
PyBUF_WRITABLE = PyBUF_WRITEABLE = 0x0001
PyBUF_ND = 0x0008
PyBUF_STRIDES = (0x0010 | PyBUF_ND)
PyBUF_CONTIG = (PyBUF_ND | PyBUF_WRITABLE)
PyBUF_CONTIG_RO = (PyBUF_ND)
PyBUF_C_CONTIGUOUS = (0x0020 | PyBUF_STRIDES)
PyBUF_F_CONTIGUOUS = (0x0040 | PyBUF_STRIDES)
PyBUF_ANY_CONTIGUOUS = (0x0080 | PyBUF_STRIDES)
PyBUF_FORMAT = 0x0004
# Python 2.6 doesn't define this...
c_ssize_t = getattr(ctypes, 'c_ssize_t', ctypes.c_ulong)
_fields_ = [
('buf', ctypes.c_void_p),
('obj', ctypes.c_void_p),
('len', c_ssize_t),
('itemsize', c_ssize_t),
('readonly', ctypes.c_int),
('ndim', ctypes.c_int),
('format', ctypes.c_char_p),
('shape', ctypes.POINTER(c_ssize_t)),
('strides', ctypes.POINTER(c_ssize_t)),
('suboffsets', ctypes.POINTER(c_ssize_t)),
]
if sys.version_info[:2] <= (2, 6) or sys.version_info[:2] >= (3, 3):
# Original structure was eventually restored in 3.3, so just
# 2.7 through 3.2 uses the "enhanced" structure below
_fields_.extend([
('internal', ctypes.c_void_p),
])
else:
# Sigh, this structure seems to have changed with Python 3.x...
_fields_.extend([
('smalltable', ctypes.c_size_t*2),
('internal', ctypes.c_void_p),
])
class Py_buffer(ctypes.Structure): # noqa
@classmethod
def from_object(cls, obj, flags):
"""Create a new Py_buffer referencing ram of object"""
if not CheckBuffer(obj):
raise TypeError(
"%s type does not support Buffer Protocol" % (obj.__class__,))
buf = cls()
# deallocation of the buf causes glibc abort :(
result = GetBuffer(obj, buf, flags)
if result != 0:
raise ValueError("Unable to retrieve Buffer from %s" % (obj,))
if not buf.buf:
raise ValueError("Null pointer result from %s" % (obj,))
return buf
_fields_ = _fields_
@property
def dims(self):
return self.shape[:self.ndim]
def __len__(self):
return self.shape[0]
@property
def dim_strides(self):
if self.strides:
return self.strides[:self.ndim]
return None
def __enter__(self):
return self
def __exit__(self, exc_type=None, exc_value=None, traceback=None):
if self.obj:
ReleaseBuffer(self)
def __del__(self):
if self.obj:
ReleaseBuffer(self)
try:
CheckBuffer = ctypes.pythonapi.PyObject_CheckBuffer
CheckBuffer.argtypes = [ctypes.py_object]
CheckBuffer.restype = ctypes.c_int
except AttributeError as err:
# Python 2.6 doesn't appear to have CheckBuffer support...
def CheckBuffer(x): # noqa
return True
IncRef = ctypes.pythonapi.Py_IncRef
IncRef.argtypes = [ctypes.py_object]
GetBuffer = ctypes.pythonapi.PyObject_GetBuffer
GetBuffer.argtypes = [ctypes.py_object, ctypes.POINTER(Py_buffer), ctypes.c_int]
GetBuffer.restype = ctypes.c_int
ReleaseBuffer = ctypes.pythonapi.PyBuffer_Release
ReleaseBuffer.argtypes = [ctypes.POINTER(Py_buffer)]
ReleaseBuffer.restype = None
|
