From 917e9e0b320775634dc2e710f7deac74fd0822f0 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Thu, 22 Oct 2020 23:30:02 +0800
Subject: * amplify shader editor

---
 .../Editor/Nodes/ImageEffects/PosterizeNode.cs     | 98 ++++++++++++++++++++++
 1 file changed, 98 insertions(+)
 create mode 100644 Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs

(limited to 'Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs')

diff --git a/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs
new file mode 100644
index 00000000..9518f64e
--- /dev/null
+++ b/Assets/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 ) );
+		}
+	}
+}
-- 
cgit v1.1-26-g67d0