Commit c3b39245 authored by Antoine Pitrou's avatar Antoine Pitrou

Issue #4580: slicing of memoryviews when itemsize != 1 is wrong.

Also fix len() to return number of items rather than length in bytes.

I'm sorry it was not possible for me to work on this without reindenting
a bit some stuff around. The indentation in memoryobject.c is a mess,
I'll open a separate bug for it.
parent 8bcddcab
...@@ -144,16 +144,18 @@ typedef int(*objobjargproc)(PyObject *, PyObject *, PyObject *); ...@@ -144,16 +144,18 @@ typedef int(*objobjargproc)(PyObject *, PyObject *, PyObject *);
typedef struct bufferinfo { typedef struct bufferinfo {
void *buf; void *buf;
PyObject *obj; /* owned reference */ PyObject *obj; /* owned reference */
Py_ssize_t len; Py_ssize_t len;
Py_ssize_t itemsize; /* This is Py_ssize_t so it can be Py_ssize_t itemsize; /* This is Py_ssize_t so it can be
pointed to by strides in simple case.*/ pointed to by strides in simple case.*/
int readonly; int readonly;
int ndim; int ndim;
char *format; char *format;
Py_ssize_t *shape; Py_ssize_t *shape;
Py_ssize_t *strides; Py_ssize_t *strides;
Py_ssize_t *suboffsets; Py_ssize_t *suboffsets;
void *internal; Py_ssize_t smalltable[2]; /* static store for shape and strides of
mono-dimensional buffers. */
void *internal;
} Py_buffer; } Py_buffer;
typedef int (*getbufferproc)(PyObject *, Py_buffer *, int); typedef int (*getbufferproc)(PyObject *, Py_buffer *, int);
......
...@@ -25,7 +25,10 @@ class Test(unittest.TestCase): ...@@ -25,7 +25,10 @@ class Test(unittest.TestCase):
v = memoryview(ob) v = memoryview(ob)
try: try:
self.failUnlessEqual(normalize(v.format), normalize(fmt)) self.failUnlessEqual(normalize(v.format), normalize(fmt))
self.failUnlessEqual(len(v), sizeof(ob)) if shape is not None:
self.failUnlessEqual(len(v), shape[0])
else:
self.failUnlessEqual(len(v) * sizeof(itemtp), sizeof(ob))
self.failUnlessEqual(v.itemsize, sizeof(itemtp)) self.failUnlessEqual(v.itemsize, sizeof(itemtp))
self.failUnlessEqual(v.shape, shape) self.failUnlessEqual(v.shape, shape)
# ctypes object always have a non-strided memory block # ctypes object always have a non-strided memory block
...@@ -37,7 +40,7 @@ class Test(unittest.TestCase): ...@@ -37,7 +40,7 @@ class Test(unittest.TestCase):
n = 1 n = 1
for dim in v.shape: for dim in v.shape:
n = n * dim n = n * dim
self.failUnlessEqual(v.itemsize * n, len(v)) self.failUnlessEqual(n * v.itemsize, len(v.tobytes()))
except: except:
# so that we can see the failing type # so that we can see the failing type
print(tp) print(tp)
...@@ -49,7 +52,10 @@ class Test(unittest.TestCase): ...@@ -49,7 +52,10 @@ class Test(unittest.TestCase):
v = memoryview(ob) v = memoryview(ob)
try: try:
self.failUnlessEqual(v.format, fmt) self.failUnlessEqual(v.format, fmt)
self.failUnlessEqual(len(v), sizeof(ob)) if shape is not None:
self.failUnlessEqual(len(v), shape[0])
else:
self.failUnlessEqual(len(v) * sizeof(itemtp), sizeof(ob))
self.failUnlessEqual(v.itemsize, sizeof(itemtp)) self.failUnlessEqual(v.itemsize, sizeof(itemtp))
self.failUnlessEqual(v.shape, shape) self.failUnlessEqual(v.shape, shape)
# ctypes object always have a non-strided memory block # ctypes object always have a non-strided memory block
...@@ -61,7 +67,7 @@ class Test(unittest.TestCase): ...@@ -61,7 +67,7 @@ class Test(unittest.TestCase):
n = 1 n = 1
for dim in v.shape: for dim in v.shape:
n = n * dim n = n * dim
self.failUnlessEqual(v.itemsize * n, len(v)) self.failUnlessEqual(n, len(v))
except: except:
# so that we can see the failing type # so that we can see the failing type
print(tp) print(tp)
......
...@@ -260,7 +260,7 @@ class IOTest(unittest.TestCase): ...@@ -260,7 +260,7 @@ class IOTest(unittest.TestCase):
def test_array_writes(self): def test_array_writes(self):
a = array.array('i', range(10)) a = array.array('i', range(10))
n = len(memoryview(a)) n = len(a.tostring())
f = io.open(support.TESTFN, "wb", 0) f = io.open(support.TESTFN, "wb", 0)
self.assertEqual(f.write(a), n) self.assertEqual(f.write(a), n)
f.close() f.close()
......
...@@ -8,24 +8,30 @@ import test.support ...@@ -8,24 +8,30 @@ import test.support
import sys import sys
import gc import gc
import weakref import weakref
import array
class CommonMemoryTests: class AbstractMemoryTests:
# source_bytes = b"abcdef"
# Tests common to direct memoryviews and sliced memoryviews
#
base_object = b"abcdef" @property
def _source(self):
return self.source_bytes
@property
def _types(self):
return filter(None, [self.ro_type, self.rw_type])
def check_getitem_with_type(self, tp): def check_getitem_with_type(self, tp):
b = tp(self.base_object) item = self.getitem_type
b = tp(self._source)
oldrefcount = sys.getrefcount(b) oldrefcount = sys.getrefcount(b)
m = self._view(b) m = self._view(b)
self.assertEquals(m[0], b"a") self.assertEquals(m[0], item(b"a"))
self.assert_(isinstance(m[0], bytes), type(m[0])) self.assert_(isinstance(m[0], bytes), type(m[0]))
self.assertEquals(m[5], b"f") self.assertEquals(m[5], item(b"f"))
self.assertEquals(m[-1], b"f") self.assertEquals(m[-1], item(b"f"))
self.assertEquals(m[-6], b"a") self.assertEquals(m[-6], item(b"a"))
# Bounds checking # Bounds checking
self.assertRaises(IndexError, lambda: m[6]) self.assertRaises(IndexError, lambda: m[6])
self.assertRaises(IndexError, lambda: m[-7]) self.assertRaises(IndexError, lambda: m[-7])
...@@ -38,14 +44,14 @@ class CommonMemoryTests: ...@@ -38,14 +44,14 @@ class CommonMemoryTests:
m = None m = None
self.assertEquals(sys.getrefcount(b), oldrefcount) self.assertEquals(sys.getrefcount(b), oldrefcount)
def test_getitem_readonly(self): def test_getitem(self):
self.check_getitem_with_type(bytes) for tp in self._types:
self.check_getitem_with_type(tp)
def test_getitem_writable(self):
self.check_getitem_with_type(bytearray)
def test_setitem_readonly(self): def test_setitem_readonly(self):
b = self.base_object if not self.ro_type:
return
b = self.ro_type(self._source)
oldrefcount = sys.getrefcount(b) oldrefcount = sys.getrefcount(b)
m = self._view(b) m = self._view(b)
def setitem(value): def setitem(value):
...@@ -57,27 +63,30 @@ class CommonMemoryTests: ...@@ -57,27 +63,30 @@ class CommonMemoryTests:
self.assertEquals(sys.getrefcount(b), oldrefcount) self.assertEquals(sys.getrefcount(b), oldrefcount)
def test_setitem_writable(self): def test_setitem_writable(self):
b = bytearray(self.base_object) if not self.rw_type:
return
tp = self.rw_type
b = self.rw_type(self._source)
oldrefcount = sys.getrefcount(b) oldrefcount = sys.getrefcount(b)
m = self._view(b) m = self._view(b)
m[0] = b"0" m[0] = tp(b"0")
self._check_contents(b, b"0bcdef") self._check_contents(tp, b, b"0bcdef")
m[1:3] = b"12" m[1:3] = tp(b"12")
self._check_contents(b, b"012def") self._check_contents(tp, b, b"012def")
m[1:1] = b"" m[1:1] = tp(b"")
self._check_contents(b, b"012def") self._check_contents(tp, b, b"012def")
m[:] = b"abcdef" m[:] = tp(b"abcdef")
self._check_contents(b, b"abcdef") self._check_contents(tp, b, b"abcdef")
# Overlapping copies of a view into itself # Overlapping copies of a view into itself
m[0:3] = m[2:5] m[0:3] = m[2:5]
self._check_contents(b, b"cdedef") self._check_contents(tp, b, b"cdedef")
m[:] = b"abcdef" m[:] = tp(b"abcdef")
m[2:5] = m[0:3] m[2:5] = m[0:3]
self._check_contents(b, b"ababcf") self._check_contents(tp, b, b"ababcf")
def setitem(key, value): def setitem(key, value):
m[key] = value m[key] = tp(value)
# Bounds checking # Bounds checking
self.assertRaises(IndexError, setitem, 6, b"a") self.assertRaises(IndexError, setitem, 6, b"a")
self.assertRaises(IndexError, setitem, -7, b"a") self.assertRaises(IndexError, setitem, -7, b"a")
...@@ -96,159 +105,224 @@ class CommonMemoryTests: ...@@ -96,159 +105,224 @@ class CommonMemoryTests:
m = None m = None
self.assertEquals(sys.getrefcount(b), oldrefcount) self.assertEquals(sys.getrefcount(b), oldrefcount)
def test_len(self):
self.assertEquals(len(self._view(self.base_object)), 6)
def test_tobytes(self): def test_tobytes(self):
m = self._view(self.base_object) for tp in self._types:
b = m.tobytes() m = self._view(tp(self._source))
self.assertEquals(b, b"abcdef") b = m.tobytes()
self.assert_(isinstance(b, bytes), type(b)) # This calls self.getitem_type() on each separate byte of b"abcdef"
expected = b"".join(
self.getitem_type(bytes([c])) for c in b"abcdef")
self.assertEquals(b, expected)
self.assert_(isinstance(b, bytes), type(b))
def test_tolist(self): def test_tolist(self):
m = self._view(self.base_object) for tp in self._types:
l = m.tolist() m = self._view(tp(self._source))
self.assertEquals(l, list(b"abcdef")) l = m.tolist()
self.assertEquals(l, list(b"abcdef"))
def test_compare(self): def test_compare(self):
# memoryviews can compare for equality with other objects # memoryviews can compare for equality with other objects
# having the buffer interface. # having the buffer interface.
m = self._view(self.base_object) for tp in self._types:
for tp in (bytes, bytearray): m = self._view(tp(self._source))
self.assertTrue(m == tp(b"abcdef")) for tp_comp in self._types:
self.assertFalse(m != tp(b"abcdef")) self.assertTrue(m == tp_comp(b"abcdef"))
self.assertFalse(m == tp(b"abcde")) self.assertFalse(m != tp_comp(b"abcdef"))
self.assertTrue(m != tp(b"abcde")) self.assertFalse(m == tp_comp(b"abcde"))
self.assertFalse(m == tp(b"abcde1")) self.assertTrue(m != tp_comp(b"abcde"))
self.assertTrue(m != tp(b"abcde1")) self.assertFalse(m == tp_comp(b"abcde1"))
self.assertTrue(m == m) self.assertTrue(m != tp_comp(b"abcde1"))
self.assertTrue(m == m[:]) self.assertTrue(m == m)
self.assertTrue(m[0:6] == m[:]) self.assertTrue(m == m[:])
self.assertFalse(m[0:5] == m) self.assertTrue(m[0:6] == m[:])
self.assertFalse(m[0:5] == m)
# Comparison with objects which don't support the buffer API
self.assertFalse(m == "abc") # Comparison with objects which don't support the buffer API
self.assertTrue(m != "abc") self.assertFalse(m == "abcdef")
self.assertFalse("abc" == m) self.assertTrue(m != "abcdef")
self.assertTrue("abc" != m) self.assertFalse("abcdef" == m)
self.assertTrue("abcdef" != m)
# Unordered comparisons
for c in (m, b"abcdef"): # Unordered comparisons
self.assertRaises(TypeError, lambda: m < c) for c in (m, b"abcdef"):
self.assertRaises(TypeError, lambda: c <= m) self.assertRaises(TypeError, lambda: m < c)
self.assertRaises(TypeError, lambda: m >= c) self.assertRaises(TypeError, lambda: c <= m)
self.assertRaises(TypeError, lambda: c > m) self.assertRaises(TypeError, lambda: m >= c)
self.assertRaises(TypeError, lambda: c > m)
def check_attributes_with_type(self, tp): def check_attributes_with_type(self, tp):
b = tp(self.base_object) m = self._view(tp(self._source))
m = self._view(b) self.assertEquals(m.format, self.format)
self.assertEquals(m.format, 'B') self.assertEquals(m.itemsize, self.itemsize)
self.assertEquals(m.itemsize, 1)
self.assertEquals(m.ndim, 1) self.assertEquals(m.ndim, 1)
self.assertEquals(m.shape, (6,)) self.assertEquals(m.shape, (6,))
self.assertEquals(len(m), 6) self.assertEquals(len(m), 6)
self.assertEquals(m.strides, (1,)) self.assertEquals(m.strides, (self.itemsize,))
self.assertEquals(m.suboffsets, None) self.assertEquals(m.suboffsets, None)
return m return m
def test_attributes_readonly(self): def test_attributes_readonly(self):
m = self.check_attributes_with_type(bytes) if not self.ro_type:
return
m = self.check_attributes_with_type(self.ro_type)
self.assertEquals(m.readonly, True) self.assertEquals(m.readonly, True)
def test_attributes_writable(self): def test_attributes_writable(self):
m = self.check_attributes_with_type(bytearray) if not self.rw_type:
return
m = self.check_attributes_with_type(self.rw_type)
self.assertEquals(m.readonly, False) self.assertEquals(m.readonly, False)
def test_getbuffer(self): def test_getbuffer(self):
# Test PyObject_GetBuffer() on a memoryview object. # Test PyObject_GetBuffer() on a memoryview object.
b = self.base_object for tp in self._types:
oldrefcount = sys.getrefcount(b) b = tp(self._source)
m = self._view(b) oldrefcount = sys.getrefcount(b)
oldviewrefcount = sys.getrefcount(m) m = self._view(b)
s = str(m, "utf-8") oldviewrefcount = sys.getrefcount(m)
self._check_contents(b, s.encode("utf-8")) s = str(m, "utf-8")
self.assertEquals(sys.getrefcount(m), oldviewrefcount) self._check_contents(tp, b, s.encode("utf-8"))
m = None self.assertEquals(sys.getrefcount(m), oldviewrefcount)
self.assertEquals(sys.getrefcount(b), oldrefcount) m = None
self.assertEquals(sys.getrefcount(b), oldrefcount)
def test_gc(self): def test_gc(self):
class MyBytes(bytes): for tp in self._types:
pass if not isinstance(tp, type):
class MyObject: # If tp is a factory rather than a plain type, skip
pass continue
class MySource(tp):
pass
class MyObject:
pass
# Create a reference cycle through a memoryview object
b = MySource(tp(b'abc'))
m = self._view(b)
o = MyObject()
b.m = m
b.o = o
wr = weakref.ref(o)
b = m = o = None
# The cycle must be broken
gc.collect()
self.assert_(wr() is None, wr())
# Variations on source objects for the buffer: bytes-like objects, then arrays
# with itemsize > 1.
# NOTE: support for multi-dimensional objects is unimplemented.
class BaseBytesMemoryTests(AbstractMemoryTests):
ro_type = bytes
rw_type = bytearray
getitem_type = bytes
itemsize = 1
format = 'B'
class BaseArrayMemoryTests(AbstractMemoryTests):
ro_type = None
rw_type = lambda self, b: array.array('i', list(b))
getitem_type = lambda self, b: array.array('i', list(b)).tostring()
itemsize = array.array('i').itemsize
format = 'i'
# Create a reference cycle through a memoryview object def test_getbuffer(self):
b = MyBytes(b'abc') # XXX Test should be adapted for non-byte buffers
m = self._view(b) pass
o = MyObject()
b.m = m def test_tolist(self):
b.o = o # XXX NotImplementedError: tolist() only supports byte views
wr = weakref.ref(o) pass
b = m = o = None
# The cycle must be broken
gc.collect()
self.assert_(wr() is None, wr())
class MemoryviewTest(unittest.TestCase, CommonMemoryTests): # Variations on indirection levels: memoryview, slice of memoryview,
# slice of slice of memoryview.
# This is important to test allocation subtleties.
class BaseMemoryviewTests:
def _view(self, obj): def _view(self, obj):
return memoryview(obj) return memoryview(obj)
def _check_contents(self, obj, contents): def _check_contents(self, tp, obj, contents):
self.assertEquals(obj, contents) self.assertEquals(obj, tp(contents))
def test_constructor(self):
ob = b'test'
self.assert_(memoryview(ob))
self.assert_(memoryview(object=ob))
self.assertRaises(TypeError, memoryview)
self.assertRaises(TypeError, memoryview, ob, ob)
self.assertRaises(TypeError, memoryview, argument=ob)
self.assertRaises(TypeError, memoryview, ob, argument=True)
def test_array_assign(self):
# Issue #4569: segfault when mutating a memoryview with itemsize != 1
from array import array
a = array('i', range(10))
m = memoryview(a)
new_a = array('i', range(9, -1, -1))
m[:] = new_a
self.assertEquals(a, new_a)
class BaseMemorySliceTests:
class MemorySliceTest(unittest.TestCase, CommonMemoryTests): source_bytes = b"XabcdefY"
base_object = b"XabcdefY"
def _view(self, obj): def _view(self, obj):
m = memoryview(obj) m = memoryview(obj)
return m[1:7] return m[1:7]
def _check_contents(self, obj, contents): def _check_contents(self, tp, obj, contents):
self.assertEquals(obj[1:7], contents) self.assertEquals(obj[1:7], tp(contents))
def test_refs(self): def test_refs(self):
m = memoryview(b"ab") for tp in self._types:
oldrefcount = sys.getrefcount(m) m = memoryview(tp(self._source))
m[1:2] oldrefcount = sys.getrefcount(m)
self.assertEquals(sys.getrefcount(m), oldrefcount) m[1:2]
self.assertEquals(sys.getrefcount(m), oldrefcount)
class MemorySliceSliceTest(unittest.TestCase, CommonMemoryTests): class BaseMemorySliceSliceTests:
base_object = b"XabcdefY" source_bytes = b"XabcdefY"
def _view(self, obj): def _view(self, obj):
m = memoryview(obj) m = memoryview(obj)
return m[:7][1:] return m[:7][1:]
def _check_contents(self, obj, contents): def _check_contents(self, tp, obj, contents):
self.assertEquals(obj[1:7], contents) self.assertEquals(obj[1:7], tp(contents))
def test_main(): # Concrete test classes
test.support.run_unittest(
MemoryviewTest, MemorySliceTest, MemorySliceSliceTest) class BytesMemoryviewTest(unittest.TestCase,
BaseMemoryviewTests, BaseBytesMemoryTests):
def test_constructor(self):
for tp in self._types:
ob = tp(self._source)
self.assert_(memoryview(ob))
self.assert_(memoryview(object=ob))
self.assertRaises(TypeError, memoryview)
self.assertRaises(TypeError, memoryview, ob, ob)
self.assertRaises(TypeError, memoryview, argument=ob)
self.assertRaises(TypeError, memoryview, ob, argument=True)
class ArrayMemoryviewTest(unittest.TestCase,
BaseMemoryviewTests, BaseArrayMemoryTests):
def test_array_assign(self):
# Issue #4569: segfault when mutating a memoryview with itemsize != 1
a = array.array('i', range(10))
m = memoryview(a)
new_a = array.array('i', range(9, -1, -1))
m[:] = new_a
self.assertEquals(a, new_a)
class BytesMemorySliceTest(unittest.TestCase,
BaseMemorySliceTests, BaseBytesMemoryTests):
pass
class ArrayMemorySliceTest(unittest.TestCase,
BaseMemorySliceTests, BaseArrayMemoryTests):
pass
class BytesMemorySliceSliceTest(unittest.TestCase,
BaseMemorySliceSliceTests, BaseBytesMemoryTests):
pass
class ArrayMemorySliceSliceTest(unittest.TestCase,
BaseMemorySliceSliceTests, BaseArrayMemoryTests):
pass
def test_main():
test.support.run_unittest(__name__)
if __name__ == "__main__": if __name__ == "__main__":
test_main() test_main()
...@@ -559,7 +559,7 @@ class SizeofTest(unittest.TestCase): ...@@ -559,7 +559,7 @@ class SizeofTest(unittest.TestCase):
check(32768*32768-1, size(vh) + 2*self.H) check(32768*32768-1, size(vh) + 2*self.H)
check(32768*32768, size(vh) + 3*self.H) check(32768*32768, size(vh) + 3*self.H)
# memory # memory
check(memoryview(b''), size(h + 'P PP2P2i5P')) check(memoryview(b''), size(h + 'P PP2P2i7P'))
# module # module
check(unittest, size(h + '3P')) check(unittest, size(h + '3P'))
# None # None
......
...@@ -12,6 +12,10 @@ What's New in Python 3.1 alpha 0 ...@@ -12,6 +12,10 @@ What's New in Python 3.1 alpha 0
Core and Builtins Core and Builtins
----------------- -----------------
- Issue #4580: Fix slicing of memoryviews when the item size is greater than
one byte. Also fixes the meaning of len() so that it returns the number of
items, rather than the size in bytes.
- Issue #4075: Use OutputDebugStringW in Py_FatalError. - Issue #4075: Use OutputDebugStringW in Py_FatalError.
- Issue #4747: When the terminal does not use utf-8, executing a script with - Issue #4747: When the terminal does not use utf-8, executing a script with
......
...@@ -3,26 +3,31 @@ ...@@ -3,26 +3,31 @@
#include "Python.h" #include "Python.h"
static void
dup_buffer(Py_buffer *dest, Py_buffer *src)
{
*dest = *src;
if (src->shape == &(src->len))
dest->shape = &(dest->len);
if (src->strides == &(src->itemsize))
dest->strides = &(dest->itemsize);
}
/* XXX The buffer API should mandate that the shape array be non-NULL, but
it would complicate some code since the (de)allocation semantics of shape
are not specified. */
static Py_ssize_t static Py_ssize_t
get_shape0(Py_buffer *buf) get_shape0(Py_buffer *buf)
{ {
if (buf->shape != NULL) if (buf->shape != NULL)
return buf->shape[0]; return buf->shape[0];
assert(buf->ndim == 1 && buf->itemsize > 0); if (buf->ndim == 0)
return buf->len / buf->itemsize; return 1;
PyErr_SetString(PyExc_TypeError,
"exported buffer does not have any shape information associated "
"to it");
return -1;
}
static void
dup_buffer(Py_buffer *dest, Py_buffer *src)
{
*dest = *src;
if (src->ndim == 1 && src->shape != NULL) {
dest->shape = &(dest->smalltable[0]);
dest->shape[0] = get_shape0(src);
}
if (src->ndim == 1 && src->strides != NULL) {
dest->strides = &(dest->smalltable[1]);
dest->strides[0] = src->strides[0];
}
} }
static int static int
...@@ -449,8 +454,6 @@ memory_tolist(PyMemoryViewObject *mem, PyObject *noargs) ...@@ -449,8 +454,6 @@ memory_tolist(PyMemoryViewObject *mem, PyObject *noargs)
return res; return res;
} }
static PyMethodDef memory_methods[] = { static PyMethodDef memory_methods[] = {
{"tobytes", (PyCFunction)memory_tobytes, METH_NOARGS, NULL}, {"tobytes", (PyCFunction)memory_tobytes, METH_NOARGS, NULL},
{"tolist", (PyCFunction)memory_tolist, METH_NOARGS, NULL}, {"tolist", (PyCFunction)memory_tolist, METH_NOARGS, NULL},
...@@ -474,19 +477,19 @@ memory_dealloc(PyMemoryViewObject *self) ...@@ -474,19 +477,19 @@ memory_dealloc(PyMemoryViewObject *self)
PyObject_CopyData(PyTuple_GET_ITEM(self->base,0), PyObject_CopyData(PyTuple_GET_ITEM(self->base,0),
PyTuple_GET_ITEM(self->base,1)); PyTuple_GET_ITEM(self->base,1));
/* The view member should have readonly == -1 in /* The view member should have readonly == -1 in
this instance indicating that the memory can this instance indicating that the memory can
be "locked" and was locked and will be unlocked be "locked" and was locked and will be unlocked
again after this call. again after this call.
*/ */
PyBuffer_Release(&(self->view)); PyBuffer_Release(&(self->view));
}
else {
PyBuffer_Release(&(self->view));
}
Py_CLEAR(self->base);
} }
PyObject_GC_Del(self); else {
PyBuffer_Release(&(self->view));
}
Py_CLEAR(self->base);
}
PyObject_GC_Del(self);
} }
static PyObject * static PyObject *
...@@ -512,16 +515,10 @@ memory_str(PyMemoryViewObject *self) ...@@ -512,16 +515,10 @@ memory_str(PyMemoryViewObject *self)
} }
/* Sequence methods */ /* Sequence methods */
static Py_ssize_t static Py_ssize_t
memory_length(PyMemoryViewObject *self) memory_length(PyMemoryViewObject *self)
{ {
Py_buffer view; return get_shape0(&self->view);
if (PyObject_GetBuffer((PyObject *)self, &view, PyBUF_FULL) < 0)
return -1;
PyBuffer_Release(&view);
return view.len;
} }
/* /*
...@@ -589,40 +586,38 @@ memory_subscript(PyMemoryViewObject *self, PyObject *key) ...@@ -589,40 +586,38 @@ memory_subscript(PyMemoryViewObject *self, PyObject *key)
} }
} }
else if (PySlice_Check(key)) { else if (PySlice_Check(key)) {
Py_ssize_t start, stop, step, slicelength; Py_ssize_t start, stop, step, slicelength;
if (PySlice_GetIndicesEx((PySliceObject*)key, get_shape0(view), if (PySlice_GetIndicesEx((PySliceObject*)key, get_shape0(view),
&start, &stop, &step, &slicelength) < 0) { &start, &stop, &step, &slicelength) < 0) {
return NULL; return NULL;
} }
if (step == 1 && view->ndim == 1) { if (step == 1 && view->ndim == 1) {
Py_buffer newview; Py_buffer newview;
void *newbuf = (char *) view->buf void *newbuf = (char *) view->buf
+ start * view->itemsize; + start * view->itemsize;
int newflags = view->readonly int newflags = view->readonly
? PyBUF_CONTIG_RO : PyBUF_CONTIG; ? PyBUF_CONTIG_RO : PyBUF_CONTIG;
/* XXX There should be an API to create a subbuffer */ /* XXX There should be an API to create a subbuffer */
if (view->obj != NULL) { if (view->obj != NULL) {
if (PyObject_GetBuffer(view->obj, if (PyObject_GetBuffer(view->obj, &newview, newflags) == -1)
&newview, newflags) == -1) return NULL;
return NULL; }
} else {
else { newview = *view;
newview = *view; }
} newview.buf = newbuf;
newview.buf = newbuf; newview.len = slicelength * newview.itemsize;
newview.len = slicelength; newview.format = view->format;
newview.format = view->format; newview.shape = &(newview.smalltable[0]);
if (view->shape == &(view->len)) newview.shape[0] = slicelength;
newview.shape = &(newview.len); newview.strides = &(newview.itemsize);
if (view->strides == &(view->itemsize)) return PyMemoryView_FromBuffer(&newview);
newview.strides = &(newview.itemsize); }
return PyMemoryView_FromBuffer(&newview); PyErr_SetNone(PyExc_NotImplementedError);
} return NULL;
PyErr_SetNone(PyExc_NotImplementedError);
return NULL;
} }
PyErr_Format(PyExc_TypeError, PyErr_Format(PyExc_TypeError,
"cannot index memory using \"%.200s\"", "cannot index memory using \"%.200s\"",
...@@ -747,7 +742,7 @@ memory_richcompare(PyObject *v, PyObject *w, int op) ...@@ -747,7 +742,7 @@ memory_richcompare(PyObject *v, PyObject *w, int op)
if (vv.itemsize != ww.itemsize || vv.len != ww.len) if (vv.itemsize != ww.itemsize || vv.len != ww.len)
goto _end; goto _end;
equal = !memcmp(vv.buf, ww.buf, vv.len * vv.itemsize); equal = !memcmp(vv.buf, ww.buf, vv.len);
_end: _end:
PyBuffer_Release(&vv); PyBuffer_Release(&vv);
......
...@@ -1205,7 +1205,7 @@ PyObject *PyUnicode_Decode(const char *s, ...@@ -1205,7 +1205,7 @@ PyObject *PyUnicode_Decode(const char *s,
/* Decode via the codec registry */ /* Decode via the codec registry */
buffer = NULL; buffer = NULL;
if (PyBuffer_FillInfo(&info, NULL, (void *)s, size, 1, PyBUF_SIMPLE) < 0) if (PyBuffer_FillInfo(&info, NULL, (void *)s, size, 1, PyBUF_FULL_RO) < 0)
goto onError; goto onError;
buffer = PyMemoryView_FromBuffer(&info); buffer = PyMemoryView_FromBuffer(&info);
if (buffer == NULL) if (buffer == NULL)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment