widget.hpp

/*
 Copyright (C) 2011 Georgia Institute of Technology

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.

 */

#include <QComboBox>
#include <string>

#include <rtxi/gen/gen_biphase.h>
#include <rtxi/gen/gen_mono.h>
#include <rtxi/gen/gen_saw.h>
#include <rtxi/gen/gen_sine.h>
#include <rtxi/gen/gen_zap.h>
#include <rtxi/widgets.hpp>

namespace SigGen
{

constexpr std::string_view MODULE_NAME = "Signal Generator";

namespace WAVEMODE
{
enum mode_t : int64_t
{
  SINE = 0,
  MONOSQUARE,
  BISQUARE,
  SAWTOOTH,
  ZAP,
};
}  // namespace WAVEMODE

enum PARAMETER : Widgets::Variable::Id
{
  SIGNAL_WAVEFORM = 0,
  DELAY,
  WIDTH,
  FREQ,
  AMPLITUDE,
  ZAP_MAX_FREQ,
  ZAP_DURATION,
};

inline  std::vector<Widgets::Variable::Info> get_default_vars() 
{
  return {
      {PARAMETER::SIGNAL_WAVEFORM,
       "Signal Waveform",
       "The current type of signal being generated. Current types are sine, "
       "monosquare, bisquare, sawtooth, and zap",     
       Widgets::Variable::INT_PARAMETER,
       int64_t{0}},
      {PARAMETER::DELAY,
       "Delay (s)",
       "Delay (s)",
       Widgets::Variable::DOUBLE_PARAMETER,
       1.0},
      {PARAMETER::WIDTH,
       "Width (s)",
       "Width (s)",
       Widgets::Variable::DOUBLE_PARAMETER,
       1.0},
      {PARAMETER::FREQ,
       "Freq (Hz)",
       "Freq (Hz), also used as minimum ZAP frequency",
       Widgets::Variable::DOUBLE_PARAMETER,
       1.0},
      {PARAMETER::AMPLITUDE,
       "Amplitude (V)",
       "Amplitude (V)",
       Widgets::Variable::DOUBLE_PARAMETER,
       1.0},
      {PARAMETER::ZAP_MAX_FREQ,
       "ZAP max Freq (Hz)",
       "Maximum ZAP frequency",
       Widgets::Variable::DOUBLE_PARAMETER,
       20.0},
      {PARAMETER::ZAP_DURATION,
       "ZAP duration (s)",
       "ZAP duration (s)",
       Widgets::Variable::DOUBLE_PARAMETER,
       10.0}
  };
}

inline std::vector<IO::channel_t> get_default_channels()
{
  return {
      {"Signal Generator Output", "Signal Generator Output", IO::OUTPUT}};
}

class Panel : public Widgets::Panel
{
  Q_OBJECT
public:
  Panel(QMainWindow* main_window, Event::Manager* ev_manager);

  void customizeGUI();

public slots:
  void refresh() override;

private:
  // QT components
  QPushButton* sineButton;
  QComboBox* waveShape;

private slots:

  void updateMode(int);
};

class Component : public Widgets::Component
{
public:
  explicit Component(Widgets::Plugin* hplugin);
  void execute() override;

private:
  void initParameters();
  void initStimulus();  // creates SigGen stimuli

  GeneratorSine sineWave;
  GeneratorMono monoWave;
  GeneratorBiphase biWave;
  GeneratorSaw sawWave;
  GeneratorZap zapWave;
  Generator* current_generator = nullptr;

  double dt;
  mode_t mode;
};

class Plugin : public Widgets::Plugin
{
public:
  explicit Plugin(Event::Manager* ev_manager);
};

}  // namespace SigGen