katboi01
/
FateViewer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
FateViewer/Assets/AmplifyShaderEditor/Plugins/Editor/Templates/TemplateDataCollector.cs

2105 lines
86 KiB
C#
Raw Normal View History

Add project files.
2023-10-09 00:51:40 +08:00
// Amplify Shader Editor - Visual Shader Editing Tool
// Copyright (c) Amplify Creations, Lda <info@amplify.pt>
using UnityEngine;
using System.Collections.Generic;
using System.Text.RegularExpressions;
using System;
namespace AmplifyShaderEditor
{
public enum NormalizeType
{
Off,
Regular,
Safe
}
public class InterpDataHelper
{
public string VarName;
public WirePortDataType VarType;
public bool IsSingleComponent;
public bool SetAtCompileTime;
public InterpDataHelper( WirePortDataType varType, string varName, bool isSingleComponent = true , bool setAtCompileTime = false )
{
VarName = varName;
VarType = varType;
IsSingleComponent = isSingleComponent;
SetAtCompileTime = setAtCompileTime;
}
}
public class TemplateCustomData
{
public WirePortDataType DataType;
public string Name;
public bool IsVertex;
public bool IsFragment;
public TemplateCustomData( string name, WirePortDataType dataType )
{
name = Name;
DataType = dataType;
IsVertex = false;
IsFragment = false;
}
}
public class TemplateInputParameters
{
public WirePortDataType Type;
public string Name;
public string Declaration;
public TemplateSemantics Semantic;
public TemplateInputParameters( WirePortDataType type, PrecisionType precision, string name, TemplateSemantics semantic, string custom = null )
{
Type = type;
Name = name;
Semantic = semantic;
Declaration = string.Format( "{0} {1} : {2}", UIUtils.PrecisionWirePortToCgType( precision, type ), Name, Semantic );
if( !string.IsNullOrEmpty( custom ) )
Declaration = custom;
}
}
public class TemplateDataCollector
{
#if UNITY_2018_2_OR_NEWER
private const int MaxUV = 8;
private int[] m_UVUsage = { 0, 0, 0, 0, 0, 0, 0, 0 };
#else
private const int MaxUV = 4;
private int[] m_UVUsage = { 0, 0, 0, 0 };
#endif
private int m_multipassSubshaderIdx = 0;
private int m_multipassPassIdx = 0;
private TemplateMultiPass m_currentTemplate;
private TemplateSRPType m_currentSRPType = TemplateSRPType.BuiltIn;
private Dictionary<string, TemplateCustomData> m_customInterpolatedData;
private Dictionary<string, TemplateVertexData> m_registeredVertexData;
private Dictionary<TemplateInfoOnSematics, InterpDataHelper> m_availableFragData;
private Dictionary<TemplateInfoOnSematics, InterpDataHelper> m_availableVertData;
private TemplateInterpData m_interpolatorData;
private Dictionary<TemplateSemantics, TemplateVertexData> m_vertexDataDict;
private TemplateData m_currentTemplateData;
private MasterNodeDataCollector m_currentDataCollector;
public Dictionary<TemplateSemantics, TemplateInputParameters> m_vertexInputParams;
public Dictionary<TemplateSemantics, TemplateInputParameters> m_fragmentInputParams;
private Dictionary<TemplateInfoOnSematics, TemplateLocalVarData> m_specialVertexLocalVars;
private Dictionary<TemplateInfoOnSematics, TemplateLocalVarData> m_specialFragmentLocalVars;
private List<PropertyDataCollector> m_lateDirectivesList = new List<PropertyDataCollector>();
private Dictionary<string, PropertyDataCollector> m_lateDirectivesDict = new Dictionary<string, PropertyDataCollector>();
private List<PropertyDataCollector> m_srpBatcherPropertiesList = new List<PropertyDataCollector>();
private List<PropertyDataCollector> m_fullSrpBatcherPropertiesList = new List<PropertyDataCollector>();
private Dictionary<string, PropertyDataCollector> m_srpBatcherPropertiesDict = new Dictionary<string, PropertyDataCollector>();
public void CopySRPPropertiesFromDataCollector( int nodeId, TemplateDataCollector dataCollector )
{
for( int i = 0; i < dataCollector.SrpBatcherPropertiesList.Count; i++ )
{
AddSRPBatcherProperty( nodeId, dataCollector.SrpBatcherPropertiesList[ i ].PropertyName );
}
}
public void AddSRPBatcherProperty( int nodeID, string property )
{
if( !m_srpBatcherPropertiesDict.ContainsKey( property ) )
{
PropertyDataCollector newValue = new PropertyDataCollector( nodeID, property );
m_srpBatcherPropertiesDict.Add( property, newValue );
m_srpBatcherPropertiesList.Add( newValue );
}
}
public void SetUVUsage( int uv, WirePortDataType type )
{
if( uv >= 0 && uv < MaxUV )
{
m_UVUsage[ uv ] = Mathf.Max( m_UVUsage[ uv ], TemplateHelperFunctions.DataTypeChannelUsage[ type ] );
}
}
public void SetUVUsage( int uv, int size )
{
if( uv >= 0 && uv < MaxUV )
{
m_UVUsage[ uv ] = Mathf.Max( m_UVUsage[ uv ], size );
}
}
public void CloseLateDirectives()
{
if( m_lateDirectivesList.Count > 0 )
{
m_lateDirectivesList.Add( new PropertyDataCollector( -1, string.Empty ) );
}
}
public void AddHDLightInfo()
{
#if !UNITY_2018_3_OR_NEWER
AddLateDirective( AdditionalLineType.Custom, "#if (SHADERPASS != SHADERPASS_FORWARD) //On forward this info is already included" );
AddLateDirective( AdditionalLineType.Include, "HDRP/Lighting/LightDefinition.cs.hlsl" );
AddLateDirective( AdditionalLineType.Include, "HDRP/Lighting/LightLoop/Shadow.hlsl" );
AddLateDirective( AdditionalLineType.Include, "HDRP/Lighting/LightLoop/LightLoopDef.hlsl" );
AddLateDirective( AdditionalLineType.Custom, "#endif // End of light info includes" );
#endif
}
public void AddLateDirective( AdditionalLineType type, string value )
{
if( !m_lateDirectivesDict.ContainsKey( value ) )
{
string formattedValue = string.Empty;
switch( type )
{
case AdditionalLineType.Include: formattedValue = string.Format( Constants.IncludeFormat, value ); break;
case AdditionalLineType.Define: formattedValue = string.Format( Constants.DefineFormat, value ); break;
case AdditionalLineType.Pragma: formattedValue = string.Format( Constants.PragmaFormat, value ); break;
case AdditionalLineType.Custom: formattedValue = value; break;
}
PropertyDataCollector property = new PropertyDataCollector( -1, formattedValue );
m_lateDirectivesDict.Add( value, property );
m_lateDirectivesList.Add( property );
}
}
public void SetMultipassInfo( TemplateMultiPass currentTemplate, int subShaderIdx, int passIdx, TemplateSRPType currentSRPType )
{
m_currentTemplate = currentTemplate;
m_multipassSubshaderIdx = subShaderIdx;
m_multipassPassIdx = passIdx;
m_currentSRPType = currentSRPType;
}
public bool HasDirective( AdditionalLineType type, string value )
{
switch( type )
{
case AdditionalLineType.Include:
{
return m_currentTemplate.SubShaders[ m_multipassSubshaderIdx ].Modules.IncludePragmaContainer.HasInclude( value ) ||
m_currentTemplate.SubShaders[ m_multipassSubshaderIdx ].Passes[ m_multipassPassIdx ].Modules.IncludePragmaContainer.HasInclude( value );
}
case AdditionalLineType.Define:
{
return m_currentTemplate.SubShaders[ m_multipassSubshaderIdx ].Modules.IncludePragmaContainer.HasDefine( value ) ||
m_currentTemplate.SubShaders[ m_multipassSubshaderIdx ].Passes[ m_multipassPassIdx ].Modules.IncludePragmaContainer.HasDefine( value );
}
case AdditionalLineType.Pragma:
{
return m_currentTemplate.SubShaders[ m_multipassSubshaderIdx ].Modules.IncludePragmaContainer.HasPragma( value ) ||
m_currentTemplate.SubShaders[ m_multipassSubshaderIdx ].Passes[ m_multipassPassIdx ].Modules.IncludePragmaContainer.HasPragma( value );
}
}
return false;
}
public void FillSpecialVariables( TemplatePass currentPass )
{
m_specialVertexLocalVars = new Dictionary<TemplateInfoOnSematics, TemplateLocalVarData>();
m_specialFragmentLocalVars = new Dictionary<TemplateInfoOnSematics, TemplateLocalVarData>();
int localVarAmount = currentPass.LocalVarsList.Count;
for( int i = 0; i < localVarAmount; i++ )
{
if( currentPass.LocalVarsList[ i ].IsSpecialVar )
{
if( currentPass.LocalVarsList[ i ].Category == MasterNodePortCategory.Vertex )
{
m_specialVertexLocalVars.Add( currentPass.LocalVarsList[ i ].SpecialVarType, currentPass.LocalVarsList[ i ] );
}
else
{
m_specialFragmentLocalVars.Add( currentPass.LocalVarsList[ i ].SpecialVarType, currentPass.LocalVarsList[ i ] );
}
}
}
}
public void BuildFromTemplateData( MasterNodeDataCollector dataCollector, TemplateData templateData )
{
m_registeredVertexData = new Dictionary<string, TemplateVertexData>();
m_customInterpolatedData = new Dictionary<string, TemplateCustomData>();
m_currentDataCollector = dataCollector;
m_currentTemplateData = templateData;
m_vertexDataDict = new Dictionary<TemplateSemantics, TemplateVertexData>();
if( templateData.VertexDataList != null )
{
for( int i = 0; i < templateData.VertexDataList.Count; i++ )
{
m_vertexDataDict.Add( templateData.VertexDataList[ i ].Semantics, new TemplateVertexData( templateData.VertexDataList[ i ] ) );
}
}
m_availableFragData = new Dictionary<TemplateInfoOnSematics, InterpDataHelper>();
if( templateData.InterpolatorData != null && templateData.FragmentFunctionData != null )
{
m_interpolatorData = new TemplateInterpData( templateData.InterpolatorData );
int fragCount = templateData.InterpolatorData.Interpolators.Count;
for( int i = 0; i < fragCount; i++ )
{
string varName = string.Empty;
if( templateData.InterpolatorData.Interpolators[ i ].ExcludeStructPrefix )
{
varName = templateData.InterpolatorData.Interpolators[ i ].VarName;
}
else if( templateData.InterpolatorData.Interpolators[ i ].IsSingleComponent )
{
varName = string.Format( TemplateHelperFunctions.TemplateVarFormat,
templateData.FragmentFunctionData.InVarName,
templateData.InterpolatorData.Interpolators[ i ].VarNameWithSwizzle );
}
else
{
varName = string.Format( templateData.InterpolatorData.Interpolators[ i ].VarNameWithSwizzle, templateData.FragmentFunctionData.InVarName );
}
m_availableFragData.Add( templateData.InterpolatorData.Interpolators[ i ].DataInfo,
new InterpDataHelper( templateData.InterpolatorData.Interpolators[ i ].SwizzleType,
varName,
templateData.InterpolatorData.Interpolators[ i ].IsSingleComponent ) );
}
}
m_availableVertData = new Dictionary<TemplateInfoOnSematics, InterpDataHelper>();
if( templateData.VertexFunctionData != null && templateData.VertexDataList != null )
{
int vertCount = templateData.VertexDataList.Count;
for( int i = 0; i < vertCount; i++ )
{
string varName = string.Empty;
if( templateData.VertexDataList[ i ].ExcludeStructPrefix )
{
varName = templateData.VertexDataList[ i ].VarName;
}
else
{
varName = string.Format( TemplateHelperFunctions.TemplateVarFormat, templateData.VertexFunctionData.InVarName, templateData.VertexDataList[ i ].VarNameWithSwizzle );
}
m_availableVertData.Add( templateData.VertexDataList[ i ].DataInfo,
new InterpDataHelper( templateData.VertexDataList[ i ].SwizzleType,
varName,
templateData.VertexDataList[ i ].IsSingleComponent ) );
}
}
}
public void RegisterFragInputParams( WirePortDataType type, PrecisionType precision, string name, TemplateSemantics semantic, string custom )
{
if( m_fragmentInputParams == null )
m_fragmentInputParams = new Dictionary<TemplateSemantics, TemplateInputParameters>();
m_fragmentInputParams.Add( semantic, new TemplateInputParameters( type, precision, name, semantic, custom ) );
}
public void RegisterFragInputParams( WirePortDataType type, PrecisionType precision, string name, TemplateSemantics semantic )
{
if( m_fragmentInputParams == null )
m_fragmentInputParams = new Dictionary<TemplateSemantics, TemplateInputParameters>();
m_fragmentInputParams.Add( semantic, new TemplateInputParameters( type, precision, name, semantic ) );
}
public void RegisterVertexInputParams( WirePortDataType type, PrecisionType precision, string name, TemplateSemantics semantic )
{
if( m_vertexInputParams == null )
m_vertexInputParams = new Dictionary<TemplateSemantics, TemplateInputParameters>();
m_vertexInputParams.Add( semantic, new TemplateInputParameters( type, precision, name, semantic ) );
}
public string GetVertexId()
{
var precision = PrecisionType.Float;
bool useMasterNodeCategory = true;
MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment;
WirePortDataType type = WirePortDataType.UINT;
if( HasInfo( TemplateInfoOnSematics.VERTEXID, useMasterNodeCategory, customCategory ) )
{
InterpDataHelper info = GetInfo( TemplateInfoOnSematics.VERTEXID, useMasterNodeCategory, customCategory );
return TemplateHelperFunctions.AutoSwizzleData( info.VarName, info.VarType, type, true );
}
else
{
MasterNodePortCategory portCategory = useMasterNodeCategory ? m_currentDataCollector.PortCategory : customCategory;
string name = "ase_vertexID";
return RegisterInfoOnSemantic( portCategory, TemplateInfoOnSematics.VERTEXID, TemplateSemantics.SV_VertexID, name, WirePortDataType.UINT, precision, true );
}
// need to review this later
//if( m_vertexInputParams != null && m_vertexInputParams.ContainsKey( TemplateSemantics.SV_VertexID ) )
//{
// if( m_currentDataCollector.PortCategory == MasterNodePortCategory.Vertex )
// return m_vertexInputParams[ TemplateSemantics.SV_VertexID ].Name;
//}
//else
//{
// RegisterVertexInputParams( WirePortDataType.UINT, PrecisionType.Float, TemplateHelperFunctions.SemanticsDefaultName[ TemplateSemantics.SV_VertexID ], TemplateSemantics.SV_VertexID );
//}
//if( m_currentDataCollector.PortCategory != MasterNodePortCategory.Vertex )
// RegisterCustomInterpolatedData( m_vertexInputParams[ TemplateSemantics.SV_VertexID ].Name, WirePortDataType.INT, PrecisionType.Float, m_vertexInputParams[ TemplateSemantics.SV_VertexID ].Name );
//return m_vertexInputParams[ TemplateSemantics.SV_VertexID ].Name;
}
#if UNITY_EDITOR_WIN
public string GetPrimitiveId()
{
if( m_fragmentInputParams != null && m_fragmentInputParams.ContainsKey( TemplateSemantics.SV_PrimitiveID ) )
return m_fragmentInputParams[ TemplateSemantics.SV_PrimitiveID ].Name;
RegisterFragInputParams( WirePortDataType.UINT, PrecisionType.Half, TemplateHelperFunctions.SemanticsDefaultName[ TemplateSemantics.SV_PrimitiveID ], TemplateSemantics.SV_PrimitiveID );
return m_fragmentInputParams[ TemplateSemantics.SV_PrimitiveID ].Name;
}
#endif
public string GetVFace( int uniqueId )
{
#if UNITY_2018_3_OR_NEWER
if( IsHDRP && ASEPackageManagerHelper.CurrentHDVersion >= ASESRPVersions.ASE_SRP_6_9_0 )
{
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateInfoOnSematics.VFACE, WirePortDataType.FLOAT, PrecisionType.Float, ref result, true, MasterNodePortCategory.Fragment ) )
{
m_currentDataCollector.AddToDirectives( "#if !defined(ASE_NEED_CULLFACE)" );
m_currentDataCollector.AddToDirectives( "#define ASE_NEED_CULLFACE 1" );
m_currentDataCollector.AddToDirectives( "#endif //ASE_NEED_CULLFACE" );
return result;
}
else
{
if( m_fragmentInputParams != null && m_fragmentInputParams.ContainsKey( TemplateSemantics.VFACE ) )
return m_fragmentInputParams[ TemplateSemantics.VFACE ].Name;
string custom = "FRONT_FACE_TYPE "+ TemplateHelperFunctions.SemanticsDefaultName[ TemplateSemantics.VFACE ] + " : FRONT_FACE_SEMANTIC";
RegisterFragInputParams( WirePortDataType.FLOAT, PrecisionType.Half, TemplateHelperFunctions.SemanticsDefaultName[ TemplateSemantics.VFACE ], TemplateSemantics.VFACE, custom );
return m_fragmentInputParams[ TemplateSemantics.VFACE ].Name;
}
}
else
#endif
{
if( m_fragmentInputParams != null && m_fragmentInputParams.ContainsKey( TemplateSemantics.VFACE ) )
return m_fragmentInputParams[ TemplateSemantics.VFACE ].Name;
RegisterFragInputParams( WirePortDataType.FLOAT, PrecisionType.Half, TemplateHelperFunctions.SemanticsDefaultName[ TemplateSemantics.VFACE ], TemplateSemantics.VFACE );
return m_fragmentInputParams[ TemplateSemantics.VFACE ].Name;
}
}
public string GetShadowCoords( int uniqueId, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
// overriding precision
var precision = PrecisionType.Float;
string worldPos = GetWorldPos( false, m_currentDataCollector.PortCategory );
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateInfoOnSematics.SHADOWCOORDS, WirePortDataType.FLOAT4, precision, ref result, useMasterNodeCategory, customCategory ) )
{
return result;
}
string varName = GeneratorUtils.ShadowCoordsStr;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string shadowCoordsValue = string.Format( "TransformWorldToShadowCoord({0})", worldPos );
if( m_currentDataCollector.PortCategory == MasterNodePortCategory.Fragment )
{
worldPos = GetWorldPos( false, MasterNodePortCategory.Vertex );
m_currentDataCollector.AddLocalVariable( uniqueId, "#if defined(REQUIRES_VERTEX_SHADOW_COORD_INTERPOLATOR) //la" );
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT4, precision, string.Format( "TransformWorldToShadowCoord({0})", worldPos ), false, MasterNodePortCategory.Fragment );
m_currentDataCollector.AddLocalVariable( uniqueId, "#elif defined(MAIN_LIGHT_CALCULATE_SHADOWS) //la" );
m_currentDataCollector.AddLocalVariable( uniqueId, precision, WirePortDataType.FLOAT4, varName, shadowCoordsValue );
m_currentDataCollector.AddLocalVariable( uniqueId, "#else //la" );
m_currentDataCollector.AddLocalVariable( uniqueId, precision, WirePortDataType.FLOAT4, varName, "0" );
m_currentDataCollector.AddLocalVariable( uniqueId, "#endif //la" );
} else
{
m_currentDataCollector.AddLocalVariable( uniqueId, precision, WirePortDataType.FLOAT4, varName, shadowCoordsValue );
}
return varName;
}
public bool HasUV( int uvChannel )
{
return ( m_currentDataCollector.PortCategory == MasterNodePortCategory.Fragment ) ? m_availableFragData.ContainsKey( TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ] ) : m_availableVertData.ContainsKey( TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ] );
}
public string GetUVName( int uvChannel, WirePortDataType dataType = WirePortDataType.FLOAT2 )
{
InterpDataHelper info = ( m_currentDataCollector.PortCategory == MasterNodePortCategory.Fragment ) ? m_availableFragData[ TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ] ] : m_availableVertData[ TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ] ];
if( dataType != info.VarType )
return info.VarName + UIUtils.GetAutoSwizzle( dataType );
else
return info.VarName;
}
public string GetTextureCoord( int uvChannel, string propertyName, int uniqueId, PrecisionType precisionType )
{
bool isVertex = ( m_currentDataCollector.PortCategory == MasterNodePortCategory.Vertex || m_currentDataCollector.PortCategory == MasterNodePortCategory.Tessellation );
string uvChannelName = string.Empty;
string propertyHelperVar = propertyName + "_ST";
m_currentDataCollector.AddToUniforms( uniqueId, "float4", propertyHelperVar, IsSRP );
string uvName = string.Empty;
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ], WirePortDataType.FLOAT2, PrecisionType.Float, ref result, false, m_currentDataCollector.PortCategory ) )
{
uvName = result;
}
else
if( m_currentDataCollector.TemplateDataCollectorInstance.HasUV( uvChannel ) )
{
uvName = m_currentDataCollector.TemplateDataCollectorInstance.GetUVName( uvChannel );
}
else
{
uvName = m_currentDataCollector.TemplateDataCollectorInstance.RegisterUV( uvChannel );
}
uvChannelName = "uv" + propertyName;
if( isVertex )
{
string value = string.Format( Constants.TilingOffsetFormat, uvName, propertyHelperVar + ".xy", propertyHelperVar + ".zw" );
string lodLevel = "0";
value = "float4( " + value + ", 0 , " + lodLevel + " )";
m_currentDataCollector.AddLocalVariable( uniqueId, precisionType, WirePortDataType.FLOAT4, uvChannelName, value );
}
else
{
m_currentDataCollector.AddLocalVariable( uniqueId, precisionType, WirePortDataType.FLOAT2, uvChannelName, string.Format( Constants.TilingOffsetFormat, uvName, propertyHelperVar + ".xy", propertyHelperVar + ".zw" ) );
}
return uvChannelName;
}
public string GenerateAutoUVs( int uvChannel, WirePortDataType size = WirePortDataType.FLOAT2 )
{
string uvName = string.Empty;
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ], size, PrecisionType.Float, ref result, false, m_currentDataCollector.PortCategory ) )
{
uvName = result;
}
else
if( HasUV( uvChannel ) )
{
uvName = GetUVName( uvChannel, size );
}
else
{
uvName = RegisterUV( uvChannel, size );
}
return uvName;
}
public string GetUV( int uvChannel, MasterNodePortCategory category = MasterNodePortCategory.Fragment, WirePortDataType size = WirePortDataType.FLOAT4 )
{
if( !HasUV( uvChannel ) )
{
RegisterUV( uvChannel, size );
}
InterpDataHelper info = ( category == MasterNodePortCategory.Fragment ) ? m_availableFragData[ TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ] ] : m_availableVertData[ TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ] ];
return info.VarName;
}
public InterpDataHelper GetUVInfo( int uvChannel )
{
return ( m_currentDataCollector.PortCategory == MasterNodePortCategory.Fragment ) ? m_availableFragData[ TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ] ] : m_availableVertData[ TemplateHelperFunctions.IntToUVChannelInfo[ uvChannel ] ];
}
public string RegisterUV( int UVChannel, WirePortDataType size = WirePortDataType.FLOAT2 )
{
int channelsSize = TemplateHelperFunctions.DataTypeChannelUsage[ size ];
WirePortDataType originalSize = size;
if( m_UVUsage[ UVChannel ] > channelsSize )
{
size = TemplateHelperFunctions.ChannelToDataType[ m_UVUsage[ UVChannel ] ];
}
if( m_currentDataCollector.PortCategory == MasterNodePortCategory.Vertex )
{
TemplateSemantics semantic = TemplateHelperFunctions.IntToSemantic[ UVChannel ];
if( m_vertexDataDict.ContainsKey( semantic ) )
{
return m_vertexDataDict[ semantic ].VarName;
}
string varName = TemplateHelperFunctions.BaseInterpolatorName + ( ( UVChannel > 0 ) ? UVChannel.ToString() : string.Empty );
m_availableVertData.Add( TemplateHelperFunctions.IntToUVChannelInfo[ UVChannel ],
new InterpDataHelper( WirePortDataType.FLOAT4,
string.Format( TemplateHelperFunctions.TemplateVarFormat,
m_currentTemplateData.VertexFunctionData.InVarName,
varName ) ) );
m_currentDataCollector.AddToVertexInput(
string.Format( TemplateHelperFunctions.TexFullSemantic,
varName,
semantic ) );
RegisterOnVertexData( semantic, size, varName );
string finalVarName = m_availableVertData[ TemplateHelperFunctions.IntToUVChannelInfo[ UVChannel ] ].VarName;
switch( size )
{
case WirePortDataType.FLOAT:
case WirePortDataType.INT:
case WirePortDataType.UINT:
finalVarName += ".x";
break;
case WirePortDataType.FLOAT2:
finalVarName += ".xy";
break;
case WirePortDataType.FLOAT3:
finalVarName += ".xyz";
break;
case WirePortDataType.FLOAT4:
case WirePortDataType.COLOR:
case WirePortDataType.FLOAT3x3:
case WirePortDataType.FLOAT4x4:
case WirePortDataType.SAMPLER1D:
case WirePortDataType.SAMPLER2D:
case WirePortDataType.SAMPLER3D:
case WirePortDataType.SAMPLERCUBE:
case WirePortDataType.SAMPLER2DARRAY:
case WirePortDataType.SAMPLERSTATE:
case WirePortDataType.OBJECT:
default:
break;
}
return finalVarName;
}
else
{
//search if the correct vertex data is set ...
TemplateInfoOnSematics info = TemplateHelperFunctions.IntToInfo[ UVChannel ];
TemplateSemantics vertexSemantics = TemplateSemantics.NONE;
foreach( KeyValuePair<TemplateSemantics, TemplateVertexData> kvp in m_vertexDataDict )
{
if( kvp.Value.DataInfo == info )
{
vertexSemantics = kvp.Key;
break;
}
}
// if not, add vertex data and create interpolator
if( vertexSemantics == TemplateSemantics.NONE )
{
vertexSemantics = TemplateHelperFunctions.IntToSemantic[ UVChannel ];
if( !m_vertexDataDict.ContainsKey( vertexSemantics ) )
{
string varName = TemplateHelperFunctions.BaseInterpolatorName + ( ( UVChannel > 0 ) ? UVChannel.ToString() : string.Empty );
m_availableVertData.Add( TemplateHelperFunctions.IntToUVChannelInfo[ UVChannel ],
new InterpDataHelper( WirePortDataType.FLOAT4,
string.Format( TemplateHelperFunctions.TemplateVarFormat,
m_currentTemplateData.VertexFunctionData.InVarName,
varName ) ) );
m_currentDataCollector.AddToVertexInput(
string.Format( TemplateHelperFunctions.TexFullSemantic,
varName,
vertexSemantics ) );
RegisterOnVertexData( vertexSemantics, size, varName );
}
}
// either way create interpolator
TemplateVertexData availableInterp = RequestNewInterpolator( size, false );
if( availableInterp != null )
{
bool isPosition = vertexSemantics == TemplateSemantics.POSITION || vertexSemantics == TemplateSemantics.POSITION;
string interpVarName = m_currentTemplateData.VertexFunctionData.OutVarName + "." + availableInterp.VarNameWithSwizzle;
InterpDataHelper vertInfo = m_availableVertData[ TemplateHelperFunctions.IntToUVChannelInfo[ UVChannel ] ];
string interpDecl = string.Format( TemplateHelperFunctions.TemplateVariableDecl, interpVarName, TemplateHelperFunctions.AutoSwizzleData( vertInfo.VarName, vertInfo.VarType, size , isPosition ) );
m_currentDataCollector.AddToVertexInterpolatorsDecl( interpDecl );
string finalVarName = m_currentTemplateData.FragmentFunctionData.InVarName + "." + availableInterp.VarNameWithSwizzle;
m_availableFragData.Add( TemplateHelperFunctions.IntToUVChannelInfo[ UVChannel ], new InterpDataHelper( size, finalVarName ) );
if( size != originalSize )
{
//finalVarName = m_currentTemplateData.FragmentFunctionData.InVarName + "." + availableInterp.VarName + UIUtils.GetAutoSwizzle( originalSize );
finalVarName = m_availableFragData[ TemplateHelperFunctions.IntToUVChannelInfo[ UVChannel ] ].VarName + UIUtils.GetAutoSwizzle( originalSize );
}
return finalVarName;
}
}
return string.Empty;
}
////////////////////////////////////////////////////////////////////////////////////////////////
bool IsSemanticUsedOnInterpolator( TemplateSemantics semantics )
{
for( int i = 0; i < m_interpolatorData.Interpolators.Count; i++ )
{
if( m_interpolatorData.Interpolators[ i ].Semantics == semantics )
{
return true;
}
}
return false;
}
public bool HasInfo( TemplateInfoOnSematics info, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
MasterNodePortCategory category = useMasterNodeCategory ? m_currentDataCollector.PortCategory : customCategory;
return ( category == MasterNodePortCategory.Fragment ) ? m_availableFragData.ContainsKey( info ) : m_availableVertData.ContainsKey( info );
}
public InterpDataHelper GetInfo( TemplateInfoOnSematics info, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
MasterNodePortCategory category = useMasterNodeCategory ? m_currentDataCollector.PortCategory : customCategory;
if( category == MasterNodePortCategory.Fragment )
{
if( !m_availableFragData[ info ].SetAtCompileTime )
{
string defineValue = string.Empty;
if( TemplateHelperFunctions.InfoToDefineFrag.TryGetValue( info, out defineValue ) )
m_currentDataCollector.AddToDefines( -1, defineValue );
}
return m_availableFragData[ info ];
}
else
{
if( !m_availableVertData[ info ].SetAtCompileTime )
{
string defineValue = string.Empty;
if( TemplateHelperFunctions.InfoToDefineVertex.TryGetValue( info, out defineValue ) )
m_currentDataCollector.AddToDefines( -1, defineValue );
}
return m_availableVertData[ info ];
}
}
public string RegisterInfoOnSemantic( TemplateInfoOnSematics info, TemplateSemantics semantic, string name, WirePortDataType dataType, PrecisionType precisionType, bool requestNewInterpolator, string dataName = null )
{
return RegisterInfoOnSemantic( m_currentDataCollector.PortCategory, info, semantic, name, dataType, precisionType, requestNewInterpolator, dataName );
}
// This should only be used to semantics outside the text coord set
public string RegisterInfoOnSemantic( MasterNodePortCategory portCategory, TemplateInfoOnSematics info, TemplateSemantics semantic, string name, WirePortDataType dataType, PrecisionType precisionType, bool requestNewInterpolator, string dataName = null )
{
if( portCategory == MasterNodePortCategory.Vertex )
{
if( m_vertexDataDict.ContainsKey( semantic ) )
{
return m_vertexDataDict[ semantic ].VarName;
}
m_availableVertData.Add( info,
new InterpDataHelper( dataType,
string.Format( TemplateHelperFunctions.TemplateVarFormat,
m_currentTemplateData.VertexFunctionData.InVarName,
name ),true,true ) );
string vertInputVarType = UIUtils.PrecisionWirePortToCgType( precisionType, dataType );
m_currentDataCollector.AddToVertexInput(
string.Format( TemplateHelperFunctions.InterpFullSemantic,
vertInputVarType,
name,
semantic ) );
RegisterOnVertexData( semantic, dataType, name );
return m_availableVertData[ info ].VarName;
}
else
{
//search if the correct vertex data is set ...
TemplateSemantics vertexSemantics = TemplateSemantics.NONE;
foreach( KeyValuePair<TemplateSemantics, TemplateVertexData> kvp in m_vertexDataDict )
{
if( kvp.Value.DataInfo == info )
{
vertexSemantics = kvp.Key;
break;
}
}
// if not, add vertex data and create interpolator
if( vertexSemantics == TemplateSemantics.NONE )
{
vertexSemantics = semantic;
if( !m_vertexDataDict.ContainsKey( vertexSemantics ) )
{
m_availableVertData.Add( info,
new InterpDataHelper( dataType,
string.Format( TemplateHelperFunctions.TemplateVarFormat,
m_currentTemplateData.VertexFunctionData.InVarName,
name ),true,true ) );
string vertInputVarType = UIUtils.PrecisionWirePortToCgType( precisionType, dataType );
m_currentDataCollector.AddToVertexInput(
string.Format( TemplateHelperFunctions.InterpFullSemantic,
vertInputVarType,
name,
vertexSemantics ) );
RegisterOnVertexData( vertexSemantics, dataType, name );
}
}
// either way create interpolator
TemplateVertexData availableInterp = null;
if( requestNewInterpolator || IsSemanticUsedOnInterpolator( semantic ) )
{
availableInterp = RequestNewInterpolator( dataType, false, dataName );
}
else
{
availableInterp = RegisterOnInterpolator( semantic, dataType, dataName );
}
if( availableInterp != null )
{
bool isPosition = vertexSemantics == TemplateSemantics.POSITION || vertexSemantics == TemplateSemantics.POSITION;
string interpVarName = m_currentTemplateData.VertexFunctionData.OutVarName + "." + availableInterp.VarNameWithSwizzle;
string interpDecl = string.Format( TemplateHelperFunctions.TemplateVariableDecl, interpVarName, TemplateHelperFunctions.AutoSwizzleData( m_availableVertData[ info ].VarName, m_availableVertData[ info ].VarType, dataType, isPosition ) );
m_currentDataCollector.AddToVertexInterpolatorsDecl( interpDecl );
string finalVarName = m_currentTemplateData.FragmentFunctionData.InVarName + "." + availableInterp.VarNameWithSwizzle;
m_availableFragData.Add( info, new InterpDataHelper( dataType, finalVarName ) );
return finalVarName;
}
}
return string.Empty;
}
TemplateVertexData RegisterOnInterpolator( TemplateSemantics semantics, WirePortDataType dataType, string vertexDataName = null )
{
if( vertexDataName == null )
{
if( TemplateHelperFunctions.SemanticsDefaultName.ContainsKey( semantics ) )
{
vertexDataName = TemplateHelperFunctions.SemanticsDefaultName[ semantics ];
}
else
{
vertexDataName = string.Empty;
Debug.LogError( "No valid name given to vertex data" );
}
}
TemplateVertexData data = new TemplateVertexData( semantics, dataType, vertexDataName );
m_interpolatorData.Interpolators.Add( data );
string interpolator = string.Format( TemplateHelperFunctions.InterpFullSemantic, UIUtils.WirePortToCgType( dataType ), data.VarName, data.Semantics );
m_currentDataCollector.AddToInterpolators( interpolator );
return data;
}
public void RegisterOnVertexData( TemplateSemantics semantics, WirePortDataType dataType, string varName )
{
m_vertexDataDict.Add( semantics, new TemplateVertexData( semantics, dataType, varName ) );
}
public TemplateVertexData RequestMacroInterpolator( string varName )
{
if( varName != null && m_registeredVertexData.ContainsKey( varName ) )
{
return m_registeredVertexData[ varName ];
}
for( int i = 0; i < m_interpolatorData.AvailableInterpolators.Count; i++ )
{
if( !m_interpolatorData.AvailableInterpolators[ i ].IsFull )
{
TemplateVertexData data = m_interpolatorData.AvailableInterpolators[ i ].RequestChannels( WirePortDataType.FLOAT4, false, varName );
if( data != null )
{
if( !m_registeredVertexData.ContainsKey( data.VarName ) )
{
m_registeredVertexData.Add( data.VarName, data );
}
if( m_interpolatorData.AvailableInterpolators[ i ].Usage == 1 )
{
string interpolator = string.Format( TemplateHelperFunctions.InterpMacro, varName, TemplateHelperFunctions.SemanticToInt[ data.Semantics ] );
m_currentDataCollector.AddToInterpolators( interpolator );
}
return data;
}
}
}
return null;
}
public bool HasRawInterpolatorOfName( string name )
{
return m_interpolatorData.HasRawInterpolatorOfName( name );
}
public TemplateVertexData RequestNewInterpolator( WirePortDataType dataType, bool isColor, string varName = null , bool noInterpolationFlag = false, bool sampleFlag = false )
{
if( varName != null && m_registeredVertexData.ContainsKey( varName ) )
{
return m_registeredVertexData[ varName ];
}
for( int i = 0; i < m_interpolatorData.AvailableInterpolators.Count; i++ )
{
if( !m_interpolatorData.AvailableInterpolators[ i ].IsFull )
{
if( m_interpolatorData.AvailableInterpolators[ i ].Usage != 0 &&
(m_interpolatorData.AvailableInterpolators[ i ].NoInterpolation != noInterpolationFlag ||
m_interpolatorData.AvailableInterpolators[ i ].Sample != sampleFlag ))
continue;
TemplateVertexData data = m_interpolatorData.AvailableInterpolators[ i ].RequestChannels( dataType, isColor, varName );
if( data != null )
{
if( !m_registeredVertexData.ContainsKey( data.VarName ) )
{
m_registeredVertexData.Add( data.VarName, data );
}
if( m_interpolatorData.AvailableInterpolators[ i ].Usage == 1 )
{
m_interpolatorData.AvailableInterpolators[ i ].NoInterpolation = noInterpolationFlag;
m_interpolatorData.AvailableInterpolators[ i ].Sample = sampleFlag;
// First time using this interpolator, so we need to register it
string interpolator = string.Format( TemplateHelperFunctions.TexFullSemantic,
data.VarName, data.Semantics );
if( noInterpolationFlag )
interpolator = "nointerpolation " + interpolator;
if( sampleFlag)
interpolator = "sample " + interpolator;
m_currentDataCollector.AddToInterpolators( interpolator );
}
return data;
}
}
}
// This area is reached if max available interpolators from shader model is reached
// Nevertheless, we register all new interpolators to that list so no imediate compilation errors are thrown
// A warning message is then thrown to warn the user about this
int newInterpId = 1 + TemplateHelperFunctions.SemanticToInt[ m_interpolatorData.AvailableInterpolators[ m_interpolatorData.AvailableInterpolators.Count - 1 ].Semantic ];
if( TemplateHelperFunctions.IntToSemantic.ContainsKey( newInterpId ) )
{
TemplateInterpElement item = new TemplateInterpElement( TemplateHelperFunctions.IntToSemantic[ newInterpId ] );
m_interpolatorData.AvailableInterpolators.Add( item );
TemplateVertexData data = item.RequestChannels( dataType, isColor, varName );
if( data != null )
{
if( !m_registeredVertexData.ContainsKey( data.VarName ) )
{
m_registeredVertexData.Add( data.VarName, data );
}
if( item.Usage == 1 )
{
string interpolator = string.Format( TemplateHelperFunctions.TexFullSemantic, data.VarName, data.Semantics );
m_currentDataCollector.AddToInterpolators( interpolator );
}
return data;
}
}
UIUtils.ShowMessage( "Maximum amount of interpolators exceeded", MessageSeverity.Error );
return null;
}
// Unused channels in interpolators must be set to something so the compiler doesn't generate warnings
public List<string> GetInterpUnusedChannels()
{
List<string> resetInstrucctions = new List<string>();
if( m_interpolatorData != null )
{
for( int i = 0; i < m_interpolatorData.AvailableInterpolators.Count; i++ )
{
if( m_interpolatorData.AvailableInterpolators[ i ].Usage > 0 && !m_interpolatorData.AvailableInterpolators[ i ].IsFull )
{
string channels = string.Empty;
bool[] availableChannels = m_interpolatorData.AvailableInterpolators[ i ].AvailableChannels;
for( int j = 0; j < availableChannels.Length; j++ )
{
if( availableChannels[ j ] )
{
channels += TemplateHelperFunctions.VectorSwizzle[ j ];
}
}
resetInstrucctions.Add( string.Format( "{0}.{1}.{2} = 0;", m_currentTemplateData.VertexFunctionData.OutVarName, m_interpolatorData.AvailableInterpolators[ i ].Name, channels ) );
}
}
}
if( resetInstrucctions.Count > 0 )
{
resetInstrucctions.Insert( 0, "\n//setting value to unused interpolator channels and avoid initialization warnings" );
}
return resetInstrucctions;
}
public bool ContainsSpecialLocalFragVar( TemplateInfoOnSematics info, WirePortDataType type, ref string result )
{
if( m_specialFragmentLocalVars.ContainsKey( info ) )
{
result = m_specialFragmentLocalVars[ info ].LocalVarName;
if( m_specialFragmentLocalVars[ info ].DataType != type )
{
result = TemplateHelperFunctions.AutoSwizzleData( result, m_specialFragmentLocalVars[ info ].DataType, type, false );
}
return true;
}
return false;
}
public bool GetCustomInterpolatedData( TemplateInfoOnSematics info, WirePortDataType type, PrecisionType precisionType, ref string result, bool useMasterNodeCategory, MasterNodePortCategory customCategory )
{
bool isPosition = info == TemplateInfoOnSematics.POSITION ||
info == TemplateInfoOnSematics.CLIP_POS ||
info == TemplateInfoOnSematics.SCREEN_POSITION ||
info == TemplateInfoOnSematics.SCREEN_POSITION_NORMALIZED ||
info == TemplateInfoOnSematics.WORLD_POSITION ||
info == TemplateInfoOnSematics.RELATIVE_WORLD_POS;
MasterNodePortCategory category = useMasterNodeCategory ? m_currentDataCollector.PortCategory : customCategory;
if( category == MasterNodePortCategory.Vertex )
{
if( m_specialVertexLocalVars.ContainsKey( info ) )
{
result = m_specialVertexLocalVars[ info ].LocalVarName;
if( m_specialVertexLocalVars[ info ].DataType != type )
{
result = TemplateHelperFunctions.AutoSwizzleData( result, m_specialVertexLocalVars[ info ].DataType, type , isPosition );
}
string defineValue = string.Empty;
if( TemplateHelperFunctions.InfoToDefineVertex.TryGetValue( info, out defineValue ) )
m_currentDataCollector.AddToDefines( -1, defineValue );
return true;
}
}
if( category == MasterNodePortCategory.Fragment )
{
if( m_specialFragmentLocalVars.ContainsKey( info ) )
{
result = m_specialFragmentLocalVars[ info ].LocalVarName;
if( m_specialFragmentLocalVars[ info ].DataType != type )
{
result = TemplateHelperFunctions.AutoSwizzleData( result, m_specialFragmentLocalVars[ info ].DataType, type, isPosition );
}
string defineValue = string.Empty;
if( TemplateHelperFunctions.InfoToDefineFrag.TryGetValue( info, out defineValue ))
m_currentDataCollector.AddToDefines( -1, defineValue );
return true;
}
if( m_availableFragData.ContainsKey( info ) )
{
if( m_availableFragData[ info ].IsSingleComponent )
{
result = m_availableFragData[ info ].VarName;
if( m_availableFragData[ info ].VarType != type )
{
result = TemplateHelperFunctions.AutoSwizzleData( result, m_availableFragData[ info ].VarType, type, isPosition );
}
return true;
}
else if( TemplateHelperFunctions.InfoToLocalVar.ContainsKey( info ) && TemplateHelperFunctions.InfoToWirePortType.ContainsKey( info ) )
{
result = TemplateHelperFunctions.InfoToLocalVar[ info ];
m_currentDataCollector.AddLocalVariable( -1, precisionType, TemplateHelperFunctions.InfoToWirePortType[ info ], result, m_availableFragData[ info ].VarName );
return true;
}
}
}
return false;
}
public string GetVertexPosition( WirePortDataType type, PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
if( HasInfo( TemplateInfoOnSematics.POSITION, useMasterNodeCategory, customCategory ) )
{
InterpDataHelper info = GetInfo( TemplateInfoOnSematics.POSITION, useMasterNodeCategory, customCategory );
if( type != WirePortDataType.OBJECT && type != info.VarType )
return TemplateHelperFunctions.AutoSwizzleData( info.VarName, info.VarType, type,true );
else
return info.VarName;
}
else
{
MasterNodePortCategory portCategory = useMasterNodeCategory ? m_currentDataCollector.PortCategory : customCategory;
string name = "ase_vertex_pos";
string varName = RegisterInfoOnSemantic( portCategory, TemplateInfoOnSematics.POSITION, TemplateSemantics.POSITION, name, WirePortDataType.FLOAT4, precisionType, true );
if( type != WirePortDataType.OBJECT && type != WirePortDataType.FLOAT4 )
return TemplateHelperFunctions.AutoSwizzleData( varName, WirePortDataType.FLOAT4, type,true );
else
return varName;
}
}
private const string InstancingLibStandard = "UnityInstancing.cginc";
private const string InstancingLibSRP = "Packages/com.unity.render-pipelines.core/ShaderLibrary/UnityInstancing.hlsl";
public void SetupInstancing()
{
if( !HasInfo( TemplateInfoOnSematics.INSTANCE_ID ) )
{
m_currentDataCollector.AddToPragmas( -1, IOUtils.InstancedPropertiesHeader );
m_currentDataCollector.AddToIncludes( -1, IsSRP ? InstancingLibSRP : InstancingLibStandard );
m_currentDataCollector.AddToVertexInput( Constants.InstanceIdMacro );
m_currentDataCollector.AddToInterpolators( Constants.InstanceIdMacro );
m_currentDataCollector.AddToLocalVariables( MasterNodePortCategory.Vertex, -1, string.Format( "UNITY_SETUP_INSTANCE_ID({0});", m_currentTemplateData.VertexFunctionData.InVarName ) );
m_currentDataCollector.AddToLocalVariables( MasterNodePortCategory.Vertex, -1, string.Format( "UNITY_TRANSFER_INSTANCE_ID({0}, {1});", m_currentTemplateData.VertexFunctionData.InVarName, m_currentTemplateData.VertexFunctionData.OutVarName ) );
m_currentDataCollector.AddToLocalVariables( MasterNodePortCategory.Fragment, -1, string.Format( "UNITY_SETUP_INSTANCE_ID({0});", m_currentTemplateData.FragmentFunctionData.InVarName ) );
}
}
public string GetVertexColor( PrecisionType precisionType )
{
if( HasInfo( TemplateInfoOnSematics.COLOR ) )
{
return GetInfo( TemplateInfoOnSematics.COLOR ).VarName;
}
else
{
string name = "ase_color";
return RegisterInfoOnSemantic( TemplateInfoOnSematics.COLOR, TemplateSemantics.COLOR, name, WirePortDataType.FLOAT4, precisionType, false );
}
}
public string GetVertexNormal( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
if( HasInfo( TemplateInfoOnSematics.NORMAL, useMasterNodeCategory, customCategory ) )
{
InterpDataHelper info = GetInfo( TemplateInfoOnSematics.NORMAL, useMasterNodeCategory, customCategory );
return TemplateHelperFunctions.AutoSwizzleData( info.VarName, info.VarType, WirePortDataType.FLOAT3 , false);
}
else
{
MasterNodePortCategory category = useMasterNodeCategory ? m_currentDataCollector.PortCategory : customCategory;
string name = "ase_normal";
return RegisterInfoOnSemantic( category, TemplateInfoOnSematics.NORMAL, TemplateSemantics.NORMAL, name, WirePortDataType.FLOAT3, precisionType, false );
}
}
public string GetWorldNormal( PrecisionType precisionType = PrecisionType.Float, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment, bool normalize = false )
{
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_NORMAL, WirePortDataType.FLOAT3, precisionType, ref result, useMasterNodeCategory, customCategory ) )
{
if( normalize )
return string.Format( "normalize( {0} )", result );
else
return result;
}
string varName = normalize ? "normalizeWorldNormal" : GeneratorUtils.WorldNormalStr;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string worldNormalValue = string.Empty;
if( !GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_NORMAL, WirePortDataType.FLOAT3, precisionType, ref worldNormalValue, false, MasterNodePortCategory.Vertex ) )
{
string vertexNormal = GetVertexNormal( precisionType, false, MasterNodePortCategory.Vertex );
string formatStr = string.Empty;
if( IsSRP )
formatStr = "TransformObjectToWorldNormal({0})";
else
formatStr = "UnityObjectToWorldNormal({0})";
worldNormalValue = string.Format( formatStr, vertexNormal );
}
if( normalize )
worldNormalValue = string.Format( "normalize( {0} )", worldNormalValue );
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT3, precisionType, worldNormalValue, useMasterNodeCategory, customCategory );
return varName;
}
public string GetWorldNormal( int uniqueId, PrecisionType precisionType, string normal, string outputId )
{
string tanToWorld0 = string.Empty;
string tanToWorld1 = string.Empty;
string tanToWorld2 = string.Empty;
GetWorldTangentTf( precisionType, out tanToWorld0, out tanToWorld1, out tanToWorld2, true );
string tanNormal = "tanNormal" + outputId;
m_currentDataCollector.AddLocalVariable( uniqueId, "float3 " + tanNormal + " = " + normal + ";" );
return string.Format( "float3(dot({1},{0}), dot({2},{0}), dot({3},{0}))", tanNormal, tanToWorld0, tanToWorld1, tanToWorld2 );
}
public string GetVertexTangent( WirePortDataType type, PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
if( HasInfo( TemplateInfoOnSematics.TANGENT, useMasterNodeCategory, customCategory ) )
{
InterpDataHelper info = GetInfo( TemplateInfoOnSematics.TANGENT, useMasterNodeCategory, customCategory );
if( type != WirePortDataType.OBJECT && type != info.VarType )
return TemplateHelperFunctions.AutoSwizzleData( info.VarName, info.VarType, type , false);
else
return info.VarName;
}
else
{
MasterNodePortCategory category = useMasterNodeCategory ? m_currentDataCollector.PortCategory : customCategory;
string name = "ase_tangent";
string varName = RegisterInfoOnSemantic( category, TemplateInfoOnSematics.TANGENT, TemplateSemantics.TANGENT, name, WirePortDataType.FLOAT4, precisionType, false );
if( type != WirePortDataType.OBJECT && type != WirePortDataType.FLOAT4 )
return TemplateHelperFunctions.AutoSwizzleData( varName, WirePortDataType.FLOAT4, type , false );
else
return varName;
}
}
public string GetVertexBitangent( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string varName = GeneratorUtils.VertexBitangentStr;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string tangentValue = GetVertexTangent( WirePortDataType.FLOAT4, precisionType, false, MasterNodePortCategory.Vertex );
string normalValue = GetVertexNormal( precisionType, false, MasterNodePortCategory.Vertex );
string bitangentValue = string.Format( "cross( {0}, {1}.xyz ) * {1}.w * unity_WorldTransformParams.w", normalValue, tangentValue );
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT3, precisionType, bitangentValue, useMasterNodeCategory, customCategory );
return varName;
}
public string GetWorldTangent( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_TANGENT, WirePortDataType.FLOAT3, precisionType, ref result, useMasterNodeCategory, customCategory ) )
{
return result;
}
string varName = GeneratorUtils.WorldTangentStr;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string worldTangentValue = string.Empty;
if( !GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_TANGENT, WirePortDataType.FLOAT3, precisionType, ref worldTangentValue, false, MasterNodePortCategory.Vertex ) )
{
string vertexTangent = GetVertexTangent( WirePortDataType.FLOAT4, precisionType, false, MasterNodePortCategory.Vertex );
string formatStr = string.Empty;
if( IsSRP )
formatStr = "TransformObjectToWorldDir({0}.xyz)";
else
formatStr = "UnityObjectToWorldDir({0})";
worldTangentValue = string.Format( formatStr, vertexTangent );
}
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT3, precisionType, worldTangentValue, useMasterNodeCategory, customCategory );
return varName;
}
public string GetTangentSign( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string varName = GeneratorUtils.VertexTangentSignStr;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string tangentValue = GetVertexTangent( WirePortDataType.FLOAT4, precisionType, false, MasterNodePortCategory.Vertex );
string tangentSignValue = string.Format( "{0}.w * unity_WorldTransformParams.w", tangentValue );
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT, precisionType, tangentSignValue, useMasterNodeCategory, customCategory );
return varName;
}
public string GetWorldBinormal( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_BITANGENT, WirePortDataType.FLOAT3, precisionType, ref result, useMasterNodeCategory, customCategory ) )
{
return result;
}
string varName = GeneratorUtils.WorldBitangentStr;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string worldBinormal = string.Empty;
if( !GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_BITANGENT, WirePortDataType.FLOAT3, precisionType, ref worldBinormal, false, MasterNodePortCategory.Vertex ) )
{
string worldNormal = GetWorldNormal( precisionType, false, MasterNodePortCategory.Vertex );
string worldtangent = GetWorldTangent( precisionType, false, MasterNodePortCategory.Vertex );
string tangentSign = GetTangentSign( precisionType, false, MasterNodePortCategory.Vertex );
worldBinormal = string.Format( "cross( {0}, {1} ) * {2}", worldNormal, worldtangent, tangentSign );
}
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT3, PrecisionType.Float, worldBinormal, useMasterNodeCategory, customCategory );
return varName;
}
public string GetWorldReflection( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment, bool normalize = false )
{
string varName = GeneratorUtils.WorldReflectionStr;//UIUtils.GetInputValueFromType( SurfaceInputs.WORLD_REFL );
if( normalize )
varName = "normalized" + varName;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string worldNormal = GetWorldNormal( precisionType );
string worldViewDir = GetViewDir();
string worldRefl = string.Format( "reflect(-{0}, {1})", worldViewDir, worldNormal );
if( normalize )
worldRefl = string.Format( "normalize( {0} )", worldRefl );
m_currentDataCollector.AddLocalVariable( -1, precisionType, WirePortDataType.FLOAT3, varName, worldRefl );
return varName;
}
public string GetWorldReflection( PrecisionType precisionType, string normal )
{
string tanToWorld0 = string.Empty;
string tanToWorld1 = string.Empty;
string tanToWorld2 = string.Empty;
GetWorldTangentTf( precisionType, out tanToWorld0, out tanToWorld1, out tanToWorld2 );
string worldRefl = GetViewDir();
return string.Format( "reflect( -{0}, float3( dot( {2}, {1} ), dot( {3}, {1} ), dot( {4}, {1} ) ) )", worldRefl, normal, tanToWorld0, tanToWorld1, tanToWorld2 );
}
public string GetLightAtten( int uniqueId, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
//string result = string.Empty;
//if( GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_POSITION, PrecisionType.Float, ref result, useMasterNodeCategory, customCategory ) )
//{
// return result;
//}
//string varName = GeneratorUtils.WorldPositionStr;//UIUtils.GetInputValueFromType( SurfaceInputs.WORLD_POS );
//if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
// return varName;
//if( !m_availableVertData.ContainsKey( TemplateInfoOnSematics.POSITION ) )
//{
// UIUtils.ShowMessage( "Attempting to access inexisting vertex position to calculate world pos" );
// return "fixed3(0,0,0)";
//}
//string vertexPos = m_availableVertData[ TemplateInfoOnSematics.POSITION ].VarName;
//string worldPosConversion = string.Format( "mul(unity_ObjectToWorld, {0}).xyz", vertexPos );
//RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT3, PrecisionType.Float, worldPosConversion, useMasterNodeCategory, customCategory );
//return varName;
m_currentDataCollector.AddToIncludes( uniqueId, Constants.UnityAutoLightLib );
m_currentDataCollector.AddToDefines( uniqueId, "ASE_SHADOWS 1" );
#if UNITY_5_6_OR_NEWER
RequestMacroInterpolator( "UNITY_SHADOW_COORDS" );
#else
RequestMacroInterpolator( "SHADOW_COORDS" );
m_currentDataCollector.AddToPragmas( uniqueId, "multi_compile_fwdbase" );
#endif
//string vOutName = CurrentTemplateData.VertexFunctionData.OutVarName;
string fInName = CurrentTemplateData.FragmentFunctionData.InVarName;
string worldPos = GetWorldPos();
m_currentDataCollector.AddLocalVariable( uniqueId, "UNITY_LIGHT_ATTENUATION(ase_atten, " + fInName + ", " + worldPos + ")" );
return "ase_atten";
}
public string GenerateRotationIndependentObjectScale( ref MasterNodeDataCollector dataCollector, int uniqueId )
{
string value = string.Empty;
if( m_currentSRPType != TemplateSRPType.BuiltIn )
{
value = "float3( length( GetWorldToObjectMatrix()[ 0 ].xyz ), length( GetWorldToObjectMatrix()[ 1 ].xyz ), length( GetWorldToObjectMatrix()[ 2 ].xyz ) )";
}
else
{
value = "float3( length( unity_WorldToObject[ 0 ].xyz ), length( unity_WorldToObject[ 1 ].xyz ), length( unity_WorldToObject[ 2 ].xyz ) )";
}
value = "( 1.0 / "+ value +" )";
dataCollector.AddLocalVariable( uniqueId, PrecisionType.Float, WirePortDataType.FLOAT3, GeneratorUtils.ParentObjectScaleStr, value );
return GeneratorUtils.ParentObjectScaleStr;
}
public string GenerateObjectScale( ref MasterNodeDataCollector dataCollector, int uniqueId )
{
string value = string.Empty;
if( m_currentSRPType != TemplateSRPType.BuiltIn )
{
value = "float3( length( GetObjectToWorldMatrix()[ 0 ].xyz ), length( GetObjectToWorldMatrix()[ 1 ].xyz ), length( GetObjectToWorldMatrix()[ 2 ].xyz ) )";
}
else
{
value = "float3( length( unity_ObjectToWorld[ 0 ].xyz ), length( unity_ObjectToWorld[ 1 ].xyz ), length( unity_ObjectToWorld[ 2 ].xyz ) )";
}
dataCollector.AddLocalVariable( uniqueId, PrecisionType.Float, WirePortDataType.FLOAT3, GeneratorUtils.ObjectScaleStr, value );
return GeneratorUtils.ObjectScaleStr;
}
public string GetWorldPos( bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
// overriding precision
var precision = PrecisionType.Float;
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_POSITION, WirePortDataType.FLOAT3, precision, ref result, useMasterNodeCategory, customCategory ) )
{
return result;
}
else if( m_currentSRPType == TemplateSRPType.HD )
{
if( GetCustomInterpolatedData( TemplateInfoOnSematics.RELATIVE_WORLD_POS, WirePortDataType.FLOAT3, precision, ref result, useMasterNodeCategory, customCategory ) )
{
string worldPosVarName = GeneratorUtils.WorldPositionStr;
string relWorldPosConversion = string.Format( "GetAbsolutePositionWS( {0} )", result );
m_currentDataCollector.AddLocalVariable( -1, precision, WirePortDataType.FLOAT3, worldPosVarName, relWorldPosConversion );
return worldPosVarName;
}
}
string varName = GeneratorUtils.WorldPositionStr;//UIUtils.GetInputValueFromType( SurfaceInputs.WORLD_POS );
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
if( !m_availableVertData.ContainsKey( TemplateInfoOnSematics.POSITION ) )
{
UIUtils.ShowMessage( "Attempting to access inexisting vertex position to calculate world pos" );
return "half3(0,0,0)";
}
string vertexPos = m_availableVertData[ TemplateInfoOnSematics.POSITION ].VarName;
string worldPosConversion = string.Empty;
//Check if world pos already defined in the vertex body
if( !GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_POSITION, WirePortDataType.FLOAT3, precision, ref worldPosConversion, false, MasterNodePortCategory.Vertex ) )
{
if( m_currentSRPType == TemplateSRPType.HD )
{
#if UNITY_2018_3_OR_NEWER
worldPosConversion = string.Format( "GetAbsolutePositionWS( TransformObjectToWorld( ({0}).xyz ) )", vertexPos );
#else
worldPosConversion = string.Format( "GetAbsolutePositionWS( mul( GetObjectToWorldMatrix(), {0}).xyz )", vertexPos );
#endif
}
else if( m_currentSRPType == TemplateSRPType.Lightweight )
{
worldPosConversion = string.Format( "mul(GetObjectToWorldMatrix(), {0}).xyz", vertexPos );
}
else
{
worldPosConversion = string.Format( "mul(unity_ObjectToWorld, {0}).xyz", vertexPos );
}
}
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT3, precision, worldPosConversion, useMasterNodeCategory, customCategory );
return varName;
}
public string GetClipPosForValue( string customVertexPos, string outputId, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string varName = GeneratorUtils.ClipPositionStr + outputId;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
if( !m_availableVertData.ContainsKey( TemplateInfoOnSematics.POSITION ) )
{
UIUtils.ShowMessage( "Attempting to access inexisting vertex position to calculate clip pos" );
return "half4(0,0,0,0)";
}
string formatStr = string.Empty;
switch( m_currentSRPType )
{
default:
case TemplateSRPType.BuiltIn:
formatStr = "UnityObjectToClipPos({0})";
break;
case TemplateSRPType.HD:
formatStr = "TransformWorldToHClip( TransformObjectToWorld({0}))";
break;
case TemplateSRPType.Lightweight:
formatStr = "TransformObjectToHClip(({0}).xyz)";
break;
}
string clipSpaceConversion = string.Format( formatStr, customVertexPos );
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT4, PrecisionType.Float, clipSpaceConversion, useMasterNodeCategory, customCategory );
return varName;
}
public string GetClipPos( bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string varName = GeneratorUtils.ClipPositionStr;// "clipPos";
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
if( !m_availableVertData.ContainsKey( TemplateInfoOnSematics.POSITION ) )
{
UIUtils.ShowMessage( "Attempting to access inexisting vertex position to calculate clip pos" );
return "half4(0,0,0,0)";
}
string vertexPos = m_availableVertData[ TemplateInfoOnSematics.POSITION ].VarName;
string formatStr = string.Empty;
switch( m_currentSRPType )
{
default:
case TemplateSRPType.BuiltIn:
formatStr = "UnityObjectToClipPos({0})";
break;
case TemplateSRPType.HD:
formatStr = "TransformWorldToHClip( TransformObjectToWorld({0}))";
break;
case TemplateSRPType.Lightweight:
formatStr = "TransformObjectToHClip(({0}).xyz)";
break;
}
string clipSpaceConversion = string.Format( formatStr, vertexPos );
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT4, PrecisionType.Float, clipSpaceConversion, useMasterNodeCategory, customCategory );
return varName;
}
public string GetScreenPosForValue( PrecisionType precision, string customVertexPos, string outputId, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
// overriding precision
precision = PrecisionType.Float;
string varName = UIUtils.GetInputValueFromType( SurfaceInputs.SCREEN_POS ) + outputId;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string clipSpacePos = GetClipPosForValue( customVertexPos, outputId, false, MasterNodePortCategory.Vertex );
string screenPosConversion = string.Empty;
if( m_currentSRPType == TemplateSRPType.HD )
{
screenPosConversion = string.Format( "ComputeScreenPos( {0} , _ProjectionParams.x )", clipSpacePos );
}
else
{
screenPosConversion = string.Format( "ComputeScreenPos({0})", clipSpacePos );
}
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT4, precision, screenPosConversion, useMasterNodeCategory, customCategory );
return varName;
}
public string GetScreenPos( PrecisionType precision, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
// overriding precision
precision = PrecisionType.Float;
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateInfoOnSematics.SCREEN_POSITION, WirePortDataType.FLOAT4, precision, ref result, useMasterNodeCategory, customCategory ) )
{
return result;
}
string varName = UIUtils.GetInputValueFromType( SurfaceInputs.SCREEN_POS );
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string clipSpacePos = GetClipPos( false, MasterNodePortCategory.Vertex );
string screenPosConversion = string.Empty;
if( m_currentSRPType == TemplateSRPType.HD )
{
screenPosConversion = string.Format( "ComputeScreenPos( {0} , _ProjectionParams.x )", clipSpacePos );
}
else
{
screenPosConversion = string.Format( "ComputeScreenPos({0})", clipSpacePos );
}
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT4, precision, screenPosConversion, useMasterNodeCategory, customCategory );
return varName;
}
public string GetScreenPosNormalized( PrecisionType precision, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateInfoOnSematics.SCREEN_POSITION_NORMALIZED, WirePortDataType.FLOAT4, precision, ref result, useMasterNodeCategory, customCategory ) )
{
return result;
}
string varName = GeneratorUtils.ScreenPositionNormalizedStr;// "norm" + UIUtils.GetInputValueFromType( SurfaceInputs.SCREEN_POS );
string screenPos = GetScreenPos( precision, useMasterNodeCategory, customCategory );
string clipPlaneTestOp = string.Format( "{0}.z = ( UNITY_NEAR_CLIP_VALUE >= 0 ) ? {0}.z : {0}.z * 0.5 + 0.5;", varName );
m_currentDataCollector.AddLocalVariable( -1, precision, WirePortDataType.FLOAT4, varName, string.Format( GeneratorUtils.NormalizedScreenPosFormat, screenPos ) );
m_currentDataCollector.AddLocalVariable( -1, clipPlaneTestOp );
return varName;
}
public string GetViewDir( bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment, NormalizeType normalizeType = NormalizeType.Regular )
{
// overriding precision
var precision = PrecisionType.Float;
string result = string.Empty;
if( GetCustomInterpolatedData( TemplateInfoOnSematics.WORLD_VIEW_DIR, WirePortDataType.FLOAT3, precision, ref result, useMasterNodeCategory, customCategory ) )
return result;
string varName = GeneratorUtils.WorldViewDirectionStr;//UIUtils.GetInputValueFromType( SurfaceInputs.VIEW_DIR );
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string worldPos = GetWorldPos();
string formatStr = string.Empty;
if( IsSRP )
formatStr = "( _WorldSpaceCameraPos.xyz - {0} )";
else
formatStr = "UnityWorldSpaceViewDir({0})";
string viewDir = string.Format( formatStr, worldPos );
m_currentDataCollector.AddLocalVariable( -1, precision, WirePortDataType.FLOAT3, varName, viewDir );
switch( normalizeType )
{
default:
case NormalizeType.Off:
break;
case NormalizeType.Regular:
m_currentDataCollector.AddLocalVariable( -1, varName + " = normalize(" + varName + ");" );
break;
case NormalizeType.Safe:
m_currentDataCollector.AddLocalVariable( -1, varName + " = " + TemplateHelperFunctions.SafeNormalize( m_currentDataCollector, varName ) + ";" );
break;
}
//RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT3, PrecisionType.Float, viewDir, useMasterNodeCategory, customCategory );
return varName;
}
public string GetTangentViewDir( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment, NormalizeType normalizeType = NormalizeType.Regular )
{
string varName = GeneratorUtils.TangentViewDirectionStr;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string tanToWorld0 = string.Empty;
string tanToWorld1 = string.Empty;
string tanToWorld2 = string.Empty;
GetWorldTangentTf( precisionType, out tanToWorld0, out tanToWorld1, out tanToWorld2 );
string viewDir = GetViewDir();
string tanViewDir = string.Format( " {0} * {3}.x + {1} * {3}.y + {2} * {3}.z", tanToWorld0, tanToWorld1, tanToWorld2, viewDir );
m_currentDataCollector.AddLocalVariable( -1, precisionType, WirePortDataType.FLOAT3, varName, tanViewDir );
switch( normalizeType )
{
default:
case NormalizeType.Off: break;
case NormalizeType.Regular:
m_currentDataCollector.AddLocalVariable( -1, varName + " = normalize(" + varName + ");" );
break;
case NormalizeType.Safe:
m_currentDataCollector.AddLocalVariable( -1, varName + " = " + TemplateHelperFunctions.SafeNormalize( m_currentDataCollector, varName ) + ";" );
break;
}
return varName;
}
public void GetWorldTangentTf( PrecisionType precisionType, out string tanToWorld0, out string tanToWorld1, out string tanToWorld2, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
tanToWorld0 = "tanToWorld0";
tanToWorld1 = "tanToWorld1";
tanToWorld2 = "tanToWorld2";
if( HasCustomInterpolatedData( tanToWorld0, useMasterNodeCategory, customCategory ) ||
HasCustomInterpolatedData( tanToWorld1, useMasterNodeCategory, customCategory ) ||
HasCustomInterpolatedData( tanToWorld2, useMasterNodeCategory, customCategory ) )
return;
string worldTangent = GetWorldTangent( precisionType, useMasterNodeCategory, customCategory );
string worldNormal = GetWorldNormal( precisionType, useMasterNodeCategory, customCategory );
string worldBinormal = GetWorldBinormal( precisionType, useMasterNodeCategory, customCategory );
string tanToWorldVar0 = string.Format( "float3( {0}.x, {1}.x, {2}.x )", worldTangent, worldBinormal, worldNormal );
string tanToWorldVar1 = string.Format( "float3( {0}.y, {1}.y, {2}.y )", worldTangent, worldBinormal, worldNormal );
string tanToWorldVar2 = string.Format( "float3( {0}.z, {1}.z, {2}.z )", worldTangent, worldBinormal, worldNormal );
if( customCategory == MasterNodePortCategory.Vertex )
{
RegisterCustomInterpolatedData( tanToWorld0, WirePortDataType.FLOAT3, precisionType, tanToWorldVar0, useMasterNodeCategory, customCategory );
RegisterCustomInterpolatedData( tanToWorld1, WirePortDataType.FLOAT3, precisionType, tanToWorldVar1, useMasterNodeCategory, customCategory );
RegisterCustomInterpolatedData( tanToWorld2, WirePortDataType.FLOAT3, precisionType, tanToWorldVar2, useMasterNodeCategory, customCategory );
}
else
{
m_currentDataCollector.AddLocalVariable( -1, precisionType, WirePortDataType.FLOAT3, tanToWorld0, tanToWorldVar0 );
m_currentDataCollector.AddLocalVariable( -1, precisionType, WirePortDataType.FLOAT3, tanToWorld1, tanToWorldVar1 );
m_currentDataCollector.AddLocalVariable( -1, precisionType, WirePortDataType.FLOAT3, tanToWorld2, tanToWorldVar2 );
}
}
public string GetTangentToWorldMatrixFast( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string worldTangent = GetWorldTangent( precisionType );
string worldNormal = GetWorldNormal( precisionType );
string worldBinormal = GetWorldBinormal( precisionType );
string varName = GeneratorUtils.TangentToWorldFastStr;
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string result = string.Format( "float3x3({0}.x,{1}.x,{2}.x,{0}.y,{1}.y,{2}.y,{0}.z,{1}.z,{2}.z)", worldTangent, worldBinormal, worldNormal );
m_currentDataCollector.AddLocalVariable( -1, precisionType, WirePortDataType.FLOAT3x3, GeneratorUtils.TangentToWorldFastStr, result );
return GeneratorUtils.TangentToWorldFastStr;
}
public string GetTangentToWorldMatrixPrecise( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string worldToTangent = GetWorldToTangentMatrix( precisionType, useMasterNodeCategory, customCategory );
GeneratorUtils.Add3x3InverseFunction( ref m_currentDataCollector, UIUtils.PrecisionWirePortToCgType( precisionType, WirePortDataType.FLOAT ) );
m_currentDataCollector.AddLocalVariable( -1, precisionType, WirePortDataType.FLOAT3x3, GeneratorUtils.TangentToWorldPreciseStr, string.Format( GeneratorUtils.Inverse3x3Header, worldToTangent ) );
return GeneratorUtils.TangentToWorldPreciseStr;
}
public string GetWorldToTangentMatrix( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
string worldTangent = GetWorldTangent( precisionType );
string worldNormal = GetWorldNormal( precisionType );
string worldBinormal = GetWorldBinormal( precisionType );
string varName = GeneratorUtils.WorldToTangentStr;// "worldToTanMat";
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string worldTanMat = string.Format( "float3x3({0},{1},{2})", worldTangent, worldBinormal, worldNormal );
m_currentDataCollector.AddLocalVariable( -1, precisionType, WirePortDataType.FLOAT3x3, varName, worldTanMat );
return varName;
}
public string GetObjectToViewPos( PrecisionType precision, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
// overriding precision
precision = PrecisionType.Float;
string varName = "objectToViewPos";
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string vertexPos = GetVertexPosition( WirePortDataType.FLOAT3, precision, false, MasterNodePortCategory.Vertex );
string formatStr = string.Empty;
if( IsSRP )
formatStr = "TransformWorldToView(TransformObjectToWorld({0}))";
else
formatStr = "UnityObjectToViewPos({0})";
string objectToViewPosValue = string.Format( formatStr, vertexPos );
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT3, precision, objectToViewPosValue, useMasterNodeCategory, customCategory );
return varName;
}
public string GetEyeDepth( PrecisionType precision, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment, int viewSpace = 0 )
{
// overriding precision
precision = PrecisionType.Float;
string varName = "eyeDepth";
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string objectToView = GetObjectToViewPos( precision, false, MasterNodePortCategory.Vertex );
string eyeDepthValue = string.Format( "-{0}.z", objectToView );
if( viewSpace == 1 )
{
eyeDepthValue += " * _ProjectionParams.w";
}
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT, precision, eyeDepthValue, useMasterNodeCategory, customCategory );
return varName;
}
public string GetObjectSpaceLightDir( PrecisionType precisionType, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
if( !IsSRP )
{
m_currentDataCollector.AddToIncludes( -1, Constants.UnityLightingLib );
m_currentDataCollector.AddToIncludes( -1, Constants.UnityAutoLightLib );
}
string varName = "objectSpaceLightDir";
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string vertexPos = GetVertexPosition( WirePortDataType.FLOAT4, precisionType, false, MasterNodePortCategory.Vertex );
string objectSpaceLightDir = string.Empty;
switch( m_currentSRPType )
{
default:
case TemplateSRPType.BuiltIn:
objectSpaceLightDir = string.Format( "ObjSpaceLightDir({0})", vertexPos );
break;
case TemplateSRPType.HD:
string worldSpaceLightDir = GetWorldSpaceLightDir( precisionType, useMasterNodeCategory, customCategory );
objectSpaceLightDir = string.Format( "mul( GetWorldToObjectMatrix(), {0} ).xyz", worldSpaceLightDir );
break;
case TemplateSRPType.Lightweight:
objectSpaceLightDir = "mul( GetWorldToObjectMatrix(), _MainLightPosition ).xyz";
break;
}
RegisterCustomInterpolatedData( varName, WirePortDataType.FLOAT3, precisionType, objectSpaceLightDir, useMasterNodeCategory, customCategory );
return varName;
}
public string GetWorldSpaceLightDir( PrecisionType precision, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
if( !IsSRP )
{
m_currentDataCollector.AddToIncludes( -1, Constants.UnityLightingLib );
m_currentDataCollector.AddToIncludes( -1, Constants.UnityAutoLightLib );
AddLateDirective( AdditionalLineType.Custom, "//This is a late directive" );
}
else
{
string lightVar;
if( m_currentSRPType == TemplateSRPType.HD )
{
AddHDLightInfo();
lightVar = "-" + string.Format( TemplateHelperFunctions.HDLightInfoFormat, "0", "forward" );
}
else
{
lightVar = "_MainLightPosition.xyz";
}
return m_currentDataCollector.SafeNormalizeLightDir ? string.Format( "SafeNormalize({0})", lightVar ) : lightVar;
}
string varName = "worldSpaceLightDir";
if( HasCustomInterpolatedData( varName, useMasterNodeCategory, customCategory ) )
return varName;
string worldPos = GetWorldPos( useMasterNodeCategory, customCategory );
string worldSpaceLightDir = string.Format( "UnityWorldSpaceLightDir({0})", worldPos );
if( m_currentDataCollector.SafeNormalizeLightDir )
{
if( IsSRP )
{
worldSpaceLightDir = string.Format( "SafeNormalize{0})", worldSpaceLightDir );
}
else
{
m_currentDataCollector.AddToIncludes( -1, Constants.UnityBRDFLib );
worldSpaceLightDir = string.Format( "Unity_SafeNormalize({0})", worldSpaceLightDir );
}
}
m_currentDataCollector.AddLocalVariable( -1, precision, WirePortDataType.FLOAT3, varName, worldSpaceLightDir );
return varName;
}
public void RegisterCustomInterpolatedData( string name, WirePortDataType dataType, PrecisionType precision, string vertexInstruction, bool useMasterNodeCategory = true,
MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment, bool noInterpolationFlag = false, bool sampleFlag = false )
{
bool addLocalVariable = !name.Equals( vertexInstruction );
MasterNodePortCategory category = useMasterNodeCategory ? m_currentDataCollector.PortCategory : customCategory;
if( !m_customInterpolatedData.ContainsKey( name ) )
{
m_customInterpolatedData.Add( name, new TemplateCustomData( name, dataType ) );
}
if( !m_customInterpolatedData[ name ].IsVertex )
{
m_customInterpolatedData[ name ].IsVertex = true;
if( addLocalVariable )
m_currentDataCollector.AddToVertexLocalVariables( -1, precision, dataType, name, vertexInstruction );
}
if( category == MasterNodePortCategory.Fragment )
{
if( !m_customInterpolatedData[ name ].IsFragment )
{
m_customInterpolatedData[ name ].IsFragment = true;
TemplateVertexData interpData = RequestNewInterpolator( dataType, false,null, noInterpolationFlag,sampleFlag );
if( interpData == null )
{
Debug.LogErrorFormat( "Could not assign interpolator of type {0} to variable {1}", dataType, name );
return;
}
m_currentDataCollector.AddToVertexLocalVariables( -1, m_currentTemplateData.VertexFunctionData.OutVarName + "." + interpData.VarNameWithSwizzle, name );
m_currentDataCollector.AddToLocalVariables( -1, precision, dataType, name, m_currentTemplateData.FragmentFunctionData.InVarName + "." + interpData.VarNameWithSwizzle );
}
}
}
public bool HasCustomInterpolatedData( string name, bool useMasterNodeCategory = true, MasterNodePortCategory customCategory = MasterNodePortCategory.Fragment )
{
if( m_customInterpolatedData.ContainsKey( name ) )
{
MasterNodePortCategory category = useMasterNodeCategory ? m_currentDataCollector.PortCategory : customCategory;
return ( category == MasterNodePortCategory.Fragment ) ? m_customInterpolatedData[ name ].IsFragment : m_customInterpolatedData[ name ].IsVertex;
}
return false;
}
public bool HasFragmentInputParams
{
get
{
if( m_fragmentInputParams != null )
return m_fragmentInputParams.Count > 0;
return false;
}
}
public string FragInputParamsStr
{
get
{
string value = string.Empty;
if( m_fragmentInputParams != null && m_fragmentInputParams.Count > 0 )
{
int count = m_fragmentInputParams.Count;
if( count > 0 )
{
value = ", ";
foreach( KeyValuePair<TemplateSemantics, TemplateInputParameters> kvp in m_fragmentInputParams )
{
value += kvp.Value.Declaration;
if( --count > 0 )
{
value += " , ";
}
}
}
}
return value;
}
}
public string VertexInputParamsStr
{
get
{
string value = string.Empty;
if( m_vertexInputParams != null && m_vertexInputParams.Count > 0 )
{
int count = m_vertexInputParams.Count;
if( count > 0 )
{
value = ", ";
foreach( KeyValuePair<TemplateSemantics, TemplateInputParameters> kvp in m_vertexInputParams )
{
value += kvp.Value.Declaration;
if( --count > 0 )
{
value += " , ";
}
}
}
}
return value;
}
}
public void Destroy()
{
m_currentTemplate = null;
m_currentTemplateData = null;
m_currentDataCollector = null;
if( m_fullSrpBatcherPropertiesList != null )
{
m_fullSrpBatcherPropertiesList.Clear();
m_fullSrpBatcherPropertiesList = null;
}
if( m_srpBatcherPropertiesList != null )
{
m_srpBatcherPropertiesList.Clear();
m_srpBatcherPropertiesList = null;
}
if( m_srpBatcherPropertiesDict != null )
{
m_srpBatcherPropertiesDict.Clear();
m_srpBatcherPropertiesDict = null;
}
if( m_lateDirectivesList != null )
{
m_lateDirectivesList.Clear();
m_lateDirectivesList = null;
}
if( m_lateDirectivesDict != null )
{
m_lateDirectivesDict.Clear();
m_lateDirectivesDict = null;
}
if( m_registeredVertexData != null )
{
m_registeredVertexData.Clear();
m_registeredVertexData = null;
}
if( m_vertexInputParams != null )
{
m_vertexInputParams.Clear();
m_vertexInputParams = null;
}
if( m_fragmentInputParams != null )
{
m_fragmentInputParams.Clear();
m_fragmentInputParams = null;
}
if( m_vertexDataDict != null )
{
m_vertexDataDict.Clear();
m_vertexDataDict = null;
}
if( m_interpolatorData != null )
{
m_interpolatorData.Destroy();
m_interpolatorData = null;
}
if( m_availableFragData != null )
{
m_availableFragData.Clear();
m_availableFragData = null;
}
if( m_availableVertData != null )
{
m_availableVertData.Clear();
m_availableVertData = null;
}
if( m_customInterpolatedData != null )
{
m_customInterpolatedData.Clear();
m_customInterpolatedData = null;
}
if( m_specialVertexLocalVars != null )
{
m_specialVertexLocalVars.Clear();
m_specialVertexLocalVars = null;
}
if( m_specialFragmentLocalVars != null )
{
m_specialFragmentLocalVars.Clear();
m_specialFragmentLocalVars = null;
}
}
public void BuildCBuffer( int nodeId )
{
m_fullSrpBatcherPropertiesList.Clear();
if( m_srpBatcherPropertiesList.Count > 0 )
{
var regex = new Regex( @"(\d)\s+\b" );
m_srpBatcherPropertiesList.Sort( ( a, b ) =>
{
var matchA = regex.Match( a.PropertyName );
int sizeA = 0;
if( matchA.Groups.Count > 1 && matchA.Groups[ 1 ].Value.Length > 0 )
sizeA = Convert.ToInt32( matchA.Groups[ 1 ].Value, System.Globalization.CultureInfo.InvariantCulture );
var matchB = regex.Match( b.PropertyName );
int sizeB = 0;
if( matchB.Groups.Count > 1 && matchB.Groups[ 1 ].Value.Length > 0 )
sizeB = Convert.ToInt32( matchB.Groups[ 1 ].Value, System.Globalization.CultureInfo.InvariantCulture );
return sizeB.CompareTo( sizeA );
} );
m_fullSrpBatcherPropertiesList.Insert(0, new PropertyDataCollector( nodeId, IOUtils.SRPCBufferPropertiesBegin ));
m_fullSrpBatcherPropertiesList.AddRange( m_srpBatcherPropertiesList );
m_fullSrpBatcherPropertiesList.Add( new PropertyDataCollector( nodeId, IOUtils.SRPCBufferPropertiesEnd ) );
}
}
public void DumpSRPBatcher()
{
for( int i = 0; i < m_srpBatcherPropertiesList.Count; i++ )
{
Debug.Log( i + "::" + m_srpBatcherPropertiesList[ i ].PropertyName );
}
}
public const string GlobalMaxInterpolatorReachedMsg = "Maximum amount of interpolators reached!\nPlease consider optmizing your shader!";
public const string MaxInterpolatorSMReachedMsg = "Maximum amount of interpolators reached for current shader model on pass {0}! Please consider increasing the shader model to {1}!";
public void CheckInterpolatorOverflow( string currShaderModel, string passName )
{
int maxInterpolatorAmount = TemplateHelperFunctions.AvailableInterpolators[ currShaderModel ];
int currInterpolatorAmount = 1 + TemplateHelperFunctions.SemanticToInt[ InterpData.AvailableInterpolators[ InterpData.AvailableInterpolators.Count - 1 ].Semantic ];
if( currInterpolatorAmount > maxInterpolatorAmount )
{
string shaderModel = string.Empty;
if( TemplateHelperFunctions.GetShaderModelForInterpolatorAmount( currInterpolatorAmount, ref shaderModel ) )
{
UIUtils.ShowMessage( string.Format( MaxInterpolatorSMReachedMsg, passName, shaderModel ), MessageSeverity.Error );
}
else
{
UIUtils.ShowMessage( GlobalMaxInterpolatorReachedMsg, MessageSeverity.Error );
}
}
}
public Dictionary<TemplateSemantics, TemplateInputParameters> FragInputParameters { get { return m_fragmentInputParams; } }
public bool HasVertexInputParams
{
get
{
if( m_vertexInputParams != null )
return m_vertexInputParams.Count > 0;
return false;
}
}
public Dictionary<TemplateSemantics, TemplateInputParameters> VertexInputParameters { get { return m_vertexInputParams; } }
public TemplateData CurrentTemplateData { get { return m_currentTemplateData; } }
public int MultipassSubshaderIdx { get { return m_multipassSubshaderIdx; } }
public int MultipassPassIdx { get { return m_multipassPassIdx; } }
public TemplateSRPType CurrentSRPType { get { return m_currentSRPType; } set { m_currentSRPType = value; } }
public bool IsHDRP { get { return m_currentSRPType == TemplateSRPType.HD; } }
public bool IsLWRP { get { return m_currentSRPType == TemplateSRPType.Lightweight; } }
public bool IsSRP { get { return ( m_currentSRPType == TemplateSRPType.Lightweight || m_currentSRPType == TemplateSRPType.HD ); } }
public TemplateInterpData InterpData { get { return m_interpolatorData; } }
public List<PropertyDataCollector> LateDirectivesList { get { return m_lateDirectivesList; } }
public List<PropertyDataCollector> SrpBatcherPropertiesList { get { return m_srpBatcherPropertiesList; } }
public List<PropertyDataCollector> FullSrpBatcherPropertiesList { get { return m_fullSrpBatcherPropertiesList; } }
public Dictionary<TemplateSemantics, TemplateVertexData> VertexDataDict { get { return m_vertexDataDict; } }
}
}