* amplify shader editor

1 files changed, 131 insertions, 0 deletions

diff --git a/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/VectorFromMatrixNode.cs b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/VectorFromMatrixNode.cs
new file mode 100644
index 00000000..1095df44
--- /dev/null
+++ b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/VectorFromMatrixNode.cs
@@ -0,0 +1,131 @@
+using System;
+using UnityEngine;
+using UnityEditor;
+
+namespace AmplifyShaderEditor
+{
+	public enum eVectorFromMatrixMode
+	{
+		Row,
+		Column
+	}
+
+	[Serializable]
+	[NodeAttributes( "Vector From Matrix", "Matrix Operators", "Retrieve vector data from a matrix" )]
+	public sealed class VectorFromMatrixNode : ParentNode
+	{
+		private const string IndexStr = "Index";
+		private const string ModeStr = "Mode";
+
+		[SerializeField]
+		private eVectorFromMatrixMode m_mode = eVectorFromMatrixMode.Row;
+
+		[SerializeField]
+		private int m_index = 0;
+
+		[SerializeField]
+		private int m_maxIndex = 3;
+
+		protected override void CommonInit( int uniqueId )
+		{
+			base.CommonInit( uniqueId );
+			AddInputPort( WirePortDataType.FLOAT4x4, false, Constants.EmptyPortValue );
+			m_inputPorts[ 0 ].CreatePortRestrictions( WirePortDataType.FLOAT3x3, WirePortDataType.FLOAT4x4 );
+			AddOutputVectorPorts( WirePortDataType.FLOAT4, "XYZW" );
+			m_useInternalPortData = true;
+			m_autoWrapProperties = true;
+		}
+		public override void OnInputPortConnected( int portId, int otherNodeId, int otherPortId, bool activateNode = true )
+		{
+			base.OnInputPortConnected( portId, otherNodeId, otherPortId, activateNode );
+			UpdatePorts();
+		}
+
+		public override void OnConnectedOutputNodeChanges( int inputPortId, int otherNodeId, int otherPortId, string name, WirePortDataType type )
+		{
+			base.OnConnectedOutputNodeChanges( inputPortId, otherNodeId, otherPortId, name, type );
+			UpdatePorts();
+		}
+
+		void UpdatePorts()
+		{
+			m_inputPorts[ 0 ].MatchPortToConnection();
+
+			if ( m_inputPorts[ 0 ].DataType == WirePortDataType.FLOAT3x3 )
+			{
+				m_outputPorts[ 0 ].ChangeType( WirePortDataType.FLOAT3, false );
+				m_outputPorts[ 0 ].Name = "XYZ";
+				m_maxIndex = 2;
+				m_outputPorts[ 4 ].Visible = false;
+			}
+			else
+			{
+				m_outputPorts[ 0 ].ChangeType( WirePortDataType.FLOAT4, false );
+				m_outputPorts[ 0 ].Name = "XYZW";
+				m_maxIndex = 3;
+				m_outputPorts[ 4 ].Visible = true;
+			}
+			m_sizeIsDirty = true;
+		}
+
+		public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
+		{
+			string value = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
+			if ( m_inputPorts[ 0 ].DataType != WirePortDataType.FLOAT4x4 &&
+				m_inputPorts[ 0 ].DataType != WirePortDataType.FLOAT3x3 )
+			{
+				value = UIUtils.CastPortType( ref dataCollector, CurrentPrecisionType, new NodeCastInfo( UniqueId, outputId ), value, m_inputPorts[ 0 ].DataType, WirePortDataType.FLOAT4x4, value );
+			}
+			if ( m_mode == eVectorFromMatrixMode.Row )
+			{
+				value += "[" + m_index + "]";
+			}
+			else
+			{
+				string formatStr = value + "[{0}]" + "[" + m_index + "]";
+				int count = 4;
+				if ( m_inputPorts[ 0 ].DataType != WirePortDataType.FLOAT3x3 )
+				{
+					value = "float4( ";
+				}
+				else
+				{
+					count = 3;
+					value = "float3( ";
+				}
+
+				for ( int i = 0; i < count; i++ )
+				{
+					value += string.Format( formatStr, i );
+					if ( i != ( count - 1 ) )
+					{
+						value += ",";
+					}
+				}
+				value += " )";
+			}
+			return GetOutputVectorItem( 0, outputId, value );
+		}
+
+		public override void DrawProperties()
+		{
+			m_mode = (eVectorFromMatrixMode)EditorGUILayoutEnumPopup( ModeStr, m_mode );
+			m_index = EditorGUILayoutIntSlider( IndexStr, m_index, 0, m_maxIndex );
+			base.DrawProperties();
+		}
+
+		public override void WriteToString( ref string nodeInfo, ref string connectionsInfo )
+		{
+			base.WriteToString( ref nodeInfo, ref connectionsInfo );
+			IOUtils.AddFieldValueToString( ref nodeInfo, m_mode );
+			IOUtils.AddFieldValueToString( ref nodeInfo, m_index );
+		}
+
+		public override void ReadFromString( ref string[] nodeParams )
+		{
+			base.ReadFromString( ref nodeParams );
+			m_mode = ( eVectorFromMatrixMode ) Enum.Parse( typeof( eVectorFromMatrixMode ), GetCurrentParam( ref nodeParams ) );
+			m_index = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
+		}
+	}
+}