Tab Capture: Backing store readbacks to YV12 VideoFrames.

ui/surface/accelerated_surface_win.h

 // Copyright (c) 2012 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 UI_SURFACE_ACCELERATED_SURFACE_WIN_H_
 #define UI_SURFACE_ACCELERATED_SURFACE_WIN_H_
 
 #include <d3d9.h>
 
 #include "base/callback_forward.h"
 #include "base/memory/ref_counted.h"
 #include "base/single_thread_task_runner.h"
 #include "base/synchronization/lock.h"
 #include "base/synchronization/waitable_event.h"
+#include "base/time.h"
 #include "base/win/scoped_comptr.h"
 #include "ui/gfx/native_widget_types.h"
 #include "ui/gfx/size.h"
 #include "ui/surface/surface_export.h"
 
 class PresentThread;
 
 namespace gfx {
 class Rect;
 }
 
+namespace media {
+class VideoFrame;
+}
+
 class SURFACE_EXPORT AcceleratedPresenter
     : public base::RefCountedThreadSafe<AcceleratedPresenter> {
  public:
   typedef base::Callback<void(bool,
                               base::TimeTicks,
                               base::TimeDelta)> CompletionTask;
 
   explicit AcceleratedPresenter(gfx::PluginWindowHandle window);
 
   // Returns a thread safe reference to the presenter for the given window or
@@ skipping 22 matching lines @@
   void SetIsSessionLocked(bool locked);
 
   // Schedule the presenter to release its reference to the shared surface.
   void ReleaseSurface();
 
   // The public member functions are called on the main thread.
   void Present(HDC dc);
   void AsyncCopyTo(const gfx::Rect& src_subrect,
                    const gfx::Size& dst_size,
                    const base::Callback<void(bool, const SkBitmap&)>& callback);
+  void AsyncCopyToVideoFrame(
+      const gfx::Rect& src_subrect,
+      const scoped_refptr<media::VideoFrame>& target,
+      const base::Callback<void(bool)>& callback);
   void Invalidate();
 
 #if defined(USE_AURA)
   // TODO(scottmg): This is a temporary hack until we have a two-worlds ash/aura
   // separation.
   void SetNewTargetWindow(gfx::PluginWindowHandle window);
 #endif
 
  private:
   friend class base::RefCountedThreadSafe<AcceleratedPresenter>;
 
   ~AcceleratedPresenter();
 
   // These member functions are called on the PresentThread with which the
   // presenter has affinity.
   void DoPresentAndAcknowledge(
       const gfx::Size& size,
       int64 surface_handle,
       const CompletionTask& completion_task);
   void DoSuspend();
   void DoPresent(const base::Closure& composite_task);
   void DoReleaseSurface();
   void DoCopyToAndAcknowledge(
       const gfx::Rect& src_subrect,
       const gfx::Size& dst_size,
       scoped_refptr<base::SingleThreadTaskRunner> callback_runner,
       const base::Callback<void(bool, const SkBitmap&)>& callback);
-  bool DoCopyTo(const gfx::Rect& src_subrect,
-                const gfx::Size& dst_size,
-                SkBitmap* bitmap);
+  void DoCopyToVideoFrameAndAcknowledge(
+      const gfx::Rect& src_subrect,
+      const scoped_refptr<media::VideoFrame>& target,
+      const scoped_refptr<base::SingleThreadTaskRunner>& callback_runner,
+      const base::Callback<void(bool)>& callback);
+  bool DoCopyToYUV(const gfx::Rect& src_subrect,
+                   const scoped_refptr<media::VideoFrame>& frame);
+  bool DoCopyToARGB(const gfx::Rect& src_subrect,
+                    const gfx::Size& dst_size,
+                    SkBitmap* bitmap);
 
   void PresentWithGDI(HDC dc);
   gfx::Size GetWindowSize();
 
   // This function tries to guess whether Direct3D will be able to reliably
   // present to the window. When the window is resizing, presenting with
   // Direct3D causes other regions of the window rendered with GDI to
   // flicker transparent / non-transparent.
   bool CheckDirect3DWillWork();
 
@@ skipping 50 matching lines @@
 };
 
 class SURFACE_EXPORT AcceleratedSurface {
  public:
   AcceleratedSurface(gfx::PluginWindowHandle window);
   ~AcceleratedSurface();
 
   // Synchronously present a frame with no acknowledgement.
   void Present(HDC dc);
 
-  // Copies the surface data to |buf|. The copied region is specified with
-  // |src_subrect| and the image data is transformed so that it fits in
-  // |dst_size|.
-  // Caller must ensure that |buf| is allocated with the size no less than
-  // |4 * dst_size.width() * dst_size.height()| bytes.
+  // Transfer the contents of the surface to an SkBitmap, and invoke a callback
+  // with the result.
   void AsyncCopyTo(const gfx::Rect& src_subrect,
                    const gfx::Size& dst_size,
                    const base::Callback<void(bool, const SkBitmap&)>& callback);
 
+  // Transfer the contents of the surface to an already-allocated YV12
+  // VideoFrame, and invoke a callback to indicate success or failure.
+  void AsyncCopyToVideoFrame(
+      const gfx::Rect& src_subrect,
+      const scoped_refptr<media::VideoFrame>& target,
+      const base::Callback<void(bool)>& callback);
+
   // Temporarily release resources until a new surface is asynchronously
   // presented. Present will not be able to represent the last surface after
   // calling this and will return false.
   void Suspend();
 
   // Indicates that the surface has become invisible.
   void WasHidden();
 
   // Called when the Windows session in locked or unlocked.
   void SetIsSessionLocked(bool locked);
 
  private:
   const scoped_refptr<AcceleratedPresenter> presenter_;
   DISALLOW_COPY_AND_ASSIGN(AcceleratedSurface);
 };
 
 #endif  // UI_SURFACE_ACCELERATED_SURFACE_WIN_H_