1 files changed, 355 insertions, 0 deletions
diff --git a/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Textures/TextureTransformNode.cs b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Textures/TextureTransformNode.cs
new file mode 100644
index 00000000..c35e550d
--- /dev/null
+++ b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/Textures/TextureTransformNode.cs
@@ -0,0 +1,355 @@
+using UnityEngine;
+using UnityEditor;
+using System;
+using System.Collections.Generic;
+
+namespace AmplifyShaderEditor
+{
+    [Serializable]
+    [NodeAttributes( "Texture Transform", "Textures", "Gives access to texture tiling and offset as set on the material inspector" )]
+    public sealed class TextureTransformNode : ParentNode
+    {
+        private readonly string[] Dummy = { string.Empty };
+		private const string InstancedLabelStr = "Instanced";
+
+		[SerializeField]
+		private bool m_instanced = false;
+
+		[SerializeField]
+        private int m_referenceSamplerId = -1;
+
+        [SerializeField]
+        private int m_referenceNodeId = -1;
+
+        [SerializeField]
+        private TexturePropertyNode m_inputReferenceNode = null;
+
+        private TexturePropertyNode m_referenceNode = null;
+
+		private Vector4Node m_texCoordsHelper;
+
+		private UpperLeftWidgetHelper m_upperLeftWidget = new UpperLeftWidgetHelper();
+
+		private int m_cachedSamplerId = -1;
+		private int m_cachedSamplerIdArray = -1;
+		private int m_cachedSamplerIdCube = -1;
+		private int m_cachedSamplerId3D = -1;
+
+		protected override void CommonInit( int uniqueId )
+        {
+            base.CommonInit( uniqueId );
+            AddInputPort( WirePortDataType.SAMPLER2D, false, "Tex" );
+            m_inputPorts[ 0 ].CreatePortRestrictions( WirePortDataType.SAMPLER1D, WirePortDataType.SAMPLER2D, WirePortDataType.SAMPLER3D, WirePortDataType.SAMPLERCUBE, WirePortDataType.OBJECT );
+            AddOutputPort( WirePortDataType.FLOAT2, "Tiling" );
+            AddOutputPort( WirePortDataType.FLOAT2, "Offset" );
+            m_textLabelWidth = 80;
+            m_autoWrapProperties = true;
+            m_hasLeftDropdown = true;
+			m_previewShaderGUID = "25ba2903568b00343ae06788994cab54";
+		}
+
+        public override void AfterCommonInit()
+        {
+            base.AfterCommonInit();
+
+            if( PaddingTitleLeft == 0 )
+            {
+                PaddingTitleLeft = Constants.PropertyPickerWidth + Constants.IconsLeftRightMargin;
+                if( PaddingTitleRight == 0 )
+                    PaddingTitleRight = Constants.PropertyPickerWidth + Constants.IconsLeftRightMargin;
+            }
+        }
+
+		public override void RenderNodePreview()
+		{
+			//Runs at least one time
+			if( !m_initialized )
+			{
+				// nodes with no preview don't update at all
+				PreviewIsDirty = false;
+				return;
+			}
+
+			if( !PreviewIsDirty )
+				return;
+
+			SetPreviewInputs();
+
+			RenderTexture temp = RenderTexture.active;
+
+			RenderTexture.active = m_outputPorts[ 0 ].OutputPreviewTexture;
+			PreviewMaterial.SetInt( "_PreviewID", 0 );
+			Graphics.Blit( null, m_outputPorts[ 0 ].OutputPreviewTexture, PreviewMaterial, m_previewMaterialPassId );
+
+			RenderTexture.active = m_outputPorts[ 1 ].OutputPreviewTexture;
+			PreviewMaterial.SetInt( "_PreviewID", 1 );
+			Graphics.Blit( null, m_outputPorts[ 1 ].OutputPreviewTexture, PreviewMaterial, m_previewMaterialPassId );
+			RenderTexture.active = temp;
+
+			PreviewIsDirty = m_continuousPreviewRefresh;
+
+			FinishPreviewRender = true;
+		}
+
+		void SetPreviewTexture( Texture newValue )
+		{
+			if( newValue is Cubemap )
+			{
+				m_previewMaterialPassId = 3;
+				if( m_cachedSamplerIdCube == -1 )
+					m_cachedSamplerIdCube = Shader.PropertyToID( "_Cube" );
+
+				PreviewMaterial.SetTexture( m_cachedSamplerIdCube, newValue as Cubemap );
+			}
+			else if( newValue is Texture2DArray )
+			{
+
+				m_previewMaterialPassId = 2;
+				if( m_cachedSamplerIdArray == -1 )
+					m_cachedSamplerIdArray = Shader.PropertyToID( "_Array" );
+
+				PreviewMaterial.SetTexture( m_cachedSamplerIdArray, newValue as Texture2DArray );
+			}
+			else if( newValue is Texture3D )
+			{
+				m_previewMaterialPassId = 1;
+				if( m_cachedSamplerId3D == -1 )
+					m_cachedSamplerId3D = Shader.PropertyToID( "_Sampler3D" );
+
+				PreviewMaterial.SetTexture( m_cachedSamplerId3D, newValue as Texture3D );
+			}
+			else
+			{
+				m_previewMaterialPassId = 0;
+				if( m_cachedSamplerId == -1 )
+					m_cachedSamplerId = Shader.PropertyToID( "_Sampler" );
+
+				PreviewMaterial.SetTexture( m_cachedSamplerId, newValue );
+			}
+		}
+
+		public override void SetPreviewInputs()
+		{
+			base.SetPreviewInputs();
+			if( m_inputPorts[ 0 ].IsConnected )
+			{
+				SetPreviewTexture( m_inputPorts[ 0 ].InputPreviewTexture( ContainerGraph ) );
+			}
+			else if( m_referenceNode != null )
+			{
+				if( m_referenceNode.Value != null )
+				{
+					SetPreviewTexture( m_referenceNode.Value );
+				}
+				else
+				{
+					SetPreviewTexture( m_referenceNode.PreviewTexture );
+				}
+			}
+		}
+
+		public override void OnInputPortConnected( int portId, int otherNodeId, int otherPortId, bool activateNode = true )
+        {
+            base.OnInputPortConnected( portId, otherNodeId, otherPortId, activateNode );
+            m_inputReferenceNode = m_inputPorts[ 0 ].GetOutputNodeWhichIsNotRelay() as TexturePropertyNode;
+			UpdateTitle();
+
+		}
+
+
+        public override void OnInputPortDisconnected( int portId )
+        {
+            base.OnInputPortDisconnected( portId );
+            m_inputReferenceNode = null;
+			UpdateTitle();
+		}
+
+
+        void UpdateTitle()
+        {
+            if( m_inputReferenceNode != null )
+            {
+                m_additionalContent.text = string.Format( Constants.PropertyValueLabel, m_inputReferenceNode.PropertyInspectorName );
+            }
+            else if( m_referenceSamplerId > -1 && m_referenceNode != null )
+            {
+                m_additionalContent.text = string.Format( Constants.PropertyValueLabel, m_referenceNode.PropertyInspectorName );
+            }
+            else
+            {
+                m_additionalContent.text = string.Empty;
+            }
+            m_sizeIsDirty = true;
+        }
+
+        public override void DrawProperties()
+        {
+			base.DrawProperties();
+			bool guiEnabledBuffer = GUI.enabled;
+			EditorGUI.BeginChangeCheck();
+			List<string> arr = new List<string>( UIUtils.TexturePropertyNodeArr() );
+
+			if( arr != null && arr.Count > 0 )
+			{
+				arr.Insert( 0, "None" );
+				GUI.enabled = true && ( !m_inputPorts[ 0 ].IsConnected );
+				m_referenceSamplerId = EditorGUILayoutPopup( Constants.AvailableReferenceStr, m_referenceSamplerId + 1, arr.ToArray() ) - 1;
+			}
+			else
+			{
+				m_referenceSamplerId = -1;
+				GUI.enabled = false;
+				EditorGUILayoutPopup( Constants.AvailableReferenceStr, m_referenceSamplerId, Dummy );
+			}
+
+			GUI.enabled = guiEnabledBuffer;
+			if( EditorGUI.EndChangeCheck() )
+			{
+				m_referenceNode = UIUtils.GetTexturePropertyNode( m_referenceSamplerId );
+				if( m_referenceNode != null )
+				{
+					m_referenceNodeId = m_referenceNode.UniqueId;
+				}
+				else
+				{
+					m_referenceNodeId = -1;
+					m_referenceSamplerId = -1;
+				}
+				UpdateTitle();
+			}
+
+			m_instanced = EditorGUILayoutToggle( InstancedLabelStr, m_instanced );
+        }
+
+        public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
+        {
+			if( !m_outputPorts[ 0 ].IsLocalValue( dataCollector.PortCategory ) )
+			{
+				base.GenerateShaderForOutput( outputId, ref dataCollector, ignoreLocalvar );
+				string texTransform = string.Empty;
+
+				if( m_inputPorts[ 0 ].IsConnected )
+				{
+					texTransform = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector ) + "_ST";
+				}
+				else if( m_referenceNode != null )
+				{
+					m_referenceNode.BaseGenerateShaderForOutput( outputId, ref dataCollector, ignoreLocalvar );
+					texTransform = m_referenceNode.PropertyName + "_ST";
+				}
+				else
+				{
+					texTransform = "_ST";
+					UIUtils.ShowMessage( UniqueId, "Please specify a texture sample on the Texture Transform Size node", MessageSeverity.Warning );
+				}
+
+				//bool excludeUniformKeyword = UIUtils.CurrentWindow.OutsideGraph.IsInstancedShader || UIUtils.CurrentWindow.OutsideGraph.IsSRP;
+				//string uniformRegister = UIUtils.GenerateUniformName( excludeUniformKeyword, WirePortDataType.FLOAT4, texTransform );
+				//dataCollector.AddToUniforms( UniqueId, uniformRegister, true );
+				if( m_texCoordsHelper == null )
+				{
+					m_texCoordsHelper = CreateInstance<Vector4Node>();
+					m_texCoordsHelper.ContainerGraph = ContainerGraph;
+					m_texCoordsHelper.SetBaseUniqueId( UniqueId, true );
+					m_texCoordsHelper.RegisterPropertyOnInstancing = false;
+					m_texCoordsHelper.AddGlobalToSRPBatcher = true;
+				}
+
+				if( m_instanced )
+				{
+					m_texCoordsHelper.CurrentParameterType = PropertyType.InstancedProperty;
+				}
+				else
+				{
+					m_texCoordsHelper.CurrentParameterType = PropertyType.Global;
+				}
+				m_texCoordsHelper.ResetOutputLocals();
+				m_texCoordsHelper.SetRawPropertyName( texTransform );
+				texTransform = m_texCoordsHelper.GenerateShaderForOutput( 0, ref dataCollector, false );
+
+				m_outputPorts[ 0 ].SetLocalValue( texTransform+ ".xy", dataCollector.PortCategory );
+				m_outputPorts[ 1 ].SetLocalValue( texTransform + ".zw", dataCollector.PortCategory );
+			}
+
+			return m_outputPorts[ outputId ].LocalValue( dataCollector.PortCategory );
+        }
+
+		public override void Draw( DrawInfo drawInfo )
+		{
+			base.Draw( drawInfo );
+
+			EditorGUI.BeginChangeCheck();
+			{
+				List<string> arr = new List<string>( UIUtils.TexturePropertyNodeArr() );
+				bool guiEnabledBuffer = GUI.enabled;
+
+				if( arr != null && arr.Count > 0 )
+				{
+					arr.Insert( 0, "None" );
+					GUI.enabled = true && ( !m_inputPorts[ 0 ].IsConnected );
+					m_referenceSamplerId = m_upperLeftWidget.DrawWidget( this, m_referenceSamplerId + 1, arr.ToArray() ) - 1;
+				}
+				else
+				{
+					m_referenceSamplerId = -1;
+					GUI.enabled = false;
+					m_upperLeftWidget.DrawWidget( this, m_referenceSamplerId, Dummy );
+				}
+				GUI.enabled = guiEnabledBuffer;
+			}
+			if( EditorGUI.EndChangeCheck() )
+			{
+				m_referenceNode = UIUtils.GetTexturePropertyNode( m_referenceSamplerId );
+				if( m_referenceNode != null )
+				{
+					m_referenceNodeId = m_referenceNode.UniqueId;
+				}
+				else
+				{
+					m_referenceNodeId = -1;
+					m_referenceSamplerId = -1;
+				}
+				UpdateTitle();
+			}
+		}
+
+		public override void RefreshExternalReferences()
+        {
+            base.RefreshExternalReferences();
+            m_referenceNode = UIUtils.GetNode( m_referenceNodeId ) as TexturePropertyNode;
+            m_referenceSamplerId = UIUtils.GetTexturePropertyNodeRegisterId( m_referenceNodeId );
+            UpdateTitle();
+        }
+
+        public override void ReadFromString( ref string[] nodeParams )
+        {
+            base.ReadFromString( ref nodeParams );
+            m_referenceNodeId = Convert.ToInt32( GetCurrentParam( ref nodeParams ) );
+			if( UIUtils.CurrentShaderVersion() > 17200 )
+			{
+				m_instanced = 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_referenceNodeId );
+			IOUtils.AddFieldValueToString( ref nodeInfo, m_instanced );
+		}
+
+        public override void Destroy()
+        {
+            base.Destroy();
+            m_referenceNode = null;
+            m_inputReferenceNode = null;
+            m_upperLeftWidget = null;
+			if( m_texCoordsHelper != null )
+			{
+				//Not calling m_texCoordsHelper.Destroy() on purpose so UIUtils does not incorrectly unregister stuff
+				DestroyImmediate( m_texCoordsHelper );
+				m_texCoordsHelper = null;
+			}
+		}
+    }
+}