AnimationBackend.h

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#pragma once

#include <ReactCommon/CallInvoker.h>
#include <folly/dynamic.h>
#include <react/renderer/core/ReactPrimitives.h>
#include <react/renderer/uimanager/UIManager.h>
#include <react/renderer/uimanager/UIManagerAnimationBackend.h>
#include <functional>
#include <memory>
#include <set>
#include <vector>
#include "AnimatedProps.h"
#include "AnimatedPropsBuilder.h"
#include "AnimatedPropsRegistry.h"
#include "AnimationBackendCommitHook.h"

namespace facebook::react {

class AnimationBackend;

class UIManagerNativeAnimatedDelegateBackendImpl : public UIManagerNativeAnimatedDelegate {
 public:
  explicit UIManagerNativeAnimatedDelegateBackendImpl(std::weak_ptr<UIManagerAnimationBackend> animationBackend);

  void runAnimationFrame() override;

 private:
  std::weak_ptr<UIManagerAnimationBackend> animationBackend_;
};

struct AnimationMutation {
  Tag tag;
  std::shared_ptr<const ShadowNodeFamily> family;
  AnimatedProps props;
  bool hasLayoutUpdates{false};
};

struct AnimationMutations {
  std::vector<AnimationMutation> batch;
  std::set<SurfaceId> asyncFlushSurfaces;
};

class AnimationBackend : public UIManagerAnimationBackend {
 public:
  using Callback = std::function<AnimationMutations(float)>;
  using ResumeCallback = std::function<void()>;
  using PauseCallback = std::function<void()>;
  using DirectManipulationCallback = std::function<void(Tag, const folly::dynamic &)>;

  std::vector<Callback> callbacks;
  const DirectManipulationCallback directManipulationCallback_;
  std::shared_ptr<AnimatedPropsRegistry> animatedPropsRegistry_;
  std::shared_ptr<UIManager> uiManager_;
  std::shared_ptr<CallInvoker> jsInvoker_;
  AnimationBackendCommitHook commitHook_;
  bool isRenderCallbackStarted_{false};

  AnimationBackend(DirectManipulationCallback &&directManipulationCallback, std::shared_ptr<UIManager> uiManager);
  void commitUpdates(SurfaceId surfaceId, SurfaceUpdates &surfaceUpdates);
  void synchronouslyUpdateProps(const std::unordered_map<Tag, AnimatedProps> &updates);
  void requestAsyncFlushForSurfaces(const std::set<SurfaceId> &surfaces);
  void clearRegistry(SurfaceId surfaceId) override;
  void registerJSInvoker(std::shared_ptr<CallInvoker> jsInvoker) override;

  void onAnimationFrame(double timestamp) override;
  void trigger() override;
  void start(const Callback &callback, bool isAsync) override;
  void stop(bool isAsync) override;
};
} // namespace facebook::react