Add optimization for CHROMIUM_subscribe_uniform extension.

content/browser/gpu/gpu_process_host_ui_shim.cc

 // 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.
 
 #include "content/browser/gpu/gpu_process_host_ui_shim.h"
 
 #include <algorithm>
 
 #include "base/bind.h"
 #include "base/callback_helpers.h"
 #include "base/debug/trace_event.h"
 #include "base/id_map.h"
 #include "base/lazy_instance.h"
 #include "base/strings/string_number_conversions.h"
 #include "content/browser/compositor/gpu_process_transport_factory.h"
 #include "content/browser/gpu/compositor_util.h"
 #include "content/browser/gpu/gpu_data_manager_impl.h"
 #include "content/browser/gpu/gpu_process_host.h"
 #include "content/browser/gpu/gpu_surface_tracker.h"
 #include "content/browser/renderer_host/render_process_host_impl.h"
 #include "content/browser/renderer_host/render_view_host_impl.h"
 #include "content/browser/renderer_host/render_widget_helper.h"
 #include "content/browser/renderer_host/render_widget_host_view_base.h"
 #include "content/common/gpu/gpu_messages.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/browser/render_widget_host_iterator.h"
 
 #if defined(OS_MACOSX)
 #include "ui/accelerated_widget_mac/accelerated_widget_mac.h"
 #endif
 
 #if defined(USE_OZONE)
 #include "ui/ozone/public/gpu_platform_support_host.h"
 #include "ui/ozone/public/ozone_platform.h"
 #endif
 
@@ skipping 187 matching lines @@
     IPC_MESSAGE_HANDLER(GpuHostMsg_AcceleratedSurfaceInitialized,
                         OnAcceleratedSurfaceInitialized)
     IPC_MESSAGE_HANDLER(GpuHostMsg_AcceleratedSurfaceBuffersSwapped,
                         OnAcceleratedSurfaceBuffersSwapped)
     IPC_MESSAGE_HANDLER(GpuHostMsg_GraphicsInfoCollected,
                         OnGraphicsInfoCollected)
     IPC_MESSAGE_HANDLER(GpuHostMsg_VideoMemoryUsageStats,
                         OnVideoMemoryUsageStatsReceived);
     IPC_MESSAGE_HANDLER(GpuHostMsg_ResourcesRelinquished,
                         OnResourcesRelinquished)
+    IPC_MESSAGE_HANDLER(GpuHostMsg_AddSubscription, OnAddSubscription);
+    IPC_MESSAGE_HANDLER(GpuHostMsg_RemoveSubscription, OnRemoveSubscription);
 
     IPC_MESSAGE_UNHANDLED_ERROR()
   IPC_END_MESSAGE_MAP()
 
   return true;
 }
 
 void GpuProcessHostUIShim::OnLogMessage(
     int level,
     const std::string& header,
@@ skipping 56 matching lines @@
   GpuDataManagerImpl::GetInstance()->UpdateVideoMemoryUsageStats(
       video_memory_usage_stats);
 }
 
 void GpuProcessHostUIShim::OnResourcesRelinquished() {
   if (!relinquish_callback_.is_null()) {
     base::ResetAndReturn(&relinquish_callback_).Run();
   }
 }
 
+void GpuProcessHostUIShim::OnAddSubscription(
+    int32 process_id, unsigned int target) {
+  scoped_ptr<RenderWidgetHostIterator> rwhi_iterator =
+      RenderWidgetHostImpl::GetRenderWidgetHosts();
+  while (RenderWidgetHost* widget = rwhi_iterator->GetNextHost()) {
+    // Count only RenderWidgetHosts in this process.
+    if (widget->GetProcess()->GetID() == process_id) {
+      RenderWidgetHostImpl* rwhi = RenderWidgetHostImpl::From(widget);
+      rwhi->OnAddSubscription(target);
+    }
+  }
+}
+
+void GpuProcessHostUIShim::OnRemoveSubscription(
+    int32 process_id, unsigned int target) {
+  scoped_ptr<RenderWidgetHostIterator> rwhi_iterator =
+      RenderWidgetHostImpl::GetRenderWidgetHosts();
+  while (RenderWidgetHost* widget = rwhi_iterator->GetNextHost()) {
+    // Count only RenderWidgetHosts in this process.
+    if (widget->GetProcess()->GetID() == process_id) {
+      RenderWidgetHostImpl* rwhi = RenderWidgetHostImpl::From(widget);
+      rwhi->OnRemoveSubscription(target);
+    }
+  }
+}
+
 }  // namespace content