Skip to content

Commit ae38935

Browse files
committed
fix: python binding for new API
1 parent 025cfae commit ae38935

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

pyudmaio/src/DataHandlerPython.cpp

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ DataHandlerPython::DataHandlerPython(std::shared_ptr<UioAxiDmaIf> dma_ptr,
2222
, _mem_ptr(mem_ptr) {}
2323

2424
void DataHandlerPython::start(int nr_pkts, size_t pkt_size, bool init_only) {
25-
_desc_ptr->init_buffers(*_mem_ptr, nr_pkts, pkt_size);
25+
_desc_ptr->init_buffers(_mem_ptr, nr_pkts, pkt_size);
2626

2727
uintptr_t first_desc = _desc_ptr->get_first_desc_addr();
2828
_dma_ptr->start(first_desc);
@@ -56,12 +56,9 @@ py::array_t<uint8_t> DataHandlerPython::numpy_read_nb() {
5656
throw std::runtime_error("DMA has experienced an error");
5757
}
5858

59-
std::vector<UioRegion> full_bufs = _desc_ptr->get_full_buffers();
60-
auto vec = new std::vector<uint8_t>();
59+
auto full_bufs = _desc_ptr->get_full_buffers();
60+
auto vec = new std::vector<uint8_t>(_desc_ptr->read_buffers(full_bufs));
6161

62-
for (auto& buf : full_bufs) {
63-
_mem_ptr->append_from_buf(buf, *vec);
64-
}
6562
// Callback for Python garbage collector
6663
py::capsule gc_callback(vec, [](void* f) {
6764
auto ptr = reinterpret_cast<std::vector<uint8_t>*>(f);

0 commit comments

Comments
 (0)