1 /*
2 * Copyright (c) 2012 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 // Unit tests for Normal class.
12
13 #include "webrtc/modules/audio_coding/neteq/normal.h"
14
15 #include <vector>
16
17 #include "testing/gtest/include/gtest/gtest.h"
18 #include "webrtc/base/scoped_ptr.h"
19 #include "webrtc/common_audio/signal_processing/include/signal_processing_library.h"
20 #include "webrtc/modules/audio_coding/neteq/audio_multi_vector.h"
21 #include "webrtc/modules/audio_coding/neteq/background_noise.h"
22 #include "webrtc/modules/audio_coding/neteq/expand.h"
23 #include "webrtc/modules/audio_coding/neteq/mock/mock_decoder_database.h"
24 #include "webrtc/modules/audio_coding/neteq/mock/mock_expand.h"
25 #include "webrtc/modules/audio_coding/neteq/random_vector.h"
26 #include "webrtc/modules/audio_coding/neteq/sync_buffer.h"
27
28 using ::testing::_;
29
30 namespace webrtc {
31
TEST(Normal,CreateAndDestroy)32 TEST(Normal, CreateAndDestroy) {
33 MockDecoderDatabase db;
34 int fs = 8000;
35 size_t channels = 1;
36 BackgroundNoise bgn(channels);
37 SyncBuffer sync_buffer(1, 1000);
38 RandomVector random_vector;
39 Expand expand(&bgn, &sync_buffer, &random_vector, fs, channels);
40 Normal normal(fs, &db, bgn, &expand);
41 EXPECT_CALL(db, Die()); // Called when |db| goes out of scope.
42 }
43
TEST(Normal,AvoidDivideByZero)44 TEST(Normal, AvoidDivideByZero) {
45 WebRtcSpl_Init();
46 MockDecoderDatabase db;
47 int fs = 8000;
48 size_t channels = 1;
49 BackgroundNoise bgn(channels);
50 SyncBuffer sync_buffer(1, 1000);
51 RandomVector random_vector;
52 MockExpand expand(&bgn, &sync_buffer, &random_vector, fs, channels);
53 Normal normal(fs, &db, bgn, &expand);
54
55 int16_t input[1000] = {0};
56 rtc::scoped_ptr<int16_t[]> mute_factor_array(new int16_t[channels]);
57 for (size_t i = 0; i < channels; ++i) {
58 mute_factor_array[i] = 16384;
59 }
60 AudioMultiVector output(channels);
61
62 // Zero input length.
63 EXPECT_EQ(
64 0,
65 normal.Process(input, 0, kModeExpand, mute_factor_array.get(), &output));
66 EXPECT_EQ(0u, output.Size());
67
68 // Try to make energy_length >> scaling = 0;
69 EXPECT_CALL(expand, SetParametersForNormalAfterExpand());
70 EXPECT_CALL(expand, Process(_));
71 EXPECT_CALL(expand, Reset());
72 // If input_size_samples < 64, then energy_length in Normal::Process() will
73 // be equal to input_size_samples. Since the input is all zeros, decoded_max
74 // will be zero, and scaling will be >= 6. Thus, energy_length >> scaling = 0,
75 // and using this as a denominator would lead to problems.
76 int input_size_samples = 63;
77 EXPECT_EQ(input_size_samples,
78 normal.Process(input,
79 input_size_samples,
80 kModeExpand,
81 mute_factor_array.get(),
82 &output));
83
84 EXPECT_CALL(db, Die()); // Called when |db| goes out of scope.
85 EXPECT_CALL(expand, Die()); // Called when |expand| goes out of scope.
86 }
87
TEST(Normal,InputLengthAndChannelsDoNotMatch)88 TEST(Normal, InputLengthAndChannelsDoNotMatch) {
89 WebRtcSpl_Init();
90 MockDecoderDatabase db;
91 int fs = 8000;
92 size_t channels = 2;
93 BackgroundNoise bgn(channels);
94 SyncBuffer sync_buffer(channels, 1000);
95 RandomVector random_vector;
96 MockExpand expand(&bgn, &sync_buffer, &random_vector, fs, channels);
97 Normal normal(fs, &db, bgn, &expand);
98
99 int16_t input[1000] = {0};
100 rtc::scoped_ptr<int16_t[]> mute_factor_array(new int16_t[channels]);
101 for (size_t i = 0; i < channels; ++i) {
102 mute_factor_array[i] = 16384;
103 }
104 AudioMultiVector output(channels);
105
106 // Let the number of samples be one sample less than 80 samples per channel.
107 size_t input_len = 80 * channels - 1;
108 EXPECT_EQ(
109 0,
110 normal.Process(
111 input, input_len, kModeExpand, mute_factor_array.get(), &output));
112 EXPECT_EQ(0u, output.Size());
113
114 EXPECT_CALL(db, Die()); // Called when |db| goes out of scope.
115 EXPECT_CALL(expand, Die()); // Called when |expand| goes out of scope.
116 }
117
118 // TODO(hlundin): Write more tests.
119
120 } // namespace webrtc
121