You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
252 lines
9.4 KiB
252 lines
9.4 KiB
/*
|
|
* Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
// MSVC++ requires this to be set before any other includes to get M_PI.
|
|
#define _USE_MATH_DEFINES
|
|
|
|
#include "common_audio/wav_file.h"
|
|
|
|
#include <cmath>
|
|
#include <limits>
|
|
|
|
#include "common_audio/wav_header.h"
|
|
#include "test/gtest.h"
|
|
#include "test/testsupport/file_utils.h"
|
|
|
|
// WavWriterTest.CPP flaky on Mac. See webrtc:9247.
|
|
#if defined(WEBRTC_MAC)
|
|
#define MAYBE_CPP DISABLED_CPP
|
|
#define MAYBE_CPPReset DISABLED_CPPReset
|
|
#else
|
|
#define MAYBE_CPP CPP
|
|
#define MAYBE_CPPReset CPPReset
|
|
#endif
|
|
|
|
namespace webrtc {
|
|
|
|
static const float kSamples[] = {0.0, 10.0, 4e4, -1e9};
|
|
|
|
// Write a tiny WAV file with the C++ interface and verify the result.
|
|
TEST(WavWriterTest, MAYBE_CPP) {
|
|
const std::string outfile = test::OutputPath() + "wavtest1.wav";
|
|
static const size_t kNumSamples = 3;
|
|
{
|
|
WavWriter w(outfile, 14099, 1);
|
|
EXPECT_EQ(14099, w.sample_rate());
|
|
EXPECT_EQ(1u, w.num_channels());
|
|
EXPECT_EQ(0u, w.num_samples());
|
|
w.WriteSamples(kSamples, kNumSamples);
|
|
EXPECT_EQ(kNumSamples, w.num_samples());
|
|
}
|
|
// Write some extra "metadata" to the file that should be silently ignored
|
|
// by WavReader. We don't use WavWriter directly for this because it doesn't
|
|
// support metadata.
|
|
static const uint8_t kMetadata[] = {101, 202};
|
|
{
|
|
FILE* f = fopen(outfile.c_str(), "ab");
|
|
ASSERT_TRUE(f);
|
|
ASSERT_EQ(1u, fwrite(kMetadata, sizeof(kMetadata), 1, f));
|
|
fclose(f);
|
|
}
|
|
static const uint8_t kExpectedContents[] = {
|
|
// clang-format off
|
|
// clang formatting doesn't respect inline comments.
|
|
'R', 'I', 'F', 'F',
|
|
42, 0, 0, 0, // size of whole file - 8: 6 + 44 - 8
|
|
'W', 'A', 'V', 'E',
|
|
'f', 'm', 't', ' ',
|
|
16, 0, 0, 0, // size of fmt block - 8: 24 - 8
|
|
1, 0, // format: PCM (1)
|
|
1, 0, // channels: 1
|
|
0x13, 0x37, 0, 0, // sample rate: 14099
|
|
0x26, 0x6e, 0, 0, // byte rate: 2 * 14099
|
|
2, 0, // block align: NumChannels * BytesPerSample
|
|
16, 0, // bits per sample: 2 * 8
|
|
'd', 'a', 't', 'a',
|
|
6, 0, 0, 0, // size of payload: 6
|
|
0, 0, // first sample: 0.0
|
|
10, 0, // second sample: 10.0
|
|
0xff, 0x7f, // third sample: 4e4 (saturated)
|
|
kMetadata[0], kMetadata[1],
|
|
// clang-format on
|
|
};
|
|
static const size_t kContentSize =
|
|
kPcmWavHeaderSize + kNumSamples * sizeof(int16_t) + sizeof(kMetadata);
|
|
static_assert(sizeof(kExpectedContents) == kContentSize, "content size");
|
|
EXPECT_EQ(kContentSize, test::GetFileSize(outfile));
|
|
FILE* f = fopen(outfile.c_str(), "rb");
|
|
ASSERT_TRUE(f);
|
|
uint8_t contents[kContentSize];
|
|
ASSERT_EQ(1u, fread(contents, kContentSize, 1, f));
|
|
EXPECT_EQ(0, fclose(f));
|
|
EXPECT_EQ(0, memcmp(kExpectedContents, contents, kContentSize));
|
|
|
|
{
|
|
WavReader r(outfile);
|
|
EXPECT_EQ(14099, r.sample_rate());
|
|
EXPECT_EQ(1u, r.num_channels());
|
|
EXPECT_EQ(kNumSamples, r.num_samples());
|
|
static const float kTruncatedSamples[] = {0.0, 10.0, 32767.0};
|
|
float samples[kNumSamples];
|
|
EXPECT_EQ(kNumSamples, r.ReadSamples(kNumSamples, samples));
|
|
EXPECT_EQ(0, memcmp(kTruncatedSamples, samples, sizeof(samples)));
|
|
EXPECT_EQ(0u, r.ReadSamples(kNumSamples, samples));
|
|
}
|
|
}
|
|
|
|
// Write a larger WAV file. You can listen to this file to sanity-check it.
|
|
TEST(WavWriterTest, LargeFile) {
|
|
constexpr int kSampleRate = 8000;
|
|
constexpr size_t kNumChannels = 2;
|
|
constexpr size_t kNumSamples = 3 * kSampleRate * kNumChannels;
|
|
for (WavFile::SampleFormat wav_format :
|
|
{WavFile::SampleFormat::kInt16, WavFile::SampleFormat::kFloat}) {
|
|
for (WavFile::SampleFormat write_format :
|
|
{WavFile::SampleFormat::kInt16, WavFile::SampleFormat::kFloat}) {
|
|
for (WavFile::SampleFormat read_format :
|
|
{WavFile::SampleFormat::kInt16, WavFile::SampleFormat::kFloat}) {
|
|
std::string outfile = test::OutputPath() + "wavtest3.wav";
|
|
float samples[kNumSamples];
|
|
for (size_t i = 0; i < kNumSamples; i += kNumChannels) {
|
|
// A nice periodic beeping sound.
|
|
static const double kToneHz = 440;
|
|
const double t =
|
|
static_cast<double>(i) / (kNumChannels * kSampleRate);
|
|
const double x = std::numeric_limits<int16_t>::max() *
|
|
std::sin(t * kToneHz * 2 * M_PI);
|
|
samples[i] = std::pow(std::sin(t * 2 * 2 * M_PI), 10) * x;
|
|
samples[i + 1] = std::pow(std::cos(t * 2 * 2 * M_PI), 10) * x;
|
|
}
|
|
{
|
|
WavWriter w(outfile, kSampleRate, kNumChannels, wav_format);
|
|
EXPECT_EQ(kSampleRate, w.sample_rate());
|
|
EXPECT_EQ(kNumChannels, w.num_channels());
|
|
EXPECT_EQ(0u, w.num_samples());
|
|
if (write_format == WavFile::SampleFormat::kFloat) {
|
|
float truncated_samples[kNumSamples];
|
|
for (size_t k = 0; k < kNumSamples; ++k) {
|
|
truncated_samples[k] = static_cast<int16_t>(samples[k]);
|
|
}
|
|
w.WriteSamples(truncated_samples, kNumSamples);
|
|
} else {
|
|
w.WriteSamples(samples, kNumSamples);
|
|
}
|
|
EXPECT_EQ(kNumSamples, w.num_samples());
|
|
}
|
|
if (wav_format == WavFile::SampleFormat::kFloat) {
|
|
EXPECT_EQ(sizeof(float) * kNumSamples + kIeeeFloatWavHeaderSize,
|
|
test::GetFileSize(outfile));
|
|
} else {
|
|
EXPECT_EQ(sizeof(int16_t) * kNumSamples + kPcmWavHeaderSize,
|
|
test::GetFileSize(outfile));
|
|
}
|
|
|
|
{
|
|
WavReader r(outfile);
|
|
EXPECT_EQ(kSampleRate, r.sample_rate());
|
|
EXPECT_EQ(kNumChannels, r.num_channels());
|
|
EXPECT_EQ(kNumSamples, r.num_samples());
|
|
|
|
if (read_format == WavFile::SampleFormat::kFloat) {
|
|
float read_samples[kNumSamples];
|
|
EXPECT_EQ(kNumSamples, r.ReadSamples(kNumSamples, read_samples));
|
|
for (size_t i = 0; i < kNumSamples; ++i) {
|
|
EXPECT_NEAR(samples[i], read_samples[i], 1);
|
|
}
|
|
EXPECT_EQ(0u, r.ReadSamples(kNumSamples, read_samples));
|
|
} else {
|
|
int16_t read_samples[kNumSamples];
|
|
EXPECT_EQ(kNumSamples, r.ReadSamples(kNumSamples, read_samples));
|
|
for (size_t i = 0; i < kNumSamples; ++i) {
|
|
EXPECT_NEAR(samples[i], static_cast<float>(read_samples[i]), 1);
|
|
}
|
|
EXPECT_EQ(0u, r.ReadSamples(kNumSamples, read_samples));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Write a tiny WAV file with the C++ interface then read-reset-read.
|
|
TEST(WavReaderTest, MAYBE_CPPReset) {
|
|
const std::string outfile = test::OutputPath() + "wavtest4.wav";
|
|
static const size_t kNumSamples = 3;
|
|
{
|
|
WavWriter w(outfile, 14099, 1);
|
|
EXPECT_EQ(14099, w.sample_rate());
|
|
EXPECT_EQ(1u, w.num_channels());
|
|
EXPECT_EQ(0u, w.num_samples());
|
|
w.WriteSamples(kSamples, kNumSamples);
|
|
EXPECT_EQ(kNumSamples, w.num_samples());
|
|
}
|
|
// Write some extra "metadata" to the file that should be silently ignored
|
|
// by WavReader. We don't use WavWriter directly for this because it doesn't
|
|
// support metadata.
|
|
static const uint8_t kMetadata[] = {101, 202};
|
|
{
|
|
FILE* f = fopen(outfile.c_str(), "ab");
|
|
ASSERT_TRUE(f);
|
|
ASSERT_EQ(1u, fwrite(kMetadata, sizeof(kMetadata), 1, f));
|
|
fclose(f);
|
|
}
|
|
static const uint8_t kExpectedContents[] = {
|
|
// clang-format off
|
|
// clang formatting doesn't respect inline comments.
|
|
'R', 'I', 'F', 'F',
|
|
42, 0, 0, 0, // size of whole file - 8: 6 + 44 - 8
|
|
'W', 'A', 'V', 'E',
|
|
'f', 'm', 't', ' ',
|
|
16, 0, 0, 0, // size of fmt block - 8: 24 - 8
|
|
1, 0, // format: PCM (1)
|
|
1, 0, // channels: 1
|
|
0x13, 0x37, 0, 0, // sample rate: 14099
|
|
0x26, 0x6e, 0, 0, // byte rate: 2 * 14099
|
|
2, 0, // block align: NumChannels * BytesPerSample
|
|
16, 0, // bits per sample: 2 * 8
|
|
'd', 'a', 't', 'a',
|
|
6, 0, 0, 0, // size of payload: 6
|
|
0, 0, // first sample: 0.0
|
|
10, 0, // second sample: 10.0
|
|
0xff, 0x7f, // third sample: 4e4 (saturated)
|
|
kMetadata[0], kMetadata[1],
|
|
// clang-format on
|
|
};
|
|
static const size_t kContentSize =
|
|
kPcmWavHeaderSize + kNumSamples * sizeof(int16_t) + sizeof(kMetadata);
|
|
static_assert(sizeof(kExpectedContents) == kContentSize, "content size");
|
|
EXPECT_EQ(kContentSize, test::GetFileSize(outfile));
|
|
FILE* f = fopen(outfile.c_str(), "rb");
|
|
ASSERT_TRUE(f);
|
|
uint8_t contents[kContentSize];
|
|
ASSERT_EQ(1u, fread(contents, kContentSize, 1, f));
|
|
EXPECT_EQ(0, fclose(f));
|
|
EXPECT_EQ(0, memcmp(kExpectedContents, contents, kContentSize));
|
|
|
|
{
|
|
WavReader r(outfile);
|
|
EXPECT_EQ(14099, r.sample_rate());
|
|
EXPECT_EQ(1u, r.num_channels());
|
|
EXPECT_EQ(kNumSamples, r.num_samples());
|
|
static const float kTruncatedSamples[] = {0.0, 10.0, 32767.0};
|
|
float samples[kNumSamples];
|
|
EXPECT_EQ(kNumSamples, r.ReadSamples(kNumSamples, samples));
|
|
EXPECT_EQ(0, memcmp(kTruncatedSamples, samples, sizeof(samples)));
|
|
EXPECT_EQ(0u, r.ReadSamples(kNumSamples, samples));
|
|
|
|
r.Reset();
|
|
EXPECT_EQ(kNumSamples, r.ReadSamples(kNumSamples, samples));
|
|
EXPECT_EQ(0, memcmp(kTruncatedSamples, samples, sizeof(samples)));
|
|
EXPECT_EQ(0u, r.ReadSamples(kNumSamples, samples));
|
|
}
|
|
}
|
|
|
|
} // namespace webrtc
|