1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "base/bind.h"
6#include "base/bind_helpers.h"
7#include "base/strings/stringprintf.h"
8#include "media/base/audio_bus.h"
9#include "media/base/audio_pull_fifo.h"
10#include "testing/gtest/include/gtest/gtest.h"
11
12namespace media {
13
14// Block diagram of a possible real-world usage:
15//
16//       | Producer | ----> | AudioPullFifo | ----> | Consumer |
17//                    push                    pull
18//          2048      ---->       (2048)      ---->     ~512
19
20// Number of channels in each audio bus.
21static int kChannels = 2;
22
23// Max number of audio framed the FIFO can contain.
24static const int kMaxFramesInFifo = 2048;
25
26class AudioPullFifoTest
27    : public testing::TestWithParam<int> {
28 public:
29  AudioPullFifoTest()
30    : pull_fifo_(kChannels, kMaxFramesInFifo, base::Bind(
31          &AudioPullFifoTest::ProvideInput, base::Unretained(this))),
32      audio_bus_(AudioBus::Create(kChannels, kMaxFramesInFifo)),
33      fill_value_(0),
34      last_frame_delay_(-1) {}
35  virtual ~AudioPullFifoTest() {}
36
37  void VerifyValue(const float data[], int size, float start_value) {
38    float value = start_value;
39    for (int i = 0; i < size; ++i) {
40      ASSERT_FLOAT_EQ(value++, data[i]) << "i=" << i;
41    }
42  }
43
44  // Consume data using different sizes, acquire audio frames from the FIFO
45  // and verify that the retrieved values matches the values written by the
46  // producer.
47  void ConsumeTest(int frames_to_consume) {
48    int start_value = 0;
49    SCOPED_TRACE(base::StringPrintf("Checking frames_to_consume %d",
50                 frames_to_consume));
51    pull_fifo_.Consume(audio_bus_.get(), frames_to_consume);
52    for (int j = 0; j < kChannels; ++j) {
53      VerifyValue(audio_bus_->channel(j), frames_to_consume, start_value);
54    }
55    start_value += frames_to_consume;
56    EXPECT_LT(last_frame_delay_, audio_bus_->frames());
57  }
58
59  // AudioPullFifo::ReadCB implementation where we increase a value for each
60  // audio frame that we provide. Note that all channels are given the same
61  // value to simplify the verification.
62  virtual void ProvideInput(int frame_delay, AudioBus* audio_bus) {
63    ASSERT_GT(frame_delay, last_frame_delay_);
64    last_frame_delay_ = frame_delay;
65
66    EXPECT_EQ(audio_bus->channels(), audio_bus_->channels());
67    EXPECT_EQ(audio_bus->frames(), kMaxFramesInFifo);
68    for (int i = 0; i < audio_bus->frames(); ++i) {
69      for (int j = 0; j < audio_bus->channels(); ++j) {
70        // Store same value in all channels.
71        audio_bus->channel(j)[i] = fill_value_;
72      }
73      fill_value_++;
74    }
75  }
76
77 protected:
78  AudioPullFifo pull_fifo_;
79  scoped_ptr<AudioBus> audio_bus_;
80  int fill_value_;
81  int last_frame_delay_;
82
83  DISALLOW_COPY_AND_ASSIGN(AudioPullFifoTest);
84};
85
86TEST_P(AudioPullFifoTest, Consume) {
87  ConsumeTest(GetParam());
88}
89
90// Test common |frames_to_consume| values which will be used as input
91// parameter to AudioPullFifo::Consume() when the consumer asks for data.
92INSTANTIATE_TEST_CASE_P(
93    AudioPullFifoTest, AudioPullFifoTest,
94    testing::Values(544, 512, 512, 512, 512, 2048, 544, 441, 440, 433, 500));
95
96}  // namespace media
97