xref: /aosp_15_r20/external/skia/tests/sksl/shared/StructIndexLookup.wgsl (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1diagnostic(off, derivative_uniformity);
2diagnostic(off, chromium.unreachable_code);
3struct FSOut {
4  @location(0) sk_FragColor: vec4<f32>,
5};
6struct _GlobalUniforms {
7  colorGreen: vec4<f32>,
8  colorRed: vec4<f32>,
9};
10@binding(0) @group(0) var<uniform> _globalUniforms: _GlobalUniforms;
11struct InnerLUT {
12  values: vec3<f32>,
13};
14struct OuterLUT {
15  inner: array<InnerLUT, 3>,
16};
17struct Root {
18  outer: array<OuterLUT, 3>,
19};
20fn _skslMain(coords: vec2<f32>) -> vec4<f32> {
21  {
22    var data: Root;
23    data.outer[0].inner[0].values = vec3<f32>(1.0, 10.0, 100.0);
24    data.outer[0].inner[1].values = vec3<f32>(2.0, 20.0, 200.0);
25    data.outer[0].inner[2].values = vec3<f32>(3.0, 30.0, 300.0);
26    data.outer[1].inner[0].values = vec3<f32>(4.0, 40.0, 400.0);
27    data.outer[1].inner[1].values = vec3<f32>(5.0, 50.0, 500.0);
28    data.outer[1].inner[2].values = vec3<f32>(6.0, 60.0, 600.0);
29    data.outer[2].inner[0].values = vec3<f32>(7.0, 70.0, 700.0);
30    data.outer[2].inner[1].values = vec3<f32>(8.0, 80.0, 800.0);
31    data.outer[2].inner[2].values = vec3<f32>(9.0, 90.0, 900.0);
32    var expected: vec3<f32> = vec3<f32>(0.0);
33    {
34      var i: i32 = 0;
35      loop {
36        {
37          {
38            var j: i32 = 0;
39            loop {
40              {
41                expected = expected + vec3<f32>(1.0, 10.0, 100.0);
42                if any(data.outer[i].inner[j].values != expected) {
43                  {
44                    return _globalUniforms.colorRed;
45                  }
46                }
47                {
48                  var k: i32 = 0;
49                  loop {
50                    {
51                      if data.outer[i].inner[j].values[k] != expected[k] {
52                        {
53                          return _globalUniforms.colorRed;
54                        }
55                      }
56                    }
57                    continuing {
58                      k = k + i32(1);
59                      break if k >= 3;
60                    }
61                  }
62                }
63              }
64              continuing {
65                j = j + i32(1);
66                break if j >= 3;
67              }
68            }
69          }
70        }
71        continuing {
72          i = i + i32(1);
73          break if i >= 3;
74        }
75      }
76    }
77    return _globalUniforms.colorGreen;
78  }
79}
80@fragment fn main() -> FSOut {
81  var _stageOut: FSOut;
82  _stageOut.sk_FragColor = _skslMain(/*fragcoord*/ vec2<f32>());
83  return _stageOut;
84}
85