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

annotate Objects/memoryobject.c @ 65035:87f4b32905df 2.7

Cosmetic fix to use print function.

author	Hirokazu Yamamoto <ocean-city@m2.ccsnet.ne.jp>
date	Fri, 01 Oct 2010 10:48:47 +0000
parents	fa4d78b9e3a2
children	eac5457ff195

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;
58643 c37f99b54d7c Issue #7385: Fix a crash in `MemoryView_FromObject` when Antoine Pitrou <solipsis@pitrou.net> parents: 56543 diff changeset	79 Py_buffer view;
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	80
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	81 if (!PyObject_CheckBuffer(base)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	82 PyErr_SetString(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	83 "cannot make memory view because object does "
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	84 "not have the buffer interface");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	85 return NULL;
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
58643 c37f99b54d7c Issue #7385: Fix a crash in `MemoryView_FromObject` when Antoine Pitrou <solipsis@pitrou.net> parents: 56543 diff changeset	88 if (PyObject_GetBuffer(base, &view, PyBUF_FULL_RO) < 0)
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	89 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	90
58643 c37f99b54d7c Issue #7385: Fix a crash in `MemoryView_FromObject` when Antoine Pitrou <solipsis@pitrou.net> parents: 56543 diff changeset	91 mview = (PyMemoryViewObject *)PyMemoryView_FromBuffer(&view);
c37f99b54d7c Issue #7385: Fix a crash in `MemoryView_FromObject` when Antoine Pitrou <solipsis@pitrou.net> parents: 56543 diff changeset	92 if (mview == NULL) {
c37f99b54d7c Issue #7385: Fix a crash in `MemoryView_FromObject` when Antoine Pitrou <solipsis@pitrou.net> parents: 56543 diff changeset	93 PyBuffer_Release(&view);
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	94 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	95 }
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 mview->base = base;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	98 Py_INCREF(base);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	99 return (PyObject *)mview;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	100 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	101
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	102 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	103 memory_new(PyTypeObject subtype, PyObject args, PyObject *kwds)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	104 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	105 PyObject *obj;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	106 static char *kwlist[] = {"object", 0};
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	107
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	108 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O:memoryview", kwlist,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	109 &obj)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	110 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	111 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	112
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	113 return PyMemoryView_FromObject(obj);
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
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 static void
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	118 _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	119 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	120 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	121 int k;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	122 Py_ssize_t outstride;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	123
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	124 if (nd==0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	125 memcpy(dest, src, itemsize);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	126 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	127 else if (nd == 1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	128 for (k = 0; k<shape[0]; k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	129 memcpy(dest, src, itemsize);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	130 dest += itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	131 src += strides[0];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	132 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	133 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	134 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	135 if (fort == 'F') {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	136 /* Copy first dimension first,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	137 second dimension second, etc...
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	138 Set up the recursive loop backwards so that final
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	139 dimension is actually copied last.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	140 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	141 outstride = itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	142 for (k=1; k<nd-1;k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	143 outstride *= shape[k];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	144 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	145 for (k=0; k<shape[nd-1]; k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	146 _strided_copy_nd(dest, src, nd-1, shape,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	147 strides, itemsize, fort);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	148 dest += outstride;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	149 src += strides[nd-1];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	150 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	151 }
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 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	154 /* Copy last dimension first,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	155 second-to-last dimension second, etc.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	156 Set up the recursion so that the
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	157 first dimension is copied last
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	158 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	159 outstride = itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	160 for (k=1; k < nd; k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	161 outstride *= shape[k];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	162 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	163 for (k=0; k<shape[0]; k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	164 _strided_copy_nd(dest, src, nd-1, shape+1,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	165 strides+1, itemsize,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	166 fort);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	167 dest += outstride;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	168 src += strides[0];
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	169 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	170 }
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 return;
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	175 static int
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	176 _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	177 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	178 Py_ssize_t *indices;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	179 int k;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	180 Py_ssize_t elements;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	181 char *ptr;
64363 fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 64345 diff changeset	182 void (func)(int, Py_ssize_t , const Py_ssize_t *);
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	183
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	184 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	185 PyErr_NoMemory();
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	186 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	187 }
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 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	190 if (indices == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	191 PyErr_NoMemory();
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	192 return -1;
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 for (k=0; k<view->ndim;k++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	195 indices[k] = 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	196 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	197
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	198 elements = 1;
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 elements *= view->shape[k];
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 if (fort == 'F') {
64345 da19d02aaaaa Merged revisions 84391 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 62965 diff changeset	203 func = _Py_add_one_to_index_F;
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	204 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	205 else {
64345 da19d02aaaaa Merged revisions 84391 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 62965 diff changeset	206 func = _Py_add_one_to_index_C;
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	207 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	208 while (elements--) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	209 func(view->ndim, indices, view->shape);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	210 ptr = PyBuffer_GetPointer(view, indices);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	211 memcpy(dest, ptr, view->itemsize);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	212 dest += view->itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	213 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	214
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	215 PyMem_Free(indices);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	216 return 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	217 }
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 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	221 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	222 separate memory area.
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 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	225 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	226 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	227 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	228
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	229 buffertype
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	230
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	231 PyBUF_READ buffer only needs to be read-only
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	232 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	233 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	234 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	235 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	236 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	237 view is de-allocated. While the shadow buffer is
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	238 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	239 the original buffer.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	240 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	241
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	242 PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	243 PyMemoryView_GetContiguous(PyObject *obj, int buffertype, char fort)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	244 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	245 PyMemoryViewObject *mem;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	246 PyObject *bytes;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	247 Py_buffer *view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	248 int flags;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	249 char *dest;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	250
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	251 if (!PyObject_CheckBuffer(obj)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	252 PyErr_SetString(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	253 "object does not have the buffer interface");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	254 return NULL;
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	257 mem = PyObject_GC_New(PyMemoryViewObject, &PyMemoryView_Type);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	258 if (mem == NULL)
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 view = &mem->view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	262 flags = PyBUF_FULL_RO;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	263 switch(buffertype) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	264 case PyBUF_WRITE:
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	265 flags = PyBUF_FULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	266 break;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	267 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	268
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	269 if (PyObject_GetBuffer(obj, view, flags) != 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	270 Py_DECREF(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	271 return NULL;
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 (PyBuffer_IsContiguous(view, fort)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	275 /* no copy needed */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	276 Py_INCREF(obj);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	277 mem->base = obj;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	278 _PyObject_GC_TRACK(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	279 return (PyObject *)mem;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	280 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	281 /* otherwise a copy is needed */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	282 if (buffertype == PyBUF_WRITE) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	283 Py_DECREF(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	284 PyErr_SetString(PyExc_BufferError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	285 "writable contiguous buffer requested "
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	286 "for a non-contiguousobject.");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	287 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	288 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	289 bytes = PyBytes_FromStringAndSize(NULL, view->len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	290 if (bytes == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	291 Py_DECREF(mem);
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 dest = PyBytes_AS_STRING(bytes);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	295 /* different copying strategy depending on whether
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	296 or not any pointer de-referencing is needed
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	297 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	298 /* strided or in-direct copy */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	299 if (view->suboffsets==NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	300 _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	301 view->strides, view->itemsize, fort);
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 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	304 if (_indirect_copy_nd(dest, view, fort) < 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	305 Py_DECREF(bytes);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	306 Py_DECREF(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	307 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	308 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	309 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	310 if (buffertype == PyBUF_SHADOW) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	311 /* return a shadowed memory-view object */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	312 view->buf = dest;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	313 mem->base = PyTuple_Pack(2, obj, bytes);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	314 Py_DECREF(bytes);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	315 if (mem->base == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	316 Py_DECREF(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	317 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	318 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	319 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	320 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	321 PyBuffer_Release(view); /* XXX ? */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	322 /* steal the reference */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	323 mem->base = bytes;
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 _PyObject_GC_TRACK(mem);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	326 return (PyObject *)mem;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	327 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	328
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 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	331 memory_format_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	332 {
56543 1617c11556af Issue #7142: Fix uses of unicode in memoryview objects Mark Dickinson <dickinsm@gmail.com> parents: 54829 diff changeset	333 return PyString_FromString(self->view.format);
52749 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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	336 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	337 memory_itemsize_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	338 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	339 return PyLong_FromSsize_t(self->view.itemsize);
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	342 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	343 _IntTupleFromSsizet(int len, Py_ssize_t *vals)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	344 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	345 int i;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	346 PyObject *o;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	347 PyObject *intTuple;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	348
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	349 if (vals == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	350 Py_INCREF(Py_None);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	351 return Py_None;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	352 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	353 intTuple = PyTuple_New(len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	354 if (!intTuple) return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	355 for(i=0; i<len; i++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	356 o = PyLong_FromSsize_t(vals[i]);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	357 if (!o) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	358 Py_DECREF(intTuple);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	359 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	360 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	361 PyTuple_SET_ITEM(intTuple, i, o);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	362 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	363 return intTuple;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	364 }
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 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	367 memory_shape_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	368 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	369 return _IntTupleFromSsizet(self->view.ndim, self->view.shape);
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	372 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	373 memory_strides_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	374 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	375 return _IntTupleFromSsizet(self->view.ndim, self->view.strides);
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	378 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	379 memory_suboffsets_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	380 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	381 return _IntTupleFromSsizet(self->view.ndim, self->view.suboffsets);
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	384 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	385 memory_readonly_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	386 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	387 return PyBool_FromLong(self->view.readonly);
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	390 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	391 memory_ndim_get(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	392 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	393 return PyLong_FromLong(self->view.ndim);
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	396 static PyGetSetDef memory_getsetlist[] ={
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	397 {"format", (getter)memory_format_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	398 {"itemsize", (getter)memory_itemsize_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	399 {"shape", (getter)memory_shape_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	400 {"strides", (getter)memory_strides_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	401 {"suboffsets", (getter)memory_suboffsets_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	402 {"readonly", (getter)memory_readonly_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	403 {"ndim", (getter)memory_ndim_get, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	404 {NULL, NULL, NULL, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	405 };
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	406
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	407
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	408 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	409 memory_tobytes(PyMemoryViewObject self, PyObject noargs)
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 Py_buffer view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	412 PyObject *res;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	413
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	414 if (PyObject_GetBuffer((PyObject *)self, &view, PyBUF_FULL) < 0)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	415 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	416
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	417 res = PyBytes_FromStringAndSize(NULL, view.len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	418 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	419 PyBuffer_Release(&view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	420 return res;
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	423 /* 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	424 each buffer item */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	425
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	426 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	427 memory_tolist(PyMemoryViewObject mem, PyObject noargs)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	428 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	429 Py_buffer *view = &(mem->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	430 Py_ssize_t i;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	431 PyObject res, item;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	432 char *buf;
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 if (strcmp(view->format, "B") \|\| view->itemsize != 1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	435 PyErr_SetString(PyExc_NotImplementedError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	436 "tolist() only supports byte views");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	437 return NULL;
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 (view->ndim != 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 one-dimensional objects");
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 res = PyList_New(view->len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	445 if (res == NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	446 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	447 buf = view->buf;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	448 for (i = 0; i < view->len; i++) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	449 item = PyInt_FromLong((unsigned char) *buf);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	450 if (item == NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	451 Py_DECREF(res);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	452 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	453 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	454 PyList_SET_ITEM(res, i, item);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	455 buf++;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	456 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	457 return res;
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	460 static PyMethodDef memory_methods[] = {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	461 {"tobytes", (PyCFunction)memory_tobytes, METH_NOARGS, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	462 {"tolist", (PyCFunction)memory_tolist, METH_NOARGS, NULL},
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	463 {NULL, NULL} /* sentinel */
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	466
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	467 static void
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	468 memory_dealloc(PyMemoryViewObject *self)
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 _PyObject_GC_UNTRACK(self);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	471 if (self->view.obj != NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	472 if (self->base && PyTuple_Check(self->base)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	473 /* Special case when first element is generic object
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	474 with buffer interface and the second element is a
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	475 contiguous "shadow" that must be copied back into
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	476 the data areay of the first tuple element before
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	477 releasing the buffer on the first element.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	478 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	479
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	480 PyObject_CopyData(PyTuple_GET_ITEM(self->base,0),
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	481 PyTuple_GET_ITEM(self->base,1));
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	482
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	483 /* The view member should have readonly == -1 in
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	484 this instance indicating that the memory can
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	485 be "locked" and was locked and will be unlocked
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	486 again after this call.
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 PyBuffer_Release(&(self->view));
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	489 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	490 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	491 PyBuffer_Release(&(self->view));
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 Py_CLEAR(self->base);
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 PyObject_GC_Del(self);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	496 }
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 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	499 memory_repr(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	500 {
56543 1617c11556af Issue #7142: Fix uses of unicode in memoryview objects Mark Dickinson <dickinsm@gmail.com> parents: 54829 diff changeset	501 return PyString_FromFormat("<memory at %p>", self);
52749 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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	504 /* Sequence methods */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	505 static Py_ssize_t
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	506 memory_length(PyMemoryViewObject *self)
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 return get_shape0(&self->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	509 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	510
54829 f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	511 /* 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	512 Used by PySeqIter_New().
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	513 */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	514 static PyObject *
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	515 memory_item(PyMemoryViewObject *self, Py_ssize_t result)
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	516 {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	517 Py_buffer *view = &(self->view);
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 if (view->ndim == 0) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	520 PyErr_SetString(PyExc_IndexError,
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	521 "invalid indexing of 0-dim memory");
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	522 return NULL;
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 == 1) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	525 /* Return a bytes object */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	526 char *ptr;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	527 ptr = (char *)view->buf;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	528 if (result < 0) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	529 result += get_shape0(view);
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	530 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	531 if ((result < 0) \|\| (result >= get_shape0(view))) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	532 PyErr_SetString(PyExc_IndexError,
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	533 "index out of bounds");
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	534 return NULL;
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 (view->strides == NULL)
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	537 ptr += view->itemsize * result;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	538 else
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	539 ptr += view->strides[0] * result;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	540 if (view->suboffsets != NULL &&
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	541 view->suboffsets[0] >= 0) {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	542 ptr = ((char *)ptr) + view->suboffsets[0];
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	543 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	544 return PyBytes_FromStringAndSize(ptr, view->itemsize);
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	545 } else {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	546 /* Return a new memory-view object */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	547 Py_buffer newview;
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	548 memset(&newview, 0, sizeof(newview));
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	549 /* 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	550 return PyMemoryView_FromBuffer(&newview);
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	551 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	552 }
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	553
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	554 /*
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	555 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	556 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	557 if it does not.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	558
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	559 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	560 not with anything else.
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	561 */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	562 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	563 memory_subscript(PyMemoryViewObject self, PyObject key)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	564 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	565 Py_buffer *view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	566 view = &(self->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	567
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	568 if (view->ndim == 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	569 if (key == Py_Ellipsis \|\|
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	570 (PyTuple_Check(key) && PyTuple_GET_SIZE(key)==0)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	571 Py_INCREF(self);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	572 return (PyObject *)self;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	573 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	574 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	575 PyErr_SetString(PyExc_IndexError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	576 "invalid indexing of 0-dim memory");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	577 return NULL;
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 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	580 if (PyIndex_Check(key)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	581 Py_ssize_t result;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	582 result = PyNumber_AsSsize_t(key, NULL);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	583 if (result == -1 && PyErr_Occurred())
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	584 return NULL;
54829 f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	585 return memory_item(self, result);
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	586 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	587 else if (PySlice_Check(key)) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	588 Py_ssize_t start, stop, step, slicelength;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	589
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	590 if (PySlice_GetIndicesEx((PySliceObject*)key, get_shape0(view),
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	591 &start, &stop, &step, &slicelength) < 0) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	592 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	593 }
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 (step == 1 && view->ndim == 1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	596 Py_buffer newview;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	597 void newbuf = (char ) view->buf
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	598 + start * view->itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	599 int newflags = view->readonly
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	600 ? PyBUF_CONTIG_RO : PyBUF_CONTIG;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	601
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	602 /* 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	603 if (view->obj != NULL) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	604 if (PyObject_GetBuffer(view->obj, &newview, newflags) == -1)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	605 return NULL;
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 else {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	608 newview = *view;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	609 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	610 newview.buf = newbuf;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	611 newview.len = slicelength * newview.itemsize;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	612 newview.format = view->format;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	613 newview.shape = &(newview.smalltable[0]);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	614 newview.shape[0] = slicelength;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	615 newview.strides = &(newview.itemsize);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	616 return PyMemoryView_FromBuffer(&newview);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	617 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	618 PyErr_SetNone(PyExc_NotImplementedError);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	619 return NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	620 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	621 PyErr_Format(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	622 "cannot index memory using \"%.200s\"",
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	623 key->ob_type->tp_name);
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	627
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	628 /* Need to support assigning memory if we can */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	629 static int
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	630 memory_ass_sub(PyMemoryViewObject self, PyObject key, PyObject *value)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	631 {
62965 b1438d26c167 Merged revisions 82814 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 58643 diff changeset	632 Py_ssize_t start, len, bytelen;
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	633 Py_buffer srcview;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	634 Py_buffer *view = &(self->view);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	635 char srcbuf, destbuf;
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 if (view->readonly) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	638 PyErr_SetString(PyExc_TypeError,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	639 "cannot modify read-only memory");
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	640 return -1;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	641 }
64363 fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 64345 diff changeset	642 if (value == NULL) {
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 64345 diff changeset	643 PyErr_SetString(PyExc_TypeError,
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 64345 diff changeset	644 "cannot delete memory");
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 64345 diff changeset	645 return -1;
fa4d78b9e3a2 Merged revisions 84408-84409 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 64345 diff changeset	646 }
52749 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);
62965 b1438d26c167 Merged revisions 82814 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 58643 diff changeset	707 else
b1438d26c167 Merged revisions 82814 via svnmerge from Antoine Pitrou <solipsis@pitrou.net> parents: 58643 diff changeset	708 memmove(destbuf, srcbuf, bytelen);
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	709
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	710 PyBuffer_Release(&srcview);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	711 return 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	712
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	713 _error:
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	714 PyBuffer_Release(&srcview);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	715 return -1;
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 static PyObject *
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	719 memory_richcompare(PyObject v, PyObject w, int op)
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 Py_buffer vv, ww;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	722 int equal = 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	723 PyObject *res;
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 vv.obj = NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	726 ww.obj = NULL;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	727 if (op != Py_EQ && op != Py_NE)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	728 goto _notimpl;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	729 if (PyObject_GetBuffer(v, &vv, PyBUF_CONTIG_RO) == -1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	730 PyErr_Clear();
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	731 goto _notimpl;
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 if (PyObject_GetBuffer(w, &ww, PyBUF_CONTIG_RO) == -1) {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	734 PyErr_Clear();
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	735 goto _notimpl;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	736 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	737
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	738 if (vv.itemsize != ww.itemsize \|\| vv.len != ww.len)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	739 goto _end;
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 equal = !memcmp(vv.buf, ww.buf, vv.len);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	742
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	743 _end:
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	744 PyBuffer_Release(&vv);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	745 PyBuffer_Release(&ww);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	746 if ((equal && op == Py_EQ) \|\| (!equal && op == Py_NE))
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	747 res = Py_True;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	748 else
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	749 res = Py_False;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	750 Py_INCREF(res);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	751 return res;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	752
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	753 _notimpl:
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	754 PyBuffer_Release(&vv);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	755 PyBuffer_Release(&ww);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	756 Py_INCREF(Py_NotImplemented);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	757 return Py_NotImplemented;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	758 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	759
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 static int
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	762 memory_traverse(PyMemoryViewObject self, visitproc visit, void arg)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	763 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	764 if (self->base != NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	765 Py_VISIT(self->base);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	766 if (self->view.obj != NULL)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	767 Py_VISIT(self->view.obj);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	768 return 0;
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	769 }
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	770
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	771 static int
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	772 memory_clear(PyMemoryViewObject *self)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	773 {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	774 Py_CLEAR(self->base);
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	775 PyBuffer_Release(&self->view);
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
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	780 /* As mapping */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	781 static PyMappingMethods memory_as_mapping = {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	782 (lenfunc)memory_length, /* mp_length */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	783 (binaryfunc)memory_subscript, /* mp_subscript */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	784 (objobjargproc)memory_ass_sub, /* mp_ass_subscript */
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
54829 f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	787 static PySequenceMethods memory_as_sequence = {
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	788 0, /* sq_length */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	789 0, /* sq_concat */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	790 0, /* sq_repeat */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	791 (ssizeargfunc)memory_item, /* sq_item */
f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	792 };
52749 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 /* Buffer methods */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	795 static PyBufferProcs memory_as_buffer = {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	796 0, /* bf_getreadbuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	797 0, /* bf_getwritebuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	798 0, /* bf_getsegcount */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	799 0, /* bf_getcharbuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	800 (getbufferproc)memory_getbuf, /* bf_getbuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	801 (releasebufferproc)memory_releasebuf, /* bf_releasebuffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	802 };
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	803
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	804
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	805 PyTypeObject PyMemoryView_Type = {
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	806 PyVarObject_HEAD_INIT(&PyType_Type, 0)
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	807 "memoryview",
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	808 sizeof(PyMemoryViewObject),
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	809 0,
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	810 (destructor)memory_dealloc, /* tp_dealloc */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	811 0, /* tp_print */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	812 0, /* tp_getattr */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	813 0, /* tp_setattr */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	814 0, /* tp_compare */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	815 (reprfunc)memory_repr, /* tp_repr */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	816 0, /* tp_as_number */
54829 f65adad3988d Issue 6329: Fix iteration for memoryviews. Raymond Hettinger <python@rcn.com> parents: 52749 diff changeset	817 &memory_as_sequence, /* tp_as_sequence */
52749 75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	818 &memory_as_mapping, /* tp_as_mapping */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	819 0, /* tp_hash */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	820 0, /* tp_call */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	821 0, /* tp_str */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	822 PyObject_GenericGetAttr, /* tp_getattro */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	823 0, /* tp_setattro */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	824 &memory_as_buffer, /* tp_as_buffer */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	825 Py_TPFLAGS_DEFAULT \| Py_TPFLAGS_HAVE_GC \|
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	826 Py_TPFLAGS_HAVE_NEWBUFFER, /* tp_flags */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	827 memory_doc, /* tp_doc */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	828 (traverseproc)memory_traverse, /* tp_traverse */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	829 (inquiry)memory_clear, /* tp_clear */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	830 memory_richcompare, /* tp_richcompare */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	831 0, /* tp_weaklistoffset */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	832 0, /* tp_iter */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	833 0, /* tp_iternext */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	834 memory_methods, /* tp_methods */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	835 0, /* tp_members */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	836 memory_getsetlist, /* tp_getset */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	837 0, /* tp_base */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	838 0, /* tp_dict */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	839 0, /* tp_descr_get */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	840 0, /* tp_descr_set */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	841 0, /* tp_dictoffset */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	842 0, /* tp_init */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	843 0, /* tp_alloc */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	844 memory_new, /* tp_new */
75efd757434b Issue #2396: backport the memoryview object. Antoine Pitrou <solipsis@pitrou.net> parents: diff changeset	845 };

Mercurial > cpython-lru_cache-weakref

annotate Objects/memoryobject.c @ 65035:87f4b32905df 2.7