# Copyright 2009 Free Software Foundation, Inc.
#
# This file is part of GNU Radio
#
# GNU Radio is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3, or (at your option)
# any later version.
#
# GNU Radio is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with GNU Radio; see the file COPYING.  If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street,
# Boston, MA 02110-1301, USA.
#

import common
from gnuradio import gr, usrp

####################################################################
# Simple USRP Source
####################################################################
class _simple_source(gr.hier_block2):
	"""A single usrp source of IO type short or complex."""

	def __init__(self, number, side='A', rx_ant='RXA'):
		"""
		USRP simple source contructor.
		@param number the unit number
		@param side the usrp side A or B
		@param rx_ant the antenna choice
		"""
		#initialize hier2 block
		gr.hier_block2.__init__(
			self, 'usrp_simple_source',
			gr.io_signature(0, 0, 0),
			gr.io_signature(1, 1, self._get_io_size()),
		)
		#create usrp object
		self._u = self._get_usrp_constructor()(number, nchan=1)
		subdev_spec = common.to_spec(side, rx_ant)
		self._u.set_mux(usrp.determine_rx_mux_value(self._u, subdev_spec))
		self._subdev = usrp.selected_subdev(self._u, subdev_spec)
		if common.is_flex(rx_ant): self._subdev.select_rx_antenna(rx_ant)
		#connect
		self.connect(self._u, self)

	def set_decim_rate(self, decim): self._u.set_decim_rate(int(decim))
	def set_frequency(self, frequency, verbose=False):
		common.set_frequency(
			u=self._u,
			which=0, #ddc0
			subdev=self._subdev,
			frequency=frequency,
			verbose=verbose,
		)
	def set_gain(self, gain): self._subdev.set_gain(gain)
	def set_auto_tr(self, auto_tr): self._subdev.set_auto_tr(auto_tr)

class simple_source_c(_simple_source, common.usrp_source_c): pass
class simple_source_s(_simple_source, common.usrp_source_s): pass

####################################################################
# Simple USRP Sink
####################################################################
class _simple_sink(gr.hier_block2):
	"""A single usrp sink of IO type short or complex."""

	def __init__(self, number, side='A'):
		"""
		USRP simple sink contructor.
		@param number the unit number
		@param side the usrp side A or B
		"""
		#initialize hier2 block
		gr.hier_block2.__init__(
			self, 'usrp_simple_sink',
			gr.io_signature(1, 1, self._get_io_size()),
			gr.io_signature(0, 0, 0),
		)
		#create usrp object
		self._u = self._get_usrp_constructor()(number, nchan=1)
		subdev_spec = common.to_spec(side)
		self._u.set_mux(usrp.determine_tx_mux_value(self._u, subdev_spec))
		self._subdev = usrp.selected_subdev(self._u, subdev_spec)
		#connect
		self.connect(self, self._u)

	def set_interp_rate(self, interp): self._u.set_interp_rate(int(interp))
	def set_frequency(self, frequency, verbose=False):
		common.set_frequency(
			u=self._u,
			which=self._subdev.which(),
			subdev=self._subdev,
			frequency=frequency,
			verbose=verbose,
		)
	def set_gain(self, gain): self._subdev.set_gain(gain)
	def set_enable(self, enable): self._subdev.set_enable(enable)
	def set_auto_tr(self, auto_tr): self._subdev.set_auto_tr(auto_tr)

class simple_sink_c(_simple_sink, common.usrp_sink_c): pass
class simple_sink_s(_simple_sink, common.usrp_sink_s): pass