AutoStreaming-Demo/Packages/net.peeweek.gameplay-ingredients/Editor/SelectionHistory/SelectionHistoryWindow.cs

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

namespace GameplayIngredients.Editor
{
    public class SelectionHistoryWindow : EditorWindow
    {
        [MenuItem("Window/General/Selection History")]
        public static void OpenSelectionHistoryWindow()
        {
            EditorWindow.GetWindow<SelectionHistoryWindow>();
        }

        Vector2 scrollPos = Vector2.zero;

        void OnGUI()
        {
            titleContent = Contents.title;

            scrollPos = EditorGUILayout.BeginScrollView(scrollPos);
            {
                Selection_OnGUI();
            }
            EditorGUILayout.EndScrollView();
        }

        void OnEnable()
        {
            lockedObjects = null;
            selectionHistory = null;
        }

        void OnDisable()
        {
            lockedObjects = null;
            selectionHistory = null;
        }

        List<GameObject> selectionHistory;
        List<GameObject> lockedObjects;

        int maxHistoryCount = 32;
        bool ignoreNextSelection = false;

        void OnSelectionChange()
        {
            if (ignoreNextSelection)
            {
                ignoreNextSelection = false;
                return;
            }

            if (selectionHistory == null) selectionHistory = new List<GameObject>();
            if (lockedObjects == null) lockedObjects = new List<GameObject>();

            if (Selection.activeGameObject != null || Selection.gameObjects.Length > 0)
            {

                foreach(var go in Selection.gameObjects)
                {
                    if (!selectionHistory.Contains(go))
                        selectionHistory.Add(go);
                }

                if (selectionHistory.Count > maxHistoryCount)
                    selectionHistory.Take(maxHistoryCount);

                Repaint();
            }

        }

        public bool CompareArray(GameObject[] a, GameObject[] b)
        {
            return a.SequenceEqual(b);
        }

        void Selection_OnGUI()
        {
            if (selectionHistory == null) selectionHistory = new List<GameObject>();
            if (lockedObjects == null) lockedObjects = new List<GameObject>();
            int i = 0;
            int toRemove = -1;

            if (lockedObjects.Count > 0)
            {
                GUILayout.Label("Favorites", EditorStyles.boldLabel);
                i = 0;
                toRemove = -1;
                foreach (var obj in lockedObjects)
                {
                    if (obj == null)
                    {
                        using (new EditorGUILayout.HorizontalScope())
                        {
                            GUILayout.Label("(object is either null or has been deleted)");
                            if (GUILayout.Button("X", GUILayout.Width(24)))
                            {
                                toRemove = i;
                            }
                        }
                    }
                    else
                    {
                        bool highlight = Selection.gameObjects.Contains(obj);
                        Color backup = GUI.color;

                        if (highlight)
                            GUI.color = Styles.highlightColor;

                        using (new EditorGUILayout.HorizontalScope())
                        {
                            var b = GUI.color;
                            GUI.color = Color.yellow * 3;
                            if (GUILayout.Button(Contents.star, Styles.icon, GUILayout.Width(24)))
                            {
                                toRemove = i;
                            }

                            GUI.color = b;

                            string label = obj.name;

                            if (GUILayout.Button(label, EditorStyles.foldout))
                            {
                                ignoreNextSelection = true;
                                Selection.activeObject = obj;
                            }
                            if (GUILayout.Button("Focus", EditorStyles.miniButton, GUILayout.Width(40)))
                            {
                                ignoreNextSelection = true;
                                Selection.activeObject = obj;
                                SceneView.lastActiveSceneView.FrameSelected();
                            }
                        }

                        GUI.color = backup;
                    }
                    i++;
                }
                if (toRemove != -1) lockedObjects.RemoveAt(toRemove);
            }

            int toAdd = -1;
            toRemove = -1;
            i = 0;
            using (new EditorGUILayout.HorizontalScope())
            {
                GUILayout.Label("History", EditorStyles.boldLabel);
                GUILayout.FlexibleSpace();
                if (GUILayout.Button("Clear", EditorStyles.miniButton))
                {
                    selectionHistory.Clear();
                    Repaint();
                }
            }

            GUILayout.Space(8);

            var reversedHistory = selectionHistory.Reverse<GameObject>().ToArray();
            foreach (var obj in reversedHistory)
            {
                if (obj != null)
                {
                    bool highlight = Selection.gameObjects.Contains(obj);
                    Color backup = GUI.color;

                    if (highlight)
                        GUI.color = Styles.highlightColor;

                    using (new EditorGUILayout.HorizontalScope())
                    {

                        if (GUILayout.Button(Contents.starDisabled, Styles.icon, GUILayout.Width(24)))
                        {
                            toAdd = i;
                        }

                        string label = obj.name;
                        if (GUILayout.Button(label, EditorStyles.foldout))
                        {
                            ignoreNextSelection = true;
                            Selection.activeObject = obj;
                        }
                        if (GUILayout.Button("Focus", Styles.historyButton, GUILayout.Width(40)))
                        {
                            ignoreNextSelection = true;
                            Selection.activeObject = obj;
                            SceneView.lastActiveSceneView.FrameSelected();
                        }
                    }
                    var rect = GUILayoutUtility.GetLastRect();
                    EditorGUI.DrawRect(rect, new Color(0.2f,0.2f,0.2f,0.5f));

                    GUI.color = backup;
                }

                i++;
            }
            if (toAdd != -1)
            {
                lockedObjects.Add(reversedHistory[toAdd]);
                Repaint();
            }
            if (toRemove != -1)
            {
                selectionHistory.RemoveAt(toRemove);
                Repaint();
            }

        }

        static class Styles
        {
            public static GUIStyle historyButton;
            public static GUIStyle highlight;
            public static Color highlightColor = new Color(2.0f, 2.0f, 2.0f);

            public static GUIStyle icon;

            static Styles()
            {
                historyButton = new GUIStyle(EditorStyles.miniButton);
                historyButton.alignment = TextAnchor.MiddleLeft;
                highlight = new GUIStyle(EditorStyles.miniLabel);
                highlight.onNormal.background = Texture2D.whiteTexture;
                highlight.onHover.background = Texture2D.whiteTexture;
                highlight.onActive.background = Texture2D.whiteTexture;
                highlight.onFocused.background = Texture2D.whiteTexture;

                icon = new GUIStyle(EditorStyles.label);
                icon.fixedHeight = 16;
                icon.padding = new RectOffset(8,2,2,0);
                icon.margin = new RectOffset();

            }
        }

        static class Contents
        {
            public static GUIContent title = new GUIContent("Selection History");
            public static GUIContent star = new GUIContent(EditorGUIUtility.IconContent("Favorite Icon").image);
            public static GUIContent starDisabled = new GUIContent(EditorGUIUtility.IconContent("Favorite").image);
        }
    }
}