| 1 | #!/usr/bin/env python |
|---|
| 2 | |
|---|
| 3 | """ |
|---|
| 4 | Read samples from the USRP and write to file formatted as binary |
|---|
| 5 | single-precision complex values. |
|---|
| 6 | |
|---|
| 7 | """ |
|---|
| 8 | |
|---|
| 9 | from gnuradio import gr, eng_notation |
|---|
| 10 | from gnuradio import usrp |
|---|
| 11 | from gnuradio.eng_option import eng_option |
|---|
| 12 | from optparse import OptionParser |
|---|
| 13 | import sys |
|---|
| 14 | |
|---|
| 15 | class my_graph(gr.flow_graph): |
|---|
| 16 | |
|---|
| 17 | def __init__(self): |
|---|
| 18 | gr.flow_graph.__init__(self) |
|---|
| 19 | |
|---|
| 20 | usage="%prog: [options] output_filename" |
|---|
| 21 | parser = OptionParser(option_class=eng_option, usage=usage) |
|---|
| 22 | parser.add_option("-R", "--rx-subdev-spec", type="subdev", default=(0, 0), |
|---|
| 23 | help="select USRP Rx side A or B (default=A)") |
|---|
| 24 | parser.add_option("-u", "--usrp", type="int", default=0, |
|---|
| 25 | help="set the usrp board to receive signal [default=%default]") |
|---|
| 26 | parser.add_option("-d", "--decim", type="int", default=16, |
|---|
| 27 | help="set fgpa decimation rate to DECIM [default=%default]") |
|---|
| 28 | parser.add_option("-f", "--freq", type="eng_float", default=None, |
|---|
| 29 | help="set frequency to FREQ", metavar="FREQ") |
|---|
| 30 | parser.add_option("-g", "--gain", type="eng_float", default=None, |
|---|
| 31 | help="set gain in dB (default is midpoint)") |
|---|
| 32 | parser.add_option("-N", "--nsamples", type="eng_float", default=None, |
|---|
| 33 | help="number of samples to collect [default=+inf]") |
|---|
| 34 | (options, args) = parser.parse_args () |
|---|
| 35 | if len(args) != 1: |
|---|
| 36 | parser.print_help() |
|---|
| 37 | raise SystemExit, 1 |
|---|
| 38 | filename = args[0] |
|---|
| 39 | |
|---|
| 40 | if options.freq is None: |
|---|
| 41 | parser.print_help() |
|---|
| 42 | sys.stderr.write('You must specify the frequency with -f FREQ\n'); |
|---|
| 43 | raise SystemExit, 1 |
|---|
| 44 | |
|---|
| 45 | # build the graph |
|---|
| 46 | self.u = usrp.source_c(which=options.usrp,decim_rate=options.decim) |
|---|
| 47 | self.dst = gr.file_sink(gr.sizeof_gr_complex, filename) |
|---|
| 48 | if options.nsamples is None: |
|---|
| 49 | self.connect(self.u, self.dst) |
|---|
| 50 | else: |
|---|
| 51 | self.head = gr.head(gr.sizeof_gr_complex, int(options.nsamples)) |
|---|
| 52 | self.connect(self.u, self.head, self.dst) |
|---|
| 53 | |
|---|
| 54 | if options.rx_subdev_spec is None: |
|---|
| 55 | options.rx_subdev_spec = usrp.pick_rx_subdevice(self.u) |
|---|
| 56 | m = usrp.determine_rx_mux_value(self.u, options.rx_subdev_spec) |
|---|
| 57 | print "mux = %#04x" % (m,) |
|---|
| 58 | self.u.set_mux(usrp.determine_rx_mux_value(self.u, options.rx_subdev_spec)) |
|---|
| 59 | #self.u.set_mux(0xf0) |
|---|
| 60 | #self.u.set_pga(options.usrp,10) |
|---|
| 61 | |
|---|
| 62 | # determine the daughterboard subdevice we're using |
|---|
| 63 | self.subdev = usrp.selected_subdev(self.u, options.rx_subdev_spec) |
|---|
| 64 | print "subdev", options.rx_subdev_spec |
|---|
| 65 | print "Using RX d'board %s" % (self.subdev.side_and_name(),) |
|---|
| 66 | input_rate = self.u.adc_freq() / self.u.decim_rate() |
|---|
| 67 | print "USB sample rate %s" % (eng_notation.num_to_str(input_rate)) |
|---|
| 68 | |
|---|
| 69 | if options.gain is None: |
|---|
| 70 | # if no gain was specified, use the mid-point in dB |
|---|
| 71 | g = self.subdev.gain_range() |
|---|
| 72 | print g[0]," ",g[1], " ",g[2]," ",float(g[0]+g[1])/2 |
|---|
| 73 | options.gain = float(g[0]+g[1])/2 |
|---|
| 74 | self.subdev.set_gain(options.gain) |
|---|
| 75 | print "gain is:",options.gain |
|---|
| 76 | |
|---|
| 77 | r = self.u.tune(0, self.subdev, options.freq) |
|---|
| 78 | if not r: |
|---|
| 79 | sys.stderr.write('Failed to set frequency\n') |
|---|
| 80 | raise SystemExit, 1 |
|---|
| 81 | |
|---|
| 82 | |
|---|
| 83 | if __name__ == '__main__': |
|---|
| 84 | try: |
|---|
| 85 | my_graph().run() |
|---|
| 86 | except KeyboardInterrupt: |
|---|
| 87 | pass |
|---|
