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