| OLD | NEW | 
|---|
| (Empty) |  | 
|  | 1 /* | 
|  | 2  *  Copyright (c) 2015 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 <memory> | 
|  | 12 | 
|  | 13 #include "webrtc/common_audio/audio_ring_buffer.h" | 
|  | 14 | 
|  | 15 #include "testing/gtest/include/gtest/gtest.h" | 
|  | 16 #include "webrtc/common_audio/channel_buffer.h" | 
|  | 17 | 
|  | 18 namespace webrtc { | 
|  | 19 | 
|  | 20 class AudioRingBufferTest : | 
|  | 21     public ::testing::TestWithParam< ::testing::tuple<int, int, int, int> > { | 
|  | 22 }; | 
|  | 23 | 
|  | 24 void ReadAndWriteTest(const ChannelBuffer<float>& input, | 
|  | 25                       size_t num_write_chunk_frames, | 
|  | 26                       size_t num_read_chunk_frames, | 
|  | 27                       size_t buffer_frames, | 
|  | 28                       ChannelBuffer<float>* output) { | 
|  | 29   const size_t num_channels = input.num_channels(); | 
|  | 30   const size_t total_frames = input.num_frames(); | 
|  | 31   AudioRingBuffer buf(num_channels, buffer_frames); | 
|  | 32   std::unique_ptr<float* []> slice(new float*[num_channels]); | 
|  | 33 | 
|  | 34   size_t input_pos = 0; | 
|  | 35   size_t output_pos = 0; | 
|  | 36   while (input_pos + buf.WriteFramesAvailable() < total_frames) { | 
|  | 37     // Write until the buffer is as full as possible. | 
|  | 38     while (buf.WriteFramesAvailable() >= num_write_chunk_frames) { | 
|  | 39       buf.Write(input.Slice(slice.get(), input_pos), num_channels, | 
|  | 40                 num_write_chunk_frames); | 
|  | 41       input_pos += num_write_chunk_frames; | 
|  | 42     } | 
|  | 43     // Read until the buffer is as empty as possible. | 
|  | 44     while (buf.ReadFramesAvailable() >= num_read_chunk_frames) { | 
|  | 45       EXPECT_LT(output_pos, total_frames); | 
|  | 46       buf.Read(output->Slice(slice.get(), output_pos), num_channels, | 
|  | 47                num_read_chunk_frames); | 
|  | 48       output_pos += num_read_chunk_frames; | 
|  | 49     } | 
|  | 50   } | 
|  | 51 | 
|  | 52   // Write and read the last bit. | 
|  | 53   if (input_pos < total_frames) { | 
|  | 54     buf.Write(input.Slice(slice.get(), input_pos), num_channels, | 
|  | 55               total_frames - input_pos); | 
|  | 56   } | 
|  | 57   if (buf.ReadFramesAvailable()) { | 
|  | 58     buf.Read(output->Slice(slice.get(), output_pos), num_channels, | 
|  | 59              buf.ReadFramesAvailable()); | 
|  | 60   } | 
|  | 61   EXPECT_EQ(0u, buf.ReadFramesAvailable()); | 
|  | 62 } | 
|  | 63 | 
|  | 64 TEST_P(AudioRingBufferTest, ReadDataMatchesWrittenData) { | 
|  | 65   const size_t kFrames = 5000; | 
|  | 66   const size_t num_channels = ::testing::get<3>(GetParam()); | 
|  | 67 | 
|  | 68   // Initialize the input data to an increasing sequence. | 
|  | 69   ChannelBuffer<float> input(kFrames, static_cast<int>(num_channels)); | 
|  | 70   for (size_t i = 0; i < num_channels; ++i) | 
|  | 71     for (size_t j = 0; j < kFrames; ++j) | 
|  | 72       input.channels()[i][j] = (i + 1) * (j + 1); | 
|  | 73 | 
|  | 74   ChannelBuffer<float> output(kFrames, static_cast<int>(num_channels)); | 
|  | 75   ReadAndWriteTest(input, | 
|  | 76                    ::testing::get<0>(GetParam()), | 
|  | 77                    ::testing::get<1>(GetParam()), | 
|  | 78                    ::testing::get<2>(GetParam()), | 
|  | 79                    &output); | 
|  | 80 | 
|  | 81   // Verify the read data matches the input. | 
|  | 82   for (size_t i = 0; i < num_channels; ++i) | 
|  | 83     for (size_t j = 0; j < kFrames; ++j) | 
|  | 84       EXPECT_EQ(input.channels()[i][j], output.channels()[i][j]); | 
|  | 85 } | 
|  | 86 | 
|  | 87 INSTANTIATE_TEST_CASE_P( | 
|  | 88     AudioRingBufferTest, AudioRingBufferTest, | 
|  | 89     ::testing::Combine(::testing::Values(10, 20, 42),  // num_write_chunk_frames | 
|  | 90                        ::testing::Values(1, 10, 17),   // num_read_chunk_frames | 
|  | 91                        ::testing::Values(100, 256),    // buffer_frames | 
|  | 92                        ::testing::Values(1, 4)));      // num_channels | 
|  | 93 | 
|  | 94 TEST_F(AudioRingBufferTest, MoveReadPosition) { | 
|  | 95   const size_t kNumChannels = 1; | 
|  | 96   const float kInputArray[] = {1, 2, 3, 4}; | 
|  | 97   const size_t kNumFrames = sizeof(kInputArray) / sizeof(*kInputArray); | 
|  | 98   ChannelBuffer<float> input(kNumFrames, kNumChannels); | 
|  | 99   input.SetDataForTesting(kInputArray, kNumFrames); | 
|  | 100   AudioRingBuffer buf(kNumChannels, kNumFrames); | 
|  | 101   buf.Write(input.channels(), kNumChannels, kNumFrames); | 
|  | 102 | 
|  | 103   buf.MoveReadPositionForward(3); | 
|  | 104   ChannelBuffer<float> output(1, kNumChannels); | 
|  | 105   buf.Read(output.channels(), kNumChannels, 1); | 
|  | 106   EXPECT_EQ(4, output.channels()[0][0]); | 
|  | 107   buf.MoveReadPositionBackward(3); | 
|  | 108   buf.Read(output.channels(), kNumChannels, 1); | 
|  | 109   EXPECT_EQ(2, output.channels()[0][0]); | 
|  | 110 } | 
|  | 111 | 
|  | 112 }  // namespace webrtc | 
| OLD | NEW | 
|---|