xref: /aosp_15_r20/external/skia/tests/sksl/shared/SwizzleIndexStore.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  colorGreen: vec4<f32>,
8*c8dee2aaSAndroid Build Coastguard Worker  colorRed: vec4<f32>,
9*c8dee2aaSAndroid Build Coastguard Worker  testMatrix3x3: mat3x3<f32>,
10*c8dee2aaSAndroid Build Coastguard Worker  testMatrix4x4: mat4x4<f32>,
11*c8dee2aaSAndroid Build Coastguard Worker};
12*c8dee2aaSAndroid Build Coastguard Worker@binding(0) @group(0) var<uniform> _globalUniforms: _GlobalUniforms;
13*c8dee2aaSAndroid Build Coastguard Workerfn test3x3_b() -> bool {
14*c8dee2aaSAndroid Build Coastguard Worker  {
15*c8dee2aaSAndroid Build Coastguard Worker    var expected: vec3<f32> = vec3<f32>(3.0, 2.0, 1.0);
16*c8dee2aaSAndroid Build Coastguard Worker    var vec: vec3<f32>;
17*c8dee2aaSAndroid Build Coastguard Worker    {
18*c8dee2aaSAndroid Build Coastguard Worker      var c: i32 = 0;
19*c8dee2aaSAndroid Build Coastguard Worker      loop {
20*c8dee2aaSAndroid Build Coastguard Worker        {
21*c8dee2aaSAndroid Build Coastguard Worker          {
22*c8dee2aaSAndroid Build Coastguard Worker            var r: i32 = 0;
23*c8dee2aaSAndroid Build Coastguard Worker            loop {
24*c8dee2aaSAndroid Build Coastguard Worker              {
25*c8dee2aaSAndroid Build Coastguard Worker                let _skTemp0 = vec3<i32>(2, 1, 0)[r];
26*c8dee2aaSAndroid Build Coastguard Worker                vec[_skTemp0] = _globalUniforms.testMatrix3x3[c][r];
27*c8dee2aaSAndroid Build Coastguard Worker              }
28*c8dee2aaSAndroid Build Coastguard Worker              continuing {
29*c8dee2aaSAndroid Build Coastguard Worker                r = r + i32(1);
30*c8dee2aaSAndroid Build Coastguard Worker                break if r >= 3;
31*c8dee2aaSAndroid Build Coastguard Worker              }
32*c8dee2aaSAndroid Build Coastguard Worker            }
33*c8dee2aaSAndroid Build Coastguard Worker          }
34*c8dee2aaSAndroid Build Coastguard Worker          if any(vec != expected) {
35*c8dee2aaSAndroid Build Coastguard Worker            {
36*c8dee2aaSAndroid Build Coastguard Worker              return false;
37*c8dee2aaSAndroid Build Coastguard Worker            }
38*c8dee2aaSAndroid Build Coastguard Worker          }
39*c8dee2aaSAndroid Build Coastguard Worker          expected = expected + 3.0;
40*c8dee2aaSAndroid Build Coastguard Worker        }
41*c8dee2aaSAndroid Build Coastguard Worker        continuing {
42*c8dee2aaSAndroid Build Coastguard Worker          c = c + i32(1);
43*c8dee2aaSAndroid Build Coastguard Worker          break if c >= 3;
44*c8dee2aaSAndroid Build Coastguard Worker        }
45*c8dee2aaSAndroid Build Coastguard Worker      }
46*c8dee2aaSAndroid Build Coastguard Worker    }
47*c8dee2aaSAndroid Build Coastguard Worker    return true;
48*c8dee2aaSAndroid Build Coastguard Worker  }
49*c8dee2aaSAndroid Build Coastguard Worker}
50*c8dee2aaSAndroid Build Coastguard Workerfn test4x4_b() -> bool {
51*c8dee2aaSAndroid Build Coastguard Worker  {
52*c8dee2aaSAndroid Build Coastguard Worker    var expected: vec4<f32> = vec4<f32>(4.0, 3.0, 2.0, 1.0);
53*c8dee2aaSAndroid Build Coastguard Worker    var vec: vec4<f32>;
54*c8dee2aaSAndroid Build Coastguard Worker    {
55*c8dee2aaSAndroid Build Coastguard Worker      var c: i32 = 0;
56*c8dee2aaSAndroid Build Coastguard Worker      loop {
57*c8dee2aaSAndroid Build Coastguard Worker        {
58*c8dee2aaSAndroid Build Coastguard Worker          {
59*c8dee2aaSAndroid Build Coastguard Worker            var r: i32 = 0;
60*c8dee2aaSAndroid Build Coastguard Worker            loop {
61*c8dee2aaSAndroid Build Coastguard Worker              {
62*c8dee2aaSAndroid Build Coastguard Worker                let _skTemp1 = vec4<i32>(3, 2, 1, 0)[r];
63*c8dee2aaSAndroid Build Coastguard Worker                vec[_skTemp1] = _globalUniforms.testMatrix4x4[c][r];
64*c8dee2aaSAndroid Build Coastguard Worker              }
65*c8dee2aaSAndroid Build Coastguard Worker              continuing {
66*c8dee2aaSAndroid Build Coastguard Worker                r = r + i32(1);
67*c8dee2aaSAndroid Build Coastguard Worker                break if r >= 4;
68*c8dee2aaSAndroid Build Coastguard Worker              }
69*c8dee2aaSAndroid Build Coastguard Worker            }
70*c8dee2aaSAndroid Build Coastguard Worker          }
71*c8dee2aaSAndroid Build Coastguard Worker          if any(vec != expected) {
72*c8dee2aaSAndroid Build Coastguard Worker            {
73*c8dee2aaSAndroid Build Coastguard Worker              return false;
74*c8dee2aaSAndroid Build Coastguard Worker            }
75*c8dee2aaSAndroid Build Coastguard Worker          }
76*c8dee2aaSAndroid Build Coastguard Worker          expected = expected + 4.0;
77*c8dee2aaSAndroid Build Coastguard Worker        }
78*c8dee2aaSAndroid Build Coastguard Worker        continuing {
79*c8dee2aaSAndroid Build Coastguard Worker          c = c + i32(1);
80*c8dee2aaSAndroid Build Coastguard Worker          break if c >= 4;
81*c8dee2aaSAndroid Build Coastguard Worker        }
82*c8dee2aaSAndroid Build Coastguard Worker      }
83*c8dee2aaSAndroid Build Coastguard Worker    }
84*c8dee2aaSAndroid Build Coastguard Worker    return true;
85*c8dee2aaSAndroid Build Coastguard Worker  }
86*c8dee2aaSAndroid Build Coastguard Worker}
87*c8dee2aaSAndroid Build Coastguard Workerfn _skslMain(coords: vec2<f32>) -> vec4<f32> {
88*c8dee2aaSAndroid Build Coastguard Worker  {
89*c8dee2aaSAndroid Build Coastguard Worker    var _skTemp2: vec4<f32>;
90*c8dee2aaSAndroid Build Coastguard Worker    var _skTemp3: bool;
91*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp4 = test3x3_b();
92*c8dee2aaSAndroid Build Coastguard Worker    if _skTemp4 {
93*c8dee2aaSAndroid Build Coastguard Worker      let _skTemp5 = test4x4_b();
94*c8dee2aaSAndroid Build Coastguard Worker      _skTemp3 = _skTemp5;
95*c8dee2aaSAndroid Build Coastguard Worker    } else {
96*c8dee2aaSAndroid Build Coastguard Worker      _skTemp3 = false;
97*c8dee2aaSAndroid Build Coastguard Worker    }
98*c8dee2aaSAndroid Build Coastguard Worker    if _skTemp3 {
99*c8dee2aaSAndroid Build Coastguard Worker      _skTemp2 = _globalUniforms.colorGreen;
100*c8dee2aaSAndroid Build Coastguard Worker    } else {
101*c8dee2aaSAndroid Build Coastguard Worker      _skTemp2 = _globalUniforms.colorRed;
102*c8dee2aaSAndroid Build Coastguard Worker    }
103*c8dee2aaSAndroid Build Coastguard Worker    return _skTemp2;
104*c8dee2aaSAndroid Build Coastguard Worker  }
105*c8dee2aaSAndroid Build Coastguard Worker}
106*c8dee2aaSAndroid Build Coastguard Worker@fragment fn main() -> FSOut {
107*c8dee2aaSAndroid Build Coastguard Worker  var _stageOut: FSOut;
108*c8dee2aaSAndroid Build Coastguard Worker  _stageOut.sk_FragColor = _skslMain(/*fragcoord*/ vec2<f32>());
109*c8dee2aaSAndroid Build Coastguard Worker  return _stageOut;
110*c8dee2aaSAndroid Build Coastguard Worker}
111