#pragma clang diagnostic ignored "-Wmissing-prototypes"
#pragma clang diagnostic ignored "-Wmissing-braces"

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

using namespace metal;

template<typename T, size_t Num>
struct spvUnsafeArray
{
    T elements[Num ? Num : 1];
    
    thread T& operator [] (size_t pos) thread
    {
        return elements[pos];
    }
    constexpr const thread T& operator [] (size_t pos) const thread
    {
        return elements[pos];
    }
    
    device T& operator [] (size_t pos) device
    {
        return elements[pos];
    }
    constexpr const device T& operator [] (size_t pos) const device
    {
        return elements[pos];
    }
    
    constexpr const constant T& operator [] (size_t pos) const constant
    {
        return elements[pos];
    }
    
    threadgroup T& operator [] (size_t pos) threadgroup
    {
        return elements[pos];
    }
    constexpr const threadgroup T& operator [] (size_t pos) const threadgroup
    {
        return elements[pos];
    }
};

template<typename T, uint N>
inline void spvArrayCopyFromConstantToStack(thread T (&dst)[N], constant T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}
template<typename T, uint V, uint N>
inline void spvArrayCopyFromConstantToStack(thread T (&dst)[N], constant vec<T, V> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = reinterpret_cast<constant T &>(src[i]);
    }
}

template<typename T, uint Vdst, uint Vsrc, uint N>
inline enable_if_t<Vdst != Vsrc> spvArrayCopyFromConstantToStack(thread vec<T, Vdst> (&dst)[N], constant vec<T, Vsrc> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = reinterpret_cast<constant vec<T, Vdst> &>(src[i]);
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromConstantToThreadGroup(threadgroup T (&dst)[N], constant T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}
template<typename T, uint V, uint N>
inline void spvArrayCopyFromConstantToThreadGroup(threadgroup T (&dst)[N], constant vec<T, V> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = reinterpret_cast<constant T &>(src[i]);
    }
}

template<typename T, uint Vdst, uint Vsrc, uint N>
inline enable_if_t<Vdst != Vsrc> spvArrayCopyFromConstantToThreadGroup(threadgroup vec<T, Vdst> (&dst)[N], constant vec<T, Vsrc> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = reinterpret_cast<constant vec<T, Vdst> &>(src[i]);
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromStackToStack(thread T (&dst)[N], thread const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromStackToThreadGroup(threadgroup T (&dst)[N], thread const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromThreadGroupToStack(thread T (&dst)[N], threadgroup const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromThreadGroupToThreadGroup(threadgroup T (&dst)[N], threadgroup const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromDeviceToDevice(device T (&dst)[N], device const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromConstantToDevice(device T (&dst)[N], constant T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromStackToDevice(device T (&dst)[N], thread const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}
template<typename T, uint V, uint N>
inline void spvArrayCopyFromStackToDevice(device vec<T, V> (&dst)[N], thread const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        reinterpret_cast<device T &>(dst[i]) = src[i];
    }
}

template<typename T, uint Vdst, uint Vsrc, uint N>
inline enable_if_t<Vdst != Vsrc> spvArrayCopyFromStackToDevice(device vec<T, Vdst> (&dst)[N], thread const vec<T, Vsrc> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        reinterpret_cast<device vec<T, Vsrc> &>(dst[i]) = src[i];
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromThreadGroupToDevice(device T (&dst)[N], threadgroup const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}
template<typename T, uint V, uint N>
inline void spvArrayCopyFromThreadGroupToDevice(device vec<T, V> (&dst)[N], threadgroup const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        reinterpret_cast<device T &>(dst[i]) = src[i];
    }
}

template<typename T, uint Vdst, uint Vsrc, uint N>
inline enable_if_t<Vdst != Vsrc> spvArrayCopyFromThreadGroupToDevice(device vec<T, Vdst> (&dst)[N], threadgroup const vec<T, Vsrc> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        reinterpret_cast<device vec<T, Vsrc> &>(dst[i]) = src[i];
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromDeviceToStack(thread T (&dst)[N], device const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}
template<typename T, uint V, uint N>
inline void spvArrayCopyFromDeviceToStack(thread T (&dst)[N], device const vec<T, V> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = reinterpret_cast<device const T &>(src[i]);
    }
}

template<typename T, uint Vdst, uint Vsrc, uint N>
inline enable_if_t<Vdst != Vsrc> spvArrayCopyFromDeviceToStack(thread vec<T, Vdst> (&dst)[N], device const vec<T, Vsrc> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = reinterpret_cast<device const vec<T, Vdst> &>(src[i]);
    }
}

template<typename T, uint N>
inline void spvArrayCopyFromDeviceToThreadGroup(threadgroup T (&dst)[N], device const T (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = src[i];
    }
}
template<typename T, uint V, uint N>
inline void spvArrayCopyFromDeviceToThreadGroup(threadgroup T (&dst)[N], device const vec<T, V> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = reinterpret_cast<device const T &>(src[i]);
    }
}

template<typename T, uint Vdst, uint Vsrc, uint N>
inline enable_if_t<Vdst != Vsrc> spvArrayCopyFromDeviceToThreadGroup(threadgroup vec<T, Vdst> (&dst)[N], device const vec<T, Vsrc> (&src)[N])
{
    for (uint i = 0; i < N; i++)
    {
        dst[i] = reinterpret_cast<device const vec<T, Vdst> &>(src[i]);
    }
}

struct SSBO
{
    float4 b[5];
    float4 c[5];
};

kernel void main0(device SSBO& _7 [[buffer(0)]])
{
    spvUnsafeArray<float, 5> a;
    spvArrayCopyFromDeviceToStack(a.elements, _7.b);
    spvArrayCopyFromDeviceToStack(a.elements, _7.b);
    spvArrayCopyFromStackToDevice(_7.b, a.elements);
    spvArrayCopyFromStackToDevice(_7.c, a.elements);
}

