1#include <metal_stdlib> 2#include <simd/simd.h> 3#ifdef __clang__ 4#pragma clang diagnostic ignored "-Wall" 5#endif 6using namespace metal; 7struct Uniforms { 8 half4 colorGreen; 9 half4 colorRed; 10}; 11struct Inputs { 12}; 13struct Outputs { 14 half4 sk_FragColor [[color(0)]]; 15}; 16 17template <typename T1, typename T2> 18bool operator==(const array_ref<T1> left, const array_ref<T2> right); 19template <typename T1, typename T2> 20bool operator!=(const array_ref<T1> left, const array_ref<T2> right); 21 22thread bool operator==(const float2x2 left, const float2x2 right); 23thread bool operator!=(const float2x2 left, const float2x2 right); 24 25template <size_t N> 26array<half, N> array_of_half_from_float(thread const array<float, N>& x) { 27 array<half, N> result; 28 for (int i = 0; i < N; ++i) { 29 result[i] = half(x[i]); 30 } 31 return result; 32} 33 34template <size_t N> 35array<float, N> array_of_float_from_half(thread const array<half, N>& x) { 36 array<float, N> result; 37 for (int i = 0; i < N; ++i) { 38 result[i] = float(x[i]); 39 } 40 return result; 41} 42 43template <size_t N> 44array<short3, N> array_of_short3_from_int3(thread const array<int3, N>& x) { 45 array<short3, N> result; 46 for (int i = 0; i < N; ++i) { 47 result[i] = short3(x[i]); 48 } 49 return result; 50} 51 52template <size_t N> 53array<int3, N> array_of_int3_from_short3(thread const array<short3, N>& x) { 54 array<int3, N> result; 55 for (int i = 0; i < N; ++i) { 56 result[i] = int3(x[i]); 57 } 58 return result; 59} 60 61template <size_t N> 62array<float2x2, N> array_of_float2x2_from_half2x2(thread const array<half2x2, N>& x) { 63 array<float2x2, N> result; 64 for (int i = 0; i < N; ++i) { 65 result[i] = float2x2(x[i]); 66 } 67 return result; 68} 69 70template <size_t N> 71array<half2x2, N> array_of_half2x2_from_float2x2(thread const array<float2x2, N>& x) { 72 array<half2x2, N> result; 73 for (int i = 0; i < N; ++i) { 74 result[i] = half2x2(x[i]); 75 } 76 return result; 77} 78 79template <typename T1, typename T2> 80bool operator==(const array_ref<T1> left, const array_ref<T2> right) { 81 if (left.size() != right.size()) { 82 return false; 83 } 84 for (size_t index = 0; index < left.size(); ++index) { 85 if (!all(left[index] == right[index])) { 86 return false; 87 } 88 } 89 return true; 90} 91 92template <typename T1, typename T2> 93bool operator!=(const array_ref<T1> left, const array_ref<T2> right) { 94 return !(left == right); 95} 96thread bool operator==(const float2x2 left, const float2x2 right) { 97 return all(left[0] == right[0]) && 98 all(left[1] == right[1]); 99} 100thread bool operator!=(const float2x2 left, const float2x2 right) { 101 return !(left == right); 102} 103fragment Outputs fragmentMain(Inputs _in [[stage_in]], constant Uniforms& _uniforms [[buffer(0)]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) { 104 Outputs _out; 105 (void)_out; 106 array<float, 4> f = array<float, 4>{1.0, 2.0, 3.0, 4.0}; 107 array<half, 4> h = array_of_half_from_float(f); 108 f = array_of_float_from_half(h); 109 h = array_of_half_from_float(f); 110 array<int3, 3> i3 = array<int3, 3>{int3(1), int3(2), int3(3)}; 111 array<short3, 3> s3 = array_of_short3_from_int3(i3); 112 i3 = array_of_int3_from_short3(s3); 113 s3 = array_of_short3_from_int3(i3); 114 array<half2x2, 2> h2x2 = array<half2x2, 2>{half2x2(half2(1.0h, 2.0h), half2(3.0h, 4.0h)), half2x2(half2(5.0h, 6.0h), half2(7.0h, 8.0h))}; 115 array<float2x2, 2> f2x2 = array_of_float2x2_from_half2x2(h2x2); 116 f2x2 = array_of_float2x2_from_half2x2(h2x2); 117 h2x2 = array_of_half2x2_from_float2x2(f2x2); 118 _out.sk_FragColor = (make_array_ref(f) == make_array_ref(array_of_float_from_half(h)) && make_array_ref(i3) == make_array_ref(array_of_int3_from_short3(s3))) && make_array_ref(f2x2) == make_array_ref(array_of_float2x2_from_half2x2(h2x2)) ? _uniforms.colorGreen : _uniforms.colorRed; 119 return _out; 120} 121