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.
 
 
 

558 lines
18 KiB

// Amplify Shader Editor - Visual Shader Editing Tool
// Copyright (c) Amplify Creations, Lda <info@amplify.pt>
using UnityEngine;
using UnityEditor;
using System;
using System.Collections.Generic;
namespace AmplifyShaderEditor
{
[Serializable]
[NodeAttributes( "Texture Coordinates", "UV Coordinates", "Texture UV coordinates set, if <b>Tex</b> is connected to a texture object it will use that texture scale factors, otherwise uses <b>Tilling</b> and <b>Offset</b> port values", null, KeyCode.U )]
public sealed class TextureCoordinatesNode : ParentNode
{
private const string DummyPropertyDec = "[HideInInspector] _DummyTex{0}( \"\", 2D ) = \"white\"";
private const string DummyUniformDec = "uniform sampler2D _DummyTex{0};";
private const string DummyTexCoordDef = "uv{0}_DummyTex{0}";
private const string DummyTexCoordSurfDef = "float2 texCoordDummy{0} = {1}.uv{2}_DummyTex{2}*{3} + {4};";
private const string DummyTexCoordSurfVar = "texCoordDummy{0}";
private readonly string[] Dummy = { string.Empty };
private const string TilingStr = "Tiling";
private const string OffsetStr = "Offset";
private const string TexCoordStr = "texcoord_";
[SerializeField]
private int m_referenceArrayId = -1;
[SerializeField]
private int m_referenceNodeId = -1;
[SerializeField]
private int m_textureCoordChannel = 0;
//[SerializeField]
//private int m_texcoordId = -1;
[SerializeField]
private int m_texcoordSize = 2;
[SerializeField]
private string m_surfaceTexcoordName = string.Empty;
[SerializeField]
private TexturePropertyNode m_inputReferenceNode = null;
private TexturePropertyNode m_referenceNode = null;
private InputPort m_texPort = null;
private InputPort m_tilingPort = null;
private InputPort m_offsetPort = null;
protected override void CommonInit( int uniqueId )
{
base.CommonInit( uniqueId );
AddInputPort( WirePortDataType.SAMPLER2D, false, "Tex", -1, MasterNodePortCategory.Fragment, 2 );
m_texPort = m_inputPorts[ m_inputPorts.Count - 1 ];
m_texPort.CreatePortRestrictions( WirePortDataType.SAMPLER1D, WirePortDataType.SAMPLER2D, WirePortDataType.SAMPLER3D, WirePortDataType.SAMPLERCUBE, WirePortDataType.OBJECT );
AddInputPort( WirePortDataType.FLOAT2, false, "Tiling", -1, MasterNodePortCategory.Fragment, 0 );
m_tilingPort = m_inputPorts[ m_inputPorts.Count - 1 ];
m_tilingPort.Vector2InternalData = new Vector2( 1, 1 );
AddInputPort( WirePortDataType.FLOAT2, false, "Offset", -1, MasterNodePortCategory.Fragment, 1 );
m_offsetPort = m_inputPorts[ m_inputPorts.Count - 1 ];
AddOutputVectorPorts( WirePortDataType.FLOAT2, "UV" );
m_outputPorts[ 1 ].Name = "U";
m_outputPorts[ 2 ].Name = "V";
AddOutputPort( WirePortDataType.FLOAT, "W" );
AddOutputPort( WirePortDataType.FLOAT, "T" );
m_textLabelWidth = 90;
m_useInternalPortData = true;
m_autoWrapProperties = true;
m_tilingPort.Category = MasterNodePortCategory.Vertex;
m_offsetPort.Category = MasterNodePortCategory.Vertex;
UpdateOutput();
m_previewShaderGUID = "085e462b2de441a42949be0e666cf5d2";
}
public override void Reset()
{
m_surfaceTexcoordName = string.Empty;
}
public override void OnInputPortConnected( int portId, int otherNodeId, int otherPortId, bool activateNode = true )
{
base.OnInputPortConnected( portId, otherNodeId, otherPortId, activateNode );
if( portId == 2 )
{
m_inputReferenceNode = m_texPort.GetOutputNode() as TexturePropertyNode;
UpdatePorts();
}
}
public override void OnInputPortDisconnected( int portId )
{
base.OnInputPortDisconnected( portId );
if( portId == 2 )
{
m_inputReferenceNode = null;
UpdatePorts();
}
}
void UpdateTitle()
{
if( m_inputReferenceNode != null )
{
m_additionalContent.text = string.Format( "Value( {0} )", m_inputReferenceNode.PropertyInspectorName );
}
else if( m_referenceArrayId > -1 && m_referenceNode != null )
{
m_referenceNode = UIUtils.GetTexturePropertyNode( m_referenceArrayId );
m_additionalContent.text = string.Format( "Value( {0} )", m_referenceNode.PropertyInspectorName );
}
else
{
m_additionalContent.text = string.Empty;
}
m_sizeIsDirty = true;
}
void UpdatePorts()
{
if( m_inputReferenceNode != null || m_texPort.IsConnected )
{
m_tilingPort.Locked = true;
m_offsetPort.Locked = true;
}
else if( m_referenceArrayId > -1 )
{
m_tilingPort.Locked = true;
m_offsetPort.Locked = true;
}
else
{
m_tilingPort.Locked = false;
m_offsetPort.Locked = false;
}
}
public override void DrawProperties()
{
bool guiEnabledBuffer = GUI.enabled;
EditorGUI.BeginChangeCheck();
List<string> arr = ( m_inputReferenceNode != null ) ? null : new List<string>( UIUtils.TexturePropertyNodeArr() );
if( arr != null && arr.Count > 0 )
{
arr.Insert( 0, "None" );
GUI.enabled = true;
m_referenceArrayId = EditorGUILayoutPopup( Constants.AvailableReferenceStr, m_referenceArrayId + 1, arr.ToArray() ) - 1;
}
else
{
m_referenceArrayId = -1;
GUI.enabled = false;
EditorGUILayoutPopup( Constants.AvailableReferenceStr, 0, Dummy );
}
GUI.enabled = guiEnabledBuffer;
if( EditorGUI.EndChangeCheck() )
{
m_referenceNode = UIUtils.GetTexturePropertyNode( m_referenceArrayId );
if( m_referenceNode != null )
{
m_referenceNodeId = m_referenceNode.UniqueId;
}
else
{
m_referenceNodeId = -1;
m_referenceArrayId = -1;
}
UpdateTitle();
UpdatePorts();
}
EditorGUI.BeginChangeCheck();
m_texcoordSize = EditorGUILayoutIntPopup( Constants.AvailableUVSizesLabel, m_texcoordSize, Constants.AvailableUVSizesStr, Constants.AvailableUVSizes );
if( EditorGUI.EndChangeCheck() )
{
UpdateOutput();
}
m_textureCoordChannel = EditorGUILayoutIntPopup( Constants.AvailableUVSetsLabel, m_textureCoordChannel, Constants.AvailableUVSetsStr, Constants.AvailableUVSets );
if( m_referenceArrayId > -1 )
GUI.enabled = false;
base.DrawProperties();
GUI.enabled = guiEnabledBuffer;
}
private void UpdateOutput()
{
if( m_texcoordSize == 3 )
{
m_outputPorts[ 0 ].ChangeType( WirePortDataType.FLOAT3, false );
m_outputPorts[ 0 ].Name = "UVW";
m_outputPorts[ 3 ].Visible = true;
m_outputPorts[ 4 ].Visible = false;
}
else if( m_texcoordSize == 4 )
{
m_outputPorts[ 0 ].ChangeType( WirePortDataType.FLOAT4, false );
m_outputPorts[ 0 ].Name = "UVWT";
m_outputPorts[ 3 ].Visible = true;
m_outputPorts[ 4 ].Visible = true;
}
else
{
m_outputPorts[ 0 ].ChangeType( WirePortDataType.FLOAT2, false );
m_outputPorts[ 0 ].Name = "UV";
m_outputPorts[ 3 ].Visible = false;
m_outputPorts[ 4 ].Visible = false;
}
m_sizeIsDirty = true;
}
public override void OnNodeLogicUpdate( DrawInfo drawInfo )
{
base.OnNodeLogicUpdate( drawInfo );
CheckReference();
}
//public override void Draw( DrawInfo drawInfo )
//{
// base.Draw( drawInfo );
// //CheckReference();
//}
void CheckReference()
{
if( m_referenceArrayId > -1 )
{
ParentNode newNode = UIUtils.GetTexturePropertyNode( m_referenceArrayId );
if( newNode == null || newNode.UniqueId != m_referenceNodeId )
{
m_referenceNode = null;
int count = UIUtils.GetTexturePropertyNodeAmount();
for( int i = 0; i < count; i++ )
{
ParentNode node = UIUtils.GetTexturePropertyNode( i );
if( node.UniqueId == m_referenceNodeId )
{
m_referenceNode = node as TexturePropertyNode;
m_referenceArrayId = i;
break;
}
}
}
}
if( m_referenceNode == null && m_referenceNodeId > -1 )
{
m_referenceNodeId = -1;
m_referenceArrayId = -1;
UpdateTitle();
UpdatePorts();
}
}
public override void ReadFromString( ref string[] nodeParams )
{
base.ReadFromString( ref nodeParams );
m_textureCoordChannel = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
if( UIUtils.CurrentShaderVersion() > 2402 )
{
if( UIUtils.CurrentShaderVersion() > 2404 )
{
m_referenceNodeId = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
}
else
{
m_referenceArrayId = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
}
}
if( UIUtils.CurrentShaderVersion() > 5001 )
{
m_texcoordSize = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
UpdateOutput();
}
}
public override void RefreshExternalReferences()
{
base.RefreshExternalReferences();
if( UIUtils.CurrentShaderVersion() > 2402 )
{
if( UIUtils.CurrentShaderVersion() > 2404 )
{
m_referenceNode = UIUtils.GetNode( m_referenceNodeId ) as TexturePropertyNode;
if( m_referenceNodeId > -1 )
m_referenceArrayId = UIUtils.GetTexturePropertyNodeRegisterId( m_referenceNodeId );
}
else
{
m_referenceNode = UIUtils.GetTexturePropertyNode( m_referenceArrayId );
if( m_referenceNode != null )
{
m_referenceNodeId = m_referenceNode.UniqueId;
}
}
UpdateTitle();
UpdatePorts();
}
}
public override void PropagateNodeData( NodeData nodeData, ref MasterNodeDataCollector dataCollector )
{
if( dataCollector != null && dataCollector.TesselationActive )
{
base.PropagateNodeData( nodeData, ref dataCollector );
return;
}
if( dataCollector.IsTemplate )
{
dataCollector.TemplateDataCollectorInstance.SetUVUsage( m_textureCoordChannel, m_texcoordSize );
}
else if( m_textureCoordChannel > 3 )
{
dataCollector.AddCustomAppData( string.Format( TemplateHelperFunctions.TexUVFullSemantic, m_textureCoordChannel ) );
}
UIUtils.SetCategoryInBitArray( ref m_category, nodeData.Category );
MasterNodePortCategory propagateCategory = ( nodeData.Category != MasterNodePortCategory.Vertex && nodeData.Category != MasterNodePortCategory.Tessellation ) ? MasterNodePortCategory.Vertex : nodeData.Category;
nodeData.Category = propagateCategory;
nodeData.GraphDepth += 1;
if( nodeData.GraphDepth > m_graphDepth )
{
m_graphDepth = nodeData.GraphDepth;
}
int count = m_inputPorts.Count;
for( int i = 0; i < count; i++ )
{
if( m_inputPorts[ i ].IsConnected )
{
//m_inputPorts[ i ].GetOutputNode().PropagateNodeCategory( category );
m_inputPorts[ i ].GetOutputNode().PropagateNodeData( nodeData, ref dataCollector );
}
}
}
public override void WriteToString( ref string nodeInfo, ref string connectionsInfo )
{
base.WriteToString( ref nodeInfo, ref connectionsInfo );
IOUtils.AddFieldValueToString( ref nodeInfo, m_textureCoordChannel );
IOUtils.AddFieldValueToString( ref nodeInfo, ( ( m_referenceNode != null ) ? m_referenceNode.UniqueId : -1 ) );
IOUtils.AddFieldValueToString( ref nodeInfo, m_texcoordSize );
}
string GetValidPropertyName()
{
string propertyName = string.Empty;
if( m_inputReferenceNode != null )
{
propertyName = m_inputReferenceNode.PropertyName;
}
else if( m_referenceArrayId > -1 )
{
m_referenceNode = UIUtils.GetTexturePropertyNode( m_referenceArrayId );
if( m_referenceNode != null )
{
propertyName = m_referenceNode.PropertyName;
}
}
return propertyName;
}
public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalVar )
{
if( dataCollector.PortCategory == MasterNodePortCategory.Tessellation )
{
UIUtils.ShowMessage( m_nodeAttribs.Name + " cannot be used on Master Node Tessellation port" );
return "-1";
}
//bool isVertex = ( dataCollector.PortCategory == MasterNodePortCategory.Vertex || dataCollector.PortCategory == MasterNodePortCategory.Tessellation );
string tiling = string.Empty;
string offset = string.Empty;
string portProperty = string.Empty;
if( m_texPort.IsConnected )
portProperty = m_texPort.GeneratePortInstructions( ref dataCollector );
if( m_referenceArrayId > -1 )
{
TexturePropertyNode temp = UIUtils.GetTexturePropertyNode( m_referenceArrayId );
if( temp != null )
{
portProperty = temp.BaseGenerateShaderForOutput( outputId, ref dataCollector, ignoreLocalVar );
}
}
//TEMPLATES
if( dataCollector.MasterNodeCategory == AvailableShaderTypes.Template )
{
if( m_outputPorts[ 0 ].IsLocalValue( dataCollector.PortCategory ) )
return GetOutputVectorItem( 0, outputId, m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory ) );
string uvName = string.Empty;
if( dataCollector.TemplateDataCollectorInstance.HasUV( m_textureCoordChannel ) )
{
uvName = dataCollector.TemplateDataCollectorInstance.GetUVName( m_textureCoordChannel, m_outputPorts[ 0 ].DataType );
}
else
{
uvName = dataCollector.TemplateDataCollectorInstance.RegisterUV( m_textureCoordChannel, m_outputPorts[ 0 ].DataType );
}
string currPropertyName = GetValidPropertyName();
if( !string.IsNullOrEmpty( portProperty ) && portProperty != "0.0" )
{
currPropertyName = portProperty;
}
if( !string.IsNullOrEmpty( currPropertyName ) )
{
string finalTexCoordName = "uv" + currPropertyName;
string dummyPropertyTexcoords = currPropertyName + "_ST";
dataCollector.AddToUniforms( UniqueId, "float4", dummyPropertyTexcoords );
if( m_texcoordSize > 2 )
{
dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, m_outputPorts[ 0 ].DataType, finalTexCoordName, uvName );
dataCollector.AddLocalVariable( UniqueId, finalTexCoordName + ".xy", string.Format( Constants.TilingOffsetFormat, uvName + ".xy", dummyPropertyTexcoords + ".xy", dummyPropertyTexcoords + ".zw" ) + ";" );
m_outputPorts[ 0 ].SetLocalValue( finalTexCoordName, dataCollector.PortCategory );
}
else
{
RegisterLocalVariable( 0, string.Format( Constants.TilingOffsetFormat, uvName, dummyPropertyTexcoords + ".xy", dummyPropertyTexcoords + ".zw" ), ref dataCollector, finalTexCoordName );
}
//RegisterLocalVariable( 0, string.Format( Constants.TilingOffsetFormat, uvName, dummyPropertyTexcoords+".xy", dummyPropertyTexcoords+".zw" ), ref dataCollector, finalTexCoordName );
}
else
{
string finalTexCoordName = "uv" + OutputId;
tiling = m_tilingPort.GeneratePortInstructions( ref dataCollector );
offset = m_offsetPort.GeneratePortInstructions( ref dataCollector );
if( m_texcoordSize > 2 )
{
dataCollector.AddLocalVariable( UniqueId, m_currentPrecisionType, m_outputPorts[ 0 ].DataType, finalTexCoordName, uvName );
dataCollector.AddLocalVariable( UniqueId, finalTexCoordName + ".xy", string.Format( Constants.TilingOffsetFormat, uvName + ".xy", tiling, offset ) + ";" );
m_outputPorts[ 0 ].SetLocalValue( finalTexCoordName, dataCollector.PortCategory );
}
else
{
RegisterLocalVariable( 0, string.Format( Constants.TilingOffsetFormat, uvName, tiling, offset ), ref dataCollector, finalTexCoordName );
}
//RegisterLocalVariable( 0, string.Format( Constants.TilingOffsetFormat, uvName, tiling, offset ), ref dataCollector, finalTexCoordName );
}
return GetOutputVectorItem( 0, outputId, m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory ) );
}
//SURFACE
string propertyName = GetValidPropertyName();
if( !string.IsNullOrEmpty( portProperty ) && portProperty != "0.0" )
{
propertyName = portProperty;
}
if( m_outputPorts[ 0 ].IsLocalValue( dataCollector.PortCategory ) )
return GetOutputVectorItem( 0, outputId, m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory ) );
if( !m_tilingPort.IsConnected && m_tilingPort.Vector2InternalData == Vector2.one )
tiling = null;
else
tiling = m_tilingPort.GeneratePortInstructions( ref dataCollector );
if( !m_offsetPort.IsConnected && m_offsetPort.Vector2InternalData == Vector2.zero )
offset = null;
else
offset = m_offsetPort.GeneratePortInstructions( ref dataCollector );
if( !string.IsNullOrEmpty( propertyName ) /*m_referenceArrayId > -1*/ )
{
m_surfaceTexcoordName = GeneratorUtils.GenerateAutoUVs( ref dataCollector, UniqueId, m_textureCoordChannel, propertyName, m_outputPorts[ 0 ].DataType, tiling, offset, OutputId );
}
else
{
m_surfaceTexcoordName = GeneratorUtils.GenerateAutoUVs( ref dataCollector, UniqueId, m_textureCoordChannel, null, m_outputPorts[ 0 ].DataType, tiling, offset, OutputId );
}
m_outputPorts[ 0 ].SetLocalValue( m_surfaceTexcoordName, dataCollector.PortCategory );
return GetOutputVectorItem( 0, outputId, m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory ) );
}
public override void ReadInputDataFromString( ref string[] nodeParams )
{
if( UIUtils.CurrentShaderVersion() > 7003 )
{
base.ReadInputDataFromString( ref nodeParams );
}
else
{
for( int i = 0; i < 2 && i < nodeParams.Length && m_currentReadParamIdx < nodeParams.Length; i++ )
{
if( UIUtils.CurrentShaderVersion() < 5003 )
{
int newId = VersionConvertInputPortId( i ) + 1;
if( UIUtils.CurrentShaderVersion() > 23 )
{
m_inputPorts[ newId ].DataType = (WirePortDataType)Enum.Parse( typeof( WirePortDataType ), nodeParams[ m_currentReadParamIdx++ ] );
}
m_inputPorts[ newId ].InternalData = nodeParams[ m_currentReadParamIdx++ ];
if( m_inputPorts[ newId ].IsEditable && UIUtils.CurrentShaderVersion() >= 3100 && m_currentReadParamIdx < nodeParams.Length )
{
m_inputPorts[ newId ].Name = nodeParams[ m_currentReadParamIdx++ ];
}
}
else
{
int portId = Convert.ToInt32( nodeParams[ m_currentReadParamIdx++ ] );
WirePortDataType DataType = (WirePortDataType)Enum.Parse( typeof( WirePortDataType ), nodeParams[ m_currentReadParamIdx++ ] );
string InternalData = nodeParams[ m_currentReadParamIdx++ ];
bool isEditable = Convert.ToBoolean( nodeParams[ m_currentReadParamIdx++ ] );
string Name = string.Empty;
if( isEditable && m_currentReadParamIdx < nodeParams.Length )
{
Name = nodeParams[ m_currentReadParamIdx++ ];
}
InputPort inputPort = GetInputPortByUniqueId( portId );
if( inputPort != null )
{
inputPort.DataType = DataType;
inputPort.InternalData = InternalData;
if( !string.IsNullOrEmpty( Name ) )
{
inputPort.Name = Name;
}
}
}
}
}
}
public override void Destroy()
{
base.Destroy();
m_referenceNode = null;
}
}
}