Merge r6461:6464 from jcorgan/t162-staging into trunk.
[debian/gnuradio] / gr-pager / src / usrp_flex_band.py
1 #!/usr/bin/env python
2
3 from gnuradio import gr, gru, usrp, optfir, eng_notation, blks2, pager
4 from gnuradio.eng_option import eng_option
5 from optparse import OptionParser
6 from string import split, join, printable
7 import time
8
9 class app_top_block(gr.top_block):
10     def __init__(self, options, queue):
11         gr.top_block.__init__(self, "usrp_flex_all")
12
13         if options.from_file is not None:
14             self.src = gr.file_source(gr.sizeof_gr_complex, options.from_file)
15             if options.verbose:
16                 print "Reading samples from file", options.from_file
17         else:
18             self.src = usrp.source_c()
19             if options.rx_subdev_spec is None:
20                 options.rx_subdev_spec = usrp.pick_rx_subdevice(self.src)
21             self.subdev = usrp.selected_subdev(self.src, options.rx_subdev_spec)
22             self.src.set_mux(usrp.determine_rx_mux_value(self.src, options.rx_subdev_spec))
23             self.src.set_decim_rate(64)
24             frequency = options.frequency+options.calibration
25             result = usrp.tune(self.src, 0, self.subdev, frequency)
26             if options.verbose:
27                 print "Using", self.subdev.name(), " for receiving."
28                 print "Tuned USRP to", frequency
29                 
30         taps = gr.firdes.low_pass(1.0,
31                                   1.0,
32                                   1.0/40.0*0.4,
33                                   1.0/40.0*0.1,
34                                   gr.firdes.WIN_HANN)
35
36         if options.verbose:
37             print "Channel filter has", len(taps), "taps"
38
39         bank = blks2.analysis_filterbank(40, taps)
40         self.connect(self.src, bank)
41
42         if options.log and options.from_file == None:
43             src_sink = gr.file_sink(gr.sizeof_gr_complex, 'usrp.dat')
44             self.connect(self.src, src_sink)
45
46         for i in range(40):
47             if i < 20:
48                 freq = options.frequency+i*25e3
49             else:
50                 freq = options.frequency-0.5e6+(i-20)*25e3
51
52             self.connect((bank, i), pager.flex_demod(queue, freq, options.verbose, options.log))
53             if options.log:
54                 self.connect((bank, i), gr.file_sink(gr.sizeof_gr_complex, 'chan_'+'%3.3f'%(freq/1e6)+'.dat'))
55
56     def __del__(self):
57         # Avoid weak-ref error
58         del self.subdev
59         
60 def main():
61     parser = OptionParser(option_class=eng_option)
62     parser.add_option("-f", "--frequency", type="eng_float", default=929.5e6,
63                       help="set receive center frequency to Hz", metavar="Hz")
64     parser.add_option("-R", "--rx-subdev-spec", type="subdev",
65                       help="select USRP Rx side A or B (default=first daughterboard found)")
66     parser.add_option("-c", "--calibration", type="eng_float", default=0.0,
67                       help="set frequency offset to Hz", metavar="Hz")
68     parser.add_option("-g", "--gain", type="int",
69                       help="set RF gain", metavar="dB")
70     parser.add_option("-F", "--from-file", default=None,
71                       help="Read from file instead of USRP")
72     parser.add_option("-l", "--log", action="store_true", default=False,
73                       help="log flowgraph to files (LOTS of data)")
74     parser.add_option("-v", "--verbose", action="store_true", default=False,
75                       help="display debug output")
76     (options, args) = parser.parse_args()
77
78     if options.verbose:
79         print options
80
81     queue = gr.msg_queue()
82     tb = app_top_block(options, queue)
83
84     try:
85         tb.start()
86         while 1:
87             if not queue.empty_p():
88                 msg = queue.delete_head() # Blocking read
89                 page = join(split(msg.to_string(), chr(128)), '|')
90                 disp = []
91                 for n in range(len(page)):
92                     if ord(page[n]) < 32:
93                         disp.append('.')
94                     else:
95                         disp.append(page[n])
96                 print join(disp, '')
97                                                 
98             else:
99                 time.sleep(1)
100
101     except KeyboardInterrupt:
102         tb.stop()
103     
104 if __name__ == "__main__":
105     main()