xref: /aosp_15_r20/external/skia/tests/sksl/blend/BlendSoftLight.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;
7constant const half sk_PrivkGuardedDivideEpsilon = half(false ? 1e-08 : 0.0);
8struct Uniforms {
9    half4 src;
10    half4 dst;
11};
12struct Inputs {
13};
14struct Outputs {
15    half4 sk_FragColor [[color(0)]];
16};
17half soft_light_component_Qhh2h2(half2 s, half2 d);
18half soft_light_component_Qhh2h2(half2 s, half2 d) {
19    if (2.0h * s.x <= s.y) {
20        return (((d.x * d.x) * (s.y - 2.0h * s.x)) / (d.y + sk_PrivkGuardedDivideEpsilon) + (1.0h - d.y) * s.x) + d.x * ((-s.y + 2.0h * s.x) + 1.0h);
21    } else if (4.0h * d.x <= d.y) {
22        half DSqd = d.x * d.x;
23        half DCub = DSqd * d.x;
24        half DaSqd = d.y * d.y;
25        half DaCub = DaSqd * d.y;
26        return (((DaSqd * (s.x - d.x * ((3.0h * s.y - 6.0h * s.x) - 1.0h)) + ((12.0h * d.y) * DSqd) * (s.y - 2.0h * s.x)) - (16.0h * DCub) * (s.y - 2.0h * s.x)) - DaCub * s.x) / (DaSqd + sk_PrivkGuardedDivideEpsilon);
27    } else {
28        return ((d.x * ((s.y - 2.0h * s.x) + 1.0h) + s.x) - sqrt(d.y * d.x) * (s.y - 2.0h * s.x)) - d.y * s.x;
29    }
30}
31fragment Outputs fragmentMain(Inputs _in [[stage_in]], constant Uniforms& _uniforms [[buffer(0)]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) {
32    Outputs _out;
33    (void)_out;
34    _out.sk_FragColor = _uniforms.dst.w == 0.0h ? _uniforms.src : half4(soft_light_component_Qhh2h2(_uniforms.src.xw, _uniforms.dst.xw), soft_light_component_Qhh2h2(_uniforms.src.yw, _uniforms.dst.yw), soft_light_component_Qhh2h2(_uniforms.src.zw, _uniforms.dst.zw), _uniforms.src.w + (1.0h - _uniforms.src.w) * _uniforms.dst.w);
35    return _out;
36}
37