MJPEG acceleration for video capture using VAAPI

content/common/gpu/client/gpu_jpeg_decode_accelerator_host.h

+// 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.
+
+#ifndef CONTENT_COMMON_GPU_CLIENT_GPU_JPEG_DECODE_ACCELERATOR_HOST_H_
+#define CONTENT_COMMON_GPU_CLIENT_GPU_JPEG_DECODE_ACCELERATOR_HOST_H_
+
+#include "base/memory/weak_ptr.h"
+#include "base/synchronization/waitable_event.h"
+#include "base/threading/non_thread_safe.h"
+#include "ipc/ipc_listener.h"
+#include "media/video/jpeg_decode_accelerator.h"
+#include "ui/gfx/geometry/size.h"
+
+namespace content {
+class GpuChannelHost;
+
+// This class is used to talk to JpegDecodeAccelerator in the GPU process
+// through IPC messages.
+class GpuJpegDecodeAcceleratorHost
+    : public IPC::Listener,
+      public media::JpegDecodeAccelerator,
+      public base::NonThreadSafe,
+      public base::SupportsWeakPtr<GpuJpegDecodeAcceleratorHost> {
+ public:
+  // |this| is guaranteed not to outlive |channel|.  (See comments for
+  // |channel_|.)
+  GpuJpegDecodeAcceleratorHost(
+      GpuChannelHost* channel,
+      int32 route_id,
+      const scoped_refptr<base::MessageLoopProxy>& io_message_loop);
+
+  // IPC::Listener implementation.
+  void OnChannelError() override;
+  bool OnMessageReceived(const IPC::Message& message) override;
+
+  // media::JpegDecodeAccelerator implementation.
+  bool Initialize(media::JpegDecodeAccelerator::Client* client) override;
+  void Decode(const media::BitstreamBuffer& bitstream_buffer,
+              const scoped_refptr<media::VideoFrame>& video_frame) override;
+  void Destroy() override;
+
+ private:
+  // Only Destroy() should be deleting |this|.
+  ~GpuJpegDecodeAcceleratorHost() override;
+
+  // Post to IO thread to notify |client_| of an error.
+  void PostNotifyError(int32_t bitstream_buffer_id, Error error);
+
+  // Disconnects channel. No more IPC messages received after this call.
+  void DisconnectChannelOnIOThread();
+
+  void Send(IPC::Message* message);
+
+  void OnVideoFrameReady(int32_t bitstream_buffer_id);
+  void OnNotifyError(int32_t bitstream_buffer_id, Error error);
+
+  // Guards |channel_|.
+  base::Lock channel_lock_;
+
+  // Unowned reference to the GpuChannelHost to send IPC messages to the GPU
+  // process.  |channel_| is always valid as long as it is not NULL.
+  GpuChannelHost* channel_;
+
+  // Used to wait on for avoiding more incoming IPC messages, before we can
+  // safely delete |this|.
+  base::WaitableEvent channel_disconnected_;
+
+  Client* client_;
+
+  // Route ID for the associated decoder in the GPU process.
+  int32 decoder_route_id_;
+
+  // IO message loop. Used for thread checking.
+  const scoped_refptr<base::MessageLoopProxy> io_message_loop_;
+
+  // WeakPtr factory for posting tasks back to itself.
+  base::WeakPtrFactory<GpuJpegDecodeAcceleratorHost> weak_this_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(GpuJpegDecodeAcceleratorHost);
+};
+
+}  // namespace content
+
+#endif  // CONTENT_COMMON_GPU_CLIENT_GPU_JPEG_DECODE_ACCELERATOR_HOST_H_