74 lines
2.3 KiB
C#
74 lines
2.3 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using UnityEditor.Graphing;
|
|
using UnityEditor.ShaderGraph.Drawing.Slots;
|
|
using UnityEditor.ShaderGraph.Internal;
|
|
using UnityEngine;
|
|
using UnityEngine.UIElements;
|
|
|
|
namespace UnityEditor.ShaderGraph
|
|
{
|
|
[Serializable]
|
|
class Vector1MaterialIntegerSlot : Vector1MaterialSlot
|
|
{
|
|
internal Vector1MaterialIntegerSlot() { }
|
|
|
|
internal Vector1MaterialIntegerSlot(int slotId, Vector1ShaderProperty property) : base(slotId, property) { }
|
|
|
|
[SerializeField]
|
|
bool m_unsigned = false;
|
|
|
|
public Vector1MaterialIntegerSlot(
|
|
int slotId,
|
|
string displayName,
|
|
string shaderOutputName,
|
|
SlotType slotType,
|
|
float value,
|
|
ShaderStageCapability stageCapability = ShaderStageCapability.All,
|
|
bool hidden = false,
|
|
bool unsigned = false)
|
|
: base(slotId, displayName, shaderOutputName, slotType, value, stageCapability: stageCapability, hidden: hidden)
|
|
{
|
|
m_unsigned = unsigned;
|
|
}
|
|
|
|
public override VisualElement InstantiateControl()
|
|
{
|
|
return new IntegerSlotControlView(this);
|
|
}
|
|
|
|
public override void CopyValuesFrom(MaterialSlot foundSlot)
|
|
{
|
|
base.CopyValuesFrom(foundSlot);
|
|
value = Mathf.RoundToInt(value);
|
|
}
|
|
|
|
class IntegerSlotControlView : VisualElement
|
|
{
|
|
Vector1MaterialIntegerSlot m_Slot;
|
|
|
|
public IntegerSlotControlView(Vector1MaterialIntegerSlot slot)
|
|
{
|
|
m_Slot = slot;
|
|
var integerField = slot.hideConnector
|
|
? new IntegerField(slot.RawDisplayName())
|
|
: new IntegerField();
|
|
|
|
integerField.value = (int)slot.value;
|
|
integerField.RegisterValueChangedCallback(OnValueChange);
|
|
Add(integerField);
|
|
}
|
|
|
|
void OnValueChange(ChangeEvent<int> evt)
|
|
{
|
|
if (evt.newValue != m_Slot.value)
|
|
{
|
|
m_Slot.owner.owner.owner.RegisterCompleteObjectUndo("Integer Change");
|
|
m_Slot.value = m_Slot.m_unsigned ? (uint)evt.newValue : (int)evt.newValue;
|
|
m_Slot.owner.Dirty(ModificationScope.Node);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|