InlineCurveEditor.cs 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314
  1. using System.Linq;
  2. using JetBrains.Annotations;
  3. using UnityEngine;
  4. using UnityEngine.Timeline;
  5. namespace UnityEditor.Timeline
  6. {
  7. interface IClipCurveEditorOwner
  8. {
  9. ClipCurveEditor clipCurveEditor { get; }
  10. bool inlineCurvesSelected { get; set; }
  11. bool showLoops { get; }
  12. TrackAsset owner { get; }
  13. }
  14. class InlineCurveResizeHandle : IBounds
  15. {
  16. public Rect boundingRect { get; private set; }
  17. public TimelineTrackGUI trackGUI { get; }
  18. public InlineCurveResizeHandle(TimelineTrackGUI trackGUI)
  19. {
  20. this.trackGUI = trackGUI;
  21. }
  22. public void Draw(Rect headerRect, Rect trackRect, WindowState state)
  23. {
  24. var rect = new Rect(headerRect.xMax + 4, headerRect.yMax - 5.0f, trackRect.width - 4, 5.0f);
  25. var handleColor = Handles.color;
  26. Handles.color = Color.black;
  27. Handles.DrawAAPolyLine(1.0f,
  28. new Vector3(rect.x, rect.yMax, 0.0f),
  29. new Vector3(rect.xMax, rect.yMax, 0.0f));
  30. Handles.color = handleColor;
  31. EditorGUIUtility.AddCursorRect(rect, MouseCursor.SplitResizeUpDown);
  32. boundingRect = trackGUI.ToWindowSpace(rect);
  33. if (Event.current.type == EventType.Repaint)
  34. {
  35. state.spacePartitioner.AddBounds(this);
  36. var dragStyle = new GUIStyle("RL DragHandle");
  37. dragStyle.Draw(rect, GUIContent.none, false, false, false, false);
  38. }
  39. }
  40. }
  41. class InlineCurveEditor : IBounds
  42. {
  43. Rect m_TrackRect;
  44. Rect m_HeaderRect;
  45. readonly TimelineTrackGUI m_TrackGUI;
  46. readonly InlineCurveResizeHandle m_ResizeHandle;
  47. bool m_LastSelectionWasClip;
  48. TimelineClipGUI m_LastSelectedClipGUI;
  49. Rect IBounds.boundingRect { get { return m_TrackGUI.ToWindowSpace(m_TrackRect); } }
  50. [UsedImplicitly] // Used in tests
  51. public TimelineClipGUI currentClipGui
  52. {
  53. get { return m_LastSelectedClipGUI; }
  54. }
  55. public IClipCurveEditorOwner currentCurveEditor
  56. {
  57. get { return m_LastSelectionWasClip ? (IClipCurveEditorOwner)m_LastSelectedClipGUI : (IClipCurveEditorOwner)m_TrackGUI; }
  58. }
  59. public InlineCurveEditor(TimelineTrackGUI trackGUI)
  60. {
  61. m_TrackGUI = trackGUI;
  62. m_ResizeHandle = new InlineCurveResizeHandle(trackGUI);
  63. }
  64. static bool MouseOverTrackArea(Rect curveRect, Rect trackRect)
  65. {
  66. curveRect.y = trackRect.y;
  67. curveRect.height = trackRect.height;
  68. // clamp the curve editor to the track. this allows the menu to scroll properly
  69. curveRect.xMin = Mathf.Max(curveRect.xMin, trackRect.xMin);
  70. curveRect.xMax = trackRect.xMax;
  71. return curveRect.Contains(Event.current.mousePosition);
  72. }
  73. static bool MouseOverHeaderArea(Rect headerRect, Rect trackRect)
  74. {
  75. headerRect.y = trackRect.y;
  76. headerRect.height = trackRect.height;
  77. return headerRect.Contains(Event.current.mousePosition);
  78. }
  79. static void DrawCurveEditor(IClipCurveEditorOwner clipCurveEditorOwner, WindowState state, Rect headerRect, Rect trackRect, Vector2 activeRange, bool locked)
  80. {
  81. ClipCurveEditor clipCurveEditor = clipCurveEditorOwner.clipCurveEditor;
  82. CurveDataSource dataSource = clipCurveEditor.dataSource;
  83. Rect curveRect = dataSource.GetBackgroundRect(state);
  84. bool newlySelected = false;
  85. if (Event.current.type == EventType.MouseDown || Event.current.type == EventType.ContextClick)
  86. newlySelected = MouseOverTrackArea(curveRect, trackRect) || MouseOverHeaderArea(headerRect, trackRect);
  87. // make sure to not use any event before drawing the curve.
  88. bool prevEnabledState = GUI.enabled;
  89. GUI.enabled = true;
  90. clipCurveEditorOwner.clipCurveEditor.DrawHeader(headerRect);
  91. GUI.enabled = prevEnabledState;
  92. bool displayAsSelected = !locked && (clipCurveEditorOwner.inlineCurvesSelected || newlySelected);
  93. using (new EditorGUI.DisabledScope(locked))
  94. {
  95. using (new GUIViewportScope(trackRect))
  96. {
  97. Rect animEditorRect = curveRect;
  98. animEditorRect.y = trackRect.y;
  99. animEditorRect.height = trackRect.height;
  100. // clamp the curve editor to the track. this allows the menu to scroll properly
  101. animEditorRect.xMin = Mathf.Max(animEditorRect.xMin, trackRect.xMin);
  102. animEditorRect.xMax = trackRect.xMax;
  103. if (activeRange == Vector2.zero)
  104. activeRange = new Vector2(animEditorRect.xMin, animEditorRect.xMax);
  105. clipCurveEditor.DrawCurveEditor(animEditorRect, state, activeRange, clipCurveEditorOwner.showLoops, displayAsSelected);
  106. }
  107. }
  108. if (newlySelected && !locked)
  109. {
  110. clipCurveEditorOwner.inlineCurvesSelected = true;
  111. HandleCurrentEvent();
  112. }
  113. }
  114. public void Draw(Rect headerRect, Rect trackRect, WindowState state)
  115. {
  116. m_TrackRect = trackRect;
  117. m_TrackRect.height -= 5.0f;
  118. if (Event.current.type == EventType.Repaint)
  119. state.spacePartitioner.AddBounds(this);
  120. // Remove the indentation of this track to render it properly, otherwise every GUI elements will be offsetted.
  121. headerRect.x -= DirectorStyles.kBaseIndent;
  122. headerRect.width += DirectorStyles.kBaseIndent;
  123. // Remove the width of the color swatch.
  124. headerRect.x += 4.0f;
  125. headerRect.width -= 4.0f;
  126. m_HeaderRect = headerRect;
  127. EditorGUI.DrawRect(m_HeaderRect, DirectorStyles.Instance.customSkin.colorAnimEditorBinding);
  128. if (ShouldShowClipCurves(state))
  129. {
  130. DrawCurveEditorsForClipsOnTrack(m_HeaderRect, m_TrackRect, state);
  131. }
  132. else if (ShouldShowTrackCurves())
  133. {
  134. DrawCurveEditorForTrack(m_HeaderRect, m_TrackRect, state);
  135. }
  136. else
  137. {
  138. DrawCurvesEditorForNothingSelected(m_HeaderRect, m_TrackRect, state);
  139. }
  140. m_ResizeHandle.Draw(headerRect, trackRect, state);
  141. // If MouseDown or ContextClick are not consumed by the curves, use the event to prevent it from going deeper into the treeview.
  142. if (Event.current.type == EventType.ContextClick)
  143. {
  144. var r = Rect.MinMaxRect(m_HeaderRect.xMin, m_HeaderRect.yMin, m_TrackRect.xMax, m_TrackRect.yMax);
  145. if (r.Contains(Event.current.mousePosition))
  146. Event.current.Use();
  147. }
  148. UpdateViewModel();
  149. }
  150. public void Refresh()
  151. {
  152. if (m_LastSelectionWasClip)
  153. RefreshInlineCurves(m_LastSelectedClipGUI);
  154. else
  155. RefreshInlineCurves(m_TrackGUI);
  156. }
  157. static void RefreshInlineCurves(IClipCurveEditorOwner guiItem)
  158. {
  159. if (guiItem.clipCurveEditor != null && guiItem.clipCurveEditor.dataSource != null)
  160. guiItem.clipCurveEditor.dataSource.RebuildCurves();
  161. }
  162. void DrawCurveEditorForTrack(Rect headerRect, Rect trackRect, WindowState state)
  163. {
  164. if (m_TrackGUI.clipCurveEditor == null)
  165. return;
  166. DrawCurveEditor(m_TrackGUI, state, headerRect, trackRect, Vector2.zero, m_TrackGUI.locked);
  167. m_LastSelectionWasClip = false;
  168. }
  169. void DrawCurveEditorsForClipsOnTrack(Rect headerRect, Rect trackRect, WindowState state)
  170. {
  171. if (m_TrackGUI.clips.Count == 0)
  172. return;
  173. if (Event.current.type == EventType.Layout)
  174. {
  175. TimelineClipGUI selectedClip = SelectionManager.SelectedClipGUI().FirstOrDefault(x => x.parent == m_TrackGUI);
  176. if (selectedClip != null)
  177. {
  178. m_LastSelectedClipGUI = selectedClip;
  179. }
  180. else if (state.recording && state.IsArmedForRecord(m_TrackGUI.track))
  181. {
  182. if (m_LastSelectedClipGUI == null || !m_TrackGUI.track.IsRecordingToClip(m_LastSelectedClipGUI.clip))
  183. {
  184. var clip = m_TrackGUI.clips.FirstOrDefault(x => m_TrackGUI.track.IsRecordingToClip(x.clip));
  185. if (clip != null)
  186. m_LastSelectedClipGUI = clip;
  187. }
  188. }
  189. if (m_LastSelectedClipGUI == null)
  190. m_LastSelectedClipGUI = m_TrackGUI.clips[0];
  191. }
  192. if (m_LastSelectedClipGUI == null || m_LastSelectedClipGUI.clipCurveEditor == null || m_LastSelectedClipGUI.isInvalid)
  193. return;
  194. var inlineCurveActiveArea = new Vector2(state.TimeToPixel(m_LastSelectedClipGUI.clip.start), state.TimeToPixel(m_LastSelectedClipGUI.clip.end));
  195. DrawCurveEditor(m_LastSelectedClipGUI, state, headerRect, trackRect, inlineCurveActiveArea, m_TrackGUI.locked);
  196. m_LastSelectionWasClip = true;
  197. }
  198. void DrawCurvesEditorForNothingSelected(Rect headerRect, Rect trackRect, WindowState state)
  199. {
  200. if (m_LastSelectionWasClip || !TrackHasCurvesToShow() && m_TrackGUI.clips.Count > 0)
  201. {
  202. DrawCurveEditorsForClipsOnTrack(headerRect, trackRect, state);
  203. }
  204. else
  205. {
  206. DrawCurveEditorForTrack(headerRect, trackRect, state);
  207. }
  208. }
  209. bool ShouldShowClipCurves(WindowState state)
  210. {
  211. if (m_TrackGUI.clips.Count == 0)
  212. return false;
  213. // Is a clip selected or being recorded to?
  214. return SelectionManager.SelectedClipGUI().FirstOrDefault(x => x.parent == m_TrackGUI) != null ||
  215. state.recording && state.IsArmedForRecord(m_TrackGUI.track) && m_TrackGUI.clips.FirstOrDefault(x => m_TrackGUI.track.IsRecordingToClip(x.clip)) != null;
  216. }
  217. bool ShouldShowTrackCurves()
  218. {
  219. if (m_TrackGUI == null)
  220. return false;
  221. var isTrackSelected = SelectionManager.SelectedTrackGUI().FirstOrDefault(x => x == m_TrackGUI) != null;
  222. if (!isTrackSelected)
  223. return false;
  224. return TrackHasCurvesToShow();
  225. }
  226. bool TrackHasCurvesToShow()
  227. {
  228. var animTrack = m_TrackGUI.track as AnimationTrack;
  229. if (animTrack != null && !animTrack.inClipMode)
  230. return true;
  231. return m_TrackGUI.track.HasAnyAnimatableParameters();
  232. }
  233. void UpdateViewModel()
  234. {
  235. var curveEditor = currentCurveEditor.clipCurveEditor;
  236. if (curveEditor == null || curveEditor.bindingHierarchy.treeViewController == null)
  237. return;
  238. var vm = TimelineWindowViewPrefs.GetTrackViewModelData(m_TrackGUI.track);
  239. vm.inlineCurvesState = curveEditor.bindingHierarchy.treeViewController.state;
  240. vm.inlineCurvesShownAreaInsideMargins = curveEditor.shownAreaInsideMargins;
  241. vm.lastInlineCurveDataID = curveEditor.dataSource.id;
  242. }
  243. static void HandleCurrentEvent()
  244. {
  245. #if UNITY_EDITOR_OSX
  246. Event.current.type = EventType.Ignore;
  247. #else
  248. Event.current.Use();
  249. #endif
  250. }
  251. }
  252. }