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.
129 lines
4.1 KiB
129 lines
4.1 KiB
using System; |
|
using UnityEditor.Graphing; |
|
using UnityEditor.ShaderGraph.Drawing.Slots; |
|
using UnityEditor.ShaderGraph.Internal; |
|
using UnityEngine; |
|
|
|
using UnityEngine.UIElements; |
|
|
|
namespace UnityEditor.ShaderGraph |
|
{ |
|
[Serializable] |
|
class DynamicMatrixMaterialSlot : MaterialSlot, IMaterialSlotHasValue<Matrix4x4> |
|
{ |
|
[SerializeField] |
|
private Matrix4x4 m_Value = Matrix4x4.identity; |
|
|
|
[SerializeField] |
|
private Matrix4x4 m_DefaultValue = Matrix4x4.identity; |
|
|
|
private ConcreteSlotValueType m_ConcreteValueType = ConcreteSlotValueType.Matrix4; |
|
|
|
public DynamicMatrixMaterialSlot() |
|
{ |
|
} |
|
|
|
public DynamicMatrixMaterialSlot( |
|
int slotId, |
|
string displayName, |
|
string shaderOutputName, |
|
SlotType slotType, |
|
ShaderStageCapability stageCapability = ShaderStageCapability.All, |
|
bool hidden = false) |
|
: base(slotId, displayName, shaderOutputName, slotType, stageCapability, hidden) |
|
{ |
|
m_Value = value; |
|
} |
|
|
|
public override VisualElement InstantiateControl() |
|
{ |
|
return new LabelSlotControlView("Identity"); |
|
} |
|
|
|
public Matrix4x4 defaultValue { get { return m_DefaultValue; } } |
|
|
|
public Matrix4x4 value |
|
{ |
|
get { return m_Value; } |
|
set { m_Value = value; } |
|
} |
|
|
|
public override bool isDefaultValue => value.Equals(defaultValue); |
|
|
|
public override SlotValueType valueType { get { return SlotValueType.DynamicMatrix; } } |
|
|
|
public override ConcreteSlotValueType concreteValueType |
|
{ |
|
get { return m_ConcreteValueType; } |
|
} |
|
|
|
public void SetConcreteType(ConcreteSlotValueType valueType) |
|
{ |
|
m_ConcreteValueType = valueType; |
|
} |
|
|
|
protected override string ConcreteSlotValueAsVariable() |
|
{ |
|
var channelCount = (int)SlotValueHelper.GetMatrixDimension(concreteValueType); |
|
var values = ""; |
|
bool isFirst = true; |
|
for (var r = 0; r < channelCount; r++) |
|
{ |
|
for (var c = 0; c < channelCount; c++) |
|
{ |
|
if (!isFirst) |
|
values += ", "; |
|
isFirst = false; |
|
values += value.GetRow(r)[c]; |
|
} |
|
} |
|
return string.Format("$precision{0}x{0}({1})", channelCount, values); |
|
} |
|
|
|
public override void AddDefaultProperty(PropertyCollector properties, GenerationMode generationMode) |
|
{ |
|
if (!generationMode.IsPreview()) |
|
return; |
|
|
|
var matOwner = owner as AbstractMaterialNode; |
|
if (matOwner == null) |
|
throw new Exception(string.Format("Slot {0} either has no owner, or the owner is not a {1}", this, typeof(AbstractMaterialNode))); |
|
|
|
AbstractShaderProperty property; |
|
switch (concreteValueType) |
|
{ |
|
case ConcreteSlotValueType.Matrix4: |
|
property = new Matrix4ShaderProperty(); |
|
break; |
|
case ConcreteSlotValueType.Matrix3: |
|
property = new Matrix3ShaderProperty(); |
|
break; |
|
case ConcreteSlotValueType.Matrix2: |
|
property = new Matrix2ShaderProperty(); |
|
break; |
|
default: |
|
throw new ArgumentOutOfRangeException(); |
|
} |
|
|
|
property.overrideReferenceName = matOwner.GetVariableNameForSlot(id); |
|
property.generatePropertyBlock = false; |
|
properties.AddShaderProperty(property); |
|
} |
|
|
|
public override void CopyValuesFrom(MaterialSlot foundSlot) |
|
{ |
|
var slot = foundSlot as DynamicMatrixMaterialSlot; |
|
if (slot != null) |
|
value = slot.value; |
|
} |
|
|
|
public override void CopyDefaultValue(MaterialSlot other) |
|
{ |
|
base.CopyDefaultValue(other); |
|
if (other is IMaterialSlotHasValue<Matrix4x4> ms) |
|
{ |
|
m_DefaultValue = ms.defaultValue; |
|
} |
|
} |
|
} |
|
}
|
|
|