Unity编辑器 - 资源批处理工具基类

Unity编辑器 - 资源批处理工具基类

using UnityEngine;
using BatchTool;
using UnityEditor;

public class TestBatchTool : BatchToolBase {

[MenuItem("BatchTool/test")]
static void test() {
GetWindow<TestBatchTool>();
}

ModelImporterAnimationType animType;
bool bImportMat;

/// <summary>
/// 初始化信息
/// </summary>
/// <param name="pTitle">进度条标题</param>
/// <param name="pInfo">进度条显示信息开头</param>
/// <param name="fileExt">需要处理文件的后缀名</param>
protected override void InitTool(out string pTitle, out string pInfo, out string fileExt) {
base.InitTool(out pTitle, out pInfo, out fileExt);
fileExt = ".fbx";
}

/// <summary>
/// 处理一个资源的逻辑
/// </summary>
/// <param name="path">资源先对路径</param>
protected override void ExcuteOnePath(string path) {
Debug.Log(path);
}

/// <summary>
/// 批处理选项
/// </summary>
protected override void GUIBatchOptions() {
base.GUIBatchOptions();
animType = (ModelImporterAnimationType)EditorGUILayout.EnumPopup("Animation Type", animType);
bImportMat = EditorGUILayout.Toggle("Import Material", bImportMat);
}
}

using System;
using System.IO;
using System.Linq;
using UnityEditor;
using UnityEngine;

namespace BatchTool {
public class BatchToolBase : EditorWindow {
private string _folderPath;
protected string FolderPath {
get {
if (string.IsNullOrEmpty(_folderPath))
_folderPath = "Assets";
return _folderPath;
}

set {
string pstr = Directory.GetParent(Application.dataPath).FullName;
pstr = pstr.Replace("\\", "/");
if (value != null && value.Contains(pstr))
_folderPath = value.Replace(pstr + "/", "");
else
_folderPath = value;
}
}

private string _processTitle = string.Empty;
private string _processInfo = string.Empty;
private string _fileNameExt = ".prefab";

private string[] _allAssetPaths;

private Rect _drawRect = Rect.zero;

protected void OnEnable() {
minSize = new Vector2(100,150);
position = new Rect(300, 300, 260, 180);
Repaint();
_allAssetPaths = AssetDatabase.GetAllAssetPaths();
InitTool(out _processTitle,out _processInfo,out _fileNameExt);
EditorUtility.ClearProgressBar();
}

protected virtual void InitTool
(out string pTitle, out string pInfo, out string fileExt) {
pTitle = "批处理";
pInfo = "正在处理： ";
fileExt = ".prefab";
}

protected void OnGUI() {
_drawRect.position = Vector2.zero;
_drawRect.size = position.size;
_drawRect.x += 5;
_drawRect.width -= 10f;
_drawRect.y += 5;
_drawRect.height -= 10f;
GUILayout.BeginArea(_drawRect);
{
GUISelectFolderPath();
GUILayout.Space(5);
GUIBatchOptions();
GUILayout.Space(5);
GUIBeginBatchBtn();
}
GUILayout.EndArea();
}

private void GUIBeginBatchBtn() {
Rect rect = GUILayoutUtility.GetRect(GUIContent.none, EditorStyles.largeLabel, GUILayout.Height(25));
rect.x += 5f;
rect.width -= 10f;
if (GUI.Button(rect, "开始批处理！")) {
if (!Directory.Exists(FolderPath)) {
Debug.LogError("文件夹不存在！");
return;
}

var filterCollection = from path in _allAssetPaths
where FilterPathFun(path)
select path;

var paths = filterCollection.ToList();

float iterateCount = 0;
float proccesCount = 0;
foreach (var path in paths) {
iterateCount++;
var procValue = iterateCount / paths.Count;
if (EditorUtility.DisplayCancelableProgressBar(_processTitle, _processInfo + path, procValue)) {
Debug.Log("取消批处理！");
EditorUtility.ClearProgressBar();
break;
}
ExcuteOnePath(path);
proccesCount++;
}

EditorUtility.ClearProgressBar();
OnComplete(paths.Count, proccesCount);
}
}

protected virtual void OnComplete(int count, float proccesCount) {
Debug.Log("遍历了 " + count + "个文件，实际处理了 " + proccesCount + " 个文件!");
}

protected virtual void ExcuteOnePath(string path) {
Debug.Log("NotImplementedException!!!!");
}

protected virtual bool FilterPathFun(string path) {
bool isExtention = Path.GetExtension(path).ToLower() == _fileNameExt;
bool isInFolder = path.Substring(0, FolderPath.Length) == FolderPath;
return isExtention && isInFolder;
}

protected virtual void GUIBatchOptions() {
GUILayout.Label("批处理设置：");
}

private void GUISelectFolderPath() {
GUILayout.BeginHorizontal();
{
EditorGUIUtility.labelWidth = 45f;

GUI.SetNextControlName("inputPath");
FolderPath = EditorGUILayout.TextField("文件夹：", FolderPath);
var rect = GUILayoutUtility.GetLastRect();
if (rect.width > 2 && GUI.GetNameOfFocusedControl() == "inputPath") {
if (Event.current.clickCount > 0 && !rect.Contains(Event.current.mousePosition)) {
GUI.FocusControl(null);
Repaint();
}
}

if (GUILayout.Button("", GUILayout.Width(25), GUILayout.Height(14))) {
FolderPath = EditorUtility.OpenFolderPanel("Select a folder", "Assets", "");
}
EditorGUIUtility.labelWidth = 0;
}
GUILayout.EndHorizontal();
}
}
}