123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344 |
- using System;
- using System.Collections.Generic;
- using UnityEngine;
- using UnityEngine.Timeline;
- namespace UnityEditor.Timeline
- {
- static class EditMode
- {
- public enum EditType
- {
- None = -1,
- Mix = 0,
- Ripple = 1,
- Replace = 2
- }
- interface ISubEditMode
- {
- IMoveItemMode moveItemMode { get; }
- IMoveItemDrawer moveItemDrawer { get; }
- ITrimItemMode trimItemMode { get; }
- ITrimItemDrawer trimItemDrawer { get; }
- IAddDeleteItemMode addDeleteItemMode { get; }
- Color color { get; }
- KeyCode clutchKey { get; }
- void Reset();
- }
- class SubEditMode<TMoveMode, TTrimMode, TAddDeleteMode>: ISubEditMode
- where TMoveMode : class, IMoveItemMode, IMoveItemDrawer, new()
- where TTrimMode : class, ITrimItemMode, ITrimItemDrawer, new()
- where TAddDeleteMode : class, IAddDeleteItemMode, new()
- {
- public SubEditMode(Color guiColor, KeyCode key)
- {
- color = guiColor;
- clutchKey = key;
- Reset();
- }
- public void Reset()
- {
- m_MoveItemMode = new TMoveMode();
- m_TrimItemMode = new TTrimMode();
- m_AddDeleteItemMode = new TAddDeleteMode();
- }
- TMoveMode m_MoveItemMode;
- TTrimMode m_TrimItemMode;
- TAddDeleteMode m_AddDeleteItemMode;
- public IMoveItemMode moveItemMode { get { return m_MoveItemMode; } }
- public IMoveItemDrawer moveItemDrawer { get { return m_MoveItemMode; } }
- public ITrimItemMode trimItemMode { get { return m_TrimItemMode; } }
- public ITrimItemDrawer trimItemDrawer { get { return m_TrimItemMode; } }
- public IAddDeleteItemMode addDeleteItemMode { get { return m_AddDeleteItemMode; } }
- public Color color { get; }
- public KeyCode clutchKey { get; }
- }
- static readonly Dictionary<EditType, ISubEditMode> k_EditModes = new Dictionary<EditType, ISubEditMode>
- {
- { EditType.Mix, new SubEditMode<MoveItemModeMix, TrimItemModeMix, AddDeleteItemModeMix>(DirectorStyles.kMixToolColor, KeyCode.Alpha1) },
- { EditType.Ripple, new SubEditMode<MoveItemModeRipple, TrimItemModeRipple, AddDeleteItemModeRipple>(DirectorStyles.kRippleToolColor, KeyCode.Alpha2) },
- { EditType.Replace, new SubEditMode<MoveItemModeReplace, TrimItemModeReplace, AddDeleteItemModeReplace>(DirectorStyles.kReplaceToolColor, KeyCode.Alpha3) }
- };
- static EditType s_CurrentEditType = EditType.Mix;
- static EditType s_OverrideEditType = EditType.None;
- static ITrimmable s_CurrentTrimItem;
- static TrimEdge s_CurrentTrimDirection;
- static MoveItemHandler s_CurrentMoveItemHandler;
- static EditModeInputHandler s_InputHandler = new EditModeInputHandler();
- static ITrimItemMode trimMode
- {
- get { return GetSubEditMode(editType).trimItemMode; }
- }
- static ITrimItemDrawer trimDrawer
- {
- get { return GetSubEditMode(editType).trimItemDrawer; }
- }
- static IMoveItemMode moveMode
- {
- get { return GetSubEditMode(editType).moveItemMode; }
- }
- static IMoveItemDrawer moveDrawer
- {
- get { return GetSubEditMode(editType).moveItemDrawer; }
- }
- static IAddDeleteItemMode addDeleteMode
- {
- get { return GetSubEditMode(editType).addDeleteItemMode; }
- }
- public static EditModeInputHandler inputHandler
- {
- get { return s_InputHandler; }
- }
- static Color modeColor
- {
- get { return GetSubEditMode(editType).color; }
- }
- public static EditType editType
- {
- get
- {
- if (s_OverrideEditType != EditType.None)
- return s_OverrideEditType;
- var window = TimelineWindow.instance;
- if (window != null)
- s_CurrentEditType = window.state.editType;
- return s_CurrentEditType;
- }
- set
- {
- s_CurrentEditType = value;
- var window = TimelineWindow.instance;
- if (window != null)
- window.state.editType = value;
- s_OverrideEditType = EditType.None;
- }
- }
- static ISubEditMode GetSubEditMode(EditType type)
- {
- var subEditMode = k_EditModes[type];
- if (subEditMode != null)
- return subEditMode;
- Debug.LogError("Unsupported editmode type");
- return null;
- }
- static EditType GetSubEditType(KeyCode key)
- {
- foreach (var subEditMode in k_EditModes)
- {
- if (subEditMode.Value.clutchKey == key)
- return subEditMode.Key;
- }
- return EditType.None;
- }
- public static void ClearEditMode()
- {
- k_EditModes[editType].Reset();
- }
- public static void BeginTrim(ITimelineItem item, TrimEdge trimDirection)
- {
- var itemToTrim = item as ITrimmable;
- if (itemToTrim == null) return;
- s_CurrentTrimItem = itemToTrim;
- s_CurrentTrimDirection = trimDirection;
- trimMode.OnBeforeTrim(itemToTrim, trimDirection);
- TimelineUndo.PushUndo(itemToTrim.parentTrack, "Trim Clip");
- }
- public static void TrimStart(ITimelineItem item, double time)
- {
- var itemToTrim = item as ITrimmable;
- if (itemToTrim == null) return;
- trimMode.TrimStart(itemToTrim, time);
- }
- public static void TrimEnd(ITimelineItem item, double time, bool affectTimeScale)
- {
- var itemToTrim = item as ITrimmable;
- if (itemToTrim == null) return;
- trimMode.TrimEnd(itemToTrim, time, affectTimeScale);
- }
- public static void DrawTrimGUI(WindowState state, TimelineItemGUI item, TrimEdge edge)
- {
- trimDrawer.DrawGUI(state, item.rect, modeColor, edge);
- }
- public static void FinishTrim()
- {
- s_CurrentTrimItem = null;
- TimelineCursors.ClearCursor();
- ClearEditMode();
- TimelineEditor.Refresh(RefreshReason.ContentsModified);
- }
- public static void BeginMove(MoveItemHandler moveItemHandler)
- {
- s_CurrentMoveItemHandler = moveItemHandler;
- moveMode.BeginMove(s_CurrentMoveItemHandler.movingItems);
- }
- public static void UpdateMove()
- {
- moveMode.UpdateMove(s_CurrentMoveItemHandler.movingItems);
- }
- public static void OnTrackDetach(IEnumerable<ItemsPerTrack> grabbedTrackItems)
- {
- moveMode.OnTrackDetach(grabbedTrackItems);
- }
- public static void HandleTrackSwitch(IEnumerable<ItemsPerTrack> grabbedTrackItems)
- {
- moveMode.HandleTrackSwitch(grabbedTrackItems);
- }
- public static bool AllowTrackSwitch()
- {
- return moveMode.AllowTrackSwitch();
- }
- public static double AdjustStartTime(WindowState state, ItemsPerTrack itemsGroup, double time)
- {
- return moveMode.AdjustStartTime(state, itemsGroup, time);
- }
- public static bool ValidateDrag(ItemsPerTrack itemsGroup)
- {
- return moveMode.ValidateMove(itemsGroup);
- }
- public static void DrawMoveGUI(WindowState state, IEnumerable<MovingItems> movingItems)
- {
- moveDrawer.DrawGUI(state, movingItems, modeColor);
- }
- public static void FinishMove()
- {
- var manipulatedItemsList = s_CurrentMoveItemHandler.movingItems;
- moveMode.FinishMove(manipulatedItemsList);
- foreach (var itemsGroup in manipulatedItemsList)
- foreach (var item in itemsGroup.items)
- item.parentTrack = itemsGroup.targetTrack;
- s_CurrentMoveItemHandler = null;
- TimelineCursors.ClearCursor();
- ClearEditMode();
- TimelineEditor.Refresh(RefreshReason.ContentsModified);
- }
- public static void FinalizeInsertItemsAtTime(IEnumerable<ItemsPerTrack> newItems, double requestedTime)
- {
- addDeleteMode.InsertItemsAtTime(newItems, requestedTime);
- }
- public static void PrepareItemsDelete(IEnumerable<ItemsPerTrack> newItems)
- {
- addDeleteMode.RemoveItems(newItems);
- }
- public static void HandleModeClutch()
- {
- if (Event.current.type == EventType.KeyDown && EditorGUI.IsEditingTextField())
- return;
- var prevType = editType;
- if (Event.current.type == EventType.KeyDown)
- {
- var clutchEditType = GetSubEditType(Event.current.keyCode);
- if (clutchEditType != EditType.None)
- {
- s_OverrideEditType = clutchEditType;
- Event.current.Use();
- }
- }
- else if (Event.current.type == EventType.KeyUp)
- {
- var clutchEditType = GetSubEditType(Event.current.keyCode);
- if (clutchEditType == s_OverrideEditType)
- {
- s_OverrideEditType = EditType.None;
- Event.current.Use();
- }
- }
- if (prevType != editType)
- {
- if (s_CurrentTrimItem != null)
- {
- trimMode.OnBeforeTrim(s_CurrentTrimItem, s_CurrentTrimDirection);
- }
- else if (s_CurrentMoveItemHandler != null)
- {
- if (s_CurrentMoveItemHandler.movingItems == null)
- {
- s_CurrentMoveItemHandler = null;
- return;
- }
- foreach (var movingItems in s_CurrentMoveItemHandler.movingItems)
- {
- if (movingItems != null && movingItems.HasAnyDetachedParents())
- {
- foreach (var items in movingItems.items)
- {
- items.parentTrack = movingItems.originalTrack;
- }
- }
- }
- var movingSelection = s_CurrentMoveItemHandler.movingItems;
- // Handle clutch key transition if needed
- GetSubEditMode(prevType).moveItemMode.OnModeClutchExit(movingSelection);
- moveMode.OnModeClutchEnter(movingSelection);
- moveMode.BeginMove(movingSelection);
- moveMode.HandleTrackSwitch(movingSelection);
- UpdateMove();
- s_CurrentMoveItemHandler.RefreshPreviewItems();
- TimelineWindow.instance.state.rebuildGraph = true; // TODO Rebuild only if parent changed
- }
- TimelineWindow.instance.Repaint(); // TODO Refresh the toolbar without doing a full repaint?
- }
- }
- }
- }
|