mathe/Library/PackageCache/com.unity.shadergraph@14.0.8/Editor/Drawing/Views/PortInputView.cs

130 lines
4.0 KiB
C#
Raw Normal View History

2024-09-20 20:30:10 +02:00
using System;
using UnityEngine;
using UnityEditor.Experimental.GraphView;
using UnityEngine.UIElements;
namespace UnityEditor.ShaderGraph.Drawing
{
class PortInputView : GraphElement, IDisposable
{
readonly CustomStyleProperty<Color> k_EdgeColorProperty = new CustomStyleProperty<Color>("--edge-color");
Color m_EdgeColor = Color.red;
public Color edgeColor
{
get { return m_EdgeColor; }
}
public MaterialSlot slot
{
get { return m_Slot; }
}
MaterialSlot m_Slot;
ConcreteSlotValueType m_SlotType;
VisualElement m_Control;
VisualElement m_Container;
EdgeControl m_EdgeControl;
public PortInputView(MaterialSlot slot)
{
styleSheets.Add(Resources.Load<StyleSheet>("Styles/PortInputView"));
pickingMode = PickingMode.Ignore;
ClearClassList();
m_Slot = slot;
m_SlotType = slot.concreteValueType;
AddToClassList("type" + m_SlotType);
m_EdgeControl = new EdgeControl
{
@from = new Vector2(232f - 21f, 11.5f),
to = new Vector2(232f, 11.5f),
edgeWidth = 2,
pickingMode = PickingMode.Ignore
};
Add(m_EdgeControl);
m_Container = new VisualElement { name = "container" };
{
CreateControl();
var slotElement = new VisualElement { name = "slot" };
{
slotElement.Add(new VisualElement { name = "dot" });
}
m_Container.Add(slotElement);
}
Add(m_Container);
m_Container.Add(new VisualElement() { name = "disabledOverlay", pickingMode = PickingMode.Ignore });
RegisterCallback<CustomStyleResolvedEvent>(OnCustomStyleResolved);
}
void OnCustomStyleResolved(CustomStyleResolvedEvent e)
{
Color colorValue;
if (e.customStyle.TryGetValue(k_EdgeColorProperty, out colorValue))
m_EdgeColor = colorValue;
m_EdgeControl.UpdateLayout();
m_EdgeControl.inputColor = edgeColor;
m_EdgeControl.outputColor = edgeColor;
}
public void UpdateSlot(MaterialSlot newSlot)
{
m_Slot = newSlot;
Recreate();
}
public void UpdateSlotType()
{
if (slot.concreteValueType != m_SlotType)
Recreate();
}
void Recreate()
{
RemoveFromClassList("type" + m_SlotType);
m_SlotType = slot.concreteValueType;
AddToClassList("type" + m_SlotType);
if (m_Control != null)
{
if (m_Control is IDisposable disposable)
disposable.Dispose();
m_Container.Remove(m_Control);
}
CreateControl();
}
void CreateControl()
{
// Specially designated properties (Use Custom Binding) are shown as a label on the slot when the slot is disconnected, with no ability to set an explicit default.
// If the port for this property is connected to, it will use the regular slot control.
m_Control = (!slot.isConnected && slot.IsConnectionTestable()) ? slot.InstantiateCustomControl() : slot.InstantiateControl();
if (m_Control != null)
{
m_Container.Insert(0, m_Control);
}
else
{
// Some slot types don't support an input control, so hide this
m_Container.visible = m_EdgeControl.visible = false;
}
}
public void Dispose()
{
if (m_Control is IDisposable disposable)
disposable.Dispose();
styleSheets.Clear();
m_Control = null;
m_EdgeControl = null;
UnregisterCallback<CustomStyleResolvedEvent>(OnCustomStyleResolved);
}
}
}