1 files changed, 93 insertions, 0 deletions
diff --git a/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Operators/PowerNode.cs b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Operators/PowerNode.cs
new file mode 100644
index 00000000..82357b67
--- /dev/null
+++ b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Operators/PowerNode.cs
@@ -0,0 +1,93 @@
+// Amplify Shader Editor - Visual Shader Editing Tool
+// Copyright (c) Amplify Creations, Lda <info@amplify.pt>
+
+using System;
+using UnityEngine;
+
+namespace AmplifyShaderEditor
+{
+	[Serializable]
+	[NodeAttributes( "Power", "Math Operators", "Base to the Exp-th power of scalars and vectors", null, KeyCode.E )]
+	public sealed class PowerNode : ParentNode
+	{
+		public const string SafePowerStr = "max( {0} , 0.0001 )";
+		public const string SafePowerLabel = "Safe Power";
+
+		[SerializeField]
+		private bool m_safePower = false;
+
+		protected override void CommonInit( int uniqueId )
+		{
+			base.CommonInit( uniqueId );
+			AddInputPort( WirePortDataType.FLOAT, false, "Base" );
+			AddInputPort( WirePortDataType.FLOAT, false, "Exp" );
+			m_inputPorts[ 1 ].FloatInternalData = 1;
+			AddOutputPort( WirePortDataType.FLOAT, Constants.EmptyPortValue );
+			m_useInternalPortData = true;
+			m_autoWrapProperties = true;
+			m_textLabelWidth = 85;
+			m_previewShaderGUID = "758cc2f8b537b4e4b93d9833075d138c";
+		}
+
+		public override void OnInputPortConnected( int portId, int otherNodeId, int otherPortId, bool activateNode = true )
+		{
+			base.OnInputPortConnected( portId, otherNodeId, otherPortId, activateNode );
+			UpdateConnections( portId );
+		}
+
+		public override void OnConnectedOutputNodeChanges( int inputPortId, int otherNodeId, int otherPortId, string name, WirePortDataType type )
+		{
+			base.OnConnectedOutputNodeChanges( inputPortId, otherNodeId, otherPortId, name, type );
+			UpdateConnections( inputPortId );
+		}
+		
+		void UpdateConnections( int inputPort )
+		{
+			m_inputPorts[ inputPort ].MatchPortToConnection();
+			if ( inputPort == 0 )
+			{
+				m_outputPorts[ 0 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
+			}
+		}
+
+		public override void DrawProperties()
+		{
+			base.DrawProperties();
+			m_safePower = EditorGUILayoutToggle( SafePowerLabel, m_safePower );
+		}
+
+		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 x = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
+			if( m_safePower )
+			{
+				string safePowerName = "saferPower" + OutputId;
+				string safePowerValue = string.Format( SafePowerStr, x );
+				dataCollector.AddLocalVariable( UniqueId, CurrentPrecisionType, m_inputPorts[ 0 ].DataType, safePowerName, safePowerValue );
+				x = safePowerName;
+			}
+
+			string y = m_inputPorts[ 1 ].GenerateShaderForOutput( ref dataCollector, m_inputPorts[ 0 ].DataType, ignoreLocalvar, true );
+			string result = "pow( " + x + " , " + y + " )";
+
+			return CreateOutputLocalVariable( 0, result, ref dataCollector );
+		}
+
+
+		public override void ReadFromString( ref string[] nodeParams )
+		{
+			base.ReadFromString( ref nodeParams );
+			if( UIUtils.CurrentShaderVersion() > 17502 )
+				m_safePower = Convert.ToBoolean( GetCurrentParam( ref nodeParams ) );
+		}
+
+		public override void WriteToString( ref string nodeInfo, ref string connectionsInfo )
+		{
+			base.WriteToString( ref nodeInfo, ref connectionsInfo );
+			IOUtils.AddFieldValueToString( ref nodeInfo, m_safePower );
+		}
+	}
+}