summaryrefslogtreecommitdiff
path: root/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs
diff options
context:
space:
mode:
Diffstat (limited to 'Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs')
-rw-r--r--Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs98
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 ) );
+ }
+ }
+}