Refactor MutatorHostClient from LayerTreeHost to LayerTree.

cc/trees/layer_tree.h

 // 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.
 
 #ifndef CC_TREES_LAYER_TREE_H_
 #define CC_TREES_LAYER_TREE_H_
 
 #include <memory>
-
 #include <unordered_map>
 #include <unordered_set>
+
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "cc/base/cc_export.h"
 #include "cc/input/event_listener_properties.h"
 #include "cc/input/layer_selection_bound.h"
 #include "cc/layers/layer_collections.h"
+#include "cc/layers/layer_list_iterator.h"
+#include "cc/trees/mutator_host_client.h"
 #include "cc/trees/property_tree.h"
 #include "third_party/skia/include/core/SkColor.h"
 #include "ui/gfx/geometry/size.h"
 
 namespace base {
 class TimeDelta;
 }  // namespace base
 
 namespace cc {
 
 namespace proto {
 class LayerTree;
 class LayerUpdate;
 }  // namespace proto
 
 class AnimationHost;
 class HeadsUpDisplayLayer;
 class Layer;
 class LayerTreeHost;
 class LayerTreeImpl;
 struct PendingPageScaleAnimation;
 
-class CC_EXPORT LayerTree {
+class CC_EXPORT LayerTree : public MutatorHostClient {
  public:
   using LayerSet = std::unordered_set<Layer*>;
   using LayerIdMap = std::unordered_map<int, Layer*>;
 
   LayerTree(std::unique_ptr<AnimationHost> animation_host,
             LayerTreeHost* layer_tree_host);
   virtual ~LayerTree();
 
   void SetRootLayer(scoped_refptr<Layer> root_layer);
   Layer* root_layer() { return inputs_.root_layer.get(); }
@@ skipping 53 matching lines @@
                                bool use_anchor,
                                float scale,
                                base::TimeDelta duration);
   bool HasPendingPageScaleAnimation() const;
 
   void SetDeviceScaleFactor(float device_scale_factor);
   float device_scale_factor() const { return inputs_.device_scale_factor; }
 
   void SetPaintedDeviceScaleFactor(float painted_device_scale_factor);
 
-  AnimationHost* animation_host() const { return animation_host_.get(); }
-
   gfx::Vector2dF elastic_overscroll() const { return elastic_overscroll_; }
 
   // Used externally by blink for setting the PropertyTrees when
   // |settings_.use_layer_lists| is true. This is a SPV2 setting.
   PropertyTrees* property_trees() { return &property_trees_; }
 
+  bool in_paint_layer_contents() const { return in_paint_layer_contents_; }
+
   // Methods which should only be used internally in cc ------------------
   void RegisterLayer(Layer* layer);
   void UnregisterLayer(Layer* layer);
   Layer* LayerById(int id) const;
   bool UpdateLayers(const LayerList& update_layer_list,
                     bool* content_is_suitable_for_gpu);
 
   void AddLayerShouldPushProperties(Layer* layer);
   void RemoveLayerShouldPushProperties(Layer* layer);
   std::unordered_set<Layer*>& LayersThatShouldPushProperties();
@@ skipping 15 matching lines @@
   bool needs_full_tree_sync() const { return needs_full_tree_sync_; }
 
   void SetNeedsCommit();
   void SetPropertyTreesNeedRebuild();
 
   void PushPropertiesTo(LayerTreeImpl* tree_impl);
 
   void ToProtobuf(proto::LayerTree* proto);
   void FromProtobuf(const proto::LayerTree& proto);
 
-  bool in_paint_layer_contents() const { return in_paint_layer_contents_; }
+  AnimationHost* animation_host() const { return animation_host_.get(); }
+
+  Layer* LayerByElementId(ElementId element_id) const;
+  void RegisterElement(ElementId element_id,
+                       ElementListType list_type,
+                       Layer* layer);
+  void UnregisterElement(ElementId element_id,
+                         ElementListType list_type,
+                         Layer* layer);
+  void SetElementIdsForTesting();
+
+  // Layer iterators.
+  LayerListIterator<Layer> begin() const;
+  LayerListIterator<Layer> end() const;
+  LayerListReverseIterator<Layer> rbegin();
+  LayerListReverseIterator<Layer> rend();
+
+  void SetNeedsDisplayOnAllLayers();
   // ---------------------------------------------------------------------
 
  private:
   friend class LayerTreeHostSerializationTest;
 
+  // MutatorHostClient implementation.
+  bool IsElementInList(ElementId element_id,
+                       ElementListType list_type) const override;
+  void SetMutatorsNeedCommit() override;
+  void SetMutatorsNeedRebuildPropertyTrees() override;
+  void SetElementFilterMutated(ElementId element_id,
+                               ElementListType list_type,
+                               const FilterOperations& filters) override;
+  void SetElementOpacityMutated(ElementId element_id,
+                                ElementListType list_type,
+                                float opacity) override;
+  void SetElementTransformMutated(ElementId element_id,
+                                  ElementListType list_type,
+                                  const gfx::Transform& transform) override;
+  void SetElementScrollOffsetMutated(
+      ElementId element_id,
+      ElementListType list_type,
+      const gfx::ScrollOffset& scroll_offset) override;
+  void ElementTransformIsAnimatingChanged(ElementId element_id,
+                                          ElementListType list_type,
+                                          AnimationChangeType change_type,
+                                          bool is_animating) override;
+  void ElementOpacityIsAnimatingChanged(ElementId element_id,
+                                        ElementListType list_type,
+                                        AnimationChangeType change_type,
+                                        bool is_animating) override;
+  void ElementFilterIsAnimatingChanged(ElementId element_id,
+                                       ElementListType list_type,
+                                       AnimationChangeType change_type,
+                                       bool is_animating) override;
+  void ScrollOffsetAnimationFinished() override {}
+  gfx::ScrollOffset GetScrollOffsetForAnimation(
+      ElementId element_id) const override;
+
   // Encapsulates the data, callbacks, interfaces received from the embedder.
   struct Inputs {
     Inputs();
     ~Inputs();
 
     scoped_refptr<Layer> root_layer;
 
     scoped_refptr<Layer> overscroll_elasticity_layer;
     scoped_refptr<Layer> page_scale_layer;
     scoped_refptr<Layer> inner_viewport_scroll_layer;
@@ skipping 33 matching lines @@
   gfx::Vector2dF elastic_overscroll_;
 
   scoped_refptr<HeadsUpDisplayLayer> hud_layer_;
 
   // Set of layers that need to push properties.
   LayerSet layers_that_should_push_properties_;
 
   // Layer id to Layer map.
   LayerIdMap layer_id_map_;
 
+  using ElementLayersMap = std::unordered_map<ElementId, Layer*, ElementIdHash>;
+  ElementLayersMap element_layers_map_;
+
   bool in_paint_layer_contents_;
 
   std::unique_ptr<AnimationHost> animation_host_;
   LayerTreeHost* layer_tree_host_;
 
   DISALLOW_COPY_AND_ASSIGN(LayerTree);
 };
 
 }  // namespace cc
 
 #endif  // CC_TREES_LAYER_TREE_H_