private static TypeSymbol ResolveIntrinsicGenericMatrixType(GenericMatrixTypeSyntax type)
{
var scalarType = SyntaxFacts.GetScalarType(type.ScalarType.TypeTokens.Select(x => x.Kind).ToArray());
var numRows = (int)type.RowsToken.Value;
var numCols = (int)type.ColsToken.Value;
return(IntrinsicTypes.GetMatrixType(scalarType, numRows, numCols));
}
private BoundMatrixType BindMatrixType(MatrixTypeSyntax node)
{
var matrixType = TypeFacts.GetMatrixType(node.TypeToken.Kind);
var scalarType = matrixType.Item1;
var numRows = matrixType.Item2;
var numCols = matrixType.Item3;
return(new BoundMatrixType(IntrinsicTypes.GetMatrixType(scalarType, numRows, numCols)));
}
private static TypeSymbol ResolveIntrinsicMatrixType(MatrixTypeSyntax type)
{
var matrixType = SyntaxFacts.GetMatrixType(type.TypeToken.Kind);
var scalarType = matrixType.Item1;
var numRows = matrixType.Item2;
var numCols = matrixType.Item3;
return(IntrinsicTypes.GetMatrixType(scalarType, numRows, numCols));
}
private BoundGenericMatrixType BindGenericMatrixType(GenericMatrixTypeSyntax node)
{
var scalarType = TypeFacts.GetScalarType(node.ScalarType);
var numRows = (int)node.RowsToken.Value;
var numCols = (int)node.ColsToken.Value;
return(new BoundGenericMatrixType(
IntrinsicTypes.GetMatrixType(scalarType, numRows, numCols),
Bind(node.ScalarType, BindScalarType)));
}
internal static IntrinsicNumericTypeSymbol GetNumericTypeWithScalarType(this IntrinsicNumericTypeSymbol type, ScalarType scalarType)
{
switch (type.Kind)
{
case SymbolKind.IntrinsicMatrixType:
var matrixType = (IntrinsicMatrixTypeSymbol)type;
return(IntrinsicTypes.GetMatrixType(scalarType, matrixType.Rows, matrixType.Cols));
case SymbolKind.IntrinsicScalarType:
return(IntrinsicTypes.GetScalarType(scalarType));
case SymbolKind.IntrinsicVectorType:
return(IntrinsicTypes.GetVectorType(scalarType, ((IntrinsicVectorTypeSymbol)type).NumComponents));
default:
throw new ArgumentOutOfRangeException();
}
}
internal static TypeSymbol GetMatchingBoolType(TypeSymbol numericType)
{
switch (numericType.Kind)
{
case SymbolKind.IntrinsicScalarType:
return(IntrinsicTypes.Bool);
case SymbolKind.IntrinsicVectorType:
return(IntrinsicTypes.GetVectorType(ScalarType.Bool, ((IntrinsicVectorTypeSymbol)numericType).NumComponents));
case SymbolKind.IntrinsicMatrixType:
var matrixType = (IntrinsicMatrixTypeSymbol)numericType;
return(IntrinsicTypes.GetMatrixType(ScalarType.Bool, matrixType.Rows, matrixType.Cols));
default:
throw new InvalidOperationException();
}
}
private static IEnumerable <FunctionSymbol> CreateSM6Functions()
{
var allFunctions = new List <FunctionSymbol>();
allFunctions.Add(Create0(
"WaveGetLaneCount",
"Returns the number of lanes in a wave on this architecture.",
IntrinsicTypes.Uint));
allFunctions.Add(Create0(
"WaveGetLaneIndex",
"Returns the index of the current lane within the current wave.",
IntrinsicTypes.Uint));
allFunctions.Add(Create0(
"WaveIsFirstLane",
"Returns true only for the active lane in the current wave with the smallest index.",
IntrinsicTypes.Bool));
allFunctions.AddRange(Create1(
"WaveActiveAnyTrue",
"Returns true if the expression is true in any of the active lanes in the current wave.",
new[] { IntrinsicTypes.Bool },
"expr",
"The boolean expression to evaluate."));
allFunctions.AddRange(Create1(
"WaveActiveAllTrue",
"Returns true if the expression is true in all active lanes in the current wave.",
new[] { IntrinsicTypes.Bool },
"expr",
"The boolean expression to evaluate."));
allFunctions.AddRange(Create1(
"WaveActiveBallot",
"Returns a 4-bit unsigned integer bitmask of the evaluation of the Boolean expression for all active lanes in the specified wave.",
new[] { IntrinsicTypes.Bool },
"expr",
"The boolean expression to evaluate.",
new[] { IntrinsicTypes.Uint4 }));
allFunctions.AddRange(Create2(
"WaveReadLaneAt",
"Returns the value of the expression for the given lane index within the specified wave.",
IntrinsicTypes.AllNumericTypes,
"expr",
"The expression to evaluate.",
"laneIndex",
"The input lane index must be uniform across the wave.",
overrideParameterType2: IntrinsicTypes.Uint));
allFunctions.AddRange(Create1(
"WaveReadLaneFirst",
"Returns the value of the expression for the active lane of the current wave with the smallest index.",
IntrinsicTypes.AllNumericTypes,
"expr",
"The expression to evaluate."));
allFunctions.AddRange(Create1(
"WaveActiveAllEqual",
"Returns true if the expression is the same for every active lane in the current wave (and thus uniform across it).",
IntrinsicTypes.AllNumericTypes,
"expr",
"The expression to evaluate.",
new[] { IntrinsicTypes.Bool }));
allFunctions.AddRange(Create1(
"WaveActiveBitAnd",
"Returns the bitwise AND of all the values of the expression across all active lanes in the current wave and replicates it back to all active lanes.",
IntrinsicTypes.AllIntTypes,
"expr",
"The expression to evaluate."));
allFunctions.AddRange(Create1(
"WaveActiveBitOr",
"Returns the bitwise OR of all the values of the expression across all active lanes in the current wave and replicates it back to all active lanes.",
IntrinsicTypes.AllIntTypes,
"expr",
"The expression to evaluate."));
allFunctions.AddRange(Create1(
"WaveActiveBitXor",
"Returns the bitwise XOR of all the values of the expression across all active lanes in the current wave and replicates it back to all active lanes.",
IntrinsicTypes.AllIntTypes,
"expr",
"The expression to evaluate."));
allFunctions.AddRange(Create1(
"WaveActiveCountBits",
"Counts the number of boolean variables which evaluate to true across all active lanes in the current wave, and replicates the result to all lanes in the wave.",
new[] { IntrinsicTypes.Bool },
"bBit",
"The boolean variables to evaluate. Providing an explicit true Boolean value returns the number of active lanes.",
new[] { IntrinsicTypes.Uint }));
allFunctions.AddRange(Create1(
"WaveActiveMax",
"Returns the maximum value of the expression across all active lanes in the current wave and replicates it back to all active lanes.",
IntrinsicTypes.AllNumericTypes,
"expr",
"The expression to evaluate."));
allFunctions.AddRange(Create1(
"WaveActiveMin",
"Returns the minimum value of the expression across all active lanes in the current wave and replicates it back to all active lanes.",
IntrinsicTypes.AllNumericTypes,
"expr",
"The expression to evaluate."));
allFunctions.AddRange(Create1(
"WaveActiveProduct",
"Multiplies the values of the expression together across all active lanes in the current wave and replicates it back to all active lanes.",
IntrinsicTypes.AllNumericTypes,
"expr",
"The expression to evaluate."));
allFunctions.AddRange(Create1(
"WaveActiveSum",
"Sums up the value of the expression across all active lanes in the current wave and replicates it to all lanes in the current wave.",
IntrinsicTypes.AllNumericTypes,
"expr",
"The expression to evaluate."));
allFunctions.AddRange(Create1(
"WavePrefixCountBits",
"Returns the sum of all the specified boolean variables set to true across all active lanes with indices smaller than the current lane.",
new[] { IntrinsicTypes.Bool },
"bBit",
"The specified boolean variables.",
new[] { IntrinsicTypes.Uint }));
allFunctions.AddRange(Create1(
"WavePrefixSum",
"Returns the sum of all of the values in the active lanes with smaller indices than this one.",
IntrinsicTypes.AllNumericTypes,
"value",
"The value to sum up."));
allFunctions.AddRange(Create1(
"WavePrefixProduct",
"Returns the product of all of the values in the active lanes in this wave with indices less than this lane.",
IntrinsicTypes.AllNumericTypes,
"value",
"The value to multiply."));
allFunctions.AddRange(Create2(
"QuadReadLaneAt",
"Returns the specified source value from the lane identified by the lane ID within the current quad.",
IntrinsicTypes.AllNumericTypes,
"sourceValue",
"The requested type.",
"quadLaneID",
"The lane ID; this will be a value from 0 to 3.",
overrideParameterType2: IntrinsicTypes.Uint));
allFunctions.AddRange(Create1(
"QuadReadAcrossDiagonal",
"Returns the specified local value which is read from the diagonally opposite lane in this quad.",
IntrinsicTypes.AllNumericTypes,
"localValue",
"The requested type."));
allFunctions.AddRange(Create1(
"QuadReadAcrossX",
"Returns the specified local value read from the other lane in this quad in the X direction.",
IntrinsicTypes.AllNumericTypes,
"localValue",
"The requested type."));
allFunctions.AddRange(Create1(
"QuadReadAcrossY",
"Returns the specified local value read from the other lane in this quad in the Y direction.",
IntrinsicTypes.AllNumericTypes,
"localValue",
"The requested type."));
// DXR Intrinsic Functions
allFunctions.Add(Create0(
"AcceptHitAndEndSearch",
"Used in an any hit shader to commit the current hit and then stop searching for more hits for the ray.",
IntrinsicTypes.Void));
// TODO: missing template feature for second parameter
//allFunctions.AddRange(Create2(
// "CallShader",
// "Invokes another shader from within a shader.",
// IntrinsicTypes.AllTypes,
// "ShaderIndex",
// "An unsigned integer representing the index into the callable shader table specified in the call to DispatchRays",
// "Parameter",
// "The user-defined parameters to pass to the callable shader. This parameter structure must match the parameter structure used in the callable shader pointed to in the shader table.",
// IntrinsicTypes.Void,
// IntrinsicTypes.Uint,
// IntrinsicTypes.Template, // TODO: this feature is missing
// ParameterDirection.Inout));
allFunctions.Add(Create0(
"IgnoreHit",
"Called from an any hit shader to reject the hit and end the shader.",
IntrinsicTypes.Void));
allFunctions.Add(Create0(
"PrimitiveIndex",
"Retrieves the autogenerated index of the primitive within the geometry inside the bottom-level acceleration structure instance.",
IntrinsicTypes.Uint));
// TODO: missing template feature for third parameter
//allFunctions.AddRange(Create3(
// "ReportHit",
// "Called by an intersection shader to report a ray intersection.",
// IntrinsicTypes.AllTypes,
// "THit",
// "A float value specifying the parametric distance of the intersection.",
// "HitKind",
// "An unsigned integer that identifies the type of hit that occurred. This is a user-specified value in the range of 0-127. The value can be read by any hit or closest hit shaders with the HitKind intrinsic.",
// "Attributes",
// "The user-defined Intersection Attribute Structure specifying the intersection attributes.",
// IntrinsicTypes.Void,
// IntrinsicTypes.Float,
// IntrinsicTypes.Uint,
// IntrinsicTypes.AllTypes /* TODO: this feature is missing */));
// TODO: missing template feature for last parameter
//allFunctions.Add(new FunctionSymbol(
// "TraceRay",
// "Sends a ray into a search for hits in an acceleration structure.",
// null,
// IntrinsicTypes.Void,
// f => new[]
// {
// new ParameterSymbol("AccelerationStructure", "The top-level acceleration structure to use. Specifying a NULL acceleration structure forces a miss.", f, IntrinsicTypes.RaytracingAccelerationStructure),
// new ParameterSymbol("RayFlags", "Valid combination of ray_flag values. Only defined ray flags are propagated by the system, i.e. are visible to the RayFlags shader intrinsic.", f, IntrinsicTypes.Uint),
// new ParameterSymbol("InstanceInclusionMask", "An unsigned integer, the bottom 8 bits of which are used to include or reject geometry instances based on the InstanceMask in each instance.", f, IntrinsicTypes.Uint),
// new ParameterSymbol("RayContributionToHitGroupIndex", "An unsigned integer specifying the offset to add into addressing calculations within shader tables for hit group indexing. Only the bottom 4 bits of this value are used.", f, IntrinsicTypes.Uint),
// new ParameterSymbol("MultiplierForGeometryContributionToShaderIndex", "An unsigned integer specifying the stride to multiply by GeometryContributionToHitGroupIndex, which is just the 0 based index the geometry was supplied by the app into its bottom-level acceleration structure. Only the bottom 16 bits of this multiplier value are used.", f, IntrinsicTypes.Uint),
// new ParameterSymbol("MissShaderIndex", "An unsigned integer specifying the index of the miss shader within a shader table.", f, IntrinsicTypes.Uint),
// new ParameterSymbol("Ray", "A RayDesc representing the ray to be traced.", f, IntrinsicTypes.RayDesc),
// new ParameterSymbol("Payload", "A user defined ray payload accessed both for both input and output by shaders invoked during raytracing. After TraceRay completes, the caller can access the payload as well.", f, IntrinsicTypes.Template, ParameterDirection.Inout) // TODO: this feature is missing
// }));
// DXR System Value Intrinsics
allFunctions.Add(Create0(
"DispatchRaysIndex",
"Gets the current location within the width, height and depth obtained with the DispatchRaysDimensions system value intrinsic.",
IntrinsicTypes.Uint3));
allFunctions.Add(Create0(
"DispatchRaysDimensions",
"The width, height and depth values from the D3D12_DISPATCH_RAYS_DESC structure specified in the originating DispatchRays call.",
IntrinsicTypes.Uint3));
allFunctions.Add(Create0(
"WorldRayOrigin",
"The world-space origin of the current ray.",
IntrinsicTypes.Float3));
allFunctions.Add(Create0(
"WorldRayDirection",
"The world-space direction for the current ray.",
IntrinsicTypes.Float3));
allFunctions.Add(Create0(
"RayTMin",
"A float representing the current parametric starting point for the ray.",
IntrinsicTypes.Float));
allFunctions.Add(Create0(
"RayTCurrent",
"A float representing the current parametric ending point for the ray.",
IntrinsicTypes.Float));
allFunctions.Add(Create0(
"RayFlags",
"An unsigned integer containing the current ray_flag flags.",
IntrinsicTypes.Uint));
allFunctions.Add(Create0(
"InstanceIndex",
"The autogenerated index of the current instance in the top-level raytracing acceleration structure.",
IntrinsicTypes.Uint));
allFunctions.Add(Create0(
"InstanceId",
"The user-provided identifier for the instance on the bottom-level acceleration structure instance within the top-level structure.",
IntrinsicTypes.Uint));
allFunctions.Add(Create0(
"PrimitiveIndex",
"Retrieves the autogenerated index of the primitive within the geometry inside the bottom-level acceleration structure instance.",
IntrinsicTypes.Uint));
allFunctions.Add(Create0(
"ObjectRayOrigin",
"The object-space origin for the current ray. Object-space refers to the space of the current bottom-level acceleration structure.",
IntrinsicTypes.Float3));
allFunctions.Add(Create0(
"ObjectRayDirection",
"The object-space direction for the current ray. Object-space refers to the space of the current bottom-level acceleration structure.",
IntrinsicTypes.Float3));
allFunctions.Add(Create0(
"ObjectToWorld3x4",
"A matrix for transforming from object-space to world-space. Object-space refers to the space of the current bottom-level acceleration structure.",
IntrinsicTypes.GetMatrixType(ScalarType.Float, 3, 4)));
allFunctions.Add(Create0(
"ObjectToWorld4x3",
"A matrix for transforming from object-space to world-space. Object-space refers to the space of the current bottom-level acceleration structure.",
IntrinsicTypes.GetMatrixType(ScalarType.Float, 4, 3)));
allFunctions.Add(Create0(
"WorldToObject3x4",
"A matrix for transforming from world-space to object-space. Object-space refers to the space of the current bottom-level acceleration structure.",
IntrinsicTypes.GetMatrixType(ScalarType.Float, 3, 4)));
allFunctions.Add(Create0(
"WorldToObject4x3",
"A matrix for transforming from world-space to object-space. Object-space refers to the space of the current bottom-level acceleration structure.",
IntrinsicTypes.GetMatrixType(ScalarType.Float, 4, 3)));
allFunctions.Add(Create0(
"HitKind",
"Returns the value passed as the HitKind parameter to ReportHit.",
IntrinsicTypes.Uint));
return(allFunctions);
}