aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2016 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 | |
Jonas Olsson | a4d8737 | 2019-07-05 19:08:33 +0200 | [diff] [blame] | 11 | #include "modules/audio_mixer/audio_mixer_impl.h" |
| 12 | |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 13 | #include <string.h> |
| 14 | |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 15 | #include <limits> |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 16 | #include <memory> |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 17 | #include <string> |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 18 | #include <utility> |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 19 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 20 | #include "api/audio/audio_mixer.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 21 | #include "modules/audio_mixer/default_output_rate_calculator.h" |
| 22 | #include "rtc_base/bind.h" |
| 23 | #include "rtc_base/checks.h" |
Jonas Olsson | 366a50c | 2018-09-06 13:41:30 +0200 | [diff] [blame] | 24 | #include "rtc_base/strings/string_builder.h" |
Tommi | 6856156 | 2018-02-13 19:47:50 +0100 | [diff] [blame] | 25 | #include "rtc_base/task_queue_for_test.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 26 | #include "test/gmock.h" |
Alex Loiko | b4977de | 2019-01-28 16:38:38 +0100 | [diff] [blame] | 27 | #include "test/gtest.h" |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 28 | |
Mirko Bonadei | 6a489f2 | 2019-04-09 15:11:12 +0200 | [diff] [blame] | 29 | using ::testing::_; |
| 30 | using ::testing::Exactly; |
| 31 | using ::testing::Invoke; |
| 32 | using ::testing::Return; |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 33 | |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 34 | namespace webrtc { |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 35 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 36 | namespace { |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 37 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 38 | constexpr int kDefaultSampleRateHz = 48000; |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 39 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 40 | // Utility function that resets the frame member variables with |
| 41 | // sensible defaults. |
| 42 | void ResetFrame(AudioFrame* frame) { |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 43 | frame->sample_rate_hz_ = kDefaultSampleRateHz; |
| 44 | frame->num_channels_ = 1; |
| 45 | |
| 46 | // Frame duration 10ms. |
| 47 | frame->samples_per_channel_ = kDefaultSampleRateHz / 100; |
| 48 | frame->vad_activity_ = AudioFrame::kVadActive; |
| 49 | frame->speech_type_ = AudioFrame::kNormalSpeech; |
| 50 | } |
| 51 | |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 52 | std::string ProduceDebugText(int sample_rate_hz, |
| 53 | int number_of_channels, |
| 54 | int number_of_sources) { |
Jonas Olsson | 366a50c | 2018-09-06 13:41:30 +0200 | [diff] [blame] | 55 | rtc::StringBuilder ss; |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 56 | ss << "Sample rate: " << sample_rate_hz << " "; |
| 57 | ss << "Number of channels: " << number_of_channels << " "; |
| 58 | ss << "Number of sources: " << number_of_sources; |
Jonas Olsson | 84df1c7 | 2018-09-14 16:59:32 +0200 | [diff] [blame] | 59 | return ss.Release(); |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 60 | } |
| 61 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 62 | AudioFrame frame_for_mixing; |
| 63 | |
| 64 | } // namespace |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 65 | |
Mirko Bonadei | 6a489f2 | 2019-04-09 15:11:12 +0200 | [diff] [blame] | 66 | class MockMixerAudioSource : public ::testing::NiceMock<AudioMixer::Source> { |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 67 | public: |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 68 | MockMixerAudioSource() |
aleloi | e891415 | 2016-10-11 06:18:31 -0700 | [diff] [blame] | 69 | : fake_audio_frame_info_(AudioMixer::Source::AudioFrameInfo::kNormal) { |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 70 | ON_CALL(*this, GetAudioFrameWithInfo(_, _)) |
aleloi | a0db81f | 2016-07-28 06:36:22 -0700 | [diff] [blame] | 71 | .WillByDefault( |
aleloi | e891415 | 2016-10-11 06:18:31 -0700 | [diff] [blame] | 72 | Invoke(this, &MockMixerAudioSource::FakeAudioFrameWithInfo)); |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 73 | ON_CALL(*this, PreferredSampleRate()) |
| 74 | .WillByDefault(Return(kDefaultSampleRateHz)); |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 75 | } |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 76 | |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 77 | MOCK_METHOD2(GetAudioFrameWithInfo, |
| 78 | AudioFrameInfo(int sample_rate_hz, AudioFrame* audio_frame)); |
aleloi | a0db81f | 2016-07-28 06:36:22 -0700 | [diff] [blame] | 79 | |
aleloi | 051f678 | 2016-10-31 03:26:40 -0700 | [diff] [blame] | 80 | MOCK_CONST_METHOD0(PreferredSampleRate, int()); |
| 81 | MOCK_CONST_METHOD0(Ssrc, int()); |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 82 | |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 83 | AudioFrame* fake_frame() { return &fake_frame_; } |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 84 | AudioFrameInfo fake_info() { return fake_audio_frame_info_; } |
| 85 | void set_fake_info(const AudioFrameInfo audio_frame_info) { |
| 86 | fake_audio_frame_info_ = audio_frame_info; |
| 87 | } |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 88 | |
| 89 | private: |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 90 | AudioFrameInfo FakeAudioFrameWithInfo(int sample_rate_hz, |
| 91 | AudioFrame* audio_frame) { |
| 92 | audio_frame->CopyFrom(fake_frame_); |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 93 | audio_frame->sample_rate_hz_ = sample_rate_hz; |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 94 | audio_frame->samples_per_channel_ = |
| 95 | rtc::CheckedDivExact(sample_rate_hz, 100); |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 96 | return fake_info(); |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 97 | } |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 98 | |
| 99 | AudioFrame fake_frame_; |
| 100 | AudioFrameInfo fake_audio_frame_info_; |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 101 | }; |
| 102 | |
aleloi | 623427c | 2016-12-08 02:37:58 -0800 | [diff] [blame] | 103 | class CustomRateCalculator : public OutputRateCalculator { |
| 104 | public: |
| 105 | explicit CustomRateCalculator(int rate) : rate_(rate) {} |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 106 | int CalculateOutputRate(const std::vector<int>& preferred_rates) override { |
aleloi | 623427c | 2016-12-08 02:37:58 -0800 | [diff] [blame] | 107 | return rate_; |
| 108 | } |
| 109 | |
| 110 | private: |
| 111 | const int rate_; |
| 112 | }; |
| 113 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 114 | // Creates participants from |frames| and |frame_info| and adds them |
| 115 | // to the mixer. Compares mixed status with |expected_status| |
| 116 | void MixAndCompare( |
| 117 | const std::vector<AudioFrame>& frames, |
aleloi | e891415 | 2016-10-11 06:18:31 -0700 | [diff] [blame] | 118 | const std::vector<AudioMixer::Source::AudioFrameInfo>& frame_info, |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 119 | const std::vector<bool>& expected_status) { |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 120 | const size_t num_audio_sources = frames.size(); |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 121 | RTC_DCHECK(frames.size() == frame_info.size()); |
| 122 | RTC_DCHECK(frame_info.size() == expected_status.size()); |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 123 | |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 124 | const auto mixer = AudioMixerImpl::Create(); |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 125 | std::vector<MockMixerAudioSource> participants(num_audio_sources); |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 126 | |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 127 | for (size_t i = 0; i < num_audio_sources; ++i) { |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 128 | participants[i].fake_frame()->CopyFrom(frames[i]); |
| 129 | participants[i].set_fake_info(frame_info[i]); |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 130 | } |
| 131 | |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 132 | for (size_t i = 0; i < num_audio_sources; ++i) { |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 133 | EXPECT_TRUE(mixer->AddSource(&participants[i])); |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 134 | EXPECT_CALL(participants[i], GetAudioFrameWithInfo(kDefaultSampleRateHz, _)) |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 135 | .Times(Exactly(1)); |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 136 | } |
| 137 | |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 138 | mixer->Mix(1, &frame_for_mixing); |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 139 | |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 140 | for (size_t i = 0; i < num_audio_sources; ++i) { |
aleloi | 3654251 | 2016-10-07 05:28:32 -0700 | [diff] [blame] | 141 | EXPECT_EQ(expected_status[i], |
| 142 | mixer->GetAudioSourceMixabilityStatusForTest(&participants[i])) |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 143 | << "Mixed status of AudioSource #" << i << " wrong."; |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 144 | } |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 145 | } |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 146 | |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 147 | void MixMonoAtGivenNativeRate(int native_sample_rate, |
| 148 | AudioFrame* mix_frame, |
| 149 | rtc::scoped_refptr<AudioMixer> mixer, |
| 150 | MockMixerAudioSource* audio_source) { |
| 151 | ON_CALL(*audio_source, PreferredSampleRate()) |
| 152 | .WillByDefault(Return(native_sample_rate)); |
| 153 | audio_source->fake_frame()->sample_rate_hz_ = native_sample_rate; |
| 154 | audio_source->fake_frame()->samples_per_channel_ = native_sample_rate / 100; |
| 155 | |
| 156 | mixer->Mix(1, mix_frame); |
| 157 | } |
| 158 | |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 159 | TEST(AudioMixer, LargestEnergyVadActiveMixed) { |
aleloi | 616df1e | 2016-08-24 01:17:12 -0700 | [diff] [blame] | 160 | constexpr int kAudioSources = |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 161 | AudioMixerImpl::kMaximumAmountOfMixedAudioSources + 3; |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 162 | |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 163 | const auto mixer = AudioMixerImpl::Create(); |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 164 | |
| 165 | MockMixerAudioSource participants[kAudioSources]; |
| 166 | |
| 167 | for (int i = 0; i < kAudioSources; ++i) { |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 168 | ResetFrame(participants[i].fake_frame()); |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 169 | |
| 170 | // We set the 80-th sample value since the first 80 samples may be |
| 171 | // modified by a ramped-in window. |
yujo | 36b1a5f | 2017-06-12 12:45:32 -0700 | [diff] [blame] | 172 | participants[i].fake_frame()->mutable_data()[80] = i; |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 173 | |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 174 | EXPECT_TRUE(mixer->AddSource(&participants[i])); |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 175 | EXPECT_CALL(participants[i], GetAudioFrameWithInfo(_, _)).Times(Exactly(1)); |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 176 | } |
| 177 | |
| 178 | // Last participant gives audio frame with passive VAD, although it has the |
| 179 | // largest energy. |
| 180 | participants[kAudioSources - 1].fake_frame()->vad_activity_ = |
| 181 | AudioFrame::kVadPassive; |
| 182 | |
| 183 | AudioFrame audio_frame; |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 184 | mixer->Mix(1, // number of channels |
aleloi | 4496809 | 2016-08-08 10:18:58 -0700 | [diff] [blame] | 185 | &audio_frame); |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 186 | |
| 187 | for (int i = 0; i < kAudioSources; ++i) { |
aleloi | 3654251 | 2016-10-07 05:28:32 -0700 | [diff] [blame] | 188 | bool is_mixed = |
| 189 | mixer->GetAudioSourceMixabilityStatusForTest(&participants[i]); |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 190 | if (i == kAudioSources - 1 || |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 191 | i < kAudioSources - 1 - |
| 192 | AudioMixerImpl::kMaximumAmountOfMixedAudioSources) { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 193 | EXPECT_FALSE(is_mixed) |
| 194 | << "Mixing status of AudioSource #" << i << " wrong."; |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 195 | } else { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 196 | EXPECT_TRUE(is_mixed) |
| 197 | << "Mixing status of AudioSource #" << i << " wrong."; |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 198 | } |
| 199 | } |
| 200 | } |
| 201 | |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 202 | TEST(AudioMixer, FrameNotModifiedForSingleParticipant) { |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 203 | const auto mixer = AudioMixerImpl::Create(); |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 204 | |
| 205 | MockMixerAudioSource participant; |
| 206 | |
| 207 | ResetFrame(participant.fake_frame()); |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 208 | const size_t n_samples = participant.fake_frame()->samples_per_channel_; |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 209 | |
| 210 | // Modify the frame so that it's not zero. |
yujo | 36b1a5f | 2017-06-12 12:45:32 -0700 | [diff] [blame] | 211 | int16_t* fake_frame_data = participant.fake_frame()->mutable_data(); |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 212 | for (size_t j = 0; j < n_samples; ++j) { |
yujo | 36b1a5f | 2017-06-12 12:45:32 -0700 | [diff] [blame] | 213 | fake_frame_data[j] = static_cast<int16_t>(j); |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 214 | } |
| 215 | |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 216 | EXPECT_TRUE(mixer->AddSource(&participant)); |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 217 | EXPECT_CALL(participant, GetAudioFrameWithInfo(_, _)).Times(Exactly(2)); |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 218 | |
| 219 | AudioFrame audio_frame; |
| 220 | // Two mix iteration to compare after the ramp-up step. |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 221 | for (int i = 0; i < 2; ++i) { |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 222 | mixer->Mix(1, // number of channels |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 223 | &audio_frame); |
| 224 | } |
| 225 | |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 226 | EXPECT_EQ(0, memcmp(participant.fake_frame()->data(), audio_frame.data(), |
| 227 | n_samples)); |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 228 | } |
| 229 | |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 230 | TEST(AudioMixer, SourceAtNativeRateShouldNeverResample) { |
| 231 | const auto mixer = AudioMixerImpl::Create(); |
| 232 | |
| 233 | MockMixerAudioSource audio_source; |
| 234 | ResetFrame(audio_source.fake_frame()); |
| 235 | |
| 236 | mixer->AddSource(&audio_source); |
| 237 | |
| 238 | for (auto frequency : {8000, 16000, 32000, 48000}) { |
| 239 | EXPECT_CALL(audio_source, GetAudioFrameWithInfo(frequency, _)) |
| 240 | .Times(Exactly(1)); |
| 241 | |
| 242 | MixMonoAtGivenNativeRate(frequency, &frame_for_mixing, mixer, |
| 243 | &audio_source); |
| 244 | } |
| 245 | } |
| 246 | |
| 247 | TEST(AudioMixer, MixerShouldMixAtNativeSourceRate) { |
| 248 | const auto mixer = AudioMixerImpl::Create(); |
| 249 | |
| 250 | MockMixerAudioSource audio_source; |
| 251 | ResetFrame(audio_source.fake_frame()); |
| 252 | |
| 253 | mixer->AddSource(&audio_source); |
| 254 | |
| 255 | for (auto frequency : {8000, 16000, 32000, 48000}) { |
| 256 | MixMonoAtGivenNativeRate(frequency, &frame_for_mixing, mixer, |
| 257 | &audio_source); |
| 258 | |
| 259 | EXPECT_EQ(frequency, frame_for_mixing.sample_rate_hz_); |
| 260 | } |
| 261 | } |
| 262 | |
| 263 | TEST(AudioMixer, MixerShouldAlwaysMixAtNativeRate) { |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 264 | const auto mixer = AudioMixerImpl::Create(); |
aleloi | 4496809 | 2016-08-08 10:18:58 -0700 | [diff] [blame] | 265 | |
| 266 | MockMixerAudioSource participant; |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 267 | ResetFrame(participant.fake_frame()); |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 268 | mixer->AddSource(&participant); |
aleloi | 4496809 | 2016-08-08 10:18:58 -0700 | [diff] [blame] | 269 | |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 270 | const int needed_frequency = 44100; |
| 271 | ON_CALL(participant, PreferredSampleRate()) |
| 272 | .WillByDefault(Return(needed_frequency)); |
| 273 | |
| 274 | // We expect mixing frequency to be native and >= needed_frequency. |
| 275 | const int expected_mix_frequency = 48000; |
| 276 | EXPECT_CALL(participant, GetAudioFrameWithInfo(expected_mix_frequency, _)) |
| 277 | .Times(Exactly(1)); |
| 278 | participant.fake_frame()->sample_rate_hz_ = expected_mix_frequency; |
| 279 | participant.fake_frame()->samples_per_channel_ = expected_mix_frequency / 100; |
| 280 | |
| 281 | mixer->Mix(1, &frame_for_mixing); |
| 282 | |
| 283 | EXPECT_EQ(48000, frame_for_mixing.sample_rate_hz_); |
| 284 | } |
| 285 | |
| 286 | // Check that the mixing rate is always >= participants preferred rate. |
| 287 | TEST(AudioMixer, ShouldNotCauseQualityLossForMultipleSources) { |
| 288 | const auto mixer = AudioMixerImpl::Create(); |
| 289 | |
| 290 | std::vector<MockMixerAudioSource> audio_sources(2); |
| 291 | const std::vector<int> source_sample_rates = {8000, 16000}; |
| 292 | for (int i = 0; i < 2; ++i) { |
| 293 | auto& source = audio_sources[i]; |
| 294 | ResetFrame(source.fake_frame()); |
| 295 | mixer->AddSource(&source); |
| 296 | const auto sample_rate = source_sample_rates[i]; |
| 297 | EXPECT_CALL(source, PreferredSampleRate()).WillOnce(Return(sample_rate)); |
| 298 | |
Mirko Bonadei | 6a489f2 | 2019-04-09 15:11:12 +0200 | [diff] [blame] | 299 | EXPECT_CALL(source, GetAudioFrameWithInfo(::testing::Ge(sample_rate), _)); |
aleloi | 4496809 | 2016-08-08 10:18:58 -0700 | [diff] [blame] | 300 | } |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 301 | mixer->Mix(1, &frame_for_mixing); |
aleloi | 4496809 | 2016-08-08 10:18:58 -0700 | [diff] [blame] | 302 | } |
| 303 | |
| 304 | TEST(AudioMixer, ParticipantNumberOfChannels) { |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 305 | const auto mixer = AudioMixerImpl::Create(); |
aleloi | 4496809 | 2016-08-08 10:18:58 -0700 | [diff] [blame] | 306 | |
| 307 | MockMixerAudioSource participant; |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 308 | ResetFrame(participant.fake_frame()); |
aleloi | 4496809 | 2016-08-08 10:18:58 -0700 | [diff] [blame] | 309 | |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 310 | EXPECT_TRUE(mixer->AddSource(&participant)); |
aleloi | 4496809 | 2016-08-08 10:18:58 -0700 | [diff] [blame] | 311 | for (size_t number_of_channels : {1, 2}) { |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 312 | EXPECT_CALL(participant, GetAudioFrameWithInfo(kDefaultSampleRateHz, _)) |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 313 | .Times(Exactly(1)); |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 314 | mixer->Mix(number_of_channels, &frame_for_mixing); |
aleloi | 4496809 | 2016-08-08 10:18:58 -0700 | [diff] [blame] | 315 | EXPECT_EQ(number_of_channels, frame_for_mixing.num_channels_); |
| 316 | } |
| 317 | } |
| 318 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 319 | // Maximal amount of participants are mixed one iteration, then |
| 320 | // another participant with higher energy is added. |
| 321 | TEST(AudioMixer, RampedOutSourcesShouldNotBeMarkedMixed) { |
| 322 | constexpr int kAudioSources = |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 323 | AudioMixerImpl::kMaximumAmountOfMixedAudioSources + 1; |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 324 | |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 325 | const auto mixer = AudioMixerImpl::Create(); |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 326 | MockMixerAudioSource participants[kAudioSources]; |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 327 | |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 328 | for (int i = 0; i < kAudioSources; ++i) { |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 329 | ResetFrame(participants[i].fake_frame()); |
| 330 | // Set the participant audio energy to increase with the index |
| 331 | // |i|. |
yujo | 36b1a5f | 2017-06-12 12:45:32 -0700 | [diff] [blame] | 332 | participants[i].fake_frame()->mutable_data()[0] = 100 * i; |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 333 | } |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 334 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 335 | // Add all participants but the loudest for mixing. |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 336 | for (int i = 0; i < kAudioSources - 1; ++i) { |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 337 | EXPECT_TRUE(mixer->AddSource(&participants[i])); |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 338 | EXPECT_CALL(participants[i], GetAudioFrameWithInfo(kDefaultSampleRateHz, _)) |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 339 | .Times(Exactly(1)); |
| 340 | } |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 341 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 342 | // First mixer iteration |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 343 | mixer->Mix(1, &frame_for_mixing); |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 344 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 345 | // All participants but the loudest should have been mixed. |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 346 | for (int i = 0; i < kAudioSources - 1; ++i) { |
aleloi | 3654251 | 2016-10-07 05:28:32 -0700 | [diff] [blame] | 347 | EXPECT_TRUE(mixer->GetAudioSourceMixabilityStatusForTest(&participants[i])) |
| 348 | << "Mixed status of AudioSource #" << i << " wrong."; |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 349 | } |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 350 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 351 | // Add new participant with higher energy. |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 352 | EXPECT_TRUE(mixer->AddSource(&participants[kAudioSources - 1])); |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 353 | for (int i = 0; i < kAudioSources; ++i) { |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 354 | EXPECT_CALL(participants[i], GetAudioFrameWithInfo(kDefaultSampleRateHz, _)) |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 355 | .Times(Exactly(1)); |
| 356 | } |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 357 | |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 358 | mixer->Mix(1, &frame_for_mixing); |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 359 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 360 | // The most quiet participant should not have been mixed. |
aleloi | 3654251 | 2016-10-07 05:28:32 -0700 | [diff] [blame] | 361 | EXPECT_FALSE(mixer->GetAudioSourceMixabilityStatusForTest(&participants[0])) |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 362 | << "Mixed status of AudioSource #0 wrong."; |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 363 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 364 | // The loudest participants should have been mixed. |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 365 | for (int i = 1; i < kAudioSources; ++i) { |
aleloi | 3654251 | 2016-10-07 05:28:32 -0700 | [diff] [blame] | 366 | EXPECT_EQ(true, |
| 367 | mixer->GetAudioSourceMixabilityStatusForTest(&participants[i])) |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 368 | << "Mixed status of AudioSource #" << i << " wrong."; |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 369 | } |
| 370 | } |
| 371 | |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 372 | // This test checks that the initialization and participant addition |
| 373 | // can be done on a different thread. |
| 374 | TEST(AudioMixer, ConstructFromOtherThread) { |
Danil Chapovalov | d26a916 | 2019-03-19 18:08:37 +0100 | [diff] [blame] | 375 | TaskQueueForTest init_queue("init"); |
tommi | c45d6d9 | 2017-07-11 06:17:10 -0700 | [diff] [blame] | 376 | rtc::scoped_refptr<AudioMixer> mixer; |
Danil Chapovalov | eb90e6f | 2019-10-15 10:04:57 +0200 | [diff] [blame] | 377 | init_queue.SendTask([&mixer]() { mixer = AudioMixerImpl::Create(); }, |
| 378 | RTC_FROM_HERE); |
tommi | c45d6d9 | 2017-07-11 06:17:10 -0700 | [diff] [blame] | 379 | |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 380 | MockMixerAudioSource participant; |
tommi | c45d6d9 | 2017-07-11 06:17:10 -0700 | [diff] [blame] | 381 | EXPECT_CALL(participant, PreferredSampleRate()) |
| 382 | .WillRepeatedly(Return(kDefaultSampleRateHz)); |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 383 | |
| 384 | ResetFrame(participant.fake_frame()); |
| 385 | |
Danil Chapovalov | d26a916 | 2019-03-19 18:08:37 +0100 | [diff] [blame] | 386 | TaskQueueForTest participant_queue("participant"); |
Tommi | 6856156 | 2018-02-13 19:47:50 +0100 | [diff] [blame] | 387 | participant_queue.SendTask( |
Danil Chapovalov | eb90e6f | 2019-10-15 10:04:57 +0200 | [diff] [blame] | 388 | [&mixer, &participant]() { mixer->AddSource(&participant); }, |
| 389 | RTC_FROM_HERE); |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 390 | |
aleloi | 6c27849 | 2016-10-20 14:24:39 -0700 | [diff] [blame] | 391 | EXPECT_CALL(participant, GetAudioFrameWithInfo(kDefaultSampleRateHz, _)) |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 392 | .Times(Exactly(1)); |
| 393 | |
| 394 | // Do one mixer iteration |
aleloi | 9561183 | 2016-11-08 06:39:50 -0800 | [diff] [blame] | 395 | mixer->Mix(1, &frame_for_mixing); |
aleloi | 311525e | 2016-09-07 06:13:12 -0700 | [diff] [blame] | 396 | } |
| 397 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 398 | TEST(AudioMixer, MutedShouldMixAfterUnmuted) { |
| 399 | constexpr int kAudioSources = |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 400 | AudioMixerImpl::kMaximumAmountOfMixedAudioSources + 1; |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 401 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 402 | std::vector<AudioFrame> frames(kAudioSources); |
| 403 | for (auto& frame : frames) { |
| 404 | ResetFrame(&frame); |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 405 | } |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 406 | |
aleloi | e891415 | 2016-10-11 06:18:31 -0700 | [diff] [blame] | 407 | std::vector<AudioMixer::Source::AudioFrameInfo> frame_info( |
| 408 | kAudioSources, AudioMixer::Source::AudioFrameInfo::kNormal); |
| 409 | frame_info[0] = AudioMixer::Source::AudioFrameInfo::kMuted; |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 410 | std::vector<bool> expected_status(kAudioSources, true); |
| 411 | expected_status[0] = false; |
| 412 | |
| 413 | MixAndCompare(frames, frame_info, expected_status); |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 414 | } |
| 415 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 416 | TEST(AudioMixer, PassiveShouldMixAfterNormal) { |
| 417 | constexpr int kAudioSources = |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 418 | AudioMixerImpl::kMaximumAmountOfMixedAudioSources + 1; |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 419 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 420 | std::vector<AudioFrame> frames(kAudioSources); |
| 421 | for (auto& frame : frames) { |
| 422 | ResetFrame(&frame); |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 423 | } |
| 424 | |
aleloi | e891415 | 2016-10-11 06:18:31 -0700 | [diff] [blame] | 425 | std::vector<AudioMixer::Source::AudioFrameInfo> frame_info( |
| 426 | kAudioSources, AudioMixer::Source::AudioFrameInfo::kNormal); |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 427 | frames[0].vad_activity_ = AudioFrame::kVadPassive; |
| 428 | std::vector<bool> expected_status(kAudioSources, true); |
| 429 | expected_status[0] = false; |
| 430 | |
| 431 | MixAndCompare(frames, frame_info, expected_status); |
aleloi | f388257 | 2016-07-29 02:12:41 -0700 | [diff] [blame] | 432 | } |
| 433 | |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 434 | TEST(AudioMixer, ActiveShouldMixBeforeLoud) { |
| 435 | constexpr int kAudioSources = |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 436 | AudioMixerImpl::kMaximumAmountOfMixedAudioSources + 1; |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 437 | |
| 438 | std::vector<AudioFrame> frames(kAudioSources); |
| 439 | for (auto& frame : frames) { |
| 440 | ResetFrame(&frame); |
| 441 | } |
| 442 | |
aleloi | e891415 | 2016-10-11 06:18:31 -0700 | [diff] [blame] | 443 | std::vector<AudioMixer::Source::AudioFrameInfo> frame_info( |
| 444 | kAudioSources, AudioMixer::Source::AudioFrameInfo::kNormal); |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 445 | frames[0].vad_activity_ = AudioFrame::kVadPassive; |
yujo | 36b1a5f | 2017-06-12 12:45:32 -0700 | [diff] [blame] | 446 | int16_t* frame_data = frames[0].mutable_data(); |
| 447 | std::fill(frame_data, frame_data + kDefaultSampleRateHz / 100, |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 448 | std::numeric_limits<int16_t>::max()); |
| 449 | std::vector<bool> expected_status(kAudioSources, true); |
| 450 | expected_status[0] = false; |
| 451 | |
| 452 | MixAndCompare(frames, frame_info, expected_status); |
| 453 | } |
| 454 | |
| 455 | TEST(AudioMixer, UnmutedShouldMixBeforeLoud) { |
| 456 | constexpr int kAudioSources = |
aleloi | 116ec6d | 2016-10-12 06:07:07 -0700 | [diff] [blame] | 457 | AudioMixerImpl::kMaximumAmountOfMixedAudioSources + 1; |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 458 | |
| 459 | std::vector<AudioFrame> frames(kAudioSources); |
| 460 | for (auto& frame : frames) { |
| 461 | ResetFrame(&frame); |
| 462 | } |
| 463 | |
aleloi | e891415 | 2016-10-11 06:18:31 -0700 | [diff] [blame] | 464 | std::vector<AudioMixer::Source::AudioFrameInfo> frame_info( |
| 465 | kAudioSources, AudioMixer::Source::AudioFrameInfo::kNormal); |
| 466 | frame_info[0] = AudioMixer::Source::AudioFrameInfo::kMuted; |
yujo | 36b1a5f | 2017-06-12 12:45:32 -0700 | [diff] [blame] | 467 | int16_t* frame_data = frames[0].mutable_data(); |
| 468 | std::fill(frame_data, frame_data + kDefaultSampleRateHz / 100, |
aleloi | 30be5d7 | 2016-08-24 01:38:44 -0700 | [diff] [blame] | 469 | std::numeric_limits<int16_t>::max()); |
| 470 | std::vector<bool> expected_status(kAudioSources, true); |
| 471 | expected_status[0] = false; |
| 472 | |
| 473 | MixAndCompare(frames, frame_info, expected_status); |
| 474 | } |
aleloi | 623427c | 2016-12-08 02:37:58 -0800 | [diff] [blame] | 475 | |
| 476 | TEST(AudioMixer, MixingRateShouldBeDecidedByRateCalculator) { |
| 477 | constexpr int kOutputRate = 22000; |
aleloi | 087613c | 2017-02-21 08:27:08 -0800 | [diff] [blame] | 478 | const auto mixer = |
| 479 | AudioMixerImpl::Create(std::unique_ptr<OutputRateCalculator>( |
| 480 | new CustomRateCalculator(kOutputRate)), |
| 481 | true); |
aleloi | 623427c | 2016-12-08 02:37:58 -0800 | [diff] [blame] | 482 | MockMixerAudioSource audio_source; |
| 483 | mixer->AddSource(&audio_source); |
| 484 | ResetFrame(audio_source.fake_frame()); |
| 485 | |
| 486 | EXPECT_CALL(audio_source, GetAudioFrameWithInfo(kOutputRate, _)) |
| 487 | .Times(Exactly(1)); |
| 488 | |
| 489 | mixer->Mix(1, &frame_for_mixing); |
| 490 | } |
| 491 | |
| 492 | TEST(AudioMixer, ZeroSourceRateShouldBeDecidedByRateCalculator) { |
| 493 | constexpr int kOutputRate = 8000; |
aleloi | 087613c | 2017-02-21 08:27:08 -0800 | [diff] [blame] | 494 | const auto mixer = |
| 495 | AudioMixerImpl::Create(std::unique_ptr<OutputRateCalculator>( |
| 496 | new CustomRateCalculator(kOutputRate)), |
| 497 | true); |
aleloi | 623427c | 2016-12-08 02:37:58 -0800 | [diff] [blame] | 498 | |
| 499 | mixer->Mix(1, &frame_for_mixing); |
| 500 | |
| 501 | EXPECT_EQ(kOutputRate, frame_for_mixing.sample_rate_hz_); |
| 502 | } |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 503 | |
| 504 | TEST(AudioMixer, NoLimiterBasicApiCalls) { |
aleloi | 087613c | 2017-02-21 08:27:08 -0800 | [diff] [blame] | 505 | const auto mixer = AudioMixerImpl::Create( |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 506 | std::unique_ptr<OutputRateCalculator>(new DefaultOutputRateCalculator()), |
| 507 | false); |
| 508 | mixer->Mix(1, &frame_for_mixing); |
| 509 | } |
| 510 | |
| 511 | TEST(AudioMixer, AnyRateIsPossibleWithNoLimiter) { |
| 512 | // No APM limiter means no AudioProcessing::NativeRate restriction |
| 513 | // on mixing rate. The rate has to be divisible by 100 since we use |
| 514 | // 10 ms frames, though. |
| 515 | for (const auto rate : {8000, 20000, 24000, 32000, 44100}) { |
| 516 | for (const size_t number_of_channels : {1, 2}) { |
| 517 | for (const auto number_of_sources : {0, 1, 2, 3, 4}) { |
| 518 | SCOPED_TRACE( |
| 519 | ProduceDebugText(rate, number_of_sources, number_of_sources)); |
| 520 | const auto mixer = |
aleloi | 087613c | 2017-02-21 08:27:08 -0800 | [diff] [blame] | 521 | AudioMixerImpl::Create(std::unique_ptr<OutputRateCalculator>( |
| 522 | new CustomRateCalculator(rate)), |
| 523 | false); |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 524 | |
| 525 | std::vector<MockMixerAudioSource> sources(number_of_sources); |
| 526 | for (auto& source : sources) { |
Alex Loiko | b4977de | 2019-01-28 16:38:38 +0100 | [diff] [blame] | 527 | ResetFrame(source.fake_frame()); |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 528 | mixer->AddSource(&source); |
| 529 | } |
| 530 | |
| 531 | mixer->Mix(number_of_channels, &frame_for_mixing); |
| 532 | EXPECT_EQ(rate, frame_for_mixing.sample_rate_hz_); |
| 533 | EXPECT_EQ(number_of_channels, frame_for_mixing.num_channels_); |
| 534 | } |
| 535 | } |
| 536 | } |
| 537 | } |
Alex Loiko | b4977de | 2019-01-28 16:38:38 +0100 | [diff] [blame] | 538 | |
| 539 | TEST(AudioMixer, MultipleChannelsOneParticipant) { |
| 540 | // Set up a participant with a 6-channel frame, and make sure a 6-channel |
| 541 | // frame with the right sample values comes out from the mixer. There are 2 |
| 542 | // Mix calls because of ramp-up. |
| 543 | constexpr size_t kNumberOfChannels = 6; |
| 544 | MockMixerAudioSource source; |
| 545 | ResetFrame(source.fake_frame()); |
| 546 | const auto mixer = AudioMixerImpl::Create(); |
| 547 | mixer->AddSource(&source); |
| 548 | mixer->Mix(1, &frame_for_mixing); |
| 549 | auto* frame = source.fake_frame(); |
| 550 | frame->num_channels_ = kNumberOfChannels; |
| 551 | std::fill(frame->mutable_data(), |
| 552 | frame->mutable_data() + AudioFrame::kMaxDataSizeSamples, 0); |
| 553 | for (size_t i = 0; i < kNumberOfChannels; ++i) { |
| 554 | frame->mutable_data()[100 * frame->num_channels_ + i] = 1000 * i; |
| 555 | } |
| 556 | |
| 557 | mixer->Mix(kNumberOfChannels, &frame_for_mixing); |
| 558 | |
| 559 | EXPECT_EQ(frame_for_mixing.num_channels_, kNumberOfChannels); |
| 560 | for (size_t i = 0; i < kNumberOfChannels; ++i) { |
| 561 | EXPECT_EQ(frame_for_mixing.data()[100 * frame_for_mixing.num_channels_ + i], |
| 562 | static_cast<int16_t>(1000 * i)); |
| 563 | } |
| 564 | } |
| 565 | |
| 566 | TEST(AudioMixer, MultipleChannelsManyParticipants) { |
| 567 | // Sets up 2 participants. One has a 6-channel frame. Make sure a 6-channel |
| 568 | // frame with the right sample values comes out from the mixer. There are 2 |
| 569 | // Mix calls because of ramp-up. |
| 570 | constexpr size_t kNumberOfChannels = 6; |
| 571 | MockMixerAudioSource source; |
| 572 | const auto mixer = AudioMixerImpl::Create(); |
| 573 | mixer->AddSource(&source); |
| 574 | ResetFrame(source.fake_frame()); |
| 575 | mixer->Mix(1, &frame_for_mixing); |
| 576 | auto* frame = source.fake_frame(); |
| 577 | frame->num_channels_ = kNumberOfChannels; |
| 578 | std::fill(frame->mutable_data(), |
| 579 | frame->mutable_data() + AudioFrame::kMaxDataSizeSamples, 0); |
| 580 | for (size_t i = 0; i < kNumberOfChannels; ++i) { |
| 581 | frame->mutable_data()[100 * frame->num_channels_ + i] = 1000 * i; |
| 582 | } |
| 583 | MockMixerAudioSource other_source; |
| 584 | ResetFrame(other_source.fake_frame()); |
| 585 | mixer->AddSource(&other_source); |
| 586 | |
| 587 | mixer->Mix(kNumberOfChannels, &frame_for_mixing); |
| 588 | |
| 589 | EXPECT_EQ(frame_for_mixing.num_channels_, kNumberOfChannels); |
| 590 | for (size_t i = 0; i < kNumberOfChannels; ++i) { |
| 591 | EXPECT_EQ(frame_for_mixing.data()[100 * frame_for_mixing.num_channels_ + i], |
| 592 | static_cast<int16_t>(1000 * i)); |
| 593 | } |
| 594 | } |
| 595 | |
| 596 | class HighOutputRateCalculator : public OutputRateCalculator { |
| 597 | public: |
| 598 | static const int kDefaultFrequency = 76000; |
| 599 | int CalculateOutputRate( |
| 600 | const std::vector<int>& preferred_sample_rates) override { |
| 601 | return kDefaultFrequency; |
| 602 | } |
| 603 | ~HighOutputRateCalculator() override {} |
| 604 | }; |
| 605 | const int HighOutputRateCalculator::kDefaultFrequency; |
| 606 | |
| 607 | TEST(AudioMixer, MultipleChannelsAndHighRate) { |
| 608 | constexpr size_t kSamplesPerChannel = |
| 609 | HighOutputRateCalculator::kDefaultFrequency / 100; |
| 610 | // As many channels as an AudioFrame can fit: |
| 611 | constexpr size_t kNumberOfChannels = |
| 612 | AudioFrame::kMaxDataSizeSamples / kSamplesPerChannel; |
| 613 | MockMixerAudioSource source; |
| 614 | const auto mixer = AudioMixerImpl::Create( |
Mirko Bonadei | 317a1f0 | 2019-09-17 17:06:18 +0200 | [diff] [blame] | 615 | std::make_unique<HighOutputRateCalculator>(), true); |
Alex Loiko | b4977de | 2019-01-28 16:38:38 +0100 | [diff] [blame] | 616 | mixer->AddSource(&source); |
| 617 | ResetFrame(source.fake_frame()); |
| 618 | mixer->Mix(1, &frame_for_mixing); |
| 619 | auto* frame = source.fake_frame(); |
| 620 | frame->num_channels_ = kNumberOfChannels; |
| 621 | frame->sample_rate_hz_ = HighOutputRateCalculator::kDefaultFrequency; |
| 622 | frame->samples_per_channel_ = kSamplesPerChannel; |
| 623 | |
| 624 | std::fill(frame->mutable_data(), |
| 625 | frame->mutable_data() + AudioFrame::kMaxDataSizeSamples, 0); |
| 626 | MockMixerAudioSource other_source; |
| 627 | ResetFrame(other_source.fake_frame()); |
| 628 | auto* other_frame = other_source.fake_frame(); |
| 629 | other_frame->num_channels_ = kNumberOfChannels; |
| 630 | other_frame->sample_rate_hz_ = HighOutputRateCalculator::kDefaultFrequency; |
| 631 | other_frame->samples_per_channel_ = kSamplesPerChannel; |
| 632 | mixer->AddSource(&other_source); |
| 633 | |
| 634 | #if RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID) |
| 635 | EXPECT_DEATH(mixer->Mix(kNumberOfChannels, &frame_for_mixing), ""); |
| 636 | #elif !RTC_DCHECK_IS_ON |
| 637 | mixer->Mix(kNumberOfChannels, &frame_for_mixing); |
| 638 | EXPECT_EQ(frame_for_mixing.num_channels_, kNumberOfChannels); |
| 639 | EXPECT_EQ(frame_for_mixing.sample_rate_hz_, |
| 640 | HighOutputRateCalculator::kDefaultFrequency); |
| 641 | #endif |
| 642 | } |
| 643 | |
aleloi | 09f4510 | 2016-07-28 03:52:15 -0700 | [diff] [blame] | 644 | } // namespace webrtc |