print "Please install SciPy to run this script (http://www.scipy.org/)"
raise SystemExit, 1
+try:
+ from matplotlib import mlab
+except ImportError:
+ print "Please install Matplotlib to run this script (http://matplotlib.sourceforge.net)"
+ raise SystemExit, 1
+
+try:
+ from PyQt4 import Qt, QtCore, QtGui
+except ImportError:
+ print "Please install PyQt4 to run this script (http://www.riverbankcomputing.co.uk/software/pyqt/download)"
+ raise SystemExit, 1
+
+try:
+ import PyQt4.Qwt5 as Qwt
+except ImportError:
+ print "Please install PyQwt5 to run this script (http://pyqwt.sourceforge.net/)"
+ raise SystemExit, 1
+
+try:
+ # FIXME: reenable this before committing
+ #from gnuradio.pyqt_plot import Ui_MainWindow
+ from pyqt_plot import Ui_MainWindow
+except ImportError:
+ print "Could not import from pyqt_plot. Please build with \"pyuic4 pyqt_plot.ui -o pyqt_plot.py\""
+ 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 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(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):
+ 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):
+ f = int(self.freq.searchsorted(x))
+ t = int(self.time.searchsorted(y))
+ return self.sp[f][t-1]
+
class gr_plot_qt(QtGui.QMainWindow):
def __init__(self, qapp, filename, options, parent=None):
self.gui = Ui_MainWindow()
self.gui.setupUi(self)
+ self.filename = None
self.block_length = options.block_length
self.start = options.start
self.sample_rate = options.sample_rate
self.winfunc = scipy.blackman
self.sizeof_data = 8
self.datatype = scipy.complex64
+ self.pen_width = 1
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)")
+ 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.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.fftComboBoxEdit)
+ self.psdFFTComboBoxEdit)
+ self.connect(self.gui.specFFTComboBox,
+ Qt.SIGNAL("activated (const QString&)"),
+ self.specFFTComboBoxEdit)
# Set up color scheme box
self.color_modes = {"Black on White" : self.color_black_on_white,
self.colorComboBoxEdit)
+ # Set up line style combo box
+ self.line_styles = {"None" : Qwt.QwtSymbol.NoSymbol,
+ "Circle" : Qwt.QwtSymbol.Ellipse,
+ "Diamond" : Qwt.QwtSymbol.Rect,
+ "Triangle" : Qwt.QwtSymbol.Triangle}
+ self.gui.lineStyleComboBox.addItems(self.line_styles.keys())
+ pos = self.gui.lineStyleComboBox.findText("None")
+ self.gui.lineStyleComboBox.setCurrentIndex(pos)
+ self.connect(self.gui.lineStyleComboBox,
+ Qt.SIGNAL("activated (const QString&)"),
+ self.lineStyleComboBoxEdit)
+
# Create zoom functionality for the plots
self.timeZoomer = Qwt.QwtPlotZoomer(self.gui.timePlot.xBottom,
self.gui.timePlot.yLeft,
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)
+ self.specZoomer = Qwt.QwtPlotZoomer(self.gui.specPlot.xBottom,
+ self.gui.specPlot.yLeft,
+ Qwt.QwtPicker.PointSelection,
+ Qwt.QwtPicker.AlwaysOn,
+ self.gui.specPlot.canvas())
# Set up action when tab is changed
self.connect(self.gui.tabGroup,
self.connect(self.gui.action_open,
Qt.SIGNAL("activated()"),
self.open_file)
-
+
+ # Connect Reload action to reload the file
+ self.connect(self.gui.action_reload,
+ Qt.SIGNAL("activated()"),
+ self.reload_file)
+ self.gui.action_reload.setShortcut(QtGui.QApplication.translate("MainWindow", "Ctrl+R",
+ None, QtGui.QApplication.UnicodeUTF8))
+
# Set up file position boxes to update current figure
self.connect(self.gui.filePosStartLineEdit,
Qt.SIGNAL("editingFinished()"),
Qt.SIGNAL("editingFinished()"),
self.file_time_length_changed)
+ stylestr = str(self.gui.lineStyleComboBox.currentText().toAscii())
+ style = self.line_styles[stylestr]
+
self.rcurve = Qwt.QwtPlotCurve("Real")
self.icurve = Qwt.QwtPlotCurve("Imaginary")
+ self.rsym = Qwt.QwtSymbol()
+ self.rsym.setStyle(style)
+ self.rsym.setSize(10)
+ self.isym = Qwt.QwtSymbol()
+ self.isym.setStyle(style)
+ self.isym.setSize(10)
+ self.rcurve.setSymbol(self.rsym)
+ self.icurve.setSymbol(self.isym)
+
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 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.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"]()
+ # When line width spin box changes, update the pen size
+ self.connect(self.gui.lineWidthSpinBox,
+ Qt.SIGNAL("valueChanged(int)"),
+ self.change_pen_width)
+ self.gui.lineWidthSpinBox.setRange(1, 10)
+
+
+ # Connect a signal for when the sample rate changes
self.set_sample_rate(self.sample_rate)
self.connect(self.gui.sampleRateLineEdit,
Qt.SIGNAL("editingFinished()"),
def open_file(self):
filename = Qt.QFileDialog.getOpenFileName(self, "Open", ".")
if(filename != ""):
- print filename
+ #print filename
self.initialize(filename)
+ def reload_file(self):
+ if(self.filename):
+ self.initialize(self.filename)
+
def initialize(self, filename):
+ self.filename = filename
self.hfile = open(filename, "r")
self.setWindowTitle(("GNU Radio File Plot Utility: %s" % filename))
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)
self.signal_size = self.hfile.tell()/self.sizeof_data
- print "Sizeof File: ", self.signal_size
+ #print "Sizeof File: ", self.signal_size
self.hfile.seek(0, os.SEEK_SET)
def get_data(self, start, end):
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())
color_func = self.color_modes[colorstr]
color_func()
+ def lineStyleComboBoxEdit(self, styleSelection):
+ stylestr = str(styleSelection.toAscii())
+ self.rsym.setStyle(self.line_styles[stylestr])
+ self.isym.setStyle(self.line_styles[stylestr])
+ self.rcurve.setSymbol(self.rsym)
+ self.icurve.setSymbol(self.isym)
+ self.gui.timePlot.replot()
+
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.get_specgram()
self.update_time_curves()
self.update_psd_curves()
+ self.update_specgram_curves()
def set_sample_rate(self, sr):
self.sample_rate = sr
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()
+ self.update_specgram_curves()
def set_file_pos_box(self, start, end):
tstart = start / self.sample_rate
self.update_time_curves()
self.update_psd_curves()
+ self.update_specgram_curves()
# If there's a non-digit character, reset box
else:
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)
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)
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)
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 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()
+
+ self.gui.specPlot.replot()
+
def tabChanged(self, index):
self.gui.timePlot.replot()
self.gui.freqPlot.replot()
+ self.gui.specPlot.replot()
+
+ def change_pen_width(self, width):
+ self.pen_width = width
+ colormode = str(self.gui.colorComboBox.currentText().toAscii())
+ color_func = self.color_modes[colormode]()
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))
+ blackPen = Qt.QPen(Qt.QBrush(Qt.QColor("black")), self.pen_width)
+ bluePen = Qt.QPen(Qt.QBrush(Qt.QColor(blue)), self.pen_width)
+ redPen = Qt.QPen(Qt.QBrush(Qt.QColor(red)), self.pen_width)
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.timeZoomer.setTrackerPen(blackPen)
+ self.timeZoomer.setRubberBandPen(blackPen)
+ self.freqZoomer.setTrackerPen(blackPen)
+ self.freqZoomer.setRubberBandPen(blackPen)
+ self.psdcurve.setPen(bluePen)
+ self.rcurve.setPen(bluePen)
+ self.icurve.setPen(redPen)
+
+ self.rsym.setPen(bluePen)
+ self.isym.setPen(redPen)
self.gui.timePlot.replot()
self.gui.freqPlot.replot()
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.timeZoomer.setTrackerPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.timeZoomer.setRubberBandPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.freqZoomer.setTrackerPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.freqZoomer.setRubberBandPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.psdcurve.setPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.rcurve.setPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.icurve.setPen(Qt.QPen(redBrush, self.pen_width))
self.gui.timePlot.replot()
self.gui.freqPlot.replot()
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.timeZoomer.setTrackerPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.timeZoomer.setRubberBandPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.freqZoomer.setTrackerPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.freqZoomer.setRubberBandPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.psdcurve.setPen(Qt.QPen(greenBrush, self.pen_width))
+ self.rcurve.setPen(Qt.QPen(greenBrush, self.pen_width))
+ self.icurve.setPen(Qt.QPen(redBrush, self.pen_width))
self.gui.timePlot.replot()
self.gui.freqPlot.replot()
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.timeZoomer.setTrackerPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.timeZoomer.setRubberBandPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.freqZoomer.setTrackerPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.freqZoomer.setRubberBandPen(Qt.QPen(whiteBrush, self.pen_width))
+ self.psdcurve.setPen(Qt.QPen(blueBrush, self.pen_width))
+ self.rcurve.setPen(Qt.QPen(blueBrush, self.pen_width))
+ self.icurve.setPen(Qt.QPen(redBrush, self.pen_width))
self.gui.timePlot.replot()
self.gui.freqPlot.replot()
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
projects / debian / gnuradio / blobdiff