GradientPicker.cs

// Unity C# reference source
// Copyright (c) Unity Technologies. For terms of use, see
// https://unity3d.com/legal/licenses/Unity_Reference_Only_License

using UnityEngine;
using UnityEditor;
using UnityEditorInternal;

namespace UnityEditor
{
    internal class GradientPicker : EditorWindow
    {
        private static GradientPicker s_GradientPicker;
        public static string presetsEditorPrefID { get { return "Gradient"; } }

        private GradientEditor m_GradientEditor;
        private PresetLibraryEditor<GradientPresetLibrary> m_GradientLibraryEditor;
        [SerializeField]
        private PresetLibraryEditorState m_GradientLibraryEditorState;
        private Gradient m_Gradient;
        private const int k_DefaultNumSteps = 0;
        private GUIView m_DelegateView;
        private System.Action<Gradient> m_Delegate;
        private bool m_HDR;
        private bool gradientChanged { get; set; }

        // Static methods
        public static void Show(Gradient newGradient, bool hdr)
        {
            GUIView currentView = GUIView.current;
            PrepareShow(hdr);
            s_GradientPicker.m_DelegateView = currentView;
            s_GradientPicker.m_Delegate = null;
            s_GradientPicker.Init(newGradient, hdr);

            GradientPreviewCache.ClearCache();
        }

        public static void Show(Gradient newGradient, bool hdr, System.Action<Gradient> onGradientChanged)
        {
            PrepareShow(hdr);
            s_GradientPicker.m_DelegateView = null;
            s_GradientPicker.m_Delegate = onGradientChanged;
            s_GradientPicker.Init(newGradient, hdr);

            GradientPreviewCache.ClearCache();
        }

        static void PrepareShow(bool hdr)
        {
            if (s_GradientPicker == null)
            {
                string title = hdr ? "HDR Gradient Editor" : "Gradient Editor";
                s_GradientPicker = (GradientPicker)GetWindow(typeof(GradientPicker), true, title, false);
                Vector2 minSize = new Vector2(360, 224);
                Vector2 maxSize = new Vector2(1900, 3000);
                s_GradientPicker.minSize = minSize;
                s_GradientPicker.maxSize = maxSize;
                s_GradientPicker.wantsMouseMove = true;
                s_GradientPicker.ShowAuxWindow(); // Use this if auto close on lost focus is wanted.
            }
            else
            {
                s_GradientPicker.Repaint(); // Ensure we get a OnGUI so we refresh if new gradient
            }
        }

        public static GradientPicker instance
        {
            get
            {
                if (!s_GradientPicker)
                    Debug.LogError("Gradient Picker not initalized, did you call Show first?");
                return s_GradientPicker;
            }
        }

        public string currentPresetLibrary
        {
            get
            {
                InitIfNeeded();
                return m_GradientLibraryEditor.currentLibraryWithoutExtension;
            }
            set
            {
                InitIfNeeded();
                m_GradientLibraryEditor.currentLibraryWithoutExtension = value;
            }
        }

        private void Init(Gradient newGradient, bool hdr)
        {
            m_Gradient = newGradient;
            m_HDR = hdr;
            if (m_GradientEditor != null)
                m_GradientEditor.Init(newGradient, k_DefaultNumSteps, m_HDR);
            Repaint();
        }

        private void SetGradientData(Gradient gradient)
        {
            m_Gradient.colorKeys = gradient.colorKeys;
            m_Gradient.alphaKeys = gradient.alphaKeys;
            m_Gradient.mode = gradient.mode;
            Init(m_Gradient, m_HDR);
        }

        public static bool visible
        {
            get { return s_GradientPicker != null; }
        }

        public static Gradient gradient
        {
            get
            {
                if (s_GradientPicker != null)
                    return s_GradientPicker.m_Gradient;
                return null;
            }
        }

        public void OnEnable()
        {
            hideFlags = HideFlags.DontSave;
            // Use these if window is not an aux window for auto closing on play/stop
            //EditorApplication.playmodeStateChanged += OnPlayModeStateChanged;
        }

        public void OnDisable()
        {
            //EditorApplication.playmodeStateChanged -= OnPlayModeStateChanged;
            if (m_GradientLibraryEditorState != null)
                m_GradientLibraryEditorState.TransferEditorPrefsState(false);

            s_GradientPicker = null;
        }

        public void OnDestroy()
        {
            m_GradientLibraryEditor.UnloadUsedLibraries();
        }

        void OnPlayModeStateChanged()
        {
            Close();
        }

        void InitIfNeeded()
        {
            // Init editor when needed
            if (m_GradientEditor == null)
            {
                m_GradientEditor = new GradientEditor();
                m_GradientEditor.Init(m_Gradient, k_DefaultNumSteps, m_HDR);
            }

            if (m_GradientLibraryEditorState == null)
            {
                m_GradientLibraryEditorState = new PresetLibraryEditorState(presetsEditorPrefID);
                m_GradientLibraryEditorState.TransferEditorPrefsState(true);
            }

            if (m_GradientLibraryEditor == null)
            {
                var saveLoadHelper = new ScriptableObjectSaveLoadHelper<GradientPresetLibrary>("gradients", SaveType.Text);
                m_GradientLibraryEditor = new PresetLibraryEditor<GradientPresetLibrary>(saveLoadHelper, m_GradientLibraryEditorState, PresetClickedCallback);
                m_GradientLibraryEditor.showHeader = true;
                m_GradientLibraryEditor.minMaxPreviewHeight = new Vector2(14f, 14f);
            }
        }

        void PresetClickedCallback(int clickCount, object presetObject)
        {
            Gradient gradient = presetObject as Gradient;
            if (gradient == null)
                Debug.LogError("Incorrect object passed " + presetObject);

            SetCurrentGradient(gradient);
            UnityEditorInternal.GradientPreviewCache.ClearCache();
            gradientChanged = true;
        }

        public void OnGUI()
        {
            // When we start play (using shortcut keys) we get two OnGui calls and m_Gradient is null: so early out.
            if (m_Gradient == null)
                return;

            InitIfNeeded();

            float gradientEditorHeight = Mathf.Min(position.height, 146);
            float distBetween = 10f;
            float presetLibraryHeight = position.height - gradientEditorHeight - distBetween;

            Rect gradientEditorRect = new Rect(10, 10, position.width - 20, gradientEditorHeight - 20);
            Rect gradientLibraryRect = new Rect(0, gradientEditorHeight + distBetween, position.width, presetLibraryHeight);

            // Separator
            EditorGUI.DrawRect(new Rect(gradientLibraryRect.x, gradientLibraryRect.y - 1, gradientLibraryRect.width, 1), new Color(0, 0, 0, 0.3f));
            EditorGUI.DrawRect(new Rect(gradientLibraryRect.x, gradientLibraryRect.y, gradientLibraryRect.width, 1), new Color(1, 1, 1, 0.1f));

            // The meat
            EditorGUI.BeginChangeCheck();
            m_GradientEditor.OnGUI(gradientEditorRect);
            if (EditorGUI.EndChangeCheck())
                gradientChanged = true;
            m_GradientLibraryEditor.OnGUI(gradientLibraryRect, m_Gradient);
            if (gradientChanged)
            {
                gradientChanged = false;
                SendEvent(true);
            }
        }

        public const string GradientPickerChangedCommand = "GradientPickerChanged";

        void SendEvent(bool exitGUI)
        {
            if (m_DelegateView)
            {
                Event e = EditorGUIUtility.CommandEvent(GradientPickerChangedCommand);
                Repaint();
                m_DelegateView.SendEvent(e);
                if (exitGUI)
                    GUIUtility.ExitGUI();
            }
            if (m_Delegate != null)
            {
                m_Delegate(gradient);
            }
        }

        public static void SetCurrentGradient(Gradient gradient)
        {
            if (s_GradientPicker == null)
                return;

            s_GradientPicker.SetGradientData(gradient);
            GUI.changed = true;
        }

        public static void CloseWindow()
        {
            if (s_GradientPicker == null)
                return;

            s_GradientPicker.Close();
            GUIUtility.ExitGUI();
        }

        public static void RepaintWindow()
        {
            if (s_GradientPicker == null)
                return;
            s_GradientPicker.Repaint();
        }
    }
} // namespace