xref: /aosp_15_r20/external/skia/tests/sksl/shared/VectorScalarMath.wgsl (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1*c8dee2aaSAndroid Build Coastguard Workerdiagnostic(off, derivative_uniformity);
2*c8dee2aaSAndroid Build Coastguard Workerdiagnostic(off, chromium.unreachable_code);
3*c8dee2aaSAndroid Build Coastguard Workerstruct FSOut {
4*c8dee2aaSAndroid Build Coastguard Worker  @location(0) sk_FragColor: vec4<f32>,
5*c8dee2aaSAndroid Build Coastguard Worker};
6*c8dee2aaSAndroid Build Coastguard Workerstruct _GlobalUniforms {
7*c8dee2aaSAndroid Build Coastguard Worker  colorRed: vec4<f32>,
8*c8dee2aaSAndroid Build Coastguard Worker  colorGreen: vec4<f32>,
9*c8dee2aaSAndroid Build Coastguard Worker  unknownInput: f32,
10*c8dee2aaSAndroid Build Coastguard Worker};
11*c8dee2aaSAndroid Build Coastguard Worker@binding(0) @group(0) var<uniform> _globalUniforms: _GlobalUniforms;
12*c8dee2aaSAndroid Build Coastguard Workerfn test_int_b() -> bool {
13*c8dee2aaSAndroid Build Coastguard Worker  {
14*c8dee2aaSAndroid Build Coastguard Worker    var ok: bool = true;
15*c8dee2aaSAndroid Build Coastguard Worker    let inputRed: vec4<i32> = vec4<i32>(_globalUniforms.colorRed);
16*c8dee2aaSAndroid Build Coastguard Worker    let inputGreen: vec4<i32> = vec4<i32>(_globalUniforms.colorGreen);
17*c8dee2aaSAndroid Build Coastguard Worker    var x: vec4<i32> = inputRed + 2;
18*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(3, 2, 2, 3));
19*c8dee2aaSAndroid Build Coastguard Worker    x = inputGreen.ywxz - 2;
20*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(-1, -1, -2, -2));
21*c8dee2aaSAndroid Build Coastguard Worker    x = inputRed + inputGreen.y;
22*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(2, 1, 1, 2));
23*c8dee2aaSAndroid Build Coastguard Worker    x = vec4<i32>((inputGreen.wyw * 9), x.w);
24*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(9, 9, 9, 2));
25*c8dee2aaSAndroid Build Coastguard Worker    x = vec4<i32>((x.zw / 4), x.zw);
26*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(2, 0, 9, 2));
27*c8dee2aaSAndroid Build Coastguard Worker    x = (inputRed * 5).yxwz;
28*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(0, 5, 5, 0));
29*c8dee2aaSAndroid Build Coastguard Worker    x = 2 + inputRed;
30*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(3, 2, 2, 3));
31*c8dee2aaSAndroid Build Coastguard Worker    x = 10 - inputGreen.ywxz;
32*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(9, 9, 10, 10));
33*c8dee2aaSAndroid Build Coastguard Worker    x = inputRed.x + inputGreen;
34*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(1, 2, 1, 2));
35*c8dee2aaSAndroid Build Coastguard Worker    x = vec4<i32>((8 * inputGreen.wyw), x.w);
36*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(8, 8, 8, 2));
37*c8dee2aaSAndroid Build Coastguard Worker    x = vec4<i32>((36 / x.zw), x.zw);
38*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(4, 18, 8, 2));
39*c8dee2aaSAndroid Build Coastguard Worker    x = (37 / x).yxwz;
40*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(2, 9, 18, 4));
41*c8dee2aaSAndroid Build Coastguard Worker    x = x + 2;
42*c8dee2aaSAndroid Build Coastguard Worker    x = x * 2;
43*c8dee2aaSAndroid Build Coastguard Worker    x = x - 4;
44*c8dee2aaSAndroid Build Coastguard Worker    x = x / 2;
45*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(2, 9, 18, 4));
46*c8dee2aaSAndroid Build Coastguard Worker    x = x + 2;
47*c8dee2aaSAndroid Build Coastguard Worker    x = x * 2;
48*c8dee2aaSAndroid Build Coastguard Worker    x = x - 4;
49*c8dee2aaSAndroid Build Coastguard Worker    x = x / 2;
50*c8dee2aaSAndroid Build Coastguard Worker    ok = ok && all(x == vec4<i32>(2, 9, 18, 4));
51*c8dee2aaSAndroid Build Coastguard Worker    return ok;
52*c8dee2aaSAndroid Build Coastguard Worker  }
53*c8dee2aaSAndroid Build Coastguard Worker}
54*c8dee2aaSAndroid Build Coastguard Workerfn _skslMain(coords: vec2<f32>) -> vec4<f32> {
55*c8dee2aaSAndroid Build Coastguard Worker  {
56*c8dee2aaSAndroid Build Coastguard Worker    var _0_ok: bool = true;
57*c8dee2aaSAndroid Build Coastguard Worker    let _1_inputRed: vec4<f32> = _globalUniforms.colorRed;
58*c8dee2aaSAndroid Build Coastguard Worker    let _2_inputGreen: vec4<f32> = _globalUniforms.colorGreen;
59*c8dee2aaSAndroid Build Coastguard Worker    var _3_x: vec4<f32> = _1_inputRed + 2.0;
60*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(3.0, 2.0, 2.0, 3.0));
61*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _2_inputGreen.ywxz - 2.0;
62*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(-1.0, -1.0, -2.0, -2.0));
63*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _1_inputRed + _2_inputGreen.y;
64*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(2.0, 1.0, 1.0, 2.0));
65*c8dee2aaSAndroid Build Coastguard Worker    _3_x = vec4<f32>((_2_inputGreen.wyw * 9.0), _3_x.w);
66*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(9.0, 9.0, 9.0, 2.0));
67*c8dee2aaSAndroid Build Coastguard Worker    _3_x = vec4<f32>((_3_x.zw * 2.0), _3_x.zw);
68*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(18.0, 4.0, 9.0, 2.0));
69*c8dee2aaSAndroid Build Coastguard Worker    _3_x = (_1_inputRed * 5.0).yxwz;
70*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(0.0, 5.0, 5.0, 0.0));
71*c8dee2aaSAndroid Build Coastguard Worker    _3_x = 2.0 + _1_inputRed;
72*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(3.0, 2.0, 2.0, 3.0));
73*c8dee2aaSAndroid Build Coastguard Worker    _3_x = 10.0 - _2_inputGreen.ywxz;
74*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(9.0, 9.0, 10.0, 10.0));
75*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _1_inputRed.x + _2_inputGreen;
76*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(1.0, 2.0, 1.0, 2.0));
77*c8dee2aaSAndroid Build Coastguard Worker    _3_x = vec4<f32>((8.0 * _2_inputGreen.wyw), _3_x.w);
78*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(8.0, 8.0, 8.0, 2.0));
79*c8dee2aaSAndroid Build Coastguard Worker    _3_x = vec4<f32>((32.0 / _3_x.zw), _3_x.zw);
80*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(4.0, 16.0, 8.0, 2.0));
81*c8dee2aaSAndroid Build Coastguard Worker    _3_x = (32.0 / _3_x).yxwz;
82*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(2.0, 8.0, 16.0, 4.0));
83*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _3_x + 2.0;
84*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _3_x * 2.0;
85*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _3_x - 4.0;
86*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _3_x * 0.5;
87*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(2.0, 8.0, 16.0, 4.0));
88*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _3_x + 2.0;
89*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _3_x * 2.0;
90*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _3_x - 4.0;
91*c8dee2aaSAndroid Build Coastguard Worker    _3_x = _3_x * 0.5;
92*c8dee2aaSAndroid Build Coastguard Worker    _0_ok = _0_ok && all(_3_x == vec4<f32>(2.0, 8.0, 16.0, 4.0));
93*c8dee2aaSAndroid Build Coastguard Worker    var _skTemp0: vec4<f32>;
94*c8dee2aaSAndroid Build Coastguard Worker    var _skTemp1: bool;
95*c8dee2aaSAndroid Build Coastguard Worker    if _0_ok {
96*c8dee2aaSAndroid Build Coastguard Worker      let _skTemp2 = test_int_b();
97*c8dee2aaSAndroid Build Coastguard Worker      _skTemp1 = _skTemp2;
98*c8dee2aaSAndroid Build Coastguard Worker    } else {
99*c8dee2aaSAndroid Build Coastguard Worker      _skTemp1 = false;
100*c8dee2aaSAndroid Build Coastguard Worker    }
101*c8dee2aaSAndroid Build Coastguard Worker    if _skTemp1 {
102*c8dee2aaSAndroid Build Coastguard Worker      _skTemp0 = _globalUniforms.colorGreen;
103*c8dee2aaSAndroid Build Coastguard Worker    } else {
104*c8dee2aaSAndroid Build Coastguard Worker      _skTemp0 = _globalUniforms.colorRed;
105*c8dee2aaSAndroid Build Coastguard Worker    }
106*c8dee2aaSAndroid Build Coastguard Worker    return _skTemp0;
107*c8dee2aaSAndroid Build Coastguard Worker  }
108*c8dee2aaSAndroid Build Coastguard Worker}
109*c8dee2aaSAndroid Build Coastguard Worker@fragment fn main() -> FSOut {
110*c8dee2aaSAndroid Build Coastguard Worker  var _stageOut: FSOut;
111*c8dee2aaSAndroid Build Coastguard Worker  _stageOut.sk_FragColor = _skslMain(/*fragcoord*/ vec2<f32>());
112*c8dee2aaSAndroid Build Coastguard Worker  return _stageOut;
113*c8dee2aaSAndroid Build Coastguard Worker}
114