import Coffee.Internal

Runtime/Internal/Utilities/FastAction.cs

@@ -0,0 +1,88 @@
+using System;
+using System.Collections.Generic;
+using UnityEngine;
+using UnityEngine.Profiling;
+
+namespace Coffee.UIParticleInternal
+{
+    /// <summary>
+    /// Base class for a fast action.
+    /// </summary>
+    internal class FastActionBase<T>
+    {
+        private static readonly ObjectPool<LinkedListNode<T>> s_NodePool =
+            new ObjectPool<LinkedListNode<T>>(() => new LinkedListNode<T>(default), _ => true, x => x.Value = default);
+
+        private readonly LinkedList<T> _delegates = new LinkedList<T>();
+
+        /// <summary>
+        /// Adds a delegate to the action.
+        /// </summary>
+        public void Add(T rhs)
+        {
+            if (rhs == null) return;
+            Profiler.BeginSample("(COF)[FastAction] Add Action");
+            var node = s_NodePool.Rent();
+            node.Value = rhs;
+            _delegates.AddLast(node);
+            Profiler.EndSample();
+        }
+
+        /// <summary>
+        /// Removes a delegate from the action.
+        /// </summary>
+        public void Remove(T rhs)
+        {
+            if (rhs == null) return;
+            Profiler.BeginSample("(COF)[FastAction] Remove Action");
+            var node = _delegates.Find(rhs);
+            if (node != null)
+            {
+                _delegates.Remove(node);
+                s_NodePool.Return(ref node);
+            }
+
+            Profiler.EndSample();
+        }
+
+        /// <summary>
+        /// Invokes the action with a callback function.
+        /// </summary>
+        protected void Invoke(Action<T> callback)
+        {
+            var node = _delegates.First;
+            while (node != null)
+            {
+                try
+                {
+                    callback(node.Value);
+                }
+                catch (Exception e)
+                {
+                    Debug.LogException(e);
+                }
+
+                node = node.Next;
+            }
+        }
+
+        public void Clear()
+        {
+            _delegates.Clear();
+        }
+    }
+
+    /// <summary>
+    /// A fast action without parameters.
+    /// </summary>
+    internal class FastAction : FastActionBase<Action>
+    {
+        /// <summary>
+        /// Invoke all the registered delegates.
+        /// </summary>
+        public void Invoke()
+        {
+            Invoke(action => action.Invoke());
+        }
+    }
+}

Runtime/Internal/Utilities/FastAction.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: a7c8c268a827b4787a8e050f1fe95ad5
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 

Runtime/Internal/Utilities/FrameCache.cs

@@ -0,0 +1,101 @@
+using System;
+using System.Collections.Generic;
+using UnityEngine;
+
+namespace Coffee.UIParticleInternal
+{
+    internal static class FrameCache
+    {
+        private static readonly Dictionary<Type, IFrameCache> s_Caches = new Dictionary<Type, IFrameCache>();
+
+        static FrameCache()
+        {
+            s_Caches.Clear();
+            UIExtraCallbacks.onLateAfterCanvasRebuild += ClearAllCache;
+        }
+
+#if UNITY_EDITOR
+        [RuntimeInitializeOnLoadMethod(RuntimeInitializeLoadType.SubsystemRegistration)]
+        private static void Clear()
+        {
+            s_Caches.Clear();
+        }
+#endif
+
+        /// <summary>
+        /// Tries to retrieve a value from the frame cache with a specified key.
+        /// </summary>
+        public static bool TryGet<T>(object key1, string key2, out T result)
+        {
+            return GetFrameCache<T>().TryGet((key1.GetHashCode(), key2.GetHashCode()), out result);
+        }
+
+        /// <summary>
+        /// Tries to retrieve a value from the frame cache with a specified key.
+        /// </summary>
+        public static bool TryGet<T>(object key1, string key2, int key3, out T result)
+        {
+            return GetFrameCache<T>().TryGet((key1.GetHashCode(), key2.GetHashCode() + key3), out result);
+        }
+
+        /// <summary>
+        /// Sets a value in the frame cache with a specified key.
+        /// </summary>
+        public static void Set<T>(object key1, string key2, T result)
+        {
+            GetFrameCache<T>().Set((key1.GetHashCode(), key2.GetHashCode()), result);
+        }
+
+        /// <summary>
+        /// Sets a value in the frame cache with a specified key.
+        /// </summary>
+        public static void Set<T>(object key1, string key2, int key3, T result)
+        {
+            GetFrameCache<T>().Set((key1.GetHashCode(), key2.GetHashCode() + key3), result);
+        }
+
+        private static void ClearAllCache()
+        {
+            foreach (var cache in s_Caches.Values)
+            {
+                cache.Clear();
+            }
+        }
+
+        private static FrameCacheContainer<T> GetFrameCache<T>()
+        {
+            var t = typeof(T);
+            if (s_Caches.TryGetValue(t, out var frameCache)) return frameCache as FrameCacheContainer<T>;
+
+            frameCache = new FrameCacheContainer<T>();
+            s_Caches.Add(t, frameCache);
+
+            return (FrameCacheContainer<T>)frameCache;
+        }
+
+        private interface IFrameCache
+        {
+            void Clear();
+        }
+
+        private class FrameCacheContainer<T> : IFrameCache
+        {
+            private readonly Dictionary<(int, int), T> _caches = new Dictionary<(int, int), T>();
+
+            public void Clear()
+            {
+                _caches.Clear();
+            }
+
+            public bool TryGet((int, int) key, out T result)
+            {
+                return _caches.TryGetValue(key, out result);
+            }
+
+            public void Set((int, int) key, T result)
+            {
+                _caches[key] = result;
+            }
+        }
+    }
+}

Runtime/Internal/Utilities/FrameCache.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 5f129e3b07ffb4d3bbb4cc5f6bd94087
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 

Runtime/Internal/Utilities/Logging.cs

@@ -0,0 +1,254 @@
+using System;
+using System.Text;
+using UnityEngine;
+using Object = UnityEngine.Object;
+#if ENABLE_COFFEE_LOGGER
+using System.Reflection;
+using System.Collections.Generic;
+#else
+using Conditional = System.Diagnostics.ConditionalAttribute;
+#endif
+
+namespace Coffee.UIParticleInternal
+{
+    internal static class Logging
+    {
+#if !ENABLE_COFFEE_LOGGER
+        private const string k_DisableSymbol = "DISABLE_COFFEE_LOGGER";
+
+        [Conditional(k_DisableSymbol)]
+#endif
+        private static void Log_Internal(LogType type, object tag, object message, Object context)
+        {
+#if ENABLE_COFFEE_LOGGER
+            AppendTag(s_Sb, tag);
+            s_Sb.Append(message);
+            switch (type)
+            {
+                case LogType.Error:
+                case LogType.Assert:
+                case LogType.Exception:
+                    Debug.LogError(s_Sb, context);
+                    break;
+                case LogType.Warning:
+                    Debug.LogWarning(s_Sb, context);
+                    break;
+                case LogType.Log:
+                    Debug.Log(s_Sb, context);
+                    break;
+            }
+
+            s_Sb.Length = 0;
+#endif
+        }
+
+#if !ENABLE_COFFEE_LOGGER
+        [Conditional(k_DisableSymbol)]
+#endif
+        public static void LogIf(bool enable, object tag, object message, Object context = null)
+        {
+            if (!enable) return;
+            Log_Internal(LogType.Log, tag, message, context ? context : tag as Object);
+        }
+
+#if !ENABLE_COFFEE_LOGGER
+        [Conditional(k_DisableSymbol)]
+#endif
+        public static void Log(object tag, object message, Object context = null)
+        {
+            Log_Internal(LogType.Log, tag, message, context ? context : tag as Object);
+        }
+
+#if !ENABLE_COFFEE_LOGGER
+        [Conditional(k_DisableSymbol)]
+#endif
+        public static void LogWarning(object tag, object message, Object context = null)
+        {
+            Log_Internal(LogType.Warning, tag, message, context ? context : tag as Object);
+        }
+
+        public static void LogError(object tag, object message, Object context = null)
+        {
+#if ENABLE_COFFEE_LOGGER
+            Log_Internal(LogType.Error, tag, message, context ? context : tag as Object);
+#else
+            Debug.LogError($"{tag}: {message}", context);
+#endif
+        }
+
+#if !ENABLE_COFFEE_LOGGER
+        [Conditional(k_DisableSymbol)]
+#endif
+        public static void LogMulticast(Type type, string fieldName, object instance = null, string message = null)
+        {
+#if ENABLE_COFFEE_LOGGER
+            AppendTag(s_Sb, instance ?? type);
+
+            var handler = type
+                .GetField(fieldName,
+                    BindingFlags.Static | BindingFlags.Instance | BindingFlags.Static | BindingFlags.NonPublic)
+                ?.GetValue(instance);
+
+            var list = ((MulticastDelegate)handler)?.GetInvocationList() ?? Array.Empty<Delegate>();
+            s_Sb.Append("<color=orange>");
+            s_Sb.Append(type.Name);
+            s_Sb.Append(".");
+            s_Sb.Append(fieldName);
+            s_Sb.Append(" has ");
+            s_Sb.Append(list.Length);
+            s_Sb.Append(" callbacks");
+            if (message != null)
+            {
+                s_Sb.Append(" (");
+                s_Sb.Append(message);
+                s_Sb.Append(")");
+            }
+
+            s_Sb.Append(":</color>");
+
+            for (var i = 0; i < list.Length; i++)
+            {
+                s_Sb.Append("\n - ");
+                s_Sb.Append(list[i].Method.DeclaringType?.Name);
+                s_Sb.Append(".");
+                s_Sb.Append(list[i].Method.Name);
+            }
+
+            Debug.Log(s_Sb);
+            s_Sb.Length = 0;
+#endif
+        }
+
+#if !ENABLE_COFFEE_LOGGER
+        [Conditional(k_DisableSymbol)]
+#endif
+        private static void AppendTag(StringBuilder sb, object tag)
+        {
+#if ENABLE_COFFEE_LOGGER
+            try
+            {
+                sb.Append("f");
+                sb.Append(Time.frameCount);
+                sb.Append(":<color=#");
+                AppendReadableCode(sb, tag);
+                sb.Append("><b>[");
+
+                switch (tag)
+                {
+                    case string name:
+                        sb.Append(name);
+                        break;
+                    case Type type:
+                        AppendType(sb, type);
+                        break;
+                    case Object uObject:
+                        AppendType(sb, tag.GetType());
+                        sb.Append(" #");
+                        sb.Append(uObject.name);
+                        break;
+                    default:
+                        AppendType(sb, tag.GetType());
+                        break;
+                }
+
+                sb.Append("]</b></color> ");
+            }
+            catch
+            {
+                sb.Append("f");
+                sb.Append(Time.frameCount);
+                sb.Append(":<b>[");
+                sb.Append(tag);
+                sb.Append("]</b> ");
+            }
+#endif
+        }
+
+#if !ENABLE_COFFEE_LOGGER
+        [Conditional(k_DisableSymbol)]
+#endif
+        private static void AppendType(StringBuilder sb, Type type)
+        {
+#if ENABLE_COFFEE_LOGGER
+            if (s_TypeNameCache.TryGetValue(type, out var name))
+            {
+                sb.Append(name);
+                return;
+            }
+
+            // New type found
+            var start = sb.Length;
+            if (0 < start && sb[start - 1] == '<' && (type.Name == "Material" || type.Name == "Color"))
+            {
+                sb.Append('@');
+            }
+
+            sb.Append(type.Name);
+            if (type.IsGenericType)
+            {
+                sb.Length -= 2;
+                sb.Append("<");
+                foreach (var gType in type.GetGenericArguments())
+                {
+                    AppendType(sb, gType);
+                    sb.Append(", ");
+                }
+
+                sb.Length -= 2;
+                sb.Append(">");
+            }
+
+            s_TypeNameCache.Add(type, sb.ToString(start, sb.Length - start));
+#endif
+        }
+
+#if !ENABLE_COFFEE_LOGGER
+        [Conditional(k_DisableSymbol)]
+#endif
+        private static void AppendReadableCode(StringBuilder sb, object tag)
+        {
+#if ENABLE_COFFEE_LOGGER
+            int hash;
+            try
+            {
+                switch (tag)
+                {
+                    case string text:
+                        hash = text.GetHashCode();
+                        break;
+                    case Type type:
+                        type = type.IsGenericType ? type.GetGenericTypeDefinition() : type;
+                        hash = type.FullName?.GetHashCode() ?? 0;
+                        break;
+                    default:
+                        hash = tag.GetType().FullName?.GetHashCode() ?? 0;
+                        break;
+                }
+            }
+            catch
+            {
+                sb.Append("FFFFFF");
+                return;
+            }
+
+            hash = hash & (s_Codes.Length - 1);
+            if (s_Codes[hash] == null)
+            {
+                var hue = hash / (float)s_Codes.Length;
+                var modifier = 1f - Mathf.Clamp01(Mathf.Abs(hue - 0.65f) / 0.2f);
+                var saturation = 0.7f + modifier * -0.2f;
+                var value = 0.8f + modifier * 0.3f;
+                s_Codes[hash] = ColorUtility.ToHtmlStringRGB(Color.HSVToRGB(hue, saturation, value));
+            }
+
+            sb.Append(s_Codes[hash]);
+#endif
+        }
+
+#if ENABLE_COFFEE_LOGGER
+        private static readonly StringBuilder s_Sb = new StringBuilder();
+        private static readonly string[] s_Codes = new string[64];
+        private static readonly Dictionary<Type, string> s_TypeNameCache = new Dictionary<Type, string>();
+#endif
+    }
+}

Runtime/Internal/Utilities/Logging.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 8255313895da84e7cbdc876be3795334
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 

Runtime/Internal/Utilities/MaterialRepository.cs

@@ -0,0 +1,92 @@
+using System;
+using UnityEngine;
+using UnityEngine.Profiling;
+
+namespace Coffee.UIParticleInternal
+{
+    /// <summary>
+    /// Provides functionality to manage materials.
+    /// </summary>
+    internal static class MaterialRepository
+    {
+        private static readonly ObjectRepository<Material> s_Repository = new ObjectRepository<Material>();
+
+        public static int count => s_Repository.count;
+
+#if UNITY_EDITOR
+        [RuntimeInitializeOnLoadMethod(RuntimeInitializeLoadType.SubsystemRegistration)]
+        private static void Clear()
+        {
+            s_Repository.Clear();
+        }
+#endif
+
+        /// <summary>
+        /// Retrieves a cached material based on the hash.
+        /// </summary>
+        public static bool Valid(Hash128 hash, Material material)
+        {
+            Profiler.BeginSample("(COF)[MaterialRegistry] Valid");
+            var ret = s_Repository.Valid(hash, material);
+            Profiler.EndSample();
+            return ret;
+        }
+
+        /// <summary>
+        /// Adds or retrieves a cached material based on the hash.
+        /// </summary>
+        public static void Get(Hash128 hash, ref Material material, Func<Material> onCreate)
+        {
+            Profiler.BeginSample("(COF)[MaterialRepository] Get");
+            s_Repository.Get(hash, ref material, onCreate);
+            Profiler.EndSample();
+        }
+
+        /// <summary>
+        /// Adds or retrieves a cached material based on the hash.
+        /// </summary>
+        public static void Get(Hash128 hash, ref Material material, string shaderName)
+        {
+            Profiler.BeginSample("(COF)[MaterialRepository] Get");
+            s_Repository.Get(hash, ref material, x => new Material(Shader.Find(x))
+            {
+                hideFlags = HideFlags.DontSave | HideFlags.NotEditable
+            }, shaderName);
+            Profiler.EndSample();
+        }
+
+        /// <summary>
+        /// Adds or retrieves a cached material based on the hash.
+        /// </summary>
+        public static void Get(Hash128 hash, ref Material material, string shaderName, string[] keywords)
+        {
+            Profiler.BeginSample("(COF)[MaterialRepository] Get");
+            s_Repository.Get(hash, ref material, x => new Material(Shader.Find(x.shaderName))
+            {
+                hideFlags = HideFlags.DontSave | HideFlags.NotEditable,
+                shaderKeywords = x.keywords
+            }, (shaderName, keywords));
+            Profiler.EndSample();
+        }
+
+        /// <summary>
+        /// Adds or retrieves a cached material based on the hash.
+        /// </summary>
+        public static void Get<T>(Hash128 hash, ref Material material, Func<T, Material> onCreate, T source)
+        {
+            Profiler.BeginSample("(COF)[MaterialRepository] Get");
+            s_Repository.Get(hash, ref material, onCreate, source);
+            Profiler.EndSample();
+        }
+
+        /// <summary>
+        /// Removes a soft mask material from the cache.
+        /// </summary>
+        public static void Release(ref Material material)
+        {
+            Profiler.BeginSample("(COF)[MaterialRepository] Release");
+            s_Repository.Release(ref material);
+            Profiler.EndSample();
+        }
+    }
+}

Runtime/Internal/Utilities/MaterialRepository.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 702912f2ee2ec49bb8003a64151ae4f7
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 

Runtime/Internal/Utilities/ObjectPool.cs

@@ -0,0 +1,85 @@
+using System;
+using System.Collections.Generic;
+
+namespace Coffee.UIParticleInternal
+{
+    /// <summary>
+    /// Object pool.
+    /// </summary>
+    internal class ObjectPool<T>
+    {
+        private readonly Func<T> _onCreate; // Delegate for creating instances
+        private readonly Action<T> _onReturn; // Delegate for returning instances to the pool
+        private readonly Predicate<T> _onValid; // Delegate for checking if instances are valid
+        private readonly Stack<T> _pool = new Stack<T>(32); // Object pool
+        private int _count; // Total count of created instances
+
+        public ObjectPool(Func<T> onCreate, Predicate<T> onValid, Action<T> onReturn)
+        {
+            _onCreate = onCreate;
+            _onValid = onValid;
+            _onReturn = onReturn;
+        }
+
+        /// <summary>
+        /// Rent an instance from the pool.
+        /// When you no longer need it, return it with <see cref="Return" />.
+        /// </summary>
+        public T Rent()
+        {
+            while (0 < _pool.Count)
+            {
+                var instance = _pool.Pop();
+                if (_onValid(instance))
+                {
+                    return instance;
+                }
+            }
+
+            // If there are no instances in the pool, create a new one.
+            Logging.Log(this, $"A new instance is created (pooled: {_pool.Count}, created: {++_count}).");
+            return _onCreate();
+        }
+
+        /// <summary>
+        /// Return an instance to the pool and assign null.
+        /// Be sure to return the instance obtained with <see cref="Rent" /> with this method.
+        /// </summary>
+        public void Return(ref T instance)
+        {
+            if (instance == null || _pool.Contains(instance)) return; // Ignore if already pooled or null.
+
+            _onReturn(instance); // Return the instance to the pool.
+            _pool.Push(instance);
+            Logging.Log(this, $"An instance is released (pooled: {_pool.Count}, created: {_count}).");
+            instance = default; // Set the reference to null.
+        }
+    }
+
+    /// <summary>
+    /// Object pool for <see cref="List{T}" />.
+    /// </summary>
+    internal static class ListPool<T>
+    {
+        private static readonly ObjectPool<List<T>> s_ListPool =
+            new ObjectPool<List<T>>(() => new List<T>(), _ => true, x => x.Clear());
+
+        /// <summary>
+        /// Rent an instance from the pool.
+        /// When you no longer need it, return it with <see cref="Return" />.
+        /// </summary>
+        public static List<T> Rent()
+        {
+            return s_ListPool.Rent();
+        }
+
+        /// <summary>
+        /// Return an instance to the pool and assign null.
+        /// Be sure to return the instance obtained with <see cref="Rent" /> with this method.
+        /// </summary>
+        public static void Return(ref List<T> toRelease)
+        {
+            s_ListPool.Return(ref toRelease);
+        }
+    }
+}

Runtime/Internal/Utilities/ObjectPool.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 632cb1ba34e6a4e80b55a32bb63ca369
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 

Runtime/Internal/Utilities/ObjectRepository.cs

@@ -0,0 +1,209 @@
+using System;
+using System.Collections.Generic;
+using UnityEngine;
+using UnityEngine.Profiling;
+using Object = UnityEngine.Object;
+
+namespace Coffee.UIParticleInternal
+{
+    internal class ObjectRepository<T> where T : Object
+    {
+        private readonly Dictionary<Hash128, Entry> _cache = new Dictionary<Hash128, Entry>(8);
+        private readonly Dictionary<int, Hash128> _objectKey = new Dictionary<int, Hash128>(8);
+        private readonly string _name;
+        private readonly Action<T> _onRelease;
+        private readonly Stack<Entry> _pool = new Stack<Entry>(8);
+
+        public ObjectRepository(Action<T> onRelease = null)
+        {
+            _name = $"{typeof(T).Name}Repository";
+            if (onRelease == null)
+            {
+                _onRelease = x =>
+                {
+#if UNITY_EDITOR
+                    if (!Application.isPlaying)
+                    {
+                        Object.DestroyImmediate(x, false);
+                    }
+                    else
+#endif
+                    {
+                        Object.Destroy(x);
+                    }
+                };
+            }
+            else
+            {
+                _onRelease = onRelease;
+            }
+
+            for (var i = 0; i < 8; i++)
+            {
+                _pool.Push(new Entry());
+            }
+        }
+
+        public int count => _cache.Count;
+
+        public void Clear()
+        {
+            foreach (var kv in _cache)
+            {
+                var entry = kv.Value;
+                if (entry == null) continue;
+
+                entry.Release(_onRelease);
+                _pool.Push(entry);
+            }
+
+            _cache.Clear();
+            _objectKey.Clear();
+        }
+
+        public bool Valid(Hash128 hash, T obj)
+        {
+            return _cache.TryGetValue(hash, out var entry) && entry.storedObject == obj;
+        }
+
+        /// <summary>
+        /// Adds or retrieves a cached object based on the hash.
+        /// </summary>
+        public void Get(Hash128 hash, ref T obj, Func<T> onCreate)
+        {
+            if (GetFromCache(hash, ref obj)) return;
+            Add(hash, ref obj, onCreate());
+        }
+
+        /// <summary>
+        /// Adds or retrieves a cached object based on the hash.
+        /// </summary>
+        public void Get<TS>(Hash128 hash, ref T obj, Func<TS, T> onCreate, TS source)
+        {
+            if (GetFromCache(hash, ref obj)) return;
+            Add(hash, ref obj, onCreate(source));
+        }
+
+        private bool GetFromCache(Hash128 hash, ref T obj)
+        {
+            // Find existing entry.
+            Profiler.BeginSample("(COF)[ObjectRepository] GetFromCache");
+            if (_cache.TryGetValue(hash, out var entry))
+            {
+                if (!entry.storedObject)
+                {
+                    Release(ref entry.storedObject);
+                    Profiler.EndSample();
+                    return false;
+                }
+
+                if (entry.storedObject != obj)
+                {
+                    // if the object is different, release the old one.
+                    Release(ref obj);
+                    ++entry.reference;
+                    obj = entry.storedObject;
+                    Logging.Log(_name, $"Get(total#{count}): {entry}");
+                }
+
+                Profiler.EndSample();
+                return true;
+            }
+
+            Profiler.EndSample();
+            return false;
+        }
+
+        private void Add(Hash128 hash, ref T obj, T newObject)
+        {
+            if (!newObject)
+            {
+                Release(ref obj);
+                obj = newObject;
+                return;
+            }
+
+            // Create and add a new entry.
+            Profiler.BeginSample("(COF)[ObjectRepository] Add");
+            var newEntry = 0 < _pool.Count ? _pool.Pop() : new Entry();
+            newEntry.storedObject = newObject;
+            newEntry.hash = hash;
+            newEntry.reference = 1;
+            _cache[hash] = newEntry;
+            _objectKey[newObject.GetInstanceID()] = hash;
+            Logging.Log(_name, $"<color=#03c700>Add</color>(total#{count}): {newEntry}");
+            Release(ref obj);
+            obj = newObject;
+            Profiler.EndSample();
+        }
+
+        /// <summary>
+        /// Release a object.
+        /// </summary>
+        public void Release(ref T obj)
+        {
+            if (ReferenceEquals(obj, null)) return;
+
+            // Find and release the entry.
+            Profiler.BeginSample("(COF)[ObjectRepository] Release");
+            var id = obj.GetInstanceID();
+            if (_objectKey.TryGetValue(id, out var hash)
+                && _cache.TryGetValue(hash, out var entry))
+            {
+                entry.reference--;
+                if (entry.reference <= 0 || !entry.storedObject)
+                {
+                    Remove(entry);
+                }
+                else
+                {
+                    Logging.Log(_name, $"Release(total#{_cache.Count}): {entry}");
+                }
+            }
+            else
+            {
+                Logging.Log(_name, $"Release(total#{_cache.Count}): <color=red>Already released: {obj}</color>");
+            }
+
+            obj = null;
+            Profiler.EndSample();
+        }
+
+        private void Remove(Entry entry)
+        {
+            if (ReferenceEquals(entry, null)) return;
+
+            Profiler.BeginSample("(COF)[ObjectRepository] Remove");
+            _cache.Remove(entry.hash);
+            _objectKey.Remove(entry.storedObject.GetInstanceID());
+            _pool.Push(entry);
+            entry.reference = 0;
+            Logging.Log(_name, $"<color=#f29e03>Remove</color>(total#{_cache.Count}): {entry}");
+            entry.Release(_onRelease);
+            Profiler.EndSample();
+        }
+
+        private class Entry
+        {
+            public Hash128 hash;
+            public int reference;
+            public T storedObject;
+
+            public void Release(Action<T> onRelease)
+            {
+                reference = 0;
+                if (storedObject)
+                {
+                    onRelease?.Invoke(storedObject);
+                }
+
+                storedObject = null;
+            }
+
+            public override string ToString()
+            {
+                return $"h{(uint)hash.GetHashCode()} (refs#{reference}), {storedObject}";
+            }
+        }
+    }
+}

Runtime/Internal/Utilities/ObjectRepository.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: a713d67bdb31e45e296e5f18460717e2
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 

Runtime/Internal/Utilities/UIExtraCallbacks.cs

@@ -0,0 +1,93 @@
+using System;
+using UnityEditor;
+using UnityEngine;
+using UnityEngine.UI;
+
+namespace Coffee.UIParticleInternal
+{
+    /// <summary>
+    /// Provides additional callbacks related to canvas and UI system.
+    /// </summary>
+    internal static class UIExtraCallbacks
+    {
+        private static bool s_IsInitializedAfterCanvasRebuild;
+        private static readonly FastAction s_AfterCanvasRebuildAction = new FastAction();
+        private static readonly FastAction s_LateAfterCanvasRebuildAction = new FastAction();
+        private static readonly FastAction s_BeforeCanvasRebuildAction = new FastAction();
+
+        static UIExtraCallbacks()
+        {
+            Canvas.willRenderCanvases += OnBeforeCanvasRebuild;
+            Logging.LogMulticast(typeof(Canvas), "willRenderCanvases", message: "ctor");
+        }
+
+        /// <summary>
+        /// Event that occurs after canvas rebuilds.
+        /// </summary>
+        public static event Action onLateAfterCanvasRebuild
+        {
+            add => s_LateAfterCanvasRebuildAction.Add(value);
+            remove => s_LateAfterCanvasRebuildAction.Remove(value);
+        }
+
+        /// <summary>
+        /// Event that occurs before canvas rebuilds.
+        /// </summary>
+        public static event Action onBeforeCanvasRebuild
+        {
+            add => s_BeforeCanvasRebuildAction.Add(value);
+            remove => s_BeforeCanvasRebuildAction.Remove(value);
+        }
+
+        /// <summary>
+        /// Event that occurs after canvas rebuilds.
+        /// </summary>
+        public static event Action onAfterCanvasRebuild
+        {
+            add => s_AfterCanvasRebuildAction.Add(value);
+            remove => s_AfterCanvasRebuildAction.Remove(value);
+        }
+
+        /// <summary>
+        /// Initializes the UIExtraCallbacks to ensure proper event handling.
+        /// </summary>
+        private static void InitializeAfterCanvasRebuild()
+        {
+            if (s_IsInitializedAfterCanvasRebuild) return;
+            s_IsInitializedAfterCanvasRebuild = true;
+
+            CanvasUpdateRegistry.IsRebuildingLayout();
+            Canvas.willRenderCanvases += OnAfterCanvasRebuild;
+            Logging.LogMulticast(typeof(Canvas), "willRenderCanvases",
+                message: "InitializeAfterCanvasRebuild");
+        }
+
+#if UNITY_EDITOR
+        [InitializeOnLoadMethod]
+#endif
+        [RuntimeInitializeOnLoadMethod(RuntimeInitializeLoadType.BeforeSceneLoad)]
+        private static void InitializeOnLoad()
+        {
+            Canvas.willRenderCanvases -= OnAfterCanvasRebuild;
+            s_IsInitializedAfterCanvasRebuild = false;
+        }
+
+        /// <summary>
+        /// Callback method called before canvas rebuilds.
+        /// </summary>
+        private static void OnBeforeCanvasRebuild()
+        {
+            s_BeforeCanvasRebuildAction.Invoke();
+            InitializeAfterCanvasRebuild();
+        }
+
+        /// <summary>
+        /// Callback method called after canvas rebuilds.
+        /// </summary>
+        private static void OnAfterCanvasRebuild()
+        {
+            s_AfterCanvasRebuildAction.Invoke();
+            s_LateAfterCanvasRebuildAction.Invoke();
+        }
+    }
+}

Runtime/Internal/Utilities/UIExtraCallbacks.cs.meta

@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 9ea318e6e3e6c46aa97c72e28230bdc9
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: