#!/usr/bin/env python
##################################################
# Gnuradio Python Flow Graph
# Title: Mono FM Radio
# Author: OZ9AEC
# Description: WFM receiver (mono only)
# Generated: Sun Dec 29 06:28:43 2013
##################################################

from gnuradio import audio
from gnuradio import blks2
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 fftsink2
from gnuradio.wxgui import forms
from grc_gnuradio import wxgui as grc_wxgui
from optparse import OptionParser
import kernel
import wx

class wfm_rx(grc_wxgui.top_block_gui):

	def __init__(self):
		grc_wxgui.top_block_gui.__init__(self, title="Mono FM Radio")

		##################################################
		# Variables
		##################################################
		self.usrp_decim = usrp_decim = 1
		self.xlate_tune = xlate_tune = 0
		self.usrp_freq = usrp_freq = 96.5e6
		self.samp_rate = samp_rate = 2048000/usrp_decim
		self.rx_freq = rx_freq = usrp_freq+xlate_tune
		self.rf_gain = rf_gain = 20
		self.filter_taps = filter_taps = firdes.low_pass(1, samp_rate, 250000, 20000, firdes.WIN_HAMMING, 6.76)
		self.af_gain = af_gain = 1.0

		##################################################
		# Blocks
		##################################################
		_xlate_tune_sizer = wx.BoxSizer(wx.VERTICAL)
		self._xlate_tune_text_box = forms.text_box(
			parent=self.GetWin(),
			sizer=_xlate_tune_sizer,
			value=self.xlate_tune,
			callback=self.set_xlate_tune,
			label="Fine frequency",
			converter=forms.float_converter(),
			proportion=0,
		)
		self._xlate_tune_slider = forms.slider(
			parent=self.GetWin(),
			sizer=_xlate_tune_sizer,
			value=self.xlate_tune,
			callback=self.set_xlate_tune,
			minimum=-250e3,
			maximum=250e3,
			num_steps=500,
			style=wx.SL_HORIZONTAL,
			cast=float,
			proportion=1,
		)
		self.GridAdd(_xlate_tune_sizer, 7, 0, 1, 5)
		self.nbook = self.nbook = wx.Notebook(self.GetWin(), style=wx.NB_TOP)
		self.nbook.AddPage(grc_wxgui.Panel(self.nbook), "Receiver")
		self.nbook.AddPage(grc_wxgui.Panel(self.nbook), "Audio")
		self.GridAdd(self.nbook, 0, 0, 5, 5)
		_af_gain_sizer = wx.BoxSizer(wx.VERTICAL)
		self._af_gain_text_box = forms.text_box(
			parent=self.GetWin(),
			sizer=_af_gain_sizer,
			value=self.af_gain,
			callback=self.set_af_gain,
			label="AF",
			converter=forms.float_converter(),
			proportion=0,
		)
		self._af_gain_slider = forms.slider(
			parent=self.GetWin(),
			sizer=_af_gain_sizer,
			value=self.af_gain,
			callback=self.set_af_gain,
			minimum=0,
			maximum=10,
			num_steps=100,
			style=wx.SL_HORIZONTAL,
			cast=float,
			proportion=1,
		)
		self.GridAdd(_af_gain_sizer, 10, 2, 1, 1)
		self.xlating_fir_filter = gr.freq_xlating_fir_filter_ccc(4, (filter_taps), -xlate_tune, samp_rate)
		self.wfm_rcv = blks2.wfm_rcv(
			quad_rate=samp_rate,
			audio_decimation=10,
		)
		_usrp_freq_sizer = wx.BoxSizer(wx.VERTICAL)
		self._usrp_freq_text_box = forms.text_box(
			parent=self.GetWin(),
			sizer=_usrp_freq_sizer,
			value=self.usrp_freq,
			callback=self.set_usrp_freq,
			label="FM frequency",
			converter=forms.float_converter(),
			proportion=0,
		)
		self._usrp_freq_slider = forms.slider(
			parent=self.GetWin(),
			sizer=_usrp_freq_sizer,
			value=self.usrp_freq,
			callback=self.set_usrp_freq,
			minimum=88e6,
			maximum=108e6,
			num_steps=200,
			style=wx.SL_HORIZONTAL,
			cast=float,
			proportion=1,
		)
		self.GridAdd(_usrp_freq_sizer, 6, 0, 1, 5)
		self._rx_freq_static_text = forms.static_text(
			parent=self.GetWin(),
			value=self.rx_freq,
			callback=self.set_rx_freq,
			label="Receive",
			converter=forms.float_converter(),
		)
		self.GridAdd(self._rx_freq_static_text, 5, 3, 1, 1)
		self.rr_stereo_right = blks2.rational_resampler_fff(
			interpolation=48,
			decimation=50,
			taps=None,
			fractional_bw=None,
		)
		_rf_gain_sizer = wx.BoxSizer(wx.VERTICAL)
		self._rf_gain_text_box = forms.text_box(
			parent=self.GetWin(),
			sizer=_rf_gain_sizer,
			value=self.rf_gain,
			callback=self.set_rf_gain,
			label="RF",
			converter=forms.float_converter(),
			proportion=0,
		)
		self._rf_gain_slider = forms.slider(
			parent=self.GetWin(),
			sizer=_rf_gain_sizer,
			value=self.rf_gain,
			callback=self.set_rf_gain,
			minimum=0,
			maximum=50,
			num_steps=50,
			style=wx.SL_HORIZONTAL,
			cast=float,
			proportion=1,
		)
		self.GridAdd(_rf_gain_sizer, 9, 2, 1, 1)
		self.kernel_libv4l2_x_0 = kernel.libv4l2_x("/dev/swradio0")
		self.kernel_libv4l2_x_0.set_samp_rate(samp_rate)
		self.kernel_libv4l2_x_0.set_center_freq(97300000)
		self.kernel_libv4l2_x_0.set_bandwidth(200000)
		self.kernel_libv4l2_x_0.set_tuner_gain(55)
		  
		self.fftsink_rf = fftsink2.fft_sink_c(
			self.nbook.GetPage(0).GetWin(),
			baseband_freq=0,
			y_per_div=10,
			y_divs=10,
			ref_level=0,
			ref_scale=13490.0,
			sample_rate=samp_rate,
			fft_size=512,
			fft_rate=10,
			average=True,
			avg_alpha=0.5,
			title="Baseband",
			peak_hold=False,
		)
		self.nbook.GetPage(0).Add(self.fftsink_rf.win)
		self.audio_sink_0 = audio.sink(48000, "", True)
		self.af_gain_stereo_left = gr.multiply_const_vff((af_gain, ))

		##################################################
		# Connections
		##################################################
		self.connect((self.rr_stereo_right, 0), (self.af_gain_stereo_left, 0))
		self.connect((self.xlating_fir_filter, 0), (self.fftsink_rf, 0))
		self.connect((self.xlating_fir_filter, 0), (self.wfm_rcv, 0))
		self.connect((self.wfm_rcv, 0), (self.rr_stereo_right, 0))
		self.connect((self.kernel_libv4l2_x_0, 0), (self.xlating_fir_filter, 0))
		self.connect((self.af_gain_stereo_left, 0), (self.audio_sink_0, 0))


	def get_usrp_decim(self):
		return self.usrp_decim

	def set_usrp_decim(self, usrp_decim):
		self.usrp_decim = usrp_decim
		self.set_samp_rate(2048000/self.usrp_decim)

	def get_xlate_tune(self):
		return self.xlate_tune

	def set_xlate_tune(self, xlate_tune):
		self.xlate_tune = xlate_tune
		self._xlate_tune_slider.set_value(self.xlate_tune)
		self._xlate_tune_text_box.set_value(self.xlate_tune)
		self.set_rx_freq(self.usrp_freq+self.xlate_tune)
		self.xlating_fir_filter.set_center_freq(-self.xlate_tune)

	def get_usrp_freq(self):
		return self.usrp_freq

	def set_usrp_freq(self, usrp_freq):
		self.usrp_freq = usrp_freq
		self._usrp_freq_slider.set_value(self.usrp_freq)
		self._usrp_freq_text_box.set_value(self.usrp_freq)
		self.set_rx_freq(self.usrp_freq+self.xlate_tune)

	def get_samp_rate(self):
		return self.samp_rate

	def set_samp_rate(self, samp_rate):
		self.samp_rate = samp_rate
		self.set_filter_taps(firdes.low_pass(1, self.samp_rate, 250000, 20000, firdes.WIN_HAMMING, 6.76))
		self.fftsink_rf.set_sample_rate(self.samp_rate)
		self.kernel_libv4l2_x_0.set_samp_rate(self.samp_rate)

	def get_rx_freq(self):
		return self.rx_freq

	def set_rx_freq(self, rx_freq):
		self.rx_freq = rx_freq
		self._rx_freq_static_text.set_value(self.rx_freq)

	def get_rf_gain(self):
		return self.rf_gain

	def set_rf_gain(self, rf_gain):
		self.rf_gain = rf_gain
		self._rf_gain_slider.set_value(self.rf_gain)
		self._rf_gain_text_box.set_value(self.rf_gain)

	def get_filter_taps(self):
		return self.filter_taps

	def set_filter_taps(self, filter_taps):
		self.filter_taps = filter_taps
		self.xlating_fir_filter.set_taps((self.filter_taps))

	def get_af_gain(self):
		return self.af_gain

	def set_af_gain(self, af_gain):
		self.af_gain = af_gain
		self._af_gain_slider.set_value(self.af_gain)
		self._af_gain_text_box.set_value(self.af_gain)
		self.af_gain_stereo_left.set_k((self.af_gain, ))

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