Create manager to track OffscreenCanvasSurfaceImpl instances

content/browser/renderer_host/offscreen_canvas_surface_manager.cc

Index: content/browser/renderer_host/offscreen_canvas_surface_manager.cc
diff --git a/content/browser/renderer_host/offscreen_canvas_surface_manager.cc b/content/browser/renderer_host/offscreen_canvas_surface_manager.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b834bb25f12adb5f56235a2dcf958a21f7d46d19
--- /dev/null
+++ b/content/browser/renderer_host/offscreen_canvas_surface_manager.cc
@@ -0,0 +1,76 @@
+// 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 "base/lazy_instance.h"
+#include "content/browser/renderer_host/offscreen_canvas_surface_manager.h"
+
+namespace content {
+
+namespace {
+base::LazyInstance<OffscreenCanvasSurfaceManager> g_manager =
+    LAZY_INSTANCE_INITIALIZER;

[xlai (Olivia), 2016/11/11 22:56:57]

I'm using a LazyInstance now, in response to dcheng@'s comment that when the
g_manager is destroyed, objects tracked in the
map unregister themselves on destruction anyway (which is true).

+}
+
+OffscreenCanvasSurfaceManager::OffscreenCanvasSurfaceManager() {}
+
+OffscreenCanvasSurfaceManager::~OffscreenCanvasSurfaceManager() {
+  registered_surface_instances_.clear();
+  registered_compositor_frame_sink_instances_.clear();
+}
+
+OffscreenCanvasSurfaceManager* OffscreenCanvasSurfaceManager::GetInstance() {
+  return g_manager.Pointer();
+}
+
+void OffscreenCanvasSurfaceManager::RegisterOffscreenCanvasSurfaceInstance(
+    cc::FrameSinkId frame_sink_id,
+    OffscreenCanvasSurfaceImpl* surface_instance) {
+  DCHECK(surface_instance);
+  DCHECK_EQ(registered_surface_instances_.count(frame_sink_id), 0u);
+  registered_surface_instances_[frame_sink_id] = surface_instance->GetWeakPtr();
+}
+
+void OffscreenCanvasSurfaceManager::UnregisterOffscreenCanvasSurfaceInstance(
+    cc::FrameSinkId frame_sink_id) {
+  DCHECK_EQ(registered_surface_instances_.count(frame_sink_id), 1u);
+  registered_surface_instances_.erase(frame_sink_id);
+}
+
+void OffscreenCanvasSurfaceManager::RegisterCompositorFrameSinkInstance(
+    cc::FrameSinkId frame_sink_id,
+    OffscreenCanvasCompositorFrameSink* surface_instance) {
+  DCHECK(surface_instance);
+  DCHECK_EQ(registered_compositor_frame_sink_instances_.count(frame_sink_id),
+            0u);
+  registered_compositor_frame_sink_instances_[frame_sink_id] =
+      surface_instance->GetWeakPtr();
+}
+
+void OffscreenCanvasSurfaceManager::UnregisterCompositorFrameSinkInstance(
+    cc::FrameSinkId frame_sink_id) {
+  DCHECK_EQ(registered_compositor_frame_sink_instances_.count(frame_sink_id),
+            1u);
+  registered_compositor_frame_sink_instances_.erase(frame_sink_id);
+}
+
+OffscreenCanvasSurfaceImpl* OffscreenCanvasSurfaceManager::GetSurfaceInstance(
+    cc::FrameSinkId frame_sink_id) {
+  auto search = registered_surface_instances_.find(frame_sink_id);
+  if (search != registered_surface_instances_.end()) {
+    return search->second.get();
+  }
+  return nullptr;
+}
+
+OffscreenCanvasCompositorFrameSink*
+OffscreenCanvasSurfaceManager::GetCompositorFrameSinkInstance(
+    cc::FrameSinkId frame_sink_id) {
+  auto search = registered_compositor_frame_sink_instances_.find(frame_sink_id);
+  if (search != registered_compositor_frame_sink_instances_.end()) {
+    return search->second.get();
+  }
+  return nullptr;
+}
+
+}  // namespace content