From b82da95b5181ac8bbae38efb13e950d5e88a4caa Mon Sep 17 00:00:00 2001 From: chai Date: Fri, 23 Oct 2020 13:08:43 +0800 Subject: =?UTF-8?q?*=E7=A7=BB=E5=8A=A8amplify=20shader=20editor=E5=88=B0th?= =?UTF-8?q?ird=20party=E7=9B=AE=E5=BD=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../Editor/Nodes/ImageEffects/PosterizeNode.cs | 98 ++++++++++++++++++++++ 1 file changed, 98 insertions(+) create mode 100644 Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs (limited to 'Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/ImageEffects/PosterizeNode.cs') 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 + +//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