| #include <metal_stdlib> |
| using namespace metal; |
| |
| template<typename T, size_t N> |
| struct tint_array { |
| const constant T& operator[](size_t i) const constant { return elements[i]; } |
| device T& operator[](size_t i) device { return elements[i]; } |
| const device T& operator[](size_t i) const device { return elements[i]; } |
| thread T& operator[](size_t i) thread { return elements[i]; } |
| const thread T& operator[](size_t i) const thread { return elements[i]; } |
| threadgroup T& operator[](size_t i) threadgroup { return elements[i]; } |
| const threadgroup T& operator[](size_t i) const threadgroup { return elements[i]; } |
| T elements[N]; |
| }; |
| |
| struct tint_module_vars_struct { |
| const constant tint_array<float3x4, 4>* u; |
| device float* s; |
| }; |
| |
| kernel void f(const constant tint_array<float3x4, 4>* u [[buffer(0)]], device float* s [[buffer(1)]]) { |
| tint_module_vars_struct const tint_module_vars = tint_module_vars_struct{.u=u, .s=s}; |
| float4x3 const t = transpose((*tint_module_vars.u)[2u]); |
| float const l = length((*tint_module_vars.u)[0u][1u].ywxz); |
| float const a = abs((*tint_module_vars.u)[0u][1u].ywxz.x); |
| float const v = (t[0u].x + float(l)); |
| (*tint_module_vars.s) = (v + float(a)); |
| } |