Chromecast: adds interfaces for hardware layer of CMA pipeline.

chromecast/media/cma/test/media_component_device_feeder_for_test.cc

+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chromecast/media/cma/test/media_component_device_feeder_for_test.h"
+
+#include <list>
+#include <vector>
+
+#include "base/basictypes.h"
+#include "base/bind.h"
+#include "base/logging.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/message_loop/message_loop.h"
+#include "base/message_loop/message_loop_proxy.h"
+#include "base/time/time.h"
+#include "chromecast/media/base/decrypt_context.h"
+#include "chromecast/media/cma/backend/audio_pipeline_device.h"
+#include "chromecast/media/cma/backend/media_clock_device.h"
+#include "chromecast/media/cma/backend/media_pipeline_device.h"
+#include "chromecast/media/cma/backend/video_pipeline_device.h"
+#include "chromecast/media/cma/base/decoder_buffer_adapter.h"
+#include "chromecast/media/cma/base/decoder_buffer_base.h"
+#include "chromecast/media/cma/test/frame_segmenter_for_test.h"
+#include "media/base/audio_decoder_config.h"
+#include "media/base/buffers.h"
+#include "media/base/decoder_buffer.h"
+#include "media/base/video_decoder_config.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace chromecast {
+namespace media {
+
+MediaComponentDeviceFeederForTest::MediaComponentDeviceFeederForTest(
+    MediaComponentDevice *device,
+    const FrameList& frames,
+    base::Callback<void(MediaComponentDeviceFeederForTest *)> eos_cb)
+    : media_component_device_(device),
+      rendering_frame_idx_(1),
+      clock_frame_idx_(1),
+      eos_cb_(eos_cb),
+      feeding_completed_(false),
+      delayed_feed_pattern_() {
+  // Copy pointers over, claiming ownership on buffers
+  frames_ = frames;
+}
+
+MediaComponentDeviceFeederForTest::~MediaComponentDeviceFeederForTest() {
+}
+
+// Accessors
+void MediaComponentDeviceFeederForTest::SetDelayedFeedPattern(
+    const std::vector<base::TimeDelta>& pattern) {
+  // Copy pattern over
+  delayed_feed_pattern_ = pattern;
+}
+
+void MediaComponentDeviceFeederForTest::Initialize() {
+  MediaComponentDevice::Client client;
+  client.eos_cb =
+      base::Bind(&MediaComponentDeviceFeederForTest::OnEos,
+                 base::Unretained(this));
+  media_component_device_->SetClient(client);
+
+  bool success =
+      media_component_device_->SetState(MediaComponentDevice::kStateIdle);
+  ASSERT_TRUE(success);
+  success = media_component_device_->SetStartPts(base::TimeDelta());
+  ASSERT_TRUE(success);
+  success =
+      media_component_device_->SetState(MediaComponentDevice::kStatePaused);
+  ASSERT_TRUE(success);
+}
+
+void MediaComponentDeviceFeederForTest::Feed() {
+  // Start rendering if needed.
+  if (rendering_frame_idx_ == 0) {
+    media_component_device_->SetState(MediaComponentDevice::kStateRunning);
+  } else {
+    rendering_frame_idx_--;
+  }
+
+  // Possibly feed one frame
+  DCHECK(!frames_.empty());
+  scoped_refptr<DecoderBufferBase> buffer = frames_.front();
+
+  MediaComponentDevice::FrameStatus status =
+      media_component_device_->PushFrame(
+          scoped_refptr<DecryptContext>(),
+          buffer,
+          base::Bind(&MediaComponentDeviceFeederForTest::OnFramePushed,
+                     base::Unretained(this)));
+  EXPECT_NE(status, MediaComponentDevice::kFrameFailed);
+  frames_.pop_front();
+
+  // Feeding is done, just wait for the end of stream callback.
+  if (buffer->end_of_stream() || frames_.empty()) {
+    if (frames_.empty() && !buffer->end_of_stream()) {
+      LOG(WARNING) << "Stream emptied without feeding EOS frame";
+    }
+
+    feeding_completed_ = true;
+    return;
+  }
+
+  if (status == MediaComponentDevice::kFramePending)
+    return;
+
+  OnFramePushed(MediaComponentDevice::kFrameSuccess);
+}
+
+void MediaComponentDeviceFeederForTest::OnFramePushed(
+    MediaComponentDevice::FrameStatus status) {
+  EXPECT_NE(status, MediaComponentDevice::kFrameFailed);
+  if (feeding_completed_)
+    return;
+
+  // Delayed feed setup
+  base::TimeDelta feed_delay = base::TimeDelta();
+  if (!delayed_feed_pattern_.empty()) {
+    feed_delay = delayed_feed_pattern_[delayed_feed_pattern_idx_];
+    delayed_feed_pattern_idx_ =
+        (delayed_feed_pattern_idx_ + 1) % delayed_feed_pattern_.size();
+  }
+
+  if (feed_delay != base::TimeDelta()) {
+    // Wait feed_delay
+    base::MessageLoopProxy::current()->PostDelayedTask(
+        FROM_HERE,
+        base::Bind(&MediaComponentDeviceFeederForTest::Feed,
+                   base::Unretained(this)),
+        feed_delay);
+  } else {
+    // Wait time is 0. Append to task queue
+    base::MessageLoopProxy::current()->PostTask(
+        FROM_HERE,
+        base::Bind(&MediaComponentDeviceFeederForTest::Feed,
+                   base::Unretained(this)));
+  }
+}
+
+void MediaComponentDeviceFeederForTest::OnEos() {
+  // Shutdown the pipeline.
+  bool success = media_component_device_->SetState(
+      MediaComponentDevice::kStateIdle);
+  ASSERT_TRUE(success);
+  success = media_component_device_->SetState(
+      MediaComponentDevice::kStateUninitialized);
+  ASSERT_TRUE(success);
+
+  if (!eos_cb_.is_null()) {
+    eos_cb_.Run(this);
+  }
+}
+
+}  // namespace media
+}  // namespace chromecast