Skip to content

Commit 576e784

Browse files
committed
Use self.elementsize internally
1 parent e5ccae6 commit 576e784

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

src/pa_ringbuffer.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -74,8 +74,6 @@ def __init__(self, elementsize, size):
7474
if res != 0:
7575
assert res == -1
7676
raise ValueError('size must be a power of 2')
77-
assert self._ptr.bufferSize == size
78-
assert self._ptr.elementSizeBytes == elementsize
7977

8078
def flush(self):
8179
"""Reset buffer to empty.
@@ -116,8 +114,7 @@ def write(self, data, size=-1):
116114
except TypeError:
117115
pass # input is not a buffer
118116
if size < 0:
119-
size, rest = divmod(self._ffi.sizeof(data),
120-
self._ptr.elementSizeBytes)
117+
size, rest = divmod(self._ffi.sizeof(data), self.elementsize)
121118
if rest:
122119
raise ValueError('data size must be multiple of elementsize')
123120
return self._lib.PaUtil_WriteRingBuffer(self._ptr, data, size)
@@ -140,10 +137,9 @@ def read(self, size=-1):
140137
"""
141138
if size < 0:
142139
size = self.read_available
143-
data = self._ffi.new(
144-
'unsigned char[]', size * self._ptr.elementSizeBytes)
140+
data = self._ffi.new('unsigned char[]', size * self.elementsize)
145141
size = self.readinto(data)
146-
return self._ffi.buffer(data, size * self._ptr.elementSizeBytes)
142+
return self._ffi.buffer(data, size * self.elementsize)
147143

148144
def readinto(self, data):
149145
"""Read data from the ring buffer into a user-provided buffer.
@@ -164,7 +160,7 @@ def readinto(self, data):
164160
data = self._ffi.from_buffer(data)
165161
except TypeError:
166162
pass # input is not a buffer
167-
size, rest = divmod(self._ffi.sizeof(data), self._ptr.elementSizeBytes)
163+
size, rest = divmod(self._ffi.sizeof(data), self.elementsize)
168164
if rest:
169165
raise ValueError('data size must be multiple of elementsize')
170166
return self._lib.PaUtil_ReadRingBuffer(self._ptr, data, size)
@@ -264,4 +260,3 @@ def elementsize(self):
264260
def __len__(self):
265261
"""Size of buffer in elements"""
266262
return self._ptr.bufferSize
267-

0 commit comments

Comments
 (0)