MJPEG acceleration for video capture using VAAPI

content/browser/renderer_host/gpu_jpeg_decoder.cc

Index: content/browser/renderer_host/gpu_jpeg_decoder.cc
diff --git a/content/browser/renderer_host/gpu_jpeg_decoder.cc b/content/browser/renderer_host/gpu_jpeg_decoder.cc
new file mode 100644
index 0000000000000000000000000000000000000000..6893a560d60e5b380a0745d2b3e8f904d7a130ad
--- /dev/null
+++ b/content/browser/renderer_host/gpu_jpeg_decoder.cc
@@ -0,0 +1,214 @@
+// Copyright 2015 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 "content/browser/renderer_host/gpu_jpeg_decoder.h"
+
+#include "base/bind.h"
+#include "base/logging.h"
+#include "base/strings/stringprintf.h"
+#include "base/trace_event/trace_event.h"
+#include "content/browser/gpu/browser_gpu_channel_host_factory.h"
+#include "content/common/gpu/client/gpu_jpeg_decode_accelerator_host.h"
+#include "content/public/browser/browser_thread.h"
+#include "media/base/video_frame.h"
+
+namespace content {
+
+// static
+bool GpuJpegDecoder::Supported() {
+// A lightweight check for caller to avoid IPC latency for known unsupported
+// platform. Initialize() can do the real platform supporting check but it
+// requires an IPC even for platforms that do not support HW decoder.
+// TODO(kcwu): move this information to GpuInfo.
+#if defined(OS_CHROMEOS) && defined(ARCH_CPU_X86_FAMILY)
+  return true;
+#endif
+  return false;
+}
+
+GpuJpegDecoder::GpuJpegDecoder(const DecodeDoneCB& decode_done_cb,
+                               const ErrorCB& error_cb)
+    : jpeg_thread_("GpuJpegDecoderThread"),
+      create_event_(false, false),
+      decode_event_(false, false),
+      decode_done_cb_(decode_done_cb),
+      error_cb_(error_cb),
+      next_bitstream_buffer_id_(0),
+      in_buffer_(media::JpegDecodeAccelerator::kInvalidBitstreamBufferId,
+                 base::SharedMemory::NULLHandle(),
+                 0) {
+}
+
+GpuJpegDecoder::~GpuJpegDecoder() {
+  DCHECK(CalledOnValidThread());

[wuchengli, 2015/05/13 15:04:05]

We need a GpuJpegDecoder::Destroy to stop jpeg thread and destroy |decoder_|. So
we don't access the member variables from VideoFrameReady when |this| is being
destructed. GpuJpegDecoder::Destroy should be called in
~VideoCaptureDeviceClient().

VideoCaptureDeviceClient owns GpuJpegDecoder. So the lifecycle should be the
same. When VideoCaptureDeviceClient is gone, GpuJpegDecoder shouldn't be alive.
We don't need to use WeakPtr for VideoCaptureDeviceClient. Using Unretained is
enough.

[kcwu, 2015/05/26 07:04:19]

Acknowledged.
piman raised the same issue in
https://codereview.chromium.org/1124423008/diff/1/content/common/gpu/client/g...
I will address it there.

+}
+
+bool GpuJpegDecoder::IsDecoding() {
+  DCHECK(CalledOnValidThread());

[wuchengli, 2015/05/13 15:04:05]

DCHECK(jpeg_thread_proxy_->BelongsToCurrentThread());

[kcwu, 2015/05/25 18:20:34]

Done.

+  return !decode_done_closure_.is_null();
+}
+
+bool GpuJpegDecoder::Initialize() {
+  DVLOG(3) << __func__;
+  DCHECK(CalledOnValidThread());
+  DCHECK(!jpeg_thread_.IsRunning());
+
+  if (!jpeg_thread_.Start()) {
+    LOG(ERROR) << "Failed to start decoding thread.";
+    return false;
+  }
+  jpeg_thread_proxy_ = jpeg_thread_.message_loop_proxy();
+
+  jpeg_thread_proxy_->PostTask(
+      FROM_HERE,
+      base::Bind(&GpuJpegDecoder::CreateDecoderOnJpegThread,
+                 base::Unretained(this)));
+  create_event_.Wait();
+
+  if (!decoder_) {
+    jpeg_thread_.Stop();
+    return false;
+  }
+
+  return true;
+}
+
+void GpuJpegDecoder::CreateDecoderOnJpegThread() {
+  DCHECK(jpeg_thread_proxy_->BelongsToCurrentThread());
+  GpuChannelHost* const host =
+      BrowserGpuChannelHostFactory::instance()->GetGpuChannel();
+  decoder_ = host->CreateJpegDecoder();
+  if (!decoder_->Initialize(this)) {
+    decoder_.reset();
+  }
+  create_event_.Signal();
+}
+
+void GpuJpegDecoder::VideoFrameReady(int32_t bitstream_buffer_id) {
+  DVLOG(3) << __func__;
+  DCHECK(jpeg_thread_proxy_->BelongsToCurrentThread());
+  TRACE_EVENT0("jpeg", "GpuJpegDecoder::VideoFrameReady");
+
+  if (bitstream_buffer_id ==
+      media::JpegDecodeAccelerator::kInvalidBitstreamBufferId)

[wuchengli, 2015/05/13 15:04:05]

When will this happen? If this is an error, remove this and just use line 102.

[kcwu, 2015/05/25 18:20:34]

Done.

+    return;
+
+  if (!IsDecoding()) {
+    LOG(ERROR) << "Got decode response while not decoding";
+    return;
+  }
+
+  if (bitstream_buffer_id != in_buffer_.id()) {
+    LOG(ERROR) << "Unexpected bitstream_buffer_id " << bitstream_buffer_id
+               << ", expected " << in_buffer_.id();
+    return;
+  }
+
+  decode_done_closure_.Run();
+  decode_done_closure_.Reset();
+
+  TRACE_EVENT_ASYNC_END0("jpeg", "GpuJpegDecoder decoding",
+                         bitstream_buffer_id);
+}
+
+void GpuJpegDecoder::NotifyError(int32_t bitstream_buffer_id,
+                                 media::JpegDecodeAccelerator::Error error) {
+  DVLOG(3) << __func__;
+  DCHECK(jpeg_thread_proxy_->BelongsToCurrentThread());
+  LOG(ERROR) << "Decode error, bitstream_buffer_id=" << bitstream_buffer_id;
+
+  decode_done_closure_.Reset();
+
+  error_cb_.Run(
+      base::StringPrintf("Decode error, bitstream_buffer_id=%d, error=%d",
+                         bitstream_buffer_id, error));
+}
+
+void GpuJpegDecoder::DecodeCapturedData(
+    const uint8* data,
+    size_t in_buffer_size,
+    const media::VideoCaptureFormat& frame_format,
+    const base::TimeTicks& timestamp,
+    scoped_ptr<media::VideoCaptureDevice::Client::Buffer> out_buffer) {
+  DCHECK(CalledOnValidThread());
+
+  jpeg_thread_proxy_->PostTask(
+      FROM_HERE,
+      base::Bind(
+          &GpuJpegDecoder::DecodeCapturedDataOnJpegThread,
+          base::Unretained(this),
+          data,
+          in_buffer_size,
+          frame_format,
+          timestamp,
+          base::Passed(&out_buffer)));
+
+  // Wait until |data| is copied out.
+  decode_event_.Wait();
+}
+
+
+void GpuJpegDecoder::DecodeCapturedDataOnJpegThread(
+    const uint8* data,
+    size_t in_buffer_size,
+    const media::VideoCaptureFormat& frame_format,
+    const base::TimeTicks& timestamp,
+    scoped_ptr<media::VideoCaptureDevice::Client::Buffer> out_buffer) {
+  DCHECK(jpeg_thread_proxy_->BelongsToCurrentThread());
+  TRACE_EVENT_ASYNC_BEGIN0("jpeg", "GpuJpegDecoder decoding",
+                           next_bitstream_buffer_id_);
+  TRACE_EVENT0("jpeg", "GpuJpegDecoder::DecodeCapturedDataOnJpegThread");
+
+  // TODO(kcwu): enqueue decode requests in case decoding is not fast enough
+  // (say, if decoding time is longer than 16ms for 60fps 4k image)
+  if (IsDecoding()) {
+    DVLOG(1) << "Drop captured frame. Previous jpeg frame is still decoding";
+    decode_event_.Signal();
+    return;
+  }
+
+  // Enlarge input buffer if necessary.
+  if (!in_shared_memory_.get() ||
+      in_buffer_size > in_shared_memory_->mapped_size()) {
+    // Reserve 2x space to avoid frequently reallocate for initial frames.
+    size_t reserved_size = 2 * in_buffer_size;
+    in_shared_memory_.reset(new base::SharedMemory);
+    if (!in_shared_memory_->CreateAndMapAnonymous(reserved_size)) {
+      error_cb_.Run(base::StringPrintf("CreateAndMapAnonymous failed, size=%ld",

[henryhsu, 2015/05/12 09:42:03]

There is a warning in arm platform. Please change %ld to %d.

[kcwu, 2015/05/25 18:20:34]

Done.

+                                       reserved_size));
+      decode_event_.Signal();
+      return;
+    }
+  }
+  memcpy(in_shared_memory_->memory(), data, in_buffer_size);
+  decode_event_.Signal();
+
+  in_buffer_ = media::BitstreamBuffer(
+      next_bitstream_buffer_id_, in_shared_memory_->handle(), in_buffer_size);
+  // Mask against 30 bits, to avoid (undefined) wraparound on signed integer.
+  next_bitstream_buffer_id_ = (next_bitstream_buffer_id_ + 1) & 0x3FFFFFFF;
+
+  const gfx::Size dimensions = frame_format.frame_size;
+  scoped_refptr<media::VideoFrame> out_frame =
+      media::VideoFrame::WrapExternalPackedMemory(
+          media::VideoFrame::I420,
+          dimensions,
+          gfx::Rect(dimensions),
+          dimensions,
+          reinterpret_cast<uint8*>(out_buffer->data()),
+          out_buffer->size(),
+          base::SharedMemoryHandle(out_buffer->AsPlatformHandle(), true),
+          0,
+          base::TimeDelta(),
+          base::Closure());
+  DCHECK(out_frame.get());
+  out_frame->metadata()->SetDouble(media::VideoFrameMetadata::FRAME_RATE,
+                                   frame_format.frame_rate);
+
+  decode_done_closure_ = base::Bind(decode_done_cb_, base::Passed(&out_buffer),
+                                    out_frame, timestamp);
+  decoder_->Decode(in_buffer_, out_frame);
+}
+
+}  // namespace content