1cb93a386Sopenharmony_ci#include <metal_stdlib>
2cb93a386Sopenharmony_ci#include <simd/simd.h>
3cb93a386Sopenharmony_ciusing namespace metal;
4cb93a386Sopenharmony_cistruct Uniforms {
5cb93a386Sopenharmony_ci    half4 colorGreen;
6cb93a386Sopenharmony_ci    half4 colorRed;
7cb93a386Sopenharmony_ci};
8cb93a386Sopenharmony_cistruct Inputs {
9cb93a386Sopenharmony_ci};
10cb93a386Sopenharmony_cistruct Outputs {
11cb93a386Sopenharmony_ci    half4 sk_FragColor [[color(0)]];
12cb93a386Sopenharmony_ci};
13cb93a386Sopenharmony_ci
14cb93a386Sopenharmony_cithread bool operator==(const half2x2 left, const half2x2 right);
15cb93a386Sopenharmony_cithread bool operator!=(const half2x2 left, const half2x2 right);
16cb93a386Sopenharmony_ci
17cb93a386Sopenharmony_cithread bool operator==(const half3x3 left, const half3x3 right);
18cb93a386Sopenharmony_cithread bool operator!=(const half3x3 left, const half3x3 right);
19cb93a386Sopenharmony_ci
20cb93a386Sopenharmony_cithread bool operator==(const half4x4 left, const half4x4 right);
21cb93a386Sopenharmony_cithread bool operator!=(const half4x4 left, const half4x4 right);
22cb93a386Sopenharmony_cithread bool operator==(const half2x2 left, const half2x2 right) {
23cb93a386Sopenharmony_ci    return all(left[0] == right[0]) &&
24cb93a386Sopenharmony_ci           all(left[1] == right[1]);
25cb93a386Sopenharmony_ci}
26cb93a386Sopenharmony_cithread bool operator!=(const half2x2 left, const half2x2 right) {
27cb93a386Sopenharmony_ci    return !(left == right);
28cb93a386Sopenharmony_ci}
29cb93a386Sopenharmony_cithread bool operator==(const half3x3 left, const half3x3 right) {
30cb93a386Sopenharmony_ci    return all(left[0] == right[0]) &&
31cb93a386Sopenharmony_ci           all(left[1] == right[1]) &&
32cb93a386Sopenharmony_ci           all(left[2] == right[2]);
33cb93a386Sopenharmony_ci}
34cb93a386Sopenharmony_cithread bool operator!=(const half3x3 left, const half3x3 right) {
35cb93a386Sopenharmony_ci    return !(left == right);
36cb93a386Sopenharmony_ci}
37cb93a386Sopenharmony_cithread bool operator==(const half4x4 left, const half4x4 right) {
38cb93a386Sopenharmony_ci    return all(left[0] == right[0]) &&
39cb93a386Sopenharmony_ci           all(left[1] == right[1]) &&
40cb93a386Sopenharmony_ci           all(left[2] == right[2]) &&
41cb93a386Sopenharmony_ci           all(left[3] == right[3]);
42cb93a386Sopenharmony_ci}
43cb93a386Sopenharmony_cithread bool operator!=(const half4x4 left, const half4x4 right) {
44cb93a386Sopenharmony_ci    return !(left == right);
45cb93a386Sopenharmony_ci}
46cb93a386Sopenharmony_ci
47cb93a386Sopenharmony_citemplate <typename T>
48cb93a386Sopenharmony_cimatrix<T, 3, 3> mat3_inverse(matrix<T, 3, 3> m) {
49cb93a386Sopenharmony_ci    T a00 = m[0][0], a01 = m[0][1], a02 = m[0][2];
50cb93a386Sopenharmony_ci    T a10 = m[1][0], a11 = m[1][1], a12 = m[1][2];
51cb93a386Sopenharmony_ci    T a20 = m[2][0], a21 = m[2][1], a22 = m[2][2];
52cb93a386Sopenharmony_ci    T b01 =  a22*a11 - a12*a21;
53cb93a386Sopenharmony_ci    T b11 = -a22*a10 + a12*a20;
54cb93a386Sopenharmony_ci    T b21 =  a21*a10 - a11*a20;
55cb93a386Sopenharmony_ci    T det = a00*b01 + a01*b11 + a02*b21;
56cb93a386Sopenharmony_ci    return matrix<T, 3, 3>(b01, (-a22*a01 + a02*a21), ( a12*a01 - a02*a11),
57cb93a386Sopenharmony_ci                           b11, ( a22*a00 - a02*a20), (-a12*a00 + a02*a10),
58cb93a386Sopenharmony_ci                           b21, (-a21*a00 + a01*a20), ( a11*a00 - a01*a10)) * (1/det);
59cb93a386Sopenharmony_ci}
60cb93a386Sopenharmony_cifragment Outputs fragmentMain(Inputs _in [[stage_in]], constant Uniforms& _uniforms [[buffer(0)]], bool _frontFacing [[front_facing]], float4 _fragCoord [[position]]) {
61cb93a386Sopenharmony_ci    Outputs _out;
62cb93a386Sopenharmony_ci    (void)_out;
63cb93a386Sopenharmony_ci    half2x2 inv2x2 = half2x2(half2(-2.0h, 1.0h), half2(1.5h, -0.5h));
64cb93a386Sopenharmony_ci    half3x3 inv3x3 = half3x3(half3(-24.0h, 18.0h, 5.0h), half3(20.0h, -15.0h, -4.0h), half3(-5.0h, 4.0h, 1.0h));
65cb93a386Sopenharmony_ci    half4x4 inv4x4 = half4x4(half4(-2.0h, -0.5h, 1.0h, 0.5h), half4(1.0h, 0.5h, 0.0h, -0.5h), half4(-8.0h, -1.0h, 2.0h, 2.0h), half4(3.0h, 0.5h, -1.0h, -0.5h));
66cb93a386Sopenharmony_ci    _out.sk_FragColor = ((half2x2(half2(-2.0h, 1.0h), half2(1.5h, -0.5h)) == inv2x2 && half3x3(half3(-24.0h, 18.0h, 5.0h), half3(20.0h, -15.0h, -4.0h), half3(-5.0h, 4.0h, 1.0h)) == inv3x3) && half4x4(half4(-2.0h, -0.5h, 1.0h, 0.5h), half4(1.0h, 0.5h, 0.0h, -0.5h), half4(-8.0h, -1.0h, 2.0h, 2.0h), half4(3.0h, 0.5h, -1.0h, -0.5h)) == inv4x4) && mat3_inverse(half3x3(half3(1.0h, 2.0h, 3.0h), half3(4.0h, 5.0h, 6.0h), half3(7.0h, 8.0h, 9.0h))) != inv3x3 ? _uniforms.colorGreen : _uniforms.colorRed;
67cb93a386Sopenharmony_ci    return _out;
68cb93a386Sopenharmony_ci}
69