1 files changed, 207 insertions, 0 deletions
diff --git a/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Operators/GradientSampleNode.cs b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Operators/GradientSampleNode.cs
new file mode 100644
index 00000000..893521dc
--- /dev/null
+++ b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Operators/GradientSampleNode.cs
@@ -0,0 +1,207 @@
+using System;
+using UnityEngine;
+using UnityEditor;
+using System.Collections;
+using System.Collections.Generic;
+
+namespace AmplifyShaderEditor
+{
+	[Serializable]
+	[NodeAttributes( "Gradient Sample", "Miscellaneous", "Samples a gradient" )]
+	public sealed class GradientSampleNode : ParentNode
+	{
+		private string m_functionHeader = "SampleGradient( {0}, {1} )";
+		private string m_functionBody = string.Empty;
+
+		private GradientNode m_gradientNode = null;
+		private InputPort m_gradPort;
+
+		private Gradient m_blankGrandient = new Gradient();
+
+		private int m_cachedTimeId = -1;
+		private int m_cachedTypeId = -1;
+		private int m_cachedColorNumId = -1;
+		private int m_cachedAlphaNumId = -1;
+
+		private Vector4 m_auxVec4 = Vector4.zero;
+
+		private string m_functionHeaderStruct = "Gradient( {0} )";
+		private string m_functionBodyStruct = string.Empty;
+
+		protected override void CommonInit( int uniqueId )
+		{
+			base.CommonInit( uniqueId );
+			AddInputPort( WirePortDataType.OBJECT, true, Constants.EmptyPortValue );
+			AddInputPort( WirePortDataType.FLOAT, false, "Time" );
+			AddOutputColorPorts( "RGBA", true );
+			m_gradPort = m_inputPorts[ 0 ];
+			m_useInternalPortData = true;
+			m_autoDrawInternalPortData = true;
+			m_drawPreviewExpander = false;
+			m_drawPreview = true;
+			m_showPreview = true;
+			m_selectedLocation = PreviewLocation.TopCenter;
+			m_previewShaderGUID = "8a09124cd6e4aa54a996e7487ec16b90";
+		}
+
+		public override void SetPreviewInputs()
+		{
+			base.SetPreviewInputs();
+
+			if( m_cachedTypeId == -1 )
+				m_cachedTypeId = Shader.PropertyToID( "_GType" );
+
+			if( m_cachedTimeId == -1 )
+				m_cachedTimeId = Shader.PropertyToID( "_GTime" );
+
+			if( m_cachedColorNumId == -1 )
+				m_cachedColorNumId = Shader.PropertyToID( "_GColorNum" );
+
+			if( m_cachedAlphaNumId == -1 )
+				m_cachedAlphaNumId = Shader.PropertyToID( "_GAlphaNum" );
+
+			PreviewMaterial.SetTexture( m_cachedTimeId, m_inputPorts[ 1 ].InputPreviewTexture( ContainerGraph ) );
+
+			Gradient curGrad = m_blankGrandient;
+			if( m_gradientNode != null )
+				curGrad = m_gradientNode.Gradient;
+
+			PreviewMaterial.SetInt( m_cachedTypeId, (int)curGrad.mode );
+			PreviewMaterial.SetInt( m_cachedColorNumId, curGrad.colorKeys.Length );
+			PreviewMaterial.SetInt( m_cachedAlphaNumId, curGrad.alphaKeys.Length );
+
+			for( int i = 0; i < 8; i++ )
+			{
+				if( i < curGrad.colorKeys.Length )
+				{
+					m_auxVec4.x = curGrad.colorKeys[ i ].color.r;
+					m_auxVec4.y = curGrad.colorKeys[ i ].color.g;
+					m_auxVec4.z = curGrad.colorKeys[ i ].color.b;
+					m_auxVec4.w = curGrad.colorKeys[ i ].time;
+					PreviewMaterial.SetVector( "_Col" + i, m_auxVec4 );
+				}
+				else
+				{
+					PreviewMaterial.SetVector( "_Col" + i, Vector4.zero );
+				}
+			}
+
+			for( int i = 0; i < 8; i++ )
+			{
+				if( i < curGrad.alphaKeys.Length )
+				{
+					m_auxVec4.x = curGrad.alphaKeys[ i ].alpha;
+					m_auxVec4.y = curGrad.alphaKeys[ i ].time;
+					PreviewMaterial.SetVector( "_Alp" + i, m_auxVec4 );
+				}
+				else
+				{
+					PreviewMaterial.SetVector( "_Alp" + i, Vector4.zero );
+				}
+			}
+		}
+
+		public override void OnNodeLogicUpdate( DrawInfo drawInfo )
+		{
+			base.OnNodeLogicUpdate( drawInfo );
+
+			if( m_gradPort.IsConnected )
+			{
+				m_gradientNode = RecursiveBackCheck( m_gradPort.GetOutputNode( 0 ) );
+			}
+			else
+			{
+				m_gradientNode = null;
+			}
+		}
+
+		GradientNode RecursiveBackCheck( ParentNode node )
+		{
+			if( node is GradientNode )
+			{
+				return node as GradientNode;
+			}
+			else
+			{
+				if( node is RelayNode || node is WireNode || node is RegisterLocalVarNode )
+				{
+					return RecursiveBackCheck( node.InputPorts[ 0 ].GetOutputNode( 0 ) );
+				}
+				else if( node is GetLocalVarNode)
+				{
+					var gnode = node as GetLocalVarNode;
+					if( gnode.CurrentSelected != null )
+						return RecursiveBackCheck( gnode.CurrentSelected );
+					else
+						return null;
+				}
+				else
+				{
+					return null;
+				}
+			}
+			
+		}
+
+		public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
+		{
+			m_functionBodyStruct = string.Empty;
+			if( !dataCollector.IsSRP )
+			{
+				GradientNode.GenerateGradientStruct( ref m_functionBodyStruct );
+				dataCollector.AddFunctions( m_functionHeaderStruct, m_functionBodyStruct, "0" );
+			}
+			else
+			{
+				dataCollector.AddToIncludes( UniqueId, "Packages/com.unity.shadergraph/ShaderGraphLibrary/Functions.hlsl" );
+			}
+
+			GenerateGradientSampler( dataCollector.IsSRP );
+
+			string gradient = "(Gradient)0";
+			if( m_inputPorts[ 0 ].IsConnected && m_gradientNode != null )
+				gradient = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
+			string time = m_inputPorts[ 1 ].GeneratePortInstructions( ref dataCollector );
+
+			string functionResult =	dataCollector.AddFunctions( m_functionHeader, m_functionBody, gradient, time );
+			return GetOutputVectorItem( 0, outputId, functionResult );
+		}
+
+		void GenerateGradientSampler( bool isSrp )
+		{
+			m_functionBody = string.Empty;
+			IOUtils.AddFunctionHeader( ref m_functionBody, "float4 SampleGradient( Gradient gradient, float time )" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "float3 color = gradient.colors[0].rgb;" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "UNITY_UNROLL" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "for (int c = 1; c < 8; c++)" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "{" );
+			if( isSrp )
+				IOUtils.AddFunctionLine( ref m_functionBody, "float colorPos = saturate((time - gradient.colors[c-1].w) / (gradient.colors[c].w - gradient.colors[c-1].w)) * step(c, gradient.colorsLength-1);" );
+			else
+				IOUtils.AddFunctionLine( ref m_functionBody, "float colorPos = saturate((time - gradient.colors[c-1].w) / (gradient.colors[c].w - gradient.colors[c-1].w)) * step(c, (float)gradient.colorsLength-1);" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "color = lerp(color, gradient.colors[c].rgb, lerp(colorPos, step(0.01, colorPos), gradient.type));" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "}" );
+
+			IOUtils.AddFunctionLine( ref m_functionBody, "#ifndef UNITY_COLORSPACE_GAMMA" );
+			if( isSrp )
+				IOUtils.AddFunctionLine( ref m_functionBody, "color = SRGBToLinear(color);" );
+			else
+				IOUtils.AddFunctionLine( ref m_functionBody, "color = half3(GammaToLinearSpaceExact(color.r), GammaToLinearSpaceExact(color.g), GammaToLinearSpaceExact(color.b));" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "#endif" );
+
+			IOUtils.AddFunctionLine( ref m_functionBody, "float alpha = gradient.alphas[0].x;" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "UNITY_UNROLL" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "for (int a = 1; a < 8; a++)" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "{" );
+			if( isSrp )
+				IOUtils.AddFunctionLine( ref m_functionBody, "float alphaPos = saturate((time - gradient.alphas[a-1].y) / (gradient.alphas[a].y - gradient.alphas[a-1].y)) * step(a, gradient.alphasLength-1);" );
+			else
+				IOUtils.AddFunctionLine( ref m_functionBody, "float alphaPos = saturate((time - gradient.alphas[a-1].y) / (gradient.alphas[a].y - gradient.alphas[a-1].y)) * step(a, (float)gradient.alphasLength-1);" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "alpha = lerp(alpha, gradient.alphas[a].x, lerp(alphaPos, step(0.01, alphaPos), gradient.type));" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "}" );
+			IOUtils.AddFunctionLine( ref m_functionBody, "return float4(color, alpha);" );
+			IOUtils.CloseFunctionBody( ref m_functionBody );
+		}
+
+	}
+}