git.gag.com Git - debian/gnuradio/blob - gr-utils/src/python/usrp_siggen.py

   1 #!/usr/bin/env python
   2
   3 from gnuradio import gr, gru
   4 from gnuradio import usrp
   5 from gnuradio.eng_option import eng_option
   6 from gnuradio import eng_notation
   7 from optparse import OptionParser
   8 import sys
   9
  10
  11 class my_top_block(gr.top_block):
  12     def __init__ (self):
  13         gr.top_block.__init__(self)
  14
  15         # controllable values
  16         self.interp = 64
  17         self.waveform_type = gr.GR_SIN_WAVE
  18         self.waveform_ampl = 16000
  19         self.waveform_freq = 100.12345e3
  20         self.waveform_offset = 0
  21         self._instantiate_blocks ()
  22         self.set_waveform_type (self.waveform_type)
  23
  24     def usb_freq (self):
  25         return self.u.dac_freq() / self.interp
  26
  27     def usb_throughput (self):
  28         return self.usb_freq () * 4
  29
  30     def set_waveform_type (self, type):
  31         '''
  32         valid waveform types are: gr.GR_SIN_WAVE, gr.GR_CONST_WAVE,
  33         gr.GR_UNIFORM and gr.GR_GAUSSIAN
  34         '''
  35         self._configure_graph (type)
  36         self.waveform_type = type
  37
  38     def set_waveform_ampl (self, ampl):
  39         self.waveform_ampl = ampl
  40         self.siggen.set_amplitude (ampl)
  41         self.noisegen.set_amplitude (ampl)
  42
  43     def set_waveform_freq (self, freq):
  44         self.waveform_freq = freq
  45         self.siggen.set_frequency (freq)
  46
  47     def set_waveform_offset (self, offset):
  48         self.waveform_offset = offset
  49         self.siggen.set_offset (offset)
  50
  51     def set_interpolator (self, interp):
  52         self.interp = interp
  53         self.siggen.set_sampling_freq (self.usb_freq ())
  54         self.u.set_interp_rate (interp)
  55
  56     def _instantiate_blocks (self):
  57         self.src = None
  58         self.u = usrp.sink_c (0, self.interp)
  59
  60         self.siggen = gr.sig_source_c (self.usb_freq (),
  61                                        gr.GR_SIN_WAVE,
  62                                        self.waveform_freq,
  63                                        self.waveform_ampl,
  64                                        self.waveform_offset)
  65
  66         self.noisegen = gr.noise_source_c (gr.GR_UNIFORM,
  67                                            self.waveform_ampl)
  68
  69         # self.file_sink = gr.file_sink (gr.sizeof_gr_complex, "siggen.dat")
  70
  71     def _configure_graph (self, type):
  72         try:
  73             self.lock()
  74             self.disconnect_all ()
  75             if type == gr.GR_SIN_WAVE or type == gr.GR_CONST_WAVE:
  76                 self.connect (self.siggen, self.u)
  77                 # self.connect (self.siggen, self.file_sink)
  78                 self.siggen.set_waveform (type)
  79                 self.src = self.siggen
  80             elif type == gr.GR_UNIFORM or type == gr.GR_GAUSSIAN:
  81                 self.connect (self.noisegen, self.u)
  82                 self.noisegen.set_type (type)
  83                 self.src = self.noisegen
  84             else:
  85                 raise ValueError, type
  86         finally:
  87             self.unlock()
  88
  89     def set_freq(self, target_freq):
  90         """
  91         Set the center frequency we're interested in.
  92
  93         @param target_freq: frequency in Hz
  94         @rypte: bool
  95
  96         Tuning is a two step process.  First we ask the front-end to
  97         tune as close to the desired frequency as it can.  Then we use
  98         the result of that operation and our target_frequency to
  99         determine the value for the digital up converter.
 100         """
 101         r = self.u.tune(self.subdev._which, self.subdev, target_freq)
 102         if r:
 103             #print "r.baseband_freq =", eng_notation.num_to_str(r.baseband_freq)
 104             #print "r.dxc_freq      =", eng_notation.num_to_str(r.dxc_freq)
 105             #print "r.residual_freq =", eng_notation.num_to_str(r.residual_freq)
 106             #print "r.inverted      =", r.inverted
 107             return True
 108
 109         return False
 110
 111
 112
 113 def main ():
 114     parser = OptionParser (option_class=eng_option)
 115     parser.add_option ("-T", "--tx-subdev-spec", type="subdev", default=(0, 0),
 116                        help="select USRP Tx side A or B")
 117     parser.add_option ("-f", "--rf-freq", type="eng_float", default=None,
 118                        help="set RF center frequency to FREQ")
 119     parser.add_option ("-i", "--interp", type="int", default=64,
 120                        help="set fgpa interpolation rate to INTERP [default=%default]")
 121
 122     parser.add_option ("--sine", dest="type", action="store_const", const=gr.GR_SIN_WAVE,
 123                        help="generate a complex sinusoid [default]", default=gr.GR_SIN_WAVE)
 124     parser.add_option ("--const", dest="type", action="store_const", const=gr.GR_CONST_WAVE,
 125                        help="generate a constant output")
 126     parser.add_option ("--gaussian", dest="type", action="store_const", const=gr.GR_GAUSSIAN,
 127                        help="generate Gaussian random output")
 128     parser.add_option ("--uniform", dest="type", action="store_const", const=gr.GR_UNIFORM,
 129                        help="generate Uniform random output")
 130
 131     parser.add_option ("-w", "--waveform-freq", type="eng_float", default=100e3,
 132                        help="set waveform frequency to FREQ [default=%default]")
 133     parser.add_option ("-a", "--amplitude", type="eng_float", default=16e3,
 134                        help="set waveform amplitude to AMPLITUDE [default=%default]", metavar="AMPL")
 135     parser.add_option ("-g", "--gain", type="eng_float", default=None,
 136                        help="set output gain to GAIN [default=%default]")
 137     parser.add_option ("-o", "--offset", type="eng_float", default=0,
 138                        help="set waveform offset to OFFSET [default=%default]")
 139     (options, args) = parser.parse_args ()
 140
 141     if len(args) != 0:
 142         parser.print_help()
 143         raise SystemExit
 144
 145     if options.rf_freq is None:
 146         sys.stderr.write("usrp_siggen: must specify RF center frequency with -f RF_FREQ\n")
 147         parser.print_help()
 148         raise SystemExit
 149
 150     tb = my_top_block()
 151     tb.set_interpolator (options.interp)
 152     tb.set_waveform_type (options.type)
 153     tb.set_waveform_freq (options.waveform_freq)
 154     tb.set_waveform_ampl (options.amplitude)
 155     tb.set_waveform_offset (options.offset)
 156
 157     # determine the daughterboard subdevice we're using
 158     if options.tx_subdev_spec is None:
 159         options.tx_subdev_spec = usrp.pick_tx_subdevice(tb.u)
 160
 161     m = usrp.determine_tx_mux_value(tb.u, options.tx_subdev_spec)
 162     #print "mux = %#04x" % (m,)
 163     tb.u.set_mux(m)
 164     tb.subdev = usrp.selected_subdev(tb.u, options.tx_subdev_spec)
 165     print "Using TX d'board %s" % (tb.subdev.side_and_name(),)
 166
 167     if options.gain is None:
 168         tb.subdev.set_gain(tb.subdev.gain_range()[1])    # set max Tx gain
 169     else:
 170         tb.subdev.set_gain(options.gain)    # set max Tx gain
 171
 172     if not tb.set_freq(options.rf_freq):
 173         sys.stderr.write('Failed to set RF frequency\n')
 174         raise SystemExit
 175
 176     tb.subdev.set_enable(True)                       # enable transmitter
 177
 178     try:
 179         tb.run()
 180     except KeyboardInterrupt:
 181         pass
 182
 183 if __name__ == '__main__':
 184     main ()