]> git.gag.com Git - debian/gnuradio/blobdiff - gnuradio-examples/python/ofdm/fftshift.py
Imported Upstream version 3.2.2
[debian/gnuradio] / gnuradio-examples / python / ofdm / fftshift.py
diff --git a/gnuradio-examples/python/ofdm/fftshift.py b/gnuradio-examples/python/ofdm/fftshift.py
new file mode 100755 (executable)
index 0000000..98abf5d
--- /dev/null
@@ -0,0 +1,44 @@
+#!/usr/bin/env python
+
+from gnuradio import gr
+
+class my_top_block(gr.top_block):
+    def __init__(self):
+        gr.top_block.__init__(self)
+
+        length = 101
+
+        data_r = range(length)
+        data_i = range(length,2*length)
+        src_r = gr.vector_source_s(data_r, False)
+        src_i = gr.vector_source_s(data_i, False)
+        s2f_r = gr.short_to_float()
+        s2f_i = gr.short_to_float()
+        f2c = gr.float_to_complex()
+        s2v = gr.stream_to_vector(gr.sizeof_gr_complex, length)
+
+        shift = True
+        ifft = gr.fft_vcc(length, False, [], shift)
+        fft  = gr.fft_vcc(length, True, [], shift)
+        
+        v2s = gr.vector_to_stream(gr.sizeof_gr_complex, length)
+        snk_in = gr.file_sink(gr.sizeof_gr_complex, "fftshift.in")
+        snk_out = gr.file_sink(gr.sizeof_gr_complex, "fftshift.out")
+
+        self.connect(src_r, s2f_r, (f2c,0))
+        self.connect(src_i, s2f_i, (f2c,1))
+        self.connect(f2c, snk_in)
+        self.connect(f2c, s2v, ifft, fft, v2s, snk_out)
+        
+
+def main():
+    tb = my_top_block()
+    tb.start()
+    tb.wait()
+
+if __name__ == "__main__":
+    try:
+        main()
+    except KeyboardInterrupt:
+        pass
+