using System; using System.Linq; using System.Reflection; using UnityEditor.UIElements; using UnityEngine; using UnityEngine.UIElements; namespace UnityEditor.ShaderGraph.Drawing.Controls { interface IEnumConversion { Enum from { get; set; } Enum to { get; set; } } [AttributeUsage(AttributeTargets.Property)] class EnumConversionControlAttribute : Attribute, IControlAttribute { public VisualElement InstantiateControl(AbstractMaterialNode node, PropertyInfo propertyInfo) { return new EnumConversionControlView(node, propertyInfo); } } class EnumConversionControlView : VisualElement { AbstractMaterialNode m_Node; PropertyInfo m_PropertyInfo; IEnumConversion value { get { return (IEnumConversion)m_PropertyInfo.GetValue(m_Node, null); } set { m_PropertyInfo.SetValue(m_Node, value, null); } } public EnumConversionControlView(AbstractMaterialNode node, PropertyInfo propertyInfo) { if (!propertyInfo.PropertyType.GetInterfaces().Any(t => t == typeof(IEnumConversion))) throw new ArgumentException("Property type must implement IEnumConversion."); m_Node = node; m_PropertyInfo = propertyInfo; styleSheets.Add(Resources.Load("Styles/Controls/EnumConversionControlView")); var currentValue = value; var ec = (IEnumConversion)propertyInfo.GetValue(m_Node, null); propertyInfo.SetValue(m_Node, ec, null); var fromField = new EnumField(currentValue.from); fromField.RegisterValueChangedCallback(OnFromChanged); Add(fromField); var arrowLabel = new Label("➔"); Add(arrowLabel); var toField = new EnumField(currentValue.to); toField.RegisterValueChangedCallback(OnToChanged); Add(toField); } void OnFromChanged(ChangeEvent evt) { var currentValue = value; if (!Equals(currentValue.from, evt.newValue)) { m_Node.owner.owner.RegisterCompleteObjectUndo("Change Colorspace From"); currentValue.from = evt.newValue; value = currentValue; } } void OnToChanged(ChangeEvent evt) { var currentValue = value; if (!Equals(currentValue.to, evt.newValue)) { m_Node.owner.owner.RegisterCompleteObjectUndo("Change Colorspace To"); currentValue.to = evt.newValue; value = currentValue; } } } }