Laurens-Packages/Packages/com.unity.render-pipelines.core/Editor/Controls/ToggleDropdown.cs

using System.Collections.Generic;
using UnityEngine;
using UnityEngine.UIElements;

namespace UnityEditor.Rendering
{
    [UxmlElement]
    internal partial class ToggleDropdown : VisualElement
    {
        const string k_StylesheetPathFormat = "Packages/com.unity.render-pipelines.core/Editor/Controls/ToggleDropdown.uss";
        const string k_MainClass = "toggle-dropdown";
        const string k_ToggleButtonClass = k_MainClass + "__toggle-button";
        const string k_DropdownButtonClass = k_MainClass + "__dropdown-button";
        const string k_SeparatorClass = k_MainClass + "__separator";
        const string k_EnabledClass = k_MainClass + "--enabled";

        private Button m_ToggleButton;
        private Button m_DropdownButton;
        private VisualElement m_DropdownArrow;
        private VisualElement m_Separator;
        private List<string> m_Options = new List<string>();
        private HashSet<int> m_SelectedIndices = new HashSet<int>();
        private int m_SelectedIndex = 0;
        private bool m_IsEnabled = false;
        private string m_Text = "Toggle Dropdown";

        [UxmlAttribute]
        public string text
        {
            get => m_Text;
            set
            {
                if (m_Text == value)
                    return;
                m_Text = value;
                m_ToggleButton.text = m_Text;
            }
        }

        [UxmlAttribute]
        private string options { get; set; } = "";

        [UxmlAttribute("selected-index")]
        private int selectedIndex
        {
            get => m_SelectedIndex;
            set
            {
                if (m_SelectedIndex == value || value < 0)
                    return;
                m_SelectedIndex = value;
            }
        }

        [UxmlAttribute("selected-indices")]
        public string selectedIndices
        {
            get
            {
                var indices = new List<string>();
                foreach (int index in m_SelectedIndices)
                {
                    indices.Add(index.ToString());
                }
                return string.Join(",", indices.ToArray());
            }
            set
            {
                m_SelectedIndices.Clear();
                if (!string.IsNullOrEmpty(value))
                {
                    var indices = value.Split(',');
                    foreach (var indexStr in indices)
                    {
                        if (int.TryParse(indexStr.Trim(), out int index) && index >= 0)
                        {
                            m_SelectedIndices.Add(index);
                        }
                    }
                }
            }
        }

        [UxmlAttribute]
        public bool value
        {
            get => m_IsEnabled;
            set
            {
                if (m_IsEnabled == value)
                    return;
                m_IsEnabled = value;
                UpdateEnabledState();
            }
        }

        /// <summary>Get the currently selected option text (first selected)</summary>
        public string selectedOption
        {
            get
            {
                foreach (int index in m_SelectedIndices)
                {
                    if (index < m_Options.Count)
                        return m_Options[index];
                }
                return "";
            }
        }

        /// <summary>Get all selected options</summary>
        public string[] selectedOptions
        {
            get
            {
                var result = new List<string>();
                foreach (int index in m_SelectedIndices)
                {
                    if (index < m_Options.Count)
                        result.Add(m_Options[index]);
                }
                return result.ToArray();
            }
        }

        /// <summary>Get all selected indices</summary>
        public int[] GetSelectedIndices()
        {
            var result = new int[m_SelectedIndices.Count];
            int i = 0;
            foreach (int index in m_SelectedIndices)
            {
                result[i++] = index;
            }
            return result;
        }

        /// <summary>Event fired when selection changes</summary>
        public event System.Action<int[]> selectionChanged;

        /// <summary>Event fired when toggle state changes</summary>
        public event System.Action<bool> toggleChanged;

        /// <summary>Constructor</summary>
        public ToggleDropdown()
        {
            styleSheets.Add(AssetDatabase.LoadAssetAtPath<StyleSheet>(k_StylesheetPathFormat));
            AddToClassList(k_MainClass);

            RegisterCallback<AttachToPanelEvent>(DelayedInit);

            m_ToggleButton = new Button(OnToggleClicked);
            m_ToggleButton.AddToClassList(k_ToggleButtonClass);
            Add(m_ToggleButton);

            m_DropdownButton = new Button(ShowCustomDropdown);
            m_DropdownButton.AddToClassList(k_DropdownButtonClass);

            m_DropdownArrow = new VisualElement();
            m_DropdownArrow.AddToClassList("unity-toolbar-menu__arrow");
            m_DropdownButton.Add(m_DropdownArrow);

            m_Separator = new VisualElement();
            m_Separator.AddToClassList(k_SeparatorClass);
            Add(m_Separator);

            Add(m_DropdownButton);
        }

        void DelayedInit(AttachToPanelEvent evt)
        {
            if (!string.IsNullOrEmpty(options))
            {
                var optionArray = options.Split(',');
                for (int i = 0; i < optionArray.Length; i++)
                {
                    optionArray[i] = optionArray[i].Trim();
                }
                SetOptions(optionArray);
            }

            if (selectedIndex >= 0)
            {
                m_SelectedIndices.Add(selectedIndex);
            }

            m_IsEnabled = value;
            m_ToggleButton.text = m_Text;
            UpdateEnabledState();
        }

        /// <summary>Set the available options for the dropdown</summary>
        public void SetOptions(string[] newOptions)
        {
            m_Options.Clear();
            if (newOptions != null)
                m_Options.AddRange(newOptions);

            if (m_SelectedIndex >= m_Options.Count)
                m_SelectedIndex = 0;
        }

        /// <summary>Set the selected indices for multi-select</summary>
        public void SetSelectedIndices(int[] indices)
        {
            m_SelectedIndices.Clear();
            if (indices != null)
            {
                foreach (int index in indices)
                {
                    if (index >= 0 && index < m_Options.Count)
                    {
                        m_SelectedIndices.Add(index);
                    }
                }
            }
            selectionChanged?.Invoke(GetSelectedIndices());
        }

        /// <summary>Toggle selection of a specific index</summary>
        public void ToggleSelection(int index)
        {
            if (index >= 0 && index < m_Options.Count)
            {
                if (m_SelectedIndices.Contains(index))
                {
                    m_SelectedIndices.Remove(index);
                }
                else
                {
                    m_SelectedIndices.Add(index);
                }
                selectionChanged?.Invoke(GetSelectedIndices());
            }
        }

        /// <summary>Check if an index is selected</summary>
        public bool IsSelected(int index)
        {
            return m_SelectedIndices.Contains(index);
        }

        /// <summary>Set the enabled state</summary>
        public new void SetEnabled(bool enabled)
        {
            if (enabled != m_IsEnabled)
            {
                m_IsEnabled = enabled;
                UpdateEnabledState();
                toggleChanged?.Invoke(enabled);
            }
        }

        void OnToggleClicked()
        {
            SetEnabled(!m_IsEnabled);
        }

        void UpdateEnabledState()
        {
            if (m_IsEnabled)
            {
                AddToClassList(k_EnabledClass);
            }
            else
            {
                RemoveFromClassList(k_EnabledClass);
            }

            MarkDirtyRepaint();
        }

        void ShowCustomDropdown()
        {
            var menu = new GenericMenu();

            for (int i = 0; i < m_Options.Count; i++)
            {
                int index = i;
                string optionName = m_Options[i];
                bool isSelected = m_SelectedIndices.Contains(index);

                menu.AddItem(new GUIContent(optionName), isSelected, () => {
                    ToggleSelection(index);
                });
            }

            var rect = this.worldBound;
            menu.DropDown(rect);
        }
    }
}