60 lines
1.7 KiB
C#
60 lines
1.7 KiB
C#
|
using System;
|
||
|
|
using UnityEditor.Graphing;
|
||
|
|
using UnityEditor.ShaderGraph.Drawing.Slots;
|
||
|
|
using UnityEditor.ShaderGraph.Internal;
|
||
|
|
using UnityEngine;
|
||
|
|
using UnityEngine.UIElements;
|
||
|
|
|
||
|
|
namespace UnityEditor.ShaderGraph
|
||
|
|
{
|
||
|
|
[Serializable]
|
||
|
|
class UVMaterialSlot : Vector2MaterialSlot, IMayRequireMeshUV
|
||
|
|
{
|
||
|
|
[SerializeField]
|
||
|
|
UVChannel m_Channel = UVChannel.UV0;
|
||
|
|
|
||
|
|
public UVChannel channel
|
||
|
|
{
|
||
|
|
get { return m_Channel; }
|
||
|
|
set { m_Channel = value; }
|
||
|
|
}
|
||
|
|
|
||
|
|
public override bool isDefaultValue => channel == UVChannel.UV0;
|
||
|
|
|
||
|
|
public UVMaterialSlot()
|
||
|
|
{ }
|
||
|
|
|
||
|
|
public UVMaterialSlot(int slotId, string displayName, string shaderOutputName, UVChannel channel,
|
||
|
|
ShaderStageCapability stageCapability = ShaderStageCapability.All, bool hidden = false)
|
||
|
|
: base(slotId, displayName, shaderOutputName, SlotType.Input, Vector2.zero, stageCapability, hidden: hidden)
|
||
|
|
{
|
||
|
|
this.channel = channel;
|
||
|
|
}
|
||
|
|
|
||
|
|
public override VisualElement InstantiateControl()
|
||
|
|
{
|
||
|
|
return new UVSlotControlView(this);
|
||
|
|
}
|
||
|
|
|
||
|
|
public override string GetDefaultValue(GenerationMode generationMode)
|
||
|
|
{
|
||
|
|
return string.Format("IN.{0}.xy", channel.GetUVName());
|
||
|
|
}
|
||
|
|
|
||
|
|
public bool RequiresMeshUV(UVChannel channel, ShaderStageCapability stageCapability)
|
||
|
|
{
|
||
|
|
if (isConnected)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
return m_Channel == channel;
|
||
|
|
}
|
||
|
|
|
||
|
|
public override void CopyValuesFrom(MaterialSlot foundSlot)
|
||
|
|
{
|
||
|
|
var slot = foundSlot as UVMaterialSlot;
|
||
|
|
if (slot != null)
|
||
|
|
channel = slot.channel;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|