draft: testing screen transitions and SETs

Common/cpp/LayoutAnimations/LayoutAnimationsManager.cpp

@@ -8,34 +8,35 @@
 
 namespace reanimated {
 
-void LayoutAnimationsManager::configureAnimation(
-    const int tag,
-    const LayoutAnimationType type,
-    const std::string &sharedTransitionTag,
-    const std::shared_ptr<Shareable> &config) {
+void LayoutAnimationsManager::configureAnimationBatch(
+    const std::vector<LayoutAnimationConfig> &layoutAnimationsBatch) {
   auto lock = std::unique_lock<std::mutex>(animationsMutex_);
-  if (type == SHARED_ELEMENT_TRANSITION ||
-      type == SHARED_ELEMENT_TRANSITION_PROGRESS) {
+  std::vector<LayoutAnimationConfig> sharedTransitionConfigs;
+  for (auto layoutAnimationConfig : layoutAnimationsBatch) {
+    auto [tag, type, config, sharedTransitionTag] = layoutAnimationConfig;
+    if (type == SHARED_ELEMENT_TRANSITION ||
+        type == SHARED_ELEMENT_TRANSITION_PROGRESS) {
+      clearSharedTransitionConfig(tag);
+      sharedTransitionConfigs.push_back(std::move(layoutAnimationConfig));
+    } else {
+      if (config == nullptr) {
+        getConfigsForType(type).erase(tag);
+      } else {
+        getConfigsForType(type)[tag] = config;
+      }
+    }
+  }
+  for (auto [tag, type, config, sharedTransitionTag] :
+       sharedTransitionConfigs) {
+    if (config == nullptr) {
+      continue;
+    }
     sharedTransitionGroups_[sharedTransitionTag].push_back(tag);
     viewTagToSharedTag_[tag] = sharedTransitionTag;
     getConfigsForType(SHARED_ELEMENT_TRANSITION)[tag] = config;
     if (type == SHARED_ELEMENT_TRANSITION) {
       ignoreProgressAnimationForTag_.insert(tag);
     }
-  } else {
-    getConfigsForType(type)[tag] = config;
-  }
-}
-
-void LayoutAnimationsManager::configureAnimationBatch(
-    const std::vector<LayoutAnimationConfig> &layoutAnimationsBatch) {
-  auto lock = std::unique_lock<std::mutex>(animationsMutex_);
-  for (auto [tag, type, config] : layoutAnimationsBatch) {
-    if (config == nullptr) {
-      getConfigsForType(type).erase(tag);
-    } else {
-      getConfigsForType(type)[tag] = config;
-    }
   }
 }
 
@@ -66,12 +67,9 @@ bool LayoutAnimationsManager::hasLayoutAnimation(
   return collection::contains(getConfigsForType(type), tag);
 }
 
-void LayoutAnimationsManager::clearLayoutAnimationConfig(const int tag) {
-  auto lock = std::unique_lock<std::mutex>(animationsMutex_);
-  enteringAnimations_.erase(tag);
-  exitingAnimations_.erase(tag);
-  layoutAnimations_.erase(tag);
-  shouldAnimateExitingForTag_.erase(tag);
+void LayoutAnimationsManager::clearSharedTransitionConfig(const int tag) {
+  // here we don't lock the mutex, since this function is only supposed to be
+  // used as a helper
 #ifndef NDEBUG
   const auto &pair = viewsScreenSharedTagMap_[tag];
   screenSharedTagSet_.erase(pair);
@@ -81,6 +79,7 @@ void LayoutAnimationsManager::clearLayoutAnimationConfig(const int tag) {
   sharedTransitionAnimations_.erase(tag);
   auto const &groupName = viewTagToSharedTag_[tag];
   if (groupName.empty()) {
+    viewTagToSharedTag_.erase(tag);
     return;
   }
   auto &group = sharedTransitionGroups_[groupName];
@@ -95,6 +94,15 @@ void LayoutAnimationsManager::clearLayoutAnimationConfig(const int tag) {
   ignoreProgressAnimationForTag_.erase(tag);
 }
 
+void LayoutAnimationsManager::clearLayoutAnimationConfig(const int tag) {
+  auto lock = std::unique_lock<std::mutex>(animationsMutex_);
+  enteringAnimations_.erase(tag);
+  exitingAnimations_.erase(tag);
+  layoutAnimations_.erase(tag);
+  shouldAnimateExitingForTag_.erase(tag);
+  clearSharedTransitionConfig(tag);
+}
+
 void LayoutAnimationsManager::startLayoutAnimation(
     jsi::Runtime &rt,
     const int tag,

Common/cpp/LayoutAnimations/LayoutAnimationsManager.h

@@ -22,17 +22,13 @@ struct LayoutAnimationConfig {
   int tag;
   LayoutAnimationType type;
   std::shared_ptr<Shareable> config;
+  std::string sharedTransitionTag;
 };
 
 class LayoutAnimationsManager {
  public:
   explicit LayoutAnimationsManager(const std::shared_ptr<JSLogger> &jsLogger)
       : jsLogger_(jsLogger) {}
-  void configureAnimation(
-      const int tag,
-      const LayoutAnimationType type,
-      const std::string &sharedTransitionTag,
-      const std::shared_ptr<Shareable> &config);
   void configureAnimationBatch(
       const std::vector<LayoutAnimationConfig> &layoutAnimationsBatch);
   void setShouldAnimateExiting(const int tag, const bool value);
@@ -54,6 +50,7 @@ class LayoutAnimationsManager {
 #endif
 
  private:
+  void clearSharedTransitionConfig(const int tag);
   std::unordered_map<int, std::shared_ptr<Shareable>> &getConfigsForType(
       const LayoutAnimationType type);
 

Common/cpp/NativeModules/NativeReanimatedModule.cpp

@@ -293,23 +293,6 @@ jsi::Value NativeReanimatedModule::configureProps(
   return jsi::Value::undefined();
 }
 
-jsi::Value NativeReanimatedModule::configureLayoutAnimation(
-    jsi::Runtime &rt,
-    const jsi::Value &viewTag,
-    const jsi::Value &type,
-    const jsi::Value &sharedTransitionTag,
-    const jsi::Value &config) {
-  layoutAnimationsManager_.configureAnimation(
-      viewTag.asNumber(),
-      static_cast<LayoutAnimationType>(type.asNumber()),
-      sharedTransitionTag.asString(rt).utf8(rt),
-      extractShareableOrThrow<ShareableObject>(
-          rt,
-          config,
-          "[Reanimated] Layout animation config must be an object."));
-  return jsi::Value::undefined();
-}
-
 jsi::Value NativeReanimatedModule::configureLayoutAnimationBatch(
     jsi::Runtime &rt,
     const jsi::Value &layoutAnimationsBatch) {
@@ -331,6 +314,16 @@ jsi::Value NativeReanimatedModule::configureLayoutAnimationBatch(
           config,
           "[Reanimated] Layout animation config must be an object.");
     }
+    if (batch[i].type != SHARED_ELEMENT_TRANSITION &&
+        batch[i].type != SHARED_ELEMENT_TRANSITION_PROGRESS) {
+      continue;
+    }
+    auto sharedTransitionTag = item.getProperty(rt, "sharedTransitionTag");
+    if (sharedTransitionTag.isUndefined()) {
+      batch[i].config = nullptr;
+    } else {
+      batch[i].sharedTransitionTag = sharedTransitionTag.asString(rt).utf8(rt);
+    }
   }
   layoutAnimationsManager_.configureAnimationBatch(batch);
   return jsi::Value::undefined();

Common/cpp/NativeModules/NativeReanimatedModule.h

@@ -80,12 +80,6 @@ class NativeReanimatedModule : public NativeReanimatedModuleSpec {
       jsi::Runtime &rt,
       const jsi::Value &uiProps,
       const jsi::Value &nativeProps) override;
-  jsi::Value configureLayoutAnimation(
-      jsi::Runtime &rt,
-      const jsi::Value &viewTag,
-      const jsi::Value &type,
-      const jsi::Value &sharedTransitionTag,
-      const jsi::Value &config) override;
   jsi::Value configureLayoutAnimationBatch(
       jsi::Runtime &rt,
       const jsi::Value &layoutAnimationsBatch) override;

Common/cpp/NativeModules/NativeReanimatedModuleSpec.cpp

@@ -150,20 +150,6 @@ static jsi::Value SPEC_PREFIX(unsubscribeFromKeyboardEvents)(
   return jsi::Value::undefined();
 }
 
-static jsi::Value SPEC_PREFIX(configureLayoutAnimation)(
-    jsi::Runtime &rt,
-    TurboModule &turboModule,
-    const jsi::Value *args,
-    size_t) {
-  return static_cast<NativeReanimatedModuleSpec *>(&turboModule)
-      ->configureLayoutAnimation(
-          rt,
-          std::move(args[0]),
-          std::move(args[1]),
-          std::move(args[2]),
-          std::move(args[3]));
-}
-
 static jsi::Value SPEC_PREFIX(configureLayoutAnimationBatch)(
     jsi::Runtime &rt,
     TurboModule &turboModule,
@@ -214,8 +200,6 @@ NativeReanimatedModuleSpec::NativeReanimatedModuleSpec(
   methodMap_["unsubscribeFromKeyboardEvents"] =
       MethodMetadata{1, SPEC_PREFIX(unsubscribeFromKeyboardEvents)};
 
-  methodMap_["configureLayoutAnimation"] =
-      MethodMetadata{4, SPEC_PREFIX(configureLayoutAnimation)};
   methodMap_["configureLayoutAnimationBatch"] =
       MethodMetadata{1, SPEC_PREFIX(configureLayoutAnimationBatch)};
   methodMap_["setShouldAnimateExitingForTag"] =

Common/cpp/NativeModules/NativeReanimatedModuleSpec.h

@@ -92,13 +92,6 @@ class JSI_EXPORT NativeReanimatedModuleSpec : public TurboModule {
       const jsi::Value &nativeProps) = 0;
 
   // layout animations
-  virtual jsi::Value configureLayoutAnimation(
-      jsi::Runtime &rt,
-      const jsi::Value &viewTag,
-      const jsi::Value &type,
-      const jsi::Value &sharedTransitionTag,
-      const jsi::Value &config) = 0;
-
   virtual jsi::Value configureLayoutAnimationBatch(
       jsi::Runtime &rt,
       const jsi::Value &layoutAnimationsBatch) = 0;

android/src/main/java/com/swmansion/reanimated/layoutReanimation/AnimationsManager.java

@@ -712,8 +712,8 @@ private static Point convertScreenLocationToViewCoordinates(Point fromPoint, Vie
     return new Point(fromPoint.x - toPoint[0], fromPoint.y - toPoint[1]);
   }
 
-  public void screenDidLayout() {
-    mSharedTransitionManager.screenDidLayout();
+  public void screenDidLayout(View view) {
+    mSharedTransitionManager.screenDidLayout(view);
   }
 
   public void viewDidLayout(View view) {
@@ -724,6 +724,10 @@ public void notifyAboutViewsRemoval(int[] tagsToDelete) {
     mSharedTransitionManager.onViewsRemoval(tagsToDelete);
   }
 
+  public void notifyAboutScreenWillDisappear() {
+    mSharedTransitionManager.onScreenWillDisappear();
+  }
+
   public void makeSnapshotOfTopScreenViews(ViewGroup stack) {
     mSharedTransitionManager.doSnapshotForTopScreenViews(stack);
   }

android/src/main/java/com/swmansion/reanimated/layoutReanimation/ReanimatedNativeHierarchyManager.java

@@ -6,9 +6,11 @@
 import android.view.ViewGroup;
 import androidx.annotation.Nullable;
 import com.facebook.react.bridge.ReactApplicationContext;
+import com.facebook.react.bridge.ReactContext;
 import com.facebook.react.bridge.UiThreadUtil;
 import com.facebook.react.uimanager.IllegalViewOperationException;
 import com.facebook.react.uimanager.NativeViewHierarchyManager;
+import com.facebook.react.uimanager.UIManagerHelper;
 import com.facebook.react.uimanager.ViewAtIndex;
 import com.facebook.react.uimanager.ViewGroupManager;
 import com.facebook.react.uimanager.ViewManager;
@@ -164,6 +166,20 @@ public void deleteView(final View view, final LayoutAnimationListener listener)
       if (parentName.equals("RNSScreenStack")) {
         mAnimationsManager.cancelAnimationsInSubviews(view);
         super.deleteView(view, listener);
+        var eventDispatcher =
+            UIManagerHelper.getEventDispatcherForReactTag(
+                (ReactContext) view.getContext(), view.getId());
+        if (eventDispatcher != null) {
+          eventDispatcher.addListener(
+              event -> {
+                // we schedule the start of transition for the ScreenWilDisappear event, so that the
+                // layout of the target screen is already calculated
+                // this allows us to make snapshots on the go, so that they are always up-to-date
+                if (event.getEventName().equals("topWillDisappear")) {
+                  getAnimationsManager().notifyAboutScreenWillDisappear();
+                }
+              });
+        }
         return;
       }
     }
@@ -264,7 +280,7 @@ public synchronized void updateLayout(
       if (container != null && viewManagerName.equals("RNSScreen") && mReaLayoutAnimator != null) {
         boolean hasHeader = checkIfTopScreenHasHeader((ViewGroup) container);
         if (!hasHeader || !container.isLayoutRequested()) {
-          mReaLayoutAnimator.getAnimationsManager().screenDidLayout();
+          mReaLayoutAnimator.getAnimationsManager().screenDidLayout(container);
         }
       }
       View view = resolveView(tag);