blob: f06465179857eee96ff9fb74cd73f0797467a746 [file] [log] [blame]
/*
* Copyright (c) 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/agc2/rnn_vad/ring_buffer.h"
#include "test/gtest.h"
namespace webrtc {
namespace rnn_vad {
namespace test {
namespace {
// Compare the elements of two given array views.
template <typename T, std::ptrdiff_t S>
void ExpectEq(rtc::ArrayView<const T, S> a, rtc::ArrayView<const T, S> b) {
for (size_t i = 0; i < S; ++i) {
SCOPED_TRACE(i);
EXPECT_EQ(a[i], b[i]);
}
}
// Test push/read sequences.
template <typename T, size_t S, size_t N>
void TestRingBuffer() {
SCOPED_TRACE(N);
SCOPED_TRACE(S);
std::array<T, S> prev_pushed_array;
std::array<T, S> pushed_array;
rtc::ArrayView<const T, S> pushed_array_view(pushed_array.data(), S);
// Init.
RingBuffer<T, S, N> ring_buf;
ring_buf.GetArrayView(0);
pushed_array.fill(0);
ring_buf.Push(pushed_array_view);
ExpectEq(pushed_array_view, ring_buf.GetArrayView(0));
// Push N times and check most recent and second most recent.
for (T v = 1; v <= static_cast<T>(N); ++v) {
SCOPED_TRACE(v);
prev_pushed_array = pushed_array;
pushed_array.fill(v);
ring_buf.Push(pushed_array_view);
ExpectEq(pushed_array_view, ring_buf.GetArrayView(0));
if (N > 1) {
pushed_array.fill(v - 1);
ExpectEq(pushed_array_view, ring_buf.GetArrayView(1));
}
}
// Check buffer.
for (size_t delay = 2; delay < N; ++delay) {
SCOPED_TRACE(delay);
T expected_value = N - static_cast<T>(delay);
pushed_array.fill(expected_value);
ExpectEq(pushed_array_view, ring_buf.GetArrayView(delay));
}
}
} // namespace
// Check that for different delays, different views are returned.
TEST(RnnVadTest, RingBufferArrayViews) {
constexpr size_t s = 3;
constexpr size_t n = 4;
RingBuffer<int, s, n> ring_buf;
std::array<int, s> pushed_array;
pushed_array.fill(1);
for (size_t k = 0; k <= n; ++k) { // Push data n + 1 times.
SCOPED_TRACE(k);
// Check array views.
for (size_t i = 0; i < n; ++i) {
SCOPED_TRACE(i);
auto view_i = ring_buf.GetArrayView(i);
for (size_t j = i + 1; j < n; ++j) {
SCOPED_TRACE(j);
auto view_j = ring_buf.GetArrayView(j);
EXPECT_NE(view_i, view_j);
}
}
ring_buf.Push(pushed_array);
}
}
TEST(RnnVadTest, RingBufferUnsigned) {
TestRingBuffer<uint8_t, 1, 1>();
TestRingBuffer<uint8_t, 2, 5>();
TestRingBuffer<uint8_t, 5, 2>();
TestRingBuffer<uint8_t, 5, 5>();
}
TEST(RnnVadTest, RingBufferSigned) {
TestRingBuffer<int, 1, 1>();
TestRingBuffer<int, 2, 5>();
TestRingBuffer<int, 5, 2>();
TestRingBuffer<int, 5, 5>();
}
TEST(RnnVadTest, RingBufferFloating) {
TestRingBuffer<float, 1, 1>();
TestRingBuffer<float, 2, 5>();
TestRingBuffer<float, 5, 2>();
TestRingBuffer<float, 5, 5>();
}
} // namespace test
} // namespace rnn_vad
} // namespace webrtc