Newer
Older
#include "python_wrapper.h"
static void
data_wrapper_dealloc(DataWrapper* self)
{
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
Py_TYPE(self)->tp_free((PyObject*)self);
}
static PyObject *
data_wrapper_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
DataWrapper *self;
self = (DataWrapper *)type->tp_alloc(type, 0);
if (self != NULL) {
new (&self->data) std::shared_ptr<loom::Data>();
}
return (PyObject *)self;
}
static PyObject *
data_wrapper_size(DataWrapper* self)
{
assert(self->data);
return PyLong_FromUnsignedLong(self->data->get_size());
}
static PyObject *
data_wrapper_read(DataWrapper* self)
{
assert(self->data);
size_t size = self->data->get_size();
const char *ptr = self->data->get_raw_data();
return PyBytes_FromStringAndSize(ptr, size);
}
static PyMethodDef data_wrapper_methods[] = {
{"size", (PyCFunction)data_wrapper_size, METH_NOARGS,
"Return the size of the data object"
},
{"read", (PyCFunction)data_wrapper_read, METH_NOARGS,
"Return byte representation of data object"
},
{NULL} /* Sentinel */
};
static PyTypeObject data_wrapper_type = {
PyVarObject_HEAD_INIT(NULL, 0)
"Data", /* tp_name */
sizeof(DataWrapper), /* tp_basicsize */
0, /* tp_itemsize */
(destructor) data_wrapper_dealloc, /* tp_dealloc */
0, /* tp_print */
0, /* tp_getattr */
0, /* tp_setattr */
0, /* tp_reserved */
0, /* tp_repr */
0, /* tp_as_number */
0, /* tp_as_sequence */
0, /* tp_as_mapping */
0, /* tp_hash */
0, /* tp_call */
0, /* tp_str */
0, /* tp_getattro */
0, /* tp_setattro */
0, /* tp_as_buffer */
Py_TPFLAGS_DEFAULT, /* tp_flags */
"DataObject", /* tp_doc */
0, /* tp_traverse */
0, /* tp_clear */
0, /* tp_richcompare */
0, /* tp_weaklistoffset */
0, /* tp_iter */
0, /* tp_iternext */
data_wrapper_methods, /* tp_methods */
0,
0, /* tp_getset */
0, /* tp_base */
0, /* tp_dict */
0, /* tp_descr_get */
0, /* tp_descr_set */
0, /* tp_dictoffset */
0, /* tp_init */
0, /* tp_alloc */
data_wrapper_new, /* tp_new */
};
void data_wrapper_init()
{
assert(!PyType_Ready(&data_wrapper_type));
}
DataWrapper *data_wrapper_create(const std::shared_ptr<loom::Data> &data)
{
DataWrapper *self = (DataWrapper*) PyObject_CallFunctionObjArgs((PyObject*) &data_wrapper_type, NULL);
assert(self);
self->data = data;
return self;
}