From c3540a056cbb3e5f03400a360905c2622c60bd93 Mon Sep 17 00:00:00 2001
From: mob-sakai <sakai861104@gmail.com>
Date: Fri, 14 Jun 2024 11:27:47 +0900
Subject: [PATCH] update coffee.internal

---
 .../Extensions/ComponentExtensions.cs         | 43 +++++++++++++++++++
 Runtime/Internal/Utilities/Logging.cs         |  4 +-
 .../Internal/Utilities/MaterialRepository.cs  | 28 ++++++++++++
 3 files changed, 74 insertions(+), 1 deletion(-)

diff --git a/Runtime/Internal/Extensions/ComponentExtensions.cs b/Runtime/Internal/Extensions/ComponentExtensions.cs
index f133e01..255ef6b 100644
--- a/Runtime/Internal/Extensions/ComponentExtensions.cs
+++ b/Runtime/Internal/Extensions/ComponentExtensions.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using UnityEditor;
 using UnityEngine;
 using UnityEngine.Profiling;
@@ -11,6 +12,48 @@ namespace Coffee.UIParticleInternal
     /// </summary>
     internal static class ComponentExtensions
     {
+        /// <summary>
+        /// Get components in children of a specific type in the hierarchy of a GameObject.
+        /// </summary>
+        public static T[] GetComponentsInChildren<T>(this Component self, int depth)
+            where T : Component
+        {
+            var results = ListPool<T>.Rent();
+            self.GetComponentsInChildren_Internal(results, depth);
+            var array = results.ToArray();
+            ListPool<T>.Return(ref results);
+            return array;
+        }
+
+        /// <summary>
+        /// Get components in children of a specific type in the hierarchy of a GameObject.
+        /// </summary>
+        public static void GetComponentsInChildren<T>(this Component self, List<T> results, int depth)
+            where T : Component
+        {
+            results.Clear();
+            self.GetComponentsInChildren_Internal(results, depth);
+        }
+
+        private static void GetComponentsInChildren_Internal<T>(this Component self, List<T> results, int depth)
+            where T : Component
+        {
+            if (!self || results == null || depth < 0) return;
+
+            var tr = self.transform;
+            if (tr.TryGetComponent<T>(out var t))
+            {
+                results.Add(t);
+            }
+
+            if (depth - 1 < 0) return;
+            var childCount = tr.childCount;
+            for (var i = 0; i < childCount; i++)
+            {
+                tr.GetChild(i).GetComponentsInChildren(results, depth - 1);
+            }
+        }
+
         /// <summary>
         /// Get or add a component of a specific type to a GameObject.
         /// </summary>
diff --git a/Runtime/Internal/Utilities/Logging.cs b/Runtime/Internal/Utilities/Logging.cs
index f03570e..d9a2f5e 100644
--- a/Runtime/Internal/Utilities/Logging.cs
+++ b/Runtime/Internal/Utilities/Logging.cs
@@ -1,11 +1,13 @@
 using System;
 using System.Text;
 using UnityEngine;
-using Conditional = System.Diagnostics.ConditionalAttribute;
 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
diff --git a/Runtime/Internal/Utilities/MaterialRepository.cs b/Runtime/Internal/Utilities/MaterialRepository.cs
index e9340a1..89ad52a 100644
--- a/Runtime/Internal/Utilities/MaterialRepository.cs
+++ b/Runtime/Internal/Utilities/MaterialRepository.cs
@@ -42,6 +42,34 @@ namespace Coffee.UIParticleInternal
             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>