#!/usr/bin/env python
##################################################
# Gnuradio Python Flow Graph
# Title: Fm Example
# Generated: Tue Jan  6 10:37:52 2015
##################################################

from gnuradio import analog
from gnuradio import audio
from gnuradio import eng_notation
from gnuradio import filter
from gnuradio import gr
from gnuradio import uhd
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 grc_gnuradio import wxgui as grc_wxgui
from optparse import OptionParser
import time
import wx

class fm_example(grc_wxgui.top_block_gui):

    def __init__(self):

        grc_wxgui.top_block_gui.__init__(self, title="Fm Example")
        _icon_path = "/usr/share/icons/hicolor/32x32/apps/gnuradio-grc.png"
        self.SetIcon(wx.Icon(_icon_path, wx.BITMAP_TYPE_ANY))

        ##################################################
        # Variables
        ##################################################
        self.samp_rate = samp_rate = 4e6
        #self.freq = freq = 88.1e6
	self.freq = freq
        ##################################################
        # Blocks
        ##################################################
        self.notebook_0 = self.notebook_0 = wx.Notebook(self.GetWin(), style=wx.NB_LEFT)
        self.notebook_0.AddPage(grc_wxgui.Panel(self.notebook_0), "RF Spectrum")
        self.notebook_0.AddPage(grc_wxgui.Panel(self.notebook_0), "Demod Spectrum")
        self.Add(self.notebook_0)
        self._freq_text_box = forms.text_box(
        	parent=self.GetWin(),
        	value=self.freq,
        	callback=self.set_freq,
        	label='freq',
        	converter=forms.float_converter(),
        )
        self.Add(self._freq_text_box)
        self.wxgui_fftsink2_1 = fftsink2.fft_sink_f(
        	self.notebook_0.GetPage(1).GetWin(),
        	baseband_freq=0,
        	y_per_div=10,
        	y_divs=10,
        	ref_level=0,
        	ref_scale=2.0,
        	sample_rate=600e3,
        	fft_size=1024,
        	fft_rate=15,
        	average=False,
        	avg_alpha=None,
        	title="FFT Plot",
        	peak_hold=False,
        	win=window.rectangular,
        )
        self.notebook_0.GetPage(1).Add(self.wxgui_fftsink2_1.win)
        self.wxgui_fftsink2_0 = fftsink2.fft_sink_c(
        	self.notebook_0.GetPage(0).GetWin(),
        	baseband_freq=freq,
        	y_per_div=10,
        	y_divs=10,
        	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,
        )
        self.notebook_0.GetPage(0).Add(self.wxgui_fftsink2_0.win)
        self.uhd_usrp_source_0 = uhd.usrp_source(
        	",".join(("serial=2R2CU0U1", "")),
        	uhd.stream_args(
        		cpu_format="fc32",
        		channels=range(1),
        	),
        )
        self.uhd_usrp_source_0.set_subdev_spec("A:0", 0)
        self.uhd_usrp_source_0.set_samp_rate(samp_rate)
        self.uhd_usrp_source_0.set_center_freq(freq, 0)
        self.uhd_usrp_source_0.set_gain(15, 0)
        self.uhd_usrp_source_0.set_antenna("RX2", 0)
        self.rational_resampler_xxx_0 = filter.rational_resampler_fff(
                interpolation=240,
                decimation=48,
                taps=None,
                fractional_bw=None,
        )
        self.low_pass_filter_0 = filter.fir_filter_ccf(20, firdes.low_pass(
        	1, samp_rate, 100e3, 10e3, firdes.WIN_HANN, 6.76))
        self.audio_sink_0 = audio.sink(48000, "", False)
        self.analog_wfm_rcv_0 = analog.wfm_rcv(
        	quad_rate=600e3,
        	audio_decimation=20,
        )

        ##################################################
        # Connections
        ##################################################
        self.connect((self.uhd_usrp_source_0, 0), (self.wxgui_fftsink2_0, 0))
        self.connect((self.uhd_usrp_source_0, 0), (self.low_pass_filter_0, 0))
        self.connect((self.analog_wfm_rcv_0, 0), (self.wxgui_fftsink2_1, 0))
        self.connect((self.low_pass_filter_0, 0), (self.analog_wfm_rcv_0, 0))
        self.connect((self.rational_resampler_xxx_0, 0), (self.audio_sink_0, 0))
        self.connect((self.analog_wfm_rcv_0, 0), (self.rational_resampler_xxx_0, 0))





    def get_samp_rate(self):
        return self.samp_rate

    def set_samp_rate(self, samp_rate):
        self.samp_rate = samp_rate
        self.wxgui_fftsink2_0.set_sample_rate(self.samp_rate)
        self.uhd_usrp_source_0.set_samp_rate(self.samp_rate)
        self.low_pass_filter_0.set_taps(firdes.low_pass(1, self.samp_rate, 100e3, 10e3, firdes.WIN_HANN, 6.76))

    def get_freq(self):
        return self.freq

    def set_freq(self, freq):
        self.freq = freq
        self.wxgui_fftsink2_0.set_baseband_freq(self.freq)
        self.uhd_usrp_source_0.set_center_freq(self.freq, 0)
        self._freq_text_box.set_value(self.freq)

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]")
    (options, args) = parser.parse_args()
    freq = 88.1e6
    #88.3
    tb1 = fm_example()
    tb1.Start(True)
    tb1.Wait()
    freq = freq+200e3
    #89.1
    tb2 = fm_example()
    tb2.Start(True)
    tb2.Wait()
    freq = freq+800e3
    #89.9
    tb3 = fm_example()
    tb3.Start(True)
    tb3.Wait()
    freq = freq+800e3
    #90.7
    tb4 = fm_example()
    tb4.Start(True)
    tb4.Wait()
    freq = freq+800e3
    #91.5
    tb5 = fm_example()
    tb5.Start(True)
    tb5.Wait()
    freq = freq+800e3
    #91.8
    tb6 = fm_example()
    tb6.Start(True)
    tb6.Wait()
    freq = freq+300e3
    #92.3
    tb7 = fm_example()
    tb7.Start(True)
    tb7.Wait()
    freq = freq+500e3
    #93.1
    tb8 = fm_example()
    tb8.Start(True)
    tb8.Wait()
    freq = freq+800e3
    #93.9
    tb9 = fm_example()
    tb9.Start(True)
    tb9.Wait()
    freq = freq+800e3
    #94.7
    tb10 = fm_example()
    tb10.Start(True)
    tb10.Wait()
    freq = freq+800e3
    #95.5
    tb11 = fm_example()
    tb11.Start(True)
    tb11.Wait()
    freq = freq+800e3
    #96.3
    tb12 = fm_example()
    tb12.Start(True)
    tb12.Wait()
    freq = freq+800e3
    #97.1
    tb13 = fm_example()
    tb13.Start(True)
    tb13.Wait()
    freq = freq+800e3
    #97.9
    tb14 = fm_example()
    tb14.Start(True)
    tb14.Wait()
    freq = freq+800e3
    #98.7
    tb15 = fm_example()
    tb15.Start(True)
    tb15.Wait()
    freq = freq+800e3
    #99.5
    tb16 = fm_example()
    tb16.Start(True)
    tb16.Wait()
    freq = freq+800e3
    #100.3
    tb17 = fm_example()
    tb17.Start(True)
    tb17.Wait()
    freq = freq+800e3
    #101.1
    tb18 = fm_example()
    tb18.Start(True)
    tb18.Wait()
    freq = freq+800e3
    #101.9
    tb19 = fm_example()
    tb19.Start(True)
    tb19.Wait()
    freq = freq+800e3
    #102.3
    tb20 = fm_example()
    tb20.Start(True)
    tb20.Wait()
    freq = freq+400e3
    #102.7
    tb21 = fm_example()
    tb21.Start(True)
    tb21.Wait()
    freq = freq+400e3
    #103.5
    tb22 = fm_example()
    tb22.Start(True)
    tb22.Wait()
    freq = freq+800e3
    #104.3 
    tb23 = fm_example()
    tb23.Start(True)
    tb23.Wait()
    freq = freq+800e3
    #105.1
    tb24 = fm_example()
    tb24.Start(True)
    tb24.Wait()
    freq = freq+800e3
    #105.9
    tb25 = fm_example()
    tb25.Start(True)
    tb25.Wait()
    freq = freq+800e3
    #106.3
    tb26 = fm_example()
    tb26.Start(True)
    tb26.Wait()
    freq = freq+400e3
    #106.7
    tb27 = fm_example()
    tb27.Start(True)
    tb27.Wait()
    freq = freq+400e3
    #107.5
    tb28 = fm_example()
    tb28.Start(True)
    tb28.Wait()
    freq = freq+800e3
    #107.9
    tb29 = fm_example()
    tb29.Start(True)
    tb29.Wait()
    freq = freq+400e3
    #108
    tb30 = fm_example()
    tb30.Start(True)
    tb30.Wait()
    freq = freq+100e3