using UnityEditor;
using UnityEngine;
using System.Collections.Generic;

[InitializeOnLoad]
class HierarchyIconDisplay
{
    static Texture2D texture;
    static List<int> markedObjects;

    static HierarchyIconDisplay()
    {
        //替换自己的图片
        texture = AssetDatabase.LoadAssetAtPath("Assets/Resources/radioButton.png", typeof(Texture2D)) as Texture2D;
        EditorApplication.update += UpdateCB;
        EditorApplication.hierarchyWindowItemOnGUI += HierarchyItemCB;
    }

    static void UpdateCB()
    {
        GameObject[] go = Object.FindObjectsOfType(typeof(GameObject)) as GameObject[];
        markedObjects = new List<int>();
        foreach (GameObject g in go)
        {
            //自定义逻辑
            if (!string.IsNullOrEmpty(g.tag)&&g.tag == "UIProperty")
                markedObjects.Add(g.GetInstanceID());
        }

    }

    static void HierarchyItemCB(int instanceID, Rect selectionRect)
    {
        Rect r = new Rect(selectionRect);
        //我用的Unity 5.6.0f3版本，我这里需要再加一个r.x才能保持一致，和Unity论坛里面的答案不一样，可能是新旧版本导致的不同吧。
        r.x = r.x + r.width - 16;
        r.width = 16;
        if (markedObjects.Contains(instanceID))
        {
            //可以替换成GUI.Button等。。。
            GUI.Label(r, texture);
        }
    }

}