| #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]; |
| }; |
| |
| int2 tint_clamp(int2 e, int2 low, int2 high) { |
| return min(max(e, low), high); |
| } |
| |
| struct GammaTransferParams { |
| /* 0x0000 */ float G; |
| /* 0x0004 */ float A; |
| /* 0x0008 */ float B; |
| /* 0x000c */ float C; |
| /* 0x0010 */ float D; |
| /* 0x0014 */ float E; |
| /* 0x0018 */ float F; |
| /* 0x001c */ uint padding; |
| }; |
| |
| struct ExternalTextureParams { |
| /* 0x0000 */ uint numPlanes; |
| /* 0x0004 */ uint doYuvToRgbConversionOnly; |
| /* 0x0008 */ tint_array<int8_t, 8> tint_pad; |
| /* 0x0010 */ float3x4 yuvToRgbConversionMatrix; |
| /* 0x0040 */ GammaTransferParams gammaDecodeParams; |
| /* 0x0060 */ GammaTransferParams gammaEncodeParams; |
| /* 0x0080 */ float3x3 gamutConversionMatrix; |
| /* 0x00b0 */ float2x2 rotationMatrix; |
| }; |
| |
| float3 gammaCorrection(float3 v, GammaTransferParams params) { |
| bool3 const cond = (fabs(v) < float3(params.D)); |
| float3 const t_1 = (sign(v) * ((params.C * fabs(v)) + params.F)); |
| float3 const f = (sign(v) * (pow(((params.A * fabs(v)) + params.B), float3(params.G)) + params.E)); |
| return select(f, t_1, cond); |
| } |
| |
| float4 textureLoadExternal(texture2d<float, access::sample> plane0, texture2d<float, access::sample> plane1, int2 coord, ExternalTextureParams params) { |
| float3 color = 0.0f; |
| if ((params.numPlanes == 1u)) { |
| color = float4(plane0.read(uint2(coord), 0)).rgb; |
| } else { |
| color = (float4(plane0.read(uint2(coord), 0)[0], float4(plane1.read(uint2(coord), 0)).rg, 1.0f) * params.yuvToRgbConversionMatrix); |
| } |
| if ((params.doYuvToRgbConversionOnly == 0u)) { |
| color = gammaCorrection(color, params.gammaDecodeParams); |
| color = (params.gamutConversionMatrix * color); |
| color = gammaCorrection(color, params.gammaEncodeParams); |
| } |
| return float4(color, 1.0f); |
| } |
| |
| kernel void tint_symbol(texture2d<float, access::sample> tint_symbol_5 [[texture(0)]], texture2d<float, access::sample> tint_symbol_6 [[texture(1)]], const constant ExternalTextureParams* tint_symbol_7 [[buffer(0)]], texture2d<float, access::write> tint_symbol_8 [[texture(2)]]) { |
| int2 const tint_symbol_1 = tint_clamp(int2(10), int2(0), int2((uint2(uint2(tint_symbol_5.get_width(), tint_symbol_5.get_height())) - uint2(1u)))); |
| float4 red = textureLoadExternal(tint_symbol_5, tint_symbol_6, tint_symbol_1, *(tint_symbol_7)); |
| int2 const tint_symbol_2 = tint_clamp(int2(0), int2(0), int2((uint2(uint2(tint_symbol_8.get_width(), tint_symbol_8.get_height())) - uint2(1u)))); |
| tint_symbol_8.write(red, uint2(tint_symbol_2)); |
| int2 const tint_symbol_3 = tint_clamp(int2(70, 118), int2(0), int2((uint2(uint2(tint_symbol_5.get_width(), tint_symbol_5.get_height())) - uint2(1u)))); |
| float4 green = textureLoadExternal(tint_symbol_5, tint_symbol_6, tint_symbol_3, *(tint_symbol_7)); |
| int2 const tint_symbol_4 = tint_clamp(int2(1, 0), int2(0), int2((uint2(uint2(tint_symbol_8.get_width(), tint_symbol_8.get_height())) - uint2(1u)))); |
| tint_symbol_8.write(green, uint2(tint_symbol_4)); |
| return; |
| } |
| |