54 lines
1.6 KiB
C++
54 lines
1.6 KiB
C++
#include "carma"
|
|
#include "lasp_biquadbank.h"
|
|
#include "lasp_siggen.h"
|
|
#include "lasp_siggen_impl.h"
|
|
#include "lasp_slm.h"
|
|
#include "lasp_window.h"
|
|
#include <iostream>
|
|
#include <pybind11/pybind11.h>
|
|
|
|
using std::cerr;
|
|
namespace py = pybind11;
|
|
|
|
void init_dsp(py::module &m) {
|
|
|
|
py::class_<Window> w(m, "Window");
|
|
|
|
py::enum_<Window::WindowType>(w, "WindowType")
|
|
.value("Hann", Window::WindowType::Hann)
|
|
.value("Hamming", Window::WindowType::Hamming)
|
|
.value("Bartlett", Window::WindowType::Bartlett)
|
|
.value("Blackman", Window::WindowType::Bartlett)
|
|
.value("Rectangular", Window::WindowType::Rectangular)
|
|
.export_values();
|
|
|
|
py::class_<Siggen, std::shared_ptr<Siggen>> siggen(m, "Siggen");
|
|
siggen.def("setLevel", &Siggen::setLevel,
|
|
"Set the level of the signal generator");
|
|
|
|
py::class_<Sine, std::shared_ptr<Sine>> sw(m, "Sine", siggen);
|
|
sw.def(py::init<const d>());
|
|
|
|
py::class_<SeriesBiquad> sbq(m, "SeriesBiquad");
|
|
sbq.def(py::init<const vd &>());
|
|
sbq.def("filter", [](SeriesBiquad &s, const vd &input) {
|
|
vd res = input;
|
|
s.filter(res);
|
|
return res;
|
|
});
|
|
|
|
py::class_<SLM> slm(m, "SLM");
|
|
slm.def_static(
|
|
"fromBiquads",
|
|
py::overload_cast<const d, const d, const us, const d, const dmat &>(
|
|
&SLM::fromBiquads));
|
|
slm.def_static(
|
|
"fromBiquads",
|
|
py::overload_cast<const d, const d, const us, const d, const vd &,
|
|
const dmat &>(&SLM::fromBiquads));
|
|
slm.def("run", &SLM::run);
|
|
slm.def_readonly("Pm", &SLM::Pm);
|
|
slm.def_readonly("Pmax", &SLM::Pmax);
|
|
slm.def_readonly("Ppeak", &SLM::Ppeak);
|
|
}
|