VFXToolbox/Editor/ImageSequencer/ImageSequencer.cs

using UnityEngine;
using UnityEngine.Rendering;
using UnityEditorInternal;
using System.Collections.Generic;
using System.Linq;

namespace UnityEditor.Experimental.VFX.Toolbox.ImageSequencer
{
    internal partial class ImageSequencer : EditorWindow
    {
        [MenuItem("Window/Visual Effects/Image Sequencer", priority = 3031)]
        public static void OpenEditor()
        {
            GetWindow(typeof(ImageSequencer));
        }

        public enum SidePanelMode
        {
            InputFrames = 0,
            Processors = 1,
            Export = 2
        }

        public ImageSequencerCanvas previewCanvas
        {
            get
            {
                return m_PreviewCanvas;
            }
        }

        public SidePanelMode sidePanelViewMode
        {
            get
            {
                return m_SidePanelViewMode;
            }
            set
            {
                m_SidePanelViewMode = value;
            }
        }

        public ProcessingNode currentProcessingNode
        {
            get
            {
                return m_CurrentProcessingNode;
            }
        }

        private ImageSequence m_CurrentAsset;
        private SerializedObject m_CurrentAssetSerializedObject;
        private SerializedObject m_SettingsReferenceSerializedObject;
        private bool m_IgnoreInheritSettings;

        private ImageSequencerCanvas m_PreviewCanvas;
        private ProcessingNodeStack m_ProcessingNodeStack;
        private bool m_AutoCook = false;
        private ProcessingNode m_CurrentProcessingNode;
        private ProcessingNode m_LockedPreviewProcessor;

        private GraphicsDeviceType m_CurrentGraphicsAPI;
        private ColorSpace m_CurrentColorSpace;

        public ImageSequencer()
        {
            Selection.selectionChanged -= OnEditorSelectionChange;
            Selection.selectionChanged += OnEditorSelectionChange;
            Undo.undoRedoPerformed -= OnUndoRedo;
            Undo.undoRedoPerformed += OnUndoRedo;
        }

        void OnEnable()
        {
            m_ProcessorDataProvider = null;

            if(Selection.activeObject != null && Selection.activeObject is ImageSequence)
            {
                LoadAsset((ImageSequence)Selection.activeObject);
                DefaultView();
            }
            else if(m_CurrentAsset != null)
            {
                LoadAsset(m_CurrentAsset);
                DefaultView();
            }

        }

        void OnDisable()
        {
            LoadAsset(null);
            Selection.selectionChanged -= OnEditorSelectionChange;
            Undo.undoRedoPerformed -= OnUndoRedo;

        }

        public void OnEditorSelectionChange()
        {
            if(Selection.activeObject != null && Selection.activeObject.GetType() == typeof(ImageSequence) && m_CurrentAsset != Selection.activeObject)
            {
                LoadAsset((ImageSequence)Selection.activeObject);
                DefaultView();
            }
            Repaint();
        }

        public void OnUndoRedo()
        {
            if (m_CurrentAsset == null)
                return;

            SidePanelMode bkpSidePanelMode = m_SidePanelViewMode;

            int hash = GetInputTexturesHashCode();

            if (m_InputFramesHashCode != hash || m_CurrentAsset.inheritSettingsReference != null)
                LoadAsset(m_CurrentAsset);

            m_ProcessingNodeStack.LoadProcessorsFromAsset(m_CurrentAsset);
            RestoreProcessorView();

            if (m_CurrentAsset.inputFrameGUIDs.Count > 0)
                m_SidePanelViewMode = bkpSidePanelMode;
            else
                m_SidePanelViewMode = SidePanelMode.InputFrames;

            foreach (ProcessingNode n in m_ProcessingNodeStack.nodes)
            {
                n.Refresh();
                n.Invalidate();
            }

            Repaint();
        }

        private ImageSequence FindSettingsReference(ImageSequence asset, ref List<ImageSequence> dependencyList)
        {
            if (asset.inheritSettingsReference != null)
            {
                if (dependencyList.Contains(asset.inheritSettingsReference))
                    return null;

                dependencyList.Add(asset.inheritSettingsReference);
                return FindSettingsReference(asset.inheritSettingsReference, ref dependencyList);
            }
            else
                return asset;
        }

        public void LoadAsset(ImageSequence asset)
        {
            m_CurrentAsset = asset;

            m_InputFramesReorderableList = null;
            m_ProcessorsReorderableList = null;
            m_LockedPreviewProcessor = null;
            m_CurrentProcessingNode = null;

            // Free resources if any
            if(m_ProcessingNodeStack != null)
                m_ProcessingNodeStack.Dispose();

            InitializeGUI();

            if(m_CurrentAsset != null)
            {
                m_ProcessingNodeStack = new ProcessingNodeStack(new ProcessingFrameSequence(null), this);

                m_CurrentAssetSerializedObject = new SerializedObject(m_CurrentAsset);

                VFXToolboxGUIUtility.DisplayProgressBar("Image Sequencer", "Loading asset....", 0.0f);

                m_LockedPreviewProcessor = null;

                VFXToolboxGUIUtility.DisplayProgressBar("Image Sequencer", "Loading Frames", 0.333333f);

                m_ProcessingNodeStack.LoadFramesFromAsset(m_CurrentAsset);
                UpdateInputTexturesHash();

                m_InputFramesReorderableList = new ReorderableList(m_ProcessingNodeStack.inputSequence.frames, typeof(Texture2D),true,false,true,true);
                m_InputFramesReorderableList.onAddCallback = AddInputFrame;
                m_InputFramesReorderableList.onRemoveCallback = RemoveInputFrame;
                m_InputFramesReorderableList.onReorderCallback = ReorderInputFrame;
                m_InputFramesReorderableList.drawElementCallback = DrawInputFrameRListElement;
                m_InputFramesReorderableList.onSelectCallback = SelectInputFrameRListElement;

                VFXToolboxGUIUtility.DisplayProgressBar("Image Sequencer", "Loading Processors", 0.66666f);

                ImageSequence inheritedSettingReference = m_CurrentAsset;

                // Loading other settings if inheriting settings
                if(m_CurrentAsset.inheritSettingsReference != null)
                {
                    var dependencyList = new List<ImageSequence>();
                    var referenceAsset = FindSettingsReference(m_CurrentAsset.inheritSettingsReference, ref dependencyList);
                    if (referenceAsset == null)
                    {
                        Debug.LogWarning("Dependency Loop detected, ignoring using external settings");
                        m_IgnoreInheritSettings = true;
                    }
                    else
                    {
                        inheritedSettingReference = referenceAsset;
                        m_IgnoreInheritSettings = false;
                    }
                }

                m_ProcessingNodeStack.LoadProcessorsFromAsset(inheritedSettingReference);
                m_ProcessorDataProvider = new ProcessorDataProvider(m_ProcessingNodeStack, m_CurrentAsset);

                // Construct the RList
                if (m_CurrentAsset.inheritSettingsReference == null)
                {
                    m_ProcessorsReorderableList = new ReorderableList(m_CurrentAssetSerializedObject, m_CurrentAssetSerializedObject.FindProperty("processorInfos"),true,false,true,true);
                    m_ProcessorsReorderableList.onAddCallback = ShowAddProcessorMenu;
                    m_ProcessorsReorderableList.onRemoveCallback = MenuRemoveProcessor;
                    m_ProcessorsReorderableList.onReorderCallback = ReorderProcessor;
                    m_ProcessorsReorderableList.onSelectCallback = MenuSelectProcessor;
                    m_ProcessorsReorderableList.drawElementCallback = DrawRListProcessorElement;
                    m_SettingsReferenceSerializedObject = null;
                }
                else
                {
                    m_SettingsReferenceSerializedObject = new SerializedObject(inheritedSettingReference);
                    m_ProcessorsReorderableList = new ReorderableList(m_SettingsReferenceSerializedObject, m_SettingsReferenceSerializedObject.FindProperty("processorInfos"),false,false,false,false);
                    m_ProcessorsReorderableList.drawElementCallback = DrawRListPreviewProcessorElement;
                    m_ProcessorsReorderableList.onSelectCallback = MenuSelectProcessor;
                }

                m_PreviewCanvas.sequence = m_ProcessingNodeStack.inputSequence;
                if(m_PreviewCanvas.sequence.length > 0)
                    m_PreviewCanvas.currentFrameIndex = 0;
                else
                    m_PreviewCanvas.currentFrameIndex = -1;

                VFXToolboxGUIUtility.DisplayProgressBar("Image Sequencer", "Finalizing...", 1.0f);

                m_ProcessingNodeStack.InvalidateAll();
                RestoreProcessorView();

                EditorUtility.ClearProgressBar();
            }
        }

        /// <summary>
        /// Setups the view for post load asset
        /// </summary>
        public void DefaultView()
        {
            if (m_ProcessingNodeStack.nodes.Count == 0 || m_ProcessingNodeStack.inputSequence.frames.Count == 0)
            {
                m_SidePanelViewMode = SidePanelMode.InputFrames;
                m_ProcessorsReorderableList.index = -1;
                SetCurrentFrameProcessor(null, false);
            }
            else
            {
                m_SidePanelViewMode = SidePanelMode.Processors;
                RestoreProcessorView();
            }

            m_PreviewCanvas.UpdateCanvasSequence();

            if(m_PreviewCanvas != null)
                m_PreviewCanvas.Recenter(false);
        }

        /// <summary>
        /// Restores the visibility and lock of processors (on load or after an undo)
        /// </summary>
        public void RestoreProcessorView()
        {
            if(m_CurrentAsset.inheritSettingsReference != null && !m_IgnoreInheritSettings)
            {
                if(m_ProcessingNodeStack.nodes.Count > 0)
                {
                    if(m_CurrentAsset.editSettings.selectedProcessor > 0)
                    {
                        m_CurrentProcessingNode = m_ProcessingNodeStack.nodes[m_CurrentAsset.editSettings.selectedProcessor];
                        m_ProcessorsReorderableList.index = m_CurrentAsset.editSettings.selectedProcessor;
                    }
                }
            }
            else
            {
                // index Checks
                m_CurrentAsset.editSettings.lockedProcessor = Mathf.Clamp(m_CurrentAsset.editSettings.lockedProcessor, -1, m_ProcessingNodeStack.nodes.Count - 1);
                m_CurrentAsset.editSettings.selectedProcessor = Mathf.Clamp(m_CurrentAsset.editSettings.selectedProcessor, -1, m_ProcessingNodeStack.nodes.Count - 1);

                // Locked processor
                if (m_CurrentAsset.editSettings.lockedProcessor != -1)
                {
                    m_ProcessorsReorderableList.index = m_CurrentAsset.editSettings.lockedProcessor;
                    m_LockedPreviewProcessor = m_ProcessingNodeStack.nodes[m_CurrentAsset.editSettings.lockedProcessor];
                    m_CurrentProcessingNode = m_ProcessingNodeStack.nodes[m_CurrentAsset.editSettings.lockedProcessor];
                }
                else
                    m_LockedPreviewProcessor = null; 

                // Selected Processor
                if(m_CurrentAsset.editSettings.selectedProcessor != -1)
                {
                    m_ProcessorsReorderableList.index = m_CurrentAsset.editSettings.selectedProcessor;

                    if (m_CurrentAsset.editSettings.lockedProcessor != -1)
                        m_CurrentProcessingNode = m_ProcessingNodeStack.nodes[m_CurrentAsset.editSettings.lockedProcessor];
                    else
                        m_CurrentProcessingNode = m_ProcessingNodeStack.nodes[m_CurrentAsset.editSettings.selectedProcessor];
                }
            }

            m_ProcessingNodeStack.InvalidateAll();
            RefreshCanvas();
        }

        private void CheckGraphicsSettings()
        {
            GraphicsDeviceType device = SystemInfo.graphicsDeviceType;
            if(m_CurrentGraphicsAPI != device)
            {
                m_CurrentGraphicsAPI = device;
                if(m_ProcessingNodeStack != null)
                    m_ProcessingNodeStack.InvalidateAll();
                Repaint();
            }
            ColorSpace colorSpace = QualitySettings.activeColorSpace;
            if(m_CurrentColorSpace != colorSpace)
            {
                m_CurrentColorSpace = colorSpace;
                if(m_ProcessingNodeStack != null)
                    m_ProcessingNodeStack.InvalidateAll();
            }
        }

        public static void CleanupAsset(ImageSequence asset)
        {
            var subAssets = AssetDatabase.LoadAllAssetsAtPath(AssetDatabase.GetAssetPath(asset));
            List<Object> toDelete = new List<Object>();

            var allSettings = asset.processorInfos.Select(i => i.Settings);

            foreach(var subAsset in subAssets)
            {
                if(subAsset is ProcessorInfo && !asset.processorInfos.Contains(subAsset))
                {
                    toDelete.Add(subAsset); 
                }
                else if(subAsset is ProcessorBase && !allSettings.Contains(subAsset))
                {
                    toDelete.Add(subAsset);
                }
            }

            foreach(var o in toDelete)
            {
                AssetDatabase.RemoveObjectFromAsset(o);
            }

            if(toDelete.Count > 0)
            {
                EditorUtility.SetDirty(asset);
            }
        }
    }
}