blob: 6952a7c9f1f980bafe1df445c2a2a91d979183d1 (
plain) (
blame)
1
2
3
4
5
6
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
|
#include <pybind11/numpy.h>
#include <pybind11/pybind11.h>
int pdm2pcm_cic3(const uint8_t *pdm, int pdm_length, int16_t *pcm,
int pcm_length);
namespace py = pybind11;
int
process(py::array_t<uint8_t> arr1, py::array_t<int16_t> arr2)
{
// Check shapes or sizes if needed
auto buf1 = arr1.request();
auto buf2 = arr2.request();
if (buf1.ndim != 1 || buf2.ndim != 1) {
throw std::runtime_error("Only 1D arrays supported");
}
if (buf1.size < buf2.size) {
throw std::runtime_error(
"Input 1 must be same or larger than input 2");
}
uint8_t *in = static_cast<uint8_t *>(buf1.ptr);
int16_t *out = static_cast<int16_t *>(buf2.ptr);
int samples = pdm2pcm_cic3(in, arr1.size(), out, arr2.size());
return samples;
}
PYBIND11_MODULE(galearn_pdm, m)
{
m.def("process", &process,
"Process two numpy arrays (uint8 and int16)");
}
|
