unity perfab 资源检查

using UnityEngine;
using UnityEditor;
using System.IO;
using System.Collections.Generic;
using System.Text.RegularExpressions;


public class ResourceTool : EditorWindow
{


    [MenuItem("Test/Resource Tools")]
    static void ShowResourceTool()
    {
        //创建窗口
        Rect rc = new Rect(0, 0, 800, 600);
        ResourceTool window = (ResourceTool)EditorWindow.GetWindow(typeof(ResourceTool), true, "Test Resource Tool");
        window.Show();
    }


    private int selectionGridInt = 0;
    string[] selectionStrings;


    Vector2 scrollPosition;


    Dictionary<string, List<string>> Dependences = new Dictionary<string, List<string>>();
    Dictionary<string, List<string>> References = new Dictionary<string, List<string>>();


    List<string> currentDepends = new List<string>();
    List<string> currentReferences = new List<string>();




    public void Awake()
    {
        this.Dependences.Clear();
        this.References.Clear();
    }


    bool disableSelectEvent = false;
    void OnSelectionChange()
    {
        if (!disableSelectEvent && Selection.activeObject != null && Dependences.Count > 0)
        {
            UnityEngine.Debug.Log("OnSelectionChange:" + Selection.activeObject.name);
            string curpath = AssetDatabase.GetAssetPath(Selection.activeObject);
            this.Dependences.TryGetValue(curpath, out this.currentDepends);
            this.References.TryGetValue(curpath, out this.currentReferences);
            this.Repaint();
        }
        disableSelectEvent = false;
    }


    void SelectAsset(string path)
    {
        disableSelectEvent = true;
        Selection.activeObject = AssetDatabase.LoadMainAssetAtPath(path);
    }


    void OnGUI()
    {


        if (GUILayout.Button("UI Resources Check Mode", GUILayout.Width(200)))
        {
            //perfab 目录可以有多个
            this.GenerateDependence(new string[] { "Assets/Game/Textures", "Assets/Game/Textures/fx/"});
        }
        if (GUILayout.Button("Cleanup UI Resources", GUILayout.Width(200)))
        {
            this.CleanupResources();
        }


        if (GUILayout.Button("Modify Particle Material", GUILayout.Width(200)))
        {
            string[] files = Directory.GetFiles("Assets/FX/Materials");
            foreach (string file in files)
            {
                Material mat = (Material)AssetDatabase.LoadMainAssetAtPath(file);
                if (mat != null)
                {
                    switch (mat.shader.name)
                    {
                        case "Particles/Additive":
                            mat.shader = Shader.Find("Mobile/Particles/Additive");
                            break;
                        case "Particles/Alpha Blended":
                            mat.shader = Shader.Find("Mobile/Particles/Alpha Blended");
                            break;
                        case "Particles/Multiply":
                            mat.shader = Shader.Find("Mobile/Particles/Multiply");
                            break;
                        default:
                            break;
                    }
                }
            }
        }
        //Dependences check area
        scrollPosition = GUILayout.BeginScrollView(scrollPosition, GUIStyle.none);
        GUILayout.BeginArea(new Rect(0, 0, 400, 1000));
        GUILayout.Label("References:");
        if (this.currentReferences != null)
        {
            foreach (string refer in this.currentReferences)
            {
                if (GUILayout.Button(refer, GUILayout.Width(380)))
                {
                    this.SelectAsset(refer);
                }
            }
        }


        GUILayout.EndArea();
        GUILayout.BeginArea(new Rect(400, 0, 400, 1000));
        GUILayout.Label("Depends:");
        if (this.currentDepends != null)
        {
            foreach (string depend in this.currentDepends)
            {
                if (GUILayout.Button(depend, GUILayout.Width(380)))
                {
                    this.SelectAsset(depend);
                }
            }
        }
        GUILayout.EndArea();


        if (selectionStrings != null)
        {
            selectionGridInt = GUILayout.SelectionGrid(selectionGridInt, selectionStrings, 1);
            if (selectionGridInt >= selectionStrings.Length)
                selectionGridInt = selectionStrings.Length - 1;
            if (selectionGridInt > 0)
                Selection.activeObject = AssetDatabase.LoadMainAssetAtPath(selectionStrings[selectionGridInt]);
        }


        GUILayout.EndScrollView();
    }


    void CleanupResources()
    {
        //查找未被使用的资源
        Dictionary<string, List<string>> AssetsRefMap = new Dictionary<string, List<string>>();
        //string[] files = Directory.GetFiles("Assets/Resources/UI/","*", SearchOption.AllDirectories);
        //foreach (string file in files)
        //{
        //    if (file.EndsWith(".meta"))
        //        continue;
        //    if (file.EndsWith(".prefab"))
        //        continue;
        //    Assets[file.Replace("/","/")] = 0;   
        //}
        string UIRES = "Assets/UI_OLD/";
        string[] files = Directory.GetFiles(UIRES, "*", SearchOption.AllDirectories);
        foreach (string file in files)
        {
            if (file.EndsWith(".meta"))
                continue;
            AssetsRefMap[file.Replace("\\", "/")] = new List<string>();
        }






        files = Directory.GetFiles("ProjectSettings/", "*.asset", SearchOption.AllDirectories);
        foreach (string key in files)
        {
            string[] depends = AssetDatabase.GetDependencies(new string[] { key });
            foreach (string dep in depends)
            {
                if (dep != key && AssetsRefMap.ContainsKey(dep))
                {
                    AssetsRefMap[dep].Add(key);
                }
            }
        }


        files = Directory.GetFiles("Assets/BuildOnlyAssets/UI/prefab/", "*.prefab", SearchOption.AllDirectories);
        foreach (string key in files)
        {
            string[] depends = AssetDatabase.GetDependencies(new string[] { key });
            foreach (string dep in depends)
            {
                if (dep != key && AssetsRefMap.ContainsKey(dep))
                {
                    AssetsRefMap[dep].Add(key);
                }
            }
        }
        files = Directory.GetFiles("Assets/Prefabs/", "*.prefab", SearchOption.AllDirectories);
        foreach (string key in files)
        {
            string[] depends = AssetDatabase.GetDependencies(new string[] { key });
            foreach (string dep in depends)
            {
                if (dep != key && AssetsRefMap.ContainsKey(dep))
                {
                    AssetsRefMap[dep].Add(key);
                }
            }
        }


        List<string> result = new List<string>();
        foreach (KeyValuePair<string, List<string>> kv in AssetsRefMap)
        {
            if (kv.Value.Count == 0)
            {
                //result.Add(kv.Key);
                File.Delete(kv.Key);
            }
            else
            {
                result.Add(kv.Key);
                foreach (string file in kv.Value)
                {
                    result.Add(file);
                }
            }


        }
        this.selectionStrings = result.ToArray();
    }


    void CheckDependsFX()
    {
        //查找未被使用的资源
        Dictionary<string, int> Assets = new Dictionary<string, int>();


        //string[] files = Directory.GetFiles("Assets/Resources/UI/","*", SearchOption.AllDirectories);
        //foreach (string file in files)
        //{
        //    if (file.EndsWith(".meta"))
        //        continue;
        //    if (file.EndsWith(".prefab"))
        //        continue;
        //    Assets[file.Replace("/","/")] = 0;   
        //}
        string[] files = Directory.GetFiles("Assets/UI/", "*", SearchOption.AllDirectories);
        foreach (string file in files)
        {
            if (file.EndsWith(".meta"))
                continue;
            Assets[file.Replace("\\", "/")] = 0;
        }


        files = Directory.GetFiles("Assets/Resources/UI/prefab/", "*.prefab", SearchOption.AllDirectories);
        foreach (string key in files)
        {
            string[] depends = AssetDatabase.GetDependencies(new string[] { key });
            foreach (string dep in depends)
            {
                if (dep != key && Assets.ContainsKey(dep))
                {
                    Assets[dep]++;
                }
            }
        }
        List<string> result = new List<string>();
        foreach (KeyValuePair<string, int> kv in Assets)
        {
            if (kv.Value == 0)
                result.Add(kv.Key);
        }
        this.selectionStrings = result.ToArray();
    }


    public void GenerateDependence(string[] paths)
    {
        Debug.Log("GenerateDependence");
        //查找未被使用的资源
        List<string> files = new List<string>();
        foreach (string path in paths)
        {
            files.AddRange(Directory.GetFiles(path, "*.prefab", SearchOption.AllDirectories));
        }
        for (int i = 0; i < files.Count; i++)
        {
            string key = files[i];
            string asset = key.Replace('\\', '/');
            string[] depends = AssetDatabase.GetDependencies(new string[] { asset });
            foreach (string dep in depends)
            {
                if (dep != asset)
                {
                    AddDepends(asset, dep);
                }
            }
            EditorUtility.DisplayProgressBar("Collect Dependences", asset, (float)i / (float)files.Count);
        }
        EditorUtility.ClearProgressBar();
    }


    void AddDepends(string asset, string depend)
    {
        if (!this.Dependences.ContainsKey(asset))
            this.Dependences[asset] = new List<string>();


        if (!this.Dependences[asset].Contains(depend))
            this.Dependences[asset].Add(depend);


        if (!this.References.ContainsKey(depend))
            this.References[depend] = new List<string>();


        if (!this.References[depend].Contains(asset))
            this.References[depend].Add(asset);
    }
}