ScriptableRenderPipeline/MaterialGraphProject/Assets/UnityShaderEditor/Editor/Drawing/Views/TitleBarButtonView.cs

using UnityEditor.Experimental.UIElements.GraphView;
using UnityEngine;
using UnityEngine.Experimental.UIElements;

namespace UnityEditor.MaterialGraph.Drawing
{
    public class TitleBarButtonView : DataWatchContainer
    {
        TitleBarButtonPresenter m_dataProvider;
        Clicker m_clicker;
        VisualElement m_label;

        public TitleBarButtonPresenter dataProvider
        {
            get { return m_dataProvider; }
            set
            {
                if (m_dataProvider == value)
                    return;
                RemoveWatch();
                m_dataProvider = value;
                OnDataChanged();
                AddWatch();
            }
        }

        public TitleBarButtonView(TitleBarButtonPresenter dataProvider)
        {
            AddToClassList("titleBarItem");

            m_clicker = new Clicker();
            m_clicker.onClick += OnClick;
            m_clicker.onStateChange += OnClickStateChanged;
            this.AddManipulator(m_clicker);

            var ve = new VisualElement();
            ve.AddToClassList("titleBarItemBorder");
            Add(ve);


            m_label = new VisualElement();
            m_label.AddToClassList("titleBarItemLabel");
            Add(m_label);

            var ve2 = new VisualElement();
            ve2.AddToClassList("titleBarItemBorder");

            this.dataProvider = dataProvider;
        }

        public override void OnDataChanged()
        {
            if (m_dataProvider == null)
                return;

            m_label.text = m_dataProvider.text;

            this.Dirty(ChangeType.Repaint);
        }

        void OnClick()
        {
            if (m_dataProvider != null && m_dataProvider.onClick != null)
                m_dataProvider.onClick();
        }

        void OnClickStateChanged(ClickerState newState)
        {
            if (newState == ClickerState.Active)
                AddToClassList("active");
            else if (newState == ClickerState.Inactive)
                RemoveFromClassList("active");
            this.Dirty(ChangeType.Repaint);
        }

        protected override Object[] toWatch
        {
            get { return new Object[] {dataProvider}; }
        }
    }
}