cpython-lru_cache-weakref: Objects/memoryobject.c annotate

annotate Objects/memoryobject.c @ 57303:4aa4c19dfaed legacy-trunk

#1757126: fix typo with the cyrillic_asian alias

author	Philip Jenvey <pjenvey@underboss.org>
date	Tue, 17 Nov 2009 02:42:26 +0000
parents	1617c11556af
children	c37f99b54d7c

rev	line source
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	1
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	2 /* Memoryview object implementation */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	3
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	4 #include "Python.h"
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	5
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	6 static Py_ssize_t
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	7 get_shape0(Py_buffer *buf)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	8 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	9 if (buf->shape != NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	10 return buf->shape[0];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	11 if (buf->ndim == 0)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	12 return 1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	13 PyErr_SetString(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	14 "exported buffer does not have any shape information associated "
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	15 "to it");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	16 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	17 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	18
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	19 static void
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	20 dup_buffer(Py_buffer dest, Py_buffer src)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	21 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	22 dest = src;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	23 if (src->ndim == 1 && src->shape != NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	24 dest->shape = &(dest->smalltable[0]);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	25 dest->shape[0] = get_shape0(src);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	26 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	27 if (src->ndim == 1 && src->strides != NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	28 dest->strides = &(dest->smalltable[1]);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	29 dest->strides[0] = src->strides[0];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	30 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	31 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	32
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	33 static int
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	34 memory_getbuf(PyMemoryViewObject self, Py_buffer view, int flags)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	35 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	36 int res = 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	37 /* XXX for whatever reason fixing the flags seems necessary */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	38 if (self->view.readonly)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	39 flags &= ~PyBUF_WRITABLE;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	40 if (self->view.obj != NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	41 res = PyObject_GetBuffer(self->view.obj, view, flags);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	42 if (view)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	43 dup_buffer(view, &self->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	44 return res;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	45 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	46
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	47 static void
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	48 memory_releasebuf(PyMemoryViewObject self, Py_buffer view)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	49 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	50 PyBuffer_Release(view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	51 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	52
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	53 PyDoc_STRVAR(memory_doc,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	54 "memoryview(object)\n\
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	55 \n\
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	56 Create a new memoryview object which references the given object.");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	57
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	58 PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	59 PyMemoryView_FromBuffer(Py_buffer *info)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	60 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	61 PyMemoryViewObject *mview;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	62
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	63 mview = (PyMemoryViewObject *)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	64 PyObject_GC_New(PyMemoryViewObject, &PyMemoryView_Type);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	65 if (mview == NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	66 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	67 mview->base = NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	68 dup_buffer(&mview->view, info);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	69 /* NOTE: mview->view.obj should already have been incref'ed as
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	70 part of PyBuffer_FillInfo(). */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	71 _PyObject_GC_TRACK(mview);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	72 return (PyObject *)mview;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	73 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	74
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	75 PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	76 PyMemoryView_FromObject(PyObject *base)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	77 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	78 PyMemoryViewObject *mview;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	79
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	80 if (!PyObject_CheckBuffer(base)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	81 PyErr_SetString(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	82 "cannot make memory view because object does "
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	83 "not have the buffer interface");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	84 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	85 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	86
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	87 mview = (PyMemoryViewObject *)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	88 PyObject_GC_New(PyMemoryViewObject, &PyMemoryView_Type);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	89 if (mview == NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	90 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	91
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	92 mview->base = NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	93 if (PyObject_GetBuffer(base, &(mview->view), PyBUF_FULL_RO) < 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	94 Py_DECREF(mview);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	95 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	96 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	97
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	98 mview->base = base;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	99 Py_INCREF(base);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	100 _PyObject_GC_TRACK(mview);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	101 return (PyObject *)mview;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	102 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	103
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	104 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	105 memory_new(PyTypeObject subtype, PyObject args, PyObject *kwds)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	106 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	107 PyObject *obj;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	108 static char *kwlist[] = {"object", 0};
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	109
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	110 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O:memoryview", kwlist,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	111 &obj)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	112 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	113 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	114
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	115 return PyMemoryView_FromObject(obj);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	116 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	117
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	118
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	119 static void
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	120 _strided_copy_nd(char dest, char src, int nd, Py_ssize_t *shape,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	121 Py_ssize_t *strides, Py_ssize_t itemsize, char fort)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	122 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	123 int k;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	124 Py_ssize_t outstride;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	125
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	126 if (nd==0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	127 memcpy(dest, src, itemsize);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	128 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	129 else if (nd == 1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	130 for (k = 0; k<shape[0]; k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	131 memcpy(dest, src, itemsize);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	132 dest += itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	133 src += strides[0];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	134 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	135 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	136 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	137 if (fort == 'F') {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	138 /* Copy first dimension first,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	139 second dimension second, etc...
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	140 Set up the recursive loop backwards so that final
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	141 dimension is actually copied last.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	142 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	143 outstride = itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	144 for (k=1; k<nd-1;k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	145 outstride *= shape[k];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	146 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	147 for (k=0; k<shape[nd-1]; k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	148 _strided_copy_nd(dest, src, nd-1, shape,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	149 strides, itemsize, fort);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	150 dest += outstride;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	151 src += strides[nd-1];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	152 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	153 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	154
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	155 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	156 /* Copy last dimension first,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	157 second-to-last dimension second, etc.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	158 Set up the recursion so that the
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	159 first dimension is copied last
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	160 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	161 outstride = itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	162 for (k=1; k < nd; k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	163 outstride *= shape[k];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	164 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	165 for (k=0; k<shape[0]; k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	166 _strided_copy_nd(dest, src, nd-1, shape+1,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	167 strides+1, itemsize,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	168 fort);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	169 dest += outstride;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	170 src += strides[0];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	171 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	172 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	173 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	174 return;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	175 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	176
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	177 void _add_one_to_index_F(int nd, Py_ssize_t index, Py_ssize_t shape);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	178 void _add_one_to_index_C(int nd, Py_ssize_t index, Py_ssize_t shape);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	179
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	180 static int
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	181 _indirect_copy_nd(char dest, Py_buffer view, char fort)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	182 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	183 Py_ssize_t *indices;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	184 int k;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	185 Py_ssize_t elements;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	186 char *ptr;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	187 void (func)(int, Py_ssize_t , Py_ssize_t *);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	188
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	189 if (view->ndim > PY_SSIZE_T_MAX / sizeof(Py_ssize_t)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	190 PyErr_NoMemory();
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	191 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	192 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	193
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	194 indices = (Py_ssize_t )PyMem_Malloc(sizeof(Py_ssize_t)view->ndim);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	195 if (indices == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	196 PyErr_NoMemory();
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	197 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	198 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	199 for (k=0; k<view->ndim;k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	200 indices[k] = 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	201 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	202
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	203 elements = 1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	204 for (k=0; k<view->ndim; k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	205 elements *= view->shape[k];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	206 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	207 if (fort == 'F') {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	208 func = _add_one_to_index_F;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	209 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	210 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	211 func = _add_one_to_index_C;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	212 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	213 while (elements--) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	214 func(view->ndim, indices, view->shape);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	215 ptr = PyBuffer_GetPointer(view, indices);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	216 memcpy(dest, ptr, view->itemsize);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	217 dest += view->itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	218 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	219
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	220 PyMem_Free(indices);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	221 return 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	222 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	223
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	224 /*
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	225 Get a the data from an object as a contiguous chunk of memory (in
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	226 either 'C' or 'F'ortran order) even if it means copying it into a
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	227 separate memory area.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	228
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	229 Returns a new reference to a Memory view object. If no copy is needed,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	230 the memory view object points to the original memory and holds a
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	231 lock on the original. If a copy is needed, then the memory view object
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	232 points to a brand-new Bytes object (and holds a memory lock on it).
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	233
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	234 buffertype
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	235
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	236 PyBUF_READ buffer only needs to be read-only
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	237 PyBUF_WRITE buffer needs to be writable (give error if not contiguous)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	238 PyBUF_SHADOW buffer needs to be writable so shadow it with
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	239 a contiguous buffer if it is not. The view will point to
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	240 the shadow buffer which can be written to and then
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	241 will be copied back into the other buffer when the memory
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	242 view is de-allocated. While the shadow buffer is
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	243 being used, it will have an exclusive write lock on
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	244 the original buffer.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	245 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	246
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	247 PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	248 PyMemoryView_GetContiguous(PyObject *obj, int buffertype, char fort)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	249 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	250 PyMemoryViewObject *mem;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	251 PyObject *bytes;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	252 Py_buffer *view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	253 int flags;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	254 char *dest;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	255
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	256 if (!PyObject_CheckBuffer(obj)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	257 PyErr_SetString(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	258 "object does not have the buffer interface");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	259 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	260 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	261
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	262 mem = PyObject_GC_New(PyMemoryViewObject, &PyMemoryView_Type);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	263 if (mem == NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	264 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	265
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	266 view = &mem->view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	267 flags = PyBUF_FULL_RO;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	268 switch(buffertype) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	269 case PyBUF_WRITE:
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	270 flags = PyBUF_FULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	271 break;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	272 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	273
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	274 if (PyObject_GetBuffer(obj, view, flags) != 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	275 Py_DECREF(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	276 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	277 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	278
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	279 if (PyBuffer_IsContiguous(view, fort)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	280 /* no copy needed */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	281 Py_INCREF(obj);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	282 mem->base = obj;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	283 _PyObject_GC_TRACK(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	284 return (PyObject *)mem;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	285 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	286 /* otherwise a copy is needed */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	287 if (buffertype == PyBUF_WRITE) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	288 Py_DECREF(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	289 PyErr_SetString(PyExc_BufferError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	290 "writable contiguous buffer requested "
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	291 "for a non-contiguousobject.");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	292 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	293 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	294 bytes = PyBytes_FromStringAndSize(NULL, view->len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	295 if (bytes == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	296 Py_DECREF(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	297 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	298 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	299 dest = PyBytes_AS_STRING(bytes);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	300 /* different copying strategy depending on whether
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	301 or not any pointer de-referencing is needed
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	302 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	303 /* strided or in-direct copy */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	304 if (view->suboffsets==NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	305 _strided_copy_nd(dest, view->buf, view->ndim, view->shape,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	306 view->strides, view->itemsize, fort);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	307 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	308 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	309 if (_indirect_copy_nd(dest, view, fort) < 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	310 Py_DECREF(bytes);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	311 Py_DECREF(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	312 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	313 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	314 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	315 if (buffertype == PyBUF_SHADOW) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	316 /* return a shadowed memory-view object */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	317 view->buf = dest;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	318 mem->base = PyTuple_Pack(2, obj, bytes);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	319 Py_DECREF(bytes);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	320 if (mem->base == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	321 Py_DECREF(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	322 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	323 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	324 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	325 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	326 PyBuffer_Release(view); /* XXX ? */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	327 /* steal the reference */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	328 mem->base = bytes;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	329 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	330 _PyObject_GC_TRACK(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	331 return (PyObject *)mem;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	332 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	333
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	334
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	335 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	336 memory_format_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	337 {
56543 1617c11556af Issue #7142: Fix uses of unicode in memoryview objects Mark Dickinson <dickinsm@gmail.com> parents: 54829 diff changeset	338 return PyString_FromString(self->view.format);
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	339 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	340
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	341 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	342 memory_itemsize_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	343 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	344 return PyLong_FromSsize_t(self->view.itemsize);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	345 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	346
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	347 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	348 _IntTupleFromSsizet(int len, Py_ssize_t *vals)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	349 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	350 int i;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	351 PyObject *o;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	352 PyObject *intTuple;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	353
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	354 if (vals == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	355 Py_INCREF(Py_None);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	356 return Py_None;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	357 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	358 intTuple = PyTuple_New(len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	359 if (!intTuple) return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	360 for(i=0; i<len; i++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	361 o = PyLong_FromSsize_t(vals[i]);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	362 if (!o) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	363 Py_DECREF(intTuple);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	364 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	365 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	366 PyTuple_SET_ITEM(intTuple, i, o);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	367 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	368 return intTuple;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	369 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	370
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	371 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	372 memory_shape_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	373 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	374 return _IntTupleFromSsizet(self->view.ndim, self->view.shape);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	375 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	376
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	377 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	378 memory_strides_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	379 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	380 return _IntTupleFromSsizet(self->view.ndim, self->view.strides);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	381 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	382
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	383 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	384 memory_suboffsets_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	385 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	386 return _IntTupleFromSsizet(self->view.ndim, self->view.suboffsets);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	387 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	388
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	389 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	390 memory_readonly_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	391 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	392 return PyBool_FromLong(self->view.readonly);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	393 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	394
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	395 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	396 memory_ndim_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	397 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	398 return PyLong_FromLong(self->view.ndim);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	399 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	400
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	401 static PyGetSetDef memory_getsetlist[] ={
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	402 {"format", (getter)memory_format_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	403 {"itemsize", (getter)memory_itemsize_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	404 {"shape", (getter)memory_shape_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	405 {"strides", (getter)memory_strides_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	406 {"suboffsets", (getter)memory_suboffsets_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	407 {"readonly", (getter)memory_readonly_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	408 {"ndim", (getter)memory_ndim_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	409 {NULL, NULL, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	410 };
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	411
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	412
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	413 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	414 memory_tobytes(PyMemoryViewObject self, PyObject noargs)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	415 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	416 Py_buffer view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	417 PyObject *res;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	418
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	419 if (PyObject_GetBuffer((PyObject *)self, &view, PyBUF_FULL) < 0)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	420 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	421
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	422 res = PyBytes_FromStringAndSize(NULL, view.len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	423 PyBuffer_ToContiguous(PyBytes_AS_STRING(res), &view, view.len, 'C');
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	424 PyBuffer_Release(&view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	425 return res;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	426 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	427
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	428 /* TODO: rewrite this function using the struct module to unpack
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	429 each buffer item */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	430
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	431 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	432 memory_tolist(PyMemoryViewObject mem, PyObject noargs)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	433 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	434 Py_buffer *view = &(mem->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	435 Py_ssize_t i;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	436 PyObject res, item;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	437 char *buf;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	438
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	439 if (strcmp(view->format, "B") \|\| view->itemsize != 1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	440 PyErr_SetString(PyExc_NotImplementedError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	441 "tolist() only supports byte views");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	442 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	443 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	444 if (view->ndim != 1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	445 PyErr_SetString(PyExc_NotImplementedError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	446 "tolist() only supports one-dimensional objects");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	447 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	448 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	449 res = PyList_New(view->len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	450 if (res == NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	451 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	452 buf = view->buf;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	453 for (i = 0; i < view->len; i++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	454 item = PyInt_FromLong((unsigned char) *buf);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	455 if (item == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	456 Py_DECREF(res);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	457 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	458 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	459 PyList_SET_ITEM(res, i, item);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	460 buf++;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	461 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	462 return res;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	463 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	464
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	465 static PyMethodDef memory_methods[] = {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	466 {"tobytes", (PyCFunction)memory_tobytes, METH_NOARGS, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	467 {"tolist", (PyCFunction)memory_tolist, METH_NOARGS, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	468 {NULL, NULL} /* sentinel */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	469 };
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	470
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	471
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	472 static void
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	473 memory_dealloc(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	474 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	475 _PyObject_GC_UNTRACK(self);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	476 if (self->view.obj != NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	477 if (self->base && PyTuple_Check(self->base)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	478 /* Special case when first element is generic object
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	479 with buffer interface and the second element is a
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	480 contiguous "shadow" that must be copied back into
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	481 the data areay of the first tuple element before
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	482 releasing the buffer on the first element.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	483 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	484
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	485 PyObject_CopyData(PyTuple_GET_ITEM(self->base,0),
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	486 PyTuple_GET_ITEM(self->base,1));
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	487
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	488 /* The view member should have readonly == -1 in
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	489 this instance indicating that the memory can
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	490 be "locked" and was locked and will be unlocked
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	491 again after this call.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	492 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	493 PyBuffer_Release(&(self->view));
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	494 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	495 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	496 PyBuffer_Release(&(self->view));
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	497 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	498 Py_CLEAR(self->base);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	499 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	500 PyObject_GC_Del(self);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	501 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	502
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	503 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	504 memory_repr(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	505 {
56543 1617c11556af Issue #7142: Fix uses of unicode in memoryview objects Mark Dickinson <dickinsm@gmail.com> parents: 54829 diff changeset	506 return PyString_FromFormat("<memory at %p>", self);
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	507 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	508
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	509 /* Sequence methods */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	510 static Py_ssize_t
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	511 memory_length(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	512 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	513 return get_shape0(&self->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	514 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	515
54829 f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	516 /* Alternate version of memory_subcript that only accepts indices.
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	517 Used by PySeqIter_New().
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	518 */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	519 static PyObject *
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	520 memory_item(PyMemoryViewObject *self, Py_ssize_t result)
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	521 {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	522 Py_buffer *view = &(self->view);
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	523
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	524 if (view->ndim == 0) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	525 PyErr_SetString(PyExc_IndexError,
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	526 "invalid indexing of 0-dim memory");
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	527 return NULL;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	528 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	529 if (view->ndim == 1) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	530 /* Return a bytes object */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	531 char *ptr;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	532 ptr = (char *)view->buf;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	533 if (result < 0) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	534 result += get_shape0(view);
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	535 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	536 if ((result < 0) \|\| (result >= get_shape0(view))) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	537 PyErr_SetString(PyExc_IndexError,
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	538 "index out of bounds");
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	539 return NULL;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	540 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	541 if (view->strides == NULL)
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	542 ptr += view->itemsize * result;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	543 else
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	544 ptr += view->strides[0] * result;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	545 if (view->suboffsets != NULL &&
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	546 view->suboffsets[0] >= 0) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	547 ptr = ((char *)ptr) + view->suboffsets[0];
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	548 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	549 return PyBytes_FromStringAndSize(ptr, view->itemsize);
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	550 } else {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	551 /* Return a new memory-view object */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	552 Py_buffer newview;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	553 memset(&newview, 0, sizeof(newview));
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	554 /* XXX: This needs to be fixed so it actually returns a sub-view */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	555 return PyMemoryView_FromBuffer(&newview);
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	556 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	557 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	558
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	559 /*
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	560 mem[obj] returns a bytes object holding the data for one element if
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	561 obj fully indexes the memory view or another memory-view object
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	562 if it does not.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	563
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	564 0-d memory-view objects can be referenced using ... or () but
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	565 not with anything else.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	566 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	567 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	568 memory_subscript(PyMemoryViewObject self, PyObject key)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	569 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	570 Py_buffer *view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	571 view = &(self->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	572
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	573 if (view->ndim == 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	574 if (key == Py_Ellipsis \|\|
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	575 (PyTuple_Check(key) && PyTuple_GET_SIZE(key)==0)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	576 Py_INCREF(self);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	577 return (PyObject *)self;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	578 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	579 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	580 PyErr_SetString(PyExc_IndexError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	581 "invalid indexing of 0-dim memory");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	582 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	583 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	584 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	585 if (PyIndex_Check(key)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	586 Py_ssize_t result;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	587 result = PyNumber_AsSsize_t(key, NULL);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	588 if (result == -1 && PyErr_Occurred())
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	589 return NULL;
54829 f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	590 return memory_item(self, result);
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	591 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	592 else if (PySlice_Check(key)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	593 Py_ssize_t start, stop, step, slicelength;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	594
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	595 if (PySlice_GetIndicesEx((PySliceObject*)key, get_shape0(view),
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	596 &start, &stop, &step, &slicelength) < 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	597 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	598 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	599
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	600 if (step == 1 && view->ndim == 1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	601 Py_buffer newview;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	602 void newbuf = (char ) view->buf
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	603 + start * view->itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	604 int newflags = view->readonly
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	605 ? PyBUF_CONTIG_RO : PyBUF_CONTIG;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	606
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	607 /* XXX There should be an API to create a subbuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	608 if (view->obj != NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	609 if (PyObject_GetBuffer(view->obj, &newview, newflags) == -1)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	610 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	611 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	612 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	613 newview = *view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	614 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	615 newview.buf = newbuf;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	616 newview.len = slicelength * newview.itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	617 newview.format = view->format;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	618 newview.shape = &(newview.smalltable[0]);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	619 newview.shape[0] = slicelength;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	620 newview.strides = &(newview.itemsize);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	621 return PyMemoryView_FromBuffer(&newview);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	622 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	623 PyErr_SetNone(PyExc_NotImplementedError);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	624 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	625 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	626 PyErr_Format(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	627 "cannot index memory using \"%.200s\"",
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	628 key->ob_type->tp_name);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	629 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	630 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	631
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	632
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	633 /* Need to support assigning memory if we can */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	634 static int
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	635 memory_ass_sub(PyMemoryViewObject self, PyObject key, PyObject *value)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	636 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	637 Py_ssize_t start, len, bytelen, i;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	638 Py_buffer srcview;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	639 Py_buffer *view = &(self->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	640 char srcbuf, destbuf;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	641
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	642 if (view->readonly) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	643 PyErr_SetString(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	644 "cannot modify read-only memory");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	645 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	646 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	647 if (view->ndim != 1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	648 PyErr_SetNone(PyExc_NotImplementedError);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	649 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	650 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	651 if (PyIndex_Check(key)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	652 start = PyNumber_AsSsize_t(key, NULL);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	653 if (start == -1 && PyErr_Occurred())
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	654 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	655 if (start < 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	656 start += get_shape0(view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	657 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	658 if ((start < 0) \|\| (start >= get_shape0(view))) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	659 PyErr_SetString(PyExc_IndexError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	660 "index out of bounds");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	661 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	662 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	663 len = 1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	664 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	665 else if (PySlice_Check(key)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	666 Py_ssize_t stop, step;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	667
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	668 if (PySlice_GetIndicesEx((PySliceObject*)key, get_shape0(view),
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	669 &start, &stop, &step, &len) < 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	670 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	671 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	672 if (step != 1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	673 PyErr_SetNone(PyExc_NotImplementedError);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	674 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	675 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	676 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	677 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	678 PyErr_Format(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	679 "cannot index memory using \"%.200s\"",
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	680 key->ob_type->tp_name);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	681 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	682 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	683 if (PyObject_GetBuffer(value, &srcview, PyBUF_CONTIG_RO) == -1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	684 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	685 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	686 /* XXX should we allow assignment of different item sizes
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	687 as long as the byte length is the same?
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	688 (e.g. assign 2 shorts to a 4-byte slice) */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	689 if (srcview.itemsize != view->itemsize) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	690 PyErr_Format(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	691 "mismatching item sizes for \"%.200s\" and \"%.200s\"",
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	692 view->obj->ob_type->tp_name, srcview.obj->ob_type->tp_name);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	693 goto _error;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	694 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	695 bytelen = len * view->itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	696 if (bytelen != srcview.len) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	697 PyErr_SetString(PyExc_ValueError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	698 "cannot modify size of memoryview object");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	699 goto _error;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	700 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	701 /* Do the actual copy */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	702 destbuf = (char ) view->buf + start view->itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	703 srcbuf = (char *) srcview.buf;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	704 if (destbuf + bytelen < srcbuf \|\| srcbuf + bytelen < destbuf)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	705 /* No overlapping */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	706 memcpy(destbuf, srcbuf, bytelen);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	707 else if (destbuf < srcbuf) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	708 /* Copy in ascending order */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	709 for (i = 0; i < bytelen; i++)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	710 destbuf[i] = srcbuf[i];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	711 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	712 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	713 /* Copy in descencing order */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	714 for (i = bytelen - 1; i >= 0; i--)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	715 destbuf[i] = srcbuf[i];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	716 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	717
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	718 PyBuffer_Release(&srcview);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	719 return 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	720
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	721 _error:
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	722 PyBuffer_Release(&srcview);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	723 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	724 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	725
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	726 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	727 memory_richcompare(PyObject v, PyObject w, int op)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	728 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	729 Py_buffer vv, ww;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	730 int equal = 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	731 PyObject *res;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	732
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	733 vv.obj = NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	734 ww.obj = NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	735 if (op != Py_EQ && op != Py_NE)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	736 goto _notimpl;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	737 if (PyObject_GetBuffer(v, &vv, PyBUF_CONTIG_RO) == -1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	738 PyErr_Clear();
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	739 goto _notimpl;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	740 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	741 if (PyObject_GetBuffer(w, &ww, PyBUF_CONTIG_RO) == -1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	742 PyErr_Clear();
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	743 goto _notimpl;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	744 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	745
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	746 if (vv.itemsize != ww.itemsize \|\| vv.len != ww.len)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	747 goto _end;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	748
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	749 equal = !memcmp(vv.buf, ww.buf, vv.len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	750
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	751 _end:
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	752 PyBuffer_Release(&vv);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	753 PyBuffer_Release(&ww);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	754 if ((equal && op == Py_EQ) \|\| (!equal && op == Py_NE))
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	755 res = Py_True;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	756 else
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	757 res = Py_False;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	758 Py_INCREF(res);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	759 return res;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	760
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	761 _notimpl:
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	762 PyBuffer_Release(&vv);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	763 PyBuffer_Release(&ww);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	764 Py_INCREF(Py_NotImplemented);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	765 return Py_NotImplemented;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	766 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	767
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	768
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	769 static int
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	770 memory_traverse(PyMemoryViewObject self, visitproc visit, void arg)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	771 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	772 if (self->base != NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	773 Py_VISIT(self->base);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	774 if (self->view.obj != NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	775 Py_VISIT(self->view.obj);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	776 return 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	777 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	778
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	779 static int
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	780 memory_clear(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	781 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	782 Py_CLEAR(self->base);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	783 PyBuffer_Release(&self->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	784 return 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	785 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	786
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	787
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	788 /* As mapping */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	789 static PyMappingMethods memory_as_mapping = {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	790 (lenfunc)memory_length, /* mp_length */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	791 (binaryfunc)memory_subscript, /* mp_subscript */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	792 (objobjargproc)memory_ass_sub, /* mp_ass_subscript */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	793 };
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	794
54829 f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	795 static PySequenceMethods memory_as_sequence = {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	796 0, /* sq_length */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	797 0, /* sq_concat */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	798 0, /* sq_repeat */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	799 (ssizeargfunc)memory_item, /* sq_item */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	800 };
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	801
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	802 /* Buffer methods */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	803 static PyBufferProcs memory_as_buffer = {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	804 0, /* bf_getreadbuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	805 0, /* bf_getwritebuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	806 0, /* bf_getsegcount */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	807 0, /* bf_getcharbuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	808 (getbufferproc)memory_getbuf, /* bf_getbuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	809 (releasebufferproc)memory_releasebuf, /* bf_releasebuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	810 };
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	811
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	812
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	813 PyTypeObject PyMemoryView_Type = {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	814 PyVarObject_HEAD_INIT(&PyType_Type, 0)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	815 "memoryview",
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	816 sizeof(PyMemoryViewObject),
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	817 0,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	818 (destructor)memory_dealloc, /* tp_dealloc */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	819 0, /* tp_print */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	820 0, /* tp_getattr */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	821 0, /* tp_setattr */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	822 0, /* tp_compare */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	823 (reprfunc)memory_repr, /* tp_repr */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	824 0, /* tp_as_number */
54829 f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	825 &memory_as_sequence, /* tp_as_sequence */
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	826 &memory_as_mapping, /* tp_as_mapping */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	827 0, /* tp_hash */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	828 0, /* tp_call */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	829 0, /* tp_str */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	830 PyObject_GenericGetAttr, /* tp_getattro */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	831 0, /* tp_setattro */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	832 &memory_as_buffer, /* tp_as_buffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	833 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	834 Py_TPFLAGS_HAVE_NEWBUFFER, /* tp_flags */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	835 memory_doc, /* tp_doc */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	836 (traverseproc)memory_traverse, /* tp_traverse */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	837 (inquiry)memory_clear, /* tp_clear */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	838 memory_richcompare, /* tp_richcompare */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	839 0, /* tp_weaklistoffset */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	840 0, /* tp_iter */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	841 0, /* tp_iternext */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	842 memory_methods, /* tp_methods */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	843 0, /* tp_members */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	844 memory_getsetlist, /* tp_getset */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	845 0, /* tp_base */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	846 0, /* tp_dict */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	847 0, /* tp_descr_get */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	848 0, /* tp_descr_set */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	849 0, /* tp_dictoffset */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	850 0, /* tp_init */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	851 0, /* tp_alloc */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	852 memory_new, /* tp_new */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	853 };

Mercurial > cpython-lru_cache-weakref

annotate Objects/memoryobject.c @ 57303:4aa4c19dfaed legacy-trunk