1#pragma clang diagnostic ignored "-Wmissing-prototypes"
2#pragma clang diagnostic ignored "-Wmissing-braces"
3
4#include <metal_stdlib>
5#include <simd/simd.h>
6
7using namespace metal;
8
9template<typename T, size_t Num>
10struct spvUnsafeArray
11{
12    T elements[Num ? Num : 1];
13    
14    thread T& operator [] (size_t pos) thread
15    {
16        return elements[pos];
17    }
18    constexpr const thread T& operator [] (size_t pos) const thread
19    {
20        return elements[pos];
21    }
22    
23    device T& operator [] (size_t pos) device
24    {
25        return elements[pos];
26    }
27    constexpr const device T& operator [] (size_t pos) const device
28    {
29        return elements[pos];
30    }
31    
32    constexpr const constant T& operator [] (size_t pos) const constant
33    {
34        return elements[pos];
35    }
36    
37    threadgroup T& operator [] (size_t pos) threadgroup
38    {
39        return elements[pos];
40    }
41    constexpr const threadgroup T& operator [] (size_t pos) const threadgroup
42    {
43        return elements[pos];
44    }
45};
46
47struct main0_out
48{
49    float4 gl_Position;
50    float gl_PointSize;
51};
52
53struct main0_patchOut
54{
55    spvUnsafeArray<float4, 2> v1;
56    float4 v3;
57};
58
59static inline __attribute__((always_inline))
60void write_in_func(threadgroup float4 (&v0)[4], thread uint& gl_InvocationID, device spvUnsafeArray<float4, 2>& v1, device float4& v3, device main0_out* thread & gl_out)
61{
62    v0[gl_InvocationID] = float4(1.0);
63    v0[gl_InvocationID].z = 3.0;
64    if (gl_InvocationID == 0)
65    {
66        v1[0] = float4(2.0);
67        ((device float*)&v1[0])[0u] = 3.0;
68        v1[1] = float4(2.0);
69        ((device float*)&v1[1])[0u] = 5.0;
70    }
71    v3 = float4(5.0);
72    gl_out[gl_InvocationID].gl_Position = float4(10.0);
73    gl_out[gl_InvocationID].gl_Position.z = 20.0;
74    gl_out[gl_InvocationID].gl_PointSize = 40.0;
75}
76
77kernel void main0(uint gl_InvocationID [[thread_index_in_threadgroup]], uint gl_PrimitiveID [[threadgroup_position_in_grid]], device main0_out* spvOut [[buffer(28)]], constant uint* spvIndirectParams [[buffer(29)]], device main0_patchOut* spvPatchOut [[buffer(27)]], device MTLQuadTessellationFactorsHalf* spvTessLevel [[buffer(26)]])
78{
79    threadgroup float4 v0[4];
80    device main0_out* gl_out = &spvOut[gl_PrimitiveID * 4];
81    device main0_patchOut& patchOut = spvPatchOut[gl_PrimitiveID];
82    write_in_func(v0, gl_InvocationID, patchOut.v1, patchOut.v3, gl_out);
83}
84
85