【m】1111

Assets/06.UIFarme/RunTime/UIFramework/Other/ObjectExtension.cs

@@ -0,0 +1,65 @@
+using System;
+using System.Collections.Generic;
+using UnityEngine;
+
+namespace Stary.Evo
+{
+    public static class ObjectExtension
+    {
+        private static readonly List<Transform> s_CachedTransforms = new List<Transform>();
+
+        public static T GetOrAddComponent<T>(this Component obj) where T : Component
+        {
+            return obj.gameObject.GetOrAddComponent<T>();
+        }
+
+        public static T GetOrAddComponent<T>(this GameObject gameObject) where T : Component
+        {
+            T t = gameObject.GetComponent<T>();
+            if (t == null)
+                t = gameObject.AddComponent<T>();
+            return t;
+        }
+
+        public static Component GetOrAddComponent(this Component obj, Type type)
+        {
+            return obj.gameObject.GetOrAddComponent(type);
+        }
+
+        public static Component GetOrAddComponent(this GameObject gameObject, Type type)
+        {
+            if (gameObject == null) return null;
+
+            Component component = gameObject.GetComponent(type);
+            if (component == null)
+                component = gameObject.AddComponent(type);
+            return component;
+        }
+
+        public static void SetParentEx(this Transform transform, Transform parent)
+        {
+            transform.SetParent(parent);
+            transform.localPosition = Vector3.zero;
+            transform.localRotation = Quaternion.identity;
+            transform.localScale = Vector3.one;
+        }
+
+        public static void SetLayerRecursively(this GameObject gameObject, int layer)
+        {
+            gameObject.GetComponentsInChildren(true, s_CachedTransforms);
+            for (int i = 0; i < s_CachedTransforms.Count; i++)
+            {
+                s_CachedTransforms[i].gameObject.layer = layer;
+            }
+            s_CachedTransforms.Clear();
+        }
+
+        public static void AddOrUpdate<TKey, TValue>(this IDictionary<TKey, TValue> dict, TKey key, TValue value)
+        {
+            if (dict.ContainsKey(key))
+                dict[key] = value;
+            else
+                dict.Add(key, value);
+        }
+    }
+}