/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "EffectSet.h"

#include "ElementAnimationData.h"
#include "mozilla/LayerAnimationInfo.h"
#include "mozilla/PseudoStyleType.h"  // For PseudoStyleRequest
#include "mozilla/RestyleManager.h"
#include "mozilla/dom/Element.h"         // For Element
#include "nsCycleCollectionNoteChild.h"  // For CycleCollectionNoteChild
#include "nsLayoutUtils.h"
#include "nsPresContext.h"

namespace mozilla {

void EffectSet::Traverse(nsCycleCollectionTraversalCallback& aCallback) {
  for (const auto& key : mEffects) {
    CycleCollectionNoteChild(aCallback, key, "EffectSet::mEffects[]",
                             aCallback.Flags());
  }
}

/* static */
EffectSet* EffectSet::Get(const dom::Element* aElement,
                          const PseudoStyleRequest& aPseudoRequest) {
  if (auto* data = aElement->GetAnimationData()) {
    return data->GetEffectSetFor(aPseudoRequest);
  }
  return nullptr;
}

/* static */
EffectSet* EffectSet::GetOrCreate(dom::Element* aElement,
                                  const PseudoStyleRequest& aPseudoRequest) {
  return &aElement->EnsureAnimationData().EnsureEffectSetFor(aPseudoRequest);
}

/* static */
EffectSet* EffectSet::GetForFrame(const nsIFrame* aFrame,
                                  const nsCSSPropertyIDSet& aProperties) {
  MOZ_ASSERT(aFrame);

  // Transform animations are run on the primary frame (but stored on the
  // content associated with the style frame).
  const nsIFrame* frameToQuery = nullptr;
  if (aProperties.IsSubsetOf(nsCSSPropertyIDSet::TransformLikeProperties())) {
    // Make sure to return nullptr if we're looking for transform animations on
    // the inner table frame.
    if (!aFrame->SupportsCSSTransforms()) {
      return nullptr;
    }
    frameToQuery = nsLayoutUtils::GetStyleFrame(aFrame);
  } else {
    MOZ_ASSERT(
        !aProperties.Intersects(nsCSSPropertyIDSet::TransformLikeProperties()),
        "We should have only transform properties or no transform properties");
    // We don't need to explicitly return nullptr when |aFrame| is NOT the style
    // frame since there will be no effect set in that case.
    frameToQuery = aFrame;
  }

  Maybe<NonOwningAnimationTarget> target =
      EffectCompositor::GetAnimationElementAndPseudoForFrame(frameToQuery);
  return target ? Get(*target) : nullptr;
}

/* static */
EffectSet* EffectSet::GetForFrame(const nsIFrame* aFrame,
                                  DisplayItemType aDisplayItemType) {
  return EffectSet::GetForFrame(
      aFrame, LayerAnimationInfo::GetCSSPropertiesFor(aDisplayItemType));
}

/* static */
EffectSet* EffectSet::GetForStyleFrame(const nsIFrame* aStyleFrame) {
  Maybe<NonOwningAnimationTarget> target =
      EffectCompositor::GetAnimationElementAndPseudoForFrame(aStyleFrame);
  return target ? Get(*target) : nullptr;
}

/* static */
EffectSet* EffectSet::GetForEffect(const dom::KeyframeEffect* aEffect) {
  NonOwningAnimationTarget target = aEffect->GetAnimationTarget();
  return target ? Get(target) : nullptr;
}

/* static */
void EffectSet::DestroyEffectSet(dom::Element* aElement,
                                 const PseudoStyleRequest& aPseudoRequest) {
  if (auto* data = aElement->GetAnimationData()) {
    data->ClearEffectSetFor(aPseudoRequest);
  }
}

void EffectSet::UpdateAnimationGeneration(nsPresContext* aPresContext) {
  mAnimationGeneration =
      aPresContext->RestyleManager()->GetAnimationGeneration();
}

void EffectSet::AddEffect(dom::KeyframeEffect& aEffect) {
  if (!mEffects.EnsureInserted(&aEffect)) {
    return;
  }

  MarkCascadeNeedsUpdate();
}

void EffectSet::RemoveEffect(dom::KeyframeEffect& aEffect) {
  if (!mEffects.EnsureRemoved(&aEffect)) {
    return;
  }

  MarkCascadeNeedsUpdate();
}

}  // namespace mozilla