File: Mp3FileReader.cpp

package info (click to toggle)
audacity 3.7.7%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 134,800 kB
  • sloc: cpp: 366,277; ansic: 198,323; lisp: 7,761; sh: 3,414; python: 1,501; xml: 1,385; perl: 854; makefile: 125
file content (73 lines) | stat: -rw-r--r-- 2,169 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/*  SPDX-License-Identifier: GPL-2.0-or-later */
/*!********************************************************************

  Audacity: A Digital Audio Editor

  Mp3FileReader.cpp

  Matthieu Hodgkinson

**********************************************************************/
#include "Mp3FileReader.h"
#include "AudioFileInfo.h"

#include <mpg123.h>

bool Mp3FileReader::Read(
   const std::string& path, std::vector<std::vector<float>>& floats,
   AudioFileInfo& info)
{
   int error = MPG123_OK;
   mpg123_handle* handle = mpg123_new(nullptr, &error);
   if (handle == nullptr)
      return false;

   if (mpg123_open(handle, path.c_str()) != MPG123_OK)
      return false;

   mpg123_param(handle, MPG123_FLAGS, MPG123_GAPLESS | MPG123_FORCE_FLOAT, 0.0);
   if (mpg123_scan(handle) != MPG123_OK)
      return false;

   if (
      mpg123_decode_frame(handle, nullptr, nullptr, nullptr) !=
      MPG123_NEW_FORMAT)
      return false;

   long long framesCount = mpg123_framelength(handle);
   long rate;
   int channels;
   int encoding = MPG123_ENC_FLOAT_32;
   mpg123_getformat(handle, &rate, &channels, &encoding);
   const auto numChannels = channels == MPG123_MONO ? 1 : 2;
   if (encoding != MPG123_ENC_FLOAT_32)
      return false;

   floats.resize(numChannels);
   off_t frameIndex { 0 };
   unsigned char* data { nullptr };
   size_t dataSize { 0 };
   std::vector<float> conversionBuffer;
   int ret = MPG123_OK;
   while ((ret = mpg123_decode_frame(handle, &frameIndex, &data, &dataSize)) ==
          MPG123_OK)
      for (auto channelIndex = 0; channelIndex < numChannels; ++channelIndex)
         for (auto frameIndex = 0;
              frameIndex < dataSize / numChannels / sizeof(float); ++frameIndex)
         {
            const auto floatIndex = channelIndex + frameIndex * numChannels;
            const auto value =
               *reinterpret_cast<float*>(data + floatIndex * sizeof(float));
            floats[channelIndex].push_back(value);
         }

   mpg123_close(handle);
   mpg123_delete(handle);
   mpg123_exit();

   info.sampleRate = rate;
   info.numChannels = numChannels;
   info.numFrames = floats[0].size();

   return true;
}