adding limit support

This commit is contained in:
sg3510 2015-02-26 14:33:41 +00:00
parent 2c961cc43f
commit 3a9e3a4da7

View file

@ -44,9 +44,12 @@ def read(filename, limit=None):
except audioop.error:
fs, _, audiofile = wavio.readwav(filename)
if limit:
audiofile = audiofile[:limit * 1000]
audiofile = audiofile.T
audiofile = audiofile.astype(np.int16)
channels = []
for chn in audiofile:
channels.append(chn)