您最多选择25个主题
主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
96 行
3.4 KiB
96 行
3.4 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using UnityEditor;
|
|
using UnityEditor.UIElements;
|
|
using UnityEngine.Experimental.Perception.Randomization.Editor;
|
|
using UnityEngine.Experimental.Perception.Randomization.Scenarios;
|
|
using UnityEngine.UIElements;
|
|
|
|
namespace UnityEngine.Experimental.Perception.Randomization.VisualElements
|
|
{
|
|
class RandomizerList : VisualElement
|
|
{
|
|
SerializedProperty m_Property;
|
|
VisualElement m_Container;
|
|
|
|
public ScenarioBase scenario => (ScenarioBase)m_Property.serializedObject.targetObject;
|
|
|
|
VisualElement inspectorContainer
|
|
{
|
|
get
|
|
{
|
|
var viewport = parent;
|
|
while (!viewport.ClassListContains("unity-inspector-main-container"))
|
|
viewport = viewport.parent;
|
|
return viewport;
|
|
}
|
|
}
|
|
|
|
public RandomizerList(SerializedProperty property)
|
|
{
|
|
m_Property = property;
|
|
AssetDatabase.LoadAssetAtPath<VisualTreeAsset>(
|
|
$"{StaticData.uxmlDir}/Randomizer/RandomizerList.uxml").CloneTree(this);
|
|
|
|
m_Container = this.Q<VisualElement>("randomizers-container");
|
|
|
|
var addRandomizerButton = this.Q<Button>("add-randomizer-button");
|
|
addRandomizerButton.clicked += () =>
|
|
{
|
|
inspectorContainer.Add(new AddRandomizerMenu(inspectorContainer, addRandomizerButton, this));
|
|
};
|
|
|
|
var expandAllButton = this.Q<Button>("expand-all");
|
|
expandAllButton.clicked += () => CollapseRandomizers(false);
|
|
|
|
var collapseAllButton = this.Q<Button>("collapse-all");
|
|
collapseAllButton.clicked += () => CollapseRandomizers(true);
|
|
|
|
RefreshList();
|
|
Undo.undoRedoPerformed += () =>
|
|
{
|
|
m_Property.serializedObject.Update();
|
|
RefreshList();
|
|
};
|
|
}
|
|
|
|
void RefreshList()
|
|
{
|
|
m_Container.Clear();
|
|
for (var i = 0; i < m_Property.arraySize; i++)
|
|
m_Container.Add(new RandomizerElement(m_Property.GetArrayElementAtIndex(i), this));
|
|
}
|
|
|
|
public void AddRandomizer(Type randomizerType)
|
|
{
|
|
Undo.RegisterCompleteObjectUndo(m_Property.serializedObject.targetObject, "Add Randomizer");
|
|
scenario.CreateRandomizer(randomizerType);
|
|
m_Property.serializedObject.Update();
|
|
RefreshList();
|
|
}
|
|
|
|
public void RemoveRandomizer(RandomizerElement element)
|
|
{
|
|
Undo.RegisterCompleteObjectUndo(m_Property.serializedObject.targetObject, "Remove Randomizer");
|
|
scenario.RemoveRandomizer(element.randomizerType);
|
|
m_Property.serializedObject.Update();
|
|
RefreshList();
|
|
}
|
|
|
|
public void ReorderRandomizer(int currentIndex, int nextIndex)
|
|
{
|
|
if (currentIndex == nextIndex)
|
|
return;
|
|
Undo.RegisterCompleteObjectUndo(m_Property.serializedObject.targetObject, "Reorder Randomizer");
|
|
scenario.ReorderRandomizer(currentIndex, nextIndex);
|
|
m_Property.serializedObject.Update();
|
|
RefreshList();
|
|
}
|
|
|
|
void CollapseRandomizers(bool collapsed)
|
|
{
|
|
foreach (var child in m_Container.Children())
|
|
((RandomizerElement)child).collapsed = collapsed;
|
|
}
|
|
}
|
|
}
|