Merged features/mp-sched -r8915:9335 into the trunk. The trunk now
[debian/gnuradio] / gr-pager / src / usrp_flex_all.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             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             src = usrp.source_c()
19             if options.rx_subdev_spec is None:
20                 options.rx_subdev_spec = usrp.pick_rx_subdevice(src)
21             subdev = usrp.selected_subdev(src, options.rx_subdev_spec)
22             src.set_mux(usrp.determine_rx_mux_value(src, options.rx_subdev_spec))
23             src.set_decim_rate(20)
24             result = usrp.tune(src, 0, subdev, 930.5e6+options.calibration)
25             if options.verbose:
26                 print "Using", subdev.name(), " for receiving."
27                 print "Tuned USRP to", 930.5e6+options.calibration
28                 
29         taps = gr.firdes.low_pass(1.0,
30                                   1.0,
31                                   1.0/128.0*0.4,
32                                   1.0/128.0*0.1,
33                                   gr.firdes.WIN_HANN)
34
35         if options.verbose:
36             print "Channel filter has", len(taps), "taps"
37
38         bank = blks2.analysis_filterbank(128, taps)
39         self.connect(src, bank)
40
41         if options.log and options.from_file == None:
42             src_sink = gr.file_sink(gr.sizeof_gr_complex, 'usrp.dat')
43             self.connect(src, src_sink)
44
45         for i in range(128):
46             if i < 64:
47                 freq = 930.5e6+i*25e3
48             else:
49                 freq = 928.9e6+(i-64)*25e3
50
51             if (freq < 929.0e6 or freq > 932.0e6):
52                 self.connect((bank, i), gr.null_sink(gr.sizeof_gr_complex))
53             else:
54                 self.connect((bank, i), pager.flex_demod(queue, freq, options.verbose, options.log))
55                 if options.log:
56                     self.connect((bank, i), gr.file_sink(gr.sizeof_gr_complex, 'chan_'+'%3.3f'%(freq/1e6)+'.dat'))
57
58 def main():
59     parser = OptionParser(option_class=eng_option)
60     parser.add_option("-R", "--rx-subdev-spec", type="subdev",
61                       help="select USRP Rx side A or B (default=first daughterboard found)")
62     parser.add_option("-c", "--calibration", type="eng_float", default=0.0,
63                       help="set frequency offset to Hz", metavar="Hz")
64     parser.add_option("-g", "--gain", type="int",
65                       help="set RF gain", metavar="dB")
66     parser.add_option("-F", "--from-file", default=None,
67                       help="Read from file instead of USRP")
68     parser.add_option("-l", "--log", action="store_true", default=False,
69                       help="log flowgraph to files (LOTS of data)")
70     parser.add_option("-v", "--verbose", action="store_true", default=False,
71                       help="display debug output")
72     (options, args) = parser.parse_args()
73
74     if options.verbose:
75         print options
76
77     queue = gr.msg_queue()
78     tb = app_top_block(options, queue)
79
80     runner = pager.top_block_runner(tb)
81     try:
82         while 1:
83             if not queue.empty_p():
84                 msg = queue.delete_head() # Blocking read
85                 page = join(split(msg.to_string(), chr(128)), '|')
86                 s = pager.make_printable(page)
87                 print s
88             elif runner.done:
89                 break
90             else:
91                 time.sleep(0.05)
92
93     except KeyboardInterrupt:
94         tb.stop()
95         runner = None
96     
97 if __name__ == "__main__":
98     main()