xref: /aosp_15_r20/external/skia/tests/sksl/shared/SwizzleAsLValueES3.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 Uniforms {
8*c8dee2aaSAndroid Build Coastguard Worker    half4 colorGreen;
9*c8dee2aaSAndroid Build Coastguard Worker    half4 colorRed;
10*c8dee2aaSAndroid Build Coastguard Worker};
11*c8dee2aaSAndroid Build Coastguard Workerstruct Inputs {
12*c8dee2aaSAndroid Build Coastguard Worker};
13*c8dee2aaSAndroid Build Coastguard Workerstruct Outputs {
14*c8dee2aaSAndroid Build Coastguard Worker    half4 sk_FragColor [[color(0)]];
15*c8dee2aaSAndroid Build Coastguard Worker};
16*c8dee2aaSAndroid Build Coastguard Workerstruct Globals {
17*c8dee2aaSAndroid Build Coastguard Worker    int gAccessCount;
18*c8dee2aaSAndroid Build Coastguard Worker};
19*c8dee2aaSAndroid Build Coastguard Workerint Z_i(thread Globals& _globals) {
20*c8dee2aaSAndroid Build Coastguard Worker    ++_globals.gAccessCount;
21*c8dee2aaSAndroid Build Coastguard Worker    return 0;
22*c8dee2aaSAndroid Build Coastguard Worker}
23*c8dee2aaSAndroid Build Coastguard Workerfragment Outputs fragmentMain(Inputs _in [[stage_in]], constant Uniforms& _uniforms [[buffer(0)]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) {
24*c8dee2aaSAndroid Build Coastguard Worker    Globals _globals{0};
25*c8dee2aaSAndroid Build Coastguard Worker    (void)_globals;
26*c8dee2aaSAndroid Build Coastguard Worker    Outputs _out;
27*c8dee2aaSAndroid Build Coastguard Worker    (void)_out;
28*c8dee2aaSAndroid Build Coastguard Worker    int _skTemp0;
29*c8dee2aaSAndroid Build Coastguard Worker    int _skTemp1;
30*c8dee2aaSAndroid Build Coastguard Worker    int _skTemp2;
31*c8dee2aaSAndroid Build Coastguard Worker    int _skTemp3;
32*c8dee2aaSAndroid Build Coastguard Worker    array<float4, 1> array;
33*c8dee2aaSAndroid Build Coastguard Worker    array[Z_i(_globals)] = float4(_uniforms.colorGreen) * 0.5;
34*c8dee2aaSAndroid Build Coastguard Worker    array[Z_i(_globals)].w = 2.0;
35*c8dee2aaSAndroid Build Coastguard Worker(_skTemp0 = Z_i(_globals),     array[_skTemp0].y = array[_skTemp0].y * 4.0);
36*c8dee2aaSAndroid Build Coastguard Worker(_skTemp1 = Z_i(_globals),     array[_skTemp1].yzw = array[_skTemp1].yzw * float3x3(0.5));
37*c8dee2aaSAndroid Build Coastguard Worker(_skTemp2 = Z_i(_globals),     array[_skTemp2].zywx = array[_skTemp2].zywx + float4(0.25, 0.0, 0.0, 0.75));
38*c8dee2aaSAndroid Build Coastguard Worker(_skTemp3 = Z_i(_globals),     array[_skTemp3].x = array[_skTemp3].x + (array[Z_i(_globals)].w <= 1.0 ? array[Z_i(_globals)].z : float(Z_i(_globals))));
39*c8dee2aaSAndroid Build Coastguard Worker    _out.sk_FragColor = _globals.gAccessCount == 8 && all(array[0] == float4(1.0, 1.0, 0.25, 1.0)) ? _uniforms.colorGreen : _uniforms.colorRed;
40*c8dee2aaSAndroid Build Coastguard Worker    return _out;
41*c8dee2aaSAndroid Build Coastguard Worker}
42