blob: bf58398d005bec34dba00809016a490a396e617d (
plain)
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
|
//DISABLE_TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj
/* A test around use of an array like container.
*/
//TEST_INPUT:ubuffer(data=[0 0 0 0], stride=4):out,name outputBuffer
RWStructuredBuffer<float> outputBuffer;
/* Here is an attempt to provide an interface to check equality.
But is the definition right anyway? Really I want the This and Type to be
the same type. I guess this enforces that but in an odd manner.
*/
interface IEquality
{
associatedtype Type;
bool isEqual(Type rhs);
};
extension int : IEquality
{
typedef int Type;
bool isEqual(Type rhs)
{
// This use of `this` is going to seem odd to a C++ programmer
return this == rhs;
}
};
struct FixedArray<T : IEquality>
{
static const int SIZE = 4;
[mutating] void setAt(int i, T value) { elements[i] = value; }
T getAt(int i) { return elements[i]; }
int indexOf(T v)
{
for (int i = 0; i < SIZE; ++i)
{
if (v.isEqual(elements[i]))
{
return i;
}
return -1;
}
}
T elements[SIZE];
};
[numthreads(4, 1, 1)]
void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
{
int index = dispatchThreadID.x;
FixedArray<int> arr;
arr.setAt(0, index);
outputBuffer[index] = 1 + arr.getAt(0);
}
|
