You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
2.9 KiB
83 lines
2.9 KiB
using System; |
|
using System.Reflection; |
|
using UnityEngine; |
|
using UnityEditor.Graphing; |
|
|
|
using UnityEngine.UIElements; |
|
|
|
namespace UnityEditor.ShaderGraph.Drawing.Controls |
|
{ |
|
[AttributeUsage(AttributeTargets.Property)] |
|
class ChannelEnumMaskControlAttribute : Attribute, IControlAttribute |
|
{ |
|
string m_Label; |
|
int m_SlotId; |
|
|
|
public ChannelEnumMaskControlAttribute(string label = null, int slotId = 0) |
|
{ |
|
m_Label = label; |
|
m_SlotId = slotId; |
|
} |
|
|
|
public VisualElement InstantiateControl(AbstractMaterialNode node, PropertyInfo propertyInfo) |
|
{ |
|
return new ChannelEnumMaskControlView(m_Label, m_SlotId, node, propertyInfo); |
|
} |
|
} |
|
|
|
class ChannelEnumMaskControlView : VisualElement, AbstractMaterialNodeModificationListener |
|
{ |
|
string m_Label; |
|
AbstractMaterialNode m_Node; |
|
PropertyInfo m_PropertyInfo; |
|
IMGUIContainer m_Container; |
|
int m_SlotId; |
|
|
|
public ChannelEnumMaskControlView(string label, int slotId, AbstractMaterialNode node, PropertyInfo propertyInfo) |
|
{ |
|
styleSheets.Add(Resources.Load<StyleSheet>("Styles/Controls/ChannelEnumMaskControlView")); |
|
m_Node = node; |
|
m_PropertyInfo = propertyInfo; |
|
m_SlotId = slotId; |
|
//if (!propertyInfo.PropertyType.IsEnum) |
|
//throw new ArgumentException("Property must be an enum.", "propertyInfo"); |
|
m_Label = label; |
|
m_Container = new IMGUIContainer(OnGUIHandler); |
|
Add(m_Container); |
|
} |
|
|
|
void OnGUIHandler() |
|
{ |
|
GUILayout.BeginHorizontal(); |
|
GUILayout.Label(m_Label); |
|
UpdatePopup(); |
|
GUILayout.EndHorizontal(); |
|
} |
|
|
|
public void OnNodeModified(ModificationScope scope) |
|
{ |
|
if (scope == ModificationScope.Graph) |
|
m_Container.MarkDirtyRepaint(); |
|
} |
|
|
|
private void UpdatePopup() |
|
{ |
|
var value = (int)m_PropertyInfo.GetValue(m_Node, null); |
|
using (var changeCheckScope = new EditorGUI.ChangeCheckScope()) |
|
{ |
|
int channelCount = SlotValueHelper.GetChannelCount(m_Node.FindSlot<MaterialSlot>(m_SlotId).concreteValueType); |
|
string[] enumEntryNames = Enum.GetNames(typeof(TextureChannel)); |
|
string[] popupEntries = new string[channelCount]; |
|
for (int i = 0; i < popupEntries.Length; i++) |
|
popupEntries[i] = enumEntryNames[i]; |
|
value = EditorGUILayout.MaskField("", value, popupEntries, GUILayout.Width(80f)); |
|
|
|
if (changeCheckScope.changed) |
|
{ |
|
m_Node.owner.owner.RegisterCompleteObjectUndo("Change " + m_Node.name); |
|
m_PropertyInfo.SetValue(m_Node, value, null); |
|
} |
|
} |
|
} |
|
} |
|
}
|
|
|