310 lines
12 KiB
C#
310 lines
12 KiB
C#
|
using System;
|
||
|
|
using System.Linq;
|
||
|
|
using UnityEditor.ShaderGraph;
|
||
|
|
using UnityEditor.UIElements;
|
||
|
|
|
||
|
|
namespace UnityEditor.Rendering.Universal.ShaderGraph
|
||
|
|
{
|
||
|
|
sealed class UniversalSpriteCustomLitSubTarget : SubTarget<UniversalTarget>
|
||
|
|
{
|
||
|
|
static readonly GUID kSourceCodeGuid = new GUID("69e608b3e7e0405bbc2f259ad9cfa196"); // UniversalUnlitSubTarget.cs
|
||
|
|
|
||
|
|
public UniversalSpriteCustomLitSubTarget()
|
||
|
|
{
|
||
|
|
displayName = "Sprite Custom Lit";
|
||
|
|
}
|
||
|
|
|
||
|
|
public override bool IsActive() => true;
|
||
|
|
|
||
|
|
public override void Setup(ref TargetSetupContext context)
|
||
|
|
{
|
||
|
|
context.AddAssetDependency(kSourceCodeGuid, AssetCollection.Flags.SourceDependency);
|
||
|
|
context.AddSubShader(SubShaders.SpriteLit(target));
|
||
|
|
}
|
||
|
|
|
||
|
|
public override void GetFields(ref TargetFieldContext context)
|
||
|
|
{
|
||
|
|
// Only support SpriteColor legacy block if BaseColor/Alpha are not active
|
||
|
|
var descs = context.blocks.Select(x => x.descriptor);
|
||
|
|
bool useLegacyBlocks = !descs.Contains(BlockFields.SurfaceDescription.BaseColor) && !descs.Contains(BlockFields.SurfaceDescription.Alpha);
|
||
|
|
context.AddField(CoreFields.UseLegacySpriteBlocks, useLegacyBlocks);
|
||
|
|
|
||
|
|
// Surface Type
|
||
|
|
context.AddField(UniversalFields.SurfaceTransparent);
|
||
|
|
context.AddField(Fields.DoubleSided);
|
||
|
|
|
||
|
|
// Blend Mode
|
||
|
|
switch (target.alphaMode)
|
||
|
|
{
|
||
|
|
case AlphaMode.Premultiply:
|
||
|
|
context.AddField(UniversalFields.BlendPremultiply);
|
||
|
|
break;
|
||
|
|
case AlphaMode.Additive:
|
||
|
|
context.AddField(UniversalFields.BlendAdd);
|
||
|
|
break;
|
||
|
|
case AlphaMode.Multiply:
|
||
|
|
context.AddField(UniversalFields.BlendMultiply);
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
context.AddField(Fields.BlendAlpha);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public override void GetActiveBlocks(ref TargetActiveBlockContext context)
|
||
|
|
{
|
||
|
|
// Only support SpriteColor legacy block if BaseColor/Alpha are not active
|
||
|
|
bool useLegacyBlocks = !context.currentBlocks.Contains(BlockFields.SurfaceDescription.BaseColor) && !context.currentBlocks.Contains(BlockFields.SurfaceDescription.Alpha);
|
||
|
|
context.AddBlock(BlockFields.SurfaceDescriptionLegacy.SpriteColor, useLegacyBlocks);
|
||
|
|
|
||
|
|
context.AddBlock(UniversalBlockFields.SurfaceDescription.SpriteMask);
|
||
|
|
context.AddBlock(BlockFields.SurfaceDescription.NormalTS);
|
||
|
|
context.AddBlock(BlockFields.SurfaceDescription.Alpha);
|
||
|
|
}
|
||
|
|
|
||
|
|
public override void GetPropertiesGUI(ref TargetPropertyGUIContext context, Action onChange, Action<String> registerUndo)
|
||
|
|
{
|
||
|
|
context.AddProperty("Blending Mode", new EnumField(AlphaMode.Alpha) { value = target.alphaMode }, (evt) =>
|
||
|
|
{
|
||
|
|
if (Equals(target.alphaMode, evt.newValue))
|
||
|
|
return;
|
||
|
|
|
||
|
|
registerUndo("Change Blend");
|
||
|
|
target.alphaMode = (AlphaMode)evt.newValue;
|
||
|
|
onChange();
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
#region SubShader
|
||
|
|
static class SubShaders
|
||
|
|
{
|
||
|
|
public static SubShaderDescriptor SpriteLit(UniversalTarget target)
|
||
|
|
{
|
||
|
|
SubShaderDescriptor result = new SubShaderDescriptor()
|
||
|
|
{
|
||
|
|
pipelineTag = UniversalTarget.kPipelineTag,
|
||
|
|
customTags = UniversalTarget.kLitMaterialTypeTag,
|
||
|
|
renderType = $"{RenderType.Transparent}",
|
||
|
|
renderQueue = $"{UnityEditor.ShaderGraph.RenderQueue.Transparent}",
|
||
|
|
generatesPreview = true,
|
||
|
|
passes = new PassCollection
|
||
|
|
{
|
||
|
|
{ SpriteLitPasses.Lit },
|
||
|
|
{ SpriteLitPasses.Normal },
|
||
|
|
// Currently neither of these passes (selection/picking) can be last for the game view for
|
||
|
|
// UI shaders to render correctly. Verify [1352225] before changing this order.
|
||
|
|
{ CorePasses._2DSceneSelection(target) },
|
||
|
|
{ CorePasses._2DScenePicking(target) },
|
||
|
|
{ SpriteLitPasses.Forward },
|
||
|
|
},
|
||
|
|
};
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#endregion
|
||
|
|
|
||
|
|
#region Passes
|
||
|
|
static class SpriteLitPasses
|
||
|
|
{
|
||
|
|
public static PassDescriptor Lit = new PassDescriptor
|
||
|
|
{
|
||
|
|
// Definition
|
||
|
|
displayName = "Sprite Lit",
|
||
|
|
referenceName = "SHADERPASS_SPRITELIT",
|
||
|
|
lightMode = "Universal2D",
|
||
|
|
useInPreview = true,
|
||
|
|
|
||
|
|
// Template
|
||
|
|
passTemplatePath = GenerationUtils.GetDefaultTemplatePath("PassMesh.template"),
|
||
|
|
sharedTemplateDirectories = GenerationUtils.GetDefaultSharedTemplateDirectories(),
|
||
|
|
|
||
|
|
// Port Mask
|
||
|
|
validVertexBlocks = CoreBlockMasks.Vertex,
|
||
|
|
validPixelBlocks = SpriteLitBlockMasks.FragmentLit,
|
||
|
|
|
||
|
|
// Fields
|
||
|
|
structs = CoreStructCollections.Default,
|
||
|
|
requiredFields = SpriteLitRequiredFields.Lit,
|
||
|
|
fieldDependencies = CoreFieldDependencies.Default,
|
||
|
|
|
||
|
|
// Conditional State
|
||
|
|
renderStates = CoreRenderStates.Default,
|
||
|
|
pragmas = CorePragmas._2DDefault,
|
||
|
|
keywords = SpriteLitKeywords.Lit,
|
||
|
|
includes = SpriteLitIncludes.Lit,
|
||
|
|
};
|
||
|
|
|
||
|
|
public static PassDescriptor Normal = new PassDescriptor
|
||
|
|
{
|
||
|
|
// Definition
|
||
|
|
displayName = "Sprite Normal",
|
||
|
|
referenceName = "SHADERPASS_SPRITENORMAL",
|
||
|
|
lightMode = "NormalsRendering",
|
||
|
|
useInPreview = true,
|
||
|
|
|
||
|
|
// Template
|
||
|
|
passTemplatePath = GenerationUtils.GetDefaultTemplatePath("PassMesh.template"),
|
||
|
|
sharedTemplateDirectories = GenerationUtils.GetDefaultSharedTemplateDirectories(),
|
||
|
|
|
||
|
|
// Port Mask
|
||
|
|
validVertexBlocks = CoreBlockMasks.Vertex,
|
||
|
|
validPixelBlocks = SpriteLitBlockMasks.FragmentNormal,
|
||
|
|
|
||
|
|
// Fields
|
||
|
|
structs = CoreStructCollections.Default,
|
||
|
|
requiredFields = SpriteLitRequiredFields.Normal,
|
||
|
|
fieldDependencies = CoreFieldDependencies.Default,
|
||
|
|
|
||
|
|
// Conditional State
|
||
|
|
renderStates = CoreRenderStates.Default,
|
||
|
|
pragmas = CorePragmas._2DDefault,
|
||
|
|
includes = SpriteLitIncludes.Normal,
|
||
|
|
};
|
||
|
|
|
||
|
|
public static PassDescriptor Forward = new PassDescriptor
|
||
|
|
{
|
||
|
|
// Definition
|
||
|
|
displayName = "Sprite Forward",
|
||
|
|
referenceName = "SHADERPASS_SPRITEFORWARD",
|
||
|
|
lightMode = "UniversalForward",
|
||
|
|
useInPreview = true,
|
||
|
|
|
||
|
|
// Template
|
||
|
|
passTemplatePath = GenerationUtils.GetDefaultTemplatePath("PassMesh.template"),
|
||
|
|
sharedTemplateDirectories = GenerationUtils.GetDefaultSharedTemplateDirectories(),
|
||
|
|
|
||
|
|
// Port Mask
|
||
|
|
validVertexBlocks = CoreBlockMasks.Vertex,
|
||
|
|
validPixelBlocks = SpriteLitBlockMasks.FragmentForward,
|
||
|
|
|
||
|
|
// Fields
|
||
|
|
structs = CoreStructCollections.Default,
|
||
|
|
requiredFields = SpriteLitRequiredFields.Forward,
|
||
|
|
keywords = SpriteLitKeywords.Forward,
|
||
|
|
fieldDependencies = CoreFieldDependencies.Default,
|
||
|
|
|
||
|
|
// Conditional State
|
||
|
|
renderStates = CoreRenderStates.Default,
|
||
|
|
pragmas = CorePragmas._2DDefault,
|
||
|
|
includes = SpriteLitIncludes.Forward,
|
||
|
|
|
||
|
|
// Custom Interpolator Support
|
||
|
|
customInterpolators = CoreCustomInterpDescriptors.Common
|
||
|
|
};
|
||
|
|
}
|
||
|
|
#endregion
|
||
|
|
|
||
|
|
#region PortMasks
|
||
|
|
static class SpriteLitBlockMasks
|
||
|
|
{
|
||
|
|
public static BlockFieldDescriptor[] FragmentLit = new BlockFieldDescriptor[]
|
||
|
|
{
|
||
|
|
BlockFields.SurfaceDescription.BaseColor,
|
||
|
|
BlockFields.SurfaceDescriptionLegacy.SpriteColor,
|
||
|
|
BlockFields.SurfaceDescription.Alpha,
|
||
|
|
UniversalBlockFields.SurfaceDescription.SpriteMask,
|
||
|
|
};
|
||
|
|
|
||
|
|
public static BlockFieldDescriptor[] FragmentNormal = new BlockFieldDescriptor[]
|
||
|
|
{
|
||
|
|
BlockFields.SurfaceDescription.Alpha,
|
||
|
|
BlockFields.SurfaceDescription.NormalTS,
|
||
|
|
};
|
||
|
|
|
||
|
|
public static BlockFieldDescriptor[] FragmentForward = new BlockFieldDescriptor[]
|
||
|
|
{
|
||
|
|
BlockFields.SurfaceDescription.BaseColor,
|
||
|
|
BlockFields.SurfaceDescriptionLegacy.SpriteColor,
|
||
|
|
BlockFields.SurfaceDescription.Alpha,
|
||
|
|
BlockFields.SurfaceDescription.NormalTS,
|
||
|
|
};
|
||
|
|
}
|
||
|
|
#endregion
|
||
|
|
|
||
|
|
#region RequiredFields
|
||
|
|
static class SpriteLitRequiredFields
|
||
|
|
{
|
||
|
|
public static FieldCollection Lit = new FieldCollection()
|
||
|
|
{
|
||
|
|
StructFields.Varyings.color,
|
||
|
|
StructFields.Varyings.positionWS,
|
||
|
|
StructFields.Varyings.texCoord0,
|
||
|
|
StructFields.Varyings.screenPosition,
|
||
|
|
};
|
||
|
|
|
||
|
|
public static FieldCollection Normal = new FieldCollection()
|
||
|
|
{
|
||
|
|
StructFields.Varyings.normalWS,
|
||
|
|
StructFields.Varyings.tangentWS,
|
||
|
|
};
|
||
|
|
|
||
|
|
public static FieldCollection Forward = new FieldCollection()
|
||
|
|
{
|
||
|
|
StructFields.Varyings.color,
|
||
|
|
StructFields.Varyings.positionWS,
|
||
|
|
StructFields.Varyings.texCoord0,
|
||
|
|
};
|
||
|
|
}
|
||
|
|
#endregion
|
||
|
|
|
||
|
|
#region Keywords
|
||
|
|
static class SpriteLitKeywords
|
||
|
|
{
|
||
|
|
public static KeywordCollection Lit = new KeywordCollection
|
||
|
|
{
|
||
|
|
{ CoreKeywordDescriptors.DebugDisplay },
|
||
|
|
};
|
||
|
|
|
||
|
|
public static KeywordCollection Forward = new KeywordCollection
|
||
|
|
{
|
||
|
|
{ CoreKeywordDescriptors.DebugDisplay },
|
||
|
|
};
|
||
|
|
}
|
||
|
|
#endregion
|
||
|
|
|
||
|
|
#region Includes
|
||
|
|
static class SpriteLitIncludes
|
||
|
|
{
|
||
|
|
const string kSpriteUnlitPass = "Packages/com.unity.render-pipelines.universal/Editor/2D/ShaderGraph/Includes/SpriteUnlitPass.hlsl";
|
||
|
|
const string k2DNormal = "Packages/com.unity.render-pipelines.universal/Shaders/2D/Include/NormalsRenderingShared.hlsl";
|
||
|
|
const string kSpriteNormalPass = "Packages/com.unity.render-pipelines.universal/Editor/2D/ShaderGraph/Includes/SpriteNormalPass.hlsl";
|
||
|
|
const string kSpriteForwardPass = "Packages/com.unity.render-pipelines.universal/Editor/2D/ShaderGraph/Includes/SpriteForwardPass.hlsl";
|
||
|
|
|
||
|
|
public static IncludeCollection Lit = new IncludeCollection
|
||
|
|
{
|
||
|
|
// Pre-graph
|
||
|
|
{ CoreIncludes.CorePregraph },
|
||
|
|
{ CoreIncludes.ShaderGraphPregraph },
|
||
|
|
|
||
|
|
// Post-graph
|
||
|
|
{ CoreIncludes.CorePostgraph },
|
||
|
|
{ kSpriteUnlitPass, IncludeLocation.Postgraph },
|
||
|
|
};
|
||
|
|
|
||
|
|
public static IncludeCollection Normal = new IncludeCollection
|
||
|
|
{
|
||
|
|
// Pre-graph
|
||
|
|
{ CoreIncludes.CorePregraph },
|
||
|
|
{ CoreIncludes.ShaderGraphPregraph },
|
||
|
|
{ k2DNormal, IncludeLocation.Pregraph },
|
||
|
|
|
||
|
|
// Post-graph
|
||
|
|
{ CoreIncludes.CorePostgraph },
|
||
|
|
{ kSpriteNormalPass, IncludeLocation.Postgraph },
|
||
|
|
};
|
||
|
|
|
||
|
|
public static IncludeCollection Forward = new IncludeCollection
|
||
|
|
{
|
||
|
|
// Pre-graph
|
||
|
|
{ CoreIncludes.CorePregraph },
|
||
|
|
{ CoreIncludes.ShaderGraphPregraph },
|
||
|
|
|
||
|
|
// Post-graph
|
||
|
|
{ CoreIncludes.CorePostgraph },
|
||
|
|
{ kSpriteForwardPass, IncludeLocation.Postgraph },
|
||
|
|
};
|
||
|
|
}
|
||
|
|
#endregion
|
||
|
|
}
|
||
|
|
}
|