[audio] Update read_wav function: just return the

first channel from multi-channel audio files.

1 files changed, 7 insertions, 2 deletions

diff --git a/simpl/audio.py b/simpl/audio.py
index d86c5ff..d79f53d 100644
--- a/simpl/audio.py
+++ b/simpl/audio.py
@@ -5,5 +5,10 @@ import simpl
 
 def read_wav(file):
     'return floating point values between -1 and 1'
-    audio = wav.read(file)
-    return np.asarray(audio[1], dtype=simpl.dtype) / 32768.0, audio[0]
+    sampling_rate, audio = wav.read(file)
+
+    # if wav file has more than 1 channel, just take the first one
+    if audio.ndim > 1:
+        audio = audio.T[0]
+
+    return np.asarray(audio, dtype=simpl.dtype) / 32768.0, sampling_rate