gr-qtgui: adding time-domain only plotter.

4 files changed, 520 insertions, 0 deletions

diff --git a/gr-qtgui/lib/qtgui_time_sink_c.cc b/gr-qtgui/lib/qtgui_time_sink_c.cc
new file mode 100644
index 000000000..d6e6f0a13
--- /dev/null
+++ b/gr-qtgui/lib/qtgui_time_sink_c.cc
@@ -0,0 +1,178 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2011 Free Software Foundation, Inc.
+ * 
+ * This file is part of GNU Radio
+ * 
+ * GNU Radio 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, or (at your option)
+ * any later version.
+ * 
+ * GNU Radio 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 GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <qtgui_time_sink_c.h>
+#include <gr_io_signature.h>
+#include <string.h>
+
+#include <QTimer>
+
+qtgui_time_sink_c_sptr
+qtgui_make_time_sink_c (int size, double bw,
+			const std::string &name,
+			QWidget *parent)
+{
+  return gnuradio::get_initial_sptr(new qtgui_time_sink_c (size, bw, name,
+							   parent));
+}
+
+qtgui_time_sink_c::qtgui_time_sink_c (int size, double bw,
+				      const std::string &name,
+				      QWidget *parent)
+  : gr_block ("time_sink_c",
+	      gr_make_io_signature (1, -1, sizeof(gr_complex)),
+	      gr_make_io_signature (0, 0, 0)),
+    d_size(size), d_bandwidth(bw), d_name(name),
+    d_parent(parent)
+{
+  d_main_gui = NULL;
+
+  d_index = 0;
+  d_residbuf = new gr_complex[d_size];
+
+  initialize();
+}
+
+qtgui_time_sink_c::~qtgui_time_sink_c()
+{
+  delete d_main_gui;
+  delete [] d_residbuf;
+}
+
+void
+qtgui_time_sink_c::forecast(int noutput_items, gr_vector_int &ninput_items_required)
+{
+  unsigned int ninputs = ninput_items_required.size();
+  for (unsigned int i = 0; i < ninputs; i++) {
+    ninput_items_required[i] = std::min(d_size, 8191);
+  }
+}
+
+void
+qtgui_time_sink_c::initialize()
+{
+  if(qApp != NULL) {
+    d_qApplication = qApp;
+  }
+  else {
+    int argc=0;
+    char **argv = NULL;
+    d_qApplication = new QApplication(argc, argv);
+  }
+
+  /*
+  uint64_t maxBufferSize = 32768;
+  d_main_gui = new SpectrumGUIClass(maxBufferSize, d_fftsize, 
+				    d_center_freq, 
+				    -d_bandwidth/2.0, 
+				    d_bandwidth/2.0);
+
+  d_main_gui->SetDisplayTitle(d_name);
+  d_main_gui->SetFFTSize(d_size);
+
+  d_main_gui->OpenSpectrumWindow(d_parent, 
+				 d_plotfreq, d_plotwaterfall,
+				 d_plottime, d_plotconst);
+  */
+
+  d_main_gui = new TimeDisplayForm(d_parent);
+
+  // initialize update time to 10 times a second
+  set_update_time(0.1);
+}
+
+
+void
+qtgui_time_sink_c::exec_()
+{
+  d_qApplication->exec();
+}
+
+QWidget*
+qtgui_time_sink_c::qwidget()
+{
+  return d_main_gui;
+}
+
+PyObject*
+qtgui_time_sink_c::pyqwidget()
+{
+  PyObject *w = PyLong_FromVoidPtr((void*)d_main_gui);
+  PyObject *retarg = Py_BuildValue("N", w);
+  return retarg;
+}
+
+void
+qtgui_time_sink_c::set_time_domain_axis(double min, double max)
+{
+  d_main_gui->SetTimeDomainAxis(min, max);
+}
+
+void
+qtgui_time_sink_c::set_update_time(double t)
+{
+  d_update_time = t;
+  d_main_gui->SetUpdateTime(d_update_time);
+}
+
+int
+qtgui_time_sink_c::general_work (int noutput_items,
+				 gr_vector_int &ninput_items,
+				 gr_vector_const_void_star &input_items,
+				 gr_vector_void_star &output_items)
+{
+  int j=0;
+  const gr_complex *in = (const gr_complex*)input_items[0];
+
+  for(int i=0; i < noutput_items; i+=d_size) {
+    unsigned int datasize = noutput_items - i;
+    unsigned int resid = d_size-d_index;
+
+    // If we have enough input for one full FFT, do it
+    if(datasize >= resid) {
+      const timespec currentTime = get_highres_clock();
+      
+      // Fill up residbuf with d_fftsize number of items
+      memcpy(d_residbuf+d_index, &in[j], sizeof(gr_complex)*resid);
+      d_index = 0;
+
+      j += resid;
+      
+      d_qApplication->postEvent(d_main_gui,
+				new TimeUpdateEvent(d_residbuf, d_size,
+						    currentTime, true));
+    }
+    // Otherwise, copy what we received into the residbuf for next time
+    else {
+      memcpy(d_residbuf+d_index, &in[j], sizeof(gr_complex)*datasize);
+      d_index += datasize;
+      j += datasize;
+    }   
+  }
+  
+  consume_each(j);
+  return j;
+}
diff --git a/gr-qtgui/lib/qtgui_time_sink_c.h b/gr-qtgui/lib/qtgui_time_sink_c.h
new file mode 100644
index 000000000..8867fa48e
--- /dev/null
+++ b/gr-qtgui/lib/qtgui_time_sink_c.h
@@ -0,0 +1,83 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2011 Free Software Foundation, Inc.
+ * 
+ * This file is part of GNU Radio
+ * 
+ * GNU Radio 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, or (at your option)
+ * any later version.
+ * 
+ * GNU Radio 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 GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifndef INCLUDED_QTGUI_TIME_SINK_C_H
+#define INCLUDED_QTGUI_TIME_SINK_C_H
+
+#include <Python.h>
+#include <gr_block.h>
+#include <gr_firdes.h>
+#include <gri_fft.h>
+#include <qapplication.h>
+#include <timedisplayform.h>
+
+class qtgui_time_sink_c;
+typedef boost::shared_ptr<qtgui_time_sink_c> qtgui_time_sink_c_sptr;
+
+qtgui_time_sink_c_sptr qtgui_make_time_sink_c(int size, double bw,
+					      const std::string &name,
+					      QWidget *parent);
+
+class qtgui_time_sink_c : public gr_block
+{
+private:
+  friend qtgui_time_sink_c_sptr qtgui_make_time_sink_c(int size, double bw,
+						       const std::string &name,
+						       QWidget *parent);
+  qtgui_time_sink_c(int size, double bw,
+		    const std::string &name,
+		    QWidget *parent);
+  
+  void forecast(int noutput_items, gr_vector_int &ninput_items_required);
+
+  void initialize();
+
+  int d_size;
+  double d_bandwidth;
+  std::string d_name;
+
+  int d_index;
+  gr_complex *d_residbuf;
+
+  double d_update_time;
+
+  QWidget *d_parent;
+  TimeDisplayForm *d_main_gui;
+
+public:
+  ~qtgui_time_sink_c();
+  void exec_();
+  QWidget*  qwidget();
+  PyObject* pyqwidget();
+
+  void set_time_domain_axis(double min, double max);
+  void set_update_time(double t);
+
+  QApplication *d_qApplication;
+
+  int general_work (int noutput_items,
+		    gr_vector_int &ninput_items,
+		    gr_vector_const_void_star &input_items,
+		    gr_vector_void_star &output_items);
+};
+
+#endif /* INCLUDED_QTGUI_TIME_SINK_C_H */
diff --git a/gr-qtgui/lib/timedisplayform.cc b/gr-qtgui/lib/timedisplayform.cc
new file mode 100644
index 000000000..809c88a9b
--- /dev/null
+++ b/gr-qtgui/lib/timedisplayform.cc
@@ -0,0 +1,180 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2011 Free Software Foundation, Inc.
+ * 
+ * This file is part of GNU Radio
+ * 
+ * GNU Radio 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, or (at your option)
+ * any later version.
+ * 
+ * GNU Radio 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 GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#include <cmath>
+#include <QColorDialog>
+#include <QMessageBox>
+#include <timedisplayform.h>
+
+TimeDisplayForm::TimeDisplayForm(QWidget* parent)
+  : QWidget(parent)
+{
+  _systemSpecifiedFlag = false;
+  _intValidator = new QIntValidator(this);
+  _intValidator->setBottom(0);
+  _timeDomainDisplayPlot = new TimeDomainDisplayPlot(this);
+
+  _numRealDataPoints = 1024;
+  
+  Reset();
+
+  // Create a timer to update plots at the specified rate
+  displayTimer = new QTimer(this);
+  connect(displayTimer, SIGNAL(timeout()), this, SLOT(UpdateGuiTimer()));
+
+  connect(_timeDomainDisplayPlot, SIGNAL(plotPointSelected(const QPointF)),
+	  this, SLOT(onTimePlotPointSelected(const QPointF)));
+}
+
+TimeDisplayForm::~TimeDisplayForm()
+{
+  // Qt deletes children when parent is deleted
+
+  // Don't worry about deleting Display Plots - they are deleted when parents are deleted
+  delete _intValidator;
+
+  displayTimer->stop();
+  delete displayTimer;
+}
+
+void
+TimeDisplayForm::newData( const TimeUpdateEvent* spectrumUpdateEvent)
+{
+  const gr_complex* timeDomainDataPoints = spectrumUpdateEvent->getTimeDomainPoints();
+  const uint64_t numTimeDomainDataPoints = spectrumUpdateEvent->getNumTimeDomainDataPoints();
+  const timespec generatedTimestamp = spectrumUpdateEvent->getDataTimestamp();
+
+  // FIXME: make time domain display take complex inputs
+  double *real = new double[numTimeDomainDataPoints];
+  double *imag = new double[numTimeDomainDataPoints];
+  for(uint64_t i = 0; i < numTimeDomainDataPoints; i++) {
+    real[i] = timeDomainDataPoints[i].real();
+    imag[i] = timeDomainDataPoints[i].imag();
+  }
+  
+  _timeDomainDisplayPlot->PlotNewData(real, imag,
+				      numTimeDomainDataPoints,
+				      d_update_time);
+
+  // Tell the system the GUI has been updated
+  //if(_systemSpecifiedFlag){
+  //  _system->SetLastGUIUpdateTime(generatedTimestamp);
+  //  _system->DecrementPendingGUIUpdateEvents();
+  //}
+
+  delete [] real;
+  delete [] imag;
+}
+
+void
+TimeDisplayForm::resizeEvent( QResizeEvent *e )
+{
+  QSize s(0,0);
+  //s.setWidth(TimeDomainDisplayFrame->width());
+  //s.setHeight(TimeDomainDisplayFrame->height());
+  emit _timeDomainDisplayPlot->resizeSlot(&s);
+}
+
+void
+TimeDisplayForm::customEvent( QEvent * e)
+{
+  if(e->type() == 10005) {
+    TimeUpdateEvent* timeUpdateEvent = (TimeUpdateEvent*)e;
+    newData(timeUpdateEvent);
+  }
+  //else if(e->type() == 10008){
+  //setWindowTitle(((SpectrumWindowCaptionEvent*)e)->getLabel());
+  //}
+  //else if(e->type() == 10009){
+  //Reset();
+  //if(_systemSpecifiedFlag){
+  //  _system->ResetPendingGUIUpdateEvents();
+  //}
+  //}
+}
+
+void
+TimeDisplayForm::UpdateGuiTimer()
+{
+  _timeDomainDisplayPlot->canvas()->update();
+}
+
+void
+TimeDisplayForm::onTimePlotPointSelected(const QPointF p)
+{
+  emit plotPointSelected(p, 3);
+}
+
+void
+TimeDisplayForm::SetFrequencyRange(const double newCenterFrequency,
+				   const double newStartFrequency, 
+				   const double newStopFrequency)
+{
+  double fdiff = std::max(fabs(newStartFrequency), fabs(newStopFrequency));
+
+  if(fdiff > 0) {
+    std::string strtime[4] = {"sec", "ms", "us", "ns"};
+    double units10 = floor(log10(fdiff));
+    double units3  = std::max(floor(units10 / 3.0), 0.0);
+    double units = pow(10, (units10-fmod(units10, 3.0)));
+    int iunit = static_cast<int>(units3);
+    
+    _startFrequency = newStartFrequency;
+    _stopFrequency = newStopFrequency;
+
+    _timeDomainDisplayPlot->SetSampleRate(_stopFrequency - _startFrequency,
+					  units, strtime[iunit]);
+  }
+}
+
+void
+TimeDisplayForm::Reset()
+{
+}
+
+
+void
+TimeDisplayForm::closeEvent( QCloseEvent *e )
+{
+  //if(_systemSpecifiedFlag){
+  //  _system->SetWindowOpenFlag(false);
+  //}
+
+  qApp->processEvents();
+
+  QWidget::closeEvent(e);
+}
+
+void
+TimeDisplayForm::SetTimeDomainAxis(double min, double max)
+{
+  _timeDomainDisplayPlot->set_yaxis(min, max);
+}
+
+void
+TimeDisplayForm::SetUpdateTime(double t)
+{
+  d_update_time = t;
+  // QTimer class takes millisecond input
+  displayTimer->start(d_update_time*1000);
+}
+
diff --git a/gr-qtgui/lib/timedisplayform.h b/gr-qtgui/lib/timedisplayform.h
new file mode 100644
index 000000000..2ba3a28c8
--- /dev/null
+++ b/gr-qtgui/lib/timedisplayform.h
@@ -0,0 +1,79 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2011 Free Software Foundation, Inc.
+ * 
+ * This file is part of GNU Radio
+ * 
+ * GNU Radio 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, or (at your option)
+ * any later version.
+ * 
+ * GNU Radio 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 GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifndef TIME_DISPLAY_FORM_H
+#define TIME_DISPLAY_FORM_H
+
+#include <spectrumUpdateEvents.h>
+#include <FrequencyDisplayPlot.h>
+#include <WaterfallDisplayPlot.h>
+#include <TimeDomainDisplayPlot.h>
+#include <ConstellationDisplayPlot.h>
+#include <QtGui/QApplication>
+#include <QValidator>
+#include <QTimer>
+#include <vector>
+
+class TimeDisplayForm : public QWidget
+{
+  Q_OBJECT
+
+  public:
+  TimeDisplayForm(QWidget* parent = 0);
+  ~TimeDisplayForm();
+  
+  void Reset();
+  
+public slots:
+  void resizeEvent( QResizeEvent * e );
+  void customEvent( QEvent * e );
+  void SetFrequencyRange( const double newCenterFrequency,
+			  const double newStartFrequency,
+			  const double newStopFrequency );
+  void closeEvent( QCloseEvent * e );
+
+  void SetTimeDomainAxis(double min, double max);
+
+  void SetUpdateTime(double t);
+
+private slots:
+  void newData( const TimeUpdateEvent* );
+  void UpdateGuiTimer();
+
+  void onTimePlotPointSelected(const QPointF p);
+
+signals:
+  void plotPointSelected(const QPointF p, int type);
+
+private:
+  uint64_t _numRealDataPoints;
+  QIntValidator* _intValidator;
+  TimeDomainDisplayPlot* _timeDomainDisplayPlot;
+  bool _systemSpecifiedFlag;
+  double _startFrequency;
+  double _stopFrequency;
+  
+  QTimer *displayTimer;
+  double d_update_time;
+};
+
+#endif /* TIME_DISPLAY_FORM_H */