git.gag.com Git - debian/gnuradio/blob - gnuradio-examples/python/usrp/flexrf_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 optparse import OptionParser
   7 from gnuradio.wxgui import stdgui, slider
   8 import wx
   9
  10 class flex_siggen (stdgui.gui_flow_graph):
  11     __slots__ = ['interp', 'waveform_type', 'waveform_ampl',
  12                  'waveform_freq', 'waveform_offset', 'fg', 'usrp',
  13                  'siggen', 'noisegen', 'src', 'file_sink' ]
  14
  15     def __init__ (self,frame,panel,vbox,argv):
  16         stdgui.gui_flow_graph.__init__ (self, frame, panel, vbox, argv)
  17
  18         self.frame = frame
  19         self.panel = panel
  20
  21         parser = OptionParser (option_class=eng_option)
  22         parser.add_option ("-a", "--amplitude", type="int", default=32000,
  23                            help="amplitude")
  24         parser.add_option ("-i", "--interp", type="int", default=64,
  25                            help="set fpga interpolation rate to INTERP")
  26         parser.add_option ("-n", "--nchannels", type="int", default=1,
  27                            help="set number of output channels to NCHANNELS")
  28         (options, args) = parser.parse_args ()
  29
  30         self.waveform_type = gr.GR_CONST_WAVE
  31         self.waveform_ampl = options.amplitude
  32         self.waveform_freq = 100.12345e3
  33         self.waveform_offset = 0
  34
  35         self.interp = options.interp
  36         self._instantiate_blocks ()
  37         self.usrp.set_nchannels (options.nchannels)
  38
  39         self.dboard=self.usrp.db[0][0]
  40
  41         self.set_waveform_type (self.waveform_type)
  42         vbox.Add(slider.slider(panel, 390, 510, self.set_rf_freq), 1, wx.EXPAND|wx.ALIGN_CENTER)
  43         vbox.Add(slider.slider(panel, -45000, +45000, self.set_if_freq), 1, wx.EXPAND|wx.ALIGN_CENTER)
  44         #vbox.Add(slider.slider(panel, 0, 4095, self.set_gain), 1, wx.EXPAND|wx.ALIGN_CENTER)
  45
  46     def usb_freq (self):
  47         return self.usrp.dac_freq() / self.interp
  48
  49     def usb_throughput (self):
  50         return self.usb_freq () * 4
  51
  52     def set_waveform_type (self, type):
  53         '''
  54         valid waveform types are: gr.GR_SIN_WAVE, gr.GR_CONST_WAVE,
  55         gr.GR_UNIFORM and gr.GR_GAUSSIAN
  56         '''
  57         self._configure_graph (type)
  58         self.waveform_type = type
  59
  60     def set_waveform_ampl (self, ampl):
  61         self.waveform_ampl = ampl
  62         self.siggen.set_amplitude (ampl)
  63         self.noisegen.set_amplitude (ampl)
  64
  65     def set_waveform_freq (self, freq):
  66         self.waveform_freq = freq
  67         self.siggen.set_frequency (freq)
  68
  69     def set_if_freq (self, freq):
  70         self.if_freq = freq
  71         self.usrp.set_tx_freq (0,freq*1e3)
  72
  73     def set_rf_freq (self, freq):
  74         self.rf_freq = freq
  75         (success,actual_freq) = self.dboard.set_freq (freq*1e6)
  76         if not success:
  77             print "Failed on ", freq
  78
  79     def set_waveform_offset (self, offset):
  80         self.waveform_offset = offset
  81         self.siggen.set_offset (offset)
  82
  83     def set_interpolator (self, interp):
  84         self.interp = interp
  85         self.siggen.set_sampling_freq (self.usb_freq ())
  86         self.usrp.set_interp_rate (interp)
  87
  88     def set_duc_freq (self, freq):
  89         self.usrp.set_tx_freq (0, freq)
  90
  91     def _instantiate_blocks (self):
  92         self.src = None
  93         self.usrp = usrp.sink_c (0, self.interp)
  94
  95         self.siggen = gr.sig_source_c (self.usb_freq (),
  96                                        gr.GR_SIN_WAVE,
  97                                        self.waveform_freq,
  98                                        self.waveform_ampl,
  99                                        self.waveform_offset)
 100
 101         self.noisegen = gr.noise_source_c (gr.GR_UNIFORM,
 102                                            self.waveform_ampl)
 103         print "done"
 104
 105     def _configure_graph (self, type):
 106         was_running = self.is_running ()
 107         if was_running:
 108             self.stop ()
 109         self.disconnect_all ()
 110         if type == gr.GR_SIN_WAVE or type == gr.GR_CONST_WAVE:
 111             self.connect (self.siggen, self.usrp)
 112             self.siggen.set_waveform (type)
 113             self.src = self.siggen
 114         elif type == gr.GR_UNIFORM or type == gr.GR_GAUSSIAN:
 115             self.connect (self.noisegen, self.usrp)
 116             self.noisegen.set_type (type)
 117             self.src = self.noisegen
 118         else:
 119             raise ValueError, type
 120         if was_running:
 121             self.start ()
 122
 123
 124 if __name__ == '__main__':
 125     parser = OptionParser (option_class=eng_option)
 126     parser.add_option ("--sine", dest="type", action="store_const", const=gr.GR_SIN_WAVE,
 127                        help="generate a complex sinusoid [default]", default=gr.GR_SIN_WAVE)
 128     parser.add_option ("--const", dest="type", action="store_const", const=gr.GR_CONST_WAVE,
 129                        help="generate a constant output")
 130     parser.add_option ("--gaussian", dest="type", action="store_const", const=gr.GR_GAUSSIAN,
 131                        help="generate Gaussian random output")
 132     parser.add_option ("--uniform", dest="type", action="store_const", const=gr.GR_UNIFORM,
 133                        help="generate Uniform random output")
 134     parser.add_option ("-f", "--freq", type="eng_float", default=100e3,
 135                        help="set waveform frequency to FREQ")
 136     parser.add_option ("-r", "--rf-freq", type="eng_float", default=910e6,
 137                        help="set waveform frequency to FREQ")
 138     parser.add_option ("-a", "--amplitude", type="eng_float", default=16e3,
 139                        help="set waveform amplitude to AMPLITUDE", metavar="AMPL")
 140     parser.add_option ("-o", "--offset", type="eng_float", default=0,
 141                        help="set waveform offset to OFFSET")
 142     parser.add_option ("-c", "--duc-freq", type="eng_float", default=0,
 143                        help="set Tx DUC frequency to FREQ", metavar="FREQ")
 144     parser.add_option ("-m", "--mux", type="intx", default=0x98,
 145                        help="set output mux register")
 146
 147     app = stdgui.stdapp (flex_siggen, "USRP FlexRF Siggen")
 148     app.MainLoop ()