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