pybind/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2026-05-13 01:36:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for nested recarrays

Browse Source
This commit is contained in:
Ivan Smirnov
2016-06-19 16:41:15 +01:00
parent 80a3785a66
commit 7f913aecab
2 changed files with 41 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
example/example20.cpp
View File

@@ -30,23 +30,36 @@ struct PackedStruct {
struct NestedStruct {
Struct a;
PackedStruct b;
};
} __attribute__((packed));
template <typename T>
py::array mkarray_via_buffer(size_t n) {
return py::array(py::buffer_info(nullptr, sizeof(T),
py::format_descriptor<T>::value(),
1, { n }, { sizeof(T) }));
}
template <typename S>
py::array_t<S> create_recarray(size_t n) {
auto arr = py::array(py::buffer_info(nullptr, sizeof(S),
py::format_descriptor<S>::value(),
1, { n }, { sizeof(S) }));
auto buf = arr.request();
auto ptr = static_cast<S*>(buf.ptr);
auto arr = mkarray_via_buffer<S>(n);
auto ptr = static_cast<S*>(arr.request().ptr);
for (size_t i = 0; i < n; i++) {
ptr[i].x = i % 2;
ptr[i].y = i;
ptr[i].z = i * 1.5;
ptr[i].x = i % 2; ptr[i].y = (uint32_t) i; ptr[i].z = (float) i * 1.5f;
}
return arr;
}
py::array_t<NestedStruct> create_nested(size_t n) {
auto arr = mkarray_via_buffer<NestedStruct>(n);
auto ptr = static_cast<NestedStruct*>(arr.request().ptr);
for (size_t i = 0; i < n; i++) {
ptr[i].a.x = i % 2; ptr[i].a.y = (uint32_t) i; ptr[i].a.z = (float) i * 1.5f;
ptr[i].b.x = (i + 1) % 2; ptr[i].b.y = (uint32_t) (i + 1); ptr[i].b.z = (float) (i + 1) * 1.5f;
}
return arr;
}
void init_ex20(py::module &m) {
PYBIND11_DTYPE(Struct, x, y, z);
PYBIND11_DTYPE(PackedStruct, x, y, z);
@@ -54,4 +67,5 @@ void init_ex20(py::module &m) {
m.def("create_rec_simple", &create_recarray<Struct>);
m.def("create_rec_packed", &create_recarray<PackedStruct>);
m.def("create_rec_nested", &create_nested);
}