lasp_siggen.cpp

Go to the documentation of this file.

#include "lasp_siggen.h"
#include "arma_npy.h"
#include "lasp_siggen_impl.h"
#include <iostream>
#include <pybind11/pybind11.h>
 
using std::cerr;
namespace py = pybind11;
 
void init_siggen(py::module &m) {
 
  py::class_<Siggen, std::shared_ptr<Siggen>> siggen(m, "Siggen");
  siggen.def("setLevel", &Siggen::setLevel,
             "Set the level of the signal generator");
  siggen.def("setDCOffset", &Siggen::setDCOffset);
  siggen.def("setMute", &Siggen::setMute);
  siggen.def("setLevel", &Siggen::setLevel);
  siggen.def("setLevel", &Siggen::setLevel, py::arg("newLevel"),
             py::arg("dB") = true);
  siggen.def("reset", &Siggen::reset);
 
  siggen.def("setFilter", &Siggen::setFilter);
 
  py::class_<Noise, std::shared_ptr<Noise>> noise(m, "Noise", siggen);
  noise.def(py::init<>());
 
  py::class_<Sine, std::shared_ptr<Sine>> sine(m, "Sine", siggen);
  sine.def(py::init<const d>());
  sine.def("setFreq", &Sine::setFreq);
 
  py::class_<Periodic, std::shared_ptr<Periodic>> periodic(m, "Periodic",
                                                           siggen);
  periodic.def("getSequence",
               [](const Sweep &s) { return ColToNpy<d>(s.getSequence()); });
 
  py::class_<Sweep, std::shared_ptr<Sweep>> sweep(m, "Sweep", periodic);
  sweep.def(py::init<const d, const d, const d, const d, const us>());
  sweep.def_readonly_static("ForwardSweep", &Sweep::ForwardSweep);
  sweep.def_readonly_static("BackwardSweep", &Sweep::BackwardSweep);
  sweep.def_readonly_static("LinearSweep", &Sweep::LinearSweep);
  sweep.def_readonly_static("LogSweep", &Sweep::LogSweep);
}