always invoke onRender via UIManagerNativeAnimatedDelegate (#54211)

Summary:

## Changelog:

[Internal] [Added] - always invoke onRender via UIManagerNativeAnimatedDelegate

Previously only android invokes `onRender` via UIManagerNativeAnimatedDelegate::runAnimationFrame()
consolidating so it's easier for debugging, also in cases where we want to add extra work per frame (e.g. add performance logging) in AnimatedNodesManagerProvider

Reviewed By: sammy-SC

Differential Revision: D85071075

Author: zeyap

packages/react-native/ReactCommon/react/renderer/animated/NativeAnimatedNodesManager.cpp

@@ -73,7 +73,7 @@ thread_local bool NativeAnimatedNodesManager::isOnRenderThread_{false};
 NativeAnimatedNodesManager::NativeAnimatedNodesManager(
     DirectManipulationCallback&& directManipulationCallback,
     FabricCommitCallback&& fabricCommitCallback,
-    StartOnRenderCallback&& startOnRenderCallback,
+    StartOnRenderCallbackInternal&& startOnRenderCallback,
     StopOnRenderCallback&& stopOnRenderCallback) noexcept
     : directManipulationCallback_(std::move(directManipulationCallback)),
       fabricCommitCallback_(std::move(fabricCommitCallback)),
@@ -539,7 +539,7 @@ void NativeAnimatedNodesManager::startRenderCallbackIfNeeded() {
   }
 
   if (startOnRenderCallback_) {
-    startOnRenderCallback_([this]() { onRender(); });
+    startOnRenderCallback_();
   }
 }
 

packages/react-native/ReactCommon/react/renderer/animated/NativeAnimatedNodesManager.h

@@ -62,12 +62,13 @@ class NativeAnimatedNodesManager {
   using FabricCommitCallback =
       std::function<void(std::unordered_map<Tag, folly::dynamic>&)>;
   using StartOnRenderCallback = std::function<void(std::function<void()>&&)>;
+  using StartOnRenderCallbackInternal = std::function<void()>;
   using StopOnRenderCallback = std::function<void()>;
 
   explicit NativeAnimatedNodesManager(
       DirectManipulationCallback&& directManipulationCallback,
       FabricCommitCallback&& fabricCommitCallback,
-      StartOnRenderCallback&& startOnRenderCallback = nullptr,
+      StartOnRenderCallbackInternal&& startOnRenderCallback = nullptr,
       StopOnRenderCallback&& stopOnRenderCallback = nullptr) noexcept;
 
   explicit NativeAnimatedNodesManager(
@@ -259,7 +260,7 @@ class NativeAnimatedNodesManager {
    * Tracks whether the render callback loop for animations is currently active.
    */
   std::atomic_bool isRenderCallbackStarted_{false};
-  const StartOnRenderCallback startOnRenderCallback_;
+  const StartOnRenderCallbackInternal startOnRenderCallback_;
   const StopOnRenderCallback stopOnRenderCallback_;
 
   std::shared_ptr<EventEmitterListener> eventEmitterListener_{nullptr};

packages/react-native/ReactCommon/react/renderer/animated/NativeAnimatedNodesManagerProvider.cpp

@@ -18,9 +18,7 @@
 
 namespace facebook::react {
 
-UIManagerNativeAnimatedDelegateImpl::UIManagerNativeAnimatedDelegateImpl(
-    std::weak_ptr<NativeAnimatedNodesManager> nativeAnimatedNodesManager)
-    : nativeAnimatedNodesManager_(std::move(nativeAnimatedNodesManager)) {}
+UIManagerNativeAnimatedDelegateImpl::UIManagerNativeAnimatedDelegateImpl() {}
 
 void UIManagerNativeAnimatedDelegateImpl::runAnimationFrame() {
   if (auto nativeAnimatedNodesManagerStrong =
@@ -67,6 +65,9 @@ NativeAnimatedNodesManagerProvider::getOrCreate(
           uiManager->synchronouslyUpdateViewOnUIThread(viewTag, props);
         };
 
+    nativeAnimatedDelegate_ =
+        std::make_shared<UIManagerNativeAnimatedDelegateImpl>();
+
     if (ReactNativeFeatureFlags::useSharedAnimatedBackend()) {
 #ifdef RN_USE_ANIMATION_BACKEND
       // TODO: this should be initialized outside of animated, but for now it
@@ -84,14 +85,27 @@ NativeAnimatedNodesManagerProvider::getOrCreate(
 
       uiManager->unstable_setAnimationBackend(animationBackend_);
     } else {
+      auto startOnRenderCallback =
+          [nativeAnimatedDelegate = nativeAnimatedDelegate_,
+           startOnRenderCallbackFn = std::move(startOnRenderCallback_)]() {
+            if (startOnRenderCallbackFn) {
+              startOnRenderCallbackFn([nativeAnimatedDelegate]() {
+                nativeAnimatedDelegate->runAnimationFrame();
+              });
+            }
+          };
       nativeAnimatedNodesManager_ =
           std::make_shared<NativeAnimatedNodesManager>(
               std::move(directManipulationCallback),
               std::move(fabricCommitCallback),
-              std::move(startOnRenderCallback_),
+              std::move(startOnRenderCallback),
               std::move(stopOnRenderCallback_));
     }
 
+    std::static_pointer_cast<UIManagerNativeAnimatedDelegateImpl>(
+        nativeAnimatedDelegate_)
+        ->setNativeAnimatedNodesManager(nativeAnimatedNodesManager_);
+
     addEventEmitterListener(
         nativeAnimatedNodesManager_->getEventEmitterListener());
 
@@ -112,10 +126,6 @@ NativeAnimatedNodesManagerProvider::getOrCreate(
           return false;
         }));
 
-    nativeAnimatedDelegate_ =
-        std::make_shared<UIManagerNativeAnimatedDelegateImpl>(
-            nativeAnimatedNodesManager_);
-
     uiManager->setNativeAnimatedDelegate(nativeAnimatedDelegate_);
 
     // TODO: remove force casting.

packages/react-native/ReactCommon/react/renderer/animated/NativeAnimatedNodesManagerProvider.h

@@ -13,20 +13,24 @@
 
 namespace facebook::react {
 
+class AnimatedMountingOverrideDelegate;
+
 class UIManagerNativeAnimatedDelegateImpl
     : public UIManagerNativeAnimatedDelegate {
  public:
-  explicit UIManagerNativeAnimatedDelegateImpl(
-      std::weak_ptr<NativeAnimatedNodesManager> nativeAnimatedNodesManager);
+  explicit UIManagerNativeAnimatedDelegateImpl();
 
   void runAnimationFrame() override;
 
+  void setNativeAnimatedNodesManager(
+      std::weak_ptr<NativeAnimatedNodesManager> manager) {
+    nativeAnimatedNodesManager_ = manager;
+  }
+
  private:
   std::weak_ptr<NativeAnimatedNodesManager> nativeAnimatedNodesManager_;
 };
 
-class AnimatedMountingOverrideDelegate;
-
 class NativeAnimatedNodesManagerProvider {
  public:
   NativeAnimatedNodesManagerProvider(