This commit is contained in:
wangjiyu 2025-04-16 16:55:10 +08:00
parent a6e6f7fdde
commit 6cb5d60eb5
4 changed files with 14 additions and 2 deletions

View File

@ -55,6 +55,9 @@ namespace np = boost::python::numpy;
np::ndarray getNumpydata() {
return RTCContext::instance().getNumpydata();
}
int16_t getDataCount() {
return RTCContext::instance().getDataCount();
}
py::object create_int16_array() {
// 1. 定义数组维度1维长度为 4
npy_intp dims[1] = {4};
@ -209,6 +212,7 @@ BOOST_PYTHON_MODULE(rtc_plugins) {
py::def("getSize", &getSize);
py::def("getData", &getData);
py::def("getNumpyData", &getNumpydata);
py::def("getDataCount", &getDataCount);
} catch (...) {
PyErr_SetString(PyExc_RuntimeError, "Module initialization failed");
}

View File

@ -58,6 +58,8 @@ while True:
print("resend succ")
size = rtc_plugins.getSize()
print(f"data size:{size}")
frame = rtc_plugins.getNumpyData()
print(f"frame:{frame}")
#frame = rtc_plugins.getNumpyData()
#print(f"frame:{frame}")
dataCount = rtc_plugins.getDataCount()
print(f"data count:{dataCount}")
time.sleep(3)

View File

@ -724,6 +724,11 @@ np::ndarray RTCContext::getNumpydata() {
}
int16_t RTCContext::getDataCount() {
std::lock_guard<std::mutex> lock(dataMutex_);
RetAudioFrame frame = getData();
return frame.dataCount;
}
int16_t RTCContext::getSize() {
std::lock_guard<std::mutex> lock(dataMutex_);
return dataSize_;

View File

@ -100,6 +100,7 @@ public:
void setData(const mrtc::MRTCAudioFrame& frame);
RetAudioFrame getData();
np::ndarray getNumpydata();
int16_t getDataCount();
void* getpData() const;
void setpData(void* pData);