SPIRV-Cross/reference/shaders-msl/comp/composite-construct.comp

#pragma clang diagnostic ignored "-Wmissing-prototypes"

#include <metal_stdlib>
#include <simd/simd.h>

using namespace metal;

struct SSBO0
{
    float4 as[1];
};

struct SSBO1
{
    float4 bs[1];
};

struct Composite
{
    float4 a;
    float4 b;
};

constant float4 _43[2] = { float4(20.0), float4(40.0) };

// Implementation of an array copy function to cover GLSL's ability to copy an array via assignment.
template<typename T, uint N>
void spvArrayCopy(thread T (&dst)[N], thread const T (&src)[N])
{
    for (uint i = 0; i < N; dst[i] = src[i], i++);
}

// An overload for constant arrays.
template<typename T, uint N>
void spvArrayCopyConstant(thread T (&dst)[N], constant T (&src)[N])
{
    for (uint i = 0; i < N; dst[i] = src[i], i++);
}

kernel void main0(device SSBO0& _16 [[buffer(0)]], device SSBO1& _32 [[buffer(1)]], uint3 gl_GlobalInvocationID [[thread_position_in_grid]], uint gl_LocalInvocationIndex [[thread_index_in_threadgroup]])
{
    float4 _37[2] = { _16.as[gl_GlobalInvocationID.x], _32.bs[gl_GlobalInvocationID.x] };
    float4 values[2];
    spvArrayCopy(values, _37);
    Composite c = Composite{ values[0], _43[1] };
    _16.as[0] = values[gl_LocalInvocationIndex];
    _32.bs[1] = c.b;
}