playtest-unity/playtest/Library/PackageCache/com.unity.timeline@1.7.4/Editor/treeview/TimelineTreeViewGUI.cs

181 lines
5.6 KiB
C#

using System.Collections.Generic;
using System.Linq;
using UnityEditor.IMGUI.Controls;
using UnityEngine;
using UnityEngine.Timeline;
namespace UnityEditor.Timeline
{
class TimelineTreeViewGUI
{
readonly TimelineAsset m_Timeline;
readonly TreeViewController m_TreeView;
readonly TimelineTreeView m_TimelineTreeView;
readonly TimelineWindow m_Window;
readonly TimelineDataSource m_DataSource;
TreeViewItem root
{
get { return m_DataSource.root; }
}
TimelineTrackBaseGUI[] visibleTrackGuis
{
get
{
int firstRow;
int lastRow;
var visibleRows = new List<TimelineTrackBaseGUI>();
m_TreeView.gui.GetFirstAndLastRowVisible(out firstRow, out lastRow);
for (int r = firstRow; r <= lastRow; r++)
{
var track = m_TreeView.data.GetItem(r) as TimelineTrackBaseGUI;
if (track != null && track != root)
{
AddVisibleTrackRecursive(ref visibleRows, track);
}
}
return visibleRows.ToArray();
}
}
public TrackAsset[] visibleTracks
{
get { return visibleTrackGuis.Select(x => x.track).ToArray(); }
}
public List<TimelineClipGUI> allClipGuis
{
get
{
TimelineDataSource dataSource = m_TreeView.data as TimelineDataSource;
if (dataSource != null && dataSource.allTrackGuis != null)
return dataSource.allTrackGuis.OfType<TimelineTrackGUI>().SelectMany(x => x.clips).ToList();
return null;
}
}
public List<TimelineTrackBaseGUI> allTrackGuis
{
get
{
var dataSource = m_TreeView.data as TimelineDataSource;
if (dataSource != null)
return dataSource.allTrackGuis;
return null;
}
}
public Vector2 contentSize
{
get { return m_TreeView.GetContentSize(); }
}
public Vector2 scrollPosition
{
get { return m_TreeView.state.scrollPos; }
set
{
Rect r = m_TreeView.GetTotalRect();
Vector2 visibleContent = m_TreeView.GetContentSize();
m_TreeView.state.scrollPos = new Vector2(value.x, Mathf.Clamp(value.y, 0, Mathf.Max(0, visibleContent.y - r.height)));
}
}
public bool showingVerticalScrollBar
{
get { return m_TreeView.showingVerticalScrollBar; }
}
public void FrameItem(TreeViewItem item)
{
m_TreeView.Frame(item.id, true, false, true);
}
public TimelineDragging timelineDragging { get { return m_TreeView.dragging as TimelineDragging; } }
public TimelineTreeViewGUI(TimelineWindow sequencerWindow, TimelineAsset timeline, Rect rect)
{
m_Timeline = timeline;
m_Window = sequencerWindow;
var treeviewState = new TreeViewState();
treeviewState.scrollPos = new Vector2(treeviewState.scrollPos.x, TimelineWindowViewPrefs.GetOrCreateViewModel(m_Timeline).verticalScroll);
m_TreeView = new TreeViewController(sequencerWindow, treeviewState);
m_TreeView.horizontalScrollbarStyle = GUIStyle.none;
m_TreeView.scrollViewStyle = GUI.skin.scrollView;
m_TimelineTreeView = new TimelineTreeView(sequencerWindow, m_TreeView);
var dragging = new TimelineDragging(m_TreeView, m_Window, m_Timeline);
m_DataSource = new TimelineDataSource(this, m_TreeView, sequencerWindow);
m_DataSource.onVisibleRowsChanged += m_TimelineTreeView.CalculateRowRects;
m_TreeView.Init(rect, m_DataSource, m_TimelineTreeView, dragging);
m_DataSource.ExpandItems(m_DataSource.root);
}
public ITreeViewGUI gui
{
get { return m_TimelineTreeView; }
}
public ITreeViewDataSource data
{
get { return m_TreeView == null ? null : m_TreeView.data; }
}
public TimelineWindow TimelineWindow
{
get { return m_Window; }
}
public void CalculateRowRects()
{
m_TimelineTreeView.CalculateRowRects();
}
public void Reload()
{
AnimationClipCurveCache.Instance.ClearCachedProxyClips();
m_TreeView.ReloadData();
m_DataSource.ExpandItems(m_DataSource.root);
m_TimelineTreeView.CalculateRowRects();
}
public void OnGUI(Rect rect)
{
int keyboardControl = GUIUtility.GetControlID(FocusType.Passive, rect);
m_TreeView.OnGUI(rect, keyboardControl);
TimelineWindowViewPrefs.GetOrCreateViewModel(m_Timeline).verticalScroll = m_TreeView.state.scrollPos.y;
}
public Rect GetRowRect(int row)
{
return m_TimelineTreeView.GetRowRect(row);
}
static void AddVisibleTrackRecursive(ref List<TimelineTrackBaseGUI> list, TimelineTrackBaseGUI track)
{
if (track == null)
return;
list.Add(track);
if (!track.isExpanded)
return;
if (track.children != null)
{
foreach (var c in track.children)
{
AddVisibleTrackRecursive(ref list, c as TimelineTrackBaseGUI);
}
}
}
}
}