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
38
|
#include <pybind11/numpy.h>
#include <pybind11/pybind11.h>
int pdm2pcm_cic3(const uint8_t *pdm, int64_t pdm_length,
int16_t *pcm, int32_t pcm_length,
uint8_t hpf_alpha, uint8_t scale_shift);
namespace py = pybind11;
int
process(py::array_t<uint8_t> arr1, py::array_t<int16_t> arr2, int hpf_alpha, int scale_shift)
{
// 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(), hpf_alpha, scale_shift);
return samples;
}
PYBIND11_MODULE(galearn_pdm, m)
{
m.def("process", &process,
"Process two numpy arrays (uint8 and int16)");
}
|
