xref: /aosp_15_r20/external/skia/tests/sksl/intrinsics/Inverse.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};
10*c8dee2aaSAndroid Build Coastguard Worker@binding(0) @group(0) var<uniform> _globalUniforms: _GlobalUniforms;
11*c8dee2aaSAndroid Build Coastguard Workerfn mat3_inverse(m: mat3x3<f32>) -> mat3x3<f32> {
12*c8dee2aaSAndroid Build Coastguard Workerlet a00 = m[0].x; let a01 = m[0].y; let a02 = m[0].z;
13*c8dee2aaSAndroid Build Coastguard Workerlet a10 = m[1].x; let a11 = m[1].y; let a12 = m[1].z;
14*c8dee2aaSAndroid Build Coastguard Workerlet a20 = m[2].x; let a21 = m[2].y; let a22 = m[2].z;
15*c8dee2aaSAndroid Build Coastguard Workerlet b01 =  a22*a11 - a12*a21;
16*c8dee2aaSAndroid Build Coastguard Workerlet b11 = -a22*a10 + a12*a20;
17*c8dee2aaSAndroid Build Coastguard Workerlet b21 =  a21*a10 - a11*a20;
18*c8dee2aaSAndroid Build Coastguard Workerlet det = a00*b01 + a01*b11 + a02*b21;
19*c8dee2aaSAndroid Build Coastguard Workerreturn mat3x3<f32>(b01, (-a22*a01 + a02*a21), ( a12*a01 - a02*a11),
20*c8dee2aaSAndroid Build Coastguard Workerb11, ( a22*a00 - a02*a20), (-a12*a00 + a02*a10),
21*c8dee2aaSAndroid Build Coastguard Workerb21, (-a21*a00 + a01*a20), ( a11*a00 - a01*a10)) * (1/det);
22*c8dee2aaSAndroid Build Coastguard Worker}
23*c8dee2aaSAndroid Build Coastguard Workerfn mat2_inverse(m: mat2x2<f32>) -> mat2x2<f32> {
24*c8dee2aaSAndroid Build Coastguard Workerreturn mat2x2<f32>(m[1].y, -m[0].y, -m[1].x, m[0].x) * (1/determinant(m));
25*c8dee2aaSAndroid Build Coastguard Worker}
26*c8dee2aaSAndroid Build Coastguard Workerfn mat4_inverse(m: mat4x4<f32>) -> mat4x4<f32>{
27*c8dee2aaSAndroid Build Coastguard Workerlet a00 = m[0].x; let a01 = m[0].y; let a02 = m[0].z; let a03 = m[0].w;
28*c8dee2aaSAndroid Build Coastguard Workerlet a10 = m[1].x; let a11 = m[1].y; let a12 = m[1].z; let a13 = m[1].w;
29*c8dee2aaSAndroid Build Coastguard Workerlet a20 = m[2].x; let a21 = m[2].y; let a22 = m[2].z; let a23 = m[2].w;
30*c8dee2aaSAndroid Build Coastguard Workerlet a30 = m[3].x; let a31 = m[3].y; let a32 = m[3].z; let a33 = m[3].w;
31*c8dee2aaSAndroid Build Coastguard Workerlet b00 = a00*a11 - a01*a10;
32*c8dee2aaSAndroid Build Coastguard Workerlet b01 = a00*a12 - a02*a10;
33*c8dee2aaSAndroid Build Coastguard Workerlet b02 = a00*a13 - a03*a10;
34*c8dee2aaSAndroid Build Coastguard Workerlet b03 = a01*a12 - a02*a11;
35*c8dee2aaSAndroid Build Coastguard Workerlet b04 = a01*a13 - a03*a11;
36*c8dee2aaSAndroid Build Coastguard Workerlet b05 = a02*a13 - a03*a12;
37*c8dee2aaSAndroid Build Coastguard Workerlet b06 = a20*a31 - a21*a30;
38*c8dee2aaSAndroid Build Coastguard Workerlet b07 = a20*a32 - a22*a30;
39*c8dee2aaSAndroid Build Coastguard Workerlet b08 = a20*a33 - a23*a30;
40*c8dee2aaSAndroid Build Coastguard Workerlet b09 = a21*a32 - a22*a31;
41*c8dee2aaSAndroid Build Coastguard Workerlet b10 = a21*a33 - a23*a31;
42*c8dee2aaSAndroid Build Coastguard Workerlet b11 = a22*a33 - a23*a32;
43*c8dee2aaSAndroid Build Coastguard Workerlet det = b00*b11 - b01*b10 + b02*b09 + b03*b08 - b04*b07 + b05*b06;
44*c8dee2aaSAndroid Build Coastguard Workerreturn mat4x4<f32>(a11*b11 - a12*b10 + a13*b09,
45*c8dee2aaSAndroid Build Coastguard Workera02*b10 - a01*b11 - a03*b09,
46*c8dee2aaSAndroid Build Coastguard Workera31*b05 - a32*b04 + a33*b03,
47*c8dee2aaSAndroid Build Coastguard Workera22*b04 - a21*b05 - a23*b03,
48*c8dee2aaSAndroid Build Coastguard Workera12*b08 - a10*b11 - a13*b07,
49*c8dee2aaSAndroid Build Coastguard Workera00*b11 - a02*b08 + a03*b07,
50*c8dee2aaSAndroid Build Coastguard Workera32*b02 - a30*b05 - a33*b01,
51*c8dee2aaSAndroid Build Coastguard Workera20*b05 - a22*b02 + a23*b01,
52*c8dee2aaSAndroid Build Coastguard Workera10*b10 - a11*b08 + a13*b06,
53*c8dee2aaSAndroid Build Coastguard Workera01*b08 - a00*b10 - a03*b06,
54*c8dee2aaSAndroid Build Coastguard Workera30*b04 - a31*b02 + a33*b00,
55*c8dee2aaSAndroid Build Coastguard Workera21*b02 - a20*b04 - a23*b00,
56*c8dee2aaSAndroid Build Coastguard Workera11*b07 - a10*b09 - a12*b06,
57*c8dee2aaSAndroid Build Coastguard Workera00*b09 - a01*b07 + a02*b06,
58*c8dee2aaSAndroid Build Coastguard Workera31*b01 - a30*b03 - a32*b00,
59*c8dee2aaSAndroid Build Coastguard Workera20*b03 - a21*b01 + a22*b00) * (1/det);
60*c8dee2aaSAndroid Build Coastguard Worker}
61*c8dee2aaSAndroid Build Coastguard Workerfn _skslMain(xy: vec2<f32>) -> vec4<f32> {
62*c8dee2aaSAndroid Build Coastguard Worker  {
63*c8dee2aaSAndroid Build Coastguard Worker    const matrix2x2: mat2x2<f32> = mat2x2<f32>(1.0, 2.0, 3.0, 4.0);
64*c8dee2aaSAndroid Build Coastguard Worker    const inv2x2: mat2x2<f32> = mat2x2<f32>(-2.0, 1.0, 1.5, -0.5);
65*c8dee2aaSAndroid Build Coastguard Worker    const inv3x3: mat3x3<f32> = mat3x3<f32>(-24.0, 18.0, 5.0, 20.0, -15.0, -4.0, -5.0, 4.0, 1.0);
66*c8dee2aaSAndroid Build Coastguard Worker    const inv4x4: mat4x4<f32> = mat4x4<f32>(-2.0, -0.5, 1.0, 0.5, 1.0, 0.5, 0.0, -0.5, -8.0, -1.0, 2.0, 2.0, 3.0, 0.5, -1.0, -0.5);
67*c8dee2aaSAndroid Build Coastguard Worker    let Zero: f32 = f32(_globalUniforms.colorGreen.z);
68*c8dee2aaSAndroid Build Coastguard Worker    const _skTemp0 = mat2x2<f32>(-2.0, 1.0, 1.5, -0.5);
69*c8dee2aaSAndroid Build Coastguard Worker    const _skTemp1 = mat3x3<f32>(-24.0, 18.0, 5.0, 20.0, -15.0, -4.0, -5.0, 4.0, 1.0);
70*c8dee2aaSAndroid Build Coastguard Worker    const _skTemp2 = mat4x4<f32>(-2.0, -0.5, 1.0, 0.5, 1.0, 0.5, 0.0, -0.5, -8.0, -1.0, 2.0, 2.0, 3.0, 0.5, -1.0, -0.5);
71*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp3 = mat3x3<f32>(1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0);
72*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp4 = mat3_inverse(_skTemp3);
73*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp5 = _skTemp4;
74*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp6 = mat2_inverse(mat2x2<f32>(matrix2x2[0] + Zero, matrix2x2[1] + Zero));
75*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp7 = _skTemp6;
76*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp8 = mat3x3<f32>(1.0, 2.0, 3.0, 0.0, 1.0, 4.0, 5.0, 6.0, 0.0);
77*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp9 = mat3_inverse(mat3x3<f32>(_skTemp8[0] + Zero, _skTemp8[1] + Zero, _skTemp8[2] + Zero));
78*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp10 = _skTemp9;
79*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp11 = mat4x4<f32>(1.0, 0.0, 0.0, 1.0, 0.0, 2.0, 1.0, 2.0, 2.0, 1.0, 0.0, 1.0, 2.0, 0.0, 1.0, 4.0);
80*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp12 = mat4_inverse(mat4x4<f32>(_skTemp11[0] + Zero, _skTemp11[1] + Zero, _skTemp11[2] + Zero, _skTemp11[3] + Zero));
81*c8dee2aaSAndroid Build Coastguard Worker    let _skTemp13 = _skTemp12;
82*c8dee2aaSAndroid Build Coastguard Worker    return select(_globalUniforms.colorRed, _globalUniforms.colorGreen, vec4<bool>(((((((all(_skTemp0[0] == inv2x2[0]) && all(_skTemp0[1] == inv2x2[1])) && (all(_skTemp1[0] == inv3x3[0]) && all(_skTemp1[1] == inv3x3[1]) && all(_skTemp1[2] == inv3x3[2]))) && (all(_skTemp2[0] == inv4x4[0]) && all(_skTemp2[1] == inv4x4[1]) && all(_skTemp2[2] == inv4x4[2]) && all(_skTemp2[3] == inv4x4[3]))) && (any(_skTemp5[0] != inv3x3[0]) || any(_skTemp5[1] != inv3x3[1]) || any(_skTemp5[2] != inv3x3[2]))) && (all(_skTemp7[0] == inv2x2[0]) && all(_skTemp7[1] == inv2x2[1]))) && (all(_skTemp10[0] == inv3x3[0]) && all(_skTemp10[1] == inv3x3[1]) && all(_skTemp10[2] == inv3x3[2]))) && (all(_skTemp13[0] == inv4x4[0]) && all(_skTemp13[1] == inv4x4[1]) && all(_skTemp13[2] == inv4x4[2]) && all(_skTemp13[3] == inv4x4[3]))));
83*c8dee2aaSAndroid Build Coastguard Worker  }
84*c8dee2aaSAndroid Build Coastguard Worker}
85*c8dee2aaSAndroid Build Coastguard Worker@fragment fn main() -> FSOut {
86*c8dee2aaSAndroid Build Coastguard Worker  var _stageOut: FSOut;
87*c8dee2aaSAndroid Build Coastguard Worker  _stageOut.sk_FragColor = _skslMain(/*fragcoord*/ vec2<f32>());
88*c8dee2aaSAndroid Build Coastguard Worker  return _stageOut;
89*c8dee2aaSAndroid Build Coastguard Worker}
90