haraldwolff
/
whisper.cpp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix reading of stereo WAV files

pull/19/head
Georgi Gerganov 2022-10-01 08:41:57 +03:00
parent 62897e8ae6
commit 6d654d192a
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
main.cpp
View File

@ -2140,20 +2140,22 @@ int main(int argc, char ** argv) {
return 5; return 5;
} }
int n = wav.totalPCMFrameCount;
std::vector<int16_t> pcm16; std::vector<int16_t> pcm16;
pcm16.resize(wav.totalPCMFrameCount); pcm16.resize(n*wav.channels);
drwav_read_pcm_frames_s16(&wav, wav.totalPCMFrameCount, pcm16.data()); drwav_read_pcm_frames_s16(&wav, n, pcm16.data());
drwav_uninit(&wav); drwav_uninit(&wav);
// convert to float // convert to mono, float
pcmf32.resize(pcm16.size()); pcmf32.resize(n);
if (wav.channels == 1) { if (wav.channels == 1) {
for (size_t i = 0; i < pcm16.size(); i++) { for (size_t i = 0; i < n; i++) {
pcmf32[i] = float(pcm16[i])/32768.0f; pcmf32[i] = float(pcm16[i])/32768.0f;
} }
} else { } else {
for (size_t i = 0; i < pcm16.size(); i++) { for (size_t i = 0; i < n; i++) {
pcmf32[i] = float(pcm16[i*2 + 0] + pcm16[i*2 + 1])/32768.0f/2.0f; pcmf32[i] = float(pcm16[2*i] + pcm16[2*i + 1])/65536.0f;
} }
} }
} }