Create manager to track OffscreenCanvasSurfaceImpl instances

content/browser/renderer_host/offscreen_canvas_compositor_frame_sink.cc

 // Copyright 2016 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/offscreen_canvas_compositor_frame_sink.h"
 
 #include "cc/surfaces/surface.h"
 #include "cc/surfaces/surface_manager.h"
 #include "content/browser/compositor/surface_utils.h"
+#include "content/browser/renderer_host/offscreen_canvas_surface_manager.h"
 #include "mojo/public/cpp/bindings/strong_binding.h"
 
 namespace content {
 
 OffscreenCanvasCompositorFrameSink::OffscreenCanvasCompositorFrameSink(
     const cc::SurfaceId& surface_id,
     cc::mojom::MojoCompositorFrameSinkClientPtr client)
-    : surface_id_(surface_id), client_(std::move(client)) {
+    : surface_id_(surface_id), client_(std::move(client)), weak_factory_(this) {

[Fady Samuel, 2016/11/11 23:33:16]

Don't store this SurfaceId maybe?

Store a const FrameSinkId. and store the LocalFrameId portion in a LocalFrameId
member variable.

[xlai (Olivia), 2016/11/14 15:13:19]

I'm going to do it in the next CL. Because right now the SubmitCompositorFrame
IPC does not submit the SurfaceId together with a frame, we still rely on a
stored surface Id to know where to send the frame.

   cc::SurfaceManager* manager = GetSurfaceManager();
   surface_factory_ = base::MakeUnique<cc::SurfaceFactory>(
       surface_id_.frame_sink_id(), manager, this);
   manager->RegisterFrameSinkId(surface_id_.frame_sink_id());
   surface_factory_->Create(surface_id_.local_frame_id());
+
+  OffscreenCanvasSurfaceManager::GetInstance()
+      ->RegisterCompositorFrameSinkInstance(surface_id_.frame_sink_id(), this);
 }
 
 OffscreenCanvasCompositorFrameSink::~OffscreenCanvasCompositorFrameSink() {
   cc::SurfaceManager* manager = GetSurfaceManager();
   if (!manager) {
     // Inform SurfaceFactory that SurfaceManager's no longer alive to
     // avoid its destruction error.
     surface_factory_->DidDestroySurfaceManager();
   } else {
     manager->InvalidateFrameSinkId(surface_id_.frame_sink_id());
   }
   surface_factory_->Destroy(surface_id_.local_frame_id());
+
+  OffscreenCanvasSurfaceManager::GetInstance()
+      ->UnregisterCompositorFrameSinkInstance(surface_id_.frame_sink_id());
 }
 
 // static
 void OffscreenCanvasCompositorFrameSink::Create(
     const cc::SurfaceId& surface_id,
     cc::mojom::MojoCompositorFrameSinkClientPtr client,
     cc::mojom::MojoCompositorFrameSinkRequest request) {
   mojo::MakeStrongBinding(base::MakeUnique<OffscreenCanvasCompositorFrameSink>(
                               surface_id, std::move(client)),
                           std::move(request));
@@ skipping 41 matching lines @@
   client_->DidReceiveCompositorFrameAck();
   DCHECK_GT(ack_pending_count_, 0);
   if (!surface_returned_resources_.empty()) {
     client_->ReclaimResources(surface_returned_resources_);
     surface_returned_resources_.clear();
   }
   ack_pending_count_--;
 }
 
 }  // namespace content