3 Copyright 2009 Free Software Foundation, Inc.
4 This file is part of GNU Radio
6 GNU Radio Companion is free software; you can redistribute it and/or
7 modify it under the terms of the GNU General Public License
8 as published by the Free Software Foundation; either version 2
9 of the License, or (at your option) any later version.
11 GNU Radio Companion is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
21 from gnuradio import usrp2
30 from gnuradio.grc.gui.Dialogs import TextDisplay
32 from gnuradio.grc.python.Platform import Platform
35 from gnuradio.grc.gui.Platform import Platform
36 platform = Platform(platform)
38 flow_graph = platform.get_new_flow_graph()
39 block = flow_graph.get_new_block('usrp2_probe')
42 usrp_interface_param = block.get_param('interface')
43 usrp_type_param = block.get_param('type')
45 class USRP2ProbeWindow(gtk.Window):
47 The main window for USRP Dignostics.
50 def delete_event(self, widget, event, data=None): return False
52 def destroy(self, widget, data=None): gtk.main_quit()
56 USRP2ProbeWindow contructor.
57 Create a new gtk Dialog with a close button, USRP2 input paramaters, and output labels.
59 self.usrp2_macs = list()
60 gtk.Window.__init__(self, gtk.WINDOW_TOPLEVEL)
62 self.connect("delete_event", self.delete_event)
63 self.connect("destroy", self.destroy)
65 self.set_title('USRP2 Probe')
66 #create decorative frame
69 #create vbox for storage
72 vbox.pack_start(usrp_interface_param.get_input_object(), False)
73 vbox.pack_start(usrp_type_param.get_input_object(), False)
74 #make the tree model for holding mac addrs
75 self.treestore = gtk.TreeStore(gobject.TYPE_STRING)
76 self.treeview = gtk.TreeView(self.treestore)
77 self.treeview.set_enable_search(False) #disable pop up search box
78 self.treeview.add_events(gtk.gdk.BUTTON_PRESS_MASK)
79 self.treeview.connect('button_press_event', self._handle_selection)
80 selection = self.treeview.get_selection()
81 selection.set_mode('single')
82 selection.connect('changed', self._handle_selection)
83 renderer = gtk.CellRendererText()
84 column = gtk.TreeViewColumn('Select a USRP2 MAC Address', renderer, text=0)
85 self.treeview.append_column(column)
86 vbox.pack_start(self.treeview, False)
88 self.probe_button = gtk.Button('Probe')
89 self.probe_button.connect('clicked', self._probe_usrp2)
90 vbox.pack_start(self.probe_button, False)
91 #Create a text box for USRP queries
92 self.query_buffer = TextDisplay()
93 self.query_buffer.set_text(block.get_doc())
94 vbox.pack_start(self.query_buffer)
98 def _probe_usrp2(self, widget=None):
99 """Probe the USRP2 device and copy the results into the query text box."""
101 args = ['find_usrps']
102 interface = usrp_interface_param.evaluate()
103 if interface: args.extend(['-e', interface])
104 p = subprocess.Popen(args=args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, shell=False, universal_newlines=True)
105 msg = p.stdout.read()
107 self.usrp2_macs = sorted(map(lambda l: l.split()[0], filter(lambda l: l.count(':') >= 5, msg.strip().splitlines())))
108 #set the tree store with the mac addrs
109 self.treestore.clear()
110 for usrp2_mac in self.usrp2_macs: self.treestore.append(None, (usrp2_mac,))
111 #set the text with the error message for 0 found, hide the list
112 #when only 1 usrp2, auto handle selection, hide the list
113 #for multiple usrp2, show the list
114 if not self.usrp2_macs:
116 self.query_buffer.set_text(msg)
117 elif len(self.usrp2_macs) == 1:
119 self.query_buffer.set_text('')
120 self._handle_selection()
123 self.query_buffer.set_text('')
125 def _handle_selection(self, *args, **kwargs):
126 """A selection change or click occured."""
128 selection = self.treeview.get_selection()
129 treestore, iter = selection.get_selected()
130 mac_addr = iter and treestore.get_value(iter, 0) or ''
131 if not mac_addr and len(self.usrp2_macs) > 1:
132 return #no empty mac addrs for when multiple found
133 #make the usrp2 object
135 'rx': (usrp2.source_32fc, 'adc_rate'),
136 'tx': (usrp2.sink_32fc, 'dac_rate'),
137 }[usrp_type_param.evaluate()]
138 interface = usrp_interface_param.evaluate()
140 if not interface and not mac_addr: u = make()
141 elif not mac_addr: u = make(interface)
142 else: u = make(interface, mac_addr)
143 msg = ">>> USRP2 Probe\n"
144 msg = "%s\nMAC Addr:\n\t%s\n"%(msg, u.mac_addr())
145 msg = "%s\nName (ID):\n\t%s\n"%(msg, u.daughterboard_id())
146 msg = "%s\nConverter Rate:\n\t%s Hz\n"%(msg, getattr(u, rate_attr)())
147 gain_min, gain_max, gain_step = u.gain_range()
148 msg = "%s\nGain Range (min, max, step size):\n\t%s\n\t%s\n\t%s\n"%(msg, gain_min, gain_max, gain_step)
149 freq_min, freq_max = u.freq_range()
150 msg = "%s\nFreq Range (min, max):\n\t%s Hz\n\t%s Hz\n"%(msg, freq_min, freq_max)
151 self.query_buffer.set_text(msg)
152 except Exception, e: #display the error message
153 self.query_buffer.set_text('>>> Error\n%s'%str(e))
155 if __name__ == '__main__':
156 #setup icon using icon theme
157 try: gtk.window_set_default_icon(gtk.IconTheme().load_icon('gnuradio-grc', 256, 0))