108 lines
3.8 KiB
C#
108 lines
3.8 KiB
C#
// 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
|