ItemsUtils.cs 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using UnityEngine.Timeline;
  5. namespace UnityEditor.Timeline
  6. {
  7. static class ItemsUtils
  8. {
  9. static readonly Dictionary<TimelineClip, ClipItem> s_ClipCache = new Dictionary<TimelineClip, ClipItem>();
  10. static readonly Dictionary<IMarker, MarkerItem> s_MarkerCache = new Dictionary<IMarker, MarkerItem>();
  11. public static IEnumerable<ItemsPerTrack> ToItemsPerTrack(this IEnumerable<ITimelineItem> items)
  12. {
  13. var groupedItems = items.GroupBy(c => c.parentTrack);
  14. foreach (var group in groupedItems)
  15. {
  16. yield return new ItemsPerTrack(group.Key, group.ToArray());
  17. }
  18. }
  19. public static ITimelineItem ToItem(this TimelineClip clip)
  20. {
  21. if (s_ClipCache.ContainsKey(clip))
  22. return s_ClipCache[clip];
  23. var ret = new ClipItem(clip);
  24. s_ClipCache.Add(clip, ret);
  25. return ret;
  26. }
  27. public static ITimelineItem ToItem(this IMarker marker)
  28. {
  29. if (s_MarkerCache.ContainsKey(marker))
  30. return s_MarkerCache[marker];
  31. var ret = new MarkerItem(marker);
  32. s_MarkerCache.Add(marker, ret);
  33. return ret;
  34. }
  35. public static IEnumerable<ITimelineItem> ToItems(this IEnumerable<TimelineClip> clips)
  36. {
  37. return clips.Select(ToItem);
  38. }
  39. public static IEnumerable<ITimelineItem> ToItems(this IEnumerable<IMarker> markers)
  40. {
  41. return markers.Select(ToItem);
  42. }
  43. public static IEnumerable<ITimelineItem> GetItems(this TrackAsset track)
  44. {
  45. var list = track.clips.Select(clip => (ITimelineItem) new ClipItem(clip)).ToList();
  46. list.AddRange(track.GetMarkers().Select(marker => (ITimelineItem) new MarkerItem(marker)));
  47. list = list.OrderBy(x => x.start).ThenBy(x => x.end).ToList();
  48. return list;
  49. }
  50. public static void GetItemRange(this TrackAsset track, out double start, out double end)
  51. {
  52. start = 0;
  53. end = 0;
  54. var items = track.GetItems().ToList();
  55. if (items.Any())
  56. {
  57. start = items.Min(p => p.start);
  58. end = items.Max(p => p.end);
  59. }
  60. }
  61. public static IEnumerable<ITimelineItem> GetItemsExcept(this TrackAsset track, IEnumerable<ITimelineItem> items)
  62. {
  63. return GetItems(track).Except(items);
  64. }
  65. public static IEnumerable<Type> GetItemTypes(IEnumerable<ITimelineItem> items)
  66. {
  67. var types = new List<Type>();
  68. if (items.OfType<ClipItem>().Any())
  69. types.Add(typeof(ClipItem));
  70. if (items.OfType<MarkerItem>().Any())
  71. types.Add(typeof(MarkerItem));
  72. return types;
  73. }
  74. public static IEnumerable<Type> GetItemTypes(IEnumerable<ItemsPerTrack> itemsGroups)
  75. {
  76. return GetItemTypes(itemsGroups.SelectMany(i => i.items)).Distinct();
  77. }
  78. public static void SetItemsStartTime(IEnumerable<ItemsPerTrack> newItems, double time)
  79. {
  80. var startTimes = newItems.Select(d => d.items.Min(x => x.start)).ToList();
  81. var min = startTimes.Min();
  82. startTimes = startTimes.Select(x => x - min + time).ToList();
  83. for (int i = 0; i < newItems.Count(); ++i)
  84. EditModeUtils.SetStart(newItems.ElementAt(i).items, startTimes[i]);
  85. }
  86. public static double TimeGapBetweenItems(ITimelineItem leftItem, ITimelineItem rightItem, WindowState state)
  87. {
  88. if (leftItem is MarkerItem && rightItem is MarkerItem)
  89. {
  90. var markerType = ((MarkerItem)leftItem).marker.GetType();
  91. var gap = state.PixelDeltaToDeltaTime(StyleManager.UssStyleForType(markerType).fixedWidth);
  92. return gap;
  93. }
  94. return 0.0;
  95. }
  96. }
  97. }