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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
/* SPDX-License-Identifier: GPL-2.0-or-later */
/*!********************************************************************
Audacity: A Digital Audio Editor
WavFileIO.cpp
Matthieu Hodgkinson
**********************************************************************/
#include "WavFileIO.h"
#include "AudioFileInfo.h"
#include <cassert>
#include <iostream>
#include <sndfile.h>
using namespace std::literals::string_literals;
bool WavFileIO::Read(
const std::string& inputPath, std::vector<std::vector<float>>& audio,
AudioFileInfo& info, const std::optional<std::chrono::seconds>& upTo)
{
SF_INFO sfInfo;
auto sndfile = sf_open(inputPath.c_str(), SFM_READ, &sfInfo);
if (!sndfile)
return false;
const auto numFramesToRead =
upTo.has_value() ?
std::min<int>(
sfInfo.frames,
static_cast<int>(upTo->count() * sfInfo.samplerate)) :
sfInfo.frames;
std::vector<float> tmp(numFramesToRead * sfInfo.channels);
const auto numReadFrames =
sf_readf_float(sndfile, tmp.data(), numFramesToRead);
sf_close(sndfile);
assert(numReadFrames == numFramesToRead);
audio.resize(sfInfo.channels);
for (auto i = 0; i < sfInfo.channels; ++i)
{
audio[i].resize(numFramesToRead);
for (auto ii = 0; ii < numFramesToRead; ++ii)
{
audio[i][ii] = tmp[ii * sfInfo.channels + i];
}
}
info.sampleRate = sfInfo.samplerate;
info.numChannels = sfInfo.channels;
info.numFrames = numFramesToRead;
return true;
}
bool WavFileIO::Write(
const std::string& outputPath, const std::vector<std::vector<float>>& audio,
int sampleRate)
{
const auto numChannels = audio.size();
const auto numFrames = audio[0].size();
SF_INFO sfInfo;
sfInfo.channels = numChannels;
sfInfo.frames = numFrames;
sfInfo.samplerate = sampleRate;
sfInfo.format = SF_FORMAT_WAV | SF_FORMAT_PCM_16;
sfInfo.sections = 1;
sfInfo.seekable = 1;
auto sndfile = sf_open(outputPath.c_str(), SFM_WRITE, &sfInfo);
if (!sndfile)
{
std::cout << "libsndfile could not open "s + outputPath + " for write."
<< std::endl;
return false;
}
std::vector<float> interleaved(numChannels * numFrames);
for (auto i = 0u; i < numChannels; ++i)
{
for (auto ii = 0u; ii < numFrames; ++ii)
{
interleaved[ii * numChannels + i] = audio[i][ii];
}
}
const auto numFramesWritten =
sf_writef_float(sndfile, interleaved.data(), numFrames);
assert(numFramesWritten == numFrames);
sf_close(sndfile);
return true;
}
|
