From: Georgi Gerganov <redacted>
Date: Sat, 1 Oct 2022 05:41:57 +0000 (+0300)
Subject: Fix reading of stereo WAV files
X-Git-Tag: upstream/1.7.4~2008
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=6d654d192a62e6cd9897d6ff683bdc97406827e9;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp

Fix reading of stereo WAV files
---

diff --git a/main.cpp b/main.cpp
index 98b838f1..fb758e37 100644
--- a/main.cpp
+++ b/main.cpp
@@ -2140,20 +2140,22 @@ int main(int argc, char ** argv) {
             return 5;
         }
 
+        int n = wav.totalPCMFrameCount;
+
         std::vector<int16_t> pcm16;
-        pcm16.resize(wav.totalPCMFrameCount);
-        drwav_read_pcm_frames_s16(&wav, wav.totalPCMFrameCount, pcm16.data());
+        pcm16.resize(n*wav.channels);
+        drwav_read_pcm_frames_s16(&wav, n, pcm16.data());
         drwav_uninit(&wav);
 
-        // convert to float
-        pcmf32.resize(pcm16.size());
+        // convert to mono, float
+        pcmf32.resize(n);
         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;
             }
         } else {
-            for (size_t i = 0; i < pcm16.size(); i++) {
-                pcmf32[i] = float(pcm16[i*2 + 0] + pcm16[i*2 + 1])/32768.0f/2.0f;
+            for (size_t i = 0; i < n; i++) {
+                pcmf32[i] = float(pcm16[2*i] + pcm16[2*i + 1])/65536.0f;
             }
         }
     }