88 lines
3.2 KiB
C#
88 lines
3.2 KiB
C#
using System;
|
|
|
|
namespace AmplifyShaderEditor
|
|
{
|
|
[Serializable]
|
|
[NodeAttributes( "HSV to RGB", "Image Effects", "Converts from HSV to RGB color space" )]
|
|
public sealed class HSVToRGBNode : ParentNode
|
|
{
|
|
public static readonly string HSVToRGBHeader = "HSVToRGB( {0}3({1},{2},{3}) )";
|
|
public static readonly string[] HSVToRGBFunction = { "{0}3 HSVToRGB( {0}3 c )\n",
|
|
"{\n",
|
|
"\t{0}4 K = {0}4( 1.0, 2.0 / 3.0, 1.0 / 3.0, 3.0 );\n",
|
|
"\t{0}3 p = abs( frac( c.xxx + K.xyz ) * 6.0 - K.www );\n",
|
|
"\treturn c.z * lerp( K.xxx, saturate( p - K.xxx ), c.y );\n",
|
|
"}\n"};
|
|
public static readonly bool[] HSVToRGBFlags = { true,
|
|
false,
|
|
true,
|
|
true,
|
|
false,
|
|
false};
|
|
|
|
protected override void CommonInit( int uniqueId )
|
|
{
|
|
base.CommonInit( uniqueId );
|
|
AddInputPort( WirePortDataType.FLOAT, false, "Hue" );
|
|
AddInputPort( WirePortDataType.FLOAT, false, "Saturation" );
|
|
AddInputPort( WirePortDataType.FLOAT, false, "Value" );
|
|
AddOutputColorPorts( "RGB", false );
|
|
m_previewShaderGUID = "fab445eb945d63047822a7a6b81b959d";
|
|
m_useInternalPortData = true;
|
|
m_autoWrapProperties = true;
|
|
m_customPrecision = true;
|
|
}
|
|
|
|
public override void DrawProperties()
|
|
{
|
|
base.DrawProperties();
|
|
DrawPrecisionProperty();
|
|
}
|
|
|
|
public static void AddHSVToRGBFunction( ref MasterNodeDataCollector dataCollector , string precisionString )
|
|
{
|
|
if( !dataCollector.HasFunction( HSVToRGBHeader ) )
|
|
{
|
|
//Hack to be used util indent is properly used
|
|
int currIndent = UIUtils.ShaderIndentLevel;
|
|
if( dataCollector.MasterNodeCategory == AvailableShaderTypes.Template )
|
|
{
|
|
UIUtils.ShaderIndentLevel = 0;
|
|
}
|
|
else
|
|
{
|
|
UIUtils.ShaderIndentLevel = 1;
|
|
UIUtils.ShaderIndentLevel++;
|
|
}
|
|
|
|
string finalFunction = string.Empty;
|
|
for( int i = 0; i < HSVToRGBFunction.Length; i++ )
|
|
{
|
|
finalFunction += UIUtils.ShaderIndentTabs + ( HSVToRGBFlags[ i ] ? string.Format( HSVToRGBFunction[ i ], precisionString ) : HSVToRGBFunction[ i ] );
|
|
}
|
|
|
|
UIUtils.ShaderIndentLevel = currIndent;
|
|
|
|
dataCollector.AddFunction( HSVToRGBHeader, finalFunction );
|
|
}
|
|
}
|
|
|
|
public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
|
|
{
|
|
if ( m_outputPorts[ 0 ].IsLocalValue( dataCollector.PortCategory ) )
|
|
return GetOutputVectorItem( 0, outputId, m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory ) );
|
|
|
|
string precisionString = UIUtils.PrecisionWirePortToCgType( CurrentPrecisionType, WirePortDataType.FLOAT );
|
|
|
|
AddHSVToRGBFunction( ref dataCollector , precisionString );
|
|
|
|
string hue = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
|
|
string saturation = m_inputPorts[ 1 ].GeneratePortInstructions( ref dataCollector );
|
|
string value = m_inputPorts[ 2 ].GeneratePortInstructions( ref dataCollector );
|
|
|
|
RegisterLocalVariable( 0, string.Format( HSVToRGBHeader, precisionString, hue, saturation, value ), ref dataCollector, "hsvTorgb" + OutputId );
|
|
return GetOutputVectorItem( 0, outputId, m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory ) );
|
|
}
|
|
}
|
|
}
|