123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521 |
- using System.Collections.Generic;
- using System.ComponentModel;
- using System.Linq;
- using JetBrains.Annotations;
- using UnityEngine;
- using UnityEngine.Timeline;
- namespace UnityEditor.Timeline
- {
- [ActiveInMode(TimelineModes.Default)]
- abstract class TrackAction : MenuItemActionBase
- {
- public abstract bool Execute(WindowState state, TrackAsset[] tracks);
- protected virtual MenuActionDisplayState GetDisplayState(WindowState state, TrackAsset[] tracks)
- {
- return tracks.Length > 0 ? MenuActionDisplayState.Visible : MenuActionDisplayState.Disabled;
- }
- protected virtual bool IsChecked(WindowState state, TrackAsset[] tracks)
- {
- return false;
- }
- protected virtual string GetDisplayName(TrackAsset[] tracks)
- {
- return menuName;
- }
- public static void Invoke<T>(WindowState state, TrackAsset[] tracks) where T : TrackAction
- {
- actions.First(x => x.GetType() == typeof(T)).Execute(state, tracks);
- }
- static List<TrackAction> s_ActionClasses;
- static List<TrackAction> actions
- {
- get
- {
- if (s_ActionClasses == null)
- s_ActionClasses =
- GetActionsOfType(typeof(TrackAction))
- .Select(x => (TrackAction)x.GetConstructors()[0].Invoke(null))
- .OrderBy(x => x.priority).ThenBy(x => x.category)
- .ToList();
- return s_ActionClasses;
- }
- }
- public static void GetMenuEntries(WindowState state, Vector2? mousePos, TrackAsset[] tracks, List<MenuActionItem> items)
- {
- var mode = TimelineWindow.instance.currentMode.mode;
- foreach (var action in actions)
- {
- if (!action.showInMenu)
- continue;
- var actionItem = action;
- items.Add(
- new MenuActionItem()
- {
- category = action.category,
- entryName = action.GetDisplayName(tracks),
- shortCut = action.shortCut,
- isChecked = action.IsChecked(state, tracks),
- isActiveInMode = IsActionActiveInMode(action, mode),
- priority = action.priority,
- state = action.GetDisplayState(state, tracks),
- callback = () =>
- {
- actionItem.mousePosition = mousePos;
- actionItem.Execute(state, tracks);
- actionItem.mousePosition = null;
- }
- }
- );
- }
- }
- public static bool HandleShortcut(WindowState state, Event evt, TrackAsset[] tracks)
- {
- foreach (var action in actions)
- {
- var attr = action.GetType().GetCustomAttributes(typeof(ShortcutAttribute), true);
- foreach (ShortcutAttribute shortcut in attr)
- {
- if (shortcut.MatchesEvent(evt))
- {
- if (s_ShowActionTriggeredByShortcut)
- Debug.Log(action.GetType().Name);
- if (!IsActionActiveInMode(action, TimelineWindow.instance.currentMode.mode))
- return false;
- return action.Execute(state, tracks);
- }
- }
- }
- return false;
- }
- // For testing
- internal MenuActionDisplayState InternalGetDisplayState(WindowState state, TrackAsset[] tracks)
- {
- return GetDisplayState(state, tracks);
- }
- }
- [MenuEntry("Edit in Animation Window", MenuOrder.TrackAction.EditInAnimationWindow)]
- class EditTrackInAnimationWindow : TrackAction
- {
- public static bool Do(WindowState state, TrackAsset track)
- {
- AnimationClip clipToEdit = null;
- AnimationTrack animationTrack = track as AnimationTrack;
- if (animationTrack != null)
- {
- if (!animationTrack.CanConvertToClipMode())
- return false;
- clipToEdit = animationTrack.infiniteClip;
- }
- else if (track.hasCurves)
- {
- clipToEdit = track.curves;
- }
- if (clipToEdit == null)
- return false;
- var gameObject = state.GetSceneReference(track);
- var timeController = TimelineAnimationUtilities.CreateTimeController(state, CreateTimeControlClipData(track));
- TimelineAnimationUtilities.EditAnimationClipWithTimeController(clipToEdit, timeController, gameObject);
- return true;
- }
- protected override MenuActionDisplayState GetDisplayState(WindowState state, TrackAsset[] tracks)
- {
- if (tracks.Length == 0)
- return MenuActionDisplayState.Hidden;
- if (tracks[0] is AnimationTrack)
- {
- var animTrack = tracks[0] as AnimationTrack;
- if (animTrack.CanConvertToClipMode())
- return MenuActionDisplayState.Visible;
- }
- else if (tracks[0].hasCurves)
- {
- return MenuActionDisplayState.Visible;
- }
- return MenuActionDisplayState.Hidden;
- }
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- return Do(state, tracks[0]);
- }
- static TimelineWindowTimeControl.ClipData CreateTimeControlClipData(TrackAsset track)
- {
- var data = new TimelineWindowTimeControl.ClipData();
- data.track = track;
- data.start = track.start;
- data.duration = track.duration;
- return data;
- }
- }
- [MenuEntry("Lock selected track only", MenuOrder.TrackAction.LockSelected)]
- class LockSelectedTrack : TrackAction
- {
- public static readonly string LockSelectedTrackOnlyText = L10n.Tr("Lock selected track only");
- public static readonly string UnlockSelectedTrackOnlyText = L10n.Tr("Unlock selected track only");
- protected override MenuActionDisplayState GetDisplayState(WindowState state, TrackAsset[] tracks)
- {
- if (tracks.Any(track => TimelineUtility.IsLockedFromGroup(track) || track is GroupTrack ||
- !track.subTracksObjects.Any()))
- return MenuActionDisplayState.Hidden;
- return MenuActionDisplayState.Visible;
- }
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- if (!tracks.Any()) return false;
- var hasUnlockedTracks = tracks.Any(x => !x.locked);
- Lock(state, tracks.Where(p => !(p is GroupTrack)).ToArray(), hasUnlockedTracks);
- return true;
- }
- protected override string GetDisplayName(TrackAsset[] tracks)
- {
- return tracks.All(t => t.locked) ? UnlockSelectedTrackOnlyText : LockSelectedTrackOnlyText;
- }
- public static void Lock(WindowState state, TrackAsset[] tracks, bool shouldlock)
- {
- if (tracks.Length == 0)
- return;
- foreach (var track in tracks.Where(t => !TimelineUtility.IsLockedFromGroup(t)))
- {
- TimelineUndo.PushUndo(track, "Lock Tracks");
- track.locked = shouldlock;
- }
- TimelineEditor.Refresh(RefreshReason.WindowNeedsRedraw);
- }
- }
- [MenuEntry("Lock", MenuOrder.TrackAction.LockTrack)]
- [Shortcut(Shortcuts.Timeline.toggleLock)]
- class LockTrack : TrackAction
- {
- public static readonly string UnlockText = L10n.Tr("Unlock");
- protected override MenuActionDisplayState GetDisplayState(WindowState state, TrackAsset[] tracks)
- {
- bool hasUnlockableTracks = tracks.Any(x => TimelineUtility.IsLockedFromGroup(x));
- if (hasUnlockableTracks)
- return MenuActionDisplayState.Disabled;
- return MenuActionDisplayState.Visible;
- }
- protected override string GetDisplayName(TrackAsset[] tracks)
- {
- return tracks.Any(x => !x.locked) ? base.GetDisplayName(tracks) : UnlockText;
- }
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- if (!tracks.Any()) return false;
- var hasUnlockedTracks = tracks.Any(x => !x.locked);
- SetLockState(tracks, hasUnlockedTracks, state);
- return true;
- }
- public static void SetLockState(TrackAsset[] tracks, bool shouldLock, WindowState state = null)
- {
- if (tracks.Length == 0)
- return;
- foreach (var track in tracks)
- {
- if (TimelineUtility.IsLockedFromGroup(track))
- continue;
- if (track as GroupTrack == null)
- SetLockState(track.GetChildTracks().ToArray(), shouldLock, state);
- TimelineUndo.PushUndo(track, "Lock Tracks");
- track.locked = shouldLock;
- }
- if (state != null)
- {
- // find the tracks we've locked. unselect anything locked and remove recording.
- foreach (var track in tracks)
- {
- if (TimelineUtility.IsLockedFromGroup(track) || !track.locked)
- continue;
- var flattenedChildTracks = track.GetFlattenedChildTracks();
- foreach (var i in track.clips)
- SelectionManager.Remove(i);
- state.UnarmForRecord(track);
- foreach (var child in flattenedChildTracks)
- {
- SelectionManager.Remove(child);
- state.UnarmForRecord(child);
- foreach (var clip in child.GetClips())
- SelectionManager.Remove(clip);
- }
- }
- // no need to rebuild, just repaint (including inspectors)
- InspectorWindow.RepaintAllInspectors();
- state.editorWindow.Repaint();
- }
- }
- }
- [UsedImplicitly]
- [MenuEntry("Show Markers", MenuOrder.TrackAction.ShowHideMarkers)]
- [ActiveInMode(TimelineModes.Default | TimelineModes.ReadOnly)]
- class ShowHideMarkers : TrackAction
- {
- protected override bool IsChecked(WindowState state, TrackAsset[] tracks)
- {
- return tracks.All(x => x.GetShowMarkers());
- }
- protected override MenuActionDisplayState GetDisplayState(WindowState state, TrackAsset[] tracks)
- {
- if (tracks.Any(x => x is GroupTrack) || tracks.Any(t => t.GetMarkerCount() == 0))
- return MenuActionDisplayState.Hidden;
- if (tracks.Any(t => t.lockedInHierarchy))
- return MenuActionDisplayState.Disabled;
- return MenuActionDisplayState.Visible;
- }
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- if (!tracks.Any()) return false;
- var hasUnlockedTracks = tracks.Any(x => !x.GetShowMarkers());
- ShowHide(state, tracks, hasUnlockedTracks);
- return true;
- }
- static void ShowHide(WindowState state, TrackAsset[] tracks, bool shouldLock)
- {
- if (tracks.Length == 0)
- return;
- var window = state.GetWindow();
- foreach (var track in tracks)
- {
- window.SetShowTrackMarkers(track, shouldLock);
- }
- TimelineEditor.Refresh(RefreshReason.WindowNeedsRedraw);
- }
- }
- [MenuEntry("Mute selected track only", MenuOrder.TrackAction.MuteSelected), UsedImplicitly]
- class MuteSelectedTrack : TrackAction
- {
- public static readonly string UnmuteSelectedText = L10n.Tr("Unmute selected track only");
- protected override MenuActionDisplayState GetDisplayState(WindowState state, TrackAsset[] tracks)
- {
- if (tracks.Any(track => TimelineUtility.IsParentMuted(track) || track is GroupTrack ||
- !track.subTracksObjects.Any()))
- return MenuActionDisplayState.Hidden;
- return MenuActionDisplayState.Visible;
- }
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- if (!tracks.Any())
- return false;
- var hasUnmutedTracks = tracks.Any(x => !x.muted);
- Mute(state, tracks.Where(p => !(p is GroupTrack)).ToArray(), hasUnmutedTracks);
- return true;
- }
- protected override string GetDisplayName(TrackAsset[] tracks)
- {
- return tracks.All(t => t.muted) ? UnmuteSelectedText : base.GetDisplayName(tracks);
- }
- public static void Mute(WindowState state, TrackAsset[] tracks, bool shouldMute)
- {
- if (tracks.Length == 0)
- return;
- foreach (var track in tracks.Where(t => !TimelineUtility.IsParentMuted(t)))
- {
- TimelineUndo.PushUndo(track, "Mute Tracks");
- track.muted = shouldMute;
- }
- state.Refresh();
- }
- }
- [MenuEntry("Mute", MenuOrder.TrackAction.MuteTrack)]
- [Shortcut(Shortcuts.Timeline.toggleMute)]
- class MuteTrack : TrackAction
- {
- public static readonly string UnMuteText = L10n.Tr("Unmute");
- protected override MenuActionDisplayState GetDisplayState(WindowState state, TrackAsset[] tracks)
- {
- if (tracks.Any(track => TimelineUtility.IsParentMuted(track)))
- return MenuActionDisplayState.Disabled;
- return MenuActionDisplayState.Visible;
- }
- protected override string GetDisplayName(TrackAsset[] tracks)
- {
- return tracks.Any(x => !x.muted) ? base.GetDisplayName(tracks) : UnMuteText;
- }
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- if (!tracks.Any() || tracks.Any(track => TimelineUtility.IsParentMuted(track)))
- return false;
- var hasUnmutedTracks = tracks.Any(x => !x.muted);
- Mute(state, tracks, hasUnmutedTracks);
- return true;
- }
- public static void Mute(WindowState state, TrackAsset[] tracks, bool shouldMute)
- {
- if (tracks.Length == 0)
- return;
- foreach (var track in tracks)
- {
- if (track as GroupTrack == null)
- Mute(state, track.GetChildTracks().ToArray(), shouldMute);
- TimelineUndo.PushUndo(track, "Mute Tracks");
- track.muted = shouldMute;
- }
- state.Refresh();
- }
- }
- class DeleteTracks : TrackAction
- {
- public static void Do(TimelineAsset timeline, TrackAsset track)
- {
- SelectionManager.Remove(track);
- TrackModifier.DeleteTrack(timeline, track);
- }
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- // disable preview mode so deleted tracks revert to default state
- // Case 956129: Disable preview mode _before_ deleting the tracks, since clip data is still needed
- state.previewMode = false;
- TimelineAnimationUtilities.UnlinkAnimationWindowFromTracks(tracks);
- foreach (var track in tracks)
- Do(state.editSequence.asset, track);
- state.Refresh();
- return true;
- }
- }
- class CopyTracksToClipboard : TrackAction
- {
- public static bool Do(WindowState state, TrackAsset[] tracks)
- {
- var action = new CopyTracksToClipboard();
- return action.Execute(state, tracks);
- }
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- TimelineEditor.clipboard.CopyTracks(tracks);
- return true;
- }
- }
- class DuplicateTracks : TrackAction
- {
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- if (tracks.Any())
- {
- SelectionManager.RemoveTimelineSelection();
- }
- foreach (var track in TrackExtensions.FilterTracks(tracks))
- {
- var newTrack = track.Duplicate(TimelineEditor.inspectedDirector, TimelineEditor.inspectedDirector);
- SelectionManager.Add(newTrack);
- foreach (var childTrack in newTrack.GetFlattenedChildTracks())
- {
- SelectionManager.Add(childTrack);
- }
- }
- state.Refresh();
- return true;
- }
- }
- [MenuEntry("Remove Invalid Markers", MenuOrder.TrackAction.RemoveInvalidMarkers), UsedImplicitly]
- class RemoveInvalidMarkersAction : TrackAction
- {
- protected override MenuActionDisplayState GetDisplayState(WindowState state, TrackAsset[] tracks)
- {
- if (tracks.Any(target => target != null && target.GetMarkerCount() != target.GetMarkersRaw().Count()))
- return MenuActionDisplayState.Visible;
- return MenuActionDisplayState.Hidden;
- }
- public override bool Execute(WindowState state, TrackAsset[] tracks)
- {
- bool anyRemoved = false;
- foreach (var target in tracks)
- {
- var invalids = target.GetMarkersRaw().Where(x => !(x is IMarker)).ToList();
- foreach (var m in invalids)
- {
- anyRemoved = true;
- target.DeleteMarkerRaw(m);
- }
- }
- if (anyRemoved)
- TimelineEditor.Refresh(RefreshReason.ContentsAddedOrRemoved);
- return anyRemoved;
- }
- }
- }
|