git.gag.com Git - debian/gnuradio/blob - gr-qtgui/src/python/usrp_display.py

   1 #!/usr/bin/env python
   2
   3 from gnuradio import gr
   4 from gnuradio import usrp
   5 from gnuradio import eng_notation
   6 from gnuradio.eng_option import eng_option
   7 from gnuradio.qtgui import qtgui
   8 from optparse import OptionParser
   9 from PyQt4 import QtGui, QtCore
  10 import sys, sip
  11
  12 class dialog_box(QtGui.QWidget):
  13     def __init__(self, display, control):
  14         QtGui.QWidget.__init__(self, None)
  15         self.setWindowTitle('USRP FFT')
  16
  17         self.boxlayout = QtGui.QBoxLayout(QtGui.QBoxLayout.LeftToRight, self)
  18         self.boxlayout.addWidget(display, 1)
  19         self.boxlayout.addWidget(control)
  20
  21         self.resize(800, 500)
  22
  23 class control_panel(QtGui.QWidget):
  24     def __init__(self, usrp, subdev, qtsink, parent=None):
  25         QtGui.QWidget.__init__(self, parent)
  26         self.setWindowTitle('USRP Control Panel')
  27
  28         self.usrp = usrp
  29         self.subdev = subdev
  30         self.qtsink = qtsink
  31         self.adc_rate = self.usrp.converter_rate()
  32
  33         self.freq = 0
  34         self.decim = 0
  35         self.bw = 0
  36         self.gain = 0
  37
  38         self.setToolTip('Set the values of the USRP')
  39         QtGui.QToolTip.setFont(QtGui.QFont('OldEnglish', 10))
  40
  41         self.layout = QtGui.QFormLayout(self)
  42
  43         # Received frequency
  44         self.freqEdit = QtGui.QLineEdit(self)
  45         self.layout.addRow("Frequency:", self.freqEdit)
  46         self.connect(self.freqEdit, QtCore.SIGNAL("editingFinished()"),
  47                      self.freqEditText)
  48
  49         # Receiver gain
  50         self.gainEdit = QtGui.QLineEdit(self)
  51         self.layout.addRow("Gain:", self.gainEdit)
  52         self.connect(self.gainEdit, QtCore.SIGNAL("editingFinished()"),
  53                      self.gainEditText)
  54
  55
  56         # Decim / Bandwidth
  57         self.decimEdit = QtGui.QLineEdit(self)
  58         self.layout.addRow("Decim Rate:", self.decimEdit)
  59         self.connect(self.decimEdit, QtCore.SIGNAL("editingFinished()"),
  60                      self.decimEditText)
  61
  62         self.quit = QtGui.QPushButton('Close', self)
  63         self.layout.addRow(self.quit)
  64
  65         self.connect(self.quit, QtCore.SIGNAL('clicked()'),
  66                      QtGui.qApp, QtCore.SLOT('quit()'))
  67
  68     def set_frequency(self, freq):
  69         self.freq = freq
  70         sfreq = eng_notation.num_to_str(self.freq)
  71         self.freqEdit.setText(QtCore.QString("%1").arg(sfreq))
  72
  73     def set_gain(self, gain):
  74         self.gain = gain
  75         self.gainEdit.setText(QtCore.QString("%1").arg(self.gain))
  76
  77     def set_decim(self, decim):
  78         self.decim = decim
  79         self.bw = self.adc_rate / float(self.decim) / 1000.0
  80         self.decimEdit.setText(QtCore.QString("%1").arg(self.decim))
  81
  82     def freqEditText(self):
  83         try:
  84             freq = eng_notation.str_to_num(self.freqEdit.text().toAscii())
  85             self.usrp.tune(0, self.subdev, freq)
  86             self.freq = freq
  87             self.qtsink.set_frequency_range(self.freq, self.freq-self.bw/2.0, self.freq+self.bw/2.0)
  88         except RuntimeError:
  89             pass
  90
  91         #self.set_frequency(self.freq)
  92
  93     def gainEditText(self):
  94         try:
  95             gain = float(self.gainEdit.text())
  96             self.subdev.set_gain(gain)
  97             self.gain = gain
  98         except ValueError:
  99             pass
 100
 101         #self.set_gain(gain)
 102
 103     def decimEditText(self):
 104         try:
 105             decim = int(self.decimEdit.text())
 106             self.usrp.set_decim_rate(decim)
 107
 108             self.decim = decim
 109             self.bw = self.adc_rate / self.decim
 110             self.qtsink.set_frequency_range(-self.bw/2.0, self.bw/2.0, self.freq)
 111
 112         except ValueError:
 113             pass
 114
 115         #self.set_decim(decim)
 116
 117
 118 class my_top_block(gr.top_block):
 119     def __init__(self):
 120         gr.top_block.__init__(self)
 121
 122         parser = OptionParser(option_class=eng_option)
 123         parser.add_option("-w", "--which", type="int", default=0,
 124                           help="select which USRP (0, 1, ...) default is %default",
 125                           metavar="NUM")
 126         parser.add_option("-R", "--rx-subdev-spec", type="subdev", default=None,
 127                           help="select USRP Rx side A or B (default=first one with a daughterboard)")
 128         parser.add_option("-A", "--antenna", default=None,
 129                           help="select Rx Antenna (only on RFX-series boards)")
 130         parser.add_option("-d", "--decim", type="int", default=16,
 131                           help="set fgpa decimation rate to DECIM [default=%default]")
 132         parser.add_option("-f", "--freq", type="eng_float", default=None,
 133                           help="set frequency to FREQ", metavar="FREQ")
 134         parser.add_option("-g", "--gain", type="eng_float", default=None,
 135                           help="set gain in dB [default is midpoint]")
 136         parser.add_option("-W", "--waterfall", action="store_true", default=False,
 137                           help="Enable waterfall display")
 138         parser.add_option("-8", "--width-8", action="store_true", default=False,
 139                           help="Enable 8-bit samples across USB")
 140         parser.add_option( "--no-hb", action="store_true", default=False,
 141                           help="don't use halfband filter in usrp")
 142         parser.add_option("-S", "--oscilloscope", action="store_true", default=False,
 143                           help="Enable oscilloscope display")
 144         parser.add_option("", "--avg-alpha", type="eng_float", default=1e-1,
 145                           help="Set fftsink averaging factor, [default=%default]")
 146         parser.add_option("", "--ref-scale", type="eng_float", default=13490.0,
 147                           help="Set dBFS=0dB input value, [default=%default]")
 148         parser.add_option("", "--fft-size", type="int", default=1024,
 149                           help="Set FFT frame size, [default=%default]");
 150
 151         (options, args) = parser.parse_args()
 152         if len(args) != 0:
 153             parser.print_help()
 154             sys.exit(1)
 155         self.options = options
 156         self.show_debug_info = True
 157
 158         # Call this before creating the Qt sink
 159         self.qapp = QtGui.QApplication(sys.argv)
 160
 161         self.u = usrp.source_c(which=options.which, decim_rate=options.decim)
 162         rx_subdev_spec = (0,0)
 163         self.u.set_mux(usrp.determine_rx_mux_value(self.u, rx_subdev_spec))
 164         self.subdev = usrp.selected_subdev(self.u, rx_subdev_spec)
 165
 166         if options.gain is None:
 167             # if no gain was specified, use the mid-point in dB
 168             g = self.subdev.gain_range()
 169             options.gain = float(g[0]+g[1])/2
 170         self.subdev.set_gain(options.gain)
 171
 172         if options.freq is None:
 173             # if no frequency was specified, use the mid-point of the subdev
 174             f = self.subdev.freq_range()
 175             options.freq = float(f[0]+f[1])/2
 176         self.u.tune(0, self.subdev, options.freq)
 177
 178         fftsize = 2048
 179         input_rate = self.u.converter_rate() / self.u.decim_rate()
 180         self.snk = qtgui.sink_c(fftsize, gr.firdes.WIN_BLACKMAN_hARRIS,
 181                                 -input_rate/2, input_rate/2,
 182                                 "USRP Display",
 183                                 True, True, False, True, False)
 184
 185         amp = gr.multiply_const_cc(0.001)
 186         self.connect(self.u, amp, self.snk)
 187
 188         self.ctrl_win = control_panel(self.u, self.subdev, self.snk)
 189
 190         self.ctrl_win.set_frequency(options.freq)
 191         self.ctrl_win.set_gain(options.gain)
 192         self.ctrl_win.set_decim(options.decim)
 193
 194         # Get the reference pointer to the SpectrumDisplayForm QWidget
 195         pyQt  = self.snk.pyqwidget()
 196
 197         # Wrap the pointer as a PyQt SIP object
 198         # This can now be manipulated as a PyQt4.QtGui.QWidget
 199         pyWin = sip.wrapinstance(pyQt, QtGui.QWidget)
 200
 201         self.main_box = dialog_box(pyWin, self.ctrl_win)
 202
 203         self.main_box.show()
 204
 205 if __name__ == "__main__":
 206     tb = my_top_block();
 207     tb.start()
 208     tb.qapp.exec_()