xref: /aosp_15_r20/external/skia/tests/sksl/metal/SwizzleHelper.metal (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1*c8dee2aaSAndroid Build Coastguard Worker#include <metal_stdlib>
2*c8dee2aaSAndroid Build Coastguard Worker#include <simd/simd.h>
3*c8dee2aaSAndroid Build Coastguard Worker#ifdef __clang__
4*c8dee2aaSAndroid Build Coastguard Worker#pragma clang diagnostic ignored "-Wall"
5*c8dee2aaSAndroid Build Coastguard Worker#endif
6*c8dee2aaSAndroid Build Coastguard Workerusing namespace metal;
7*c8dee2aaSAndroid Build Coastguard Workerstruct Inputs {
8*c8dee2aaSAndroid Build Coastguard Worker};
9*c8dee2aaSAndroid Build Coastguard Workerstruct Outputs {
10*c8dee2aaSAndroid Build Coastguard Worker    half4 sk_FragColor [[color(0)]];
11*c8dee2aaSAndroid Build Coastguard Worker};
12*c8dee2aaSAndroid Build Coastguard Workerstruct Globals {
13*c8dee2aaSAndroid Build Coastguard Worker    half2 glob;
14*c8dee2aaSAndroid Build Coastguard Worker};
15*c8dee2aaSAndroid Build Coastguard Workerhalf4 fn_h4hh2h2h3(thread Globals& _globals, half a, thread half2& b, thread half2& c, thread half3& d) {
16*c8dee2aaSAndroid Build Coastguard Worker    a = _out.sk_FragColor.x + a;
17*c8dee2aaSAndroid Build Coastguard Worker    b = _out.sk_FragColor.yz - _globals.glob.y;
18*c8dee2aaSAndroid Build Coastguard Worker    c *= a;
19*c8dee2aaSAndroid Build Coastguard Worker    d = _out.sk_FragColor.www / d;
20*c8dee2aaSAndroid Build Coastguard Worker    return half4(a, b.x, c.y, d.x);
21*c8dee2aaSAndroid Build Coastguard Worker}
22*c8dee2aaSAndroid Build Coastguard Workerfragment Outputs fragmentMain(Inputs _in [[stage_in]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) {
23*c8dee2aaSAndroid Build Coastguard Worker    Globals _globals{half2(1.0h)};
24*c8dee2aaSAndroid Build Coastguard Worker    (void)_globals;
25*c8dee2aaSAndroid Build Coastguard Worker    Outputs _out;
26*c8dee2aaSAndroid Build Coastguard Worker    (void)_out;
27*c8dee2aaSAndroid Build Coastguard Worker    half2 _skTemp0;
28*c8dee2aaSAndroid Build Coastguard Worker    half2 _skTemp1;
29*c8dee2aaSAndroid Build Coastguard Worker    half3 _skTemp2;
30*c8dee2aaSAndroid Build Coastguard Worker    half4 _skTemp3;
31*c8dee2aaSAndroid Build Coastguard Worker    half2 a = half2(1.0h);
32*c8dee2aaSAndroid Build Coastguard Worker    half3 b = half3(2.0h);
33*c8dee2aaSAndroid Build Coastguard Worker    half3x3 d = half3x3(4.0h);
34*c8dee2aaSAndroid Build Coastguard Worker    _out.sk_FragColor = ((_skTemp3 = fn_h4hh2h2h3(_globals, a.x, _skTemp0, (_skTemp1 = _globals.glob.yx), (_skTemp2 = d[1].zyx))), (b.yz = _skTemp0), (_globals.glob.yx = _skTemp1), (d[1].zyx = _skTemp2), _skTemp3);
35*c8dee2aaSAndroid Build Coastguard Worker    return _out;
36*c8dee2aaSAndroid Build Coastguard Worker}
37