939 lines
32 KiB
C#
939 lines
32 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Nullframes.Intrigues.EDITOR.Utils;
|
|
using Nullframes.Intrigues.Graph;
|
|
using Nullframes.Intrigues.Utils;
|
|
using UnityEditor;
|
|
using UnityEditor.UIElements;
|
|
using UnityEngine;
|
|
using UnityEngine.UIElements;
|
|
|
|
namespace Nullframes.Intrigues.EDITOR {
|
|
public class AssetDbWindow : EditorWindow {
|
|
private static IEDatabase database;
|
|
private bool isDbLoaded;
|
|
private bool isUIInitialized;
|
|
|
|
private VisualElement root;
|
|
private VisualElement scrollView;
|
|
private VisualElement tagList;
|
|
|
|
private List< VisualElement > errorMsg;
|
|
private ObjectField databaseField;
|
|
|
|
private TextField searchField;
|
|
private TextField newCategoryField;
|
|
|
|
private Button newCategoryBtn;
|
|
|
|
private readonly List< VisualElement > selectedItems = new();
|
|
private VisualElement lastSelected;
|
|
|
|
private bool isCtrlPressed;
|
|
|
|
private List< VisualElement > currentTags = new();
|
|
|
|
private List< VisualElement > categories {
|
|
get {
|
|
return scrollView.Query< VisualElement >()
|
|
.Where(e => e.ClassListContains("categoryField"))
|
|
.ToList();
|
|
}
|
|
}
|
|
|
|
private void OnEnable() {
|
|
isDbLoaded = false;
|
|
isUIInitialized = false;
|
|
|
|
Undo.undoRedoPerformed += OnUndoRedoPerformed;
|
|
EditorApplication.update += OnEditorUpdate;
|
|
}
|
|
|
|
private void OnDisable() {
|
|
Undo.undoRedoPerformed -= OnUndoRedoPerformed;
|
|
EditorApplication.update -= OnEditorUpdate;
|
|
}
|
|
|
|
private void OnUndoRedoPerformed() {
|
|
if ( database == null ) return;
|
|
var _selectedItems = new List< VisualElement >(selectedItems);
|
|
UpdateDbList(database);
|
|
|
|
if ( _selectedItems.Count > 0 ) {
|
|
var groupedByParent = _selectedItems
|
|
.Where(item => item is { parent: not null })
|
|
.GroupBy(item => item.parent)
|
|
.OrderByDescending(g => g.Count())
|
|
.FirstOrDefault();
|
|
|
|
if ( groupedByParent != null ) {
|
|
foreach ( var item in groupedByParent ) {
|
|
Select(item);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private void OnEditorUpdate() {
|
|
if ( isCtrlPressed ) {
|
|
foreach ( var tag in currentTags ) {
|
|
tag.Hide();
|
|
}
|
|
}
|
|
}
|
|
|
|
[MenuItem("Tools/Nullframes/Sprite dB")]
|
|
public static void Open() {
|
|
var wnd = GetWindow< AssetDbWindow >();
|
|
wnd.position = new Rect(100, 100, 800, 800);
|
|
wnd.titleContent = new GUIContent("SpriteDb");
|
|
}
|
|
|
|
public static void Open(IEDatabase db) {
|
|
if ( db == null ) return;
|
|
|
|
database = db;
|
|
EditorPrefs.SetString("Nullframes.PendingDatabasePath", AssetDatabase.GetAssetPath(db));
|
|
|
|
var wnd = GetWindow< AssetDbWindow >();
|
|
wnd.position = new Rect(100, 100, 800, 800);
|
|
wnd.titleContent = new GUIContent("SpriteDb");
|
|
}
|
|
|
|
private void OnGUI() {
|
|
if ( !isUIInitialized ) return;
|
|
|
|
if ( database == null ) {
|
|
errorMsg[ 0 ].Show();
|
|
errorMsg[ 1 ].Hide();
|
|
titleContent = new GUIContent("SpriteDb - Not Initialized");
|
|
|
|
isDbLoaded = false;
|
|
return;
|
|
}
|
|
|
|
if ( !isDbLoaded ) {
|
|
LoadDb();
|
|
if ( !isDbLoaded ) {
|
|
return;
|
|
}
|
|
}
|
|
|
|
errorMsg[ 0 ].Hide();
|
|
|
|
if ( database.spriteDatabase.Count > 0 ) {
|
|
errorMsg[ 1 ].Hide();
|
|
} else {
|
|
errorMsg[ 1 ].Show();
|
|
}
|
|
|
|
titleContent = new GUIContent("SpriteDb - " + database.name);
|
|
|
|
// Key handler
|
|
|
|
Event e = Event.current;
|
|
if ( e.type == EventType.KeyDown && e.control ) {
|
|
isCtrlPressed = true;
|
|
currentTags = scrollView.Query< VisualElement >()
|
|
.Where(t => t.ClassListContains("tagLabel"))
|
|
.ToList();
|
|
} else if ( e.type == EventType.KeyUp && e.keyCode == KeyCode.LeftControl ||
|
|
e.keyCode == KeyCode.RightControl ) {
|
|
isCtrlPressed = false;
|
|
foreach ( var tag in currentTags ) {
|
|
tag.Show();
|
|
}
|
|
|
|
currentTags.Clear();
|
|
}
|
|
|
|
if ( Event.current.type == EventType.KeyDown ) {
|
|
if ( Event.current.keyCode == KeyCode.Delete || Event.current.keyCode == KeyCode.Backspace ) {
|
|
if ( selectedItems.Count > 0 ) {
|
|
DeleteSelectedItems();
|
|
Save();
|
|
Event.current.Use();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public void CreateGUI() {
|
|
root = rootVisualElement;
|
|
|
|
var visualTree = (VisualTreeAsset)EditorGUIUtility.Load("Nullframes/AssetDb.uxml");
|
|
visualTree.CloneTree(root);
|
|
|
|
var styleSheet = (StyleSheet)EditorGUIUtility.Load(EditorGUIUtility.isProSkin
|
|
? "Nullframes/AssetDb_Dark.uss"
|
|
: "Nullframes/AssetDb_Light.uss");
|
|
root.styleSheets.Add(styleSheet);
|
|
|
|
var toolbar = new Toolbar();
|
|
|
|
var flexibleSpacer = new VisualElement {
|
|
style = {
|
|
flexGrow = 1
|
|
}
|
|
};
|
|
toolbar.Add(flexibleSpacer);
|
|
|
|
var docButton = new Button(() => {
|
|
Application.OpenURL("https://www.wlabsocks.com/wiki/index.php?title=Sprite_dB");
|
|
}) { text = "?" };
|
|
|
|
toolbar.Add(docButton);
|
|
root.Insert(0, toolbar);
|
|
|
|
errorMsg = new List< VisualElement > {
|
|
root.Q< VisualElement >("errorMsg"),
|
|
root.Q< VisualElement >("errorMsg2")
|
|
};
|
|
|
|
searchField = root.Q< TextField >("searchField");
|
|
newCategoryField = root.Q< TextField >("newCategoryField");
|
|
databaseField = root.Q< ObjectField >("databaseField");
|
|
databaseField.objectType = typeof( IEDatabase );
|
|
|
|
newCategoryBtn = root.Q< Button >("newCategoryBtn");
|
|
|
|
scrollView = rootVisualElement.Q< ScrollView >("scrollView");
|
|
tagList = rootVisualElement.Q< ScrollView >("tagList");
|
|
|
|
#region New Category Button
|
|
|
|
newCategoryBtn.clicked += () => {
|
|
if ( database != null && newCategoryField.text != string.Empty ) {
|
|
if ( database.spriteDatabase.ContainsKey(newCategoryField.text) ) return;
|
|
// database.assetDatabase.Add(newCategoryField.text, new AssetDb());
|
|
|
|
var _categoryField = CreateCategory(newCategoryField.text);
|
|
|
|
scrollView.Add(_categoryField);
|
|
|
|
var assetField = CreateAssetField(scrollView);
|
|
assetField.userData = _categoryField;
|
|
|
|
_categoryField.userData = assetField;
|
|
|
|
Save();
|
|
}
|
|
};
|
|
|
|
#endregion
|
|
|
|
#region Object Field
|
|
|
|
databaseField.RegisterValueChangedCallback(_ => {
|
|
string path = string.Empty;
|
|
|
|
if ( databaseField.value != null ) {
|
|
database = (IEDatabase)databaseField.value;
|
|
path = AssetDatabase.GetAssetPath(database);
|
|
} else {
|
|
database = null;
|
|
}
|
|
|
|
UpdateDbList(database);
|
|
|
|
EditorPrefs.SetString("Nullframes.PendingDatabasePath", path);
|
|
});
|
|
|
|
#endregion
|
|
|
|
#region Search Field
|
|
|
|
var searchField_placeHolder = searchField.Q< Label >("placeHolder");
|
|
|
|
searchField.RegisterValueChangedCallback(_ => {
|
|
if ( string.IsNullOrEmpty(searchField.value) ) {
|
|
searchField_placeHolder.style.display = DisplayStyle.Flex;
|
|
foreach ( var category in categories ) {
|
|
var _assetField = (VisualElement)category.userData;
|
|
|
|
foreach ( var _asset in _assetField.Children() ) {
|
|
var obj = _asset.userData as Sprite;
|
|
if ( obj == null ) continue;
|
|
_asset.Show();
|
|
}
|
|
|
|
category.style.display = DisplayStyle.Flex;
|
|
}
|
|
} else {
|
|
searchField_placeHolder.style.display = DisplayStyle.None;
|
|
|
|
foreach ( var category in categories ) {
|
|
var _assetField = (VisualElement)category.userData;
|
|
|
|
foreach ( var _asset in _assetField.Children() ) {
|
|
var obj = _asset.userData as Sprite;
|
|
if ( obj == null ) continue;
|
|
if ( obj.name.Contains(searchField.value, StringComparison.OrdinalIgnoreCase) ) {
|
|
_asset.Show();
|
|
} else {
|
|
_asset.Hide();
|
|
}
|
|
}
|
|
|
|
if ( _assetField.Children().Count(c => c.style.display == DisplayStyle.Flex) == 0 ) {
|
|
category.style.display = DisplayStyle.None;
|
|
}
|
|
}
|
|
}
|
|
});
|
|
|
|
#endregion
|
|
|
|
#region Category Field
|
|
|
|
var categoryField_placeHolder = newCategoryField.Q< Label >("placeHolder");
|
|
|
|
newCategoryField.RegisterValueChangedCallback(_ => {
|
|
categoryField_placeHolder.style.display = string.IsNullOrEmpty(newCategoryField.value)
|
|
? DisplayStyle.Flex
|
|
: DisplayStyle.None;
|
|
});
|
|
|
|
#endregion
|
|
|
|
bool isAnyChanged = false;
|
|
|
|
root.RegisterCallback< MouseDownEvent >(evt => {
|
|
if ( evt.button != 0 ) return;
|
|
|
|
ClearSelection();
|
|
|
|
|
|
var tagFields = scrollView.Query< TextField >()
|
|
.Where(e => e.ClassListContains("tagField"))
|
|
.ToList();
|
|
|
|
var tagLabels = scrollView.Query< Label >()
|
|
.Where(e => e.ClassListContains("tagLabel"))
|
|
.ToList();
|
|
|
|
foreach ( var tagLbl in tagLabels ) {
|
|
tagLbl.Show();
|
|
}
|
|
|
|
foreach (var tagField in tagFields)
|
|
{
|
|
tagField.Hide();
|
|
|
|
if (tagField.userData is ValueTuple<bool, Label> { Item1: true } data)
|
|
{
|
|
tagField.userData = (false, data.Item2);
|
|
tagField.SetValueWithoutNotify(string.Empty);
|
|
isAnyChanged = true;
|
|
}
|
|
}
|
|
|
|
if ( isAnyChanged ) {
|
|
isAnyChanged = false;
|
|
Save();
|
|
}
|
|
});
|
|
|
|
EditorRoutine.CallNextFrame(LoadDb);
|
|
|
|
isUIInitialized = true;
|
|
}
|
|
|
|
private void LoadDb() {
|
|
if ( isDbLoaded ) return;
|
|
|
|
if ( database == null ) {
|
|
string path = EditorPrefs.GetString("Nullframes.PendingDatabasePath", null);
|
|
if ( !string.IsNullOrEmpty(path) ) {
|
|
database = AssetDatabase.LoadAssetAtPath< IEDatabase >(path);
|
|
}
|
|
}
|
|
|
|
if ( database == null ) return;
|
|
|
|
databaseField.value = database;
|
|
|
|
UpdateDbList(database);
|
|
|
|
//
|
|
|
|
UpdateTagList();
|
|
|
|
isDbLoaded = true;
|
|
}
|
|
|
|
private void UpdateTagList() {
|
|
var uniqueTags = scrollView.Query<Label>()
|
|
.Where(c => c.ClassListContains("tagLabel"))
|
|
.ToList()
|
|
.Select(label => label.text)
|
|
.Distinct()
|
|
.ToList();
|
|
|
|
tagList.Clear();
|
|
|
|
foreach ( var tag in uniqueTags ) {
|
|
var tagLabel = CreateTagLabel(tag, null, tagList);
|
|
tagLabel.style.marginRight = (4);
|
|
tagList.Add(tagLabel);
|
|
}
|
|
|
|
isDbLoaded = true;
|
|
}
|
|
|
|
private void UpdateDbList(IEDatabase db) {
|
|
scrollView.Clear();
|
|
ClearSelection();
|
|
|
|
if ( db == null ) return;
|
|
|
|
foreach ( var _db in db.spriteDatabase ) {
|
|
var _categoryField = CreateCategory(_db.Key);
|
|
|
|
scrollView.Add(_categoryField);
|
|
|
|
var assetField = CreateAssetField(scrollView);
|
|
assetField.userData = _categoryField;
|
|
|
|
_categoryField.userData = assetField;
|
|
|
|
foreach ( var asset in _db.Value.Sprites ) {
|
|
var _asset = CreateAsset(asset.Key, asset.Value);
|
|
assetField.Add(_asset);
|
|
}
|
|
}
|
|
}
|
|
|
|
private VisualElement CreateCategory(string categoryName) {
|
|
var _categoryField = new VisualElement() {
|
|
name = categoryName
|
|
};
|
|
|
|
_categoryField.AddManipulator(new ContextualMenuManipulator(tagEvt => {
|
|
tagEvt.menu.InsertAction(0, "Remove Category", _ => {
|
|
var _assetField = (VisualElement)_categoryField.userData;
|
|
|
|
_categoryField.RemoveFromHierarchy();
|
|
_assetField.RemoveFromHierarchy();
|
|
|
|
Save();
|
|
});
|
|
}));
|
|
|
|
_categoryField.AddToClassList("categoryField");
|
|
|
|
var _foldButton = new VisualElement();
|
|
_foldButton.AddToClassList("foldoutIcon");
|
|
|
|
var _categoryLabel = new Label(categoryName);
|
|
_categoryLabel.AddToClassList("categoryLabel");
|
|
|
|
_categoryField.Add(_foldButton);
|
|
_categoryField.Add(_categoryLabel);
|
|
|
|
_categoryField.RegisterCallback< DragUpdatedEvent >(evt => {
|
|
DragAndDrop.visualMode = DragAndDropVisualMode.Copy;
|
|
evt.StopPropagation();
|
|
});
|
|
|
|
_categoryField.RegisterCallback< DragPerformEvent >(evt => {
|
|
var _assetField = _categoryField.userData as VisualElement;
|
|
|
|
if ( _assetField == null ) return;
|
|
|
|
DragAndDrop.AcceptDrag();
|
|
|
|
if ( DragAndDrop.GetGenericData("DraggedAssetsKey") is List< VisualElement > list ) {
|
|
foreach ( var item in list ) {
|
|
item.RemoveFromHierarchy();
|
|
_assetField.Add(item);
|
|
}
|
|
}
|
|
|
|
if (DragAndDrop.objectReferences != null)
|
|
{
|
|
foreach (var obj in DragAndDrop.objectReferences)
|
|
{
|
|
string path = AssetDatabase.GetAssetPath(obj);
|
|
if (string.IsNullOrEmpty(path))
|
|
continue;
|
|
|
|
var sprites = AssetDatabase.LoadAllAssetsAtPath(path).OfType<Sprite>();
|
|
|
|
foreach (var sprite in sprites)
|
|
{
|
|
if (database.spriteDatabase[categoryName].Sprites.ContainsKey(sprite))
|
|
continue;
|
|
|
|
var _asset = CreateAsset(sprite, new List<string>());
|
|
_assetField.Add(_asset);
|
|
}
|
|
}
|
|
}
|
|
|
|
Save();
|
|
|
|
evt.StopPropagation();
|
|
});
|
|
|
|
bool foldOut = false;
|
|
|
|
_foldButton.RegisterCallback< MouseDownEvent >(evt => {
|
|
if ( evt.button != 0 ) return;
|
|
Foldout();
|
|
|
|
evt.StopPropagation();
|
|
});
|
|
|
|
void Foldout() {
|
|
foldOut = !foldOut;
|
|
|
|
var assetField = (VisualElement)_categoryField.userData;
|
|
|
|
if ( foldOut ) {
|
|
_foldButton.style.rotate = new StyleRotate(new Rotate(new Angle(180f)));
|
|
foreach ( var _child in assetField.Children() ) {
|
|
_child.Hide();
|
|
}
|
|
} else {
|
|
_foldButton.style.rotate = new StyleRotate(new Rotate(new Angle(270f)));
|
|
foreach ( var _child in assetField.Children() ) {
|
|
_child.Show();
|
|
}
|
|
}
|
|
}
|
|
|
|
return _categoryField;
|
|
}
|
|
|
|
private VisualElement CreateAssetField(VisualElement categoryField) {
|
|
var _assetField = new VisualElement();
|
|
_assetField.AddToClassList("assetField");
|
|
|
|
categoryField.userData = _assetField;
|
|
|
|
categoryField.Add(_assetField);
|
|
|
|
return _assetField;
|
|
}
|
|
|
|
private VisualElement CreateAsset(Sprite obj, IEnumerable< string > tags) {
|
|
var _asset = new VisualElement() { userData = obj };
|
|
_asset.AddToClassList("assetPreview");
|
|
|
|
var tagNameField = IEGraphUtility.CreateTextField();
|
|
tagNameField.userData = false;
|
|
tagNameField.Hide();
|
|
tagNameField.AddToClassList("tagField");
|
|
_asset.Add(tagNameField);
|
|
|
|
bool isAnyChanges = false;
|
|
|
|
tagNameField.RegisterValueChangedCallback(evt => {
|
|
if ( tagNameField.userData is ValueTuple< bool, Label > { Item2: not null } data ) {
|
|
var selected = data.Item2;
|
|
|
|
tagNameField.value = tagNameField.text.RemoveSpecialCharacters();
|
|
selected.text = "#" + tagNameField.value;
|
|
|
|
if ( evt.previousValue != evt.newValue ) {
|
|
isAnyChanges = true;
|
|
tagNameField.userData = ( true, selected );
|
|
}
|
|
}
|
|
});
|
|
|
|
tagNameField.RegisterCallback< KeyDownEvent >(evt => {
|
|
if ( evt.keyCode == KeyCode.Return || evt.keyCode == KeyCode.KeypadEnter ) {
|
|
tagNameField.Hide();
|
|
|
|
var tagLabels = scrollView.Query< Label >()
|
|
.Where(e => e.ClassListContains("tagLabel"))
|
|
.ToList();
|
|
|
|
foreach ( var tagLabel in tagLabels )
|
|
tagLabel.Show();
|
|
|
|
if ( isAnyChanges )
|
|
Save();
|
|
|
|
evt.StopPropagation();
|
|
}
|
|
});
|
|
|
|
|
|
foreach ( var tag in tags ) {
|
|
var tagLabel = CreateTagLabel(tag, tagNameField, _asset);
|
|
_asset.Add(tagLabel);
|
|
}
|
|
|
|
var _label = new Label() { text = obj.name.Shortener() };
|
|
_label.AddToClassList("assetLabel");
|
|
_asset.tooltip = obj.name;
|
|
_asset.Add(_label);
|
|
|
|
Texture2D texture = obj.texture;
|
|
_asset.style.backgroundImage = texture;
|
|
|
|
// EditorApplication.update += TryUpdatePreview;
|
|
//
|
|
// void TryUpdatePreview()
|
|
// {
|
|
// Texture2D bgTexture = AssetPreview.GetAssetPreview(obj);
|
|
// if (bgTexture != null)
|
|
// {
|
|
// _asset.style.backgroundImage = bgTexture;
|
|
// EditorApplication.update -= TryUpdatePreview;
|
|
// }
|
|
// }
|
|
|
|
SetupSelectable(_asset);
|
|
|
|
bool isDragging = false;
|
|
Vector2 dragStart = Vector2.zero;
|
|
|
|
_asset.RegisterCallback< MouseDownEvent >(evt => {
|
|
if ( evt.button != 0 ) return;
|
|
dragStart = evt.mousePosition;
|
|
isDragging = true;
|
|
evt.StopPropagation();
|
|
});
|
|
|
|
_asset.RegisterCallback< MouseMoveEvent >(evt => {
|
|
if ( !isDragging ) return;
|
|
|
|
if ( ( evt.mousePosition - dragStart ).sqrMagnitude > 16f ) {
|
|
DragAndDrop.PrepareStartDrag();
|
|
var itemsToDrag = selectedItems.Contains(_asset)
|
|
? selectedItems.ToList()
|
|
: new List< VisualElement > { _asset };
|
|
|
|
DragAndDrop.SetGenericData("DraggedAssetsKey", itemsToDrag);
|
|
DragAndDrop.SetGenericData("MyAssetKey", _asset);
|
|
DragAndDrop.StartDrag("Dragging Element");
|
|
|
|
isDragging = false;
|
|
}
|
|
});
|
|
|
|
_asset.RegisterCallback< MouseUpEvent >(evt => {
|
|
if ( evt.button == 0 )
|
|
isDragging = false;
|
|
});
|
|
|
|
_asset.RegisterCallback< DragUpdatedEvent >(evt => {
|
|
DragAndDrop.visualMode = DragAndDropVisualMode.Copy;
|
|
evt.StopPropagation();
|
|
});
|
|
|
|
_asset.RegisterCallback< DragPerformEvent >(evt => {
|
|
DragAndDrop.AcceptDrag();
|
|
|
|
var parent = _asset.parent;
|
|
if ( parent == null )
|
|
return;
|
|
|
|
int baseIndex = parent.IndexOf(_asset);
|
|
if ( baseIndex < 0 )
|
|
return;
|
|
|
|
if ( DragAndDrop.objectReferences != null ) {
|
|
string categoryName = ( (VisualElement)parent.userData ).name;
|
|
|
|
int insertIndex = baseIndex;
|
|
foreach (var _obj in DragAndDrop.objectReferences)
|
|
{
|
|
string path = AssetDatabase.GetAssetPath(_obj);
|
|
|
|
var sprites = AssetDatabase.LoadAllAssetsAtPath(path).OfType<Sprite>();
|
|
|
|
foreach (var sprite in sprites)
|
|
{
|
|
if (database.spriteDatabase[categoryName].Sprites.ContainsKey(sprite))
|
|
continue;
|
|
|
|
var newAsset = CreateAsset(sprite, new List<string>());
|
|
parent.Insert(insertIndex++, newAsset);
|
|
}
|
|
}
|
|
}
|
|
|
|
if ( DragAndDrop.GetGenericData("MyTagKey") is Label draggedTag ) {
|
|
var _selectedItems = new List< VisualElement >(selectedItems) { _asset };
|
|
string tagText = draggedTag.text.TrimStart('#');
|
|
|
|
foreach ( var selectedItem in _selectedItems ) {
|
|
var existingTags = selectedItem.Query< Label >()
|
|
.Where(e => e.ClassListContains("tagLabel"))
|
|
.ToList() // <- UQueryBuilder<Label> → List<Label>
|
|
.Select(lbl => lbl.text.TrimStart('#'))
|
|
.ToList();
|
|
|
|
if ( existingTags.Contains(tagText) || existingTags.Count >= 5 )
|
|
continue;
|
|
|
|
var tagField = selectedItem.Q< TextField >();
|
|
|
|
var tagLabel = CreateTagLabel(tagText, tagField, selectedItem);
|
|
selectedItem.Add(tagLabel);
|
|
}
|
|
|
|
Save();
|
|
}
|
|
|
|
if ( DragAndDrop.GetGenericData("MyAssetKey") is VisualElement draggedItem ) {
|
|
if ( draggedItem == _asset || selectedItems.Contains(_asset) )
|
|
return;
|
|
|
|
var itemsToInsert = selectedItems.Contains(draggedItem)
|
|
? selectedItems.ToList()
|
|
: selectedItems.Append(draggedItem).ToList();
|
|
|
|
foreach ( var item in itemsToInsert ) {
|
|
if ( item != null && item != _asset )
|
|
item.RemoveFromHierarchy();
|
|
}
|
|
|
|
int insertIndex = Mathf.Clamp(baseIndex, 0, parent.childCount);
|
|
foreach ( var item in itemsToInsert ) {
|
|
if ( item != null && item != _asset )
|
|
parent.Insert(insertIndex++, item);
|
|
}
|
|
}
|
|
|
|
Save();
|
|
evt.StopPropagation();
|
|
});
|
|
|
|
_asset.AddManipulator(new ContextualMenuManipulator(evt => {
|
|
evt.menu.InsertAction(0, "Add Tag", _ => {
|
|
var _selectedItems = new HashSet< VisualElement >(selectedItems) { _asset };
|
|
|
|
foreach ( var selectedItem in _selectedItems ) {
|
|
var tagLabels = selectedItem.Query< Label >()
|
|
.Where(e => e.ClassListContains("tagLabel"))
|
|
.ToList();
|
|
|
|
if ( tagLabels.Count > 4 ) continue;
|
|
|
|
var tagField = selectedItem.Q< TextField >();
|
|
|
|
var tagLabel = CreateTagLabel("Tag", tagField, selectedItem);
|
|
selectedItem.Add(tagLabel);
|
|
}
|
|
|
|
Save();
|
|
});
|
|
|
|
evt.menu.InsertSeparator("", 1);
|
|
evt.menu.InsertAction(2, "Remove Asset", _ => {
|
|
_asset.RemoveFromHierarchy();
|
|
Save();
|
|
});
|
|
evt.menu.InsertSeparator("", 3);
|
|
evt.menu.InsertAction(4, "Clear Tag", _ => {
|
|
var _selectedItems = new HashSet< VisualElement >(selectedItems) { _asset };
|
|
|
|
foreach ( var tag in _selectedItems.Select(selectedItem => selectedItem.Query< Label >()
|
|
.Where(e => e.ClassListContains("tagLabel"))
|
|
.ToList()).SelectMany(tagLabels => tagLabels) ) { tag.RemoveFromHierarchy(); }
|
|
|
|
Save();
|
|
});
|
|
}));
|
|
|
|
return _asset;
|
|
}
|
|
|
|
private Label CreateTagLabel(string tag, TextField tagNameField, VisualElement _asset) {
|
|
var tagLabel = new Label() {
|
|
text = "#" + tag.RemoveSpecialCharacters()
|
|
};
|
|
tagLabel.AddToClassList("tagLabel");
|
|
|
|
tagLabel.AddManipulator(new ContextualMenuManipulator(tagEvt => {
|
|
tagEvt.menu.InsertAction(0, "Remove Tag", _ => {
|
|
tagLabel.RemoveFromHierarchy();
|
|
Save();
|
|
});
|
|
}));
|
|
|
|
tagLabel.RegisterCallback< MouseDownEvent >(e => e.StopPropagation());
|
|
tagLabel.RegisterCallback< ClickEvent >(e => e.StopPropagation());
|
|
|
|
tagLabel.RegisterCallback< MouseUpEvent >(tagEvt => {
|
|
if ( tagEvt.button != 0 || tagNameField == null) return;
|
|
|
|
var parent = tagNameField.parent;
|
|
if ( parent != null ) {
|
|
tagNameField.RemoveFromHierarchy();
|
|
parent.Add(tagNameField);
|
|
}
|
|
|
|
var tagLabels = _asset.Query< Label >()
|
|
.Where(e => e.ClassListContains("tagLabel"))
|
|
.ToList();
|
|
|
|
foreach ( var tagLbl in tagLabels )
|
|
tagLbl.Hide();
|
|
|
|
tagNameField.SetValueWithoutNotify(tagLabel.text);
|
|
tagNameField.userData = ( true, tagLabel );
|
|
tagNameField.Show();
|
|
tagNameField.Focus();
|
|
tagNameField.SelectAll();
|
|
|
|
tagEvt.StopPropagation();
|
|
});
|
|
|
|
bool isDragging = false;
|
|
Vector2 dragStart = Vector2.zero;
|
|
|
|
tagLabel.RegisterCallback< MouseDownEvent >(evt => {
|
|
if ( evt.button != 0 ) return;
|
|
dragStart = evt.mousePosition;
|
|
isDragging = true;
|
|
evt.StopPropagation();
|
|
});
|
|
|
|
tagLabel.RegisterCallback< MouseMoveEvent >(evt => {
|
|
if ( !isDragging ) return;
|
|
|
|
if ( ( evt.mousePosition - dragStart ).sqrMagnitude > 16f ) {
|
|
DragAndDrop.PrepareStartDrag();
|
|
DragAndDrop.SetGenericData("MyTagKey", tagLabel);
|
|
DragAndDrop.StartDrag("Dragging Element");
|
|
isDragging = false;
|
|
}
|
|
});
|
|
|
|
tagLabel.RegisterCallback< MouseUpEvent >(evt => {
|
|
if ( evt.button == 0 )
|
|
isDragging = false;
|
|
});
|
|
|
|
return tagLabel;
|
|
}
|
|
|
|
|
|
private float lastClickTime;
|
|
private const float doubleClickThreshold = 0.3f;
|
|
private VisualElement lastClickedItem;
|
|
|
|
private void SetupSelectable(VisualElement item) {
|
|
item.RegisterCallback< ClickEvent >(evt => {
|
|
if ( evt.button != 0 ) return;
|
|
bool shift = evt.shiftKey;
|
|
bool ctrl = evt.ctrlKey || evt.commandKey;
|
|
|
|
var parent = item.parent;
|
|
int index = parent.IndexOf(item);
|
|
|
|
if ( shift && lastSelected != null ) {
|
|
int lastIndex = parent.IndexOf(lastSelected);
|
|
if ( lastIndex < 0 ) {
|
|
ClearSelection();
|
|
Select(item);
|
|
lastSelected = item;
|
|
return;
|
|
}
|
|
|
|
int from = Mathf.Min(index, lastIndex);
|
|
int to = Mathf.Max(index, lastIndex);
|
|
|
|
for ( int i = from; i <= to && i < parent.childCount; i++ )
|
|
Select(parent[ i ]);
|
|
} else if ( ctrl ) {
|
|
if ( selectedItems.Contains(item) )
|
|
Deselect(item);
|
|
else
|
|
Select(item);
|
|
} else {
|
|
ClearSelection();
|
|
Select(item);
|
|
}
|
|
|
|
// Double-CLick
|
|
|
|
float time = Time.realtimeSinceStartup;
|
|
|
|
if ( item == lastClickedItem && time - lastClickTime < doubleClickThreshold ) {
|
|
var asset = item.userData as Sprite;
|
|
Selection.activeObject = asset;
|
|
EditorGUIUtility.PingObject(asset);
|
|
|
|
lastClickedItem = null;
|
|
lastClickTime = 0;
|
|
return;
|
|
}
|
|
|
|
lastClickedItem = item;
|
|
lastClickTime = time;
|
|
|
|
//
|
|
|
|
lastSelected = item;
|
|
|
|
evt.StopPropagation();
|
|
});
|
|
}
|
|
|
|
private void ClearSelection() {
|
|
foreach ( var selected in selectedItems.ToList() )
|
|
Deselect(selected);
|
|
}
|
|
|
|
private void DeleteSelectedItems() {
|
|
foreach ( var item in selectedItems.ToList() ) {
|
|
item.RemoveFromHierarchy();
|
|
selectedItems.Remove(item);
|
|
}
|
|
}
|
|
|
|
private void Select(VisualElement item) {
|
|
if ( selectedItems.Contains(item) ) return;
|
|
selectedItems.Add(item);
|
|
item.AddToClassList("selected");
|
|
}
|
|
|
|
private void Deselect(VisualElement item) {
|
|
selectedItems.Remove(item);
|
|
item.RemoveFromClassList("selected");
|
|
}
|
|
|
|
private void Save() {
|
|
Undo.RecordObject(database, "_intrigues_assetDb");
|
|
|
|
database.spriteDatabase = new SerializableDictionary< string, AssetDb >();
|
|
|
|
foreach ( var category in categories ) {
|
|
var _assetDb = new AssetDb();
|
|
|
|
var _assetField = (VisualElement)category.userData;
|
|
|
|
foreach ( var _asset in _assetField.Children() ) {
|
|
var obj = _asset.userData as Sprite;
|
|
if ( obj == null ) continue;
|
|
|
|
var tags = _asset.Query< Label >()
|
|
.Where(label => label.ClassListContains("tagLabel"))
|
|
.ToList() // <- UQueryBuilder<Label> → List<Label>
|
|
.Select(label => {
|
|
var text = label.text.RemoveSpecialCharacters();
|
|
return string.IsNullOrEmpty(text) ? "Tag" : text;
|
|
})
|
|
.ToList();
|
|
|
|
_assetDb.Sprites.Add(obj, tags);
|
|
}
|
|
|
|
database.spriteDatabase.Add(category.name, _assetDb);
|
|
}
|
|
|
|
UpdateTagList();
|
|
|
|
EditorUtility.SetDirty(database);
|
|
}
|
|
}
|
|
} |