mirror
/
HiFiScan
kopia lustrzana https://github.com/erdewit/HiFiScan
1
0
Forkuj 0
Kod Zgłoszenia Projekty Wydania Wiki Aktywność

Simplify/optimize read_wav

pull/14/head
Ewald de Wit 2022-09-24 09:54:58 +02:00
rodzic 9f257e7872
commit 34050fe2a1
1 zmienionych plików z 7 dodań i 5 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

12
hifiscan/io_.py
Wyświetl plik

@ -58,20 +58,22 @@ def read_wav(path: str) -> Sound:
frames = wav.readframes(n)
if width == 4:
buff = np.frombuffer(frames, np.int32)
data = buff.astype('f') / np.float32(2 ** 31 - 1)
norm = 2 ** 31 - 1
elif width == 3:
buff = np.frombuffer(frames, np.uint8)
uints = buff[0::3].astype(np.uint32) << 8 \
| buff[1::3].astype(np.uint32) << 16 \
| buff[2::3].astype(np.uint32) << 24
data = uints.view(np.int32).astype('f') / np.float32(
2 ** 31 - 2 ** 8)
buff = uints.view(np.int32)
norm = 2 ** 31 - 2 ** 8
elif width == 2:
buff = np.frombuffer(frames, np.int16)
data = buff.astype('f') / np.float32(2 ** 15 - 1)
norm = 2 ** 15 - 1
else:
buff = np.frombuffer(frames, np.int8)
data = buff.astype('f') / np.float32(2 ** 7 - 1)
norm = 2 ** 7 - 1
data = buff.astype('f')
data /= np.float32(norm)
data = data.reshape((-1, ch)).T
return Sound(data, rate, width)