diff options
Diffstat (limited to 'gr-qtgui/src/python/usrp2_display.py')
| -rwxr-xr-x | gr-qtgui/src/python/usrp2_display.py | 231 |
1 files changed, 231 insertions, 0 deletions
diff --git a/gr-qtgui/src/python/usrp2_display.py b/gr-qtgui/src/python/usrp2_display.py new file mode 100755 index 0000000000..e5f01ef0e9 --- /dev/null +++ b/gr-qtgui/src/python/usrp2_display.py @@ -0,0 +1,231 @@ +#!/usr/bin/env python +# +# Copyright 2009 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. +# + +from gnuradio import gr, gru +from gnuradio import usrp2 +from gnuradio import eng_notation +from gnuradio.eng_option import eng_option +from gnuradio.qtgui import qtgui +from optparse import OptionParser +from PyQt4 import QtGui, QtCore +import sys, sip + +class dialog_box(QtGui.QWidget): + def __init__(self, display, control): + QtGui.QWidget.__init__(self, None) + self.setWindowTitle('USRP2 Display') + + self.boxlayout = QtGui.QBoxLayout(QtGui.QBoxLayout.LeftToRight, self) + self.boxlayout.addWidget(display, 1) + self.boxlayout.addWidget(control) + + self.resize(800, 500) + +class control_panel(QtGui.QWidget): + def __init__(self, usrp, qtsink, parent=None): + QtGui.QWidget.__init__(self, parent) + self.setWindowTitle('USRP2 Control Panel') + + self.usrp = usrp + self.qtsink = qtsink + self.adc_rate = self.usrp.adc_rate() + + self.freq = 0 + self.decim = 0 + self.bw = 0 + self.gain = 0 + + self.setToolTip('Set the values of the USRP2') + QtGui.QToolTip.setFont(QtGui.QFont('OldEnglish', 10)) + + self.layout = QtGui.QFormLayout(self) + + # Received frequency + self.freqEdit = QtGui.QLineEdit(self) + self.layout.addRow("Frequency:", self.freqEdit) + self.connect(self.freqEdit, QtCore.SIGNAL("editingFinished()"), + self.freqEditText) + + # Receiver gain + self.gainEdit = QtGui.QLineEdit(self) + self.layout.addRow("Gain:", self.gainEdit) + self.connect(self.gainEdit, QtCore.SIGNAL("editingFinished()"), + self.gainEditText) + + + # Decim / Bandwidth + self.decimEdit = QtGui.QLineEdit(self) + self.layout.addRow("Decim Rate:", self.decimEdit) + self.connect(self.decimEdit, QtCore.SIGNAL("editingFinished()"), + self.decimEditText) + + self.quit = QtGui.QPushButton('Close', self) + self.layout.addRow(self.quit) + + self.connect(self.quit, QtCore.SIGNAL('clicked()'), + QtGui.qApp, QtCore.SLOT('quit()')) + + def set_frequency(self, freq): + self.freq = freq + sfreq = eng_notation.num_to_str(self.freq) + self.freqEdit.setText(QtCore.QString("%1").arg(sfreq)) + + def set_gain(self, gain): + self.gain = gain + self.gainEdit.setText(QtCore.QString("%1").arg(self.gain)) + + def set_decim(self, decim): + self.decim = decim + self.bw = self.adc_rate / float(self.decim) / 1000.0 + self.decimEdit.setText(QtCore.QString("%1").arg(self.decim)) + + def freqEditText(self): + try: + freq = eng_notation.str_to_num(self.freqEdit.text().toAscii()) + r = self.usrp.set_center_freq(freq) + self.freq = freq + self.qtsink.set_frequency_range(self.freq, self.freq-self.bw/2.0, self.freq+self.bw/2.0) + except RuntimeError: + pass + + #self.set_frequency(self.freq) + + def gainEditText(self): + try: + gain = float(self.gainEdit.text()) + self.usrp.set_gain(gain) + self.gain = gain + except ValueError: + pass + + #self.set_gain(gain) + + def decimEditText(self): + try: + decim = int(self.decimEdit.text()) + self.usrp.set_decim(decim) + + self.decim = decim + self.bw = self.adc_rate / self.decim + self.qtsink.set_frequency_range(-self.bw/2.0, self.bw/2.0, self.freq) + + except ValueError: + pass + + #self.set_decim(decim) + +class app_top_block(gr.top_block): + def __init__(self): + gr.top_block.__init__(self) + + parser = OptionParser(option_class=eng_option) + parser.add_option("-e", "--interface", type="string", default="eth0", + help="select Ethernet interface, default is eth0") + parser.add_option("-m", "--mac-addr", type="string", default="", + help="select USRP by MAC address, default is auto-select") + parser.add_option("-d", "--decim", type="int", default=16, + help="set fgpa decimation rate to DECIM [default=%default]") + parser.add_option("-f", "--freq", type="eng_float", default=None, + help="set frequency to FREQ", metavar="FREQ") + parser.add_option("-g", "--gain", type="eng_float", default=None, + help="set gain in dB (default is midpoint)") + parser.add_option("--fft-size", type="int", default=1024, + help="Set number of FFT bins [default=%default]") + (options, args) = parser.parse_args() + + if len(args) != 0: + parser.print_help() + sys.exit(1) + self.options = options + self.show_debug_info = True + + self.qapp = QtGui.QApplication(sys.argv) + + self.u = usrp2.source_32fc(options.interface, options.mac_addr) + self.u.set_decim(options.decim) + + input_rate = self.u.adc_rate() / self.u.decim() + + self.snk = qtgui.sink_c(options.fft_size, gr.firdes.WIN_BLACKMAN_hARRIS, + -input_rate/2, input_rate/2, + "USRP2 Display", + True, True, False, True, False) + + self.connect(self.u, self.snk) + + if options.gain is None: + # if no gain was specified, use the mid-point in dB + g = self.u.gain_range() + options.gain = float(g[0]+g[1])/2 + + if options.freq is None: + # if no freq was specified, use the mid-point + r = self.u.freq_range() + options.freq = float(r[0]+r[1])/2 + + self.set_gain(options.gain) + r = self.u.set_center_freq(options.freq) + + if self.show_debug_info: + print "Decimation rate: ", self.u.decim() + print "Bandwidth: ", input_rate + print "D'board: ", self.u.daughterboard_id() + + + self.ctrl_win = control_panel(self.u, self.snk) + + self.ctrl_win.set_frequency(options.freq) + self.ctrl_win.set_gain(options.gain) + self.ctrl_win.set_decim(options.decim) + + # Get the reference pointer to the SpectrumDisplayForm QWidget + pyQt = self.snk.pyqwidget() + + # Wrap the pointer as a PyQt SIP object + # This can now be manipulated as a PyQt4.QtGui.QWidget + pyWin = sip.wrapinstance(pyQt, QtGui.QWidget) + + self.main_box = dialog_box(pyWin, self.ctrl_win) + + self.main_box.show() + + def set_gain(self, gain): + self.u.set_gain(gain) + + def set_decim(self, decim): + ok = self.u.set_decim(decim) + if not ok: + print "set_decim failed" + input_rate = self.u.adc_rate() / self.u.decim() + return ok + +def main (): + tb = app_top_block() + tb.start() + tb.snk.exec_(); + +if __name__ == '__main__': + try: + main () + except KeyboardInterrupt: + pass + |