1 files changed, 484 insertions, 0 deletions
diff --git a/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/WireNode.cs b/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/WireNode.cs
new file mode 100644
index 00000000..4bcc5005
--- /dev/null
+++ b/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/WireNode.cs
@@ -0,0 +1,484 @@
+using System;
+using UnityEngine;
+
+namespace AmplifyShaderEditor
+{
+	[Serializable]
+	[NodeAttributes( "Wire Node", "Miscellaneous", "Wire Node", null, KeyCode.None, false )]
+	public sealed class WireNode : ParentNode
+	{
+		private bool m_markedToDelete = false;
+
+		[SerializeField]
+		private WirePortDataType m_visualDataType = WirePortDataType.FLOAT;
+
+		bool m_forceVisualDataUpdate = false;
+
+		protected override void CommonInit( int uniqueId )
+		{
+			base.CommonInit( uniqueId );
+			AddInputPort( WirePortDataType.OBJECT, false, string.Empty );
+			AddOutputPort( WirePortDataType.OBJECT, Constants.EmptyPortValue );
+			m_tooltipText = string.Empty;
+			m_drawPreview = false;
+			m_drawPreviewExpander = false;
+			m_canExpand = false;
+			m_previewShaderGUID = "fa1e3e404e6b3c243b5527b82739d682";
+		}
+
+		public WirePortDataType GetLastInputDataTypeRecursively()
+		{
+			if( m_outputPorts[ 0 ].ExternalReferences.Count > 0 )
+			{
+				WireNode rightWire = m_outputPorts[ 0 ].GetInputNode( 0 ) as WireNode;
+				if( rightWire != null )
+					return rightWire.GetLastInputDataTypeRecursively();
+				else
+				{
+					return m_outputPorts[ 0 ].GetInputConnection( 0 ).DataType;
+				}
+			}
+
+			if( m_containerGraph.ParentWindow.WireReferenceUtils.OutputPortReference.IsValid )
+				return m_containerGraph.ParentWindow.WireReferenceUtils.OutputPortReference.DataType;
+			else
+				return m_visualDataType;
+		}
+
+		public override WirePortDataType GetInputPortVisualDataTypeByArrayIdx( int portArrayIdx )
+		{
+			return m_visualDataType;
+		}
+
+		public override WirePortDataType GetOutputPortVisualDataTypeById( int portId )
+		{
+			return m_visualDataType;
+		}
+
+		public override void OnInputPortConnected( int portId, int otherNodeId, int otherPortId, bool activateNode = true )
+		{
+			base.OnInputPortConnected( portId, otherNodeId, otherPortId, activateNode );
+			m_inputPorts[ 0 ].MatchPortToConnection();
+			m_outputPorts[ 0 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
+
+			m_forceVisualDataUpdate = true;
+		}
+
+		public override void OnOutputPortConnected( int portId, int otherNodeId, int otherPortId )
+		{
+			base.OnOutputPortConnected( portId, otherNodeId, otherPortId );
+
+			if( m_outputPorts[ portId ].ConnectionCount > 1 )
+			{
+				for( int i = 0; i < m_outputPorts[ portId ].ExternalReferences.Count; i++ )
+				{
+					if( m_outputPorts[ portId ].ExternalReferences[ i ].PortId != otherPortId )
+					{
+						UIUtils.DeleteConnection( true, m_outputPorts[ portId ].ExternalReferences[ i ].NodeId, m_outputPorts[ portId ].ExternalReferences[ i ].PortId, false, true );
+					}
+				}
+			}
+
+			m_inputPorts[ 0 ].NotifyExternalRefencesOnChange();
+			m_forceVisualDataUpdate = true;
+		}
+
+		public override void OnConnectedInputNodeChanges( int portId, int otherNodeId, int otherPortId, string name, WirePortDataType type )
+		{
+			base.OnConnectedInputNodeChanges( portId, otherNodeId, otherPortId, name, type );
+
+			m_inputPorts[ 0 ].NotifyExternalRefencesOnChange();
+			m_forceVisualDataUpdate = true;
+		}
+
+		public override void OnConnectedOutputNodeChanges( int outputPortId, int otherNodeId, int otherPortId, string name, WirePortDataType type )
+		{
+			base.OnConnectedOutputNodeChanges( outputPortId, otherNodeId, otherPortId, name, type );
+			m_inputPorts[ 0 ].MatchPortToConnection();
+			m_outputPorts[ 0 ].ChangeType( m_inputPorts[ 0 ].DataType, false );
+
+			m_forceVisualDataUpdate = true;
+		}
+
+		public override void OnInputPortDisconnected( int portId )
+		{
+			base.OnInputPortDisconnected( portId );
+			TestIfValid();
+
+			m_forceVisualDataUpdate = true;
+			m_outputPorts[ 0 ].NotifyExternalRefencesOnChange();
+		}
+
+		public override void OnOutputPortDisconnected( int portId )
+		{
+			base.OnOutputPortDisconnected( portId );
+			TestIfValid();
+
+			m_forceVisualDataUpdate = true;
+			m_inputPorts[ 0 ].NotifyExternalRefencesOnChange();
+		}
+
+		public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
+		{
+			base.GenerateShaderForOutput( outputId, ref dataCollector, ignoreLocalvar );
+			return m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
+		}
+
+		public override void DrawProperties()
+		{
+			if( m_markedToDelete )
+				return;
+
+			base.DrawProperties();
+		}
+
+		public override void OnNodeLayout( DrawInfo drawInfo )
+		{
+			if( m_firstDraw )
+			{
+				m_firstDraw = false;
+				AfterCommonInit();
+				OnNodeChange();
+			}
+
+			if( m_forceVisualDataUpdate )
+			{
+				m_forceVisualDataUpdate = false;
+				m_visualDataType = GetLastInputDataTypeRecursively();
+			}
+
+			if( m_repopulateDictionaries )
+			{
+				m_repopulateDictionaries = false;
+
+				m_inputPortsDict.Clear();
+				int inputCount = m_inputPorts.Count;
+				for( int i = 0; i < inputCount; i++ )
+				{
+					m_inputPortsDict.Add( m_inputPorts[ i ].PortId, m_inputPorts[ i ] );
+				}
+
+				m_outputPortsDict.Clear();
+				int outputCount = m_outputPorts.Count;
+				for( int i = 0; i < outputCount; i++ )
+				{
+					m_outputPortsDict.Add( m_outputPorts[ i ].PortId, m_outputPorts[ i ] );
+				}
+			}
+
+			if( m_sizeIsDirty )
+			{
+				m_sizeIsDirty = false;
+				m_extraSize.Set( 20f, 20f );
+				m_position.width = m_extraSize.x + UIUtils.PortsSize.x;
+				m_position.height = m_extraSize.y + UIUtils.PortsSize.y;
+
+				Vec2Position -= Position.size * 0.5f;
+				if( OnNodeChangeSizeEvent != null )
+				{
+					OnNodeChangeSizeEvent( this );
+				}
+
+				ChangeSizeFinished();
+				//ChangeSize();
+			}
+
+			CalculatePositionAndVisibility( drawInfo );
+
+			// Input Ports
+			{
+				m_currInputPortPos = m_globalPosition;
+				m_currInputPortPos.width = drawInfo.InvertedZoom * UIUtils.PortsSize.x;
+				m_currInputPortPos.height = drawInfo.InvertedZoom * UIUtils.PortsSize.y;
+				m_currInputPortPos.position = m_globalPosition.center - m_currInputPortPos.size * 0.5f;
+				int inputCount = m_inputPorts.Count;
+
+				for( int i = 0; i < inputCount; i++ )
+				{
+					if( m_inputPorts[ i ].Visible )
+					{
+						// Button
+						m_inputPorts[ i ].Position = m_currInputPortPos;
+
+						if( !m_inputPorts[ i ].Locked )
+						{
+							float overflow = 2;
+							float scaledOverflow = 3 * drawInfo.InvertedZoom;
+							m_auxRect = m_currInputPortPos;
+							m_auxRect.yMin -= scaledOverflow + overflow;
+							m_auxRect.yMax += scaledOverflow + overflow;
+							m_auxRect.xMin -= Constants.PORT_INITIAL_X * drawInfo.InvertedZoom + scaledOverflow + overflow;
+							m_auxRect.xMax += m_inputPorts[ i ].LabelSize.x + Constants.PORT_TO_LABEL_SPACE_X * drawInfo.InvertedZoom + scaledOverflow + overflow;
+							m_inputPorts[ i ].ActivePortArea = m_auxRect;
+						}
+						m_currInputPortPos.y += drawInfo.InvertedZoom * ( m_fontHeight + Constants.INPUT_PORT_DELTA_Y );
+					}
+				}
+			}
+
+			// Output Ports
+			{
+				m_currOutputPortPos = m_globalPosition;
+				m_currOutputPortPos.width = drawInfo.InvertedZoom * UIUtils.PortsSize.x;
+				m_currOutputPortPos.height = drawInfo.InvertedZoom * UIUtils.PortsSize.y;
+				m_currOutputPortPos.position = m_globalPosition.center - m_currOutputPortPos.size * 0.5f;
+				//m_currOutputPortPos.x += ( m_globalPosition.width - drawInfo.InvertedZoom * ( Constants.PORT_INITIAL_X + m_anchorAdjust ) );
+				//m_currOutputPortPos.y += drawInfo.InvertedZoom * Constants.PORT_INITIAL_Y;// + m_extraHeaderHeight * drawInfo.InvertedZoom;
+				int outputCount = m_outputPorts.Count;
+
+				for( int i = 0; i < outputCount; i++ )
+				{
+					if( m_outputPorts[ i ].Visible )
+					{
+						//Button
+						m_outputPorts[ i ].Position = m_currOutputPortPos;
+
+						if( !m_outputPorts[ i ].Locked )
+						{
+							float overflow = 2;
+							float scaledOverflow = 3 * drawInfo.InvertedZoom;
+							m_auxRect = m_currOutputPortPos;
+							m_auxRect.yMin -= scaledOverflow + overflow;
+							m_auxRect.yMax += scaledOverflow + overflow;
+							m_auxRect.xMin -= m_outputPorts[ i ].LabelSize.x + Constants.PORT_TO_LABEL_SPACE_X * drawInfo.InvertedZoom + scaledOverflow + overflow;
+							m_auxRect.xMax += Constants.PORT_INITIAL_X * drawInfo.InvertedZoom + scaledOverflow + overflow;
+							m_outputPorts[ i ].ActivePortArea = m_auxRect;
+						}
+						m_currOutputPortPos.y += drawInfo.InvertedZoom * ( m_fontHeight + Constants.INPUT_PORT_DELTA_Y );
+					}
+				}
+			}
+		}
+
+		public override void OnNodeRepaint( DrawInfo drawInfo )
+		{
+			//base.OnRepaint( drawInfo );
+			//return;
+			if( !m_isVisible )
+				return;
+
+			m_colorBuffer = GUI.color;
+
+			// Output Ports
+			int outputCount = m_outputPorts.Count;
+			for( int i = 0; i < outputCount; i++ )
+			{
+				if( m_outputPorts[ i ].Visible )
+				{
+					// Output Port Icon
+					if( ContainerGraph.LodLevel <= ParentGraph.NodeLOD.LOD4 )
+					{
+						if( m_outputPorts[ i ].Locked )
+							GUI.color = Constants.LockedPortColor;
+						else if( ContainerGraph.ParentWindow.Options.ColoredPorts )
+							GUI.color = UIUtils.GetColorForDataType( m_visualDataType, false, false );
+						else
+							GUI.color = m_outputPorts[ i ].HasCustomColor ? m_outputPorts[ i ].CustomColor : UIUtils.GetColorForDataType( m_visualDataType, true, false );
+
+						GUIStyle style = m_outputPorts[ i ].IsConnected ? UIUtils.GetCustomStyle( CustomStyle.PortFullIcon ) : UIUtils.GetCustomStyle( CustomStyle.PortEmptyIcon );
+						GUI.Label( m_outputPorts[ i ].Position, string.Empty, style );
+
+						GUI.color = m_colorBuffer;
+					}
+
+					// Output Port Label
+					if( ContainerGraph.LodLevel <= ParentGraph.NodeLOD.LOD3 )
+					{
+						if( m_outputPorts[ i ].Locked )
+						{
+							GUI.color = Constants.PortLockedTextColor;
+							GUI.Label( m_outputPorts[ i ].LabelPosition, m_outputPorts[ i ].Name, UIUtils.OutputPortLabel );
+							GUI.color = m_colorBuffer;
+						}
+						else
+						{
+							GUI.Label( m_outputPorts[ i ].LabelPosition, m_outputPorts[ i ].Name, UIUtils.OutputPortLabel );
+						}
+					}
+				}
+			}
+
+			// Selection Box
+			if( m_selected )
+			{
+				Rect selectionBox = m_globalPosition;
+				selectionBox.size = Vector2.one * 16 * drawInfo.InvertedZoom + Vector2.one * 4;
+				selectionBox.center = m_globalPosition.center;
+				GUI.DrawTexture( selectionBox, UIUtils.WireNodeSelection );
+				GUI.color = m_colorBuffer;
+			}
+		}
+
+		public override void DrawGUIControls( DrawInfo drawInfo )
+		{
+			//base.DrawGUIControls( drawInfo );
+		}
+
+		public override void Draw( DrawInfo drawInfo )
+		{
+			if( m_markedToDelete )
+				return;
+
+			if( drawInfo.CurrentEventType == EventType.Repaint )
+				OnNodeRepaint( drawInfo );
+			//base.Draw( drawInfo );
+
+			if( drawInfo.CurrentEventType == EventType.Repaint )
+				TestIfValid();
+		}
+
+		bool TestIfValid()
+		{
+			if( !Alive )
+				return false;
+
+			bool result = true;
+			if( !m_inputPorts[ 0 ].IsConnected )
+			{
+				if( !m_containerGraph.ParentWindow.WireReferenceUtils.InputPortReference.IsValid || m_containerGraph.ParentWindow.WireReferenceUtils.InputPortReference.IsValid && m_containerGraph.ParentWindow.WireReferenceUtils.InputPortReference.NodeId != UniqueId )
+				{
+					ContainerGraph.MarkWireNodeSequence( this, true );
+					result = false;
+				}
+			}
+
+			if( !m_outputPorts[ 0 ].IsConnected )
+			{
+				if( !m_containerGraph.ParentWindow.WireReferenceUtils.OutputPortReference.IsValid || m_containerGraph.ParentWindow.WireReferenceUtils.OutputPortReference.IsValid && m_containerGraph.ParentWindow.WireReferenceUtils.OutputPortReference.NodeId != UniqueId )
+				{
+					ContainerGraph.MarkWireNodeSequence( this, false );
+					result = false;
+				}
+			}
+			return result;
+		}
+
+		public Vector3 TangentDirection
+		{
+			get
+			{
+				ParentNode otherInputNode = null;
+				ParentNode otherOutputNode = null;
+
+				//defaults to itself so it can still calculate tangents
+				WirePort otherInputPort = m_outputPorts[ 0 ];
+				WirePort otherOutputPort = m_inputPorts[ 0 ];
+
+				if( m_outputPorts[ 0 ].ConnectionCount > 0 )
+				{
+					otherInputNode = m_containerGraph.GetNode( m_outputPorts[ 0 ].ExternalReferences[ 0 ].NodeId );
+					otherInputPort = otherInputNode.GetInputPortByUniqueId( m_outputPorts[ 0 ].ExternalReferences[ 0 ].PortId );
+				}
+
+				if( m_inputPorts[ 0 ].ConnectionCount > 0 )
+				{
+					otherOutputNode = m_containerGraph.GetNode( m_inputPorts[ 0 ].ExternalReferences[ 0 ].NodeId );
+					otherOutputPort = otherOutputNode.GetOutputPortByUniqueId( m_inputPorts[ 0 ].ExternalReferences[ 0 ].PortId );
+				}
+
+				//TODO: it still generates crooked lines if wire nodes get too close to non-wire nodes (the fix would be to calculate the non-wire nodes magnitude properly)
+				float mag = Constants.HORIZONTAL_TANGENT_SIZE * ContainerGraph.ParentWindow.CameraDrawInfo.InvertedZoom;
+
+				Vector2 outPos;
+				if( otherOutputNode != null && otherOutputNode.GetType() != typeof( WireNode ) )
+					outPos = otherOutputPort.Position.position + Vector2.right * mag * 0.66f;
+				else
+					outPos = otherOutputPort.Position.position;
+
+				Vector2 inPos;
+				if( otherInputNode != null && otherInputNode.GetType() != typeof( WireNode ) )
+					inPos = otherInputPort.Position.position - Vector2.right * mag * 0.66f;
+				else
+					inPos = otherInputPort.Position.position;
+
+				Vector2 tangent = ( outPos - inPos ).normalized;
+				return new Vector3( tangent.x, tangent.y );
+			}
+		}
+
+		public override void RefreshExternalReferences()
+		{
+			base.RefreshExternalReferences();
+
+			m_extraSize.Set( 20f, 20f );
+			m_position.width = m_extraSize.x + UIUtils.PortsSize.x;
+			m_position.height = m_extraSize.y + UIUtils.PortsSize.y;
+
+			Vec2Position += Position.size * 0.5f;
+		}
+
+		public override void OnAfterDeserialize()
+		{
+			base.OnAfterDeserialize();
+			m_sizeIsDirty = false;
+		}
+
+		public WireReference FindNewValidInputNode( WireNode current )
+		{
+			if( current.InputPorts[ 0 ].IsConnected )
+			{
+				ParentNode node = m_containerGraph.GetNode( current.InputPorts[ 0 ].ExternalReferences[ 0 ].NodeId );
+				if( node != null )
+				{
+					WireNode wireNode = node as WireNode;
+					if( wireNode != null && wireNode.MarkToDelete )
+					{
+						return FindNewValidInputNode( wireNode );
+					}
+					else
+					{
+						return current.InputPorts[ 0 ].ExternalReferences[ 0 ];
+					}
+				}
+			}
+			return null;
+		}
+
+		public WireReference FindNewValidOutputNode( WireNode current )
+		{
+			if( current.OutputPorts[ 0 ].IsConnected )
+			{
+				ParentNode node = m_containerGraph.GetNode( current.OutputPorts[ 0 ].ExternalReferences[ 0 ].NodeId );
+
+				if( node != null )
+				{
+					WireNode wireNode = node as WireNode;
+					if( wireNode != null && wireNode.MarkToDelete )
+					{
+						return FindNewValidOutputNode( wireNode );
+					}
+					else
+					{
+						return current.OutputPorts[ 0 ].ExternalReferences[ 0 ];
+					}
+				}
+			}
+			return null;
+		}
+
+		public override void Rewire()
+		{
+			//if ( m_inputPorts[ 0 ].ExternalReferences != null && m_inputPorts[ 0 ].ExternalReferences.Count > 0 )
+			//{
+			//WireReference backPort = m_inputPorts[ 0 ].ExternalReferences[ 0 ];
+			//for ( int i = 0; i < m_outputPorts[ 0 ].ExternalReferences.Count; i++ )
+			//{
+			//	UIUtils.CurrentWindow.ConnectInputToOutput( m_outputPorts[ 0 ].ExternalReferences[ i ].NodeId, m_outputPorts[ 0 ].ExternalReferences[ i ].PortId, backPort.NodeId, backPort.PortId );
+			//}
+			//}
+			MarkToDelete = true;
+			WireReference outputReference = FindNewValidInputNode( this );
+			WireReference inputReference = FindNewValidOutputNode( this );
+			if( outputReference != null && inputReference != null )
+			{
+				ContainerGraph.ParentWindow.ConnectInputToOutput( inputReference.NodeId, inputReference.PortId, outputReference.NodeId, outputReference.PortId );
+			}
+		}
+
+		public bool MarkToDelete
+		{
+			get { return m_markedToDelete; }
+			set { m_markedToDelete = value; }
+		}
+	}
+}