blob: 7c0c0ef665d9e978ca0c5806e1ed3bbdf04ad519 [file] [log] [blame]
Per Åhgren4dd56a32019-11-19 20:00:591/*
2 * Copyright (c) 2013 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_coding/acm2/acm_remixing.h"
12
13#include <vector>
14
15#include "api/audio/audio_frame.h"
16#include "system_wrappers/include/clock.h"
17#include "test/gmock.h"
18#include "test/gtest.h"
19#include "test/testsupport/file_utils.h"
20
21using ::testing::AllOf;
22using ::testing::Each;
23using ::testing::ElementsAreArray;
24using ::testing::SizeIs;
25
26namespace webrtc {
27
28TEST(AcmRemixing, DownMixFrame) {
29 std::vector<int16_t> out(480, 0);
30 AudioFrame in;
Tommi5d3e6802024-05-24 14:43:5531 InterleavedView<int16_t> const in_data = in.mutable_data(480, 2);
32 for (size_t k = 0; k < in_data.samples_per_channel(); ++k) {
Per Åhgren4dd56a32019-11-19 20:00:5933 in_data[2 * k] = 2;
34 in_data[2 * k + 1] = 0;
35 }
36
37 DownMixFrame(in, out);
38
39 EXPECT_THAT(out, AllOf(SizeIs(480), Each(1)));
40}
41
42TEST(AcmRemixing, DownMixMutedFrame) {
43 std::vector<int16_t> out(480, 0);
44 AudioFrame in;
45 in.num_channels_ = 2;
46 in.samples_per_channel_ = 480;
47
48 int16_t* const in_data = in.mutable_data();
49 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
50 in_data[2 * k] = 2;
51 in_data[2 * k + 1] = 0;
52 }
53
54 in.Mute();
55
56 DownMixFrame(in, out);
57
58 EXPECT_THAT(out, AllOf(SizeIs(480), Each(0)));
59}
60
61TEST(AcmRemixing, RemixMutedStereoFrameTo6Channels) {
62 std::vector<int16_t> out(480, 0);
63 AudioFrame in;
64 in.num_channels_ = 2;
65 in.samples_per_channel_ = 480;
66
67 int16_t* const in_data = in.mutable_data();
68 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
69 in_data[2 * k] = 1;
70 in_data[2 * k + 1] = 2;
71 }
72 in.Mute();
73
74 ReMixFrame(in, 6, &out);
75 EXPECT_EQ(6 * 480u, out.size());
76
77 EXPECT_THAT(out, AllOf(SizeIs(in.samples_per_channel_ * 6), Each(0)));
78}
79
80TEST(AcmRemixing, RemixStereoFrameTo6Channels) {
81 std::vector<int16_t> out(480, 0);
82 AudioFrame in;
83 in.num_channels_ = 2;
84 in.samples_per_channel_ = 480;
85
86 int16_t* const in_data = in.mutable_data();
87 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
88 in_data[2 * k] = 1;
89 in_data[2 * k + 1] = 2;
90 }
91
92 ReMixFrame(in, 6, &out);
93 EXPECT_EQ(6 * 480u, out.size());
94
95 std::vector<int16_t> expected_output(in.samples_per_channel_ * 6);
96 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
97 expected_output[6 * k] = 1;
98 expected_output[6 * k + 1] = 2;
99 }
100
101 EXPECT_THAT(out, ElementsAreArray(expected_output));
102}
103
104TEST(AcmRemixing, RemixMonoFrameTo6Channels) {
105 std::vector<int16_t> out(480, 0);
106 AudioFrame in;
107 in.num_channels_ = 1;
108 in.samples_per_channel_ = 480;
109
110 int16_t* const in_data = in.mutable_data();
111 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
112 in_data[k] = 1;
113 }
114
115 ReMixFrame(in, 6, &out);
116 EXPECT_EQ(6 * 480u, out.size());
117
118 std::vector<int16_t> expected_output(in.samples_per_channel_ * 6, 0);
119 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
120 expected_output[6 * k] = 1;
121 expected_output[6 * k + 1] = 1;
122 }
123
124 EXPECT_THAT(out, ElementsAreArray(expected_output));
125}
126
127TEST(AcmRemixing, RemixStereoFrameToMono) {
128 std::vector<int16_t> out(480, 0);
129 AudioFrame in;
130 in.num_channels_ = 2;
131 in.samples_per_channel_ = 480;
132
133 int16_t* const in_data = in.mutable_data();
134 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
135 in_data[2 * k] = 2;
136 in_data[2 * k + 1] = 0;
137 }
138
139 ReMixFrame(in, 1, &out);
140 EXPECT_EQ(480u, out.size());
141
142 EXPECT_THAT(out, AllOf(SizeIs(in.samples_per_channel_), Each(1)));
143}
144
145TEST(AcmRemixing, RemixMonoFrameToStereo) {
146 std::vector<int16_t> out(480, 0);
147 AudioFrame in;
148 in.num_channels_ = 1;
149 in.samples_per_channel_ = 480;
150
151 int16_t* const in_data = in.mutable_data();
152 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
153 in_data[k] = 1;
154 }
155
156 ReMixFrame(in, 2, &out);
157 EXPECT_EQ(960u, out.size());
158
159 EXPECT_THAT(out, AllOf(SizeIs(2 * in.samples_per_channel_), Each(1)));
160}
161
162TEST(AcmRemixing, Remix3ChannelFrameToStereo) {
163 std::vector<int16_t> out(480, 0);
164 AudioFrame in;
165 in.num_channels_ = 3;
166 in.samples_per_channel_ = 480;
167
168 int16_t* const in_data = in.mutable_data();
169 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
170 for (size_t j = 0; j < 3; ++j) {
171 in_data[3 * k + j] = j;
172 }
173 }
174
175 ReMixFrame(in, 2, &out);
176 EXPECT_EQ(2 * 480u, out.size());
177
178 std::vector<int16_t> expected_output(in.samples_per_channel_ * 2);
179 for (size_t k = 0; k < in.samples_per_channel_; ++k) {
180 for (size_t j = 0; j < 2; ++j) {
181 expected_output[2 * k + j] = static_cast<int>(j);
182 }
183 }
184
185 EXPECT_THAT(out, ElementsAreArray(expected_output));
186}
187
188} // namespace webrtc