xref: /aosp_15_r20/external/skia/tests/sksl/shared/MatrixSwizzleStore.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  testMatrix3x3: mat3x3<f32>,
10  testMatrix4x4: mat4x4<f32>,
11};
12@binding(0) @group(0) var<uniform> _globalUniforms: _GlobalUniforms;
13fn test4x4_b() -> bool {
14  {
15    var matrix: mat4x4<f32>;
16    var values: vec4<f32> = vec4<f32>(4.0, 3.0, 2.0, 1.0);
17    {
18      var index: i32 = 0;
19      loop {
20        {
21          matrix[index] = vec4<f32>((values.xw), matrix[index].yz).yzwx;
22          matrix[index] = vec4<f32>(matrix[index].xw, (values.yz)).xwzy;
23          values = values + 4.0;
24        }
25        continuing {
26          index = index + i32(1);
27          break if index >= 4;
28        }
29      }
30    }
31    return (all(matrix[0] == _globalUniforms.testMatrix4x4[0]) && all(matrix[1] == _globalUniforms.testMatrix4x4[1]) && all(matrix[2] == _globalUniforms.testMatrix4x4[2]) && all(matrix[3] == _globalUniforms.testMatrix4x4[3]));
32  }
33}
34fn _skslMain(coords: vec2<f32>) -> vec4<f32> {
35  {
36    var _0_matrix: mat3x3<f32>;
37    var _1_values: vec3<f32> = vec3<f32>(3.0, 2.0, 1.0);
38    {
39      var _2_index: i32 = 0;
40      loop {
41        {
42          _0_matrix[_2_index] = vec3<f32>((_1_values.xz), _0_matrix[_2_index].y).yzx;
43          _0_matrix[_2_index].y = _1_values.y;
44          _1_values = _1_values + 3.0;
45        }
46        continuing {
47          _2_index = _2_index + i32(1);
48          break if _2_index >= 3;
49        }
50      }
51    }
52    var _skTemp0: vec4<f32>;
53    var _skTemp1: bool;
54    if (all(_0_matrix[0] == _globalUniforms.testMatrix3x3[0]) && all(_0_matrix[1] == _globalUniforms.testMatrix3x3[1]) && all(_0_matrix[2] == _globalUniforms.testMatrix3x3[2])) {
55      let _skTemp2 = test4x4_b();
56      _skTemp1 = _skTemp2;
57    } else {
58      _skTemp1 = false;
59    }
60    if _skTemp1 {
61      _skTemp0 = _globalUniforms.colorGreen;
62    } else {
63      _skTemp0 = _globalUniforms.colorRed;
64    }
65    return _skTemp0;
66  }
67}
68@fragment fn main() -> FSOut {
69  var _stageOut: FSOut;
70  _stageOut.sk_FragColor = _skslMain(/*fragcoord*/ vec2<f32>());
71  return _stageOut;
72}
73