bindings/galearn_pdm.cpp


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

#include <pybind11/pybind11.h>
#include <pybind11/numpy.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)");
}