1 /*
2 * Copyright (c) 2018 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "modules/audio_processing/agc2/rnn_vad/sequence_buffer.h"
12
13 #include <algorithm>
14 #include <array>
15
16 #include "test/gtest.h"
17
18 namespace webrtc {
19 namespace rnn_vad {
20 namespace test {
21 namespace {
22
23 template <typename T, size_t S, size_t N>
TestSequenceBufferPushOp()24 void TestSequenceBufferPushOp() {
25 SCOPED_TRACE(S);
26 SCOPED_TRACE(N);
27 SequenceBuffer<T, S, N> seq_buf;
28 auto seq_buf_view = seq_buf.GetBufferView();
29 std::array<T, N> chunk;
30
31 // Check that a chunk is fully gone after ceil(S / N) push ops.
32 chunk.fill(1);
33 seq_buf.Push(chunk);
34 chunk.fill(0);
35 constexpr size_t required_push_ops = (S % N) ? S / N + 1 : S / N;
36 for (size_t i = 0; i < required_push_ops - 1; ++i) {
37 SCOPED_TRACE(i);
38 seq_buf.Push(chunk);
39 // Still in the buffer.
40 const auto* m = std::max_element(seq_buf_view.begin(), seq_buf_view.end());
41 EXPECT_EQ(1, *m);
42 }
43 // Gone after another push.
44 seq_buf.Push(chunk);
45 const auto* m = std::max_element(seq_buf_view.begin(), seq_buf_view.end());
46 EXPECT_EQ(0, *m);
47
48 // Check that the last item moves left by N positions after a push op.
49 if (S > N) {
50 // Fill in with non-zero values.
51 for (size_t i = 0; i < N; ++i)
52 chunk[i] = static_cast<T>(i + 1);
53 seq_buf.Push(chunk);
54 // With the next Push(), |last| will be moved left by N positions.
55 const T last = chunk[N - 1];
56 for (size_t i = 0; i < N; ++i)
57 chunk[i] = static_cast<T>(last + i + 1);
58 seq_buf.Push(chunk);
59 EXPECT_EQ(last, seq_buf_view[S - N - 1]);
60 }
61 }
62
63 } // namespace
64
TEST(RnnVadTest,SequenceBufferGetters)65 TEST(RnnVadTest, SequenceBufferGetters) {
66 constexpr size_t buffer_size = 8;
67 constexpr size_t chunk_size = 8;
68 SequenceBuffer<int, buffer_size, chunk_size> seq_buf;
69 EXPECT_EQ(buffer_size, seq_buf.size());
70 EXPECT_EQ(chunk_size, seq_buf.chunks_size());
71 // Test view.
72 auto seq_buf_view = seq_buf.GetBufferView();
73 EXPECT_EQ(0, seq_buf_view[0]);
74 EXPECT_EQ(0, seq_buf_view[seq_buf_view.size() - 1]);
75 constexpr std::array<int, chunk_size> chunk = {10, 20, 30, 40,
76 50, 60, 70, 80};
77 seq_buf.Push(chunk);
78 EXPECT_EQ(10, *seq_buf_view.begin());
79 EXPECT_EQ(80, *(seq_buf_view.end() - 1));
80 }
81
TEST(RnnVadTest,SequenceBufferPushOpsUnsigned)82 TEST(RnnVadTest, SequenceBufferPushOpsUnsigned) {
83 TestSequenceBufferPushOp<uint8_t, 32, 8>(); // Chunk size: 25%.
84 TestSequenceBufferPushOp<uint8_t, 32, 16>(); // Chunk size: 50%.
85 TestSequenceBufferPushOp<uint8_t, 32, 32>(); // Chunk size: 100%.
86 TestSequenceBufferPushOp<uint8_t, 23, 7>(); // Non-integer ratio.
87 }
88
TEST(RnnVadTest,SequenceBufferPushOpsSigned)89 TEST(RnnVadTest, SequenceBufferPushOpsSigned) {
90 TestSequenceBufferPushOp<int, 32, 8>(); // Chunk size: 25%.
91 TestSequenceBufferPushOp<int, 32, 16>(); // Chunk size: 50%.
92 TestSequenceBufferPushOp<int, 32, 32>(); // Chunk size: 100%.
93 TestSequenceBufferPushOp<int, 23, 7>(); // Non-integer ratio.
94 }
95
TEST(RnnVadTest,SequenceBufferPushOpsFloating)96 TEST(RnnVadTest, SequenceBufferPushOpsFloating) {
97 TestSequenceBufferPushOp<float, 32, 8>(); // Chunk size: 25%.
98 TestSequenceBufferPushOp<float, 32, 16>(); // Chunk size: 50%.
99 TestSequenceBufferPushOp<float, 32, 32>(); // Chunk size: 100%.
100 TestSequenceBufferPushOp<float, 23, 7>(); // Non-integer ratio.
101 }
102
103 } // namespace test
104 } // namespace rnn_vad
105 } // namespace webrtc
106