blob: 30f1d8e0c3f4bda0bf694ca0665064710a4e2f4b [file] [log] [blame] [edit]
/*
* Copyright 2018 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.
*/
#include "modules/audio_processing/include/audio_frame_view.h"
#include <array>
#include "common_audio/channel_buffer.h"
#include "modules/audio_processing/audio_buffer.h"
#include "rtc_base/arraysize.h"
#include "test/gtest.h"
namespace webrtc {
TEST(AudioFrameTest, ConstructFromAudioBuffer) {
constexpr int kSampleRateHz = 48000;
constexpr int kNumChannels = 2;
constexpr float kFloatConstant = 1272.f;
constexpr float kIntConstant = 17252;
const StreamConfig stream_config(kSampleRateHz, kNumChannels);
AudioBuffer buffer(
stream_config.sample_rate_hz(), stream_config.num_channels(),
stream_config.sample_rate_hz(), stream_config.num_channels(),
stream_config.sample_rate_hz(), stream_config.num_channels());
AudioFrameView<float> non_const_view(buffer.channels(), buffer.num_channels(),
buffer.num_frames());
// Modification is allowed.
non_const_view.channel(0)[0] = kFloatConstant;
EXPECT_EQ(buffer.channels()[0][0], kFloatConstant);
AudioFrameView<const float> const_view(
buffer.channels(), buffer.num_channels(), buffer.num_frames());
// Modification is not allowed.
// const_view.channel(0)[0] = kFloatConstant;
// Assignment is allowed.
AudioFrameView<const float> other_const_view = non_const_view;
static_cast<void>(other_const_view);
// But not the other way. The following will fail:
// non_const_view = other_const_view;
AudioFrameView<float> non_const_float_view(
buffer.channels(), buffer.num_channels(), buffer.num_frames());
non_const_float_view.channel(0)[0] = kIntConstant;
EXPECT_EQ(buffer.channels()[0][0], kIntConstant);
}
TEST(AudioFrameTest, ConstructFromChannelBuffer) {
ChannelBuffer<float> buffer(480, 2);
AudioFrameView<float> view(buffer.channels(), buffer.num_channels(),
buffer.num_frames());
EXPECT_EQ(view.num_channels(), 2);
EXPECT_EQ(view.samples_per_channel(), 480);
}
TEST(AudioFrameTest, ToDeinterleavedView) {
ChannelBuffer<float> buffer(480, 2);
AudioFrameView<float> view(buffer.channels(), buffer.num_channels(),
buffer.num_frames());
DeinterleavedView<float> non_const_view = view.view();
DeinterleavedView<const float> const_view =
static_cast<const AudioFrameView<float>&>(view).view();
ASSERT_EQ(non_const_view.num_channels(), 2u);
ASSERT_EQ(const_view.num_channels(), 2u);
for (size_t i = 0; i < non_const_view.num_channels(); ++i) {
EXPECT_EQ(non_const_view[i].data(), const_view[i].data());
EXPECT_EQ(non_const_view[i].data(), view.channel(i).data());
}
}
TEST(AudioFrameTest, FromDeinterleavedView) {
std::array<float, 480 * 2> buffer;
DeinterleavedView<float> view(buffer.data(), buffer.size() / 2u, 2u);
AudioFrameView<float> frame_view(view);
EXPECT_EQ(static_cast<size_t>(frame_view.num_channels()),
view.num_channels());
EXPECT_EQ(frame_view[0], view[0]);
EXPECT_EQ(frame_view[1], view[1]);
}
} // namespace webrtc