xref: /aosp_15_r20/external/skia/tests/sksl/realistic/GaussianBlur.metal (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17) 
1#include <metal_stdlib>
2#include <simd/simd.h>
3#ifdef __clang__
4#pragma clang diagnostic ignored "-Wall"
5#endif
6using namespace metal;
7
8struct sampler2D {
9    texture2d<half> tex;
10    sampler smp;
11};
12half4 sample(sampler2D i, float2 p, float b=0) { return i.tex.sample(i.smp, p, bias(b)); }
13half4 sample(sampler2D i, float3 p, float b=0) { return i.tex.sample(i.smp, p.xy / p.z, bias(b)); }
14half4 sampleLod(sampler2D i, float2 p, float lod) { return i.tex.sample(i.smp, p, level(lod)); }
15half4 sampleLod(sampler2D i, float3 p, float lod) {
16    return i.tex.sample(i.smp, p.xy / p.z, level(lod));
17}
18half4 sampleGrad(sampler2D i, float2 p, float2 dPdx, float2 dPdy) {
19    return i.tex.sample(i.smp, p, gradient2d(dPdx, dPdy));
20}
21
22struct Inputs {
23    float2 vLocalCoord_Stage0 [[user(locn0)]];
24};
25struct Outputs {
26    half4 sk_FragColor [[color(0)]];
27};
28struct uniformBuffer {
29    float4 sk_RTAdjust;
30    half2 uIncrement_Stage1_c0;
31    char pad0[12];
32    array<half4, 7> uKernel_Stage1_c0;
33    char pad1[56];
34    float3x3 umatrix_Stage1_c0_c0;
35    half4 uborder_Stage1_c0_c0_c0;
36    char pad2[8];
37    float4 usubset_Stage1_c0_c0_c0;
38    float4 unorm_Stage1_c0_c0_c0;
39};
40struct Globals {
41    sampler2D uTextureSampler_0_Stage1;
42    constant uniformBuffer* _anonInterface0;
43};
44half4 MatrixEffect_Stage1_c0_c0_h4h4f2(thread Globals& _globals, half4 _input, float2 _coords) {
45    float2 _1_inCoord = (_globals._anonInterface0->umatrix_Stage1_c0_c0 * float3(_coords, 1.0)).xy;
46    _1_inCoord *= _globals._anonInterface0->unorm_Stage1_c0_c0_c0.xy;
47    float2 _2_subsetCoord;
48    _2_subsetCoord.x = _1_inCoord.x;
49    _2_subsetCoord.y = _1_inCoord.y;
50    float2 _3_clampedCoord = _2_subsetCoord;
51    half4 _4_textureColor = sample(_globals.uTextureSampler_0_Stage1, _3_clampedCoord * _globals._anonInterface0->unorm_Stage1_c0_c0_c0.zw);
52    float _5_snappedX = floor(_1_inCoord.x + 0.001) + 0.5;
53    if (_5_snappedX < _globals._anonInterface0->usubset_Stage1_c0_c0_c0.x || _5_snappedX > _globals._anonInterface0->usubset_Stage1_c0_c0_c0.z) {
54        _4_textureColor = _globals._anonInterface0->uborder_Stage1_c0_c0_c0;
55    }
56    return _4_textureColor;
57}
58fragment Outputs fragmentMain(Inputs _in [[stage_in]], texture2d<half> uTextureSampler_0_Stage1_Tex [[texture(0)]], sampler uTextureSampler_0_Stage1_Smplr [[sampler(0)]], constant uniformBuffer& _anonInterface0 [[buffer(0)]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) {
59    Globals _globals{{uTextureSampler_0_Stage1_Tex, uTextureSampler_0_Stage1_Smplr}, &_anonInterface0};
60    (void)_globals;
61    Outputs _out;
62    (void)_out;
63    half4 outputColor_Stage0;
64    half4 outputCoverage_Stage0;
65    {
66        outputColor_Stage0 = half4(1.0h);
67        outputCoverage_Stage0 = half4(1.0h);
68    }
69    half4 _6_output = half4(0.0h);
70    float2 _7_coord = _in.vLocalCoord_Stage0 - float2(12.0h * _globals._anonInterface0->uIncrement_Stage1_c0);
71    float2 _8_coordSampled = float2(0.0);
72    _8_coordSampled = _7_coord;
73    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[0].x;
74    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
75    _8_coordSampled = _7_coord;
76    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[0].y;
77    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
78    _8_coordSampled = _7_coord;
79    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[0].z;
80    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
81    _8_coordSampled = _7_coord;
82    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[0].w;
83    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
84    _8_coordSampled = _7_coord;
85    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[1].x;
86    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
87    _8_coordSampled = _7_coord;
88    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[1].y;
89    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
90    _8_coordSampled = _7_coord;
91    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[1].z;
92    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
93    _8_coordSampled = _7_coord;
94    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[1].w;
95    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
96    _8_coordSampled = _7_coord;
97    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[2].x;
98    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
99    _8_coordSampled = _7_coord;
100    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[2].y;
101    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
102    _8_coordSampled = _7_coord;
103    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[2].z;
104    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
105    _8_coordSampled = _7_coord;
106    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[2].w;
107    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
108    _8_coordSampled = _7_coord;
109    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[3].x;
110    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
111    _8_coordSampled = _7_coord;
112    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[3].y;
113    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
114    _8_coordSampled = _7_coord;
115    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[3].z;
116    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
117    _8_coordSampled = _7_coord;
118    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[3].w;
119    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
120    _8_coordSampled = _7_coord;
121    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[4].x;
122    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
123    _8_coordSampled = _7_coord;
124    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[4].y;
125    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
126    _8_coordSampled = _7_coord;
127    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[4].z;
128    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
129    _8_coordSampled = _7_coord;
130    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[4].w;
131    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
132    _8_coordSampled = _7_coord;
133    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[5].x;
134    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
135    _8_coordSampled = _7_coord;
136    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[5].y;
137    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
138    _8_coordSampled = _7_coord;
139    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[5].z;
140    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
141    _8_coordSampled = _7_coord;
142    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[5].w;
143    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
144    _8_coordSampled = _7_coord;
145    _6_output += MatrixEffect_Stage1_c0_c0_h4h4f2(_globals, outputColor_Stage0, _8_coordSampled) * _globals._anonInterface0->uKernel_Stage1_c0[6].x;
146    _7_coord += float2(_globals._anonInterface0->uIncrement_Stage1_c0);
147    _6_output *= outputColor_Stage0;
148    half4 output_Stage1 = _6_output;
149    {
150        _out.sk_FragColor = output_Stage1 * outputCoverage_Stage0;
151    }
152    return _out;
153}
154