OLD | NEW |
| (Empty) |
1 /* | |
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 "webrtc/modules/audio_device/android/fine_audio_buffer.h" | |
12 | |
13 #include <limits.h> | |
14 #include <memory> | |
15 | |
16 #include "testing/gmock/include/gmock/gmock.h" | |
17 #include "testing/gtest/include/gtest/gtest.h" | |
18 #include "webrtc/base/scoped_ptr.h" | |
19 #include "webrtc/modules/audio_device/mock_audio_device_buffer.h" | |
20 | |
21 using ::testing::_; | |
22 using ::testing::InSequence; | |
23 using ::testing::Return; | |
24 | |
25 namespace webrtc { | |
26 | |
27 // The fake audio data is 0,1,..SCHAR_MAX-1,0,1,... This is to make it easy | |
28 // to detect errors. This function verifies that the buffers contain such data. | |
29 // E.g. if there are two buffers of size 3, buffer 1 would contain 0,1,2 and | |
30 // buffer 2 would contain 3,4,5. Note that SCHAR_MAX is 127 so wrap-around | |
31 // will happen. | |
32 // |buffer| is the audio buffer to verify. | |
33 bool VerifyBuffer(const int8_t* buffer, int buffer_number, int size) { | |
34 int start_value = (buffer_number * size) % SCHAR_MAX; | |
35 for (int i = 0; i < size; ++i) { | |
36 if (buffer[i] != (i + start_value) % SCHAR_MAX) { | |
37 return false; | |
38 } | |
39 } | |
40 return true; | |
41 } | |
42 | |
43 // This function replaces GetPlayoutData when it's called (which is done | |
44 // implicitly when calling GetBufferData). It writes the sequence | |
45 // 0,1,..SCHAR_MAX-1,0,1,... to the buffer. Note that this is likely a buffer of | |
46 // different size than the one VerifyBuffer verifies. | |
47 // |iteration| is the number of calls made to UpdateBuffer prior to this call. | |
48 // |samples_per_10_ms| is the number of samples that should be written to the | |
49 // buffer (|arg0|). | |
50 ACTION_P2(UpdateBuffer, iteration, samples_per_10_ms) { | |
51 int8_t* buffer = static_cast<int8_t*>(arg0); | |
52 int bytes_per_10_ms = samples_per_10_ms * static_cast<int>(sizeof(int16_t)); | |
53 int start_value = (iteration * bytes_per_10_ms) % SCHAR_MAX; | |
54 for (int i = 0; i < bytes_per_10_ms; ++i) { | |
55 buffer[i] = (i + start_value) % SCHAR_MAX; | |
56 } | |
57 return samples_per_10_ms; | |
58 } | |
59 | |
60 void RunFineBufferTest(int sample_rate, int frame_size_in_samples) { | |
61 const int kSamplesPer10Ms = sample_rate * 10 / 1000; | |
62 const int kFrameSizeBytes = frame_size_in_samples * | |
63 static_cast<int>(sizeof(int16_t)); | |
64 const int kNumberOfFrames = 5; | |
65 // Ceiling of integer division: 1 + ((x - 1) / y) | |
66 const int kNumberOfUpdateBufferCalls = | |
67 1 + ((kNumberOfFrames * frame_size_in_samples - 1) / kSamplesPer10Ms); | |
68 | |
69 MockAudioDeviceBuffer audio_device_buffer; | |
70 EXPECT_CALL(audio_device_buffer, RequestPlayoutData(_)) | |
71 .WillRepeatedly(Return(kSamplesPer10Ms)); | |
72 { | |
73 InSequence s; | |
74 for (int i = 0; i < kNumberOfUpdateBufferCalls; ++i) { | |
75 EXPECT_CALL(audio_device_buffer, GetPlayoutData(_)) | |
76 .WillOnce(UpdateBuffer(i, kSamplesPer10Ms)) | |
77 .RetiresOnSaturation(); | |
78 } | |
79 } | |
80 FineAudioBuffer fine_buffer(&audio_device_buffer, kFrameSizeBytes, | |
81 sample_rate); | |
82 | |
83 rtc::scoped_ptr<int8_t[]> out_buffer; | |
84 out_buffer.reset( | |
85 new int8_t[fine_buffer.RequiredBufferSizeBytes()]); | |
86 for (int i = 0; i < kNumberOfFrames; ++i) { | |
87 fine_buffer.GetBufferData(out_buffer.get()); | |
88 EXPECT_TRUE(VerifyBuffer(out_buffer.get(), i, kFrameSizeBytes)); | |
89 } | |
90 } | |
91 | |
92 TEST(FineBufferTest, BufferLessThan10ms) { | |
93 const int kSampleRate = 44100; | |
94 const int kSamplesPer10Ms = kSampleRate * 10 / 1000; | |
95 const int kFrameSizeSamples = kSamplesPer10Ms - 50; | |
96 RunFineBufferTest(kSampleRate, kFrameSizeSamples); | |
97 } | |
98 | |
99 TEST(FineBufferTest, GreaterThan10ms) { | |
100 const int kSampleRate = 44100; | |
101 const int kSamplesPer10Ms = kSampleRate * 10 / 1000; | |
102 const int kFrameSizeSamples = kSamplesPer10Ms + 50; | |
103 RunFineBufferTest(kSampleRate, kFrameSizeSamples); | |
104 } | |
105 | |
106 } // namespace webrtc | |
OLD | NEW |