james/ndi-python
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

get video array

Browse Source
This commit is contained in:
Naoto Kondo
2020-04-25 07:18:46 +09:00
parent 9ee4cc1b6a
commit 300075443b
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/main.cpp
View File

@@ -103,7 +103,16 @@ PYBIND11_MODULE(NDIlib, m) {
&NDIlib_video_frame_v2_t::frame_format_type) &NDIlib_video_frame_v2_t::frame_format_type)
.def_readwrite("timecode", &NDIlib_video_frame_v2_t::timecode) .def_readwrite("timecode", &NDIlib_video_frame_v2_t::timecode)
.def_property( .def_property(
"data", nullptr, "data",
[](const NDIlib_video_frame_v2_t &self) {
auto buffer_info =
py::buffer_info(self.p_data, sizeof(uint8_t),
py::format_descriptor<uint8_t>::format(), 3,
{self.yres, self.xres, 4},
{sizeof(uint8_t) * self.xres * 4,
sizeof(uint8_t) * 4, sizeof(uint8_t)});
return py::array(buffer_info);
},
[](NDIlib_video_frame_v2_t &self, const py::array_t<uint8_t> &array) { [](NDIlib_video_frame_v2_t &self, const py::array_t<uint8_t> &array) {
auto info = array.request(); auto info = array.request();
self.p_data = reinterpret_cast<uint8_t *>(info.ptr); self.p_data = reinterpret_cast<uint8_t *>(info.ptr);