mirror of
https://github.com/muskit/H3VR-TNH-Quality-of-Life-Improvements.git
synced 2026-06-03 04:34:26 -07:00
Initial commit
This commit is contained in:
msk
@@ -0,0 +1,300 @@
|
||||
using UnityEditor;
|
||||
using UnityEngine;
|
||||
using System.Collections.Generic;
|
||||
using AssetBundleBrowser.AssetBundleModel;
|
||||
using UnityEditor.IMGUI.Controls;
|
||||
|
||||
namespace AssetBundleBrowser
|
||||
{
|
||||
internal class BundleDetailItem : TreeViewItem
|
||||
{
|
||||
internal BundleDetailItem(int id, int depth, string displayName, MessageType type) : base(id, depth, displayName)
|
||||
{
|
||||
MessageLevel = type;
|
||||
}
|
||||
|
||||
internal MessageType MessageLevel { get; set; }
|
||||
}
|
||||
|
||||
internal class TogglePathTreeViewItem : TreeViewItem
|
||||
{
|
||||
private static bool m_DisplayAlt = false;
|
||||
|
||||
private string m_DisplayNamePrefix;
|
||||
private string m_Path;
|
||||
|
||||
public string Path
|
||||
{
|
||||
get { return m_Path; }
|
||||
}
|
||||
|
||||
public string DisplayNamePrefix
|
||||
{
|
||||
get { return m_DisplayNamePrefix; }
|
||||
}
|
||||
|
||||
public TogglePathTreeViewItem(int id, int depth, string displayName, string path)
|
||||
{
|
||||
base.depth = depth;
|
||||
base.id = id;
|
||||
base.displayName = displayName;
|
||||
m_Path = path;
|
||||
m_DisplayNamePrefix = "";
|
||||
}
|
||||
|
||||
public TogglePathTreeViewItem(int id, int depth, string displayNamePrefix, string displayName, string path)
|
||||
{
|
||||
base.depth = depth;
|
||||
base.id = id;
|
||||
base.displayName = displayName;
|
||||
m_Path = path;
|
||||
m_DisplayNamePrefix = displayNamePrefix;
|
||||
}
|
||||
|
||||
public override string displayName
|
||||
{
|
||||
get
|
||||
{
|
||||
// TODO this is a bit unresponsive here in large projects, see if can be better elsewhere
|
||||
Event e = Event.current;
|
||||
if (e.alt && e.type == EventType.MouseDown)
|
||||
m_DisplayAlt = !m_DisplayAlt;
|
||||
|
||||
return m_DisplayNamePrefix + (m_DisplayAlt ? m_Path : base.displayName);
|
||||
}
|
||||
set { base.displayName = value; }
|
||||
}
|
||||
}
|
||||
|
||||
internal class BundleDetailList : TreeView
|
||||
{
|
||||
private HashSet<BundleDataInfo> m_Selecteditems;
|
||||
private Rect m_TotalRect;
|
||||
|
||||
private const float k_DoubleIndent = 32f;
|
||||
private const string k_SizeHeader = "Size: ";
|
||||
private const string k_DependencyHeader = "Dependent On:";
|
||||
private const string k_DependencyEmpty = k_DependencyHeader + " - None";
|
||||
private const string k_MessageHeader = "Messages:";
|
||||
private const string k_MessageEmpty = k_MessageHeader + " - None";
|
||||
private const string k_ReferencedPrefix = "- ";
|
||||
|
||||
|
||||
internal BundleDetailList(TreeViewState state) : base(state)
|
||||
{
|
||||
m_Selecteditems = new HashSet<BundleDataInfo>();
|
||||
showBorder = true;
|
||||
}
|
||||
|
||||
internal void Update()
|
||||
{
|
||||
bool dirty = false;
|
||||
foreach (BundleDataInfo bundle in m_Selecteditems) dirty |= bundle.dirty;
|
||||
if (dirty)
|
||||
{
|
||||
Reload();
|
||||
ExpandAll(2);
|
||||
}
|
||||
}
|
||||
|
||||
protected override TreeViewItem BuildRoot()
|
||||
{
|
||||
TreeViewItem root = new TreeViewItem(-1, -1);
|
||||
root.children = new List<TreeViewItem>();
|
||||
if (m_Selecteditems != null)
|
||||
foreach (BundleDataInfo bundle in m_Selecteditems)
|
||||
root.AddChild(AppendBundleToTree(bundle));
|
||||
return root;
|
||||
}
|
||||
|
||||
protected override void RowGUI(RowGUIArgs args)
|
||||
{
|
||||
if (args.item as BundleDetailItem != null)
|
||||
{
|
||||
EditorGUI.HelpBox(
|
||||
new Rect(args.rowRect.x + k_DoubleIndent, args.rowRect.y, args.rowRect.width - k_DoubleIndent, args.rowRect.height),
|
||||
args.item.displayName,
|
||||
(args.item as BundleDetailItem).MessageLevel);
|
||||
}
|
||||
else
|
||||
{
|
||||
Color old = GUI.color;
|
||||
if (args.item.depth == 1 &&
|
||||
(args.item.displayName == k_MessageEmpty || args.item.displayName == k_DependencyEmpty))
|
||||
GUI.color = Model.k_LightGrey;
|
||||
base.RowGUI(args);
|
||||
GUI.color = old;
|
||||
}
|
||||
}
|
||||
|
||||
public override void OnGUI(Rect rect)
|
||||
{
|
||||
m_TotalRect = rect;
|
||||
base.OnGUI(rect);
|
||||
}
|
||||
|
||||
protected override float GetCustomRowHeight(int row, TreeViewItem item)
|
||||
{
|
||||
if (item as BundleDetailItem != null)
|
||||
{
|
||||
float height = DefaultStyles.backgroundEven.CalcHeight(new GUIContent(item.displayName), m_TotalRect.width);
|
||||
return height + 3f;
|
||||
}
|
||||
|
||||
return base.GetCustomRowHeight(row, item);
|
||||
}
|
||||
|
||||
|
||||
protected override void SelectionChanged(IList<int> selectedIds)
|
||||
{
|
||||
base.SelectionChanged(selectedIds);
|
||||
List<string> pathList = new List<string>();
|
||||
|
||||
for (int i = 0; i < selectedIds.Count; ++i)
|
||||
{
|
||||
TreeViewItem item = FindItem(selectedIds[i], rootItem);
|
||||
if (item != null) AddDependentAssetsRecursive(item, pathList);
|
||||
}
|
||||
|
||||
AssetBundleBrowserMain.instance.m_ManageTab.SetAssetListSelection(pathList);
|
||||
}
|
||||
|
||||
private void AddDependentAssetsRecursive(TreeViewItem item, List<string> pathList)
|
||||
{
|
||||
TogglePathTreeViewItem pathItem = item as TogglePathTreeViewItem;
|
||||
if (pathItem != null)
|
||||
if (string.IsNullOrEmpty(pathItem.DisplayNamePrefix) == false && pathList.Contains(pathItem.Path) == false)
|
||||
pathList.Add(pathItem.Path);
|
||||
|
||||
if (item.hasChildren)
|
||||
for (int i = 0; i < item.children.Count; ++i)
|
||||
AddDependentAssetsRecursive(item.children[i], pathList);
|
||||
}
|
||||
|
||||
protected override void DoubleClickedItem(int id)
|
||||
{
|
||||
base.DoubleClickedItem(id);
|
||||
TreeViewItem item = FindItem(id, rootItem);
|
||||
if (item != null)
|
||||
{
|
||||
TogglePathTreeViewItem pathItem = item as TogglePathTreeViewItem;
|
||||
if (pathItem != null)
|
||||
{
|
||||
Object o = AssetDatabase.LoadAssetAtPath<Object>(pathItem.Path);
|
||||
if (o != null)
|
||||
{
|
||||
Selection.activeObject = o;
|
||||
EditorGUIUtility.PingObject(o);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
internal static TreeViewItem AppendBundleToTree(BundleDataInfo bundle)
|
||||
{
|
||||
string itemName = bundle.m_Name.fullNativeName;
|
||||
TreeViewItem bunRoot = new TreeViewItem(itemName.GetHashCode(), 0, itemName);
|
||||
|
||||
string str = itemName + k_SizeHeader;
|
||||
TreeViewItem sz = new TreeViewItem(str.GetHashCode(), 1, k_SizeHeader + bundle.TotalSize());
|
||||
|
||||
str = itemName + k_DependencyHeader;
|
||||
TreeViewItem dependency = new TreeViewItem(str.GetHashCode(), 1, k_DependencyEmpty);
|
||||
List<BundleDependencyInfo> depList = bundle.GetBundleDependencies();
|
||||
if (depList.Count > 0)
|
||||
{
|
||||
dependency.displayName = k_DependencyHeader;
|
||||
foreach (BundleDependencyInfo dep in bundle.GetBundleDependencies())
|
||||
{
|
||||
str = itemName + dep.m_BundleName;
|
||||
TreeViewItem newItem = new TreeViewItem(str.GetHashCode(), 2, dep.m_BundleName);
|
||||
newItem.icon = Model.GetBundleIcon();
|
||||
dependency.AddChild(newItem);
|
||||
|
||||
Dictionary<string, TogglePathTreeViewItem> toAssetItems = new Dictionary<string, TogglePathTreeViewItem>();
|
||||
|
||||
for (int i = 0; i < dep.m_FromAssets.Count; ++i)
|
||||
{
|
||||
TogglePathTreeViewItem item = null;
|
||||
|
||||
if (!toAssetItems.TryGetValue(dep.m_ToAssets[i].fullAssetName, out item))
|
||||
{
|
||||
str = itemName + dep.m_BundleName + dep.m_ToAssets[i].displayName;
|
||||
item = new TogglePathTreeViewItem(str.GetHashCode(), 3, "/" + dep.m_ToAssets[i].displayName, "/" + dep.m_ToAssets[i].fullAssetName);
|
||||
item.icon = AssetDatabase.GetCachedIcon(dep.m_ToAssets[i].fullAssetName) as Texture2D;
|
||||
newItem.AddChild(item);
|
||||
toAssetItems.Add(dep.m_ToAssets[i].fullAssetName, item);
|
||||
}
|
||||
|
||||
str = str + dep.m_FromAssets[i].displayName;
|
||||
TreeViewItem refItem = new TogglePathTreeViewItem(str.GetHashCode(), 4, k_ReferencedPrefix,
|
||||
dep.m_FromAssets[i].displayName, dep.m_FromAssets[i].fullAssetName);
|
||||
refItem.icon = AssetDatabase.GetCachedIcon(dep.m_FromAssets[i].fullAssetName) as Texture2D;
|
||||
item.AddChild(refItem);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
str = itemName + k_MessageHeader;
|
||||
TreeViewItem msg = new TreeViewItem(str.GetHashCode(), 1, k_MessageEmpty);
|
||||
if (bundle.HasMessages())
|
||||
{
|
||||
msg.displayName = k_MessageHeader;
|
||||
List<MessageSystem.Message> currMessages = bundle.GetMessages();
|
||||
|
||||
foreach (MessageSystem.Message currMsg in currMessages)
|
||||
{
|
||||
str = itemName + currMsg.message;
|
||||
msg.AddChild(new BundleDetailItem(str.GetHashCode(), 2, currMsg.message, currMsg.severity));
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
bunRoot.AddChild(sz);
|
||||
bunRoot.AddChild(dependency);
|
||||
bunRoot.AddChild(msg);
|
||||
|
||||
return bunRoot;
|
||||
}
|
||||
|
||||
|
||||
internal void SetItems(IEnumerable<BundleInfo> items)
|
||||
{
|
||||
m_Selecteditems.Clear();
|
||||
foreach (BundleInfo item in items) CollectBundles(item);
|
||||
SetSelection(new List<int>());
|
||||
Reload();
|
||||
ExpandAll(2);
|
||||
}
|
||||
|
||||
internal void CollectBundles(BundleInfo bundle)
|
||||
{
|
||||
BundleDataInfo bunData = bundle as BundleDataInfo;
|
||||
if (bunData != null)
|
||||
{
|
||||
m_Selecteditems.Add(bunData);
|
||||
}
|
||||
else
|
||||
{
|
||||
BundleFolderInfo bunFolder = bundle as BundleFolderInfo;
|
||||
foreach (BundleInfo bun in bunFolder.GetChildList()) CollectBundles(bun);
|
||||
}
|
||||
}
|
||||
|
||||
internal void ExpandAll(int maximumDepth)
|
||||
{
|
||||
List<int> expanded = new List<int>(GetExpanded());
|
||||
FindItems(rootItem, maximumDepth, expanded);
|
||||
SetExpanded(expanded);
|
||||
}
|
||||
|
||||
internal void FindItems(TreeViewItem item, int maximumDepth, List<int> expanded)
|
||||
{
|
||||
if (item.depth >= maximumDepth || !item.hasChildren)
|
||||
return;
|
||||
|
||||
expanded.Add(item.id);
|
||||
for (int i = 0; i < item.children.Count; ++i) FindItems(item.children[i], maximumDepth, expanded);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user