Add tracing instrumentation to shared animation backend

Summary:
Add TraceSection annotations to the shared animated backend so the work it performs becomes visible in systrace / Perfetto / Android Studio System Trace captures.

Instrumented entry points:
- AnimationBackend::onAnimationFrame
- AnimationBackend::applySurfaceUpdates
- AnimationBackend::commitUpdates
- AnimationBackend::synchronouslyUpdateProps and its three platform variants
  (Buffered, Unbuffered, WithAnimatedProps)
- AnimationBackend::requestAsyncFlushForSurfaces
- AnimationBackendCommitHook::shadowTreeWillCommit
- AnimatedPropsRegistry::update
- AnimatedPropsRegistry::getMap

Each section includes contextual args (surfaceId / surface count / update count) where available to make traces easier to correlate when investigating animation perf.

Changelog:
[General][Added] - Add `TraceSection` instrumentation to the shared animation backend so its work shows up in systrace / Perfetto / Android Studio System Trace captures

Reviewed By: christophpurrer

Differential Revision: D102800179

Author: Bartlomiej Bloniarz

packages/react-native/ReactCommon/react/renderer/animationbackend/AnimatedPropsRegistry.cpp

@@ -6,13 +6,16 @@
  */
 
 #include "AnimatedPropsRegistry.h"
+#include <cxxreact/TraceSection.h>
 #include <react/renderer/core/PropsParserContext.h>
 #include "AnimatedProps.h"
 
 namespace facebook::react {
 
 void AnimatedPropsRegistry::update(
     const std::unordered_map<SurfaceId, SurfaceUpdates>& surfaceUpdates) {
+  TraceSection s(
+      "AnimatedPropsRegistry::update", "surfaceCount", surfaceUpdates.size());
   auto lock = std::lock_guard(mutex_);
   for (const auto& [surfaceId, updates] : surfaceUpdates) {
     auto& surfaceContext = surfaceContexts_[surfaceId];
@@ -59,6 +62,7 @@ std::pair<
     std::unordered_set<std::shared_ptr<const ShadowNodeFamily>>&,
     SnapshotMap&>
 AnimatedPropsRegistry::getMap(SurfaceId surfaceId) {
+  TraceSection s("AnimatedPropsRegistry::getMap", "surfaceId", surfaceId);
   auto lock = std::lock_guard(mutex_);
   auto& [pendingMap, map, pendingFamilies, families] =
       surfaceContexts_[surfaceId];

packages/react-native/ReactCommon/react/renderer/animationbackend/AnimationBackend.cpp

@@ -8,6 +8,7 @@
 #include "AnimationBackend.h"
 #include "AnimatedPropsRegistry.h"
 
+#include <cxxreact/TraceSection.h>
 #include <react/debug/react_native_assert.h>
 #include <react/featureflags/ReactNativeFeatureFlags.h>
 #include <react/renderer/animationbackend/AnimatedPropsSerializer.h>
@@ -78,6 +79,10 @@ void AnimationBackend::unpackMutations(
 void AnimationBackend::applySurfaceUpdates(
     std::unordered_map<SurfaceId, SurfaceUpdates>& surfaceUpdates,
     const std::set<SurfaceId>& asyncFlushSurfaces) {
+  TraceSection s(
+      "AnimationBackend::applySurfaceUpdates",
+      "surfaceCount",
+      surfaceUpdates.size());
   animatedPropsRegistry_->update(surfaceUpdates);
 
   for (auto& [surfaceId, updates] : surfaceUpdates) {
@@ -99,6 +104,7 @@ void AnimationBackend::applyMutations(AnimationMutations mutations) {
 }
 
 void AnimationBackend::onAnimationFrame(AnimationTimestamp timestamp) {
+  TraceSection s("AnimationBackend::onAnimationFrame");
   std::vector<CallbackWithId> callbacksCopy;
 
   {
@@ -158,6 +164,12 @@ void AnimationBackend::pushAnimationMutations(const Callback& callback) {
 void AnimationBackend::commitUpdates(
     SurfaceId surfaceId,
     SurfaceUpdates& surfaceUpdates) {
+  TraceSection s(
+      "AnimationBackend::commitUpdates",
+      "surfaceId",
+      surfaceId,
+      "updateCount",
+      surfaceUpdates.propsMap.size());
   auto uiManager = uiManager_.lock();
   if (!uiManager) {
     return;
@@ -195,6 +207,10 @@ void AnimationBackend::commitUpdates(
 
 void AnimationBackend::synchronouslyUpdatePropsUnbuffered(
     const std::unordered_map<Tag, AnimatedProps>& updates) {
+  TraceSection s(
+      "AnimationBackend::synchronouslyUpdatePropsUnbuffered",
+      "updateCount",
+      updates.size());
   for (auto& [tag, animatedProps] : updates) {
     auto dyn = animationbackend::packAnimatedProps(animatedProps);
     if (auto uiManager = uiManager_.lock()) {
@@ -206,6 +222,12 @@ void AnimationBackend::synchronouslyUpdatePropsUnbuffered(
 void AnimationBackend::synchronouslyUpdateProps(
     SurfaceId surfaceId,
     const std::unordered_map<Tag, AnimatedProps>& updates) {
+  TraceSection s(
+      "AnimationBackend::synchronouslyUpdateProps",
+      "surfaceId",
+      surfaceId,
+      "updateCount",
+      updates.size());
   if (ReactNativeFeatureFlags::optimizedAnimatedPropUpdates()) {
     if (auto uiManager = uiManager_.lock()) {
       uiManager->synchronouslyUpdateAnimatedPropsOnUIThread(surfaceId, updates);
@@ -217,6 +239,10 @@ void AnimationBackend::synchronouslyUpdateProps(
 
 void AnimationBackend::requestAsyncFlushForSurfaces(
     const std::set<SurfaceId>& surfaces) {
+  TraceSection s(
+      "AnimationBackend::requestAsyncFlushForSurfaces",
+      "surfaceCount",
+      surfaces.size());
   react_native_assert(
       jsInvoker_ != nullptr ||
       surfaces.empty() && "jsInvoker_ was not provided");

packages/react-native/ReactCommon/react/renderer/animationbackend/AnimationBackendCommitHook.cpp

@@ -7,6 +7,7 @@
 
 #include <react/renderer/animationbackend/AnimationBackendCommitHook.h>
 
+#include <cxxreact/TraceSection.h>
 #include <utility>
 
 namespace facebook::react {
@@ -23,6 +24,10 @@ RootShadowNode::Unshared AnimationBackendCommitHook::shadowTreeWillCommit(
     const RootShadowNode::Shared& oldRootShadowNode,
     const RootShadowNode::Unshared& newRootShadowNode,
     const ShadowTreeCommitOptions& commitOptions) noexcept {
+  TraceSection s(
+      "AnimationBackendCommitHook::shadowTreeWillCommit",
+      "surfaceId",
+      shadowTree.getSurfaceId());
   if (commitOptions.source != ShadowTreeCommitSource::React &&
       commitOptions.source != ShadowTreeCommitSource::AnimationEndSync) {
     return newRootShadowNode;