1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
// Copyright (C) 2016 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
#include "shibokenbuffer.h"
#include <cstdlib>
#include <cstring>
bool Shiboken::Buffer::checkType(PyObject *pyObj)
{
return PyObject_CheckBuffer(pyObj) != 0;
}
void *Shiboken::Buffer::getPointer(PyObject *pyObj, Py_ssize_t *size)
{
Py_buffer view;
if (PyObject_GetBuffer(pyObj, &view, PyBUF_ND) == 0) {
if (size)
*size = view.len;
PyBuffer_Release(&view);
return view.buf;
}
return nullptr;
}
void *Shiboken::Buffer::copyData(PyObject *pyObj, Py_ssize_t *sizeIn)
{
void *result = nullptr;
Py_ssize_t size = 0;
Py_buffer view;
if (PyObject_GetBuffer(pyObj, &view, PyBUF_ND) == 0) {
size = view.len;
if (size) {
result = std::malloc(size);
if (result != nullptr)
std::memcpy(result, view.buf, size);
else
size = 0;
}
PyBuffer_Release(&view);
}
if (sizeIn != nullptr)
*sizeIn = size;
return result;
}
PyObject *Shiboken::Buffer::newObject(void *memory, Py_ssize_t size, Type type)
{
if (size == 0)
Py_RETURN_NONE;
Py_buffer view;
memset(&view, 0, sizeof(Py_buffer));
view.buf = memory;
view.len = size;
view.readonly = type == Shiboken::Buffer::ReadOnly;
view.ndim = 1;
view.itemsize = sizeof(char);
Py_ssize_t shape[] = { size };
view.shape = shape;
// Pep384: This is way too complicated and impossible with the limited api:
//return PyMemoryView_FromBuffer(&view);
return PyMemoryView_FromMemory(reinterpret_cast<char *>(view.buf),
size, type == ReadOnly ? PyBUF_READ : PyBUF_WRITE);
}
PyObject *Shiboken::Buffer::newObject(const void *memory, Py_ssize_t size)
{
return newObject(const_cast<void *>(memory), size, ReadOnly);
}
|