From 471ebf621da0426fefda870dfe29d75cd9ef2da6 Mon Sep 17 00:00:00 2001
From: trondeau <trondeau@221aa14e-8319-0410-a670-987f0aec2ac5>
Date: Mon, 6 Jul 2009 04:39:24 +0000
Subject: Merging trondeau/qt branch r11231:11360. This merge improves the
usability and examples of the QT-based interface.
git-svn-id: http://gnuradio.org/svn/gnuradio/trunk@11361 221aa14e-8319-0410-a670-987f0aec2ac5
---
gnuradio-examples/python/usrp2/usrp2_wfm_qt.py | 352 +++++++++++++++++++++++++
1 file changed, 352 insertions(+)
create mode 100755 gnuradio-examples/python/usrp2/usrp2_wfm_qt.py
(limited to 'gnuradio-examples/python/usrp2/usrp2_wfm_qt.py')
diff --git a/gnuradio-examples/python/usrp2/usrp2_wfm_qt.py b/gnuradio-examples/python/usrp2/usrp2_wfm_qt.py
new file mode 100755
index 0000000000..0be21ceb98
--- /dev/null
+++ b/gnuradio-examples/python/usrp2/usrp2_wfm_qt.py
@@ -0,0 +1,352 @@
+#!/usr/bin/env python
+#
+# Copyright 2005,2006,2007,2008,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, eng_notation, optfir
+from gnuradio import audio
+from gnuradio import usrp2
+from gnuradio import blks2
+from gnuradio.eng_option import eng_option
+from optparse import OptionParser
+import sys
+import math
+
+
+try:
+ from gnuradio.qtgui import qtgui
+ from PyQt4 import QtGui, QtCore
+ import sip
+except ImportError:
+ print "Please install gr-qtgui."
+ sys.exit(1)
+
+try:
+ from qt_wfm_interface import Ui_InterfaceWindow
+except ImportError:
+ print "Error: could not find qt_wfm_interface.py:"
+ print "\tPlease run: \"pyuic4 qt_wfm_interface.ui -o qt_wfm_interface.py\""
+ sys.exit(1)
+
+print "This program is not in a proper working state. Comment this out if you want to play."
+sys.exit(1)
+
+
+# ////////////////////////////////////////////////////////////////////
+# Define the QT Interface and Control Dialog
+# ////////////////////////////////////////////////////////////////////
+
+
+class dialog_box(QtGui.QMainWindow):
+ def __init__(self, snk_usrp, snk_vol, fg, parent=None):
+
+ QtGui.QWidget.__init__(self, parent)
+ self.gui = Ui_InterfaceWindow()
+ self.gui.setupUi(self)
+
+ self.fg = fg
+
+ # Set USRP parameters
+ self.set_bw(self.fg.usrp_bw())
+ self.set_freq(self.fg.freq())
+ self.set_gain(self.fg.gain())
+ self.set_volume(self.fg.volume())
+
+ # Add the qtsnk widgets to the hlayout box
+ self.gui.sinkLayout.addWidget(snk_usrp)
+ self.gui.sinkLayout.addWidget(snk_vol)
+
+
+ # Connect up some signals
+ self.connect(self.gui.pauseButton, QtCore.SIGNAL("clicked()"),
+ self.pauseFg)
+
+ self.connect(self.gui.bandwidthEdit, QtCore.SIGNAL("editingFinished()"),
+ self.bwEditText)
+ self.connect(self.gui.freqEdit, QtCore.SIGNAL("editingFinished()"),
+ self.freqEditText)
+ self.connect(self.gui.gainEdit, QtCore.SIGNAL("editingFinished()"),
+ self.gainEditText)
+
+ self.connect(self.gui.volumeEdit, QtCore.SIGNAL("editingFinished()"),
+ self.volumeEditText)
+
+
+ def pauseFg(self):
+ if(self.gui.pauseButton.text() == "Pause"):
+ self.fg.stop()
+ self.fg.wait()
+ self.gui.pauseButton.setText("Unpause")
+ else:
+ self.fg.start()
+ self.gui.pauseButton.setText("Pause")
+
+
+ # Accessor functions for Gui to manipulate USRP
+ def set_bw(self, bw):
+ self.gui.bandwidthEdit.setText(QtCore.QString("%1").arg(bw))
+
+ def set_freq(self, freq):
+ self.gui.freqEdit.setText(QtCore.QString("%1").arg(freq))
+
+ def set_gain(self, gain):
+ self.gui.gainEdit.setText(QtCore.QString("%1").arg(gain))
+
+ def set_volume(self, vol):
+ self.gui.volumeEdit.setText(QtCore.QString("%1").arg(vol))
+
+ def bwEditText(self):
+ try:
+ bw = self.gui.bandwidthEdit.text().toDouble()[0]
+ self.fg.set_usrp_bw(bw)
+ except RuntimeError:
+ pass
+
+ def freqEditText(self):
+ try:
+ freq = self.gui.freqEdit.text().toDouble()[0]
+ self.fg.set_freq(freq)
+ except RuntimeError:
+ pass
+
+ def gainEditText(self):
+ try:
+ gain = self.gui.gainEdit.text().toDouble()[0]
+ self.fg.set_gain(gain)
+ except RuntimeError:
+ pass
+
+ def volumeEditText(self):
+ try:
+ vol = self.gui.volumeEdit.text().toDouble()[0]
+ self.fg.set_volume(vol)
+ except RuntimeError:
+ pass
+
+
+
+
+# ////////////////////////////////////////////////////////////////////
+# Define the GNU Radio Top Block
+# ////////////////////////////////////////////////////////////////////
+
+
+class wfm_rx_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("-A", "--antenna", default=None,
+ # help="select Rx Antenna (only on RFX-series boards)")
+ parser.add_option("-f", "--freq", type="eng_float", default=100.1,
+ 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("-V", "--volume", type="eng_float", default=None,
+ help="set volume (default is midpoint)")
+ parser.add_option("-O", "--audio-output", type="string", default="",
+ help="pcm device name. E.g., hw:0,0 or surround51 or /dev/dsp")
+
+ (options, args) = parser.parse_args()
+ if len(args) != 0:
+ parser.print_help()
+ sys.exit(1)
+
+ self._volume = options.volume
+ self._usrp_freq = options.freq
+ self._usrp_gain = options.gain
+ self._audio_rate = int(32e3)
+
+ # build graph
+
+ self.u = usrp2.source_32fc(options.interface, options.mac_addr)
+
+ # calculate decimation values to get USRP BW at 320 kHz
+ self.calculate_usrp_bw(320e3)
+
+ self.set_decim(self._usrp_decim)
+
+ #FIXME: need named constants and text descriptions available to (gr-)usrp2 even
+ #when usrp(1) module is not built. A usrp_common module, perhaps?
+ dbid = self.u.daughterboard_id()
+ print "Using RX d'board 0x%04X" % (dbid,)
+ #if not (dbid == 0x0001 or #usrp_dbid.BASIC_RX
+ # dbid == 0x0003 or #usrp_dbid.TV_RX
+ # dbid == 0x000c or #usrp_dbid.TV_RX_REV_2
+ # dbid == 0x0040): #usrp_dbid.TV_RX_REV_3
+ # print "This daughterboard does not cover the required frequency range"
+ # print "for this application. Please use a BasicRX or TVRX daughterboard."
+ # raw_input("Press ENTER to continue anyway, or Ctrl-C to exit.")
+
+ chan_filt_coeffs = optfir.low_pass (1, # gain
+ self._usrp_rate, # sampling rate
+ 80e3, # passband cutoff
+ 115e3, # stopband cutoff
+ 0.1, # passband ripple
+ 60) # stopband attenuation
+ #print len(chan_filt_coeffs)
+ chan_filt = gr.fir_filter_ccf (self._chanfilt_decim, chan_filt_coeffs)
+
+ self.guts = blks2.wfm_rcv (self._demod_rate, self._audio_decim)
+
+ self.volume_control = gr.multiply_const_ff(1)
+
+ # sound card as final sink
+ #audio_sink = audio.sink (int (audio_rate),
+ # options.audio_output,
+ # False) # ok_to_block
+ audio_sink = audio.sink (self._audio_rate,
+ options.audio_output)
+
+
+ if self._usrp_gain is None:
+ # if no gain was specified, use the mid-point in dB
+ g = self.u.gain_range()
+ print "Gain range: ", g
+ self._usrp_gain = float(g[0]+g[1])/2
+
+ if self._volume is None:
+ g = self.volume_range()
+ self._volume = float(g[0]+g[1])/2
+
+ if abs(self._usrp_freq) < 1e6:
+ self._usrp_freq *= 1e6
+
+ # set initial values
+ self.set_gain(self._usrp_gain)
+ self.set_volume(self._volume)
+ if not(self.set_freq(self._usrp_freq)):
+ print ("Failed to set initial frequency")
+
+
+ # Define a GUI sink to display the received signal
+ self.qapp = QtGui.QApplication(sys.argv)
+ fftsize = 2048
+
+ self.usrp_rx = qtgui.sink_c(fftsize, gr.firdes.WIN_BLACKMAN_hARRIS,
+ -self._usrp_rate/2.0, self._usrp_rate/2.0,
+ "Received Signal", True, True, False, True, False,
+ use_openGL=False)
+ self.usrp_rx2 = qtgui.sink_f(fftsize, gr.firdes.WIN_BLACKMAN_hARRIS,
+ -self._usrp_rate/2.0, self._usrp_rate/2.0,
+ "Received Signal", True, True, False, True, False)
+
+ # now wire it all together
+ self.connect (self.u, chan_filt, self.guts, self.volume_control, audio_sink)
+ self.connect (self.u, self.usrp_rx)
+ self.connect (self.volume_control, self.usrp_rx2)
+
+ usrp_rx_widget = sip.wrapinstance(self.usrp_rx.pyqwidget(), QtGui.QWidget)
+ usrp_rx2_widget = sip.wrapinstance(self.usrp_rx2.pyqwidget(), QtGui.QWidget)
+
+ self.main_box = dialog_box(usrp_rx_widget, usrp_rx2_widget, self)
+ self.main_box.show()
+
+
+ def calculate_usrp_bw(self, bw):
+ """
+ Calculate the different decimation rates that make the USRP BW equal to the
+ input bandwidth parameter 'bw' and the audio bandwidth equal to the system-
+ wide bandwidth 'self._audio_rate'
+ """
+
+ adc_rate = self.u.adc_rate()
+ d_usrp = int(adc_rate/bw)
+ bw_real = adc_rate / float(d_usrp)
+
+ d_chan = 1
+ demod_rate = bw_real / d_chan
+
+ d_audio = int(bw_real / self._audio_rate)
+ audio_rate = demod_rate / d_audio
+
+ self._usrp_decim = d_usrp
+ self._chanfilt_decim = d_chan
+ self._audio_decim = d_audio
+ self._demod_rate = demod_rate
+ self._usrp_rate = bw_real
+
+ print "USRP Decimation: ", self._usrp_decim
+ print "USRP Bandwidth: ", bw_real
+ print "Audio Decimation: ", self._audio_decim
+ print "Audio Bandwidth: ", audio_rate
+
+ def set_volume (self, vol):
+ g = self.volume_range()
+ self._volume = max(g[0], min(g[1], vol))
+ self.volume_control.set_k(10**(self._volume/10))
+
+ def set_freq(self, target_freq):
+ """
+ Set the center frequency we're interested in.
+
+ @param target_freq: frequency in Hz
+ @rypte: bool
+
+ Tuning is a two step process. First we ask the front-end to
+ tune as close to the desired frequency as it can. Then we use
+ the result of that operation and our target_frequency to
+ determine the value for the digital down converter.
+ """
+ r = self.u.set_center_freq(target_freq)
+ if r:
+ self._usrp_freq = target_freq
+ return True
+ return False
+
+ def set_usrp_bw(self, bw):
+ self.calculate_usrp_bw(bw)
+
+ def set_gain(self, gain):
+ self._usrp_gain = gain
+ self.u.set_gain(gain)
+
+ def set_decim(self, decim):
+ self._usrp_decim = int(decim)
+ self.u.set_decim(self._usrp_decim)
+
+ def volume(self):
+ return self._volume
+
+ def freq(self):
+ return self._usrp_freq
+
+ def usrp_bw(self):
+ return self._usrp_rate
+
+ def gain(self):
+ return self._usrp_gain
+
+ def decim(self):
+ return self._usrp_decim
+
+ def volume_range(self):
+ return (-20.0, 0.0, 0.5)
+
+
+if __name__ == '__main__':
+ tb = wfm_rx_block()
+ tb.start()
+ tb.qapp.exec_()
+
--
cgit v1.2.3