using System.Collections.Generic ;
using System.Linq ;
using System.Text.RegularExpressions ;
using UnityEditor.Compilation ;
using UnityEngine ;
using UnityEditor.Graphing ;
using UnityEditor.Graphing.Util ;
[Serializable]
public abstract class AbstractMaterialGraph : SerializableGraph , IGenerateProperties
public abstract class AbstractMaterialGraph : IGraph , ISerializationCallbackReceiver , IGenerateProperties
public IGraphObject owner { get ; set ; }
#region Property data
public IEnumerable < IShaderProperty > properties
{
get { return m_Properties ; }
}
[SerializeField]
List < SerializationHelper . JSONSerializedElement > m_SerializedProperties = new List < SerializationHelper . JSONSerializedElement > ( ) ;
public IEnumerable < IShaderProperty > addedProperties
{
get { return m_AddedProperties ; }
}
public IEnumerable < Guid > removedProperties
{
get { return m_RemovedProperties ; }
}
#endregion
#region Node data
public InspectorPreviewData previewData = new InspectorPreviewData ( ) ;
Dictionary < Guid , INode > m_Nodes = new Dictionary < Guid , INode > ( ) ;
public Mesh previewMesh
public IEnumerable < T > GetNodes < T > ( ) where T : INode
get { return previewData . mesh ; }
set { previewData . mesh = value ; }
return m_Nodes . Values . OfType < T > ( ) ;
public IEnumerable < IShaderProperty > properties
[SerializeField]
List < SerializationHelper . JSONSerializedElement > m_SerializableNodes = new List < SerializationHelper . JSONSerializedElement > ( ) ;
[NonSerialized]
List < INode > m_AddedNodes = new List < INode > ( ) ;
public IEnumerable < INode > addedNodes
get { return m_Properties ; }
get { return m_AddedNodes ; }
public IEnumerable < IShaderProperty > addedProperties
[NonSerialized]
List < INode > m_RemovedNodes = new List < INode > ( ) ;
public IEnumerable < INode > removedNodes
get { return m_AddedProperties ; }
get { return m_RemovedNodes ; }
public IEnumerable < Guid > removedProperties
#endregion
#region Edge data
[NonSerialized]
List < IEdge > m_Edges = new List < IEdge > ( ) ;
public IEnumerable < IEdge > edges
get { return m_RemovedProperties ; }
get { return m_Edges ; }
public override void ClearChanges ( )
[SerializeField]
List < SerializationHelper . JSONSerializedElement > m_SerializableEdges = new List < SerializationHelper . JSONSerializedElement > ( ) ;
[NonSerialized]
Dictionary < Guid , List < IEdge > > m_NodeEdges = new Dictionary < Guid , List < IEdge > > ( ) ;
[NonSerialized]
List < IEdge > m_AddedEdges = new List < IEdge > ( ) ;
public IEnumerable < IEdge > addedEdges
base . ClearChanges ( ) ;
get { return m_AddedEdges ; }
}
[NonSerialized]
List < IEdge > m_RemovedEdges = new List < IEdge > ( ) ;
public IEnumerable < IEdge > removedEdges
{
get { return m_RemovedEdges ; }
}
#endregion
[SerializeField]
InspectorPreviewData m_PreviewData = new InspectorPreviewData ( ) ;
public InspectorPreviewData previewData
{
get { return m_PreviewData ; }
set { m_PreviewData = value ; }
}
public void ClearChanges ( )
{
m_AddedNodes . Clear ( ) ;
m_RemovedNodes . Clear ( ) ;
m_AddedEdges . Clear ( ) ;
m_RemovedEdges . Clear ( ) ;
public override void AddNode ( INode node )
public virtual void AddNode ( INode node )
base . AddNode ( node ) ;
AddNodeNoValidate ( node ) ;
ValidateGraph ( ) ;
}
else
{
void AddNodeNoValidate ( INode node )
{
m_Nodes . Add ( node . guid , node ) ;
node . owner = this ;
m_AddedNodes . Add ( node ) ;
}
public void RemoveNode ( INode node )
{
if ( ! node . canDeleteNode )
return ;
m_Nodes . Remove ( node . guid ) ;
m_RemovedNodes . Add ( node ) ;
ValidateGraph ( ) ;
}
void RemoveNodeNoValidate ( INode node )
{
if ( ! node . canDeleteNode )
return ;
m_Nodes . Remove ( node . guid ) ;
m_RemovedNodes . Add ( node ) ;
}
void AddEdgeToNodeEdges ( IEdge edge )
{
List < IEdge > inputEdges ;
if ( ! m_NodeEdges . TryGetValue ( edge . inputSlot . nodeGuid , out inputEdges ) )
m_NodeEdges [ edge . inputSlot . nodeGuid ] = inputEdges = new List < IEdge > ( ) ;
inputEdges . Add ( edge ) ;
List < IEdge > outputEdges ;
if ( ! m_NodeEdges . TryGetValue ( edge . outputSlot . nodeGuid , out outputEdges ) )
m_NodeEdges [ edge . outputSlot . nodeGuid ] = outputEdges = new List < IEdge > ( ) ;
outputEdges . Add ( edge ) ;
}
IEdge ConnectNoValidate ( SlotReference fromSlotRef , SlotReference toSlotRef )
{
var fromNode = GetNodeFromGuid ( fromSlotRef . nodeGuid ) ;
var toNode = GetNodeFromGuid ( toSlotRef . nodeGuid ) ;
if ( fromNode = = null | | toNode = = null )
return null ;
// if fromNode is already connected to toNode
// do now allow a connection as toNode will then
// have an edge to fromNode creating a cycle.
// if this is parsed it will lead to an infinite loop.
var dependentNodes = new List < INode > ( ) ;
NodeUtils . CollectNodesNodeFeedsInto ( dependentNodes , toNode ) ;
if ( dependentNodes . Contains ( fromNode ) )
return null ;
var fromSlot = fromNode . FindSlot < ISlot > ( fromSlotRef . slotId ) ;
var toSlot = toNode . FindSlot < ISlot > ( toSlotRef . slotId ) ;
if ( fromSlot . isOutputSlot = = toSlot . isOutputSlot )
return null ;
var outputSlot = fromSlot . isOutputSlot ? fromSlotRef : toSlotRef ;
var inputSlot = fromSlot . isInputSlot ? fromSlotRef : toSlotRef ;
s_TempEdges . Clear ( ) ;
GetEdges ( inputSlot , s_TempEdges ) ;
// remove any inputs that exits before adding
foreach ( var edge in s_TempEdges )
{
RemoveEdgeNoValidate ( edge ) ;
}
var newEdge = new Edge ( outputSlot , inputSlot ) ;
m_Edges . Add ( newEdge ) ;
m_AddedEdges . Add ( newEdge ) ;
AddEdgeToNodeEdges ( newEdge ) ;
//Debug.LogFormat("Connected edge: {0} -> {1} ({2} -> {3})\n{4}", newEdge.outputSlot.nodeGuid, newEdge.inputSlot.nodeGuid, fromNode.name, toNode.name, Environment.StackTrace);
return newEdge ;
}
public virtual IEdge Connect ( SlotReference fromSlotRef , SlotReference toSlotRef )
{
var newEdge = ConnectNoValidate ( fromSlotRef , toSlotRef ) ;
ValidateGraph ( ) ;
return newEdge ;
}
public virtual void RemoveEdge ( IEdge e )
{
RemoveEdgeNoValidate ( e ) ;
ValidateGraph ( ) ;
}
public void RemoveElements ( IEnumerable < INode > nodes , IEnumerable < IEdge > edges )
{
foreach ( var edge in edges . ToArray ( ) )
RemoveEdgeNoValidate ( edge ) ;
foreach ( var serializableNode in nodes . ToArray ( ) )
RemoveNodeNoValidate ( serializableNode ) ;
ValidateGraph ( ) ;
}
protected void RemoveEdgeNoValidate ( IEdge e )
{
e = m_Edges . FirstOrDefault ( x = > x . Equals ( e ) ) ;
if ( e = = null )
throw new ArgumentException ( "Trying to remove an edge that does not exist." , "e" ) ;
m_Edges . Remove ( e ) ;
List < IEdge > inputNodeEdges ;
if ( m_NodeEdges . TryGetValue ( e . inputSlot . nodeGuid , out inputNodeEdges ) )
inputNodeEdges . Remove ( e ) ;
List < IEdge > outputNodeEdges ;
if ( m_NodeEdges . TryGetValue ( e . outputSlot . nodeGuid , out outputNodeEdges ) )
outputNodeEdges . Remove ( e ) ;
m_RemovedEdges . Add ( e ) ;
}
public INode GetNodeFromGuid ( Guid guid )
{
INode node ;
m_Nodes . TryGetValue ( guid , out node ) ;
return node ;
}
public bool ContainsNodeGuid ( Guid guid )
{
return m_Nodes . ContainsKey ( guid ) ;
}
public T GetNodeFromGuid < T > ( Guid guid ) where T : INode
{
var node = GetNodeFromGuid ( guid ) ;
if ( node is T )
return ( T ) node ;
return default ( T ) ;
}
public void GetEdges ( SlotReference s , List < IEdge > foundEdges )
{
var node = GetNodeFromGuid ( s . nodeGuid ) ;
if ( node = = null )
{
Debug . LogWarning ( "Node does not exist" ) ;
return ;
}
ISlot slot = node . FindSlot < ISlot > ( s . slotId ) ;
List < IEdge > candidateEdges ;
if ( ! m_NodeEdges . TryGetValue ( s . nodeGuid , out candidateEdges ) )
return ;
foreach ( var edge in candidateEdges )
{
var cs = slot . isInputSlot ? edge . inputSlot : edge . outputSlot ;
if ( cs . nodeGuid = = s . nodeGuid & & cs . slotId = = s . slotId )
foundEdges . Add ( edge ) ;
}
}
public virtual void CollectShaderProperties ( PropertyCollector collector , GenerationMode generationMode )
{
foreach ( var prop in properties )
public virtual void AddShaderProperty ( IShaderProperty property )
public void AddShaderProperty ( IShaderProperty property )
{
if ( property = = null )
return ;
public void RemoveShaderProperty ( Guid guid )
{
var propertyNodes = GetNodes < PropertyNode > ( ) . Where ( x = > x . propertyGuid = = guid ) . ToList ( ) ;
foreach ( var propNode in propertyNodes )
ReplacePropertyNodeWithConcreteNodeNoValidate ( propNode ) ;
var propertyNodes = GetNodes < PropertyNode > ( ) . Where ( x = > x . propertyGuid = = guid ) . ToList ( ) ;
foreach ( var propNode in propertyNodes )
ReplacePropertyNodeWithConcreteNode ( propNode ) ;
if ( m_Properties . RemoveAll ( x = > x . guid = = guid ) > 0 )
m_RemovedProperties . Add ( guid ) ;
}
public void ReplacePropertyNodeWithConcreteNode ( PropertyNode propertyNode )
{
var property = properties . FirstOrDefault ( x = > x . guid = = propertyNode . propertyGuid ) ;
if ( property ! = null )
{
AbstractMaterialNode node = null ;
int slotId = - 1 ;
if ( property is FloatShaderProperty )
{
var createdNode = new Vector1Node ( ) ;
createdNode . value = ( ( FloatShaderProperty ) property ) . value ;
slotId = Vector1Node . OutputSlotId ;
node = createdNode ;
ReplacePropertyNodeWithConcreteNodeNoValidate ( propertyNode ) ;
ValidateGraph ( ) ;
else if ( property is Vector2ShaderProperty )
void ReplacePropertyNodeWithConcreteNodeNoValidate ( PropertyNode propertyNode )
var createdNode = new Vector2Node ( ) ;
createdNode . value = ( ( Vector2ShaderProperty ) property ) . value ;
slotId = Vector2Node . OutputSlotId ;
node = createdNode ;
}
else if ( property is Vector3ShaderProperty )
{
var createdNode = new Vector3Node ( ) ;
createdNode . value = ( ( Vector3ShaderProperty ) property ) . value ;
slotId = Vector3Node . OutputSlotId ;
node = createdNode ;
}
else if ( property is Vector4ShaderProperty )
{
var createdNode = new Vector4Node ( ) ;
createdNode . value = ( ( Vector4ShaderProperty ) property ) . value ;
slotId = Vector4Node . OutputSlotId ;
node = createdNode ;
}
else if ( property is ColorShaderProperty )
{
var createdNode = new ColorNode ( ) ;
createdNode . color = ( ( ColorShaderProperty ) property ) . value ;
slotId = ColorNode . OutputSlotId ;
node = createdNode ;
}
else if ( property is TextureShaderProperty )
{
var createdNode = new Texture2DAssetNode ( ) ;
createdNode . texture = ( ( TextureShaderProperty ) property ) . value . texture ;
slotId = Texture2DAssetNode . OutputSlotId ;
node = createdNode ;
}
else if ( property is CubemapShaderProperty )
{
var createdNode = new CubemapAssetNode ( ) ;
createdNode . cubemap = ( ( CubemapShaderProperty ) property ) . value . cubemap ;
slotId = CubemapAssetNode . OutputSlotId ;
node = createdNode ;
}
var property = properties . FirstOrDefault ( x = > x . guid = = propertyNode . propertyGuid ) ;
if ( property = = null )
return ;
if ( node = = null )
var node = property . ToConcreteNode ( ) ;
if ( ! ( node is AbstractMaterialNode ) )
var newSlot = node . GetOutputSlots < MaterialSlot > ( ) . FirstOrDefault ( s = > s . valueType = = slot . valueType ) ;
if ( newSlot = = null )
return ;
s_TempEdges . Clear ( ) ;
GetEdges ( slot . slotReference , s_TempEdges ) ;
foreach ( var edge in s_TempEdges )
ConnectNoValidate ( node . GetSlotReference ( slotId ) , edge . inputSlot ) ;
foreach ( var edge in this . GetEdges ( slot . slotReference ) )
ConnectNoValidate ( newSlot . slotReference , edge . inputSlot ) ;
}
public override void ValidateGraph ( )
public void ValidateGraph ( )
ReplacePropertyNodeWithConcreteNode ( pNode ) ;
base . ValidateGraph ( ) ;
}
ReplacePropertyNodeWithConcreteNodeNoValidate ( pNode ) ;
public override Dictionary < SerializationHelper . TypeSerializationInfo , SerializationHelper . TypeSerializationInfo > GetLegacyTypeRemapping ( )
//First validate edges, remove any
//orphans. This can happen if a user
//manually modifies serialized data
//of if they delete a node in the inspector
//debug view.
foreach ( var edge in edges . ToArray ( ) )
var result = base . GetLegacyTypeRemapping ( ) ;
var viewNode = new SerializationHelper . TypeSerializationInfo
{
fullName = "UnityEngine.MaterialGraph.ViewDirectionNode"
} ;
result [ viewNode ] = SerializationHelper . GetTypeSerializableAsString ( typeof ( ViewDirectionNode ) ) ;
var outputNode = GetNodeFromGuid ( edge . outputSlot . nodeGuid ) ;
var inputNode = GetNodeFromGuid ( edge . inputSlot . nodeGuid ) ;
var normalNode = new SerializationHelper . TypeSerializationInfo
if ( outputNode = = null
| | inputNode = = null
| | outputNode . FindOutputSlot < ISlot > ( edge . outputSlot . slotId ) = = null
| | inputNode . FindInputSlot < ISlot > ( edge . inputSlot . slotId ) = = null )
fullName = "UnityEngine.MaterialGraph.NormalNode"
} ;
result [ normalNode ] = SerializationHelper . GetTypeSerializableAsString ( typeof ( NormalVectorNode ) ) ;
//orphaned edge
RemoveEdgeNoValidate ( edge ) ;
}
}
var worldPosNode = new SerializationHelper . TypeSerializationInfo
{
fullName = "UnityEngine.MaterialGraph.WorldPosNode"
} ;
result [ worldPosNode ] = SerializationHelper . GetTypeSerializableAsString ( typeof ( PositionNode ) ) ;
foreach ( var node in GetNodes < INode > ( ) )
node . ValidateNode ( ) ;
return result ;
foreach ( var edge in m_AddedEdges . ToList ( ) )
{
if ( ! ContainsNodeGuid ( edge . outputSlot . nodeGuid ) | | ! ContainsNodeGuid ( edge . inputSlot . nodeGuid ) )
{
Debug . LogWarningFormat ( "Added edge is invalid: {0} -> {1}\n{2}" , edge . outputSlot . nodeGuid , edge . inputSlot . nodeGuid , Environment . StackTrace ) ;
m_AddedEdges . Remove ( edge ) ;
}
}
public override void ReplaceWith ( IGraph other )
{
var otherMG = other as AbstractMaterialGraph ;
if ( otherMG ! = null )
public void ReplaceWith ( IGraph other )
var otherMg = other as AbstractMaterialGraph ;
if ( otherMg = = null )
throw new ArgumentException ( "Can only replace with another AbstractMaterialGraph" , "other" ) ;
using ( var removedPropertiesPooledObject = ListPool < Guid > . GetDisposable ( ) )
{
var removedPropertyGuids = removedPropertiesPooledObject . value ;
RemoveShaderPropertyNoValidate ( propertyGuid ) ;
}
foreach ( var otherProperty in otherMG . properties )
foreach ( var otherProperty in otherMg . properties )
}
base . ReplaceWith ( other ) ;
}
public override void OnBeforeSerialize ( )
{
base . OnBeforeSerialize ( ) ;
m_SerializedProperties = SerializationHelper . Serialize < IShaderProperty > ( m_Properties ) ;
}
other . ValidateGraph ( ) ;
ValidateGraph ( ) ;
public override void OnAfterDeserialize ( )
// Current tactic is to remove all nodes and edges and then re-add them, such that depending systems
// will re-initialize with new references.
using ( var pooledList = ListPool < IEdge > . GetDisposable ( ) )
// have to deserialize 'globals' before nodes
m_Properties = SerializationHelper . Deserialize < IShaderProperty > ( m_SerializedProperties , null ) ;
base . OnAfterDeserialize ( ) ;
var removedNodeEdges = pooledList . value ;
removedNodeEdges . AddRange ( m_Edges ) ;
foreach ( var edge in removedNodeEdges )
RemoveEdgeNoValidate ( edge ) ;
internal static ShaderGraphRequirements GetRequirements ( List < INode > nodes )
{
NeededCoordinateSpace requiresNormal = nodes . OfType < IMayRequireNormal > ( ) . Aggregate ( NeededCoordinateSpace . None , ( mask , node ) = > mask | node . RequiresNormal ( ) ) ;
NeededCoordinateSpace requiresBitangent = nodes . OfType < IMayRequireBitangent > ( ) . Aggregate ( NeededCoordinateSpace . None , ( mask , node ) = > mask | node . RequiresBitangent ( ) ) ;
NeededCoordinateSpace requiresTangent = nodes . OfType < IMayRequireTangent > ( ) . Aggregate ( NeededCoordinateSpace . None , ( mask , node ) = > mask | node . RequiresTangent ( ) ) ;
NeededCoordinateSpace requiresViewDir = nodes . OfType < IMayRequireViewDirection > ( ) . Aggregate ( NeededCoordinateSpace . None , ( mask , node ) = > mask | node . RequiresViewDirection ( ) ) ;
NeededCoordinateSpace requiresPosition = nodes . OfType < IMayRequirePosition > ( ) . Aggregate ( NeededCoordinateSpace . None , ( mask , node ) = > mask | node . RequiresPosition ( ) ) ;
bool requiresScreenPosition = nodes . OfType < IMayRequireScreenPosition > ( ) . Any ( x = > x . RequiresScreenPosition ( ) ) ;
bool requiresVertexColor = nodes . OfType < IMayRequireVertexColor > ( ) . Any ( x = > x . RequiresVertexColor ( ) ) ;
var meshUV = new List < UVChannel > ( ) ;
for ( int uvIndex = 0 ; uvIndex < ShaderGeneratorNames . UVCount ; + + uvIndex )
using ( var removedNodesPooledObject = ListPool < Guid > . GetDisposable ( ) )
var channel = ( UVChannel ) uvIndex ;
if ( nodes . OfType < IMayRequireMeshUV > ( ) . Any ( x = > x . RequiresMeshUV ( channel ) ) )
meshUV . Add ( channel ) ;
var removedNodeGuids = removedNodesPooledObject . value ;
removedNodeGuids . AddRange ( m_Nodes . Keys ) ;
foreach ( var nodeGuid in removedNodeGuids )
RemoveNodeNoValidate ( m_Nodes [ nodeGuid ] ) ;
// if anything needs tangentspace we have make
// sure to have our othonormal basis!
var compoundSpaces = requiresBitangent | requiresNormal | requiresPosition
| requiresTangent | requiresViewDir | requiresPosition
| requiresNormal ;
ValidateGraph ( ) ;
var needsTangentSpace = ( compoundSpaces & NeededCoordinateSpace . Tangent ) > 0 ;
if ( needsTangentSpace )
{
requiresBitangent | = NeededCoordinateSpace . Object ;
requiresNormal | = NeededCoordinateSpace . Object ;
requiresTangent | = NeededCoordinateSpace . Object ;
}
var reqs = new ShaderGraphRequirements ( )
{
requiresNormal = requiresNormal ,
requiresBitangent = requiresBitangent ,
requiresTangent = requiresTangent ,
requiresViewDir = requiresViewDir ,
requiresPosition = requiresPosition ,
requiresScreenPosition = requiresScreenPosition ,
requiresVertexColor = requiresVertexColor ,
requiresMeshUVs = meshUV
} ;
foreach ( var node in other . GetNodes < INode > ( ) )
AddNodeNoValidate ( node ) ;
return reqs ;
}
foreach ( var edge in other . edges )
ConnectNoValidate ( edge . outputSlot , edge . inputSlot ) ;
public string GetPreviewShader ( AbstractMaterialNode node , out PreviewMode previewMode )
{
List < PropertyCollector . TextureInfo > configuredTextures ;
FloatShaderProperty outputIdProperty ;
return GetShader ( node , GenerationMode . Preview , string . Format ( "hidden/preview/{0}" , node . GetVariableNameForNode ( ) ) , out configuredTextures , out previewMode , out outputIdProperty ) ;
ValidateGraph ( ) ;
public string GetUberPreviewShader ( Dictionary < Guid , int > ids , out FloatShaderProperty outputIdProperty )
public void OnBeforeSerialize ( )
List < PropertyCollector . TextureInfo > configuredTextures ;
PreviewMode previewMode ;
return GetShader ( null , GenerationMode . Preview , "hidden/preview" , out configuredTextures , out previewMode , out outputIdProperty , ids ) ;
m_SerializableNodes = SerializationHelper . Serialize < INode > ( m_Nodes . Values ) ;
m_SerializableEdges = SerializationHelper . Serialize < IEdge > ( m_Edges ) ;
m_SerializedProperties = SerializationHelper . Serialize < IShaderProperty > ( m_Properties ) ;
internal static void GenerateSurfaceDescriptionStruct ( ShaderGenerator surfaceDescriptionStruct , List < MaterialSlot > slots , bool isMaster )
public virtual void OnAfterDeserialize ( )
surfaceDescriptionStruct . AddShaderChunk ( "struct SurfaceDescription{" , false ) ;
surfaceDescriptionStruct . Indent ( ) ;
if ( isMaster )
// have to deserialize 'globals' before nodes
m_Properties = SerializationHelper . Deserialize < IShaderProperty > ( m_SerializedProperties , null ) ;
var nodes = SerializationHelper . Deserialize < INode > ( m_SerializableNodes , GraphUtil . GetLegacyTypeRemapping ( ) ) ;
m_Nodes = new Dictionary < Guid , INode > ( nodes . Count ) ;
foreach ( var node in nodes )
foreach ( var slot in slots )
surfaceDescriptionStruct . AddShaderChunk ( string . Format ( "{0} {1};" , AbstractMaterialNode . ConvertConcreteSlotValueTypeToString ( AbstractMaterialNode . OutputPrecision . @float , slot . concreteValueType ) , AbstractMaterialNode . GetHLSLSafeName ( slot . shaderOutputName ) ) , false ) ;
surfaceDescriptionStruct . Deindent ( ) ;
}
else
{
surfaceDescriptionStruct . AddShaderChunk ( "float4 PreviewOutput;" , false ) ;
}
surfaceDescriptionStruct . Deindent ( ) ;
surfaceDescriptionStruct . AddShaderChunk ( "};" , false ) ;
node . owner = this ;
node . UpdateNodeAfterDeserialization ( ) ;
m_Nodes . Add ( node . guid , node ) ;
internal static void GenerateSurfaceDescription (
List < INode > activeNodeList ,
AbstractMaterialNode masterNode ,
AbstractMaterialGraph graph ,
ShaderGenerator surfaceDescriptionFunction ,
ShaderGenerator shaderFunctionVisitor ,
PropertyCollector shaderProperties ,
ShaderGraphRequirements requirements ,
GenerationMode mode ,
string functionName = "PopulateSurfaceData" ,
string surfaceDescriptionName = "SurfaceDescription" ,
FloatShaderProperty outputIdProperty = null ,
Dictionary < Guid , int > ids = null ,
IEnumerable < MaterialSlot > slots = null )
{
if ( graph = = null )
return ;
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "{0} {1}(SurfaceInputs IN) {{" , surfaceDescriptionName , functionName ) , false ) ;
surfaceDescriptionFunction . Indent ( ) ;
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "{0} surface = ({0})0;" , surfaceDescriptionName ) , false ) ;
m_SerializableNodes = null ;
foreach ( CoordinateSpace space in Enum . GetValues ( typeof ( CoordinateSpace ) ) )
{
var neededCoordinateSpace = space . ToNeededCoordinateSpace ( ) ;
if ( ( requirements . requiresNormal & neededCoordinateSpace ) > 0 )
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "float3 {0} = IN.{0};" , space . ToVariableName ( InterpolatorType . Normal ) ) , false ) ;
if ( ( requirements . requiresTangent & neededCoordinateSpace ) > 0 )
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "float3 {0} = IN.{0};" , space . ToVariableName ( InterpolatorType . Tangent ) ) , false ) ;
if ( ( requirements . requiresBitangent & neededCoordinateSpace ) > 0 )
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "float3 {0} = IN.{0};" , space . ToVariableName ( InterpolatorType . BiTangent ) ) , false ) ;
if ( ( requirements . requiresViewDir & neededCoordinateSpace ) > 0 )
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "float3 {0} = IN.{0};" , space . ToVariableName ( InterpolatorType . ViewDirection ) ) , false ) ;
if ( ( requirements . requiresPosition & neededCoordinateSpace ) > 0 )
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "float3 {0} = IN.{0};" , space . ToVariableName ( InterpolatorType . Position ) ) , false ) ;
m_Edges = SerializationHelper . Deserialize < IEdge > ( m_SerializableEdges , null ) ;
m_SerializableEdges = null ;
foreach ( var edge in m_Edges )
AddEdgeToNodeEdges ( edge ) ;
if ( requirements . requiresScreenPosition )
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "float4 {0} = IN.{0};" , ShaderGeneratorNames . ScreenPosition ) , false ) ;
if ( requirements . requiresVertexColor )
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "float4 {0} = IN.{0};" , ShaderGeneratorNames . VertexColor ) , false ) ;
foreach ( var channel in requirements . requiresMeshUVs . Distinct ( ) )
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "half4 {0} = IN.{0};" , channel . GetUVName ( ) ) , false ) ;
graph . CollectShaderProperties ( shaderProperties , mode ) ;
var currentId = - 1 ;
foreach ( var activeNode in activeNodeList . OfType < AbstractMaterialNode > ( ) )
{
if ( activeNode is IGeneratesFunction )
( activeNode as IGeneratesFunction ) . GenerateNodeFunction ( shaderFunctionVisitor , mode ) ;
if ( activeNode is IGeneratesBodyCode )
( activeNode as IGeneratesBodyCode ) . GenerateNodeCode ( surfaceDescriptionFunction , mode ) ;
if ( masterNode = = null & & activeNode . hasPreview )
public void OnEnable ( )
var outputSlot = activeNode . GetOutputSlots < MaterialSlot > ( ) . FirstOrDefault ( ) ;
if ( outputSlot ! = null )
foreach ( var node in GetNodes < INode > ( ) . OfType < IOnAssetEnabled > ( ) )
currentId + + ;
ids [ activeNode . guid ] = currentId ;
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "if ({0} == {1}) {{ surface.PreviewOutput = {2}; return surface; }}" , outputIdProperty . referenceName , currentId , ShaderGenerator . AdaptNodeOutputForPreview ( activeNode , outputSlot . id , activeNode . GetVariableNameForSlot ( outputSlot . id ) ) ) , false ) ;
node . OnEnable ( ) ;
activeNode . CollectShaderProperties ( shaderProperties , mode ) ;
if ( masterNode ! = null )
{
if ( masterNode is IMasterNode )
[Serializable]
public class InspectorPreviewData
var usedSlots = slots ? ? masterNode . GetInputSlots < MaterialSlot > ( ) ;
foreach ( var input in usedSlots )
{
var foundEdges = graph . GetEdges ( input . slotReference ) . ToArray ( ) ;
if ( foundEdges . Any ( ) )
{
var outputRef = foundEdges [ 0 ] . outputSlot ;
var fromNode = graph . GetNodeFromGuid < AbstractMaterialNode > ( outputRef . nodeGuid ) ;
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "surface.{0} = {1};" , AbstractMaterialNode . GetHLSLSafeName ( input . shaderOutputName ) , fromNode . GetVariableNameForSlot ( outputRef . slotId ) ) , true ) ;
}
else
{
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "surface.{0} = {1};" , AbstractMaterialNode . GetHLSLSafeName ( input . shaderOutputName ) , input . GetDefaultValue ( mode ) ) , true ) ;
}
}
}
else if ( masterNode . hasPreview )
{
foreach ( var slot in masterNode . GetOutputSlots < MaterialSlot > ( ) )
surfaceDescriptionFunction . AddShaderChunk ( string . Format ( "surface.{0} = {1};" , AbstractMaterialNode . GetHLSLSafeName ( slot . shaderOutputName ) , masterNode . GetVariableNameForSlot ( slot . id ) ) , true ) ;
}
}
surfaceDescriptionFunction . AddShaderChunk ( "return surface;" , false ) ;
surfaceDescriptionFunction . Deindent ( ) ;
surfaceDescriptionFunction . AddShaderChunk ( "}" , false ) ;
}
static void Visit ( List < INode > outputList , Dictionary < Guid , INode > unmarkedNodes , INode node )
{
if ( ! unmarkedNodes . ContainsKey ( node . guid ) )
return ;
foreach ( var slot in node . GetInputSlots < ISlot > ( ) )
{
foreach ( var edge in node . owner . GetEdges ( slot . slotReference ) )
{
var inputNode = node . owner . GetNodeFromGuid ( edge . outputSlot . nodeGuid ) ;
Visit ( outputList , unmarkedNodes , inputNode ) ;
}
}
unmarkedNodes . Remove ( node . guid ) ;
outputList . Add ( node ) ;
}
public string GetShader ( AbstractMaterialNode node , GenerationMode mode , string name , out List < PropertyCollector . TextureInfo > configuredTextures , out PreviewMode previewMode , out FloatShaderProperty outputIdProperty , Dictionary < Guid , int > ids = null )
{
bool isUber = node = = null ;
var vertexShader = new ShaderGenerator ( ) ;
var surfaceDescriptionFunction = new ShaderGenerator ( ) ;
var surfaceDescriptionStruct = new ShaderGenerator ( ) ;
var shaderFunctionVisitor = new ShaderGenerator ( ) ;
var surfaceInputs = new ShaderGenerator ( ) ;
var graphVertexInput = @ "
struct GraphVertexInput
{
float4 vertex : POSITION ;
float3 normal : NORMAL ;
float4 tangent : TANGENT ;
float4 color : COLOR ;
float4 texcoord0 : TEXCOORD0 ;
float4 lightmapUV : TEXCOORD1 ;
//UNITY_VERTEX_INPUT_INSTANCE_ID
} ; ";
surfaceInputs . AddShaderChunk ( "struct SurfaceInputs{" , false ) ;
surfaceInputs . Indent ( ) ;
var activeNodeList = ListPool < INode > . Get ( ) ;
if ( isUber )
{
var unmarkedNodes = GetNodes < INode > ( ) . Where ( x = > ! ( x is IMasterNode ) ) . ToDictionary ( x = > x . guid ) ;
while ( unmarkedNodes . Any ( ) )
{
var unmarkedNode = unmarkedNodes . FirstOrDefault ( ) ;
Visit ( activeNodeList , unmarkedNodes , unmarkedNode . Value ) ;
}
}
else
{
NodeUtils . DepthFirstCollectNodesFromNode ( activeNodeList , node ) ;
}
public SerializableMesh serializedMesh = new SerializableMesh ( ) ;
var requirements = GetRequirements ( activeNodeList ) ;
ShaderGenerator . GenerateSpaceTranslationSurfaceInputs ( requirements . requiresNormal , InterpolatorType . Normal , surfaceInputs ) ;
ShaderGenerator . GenerateSpaceTranslationSurfaceInputs ( requirements . requiresTangent , InterpolatorType . Tangent , surfaceInputs ) ;
ShaderGenerator . GenerateSpaceTranslationSurfaceInputs ( requirements . requiresBitangent , InterpolatorType . BiTangent , surfaceInputs ) ;
ShaderGenerator . GenerateSpaceTranslationSurfaceInputs ( requirements . requiresViewDir , InterpolatorType . ViewDirection , surfaceInputs ) ;
ShaderGenerator . GenerateSpaceTranslationSurfaceInputs ( requirements . requiresPosition , InterpolatorType . Position , surfaceInputs ) ;
if ( requirements . requiresVertexColor )
surfaceInputs . AddShaderChunk ( string . Format ( "float4 {0};" , ShaderGeneratorNames . VertexColor ) , false ) ;
if ( requirements . requiresScreenPosition )
surfaceInputs . AddShaderChunk ( string . Format ( "float4 {0};" , ShaderGeneratorNames . ScreenPosition ) , false ) ;
previewMode = PreviewMode . Preview3D ;
if ( ! isUber )
{
foreach ( var pNode in activeNodeList . OfType < AbstractMaterialNode > ( ) )
{
if ( pNode . previewMode = = PreviewMode . Preview3D )
{
previewMode = PreviewMode . Preview3D ;
break ;
}
}
}
foreach ( var channel in requirements . requiresMeshUVs . Distinct ( ) )
surfaceInputs . AddShaderChunk ( string . Format ( "half4 {0};" , channel . GetUVName ( ) ) , false ) ;
surfaceInputs . Deindent ( ) ;
surfaceInputs . AddShaderChunk ( "};" , false ) ;
vertexShader . AddShaderChunk ( "GraphVertexInput PopulateVertexData(GraphVertexInput v){" , false ) ;
vertexShader . Indent ( ) ;
vertexShader . AddShaderChunk ( "return v;" , false ) ;
vertexShader . Deindent ( ) ;
vertexShader . AddShaderChunk ( "}" , false ) ;
var slots = new List < MaterialSlot > ( ) ;
foreach ( var activeNode in isUber ? activeNodeList . Where ( n = > ( ( AbstractMaterialNode ) n ) . hasPreview ) : ( ( INode ) node ) . ToEnumerable ( ) )
{
if ( activeNode is IMasterNode )
slots . AddRange ( activeNode . GetInputSlots < MaterialSlot > ( ) ) ;
else
slots . AddRange ( activeNode . GetOutputSlots < MaterialSlot > ( ) ) ;
}
GenerateSurfaceDescriptionStruct ( surfaceDescriptionStruct , slots , ! isUber ) ;
var shaderProperties = new PropertyCollector ( ) ;
outputIdProperty = new FloatShaderProperty
{
displayName = "OutputId" ,
generatePropertyBlock = false ,
value = - 1
} ;
if ( isUber )
shaderProperties . AddShaderProperty ( outputIdProperty ) ;
GenerateSurfaceDescription (
activeNodeList ,
node ,
this ,
surfaceDescriptionFunction ,
shaderFunctionVisitor ,
shaderProperties ,
requirements ,
mode ,
outputIdProperty : outputIdProperty ,
ids : ids ) ;
var finalShader = new ShaderGenerator ( ) ;
finalShader . AddShaderChunk ( string . Format ( @"Shader ""{0}""" , name ) , false ) ;
finalShader . AddShaderChunk ( "{" , false ) ;
finalShader . Indent ( ) ;
finalShader . AddShaderChunk ( "Properties" , false ) ;
finalShader . AddShaderChunk ( "{" , false ) ;
finalShader . Indent ( ) ;
finalShader . AddShaderChunk ( shaderProperties . GetPropertiesBlock ( 2 ) , false ) ;
finalShader . Deindent ( ) ;
finalShader . AddShaderChunk ( "}" , false ) ;
finalShader . AddShaderChunk ( "HLSLINCLUDE" , false ) ;
finalShader . AddShaderChunk ( "#define USE_LEGACY_UNITY_MATRIX_VARIABLES" , false ) ;
finalShader . AddShaderChunk ( "#include \"Common.hlsl\"" , false ) ;
finalShader . AddShaderChunk ( "#include \"Packing.hlsl\"" , false ) ;
finalShader . AddShaderChunk ( "#include \"ShaderVariables.hlsl\"" , false ) ;
finalShader . AddShaderChunk ( "#include \"ShaderVariablesFunctions.hlsl\"" , false ) ;
finalShader . AddShaderChunk ( shaderFunctionVisitor . GetShaderString ( 2 ) , false ) ;
finalShader . AddShaderChunk ( graphVertexInput , false ) ;
finalShader . AddShaderChunk ( surfaceInputs . GetShaderString ( 2 ) , false ) ;
finalShader . AddShaderChunk ( surfaceDescriptionStruct . GetShaderString ( 2 ) , false ) ;
finalShader . AddShaderChunk ( shaderProperties . GetPropertiesDeclaration ( 2 ) , false ) ;
finalShader . AddShaderChunk ( vertexShader . GetShaderString ( 2 ) , false ) ;
finalShader . AddShaderChunk ( surfaceDescriptionFunction . GetShaderString ( 2 ) , false ) ;
finalShader . AddShaderChunk ( "ENDHLSL" , false ) ;
finalShader . AddShaderChunk ( ShaderGenerator . GetPreviewSubShader ( node , requirements ) , false ) ;
ListPool < INode > . Release ( activeNodeList ) ;
finalShader . Deindent ( ) ;
finalShader . AddShaderChunk ( "}" , false ) ;
configuredTextures = shaderProperties . GetConfiguredTexutres ( ) ;
return finalShader . GetShaderString ( 0 ) ;
}
}
public class InspectorPreviewData
{
public Mesh mesh ;
[NonSerialized]
public float scale = 1f ;
}
}
Tim Cooper
7 年前