Merged r4767:4859 from developer branch jcorgan/channel, passes distcheck.

[debian/gnuradio] / gnuradio-examples / python / hier / sounder / usrp_sounder_tx.py

diff --git a/gnuradio-examples/python/hier/sounder/usrp_sounder_tx.py b/gnuradio-examples/python/hier/sounder/usrp_sounder_tx.py
index c96313c67add7c00b4c656a073326ae2a5ccb419..00146146ce93d5bba68d23979520b81262218655 100755 (executable)
--- a/gnuradio-examples/python/hier/sounder/usrp_sounder_tx.py
+++ b/gnuradio-examples/python/hier/sounder/usrp_sounder_tx.py
@@ -23,8 +23,49 @@
 from gnuradio import gr, eng_notation
 from gnuradio.eng_option import eng_option
 from optparse import OptionParser
+from usrp_sink import usrp_sink_c
 from sounder_tx import sounder_tx
 
+n2s = eng_notation.num_to_str
+
+class usrp_sounder_tx(gr.hier_block2):
+    def __init__(self, subdev_spec, freq, cal, verbose, degree, chip_rate, amplitude):
+
+        # Call hierarchical block constructor
+        # Top-level blocks have no inputs or outputs
+        gr.hier_block2.__init__(self,
+                                "usrp_sounder_tx",      # Block typename
+                                gr.io_signature(0,0,0), # Input signature
+                                gr.io_signature(0,0,0)) # Output signature
+
+        self._freq = freq
+        self._cal = cal
+        self._verbose = verbose
+        self._degree = degree
+        self._length = 2**degree-1
+       self._amplitude = amplitude
+       
+        self._u = usrp_sink_c(0, subdev_spec, chip_rate, self._freq, self._cal, self._verbose)
+       self.define_component("usrp", self._u)
+        self._chip_rate = self._u._if_rate
+        self._max_time = float(self._length)/self._chip_rate
+       self.define_component("pn", sounder_tx(self._degree, self._chip_rate, self._verbose))
+        self.define_component("gain", gr.multiply_const_ff(amplitude));
+        self.define_component("f2c", gr.float_to_complex())
+        
+        if self._verbose:
+            print "Chip rate is", n2s(self._chip_rate), "chips/sec"
+            print "Resolution is", n2s(1.0/self._chip_rate), "sec"
+            print "Using PN code of degree", self._degree, "length", 2**self._degree-1
+            print "Maximum measurable impulse response is", n2s(self._max_time), "sec"
+            print "Output amplitude is", amplitude
+
+       # Ultimately this will be
+       # self.connect("pn gain f2c usrp")
+       self.connect("pn", 0, "gain", 0)
+        self.connect("gain", 0, "f2c", 0)
+        self.connect("f2c", 0, "usrp", 0)
+                              
 def main():
        parser = OptionParser(option_class=eng_option)
 
@@ -37,22 +78,35 @@ def main():
                           help="set frequency calibration offset (default=%default)")
        parser.add_option("-v", "--verbose", action="store_true", default=False,
                          help="print extra debugging info")
-       parser.add_option("-d", "--max-delay", type="eng_float", default=10e-6,
-                         help="set maximum delay spread (default=%default)")
+       parser.add_option("-d", "--degree", type="int", default=10,
+                         help="set PN code degree (length=2**degree-1, default=%default)")
        parser.add_option("-r", "--chip-rate", type="eng_float", default=8e6,
                          help="set sounder chip rate (default=%default)")
-       parser.add_option("-g", "--amplitude", type="eng_float", default=32000.0,
+       parser.add_option("-g", "--amplitude", type="eng_float", default=8000.0,
                          help="set output amplitude (default=%default)")
+        parser.add_option("", "--real-time", action="store_true", default=False,
+                          help="Attempt to enable real-time scheduling")
         (options, args) = parser.parse_args()
 
        if len(args) != 0:
             parser.print_help()
             sys.exit(1)
 
+        if not options.real_time:
+            realtime = False
+        else:
+            # Attempt to enable realtime scheduling
+            r = gr.enable_realtime_scheduling()
+            if r == gr.RT_OK:
+                realtime = True
+            else:
+                realtime = False
+                print "Note: failed to enable realtime scheduling"
+
        # Create an instance of a hierarchical block
-       top_block = sounder_tx(options.tx_subdev_spec, options.freq, options.cal,
-                              options.verbose, options.max_delay, options.chip_rate,
-                              options.amplitude)
+       top_block = usrp_sounder_tx(options.tx_subdev_spec, options.freq, options.cal,
+                                   options.verbose, options.degree, options.chip_rate,
+                                   options.amplitude)
                              
        # Create an instance of a runtime, passing it the top block
        # to process