1 # Copyright 2009, 2010 Free Software Foundation, Inc.
3 # This file is part of GNU Radio
5 # GNU Radio is free software; you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation; either version 3, or (at your option)
10 # GNU Radio is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details.
15 # You should have received a copy of the GNU General Public License
16 # along with GNU Radio; see the file COPYING. If not, write to
17 # the Free Software Foundation, Inc., 51 Franklin Street,
18 # Boston, MA 02110-1301, USA.
22 from gnuradio import gr
24 ####################################################################
26 ####################################################################
27 class _dual_source(gr.hier_block2):
28 """A dual usrp source of IO type short or complex."""
30 def __init__(self, which, rx_ant_a='RXA', rx_ant_b='RXA', rx_source_a='A', rx_source_b='B'):
32 USRP dual source contructor.
33 @param which the unit number
34 @param rx_ant_a the antenna choice
35 @param rx_ant_b the antenna choice
37 #initialize hier2 block
38 gr.hier_block2.__init__(
39 self, 'usrp_dual_source',
40 gr.io_signature(0, 0, 0),
41 gr.io_signature(2, 2, self._get_io_size()),
44 self._make_usrp(which=which, nchan=2)
45 subdev_spec_a = common.to_spec(rx_source_a, rx_ant_a)
46 subdev_spec_b = common.to_spec(rx_source_b, rx_ant_b)
47 self._get_u().set_mux(self._get_u().determine_rx_mux_value(subdev_spec_a, subdev_spec_b))
48 self._subdev_a = self._get_u().selected_subdev(subdev_spec_a)
49 self._subdev_b = self._get_u().selected_subdev(subdev_spec_b)
51 deinter = gr.deinterleave(self._get_io_size())
52 self.connect(self._get_u(), deinter)
53 for i in range(2): self.connect((deinter, i), (self, i))
55 def set_decim_rate(self, decim): self._get_u().set_decim_rate(int(decim))
56 def set_frequency_a(self, frequency, verbose=False, lo_offset=None):
57 if lo_offset is not None: self._subdev_a.set_lo_offset(lo_offset)
60 subdev=self._subdev_a,
64 def set_frequency_b(self, frequency, verbose=False, lo_offset=None):
65 if lo_offset is not None: self._subdev_b.set_lo_offset(lo_offset)
68 subdev=self._subdev_b,
72 def set_gain_a(self, gain): self._subdev_a.set_gain(gain)
73 def set_gain_b(self, gain): self._subdev_b.set_gain(gain)
75 class dual_source_c(_dual_source, common.usrp_source_c): pass
76 class dual_source_s(_dual_source, common.usrp_source_s): pass
78 ####################################################################
80 ####################################################################
81 class _dual_sink(gr.hier_block2):
82 """A dual usrp sink of IO type short or complex."""
84 def __init__(self, which):
86 USRP simple sink contructor.
87 @param which the unit number
89 #initialize hier2 block
90 gr.hier_block2.__init__(
91 self, 'usrp_dual_sink',
92 gr.io_signature(2, 2, self._get_io_size()),
93 gr.io_signature(0, 0, 0),
96 self._make_usrp(which=which, nchan=2)
97 subdev_spec_a = common.to_spec('A')
98 subdev_spec_b = common.to_spec('B')
99 self._get_u().set_mux(self._get_u().determine_tx_mux_value(subdev_spec_a, subdev_spec_b))
100 self._subdev_a = self._get_u().selected_subdev(subdev_spec_a)
101 self._subdev_b = self._get_u().selected_subdev(subdev_spec_b)
103 inter = gr.interleave(self._get_io_size())
104 self.connect(inter, self._get_u())
105 for i in range(2): self.connect((self, i), (inter, i))
107 def set_interp_rate(self, interp): self._get_u().set_interp_rate(int(interp))
108 def set_frequency_a(self, frequency, verbose=False, lo_offset=None):
109 if lo_offset is not None: self._subdev_a.set_lo_offset(lo_offset)
111 chan=self._subdev_a.which(),
112 subdev=self._subdev_a,
116 def set_frequency_b(self, frequency, verbose=False, lo_offset=None):
117 if lo_offset is not None: self._subdev_b.set_lo_offset(lo_offset)
119 chan=self._subdev_b.which(),
120 subdev=self._subdev_b,
124 def set_gain_a(self, gain): self._subdev_a.set_gain(gain)
125 def set_gain_b(self, gain): self._subdev_b.set_gain(gain)
126 def set_enable_a(self, enable): self._subdev_a.set_enable(enable)
127 def set_enable_b(self, enable): self._subdev_b.set_enable(enable)
128 def set_auto_tr_a(self, auto_tr): self._subdev_a.set_auto_tr(auto_tr)
129 def set_auto_tr_b(self, auto_tr): self._subdev_b.set_auto_tr(auto_tr)
131 class dual_sink_c(_dual_sink, common.usrp_sink_c): pass
132 class dual_sink_s(_dual_sink, common.usrp_sink_s): pass