Client/GameClient/Assets/YooAsset/Editor/ShaderVariantCollector/ShaderVariantCollectorWindow.cs

#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;

namespace YooAsset.Editor
{
	public class ShaderVariantCollectorWindow : EditorWindow
	{
		[MenuItem("YooAsset/ShaderVariant Collector", false, 201)]
		public static void OpenWindow()
		{
			ShaderVariantCollectorWindow window = GetWindow<ShaderVariantCollectorWindow>("着色器变种收集工具", true, WindowsDefine.DockedWindowTypes);
			window.minSize = new Vector2(800, 600);
		}

		private List<string> _packageNames;

		private Button _saveButton;
		private Button _collectButton;
		private TextField _collectOutputField;
		private Label _currentShaderCountField;
		private Label _currentVariantCountField;
		private SliderInt _processCapacitySlider;
		private PopupField<string> _packageField;

		public void CreateGUI()
		{
			try
			{
				VisualElement root = this.rootVisualElement;

				// 加载布局文件
				var visualAsset = UxmlLoader.LoadWindowUXML<ShaderVariantCollectorWindow>();
				if (visualAsset == null)
					return;

				visualAsset.CloneTree(root);

				// 配置保存按钮
				_saveButton = root.Q<Button>("SaveButton");
				_saveButton.clicked += SaveBtn_clicked;

				// 包裹名称列表
				_packageNames = GetBuildPackageNames();

				// 文件输出目录
				_collectOutputField = root.Q<TextField>("CollectOutput");
				_collectOutputField.SetValueWithoutNotify(ShaderVariantCollectorSettingData.Setting.SavePath);
				_collectOutputField.RegisterValueChangedCallback(evt =>
				{
					ShaderVariantCollectorSettingData.IsDirty = true;
					ShaderVariantCollectorSettingData.Setting.SavePath = _collectOutputField.value;
				});

				// 收集的包裹
				var packageContainer = root.Q("PackageContainer");
				if (_packageNames.Count > 0)
				{
					int defaultIndex = GetDefaultPackageIndex(ShaderVariantCollectorSettingData.Setting.CollectPackage);
					_packageField = new PopupField<string>(_packageNames, defaultIndex);
					_packageField.label = "Package";
					_packageField.style.width = 350;
					_packageField.RegisterValueChangedCallback(evt =>
					{
						ShaderVariantCollectorSettingData.IsDirty = true;
						ShaderVariantCollectorSettingData.Setting.CollectPackage = _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(ShaderVariantCollectorSettingData.Setting.ProcessCapacity);
#if !UNITY_2020_3_OR_NEWER
				_processCapacitySlider.label = $"Capacity ({_processCapacitySlider.value})";
				_processCapacitySlider.RegisterValueChangedCallback(evt =>
				{
					ShaderVariantCollectorSettingData.IsDirty = true;
					ShaderVariantCollectorSettingData.Setting.ProcessCapacity = _processCapacitySlider.value;
					_processCapacitySlider.label = $"Capacity ({_processCapacitySlider.value})";
				});
#else
				_processCapacitySlider.RegisterValueChangedCallback(evt =>
				{
					ShaderVariantCollectorSettingData.IsDirty = true;
					ShaderVariantCollectorSettingData.Setting.ProcessCapacity = _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());
			}
		}
		public void OnDestroy()
		{
			if (ShaderVariantCollectorSettingData.IsDirty)
				ShaderVariantCollectorSettingData.SaveFile();
		}
		private void Update()
		{
			if (_saveButton != null)
			{
				if (ShaderVariantCollectorSettingData.IsDirty)
				{
					if (_saveButton.enabledSelf == false)
						_saveButton.SetEnabled(true);
				}
				else
				{
					if (_saveButton.enabledSelf)
						_saveButton.SetEnabled(false);
				}
			}

			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 SaveBtn_clicked()
		{
			ShaderVariantCollectorSettingData.SaveFile();
		}
		private void CollectButton_clicked()
		{
			string savePath = ShaderVariantCollectorSettingData.Setting.SavePath;
			string packageName = ShaderVariantCollectorSettingData.Setting.CollectPackage;
			int processCapacity = _processCapacitySlider.value;
			ShaderVariantCollector.Run(savePath, packageName, processCapacity, null);
		}

		// 构建包裹相关
		private int GetDefaultPackageIndex(string packageName)
		{
			for (int index = 0; index < _packageNames.Count; index++)
			{
				if (_packageNames[index] == packageName)
				{
					return index;
				}
			}

			ShaderVariantCollectorSettingData.Setting.CollectPackage = _packageNames[0];
			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