diff options
Diffstat (limited to 'Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs')
-rw-r--r-- | Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs | 98 |
1 files changed, 98 insertions, 0 deletions
diff --git a/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs b/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs new file mode 100644 index 00000000..9518f64e --- /dev/null +++ b/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs @@ -0,0 +1,98 @@ +// Amplify Shader Editor - Visual Shader Editing Tool +// Copyright (c) Amplify Creations, Lda <info@amplify.pt> + +//https://www.shadertoy.com/view/ldX3D4 +using UnityEngine; +using UnityEditor; +using System; + +namespace AmplifyShaderEditor +{ + [Serializable] + [NodeAttributes( "Posterize", "Image Effects", "Converts a continuous gradation of tones to multiple regions of fewer tones" )] + public sealed class PosterizeNode : ParentNode + { + private const string PosterizationPowerStr = "Power"; + [SerializeField] + private int m_posterizationPower = 1; + + protected override void CommonInit( int uniqueId ) + { + base.CommonInit( uniqueId ); + AddInputPort( WirePortDataType.COLOR, false, "RGBA", -1, MasterNodePortCategory.Fragment, 1 ); + AddInputPort( WirePortDataType.INT, false, "Power", -1, MasterNodePortCategory.Fragment, 0 ); + m_inputPorts[ 1 ].AutoDrawInternalData = true; + AddOutputPort( WirePortDataType.COLOR, Constants.EmptyPortValue ); + m_textLabelWidth = 60; + m_autoWrapProperties = true; + m_previewShaderGUID = "ecb3048ef0eec1645bad1d72a98d8279"; + } + + public override void DrawProperties() + { + base.DrawProperties(); + if( !m_inputPorts[ 1 ].IsConnected ) + { + EditorGUILayout.BeginVertical(); + { + EditorGUI.BeginChangeCheck(); + m_posterizationPower = EditorGUILayoutIntSlider( PosterizationPowerStr, m_posterizationPower, 1, 256 ); + if( EditorGUI.EndChangeCheck() ) + { + GetInputPortByUniqueId( 0 ).IntInternalData = m_posterizationPower; + } + } + EditorGUILayout.EndVertical(); + } + else + { + EditorGUILayout.Space(); + } + } + + public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalVar ) + { + if( m_outputPorts[ 0 ].IsLocalValue( dataCollector.PortCategory ) ) + return m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory ); + + string posterizationPower = "1"; + if( m_inputPorts[ 1 ].IsConnected ) + { + posterizationPower = m_inputPorts[ 1 ].GeneratePortInstructions( ref dataCollector ); + } + else + { + posterizationPower = m_posterizationPower.ToString(); + } + + string colorTarget = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector ); + + string divVar = "div" + OutputId; + dataCollector.AddLocalVariable( UniqueId, "float " + divVar + "=256.0/float(" + posterizationPower + ");" ); + string result = "( floor( " + colorTarget + " * " + divVar + " ) / " + divVar + " )"; + + RegisterLocalVariable( 0, result, ref dataCollector, "posterize" + OutputId ); + + return m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory ); + } + + public override void RefreshExternalReferences() + { + base.RefreshExternalReferences(); + m_inputPorts[ 0 ].ChangeType( WirePortDataType.COLOR, false ); + m_inputPorts[ 1 ].ChangeType( WirePortDataType.INT, false ); + } + + public override void WriteToString( ref string nodeInfo, ref string connectionsInfo ) + { + base.WriteToString( ref nodeInfo, ref connectionsInfo ); + IOUtils.AddFieldValueToString( ref nodeInfo, m_posterizationPower ); + } + + public override void ReadFromString( ref string[] nodeParams ) + { + base.ReadFromString( ref nodeParams ); + m_posterizationPower = Convert.ToInt32( GetCurrentParam( ref nodeParams ) ); + } + } +} |