Add content::SurfaceCapturer

content/browser/aura/browser_compositor_output_surface.cc

 // Copyright (c) 2013 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/aura/browser_compositor_output_surface.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/location.h"
 #include "base/message_loop/message_loop_proxy.h"
 #include "base/strings/string_number_conversions.h"
 #include "cc/output/compositor_frame.h"
+#include "content/browser/aura/browser_compositor_output_surface_capturer.h"
 #include "content/browser/aura/reflector_impl.h"
 #include "content/common/gpu/client/webgraphicscontext3d_command_buffer_impl.h"
 #include "ui/compositor/compositor.h"
 #include "ui/compositor/compositor_switches.h"
 
 namespace content {
 
 BrowserCompositorOutputSurface::BrowserCompositorOutputSurface(
     scoped_ptr<WebKit::WebGraphicsContext3D> context,
     int surface_id,
     IDMap<BrowserCompositorOutputSurface>* output_surface_map,
-    base::MessageLoopProxy* compositor_message_loop,
+    const scoped_refptr<base::MessageLoopProxy>& compositor_message_loop,
     base::WeakPtr<ui::Compositor> compositor)
     : OutputSurface(context.Pass()),
       surface_id_(surface_id),
       output_surface_map_(output_surface_map),
       compositor_message_loop_(compositor_message_loop),
       compositor_(compositor) {
   CommandLine* command_line = CommandLine::ForCurrentProcess();
   if (command_line->HasSwitch(switches::kUIMaxFramesPending)) {
     std::string string_value = command_line->GetSwitchValueASCII(
         switches::kUIMaxFramesPending);
     int int_value;
     if (base::StringToInt(string_value, &int_value))
       capabilities_.max_frames_pending = int_value;
     else
       LOG(ERROR) << "Trouble parsing --" << switches::kUIMaxFramesPending;
   }
   capabilities_.adjust_deadline_for_parent = false;
   DetachFromThread();
 }
 
 BrowserCompositorOutputSurface::~BrowserCompositorOutputSurface() {
   DCHECK(CalledOnValidThread());
   if (!HasClient())
     return;
   output_surface_map_->Remove(surface_id_);
+  // Don't delete the actual object; we don't own it.
+  FOR_EACH_OBSERVER(BrowserCompositorOutputSurfaceCapturer,
+                    capturer_list_,
+                    DoDestroyOnImplThread());
 }
 
 bool BrowserCompositorOutputSurface::BindToClient(
     cc::OutputSurfaceClient* client) {
   DCHECK(CalledOnValidThread());
 
   if (!OutputSurface::BindToClient(client))
     return false;
 
   output_surface_map_->AddWithID(this, surface_id_);
   return true;
 }
 
 void BrowserCompositorOutputSurface::Reshape(gfx::Size size,
                                              float scale_factor) {
   OutputSurface::Reshape(size, scale_factor);
   if (reflector_.get())
     reflector_->OnReshape(size);
 }
 
 void BrowserCompositorOutputSurface::SwapBuffers(cc::CompositorFrame* frame) {
   DCHECK(frame->gl_frame_data);
 
-  WebGraphicsContext3DCommandBufferImpl* command_buffer =
-      static_cast<WebGraphicsContext3DCommandBufferImpl*>(context3d());
   CommandBufferProxyImpl* command_buffer_proxy =
-      command_buffer->GetCommandBufferProxy();
+      GetContext3DCommandBufferImpl()->GetCommandBufferProxy();
   DCHECK(command_buffer_proxy);
   context3d()->shallowFlushCHROMIUM();
   command_buffer_proxy->SetLatencyInfo(frame->metadata.latency_info);
 
+  FOR_EACH_OBSERVER(BrowserCompositorOutputSurfaceCapturer,
+                    capturer_list_,
+                    NotifySwapBuffersOnImplThread());
+
   if (reflector_.get()) {
     if (frame->gl_frame_data->sub_buffer_rect ==
         gfx::Rect(frame->gl_frame_data->size))
       reflector_->OnSwapBuffers();
     else
       reflector_->OnPostSubBuffer(frame->gl_frame_data->sub_buffer_rect);
   }
 
   OutputSurface::SwapBuffers(frame);
 }
 
 void BrowserCompositorOutputSurface::OnUpdateVSyncParameters(
     base::TimeTicks timebase,
     base::TimeDelta interval) {
   DCHECK(CalledOnValidThread());
   DCHECK(HasClient());
   OnVSyncParametersChanged(timebase, interval);
   compositor_message_loop_->PostTask(
       FROM_HERE,
       base::Bind(&ui::Compositor::OnUpdateVSyncParameters,
                  compositor_, timebase, interval));
 }
 
 void BrowserCompositorOutputSurface::SetReflector(ReflectorImpl* reflector) {
   reflector_ = reflector;
 }
 
+void BrowserCompositorOutputSurface::AddCapturer(
+    BrowserCompositorOutputSurfaceCapturer* capturer) {
+  DCHECK(CalledOnValidThread());
+  capturer_list_.AddObserver(capturer);
+}
+
+void BrowserCompositorOutputSurface::RemoveCapturer(
+    BrowserCompositorOutputSurfaceCapturer* capturer) {
+  DCHECK(CalledOnValidThread());
+  capturer_list_.RemoveObserver(capturer);
+}
+
+WebGraphicsContext3DCommandBufferImpl*
+BrowserCompositorOutputSurface::GetContext3DCommandBufferImpl() {
+  return static_cast<WebGraphicsContext3DCommandBufferImpl*>(context3d());
+}
+
 }  // namespace content