VFXToolbox/ImageSequencer/Editor/FrameProcessorStack.cs

using System;
using System.Linq;
using System.Reflection;
using System.Collections.Generic;

namespace UnityEditor.VFXToolbox.ImageSequencer
{
    internal partial class FrameProcessorStack
    {
        public ProcessingFrameSequence inputSequence
        {
            get
            {
                return m_InputSequence;
            }
        }

        public ProcessingFrameSequence outputSequence
        {
            get
            {
                if (m_Processors.Count > 0)
                    return m_Processors[m_Processors.Count - 1].OutputSequence;
                else
                    return m_InputSequence;
            }
        }

        public ImageSequencer imageSequencer
        {
            get { return m_ImageSequencer; }
        }

        public List<FrameProcessor> processors
        {
            get
            {
                return m_Processors;
            }
        }

        private List<FrameProcessor> m_Processors;
        private ProcessingFrameSequence m_InputSequence;
        private ImageSequencer m_ImageSequencer;

        public FrameProcessorStack(ProcessingFrameSequence inputSequence, ImageSequencer imageSequencer)
        {
            m_InputSequence = inputSequence;
            m_Processors = new List<FrameProcessor>();
            m_ImageSequencer = imageSequencer;
            
        }

        public void Dispose()
        {
            foreach(FrameProcessor p in m_Processors)
            {
                p.Dispose();
            }
            m_Processors.Clear();
        }

        public ProcessingFrameSequence GetOutputSequence()
        {
            if(m_Processors.Count > 0)
            {
                return m_Processors[m_Processors.Count - 1].OutputSequence;
            }
            else
            {
                return inputSequence;
            }
        }

        public ProcessingFrameSequence GetInputSequence(FrameProcessor processor)
        {
            int index = m_Processors.IndexOf(processor);

            if (index > 0)
            {
                return m_Processors[index - 1].OutputSequence;
            }
            else
                return m_InputSequence;
        }

        public FrameProcessor GetNextProcessor(FrameProcessor processor)
        {
            int index = m_Processors.IndexOf(processor);
            if(index < m_Processors.Count-1)
            {
                return m_Processors[index + 1];
            }
            return null;
        }

        public void Invalidate(FrameProcessor processor)
        {
            int index = m_Processors.IndexOf(processor);
            if(index != -1)
                m_Processors[index].Invalidate();
        }

        public void InvalidateAll()
        {
            if (m_Processors.Count > 0)
                m_Processors[0].Invalidate();
        }


        public Dictionary<Type, ProcessorAttribute> settingsDefinitions { get; private set; }

        public void UpdateProcessorsFromAssembly()
        {
            settingsDefinitions = new Dictionary<Type, ProcessorAttribute>();

            var assembly = Assembly.GetAssembly(typeof(ProcessorSettingsBase));
            var types = assembly.GetTypes();
            var processorSettingsType = typeof(ProcessorSettingsBase);
            var attrType = typeof(ProcessorAttribute);

            var processorSettingTypes = types
                .Where(t => t.IsClass
                    && !t.IsAbstract
                    && t.IsSubclassOf(processorSettingsType)
                    && t.IsDefined(attrType, false));

            foreach (var processorSettingType in processorSettingTypes)
            {
                var attr = (ProcessorAttribute)processorSettingType.GetCustomAttributes(attrType, false)[0];
                Type processorType = attr.processorType;

                if (!processorType.IsClass || !processorType.IsSubclassOf(typeof(FrameProcessor)) || processorType.IsAbstract)
                    throw new InvalidOperationException("Invalid RendererAttribute parameter, type must be a non-abstract class that extends GPUFrameProcessor<>");

                settingsDefinitions.Add(processorSettingType, attr);
            }
        }


    }

}