1#version 320 es
2
3#extension GL_KHR_shader_subgroup_quad: enable
4
5layout (local_size_x = 8) in;
6
7layout(binding = 0) buffer Buffers
8{
9    vec4  f4;
10    ivec4 i4;
11    uvec4 u4;
12} data[4];
13
14void main()
15{
16    uint invocation = (gl_SubgroupInvocationID + gl_SubgroupSize) % 4u;
17
18    data[0].f4.x   = subgroupQuadBroadcast(data[0].f4.x, 1u);
19    data[0].f4.xy  = subgroupQuadBroadcast(data[1].f4.xy, 1u);
20    data[0].f4.xyz = subgroupQuadBroadcast(data[2].f4.xyz, 1u);
21    data[0].f4     = subgroupQuadBroadcast(data[3].f4, 1u);
22
23    data[0].i4.x   = subgroupQuadBroadcast(data[0].i4.x, 1u);
24    data[0].i4.xy  = subgroupQuadBroadcast(data[1].i4.xy, 1u);
25    data[0].i4.xyz = subgroupQuadBroadcast(data[2].i4.xyz, 1u);
26    data[0].i4     = subgroupQuadBroadcast(data[3].i4, 1u);
27
28    data[0].u4.x   = subgroupQuadBroadcast(data[0].u4.x, 1u);
29    data[0].u4.xy  = subgroupQuadBroadcast(data[1].u4.xy, 1u);
30    data[0].u4.xyz = subgroupQuadBroadcast(data[2].u4.xyz, 1u);
31    data[0].u4     = subgroupQuadBroadcast(data[3].u4, 1u);
32
33    data[1].i4.x   =   int(subgroupQuadBroadcast(data[0].i4.x < 0, 1u));
34    data[1].i4.xy  = ivec2(subgroupQuadBroadcast(lessThan(data[1].i4.xy, ivec2(0)), 1u));
35    data[1].i4.xyz = ivec3(subgroupQuadBroadcast(lessThan(data[1].i4.xyz, ivec3(0)), 1u));
36    data[1].i4     = ivec4(subgroupQuadBroadcast(lessThan(data[1].i4, ivec4(0)), 1u));
37
38    data[1].f4.x   = subgroupQuadSwapHorizontal(data[0].f4.x);
39    data[1].f4.xy  = subgroupQuadSwapHorizontal(data[1].f4.xy);
40    data[1].f4.xyz = subgroupQuadSwapHorizontal(data[2].f4.xyz);
41    data[1].f4     = subgroupQuadSwapHorizontal(data[3].f4);
42
43    data[1].i4.x   = subgroupQuadSwapHorizontal(data[0].i4.x);
44    data[1].i4.xy  = subgroupQuadSwapHorizontal(data[1].i4.xy);
45    data[1].i4.xyz = subgroupQuadSwapHorizontal(data[2].i4.xyz);
46    data[1].i4     = subgroupQuadSwapHorizontal(data[3].i4);
47
48    data[1].u4.x   = subgroupQuadSwapHorizontal(data[0].u4.x);
49    data[1].u4.xy  = subgroupQuadSwapHorizontal(data[1].u4.xy);
50    data[1].u4.xyz = subgroupQuadSwapHorizontal(data[2].u4.xyz);
51    data[1].u4     = subgroupQuadSwapHorizontal(data[3].u4);
52
53    data[2].i4.x   =   int(subgroupQuadSwapHorizontal(data[0].i4.x < 0));
54    data[2].i4.xy  = ivec2(subgroupQuadSwapHorizontal(lessThan(data[1].i4.xy, ivec2(0))));
55    data[2].i4.xyz = ivec3(subgroupQuadSwapHorizontal(lessThan(data[1].i4.xyz, ivec3(0))));
56    data[2].i4     = ivec4(subgroupQuadSwapHorizontal(lessThan(data[1].i4, ivec4(0))));
57
58    data[2].f4.x   = subgroupQuadSwapVertical(data[0].f4.x);
59    data[2].f4.xy  = subgroupQuadSwapVertical(data[1].f4.xy);
60    data[2].f4.xyz = subgroupQuadSwapVertical(data[2].f4.xyz);
61    data[2].f4     = subgroupQuadSwapVertical(data[3].f4);
62
63    data[2].i4.x   = subgroupQuadSwapVertical(data[0].i4.x);
64    data[2].i4.xy  = subgroupQuadSwapVertical(data[1].i4.xy);
65    data[2].i4.xyz = subgroupQuadSwapVertical(data[2].i4.xyz);
66    data[2].i4     = subgroupQuadSwapVertical(data[3].i4);
67
68    data[2].u4.x   = subgroupQuadSwapVertical(data[0].u4.x);
69    data[2].u4.xy  = subgroupQuadSwapVertical(data[1].u4.xy);
70    data[2].u4.xyz = subgroupQuadSwapVertical(data[2].u4.xyz);
71    data[2].u4     = subgroupQuadSwapVertical(data[3].u4);
72
73    data[3].i4.x   =   int(subgroupQuadSwapVertical(data[0].i4.x < 0));
74    data[3].i4.xy  = ivec2(subgroupQuadSwapVertical(lessThan(data[1].i4.xy, ivec2(0))));
75    data[3].i4.xyz = ivec3(subgroupQuadSwapVertical(lessThan(data[1].i4.xyz, ivec3(0))));
76    data[3].i4     = ivec4(subgroupQuadSwapVertical(lessThan(data[1].i4, ivec4(0))));
77
78    data[3].f4.x   = subgroupQuadSwapDiagonal(data[0].f4.x);
79    data[3].f4.xy  = subgroupQuadSwapDiagonal(data[1].f4.xy);
80    data[3].f4.xyz = subgroupQuadSwapDiagonal(data[2].f4.xyz);
81    data[3].f4     = subgroupQuadSwapDiagonal(data[3].f4);
82
83    data[3].i4.x   = subgroupQuadSwapDiagonal(data[0].i4.x);
84    data[3].i4.xy  = subgroupQuadSwapDiagonal(data[1].i4.xy);
85    data[3].i4.xyz = subgroupQuadSwapDiagonal(data[2].i4.xyz);
86    data[3].i4     = subgroupQuadSwapDiagonal(data[3].i4);
87
88    data[3].u4.x   = subgroupQuadSwapDiagonal(data[0].u4.x);
89    data[3].u4.xy  = subgroupQuadSwapDiagonal(data[1].u4.xy);
90    data[3].u4.xyz = subgroupQuadSwapDiagonal(data[2].u4.xyz);
91    data[3].u4     = subgroupQuadSwapDiagonal(data[3].u4);
92
93    data[3].i4.x   =   int(subgroupQuadSwapDiagonal(data[0].i4.x < 0));
94    data[3].i4.xy  = ivec2(subgroupQuadSwapDiagonal(lessThan(data[1].i4.xy, ivec2(0))));
95    data[3].i4.xyz = ivec3(subgroupQuadSwapDiagonal(lessThan(data[1].i4.xyz, ivec3(0))));
96    data[3].i4     = ivec4(subgroupQuadSwapDiagonal(lessThan(data[1].i4, ivec4(0))));
97}
98