Changeset 1e4d90f for python/demos

Timestamp:

May 10, 2016, 10:08:06 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:

0c6e3b0

Parents:

8fb567c

Message:

python/demos/demo_reading_speed.py: disable other packages by default

File:

• python/demos/demo_reading_speed.py (modified) (10 diffs)

python/demos/demo_reading_speed.py

@@ -6 +6 @@
 Compare the speed of several methods for reading and loading a sound file.
 
-This file depends on the following packages:
+Optionally, this file can make use of the following packages:
 
     - audioread     https://github.com/beetbox/audioread
+    - scipy         https://scipy.org
     - librosa       https://github.com/bmcfee/librosa
     - pydub         https://github.com/jiaaro/pydub
 
+Uncomment the function names below and send us your speed results!
+
 """
 
+
+test_functions = [
+            "read_file_aubio",
+            "load_file_aubio",
+            #"load_file_scipy",
+            #"load_file_scipy_mmap",
+            #"read_file_audioread",
+            #"load_file_librosa",
+            #"read_file_pydub",
+            #"load_file_pydub",
+            ]
+
+
 import numpy as np
-import aubio
-"""
-import audioread
-import librosa
-import scipy.io.wavfile
-from pydub import AudioSegment
-"""
 
 def read_file_audioread(filename):
+    import audioread
     # taken from librosa.util.utils
     def convert_buffer_to_float(buf, n_bytes = 2, dtype = np.float32):
@@ -43 +53 @@
 
 def load_file_librosa(filename):
+    import librosa
     y, sr = librosa.load(filename, sr = None)
     #print y.mean(), y.shape
@@ -48 +59 @@
 
 def load_file_scipy(filename):
+    import scipy.io.wavfile
     sr, y = scipy.io.wavfile.read(filename)
     y = y.astype('float32') / 32767
@@ -54 +66 @@
 
 def load_file_scipy_mmap(filename):
+    import scipy.io.wavfile
     sr, y = scipy.io.wavfile.read(filename, mmap = True)
     #print y.mean(), y.shape
@@ -59 +72 @@
 
 def read_file_pydub(filename):
+    from pydub import AudioSegment
     song = AudioSegment.from_file(filename)
     song.get_array_of_samples()
@@ -64 +78 @@
 
 def load_file_pydub(filename):
+    from pydub import AudioSegment
     song = AudioSegment.from_file(filename)
     y = np.asarray(song.get_array_of_samples(), dtype = 'float32')
@@ -70 +85 @@
 
 def read_file_aubio(filename):
+    import aubio
     f = aubio.source(filename, hop_size = 1024)
     total_frames = 0
@@ -79 +95 @@
 
 def load_file_aubio(filename):
+    import aubio
     f = aubio.source(filename, hop_size = 1024)
     y = np.zeros(f.duration, dtype = aubio.float_type)
@@ -95 +112 @@
     for i in range(10):
         start = time.time()
-        total_frames, samplerate = function(filename)
+        try:
+            total_frames, samplerate = function(filename)
+        except ImportError as e:
+            print ("error: failed importing {:s}".format(e))
+            return
         elapsed = time.time() - start
         #print ("{:5f} ".format(elapsed)),
         times.append(elapsed)
+
     #print
     times = np.array(times)
@@ -112 +134 @@
     filename = sys.argv[1]
 
-    functions = [
-            read_file_aubio,
-            load_file_aubio,
-            #load_file_scipy,
-            #load_file_scipy_mmap,
-            #read_file_audioread,
-            #load_file_librosa,
-            #read_file_pydub,
-            #load_file_pydub,
-            ]
-
-    for f in functions:
-        test_speed(f, filename)
+    for f in test_functions:
+        test_speed(eval(f), filename)