#!/usr/bin/env python ################################################## # Gnuradio Python Flow Graph # Title: UHD FFT # Author: Example # Description: UHD FFT Waveform Plotter # Generated: Sun May 25 01:21:34 2014 ################################################## from gnuradio import analog from gnuradio import blocks from gnuradio import eng_notation from gnuradio import gr from gnuradio import wxgui from gnuradio.eng_option import eng_option from gnuradio.fft import window from gnuradio.filter import firdes from gnuradio.wxgui import fftsink2 from gnuradio.wxgui import forms from gnuradio.wxgui import scopesink2 from gnuradio.wxgui import waterfallsink2 from grc_gnuradio import wxgui as grc_wxgui from optparse import OptionParser import numpy import wx class uhd_fft(grc_wxgui.top_block_gui): def __init__(self, param_freq=2.45e9, param_gain=0, address="addr=192.168.11.2", param_samp_rate=1e6): grc_wxgui.top_block_gui.__init__(self, title="UHD FFT") _icon_path = "/usr/share/icons/hicolor/32x32/apps/gnuradio-grc.png" self.SetIcon(wx.Icon(_icon_path, wx.BITMAP_TYPE_ANY)) ################################################## # Parameters ################################################## self.param_freq = param_freq self.param_gain = param_gain self.address = address self.param_samp_rate = param_samp_rate ################################################## # Variables ################################################## self.samp_rate = samp_rate = param_samp_rate self.gain = gain = param_gain self.freq = freq = param_freq self.ant = ant = "RX2" ################################################## # Blocks ################################################## self._samp_rate_text_box = forms.text_box( parent=self.GetWin(), value=self.samp_rate, callback=self.set_samp_rate, label="Sample Rate", converter=forms.float_converter(), ) self.GridAdd(self._samp_rate_text_box, 1, 0, 1, 3) self.nb0 = self.nb0 = wx.Notebook(self.GetWin(), style=wx.NB_TOP) self.nb0.AddPage(grc_wxgui.Panel(self.nb0), "FFT") self.nb0.AddPage(grc_wxgui.Panel(self.nb0), "Waterfall") self.nb0.AddPage(grc_wxgui.Panel(self.nb0), "Scope") self.GridAdd(self.nb0, 0, 0, 1, 1) _freq_sizer = wx.BoxSizer(wx.VERTICAL) self._freq_text_box = forms.text_box( parent=self.GetWin(), sizer=_freq_sizer, value=self.freq, callback=self.set_freq, label="RX Tune Frequency", converter=forms.float_converter(), proportion=0, ) self._freq_slider = forms.slider( parent=self.GetWin(), sizer=_freq_sizer, value=self.freq, callback=self.set_freq, minimum=50e6, maximum=6e9, num_steps=1000, style=wx.SL_HORIZONTAL, cast=float, proportion=1, ) self.Add(_freq_sizer) self.wxgui_waterfallsink2_0 = waterfallsink2.waterfall_sink_c( self.nb0.GetPage(1).GetWin(), baseband_freq=0, dynamic_range=100, ref_level=0, ref_scale=2.0, sample_rate=samp_rate, fft_size=512, fft_rate=15, average=False, avg_alpha=None, title="Waterfall Plot", size=((-1, 400)), ) self.nb0.GetPage(1).Add(self.wxgui_waterfallsink2_0.win) self.wxgui_scopesink2_0 = scopesink2.scope_sink_c( self.nb0.GetPage(2).GetWin(), title="Scope Plot", sample_rate=samp_rate, v_scale=0, v_offset=0, t_scale=0, ac_couple=False, xy_mode=False, num_inputs=1, trig_mode=wxgui.TRIG_MODE_AUTO, y_axis_label="Counts", ) self.nb0.GetPage(2).Add(self.wxgui_scopesink2_0.win) _gain_sizer = wx.BoxSizer(wx.VERTICAL) self._gain_text_box = forms.text_box( parent=self.GetWin(), sizer=_gain_sizer, value=self.gain, callback=self.set_gain, label="RX Gain", converter=forms.float_converter(), proportion=0, ) self._gain_slider = forms.slider( parent=self.GetWin(), sizer=_gain_sizer, value=self.gain, callback=self.set_gain, minimum=0, maximum=31.5, num_steps=63, style=wx.SL_HORIZONTAL, cast=float, proportion=1, ) self.GridAdd(_gain_sizer, 2, 0, 1, 8) self.fft = fftsink2.fft_sink_c( self.nb0.GetPage(0).GetWin(), baseband_freq=freq, y_per_div=10, y_divs=15, ref_level=0, ref_scale=2.0, sample_rate=samp_rate, fft_size=1024, fft_rate=15, average=False, avg_alpha=None, title="FFT Plot", peak_hold=False, size=((-1, 400)), ) self.nb0.GetPage(0).Add(self.fft.win) self.blocks_throttle_0 = blocks.throttle(gr.sizeof_gr_complex*1, samp_rate,True) self._ant_chooser = forms.radio_buttons( parent=self.GetWin(), value=self.ant, callback=self.set_ant, label="Antenna", choices=["RX2","TX/RX"], labels=["RX2","TX/RX"], style=wx.RA_HORIZONTAL, ) self.GridAdd(self._ant_chooser, 1, 4, 1, 2) self.analog_sig_source_x_0 = analog.sig_source_c(samp_rate, analog.GR_COS_WAVE, 50, 1, 0) ################################################## # Connections ################################################## self.connect((self.analog_sig_source_x_0, 0), (self.blocks_throttle_0, 0)) self.connect((self.blocks_throttle_0, 0), (self.wxgui_scopesink2_0, 0)) self.connect((self.blocks_throttle_0, 0), (self.fft, 0)) self.connect((self.blocks_throttle_0, 0), (self.wxgui_waterfallsink2_0, 0)) # QT sink close method reimplementation def get_param_freq(self): return self.param_freq def set_param_freq(self, param_freq): self.param_freq = param_freq self.set_freq(self.param_freq) def get_param_gain(self): return self.param_gain def set_param_gain(self, param_gain): self.param_gain = param_gain self.set_gain(self.param_gain) def get_address(self): return self.address def set_address(self, address): self.address = address def get_param_samp_rate(self): return self.param_samp_rate def set_param_samp_rate(self, param_samp_rate): self.param_samp_rate = param_samp_rate self.set_samp_rate(self.param_samp_rate) def get_samp_rate(self): return self.samp_rate def set_samp_rate(self, samp_rate): self.samp_rate = samp_rate self.analog_sig_source_x_0.set_sampling_freq(self.samp_rate) self.wxgui_scopesink2_0.set_sample_rate(self.samp_rate) self.fft.set_sample_rate(self.samp_rate) self.wxgui_waterfallsink2_0.set_sample_rate(self.samp_rate) self.blocks_throttle_0.set_sample_rate(self.samp_rate) self._samp_rate_text_box.set_value(self.samp_rate) def get_gain(self): return self.gain def set_gain(self, gain): self.gain = gain self._gain_slider.set_value(self.gain) self._gain_text_box.set_value(self.gain) def get_freq(self): return self.freq def set_freq(self, freq): self.freq = freq self.fft.set_baseband_freq(self.freq) self._freq_slider.set_value(self.freq) self._freq_text_box.set_value(self.freq) def get_ant(self): return self.ant def set_ant(self, ant): self.ant = ant self._ant_chooser.set_value(self.ant) if __name__ == '__main__': import ctypes import sys if sys.platform.startswith('linux'): try: x11 = ctypes.cdll.LoadLibrary('libX11.so') x11.XInitThreads() except: print "Warning: failed to XInitThreads()" parser = OptionParser(option_class=eng_option, usage="%prog: [options]") parser.add_option("-f", "--param-freq", dest="param_freq", type="eng_float", default=eng_notation.num_to_str(2.45e9), help="Set Default Frequency [default=%default]") parser.add_option("-g", "--param-gain", dest="param_gain", type="eng_float", default=eng_notation.num_to_str(0), help="Set Default Gain [default=%default]") parser.add_option("-a", "--address", dest="address", type="string", default="addr=192.168.11.2", help="Set IP Address [default=%default]") parser.add_option("-s", "--param-samp-rate", dest="param_samp_rate", type="eng_float", default=eng_notation.num_to_str(1e6), help="Set Sample Rate [default=%default]") (options, args) = parser.parse_args() tb = uhd_fft(param_freq=options.param_freq, param_gain=options.param_gain, address=options.address, param_samp_rate=options.param_samp_rate) tb.Start(True) tb.Wait()