diff options
| author | Josh Morman <jmorman@gnuradio.org> | 2021-11-24 12:34:58 -0500 |
|---|---|---|
| committer | mormj <34754695+mormj@users.noreply.github.com> | 2021-11-24 14:41:53 -0500 |
| commit | 09cd2d7cccf0a472ee2623c4a388a4ce950c0c5c (patch) | |
| tree | 69655be1c0d244a68c1fe054969263d27ef99c07 /gr-filter/examples/reconstruction.py | |
| parent | b4aaa48f69239e16341a59628ca7795cfeebb4da (diff) | |
filter: pep8 formatting
Signed-off-by: Josh Morman <jmorman@gnuradio.org>
Diffstat (limited to 'gr-filter/examples/reconstruction.py')
| -rw-r--r-- | gr-filter/examples/reconstruction.py | 46 |
1 files changed, 22 insertions, 24 deletions
diff --git a/gr-filter/examples/reconstruction.py b/gr-filter/examples/reconstruction.py index 8b65f70deb..2a34f9e33f 100644 --- a/gr-filter/examples/reconstruction.py +++ b/gr-filter/examples/reconstruction.py @@ -29,11 +29,12 @@ except ImportError: fftlen = 8192 + def main(): N = 10000 fs = 2000.0 Ts = 1.0 / fs - t = numpy.arange(0, N*Ts, Ts) + t = numpy.arange(0, N * Ts, Ts) # When playing with the number of channels, be careful about the filter # specs and the channel map of the synthesizer set below. @@ -42,12 +43,11 @@ def main(): # Build the filter(s) bw = 1000 tb = 400 - proto_taps = filter.firdes.low_pass_2(1, nchans*fs, + proto_taps = filter.firdes.low_pass_2(1, nchans * fs, bw, tb, 80, window.WIN_BLACKMAN_hARRIS) print("Filter length: ", len(proto_taps)) - # Create a modulated signal npwr = 0.01 data = numpy.random.randint(0, 256, N) @@ -62,7 +62,7 @@ def main(): channelizer = filter.pfb.channelizer_ccf(nchans, proto_taps, 2) # Put the pieces back together again - syn_taps = [nchans*t for t in proto_taps] + syn_taps = [nchans * t for t in proto_taps] synthesizer = filter.pfb_synthesizer_ccf(nchans, syn_taps, True) src_snk = blocks.vector_sink_c() snk = blocks.vector_sink_c() @@ -70,7 +70,7 @@ def main(): # Remap the location of the channels # Can be done in synth or channelizer (watch out for rotattions in # the channelizer) - synthesizer.set_channel_map([ 0, 1, 2, 3, 4, + synthesizer.set_channel_map([0, 1, 2, 3, 4, 15, 16, 17, 18, 19]) tb = gr.top_block() @@ -79,34 +79,33 @@ def main(): vsnk = [] for i in range(nchans): - tb.connect((channelizer,i), (synthesizer, i)) + tb.connect((channelizer, i), (synthesizer, i)) vsnk.append(blocks.vector_sink_c()) - tb.connect((channelizer,i), vsnk[i]) + tb.connect((channelizer, i), vsnk[i]) tb.connect(synthesizer, snk) tb.run() - sin = numpy.array(src_snk.data()[1000:]) + sin = numpy.array(src_snk.data()[1000:]) sout = numpy.array(snk.data()[1000:]) - # Plot original signal - fs_in = nchans*fs - f1 = pyplot.figure(1, figsize=(16,12), facecolor='w') - s11 = f1.add_subplot(2,2,1) + fs_in = nchans * fs + f1 = pyplot.figure(1, figsize=(16, 12), facecolor='w') + s11 = f1.add_subplot(2, 2, 1) s11.psd(sin, NFFT=fftlen, Fs=fs_in) s11.set_title("PSD of Original Signal") s11.set_ylim([-200, -20]) - s12 = f1.add_subplot(2,2,2) + s12 = f1.add_subplot(2, 2, 2) s12.plot(sin.real[1000:1500], "o-b") s12.plot(sin.imag[1000:1500], "o-r") s12.set_title("Original Signal in Time") start = 1 - skip = 2 - s13 = f1.add_subplot(2,2,3) + skip = 2 + s13 = f1.add_subplot(2, 2, 3) s13.plot(sin.real[start::skip], sin.imag[start::skip], "o") s13.set_title("Constellation") s13.set_xlim([-2, 2]) @@ -116,29 +115,29 @@ def main(): nrows = int(numpy.sqrt(nchans)) ncols = int(numpy.ceil(float(nchans) / float(nrows))) - f2 = pyplot.figure(2, figsize=(16,12), facecolor='w') + f2 = pyplot.figure(2, figsize=(16, 12), facecolor='w') for n in range(nchans): - s = f2.add_subplot(nrows, ncols, n+1) + s = f2.add_subplot(nrows, ncols, n + 1) s.psd(vsnk[n].data(), NFFT=fftlen, Fs=fs_in) s.set_title("Channel {0}".format(n)) s.set_ylim([-200, -20]) # Plot reconstructed signal - fs_out = 2*nchans*fs - f3 = pyplot.figure(3, figsize=(16,12), facecolor='w') - s31 = f3.add_subplot(2,2,1) + fs_out = 2 * nchans * fs + f3 = pyplot.figure(3, figsize=(16, 12), facecolor='w') + s31 = f3.add_subplot(2, 2, 1) s31.psd(sout, NFFT=fftlen, Fs=fs_out) s31.set_title("PSD of Reconstructed Signal") s31.set_ylim([-200, -20]) - s32 = f3.add_subplot(2,2,2) + s32 = f3.add_subplot(2, 2, 2) s32.plot(sout.real[1000:1500], "o-b") s32.plot(sout.imag[1000:1500], "o-r") s32.set_title("Reconstructed Signal in Time") start = 0 - skip = 4 - s33 = f3.add_subplot(2,2,3) + skip = 4 + s33 = f3.add_subplot(2, 2, 3) s33.plot(sout.real[start::skip], sout.imag[start::skip], "o") s33.set_title("Constellation") s33.set_xlim([-2, 2]) @@ -152,4 +151,3 @@ if __name__ == "__main__": main() except KeyboardInterrupt: pass - |