#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;

/**
 * \ingroup pybind
 * @{
 *
 */

/**
 * @brief Initialize Siggen wrappers
 *
 * @param m The Python module to add classes and methods to
 */
void init_siggen(py::module &m) {

  /// Siggen
  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("setInterruptPeriod", &Siggen::setInterruptPeriod);

  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);
}
/** @} */