给Unity的GameObject拓展接口

最新推荐文章于 2023-05-18 17:46:34 发布

原创最新推荐文章于 2023-05-18 17:46:34 发布 · 1.9k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#扩展 #unity #接口

本文介绍了一个Unity的扩展方法，该方法提供了一种便捷的方式来获取或添加GameObject上的组件。通过使用GetOrAddComponent泛型方法，可以简化游戏开发中组件管理的工作流程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

现在，我们想给GameObject拓展一些接口，我们知道GameObject有AddComponent和GetComponent接口

我们如何给GameObject拓展一个GetOrAddComponent接口呢？

代码如下：

using UnityEngine;

//Unity的一些扩展方法
static public class MethodExtensionForUnity
{
    /// <summary>
    /// Gets or add a component. Usage example:
    /// BoxCollider boxCollider = transform.GetOrAddComponent<BoxCollider>();
    /// </summary>
    static public T GetOrAddComponent<T>(this Component child, bool set_enable = false) where T : Component
    {
        T result = child.GetComponent<T>();
        if (result == null)
        {
            result = child.gameObject.AddComponent<T>();
        }
        var bcomp = result as Behaviour;
        if (set_enable)
        {
            if (bcomp != null) bcomp.enabled = true;
        }
        return result;
    }

    static public T GetOrAddComponent<T>(this GameObject go) where T : Component
    {
        T result = go.transform.GetComponent<T>();
        if (result == null)
        {
            result = go.AddComponent<T>();
        }
        var bcomp = result as Behaviour;
        if (bcomp != null) bcomp.enabled = true;
        return result;
    }


    public static void walk(this GameObject o, System.Action<GameObject> f)
    {
        f(o);

        int numChildren = o.transform.childCount;

        for (int i = 0; i < numChildren; ++i)
        {
            walk(o.transform.GetChild(i).gameObject, f);
        }
    }


}