Changeset 01f7598 for python/demos/demo_filter.py

Timestamp:

May 10, 2016, 6:50:36 PM (9 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch, yinfft+

Children:

a4575c4

Parents:

73d38d5

Message:

python/demos/demo_filter.py: clean-up

File:

• python/demos/demo_filter.py (modified) (1 diff)

python/demos/demo_filter.py

@@ -2 +2 @@
 
 
-def apply_filter(path, params = {}):
+def apply_filter(path):
     from aubio import source, sink, digital_filter
-    from os.path import basename, splitex, splitextt
+    from os.path import basename, splitext
+
+    # open input file, get its samplerate
     s = source(path)
+    samplerate = s.samplerate
+
+    # create an A-weighting filter
     f = digital_filter(7)
-    f.set_a_weighting(s.samplerate)
-    #f = digital_filter(3)
-    #f.set_biquad(...)
-    o = sink("filtered_" + splitext(basename(path))[0] + ".wav")
-    # Total number of frames read
+    f.set_a_weighting(samplerate)
+    # alternatively, apply another filter
+
+    # create output file
+    o = sink("filtered_" + splitext(basename(path))[0] + ".wav", samplerate)
+
     total_frames = 0
-
     while True:
         samples, read = s()
         filtered_samples = f(samples)
-        o(samples, read)
+        o(filtered_samples, read)
         total_frames += read
         if read < s.hop_size: break
-    print "filtered", s.uri, "to", o.uri, "using an A-weighting filter"
+
+    duration = total_frames / float(samplerate)
+    print ("read {:s}".format(s.uri))
+    print ("applied A-weighting filtered ({:d} Hz)".format(samplerate))
+    print ("wrote {:s} ({:.2f} s)".format(o.uri, duration))
 
 if __name__ == '__main__':