1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
// hit-object-reorder-thread.slang
//TEST:SIMPLE: -target dxil -entry rayGenerationMain -stage raygeneration -profile sm_6_5 -DNV_SHADER_EXTN_SLOT=u0
//TEST:SIMPLE(filecheck=SPIRV): -target spirv -entry rayGenerationMain -stage raygeneration -profile sm_6_5 -line-directive-mode none
//TEST:SIMPLE(filecheck=SPIRV): -target spirv -entry rayGenerationMain -stage raygeneration -emit-spirv-directly
//DISABLE_TEST(compute):COMPARE_COMPUTE:-d3d12 -output-using-type -profile sm_6_5 -render-feature ray-query
//DISABLE_TEST(compute):COMPARE_COMPUTE:-vk -output-using-type -render-feature ray-query
//TEST_INPUT: set scene = AccelerationStructure
uniform RaytracingAccelerationStructure scene;
//TEST_INPUT:set outputBuffer = out ubuffer(data=[0, 0, 0, 0], stride=4)
RWStructuredBuffer<uint> outputBuffer;
struct SomeValues
{
int a;
float b;
};
uint calcValue(HitObject hit)
{
uint r = 0;
if (hit.IsHit())
{
uint instanceIndex = hit.GetInstanceIndex();
uint instanceID = hit.GetInstanceID();
uint geometryIndex = hit.GetGeometryIndex();
uint primitiveIndex = hit.GetPrimitiveIndex();
SomeValues objSomeValues = hit.GetAttributes<SomeValues>();
r += instanceIndex;
r += instanceID;
r += geometryIndex;
r += primitiveIndex;
r += objSomeValues.a;
}
return r;
}
void rayGenerationMain()
{
int2 launchID = int2(DispatchRaysIndex().xy);
int2 launchSize = int2(DispatchRaysDimensions().xy);
int idx = launchID.x;
SomeValues someValues = { idx, idx * 2.0f };
RayDesc ray;
ray.Origin = float3(idx, 0, 0);
ray.TMin = 0.01f;
ray.Direction = float3(0, 1, 0);
ray.TMax = 1e4f;
RAY_FLAG rayFlags = RAY_FLAG_ACCEPT_FIRST_HIT_AND_END_SEARCH | RAY_FLAG_CULL_BACK_FACING_TRIANGLES;
uint instanceInclusionMask = 0xff;
uint rayContributionToHitGroupIndex = 0;
uint multiplierForGeometryContributionToHitGroupIndex = 4;
uint missShaderIndex = 0;
// SPIRV: OpHitObjectTraceRayNV
HitObject hit = HitObject::TraceRay(scene,
rayFlags,
instanceInclusionMask,
rayContributionToHitGroupIndex,
multiplierForGeometryContributionToHitGroupIndex,
missShaderIndex,
ray,
someValues);
uint r = calcValue(hit);
// SPIRV: OpReorderThreadWithHitObjectNV
ReorderThread( hit );
// Change the payload
SomeValues otherValues = { idx * -1, idx * 4.0f };
// Now Invoke to cast another ray, with the new payload
// SPIRV: OpHitObjectExecuteShaderNV
HitObject::Invoke( scene, hit, otherValues );
r += calcValue(hit);
// !!! TODO(JS) !!!
// NOTE! If I enable this I end up with a recursive failure in AST traversal, if
// otherValues is redefined.
// Reorder
// SPIRV: OpReorderThreadWithHitObjectNV
ReorderThread(hit, uint(idx & 3), 2);
// Change the payload
otherValues = { idx * -2, idx * 8.0f };
// Now Invoke to cast another ray, with the new payload
// SPIRV: OpHitObjectExecuteShaderNV
HitObject::Invoke( scene, hit, otherValues );
r += calcValue(hit);
// Reorder
// SPIRV: OpReorderThreadWithHintNV
ReorderThread(uint(idx & 1), 1);
// Change the payload
otherValues = { idx * -4, idx * 16.0f };
// Now Invoke to cast another ray, with the new payload
// SPIRV: OpHitObjectExecuteShaderNV
HitObject::Invoke( scene, hit, otherValues );
r += calcValue(hit);
outputBuffer[idx] = r;
}
|
