* amplify shader editor

1 files changed, 346 insertions, 0 deletions

diff --git a/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/RegisterLocalVarNode.cs b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/RegisterLocalVarNode.cs
new file mode 100644
index 00000000..15bf0313
--- /dev/null
+++ b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Misc/RegisterLocalVarNode.cs
@@ -0,0 +1,346 @@
+// Amplify Shader Editor - Visual Shader Editing Tool
+// Copyright (c) Amplify Creations, Lda <info@amplify.pt>
+
+using UnityEngine;
+using UnityEditor;
+using System;
+using System.Collections.Generic;
+
+namespace AmplifyShaderEditor
+{
+	[Serializable]
+	[NodeAttributes( "Register Local Var", "Miscellaneous", "Forces a local variable to be written with the given name. Can then be fetched at any place with a <b>Get Local Var</b> node.", null, KeyCode.R )]
+	public sealed class RegisterLocalVarNode : ParentNode
+	{
+		private const double MaxEditingTimestamp = 1;
+
+		private const string LocalDefaultNameStr = "myVarName";
+		private const string LocalVarNameStr = "Var Name";
+		private const string OrderIndexStr = "Order Index";
+		private const string AutoOrderIndexStr = "Auto Order";
+		private const string ReferencesStr = "References";
+
+		private const string GetLocalVarLabel = "( {0} ) Get Local Var";
+		private string m_oldName = string.Empty;
+		private bool m_reRegisterName = false;
+		private int m_autoOrderIndex = int.MaxValue;
+		private bool m_forceUpdate = true;
+		private bool m_refSelect = false;
+
+		private bool m_referencesVisible = false;
+
+		[SerializeField]
+		private string m_variableName = LocalDefaultNameStr;
+
+		[SerializeField]
+		private int m_orderIndex = -1;
+
+		[SerializeField]
+		private bool m_autoIndexActive = true;
+
+		[SerializeField]
+		private List<GetLocalVarNode> m_registeredGetLocalVars = new List<GetLocalVarNode>();
+
+		[NonSerialized]
+		private double m_editingTimestamp;
+
+		[NonSerialized]
+		private bool m_editingTimestampFlag;
+
+		protected override void CommonInit( int uniqueId )
+		{
+			base.CommonInit( uniqueId );
+			AddInputPort( WirePortDataType.FLOAT, false, Constants.EmptyPortValue );
+			AddOutputPort( WirePortDataType.FLOAT, Constants.EmptyPortValue );
+			m_textLabelWidth = 85;
+			m_customPrecision = true;
+
+			if( m_containerGraph != null )
+				m_variableName += m_containerGraph.LocalVarNodes.NodesList.Count;
+
+			m_oldName = m_variableName;
+			UpdateTitle();
+			m_previewShaderGUID = "5aaa1d3ea9e1fa64781647e035a82334";
+		}
+
+		protected override void OnUniqueIDAssigned()
+		{
+			base.OnUniqueIDAssigned();
+			m_containerGraph.LocalVarNodes.AddNode( this );
+		}
+
+		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 );
+		}
+
+		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 );
+		}
+
+		void UpdateTitle()
+		{
+			SetAdditonalTitleText( string.Format( Constants.SubTitleVarNameFormatStr, m_variableName ) );
+		}
+
+		public override void OnNodeLogicUpdate( DrawInfo drawInfo )
+		{
+			base.OnNodeLogicUpdate( drawInfo );
+			if( m_editingTimestampFlag && ( EditorApplication.timeSinceStartup - m_editingTimestamp ) > MaxEditingTimestamp )
+			{
+				m_editingTimestampFlag = false;
+				CheckAndChangeName();
+			}
+		}
+
+		void DrawMainProperties()
+		{
+			EditorGUI.BeginChangeCheck();
+			m_variableName = EditorGUILayoutTextField( LocalVarNameStr, m_variableName );
+			if( EditorGUI.EndChangeCheck() )
+			{
+				m_editingTimestampFlag = true;
+				m_editingTimestamp = EditorApplication.timeSinceStartup;
+				//CheckAndChangeName();
+			}
+
+			DrawPrecisionProperty();
+		}
+
+		public override void AfterDuplication()
+		{
+			base.AfterDuplication();
+			CheckAndChangeName();
+		}
+
+		void CheckAndChangeName()
+		{
+			m_variableName = UIUtils.RemoveInvalidCharacters( m_variableName );
+			if( string.IsNullOrEmpty( m_variableName ) )
+			{
+				m_variableName = LocalDefaultNameStr + OutputId;
+			}
+			bool isNumericName = UIUtils.IsNumericName( m_variableName );
+			bool isLocalVarNameAvailable = m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.IsLocalvariableNameAvailable( m_variableName );
+			if( !isNumericName && isLocalVarNameAvailable )
+			{
+				m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.ReleaseLocalVariableName( UniqueId, m_oldName );
+				m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.RegisterLocalVariableName( UniqueId, m_variableName );
+				m_oldName = m_variableName;
+				m_containerGraph.LocalVarNodes.UpdateDataOnNode( UniqueId, m_variableName );
+				UpdateTitle();
+				m_forceUpdate = true;
+			}
+			else
+			{
+				if( isNumericName )
+				{
+					UIUtils.ShowMessage( UniqueId, string.Format( "Local variable name '{0}' cannot start or be numerical values. Reverting back to '{1}'.", m_variableName, m_oldName ), MessageSeverity.Warning );
+				}
+				else if( !isLocalVarNameAvailable )
+				{
+					UIUtils.ShowMessage( UniqueId, string.Format( "Local variable name '{0}' already being used. Reverting back to '{1}'.", m_variableName, m_oldName ), MessageSeverity.Warning );
+				}
+
+				m_variableName = m_oldName;
+				m_containerGraph.LocalVarNodes.UpdateDataOnNode( UniqueId, m_variableName );
+			}
+		}
+
+		void DrawReferences()
+		{
+			int count = m_registeredGetLocalVars.Count;
+			if( m_registeredGetLocalVars.Count > 0 )
+			{
+				for( int i = 0; i < count; i++ )
+				{
+					EditorGUILayout.BeginHorizontal();
+					EditorGUILayout.LabelField( string.Format( GetLocalVarLabel, m_registeredGetLocalVars[ i ].UniqueId ) );
+					if( GUILayout.Button( "\u25BA", "minibutton", GUILayout.Width( 17 ) ) )
+					{
+						m_containerGraph.ParentWindow.FocusOnNode( m_registeredGetLocalVars[ i ], 0, false );
+					}
+					EditorGUILayout.EndHorizontal();
+				}
+
+				if( GUILayout.Button( "Back" ) )
+				{
+					m_containerGraph.ParentWindow.FocusOnNode( this, 0, false );
+				}
+			}
+			else
+			{
+				EditorGUILayout.HelpBox( "This node is not being referenced by any Get Local Var.", MessageType.Info );
+			}
+		}
+
+		public override void DrawProperties()
+		{
+			NodeUtils.DrawPropertyGroup( ref m_propertiesFoldout, Constants.ParameterLabelStr, DrawMainProperties );
+			NodeUtils.DrawPropertyGroup( ref m_referencesVisible, ReferencesStr, DrawReferences );
+			//EditorGUILayout.LabelField(ConnStatus.ToString()+" "+m_activeConnections);
+		}
+
+		public override void OnEnable()
+		{
+			base.OnEnable();
+			m_reRegisterName = true;
+		}
+
+		public void CheckReferenceSelection()
+		{
+			m_refSelect = false;
+			int count = m_registeredGetLocalVars.Count;
+			for( int i = 0; i < count; i++ )
+			{
+				if( m_registeredGetLocalVars[ i ].Selected )
+					m_refSelect = true;
+			}
+		}
+
+		public override void OnNodeRepaint( DrawInfo drawInfo )
+		{
+			base.OnNodeRepaint( drawInfo );
+			if( m_isVisible && m_refSelect && !m_selected )
+			{
+				GUI.color = Constants.SpecialRegisterLocalVarSelectionColor;
+				GUI.Label( m_globalPosition, string.Empty, UIUtils.GetCustomStyle( CustomStyle.NodeWindowOn ) );
+				GUI.color = m_colorBuffer;
+			}
+		}
+
+		public override void Draw( DrawInfo drawInfo )
+		{
+
+			base.Draw( drawInfo );
+			if( m_reRegisterName )
+			{
+				m_reRegisterName = false;
+				m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.RegisterLocalVariableName( UniqueId, m_variableName );
+			}
+
+			if( m_forceUpdate )
+			{
+				m_forceUpdate = false;
+				UpdateTitle();
+			}
+		}
+
+		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 result = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
+			if( m_inputPorts[ 0 ].DataType == WirePortDataType.OBJECT )
+				m_outputPorts[ 0 ].SetLocalValue( result, dataCollector.PortCategory );
+			else
+				RegisterLocalVariable( 0, result, ref dataCollector, m_variableName + OutputId );
+			return m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory );
+		}
+
+		public override void ReadFromString( ref string[] nodeParams )
+		{
+			base.ReadFromString( ref nodeParams );
+			m_variableName = GetCurrentParam( ref nodeParams );
+			m_oldName = m_variableName;
+			if( UIUtils.CurrentShaderVersion() > 14 )
+				m_orderIndex = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
+
+			if( UIUtils.CurrentShaderVersion() > 3106 )
+			{
+				m_autoIndexActive = Convert.ToBoolean( GetCurrentParam( ref nodeParams ) );
+			}
+			else
+			{
+				m_autoIndexActive = false;
+			}
+			if( !m_isNodeBeingCopied )
+			{
+				m_containerGraph.LocalVarNodes.UpdateDataOnNode( UniqueId, m_variableName );
+				m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.ReleaseLocalVariableName( UniqueId, m_oldName );
+				m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.RegisterLocalVariableName( UniqueId, m_variableName );
+			}
+			m_forceUpdate = true;
+		}
+
+		public override void WriteToString( ref string nodeInfo, ref string connectionsInfo )
+		{
+			base.WriteToString( ref nodeInfo, ref connectionsInfo );
+			IOUtils.AddFieldValueToString( ref nodeInfo, m_variableName );
+			IOUtils.AddFieldValueToString( ref nodeInfo, m_orderIndex );
+			IOUtils.AddFieldValueToString( ref nodeInfo, m_autoIndexActive );
+		}
+
+		public override void PropagateNodeData( NodeData nodeData, ref MasterNodeDataCollector dataCollector )
+		{
+			if( m_autoOrderIndex < nodeData.OrderIndex )
+			{
+				nodeData.OrderIndex = m_autoOrderIndex - 1;
+			}
+			else
+			{
+				m_autoOrderIndex = nodeData.OrderIndex;
+				nodeData.OrderIndex -= 1;
+			}
+
+			base.PropagateNodeData( nodeData, ref dataCollector );
+		}
+
+		public override void ResetNodeData()
+		{
+			base.ResetNodeData();
+			m_autoOrderIndex = int.MaxValue;
+		}
+
+		public void RegisterGetLocalVar( GetLocalVarNode node )
+		{
+			if( !m_registeredGetLocalVars.Contains( node ) )
+			{
+				m_registeredGetLocalVars.Add( node );
+				CheckReferenceSelection();
+			}
+		}
+
+		public void UnregisterGetLocalVar( GetLocalVarNode node )
+		{
+			if( m_registeredGetLocalVars.Contains( node ) )
+			{
+				m_registeredGetLocalVars.Remove( node );
+				CheckReferenceSelection();
+			}
+		}
+
+		public override void Destroy()
+		{
+			base.Destroy();
+			m_containerGraph.LocalVarNodes.RemoveNode( this );
+			m_containerGraph.ParentWindow.DuplicatePrevBufferInstance.ReleaseLocalVariableName( UniqueId, m_variableName );
+
+			int count = m_registeredGetLocalVars.Count;
+			for( int i = 0; i < count; i++ )
+			{
+				//GetLocalVarNode node =  m_containerGraph.GetNode( m_registeredGetLocalVars[ i ] ) as GetLocalVarNode;
+				if( m_registeredGetLocalVars[ i ] != null )
+					m_registeredGetLocalVars[ i ].ResetReference();
+			}
+			m_registeredGetLocalVars.Clear();
+			m_registeredGetLocalVars = null;
+
+			m_containerGraph.LocalVarNodes.RemoveNode( this );
+		}
+
+		public override void ActivateNode( int signalGenNodeId, int signalGenPortId, Type signalGenNodeType )
+		{
+			base.ActivateNode( signalGenNodeId, signalGenPortId, signalGenNodeType );
+		}
+		public override string DataToArray { get { return m_variableName; } }
+		public List<GetLocalVarNode> NodeReferences { get { return m_registeredGetLocalVars; } }
+	}
+}