GlobalGameJame/Library/PackageCache/com.unity.timeline@1.2.10/Editor/Animation/BindingTreeViewDataSource.cs

140 lines
4.5 KiB
C#
Raw Normal View History

2020-01-31 15:10:04 +00:00
using System.Collections.Generic;
using System.Linq;
using UnityEditor;
using UnityEditor.IMGUI.Controls;
using UnityEditor.Timeline;
using UnityEngine;
namespace UnityEditorInternal
{
class BindingTreeViewDataSource : TreeViewDataSource
{
public const int RootID = int.MinValue;
public const int GroupID = -1;
AnimationClip m_Clip;
CurveDataSource m_CurveDataSource;
public BindingTreeViewDataSource(
TreeViewController treeView, AnimationClip clip, CurveDataSource curveDataSource)
: base(treeView)
{
m_Clip = clip;
showRootItem = false;
m_CurveDataSource = curveDataSource;
}
void SetupRootNodeSettings()
{
showRootItem = false;
SetExpanded(RootID, true);
SetExpanded(GroupID, true);
}
static string GroupName(EditorCurveBinding binding)
{
string property = AnimationWindowUtility.NicifyPropertyGroupName(binding.type, binding.propertyName);
if (!string.IsNullOrEmpty(binding.path))
{
property = binding.path + " : " + property;
}
int lastArrayIdx = property.LastIndexOf("Array.");
if (lastArrayIdx != -1)
{
property = property.Substring(0, lastArrayIdx - 1);
}
return property;
}
static string PropertyName(EditorCurveBinding binding, string arrayPrefixToRemove = "")
{
string propertyName = AnimationWindowUtility.GetPropertyDisplayName(binding.propertyName);
if (propertyName.Contains("Array"))
{
propertyName = propertyName.Replace("Array.", "");
propertyName = propertyName.Replace(arrayPrefixToRemove, "");
propertyName = propertyName.TrimStart('.');
}
return propertyName;
}
public override void FetchData()
{
if (m_Clip == null)
return;
var bindings = AnimationUtility.GetCurveBindings(m_Clip)
.Union(AnimationUtility.GetObjectReferenceCurveBindings(m_Clip))
.ToArray();
var results = bindings.GroupBy(p => GroupName(p), p => p, (key, g) => new
{
parent = key,
bindings = g.ToList()
}).OrderBy(t =>
{
//Force transform order first
if (t.parent == "Position") return -3;
if (t.parent == "Rotation") return -2;
if (t.parent == "Scale") return -1;
return 0;
}).ThenBy(t => t.parent);
m_RootItem = new CurveTreeViewNode(RootID, null, "root", null)
{
children = new List<TreeViewItem>(1)
};
var groupingNode = new CurveTreeViewNode(GroupID, m_RootItem, m_CurveDataSource.groupingName, bindings)
{
children = new List<TreeViewItem>()
};
m_RootItem.children.Add(groupingNode);
foreach (var r in results)
{
var newNode = new CurveTreeViewNode(r.parent.GetHashCode(), groupingNode, r.parent, r.bindings.ToArray());
groupingNode.children.Add(newNode);
if (r.bindings.Count > 1)
{
for (int b = 0; b < r.bindings.Count; b++)
{
if (newNode.children == null)
newNode.children = new List<TreeViewItem>();
var binding = r.bindings[b];
var bindingNode = new CurveTreeViewNode(binding.GetHashCode(), newNode, PropertyName(binding, newNode.displayName), new[] {binding});
newNode.children.Add(bindingNode);
}
}
}
SetupRootNodeSettings();
m_NeedRefreshRows = true;
}
public void UpdateData()
{
m_TreeView.ReloadData();
}
}
class CurveTreeViewNode : TreeViewItem
{
EditorCurveBinding[] m_Bindings;
public EditorCurveBinding[] bindings
{
get { return m_Bindings; }
}
public CurveTreeViewNode(int id, TreeViewItem parent, string displayName, EditorCurveBinding[] bindings)
: base(id, parent != null ? parent.depth + 1 : -1, parent, displayName)
{
m_Bindings = bindings;
}
}
}