[compositorworker] Register compositor proxies with proxy client

third_party/WebKit/Source/web/CompositorProxyClientImpl.cpp

 // Copyright 2015 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 "web/CompositorProxyClientImpl.h"
 
 #include "core/dom/CompositorProxy.h"
 #include "modules/compositorworker/CompositorWorkerGlobalScope.h"
 #include "platform/TraceEvent.h"
 #include "web/CompositorMutatorImpl.h"
 #include "wtf/CurrentTime.h"
 
 namespace blink {
 
 CompositorProxyClientImpl::CompositorProxyClientImpl(CompositorMutatorImpl* mutator)
     : m_mutator(mutator)
     , m_globalScope(nullptr)
 {
 }
 
 DEFINE_TRACE(CompositorProxyClientImpl)
 {
     CompositorProxyClient::trace(visitor);
     visitor->trace(m_mutator);
     visitor->trace(m_globalScope);
+    visitor->trace(m_proxyMap);
 }
 
 void CompositorProxyClientImpl::setGlobalScope(WorkerGlobalScope* scope)
 {
     TRACE_EVENT0("compositor-worker", "CompositorProxyClientImpl::setGlobalScope");
     DCHECK(!m_globalScope);
     DCHECK(scope);
     m_globalScope = static_cast<CompositorWorkerGlobalScope*>(scope);
     m_mutator->registerProxyClient(this);
 }
@@ skipping 21 matching lines @@
 
 bool CompositorProxyClientImpl::executeAnimationFrameCallbacks(double monotonicTimeNow)
 {
     TRACE_EVENT0("compositor-worker", "CompositorProxyClientImpl::executeAnimationFrameCallbacks");
     // Convert to zero based document time in milliseconds consistent with requestAnimationFrame.
     double highResTimeMs = 1000.0 * (monotonicTimeNow - m_globalScope->timeOrigin());
     const bool shouldReinvoke = m_globalScope->executeAnimationFrameCallbacks(highResTimeMs);
     return shouldReinvoke;
 }
 
+void CompositorProxyClientImpl::registerCompositorProxy(CompositorProxy* proxy)
+{
+    uint64_t elementId = proxy->elementId();
+
+    TRACE_EVENT2("compositor-worker,compositor-proxy",

[haraken, 2016/06/08 00:10:36]

registerCompositorProxy should be cheap, so it won't make sense to add a trace
event.

[majidvp, 2016/06/09 20:34:21]

Removed. We were using them as a way to trace the call stack for debugging.

+        "CompositorProxyClientImpl::registerCompositorProxy",
+        "elementId", static_cast<int>(elementId),
+        "proxy", proxy);
+
+    ProxyMap::AddResult entry = m_proxyMap.add(elementId, new ProxyVector);
+    ProxyVector* proxies = entry.storedValue->value;
+    if (!proxies->contains(proxy))
+        proxies->append(proxy);
+}
+
+void CompositorProxyClientImpl::unregisterCompositorProxy(CompositorProxy* proxy)
+{
+    uint64_t elementId = proxy->elementId();
+    if (ProxyVector* proxies = m_proxyMap.get(elementId)) {
+        TRACE_EVENT2("compositor-worker,compositor-proxy",

[haraken, 2016/06/08 00:10:36]

Ditto.

+            "CompositorProxyClientImpl::unregisterCompositorProxy",
+            "elementId", static_cast<int>(elementId),
+            "proxy", proxy);
+
+        size_t position = proxies->find(proxy);
+        if (position == kNotFound)
+            return;
+
+        proxies->remove(position);
+        if (proxies->isEmpty())
+            m_proxyMap.remove(elementId);
+    }
+}
+
 } // namespace blink