From 09cd2d7cccf0a472ee2623c4a388a4ce950c0c5c Mon Sep 17 00:00:00 2001
From: Josh Morman <jmorman@gnuradio.org>
Date: Wed, 24 Nov 2021 12:34:58 -0500
Subject: filter: pep8 formatting
Signed-off-by: Josh Morman <jmorman@gnuradio.org>
---
gr-filter/examples/resampler.py | 32 +++++++++++++++++---------------
1 file changed, 17 insertions(+), 15 deletions(-)
(limited to 'gr-filter/examples/resampler.py')
diff --git a/gr-filter/examples/resampler.py b/gr-filter/examples/resampler.py
index 0208bac188..282eaa1286 100644
--- a/gr-filter/examples/resampler.py
+++ b/gr-filter/examples/resampler.py
@@ -23,15 +23,17 @@ except ImportError:
try:
from matplotlib import pyplot
except ImportError:
- sys.stderr.write("Error: Program requires matplotlib (see: matplotlib.sourceforge.net).\n")
+ sys.stderr.write(
+ "Error: Program requires matplotlib (see: matplotlib.sourceforge.net).\n")
sys.exit(1)
+
class mytb(gr.top_block):
def __init__(self, fs_in, fs_out, fc, N=10000):
gr.top_block.__init__(self)
rerate = float(fs_out) / float(fs_in)
- print("Resampling from %f to %f by %f " %(fs_in, fs_out, rerate))
+ print("Resampling from %f to %f by %f " % (fs_in, fs_out, rerate))
# Creating our own taps
taps = filter.firdes.low_pass_2(32, 32, 0.25, 0.1, 80)
@@ -59,6 +61,7 @@ class mytb(gr.top_block):
self.connect(self.head, self.resamp_0, self.snk_0)
self.connect(self.head, self.resamp_1, self.snk_1)
+
def main():
fs_in = 8000
fs_out = 20000
@@ -68,22 +71,21 @@ def main():
tb = mytb(fs_in, fs_out, fc, N)
tb.run()
-
# Plot PSD of signals
nfftsize = 2048
- fig1 = pyplot.figure(1, figsize=(10,10), facecolor="w")
- sp1 = fig1.add_subplot(2,1,1)
+ fig1 = pyplot.figure(1, figsize=(10, 10), facecolor="w")
+ sp1 = fig1.add_subplot(2, 1, 1)
sp1.psd(tb.snk_in.data(), NFFT=nfftsize,
- noverlap=nfftsize / 4, Fs = fs_in)
+ noverlap=nfftsize / 4, Fs=fs_in)
sp1.set_title(("Input Signal at f_s=%.2f kHz" % (fs_in / 1000.0)))
sp1.set_xlim([-fs_in / 2, fs_in / 2])
- sp2 = fig1.add_subplot(2,1,2)
+ sp2 = fig1.add_subplot(2, 1, 2)
sp2.psd(tb.snk_0.data(), NFFT=nfftsize,
- noverlap=nfftsize / 4, Fs = fs_out,
+ noverlap=nfftsize / 4, Fs=fs_out,
label="With our filter")
sp2.psd(tb.snk_1.data(), NFFT=nfftsize,
- noverlap=nfftsize / 4, Fs = fs_out,
+ noverlap=nfftsize / 4, Fs=fs_out,
label="With auto-generated filter")
sp2.set_title(("Output Signals at f_s=%.2f kHz" % (fs_out / 1000.0)))
sp2.set_xlim([-fs_out / 2, fs_out / 2])
@@ -92,16 +94,16 @@ def main():
# Plot signals in time
Ts_in = 1.0 / fs_in
Ts_out = 1.0 / fs_out
- t_in = numpy.arange(0, len(tb.snk_in.data())*Ts_in, Ts_in)
- t_out = numpy.arange(0, len(tb.snk_0.data())*Ts_out, Ts_out)
+ t_in = numpy.arange(0, len(tb.snk_in.data()) * Ts_in, Ts_in)
+ t_out = numpy.arange(0, len(tb.snk_0.data()) * Ts_out, Ts_out)
- fig2 = pyplot.figure(2, figsize=(10,10), facecolor="w")
- sp21 = fig2.add_subplot(2,1,1)
+ fig2 = pyplot.figure(2, figsize=(10, 10), facecolor="w")
+ sp21 = fig2.add_subplot(2, 1, 1)
sp21.plot(t_in, tb.snk_in.data())
sp21.set_title(("Input Signal at f_s=%.2f kHz" % (fs_in / 1000.0)))
sp21.set_xlim([t_in[100], t_in[200]])
- sp22 = fig2.add_subplot(2,1,2)
+ sp22 = fig2.add_subplot(2, 1, 2)
sp22.plot(t_out, tb.snk_0.data(),
label="With our filter")
sp22.plot(t_out, tb.snk_1.data(),
@@ -113,6 +115,6 @@ def main():
pyplot.show()
+
if __name__ == "__main__":
main()
-
--
cgit v1.2.3