// Configures an HDRP Subshader with a VFX context.
private static SubShaderDescriptor PostProcessSubShader(SubShaderDescriptor subShaderDescriptor, VFXContext context, VFXContextCompiledData data)
{
var attributesStruct = GenerateVFXAttributesStruct(context, VFXAttributeType.Current);
var sourceAttributesStruct = GenerateVFXAttributesStruct(context, VFXAttributeType.Source);
// Defer to VFX to generate various misc. code-gen that ShaderGraph currently can't handle.
// We use the AdditionalCommand descriptors for ShaderGraph generation to splice these in.
// ( i.e. VFX Graph Block Function declaration + calling, Property Mapping, etc. )
GenerateVFXAdditionalCommands(
context, data,
out var loadAttributeDescriptor,
out var blockFunctionDescriptor,
out var blockCallFunctionDescriptor,
out var interpolantsGenerationDescriptor,
out var buildVFXFragInputs,
out var pixelPropertiesAssignDescriptor,
out var defineSpaceDescriptor,
out var parameterBufferDescriptor,
out var additionalDefinesDescriptor,
out var loadPositionAttributeDescriptor,
out var loadCropFactorAttributesDescriptor,
out var loadTexcoordAttributesDescriptor,
out var vertexPropertiesGenerationDescriptor,
out var vertexPropertiesAssignDescriptor
);
// Omit MV or Shadow Pass if disabled on the context.
var filteredPasses = subShaderDescriptor.passes.AsEnumerable();
var outputContext = (VFXAbstractParticleOutput)context;
if (!outputContext.hasMotionVector)
{
filteredPasses = filteredPasses.Where(o => o.descriptor.lightMode != "MotionVectors");
}
if (!outputContext.hasShadowCasting)
{
filteredPasses = filteredPasses.Where(o => o.descriptor.lightMode != "ShadowCaster");
}
var passes = filteredPasses.ToArray();
PassCollection vfxPasses = new PassCollection();
for (int i = 0; i < passes.Length; i++)
{
var passDescriptor = passes[i].descriptor;
// Warning: We are replacing the struct provided in the regular pass. It is ok as for now the VFX editor don't support
// tessellation or raytracing
passDescriptor.structs = new StructCollection
{
AttributesMeshVFX, // TODO: Could probably re-use the original HD Attributes Mesh and just ensure Instancing enabled.
Structs.VertexDescriptionInputs,
Structs.SurfaceDescriptionInputs,
AppendVFXInterpolator(HDStructs.VaryingsMeshToPS, context, data),
attributesStruct,
sourceAttributesStruct
};
// Add additional VFX dependencies
passDescriptor.fieldDependencies = passDescriptor.fieldDependencies == null ? new DependencyCollection() : new DependencyCollection {
passDescriptor.fieldDependencies
}; // Duplicate fieldDependencies to avoid side effects (static list modification)
passDescriptor.fieldDependencies.Add(VFXHDRPSubTarget.ElementSpaceDependencies);
passDescriptor.additionalCommands = new AdditionalCommandCollection
{
loadAttributeDescriptor,
blockFunctionDescriptor,
blockCallFunctionDescriptor,
interpolantsGenerationDescriptor,
buildVFXFragInputs,
pixelPropertiesAssignDescriptor,
defineSpaceDescriptor,
parameterBufferDescriptor,
additionalDefinesDescriptor,
loadPositionAttributeDescriptor,
loadCropFactorAttributesDescriptor,
loadTexcoordAttributesDescriptor,
vertexPropertiesGenerationDescriptor,
vertexPropertiesAssignDescriptor,
GenerateFragInputs(context, data)
};
vfxPasses.Add(passDescriptor, passes[i].fieldConditions);
}
subShaderDescriptor.passes = vfxPasses;
return(subShaderDescriptor);
}
// A key difference between Material Shader and VFX Shader generation is how surface properties are provided. Material Shaders
// simply provide properties via UnityPerMaterial cbuffer. VFX expects these same properties to be computed in the vertex
// stage (because we must evaluate them with the VFX blocks), and packed with the interpolators for the fragment stage.
static StructDescriptor AppendVFXInterpolator(StructDescriptor interpolator, VFXContext context, VFXContextCompiledData contextData)
{
var fields = interpolator.fields.ToList();
var expressionToName = context.GetData().GetAttributes().ToDictionary(o => new VFXAttributeExpression(o.attrib) as VFXExpression, o => (new VFXAttributeExpression(o.attrib)).GetCodeString(null));
expressionToName = expressionToName.Union(contextData.uniformMapper.expressionToCode).ToDictionary(s => s.Key, s => s.Value);
var mainParameters = contextData.gpuMapper.CollectExpression(-1).ToArray();
// Warning/TODO: FragmentParameters are created from the ShaderGraphVfxAsset.
// We may ultimately need to move this handling of VFX Interpolators + SurfaceDescriptionFunction function signature directly into the SG Generator (since it knows about the exposed properties).
foreach (string fragmentParameter in context.fragmentParameters)
{
var filteredNamedExpression = mainParameters.FirstOrDefault(o => fragmentParameter == o.name &&
!(expressionToName.ContainsKey(o.exp) && expressionToName[o.exp] == o.name)); // if parameter already in the global scope, there's nothing to do
if (filteredNamedExpression.exp != null)
{
var type = VFXExpression.TypeToType(filteredNamedExpression.exp.valueType);
if (!kVFXShaderValueTypeMap.TryGetValue(type, out var shaderValueType))
{
continue;
}
// TODO: NoInterpolation only for non-strips.
fields.Add(new FieldDescriptor(HDStructFields.VaryingsMeshToPS.name, filteredNamedExpression.name, "", shaderValueType, subscriptOptions: StructFieldOptions.Static, interpolation: "nointerpolation"));
}
}
// VFX Object Space Interpolators
fields.Add(HDStructFields.VaryingsMeshToPS.worldToElement0);
fields.Add(HDStructFields.VaryingsMeshToPS.worldToElement1);
fields.Add(HDStructFields.VaryingsMeshToPS.worldToElement2);
fields.Add(HDStructFields.VaryingsMeshToPS.elementToWorld0);
fields.Add(HDStructFields.VaryingsMeshToPS.elementToWorld1);
fields.Add(HDStructFields.VaryingsMeshToPS.elementToWorld2);
interpolator.fields = fields.ToArray();
return(interpolator);
}
static void GenerateVFXAdditionalCommands(VFXContext context, VFXContextCompiledData contextData,
out AdditionalCommandDescriptor loadAttributeDescriptor,
out AdditionalCommandDescriptor blockFunctionDescriptor,
out AdditionalCommandDescriptor blockCallFunctionDescriptor,
out AdditionalCommandDescriptor interpolantsGenerationDescriptor,
out AdditionalCommandDescriptor buildVFXFragInputsDescriptor,
out AdditionalCommandDescriptor pixelPropertiesAssignDescriptor,
out AdditionalCommandDescriptor defineSpaceDescriptor,
out AdditionalCommandDescriptor parameterBufferDescriptor,
out AdditionalCommandDescriptor additionalDefinesDescriptor,
out AdditionalCommandDescriptor loadPositionAttributeDescriptor,
out AdditionalCommandDescriptor loadCropFactorAttributesDescriptor,
out AdditionalCommandDescriptor loadTexcoordAttributesDescriptor,
out AdditionalCommandDescriptor vertexPropertiesGenerationDescriptor,
out AdditionalCommandDescriptor vertexPropertiesAssignDescriptor)
{
// TODO: Clean all of this up. Currently just an adapter between VFX Code Gen + SG Code Gen and *everything* has been stuffed here.
// Load Attributes
loadAttributeDescriptor = new AdditionalCommandDescriptor("VFXLoadAttribute", VFXCodeGenerator.GenerateLoadAttribute(".", context).ToString());
// Graph Blocks
VFXCodeGenerator.BuildContextBlocks(context, contextData, out var blockFunction, out var blockCallFunction);
blockFunctionDescriptor = new AdditionalCommandDescriptor("VFXGeneratedBlockFunction", blockFunction);
blockCallFunctionDescriptor = new AdditionalCommandDescriptor("VFXProcessBlocks", blockCallFunction);
// Vertex Input
VFXCodeGenerator.BuildVertexProperties(context, contextData, out var vertexPropertiesGeneration);
vertexPropertiesGenerationDescriptor = new AdditionalCommandDescriptor("VFXVertexPropertiesGeneration", vertexPropertiesGeneration);
VFXCodeGenerator.BuildVertexPropertiesAssign(context, contextData, out var vertexPropertiesAssign);
vertexPropertiesAssignDescriptor = new AdditionalCommandDescriptor("VFXVertexPropertiesAssign", vertexPropertiesAssign);
// Interpolator
VFXCodeGenerator.BuildInterpolatorBlocks(context, contextData, out var interpolatorsGeneration);
interpolantsGenerationDescriptor = new AdditionalCommandDescriptor("VFXInterpolantsGeneration", interpolatorsGeneration);
// Frag Inputs - Only VFX will know if frag inputs come from interpolator or the CBuffer.
VFXCodeGenerator.BuildFragInputsGeneration(context, contextData, out var buildFragInputsGeneration);
buildVFXFragInputsDescriptor = new AdditionalCommandDescriptor("VFXSetFragInputs", buildFragInputsGeneration);
VFXCodeGenerator.BuildPixelPropertiesAssign(context, contextData, out var pixelPropertiesAssign);
pixelPropertiesAssignDescriptor = new AdditionalCommandDescriptor("VFXPixelPropertiesAssign", pixelPropertiesAssign);
// Define coordinate space
var defineSpaceDescriptorContent = string.Empty;
if (context.GetData() is ISpaceable)
{
var spaceable = context.GetData() as ISpaceable;
defineSpaceDescriptorContent =
$"#define {(spaceable.space == VFXCoordinateSpace.World ? "VFX_WORLD_SPACE" : "VFX_LOCAL_SPACE")} 1";
}
defineSpaceDescriptor = new AdditionalCommandDescriptor("VFXDefineSpace", defineSpaceDescriptorContent);
// Parameter Cbuffer
VFXCodeGenerator.BuildParameterBuffer(contextData, out var parameterBuffer);
parameterBufferDescriptor = new AdditionalCommandDescriptor("VFXParameterBuffer", parameterBuffer);
// Defines & Headers - Not all are necessary, however some important ones are mixed in like indirect draw, strips, flipbook, particle strip info...
ShaderStringBuilder additionalDefines = new ShaderStringBuilder();
// TODO: Need to add defines for current/source usage (i.e. scale).
var allCurrentAttributes = context.GetData().GetAttributes().Where(a =>
(context.GetData().IsCurrentAttributeUsed(a.attrib, context)) ||
(context.contextType == VFXContextType.Init && context.GetData().IsAttributeStored(a.attrib))); // In init, needs to declare all stored attributes for intialization
var allSourceAttributes = context.GetData().GetAttributes().Where(a => (context.GetData().IsSourceAttributeUsed(a.attrib, context)));
foreach (var attribute in allCurrentAttributes)
{
additionalDefines.AppendLine("#define VFX_USE_{0}_{1} 1", attribute.attrib.name.ToUpper(CultureInfo.InvariantCulture), "CURRENT");
}
foreach (var attribute in allSourceAttributes)
{
additionalDefines.AppendLine("#define VFX_USE_{0}_{1} 1", attribute.attrib.name.ToUpper(CultureInfo.InvariantCulture), "SOURCE");
}
foreach (var header in context.additionalDataHeaders)
{
additionalDefines.AppendLine(header);
}
foreach (var define in context.additionalDefines)
{
additionalDefines.AppendLine($"#define {define} 1");
}
additionalDefinesDescriptor = new AdditionalCommandDescriptor("VFXDefines", additionalDefines.ToString());
// Load Position Attribute
loadPositionAttributeDescriptor = new AdditionalCommandDescriptor("VFXLoadPositionAttribute", VFXCodeGenerator.GenerateLoadAttribute("position", context).ToString().ToString());
// Load Crop Factor Attribute
var mainParameters = contextData.gpuMapper.CollectExpression(-1).ToArray();
var expressionToName = context.GetData().GetAttributes().ToDictionary(o => new VFXAttributeExpression(o.attrib) as VFXExpression, o => (new VFXAttributeExpression(o.attrib)).GetCodeString(null));
expressionToName = expressionToName.Union(contextData.uniformMapper.expressionToCode).ToDictionary(s => s.Key, s => s.Value);
loadCropFactorAttributesDescriptor = new AdditionalCommandDescriptor("VFXLoadCropFactorParameter", VFXCodeGenerator.GenerateLoadParameter("cropFactor", mainParameters, expressionToName).ToString().ToString());
loadTexcoordAttributesDescriptor = new AdditionalCommandDescriptor("VFXLoadTexcoordParameter", VFXCodeGenerator.GenerateLoadParameter("texCoord", mainParameters, expressionToName).ToString().ToString());
}
public override ShaderGraphBinder GetShaderGraphDescriptor(VFXContext context, VFXContextCompiledData data)
{
return(new ShaderGraphBinder
{
structs = new StructCollection
{
AttributesMeshVFX, // TODO: Could probably re-use the original HD Attributes Mesh and just ensure Instancing enabled.
Structs.VertexDescriptionInputs,
Structs.SurfaceDescriptionInputs,
AppendVFXInterpolator(HDStructs.VaryingsMeshToPS, context, data),
},
fieldDependencies = ElementSpaceDependencies,
useFragInputs = true
});
}