#! /usr/bin/env python

from gnuradio import eng_notation
from gnuradio import gr
from gnuradio import window
from gnuradio.eng_option import eng_option
from gnuradio.gr import firdes
from gnuradio.wxgui import waterfallsink2
from grc_gnuradio import wxgui as grc_wxgui
from optparse import OptionParser
import wx

class top_block(grc_wxgui.top_block_gui):

    def __init__(self):
        grc_wxgui.top_block_gui.__init__(self, title="Top Block")

        self.samp_rate = samp_rate = 20000000

        self.gr_file_source_0 = gr.file_source(gr.sizeof_short*1, "/tmp/radio", False)
        self.gr_short_to_float_0 = gr.short_to_float()
        self.gr_throttle_0 = gr.throttle(gr.sizeof_short*1, samp_rate)
        self.wxgui_waterfallsink2_0 = waterfallsink2.waterfall_sink_f(
            self.GetWin(),
            baseband_freq=0,
            dynamic_range = 100,
            ref_level=50,
            ref_scale=2.0,
            sample_rate=samp_rate,
            fft_size=1024,
            fft_rate=20,
            average=False,
            avg_alpha=None,
            title="Waterfall Plot",
            win=window.flattop,
        )

        self.Add(self.wxgui_waterfallsink2_0.win)

        self.connect((self.gr_file_source_0, 0), (self.gr_throttle_0, 0))
        self.connect((self.gr_throttle_0, 0), (self.gr_short_to_float_0, 0))
        self.connect((self.gr_short_to_float_0, 0), (self.wxgui_waterfallsink2_0, 0))

    def set_samp_rate(self, samp_rate):
        self.samp_rate = samp_rate
        self.wxgui_waterfallsink2_0.set_sample_rate(self.sample_rate)

if __name__ == '__main__':
    parser = OptionParser(option_class=eng_option, usage="%prog: [options]")
    (options, args) = parser.parse_args()
    tb = top_block()
    tb.Run(True)