123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445 |
- using UnityEditor;
- using UnityEngine;
- using System.Collections.Generic;
- using UnityEditor.IMGUI.Controls;
- using System.Linq;
- //using System;
- namespace AssetBundleBrowser
- {
- internal class AssetListTree : TreeView
- {
- List<AssetBundleModel.BundleInfo> m_SourceBundles = new List<AssetBundleModel.BundleInfo>();
- AssetBundleManageTab m_Controller;
- List<UnityEngine.Object> m_EmptyObjectList = new List<UnityEngine.Object>();
- internal static MultiColumnHeaderState CreateDefaultMultiColumnHeaderState()
- {
- return new MultiColumnHeaderState(GetColumns());
- }
- private static MultiColumnHeaderState.Column[] GetColumns()
- {
- var retVal = new MultiColumnHeaderState.Column[] {
- new MultiColumnHeaderState.Column(),
- new MultiColumnHeaderState.Column(),
- new MultiColumnHeaderState.Column(),
- new MultiColumnHeaderState.Column()
- };
- retVal[0].headerContent = new GUIContent("Asset", "Short name of asset. For full name select asset and see message below");
- retVal[0].minWidth = 50;
- retVal[0].width = 100;
- retVal[0].maxWidth = 300;
- retVal[0].headerTextAlignment = TextAlignment.Left;
- retVal[0].canSort = true;
- retVal[0].autoResize = true;
- retVal[1].headerContent = new GUIContent("Bundle", "Bundle name. 'auto' means asset was pulled in due to dependency");
- retVal[1].minWidth = 50;
- retVal[1].width = 100;
- retVal[1].maxWidth = 300;
- retVal[1].headerTextAlignment = TextAlignment.Left;
- retVal[1].canSort = true;
- retVal[1].autoResize = true;
- retVal[2].headerContent = new GUIContent("Size", "Size on disk");
- retVal[2].minWidth = 30;
- retVal[2].width = 75;
- retVal[2].maxWidth = 100;
- retVal[2].headerTextAlignment = TextAlignment.Left;
- retVal[2].canSort = true;
- retVal[2].autoResize = true;
- retVal[3].headerContent = new GUIContent("!", "Errors, Warnings, or Info");
- retVal[3].minWidth = 16;
- retVal[3].width = 16;
- retVal[3].maxWidth = 16;
- retVal[3].headerTextAlignment = TextAlignment.Left;
- retVal[3].canSort = true;
- retVal[3].autoResize = false;
- return retVal;
- }
- enum MyColumns
- {
- Asset,
- Bundle,
- Size,
- Message
- }
- internal enum SortOption
- {
- Asset,
- Bundle,
- Size,
- Message
- }
- SortOption[] m_SortOptions =
- {
- SortOption.Asset,
- SortOption.Bundle,
- SortOption.Size,
- SortOption.Message
- };
- internal AssetListTree(TreeViewState state, MultiColumnHeaderState mchs, AssetBundleManageTab ctrl ) : base(state, new MultiColumnHeader(mchs))
- {
- m_Controller = ctrl;
- showBorder = true;
- showAlternatingRowBackgrounds = true;
- DefaultStyles.label.richText = true;
- multiColumnHeader.sortingChanged += OnSortingChanged;
- }
- internal void Update()
- {
- bool dirty = false;
- foreach (var bundle in m_SourceBundles)
- {
- dirty |= bundle.dirty;
- }
- if (dirty)
- Reload();
- }
- public override void OnGUI(Rect rect)
- {
- base.OnGUI(rect);
- if (Event.current.type == EventType.MouseDown && Event.current.button == 0 && rect.Contains(Event.current.mousePosition))
- {
- SetSelection(new int[0], TreeViewSelectionOptions.FireSelectionChanged);
- }
- }
- protected override IList<TreeViewItem> BuildRows(TreeViewItem root)
- {
- var rows = base.BuildRows(root);
- SortIfNeeded(root, rows);
- return rows;
- }
- internal void SetSelectedBundles(IEnumerable<AssetBundleModel.BundleInfo> bundles)
- {
- m_Controller.SetSelectedItems(null);
- m_SourceBundles = bundles.ToList();
- SetSelection(new List<int>());
- Reload();
- }
- protected override TreeViewItem BuildRoot()
- {
- var root = AssetBundleModel.Model.CreateAssetListTreeView(m_SourceBundles);
- return root;
- }
- protected override void RowGUI(RowGUIArgs args)
- {
- for (int i = 0; i < args.GetNumVisibleColumns(); ++i)
- CellGUI(args.GetCellRect(i), args.item as AssetBundleModel.AssetTreeItem, args.GetColumn(i), ref args);
- }
- private void CellGUI(Rect cellRect, AssetBundleModel.AssetTreeItem item, int column, ref RowGUIArgs args)
- {
- Color oldColor = GUI.color;
- CenterRectUsingSingleLineHeight(ref cellRect);
- if(column != 3)
- GUI.color = item.itemColor;
- switch (column)
- {
- case 0:
- {
- var iconRect = new Rect(cellRect.x + 1, cellRect.y + 1, cellRect.height - 2, cellRect.height - 2);
- if(item.icon != null)
- GUI.DrawTexture(iconRect, item.icon, ScaleMode.ScaleToFit);
- DefaultGUI.Label(
- new Rect(cellRect.x + iconRect.xMax + 1, cellRect.y, cellRect.width - iconRect.width, cellRect.height),
- item.displayName,
- args.selected,
- args.focused);
- }
- break;
- case 1:
- DefaultGUI.Label(cellRect, item.asset.bundleName, args.selected, args.focused);
- break;
- case 2:
- DefaultGUI.Label(cellRect, item.asset.GetSizeString(), args.selected, args.focused);
- break;
- case 3:
- var icon = item.MessageIcon();
- if (icon != null)
- {
- var iconRect = new Rect(cellRect.x, cellRect.y, cellRect.height, cellRect.height);
- GUI.DrawTexture(iconRect, icon, ScaleMode.ScaleToFit);
- }
- break;
- }
- GUI.color = oldColor;
- }
- protected override void DoubleClickedItem(int id)
- {
- var assetItem = FindItem(id, rootItem) as AssetBundleModel.AssetTreeItem;
- if (assetItem != null)
- {
- Object o = AssetDatabase.LoadAssetAtPath<Object>(assetItem.asset.fullAssetName);
- EditorGUIUtility.PingObject(o);
- Selection.activeObject = o;
- }
- }
- protected override void SelectionChanged(IList<int> selectedIds)
- {
- List<Object> selectedObjects = new List<Object>();
- List<AssetBundleModel.AssetInfo> selectedAssets = new List<AssetBundleModel.AssetInfo>();
- foreach (var id in selectedIds)
- {
- var assetItem = FindItem(id, rootItem) as AssetBundleModel.AssetTreeItem;
- if (assetItem != null)
- {
- Object o = AssetDatabase.LoadAssetAtPath<Object>(assetItem.asset.fullAssetName);
- selectedObjects.Add(o);
- Selection.activeObject = o;
- selectedAssets.Add(assetItem.asset);
- }
- }
- m_Controller.SetSelectedItems(selectedAssets);
- Selection.objects = selectedObjects.ToArray();
- }
- protected override bool CanBeParent(TreeViewItem item)
- {
- return false;
- }
- protected override bool CanStartDrag(CanStartDragArgs args)
- {
- args.draggedItemIDs = GetSelection();
- return true;
- }
- protected override void SetupDragAndDrop(SetupDragAndDropArgs args)
- {
- DragAndDrop.PrepareStartDrag();
- DragAndDrop.objectReferences = m_EmptyObjectList.ToArray();
- List<AssetBundleModel.AssetTreeItem> items =
- new List<AssetBundleModel.AssetTreeItem>(args.draggedItemIDs.Select(id => FindItem(id, rootItem) as AssetBundleModel.AssetTreeItem));
- DragAndDrop.paths = items.Select(a => a.asset.fullAssetName).ToArray();
- DragAndDrop.SetGenericData("AssetListTreeSource", this);
- DragAndDrop.StartDrag("AssetListTree");
- }
-
- protected override DragAndDropVisualMode HandleDragAndDrop(DragAndDropArgs args)
- {
- if(IsValidDragDrop(args))
- {
- if (args.performDrop)
- {
- AssetBundleModel.Model.MoveAssetToBundle(DragAndDrop.paths, m_SourceBundles[0].m_Name.bundleName, m_SourceBundles[0].m_Name.variant);
- AssetBundleModel.Model.ExecuteAssetMove();
- foreach (var bundle in m_SourceBundles)
- {
- bundle.RefreshAssetList();
- }
- m_Controller.UpdateSelectedBundles(m_SourceBundles);
- }
- return DragAndDropVisualMode.Copy;//Move;
- }
- return DragAndDropVisualMode.Rejected;
- }
- protected bool IsValidDragDrop(DragAndDropArgs args)
- {
- //can't do drag & drop if data source is read only
- if (AssetBundleModel.Model.DataSource.IsReadOnly ())
- return false;
- //can't drag onto none or >1 bundles
- if (m_SourceBundles.Count == 0 || m_SourceBundles.Count > 1)
- return false;
-
- //can't drag nothing
- if (DragAndDrop.paths == null || DragAndDrop.paths.Length == 0)
- return false;
- //can't drag into a folder
- var folder = m_SourceBundles[0] as AssetBundleModel.BundleFolderInfo;
- if (folder != null)
- return false;
- var data = m_SourceBundles[0] as AssetBundleModel.BundleDataInfo;
- if(data == null)
- return false; // this should never happen.
- var thing = DragAndDrop.GetGenericData("AssetListTreeSource") as AssetListTree;
- if (thing != null)
- return false;
-
- if(data.IsEmpty())
- return true;
- if (data.isSceneBundle)
- {
- foreach (var assetPath in DragAndDrop.paths)
- {
- if ((AssetDatabase.GetMainAssetTypeAtPath(assetPath) != typeof(SceneAsset)) &&
- (!AssetDatabase.IsValidFolder(assetPath)))
- return false;
- }
- }
- else
- {
- foreach (var assetPath in DragAndDrop.paths)
- {
- if (AssetDatabase.GetMainAssetTypeAtPath(assetPath) == typeof(SceneAsset))
- return false;
- }
- }
- return true;
- }
- protected override void ContextClickedItem(int id)
- {
- if (AssetBundleModel.Model.DataSource.IsReadOnly ()) {
- return;
- }
- List<AssetBundleModel.AssetTreeItem> selectedNodes = new List<AssetBundleModel.AssetTreeItem>();
- foreach(var nodeID in GetSelection())
- {
- selectedNodes.Add(FindItem(nodeID, rootItem) as AssetBundleModel.AssetTreeItem);
- }
- if(selectedNodes.Count > 0)
- {
- GenericMenu menu = new GenericMenu();
- menu.AddItem(new GUIContent("Remove asset(s) from bundle."), false, RemoveAssets, selectedNodes);
- menu.ShowAsContext();
- }
- }
- void RemoveAssets(object obj)
- {
- var selectedNodes = obj as List<AssetBundleModel.AssetTreeItem>;
- var assets = new List<AssetBundleModel.AssetInfo>();
- //var bundles = new List<AssetBundleModel.BundleInfo>();
- foreach (var node in selectedNodes)
- {
- if (node.asset.bundleName != string.Empty)
- assets.Add(node.asset);
- }
- AssetBundleModel.Model.MoveAssetToBundle(assets, string.Empty, string.Empty);
- AssetBundleModel.Model.ExecuteAssetMove();
- foreach (var bundle in m_SourceBundles)
- {
- bundle.RefreshAssetList();
- }
- m_Controller.UpdateSelectedBundles(m_SourceBundles);
- //ReloadAndSelect(new List<int>());
- }
- protected override void KeyEvent()
- {
- if (m_SourceBundles.Count > 0 && Event.current.keyCode == KeyCode.Delete && GetSelection().Count > 0)
- {
- List<AssetBundleModel.AssetTreeItem> selectedNodes = new List<AssetBundleModel.AssetTreeItem>();
- foreach (var nodeID in GetSelection())
- {
- selectedNodes.Add(FindItem(nodeID, rootItem) as AssetBundleModel.AssetTreeItem);
- }
- RemoveAssets(selectedNodes);
- }
- }
- void OnSortingChanged(MultiColumnHeader multiColumnHeader)
- {
- SortIfNeeded(rootItem, GetRows());
- }
- void SortIfNeeded(TreeViewItem root, IList<TreeViewItem> rows)
- {
- if (rows.Count <= 1)
- return;
- if (multiColumnHeader.sortedColumnIndex == -1)
- return;
- SortByColumn();
- rows.Clear();
- for (int i = 0; i < root.children.Count; i++)
- rows.Add(root.children[i]);
- Repaint();
- }
- void SortByColumn()
- {
- var sortedColumns = multiColumnHeader.state.sortedColumns;
- if (sortedColumns.Length == 0)
- return;
- List<AssetBundleModel.AssetTreeItem> assetList = new List<AssetBundleModel.AssetTreeItem>();
- foreach(var item in rootItem.children)
- {
- assetList.Add(item as AssetBundleModel.AssetTreeItem);
- }
- var orderedItems = InitialOrder(assetList, sortedColumns);
- rootItem.children = orderedItems.Cast<TreeViewItem>().ToList();
- }
- IOrderedEnumerable<AssetBundleModel.AssetTreeItem> InitialOrder(IEnumerable<AssetBundleModel.AssetTreeItem> myTypes, int[] columnList)
- {
- SortOption sortOption = m_SortOptions[columnList[0]];
- bool ascending = multiColumnHeader.IsSortedAscending(columnList[0]);
- switch (sortOption)
- {
- case SortOption.Asset:
- return myTypes.Order(l => l.displayName, ascending);
- case SortOption.Size:
- return myTypes.Order(l => l.asset.fileSize, ascending);
- case SortOption.Message:
- return myTypes.Order(l => l.HighestMessageLevel(), ascending);
- case SortOption.Bundle:
- default:
- return myTypes.Order(l => l.asset.bundleName, ascending);
- }
-
- }
- private void ReloadAndSelect(IList<int> hashCodes)
- {
- Reload();
- SetSelection(hashCodes);
- SelectionChanged(hashCodes);
- }
- }
- static class MyExtensionMethods
- {
- internal static IOrderedEnumerable<T> Order<T, TKey>(this IEnumerable<T> source, System.Func<T, TKey> selector, bool ascending)
- {
- if (ascending)
- {
- return source.OrderBy(selector);
- }
- else
- {
- return source.OrderByDescending(selector);
- }
- }
- internal static IOrderedEnumerable<T> ThenBy<T, TKey>(this IOrderedEnumerable<T> source, System.Func<T, TKey> selector, bool ascending)
- {
- if (ascending)
- {
- return source.ThenBy(selector);
- }
- else
- {
- return source.ThenByDescending(selector);
- }
- }
- }
- }
|