From b82da95b5181ac8bbae38efb13e950d5e88a4caa Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
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

---
 .../Plugins/Editor/Menu/NodeWireReferencesUtils.cs | 56 ++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Menu/NodeWireReferencesUtils.cs

(limited to 'Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Menu/NodeWireReferencesUtils.cs')

diff --git a/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Menu/NodeWireReferencesUtils.cs b/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Menu/NodeWireReferencesUtils.cs
new file mode 100644
index 00000000..f1793a24
--- /dev/null
+++ b/Assets/ThirdParty/AmplifyShaderEditor/Plugins/Editor/Menu/NodeWireReferencesUtils.cs
@@ -0,0 +1,56 @@
+using UnityEngine;
+
+namespace AmplifyShaderEditor
+{
+	public class NodeWireReferencesUtils
+	{
+		public WireReference InputPortReference = new WireReference();
+		public WireReference SwitchPortReference = new WireReference();
+		public WireReference OutputPortReference = new WireReference();
+
+		public Vector2 SnapPosition = Vector2.zero;
+		public bool SnapEnabled = false;
+		public WireReference SnapPort = new WireReference();
+
+		public bool ValidReferences()
+		{
+			return ( InputPortReference.IsValid || OutputPortReference.IsValid );
+		}
+
+		public void InvalidateReferences()
+		{
+			InputPortReference.Invalidate();
+			OutputPortReference.Invalidate();
+			SnapPort.Invalidate();
+			SnapEnabled = false;
+		}
+
+
+		public void SetOutputReference( int nodeId, int portId, WirePortDataType dataType, bool typeLocked )
+		{
+			if( InputPortReference.IsValid )
+				InputPortReference.Invalidate();
+			OutputPortReference.SetReference( nodeId, portId, dataType, typeLocked );
+		}
+
+		public void SetInputReference( int nodeId, int portId, WirePortDataType dataType, bool typeLocked )
+		{
+			if( OutputPortReference.IsValid )
+				OutputPortReference.Invalidate();
+			InputPortReference.SetReference( nodeId, portId, dataType, typeLocked );
+		}
+
+		public void ActivateSnap( Vector2 position, WirePort port )
+		{
+			SnapPort.SetReference( port );
+			SnapEnabled = true;
+			SnapPosition = position;
+		}
+
+		public void DeactivateSnap()
+		{
+			SnapEnabled = false;
+			SnapPort.Invalidate();
+		}
+	}
+}
-- 
cgit v1.1-26-g67d0