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 FragColor_0 [[color(0)]];
50    float4 FragColor_1 [[color(1)]];
51    float4 FragColor_2 [[color(2)]];
52    float4 FragColor_3 [[color(3)]];
53};
54
55struct main0_in
56{
57    float4 vA [[user(locn0)]];
58    float4 vB [[user(locn1)]];
59};
60
61// Implementation of the GLSL mod() function, which is slightly different than Metal fmod()
62template<typename Tx, typename Ty>
63inline Tx mod(Tx x, Ty y)
64{
65    return x - y * floor(x / y);
66}
67
68fragment main0_out main0(main0_in in [[stage_in]])
69{
70    main0_out out = {};
71    spvUnsafeArray<float4, 4> FragColor = {};
72    FragColor[0] = mod(in.vA, in.vB);
73    FragColor[1] = in.vA + in.vB;
74    FragColor[2] = in.vA - in.vB;
75    FragColor[3] = in.vA * in.vB;
76    out.FragColor_0 = FragColor[0];
77    out.FragColor_1 = FragColor[1];
78    out.FragColor_2 = FragColor[2];
79    out.FragColor_3 = FragColor[3];
80    return out;
81}
82
83