From 917e9e0b320775634dc2e710f7deac74fd0822f0 Mon Sep 17 00:00:00 2001 From: chai Date: Thu, 22 Oct 2020 23:30:02 +0800 Subject: * amplify shader editor --- .../Plugins/Editor/Nodes/Misc/WeightedBlendNode.cs | 58 ++++++++++++++++++++++ 1 file changed, 58 insertions(+) create mode 100644 Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/WeightedBlendNode.cs (limited to 'Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/WeightedBlendNode.cs') diff --git a/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/WeightedBlendNode.cs b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/WeightedBlendNode.cs new file mode 100644 index 00000000..dc5fc1b0 --- /dev/null +++ b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/WeightedBlendNode.cs @@ -0,0 +1,58 @@ +// Amplify Shader Editor - Visual Shader Editing Tool +// Copyright (c) Amplify Creations, Lda + +using UnityEngine; +using System; +using UnityEditor; + +namespace AmplifyShaderEditor +{ + [Serializable] + [NodeAttributes( "Weighted Blend", "Miscellaneous", "Mix all channels through weighted average sum", null, KeyCode.None, true )] + public sealed class WeightedBlendNode : WeightedAvgNode + { + protected override void CommonInit( int uniqueId ) + { + base.CommonInit( uniqueId ); + m_inputData = new string[ 6 ]; + m_previewShaderGUID = "6076cbeaa41ebb14c85ff81b58df7d88"; + } + + 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 ); + + GetInputData( ref dataCollector, ignoreLocalvar ); + + string result = string.Empty; + string avgSum = string.Empty; + + string localVarName = "weightedBlendVar" + OutputId; + dataCollector.AddLocalVariable( UniqueId, CurrentPrecisionType, m_inputPorts[ 0 ].DataType, localVarName, m_inputData[ 0 ] ); + + if ( m_activeCount < 2 ) + { + return CreateOutputLocalVariable( 0, m_inputData[ 1 ], ref dataCollector ); + } + else + { + for ( int i = 0; i < m_activeCount; i++ ) + { + result += localVarName + Constants.VectorSuffixes[ i ] + "*" + m_inputData[ i + 1 ]; + avgSum += localVarName + Constants.VectorSuffixes[ i ]; + if ( i != ( m_activeCount - 1 ) ) + { + result += " + "; + avgSum += " + "; + } + } + } + + result = UIUtils.AddBrackets( result ) + "/" + UIUtils.AddBrackets( avgSum ); + result = UIUtils.AddBrackets( result ); + RegisterLocalVariable( 0, result, ref dataCollector, "weightedAvg" + OutputId ); + return m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory ); + } + } +} -- cgit v1.1-26-g67d0