FateViewer/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/HelperFuncs/LinearToGammaNode.cs

128 lines
4.7 KiB
C#

// Amplify Shader Editor - Visual Shader Editing Tool
// Copyright (c) Amplify Creations, Lda <info@amplify.pt>
using UnityEngine;
using UnityEditor;
using System;
namespace AmplifyShaderEditor
{
[Serializable]
[NodeAttributes( "Linear To Gamma", "Image Effects", "Converts color from linear space to gamma space" )]
public sealed class LinearToGammaNode : HelperParentNode
{
//[SerializeField]
//private bool m_exact = false;
//private readonly static GUIContent LGExactContent = new GUIContent( "Exact Conversion", "Uses a precise version of the conversion, it's more expensive and often not needed." );
public readonly static string[] ModeListStr = { "Fast Linear to sRGB", "Exact Linear to sRGB" };
public readonly static int[] ModeListInt = { 0, 1 };
public readonly static string[] ModeListStrLW = { "Fast Linear to sRGB", "Exact Linear to sRGB", "Linear to Gamma 2.0", "Linear to Gamma 2.2" };
public readonly static int[] ModeListIntLW = { 0, 1, 2, 3 };
[SerializeField]
public int m_selectedMode = 0;
protected override void CommonInit( int uniqueId )
{
base.CommonInit( uniqueId );
m_funcType = "LinearToGammaSpace";
m_inputPorts[ 0 ].ChangeType( WirePortDataType.FLOAT3, false );
m_inputPorts[ 0 ].Name = "RGB";
m_outputPorts[ 0 ].ChangeType( WirePortDataType.FLOAT3, false );
m_autoWrapProperties = true;
m_previewShaderGUID = "9027c408b928c5c4d8b450712049d541";
m_textLabelWidth = 120;
}
protected override void OnUniqueIDAssigned()
{
base.OnUniqueIDAssigned();
m_localVarName = "linearToGamma" + OutputId;
}
public override void DrawProperties()
{
base.DrawProperties();
if( ContainerGraph.IsSRP )
{
m_selectedMode = EditorGUILayoutIntPopup( "Mode", m_selectedMode, ModeListStrLW, ModeListIntLW );
EditorGUILayout.HelpBox( "Fast Linear: fast approximation from Linear to sRGB\n\nExact Linear: a more expensive but exact calculation from Linear to sRGB.\n\nLinear 2.0: crude approximation from Linear to Gamma using a power of 1/2.0 gamma value\n\nLinear 2.2: an approximation from Linear to Gamma using a power of 1/2.2 gamma value", MessageType.None );
}
else
{
m_selectedMode = EditorGUILayoutIntPopup( "Mode", m_selectedMode, ModeListStr, ModeListInt );
EditorGUILayout.HelpBox( "Fast Linear: fast approximation from Linear to sRGB\n\nExact Linear: a more expensive but exact calculation from Linear to sRGB.", MessageType.None );
}
}
public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
{
string result = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
if( !dataCollector.IsSRP )
{
m_selectedMode = Mathf.Min( m_selectedMode, 1 );
if( m_selectedMode == 1 )
{
dataCollector.AddLocalVariable( UniqueId, "half3 " + m_localVarName + " = " + result + ";" );
dataCollector.AddLocalVariable( UniqueId, m_localVarName + " = half3( LinearToGammaSpaceExact(" + m_localVarName + ".r), LinearToGammaSpaceExact(" + m_localVarName + ".g), LinearToGammaSpaceExact(" + m_localVarName + ".b) );" );
return m_localVarName;
}
return base.GenerateShaderForOutput( outputId, ref dataCollector, ignoreLocalvar );
}
else
{
dataCollector.AddToIncludes( UniqueId, TemplateHelperFunctions.CoreCommonLib );
dataCollector.AddToIncludes( UniqueId, TemplateHelperFunctions.CoreColorLib );
switch( m_selectedMode )
{
default:
case 0:
m_funcLWFormatOverride = "FastLinearToSRGB( {0} )";
m_funcHDFormatOverride = "FastLinearToSRGB( {0} )";
break;
case 1:
m_funcLWFormatOverride = "LinearToSRGB( {0} )";
m_funcHDFormatOverride = "LinearToSRGB( {0} )";
break;
case 2:
m_funcLWFormatOverride = "LinearToGamma20( {0} )";
m_funcHDFormatOverride = "LinearToGamma20( {0} )";
break;
case 3:
m_funcLWFormatOverride = "LinearToGamma22( {0} )";
m_funcHDFormatOverride = "LinearToGamma22( {0} )";
break;
}
return base.GenerateShaderForOutput( outputId, ref dataCollector, ignoreLocalvar );
}
}
public override void WriteToString( ref string nodeInfo, ref string connectionsInfo )
{
base.WriteToString( ref nodeInfo, ref connectionsInfo );
IOUtils.AddFieldValueToString( ref nodeInfo, m_selectedMode );
}
public override void ReadFromString( ref string[] nodeParams )
{
base.ReadFromString( ref nodeParams );
if( UIUtils.CurrentShaderVersion() > 11003 && UIUtils.CurrentShaderVersion() <= 14503 )
{
bool fast = Convert.ToBoolean( GetCurrentParam( ref nodeParams ) );
if( fast )
m_selectedMode = 1;
}
if( UIUtils.CurrentShaderVersion() > 14503 )
{
m_selectedMode = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
}
}
}
}