https://github.com/annulusgames/LitMotion/pull/240

Fixing LitMotion to work with Enter the Play Mode annulusgames#240

Author: duynt-hypersol

src/LitMotion/Assets/LitMotion.Animation/Editor/LitMotionAnimationEditor.cs

@@ -3,6 +3,7 @@
 using UnityEditor.UIElements;
 using System.Collections.Generic;
 using UnityEngine;
+using UnityEditor.SceneManagement;
 
 namespace LitMotion.Animation.Editor
 {

src/LitMotion/Assets/LitMotion.Animation/Runtime/Internal.meta

@@ -0,0 +1,43 @@
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+
+#if UNITY_EDITOR
+using UnityEditor;
+
+namespace LitMotion.Runtime.Internal
+{
+    internal class EnterThePlayModeHelper
+    {
+        private static readonly List<Action> _scheduledActions = new();
+
+        [InitializeOnEnterPlayMode]
+        private static void Reset()
+        {
+            foreach (var action in _scheduledActions)
+                action();
+            MotionDispatcher.Clear();
+        }
+        
+        [Conditional("UNITY_EDITOR")]
+        public static void Register<TValue, TOptions, TAdapter>(MotionStorage<TValue, TOptions, TAdapter> storage)
+            where TValue : unmanaged
+            where TOptions : unmanaged, IMotionOptions
+            where TAdapter : unmanaged, IMotionAdapter<TValue, TOptions>
+        {
+            if (EditorSettings.enterPlayModeOptionsEnabled)
+                _scheduledActions.Add(storage.Reset);
+        }
+        
+        [Conditional("UNITY_EDITOR")]
+        public static void Register<TValue, TOptions, TAdapter>(UpdateRunner<TValue, TOptions, TAdapter> runner)
+            where TValue : unmanaged
+            where TOptions : unmanaged, IMotionOptions
+            where TAdapter : unmanaged, IMotionAdapter<TValue, TOptions>
+        {
+            if (EditorSettings.enterPlayModeOptionsEnabled)
+                _scheduledActions.Add(runner.Reset);
+        }
+    }
+}
+#endif

src/LitMotion/Assets/LitMotion/Runtime/Internal/EnterThePlayModeHelper.cs

@@ -115,5 +115,14 @@ static void CheckTypeId(in MotionHandle handle)
                 throw new ArgumentException("Invalid type id.");
             }
         }
+
+#if UNITY_EDITOR
+        [UnityEditor.InitializeOnEnterPlayMode]
+        private static void Clear()
+        {
+            MotionTypeCount = 0;
+            list.Clear();
+        }
+#endif
     }
 }

src/LitMotion/Assets/LitMotion/Runtime/Internal/EnterThePlayModeHelper.cs.meta

@@ -4,6 +4,7 @@
 using LitMotion.Collections;
 
 #if UNITY_EDITOR
+using LitMotion.Runtime.Internal;
 using UnityEditor;
 #endif
 
@@ -51,6 +52,9 @@ static MotionStorage<TValue, TOptions, TAdapter> CreateIfNull(ref MotionStorage<
                 {
                     storage = new MotionStorage<TValue, TOptions, TAdapter>(MotionManager.MotionTypeCount);
                     MotionManager.Register(storage);
+#if UNITY_EDITOR
+                    EnterThePlayModeHelper.Register(storage);
+#endif
                 }
                 return storage;
             }
@@ -100,6 +104,9 @@ public static (UpdateRunner<TValue, TOptions, TAdapter> runner, bool isCreated)
                         runner = new UpdateRunner<TValue, TOptions, TAdapter>(storage, Time.timeAsDouble, Time.unscaledTimeAsDouble, Time.realtimeSinceStartupAsDouble);
                     }
                     GetRunnerList(playerLoopTiming).Add(runner);
+#if UNITY_EDITOR
+                    EnterThePlayModeHelper.Register(runner);
+#endif
                     return (runner, true);
                 }
                 return (runner, false);
@@ -251,6 +258,7 @@ public static MotionStorage<TValue, TOptions, TAdapter> GetOrCreateStorage()
                 {
                     storage = new MotionStorage<TValue, TOptions, TAdapter>(MotionManager.MotionTypeCount);
                     MotionManager.Register(storage);
+                    EnterThePlayModeHelper.Register(storage);
                 }
                 return storage;
             }