CC Animation: Merge LayerAnimationController into ElementAnimations.

cc/animation/animation_host.cc

 // 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 "cc/animation/animation_host.h"
 
 #include <algorithm>
 
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/trace_event/trace_event.h"
 #include "base/trace_event/trace_event_argument.h"
 #include "cc/animation/animation_delegate.h"
 #include "cc/animation/animation_events.h"
 #include "cc/animation/animation_id_provider.h"
 #include "cc/animation/animation_player.h"
 #include "cc/animation/animation_timeline.h"
 #include "cc/animation/element_animations.h"
-#include "cc/animation/layer_animation_controller.h"
 #include "cc/animation/scroll_offset_animation_curve.h"
 #include "cc/animation/timing_function.h"
 #include "ui/gfx/geometry/box_f.h"
 #include "ui/gfx/geometry/scroll_offset.h"
 
 namespace cc {
 
 class AnimationHost::ScrollOffsetAnimations : public AnimationDelegate {
  public:
   explicit ScrollOffsetAnimations(AnimationHost* animation_host)
@@ skipping 118 matching lines @@
 };
 
 std::unique_ptr<AnimationHost> AnimationHost::Create(
     ThreadInstance thread_instance) {
   return base::WrapUnique(new AnimationHost(thread_instance));
 }
 
 AnimationHost::AnimationHost(ThreadInstance thread_instance)
     : mutator_host_client_(nullptr),
       thread_instance_(thread_instance),
-      supports_scroll_animations_(false) {
+      supports_scroll_animations_(false),
+      animation_waiting_for_deletion_(false) {
   if (thread_instance_ == ThreadInstance::IMPL)
     scroll_offset_animations_ =
         base::WrapUnique(new ScrollOffsetAnimations(this));
 }
 
 AnimationHost::~AnimationHost() {
   scroll_offset_animations_ = nullptr;
 
   ClearTimelines();
   DCHECK(!mutator_host_client());
   DCHECK(layer_to_element_animations_map_.empty());
-
-  AnimationControllerMap copy = all_animation_controllers_;
-  for (AnimationControllerMap::iterator iter = copy.begin(); iter != copy.end();
-       ++iter)
-    (*iter).second->SetAnimationHost(nullptr);
 }
 
 AnimationTimeline* AnimationHost::GetTimelineById(int timeline_id) const {
   auto f = id_to_timeline_map_.find(timeline_id);
   return f == id_to_timeline_map_.end() ? nullptr : f->second.get();
 }
 
 void AnimationHost::ClearTimelines() {
   for (auto& kv : id_to_timeline_map_)
     EraseTimeline(kv.second);
@@ skipping 35 matching lines @@
 }
 
 void AnimationHost::RegisterPlayerForLayer(int layer_id,
                                            AnimationPlayer* player) {
   DCHECK(layer_id);
   DCHECK(player);
 
   scoped_refptr<ElementAnimations> element_animations =
       GetElementAnimationsForLayerId(layer_id);
   if (!element_animations) {
-    element_animations = ElementAnimations::Create(this);
-    layer_to_element_animations_map_[layer_id] = element_animations;
-
-    element_animations->CreateLayerAnimationController(layer_id);
+    element_animations = ElementAnimations::Create();
+    element_animations->SetLayerId(layer_id);
+    RegisterElementAnimations(element_animations.get());
   }
 
-  DCHECK(element_animations);
+  element_animations->SetAnimationHost(this);
+  element_animations->InitValueObservations();
+
   element_animations->AddPlayer(player);
 }
 
 void AnimationHost::UnregisterPlayerForLayer(int layer_id,
                                              AnimationPlayer* player) {
   DCHECK(layer_id);
   DCHECK(player);
 
   scoped_refptr<ElementAnimations> element_animations =
       GetElementAnimationsForLayerId(layer_id);
   DCHECK(element_animations);
   element_animations->RemovePlayer(player);
 
   if (element_animations->IsEmpty()) {
-    element_animations->DestroyLayerAnimationController();
-    layer_to_element_animations_map_.erase(layer_id);
+    element_animations->ClearValueObservations();
+    UnregisterElementAnimations(element_animations.get());
+    element_animations->SetAnimationHost(nullptr);
   }
 }
 
 void AnimationHost::SetMutatorHostClient(MutatorHostClient* client) {
   if (mutator_host_client_ == client)
     return;
 
   mutator_host_client_ = client;
 }
 
 void AnimationHost::SetNeedsCommit() {
   DCHECK(mutator_host_client_);
   mutator_host_client_->SetMutatorsNeedCommit();
 }
 
 void AnimationHost::SetNeedsRebuildPropertyTrees() {
   DCHECK(mutator_host_client_);
   mutator_host_client_->SetMutatorsNeedRebuildPropertyTrees();
 }
 
 void AnimationHost::PushPropertiesTo(AnimationHost* host_impl) {
   PushTimelinesToImplThread(host_impl);
   RemoveTimelinesFromImplThread(host_impl);
   PushPropertiesToImplThread(host_impl);
+  animation_waiting_for_deletion_ = false;
 }
 
 void AnimationHost::PushTimelinesToImplThread(AnimationHost* host_impl) const {
   for (auto& kv : id_to_timeline_map_) {
     auto& timeline = kv.second;
     AnimationTimeline* timeline_impl =
         host_impl->GetTimelineById(timeline->id());
     if (timeline_impl)
       continue;
 
@@ skipping 12 matching lines @@
     if (timeline_impl->is_impl_only() || GetTimelineById(timeline_impl->id())) {
       ++it;
     } else {
       host_impl->EraseTimeline(it->second);
       it = timelines_impl.erase(it);
     }
   }
 }
 
 void AnimationHost::PushPropertiesToImplThread(AnimationHost* host_impl) {
-  // Firstly, sync all players with impl thread to create ElementAnimations and
-  // layer animation controllers.
+  // Firstly, sync all players with impl thread to create ElementAnimations.
   for (auto& kv : id_to_timeline_map_) {
     AnimationTimeline* timeline = kv.second.get();
     AnimationTimeline* timeline_impl =
         host_impl->GetTimelineById(timeline->id());
     if (timeline_impl)
       timeline->PushPropertiesTo(timeline_impl);
   }
 
-  // Secondly, sync properties for created layer animation controllers.
+  // Secondly, sync properties for created ElementAnimations.
   for (auto& kv : layer_to_element_animations_map_) {
     const auto& element_animations = kv.second;
     auto element_animations_impl =
         host_impl->GetElementAnimationsForLayerId(kv.first);
     if (element_animations_impl)
       element_animations->PushPropertiesTo(std::move(element_animations_impl));
   }
 }
 
-LayerAnimationController* AnimationHost::GetControllerForLayerId(
-    int layer_id) const {
-  const scoped_refptr<ElementAnimations> element_animations =
-      GetElementAnimationsForLayerId(layer_id);
-  if (!element_animations)
-    return nullptr;
-
-  return element_animations->layer_animation_controller_.get();
-}
-
 scoped_refptr<ElementAnimations> AnimationHost::GetElementAnimationsForLayerId(
     int layer_id) const {
   DCHECK(layer_id);
   auto iter = layer_to_element_animations_map_.find(layer_id);
   return iter == layer_to_element_animations_map_.end() ? nullptr
                                                         : iter->second;
 }
 
 void AnimationHost::SetSupportsScrollAnimations(
     bool supports_scroll_animations) {
   supports_scroll_animations_ = supports_scroll_animations;
 }
 
 bool AnimationHost::SupportsScrollAnimations() const {
   return supports_scroll_animations_;
 }
 
 bool AnimationHost::NeedsAnimateLayers() const {
-  return !active_animation_controllers_.empty();
+  return !active_element_animations_map_.empty();
 }
 
 bool AnimationHost::ActivateAnimations() {
   if (!NeedsAnimateLayers())
     return false;
 
   TRACE_EVENT0("cc", "AnimationHost::ActivateAnimations");
-  AnimationControllerMap active_controllers_copy =
-      active_animation_controllers_;
-  for (auto& it : active_controllers_copy)
+  LayerToElementAnimationsMap active_element_animations_map_copy =
+      active_element_animations_map_;
+  for (auto& it : active_element_animations_map_copy)
     it.second->ActivateAnimations();
 
   return true;
 }
 
 bool AnimationHost::AnimateLayers(base::TimeTicks monotonic_time) {
   if (!NeedsAnimateLayers())
     return false;
 
   TRACE_EVENT0("cc", "AnimationHost::AnimateLayers");
-  AnimationControllerMap controllers_copy = active_animation_controllers_;
-  for (auto& it : controllers_copy)
+  LayerToElementAnimationsMap active_element_animations_map_copy =
+      active_element_animations_map_;
+  for (auto& it : active_element_animations_map_copy)
     it.second->Animate(monotonic_time);
 
   return true;
 }
 
 bool AnimationHost::UpdateAnimationState(bool start_ready_animations,
                                          AnimationEvents* events) {
   if (!NeedsAnimateLayers())
     return false;
 
   TRACE_EVENT0("cc", "AnimationHost::UpdateAnimationState");
-  AnimationControllerMap active_controllers_copy =
-      active_animation_controllers_;
-  for (auto& it : active_controllers_copy)
+  LayerToElementAnimationsMap active_element_animations_map_copy =
+      active_element_animations_map_;
+  for (auto& it : active_element_animations_map_copy)
     it.second->UpdateState(start_ready_animations, events);
 
   return true;
 }
 
 std::unique_ptr<AnimationEvents> AnimationHost::CreateEvents() {
   return base::WrapUnique(new AnimationEvents());
 }
 
 void AnimationHost::SetAnimationEvents(
     std::unique_ptr<AnimationEvents> events) {
   for (size_t event_index = 0; event_index < events->events_.size();
        ++event_index) {
     int event_layer_id = events->events_[event_index].layer_id;
 
-    // Use the map of all controllers, not just active ones, since non-active
-    // controllers may still receive events for impl-only animations.
-    const AnimationControllerMap& animation_controllers =
-        all_animation_controllers_;
-    auto iter = animation_controllers.find(event_layer_id);
-    if (iter != animation_controllers.end()) {
+    // Use the map of all ElementAnimations, not just active ones, since
+    // non-active ElementAnimations may still receive events for impl-only
+    // animations.
+    const LayerToElementAnimationsMap& all_element_animations =
+        layer_to_element_animations_map_;
+    auto iter = all_element_animations.find(event_layer_id);
+    if (iter != all_element_animations.end()) {
       switch (events->events_[event_index].type) {
         case AnimationEvent::STARTED:
           (*iter).second->NotifyAnimationStarted(events->events_[event_index]);
           break;
 
         case AnimationEvent::FINISHED:
           (*iter).second->NotifyAnimationFinished(events->events_[event_index]);
           break;
 
         case AnimationEvent::ABORTED:
           (*iter).second->NotifyAnimationAborted(events->events_[event_index]);
           break;
 
         case AnimationEvent::PROPERTY_UPDATE:
           (*iter).second->NotifyAnimationPropertyUpdate(
               events->events_[event_index]);
           break;
 
         case AnimationEvent::TAKEOVER:
           (*iter).second->NotifyAnimationTakeover(events->events_[event_index]);
           break;
       }
     }
   }
 }
 
 bool AnimationHost::ScrollOffsetAnimationWasInterrupted(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller ? controller->scroll_offset_animation_was_interrupted()
-                    : false;
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->scroll_offset_animation_was_interrupted()
+             : false;
 }
 
-static LayerAnimationController::ObserverType ObserverTypeFromTreeType(
+static ElementAnimations::ObserverType ObserverTypeFromTreeType(
     LayerTreeType tree_type) {
   return tree_type == LayerTreeType::ACTIVE
-             ? LayerAnimationController::ObserverType::ACTIVE
-             : LayerAnimationController::ObserverType::PENDING;
+             ? ElementAnimations::ObserverType::ACTIVE
+             : ElementAnimations::ObserverType::PENDING;
 }
 
 bool AnimationHost::IsAnimatingFilterProperty(int layer_id,
                                               LayerTreeType tree_type) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller
-             ? controller->IsCurrentlyAnimatingProperty(
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->IsCurrentlyAnimatingProperty(
                    TargetProperty::FILTER, ObserverTypeFromTreeType(tree_type))
              : false;
 }
 
 bool AnimationHost::IsAnimatingOpacityProperty(int layer_id,
                                                LayerTreeType tree_type) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller
-             ? controller->IsCurrentlyAnimatingProperty(
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->IsCurrentlyAnimatingProperty(
                    TargetProperty::OPACITY, ObserverTypeFromTreeType(tree_type))
              : false;
 }
 
 bool AnimationHost::IsAnimatingTransformProperty(
     int layer_id,
     LayerTreeType tree_type) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller
-             ? controller->IsCurrentlyAnimatingProperty(
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->IsCurrentlyAnimatingProperty(
                    TargetProperty::TRANSFORM,
                    ObserverTypeFromTreeType(tree_type))
              : false;
 }
 
 bool AnimationHost::HasPotentiallyRunningFilterAnimation(
     int layer_id,
     LayerTreeType tree_type) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller
-             ? controller->IsPotentiallyAnimatingProperty(
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->IsPotentiallyAnimatingProperty(
                    TargetProperty::FILTER, ObserverTypeFromTreeType(tree_type))
              : false;
 }
 
 bool AnimationHost::HasPotentiallyRunningOpacityAnimation(
     int layer_id,
     LayerTreeType tree_type) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller
-             ? controller->IsPotentiallyAnimatingProperty(
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->IsPotentiallyAnimatingProperty(
                    TargetProperty::OPACITY, ObserverTypeFromTreeType(tree_type))
              : false;
 }
 
 bool AnimationHost::HasPotentiallyRunningTransformAnimation(
     int layer_id,
     LayerTreeType tree_type) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller
-             ? controller->IsPotentiallyAnimatingProperty(
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->IsPotentiallyAnimatingProperty(
                    TargetProperty::TRANSFORM,
                    ObserverTypeFromTreeType(tree_type))
              : false;
 }
 
 bool AnimationHost::HasAnyAnimationTargetingProperty(
     int layer_id,
     TargetProperty::Type property) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  if (!controller)
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  if (!element_animations)
     return false;
 
-  return !!controller->GetAnimation(property);
+  return !!element_animations->GetAnimation(property);
 }
 
 bool AnimationHost::FilterIsAnimatingOnImplOnly(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  if (!controller)
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  if (!element_animations)
     return false;
 
-  Animation* animation = controller->GetAnimation(TargetProperty::FILTER);
+  Animation* animation =
+      element_animations->GetAnimation(TargetProperty::FILTER);
   return animation && animation->is_impl_only();
 }
 
 bool AnimationHost::OpacityIsAnimatingOnImplOnly(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  if (!controller)
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  if (!element_animations)
     return false;
 
-  Animation* animation = controller->GetAnimation(TargetProperty::OPACITY);
+  Animation* animation =
+      element_animations->GetAnimation(TargetProperty::OPACITY);
   return animation && animation->is_impl_only();
 }
 
 bool AnimationHost::ScrollOffsetIsAnimatingOnImplOnly(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  if (!controller)
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  if (!element_animations)
     return false;
 
   Animation* animation =
-      controller->GetAnimation(TargetProperty::SCROLL_OFFSET);
+      element_animations->GetAnimation(TargetProperty::SCROLL_OFFSET);
   return animation && animation->is_impl_only();
 }
 
 bool AnimationHost::TransformIsAnimatingOnImplOnly(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  if (!controller)
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  if (!element_animations)
     return false;
 
-  Animation* animation = controller->GetAnimation(TargetProperty::TRANSFORM);
+  Animation* animation =
+      element_animations->GetAnimation(TargetProperty::TRANSFORM);
   return animation && animation->is_impl_only();
 }
 
 bool AnimationHost::HasFilterAnimationThatInflatesBounds(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller ? controller->HasFilterAnimationThatInflatesBounds()
-                    : false;
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->HasFilterAnimationThatInflatesBounds()
+             : false;
 }
 
 bool AnimationHost::HasTransformAnimationThatInflatesBounds(
     int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller ? controller->HasTransformAnimationThatInflatesBounds()
-                    : false;
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->HasTransformAnimationThatInflatesBounds()
+             : false;
 }
 
 bool AnimationHost::HasAnimationThatInflatesBounds(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller ? controller->HasAnimationThatInflatesBounds() : false;
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->HasAnimationThatInflatesBounds()
+             : false;
 }
 
 bool AnimationHost::FilterAnimationBoundsForBox(int layer_id,
                                                 const gfx::BoxF& box,
                                                 gfx::BoxF* bounds) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller ? controller->FilterAnimationBoundsForBox(box, bounds)
-                    : false;
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->FilterAnimationBoundsForBox(box, bounds)
+             : false;
 }
 
 bool AnimationHost::TransformAnimationBoundsForBox(int layer_id,
                                                    const gfx::BoxF& box,
                                                    gfx::BoxF* bounds) const {
   *bounds = gfx::BoxF();
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller ? controller->TransformAnimationBoundsForBox(box, bounds)
-                    : true;
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->TransformAnimationBoundsForBox(box, bounds)
+             : true;
 }
 
 bool AnimationHost::HasOnlyTranslationTransforms(
     int layer_id,
     LayerTreeType tree_type) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller
-             ? controller->HasOnlyTranslationTransforms(
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->HasOnlyTranslationTransforms(
                    ObserverTypeFromTreeType(tree_type))
              : true;
 }
 
 bool AnimationHost::AnimationsPreserveAxisAlignment(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller ? controller->AnimationsPreserveAxisAlignment() : true;
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->AnimationsPreserveAxisAlignment()
+             : true;
 }
 
 bool AnimationHost::MaximumTargetScale(int layer_id,
                                        LayerTreeType tree_type,
                                        float* max_scale) const {
   *max_scale = 0.f;
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller
-             ? controller->MaximumTargetScale(
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->MaximumTargetScale(
                    ObserverTypeFromTreeType(tree_type), max_scale)
              : true;
 }
 
 bool AnimationHost::AnimationStartScale(int layer_id,
                                         LayerTreeType tree_type,
                                         float* start_scale) const {
   *start_scale = 0.f;
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller
-             ? controller->AnimationStartScale(
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations
+             ? element_animations->AnimationStartScale(
                    ObserverTypeFromTreeType(tree_type), start_scale)
              : true;
 }
 
 bool AnimationHost::HasAnyAnimation(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller ? controller->has_any_animation() : false;
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations ? element_animations->has_any_animation() : false;
 }
 
 bool AnimationHost::HasActiveAnimationForTesting(int layer_id) const {
-  LayerAnimationController* controller = GetControllerForLayerId(layer_id);
-  return controller ? controller->HasActiveAnimation() : false;
+  auto element_animations = GetElementAnimationsForLayerId(layer_id);
+  return element_animations ? element_animations->HasActiveAnimation() : false;
 }
 
 void AnimationHost::ImplOnlyScrollAnimationCreate(
     int layer_id,
     const gfx::ScrollOffset& target_offset,
     const gfx::ScrollOffset& current_offset) {
   DCHECK(scroll_offset_animations_);
   scroll_offset_animations_->ScrollAnimationCreate(layer_id, target_offset,
                                                    current_offset);
 }
 
 bool AnimationHost::ImplOnlyScrollAnimationUpdateTarget(
     int layer_id,
     const gfx::Vector2dF& scroll_delta,
     const gfx::ScrollOffset& max_scroll_offset,
     base::TimeTicks frame_monotonic_time) {
   DCHECK(scroll_offset_animations_);
   return scroll_offset_animations_->ScrollAnimationUpdateTarget(
       layer_id, scroll_delta, max_scroll_offset, frame_monotonic_time);
 }
 
 void AnimationHost::ScrollAnimationAbort(bool needs_completion) {
   DCHECK(scroll_offset_animations_);
   return scroll_offset_animations_->ScrollAnimationAbort(needs_completion);
 }
 
-scoped_refptr<LayerAnimationController>
-AnimationHost::GetAnimationControllerForId(int id) {
-  scoped_refptr<LayerAnimationController> to_return;
-  if (!ContainsKey(all_animation_controllers_, id)) {
-    to_return = LayerAnimationController::Create(id);
-    to_return->SetAnimationHost(this);
-    all_animation_controllers_[id] = to_return.get();
-  } else {
-    to_return = all_animation_controllers_[id];
-  }
-  return to_return;
+void AnimationHost::DidActivateElementAnimations(
+    ElementAnimations* element_animations) {
+  DCHECK(element_animations->layer_id());
+  active_element_animations_map_[element_animations->layer_id()] =
+      element_animations;
 }
 
-void AnimationHost::DidActivateAnimationController(
-    LayerAnimationController* controller) {
-  active_animation_controllers_[controller->id()] = controller;
+void AnimationHost::DidDeactivateElementAnimations(
+    ElementAnimations* element_animations) {
+  DCHECK(element_animations->layer_id());
+  active_element_animations_map_.erase(element_animations->layer_id());
 }
 
-void AnimationHost::DidDeactivateAnimationController(
-    LayerAnimationController* controller) {
-  if (ContainsKey(active_animation_controllers_, controller->id()))
-    active_animation_controllers_.erase(controller->id());
+void AnimationHost::RegisterElementAnimations(
+    ElementAnimations* element_animations) {
+  DCHECK(element_animations->layer_id());
+  layer_to_element_animations_map_[element_animations->layer_id()] =
+      element_animations;
 }
 
-void AnimationHost::RegisterAnimationController(
-    LayerAnimationController* controller) {
-  all_animation_controllers_[controller->id()] = controller;
+void AnimationHost::UnregisterElementAnimations(
+    ElementAnimations* element_animations) {
+  DCHECK(element_animations->layer_id());
+  layer_to_element_animations_map_.erase(element_animations->layer_id());
+  DidDeactivateElementAnimations(element_animations);
 }
 
-void AnimationHost::UnregisterAnimationController(
-    LayerAnimationController* controller) {
-  if (ContainsKey(all_animation_controllers_, controller->id()))
-    all_animation_controllers_.erase(controller->id());
-  DidDeactivateAnimationController(controller);
+const AnimationHost::LayerToElementAnimationsMap&
+AnimationHost::active_element_animations_for_testing() const {
+  return active_element_animations_map_;
 }
 
-const AnimationHost::AnimationControllerMap&
-AnimationHost::active_animation_controllers_for_testing() const {
-  return active_animation_controllers_;
+const AnimationHost::LayerToElementAnimationsMap&
+AnimationHost::all_element_animations_for_testing() const {
+  return layer_to_element_animations_map_;
 }
 
-const AnimationHost::AnimationControllerMap&
-AnimationHost::all_animation_controllers_for_testing() const {
-  return all_animation_controllers_;
+void AnimationHost::OnAnimationWaitingForDeletion() {
+  animation_waiting_for_deletion_ = true;
 }
 
 }  // namespace cc