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
122
123
124
125
126
127
128
129
130
131
|
import fcpw;
#define UNDEFINED_BVH_TYPE 0
#define LINE_SEGMENT_BVH 1
#define TRIANGLE_BVH 2
#define LINE_SEGMENT_SNCH 3
#define TRIANGLE_SNCH 4
#ifndef _BVH_TYPE
#define _BVH_TYPE UNDEFINED_BVH_TYPE
#endif
#if _BVH_TYPE == LINE_SEGMENT_BVH
uniform ParameterBlock<Bvh<BvhNode, LineSegment, NoSilhouette>> gBvh;
#define _BVH_HAS_SILHOUETTE_DATA 0
#elif _BVH_TYPE == TRIANGLE_BVH
uniform ParameterBlock<Bvh<BvhNode, Triangle, NoSilhouette>> gBvh;
#define _BVH_HAS_SILHOUETTE_DATA 0
#elif _BVH_TYPE == LINE_SEGMENT_SNCH
uniform ParameterBlock<Bvh<SnchNode, LineSegment, Vertex>> gBvh;
#define _BVH_HAS_SILHOUETTE_DATA 1
#elif _BVH_TYPE == TRIANGLE_SNCH
uniform ParameterBlock<Bvh<SnchNode, Triangle, Edge>> gBvh;
#define _BVH_HAS_SILHOUETTE_DATA 1
#else
// Compile time error
#error _BVH_TYPE is not set to a supported type
#endif
[shader("compute")]
[numthreads(256, 1, 1)]
void rayIntersection(uint3 threadId: SV_DispatchThreadID,
uniform StructuredBuffer<Ray> rays,
uniform bool checkForOcclusion,
uniform RWStructuredBuffer<Interaction> interactions,
uniform uint nQueries)
{
uint index = threadId.x;
if (index >= nQueries)
{
return;
}
Ray r = rays[index];
Interaction i;
bool didIntersect = gBvh.intersect(r, checkForOcclusion, i);
if (didIntersect)
{
interactions[index] = i;
}
}
[shader("compute")]
[numthreads(256, 1, 1)]
void sphereIntersection(uint3 threadId: SV_DispatchThreadID,
uniform StructuredBuffer<BoundingSphere> boundingSpheres,
uniform StructuredBuffer<float3> randNums,
uniform RWStructuredBuffer<Interaction> interactions,
uniform uint nQueries)
{
uint index = threadId.x;
if (index >= nQueries)
{
return;
}
BoundingSphere s = boundingSpheres[index];
float3 randNum = randNums[index];
ConstantBranchTraversalWeight branchTraversalWeight;
Interaction i;
bool didIntersect = gBvh.intersect<ConstantBranchTraversalWeight>(s, randNum, branchTraversalWeight, i);
if (didIntersect)
{
interactions[index] = i;
}
}
[shader("compute")]
[numthreads(256, 1, 1)]
void closestPoint(uint3 threadId: SV_DispatchThreadID,
uniform StructuredBuffer<BoundingSphere> boundingSpheres,
uniform RWStructuredBuffer<Interaction> interactions,
uniform bool recordNormals,
uniform uint nQueries)
{
uint index = threadId.x;
if (index >= nQueries)
{
return;
}
BoundingSphere s = boundingSpheres[index];
Interaction i;
bool found = gBvh.findClosestPoint(s, i, recordNormals);
if (found)
{
interactions[index] = i;
}
}
[shader("compute")]
[numthreads(256, 1, 1)]
void closestSilhouettePoint(uint3 threadId: SV_DispatchThreadID,
uniform StructuredBuffer<BoundingSphere> boundingSpheres,
uniform StructuredBuffer<uint> flipNormalOrientation,
uniform float squaredMinRadius,
uniform float precision,
uniform RWStructuredBuffer<Interaction> interactions,
uniform uint nQueries)
{
uint index = threadId.x;
if (index >= nQueries)
{
return;
}
Interaction i;
#if _BVH_HAS_SILHOUETTE_DATA
BoundingSphere s = boundingSpheres[index];
bool flipNormal = flipNormalOrientation[index] == 1 ? true : false;
bool found = gBvh.findClosestSilhouettePoint(s, flipNormal, squaredMinRadius, precision, i);
if (found)
{
interactions[index] = i;
}
#endif
}
|
