1 files changed, 107 insertions, 0 deletions
diff --git a/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/SRP/BakedGINode.cs b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/SRP/BakedGINode.cs
new file mode 100644
index 00000000..7a67f480
--- /dev/null
+++ b/Assets/AmplifyShaderEditor/Plugins/Editor/Nodes/SRP/BakedGINode.cs
@@ -0,0 +1,107 @@
+// Amplify Shader Editor - Visual Shader Editing Tool
+// Copyright (c) Amplify Creations, Lda <info@amplify.pt>
+
+#if UNITY_2018_3_OR_NEWER
+using System;
+using UnityEngine;
+
+namespace AmplifyShaderEditor
+{
+	[Serializable]
+	[NodeAttributes( "SRP Baked GI", "Miscellaneous", "Gets Baked GI info." )]
+	public sealed class BakedGINode : ParentNode
+	{
+		private const string HDBakedGIHeader = "ASEBakedGI( {0}, {1}, {2}, {3} )";
+		private readonly string[] HDBakedGIBody =
+		{
+			"float3 ASEBakedGI( float3 positionWS, float3 normalWS, float2 uvStaticLightmap, float2 uvDynamicLightmap )\n",
+			"{\n",
+			"\tfloat3 positionRWS = GetCameraRelativePositionWS( positionWS );\n",
+			"\treturn SampleBakedGI( positionRWS, normalWS, uvStaticLightmap, uvDynamicLightmap );\n",
+			"}\n"
+		};
+
+		private readonly string LWBakedGIHeader = "ASEBakedGI( {0}, {1}, {2})";
+		private readonly string[] LWBakedGIBody =
+		{
+		"float3 ASEBakedGI( float3 normalWS, float2 uvStaticLightmap, bool applyScaling )\n",
+		"{\n",
+		"#ifdef LIGHTMAP_ON\n",
+		"\tif( applyScaling )\n",
+		"\t\tuvStaticLightmap = uvStaticLightmap * unity_LightmapST.xy + unity_LightmapST.zw;\n",
+		"\treturn SampleLightmap( uvStaticLightmap, normalWS );\n",
+		"#else\n",
+		"\treturn SampleSH(normalWS);\n",
+		"#endif\n",
+		"}\n"
+		};
+
+		private const string ApplyScalingStr = "Apply Scaling";
+
+		[SerializeField]
+		private bool m_applyScaling = true;
+
+		protected override void CommonInit( int uniqueId )
+		{
+			base.CommonInit( uniqueId );
+			AddInputPort( WirePortDataType.FLOAT3, false, "World Position" );
+			AddInputPort( WirePortDataType.FLOAT3, false, "World Normal" );
+			AddInputPort( WirePortDataType.FLOAT2, false, "Static UV" );
+			AddInputPort( WirePortDataType.FLOAT2, false, "Dynamic UV" );
+			AddOutputPort( WirePortDataType.FLOAT3, Constants.EmptyPortValue );
+			m_textLabelWidth = 95;
+			m_autoWrapProperties = true;
+		}
+
+		public override void DrawProperties()
+		{
+			base.DrawProperties();
+			m_applyScaling = EditorGUILayoutToggle( ApplyScalingStr, m_applyScaling );
+		}
+
+		public override string GenerateShaderForOutput( int outputId, ref MasterNodeDataCollector dataCollector, bool ignoreLocalvar )
+		{
+			if( !dataCollector.IsSRP )
+			{
+				UIUtils.ShowMessage( "Node only intended to use on HD and Lightweight rendering pipelines" );
+				return GenerateErrorValue();
+			}
+
+			if( m_outputPorts[ 0 ].IsLocalValue( dataCollector.PortCategory ) )
+				return m_outputPorts[ 0 ].LocalValue( dataCollector.PortCategory );
+
+			string positionWS = m_inputPorts[ 0 ].GeneratePortInstructions( ref dataCollector );
+			string normalWS = m_inputPorts[ 1 ].GeneratePortInstructions( ref dataCollector );
+			string uvStaticLightmap = m_inputPorts[ 2 ].GeneratePortInstructions( ref dataCollector );
+			string uvDynamicLightmap = m_inputPorts[ 3 ].GeneratePortInstructions( ref dataCollector );
+			string localVarName = "bakedGI" + OutputId;
+
+			if( dataCollector.TemplateDataCollectorInstance.IsHDRP )
+			{
+				dataCollector.AddFunction( HDBakedGIBody[ 0 ], HDBakedGIBody, false );
+				RegisterLocalVariable( 0, string.Format( HDBakedGIHeader, positionWS, normalWS, uvStaticLightmap, uvDynamicLightmap ), ref dataCollector, localVarName );
+			}
+			else
+			{
+				dataCollector.AddFunction( LWBakedGIBody[ 0 ], LWBakedGIBody, false );
+				RegisterLocalVariable( 0, string.Format( LWBakedGIHeader, normalWS, uvStaticLightmap, m_applyScaling?"true":"false" ), ref dataCollector, localVarName );
+			}
+			return localVarName;
+		}
+
+		public override void ReadFromString( ref string[] nodeParams )
+		{
+			base.ReadFromString( ref nodeParams );
+			m_applyScaling = 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_applyScaling );
+		}
+
+
+	}
+}
+#endif