Unity gadget displays the file size of the resource directory

This editor extension tool script mainly displays the storage space occupied by all files in the resource directory , It has little practical significance , In the actual development process, we can better allocate our resources by understanding the file size of each module .

Shortcut key Alt+K Can be opened or hidden

Put the following code into Editor Folder that will do

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

public class FileCapacity
{
    
    private const string REMOVE_STR = "Assets";
    private const string FILESIZE = "FileSize";

    private static readonly int mRemoveCount = REMOVE_STR.Length;
    private static readonly Color professionalColor = new Color(56f / 255, 56f / 255, 56f / 255, 1);
    private static readonly Color personaloColor = new Color(194f / 255, 194f / 255, 194f / 255, 1);
    private static Dictionary<string, long> DirSizeDictionary = new Dictionary<string, long>();
    private static List<string> DirList = new List<string>();
    private static bool isShowSize = true;
    [MenuItem(" Kill gadget / Shortcut / Show file size  &K")]
    private static void OpenPlaySize()
    {
    
        isShowSize = !isShowSize;
        EditorPrefs.SetBool(FILESIZE, isShowSize);
        GetPropjectDirs();
        AssetDatabase.Refresh();
    }

    [InitializeOnLoadMethod]
    private static void InitializeOnLoadMethod()
    {
    
        EditorApplication.projectChanged += GetPropjectDirs;
        // stay ProjectWindow in , Delegate for each visible list item OnGUI event .
        EditorApplication.projectWindowItemOnGUI += OnGUI;
    }

    [UnityEditor.Callbacks.DidReloadScripts]
    private static void OnScriptsReloaded()
    {
    
        GetPropjectDirs();
    }

    private static void GetPropjectDirs()
    {
    
        Init();
        if (isShowSize == false)
            return;
        GetAllDirecotries(Application.dataPath);
        foreach (string path in DirList)
        {
    
            string newPath = path.Replace("\\", "/");
            DirSizeDictionary.Add(newPath, GetDirectoriesSize(path));
        }
    }
    private static void Init()
    {
    
        isShowSize = EditorPrefs.GetBool(FILESIZE);
        DirSizeDictionary.Clear();
        DirList.Clear();
    }
    // Refresh editor ui
    private static void OnGUI(string guid, Rect selectionRect)
    {
    
        if (isShowSize == false || selectionRect.height > 16)
            return;//>16 To prevent typographical confusion caused by the scaling of file icons 
        var dataPath = Application.dataPath;
        var startIndex = dataPath.LastIndexOf(REMOVE_STR);
        var dir = dataPath.Remove(startIndex, mRemoveCount);
        var path = dir + AssetDatabase.GUIDToAssetPath(guid);
        string text = null;

        long fileSize = 0;
        if (DirSizeDictionary.ContainsKey(path))
        {
    
            fileSize = DirSizeDictionary[path];
        }
        else if (File.Exists(path))
        {
    
            fileSize = new FileInfo(path).Length;
        }
        else
        {
    
            return;
        }
        text = GetFormatSizeString((int)fileSize);

        var label = EditorStyles.label;
        var content = new GUIContent(text);
        var width = label.CalcSize(content).x + 10;

        var pos = selectionRect;
        pos.x = pos.xMax - width;
        pos.width = width;

        EditorGUI.DrawRect(pos, UseDark() ? professionalColor : personaloColor);
        Color defaultC = GUI.color;

        if (fileSize > 1024 * 1024 * 10)
        {
    
            GUI.color = Color.red;
        }
        else if (fileSize > 1024 * 1024)
        {
    
            GUI.color = Color.yellow;
        }

        GUI.Label(pos, text);
        GUI.color = defaultC;
    }

    /// <summary>
    ///  Get the skin 
    /// </summary>
    private static bool UseDark()
    {
    
        PropertyInfo propertyInfo = typeof(EditorGUIUtility).GetProperty("skinIndex", BindingFlags.Static | BindingFlags.NonPublic);
        bool useDark = (int)propertyInfo.GetValue(null) == 1;
        return useDark;
    }
    // Calculate file size 
    private static string GetFormatSizeString(int size)
    {
    
        string[] ns = new string[] {
     "Byte", "KB", "MB", "GB", "TB", "PB" };
        double baseNum = 1024;
        if (size <= 0)
        {
    
            return $"{
      0.ToString("F2")} {
      ns[0]}";
        }
        int pow = Math.Min((int)Math.Floor(Math.Log(size, baseNum)), ns.Length - 1);

        return $"{
      (size / Math.Pow(baseNum, pow)).ToString("F2")} {
      ns[pow]}";
    }

    // obtain Asset All file paths in the directory 
    private static void GetAllDirecotries(string dirPath)
    {
    
        if (Directory.Exists(dirPath) == false)
        {
    
            return;
        }
        DirList.Add(dirPath);
        DirectoryInfo[] dirArray = new DirectoryInfo(dirPath).GetDirectories();
        foreach (DirectoryInfo item in dirArray)
        {
    
            GetAllDirecotries(item.FullName);
        }
    }
    // Get the size of the specific path file 
    private static long GetDirectoriesSize(string dirPath)
    {
    
        if (Directory.Exists(dirPath) == false)
        {
    
            return 0;
        }

        long size = 0;
        DirectoryInfo dir = new DirectoryInfo(dirPath);
        foreach (FileInfo info in dir.GetFiles())
        {
    
            size += info.Length;
        }

        DirectoryInfo[] dirBotton = dir.GetDirectories();
        foreach (DirectoryInfo info in dirBotton)
        {
    
            size += GetDirectoriesSize(info.FullName);
        }
        return size;
    }
}

The article is reproduced in https://blog.csdn.net/qq_37310110/article/details/124295956?spm=1001.2014.3001.5501