From 392227182979367a981606e86c7ec24314555c04 Mon Sep 17 00:00:00 2001
From: Tom <trondeau@vt.edu>
Date: Fri, 14 Aug 2009 23:29:17 -0400
Subject: Adding a QT-based data file viewer.
This is the start of an application to read in data files and display them in a useful manner.
---
gr-utils/src/python/gr_plot_qt.py | 504 ++++++++++++++++++++++++++++++++++++++
1 file changed, 504 insertions(+)
create mode 100755 gr-utils/src/python/gr_plot_qt.py
(limited to 'gr-utils/src/python/gr_plot_qt.py')
diff --git a/gr-utils/src/python/gr_plot_qt.py b/gr-utils/src/python/gr_plot_qt.py
new file mode 100755
index 0000000000..7c744f2c69
--- /dev/null
+++ b/gr-utils/src/python/gr_plot_qt.py
@@ -0,0 +1,504 @@
+#!/usr/bin/env python
+
+try:
+ import scipy
+ from scipy import fftpack
+except ImportError:
+ print "Please install SciPy to run this script (http://www.scipy.org/)"
+ raise SystemExit, 1
+
+import sys, os
+from PyQt4 import Qt, QtCore, QtGui
+import PyQt4.Qwt5 as Qwt
+from matplotlib import mlab
+from optparse import OptionParser
+from gnuradio import eng_notation
+
+from pyqt_plot import Ui_MainWindow
+
+class gr_plot_qt(QtGui.QMainWindow):
+ def __init__(self, qapp, filename, options, parent=None):
+ QtGui.QWidget.__init__(self, parent)
+ self.gui = Ui_MainWindow()
+ self.gui.setupUi(self)
+
+ self.block_length = options.block_length
+ self.start = options.start
+ self.sample_rate = options.sample_rate
+ self.psdfftsize = options.psd_size
+ self.specfftsize = options.spec_size
+ self.winfunc = scipy.blackman
+ self.sizeof_data = 8
+ self.datatype = scipy.complex64
+ self.iq = list()
+ self.time = list()
+
+ # Set up basic plot attributes
+ self.gui.timePlot.setAxisTitle(self.gui.timePlot.xBottom, "Time (sec)")
+ self.gui.timePlot.setAxisTitle(self.gui.timePlot.yLeft, "Amplitude (V)")
+ self.gui.freqPlot.setAxisTitle(self.gui.freqPlot.xBottom, "Frequency (Hz)")
+ self.gui.freqPlot.setAxisTitle(self.gui.freqPlot.yLeft, "Magnitude (dB)")
+
+ # Set up FFT size combo box
+ self.gui.fftComboBox.addItems(["128", "256", "512", "1024", "2048",
+ "4096", "8192", "16384", "32768"])
+ pos = self.gui.fftComboBox.findText(Qt.QString("%1").arg(self.psdfftsize))
+ self.gui.fftComboBox.setCurrentIndex(pos)
+ self.connect(self.gui.fftComboBox,
+ Qt.SIGNAL("activated (const QString&)"),
+ self.fftComboBoxEdit)
+
+ # Set up color scheme box
+ self.color_modes = {"Black on White" : self.color_black_on_white,
+ "White on Black" : self.color_white_on_black,
+ "Blue on Black" : self.color_blue_on_black,
+ "Green on Black" : self.color_green_on_black}
+ self.gui.colorComboBox.addItems(self.color_modes.keys())
+ pos = self.gui.colorComboBox.findText("Blue on Black")
+ self.gui.colorComboBox.setCurrentIndex(pos)
+ self.connect(self.gui.colorComboBox,
+ Qt.SIGNAL("activated (const QString&)"),
+ self.colorComboBoxEdit)
+
+
+ # Create zoom functionality for the plots
+ self.timeZoomer = Qwt.QwtPlotZoomer(self.gui.timePlot.xBottom,
+ self.gui.timePlot.yLeft,
+ Qwt.QwtPicker.PointSelection,
+ Qwt.QwtPicker.AlwaysOn,
+ self.gui.timePlot.canvas())
+
+ self.freqZoomer = Qwt.QwtPlotZoomer(self.gui.freqPlot.xBottom,
+ self.gui.freqPlot.yLeft,
+ Qwt.QwtPicker.PointSelection,
+ Qwt.QwtPicker.AlwaysOn,
+ self.gui.freqPlot.canvas())
+
+ self.picker = Qwt.QwtPlotPicker(self.gui.timePlot.xBottom,
+ self.gui.timePlot.yLeft,
+ Qwt.QwtPicker.PointSelection,
+ Qwt.QwtPlotPicker.CrossRubberBand,
+ Qwt.QwtPicker.AlwaysOn,
+ self.gui.timePlot.canvas())
+ self.picker.connect(self.picker,
+ Qt.SIGNAL('selected(const QwtDoublePoint&)'),
+ self.clickMe)
+
+ # Set up action when tab is changed
+ self.connect(self.gui.tabGroup,
+ Qt.SIGNAL("currentChanged (int)"),
+ self.tabChanged)
+
+ # Add a legend to the Time plot
+ legend_real = Qwt.QwtLegend()
+ self.gui.timePlot.insertLegend(legend_real)
+
+ # Set up slider
+ self.gui.plotHBar.setSingleStep(1)
+ self.gui.plotHBar.setPageStep(self.block_length)
+ self.gui.plotHBar.setMinimum(0)
+ self.gui.plotHBar.setMaximum(self.block_length)
+ self.connect(self.gui.plotHBar,
+ Qt.SIGNAL("valueChanged(int)"),
+ self.sliderMoved)
+
+ # Connect Open action to Open Dialog box
+ self.connect(self.gui.action_open,
+ Qt.SIGNAL("activated()"),
+ self.open_file)
+
+ # Set up file position boxes to update current figure
+ self.connect(self.gui.filePosStartLineEdit,
+ Qt.SIGNAL("editingFinished()"),
+ self.file_position_changed)
+ self.connect(self.gui.filePosStopLineEdit,
+ Qt.SIGNAL("editingFinished()"),
+ self.file_position_changed)
+ self.connect(self.gui.filePosLengthLineEdit,
+ Qt.SIGNAL("editingFinished()"),
+ self.file_length_changed)
+
+ self.connect(self.gui.fileTimeStartLineEdit,
+ Qt.SIGNAL("editingFinished()"),
+ self.file_time_changed)
+ self.connect(self.gui.fileTimeStopLineEdit,
+ Qt.SIGNAL("editingFinished()"),
+ self.file_time_changed)
+ self.connect(self.gui.fileTimeLengthLineEdit,
+ Qt.SIGNAL("editingFinished()"),
+ self.file_time_length_changed)
+
+ self.rcurve = Qwt.QwtPlotCurve("Real")
+ self.icurve = Qwt.QwtPlotCurve("Imaginary")
+
+ self.icurve.attach(self.gui.timePlot)
+ self.rcurve.attach(self.gui.timePlot)
+
+ self.psdcurve = Qwt.QwtPlotCurve("PSD")
+ self.psdcurve.attach(self.gui.freqPlot)
+
+ # Set up initial color scheme
+ self.color_modes["Blue on Black"]()
+
+ self.set_sample_rate(self.sample_rate)
+ self.connect(self.gui.sampleRateLineEdit,
+ Qt.SIGNAL("editingFinished()"),
+ self.sample_rate_changed)
+
+ if(filename is not None):
+ self.initialize(filename)
+
+ self.show()
+
+ def open_file(self):
+ filename = Qt.QFileDialog.getOpenFileName(self, "Open", ".")
+ print filename
+ self.initialize(filename)
+
+ def initialize(self, filename):
+ self.hfile = open(filename, "r")
+
+ self.setWindowTitle(("GNU Radio File Plot Utility: %s" % filename))
+
+ self.cur_start = 0
+ self.cur_stop = self.block_length
+
+ self.init_data_input()
+ self.get_data(self.cur_start, self.cur_stop)
+ self.get_psd()
+ self.gui.plotHBar.setSliderPosition(0)
+ self.gui.plotHBar.setMaximum(self.signal_size)
+
+ self.update_time_curves()
+ self.update_psd_curves()
+
+ def init_data_input(self):
+ self.hfile.seek(0, os.SEEK_END)
+ self.signal_size = self.hfile.tell()/self.sizeof_data
+ print "Sizeof File: ", self.signal_size
+ self.hfile.seek(0, os.SEEK_SET)
+
+ def get_data(self, start, end):
+ if(end > start):
+ self.hfile.seek(start*self.sizeof_data, os.SEEK_SET)
+ self.position = start
+ iq = scipy.fromfile(self.hfile, dtype=self.datatype,
+ count=end-start)
+ if(len(iq) < (end-start)):
+ print "End of File"
+ else:
+ tstep = 1.0 / self.sample_rate
+ self.iq = iq
+ self.time = [tstep*(self.position + i) for i in xrange(len(self.iq))]
+
+ self.set_file_pos_box(start, end)
+ else:
+ # Do we want to do anything about this?
+ pass
+
+ def get_psd(self):
+ winpoints = self.winfunc(self.psdfftsize)
+ iq_psd, freq = mlab.psd(self.iq, Fs=self.sample_rate,
+ NFFT=self.psdfftsize,
+ noverlap=self.psdfftsize/4.0,
+ window=winpoints,
+ scale_by_freq=False)
+ self.iq_psd = 10.0*scipy.log10(abs(fftpack.fftshift(iq_psd)))
+ self.freq = freq - self.sample_rate/2.0
+
+
+ def clickMe(self, qPoint):
+ print qPoint.x()
+
+ def fftComboBoxEdit(self, fftSize):
+ self.psdfftsize = fftSize.toInt()[0]
+ self.get_psd()
+ self.update_psd_curves()
+
+ def colorComboBoxEdit(self, colorSelection):
+ colorstr = str(colorSelection.toAscii())
+ color_func = self.color_modes[colorstr]
+ color_func()
+
+ def sliderMoved(self, value):
+ pos_start = value
+ pos_end = value + self.gui.plotHBar.pageStep()
+
+ self.get_data(pos_start, pos_end)
+ self.get_psd()
+ self.update_time_curves()
+ self.update_psd_curves()
+
+ def set_sample_rate(self, sr):
+ self.sample_rate = sr
+ srstr = eng_notation.num_to_str(self.sample_rate)
+ self.gui.sampleRateLineEdit.setText(Qt.QString("%1").arg(srstr))
+
+ def sample_rate_changed(self):
+ srstr = self.gui.sampleRateLineEdit.text().toAscii()
+ self.sample_rate = eng_notation.str_to_num(srstr)
+ self.set_file_pos_box(self.cur_start, self.cur_stop)
+ self.get_data(self.cur_start, self.cur_stop)
+ self.get_psd()
+ self.update_time_curves()
+ self.update_psd_curves()
+
+ def set_file_pos_box(self, start, end):
+ tstart = start / self.sample_rate
+ tend = end / self.sample_rate
+
+ self.gui.filePosStartLineEdit.setText(Qt.QString("%1").arg(start))
+ self.gui.filePosStopLineEdit.setText(Qt.QString("%1").arg(end))
+ self.gui.filePosLengthLineEdit.setText(Qt.QString("%1").arg(end-start))
+
+ self.gui.fileTimeStartLineEdit.setText(Qt.QString("%1").arg(tstart))
+ self.gui.fileTimeStopLineEdit.setText(Qt.QString("%1").arg(tend))
+ self.gui.fileTimeLengthLineEdit.setText(Qt.QString("%1").arg(tend-tstart))
+
+ def file_position_changed(self):
+ start = self.gui.filePosStartLineEdit.text().toInt()
+ end = self.gui.filePosStopLineEdit.text().toInt()
+ if((start[1] == True) and (end[1] == True)):
+ self.cur_start = start[0]
+ self.cur_stop = end[0]
+
+ tstart = self.cur_start / self.sample_rate
+ tend = self.cur_stop / self.sample_rate
+ self.gui.fileTimeStartLineEdit.setText(Qt.QString("%1").arg(tstart))
+ self.gui.fileTimeStopLineEdit.setText(Qt.QString("%1").arg(tend))
+
+ self.get_data(self.cur_start, self.cur_stop)
+
+ self.update_time_curves()
+ self.update_psd_curves()
+
+ # If there's a non-digit character, reset box
+ else:
+ self.set_file_pos_box(self.cur_start, self.cur_stop)
+
+ def file_time_changed(self):
+ tstart = self.gui.fileTimeStartLineEdit.text().toDouble()
+ tstop = self.gui.fileTimeStopLineEdit.text().toDouble()
+ if((tstart[1] == True) and (tstop[1] == True)):
+ self.cur_start = int(tstart[0] * self.sample_rate)
+ self.cur_stop = int(tstop[0] * self.sample_rate)
+ self.get_data(self.cur_start, self.cur_stop)
+
+ self.gui.filePosStartLineEdit.setText(Qt.QString("%1").arg(self.cur_start))
+ self.gui.filePosStopLineEdit.setText(Qt.QString("%1").arg(self.cur_stop))
+
+ self.update_time_curves()
+ self.update_psd_curves()
+ # If there's a non-digit character, reset box
+ else:
+ self.set_file_pos_box(self.cur_start, self.cur_stop)
+
+ def file_length_changed(self):
+ start = self.gui.filePosStartLineEdit.text().toInt()
+ length = self.gui.filePosLengthLineEdit.text().toInt()
+ if((start[1] == True) and (length[1] == True)):
+ self.cur_start = start[0]
+ self.block_length = length[0]
+ self.cur_stop = self.cur_start + self.block_length
+
+ tstart = self.cur_start / self.sample_rate
+ tend = self.cur_stop / self.sample_rate
+ tlen = self.block_length / self.sample_rate
+ self.gui.fileTimeStartLineEdit.setText(Qt.QString("%1").arg(tstart))
+ self.gui.fileTimeStopLineEdit.setText(Qt.QString("%1").arg(tend))
+ self.gui.fileTimeLengthLineEdit.setText(Qt.QString("%1").arg(tlen))
+
+ self.gui.plotHBar.setPageStep(self.block_length)
+
+ self.get_data(self.cur_start, self.cur_stop)
+ self.get_psd()
+
+ self.update_time_curves()
+ self.update_psd_curves()
+ # If there's a non-digit character, reset box
+ else:
+ self.set_file_pos_box(self.cur_start, self.cur_stop)
+
+ def file_time_length_changed(self):
+ tstart = self.gui.fileTimeStartLineEdit.text().toDouble()
+ tlength = self.gui.fileTimeLengthLineEdit.text().toDouble()
+ if((tstart[1] == True) and (tlength[1] == True)):
+ self.cur_start = int(tstart[0] * self.sample_rate)
+ self.block_length = int(tlength[0] * self.sample_rate)
+ self.cur_stop = self.cur_start + self.block_length
+
+ tstart = self.cur_start / self.sample_rate
+ tend = self.cur_stop / self.sample_rate
+ tlen = self.block_length / self.sample_rate
+ self.gui.fileTimeStartLineEdit.setText(Qt.QString("%1").arg(tstart))
+ self.gui.fileTimeStopLineEdit.setText(Qt.QString("%1").arg(tend))
+ self.gui.fileTimeLengthLineEdit.setText(Qt.QString("%1").arg(tlen))
+
+ self.get_data(self.cur_start, self.cur_stop)
+ self.get_psd()
+
+ self.update_time_curves()
+ self.update_psd_curves()
+ # If there's a non-digit character, reset box
+ else:
+ self.set_file_pos_box(self.cur_start, self.cur_stop)
+
+
+ def update_time_curves(self):
+ self.icurve.setData(self.time, self.iq.imag)
+ self.rcurve.setData(self.time, self.iq.real)
+
+ # Reset the x-axis to the new time scale
+ iqmax = 1.5 * max(max(self.iq.real), max(self.iq.imag))
+ iqmin = 1.5 * min(min(self.iq.real), min(self.iq.imag))
+ self.gui.timePlot.setAxisScale(self.gui.timePlot.xBottom,
+ min(self.time),
+ max(self.time))
+ self.gui.timePlot.setAxisScale(self.gui.timePlot.yLeft,
+ iqmin,
+ iqmax)
+
+ # Set the zoomer base to unzoom to the new axis
+ self.timeZoomer.setZoomBase()
+
+ self.gui.timePlot.replot()
+
+ def update_psd_curves(self):
+ self.psdcurve.setData(self.freq, self.iq_psd)
+
+ self.gui.freqPlot.setAxisScale(self.gui.freqPlot.xBottom,
+ min(self.freq),
+ max(self.freq))
+
+ # Set the zoomer base to unzoom to the new axis
+ self.freqZoomer.setZoomBase()
+
+ self.gui.freqPlot.replot()
+
+ def tabChanged(self, index):
+ self.gui.timePlot.replot()
+ self.gui.freqPlot.replot()
+
+ def color_black_on_white(self):
+ blue = QtGui.qRgb(0x00, 0x00, 0xFF)
+ red = QtGui.qRgb(0xFF, 0x00, 0x00)
+
+ blackBrush = Qt.QBrush(Qt.QColor("black"))
+ blueBrush = Qt.QBrush(Qt.QColor(blue))
+ redBrush = Qt.QBrush(Qt.QColor(red))
+
+ self.gui.timePlot.setCanvasBackground(Qt.QColor("white"))
+ self.gui.freqPlot.setCanvasBackground(Qt.QColor("white"))
+ self.picker.setTrackerPen(Qt.QPen(blackBrush, 2))
+ self.timeZoomer.setTrackerPen(Qt.QPen(blackBrush, 2))
+ self.timeZoomer.setRubberBandPen(Qt.QPen(blackBrush, 2))
+ self.freqZoomer.setTrackerPen(Qt.QPen(blackBrush, 2))
+ self.freqZoomer.setRubberBandPen(Qt.QPen(blackBrush, 2))
+ self.psdcurve.setPen(Qt.QPen(blueBrush, 1))
+ self.rcurve.setPen(Qt.QPen(blueBrush, 2))
+ self.icurve.setPen(Qt.QPen(redBrush, 2))
+
+ self.gui.timePlot.replot()
+ self.gui.freqPlot.replot()
+
+ def color_white_on_black(self):
+ white = QtGui.qRgb(0xFF, 0xFF, 0xFF)
+ red = QtGui.qRgb(0xFF, 0x00, 0x00)
+
+ whiteBrush = Qt.QBrush(Qt.QColor("white"))
+ whiteBrush = Qt.QBrush(Qt.QColor(white))
+ redBrush = Qt.QBrush(Qt.QColor(red))
+
+ self.gui.timePlot.setCanvasBackground(QtGui.QColor("black"))
+ self.gui.freqPlot.setCanvasBackground(QtGui.QColor("black"))
+ self.picker.setTrackerPen(Qt.QPen(whiteBrush, 2))
+ self.timeZoomer.setTrackerPen(Qt.QPen(whiteBrush, 2))
+ self.timeZoomer.setRubberBandPen(Qt.QPen(whiteBrush, 2))
+ self.freqZoomer.setTrackerPen(Qt.QPen(whiteBrush, 2))
+ self.freqZoomer.setRubberBandPen(Qt.QPen(whiteBrush, 2))
+ self.psdcurve.setPen(Qt.QPen(whiteBrush, 1))
+ self.rcurve.setPen(Qt.QPen(whiteBrush, 2))
+ self.icurve.setPen(Qt.QPen(redBrush, 2))
+
+ self.gui.timePlot.replot()
+ self.gui.freqPlot.replot()
+
+
+ def color_green_on_black(self):
+ green = QtGui.qRgb(0x00, 0xFF, 0x00)
+ red = QtGui.qRgb(0xFF, 0x00, 0x50)
+
+ whiteBrush = Qt.QBrush(Qt.QColor("white"))
+ greenBrush = Qt.QBrush(Qt.QColor(green))
+ redBrush = Qt.QBrush(Qt.QColor(red))
+
+ self.gui.timePlot.setCanvasBackground(QtGui.QColor("black"))
+ self.gui.freqPlot.setCanvasBackground(QtGui.QColor("black"))
+ self.picker.setTrackerPen(Qt.QPen(whiteBrush, 2))
+ self.timeZoomer.setTrackerPen(Qt.QPen(whiteBrush, 2))
+ self.timeZoomer.setRubberBandPen(Qt.QPen(whiteBrush, 2))
+ self.freqZoomer.setTrackerPen(Qt.QPen(whiteBrush, 2))
+ self.freqZoomer.setRubberBandPen(Qt.QPen(whiteBrush, 2))
+ self.psdcurve.setPen(Qt.QPen(greenBrush, 1))
+ self.rcurve.setPen(Qt.QPen(greenBrush, 2))
+ self.icurve.setPen(Qt.QPen(redBrush, 2))
+
+ self.gui.timePlot.replot()
+ self.gui.freqPlot.replot()
+
+ def color_blue_on_black(self):
+ blue = QtGui.qRgb(0x00, 0x00, 0xFF)
+ red = QtGui.qRgb(0xFF, 0x00, 0x00)
+
+ whiteBrush = Qt.QBrush(Qt.QColor("white"))
+ blueBrush = Qt.QBrush(Qt.QColor(blue))
+ redBrush = Qt.QBrush(Qt.QColor(red))
+
+ self.gui.timePlot.setCanvasBackground(QtGui.QColor("black"))
+ self.gui.freqPlot.setCanvasBackground(QtGui.QColor("black"))
+ self.picker.setTrackerPen(Qt.QPen(whiteBrush, 2))
+ self.timeZoomer.setTrackerPen(Qt.QPen(whiteBrush, 2))
+ self.timeZoomer.setRubberBandPen(Qt.QPen(whiteBrush, 2))
+ self.freqZoomer.setTrackerPen(Qt.QPen(whiteBrush, 2))
+ self.freqZoomer.setRubberBandPen(Qt.QPen(whiteBrush, 2))
+ self.psdcurve.setPen(Qt.QPen(blueBrush, 1))
+ self.rcurve.setPen(Qt.QPen(blueBrush, 2))
+ self.icurve.setPen(Qt.QPen(redBrush, 2))
+
+ self.gui.timePlot.replot()
+ self.gui.freqPlot.replot()
+
+def setup_options():
+ usage="%prog: [options] (input_filename)"
+ description = ""
+
+ parser = OptionParser(conflict_handler="resolve", usage=usage, description=description)
+ parser.add_option("-B", "--block-length", type="int", default=8192,
+ help="Specify the block size [default=%default]")
+ parser.add_option("-s", "--start", type="int", default=0,
+ help="Specify where to start in the file [default=%default]")
+ parser.add_option("-R", "--sample-rate", type="float", default=1.0,
+ help="Set the sampler rate of the data [default=%default]")
+ parser.add_option("", "--psd-size", type="int", default=2048,
+ help="Set the size of the PSD FFT [default=%default]")
+ parser.add_option("", "--spec-size", type="int", default=256,
+ help="Set the size of the spectrogram FFT [default=%default]")
+
+ return parser
+
+def main(args):
+ parser = setup_options()
+ (options, args) = parser.parse_args ()
+
+ if(len(args) == 1):
+ filename = args[0]
+ else:
+ filename = None
+
+ app = Qt.QApplication(args)
+ gplt = gr_plot_qt(app, filename, options)
+ app.exec_()
+
+if __name__ == '__main__':
+ main(sys.argv)
+
--
cgit v1.2.3
From d4ba08af653a00af6530add2df14c1341d6a9b90 Mon Sep 17 00:00:00 2001
From: Tom Rondeau <trondeau@molly.home>
Date: Tue, 25 Aug 2009 12:12:17 -0400
Subject: GR plotter tool handles end of file and files shorter than the block
length.
---
gr-utils/src/python/gr_plot_qt.py | 34 +++++++++++++++++++++++++---------
1 file changed, 25 insertions(+), 9 deletions(-)
(limited to 'gr-utils/src/python/gr_plot_qt.py')
diff --git a/gr-utils/src/python/gr_plot_qt.py b/gr-utils/src/python/gr_plot_qt.py
index 7c744f2c69..e772151f0b 100755
--- a/gr-utils/src/python/gr_plot_qt.py
+++ b/gr-utils/src/python/gr_plot_qt.py
@@ -152,14 +152,20 @@ class gr_plot_qt(QtGui.QMainWindow):
def open_file(self):
filename = Qt.QFileDialog.getOpenFileName(self, "Open", ".")
- print filename
- self.initialize(filename)
+ if(filename != ""):
+ print filename
+ self.initialize(filename)
def initialize(self, filename):
self.hfile = open(filename, "r")
self.setWindowTitle(("GNU Radio File Plot Utility: %s" % filename))
-
+
+ self.gui.filePosStartLineEdit.setText("0")
+ self.gui.filePosStopLineEdit.setText("0")
+ self.gui.fileTimeStartLineEdit.setText("0")
+ self.gui.fileTimeStopLineEdit.setText("0")
+
self.cur_start = 0
self.cur_stop = self.block_length
@@ -169,6 +175,7 @@ class gr_plot_qt(QtGui.QMainWindow):
self.gui.plotHBar.setSliderPosition(0)
self.gui.plotHBar.setMaximum(self.signal_size)
+
self.update_time_curves()
self.update_psd_curves()
@@ -182,16 +189,24 @@ class gr_plot_qt(QtGui.QMainWindow):
if(end > start):
self.hfile.seek(start*self.sizeof_data, os.SEEK_SET)
self.position = start
- iq = scipy.fromfile(self.hfile, dtype=self.datatype,
- count=end-start)
- if(len(iq) < (end-start)):
- print "End of File"
- else:
+ try:
+ iq = scipy.fromfile(self.hfile, dtype=self.datatype,
+ count=end-start)
+
+ if(len(iq) < (end-start)):
+ end = len(iq)
+ self.gui.filePosLengthLineEdit.setText(Qt.QString("%1").arg(end))
+ self.gui.plotHBar.setMaximum(end)
+ self.gui.plotHBar.setSingleStep(end)
+ self.file_length_changed()
+
tstep = 1.0 / self.sample_rate
self.iq = iq
self.time = [tstep*(self.position + i) for i in xrange(len(self.iq))]
- self.set_file_pos_box(start, end)
+ self.set_file_pos_box(start, end)
+ except MemoryError:
+ pass
else:
# Do we want to do anything about this?
pass
@@ -296,6 +311,7 @@ class gr_plot_qt(QtGui.QMainWindow):
def file_length_changed(self):
start = self.gui.filePosStartLineEdit.text().toInt()
length = self.gui.filePosLengthLineEdit.text().toInt()
+
if((start[1] == True) and (length[1] == True)):
self.cur_start = start[0]
self.block_length = length[0]
--
cgit v1.2.3
From f87944705b9fb251af9fe0c16ef2b47424c867d5 Mon Sep 17 00:00:00 2001
From: Tom Rondeau <trondeau@molly.home>
Date: Tue, 25 Aug 2009 17:35:56 -0400
Subject: Adding a spectrogram plot. The axis need work.
---
gr-utils/src/python/gr_plot_qt.py | 117 +++++++++++++--
gr-utils/src/python/pyqt_plot.py | 56 +++++--
gr-utils/src/python/pyqt_plot.ui | 307 ++++++++++++++++++++++----------------
3 files changed, 328 insertions(+), 152 deletions(-)
(limited to 'gr-utils/src/python/gr_plot_qt.py')
diff --git a/gr-utils/src/python/gr_plot_qt.py b/gr-utils/src/python/gr_plot_qt.py
index e772151f0b..958be19caf 100755
--- a/gr-utils/src/python/gr_plot_qt.py
+++ b/gr-utils/src/python/gr_plot_qt.py
@@ -16,6 +16,36 @@ from gnuradio import eng_notation
from pyqt_plot import Ui_MainWindow
+class SpectrogramData(Qwt.QwtRasterData):
+
+ def __init__(self, f, t):
+ Qwt.QwtArrayData.__init__(self, Qt.QRectF(0, 0, 0, 0))
+ self.sp = scipy.array([[0], [0]])
+
+ def set_data(self, xfreq, ytime, data):
+ self.sp = data
+ self.freq = xfreq
+ self.time = ytime
+ boundingBox = Qt.QRectF(0, 0, self.freq.size, self.time.size)
+ self.setBoundingRect(boundingBox)
+
+ def rasterHint(self, rect):
+ return Qt.QSize(self.sp.shape[0], self.sp.shape[1])
+
+ def copy(self):
+ return self
+
+ def range(self):
+
+ return Qwt.QwtDoubleInterval(self.sp.min(), self.sp.max())
+
+ def value(self, x, y):
+ #print x, y
+ x = int(x)
+ y = int(y)
+ return self.sp[x][y-1]
+
+
class gr_plot_qt(QtGui.QMainWindow):
def __init__(self, qapp, filename, options, parent=None):
QtGui.QWidget.__init__(self, parent)
@@ -32,7 +62,7 @@ class gr_plot_qt(QtGui.QMainWindow):
self.datatype = scipy.complex64
self.iq = list()
self.time = list()
-
+
# Set up basic plot attributes
self.gui.timePlot.setAxisTitle(self.gui.timePlot.xBottom, "Time (sec)")
self.gui.timePlot.setAxisTitle(self.gui.timePlot.yLeft, "Amplitude (V)")
@@ -40,13 +70,21 @@ class gr_plot_qt(QtGui.QMainWindow):
self.gui.freqPlot.setAxisTitle(self.gui.freqPlot.yLeft, "Magnitude (dB)")
# Set up FFT size combo box
- self.gui.fftComboBox.addItems(["128", "256", "512", "1024", "2048",
- "4096", "8192", "16384", "32768"])
- pos = self.gui.fftComboBox.findText(Qt.QString("%1").arg(self.psdfftsize))
- self.gui.fftComboBox.setCurrentIndex(pos)
- self.connect(self.gui.fftComboBox,
+ self.fftsizes = ["128", "256", "512", "1024", "2048",
+ "4096", "8192", "16384", "32768"]
+ self.gui.psdFFTComboBox.addItems(self.fftsizes)
+ self.gui.specFFTComboBox.addItems(self.fftsizes)
+ pos = self.gui.psdFFTComboBox.findText(Qt.QString("%1").arg(self.psdfftsize))
+ self.gui.psdFFTComboBox.setCurrentIndex(pos)
+ pos = self.gui.specFFTComboBox.findText(Qt.QString("%1").arg(self.specfftsize))
+ self.gui.specFFTComboBox.setCurrentIndex(pos)
+
+ self.connect(self.gui.psdFFTComboBox,
+ Qt.SIGNAL("activated (const QString&)"),
+ self.psdFFTComboBoxEdit)
+ self.connect(self.gui.specFFTComboBox,
Qt.SIGNAL("activated (const QString&)"),
- self.fftComboBoxEdit)
+ self.specFFTComboBoxEdit)
# Set up color scheme box
self.color_modes = {"Black on White" : self.color_black_on_white,
@@ -74,6 +112,12 @@ class gr_plot_qt(QtGui.QMainWindow):
Qwt.QwtPicker.AlwaysOn,
self.gui.freqPlot.canvas())
+ self.specZoomer = Qwt.QwtPlotZoomer(self.gui.specPlot.xBottom,
+ self.gui.specPlot.yLeft,
+ Qwt.QwtPicker.PointSelection,
+ Qwt.QwtPicker.AlwaysOn,
+ self.gui.specPlot.canvas())
+
self.picker = Qwt.QwtPlotPicker(self.gui.timePlot.xBottom,
self.gui.timePlot.yLeft,
Qwt.QwtPicker.PointSelection,
@@ -137,6 +181,21 @@ class gr_plot_qt(QtGui.QMainWindow):
self.psdcurve = Qwt.QwtPlotCurve("PSD")
self.psdcurve.attach(self.gui.freqPlot)
+ # Set up specTab plot as a spectrogram
+ self.specdata = SpectrogramData(range(0, 10), range(0, 10))
+
+ colorMap = Qwt.QwtLinearColorMap(Qt.Qt.darkCyan, Qt.Qt.red)
+ colorMap.addColorStop(0.1, Qt.Qt.cyan)
+ colorMap.addColorStop(0.6, Qt.Qt.green)
+ colorMap.addColorStop(0.95, Qt.Qt.yellow)
+
+ self.spec = Qwt.QwtPlotSpectrogram()
+ self.spec.setColorMap(colorMap)
+ self.spec.attach(self.gui.specPlot)
+ self.spec.setContourLevels([0.5, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5, 8.5, 9.5])
+ self.spec.setDisplayMode(Qwt.QwtPlotSpectrogram.ImageMode, True)
+ self.spec.setData(self.specdata)
+
# Set up initial color scheme
self.color_modes["Blue on Black"]()
@@ -172,12 +231,14 @@ class gr_plot_qt(QtGui.QMainWindow):
self.init_data_input()
self.get_data(self.cur_start, self.cur_stop)
self.get_psd()
+ self.get_specgram()
self.gui.plotHBar.setSliderPosition(0)
self.gui.plotHBar.setMaximum(self.signal_size)
self.update_time_curves()
self.update_psd_curves()
+ self.update_specgram_curves()
def init_data_input(self):
self.hfile.seek(0, os.SEEK_END)
@@ -218,17 +279,34 @@ class gr_plot_qt(QtGui.QMainWindow):
noverlap=self.psdfftsize/4.0,
window=winpoints,
scale_by_freq=False)
+
self.iq_psd = 10.0*scipy.log10(abs(fftpack.fftshift(iq_psd)))
self.freq = freq - self.sample_rate/2.0
+ def get_specgram(self):
+ winpoints = self.winfunc(self.specfftsize)
+ iq_spec, f, t = mlab.specgram(self.iq, Fs=self.sample_rate,
+ NFFT=self.specfftsize,
+ noverlap=self.specfftsize/4.0,
+ window=winpoints,
+ scale_by_freq=False)
+
+ self.iq_spec = 10.0*scipy.log10(abs(iq_spec))
+ self.spec_f = f
+ self.spec_t = t
def clickMe(self, qPoint):
print qPoint.x()
- def fftComboBoxEdit(self, fftSize):
+ def psdFFTComboBoxEdit(self, fftSize):
self.psdfftsize = fftSize.toInt()[0]
self.get_psd()
self.update_psd_curves()
+
+ def specFFTComboBoxEdit(self, fftSize):
+ self.specfftsize = fftSize.toInt()[0]
+ self.get_specgram()
+ self.update_specgram_curves()
def colorComboBoxEdit(self, colorSelection):
colorstr = str(colorSelection.toAscii())
@@ -241,8 +319,10 @@ class gr_plot_qt(QtGui.QMainWindow):
self.get_data(pos_start, pos_end)
self.get_psd()
+ self.get_specgram()
self.update_time_curves()
self.update_psd_curves()
+ self.update_specgram_curves()
def set_sample_rate(self, sr):
self.sample_rate = sr
@@ -255,6 +335,7 @@ class gr_plot_qt(QtGui.QMainWindow):
self.set_file_pos_box(self.cur_start, self.cur_stop)
self.get_data(self.cur_start, self.cur_stop)
self.get_psd()
+ self.get_specgram()
self.update_time_curves()
self.update_psd_curves()
@@ -286,6 +367,7 @@ class gr_plot_qt(QtGui.QMainWindow):
self.update_time_curves()
self.update_psd_curves()
+ self.update_specgram_curves()
# If there's a non-digit character, reset box
else:
@@ -304,6 +386,7 @@ class gr_plot_qt(QtGui.QMainWindow):
self.update_time_curves()
self.update_psd_curves()
+ self.update_specgram_curves()
# If there's a non-digit character, reset box
else:
self.set_file_pos_box(self.cur_start, self.cur_stop)
@@ -328,9 +411,11 @@ class gr_plot_qt(QtGui.QMainWindow):
self.get_data(self.cur_start, self.cur_stop)
self.get_psd()
+ self.get_specgram()
self.update_time_curves()
self.update_psd_curves()
+ self.update_specgram_curves()
# If there's a non-digit character, reset box
else:
self.set_file_pos_box(self.cur_start, self.cur_stop)
@@ -352,9 +437,11 @@ class gr_plot_qt(QtGui.QMainWindow):
self.get_data(self.cur_start, self.cur_stop)
self.get_psd()
+ self.get_specgram()
self.update_time_curves()
self.update_psd_curves()
+ self.update_specgram_curves()
# If there's a non-digit character, reset box
else:
self.set_file_pos_box(self.cur_start, self.cur_stop)
@@ -385,12 +472,22 @@ class gr_plot_qt(QtGui.QMainWindow):
self.gui.freqPlot.setAxisScale(self.gui.freqPlot.xBottom,
min(self.freq),
max(self.freq))
-
+
# Set the zoomer base to unzoom to the new axis
self.freqZoomer.setZoomBase()
self.gui.freqPlot.replot()
+ def update_specgram_curves(self):
+ # We don't have to reset the data for the speccurve here
+ # since this is taken care of in the SpectrogramData class
+ self.specdata.set_data(self.spec_f, self.spec_t, self.iq_spec)
+
+ # Set the zoomer base to unzoom to the new axis
+ self.specZoomer.setZoomBase()
+
+ self.gui.specPlot.replot()
+
def tabChanged(self, index):
self.gui.timePlot.replot()
self.gui.freqPlot.replot()
@@ -497,7 +594,7 @@ def setup_options():
help="Set the sampler rate of the data [default=%default]")
parser.add_option("", "--psd-size", type="int", default=2048,
help="Set the size of the PSD FFT [default=%default]")
- parser.add_option("", "--spec-size", type="int", default=256,
+ parser.add_option("", "--spec-size", type="int", default=2048,
help="Set the size of the spectrogram FFT [default=%default]")
return parser
diff --git a/gr-utils/src/python/pyqt_plot.py b/gr-utils/src/python/pyqt_plot.py
index 796b6a238d..2724d85461 100644
--- a/gr-utils/src/python/pyqt_plot.py
+++ b/gr-utils/src/python/pyqt_plot.py
@@ -2,8 +2,8 @@
# Form implementation generated from reading ui file 'pyqt_plot.ui'
#
-# Created: Tue Aug 11 23:12:27 2009
-# by: PyQt4 UI code generator 4.4.4
+# Created: Tue Aug 25 15:28:55 2009
+# by: PyQt4 UI code generator 4.4.3
#
# WARNING! All changes made in this file will be lost!
@@ -38,22 +38,43 @@ class Ui_MainWindow(object):
self.fftPropBox = QtGui.QGroupBox(self.freqTab)
self.fftPropBox.setMinimumSize(QtCore.QSize(160, 0))
self.fftPropBox.setObjectName("fftPropBox")
- self.formLayoutWidget = QtGui.QWidget(self.fftPropBox)
- self.formLayoutWidget.setGeometry(QtCore.QRect(0, 20, 151, 191))
- self.formLayoutWidget.setObjectName("formLayoutWidget")
- self.formLayout = QtGui.QFormLayout(self.formLayoutWidget)
- self.formLayout.setObjectName("formLayout")
- self.fftSizeLabel = QtGui.QLabel(self.formLayoutWidget)
- self.fftSizeLabel.setObjectName("fftSizeLabel")
- self.formLayout.setWidget(0, QtGui.QFormLayout.LabelRole, self.fftSizeLabel)
- self.fftComboBox = QtGui.QComboBox(self.formLayoutWidget)
- self.fftComboBox.setObjectName("fftComboBox")
- self.formLayout.setWidget(0, QtGui.QFormLayout.FieldRole, self.fftComboBox)
+ self.formLayout_4 = QtGui.QFormLayout(self.fftPropBox)
+ self.formLayout_4.setFieldGrowthPolicy(QtGui.QFormLayout.AllNonFixedFieldsGrow)
+ self.formLayout_4.setObjectName("formLayout_4")
+ self.psdFFTComboBox = QtGui.QComboBox(self.fftPropBox)
+ self.psdFFTComboBox.setMinimumSize(QtCore.QSize(96, 0))
+ self.psdFFTComboBox.setMaximumSize(QtCore.QSize(96, 16777215))
+ self.psdFFTComboBox.setObjectName("psdFFTComboBox")
+ self.formLayout_4.setWidget(0, QtGui.QFormLayout.FieldRole, self.psdFFTComboBox)
+ self.psdFFTSizeLabel = QtGui.QLabel(self.fftPropBox)
+ self.psdFFTSizeLabel.setObjectName("psdFFTSizeLabel")
+ self.formLayout_4.setWidget(0, QtGui.QFormLayout.LabelRole, self.psdFFTSizeLabel)
self.horizontalLayout_2.addWidget(self.fftPropBox)
self.freqPlot = Qwt5.QwtPlot(self.freqTab)
self.freqPlot.setObjectName("freqPlot")
self.horizontalLayout_2.addWidget(self.freqPlot)
self.tabGroup.addTab(self.freqTab, "")
+ self.specTab = QtGui.QWidget()
+ self.specTab.setObjectName("specTab")
+ self.horizontalLayout_3 = QtGui.QHBoxLayout(self.specTab)
+ self.horizontalLayout_3.setObjectName("horizontalLayout_3")
+ self.groupBox = QtGui.QGroupBox(self.specTab)
+ self.groupBox.setObjectName("groupBox")
+ self.formLayout_3 = QtGui.QFormLayout(self.groupBox)
+ self.formLayout_3.setObjectName("formLayout_3")
+ self.specFFTLabel = QtGui.QLabel(self.groupBox)
+ self.specFFTLabel.setObjectName("specFFTLabel")
+ self.formLayout_3.setWidget(1, QtGui.QFormLayout.LabelRole, self.specFFTLabel)
+ self.specFFTComboBox = QtGui.QComboBox(self.groupBox)
+ self.specFFTComboBox.setMinimumSize(QtCore.QSize(96, 0))
+ self.specFFTComboBox.setMaximumSize(QtCore.QSize(96, 16777215))
+ self.specFFTComboBox.setObjectName("specFFTComboBox")
+ self.formLayout_3.setWidget(1, QtGui.QFormLayout.FieldRole, self.specFFTComboBox)
+ self.horizontalLayout_3.addWidget(self.groupBox)
+ self.specPlot = Qwt5.QwtPlot(self.specTab)
+ self.specPlot.setObjectName("specPlot")
+ self.horizontalLayout_3.addWidget(self.specPlot)
+ self.tabGroup.addTab(self.specTab, "")
self.gridLayout.addWidget(self.tabGroup, 1, 0, 1, 1)
self.filePosBox = QtGui.QGroupBox(self.centralwidget)
self.filePosBox.setMinimumSize(QtCore.QSize(0, 120))
@@ -137,7 +158,7 @@ class Ui_MainWindow(object):
self.gridLayout.addWidget(self.filePosBox, 3, 0, 1, 1)
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtGui.QMenuBar(MainWindow)
- self.menubar.setGeometry(QtCore.QRect(0, 0, 927, 25))
+ self.menubar.setGeometry(QtCore.QRect(0, 0, 927, 24))
self.menubar.setObjectName("menubar")
self.menu_File = QtGui.QMenu(self.menubar)
self.menu_File.setObjectName("menu_File")
@@ -154,7 +175,7 @@ class Ui_MainWindow(object):
self.menubar.addAction(self.menu_File.menuAction())
self.retranslateUi(MainWindow)
- self.tabGroup.setCurrentIndex(0)
+ self.tabGroup.setCurrentIndex(2)
QtCore.QObject.connect(self.action_exit, QtCore.SIGNAL("activated()"), MainWindow.close)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
@@ -162,8 +183,11 @@ class Ui_MainWindow(object):
MainWindow.setWindowTitle(QtGui.QApplication.translate("MainWindow", "MainWindow", None, QtGui.QApplication.UnicodeUTF8))
self.tabGroup.setTabText(self.tabGroup.indexOf(self.timeTab), QtGui.QApplication.translate("MainWindow", "Time Domain", None, QtGui.QApplication.UnicodeUTF8))
self.fftPropBox.setTitle(QtGui.QApplication.translate("MainWindow", "FFT Properties", None, QtGui.QApplication.UnicodeUTF8))
- self.fftSizeLabel.setText(QtGui.QApplication.translate("MainWindow", "FFT Size", None, QtGui.QApplication.UnicodeUTF8))
+ self.psdFFTSizeLabel.setText(QtGui.QApplication.translate("MainWindow", "FFT Size", None, QtGui.QApplication.UnicodeUTF8))
self.tabGroup.setTabText(self.tabGroup.indexOf(self.freqTab), QtGui.QApplication.translate("MainWindow", "Frequency Domain", None, QtGui.QApplication.UnicodeUTF8))
+ self.groupBox.setTitle(QtGui.QApplication.translate("MainWindow", "Spectrogram Properties", None, QtGui.QApplication.UnicodeUTF8))
+ self.specFFTLabel.setText(QtGui.QApplication.translate("MainWindow", "FFT Size", None, QtGui.QApplication.UnicodeUTF8))
+ self.tabGroup.setTabText(self.tabGroup.indexOf(self.specTab), QtGui.QApplication.translate("MainWindow", "Spectrogram", None, QtGui.QApplication.UnicodeUTF8))
self.filePosBox.setTitle(QtGui.QApplication.translate("MainWindow", "File Position", None, QtGui.QApplication.UnicodeUTF8))
self.filePosStartLabel.setText(QtGui.QApplication.translate("MainWindow", "Start", None, QtGui.QApplication.UnicodeUTF8))
self.filePosStopLabel.setText(QtGui.QApplication.translate("MainWindow", "Stop", None, QtGui.QApplication.UnicodeUTF8))
diff --git a/gr-utils/src/python/pyqt_plot.ui b/gr-utils/src/python/pyqt_plot.ui
index f298735c41..62125962b6 100644
--- a/gr-utils/src/python/pyqt_plot.ui
+++ b/gr-utils/src/python/pyqt_plot.ui
@@ -1,8 +1,7 @@
-<?xml version="1.0" encoding="UTF-8"?>
-<ui version="4.0">
+<ui version="4.0" >
<class>MainWindow</class>
- <widget class="QMainWindow" name="MainWindow">
- <property name="geometry">
+ <widget class="QMainWindow" name="MainWindow" >
+ <property name="geometry" >
<rect>
<x>0</x>
<y>0</y>
@@ -10,93 +9,149 @@
<height>696</height>
</rect>
</property>
- <property name="windowTitle">
+ <property name="windowTitle" >
<string>MainWindow</string>
</property>
- <widget class="QWidget" name="centralwidget">
- <layout class="QGridLayout" name="gridLayout">
- <item row="2" column="0" colspan="2">
- <widget class="QScrollBar" name="plotHBar">
- <property name="orientation">
+ <widget class="QWidget" name="centralwidget" >
+ <layout class="QGridLayout" name="gridLayout" >
+ <item row="2" column="0" colspan="2" >
+ <widget class="QScrollBar" name="plotHBar" >
+ <property name="orientation" >
<enum>Qt::Horizontal</enum>
</property>
</widget>
</item>
- <item row="1" column="0">
- <widget class="QTabWidget" name="tabGroup">
- <property name="currentIndex">
- <number>0</number>
+ <item row="1" column="0" >
+ <widget class="QTabWidget" name="tabGroup" >
+ <property name="currentIndex" >
+ <number>2</number>
</property>
- <widget class="QWidget" name="timeTab">
- <attribute name="title">
+ <widget class="QWidget" name="timeTab" >
+ <attribute name="title" >
<string>Time Domain</string>
</attribute>
- <layout class="QHBoxLayout" name="horizontalLayout">
+ <layout class="QHBoxLayout" name="horizontalLayout" >
<item>
- <widget class="QwtPlot" name="timePlot"/>
+ <widget class="QwtPlot" name="timePlot" />
</item>
</layout>
</widget>
- <widget class="QWidget" name="freqTab">
- <attribute name="title">
+ <widget class="QWidget" name="freqTab" >
+ <attribute name="title" >
<string>Frequency Domain</string>
</attribute>
- <layout class="QHBoxLayout" name="horizontalLayout_2">
+ <layout class="QHBoxLayout" name="horizontalLayout_2" >
<item>
- <widget class="QGroupBox" name="fftPropBox">
- <property name="minimumSize">
+ <widget class="QGroupBox" name="fftPropBox" >
+ <property name="minimumSize" >
<size>
<width>160</width>
<height>0</height>
</size>
</property>
- <property name="title">
+ <property name="title" >
<string>FFT Properties</string>
</property>
- <widget class="QWidget" name="formLayoutWidget">
- <property name="geometry">
- <rect>
- <x>0</x>
- <y>20</y>
- <width>151</width>
- <height>191</height>
- </rect>
+ <layout class="QFormLayout" name="formLayout_4" >
+ <property name="fieldGrowthPolicy" >
+ <enum>QFormLayout::AllNonFixedFieldsGrow</enum>
</property>
- <layout class="QFormLayout" name="formLayout">
- <item row="0" column="0">
- <widget class="QLabel" name="fftSizeLabel">
- <property name="text">
- <string>FFT Size</string>
- </property>
- </widget>
- </item>
- <item row="0" column="1">
- <widget class="QComboBox" name="fftComboBox"/>
- </item>
- </layout>
- </widget>
+ <item row="0" column="1" >
+ <widget class="QComboBox" name="psdFFTComboBox" >
+ <property name="minimumSize" >
+ <size>
+ <width>96</width>
+ <height>0</height>
+ </size>
+ </property>
+ <property name="maximumSize" >
+ <size>
+ <width>96</width>
+ <height>16777215</height>
+ </size>
+ </property>
+ </widget>
+ </item>
+ <item row="0" column="0" >
+ <widget class="QLabel" name="psdFFTSizeLabel" >
+ <property name="text" >
+ <string>FFT Size</string>
+ </property>
+ </widget>
+ </item>
+ </layout>
+ <zorder>formLayoutWidget</zorder>
+ <zorder>psdFFTComboBox</zorder>
+ <zorder>psdFFTSizeLabel</zorder>
+ </widget>
+ </item>
+ <item>
+ <widget class="QwtPlot" name="freqPlot" />
+ </item>
+ </layout>
+ </widget>
+ <widget class="QWidget" name="specTab" >
+ <attribute name="title" >
+ <string>Spectrogram</string>
+ </attribute>
+ <layout class="QHBoxLayout" name="horizontalLayout_3" >
+ <item>
+ <widget class="QGroupBox" name="groupBox" >
+ <property name="title" >
+ <string>Spectrogram Properties</string>
+ </property>
+ <layout class="QFormLayout" name="formLayout_3" >
+ <item row="1" column="0" >
+ <widget class="QLabel" name="specFFTLabel" >
+ <property name="text" >
+ <string>FFT Size</string>
+ </property>
+ </widget>
+ </item>
+ <item row="1" column="1" >
+ <widget class="QComboBox" name="specFFTComboBox" >
+ <property name="minimumSize" >
+ <size>
+ <width>96</width>
+ <height>0</height>
+ </size>
+ </property>
+ <property name="maximumSize" >
+ <size>
+ <width>96</width>
+ <height>16777215</height>
+ </size>
+ </property>
+ </widget>
+ </item>
+ </layout>
+ <zorder>specFFTLabel</zorder>
+ <zorder>specFFTComboBox</zorder>
+ <zorder>specFFTComboBox</zorder>
</widget>
</item>
<item>
- <widget class="QwtPlot" name="freqPlot"/>
+ <widget class="QwtPlot" name="specPlot" />
</item>
</layout>
+ <zorder>specPlot</zorder>
+ <zorder>groupBox</zorder>
</widget>
</widget>
</item>
- <item row="3" column="0">
- <widget class="QGroupBox" name="filePosBox">
- <property name="minimumSize">
+ <item row="3" column="0" >
+ <widget class="QGroupBox" name="filePosBox" >
+ <property name="minimumSize" >
<size>
<width>0</width>
<height>120</height>
</size>
</property>
- <property name="title">
+ <property name="title" >
<string>File Position</string>
</property>
- <widget class="QWidget" name="formLayoutWidget_2">
- <property name="geometry">
+ <widget class="QWidget" name="formLayoutWidget_2" >
+ <property name="geometry" >
<rect>
<x>0</x>
<y>20</y>
@@ -104,41 +159,41 @@
<height>92</height>
</rect>
</property>
- <layout class="QFormLayout" name="filePosLayout">
- <item row="0" column="0">
- <widget class="QLabel" name="filePosStartLabel">
- <property name="text">
+ <layout class="QFormLayout" name="filePosLayout" >
+ <item row="0" column="0" >
+ <widget class="QLabel" name="filePosStartLabel" >
+ <property name="text" >
<string>Start</string>
</property>
</widget>
</item>
- <item row="0" column="1">
- <widget class="QLineEdit" name="filePosStartLineEdit"/>
+ <item row="0" column="1" >
+ <widget class="QLineEdit" name="filePosStartLineEdit" />
</item>
- <item row="1" column="0">
- <widget class="QLabel" name="filePosStopLabel">
- <property name="text">
+ <item row="1" column="0" >
+ <widget class="QLabel" name="filePosStopLabel" >
+ <property name="text" >
<string>Stop</string>
</property>
</widget>
</item>
- <item row="1" column="1">
- <widget class="QLineEdit" name="filePosStopLineEdit"/>
+ <item row="1" column="1" >
+ <widget class="QLineEdit" name="filePosStopLineEdit" />
</item>
- <item row="2" column="0">
- <widget class="QLabel" name="filePosLengthLabel">
- <property name="text">
+ <item row="2" column="0" >
+ <widget class="QLabel" name="filePosLengthLabel" >
+ <property name="text" >
<string>Length</string>
</property>
</widget>
</item>
- <item row="2" column="1">
- <widget class="QLineEdit" name="filePosLengthLineEdit"/>
+ <item row="2" column="1" >
+ <widget class="QLineEdit" name="filePosLengthLineEdit" />
</item>
</layout>
</widget>
- <widget class="QWidget" name="formLayoutWidget_4">
- <property name="geometry">
+ <widget class="QWidget" name="formLayoutWidget_4" >
+ <property name="geometry" >
<rect>
<x>180</x>
<y>20</y>
@@ -146,41 +201,41 @@
<height>92</height>
</rect>
</property>
- <layout class="QFormLayout" name="fileTimeLayout">
- <item row="0" column="0">
- <widget class="QLabel" name="fileTimeStartLabel">
- <property name="text">
+ <layout class="QFormLayout" name="fileTimeLayout" >
+ <item row="0" column="0" >
+ <widget class="QLabel" name="fileTimeStartLabel" >
+ <property name="text" >
<string>time start (sec)</string>
</property>
</widget>
</item>
- <item row="0" column="1">
- <widget class="QLineEdit" name="fileTimeStartLineEdit"/>
+ <item row="0" column="1" >
+ <widget class="QLineEdit" name="fileTimeStartLineEdit" />
</item>
- <item row="1" column="0">
- <widget class="QLabel" name="fileTimeStopLabel">
- <property name="text">
+ <item row="1" column="0" >
+ <widget class="QLabel" name="fileTimeStopLabel" >
+ <property name="text" >
<string>time stop (sec)</string>
</property>
</widget>
</item>
- <item row="1" column="1">
- <widget class="QLineEdit" name="fileTimeStopLineEdit"/>
+ <item row="1" column="1" >
+ <widget class="QLineEdit" name="fileTimeStopLineEdit" />
</item>
- <item row="2" column="0">
- <widget class="QLabel" name="fileTimeLengthLabel">
- <property name="text">
+ <item row="2" column="0" >
+ <widget class="QLabel" name="fileTimeLengthLabel" >
+ <property name="text" >
<string>time length (sec)</string>
</property>
</widget>
</item>
- <item row="2" column="1">
- <widget class="QLineEdit" name="fileTimeLengthLineEdit"/>
+ <item row="2" column="1" >
+ <widget class="QLineEdit" name="fileTimeLengthLineEdit" />
</item>
</layout>
</widget>
- <widget class="QGroupBox" name="sysGroupBox">
- <property name="geometry">
+ <widget class="QGroupBox" name="sysGroupBox" >
+ <property name="geometry" >
<rect>
<x>530</x>
<y>0</y>
@@ -188,17 +243,17 @@
<height>120</height>
</rect>
</property>
- <property name="minimumSize">
+ <property name="minimumSize" >
<size>
<width>200</width>
<height>0</height>
</size>
</property>
- <property name="title">
+ <property name="title" >
<string>System Properties</string>
</property>
- <widget class="QWidget" name="formLayoutWidget_3">
- <property name="geometry">
+ <widget class="QWidget" name="formLayoutWidget_3" >
+ <property name="geometry" >
<rect>
<x>0</x>
<y>20</y>
@@ -206,17 +261,17 @@
<height>91</height>
</rect>
</property>
- <layout class="QFormLayout" name="formLayout_2">
- <item row="0" column="0">
- <widget class="QLabel" name="sampleRateLabel">
- <property name="text">
+ <layout class="QFormLayout" name="formLayout_2" >
+ <item row="0" column="0" >
+ <widget class="QLabel" name="sampleRateLabel" >
+ <property name="text" >
<string>Sample Rate</string>
</property>
</widget>
</item>
- <item row="0" column="1">
- <widget class="QLineEdit" name="sampleRateLineEdit">
- <property name="minimumSize">
+ <item row="0" column="1" >
+ <widget class="QLineEdit" name="sampleRateLineEdit" >
+ <property name="minimumSize" >
<size>
<width>0</width>
<height>0</height>
@@ -227,8 +282,8 @@
</layout>
</widget>
</widget>
- <widget class="QGroupBox" name="displayGroupBox">
- <property name="geometry">
+ <widget class="QGroupBox" name="displayGroupBox" >
+ <property name="geometry" >
<rect>
<x>730</x>
<y>0</y>
@@ -236,17 +291,17 @@
<height>120</height>
</rect>
</property>
- <property name="minimumSize">
+ <property name="minimumSize" >
<size>
<width>170</width>
<height>0</height>
</size>
</property>
- <property name="title">
+ <property name="title" >
<string>Display Properties</string>
</property>
- <widget class="QWidget" name="verticalLayoutWidget">
- <property name="geometry">
+ <widget class="QWidget" name="verticalLayoutWidget" >
+ <property name="geometry" >
<rect>
<x>0</x>
<y>20</y>
@@ -254,16 +309,16 @@
<height>91</height>
</rect>
</property>
- <layout class="QVBoxLayout" name="verticalLayout">
+ <layout class="QVBoxLayout" name="verticalLayout" >
<item>
- <widget class="QComboBox" name="colorComboBox"/>
+ <widget class="QComboBox" name="colorComboBox" />
</item>
<item>
- <spacer name="verticalSpacer">
- <property name="orientation">
+ <spacer name="verticalSpacer" >
+ <property name="orientation" >
<enum>Qt::Vertical</enum>
</property>
- <property name="sizeHint" stdset="0">
+ <property name="sizeHint" stdset="0" >
<size>
<width>20</width>
<height>40</height>
@@ -278,35 +333,35 @@
</item>
</layout>
</widget>
- <widget class="QMenuBar" name="menubar">
- <property name="geometry">
+ <widget class="QMenuBar" name="menubar" >
+ <property name="geometry" >
<rect>
<x>0</x>
<y>0</y>
<width>927</width>
- <height>25</height>
+ <height>24</height>
</rect>
</property>
- <widget class="QMenu" name="menu_File">
- <property name="title">
+ <widget class="QMenu" name="menu_File" >
+ <property name="title" >
<string>&File</string>
</property>
- <addaction name="action_open"/>
- <addaction name="action_exit"/>
+ <addaction name="action_open" />
+ <addaction name="action_exit" />
</widget>
- <addaction name="menu_File"/>
+ <addaction name="menu_File" />
</widget>
- <widget class="QStatusBar" name="statusbar"/>
- <action name="action_open">
- <property name="text">
+ <widget class="QStatusBar" name="statusbar" />
+ <action name="action_open" >
+ <property name="text" >
<string>&Open</string>
</property>
- <property name="shortcut">
+ <property name="shortcut" >
<string>Ctrl+O</string>
</property>
</action>
- <action name="action_exit">
- <property name="text">
+ <action name="action_exit" >
+ <property name="text" >
<string>E&xit</string>
</property>
</action>
@@ -326,11 +381,11 @@
<receiver>MainWindow</receiver>
<slot>close()</slot>
<hints>
- <hint type="sourcelabel">
+ <hint type="sourcelabel" >
<x>-1</x>
<y>-1</y>
</hint>
- <hint type="destinationlabel">
+ <hint type="destinationlabel" >
<x>399</x>
<y>347</y>
</hint>
--
cgit v1.2.3
From ab88e3c4fcf96093637eaadeae360df0d04282fa Mon Sep 17 00:00:00 2001
From: Tom Rondeau <trondeau@molly.home>
Date: Tue, 25 Aug 2009 17:54:36 -0400
Subject: Fixed spectrogram plotting axis.
---
gr-utils/src/python/gr_plot_qt.py | 22 +++++++++++++++++-----
1 file changed, 17 insertions(+), 5 deletions(-)
(limited to 'gr-utils/src/python/gr_plot_qt.py')
diff --git a/gr-utils/src/python/gr_plot_qt.py b/gr-utils/src/python/gr_plot_qt.py
index 958be19caf..5579bbce29 100755
--- a/gr-utils/src/python/gr_plot_qt.py
+++ b/gr-utils/src/python/gr_plot_qt.py
@@ -26,7 +26,9 @@ class SpectrogramData(Qwt.QwtRasterData):
self.sp = data
self.freq = xfreq
self.time = ytime
- boundingBox = Qt.QRectF(0, 0, self.freq.size, self.time.size)
+ boundingBox = Qt.QRectF(self.freq.min(), self.time.min(),
+ self.freq.max() - self.freq.min(),
+ self.time.max() - self.time.min())
self.setBoundingRect(boundingBox)
def rasterHint(self, rect):
@@ -40,10 +42,9 @@ class SpectrogramData(Qwt.QwtRasterData):
return Qwt.QwtDoubleInterval(self.sp.min(), self.sp.max())
def value(self, x, y):
- #print x, y
- x = int(x)
- y = int(y)
- return self.sp[x][y-1]
+ f = int(self.freq.searchsorted(x))
+ t = int(self.time.searchsorted(y))
+ return self.sp[f][t-1]
class gr_plot_qt(QtGui.QMainWindow):
@@ -68,6 +69,8 @@ class gr_plot_qt(QtGui.QMainWindow):
self.gui.timePlot.setAxisTitle(self.gui.timePlot.yLeft, "Amplitude (V)")
self.gui.freqPlot.setAxisTitle(self.gui.freqPlot.xBottom, "Frequency (Hz)")
self.gui.freqPlot.setAxisTitle(self.gui.freqPlot.yLeft, "Magnitude (dB)")
+ self.gui.specPlot.setAxisTitle(self.gui.specPlot.xBottom, "Frequency (Hz)")
+ self.gui.specPlot.setAxisTitle(self.gui.specPlot.yLeft, "Time (sec)")
# Set up FFT size combo box
self.fftsizes = ["128", "256", "512", "1024", "2048",
@@ -199,6 +202,7 @@ class gr_plot_qt(QtGui.QMainWindow):
# Set up initial color scheme
self.color_modes["Blue on Black"]()
+ # Connect a signal for when the sample rate changes
self.set_sample_rate(self.sample_rate)
self.connect(self.gui.sampleRateLineEdit,
Qt.SIGNAL("editingFinished()"),
@@ -338,6 +342,7 @@ class gr_plot_qt(QtGui.QMainWindow):
self.get_specgram()
self.update_time_curves()
self.update_psd_curves()
+ self.update_specgram_curves()
def set_file_pos_box(self, start, end):
tstart = start / self.sample_rate
@@ -483,6 +488,13 @@ class gr_plot_qt(QtGui.QMainWindow):
# since this is taken care of in the SpectrogramData class
self.specdata.set_data(self.spec_f, self.spec_t, self.iq_spec)
+ self.gui.specPlot.setAxisScale(self.gui.specPlot.xBottom,
+ min(self.spec_f),
+ max(self.spec_f))
+ self.gui.specPlot.setAxisScale(self.gui.specPlot.yLeft,
+ min(self.spec_t),
+ max(self.spec_t))
+
# Set the zoomer base to unzoom to the new axis
self.specZoomer.setZoomBase()
--
cgit v1.2.3
From 785cacd477d50aba8a412bae16c57681c52a7fd2 Mon Sep 17 00:00:00 2001
From: Tom Rondeau <trondeau@molly.home>
Date: Tue, 25 Aug 2009 18:22:06 -0400
Subject: Added colorbar to spectrogram for magnitude measurement.
---
gr-utils/src/python/gr_plot_qt.py | 18 ++++++++++++++++--
gr-utils/src/python/pyqt_plot.py | 4 ++--
gr-utils/src/python/pyqt_plot.ui | 8 +-------
3 files changed, 19 insertions(+), 11 deletions(-)
(limited to 'gr-utils/src/python/gr_plot_qt.py')
diff --git a/gr-utils/src/python/gr_plot_qt.py b/gr-utils/src/python/gr_plot_qt.py
index 5579bbce29..a5e3463c8e 100755
--- a/gr-utils/src/python/gr_plot_qt.py
+++ b/gr-utils/src/python/gr_plot_qt.py
@@ -195,10 +195,16 @@ class gr_plot_qt(QtGui.QMainWindow):
self.spec = Qwt.QwtPlotSpectrogram()
self.spec.setColorMap(colorMap)
self.spec.attach(self.gui.specPlot)
- self.spec.setContourLevels([0.5, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5, 8.5, 9.5])
self.spec.setDisplayMode(Qwt.QwtPlotSpectrogram.ImageMode, True)
self.spec.setData(self.specdata)
+ self.rightAxis = self.gui.specPlot.axisWidget(Qwt.QwtPlot.yRight)
+ self.rightAxis.setTitle("Magnitude (dBm)")
+ self.rightAxis.setColorBarEnabled(True)
+ self.rightAxis.setColorMap(self.spec.data().range(),
+ self.spec.colorMap())
+ self.gui.specPlot.enableAxis(Qwt.QwtPlot.yRight)
+
# Set up initial color scheme
self.color_modes["Blue on Black"]()
@@ -488,13 +494,21 @@ class gr_plot_qt(QtGui.QMainWindow):
# since this is taken care of in the SpectrogramData class
self.specdata.set_data(self.spec_f, self.spec_t, self.iq_spec)
+ # Set the color map based on the new data
+ self.rightAxis.setColorMap(self.spec.data().range(),
+ self.spec.colorMap())
+
+ # Set the new axis base; include right axis for the intenisty color bar
self.gui.specPlot.setAxisScale(self.gui.specPlot.xBottom,
min(self.spec_f),
max(self.spec_f))
self.gui.specPlot.setAxisScale(self.gui.specPlot.yLeft,
min(self.spec_t),
max(self.spec_t))
-
+ self.gui.specPlot.setAxisScale(self.gui.specPlot.yRight,
+ self.iq_spec.min(),
+ self.iq_spec.max())
+
# Set the zoomer base to unzoom to the new axis
self.specZoomer.setZoomBase()
diff --git a/gr-utils/src/python/pyqt_plot.py b/gr-utils/src/python/pyqt_plot.py
index 2724d85461..74c43c3ebe 100644
--- a/gr-utils/src/python/pyqt_plot.py
+++ b/gr-utils/src/python/pyqt_plot.py
@@ -2,7 +2,7 @@
# Form implementation generated from reading ui file 'pyqt_plot.ui'
#
-# Created: Tue Aug 25 15:28:55 2009
+# Created: Tue Aug 25 18:18:14 2009
# by: PyQt4 UI code generator 4.4.3
#
# WARNING! All changes made in this file will be lost!
@@ -175,7 +175,7 @@ class Ui_MainWindow(object):
self.menubar.addAction(self.menu_File.menuAction())
self.retranslateUi(MainWindow)
- self.tabGroup.setCurrentIndex(2)
+ self.tabGroup.setCurrentIndex(0)
QtCore.QObject.connect(self.action_exit, QtCore.SIGNAL("activated()"), MainWindow.close)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
diff --git a/gr-utils/src/python/pyqt_plot.ui b/gr-utils/src/python/pyqt_plot.ui
index 62125962b6..19a62adf54 100644
--- a/gr-utils/src/python/pyqt_plot.ui
+++ b/gr-utils/src/python/pyqt_plot.ui
@@ -24,7 +24,7 @@
<item row="1" column="0" >
<widget class="QTabWidget" name="tabGroup" >
<property name="currentIndex" >
- <number>2</number>
+ <number>0</number>
</property>
<widget class="QWidget" name="timeTab" >
<attribute name="title" >
@@ -80,9 +80,6 @@
</widget>
</item>
</layout>
- <zorder>formLayoutWidget</zorder>
- <zorder>psdFFTComboBox</zorder>
- <zorder>psdFFTSizeLabel</zorder>
</widget>
</item>
<item>
@@ -125,9 +122,6 @@
</widget>
</item>
</layout>
- <zorder>specFFTLabel</zorder>
- <zorder>specFFTComboBox</zorder>
- <zorder>specFFTComboBox</zorder>
</widget>
</item>
<item>
--
cgit v1.2.3