blob: c5283fb7e9026c93b636965139f79bde91b97c7a (
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
|
//DISABLE_TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj
/* A test to use generics to treat an enum as a set.
Here we simplify previous example, by not allowing the backing type be
a specialization parameter (and just use int).
Moreover here E is constrained to __EnumType.
Works
*/
//TEST_INPUT:ubuffer(data=[0 0 0 0], stride=4):out,name outputBuffer
RWStructuredBuffer<float> outputBuffer;
enum Enum
{
A = 0x1,
B = 0x2,
C = 0x4,
};
__generic<E : __EnumType>
struct Flags
{
[mutating] void set(E e) { value |= (int)e; }
bool isSet(E e) { return (((int)e) & value) != 0; }
int value = 0;
};
[numthreads(4, 1, 1)]
void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
{
int index = dispatchThreadID.x;
Flags<Enum> flags;
if (index & 1)
{
flags.set(Enum::A);
}
bool isASet = flags.isSet(Enum::A);
outputBuffer[index] = isASet ? 2 : 1;
}
|
