更新YooAsset 2.3.3 -> 2.3.7 优化YooAsset.RuntimeExtension以及YooAsset.EditorExtension目录结构

2025-08-14 16:51:28 +00:00 · 2025-04-17 12:59:23 +08:00
parent 32418326b1
commit 227283864f
206 changed files with 1641 additions and 461 deletions
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectionHelper.cs
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectionHelper.cs
@@ -0,0 +1,38 @@
+using System;
+using System.Collections;
+using System.Collections.Generic;
+using System.Diagnostics;
+using UnityEngine;
+using UnityEngine.Rendering;
+using UnityEditor;
+using YooAsset.Editor;
+
+public static class ShaderVariantCollectionHelper
+{
+    public static void ClearCurrentShaderVariantCollection()
+    {
+        EditorTools.InvokeNonPublicStaticMethod(typeof(ShaderUtil), "ClearCurrentShaderVariantCollection");
+    }
+    public static void SaveCurrentShaderVariantCollection(string savePath)
+    {
+        EditorTools.InvokeNonPublicStaticMethod(typeof(ShaderUtil), "SaveCurrentShaderVariantCollection", savePath);
+    }
+    public static int GetCurrentShaderVariantCollectionShaderCount()
+    {
+        return (int)EditorTools.InvokeNonPublicStaticMethod(typeof(ShaderUtil), "GetCurrentShaderVariantCollectionShaderCount");
+    }
+    public static int GetCurrentShaderVariantCollectionVariantCount()
+    {
+        return (int)EditorTools.InvokeNonPublicStaticMethod(typeof(ShaderUtil), "GetCurrentShaderVariantCollectionVariantCount");
+    }
+
+    /// <summary>
+    /// 获取着色器的变种总数量
+    /// </summary>
+    public static string GetShaderVariantCount(string assetPath)
+    {
+        Shader shader = AssetDatabase.LoadAssetAtPath<Shader>(assetPath);
+        var variantCount = EditorTools.InvokeNonPublicStaticMethod(typeof(ShaderUtil), "GetVariantCount", shader, true);
+        return variantCount.ToString();
+    }
+}
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectionHelper.cs.meta
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectionHelper.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: ab74d4ff4a2805147883de70a1559a0a
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectionManifest.cs
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectionManifest.cs
@@ -0,0 +1,190 @@
+using System;
+using System.Collections;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.Linq;
+using UnityEngine;
+using UnityEngine.Rendering;
+using UnityEditor;
+
+[Serializable]
+public class ShaderVariantCollectionManifest
+{
+    [Serializable]
+    public class ShaderVariantElement : IComparable<ShaderVariantElement>
+    {
+        public string SortValue { private set; get; }
+
+        /// <summary>
+        ///  Pass type to use in this variant.
+        /// </summary>
+        public PassType PassType;
+
+        /// <summary>
+        /// Array of shader keywords to use in this variant.
+        /// </summary>
+        public string[] Keywords;
+
+        public void MakeSortValue()
+        {
+            string combineKeyword = string.Empty;
+            for (int i = 0; i < Keywords.Length; i++)
+            {
+                if (i == 0)
+                    combineKeyword = Keywords[0];
+                else
+                    combineKeyword = $"{combineKeyword}+{Keywords[0]}";
+            }
+
+            SortValue = $"{PassType}+{combineKeyword}";
+        }
+        public int CompareTo(ShaderVariantElement other)
+        {
+            return SortValue.CompareTo(other.SortValue);
+        }
+    }
+
+    [Serializable]
+    public class ShaderVariantInfo : IComparable<ShaderVariantInfo>
+    {
+        public string SortValue { private set; get; }
+
+        /// <summary>
+        /// 着色器资源路径.
+        /// </summary>
+        public string AssetPath;
+
+        /// <summary>
+        /// 着色器名称
+        /// </summary>
+        public string ShaderName;
+
+        /// <summary>
+        /// 着色器变种总数
+        /// </summary>
+        public int ShaderVariantCount = 0;
+
+        /// <summary>
+        /// 着色器变种列表
+        /// </summary>
+        public List<ShaderVariantElement> ShaderVariantElements = new List<ShaderVariantElement>(1000);
+
+        public void MakeSortValue()
+        {
+            SortValue = AssetPath + "+" + ShaderName;
+        }
+        public int CompareTo(ShaderVariantInfo other)
+        {
+            return SortValue.CompareTo(other.SortValue);
+        }
+    }
+
+
+    /// <summary>
+    /// Number of shaders in this collection
+    /// </summary>
+    public int ShaderTotalCount;
+
+    /// <summary>
+    /// Number of total varians in this collection
+    /// </summary>
+    public int VariantTotalCount;
+
+    /// <summary>
+    /// Shader variants info list.
+    /// </summary>
+    public List<ShaderVariantInfo> ShaderVariantInfos = new List<ShaderVariantInfo>(1000);
+
+    /// <summary>
+    /// 添加着色器变种信息
+    /// </summary>
+    public void AddShaderVariant(string assetPath, string shaderName, PassType passType, string[] keywords)
+    {
+        // 排序Keyword列表
+        List<string> temper = new List<string>(keywords);
+        temper.Sort();
+
+        var info = GetOrCreateShaderVariantInfo(assetPath, shaderName);
+        ShaderVariantElement element = new ShaderVariantElement();
+        element.PassType = passType;
+        element.Keywords = temper.ToArray();
+        element.MakeSortValue();
+        info.ShaderVariantElements.Add(element);
+        info.ShaderVariantCount++;
+    }
+    private ShaderVariantInfo GetOrCreateShaderVariantInfo(string assetPath, string shaderName)
+    {
+        var selectList = ShaderVariantInfos.Where(t => t.ShaderName == shaderName && t.AssetPath == assetPath).ToList();
+        if (selectList.Count == 0)
+        {
+            ShaderVariantInfo newInfo = new ShaderVariantInfo();
+            newInfo.AssetPath = assetPath;
+            newInfo.ShaderName = shaderName;
+            newInfo.MakeSortValue();
+            ShaderVariantInfos.Add(newInfo);
+            return newInfo;
+        }
+
+        if (selectList.Count != 1)
+            throw new Exception("Should never get here !");
+
+        return selectList[0];
+    }
+
+
+    /// <summary>
+    /// 解析SVC文件并将数据写入到清单
+    /// </summary>
+    public static ShaderVariantCollectionManifest Extract(ShaderVariantCollection svc)
+    {
+        var manifest = new ShaderVariantCollectionManifest();
+        manifest.ShaderTotalCount = ShaderVariantCollectionHelper.GetCurrentShaderVariantCollectionShaderCount();
+        manifest.VariantTotalCount = ShaderVariantCollectionHelper.GetCurrentShaderVariantCollectionVariantCount();
+
+        using (var so = new SerializedObject(svc))
+        {
+            var shaderArray = so.FindProperty("m_Shaders.Array");
+            if (shaderArray != null && shaderArray.isArray)
+            {
+                for (int i = 0; i < shaderArray.arraySize; ++i)
+                {
+                    var shaderRef = shaderArray.FindPropertyRelative($"data[{i}].first");
+                    var shaderVariantsArray = shaderArray.FindPropertyRelative($"data[{i}].second.variants");
+                    if (shaderRef != null && shaderRef.propertyType == SerializedPropertyType.ObjectReference && shaderVariantsArray != null && shaderVariantsArray.isArray)
+                    {
+                        var shader = shaderRef.objectReferenceValue as Shader;
+                        if (shader == null)
+                        {
+                            throw new Exception("Invalid shader in ShaderVariantCollection file.");
+                        }
+
+                        string shaderAssetPath = AssetDatabase.GetAssetPath(shader);
+                        string shaderName = shader.name;
+
+                        // 添加变种信息
+                        for (int j = 0; j < shaderVariantsArray.arraySize; ++j)
+                        {
+                            var propKeywords = shaderVariantsArray.FindPropertyRelative($"Array.data[{j}].keywords");
+                            var propPassType = shaderVariantsArray.FindPropertyRelative($"Array.data[{j}].passType");
+                            if (propKeywords != null && propPassType != null && propKeywords.propertyType == SerializedPropertyType.String)
+                            {
+                                string[] keywords = propKeywords.stringValue.Split(' ');
+                                PassType pathType = (PassType)propPassType.intValue;
+                                manifest.AddShaderVariant(shaderAssetPath, shaderName, pathType, keywords);
+                            }
+                        }
+                    }
+                }
+            }
+        }
+
+        // 重新排序
+        manifest.ShaderVariantInfos.Sort();
+        foreach (var shaderVariantInfo in manifest.ShaderVariantInfos)
+        {
+            shaderVariantInfo.ShaderVariantElements.Sort();
+        }
+
+        return manifest;
+    }
+}
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectionManifest.cs.meta
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectionManifest.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 97098b04691f5c046ac4829f1d72f425
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollector.cs
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollector.cs
@@ -0,0 +1,251 @@
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.Linq;
+using System.IO;
+using UnityEngine;
+using UnityEditor;
+using UnityEditor.SceneManagement;
+using YooAsset.Editor;
+
+public static class ShaderVariantCollector
+{
+    private enum ESteps
+    {
+        None,
+        Prepare,
+        CollectAllMaterial,
+        CollectVariants,
+        CollectSleeping,
+        WaitingDone,
+    }
+
+    private const float WaitMilliseconds = 3000f;
+    private const float SleepMilliseconds = 3000f;
+    private static string _savePath;
+    private static string _packageName;
+    private static int _processMaxNum;
+    private static Action _completedCallback;
+
+    private static ESteps _steps = ESteps.None;
+    private static Stopwatch _elapsedTime;
+    private static List<string> _allMaterials;
+    private static List<GameObject> _allSpheres = new List<GameObject>(1000);
+
+
+    /// <summary>
+    /// 开始收集
+    /// </summary>
+    public static void Run(string savePath, string packageName, int processMaxNum, Action completedCallback)
+    {
+        if (_steps != ESteps.None)
+            return;
+
+        if (Path.HasExtension(savePath) == false)
+            savePath = $"{savePath}.shadervariants";
+        if (Path.GetExtension(savePath) != ".shadervariants")
+            throw new System.Exception("Shader variant file extension is invalid.");
+        if (string.IsNullOrEmpty(packageName))
+            throw new System.Exception("Package name is null or empty !");
+
+        // 注意：先删除再保存，否则ShaderVariantCollection内容将无法及时刷新
+        AssetDatabase.DeleteAsset(savePath);
+        EditorTools.CreateFileDirectory(savePath);
+        _savePath = savePath;
+        _packageName = packageName;
+        _processMaxNum = processMaxNum;
+        _completedCallback = completedCallback;
+
+        // 聚焦到游戏窗口
+        EditorTools.FocusUnityGameWindow();
+
+        // 创建临时测试场景
+        CreateTempScene();
+
+        _steps = ESteps.Prepare;
+        EditorApplication.update += EditorUpdate;
+    }
+
+    private static void EditorUpdate()
+    {
+        if (_steps == ESteps.None)
+            return;
+
+        if (_steps == ESteps.Prepare)
+        {
+            ShaderVariantCollectionHelper.ClearCurrentShaderVariantCollection();
+            _steps = ESteps.CollectAllMaterial;
+            return; //等待一帧
+        }
+
+        if (_steps == ESteps.CollectAllMaterial)
+        {
+            _allMaterials = GetAllMaterials();
+            _steps = ESteps.CollectVariants;
+            return; //等待一帧
+        }
+
+        if (_steps == ESteps.CollectVariants)
+        {
+            int count = Mathf.Min(_processMaxNum, _allMaterials.Count);
+            List<string> range = _allMaterials.GetRange(0, count);
+            _allMaterials.RemoveRange(0, count);
+            CollectVariants(range);
+
+            if (_allMaterials.Count > 0)
+            {
+                _elapsedTime = Stopwatch.StartNew();
+                _steps = ESteps.CollectSleeping;
+            }
+            else
+            {
+                _elapsedTime = Stopwatch.StartNew();
+                _steps = ESteps.WaitingDone;
+            }
+        }
+
+        if (_steps == ESteps.CollectSleeping)
+        {
+            if (_elapsedTime.ElapsedMilliseconds > SleepMilliseconds)
+            {
+                DestroyAllSpheres();
+                _elapsedTime.Stop();
+                _steps = ESteps.CollectVariants;
+            }
+        }
+
+        if (_steps == ESteps.WaitingDone)
+        {
+            // 注意：一定要延迟保存才会起效
+            if (_elapsedTime.ElapsedMilliseconds > WaitMilliseconds)
+            {
+                _elapsedTime.Stop();
+                _steps = ESteps.None;
+
+                // 保存结果并创建清单
+                ShaderVariantCollectionHelper.SaveCurrentShaderVariantCollection(_savePath);
+                CreateManifest();
+
+                UnityEngine.Debug.Log($"搜集SVC完毕！");
+                EditorApplication.update -= EditorUpdate;
+                _completedCallback?.Invoke();
+            }
+        }
+    }
+    private static void CreateTempScene()
+    {
+        EditorSceneManager.NewScene(NewSceneSetup.DefaultGameObjects);
+    }
+    private static List<string> GetAllMaterials()
+    {
+        // 获取所有打包的资源
+        CollectResult collectResult = AssetBundleCollectorSettingData.Setting.BeginCollect(_packageName, false, false);
+
+        // 搜集所有材质球
+        int progressValue = 0;
+        HashSet<string> result = new HashSet<string>();
+        foreach (var collectAssetInfo in collectResult.CollectAssets)
+        {
+            if (collectAssetInfo.AssetInfo.AssetType == typeof(UnityEngine.Material))
+            {
+                string assetPath = collectAssetInfo.AssetInfo.AssetPath;
+                if (result.Contains(assetPath) == false)
+                    result.Add(assetPath);
+            }
+            foreach(var dependAssetInfo in collectAssetInfo.DependAssets)
+            {
+                if (dependAssetInfo.AssetType == typeof(UnityEngine.Material))
+                {
+                    string assetPath = dependAssetInfo.AssetPath;
+                    if (result.Contains(assetPath) == false)
+                        result.Add(assetPath);
+                }
+            }
+            EditorTools.DisplayProgressBar("搜集所有材质球", ++progressValue, collectResult.CollectAssets.Count);
+        }
+        EditorTools.ClearProgressBar();
+
+        // 返回结果
+        return result.ToList();
+    }
+    private static void CollectVariants(List<string> materials)
+    {
+        Camera camera = Camera.main;
+        if (camera == null)
+            throw new System.Exception("Not found main camera.");
+
+        // 设置主相机
+        float aspect = camera.aspect;
+        int totalMaterials = materials.Count;
+        float height = Mathf.Sqrt(totalMaterials / aspect) + 1;
+        float width = Mathf.Sqrt(totalMaterials / aspect) * aspect + 1;
+        float halfHeight = Mathf.CeilToInt(height / 2f);
+        float halfWidth = Mathf.CeilToInt(width / 2f);
+        camera.orthographic = true;
+        camera.orthographicSize = halfHeight;
+        camera.transform.position = new Vector3(0f, 0f, -10f);
+
+        // 创建测试球体
+        int xMax = (int)(width - 1);
+        int x = 0, y = 0;
+        int progressValue = 0;
+        for (int i = 0; i < materials.Count; i++)
+        {
+            var material = materials[i];
+            var position = new Vector3(x - halfWidth + 1f, y - halfHeight + 1f, 0f);
+            var go = CreateSphere(material, position, i);
+            if (go != null)
+                _allSpheres.Add(go);
+            if (x == xMax)
+            {
+                x = 0;
+                y++;
+            }
+            else
+            {
+                x++;
+            }
+            EditorTools.DisplayProgressBar("照射所有材质球", ++progressValue, materials.Count);
+        }
+        EditorTools.ClearProgressBar();
+    }
+    private static GameObject CreateSphere(string assetPath, Vector3 position, int index)
+    {
+        var material = AssetDatabase.LoadAssetAtPath<Material>(assetPath);
+        var shader = material.shader;
+        if (shader == null)
+            return null;
+
+        var go = GameObject.CreatePrimitive(PrimitiveType.Sphere);
+        go.GetComponent<Renderer>().sharedMaterial = material;
+        go.transform.position = position;
+        go.name = $"Sphere_{index} | {material.name}";
+        return go;
+    }
+    private static void DestroyAllSpheres()
+    {
+        foreach (var go in _allSpheres)
+        {
+            GameObject.DestroyImmediate(go);
+        }
+        _allSpheres.Clear();
+
+        // 尝试释放编辑器加载的资源
+        EditorUtility.UnloadUnusedAssetsImmediate(true);
+    }
+    private static void CreateManifest()
+    {
+        AssetDatabase.Refresh(ImportAssetOptions.ForceUpdate);
+
+        ShaderVariantCollection svc = AssetDatabase.LoadAssetAtPath<ShaderVariantCollection>(_savePath);
+        if (svc != null)
+        {
+            var wrapper = ShaderVariantCollectionManifest.Extract(svc);
+            string jsonData = JsonUtility.ToJson(wrapper, true);
+            string savePath = _savePath.Replace(".shadervariants", ".json");
+            File.WriteAllText(savePath, jsonData);
+        }
+
+        AssetDatabase.Refresh(ImportAssetOptions.ForceUpdate);
+    }
+}
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollector.cs.meta
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollector.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 21b4cc6bf4c0c064d8e2687024e24c86
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorSetting.cs
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorSetting.cs
@@ -0,0 +1,29 @@
+using UnityEngine;
+using UnityEditor;
+
+public class ShaderVariantCollectorSetting : ScriptableObject
+{
+    private const string DefaultSavePath = "Assets/MyShaderVariants.shadervariants";
+
+    public static string GeFileSavePath(string packageName)
+    {
+        string key = $"{Application.productName}_{packageName}_GeFileSavePath";
+        return EditorPrefs.GetString(key, DefaultSavePath);
+    }
+    public static void SetFileSavePath(string packageName, string savePath)
+    {
+        string key = $"{Application.productName}_{packageName}_GeFileSavePath";
+        EditorPrefs.SetString(key, savePath);
+    }
+
+    public static int GeProcessCapacity(string packageName)
+    {
+        string key = $"{Application.productName}_{packageName}_GeProcessCapacity";
+        return EditorPrefs.GetInt(key, 1000);
+    }
+    public static void SetProcessCapacity(string packageName, int capacity)
+    {
+        string key = $"{Application.productName}_{packageName}_GeProcessCapacity";
+        EditorPrefs.SetInt(key, capacity);
+    }
+}
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorSetting.cs.meta
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorSetting.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 44454e58a49818040a1aef5799e71b30
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorWindow.cs
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorWindow.cs
@@ -0,0 +1,150 @@
+#if UNITY_2019_4_OR_NEWER
+using System;
+using System.Linq;
+using System.Collections.Generic;
+using UnityEditor;
+using UnityEngine;
+using UnityEditor.UIElements;
+using UnityEngine.UIElements;
+using YooAsset.Editor;
+
+public class ShaderVariantCollectorWindow : EditorWindow
+{
+    [MenuItem("Tools/着色器变种收集器", false, 100)]
+    public static void OpenWindow()
+    {
+        ShaderVariantCollectorWindow window = GetWindow<ShaderVariantCollectorWindow>("着色器变种收集工具", true);
+        window.minSize = new Vector2(800, 600);
+    }
+
+    private Button _collectButton;
+    private TextField _collectOutputField;
+    private Label _currentShaderCountField;
+    private Label _currentVariantCountField;
+    private SliderInt _processCapacitySlider;
+    private PopupField<string> _packageField;
+
+    private List<string> _packageNames;
+    private string _currentPackageName;
+
+    public void CreateGUI()
+    {
+        try
+        {
+            VisualElement root = this.rootVisualElement;
+
+            // 加载布局文件
+            var visualAsset = UxmlLoader.LoadWindowUXML<ShaderVariantCollectorWindow>();
+            if (visualAsset == null)
+                return;
+
+            visualAsset.CloneTree(root);
+
+            // 包裹名称列表
+            _packageNames = GetBuildPackageNames();
+            _currentPackageName = _packageNames[0];
+
+            // 文件输出目录
+            _collectOutputField = root.Q<TextField>("CollectOutput");
+            _collectOutputField.SetValueWithoutNotify(ShaderVariantCollectorSetting.GeFileSavePath(_currentPackageName));
+            _collectOutputField.RegisterValueChangedCallback(evt =>
+            {
+                ShaderVariantCollectorSetting.SetFileSavePath(_currentPackageName, _collectOutputField.value);
+            });
+
+            // 收集的包裹
+            var packageContainer = root.Q("PackageContainer");
+            if (_packageNames.Count > 0)
+            {
+                int defaultIndex = GetDefaultPackageIndex(_currentPackageName);
+                _packageField = new PopupField<string>(_packageNames, defaultIndex);
+                _packageField.label = "Package";
+                _packageField.style.width = 350;
+                _packageField.RegisterValueChangedCallback(evt =>
+                {
+                    _currentPackageName = _packageField.value;
+                });
+                packageContainer.Add(_packageField);
+            }
+            else
+            {
+                _packageField = new PopupField<string>();
+                _packageField.label = "Package";
+                _packageField.style.width = 350;
+                packageContainer.Add(_packageField);
+            }
+
+            // 容器值
+            _processCapacitySlider = root.Q<SliderInt>("ProcessCapacity");
+            _processCapacitySlider.SetValueWithoutNotify(ShaderVariantCollectorSetting.GeProcessCapacity(_currentPackageName));
+#if !UNITY_2020_3_OR_NEWER
+            _processCapacitySlider.label = $"Capacity ({_processCapacitySlider.value})";
+            _processCapacitySlider.RegisterValueChangedCallback(evt =>
+            {
+                ShaderVariantCollectorSetting.SetProcessCapacity(_currentPackageName, _processCapacitySlider.value);
+                _processCapacitySlider.label = $"Capacity ({_processCapacitySlider.value})";
+            });
+#else
+            _processCapacitySlider.RegisterValueChangedCallback(evt =>
+            {
+                ShaderVariantCollectorSetting.SetProcessCapacity(_currentPackageName, _processCapacitySlider.value);
+            });
+#endif
+
+            _currentShaderCountField = root.Q<Label>("CurrentShaderCount");
+            _currentVariantCountField = root.Q<Label>("CurrentVariantCount");
+
+            // 变种收集按钮
+            _collectButton = root.Q<Button>("CollectButton");
+            _collectButton.clicked += CollectButton_clicked;
+        }
+        catch (Exception e)
+        {
+            Debug.LogError(e.ToString());
+        }
+    }
+    private void Update()
+    {
+        if (_currentShaderCountField != null)
+        {
+            int currentShaderCount = ShaderVariantCollectionHelper.GetCurrentShaderVariantCollectionShaderCount();
+            _currentShaderCountField.text = $"Current Shader Count : {currentShaderCount}";
+        }
+
+        if (_currentVariantCountField != null)
+        {
+            int currentVariantCount = ShaderVariantCollectionHelper.GetCurrentShaderVariantCollectionVariantCount();
+            _currentVariantCountField.text = $"Current Variant Count : {currentVariantCount}";
+        }
+    }
+
+    private void CollectButton_clicked()
+    {
+        string savePath = ShaderVariantCollectorSetting.GeFileSavePath(_currentPackageName);
+        int processCapacity = _processCapacitySlider.value;
+        ShaderVariantCollector.Run(savePath, _currentPackageName, processCapacity, null);
+    }
+
+    // 构建包裹相关
+    private int GetDefaultPackageIndex(string packageName)
+    {
+        for (int index = 0; index < _packageNames.Count; index++)
+        {
+            if (_packageNames[index] == packageName)
+            {
+                return index;
+            }
+        }
+        return 0;
+    }
+    private List<string> GetBuildPackageNames()
+    {
+        List<string> result = new List<string>();
+        foreach (var package in AssetBundleCollectorSettingData.Setting.Packages)
+        {
+            result.Add(package.PackageName);
+        }
+        return result;
+    }
+}
+#endif
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorWindow.cs.meta
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorWindow.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 70401cc80b9807e46bd8283e01b4302f
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorWindow.uxml
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorWindow.uxml
@@ -0,0 +1,11 @@
+<ui:UXML xmlns:ui="UnityEngine.UIElements" xmlns:uie="UnityEditor.UIElements" editor-extension-mode="False">
+    <uie:Toolbar name="Toolbar" style="display: flex; flex-direction: row-reverse;" />
+    <ui:VisualElement name="CollectContainer">
+        <ui:TextField picking-mode="Ignore" label="文件保存路径" name="CollectOutput" style="height: 22px;" />
+        <ui:VisualElement name="PackageContainer" style="height: 24px;" />
+        <ui:Label text="Current Shader Count" display-tooltip-when-elided="true" name="CurrentShaderCount" style="height: 20px; padding-left: 4px;" />
+        <ui:Label text="Current Variant Count" display-tooltip-when-elided="true" name="CurrentVariantCount" style="height: 20px; padding-left: 4px;" />
+        <ui:SliderInt picking-mode="Ignore" label="Capacity" value="9999" high-value="1000" name="ProcessCapacity" low-value="10" show-input-field="true" />
+        <ui:Button text="开始搜集" display-tooltip-when-elided="true" name="CollectButton" style="height: 50px; background-color: rgb(40, 106, 42); margin-top: 10px;" />
+    </ui:VisualElement>
+</ui:UXML>
--- a/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorWindow.uxml.meta
+++ b/UnityProject/Packages/YooAsset/EditorExtension/ShaderVariantCollector/ShaderVariantCollectorWindow.uxml.meta
@@ -0,0 +1,10 @@
+fileFormatVersion: 2
+guid: 9bff4878063eaf04dab8713e1e662ac5
+ScriptedImporter:
+  internalIDToNameTable: []
+  externalObjects: {}
+  serializedVersion: 2
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
+  script: {fileID: 13804, guid: 0000000000000000e000000000000000, type: 0}