您最多选择25个主题
主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
118 行
3.5 KiB
118 行
3.5 KiB
using System;
|
|
using UnityEngine;
|
|
using UnityEditor.Experimental.GraphView;
|
|
using UnityEngine.UIElements;
|
|
using UnityEngine.UIElements.StyleSheets;
|
|
|
|
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" };
|
|
{
|
|
m_Control = this.slot.InstantiateControl();
|
|
if (m_Control != null)
|
|
m_Container.Add(m_Control);
|
|
|
|
var slotElement = new VisualElement { name = "slot" };
|
|
{
|
|
slotElement.Add(new VisualElement { name = "dot" });
|
|
}
|
|
m_Container.Add(slotElement);
|
|
}
|
|
Add(m_Container);
|
|
|
|
m_Container.visible = m_EdgeControl.visible = m_Control != null;
|
|
|
|
RegisterCallback<CustomStyleResolvedEvent>(OnCustomStyleResolved);
|
|
}
|
|
|
|
private 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)
|
|
{
|
|
var disposable = m_Control as IDisposable;
|
|
if (disposable != null)
|
|
disposable.Dispose();
|
|
m_Container.Remove(m_Control);
|
|
}
|
|
m_Control = slot.InstantiateControl();
|
|
if (m_Control != null)
|
|
m_Container.Insert(0, m_Control);
|
|
|
|
m_Container.visible = m_EdgeControl.visible = m_Control != null;
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
var disposable = m_Control as IDisposable;
|
|
if (disposable != null)
|
|
disposable.Dispose();
|
|
}
|
|
}
|
|
}
|