You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

889 lines
25 KiB

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