Populate cc element id maps for transform/effect nodes.

third_party/WebKit/Source/platform/graphics/compositing/PropertyTreeManager.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 "platform/graphics/compositing/PropertyTreeManager.h"
 
 #include "cc/layers/layer.h"
 #include "cc/trees/clip_node.h"
 #include "cc/trees/effect_node.h"
 #include "cc/trees/property_tree.h"
@@ skipping 44 matching lines @@
 
 const EffectPaintPropertyNode* PropertyTreeManager::currentEffectNode() const {
   return m_effectStack.back().effect;
 }
 
 void PropertyTreeManager::setupRootTransformNode() {
   // cc is hardcoded to use transform node index 1 for device scale and
   // transform.
   cc::TransformTree& transformTree = m_propertyTrees.transform_tree;
   transformTree.clear();
+  m_propertyTrees.element_id_to_transform_node_index.clear();
   cc::TransformNode& transformNode = *transformTree.Node(
       transformTree.Insert(cc::TransformNode(), kRealRootNodeId));
   DCHECK_EQ(transformNode.id, kSecondaryRootNodeId);
   transformNode.source_node_id = transformNode.parent_id;
   transformTree.SetTargetId(transformNode.id, kRealRootNodeId);
   transformTree.SetContentTargetId(transformNode.id, kRealRootNodeId);
 
   // TODO(jaydasika): We shouldn't set ToScreen and FromScreen of root
   // transform node here. They should be set while updating transform tree in
   // cc.
@@ skipping 33 matching lines @@
 
   m_clipNodeMap.set(ClipPaintPropertyNode::root(), clipNode.id);
   m_rootLayer->SetClipTreeIndex(clipNode.id);
 }
 
 void PropertyTreeManager::setupRootEffectNode() {
   // cc is hardcoded to use effect node index 1 for root render surface.
   cc::EffectTree& effectTree = m_propertyTrees.effect_tree;
   effectTree.clear();
   m_propertyTrees.layer_id_to_effect_node_index.clear();
+  m_propertyTrees.element_id_to_effect_node_index.clear();
   cc::EffectNode& effectNode =
       *effectTree.Node(effectTree.Insert(cc::EffectNode(), kInvalidNodeId));
   DCHECK_EQ(effectNode.id, kSecondaryRootNodeId);
   effectNode.owning_layer_id = m_rootLayer->id();
   effectNode.transform_id = kRealRootNodeId;
   effectNode.clip_id = kSecondaryRootNodeId;
   effectNode.has_render_surface = true;
   m_propertyTrees.layer_id_to_effect_node_index[effectNode.owning_layer_id] =
       effectNode.id;
 
@@ skipping 50 matching lines @@
   compositorNode.flattens_inherited_transform =
       transformNode->flattensInheritedTransform();
   compositorNode.sorting_context_id = transformNode->renderingContextId();
 
   m_rootLayer->AddChild(dummyLayer);
   dummyLayer->SetTransformTreeIndex(id);
   dummyLayer->SetClipTreeIndex(kSecondaryRootNodeId);
   dummyLayer->SetEffectTreeIndex(kSecondaryRootNodeId);
   dummyLayer->SetScrollTreeIndex(kRealRootNodeId);
   dummyLayer->set_property_tree_sequence_number(kPropertyTreeSequenceNumber);
+  CompositorElementId compositorElementId =
+      transformNode->compositorElementId();
+  if (compositorElementId) {
+    m_propertyTrees.element_id_to_transform_node_index[compositorElementId] =
+        id;
+  }
 
   auto result = m_transformNodeMap.set(transformNode, id);
   DCHECK(result.isNewEntry);
   transformTree().set_needs_update(true);
   return id;
 }
 
 int PropertyTreeManager::ensureCompositorClipNode(
     const ClipPaintPropertyNode* clipNode) {
   DCHECK(clipNode);
@@ skipping 45 matching lines @@
     return kSecondaryRootNodeId;
 
   auto it = m_scrollNodeMap.find(scrollNode);
   if (it != m_scrollNodeMap.end())
     return it->value;
 
   int parentId = ensureCompositorScrollNode(scrollNode->parent());
   int id = scrollTree().Insert(cc::ScrollNode(), parentId);
 
   cc::ScrollNode& compositorNode = *scrollTree().Node(id);
+  // TODO(wkorman): Fix owning layer id to reference a layer id rather than a
+  // scroll node index.
   compositorNode.owning_layer_id = parentId;
   m_propertyTrees
       .layer_id_to_scroll_node_index[compositorNode.owning_layer_id] = id;
 
   compositorNode.scrollable = true;
 
   compositorNode.scroll_clip_layer_bounds.SetSize(scrollNode->clip().width(),
                                                   scrollNode->clip().height());
   compositorNode.bounds.SetSize(scrollNode->bounds().width(),
                                 scrollNode->bounds().height());
@@ skipping 116 matching lines @@
   // decision until later phase of the pipeline. Remove premature optimization
   // here once the work is ready.
   if (!nextEffect->filter().isEmpty() ||
       nextEffect->blendMode() != SkBlendMode::kSrcOver)
     effectNode.has_render_surface = true;
   effectNode.opacity = nextEffect->opacity();
   effectNode.filters = nextEffect->filter().asCcFilterOperations();
   effectNode.blend_mode = nextEffect->blendMode();
   m_propertyTrees.layer_id_to_effect_node_index[effectNode.owning_layer_id] =
       effectNode.id;
+  CompositorElementId compositorElementId = nextEffect->compositorElementId();
+  if (compositorElementId) {
+    m_propertyTrees.element_id_to_effect_node_index[compositorElementId] =
+        effectNode.id;
+  }
   m_effectStack.push_back(BlinkEffectAndCcIdPair{nextEffect, effectNode.id});
 
   dummyLayer->set_property_tree_sequence_number(kPropertyTreeSequenceNumber);
   dummyLayer->SetTransformTreeIndex(kSecondaryRootNodeId);
   dummyLayer->SetClipTreeIndex(outputClipId);
   dummyLayer->SetEffectTreeIndex(effectNode.id);
   dummyLayer->SetScrollTreeIndex(kRealRootNodeId);
 }
 
 }  // namespace blink