PyAPI: buffer protocol support for mathutils types

Adding buffer protocol support increases the speed of copying a Vector
(3D) array into a `numpy.array` by up to x3.8.

Ref !144401
This commit is contained in:
Oxicid
2025-08-16 06:14:19 +00:00
committed by Campbell Barton
parent ebfa7edeb1
commit b856b6010e
8 changed files with 402 additions and 6 deletions

View File

@@ -383,6 +383,59 @@ static PyObject *Euler_str(EulerObject *self)
/** \} */
/* -------------------------------------------------------------------- */
/** \name Euler Type: Buffer Protocol
* \{ */
static int Euler_getbuffer(PyObject *obj, Py_buffer *view, int flags)
{
EulerObject *self = (EulerObject *)obj;
if (UNLIKELY(BaseMath_Prepare_ForBufferAccess(self, view, flags) == -1)) {
return -1;
}
if (UNLIKELY(BaseMath_ReadCallback(self) == -1)) {
return -1;
}
memset(view, 0, sizeof(*view));
view->obj = (PyObject *)self;
view->buf = (void *)self->eul;
view->len = Py_ssize_t(EULER_SIZE * sizeof(float));
view->itemsize = sizeof(float);
view->ndim = 1;
if ((flags & PyBUF_WRITABLE) == 0) {
view->readonly = 1;
}
if (flags & PyBUF_FORMAT) {
view->format = (char *)"f";
}
self->flag |= BASE_MATH_FLAG_HAS_BUFFER_VIEW;
Py_INCREF(self);
return 0;
}
static void Euler_releasebuffer(PyObject * /*exporter*/, Py_buffer *view)
{
EulerObject *self = (EulerObject *)view->obj;
self->flag &= ~BASE_MATH_FLAG_HAS_BUFFER_VIEW;
if (view->readonly == 0) {
if (UNLIKELY(BaseMath_WriteCallback(self) == -1)) {
PyErr_Print();
}
}
}
static PyBufferProcs Euler_as_buffer = {
(getbufferproc)Euler_getbuffer,
(releasebufferproc)Euler_releasebuffer,
};
/** \} */
/* -------------------------------------------------------------------- */
/** \name Euler Type: Rich Compare
* \{ */
@@ -854,7 +907,7 @@ PyTypeObject euler_Type = {
/*tp_str*/ (reprfunc)Euler_str,
/*tp_getattro*/ nullptr,
/*tp_setattro*/ nullptr,
/*tp_as_buffer*/ nullptr,
/*tp_as_buffer*/ &Euler_as_buffer,
/*tp_flags*/ Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC,
/*tp_doc*/ euler_doc,
/*tp_traverse*/ (traverseproc)BaseMathObject_traverse,